#include "mod_socket.h"
#include "shm_mm_wraper.h"
#include "usg_common.h"

#define SERVER "server"
#define CLIENT "client"
#define DATE "DATE"

char *date(void) {
  time_t now = time(&now);
  struct tm *info = localtime(&now);
  char *text = asctime(info);
  text[strlen(text) - 1] = '\0'; // remove '\n'
  return (text);
}

int server(const int port) {
  int rv;

  void *socket = mod_open_socket(PUB_SUB);
  mod_socket_bind(socket, port);
  if ((rv = mod_listen(socket)) != 0) {
    printf("mod_listen");
  }
  for (;;) {
    printf("SERVER: SENDING DATE SURVEY REQUEST\n");
    if ((rv = mod_send(socket, DATE, strlen(DATE) + 1)) != 0) {
      printf("mod_send");
    }

    for (;;) {
      char *buf = NULL;
      int sz;
      rv = mod_recv(socket, (void **)&buf, &sz);
      
      if (rv != 0) {
        printf("mod_recv");
      }
      printf("SERVER: RECEIVED \"%s\" SURVEY RESPONSE\n", buf);
      mod_free(buf, sz);
    }

    printf("SERVER: SURVEY COMPLETE\n");
  }
}

int client(const int port, const char *name) {
  int rv;

  void *socket = mod_open_socket(PUB_SUB);
  if ((rv = mod_connect(socket, port)) != 0) {
    printf("mod_connect");
  }
  for (;;) {
    char *buf = NULL;
    int sz;
    if ((rv = mod_recv(socket, (void **)&buf, &sz)) == 0) {
      printf("CLIENT (%s): RECEIVED \"%s\" SURVEY REQUEST\n", name, buf);
      mod_free(buf, sz);
      char *d = date();
      printf("CLIENT (%s): SENDING DATE SURVEY RESPONSE\n", name);
      if ((rv = mod_send(socket, d, strlen(d) + 1, 0)) != 0) {
        printf("mod_send");
      }
    }
  }
}

int main(const int argc, const char **argv) {
  shm_mm_wrapper_init(512);
  int port;
  if ((argc >= 2) && (strcmp(SERVER, argv[1]) == 0)) {
    port = atoi(argv[2]);
    (server(port));
  }
  else if ((argc >= 3) && (strcmp(CLIENT, argv[1]) == 0)) {
    port = atoi(argv[2]);
    (client(port, argv[3]));
  } else {
    fprintf(stderr, "Usage: survey %s|%s <URL> <ARG> ...\n", SERVER, CLIENT);
  }

  shm_mm_wrapper_destroy();

  return 0;
}