add support for {{.}}

sps · spullara · commit 30bfa4573108 · 2011-03-28T12:50:18.000+08:00
this is useful for data structures that are not associative, like an array of primitives.
for example:

{ list:[1,2,3] }

{{#list}}
    {{.}}
{{/list}}
diff --git a/src/main/java/com/sampullara/mustache/Mustache.java b/src/main/java/com/sampullara/mustache/Mustache.java
@@ -37,9 +37,13 @@
  * Time: 10:12:47 AM
  */
 public abstract class Mustache {
+  
   protected static Logger logger = Logger.getLogger(Mustache.class.getName());
   private static final boolean debug = Boolean.getBoolean("mustache.debug");
   public static final boolean trace = Boolean.getBoolean("mustache.trace");
+  
+  private static final String IMPLICIT_CURRENT_ELEMENT_TOKEN = ".";
+  
   private File root;
   private String path;
 
@@ -485,7 +489,13 @@ protected Iterable<Scope> inverted(final Scope s, final String name) {
 
   protected Object getValue(Scope s, String name) {
     try {
+        
       Object o = s.get(name);
+      
+      if (o == null && IMPLICIT_CURRENT_ELEMENT_TOKEN.equals(name)) {
+          o = s.values().iterator().next();
+      }
+      
       if (o == null && debug) {
         StringBuilder sb = new StringBuilder();
         for (StackTraceElement ste : Thread.currentThread().getStackTrace()) {
diff --git a/src/test/java/com/sampullara/mustache/CompilerTest.java b/src/test/java/com/sampullara/mustache/CompilerTest.java
@@ -306,6 +306,36 @@ public void testJson() throws IOException, MustacheException {
 
   }
 
+  @SuppressWarnings("serial")
+  public void testCurrentElementInArray() throws IOException, MustacheException {
+      
+      MustacheCompiler c = init();
+      Mustache m = c.parseFile("simple_array.html");
+      StringWriter sw = new StringWriter();
+      FutureWriter writer = new FutureWriter(sw);
+      m.execute(writer, new Scope(new HashMap<String, Object>() {
+          {
+              put("list", Arrays.asList(1,2,3));
+          }
+      }));
+      writer.flush();
+      assertEquals(getContents(root, "simple_array.txt"), sw.toString());
+      
+      /*
+       * verify null elements in a list are properly handled when using {{.}} 
+       */
+      sw = new StringWriter();
+      writer = new FutureWriter(sw);
+      m.execute(writer, new Scope(new HashMap<String, Object>() {
+          {
+              put("list", Arrays.asList(null,null));
+          }
+      }));
+      writer.flush();
+      assertEquals("\n\n", sw.toString());
+      
+  }
+  
   public void testReadme() throws MustacheException, IOException {
     MustacheCompiler c = init();
     Mustache m = c.parseFile("items.html");
diff --git a/src/test/resources/simple_array.html b/src/test/resources/simple_array.html
@@ -0,0 +1,3 @@
+{{#list}}
+{{.}}
+{{/list}}
diff --git a/src/test/resources/simple_array.txt b/src/test/resources/simple_array.txt
@@ -0,0 +1,3 @@
+1
+2
+3
