org.watertemplate.i18n.developer.DirectoryWatcher Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of watertemplate-i18n-maven-plugin Show documentation
Maven plugin that internationalizes your template files
There is a newer version: 1.2.2
Show newest version
package org.watertemplate.i18n.developer;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.nio.file.*;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.HashMap;
import java.util.Map;

import static com.sun.nio.file.SensitivityWatchEventModifier.HIGH;
import static java.nio.file.LinkOption.NOFOLLOW_LINKS;
import static java.nio.file.StandardWatchEventKinds.*;

public class DirectoryWatcher {

    private final WatchService watcher;
    private final Map keys;

    private static final Logger LOGGER = LoggerFactory.getLogger(DirectoryWatcher.class);

    public DirectoryWatcher(final String... dirs) {
        try {
            this.watcher = FileSystems.getDefault().newWatchService();
            this.keys = new HashMap<>();

            for (String dir : dirs) {
                registerAll(Paths.get(dir));
            }

            LOGGER.info("Watching {}", dirs);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    public void watchRunning(final Runnable r) {
        final Thread thread = new Thread() {
            @Override
            public void run() {
                while (true) {
                    try {
                        WatchKey key = watcher.take();
                        Path dir = keys.get(key);

                        for (WatchEvent event : key.pollEvents()) {
                            WatchEvent ev = cast(event);
                            Path child = dir.resolve(ev.context());

                            if (child.toFile().isFile()) {
                                LOGGER.info("Rebuilding..");

                                try {
                                    r.run();
                                } catch (Exception e) {
                                    LOGGER.error("Error..", e);
                                }

                                if (event.kind() == ENTRY_CREATE) {
                                    if (Files.isDirectory(child, NOFOLLOW_LINKS)) {
                                        registerAll(child);
                                    }
                                }
                            }

                            boolean valid = key.reset();
                            if (!valid) {
                                keys.remove(key);
                                if (keys.isEmpty()) {
                                    break;
                                }
                            }
                        }
                    } catch (Exception ignored) {
                    }
                }
            }
        };

        thread.start();
    }

    private void registerAll(Path dirPath) throws IOException {
        final WatchEvent.Kind[] events = {ENTRY_CREATE, ENTRY_DELETE, ENTRY_MODIFY};
        Files.walkFileTree(dirPath, new SimpleFileVisitor() {
            @Override
            public FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) throws IOException {
                WatchKey key = dir.register(watcher, events, HIGH);
                keys.put(key, dir);
                return FileVisitResult.CONTINUE;
            }
        });
    }

    @SuppressWarnings("unchecked")
    private static  WatchEvent cast(WatchEvent event) {
        return (WatchEvent) event;
    }
}