From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Tue, 16 Jul 2019 19:02:45 +0000 (+0100)
Subject: [Minor] Selectors: Add append/prepend transforms
X-Git-Tag: 2.0~568
X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=e872fbfb5ea795568017dd97db8fffc827d89da3;p=rspamd.git

[Minor] Selectors: Add append/prepend transforms
---

diff --git a/lualib/lua_selectors.lua b/lualib/lua_selectors.lua
index 3bc7b9d75..a55d0742b 100644
--- a/lualib/lua_selectors.lua
+++ b/lualib/lua_selectors.lua
@@ -494,6 +494,32 @@ the second argument is optional hash type (`blake2`, `sha256`, `sha1`, `sha512`,
     ['args_schema'] = {(ts.number + ts.string / tonumber):is_optional(),
                        (ts.number + ts.string / tonumber):is_optional()}
   },
+  -- Prepends a string or a strings list
+  ['prepend'] = {
+    ['types'] = {
+      ['string'] = true
+    },
+    ['map_type'] = 'string',
+    ['process'] = function(inp, _, args)
+      local prepend = table.concat(args, '')
+
+      return prepend .. inp, 'string'
+    end,
+    ['description'] = 'Prepends a string or a strings list',
+  },
+  -- Appends a string or a strings list
+  ['append'] = {
+    ['types'] = {
+      ['string'] = true
+    },
+    ['map_type'] = 'string',
+    ['process'] = function(inp, _, args)
+      local append = table.concat(args, '')
+
+      return inp .. append, 'string'
+    end,
+    ['description'] = 'Appends a string or a strings list',
+  },
   -- Regexp matching
   ['regexp'] = {
     ['types'] = {