#pragma once

#include <cstdint>

#include <functional>

#include <string>

#include "envoy/config/core/v3/config_source.pb.h"

#include "envoy/config/route/v3/route_components.pb.h"

#include "envoy/config/route/v3/route_components.pb.validate.h"

#include "envoy/config/subscription.h"

#include "envoy/http/codes.h"

#include "envoy/local_info/local_info.h"

#include "envoy/router/rds.h"

#include "envoy/router/route_config_update_receiver.h"

#include "envoy/server/filter_config.h"

#include "envoy/service/discovery/v3/discovery.pb.h"

#include "envoy/singleton/instance.h"

#include "envoy/stats/scope.h"

#include "envoy/thread_local/thread_local.h"

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

#include "source/common/config/subscription_base.h"

#include "source/common/init/target_impl.h"

#include "source/common/protobuf/utility.h"

#include "absl/container/node_hash_set.h"

namespace Envoy {

namespace Router {

#define ALL_VHDS_STATS(COUNTER)                                                                    \

  COUNTER(config_reload)                                                                           \

  COUNTER(update_empty)

struct VhdsStats {

  ALL_VHDS_STATS(GENERATE_COUNTER_STRUCT)

};

class VhdsSubscription : Envoy::Config::SubscriptionBase<envoy::config::route::v3::VirtualHost>,

                         Logger::Loggable<Logger::Id::router> {

public:

  static absl::StatusOr<std::unique_ptr<VhdsSubscription>>

  createVhdsSubscription(RouteConfigUpdatePtr& config_update_info,

                         Server::Configuration::ServerFactoryContext& factory_context,

                         const std::string& stat_prefix,

                         Rds::RouteConfigProvider* route_config_provider);

  ~VhdsSubscription() override { init_target_.ready(); }

  void registerInitTargetWithInitManager(Init::Manager& m) { m.add(init_target_); }

  void updateOnDemand(const std::string& with_route_config_name_prefix);

  static std::string domainNameToAlias(const std::string& route_config_name,

                                       const std::string& domain) {

    return route_config_name + "/" + domain;

  }

  static std::string aliasToDomainName(const std::string& alias) {

    const auto pos = alias.find_last_of('/');

    return pos == std::string::npos ? alias : alias.substr(pos + 1);

  }

private:

  VhdsSubscription(RouteConfigUpdatePtr& config_update_info,

                   Server::Configuration::ServerFactoryContext& factory_context,

                   const std::string& stat_prefix, Rds::RouteConfigProvider* route_config_provider);

  // Config::SubscriptionCallbacks

  absl::Status onConfigUpdate(const std::vector<Envoy::Config::DecodedResourceRef>&,

                              const std::string&) override {

    return absl::OkStatus();

  }

  absl::Status onConfigUpdate(const std::vector<Envoy::Config::DecodedResourceRef>&,

                              const Protobuf::RepeatedPtrField<std::string>&,

                              const std::string&) override;

  void onConfigUpdateFailed(Envoy::Config::ConfigUpdateFailureReason reason,

                            const EnvoyException* e) override;

  RouteConfigUpdatePtr& config_update_info_;

  Stats::ScopeSharedPtr scope_;

  VhdsStats stats_;

  Envoy::Config::SubscriptionPtr subscription_;

  Init::TargetImpl init_target_;

  Rds::RouteConfigProvider* route_config_provider_;

};

using VhdsSubscriptionPtr = std::unique_ptr<VhdsSubscription>;

} // namespace Router

} // namespace Envoy