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

import java.util.Comparator;
import java.util.LinkedHashSet;
import java.util.Set;

import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlElementWrapper;
import javax.xml.bind.annotation.XmlTransient;
import javax.xml.bind.annotation.XmlType;

import gudusoft.gsqlparser.TSourceToken;
import gudusoft.gsqlparser.dlineage.dataflow.model.ModelBindingManager;
import gudusoft.gsqlparser.dlineage.dataflow.model.Transform;
import gudusoft.gsqlparser.dlineage.util.DlineageUtil;
import gudusoft.gsqlparser.dlineage.util.LegacyDlineageWireNamesV1;
import gudusoft.gsqlparser.dlineage.util.Pair3;
import gudusoft.gsqlparser.util.SQLUtil;

@XmlType(propOrder = { "id", "name", "column", "value", "source_id", "source_name", "column_type", "parent_id", "parent_name", "parent_alias",
                "coordinate", "clauseType", "source", "type", "transforms", "candidateParents"})
public class sourceColumn {

        private String id;

        private String name;
        
        private String column;

        private String value;

        private String source_id;

        private String source_name;

        private String column_type;

        private String parent_id;

        private String parent_name;
        
        private String parent_alias;

        private String coordinate;

        private String clauseType;

        private String source;
        
        private String type;
        
        private Set<transform> transforms;
        
        private Set<candidateTable> candidateParents;
        
        @XmlTransient
        private Boolean struct;

        @XmlTransient
        private boolean positionUnproven;

        private String dedupCoordinate;

        @XmlElementWrapper(name="candidateTables") 
        @XmlElement(name="candidateTable", required = false)
        public Set<candidateTable> getCandidateParents() {
                if (ModelBindingManager.getGlobalOption()!=null && !ModelBindingManager.getGlobalOption().isShowCandidateTable()) {
                        return null;
                }
                return candidateParents;
        }
        
        public void setCandidateParents(Set<candidateTable> candidateParents) {
                if (ModelBindingManager.getGlobalOption()!=null && !ModelBindingManager.getGlobalOption().isShowCandidateTable()) {
                        return;
                }
                this.candidateParents = candidateParents;
        }
        
        public void addCandidateParent(candidateTable candidateParent) {
                if (ModelBindingManager.getGlobalOption()!=null && !ModelBindingManager.getGlobalOption().isShowCandidateTable()) {
                        return;
                }
                if(candidateParent == null)
                        return;
                if(candidateParents == null) {
                        candidateParents = new LinkedHashSet<candidateTable>();
                }
                candidateParents.add(candidateParent);
        }
        
        @XmlElementWrapper(name="transforms") 
        @XmlElement(name="transform", required = false)
        public Set<transform> getTransforms() {
                if (ModelBindingManager.getGlobalOption()!=null && !ModelBindingManager.getGlobalOption().isTransform()) {
                        return null;
                }
                return transforms;
        }
        
        public void setTransforms(Set<transform> transforms) {
                if (ModelBindingManager.getGlobalOption()!=null && !ModelBindingManager.getGlobalOption().isTransform()) {
                        return;
                }
                this.transforms = transforms;
        }
        
