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

import gudusoft.gsqlparser.dlineage.dataflow.model.json.Coordinate;
import gudusoft.gsqlparser.util.SQLUtil;

import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlTransient;
import javax.xml.bind.annotation.XmlType;
import java.util.Comparator;
import java.util.Objects;
import java.util.TreeSet;

@XmlType(propOrder = { "id", "name", "displayName", "coordinate", "source", "qualifiedTable", "isFunction", "dataType", "primaryKey", "unqiueKey", "indexKey", "foreignKey" })
public class column {
        private gudusoft.gsqlparser.sqlenv.CanonKey identifierKey;

        /** Producer-only identity; never serialized or reconstructed from display text. */
        @XmlTransient
        public gudusoft.gsqlparser.sqlenv.CanonKey getIdentifierKey() {
                return "system".equals(source) ? null : identifierKey;
        }

        public void setIdentifierKey(gudusoft.gsqlparser.sqlenv.CanonKey identifierKey) {
                this.identifierKey = identifierKey;
        }

        private Object fieldPathKey;

        /**
         * Producer-only: the structural identity of a STRUCT field path node, its declared column and its path, by which
         * an output merge tells two field nodes apart whose displays collide ({@code s."a.b".c} and {@code s.a.b.c}) and
         * keeps one node for two spellings of one field; null for any other column. Never serialized (plan section 4.7).
         */
        @XmlTransient
        public Object getFieldPathKey() {
                return fieldPathKey;
        }

        public void setFieldPathKey(Object fieldPathKey) {
                this.fieldPathKey = fieldPathKey;
        }

        private String sqlSpelling;

        /**
         * Producer-only: the column's spelling as SQL when the producer knows it (the column a star link of an identity
         * names), from which an output merge gives two columns of one table whose names collide names that tell them
         * apart. Never serialized.
         */
        @XmlTransient
        public String getSqlSpelling() {
                return sqlSpelling;
        }

        public void setSqlSpelling(String sqlSpelling) {
                this.sqlSpelling = sqlSpelling;
        }

        private String id;

        private String name;

        private String displayName;

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

        private StringBuffer coordinate = new StringBuffer();

        private String source;

        private String qualifiedTable;

        private String isFunction;

        private String dataType;

        private Boolean primaryKey;

        private Boolean unqiueKey;

        private Boolean indexKey;

        private Boolean foreignKey;

        private Boolean nullable;

        private Boolean identity;

        private Boolean computed;

        private Boolean persisted;

        private String computedDefinition;

        private String defaultValue;

        private String defaultName;

        private Boolean star;

        /**
         * Whether this column stands for a star ({@code *}, {@code t.*}, a whole-row pseudo column). Output version 1
         * names a star {@code *}; a column NAMED {@code *} (SQL Server {@code [*]}, BigQuery {@code `*`}) is shown the
         * same way, so the document says {@code star="false"} on it. Never decided from the name alone.
         */
        public boolean standsForStar() {
                return "*".equals(name) && !Boolean.FALSE.equals(star);
        }

        /** The star marker as written: {@code false} on a column named {@code *} that is not a star, else absent. */
        @XmlAttribute(name = "star", required = false)
        public Boolean getStar() {
                return star;
        }

        public void setStar(Boolean star) {
                this.star = star;
        }

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

