Support YAML templates

features/task_template.feature

@@ -35,3 +35,15 @@ Feature: `template' task keyword
       """
     When I execute the recipe
     Then the output must contain "basic template"
+
+  Scenario: parses a yaml file
+    Given a file named "templates/basic.yaml" with:
+      """
+      foo: bar
+      """
+    And a recipe with:
+      """
+      task(:echo_template) { echo template('basic')['foo'] }
+      """
+    When I execute the recipe
+    Then the output must match /^bar$/

lib/producer/core.rb

@@ -3,6 +3,7 @@ require 'etc'
 require 'forwardable'
 require 'optparse'
 require 'pathname'
+require 'yaml'
 
 require 'net/ssh'
 require 'net/sftp'

lib/producer/core/template.rb

@@ -4,21 +4,40 @@ module Producer
       SEARCH_PATH = 'templates'.freeze
 
       def initialize(path, search_path: SEARCH_PATH)
-        @path         = Pathname.new("#{path}.erb")
+        @path         = Pathname.new(path)
         @search_path  = Pathname.new(search_path)
       end
 
       def render(variables = {})
-        tpl = ERB.new(File.read(resolve_path), nil, '-')
+        case (file_path = resolve_path).extname
-        tpl.filename = resolve_path.to_s
+        when '.yaml'  then render_yaml file_path
-        tpl.result build_erb_binding variables
+        when '.erb'   then render_erb file_path, variables
+        end
       end
 
 
       private
 
+      def render_erb(file_path, variables = {})
+        tpl = ERB.new(File.read(file_path), nil, '-')
+        tpl.filename = file_path.to_s
+        tpl.result build_erb_binding variables
+      end
+
+      def render_yaml(file_path)
+        YAML.load(File.read(file_path))
+      end
+
       def resolve_path
-        if @path.to_s =~ /\A\.\// then @path else @search_path + @path end
+        if @path.to_s =~ /\A\.\//
+          resolve_suffix @path
+        else
+          resolve_suffix @search_path + @path
+        end
+      end
+
+      def resolve_suffix(path)
+        Pathname.glob("#{path}.{erb,yaml}").first
       end
 
       def build_erb_binding(variables)

spec/fixtures/templates/basic_yaml.yaml

@@ -0,0 +1 @@
+foo: bar

spec/producer/core/template_spec.rb

@@ -13,6 +13,14 @@ module Producer::Core
         expect(template.render).to eq "basic template\n"
       end
 
+      context 'yaml templates' do
+        let(:path) { 'basic_yaml' }
+
+        it 'renders yaml templates' do
+          expect(template.render).to eq({ 'foo' => 'bar' })
+        end
+      end
+
       context 'when variables are given' do
         let(:path) { 'variables' }