Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.modeling / src / org / simantics / modeling / migration / DocumentCleanupMigrationStep.java

diff --git a/bundles/org.simantics.modeling/src/org/simantics/modeling/migration/DocumentCleanupMigrationStep.java b/bundles/org.simantics.modeling/src/org/simantics/modeling/migration/DocumentCleanupMigrationStep.java
new file mode 100644 (file)
index 0000000..816b03f
--- /dev/null
+++ b/bundles/org.simantics.modeling/src/org/simantics/modeling/migration/DocumentCleanupMigrationStep.java
@@ -0,0 +1,48 @@
+package org.simantics.modeling.migration;\r
+\r
+import org.eclipse.core.runtime.IProgressMonitor;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.Session;\r
+import org.simantics.db.WriteGraph;\r
+import org.simantics.db.common.request.WriteRequest;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.migration.MigrationState;\r
+import org.simantics.db.layer0.migration.MigrationStep;\r
+import org.simantics.db.layer0.migration.MigrationUtils;\r
+import org.simantics.db.layer0.util.RemoverUtil;\r
+import org.simantics.document.DocumentResource;\r
+import org.simantics.layer0.Layer0;\r
+import org.simantics.modeling.ModelingUtils;\r
+\r
+public class DocumentCleanupMigrationStep implements MigrationStep {\r
+\r
+    @Override\r
+    public void applyTo(IProgressMonitor monitor, Session session, MigrationState state) throws DatabaseException {\r
+       \r
+        final Resource indexRoot = MigrationUtils.getResource(monitor, session, state);\r
+        if(indexRoot == null) return;\r
+\r
+        session.sync(new WriteRequest() {\r
+               \r
+            @Override\r
+            public void perform(WriteGraph graph) throws DatabaseException {\r
+               \r
+               Layer0 L0 = Layer0.getInstance(graph);\r
+               DocumentResource DOC = DocumentResource.getInstance(graph);\r
+               \r
+               for(Resource model : graph.getObjects(indexRoot, L0.ConsistsOf)) {\r
+                               for(Resource instance : ModelingUtils.searchByType(graph, model, DOC.ScenegraphDocument)) {\r
+                                       if(!graph.hasStatement(instance, DOC.HasDocumentation_Inverse)) {\r
+                                               System.err.println("Removing stray document " + graph.getURI(instance));\r
+                                               RemoverUtil.remove(graph, instance);\r
+                                       }\r
+                               }\r
+               }\r
+\r
+            }\r
+\r
+        });\r
+\r
+    }\r
+\r
+}\r