degesch: expose channels and users to Lua

1 files changed, 226 insertions, 5 deletions

diff --git a/degesch.c b/degesch.c
index c21dba7..f1eceb8 100644
--- a/degesch.c
+++ b/degesch.c
@@ -8322,6 +8322,183 @@ lua_plugin_parse (lua_State *L)
 
 // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
 
+#define XLUA_USER_METATABLE "user"      ///< Identifier for the Lua metatable
+
+struct lua_user
+{
+	struct lua_plugin *plugin;          ///< The plugin we belong to
+	struct user *user;                  ///< The user
+	struct weak_ref_link *weak_ref;     ///< A weak reference link
+};
+
+static void
+lua_user_invalidate (void *object, void *user_data)
+{
+	struct lua_user *wrapper = user_data;
+	wrapper->user = NULL;
+	wrapper->weak_ref = NULL;
+	// This can in theory call the GC, order isn't arbitrary here
+	lua_cache_invalidate (wrapper->plugin->L, object);
+}
+
+static void
+lua_plugin_push_user (struct lua_plugin *plugin, struct user *user)
+{
+	lua_State *L = plugin->L;
+	if (lua_cache_get (L, user))
+		return;
+
+	struct lua_user *wrapper = lua_newuserdata (L, sizeof *wrapper);
+	luaL_setmetatable (L, XLUA_USER_METATABLE);
+	wrapper->plugin = plugin;
+	wrapper->user = user;
+	wrapper->weak_ref = user_weak_ref (user, lua_user_invalidate, wrapper);
+	lua_cache_store (L, user, -1);
+}
+
+// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+
+/// Identifier for the Lua metatable
+#define XLUA_CHANNEL_METATABLE "channel"
+
+struct lua_channel
+{
+	struct lua_plugin *plugin;          ///< The plugin we belong to
+	struct channel *channel;            ///< The channel
+	struct weak_ref_link *weak_ref;     ///< A weak reference link
+};
+
+static void
+lua_channel_invalidate (void *object, void *user_data)
+{
+	struct lua_channel *wrapper = user_data;
+	wrapper->channel = NULL;
+	wrapper->weak_ref = NULL;
+	// This can in theory call the GC, order isn't arbitrary here
+	lua_cache_invalidate (wrapper->plugin->L, object);
+}
+
+static void
+lua_plugin_push_channel (struct lua_plugin *plugin, struct channel *channel)
+{
+	lua_State *L = plugin->L;
+	if (lua_cache_get (L, channel))
+		return;
+
+	struct lua_channel *wrapper = lua_newuserdata (L, sizeof *wrapper);
+	luaL_setmetatable (L, XLUA_CHANNEL_METATABLE);
+	wrapper->plugin = plugin;
+	wrapper->channel = channel;
+	wrapper->weak_ref = channel_weak_ref
+		(channel, lua_channel_invalidate, wrapper);
+	lua_cache_store (L, channel, -1);
+}
+
+// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+
+static int
+lua_user_gc (lua_State *L)
+{
+	struct lua_user *wrapper = luaL_checkudata (L, 1, XLUA_USER_METATABLE);
+	if (wrapper->user)
+	{
+		lua_cache_invalidate (L, wrapper->user);
+		user_weak_unref (wrapper->user, &wrapper->weak_ref);
+		wrapper->user = NULL;
+	}
+	return 0;
+}
+
+static int
+lua_user_get_nickname (lua_State *L)
+{
+	struct lua_user *wrapper = luaL_checkudata (L, 1, XLUA_USER_METATABLE);
+	luaL_argcheck (L, wrapper->user, 1, "dead reference used");
+	lua_pushstring (L, wrapper->user->nickname);
+	return 1;
+}
+
+static int
+lua_user_get_channels (lua_State *L)
+{
+	struct lua_user *wrapper = luaL_checkudata (L, 1, XLUA_USER_METATABLE);
+	luaL_argcheck (L, wrapper->user, 1, "dead reference used");
+
+	int i = 1;
+	lua_newtable (L);
+	LIST_FOR_EACH (struct user_channel, iter, wrapper->user->channels)
+	{
+		lua_plugin_push_channel (wrapper->plugin, iter->channel);
+		lua_rawseti (L, -2, i++);
+	}
+	return 1;
+}
+
+static luaL_Reg lua_user_table[] =
+{
+	{ "__gc",         lua_user_gc           },
+	{ "get_nickname", lua_user_get_nickname },
+	{ "get_channels", lua_user_get_channels },
+	{ NULL,           NULL                  }
+};
+
+// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+
+static int
+lua_channel_gc (lua_State *L)
+{
+	struct lua_channel *wrapper =
+		luaL_checkudata (L, 1, XLUA_CHANNEL_METATABLE);
+	if (wrapper->channel)
+	{
+		lua_cache_invalidate (L, wrapper->channel);
+		channel_weak_unref (wrapper->channel, &wrapper->weak_ref);
+		wrapper->channel = NULL;
+	}
+	return 0;
+}
+
+static int
+lua_channel_get_name (lua_State *L)
+{
+	struct lua_channel *wrapper =
+		luaL_checkudata (L, 1, XLUA_CHANNEL_METATABLE);
+	luaL_argcheck (L, wrapper->channel, 1, "dead reference used");
+	lua_pushstring (L, wrapper->channel->name);
+	return 1;
+}
+
+static int
+lua_channel_get_users (lua_State *L)
+{
+	struct lua_channel *wrapper =
+		luaL_checkudata (L, 1, XLUA_CHANNEL_METATABLE);
+	luaL_argcheck (L, wrapper->channel, 1, "dead reference used");
+
+	int i = 1;
+	lua_newtable (L);
+	LIST_FOR_EACH (struct channel_user, iter, wrapper->channel->users)
+	{
+		lua_createtable (L, 0, 2);
+		lua_plugin_push_user (wrapper->plugin, iter->user);
+		lua_setfield (L, -2, "user");
+		lua_plugin_kv (L, "prefixes", iter->prefixes.str);
+
+		lua_rawseti (L, -2, i++);
+	}
+	return 1;
+}
+
+static luaL_Reg lua_channel_table[] =
+{
+	{ "__gc",      lua_channel_gc        },
+	{ "get_name",  lua_channel_get_name  },
+	{ "get_users", lua_channel_get_users },
+	{ NULL,        NULL                  }
+};
+
+// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+
 #define XLUA_BUFFER_METATABLE "buffer"  ///< Identifier for the Lua metatable
 
 struct lua_buffer
