/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the "Elastic License
 * 2.0", the "GNU Affero General Public License v3.0 only", and the "Server Side
 * Public License v 1"; you may not use this file except in compliance with, at
 * your election, the "Elastic License 2.0", the "GNU Affero General Public
 * License v3.0 only", or the "Server Side Public License, v 1".
 */

package org.elasticsearch.search.retriever;

import org.elasticsearch.action.ActionRequestValidationException;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.search.SearchModule;
import org.elasticsearch.search.builder.SearchSourceBuilder;
import org.elasticsearch.test.ESTestCase;
import org.elasticsearch.xcontent.NamedXContentRegistry;
import org.elasticsearch.xcontent.XContentParser;
import org.elasticsearch.xcontent.json.JsonXContent;

import java.io.IOException;
import java.util.List;

import static org.hamcrest.Matchers.containsString;

/**
 * Tests exceptions related to usage of restricted global values with a retriever.
 */
public class RetrieverBuilderErrorTests extends ESTestCase {

    public void testRetrieverExtractionErrors() throws IOException {
        try (
            XContentParser parser = createParser(
                JsonXContent.jsonXContent,
                "{\"query\": {\"match_all\": {}}, \"retriever\":{\"standard\":{}}}"
            )
        ) {
            SearchSourceBuilder ssb = new SearchSourceBuilder();
            ssb.parseXContent(parser, true, nf -> true);
            ActionRequestValidationException iae = ssb.validate(null, false, false);
            assertNotNull(iae);
            assertThat(iae.getMessage(), containsString("cannot specify [retriever] and [query]"));
        }

        try (
            XContentParser parser = createParser(
                JsonXContent.jsonXContent,
                "{\"knn\":{\"field\": \"test\", \"k\": 2, \"num_candidates\": 5,"
                    + " \"query_vector\": [1, 2, 3]}, \"retriever\":{\"standard\":{}}}"
            )
        ) {
            SearchSourceBuilder ssb = new SearchSourceBuilder();
            ssb.parseXContent(parser, true, nf -> true);
            ActionRequestValidationException iae = ssb.validate(null, false, false);
            assertNotNull(iae);
            assertThat(iae.getMessage(), containsString("cannot specify [retriever] and [knn]"));
        }

        try (XContentParser parser = createParser(JsonXContent.jsonXContent, "{\"search_after\": [1], \"retriever\":{\"standard\":{}}}")) {
            SearchSourceBuilder ssb = new SearchSourceBuilder();
            ssb.parseXContent(parser, true, nf -> true);
            ActionRequestValidationException iae = ssb.validate(null, false, false);
            assertNotNull(iae);
            assertThat(iae.getMessage(), containsString("cannot specify [retriever] and [search_after]"));

        }

        try (XContentParser parser = createParser(JsonXContent.jsonXContent, "{\"terminate_after\": 1, \"retriever\":{\"standard\":{}}}")) {
            SearchSourceBuilder ssb = new SearchSourceBuilder();
            ssb.parseXContent(parser, true, nf -> true);
            ActionRequestValidationException iae = ssb.validate(null, false, false);
            assertNotNull(iae);
            assertThat(iae.getMessage(), containsString("cannot specify [retriever] and [terminate_after]"));
        }

        try (
            XContentParser parser = createParser(
                JsonXContent.jsonXContent,
                "{\"rescore\": {\"query\": {\"rescore_query\": {\"match_all\": {}}}}, \"retriever\":{\"standard\":{}}}"
            )
        ) {
            SearchSourceBuilder ssb = new SearchSourceBuilder();
            ssb.parseXContent(parser, true, nf -> true);
            ActionRequestValidationException iae = ssb.validate(null, false, false);
            assertNotNull(iae);
            assertThat(iae.getMessage(), containsString("cannot specify [retriever] and [rescore]"));
        }

        try (
            XContentParser parser = createParser(
                JsonXContent.jsonXContent,
                "" + "{\"min_score\": 2, \"query\": {\"match_all\": {}}, \"retriever\":{\"standard\":{}}, \"terminate_after\": 1}"
            )
        ) {
            SearchSourceBuilder ssb = new SearchSourceBuilder();
            ssb.parseXContent(parser, true, nf -> true);
            ActionRequestValidationException iae = ssb.validate(null, false, false);
            assertNotNull(iae);
            assertThat(iae.getMessage(), containsString("cannot specify [retriever] and [query, terminate_after]"));
        }
    }

    @Override
    protected NamedXContentRegistry xContentRegistry() {
        return new NamedXContentRegistry(new SearchModule(Settings.EMPTY, List.of()).getNamedXContents());
    }
}