Support dependency embedding in generic model export/import

[simantics/platform.git] / bundles / org.simantics.db.layer0 / src / org / simantics / db / layer0 / util / ModelDependenciesBean.java

diff --git a/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/util/ModelDependenciesBean.java b/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/util/ModelDependenciesBean.java
index d4028bd1d6353fea1f8d2ce17da2162600de3c5c..f448340dd8847300ad3564560e7ab3b3fadaa744 100644 (file)
--- a/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/util/ModelDependenciesBean.java
+++ b/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/util/ModelDependenciesBean.java
@@ -1,22 +1,81 @@
-package org.simantics.db.layer0.util;\r
-\r
-import org.simantics.databoard.Bindings;\r
-import org.simantics.databoard.binding.Binding;\r
-\r
-public class ModelDependenciesBean {\r
-\r
-       public static final String EXTENSION_KEY = ModelDependenciesBean.class.getSimpleName();\r
-       \r
-       public static final Binding BINDING = Bindings.getBindingUnchecked(ModelDependenciesBean.class);\r
-       \r
-       public ModelDependency[] dependencies;\r
-       \r
-       public ModelDependenciesBean() {\r
-               dependencies = new ModelDependency[0];\r
-       }\r
-\r
-       public ModelDependenciesBean(ModelDependency[] dependencies) {\r
-               this.dependencies = dependencies;\r
-       }\r
-       \r
-}\r
+package org.simantics.db.layer0.util;
+
+import java.util.LinkedList;
+import java.util.Map;
+import java.util.Set;
+
+import org.simantics.databoard.Bindings;
+import org.simantics.databoard.adapter.AdaptException;
+import org.simantics.databoard.binding.Binding;
+import org.simantics.databoard.binding.mutable.Variant;
+import org.simantics.db.ReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.common.request.ObjectsWithType;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.layer0.adapter.CopyHandler;
+import org.simantics.db.layer0.migration.MigrationState;
+import org.simantics.db.layer0.migration.MigrationStateKeys;
+import org.simantics.db.layer0.util.SimanticsClipboard.Representation;
+import org.simantics.graph.representation.TransferableGraph1;
+import org.simantics.layer0.Layer0;
+
+public class ModelDependenciesBean {
+
+       public static final String EXTENSION_KEY = ModelDependenciesBean.class.getSimpleName();
+
+       public static final Binding BINDING = Bindings.getBindingUnchecked(ModelDependenciesBean.class);
+
+       public ModelDependency[] dependencies;
+
+       public ModelDependenciesBean() {
+               dependencies = new ModelDependency[0];
+       }
+
+       public ModelDependenciesBean(ModelDependency[] dependencies) {
+               this.dependencies = dependencies;
+       }
+
+       private static void collectDependencies(ReadGraph graph, Resource resource, LinkedList<ModelDependency> modelDependencies) throws DatabaseException {
+
+               Layer0 L0 = Layer0.getInstance(graph);
+               libs: for(Resource library : graph.syncRequest(new ObjectsWithType(resource, L0.IsLinkedTo, L0.SharedOntology))) {
+                       String uri = graph.getPossibleURI(library);
+                       if(uri == null) continue;
+                       for(ModelDependency dep : modelDependencies)
+                               if(dep.uri.equals(uri)) continue libs;
+                       CopyHandler ch = graph.adapt(library, CopyHandler.class);
+                       SimanticsClipboardImpl clipboard = new SimanticsClipboardImpl();
+                       ch.copyToClipboard(graph, clipboard);
+                       for (Set<Representation> object : clipboard.getContents()) {
+                               TransferableGraph1 tg = ClipboardUtils.accept(graph, object, SimanticsKeys.KEY_TRANSFERABLE_GRAPH);
+                               if(tg != null) {
+                                       modelDependencies.addFirst(new ModelDependency(uri, tg));
+                               }
+                       }
+                       collectDependencies(graph, library, modelDependencies);
+               }
+
+       }
+
+       public static ModelDependenciesBean create(ReadGraph graph, Resource resource) throws DatabaseException {
+
+               LinkedList<ModelDependency> dependencies = new LinkedList<>();
+               collectDependencies(graph, resource, dependencies);
+               return new ModelDependenciesBean(dependencies.toArray(new ModelDependency[dependencies.size()]));
+
+       }
+
+       public static ModelDependenciesBean fromMigrationState(MigrationState state) throws DatabaseException {
+               Map<String,Variant> extensions = state.getProperty(MigrationStateKeys.TG_EXTENSIONS);
+               final Variant variant = extensions.get(ModelDependenciesBean.EXTENSION_KEY);
+               if (variant != null) {
+                       try {
+                               return (ModelDependenciesBean) variant.getValue(ModelDependenciesBean.BINDING);
+                       } catch (AdaptException e) {
+                               e.printStackTrace();
+                       }
+               }
+               return null;
+       }       
+       
+}