Add possibility to stop a running insert.
[jSite.git] / src / de / todesbaum / jsite / application / ProjectInserter.java
index 48fd36b..756e121 100644 (file)
@@ -40,7 +40,6 @@ import java.util.logging.Logger;
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipOutputStream;
 
-import de.todesbaum.jsite.application.ProjectInserter.CheckReport.Issue;
 import de.todesbaum.jsite.gui.FileScanner;
 import de.todesbaum.jsite.gui.FileScannerListener;
 import de.todesbaum.util.freenet.fcp2.Client;
@@ -89,6 +88,12 @@ public class ProjectInserter implements FileScannerListener, Runnable {
        /** The temp directory. */
        private String tempDirectory;
 
+       /** The current connection. */
+       private Connection connection;
+
+       /** Whether the insert is cancelled. */
+       private volatile boolean cancelled = false;
+
        /**
         * Adds a listener to the list of registered listeners.
         *
@@ -218,12 +223,25 @@ public class ProjectInserter implements FileScannerListener, Runnable {
         * Starts the insert.
         */
        public void start() {
+               cancelled = false;
                fileScanner = new FileScanner(project);
                fileScanner.addFileScannerListener(this);
                new Thread(fileScanner).start();
        }
 
        /**
+        * Stops the current insert.
+        */
+       public void stop() {
+               cancelled = true;
+               synchronized (lockObject) {
+                       if (connection != null) {
+                               connection.disconnect();
+                       }
+               }
+       }
+
+       /**
         * Creates an input stream that delivers the given file, replacing edition
         * tokens in the file’s content, if necessary.
         *
@@ -463,7 +481,9 @@ public class ProjectInserter implements FileScannerListener, Runnable {
                List<String> files = fileScanner.getFiles();
 
                /* create connection to node */
-               Connection connection = freenetInterface.getConnection("project-insert-" + random + counter++);
+               synchronized (lockObject) {
+                       connection = freenetInterface.getConnection("project-insert-" + random + counter++);
+               }
                connection.setTempDirectory(tempDirectory);
                boolean connected = false;
                Throwable cause = null;
@@ -473,8 +493,8 @@ public class ProjectInserter implements FileScannerListener, Runnable {
                        cause = e1;
                }
 
-               if (!connected) {
-                       fireProjectInsertFinished(false, cause);
+               if (!connected || cancelled) {
+                       fireProjectInsertFinished(false, cancelled ? new AbortedException() : cause);
                        return;
                }
 
@@ -494,6 +514,7 @@ public class ProjectInserter implements FileScannerListener, Runnable {
                }
                putDir.setVerbosity(Verbosity.ALL);
                putDir.setMaxRetries(-1);
+               putDir.setEarlyEncode(false);
                for (String filename : files) {
                        FileEntry fileEntry = createFileEntry(filename, edition, containerFiles);
                        if (fileEntry != null) {
@@ -520,7 +541,7 @@ public class ProjectInserter implements FileScannerListener, Runnable {
                boolean success = false;
                boolean finished = false;
                boolean disconnected = false;
-               while (!finished) {
+               while (!finished && !cancelled) {
                        Message message = client.readMessage();
                        finished = (message == null) || (disconnected = client.isDisconnected());
                        if (firstMessage) {
@@ -549,7 +570,6 @@ public class ProjectInserter implements FileScannerListener, Runnable {
                }
 
                /* post-insert work */
-               fireProjectInsertFinished(success, disconnected ? new IOException("Connection terminated") : null);
                if (success) {
                        @SuppressWarnings("null")
                        String editionPart = finalURI.substring(finalURI.lastIndexOf('/') + 1);
@@ -557,6 +577,7 @@ public class ProjectInserter implements FileScannerListener, Runnable {
                        project.setEdition(newEdition);
                        project.setLastInsertionTime(System.currentTimeMillis());
                }
+               fireProjectInsertFinished(success, cancelled ? new AbortedException() : (disconnected ? new IOException("Connection terminated") : null));
        }
 
        //
@@ -639,72 +660,72 @@ public class ProjectInserter implements FileScannerListener, Runnable {
                        return issues.size();
                }
 
+       }
+
+       /**
+        * Container class for a single issue. An issue contains an error key
+        * that describes the error, and a fatality flag that determines whether
+        * the insert has to be aborted (if the flag is {@code true}) or if it
+        * can still be performed and only a warning should be generated (if the
+        * flag is {@code false}).
+        *
+        * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’
+        *         Roden</a>
+        */
+       public static class Issue {
+
+               /** The error key. */
+               private final String errorKey;
+
+               /** The fatality flag. */
+               private final boolean fatal;
+
+               /** Additional parameters. */
+               private String[] parameters;
+
                /**
-                * Container class for a single issue. An issue contains an error key
-                * that describes the error, and a fatality flag that determines whether
-                * the insert has to be aborted (if the flag is {@code true}) or if it
-                * can still be performed and only a warning should be generated (if the
-                * flag is {@code false}).
+                * Creates a new issue.
                 *
-                * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’
-                *         Roden</a>
+                * @param errorKey
+                *            The error key
+                * @param fatal
+                *            The fatality flag
+                * @param parameters
+                *            Any additional parameters
                 */
-               public static class Issue {
-
-                       /** The error key. */
-                       private final String errorKey;
-
-                       /** The fatality flag. */
-                       private final boolean fatal;
-
-                       /** Additional parameters. */
-                       private String[] parameters;
-
-                       /**
-                        * Creates a new issue.
-                        *
-                        * @param errorKey
-                        *            The error key
-                        * @param fatal
-                        *            The fatality flag
-                        * @param parameters
-                        *            Any additional parameters
-                        */
-                       protected Issue(String errorKey, boolean fatal, String... parameters) {
-                               this.errorKey = errorKey;
-                               this.fatal = fatal;
-                               this.parameters = parameters;
-                       }
-
-                       /**
-                        * Returns the key of the encountered error.
-                        *
-                        * @return The error key
-                        */
-                       public String getErrorKey() {
-                               return errorKey;
-                       }
+               protected Issue(String errorKey, boolean fatal, String... parameters) {
+                       this.errorKey = errorKey;
+                       this.fatal = fatal;
+                       this.parameters = parameters;
+               }
 
-                       /**
-                        * Returns whether the issue is fatal and the insert has to be
-                        * aborted. Otherwise only a warning should be shown.
-                        *
-                        * @return {@code true} if the insert needs to be aborted, {@code
-                        *         false} otherwise
-                        */
-                       public boolean isFatal() {
-                               return fatal;
-                       }
+               /**
+                * Returns the key of the encountered error.
+                *
+                * @return The error key
+                */
+               public String getErrorKey() {
+                       return errorKey;
+               }
 
-                       /**
-                        * Returns any additional parameters.
-                        *
-                        * @return The additional parameters
-                        */
-                       public String[] getParameters() {
-                               return parameters;
-                       }
+               /**
+                * Returns whether the issue is fatal and the insert has to be
+                * aborted. Otherwise only a warning should be shown.
+                *
+                * @return {@code true} if the insert needs to be aborted, {@code
+                *         false} otherwise
+                */
+               public boolean isFatal() {
+                       return fatal;
+               }
 
+               /**
+                * Returns any additional parameters.
+                *
+                * @return The additional parameters
+                */
+               public String[] getParameters() {
+                       return parameters;
                }
 
        }