add project manager

[jSite2.git] / src / net / pterodactylus / jsite / core / ProjectManager.java

diff --git a/src/net/pterodactylus/jsite/core/ProjectManager.java b/src/net/pterodactylus/jsite/core/ProjectManager.java
new file mode 100644 (file)
index 0000000..8034189
--- /dev/null
+++ b/src/net/pterodactylus/jsite/core/ProjectManager.java
@@ -0,0 +1,162 @@
+/*
+ * jSite2 - ProjectManager.java -
+ * Copyright © 2008 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 net.pterodactylus.jsite.core;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Properties;
+
+import net.pterodactylus.util.io.Closer;
+
+/**
+ * Manages projects, taking care of persistence, lifetime statistics, and other
+ * things.
+ * 
+ * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
+ * @version $Id$
+ */
+public class ProjectManager {
+
+       /** The directory the projects are stored in. */
+       private final String directory;
+
+       /** The list of project names. */
+       private final List<String> projectNames = new ArrayList<String>();
+
+       /** Mapping from project name to project. */
+       private final Map<String, Project> projects = new HashMap<String, Project>();
+
+       /** Mapping from project to lifetime statistics. */
+       @SuppressWarnings("unused")
+       private final Map<Project, ProjectLifetime> projectLifetimes = new HashMap<Project, ProjectLifetime>();
+
+       /**
+        * Creates a new project manager that saves and restores its state to/from
+        * the given directory.
+        * 
+        * @param directory
+        *            The directory to save and restore states to/from
+        */
+       public ProjectManager(String directory) {
+               this.directory = directory;
+       }
+
+       //
+       // ACCESSORS
+       //
+
+       /**
+        * Returns the directory the projects are loaded from and saved to.
+        * 
+        * @return The directory for storing the projects
+        */
+       public String getDirectory() {
+               return directory;
+       }
+
+       /**
+        * Returns a list of all projects, sorted by their name.
+        * 
+        * @return A list of all projects
+        */
+       public List<Project> getProjects() {
+               List<Project> projects = new ArrayList<Project>();
+               for (String projectName: projectNames) {
+                       projects.add(this.projects.get(projectName));
+               }
+               return projects;
+       }
+
+       //
+       // ACTIONS
+       //
+
+       /**
+        * Loads projects and statistics.
+        * 
+        * @throws IOException
+        *             if an I/O error occurs
+        */
+       public void load() throws IOException {
+               File directoryFile = new File(directory);
+               File projectFile = new File(directoryFile, "projects.properties");
+               if (!projectFile.exists() || !projectFile.isFile() || !projectFile.canRead()) {
+                       return;
+               }
+               Properties projectProperties = new Properties();
+               InputStream projectInputStream = null;
+               try {
+                       projectInputStream = new FileInputStream(projectFile);
+                       projectProperties.load(projectInputStream);
+               } finally {
+                       Closer.close(projectInputStream);
+               }
+               int projectIndex = 0;
+               while (projectProperties.containsKey("projects." + projectIndex + ".name")) {
+                       String projectPrefix = "projects." + projectIndex;
+                       String projectName = projectProperties.getProperty(projectPrefix + ".name");
+                       Project project = new Project();
+                       project.setName(projectName);
+                       projectNames.add(projectName);
+                       projects.put(projectName, project);
+                       projectIndex++;
+               }
+       }
+
+       /**
+        * Saves projects and statistics.
+        * 
+        * @throws IOException
+        *             if an I/O error occurs
+        */
+       public void save() throws IOException {
+               File directoryFile = new File(directory);
+               if (!directoryFile.exists()) {
+                       if (!directoryFile.mkdirs()) {
+                               throw new IOException("could not create directory: " + directory);
+                       }
+               }
+               Properties projectProperties = new Properties();
+               int projectIndex = 0;
+               for (String projectName: projectNames) {
+                       String projectPrefix = "projects." + projectIndex;
+                       Project project = projects.get(projectName);
+                       projectProperties.setProperty("projects." + projectPrefix + ".name", project.getName());
+                       projectIndex++;
+               }
+               File projectFile = new File(directoryFile, "projects.properties");
+               OutputStream projectOutputStream = null;
+               try {
+                       projectOutputStream = new FileOutputStream(projectFile);
+                       projectProperties.store(projectOutputStream, "jSite projects");
+               } finally {
+                       Closer.close(projectOutputStream);
+               }
+       }
+
+}