Hyprland/hyprctl/main.cpp

273 lines
6.8 KiB
C++
Raw Normal View History

2022-03-20 16:51:14 +01:00
#include <ctype.h>
2022-03-21 18:29:41 +01:00
#include <netdb.h>
#include <netinet/in.h>
2022-03-20 16:51:14 +01:00
#include <stdio.h>
#include <stdlib.h>
2022-03-21 18:29:41 +01:00
#include <string.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/un.h>
2022-03-20 16:51:14 +01:00
#include <unistd.h>
2022-07-19 14:27:52 +02:00
#include <ranges>
2022-07-19 14:28:37 +02:00
#include <algorithm>
2022-03-20 16:51:14 +01:00
#include <iostream>
2022-03-21 18:29:41 +01:00
#include <string>
#include <fstream>
2022-03-20 16:51:14 +01:00
#include <string>
2022-07-15 11:58:09 +02:00
#include <deque>
2022-03-20 16:51:14 +01:00
2022-07-15 11:58:09 +02:00
const std::string USAGE = R"#(usage: hyprctl [(opt)flags] [command] [(opt)args]
2022-03-20 16:51:14 +01:00
2022-07-12 23:21:00 +02:00
commands:
2022-03-20 16:51:14 +01:00
monitors
workspaces
clients
2022-03-30 22:40:53 +02:00
activewindow
layers
2022-06-02 13:59:33 +02:00
devices
2022-04-21 16:11:29 +02:00
dispatch
2022-04-21 16:56:27 +02:00
keyword
2022-04-22 18:14:25 +02:00
version
2022-06-27 13:42:20 +02:00
kill
2022-07-10 15:41:26 +02:00
splash
2022-07-02 18:27:44 +02:00
hyprpaper
2022-07-12 23:21:00 +02:00
reload
flags:
2022-07-13 15:50:46 +02:00
j -> output in JSON
2022-07-12 23:21:00 +02:00
)#";
2022-03-20 16:51:14 +01:00
2022-03-21 18:29:41 +01:00
void request(std::string arg) {
const auto SERVERSOCKET = socket(AF_UNIX, SOCK_STREAM, 0);
2022-03-20 16:51:14 +01:00
2022-03-21 18:29:41 +01:00
if (SERVERSOCKET < 0) {
std::cout << "Couldn't open a socket (1)";
return;
}
2022-03-20 16:51:14 +01:00
2022-03-21 18:29:41 +01:00
const auto SERVER = gethostbyname("localhost");
2022-03-20 16:51:14 +01:00
2022-03-21 18:29:41 +01:00
if (!SERVER) {
std::cout << "Couldn't get host (2)";
return;
}
2022-03-20 16:51:14 +01:00
// get the instance signature
auto instanceSig = getenv("HYPRLAND_INSTANCE_SIGNATURE");
if (!instanceSig) {
std::cout << "HYPRLAND_INSTANCE_SIGNATURE was not set! (Is Hyprland running?)";
return;
}
std::string instanceSigStr = std::string(instanceSig);
sockaddr_un serverAddress = {0};
serverAddress.sun_family = AF_UNIX;
std::string socketPath = "/tmp/hypr/" + instanceSigStr + "/.socket.sock";
strcpy(serverAddress.sun_path, socketPath.c_str());
2022-03-20 16:51:14 +01:00
if (connect(SERVERSOCKET, (sockaddr*)&serverAddress, SUN_LEN(&serverAddress)) < 0) {
std::cout << "Couldn't connect to " << socketPath << ". (3)";
2022-03-21 18:29:41 +01:00
return;
}
2022-03-20 16:51:14 +01:00
2022-03-21 18:29:41 +01:00
auto sizeWritten = write(SERVERSOCKET, arg.c_str(), arg.length());
2022-03-20 16:51:14 +01:00
2022-03-21 18:29:41 +01:00
if (sizeWritten < 0) {
std::cout << "Couldn't write (4)";
return;
}
2022-03-20 16:51:14 +01:00
2022-03-21 18:29:41 +01:00
char buffer[8192] = {0};
2022-03-20 16:51:14 +01:00
2022-07-02 18:27:44 +02:00
sizeWritten = read(SERVERSOCKET, buffer, 8192);
if (sizeWritten < 0) {
std::cout << "Couldn't read (5)";
return;
}
close(SERVERSOCKET);
std::cout << std::string(buffer);
}
void requestHyprpaper(std::string arg) {
const auto SERVERSOCKET = socket(AF_UNIX, SOCK_STREAM, 0);
if (SERVERSOCKET < 0) {
std::cout << "Couldn't open a socket (1)";
return;
}
const auto SERVER = gethostbyname("localhost");
if (!SERVER) {
std::cout << "Couldn't get host (2)";
return;
}
// get the instance signature
auto instanceSig = getenv("HYPRLAND_INSTANCE_SIGNATURE");
if (!instanceSig) {
std::cout << "HYPRLAND_INSTANCE_SIGNATURE was not set! (Is Hyprland running?)";
return;
}
std::string instanceSigStr = std::string(instanceSig);
sockaddr_un serverAddress = {0};
serverAddress.sun_family = AF_UNIX;
std::string socketPath = "/tmp/hypr/" + instanceSigStr + "/.hyprpaper.sock";
strcpy(serverAddress.sun_path, socketPath.c_str());
if (connect(SERVERSOCKET, (sockaddr*)&serverAddress, SUN_LEN(&serverAddress)) < 0) {
std::cout << "Couldn't connect to " << socketPath << ". (3)";
return;
}
auto sizeWritten = write(SERVERSOCKET, arg.c_str(), arg.length());
if (sizeWritten < 0) {
std::cout << "Couldn't write (4)";
return;
}
char buffer[8192] = {0};
sizeWritten = read(SERVERSOCKET, buffer, 8192);
2022-03-20 16:51:14 +01:00
2022-03-21 18:29:41 +01:00
if (sizeWritten < 0) {
std::cout << "Couldn't read (5)";
return;
}
2022-03-20 16:51:14 +01:00
2022-03-21 18:29:41 +01:00
close(SERVERSOCKET);
2022-03-20 16:51:14 +01:00
2022-03-21 18:29:41 +01:00
std::cout << std::string(buffer);
2022-03-20 16:51:14 +01:00
}
2022-04-21 16:11:29 +02:00
void dispatchRequest(int argc, char** argv) {
if (argc < 4) {
std::cout << "dispatch requires 2 params";
return;
}
std::string rq = "/dispatch " + std::string(argv[2]) + " " + std::string(argv[3]);
2022-04-21 16:11:29 +02:00
request(rq);
}
2022-04-21 16:56:27 +02:00
void keywordRequest(int argc, char** argv) {
if (argc < 4) {
std::cout << "keyword requires 2 params";
return;
}
std::string rq = "keyword " + std::string(argv[2]) + " " + std::string(argv[3]);
request(rq);
}
2022-07-02 18:27:44 +02:00
void hyprpaperRequest(int argc, char** argv) {
if (argc < 4) {
std::cout << "hyprpaper requires 2 params";
return;
}
std::string rq = std::string(argv[2]) + " " + std::string(argv[3]);
requestHyprpaper(rq);
}
2022-07-13 00:34:28 +02:00
void batchRequest(std::string arg) {
std::string rq = "[[BATCH]]" + arg.substr(arg.find_first_of(" ") + 1);
2022-04-29 19:44:09 +02:00
request(rq);
}
2022-07-15 11:58:09 +02:00
std::deque<std::string> splitArgs(int argc, char** argv) {
std::deque<std::string> result;
for (auto i = 1 /* skip the executable */; i < argc; ++i)
result.push_back(std::string(argv[i]));
return result;
}
2022-07-19 14:24:03 +02:00
bool isNumber(const std::string& str, bool allowfloat) {
return std::ranges::all_of(str.begin(), str.end(), [&](char c) { return isdigit(c) != 0 || c == '-' || (allowfloat && c == '.'); });
}
2022-03-20 16:51:14 +01:00
int main(int argc, char** argv) {
int bflag = 0, sflag = 0, index, c;
if (argc < 2) {
printf("%s\n", USAGE.c_str());
2022-03-20 16:51:14 +01:00
return 1;
}
2022-07-13 00:34:28 +02:00
std::string fullRequest = "";
2022-07-15 11:58:09 +02:00
std::string fullArgs = "";
const auto ARGS = splitArgs(argc, argv);
for (auto i = 0; i < ARGS.size(); ++i) {
2022-07-19 14:24:03 +02:00
if (ARGS[i][0] == '-' && !isNumber(ARGS[i], true) /* For stuff like -2 */) {
2022-07-15 11:58:09 +02:00
// parse
if (ARGS[i] == "-j" && !fullArgs.contains("j")) {
fullArgs += "j";
} else if (ARGS[i] == "--batch") {
fullRequest = "--batch ";
} else {
printf("%s\n", USAGE.c_str());
return 1;
}
continue;
}
fullRequest += ARGS[i] + " ";
2022-07-13 00:34:28 +02:00
}
2022-07-15 11:58:09 +02:00
if (fullRequest.empty()) {
printf("%s\n", USAGE.c_str());
return 1;
}
2022-07-13 00:34:28 +02:00
fullRequest.pop_back(); // remove trailing space
2022-07-12 23:17:55 +02:00
2022-07-15 11:58:09 +02:00
fullRequest = fullArgs + "/" + fullRequest;
2022-07-12 23:17:55 +02:00
2022-07-13 00:34:28 +02:00
if (fullRequest.contains("/--batch")) batchRequest(fullRequest);
else if (fullRequest.contains("/monitors")) request(fullRequest);
2022-07-12 23:17:55 +02:00
else if (fullRequest.contains("/clients")) request(fullRequest);
else if (fullRequest.contains("/workspaces")) request(fullRequest);
else if (fullRequest.contains("/activewindow")) request(fullRequest);
else if (fullRequest.contains("/layers")) request(fullRequest);
else if (fullRequest.contains("/version")) request(fullRequest);
else if (fullRequest.contains("/kill")) request(fullRequest);
else if (fullRequest.contains("/splash")) request(fullRequest);
else if (fullRequest.contains("/devices")) request(fullRequest);
else if (fullRequest.contains("/reload")) request(fullRequest);
else if (fullRequest.contains("/dispatch")) dispatchRequest(argc, argv);
else if (fullRequest.contains("/keyword")) keywordRequest(argc, argv);
else if (fullRequest.contains("/hyprpaper")) hyprpaperRequest(argc, argv);
else if (fullRequest.contains("/--help")) printf("%s", USAGE.c_str());
2022-03-20 16:51:14 +01:00
else {
printf("%s\n", USAGE.c_str());
2022-03-20 16:51:14 +01:00
return 1;
}
printf("\n");
2022-03-20 16:51:14 +01:00
return 0;
2022-06-17 22:02:57 +02:00
}