cached_column 0.1 → 0.2

data/cached_column.gemspec

@@ -12,7 +12,7 @@ Gem::Specification.new do |gem|
   gem.test_files    = gem.files.grep(%r{^(test|spec|features)/})
   gem.name          = "cached_column"
   gem.require_paths = ["lib"]
-  gem.version       = "0.1"
+  gem.version       = "0.2"
 
   gem.add_dependency "activerecord", "~> 3.0"
   gem.add_development_dependency "rspec"

data/lib/cached_column.rb

@@ -1,22 +1,29 @@
 require 'active_record'
 
 class CachedColumn
-  attr_accessor :column, :options, :instance_method
+  attr_accessor :column, :options, :block
 
-  def initialize(column, instance_method, options = {})
-    @column          = column
-    @options         = options
-    @instance_method = instance_method
+  def initialize(column, options = {}, &block)
+    @column  = column
+    @options = options
+    @block   = block
   end
 
   def before_save(record)
-    record.send("#{column}=", instance_method.bind(record).call)
+    record.send("#{column}=", computed_value(record))
+  end
+
+  def computed_value(record)
+    if block
+      record.instance_eval(&block)
+    else
+      record.send(options[:method] || column)
+    end
   end
 end
 
 class ActiveRecord::Base
-  def self.cached_column(column, options = {})
-    before_save CachedColumn.new(column, instance_method(options[:method] || column), options)
-    define_method_attribute(column.to_s)
+  def self.cached_column(column, options = {}, &block)
+    before_save CachedColumn.new(column, options, &block)
   end
 end

data/spec/cached_column_spec.rb

@@ -34,6 +34,13 @@ describe CachedColumn do
       instance[:cached_column].should == 43
     end
 
+    it "sets the attribute value to the result of instance evaluating the block" do
+      model.cached_column(:cached_column) { calculating_method }
+      instance = model.new
+      instance.save
+      instance[:cached_column].should == 43
+    end
+
     it "allows the original method to be called" do
       model.cached_column :cached_column
       instance = model.new

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cached_column
 version: !ruby/object:Gem::Version
-  version: '0.1'
+  version: '0.2'
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-15 00:00:00.000000000 Z
+date: 2012-10-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord