#pragma once

#include <cstdint>

#include <string>

#include "envoy/config/metrics/v3/stats.pb.h"

#include "envoy/stats/histogram.h"

#include "envoy/stats/stats.h"

#include "envoy/stats/store.h"

#include "source/common/common/matchers.h"

#include "source/common/common/non_copyable.h"

#include "source/common/stats/metric_impl.h"

#include "circllhist.h"

namespace Envoy {

namespace Stats {

class HistogramSettingsImpl : public HistogramSettings {

public:

  HistogramSettingsImpl(const envoy::config::metrics::v3::StatsConfig& config,

                        Server::Configuration::CommonFactoryContext& context);

  // HistogramSettings

  const ConstSupportedBuckets& buckets(absl::string_view stat_name) const override;

  absl::optional<uint32_t> bins(absl::string_view stat_name) const override;

  static ConstSupportedBuckets& defaultBuckets();

private:

  struct Config {

    Matchers::StringMatcherImpl matcher_;

    absl::optional<ConstSupportedBuckets> buckets_;

    absl::optional<uint32_t> bins_;

  };

  const std::vector<Config> configs_{};

};

/**

 * Implementation of HistogramStatistics for circllhist.

 */

class HistogramStatisticsImpl final : public HistogramStatistics, NonCopyable {

public:

  HistogramStatisticsImpl();

  /**

   * HistogramStatisticsImpl object is constructed using the passed in histogram.

   * @param histogram_ptr pointer to the histogram for which stats will be calculated. This pointer

   * will not be retained.

   */

  HistogramStatisticsImpl(

      const histogram_t* histogram_ptr, Histogram::Unit unit = Histogram::Unit::Unspecified,

      ConstSupportedBuckets& supported_buckets = HistogramSettingsImpl::defaultBuckets());

  void refresh(const histogram_t* new_histogram_ptr);

  // HistogramStatistics

  std::string quantileSummary() const override;

  std::string bucketSummary() const override;

  const std::vector<double>& supportedQuantiles() const final;

  std::vector<uint64_t> computeDisjointBuckets() const override;

private:

  ConstSupportedBuckets& supported_buckets_;

  std::vector<double> computed_quantiles_;

  std::vector<uint64_t> computed_buckets_;

  uint64_t sample_count_{0};

  uint64_t out_of_bound_count_{0};

  double sample_sum_{0};

  const Histogram::Unit unit_{Histogram::Unit::Unspecified};

};

class HistogramImplHelper : public MetricImpl<Histogram> {

public:

  HistogramImplHelper(StatName name, StatName tag_extracted_name,

                      const StatNameTagVector& stat_name_tags, SymbolTable& symbol_table)

  // RefcountInterface

private:

  RefcountHelper refcount_helper_;

};

/**

 * Histogram implementation for the heap.

 */

class HistogramImpl : public HistogramImplHelper {

public:

  HistogramImpl(StatName name, Unit unit, Store& parent, StatName tag_extracted_name,

                const StatNameTagVector& stat_name_tags)

    // We must explicitly free the StatName here in order to supply the

    // SymbolTable reference. An RAII alternative would be to store a

    // reference to the SymbolTable in MetricImpl, which would cost 8 bytes

    // per stat.

  // Stats::Histogram

private:

  Unit unit_;

  // This is used for delivering the histogram data to sinks.

  Store& parent_;

};

/**

 * Null histogram implementation.

 * No-ops on all calls and requires no underlying metric or data.

 */

class NullHistogramImpl : public HistogramImplHelper {

public:

  explicit NullHistogramImpl(SymbolTable& symbol_table)

private:

  SymbolTable& symbol_table_;

};

} // namespace Stats

} // namespace Envoy