Add check_for_mime('mime_attachement') function

1 files changed, 23 insertions, 0 deletions

diff --git a/src/plugins/lua/spamassassin.lua b/src/plugins/lua/spamassassin.lua
index 7f5376019..fd1d62854 100644
--- a/src/plugins/lua/spamassassin.lua
+++ b/src/plugins/lua/spamassassin.lua
@@ -291,6 +291,29 @@ local function gen_eval_rule(arg)
         return 0
       end
     },
+    {
+      'check_for_mime',
+      function(task, remain)
+        local arg = string.match(remain, "^%(%s*['\"]([^%s]+)['\"]%s*%)$")
+
+        if arg then
+          if arg == 'mime_attachment' then
+            local parts = task:get_parts()
+            if parts then
+              for i,p in ipairs(parts) do
+                if p:get_filename() then
+                  return 1
+                end
+              end
+            end
+          else
+            rspamd_logger.infox(task, 'unimplemented mime check %1', arg)
+          end
+        end
+
+        return 0
+      end
+    }
   }
 
   for k,f in ipairs(eval_funcs) do