D-Net

package eu.dnetlib.data.search.utils.cql;

import java.util.List;

/**

 * Created by kiatrop on 28/9/2016.

 */

public class CQLQueryBuilder {

    public enum Operator {

        AND("and"),

        OR("or"),

        NOT("not"),

        EQUAL("="),

        EXACT("exact"),

        NOTEQUAL("<>");

        private String value;

        Operator(String value) {

            this.value = value;

        }

        private String getValue() {

            return value;

        }

        @Override

        public String toString() {

            return this.getValue();

        }

    }

    public static void appendKeywords(StringBuilder stringBuilder, String keywords) {

        if (keywords != null && !keywords.trim().isEmpty()) {

            appendSimpleTerm(stringBuilder, CQLQueryBuilder.Operator.AND, keywords);

        }

    }

    public static void appendSimpleTerm(StringBuilder stringBuilder, Operator queryOperator, String value) {

        stringBuilder.append(' ').append(queryOperator).append(" (").append(value).append(')');

    }

    public static void appendSimpleQuotedTerm(StringBuilder stringBuilder, Operator queryOperator, String value) {

        stringBuilder.append(' ').append(queryOperator).append(" (\"").append(value).append("\")");

    }

    public static void appendFieldTerm(StringBuilder stringBuilder, Operator queryOperator, String indexFieldName, Operator termOperator, String fieldValue) {

            stringBuilder.append(' ').append(queryOperator).

                    append(" (").append(indexFieldName).append(' ').append(termOperator).append(' ').append(fieldValue).append(")");

    }

    public static void appendFieldQuotedTerm(StringBuilder stringBuilder, Operator queryOperator, String indexFieldName, Operator termOperator, String fieldValue) {

            stringBuilder.append(' ').append(queryOperator).

                    append(" (").append(indexFieldName).append(' ').append(termOperator).append(" \"").append(fieldValue).append("\")");

    }

    public static void appendFieldTerms(StringBuilder stringBuilder, Operator queryOperator, String indexFieldName, Operator termOperator, List<String> fieldValues) {

        for (String fieldValue: fieldValues) {

            appendFieldTerm(stringBuilder, queryOperator, indexFieldName, termOperator, fieldValue);

        }

    }

    public static void appendFieldQuotedTerms(StringBuilder stringBuilder, Operator queryOperator, String indexFieldName, Operator termOperator, List<String> fieldValues) {

        for (String fieldValue: fieldValues) {

            appendFieldQuotedTerm(stringBuilder, queryOperator, indexFieldName, termOperator, fieldValue);

        }

    }

}