#include "Server.hpp"
#include "Request.hpp"
#include "Response.hpp"

#include <sys/socket.h>
#include <netinet/in.h>
#include <stdexcept>
#include <unistd.h>
#include <iostream>

namespace eRPC
{
  Server::Server(int port) : sockfd(-1), running(false)
  {
    if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
    {
      throw std::runtime_error("Failed to create socket");
    }

    sockaddr_in serv_addr;
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = INADDR_ANY;
    serv_addr.sin_port = htons(port);

    if (bind(sockfd, (sockaddr *)&serv_addr, sizeof(serv_addr)) < 0)
    {
      throw std::runtime_error("Failed to bind socket");
    }

    if (listen(sockfd, 5) < 0)
    {
      throw std::runtime_error("Failed to listen on socket");
    }

    std::cout << "Server listening on port " << port << std::endl;
  }

  Server::~Server()
  {
    stop();
  }

  void Server::start()
  {
    if (running)
    {
      return;
    }

    running = true;

    while (running)
    {
      std::cout << "Waiting for connection" << std::endl;

      int connfd = accept(sockfd, (sockaddr *)NULL, NULL);

      char buffer[1024] = {0};
      read(connfd, buffer, 1024);
      Request request(buffer);
      std::cout << "Received request:\n"
                << request.serialize() << std::endl;

      Response response;
      if (methods.find(request.getMethod()) != methods.end())
      {
        std::function<void *()> method = methods[request.getMethod()];
        method();

        response = Response(
            request.getMsgid(),
            true,
            "RESULTS HERE");
      }
      else
      {
        response = Response(
            request.getMsgid(),
            false,
            "Method \"" + request.getMethod() + "\" not found");
      }

      std::string serialized = response.serialize();
      write(connfd, serialized.c_str(), serialized.size() + 1);

      close(connfd);
    }

    close(sockfd);
  }

  void Server::stop()
  {
    running = false;
  }

  void Server::bindMethod(std::string method, std::function<void *()> callback)
  {
    methods[method] = callback;
  }
}