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

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