package gudusoft.gsqlparser.nodes.starrocks;

import gudusoft.gsqlparser.TSourceToken;
import gudusoft.gsqlparser.nodes.*;

/**
 * AST node representing a pipe operator operation in StarRocks pipe query syntax.
 *
 * The pipe operator (|>) chains SQL operations together in a linear flow:
 * FROM table |> SELECT * |> WHERE cond |> ORDER BY col |> LIMIT n
 *
 * Each TPipeOperator represents one operation after a |> symbol.
 */
public class TPipeOperator extends TParseTreeNode {

    /**
     * Enum for different types of pipe operations
     */
    public enum EPipeOperatorType {
        SELECT,       // |> SELECT columns
        WHERE,        // |> WHERE condition
        ORDER_BY,     // |> ORDER BY columns
        LIMIT,        // |> LIMIT n [OFFSET m]
        AGGREGATE,    // |> AGGREGATE expr GROUP BY ...
        EXTEND,       // |> EXTEND expr AS alias
        SET,          // |> SET col = expr
        DROP,         // |> DROP col
        AS,           // |> AS alias
        JOIN,         // |> JOIN table ON cond
        UNION,        // |> UNION [ALL] (query)
        INTERSECT,    // |> INTERSECT [ALL] (query)
        EXCEPT,       // |> EXCEPT [ALL] (query)
        TABLESAMPLE,  // |> TABLESAMPLE
        PIVOT,        // |> PIVOT
        UNPIVOT       // |> UNPIVOT
    }

    private TSourceToken pipeToken;         // The |> token
    private EPipeOperatorType operatorType;

    // For SELECT operation
    private TResultColumnList selectList;
    private TSelectDistinct selectDistinct;

    // For WHERE operation
    private TExpression whereCondition;

    // For ORDER BY operation
    private TOrderBy orderByClause;

    // For LIMIT operation
    private TLimitClause limitClause;

    // For AGGREGATE operation
    private TResultColumnList aggregateList;
    private TGroupBy groupByClause;

    // For EXTEND/SET operations (expression list)
    private TResultColumnList extendList;

    // For DROP operation (column list)
    private TObjectNameList dropColumns;

    // For AS operation (table alias)
    private TObjectName tableAlias;

    // For JOIN operations
    private TJoinExpr joinExpr;
    private TFromTable joinTable;
    private TExpression joinCondition;
    private gudusoft.gsqlparser.EJoinType joinType;

    // For set operations (UNION/INTERSECT/EXCEPT)
    private TParseTreeNode setOperand;
    private boolean setAll;

    // Generic operator keyword token (SELECT, WHERE, etc.)
    private TSourceToken operatorKeyword;

    // Getters and setters
    public TSourceToken getPipeToken() {
        return pipeToken;
    }

    public void setPipeToken(TSourceToken pipeToken) {
        this.pipeToken = pipeToken;
    }

    public EPipeOperatorType getOperatorType() {
        return operatorType;
    }

    public void setOperatorType(EPipeOperatorType operatorType) {
        this.operatorType = operatorType;
    }

    public TSourceToken getOperatorKeyword() {
        return operatorKeyword;
    }

    public void setOperatorKeyword(TSourceToken operatorKeyword) {
        this.operatorKeyword = operatorKeyword;
    }

    public TResultColumnList getSelectList() {
        return selectList;
    }

    public void setSelectList(TResultColumnList selectList) {
        this.selectList = selectList;
    }

    public TSelectDistinct getSelectDistinct() {
        return selectDistinct;
    }

    public void setSelectDistinct(TSelectDistinct selectDistinct) {
        this.selectDistinct = selectDistinct;
    }

    public TExpression getWhereCondition() {
        return whereCondition;
    }

    public void setWhereCondition(TExpression whereCondition) {
        this.whereCondition = whereCondition;
    }

    public TOrderBy getOrderByClause() {
        return orderByClause;
    }

    public void setOrderByClause(TOrderBy orderByClause) {
        this.orderByClause = orderByClause;
    }

    public TLimitClause getLimitClause() {
        return limitClause;
    }

    public void setLimitClause(TLimitClause limitClause) {
        this.limitClause = limitClause;
    }

    public TResultColumnList getAggregateList() {
        return aggregateList;
    }

    public void setAggregateList(TResultColumnList aggregateList) {
        this.aggregateList = aggregateList;
    }

    public TGroupBy getGroupByClause() {
        return groupByClause;
    }

    public void setGroupByClause(TGroupBy groupByClause) {
        this.groupByClause = groupByClause;
    }

    public TResultColumnList getExtendList() {
        return extendList;
    }

    public void setExtendList(TResultColumnList extendList) {
        this.extendList = extendList;
    }

    public TObjectNameList getDropColumns() {
        return dropColumns;
    }

    public void setDropColumns(TObjectNameList dropColumns) {
        this.dropColumns = dropColumns;
    }

    public TObjectName getTableAlias() {
        return tableAlias;
    }

    public void setTableAlias(TObjectName tableAlias) {
        this.tableAlias = tableAlias;
    }

    public TJoinExpr getJoinExpr() {
        return joinExpr;
    }

