bespoke 0.1.1 → 0.1.2

data/lib/bespoke/exportable.rb

@@ -14,13 +14,39 @@ class Bespoke
       @name = name.to_sym
       @fields = {}
       @joins = {}
+      @helpers = {}
       @logger = logger || Logger.new(STDERR)
     end
 
+    class HelperClass
+      def initialize(helpers, context)
+        @helpers, @context = helpers, context
+      end
+
+      def has_key?(key)
+        @helpers.has_key?(key) ||
+        @context.has_key?(key)
+      end
+
+      def [](key)
+        if @helpers.has_key?(key)
+          @helpers[key].call(@context)
+        elsif @context.has_key?(key)
+          @context[key]
+        else
+          super(key)
+        end
+      end
+    end
+
     def headers
       @fields.keys
     end
 
+    def helper(name, &block)
+      @helpers[name] = block
+    end
+
     def field(name, template_string)
       fields[name] = Mustache::Template.new(template_string)
     end
@@ -33,6 +59,7 @@ class Bespoke
       raise "hashes missing #{@name.inspect} (of: #{hashes.keys.inspect})" unless hashes.has_key?(@name)
       hashes[@name].map do |main_key, row|
         context = { @name => row }
+        context[:helper] = HelperClass.new(@helpers, context)
         @joins.each_pair do |join_name, key|
           if other_table = hashes[join_name.to_sym]
             if other_table.has_key?(row[key])

data/lib/bespoke/version.rb

@@ -1,3 +1,3 @@
 class Bespoke
-  VERSION = '0.1.1' unless defined?(Bespoke::VERSION)
+  VERSION = '0.1.2' unless defined?(Bespoke::VERSION)
 end

data/spec/exportable_spec.rb

@@ -5,20 +5,51 @@ require 'stringio'
 describe Bespoke::Exportable do
   let(:logger) { nil }
   let(:export) { Bespoke::Exportable.new(:test, logger) }
+  let(:test_data) {
+    {
+      1 => {:one => 1, :two => 2},
+      2 => {:one => 5, :two => 10}
+    }
+  }
+
+  before do
+    export.field(:column, "{{test.one}}-{{test.two}}")
+  end
 
   it "initializes" do
     export
   end
 
   it "exports templated rows" do
-    export.field(:column, "{{test.one}}-{{test.two}}")
-    data = export.export({:test => {
-      1 => {:one => 1, :two => 2},
-      2 => {:one => 5, :two => 10}
-    }})
+    data = export.export(:test => test_data)
     data.should == [["1-2"], ["5-10"]]
   end
 
+  context "helper methods" do
+    it "can be declared" do
+      export.field(:helped, "{{helper.helper_method}}")
+      export.helper(:helper_method) { |row| "help!" }
+
+      data = export.export(:test => test_data)
+      data.should == [["1-2", "help!"], ["5-10", "help!"]]
+    end
+
+    it "has access to row data" do
+      export.field(:helped, "{{helper.fallback}}")
+      export.helper(:fallback) do |row|
+        row[:test][:one] || row[:test][:two]
+      end
+
+      data = export.export(:test =>
+        {
+          1 => {:two => 2},
+          2 => {:one => 5, :two => 10}
+        }
+      )
+      data.should == [["-2", "2"], ["5-10", "5"]]
+    end
+  end
+
   context "logging" do
     let(:output) { StringIO.new }
     let(:logger) { Logger.new(output) }

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bespoke
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-02-07 00:00:00.000000000 Z
+date: 2014-02-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler