[Feature] Print map id when using it in lua logger

1 files changed, 24 insertions, 0 deletions

diff --git a/src/lua/lua_common.h b/src/lua/lua_common.h
index ed284ff5b..69e02142b 100644
--- a/src/lua/lua_common.h
+++ b/src/lua/lua_common.h
@@ -82,6 +82,28 @@ struct rspamd_lua_regexp {
 	gint re_flags;
 };
 
+struct rspamd_map;
+struct lua_map_callback_data;
+struct radix_tree_compressed;
+
+enum rspamd_lua_map_type {
+	RSPAMD_LUA_MAP_RADIX = 0,
+	RSPAMD_LUA_MAP_SET,
+	RSPAMD_LUA_MAP_HASH,
+	RSPAMD_LUA_MAP_CALLBACK
+};
+
+struct rspamd_lua_map {
+	struct rspamd_map *map;
+	enum rspamd_lua_map_type type;
+
+	union {
+		struct radix_tree_compressed *radix;
+		GHashTable *hash;
+		struct lua_map_callback_data *cbdata;
+	} data;
+};
+
 /* Common utility functions */
 
 /**
@@ -176,6 +198,8 @@ gint rspamd_lua_push_header (lua_State * L,
  */
 struct rspamd_task *lua_check_task (lua_State * L, gint pos);
 
+struct rspamd_lua_map *lua_check_map (lua_State * L, gint pos);
+
 /**
  * Push ip address from a string (nil is pushed if a string cannot be converted)
  */