mover_postgres 0.3.6

data/.gitignore

@@ -0,0 +1,8 @@
+.DS_Store
+*.gem
+coverage
+pkg
+spec/config/database.yml
+spec/log
+tmp
+.rvmrc

data/LICENSE

@@ -0,0 +1,18 @@
+Copyright (c) 2010
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of 
+this software and associated documentation files (the "Software"), to deal in 
+the Software without restriction, including without limitation the rights to 
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of 
+the Software, and to permit persons to whom the Software is furnished to do so, 
+subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all 
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS 
+FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR 
+COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER 
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN 
+CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,101 @@
+Mover
+=====
+
+Move ActiveRecord records across tables like it ain't no thang.
+
+Requirements
+------------
+
+<pre>
+sudo gem install mover
+</pre>
+
+Move records
+------------
+
+Move the last article:
+
+<pre>
+Article.last.move_to(ArticleArchive)
+</pre>
+
+Move today's articles:
+
+<pre>
+Article.move_to(
+  ArticleArchive,
+  :conditions => [ "created_at > ?", Date.today ]
+)
+</pre>
+
+The two tables do not have to be identical. Only shared columns transfer.
+
+If a primary key collision occurs, the destination record is updated.
+
+Callbacks
+---------
+
+In this example, we want an "archive" table for articles and comments.
+
+We also want the article's comments to be archived when the article is.
+
+<pre>
+class Article &lt; ActiveRecord::Base
+  has_many :comments
+  before_move :ArticleArchive do
+    comments.each { |c| c.move_to(CommentArchive) }
+  end
+end
+
+class ArticleArchive &lt; ActiveRecord::Base
+  has_many :comments, :class_name => 'CommentArchive', :foreign_key => 'article_id'
+  before_move :Article do
+    comments.each { |c| c.move_to(Comment) }
+  end
+end
+
+class Comment &lt; ActiveRecord::Base
+  belongs_to :article
+end
+
+class CommentArchive &lt; ActiveRecord::Base
+  belongs_to :article, :class_name => 'ArticleArchive', :foreign_key => 'article_id'
+end
+</pre>
+
+The <code>after\_move</code> callback is also available.
+
+Magic column
+------------
+
+If a table contains a <code>moved_at</code> column, it will magically populate with the date and time it was moved.
+
+Options
+-------
+
+There are other options, in addition to <code>conditions</code>:
+
+<pre>
+Article.move_to(
+  ArticleArchive,
+  :copy => true,          # Do not delete Article after move
+  :generic => true,       # UPDATE using a JOIN instead of ON DUPLICATE KEY UPDATE (default: false on MySQL engines)
+  :magic => 'updated_at', # Custom magic column
+  :migrate => true,       # Copies the original value of the magic column
+  :quick => true          # You are certain only INSERTs are necessary, no primary key collisions possible
+                          # May only be a little faster on MySQL, but dramatically faster on other engines
+)
+</pre>
+
+You can access these options from callbacks using <code>move_options</code>.
+
+Reserve a spot
+--------------
+
+Before you create a record, you can "reserve a spot" on a table that you will move the record to later.
+
+<pre>
+archive = ArticleArchive.new
+archive.id = Article.reserve_id
+archive.save
+</pre>

data/Rakefile

@@ -0,0 +1,90 @@
+require File.dirname(__FILE__) + '/lib/mover/gems'
+
+Mover::Gems.activate %w(rake rspec)
+
+require 'rake'
+require 'spec/rake/spectask'
+
+def gemspec
+  @gemspec ||= begin
+    file = File.expand_path('../mover.gemspec', __FILE__)
+    eval(File.read(file), binding, file)
+  end
+end
+
+if defined?(Spec::Rake::SpecTask)
+  desc "Run specs"
+  Spec::Rake::SpecTask.new do |t|
+    t.spec_files = FileList['spec/**/*_spec.rb']
+    t.spec_opts = %w(-fs --color)
+    t.warning = true
+  end
+  task :spec
+  task :default => :spec
+end
+
+desc "Build gem(s)"
+task :gem do
+  old_gemset = ENV['GEMSET']
+  root = File.expand_path('../', __FILE__)
+  pkg = "#{root}/pkg"
+  system "rm -Rf #{pkg}"
+  Mover::Gems.gemset_names.each do |gemset|
+    ENV['GEMSET'] = gemset.to_s
+    system "cd #{root} && gem build mover.gemspec"
+    system "mkdir -p #{pkg} && mv *.gem pkg"
+  end
+  ENV['GEMSET'] = old_gemset
+end
+
+namespace :gem do
+  desc "Install gem(s)"
+  task :install do
+    Rake::Task['gem'].invoke
+    Dir["#{File.dirname(__FILE__)}/pkg/*.gem"].each do |pkg|
+      system "gem install #{pkg} --no-ri --no-rdoc"
+    end
+  end
+  
+  desc "Push gem(s)"
+  task :push do
+    Rake::Task['gem'].invoke
+    Dir["#{File.dirname(__FILE__)}/pkg/*.gem"].each do |pkg|
+      system "gem push #{pkg}"
+    end
+  end
+end
+
+namespace :gems do
+  desc "Install gem dependencies (DEV=0 DOCS=0 GEMSPEC=default SUDO=0)"
+  task :install do
+    dev = ENV['DEV'] == '1'
+    docs = ENV['DOCS'] == '1' ? '' : '--no-ri --no-rdoc'
+    gemset = ENV['GEMSET']
+    sudo = ENV['SUDO'] == '1' ? 'sudo' : ''
+    
+    Mover::Gems.gemset = gemset if gemset
+    
+    if dev
+      gems = Mover::Gems.gemspec.development_dependencies
+    else
+      gems = Mover::Gems.gemspec.dependencies
+    end
+    
+    gems.each do |name|
+      name = name.to_s
+      version = Mover::Gems.versions[name]
+      if Gem.source_index.find_name(name, version).empty?
+        version = version ? "-v #{version}" : ''
+        system "#{sudo} gem install #{name} #{version} #{docs}"
+      else
+        puts "already installed: #{name} #{version}"
+      end
+    end
+  end
+end
+
+desc "Validate the gemspec"
+task :gemspec do
+  gemspec.validate
+end

