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.TPathSqlNode;
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();

        // Mantis #4425: when the FROM clause uses a quoted string literal
        // (e.g. FROM '@db.schema.stage/path' or FROM 's3://bucket/path'),
        // the grammar stores it as fromSourceLocation rather than as a
        // structured stage location, leaving stageLocation null.
        // Parse the literal and populate stageLocation so the public API
        // is consistent with the unquoted form.
        if (this.stageLocation == null && fromSourceLocation != null) {
            populateStageLocationFromLiteral(fromSourceLocation);
        }

        return 0;
    }

    private void populateStageLocationFromLiteral(TObjectName source) {
        String raw = source.toString();
        if (raw == null || raw.isEmpty()) return;

        // Strip surrounding single quotes, if any.
        String content = raw;
        if (content.length() >= 2 && content.startsWith("'") && content.endsWith("'")) {
            content = content.substring(1, content.length() - 1);
        }
        if (content.isEmpty()) return;

        if (content.startsWith("@")) {
            populateInternalStageLocation(content);
        } else if (looksLikeExternalUrl(content)) {
            TStageLocation loc = new TStageLocation();
            loc.init(TStageLocation.EStageLocationType.location);
            loc.setStage(false);
            TObjectName extLoc = new TObjectName();
            extLoc.setString(raw);
            loc.setExternalLocation(extLoc);
            this.stageLocation = loc;
        }
    }

    private void populateInternalStageLocation(String content) {
        // content begins with '@'
        String afterAt = content.substring(1);

        // Split into name part and path part at the first '/'.
        String namePart = afterAt;
        String pathPart = null;
        int slashIdx = afterAt.indexOf('/');
        if (slashIdx >= 0) {
            namePart = afterAt.substring(0, slashIdx);
            pathPart = afterAt.substring(slashIdx);
        }

        TStageLocation loc = new TStageLocation();
        loc.setStage(true);

        if (namePart.isEmpty() || namePart.equals("~") || namePart.startsWith("~")) {
            // @~ or @~/path -> user stage
            loc.init(TStageLocation.EStageLocationType.internalUser);
        } else if (namePart.startsWith("%")) {
            // @%table or @%table/path -> table stage
            loc.init(TStageLocation.EStageLocationType.internalTable);
            TObjectName tblName = new TObjectName();
            tblName.setString(namePart.substring(1));
            loc.setTableName(tblName);
        } else {
            // @stage, @schema.stage, or @db.schema.stage (with optional path)
            TObjectName stageName = new TObjectName();
            stageName.setString(namePart);
            loc.init(TStageLocation.EStageLocationType.internalNamed, stageName);
        }

        if (pathPart != null && !pathPart.isEmpty()) {
            TPathSqlNode path = new TPathSqlNode();
            path.setString(pathPart);
            loc.setPath(path);
        }

        this.stageLocation = loc;
    }

    private boolean looksLikeExternalUrl(String content) {
        String lower = content.toLowerCase();
        return lower.startsWith("s3://")
            || lower.startsWith("azure://")
            || lower.startsWith("gcs://")
            || lower.startsWith("https://")
            || lower.startsWith("http://")
            || lower.startsWith("file://");
    }

    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);
    }

}