package org.simantics.document.server;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Set;
import java.util.SortedMap;
import java.util.TreeMap;

import org.simantics.databoard.Bindings;
import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
import org.simantics.db.common.request.UnaryRead;
import org.simantics.db.exception.DatabaseException;
import org.simantics.db.layer0.request.VariableRead;
import org.simantics.db.layer0.variable.ProxyChildVariable;
import org.simantics.db.layer0.variable.Variable;
import org.simantics.db.layer0.variable.Variables;
import org.simantics.document.base.ontology.DocumentationResource;
import org.simantics.document.server.Functions.RootVariable;
import org.simantics.structural2.variables.Connection;
import org.simantics.structural2.variables.VariableConnectionPointDescriptor;
import org.simantics.utils.datastructures.Pair;
import org.simantics.utils.strings.AlphanumComparator;

public class DocumentServerUtils {

    public static Collection<Variable> getChildren(ReadGraph graph, Variable variable) throws DatabaseException {

    	DocumentationResource DOC = DocumentationResource.getInstance(graph);
    	
		ArrayList<Variable> result = new ArrayList<Variable>();
		for(Variable property : variable.getProperties(graph)) {
			Collection<String> classifications = property.getPossiblePropertyValue(graph, Variables.CLASSIFICATIONS);
			if(classifications != null) {
				if(classifications.contains(DocumentationResource.URIs.Document_ChildRelation)) {
					Connection conn = property.getValue(graph);
					Variable childConnectionPoint = DocumentServerUtils.getPossibleOtherConnectionPoint(graph, property, conn);
					if(childConnectionPoint != null) {
						result.add(childConnectionPoint.getParent(graph));
					}
				} else if (DOC.Relations_partN.equals(property.getPossiblePredicateResource(graph))) {
					Connection conn = property.getValue(graph);
					for (Variable childConnectionPoint : DocumentServerUtils.getOtherConnectionPoints(graph, property, conn)) {
						result.add(childConnectionPoint.getParent(graph));
					}
				}
			}
		}
		return result;

	}
    
	public static String findManualOrdinal(ReadGraph graph, Variable v) throws DatabaseException {
		DocumentationResource DOC = DocumentationResource.getInstance(graph);
		Integer j = null;
		while (j == null && v != null) {
			j = v.getPossiblePropertyValue(graph, DOC.Components_Component_manualOrdinal);
			v = v.getParent(graph);
		}
		if (j != null) {
			return Integer.toString(j);
		} else {
			return null;
		}
	}
    
    public static Collection<Variable> getChildrenInOrdinalOrder(ReadGraph graph, Variable variable) throws DatabaseException {
    	DocumentationResource DOC = DocumentationResource.getInstance(graph);
    	
		SortedMap<String, Variable> childMap = new TreeMap<String, Variable>(AlphanumComparator.COMPARATOR);
		
		for(Variable property : variable.getProperties(graph)) {
			Collection<String> classifications = property.getPossiblePropertyValue(graph, Variables.CLASSIFICATIONS);
			if(classifications != null) {
				if(classifications.contains(DocumentationResource.URIs.Document_ChildRelation)) {
					Resource cp = property.getPossiblePredicateResource(graph);
					String i = graph.getRelatedValue(cp, DOC.Document_ChildRelation_ordinal, Bindings.STRING);
					
					Connection conn = property.getValue(graph);
					Variable childConnectionPoint = DocumentServerUtils.getPossibleOtherConnectionPoint(graph, property, conn);
					if(childConnectionPoint != null) {
						childMap.put(i, childConnectionPoint.getParent(graph));
					}
				} else if (DOC.Relations_partN.equals(property.getPossiblePredicateResource(graph))) {
					Connection conn = property.getValue(graph);

					for (Variable childConnectionPoint : DocumentServerUtils.getOtherConnectionPoints(graph, property, conn)) {
						Variable child = childConnectionPoint.getParent(graph);
						String i = findManualOrdinal(graph, child);
						if (i == null) {
							i = "0";
						}
						childMap.put(i, child);
					}
				}
			}
		}
		return childMap.values();

	}    

