Add plugins to template factory.
[Sone.git] / src / main / java / net / pterodactylus / sone / web / WebInterface.java
1 /*
2  * FreenetSone - WebInterface.java - Copyright © 2010 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 3 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, see <http://www.gnu.org/licenses/>.
16  */
17
18 package net.pterodactylus.sone.web;
19
20 import java.io.InputStream;
21 import java.io.InputStreamReader;
22 import java.io.Reader;
23 import java.io.UnsupportedEncodingException;
24 import java.net.URI;
25 import java.net.URISyntaxException;
26 import java.util.ArrayList;
27 import java.util.List;
28 import java.util.logging.Level;
29 import java.util.logging.Logger;
30
31 import net.pterodactylus.sone.core.Core;
32 import net.pterodactylus.sone.data.Post;
33 import net.pterodactylus.sone.data.Sone;
34 import net.pterodactylus.sone.freenet.L10nFilter;
35 import net.pterodactylus.sone.main.SonePlugin;
36 import net.pterodactylus.sone.template.GetPagePlugin;
37 import net.pterodactylus.sone.template.PostAccessor;
38 import net.pterodactylus.sone.template.RequestChangeFilter;
39 import net.pterodactylus.sone.template.SoneAccessor;
40 import net.pterodactylus.sone.template.SubstringFilter;
41 import net.pterodactylus.sone.web.ajax.GetSoneStatusPage;
42 import net.pterodactylus.sone.web.ajax.GetTranslationPage;
43 import net.pterodactylus.sone.web.page.PageToadlet;
44 import net.pterodactylus.sone.web.page.PageToadletFactory;
45 import net.pterodactylus.sone.web.page.StaticPage;
46 import net.pterodactylus.util.logging.Logging;
47 import net.pterodactylus.util.service.AbstractService;
48 import net.pterodactylus.util.template.DateFilter;
49 import net.pterodactylus.util.template.DefaultTemplateFactory;
50 import net.pterodactylus.util.template.PaginationPlugin;
51 import net.pterodactylus.util.template.ReflectionAccessor;
52 import net.pterodactylus.util.template.Template;
53 import net.pterodactylus.util.template.TemplateException;
54 import net.pterodactylus.util.template.TemplateFactory;
55 import net.pterodactylus.util.template.TemplateProvider;
56 import net.pterodactylus.util.template.XmlFilter;
57 import freenet.clients.http.SessionManager;
58 import freenet.clients.http.ToadletContainer;
59 import freenet.l10n.BaseL10n;
60
61 /**
62  * Bundles functionality that a web interface of a Freenet plugin needs, e.g.
63  * references to l10n helpers.
64  *
65  * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
66  */
67 public class WebInterface extends AbstractService {
68
69         /** The logger. */
70         private static final Logger logger = Logging.getLogger(WebInterface.class);
71
72         /** The Sone plugin. */
73         private final SonePlugin sonePlugin;
74
75         /** The registered toadlets. */
76         private final List<PageToadlet> pageToadlets = new ArrayList<PageToadlet>();
77
78         /**
79          * Creates a new web interface.
80          *
81          * @param sonePlugin
82          *            The Sone plugin
83          */
84         public WebInterface(SonePlugin sonePlugin) {
85                 super("Sone Web Interface", false);
86                 this.sonePlugin = sonePlugin;
87         }
88
89         //
90         // ACCESSORS
91         //
92
93         /**
94          * Returns the Sone core used by the Sone plugin.
95          *
96          * @return The Sone core
97          */
98         public Core core() {
99                 return sonePlugin.core();
100         }
101
102         /**
103          * Returns the l10n helper of the node.
104          *
105          * @return The node’s l10n helper
106          */
107         public BaseL10n l10n() {
108                 return sonePlugin.l10n().getBase();
109         }
110
111         /**
112          * Returns the session manager of the node.
113          *
114          * @return The node’s session manager
115          */
116         public SessionManager sessionManager() {
117                 try {
118                         return sonePlugin.pluginRespirator().getSessionManager(new URI("/"));
119                 } catch (URISyntaxException use1) {
120                         logger.log(Level.SEVERE, "Could not get Session Manager!", use1);
121                         return null;
122                 }
123         }
124
125         //
126         // SERVICE METHODS
127         //
128
129         /**
130          * {@inheritDoc}
131          */
132         @Override
133         protected void serviceStart() {
134                 registerToadlets();
135         }
136
137         /**
138          * {@inheritDoc}
139          */
140         @Override
141         protected void serviceStop() {
142                 unregisterToadlets();
143         }
144
145         //
146         // PRIVATE METHODS
147         //
148
149         /**
150          * Register all toadlets.
151          */
152         private void registerToadlets() {
153                 DefaultTemplateFactory templateFactory = new DefaultTemplateFactory();
154                 templateFactory.addAccessor(Object.class, new ReflectionAccessor());
155                 templateFactory.addAccessor(Sone.class, new SoneAccessor(core()));
156                 templateFactory.addAccessor(Post.class, new PostAccessor(core()));
157                 templateFactory.addFilter("date", new DateFilter());
158                 templateFactory.addFilter("l10n", new L10nFilter(l10n()));
159                 templateFactory.addFilter("substring", new SubstringFilter());
160                 templateFactory.addFilter("xml", new XmlFilter());
161                 templateFactory.addFilter("change", new RequestChangeFilter());
162                 templateFactory.addPlugin("getpage", new GetPagePlugin());
163                 templateFactory.addPlugin("paginate", new PaginationPlugin());
164                 templateFactory.setTemplateProvider(new ClassPathTemplateProvider(templateFactory));
165                 templateFactory.addTemplateObject("formPassword", sonePlugin.pluginRespirator().getToadletContainer().getFormPassword());
166
167                 Template loginTemplate = templateFactory.createTemplate(createReader("/templates/login.html"));
168                 Template indexTemplate = templateFactory.createTemplate(createReader("/templates/index.html"));
169                 Template addSoneTemplate = templateFactory.createTemplate(createReader("/templates/addSone.html"));
170                 Template loadSoneTemplate = templateFactory.createTemplate(createReader("/templates/loadSone.html"));
171                 Template knownSonesTemplate = templateFactory.createTemplate(createReader("/templates/knownSones.html"));
172                 Template createSoneTemplate = templateFactory.createTemplate(createReader("/templates/createSone.html"));
173                 Template createPostTemplate = templateFactory.createTemplate(createReader("/templates/createPost.html"));
174                 Template createReplyTemplate = templateFactory.createTemplate(createReader("/templates/createReply.html"));
175                 Template editProfileTemplate = templateFactory.createTemplate(createReader("/templates/editProfile.html"));
176                 Template backupProfileTemplate = templateFactory.createTemplate(createReader("/templates/backup.xml"));
177                 Template viewSoneTemplate = templateFactory.createTemplate(createReader("/templates/viewSone.html"));
178                 Template blockSoneTemplate = templateFactory.createTemplate(createReader("/templates/blockSone.html"));
179                 Template unblockSoneTemplate = templateFactory.createTemplate(createReader("/templates/unblockSone.html"));
180                 Template viewPostTemplate = templateFactory.createTemplate(createReader("/templates/viewPost.html"));
181                 Template deletePostTemplate = templateFactory.createTemplate(createReader("/templates/deletePost.html"));
182                 Template deleteReplyTemplate = templateFactory.createTemplate(createReader("/templates/deleteReply.html"));
183                 Template followSoneTemplate = templateFactory.createTemplate(createReader("/templates/followSone.html"));
184                 Template unfollowSoneTemplate = templateFactory.createTemplate(createReader("/templates/unfollowSone.html"));
185                 Template deleteSoneTemplate = templateFactory.createTemplate(createReader("/templates/deleteSone.html"));
186                 Template noPermissionTemplate = templateFactory.createTemplate(createReader("/templates/noPermission.html"));
187                 Template logoutTemplate = templateFactory.createTemplate(createReader("/templates/logout.html"));
188
189                 PageToadletFactory pageToadletFactory = new PageToadletFactory(sonePlugin.pluginRespirator().getHLSimpleClient(), "/Sone/");
190                 pageToadlets.add(pageToadletFactory.createPageToadlet(new IndexPage(indexTemplate, this), "Index"));
191                 pageToadlets.add(pageToadletFactory.createPageToadlet(new CreateSonePage(createSoneTemplate, this), "CreateSone"));
192                 pageToadlets.add(pageToadletFactory.createPageToadlet(new LoadSonePage(loadSoneTemplate, this)));
193                 pageToadlets.add(pageToadletFactory.createPageToadlet(new AddSonePage(addSoneTemplate, this)));
194                 pageToadlets.add(pageToadletFactory.createPageToadlet(new KnownSonesPage(knownSonesTemplate, this), "KnownSones"));
195                 pageToadlets.add(pageToadletFactory.createPageToadlet(new EditProfilePage(editProfileTemplate, this), "EditProfile"));
196                 pageToadlets.add(pageToadletFactory.createPageToadlet(new BackupProfilePage(backupProfileTemplate, this)));
197                 pageToadlets.add(pageToadletFactory.createPageToadlet(new CreatePostPage(createPostTemplate, this)));
198                 pageToadlets.add(pageToadletFactory.createPageToadlet(new CreateReplyPage(createReplyTemplate, this)));
199                 pageToadlets.add(pageToadletFactory.createPageToadlet(new ViewSonePage(viewSoneTemplate, this)));
200                 pageToadlets.add(pageToadletFactory.createPageToadlet(new BlockSonePage(blockSoneTemplate, this)));
201                 pageToadlets.add(pageToadletFactory.createPageToadlet(new UnblockSonePage(unblockSoneTemplate, this)));
202                 pageToadlets.add(pageToadletFactory.createPageToadlet(new ViewPostPage(viewPostTemplate, this)));
203                 pageToadlets.add(pageToadletFactory.createPageToadlet(new DeletePostPage(deletePostTemplate, this)));
204                 pageToadlets.add(pageToadletFactory.createPageToadlet(new DeleteReplyPage(deleteReplyTemplate, this)));
205                 pageToadlets.add(pageToadletFactory.createPageToadlet(new FollowSonePage(followSoneTemplate, this)));
206                 pageToadlets.add(pageToadletFactory.createPageToadlet(new UnfollowSonePage(unfollowSoneTemplate, this)));
207                 pageToadlets.add(pageToadletFactory.createPageToadlet(new DeleteSonePage(deleteSoneTemplate, this), "DeleteSone"));
208                 pageToadlets.add(pageToadletFactory.createPageToadlet(new LoginPage(loginTemplate, this), "Login"));
209                 pageToadlets.add(pageToadletFactory.createPageToadlet(new LogoutPage(logoutTemplate, this), "Logout"));
210                 pageToadlets.add(pageToadletFactory.createPageToadlet(new SoneTemplatePage("noPermission.html", noPermissionTemplate, "Page.NoPermission.Title", this)));
211                 pageToadlets.add(pageToadletFactory.createPageToadlet(new StaticPage("css/", "/static/css/", "text/css")));
212                 pageToadlets.add(pageToadletFactory.createPageToadlet(new StaticPage("javascript/", "/static/javascript/", "text/javascript")));
213                 pageToadlets.add(pageToadletFactory.createPageToadlet(new StaticPage("images/", "/static/images/", "image/png")));
214                 pageToadlets.add(pageToadletFactory.createPageToadlet(new GetTranslationPage(this)));
215                 pageToadlets.add(pageToadletFactory.createPageToadlet(new GetSoneStatusPage(core())));
216
217                 ToadletContainer toadletContainer = sonePlugin.pluginRespirator().getToadletContainer();
218                 toadletContainer.getPageMaker().addNavigationCategory("/Sone/index.html", "Navigation.Menu.Name", "Navigation.Menu.Tooltip", sonePlugin);
219                 for (PageToadlet toadlet : pageToadlets) {
220                         String menuName = toadlet.getMenuName();
221                         if (menuName != null) {
222                                 toadletContainer.register(toadlet, "Navigation.Menu.Name", toadlet.path(), true, "Navigation.Menu.Item." + menuName + ".Name", "Navigation.Menu.Item." + menuName + ".Tooltip", false, toadlet);
223                         } else {
224                                 toadletContainer.register(toadlet, null, toadlet.path(), true, false);
225                         }
226                 }
227         }
228
229         /**
230          * Unregisters all toadlets.
231          */
232         private void unregisterToadlets() {
233                 ToadletContainer toadletContainer = sonePlugin.pluginRespirator().getToadletContainer();
234                 for (PageToadlet pageToadlet : pageToadlets) {
235                         toadletContainer.unregister(pageToadlet);
236                 }
237                 toadletContainer.getPageMaker().removeNavigationCategory("Navigation.Menu.Name");
238         }
239
240         /**
241          * Creates a {@link Reader} from the {@link InputStream} for the resource
242          * with the given name.
243          *
244          * @param resourceName
245          *            The name of the resource
246          * @return A {@link Reader} for the resource
247          */
248         private Reader createReader(String resourceName) {
249                 try {
250                         return new InputStreamReader(getClass().getResourceAsStream(resourceName), "UTF-8");
251                 } catch (UnsupportedEncodingException uee1) {
252                         return null;
253                 }
254         }
255
256         /**
257          * Template provider implementation that uses
258          * {@link WebInterface#createReader(String)} to load templates for
259          * inclusion.
260          *
261          * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
262          */
263         private class ClassPathTemplateProvider implements TemplateProvider {
264
265                 /** The template factory. */
266                 private final TemplateFactory templateFactory;
267
268                 /**
269                  * Creates a new template provider that locates templates on the
270                  * classpath.
271                  *
272                  * @param templateFactory
273                  *            The template factory to create the templates
274                  */
275                 public ClassPathTemplateProvider(TemplateFactory templateFactory) {
276                         this.templateFactory = templateFactory;
277                 }
278
279                 /**
280                  * {@inheritDoc}
281                  */
282                 @Override
283                 @SuppressWarnings("synthetic-access")
284                 public Template getTemplate(String templateName) {
285                         Reader templateReader = createReader("/templates/" + templateName);
286                         if (templateReader == null) {
287                                 return null;
288                         }
289                         Template template = templateFactory.createTemplate(templateReader);
290                         try {
291                                 template.parse();
292                         } catch (TemplateException te1) {
293                                 logger.log(Level.WARNING, "Could not parse template “" + templateName + "” for inclusion!", te1);
294                         }
295                         return template;
296                 }
297
298         }
299 }