#include "common.h"
#include "context.h"
#include "errors.h"
#include "modules.h"
#include "resources.h"

namespace user32 {
	constexpr uint32_t RT_STRING_ID = 6;
	constexpr uintptr_t kDefaultKeyboardLayout = 0x04090409;
	constexpr int UOI_FLAGS = 1;
	constexpr DWORD WSF_VISIBLE = 0x0001;

	struct USEROBJECTFLAGS {
		BOOL fInherit;
		BOOL fReserved;
		DWORD dwFlags;
	};

	int WIN_FUNC LoadStringA(void* hInstance, unsigned int uID, char* lpBuffer, int cchBufferMax) {
		HOST_CONTEXT_GUARD();
		DEBUG_LOG("LoadStringA(%p, %u, %p, %d)\n", hInstance, uID, lpBuffer, cchBufferMax);
		if (!lpBuffer || cchBufferMax <= 0) {
			return 0;
		}
		wibo::Executable *mod = wibo::executableFromModule((HMODULE) hInstance);
		if (!mod) {
			return 0;
		}
		wibo::ResourceIdentifier type = wibo::ResourceIdentifier::fromID(RT_STRING_ID);
		wibo::ResourceIdentifier table = wibo::ResourceIdentifier::fromID((uID >> 4) + 1);
		wibo::ResourceLocation loc;
		if (!mod->findResource(type, table, std::nullopt, loc)) {
			return 0;
		}
		const uint16_t *cursor = reinterpret_cast<const uint16_t *>(loc.data);
		const uint16_t *end = cursor + (loc.size / sizeof(uint16_t));
		unsigned int entryIndex = uID & 0x0Fu;
		for (unsigned int i = 0; i < entryIndex; ++i) {
			if (cursor >= end) {
				return 0;
			}
			uint16_t length = *cursor++;
			if (cursor + length > end) {
				return 0;
			}
			cursor += length;
		}
		if (cursor >= end) {
			return 0;
		}
		uint16_t length = *cursor++;
		if (cursor + length > end) {
			return 0;
		}
		int copyLength = length;
		if (copyLength > cchBufferMax - 1) {
			copyLength = cchBufferMax - 1;
		}
		for (int i = 0; i < copyLength; ++i) {
			lpBuffer[i] = static_cast<char>(cursor[i] & 0xFF);
		}
		lpBuffer[copyLength] = 0;
		DEBUG_LOG("LoadStringA -> %.*s\n", copyLength, lpBuffer);
		return copyLength;
	}

	int WIN_FUNC LoadStringW(void* hInstance, unsigned int uID, uint16_t* lpBuffer, int cchBufferMax) {
		HOST_CONTEXT_GUARD();
		DEBUG_LOG("LoadStringW(%p, %u, %p, %d)\n", hInstance, uID, lpBuffer, cchBufferMax);
		wibo::Executable *mod = wibo::executableFromModule((HMODULE) hInstance);
		if (!mod) {
			return 0;
		}
		wibo::ResourceIdentifier type = wibo::ResourceIdentifier::fromID(RT_STRING_ID);
		wibo::ResourceIdentifier table = wibo::ResourceIdentifier::fromID((uID >> 4) + 1);
		wibo::ResourceLocation loc;
		if (!mod->findResource(type, table, std::nullopt, loc)) {
			return 0;
		}
		const uint16_t *cursor = reinterpret_cast<const uint16_t *>(loc.data);
		const uint16_t *end = cursor + (loc.size / sizeof(uint16_t));
		unsigned int entryIndex = uID & 0x0Fu;
		for (unsigned int i = 0; i < entryIndex; ++i) {
			if (cursor >= end) {
				return 0;
			}
			uint16_t length = *cursor++;
			if (cursor + length > end) {
				return 0;
			}
			cursor += length;
		}
		if (cursor >= end) {
			return 0;
		}
		uint16_t length = *cursor++;
		if (cursor + length > end) {
			return 0;
		}
		if (cchBufferMax == 0) {
			if (lpBuffer) {
				*reinterpret_cast<uint16_t **>(lpBuffer) = const_cast<uint16_t *>(cursor);
			}
			return length;
		}
		if (!lpBuffer || cchBufferMax <= 0) {
			return 0;
		}
		int copyLength = length;
		if (copyLength > cchBufferMax - 1) {
			copyLength = cchBufferMax - 1;
		}
		for (int i = 0; i < copyLength; ++i) {
			lpBuffer[i] = cursor[i];
		}
		lpBuffer[copyLength] = 0;
		DEBUG_LOG("LoadStringW -> length %d\n", copyLength);
		return copyLength;
	}

