package gudusoft.gsqlparser.stmt.mssql;

import gudusoft.gsqlparser.*;
import gudusoft.gsqlparser.nodes.*;
import gudusoft.gsqlparser.nodes.mssql.TMssqlCreateTriggerUpdateColumn;
import gudusoft.gsqlparser.nodes.mssql.TMssqlCreateTriggerUpdateColumnList;

import java.util.Locale;

/**
 * This class represents if and while SQL statement of SQL Server.
 */
public class TMssqlIfElse extends TCustomSqlStatement {
     public TMssqlIfElse (EDbVendor dbvendor){
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sstmssqlif ;
        }

    void buildsql() {
    }

    void clear() {
    }

    String getasprettytext() {
        return "";
    }

    void iterate(TVisitorAbs pvisitor) {
    }

    private int stmtType = TBaseType.condition_statement_if;
    private TExpression condition = null;
    private TMssqlCreateTriggerUpdateColumnList updateColumnList = null;
    private TCustomSqlStatement stmt = null;

    public TCustomSqlStatement getElseStmt() {
        return elseStmt;
    }

    public TCustomSqlStatement getStmt() {
        return stmt;
    }

    private TCustomSqlStatement elseStmt = null;
    

    public TExpression getCondition() {
        return condition;
    }

    public TMssqlCreateTriggerUpdateColumnList getUpdateColumnList() {
        return updateColumnList;
    }

    /**
     * if or while SQL statement this class represents for.
     * @return
     */
    public int getStmtType() {
        return stmtType;
    }

    public int doParseStatement(TCustomSqlStatement psql) {
        if (rootNode == null) return -1;
        TIfSqlNode ifSqlNode = (TIfSqlNode)rootNode;
        super.doParseStatement(psql);
        this.stmtType = ifSqlNode.getStmtType();
        this.condition = ifSqlNode.getCondition();
        if (this.condition != null){
            this.condition.doParse(this,ESqlClause.unknown);
        }
        this.updateColumnList = ifSqlNode.getUpdateColumnList();
        if (updateColumnList != null){
            TCustomSqlStatement ancestor = psql == null ? null : psql.getAncestorStmt();
            for(int i = 0 ;i<updateColumnList.size();i++){
                if ((ancestor != null)
                        && (ancestor.sqlstatementtype == ESqlStatementType.sstcreatetrigger)
                        && (ancestor.tables.size() > 0)){
                    TMssqlCreateTriggerUpdateColumn updateColumn = updateColumnList.getCreateTriggerUpdateColumn(i);
                    if (updateColumn.getColumnName() != null){
                        ancestor.tables.getTable(0).getLinkedColumns().addObjectName(updateColumn.getColumnName());
                        updateColumn.getColumnName().setSourceTable(ancestor.tables.getTable(0));
                    }else if (updateColumn.getCondition() != null){
                        // UPDATE([ProductID2]) OR UPDATE([OrderQty2])
                        TExpression condition  = updateColumn.getCondition();
                        if( (condition.toString().toUpperCase().startsWith("UPDATE")) && (condition.getExpressionType() == EExpressionType.function_t)){
                            TFunctionCall f =  condition.getFunctionCall();
                            TObjectName name = f.getArgs().getExpression(0).getObjectOperand();

                            ancestor.tables.getTable(0).getLinkedColumns().addObjectName(name);
                            name.setSourceTable(ancestor.tables.getTable(0));
                        }
                    }
                }
                //psql.tables.getTable(0)
            }
        }

        TStatementSqlNode stmtSqlNode = ifSqlNode.getStmtSqlNode();
        if (stmtSqlNode == null) return reportMissingStatementBody(ifSqlNode);
        stmtSqlNode.doParse(this,ESqlClause.unknown);
        stmt = stmtSqlNode.getStmt();
        if ((stmt == null) || (stmt.getStartToken() == null) || (stmt.getEndToken() == null)) {
            return reportMissingStatementBody(ifSqlNode);
        }
        if (this.getEndToken() == null){
            this.setEndToken(stmt.getEndToken());
        }

        if (ifSqlNode.getElseStmtSqlNode() != null){
            ifSqlNode.getElseStmtSqlNode().doParse(this,ESqlClause.unknown);
            elseStmt = ifSqlNode.getElseStmtSqlNode().getStmt();
            if ((elseStmt == null) || (elseStmt.getStartToken() == null)
                    || (elseStmt.getEndToken() == null)) {
                return reportMissingStatementBody(ifSqlNode);
            }
            if (this.getEndToken() == null){
                this.setEndToken(elseStmt.getEndToken());
            }
        }
        
        return 0;
    }

    private int reportMissingStatementBody(TIfSqlNode ifSqlNode) {
        TSourceToken errorToken = ifSqlNode.getEndToken();
        if (errorToken != null) {
            parseerrormessagehandle(new TSyntaxError(errorToken,
                    "missing IF/WHILE statement body", EErrorType.sperror,
                    TBaseType.MSG_ERROR_SYNTAX_ERROR, this));
        } else {
            parseerrormessagehandle(new TSyntaxError("end of input", 0, 0,
                    "missing IF/WHILE statement body", EErrorType.sperror,
                    TBaseType.MSG_ERROR_SYNTAX_ERROR, this, -1));
        }
        return TBaseType.MSG_ERROR_SYNTAX_ERROR;
    }

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

    public void acceptChildren(TParseTreeVisitor v){
        v.preVisit(this);
        if (this.condition != null) condition.acceptChildren(v);
        // Visit UPDATE() function arguments in trigger IF statements
        // These contain column references that need to be resolved
        if (this.updateColumnList != null) {
            updateColumnList.acceptChildren(v);
        }
        if (stmt != null) stmt.acceptChildren(v);
        if (elseStmt != null) elseStmt.acceptChildren(v);
        v.postVisit(this);
    }

    public void setStmtType(int stmtType) {
        this.stmtType = stmtType;
    }

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

    public void setUpdateColumnList(TMssqlCreateTriggerUpdateColumnList updateColumnList) {
        this.updateColumnList = updateColumnList;
    }

    public void setStmt(TCustomSqlStatement stmt) {
        this.stmt = stmt;
    }

    public void setElseStmt(TCustomSqlStatement elseStmt) {
        this.elseStmt = elseStmt;
    }
}