From a492727ffcefcc4ae7c6b1790da04a1b61aaa5f1 Mon Sep 17 00:00:00 2001
From: Alexander Moisseev <moiseev@mezonplus.ru>
Date: Wed, 18 Jun 2025 16:56:55 +0300
Subject: [PATCH] [WebUI] Add Bayes classifier selector to Learn card

This enables users to choose a specific Bayes classifier for training from WebUI.
---
 interface/index.html       |  9 ++++++---
 interface/js/app/rspamd.js |  2 ++
 interface/js/app/upload.js | 18 +++++++++++++++++-
 src/controller.c           |  2 +-
 4 files changed, 26 insertions(+), 5 deletions(-)

diff --git a/interface/index.html b/interface/index.html
index b176d65273..61487ce295 100644
--- a/interface/index.html
+++ b/interface/index.html
@@ -479,9 +479,12 @@
 								<div class="card bg-light shadow card-body card p-2">
 									<p>Learn Bayesian classifier:</p>
 									<form>
-										<div class="btn-group">
-											<button class="btn btn-success d-flex align-items-center" type="button" data-upload="learnham" disabled><i class="fas fa-thumbs-up me-2"></i>Upload HAM</button>
-											<button class="btn btn-danger d-flex align-items-center" type="button" data-upload="learnspam" disabled><i class="fas fa-thumbs-down me-2"></i>Upload SPAM</button>
+										<div class="d-flex flex-wrap flex-lg-column align-items-start align-items-lg-stretch gap-2">
+											<select id="classifier" class="form-select w-auto"></select>
+											<div class="btn-group">
+												<button class="btn btn-success d-flex align-items-center" type="button" data-upload="learnham" disabled><i class="fas fa-thumbs-up me-2"></i>Upload HAM</button>
+												<button class="btn btn-danger d-flex align-items-center" type="button" data-upload="learnspam" disabled><i class="fas fa-thumbs-down me-2"></i>Upload SPAM</button>
+											</div>
 										</div>
 									</form>
 								</div>
diff --git a/interface/js/app/rspamd.js b/interface/js/app/rspamd.js
index 6d047d6f66..da4e9bccfe 100644
--- a/interface/js/app/rspamd.js
+++ b/interface/js/app/rspamd.js
@@ -236,6 +236,8 @@ define(["jquery", "app/common", "stickytabs", "visibility",
             complete: function () {
                 ajaxSetup(localStorage.getItem("ajax_timeout"));
 
+                if (require.defined("app/upload")) require(["app/upload"], (module) => module.getClassifiers());
+
                 if (common.read_only) {
                     $(".ro-disable").attr("disabled", true);
                     $(".ro-hide").hide();
diff --git a/interface/js/app/upload.js b/interface/js/app/upload.js
index a5d30b59ef..0960ebf25b 100644
--- a/interface/js/app/upload.js
+++ b/interface/js/app/upload.js
@@ -217,7 +217,10 @@ define(["jquery", "app/common", "app/libft"],
                     getFuzzyHashes(data);
                 } else {
                     let headers = {};
-                    if (source === "fuzzyadd") {
+                    if (source === "learnham" || source === "learnspam") {
+                        const classifier = $("#classifier").val();
+                        if (classifier) headers = {classifier: classifier};
+                    } else if (source === "fuzzyadd") {
                         headers = {
                             flag: $("#fuzzyFlagText").val(),
                             weight: $("#fuzzyWeightText").val()
@@ -296,5 +299,18 @@ define(["jquery", "app/common", "app/libft"],
 
         common.fileUtils.setupFileHandling("#scanMsgSource", "#formFile", fileSet, enable_disable_scan_btn, multiple_files_cb);
 
+        ui.getClassifiers = function () {
+            if (!common.read_only) {
+                const sel = $("#classifier").empty().append($("<option>", {value: "", text: "All classifiers"}));
+                common.query("/bayes/classifiers", {
+                    success: function (data) {
+                        data[0].data.forEach((c) => sel.append($("<option>", {value: c, text: c})));
+                    },
+                    server: common.getServer()
+                });
+            }
+        };
+        ui.getClassifiers();
+
         return ui;
     });
diff --git a/src/controller.c b/src/controller.c
index 9ba95fcaba..0550ba6b86 100644
--- a/src/controller.c
+++ b/src/controller.c
@@ -3292,7 +3292,7 @@ rspamd_controller_handle_unknown(struct rspamd_http_connection_entry *conn_ent,
 		rspamd_http_message_add_header(rep, "Access-Control-Allow-Methods",
 									   "POST, GET, OPTIONS");
 		rspamd_http_message_add_header(rep, "Access-Control-Allow-Headers",
-									   "Content-Type,Password,Map,Weight,Flag,Hash");
+									   "Classifier,Content-Type,Password,Map,Weight,Flag,Hash");
 		rspamd_http_connection_reset(conn_ent->conn);
 		rspamd_http_router_insert_headers(conn_ent->rt, rep);
 		rspamd_http_connection_write_message(conn_ent->conn,
-- 
2.47.3