#include "client.hpp"
#include "definitions.hpp"
#include "server.hpp"
#include <array>
#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 proto::Message &msg) {
  uint8_t buf[msg.ByteSizeLong()];
  msg.SerializeToArray(buf, msg.ByteSizeLong());
  auto bytes = send(sock, buf, msg.ByteSizeLong(), 0);
  return bytes;
}

int Client::receiveMessage() {
  int nbytesrecv = recv(sock, inputBuffer, INPUT_BUFFER_SIZE, 0);
  if (nbytesrecv <= 0) {
    disconnectFromServer();
    perror("recv() failed");
    return -1;
  }
  assert(onReceiveCB != nullptr);
  onReceiveCB(sock, inputBuffer);
  bzero(&inputBuffer, INPUT_BUFFER_SIZE);
  return 0;
}

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