de.jakusys.jackhammer.cli.upload.command.WatchCommand.java Source code

Introduction

Here is the source code for de.jakusys.jackhammer.cli.upload.command.WatchCommand.java

Source

/**
 * Copyright (C) 2013 Jakob Klzer (jakob.kuelzer@gmail.com)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package de.jakusys.jackhammer.cli.upload.command;

import com.google.common.eventbus.EventBus;
import com.google.inject.Inject;
import de.jakusys.jackhammer.cli.path.Path;
import de.jakusys.jackhammer.cli.upload.event.LoggingEventListener;
import de.jakusys.jackhammer.cli.upload.event.UploadingEventListener;
import de.jakusys.jackhammer.cli.upload.handler.factory.DefaultFileHandlerFactory;
import de.jakusys.jackhammer.cli.upload.listener.JackhammerFileAlternationListener;
import de.jakusys.jackhammer.cli.util.PathRelativizer;
import io.airlift.command.Arguments;
import io.airlift.command.Command;
import io.airlift.command.Option;
import org.apache.commons.io.monitor.FileAlterationListener;
import org.apache.commons.io.monitor.FileAlterationMonitor;
import org.apache.commons.io.monitor.FileAlterationObserver;

import javax.jcr.Node;
import javax.jcr.RepositoryException;
import javax.jcr.Session;
import java.io.File;

/**
 * @author Jakob Klzer
 */
@Command(name = "watch", description = "Watch a directory for changes and automatically upload them")
public class WatchCommand implements Runnable {

    @Option(name = "--to", description = "Root path in the JCR for uploads, defaults to \"/\"")
    private String rootPath = "";

    @Arguments(title = "directory", description = "Directory to watch for filesystem changes")
    private File directory;

    @Inject
    private Session session;

    @Override
    public void run() {

        if (!directory.exists()) {
            System.out.println(
                    "Cannot watch directory " + directory.getPath() + ". The specified path could not be found.");
            return;
        }

        Node node;
        try {

            String tmp = rootPath;
            if (rootPath.startsWith("/"))
                tmp = rootPath.substring(1);

            if ("".equals(tmp)) {
                node = session.getRootNode();
            } else {
                node = session.getRootNode().getNode(tmp);
            }
        } catch (RepositoryException e) {
            throw new RuntimeException("Unable to get node", e);
        }

        final DefaultFileHandlerFactory fileHandlerFactory = new DefaultFileHandlerFactory(
                new PathRelativizer(directory));

        final EventBus eventBus = new EventBus();
        eventBus.register(new LoggingEventListener(new Path(rootPath)));
        eventBus.register(new UploadingEventListener(fileHandlerFactory, session, node));

        final FileAlterationObserver observer = new FileAlterationObserver(directory);
        final FileAlterationMonitor monitor = new FileAlterationMonitor(1000);

        PathRelativizer pathRelativizer = new PathRelativizer(directory);

        FileAlterationListener listener = new JackhammerFileAlternationListener(eventBus, pathRelativizer);

        observer.addListener(listener);

        monitor.addObserver(observer);
        try {
            System.out.println(
                    "Watching " + directory.getCanonicalPath() + " and uploading changes to " + node.getPath());
            monitor.start();

        } catch (Exception e) {
            throw new RuntimeException("Unable to monitor files", e);
        }
    }

}