package gudusoft.gsqlparser.stmt;


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

import java.util.EnumSet;

/**
 * Create trigger statement.
 *
 * supported databases:
 * <ul>
 *     <li>DB2</li>
 *     <li>Hana</li>
 *     <li>Greenplum, represented by {@link gudusoft.gsqlparser.stmt.oracle.TPlsqlCreateTrigger}</li>
 *     <li>Informix</li>
 *     <li>MySQL</li>
 *     <li>Openedge</li>
 *     <li>Oracle, represented by {@link gudusoft.gsqlparser.stmt.oracle.TPlsqlCreateTrigger}
 *     support dml, non-dml and database trigger.
 *     </li>
 *     <li>PostgreSQL</li>
 *     <li>SQL Server,
 *     support dml, ddl trigger
 *     </li>
 *     <li>Sybase</li>
 *     <li>Teradata</li>
 * </ul>
 *
 * @see TTriggeringClause
 */
public class TCreateTriggerStmt extends TStoredProcedureSqlStatement {
    public void setTriggeringClause(TTriggeringClause triggeringClause) {
        this.triggeringClause = triggeringClause;
    }

    public TCreateTriggerStmt (EDbVendor dbvendor){
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sstcreatetrigger ;
    }


    void buildsql() {
    }

    void clear() {
    }

    String getasprettytext() {
        return "";
    }

    void iterate(TVisitorAbs pvisitor) {
    }

    private TExpression whenCondition = null;

    /**
     * When condition
     * @return when condition
     */
    public TExpression getWhenCondition() {
        return whenCondition;
    }

    private boolean alterTrigger;

    public void setAlterTrigger(boolean alterTrigger) {
        this.alterTrigger = alterTrigger;
    }

    public boolean isAlterTrigger() {

        return alterTrigger;
    }
    @Override
    public TObjectName getStoredProcedureName(){
        return triggerName;
    }

    private TTable onTable = null;

    /**
     * @deprecated As of v2.0.1.0,  use {@link #getTriggeringClause()} instead
     *
     * @return onTable
     */
    public TTable getOnTable() {
        return onTable;
    }


    public TObjectName getTriggerName() {
        return triggerName;
    }

    private TObjectName triggerName = null;

    public void setTimingPoint(ETriggerTimingPoint timingPoint) {
        this.timingPoint = timingPoint;
    }

    public void setDmlTypes(EnumSet<ETriggerDmlType> dmlTypes) {
        this.dmlTypes = dmlTypes;
    }

    /**
     * @deprecated As of v2.0.1.0, replaced by {@link #getTriggeringClause()}
     * @return timing point
     */
    public ETriggerTimingPoint getTimingPoint() {

        return timingPoint;
    }

    /**
     * @deprecated As of v2.0.1.0, replaced by {@link #getTriggeringClause()}
     * @return dml
     */
    public EnumSet<ETriggerDmlType> getDmlTypes() {
        return dmlTypes;
    }

    private TFunctionCall functionCall;//postgresql

    public void setFunctionCall(TFunctionCall functionCall) {
        this.functionCall = functionCall;
    }

    /**
     * Greenplum,postgresql executed function
     *
     * @return function call
     */
    public TFunctionCall getFunctionCall() {

        return functionCall;
    }

