#pragma once

#include <llarp/handlers/tun.hpp>
#include <llarp/net/net.hpp>
#include <llarp/config/config.hpp>
#include <unordered_map>

#include "endpoint.hpp"

/*
  TODO:
    - if we are only using one local endpoint, does this need to change to account for that?
*/

namespace llarp::service
{
  /// holds all the hidden service endpoints we own
  /// this should be refactored (removed entirely...?) now that lokinet
  ///       only supports one endpoint per instance
  struct Context
  {
    explicit Context(Router* r);
    ~Context();

    void
    Tick(llarp_time_t now);

    /// stop all held services
    bool
    StopAll();

    util::StatusObject
    ExtractStatus() const;

    bool
    hasEndpoints();

    /// function visitor returns false to prematurely break iteration
    void
    ForEachService(std::function<bool(const std::string&, const Endpoint_ptr&)> visit) const;

    /// Pumps the hidden service endpoints, called during Router::PumpLL
    void
    Pump();

    /// add endpoint via config
    void
    AddEndpoint(const Config& conf, bool autostart = false);

    /// inject endpoint instance
    void
    InjectEndpoint(std::string name, std::shared_ptr<Endpoint> ep);

    /// stop and remove an endpoint by name
    /// return false if we don't have the hidden service with that name
    bool
    RemoveEndpoint(const std::string& name);

    Endpoint_ptr
    GetEndpointByName(const std::string& name) const;

    Endpoint_ptr
    GetDefault() const
    {
      return GetEndpointByName("default");
    }

    bool
    StartAll();

   private:
    Router* const m_Router;
    std::unordered_map<std::string, std::shared_ptr<Endpoint>> m_Endpoints;
    std::list<std::shared_ptr<Endpoint>> m_Stopped;
  };
}  // namespace llarp::service