/*
 * Copyright (c) 2011, 2021 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0,
 * or the Eclipse Distribution License v. 1.0 which is available at
 * http://www.eclipse.org/org/documents/edl-v10.php.
 *
 * SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause
 */

// Contributors:
//     Oracle - initial API and implementation
//
package org.eclipse.persistence.jpa.jpql.tools.model.query;

import java.io.IOException;
import java.util.List;
import org.eclipse.persistence.jpa.jpql.parser.AbstractSelectStatement;
import org.eclipse.persistence.jpa.jpql.tools.spi.IEntity;
import org.eclipse.persistence.jpa.jpql.utility.iterable.ListIterable;
import static org.eclipse.persistence.jpa.jpql.parser.AbstractExpression.*;

/**
 * This state object represents the select statement, which has at least a <code><b>SELECT</b></code>
 * clause and a <code><b>FROM</b></code> clause.
 *
 * @see SelectStatementStateObject
 * @see FromClauseStateObject
 * @see GroupByClauseStateObject
 * @see HavingClauseStateObject
 * @see SelectClauseStateObject
 * @see WhereClauseStateObject
 *
 * @see AbstractSelectStatement
 *
 * @version 2.5
 * @since 2.4
 * @author Pascal Filion
 */
@SuppressWarnings("nls")
public abstract class AbstractSelectStatementStateObject extends AbstractStateObject {

    /**
     * The state object representing the <code><b>FROM</b></code> clause.
     */
    private AbstractFromClauseStateObject fromClause;

    /**
     * The state object representing the <code><b>GROUP BY</b></code> clause.
     */
    private GroupByClauseStateObject groupByClause;

    /**
     * The state object representing the <code><b>HAVING</b></code> clause.
     */
    private HavingClauseStateObject havingClause;

    /**
     * The state object representing the <code><b>SELECT</b></code> clause.
     */
    private AbstractSelectClauseStateObject selectClause;

    /**
     * The state object representing the <code><b>WHERE</b></code> clause.
     */
    private WhereClauseStateObject whereClause;

    /**
     * Notify the state object representing the <code><b>GROUP BY</b></code> clause has changed.
     */
    public static final String GROUP_BY_CLAUSE_PROPERTY = "groupByClause";

    /**
     * Notify the state object representing the <code><b>HAVING</b></code> clause has changed.
     */
    public static final String HAVING_CLAUSE_PROPERTY = "havingClause";

    /**
     * Notify the state object representing the <code><b>WHERE</b></code> clause has changed.
     */
    public static final String WHERE_CLAUSE_PROPERTY = "whereClause";

    /**
     * Creates a new <code>AbstractSelectStatementStateObject</code>.
     *
     * @param parent The parent of this state object, which cannot be <code>null</code>
     * @exception NullPointerException The given parent cannot be <code>null</code>
     */
    protected AbstractSelectStatementStateObject(StateObject parent) {
        super(parent);
    }

    @Override
    protected void addChildren(List<StateObject> children) {

        super.addChildren(children);
        children.add(selectClause);
        children.add(fromClause);

        if (whereClause != null) {
            children.add(whereClause);
        }

        if (groupByClause != null) {
            children.add(groupByClause);
        }

        if (havingClause != null) {
            children.add(havingClause);
        }
    }

    /**
     * Adds a new collection declaration to the <code><b>FROM</b></code> clause.
     *
     * @return The {@link CollectionMemberDeclarationStateObject} representing the collection
     * declaration
     */
    public CollectionMemberDeclarationStateObject addCollectionDeclaration() {
        return getFromClause().addCollectionDeclaration();
    }

    /**
     * Adds a new collection declaration to the <code><b>FROM</b></code> clause.
     *
     * @param collectionValuedPath The collection-valued path expression
     * @param identificationVariable The variable defining the collection-valued path expression
     * @return The {@link CollectionMemberDeclarationStateObject} representing the collection
     * declaration
     */
    public CollectionMemberDeclarationStateObject addCollectionDeclaration(String collectionValuedPath,
                                                                           String identificationVariable) {

        return getFromClause().addCollectionDeclaration(collectionValuedPath, identificationVariable);
    }

