#include "..\include_eapi_header.h"

// 调用格式: SDT_BOOL 添加IE工具条按钮, 命令说明: "向IE工具栏添加自己的按钮。"
// 参数<1>: 按钮文字 SDT_TEXT, 参数说明: "标题"
// 参数<2>: 图标文件名 SDT_TEXT, 参数说明: "图标路径"
// 参数<3>: 执行动作 SDT_TEXT, 参数说明: "在IE中按下按钮后执行一个可执行文件或打开网页,就是命令行。"
// 参数<4>: [按钮热点图标 SDT_TEXT], 参数说明: "按钮处于热点状态显示的图标,默认不变更"
// 参数<5>: [按钮是否被显示 SDT_BOOL], 参数说明: "是否在工具栏显示,默认为真"
EAPI_EXTERN_C void eapi_AddButtonToIE_35_eapi(PMDATA_INF pRetData, INT nArgCount, PMDATA_INF pArgInf)
{
	char* ButtonTitle = nullptr,
		* Icon = nullptr,
		* Comline = nullptr,
		* HotButton = nullptr;
	BOOL IsVisible = TRUE;

	if (pArgInf->m_pText) {

		ButtonTitle = pArgInf->m_pText;
	}
	if (pArgInf[1].m_pText)
	{
		Icon = pArgInf[1].m_pText;
		HotButton = pArgInf[1].m_pText;
	}
	if (pArgInf[2].m_pText)
	{
		Comline = pArgInf[2].m_pText;
	}
	if (pArgInf[3].m_dtDataType && pArgInf[3].m_bool == FALSE)
	{
		IsVisible = FALSE;
	}
	//所有都内联lambda是不是就不用考虑命名冲突了
	auto AddButtonToIE = [](const char* ButtTitle, const char* Icon, const char* Comline, const char* HotIcon, bool IsVisible)
	{
		bool result = false;
		GUID pguid;
		DWORD dwDisposition;
		if (CoCreateGuid(&pguid) == S_OK)
		{
			HKEY phkResult = 0;
			char lpSubKey[260] = { 0 };
			sprintf_s(lpSubKey,
				"Software\\Microsoft\\Internet Explorer\\Extensions\\{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
				pguid.Data1,
				pguid.Data2,
				pguid.Data3,
				pguid.Data4[0],
				pguid.Data4[1],
				pguid.Data4[2],
				pguid.Data4[3],
				pguid.Data4[4],
				pguid.Data4[5],
				pguid.Data4[6],
				pguid.Data4[7]);
			if (RegCreateKeyExA(HKEY_LOCAL_MACHINE, lpSubKey, 0, nullptr, 0, KEY_ALL_ACCESS, nullptr, &phkResult, &dwDisposition) == ERROR_SUCCESS)
			{
				RegSetValueExA(phkResult, "CLSID", 0, REG_SZ, (const BYTE*)"{1FBA04EE-3024-11D2-8F1F-0000F87ABD16}", sizeof("{1FBA04EE-3024-11D2-8F1F-0000F87ABD16}"));
				const char* BtutonVisible = IsVisible ? "Yes" : "No";
				RegSetValueExA(phkResult, "Default Visible", 0, REG_SZ, (const BYTE*)BtutonVisible, strlen(BtutonVisible));
				RegSetValueExA(phkResult, "ButtonText", 0, REG_SZ, (const BYTE*)ButtTitle, strlen(ButtTitle));
				RegSetValueExA(phkResult, "Icon", 0, REG_SZ, (const BYTE*)Icon, strlen(Icon));
				RegSetValueExA(phkResult, "HotIcon", 0, REG_SZ, (const BYTE*)HotIcon, strlen(HotIcon));
				RegSetValueExA(phkResult, "Exec", 0, REG_SZ, (const BYTE*)Comline, strlen(Comline));
				if (phkResult)
					RegCloseKey(phkResult);
				result = true;
			}
		}
		return result;
	};
	pRetData->m_bool = AddButtonToIE(ButtonTitle, Icon, Comline, HotButton, IsVisible);
}