    public int doParseStatement(TCustomSqlStatement psql) {
        if (rootNode == null) return -1;
        if(!(rootNode instanceof TCreateTriggerSqlNode)) return -1;
        TCreateTriggerSqlNode createTriggerNode = (TCreateTriggerSqlNode)rootNode;

        super.doParseStatement(psql);
        this.triggerName = createTriggerNode.getTriggerName();
        if (getSqlEnv().getDefaultCatalogName() != null){
            if (triggerName.getDatabaseToken() == null){
                triggerName.setDatabaseToken(new TSourceToken(getSqlEnv().getDefaultCatalogName()));
            }
        }
        alterTrigger = createTriggerNode.isAlterTrigger();
        this.fireMode = createTriggerNode.getFireMode();
        timingPoint = createTriggerNode.getTimingPoint();
        triggeringClause = createTriggerNode.getTriggeringClause();
        triggeringClause.doParse(this,ESqlClause.unknown);
        functionCall = createTriggerNode.getFunctionCall();

        onTable = tables.getTable(0);

        switch (dbvendor){
            case dbvdb2:
                TTriggerAction triggerAction = createTriggerNode.getTriggerAction();
                this.whenCondition = triggerAction.getWhenExpr();

                if (triggerAction.getStmtNode() != null){
                    triggerAction.getStmtNode().doParse(this,ESqlClause.unknown);
                    this.getBodyStatements().add(triggerAction.getStmtNode().getStmt());
                }else if (triggerAction.getCompoundSqlNode() != null){
                    TCompoundSqlNode compoundSqlNode = triggerAction.getCompoundSqlNode();
                    if (compoundSqlNode.getDeclareStmts() != null){
                        compoundSqlNode.getDeclareStmts().doParse(this,ESqlClause.unknown);

                        // push variable declare into symbolTable, and add to declareStatements
                        for(int i=0;i<compoundSqlNode.getDeclareStmts().size();i++){
                            this.getTopStatement().getSymbolTable().push( new TSymbolTableItem(TObjectName.ttobjVariable,this,compoundSqlNode.getDeclareStmts().getStatementSqlNode(i).getStmt() ));
                            this.getDeclareStatements().add(compoundSqlNode.getDeclareStmts().getStatementSqlNode(i).getStmt());
                        }
                    }

                    if (compoundSqlNode.getStmts() != null){
                        compoundSqlNode.getStmts().doParse(this,ESqlClause.unknown);

                        for(int i= 0; i<compoundSqlNode.getStmts().size();i++){
                            this.getBodyStatements().add(compoundSqlNode.getStmts().getStatementSqlNode(i).getStmt());
                        }
                    }

                    if (compoundSqlNode.getDeclareStmts() != null){
                        // pop variable declare from symbolTable
                        for(int i=0;i<compoundSqlNode.getDeclareStmts().size();i++){
                            this.getTopStatement().getSymbolTable().pop();
                        }
                    }
                }
                break;
            case dbvmysql:
                if (createTriggerNode.getStmt() != null){
                    createTriggerNode.getStmt().doParse(this,ESqlClause.unknown);
                    this.getBodyStatements().add(createTriggerNode.getStmt().getStmt());
                }
                else if (createTriggerNode.getBlcok() != null){
                    if (createTriggerNode.getBlcok().getStmts() != null){
                        createTriggerNode.getBlcok().getStmts().doParse(this,ESqlClause.unknown);

                        for(int i=0;i<createTriggerNode.getBlcok().getStmts().size();i++){
                            this.getBodyStatements().add(createTriggerNode.getBlcok().getStmts().getStatementSqlNode(i).getStmt());
                        }

                    }
                }
                break;
            default:
                if (createTriggerNode.getStmts() != null){
                    createTriggerNode.getStmts().doParse(this,ESqlClause.unknown);
                    for(int i=0;i<createTriggerNode.getStmts().size();i++){
                        this.getBodyStatements().add(createTriggerNode.getStmts().getStatementSqlNode(i).getStmt());
                    }
                }
                break;
        }


        return 0;
    }

    private TTriggeringClause triggeringClause;

    public TTriggeringClause getTriggeringClause() {
        return triggeringClause;
    }

    private int fireMode = TBaseType.tfmFor;
    private ETriggerTimingPoint timingPoint = ETriggerTimingPoint.ttpAfter;
    private EnumSet <ETriggerDmlType> dmlTypes = EnumSet.noneOf(ETriggerDmlType.class);

    /**
     * @deprecated As of v1.7.2.5, replaced by {@link gudusoft.gsqlparser.ETriggerDmlType}
     * @return insert/delete/update action
     */
    public TSourceTokenList getDmlTpyes() {
        if (this.dmlTpyes == null){
            this.dmlTpyes = new TSourceTokenList();
        }
        return this.dmlTpyes;
    }

    private TSourceTokenList dmlTpyes = null;

    /**
     * @deprecated As of v1.7.2.5, replaced by {@link gudusoft.gsqlparser.ETriggerTimingPoint}
     * @return trigger timing point
     */
    public int getFireMode() {
        return fireMode;
    }

    public void accept(TParseTreeVisitor v){
        v.preVisit(this);

        v.postVisit(this);
    }
    public void acceptChildren(TParseTreeVisitor v){
        v.preVisit(this);
        if (getBodyStatements().size() > 0) getBodyStatements().acceptChildren(v);
        v.postVisit(this);
    }

    public void setOnTable(TTable onTable) {
        this.onTable = onTable;
    }

    public void setTriggerName(TObjectName triggerName) {
        this.triggerName = triggerName;
    }

    public void setFireMode(int fireMode) {
        this.fireMode = fireMode;
    }
}