	public static Collection<Variable> collectNodes(ReadGraph graph, Variable variable, Collection<Variable> nodes) throws DatabaseException {

		DocumentationResource DOC = DocumentationResource.getInstance(graph);

		Resource type = variable.getPossibleType(graph);
		if(type == null) return nodes;
		
		if(!graph.isInheritedFrom(type, DOC.Components_Component)) return nodes;

		Boolean enabled = variable.getPossiblePropertyValue(graph, DOC.Properties_exists, Bindings.BOOLEAN);
		if(enabled != null && !enabled) return nodes;

		if(graph.isInheritedFrom(type, DOC.Components_PrimitiveComponent)) {
			nodes.add(variable);
		} else {
			for(Variable child : variable.getChildren(graph)) {
				collectNodes(graph, child, nodes);
			}
		}

		return nodes;

	}

	public static Variable getPossibleOtherConnectionPoint(ReadGraph graph, Variable connectionPoint, Connection conn) throws DatabaseException {

		Collection<VariableConnectionPointDescriptor> descs = conn.getConnectionPointDescriptors(graph, null);
		if(descs.size() != 2) return null;

		for(VariableConnectionPointDescriptor desc : descs) {
			if(desc.isFlattenedFrom(graph, connectionPoint)) continue;
			return desc.getVariable(graph);
		}
		
		return null;

	}
	
	public static Collection<Variable> getOtherConnectionPoints(ReadGraph graph, Variable connectionPoint, Connection conn) throws DatabaseException {

		ArrayList<Variable> connectionPoints = new ArrayList<Variable>();
		
		Collection<VariableConnectionPointDescriptor> descs = conn.getConnectionPointDescriptors(graph, null);

		for(VariableConnectionPointDescriptor desc : descs) {
			if(desc.isFlattenedFrom(graph, connectionPoint)) continue;
			connectionPoints.add(desc.getVariable(graph));
		}
		
		return connectionPoints;

	}

	public static Variable getPossibleCommandTriggerConnectionPoint(ReadGraph graph, Variable connectionPoint, Connection conn) throws DatabaseException {

 		Collection<Variable> cpts = conn.getConnectionPoints(graph, null);

		Variable result = null;
		
		for(Variable cpt : cpts) {
			Set<String> classifications = cpt.getClassifications(graph);
			if(classifications.contains(DocumentationResource.URIs.Relations_commandExecutorRelation)) {
				if(result != null) throw new DatabaseException("Multiple executor connection points in command connection");
				result = cpt;
			}
		}
		
		return result;

	}

	public static Collection<Variable> getPossibleOtherConnectionPoints(ReadGraph graph, Variable connectionPoint, Connection conn) throws DatabaseException {

	    Collection<Variable> cpts = conn.getConnectionPoints(graph, null);
	    if(cpts.size() < 2) 
	        return Collections.emptyList();

	    ArrayList<Variable> result = new ArrayList<Variable>();
	    for(Variable cpt : cpts) {
	        if(!cpt.equals(connectionPoint)) 
	            result.add(cpt);
	    }
	    return result;
	}

	public static String getId(ReadGraph graph, Variable node) throws DatabaseException {

		if(node == null) return "root";
		else {
			String uri = node.getURI(graph); 
			int l = uri.lastIndexOf(ProxyChildVariable.CONTEXT_END);
			return uri.substring(l+4);
		}

	}

	public static class DocumentValue extends VariableRead<Object> {

		public DocumentValue(Variable variable) {
			super(variable);
		}

		@Override
		public Object perform(ReadGraph graph) throws DatabaseException {
			return variable.getValue(graph);
		}
		
		@Override
		public String toString() {
			return "DocumentValue[" + variable + "]";
		}
		
	}
	
	public static Object getValue(ReadGraph graph, Variable attrib) throws DatabaseException {
		return graph.syncRequest(new DocumentValue(attrib));
	}

	public static Variable getParentConnectionPoint(ReadGraph graph, Variable component) throws DatabaseException {

		Variable connectionPoint = component.getPossibleProperty(graph, "parent");
		if(connectionPoint == null) {
			DocumentationResource DOC = DocumentationResource.getInstance(graph);
			Collection<Variable> cps = component.getProperties(graph, DOC.Relations_parentRelation);
			if(cps.size() == 1) {
				connectionPoint = cps.iterator().next();
			} else {
				return null;
			}
		}
		
		Connection conn = connectionPoint.getValue(graph);
		Variable otherCp = DocumentServerUtils.getPossibleOtherConnectionPoint(graph, connectionPoint, conn);
		if (otherCp != null) {
			return otherCp;
		} else {
			Variable parentCp = graph.sync(new UnaryRead<Connection, Variable>(conn) {
	            @Override
	            public Variable perform(ReadGraph graph) throws DatabaseException {
	            	DocumentationResource DOC = DocumentationResource.getInstance(graph);
	            	Collection<VariableConnectionPointDescriptor> descs = parameter.getConnectionPointDescriptors(graph, null);

	        		for(VariableConnectionPointDescriptor desc : descs) {
	        			if (DOC.Relations_partN.equals(desc.getConnectionPointResource(graph))) {
	        				return desc.getVariable(graph);
	        			}
	        		}
	        		return null;
	            }
	        });
			if (parentCp != null) {
				return parentCp;
			}
		}
		return null;

	}
	
