From d4d22517abb0f17f65e579c4bb8d9e784421247d Mon Sep 17 00:00:00 2001
From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Fri, 1 Sep 2017 15:09:34 +0100
Subject: [PATCH] [Feature] Read global maps for lua

---
 rules/rspamd.lua | 41 ++++++++++++++++++++++++++++-------------
 1 file changed, 28 insertions(+), 13 deletions(-)

diff --git a/rules/rspamd.lua b/rules/rspamd.lua
index 3cdd626f9..1c0ddaa8a 100644
--- a/rules/rspamd.lua
+++ b/rules/rspamd.lua
@@ -19,6 +19,7 @@ limitations under the License.
 require "global_functions" ()
 
 config['regexp'] = {}
+rspamd_maps = {} -- Global maps
 
 local local_conf = rspamd_paths['CONFDIR']
 local local_rules = rspamd_paths['RULESDIR']
@@ -36,24 +37,38 @@ dofile(local_rules .. '/forwarding.lua')
 dofile(local_rules .. '/mid.lua')
 
 local function file_exists(filename)
-	local file = io.open(filename)
-	if file then
-		io.close(file)
-		return true
-	else
-		return false
-	end
+  local file = io.open(filename)
+  if file then
+    io.close(file)
+    return true
+  else
+    return false
+  end
 end
 
 if file_exists(local_conf .. '/rspamd.local.lua') then
-	dofile(local_conf .. '/rspamd.local.lua')
+  dofile(local_conf .. '/rspamd.local.lua')
 else
-	-- Legacy lua/rspamd.local.lua
-	if file_exists(local_conf .. '/lua/rspamd.local.lua') then
-		dofile(local_conf .. '/lua/rspamd.local.lua')
-	end
+  -- Legacy lua/rspamd.local.lua
+  if file_exists(local_conf .. '/lua/rspamd.local.lua') then
+    dofile(local_conf .. '/lua/rspamd.local.lua')
+  end
 end
 
 if file_exists(local_rules .. '/rspamd.classifiers.lua') then
-	dofile(local_rules .. '/rspamd.classifiers.lua')
+  dofile(local_rules .. '/rspamd.classifiers.lua')
 end
+
+local rmaps =  rspamd_config:get_all_opt("lua_maps")
+if rmaps and type(rmaps) == 'table' then
+  local rspamd_logger = require "rspamd_logger"
+  for k,v in pairs(rmaps) do
+    local status,map_or_err = pcall(rspamd_config:add_map(v))
+
+    if not status then
+      rspamd_logger.errx(rspamd_config, "cannot add map %s: %s", k, map_or_err)
+    else
+      rspamd_maps[k] = map_or_err
+    end
+  end
+end
\ No newline at end of file
-- 
2.39.5