fix #2835 Add interface for rewriting search request parameters befor…

…e processing
codelibs · Jul 26, 2024 · 8c48ac1 · 8c48ac1
1 parent 09fb29e
commit 8c48ac1
Showing 1 changed file with 28 additions and 1 deletion.
diff --git a/src/main/java/org/codelibs/fess/helper/SearchHelper.java b/src/main/java/org/codelibs/fess/helper/SearchHelper.java
@@ -16,6 +16,7 @@
 package org.codelibs.fess.helper;
 
 import java.text.NumberFormat;
+import java.util.Arrays;
 import java.util.Enumeration;
 import java.util.HashMap;
 import java.util.HashSet;
@@ -72,15 +73,24 @@ public class SearchHelper {
 
     private static final Logger logger = LogManager.getLogger(SearchHelper.class);
 
+    // ===================================================================================
+    //                                                                            Variable
+    //
+
+    protected SearchRequestParamsRewriter[] searchRequestParamsRewriters = new SearchRequestParamsRewriter[0];
+
     // ===================================================================================
     //                                                                              Method
     //                                                                      ==============
 
-    public void search(final SearchRequestParams params, final SearchRenderData data, final OptionalThing<FessUserBean> userBean) {
+    public void search(final SearchRequestParams searchRequestParams, final SearchRenderData data,
+            final OptionalThing<FessUserBean> userBean) {
         final SystemHelper systemHelper = ComponentUtil.getSystemHelper();
         final long startTime = systemHelper.getCurrentTimeAsLong();
         final long requestedTime = startTime;
 
+        final SearchRequestParams params = rewrite(searchRequestParams);
+
         LaRequestUtil.getOptionalRequest().ifPresent(request -> {
             request.setAttribute(Constants.REQUEST_LANGUAGES, params.getLanguages());
             request.setAttribute(Constants.REQUEST_QUERIES, params.getQuery());
@@ -354,4 +364,21 @@ public boolean bulkUpdate(final Consumer<BulkRequestBuilder> consumer) {
             throw new SearchEngineClientException("Failed to update bulk data.", e);
         }
     }
+
+    protected SearchRequestParams rewrite(final SearchRequestParams params) {
+        SearchRequestParams newParams = params;
+        for (final SearchRequestParamsRewriter rewriter : searchRequestParamsRewriters) {
+            newParams = rewriter.rewrite(newParams);
+        }
+        return newParams;
+    }
+
+    public void addRewriter(final SearchRequestParamsRewriter rewriter) {
+        searchRequestParamsRewriters = Arrays.copyOf(searchRequestParamsRewriters, searchRequestParamsRewriters.length + 1);
+        searchRequestParamsRewriters[searchRequestParamsRewriters.length - 1] = rewriter;
+    }
+
+    public interface SearchRequestParamsRewriter {
+        SearchRequestParams rewrite(SearchRequestParams params);
+    }
 }