Replace edit profile page with Kotlin version
[Sone.git] / src / main / java / net / pterodactylus / sone / template / FilesystemTemplate.java
1 package net.pterodactylus.sone.template;
2
3 import java.io.File;
4 import java.io.FileInputStream;
5 import java.io.IOException;
6 import java.io.InputStream;
7 import java.io.InputStreamReader;
8 import java.io.Reader;
9 import java.io.Writer;
10 import java.util.ArrayList;
11 import java.util.Iterator;
12 import java.util.List;
13 import java.util.concurrent.atomic.AtomicReference;
14
15 import net.pterodactylus.util.template.Part;
16 import net.pterodactylus.util.template.Template;
17 import net.pterodactylus.util.template.TemplateContext;
18 import net.pterodactylus.util.template.TemplateException;
19 import net.pterodactylus.util.template.TemplateParser;
20
21 import com.google.common.base.Charsets;
22
23 /**
24  * {@link Template} implementation that can be reloaded from the filesystem.
25  *
26  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
27  */
28 public class FilesystemTemplate extends Template {
29
30         private final String filename;
31         private final AtomicReference<LastLoadedTemplate> lastTemplate = new AtomicReference<LastLoadedTemplate>();
32         private final TemplateContext initialContext = new TemplateContext();
33         private final List<Part> parts = new ArrayList<Part>();
34
35         public FilesystemTemplate(String filename) {
36                 this.filename = filename;
37         }
38
39         @Override
40         public TemplateContext getInitialContext() {
41                 loadTemplate();
42                 return initialContext;
43         }
44
45         private void loadTemplate() {
46                 File templateFile = new File(filename);
47                 if (templateWasLoaded() && !templateFileHasBeenModifiedAfterLoading(templateFile)) {
48                         return;
49                 }
50                 try (InputStream templateInputStream = new FileInputStream(templateFile);
51                                 Reader templateReader = new InputStreamReader(templateInputStream, Charsets.UTF_8)) {
52                         Template template = TemplateParser.parse(templateReader);
53                         lastTemplate.set(new LastLoadedTemplate(template));
54                         template.getInitialContext().mergeContext(initialContext);
55                         for (Part part : parts) {
56                                 template.add(part);
57                         }
58                 } catch (IOException e) {
59                         throw new TemplateFileNotFoundException(filename);
60                 }
61         }
62
63         private boolean templateWasLoaded() {
64                 return lastTemplate.get() != null;
65         }
66
67         private boolean templateFileHasBeenModifiedAfterLoading(File templateFile) {
68                 return templateFile.lastModified() > lastTemplate.get().getLoadTime();
69         }
70
71         @Override
72         public void add(Part part) {
73                 loadTemplate();
74                 parts.add(part);
75                 lastTemplate.get().getTemplate().add(part);
76         }
77
78         @Override
79         public void render(TemplateContext templateContext, Writer writer) throws TemplateException {
80                 loadTemplate();
81                 lastTemplate.get().getTemplate().render(templateContext, writer);
82         }
83
84         @Override
85         public Iterator<Part> iterator() {
86                 loadTemplate();
87                 return lastTemplate.get().getTemplate().iterator();
88         }
89
90         @Override
91         public int getLine() {
92                 loadTemplate();
93                 return lastTemplate.get().getTemplate().getLine();
94         }
95
96         @Override
97         public int getColumn() {
98                 loadTemplate();
99                 return lastTemplate.get().getTemplate().getColumn();
100         }
101
102         private static class LastLoadedTemplate {
103
104                 private final Template template;
105                 private final long loadTime = System.currentTimeMillis();
106
107                 private LastLoadedTemplate(Template template) {
108                         this.template = template;
109                 }
110
111                 public Template getTemplate() {
112                         return template;
113                 }
114
115                 public long getLoadTime() {
116                         return loadTime;
117                 }
118
119         }
120
121         /**
122          * Exception that signals that a template file could not be found.
123          *
124          * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
125          */
126         public static class TemplateFileNotFoundException extends RuntimeException {
127
128                 public TemplateFileNotFoundException(String filename) {
129                         super(filename);
130                 }
131
132         }
133
134 }