#define _GNU_SOURCE
#include "login.h"
#include "data.h"
#include "server.h"
#include "util.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <unistd.h>
#include <argon2.h>

void handle_player(int conn_fd) {
  printf("Let's make a new player object\n");
  playerc_t plrc = {0};
  plrc.state = LoggedOut;
  plrc.conn = conn_fd;
  player_t plr = {0};
  plrc.plr = plr;

  printf("No sense in having a player if they can't log in!\n");

  int err = step_login(&plrc, conn_fd);
  if (err) {
    err = send_to_fd(conn_fd, "Error in login process, exiting");
    if (err) {
      printf("Could not send error message");
    }
    close(conn_fd);
    return;
  }

  close(conn_fd);
}

int step_login(playerc_t *player, int conn_fd) {
  int err = 0;

  for (;;) {
    switch (player->state) {
    case LoggedOut: /* the player has yet to log in, this is a fresh connection!
                     */
      printf("The player is logged out\n");
      send_to_fd(player->conn,
                 "Welcome! Please enter your **username** below\n");
      char *buf = (char *)malloc(1 << 10);
      printf("Waiting to receive...\n");
      recv(player->conn, buf, 1024, 0);
      printf("Received!\n");
      char* playername = trimwhitespace(buf);
      err = try_load_plr(playername, player);
      printf("Finished trying to load the player\n");
      if (err) {
        // File doesn't exist, let's create a player!
	player->plr.name = (char *)malloc(32);
        strcpy(player->plr.name, playername);
        send_to_fd(player->conn, "Howdy! Want to introduce yourself? [y|n]\n");
        recv(player->conn, buf, sizeof(buf), 0);
        if (buf[0] == 'y') {
          player->state = WantMakeAccount;
        } else if (buf[0] == 'n') {
          send_to_fd(player->conn,
                     "Awww :(. Oh well, hope to see you again soon!\n");
          return 4;
        }
        continue;
      } else {
        player->state = LoggedOut;
        continue;
      }
    case WantMakeAccount:
      asprintf(&buf, "Welcome aboard %s! Why don't you start us off by giving me a password.\n", player->plr.name);
      send_to_fd(player->conn, buf);
      recv(player->conn, buf, sizeof(buf), 0);
        if (strlen(buf) <= 5) {
            send_to_fd(player->conn, "We're sorry, but passwords need to be at least 5 characters");
        } else {
            player->plr.pw_hash = (char *)malloc(512);
            char* hashbuf = (char *)malloc (1 << 5);
            err = argon2i_hash_raw(32, 512, 1, buf, strlen(buf), "adSAaVWIueBwP2jHfAl7diRJ9ijC1Ysb9e920d6d9+I=", strlen("adSAaVWIueBwP2jHfAl7diRJ9ijC1Ysb9e920d6d9+I="), player->plr.pw_hash, sizeof(player->plr.pw_hash));

            if (err != ARGON2_OK) {
                send_to_fd(player->conn, "Uhhh, something happened. Try again.");
                printf("%d\n", err);
                player->state = LoggedOut;
                continue;
            } else {
                send_to_fd(player->conn, "Sounds good! let's drop you into a world!");
                player->state = Complete;
                player->plr.hp = player->plr.max_hp = 20;
                player->plr.level = 1;
                player->plr.xp = 0;
                player->plr.location_id = 0;
                try_write_plr(&player->plr);
                continue;
            }
        }
    }
  }

  return err;
}