#include "hanoi.h"

int main(int argc, char **argv)
{
	t_tower	towers[3];
	size_t disk_num = 3;
	if (argc == 2)
		if (sscanf(argv[1], "%lu", &disk_num) != 1)
		{
			fprintf(stderr, "%s: error: `%s` is not a valid number of disk", argv[0], argv[1]);
			return (1);
		}
	towers_init(towers, disk_num);

	initscr();
	noecho();
	cbreak();
	keypad(stdscr, TRUE);
	curs_set(0);

	int width, height;
	getmaxyx(stdscr, height, width);
	(void)height;

	int win_width = width / 3;

	WINDOW	*wins[3];
	refresh();
	for (int i = 0; i < 3; i++)
	{
		if ((wins[i] = newwin(0, win_width, 0, i * win_width)) == NULL)
			abort();
	}

	enum e_mode mode = MODE_SELECT_FROM;
	int from_selection = 0;
	int current = 0;
	bool running = true;
	while (running)
	{
		for (int i = 0; i < 3; i++)
		{
			enum e_highlight highlight = HIGHLIGHT_NONE;
			if (mode == MODE_SELECT_TO && i == from_selection)
				highlight |= HIGHLIGHT_FROM;
			if (i == current)
				highlight |= HIGHLIGHT_CURRENT;
			tower_put(&towers[i], wins[i], highlight);
		}

		int c;
		c = getch();

		switch (c)
		{
			case 'q':
				running = false;
				break;
			case 'j':
			case KEY_LEFT:
				current--;
				break;
			case 'k':
			case KEY_RIGHT:
				current++;
				break;
			case '\n':
				if (mode == MODE_SELECT_FROM)
				{
					if (towers[current].len == 0)
						mvprintw(0, 0, "Tower %d is empty");
					else
					{
						from_selection = current;
						mode = MODE_SELECT_TO;
					}
				}
				else if (mode == MODE_SELECT_TO)
				{
					if (!towers[current].len == 0 &&
						tower_peek(&towers[current]) < tower_peek(&towers[from_selection]))
					{
						mvprintw(0, 0, "Top disk of %d is smaller than top disk of %d",
								current, from_selection);
					}
					else
						towers_move(towers, from_selection, current);
					mode = MODE_SELECT_FROM;
				}
				break;
		}
		if (current < 0)
			current = 2;
		else if (current > 2)
			current = 0;
	}

	for (int i = 0; i < 3; i++)
		delwin(wins[3]);
	endwin();
	return 0;
}