package gudusoft.gsqlparser.stmt;

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

/**
 * The IF statement executes or skips a sequence of statements, depending on the value
 * of a Boolean expression
 */

public class TIfStmt extends TCustomSqlStatement {
    public TIfStmt(){
        this(EDbVendor.dbvoracle);
    }

     public TIfStmt(EDbVendor dbvendor){
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sst_ifstmt;
        }

    void buildsql() {
    }

    void clear() {
    }

    String getasprettytext() {
        return "";
    }

    void iterate(TVisitorAbs pvisitor) {
    }

    /**
     * @return If and only if the value of this expression is TRUE,
     * the statements following THEN execute
     */
    public TExpression getCondition() {
        return condition;
    }

    private TExpression condition = null;
    private TStatementListSqlNode thenStmts = null;
    private TStatementListSqlNode elseStmts = null;
    private TStatementListSqlNode elsifStmts = null;

    /**
     *
     * @return elseif statement list, type of {@link TElsifStmt}.
     */
    public TStatementList getElseifStatements() {
        if (elseifStatements == null){
          elseifStatements = new  TStatementList();
        }
        return elseifStatements;
    }

    private TStatementList elseifStatements = null;

    private TStatementList thenStatements = null;

    /**
     * 
     * @return the statements after the THEN keyword
     */
    public TStatementList getThenStatements() {
        if (thenStatements == null){
          thenStatements = new  TStatementList();  
        }
        return thenStatements;
    }

    /**
     * 
     * @return   the statements after the ELSE keyword
     */
    public TStatementList getElseStatements() {
        if (elseStatements == null){
          elseStatements = new  TStatementList();  
        }

        return elseStatements;
    }

    private TStatementList elseStatements = null;


    public void setElsifStmts(TStatementListSqlNode elsifStmts) {
        this.elsifStmts = elsifStmts;
    }


    public void init(Object arg1,Object arg2,Object arg3)
    {
        condition = (TExpression)arg1;
        thenStmts = (TStatementListSqlNode)arg2;
        elseStmts = (TStatementListSqlNode)arg3;
    }

    private TElseIfSqlNodeList elseIfList = null;

    public int doParseStatement(TCustomSqlStatement psql) {
        super.doParseStatement(psql);

        switch (dbvendor){
            case dbvbigquery:
            case dbvdb2:
            case dbvhana:
                TIfSqlNode ifSqlNode = (TIfSqlNode)rootNode;
                this.condition = ifSqlNode.getCondition();
                this.condition.doParse(this,ESqlClause.unknown);
                if (ifSqlNode.getThenStmts() != null){
                    ifSqlNode.getThenStmts().doParse(this,ESqlClause.unknown);
                    for(int i=0;i<ifSqlNode.getThenStmts().size();i++){
                        this.getThenStatements().add(ifSqlNode.getThenStmts().getStatementSqlNode(i).getStmt());
                    }
                }

                if (ifSqlNode.getElseIfList() != null){
                    ifSqlNode.getElseIfList().doParse(this,ESqlClause.unknown);
                    this.elseIfList = ifSqlNode.getElseIfList();
                }

                if (ifSqlNode.getElsifStmts() != null){
                    ifSqlNode.getElsifStmts().doParse(this,ESqlClause.unknown);
                    for(int i=0;i<ifSqlNode.getElsifStmts().size();i++){
                        this.getElseifStatements().add(ifSqlNode.getElsifStmts().getStatementSqlNode(i).getStmt());
                    }
                }

                if (ifSqlNode.getElseStmts() != null){
                    ifSqlNode.getElseStmts().doParse(this,ESqlClause.unknown);
                    for(int i=0;i<ifSqlNode.getElseStmts().size();i++){
                        this.getElseStatements().add(ifSqlNode.getElseStmts().getStatementSqlNode(i).getStmt());
                    }
                }
                break;
            default:
                condition.doParse(this,ESqlClause.unknown);
                if (thenStmts != null){
                    thenStmts.doParse(this,ESqlClause.unknown);
                    for(int i=0;i<thenStmts.size();i++){
                        this.getThenStatements().add(thenStmts.getStatementSqlNode(i).getStmt());
                    }
                }

                if (elsifStmts != null){
                    elsifStmts.doParse(this,ESqlClause.unknown);
                    for(int i=0;i<elsifStmts.size();i++){
                        this.getElseifStatements().add(elsifStmts.getStatementSqlNode(i).getStmt());
                    }
                }

                if (elseStmts != null){
                    elseStmts.doParse(this,ESqlClause.unknown);
                    for(int i=0;i<elseStmts.size();i++){
                        this.getElseStatements().add(elseStmts.getStatementSqlNode(i).getStmt());
                    }
                }
                break;
        }



        return 0;
    }

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

    public void acceptChildren(TParseTreeVisitor v){
        v.preVisit(this);
        condition.acceptChildren(v);
        //if (thenStmts != null) thenStmts.acceptChildren(v);
        getThenStatements().acceptChildren(v);
        if (elseifStatements != null) elseifStatements.acceptChildren(v);
        if (elseStatements != null) elseStatements.acceptChildren(v);
        v.postVisit(this);
    }

    public void setCondition(TExpression condition) {
        this.condition = condition;
    }

    public void setElseifStatements(TStatementList elseifStatements) {
        this.elseifStatements = elseifStatements;
    }

    public void setThenStatements(TStatementList thenStatements) {
        this.thenStatements = thenStatements;
    }

    public void setElseStatements(TStatementList elseStatements) {
        this.elseStatements = elseStatements;
    }
}