Store the manifest putter in the configuration.
[jSite.git] / src / de / todesbaum / jsite / main / Configuration.java
1 /*
2  * jSite - Configuration.java - Copyright © 2006–2012 David Roden
3  *
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.
8  *
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.
13  *
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.
17  */
18
19 package de.todesbaum.jsite.main;
20
21 import java.io.ByteArrayInputStream;
22 import java.io.ByteArrayOutputStream;
23 import java.io.File;
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;
33 import java.util.Map;
34 import java.util.Map.Entry;
35 import java.util.logging.Level;
36 import java.util.logging.Logger;
37
38 import de.todesbaum.jsite.application.FileOption;
39 import de.todesbaum.jsite.application.Node;
40 import de.todesbaum.jsite.application.Project;
41 import de.todesbaum.jsite.main.ConfigurationLocator.ConfigurationLocation;
42 import de.todesbaum.util.freenet.fcp2.ClientPutDir.ManifestPutter;
43 import de.todesbaum.util.freenet.fcp2.PriorityClass;
44 import de.todesbaum.util.io.Closer;
45 import de.todesbaum.util.io.StreamCopier;
46 import de.todesbaum.util.xml.SimpleXML;
47 import de.todesbaum.util.xml.XML;
48
49 /**
50  * The configuration.
51  *
52  * @author David ‘Bombe’ Roden <bombe@freenetproject.org>
53  */
54 public class Configuration {
55
56         /** The root node of the configuration. */
57         private SimpleXML rootNode;
58
59         /** The configuration locator. */
60         private final ConfigurationLocator configurationLocator;
61
62         /** Where the configuration resides. */
63         private ConfigurationLocation configurationLocation;
64
65         /**
66          * Creates a new configuration that is read from the given file.
67          *
68          * @param configurationLocator
69          *            The configuration locator
70          * @param configurationLocation
71          *            The configuration directory
72          */
73         public Configuration(ConfigurationLocator configurationLocator, ConfigurationLocation configurationLocation) {
74                 this.configurationLocator = configurationLocator;
75                 this.configurationLocation = configurationLocation;
76                 readConfiguration(configurationLocator.getFile(configurationLocation));
77         }
78
79         //
80         // ACCESSORS
81         //
82
83         /**
84          * Returns the configuration locator.
85          *
86          * @return The configuration locator
87          */
88         public ConfigurationLocator getConfigurationLocator() {
89                 return configurationLocator;
90         }
91
92         /**
93          * Returns the location the configuration will be written to when calling
94          * {@link #save()}.
95          *
96          * @return The location the configuration will be written to
97          */
98         public ConfigurationLocation getConfigurationDirectory() {
99                 return configurationLocation;
100         }
101
102         /**
103          * Sets the location the configuration will be written to when calling
104          * {@link #save()}.
105          *
106          * @param configurationLocation
107          *            The location to write the configuration to
108          */
109         public void setConfigurationLocation(ConfigurationLocation configurationLocation) {
110                 this.configurationLocation = configurationLocation;
111         }
112
113         /**
114          * Reads the configuration from the file.
115          *
116          * @param filename
117          *            The name of the file to read the configuration from
118          */
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;
126                                 try {
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));
133                                         return;
134                                 } catch (FileNotFoundException e) {
135                                         /* ignore. */
136                                 } catch (IOException e) {
137                                         /* ignore. */
138                                 } finally {
139                                         Closer.close(fileInputStream);
140                                         Closer.close(fileByteOutputStream);
141                                 }
142                         }
143                 }
144                 rootNode = new SimpleXML("configuration");
145         }
146
147         /**
148          * Saves the configuration.
149          *
150          * @return <code>true</code> if the configuration could be saved,
151          *         <code>false</code> otherwise
152          */
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()) {
159                                 return false;
160                         }
161                 }
162                 FileOutputStream fileOutputStream = null;
163                 ByteArrayInputStream configurationInputStream = null;
164                 try {
165                         byte[] configurationBytes = XML.transformToByteArray(rootNode.getDocument());
166                         configurationInputStream = new ByteArrayInputStream(configurationBytes);
167                         fileOutputStream = new FileOutputStream(configurationFile);
168                         StreamCopier.copy(configurationInputStream, fileOutputStream, configurationBytes.length);
169                         return true;
170                 } catch (IOException ioe1) {
171                         /* ignore. */
172                 } finally {
173                         Closer.close(configurationInputStream);
174                         Closer.close(fileOutputStream);
175                 }
176                 return false;
177         }
178
179         /**
180          * Returns the value of a node.
181          *
182          * @param nodeNames
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
187          *         be found
188          */
189         private String getNodeValue(String[] nodeNames, String defaultValue) {
190                 SimpleXML node = rootNode;
191                 int nodeIndex = 0;
192                 while ((node != null) && (nodeIndex < nodeNames.length)) {
193                         node = node.getNode(nodeNames[nodeIndex++]);
194                 }
195                 if (node == null) {
196                         return defaultValue;
197                 }
198                 return node.getValue();
199         }
200
201         /**
202          * Returns the integer value of a node.
203          *
204          * @param nodeNames
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
210          */
211         private int getNodeIntValue(String[] nodeNames, int defaultValue) {
212                 try {
213                         return Integer.parseInt(getNodeValue(nodeNames, String.valueOf(defaultValue)));
214                 } catch (NumberFormatException nfe1) {
215                         /* ignore. */
216                 }
217                 return defaultValue;
218         }
219
220         /**
221          * Returns the boolean value of a node.
222          *
223          * @param nodeNames
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
228          *         not be found
229          */
230         private boolean getNodeBooleanValue(String[] nodeNames, boolean defaultValue) {
231                 String nodeValue = getNodeValue(nodeNames, null);
232                 if (nodeValue == null) {
233                         return defaultValue;
234                 }
235                 return Boolean.parseBoolean(nodeValue);
236         }
237
238         /**
239          * Returns the hostname of the node.
240          *
241          * @return The hostname of the node
242          * @deprecated Use {@link #getSelectedNode()} instead
243          */
244         @Deprecated
245         public String getNodeAddress() {
246                 return getNodeValue(new String[] { "node-address" }, "localhost");
247         }
248
249         /**
250          * Sets the hostname of the node.
251          *
252          * @param nodeAddress
253          *            The hostname of the node
254          * @deprecated Use {@link #setSelectedNode(Node)} instead
255          */
256         @Deprecated
257         public void setNodeAddress(String nodeAddress) {
258                 rootNode.replace("node-address", nodeAddress);
259         }
260
261         /**
262          * The port number of the node
263          *
264          * @return The port number of the node
265          * @deprecated Use {@link #getSelectedNode()} instead.
266          */
267         @Deprecated
268         public int getNodePort() {
269                 return getNodeIntValue(new String[] { "node-port" }, 9481);
270         }
271
272         /**
273          * Sets the port number of the node.
274          *
275          * @param nodePort
276          *            The port number of the node
277          * @deprecated Use {@link #setSelectedNode(Node)} instead
278          */
279         @Deprecated
280         public void setNodePort(int nodePort) {
281                 rootNode.replace("node-port", String.valueOf(nodePort));
282         }
283
284         /**
285          * Returns whether the node configuration page should be skipped on startup.
286          *
287          * @return <code>true</code> to skip the node configuration page on startup,
288          *         <code>false</code> to show it
289          */
290         public boolean isSkipNodePage() {
291                 return getNodeBooleanValue(new String[] { "skip-node-page" }, false);
292         }
293
294         /**
295          * Sets whether the node configuration page should be skipped on startup.
296          *
297          * @param skipNodePage
298          *            <code>true</code> to skip the node configuration page on
299          *            startup, <code>false</code> to show it
300          */
301         public void setSkipNodePage(boolean skipNodePage) {
302                 rootNode.replace("skip-node-page", String.valueOf(skipNodePage));
303         }
304
305         /**
306          * Returns all configured projects.
307          *
308          * @return A list of all projects
309          */
310         public 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) {
316                                 try {
317                                         Project project = new Project();
318                                         projects.add(project);
319                                         project.setDescription(projectNode.getNode("description").getValue(""));
320                                         String indexFile = projectNode.getNode("index-file").getValue("");
321                                         if (indexFile.indexOf('/') > -1) {
322                                                 indexFile = "";
323                                         }
324                                         project.setIndexFile(indexFile);
325                                         project.setLastInsertionTime(Long.parseLong(projectNode.getNode("last-insertion-time").getValue("0")));
326                                         project.setLocalPath(projectNode.getNode("local-path").getValue(""));
327                                         project.setName(projectNode.getNode("name").getValue(""));
328                                         project.setPath(projectNode.getNode("path").getValue(""));
329                                         if ((project.getPath() != null) && (project.getPath().indexOf("/") != -1)) {
330                                                 project.setPath(project.getPath().replaceAll("/", ""));
331                                         }
332                                         project.setEdition(Integer.parseInt(projectNode.getNode("edition").getValue("0")));
333                                         project.setInsertURI(projectNode.getNode("insert-uri").getValue(""));
334                                         project.setRequestURI(projectNode.getNode("request-uri").getValue(""));
335                                         if (projectNode.getNode("ignore-hidden-files") != null) {
336                                                 project.setIgnoreHiddenFiles(Boolean.parseBoolean(projectNode.getNode("ignore-hidden-files").getValue("true")));
337                                         } else {
338                                                 project.setIgnoreHiddenFiles(true);
339                                         }
340
341                                         /* load last insert hashes. */
342                                         Map<String, FileOption> fileOptions = new HashMap<String, FileOption>();
343                                         SimpleXML lastInsertHashesNode = projectNode.getNode("last-insert-hashes");
344                                         if (lastInsertHashesNode != null) {
345                                                 for (SimpleXML fileNode : lastInsertHashesNode.getNodes("file")) {
346                                                         String filename = fileNode.getNode("filename").getValue();
347                                                         String lastInsertHash = fileNode.getNode("last-insert-hash").getValue();
348                                                         int lastInsertEdition = Integer.valueOf(fileNode.getNode("last-insert-edition").getValue());
349                                                         FileOption fileOption = project.getFileOption(filename);
350                                                         fileOption.setLastInsertHash(lastInsertHash).setLastInsertEdition(lastInsertEdition);
351                                                         fileOptions.put(filename, fileOption);
352                                                 }
353                                         }
354
355                                         SimpleXML fileOptionsNode = projectNode.getNode("file-options");
356                                         if (fileOptionsNode != null) {
357                                                 SimpleXML[] fileOptionNodes = fileOptionsNode.getNodes("file-option");
358                                                 for (SimpleXML fileOptionNode : fileOptionNodes) {
359                                                         String filename = fileOptionNode.getNode("filename").getValue();
360                                                         FileOption fileOption = project.getFileOption(filename);
361                                                         fileOption.setInsert(Boolean.parseBoolean(fileOptionNode.getNode("insert").getValue()));
362                                                         if (fileOptionNode.getNode("insert-redirect") != null) {
363                                                                 fileOption.setInsertRedirect(Boolean.parseBoolean(fileOptionNode.getNode("insert-redirect").getValue()));
364                                                         }
365                                                         fileOption.setCustomKey(fileOptionNode.getNode("custom-key").getValue(""));
366                                                         if (fileOptionNode.getNode("changed-name") != null) {
367                                                                 fileOption.setChangedName(fileOptionNode.getNode("changed-name").getValue());
368                                                         }
369                                                         fileOption.setMimeType(fileOptionNode.getNode("mime-type").getValue(""));
370                                                         fileOptions.put(filename, fileOption);
371                                                 }
372                                         }
373                                         project.setFileOptions(fileOptions);
374                                 } catch (NumberFormatException nfe1) {
375                                         nfe1.printStackTrace();
376                                 }
377                         }
378                 }
379                 return projects.toArray(new Project[projects.size()]);
380         }
381
382         /**
383          * Sets the list of all projects.
384          *
385          * @param projects
386          *            The list of all projects
387          */
388         public void setProjects(Project[] projects) {
389                 SimpleXML projectsNode = new SimpleXML("project-list");
390                 for (Project project : projects) {
391                         SimpleXML projectNode = projectsNode.append("project");
392                         projectNode.append("edition", String.valueOf(project.getEdition()));
393                         projectNode.append("description", project.getDescription());
394                         projectNode.append("index-file", project.getIndexFile());
395                         projectNode.append("last-insertion-time", String.valueOf(project.getLastInsertionTime()));
396                         projectNode.append("local-path", project.getLocalPath());
397                         projectNode.append("name", project.getName());
398                         projectNode.append("path", project.getPath());
399                         projectNode.append("insert-uri", project.getInsertURI());
400                         projectNode.append("request-uri", project.getRequestURI());
401                         projectNode.append("ignore-hidden-files", String.valueOf(project.isIgnoreHiddenFiles()));
402
403                         /* store last insert hashes. */
404                         SimpleXML lastInsertHashesNode = projectNode.append("last-insert-hashes");
405                         for (Entry<String, FileOption> fileOption : project.getFileOptions().entrySet()) {
406                                 if ((fileOption.getValue().getLastInsertHash() == null) || (fileOption.getValue().getLastInsertHash().length() == 0)) {
407                                         continue;
408                                 }
409                                 SimpleXML fileNode = lastInsertHashesNode.append("file");
410                                 fileNode.append("filename", fileOption.getKey());
411                                 fileNode.append("last-insert-hash", fileOption.getValue().getLastInsertHash());
412                                 fileNode.append("last-insert-edition", String.valueOf(fileOption.getValue().getLastInsertEdition()));
413                         }
414
415                         SimpleXML fileOptionsNode = projectNode.append("file-options");
416                         Iterator<Entry<String, FileOption>> entries = project.getFileOptions().entrySet().iterator();
417                         while (entries.hasNext()) {
418                                 Entry<String, FileOption> entry = entries.next();
419                                 FileOption fileOption = entry.getValue();
420                                 if (fileOption.isCustom()) {
421                                         SimpleXML fileOptionNode = fileOptionsNode.append("file-option");
422                                         fileOptionNode.append("filename", entry.getKey());
423                                         fileOptionNode.append("insert", String.valueOf(fileOption.isInsert()));
424                                         fileOptionNode.append("insert-redirect", String.valueOf(fileOption.isInsertRedirect()));
425                                         fileOptionNode.append("custom-key", fileOption.getCustomKey());
426                                         fileOptionNode.append("changed-name", fileOption.getChangedName());
427                                         fileOptionNode.append("mime-type", fileOption.getMimeType());
428                                 }
429                         }
430                 }
431                 rootNode.replace(projectsNode);
432         }
433
434         /**
435          * Returns the stored locale.
436          *
437          * @return The stored locale
438          */
439         public Locale getLocale() {
440                 String language = getNodeValue(new String[] { "i18n", "language" }, "en");
441                 String country = getNodeValue(new String[] { "i18n", "country" }, null);
442                 if (country != null) {
443                         return new Locale(language, country);
444                 }
445                 return new Locale(language);
446         }
447
448         /**
449          * Sets the locale to store.
450          *
451          * @param locale
452          *            The locale to store
453          */
454         public void setLocale(Locale locale) {
455                 SimpleXML i18nNode = new SimpleXML("i18n");
456                 if (locale.getCountry().length() != 0) {
457                         i18nNode.append("country", locale.getCountry());
458                 }
459                 i18nNode.append("language", locale.getLanguage());
460                 rootNode.replace(i18nNode);
461                 return;
462         }
463
464         /**
465          * Returns a list of configured nodes.
466          *
467          * @return The list of the configured nodes
468          */
469         public Node[] getNodes() {
470                 SimpleXML nodesNode = rootNode.getNode("nodes");
471                 if (nodesNode == null) {
472                         String hostname = getNodeAddress();
473                         int port = getNodePort();
474                         if (hostname == null) {
475                                 hostname = "127.0.0.1";
476                                 port = 9481;
477                         }
478                         return new Node[] { new Node(hostname, port, "Node") };
479                 }
480                 SimpleXML[] nodeNodes = nodesNode.getNodes("node");
481                 Node[] returnNodes = new Node[nodeNodes.length];
482                 int nodeIndex = 0;
483                 for (SimpleXML nodeNode : nodeNodes) {
484                         String name = nodeNode.getNode("name").getValue();
485                         String hostname = nodeNode.getNode("hostname").getValue();
486                         int port = Integer.parseInt(nodeNode.getNode("port").getValue());
487                         Node node = new Node(hostname, port, name);
488                         returnNodes[nodeIndex++] = node;
489                 }
490                 return returnNodes;
491         }
492
493         /**
494          * Sets the list of configured nodes.
495          *
496          * @param nodes
497          *            The list of configured nodes
498          */
499         public void setNodes(Node[] nodes) {
500                 SimpleXML nodesNode = new SimpleXML("nodes");
501                 for (Node node : nodes) {
502                         SimpleXML nodeNode = nodesNode.append("node");
503                         nodeNode.append("name", node.getName());
504                         nodeNode.append("hostname", node.getHostname());
505                         nodeNode.append("port", String.valueOf(node.getPort()));
506                 }
507                 rootNode.replace(nodesNode);
508                 rootNode.remove("node-address");
509                 rootNode.remove("node-port");
510         }
511
512         /**
513          * Sets the selected node.
514          *
515          * @param selectedNode
516          *            The selected node
517          */
518         public void setSelectedNode(Node selectedNode) {
519                 SimpleXML selectedNodeNode = new SimpleXML("selected-node");
520                 selectedNodeNode.append("name", selectedNode.getName());
521                 selectedNodeNode.append("hostname", selectedNode.getHostname());
522                 selectedNodeNode.append("port", String.valueOf(selectedNode.getPort()));
523                 rootNode.replace(selectedNodeNode);
524         }
525
526         /**
527          * Returns the selected node.
528          *
529          * @return The selected node
530          */
531         public Node getSelectedNode() {
532                 SimpleXML selectedNodeNode = rootNode.getNode("selected-node");
533                 if (selectedNodeNode == null) {
534                         String hostname = getNodeAddress();
535                         int port = getNodePort();
536                         if (hostname == null) {
537                                 hostname = "127.0.0.1";
538                                 port = 9481;
539                         }
540                         return new Node(hostname, port, "Node");
541                 }
542                 String name = selectedNodeNode.getNode("name").getValue();
543                 String hostname = selectedNodeNode.getNode("hostname").getValue();
544                 int port = Integer.valueOf(selectedNodeNode.getNode("port").getValue());
545                 return new Node(hostname, port, name);
546         }
547
548         /**
549          * Returns the temp directory to use.
550          *
551          * @return The temp directoy, or {@code null} to use the default temp
552          *         directory
553          */
554         public String getTempDirectory() {
555                 return getNodeValue(new String[] { "temp-directory" }, null);
556         }
557
558         /**
559          * Sets the temp directory to use.
560          *
561          * @param tempDirectory
562          *            The temp directory to use, or {@code null} to use the default
563          *            temp directory
564          */
565         public void setTempDirectory(String tempDirectory) {
566                 if (tempDirectory != null) {
567                         SimpleXML tempDirectoryNode = new SimpleXML("temp-directory");
568                         tempDirectoryNode.setValue(tempDirectory);
569                         rootNode.replace(tempDirectoryNode);
570                 } else {
571                         rootNode.remove("temp-directory");
572                 }
573         }
574
575         /**
576          * Returns whether to use the “early encode“ flag for the insert.
577          *
578          * @return {@code true} to set the “early encode” flag for the insert,
579          *         {@code false} otherwise
580          */
581         public boolean useEarlyEncode() {
582                 return getNodeBooleanValue(new String[] { "use-early-encode" }, false);
583         }
584
585         /**
586          * Sets whether to use the “early encode“ flag for the insert.
587          *
588          * @param useEarlyEncode
589          *            {@code true} to set the “early encode” flag for the insert,
590          *            {@code false} otherwise
591          * @return This configuration
592          */
593         public Configuration setUseEarlyEncode(boolean useEarlyEncode) {
594                 rootNode.replace("use-early-encode", String.valueOf(useEarlyEncode));
595                 return this;
596         }
597
598         /**
599          * Returns the insert priority.
600          *
601          * @return The insert priority
602          */
603         public PriorityClass getPriority() {
604                 return PriorityClass.valueOf(getNodeValue(new String[] { "insert-priority" }, "interactive"));
605         }
606
607         /**
608          * Sets the insert priority.
609          *
610          * @param priority
611          *            The insert priority
612          * @return This configuration
613          */
614         public Configuration setPriority(PriorityClass priority) {
615                 rootNode.replace("insert-priority", priority.toString());
616                 return this;
617         }
618
619         /**
620          * Returns the manifest putter.
621          *
622          * @return The manifest putter
623          */
624         public ManifestPutter getManifestPutter() {
625                 return ManifestPutter.valueOf(getNodeValue(new String[] { "manifest-putter" }, "simple").toUpperCase());
626         }
627
628         /**
629          * Sets the manifest putter.
630          *
631          * @param manifestPutter
632          *            The manifest putter
633          * @return This configuration
634          */
635         public Configuration setManifestPutter(ManifestPutter manifestPutter) {
636                 rootNode.replace("manifest-putter", manifestPutter.name().toLowerCase());
637                 return this;
638         }
639
640 }