moargration 0.0.2 → 0.0.3

data/README.md

@@ -28,7 +28,7 @@ Worry not, Moargration won't impact your boot time until you actually need it.
 
 So it's time to drop some columns.
 
-Knowing that this can bring your app down, and that you normally need to do two deploys to safely dro pcolumns, you can use Moargration to setup the first step for you.
+Knowing that this can bring your app down, and that you normally need to do two deploys to safely drop columns, you can use Moargration to setup the first step for you.
 
 Just set an `ENV` var like:
 

data/lib/moargration.rb

@@ -5,8 +5,12 @@ module Moargration
 
   def init
     return unless ignore = ENV["MOARGRATION_IGNORE"]
-    spec = parse(ignore)
-    hack_active_record(spec)
+    columns_to_ignore = parse(ignore)
+    hack_active_record!
+  end
+
+  def columns_to_ignore=(columns)
+    @@columns_to_ignore = columns
   end
 
   def columns_to_ignore
@@ -14,20 +18,21 @@ module Moargration
   end
 
   def parse(text)
-    text.strip.split(" ").inject({}) do |parsed, definition|
+    @@columns_to_ignore = text.strip.split(" ").inject({}) do |parsed, definition|
       table, fields = definition.split(":", 2)
       parsed[table] = fields.split(",") if fields
       parsed
     end
   end
 
-  def hack_active_record(spec)
-    @@columns_to_ignore = spec
-
+  def hack_active_record!
     ActiveRecord::Base.class_eval do
-      def self.columns
-        super.reject do |column|
-          (Moargration.columns_to_ignore[table_name] || []).include?(column.name)
+      class << self
+        alias :columns_without_moargration :columns
+        def columns
+          columns_without_moargration.reject do |column|
+            (Moargration.columns_to_ignore[table_name] || []).include?(column.name)
+          end
         end
       end
     end

data/moargration.gemspec

@@ -4,7 +4,7 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |gem|
   gem.name          = "moargration"
-  gem.version       = "0.0.2"
+  gem.version       = "0.0.3"
   gem.authors       = ["Pedro Belo"]
   gem.email         = ["pedro@heroku.com"]
   gem.description   = %q{Helping you migrate, MOAR.}

data/test/moargration_test.rb

@@ -21,12 +21,12 @@ describe Moargration do
 
   describe "hack_active_record" do
     it "hacks the model to ignore the specified columns" do
-      Moargration.hack_active_record("samples" => %w( f1 ))
+      Moargration.columns_to_ignore = { "samples" => %w( f1 ) }
       Sample.columns.map(&:name).wont_include("f1")
     end
 
     it "doesn't affect other models" do
-      Moargration.hack_active_record("samples" => %w( f1 ))
+      Moargration.columns_to_ignore = { "samples" => %w( f1 ) }
       User.columns.map(&:name).must_include("f1")
     end
   end

data/test/test_helper.rb

@@ -19,4 +19,8 @@ class Sample < ActiveRecord::Base
 end
 
 class User < ActiveRecord::Base
-end
+end
+
+# enable the hack
+ENV["MOARGRATION_IGNORE"] = ""
+Moargration.hack_active_record!

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: moargration
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors: