From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Thu, 23 Mar 2017 14:11:29 +0000 (+0000)
Subject: [Feature] Support tag_exists SA function
X-Git-Tag: 1.5.4~48
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=8e865ebfcb0fe584841c2e082acd894646569213;p=thirdparty%2Frspamd.git

[Feature] Support tag_exists SA function
---

diff --git a/src/plugins/lua/spamassassin.lua b/src/plugins/lua/spamassassin.lua
index bb80ce6b20..2a3c04bfc5 100644
--- a/src/plugins/lua/spamassassin.lua
+++ b/src/plugins/lua/spamassassin.lua
@@ -37,6 +37,7 @@ local known_plugins = {
   'Mail::SpamAssassin::Plugin::BodyEval',
   'Mail::SpamAssassin::Plugin::MIMEHeader',
   'Mail::SpamAssassin::Plugin::WLBLEval',
+  'Mail::SpamAssassin::Plugin::HTMLEval',
 }
 
 -- Table that replaces SA symbol with rspamd equialent
@@ -520,6 +521,24 @@ local function gen_eval_rule(arg)
         return 0
       end
     },
+    {
+      'html_tag_exists',
+      function(task, remain)
+        local tp = task:get_text_parts()
+
+        for _,p in ipairs(tp) do
+          if p:is_html() then
+            local hc = p:get_html()
+
+            if hc:has_tag(remain) then
+              return 1
+            end
+          end
+        end
+
+        return 0
+      end
+    }
   }
 
   for _,f in ipairs(eval_funcs) do