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.sqlenv.ESQLDataObjectType;
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", "fileType", "fileFormat", "location", "namespace", "isTarget", "coordinate", "columns", "parent", "more", "fromDDL"}
)
public class table implements Cloneable {

    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 isTarget;

    private StringBuffer coordinate = new StringBuffer();

    private List<column> columns;

    private String parent;

    private String fileType;

    private String fileFormat;

    private String location;

    private String namespace;

    @XmlTransient
    private String starStmt;

    private Boolean more;
    
    @XmlTransient
    private String isDetermined;

    private String fromDDL;

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

    @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(DlineageUtil.getIdentifierNormalName(database, ESQLDataObjectType.dotCatalog)).append(".");
            }
            else{
                fullName.append(database).append(".");
            }
        }
        if (!SQLUtil.isEmpty(schema)) {
            if(ModelBindingManager.getGlobalVendor()!=null) {
                fullName.append(DlineageUtil.getIdentifierNormalName(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;
    }

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