1 files changed, 97 insertions, 0 deletions

diff --git a/sway/commands/bar/tray_bind.c b/sway/commands/bar/tray_bind.c
new file mode 100644
index 00000000..48a15462
--- /dev/null
+++ b/sway/commands/bar/tray_bind.c
@@ -0,0 +1,97 @@
+#include <strings.h>
+#include "config.h"
+#include "sway/commands.h"
+#include "sway/config.h"
+#include "sway/input/cursor.h"
+#include "log.h"
+
+static struct cmd_results *tray_bind(int argc, char **argv, bool code) {
+#if HAVE_TRAY
+        const char *command = code ? "bar tray_bindcode" : "bar tray_bindsym";
+        struct cmd_results *error = NULL;
+        if ((error = checkarg(argc, command, EXPECTED_EQUAL_TO, 2))) {
+                return error;
+        }
+        if (!config->current_bar) {
+                return cmd_results_new(CMD_FAILURE, "No bar defined.");
+        }
+
+        struct tray_binding *binding = calloc(1, sizeof(struct tray_binding));
+        if (!binding) {
+                return cmd_results_new(CMD_FAILURE, "Unable to allocate tray binding");
+        }
+
+        char *message = NULL;
+        if (code) {
+                binding->button = get_mouse_bindcode(argv[0], &message);
+        } else {
+                binding->button = get_mouse_bindsym(argv[0], &message);
+        }
+        if (message) {
+                free(binding);
+                error = cmd_results_new(CMD_INVALID, message);
+                free(message);
+                return error;
+        } else if (!binding->button) {
+                free(binding);
+                return cmd_results_new(CMD_INVALID, "Unknown button %s", argv[0]);
+        }
+        const char *name = get_mouse_button_name(binding->button);
+
+        static const char *commands[] = {
+                "ContextMenu",
+                "Activate",
+                "SecondaryActivate",
+                "ScrollDown",
+                "ScrollLeft",
+                "ScrollRight",
+                "ScrollUp",
+                "nop"
+        };
+
+        for (size_t i = 0; i < sizeof(commands) / sizeof(commands[0]); ++i) {
+                if (strcasecmp(argv[1], commands[i]) == 0) {
+                        binding->command = commands[i];
+                }
+        }
+        if (!binding->command) {
+                return cmd_results_new(CMD_INVALID, "[Bar %s] Invalid tray command %s",
+                                config->current_bar->id, argv[1]);
+        }
+
+        bool overwritten = false;
+        struct tray_binding *other = NULL;
+        wl_list_for_each(other, &config->current_bar->tray_bindings, link) {
+                if (other->button == binding->button) {
+                        overwritten = true;
+                        other->command = binding->command;
+                        free(binding);
+                        binding = other;
+                        wlr_log(WLR_DEBUG,
+                                        "[bar %s] Updated tray binding for %u (%s) to %s",
+                                        config->current_bar->id, binding->button, name,
+                                        binding->command);
+                        break;
+                }
+        }
+        if (!overwritten) {
+                wl_list_insert(&config->current_bar->tray_bindings, &binding->link);
+                wlr_log(WLR_DEBUG, "[bar %s] Added tray binding for %u (%s) to %s",
+                                config->current_bar->id, binding->button, name,
+                                binding->command);
+        }
+
+        return cmd_results_new(CMD_SUCCESS, NULL);
+#else
+        return cmd_results_new(CMD_INVALID,
+                        "Sway has been compiled without tray support");
+#endif
+}
+
+struct cmd_results *bar_cmd_tray_bindcode(int argc, char **argv) {
+        return tray_bind(argc, argv, true);
+}
+
+struct cmd_results *bar_cmd_tray_bindsym(int argc, char **argv) {
+        return tray_bind(argc, argv, false);
+}