#include "ui.h"
#include "cardstack.h"

// Definition of ncurses color pair identifiers
#define CP_WHITE_ON_BLACK	1
#define CP_YELLOW_ON_BLACK	2
#define CP_BLUE_ON_BLACK	3
#define CP_MAGENTA_ON_BLACK	4
#define CP_RED_ON_BLACK		5

#define KEY_ESCAPE	27
#define KEY_RETURN	10

#define KEY_VI_LEFT	'h'
#define KEY_VI_RIGHT	'l'
#define KEY_VI_UP	'k'
#define KEY_VI_DOWN	'j'

static bool colors = false;
static WINDOW *w_table_cards;
static WINDOW *w_stack_points;
static WINDOW *w_current_state;
static WINDOW *w_hand_cards;

static void draw_card(WINDOW *w, const uint8_t row, const uint8_t col, const bool highlight, const card c)
{
	unsigned color_pair = 0;

	// If card is 0, draw a placeholder (empty space) instead of a card. This is used to remove a card that does not longer exist but has visible remains on the screen.
	if (0 == c)
	{
		mvwhline(w, row, col, ' ', 7);
		mvwhline(w, row+1, col, ' ', 7);
		mvwhline(w, row+2, col, ' ', 7);
		mvwhline(w, row+3, col, ' ', 7);
		mvwhline(w, row+4, col, ' ', 7);
		return;
	}

	if (highlight)
		wattron(w, A_BOLD);

	if (colors)
	{
		switch (card_get_points(c))
		{
			case 7:
				color_pair = CP_MAGENTA_ON_BLACK;
				break;
			case 5:
				color_pair = CP_RED_ON_BLACK;
				break;
			case 3:
				color_pair = CP_YELLOW_ON_BLACK;
				break;
			case 2:
				color_pair = CP_BLUE_ON_BLACK;
				break;
			default:
				color_pair = CP_WHITE_ON_BLACK;
		}

		wattron(w, COLOR_PAIR(color_pair));
	}

	mvwaddch(w, row, col, ACS_ULCORNER); // Upper left corner
	mvwhline(w, row, col+1, ACS_HLINE, 5); // Upper horizontal line
	mvwaddch(w, row, col+6, ACS_URCORNER); // Upper right corner
	mvwvline(w, row+1, col, ACS_VLINE, 3); // Left vertical line
	mvwvline(w, row+1, col+6, ACS_VLINE, 3); // Right vertical line
	mvwaddch(w, row+4, col, ACS_LLCORNER); // Lower left corner
	mvwhline(w, row+4, col+1, ACS_HLINE, 5); // Lower horizontal line
	mvwaddch(w, row+4, col+6, ACS_LRCORNER); // Lower right corner
	mvwprintw(w, row+1, col+1, " %3d ", c); // Card number, 3 digits wide, filled with spaces, if number has fewer digits

	// Clear the rest of the card first
	mvwprintw(w, row+2, col+1, "     ");
	mvwprintw(w, row+3, col+1, "     ");

	switch (card_get_points(c))
	{
		case 7:
			// "** **"
			// " *** "
			mvwhline(w, row+2, col+1, ACS_DIAMOND, 2);
			mvwhline(w, row+2, col+4, ACS_DIAMOND, 2);
			mvwhline(w, row+3, col+2, ACS_DIAMOND, 3);
			break;
		case 5:
			// "* * *"
			// " * * "
			mvwaddch(w, row+2, col+1, ACS_DIAMOND);
			mvwaddch(w, row+2, col+3, ACS_DIAMOND);
			mvwaddch(w, row+2, col+5, ACS_DIAMOND);
			mvwaddch(w, row+3, col+2, ACS_DIAMOND);
			mvwaddch(w, row+3, col+4, ACS_DIAMOND);
			break;
		case 3:
			// " * * "
			// "  *  "
			mvwaddch(w, row+2, col+2, ACS_DIAMOND);
			mvwaddch(w, row+2, col+4, ACS_DIAMOND);
			mvwaddch(w, row+3, col+3, ACS_DIAMOND);
			break;
		case 2:
			// " * * "
			// "     "
			mvwaddch(w, row+2, col+2, ACS_DIAMOND);
			mvwaddch(w, row+2, col+4, ACS_DIAMOND);
			break;
		default:
			// "  *  "
			// "     "
			mvwaddch(w, row+2, col+3, ACS_DIAMOND);
	}

	if (colors)
	{
		wattroff(w, COLOR_PAIR(color_pair));
	}
	
	if (highlight)
		wattroff(w, A_BOLD);
}