        public void addTransform(Transform transform) {
                if (ModelBindingManager.getGlobalOption()!=null && !ModelBindingManager.getGlobalOption().isTransform()) {
                        return;
                }
                if(transform == null)
                        return;
                if(transforms == null) {
                        transforms = new LinkedHashSet<transform>();
                }
                
                transform t = new transform();
                if (transform.getCode() != null) {
                        if (ModelBindingManager.getGlobalOption().isTransformCoordinate()) {
                                TSourceToken startToken = transform.getCode().getStartToken();
                                TSourceToken endToken = transform.getCode().getEndToken();
                                Pair3<Long, Long, String> startPosition = new Pair3<Long, Long, String>(startToken.lineNo,
                                                startToken.columnNo, ModelBindingManager.getGlobalHash());
                                Pair3<Long, Long, String> endPosition = new Pair3<Long, Long, String>(endToken.lineNo,
                                                endToken.columnNo + SQLUtil.endTrim(endToken.getAstext()).length(), ModelBindingManager.getGlobalHash());
                                t.setCoordinate(convertCoordinate(startPosition) + "," + convertCoordinate(endPosition));
                        }
                        t.setCode(transform.getCode().toString());
                } else if (!SQLUtil.isEmpty(transform.getCodeString())) {
                        if (ModelBindingManager.getGlobalOption().isTransformCoordinate()) {
                                TSourceToken startToken = transform.getStartToken();
                                TSourceToken endToken = transform.getEndToken();
                                Pair3<Long, Long, String> startPosition = new Pair3<Long, Long, String>(startToken.lineNo,
                                                startToken.columnNo, ModelBindingManager.getGlobalHash());
                                Pair3<Long, Long, String> endPosition = new Pair3<Long, Long, String>(endToken.lineNo,
                                                endToken.columnNo + SQLUtil.endTrim(endToken.getAstext()).length(), ModelBindingManager.getGlobalHash());
                                t.setCoordinate(convertCoordinate(startPosition) + "," + convertCoordinate(endPosition));
                        }
                        t.setCode(transform.getCodeString());
                }
                t.setType(transform.getType());
                transforms.add(t);
        }
        
        public void addTransform(transform transform) {
                if (ModelBindingManager.getGlobalOption()!=null && !ModelBindingManager.getGlobalOption().isTransform()) {
                        return;
                }
                if(transform == null)
                        return;
                if(transforms == null) {
                        transforms = new LinkedHashSet<transform>();
                }
                transforms.add(transform);
        }
        
        private Pair3<Long, Long, Integer> convertCoordinate(Pair3<Long, Long, String> position) {
                return new Pair3<Long, Long, Integer>(position.first, position.second,
                                ModelBindingManager.getGlobalSqlInfo().getIndexOf(position.third));
        }
        
        @XmlAttribute(required = false)
        public String getCoordinate() {
//              if (ModelBindingManager.getGlobalOption()!=null && ModelBindingManager.getGlobalOption().isIgnoreCoordinate()) {
//                      return null;
//              }
                return coordinate;
        }

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

        @XmlAttribute(required = false)
        public String getColumn() {
                return column;
        }

        public void setColumn(String column) {
                this.column = column;
        }

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

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

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

        public void setParent_id(String parent_id) {
                this.parent_id = parent_id;
        }

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

        public void setParent_name(String parent_name) {
                this.parent_name = parent_name;
        }
        
        @XmlAttribute(name = "parent_alias", required = false)
        public String getParent_alias() {
                return parent_alias;
        }

        public void setParent_alias(String parent_alias) {
                this.parent_alias = parent_alias;
        }

        @XmlAttribute(required = false)
        public String getValue() {
                return value;
        }

        public void setValue(String value) {
                this.value = value;
        }

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

        public void setSource_name(String source_name) {
                this.source_name = source_name;
        }

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

        public void setSource_id(String source_id) {
                this.source_id = source_id;
        }

        @XmlAttribute(required = false)
        public String getClauseType() {
                return clauseType;
        }

        public void setClauseType(String clauseType) {
                this.clauseType = clauseType;
        }

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

        public void setColumn_type(String column_type) {
                this.column_type = column_type;
        }

        @XmlAttribute(required = false)
        public String getSource() {
                return source;
        }

        public void setSource(String source) {
                this.source = source;
        }
        
        @XmlAttribute(required = false)
        public String getType() {
                return type;
        }

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

        @Override
        public int hashCode() {
                final int prime = 31;
                int result = 1;
                String columnName = getColumnName(this);
                result = prime * result + ((columnName == null) ? 0 : columnName.hashCode());
                result = prime * result + ((coordinate == null) ? 0 : coordinate.hashCode());
                result = prime * result + ((parent_id == null) ? 0 : parent_id.hashCode());
                return result;
        }

        private String getColumnName(sourceColumn column) {
                if ("constant".equals(column.getColumn_type())) {
                        return column.column;
                }
                if (ModelBindingManager.getGlobalVendor() != null) {
                        // equal published endpoints: the label's output-v1 bucket, beside the coordinate and the parent
                        return LegacyDlineageWireNamesV1.columnBucket(column.column);
                }
                return column.column;
        }

