#include "..\include_eapi_header.h"
#include<shlwapi.h>
#include<shlobj_core.h>
#pragma comment(lib,"shlwapi.lib")


inline bool CancelAutoRun(const std::string& valueName)
{
	//注册表
	bool ret = false;
	HKEY hKeyAutoStart;
	if (RegOpenKeyExA(HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Run", 0, KEY_ALL_ACCESS, &hKeyAutoStart) == ERROR_SUCCESS)
	{
		if (RegDeleteValueA(hKeyAutoStart, valueName.c_str()) == ERROR_SUCCESS) {
			ret = true;
		}
		RegCloseKey(hKeyAutoStart);
	}
	//启动文件夹
	char pszPath[MAX_PATH];
	SHGetSpecialFolderPathA(0, pszPath, CSIDL_STARTUP, 0);
	std::string linkPath = std::string(pszPath) + "\\" + valueName + ".lnk";
	if (DeleteFileA(linkPath.c_str())) {
		ret = true;
	}
	//刷新文件夹
	auto notifyShell = [](LONG wEventId, const std::string& path)
	{
		SHChangeNotify(wEventId, SHCNF_FLUSH | SHCNF_PATH, path.c_str(), 0);
		size_t pos = path.find_last_of("\\");
		std::string parentPath = path.substr(0, pos);
		SHChangeNotify(SHCNE_UPDATEDIR | SHCNE_INTERRUPT, SHCNF_FLUSH | SHCNF_PATH, parentPath.c_str(), 0);
	};
	notifyShell(SHCNE_DELETE | SHCNE_INTERRUPT, linkPath);

	return ret;
}

static bool SetAutoRun(const std::string& lpValueName, const std::string& path, bool toRegKey)
{
	HKEY phkResult;
	if (!PathFileExistsA(path.c_str()))
		return false;

	//如果是修改注册表启动项
	if (toRegKey)
	{
		if (!RegOpenKeyA(HKEY_LOCAL_MACHINE, "Software\\Microsoft\\Windows\\CurrentVersion\\Run", &phkResult))
		{
			if (!RegSetValueExA(phkResult, lpValueName.c_str(), 0, REG_SZ, (LPBYTE)path.c_str(), (path.size() + 1) * sizeof(char)))
			{
				RegCloseKey(phkResult);
				return true;
			}
			RegCloseKey(phkResult);
		}
		return false;
	}

	//否则创建启动文件夹快捷方式
	char pszPath[MAX_PATH] = { 0 };
	::SHGetSpecialFolderPathA(NULL, pszPath, CSIDL_STARTUP, TRUE);

	std::string lnkPath = pszPath;
	lnkPath += "\\";
	lnkPath += lpValueName;
	lnkPath += ".lnk";

	bool result = false;
	IShellLinkA* piShellLink = NULL;
	IPersistFile* piPersistFile = NULL;
	WCHAR WideCharStr[MAX_PATH] = { 0 };

	if (!SUCCEEDED(CoInitialize(NULL)))
		return false;

	if (CoCreateInstance(CLSID_ShellLink, 0, CLSCTX_INPROC_SERVER, IID_IShellLinkA, (void**)&piShellLink) >= 0)
	{
		piShellLink->SetPath(path.c_str());
		if (piShellLink->QueryInterface(IID_IPersistFile, (void**)&piPersistFile) >= 0)
		{
			MultiByteToWideChar(0, 0, lnkPath.c_str(), -1, WideCharStr, MAX_PATH);
			if (piPersistFile->Save(WideCharStr, 2) == S_OK)
				result = true;
			piPersistFile->Release();
		}
		piShellLink->Release();
	}
	//成功刷新目录
	if (result) {
		//和上方lambda其实一样，这里没用string就避免了额外的拷贝
		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);
	}
	return result;
}













// 调用格式: SDT_BOOL 设置自动运行, 命令说明: "设置程序在开机时自动运行。本命令为初级命令"
// 参数<1>: 程序类名 SDT_TEXT, 参数说明: "程序类名、置于启动组或者注册表中的名称"
// 参数<2>: 程序路径 SDT_TEXT, 参数说明: "要设置为自动运行的程序文件名（包含绝对路径）。"
// 参数<3>: [位置 SDT_INT], 参数说明: "可以被省略。为以下常量值之一：0、#接口常量.到注册表；1、#接口常量.到启动组。"
EAPI_EXTERN_C void eapi_SetAutoRun_25_eapi(PMDATA_INF pRetData, INT nArgCount, PMDATA_INF pArgInf)
{
	std::string Name, Path;
	BOOL isres = TRUE;
	if (pArgInf[0].m_dtDataType && pArgInf[0].m_pText && *pArgInf[0].m_pText != 0)
	{
		Name = pArgInf[0].m_pText;
	}
	if (pArgInf[1].m_dtDataType && pArgInf[1].m_pText)
	{
		Path = pArgInf[1].m_pText;

	}
	if (pArgInf[2].m_dtDataType)
	{
		isres = pArgInf[2].m_int == 0 ? TRUE : FALSE;
	}
	pRetData->m_bool = SetAutoRun(Name, Path, isres);

}