    /**
     * Adds the <code><b>GROUP BY</b></code> clause. The clause is not added if it's already present.
     *
     * @return The {@link GroupByClauseStateObject}
     */
    public GroupByClauseStateObject addGroupByClause() {
        if (!hasGroupByClause()) {
            setGroupByClause(new GroupByClauseStateObject(this));
        }
        return groupByClause;
    }

    /**
     * Adds the <code><b>GROUP BY</b></code> clause and parses the given JPQL fragment. The clause is
     * not added if it's already present.
     *
     * @param jpqlFragment The fragment of the JPQL to parse that represents the group by items, the
     * fragment cannot start with <code><b>GROUP BY</b></code>
     * @return The {@link GroupByClauseStateObject}
     */
    public GroupByClauseStateObject addGroupByClause(String jpqlFragment) {
        GroupByClauseStateObject stateObject = addGroupByClause();
        stateObject.parse(jpqlFragment);
        return stateObject;
    }

    /**
     * Adds the <code><b>HAVING</b></code> clause. The clause is not added if it's already present.
     *
     * @return The {@link GroupByClauseStateObject}
     */
    public HavingClauseStateObject addHavingClause() {
        if (!hasHavingClause()) {
            setHavingClause(new HavingClauseStateObject(this));
        }
        return havingClause;
    }

    /**
     * Adds the <code><b>HAVING</b></code> clause and parses the given JPQL fragment. The clause is
     * not added if it's already present.
     *
     * @param jpqlFragment The fragment of the JPQL to parse that represents the conditional expression,
     * the fragment cannot start with <code><b>HAVING</b></code>
     * @return The {@link HavingClauseStateObject}
     */
    public HavingClauseStateObject addHavingClause(String jpqlFragment) {
        HavingClauseStateObject stateObject = addHavingClause();
        stateObject.parse(jpqlFragment);
        return stateObject;
    }

    /**
     * Adds a new range variable declaration to the <code><b>FROM</b></code> clause.
     *
     * @return The {@link StateObject} representing the new range variable declaration
     */
    public IdentificationVariableDeclarationStateObject addRangeDeclaration() {
        return getFromClause().addRangeDeclaration();
    }

    /**
     * Adds to this select statement a new range variable declaration.
     *
     * @param entity The external form of the entity to add to the declaration list
     * @param identificationVariable The unique identifier identifying the given {@link IEntity}
     * @return The {@link StateObject} representing the new range variable declaration
     */
    public IdentificationVariableDeclarationStateObject addRangeDeclaration(IEntity entity,
                                                                            String identificationVariable) {

        return getFromClause().addRangeDeclaration(entity, identificationVariable);
    }

    /**
     * Adds to this select statement a new range variable declaration.
     *
     * @param entityName The name of the entity
     * @param identificationVariable The unique identifier identifying the entity
     * @return The {@link StateObject} representing the range variable declaration
     */
    public IdentificationVariableDeclarationStateObject addRangeDeclaration(String entityName,
                                                                            String identificationVariable) {

        return getFromClause().addRangeDeclaration(entityName, identificationVariable);
    }

    /**
     * Adds the <code><b>WHERE</b></code> clause. The clause is not added if it's already present.
     *
     * @return The {@link GroupByClauseStateObject}
     */
    public WhereClauseStateObject addWhereClause() {
        if (!hasWhereClause()) {
            setWhereClause(new WhereClauseStateObject(this));
        }
        return whereClause;
    }

    /**
     * Adds the <code><b>WHERE</b></code> clause and parses the given JPQL fragment. The clause is
     * not added if it's already present.
     *
     * @param jpqlFragment The fragment of the JPQL to parse that represents the conditional expression,
     * the fragment cannot start with <code><b>WHERE</b></code>
     * @return The {@link WhereClauseStateObject}
     */
    public WhereClauseStateObject addWhereClause(String jpqlFragment) {
        WhereClauseStateObject stateObject = addWhereClause();
        stateObject.parse(jpqlFragment);
        return stateObject;
    }