	/* Children */
	public static Collection<Variable> getChildConnections(ReadGraph graph, Variable variable) throws DatabaseException {
		return variable.getProperties(graph, DocumentationResource.getInstance(graph).Document_ChildRelation);
	}

	/* Command sequence */
	public static Collection<Variable> getTriggerCommands(ReadGraph graph, Variable variable) throws DatabaseException {
		ArrayList<Variable> result = new ArrayList<Variable>();
		DocumentationResource DOC = DocumentationResource.getInstance(graph);
		for(Variable var : variable.getProperties(graph, DOC.Document_CommandRelation)) {
            if(DOC.Relations_broadcasted.equals(var.getPredicateResource(graph))) continue;
            result.add(var);
		}
		return result;
	}

	public static Collection<Variable> getCommands(ReadGraph graph, Variable variable) throws DatabaseException {
		return variable.getProperties(graph, DocumentationResource.getInstance(graph).Document_CommandRelation);
	}

	/* Data definition */
	public static Collection<Variable> getDataDefinitions(ReadGraph graph, Variable variable) throws DatabaseException {
		return variable.getProperties(graph, DocumentationResource.getInstance(graph).Document_DataDefinitionRelation);
	}

	/* Data relation */
	public static Collection<Variable> getDataRelations(ReadGraph graph, Variable variable) throws DatabaseException {
		return variable.getProperties(graph, DocumentationResource.getInstance(graph).Document_DataRelation);
	}

	/* Attributes */
	public static Collection<Variable> getAttributes(ReadGraph graph, DocumentationResource DOC, Variable variable) throws DatabaseException {
		return variable.getProperties(graph, DOC.Document_AttributeRelation);
	}
	
	static class AttributesRequest extends VariableRead<Pair<Collection<Variable>, Collection<Variable>>> {

		public AttributesRequest(Variable variable) {
			super(variable);
		}

		@Override
		public Pair<Collection<Variable>,Collection<Variable>> perform(ReadGraph graph) throws DatabaseException {
	        ArrayList<Variable> statics = new ArrayList<Variable>();
	        ArrayList<Variable> dynamics = new ArrayList<Variable>();
	        DocumentationResource DOC = DocumentationResource.getInstance(graph);
	        for(Variable property : getAttributes(graph, DOC, variable)) {
	        	Boolean defaultProperty = property.getPossiblePropertyValue(graph, "default");
	        	if(defaultProperty != null && defaultProperty) {
//	        		System.err.println("" + property.getURI(graph) + " is default");
	        		continue;
	        	} 
//	        	else {
//	        		System.err.println("" + property.getURI(graph) + " is not default");
//	        	}
	        	Boolean dynamicProperty = property.getPossiblePropertyValue(graph, DOC.Document_AttributeRelation_dynamic);
	        	if(dynamicProperty != null && dynamicProperty) dynamics.add(property);
	        	else statics.add(property);
	        }
	        return new Pair<Collection<Variable>, Collection<Variable>>(statics, dynamics);
		}
		
	}
	
    public static Collection<Variable> getStaticAttributes(ReadGraph graph, DocumentationResource DOC, Variable variable) throws DatabaseException {
    	Pair<Collection<Variable>, Collection<Variable>> attribs = graph.syncRequest(new AttributesRequest(variable));
    	return attribs.first;
    }

	public static Collection<Variable> getDynamicAttributes(ReadGraph graph, final DocumentationResource DOC, Variable variable) throws DatabaseException {
    	Pair<Collection<Variable>, Collection<Variable>> attribs = graph.syncRequest(new AttributesRequest(variable));
    	return attribs.second;
	}
	
	public static Variable getPossibleDocumentRootVariable(ReadGraph graph, Variable documentPart) throws DatabaseException {
		if(documentPart instanceof RootVariable) return documentPart;
		Variable parent = documentPart.getParent(graph);
		if(parent == null) return null;
		return getPossibleDocumentRootVariable(graph, parent);
	}

}