Allow specifying alternate suffixes

phly · Aug 30, 2010 · 95ec004 · 95ec004
1 parent fcfa41e
commit 95ec004
Show file tree

Hide file tree

Showing 5 changed files with 35 additions and 2 deletions.
diff --git a/.project b/.project
@@ -48,6 +48,7 @@ mustache=/home/matthew/git/phly_mustache CD=. filter="*.php *.html *.xml *.txt"
      ViewWithTraversableObject.php
     }
     templates=templates filter="*.mustache" {
+     alternate-suffix.html
      partial-template.mustache
      renders-file-templates.mustache
      template-with-aliased-partial.mustache

diff --git a/TODO b/TODO
@@ -1,4 +1,3 @@
-Allow specifying a template suffix
 API documentation
 Tests 
     for comments

diff --git a/library/Phly/Mustache/Mustache.php b/library/Phly/Mustache/Mustache.php
@@ -26,6 +26,9 @@ class Mustache
     /** @var Renderer */
     protected $renderer;
 
+    /** @var string Suffix used when resolving templates */
+    protected $suffix = '.mustache';
+
     /**
      * Constructor
      * 
@@ -104,6 +107,28 @@ public function setTemplatePath($path)
         return $this;
     }
 
+    /**
+     * Set suffix used when resolving templates
+     * 
+     * @param  string $suffix 
+     * @return Mustache
+     */
+    public function setSuffix($suffix)
+    {
+        $this->suffix = '.' . ltrim($suffix, '.');
+        return $this;
+    }
+
+    /**
+     * Get template suffix
+     * 
+     * @return string
+     */
+    public function getSuffix()
+    {
+        return $this->suffix;
+    }
+
     /**
      * Render a template using a view, and optionally a list of partials
      * 
@@ -174,7 +199,7 @@ public function tokenize($template)
     protected function fetchTemplate($template)
     {
         foreach ($this->templatePath as $path) {
-            $file = $path . DIRECTORY_SEPARATOR . $template . '.mustache';
+            $file = $path . DIRECTORY_SEPARATOR . $template . $this->getSuffix();
             if (file_exists($file)) {
                 $content = file_get_contents($file);
                 $this->cachedTemplates[$template] = $content;

diff --git a/tests/PhlyTest/Mustache/MustacheTest.php b/tests/PhlyTest/Mustache/MustacheTest.php
@@ -332,4 +332,11 @@ public function testHonorsImplicitIteratorPragma()
 EOT;
         $this->assertEquals($expected, $test);
     }
+
+    public function testAllowsSettingAlternateTemplateSuffix()
+    {
+        $this->mustache->setSuffix('html');
+        $test = $this->mustache->render('alternate-suffix', array());
+        $this->assertContains('alternate template suffix', $test);
+    }
 }
diff --git a/tests/PhlyTest/Mustache/templates/alternate-suffix.html b/tests/PhlyTest/Mustache/templates/alternate-suffix.html
@@ -0,0 +1 @@
+alternate template suffix