package gudusoft.gsqlparser.stmt;

import gudusoft.gsqlparser.*;
import gudusoft.gsqlparser.compiler.TSymbolTableManager;
import gudusoft.gsqlparser.compiler.TVariable;
import gudusoft.gsqlparser.nodes.*;


/**
 * execute statement
 *
 * db: couchbase, netezza,greenplum,mysql, postgresql,redshift
 *
 * @TODO: 2024/2/7  {@link  gudusoft.gsqlparser.stmt.mssql.TMssqlExecute}, {@link TExecImmeStmt} should merge into this class
 */
public class TExecuteSqlStatement extends TCustomSqlStatement {

    private String preparedSqlText;

    public void setPreparedSqlText(String preparedSqlText) {
        this.preparedSqlText = preparedSqlText;
    }

    public String getPreparedSqlText() {
        return preparedSqlText;
    }

    private EExecType executeType = EExecType.unknown;

    public EExecType getExecuteType() {
        return executeType;
    }

    private TCustomSqlStatement stmt;

    public TCustomSqlStatement getStmt() {
        return stmt;
    }

    public TExecuteSqlStatement(){
        this(EDbVendor.dbvpostgresql);
    }

    public TExecuteSqlStatement(EDbVendor dbvendor) {
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sstExecute;
    }

    public int doParseStatement(TCustomSqlStatement psql) {

        if (rootNode == null) return -1;
        super.doParseStatement(psql);

        TExecuteSqlNode sqlNode = (TExecuteSqlNode)rootNode;
        this.executeType = sqlNode.getExecuteType();

        switch (dbvendor){
            case dbvpostgresql:
                this.stmtString = sqlNode.getStmtString();
                this.intoVariable = sqlNode.getIntoVariable();
                this.usingVariables = sqlNode.getUsingVariables();

                sqlText = stmtString.toString();
                if (stmtString.getExpressionType() == EExpressionType.function_t){
                    if (stmtString.toString().startsWith("format")){
                        //postgresql format function
                        TFunctionCall functionCall = stmtString.getFunctionCall();
                        sqlText = functionCall.getArgs().getExpression(0).toString();
                        sqlText = sqlText.replaceAll("%s","PLACEHOLDER");
                    }
                }
               // System.out.println(sqlText);

                this.moduleName = sqlNode.getModuleName();
                this.statementName = sqlNode.getModuleName();
                this.parameters = sqlNode.getStringValues();

                break;
            case dbvsnowflake:
                this.stmtString = sqlNode.getStmtString();
                //System.out.println(stmtString.toString());
                TSourceToken st = stmtString.getStartToken();
                if (st.toString().startsWith("$$")){
                    TGSqlParser parser = new TGSqlParser(EDbVendor.dbvsnowflake);
                    parser.sqltext = TBaseType.stringBlock((int)st.lineNo - 1,(int)st.columnNo)+ TBaseType.getStringInsideLiteral(st.toString());
                    int ret = parser.parse();
                    if (ret == 0){
                        stmt = parser.sqlstatements.get(0);
                    }else{
                        for(int j=0;j<parser.getErrorCount();j++){
                            this.parseerrormessagehandle(parser.getSyntaxErrors().get(j));
                        }
                    }
                }

                break;
            case dbvnetezza:
                this.moduleName = sqlNode.getModuleName();
                this.statementName = sqlNode.getModuleName();
                this.parameters = sqlNode.getStringValues();
                preparedValue = sqlNode.getPreparedValue();
                if (preparedValue != null){
                    preparedValue.doParse(this, ESqlClause.unknown);
                }

                switch (executeType){
                    case expr:
                        if (sqlNode.getPreparedValue().getExpressionType() == EExpressionType.function_t){
                            TFunctionCall functionCall = sqlNode.getPreparedValue().getFunctionCall();
                            this.moduleName = functionCall.getFunctionName();
                            this.parameters = functionCall.getArgs();
                            this.executeType = EExecType.module;
                        }
                        break;
                    default:
                        break;
                }

                break;
            case dbvredshift:
                this.moduleName = sqlNode.getModuleName();
                this.statementName = sqlNode.getModuleName();
                this.parameters = sqlNode.getStringValues();
                preparedValue = sqlNode.getPreparedValue();
                if (preparedValue != null){
                    preparedValue.doParse(this, ESqlClause.unknown);

                    if (this.preparedValue.getExpressionType() == EExpressionType.simple_object_name_t){
                        TObjectName var = this.preparedValue.getObjectOperand();
                        //  if (var.getDbObjectType() == EDbObjectType.variable){
                        TVariable symbolVariable =  TSymbolTableManager.searchSymbolVariable(this.getFrameStack(),var.toString());
                        if (symbolVariable != null){
                            this.sqlText = symbolVariable.getVariableStr();
                            //System.out.println(this.sqlText);
                        }
                        // }
                    }

                }
                break;
            default:
                this.moduleName = sqlNode.getModuleName();
                this.statementName = sqlNode.getModuleName();
                this.parameters = sqlNode.getStringValues();
                preparedValue = sqlNode.getPreparedValue();
                if (preparedValue != null){
                    preparedValue.doParse(this, ESqlClause.unknown);
                }

                // available in TMssqlExecute currently, maybe will be moved to here in later version
                //this.stringValues = sqlNode.getStringValues();
                //this.linkServerName = sqlNode.getLinkServerName();

                break;
        }
        return 0;
    }

