.{memory, log, buffer, string} := @use("../../../libraries/stn/src/lib.hb");
.{MouseEvent} := @use("../../../libraries/intouch/src/lib.hb").events;
.{bit0, bit1, bit2, bit3, bit4} := @use("bits.hb")
devices := @use("devices.hb")
controller := @use("controller.hb");
.{Info, Port} := controller
mouse := @use("mouse.hb")
format_page := memory.dangling(u8)

mouse_buffer := 0
keyboard_buffer := 0
info := Info.(0)

send_command := fn(port: ^Port, byte: u8): void {
	tries := 3
	loop if tries == 0 break else {
		controller.send_byte(port, byte)
		loop {
			info = controller.get_info()
			if controller.has_input(info) == false {
				continue
			}
			input := controller.get_input()
			if controller.get_port(info) != port {
				if check_complete(port) == false {
					port.packet[port.packet_length] = input
					port.packet_length += 1
				}
				continue
			}
			if input == 0xFA {
				return
			} else {
				break
			}
		}
		tries -= 1
	}
}

enable_streaming := fn(port: ^Port): void {
	@inline(send_command, port, 0xF4)
}

process := fn(port: ^controller.Port): void {
	if port.device.value < devices.MOUSE_5_BUTTON.value {
		event := MouseEvent.(0, 0, false, false, false)

		event.left = bit0(port.packet[0])
		event.right = bit1(port.packet[0])
		event.middle = bit2(port.packet[0])

		event.x_change = @intcast(port.packet[1])
		event.y_change = @intcast(port.packet[2])

		buffer.write(MouseEvent, mouse_buffer, &event)
	} else if port.device == devices.MOUSE_INIT_1 {
		port.device.value = port.packet[0]
		if port.device != devices.MOUSE_SCROLLWHEEL {
			enable_streaming(port)
			return
		}
		port.device = devices.MOUSE_INIT_2
	} else if port.device == devices.MOUSE_INIT_2 {
		port.device.value = port.packet[0]
	} else if port.device == devices.NO_DEVICE {
		if port.packet_length == 1 {
			port.device.value = port.packet[0]
			enable_streaming(port)
			//TODO: Upgrade mouse.
		} else {
			port.device.value = port.packet[1] | port.packet[0] << 8
			enable_streaming(port)
		}
		log.info("Identified device!\0")
		log.info(string.display_int(port.device.value, format_page, 16))
	} else {
		log.info("KEY PRESSED\0")
	}
}

check_complete := fn(port: ^controller.Port): bool {
	last_value := port.packet[port.packet_length - 1]
	if port.device == devices.NO_DEVICE {
		if last_value == 0 | last_value == 3 | last_value == 4 {
			return true
		} else if port.packet_length == 2 {
			return true
		}
	} else if port.device == devices.MOUSE_3_BUTTON {
		if port.packet_length == 3 return true
	} else if port.device == devices.MOUSE_SCROLLWHEEL | port.device == devices.MOUSE_5_BUTTON {
		if port.packet_length == 4 return true
	} else {
		if port.packet[0] == 0xE1 {
			if port.packet_length == 6 {
				return true
			}
		} else if port.packet[0] != 0xE0 {
			return true
		} else if port.packet_length == 2 & port.packet[1] != 0x2A & port.packet[1] != 0xB7 {
			return true
		} else if port.packet_length == 4 {
			return true
		}
	}
	return false
}

main := fn(): void {
	mouse_buffer = buffer.create("PS/2 Mouse\0")
	format_page = memory.alloc(u8, 1024)

	controller.init()

	if controller.port1.exists {
		//log.info("Port 1 exists.\0")
		controller.send_byte(@bitcast(0), 0xF4)
	}
	if controller.port2.exists {
		//controller.send_byte(&controller.port2, 0xF4)
	}

	loop {
		info = controller.get_info()

		if controller.timed_out(info) {
			log.error("Timeout error! Cannot handle these!\0")
		}
		if controller.check_parity(info) {
			log.error("Parity error! Cannot handle these!\0")
		}
		/*
		if controller.has_input(info) {
			port := controller.get_port(info)
			if port.packet_length > 0 & check_complete(port) {
				process(port)
			}
			input := controller.get_input()
			/*if input == 0xAA & port.can_hot_plug {
				port.device = devices.NO_DEVICE
				controller.send_byte(port, 0xF4)
			}*/
			port.packet[port.packet_length] = input
			port.packet_length += 1
			if check_complete(port) {
				process(port)
				port.packet_length = 0
			}
		}*/
	}
}