#include "..\include_eapi_header.h"
#include<shlwapi.h>
#include<shlobj_core.h>
#pragma comment(lib,"shlwapi.lib")
// 调用格式: SDT_BOOL 创建程序项, 命令说明: "在指定的特殊位置位置程序组文件夹中创建快捷方式。"
// 参数<1>: 程序组文件夹 SDT_TEXT, 参数说明: "要创建的快捷方式的程序组文件夹名称，非路径。"
// 参数<2>: 程序项文件夹 SDT_TEXT, 参数说明: "要创建的快捷方式名称"
// 参数<3>: 快捷方式链接的程序路径 SDT_TEXT, 参数说明: "绑定执行程序文件名的路径。"
// 参数<4>: [特殊目录位置 SDT_INT], 参数说明: "程序组的起始位置。可以为以下常量值之一：0、#接口常量.开始菜单；1、#接口常量.开始菜单程序；2、#接口常量.桌面。"
namespace eapi_create_link {
	BOOL CreateLink(std::string_view LinkName, std::string_view FilePath, std::string_view remark, WORD wHotkey, std::string_view comline, std::string_view workdir)
	{
		BOOL blSucceeded = FALSE;
		IShellLinkA* piShellLink;
		if (SUCCEEDED(CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (void**)&piShellLink)))
		{
			IPersistFile* piPersistFile;
			if (SUCCEEDED(piShellLink->QueryInterface(IID_IPersistFile, (void**)&piPersistFile)))
			{
				piShellLink->SetPath(FilePath.data());
				piShellLink->SetShowCmd(SW_SHOW);
				if (remark.empty() == FALSE)  piShellLink->SetDescription(remark.data());
				if (comline.empty() == FALSE)  piShellLink->SetArguments(comline.data());
				if (wHotkey != 0)  piShellLink->SetHotkey(wHotkey);
				if (workdir.empty() == FALSE)  piShellLink->SetWorkingDirectory(workdir.data());
				const char* FileNmae = (strchr(LinkName.data(), '.') != NULL ? LinkName.data() : (std::string(LinkName) + ".lnk").c_str());
				WCHAR WideCharStr[MAX_PATH] = { 0 };
				MultiByteToWideChar(0, 0, FileNmae, -1, WideCharStr, MAX_PATH);
				blSucceeded = SUCCEEDED(piPersistFile->Save(WideCharStr, TRUE));
				piPersistFile->Release();
			}
			piShellLink->Release();
		}
		return blSucceeded;
	}
}

inline BOOL CreateProgramItem(const char* GroupName, const char* ItemName, const char* ItemPath, int position)
{

	BOOL result = FALSE;
	if (PathFileExistsA(ItemPath))
	{
		char pszPath[MAX_PATH] = { 0 };
		SHGetSpecialFolderPathA(0, pszPath, position, 0);
		wsprintfA(pszPath, "%s\\%s\\%s.lnk", pszPath, GroupName, ItemName);
		if (eapi_create_link::CreateLink(pszPath, ItemPath, "", 0, "", ""))
		{
			auto  notifyShell = [](LONG wEventId, CHAR* szPath)
			{
				SHChangeNotify(wEventId, SHCNF_FLUSH | SHCNF_PATH, szPath, 0);
				CHAR* p;
				for (p = szPath + strlen(szPath) - 1; *p != '\\'; p--)
					*p = '\0';
				SHChangeNotify(SHCNE_UPDATEDIR | SHCNE_INTERRUPT, SHCNF_FLUSH | SHCNF_PATH, szPath, 0);
			};
			notifyShell(SHCNE_CREATE | SHCNE_INTERRUPT, pszPath);
			result = TRUE;
		}
	}
	return result;
}

EAPI_EXTERN_C void eapi_CreateProgramItem_39_eapi(PMDATA_INF pRetData, INT nArgCount, PMDATA_INF pArgInf)
{
	//开始菜单
	int pos = CSIDL_STARTMENU;
	if (pArgInf[3].m_dtDataType && pArgInf[3].m_int) {
		if (pArgInf[3].m_int == 2)
		{
			//桌面
			pos = CSIDL_DESKTOPDIRECTORY;
		}
		else if (pArgInf[3].m_int == 2)
		{
			//程序组
			pos = CSIDL_PROGRAMS;
		}
	}
	pRetData->m_bool = CreateProgramItem(pArgInf->m_pText, pArgInf[1].m_pText, pArgInf[2].m_pText, pos);
}