#include <stdlib.h>
#include <stdbool.h>
#include "hedley.h"

typedef unsigned char uchar;

typedef enum : uchar {
	NONE  = ' ',
	CROSS = 'X',
	NOT   = 'O'
} Cell;

static Cell* const board = (Cell*)0x7F00;

static uchar readchar(void);
static bool checkwin(Cell c);
static void putchar(uchar);
static void printboard(void);

static uchar numpadkeys[11];

void _start(void) {
	__asm__ volatile (
		"movl $0x8000, %esp\n"
		"ljmp $(0),$(start)\n"
	"start:\n"
	);
	Cell cur = CROSS;
	for (uchar i = 0; i < 9; i++)
		board[i] = NONE;
	while (1) {
		printboard();
		uchar k = readchar();
		uchar o;
		if      (k > 0x46)
			o = 0x47;
		else if (k < 0x13)
			o =  0x10;
		else if (k < 0x21)
			o = (0x1E - 4);
		else
			o = (0x2C - 8);
		uchar c = numpadkeys[k - o];
		if (board[c] != NONE)
			continue;
		board[c] = cur;
		if (checkwin(cur))
			break;
		cur ^= (CROSS ^ NOT);
	}
	putchar(cur);
	printboard();
	while(1)
		;
	HEDLEY_UNREACHABLE();
}
// numpadkeys[x - 0x47]
static uchar numpadkeys[11] = {0, 1, 2, false, 3, 4, 5, false, 6, 7, 8};


static void printboard(void) {
	__asm__ volatile (
		"movb $0x02, %%ah;"
		"movb $0x00, %%bh;"
		"movw $0x0000, %%dx;"
		"int $0x10;"
		:
		:
		:"%ah","%bh","%dx"
	);
	for (uchar i = 0; i < 3; i++) {
		for (uchar j = 0; j < 3; j++) {
			putchar(board[i * 3 + j]);
		}
		putchar('\n');
		putchar('\r');
	}
}

static bool checkwin(Cell c) {
	for (size_t i = 0; i <= 2; i++)
		if (((board[i] == c) && (board[i + 3] == c) && (board[i + 6] == c)) &&
		   ((board[3*i] == c) && (board[3*i + 1] == c) && (board[3*i + 2] == c)))
			return true;
	if (board[0] == c && board[4] == c && board[8] == c)
		return true;
	if (board[2] == c && board[4] == c && board[6] == c)
		return true;
	return false;
}
/*
 * 012
 * 345
 * 678
 */

static uchar readchar(void) {
	char out;
__asm__ volatile (
	"movb $0x00, %%ah;"
	"int $0x16;"
	"movb %%ah, %0;"
	:"=r"(out)
	:
	:"%ax"
);
	return out;
}

static void putchar(uchar out) {
__asm__ volatile (
	"movb $0x0e, %%ah;"
	"movb %0, %%al;"
	"int $0x10;"
	:
	:"r"(out)
	:"%ax"
);
}