#include "checkouts.h"
#include "comp.h"

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <time.h>

#include <emscripten/emscripten.h>

#define P_USER 1
#define P_BOAT 2

int delay_ms = 1000;

// TODO refactor *everything*

struct match_state {
	struct leg *l1, *l2;
	int complete;
};

struct match_state *state;

void set_user_active() {
	EM_ASM({setPlayerActive($0)}, P_USER);
	EM_ASM({setPromptHandler($0)}, "visit");
}

void set_boat_active() {
	EM_ASM({setPlayerActive($0)}, P_BOAT);
	EM_ASM({setPromptHandler()}, "");
}

void suggested(int rem, char *buf) {
	buf[0] = 0;

	if (rem > 1 && rem <= 170) {
		char *target = CHECKOUTS[2][rem-1];
		if (target) {
			int trem = rem - segment_points(segment_from_name(target));
			int len = sprintf(buf, "%s", target);
			if (trem) {
				target = CHECKOUTS[1][trem-1];
				len += sprintf(buf + len, "-%s", target);
				trem = trem - segment_points(segment_from_name(target));

				if (trem) {
					target = CHECKOUTS[0][trem-1];
					len += sprintf(buf + len, "-%s", target);
				}
			}
		}
	}
}

EMSCRIPTEN_KEEPALIVE void get_suggested(int rem, int player) {
	static char buf[20];
	suggested(rem, buf);
	EM_ASM({updatePlayerSugg($0, $1)}, player, buf);
}

EMSCRIPTEN_KEEPALIVE int is_match_over() {
	if (state->l1->rem <= 0 || state->l2->rem <= 0)
		return 1;
	return 0;
}

void match_over() {
	EM_ASM({setPromptHandler($0)}, "init");
	EM_ASM({promptMsgR($0)}, "Press OK to play again.");
	EM_ASM(setPlayerActive());
}

EMSCRIPTEN_KEEPALIVE void draw_match() {
	struct leg *l1 = state->l1;
	struct leg *l2 = state->l2;

	char visit_no[10], u_pts[10], u_rem[10], b_pts[10], b_rem[10], b_darts[100];
	visit_no[0] = u_pts[0] = u_rem[0] = b_pts[0] = b_rem[0] = b_darts[0] = 0;

	EM_ASM({drawVisit($0, $1, $2, $3, $4, $5)},
		"0", "", "501", "", "501", "");

	int n_visits = l1->n_visits > l2->n_visits ? l1->n_visits : l2->n_visits;
	for (int i = 0; i < n_visits; ++i) {
		sprintf(visit_no, "%d", i + 1);

		struct visit *v = l1->visits + i;
		sprintf(u_pts, "%d", v->points);
		sprintf(u_rem, "%d", v->rem);

		if (i < l2->n_visits) {
			v = l2->visits + i;
			sprintf(b_pts, "%d", v->points);
			sprintf(b_rem, "%d", v->rem);

			for (int j = 0; j < v->n_darts; ++j) {
				char *n = segment_name(v->darts[j]);
				sprintf(b_darts + (j ? (j * 5 - 1) : 0), j == 0 ? "%4s" : " %4s", n);
				free(n);
			}
		}

		EM_ASM({drawVisit($0, $1, $2, $3, $4, $5)},
			visit_no, u_pts, u_rem, b_pts, b_rem, b_darts);

		visit_no[0] = 0;
		u_pts[0] = 0;
		u_rem[0] = 0;
		b_pts[0] = 0;
		b_rem[0] = 0;
		b_darts[0] = 0;
	}

	if (l1->rem <= 0) {
		EM_ASM({promptMsgL($0)}, "Darts needed?");
		EM_ASM({setPromptHandler($0)}, "num_darts");
	} else if (l2->rem <= 0) {
		EM_ASM({promptMsgL($0)}, "Bot wins. :(");
		match_over();
	} else {
		EM_ASM({promptMsgL($0)}, "Enter points:");
	}
}

char *str_for_rem_state(int rem) {
	char *buf = malloc(10);

	if (rem < 0 || rem == 1)
		strcpy(buf, "BUST");
	else
		sprintf(buf, "%d", rem);

	return buf;
}

EMSCRIPTEN_KEEPALIVE void update_user_rem(int rem) {
	char *str = str_for_rem_state(rem);
	EM_ASM({updatePlayerRem($0, $1)}, P_USER, str);
	free(str);

	get_suggested(rem, P_USER);
}

EMSCRIPTEN_KEEPALIVE void update_user_rem_from_pts(int pts) {
	update_user_rem(state->l1->rem - pts);
}

EMSCRIPTEN_KEEPALIVE void update_boat_rem(int rem) {
	char *str = str_for_rem_state(rem);
	EM_ASM({updatePlayerRem($0, $1)}, P_BOAT, str);
	free(str);

	get_suggested(rem, P_BOAT);
}

void end_boat_throwing(int, int); // FIXME

