rspamd.local.lua 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. rspamd_config.MAILCOW_AUTH = {
  2. callback = function(task)
  3. local uname = task:get_user()
  4. if uname then
  5. return 1
  6. end
  7. end
  8. }
  9. rspamd_config.MAILCOW_MOO = function (task)
  10. return true
  11. end
  12. local modify_subject_map = rspamd_config:add_map({
  13. url = 'http://nginx:8081/tags.php',
  14. type = 'map',
  15. description = 'Map of users to use subject tags for'
  16. })
  17. local auth_domain_map = rspamd_config:add_map({
  18. url = 'http://nginx:8081/authoritative.php',
  19. type = 'map',
  20. description = 'Map of domains we are authoritative for'
  21. })
  22. rspamd_config.ADD_DELIMITER_TAG = {
  23. callback = function(task)
  24. local util = require("rspamd_util")
  25. local rspamd_logger = require "rspamd_logger"
  26. local user_tagged = task:get_recipients(1)[1]['user']
  27. local domain = task:get_recipients(1)[1]['domain']
  28. local user, tag = user_tagged:match("([^+]+)+(.*)")
  29. local authdomain = auth_domain_map:get_key(domain)
  30. if tag and authdomain then
  31. rspamd_logger.infox("Domain %s is part of mailcow, start reading tag settings", domain)
  32. local user_untagged = user .. '@' .. domain
  33. rspamd_logger.infox("Querying tag settings for user %1", user_untagged)
  34. if modify_subject_map:get_key(user_untagged) then
  35. rspamd_logger.infox("User wants subject modified for tagged mail")
  36. local sbj = task:get_header('Subject')
  37. if tag then
  38. rspamd_logger.infox("Found tag %1, will modify subject header", tag)
  39. new_sbj = '=?UTF-8?B?' .. tostring(util.encode_base64('[' .. tag .. '] ' .. sbj)) .. '?='
  40. task:set_rmilter_reply({
  41. remove_headers = {['Subject'] = 1},
  42. add_headers = {['Subject'] = new_sbj}
  43. })
  44. end
  45. else
  46. rspamd_logger.infox("Add X-Moo-Tag header")
  47. task:set_rmilter_reply({
  48. add_headers = {['X-Moo-Tag'] = 'YES'}
  49. })
  50. end
  51. else
  52. rspamd_logger.infox("Skip delimiter handling for untagged message or authenticated user")
  53. end
  54. return false
  55. end
  56. }