+package org.simantics.document.ui.graphfile;\r
+\r
+import java.io.BufferedReader;\r
+import java.io.File;\r
+import java.io.FileInputStream;\r
+import java.io.IOException;\r
+import java.io.InputStreamReader;\r
+import java.io.PrintStream;\r
+import java.util.Collection;\r
+import java.util.HashSet;\r
+import java.util.Set;\r
+\r
+import org.eclipse.core.runtime.IProgressMonitor;\r
+import org.simantics.Simantics;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.WriteGraph;\r
+import org.simantics.db.common.request.ReadRequest;\r
+import org.simantics.db.common.request.WriteRequest;\r
+import org.simantics.db.common.request.WriteResultRequest;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.document.DocumentResource;\r
+import org.simantics.graphfile.ontology.GraphFileResource;\r
+import org.simantics.graphfile.util.GraphFileUtil;\r
+import org.simantics.layer0.Layer0;\r
+import org.simantics.utils.ui.ExceptionUtils;\r
+\r
+/**\r
+ * \r
+ * @author Marko Luukkainen <marko.luukkainen@vtt.fi>\r
+ *\r
+ */\r
+public class FileDocumentUtil {\r
+ \r
+ /**\r
+ * Imports file, sets its L0.hasName, and and adds it to a library\r
+ * \r
+ * Note: if library relation is L0.ConsistsOf, L0.HasName is set to next available unique name.\r
+ * \r
+ * @param fileName\r
+ * @param lib\r
+ * @param rel\r
+ * @throws DatabaseException \r
+ */\r
+ public static Resource importFile(final String fileName, final Resource lib, final Resource rel) throws DatabaseException {\r
+ return Simantics.getSession().syncRequest(new WriteResultRequest<Resource>() {\r
+ @Override\r
+ public Resource perform(WriteGraph graph) throws DatabaseException {\r
+ return importFile(graph, fileName,lib,rel);\r
+ }\r
+ });\r
+ \r
+ \r
+ }\r
+ \r
+ public static void importFileAsync(final String fileName, final Resource lib, final Resource rel) {\r
+ Simantics.getSession().asyncRequest(new WriteRequest() {\r
+ \r
+ @Override\r
+ public void perform(WriteGraph graph) throws DatabaseException {\r
+ importFile(graph, fileName,lib,rel);\r
+ \r
+ }\r
+ },new org.simantics.utils.datastructures.Callback<DatabaseException>() {\r
+ \r
+ @Override\r
+ public void run(DatabaseException parameter) {\r
+ if (parameter != null)\r
+ ExceptionUtils.logAndShowError("Cannot import file " + fileName, parameter);\r
+ \r
+ }\r
+ });\r
+ \r
+ \r
+ }\r
+ \r
+ /**\r
+ * Imports file, sets its L0.HasName, and and adds it to a library\r
+ * \r
+ * Note: if library relation is L0.ConsistsOf, L0.HasName is set to next available unique name.\r
+ * \r
+ * @param graph\r
+ * @param fileName\r
+ * @param lib\r
+ * @param rel\r
+ * @throws DatabaseException\r
+ */\r
+ public static Resource importFile(WriteGraph graph, String fileName, Resource lib, Resource rel) throws DatabaseException{\r
+ Layer0 l0 = Layer0.getInstance(graph);\r
+ Resource fileResource = importFile(graph, fileName);\r
+ graph.claim(lib, rel, fileResource);\r
+ File file = new File(fileName);\r
+ String name = file.getName();\r
+ graph.claimLiteral(fileResource, l0.HasName, name);\r
+ setUniqueName(graph, fileResource, lib, rel);\r
+ return fileResource;\r
+ }\r
+ \r
+ public static Resource importFileWithName(WriteGraph graph, String fileName) throws DatabaseException{\r
+ Layer0 l0 = Layer0.getInstance(graph);\r
+ Resource fileResource = importFile(graph, fileName);\r
+ File file = new File(fileName);\r
+ String name = file.getName();\r
+ graph.claimLiteral(fileResource, l0.HasName, name);\r
+ return fileResource;\r
+ }\r
+ \r
+ /**\r
+ * Imports folder of documents recursively (including all sub folders). \r
+ * @param graph\r
+ * @param folderName Name of imported folder\r
+ * @param lib Library, where imported folder is attached.\r
+ * @param folderType Type of folders\r
+ * @param relation Relation used to create file/folder hierarchy\r
+ * @return the imported folder\r
+ * @throws DatabaseException\r
+ */\r
+ public static Resource importFolderWithName(WriteGraph graph, String folderName, Resource lib, Resource folderType, Resource relation, IProgressMonitor monitor) throws Exception{\r
+ Resource folderRes = importFolderWithName(graph, folderName, folderType, relation,monitor);\r
+ graph.claim(lib, relation, folderRes);\r
+ FileDocumentUtil.createUniqueName(graph, folderRes);\r
+ return folderRes;\r
+ }\r
+ \r
+ /**\r
+ * Imports folder of documents recursively (including all sub folders). \r
+ * @param graph\r
+ * @param folderName Name of imported folder\r
+ * @param folderType Type of folders\r
+ * @param relation Relation used to create file/folder hierarchy\r
+ * @param monitor ProgessMonitor or null\r
+ * @return the imported folder\r
+ * @throws DatabaseException\r
+ */\r
+ public static Resource importFolderWithName(WriteGraph graph, String folderName, Resource folderType, Resource relation, IProgressMonitor monitor) throws Exception{\r
+ Layer0 l0 = Layer0.getInstance(graph);\r
+ File folder = new File(folderName);\r
+ Resource rootFolderRes = graph.newResource();\r
+ graph.claim(rootFolderRes, l0.InstanceOf, folderType);\r
+ graph.claimLiteral(rootFolderRes, l0.HasName, folder.getName());\r
+ importFolder(graph, folder, rootFolderRes, relation, monitor);\r
+ return rootFolderRes;\r
+ }\r
+ \r
+ /**\r
+ * Imports folder of documents recursively (including all sub folders).\r
+ * @param graph\r
+ * @param folder Imported folder\r
+ * @param folderResource Resource folder matching file system folder\r
+ * @param relation Relation used to create file/folder hierarchy\r
+ * @throws DatabaseException\r
+ */\r
+ public static void importFolder(WriteGraph graph, File folder, Resource folderResource, Resource relation, IProgressMonitor monitor) throws Exception{\r
+ if (monitor != null) {\r
+ int count = _countFiles(folder);\r
+ monitor.beginTask("Import files", count);\r
+ }\r
+ _importFolder(graph, folder, folderResource, relation, monitor);\r
+ if (monitor != null)\r
+ monitor.done();\r
+ }\r
+ \r
+ private static void _importFolder(WriteGraph graph, File folder, Resource folderResource, Resource relation, IProgressMonitor monitor) throws Exception{\r
+ Layer0 l0 = Layer0.getInstance(graph);\r
+ File files[] = folder.listFiles();\r
+ for (File f : files) {\r
+ if (f.isDirectory()) {\r
+ Resource newFolderRes = graph.newResource();\r
+ graph.claim(newFolderRes, l0.InstanceOf, graph.getSingleType(folderResource));\r
+ graph.claim(folderResource, relation, newFolderRes);\r
+ graph.claimLiteral(newFolderRes, l0.HasName, f.getName());\r
+ _importFolder(graph, f, newFolderRes, relation,monitor);\r
+ } else {\r
+ Resource fileRes = null;\r
+ if (isUrl(f)) {\r
+ } else {\r
+ fileRes = importURL(graph, f);\r
+ fileRes = importFileWithName(graph, f.getAbsolutePath());\r
+ }\r
+ graph.claim(folderResource, relation, fileRes);\r
+ if (monitor != null)\r
+ monitor.worked(1);\r
+ }\r
+ }\r
+ }\r
+ \r
+ private static int _countFiles(File folder) {\r
+ \r
+ int count = 0;\r
+ File files[] = folder.listFiles();\r
+ for (File f : files) {\r
+ if (f.isDirectory()) {\r
+ count += _countFiles(f);\r
+ } else {\r
+ count++;\r
+ }\r
+ }\r
+ return count;\r
+ }\r
+ \r
+ \r
+ public static void createUniqueName(WriteGraph graph, Resource document) throws DatabaseException {\r
+ Layer0 l0 = Layer0.getInstance(graph);\r
+ Resource lib = graph.getPossibleObject(document, l0.PartOf);\r
+ if (lib == null)\r
+ return;\r
+ setUniqueName(graph, document, lib, l0.ConsistsOf);\r
+ }\r
+ \r
+ public static void setUniqueName(WriteGraph graph, Resource res, Resource lib, Resource rel) throws DatabaseException{\r
+ Layer0 l0 = Layer0.getInstance(graph);\r
+ Set<String> names = new HashSet<String>();\r
+ for (Resource r : graph.getObjects(lib, rel)) {\r
+ if (r.equals(res))\r
+ continue;\r
+ names.add((String)graph.getRelatedValue(r, l0.HasName));\r
+ }\r
+ String name = graph.getRelatedValue(res, l0.HasName);\r
+ if (!names.contains(name))\r
+ return;\r
+ int i = 1;\r
+ while (true) {\r
+ String proposal = name +" (" + i +")";\r
+ if (!names.contains(proposal)) {\r
+ graph.claimLiteral(res, l0.HasName, proposal);\r
+ return;\r
+ }\r
+ i++;\r
+ }\r
+ \r
+ }\r
+ \r
+ /**\r
+ * Imports a file\r
+ * \r
+ * @param graph\r
+ * @param fileName\r
+ * @return\r
+ * @throws DatabaseException\r
+ */\r
+ public static Resource importFile(WriteGraph graph, String fileName) throws DatabaseException{\r
+ Layer0 l0 = Layer0.getInstance(graph);\r
+ DocumentResource doc = DocumentResource.getInstance(graph);\r
+ \r
+ Resource fileResource = graph.newResource();\r
+ graph.claim(fileResource, l0.InstanceOf, doc.FileDocument);\r
+ try {\r
+ GraphFileUtil.toGraph(graph,fileName, fileResource);\r
+ \r
+ } catch (IOException e) {\r
+ throw new DatabaseException(e);\r
+ }\r
+ return fileResource;\r
+ \r
+ }\r
+ \r
+ /**\r
+ * Exports graph folder recursively to file system. \r
+ * @param graph\r
+ * @param folderResource\r
+ * @param folder\r
+ * @param relation\r
+ * @throws DatabaseException\r
+ */\r
+ public static void exportDocumentFolder(final Resource folderResource, final File folder, final Resource relation, final IProgressMonitor monitor) throws Exception{\r
+ Simantics.getSession().syncRequest(new ReadRequest() {\r
+ \r
+ @Override\r
+ public void run(ReadGraph graph) throws DatabaseException {\r
+ try {\r
+ exportDocumentFolder(graph, folderResource, folder, relation, monitor);\r
+ } catch (Exception e) {\r
+ throw new DatabaseException(e);\r
+ }\r
+ \r
+ }\r
+ });\r
+ }\r
+ \r
+ \r
+ /**\r
+ * Exports graph folder recursively to file system. \r
+ * @param graph\r
+ * @param folderResource\r
+ * @param folder\r
+ * @param relation\r
+ * @throws DatabaseException\r
+ */\r
+ public static void exportDocumentFolder(ReadGraph graph, Resource folderResource, File folder, Resource relation, IProgressMonitor monitor) throws Exception{\r
+ Layer0 l0 = Layer0.getInstance(graph);\r
+ DocumentResource doc = DocumentResource.getInstance(graph);\r
+ GraphFileResource gf = GraphFileResource.getInstance(graph);\r
+ Set<String> names = new HashSet<String>();\r
+ Collection<Resource> folderType = graph.getPrincipalTypes(folderResource);\r
+ for (Resource r : graph.getObjects(folderResource, relation)) {\r
+ if (graph.isInstanceOf(r, doc.Document)) {\r
+ String name = null;\r
+ boolean canExport = false;\r
+ if (graph.isInstanceOf(r, doc.FileDocument)) {\r
+ name = graph.getRelatedValue(r, gf.HasResourceName);\r
+ canExport = true;\r
+ } else if (graph.isInstanceOf(r, doc.UrlDocument)) {\r
+ name = graph.getRelatedValue(r, l0.HasName) +".url";\r
+ canExport = true;\r
+ }\r
+ if (canExport) {\r
+ name = resolveName(folder, name, names, true);\r
+ File file = new File(folder.getAbsolutePath()+"/"+name);\r
+ if (graph.isInstanceOf(r, doc.FileDocument)) {\r
+ GraphFileUtil.writeDataToFile(graph,r, file);\r
+ } else if (graph.isInstanceOf(r, doc.UrlDocument)) {\r
+ String url = graph.getRelatedValue(r, doc.HasUrl);\r
+ String n = graph.getRelatedValue(r, l0.HasName);\r
+ exportUrl(file, n, url);\r
+ }\r
+ if (monitor != null)\r
+ monitor.worked(1);\r
+ }\r
+ \r
+ } else {\r
+ Collection<Resource> type = graph.getPrincipalTypes(r);\r
+ if (type.size() == folderType.size() && folderType.containsAll(type)) {\r
+ String name = graph.getRelatedValue(r, l0.HasName);\r
+ name = resolveName(folder, name, names, false);\r
+ File subFolder = new File(folder.getAbsolutePath()+"/"+name);\r
+ if (!subFolder.exists()) {\r
+ if (!subFolder.mkdir()) {\r
+ // TODO : error.\r
+ continue;\r
+ }\r
+ }\r
+ exportDocumentFolder(graph, r, subFolder, relation,monitor);\r
+ }\r
+ }\r
+ }\r
+ }\r
+ \r
+ /**\r
+ * Print URL to a file (Windows specific format?)\r
+ * @param toFile\r
+ * @param url\r
+ * @throws DatabaseException\r
+ */\r
+ private static void exportUrl(File toFile, String name, String url) throws Exception{\r
+ PrintStream os = new PrintStream(toFile,"UTF-8");\r
+ os.println("[InternetShortcut]");\r
+ os.println("URL="+url);\r
+ os.println("name="+name);\r
+ os.flush();\r
+ os.close();\r
+ }\r
+ \r
+ public static Resource importURL(WriteGraph graph, File file) throws Exception{\r
+ String s = null;\r
+ String url = null;\r
+ String name = null;\r
+ BufferedReader is = null;\r
+ try {\r
+ is = new BufferedReader(new InputStreamReader(new FileInputStream(file), "UTF-8"));\r
+ while ((s = is.readLine()) != null) {\r
+ if (s.startsWith("URL=")) {\r
+ url = s.substring(4);\r
+ } else if (s.startsWith("name=")) {\r
+ name = s.substring(5);\r
+ }\r
+ }\r
+ } finally {\r
+ if (is != null)\r
+ is.close();\r
+ }\r
+ \r
+ if (url == null)\r
+ return null;\r
+ \r
+ Layer0 l0 = Layer0.getInstance(graph);\r
+ DocumentResource doc = DocumentResource.getInstance(graph);\r
+ \r
+ Resource fileResource = graph.newResource();\r
+ graph.claim(fileResource, l0.InstanceOf, doc.UrlDocument);\r
+ if (name == null) {\r
+ name = file.getName();\r
+ name = unescape(name);\r
+ name = name.substring(0,name.length()-4);\r
+ }\r
+ graph.claimLiteral(fileResource, l0.HasName, name);\r
+ graph.claimLiteral(fileResource, doc.HasUrl, url);\r
+ return fileResource;\r
+ }\r
+ \r
+ private static boolean isUrl(File file) throws Exception{\r
+ return (file.getAbsolutePath().endsWith(".url"));\r
+ }\r
+ \r
+ private static final char ESCAPE = '%';\r
+ \r
+ private static String escape(String s) {\r
+ \r
+ int len = s.length();\r
+ StringBuilder sb = new StringBuilder(len);\r
+ for (int i = 0; i < len; i++) {\r
+ char ch = s.charAt(i);\r
+ if (ch < ' ' || ch >= 0x7F || ch == '/' || ch == '\\' || ch == ':' || ch == ESCAPE) {\r
+ sb.append(ESCAPE);\r
+ if (ch < 0x10) {\r
+ sb.append('0');\r
+ }\r
+ sb.append(Integer.toHexString(ch));\r
+ } else {\r
+ sb.append(ch);\r
+ }\r
+ }\r
+ return sb.toString();\r
+ }\r
+ \r
+ private static String unescape(String s) {\r
+ int len = s.length();\r
+ StringBuilder sb = new StringBuilder(len);\r
+ for (int i = 0; i < len; i++) {\r
+ char ch = s.charAt(i);\r
+ if (ch == ESCAPE) {\r
+ String num = "0x";\r
+ num += s.charAt(++i);\r
+ num += s.charAt(++i);\r
+ ch = (char)Integer.decode(num).intValue();\r
+ }\r
+ sb.append(ch);\r
+ }\r
+ return sb.toString();\r
+ \r
+ }\r
+ \r
+ private static String resolveName(File parentFolder, String proposal, Set<String> used, boolean file) {\r
+ String current = escape(proposal);\r
+ int i = 0;\r
+ if (file) {\r
+ while (true) {\r
+ i++;\r
+ if (used.contains(current)) {\r
+ current = createFileName(proposal, i);\r
+ } else {\r
+ File subFile = new File(parentFolder.getAbsolutePath()+"/"+current);\r
+ if (!subFile.exists())\r
+ break;\r
+ if (subFile.exists() && subFile.isFile() && subFile.canWrite()) {\r
+ break;\r
+ }\r
+ }\r
+ }\r
+ } else {\r
+ while (true) {\r
+ i++;\r
+ if (used.contains(current)) {\r
+ current = proposal+i;\r
+ } else {\r
+ File subFolder = new File(parentFolder.getAbsolutePath()+"/"+current);\r
+ if (!subFolder.exists())\r
+ break;\r
+ if (subFolder.exists() && subFolder.isDirectory()) {\r
+ break;\r
+ }\r
+ }\r
+ }\r
+ }\r
+ used.add(current);\r
+ return current;\r
+ }\r
+ \r
+ private static String createFileName(String original, int i) {\r
+ int extIndex = original.lastIndexOf(".");\r
+ if (extIndex == -1)\r
+ return original+i;\r
+ return original.substring(0,extIndex) + i + original.substring(extIndex);\r
+ }\r
+\r
+}\r