EMSCRIPTEN_KEEPALIVE void boat_visit() {
	if (state->l1->rem <= 0 || state->l2->rem <= 0)
		return;

	EM_ASM({promptMsgL($0)}, "Bot is throwing…");

	struct leg *l = state->l2;
	comp_visit(l);

	struct visit *v = l->visits + l->n_visits - 1;

	double avg;
	if (v->rem > 0)
		avg = (double)(l->start - l->rem) / l->n_visits;
	else
		avg = ((double)l->start / (((l->n_visits - 1) * 3) + v->n_darts)) * 3;

	if (!delay_ms) {
		end_boat_throwing(l->rem, avg);
		return;
	}

	int pts = 0;
	char str[20] = {0};

	for (int i = 0; i < v->n_darts; ++i) {
		pts += segment_points(v->darts[i]);
		char *seg_name = segment_name(v->darts[i]);
		if (i) strcat(str, "-");
		strcat(str, seg_name);
		free(seg_name);
		EM_ASM({scheduleCCall($0, $1, $2, $3, $4, $5)},
			"draw_boat_throwing", delay_ms * (i+1),
			"number", pts, "string", str);
	}

	EM_ASM({scheduleCCall($0, $1, $2, $3, $4, $5)},
		"end_boat_throwing", delay_ms * (v->n_darts + 1),
		"number", l->rem, "number", avg);
}

EMSCRIPTEN_KEEPALIVE bool user_visit(int points) {
	if (state->l1->rem <= 0 || state->l2->rem <= 0 ||
		points > state->l1->rem || state->l1->rem - points == 1 ||
		points > 180 || points == 179 || points == 178 || points == 176 ||
		points == 175 || points == 173 || points == 172 || points == 169 ||
		points == 166 || points == 163) {

		EM_ASM(oi());
		return false;
	}

	struct leg *l = state->l1;
	struct visit *v = l->visits + l->n_visits++;
	v->points = points;
	l->rem -= points;
	v->rem = l->rem;
	update_user_rem(l->rem);

	if (v->rem > 0)
		EM_ASM({updatePlayerAvg($0, $1)}, P_USER,
			(double)(l->start-l->rem)/l->n_visits);

	EM_ASM({clearVisits()});
	draw_match();

	if (!is_match_over()) {
		set_boat_active();
		boat_visit();
	}

	return true;
}

// FIXME neg
EMSCRIPTEN_KEEPALIVE void user_visit_to_rem(int rem) {
	user_visit(state->l1->rem - rem);
}

EMSCRIPTEN_KEEPALIVE void draw_boat_throwing(int pts, char *str) {
	char buf[10];
	sprintf(buf, "%d", pts);

	int rem = state->l2->n_visits > 1 ? state->l2->visits[state->l2->n_visits-2].rem : state->l2->start;

	update_boat_rem(rem - pts);
	EM_ASM({setPromptInput($0)}, buf);
	EM_ASM({promptMsgR($0)}, str);
}

EMSCRIPTEN_KEEPALIVE void end_boat_throwing(int rem, int avg) {
	update_boat_rem(rem);
	EM_ASM({updatePlayerAvg($0, $1)}, P_BOAT, avg);
	EM_ASM({setPromptInput($0)}, "");
	EM_ASM(clearVisits());
	EM_ASM({promptMsgR($0)}, "");
	draw_match();

	if (!is_match_over())
		set_user_active();
}

EMSCRIPTEN_KEEPALIVE void resp_numdarts(int n_darts) {
	if (n_darts < 1 || n_darts > 3) {
		EM_ASM(oi());
		return;
	}

	struct leg *l = state->l1;

	EM_ASM({updatePlayerAvg($0, $1)}, P_USER,
		((double)l->start / (((l->n_visits - 1) * 3) + n_darts)) * 3);
	EM_ASM({promptMsgL($0)}, "You win! :)");
	match_over();
	EM_ASM({setPromptHandler($0)}, "init");
}

void init_boat() {
	static int ran;

	if (ran) return;
	ran = 1;

	srand(time(NULL));
	init_board();

	EM_ASM({updateStdev($0)}, horizontal_stdev);
	EM_ASM({updateDelay($0)}, delay_ms);
}

EMSCRIPTEN_KEEPALIVE void free_match() {
	leg_free(state->l1);
	leg_free(state->l2);
	free(state);
	state = NULL;
}

EMSCRIPTEN_KEEPALIVE void start_match() {
	if (state) free_match();

	init_boat();
	state = calloc(1, sizeof(struct match_state));
	state->l1 = leg_init(501, "User");
	state->l2 = leg_init(501, "Bot");

	EM_ASM({updatePlayerName($0, $1)}, P_USER, state->l1->name);
	EM_ASM({updatePlayerName($0, $1)}, P_BOAT, state->l2->name);
	update_user_rem(state->l1->rem);
	update_boat_rem(state->l2->rem);

	EM_ASM(clearVisits());
	set_user_active();
	EM_ASM({promptMsgR($0)}, "");
	draw_match();
}

EMSCRIPTEN_KEEPALIVE void set_stdev(float hstdev, float vstdev) {
	horizontal_stdev = hstdev;
	vertical_stdev = vstdev;
}

EMSCRIPTEN_KEEPALIVE void set_delay(int delay) {
	delay_ms = delay;
}

int main() {
	start_match();
}