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

   1 /*
   2  * vim:ts=4:sw=4:expandtab
   3  *
   4  * i3bar - an xcb-based status- and ws-bar for i3
   5  * © 2010 Axel Wagner and contributors (see also: LICENSE)
   6  *
   7  */
   8 #include <stdio.h>
   9 #include <i3/ipc.h>
  10 #include <string.h>
  11 #include <unistd.h>
  12 #include <stdlib.h>
  13 #include <errno.h>
  14 #include <ev.h>
  15 #include <getopt.h>
  16 #include <glob.h>
  17
  18 #include "common.h"
  19
  20 /*
  21  * Having verboselog(), errorlog() and debuglog() is necessary when using libi3.
  22  *
  23  */
  24 void verboselog(char *fmt, ...) {
  25     va_list args;
  26
  27     va_start(args, fmt);
  28     vfprintf(stdout, fmt, args);
  29     va_end(args);
  30 }
  31
  32 void errorlog(char *fmt, ...) {
  33     va_list args;
  34
  35     va_start(args, fmt);
  36     vfprintf(stderr, fmt, args);
  37     va_end(args);
  38 }
  39
  40 void debuglog(char *fmt, ...) {
  41 }
  42
  43 /*
  44  * Glob path, i.e. expand ~
  45  *
  46  */
  47 char *expand_path(char *path) {
  48     static glob_t globbuf;
  49     if (glob(path, GLOB_NOCHECK | GLOB_TILDE, NULL, &globbuf) < 0) {
  50         ELOG("glob() failed\n");
  51         exit(EXIT_FAILURE);
  52     }
  53     char *result = sstrdup(globbuf.gl_pathc > 0 ? globbuf.gl_pathv[0] : path);
  54     globfree(&globbuf);
  55     return result;
  56 }
  57
  58 void print_usage(char *elf_name) {
  59     printf("Usage: %s -b bar_id [-s sock_path] [-h] [-v]\n", elf_name);
  60     printf("\n");
  61     printf("-b, --bar_id  <bar_id>\tBar ID for which to get the configuration\n");
  62     printf("-s, --socket  <sock_path>\tConnect to i3 via <sock_path>\n");
  63     printf("-h, --help    Display this help message and exit\n");
  64     printf("-v, --version Display version number and exit\n");
  65     printf("\n");
  66     printf(" PLEASE NOTE that i3bar will be automatically started by i3\n"
  67            " as soon as there is a 'bar' configuration block in your\n"
  68            " config file. You should never need to start it manually.\n");
  69     printf("\n");
  70 }
  71
  72 /*
  73  * We watch various signals, that are there to make our application stop.
  74  * If we get one of those, we ev_unloop() and invoke the cleanup routines
  75  * in main() with that
  76  *
  77  */
  78 void sig_cb(struct ev_loop *loop, ev_signal *watcher, int revents) {
  79     switch (watcher->signum) {
  80         case SIGTERM:
  81             DLOG("Got a SIGTERM, stopping\n");
  82             break;
  83         case SIGINT:
  84             DLOG("Got a SIGINT, stopping\n");
  85             break;
  86         case SIGHUP:
  87             DLOG("Got a SIGHUP, stopping\n");
  88     }
  89     ev_unloop(main_loop, EVUNLOOP_ALL);
  90 }
  91
  92 int main(int argc, char **argv) {
  93     int opt;
  94     int option_index = 0;
  95     char *socket_path = getenv("I3SOCK");
  96     if (socket_path != NULL) {
  97         socket_path = sstrdup(socket_path);
  98     }
  99     char *i3_default_sock_path = "/tmp/i3-ipc.sock";
 100
 101     /* Initialize the standard config to use 0 as default */
 102     memset(&config, '\0', sizeof(config_t));
 103
 104     static struct option long_opt[] = {
 105         {"socket", required_argument, 0, 's'},
 106         {"bar_id", required_argument, 0, 'b'},
 107         {"help", no_argument, 0, 'h'},
 108         {"version", no_argument, 0, 'v'},
 109         {NULL, 0, 0, 0}};
 110
 111     while ((opt = getopt_long(argc, argv, "b:s:hv", long_opt, &option_index)) != -1) {
 112         switch (opt) {
 113             case 's':
 114                 socket_path = expand_path(optarg);
 115                 break;
 116             case 'v':
 117                 printf("i3bar version " I3_VERSION " © 2010 Axel Wagner and contributors\n");
 118                 exit(EXIT_SUCCESS);
 119                 break;
 120             case 'b':
 121                 config.bar_id = sstrdup(optarg);
 122                 break;
 123             default:
 124                 print_usage(argv[0]);
 125                 exit(EXIT_SUCCESS);
 126                 break;
 127         }
 128     }
 129
 130     if (!config.bar_id) {
 131         /* TODO: maybe we want -f which will automatically ask i3 for the first
 132          * configured bar (and error out if there are too many)? */
 133         ELOG("No bar_id passed. Please let i3 start i3bar or specify --bar_id\n");
 134         exit(EXIT_FAILURE);
 135     }
 136
 137     main_loop = ev_default_loop(0);
 138
 139     char *atom_sock_path = init_xcb_early();
 140
 141     if (socket_path == NULL) {
 142         socket_path = atom_sock_path;
 143     } else {
 144         free(atom_sock_path);
 145     }
 146
 147     if (socket_path == NULL) {
 148         ELOG("No socket path specified, default to %s\n", i3_default_sock_path);
 149         socket_path = expand_path(i3_default_sock_path);
 150     }
 151
 152     init_outputs();
 153     if (init_connection(socket_path)) {
 154         /* Request the bar configuration. When it arrives, we fill the config array. */
 155         i3_send_msg(I3_IPC_MESSAGE_TYPE_GET_BAR_CONFIG, config.bar_id);
 156     }
 157     free(socket_path);
 158
 159     /* We listen to SIGTERM/QUIT/INT and try to exit cleanly, by stopping the main loop.
 160      * We only need those watchers on the stack, so putting them on the stack saves us
 161      * some calls to free() */
 162     ev_signal *sig_term = smalloc(sizeof(ev_signal));
 163     ev_signal *sig_int = smalloc(sizeof(ev_signal));
 164     ev_signal *sig_hup = smalloc(sizeof(ev_signal));
 165
 166     ev_signal_init(sig_term, &sig_cb, SIGTERM);
 167     ev_signal_init(sig_int, &sig_cb, SIGINT);
 168     ev_signal_init(sig_hup, &sig_cb, SIGHUP);
 169
 170     ev_signal_start(main_loop, sig_term);
 171     ev_signal_start(main_loop, sig_int);
 172     ev_signal_start(main_loop, sig_hup);
 173
 174     /* From here on everything should run smooth for itself, just start listening for
 175      * events. We stop simply stop the event loop, when we are finished */
 176     ev_loop(main_loop, 0);
 177
 178     kill_child();
 179
 180     clean_xcb();
 181     ev_default_destroy();
 182
 183     free_workspaces();
 184
 185     return 0;
 186 }