add a scalaobjecthandler that doesnt require the twitter libraries

Sam Pullara · Sam Pullara · commit 958198b25c6f · 2012-09-02T09:56:19.000-07:00
diff --git a/compiler/src/main/scala/com/twitter/mustache/ScalaObjectHandler.scala b/compiler/src/main/scala/com/twitter/mustache/ScalaObjectHandler.scala
@@ -0,0 +1,58 @@
+package com.twitter.mustache
+
+import collection.JavaConversions._
+import com.github.mustachejava.Iteration
+import com.github.mustachejava.reflect.ReflectionObjectHandler
+import java.io.Writer
+import java.lang.reflect.{Field, Method}
+
+/**
+ * Plain old scala handler that doesn't depend on Twitter libraries.
+ */
+class ScalaObjectHandler extends ReflectionObjectHandler {
+
+  // Allow any method or field
+  override def checkMethod(member: Method) {}
+
+  override def checkField(member: Field) {}
+
+  override def coerce(value: Object) = {
+    value match {
+      case m: Map[_, _] => asJavaMap(m)
+      case o: Option[_] => o match {
+        case Some(some: Object) => coerce(some)
+        case None => null
+      }
+      case _ => value
+    }
+  }
+
+  override def iterate(iteration: Iteration, writer: Writer, value: Object, scopes: Array[Object]) = {
+    value match {
+      case t: Traversable[AnyRef] => {
+        var newWriter = writer
+        t map {
+          next =>
+            newWriter = iteration.next(newWriter, coerce(next), scopes)
+        }
+        newWriter
+      }
+      case n: Number => if (n == 0) writer else iteration.next(writer, coerce(value), scopes)
+      case _ => super.iterate(iteration, writer, value, scopes)
+    }
+  }
+
+  override def falsey(iteration: Iteration, writer: Writer, value: Object, scopes: Array[Object]) = {
+    value match {
+      case t: Traversable[AnyRef] => {
+        if (t.isEmpty) {
+          iteration.next(writer, value, scopes)
+        } else {
+          writer
+        }
+      }
+      case n: Number => if (n == 0) iteration.next(writer, coerce(value), scopes) else writer
+      case _ => super.falsey(iteration, writer, value, scopes)
+    }
+  }
+}
diff --git a/compiler/src/test/scala/com/twitter/mustache/ObjectHandlerTest.scala b/compiler/src/test/scala/com/twitter/mustache/ObjectHandlerTest.scala
@@ -3,12 +3,12 @@ package com.twitter.mustache
 import com.github.mustachejava.DefaultMustacheFactory
 import com.twitter.util.FuturePool
 import java.io.{StringWriter, StringReader}
-import java.util.concurrent.Executors
+import java.util.concurrent.{Callable, Executors}
 import org.junit.{Assert, Test}
 
 class ObjectHandlerTest {
   @Test
-  def testHandler() {
+  def testTwitterHandler() {
     val pool = Executors.newCachedThreadPool()
     val futurePool = FuturePool(pool)
     val mf = new DefaultMustacheFactory()
@@ -43,4 +43,40 @@ class ObjectHandlerTest {
     writer.close()
     Assert.assertEquals("Hello, world!?!\nGoodbye, thanks for all the fish!!?test\n", sw.toString)
   }
+
+  @Test
+  def testScalaHandler() {
+    val pool = Executors.newCachedThreadPool()
+    val mf = new DefaultMustacheFactory()
+    mf.setObjectHandler(new TwitterObjectHandler)
+    mf.setExecutorService(pool)
+    val m = mf.compile(
+      new StringReader("{{#list}}{{optionalHello}}, {{futureWorld}}!" +
+              "{{#test}}?{{/test}}{{^test}}!{{/test}}{{#num}}?{{/num}}{{^num}}!{{/num}}" +
+              "{{#map}}{{value}}{{/map}}\n{{/list}}"),
+      "helloworld"
+    )
+    val sw = new StringWriter
+    val writer = m.execute(sw, new {
+      val list = Seq(new {
+        lazy val optionalHello = Some("Hello")
+        val futureWorld = pool.submit(new Callable[String] {
+          def call(): String = "world"
+        })
+        val test = true
+        val num = 0
+      }, new {
+        val optionalHello = Some("Goodbye")
+        val futureWorld = pool.submit(new Callable[String] {
+          def call(): String = "thanks for all the fish"
+        })
+        lazy val test = false
+        val map = Map(("value", "test"))
+        val num = 1
+      })
+    })
+    // You must use close if you use concurrent latched writers
+    writer.close()
+    Assert.assertEquals("Hello, world!?!\nGoodbye, thanks for all the fish!!?test\n", sw.toString)
+  }
 }
