mirror of
https://github.com/hyprwm/Hyprland.git
synced 2025-08-15 12:03:48 -07:00
225 lines
8.4 KiB
C++
225 lines
8.4 KiB
C++
#include "helpers/Colors.hpp"
|
|
#include "helpers/StringUtils.hpp"
|
|
#include "core/PluginManager.hpp"
|
|
#include "core/DataState.hpp"
|
|
#include "helpers/Sys.hpp"
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
#include <print>
|
|
|
|
#include <hyprutils/utils/ScopeGuard.hpp>
|
|
using namespace Hyprutils::Utils;
|
|
|
|
constexpr std::string_view HELP = R"#(┏ hyprpm, a Hyprland Plugin Manager
|
|
┃
|
|
┣ add [url] [git rev] → Install a new plugin repository from git. Git revision
|
|
┃ is optional, when set, commit locks are ignored.
|
|
┣ remove [url/name] → Remove an installed plugin repository.
|
|
┣ enable [name] → Enable a plugin.
|
|
┣ disable [name] → Disable a plugin.
|
|
┣ update → Check and update all plugins if needed.
|
|
┣ reload → Reload hyprpm state. Ensure all enabled plugins are loaded.
|
|
┣ list → List all installed plugins.
|
|
┣ purge-cache → Remove the entire hyprpm cache, built plugins, hyprpm settings and headers.
|
|
┃
|
|
┣ Flags:
|
|
┃
|
|
┣ --notify | -n → Send a hyprland notification for important events (including both successes and fail events).
|
|
┣ --notify-fail | -nn → Send a hyprland notification for fail events only.
|
|
┣ --help | -h → Show this menu.
|
|
┣ --verbose | -v → Enable too much logging.
|
|
┣ --force | -f → Force an operation ignoring checks (e.g. update -f).
|
|
┣ --no-shallow | -s → Disable shallow cloning of Hyprland sources.
|
|
┣ --hl-url | → Pass a custom hyprland source url.
|
|
┗
|
|
)#";
|
|
|
|
int main(int argc, char** argv, char** envp) {
|
|
std::vector<std::string> ARGS{argc};
|
|
for (int i = 0; i < argc; ++i) {
|
|
ARGS[i] = std::string{argv[i]};
|
|
}
|
|
|
|
if (ARGS.size() < 2) {
|
|
std::println(stderr, "{}", HELP);
|
|
return 1;
|
|
}
|
|
|
|
std::vector<std::string> command;
|
|
bool notify = false, notifyFail = false, verbose = false, force = false, noShallow = false;
|
|
std::string customHlUrl;
|
|
|
|
for (int i = 1; i < argc; ++i) {
|
|
if (ARGS[i].starts_with("-")) {
|
|
if (ARGS[i] == "--help" || ARGS[i] == "-h") {
|
|
std::println("{}", HELP);
|
|
return 0;
|
|
} else if (ARGS[i] == "--notify" || ARGS[i] == "-n") {
|
|
notify = true;
|
|
} else if (ARGS[i] == "--notify-fail" || ARGS[i] == "-nn") {
|
|
notifyFail = notify = true;
|
|
} else if (ARGS[i] == "--verbose" || ARGS[i] == "-v") {
|
|
verbose = true;
|
|
} else if (ARGS[i] == "--no-shallow" || ARGS[i] == "-s") {
|
|
noShallow = true;
|
|
} else if (ARGS[i] == "--hl-url") {
|
|
if (i + 1 >= argc) {
|
|
std::println(stderr, "Missing argument for --hl-url");
|
|
return 1;
|
|
}
|
|
customHlUrl = ARGS[i + 1];
|
|
i++;
|
|
} else if (ARGS[i] == "--force" || ARGS[i] == "-f") {
|
|
force = true;
|
|
std::println("{}", statusString("!", Colors::RED, "Using --force, I hope you know what you are doing."));
|
|
} else {
|
|
std::println(stderr, "Unrecognized option {}", ARGS[i]);
|
|
return 1;
|
|
}
|
|
} else
|
|
command.push_back(ARGS[i]);
|
|
}
|
|
|
|
if (command.empty()) {
|
|
std::println(stderr, "{}", HELP);
|
|
return 1;
|
|
}
|
|
|
|
g_pPluginManager = std::make_unique<CPluginManager>();
|
|
g_pPluginManager->m_bVerbose = verbose;
|
|
g_pPluginManager->m_bNoShallow = noShallow;
|
|
g_pPluginManager->m_szCustomHlUrl = customHlUrl;
|
|
|
|
if (command[0] == "add") {
|
|
if (command.size() < 2) {
|
|
std::println(stderr, "{}", failureString("Not enough args for add."));
|
|
return 1;
|
|
}
|
|
|
|
std::string rev = "";
|
|
if (command.size() >= 3)
|
|
rev = command[2];
|
|
|
|
const auto HLVER = g_pPluginManager->getHyprlandVersion();
|
|
auto GLOBALSTATE = DataState::getGlobalState();
|
|
|
|
if (GLOBALSTATE.headersHashCompiled != HLVER.hash) {
|
|
std::println(stderr, "{}", failureString("Headers outdated, please run hyprpm update."));
|
|
return 1;
|
|
}
|
|
|
|
NSys::root::cacheSudo();
|
|
CScopeGuard x([] { NSys::root::dropSudo(); });
|
|
|
|
g_pPluginManager->updateHeaders(false);
|
|
|
|
return g_pPluginManager->addNewPluginRepo(command[1], rev) ? 0 : 1;
|
|
} else if (command[0] == "remove") {
|
|
if (command.size() < 2) {
|
|
std::println(stderr, "{}", failureString("Not enough args for remove."));
|
|
return 1;
|
|
}
|
|
|
|
NSys::root::cacheSudo();
|
|
CScopeGuard x([] { NSys::root::dropSudo(); });
|
|
|
|
return g_pPluginManager->removePluginRepo(command[1]) ? 0 : 1;
|
|
} else if (command[0] == "update") {
|
|
NSys::root::cacheSudo();
|
|
CScopeGuard x([] { NSys::root::dropSudo(); });
|
|
|
|
bool headersValid = g_pPluginManager->headersValid() == HEADERS_OK;
|
|
bool headers = g_pPluginManager->updateHeaders(force);
|
|
|
|
if (headers) {
|
|
const auto HLVER = g_pPluginManager->getHyprlandVersion(false);
|
|
auto GLOBALSTATE = DataState::getGlobalState();
|
|
const auto COMPILEDOUTDATED = HLVER.hash != GLOBALSTATE.headersHashCompiled;
|
|
|
|
bool ret1 = g_pPluginManager->updatePlugins(!headersValid || force || COMPILEDOUTDATED);
|
|
|
|
if (!ret1)
|
|
return 1;
|
|
|
|
auto ret2 = g_pPluginManager->ensurePluginsLoadState();
|
|
|
|
if (ret2 == LOADSTATE_HYPRLAND_UPDATED)
|
|
g_pPluginManager->notify(ICON_INFO, 0, 10000, "[hyprpm] Updated plugins, but Hyprland was updated. Please restart Hyprland.");
|
|
|
|
if (ret2 != LOADSTATE_OK)
|
|
return 1;
|
|
} else if (notify)
|
|
g_pPluginManager->notify(ICON_ERROR, 0, 10000, "[hyprpm] Couldn't update headers");
|
|
} else if (command[0] == "enable") {
|
|
if (command.size() < 2) {
|
|
std::println(stderr, "{}", failureString("Not enough args for enable."));
|
|
return 1;
|
|
}
|
|
|
|
if (!g_pPluginManager->enablePlugin(command[1])) {
|
|
std::println(stderr, "{}", failureString("Couldn't enable plugin (missing?)"));
|
|
return 1;
|
|
}
|
|
|
|
NSys::root::cacheSudo();
|
|
CScopeGuard x([] { NSys::root::dropSudo(); });
|
|
|
|
auto ret = g_pPluginManager->ensurePluginsLoadState();
|
|
|
|
if (ret == LOADSTATE_HYPRLAND_UPDATED)
|
|
g_pPluginManager->notify(ICON_INFO, 0, 10000, "[hyprpm] Enabled plugin, but Hyprland was updated. Please restart Hyprland.");
|
|
|
|
if (ret != LOADSTATE_OK)
|
|
return 1;
|
|
} else if (command[0] == "disable") {
|
|
if (command.size() < 2) {
|
|
std::println(stderr, "{}", failureString("Not enough args for disable."));
|
|
return 1;
|
|
}
|
|
|
|
if (!g_pPluginManager->disablePlugin(command[1])) {
|
|
std::println(stderr, "{}", failureString("Couldn't disable plugin (missing?)"));
|
|
return 1;
|
|
}
|
|
|
|
NSys::root::cacheSudo();
|
|
CScopeGuard x([] { NSys::root::dropSudo(); });
|
|
|
|
auto ret = g_pPluginManager->ensurePluginsLoadState();
|
|
|
|
if (ret != LOADSTATE_OK)
|
|
return 1;
|
|
} else if (command[0] == "reload") {
|
|
auto ret = g_pPluginManager->ensurePluginsLoadState(force);
|
|
|
|
if (ret != LOADSTATE_OK) {
|
|
if (notify) {
|
|
switch (ret) {
|
|
case LOADSTATE_FAIL:
|
|
case LOADSTATE_PARTIAL_FAIL: g_pPluginManager->notify(ICON_ERROR, 0, 10000, "[hyprpm] Failed to load plugins"); break;
|
|
case LOADSTATE_HEADERS_OUTDATED:
|
|
g_pPluginManager->notify(ICON_ERROR, 0, 10000, "[hyprpm] Failed to load plugins: Outdated headers. Please run hyprpm update manually.");
|
|
break;
|
|
default: break;
|
|
}
|
|
}
|
|
|
|
return 1;
|
|
} else if (notify && !notifyFail) {
|
|
g_pPluginManager->notify(ICON_OK, 0, 4000, "[hyprpm] Loaded plugins");
|
|
}
|
|
} else if (command[0] == "purge-cache") {
|
|
NSys::root::cacheSudo();
|
|
CScopeGuard x([] { NSys::root::dropSudo(); });
|
|
DataState::purgeAllCache();
|
|
} else if (command[0] == "list") {
|
|
g_pPluginManager->listAllPlugins();
|
|
} else {
|
|
std::println(stderr, "{}", HELP);
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|