    static TGSqlParser sqlparser = null;
    static {
        // Todo: hardcode to redshift, need to change according to dbvendor
        sqlparser = new TGSqlParser(EDbVendor.dbvredshift);
    }

    private TStatementList dynamicStatements = null;

    /**
     *
     * @return sql statement instance that generated dynamically based on {@link #sqlText}
     */
    public synchronized TStatementList getDynamicStatements() {
        if (this.dynamicStatements != null) return this.dynamicStatements;
        if (this.sqlText == null) return null;


        String query = this.sqlText;
        if((this.preparedValue != null) && (this.preparedValue.getPlainTextLineNo() != -1)){
            long lineNo = this.preparedValue.getPlainTextLineNo();
            long columnNo = this.preparedValue.getPlainTextColumnNo();
            query = TBaseType.stringBlock((int)lineNo - 1,(int)columnNo)+ this.sqlText;
        }


        sqlparser.sqltext = query;
        int ret = sqlparser.parse();

        if ( ret != 0){
            for(int j=0;j<sqlparser.getErrorCount();j++){
                this.parseerrormessagehandle(sqlparser.getSyntaxErrors().get(j));
            }

            return null;
        }

        this.dynamicStatements = new TStatementList();
        for(int i=0;i<sqlparser.sqlstatements.size();i++){
            if (this.getParentStmt() == null){
                sqlparser.sqlstatements.get(i).setParentStmt(this);
            }else{
                sqlparser.sqlstatements.get(i).setParentStmt(this.getParentStmt());
            }

            this.dynamicStatements.add(sqlparser.sqlstatements.get(i));
        }
        return dynamicStatements;
    }

    public TObjectName getLinkServerName() {
        return linkServerName;
    }

    private TObjectName linkServerName;
    private TExpressionList stringValues = null;

    public TExpressionList getStringValues() {
        return stringValues;
    }

    private TObjectName moduleName;

    public TObjectName getModuleName() {
        return moduleName;
    }

    private TExpression stmtString;
    private TObjectName intoVariable;
    private TExpressionList usingVariables;

    private TObjectName statementName;
    private TExpressionList parameters;
    private TExpression preparedValue;//couchbase

    public void setSqlText(String sqlText) {
        this.sqlText = sqlText;
    }

    private String sqlText = null;

    public String getSqlText() {
        return sqlText;
    }

    public void init(Object arg1){
        stmtString = (TExpression)arg1;
        sqlText = stmtString.toString();
        switch (stmtString.getExpressionType()){
            case simple_object_name_t:
                moduleName = stmtString.getObjectOperand();
                //moduleName.setObjectType(TObjectName.ttobjProcedureName);
                moduleName.setDbObjectType(EDbObjectType.procedure);
                break;
            case function_t:
                if (stmtString.toString().startsWith("format")){
                    //postgresql format function
                    TFunctionCall functionCall = stmtString.getFunctionCall();
                    sqlText = functionCall.getArgs().getExpression(0).toString();
                    sqlText = sqlText.replaceAll("%s","PLACEHOLDER");
                }
                break;
        }

    }
    public void init(Object arg1,Object arg2){
        init(arg1);
        intoVariable = (TObjectName)arg2;

    }

    public TObjectName getIntoVariable() {
        return intoVariable;
    }

    public TExpression getStmtString() {
        return stmtString;
    }

    public TExpressionList getUsingVariables() {
        return usingVariables;
    }

    public void init(Object arg1,Object arg2, Object arg3){
        init(arg1,arg2);
        usingVariables = (TExpressionList)arg3;

    }

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

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

    public void setStmtString(TExpression stmtString) {
        this.stmtString = stmtString;
    }

    public void setIntoVariable(TObjectName intoVariable) {
        this.intoVariable = intoVariable;
    }

    public void setUsingVariables(TExpressionList usingVariables) {
        this.usingVariables = usingVariables;
    }

    public TObjectName getStatementName() {
        return statementName;
    }
    public TExpressionList getParameters() {
        return parameters;
    }
    public void setStatementName(TObjectName statementName) {
        this.statementName = statementName;
    }
    public void setParameters(TExpressionList parameters) {
        this.parameters = parameters;
    }
    public TExpression getPreparedValue() {
        return preparedValue;
    }

}