]> gerrit.simantics Code Review - simantics/platform.git/blobdiff - bundles/org.simantics.document/src/org/simantics/document/DocumentDialect.java
Several Wiki documentation view improvements.
[simantics/platform.git] / bundles / org.simantics.document / src / org / simantics / document / DocumentDialect.java
index 9fc8e98a7ebb85927f2e76b13dcf8fc9b0094959..7cb6bbee0f669a1f7d2b65536edf87b014a0a475 100644 (file)
  *******************************************************************************/\r
 package org.simantics.document;\r
 \r
-import java.io.File;\r
 import java.io.IOException;\r
 import java.math.BigInteger;\r
+import java.nio.file.Files;\r
+import java.nio.file.Path;\r
 import java.security.MessageDigest;\r
 import java.security.NoSuchAlgorithmException;\r
 import java.util.regex.Matcher;\r
@@ -26,78 +27,109 @@ import org.simantics.db.Resource;
 import org.simantics.db.common.request.PossibleIndexRoot;\r
 import org.simantics.db.exception.DatabaseException;\r
 import org.simantics.image2.ontology.ImageResource;\r
-import org.simantics.utils.FileUtils;\r
-\r
 \r
 public class DocumentDialect {\r
-    \r
-    final Pattern imagePattern = Pattern.compile("\\[\\[Image(?::([^\\]]+))?\\]\\]");\r
 \r
        final static public DocumentDialect INSTANCE = new DocumentDialect();\r
 \r
-       public String transform(ReadGraph graph, Resource res, String options) {\r
-               \r
+       public static final String SIMANTICS_INTERNAL_URI_PREFIX = "http://simantics-internal/";\r
+\r
+       private static final String HTTP = "http://";\r
+       private static final String ROOT = "root:/";\r
+       private static final String IMAGE = "image";\r
+       private static final String MEDIA = "media";\r
+\r
+       final Pattern imageOrMediaPattern = Pattern.compile("\\[\\[([Ii]mage|[Mm]edia)(?::([^\\]]+))?\\]\\]");\r
+\r
+       private static String digest(byte[] bytes) throws NoSuchAlgorithmException {\r
+               MessageDigest md = MessageDigest.getInstance("MD5");\r
+               md.update(bytes);\r
+               BigInteger number = new BigInteger(1, md.digest());\r
+               return number.toString(16);\r
+       }\r
+\r
+       private static String imageExtension(ReadGraph graph, Resource image) throws DatabaseException {\r
+                  ImageResource IMAGE = ImageResource.getInstance(graph);\r
+               if (graph.isInstanceOf(image, IMAGE.PngImage))\r
+                       return ".png";\r
+               else if (graph.isInstanceOf(image, IMAGE.JpegImage))\r
+                       return ".jpg";\r
+               return null;\r
+       }\r
+\r
+       public String transform(ReadGraph graph, Resource res, String type, String options) {\r
                try {\r
-                       \r
                        String[] parts = options.split("\\|");\r
-                       if(parts.length > 0) {\r
-                               \r
+                       if (parts.length > 0) {\r
                                String uri = parts[0];\r
+\r
                                Resource indexRoot = graph.syncRequest(new PossibleIndexRoot(res));\r
-                               if(indexRoot == null) return null;\r
+                               if (indexRoot == null) return null;\r
                                String rootURI = graph.getURI(indexRoot);\r
-                               uri = uri.replace("root:/", rootURI);\r
-                               // (Apros #12268) For easier image linking, allow users to\r
-                               // write white space as ' ' in the wiki-style image links.\r
+\r
+                               uri = uri.replace(ROOT, rootURI);\r
+\r
+                               // (Apros #12268) For more user-friendly linking, allow users to\r
+                               // write white space as ' ' in the wiki-style image links instead\r
+                               // of having to write %20.\r
                                uri = uri.replace(" ", "%20");\r
-                               Resource image = graph.getPossibleResource(uri);\r
-                               if(image == null) return null;\r
-                               byte[] bytes = graph.getValue(image, Bindings.BYTE_ARRAY);\r
-                               MessageDigest md;\r
-                               md = MessageDigest.getInstance("MD5");\r
-                               md.update(bytes);\r
-                               BigInteger number = new BigInteger(1, md.digest());\r
-                               String digest = number.toString(16);\r
-                               File dir = Simantics.getTemporaryDirectory("documentImages");\r
-                               dir.mkdirs();\r
-                               \r
-                               ImageResource IMAGE = ImageResource.getInstance(graph);\r
-                               \r
-                               String extension;\r
-                               if(graph.isInstanceOf(image, IMAGE.PngImage)) extension = ".png";\r
-                               else if(graph.isInstanceOf(image, IMAGE.JpegImage)) extension = ".jpg";\r
-                               else return null;\r
-                               \r
-                               File f = new File(dir, digest + extension);\r
-                               if(!f.exists())\r
-                                       FileUtils.writeFile(f, bytes);\r
-                               StringBuilder sb = new StringBuilder();\r
-                               sb.append("[[File: " + f.toURI());\r
-                               for(int i=1;i<parts.length;i++) {\r
-                                       sb.append("|");\r
-                                       sb.append(parts[i]);\r
+\r
+                               if (IMAGE.equals(type)) {\r
+                                       Resource image = graph.getPossibleResource(uri);\r
+                                       if (image == null)\r
+                                               return null;\r
+\r
+                                       String extension = imageExtension(graph, image);\r
+                                       if (extension == null)\r
+                                               return null;\r
+\r
+                                       byte[] bytes = graph.getValue(image, Bindings.BYTE_ARRAY);\r
+                                       String digest = digest(bytes);\r
+\r
+                                       Path dir = Simantics.getTemporaryDirectory("documentImages").toPath();\r
+                                       Files.createDirectories(dir);\r
+\r
+                                       Path f = dir.resolve(digest + extension);\r
+                                       if (!Files.exists(f))\r
+                                               Files.write(f, bytes);\r
+\r
+                                       StringBuilder sb = new StringBuilder(128);\r
+                                       sb.append("[[File: ").append(f.toUri());\r
+                                       for (int i = 1; i < parts.length; i++)\r
+                                               sb.append("|").append(parts[i]);\r
+                                       sb.append("]]");\r
+                                       return sb.toString();\r
+                               } else if (MEDIA.equals(type)) {\r
+                                       Resource image = graph.getPossibleResource(uri);\r
+                                       if (image == null)\r
+                                               return null;\r
+\r
+                                       StringBuilder sb = new StringBuilder();\r
+                                       sb.append("[").append(SIMANTICS_INTERNAL_URI_PREFIX).append(uri);\r
+                                       for (int i = 1; i < parts.length; i++)\r
+                                               sb.append(" ").append(parts[i]);\r
+                                       sb.append("]");\r
+                                       return sb.toString();\r
                                }\r
-                               sb.append("]]");\r
-                               return sb.toString();\r
+\r
                        }\r
-                       \r
                } catch (DatabaseException e) {\r
                } catch (NoSuchAlgorithmException e) {\r
                } catch (IOException e) {\r
                }\r
                \r
                return null;\r
-               \r
        }\r
 \r
        public String apply(ReadGraph graph, Resource res, String markup) throws DatabaseException {\r
 \r
                StringBuffer sb = new StringBuffer();\r
-               Matcher matcher = imagePattern.matcher(markup);\r
+               Matcher matcher = imageOrMediaPattern.matcher(markup);\r
                while(matcher.find()) {\r
                        matcher.appendReplacement(sb, "");\r
-                       String options = matcher.group(1);\r
-                       String match = transform(graph, res, options);\r
+                       String type = matcher.group(1);\r
+                       String options = matcher.group(2);\r
+                       String match = transform(graph, res, type.toLowerCase(), options);\r
                        if(match != null) sb.append(match);\r
                        else sb.append("[[Image:" + options + "]]");\r
                }\r