Replace file extension properly in FixExportedOntology

[simantics/platform.git] / bundles / org.simantics.graph / src / org / simantics / graph / refactoring / FixExportedOntology.java

diff --git a/bundles/org.simantics.graph/src/org/simantics/graph/refactoring/FixExportedOntology.java b/bundles/org.simantics.graph/src/org/simantics/graph/refactoring/FixExportedOntology.java
index 09931e604bf6fdddf9ab751f20e0f910090210c6..33291cdf108e75cef32b35e64e69b19d8583c589 100644 (file)
--- a/bundles/org.simantics.graph/src/org/simantics/graph/refactoring/FixExportedOntology.java
+++ b/bundles/org.simantics.graph/src/org/simantics/graph/refactoring/FixExportedOntology.java
@@ -1,51 +1,81 @@
-package org.simantics.graph.refactoring;\r
-\r
-import java.io.BufferedInputStream;\r
-import java.io.DataInput;\r
-import java.io.DataInputStream;\r
-import java.io.InputStream;\r
-import java.nio.file.Files;\r
-import java.nio.file.Path;\r
-import java.nio.file.Paths;\r
-\r
-import org.simantics.databoard.binding.Binding;\r
-import org.simantics.databoard.binding.mutable.Variant;\r
-import org.simantics.databoard.container.DataContainer;\r
-import org.simantics.databoard.container.DataContainers;\r
-import org.simantics.graph.representation.TransferableGraph1;\r
-\r
-/**\r
- * @author Antti Villberg\r
- * @since 1.24.0\r
- */\r
-public class FixExportedOntology {\r
-       \r
-       static void convertExportedSharedOntologyIntoBundleOntology(Path input, Path output) throws Exception {\r
-               System.out.format("Converting exported shared ontology%n\t" + input.toString() + "%nto bundle-compatible ontology%n\t" + output.toString());\r
-               try (InputStream is = new BufferedInputStream(Files.newInputStream(input), 128*1024)) {\r
-                       DataInput dis = new DataInputStream(is);\r
-                       org.simantics.databoard.container.DataContainer container = \r
-                                       DataContainers.readFile(dis); \r
-                       Binding binding = TransferableGraph1.BINDING;\r
-                       TransferableGraph1 graph = (TransferableGraph1)container.content.getValue(binding);\r
-                       GraphRefactoringUtils.fixOntologyExport(graph);\r
-\r
-                       DataContainers.writeFile(output.toFile(), new DataContainer(\r
-                                       container.format, container.version,\r
-                                       container.metadata, new Variant(TransferableGraph1.BINDING, graph)));\r
-               }\r
-       }\r
-\r
-       public static void main(String[] args) throws Exception {\r
-               if (args.length < 1) {\r
-                       System.out.println("Required arguments: <input .sharedOntology file> [<output .tg file>]");\r
-               } else if (args.length < 2) {\r
-                       Path input = Paths.get(args[0]);\r
-                       Path output = input.getParent().resolve(input.getName(input.getNameCount()-1) + ".fixed");\r
-                       convertExportedSharedOntologyIntoBundleOntology(input, output);\r
-               } else {\r
-                       convertExportedSharedOntologyIntoBundleOntology(Paths.get(args[0]), Paths.get(args[1]));\r
-               }\r
-       }\r
-\r
-}\r
+package org.simantics.graph.refactoring;
+
+import java.io.BufferedInputStream;
+import java.io.DataInputStream;
+import java.io.InputStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardOpenOption;
+
+import org.simantics.databoard.binding.Binding;
+import org.simantics.databoard.container.DataContainer;
+import org.simantics.databoard.container.DataContainers;
+import org.simantics.graph.representation.PrettyPrintTG;
+import org.simantics.graph.representation.TransferableGraph1;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * @author Antti Villberg
+ * @since 1.24.0
+ */
+public class FixExportedOntology {
+
+       private static final Logger LOGGER = LoggerFactory.getLogger(FixExportedOntology.class);
+
+       static TransferableGraph1 convertExportedSharedOntologyIntoBundleOntology(Path input, Path output) throws Exception {
+               LOGGER.info("Converting exported shared ontology\n\t{}\nto bundle-compatible ontology\n\t{}", input.toString(), output.toString());
+               try (InputStream is = new BufferedInputStream(Files.newInputStream(input), 128*1024)) {
+                       Binding tgBinding = TransferableGraph1.BINDING;
+                       DataContainer container = DataContainers.readFile(new DataInputStream(is), tgBinding); 
+                       TransferableGraph1 graph = (TransferableGraph1) container.content.getValue(tgBinding);
+
+                       GraphRefactoringUtils.fixOntologyExport(graph);
+                       container = TransferableGraphHasher.addHashToTG(container, graph);
+
+                       DataContainers.writeFile(output.toFile(), container);
+                       return graph;
+               }
+       }
+
+       private static Path replaceExtension(Path p, String newExtension) {
+               String newName = p.getFileName().toString();
+               int lastDot = newName.lastIndexOf('.');
+               if (lastDot > -1)
+                       newName = newName.substring(0, lastDot);
+               return p.resolveSibling(newName + newExtension);
+       }
+
+       private static void createTg(Path input, Path output) throws Exception {
+               convertExportedSharedOntologyIntoBundleOntology(input, output);
+       }
+
+       public static void createTg(Path input) throws Exception {
+               createTg(input, replaceExtension(input, ".tg"));
+       }
+
+       public static void createTGAndPGraph(Path input) throws Exception {
+               createTGAndPGraph(input, true);
+       }
+
+       public static void createTGAndPGraph(Path input, boolean writePGraph) throws Exception {
+               TransferableGraph1 tg = convertExportedSharedOntologyIntoBundleOntology(input, replaceExtension(input, ".tg"));
+               if (writePGraph) {
+                       String listing = PrettyPrintTG.print(tg, false);
+                       Files.write(replaceExtension(input, ".pgraph"), listing.getBytes(),StandardOpenOption.CREATE);
+               }
+       }
+
+       public static void main(String[] args) throws Exception {
+               if (args.length == 0) {
+                       System.out.println("Required arguments: <input .sharedLibrary file> [<output .tg file>]");
+               } else if (args.length == 1) {
+                       Path input = Paths.get(args[0]);
+                       createTGAndPGraph(input);
+               } else {
+                       convertExportedSharedOntologyIntoBundleOntology(Paths.get(args[0]), Paths.get(args[1]));
+               }
+       }
+
+}