#include <std_include.hpp>
#include "loader/component_loader.hpp"
#include <utils/nt.hpp>

namespace resources
{
	namespace
	{
		HICON icon;
		HANDLE splash;

		HANDLE WINAPI load_image_a(const HINSTANCE handle, LPCSTR name, const UINT type, const int c_x, const int c_y,
			const UINT load)
		{
			const utils::nt::library self;
			if (self.get_handle() == handle && name == LPCSTR(0x64)) return splash;

			return LoadImageA(handle, name, type, c_x, c_y, load);
		}

		HICON WINAPI load_icon_a(const HINSTANCE handle, const LPCSTR name)
		{
			const utils::nt::library self;
			if (self.get_handle() == handle && name == LPCSTR(1)) return icon;

			return LoadIconA(handle, name);
		}
	}

	class component final : public component_interface
	{
	public:
		~component()
		{
			if (utils::nt::is_wine())
			{
				return;
			}

			if (icon) DestroyIcon(icon);
			if (splash) DeleteObject(splash);
		}

		void post_start() override
		{
			if (utils::nt::is_wine())
			{
				return;
			}

			const utils::nt::library self;
			icon = LoadIconA(self.get_handle(), MAKEINTRESOURCEA(ID_ICON));
			splash = LoadImageA(self.get_handle(), MAKEINTRESOURCEA(IMAGE_SPLASH), 0, 0, 0, LR_COPYFROMRESOURCE);
		}

		void* load_import(const std::string& library, const std::string& function) override
		{
			if (utils::nt::is_wine())
			{
				return nullptr;
			}

			if (library == "USER32.dll")
			{
				if (function == "LoadIconA")
				{
					return load_icon_a;
				}

				if (function == "LoadImageA")
				{
					return load_image_a;
				}
			}

			return nullptr;
		}
	};
}

REGISTER_COMPONENT(resources::component)