#include "daemon.h"

#include "timer.h"

#include <signal.h>
#include <errno.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <pthread.h>
#include <assert.h>

static struct sockaddr_un addr = (struct sockaddr_un){.sun_path = "/tmp/pomo.sock", .sun_family = AF_UNIX};

int socket_create() {
  errno = 0;  
  int fd = socket(AF_UNIX, SOCK_STREAM, 0);
  if (fd == -1) {
    printf("Socket creation failed! %s\n", strerror(errno));
    fflush(stdout);        
    exit(1);
  }
  return fd;  
}

void socket_set_server(int sockfd) {
  errno = 0;
  if (bind(sockfd, (struct sockaddr*)&addr, sizeof(addr)) == -1) {
    printf("Socket bind failed! %s\n", strerror(errno));
    fflush(stdout);        
    exit(1);
  }
  errno = 0;
  if (listen(sockfd, 1) == -1) {
    printf("Socket listen() failed! %s\n", strerror(errno));
    exit(1);    
  }    
}

void socket_set_client(int sockfd) {
  errno = 0;
  if(connect(sockfd, (struct sockaddr*)&addr, sizeof(addr)) == -1) {
    printf("Socket connect failed! %s\n", strerror(errno));
    fflush(stdout);        
    exit(1);
  }
}

void socket_bind_clean(int sig) {
  unlink("/tmp/pomo.sock");
  exit(0);
}

void daemon_init(Daemon* d) {
  assert(d != NULL);
  socket_set_server(d->sockfd);
  signal(SIGINT, socket_bind_clean);
  signal(SIGTERM, socket_bind_clean);
  atexit((void *)socket_bind_clean);
}

void* daemon_listener_loop(void *d) {
  assert(d != NULL);
  while(1) {
    int clientfd = accept(((Daemon *)d)->sockfd, NULL, NULL);
      
    if (clientfd == -1) {
      printf("Incoming connection failed.\n");
      continue;
    }

    char buffer[1024];
    read(clientfd, buffer, 1024 - 1);

    // TODO: Create a mechanism to fetch commands without writing everything again.
    // TODO: Make it so that a more efficient approach is used instead of a string.
    if(strncmp(buffer, "start", sizeof("start")) == 0) { 
      timer_start(((Daemon *)d)->timer);
    }

    if(strncmp(buffer, "toggle", sizeof("toggle")) == 0) {
      timer_toggle_state(((Daemon *)d)->timer);
    }

    if(strncmp(buffer, "reset", sizeof("reset")) == 0) {
      timer_reset(((Daemon *)d)->timer);
    }
    
    /* send(clientfd, "done", sizeof("done"), 0); */
    close(clientfd);
  }
  return NULL;
}

void daemon_run(Daemon* d) {
  assert(d != NULL);

  Timer t;
  timer_init(&t, 3000);
  d->timer = &t;
  pthread_t thread;
  if(pthread_create(&thread, NULL, daemon_listener_loop, d) != 0) {
    printf("Error: daemon_listener_loop thread cannot be created!");
    exit(1);
  }
  
  timer_loop(&t);

  void* retval;
  pthread_join(thread, &retval);
}