static void draw_cardstack(WINDOW *w, const uint8_t row, const uint8_t col, const bool highlight, const cardstack cs)
{
	// Clear old card stack first
	mvwhline(w, row, col, ' ', 15);
	mvwhline(w, row+1, col, ' ', 15);
	mvwhline(w, row+2, col, ' ', 15);
	mvwhline(w, row+3, col, ' ', 15);
	mvwhline(w, row+4, col, ' ', 15);

	for (uint8_t i = 0; i < MAX_CARDSTACK_SIZE; i++)
	{
		if (cs[i] != 0)
		{
			draw_card(w, row, i*2, highlight, cs[i]);
		}
	}
}

/**
 * Displays the table cards window. Draws table stacks and can highlight a stack.
 * @param[in] ts The table stacks to display
 * @param[in] highlight If true, a stack will be highlighted
 * @param[in] highlighted_stack The stack to highlight. Only used, if highlight is true
*/
void ui_display_wnd_table_cards(const tablestacks ts, const bool highlight, const uint8_t highlighted_stack)
{
	wattron(w_table_cards, A_BOLD);
	mvwprintw(w_table_cards, 0, 2, "Table Cards:");
	wattroff(w_table_cards, A_BOLD);

	for (uint8_t i = 0; i < NUM_TABLESTACKS; i++)
	{
		if (highlight && i == highlighted_stack)
		{
			draw_cardstack(w_table_cards, 1 + i*5, 0, true, ts[i]);
			continue;
		}

		draw_cardstack(w_table_cards, 1 + i*5, 0, false, ts[i]);
	}

	wrefresh(w_table_cards);
}

/**
 * Displays the stack points window.
 * @param[in] ts The table stacks used for calculation of stack points
 * @param[in] highlight If true, stack poins will be highlighted
 * @param[in] highlighted_points The stack points to highlight. Only used, if highlight is true
*/
void ui_display_wnd_stack_points(const tablestacks ts, const bool highlight, const uint8_t highlighted_points)
{
	wattron(w_stack_points, A_BOLD);
	mvwprintw(w_stack_points, 0, 0, "Pts:");
	wattroff(w_stack_points, A_BOLD);

	for (uint8_t i = 0; i < NUM_TABLESTACKS; i++)
	{
		if (highlight && i == highlighted_points)
		{
			wattron(w_stack_points, A_BOLD);
			mvwprintw(w_stack_points, 3 + i*5, 1, "%2d", cardstack_get_points(ts[i]));
			wattroff(w_stack_points, A_BOLD);
			continue;
		}

		mvwprintw(w_stack_points, 3 + i*5, 1, "%2d", cardstack_get_points(ts[i]));
	}

	wrefresh(w_stack_points);
}

