#ifndef INCLUDE_CCAPI_CPP_CCAPI_WS_CONNECTION_H_
#define INCLUDE_CCAPI_CPP_CCAPI_WS_CONNECTION_H_
#ifndef CCAPI_USE_BOOST_BEAST_WEBSOCKET
#include <string>

#include "ccapi_cpp/ccapi_logger.h"
#include "ccapi_cpp/ccapi_subscription.h"
namespace wspp = websocketpp;
namespace ccapi {
/**
 * This class represents a TCP socket connection for the websocket API.
 */
class WsConnection CCAPI_FINAL {
 public:
  WsConnection(std::string url, std::string group, std::vector<Subscription> subscriptionList, std::map<std::string, std::string> credential)
      : url(url), group(group), subscriptionList(subscriptionList), credential(credential) {
    this->assignDummyId();
  }
  WsConnection() {}
  void assignDummyId() {
    this->id = this->url + "||" + this->group + "||" + ccapi::toString(this->subscriptionList) + "||" + ccapi::toString(this->credential);
    this->hdl.reset();
  }
  std::string toString() const {
    std::map<std::string, std::string> shortCredential;
    for (const auto& x : credential) {
      shortCredential.insert(std::make_pair(x.first, UtilString::firstNCharacter(x.second, CCAPI_CREDENTIAL_DISPLAY_LENGTH)));
    }
    std::string output = "WsConnection [id = " + id + ", url = " + url + ", group = " + group + ", subscriptionList = " + ccapi::toString(subscriptionList) +
                         ", credential = " + ccapi::toString(shortCredential) + ", status = " + statusToString(status) +
                         ", headers = " + ccapi::toString(headers) + "]";
    return output;
  }
  enum class Status {
    UNKNOWN,
    CONNECTING,
    OPEN,
    FAILED,
    CLOSING,
    CLOSED,
  };
  static std::string statusToString(Status status) {
    std::string output;
    switch (status) {
      case Status::UNKNOWN:
        output = "UNKNOWN";
        break;
      case Status::CONNECTING:
        output = "CONNECTING";
        break;
      case Status::OPEN:
        output = "OPEN";
        break;
      case Status::FAILED:
        output = "FAILED";
        break;
      case Status::CLOSING:
        output = "CLOSING";
        break;
      case Status::CLOSED:
        output = "CLOSED";
        break;
      default:
        CCAPI_LOGGER_FATAL(CCAPI_UNSUPPORTED_VALUE);
    }
    return output;
  }
  std::string id;
  std::string url;
  std::string group;
  std::vector<Subscription> subscriptionList;
  Status status{Status::UNKNOWN};
  wspp::connection_hdl hdl = wspp::lib::weak_ptr<void>();
  std::map<std::string, std::string> headers;
  std::map<std::string, std::string> credential;
};
} /* namespace ccapi */
#else
#include <string>

#include "ccapi_cpp/ccapi_logger.h"
#include "ccapi_cpp/ccapi_subscription.h"
namespace ccapi {
/**
 * This class represents a TCP socket connection for the websocket API.
 */
class WsConnection CCAPI_FINAL {
 public:
  WsConnection(std::string url, std::string group, std::vector<Subscription> subscriptionList, std::map<std::string, std::string> credential,
               std::shared_ptr<beast::websocket::stream<beast::ssl_stream<beast::tcp_stream> > > streamPtr)
      : url(url), group(group), subscriptionList(subscriptionList), credential(credential), streamPtr(streamPtr) {
    this->id = this->url + "||" + this->group + "||" + ccapi::toString(this->subscriptionList) + "||" + ccapi::toString(this->credential);
    this->correlationIdList.reserve(subscriptionList.size());
    std::transform(subscriptionList.cbegin(), subscriptionList.cend(), this->correlationIdList.begin(),
                   [](Subscription subscription) { return subscription.getCorrelationId(); });
    auto splitted1 = UtilString::split(url, "://");
    auto foundSlash = splitted1[1].find_first_of('/');
    auto foundQuestionMark = splitted1[1].find_first_of('?');
    if (foundSlash == std::string::npos && foundQuestionMark == std::string::npos) {
      this->path = "/";
    } else if (foundSlash == std::string::npos && foundQuestionMark != std::string::npos) {
      this->path = "/" + splitted1[1].substr(foundQuestionMark);
    } else if (foundSlash != std::string::npos && foundQuestionMark == std::string::npos) {
      this->path = splitted1[1].substr(foundSlash);
    } else {
      this->path = splitted1[1].substr(foundSlash);
    }
  }
  WsConnection() {}
  std::string toString() const {
    std::map<std::string, std::string> shortCredential;
    for (const auto& x : credential) {
      shortCredential.insert(std::make_pair(x.first, UtilString::firstNCharacter(x.second, CCAPI_CREDENTIAL_DISPLAY_LENGTH)));
    }
    std::ostringstream oss;
    oss << streamPtr;
    std::string output = "WsConnection [id = " + id + ", url = " + url + ", group = " + group + ", subscriptionList = " + ccapi::toString(subscriptionList) +
                         ", credential = " + ccapi::toString(shortCredential) + ", status = " + statusToString(status) +
                         ", headers = " + ccapi::toString(headers) + ", streamPtr = " + oss.str() + "]";
    return output;
  }
  enum class Status {
    UNKNOWN,
    CONNECTING,
    OPEN,
    FAILED,
    CLOSING,
    CLOSED,
  };
  static std::string statusToString(Status status) {
    std::string output;
    switch (status) {
      case Status::UNKNOWN:
        output = "UNKNOWN";
        break;
      case Status::CONNECTING:
        output = "CONNECTING";
        break;
      case Status::OPEN:
        output = "OPEN";
        break;
      case Status::FAILED:
        output = "FAILED";
        break;
      case Status::CLOSING:
        output = "CLOSING";
        break;
      case Status::CLOSED:
        output = "CLOSED";
        break;
      default:
        CCAPI_LOGGER_FATAL(CCAPI_UNSUPPORTED_VALUE);
    }
    return output;
  }
  std::string id;
  std::string url;
  std::string group;
  std::vector<Subscription> subscriptionList;
  std::vector<std::string> correlationIdList;
  Status status{Status::UNKNOWN};
  std::map<std::string, std::string> headers;
  std::map<std::string, std::string> credential;
  std::shared_ptr<beast::websocket::stream<beast::ssl_stream<beast::tcp_stream> > > streamPtr;
  beast::websocket::close_code remoteCloseCode;
  beast::websocket::close_reason remoteCloseReason;
  std::string hostHttpHeaderValue;
  std::string path;
};
} /* namespace ccapi */
#endif
#endif  // INCLUDE_CCAPI_CPP_CCAPI_WS_CONNECTION_H_