commit ef130e1: [Feature] Selectors: Add flatten function

[Vsevolod Stakhov]

Author: Vsevolod Stakhov
Date: 2019-08-21 15:30:35 +0100
URL: https://github.com/rspamd/rspamd/commit/ef130e10c0577a1f2e5baf175767a7a0a474969f

[Feature] Selectors: Add flatten function

---
 lualib/lua_selectors/init.lua | 32 ++++++++++++++++++++++++++++++--
 1 file changed, 30 insertions(+), 2 deletions(-)

diff --git a/lualib/lua_selectors/init.lua b/lualib/lua_selectors/init.lua
index 817330598..f148c4d46 100644
--- a/lualib/lua_selectors/init.lua
+++ b/lualib/lua_selectors/init.lua
@@ -461,10 +461,34 @@ exports.combine_selectors = function(_, selectors, delimiter)
   end
 end
 
+--[[[
+-- @function lua_selectors.flatten_selectors(selectors)
+-- Convert selectors to a flat table of elements
+--]]
+exports.flatten_selectors = function(selectors)
+  local res = {}
+
+  local function fill(tbl)
+    for _,s in ipairs(tbl) do
+      if type(s) == 'string' then
+        rawset(res, #res + 1, s)
+      elseif type(s) == 'userdata' then
+        rawset(res, #res + 1, tostring(s))
+      else
+        fill(s)
+      end
+    end
+  end
+
+  fill(selectors)
+
+  return res
+end
+
 --[[[
 -- @function lua_selectors.create_closure(cfg, selector_str, delimiter='')
 --]]
-exports.create_selector_closure = function(cfg, selector_str, delimiter)
+exports.create_selector_closure = function(cfg, selector_str, delimiter, flatten)
   local selector = exports.parse_selector(cfg, selector_str)
 
   if not selector then
@@ -475,7 +499,11 @@ exports.create_selector_closure = function(cfg, selector_str, delimiter)
     local res = exports.process_selectors(task, selector)
 
     if res then
-      return exports.combine_selectors(nil, res, delimiter)
+      if flatten then
+        return exports.flatten_selectors(res)
+      else
+        return exports.combine_selectors(nil, res, delimiter)
+      end
     end
 
     return nil