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

	Copyright SoftwareLabs

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

#include <NewKit/Utils.hpp>
#include <KernelKit/DebugOutput.hpp>
#include <cstddef>

namespace NewOS
{
	Int rt_string_cmp(const Char* src, const Char* cmp, Size size)
	{
		Int32 counter = 0;

		for (Size index = 0; index < size; ++index)
		{
			if (src[index] != cmp[index])
				++counter;
		}

		return counter;
	}

	void rt_zero_memory(voidPtr pointer, Size len)
	{
		rt_set_memory(pointer, 0, len);
	}

	Size rt_string_len(const Char* str, SizeT _len)
	{
		if (*str == '\0')
			return 0;

		Size len{0};
		while (str[len] != '\0')
		{
			if (len > _len)
			{
				break;
			}

			++len;
		}

		return len;
	}

	Size rt_string_len(const Char* ptr)
	{
		if (!ptr)
			return 0;

		SizeT cnt = 0;

		while (*ptr != (Char)0)
		{
			++ptr;
			++cnt;
		}

		return cnt;
	}

	voidPtr rt_set_memory(voidPtr src, char value, Size len)
	{
		if (!src || len < 1)
			return nullptr;
		char* start = reinterpret_cast<Char*>(src);

		while (len)
		{
			*start = value;
			++start;
			--len;
		}

		return (voidPtr)start;
	}

	Int rt_move_memory(const voidPtr src, voidPtr dst, Size len)
	{
		if (len < 1)
			return -2;
		if (!src || !dst)
			return -1;

		char* srcChr  = reinterpret_cast<Char*>(src);
		char* dstChar = reinterpret_cast<Char*>(dst);
		Size  index	  = 0;

		while (index < len)
		{
			dstChar[index] = srcChr[index];
			srcChr[index]  = 0;

			++index;
		}

		return 0;
	}

	Int rt_copy_memory(const voidPtr src, voidPtr dst, Size len)
	{
		if (len < 1)
			return -2;

		char* srcChr  = reinterpret_cast<char*>(src);
		char* dstChar = reinterpret_cast<char*>(dst);
		Size  index	  = 0;

		while (index < len)
		{
			dstChar[index] = srcChr[index];
			++index;
		}

		return index;
	}

	const Char* alloc_string(const Char* text)
	{
		if (!text)
			return nullptr;

		const Char* string = new Char[rt_string_len(text)];
		if (!string)
			return nullptr;

		voidPtr vText = reinterpret_cast<voidPtr>(const_cast<char*>(text));
		voidPtr vStr  = reinterpret_cast<voidPtr>(const_cast<char*>(string));
		rt_copy_memory(vText, vStr, rt_string_len(text));

		return string;
	}

	Int rt_to_uppercase(Int character)
	{
		if (character >= 'a' && character <= 'z')
			return character - 0x20;

		return character;
	}

	Int rt_to_lower(Int character)
	{
		if (character >= 'A' && character <= 'Z')
			return character + 0x20;

		return character;
	}

	bool rt_to_string(Char* str, Int limit, Int base)
	{
		if (limit == 0)
			return false;

		Int copy_limit = limit;
		Int cnt		   = 0;
		Int ret		   = base;

		while (limit != 1)
		{
			ret		 = ret % 10;
			str[cnt] = ret;

			++cnt;
			--limit;
			--ret;
		}

		str[copy_limit] = '\0';
		return true;
	}

	Boolean is_space(Char chr)
	{
		return chr == ' ';
	}

	Boolean is_newln(Char chr)
	{
		return chr == '\n';
	}

	voidPtr rt_string_in_string(const char* in, const char* needle)
	{
		for (size_t i = 0; i < rt_string_len(in); ++i)
		{
			if (rt_string_cmp(in + i, needle, rt_string_len(needle)) == 0)
				return reinterpret_cast<voidPtr>(const_cast<char*>(in + i));
		}

		return nullptr;
	}

	// @brief Checks for a string start at the character.

	char* rt_string_has_char(char* str, const char chr)
	{
		while (*str != chr)
		{
			++str;

			if (*str == 0)
				return nullptr;
		}

		return str;
	}
} // namespace NewOS

////////////////////////////////////////////////////////////////////////////////////////
/// Exported C functions
////////////////////////////////////////////////////////////////////////////////////////

/// @brief memset in C++
EXTERN_C void memset(void* dst, char src, size_t len)
{
	NewOS::rt_set_memory(dst, src, len);
}

/// @brief memcpy in C++
EXTERN_C void memcpy(void* dst, void* src, size_t len)
{
	NewOS::rt_copy_memory(src, dst, len);
}

/// @brief memmove in C++
EXTERN_C void* memmove(void* dst, void* src, size_t len)
{
	NewOS::rt_copy_memory(src, dst, len);
	return dst;
}

/// @brief strlen definition in C++.
EXTERN_C size_t strlen(const char* whatToCheck)
{
	return NewOS::rt_string_len(whatToCheck);
}

/// @brief memcmp in C++
EXTERN_C NewOS::SizeT memcmp(void* dst, void* src, size_t len)
{
	return NewOS::rt_string_cmp((char*)src, (char*)dst, len);
}

/// @brief strcmp in C++
EXTERN_C NewOS::SizeT strcmp(char* dst, char* src, size_t len)
{
	return NewOS::rt_string_cmp(src, dst, len);
}