From: Vsevolod Stakhov <vsevolod@rspamd.com>
Date: Tue, 10 Dec 2024 14:57:22 +0000 (+0000)
Subject: [Minor] GPT: use lua_mime routine to select text part
X-Git-Tag: 3.11.0~8
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=1b78893987deb9913e0cd7e5534c592f4a3ad1a9;p=thirdparty%2Frspamd.git

[Minor] GPT: use lua_mime routine to select text part
---

diff --git a/src/plugins/lua/gpt.lua b/src/plugins/lua/gpt.lua
index 014f47d72f..10cff0fb56 100644
--- a/src/plugins/lua/gpt.lua
+++ b/src/plugins/lua/gpt.lua
@@ -56,6 +56,7 @@ end
 local lua_util = require "lua_util"
 local rspamd_http = require "rspamd_http"
 local rspamd_logger = require "rspamd_logger"
+local lua_mime = require "lua_mime"
 local ucl = require "ucl"
 local fun = require "fun"
 
@@ -128,19 +129,7 @@ local function default_condition(task)
   end
 
   -- Check if we have text at all
-  local mp = task:get_parts() or {}
-  local sel_part
-  for _, mime_part in ipairs(mp) do
-    if mime_part:is_text() then
-      local part = mime_part:get_text()
-      if part:is_html() then
-        -- We prefer html content
-        sel_part = part
-      elseif not sel_part then
-        sel_part = part
-      end
-    end
-  end
+  local sel_part = lua_mime.get_displayed_text_part(task)
 
   if not sel_part then
     return false, 'no text part found'