git.sur5r.net Git - i3/i3/blob - src/load_layout.c

   1 #undef I3__FILE__
   2 #define I3__FILE__ "load_layout.c"
   3 /*
   4  * vim:ts=4:sw=4:expandtab
   5  *
   6  * i3 - an improved dynamic tiling window manager
   7  * © 2009-2012 Michael Stapelberg and contributors (see also: LICENSE)
   8  *
   9  * load_layout.c: Restore (parts of) the layout, for example after an inplace
  10  *                restart.
  11  *
  12  */
  13 #include "all.h"
  14
  15 #include <yajl/yajl_common.h>
  16 #include <yajl/yajl_gen.h>
  17 #include <yajl/yajl_parse.h>
  18 #include <yajl/yajl_version.h>
  19
  20 /* TODO: refactor the whole parsing thing */
  21
  22 static char *last_key;
  23 static Con *json_node;
  24 static Con *to_focus;
  25 static bool parsing_swallows;
  26 static bool parsing_rect;
  27 static bool parsing_window_rect;
  28 static bool parsing_geometry;
  29 static bool parsing_focus;
  30 struct Match *current_swallow;
  31
  32 /* This list is used for reordering the focus stack after parsing the 'focus'
  33  * array. */
  34 struct focus_mapping {
  35     int old_id;
  36     TAILQ_ENTRY(focus_mapping) focus_mappings;
  37 };
  38
  39 static TAILQ_HEAD(focus_mappings_head, focus_mapping) focus_mappings =
  40   TAILQ_HEAD_INITIALIZER(focus_mappings);
  41
  42 static int json_start_map(void *ctx) {
  43     LOG("start of map, last_key = %s\n", last_key);
  44     if (parsing_swallows) {
  45         LOG("creating new swallow\n");
  46         current_swallow = smalloc(sizeof(Match));
  47         match_init(current_swallow);
  48         TAILQ_INSERT_TAIL(&(json_node->swallow_head), current_swallow, matches);
  49     } else {
  50         if (!parsing_rect && !parsing_window_rect && !parsing_geometry) {
  51             if (last_key && strcasecmp(last_key, "floating_nodes") == 0) {
  52                 DLOG("New floating_node\n");
  53                 Con *ws = con_get_workspace(json_node);
  54                 json_node = con_new(NULL, NULL);
  55                 json_node->parent = ws;
  56                 DLOG("Parent is workspace = %p\n", ws);
  57             } else {
  58                 Con *parent = json_node;
  59                 json_node = con_new(NULL, NULL);
  60                 json_node->parent = parent;
  61             }
  62         }
  63     }
  64     return 1;
  65 }
  66
  67 static int json_end_map(void *ctx) {
  68     LOG("end of map\n");
  69     if (!parsing_swallows && !parsing_rect && !parsing_window_rect && !parsing_geometry) {
  70         LOG("attaching\n");
  71         con_attach(json_node, json_node->parent, true);
  72         json_node = json_node->parent;
  73     }
  74     if (parsing_rect)
  75         parsing_rect = false;
  76     if (parsing_window_rect)
  77         parsing_window_rect = false;
  78     if (parsing_geometry)
  79         parsing_geometry = false;
  80     return 1;
  81 }
  82
  83 static int json_end_array(void *ctx) {
  84     LOG("end of array\n");
  85     parsing_swallows = false;
  86     if (parsing_focus) {
  87         /* Clear the list of focus mappings */
  88         struct focus_mapping *mapping;
  89         TAILQ_FOREACH_REVERSE(mapping, &focus_mappings, focus_mappings_head, focus_mappings) {
  90             LOG("focus (reverse) %d\n", mapping->old_id);
  91             Con *con;
  92             TAILQ_FOREACH(con, &(json_node->focus_head), focused) {
  93                 if (con->old_id != mapping->old_id)
  94                     continue;
  95                 LOG("got it! %p\n", con);
  96                 /* Move this entry to the top of the focus list. */
  97                 TAILQ_REMOVE(&(json_node->focus_head), con, focused);
  98                 TAILQ_INSERT_HEAD(&(json_node->focus_head), con, focused);
  99                 break;
 100             }
 101         }
 102         while (!TAILQ_EMPTY(&focus_mappings)) {
 103             mapping = TAILQ_FIRST(&focus_mappings);
 104             TAILQ_REMOVE(&focus_mappings, mapping, focus_mappings);
 105             free(mapping);
 106         }
 107         parsing_focus = false;
 108     }
 109     return 1;
 110 }
 111
 112 #if YAJL_MAJOR < 2
 113 static int json_key(void *ctx, const unsigned char *val, unsigned int len) {
 114 #else
 115 static int json_key(void *ctx, const unsigned char *val, size_t len) {
 116 #endif
 117     LOG("key: %.*s\n", (int)len, val);
 118     FREE(last_key);
 119     last_key = scalloc((len+1) * sizeof(char));
 120     memcpy(last_key, val, len);
 121     if (strcasecmp(last_key, "swallows") == 0)
 122         parsing_swallows = true;
 123
 124     if (strcasecmp(last_key, "rect") == 0)
 125         parsing_rect = true;
 126
 127     if (strcasecmp(last_key, "window_rect") == 0)
 128         parsing_window_rect = true;
 129
 130     if (strcasecmp(last_key, "geometry") == 0)
 131         parsing_geometry = true;
 132
 133     if (strcasecmp(last_key, "focus") == 0)
 134         parsing_focus = true;
 135
 136     return 1;
 137 }
 138
 139 #if YAJL_MAJOR >= 2
 140 static int json_string(void *ctx, const unsigned char *val, size_t len) {
 141 #else
 142 static int json_string(void *ctx, const unsigned char *val, unsigned int len) {
 143 #endif
 144     LOG("string: %.*s for key %s\n", (int)len, val, last_key);
 145     if (parsing_swallows) {
 146         /* TODO: the other swallowing keys */
 147         if (strcasecmp(last_key, "class") == 0) {
 148             current_swallow->class = scalloc((len+1) * sizeof(char));
 149             memcpy(current_swallow->class, val, len);
 150         }
 151         LOG("unhandled yet: swallow\n");
 152     } else {
 153         if (strcasecmp(last_key, "name") == 0) {
 154             json_node->name = scalloc((len+1) * sizeof(char));
 155             memcpy(json_node->name, val, len);
 156         } else if (strcasecmp(last_key, "sticky_group") == 0) {
 157             json_node->sticky_group = scalloc((len+1) * sizeof(char));
 158             memcpy(json_node->sticky_group, val, len);
 159             LOG("sticky_group of this container is %s\n", json_node->sticky_group);
 160         } else if (strcasecmp(last_key, "orientation") == 0) {
 161             /* Upgrade path from older versions of i3 (doing an inplace restart
 162              * to a newer version):
 163              * "orientation" is dumped before "layout". Therefore, we store
 164              * whether the orientation was horizontal or vertical in the
 165              * last_split_layout. When we then encounter layout == "default",
 166              * we will use the last_split_layout as layout instead. */
 167             char *buf = NULL;
 168             sasprintf(&buf, "%.*s", (int)len, val);
 169             if (strcasecmp(buf, "none") == 0 ||
 170                 strcasecmp(buf, "horizontal") == 0)
 171                 json_node->last_split_layout = L_SPLITH;
 172             else if (strcasecmp(buf, "vertical") == 0)
 173                 json_node->last_split_layout = L_SPLITV;
 174             else LOG("Unhandled orientation: %s\n", buf);
 175
 176             /* What used to be an implicit check whether orientation !=
 177              * NO_ORIENTATION is now a proper separate flag. */
 178             if (strcasecmp(buf, "none") != 0)
 179                 json_node->split = true;
 180             free(buf);
 181         } else if (strcasecmp(last_key, "border") == 0) {
 182             char *buf = NULL;
 183             sasprintf(&buf, "%.*s", (int)len, val);
 184             if (strcasecmp(buf, "none") == 0)
 185                 json_node->border_style = BS_NONE;
 186             else if (strcasecmp(buf, "1pixel") == 0)
 187                 json_node->border_style = BS_1PIXEL;
 188             else if (strcasecmp(buf, "normal") == 0)
 189                 json_node->border_style = BS_NORMAL;
 190             else LOG("Unhandled \"border\": %s\n", buf);
 191             free(buf);
 192         } else if (strcasecmp(last_key, "layout") == 0) {
 193             char *buf = NULL;
 194             sasprintf(&buf, "%.*s", (int)len, val);
 195             if (strcasecmp(buf, "default") == 0)
 196                 /* This set above when we read "orientation". */
 197                 json_node->layout = json_node->last_split_layout;
 198             else if (strcasecmp(buf, "stacked") == 0)
 199                 json_node->layout = L_STACKED;
 200             else if (strcasecmp(buf, "tabbed") == 0)
 201                 json_node->layout = L_TABBED;
 202             else if (strcasecmp(buf, "dockarea") == 0) {
 203                 json_node->layout = L_DOCKAREA;
 204                 /* Necessary for migrating from older versions of i3. */
 205                 json_node->split = false;
 206             } else if (strcasecmp(buf, "output") == 0)
 207                 json_node->layout = L_OUTPUT;
 208             else if (strcasecmp(buf, "splith") == 0)
 209                 json_node->layout = L_SPLITH;
 210             else if (strcasecmp(buf, "splitv") == 0)
 211                 json_node->layout = L_SPLITV;
 212             else LOG("Unhandled \"layout\": %s\n", buf);
 213             free(buf);
 214         } else if (strcasecmp(last_key, "workspace_layout") == 0) {
 215             char *buf = NULL;
 216             sasprintf(&buf, "%.*s", (int)len, val);
 217             if (strcasecmp(buf, "default") == 0)
 218                 json_node->workspace_layout = L_DEFAULT;
 219             else if (strcasecmp(buf, "stacked") == 0)
 220                 json_node->workspace_layout = L_STACKED;
 221             else if (strcasecmp(buf, "tabbed") == 0)
 222                 json_node->workspace_layout = L_TABBED;
 223             else LOG("Unhandled \"workspace_layout\": %s\n", buf);
 224             free(buf);
 225         } else if (strcasecmp(last_key, "last_split_layout") == 0) {
 226             char *buf = NULL;
 227             sasprintf(&buf, "%.*s", (int)len, val);
 228             if (strcasecmp(buf, "splith") == 0)
 229                 json_node->last_split_layout = L_SPLITH;
 230             else if (strcasecmp(buf, "splitv") == 0)
 231                 json_node->last_split_layout = L_SPLITV;
 232             else LOG("Unhandled \"last_splitlayout\": %s\n", buf);
 233             free(buf);
 234         } else if (strcasecmp(last_key, "mark") == 0) {
 235             char *buf = NULL;
 236             sasprintf(&buf, "%.*s", (int)len, val);
 237             json_node->mark = buf;
 238         } else if (strcasecmp(last_key, "floating") == 0) {
 239             char *buf = NULL;
 240             sasprintf(&buf, "%.*s", (int)len, val);
 241             if (strcasecmp(buf, "auto_off") == 0)
 242                 json_node->floating = FLOATING_AUTO_OFF;
 243             else if (strcasecmp(buf, "auto_on") == 0)
 244                 json_node->floating = FLOATING_AUTO_ON;
 245             else if (strcasecmp(buf, "user_off") == 0)
 246                 json_node->floating = FLOATING_USER_OFF;
 247             else if (strcasecmp(buf, "user_on") == 0)
 248                 json_node->floating = FLOATING_USER_ON;
 249             free(buf);
 250         } else if (strcasecmp(last_key, "scratchpad_state") == 0) {
 251             char *buf = NULL;
 252             sasprintf(&buf, "%.*s", (int)len, val);
 253             if (strcasecmp(buf, "none") == 0)
 254                 json_node->scratchpad_state = SCRATCHPAD_NONE;
 255             else if (strcasecmp(buf, "fresh") == 0)
 256                 json_node->scratchpad_state = SCRATCHPAD_FRESH;
 257             else if (strcasecmp(buf, "changed") == 0)
 258                 json_node->scratchpad_state = SCRATCHPAD_CHANGED;
 259             free(buf);
 260         }
 261     }
 262     return 1;
 263 }
 264
 265 #if YAJL_MAJOR >= 2
 266 static int json_int(void *ctx, long long val) {
 267     LOG("int %lld for key %s\n", val, last_key);
 268 #else
 269 static int json_int(void *ctx, long val) {
 270     LOG("int %ld for key %s\n", val, last_key);
 271 #endif
 272     if (strcasecmp(last_key, "type") == 0)
 273         json_node->type = val;
 274
 275     if (strcasecmp(last_key, "fullscreen_mode") == 0)
 276         json_node->fullscreen_mode = val;
 277
 278     if (strcasecmp(last_key, "num") == 0)
 279         json_node->num = val;
 280
 281     if (!parsing_swallows && strcasecmp(last_key, "id") == 0)
 282         json_node->old_id = val;
 283
 284     if (parsing_focus) {
 285         struct focus_mapping *focus_mapping = scalloc(sizeof(struct focus_mapping));
 286         focus_mapping->old_id = val;
 287         TAILQ_INSERT_TAIL(&focus_mappings, focus_mapping, focus_mappings);
 288     }
 289
 290     if (parsing_rect || parsing_window_rect || parsing_geometry) {
 291         Rect *r;
 292         if (parsing_rect)
 293             r = &(json_node->rect);
 294         else if (parsing_window_rect)
 295             r = &(json_node->window_rect);
 296         else r = &(json_node->geometry);
 297         if (strcasecmp(last_key, "x") == 0)
 298             r->x = val;
 299         else if (strcasecmp(last_key, "y") == 0)
 300             r->y = val;
 301         else if (strcasecmp(last_key, "width") == 0)
 302             r->width = val;
 303         else if (strcasecmp(last_key, "height") == 0)
 304             r->height = val;
 305         else printf("WARNING: unknown key %s in rect\n", last_key);
 306         printf("rect now: (%d, %d, %d, %d)\n",
 307                 r->x, r->y, r->width, r->height);
 308     }
 309     if (parsing_swallows) {
 310         if (strcasecmp(last_key, "id") == 0) {
 311             current_swallow->id = val;
 312         }
 313         if (strcasecmp(last_key, "dock") == 0) {
 314             current_swallow->dock = val;
 315         }
 316         if (strcasecmp(last_key, "insert_where") == 0) {
 317             current_swallow->insert_where = val;
 318         }
 319     }
 320
 321     return 1;
 322 }
 323
 324 static int json_bool(void *ctx, int val) {
 325     LOG("bool %d for key %s\n", val, last_key);
 326     if (strcasecmp(last_key, "focused") == 0 && val) {
 327         to_focus = json_node;
 328     }
 329
 330     if (strcasecmp(last_key, "split") == 0)
 331         json_node->split = val;
 332
 333     if (parsing_swallows) {
 334         if (strcasecmp(last_key, "restart_mode") == 0)
 335             current_swallow->restart_mode = val;
 336     }
 337
 338     return 1;
 339 }
 340
 341 static int json_double(void *ctx, double val) {
 342     LOG("double %f for key %s\n", val, last_key);
 343     if (strcasecmp(last_key, "percent") == 0) {
 344         json_node->percent = val;
 345     }
 346     return 1;
 347 }
 348
 349 void tree_append_json(const char *filename) {
 350     /* TODO: percent of other windows are not correctly fixed at the moment */
 351     FILE *f;
 352     if ((f = fopen(filename, "r")) == NULL) {
 353         LOG("Cannot open file\n");
 354         return;
 355     }
 356     char *buf = malloc(65535); /* TODO */
 357     int n = fread(buf, 1, 65535, f);
 358     LOG("read %d bytes\n", n);
 359     yajl_gen g;
 360     yajl_handle hand;
 361     yajl_callbacks callbacks;
 362     memset(&callbacks, '\0', sizeof(yajl_callbacks));
 363     callbacks.yajl_start_map = json_start_map;
 364     callbacks.yajl_end_map = json_end_map;
 365     callbacks.yajl_end_array = json_end_array;
 366     callbacks.yajl_string = json_string;
 367     callbacks.yajl_map_key = json_key;
 368     callbacks.yajl_integer = json_int;
 369     callbacks.yajl_double = json_double;
 370     callbacks.yajl_boolean = json_bool;
 371 #if YAJL_MAJOR >= 2
 372     g = yajl_gen_alloc(NULL);
 373     hand = yajl_alloc(&callbacks, NULL, (void*)g);
 374 #else
 375     g = yajl_gen_alloc(NULL, NULL);
 376     hand = yajl_alloc(&callbacks, NULL, NULL, (void*)g);
 377 #endif
 378     yajl_status stat;
 379     json_node = focused;
 380     to_focus = NULL;
 381     parsing_rect = false;
 382     parsing_window_rect = false;
 383     parsing_geometry = false;
 384     setlocale(LC_NUMERIC, "C");
 385     stat = yajl_parse(hand, (const unsigned char*)buf, n);
 386     if (stat != yajl_status_ok)
 387     {
 388         unsigned char * str = yajl_get_error(hand, 1, (const unsigned char*)buf, n);
 389         fprintf(stderr, "%s\n", (const char *) str);
 390         yajl_free_error(hand, str);
 391     }
 392
 393     setlocale(LC_NUMERIC, "");
 394 #if YAJL_MAJOR >= 2
 395     yajl_complete_parse(hand);
 396 #else
 397     yajl_parse_complete(hand);
 398 #endif
 399
 400     fclose(f);
 401     if (to_focus)
 402         con_focus(to_focus);
 403 }