@@ -8410,6 +8587,32 @@ lua_buffer_gc (lua_State *L)
 }
 
 static int
+lua_buffer_get_user (lua_State *L)
+{
+	struct lua_buffer *wrapper = luaL_checkudata (L, 1, XLUA_BUFFER_METATABLE);
+	luaL_argcheck (L, wrapper->buffer, 1, "dead reference used");
+
+	if (wrapper->buffer->user)
+		lua_plugin_push_user (wrapper->plugin, wrapper->buffer->user);
+	else
+		lua_pushnil (L);
+	return 1;
+}
+
+static int
+lua_buffer_get_channel (lua_State *L)
+{
+	struct lua_buffer *wrapper = luaL_checkudata (L, 1, XLUA_BUFFER_METATABLE);
+	luaL_argcheck (L, wrapper->buffer, 1, "dead reference used");
+
+	if (wrapper->buffer->channel)
+		lua_plugin_push_channel (wrapper->plugin, wrapper->buffer->channel);
+	else
+		lua_pushnil (L);
+	return 1;
+}
+
+static int
 lua_buffer_get_server (lua_State *L)
 {
 	struct lua_buffer *wrapper = luaL_checkudata (L, 1, XLUA_BUFFER_METATABLE);
@@ -8450,11 +8653,13 @@ lua_buffer_execute (lua_State *L)
 static luaL_Reg lua_buffer_table[] =
 {
 	// TODO: some useful methods or values
-	{ "__gc",       lua_buffer_gc         },
-	{ "get_server", lua_buffer_get_server },
-	{ "log",        lua_buffer_log        },
-	{ "execute",    lua_buffer_execute    },
-	{ NULL,         NULL                  }
+	{ "__gc",        lua_buffer_gc          },
+	{ "get_user",    lua_buffer_get_user    },
+	{ "get_channel", lua_buffer_get_channel },
+	{ "get_server",  lua_buffer_get_server  },
+	{ "log",         lua_buffer_log         },
+	{ "execute",     lua_buffer_execute     },
+	{ NULL,          NULL                   }
 };
 
 // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
@@ -8473,6 +8678,19 @@ lua_server_gc (lua_State *L)
 }
 
 static int
+lua_server_get_user (lua_State *L)
+{
+	struct lua_server *wrapper = luaL_checkudata (L, 1, XLUA_SERVER_METATABLE);
+	luaL_argcheck (L, wrapper->server, 1, "dead reference used");
+
+	if (wrapper->server->irc_user)
+		lua_plugin_push_user (wrapper->plugin, wrapper->server->irc_user);
+	else
+		lua_pushnil (L);
+	return 1;
+}
+
+static int
 lua_server_get_buffer (lua_State *L)
 {
 	struct lua_server *wrapper = luaL_checkudata (L, 1, XLUA_SERVER_METATABLE);
@@ -8500,6 +8718,7 @@ static luaL_Reg lua_server_table[] =
 {
 	// TODO: some useful methods or values
 	{ "__gc",       lua_server_gc         },
+	{ "get_user",   lua_server_get_user   },
 	{ "get_buffer", lua_server_get_buffer },
 	{ "send",       lua_server_send       },
 	{ NULL,         NULL                  }
@@ -9631,6 +9850,8 @@ lua_plugin_load (struct app_context *ctx, const char *filename,
 
 	// Create metatables for our objects
 	lua_plugin_create_meta (L, XLUA_HOOK_METATABLE,       lua_hook_table);
+	lua_plugin_create_meta (L, XLUA_USER_METATABLE,       lua_user_table);
+	lua_plugin_create_meta (L, XLUA_CHANNEL_METATABLE,    lua_channel_table);
 	lua_plugin_create_meta (L, XLUA_BUFFER_METATABLE,     lua_buffer_table);
 	lua_plugin_create_meta (L, XLUA_SERVER_METATABLE,     lua_server_table);
 	lua_plugin_create_meta (L, XLUA_SCHEMA_METATABLE,     lua_schema_table);