HomeDir.h

// SPDX-License-Identifier: GPL-2.0-only

#pragma once

#include <filesystem>
#include <pwd.h>
#include <sys/types.h>
#include <unistd.h>

namespace SlHelpers {

class HomeDir {
public:
    HomeDir() = delete;

    static std::filesystem::path get() noexcept {
        std::filesystem::path dir;

        if (const auto homeDir = std::getenv("HOME")) {
            dir = homeDir;
            if (std::filesystem::exists(dir))
                return dir;
        }

        dir = ::getpwuid(::getuid())->pw_dir;
        if (std::filesystem::exists(dir))
            return dir;

        return {};
    }

    static std::filesystem::path getCacheDir() noexcept {
        if (const auto xdgCacheDir = std::getenv("XDG_CACHE_HOME"))
            return xdgCacheDir;

        const auto home_dir = get();
        if (home_dir.empty())
            return {};

        return std::filesystem::path(home_dir) / ".cache";
    }

    static std::filesystem::path createCacheDir(const std::filesystem::path &subdir) noexcept {
        auto cache = getCacheDir();
        if (cache.empty())
            return {};
        cache /= subdir;

        std::error_code ec;
        std::filesystem::create_directories(cache, ec);
        if (ec)
            return {};

        return cache;
    }
};

}