screen: add luaA_pushscreen

Signed-off-by: Julien Danjou <julien@danjou.info>

screen.c

@@ -338,6 +338,20 @@ screen_client_moveto(client_t *c, screen_t *new_screen, bool dotag, bool doresiz
     hook_property(client, c, "screen");
 }
 
+/** Push a screen onto the stack.
+ * \param L The Lua VM state.
+ * \param s The scren to push.
+ * \return The number of elements pushed on stack.
+ */
+int
+luaA_pushscreen(lua_State *L, screen_t *s)
+{
+    lua_pushlightuserdata(L, s);
+    luaL_getmetatable(L, "screen");
+    lua_setmetatable(L, -2);
+    return 1;
+}
+
 /** Screen module.
  * \param L The Lua VM state.
  * \return The number of elements pushed on stack.
@@ -348,10 +362,8 @@ static int
 luaA_screen_module_index(lua_State *L)
 {
     int screen = luaL_checknumber(L, 2) - 1;
-
     luaA_checkscreen(screen);
-    lua_pushlightuserdata(L, &globalconf.screens.tab[screen]);
-    return luaA_settype(L, "screen");
+    return luaA_pushscreen(L, &globalconf.screens.tab[screen]);
 }
 
 /** Get or set screen tags.

screen.h

@@ -56,6 +56,7 @@ area_t screen_area_get(screen_t *, bool);
 area_t display_area_get(int);
 int screen_virttophys(int);
 void screen_client_moveto(client_t *, screen_t *, bool, bool);
+int luaA_pushscreen(lua_State *, screen_t *);
 
 #endif
 // vim: filetype=c:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:encoding=utf-8:textwidth=80