package gudusoft.gsqlparser.dlineage.dataflow.model.xml;

import gudusoft.gsqlparser.dlineage.dataflow.model.ModelBindingManager;
import gudusoft.gsqlparser.dlineage.dataflow.model.json.Coordinate;
import gudusoft.gsqlparser.dlineage.util.DlineageUtil;
import gudusoft.gsqlparser.dlineage.util.LegacyDlineageWireNamesV1;
import gudusoft.gsqlparser.sqlenv.ESQLDataObjectType;
import gudusoft.gsqlparser.sqlenv.QualifiedCanonKey;
import gudusoft.gsqlparser.sqlenv.TSQLEnv;
import gudusoft.gsqlparser.util.SQLUtil;

import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlTransient;
import javax.xml.bind.annotation.XmlType;
import java.util.ArrayList;
import java.util.LinkedHashSet;
import java.util.List;

@XmlType(
        propOrder = {"id", "server", "userName", "database", "schema", "name", "displayName", "alias", "uri", "type", "subType",
                "processIds", "procedureId", "fileType", "fileFormat", "location", "dataSourceName", "namespace", "isTarget", "coordinate", "columns", "parent", "more", "fromDDL", "candidateTables",
                "endpointKind", "endpointIntroduction", "createdInSql", "normalizedName", "templatedParts"}
)
public class table implements Cloneable {

    private QualifiedCanonKey identifierKey;

    /** Same-run identity supplied by the producer, never inferred from display text. */
    @XmlTransient
    public QualifiedCanonKey getIdentifierKey() {
        return identifierKey;
    }

    public void setIdentifierKey(QualifiedCanonKey identifierKey) {
        this.identifierKey = identifierKey;
    }

    private boolean scopedTemporary;

    /**
     * Same-run fact, never serialized: this is one routine body's temporary table (see
     * {@link gudusoft.gsqlparser.sqlenv.TemporaryTableSemantics}), a different table from any
     * other of the same name, so it is never merged with one.
     */
    @XmlTransient
    public boolean isScopedTemporary() {
        return scopedTemporary;
    }

    public void setScopedTemporary(boolean scopedTemporary) {
        this.scopedTemporary = scopedTemporary;
    }

    private boolean arrayRelation;

    /**
     * Same-run fact, never serialized: this is the relation a table-generating construct adds to one SELECT (a
     * Hive-family LATERAL VIEW), a different table from any other of the same name, so it is never merged with one
     * (nested-access plan section 4.3, Mantis 4890).
     */
    @XmlTransient
    public boolean isArrayRelation() {
        return arrayRelation;
    }

    public void setArrayRelation(boolean arrayRelation) {
        this.arrayRelation = arrayRelation;
    }

    private QualifiedCanonKey mergeIdentity;

    /**
     * For a function table (a call): the identity of the routine it calls, from the call's name as written, taken
     * when the analysis published it. Normalized output renames the table afterwards, and an XML document drops it.
     */
    private Object routineKey;

    @XmlTransient
    public Object getRoutineKey() {
        return routineKey;
    }

    public void setRoutineKey(Object routineKey) {
        this.routineKey = routineKey;
    }

    /**
     * Same-run identity of a physical table or view under the analysis's identifier rules, for every
     * vendor, as the producing model held it. Merges split two tables that share a folded name only
     * when these differ; it is never inferred from the published name, and it does not survive
     * serialization.
     */
    @XmlTransient
    public QualifiedCanonKey getMergeIdentity() {
        return mergeIdentity;
    }

    public void setMergeIdentity(QualifiedCanonKey mergeIdentity) {
        this.mergeIdentity = mergeIdentity;
    }

    private boolean publishedNameIsSqlSpelling;

    /**
     * Same-run marker, set by producers whose published names are SQL spellings by construction (the
     * SQLDep, Grabit and sqlflow metadata readers): the table's identity is its published name read as
     * SQL under the analysis's identifier rules, which a merge applies with its own service. Not
     * serialized.
     */
    @XmlTransient
    public boolean isPublishedNameSqlSpelling() {
        return publishedNameIsSqlSpelling;
    }

    public void setPublishedNameSqlSpelling(boolean publishedNameIsSqlSpelling) {
        this.publishedNameIsSqlSpelling = publishedNameIsSqlSpelling;
    }

    private table mintedNames;

