util/wmmenugen_parse_xdg.c

   1 /*
   2  * wmmenugen - Window Maker PropList menu generator
   3  *
   4  * Desktop Entry Specification parser functions
   5  *
   6  * Copyright (c) 2010. Tamas Tevesz <ice@extreme.hu>
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License along
  19  * with this program; if not, write to the Free Software Foundation, Inc.,
  20  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21  */
  22
  23 /*
  24  * http://standards.freedesktop.org/desktop-entry-spec/desktop-entry-spec-1.1.html
  25  * http://standards.freedesktop.org/menu-spec/menu-spec-1.1.html
  26  *
  27  * We will only deal with Type == "Application" entries in [Desktop Entry]
  28  * groups. Since there is no passing of file name arguments or anything of
  29  * the sort to applications from the menu, execname is determined as follows:
  30  * - If `TryExec' is present, use that;
  31  * - else use `Exec' with any switches stripped
  32  *
  33  * Only the (first, though there should not be more than one) `Main Category'
  34  * is used to place the entry in a submenu.
  35  *
  36  * Basic validation of the .desktop file is done.
  37  */
  38
  39 #include <sys/types.h>
  40 #include <sys/stat.h>
  41
  42 #include <ctype.h>
  43 #include <ftw.h>
  44 #if DEBUG
  45 #include <errno.h>
  46 #endif
  47 #include <stdio.h>
  48 #include <stdlib.h>
  49 #include <string.h>
  50
  51 #include "wmmenugen.h"
  52
  53 /* LocaleString match levels */
  54 enum {
  55         MATCH_DEFAULT,
  56         MATCH_LANG,
  57         MATCH_LANG_MODIFIER,
  58         MATCH_LANG_COUNTRY,
  59         MATCH_LANG_COUNTRY_MODIFIER
  60 };
  61
  62 typedef struct {
  63         char    *Name;          /* Name */                              /* localestring */
  64         int      MatchLevel;    /* LocaleString match type */           /* int */
  65         char    *TryExec;       /* TryExec */                           /* string */
  66         char    *Exec;          /* Exec */                              /* string */
  67         char    *Path;          /* Path */                              /* string */
  68         int      Flags;         /* Flags */
  69         char    *Category;      /* Categories (first item only) */      /* string */
  70 } XDGMenuEntry;
  71
  72 static void  getKey(char **target, const char *line);
  73 static void  getStringValue(char **target, const char *line);
  74 static void  getLocalizedStringValue(char **target, const char *line, int *match_level);
  75 static int   getBooleanValue(const char *line);
  76 static void  getMenuHierarchyFor(char **xdgmenuspec);
  77 static int   compare_matchlevel(int *current_level, const char *found_locale);
  78 static Bool  xdg_to_wm(XDGMenuEntry **xdg, WMMenuEntry **wmentry);
  79 static void init_xdg_storage(XDGMenuEntry **xdg);
  80 static void init_wm_storage(WMMenuEntry **wm);
  81
  82
  83 void parse_xdg(const char *file, cb_add_menu_entry *addWMMenuEntryCallback)
  84 {
  85         FILE *fp;
  86         char buf[1024];
  87         char *p, *tmp, *key;
  88         WMMenuEntry *wm;
  89         XDGMenuEntry *xdg;
  90         int InGroup;
  91
  92         fp = fopen(file, "r");
  93         if (!fp) {
  94 #if DEBUG
  95                 fprintf(stderr, "Error opening file %s: %s\n", file, strerror(errno));
  96 #endif
  97                 return;
  98         }
  99
 100         xdg = (XDGMenuEntry *)wmalloc(sizeof(XDGMenuEntry));
 101         wm = (WMMenuEntry *)wmalloc(sizeof(WMMenuEntry));
 102         InGroup = 0;
 103         memset(buf, 0, sizeof(buf));
 104
 105         while (fgets(buf, sizeof(buf), fp)) {
 106
 107                 p = buf;
 108
 109                 /* skip whitespaces */
 110                 while (isspace(*p))
 111                         p++;
 112                 /* skip comments, empty lines */
 113                 if (*p == '\r' || *p == '\n' || *p == '#') {
 114                         memset(buf, 0, sizeof(buf));
 115                         continue;
 116                 }
 117                 /* trim crlf */
 118                 buf[strcspn(buf, "\r\n")] = '\0';
 119                 if (strlen(buf) == 0)
 120                         continue;
 121
 122                 if (strcmp(p, "[Desktop Entry]") == 0) {
 123                         /* if currently processing a group, we've just hit the
 124                          * end of its definition, try processing it
 125                          */
 126                         if (InGroup && xdg_to_wm(&xdg, &wm)) {
 127                                 (*addWMMenuEntryCallback)(wm);
 128                         }
 129                         init_xdg_storage(&xdg);
 130                         init_wm_storage(&wm);
 131                         InGroup = 1;
 132                         /* start processing group */
 133                         memset(buf, 0, sizeof(buf));
 134                         continue;
 135                 } else if (p[0] == '[') {
 136                         /* If we find a new group and the previous group was the main one,
 137                          * we stop all further processing
 138                          */
 139                         if (InGroup) break;
 140                 }
 141
 142                 if (!InGroup) {
 143                         memset(buf, 0, sizeof(buf));
 144                         continue;
 145                 }
 146
 147                 getKey(&key, p);
 148                 if (key == NULL) { /* not `key' = `value' */
 149                         memset(buf, 0, sizeof(buf));
 150                         continue;
 151                 }
 152
 153                 if (strcmp(key, "Type") == 0) {
 154                         getStringValue(&tmp, p);
 155                         if (strcmp(tmp, "Application") != 0)
 156                                 InGroup = 0;    /* if not application, skip current group */
 157                         wfree(tmp);
 158                         tmp = NULL;
 159                 } else if (strcmp(key, "Name") == 0) {
 160                         getLocalizedStringValue(&xdg->Name, p, &xdg->MatchLevel);
 161                 } else if (strcmp(key, "NoDisplay") == 0) {
 162                         if (getBooleanValue(p)) /* if nodisplay, skip current group */
 163                                 InGroup = 0;
 164                 } else if (strcmp(key, "Hidden") == 0) {
 165                         if (getBooleanValue(p))
 166                                 InGroup = 0;    /* if hidden, skip current group */
 167                 } else if (strcmp(key, "TryExec") == 0) {
 168                         getStringValue(&xdg->TryExec, p);
 169                 } else if (strcmp(key, "Exec") == 0) {
 170                         getStringValue(&xdg->Exec, p);
 171                 } else if (strcmp(key, "Path") == 0) {
 172                         getStringValue(&xdg->Path, p);
 173                 } else if (strcmp(key, "Terminal") == 0) {
 174                         if (getBooleanValue(p))
 175                                 xdg->Flags |= F_TERMINAL;
 176                 } else if (strcmp(key, "Categories") == 0) {
 177                         getStringValue(&xdg->Category, p);
 178                         getMenuHierarchyFor(&xdg->Category);
 179                 }
 180
 181                 if (xdg->Category == NULL)
 182                         xdg->Category = wstrdup(_("Other"));
 183
 184                 wfree(key);
 185                 key = NULL;
 186         }
 187
 188         fclose(fp);
 189
 190         /* at the end of the file, might as well try to menuize what we have
 191          * unless there was no group at all or it was marked as hidden
 192          */
 193         if (InGroup && xdg_to_wm(&xdg, &wm))
 194                 (*addWMMenuEntryCallback)(wm);
 195
 196 }
 197
 198
 199 /* coerce an xdg entry type into a wm entry type
 200  */
 201 static Bool xdg_to_wm(XDGMenuEntry **xdg, WMMenuEntry **wm)
 202 {
 203         char *p;
 204
 205         /* Exec or TryExec is mandatory */
 206         if (!((*xdg)->Exec || (*xdg)->TryExec))
 207                 return False;
 208
 209         /* if there's no Name, use the first word of Exec or TryExec
 210          */
 211         if ((*xdg)->Name) {
 212                 (*wm)->Name = (*xdg)->Name;
 213         } else  {
 214                 if ((*xdg)->Exec)
 215                         (*wm)->Name = wstrdup((*xdg)->Exec);
 216                 else /* (*xdg)->TryExec */
 217                         (*wm)->Name = wstrdup((*xdg)->TryExec);
 218
 219                 p = strchr((*wm)->Name, ' ');
 220                 if (p)
 221                         *p = '\0';
 222         }
 223
 224         if ((*xdg)->Exec)
 225                 (*wm)->CmdLine = (*xdg)->Exec;
 226         else                                    /* (*xdg)->TryExec */
 227                 (*wm)->CmdLine = (*xdg)->TryExec;
 228
 229         (*wm)->SubMenu = (*xdg)->Category;
 230         (*wm)->Flags = (*xdg)->Flags;
 231
 232         return True;
 233 }
 234
 235 /* (re-)initialize a XDGMenuEntry storage
 236  */
 237 static void init_xdg_storage(XDGMenuEntry **xdg)
 238 {
 239
 240         if ((*xdg)->Name)
 241                 wfree((*xdg)->Name);
 242         if ((*xdg)->TryExec)
 243                 wfree((*xdg)->TryExec);
 244         if ((*xdg)->Exec)
 245                 wfree((*xdg)->Exec);
 246         if ((*xdg)->Category)
 247                 wfree((*xdg)->Category);
 248         if ((*xdg)->Path)
 249                 wfree((*xdg)->Path);
 250
 251         (*xdg)->Name = NULL;
 252         (*xdg)->TryExec = NULL;
 253         (*xdg)->Exec = NULL;
 254         (*xdg)->Category = NULL;
 255         (*xdg)->Path = NULL;
 256         (*xdg)->Flags = 0;
 257         (*xdg)->MatchLevel = -1;
 258 }
 259
 260 /* (re-)initialize a WMMenuEntry storage
 261  */
 262 static void init_wm_storage(WMMenuEntry **wm)
 263 {
 264         (*wm)->Name = NULL;
 265         (*wm)->CmdLine = NULL;
 266         (*wm)->Flags = 0;
 267 }
 268
 269 /* get a key from line. allocates target, which must be wfreed later */
 270 static void getKey(char **target, const char *line)
 271 {
 272         const char *p;
 273         int kstart, kend;
 274
 275         p = line;
 276
 277         if (strchr(p, '=') == NULL) {           /* not `key' = `value' */
 278                 *target = NULL;
 279                 return;
 280         }
 281
 282         kstart = 0;
 283
 284         /* skip whitespace */
 285         while (isspace(*(p + kstart)))
 286                 kstart++;
 287
 288         /* skip up until first whitespace or '[' (localestring) or '=' */
 289         kend = kstart + 1;
 290         while (*(p + kend) && !isspace(*(p + kend)) && *(p + kend) != '=' && *(p + kend) != '[')
 291                 kend++;
 292
 293         *target = wstrndup(p + kstart, kend - kstart);
 294 }
 295
 296 /* get a string value from line. allocates target, which must be wfreed later. */
 297 static void getStringValue(char **target, const char *line)
 298 {
 299         const char *p;
 300         int kstart;
 301
 302         p = line;
 303         kstart = 0;
 304
 305         /* skip until after '=' */
 306         while (*(p + kstart) && *(p + kstart) != '=')
 307                 kstart++;
 308         kstart++;
 309
 310         /* skip whitespace */
 311         while (*(p + kstart) && isspace(*(p + kstart)))
 312                 kstart++;
 313
 314         *target = wstrdup(p + kstart);
 315 }
 316
 317 /* get a localized string value from line. allocates target, which must be wfreed later.
 318  * matching is dependent on the current value of target as well as on the
 319  * level the current value is matched on. guts matching algorithm is in
 320  * compare_matchlevel().
 321  */
 322 static void getLocalizedStringValue(char **target, const char *line, int *match_level)
 323 {
 324         const char *p;
 325         char *locale;
 326         int kstart;
 327         int sqbstart, sqbend;
 328
 329         p = line;
 330         kstart = 0;
 331         sqbstart = 0;
 332         sqbend = 0;
 333         locale = NULL;
 334
 335         /* skip until after '=', mark if '[' and ']' is found */
 336         while (*(p + kstart) && *(p + kstart) != '=') {
 337                 switch (*(p + kstart)) {
 338                         case '[': sqbstart = kstart + 1;break;
 339                         case ']': sqbend = kstart;      break;
 340                         default :                       break;
 341                 }
 342                 kstart++;
 343         }
 344         kstart++;
 345
 346         /* skip whitespace */
 347         while (isspace(*(p + kstart)))
 348                 kstart++;
 349
 350         if (sqbstart > 0 && sqbend > sqbstart)
 351                 locale = wstrndup(p + sqbstart, sqbend - sqbstart);
 352
 353         /* if there is no value yet and this is the default key, return */
 354         if (!*target && !locale) {
 355                 *match_level = MATCH_DEFAULT;
 356                 *target = wstrdup(p + kstart);
 357                 return;
 358         }
 359
 360         if (compare_matchlevel(match_level, locale)) {
 361                 wfree(locale);
 362                 *target = wstrdup(p + kstart);
 363                 return;
 364         }
 365
 366         return;
 367 }
 368
 369 /* get a boolean value from line */
 370 static Bool getBooleanValue(const char *line)
 371 {
 372         char *p;
 373         int ret;
 374
 375         getStringValue(&p, line);
 376         ret = strcmp(p, "true") == 0 ? True : False;
 377         wfree(p);
 378
 379         return ret;
 380 }
 381
 382 /* perform locale matching by implementing the algorithm specified in
 383  * xdg desktop entry specification, section "localized values for keys".
 384  */
 385 static Bool compare_matchlevel(int *current_level, const char *found_locale)
 386 {
 387         /* current key locale */
 388         char *key_lang, *key_ctry, *key_enc, *key_mod;
 389
 390         parse_locale(found_locale, &key_lang, &key_ctry, &key_enc, &key_mod);
 391
 392         if (env_lang && key_lang &&             /* Shortcut: if key and env languages don't match, */
 393             strcmp(env_lang, key_lang) != 0)    /* don't even bother. This takes care of the great */
 394                 return False;                   /* majority of the cases without having to go through */
 395                                                 /* the more theoretical parts of the spec'd algo. */
 396
 397         if (!env_mod && key_mod)        /* If LC_MESSAGES does not have a MODIFIER field, */
 398                 return False;           /* then no key with a modifier will be matched. */
 399
 400         if (!env_ctry && key_ctry)      /* Similarly, if LC_MESSAGES does not have a COUNTRY field, */
 401                 return False;           /* then no key with a country specified will be matched. */
 402
 403         /* LC_MESSAGES value: lang_COUNTRY@MODIFIER */
 404         if (env_lang && env_ctry && env_mod) {          /* lang_COUNTRY@MODIFIER */
 405                 if (key_lang && key_ctry && key_mod &&
 406                     strcmp(env_lang, key_lang) == 0 &&
 407                     strcmp(env_ctry, key_ctry) == 0 &&
 408                     strcmp(env_mod,  key_mod) == 0) {
 409                         *current_level = MATCH_LANG_COUNTRY_MODIFIER;
 410                         return True;
 411                 } else if (key_lang && key_ctry &&      /* lang_COUNTRY */
 412                     strcmp(env_lang, key_lang) == 0 &&
 413                     strcmp(env_ctry, key_ctry) == 0 &&
 414                     *current_level < MATCH_LANG_COUNTRY) {
 415                         *current_level = MATCH_LANG_COUNTRY;
 416                         return True;
 417                 } else if (key_lang && key_mod &&       /* lang@MODIFIER */
 418                     strcmp(env_lang, key_lang) == 0 &&
 419                     strcmp(env_mod,  key_mod) == 0 &&
 420                     *current_level < MATCH_LANG_MODIFIER) {
 421                         *current_level = MATCH_LANG_MODIFIER;
 422                         return True;
 423                 } else if (key_lang &&                  /* lang */
 424                     strcmp(env_lang, key_lang) == 0 &&
 425                     *current_level < MATCH_LANG) {
 426                         *current_level = MATCH_LANG;
 427                         return True;
 428                 } else {
 429                         return False;
 430                 }
 431         }
 432
 433         /* LC_MESSAGES value: lang_COUNTRY */
 434         if (env_lang && env_ctry) {                     /* lang_COUNTRY */
 435                 if (key_lang && key_ctry &&
 436                     strcmp(env_lang, key_lang) == 0 &&
 437                     strcmp(env_ctry, key_ctry) == 0 &&
 438                     *current_level < MATCH_LANG_COUNTRY) {
 439                         *current_level = MATCH_LANG_COUNTRY;
 440                         return True;
 441                 } else if (key_lang &&                  /* lang */
 442                     strcmp(env_lang, key_lang) == 0 &&
 443                     *current_level < MATCH_LANG) {
 444                         *current_level = MATCH_LANG;
 445                         return True;
 446                 } else {
 447                         return False;
 448                 }
 449         }
 450
 451         /* LC_MESSAGES value: lang@MODIFIER */
 452         if (env_lang && env_mod) {                      /* lang@MODIFIER */
 453                 if (key_lang && key_mod &&
 454                     strcmp(env_lang, key_lang) == 0 &&
 455                     strcmp(env_mod,  key_mod) == 0 &&
 456                     *current_level < MATCH_LANG_MODIFIER) {
 457                         *current_level = MATCH_LANG_MODIFIER;
 458                         return True;
 459                 } else if (key_lang &&                  /* lang */
 460                     strcmp(env_lang, key_lang) == 0 &&
 461                     *current_level < MATCH_LANG) {
 462                         *current_level = MATCH_LANG;
 463                         return True;
 464                 } else {
 465                         return False;
 466                 }
 467         }
 468
 469         /* LC_MESSAGES value: lang */
 470         if (env_lang) {                                 /* lang */
 471                 if (key_lang &&
 472                     strcmp(env_lang, key_lang) == 0 &&
 473                     *current_level < MATCH_LANG) {
 474                         *current_level = MATCH_LANG;
 475                         return True;
 476                 } else {
 477                         return False;
 478                 }
 479         }
 480
 481         /* MATCH_DEFAULT is handled in getLocalizedStringValue */
 482
 483         return False;
 484 }
 485
 486 /* get the (first) xdg main category from a list of categories
 487  */
 488 static void  getMenuHierarchyFor(char **xdgmenuspec)
 489 {
 490         char *category, *p;
 491         char buf[1024];
 492
 493         if (!*xdgmenuspec || !**xdgmenuspec)
 494                 return;
 495
 496         category = wstrdup(*xdgmenuspec);
 497         wfree(*xdgmenuspec);
 498         memset(buf, 0, sizeof(buf));
 499
 500         p = strtok(category, ";");
 501         while (p) {             /* get a known category */
 502                 if (strcmp(p, "AudioVideo") == 0) {
 503                         snprintf(buf, sizeof(buf), "%s", _("Audio & Video"));
 504                         break;
 505                 } else if (strcmp(p, "Audio") == 0) {
 506                         snprintf(buf, sizeof(buf), "%s", _("Audio"));
 507                         break;
 508                 } else if (strcmp(p, "Video") == 0) {
 509                         snprintf(buf, sizeof(buf), "%s", _("Video"));
 510                         break;
 511                 } else if (strcmp(p, "Development") == 0) {
 512                         snprintf(buf, sizeof(buf), "%s", _("Development"));
 513                         break;
 514                 } else if (strcmp(p, "Education") == 0) {
 515                         snprintf(buf, sizeof(buf), "%s", _("Education"));
 516                         break;
 517                 } else if (strcmp(p, "Game") == 0) {
 518                         snprintf(buf, sizeof(buf), "%s", _("Game"));
 519                         break;
 520                 } else if (strcmp(p, "Graphics") == 0) {
 521                         snprintf(buf, sizeof(buf), "%s", _("Graphics"));
 522                         break;
 523                 } else if (strcmp(p, "Network") == 0) {
 524                         snprintf(buf, sizeof(buf), "%s", _("Network"));
 525                         break;
 526                 } else if (strcmp(p, "Office") == 0) {
 527                         snprintf(buf, sizeof(buf), "%s", _("Office"));
 528                         break;
 529                 } else if (strcmp(p, "Science") == 0) {
 530                         snprintf(buf, sizeof(buf), "%s", _("Science"));
 531                         break;
 532                 } else if (strcmp(p, "Settings") == 0) {
 533                         snprintf(buf, sizeof(buf), "%s", _("Settings"));
 534                         break;
 535                 } else if (strcmp(p, "System") == 0) {
 536                         snprintf(buf, sizeof(buf), "%s", _("System"));
 537                         break;
 538                 } else if (strcmp(p, "Utility") == 0) {
 539                         snprintf(buf, sizeof(buf), "%s", _("Utility"));
 540                         break;
 541                 /* reserved categories */
 542                 } else if (strcmp(p, "Screensaver") == 0) {
 543                         snprintf(buf, sizeof(buf), "%s", _("Screensaver"));
 544                         break;
 545                 } else if (strcmp(p, "TrayIcon") == 0) {
 546                         snprintf(buf, sizeof(buf), "%s", _("Tray Icon"));
 547                         break;
 548                 } else if (strcmp(p, "Applet") == 0) {
 549                         snprintf(buf, sizeof(buf), "%s", _("Applet"));
 550                         break;
 551                 } else if (strcmp(p, "Shell") == 0) {
 552                         snprintf(buf, sizeof(buf), "%s", _("Shell"));
 553                         break;
 554                 }
 555                 p = strtok(NULL, ";");
 556         }
 557
 558
 559         if (!*buf)              /* come up with something if nothing found */
 560                 snprintf(buf, sizeof(buf), "%s", _("Other"));
 561
 562         *xdgmenuspec = wstrdup(buf);
 563 }