[Minor] Add alternative format of metatokens

diff --git a/lualib/lua_meta.lua b/lualib/lua_meta.lua
index 96404192d72a117b0df3210c37c0f5c6cc11951f..0161eb5c9200240a61685d252388bca9ca5665e2 100644 (file)
--- a/lualib/lua_meta.lua
+++ b/lualib/lua_meta.lua
@@ -380,6 +380,23 @@ local function rspamd_gen_metatokens(task)
 end
 
 exports.rspamd_gen_metatokens = rspamd_gen_metatokens
+exports.gen_metatokens = rspamd_gen_metatokens
+
+local function rspamd_gen_metatokens_table(task)
+  local metatokens = {}
+
+  for _,mt in ipairs(metafunctions) do
+    local ct = mt.cb(task)
+    for i,tok in ipairs(ct) do
+      metatokens[mt.desc[i]] = tok
+    end
+  end
+
+  return metatokens
+end
+
+exports.rspamd_gen_metatokens_table = rspamd_gen_metatokens_table
+exports.gen_metatokens_table = rspamd_gen_metatokens_table
 
 local function rspamd_count_metatokens()
   local ipairs = ipairs
@@ -392,5 +409,6 @@ local function rspamd_count_metatokens()
 end
 
 exports.rspamd_count_metatokens = rspamd_count_metatokens
+exports.count_metatokens = rspamd_count_metatokens
 
 return exports