/**
 * Displays the current state window, showing all player names and their open cards and the client player's score.
 * @param[in] pnoc Array of (player name, open card) tuples. Array has to be sorted by open card, lowest open card first
 * @param[in] num_players The number of players to show a player name and open card for
 * @param[in] active_players The currently active player
 * @param[in] score The players score
*/
void ui_display_wnd_current_state(const player_name_open_card_tuple pnoc[], const uint8_t num_players, const uint8_t active_player, const uint32_t score)
{
	wattron(w_current_state, A_BOLD);
	mvwprintw(w_current_state, 0, 0, "Current state:");
	mvwprintw(w_current_state, 0, 22, "Your Score: %3d", score);
	wattroff(w_current_state, A_BOLD);
	mvwprintw(w_current_state, 1, 1, "Player      Card");
	mvwprintw(w_current_state, 1, 22, "Player      Card");
	mvwvline(w_current_state, 1, 19, ACS_VLINE, 6); // Vertical line

	for (uint8_t i = 0; i < num_players; i++)
	{
		if (i == active_player)
			wattron(w_current_state, COLOR_PAIR(CP_YELLOW_ON_BLACK));
		
		if (i < 5)
		{
			mvwprintw(w_current_state, 2+i, 1, "%-s", pnoc[i].player_name);
			mvwprintw(w_current_state, 2+i, 13, "%3d", pnoc[i].open_card);
		}
		else
		{
			mvwprintw(w_current_state, 2+(i-5), 22, "%-s", pnoc[i].player_name);
			mvwprintw(w_current_state, 2+(i-5), 34, "%3d", pnoc[i].open_card);
		}

		if (i == active_player)
			wattroff(w_current_state, COLOR_PAIR(CP_YELLOW_ON_BLACK));
	}

	wrefresh(w_current_state);
}

/**
 * Displays the hand cards window.
 * @param[in] h The hand that will be displayed. h must be sorted in ascending order
 * @param[in] highlight If true, a card will be highlighted
 * @param[in] highlighted_card The card to highlight. Only used, if highlight is true
*/
void ui_display_wnd_hand_cards(const hand h, const bool highlight, const uint8_t highlighted_card)
{
	uint8_t num_zero_cards = 0;

	wattron(w_hand_cards, A_BOLD);
	mvwprintw(w_hand_cards, 0, 0, "Hand Cards:");
	wattroff(w_hand_cards, A_BOLD);

	// Clear old cards from screen first
	for (uint8_t i = 0; i < MAX_HAND_CARDS; i++)
	{
		if (i - num_zero_cards < 5) // Start with the first row of cards
			draw_card(w_hand_cards, 1, (i - num_zero_cards) * 8, false, 0);
		else // And then draw the second row
			draw_card(w_hand_cards, 6, (i - num_zero_cards - 5) * 8, false, 0);
	}

	for (uint8_t i = 0; i < MAX_HAND_CARDS; i++)
	{
		// Cound all 0 cards and don't draw them
		if (0 == h[i])
		{
			num_zero_cards++;
			continue;
		}

		if (highlight && i == highlighted_card)
		{
			if (i - num_zero_cards < 5) // Start with the first row of cards
				draw_card(w_hand_cards, 1, (i - num_zero_cards) * 8, true, h[i]);
			else // And then draw the second row
				draw_card(w_hand_cards, 6, (i - num_zero_cards - 5) * 8, true, h[i]);
			continue;
		}

		if (i - num_zero_cards < 5) // Start with the first row of cards
			draw_card(w_hand_cards, 1, (i - num_zero_cards) * 8, false, h[i]);
		else // And then draw the second row
			draw_card(w_hand_cards, 6, (i - num_zero_cards - 5) * 8, false, h[i]);
	}

	wrefresh(w_hand_cards);
}

