[Project] Use lua_mime to modify headers

diff --git a/src/plugins/lua/arc.lua b/src/plugins/lua/arc.lua
index fd062ff5c960bd6be03a7a51902196b07546f6fe..0e575c5fdf09b9740ec8608819fabe295e71ee1b 100644 (file)
--- a/src/plugins/lua/arc.lua
+++ b/src/plugins/lua/arc.lua
@@ -23,6 +23,7 @@ local rspamd_rsa = require "rspamd_rsa"
 local fun = require "fun"
 local auth_results = require "lua_auth_results"
 local hash = require "rspamd_cryptobox_hash"
+local lua_mime = require "lua_mime"
 
 if confighelp then
   return
@@ -534,12 +535,12 @@ local function arc_sign_seal(task, params, header)
   cur_arc_seal = string.format('%s%s', cur_arc_seal,
     sig:base64(70, nl_type))
 
-  task:set_milter_reply({
-    add_headers = {
+  lua_mime.modify_headers(task, {
+    add = {
       ['ARC-Authentication-Results'] = {order = 1, value = cur_auth_results},
       ['ARC-Message-Signature'] = {order = 1, value = header},
       ['ARC-Seal'] = {order = 1, value = lua_util.fold_header(task,
-        'ARC-Seal', cur_arc_seal) }
+              'ARC-Seal', cur_arc_seal) }
     }
   })
   task:insert_result(settings.sign_symbol, 1.0,

diff --git a/src/plugins/lua/dkim_signing.lua b/src/plugins/lua/dkim_signing.lua
index e42e0f75c97060835026216e34383540e2bb25de..1e5a35be74b7d231f578cceafcb6a2feb7b2f757 100644 (file)
--- a/src/plugins/lua/dkim_signing.lua
+++ b/src/plugins/lua/dkim_signing.lua
@@ -19,6 +19,7 @@ local lua_util = require "lua_util"
 local rspamd_logger = require "rspamd_logger"
 local dkim_sign_tools = require "lua_dkim_tools"
 local lua_redis = require "lua_redis"
+local lua_mime = require "lua_mime"
 
 if confighelp then
   return
@@ -55,8 +56,8 @@ local sign_func = rspamd_plugins.dkim.sign
 
 local function insert_sign_results(task, ret, hdr, dkim_params)
   if settings.use_milter_headers then
-    task:set_milter_reply({
-      add_headers = {
+    lua_mime.modify_headers(task, {
+      add = {
         ['DKIM-Signature'] = {order = 1, value = hdr},
       }
     })

diff --git a/src/plugins/lua/milter_headers.lua b/src/plugins/lua/milter_headers.lua
index dd2c27f3c0a3638f245b0087f6bd9cd087dec3a9..b6595402594bf4cdf3e5d88769bf548af641d962 100644 (file)
--- a/src/plugins/lua/milter_headers.lua
+++ b/src/plugins/lua/milter_headers.lua
@@ -26,6 +26,7 @@ local util = require "rspamd_util"
 local N = 'milter_headers'
 local lua_util = require "lua_util"
 local lua_maps = require "lua_maps"
+local lua_mime = require "lua_mime"
 local ts = require("tableshape").types
 local E = {}
 
@@ -580,9 +581,10 @@ local function milter_headers(task)
   if not next(add) then add = nil end
   if not next(remove) then remove = nil end
   if add or remove then
-    task:set_milter_reply({
-      add_headers = add,
-      remove_headers = remove
+
+    lua_mime.modify_headers(task, {
+      add = add,
+      remove = remove
     })
   end
 end