luah.c

   1 /*
   2  * luah.c - Lua helper functions
   3  *
   4  * Copyright (C) 2010 Mason Larobina <mason.larobina@gmail.com>
   5  * Copyright (C) 2008-2009 Julien Danjou <julien@danjou.info>
   6  *
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation, either version 3 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  *
  20  */
  21
  22 #include <gtk/gtk.h>
  23 #include <basedir_fs.h>
  24 #include <stdlib.h>
  25 #include "common/util.h"
  26 #include "common/lualib.h"
  27 #include "luakit.h"
  28 #include "widget.h"
  29 #include "luah.h"
  30
  31 void
  32 luaH_modifier_table_push(lua_State *L, guint state) {
  33     gint i = 1;
  34     lua_newtable(L);
  35     if (state & GDK_MODIFIER_MASK) {
  36
  37 #define MODKEY(key, name)           \
  38     if (state & GDK_##key##_MASK) { \
  39         lua_pushstring(L, name);    \
  40         lua_rawseti(L, -2, i++);    \
  41     }
  42
  43         MODKEY(SHIFT, "Shift");
  44         MODKEY(LOCK, "Lock");
  45         MODKEY(CONTROL, "Control");
  46         MODKEY(MOD1, "Mod1");
  47         MODKEY(MOD2, "Mod2");
  48         MODKEY(MOD3, "Mod3");
  49         MODKEY(MOD4, "Mod4");
  50         MODKEY(MOD5, "Mod5");
  51
  52 #undef MODKEY
  53
  54     }
  55 }
  56
  57 void
  58 luaH_keystr_push(lua_State *L, guint keyval)
  59 {
  60     gchar ucs[7];
  61     guint ulen;
  62     guint32 ukval = gdk_keyval_to_unicode(keyval);
  63
  64     /* check for printable unicode character */
  65     if (g_unichar_isgraph(ukval)) {
  66         ulen = g_unichar_to_utf8(ukval, ucs);
  67         ucs[ulen] = 0;
  68         lua_pushstring(L, ucs);
  69     }
  70     /* sent keysym for non-printable characters */
  71     else
  72         lua_pushstring(L, gdk_keyval_name(keyval));
  73 }
  74
  75 /* UTF-8 aware string length computing.
  76  * Returns the number of elements pushed on the stack. */
  77 static gint
  78 luaH_utf8_strlen(lua_State *L)
  79 {
  80     const gchar *cmd  = luaL_checkstring(L, 1);
  81     lua_pushnumber(L, (ssize_t) g_utf8_strlen(NONULL(cmd), -1));
  82     return 1;
  83 }
  84
  85 /* Overload standard Lua next function to use __next key on metatable.
  86  * Returns the number of elements pushed on stack. */
  87 static gint
  88 luaHe_next(lua_State *L)
  89 {
  90     if(luaL_getmetafield(L, 1, "__next")) {
  91         lua_insert(L, 1);
  92         lua_call(L, lua_gettop(L) - 1, LUA_MULTRET);
  93         return lua_gettop(L);
  94     }
  95     luaL_checktype(L, 1, LUA_TTABLE);
  96     lua_settop(L, 2);
  97     if(lua_next(L, 1))
  98         return 2;
  99     lua_pushnil(L);
 100     return 1;
 101 }
 102
 103 /* Overload lua_next() function by using __next metatable field to get
 104  * next elements. `idx` is the index number of elements in stack.
 105  * Returns 1 if more elements to come, 0 otherwise. */
 106 gint
 107 luaH_next(lua_State *L, gint idx)
 108 {
 109     if(luaL_getmetafield(L, idx, "__next")) {
 110         /* if idx is relative, reduce it since we got __next */
 111         if(idx < 0) idx--;
 112         /* copy table and then move key */
 113         lua_pushvalue(L, idx);
 114         lua_pushvalue(L, -3);
 115         lua_remove(L, -4);
 116         lua_pcall(L, 2, 2, 0);
 117         /* next returned nil, it's the end */
 118         if(lua_isnil(L, -1)) {
 119             /* remove nil */
 120             lua_pop(L, 2);
 121             return 0;
 122         }
 123         return 1;
 124     }
 125     else if(lua_istable(L, idx))
 126         return lua_next(L, idx);
 127     /* remove the key */
 128     lua_pop(L, 1);
 129     return 0;
 130 }
 131
 132 /* Generic pairs function.
 133  * Returns the number of elements pushed on stack. */
 134 static gint
 135 luaH_generic_pairs(lua_State *L)
 136 {
 137     lua_pushvalue(L, lua_upvalueindex(1));  /* return generator, */
 138     lua_pushvalue(L, 1);  /* state, */
 139     lua_pushnil(L);  /* and initial value */
 140     return 3;
 141 }
 142
 143 /* Overload standard pairs function to use __pairs field of metatables.
 144  * Returns the number of elements pushed on stack. */
 145 static gint
 146 luaHe_pairs(lua_State *L)
 147 {
 148     if(luaL_getmetafield(L, 1, "__pairs")) {
 149         lua_insert(L, 1);
 150         lua_call(L, lua_gettop(L) - 1, LUA_MULTRET);
 151         return lua_gettop(L);
 152     }
 153     luaL_checktype(L, 1, LUA_TTABLE);
 154     return luaH_generic_pairs(L);
 155 }
 156
 157 static gint
 158 luaH_ipairs_aux(lua_State *L)
 159 {
 160     gint i = luaL_checkint(L, 2) + 1;
 161     luaL_checktype(L, 1, LUA_TTABLE);
 162     lua_pushinteger(L, i);
 163     lua_rawgeti(L, 1, i);
 164     return (lua_isnil(L, -1)) ? 0 : 2;
 165 }
 166
 167 /* Overload standard ipairs function to use __ipairs field of metatables.
 168  * Returns the number of elements pushed on stack. */
 169 static gint
 170 luaHe_ipairs(lua_State *L)
 171 {
 172     if(luaL_getmetafield(L, 1, "__ipairs")) {
 173         lua_insert(L, 1);
 174         lua_call(L, lua_gettop(L) - 1, LUA_MULTRET);
 175         return lua_gettop(L);
 176     }
 177
 178     luaL_checktype(L, 1, LUA_TTABLE);
 179     lua_pushvalue(L, lua_upvalueindex(1));
 180     lua_pushvalue(L, 1);
 181     lua_pushinteger(L, 0);  /* and initial value */
 182     return 3;
 183 }
 184
 185 /* Enhanced type() function which recognize luakit objects.
 186  * \param L The Lua VM state.
 187  * \return The number of arguments pushed on the stack.
 188  */
 189 static gint
 190 luaHe_type(lua_State *L)
 191 {
 192     luaL_checkany(L, 1);
 193     lua_pushstring(L, luaH_typename(L, 1));
 194     return 1;
 195 }
 196
 197
 198 /* Fix up and add handy standard lib functions */
 199 static void
 200 luaH_fixups(lua_State *L)
 201 {
 202     /* export string.wlen */
 203     lua_getglobal(L, "string");
 204     lua_pushcfunction(L, &luaH_utf8_strlen);
 205     lua_setfield(L, -2, "wlen");
 206     lua_pop(L, 1);
 207     /* replace next */
 208     lua_pushliteral(L, "next");
 209     lua_pushcfunction(L, luaHe_next);
 210     lua_settable(L, LUA_GLOBALSINDEX);
 211     /* replace pairs */
 212     lua_pushliteral(L, "pairs");
 213     lua_pushcfunction(L, luaHe_next);
 214     lua_pushcclosure(L, luaHe_pairs, 1); /* pairs get next as upvalue */
 215     lua_settable(L, LUA_GLOBALSINDEX);
 216     /* replace ipairs */
 217     lua_pushliteral(L, "ipairs");
 218     lua_pushcfunction(L, luaH_ipairs_aux);
 219     lua_pushcclosure(L, luaHe_ipairs, 1);
 220     lua_settable(L, LUA_GLOBALSINDEX);
 221     /* replace type */
 222     lua_pushliteral(L, "type");
 223     lua_pushcfunction(L, luaHe_type);
 224     lua_settable(L, LUA_GLOBALSINDEX);
 225 }
 226
 227 /* Look for an item: table, function, etc.
 228  * \param L The Lua VM state.
 229  * \param item The pointer item.
 230  */
 231 gboolean
 232 luaH_hasitem(lua_State *L, gconstpointer item)
 233 {
 234     lua_pushnil(L);
 235     while(luaH_next(L, -2)) {
 236         if(lua_topointer(L, -1) == item) {
 237             /* remove value and key */
 238             lua_pop(L, 2);
 239             return TRUE;
 240         }
 241         if(lua_istable(L, -1))
 242             if(luaH_hasitem(L, item)) {
 243                 /* remove key and value */
 244                 lua_pop(L, 2);
 245                 return TRUE;
 246             }
 247         /* remove value */
 248         lua_pop(L, 1);
 249     }
 250     return FALSE;
 251 }
 252
 253 /* Browse a table pushed on top of the index, and put all its table and
 254  * sub-table ginto an array.
 255  * \param L The Lua VM state.
 256  * \param elems The elements array.
 257  * \return False if we encounter an elements already in list.
 258  */
 259 static gboolean
 260 luaH_isloop_check(lua_State *L, GPtrArray *elems)
 261 {
 262     if(lua_istable(L, -1)) {
 263         gconstpointer object = lua_topointer(L, -1);
 264
 265         /* Check that the object table is not already in the list */
 266         for(guint i = 0; i < elems->len; i++)
 267             if(elems->pdata[i] == object)
 268                 return FALSE;
 269
 270         /* push the table in the elements list */
 271         g_ptr_array_add(elems, (gpointer) object);
 272
 273         /* look every object in the "table" */
 274         lua_pushnil(L);
 275         while(luaH_next(L, -2)) {
 276             if(!luaH_isloop_check(L, elems)) {
 277                 /* remove key and value */
 278                 lua_pop(L, 2);
 279                 return FALSE;
 280             }
 281             /* remove value, keep key for next iteration */
 282             lua_pop(L, 1);
 283         }
 284     }
 285     return TRUE;
 286 }
 287
 288 /* Check if a table is a loop. When using tables as direct acyclic digram,
 289  * this is useful.
 290  * \param L The Lua VM state.
 291  * \param idx The index of the table in the stack
 292  * \return True if the table loops.
 293  */
 294 gboolean
 295 luaH_isloop(lua_State *L, gint idx)
 296 {
 297     /* elems is an elements array that we will fill with all array we
 298      * encounter while browsing the tables */
 299     GPtrArray *elems = g_ptr_array_new();
 300
 301     /* push table on top */
 302     lua_pushvalue(L, idx);
 303
 304     gboolean ret = luaH_isloop_check(L, elems);
 305
 306     /* remove pushed table */
 307     lua_pop(L, 1);
 308
 309     g_ptr_array_free(elems, TRUE);
 310
 311     return !ret;
 312 }
 313
 314 /* Returns a string from X selection.
 315  * \param L The Lua VM state.
 316  * \return The number of elements pushed on stack (1).
 317  */
 318 static gint
 319 luaH_luakit_selection(lua_State *L)
 320 {
 321     int n = lua_gettop(L);
 322     GdkAtom atom = GDK_SELECTION_PRIMARY;
 323
 324     if (n)
 325     {
 326       const gchar *arg = luaL_checklstring(L, 1, NULL);
 327
 328       /* Follow xclip(1) behavior: check only the first character of argument */
 329       switch (arg[0]) {
 330           case 'p':
 331               break;
 332           case 's':
 333               atom = GDK_SELECTION_SECONDARY;
 334               break;
 335           case 'c':
 336               atom = GDK_SELECTION_CLIPBOARD;
 337               break;
 338           default:
 339               luaL_argerror(L, 1, "should be 'primary', 'secondary' or 'clipboard'");
 340               break;
 341       }
 342     }
 343     GtkClipboard *selection = gtk_clipboard_get(atom);
 344     gchar *text = gtk_clipboard_wait_for_text(selection);
 345     lua_pushstring(L, text);
 346     g_free(text);
 347     return 1;
 348 }
 349
 350 /* luakit global table.
 351  * \param L The Lua VM state.
 352  * \return The number of elements pushed on stack.
 353  * \luastack
 354  * \lfield font The default font.
 355  * \lfield font_height The default font height.
 356  * \lfield conffile The configuration file which has been loaded.
 357  */
 358 static gint
 359 luaH_luakit_index(lua_State *L)
 360 {
 361     if(luaH_usemetatable(L, 1, 2))
 362         return 1;
 363
 364     size_t len;
 365     widget_t *w;
 366     const gchar *prop = luaL_checklstring(L, 2, &len);
 367     luakit_token_t token = l_tokenize(prop, len);
 368
 369     switch(token)
 370     {
 371       case L_TK_WINDOWS:
 372         lua_newtable(L);
 373         for (guint i = 0; i < globalconf.windows->len; i++) {
 374             w = globalconf.windows->pdata[i];
 375             luaH_object_push(L, w->ref);
 376             lua_rawseti(L, -2, i+1);
 377         }
 378         return 1;
 379
 380       case L_TK_SELECTION:
 381         lua_pushcfunction(L, luaH_luakit_selection);
 382         return 1;
 383
 384       default:
 385         break;
 386     }
 387     return 0;
 388 }
 389
 390 /* Newindex function for the luakit global table.
 391  * \param L The Lua VM state.
 392  * \return The number of elements pushed on stack.
 393  */
 394 static gint
 395 luaH_luakit_newindex(lua_State *L)
 396 {
 397     if(luaH_usemetatable(L, 1, 2))
 398         return 1;
 399
 400     size_t len;
 401     const gchar *buf = luaL_checklstring(L, 2, &len);
 402
 403     debug("Luakit newindex %s", buf);
 404
 405     return 0;
 406 }
 407
 408 /* Add a global signal.
 409  * Returns the number of elements pushed on stack.
 410  * \luastack
 411  * \lparam A string with the event name.
 412  * \lparam The function to call.
 413  */
 414 static gint
 415 luaH_luakit_add_signal(lua_State *L)
 416 {
 417     const gchar *name = luaL_checkstring(L, 1);
 418     luaH_checkfunction(L, 2);
 419     signal_add(globalconf.signals, name, luaH_object_ref(L, 2));
 420     return 0;
 421 }
 422
 423 /* Remove a global signal.
 424  * \param L The Lua VM state.
 425  * \return The number of elements pushed on stack.
 426  * \luastack
 427  * \lparam A string with the event name.
 428  * \lparam The function to call.
 429  */
 430 static gint
 431 luaH_luakit_remove_signal(lua_State *L)
 432 {
 433     const gchar *name = luaL_checkstring(L, 1);
 434     luaH_checkfunction(L, 2);
 435     gpointer func = (gpointer) lua_topointer(L, 2);
 436     signal_remove(globalconf.signals, name, func);
 437     luaH_object_unref(L, (void *) func);
 438     return 0;
 439 }
 440
 441 /* Emit a global signal.
 442  * \param L The Lua VM state.
 443  * \return The number of elements pushed on stack.
 444  * \luastack
 445  * \lparam A string with the event name.
 446  * \lparam The function to call.
 447  */
 448 static gint
 449 luaH_luakit_emit_signal(lua_State *L)
 450 {
 451     signal_object_emit(L, globalconf.signals, luaL_checkstring(L, 1), lua_gettop(L) - 1);
 452     return 0;
 453 }
 454
 455 static gint
 456 luaH_panic(lua_State *L)
 457 {
 458     warn("unprotected error in call to Lua API (%s)", lua_tostring(L, -1));
 459     return 0;
 460 }
 461
 462 static gint
 463 luaH_quit(lua_State *L)
 464 {
 465     (void) L;
 466     gtk_main_quit();
 467     return 0;
 468 }
 469
 470 static gint
 471 luaH_dofunction_on_error(lua_State *L)
 472 {
 473     /* duplicate string error */
 474     lua_pushvalue(L, -1);
 475     /* emit error signal */
 476     signal_object_emit(L, globalconf.signals, "debug::error", 1);
 477
 478     if(!luaL_dostring(L, "return debug.traceback(\"error while running function\", 3)"))
 479     {
 480         /* Move traceback before error */
 481         lua_insert(L, -2);
 482         /* Insert sentence */
 483         lua_pushliteral(L, "\nerror: ");
 484         /* Move it before error */
 485         lua_insert(L, -2);
 486         lua_concat(L, 3);
 487     }
 488     return 1;
 489 }
 490
 491 void
 492 luaH_init(xdgHandle *xdg)
 493 {
 494     lua_State *L;
 495
 496     static const struct luaL_reg luakit_lib[] = {
 497         { "quit", luaH_quit },
 498         { "add_signal", luaH_luakit_add_signal },
 499         { "remove_signal", luaH_luakit_remove_signal },
 500         { "emit_signal", luaH_luakit_emit_signal },
 501         { "__index", luaH_luakit_index },
 502         { "__newindex", luaH_luakit_newindex },
 503         { NULL, NULL }
 504     };
 505
 506     /* Lua VM init */
 507     L = globalconf.L = luaL_newstate();
 508
 509     /* Set panic fuction */
 510     lua_atpanic(L, luaH_panic);
 511
 512     /* Set error handling function */
 513     lualib_dofunction_on_error = luaH_dofunction_on_error;
 514
 515     luaL_openlibs(L);
 516
 517     luaH_fixups(L);
 518
 519     luaH_object_setup(L);
 520
 521     /* Export luakit lib */
 522     luaH_openlib(L, "luakit", luakit_lib, luakit_lib);
 523
 524     /* Export widget */
 525     widget_class_setup(L);
 526
 527     /* add Lua search paths */
 528     lua_getglobal(L, "package");
 529     if(LUA_TTABLE != lua_type(L, 1)) {
 530         warn("package is not a table");
 531         return;
 532     }
 533     lua_getfield(L, 1, "path");
 534     if(LUA_TSTRING != lua_type(L, 2)) {
 535         warn("package.path is not a string");
 536         lua_pop(L, 1);
 537         return;
 538     }
 539
 540     /* allows for testing luakit in the project directory
 541      * (i.e. `./luakit -c rc.lua ...` */
 542     lua_pushliteral(L, ";./lib/?.lua;");
 543     lua_pushliteral(L, ";./lib/?/init.lua");
 544     lua_concat(L, 2);
 545
 546     /* add XDG_CONFIG_DIR as an include path */
 547     const gchar * const *xdgconfigdirs = xdgSearchableConfigDirectories(xdg);
 548     for(; *xdgconfigdirs; xdgconfigdirs++)
 549     {
 550         size_t len = l_strlen(*xdgconfigdirs);
 551         lua_pushliteral(L, ";");
 552         lua_pushlstring(L, *xdgconfigdirs, len);
 553         lua_pushliteral(L, "/luakit/?.lua");
 554         lua_concat(L, 3);
 555
 556         lua_pushliteral(L, ";");
 557         lua_pushlstring(L, *xdgconfigdirs, len);
 558         lua_pushliteral(L, "/luakit/?/init.lua");
 559         lua_concat(L, 3);
 560
 561         lua_concat(L, 3); /* concatenate with package.path */
 562     }
 563
 564     /* add Lua lib path (/usr/share/luakit/lib by default) */
 565     lua_pushliteral(L, ";" LUAKIT_LUA_LIB_PATH "/?.lua");
 566     lua_pushliteral(L, ";" LUAKIT_LUA_LIB_PATH "/?/init.lua");
 567     lua_concat(L, 3); /* concatenate with package.path */
 568     lua_setfield(L, 1, "path"); /* package.path = "concatenated string" */
 569 }
 570
 571 gboolean
 572 luaH_loadrc(const gchar *confpath, gboolean run)
 573 {
 574     lua_State *L = globalconf.L;
 575
 576     if(!luaL_loadfile(L, confpath)) {
 577         if(run) {
 578             if(lua_pcall(L, 0, LUA_MULTRET, 0)) {
 579                 g_fprintf(stderr, "%s\n", lua_tostring(L, -1));
 580             } else
 581                 return TRUE;
 582         } else
 583             lua_pop(L, 1);
 584         return TRUE;
 585     } else
 586         g_fprintf(stderr, "%s\n", lua_tostring(L, -1));
 587     return FALSE;
 588 }
 589
 590 /* Load a configuration file.
 591  *
 592  * param xdg An xdg handle to use to get XDG basedir.
 593  * param confpatharg The configuration file to load.
 594  * param run Run the configuration file.
 595  */
 596 gboolean
 597 luaH_parserc(xdgHandle* xdg, const gchar *confpatharg, gboolean run)
 598 {
 599     gchar *confpath = NULL;
 600     gboolean ret = FALSE;
 601
 602     /* try to load, return if it's ok */
 603     if(confpatharg) {
 604         debug("Attempting to load rc file: %s", confpatharg);
 605         if(luaH_loadrc(confpatharg, run))
 606             ret = TRUE;
 607         goto bailout;
 608     }
 609     confpath = xdgConfigFind("luakit/rc.lua", xdg);
 610     gchar *tmp = confpath;
 611
 612     /* confpath is "string1\0string2\0string3\0\0" */
 613     while(*tmp) {
 614         debug("Loading rc file: %s", tmp);
 615         if(luaH_loadrc(tmp, run)) {
 616             globalconf.confpath = g_strdup(tmp);
 617             ret = TRUE;
 618             goto bailout;
 619         } else if(!run)
 620             goto bailout;
 621         tmp += l_strlen(tmp) + 1;
 622     }
 623
 624 bailout:
 625
 626     if (confpath) g_free(confpath);
 627     return ret;
 628 }
 629
 630 gint
 631 luaH_class_index_miss_property(lua_State *L, lua_object_t *obj)
 632 {
 633     (void) obj;
 634     signal_object_emit(L, globalconf.signals, "debug::index::miss", 2);
 635     return 0;
 636 }
 637
 638 gint
 639 luaH_class_newindex_miss_property(lua_State *L, lua_object_t *obj)
 640 {
 641     (void) obj;
 642     signal_object_emit(L, globalconf.signals, "debug::newindex::miss", 3);
 643     return 0;
 644 }
 645
 646 // vim: ft=c:et:sw=4:ts=8:sts=4:tw=80