Add other rules types.

1 files changed, 43 insertions, 0 deletions

diff --git a/src/plugins/lua/spamassassin.lua b/src/plugins/lua/spamassassin.lua
index 2cc59ff6e..03387255b 100644
--- a/src/plugins/lua/spamassassin.lua
+++ b/src/plugins/lua/spamassassin.lua
@@ -153,6 +153,7 @@ _.each(function(k, r)
           -- Subject for optimization
           if (r['re']:match(rh['decoded'])) then
             task:insert_result(k, 1.0)
+            return
           end
         end
       end
@@ -165,4 +166,46 @@ _.each(function(k, r)
   _.filter(function(k, r)
       return r['type'] == 'header' and r['header']
     end,
+    rules))
+
+-- Parts rules
+_.each(function(k, r)
+    local f = function(task)
+      local parts = task:get_parts()
+      if parts then
+        for n, part in ipairs(parts) do
+          -- Subject for optimization
+          if (r['re']:match(part:get_content())) then
+            task:insert_result(k, 1.0)
+            return
+          end
+        end
+      end
+    end
+    rspamd_config:register_symbol(k, 1.0, f)
+    if r['score'] then
+      rspamd_config:set_metric_symbol(k, r['score'], r['description'])
+    end
+  end,
+  _.filter(function(k, r)
+      return r['type'] == 'part'
+    end,
+    rules))
+
+-- Raw body rules
+_.each(function(k, r)
+    local f = function(task)
+      if (r['re']:match(task:get_content())) then
+        task:insert_result(k, 1.0)
+        return
+      end
+    end
+    rspamd_config:register_symbol(k, 1.0, f)
+    if r['score'] then
+      rspamd_config:set_metric_symbol(k, r['score'], r['description'])
+    end
+  end,
+  _.filter(function(k, r)
+      return r['type'] == 'message'
+    end,
     rules))
\ No newline at end of file