#include <iostream>
#include <windows.h>
#include <shlobj.h>
#pragma comment(lib, "shell32.lib")

using namespace std;

// 获取启动文件夹路径的函数
bool GetStartupFolder(char* path, DWORD size) {
    LPITEMIDLIST pidl;
    if (SUCCEEDED(SHGetFolderLocation(NULL, CSIDL_STARTUP, NULL, 0, &pidl))) {
        SHGetPathFromIDList(pidl, path);
        CoTaskMemFree(pidl);
        return true;
    }
    return false;
}

int main() {
    char startupPath[MAX_PATH];
    if (GetStartupFolder(startupPath, MAX_PATH)) {
        // 1. 获取当前自身 exe 程序的路径
        char exePath[MAX_PATH];
        GetModuleFileName(NULL, exePath, MAX_PATH);

        // 2. 设置快捷方式的保存路径 (启动文件夹内)
        string lnkPath = string(startupPath) + "\\MyDevApp.lnk";

        // 3. 构造 PowerShell 命令来静默创建快捷方式
        // 语法: powershell -Command "$s=(New-Object -ComObject WScript.Shell).CreateShortcut('快捷方式路径'); $s.TargetPath='目标exe路径'; $s.Save()"
        string cmd = "powershell -Command \"$s=(New-Object -ComObject WScript.Shell).CreateShortcut('" + lnkPath + "'); $s.TargetPath='" + exePath + "'; $s.Save()\"";

        // 4. 执行命令
        int result = system(cmd.c_str());

        if (result == 0) {
            cout << "成功设置开机自启！" << endl;
        } else {
            cout << "设置失败！" << endl;
        }
    } else {
        cout << "无法找到启动文件夹！" << endl;
    }
    
    return 0;
}