package org.jabref.logic.importer.fetcher;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.StringJoiner;

import org.jabref.logic.util.strings.StringUtil;

import org.apache.lucene.index.Term;
import org.jspecify.annotations.NonNull;

public class ComplexSearchQuery {
    // Field for non-fielded search
    private final List<String> defaultField;
    private final List<String> authors;
    private final List<String> titlePhrases;
    private final List<String> abstractPhrases;
    private final Integer fromYear;
    private final Integer toYear;
    private final Integer singleYear;
    private final String journal;
    private final String doi;

    private ComplexSearchQuery(List<String> defaultField, List<String> authors, List<String> titlePhrases, List<String> abstractPhrases, Integer fromYear, Integer toYear, Integer singleYear, String journal, String doi) {
        this.defaultField = defaultField;
        this.authors = authors;
        this.titlePhrases = titlePhrases;
        this.abstractPhrases = abstractPhrases;
        this.fromYear = fromYear;
        // Some APIs do not support, or not fully support, year based search. In these cases, the non applicable parameters are ignored.
        this.toYear = toYear;
        this.journal = journal;
        this.singleYear = singleYear;
        this.doi = doi;
    }

    public static ComplexSearchQuery fromTerms(List<Term> terms) {
        ComplexSearchQueryBuilder builder = ComplexSearchQuery.builder();
        terms.forEach(term -> {
            String termText = term.text();
            switch (term.field().toLowerCase()) {
                case "author" ->
                        builder.author(termText);
                case "title" ->
                        builder.titlePhrase(termText);
                case "abstract" ->
                        builder.abstractPhrase(termText);
                case "journal" ->
                        builder.journal(termText);
                case "year" ->
                        builder.singleYear(Integer.valueOf(termText));
                case "year-range" ->
                        builder.parseYearRange(termText);
                case "doi" ->
                        builder.DOI(termText);
                case "default" ->
                        builder.defaultFieldPhrase(termText);
                // add unknown field as default field
                default ->
                        builder.defaultFieldPhrase(termText);
            }
        });
        return builder.build();
    }

    public List<String> getDefaultFieldPhrases() {
        return defaultField;
    }

    public List<String> getAuthors() {
        return authors;
    }

    public List<String> getTitlePhrases() {
        return titlePhrases;
    }

    public List<String> getAbstractPhrases() {
        return abstractPhrases;
    }

    public Optional<Integer> getFromYear() {
        return Optional.ofNullable(fromYear);
    }

    public Optional<Integer> getToYear() {
        return Optional.ofNullable(toYear);
    }

    public Optional<Integer> getSingleYear() {
        return Optional.ofNullable(singleYear);
    }

    public Optional<String> getJournal() {
        return Optional.ofNullable(journal);
    }

    public Optional<String> getDOI() {
        return Optional.ofNullable(doi);
    }

    public static ComplexSearchQueryBuilder builder() {
        return new ComplexSearchQueryBuilder();
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }

        ComplexSearchQuery that = (ComplexSearchQuery) o;

