diff --git a/src/main/java/org/codelibs/fess/helper/SearchHelper.java b/src/main/java/org/codelibs/fess/helper/SearchHelper.java index c640f7783..b46437129 100644 --- 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 userBean) { + public void search(final SearchRequestParams searchRequestParams, final SearchRenderData data, + final OptionalThing 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 class SearchHelper { 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); + } }