#include <httq/AbstractServer.h>
#include <httq/HttpRequest.h>
#include <httq/Logger.h>

#include <QTcpServer>
#include <QTcpSocket>
#include <QWebSocketServer>


namespace httq
{
AbstractServer::AbstractServer(QObject *parent)
  : QObject(parent)
  , mSvr(new QTcpServer(this))
  , mLoggerFactory(nullptr)
  , mLogger(nullptr)
{}


LoggerFactory *AbstractServer::createLoggerFactory()
{
  return new LoggerFactory(this);
}


LoggerFactory *AbstractServer::getLoggerFactory()
{
  if (mLoggerFactory == nullptr)
    mLoggerFactory = createLoggerFactory();
  return mLoggerFactory;
}


void AbstractServer::slotNewConnection()
{
  mLogger->debug(QStringLiteral("newConnection"));

  while (QTcpSocket *cli = mSvr->nextPendingConnection())
  {
    mLogger->debug(QStringLiteral("nextPendingConnection"));

    if (cli == nullptr)
    {
      mLogger->debug(QStringLiteral("invalid tcp client"));
      return;
    }

    HttpRequest *req = new HttpRequest(cli, this);

    connect(req, &HttpRequest::signalUpgrade,
            this, [this, req](QTcpSocket *sock)
    {
      mLogger->debug(QStringLiteral("signalUpgrade"));

      QWebSocketServer *wsSvr = webSocketServer();

      if (wsSvr == nullptr)
      {
        mLogger->debug(QStringLiteral("no websocket server available for websocket upgrade"));
        sock->close();
        sock->deleteLater();
        return;
      }

      wsSvr->handleConnection(sock);
      emit sock->readyRead(); 
    });

    connect(req, &HttpRequest::signalReady,
            this, [this, req]()
    {
      mLogger->debug(QStringLiteral("signalReady"));
      newHttpConnection(req); 
    });
  }
}


bool AbstractServer::listen(qint16 port, const QHostAddress &host)
{
  mLoggerFactory = createLoggerFactory();
  mLogger = mLoggerFactory->createLogger(this);

  mLogger->debug(QStringLiteral("listen"));

  connect(mSvr, &QTcpServer::newConnection,
          this, &AbstractServer::slotNewConnection);

  return mSvr->listen(host, port);
}
}