    public void setJoinExpr(TJoinExpr joinExpr) {
        this.joinExpr = joinExpr;
    }

    public TFromTable getJoinTable() {
        return joinTable;
    }

    public void setJoinTable(TFromTable joinTable) {
        this.joinTable = joinTable;
    }

    public TExpression getJoinCondition() {
        return joinCondition;
    }

    public void setJoinCondition(TExpression joinCondition) {
        this.joinCondition = joinCondition;
    }

    public gudusoft.gsqlparser.EJoinType getJoinType() {
        return joinType;
    }

    public void setJoinType(gudusoft.gsqlparser.EJoinType joinType) {
        this.joinType = joinType;
    }

    public TParseTreeNode getSetOperand() {
        return setOperand;
    }

    public void setSetOperand(TParseTreeNode setOperand) {
        this.setOperand = setOperand;
    }

    public boolean isSetAll() {
        return setAll;
    }

    public void setSetAll(boolean setAll) {
        this.setAll = setAll;
    }

    /**
     * Initialize for SELECT pipe operation
     */
    public void initSelect(TSourceToken pipeToken, TSourceToken selectKeyword,
                           TSelectDistinct distinct, TResultColumnList columns) {
        this.pipeToken = pipeToken;
        this.operatorKeyword = selectKeyword;
        this.operatorType = EPipeOperatorType.SELECT;
        this.selectDistinct = distinct;
        this.selectList = columns;
    }

    /**
     * Initialize for WHERE pipe operation
     */
    public void initWhere(TSourceToken pipeToken, TSourceToken whereKeyword, TExpression condition) {
        this.pipeToken = pipeToken;
        this.operatorKeyword = whereKeyword;
        this.operatorType = EPipeOperatorType.WHERE;
        this.whereCondition = condition;
    }

    /**
     * Initialize for ORDER BY pipe operation
     */
    public void initOrderBy(TSourceToken pipeToken, TOrderBy orderBy) {
        this.pipeToken = pipeToken;
        this.operatorType = EPipeOperatorType.ORDER_BY;
        this.orderByClause = orderBy;
    }

    /**
     * Initialize for LIMIT pipe operation
     */
    public void initLimit(TSourceToken pipeToken, TLimitClause limit) {
        this.pipeToken = pipeToken;
        this.operatorType = EPipeOperatorType.LIMIT;
        this.limitClause = limit;
    }

    /**
     * Initialize for AGGREGATE pipe operation
     */
    public void initAggregate(TSourceToken pipeToken, TSourceToken aggKeyword,
                              TResultColumnList aggregates, TGroupBy groupBy) {
        this.pipeToken = pipeToken;
        this.operatorKeyword = aggKeyword;
        this.operatorType = EPipeOperatorType.AGGREGATE;
        this.aggregateList = aggregates;
        this.groupByClause = groupBy;
    }

    /**
     * Initialize for EXTEND pipe operation
     */
    public void initExtend(TSourceToken pipeToken, TSourceToken extendKeyword,
                           TResultColumnList extendColumns) {
        this.pipeToken = pipeToken;
        this.operatorKeyword = extendKeyword;
        this.operatorType = EPipeOperatorType.EXTEND;
        this.extendList = extendColumns;
    }

    /**
     * Initialize for JOIN pipe operation
     */
    public void initJoin(TSourceToken pipeToken, TJoinExpr join) {
        this.pipeToken = pipeToken;
        this.operatorType = EPipeOperatorType.JOIN;
        this.joinExpr = join;
    }

    /**
     * Initialize for JOIN pipe operation with table and condition
     */
    public void initJoin(TSourceToken pipeToken, gudusoft.gsqlparser.EJoinType joinType,
                         TFromTable table, TExpression condition) {
        this.pipeToken = pipeToken;
        this.operatorType = EPipeOperatorType.JOIN;
        this.joinType = joinType;
        this.joinTable = table;
        this.joinCondition = condition;
    }

    /**
     * Initialize for UNION pipe operation
     */
    public void initUnion(TSourceToken pipeToken, TSourceToken unionKeyword,
                          TParseTreeNode operand, boolean all) {
        this.pipeToken = pipeToken;
        this.operatorKeyword = unionKeyword;
        this.operatorType = EPipeOperatorType.UNION;
        this.setOperand = operand;
        this.setAll = all;
    }

    /**
     * Initialize for INTERSECT pipe operation
     */
    public void initIntersect(TSourceToken pipeToken, TSourceToken intersectKeyword,
                              TParseTreeNode operand, boolean all) {
        this.pipeToken = pipeToken;
        this.operatorKeyword = intersectKeyword;
        this.operatorType = EPipeOperatorType.INTERSECT;
        this.setOperand = operand;
        this.setAll = all;
    }

    /**
     * Initialize for EXCEPT pipe operation
     */
    public void initExcept(TSourceToken pipeToken, TSourceToken exceptKeyword,
                           TParseTreeNode operand, boolean all) {
        this.pipeToken = pipeToken;
        this.operatorKeyword = exceptKeyword;
        this.operatorType = EPipeOperatorType.EXCEPT;
        this.setOperand = operand;
        this.setAll = all;
    }
}