AudioCppTray/AutoStart.cpp at main · spicchio72/AudioCppTray · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include "AutoStart.h"

#include <windows.h>
#include <shlobj.h>
#include <string>


std::wstring GetStartupPath()
{
    wchar_t path[MAX_PATH];

    SHGetFolderPathW(
        nullptr,
        CSIDL_STARTUP,
        nullptr,
        0,
        path
    );

    return std::wstring(path);
}


std::wstring GetShortcutPath()
{
    return GetStartupPath() +
        L"\\AudioCppTray.lnk";
}


bool IsAutoStartEnabled()
{
    DWORD attr = GetFileAttributesW(
        GetShortcutPath().c_str()
    );

    return attr != INVALID_FILE_ATTRIBUTES;
}


void EnableAutoStart()
{
    wchar_t exe[MAX_PATH];

    GetModuleFileNameW(
        nullptr,
        exe,
        MAX_PATH
    );


    CoInitialize(nullptr);


    IShellLinkW* link = nullptr;


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


    if (SUCCEEDED(hr))
    {
        link->SetPath(exe);

        IPersistFile* file = nullptr;

        link->QueryInterface(
            IID_IPersistFile,
            (void**)&file
        );


        if (file)
        {
            file->Save(
                GetShortcutPath().c_str(),
                TRUE
            );

            file->Release();
        }

        link->Release();
    }


    CoUninitialize();
}


void DisableAutoStart()
{
    DeleteFileW(
        GetShortcutPath().c_str()
    );
}