package org.codelibs.fess.score;

import java.util.Map;

import org.opensearch.search.rescore.RescorerBuilder;

/**
 * Interface for query rescoring implementations.
 */
public interface QueryRescorer {
    /**
     * Evaluates the rescorer with the given parameters.
     *
     * @param params the parameters for rescoring
     * @return the rescorer builder
     */
    RescorerBuilder<?> evaluate(final Map<String, Object> params);

    }

    /**
     * Adds a query rescorer for post-processing search results.
     * Query rescorers allow modification of search scores after the initial query execution.
     *
     * @param rescorer the query rescorer to add
     */
    public void addQueryRescorer(final QueryRescorer rescorer) {
        queryRescorerList.add(rescorer);
    }

import org.codelibs.fess.opensearch.query.StoredLtrQueryBuilder;
import org.codelibs.fess.util.ComponentUtil;
import org.opensearch.search.rescore.QueryRescorerBuilder;
import org.opensearch.search.rescore.RescorerBuilder;

/**
 * Learning to Rank query rescorer implementation.
 */
public class LtrQueryRescorer implements QueryRescorer {

    /**
     * Default constructor.
     */

            params.put("query", query);
            params.put("responseFields", responseFields);
            params.put("offset", offset);
            params.put("size", size);
            // TODO support rescorer(convert to map)
            // params.put("geoInfo", geoInfo);
            // params.put("facetInfo", facetInfo);
            params.put("similarDocHash", similarDocHash);
            return params;
        }

        // Test with first implementation
        QueryRescorer rescorer1 = new QueryRescorer() {
            @Override
            public RescorerBuilder<?> evaluate(Map<String, Object> params) {
                return createMockRescorerBuilder();
            }
        };

        // Test with second implementation
        QueryRescorer rescorer2 = new QueryRescorer() {
            @Override

            searchPaging(data, form);
        });
    }

    /**
     * Sets up pagination data and restores search form values from the pager.
     *
     * @param data the render data to populate with failure URL items
     * @param form the search form to restore values into
     */
    protected void searchPaging(final RenderData data, final SearchForm form) {

import org.codelibs.fess.unit.UnitFessTestCase;
import org.codelibs.fess.util.ComponentUtil;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.TestInfo;
import org.opensearch.search.rescore.QueryRescorerBuilder;
import org.opensearch.search.rescore.RescorerBuilder;

public class LtrQueryRescorerTest extends UnitFessTestCase {

    private LtrQueryRescorer ltrQueryRescorer;

    @Override

        try {
            String result = SystemUtil.getSearchEngineHttpAddress();
            assertNull(result);
        } finally {
            // Restore original value if it existed
            if (originalValue != null) {
                System.setProperty(Constants.FESS_SEARCH_ENGINE_HTTP_ADDRESS, originalValue);
            }
        }
    }

    @Test

            searchPaging(data, form);
        });
    }

    /**
     * Sets up pagination data for the web config search results.
     * Registers web config items and restores search criteria from the pager.
     *
     * @param data the render data to populate with search results
     * @param form the search form containing filter criteria
     */

        originalTimeZone = TimeZone.getDefault();
        originalProvider = getDBFluteSystemTimeZoneProvider();
    }

    @Override
    protected void tearDown(TestInfo testInfo) throws Exception {
        // Restore original timezone settings
        TimeZone.setDefault(originalTimeZone);
        if (originalProvider != null) {
            DBFluteSystem.unlock();
            DBFluteSystem.setFinalTimeZoneProvider(originalProvider);