#include "Sys.hpp"
#include "Die.hpp"
#include "StringUtils.hpp"
#include <pwd.h>
#include <unistd.h>
#include <print>
#include <filesystem>

#include <hyprutils/os/Process.hpp>
#include <hyprutils/string/VarList.hpp>
using namespace Hyprutils::OS;
using namespace Hyprutils::String;

static const std::vector<const char*> SUPERUSER_BINARIES = {
    "sudo",
    "doas",
    "run0",
};

static bool executableExistsInPath(const std::string& exe) {
    if (!getenv("PATH"))
        return false;

    static CVarList paths(getenv("PATH"), 0, ':', true);

    for (auto& p : paths) {
        std::string     path = p + std::string{"/"} + exe;
        std::error_code ec;
        if (!std::filesystem::exists(path, ec) || ec)
            continue;

        if (!std::filesystem::is_regular_file(path, ec) || ec)
            continue;

        auto stat = std::filesystem::status(path, ec);
        if (ec)
            continue;

        auto perms = stat.permissions();

        return std::filesystem::perms::none != (perms & std::filesystem::perms::others_exec);
    }

    return false;
}

static std::pair<std::string, int> execAndGet(std::string cmd, bool noRedirect = false) {
    if (!noRedirect)
        cmd += " 2>&1";

    CProcess proc("/bin/sh", {"-c", cmd});

    if (!proc.runSync())
        return {"error", 1};

    return {proc.stdOut(), proc.exitCode()};
}

int NSys::getUID() {
    const auto UID   = getuid();
    const auto PWUID = getpwuid(UID);
    return PWUID ? PWUID->pw_uid : UID;
}

int NSys::getEUID() {
    const auto UID   = geteuid();
    const auto PWUID = getpwuid(UID);
    return PWUID ? PWUID->pw_uid : UID;
}

bool NSys::isSuperuser() {
    return getuid() != geteuid() || !geteuid();
}

std::string NSys::runAsSuperuser(const std::string& cmd) {
    for (const auto& SB : SUPERUSER_BINARIES) {
        if (!executableExistsInPath(SB))
            continue;

        const auto RESULT = execAndGet(std::string{SB} + " /bin/sh -c \"" + cmd + "\"", true);

        if (RESULT.second != 0)
            Debug::die("Failed to run a command as sudo. This could be due to an invalid password, or a hyprpm bug.");

        return RESULT.first;
    }

    Debug::die("Failed to find a superuser binary. Supported: sudo, doas, run0.");
    return "";
}

void NSys::cacheSudo() {
    // "caches" the sudo so that the prompt later doesn't pop up in a weird spot
    // sudo will not ask us again for a moment
    runAsSuperuser("echo e > /dev/null");
}

void NSys::dropSudo() {
    for (const auto& SB : SUPERUSER_BINARIES) {
        if (!executableExistsInPath(SB))
            continue;

        if (SB == std::string_view{"sudo"})
            execAndGet("sudo -k");
        else
            std::println("{}", infoString("Don't know how to drop timestamp for {}, ignoring.", SB));

        return;
    }
}