obt/xml.c

   1 /* -*- indent-tabs-mode: nil; tab-width: 4; c-basic-offset: 4; -*-
   2
   3    obt/xml.c for the Openbox window manager
   4    Copyright (c) 2003-2007   Dana Jansens
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 2 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    See the COPYING file for a copy of the GNU General Public License.
  17 */
  18
  19 #include "obt/xml.h"
  20 #include "obt/paths.h"
  21
  22 #include <libxml/xinclude.h>
  23 #include <glib.h>
  24
  25 #ifdef HAVE_STDLIB_H
  26 #  include <stdlib.h>
  27 #endif
  28 #ifdef HAVE_SYS_STAT_H
  29 #  include <sys/stat.h>
  30 #endif
  31 #ifdef HAVE_SYS_TYPES_H
  32 #  include <sys/types.h>
  33 #endif
  34 #ifdef HAVE_UNISTD_H
  35 #  include <unistd.h>
  36 #endif
  37
  38 struct Callback {
  39     gchar *tag;
  40     ObtXmlCallback func;
  41     gpointer data;
  42 };
  43
  44 struct _ObtXmlInst {
  45     gint ref;
  46     ObtPaths *xdg_paths;
  47     GHashTable *callbacks;
  48     xmlDocPtr doc;
  49     xmlNodePtr root;
  50     gchar *path;
  51 };
  52
  53 static void destfunc(struct Callback *c)
  54 {
  55     g_free(c->tag);
  56     g_slice_free(struct Callback, c);
  57 }
  58
  59 ObtXmlInst* obt_xml_instance_new(void)
  60 {
  61     ObtXmlInst *i = g_slice_new(ObtXmlInst);
  62     i->ref = 1;
  63     i->xdg_paths = obt_paths_new();
  64     i->callbacks = g_hash_table_new_full(g_str_hash, g_str_equal, NULL,
  65                                          (GDestroyNotify)destfunc);
  66     i->doc = NULL;
  67     i->root = NULL;
  68     i->path = NULL;
  69     return i;
  70 }
  71
  72 void obt_xml_instance_ref(ObtXmlInst *i)
  73 {
  74     ++i->ref;
  75 }
  76
  77 void obt_xml_instance_unref(ObtXmlInst *i)
  78 {
  79     if (i && --i->ref == 0) {
  80         obt_paths_unref(i->xdg_paths);
  81         g_hash_table_destroy(i->callbacks);
  82         g_slice_free(ObtXmlInst, i);
  83     }
  84 }
  85
  86 xmlDocPtr obt_xml_doc(ObtXmlInst *i)
  87 {
  88     g_assert(i->doc); /* a doc is open? */
  89     return i->doc;
  90 }
  91
  92 xmlNodePtr obt_xml_root(ObtXmlInst *i)
  93 {
  94     g_assert(i->doc); /* a doc is open? */
  95     return i->root;
  96 }
  97
  98 void obt_xml_register(ObtXmlInst *i, const gchar *tag,
  99                       ObtXmlCallback func, gpointer data)
 100 {
 101     struct Callback *c;
 102
 103     if (g_hash_table_lookup(i->callbacks, tag)) {
 104         g_error("Tag '%s' already registered", tag);
 105         return;
 106     }
 107
 108     c = g_slice_new(struct Callback);
 109     c->tag = g_strdup(tag);
 110     c->func = func;
 111     c->data = data;
 112     g_hash_table_insert(i->callbacks, c->tag, c);
 113 }
 114
 115 void obt_xml_unregister(ObtXmlInst *i, const gchar *tag)
 116 {
 117     g_hash_table_remove(i->callbacks, tag);
 118 }
 119
 120 void obt_xml_new_file(ObtXmlInst *i,
 121                       const gchar *root_node)
 122 {
 123     xmlNodePtr old;
 124
 125     g_assert(i->doc == NULL); /* another doc isn't open already? */
 126
 127     i->doc = xmlNewDoc((xmlChar*)"1.0");
 128     i->root = xmlNewDocRawNode(i->doc, NULL, (xmlChar*)root_node, NULL);
 129     old = xmlDocSetRootElement(i->doc, i->root);
 130     if (old) xmlFreeNode(old);
 131 }
 132
 133 static gboolean load_file(ObtXmlInst *i,
 134                           const gchar *domain,
 135                           const gchar *filename,
 136                           const gchar *root_node,
 137                           GSList *paths)
 138 {
 139     GSList *it;
 140     gboolean r = FALSE;
 141
 142     g_assert(i->doc == NULL); /* another doc isn't open already? */
 143
 144     for (it = paths; !r && it; it = g_slist_next(it)) {
 145         gchar *path;
 146         struct stat s;
 147
 148         if (!domain && !filename) /* given a full path to the file */
 149             path = g_strdup(it->data);
 150         else
 151             path = g_build_filename(it->data, domain, filename, NULL);
 152
 153         if (stat(path, &s) >= 0) {
 154             /* XML_PARSE_BLANKS is needed apparently, or the tree can end up
 155                with extra nodes in it. */
 156             i->doc = xmlReadFile(path, NULL, (XML_PARSE_NOBLANKS |
 157                                               XML_PARSE_RECOVER));
 158             xmlXIncludeProcessFlags(i->doc, (XML_PARSE_NOBLANKS |
 159                                              XML_PARSE_RECOVER));
 160             if (i->doc) {
 161                 i->root = xmlDocGetRootElement(i->doc);
 162                 if (!i->root) {
 163                     xmlFreeDoc(i->doc);
 164                     i->doc = NULL;
 165                     g_message("%s is an empty XML document", path);
 166                 }
 167                 else if (xmlStrcmp(i->root->name,
 168                                    (const xmlChar*)root_node)) {
 169                     xmlFreeDoc(i->doc);
 170                     i->doc = NULL;
 171                     i->root = NULL;
 172                     g_message("XML document %s is of wrong type. Root "
 173                               "node is not '%s'", path, root_node);
 174                 }
 175                 else {
 176                     i->path = g_strdup(path);
 177                     r = TRUE; /* ok! */
 178                 }
 179             }
 180         }
 181
 182         g_free(path);
 183     }
 184
 185     return r;
 186 }
 187
 188 gboolean obt_xml_load_file(ObtXmlInst *i,
 189                            const gchar *path,
 190                            const gchar *root_node)
 191 {
 192     GSList *paths;
 193     gboolean r;
 194
 195     paths = g_slist_append(NULL, g_strdup(path));
 196
 197     r = load_file(i, NULL, NULL, root_node, paths);
 198
 199     while (paths) {
 200         g_free(paths->data);
 201         paths = g_slist_delete_link(paths, paths);
 202     }
 203     return r;
 204 }
 205
 206 gboolean obt_xml_load_cache_file(ObtXmlInst *i,
 207                                  const gchar *domain,
 208                                  const gchar *filename,
 209                                  const gchar *root_node)
 210 {
 211     GSList *paths = NULL;
 212     gboolean r;
 213
 214     paths = g_slist_append(paths,
 215                            g_strdup(obt_paths_cache_home(i->xdg_paths)));
 216
 217     r = load_file(i, domain, filename, root_node, paths);
 218
 219     while (paths) {
 220         g_free(paths->data);
 221         paths = g_slist_delete_link(paths, paths);
 222     }
 223     return r;
 224 }
 225
 226 gboolean obt_xml_load_config_file(ObtXmlInst *i,
 227                                   const gchar *domain,
 228                                   const gchar *filename,
 229                                   const gchar *root_node)
 230 {
 231     GSList *it, *paths = NULL;
 232     gboolean r;
 233
 234     for (it = obt_paths_config_dirs(i->xdg_paths); it; it = g_slist_next(it))
 235         paths = g_slist_append(paths, g_strdup(it->data));
 236
 237     r = load_file(i, domain, filename, root_node, paths);
 238
 239     while (paths) {
 240         g_free(paths->data);
 241         paths = g_slist_delete_link(paths, paths);
 242     }
 243     return r;
 244 }
 245
 246 gboolean obt_xml_load_data_file(ObtXmlInst *i,
 247                                 const gchar *domain,
 248                                 const gchar *filename,
 249                                 const gchar *root_node)
 250 {
 251     GSList *it, *paths = NULL;
 252     gboolean r;
 253
 254     for (it = obt_paths_data_dirs(i->xdg_paths); it; it = g_slist_next(it))
 255         paths = g_slist_append(paths, g_strdup(it->data));
 256
 257     r = load_file(i, domain, filename, root_node, paths);
 258
 259     while (paths) {
 260         g_free(paths->data);
 261         paths = g_slist_delete_link(paths, paths);
 262     }
 263     return r;
 264 }
 265
 266 gboolean obt_xml_load_theme_file(ObtXmlInst *i,
 267                                  const gchar *theme,
 268                                  const gchar *domain,
 269                                  const gchar *filename,
 270                                  const gchar *root_node)
 271 {
 272     GSList *it, *paths = NULL;
 273     gboolean r;
 274
 275     /* use ~/.themes for backwards compatibility */
 276     paths = g_slist_append
 277         (paths, g_build_filename(g_get_home_dir(), ".themes", theme, NULL));
 278
 279     for (it = obt_paths_data_dirs(i->xdg_paths); it; it = g_slist_next(it))
 280         paths = g_slist_append
 281             (paths, g_build_filename(it->data, "themes", theme, NULL));
 282
 283     r = load_file(i, domain, filename, root_node, paths);
 284
 285     while (paths) {
 286         g_free(paths->data);
 287         paths = g_slist_delete_link(paths, paths);
 288     }
 289     return r;
 290 }
 291
 292
 293 gboolean obt_xml_load_mem(ObtXmlInst *i,
 294                           gpointer data, guint len, const gchar *root_node)
 295 {
 296     gboolean r = FALSE;
 297
 298     g_assert(i->doc == NULL); /* another doc isn't open already? */
 299
 300     i->doc = xmlParseMemory(data, len);
 301     if (i) {
 302         i->root = xmlDocGetRootElement(i->doc);
 303         if (!i->root) {
 304             xmlFreeDoc(i->doc);
 305             i->doc = NULL;
 306             g_message("Given memory is an empty document");
 307         }
 308         else if (xmlStrcmp(i->root->name, (const xmlChar*)root_node)) {
 309             xmlFreeDoc(i->doc);
 310             i->doc = NULL;
 311             i->root = NULL;
 312             g_message("XML Document in given memory is of wrong "
 313                       "type. Root node is not '%s'\n", root_node);
 314         }
 315         else
 316             r = TRUE; /* ok ! */
 317     }
 318     return r;
 319 }
 320
 321 gboolean obt_xml_save_file(ObtXmlInst *inst,
 322                            const gchar *path,
 323                            gboolean pretty)
 324 {
 325     return xmlSaveFormatFile(path, inst->doc, pretty) != -1;
 326 }
 327
 328 void obt_xml_close(ObtXmlInst *i)
 329 {
 330     if (i && i->doc) {
 331         xmlFreeDoc(i->doc);
 332         g_free(i->path);
 333         i->doc = NULL;
 334         i->root = NULL;
 335         i->path = NULL;
 336     }
 337 }
 338
 339 void obt_xml_tree(ObtXmlInst *i, xmlNodePtr node)
 340 {
 341     g_assert(i->doc); /* a doc is open? */
 342
 343     while (node) {
 344         if (node->name) {
 345             struct Callback *c = g_hash_table_lookup(i->callbacks, node->name);
 346             if (c) c->func(node, c->data);
 347         }
 348         node = node->next;
 349     }
 350 }
 351
 352 void obt_xml_tree_from_root(ObtXmlInst *i)
 353 {
 354     obt_xml_tree(i, i->root->children);
 355 }
 356
 357 gchar *obt_xml_node_string(xmlNodePtr node)
 358 {
 359     xmlChar *c = xmlNodeGetContent(node);
 360     gchar *s;
 361     if (c) g_strstrip((char*)c); /* strip leading/trailing whitespace */
 362     s = g_strdup(c ? (gchar*)c : "");
 363     xmlFree(c);
 364     return s;
 365 }
 366
 367 gint obt_xml_node_int(xmlNodePtr node)
 368 {
 369     xmlChar *c = xmlNodeGetContent(node);
 370     gint i;
 371     if (c) g_strstrip((char*)c); /* strip leading/trailing whitespace */
 372     i = c ? atoi((gchar*)c) : 0;
 373     xmlFree(c);
 374     return i;
 375 }
 376
 377 gboolean obt_xml_node_bool(xmlNodePtr node)
 378 {
 379     xmlChar *c = xmlNodeGetContent(node);
 380     gboolean b = FALSE;
 381     if (c) g_strstrip((char*)c); /* strip leading/trailing whitespace */
 382     if (c && !xmlStrcasecmp(c, (const xmlChar*) "true"))
 383         b = TRUE;
 384     else if (c && !xmlStrcasecmp(c, (const xmlChar*) "yes"))
 385         b = TRUE;
 386     else if (c && !xmlStrcasecmp(c, (const xmlChar*) "on"))
 387         b = TRUE;
 388     xmlFree(c);
 389     return b;
 390 }
 391
 392 gboolean obt_xml_node_contains(xmlNodePtr node, const gchar *val)
 393 {
 394     xmlChar *c = xmlNodeGetContent(node);
 395     gboolean r;
 396     if (c) g_strstrip((char*)c); /* strip leading/trailing whitespace */
 397     r = !xmlStrcasecmp(c, (const xmlChar*) val);
 398     xmlFree(c);
 399     return r;
 400 }
 401
 402 xmlNodePtr obt_xml_find_sibling(xmlNodePtr node, const gchar *tag)
 403 {
 404     while (node) {
 405         if (!xmlStrcmp(node->name, (const xmlChar*) tag))
 406             return node;
 407         node = node->next;
 408     }
 409     return NULL;
 410 }
 411
 412 gboolean obt_xml_attr_bool(xmlNodePtr node, const gchar *name,
 413                            gboolean *value)
 414 {
 415     xmlChar *c = xmlGetProp(node, (const xmlChar*) name);
 416     gboolean r = FALSE;
 417     if (c) {
 418         g_strstrip((char*)c); /* strip leading/trailing whitespace */
 419         if (!xmlStrcasecmp(c, (const xmlChar*) "true"))
 420             *value = TRUE, r = TRUE;
 421         else if (!xmlStrcasecmp(c, (const xmlChar*) "yes"))
 422             *value = TRUE, r = TRUE;
 423         else if (!xmlStrcasecmp(c, (const xmlChar*) "on"))
 424             *value = TRUE, r = TRUE;
 425         else if (!xmlStrcasecmp(c, (const xmlChar*) "false"))
 426             *value = FALSE, r = TRUE;
 427         else if (!xmlStrcasecmp(c, (const xmlChar*) "no"))
 428             *value = FALSE, r = TRUE;
 429         else if (!xmlStrcasecmp(c, (const xmlChar*) "off"))
 430             *value = FALSE, r = TRUE;
 431     }
 432     xmlFree(c);
 433     return r;
 434 }
 435
 436 gboolean obt_xml_attr_int(xmlNodePtr node, const gchar *name, gint *value)
 437 {
 438     xmlChar *c = xmlGetProp(node, (const xmlChar*) name);
 439     gboolean r = FALSE;
 440     if (c) {
 441         g_strstrip((char*)c); /* strip leading/trailing whitespace */
 442         *value = atoi((gchar*)c);
 443         r = TRUE;
 444     }
 445     xmlFree(c);
 446     return r;
 447 }
 448
 449 gboolean obt_xml_attr_string(xmlNodePtr node, const gchar *name,
 450                              gchar **value)
 451 {
 452     xmlChar *c = xmlGetProp(node, (const xmlChar*) name);
 453     gboolean r = FALSE;
 454     if (c) {
 455         g_strstrip((char*)c); /* strip leading/trailing whitespace */
 456         *value = g_strdup((gchar*)c);
 457         r = TRUE;
 458     }
 459     xmlFree(c);
 460     return r;
 461 }
 462
 463 gboolean obt_xml_attr_contains(xmlNodePtr node, const gchar *name,
 464                                const gchar *val)
 465 {
 466     xmlChar *c = xmlGetProp(node, (const xmlChar*) name);
 467     gboolean r = FALSE;
 468     if (c) {
 469         g_strstrip((char*)c); /* strip leading/trailing whitespace */
 470         r = !xmlStrcasecmp(c, (const xmlChar*) val);
 471     }
 472     xmlFree(c);
 473     return r;
 474 }
 475
 476 /*! Finds a sibling which matches a name and attributes.
 477   @first The first sibling in the list.
 478   @attrs NULL-terminated array of strings.  The first string is the name of
 479     the node.  The remaining values are key,value pairs for attributes the
 480     node should have.
 481 */
 482 static xmlNodePtr find_sibling(xmlNodePtr first, gchar **attrs)
 483 {
 484     xmlNodePtr c;
 485     gboolean ok;
 486
 487     ok = FALSE;
 488     c = obt_xml_find_sibling(first, attrs[0]);
 489     while (c && !ok) {
 490         gint i;
 491
 492         ok = TRUE;
 493         for (i = 1; attrs[i]; ++i) {
 494             gchar **eq = g_strsplit(attrs[i], "=", 2);
 495             if (eq[1] && !obt_xml_attr_contains(c, eq[0], eq[1]))
 496                 ok = FALSE;
 497             g_strfreev(eq);
 498         }
 499         if (!ok)
 500             c = obt_xml_find_sibling(c->next, attrs[0]);
 501     }
 502     return ok ? c : NULL;
 503 }
 504
 505 static xmlNodePtr create_child(xmlNodePtr parent, gchar *const*attrs,
 506                                const gchar *value)
 507 {
 508     xmlNodePtr c;
 509     gint i;
 510
 511     c = xmlNewTextChild(parent, NULL, (xmlChar*)attrs[0], (xmlChar*)value);
 512
 513     for (i = 1; attrs[i]; ++i) {
 514         gchar **eq = g_strsplit(attrs[i], "=", 2);
 515         if (eq[1])
 516             xmlNewProp(c, (xmlChar*)eq[0], (xmlChar*)eq[1]);
 517         g_strfreev(eq);
 518     }
 519     return c;
 520 }
 521
 522 xmlNodePtr obt_xml_path_get_node(xmlNodePtr subtree, const gchar *path,
 523                                  const gchar *default_value)
 524 {
 525     xmlNodePtr n, c;
 526     gchar **nodes;
 527     gchar **it, **next;
 528
 529     g_return_val_if_fail(subtree != NULL, NULL);
 530
 531     n = subtree;
 532
 533     nodes = g_strsplit(path, "/", 0);
 534     for (it = nodes; *it; it = next) {
 535         gchar **attrs;
 536
 537         attrs = g_strsplit(*it, ":", 0);
 538         next = it + 1;
 539
 540         /* match attributes */
 541         c = find_sibling(n->children, attrs);
 542
 543         if (!c) {
 544             if (*next)
 545                 c = create_child(n, attrs, NULL);
 546             else if (default_value)
 547                 c = create_child(n, attrs, default_value);
 548         }
 549         n = c;
 550
 551         g_strfreev(attrs);
 552     }
 553
 554     g_strfreev(nodes);
 555
 556     return n;
 557 }
 558
 559 GList* obt_xml_path_get_list(xmlNodePtr subtree, const gchar *path)
 560 {
 561     xmlNodePtr n, c;
 562     gchar **nodes;
 563     gchar **it, **next;
 564     GList *list;
 565
 566     g_return_val_if_fail(subtree != NULL, NULL);
 567
 568     n = subtree;
 569     list = NULL;
 570
 571     nodes = g_strsplit(path, "/", 0);
 572     if (nodes[0]) {
 573         gchar **attrs;
 574
 575         for (it = nodes, next = it + 1; *next; it = next) {
 576             attrs = g_strsplit(*it, ":", 0);
 577             next = it + 1;
 578
 579             /* match attributes */
 580             c = find_sibling(n->children, attrs);
 581             if (!c) c = create_child(n, attrs, NULL);
 582             n = c;
 583
 584             g_strfreev(attrs);
 585         }
 586
 587         attrs = g_strsplit(*next, ":", 0);
 588         c = n->children;
 589         while (c && (c = find_sibling(c, attrs))) {
 590             list = g_list_append(list, c);
 591             c = c->next;
 592         }
 593         g_strfreev(attrs);
 594     }
 595
 596     g_strfreev(nodes);
 597
 598     return list;
 599 }
 600
 601
 602 void obt_xml_path_delete_node(xmlNodePtr subtree, const gchar *path)
 603 {
 604     xmlNodePtr n;
 605
 606     n = obt_xml_path_get_node(subtree, path, NULL);
 607     xmlUnlinkNode(n);
 608     xmlFreeNode(n);
 609 }