    /**
     * Same-run: set on a merged table whose names no minting model published (a metadata reader's spelling,
     * or a table's that carries no identity) while a table merged into it had such names; the first such
     * table's names and identity, as a table holding only those. A later merge that joins across legacy
     * folded names publishes these, as one merge of all its inputs would. Not serialized.
     */
    @XmlTransient
    public table getMintedNames() {
        return mintedNames;
    }

    public void setMintedNames(table mintedNames) {
        this.mintedNames = mintedNames;
    }

    private boolean acrossLegacyNames;

    /**
     * Same-run: whether this table merges tables whose legacy folded names differ, which only identities
     * join. A later merge that includes it chooses names as a merge across legacy names does. Not
     * serialized.
     */
    @XmlTransient
    public boolean isAcrossLegacyNames() {
        return acrossLegacyNames;
    }

    public void setAcrossLegacyNames(boolean acrossLegacyNames) {
        this.acrossLegacyNames = acrossLegacyNames;
    }

    private java.util.Map<String, String> carriedPublishedNames;

    /**
     * Same-run: on a merged table, the exact names that the tables carrying an identity merged into it published,
     * other than the one it publishes itself, each with its legacy folded key, in merge order. A later merge
     * matches a table that carries no identity on these as on the table's own name (an exact published name),
     * as one merge of all the inputs would. Null when there are none. Not serialized.
     */
    @XmlTransient
    public java.util.Map<String, String> getCarriedPublishedNames() {
        return carriedPublishedNames;
    }

    public void setCarriedPublishedNames(java.util.Map<String, String> carriedPublishedNames) {
        this.carriedPublishedNames = carriedPublishedNames;
    }

    private String id;

    private String server;

    private String userName;

    private String database;

    private String schema;

    private String name;

    private String displayName;

    private String alias;

    private String type;

    private String subType;

    private String uri;

    private List<String> processIds;

    private String procedureId;

    private String isTarget;

    private StringBuffer coordinate = new StringBuffer();

    private List<column> columns;

    private String parent;

    private String fileType;

    private String fileFormat;

    private String location;

    private String dataSourceName;

    private String namespace;

    @XmlTransient
    private String starStmt;

    private Boolean more;
    
    @XmlTransient
    private String isDetermined;

    private String fromDDL;

    /**
     * Qualified candidate table names when this one-part reference is ambiguous
     * across two or more schemas of the default catalog (e.g. {@code testdb.s1.orders},
     * {@code testdb.s2.orders}) and no object of that name exists in the default
     * schema. Empty/absent for normally-resolved references. See
     * docs/tmp/unqualified-name-default-schema-masks-cross-schema-ambiguity.md.
     */
    private List<String> candidateTables;

    /**
     * Authoritative endpoint classification (additive). Carried as Strings so a null
     * value is omitted from XML, keeping existing output byte-identical for endpoints
     * that aren't classified. Populated from the internal model only when non-default.
     * See docs/tmp/dlineage-authoritative-endpoint-classification.md.
     */
    private String endpointKind;

    private String endpointIntroduction;

    private String createdInSql;

    private String normalizedName;

    /**
     * Which identifier segments of {@link #getName()} a dynamic-SQL fold assembled at
     * runtime, pipe-joined outermost-first ({@code "db"}, {@code "db|name"}, ...). Emitted
     * only for {@code endpointKind} endpoints the analyzer marked inside a fold context;
     * absent otherwise, so existing output stays byte-identical. See
     * docs/tmp/dynamic-template-endpoint-provenance.md.
     */
    private String templatedParts;

    @XmlTransient
    private LinkedHashSet<String> coordinateItems = new LinkedHashSet<String>();

    @XmlAttribute(required = false)
    public String getAlias() {
        return alias;
    }

    public void setAlias(String alias) {
        this.alias = alias;
    }

    @XmlElement(name = "column", required = false)
    public List<column> getColumns() {
        if (this.columns == null) {
            this.columns = new ArrayList<column>();
        }
        return columns;
    }

    public void setColumns(List<column> columns) {
        this.columns = columns;
    }

    @XmlAttribute(required = false)
    public String getCoordinate() {
        String result = coordinate.toString();
        if (SQLUtil.isEmpty(result))
            return null;
        return result;
    }

    public void appendCoordinate(String coordinate) {
        if (!coordinateItems.contains(coordinate)) {
            coordinateItems.add(coordinate);
            rebuildCoordinate();
        }
    }
    
