package gudusoft.gsqlparser.stmt.mssql;

import gudusoft.gsqlparser.*;
import gudusoft.gsqlparser.compiler.TFrame;
import gudusoft.gsqlparser.compiler.TVariable;
import gudusoft.gsqlparser.nodes.*;
import gudusoft.gsqlparser.stmt.TSelectSqlStatement;

import java.util.ArrayList;


public class TMssqlDeclare extends TCustomSqlStatement {

    /**
     * with return only clause
     *
     * @return
     */
    public boolean isWithReturnOnly() {
        return withReturnOnly;
    }

    private boolean withReturnOnly = false;

     public TMssqlDeclare (EDbVendor dbvendor){
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sstmssqldeclare ;
        }

    private TSelectSqlStatement subquery = null;

    void buildsql() {
    }

    void clear() {
    }

    String getasprettytext() {
        return "";
    }

    void iterate(TVisitorAbs pvisitor) {
    }

    private TObjectName cursorName = null;
    private EDeclareType declareType = EDeclareType.variable;

    public TObjectName getCursorName() {
        return cursorName;
    }

    /**
     *
     * @return   EDeclareType.variable or EDeclareType.cursor
     */
    public EDeclareType getDeclareType() {
        return declareType;
    }

    public TSelectSqlStatement getSubquery() {
        return subquery;
    }

    public TDeclareVariableList getVariables() {
        return variables;
    }

    private TDeclareVariableList variables = null;

    protected TStatementList bodyStatements = null;

    public TStatementList getBodyStatements() {
        if (this.bodyStatements == null){
            this.bodyStatements = new TStatementList();
        }
        return bodyStatements;
    }

    private TConstant stateValue;
    private TConstant errorCode;

    public TConstant getStateValue() {
        return stateValue;
    }

    public TConstant getErrorCode() {
        return errorCode;
    }

    private TObjectName conditionName = null;//mysql

    public TObjectName getConditionName() {
        return conditionName;
    }

    private TBlockSqlNode handleBlock;

    public TBlockSqlNode getHandleBlock() {
        return handleBlock;
    }

    private ArrayList<THandlerForCondition> handlerForConditions;

    public ArrayList<THandlerForCondition> getHandlerForConditions() {
        return handlerForConditions;
    }

    public int doParseStatement(TCustomSqlStatement psql) {
        if (rootNode == null) return -1;
        TDeclareSqlNode declareSqlNode = (TDeclareSqlNode)rootNode;

        super.doParseStatement(psql);
        this.declareType = declareSqlNode.getDeclareType();
        this.cursorName = declareSqlNode.getCursorName();
        this.conditionName = declareSqlNode.getConditionName();
        this.stateValue = declareSqlNode.getStateValue();
        this.errorCode = declareSqlNode.getErrorCode();

        if (this.declareType == EDeclareType.variable){
            this.variables = declareSqlNode.getVariables();
            for(int i=0;i<variables.size();i++){
                variables.getDeclareVariable(i).doParse(this,ESqlClause.unknown);
                if (variables.getDeclareVariable(i).getVariableName() != null){
                    TFrame stackFrame = getFrameStack().peek();
                    TDeclareVariable declareVariable = variables.getDeclareVariable(i);
                    TVariable v = new TVariable(declareVariable.getVariableName(),this);
                    if (declareVariable.getDefaultValue() != null){
                        declareVariable.getDefaultValue().evaluate(this.getFrameStack(),this);
                        v.setVariableStr(declareVariable.getDefaultValue().getPlainText());
                        //System.out.println(declareVariable.getDefaultValue().getPlainText());
                    }
                    stackFrame.getScope().addSymbol(v);
                }
            }
        }

        this.handlerForConditions = declareSqlNode.getHandlerForConditions();
        
        if(declareSqlNode.getSelectSqlNode() != null){
            subquery = new TSelectSqlStatement(this.dbvendor);
            subquery.rootNode = declareSqlNode.getSelectSqlNode();
            subquery.doParseStatement(this);
        }

        if (declareSqlNode.getHandleStmt() != null){
            declareSqlNode.getHandleStmt().doParse(this,ESqlClause.unknown);
            this.getBodyStatements().add(declareSqlNode.getHandleStmt().getStmt());
        }
        else if (declareSqlNode.getHandlerBlock() != null){
            handleBlock = declareSqlNode.getHandlerBlock();
            handleBlock.doParse(this,ESqlClause.unknown);
//            declareSqlNode.getHandlerBlock().getStmts().doParse(this,ESqlClause.unknown);
//
//            for(int i=0;i<declareSqlNode.getHandlerBlock().getStmts().size();i++){
//                this.getBodyStatements().add(declareSqlNode.getHandlerBlock().getStmts().getStatementSqlNode(i).getStmt());
//            }
        }else if (declareSqlNode.getStmtSqlNode() != null){
            //teradata declare handler for sqlblock/sqlstmt
            declareSqlNode.getStmtSqlNode().doParse(this,ESqlClause.unknown);
            this.getBodyStatements().add(declareSqlNode.getStmtSqlNode().getStmt());
        }

        withReturnOnly = declareSqlNode.isWithReturnOnly();

        return 0;
    }

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

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

        if (this.cursorName != null){
            this.cursorName.acceptChildren(v);
        }

        if (this.declareType == EDeclareType.variable){
            this.variables.acceptChildren(v);
        }


        if (subquery != null) subquery.acceptChildren(v);
        v.postVisit(this);
    }

    public void setSubquery(TSelectSqlStatement subquery) {
        this.subquery = subquery;
    }

    public void setCursorName(TObjectName cursorName) {
        this.cursorName = cursorName;
    }

    public void setDeclareType(EDeclareType declareType) {
        this.declareType = declareType;
    }

    public void setVariables(TDeclareVariableList variables) {
        this.variables = variables;
    }
}