        public void appendCoordinate(String coordinate) {
                if(SQLUtil.isEmpty(coordinate)){
                        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(SQLUtil.isEmpty(coordinate)){
                        return;
                }
                this.coordinate.setLength(0);
                this.coordinate.append(coordinate);
        }
        
    public void clearCoordinate() {
        this.coordinate = new StringBuffer();
    }


        @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 String getSource() {
                return source;
        }

        public void setSource(String source) {
                this.source = source;
        }

        @XmlAttribute(required = false)
        public String getQualifiedTable() {
                return qualifiedTable;
        }

        public void setQualifiedTable(String qualifiedTable) {
                this.qualifiedTable = qualifiedTable;
        }

        @Override
        public int hashCode() {
                final int prime = 31;
                int result = 1;
                result = prime * result + ((coordinate == null) ? 0 : coordinate.toString().hashCode());
                result = prime * result + ((id == null) ? 0 : id.hashCode());
                result = prime * result + ((name == null) ? 0 : name.hashCode());
                result = prime * result + ((qualifiedTable == null) ? 0 : qualifiedTable.hashCode());
                result = prime * result + ((source == null) ? 0 : source.hashCode());
                return result;
        }

        @Override
        public boolean equals(Object obj) {
                if (this == obj)
                        return true;
                if (obj == null)
                        return false;
                if (getClass() != obj.getClass())
                        return false;
                column other = (column) obj;
                if (coordinate == null) {
                        if (other.coordinate != null)
                                return false;
                } else if (!Objects.equals(coordinate.toString(),other.coordinate.toString()))
                        return false;
                if (id == null) {
                        if (other.id != null)
                                return false;
                } else if (!id.equals(other.id))
                        return false;
                if (name == null) {
                        if (other.name != null)
                                return false;
                } else if (!name.equals(other.name))
                        return false;
                if (qualifiedTable == null) {
                        if (other.qualifiedTable != null)
                                return false;
                } else if (!qualifiedTable.equals(other.qualifiedTable))
                        return false;
                if (source == null) {
                        if (other.source != null)
                                return false;
                } else if (!source.equals(other.source))
                        return false;
                return true;
        }

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

        public void setIsFunction(String isFunction) {
                this.isFunction = isFunction;
        }

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

        @XmlAttribute(name = "primaryKey", required = false)
        public Boolean isPrimaryKey() {
                return primaryKey;
        }

        public void setPrimaryKey(Boolean primaryKey) {
                if(primaryKey == null || Boolean.FALSE.equals(primaryKey)){
                        return;
                }
                this.primaryKey = primaryKey;
        }

        @XmlAttribute(name = "unqiueKey", required = false)
        public Boolean isUnqiueKey() {
                return unqiueKey;
        }

        public void setUnqiueKey(Boolean unqiueKey) {
                if(unqiueKey == null || Boolean.FALSE.equals(unqiueKey)){
                        return;
                }
                this.unqiueKey = unqiueKey;
        }

        @XmlAttribute(name = "indexKey", required = false)
        public Boolean isIndexKey() {
                return indexKey;
        }

        public void setIndexKey(Boolean indexKey) {
                if(indexKey == null || Boolean.FALSE.equals(indexKey)){
                        return;
                }
                this.indexKey = indexKey;
        }

        @XmlAttribute(name = "foreignKey", required = false)
        public Boolean isForeignKey() {
                return foreignKey;
        }

        public void setForeignKey(Boolean foreignKey) {
                if(foreignKey == null || Boolean.FALSE.equals(foreignKey)){
                        return;
                }
                this.foreignKey = foreignKey;
        }

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

        public void setDataType(String dataType) {
                this.dataType = dataType;
        }

        @XmlAttribute(name = "nullable", required = false)
        public Boolean isNullable() {
                return nullable;
        }

        public void setNullable(Boolean nullable) {
                this.nullable = nullable;
        }

        @XmlAttribute(name = "identity", required = false)
        public Boolean isIdentity() {
                return identity;
        }

        public void setIdentity(Boolean identity) {
                this.identity = identity;
        }

        @XmlAttribute(name = "computed", required = false)
        public Boolean isComputed() {
                return computed;
        }

        public void setComputed(Boolean computed) {
                this.computed = computed;
        }

        @XmlAttribute(name = "persisted", required = false)
        public Boolean isPersisted() {
                return persisted;
        }

        public void setPersisted(Boolean persisted) {
                this.persisted = persisted;
        }

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

        public void setComputedDefinition(String computedDefinition) {
                this.computedDefinition = computedDefinition;
        }

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

        public void setDefaultValue(String defaultValue) {
                this.defaultValue = defaultValue;
        }

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

        public void setDefaultName(String defaultName) {
                this.defaultName = defaultName;
        }
}