vstakhov/footer.lua Secret

## footer.lua
local lua_mime = require "lua_mime"
local rspamd_logger = require "rspamd_logger"
local ucl = require "ucl"
local lua_util = require "lua_util"
local N = "footer"
local opts =  rspamd_config:get_all_opt(N)

local function unescape(str)
  str = string.gsub(str, "%%20", " ")
  str = string.gsub(str, "%%0A", "\n")
  str = string.gsub(str, "%%0D", "\r")
  return string.gsub(str, "%%09", "\t")
end

local function newline(task)
  local t = task:get_newlines_type()

  if t == 'cr' then
    return '\r'
  elseif t == 'lf' then
    return '\n'
  end

  return '\r\n'
end

rspamd_config.TEST_FOOTER = {
  callback = function(task)
    local footer = {
      html = "<br><p>html footer</p>",
      plain = "\r\nplain footer"
    }
    local out = {} -- Start with headers
    local rewrite = lua_mime.add_text_footer(task, footer.html, footer.plain) or {}

    local seen_cte
    local newline_s = newline(task)

    local function rewrite_ct_cb(name, hdr)
      if rewrite.need_rewrite_ct then
        if name:lower() == 'content-type' then
          local nct = string.format('%s: %s/%s; charset=utf-8',
              'Content-Type', rewrite.new_ct.type, rewrite.new_ct.subtype)
          out[#out + 1] = nct
          return
        elseif name:lower() == 'content-transfer-encoding' then
          out[#out + 1] = string.format('%s: %s',
              'Content-Transfer-Encoding', 'quoted-printable')
          seen_cte = true
          return
        end
      end
      out[#out + 1] = hdr.raw:gsub('\r?\n?$', '')
    end

    task:headers_foreach(rewrite_ct_cb, {full = true})


    if not seen_cte and rewrite.need_rewrite_ct then
      out[#out + 1] = string.format('%s: %s', 'Content-Transfer-Encoding', 'quoted-printable')
    end

    -- End of headers
    out[#out + 1] = newline_s

    if rewrite.out then
      for _,o in ipairs(rewrite.out) do
        out[#out + 1] = o
      end
    else
      out[#out + 1] = task:get_rawbody()
    end
    local out_parts = {}
    for _,o in ipairs(out) do
       if type(o) ~= 'table' then
         out_parts[#out_parts + 1] = o
         --out_parts[#out_parts + 1] = newline_s
       else
         out_parts[#out_parts + 1] = o[1]
         if o[2] then
           out_parts[#out_parts + 1] = newline_s
         end
       end
    end
    task:set_message(out_parts)

    return true
  end,
  priority = 1,
  type = 'prefilter'
}