git.sur5r.net Git - i3/i3/blob - i3bar/src/main.c

   1 /*
   2  * i3bar - an xcb-based status- and ws-bar for i3
   3  *
   4  * © 2010 Axel Wagner and contributors
   5  *
   6  * See file LICNSE for license information
   7  *
   8  */
   9 #include <stdio.h>
  10 #include <i3/ipc.h>
  11 #include <string.h>
  12 #include <unistd.h>
  13 #include <stdlib.h>
  14 #include <errno.h>
  15 #include <ev.h>
  16 #include <getopt.h>
  17 #include <glob.h>
  18
  19 #include "common.h"
  20
  21 /*
  22  * Glob path, i.e. expand ~
  23  *
  24  */
  25 char *expand_path(char *path) {
  26     static glob_t globbuf;
  27     if (glob(path, GLOB_NOCHECK | GLOB_TILDE, NULL, &globbuf) < 0) {
  28         ELOG("glob() failed\n");
  29         exit(EXIT_FAILURE);
  30     }
  31     char *result = strdup(globbuf.gl_pathc > 0 ? globbuf.gl_pathv[0] : path);
  32     if (result == NULL) {
  33         ELOG("malloc() failed: %s\n", strerror(errno));
  34         exit(EXIT_FAILURE);
  35     }
  36     globfree(&globbuf);
  37     return result;
  38 }
  39
  40 static void read_color(char **color) {
  41     int len = strlen(optarg);
  42     if (len == 6 || (len == 7 && optarg[0] == '#')) {
  43         int offset = len - 6;
  44         int good = 1, i;
  45         for (i = offset; good && i < 6 + offset; ++i) {
  46             char c = optarg[i];
  47             if (!(c >= 'a' && c <= 'f')
  48                     && !(c >= 'A' && c <= 'F')
  49                     && !(c >= '0' && c <= '9')) {
  50                 good = 0;
  51                 break;
  52             }
  53         }
  54         if (good) {
  55             *color = strdup(optarg + offset);
  56             return;
  57         }
  58     }
  59
  60     fprintf(stderr, "Bad color value \"%s\"\n", optarg);
  61     exit(EXIT_FAILURE);
  62 }
  63
  64 static void free_colors(struct xcb_color_strings_t *colors) {
  65 #define FREE_COLOR(x) \
  66     do { \
  67         if (colors->x) \
  68             free(colors->x); \
  69     } while (0)
  70     FREE_COLOR(bar_fg);
  71     FREE_COLOR(bar_bg);
  72     FREE_COLOR(active_ws_fg);
  73     FREE_COLOR(active_ws_bg);
  74     FREE_COLOR(inactive_ws_fg);
  75     FREE_COLOR(inactive_ws_bg);
  76     FREE_COLOR(urgent_ws_fg);
  77     FREE_COLOR(urgent_ws_bg);
  78 #undef FREE_COLOR
  79 }
  80
  81 void print_usage(char *elf_name) {
  82     printf("Usage: %s [-s sock_path] [-c command] [-m] [-f font] [-V] [-h]\n", elf_name);
  83     printf("-s <sock_path>\tConnect to i3 via <sock_path>\n");
  84     printf("-c <command>\tExecute <command> to get stdin\n");
  85     printf("-m\t\tHide the bars, when mod4 is not pressed.\n");
  86     printf("\t\tIf -c is specified, the childprocess is sent a SIGSTOP on hiding,\n");
  87     printf("\t\tand a SIGCONT on unhiding of the bars\n");
  88     printf("-f <font>\tUse X-Core-Font <font> for display\n");
  89     printf("-V\t\tBe (very) verbose with the debug-output\n");
  90     printf("-h\t\tDisplay this help-message and exit\n");
  91 }
  92
  93 /*
  94  * We watch various signals, that are there to make our application stop.
  95  * If we get one of those, we ev_unloop() and invoke the cleanup-routines
  96  * in main() with that
  97  *
  98  */
  99 void sig_cb(struct ev_loop *loop, ev_signal *watcher, int revents) {
 100     switch (watcher->signum) {
 101         case SIGTERM:
 102             DLOG("Got a SIGTERM, stopping\n");
 103             break;
 104         case SIGINT:
 105             DLOG("Got a SIGINT, stopping\n");
 106             break;
 107         case SIGHUP:
 108             DLOG("Got a SIGHUP, stopping\n");
 109     }
 110     ev_unloop(main_loop, EVUNLOOP_ALL);
 111 }
 112
 113 int main(int argc, char **argv) {
 114     int opt;
 115     int option_index = 0;
 116     char *socket_path = getenv("I3SOCK");
 117     char *command = NULL;
 118     char *fontname = NULL;
 119     char *i3_default_sock_path = "~/.i3/ipc.sock";
 120     struct xcb_color_strings_t colors = { NULL, };
 121
 122     /* Definition of the standard-config */
 123     config.hide_on_modifier = 0;
 124
 125     static struct option long_opt[] = {
 126         { "socket",               required_argument, 0, 's' },
 127         { "command",              required_argument, 0, 'c' },
 128         { "hide",                 no_argument,       0, 'm' },
 129         { "font",                 required_argument, 0, 'f' },
 130         { "help",                 no_argument,       0, 'h' },
 131         { "version",              no_argument,       0, 'v' },
 132         { "verbose",              no_argument,       0, 'V' },
 133         { "color-bar-fg",         required_argument, 0, 'A' },
 134         { "color-bar-bg",         required_argument, 0, 'B' },
 135         { "color-active-ws-fg",   required_argument, 0, 'C' },
 136         { "color-active-ws-bg",   required_argument, 0, 'D' },
 137         { "color-inactive-ws-fg", required_argument, 0, 'E' },
 138         { "color-inactive-ws-bg", required_argument, 0, 'F' },
 139         { "color-urgent-ws-bg",   required_argument, 0, 'G' },
 140         { "color-urgent-ws-fg",   required_argument, 0, 'H' },
 141         { NULL,                   0,                 0, 0}
 142     };
 143
 144     while ((opt = getopt_long(argc, argv, "s:c:mf:hvVA:B:C:D:E:F:G:H:", long_opt, &option_index)) != -1) {
 145         switch (opt) {
 146             case 's':
 147                 socket_path = expand_path(optarg);
 148                 break;
 149             case 'c':
 150                 command = strdup(optarg);
 151                 break;
 152             case 'm':
 153                 config.hide_on_modifier = 1;
 154                 break;
 155             case 'f':
 156                 fontname = strdup(optarg);
 157                 break;
 158             case 'v':
 159                 printf("i3bar version " I3BAR_VERSION " © 2010 Axel Wagner and contributors\n");
 160                 exit(EXIT_SUCCESS);
 161                 break;
 162             case 'V':
 163                 config.verbose = 1;
 164                 break;
 165             case 'A':
 166                 read_color(&colors.bar_fg);
 167                 break;
 168             case 'B':
 169                 read_color(&colors.bar_bg);
 170                 break;
 171             case 'C':
 172                 read_color(&colors.active_ws_fg);
 173                 break;
 174             case 'D':
 175                 read_color(&colors.active_ws_bg);
 176                 break;
 177             case 'E':
 178                 read_color(&colors.inactive_ws_fg);
 179                 break;
 180             case 'F':
 181                 read_color(&colors.inactive_ws_bg);
 182                 break;
 183             case 'G':
 184                 read_color(&colors.urgent_ws_bg);
 185                 break;
 186             case 'H':
 187                 read_color(&colors.urgent_ws_fg);
 188                 break;
 189             default:
 190                 print_usage(argv[0]);
 191                 exit(EXIT_SUCCESS);
 192                 break;
 193         }
 194     }
 195
 196     if (fontname == NULL) {
 197         /* This is a very restrictive default. More sensefull would be something like
 198          * "-misc-*-*-*-*--*-*-*-*-*-*-*-*". But since that produces very ugly results
 199          * on my machine, let's stick with this until we have a configfile */
 200         fontname = "-misc-fixed-medium-r-semicondensed--12-110-75-75-c-60-iso10646-1";
 201     }
 202
 203     if (socket_path == NULL) {
 204         ELOG("No Socket Path Specified, default to %s\n", i3_default_sock_path);
 205         socket_path = expand_path(i3_default_sock_path);
 206     }
 207
 208     main_loop = ev_default_loop(0);
 209
 210     init_colors(&colors);
 211     init_xcb(fontname);
 212
 213     free_colors(&colors);
 214
 215     init_outputs();
 216     init_connection(socket_path);
 217
 218     /* We subscribe to the i3-events we need */
 219     subscribe_events();
 220
 221     /* We initiate the main-function by requesting infos about the outputs and
 222      * workspaces. Everything else (creating the bars, showing the right workspace-
 223      * buttons and more) is taken care of by the event-driveniness of the code */
 224     i3_send_msg(I3_IPC_MESSAGE_TYPE_GET_OUTPUTS, NULL);
 225     i3_send_msg(I3_IPC_MESSAGE_TYPE_GET_WORKSPACES, NULL);
 226
 227     /* The name of this function is actually misleading. Even if no -c is specified,
 228      * this function initiates the watchers to listen on stdin and react accordingly */
 229     start_child(command);
 230     FREE(command);
 231
 232     /* We listen to SIGTERM/QUIT/INT and try to exit cleanly, by stopping the main-loop.
 233      * We only need those watchers on the stack, so putting them on the stack saves us
 234      * some calls to free() */
 235     ev_signal *sig_term = malloc(sizeof(ev_signal));
 236     ev_signal *sig_int = malloc(sizeof(ev_signal));
 237     ev_signal *sig_hup = malloc(sizeof(ev_signal));
 238
 239     if (sig_term == NULL || sig_int == NULL || sig_hup == NULL) {
 240         ELOG("malloc() failed: %s\n", strerror(errno));
 241         exit(EXIT_FAILURE);
 242     }
 243
 244     ev_signal_init(sig_term, &sig_cb, SIGTERM);
 245     ev_signal_init(sig_int, &sig_cb, SIGINT);
 246     ev_signal_init(sig_hup, &sig_cb, SIGHUP);
 247
 248     ev_signal_start(main_loop, sig_term);
 249     ev_signal_start(main_loop, sig_int);
 250     ev_signal_start(main_loop, sig_hup);
 251
 252     /* From here on everything should run smooth for itself, just start listening for
 253      * events. We stop simply stop the event-loop, when we are finished */
 254     ev_loop(main_loop, 0);
 255
 256     kill_child();
 257
 258     FREE(statusline_buffer);
 259
 260     clean_xcb();
 261     ev_default_destroy();
 262
 263     free_workspaces();
 264
 265     return 0;
 266 }