Merge pull request #2798 from RedSoxFan/bar-bindsym

Implement bar bindsym

10 files changed, 182 insertions, 2 deletions

diff --git a/include/sway/config.h b/include/sway/config.h
index 0325042c..bc02c0fd 100644
--- a/include/sway/config.h
+++ b/include/sway/config.h
@@ -239,6 +239,12 @@ struct bar_config {
         } colors;
 };
 
+struct bar_binding {
+        uint32_t button;
+        bool release;
+        char *command;
+};
+
 struct border_colors {
         float border[4];
         float background[4];
@@ -533,6 +539,8 @@ struct bar_config *default_bar_config(void);
 
 void free_bar_config(struct bar_config *bar);
 
+void free_bar_binding(struct bar_binding *binding);
+
 void free_workspace_config(struct workspace_config *wsc);
 
 /**
diff --git a/include/swaybar/config.h b/include/swaybar/config.h
index 5f5688cf..d0336c27 100644
--- a/include/swaybar/config.h
+++ b/include/swaybar/config.h
@@ -3,6 +3,7 @@
 #include <stdbool.h>
 #include <stdint.h>
 #include <wayland-client.h>
+#include "list.h"
 #include "util.h"
 
 struct box_colors {
@@ -17,6 +18,12 @@ struct config_output {
         size_t index;
 };
 
+struct swaybar_binding {
+        uint32_t button;
+        char *command;
+        bool release;
+};
+
 struct swaybar_config {
         char *status_command;
         bool pango_markup;
@@ -29,6 +36,7 @@ struct swaybar_config {
         bool binding_mode_indicator;
         bool wrap_scroll;
         bool workspace_buttons;
+        list_t *bindings;
         struct wl_list outputs; // config_output::link
         bool all_outputs;
         int height;
diff --git a/include/swaybar/ipc.h b/include/swaybar/ipc.h
index 81e48a6b..8731dac2 100644
--- a/include/swaybar/ipc.h
+++ b/include/swaybar/ipc.h
@@ -7,5 +7,6 @@ bool ipc_initialize(struct swaybar *bar, const char *bar_id);
 bool handle_ipc_readable(struct swaybar *bar);
 void ipc_get_workspaces(struct swaybar *bar);
 void ipc_send_workspace_command(struct swaybar *bar, const char *ws);
+void ipc_execute_binding(struct swaybar *bar, struct swaybar_binding *bind);
 
 #endif
diff --git a/sway/commands/bar/bindsym.c b/sway/commands/bar/bindsym.c
index ac09a03f..4eea3e6a 100644
--- a/sway/commands/bar/bindsym.c
+++ b/sway/commands/bar/bindsym.c
@@ -1,5 +1,7 @@
+#define _XOPEN_SOURCE 500
 #include <stdlib.h>
 #include <string.h>
+#include <strings.h>
 #include "sway/commands.h"
 #include "sway/config.h"
 #include "list.h"
@@ -7,5 +9,61 @@
 #include "stringop.h"
 
 struct cmd_results *bar_cmd_bindsym(int argc, char **argv) {
-        return cmd_results_new(CMD_FAILURE, "bindsym", "TODO"); // TODO
+        struct cmd_results *error = NULL;
+        if ((error = checkarg(argc, "bar bindsym", EXPECTED_MORE_THAN, 1))) {
+                return error;
+        }
+        if (!config->current_bar) {
+                return cmd_results_new(CMD_FAILURE, "bar bindsym", "No bar defined.");
+        }
+
+        struct bar_binding *binding = calloc(1, sizeof(struct bar_binding));
+        if (!binding) {
+                return cmd_results_new(CMD_FAILURE, "bar bindsym",
+                                "Unable to allocate bar binding");
+        }
+
+        binding->release = false;
+        if (strcmp("--release", argv[0]) == 0) {
+                binding->release = true;
+                argv++;
+                argc--;
+        }
+
+        binding->button = 0;
+        if (strncasecmp(argv[0], "button", strlen("button")) == 0 &&
+                        strlen(argv[0]) == strlen("button0")) {
+                binding->button = argv[0][strlen("button")] - '0';
+        }
+        if (binding->button < 1 || binding->button > 9) {
+                free_bar_binding(binding);
+                return cmd_results_new(CMD_FAILURE, "bar bindsym",
+                                "Only button<1-9> is supported");
+        }
+
+        binding->command = join_args(argv + 1, argc - 1);
+
+        list_t *bindings = config->current_bar->bindings;
+        bool overwritten = false;
+        for (int i = 0; i < bindings->length; i++) {
+                struct bar_binding *other = bindings->items[i];
+                if (other->button == binding->button &&
+                                other->release == binding->release) {
+                        overwritten = true;
+                        bindings->items[i] = binding;
+                        free_bar_binding(other);
+                        wlr_log(WLR_DEBUG, "[bar %s] Updated binding for button%u%s",
+                                        config->current_bar->id, binding->button,
+                                        binding->release ? " (release)" : "");
+                        break;
+                }
+        }
+        if (!overwritten) {
+                list_add(bindings, binding);
+                wlr_log(WLR_DEBUG, "[bar %s] Added binding for button%u%s",
+                                config->current_bar->id, binding->button,
+                                binding->release ? " (release)" : "");
+        }
+
+        return cmd_results_new(CMD_SUCCESS, NULL, NULL);
 }
diff --git a/sway/config/bar.c b/sway/config/bar.c
index b8695798..c6899f57 100644
--- a/sway/config/bar.c
+++ b/sway/config/bar.c
@@ -28,6 +28,14 @@ static void terminate_swaybar(pid_t pid) {
         }
 }
 
+void free_bar_binding(struct bar_binding *binding) {
+        if (!binding) {
+                return;
+        }
+        free(binding->command);
+        free(binding);
+}
+
 void free_bar_config(struct bar_config *bar) {
         if (!bar) {
                 return;
@@ -39,7 +47,10 @@ void free_bar_config(struct bar_config *bar) {
         free(bar->status_command);
         free(bar->font);
         free(bar->separator_symbol);
-        // TODO: Free mouse bindings
+        for (int i = 0; i < bar->bindings->length; i++) {
+                struct bar_binding *binding = bar->bindings->items[i];
+                free_bar_binding(binding);
+        }
         list_free(bar->bindings);
         if (bar->outputs) {
                 free_flat_list(bar->outputs);
diff --git a/sway/ipc-json.c b/sway/ipc-json.c
index f02f370b..54d611f2 100644
--- a/sway/ipc-json.c
+++ b/sway/ipc-json.c
@@ -623,6 +623,22 @@ json_object *ipc_json_describe_bar_config(struct bar_config *bar) {
 
         json_object_object_add(json, "colors", colors);
 
+        if (bar->bindings->length > 0) {
+                json_object *bindings = json_object_new_array();
+                for (int i = 0; i < bar->bindings->length; ++i) {
+                        struct bar_binding *binding = bar->bindings->items[i];
+                        json_object *bind = json_object_new_object();
+                        json_object_object_add(bind, "input_code",
+                                        json_object_new_int(binding->button));
+                        json_object_object_add(bind, "command",
+                                        json_object_new_string(binding->command));
+                        json_object_object_add(bind, "release",
+                                        json_object_new_boolean(binding->release));
+                        json_object_array_add(bindings, bind);
+                }
+                json_object_object_add(json, "bindings", bindings);
+        }
+
         // Add outputs if defined
         if (bar->outputs && bar->outputs->length > 0) {
                 json_object *outputs = json_object_new_array();
diff --git a/sway/sway-bar.5.scd b/sway/sway-bar.5.scd
index 8c7be8e7..6729c9ac 100644
--- a/sway/sway-bar.5.scd
+++ b/sway/sway-bar.5.scd
@@ -60,6 +60,11 @@ Sway allows configuring swaybar in the sway configuration file.
 *height* <height>
         Sets the height of the bar. Default height will match the font size.
 
+*bindsym* [--release] button<n> <command>
+        Executes _command_ when mouse button _n_ has been pressed (or if _released_
+        is given, when mouse button _n_ has been released). To disable the default
+        behavior for a button, use the command _nop_.
+
 ## TRAY
 
 Swaybar provides a system tray where third-party applications may place icons.
diff --git a/swaybar/bar.c b/swaybar/bar.c
index 3990f1ca..5b7fea71 100644
--- a/swaybar/bar.c
+++ b/swaybar/bar.c
@@ -144,6 +144,19 @@ static void wl_pointer_motion(void *data, struct wl_pointer *wl_pointer,
         bar->pointer.y = wl_fixed_to_int(surface_y);
 }
 
+static bool check_bindings(struct swaybar *bar, uint32_t x11_button,
+                uint32_t state) {
+        bool released = state == WL_POINTER_BUTTON_STATE_RELEASED;
+        for (int i = 0; i < bar->config->bindings->length; i++) {
+                struct swaybar_binding *binding = bar->config->bindings->items[i];
+                if (binding->button == x11_button && binding->release == released) {
+                        ipc_execute_binding(bar, binding);
+                        return true;
+                }
+        }
+        return false;
+}
+
 static void wl_pointer_button(void *data, struct wl_pointer *wl_pointer,
                 uint32_t serial, uint32_t time, uint32_t button, uint32_t state) {
         struct swaybar *bar = data;
@@ -152,6 +165,11 @@ static void wl_pointer_button(void *data, struct wl_pointer *wl_pointer,
         if (!sway_assert(output, "button with no active output")) {
                 return;
         }
+
+        if (check_bindings(bar, wl_button_to_x11_button(button), state)) {
+                return;
+        }
+
         if (state != WL_POINTER_BUTTON_STATE_PRESSED) {
                 return;
         }
@@ -180,6 +198,15 @@ static void wl_pointer_axis(void *data, struct wl_pointer *wl_pointer,
                 return;
         }
 
+        // If there is a button press binding, execute it, skip default behavior,
+        // and check button release bindings
+        if (check_bindings(bar, wl_axis_to_x11_button(axis, value),
+                        WL_POINTER_BUTTON_STATE_PRESSED)) {
+                check_bindings(bar, wl_axis_to_x11_button(axis, value),
+                                WL_POINTER_BUTTON_STATE_RELEASED);
+                return;
+        }
+
         struct swaybar_hotspot *hotspot;
         wl_list_for_each(hotspot, &output->hotspots, link) {
                 double x = pointer->x * output->scale;
@@ -247,6 +274,10 @@ static void wl_pointer_axis(void *data, struct wl_pointer *wl_pointer,
         }
 
         ipc_send_workspace_command(bar, new->name);
+
+        // Check button release bindings
+        check_bindings(bar, wl_axis_to_x11_button(axis, value),
+                        WL_POINTER_BUTTON_STATE_RELEASED);
 }
 
 static void wl_pointer_frame(void *data, struct wl_pointer *wl_pointer) {
diff --git a/swaybar/config.c b/swaybar/config.c
index 4e851cca..09d40c24 100644
--- a/swaybar/config.c
+++ b/swaybar/config.c
@@ -3,6 +3,8 @@
 #include <string.h>
 #include "swaybar/config.h"
 #include "wlr-layer-shell-unstable-v1-client-protocol.h"
+#include "stringop.h"
+#include "list.h"
 
 uint32_t parse_position(const char *position) {
         uint32_t horiz = ZWLR_LAYER_SURFACE_V1_ANCHOR_LEFT |
@@ -34,6 +36,7 @@ struct swaybar_config *init_config(void) {
         config->binding_mode_indicator = true;
         config->wrap_scroll = false;
         config->workspace_buttons = true;
+        config->bindings = create_list();
         wl_list_init(&config->outputs);
 
         /* height */
@@ -69,11 +72,24 @@ struct swaybar_config *init_config(void) {
         return config;
 }
 
+static void free_binding(struct swaybar_binding *binding) {
+        if (!binding) {
+                return;
+        }
+        free(binding->command);
+        free(binding);
+}
+
 void free_config(struct swaybar_config *config) {
         free(config->status_command);
         free(config->font);
         free(config->mode);
         free(config->sep_symbol);
+        for (int i = 0; i < config->bindings->length; i++) {
+                struct swaybar_binding *binding = config->bindings->items[i];
+                free_binding(binding);
+        }
+        list_free(config->bindings);
         struct config_output *coutput, *tmp;
         wl_list_for_each_safe(coutput, tmp, &config->outputs, link) {
                 wl_list_remove(&coutput->link);
diff --git a/swaybar/ipc.c b/swaybar/ipc.c
index 7c53a44f..a67814c1 100644
--- a/swaybar/ipc.c
+++ b/swaybar/ipc.c
@@ -7,6 +7,7 @@
 #include "swaybar/config.h"
 #include "swaybar/ipc.h"
 #include "ipc-client.h"
+#include "list.h"
 
 void ipc_send_workspace_command(struct swaybar *bar, const char *ws) {
         const char *fmt = "workspace \"%s\"";
@@ -154,6 +155,7 @@ static bool ipc_parse_config(
         json_object *markup, *mode, *hidden_bar, *position, *status_command;
         json_object *font, *bar_height, *wrap_scroll, *workspace_buttons, *strip_workspace_numbers;
         json_object *binding_mode_indicator, *verbose, *colors, *sep_symbol, *outputs;
+        json_object *bindings;
         json_object_object_get_ex(bar_config, "mode", &mode);
         json_object_object_get_ex(bar_config, "hidden_bar", &hidden_bar);
         json_object_object_get_ex(bar_config, "position", &position);
@@ -169,6 +171,7 @@ static bool ipc_parse_config(
         json_object_object_get_ex(bar_config, "colors", &colors);
         json_object_object_get_ex(bar_config, "outputs", &outputs);
         json_object_object_get_ex(bar_config, "pango_markup", &markup);
+        json_object_object_get_ex(bar_config, "bindings", &bindings);
         if (status_command) {
                 free(config->status_command);
                 config->status_command = strdup(json_object_get_string(status_command));
@@ -202,6 +205,21 @@ static bool ipc_parse_config(
         if (markup) {
                 config->pango_markup = json_object_get_boolean(markup);
         }
+        if (bindings) {
+                int length = json_object_array_length(bindings);
+                for (int i = 0; i < length; ++i) {
+                        json_object *bindobj = json_object_array_get_idx(bindings, i);
+                        struct swaybar_binding *binding =
+                                calloc(1, sizeof(struct swaybar_binding));
+                        binding->button = json_object_get_int(
+                                        json_object_object_get(bindobj, "input_code"));
+                        binding->command = strdup(json_object_get_string(
+                                        json_object_object_get(bindobj, "command")));
+                        binding->release = json_object_get_boolean(
+                                        json_object_object_get(bindobj, "release"));
+                        list_add(config->bindings, binding);
+                }
+        }
 
         struct config_output *output, *tmp;
         wl_list_for_each_safe(output, tmp, &config->outputs, link) {
@@ -319,6 +337,14 @@ static void ipc_get_outputs(struct swaybar *bar) {
         free(res);
 }
 
+void ipc_execute_binding(struct swaybar *bar, struct swaybar_binding *bind) {
+        wlr_log(WLR_DEBUG, "Executing binding for button %u (release=%d): `%s`",
+                        bind->button, bind->release, bind->command);
+        uint32_t len = strlen(bind->command);
+        free(ipc_single_command(bar->ipc_socketfd,
+                        IPC_COMMAND, bind->command, &len));
+}
+
 bool ipc_initialize(struct swaybar *bar, const char *bar_id) {
         uint32_t len = strlen(bar_id);
         char *res = ipc_single_command(bar->ipc_socketfd,