From f57d28d89475fc972e717e98e4512a54702a2230 Mon Sep 17 00:00:00 2001
From: Julien Danjou <julien@danjou.info>
Date: Thu, 18 Sep 2008 15:11:18 +0200
Subject: [PATCH] client: client.get() supports screen as argument

Signed-off-by: Julien Danjou <julien@danjou.info>
---
 client.c | 22 ++++++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/client.c b/client.c
index a19df7c53..cb4e97497 100644
--- a/client.c
+++ b/client.c
@@ -878,20 +878,34 @@ client_kill(client_t *c)
  * \param L The Lua VM state.
  * \return The number of elements pushed on stack.
  * \luastack
+ * \lparam An optional screen nunmber.
  * \lreturn A table with all clients.
  */
 static int
 luaA_client_get(lua_State *L)
 {
-    int i = 1;
+    int i = 1, screen;
     client_t *c;
 
+    screen = luaL_optnumber(L, 1, 0) - 1;
+
     lua_newtable(L);
 
-    for(c = globalconf.clients; c; c = c->next)
+    if(screen == SCREEN_UNDEF)
+        for(c = globalconf.clients; c; c = c->next)
+        {
+            luaA_client_userdata_new(globalconf.L, c);
+            lua_rawseti(L, -2, i++);
+        }
+    else
     {
-        luaA_client_userdata_new(globalconf.L, c);
-        lua_rawseti(L, -2, i++);
+        luaA_checkscreen(screen);
+        for(c = globalconf.clients; c; c = c->next)
+            if(c->screen == screen)
+            {
+                luaA_client_userdata_new(globalconf.L, c);
+                lua_rawseti(L, -2, i++);
+            }
     }
 
     return 1;