SendToDesktopShortcut11.cpp

#define INITGUID

#include "common.h"

#include <shlobj.h>   // SHGetKnownFolderPath
#include <shobjidl.h> // IShellLink
#include <objbase.h>
#include <knownfolders.h>
#include <string>

std::wstring GetDesktopPath()
{
    PWSTR path = nullptr;

    HRESULT hr = SHGetKnownFolderPath(FOLDERID_Desktop, 0, NULL, &path);
    if (FAILED(hr))
        return L"";

    std::wstring result(path);
    CoTaskMemFree(path);
    return result;
}

std::wstring GetFileName(const std::wstring &fullPath)
{
    if (fullPath.empty())
        return fullPath;

    std::wstring path = fullPath;

    // Remove trailing slashes
    while (path.length() > 1 &&
           (path.back() == L'\\' || path.back() == L'/'))
    {
        // Keep paths like "C:\"
        if (path.length() == 3 &&
            path[1] == L':' &&
            (path[2] == L'\\' || path[2] == L'/'))
        {
            break;
        }

        path.pop_back();
    }

    // "C:" -> "C"
    if (path.length() == 2 && path[1] == L':')
    {
        return path.substr(0, 1);
    }

    // "C:\" -> "C"
    if (path.length() == 3 &&
        path[1] == L':' &&
        (path[2] == L'\\' || path[2] == L'/'))
    {
        return path.substr(0, 1);
    }

    // "\\mynas" -> "mynas"
    if (path.size() >= 2 &&
        path[0] == L'\\' &&
        path[1] == L'\\' &&
        path.find(L'\\', 2) == std::wstring::npos)
    {
        return path.substr(2);
    }

    size_t pos = path.find_last_of(L"\\/");

    if (pos == std::wstring::npos)
        return path;

    return path.substr(pos + 1);
}

bool CreateShortcut(const std::wstring &targetPath)
{
    HRESULT hr = CoInitialize(nullptr);
    if (FAILED(hr))
        return false;

    IShellLinkW *pShellLink = nullptr;

    hr = CoCreateInstance(
        CLSID_ShellLink,
        nullptr,
        CLSCTX_INPROC_SERVER,
        IID_IShellLinkW,
        (void **)&pShellLink);

    if (FAILED(hr))
    {
        CoUninitialize();
        return false;
    }

    pShellLink->SetPath(targetPath.c_str());

    IPersistFile *pPersistFile = nullptr;
    hr = pShellLink->QueryInterface(IID_IPersistFile, (void **)&pPersistFile);

    if (SUCCEEDED(hr))
    {
        std::wstring desktop = GetDesktopPath();
        std::wstring name = GetFileName(targetPath);

        size_t dot = name.find_last_of(L'.');
        if (dot != std::wstring::npos && name.substr(dot) == L".exe") // .exe only
        {
            name = name.substr(0, dot);
        }

        std::wstring shortcutPath = desktop + L"\\" + name + L".lnk";

        hr = pPersistFile->Save(shortcutPath.c_str(), TRUE);
        pPersistFile->Release();
    }

    pShellLink->Release();
    CoUninitialize();

    return SUCCEEDED(hr);
}

int wmain(int argc, wchar_t *argv[])
{
    if (argc < 2)
    {
        std::wstring aboutMessage = L"";

        aboutMessage += L"SendToDesktopShortcut11";
        aboutMessage += L" 0.3.0\n";

        if (GetUserDefaultUILanguage() == 2052)
        {
            aboutMessage += L"为 Windows 11 设计的右键菜单扩展，可将选中的文件或文件夹发送到桌面快捷方式\n\n";
            aboutMessage += L"仓库地址：https://github.com/SJEllipses/SendToDesktopShortcut11\n\n";

            aboutMessage += L"当前右键菜单拓展状态：";
            aboutMessage += IsExtensionEnabled() ? L"已启用\n\n" : L"已禁用\n\n";
            
            aboutMessage += L"设置右键菜单拓展状态：\n";
            aboutMessage += L"是 (Y) = 启用\n否 (N) = 禁用\n取消 = 不做更改";
        }
        else
        {
            aboutMessage += L"Context menu extension designed for Windows 11, which can send selected files or folders to desktop shortcuts\n\n";
            aboutMessage += L"Repo: https://github.com/SJEllipses/SendToDesktopShortcut11\n";

            aboutMessage += L"Current context menu extension status: ";
            aboutMessage += IsExtensionEnabled() ? L"Enabled\n\n" : L"Disabled\n\n";

            aboutMessage += L"Set context menu extension status: \n";
            aboutMessage += L"Yes (Y) = Enable\nNo (N) = Disable\nCancel = Do not change";
        }

        int choice = MessageBoxW(nullptr, aboutMessage.c_str(), L"SendToDesktopShortcut11", MB_YESNOCANCEL);
        if (choice == IDYES)
        {
            if(SetExtensionEnabled(true) == ERROR_SUCCESS)
            {
                if (GetUserDefaultUILanguage() == 2052)
                    MessageBoxW(nullptr, L"右键菜单拓展已启用", L"SendToDesktopShortcut11", MB_OK);
                else
                    MessageBoxW(nullptr, L"Context menu extension enabled", L"SendToDesktopShortcut11", MB_OK);
            }
            else
            {
                if (GetUserDefaultUILanguage() == 2052)
                    MessageBoxW(nullptr, L"右键菜单拓展启用失败", L"SendToDesktopShortcut11", MB_OK);
                else
                    MessageBoxW(nullptr, L"Failed to enable context menu extension", L"SendToDesktopShortcut11", MB_OK);
            }
        }
        else if (choice == IDNO)
        {
            if(SetExtensionEnabled(false) == ERROR_SUCCESS)
            {
                if (GetUserDefaultUILanguage() == 2052)
                    MessageBoxW(nullptr, L"右键菜单拓展已禁用", L"SendToDesktopShortcut11", MB_OK);
                else
                    MessageBoxW(nullptr, L"Context menu extension disabled", L"SendToDesktopShortcut11", MB_OK);
            }
            else
            {
                if (GetUserDefaultUILanguage() == 2052)
                    MessageBoxW(nullptr, L"右键菜单拓展禁用失败", L"SendToDesktopShortcut11", MB_OK);
                else
                    MessageBoxW(nullptr, L"Failed to disable context menu extension", L"SendToDesktopShortcut11", MB_OK);
            }
        }

        return 1;
    }
    else
    {
        for (int i = 1; i < argc; i++)
        {
            HRESULT hr = CreateShortcut(argv[i]);
            if (!hr)
            {
                if (GetUserDefaultUILanguage() == 2052)
                    MessageBoxW(nullptr, L"  创建快捷方式失败", L"SendToDesktopShortcut11", MB_ICONERROR | MB_OK);
                else
                    MessageBoxW(nullptr, L"  Failed to create shortcut.", L"SendToDesktopShortcut11", MB_ICONERROR | MB_OK);
                return hr;
            }
        }

        return 0;
    }
}