From 6b7da8376218972db4d0a98a9e0220f9dc81dbd5 Mon Sep 17 00:00:00 2001
From: Jesse Luehrs <doy@tozt.net>
Date: Wed, 3 Oct 2012 18:34:41 -0500
Subject: make suffixes work for partials

---
 lib/Text/Handlebars.pm          |  5 ++++-
 lib/Text/Handlebars/Compiler.pm | 12 ++++++++++++
 t/mustache.t                    |  4 +---
 3 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/lib/Text/Handlebars.pm b/lib/Text/Handlebars.pm
index 4cb1723..7c975b9 100644
--- a/lib/Text/Handlebars.pm
+++ b/lib/Text/Handlebars.pm
@@ -81,7 +81,10 @@ sub _register_builtin_methods {
     };
     $funcs->{'(find_file)'} = sub {
         my ($filename) = @_;
-        return try { $weakself->find_file($filename); 1 } catch { undef };
+        return 1 if try { $weakself->find_file($filename); 1 };
+        $filename .= $weakself->{suffix};
+        return 1 if try { $weakself->find_file($filename); 1 };
+        return 0;
     };
 }
 
diff --git a/lib/Text/Handlebars/Compiler.pm b/lib/Text/Handlebars/Compiler.pm
index 5f112fc..2fc7422 100644
--- a/lib/Text/Handlebars/Compiler.pm
+++ b/lib/Text/Handlebars/Compiler.pm
@@ -3,6 +3,8 @@ use Any::Moose;
 
 extends 'Text::Xslate::Compiler';
 
+use Try::Tiny;
+
 has '+syntax' => (
     default => 'Handlebars',
 );
@@ -19,6 +21,16 @@ sub _generate_block {
     return @compiled;
 }
 
+sub _generate_include {
+    my $self = shift;
+    my ($node) = @_;
+
+    my $file = $node->first;
+    $file->id($file->id . $self->engine->{suffix})
+        unless try { $self->engine->find_file($file->id); 1 };
+    return $self->SUPER::_generate_include($node);
+}
+
 if (0) {
     our $_recursing;
     around compile_ast => sub {
diff --git a/t/mustache.t b/t/mustache.t
index bbd9227..8034db2 100644
--- a/t/mustache.t
+++ b/t/mustache.t
@@ -152,9 +152,8 @@ RENDERED
     "comments"
 );
 
-{ local $TODO = "unimplemented";
 render_file_ok(
-    { path => ['t/mustache/partials'] },
+    { path => ['t/mustache/partials'], suffix => '.mustache' },
     'base.mustache',
     {
         names => [
@@ -171,7 +170,6 @@ render_file_ok(
 EXPECTED
     "partials"
 );
-}
 
 render_ok(
     <<'TEMPLATE',
-- 
cgit v1.2.3-54-g00ecf