#pragma once

#include "mgm/Namespace.hh"
#include "common/AssistedThread.hh"
#include "common/Mapping.hh"
#include "common/Logging.hh"
#include "GrpcRestGwInterface.hh"

#ifdef EOS_GRPC_GATEWAY
#include <grpc++/grpc++.h>
#endif // EOS_GRPC_GATEWAY

EOSMGMNAMESPACE_BEGIN

/**
 * @file   GrpcRestGwServer.hh
 *
 * @brief  This class implements a GRPC server with a grpc-gateway
           for accessing all EOS console commands through HTTP requests
 */
class GrpcRestGwServer: public eos::common::LogId
{
private:
  int mPort;
  bool mSSL;
  std::string mSSLCert;
  std::string mSSLKey;
  std::string mSSLCa;
  std::string mSSLCertFile;
  std::string mSSLKeyFile;
  std::string mSSLCaFile;
  AssistedThread mThread; // Thread running GRPC service

#ifdef EOS_GRPC_GATEWAY
  std::unique_ptr<grpc::Server> mRestGwServer;
#endif /// EOS_GRPC_GATEWAY

public:

  /* Default Constructor - enabling port 50054 by default
   */
  GrpcRestGwServer(int port = 50054) : mPort(port), mSSL(false) {  }

  ~GrpcRestGwServer()
  {
#ifdef EOS_GRPC_GATEWAY

    if (mRestGwServer) {
      mRestGwServer->Shutdown();
    }

#endif // EOS_GRPC_GATEWAY
    mThread.join();
  }

  /* Run function */
  void Run(ThreadAssistant& assistant) noexcept;

  /* Startup function */
  void Start()
  {
    mThread.reset(&GrpcRestGwServer::Run, this);
  }

#ifdef EOS_GRPC_GATEWAY

  /* return client DN*/
  static std::string DN(grpc::ServerContext* context);
  /* return client IP*/
  static std::string IP(grpc::ServerContext* context, std::string* id = 0,
                        std::string* port = 0);
  /* return VID for a given call */
  static void Vid(grpc::ServerContext* context,
                  eos::common::VirtualIdentity& vid);

#endif // EOS_GRPC_GATEWAY
};

EOSMGMNAMESPACE_END