    /**
     * Creates the state object representing the <code><b>FROM</b></code> clause.
     *
     * @return A concrete instance of {@link AbstractFromClauseStateObject}
     */
    protected abstract AbstractFromClauseStateObject buildFromClause();

    /**
     * Creates the state object representing the <code><b>SELECT</b></code> clause.
     *
     * @return A concrete instance of {@link AbstractSelectClauseStateObject}
     */
    protected abstract AbstractSelectClauseStateObject buildSelectClause();

    /**
     * Returns the list of {@link VariableDeclarationStateObject} defining the variable declarations,
     * which are mapping an entity to a variable or a collection-valued member to a variable.
     * <p>
     * Example:
     * <ul>
     * <li><code>Employee e</code></li>
     * <li><code>IN (e.employees) AS emps</code></li>
     * </ul>
     *
     * @return The list of {@link VariableDeclarationStateObject}
     */
    public ListIterable<? extends VariableDeclarationStateObject> declarations() {
        return fromClause.items();
    }

    @Override
    public IdentificationVariableStateObject findIdentificationVariable(String identificationVariable) {
        return fromClause.findIdentificationVariable(identificationVariable);
    }

    @Override
    public DeclarationStateObject getDeclaration() {
        return fromClause;
    }

    @Override
    public AbstractSelectStatement getExpression() {
        return (AbstractSelectStatement) super.getExpression();
    }

    /**
     * Returns the state object representing the <code><b>FROM</b></code> clause.
     *
     * @return The state object representing the <code><b>FROM</b></code> clause, which is never
     * <code>null</code>
     */
    public AbstractFromClauseStateObject getFromClause() {
        return fromClause;
    }

    /**
     * Returns the state object representing the <code><b>GROUP BY</b></code> clause.
     *
     * @return Either the actual state object representing the <code><b>GROUP BY</b></code> clause or
     * <code>null</code> if it's not present
     */
    public GroupByClauseStateObject getGroupByClause() {
        return groupByClause;
    }

    /**
     * Returns the state object representing the <code><b>HAVING</b></code> clause.
     *
     * @return Either the actual state object representing the <code><b>HAVING</b></code> clause or
     * <code>null</code> if it's not present
     */
    public HavingClauseStateObject getHavingClause() {
        return havingClause;
    }

    /**
     * Returns the state object representing the <code><b>SELECT</b></code> clause.
     *
     * @return Either the actual state object representing the <code><b>SELECT</b></code> clause,
     * which is never <code>null</code>
     */
    public AbstractSelectClauseStateObject getSelectClause() {
        return selectClause;
    }

    /**
     * Returns the state object representing the <code><b>WHERE</b></code> clause.
     *
     * @return Either the actual state object representing the <code><b>WHERE</b></code> clause or
     * the <code>null</code> state object since <code>null</code> is never returned
     */
    public WhereClauseStateObject getWhereClause() {
        return whereClause;
    }

    /**
     * Returns the state object representing the <code><b>GROUP BY</b></code> clause.
     *
     * @return Either the actual state object representing the <code><b>GROUP BY</b></code> clause or
     * <code>null</code> if it's not present
     */
    public boolean hasGroupByClause() {
        return groupByClause != null;
    }

    /**
     * Returns the state object representing the <code><b>HAVING</b></code> clause.
     *
     * @return Either the actual state object representing the <code><b>HAVING</b></code> clause  or
     * <code>null</code> if it's not present
     */
    public boolean hasHavingClause() {
        return havingClause != null;
    }

    /**
     * Returns the state object representing the <code><b>WHERE</b></code> clause.
     *
     * @return Either the actual state object representing the <code><b>WHERE</b></code> clause or
     * <code>null</code> if it's not present
     */
    public boolean hasWhereClause() {
        return whereClause != null;
    }

