redundant_column 0.0.1 → 0.1.0

data/.gitignore

@@ -15,3 +15,5 @@ spec/reports
 test/tmp
 test/version_tmp
 tmp
+*~
+nbproject/*

data/README.md

@@ -1,6 +1,9 @@
 # RedundantColumn
 
-TODO: Write a gem description
+Auto maintain the consistency of redundant column.
+
+1. Auto update redundant column.
+2. Auto fill redundant column value from redundant target object.
 
 ## Installation
 
@@ -18,7 +21,26 @@ Or install it yourself as:
 
 ## Usage
 
-TODO: Write usage instructions here
+    class User < ActiveRecord::Base
+      has_many :topics, :redundant_column => {:status => :user_status}
+    end
+
+    class Topic < ActiveRecord::Base
+      belongs_to :user, :redundant_column => {:user_status => :status}
+    end
+
+    user = User.create(:name => "Hooopo", :status => "active")
+
+    topic = Topic.create(:user => user, :title => "this is title")
+
+    assert_equal topic.user_status, user.status
+
+    user.status = "disable"
+    user.save
+
+    topic.reload
+
+    assert_equal topic.user_status, user.status
 
 ## Contributing
 

data/lib/redundant_column.rb

@@ -1,34 +1,3 @@
 require "redundant_column/version"
-
-module RedundantColumn
-  def redundant_column(name, target_model, options = {}) 
-    set_callback :update, :before, CallbackBuilder.new(name, target_model, options)
-  end
-
-  class CallbackBuilder
-    attr_reader :name, :target_model, :target_name, :association_name
-
-    def initialize(name, target_model, options = {})
-      @name = name
-      @target_model = target_model
-      @target_name =  options[:target_name] 
-      @association_name = options[:association_name] || "#{target_model.name.downcase.pluralize}"
-    end
-
-    def before_update(caller) 
-      if caller.send("#{name}_changed?")
-        if caller.respond_to?(association_name)
-          new_column_value = caller.send(name)
-          target_name ||= "#{caller.class.name.downcase}_#{name}"
-          update_sql = new_column_value.is_a?(String) ? %Q|#{target_name} = '#{new_column_value.gsub("'", "\\'")}'| : %Q|#{target_name} = #{new_column_value}|
-          caller.send(association_name).update_all(update_sql)
-        else
-          STDERR.puts "You must set association in #{caller.class} model:"
-          STDERR.puts "has_many :#{association_name}, ..."
-        end
-      end
-    end
-  end
-end
-
-ActiveRecord::Base.send :extend, RedundantColumn
+require 'redundant_column/has_many'
+require 'redundant_column/belongs_to'

data/lib/redundant_column/belongs_to.rb

@@ -0,0 +1,45 @@
+module RedundantColumn
+  module BelongsTo
+    class CallbackBuilder
+      attr_reader :name, :target_name, :association_name
+
+      def initialize(name, target_name, association_name)
+        @name = name
+        @target_name =  target_name
+        @association_name = association_name
+      end
+
+      def before_create(caller)
+        if caller.read_attribute(name).nil?
+          column_value = caller.send(association_name).try(target_name)
+          caller.send("#{name}=", column_value)
+          true
+        end
+      end
+    end
+
+    def self.included(base)
+      base.valid_options += [:redundant_column]
+    end
+
+    def build
+      reflection = super
+      add_counter_cache_callbacks(reflection) if options[:counter_cache]
+      add_touch_callbacks(reflection)         if options[:touch]
+      configure_dependency
+      configure_auto_fill_redundant_column_callback
+      reflection
+    end
+
+    private
+
+    def configure_auto_fill_redundant_column_callback
+      if columns = options[:redundant_column]
+        column = columns.to_a.first
+        target_name = columns.to_a.last
+        model.set_callback :create, :before, CallbackBuilder.new(column, target_name, self.name)
+      end
+    end
+  end
+end
+ActiveRecord::Associations::Builder::BelongsTo.send(:include, RedundantColumn::BelongsTo)

data/lib/redundant_column/has_many.rb

@@ -0,0 +1,62 @@
+module RedundantColumn
+  module HasMany
+    class CallbackBuilder
+      attr_reader :name, :target_name, :association_name
+
+      def initialize(name, target_name, association_name)
+        @name = name
+        @target_name =  target_name
+        @association_name = association_name
+      end
+
+      def before_update(caller)
+        if caller.send("#{name}_changed?")
+          if caller.respond_to?(association_name)
+            new_column_value = caller.send(name)
+
+            update_sql =  if new_column_value.is_a?(String)
+              %Q|#{target_name} = '#{new_column_value.gsub("'", "\\'")}'|
+            else
+              %Q|#{target_name} = #{new_column_value}|
+            end
+            caller.send(association_name).update_all(update_sql)
+          else
+            STDERR.puts "You must set association in #{caller.class} model:"
+            STDERR.puts "has_many :#{association_name}, ..."
+          end
+        end
+      end
+    end
+
+
+    def self.included(base)
+      base.valid_options += [:redundant_column]
+    end
+
+    def build
+      reflection = super
+      configure_dependency
+      configure_update_redundant_column_callback
+      reflection
+    end
+
+    private
+
+    def configure_update_redundant_column_callback
+      if columns = options[:redundant_column]
+        if columns.is_a?(Array)
+          columns.each do |column|
+            model.set_callback :update, :before, CallbackBuilder.new(column, model.name.downcase + "_#{column}", self.name)
+          end
+        elsif columns.is_a?(Hash)
+          columns.each do |column, target_name|
+            model.set_callback :update, :before, CallbackBuilder.new(column, target_name, self.name)
+          end
+        else
+          model.set_callback :update, :before, CallbackBuilder.new(column, model.name.downcase + "_#{columns}", self.name)
+        end
+      end
+    end
+  end
+end
+ActiveRecord::Associations::Builder::HasMany.send(:include, RedundantColumn::HasMany)

data/lib/redundant_column/version.rb

@@ -1,3 +1,3 @@
 module RedundantColumn
-  VERSION = "0.0.1"
+  VERSION = "0.1.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: redundant_column
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-07-02 00:00:00.000000000 Z
+date: 2012-07-12 00:00:00.000000000 Z
 dependencies: []
 description: redundant column for activerecord
 email:
@@ -24,6 +24,8 @@ files:
 - README.md
 - Rakefile
 - lib/redundant_column.rb
+- lib/redundant_column/belongs_to.rb
+- lib/redundant_column/has_many.rb
 - lib/redundant_column/version.rb
 - redundant_column.gemspec
 homepage: ''