2 * jSite - Configuration.java - Copyright © 2006–2012 David Roden
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
19 package de.todesbaum.jsite.main;
21 import java.io.ByteArrayInputStream;
22 import java.io.ByteArrayOutputStream;
24 import java.io.FileInputStream;
25 import java.io.FileNotFoundException;
26 import java.io.FileOutputStream;
27 import java.io.IOException;
28 import java.util.ArrayList;
29 import java.util.HashMap;
30 import java.util.Iterator;
31 import java.util.List;
32 import java.util.Locale;
34 import java.util.Map.Entry;
35 import java.util.logging.Level;
36 import java.util.logging.Logger;
38 import net.pterodactylus.util.io.Closer;
39 import net.pterodactylus.util.io.StreamCopier;
40 import net.pterodactylus.util.xml.SimpleXML;
41 import net.pterodactylus.util.xml.XML;
42 import de.todesbaum.jsite.application.FileOption;
43 import de.todesbaum.jsite.application.Node;
44 import de.todesbaum.jsite.application.Project;
45 import de.todesbaum.jsite.main.ConfigurationLocator.ConfigurationLocation;
46 import de.todesbaum.util.freenet.fcp2.ClientPutDir.ManifestPutter;
47 import de.todesbaum.util.freenet.fcp2.PriorityClass;
52 * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
54 public class Configuration {
56 /** The root node of the configuration. */
57 private SimpleXML rootNode;
59 /** The configuration locator. */
60 private final ConfigurationLocator configurationLocator;
62 /** Where the configuration resides. */
63 private ConfigurationLocation configurationLocation;
66 * Creates a new configuration that is read from the given file.
68 * @param configurationLocator
69 * The configuration locator
70 * @param configurationLocation
71 * The configuration directory
73 public Configuration(ConfigurationLocator configurationLocator, ConfigurationLocation configurationLocation) {
74 this.configurationLocator = configurationLocator;
75 this.configurationLocation = configurationLocation;
76 readConfiguration(configurationLocator.getFile(configurationLocation));
84 * Returns the configuration locator.
86 * @return The configuration locator
88 public ConfigurationLocator getConfigurationLocator() {
89 return configurationLocator;
93 * Returns the location the configuration will be written to when calling
96 * @return The location the configuration will be written to
98 public ConfigurationLocation getConfigurationDirectory() {
99 return configurationLocation;
103 * Sets the location the configuration will be written to when calling
106 * @param configurationLocation
107 * The location to write the configuration to
109 public void setConfigurationLocation(ConfigurationLocation configurationLocation) {
110 this.configurationLocation = configurationLocation;
114 * Reads the configuration from the file.
117 * The name of the file to read the configuration from
119 private void readConfiguration(String filename) {
120 Logger.getLogger(Configuration.class.getName()).log(Level.CONFIG, "Reading configuration from: " + filename);
121 if (filename != null) {
122 File configurationFile = new File(filename);
123 if (configurationFile.exists()) {
124 ByteArrayOutputStream fileByteOutputStream = null;
125 FileInputStream fileInputStream = null;
127 fileByteOutputStream = new ByteArrayOutputStream();
128 fileInputStream = new FileInputStream(configurationFile);
129 StreamCopier.copy(fileInputStream, fileByteOutputStream, configurationFile.length());
130 fileByteOutputStream.close();
131 byte[] fileBytes = fileByteOutputStream.toByteArray();
132 rootNode = SimpleXML.fromDocument(XML.transformToDocument(fileBytes));
134 } catch (FileNotFoundException e) {
136 } catch (IOException e) {
139 Closer.close(fileInputStream);
140 Closer.close(fileByteOutputStream);
144 rootNode = new SimpleXML("configuration");
148 * Saves the configuration.
150 * @return <code>true</code> if the configuration could be saved,
151 * <code>false</code> otherwise
153 public boolean save() {
154 Logger.getLogger(Configuration.class.getName()).log(Level.CONFIG, "Trying to save configuration to: " + configurationLocation);
155 File configurationFile = new File(configurationLocator.getFile(configurationLocation));
156 if (!configurationFile.exists()) {
157 File configurationFilePath = configurationFile.getAbsoluteFile().getParentFile();
158 if (!configurationFilePath.exists() && !configurationFilePath.mkdirs()) {
162 FileOutputStream fileOutputStream = null;
163 ByteArrayInputStream configurationInputStream = null;
165 byte[] configurationBytes = XML.transformToByteArray(rootNode.getDocument());
166 configurationInputStream = new ByteArrayInputStream(configurationBytes);
167 fileOutputStream = new FileOutputStream(configurationFile);
168 StreamCopier.copy(configurationInputStream, fileOutputStream, configurationBytes.length);
170 } catch (IOException ioe1) {
173 Closer.close(configurationInputStream);
174 Closer.close(fileOutputStream);
180 * Returns the value of a node.
183 * The name of all nodes in the chain
184 * @param defaultValue
185 * The default value to return if the node could not be found
186 * @return The value of the node, or the default value if the node could not
189 private String getNodeValue(String[] nodeNames, String defaultValue) {
190 SimpleXML node = rootNode;
192 while ((node != null) && (nodeIndex < nodeNames.length)) {
193 node = node.getNode(nodeNames[nodeIndex++]);
198 return node.getValue();
202 * Returns the integer value of a node.
205 * The names of all nodes in the chain
206 * @param defaultValue
207 * The default value to return if the node can not be found
208 * @return The parsed integer value, or the default value if the node can
209 * not be found or the value can not be parsed into an integer
211 private int getNodeIntValue(String[] nodeNames, int defaultValue) {
213 return Integer.parseInt(getNodeValue(nodeNames, String.valueOf(defaultValue)));
214 } catch (NumberFormatException nfe1) {
221 * Returns the boolean value of a node.
224 * The names of all nodes in the chain
225 * @param defaultValue
226 * The default value to return if the node can not be found
227 * @return The parsed boolean value, or the default value if the node can
230 private boolean getNodeBooleanValue(String[] nodeNames, boolean defaultValue) {
231 String nodeValue = getNodeValue(nodeNames, null);
232 if (nodeValue == null) {
235 return Boolean.parseBoolean(nodeValue);
239 * Returns the hostname of the node.
241 * @return The hostname of the node
242 * @deprecated Use {@link #getSelectedNode()} instead
245 public String getNodeAddress() {
246 return getNodeValue(new String[] { "node-address" }, "localhost");
250 * Sets the hostname of the node.
253 * The hostname of the node
254 * @deprecated Use {@link #setSelectedNode(Node)} instead
257 public void setNodeAddress(String nodeAddress) {
258 rootNode.replace("node-address", nodeAddress);
262 * The port number of the node
264 * @return The port number of the node
265 * @deprecated Use {@link #getSelectedNode()} instead.
268 public int getNodePort() {
269 return getNodeIntValue(new String[] { "node-port" }, 9481);
273 * Sets the port number of the node.
276 * The port number of the node
277 * @deprecated Use {@link #setSelectedNode(Node)} instead
280 public void setNodePort(int nodePort) {
281 rootNode.replace("node-port", String.valueOf(nodePort));
285 * Returns whether the node configuration page should be skipped on startup.
287 * @return <code>true</code> to skip the node configuration page on startup,
288 * <code>false</code> to show it
290 public boolean isSkipNodePage() {
291 return getNodeBooleanValue(new String[] { "skip-node-page" }, false);
295 * Sets whether the node configuration page should be skipped on startup.
297 * @param skipNodePage
298 * <code>true</code> to skip the node configuration page on
299 * startup, <code>false</code> to show it
301 public void setSkipNodePage(boolean skipNodePage) {
302 rootNode.replace("skip-node-page", String.valueOf(skipNodePage));
306 * Returns all configured projects.
308 * @return A list of all projects
310 public List<Project> getProjects() {
311 List<Project> projects = new ArrayList<Project>();
312 SimpleXML projectsNode = rootNode.getNode("project-list");
313 if (projectsNode != null) {
314 SimpleXML[] projectNodes = projectsNode.getNodes("project");
315 for (SimpleXML projectNode : projectNodes) {
317 Project project = new Project();
318 projects.add(project);
319 project.setDescription(projectNode.getValue("description", ""));
320 String indexFile = projectNode.getValue("index-file", "");
321 if (indexFile.indexOf('/') > -1) {
324 project.setIndexFile(indexFile);
325 project.setLastInsertionTime(Long.parseLong(projectNode.getValue("last-insertion-time", "0")));
326 project.setLocalPath(projectNode.getValue("local-path", ""));
327 project.setName(projectNode.getValue("name", ""));
328 project.setPath(projectNode.getValue("path", ""));
329 if ((project.getPath() != null) && (project.getPath().indexOf("/") != -1)) {
330 project.setPath(project.getPath().replaceAll("/", ""));
332 project.setEdition(Integer.parseInt(projectNode.getValue("edition", "0")));
333 project.setInsertURI(projectNode.getValue("insert-uri", ""));
334 project.setRequestURI(projectNode.getValue("request-uri", ""));
335 if (projectNode.getNode("ignore-hidden-files") != null) {
336 project.setIgnoreHiddenFiles(Boolean.parseBoolean(projectNode.getValue("ignore-hidden-files", "true")));
338 project.setIgnoreHiddenFiles(true);
340 project.setAlwaysForceInsert(Boolean.parseBoolean(projectNode.getValue("always-force-insert", "false")));
342 /* load last insert hashes. */
343 Map<String, FileOption> fileOptions = new HashMap<String, FileOption>();
344 SimpleXML lastInsertHashesNode = projectNode.getNode("last-insert-hashes");
345 if (lastInsertHashesNode != null) {
346 for (SimpleXML fileNode : lastInsertHashesNode.getNodes("file")) {
347 String filename = fileNode.getNode("filename").getValue();
348 String lastInsertHash = fileNode.getNode("last-insert-hash").getValue();
349 int lastInsertEdition = Integer.valueOf(fileNode.getNode("last-insert-edition").getValue());
350 String lastInsertFilename = filename;
351 if (fileNode.getNode("last-insert-filename") != null) {
352 lastInsertFilename = fileNode.getNode("last-insert-filename").getValue();
354 FileOption fileOption = project.getFileOption(filename);
355 fileOption.setLastInsertHash(lastInsertHash).setLastInsertEdition(lastInsertEdition).setLastInsertFilename(lastInsertFilename);
356 fileOptions.put(filename, fileOption);
360 SimpleXML fileOptionsNode = projectNode.getNode("file-options");
361 if (fileOptionsNode != null) {
362 SimpleXML[] fileOptionNodes = fileOptionsNode.getNodes("file-option");
363 for (SimpleXML fileOptionNode : fileOptionNodes) {
364 String filename = fileOptionNode.getNode("filename").getValue();
365 FileOption fileOption = project.getFileOption(filename);
366 fileOption.setInsert(Boolean.parseBoolean(fileOptionNode.getNode("insert").getValue()));
367 if (fileOptionNode.getNode("insert-redirect") != null) {
368 fileOption.setInsertRedirect(Boolean.parseBoolean(fileOptionNode.getNode("insert-redirect").getValue()));
370 fileOption.setCustomKey(fileOptionNode.getValue("custom-key", ""));
371 if (fileOptionNode.getNode("changed-name") != null) {
372 fileOption.setChangedName(fileOptionNode.getNode("changed-name").getValue());
374 fileOption.setMimeType(fileOptionNode.getValue("mime-type", ""));
375 fileOptions.put(filename, fileOption);
378 project.setFileOptions(fileOptions);
379 } catch (NumberFormatException nfe1) {
380 nfe1.printStackTrace();
388 * Sets the list of all projects.
391 * The list of all projects
393 public void setProjects(List<Project> projects) {
394 SimpleXML projectsNode = new SimpleXML("project-list");
395 for (Project project : projects) {
396 SimpleXML projectNode = projectsNode.append("project");
397 projectNode.append("edition", String.valueOf(project.getEdition()));
398 projectNode.append("description", project.getDescription());
399 projectNode.append("index-file", project.getIndexFile());
400 projectNode.append("last-insertion-time", String.valueOf(project.getLastInsertionTime()));
401 projectNode.append("local-path", project.getLocalPath());
402 projectNode.append("name", project.getName());
403 projectNode.append("path", project.getPath());
404 projectNode.append("insert-uri", project.getInsertURI());
405 projectNode.append("request-uri", project.getRequestURI());
406 projectNode.append("ignore-hidden-files", String.valueOf(project.isIgnoreHiddenFiles()));
407 projectNode.append("always-force-insert", String.valueOf(project.isAlwaysForceInsert()));
409 /* store last insert hashes. */
410 SimpleXML lastInsertHashesNode = projectNode.append("last-insert-hashes");
411 for (Entry<String, FileOption> fileOption : project.getFileOptions().entrySet()) {
412 if ((fileOption.getValue().getLastInsertHash() == null) || (fileOption.getValue().getLastInsertHash().length() == 0)) {
415 SimpleXML fileNode = lastInsertHashesNode.append("file");
416 fileNode.append("filename", fileOption.getKey());
417 fileNode.append("last-insert-hash", fileOption.getValue().getLastInsertHash());
418 fileNode.append("last-insert-edition", String.valueOf(fileOption.getValue().getLastInsertEdition()));
419 fileNode.append("last-insert-filename", fileOption.getValue().getLastInsertFilename());
422 SimpleXML fileOptionsNode = projectNode.append("file-options");
423 Iterator<Entry<String, FileOption>> entries = project.getFileOptions().entrySet().iterator();
424 while (entries.hasNext()) {
425 Entry<String, FileOption> entry = entries.next();
426 FileOption fileOption = entry.getValue();
427 if (fileOption.isCustom()) {
428 SimpleXML fileOptionNode = fileOptionsNode.append("file-option");
429 fileOptionNode.append("filename", entry.getKey());
430 fileOptionNode.append("insert", String.valueOf(fileOption.isInsert()));
431 fileOptionNode.append("insert-redirect", String.valueOf(fileOption.isInsertRedirect()));
432 fileOptionNode.append("custom-key", fileOption.getCustomKey());
433 fileOptionNode.append("changed-name", fileOption.getChangedName().orNull());
434 fileOptionNode.append("mime-type", fileOption.getMimeType());
438 rootNode.replace(projectsNode);
442 * Returns the stored locale.
444 * @return The stored locale
446 public Locale getLocale() {
447 String language = getNodeValue(new String[] { "i18n", "language" }, "en");
448 String country = getNodeValue(new String[] { "i18n", "country" }, null);
449 if (country != null) {
450 return new Locale(language, country);
452 return new Locale(language);
456 * Sets the locale to store.
459 * The locale to store
461 public void setLocale(Locale locale) {
462 SimpleXML i18nNode = new SimpleXML("i18n");
463 if (locale.getCountry().length() != 0) {
464 i18nNode.append("country", locale.getCountry());
466 i18nNode.append("language", locale.getLanguage());
467 rootNode.replace(i18nNode);
472 * Returns a list of configured nodes.
474 * @return The list of the configured nodes
476 public Node[] getNodes() {
477 SimpleXML nodesNode = rootNode.getNode("nodes");
478 if (nodesNode == null) {
479 String hostname = getNodeAddress();
480 int port = getNodePort();
481 if (hostname == null) {
482 hostname = "127.0.0.1";
485 return new Node[] { new Node(hostname, port, "Node") };
487 SimpleXML[] nodeNodes = nodesNode.getNodes("node");
488 Node[] returnNodes = new Node[nodeNodes.length];
490 for (SimpleXML nodeNode : nodeNodes) {
491 String name = nodeNode.getNode("name").getValue();
492 String hostname = nodeNode.getNode("hostname").getValue();
493 int port = Integer.parseInt(nodeNode.getNode("port").getValue());
494 Node node = new Node(hostname, port, name);
495 returnNodes[nodeIndex++] = node;
501 * Sets the list of configured nodes.
504 * The list of configured nodes
506 public void setNodes(Node[] nodes) {
507 SimpleXML nodesNode = new SimpleXML("nodes");
508 for (Node node : nodes) {
509 SimpleXML nodeNode = nodesNode.append("node");
510 nodeNode.append("name", node.getName());
511 nodeNode.append("hostname", node.getHostname());
512 nodeNode.append("port", String.valueOf(node.getPort()));
514 rootNode.replace(nodesNode);
515 rootNode.remove("node-address");
516 rootNode.remove("node-port");
520 * Sets the selected node.
522 * @param selectedNode
525 public void setSelectedNode(Node selectedNode) {
526 SimpleXML selectedNodeNode = new SimpleXML("selected-node");
527 selectedNodeNode.append("name", selectedNode.getName());
528 selectedNodeNode.append("hostname", selectedNode.getHostname());
529 selectedNodeNode.append("port", String.valueOf(selectedNode.getPort()));
530 rootNode.replace(selectedNodeNode);
534 * Returns the selected node.
536 * @return The selected node
538 public Node getSelectedNode() {
539 SimpleXML selectedNodeNode = rootNode.getNode("selected-node");
540 if (selectedNodeNode == null) {
541 String hostname = getNodeAddress();
542 int port = getNodePort();
543 if (hostname == null) {
544 hostname = "127.0.0.1";
547 return new Node(hostname, port, "Node");
549 String name = selectedNodeNode.getNode("name").getValue();
550 String hostname = selectedNodeNode.getNode("hostname").getValue();
551 int port = Integer.valueOf(selectedNodeNode.getNode("port").getValue());
552 return new Node(hostname, port, name);
556 * Returns the temp directory to use.
558 * @return The temp directoy, or {@code null} to use the default temp
561 public String getTempDirectory() {
562 return getNodeValue(new String[] { "temp-directory" }, null);
566 * Sets the temp directory to use.
568 * @param tempDirectory
569 * The temp directory to use, or {@code null} to use the default
572 public void setTempDirectory(String tempDirectory) {
573 if (tempDirectory != null) {
574 SimpleXML tempDirectoryNode = new SimpleXML("temp-directory");
575 tempDirectoryNode.setValue(tempDirectory);
576 rootNode.replace(tempDirectoryNode);
578 rootNode.remove("temp-directory");
583 * Returns whether to use the “early encode“ flag for the insert.
585 * @return {@code true} to set the “early encode” flag for the insert,
586 * {@code false} otherwise
588 public boolean useEarlyEncode() {
589 return getNodeBooleanValue(new String[] { "use-early-encode" }, false);
593 * Sets whether to use the “early encode“ flag for the insert.
595 * @param useEarlyEncode
596 * {@code true} to set the “early encode” flag for the insert,
597 * {@code false} otherwise
598 * @return This configuration
600 public Configuration setUseEarlyEncode(boolean useEarlyEncode) {
601 rootNode.replace("use-early-encode", String.valueOf(useEarlyEncode));
606 * Returns the insert priority.
608 * @return The insert priority
610 public PriorityClass getPriority() {
611 return PriorityClass.valueOf(getNodeValue(new String[] { "insert-priority" }, "interactive"));
615 * Sets the insert priority.
618 * The insert priority
619 * @return This configuration
621 public Configuration setPriority(PriorityClass priority) {
622 rootNode.replace("insert-priority", priority.toString());
627 * Returns the manifest putter.
629 * @return The manifest putter
631 public ManifestPutter getManifestPutter() {
632 return ManifestPutter.valueOf(getNodeValue(new String[] { "manifest-putter" }, "simple").toUpperCase());
636 * Sets the manifest putter.
638 * @param manifestPutter
639 * The manifest putter
640 * @return This configuration
642 public Configuration setManifestPutter(ManifestPutter manifestPutter) {
643 rootNode.replace("manifest-putter", manifestPutter.name().toLowerCase());