From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Fri, 21 Aug 2020 20:36:28 +0000 (+0100)
Subject: [Minor] Allow to add all parameters into `rcpts` and `from` selectors
X-Git-Tag: 2.6~135
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=450d49c880e636c35df96e65339803ede0fc0b2c;p=thirdparty%2Frspamd.git

[Minor] Allow to add all parameters into `rcpts` and `from` selectors
---

diff --git a/lualib/lua_selectors/extractors.lua b/lualib/lua_selectors/extractors.lua
index 4727b389c6..515d522cb3 100644
--- a/lualib/lua_selectors/extractors.lua
+++ b/lualib/lua_selectors/extractors.lua
@@ -58,7 +58,12 @@ For example, `list('foo', 'bar')` returns a list {'foo', 'bar'}]],
   -- Get MIME from
   ['from'] = {
     ['get_value'] = function(task, args)
-      local from = task:get_from(args[1] or 0)
+      local from
+      if type(args) == 'table' then
+        from = task:get_from(args)
+      else
+        from = task:get_from(0)
+      end
       if ((from or E)[1] or E).addr then
         return from[1],'table'
       end
@@ -69,7 +74,12 @@ uses any type by default)]],
   },
   ['rcpts'] = {
     ['get_value'] = function(task, args)
-      local rcpts = task:get_recipients(args[1] or 0)
+      local rcpts
+      if type(args) == 'table' then
+        rcpts = task:get_recipients(args)
+      else
+        rcpts = task:get_recipients(0)
+      end
       if ((rcpts or E)[1] or E).addr then
         return rcpts,'table_list'
       end