	int WIN_FUNC MessageBoxA(void *hwnd, const char *lpText, const char *lpCaption, unsigned int uType) {
		HOST_CONTEXT_GUARD();
		printf("MESSAGE BOX: [%s] %s\n", lpCaption, lpText);
		fflush(stdout);
		return 1;
	}

	HKL WIN_FUNC GetKeyboardLayout(DWORD idThread) {
		HOST_CONTEXT_GUARD();
		DEBUG_LOG("GetKeyboardLayout(%u)\n", idThread);
		(void)idThread;
		wibo::lastError = ERROR_SUCCESS;
		return reinterpret_cast<HKL>(kDefaultKeyboardLayout);
	}

	HWINSTA WIN_FUNC GetProcessWindowStation() {
		DEBUG_LOG("GetProcessWindowStation()\n");
		static int kWindowStationStub;
		wibo::lastError = ERROR_SUCCESS;
		return reinterpret_cast<HWINSTA>(&kWindowStationStub);
	}

	BOOL WIN_FUNC GetUserObjectInformationA(HANDLE hObj, int nIndex, PVOID pvInfo, DWORD nLength,
								 LPDWORD lpnLengthNeeded) {
		DEBUG_LOG("GetUserObjectInformationA(%p, %d, %p, %u, %p)\n", hObj, nIndex, pvInfo, nLength,
			  lpnLengthNeeded);
		(void)hObj;

		if (lpnLengthNeeded) {
			*lpnLengthNeeded = sizeof(USEROBJECTFLAGS);
		}

		if (nIndex != UOI_FLAGS) {
			wibo::lastError = ERROR_CALL_NOT_IMPLEMENTED;
			return FALSE;
		}

		if (!pvInfo || nLength < sizeof(USEROBJECTFLAGS)) {
			wibo::lastError = ERROR_INSUFFICIENT_BUFFER;
			return FALSE;
		}

		auto *flags = reinterpret_cast<USEROBJECTFLAGS *>(pvInfo);
		flags->fInherit = FALSE;
		flags->fReserved = FALSE;
		flags->dwFlags = WSF_VISIBLE;

		wibo::lastError = ERROR_SUCCESS;
		return TRUE;
	}

	HWND WIN_FUNC GetActiveWindow() {
		DEBUG_LOG("GetActiveWindow()\n");
		wibo::lastError = ERROR_SUCCESS;
		return nullptr;
	}
}


static void *resolveByName(const char *name) {
	if (strcmp(name, "LoadStringA") == 0) return (void *) user32::LoadStringA;
	if (strcmp(name, "LoadStringW") == 0) return (void *) user32::LoadStringW;
	if (strcmp(name, "MessageBoxA") == 0) return (void *) user32::MessageBoxA;
	if (strcmp(name, "GetKeyboardLayout") == 0) return (void *) user32::GetKeyboardLayout;
	if (strcmp(name, "GetProcessWindowStation") == 0) return (void *) user32::GetProcessWindowStation;
	if (strcmp(name, "GetUserObjectInformationA") == 0) return (void *) user32::GetUserObjectInformationA;
	if (strcmp(name, "GetActiveWindow") == 0) return (void *) user32::GetActiveWindow;
	return nullptr;
}

wibo::ModuleStub lib_user32 = {
	(const char *[]){
		"user32",
		nullptr,
	},
	resolveByName,
	nullptr,
};