    /**
     * Returns the {@link IdentificationVariableStateObject IdentificationVariableStateObjects}
     * holding onto the identification variables, which are the variables defined in the
     * <code><b>FROM</b></code> clause.
     * <p>
     * Example:
     * <ul>
     * <li><code>Employee e</code>; <i>e</i> is returned</li>
     * <li><code>IN (e.employees) AS emps</code>; <i>emps</i> is returned</li>
     * <li><code>Manager m JOIN m.employees emps</code>; <i>m</i> and <i>emps</i> are returned</li>
     * </ul>
     *
     * @return The list of {@link IdentificationVariableStateObject IdentificationVariableStateObjects}
     */
    public Iterable<IdentificationVariableStateObject> identificationVariables() {
        return fromClause.identificationVariables();
    }

    @Override
    protected void initialize() {
        super.initialize();
        selectClause = buildSelectClause();
        fromClause   = buildFromClause();
    }

    @Override
    public boolean isEquivalent(StateObject stateObject) {

        if (super.isEquivalent(stateObject)) {
            AbstractSelectStatementStateObject select = (AbstractSelectStatementStateObject) stateObject;
            return areEquivalent(selectClause,  select.selectClause)  &&
                   areEquivalent(fromClause,    select.fromClause)    &&
                   areEquivalent(fromClause,    select.fromClause)    &&
                   areEquivalent(whereClause,   select.whereClause)   &&
                   areEquivalent(groupByClause, select.groupByClause) &&
                   areEquivalent(havingClause,  select.havingClause);
        }

        return false;
    }

    /**
     * Parses the given JPQL fragment and create the select item. For the top-level query, the
     * fragment can contain several select items but for a subquery, it can represent only one.
     *
     * @param jpqlFragment The portion of the query representing one or several select items
     */
    public void parseSelect(String jpqlFragment) {
        getSelectClause().parse(jpqlFragment);
    }

    /**
     * Removes the <code><b>GROUP BY</b></code> clause.
     */
    public void removeGroupByClause() {
        setGroupByClause(null);
    }

    /**
     * Removes the <code><b>HAVING</b></code> clause.
     */
    public void removeHavingClause() {
        setHavingClause(null);
    }

    /**
     * Removes the <code><b>WHERE</b></code> clause.
     */
    public void removeWhereClause() {
        setWhereClause(null);
    }

    private void setGroupByClause(GroupByClauseStateObject groupByClause) {
        GroupByClauseStateObject oldGroupByClause = this.groupByClause;
        this.groupByClause = groupByClause;
        firePropertyChanged(GROUP_BY_CLAUSE_PROPERTY, oldGroupByClause, groupByClause);
    }

    private void setHavingClause(HavingClauseStateObject havingClause) {
        HavingClauseStateObject oldHavingClause = this.havingClause;
        this.havingClause = havingClause;
        firePropertyChanged(HAVING_CLAUSE_PROPERTY, oldHavingClause, havingClause);
    }

    private void setWhereClause(WhereClauseStateObject whereClause) {
        WhereClauseStateObject oldWhereClause = this.whereClause;
        this.whereClause = whereClause;
        firePropertyChanged(WHERE_CLAUSE_PROPERTY, oldWhereClause, whereClause);
    }

    /**
     * Either adds or removes the state object representing the <code><b>GROUP BY</b></code> clause.
     */
    public void toggleGroupByClause() {
        if (hasGroupByClause()) {
            removeGroupByClause();
        }
        else {
            addGroupByClause();
        }
    }

    /**
     * Either adds or removes the state object representing the <code><b>HAVING</b></code> clause.
     */
    public void toggleHavingClause() {
        if (hasHavingClause()) {
            removeHavingClause();
        }
        else {
            addHavingClause();
        }
    }

    /**
     * Either adds or removes the state object representing the <code><b>WHERE</b></code> clause.
     */
    public void toggleWhereClause() {
        if (hasWhereClause()) {
            removeWhereClause();
        }
        else {
            addWhereClause();
        }
    }

    @Override
    protected void toTextInternal(Appendable writer) throws IOException {

        selectClause.toString(writer);
        writer.append(SPACE);
        fromClause.toString(writer);

        if (whereClause != null) {
            writer.append(SPACE);
            whereClause.toString(writer);
        }

        if (groupByClause != null) {
            writer.append(SPACE);
            groupByClause.toString(writer);
        }

        if (havingClause != null) {
            writer.append(SPACE);
            havingClause.toString(writer);
        }
    }
}