X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.fileimport%2Fsrc%2Forg%2Fsimantics%2Ffileimport%2Fdropins%2FFileImportDropins.java;fp=bundles%2Forg.simantics.fileimport%2Fsrc%2Forg%2Fsimantics%2Ffileimport%2Fdropins%2FFileImportDropins.java;h=a32dd98775639f0f15b5224d011b72b54e7f17b2;hb=96bb7ef9cbe42d82eb58306d8f9b62392cc29ba8;hp=0000000000000000000000000000000000000000;hpb=ae5bb63c5c88f6569518fed2a24df86fbd0570ff;p=simantics%2Fplatform.git

diff --git a/bundles/org.simantics.fileimport/src/org/simantics/fileimport/dropins/FileImportDropins.java b/bundles/org.simantics.fileimport/src/org/simantics/fileimport/dropins/FileImportDropins.java
new file mode 100644
index 000000000..a32dd9877
--- /dev/null
+++ b/bundles/org.simantics.fileimport/src/org/simantics/fileimport/dropins/FileImportDropins.java
@@ -0,0 +1,140 @@
+package org.simantics.fileimport.dropins;
+
+import static java.nio.file.StandardWatchEventKinds.OVERFLOW;
+import static java.nio.file.StandardWatchEventKinds.ENTRY_CREATE;
+import static java.nio.file.StandardWatchEventKinds.ENTRY_DELETE;
+import static java.nio.file.StandardWatchEventKinds.ENTRY_MODIFY;
+
+import java.io.IOException;
+import java.nio.file.FileSystem;
+import java.nio.file.FileVisitResult;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.SimpleFileVisitor;
+import java.nio.file.WatchEvent;
+import java.nio.file.WatchEvent.Kind;
+import java.nio.file.attribute.BasicFileAttributes;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Optional;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.nio.file.WatchKey;
+import java.nio.file.WatchService;
+
+import org.simantics.fileimport.Activator;
+import org.simantics.fileimport.FileImportService;
+
+public class FileImportDropins {
+    
+    private static Thread watcherThread = null;
+    private static DropinsFolderWatcher watcher = null;
+
+    public static void watchDropinsFolder() {
+        if (watcher == null && watcherThread == null) {
+            try {
+                watcher = new DropinsFolderWatcher(Activator.getDropinsFolder());
+                watcherThread = new Thread(watcher, "Simantics Dropins Folder watcher thread");
+                watcherThread.setDaemon(true);
+                watcherThread.start();
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+    
+    public static void unwatchDropinsFolder() {
+        watcher.stop();
+        try {
+            watcherThread.join(500);
+            if (watcherThread.isAlive())
+                watcherThread.interrupt();
+        } catch (InterruptedException e) {
+            e.printStackTrace();
+        }
+        watcherThread = null;
+        watcher = null;
+    }
+    
+    private static class DropinsFolderWatcher implements Runnable {
+
+        private final Path dropinsFolder;
+        private final WatchService ws;
+        private final AtomicBoolean stopped = new AtomicBoolean(true);
+        
+        private final Map<WatchKey, Path> keys = new HashMap<>();
+        
+        public DropinsFolderWatcher(Path dropinsFolder) throws IOException {
+            this.dropinsFolder = dropinsFolder;
+            FileSystem fs = dropinsFolder.getFileSystem();
+            this.ws = fs.newWatchService();
+            registerAll(this.dropinsFolder);
+        }
+        
+        @Override
+        public void run() {
+            stopped.set(false);
+
+            while (!stopped.get()) {
+                try {
+                    WatchKey key = ws.take();
+                    for (WatchEvent<?> watchEvent : key.pollEvents()) {
+                        if (OVERFLOW == watchEvent.kind())
+                            continue; // loop
+                        
+                        @SuppressWarnings("unchecked")
+                        WatchEvent<Path> pathEvent = (WatchEvent<Path>) watchEvent;
+                        Kind<Path> kind = pathEvent.kind();
+                        
+                        Path parent = keys.get(key);
+                        Path newPath = parent.resolve(pathEvent.context());
+                        if (FileImportService.DB_FILE.equals(newPath.getFileName().toString()))
+                            continue;
+                        if (ENTRY_CREATE == kind) {
+                            System.out.println("New path created: " + newPath);
+                            FileImportService.performFileImport(newPath, Optional.empty());
+                            register(newPath);
+                        } else if (ENTRY_MODIFY == kind) {
+                            System.out.println("New path modified: " + newPath);
+                        } else if (ENTRY_DELETE == kind) {
+                            System.out.println("New path deleted: " + newPath);
+                            FileImportService.removeResourceForFile(newPath.toAbsolutePath(), Optional.empty());
+                        }
+                    }
+                    if (!key.reset()) {
+                        keys.remove(key);
+//                        break; // loop
+                    }
+                } catch (IOException e) {
+                    e.printStackTrace();
+                } catch (InterruptedException e) {
+                    if (!stopped.get())
+                        e.printStackTrace();
+                } catch (Throwable t) {
+                    t.printStackTrace();
+                }
+            }
+        }
+        
+        public void stop() {
+            stopped.set(true);
+        }
+        
+        private void registerAll(Path path) throws IOException {
+            Files.walkFileTree(path, new SimpleFileVisitor<Path>() {
+                
+                @Override
+                public FileVisitResult preVisitDirectory(Path file, BasicFileAttributes attrs) throws IOException {
+                    register(file);
+                    return FileVisitResult.CONTINUE;
+                }
+            });
+        }
+
+        private void register(Path path) throws IOException {
+            if (Files.isDirectory(path)) {
+                WatchKey key = path.toAbsolutePath().register(ws, ENTRY_CREATE, ENTRY_DELETE, ENTRY_MODIFY);
+                keys.put(key, path);
+            }
+        }
+    }
+}