package gudusoft.gsqlparser.dlineage.dataflow.model;

import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.TCustomSqlStatement;
import gudusoft.gsqlparser.TSourceToken;
import gudusoft.gsqlparser.dlineage.util.DlineageUtil;
import gudusoft.gsqlparser.dlineage.util.Pair3;
import gudusoft.gsqlparser.nodes.TParseTreeNode;
import gudusoft.gsqlparser.sqlenv.TSQLEnv;
import gudusoft.gsqlparser.util.Logger;
import gudusoft.gsqlparser.util.LoggerFactory;
import gudusoft.gsqlparser.util.SQLUtil;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;

public class ResultSet {
    private static final Logger logger = LoggerFactory.getLogger(ResultSet.class);
    private long id;
    protected String server;
    protected String schema;
    protected String database;
    private Pair3<Long, Long, String> startPosition;
    private Pair3<Long, Long, String> endPosition;
    private List<ResultColumn> columns = new ArrayList<ResultColumn>();

    private TParseTreeNode gspObject;
    private boolean isTarget;
    private boolean isDetermined;

    private ResultSetRelationRows relationRows = new ResultSetRelationRows(this);

    private Map<String, AtomicInteger> aliasMap = new HashMap<>();
    
    public ResultSet(TParseTreeNode gspObject, boolean isTarget) {
        if (gspObject == null) {
            throw new IllegalArgumentException("ResultSet arguments can't be null.");
        }

        id = ++ModelBindingManager.get().TABLE_COLUMN_ID;

        this.gspObject = gspObject;
        this.isTarget = isTarget;

        TSourceToken startToken = gspObject.getStartToken();
        TSourceToken endToken = gspObject.getEndToken();
        if (startToken != null) {
            this.startPosition = new Pair3<Long, Long, String>(startToken.lineNo,
                    startToken.columnNo, ModelBindingManager.getGlobalHash());
        }
        
        if (endToken != null) {
            this.endPosition = new Pair3<Long, Long, String>(endToken.lineNo,
                    endToken.columnNo + SQLUtil.endTrim(endToken.astext).length(), ModelBindingManager.getGlobalHash());
        }
        
        this.schema = ModelBindingManager.getGlobalSchema();
        this.database = ModelBindingManager.getGlobalDatabase();

        EDbVendor vendor = ModelBindingManager.getGlobalOption().getVendor();
        boolean supportCatalog = TSQLEnv.supportCatalog(vendor);
        boolean supportSchema = TSQLEnv.supportSchema(vendor);
        
        fillSchemaInfo();
        
        if (!supportCatalog) {
            this.database = null;
        } else if (this.database == null && !TSQLEnv.DEFAULT_DB_NAME.equals(getDefaultDatabase())) {
            this.database = getDefaultDatabase();
        }

        if (!supportSchema) {
            this.schema = null;
        } else if (this.schema == null && !TSQLEnv.DEFAULT_SCHEMA_NAME.equals(getDefaultSchema())) {
            this.schema = getDefaultSchema();
        }

        if (this.server == null && !TSQLEnv.DEFAULT_SERVER_NAME.equals(getDefaultServer())) {
            this.server = getDefaultServer();
        }
    }

    private void fillSchemaInfo() {
                TCustomSqlStatement stmt = DlineageUtil.getTopStmt(ModelBindingManager.getGlobalStmtStack().peek());
                String sqlComment = null;
                try {
                        sqlComment = stmt.getCommentBeforeNode();
                } catch (Exception e) {
                }
                if (!SQLUtil.isEmpty(sqlComment) && (sqlComment.indexOf("db") != -1 || sqlComment.indexOf("schema") != -1)) {
                        Properties properties = new Properties();
                        try {
                                properties.load(
                                                new ByteArrayInputStream(sqlComment.replace("--", "").trim().replace(",", "\n").getBytes()));
                                if (SQLUtil.isEmpty(this.server) && properties.containsKey("db-instance")) {
                                        this.server = properties.getProperty("db-instance");
                                }
                                if (SQLUtil.isEmpty(this.database) && properties.containsKey("db")) {
                                        this.database = properties.getProperty("db");
                                        if(this.database.indexOf(".")!=-1) {
                                                String delimitedChar = TSQLEnv.delimitedChar(ModelBindingManager.getGlobalOption().getVendor());
                                                this.database = delimitedChar + SQLUtil.trimColumnStringQuote(this.database) + delimitedChar;
                                        }
                                }
                                if (SQLUtil.isEmpty(this.schema) && properties.containsKey("schema")) {
                                        this.schema = properties.getProperty("schema");
                                        if(this.schema.indexOf(".")!=-1) {
                                                String delimitedChar = TSQLEnv.delimitedChar(ModelBindingManager.getGlobalOption().getVendor());
                                                this.schema = delimitedChar + SQLUtil.trimColumnStringQuote(this.schema) + delimitedChar;
                                        }
                                }
                        } catch (IOException e) {
                logger.error("load sql comment properties failed.", e);
                        }
                }
        }
    
    protected String getDefaultServer() {
        String defaultServer = null;
        if (ModelBindingManager.getGlobalSQLEnv() != null) {
            defaultServer = ModelBindingManager.getGlobalSQLEnv().getDefaultServerName();
        }
        if (!SQLUtil.isEmpty(defaultServer))
            return defaultServer;
        return TSQLEnv.DEFAULT_SERVER_NAME;
    }

    protected String getDefaultSchema() {
        String defaultSchema = null;
        if (ModelBindingManager.getGlobalSQLEnv() != null) {
            defaultSchema = ModelBindingManager.getGlobalSQLEnv().getDefaultSchemaName();
        }
        if (!SQLUtil.isEmpty(defaultSchema))
            return defaultSchema;
        return TSQLEnv.DEFAULT_SCHEMA_NAME;
    }

    protected String getDefaultDatabase() {
        String defaultDatabase = null;
        if (ModelBindingManager.getGlobalSQLEnv() != null) {
            defaultDatabase = ModelBindingManager.getGlobalSQLEnv().getDefaultCatalogName();
        }
        if (!SQLUtil.isEmpty(defaultDatabase))
            return defaultDatabase;
        return TSQLEnv.DEFAULT_DB_NAME;
    }

    public Pair3<Long, Long, String> getStartPosition() {
        return startPosition;
    }

    public Pair3<Long, Long, String> getEndPosition() {
        return endPosition;
    }

    public List<ResultColumn> getColumns() {
        return columns;
    }

    public void addColumn(ResultColumn column) {
        if (column != null && !columns.contains(column)) {
            this.columns.add(column);
        }
    }

    public TParseTreeNode getGspObject() {
        return gspObject;
    }

    public long getId() {
        return id;
    }

    public boolean isTarget() {
        return isTarget;
    }

        public String getSchema() {
                return schema;
        }

        public String getDatabase() {
                return database;
        }
        
        public ResultSetRelationRows getRelationRows(){
                return relationRows;
        }

    public String getServer() {
        return server;
    }

        public boolean isDetermined() {
                return isDetermined;
        }

        public void setDetermined(boolean isDetermined) {
                this.isDetermined = isDetermined;
        }

        public void setTarget(boolean isTarget) {
                this.isTarget = isTarget;
        }

    Map<String, AtomicInteger> getAliasMap() {
        return aliasMap;
    }
}