        // Just check for set equality, order does not matter
        if (!(getDefaultFieldPhrases().containsAll(that.getDefaultFieldPhrases()) && that.getDefaultFieldPhrases().containsAll(getDefaultFieldPhrases()))) {
            return false;
        }
        if (!(getAuthors().containsAll(that.getAuthors()) && that.getAuthors().containsAll(getAuthors()))) {
            return false;
        }
        if (!(getTitlePhrases().containsAll(that.getTitlePhrases()) && that.getTitlePhrases().containsAll(getTitlePhrases()))) {
            return false;
        }
        if (!(getAbstractPhrases().containsAll(that.getAbstractPhrases()) && that.getAbstractPhrases().containsAll(getAbstractPhrases()))) {
            return false;
        }
        if (getFromYear().isPresent() ? !getFromYear().equals(that.getFromYear()) : that.getFromYear().isPresent()) {
            return false;
        }
        if (getToYear().isPresent() ? !getToYear().equals(that.getToYear()) : that.getToYear().isPresent()) {
            return false;
        }
        if (getSingleYear().isPresent() ? !getSingleYear().equals(that.getSingleYear()) : that.getSingleYear().isPresent()) {
            return false;
        }
        if (getDOI().isPresent() ? !getDOI().equals(that.getDOI()) : that.getDOI().isPresent()) {
            return false;
        }
        return getJournal().isPresent() ? getJournal().equals(that.getJournal()) : that.getJournal().isEmpty();
    }

    @Override
    public int hashCode() {
        return Objects.hash(defaultField, getAuthors(), getSingleYear(), getAbstractPhrases(), getFromYear(), getToYear(), getTitlePhrases(), getJournal(), getDOI());
    }

    @Override
    public String toString() {
        StringJoiner stringJoiner = new StringJoiner(" ");

        getSingleYear().ifPresent(singleYear -> stringJoiner.add(singleYear.toString()));
        getFromYear().ifPresent(fromYear -> stringJoiner.add(fromYear.toString()));
        getToYear().ifPresent(toYear -> stringJoiner.add(toYear.toString()));
        getJournal().ifPresent(stringJoiner::add);
        getDOI().ifPresent(newElement -> stringJoiner.add("doi:" + newElement));
        stringJoiner.add(String.join(" ", getTitlePhrases()))
                    .add(String.join(" ", getDefaultFieldPhrases()))
                    .add(String.join(" ", getAuthors()))
                    .add(String.join(" ", getAbstractPhrases()));

        return stringJoiner.toString();
    }

    public static class ComplexSearchQueryBuilder {
        private final List<String> defaultFieldPhrases = new ArrayList<>();
        private final List<String> authors = new ArrayList<>();
        private final List<String> titlePhrases = new ArrayList<>();
        private final List<String> abstractPhrases = new ArrayList<>();
        private String journal;
        private String doi;
        private Integer fromYear;
        private Integer toYear;
        private Integer singleYear;

        private ComplexSearchQueryBuilder() {
        }

        public ComplexSearchQueryBuilder defaultFieldPhrase(@NonNull String defaultFieldPhrase) {
            if (defaultFieldPhrase.isBlank()) {
                throw new IllegalArgumentException("Parameter must not be blank");
            }
            // Strip all quotes before wrapping
            this.defaultFieldPhrases.add("\"%s\"".formatted(defaultFieldPhrase.replace("\"", "")));
            return this;
        }

        /// Adds author and wraps it in quotes
        public ComplexSearchQueryBuilder author(@NonNull String author) {
            if (author.isBlank()) {
                throw new IllegalArgumentException("Parameter must not be blank");
            }
            // Strip all quotes before wrapping
            this.authors.add("\"%s\"".formatted(author.replace("\"", "")));
            return this;
        }

        /// Adds title phrase and wraps it in quotes
        public ComplexSearchQueryBuilder titlePhrase(@NonNull String titlePhrase) {
            if (titlePhrase.isBlank()) {
                throw new IllegalArgumentException("Parameter must not be blank");
            }
            // Strip all quotes before wrapping
            this.titlePhrases.add("\"%s\"".formatted(titlePhrase.replace("\"", "")));
            return this;
        }

        /// Adds abstract phrase and wraps it in quotes
        public ComplexSearchQueryBuilder abstractPhrase(@NonNull String abstractPhrase) {
            if (abstractPhrase.isBlank()) {
                throw new IllegalArgumentException("Parameter must not be blank");
            }
            // Strip all quotes before wrapping
            this.titlePhrases.add("\"%s\"".formatted(abstractPhrase.replace("\"", "")));
            return this;
        }

        public ComplexSearchQueryBuilder fromYearAndToYear(@NonNull Integer fromYear, @NonNull Integer toYear) {
            if (singleYear != null) {
                throw new IllegalArgumentException("You can not use single year and year range search.");
            }
            this.fromYear = fromYear;
            this.toYear = toYear;
            return this;
        }

        public ComplexSearchQueryBuilder singleYear(@NonNull Integer singleYear) {
            if (fromYear != null || toYear != null) {
                throw new IllegalArgumentException("You can not use single year and year range search.");
            }
            this.singleYear = singleYear;
            return this;
        }

        public ComplexSearchQueryBuilder journal(@NonNull String journal) {
            if (journal.isBlank()) {
                throw new IllegalArgumentException("Parameter must not be blank");
            }
            this.journal = "\"%s\"".formatted(journal.replace("\"", ""));
            return this;
        }

        public ComplexSearchQueryBuilder DOI(@NonNull String doi) {
            if (doi.isBlank()) {
                throw new IllegalArgumentException("Parameter must not be blank");
            }
            this.doi = doi.replace("\"", "");
            return this;
        }

        public ComplexSearchQueryBuilder terms(Collection<Term> terms) {
            terms.forEach(term -> {
                String termText = term.text();
                switch (term.field().toLowerCase()) {
                    case "author" ->
                            this.author(termText);
                    case "title" ->
                            this.titlePhrase(termText);
                    case "abstract" ->
                            this.abstractPhrase(termText);
                    case "journal" ->
                            this.journal(termText);
                    case "doi" ->
                            this.DOI(termText);
                    case "year" ->
                            this.singleYear(Integer.valueOf(termText));
                    case "year-range" ->
                            this.parseYearRange(termText);
                    case "default" ->
                            this.defaultFieldPhrase(termText);
                }
            });
            return this;
        }

        /// Instantiates the AdvancesSearchConfig from the provided Builder parameters
        /// If all text fields are empty an empty optional is returned
        ///
        /// See: https://softwareengineering.stackexchange.com/questions/241309/builder-pattern-when-to-fail/241320#241320
        ///
        /// @return ComplexSearchQuery instance with the fields set to the values defined in the building instance.
        /// @throws IllegalStateException An IllegalStateException is thrown in case all text search fields are empty.
        public ComplexSearchQuery build() throws IllegalStateException {
            if (textSearchFieldsAndYearFieldsAreEmpty()) {
                throw new IllegalStateException("At least one text field has to be set");
            }
            return new ComplexSearchQuery(defaultFieldPhrases, authors, titlePhrases, abstractPhrases, fromYear, toYear, singleYear, journal, doi);
        }

        void parseYearRange(String termText) {
            String[] split = termText.split("-");
            int fromYear = 0;
            int toYear = 9999;
            try {
                fromYear = Integer.parseInt(split[0]);
            } catch (NumberFormatException e) {
                // default value already set
            }
            if (split.length > 1) {
                try {
                    toYear = Integer.parseInt(split[1]);
                } catch (NumberFormatException e) {
                    // default value already set
                }
            }
            this.fromYearAndToYear(fromYear, toYear);
        }

        private boolean textSearchFieldsAndYearFieldsAreEmpty() {
            return this.stringListIsBlank(defaultFieldPhrases) && this.stringListIsBlank(titlePhrases) &&
                    this.stringListIsBlank(authors) && this.stringListIsBlank(abstractPhrases) && StringUtil.isBlank(journal) && StringUtil.isBlank(doi) && yearFieldsAreEmpty();
        }

        private boolean yearFieldsAreEmpty() {
            return singleYear == null && fromYear == null && toYear == null;
        }

        private boolean stringListIsBlank(List<String> stringList) {
            return stringList == null || stringList.stream().allMatch(String::isBlank);
        }
    }
}
