From 75bf253a7896d3a686ae8dcf70a34133b168d303 Mon Sep 17 00:00:00 2001 From: coast Date: Sun, 21 Dec 2025 10:21:49 +0330 Subject: init --- LICENSE | 202 +++++++++++++++++++++++++++++++ README.md | 18 +++ makefile | 31 +++++ pissrc.example.lua | 47 ++++++++ src/daemon.c | 349 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 5 files changed, 647 insertions(+) create mode 100644 LICENSE create mode 100644 README.md create mode 100644 makefile create mode 100755 pissrc.example.lua create mode 100644 src/daemon.c diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..d645695 --- /dev/null +++ b/LICENSE @@ -0,0 +1,202 @@ + + Apache License + Version 2.0, January 2004 + http://www.apache.org/licenses/ + + TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION + + 1. Definitions. + + "License" shall mean the terms and conditions for use, reproduction, + and distribution as defined by Sections 1 through 9 of this document. + + "Licensor" shall mean the copyright owner or entity authorized by + the copyright owner that is granting the License. + + "Legal Entity" shall mean the union of the acting entity and all + other entities that control, are controlled by, or are under common + control with that entity. For the purposes of this definition, + "control" means (i) the power, direct or indirect, to cause the + direction or management of such entity, whether by contract or + otherwise, or (ii) ownership of fifty percent (50%) or more of the + outstanding shares, or (iii) beneficial ownership of such entity. + + "You" (or "Your") shall mean an individual or Legal Entity + exercising permissions granted by this License. + + "Source" form shall mean the preferred form for making modifications, + including but not limited to software source code, documentation + source, and configuration files. + + "Object" form shall mean any form resulting from mechanical + transformation or translation of a Source form, including but + not limited to compiled object code, generated documentation, + and conversions to other media types. + + "Work" shall mean the work of authorship, whether in Source or + Object form, made available under the License, as indicated by a + copyright notice that is included in or attached to the work + (an example is provided in the Appendix below). + + "Derivative Works" shall mean any work, whether in Source or Object + form, that is based on (or derived from) the Work and for which the + editorial revisions, annotations, elaborations, or other modifications + represent, as a whole, an original work of authorship. For the purposes + of this License, Derivative Works shall not include works that remain + separable from, or merely link (or bind by name) to the interfaces of, + the Work and Derivative Works thereof. + + "Contribution" shall mean any work of authorship, including + the original version of the Work and any modifications or additions + to that Work or Derivative Works thereof, that is intentionally + submitted to Licensor for inclusion in the Work by the copyright owner + or by an individual or Legal Entity authorized to submit on behalf of + the copyright owner. For the purposes of this definition, "submitted" + means any form of electronic, verbal, or written communication sent + to the Licensor or its representatives, including but not limited to + communication on electronic mailing lists, source code control systems, + and issue tracking systems that are managed by, or on behalf of, the + Licensor for the purpose of discussing and improving the Work, but + excluding communication that is conspicuously marked or otherwise + designated in writing by the copyright owner as "Not a Contribution." + + "Contributor" shall mean Licensor and any individual or Legal Entity + on behalf of whom a Contribution has been received by Licensor and + subsequently incorporated within the Work. + + 2. Grant of Copyright License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + copyright license to reproduce, prepare Derivative Works of, + publicly display, publicly perform, sublicense, and distribute the + Work and such Derivative Works in Source or Object form. + + 3. Grant of Patent License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + (except as stated in this section) patent license to make, have made, + use, offer to sell, sell, import, and otherwise transfer the Work, + where such license applies only to those patent claims licensable + by such Contributor that are necessarily infringed by their + Contribution(s) alone or by combination of their Contribution(s) + with the Work to which such Contribution(s) was submitted. If You + institute patent litigation against any entity (including a + cross-claim or counterclaim in a lawsuit) alleging that the Work + or a Contribution incorporated within the Work constitutes direct + or contributory patent infringement, then any patent licenses + granted to You under this License for that Work shall terminate + as of the date such litigation is filed. + + 4. Redistribution. You may reproduce and distribute copies of the + Work or Derivative Works thereof in any medium, with or without + modifications, and in Source or Object form, provided that You + meet the following conditions: + + (a) You must give any other recipients of the Work or + Derivative Works a copy of this License; and + + (b) You must cause any modified files to carry prominent notices + stating that You changed the files; and + + (c) You must retain, in the Source form of any Derivative Works + that You distribute, all copyright, patent, trademark, and + attribution notices from the Source form of the Work, + excluding those notices that do not pertain to any part of + the Derivative Works; and + + (d) If the Work includes a "NOTICE" text file as part of its + distribution, then any Derivative Works that You distribute must + include a readable copy of the attribution notices contained + within such NOTICE file, excluding those notices that do not + pertain to any part of the Derivative Works, in at least one + of the following places: within a NOTICE text file distributed + as part of the Derivative Works; within the Source form or + documentation, if provided along with the Derivative Works; or, + within a display generated by the Derivative Works, if and + wherever such third-party notices normally appear. The contents + of the NOTICE file are for informational purposes only and + do not modify the License. You may add Your own attribution + notices within Derivative Works that You distribute, alongside + or as an addendum to the NOTICE text from the Work, provided + that such additional attribution notices cannot be construed + as modifying the License. + + You may add Your own copyright statement to Your modifications and + may provide additional or different license terms and conditions + for use, reproduction, or distribution of Your modifications, or + for any such Derivative Works as a whole, provided Your use, + reproduction, and distribution of the Work otherwise complies with + the conditions stated in this License. + + 5. Submission of Contributions. Unless You explicitly state otherwise, + any Contribution intentionally submitted for inclusion in the Work + by You to the Licensor shall be under the terms and conditions of + this License, without any additional terms or conditions. + Notwithstanding the above, nothing herein shall supersede or modify + the terms of any separate license agreement you may have executed + with Licensor regarding such Contributions. + + 6. Trademarks. This License does not grant permission to use the trade + names, trademarks, service marks, or product names of the Licensor, + except as required for reasonable and customary use in describing the + origin of the Work and reproducing the content of the NOTICE file. + + 7. Disclaimer of Warranty. Unless required by applicable law or + agreed to in writing, Licensor provides the Work (and each + Contributor provides its Contributions) on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or + implied, including, without limitation, any warranties or conditions + of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A + PARTICULAR PURPOSE. You are solely responsible for determining the + appropriateness of using or redistributing the Work and assume any + risks associated with Your exercise of permissions under this License. + + 8. Limitation of Liability. In no event and under no legal theory, + whether in tort (including negligence), contract, or otherwise, + unless required by applicable law (such as deliberate and grossly + negligent acts) or agreed to in writing, shall any Contributor be + liable to You for damages, including any direct, indirect, special, + incidental, or consequential damages of any character arising as a + result of this License or out of the use or inability to use the + Work (including but not limited to damages for loss of goodwill, + work stoppage, computer failure or malfunction, or any and all + other commercial damages or losses), even if such Contributor + has been advised of the possibility of such damages. + + 9. Accepting Warranty or Additional Liability. While redistributing + the Work or Derivative Works thereof, You may choose to offer, + and charge a fee for, acceptance of support, warranty, indemnity, + or other liability obligations and/or rights consistent with this + License. However, in accepting such obligations, You may act only + on Your own behalf and on Your sole responsibility, not on behalf + of any other Contributor, and only if You agree to indemnify, + defend, and hold each Contributor harmless for any liability + incurred by, or claims asserted against, such Contributor by reason + of your accepting any such warranty or additional liability. + + END OF TERMS AND CONDITIONS + + APPENDIX: How to apply the Apache License to your work. + + To apply the Apache License to your work, attach the following + boilerplate notice, with the fields enclosed by brackets "[]" + replaced with your own identifying information. (Don't include + the brackets!) The text should be enclosed in the appropriate + comment syntax for the file format. We also recommend that a + file or class name and description of purpose be included on the + same "printed page" as the copyright notice for easier + identification within third-party archives. + + Copyright [yyyy] [name of copyright owner] + + Licensed under the Apache License, Version 2.0 (the "License"); + you may not use this file except in compliance with the License. + You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + + Unless required by applicable law or agreed to in writing, software + distributed under the License is distributed on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + See the License for the specific language governing permissions and + limitations under the License. diff --git a/README.md b/README.md new file mode 100644 index 0000000..7dcd4de --- /dev/null +++ b/README.md @@ -0,0 +1,18 @@ +Simple X hotkey daemon, with a `bind` argument + + +The config file is a script that calls `piss bind` commands. Each bind consists of a key combination in quotes. + +Config example: +```sh +#!/usr/bin/env sh +piss bind "Super + Return" "st" +``` +Or alternatively, in Lua: +```lua +#!/usr/bin/env lua +os.execute('piss bind "Super + Return" "st"') +``` + +The daemon automatically detects the shebang and executes your config! +My config can be seen [here.](pissrc.example.lua) diff --git a/makefile b/makefile new file mode 100644 index 0000000..6fbd1c1 --- /dev/null +++ b/makefile @@ -0,0 +1,31 @@ +PREFIX = /usr/local + +X11INC = /usr/X11R6/include +X11LIB = /usr/X11R6/lib + +INCS = -I$(X11INC) +LIBS = -L$(X11LIB) -lX11 + +CPPFLAGS = -D_DEFAULT_SOURCE -D_BSD_SOURCE -D_POSIX_C_SOURCE=200809L +CFLAGS = -std=c99 -pedantic -Wall -Wextra -Os $(INCS) $(CPPFLAGS) +LDFLAGS = $(LIBS) + +CC = cc + +all: piss + +piss: src/daemon.c + $(CC) $(CFLAGS) -o piss src/daemon.c $(LDFLAGS) + +clean: + rm -f piss + +install: all + mkdir -p $(DESTDIR)$(PREFIX)/bin + cp -f piss $(DESTDIR)$(PREFIX)/bin + chmod 755 $(DESTDIR)$(PREFIX)/bin/piss + +uninstall: + rm -f $(DESTDIR)$(PREFIX)/bin/piss + +.PHONY: all clean install uninstall diff --git a/pissrc.example.lua b/pissrc.example.lua new file mode 100755 index 0000000..3097c79 --- /dev/null +++ b/pissrc.example.lua @@ -0,0 +1,47 @@ +#!/usr/bin/lua + +-- NOTE THAT THE CONFIG DOES __NOT__ HAVE TO BE IN LUA. Do whatever you want! + +local function bind(key, cmd) + os.execute(string.format('piss bind "%s" "%s"', key, cmd)) +end + +bind("Super + Ctrl + l", "slock") +bind("Super + Return", "st") +bind("Super + s", "bspc node -c") +bind("Super + Ctrl + r", "bspc wm -r") + +for _, dir in ipairs({ + { "h", "west" }, + { "j", "south" }, + { "k", "north" }, + { "l", "east" } +}) do + bind("Super + " .. dir[1], "bspc node -f " .. dir[2]) +end + +for _, dir in ipairs({ + { "h", "west" }, + { "j", "south" }, + { "k", "north" }, + { "l", "east" } +}) do + bind("Super + Shift + " .. dir[1], "bspc node -s " .. dir[2]) +end + +bind("Super + p", "pkill picom && picom --backend glx & disown") +bind("Super + Shift + p", "pkill piss && piss ~/.config/piss/pissrc & disown") + +local desktops = { "I", "II", "III", "IV", "V", "VI", "VII", "VIII", "IX" } +for i = 1, 9 do + bind("Super + " .. i, "bspc desktop -f " .. desktops[i]) + bind("Super + Shift + " .. i, "bspc node -d " .. desktops[i]) +end + +bind("Super + r", "rofi -show drun -config ~/.config/rofi/gruvbox.rasi") +bind("Super + Shift + Return", "rofi -show run -config ~/.config/rofi/gruvbox.rasi") +bind("Super + Shift + q", "bspc quit") +bind("Super + Shift + s", "flameshot gui") +bind("Super + f", "pcmanfm") +bind("Super + space", "bspc node -t floating") +bind("Super + Shift + space", "bspc node -t tiled") diff --git a/src/daemon.c b/src/daemon.c new file mode 100644 index 0000000..40ed473 --- /dev/null +++ b/src/daemon.c @@ -0,0 +1,349 @@ +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#define MAX_BINDS 256 +#define MAX_CMD_LEN 512 +#define SOCK_PATH "/tmp/piss.sock" + +typedef struct { + unsigned int mod; + KeySym keysym; + char cmd[MAX_CMD_LEN]; +} Binding; + +static Display *dpy; +static Window root; +static Binding bindings[MAX_BINDS]; +static int bind_count = 0; + +unsigned int parse_modifiers(const char *mod_str) { + unsigned int mod = 0; + char *str = strdup(mod_str); + char *token = strtok(str, "+"); + + while (token) { + while (*token == ' ') + token++; + char *end = token + strlen(token) - 1; + while (end > token && *end == ' ') + end--; + *(end + 1) = '\0'; + + if (strcasecmp(token, "super") == 0 || strcasecmp(token, "mod4") == 0) + mod |= Mod4Mask; + else if (strcasecmp(token, "shift") == 0) + mod |= ShiftMask; + else if (strcasecmp(token, "ctrl") == 0 || + strcasecmp(token, "control") == 0) + mod |= ControlMask; + else if (strcasecmp(token, "alt") == 0 || strcasecmp(token, "mod1") == 0) + mod |= Mod1Mask; + + token = strtok(NULL, "+"); + } + + free(str); + return mod; +} + +KeySym parse_key(const char *key_str) { + char *str = strdup(key_str); + char *s = str; + while (*s == ' ') + s++; + char *end = s + strlen(s) - 1; + while (end > s && *end == ' ') + end--; + *(end + 1) = '\0'; + + KeySym ks = XStringToKeysym(s); + free(str); + return ks; +} + +void add_binding(const char *keycombo, const char *cmd) { + if (bind_count >= MAX_BINDS) { + fprintf(stderr, "piss: max bindings reached\n"); + return; + } + + char *combo = strdup(keycombo); + char *end = combo + strlen(combo) - 1; + while (end > combo && (*end == ' ' || *end == '\t' || *end == '\n')) { + *end = '\0'; + end--; + } + + char *plus = strrchr(combo, '+'); + + if (!plus) { + fprintf(stderr, "piss: invalid key combo: %s\n", keycombo); + free(combo); + return; + } + + *plus = '\0'; + char *mod_str = combo; + char *key_str = plus + 1; + + bindings[bind_count].mod = parse_modifiers(mod_str); + bindings[bind_count].keysym = parse_key(key_str); + + if (bindings[bind_count].keysym == NoSymbol) { + fprintf(stderr, "piss: invalid key: %s (from %s)\n", key_str, keycombo); + free(combo); + return; + } + + strncpy(bindings[bind_count].cmd, cmd, MAX_CMD_LEN - 1); + bindings[bind_count].cmd[MAX_CMD_LEN - 1] = '\0'; + + KeyCode kc = XKeysymToKeycode(dpy, bindings[bind_count].keysym); + if (kc != 0) { + XSync(dpy, False); + + XGrabKey(dpy, kc, bindings[bind_count].mod, root, True, GrabModeAsync, + GrabModeAsync); + XGrabKey(dpy, kc, bindings[bind_count].mod | Mod2Mask, root, True, + GrabModeAsync, GrabModeAsync); + XGrabKey(dpy, kc, bindings[bind_count].mod | LockMask, root, True, + GrabModeAsync, GrabModeAsync); + XGrabKey(dpy, kc, bindings[bind_count].mod | Mod2Mask | LockMask, root, + True, GrabModeAsync, GrabModeAsync); + + XSync(dpy, False); + + fprintf(stderr, "piss: bound [%s] keysym=%lu keycode=%u mod=%u -> %s\n", + keycombo, bindings[bind_count].keysym, kc, bindings[bind_count].mod, + cmd); + + bind_count++; + } else { + fprintf(stderr, "piss: couldn't get keycode for: %s\n", keycombo); + } + + free(combo); +} + +void execute_command(const char *cmd) { + if (fork() == 0) { + setsid(); + execl("/bin/sh", "sh", "-c", cmd, NULL); + exit(0); + } +} + +void handle_keypress(XEvent *ev) { + XKeyEvent *e = &ev->xkey; + KeySym ks = XkbKeycodeToKeysym(dpy, e->keycode, 0, 0); + unsigned int cleanmask = e->state & ~(Mod2Mask | LockMask); + + fprintf(stderr, "piss: key pressed: keysym=%lu mod=%u\n", ks, cleanmask); + + for (int i = 0; i < bind_count; i++) { + if (bindings[i].keysym == ks && bindings[i].mod == cleanmask) { + fprintf(stderr, "piss: executing: %s\n", bindings[i].cmd); + execute_command(bindings[i].cmd); + break; + } + } +} + +char *get_default_config() { + static char path[512]; + const char *home = getenv("HOME"); + if (!home) { + struct passwd *pw = getpwuid(getuid()); + home = pw->pw_dir; + } + snprintf(path, sizeof(path), "%s/.config/piss/pissrc", home); + return path; +} + +int send_bind_to_daemon(const char *keycombo, const char *cmd) { + int sock; + struct sockaddr_un addr; + char msg[MAX_CMD_LEN * 2]; + + sock = socket(AF_UNIX, SOCK_STREAM, 0); + if (sock < 0) { + return 1; + } + + memset(&addr, 0, sizeof(addr)); + addr.sun_family = AF_UNIX; + strncpy(addr.sun_path, SOCK_PATH, sizeof(addr.sun_path) - 1); + + if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) { + close(sock); + return 1; + } + + snprintf(msg, sizeof(msg), "%s|||%s", keycombo, cmd); + write(sock, msg, strlen(msg)); + close(sock); + return 0; +} + +void handle_socket_connection(int client_sock) { + char buf[MAX_CMD_LEN * 2]; + int n = read(client_sock, buf, sizeof(buf) - 1); + if (n > 0) { + buf[n] = '\0'; + + char *sep = strstr(buf, "|||"); + if (sep) { + *sep = '\0'; + char *keycombo = buf; + char *cmd = sep + 3; + add_binding(keycombo, cmd); + } + } + close(client_sock); +} + +int main(int argc, char *argv[]) { + if (argc >= 4 && strcmp(argv[1], "bind") == 0) { + return send_bind_to_daemon(argv[2], argv[3]); + } + + char *config = (argc >= 2) ? argv[1] : get_default_config(); + + dpy = XOpenDisplay(NULL); + if (!dpy) { + fprintf(stderr, "piss: cannot open display\n"); + return 1; + } + + root = DefaultRootWindow(dpy); + + int sock, client_sock; + struct sockaddr_un addr; + + unlink(SOCK_PATH); + sock = socket(AF_UNIX, SOCK_STREAM, 0); + if (sock < 0) { + perror("socket"); + return 1; + } + + memset(&addr, 0, sizeof(addr)); + addr.sun_family = AF_UNIX; + strncpy(addr.sun_path, SOCK_PATH, sizeof(addr.sun_path) - 1); + + if (bind(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) { + perror("bind"); + return 1; + } + + if (listen(sock, 5) < 0) { + perror("listen"); + return 1; + } + + char piss_path[512]; + ssize_t len = readlink("/proc/self/exe", piss_path, sizeof(piss_path) - 1); + if (len != -1) { + piss_path[len] = '\0'; + setenv("PISS_BIN", piss_path, 1); + } + + pid_t config_pid = fork(); + if (config_pid == 0) { + usleep(50000); + + FILE *f = fopen(config, "r"); + if (f) { + char line[256]; + if (fgets(line, sizeof(line), f) && strncmp(line, "#!", 2) == 0) { + fclose(f); + execl(config, config, NULL); + } else { + fclose(f); + execl("/bin/sh", "sh", config, NULL); + } + } + + perror("execl"); + exit(1); + } + + fd_set readfds; + struct timeval tv; + int status; + + while (waitpid(config_pid, &status, WNOHANG) == 0) { + FD_ZERO(&readfds); + FD_SET(sock, &readfds); + tv.tv_sec = 0; + tv.tv_usec = 50000; + + if (select(sock + 1, &readfds, NULL, NULL, &tv) > 0) { + client_sock = accept(sock, NULL, NULL); + if (client_sock >= 0) { + handle_socket_connection(client_sock); + } + } + } + + for (int i = 0; i < 5; i++) { + FD_ZERO(&readfds); + FD_SET(sock, &readfds); + tv.tv_sec = 0; + tv.tv_usec = 10000; + + if (select(sock + 1, &readfds, NULL, NULL, &tv) > 0) { + client_sock = accept(sock, NULL, NULL); + if (client_sock >= 0) { + handle_socket_connection(client_sock); + } + } + } + + XSelectInput(dpy, root, KeyPressMask); + + printf("piss: running with %d bindings\n", bind_count); + + int x11_fd = ConnectionNumber(dpy); + + fprintf(stderr, "piss: listening for keypresses (X11 fd=%d)...\n", x11_fd); + + while (1) { + FD_ZERO(&readfds); + FD_SET(x11_fd, &readfds); + FD_SET(sock, &readfds); + int max_fd = (x11_fd > sock) ? x11_fd : sock; + + if (select(max_fd + 1, &readfds, NULL, NULL, NULL) > 0) { + if (FD_ISSET(x11_fd, &readfds)) { + while (XPending(dpy)) { + XEvent ev; + XNextEvent(dpy, &ev); + fprintf(stderr, "piss: received X event type=%d\n", ev.type); + if (ev.type == KeyPress) + handle_keypress(&ev); + } + } + + if (FD_ISSET(sock, &readfds)) { + client_sock = accept(sock, NULL, NULL); + if (client_sock >= 0) { + handle_socket_connection(client_sock); + } + } + } + } + + unlink(SOCK_PATH); + XCloseDisplay(dpy); + return 0; +} -- cgit v1.3