        @Override
        public boolean equals(Object obj) {
                if (this == obj)
                        return true;
                if (obj == null)
                        return false;
                if (getClass() != obj.getClass())
                        return false;
                sourceColumn other = (sourceColumn) obj;
                if (column == null) {
                        if (other.column != null)
                                return false;
                } else if (!getColumnName(this).equals(getColumnName(other)))
                        return false;
                if (coordinate == null) {
                        if (other.coordinate != null)
                                return false;
                } else if (coordinate.hashCode() != other.coordinate.hashCode())
                        return false;
                if (parent_id == null) {
                        if (other.parent_id != null)
                                return false;
                } else if (!parent_id.equals(other.parent_id))
                        return false;
                return true;
        }

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

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

        /**
         * Whether this source is a {@code *} read at a position its members' unknown order cannot tell (Mantis 4856):
         * the simple output follows it to the {@code *} of the table that fills it, never to the column named like the
         * target. Never serialized.
         */
        @XmlTransient
        public boolean isPositionUnproven() {
                return positionUnproven;
        }

        public void setPositionUnproven(boolean positionUnproven) {
                this.positionUnproven = positionUnproven;
        }

        @XmlTransient
        public Boolean isStruct() {
                return struct;
        }

        public void setStruct(Boolean struct) {
                this.struct = struct;
        }

        public String toDedupKey() {
                StringBuilder sb = new StringBuilder(128);
                appendDedupKey(sb);
                return sb.toString();
        }

        public long toDedupHash() {
                StringBuilder sb = new StringBuilder(128);
                appendDedupKey(sb);
                return SQLUtil.hash64(sb);
        }

        /**
         * The coordinate this source would have carried before reference-site
         * publishing — the column model's definition span. The table-merge pass
         * stamps it in for its dedup steps (equals/hashCode and toDedupHash include
         * the coordinate), so sources and relations keep collapsing exactly as they
         * did when every reference was rendered with the model span, then restores
         * the published reference-site coordinate on the survivors. Never
         * serialized.
         */
        @XmlTransient
        public String getDedupCoordinate() {
                return dedupCoordinate;
        }

        public void setDedupCoordinate(String dedupCoordinate) {
                this.dedupCoordinate = dedupCoordinate;
        }

        void appendDedupKey(StringBuilder sb) {
                if (name != null) { sb.append("name=").append(name).append('|'); }
                if (column != null) { sb.append("column=").append(column).append('|'); }
                if (parent_id != null) { sb.append("parent_id=").append(parent_id).append('|'); }
                if (source_id != null) { sb.append("source_id=").append(source_id).append('|'); }
                if (value != null) { sb.append("value=").append(value).append('|'); }
                if (source_name != null) { sb.append("source_name=").append(source_name).append('|'); }
                if (column_type != null) { sb.append("column_type=").append(column_type).append('|'); }
                if (parent_name != null) { sb.append("parent_name=").append(parent_name).append('|'); }
                if (parent_alias != null) { sb.append("parent_alias=").append(parent_alias).append('|'); }
                if (coordinate != null) { sb.append("coordinate=").append(coordinate).append('|'); }
                if (clauseType != null) { sb.append("clauseType=").append(clauseType).append('|'); }
                if (source != null) { sb.append("source=").append(source).append('|'); }
                if (type != null) { sb.append("type=").append(type).append('|'); }
                if (transforms != null && !transforms.isEmpty()) {
                        sb.append("transforms=[");
                        transforms.stream().sorted(Comparator.comparing(transform::getType, Comparator.nullsFirst(Comparator.<String>naturalOrder())))
                                        .forEach(t -> { sb.append('{'); t.appendDedupKey(sb); sb.append("},"); });
                        sb.setLength(sb.length() - 1);
                        sb.append("]|");
                }
                if (candidateParents != null && !candidateParents.isEmpty()) {
                        sb.append("candidateParents=[");
                        candidateParents.stream().sorted(Comparator.comparing(candidateTable::getId, Comparator.nullsFirst(Comparator.<String>naturalOrder())))
                                        .forEach(c -> { sb.append('{'); c.appendDedupKey(sb); sb.append("},"); });
                        sb.setLength(sb.length() - 1);
                        sb.append("]|");
                }
        }
}