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

import gudusoft.gsqlparser.EDbVendor;
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.Comparator;
import java.util.List;
import java.util.TreeSet;

@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 TreeSet<String> coordinateItems = new TreeSet<String>(new Comparator<String>() {

        @Override
        public int compare(String o1, String o2) {
            Coordinate[] c1 = Coordinate.parse(o1);
            Coordinate[] c2 = Coordinate.parse(o2);
            if (c1[0].getY() == -1 && c2[0].getY() != -1) {
                return 1;
            }
            if (c1[0].getY() != -1 && c2[0].getY() == -1) {
                return -1;
            }
            return Long.compare(c1[0].getY(), c1[0].getY());
        }
    });

    @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() {
        if (ModelBindingManager.getGlobalOption() != null && ModelBindingManager.getGlobalOption().isIgnoreCoordinate()) {
            return null;
        }
        String result = coordinate.toString();
        if (SQLUtil.isEmpty(result))
            return null;
        return result;
    }

    public void appendCoordinate(String coordinate) {
        if (ModelBindingManager.getGlobalOption() != null && ModelBindingManager.getGlobalOption().isIgnoreCoordinate()) {
            return;
        }
        if (!coordinateItems.contains(coordinate)) {
            coordinateItems.add(coordinate);
            this.coordinate.setLength(0);
            for (String item : coordinateItems) {
                if (this.coordinate.length() > 0) {
                    if (item.indexOf("-1") != -1) {
                        continue;
                    }
                    this.coordinate.append(",");
                }
                this.coordinate.append(item);
            }
        }
    }

    public void setCoordinate(String coordinate) {
        if (ModelBindingManager.getGlobalOption() != null && ModelBindingManager.getGlobalOption().isIgnoreCoordinate()) {
            return;
        }
        this.coordinate.setLength(0);
        this.coordinate.append(coordinate);
    }

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