swaybar: Implement strip_workspace_numbers

1 files changed, 38 insertions, 4 deletions

diff --git a/swaybar/main.c b/swaybar/main.c
index cbedb200..34c54287 100644
--- a/swaybar/main.c
+++ b/swaybar/main.c
@@ -76,6 +76,7 @@ struct registry *registry;
 struct window *window;
 bool dirty = true;
 char *separator_symbol = NULL;
+bool strip_workspace_numbers = false;
 typedef enum {UNDEF, TEXT, I3BAR} command_protocol;
 command_protocol protocol = UNDEF;
 
@@ -289,7 +290,7 @@ void bar_ipc_init(int outputi, const char *bar_id) {
 
         json_object *bar_config = json_tokener_parse(res);
         json_object *tray_output, *mode, *hidden_state, *position, *_status_command;
-        json_object *font, *bar_height, *workspace_buttons, *strip_workspace_numbers;
+        json_object *font, *bar_height, *workspace_buttons, *_strip_workspace_numbers;
         json_object *binding_mode_indicator, *verbose, *_colors, *sep_symbol;
         json_object_object_get_ex(bar_config, "tray_output", &tray_output);
         json_object_object_get_ex(bar_config, "mode", &mode);
@@ -299,7 +300,7 @@ void bar_ipc_init(int outputi, const char *bar_id) {
         json_object_object_get_ex(bar_config, "font", &font);
         json_object_object_get_ex(bar_config, "bar_height", &bar_height);
         json_object_object_get_ex(bar_config, "workspace_buttons", &workspace_buttons);
-        json_object_object_get_ex(bar_config, "strip_workspace_numbers", &strip_workspace_numbers);
+        json_object_object_get_ex(bar_config, "strip_workspace_numbers", &_strip_workspace_numbers);
         json_object_object_get_ex(bar_config, "binding_mode_indicator", &binding_mode_indicator);
         json_object_object_get_ex(bar_config, "verbose", &verbose);
         json_object_object_get_ex(bar_config, "separator_symbol", &sep_symbol);
@@ -323,6 +324,10 @@ void bar_ipc_init(int outputi, const char *bar_id) {
                 separator_symbol = strdup(json_object_get_string(sep_symbol));
         }
 
+        if (_strip_workspace_numbers) {
+                strip_workspace_numbers = json_object_get_boolean(_strip_workspace_numbers);
+        }
+
         if (bar_height) {
                 int width, height;
                 get_text_size(window, &width, &height, "Test string for measuring purposes");
@@ -567,9 +572,36 @@ void render_block(struct status_block *block, double *x, bool edge) {
 
 }
 
+char *handle_workspace_number(bool strip_num, const char *ws_name) {
+        bool strip = false;
+        int i;
+
+        if (strip_num) {
+                int len = strlen(ws_name);
+                for (i = 0; i < len; ++i) {
+                        if (!('0' <= ws_name[i] && ws_name[i] <= '9')) {
+                                if (':' == ws_name[i] && i < len-1) {
+                                        strip = true;
+                                        ++i;
+                                }
+                                break;
+                        }
+                }
+        }
+
+        if (strip) {
+                return strdup(ws_name + i);
+        }
+
+        return strdup(ws_name);
+}
+
 void render_workspace_button(struct workspace *ws, double *x) {
+        // strip workspace numbers if required
+        char *name = handle_workspace_number(strip_workspace_numbers, ws->name);
+
         int width, height;
-        get_text_size(window, &width, &height, "%s", ws->name);
+        get_text_size(window, &width, &height, "%s", name);
         struct box_colors box_colors;
         if (ws->urgent) {
                 box_colors = colors.urgent_workspace;
@@ -596,9 +628,11 @@ void render_workspace_button(struct workspace *ws, double *x) {
         // text
         cairo_set_source_u32(window->cairo, box_colors.text);
         cairo_move_to(window->cairo, (int)*x + ws_hor_padding, margin);
-        pango_printf(window, "%s", ws->name);
+        pango_printf(window, "%s", name);
 
         *x += width + ws_hor_padding * 2 + ws_spacing;
+
+        free(name);
 }
 
 void render() {