#define _POSIX_C_SOURCE 200809L
#include "modules/sway/ipc/client.hpp"
#include <cstdio>
#include <string>
#include <sys/socket.h>
#include <sys/un.h>

static const char ipc_magic[] = {'i', '3', '-', 'i', 'p', 'c'};
static const size_t ipc_header_size = sizeof(ipc_magic)+8;

std::string getSocketPath() {
  const char *env = getenv("SWAYSOCK");
  if (env != nullptr) {
    return std::string(env);
  }
  std::string str;
  {
    std::string str_buf;
    FILE*  in;
    char  buf[512] = { 0 };
    if ((in = popen("sway --get-socketpath 2>/dev/null", "r")) == nullptr) {
      throw std::runtime_error("Failed to get socket path");
    }
    while (fgets(buf, sizeof(buf), in) != nullptr) {
      str_buf.append(buf, sizeof(buf));
    }
    pclose(in);
    str = str_buf;
  }
  if (str.back() == '\n') {
    str.pop_back();
  }
  return str;
}

int ipcOpenSocket(const std::string &socketPath) {
  struct sockaddr_un addr = {0};
  int socketfd;
  if ((socketfd = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
    throw std::runtime_error("Unable to open Unix socket");
  }
  addr.sun_family = AF_UNIX;
  strncpy(addr.sun_path, socketPath.c_str(), sizeof(addr.sun_path) - 1);
  addr.sun_path[sizeof(addr.sun_path) - 1] = 0;
  int l = sizeof(struct sockaddr_un);
  if (connect(socketfd, reinterpret_cast<struct sockaddr *>(&addr), l) == -1) {
    throw std::runtime_error("Unable to connect to " + socketPath);
  }
  return socketfd;
}

struct ipc_response ipcRecvResponse(int socketfd) {
  struct ipc_response response;
  char data[ipc_header_size];
  auto data32 = reinterpret_cast<uint32_t *>(data + sizeof(ipc_magic));
  size_t total = 0;

  while (total < ipc_header_size) {
    ssize_t received = recv(socketfd, data + total, ipc_header_size - total, 0);
    if (received <= 0) {
      throw std::runtime_error("Unable to receive IPC response");
    }
    total += received;
  }

  total = 0;
  response.size = data32[0];
  response.type = data32[1];
  char payload[response.size + 1];

  while (total < response.size) {
    ssize_t received = recv(socketfd, payload + total, response.size - total, 0);
    if (received < 0) {
      throw std::runtime_error("Unable to receive IPC response");
    }
    total += received;
  }
  payload[response.size] = '\0';
  response.payload = std::string(payload);
  return response;
}

std::string ipcSingleCommand(int socketfd, uint32_t type, const char *payload, uint32_t *len) {
  char data[ipc_header_size];
  auto data32 = reinterpret_cast<uint32_t *>(data + sizeof(ipc_magic));
  memcpy(data, ipc_magic, sizeof(ipc_magic));
  data32[0] = *len;
  data32[1] = type;

  if (send(socketfd, data, ipc_header_size, 0) == -1) {
    throw std::runtime_error("Unable to send IPC header");
  }
  if (send(socketfd, payload, *len, 0) == -1) {
    throw std::runtime_error("Unable to send IPC payload");
  }
  struct ipc_response resp = ipcRecvResponse(socketfd);
  *len = resp.size;
  return resp.payload;
}