#pragma once

#include <memory>

#include <tuple>

#include "envoy/extensions/http/stateful_session/envelope/v3/envelope.pb.h"

#include "envoy/http/stateful_session.h"

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

#include "source/common/http/headers.h"

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

namespace Envoy {

namespace Extensions {

namespace Http {

namespace StatefulSession {

namespace Envelope {

using EnvelopeSessionStateProto =

    envoy::extensions::http::stateful_session::envelope::v3::EnvelopeSessionState;

class EnvelopeSessionStateFactory : public Envoy::Http::SessionStateFactory,

                                    public Logger::Loggable<Logger::Id::http> {

public:

  class SessionStateImpl : public Envoy::Http::SessionState {

  public:

    SessionStateImpl(absl::optional<std::string> address,

                     const EnvelopeSessionStateFactory& factory)

    bool onUpdate(absl::string_view host_address, Envoy::Http::ResponseHeaderMap& headers) override;

  private:

    absl::optional<std::string> upstream_address_;

    const EnvelopeSessionStateFactory& factory_;

  };

  EnvelopeSessionStateFactory(const EnvelopeSessionStateProto& config);

private:

  absl::optional<std::string> parseAddress(Envoy::Http::RequestHeaderMap& headers) const;

  const Envoy::Http::LowerCaseString name_;

};

} // namespace Envelope

} // namespace StatefulSession

} // namespace Http

} // namespace Extensions

} // namespace Envoy