X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Fmain%2Fjava%2Fde%2Ftodesbaum%2Fjsite%2Fmain%2FCLI.java;fp=src%2Fmain%2Fjava%2Fde%2Ftodesbaum%2Fjsite%2Fmain%2FCLI.java;h=bbe76c7136a33f9e5f2ff9561ee540239c378116;hb=38bdc433e50669e8244a63b5af59e597f88f1d29;hp=0000000000000000000000000000000000000000;hpb=f14b9fbe6d88e23920b10a75ebeba4d38390301b;p=jSite.git

diff --git a/src/main/java/de/todesbaum/jsite/main/CLI.java b/src/main/java/de/todesbaum/jsite/main/CLI.java
new file mode 100644
index 0000000..bbe76c7
--- /dev/null
+++ b/src/main/java/de/todesbaum/jsite/main/CLI.java
@@ -0,0 +1,303 @@
+/*
+ * jSite - CLI.java - Copyright © 2006–2012 David Roden
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ */
+
+package de.todesbaum.jsite.main;
+
+import java.io.PrintWriter;
+
+import net.pterodactylus.util.io.StreamCopier.ProgressListener;
+import de.todesbaum.jsite.application.Freenet7Interface;
+import de.todesbaum.jsite.application.InsertListener;
+import de.todesbaum.jsite.application.Node;
+import de.todesbaum.jsite.application.Project;
+import de.todesbaum.jsite.application.ProjectInserter;
+
+/**
+ * Command-line interface for jSite.
+ *
+ * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
+ */
+public class CLI implements InsertListener {
+
+	/** Object used for synchronization. */
+	private Object lockObject = new Object();
+
+	/** Writer for the console. */
+	private PrintWriter outputWriter = new PrintWriter(System.out, true);
+
+	/** The freenet interface. */
+	private Freenet7Interface freenetInterface;
+
+	/** The project inserter. */
+	private ProjectInserter projectInserter = new ProjectInserter();
+
+	/** The list of nodes. */
+	private Node[] nodes;
+
+	/** The projects. */
+	private Project[] projects;
+
+	/** Whether the insert has finished. */
+	private boolean finished = false;
+
+	/** Whether the insert finished successfully. */
+	private boolean success;
+
+	/**
+	 * Creates a new command-line interface.
+	 *
+	 * @param args
+	 *            The command-line arguments
+	 */
+	private CLI(String[] args) {
+
+		if ((args.length == 0) || args[0].equals("-h") || args[0].equals("--help")) {
+			outputWriter.println("\nParameters:\n");
+			outputWriter.println("  --config-file=<configuration file>");
+			outputWriter.println("  --node=<node name>");
+			outputWriter.println("  --project=<project name>");
+			outputWriter.println("  --local-directory=<local directory>");
+			outputWriter.println("  --path=<path>");
+			outputWriter.println("  --edition=<edition>");
+			outputWriter.println("\nA project gets inserted when a new project is loaded on the command line,");
+			outputWriter.println("or when the command line is finished. --local-directory, --path, and --edition");
+			outputWriter.println("override the parameters in the project.");
+			return;
+		}
+
+		String configFile = System.getProperty("user.home") + "/.jSite/config7";
+		for (String argument : args) {
+			String value = argument.substring(argument.indexOf('=') + 1).trim();
+			if (argument.startsWith("--config-file=")) {
+				configFile = value;
+			}
+		}
+
+		ConfigurationLocator configurationLocator = new ConfigurationLocator();
+		if (configFile != null) {
+			configurationLocator.setCustomLocation(configFile);
+		}
+		Configuration configuration = new Configuration(configurationLocator, configurationLocator.findPreferredLocation());
+
+		projectInserter.addInsertListener(this);
+		projects = configuration.getProjects();
+		Node node = configuration.getSelectedNode();
+		nodes = configuration.getNodes();
+
+		freenetInterface = new Freenet7Interface();
+		freenetInterface.setNode(node);
+
+		projectInserter.setFreenetInterface(freenetInterface);
+
+		Project currentProject = null;
+		for (String argument : args) {
+			if (argument.startsWith("--config-file=")) {
+				/* we already parsed this one. */
+				continue;
+			}
+			String value = argument.substring(argument.indexOf('=') + 1).trim();
+			if (argument.startsWith("--node=")) {
+				Node newNode = getNode(value);
+				if (newNode == null) {
+					outputWriter.println("Node \"" + value + "\" not found.");
+					return;
+				}
+				node = newNode;
+				freenetInterface.setNode(node);
+			} else if (argument.startsWith("--project=")) {
+				if (currentProject != null) {
+					if (insertProject(currentProject)) {
+						outputWriter.println("Project \"" + currentProject.getName() + "\" successfully inserted.");
+					} else {
+						outputWriter.println("Project \"" + currentProject.getName() + "\" was not successfully inserted.");
+					}
+					currentProject = null;
+				}
+				currentProject = getProject(value);
+				if (currentProject == null) {
+					outputWriter.println("Project \"" + value + "\" not found.");
+				}
+			} else if (argument.startsWith("--local-directory")) {
+				if (currentProject == null) {
+					outputWriter.println("You can't specifiy --local-directory before --project.");
+					return;
+				}
+				currentProject.setLocalPath(value);
+			} else if (argument.startsWith("--path=")) {
+				if (currentProject == null) {
+					outputWriter.println("You can't specify --path before --project.");
+					return;
+				}
+				currentProject.setPath(value);
+			} else if (argument.startsWith("--edition=")) {
+				if (currentProject == null) {
+					outputWriter.println("You can't specify --edition before --project.");
+					return;
+				}
+				currentProject.setEdition(Integer.parseInt(value));
+			} else {
+				outputWriter.println("Unknown parameter: " + argument);
+				return;
+			}
+		}
+
+		int errorCode = 1;
+		if (currentProject != null) {
+			if (insertProject(currentProject)) {
+				outputWriter.println("Project \"" + currentProject.getName() + "\" successfully inserted.");
+				errorCode = 0;
+			} else {
+				outputWriter.println("Project \"" + currentProject.getName() + "\" was not successfully inserted.");
+			}
+		}
+
+		configuration.setProjects(projects);
+		configuration.save();
+
+		System.exit(errorCode);
+	}
+
+	/**
+	 * Returns the project with the given name.
+	 *
+	 * @param name
+	 *            The name of the project
+	 * @return The project, or <code>null</code> if no project could be found
+	 */
+	private Project getProject(String name) {
+		for (Project project : projects) {
+			if (project.getName().equals(name)) {
+				return project;
+			}
+		}
+		return null;
+	}
+
+	/**
+	 * Returns the node with the given name.
+	 *
+	 * @param name
+	 *            The name of the node
+	 * @return The node, or <code>null</code> if no node could be found
+	 */
+	private Node getNode(String name) {
+		for (Node node : nodes) {
+			if (node.getName().equals(name)) {
+				return node;
+			}
+		}
+		return null;
+	}
+
+	/**
+	 * Inserts the given project.
+	 *
+	 * @param currentProject
+	 *            The project to insert
+	 * @return <code>true</code> if the insert finished successfully,
+	 *         <code>false</code> otherwise
+	 */
+	private boolean insertProject(Project currentProject) {
+		if (!freenetInterface.hasNode()) {
+			outputWriter.println("Node is not running!");
+			return false;
+		}
+		projectInserter.setProject(currentProject);
+		projectInserter.start(new ProgressListener() {
+
+			@Override
+			public void onProgress(long copied, long length) {
+				System.out.print("Uploaded: " + copied + " / " + length + " bytes...\r");
+			}
+		});
+		synchronized (lockObject) {
+			while (!finished) {
+				try {
+					lockObject.wait();
+				} catch (InterruptedException e) {
+					/* ignore, we're in a loop. */
+				}
+			}
+		}
+		return success;
+	}
+
+	//
+	// INTERFACE InsertListener
+	//
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	public void projectInsertStarted(Project project) {
+		outputWriter.println("Starting Insert of project \"" + project.getName() + "\".");
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	public void projectUploadFinished(Project project) {
+		outputWriter.println("Project \"" + project.getName() + "\" has been uploaded, starting insert...");
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	public void projectURIGenerated(Project project, String uri) {
+		outputWriter.println("URI: " + uri);
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	public void projectInsertProgress(Project project, int succeeded, int failed, int fatal, int total, boolean finalized) {
+		outputWriter.println("Progress: " + succeeded + " done, " + failed + " failed, " + fatal + " fatal, " + total + " total" + (finalized ? " (finalized)" : "") + ", " + ((succeeded + failed + fatal) * 100 / total) + "%");
+	}
+
+	/**
+	 * {@inheritDoc}
+	 */
+	@Override
+	public void projectInsertFinished(Project project, boolean success, Throwable cause) {
+		outputWriter.println("Request URI: " + project.getFinalRequestURI(0));
+		finished = true;
+		this.success = success;
+		synchronized (lockObject) {
+			lockObject.notify();
+		}
+	}
+
+	//
+	// MAIN
+	//
+
+	/**
+	 * Creates a new command-line interface with the given arguments.
+	 *
+	 * @param args
+	 *            The command-line arguments
+	 */
+	public static void main(String[] args) {
+		new CLI(args);
+	}
+
+}