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

#include <tak.h>
#include <negamax_cnn1986.h>
#include <lcdlib.h>

static char *gamelog = 0;
static int human;

static void
append_to_gamelog(const char *line, const uint8_t win_line) {
	// I _could_ dynamically compute the size but ... don't let
	// `perfect' be the enemy of `good' ?
	char prepend[8];

	if (win_line) {
		prepend[0] = '\n';
		prepend[1] = 0;
	} else if (ply & 1) {
		snprintf(prepend, 7, "%s%d. ",
						 (ply == 1) ? "" : "\n", ply/2+1);
	} else {
		strcpy(prepend, " ");
	}
	// Make room for this line
	gamelog = realloc(gamelog,
										strlen(gamelog)
										+ strlen(prepend)
										+ strlen(line) + 1);
	strcat(gamelog,prepend);
	strcat(gamelog,line);
}

static void
end_game(char *line, char *win) {
	append_to_gamelog(line, 0);
	append_to_gamelog(win, 1);
	lcd_printf_line(L_SCROLL, "Game over: %s", win);
}

static int
handle_turn(char *line) {
	// Track win state
	uint8_t new_win = (won == 0xFF);
	switch (do_ptn(line)) {
		// Errors
		case PTN_INVALID:  { lcd_put_line(L_SCROLL, "Invalid PTN.");  break; }
		case ACT_ILLEGAL:  { lcd_put_line(L_SCROLL, "Illegal ply.");  break; }
		case ACT_OVERFLOW: { lcd_put_line(L_SCROLL, "Overflow.");     break; }
			// Game has ended
		case GAME_END: {
			// Did it end this turn?
			if (new_win) {
				switch (won) {
					case WIN_DRAW:       { end_game(line,"1/2-1/2"); break; }
					case WIN_FLAT_BLACK: { end_game(line,"0-F");     break; }
					case WIN_FLAT_WHITE: { end_game(line,"F-0");     break; }
					case WIN_ROAD_BLACK: { end_game(line,"0-R");     break; }
					case WIN_ROAD_WHITE: { end_game(line,"R-0");     break; }
				}
				return EXIT_SUCCESS;
			} else {
				lcd_put_line(L_SCROLL, "Game over.");
				break;
			}
		}
			// Valid, append to game log
		default: {
			append_to_gamelog(line, 0);
			return EXIT_SUCCESS;
		}
	}
	return EXIT_FAILURE;
}

static void
new_game(uint8_t size) {
	reset_state(size);
	lcd_put_line(L_SCROLL, "New 5x5 game.");
	if (gamelog) gamelog = realloc(gamelog, sizeof(char));
	else gamelog = malloc(sizeof(char));
	gamelog[0] = 0;
}

// Set up output function for negamax_cnn1986
static uint8_t perc;

inline void
negamax_cnn1986_display_progress(const uint8_t depth) {
	if (depth == 0) {
		perc++;
		lcd_printf_line(L_OVERWRITE, "Computing: %d%%", perc*4);
	}
}

static int
negamax_cnn1986_turn() {
	// Prepare progress bar
	lcd_put_line(L_SCROLL, "Computing: 0%");
	// Run the minimax
	perc = 0;
	float minimax = negamax_cnn1986_generate();
	// Failed to find a move?
	if (minimax <= -infty) {
		lcd_printf_line(L_SCROLL, "%s concedes!",
										(ply & 1) ? "Black" : "White");
		return EXIT_FAILURE;
	} else {
		lcd_printf_line(L_SCROLL, "%s: %s",
										(ply & 1) ? "Black" : "White",
										negamax_cnn1986_ptn);
		handle_turn(negamax_cnn1986_ptn);
		return EXIT_SUCCESS;
	}
}

static void
do_game_log(void) {
	lcd_put_line(L_SCROLL, "Not implemented.");
}

static int
input_is_not_turn(const char *line) {
	switch (line[0]) {
		case 'l': { do_game_log(); break; }
		case 'n': { new_game(5);   break; }
		case 's': {
			negamax_cnn1986_search_depth = line[1] - '0';
			lcd_printf_line(L_SCROLL, "Search depth: %d",
											negamax_cnn1986_search_depth);
			break;
		}
		case 'B': { human = 1; new_game(5); break; }
		case 'W': { human = 0; new_game(5); break; }
		default: return EXIT_FAILURE;
	}
	return EXIT_SUCCESS;
}

int
main(int argc, char **argv) {
	(void)(argc);
	(void)(argv);

	if (lcd_begin() != EXIT_SUCCESS)
		return EXIT_FAILURE;

	human = 0;
	negamax_cnn1986_search_depth = 3;
	new_game(5);

	char *line = NULL;
	ssize_t read;
	size_t alloc_size;

	for (int playing = 1; playing;) {
		while (human == 0) {
			if (ply & 1) lcd_put_line(L_SCROLL, "Black: ");
			else lcd_put_line(L_SCROLL, "White: ");
			read = getline(&line, &alloc_size, stdin);
			if (read > 0) {
				line[read - 1] = 0;
				if (input_is_not_turn(line)) {
					int r = handle_turn(line);
					if (r == EXIT_SUCCESS && won == 0xFF) {
						human = 1;
					}
				}
			} else {
				playing = 0;
				break;
			}
		}
		if (playing) {
			negamax_cnn1986_turn();
			human = 0;
		}
	}

	lcd_end();
	return EXIT_SUCCESS;
}