data/config/gemsets.yml

@@ -0,0 +1,5 @@
+mover:
+
+  rspec: ~>1.0
+  default:
+    active_wrapper: =0.4.4

data/config/gemspec.yml

@@ -0,0 +1,13 @@
+name: mover_postgres
+version: 0.3.6
+authors:
+  - Winton Welsh
+email: james@njtechnologies.co.uk
+homepage: https://github.com/jamesharker/mover
+summary: Move ActiveRecord records across tables like it ain't no thang
+description: Move ActiveRecord records across tables like it ain't no thang
+dependencies: null
+development_dependencies:
+  - active_wrapper
+  - rake
+  - rspec

data/init.rb

@@ -0,0 +1 @@
+require File.dirname(__FILE__) + "/rails/init"

data/lib/mover.rb

@@ -0,0 +1,213 @@
+require File.dirname(__FILE__) + '/mover/gems'
+
+$:.unshift File.dirname(__FILE__)
+
+module Mover
+  
+  def self.included(base)
+    unless base.included_modules.include?(InstanceMethods)
+      base.extend ClassMethods
+      base.send :include, InstanceMethods
+      base.send :attr_accessor, :move_options
+    end
+  end
+  
+  module ClassMethods
+    
+    def after_move(*to_class, &block)
+      @after_move ||= []
+      @after_move << [ to_class, block ]
+    end
+    
+    def before_move(*to_class, &block)
+      @before_move ||= []
+      @before_move << [ to_class, block ]
+    end
+
+    def move_to(to_class, options={})
+      from_class = self
+      
+      # Conditions
+      conditions = options[:conditions] || '1'
+      conditions = self.sanitize_sql(conditions)
+      where = "WHERE #{conditions}"
+      
+      # Columns
+      magic = options[:magic] || 'moved_at'
+      from = {
+        :columns => from_class.column_names,
+        :table => from_class.table_name
+      }
+      to = {
+        :columns => to_class.column_names,
+        :table => to_class.table_name
+      }
+      
+      # insert[column] = value
+      insert = (from[:columns] & to[:columns]).inject({}) do |hash, column|
+        if column == magic && !options[:migrate]
+          hash[column] = Time.now.utc
+        else
+          hash[column] = column
+        end
+        hash
+      end
+      
+      # Magic column not in "from" table
+      if to[:columns].include?(magic) && !from[:columns].include?(magic)
+        insert[magic] = Time.now.utc
+      end
+      
+      # Quote everything
+      insert = insert.inject({}) do |hash, (column, value)|
+        if value.is_a?(::Time)
+          hash[connection.quote_column_name(column)] = connection.quote(value)
+        else
+          hash[connection.quote_column_name(column)] = connection.quote_column_name(value)
+        end
+        hash
+      end
+      
+      # Callbacks
+      collector = lambda do |(classes, block)|
+        classes.collect! { |c| eval(c.to_s) }
+        block if classes.include?(to_class) || classes.empty?
+      end
+      before = (@before_move || []).collect(&collector).compact
+      after = (@after_move || []).collect(&collector).compact
+      
+      # Instances
+      instances =
+        if options[:instance]
+          [ options[:instance] ]
+        elsif before.empty? && after.empty?
+          []
+        else
+          self.find(:all, :conditions => conditions)
+        end
+      options.delete(:instance)
+      
+      # Callback executor
+      exec_callbacks = lambda do |callbacks|
+        callbacks.each do |block|
+          instances.each do |instance|
+            instance.move_options = options
+            instance.instance_eval(&block)
+            instance.move_options = nil
+          end
+        end
+      end
+      
+      # Execute
+      transaction do
+        exec_callbacks.call before
+        
+        if options[:quick]
+          connection.execute(<<-SQL)
+            INSERT INTO #{to[:table]} (#{insert.keys.join(', ')})
+            SELECT #{insert.values.join(', ')}
+            FROM #{from[:table]}
+            #{where}
+          SQL
+        elsif !options[:generic] && connection.class.to_s.include?('Mysql')
+          update = insert.collect do |column, value|
+            if value.include?("'")
+              "#{to[:table]}.#{column} = #{value}"
+            else
+              "#{to[:table]}.#{column} = #{from[:table]}.#{value}"
+            end
+          end
+          
+          connection.execute(<<-SQL)
+            INSERT INTO #{to[:table]} (#{insert.keys.join(', ')})
+            SELECT #{insert.values.join(', ')}
+            FROM #{from[:table]}
+            #{where}
+            ON DUPLICATE KEY
+            UPDATE #{update.join(', ')};
+          SQL
+        else
+          conditions.gsub!(to[:table], 't')
+          conditions.gsub!(from[:table], 'f')
+          conditions.gsub!(/\"id\"/,'f.id') if connection.class.to_s.include?('PostgreSQL')
+          
+          select = insert.values.collect { |i| i.include?("'") ? i : "f.#{i}" }
+          set = insert.collect do |column, value|
+            prefix = 't.' unless connection.class.to_s.include?('PostgreSQL')
+            if value.include?("'")
+              "#{prefix}#{column} = #{value}"
+            else
+              "#{prefix}#{column} = f.#{value}"
+            end
+          end
+          
+          if connection.class.to_s.include?('PostgreSQL')
+            connection.execute(<<-SQL)
+              UPDATE #{to[:table]}
+                AS t
+              SET #{set.join(', ')}
+              FROM #{from[:table]}
+                AS f
+              WHERE f.id = t.id
+                AND #{conditions}
+            SQL
+          else
+            connection.execute(<<-SQL)
+              UPDATE #{to[:table]}
+                AS t
+              INNER JOIN #{from[:table]}
+                AS f
+              ON f.id = t.id
+                AND #{conditions}
+              SET #{set.join(', ')}
+            SQL
+          end
+      
+          connection.execute(<<-SQL)
+            INSERT INTO #{to[:table]} (#{insert.keys.join(', ')})
+            SELECT #{select.join(', ')}
+            FROM #{from[:table]}
+              AS f
+            LEFT OUTER JOIN #{to[:table]}
+              AS t
+            ON f.id = t.id
+            WHERE (
+              t.id IS NULL
+              AND #{conditions}
+            )
+          SQL
+        end
+        
+        unless options[:copy]
+          connection.execute("DELETE FROM #{from[:table]} #{where}")
+        end
+        
+        exec_callbacks.call after
+      end
+    end
+
+    def reserve_id
+      id = nil
+      transaction do
+        if connection.class.to_s.include?('PostgreSQL')
+          id = connection.insert("INSERT INTO #{self.table_name} (id) VALUES (nextval('#{self.table_name}_id_seq'::regclass))").to_i
+        else
+          id = connection.insert("INSERT INTO #{self.table_name} () VALUES ()")
+        end
+        connection.execute("DELETE FROM #{self.table_name} WHERE id = #{id}") if id
+      end
+      id
+    end
+  end
+  
+  module InstanceMethods
+    
+    def move_to(to_class, options={})
+      options[:conditions] = "#{self.class.table_name}.#{self.class.primary_key} = #{id}"
+      options[:instance] = self
+      self.class.move_to(to_class, options)
+    end
+  end
+end
+
+ActiveRecord::Base.send(:include, Mover)

data/lib/mover/gems.rb

@@ -0,0 +1,154 @@
+unless defined?(Mover::Gems)
+  
+  require 'yaml'
+  
+  module Mover
+    module Gems
+      class <<self
+        
+        attr_accessor :config
+        attr_reader :gemset, :gemsets, :versions
+        
+        class SimpleStruct
+          attr_reader :hash
+          
+          def initialize(hash)
+            @hash = hash
+            @hash.each do |key, value|
+              self.class.send(:define_method, key) { @hash[key] }
+              self.class.send(:define_method, "#{key}=") { |v| @hash[key] = v }
+            end
+          end
+        end
+        
+        Gems.config = SimpleStruct.new(
+          :gemsets => [ "#{File.expand_path('../../../', __FILE__)}/config/gemsets.yml" ],
+          :gemspec => "#{File.expand_path('../../../', __FILE__)}/config/gemspec.yml",
+          :warn => true
+        )
+        
+        def activate(*gems)
+          begin
+            require 'rubygems' unless defined?(::Gem)
+          rescue LoadError
+            puts "rubygems library could not be required" if @config.warn
+          end
+          
+          self.gemset ||= gemset_from_loaded_specs
+          
+          gems.flatten.collect(&:to_sym).each do |name|
+            version = @versions[name]
+            vendor = File.expand_path("../../../vendor/#{name}/lib", __FILE__)
+            if File.exists?(vendor)
+              $:.unshift vendor
+            elsif defined?(gem)
+              gem name.to_s, version
+            else
+              puts "#{name} #{"(#{version})" if version} failed to activate" if @config.warn
+            end
+          end
+        end
+        
+        def dependencies
+          dependency_filter(@gemspec.dependencies, @gemset)
+        end
+        
+        def development_dependencies
+          dependency_filter(@gemspec.development_dependencies, @gemset)
+        end
+        
+        def gemset=(gemset)
+          if gemset
+            @gemset = gemset.to_sym
+        
+            @gemsets = @config.gemsets.reverse.collect { |config|
+              if config.is_a?(::String)
+                YAML::load(File.read(config)) rescue {}
+              elsif config.is_a?(::Hash)
+                config
+              end
+            }.inject({}) do |hash, config|
+              deep_merge(hash, symbolize_keys(config))
+            end
+            
+            @versions = (@gemsets[gemspec.name.to_sym] || {}).inject({}) do |hash, (key, value)|
+              if !value.is_a?(::Hash) && value
+                hash[key] = value
+              elsif key == @gemset
+                (value || {}).each { |k, v| hash[k] = v }
+              end
+              hash
+            end
+          else
+            @gemset = nil
+            @gemsets = nil
+            @versions = nil
+          end
+        end
+        
+        def gemset_names
+          (
+            [ :default ] +
+            @gemsets[gemspec.name.to_sym].inject([]) { |array, (key, value)|
+              array.push(key) if value.is_a?(::Hash) || value.nil?
+              array
+            }
+          ).uniq
+        end
+        
+        def gemspec(reload=false)
+          if @gemspec && !reload
+            @gemspec
+          else
+            data = YAML::load(File.read(@config.gemspec)) rescue {}
+            @gemspec = SimpleStruct.new(data)
+          end
+        end
+      
+        private
+      
+        def deep_merge(first, second)
+          merger = lambda do |key, v1, v2|
+            Hash === v1 && Hash === v2 ? v1.merge(v2, &merger) : v2
+          end
+          first.merge(second, &merger)
+        end
+        
+        def dependency_filter(dependencies, match)
+          (dependencies || []).inject([]) { |array, value|
+            if value.is_a?(::Hash)
+              array += value[match.to_s] if value[match.to_s]
+            else
+              array << value
+            end
+            array
+          }.uniq.collect(&:to_sym)
+        end
+        
+        def gemset_from_loaded_specs
+          if defined?(Gem)
+            Gem.loaded_specs.each do |name, spec|
+              if name == gemspec.name
+                return :default
+              elsif name[0..gemspec.name.length] == "#{gemspec.name}-"
+                return name[gemspec.name.length+1..-1].to_sym
+              end
+            end
+            :default
+          else
+            :none
+          end
+        end
+      
+        def symbolize_keys(hash)
+          return {} unless hash.is_a?(::Hash)
+          hash.inject({}) do |options, (key, value)|
+            value = symbolize_keys(value) if value.is_a?(::Hash)
+            options[(key.to_sym rescue key) || key] = value
+            options
+          end
+        end
+      end
+    end
+  end
+end

data/mover.gemspec

@@ -0,0 +1,32 @@
+# -*- encoding: utf-8 -*-
+root = File.expand_path('../', __FILE__)
+lib = "#{root}/lib"
+$:.unshift lib unless $:.include?(lib)
+ 
+require 'mover/gems'
+Mover::Gems.gemset ||= ENV['GEMSET'] || :default
+
+Gem::Specification.new do |s|
+  Mover::Gems.gemspec.hash.each do |key, value|
+    if key == 'name' && Mover::Gems.gemset != :default
+      s.name = "#{value}-#{Mover::Gems.gemset}"
+    elsif key == 'summary' && Mover::Gems.gemset == :solo
+      s.summary = value + " (no dependencies)"
+    elsif !%w(dependencies development_dependencies).include?(key)
+      s.send "#{key}=", value
+    end
+  end
+
+  Mover::Gems.dependencies.each do |g|
+    s.add_dependency g.to_s, Mover::Gems.versions[g]
+  end
+  
+  Mover::Gems.development_dependencies.each do |g|
+    s.add_development_dependency g.to_s, Mover::Gems.versions[g]
+  end
+
+  s.executables = `cd #{root} && git ls-files -- {bin}/*`.split("\n").collect { |f| File.basename(f) }
+  s.files = `cd #{root} && git ls-files`.split("\n")
+  s.require_paths = %w(lib)
+  s.test_files = `cd #{root} && git ls-files -- {features,test,spec}/*`.split("\n")
+end

data/rails/init.rb

@@ -0,0 +1 @@
+require File.expand_path(File.dirname(__FILE__) + '/../lib/mover')

data/spec/Rakefile

@@ -0,0 +1,9 @@
+require File.expand_path(File.dirname(__FILE__) + '/../lib/mover/gems')
+Mover::Gems.activate(:active_wrapper)
+
+require 'active_wrapper/tasks'
+
+ActiveWrapper::Tasks.new(
+  :base => File.dirname(__FILE__),
+  :env => ENV['ENV']
+)

data/spec/config/database.yml.example

@@ -0,0 +1,6 @@
+test:
+  adapter: mysql
+  database: mover
+  username: root
+  password: 
+  host: localhost

data/spec/db/migrate/001_create_fixtures.rb

@@ -0,0 +1,28 @@
+class CreateFixtures < ActiveRecord::Migration
+  def self.up
+    [ :articles, :article_archives ].each do |table|
+      create_table table do |t|
+        t.string :title
+        t.string :body
+        t.boolean :read
+        t.datetime :moved_at
+      end
+    end
+    
+    [ :comments, :comment_archives ].each do |table|
+      create_table table do |t|
+        t.string :title
+        t.string :body
+        t.boolean :read
+        t.integer :article_id
+        t.datetime :moved_at
+      end
+    end
+  end
+
+  def self.down
+    [ :articles, :article_archives, :comments, :comment_archives ].each do |table|
+      drop_table table
+    end
+  end
+end

data/spec/fixtures/article.rb

@@ -0,0 +1,6 @@
+class Article < ActiveRecord::Base
+  has_many :comments
+  before_move :ArticleArchive do
+    comments.each { |c| c.move_to(CommentArchive, move_options) }
+  end
+end

data/spec/fixtures/article_archive.rb

@@ -0,0 +1,6 @@
+class ArticleArchive < ActiveRecord::Base
+  has_many :comments, :class_name => 'CommentArchive', :foreign_key => 'article_id'
+  before_move :Article do
+    comments.each { |c| c.move_to(Comment, move_options) }
+  end
+end

data/spec/fixtures/comment.rb

@@ -0,0 +1,3 @@
+class Comment < ActiveRecord::Base
+  belongs_to :article
+end

data/spec/fixtures/comment_archive.rb

@@ -0,0 +1,3 @@
+class CommentArchive < ActiveRecord::Base
+  belongs_to :article, :class_name => 'ArticleArchive', :foreign_key => 'article_id'
+end

data/spec/fixtures/gemsets.yml

@@ -0,0 +1,9 @@
+name:
+  rake: =0.8.7
+  default:
+    mysql: =2.8.1
+    rspec: =1.3.1
+  rspec2:
+    mysql2: =0.2.6
+    rspec: =2.3.0
+  solo: null

data/spec/fixtures/gemspec.yml

@@ -0,0 +1,15 @@
+name: name
+version: 0.1.0
+authors:
+  - Author
+email: email@email.com
+homepage: http://github.com/author/name
+summary: Summary
+description: Description
+dependencies:
+  - rake
+  - default:
+    - mysql
+  - rspec2:
+    - mysql2
+development_dependencies: null

data/spec/mover/gems_spec.rb

@@ -0,0 +1,249 @@
+require 'spec_helper'
+
+describe Mover::Gems do
+  
+  before(:each) do
+    @old_config = Mover::Gems.config
+    
+    Mover::Gems.config.gemspec = "#{$root}/spec/fixtures/gemspec.yml"
+    Mover::Gems.config.gemsets = [
+      "#{$root}/spec/fixtures/gemsets.yml"
+    ]
+    Mover::Gems.config.warn = true
+    
+    Mover::Gems.gemspec true
+    Mover::Gems.gemset = nil
+  end
+  
+  after(:each) do
+    Mover::Gems.config = @old_config
+  end
+  
+  describe :activate do
+    it "should activate gems" do
+      Mover::Gems.stub!(:gem)
+      Mover::Gems.should_receive(:gem).with('rspec', '=1.3.1')
+      Mover::Gems.should_receive(:gem).with('rake', '=0.8.7')
+      Mover::Gems.activate :rspec, 'rake'
+    end
+  end
+  
+  describe :gemset= do
+    before(:each) do
+      Mover::Gems.config.gemsets = [
+        {
+          :name => {
+            :rake => '>0.8.6',
+            :default => {
+              :externals => '=1.0.2'
+            }
+          }
+        },
+        "#{$root}/spec/fixtures/gemsets.yml"
+      ]
+    end
+    
+    describe :default do
+      before(:each) do
+        Mover::Gems.gemset = :default
+      end
+      
+      it "should set @gemset" do
+        Mover::Gems.gemset.should == :default
+      end
+    
+      it "should set @gemsets" do
+        Mover::Gems.gemsets.should == {
+          :name => {
+            :rake => ">0.8.6",
+            :default => {
+              :externals => '=1.0.2',
+              :mysql => "=2.8.1",
+              :rspec => "=1.3.1"
+            },
+            :rspec2 => {
+              :mysql2 => "=0.2.6",
+              :rspec => "=2.3.0"
+            },
+            :solo => nil
+          }
+        }
+      end
+    
+      it "should set Gems.versions" do
+        Mover::Gems.versions.should == {
+          :externals => "=1.0.2",
+          :mysql => "=2.8.1",
+          :rake => ">0.8.6",
+          :rspec => "=1.3.1"
+        }
+      end
+      
+      it "should return proper values for Gems.dependencies" do
+        Mover::Gems.dependencies.should == [ :rake, :mysql ]
+        Mover::Gems.development_dependencies.should == []
+      end
+      
+      it "should return proper values for Gems.gemset_names" do
+        Mover::Gems.gemset_names.should == [ :default, :rspec2, :solo ]
+      end
+    end
+    
+    describe :rspec2 do
+      before(:each) do
+        Mover::Gems.gemset = "rspec2"
+      end
+      
+      it "should set @gemset" do
+        Mover::Gems.gemset.should == :rspec2
+      end
+    
+      it "should set @gemsets" do
+        Mover::Gems.gemsets.should == {
+          :name => {
+            :rake => ">0.8.6",
+            :default => {
+              :externals => '=1.0.2',
+              :mysql => "=2.8.1",
+              :rspec => "=1.3.1"
+            },
+            :rspec2 => {
+              :mysql2=>"=0.2.6",
+              :rspec => "=2.3.0"
+            },
+            :solo => nil
+          }
+        }
+      end
+    
+      it "should set Gems.versions" do
+        Mover::Gems.versions.should == {
+          :mysql2 => "=0.2.6",
+          :rake => ">0.8.6",
+          :rspec => "=2.3.0"
+        }
+      end
+      
+      it "should return proper values for Gems.dependencies" do
+        Mover::Gems.dependencies.should == [ :rake, :mysql2 ]
+        Mover::Gems.development_dependencies.should == []
+      end
+      
+      it "should return proper values for Gems.gemset_names" do
+        Mover::Gems.gemset_names.should == [ :default, :rspec2, :solo ]
+      end
+    end
+    
+    describe :solo do
+      before(:each) do
+        Mover::Gems.gemset = :solo
+      end
+      
+      it "should set @gemset" do
+        Mover::Gems.gemset.should == :solo
+      end
+    
+      it "should set @gemsets" do
+        Mover::Gems.gemsets.should == {
+          :name => {
+            :rake => ">0.8.6",
+            :default => {
+              :externals => '=1.0.2',
+              :mysql => "=2.8.1",
+              :rspec => "=1.3.1"
+            },
+            :rspec2 => {
+              :mysql2=>"=0.2.6",
+              :rspec => "=2.3.0"
+            },
+            :solo => nil
+          }
+        }
+      end
+    
+      it "should set Gems.versions" do
+        Mover::Gems.versions.should == {:rake=>">0.8.6"}
+      end
+      
+      it "should return proper values for Gems.dependencies" do
+        Mover::Gems.dependencies.should == [:rake]
+        Mover::Gems.development_dependencies.should == []
+      end
+      
+      it "should return proper values for Gems.gemset_names" do
+        Mover::Gems.gemset_names.should == [ :default, :rspec2, :solo ]
+      end
+    end
+    
+    describe :nil do
+      before(:each) do
+        Mover::Gems.gemset = nil
+      end
+      
+      it "should set everything to nil" do
+        Mover::Gems.gemset.should == nil
+        Mover::Gems.gemsets.should == nil
+        Mover::Gems.versions.should == nil
+      end
+    end
+  end
+  
+  describe :gemset_from_loaded_specs do
+    before(:each) do
+      Gem.stub!(:loaded_specs)
+    end
+    
+    it "should return the correct gemset for name gem" do
+      Gem.should_receive(:loaded_specs).and_return({ "name" => nil })
+      Mover::Gems.send(:gemset_from_loaded_specs).should == :default
+    end
+    
+    it "should return the correct gemset for name-rspec gem" do
+      Gem.should_receive(:loaded_specs).and_return({ "name-rspec2" => nil })
+      Mover::Gems.send(:gemset_from_loaded_specs).should == :rspec2
+    end
+  end
+  
+  describe :reload_gemspec do
+    it "should populate @gemspec" do
+      Mover::Gems.gemspec.hash.should == {
+        "name" => "name",
+        "version" => "0.1.0",
+        "authors" => ["Author"],
+        "email" => "email@email.com",
+        "homepage" => "http://github.com/author/name",
+        "summary" => "Summary",
+        "description" => "Description",
+        "dependencies" => [
+          "rake",
+          { "default" => [ "mysql" ] },
+          { "rspec2" => [ "mysql2" ] }
+        ],
+        "development_dependencies" => nil
+       }
+    end
+  
+    it "should create methods from keys of @gemspec" do
+      Mover::Gems.gemspec.name.should == "name"
+      Mover::Gems.gemspec.version.should == "0.1.0"
+      Mover::Gems.gemspec.authors.should == ["Author"]
+      Mover::Gems.gemspec.email.should == "email@email.com"
+      Mover::Gems.gemspec.homepage.should == "http://github.com/author/name"
+      Mover::Gems.gemspec.summary.should == "Summary"
+      Mover::Gems.gemspec.description.should == "Description"
+      Mover::Gems.gemspec.dependencies.should == [
+        "rake",
+        { "default" => ["mysql"] },
+        { "rspec2" => [ "mysql2" ] }
+      ]
+      Mover::Gems.gemspec.development_dependencies.should == nil
+    end
+  
+    it "should produce a valid gemspec" do
+      Mover::Gems.gemset = :default
+      gemspec = File.expand_path("../../../mover.gemspec", __FILE__)
+      gemspec = eval(File.read(gemspec), binding, gemspec)
+      gemspec.validate.should == true
+    end
+  end
+end

data/spec/mover_spec.rb

@@ -0,0 +1,137 @@
+require 'spec_helper'
+
+describe Mover do
+    
+  [ :none, :generic, :quick ].each do |option|
+    
+    describe "(option: #{option})" do
+      
+      before(:all) do
+        @options = {}
+        @options[option] = true
+        $db.drop_db
+        $db.create_db
+        $db.establish_connection
+      end
+    
+      before(:each) do
+        [ 1, 0, 1 ].each { |v| $db.migrate(v) }
+        @articles = create_records(Article)
+        @comments = create_records(Comment)
+        @articles[0].move_to(ArticleArchive, @options)
+        Article.move_to(
+          ArticleArchive,
+          @options.merge(:conditions => [ 'articles.id = ?', 2 ])
+        )
+      end
+            
+      describe 'move records' do
+
+        it "should move both articles and their associations" do
+          Article.count.should == 3
+          Comment.count.should == 3
+          ArticleArchive.count.should == 2
+          CommentArchive.count.should == 2
+          Article.find_by_id(1).nil?.should == true
+          Comment.find_by_id(2).nil?.should == true
+          ArticleArchive.find_by_id(1).nil?.should == false
+          CommentArchive.find_by_id(2).nil?.should == false
+          comments = ArticleArchive.find_by_id(1).comments
+          comments.length.should == 1
+          comments.first.id.should == 1
+          comments = ArticleArchive.find_by_id(2).comments
+          comments.length.should == 1
+          comments.first.id.should == 2
+        end
+
+        it "should assign moved_at" do
+          ArticleArchive.find_by_id(1).moved_at.utc.to_s.should == Time.now.utc.to_s
+        end
+      end
+
+      describe 'move records back' do
+
+        before(:each) do
+          ArticleArchive.find(1).move_to(Article, @options)
+          ArticleArchive.move_to(
+            Article,
+            @options.merge(:conditions => [ 'article_archives.id = ?', 2 ])
+          )
+        end
+
+        it "should move both articles and their associations" do
+          Article.count.should == 5
+          Comment.count.should == 5
+          ArticleArchive.count.should == 0
+          CommentArchive.count.should == 0
+          Article.find_by_id(1).nil?.should == false
+          Comment.find_by_id(2).nil?.should == false
+          ArticleArchive.find_by_id(1).nil?.should == true
+          CommentArchive.find_by_id(2).nil?.should == true
+          comments = Article.find_by_id(1).comments
+          comments.length.should == 1
+          comments.first.id.should == 1
+          comments = Article.find_by_id(2).comments
+          comments.length.should == 1
+          comments.first.id.should == 2
+        end
+        
+        unless option == :quick
+          it "should copy articles" do
+            article_1 = Article.find(1)
+            article_1.update_attributes(:title => 'edited')
+            article_2 = Article.find(2)
+            article_2.update_attributes(:title => 'edited')
+            article_1.move_to(ArticleArchive, @options.merge(:copy => true))
+            Article.move_to(
+              ArticleArchive,
+              @options.merge(
+                :conditions => [ 'articles.id = ?', 2 ],
+                :copy => true
+              )
+            )
+            ArticleArchive.find(1).title.should == 'edited'
+            ArticleArchive.find(2).title.should == 'edited'
+            Article.count.should == 5
+            Comment.count.should == 5
+            ArticleArchive.count.should == 2
+            CommentArchive.count.should == 2
+          end
+        end
+      end
+    end
+  end
+  
+  describe 'migrate magic column' do
+    
+    before(:each) do
+      [ 1, 0, 1 ].each { |v| $db.migrate(v) }
+      @moved_at = Time.now.utc - 60
+      @articles = create_records(Article, :moved_at => @moved_at)
+      @comments = create_records(Comment)
+      @articles[0].move_to(ArticleArchive, :migrate => true)
+      Article.move_to(
+        ArticleArchive,
+        :conditions => [ 'articles.id = ?', 2 ],
+        :migrate => true
+      )
+    end
+    
+    it "should migrate magic column" do
+      ArticleArchive.find(1).moved_at.to_s.should == @moved_at.to_s
+      ArticleArchive.find(2).moved_at.to_s.should == @moved_at.to_s
+    end
+  end
+  
+  describe :reserve_id do
+    
+    it "should return an id" do
+      Article.reserve_id.class.should == Fixnum
+    end
+  
+    it "should delete the record" do
+      id = Article.reserve_id
+      Article.find_by_id(id).should == nil
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,58 @@
+require 'pp'
+
+$root = File.expand_path('../../', __FILE__)
+require "#{$root}/lib/mover/gems"
+
+Mover::Gems.activate %w(active_wrapper rspec)
+
+require 'active_wrapper'
+require 'fileutils'
+
+require "#{$root}/lib/mover"
+
+require "#{$root}/spec/fixtures/article"
+require "#{$root}/spec/fixtures/article_archive"
+require "#{$root}/spec/fixtures/comment"
+require "#{$root}/spec/fixtures/comment_archive"
+
+Spec::Runner.configure do |config|
+end
+
+$db, $log = ActiveWrapper.setup(
+  :base => File.dirname(__FILE__),
+  :env => 'test'
+)
+$db.establish_connection
+
+def record_match?(original, copy)
+  (original.class.column_names & copy.class.column_names).each do |col|
+    copy.send(col).should == original.send(col)
+  end
+end
+
+def columns(table)
+  connection.columns(table).collect(&:name)
+end
+
+def connection
+  ActiveRecord::Base.connection
+end
+
+def create_records(klass, values={})
+  klass.delete_all
+  keys = values.keys
+  (1..5).collect do |x|
+    klass.column_names.each do |column|
+      next if column == 'id'
+      if column == 'article_id' && !keys.include?(:article_id)
+        values[:article_id] = x
+      elsif !keys.include?(column.intern)
+        values[column.intern] = "#{klass} #{x} #{column}"
+      end
+    end
+    record = klass.new
+    record.id = x
+    record.update_attributes(values)
+    record
+  end
+end

metadata

@@ -0,0 +1,128 @@
+--- !ruby/object:Gem::Specification
+name: mover_postgres
+version: !ruby/object:Gem::Version
+  version: 0.3.6
+  prerelease: 
+platform: ruby
+authors:
+- Winton Welsh
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-07-25 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: active_wrapper
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Move ActiveRecord records across tables like it ain't no thang
+email: james@njtechnologies.co.uk
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- LICENSE
+- README.md
+- Rakefile
+- config/gemsets.yml
+- config/gemspec.yml
+- init.rb
+- lib/mover.rb
+- lib/mover/gems.rb
+- mover.gemspec
+- rails/init.rb
+- spec/Rakefile
+- spec/config/database.yml.example
+- spec/db/migrate/001_create_fixtures.rb
+- spec/fixtures/article.rb
+- spec/fixtures/article_archive.rb
+- spec/fixtures/comment.rb
+- spec/fixtures/comment_archive.rb
+- spec/fixtures/gemsets.yml
+- spec/fixtures/gemspec.yml
+- spec/mover/gems_spec.rb
+- spec/mover_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/jamesharker/mover
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Move ActiveRecord records across tables like it ain't no thang
+test_files:
+- spec/Rakefile
+- spec/config/database.yml.example
+- spec/db/migrate/001_create_fixtures.rb
+- spec/fixtures/article.rb
+- spec/fixtures/article_archive.rb
+- spec/fixtures/comment.rb
+- spec/fixtures/comment_archive.rb
+- spec/fixtures/gemsets.yml
+- spec/fixtures/gemspec.yml
+- spec/mover/gems_spec.rb
+- spec/mover_spec.rb
+- spec/spec_helper.rb
+has_rdoc: