#pragma once

#include "envoy/config/tap/v3/common.pb.h"

#include "envoy/extensions/common/tap/v3/common.pb.h"

#include "envoy/thread_local/thread_local.h"

#include "source/extensions/common/tap/admin.h"

#include "source/extensions/common/tap/tap.h"

namespace Envoy {

namespace Extensions {

namespace Common {

namespace Tap {

/**

 * Base class for tap extension configuration. Used by all tap extensions.

 */

class ExtensionConfigBase : public ExtensionConfig, Logger::Loggable<Logger::Id::tap> {

public:

  // Extensions::Common::Tap::ExtensionConfig

  void clearTapConfig() override;

  const absl::string_view adminId() override;

  void newTapConfig(const envoy::config::tap::v3::TapConfig& proto_config,

                    Sink* admin_streamer) override;

protected:

  ExtensionConfigBase(const envoy::extensions::common::tap::v3::CommonExtensionConfig proto_config,

                      TapConfigFactoryPtr&& config_factory, OptRef<Server::Admin> admin,

                      Singleton::Manager& singleton_manager, ThreadLocal::SlotAllocator& tls,

                      Event::Dispatcher& main_thread_dispatcher);

  ~ExtensionConfigBase() override;

  // All tap configurations derive from TapConfig for type safety. In order to use a common

  // extension base class (with TLS logic, etc.) we must dynamic cast to the actual tap

  // configuration type that the extension expects (and is created by the configuration factory).

private:

  // Holds the functionality of installing a new tap config. This is the underlying method to the

  // virtual method newTapConfig.

  void installNewTap(const envoy::config::tap::v3::TapConfig& proto_config, Sink* admin_streamer);

  struct TlsFilterConfig : public ThreadLocal::ThreadLocalObject {

    TapConfigSharedPtr config_;

  };

  const envoy::extensions::common::tap::v3::CommonExtensionConfig proto_config_;

  TapConfigFactoryPtr config_factory_;

  ThreadLocal::TypedSlot<TlsFilterConfig> tls_slot_;

  AdminHandlerSharedPtr admin_handler_;

};

} // namespace Tap

} // namespace Common

} // namespace Extensions

} // namespace Envoy