/*******************************************************************************
 * Copyright (c) 2007, 2010 Association for Decentralized Information Management
 * in Industry THTH ry.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     VTT Technical Research Centre of Finland - initial API and implementation
 *******************************************************************************/
package org.simantics.modeling.ui.diagramEditor;

import java.util.Collection;
import java.util.Collections;

import org.eclipse.ui.PlatformUI;
import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
import org.simantics.db.common.request.PossibleIndexRoot;
import org.simantics.db.common.request.PossibleTypedParent;
import org.simantics.db.common.request.ReadRequest;
import org.simantics.db.exception.DatabaseException;
import org.simantics.db.layer0.variable.RVI;
import org.simantics.db.layer0.variable.Variable;
import org.simantics.db.layer0.variable.Variables;
import org.simantics.diagram.stubs.DiagramResource;
import org.simantics.g2d.canvas.ICanvasContext;
import org.simantics.g2d.diagram.DiagramHints;
import org.simantics.layer0.Layer0;
import org.simantics.modeling.ComponentUtils;
import org.simantics.modeling.actions.NavigateToTarget;
import org.simantics.modeling.ui.Activator;
import org.simantics.structural.stubs.StructuralResource2;
import org.simantics.ui.SimanticsUI;
import org.simantics.ui.workbench.editor.AbstractResourceEditorAdapter;
import org.simantics.utils.datastructures.Pair;
import org.simantics.utils.threads.ThreadUtils;

/**
 * @author Tuukka Lehtonen
 */
public class OpenDiagramFromConfigurationAdapter extends AbstractResourceEditorAdapter {

    private static final String EDITOR_ID = "org.simantics.modeling.ui.diagramEditor";

    public OpenDiagramFromConfigurationAdapter() {
        super("Diagram Editor", Activator.COMPOSITE_ICON);
    }

    protected String getEditorId() {
        return EDITOR_ID;
    }

    @Override
    public boolean canHandle(ReadGraph g, Resource r) throws DatabaseException {
    	Resource diagram = getDiagram(g, r, Collections.emptySet());
    	if(diagram == null)
    	    return false;
    	
    	// Check if the diagram is locked, because of the locked component type
    	return !isLocked(g, diagram);
    	
//        return ComponentUtils.compositeHasDiagram(g, r) /*|| ComponentUtils.componentHasDiagram(g, r)*/;
    }
    
    public static boolean isLocked(ReadGraph g, Resource diagram) throws DatabaseException {
        StructuralResource2 STR = StructuralResource2.getInstance(g);
        Resource componentType = g.syncRequest(new PossibleTypedParent(diagram, STR.ComponentType));
        if(componentType == null)
            return false; // Not part of a component type
        
        return g.hasStatement(componentType, STR.ComponentType_Locked);
    }

    @Override
    public void openEditor(final Resource r) throws Exception {
        SimanticsUI.getSession().asyncRequest(new ReadRequest() {
            @Override
            public void run(ReadGraph g) throws DatabaseException {
                openEditor(g, r, getEditorId());
            }
        });
    }

    /**
     * @param g
     * @param configurationComposite
     * @param editorId
     * @throws DatabaseException
     */
    public static boolean openEditor(ReadGraph g, Resource configurationComposite, final String editorId) throws DatabaseException {
        return openEditor(g, configurationComposite, editorId, Collections.emptySet());
    }

    protected static Resource getDiagram(ReadGraph graph, Resource r, final Collection<Object> selectedObjects) throws DatabaseException {

    	Layer0 L0 = Layer0.getInstance(graph);
    	DiagramResource DIA = DiagramResource.getInstance(graph);
    	
    	if(graph.isInstanceOf(r, DIA.Diagram)) return r;
    	
        Resource diagram = ComponentUtils.getPossibleCompositeDiagram(graph, r);
        if(diagram != null) return diagram;
        
        if(selectedObjects.size() == 1) {
        	Object o = selectedObjects.iterator().next();
        	if(o instanceof Resource) {
        		Resource res = (Resource)o;
        		if(graph.isInstanceOf(res, DIA.Element)) {
        			return graph.getPossibleObject(res, L0.PartOf);
        		}
        	}
        }
        
        return null;

    }

    public static Pair<Resource, RVI> getModelAndRVI(ReadGraph graph, Resource configurationComposite) throws DatabaseException {
        Variable compositeVariable = Variables.getPossibleVariable(graph, configurationComposite);
        if (compositeVariable == null)
            return null;

        Layer0 L0 = Layer0.getInstance(graph);
        StructuralResource2 STR = StructuralResource2.getInstance(graph);
        boolean isComponentType = graph.hasStatement(configurationComposite, STR.Defines);
        Resource rviContext = graph.syncRequest(new PossibleTypedParent(configurationComposite, L0.RVIContext));

        if (isComponentType || rviContext == null) {
            Resource indexRoot = graph.sync(new PossibleIndexRoot(configurationComposite));
            if (indexRoot == null)
                return null;

            return Pair.make(indexRoot, null);
        }

        final Resource model = Variables.getPossibleModel(graph, compositeVariable);
        if (model == null)
            return null;
        final RVI rvi = compositeVariable.getPossibleRVI(graph);

        return Pair.make(model, rvi);
    }

    /**
     * @param g
     * @param configurationComposite
     * @param editorId
     * @param selectedObjects
     * @throws DatabaseException
     */
    public static boolean openEditor(ReadGraph g, Resource r, final String editorId, final Collection<Object> selectedObjects) throws DatabaseException {

        final Resource diagram = getDiagram(g, r, selectedObjects);
        if (diagram == null)
            return false;

        final Resource configurationComposite = ComponentUtils.getPossibleDiagramComposite(g, diagram);
        if (configurationComposite == null)
            return false;

        Pair<Resource, RVI> modelAndRVI = getModelAndRVI(g, configurationComposite);
        //System.out.println("modelAndRVI: " + modelAndRVI);
        if (modelAndRVI == null)
            return false;

        final Runnable editorActivator = NavigateToTarget.editorActivator(editorId, diagram, modelAndRVI.first, modelAndRVI.second, part -> {
            if (selectedObjects.isEmpty())
                return;

            final ICanvasContext openedCanvas = (ICanvasContext) part.getAdapter(ICanvasContext.class);
            assert openedCanvas != null;
            // CanvasContext-wide denial of initial zoom-to-fit on diagram open.
            openedCanvas.getDefaultHintContext().setHint(DiagramHints.KEY_INITIAL_ZOOM_TO_FIT, Boolean.FALSE);
            //System.out.println("ASDF: " + element);
            ThreadUtils.asyncExec(openedCanvas.getThreadAccess(),
                    NavigateToTarget.elementSelectorZoomer(openedCanvas, selectedObjects, false));
        });

        PlatformUI.getWorkbench().getDisplay().asyncExec(new Runnable() {
            @Override
            public void run() {
                editorActivator.run();
            }
        });

        return true;
    }

}