/*
	This file is part of Fennix Drivers.

	Fennix Drivers is free software: you can redistribute it and/or
	modify it under the terms of the GNU General Public License as
	published by the Free Software Foundation, either version 3 of
	the License, or (at your option) any later version.

	Fennix Drivers is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with Fennix Drivers. If not, see <https://www.gnu.org/licenses/>.
*/

#include "aip.h"

#include <errno.h>
#include <base.h>

bool IsKeyboard(uint8_t ID)
{
	/* Common keyboard IDs */
	return ID == 0xAB || ID == 0xAC || ID == 0x5D ||
		   ID == 0x2B || ID == 0x47 || ID == 0x60;
}

bool IsMouse(uint8_t ID)
{
	/* Common mouse IDs */
	return ID == 0x00 || ID == 0x03 || ID == 0x04;
}

const char *GetPS2DeviceName(uint8_t ID, uint8_t SubID)
{
	switch (ID)
	{
	case 0x00:
		return "Standard PS/2 Mouse";
	case 0x03:
		return "Mouse with scroll wheel";
	case 0x04:
		return "Mouse 5 buttons";
	case 0xAB:
	{
		switch (SubID)
		{
		case 0x83: /* Normal */
		case 0x41: /* Translated */
		case 0xC1: /* Normal + Translated */
			return "Standard PS/2 Keyboard";
		case 0x84:
		case 0x54:
			return "IBM Thinkpad/Spacesaver Keyboard";
		case 0x85:
			return "NCD N-97/122-Key Host Connect(ed) Keyboard";
		case 0x86:
			return "122-Key Keyboard";
		case 0x90:
			return "Japanese \"G\" Keyboard";
		case 0x91:
			return "Japanese \"P\" Keyboard";
		case 0x92:
			return "Japanese \"A\" Keyboard";
		default:
			return "Unknown PS/2 Keyboard";
		}
	}
	case 0xAC:
	{
		switch (SubID)
		{
		case 0xA1:
			return "NCD Sun Keyboard";
		default:
			return "Unknown NCD Sun Keyboard";
		}
	}
	case 0x5D:
	case 0x2B:
		return "Trust Keyboard";
	case 0x47:
	case 0x60:
		return "NMB SGI Keyboard";
	default:
		return "Unknown PS/2 Device";
	}
}

uint8_t Device1ID[2] = {0x00, 0x00};
uint8_t Device2ID[2] = {0x00, 0x00};
bool DualChannel = false;

int DriverEntry()
{
	PS2WriteCommand(PS2_CMD_DISABLE_PORT_1);
	PS2WriteCommand(PS2_CMD_DISABLE_PORT_2);
	PS2ClearOutputBuffer();

	PS2WriteCommand(PS2_CMD_READ_CONFIG);
	PS2_CONFIGURATION cfg = {.Raw = PS2ReadData()};

	DualChannel = cfg.Port2Clock;
	if (DualChannel)
		Log("Dual channel PS/2 controller detected");
	cfg.Port1Interrupt = 1;
	cfg.Port2Interrupt = 1;
	cfg.Port1Translation = 1;

	PS2WriteCommand(PS2_CMD_WRITE_CONFIG);
	PS2WriteData(cfg.Raw);

	PS2WriteCommand(PS2_CMD_TEST_CONTROLLER);
	uint8_t test = PS2ReadData();
	if (test != PS2_TEST_PASSED)
	{
		Log("PS/2 controller self test failed (%#x)", test);
		return -EFAULT;
	}

	PS2WriteCommand(PS2_CMD_WRITE_CONFIG);
	PS2WriteData(cfg.Raw);

	// bool port2avail = false;
	// if (DualChannel)
	// {
	// 	PS2WriteCommand(PS2_CMD_ENABLE_PORT_1);
	// 	PS2WriteCommand(PS2_CMD_READ_CONFIG);
	// 	cfg.Raw = PS2ReadData();
	// 	port2avail = cfg.Port2Clock;
	// 	PS2WriteCommand(PS2_CMD_DISABLE_PORT_1);
	// }

	PS2WriteCommand(PS2_CMD_TEST_PORT_1);
	test = PS2ReadData();
	if (test != 0x00)
	{
		Log("PS/2 Port 1 self test failed (%#x)", test);
		return -EFAULT;
	}

	if (DualChannel)
	{
		PS2WriteCommand(PS2_CMD_TEST_PORT_2);
		test = PS2ReadData();
		if (test != 0x00)
		{
			Log("PS/2 Port 2 self test failed (%#x)", test);
			return -EFAULT;
		}
	}

	PS2WriteCommand(PS2_CMD_ENABLE_PORT_1);
	if (DualChannel)
		PS2WriteCommand(PS2_CMD_ENABLE_PORT_2);

	int errK = InitializeKeyboard();

	int errM = 0;
	if (DualChannel)
		errM = InitializeMouse();

	/** A device may fail, but if the other one works,
	 * we can still use it.
	 */
	if (errK != 0 && errM != 0)
		return -ENODEV;
	return 0;
}

int DriverFinal()
{
	FinalizeKeyboard();
	FinalizeMouse();
	PS2WriteCommand(PS2_CMD_DISABLE_PORT_1);
	PS2WriteCommand(PS2_CMD_DISABLE_PORT_2);
	return 0;
}

int DriverPanic()
{
	PS2WriteCommand(PS2_CMD_DISABLE_PORT_1);
	PS2WriteCommand(PS2_CMD_DISABLE_PORT_2);
	return 0;
}

void __intStub() {}
int DriverProbe()
{
	RegisterInterruptHandler(1, __intStub);
	RegisterInterruptHandler(12, __intStub);

	int kbd = DetectPS2Keyboard();
	int mouse = DetectPS2Mouse();

	UnregisterAllInterruptHandlers(__intStub);

	if (kbd != 0 && mouse != 0)
		return -ENODEV;

	if (kbd == 0)
	{
		if (!IsKeyboard(Device1ID[0]))
		{
			Log("PS/2 Port 1 is not a keyboard");
			// return -EINVAL;
		}
	}

	if (mouse == 0)
	{
		if (!IsMouse(Device2ID[0]))
		{
			Log("PS/2 Port 2 is not a mouse");
			// return -EINVAL;
		}
	}

	KPrint("PS/2 Port 1: %s (0x%X 0x%X)",
		   GetPS2DeviceName(Device1ID[0], Device1ID[1]),
		   Device1ID[0], Device1ID[1]);
	KPrint("PS/2 Port 2: %s (0x%X 0x%X)",
		   GetPS2DeviceName(Device2ID[0], Device2ID[1]),
		   Device2ID[0], Device2ID[1]);
	return 0;
}

DriverInfo("aip",
		   "Advanced Integrated Peripheral Driver",
		   "EnderIce2",
		   "0.1",
		   "GPLv3");