package gudusoft.gsqlparser.stmt.snowflake;


import gudusoft.gsqlparser.EDbVendor;
import gudusoft.gsqlparser.ESqlClause;
import gudusoft.gsqlparser.ESqlStatementType;
import gudusoft.gsqlparser.TCustomSqlStatement;
import gudusoft.gsqlparser.ETableSource;
import gudusoft.gsqlparser.nodes.TCreateTableOption;
import gudusoft.gsqlparser.nodes.TObjectName;
import gudusoft.gsqlparser.nodes.TObjectNameList;
import gudusoft.gsqlparser.nodes.TParseTreeVisitor;
import gudusoft.gsqlparser.nodes.TStageLocation;
import gudusoft.gsqlparser.nodes.TTable;
import gudusoft.gsqlparser.nodes.snowflake.TCopyIntoNode;
import gudusoft.gsqlparser.nodes.snowflake.TStageReference;
import gudusoft.gsqlparser.stmt.TSelectSqlStatement;

import java.util.ArrayList;

public class TSnowflakeCopyIntoStmt extends TCustomSqlStatement {

    private ArrayList<String> fileList = new ArrayList<>();


    public ArrayList<String> getFileList() {
        return fileList;
    }

    private String fileFormatName = null;
    private String fileFormatType = null;

    public void setFileFormatName(String fileFormatName) {
        this.fileFormatName = fileFormatName;
    }

    public void setFileFormatType(String fileFormatType) {
        this.fileFormatType = fileFormatType;
    }

    public String getFileFormatName() {
        return fileFormatName;
    }

    public String getFileFormatType() {
        return fileFormatType;
    }

    public static int COPY_INTO_TABLE = 0;
    public static int COPY_INTO_LOCATION = 1;

    private String regex_pattern;

    public void setRegex_pattern(String regex_pattern) {
        this.regex_pattern = regex_pattern;
    }

    public String getRegex_pattern() {
        return regex_pattern;
    }

    private int copyIntoType = TSnowflakeCopyIntoStmt.COPY_INTO_TABLE;

    public int getCopyIntoType() {
        return copyIntoType;
    }

    private TStageLocation stageLocation;

    public TStageLocation getStageLocation() {
        return stageLocation;
    }

    private TObjectName tableName;
    private TSelectSqlStatement subQuery;
    public TSnowflakeCopyIntoStmt(EDbVendor dbvendor) {
        super(dbvendor);
        sqlstatementtype = ESqlStatementType.sstCopyInto;
    }

    public void setTableName(TObjectName tableName) {
        this.tableName = tableName;
    }

    public void setSubQuery(TSelectSqlStatement subQuery) {
        this.subQuery = subQuery;
    }

    public TObjectName getTableName() {

        return tableName;
    }

    public TSelectSqlStatement getSubQuery() {
        return subQuery;
    }

    private TObjectName fromSourceLocation;
    public TObjectName getFromSourceLocation() {
        return fromSourceLocation;
    }

    private TObjectNameList tableColumnList;

    public void setTableColumnList(TObjectNameList tableColumnList) {
        this.tableColumnList = tableColumnList;
    }

    public TObjectNameList getTableColumnList() {
        return tableColumnList;
    }


    public int doParseStatement(TCustomSqlStatement psql) {
        if (rootNode == null) return -1;
        super.doParseStatement(psql);
        TCopyIntoNode node = (TCopyIntoNode)(rootNode);
        tableName = node.getTableName();
        this.copyIntoType = node.getCopyIntoType();
        this.stageLocation = node.getStageLocation();

        if (node.getSubquery() != null){
            subQuery = new TSelectSqlStatement(this.dbvendor);
            subQuery.rootNode = node.getSubquery();
            subQuery.doParseStatement(this);

            // When the stage location is inside the subquery's FROM clause
            // (e.g., COPY INTO t FROM (SELECT ... FROM @stage/path)),
            // propagate it to the COPY INTO statement for API accessibility.
            if (this.stageLocation == null && subQuery.getTables() != null) {
                for (int i = 0; i < subQuery.getTables().size(); i++) {
                    TTable table = subQuery.getTables().getTable(i);
                    if (table.getTableType() == ETableSource.stageReference
                            && table.getStageReference() != null) {
                        TStageReference ref = table.getStageReference();
                        this.stageLocation = new TStageLocation();
                        this.stageLocation.setStage(true);
                        if (ref.getStageName() != null) {
                            String name = ref.getStageName().toString();
                            if (name.startsWith("~")) {
                                this.stageLocation.init(TStageLocation.EStageLocationType.internalUser);
                            } else if (name.startsWith("%")) {
                                this.stageLocation.init(TStageLocation.EStageLocationType.internalTable);
                                TObjectName tblName = new TObjectName();
                                tblName.setString(name.substring(1));
                                this.stageLocation.setTableName(tblName);
                            } else {
                                this.stageLocation.init(TStageLocation.EStageLocationType.internalNamed, ref.getStageName());
                            }
                        }
                        if (ref.getStagePath() != null) {
                            this.stageLocation.setPath(ref.getStagePath());
                        }
                        break;
                    }
                }
            }
        }

        ArrayList<TCreateTableOption> tableOptions = node.tableOptions;
        if (tableOptions != null){
            for(int i=0;i<tableOptions.size();i++){
                tableOptions.get(i).doParse(this, ESqlClause.unknown);
            }
        }

        fromSourceLocation = node.getFromSourceLocation();
        tableColumnList = node.getTableColumnList();

        return 0;
    }

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

    public void acceptChildren(TParseTreeVisitor v){
        v.preVisit(this);
        if (this.subQuery != null){
            this.subQuery.acceptChildren(v);
        }
        v.postVisit(this);
    }

}