uint8_t ui_choose_card(hand h)
{
	int key;
	uint8_t chosen_card_idx = 0;
	int8_t i;	// Has to be signed for modulo calculation

	hand_sort(h);

	// Select card with lowest index as default
	if (0 == h[chosen_card_idx])
	{
		i = (chosen_card_idx + 1) % MAX_HAND_CARDS;
		while(0 == h[i])
			i = (i + 1) % MAX_HAND_CARDS;
		chosen_card_idx = i;
	}

	ui_display_wnd_hand_cards(h, true, chosen_card_idx);

	while (KEY_RETURN != (key = wgetch(w_hand_cards)))
	{
		switch (key)
		{
			case KEY_VI_LEFT:
				// Fall through
			case KEY_LEFT:
				i = (chosen_card_idx - 1 + MAX_HAND_CARDS) % MAX_HAND_CARDS;
				while(0 == h[i])
					i = (i - 1 + MAX_HAND_CARDS) % MAX_HAND_CARDS;
				chosen_card_idx = i;
				break;

			case KEY_VI_RIGHT:
				// Fall through
			case KEY_RIGHT:
				i = (chosen_card_idx + 1) % MAX_HAND_CARDS;
				while(0 == h[i])
					i = (i + 1) % MAX_HAND_CARDS;
				chosen_card_idx = i;
				break;

			default:
				// No default
				break;
		}

		ui_display_wnd_hand_cards(h, true, chosen_card_idx);
	}

	return chosen_card_idx;
}

uint8_t ui_choose_stack(const tablestacks ts)
{
	int key;
	uint8_t chosen_stack_idx = 0;

	ui_display_wnd_table_cards(ts, true, chosen_stack_idx);
	ui_display_wnd_stack_points(ts, true, chosen_stack_idx);
	
	while (KEY_RETURN != (key = wgetch(w_table_cards)))
	{
		switch (key)
		{
			case KEY_VI_UP:
				// Fall through
			case KEY_LEFT:
				// Fall through
			case KEY_UP:
				if (0 == chosen_stack_idx)
					chosen_stack_idx = NUM_TABLESTACKS - 1;
				else
					chosen_stack_idx--;
				break;

			case KEY_VI_DOWN:
				// Fall through
			case KEY_RIGHT:
				// Fall through
			case KEY_DOWN:
				if (NUM_TABLESTACKS - 1 == chosen_stack_idx)
					chosen_stack_idx = 0;
				else
					chosen_stack_idx++;
				break;

			default:
				// No default
				break;
		}

		ui_display_wnd_table_cards(ts, true, chosen_stack_idx);
		ui_display_wnd_stack_points(ts, true, chosen_stack_idx);
	}

	return chosen_stack_idx;
}

/**
 * Initializes ncurses.
*/
void ui_init(void)
{
	initscr();	// Start curses mode
	if (TRUE == has_colors())
	{
		colors = true;
		start_color();

		init_pair(CP_WHITE_ON_BLACK, COLOR_WHITE, COLOR_BLACK);
		init_pair(CP_YELLOW_ON_BLACK, COLOR_YELLOW, COLOR_BLACK);
		init_pair(CP_BLUE_ON_BLACK, COLOR_BLUE, COLOR_BLACK);
		init_pair(CP_MAGENTA_ON_BLACK, COLOR_MAGENTA, COLOR_BLACK);
		init_pair(CP_RED_ON_BLACK, COLOR_RED, COLOR_BLACK);
		bkgd(COLOR_PAIR(0));
	}

	raw();		// Line buffering disabled
	noecho();	// Don't echo() while we do getch
	curs_set(0);	// Make the cursor invisible
	clear();
	refresh();

	// Window positions and sizes
	w_table_cards = newwin(21, 15, 0, 0);
	w_stack_points = newwin(19, 4, 0, 17);
	w_current_state = newwin(10, 38, 0, 24);
	w_hand_cards = newwin(11, 39, 10, 24);

	// Set input settings for windows
	keypad(w_table_cards, true); // We get F1, F2 etc..
	keypad(w_hand_cards, true); // We get F1, F2 etc..

	// Draw user interface
	mvvline(0, 22, ACS_VLINE, 21); // Vertical line
	refresh();
}

/**
 * Finalizes ncurses.
*/
void ui_fini(void)
{
	delwin(w_table_cards);
	delwin(w_stack_points);
	delwin(w_current_state);
	delwin(w_hand_cards);
	endwin();	// End curses mode
}