/* -------------------------------------------

	Copyright SoftwareLabs

------------------------------------------- */

#include <ArchKit/ArchKit.hpp>
#include <KernelKit/DebugOutput.hpp>
#include <KernelKit/Framebuffer.hpp>
#include <NewKit/Utils.hpp>

namespace NewOS
{
	enum CommStatus
	{
		kStateInvalid,
		kStateReady	   = 0xCF,
		kStateTransmit = 0xFC,
		kStateCnt	   = 3
	};

	namespace Detail
	{
		constexpr short PORT = 0x3F8;

		static int kState = kStateInvalid;

		/// @brief Init COM1.
		/// @return
		bool serial_init() noexcept
		{
#ifdef __DEBUG__
			if (kState == kStateReady || kState == kStateTransmit)
				return true;

			HAL::Out8(PORT + 1, 0x00); // Disable all interrupts
			HAL::Out8(PORT + 3, 0x80); // Enable DLAB (set baud rate divisor)
			HAL::Out8(PORT + 0, 0x03); // Set divisor to 3 (lo byte) 38400 baud
			HAL::Out8(PORT + 1, 0x00); //                  (hi byte)
			HAL::Out8(PORT + 3, 0x03); // 8 bits, no parity, one stop bit
			HAL::Out8(PORT + 2, 0xC7); // Enable FIFO, clear them, with 14-byte threshold
			HAL::Out8(PORT + 4, 0x0B); // IRQs enabled, RTS/DSR set
			HAL::Out8(PORT + 4, 0x1E); // Set in loopback mode, test the serial chip
			HAL::Out8(PORT + 0, 0xAE); // Test serial chip (send byte 0xAE and check if
									   // serial returns same byte)

			// Check if serial is faulty (i.e: not same byte as sent)
			if (HAL::In8(PORT) != 0xAE)
			{
				ke_stop(RUNTIME_CHECK_HANDSHAKE);
			}

			kState = kStateReady;

			// If serial is not faulty set it in normal operation mode
			// (not-loopback with IRQs enabled and OUT#1 and OUT#2 bits enabled)
			HAL::Out8(Detail::PORT + 4, 0x0F);
#endif // __DEBUG__

			return true;
		}
	} // namespace Detail

	EXTERN_C void ke_io_write(const char* bytes)
	{
#ifdef __DEBUG__
		Detail::serial_init();

		if (!bytes || Detail::kState != kStateReady)
			return;
		if (*bytes == 0)
			return;

		Detail::kState = kStateTransmit;

		SizeT index = 0;
		SizeT len	= rt_string_len(bytes, 256);

		while (index < len)
		{
			if (bytes[index] == '\r')
				HAL::Out8(Detail::PORT, '\r');

			HAL::Out8(Detail::PORT, bytes[index] == '\r' ? '\n' : bytes[index]);
			++index;
		}

		Detail::kState = kStateReady;
#endif // __DEBUG__
	}

	EXTERN_C void ke_io_read(const char* bytes)
	{
#ifdef __DEBUG__
		Detail::serial_init();

		if (!bytes || Detail::kState != kStateReady)
			return;

		Detail::kState = kStateTransmit;

		SizeT index = 0;

		///! TODO: Look on how to wait for the UART to complete.
		while (true)
		{
			auto in = HAL::In8(Detail::PORT);

			///! If enter pressed then break.
			if (in == 0xD)
			{
				break;
			}

			if (in < '0' || in < 'A' || in < 'a')
			{
				if (in != '@' || in != '!' || in != '?' || in != '.' || in != '/' ||
					in != ':')
				{
					continue;
				}
			}

			((char*)bytes)[index] = in;

			++index;
		}

		((char*)bytes)[index] = 0;

		Detail::kState = kStateReady;
#endif // __DEBUG__
	}

	TerminalDevice& TerminalDevice::The() noexcept
	{
		static TerminalDevice* out = nullptr;

		if (!out)
			out = new TerminalDevice(NewOS::ke_io_write, NewOS::ke_io_read);

		return *out;
	}

} // namespace NewOS