    private void rebuildCoordinate() {
        this.coordinate.setLength(0);
        
        List<String> itemsList = new ArrayList<>(coordinateItems);
        boolean hasMultiplePairs = itemsList.size() > 1;
        
        String separator = "";
        for (String coordPair : itemsList) {
            // 只有多对坐标时才过滤无效对
            if (!hasMultiplePairs || !isBothCoordsInvalid(coordPair)) {
                this.coordinate.append(separator).append(coordPair);
                separator = ",";
            }
        }
    }
    
    /**
     * 判断一对坐标是否两个都是无效坐标（[-1,-1,x] 格式）
     */
    private static boolean isBothCoordsInvalid(String coordPair) {
        return coordPair != null 
            && coordPair.contains("[-1,-1,") 
            && countOccurrences(coordPair, "[-1,-1,") >= 2;
    }
    
    private static int countOccurrences(String str, String target) {
        if (str == null || target == null || target.isEmpty()) {
            return 0;
        }
        int count = 0;
        int idx = 0;
        while ((idx = str.indexOf(target, idx)) != -1) {
            count++;
            idx += target.length();
        }
        return count;
    }

    public void setCoordinate(String coordinate) {
        if (SQLUtil.isEmpty(coordinate)) {
                return;
        }
        this.coordinate.setLength(0);
        this.coordinateItems.clear();
        
        // 先分割出所有坐标 [line,col,fileIdx]
        List<String> allCoords = new ArrayList<>();
        int start = 0;
        while (start < coordinate.length()) {
            int open = coordinate.indexOf('[', start);
            if (open < 0) break;
            int close = coordinate.indexOf(']', open);
            if (close < 0) break;
            allCoords.add(coordinate.substring(open, close + 1));
            start = close + 1;
        }
        
        // 计算坐标对数量（每两个坐标组成一对）
        int pairCount = (allCoords.size() + 1) / 2;
        
        // 如果只有一对坐标，直接保留
        if (pairCount <= 1) {
            this.coordinate.append(coordinate);
            this.coordinateItems.add(coordinate);
            return;
        }
        
        // 有多对坐标时，过滤掉无效的坐标对（两个坐标都是 [-1,-1,x] 格式）
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < allCoords.size(); i += 2) {
            String coord1 = allCoords.get(i);
            String coord2 = (i + 1 < allCoords.size()) ? allCoords.get(i + 1) : "";
            
            // 检查这一对是否都是无效坐标
            boolean pairIsInvalid = isInvalidCoord(coord1) && isInvalidCoord(coord2);
            
            if (!pairIsInvalid) {
                if (sb.length() > 0) {
                    sb.append(",");
                }
                sb.append(coord1);
                this.coordinateItems.add(coord1);
                if (!coord2.isEmpty()) {
                    sb.append(",").append(coord2);
                    this.coordinateItems.add(coord2);
                }
            }
        }
        this.coordinate.append(sb.toString());
    }
    
    private static boolean isInvalidCoord(String coord) {
        if (coord == null || coord.isEmpty()) {
            return false;
        }
        String trimmed = coord.trim();
        if (trimmed.indexOf("-1") != -1 && trimmed.startsWith("[") && trimmed.endsWith("]")) {
            String inner = trimmed.substring(1, trimmed.length() - 1);
            String[] coords = inner.split(",");
            if (coords.length >= 2 && "-1".equals(coords[0].trim()) && "-1".equals(coords[1].trim())) {
                return true;
            }
        }
        return false;
    }
    
    public void clearCoordinate() {
        this.coordinate = new StringBuffer();
    }

    @XmlAttribute(required = false)
    public String getUserName() {
        return userName;
    }

    public void setUserName(String userName) {
        this.userName = userName;
    }

    @XmlAttribute(required = false)
    public String getServer() {
        return server;
    }

    public void setServer(String server) {
        this.server = server;
    }

    @XmlAttribute(required = false)
    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    @XmlAttribute(required = false)
    public String getDisplayName() {
        return displayName;
    }

    public void setDisplayName(String displayName) {
        this.displayName = displayName;
    }

    @XmlAttribute(required = false)
    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    @XmlAttribute(required = false)
    public List<String> getProcessIds() {
        return processIds;
    }

    public void setProcessIds(List<String> processIds) {
        this.processIds = processIds;
    }

    @XmlAttribute(required = false)
    public String getProcedureId() {
        return procedureId;
    }

    public void setProcedureId(String procedureId) {
        this.procedureId = procedureId;
    }

    @XmlElement(name = "candidate", required = false)
    public List<String> getCandidateTables() {
        return candidateTables;
    }

    public void setCandidateTables(List<String> candidateTables) {
        this.candidateTables = candidateTables;
    }

    @XmlAttribute(required = false)
    public String getType() {
        return type;
    }

    public void setType(String type) {
        this.type = type;
    }

    @XmlAttribute(required = false)
    public String getUri() {
        return uri;
    }

    public void setUri(String uri) {
        this.uri = uri;
    }

    @XmlAttribute(required = false)
    public String getFileType() {
        return fileType;
    }

    public void setFileType(String fileType) {
        this.fileType = fileType;
    }

    @XmlAttribute(required = false)
    public String getFileFormat() {
        return fileFormat;
    }

    public void setFileFormat(String fileFormat) {
        this.fileFormat = fileFormat;
    }

    @XmlAttribute(required = false)
    public String getLocation() {
        return location;
    }

    public void setLocation(String location) {
        this.location = location;
    }

    /** The external data source a file path is read through (SQL Server BULK INSERT ... WITH (DATA_SOURCE = ...)). */
    @XmlAttribute(required = false)
    public String getDataSourceName() {
        return dataSourceName;
    }

    public void setDataSourceName(String dataSourceName) {
        this.dataSourceName = dataSourceName;
    }

    @XmlAttribute(required = false)
    public String getNamespace() {
        return namespace;
    }

    public void setNamespace(String namespace) {
        this.namespace = namespace;
    }

    @XmlTransient
    public String getStarStmt() {
        return starStmt;
    }

    public void setStarStmt(String starStmt) {
        this.starStmt = starStmt;
    }

    @XmlTransient
    public String getIsDetermined() {
                return isDetermined;
        }

        public void setIsDetermined(String isDetermined) {
                this.isDetermined = isDetermined;
        }

        public boolean isFunction() {
        return "function".equals(type) || "function".equals(subType);
    }

    public boolean isView() {
        return "view".equals(type);
    }

    public boolean isDatabaseType() {
        return "database".equals(type);
    }

    public boolean isSchemaType() {
        return "schema".equals(type);
    }

    public boolean isSequence() {
        return "sequence".equals(type);
    }

    public boolean isStage() {
        return "stage".equals(type);
    }

    public boolean isDataSource() {
        return "dataSource".equals(type);
    }

    public boolean isStream() {
        return "stream".equals(type);
    }

    public boolean isVariable() {
        return "variable".equals(type);
    }

    public boolean isCursor() {
        return "cursor".equals(type);
    }

    public boolean isFile() {
        return "file".equals(type) || "path".equals(type);
    }

    public boolean isTable() {
        return "table".equals(type) || "pseudoTable".equals(type) || "constantTable".equals(type);
    }

    public boolean isPseudoTable() {
        return "pseudoTable".equals(type);
    }

    public boolean isConstantTable() {
        return "pseudoTable".equals(type);
    }

    public boolean isResultSet() {
        return type != null && !isView() && !isCursor() && !isTable() && !isStage() && !isSequence() && !isDataSource() && !isDatabaseType() && !isSchemaType() && !isStream() && !isVariable() && !isFile();
    }

    @XmlAttribute(name = "isTarget", required = false)
    public String getIsTarget() {
        return isTarget;
    }

    public boolean isTarget() {
        return "true".equals(isTarget);
    }

    @XmlAttribute(required = false)
    public String getParent() {
        return parent;
    }

    public void setParent(String parent) {
        this.parent = parent;
    }

    @XmlAttribute(required = false)
    public String getDatabase() {
        return database;
    }

    public void setDatabase(String database) {
        if (SQLUtil.parseNames(database).size() > 1) {
            database = "\"" + database + "\"";
        }
        this.database = database;
    }

    @XmlAttribute(required = false)
    public String getSchema() {
        return schema;
    }

    public void setSchema(String schema) {
        if (SQLUtil.parseNames(schema).size() > 1) {
            schema = "\"" + schema + "\"";
        }
        this.schema = schema;
    }

    @XmlAttribute(required = false)
    public String getSubType() {
        return subType;
    }

    public void setSubType(String subType) {
        this.subType = subType;
    }

    public String getFullName() {
        if (isDatabaseType()) {
            return database;
        }
        StringBuilder fullName = new StringBuilder();
        if (!SQLUtil.isEmpty(database)) {
            fullName.append(database).append(".");
        }
        if (!SQLUtil.isEmpty(schema)) {
            fullName.append(schema).append(".");
        }
        if (fullName.length() > 0) {
            fullName.append(getTableNameOnly());
        } else {
            fullName.append(name);
        }
        return fullName.toString();
    }

    public String getFullSchemaName() {
        StringBuilder fullName = new StringBuilder();
        if (!SQLUtil.isEmpty(database)) {
            if(ModelBindingManager.getGlobalVendor()!=null) {
                fullName.append(LegacyDlineageWireNamesV1.name(database, ESQLDataObjectType.dotCatalog)).append(".");
            }
            else{
                fullName.append(database).append(".");
            }
        }
        if (!SQLUtil.isEmpty(schema)) {
            if(ModelBindingManager.getGlobalVendor()!=null) {
                fullName.append(LegacyDlineageWireNamesV1.name(schema, ESQLDataObjectType.dotSchema));
            }
            else {
                fullName.append(schema).append(".");
            }
        }
        String fullSchemaName = fullName.toString();
        if (fullSchemaName.endsWith(".")) {
            fullSchemaName = fullSchemaName.substring(0, fullSchemaName.length() - 1);
        }
        if (fullSchemaName.length() == 0) {
            fullSchemaName = TSQLEnv.DEFAULT_SCHEMA_NAME;
        }
        return fullSchemaName;
    }

    public String getTableNameOnly() {
        if (name.indexOf("@") != -1 && SQLUtil.trimColumnStringQuote(name.substring(name.lastIndexOf("@") + 1).trim()).equals(SQLUtil.trimColumnStringQuote(database))) {
            List<String> segments = SQLUtil.parseNames(name.substring(0, name.lastIndexOf("@")).trim());
            if (segments.size() > 2) {
                return SQLUtil.mergeSegments(segments, 2);
            }
            return segments.get(segments.size() - 1);
        } else {
            List<String> segments = SQLUtil.parseNames(name);
            if (segments.size() > 2) {
                return SQLUtil.mergeSegments(segments, 2);
            }
            return segments.get(segments.size() - 1);
        }
    }

    public void setIsTarget(String isTarget) {
        this.isTarget = isTarget;
    }

    public int getOccurrencesNumber() {
        return PositionUtil.getOccurrencesNumber(coordinate.toString());
    }

    public Coordinate getStartPos(int index) {
        return PositionUtil.getStartPos(coordinate.toString(), index);
    }

    public Coordinate getEndPos(int index) {
        return PositionUtil.getEndPos(coordinate.toString(), index);
    }

    public Boolean getMore() {
        return more;
    }

    public void setMore(Boolean more) {
        this.more = more;
    }

    @XmlAttribute(required = false)
    public String getFromDDL() {
        return fromDDL;
    }

    public void setFromDDL(String fromDDL) {
        this.fromDDL = fromDDL;
    }

    @XmlAttribute(required = false)
    public String getEndpointKind() {
        return endpointKind;
    }

    public void setEndpointKind(String endpointKind) {
        this.endpointKind = endpointKind;
    }

    @XmlAttribute(required = false)
    public String getEndpointIntroduction() {
        return endpointIntroduction;
    }

    public void setEndpointIntroduction(String endpointIntroduction) {
        this.endpointIntroduction = endpointIntroduction;
    }

    @XmlAttribute(required = false)
    public String getCreatedInSql() {
        return createdInSql;
    }

    public void setCreatedInSql(String createdInSql) {
        this.createdInSql = createdInSql;
    }

    @XmlAttribute(required = false)
    public String getNormalizedName() {
        return normalizedName;
    }

    public void setNormalizedName(String normalizedName) {
        this.normalizedName = normalizedName;
    }

    @XmlAttribute(required = false)
    public String getTemplatedParts() {
        return templatedParts;
    }

    public void setTemplatedParts(String templatedParts) {
        this.templatedParts = templatedParts;
    }

    @Override
    public Object clone() throws CloneNotSupportedException {
        return super.clone();
    }
}