#include "client.hpp"
#include "definitions.hpp"
#include "server.hpp"
#include <cstdio>
#include <cstring>
#include <unistd.h>

Client::Client() : sock(), serverAddr() {
  sock = socket(AF_INET, SOCK_STREAM, 0);
  serverAddr.sin_family = AF_INET;
  serverAddr.sin_port = htons(DEFAULT_PORT);
  inet_aton(DEFAULT_IP, &serverAddr.sin_addr);
}

Client::Client(const std::string &addr, const int &port) {
  sock = socket(AF_INET, SOCK_STREAM, 0);
  serverAddr.sin_family = AF_INET;
  serverAddr.sin_port = htons(port);
  inet_aton(addr.c_str(), &serverAddr.sin_addr);
}

int Client::connectToServer() {
  int rc = connect(sock, (struct sockaddr *)&serverAddr, sizeof(serverAddr));
  if (rc < 0) {
    perror("connect() failed");
  }
  return rc;
}

int Client::disconnectFromServer() {
  int rc = close(sock);
  if (rc < 0) {
    perror("disconnect() failed");
  }
  return rc;
}

uint16_t Client::sendMessage(const std::string &msg) {
  auto bytes = send(sock, msg.c_str(), strlen(msg.c_str()) + 1, 0);
  return bytes;
}

int Client::receiveMessage() {
  int nbytesrecv = recv(sock, inputBuffer, INPUT_BUFFER_SIZE, 0);
  if (nbytesrecv <= 0) {
    if (nbytesrecv == 0) {
      disconnectFromServer();
      return -1;
    } else {
      perror("recv() failed");
    }
    close(sock);
    return -1;
  }
  std::string msg(inputBuffer);
  onReceiveCB(sock, msg);
  bzero(&inputBuffer, INPUT_BUFFER_SIZE);
  return nbytesrecv;
}

void Client::onReceive(receiveCallbackType fn) {
  if (onReceiveCB != nullptr)
    onReceiveCB = fn;
}