Merge pull request #103 from devinrsmith/feature/mustache-source

added MustacheCustomContext

README.md

@@ -75,6 +75,7 @@ void executeTemplate (Reader template, Writer out, Map<String, String> data) {
 The execution context can be any Java object. Variables will be resolved via the following
 mechanisms:
 
+  * If the context is a `MustacheCustomContext`, `MustacheCustomContext.get` will be used.
   * If the context is a `Map`, `Map.get` will be used.
   * If a non-void method with the same name as the variable exists, it will be called.
   * If a non-void method named (for variable `foo`) `getFoo` exists, it will be called.

src/main/java/com/samskivert/mustache/BasicCollector.java

@@ -35,6 +35,7 @@ public abstract class BasicCollector implements Mustache.Collector
     }
 
     public Mustache.VariableFetcher createFetcher (Object ctx, String name) {
+        if (ctx instanceof MustacheCustomContext) return CUSTOM_FETCHER;
         if (ctx instanceof Map<?,?>) return MAP_FETCHER;
 
         // if the name looks like a number, potentially use one of our 'indexing' fetchers
@@ -65,6 +66,17 @@ public abstract class BasicCollector implements Mustache.Collector
         return null;
     }
 
+    protected static final Mustache.VariableFetcher CUSTOM_FETCHER = new Mustache.VariableFetcher() {
+        public Object get (Object ctx, String name) throws Exception {
+            MustacheCustomContext custom = (MustacheCustomContext)ctx;
+            Object val = custom.get(name);
+            return val == null ? Template.NO_FETCHER_FOUND : val;
+        }
+        @Override public String toString () {
+            return "CUSTOM_FETCHER";
+        }
+    };
+
     protected static final Mustache.VariableFetcher MAP_FETCHER = new Mustache.VariableFetcher() {
         public Object get (Object ctx, String name) throws Exception {
             Map<?,?> map = (Map<?,?>)ctx;

src/main/java/com/samskivert/mustache/MustacheCustomContext.java

@@ -0,0 +1,8 @@
+package com.samskivert.mustache;
+
+/**
+ * Provides a simple interface for callers to implement their own logic for contextual values.
+ */
+public interface MustacheCustomContext {
+    Object get(String name) throws Exception;
+}

src/test/java/com/samskivert/mustache/MustacheTest.java

@@ -45,6 +45,15 @@ public class MustacheTest extends SharedTests
         });
     }
 
+    @Test public void testMustacheCustomContext() {
+        test("bar", "{{foo}}", new MustacheCustomContext() {
+            @Override
+            public Object get(String name) {
+                return "foo".equals(name) ? "bar" : null;
+            }
+        });
+    }
+
     public interface HasDefault {
         default String getFoo () { return "bar"; }
     }