Add accessor for Notifications.

diff --git a/src/main/java/net/pterodactylus/sone/template/NotificationAccessor.java b/src/main/java/net/pterodactylus/sone/template/NotificationAccessor.java
new file mode 100644 (file)
index 0000000..22c53c3
--- /dev/null
+++ b/src/main/java/net/pterodactylus/sone/template/NotificationAccessor.java
@@ -0,0 +1,57 @@
+/*
+ * Sone - NotificationAccessor.java - Copyright © 2010 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 3 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, see <http://www.gnu.org/licenses/>.
+ */
+
+package net.pterodactylus.sone.template;
+
+import java.io.IOException;
+import java.io.StringWriter;
+
+import net.pterodactylus.util.notify.Notification;
+import net.pterodactylus.util.template.DataProvider;
+import net.pterodactylus.util.template.ReflectionAccessor;
+
+/**
+ * Adds additional properties to a {@link Notification}.
+ * <dl>
+ * <dd>render</dd>
+ * <dt>Returns the rendered notification.</dt>
+ * </dl>
+ *
+ * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
+ */
+public class NotificationAccessor extends ReflectionAccessor {
+
+       /**
+        * {@inheritDoc}
+        */
+       @Override
+       public Object get(DataProvider dataProvider, Object object, String member) {
+               Notification notification = (Notification) object;
+               if ("render".equals(member)) {
+                       StringWriter stringWriter = new StringWriter();
+                       try {
+                               notification.render(stringWriter);
+                       } catch (IOException ioe1) {
+                               /* TODO - log. */
+                               /* ignore. */
+                       }
+                       return stringWriter.toString();
+               }
+               return super.get(dataProvider, object, member);
+       }
+
+}