awesome/lua.c

659 lines
17 KiB
C
Raw Normal View History

/*
* lua.c - Lua configuration management
*
* Copyright © 2008 Julien Danjou <julien@danjou.info>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
*/
#include <errno.h>
#include <stdio.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <unistd.h>
#include <ev.h>
#include <lua.h>
#include <lauxlib.h>
#include <lualib.h>
#include <xcb/xcb.h>
#include <xcb/xcb_aux.h>
#include "ewmh.h"
#include "config.h"
#include "structs.h"
#include "lua.h"
#include "tag.h"
#include "client.h"
#include "window.h"
#include "statusbar.h"
#include "titlebar.h"
#include "screen.h"
#include "layouts/tile.h"
#include "common/socket.h"
extern awesome_t globalconf;
extern const name_func_link_t FloatingPlacementList[];
extern const struct luaL_reg awesome_keygrabber_lib[];
extern const struct luaL_reg awesome_mouse_methods[];
extern const struct luaL_reg awesome_mouse_meta[];
extern const struct luaL_reg awesome_client_methods[];
extern const struct luaL_reg awesome_client_meta[];
extern const struct luaL_reg awesome_titlebar_methods[];
extern const struct luaL_reg awesome_titlebar_meta[];
extern const struct luaL_reg awesome_tag_methods[];
extern const struct luaL_reg awesome_tag_meta[];
extern const struct luaL_reg awesome_widget_methods[];
extern const struct luaL_reg awesome_widget_meta[];
extern const struct luaL_reg awesome_statusbar_methods[];
extern const struct luaL_reg awesome_statusbar_meta[];
extern const struct luaL_reg awesome_keybinding_methods[];
extern const struct luaL_reg awesome_keybinding_meta[];
static struct sockaddr_un *addr;
static ev_io csio = { .fd = -1 };
/** Add a global mouse binding. This binding will be available when you'll
* click on root window.
* \param L The Lua VM state.
*
* \luastack
* \lparam A mouse button binding.
*/
static int
luaA_mouse_add(lua_State *L)
{
button_t **button = luaA_checkudata(L, 1, "mouse");
button_list_push(&globalconf.buttons.root, *button);
button_ref(button);
return 0;
}
/** Set the floating placement algorithm. This will be used to compute the
* initial floating position of floating windows.
* \param L The Lua VM state.
*
* \luastack
* \lparam An algorith name, either `none', `smart' or `mouse'.
*/
static int
luaA_floating_placement_set(lua_State *L)
{
const char *pl = luaL_checkstring(L, 1);
globalconf.floating_placement = name_func_lookup(pl, FloatingPlacementList);
return 0;
}
/** Quit awesome.
*/
static int
luaA_quit(lua_State *L __attribute__ ((unused)))
{
ev_unloop(globalconf.loop, 1);
return 0;
}
/** Execute another application, probably a window manager, to replace
* awesome.
* \param L The Lua VM state.
*
* \luastack
* \lparam The command line to execute.
*/
static int
luaA_exec(lua_State *L)
{
client_t *c;
const char *cmd = luaL_checkstring(L, 1);
for(c = globalconf.clients; c; c = c->next)
client_unban(c);
xcb_aux_sync(globalconf.connection);
xcb_disconnect(globalconf.connection);
a_exec(cmd);
return 0;
}
/** Restart awesome.
*/
static int
luaA_restart(lua_State *L __attribute__ ((unused)))
{
a_exec(globalconf.argv);
return 0;
}
/** Set the screen padding. This can be used to define margin around the
* screen. awesome will not use this area.
* \param L The Lua VM state.
*
* \luastack
* \lparam A screen number.
* \lparam A table with a list of margin for `right', `left', `top' and
* `bottom'.
*/
static int
luaA_padding_set(lua_State *L)
{
int screen = luaL_checknumber(L, 1) - 1;
if(screen >= 0 && screen < globalconf.screens_info->nscreen)
{
luaA_checktable(L, 2);
globalconf.screens[screen].padding.right = luaA_getopt_number(L, 2, "right", 0);
globalconf.screens[screen].padding.left = luaA_getopt_number(L, 2, "left", 0);
globalconf.screens[screen].padding.top = luaA_getopt_number(L, 2, "top", 0);
globalconf.screens[screen].padding.bottom = luaA_getopt_number(L, 2, "bottom", 0);
}
ewmh_update_workarea(screen_virttophys(screen));
return 0;
}
/** Define if awesome should respect applications size hints when resizing
* windows in tiled mode. If you set this to true, you will experience gaps
* between windows, but they will have the best size they can have.
* \param L The Lua VM state.
*
* \luastack
* \lparam A boolean value, true to enable, false to disable.
*/
static int
luaA_resizehints_set(lua_State *L)
{
globalconf.resize_hints = luaA_checkboolean(L, 1);
return 0;
}
/** Get the screen count.
* \param L The Lua VM state.
*
* \luastack
* \lreturn The screen count, at least 1.
*/
static int
luaA_screen_count(lua_State *L)
{
lua_pushnumber(L, globalconf.screens_info->nscreen);
return 1;
}
/** Give the focus to a screen.
* \param L The Lua VM state.
*
* \luastack
* \lparam A screen number
*/
static int
luaA_screen_focus(lua_State *L)
{
/* Our table begin at 0, Lua begins at 1 */
int screen = luaL_checknumber(L, 1) - 1;
luaA_checkscreen(screen);
client_focus(NULL, screen);
return 0;
}
static int
luaA_screen_coords_get(lua_State *L)
{
int screen = luaL_checknumber(L, 1) - 1;
luaA_checkscreen(screen);
lua_newtable(L);
lua_pushnumber(L, globalconf.screens_info->geometry[screen].x);
lua_setfield(L, -2, "x");
lua_pushnumber(L, globalconf.screens_info->geometry[screen].y);
lua_setfield(L, -2, "y");
lua_pushnumber(L, globalconf.screens_info->geometry[screen].width);
lua_setfield(L, -2, "width");
lua_pushnumber(L, globalconf.screens_info->geometry[screen].height);
lua_setfield(L, -2, "height");
return 1;
}
/** Set the function called each time a client gets focus. This function is
* called with the client object as argument.
* \param L The Lua VM state.
*
* \luastack
* \lparam A function to call each time a client gets focus.
*/
static int
luaA_hooks_focus(lua_State *L)
{
luaA_checkfunction(L, 1);
if(globalconf.hooks.focus)
luaL_unref(L, LUA_REGISTRYINDEX, globalconf.hooks.focus);
globalconf.hooks.focus = luaL_ref(L, LUA_REGISTRYINDEX);
return 0;
}
/** Set the function called each time a client loses focus. This function is
* called with the client object as argument.
* \param L The Lua VM state.
*
* \luastack
* \lparam A function to call each time a client loses focus.
*/
static int
luaA_hooks_unfocus(lua_State *L)
{
luaA_checkfunction(L, 1);
if(globalconf.hooks.unfocus)
luaL_unref(L, LUA_REGISTRYINDEX, globalconf.hooks.unfocus);
globalconf.hooks.unfocus = luaL_ref(L, LUA_REGISTRYINDEX);
return 0;
}
/** Set the function called each time a new client appears. This function is
* called with the client object as argument.
* \param L The Lua VM state.
*
* \luastack
* \lparam A function to call on each new client.
*/
static int
luaA_hooks_manage(lua_State *L)
{
luaA_checkfunction(L, 1);
if(globalconf.hooks.manage)
luaL_unref(L, LUA_REGISTRYINDEX, globalconf.hooks.manage);
globalconf.hooks.manage = luaL_ref(L, LUA_REGISTRYINDEX);
return 0;
}
/** Set the function called each time a client goes away. This function is
* called with the client object as argument.
* \param L The Lua VM state.
*
* \luastack
* \lparam A function to call when a client goes away.
*/
static int
luaA_hooks_unmanage(lua_State *L)
{
luaA_checkfunction(L, 1);
if(globalconf.hooks.unmanage)
luaL_unref(L, LUA_REGISTRYINDEX, globalconf.hooks.unmanage);
globalconf.hooks.unmanage = luaL_ref(L, LUA_REGISTRYINDEX);
return 0;
}
/** Set the function called each time the mouse enter a new window. This
* function is called with the client object as argument.
* \param L The Lua VM state.
*
* \luastack
* \lparam A function to call each time a client gets mouse over it.
*/
static int
luaA_hooks_mouseover(lua_State *L)
{
luaA_checkfunction(L, 1);
if(globalconf.hooks.mouseover)
luaL_unref(L, LUA_REGISTRYINDEX, globalconf.hooks.mouseover);
globalconf.hooks.mouseover = luaL_ref(L, LUA_REGISTRYINDEX);
return 0;
}
/** Set the function called on each screen arrange. This function is called
* with the screen number as argument.
* \param L The Lua VM state.
*
* \luastack
* \lparam A function to call on each screen arrange.
*/
static int
luaA_hooks_arrange(lua_State *L)
{
luaA_checkfunction(L, 1);
if(globalconf.hooks.arrange)
luaL_unref(L, LUA_REGISTRYINDEX, globalconf.hooks.arrange);
globalconf.hooks.arrange = luaL_ref(L, LUA_REGISTRYINDEX);
return 0;
}
/** Set the function called on each title update. This function is called with
* the client object as argument.
* \param L The Lua VM state.
*
* \luastack
* \lparam A function to call on each title update of each client.
*/
static int
luaA_hooks_titleupdate(lua_State *L)
{
luaA_checkfunction(L, 1);
if(globalconf.hooks.titleupdate)
luaL_unref(L, LUA_REGISTRYINDEX, globalconf.hooks.titleupdate);
globalconf.hooks.titleupdate = luaL_ref(L, LUA_REGISTRYINDEX);
return 0;
}
/** Set the function called when a client get urgency flag. This function is called with
* the client object as argument.
* \param L The Lua VM state.
*
* \luastack
* \lparam A function to call when a client get the urgent flag.
*/
static int
luaA_hooks_urgent(lua_State *L)
{
luaA_checkfunction(L, 1);
if(globalconf.hooks.urgent)
luaL_unref(L, LUA_REGISTRYINDEX, globalconf.hooks.urgent);
globalconf.hooks.urgent = luaL_ref(L, LUA_REGISTRYINDEX);
return 0;
}
/** Set the function to be called every N seconds.
* \param L The Lua VM state.
*
* \luastack
* \lparam The number of seconds to run function every. Set 0 to disable.
* \lparam A function to call every N seconds (optional).
*/
static int
luaA_hooks_timer(lua_State *L)
{
globalconf.timer.repeat = luaL_checknumber(L, 1);
if(lua_gettop(L) == 2 && !lua_isnil(L, 2))
{
luaA_checkfunction(L, 2);
if(globalconf.hooks.timer)
luaL_unref(L, LUA_REGISTRYINDEX, globalconf.hooks.timer);
globalconf.hooks.timer = luaL_ref(L, LUA_REGISTRYINDEX);
}
ev_timer_again(globalconf.loop, &globalconf.timer);
return 0;
}
/** Set default font.
* \param L The Lua VM state.
*
* \luastack
* \lparam A string with a font name in Pango format.
*/
static int
luaA_font_set(lua_State *L)
{
const char *font = luaL_checkstring(L, 1);
draw_font_delete(&globalconf.font);
globalconf.font = draw_font_new(globalconf.connection,
globalconf.default_screen, font);
return 0;
}
/** Set default colors.
* \param L The Lua VM state.
*
* \luastack
* \lparam A table with `fg' and `bg' elements, containing colors.
*/
static int
luaA_colors_set(lua_State *L)
{
const char *fg, *bg;
luaA_checktable(L, 1);
if((fg = luaA_getopt_string(L, 1, "fg", NULL)))
xcolor_new(globalconf.connection, globalconf.default_screen,
fg, &globalconf.colors.fg);
if((bg = luaA_getopt_string(L, 1, "bg",NULL)))
xcolor_new(globalconf.connection, globalconf.default_screen,
bg, &globalconf.colors.bg);
return 0;
}
/** Push a pointer onto the stack according to its type.
* \param p The pointer.
* \param type Its type.
*/
void
luaA_pushpointer(void *p, awesome_type_t type)
{
switch(type)
{
case AWESOME_TYPE_STATUSBAR:
luaA_statusbar_userdata_new(p);
break;
case AWESOME_TYPE_TITLEBAR:
luaA_titlebar_userdata_new(p);
break;
}
}
static void
luaA_openlib(lua_State *L, const char *name,
const struct luaL_reg methods[],
const struct luaL_reg meta[])
{
luaL_newmetatable(L, name);
lua_pushliteral(L, "__index");
lua_pushvalue(L, -2); /* dup metatable*/
lua_settable(L, -3); /* metatable.__index = metatable */
luaL_register(L, NULL, meta);
luaL_register(L, name, methods);
}
/** Initialize the Lua VM
*/
void
luaA_init(void)
{
lua_State *L;
static const struct luaL_reg awesome_lib[] =
{
{ "quit", luaA_quit },
{ "exec", luaA_exec },
{ "restart", luaA_restart },
{ "floating_placement_set", luaA_floating_placement_set },
{ "padding_set", luaA_padding_set },
{ "mouse_add", luaA_mouse_add },
{ "resizehints_set", luaA_resizehints_set },
{ "font_set", luaA_font_set },
{ "colors_set", luaA_colors_set },
{ NULL, NULL }
};
static const struct luaL_reg awesome_screen_lib[] =
{
{ "coords_get", luaA_screen_coords_get },
{ "count", luaA_screen_count },
{ "focus", luaA_screen_focus },
{ NULL, NULL }
};
static const struct luaL_reg awesome_hooks_lib[] =
{
{ "focus", luaA_hooks_focus },
{ "unfocus", luaA_hooks_unfocus },
{ "manage", luaA_hooks_manage },
{ "unmanage", luaA_hooks_unmanage },
{ "mouseover", luaA_hooks_mouseover },
{ "arrange", luaA_hooks_arrange },
{ "titleupdate", luaA_hooks_titleupdate },
{ "urgent", luaA_hooks_urgent },
{ "timer", luaA_hooks_timer },
{ NULL, NULL }
};
L = globalconf.L = lua_open();
luaL_openlibs(L);
/* Export awesome lib */
luaL_register(L, "awesome", awesome_lib);
/* Export screen lib */
luaL_register(L, "screen", awesome_screen_lib);
/* Export hooks lib */
luaL_register(L, "hooks", awesome_hooks_lib);
/* Export keygrabber lib */
luaL_register(L, "keygrabber", awesome_keygrabber_lib);
/* Export mouse */
luaA_openlib(L, "mouse", awesome_mouse_methods, awesome_mouse_meta);
/* Export tag */
luaA_openlib(L, "tag", awesome_tag_methods, awesome_tag_meta);
/* Export statusbar */
luaA_openlib(L, "statusbar", awesome_statusbar_methods, awesome_statusbar_meta);
/* Export widget */
luaA_openlib(L, "widget", awesome_widget_methods, awesome_widget_meta);
/* Export client */
luaA_openlib(L, "client", awesome_client_methods, awesome_client_meta);
/* Export titlebar */
luaA_openlib(L, "titlebar", awesome_titlebar_methods, awesome_titlebar_meta);
/* Export keys */
luaA_openlib(L, "keybinding", awesome_keybinding_methods, awesome_keybinding_meta);
lua_pushliteral(L, "AWESOME_VERSION");
lua_pushliteral(L, VERSION);
lua_settable(L, LUA_GLOBALSINDEX);
luaA_dostring(L, "package.path = package.path .. \";" AWESOME_LUA_LIB_PATH "/?.lua\"");
}
/** Load a configuration file
*
* \param rcfile The configuration file to load.
* \return True on succes, false on failure.
*/
bool
luaA_parserc(const char* rcfile)
{
int screen;
if(luaL_dofile(globalconf.L, rcfile))
{
fprintf(stderr, "%s\n", lua_tostring(globalconf.L, -1));
return false;
}
/* Assure there's at least one tag */
for(screen = 0; screen < globalconf.screens_info->nscreen; screen++)
if(!globalconf.screens[screen].tags)
tag_append_to_screen(tag_new("default", layout_tile, 0.5, 1, 0), screen);
return true;
}
/** Parse a command.
* \param cmd The buffer to parse.
* \return true on succes, false on failure.
*/
static void
luaA_docmd(char *cmd)
{
char *p, *curcmd = cmd;
if(a_strlen(cmd))
while((p = strchr(curcmd, '\n')))
{
*p = '\0';
luaA_dostring(globalconf.L, curcmd);
curcmd = p + 1;
}
}
static void
luaA_cb(EV_P_ ev_io *w, int revents)
{
char buf[1024];
int r;
switch(r = recv(w->fd, buf, sizeof(buf)-1, MSG_TRUNC))
{
case -1:
warn("error reading UNIX domain socket: %s", strerror(errno));
luaA_cs_cleanup();
break;
case 0:
break;
default:
if(r >= ssizeof(buf))
break;
buf[r] = '\0';
luaA_docmd(buf);
}
}
void
luaA_cs_init(void)
{
int csfd = socket_getclient();
if (csfd < 0)
return;
addr = socket_getaddr(getenv("DISPLAY"));
if(bind(csfd, (const struct sockaddr *) addr, SUN_LEN(addr)))
{
if(errno == EADDRINUSE)
{
if(unlink(addr->sun_path))
warn("error unlinking existing file: %s", strerror(errno));
if(bind(csfd, (const struct sockaddr *) addr, SUN_LEN(addr)))
warn("error binding UNIX domain socket: %s", strerror(errno));
}
else
warn("error binding UNIX domain socket: %s", strerror(errno));
}
ev_io_init(&csio, &luaA_cb, csfd, EV_READ);
ev_io_start(EV_DEFAULT_UC_ &csio);
ev_unref(EV_DEFAULT_UC);
}
void
luaA_cs_cleanup(void)
{
if(csio.fd < 0)
return;
ev_ref(EV_DEFAULT_UC);
ev_io_stop(EV_DEFAULT_UC_ &csio);
if (close(csio.fd))
warn("error closing UNIX domain socket: %s", strerror(errno));
if(unlink(addr->sun_path))
warn("error unlinking UNIX domain socket: %s", strerror(errno));
p_delete(&addr);
csio.fd = -1;
}
void
luaA_on_timer(EV_P_ ev_timer *w, int revents)
{
luaA_dofunction(globalconf.L, globalconf.hooks.timer, 0);
}