[WIN32] Enumerate all .lnk files in the StartMenu folders.

EnumStartMenuLinkFiles.cpp

// Enumerate all .lnk files in the StartMenu folders.
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include <shlobj.h>         // REFKNOWNFOLDERID, SHGetKnownFolderPath
#include <functional>
#include <filesystem>

using EnumStartMenuLnkFilesCallbackFunc = std::function<void(const std::filesystem::path& rootPath, const std::filesystem::path& entryPath)>;

void enumStartMenuLnkFiles(const EnumStartMenuLnkFilesCallbackFunc& callback) {
    const auto getKnownFolderPath = [](REFKNOWNFOLDERID id, DWORD flags = 0) -> std::wstring {
        std::wstring result;
        wchar_t* path = nullptr;
        const auto hr = SHGetKnownFolderPath(id, flags, nullptr, &path);
        if(SUCCEEDED(hr)) {
            result = std::wstring(path);
        }
        if(path) { CoTaskMemFree(path); }
        return result;
    };

    namespace fs = std::filesystem;

    const auto rootPaths = std::vector<fs::path> {
        fs::path(getKnownFolderPath(FOLDERID_CommonStartMenu)),
        fs::path(getKnownFolderPath(FOLDERID_StartMenu))
    };

    for(const auto& rootPath : rootPaths) {
        for(auto& p: fs::recursive_directory_iterator(rootPath)) {
            const auto path = p.path();
            if(fs::is_regular_file(p) && path.extension().compare(fs::path(".lnk")) == 0) {
                callback(rootPath, path);
            }
        }
    }
}

int main() {
    namespace fs = std::filesystem;
    setlocale(LC_ALL, "");

    enumStartMenuLnkFiles([&](const std::filesystem::path& rootPath, const std::filesystem::path& entryPath) {
        const auto relPath = fs::relative(entryPath, rootPath);
        wprintf(L"%s\n", relPath.c_str());
    });
}