#include "host/util.h"
#include <sstream>

namespace
{
	char* GetCppExceptionInfo(EXCEPTION_POINTERS* exception)
	{
		// See https://blogs.msdn.microsoft.com/oldnewthing/20100730-00/?p=13273
		// Not very reliable so use __try
		__try { return ((char****)exception->ExceptionRecord->ExceptionInformation[2])[3][1][1] + 8; }
		__except (EXCEPTION_EXECUTE_HANDLER) { return "Could not find"; }
	}

	thread_local std::wstring lastError = L"Unknown error";

	__declspec(noreturn) void Terminate()
	{
		MessageBoxW(NULL, lastError.c_str(), L"Textractor ERROR", MB_ICONERROR);
		std::abort();
	}

	LONG WINAPI ExceptionLogger(EXCEPTION_POINTERS* exception)
	{
		thread_local static auto _ = std::invoke([]
		{
			std::set_terminate(Terminate);
			return 0;
		});

		MEMORY_BASIC_INFORMATION info = {};
		VirtualQuery(exception->ExceptionRecord->ExceptionAddress, &info, sizeof(info));

		std::wstringstream errorMsg;
		errorMsg << std::uppercase << std::hex <<
			L"Error code: " << exception->ExceptionRecord->ExceptionCode << std::endl <<
			L"Error address: " << exception->ExceptionRecord->ExceptionAddress << std::endl <<
			L"Error in module: " << Util::GetModuleFilename((HMODULE)info.AllocationBase).value_or(L"Could not find") << std::endl <<
			L"Additional info: " << info.AllocationBase << std::endl;

		if (exception->ExceptionRecord->ExceptionCode == 0xE06D7363)
			errorMsg << L"Additional info: " << GetCppExceptionInfo(exception) << std::endl;

		for (int i = 0; i < exception->ExceptionRecord->NumberParameters; ++i)
			errorMsg << L"Additional info: " << exception->ExceptionRecord->ExceptionInformation[i] << std::endl;

		lastError = errorMsg.str();
		return EXCEPTION_CONTINUE_SEARCH;
	}

	auto _ = std::invoke([]
	{
		AddVectoredExceptionHandler(FALSE, ExceptionLogger);
		SetUnhandledExceptionFilter([](auto) -> LONG { Terminate(); });
		return 0;
	});
}