migration_collapser 0.1.0

data/MIT_LICENSE

@@ -0,0 +1,22 @@
+Copyright (c) 2009 Scott Taylor (smtlaissezfaire) <scott@railsnewbie.com>
+
+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.rdoc

@@ -0,0 +1,21 @@
+= Migration Collapser
+
+Collapse your rails migrations
+
+== Installation
+
+  gem install migration_collapser
+
+== Usage
+
+First, load your production database:
+
+  mysql -u root my_dev_db < my_production_db.sql
+
+Dump your production database (without data) in db/migrate/initial_schema.sql
+
+  mysqldump -u root --no-data my_dev_db > db/migrate/initial_schema.sql
+
+Run the migration collapser from RAILS_ROOT:
+
+  collapse_migrations

data/Rakefile

@@ -0,0 +1,6 @@
+
+Dir.glob(File.dirname(__FILE__) + "/rake_tasks/**/**").each do |file|
+  require file
+end
+
+task :default => :spec

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/bin/collapse_migrations

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+
+require File.expand_path(File.dirname(__FILE__) + "/../lib/migration_collapser")
+MigrationCollapser::Runner.run

data/lib/migration_collapser/file_replacer.rb

@@ -0,0 +1,64 @@
+module MigrationCollapser
+  class FileReplacer
+    PATTERN = /(\d+)\_.+\.rb/
+
+    def initialize(revision)
+      self.revision = revision
+    end
+
+    attr_reader :revision
+
+    def revision=(num)
+      @revision = num.to_i
+    end
+
+    def replace
+      delete_files
+      create_migration
+    end
+
+    def delete_files
+      files = Dir.glob("db/migrate/*.rb")
+      file_map = new_migration_map(files)
+
+      check_files(file_map)
+
+      file_map.each do |file, number|
+        if number <= revision
+          FileUtils.rm(file)
+        end
+      end
+    end
+
+    def create_migration
+      cp "templates/initial_schema_migration.rb", "db/migrate/#{revision}_initial_schema.rb"
+    end
+
+  private
+
+    def cp(template, path)
+      FileUtils.cp(File.dirname(__FILE__) + "/#{template}", path)
+    end
+
+    def check_files(file_map)
+      raise "No files in db/migrate" if file_map.empty?
+
+      if file_map.none? { |_, number| number == revision }
+        raise "No migration matching revision #{revision}"
+      end
+    end
+
+    def new_migration_map(files)
+      hash = {}
+
+      files.each do |file|
+        file =~ PATTERN
+        number = $1.to_i
+
+        hash[file] = number
+      end
+
+      hash
+    end
+  end
+end

data/lib/migration_collapser/rails_loader.rb

@@ -0,0 +1,19 @@
+module MigrationCollapser
+  module RailsLoader
+    class << self
+      def load
+        require expand_path(Dir.getwd, "config", "environment")
+      end
+
+    private
+
+      def expand_path(*dirs)
+        File.expand_path(join(*dirs))
+      end
+
+      def join(*dirs)
+        File.join(*dirs)
+      end
+    end
+  end
+end

data/lib/migration_collapser/revision_finder.rb

@@ -0,0 +1,13 @@
+require 'active_record'
+
+module MigrationCollapser
+  class RevisionFinder
+    class SchemaInfo < ActiveRecord::Base
+      set_table_name "schema_migrations"
+    end
+
+    def self.find
+      SchemaInfo.find(:first, :order => "version DESC").version
+    end
+  end
+end

data/lib/migration_collapser/runner.rb

@@ -0,0 +1,9 @@
+module MigrationCollapser
+  class Runner
+    def self.run
+      RailsLoader.load
+      revision_num = RevisionFinder.find
+      FileReplacer.new(revision_num).replace
+    end
+  end
+end

data/lib/migration_collapser/templates/initial_schema_migration.rb

@@ -0,0 +1,70 @@
+class InitialSchema < ActiveRecord::Migration
+  class << self
+    def up
+      class << connection = ActiveRecord::Base.connection
+        def username
+          @config[:username]
+        end
+
+        def host
+          @config[:host]
+        end
+      end
+
+      file = File.read(File.dirname(__FILE__) + "/initial_schema.sql")
+
+      file.split(";").each do |sql_line|
+        execute(sql_line, true)
+      end
+    end
+
+    def down
+      drop_views
+      drop_tables
+    end
+
+  private
+
+    def execute(sql_string, verbose = false)
+      if verbose
+        puts "* Executing: #{sql_string}"
+      end
+
+      ActiveRecord::Base.connection.execute(sql_string)
+    end
+
+    def drop_tables
+      tables.each do |table|
+        drop_table(table)
+      end
+    end
+
+    def drop_views
+      views.each do |view|
+        drop_view(view)
+      end
+    end
+
+    def drop_table(table_name)
+      puts "* Dropping table: #{table_name}"
+      execute "DROP TABLE `#{table_name}`"
+    end
+
+    def drop_view(table_name)
+      puts "* Dropping view: #{table_name}"
+      execute "DROP VIEW `#{table_name}`"
+    end
+
+    def tables
+      @tables ||= all_tables - ["schema_migrations"]
+    end
+
+    def all_tables
+      execute("SHOW FULL TABLES WHERE table_type = 'VIEW'").all_hashes.map { |x| x.values.last }
+    end
+
+    def views
+      @views ||= execute("SHOW FULL TABLES WHERE table_type = 'VIEW'").all_hashes.map { |x| x.values.last }
+    end
+  end
+end

data/lib/migration_collapser/version.rb

@@ -0,0 +1,9 @@
+module MigrationCollapser
+  module Version
+    MAJOR = 0
+    MINOR = 1
+    TINY  = 0
+
+    STRING = "#{MAJOR}.#{MINOR}.#{TINY}"
+  end
+end

data/lib/migration_collapser.rb

@@ -0,0 +1,11 @@
+require "using"
+
+module MigrationCollapser
+  extend Using
+
+  using :Version
+  using :RevisionFinder
+  using :RailsLoader
+  using :FileReplacer
+  using :Runner
+end

data/rake_tasks/flog.rb

@@ -0,0 +1,10 @@
+desc "Feel the pain of my code, and submit a refactoring patch"
+task :flog do
+  puts %x(find lib | grep ".rb$" | xargs flog)
+end
+
+task :flog_to_disk => :create_doc_directory do
+  puts "Flogging..."
+  %x(find lib | grep ".rb$" | xargs flog > doc/flog.txt)
+  puts "Done Flogging...\n"
+end

data/rake_tasks/gem.rb

@@ -0,0 +1,27 @@
+require File.expand_path(File.dirname(__FILE__) + "/../lib/migration_collapser")
+
+begin
+  require 'jeweler'
+
+  def set_version_for_jewler
+    version = MigrationCollapser::Version::STRING
+
+    File.open(File.expand_path(File.dirname(__FILE__) + "/../VERSION"), "w") do |f|
+      f << version
+    end
+  end
+
+  Jeweler::Tasks.new do |gemspec|
+    set_version_for_jewler
+
+    gemspec.name           = "migration_collapser"
+    gemspec.summary        = "Collapse rails migrations"
+    gemspec.description    = "Collapse migrations already run in production"
+    gemspec.email          = "scott@railsnewbie.com"
+    gemspec.homepage       = "http://github.com/smtlaissezfaire/migration_collapser"
+    gemspec.authors        = ["Scott Taylor"]
+    gemspec.add_dependency "using"
+  end
+rescue LoadError
+  puts "Jeweler not available. Install it with: sudo gem install technicalpickles-jeweler -s http://gems.github.com"
+end

data/rake_tasks/rdoc.rb

@@ -0,0 +1,16 @@
+require 'rake'
+require 'hanna/rdoctask'
+
+DOC_DIRECTORY = File.dirname(__FILE__) + "/../doc"
+
+Rake::RDocTask.new do |rdoc|
+  rdoc.rdoc_dir = DOC_DIRECTORY
+  rdoc.title    = 'migration_collapser'
+  rdoc.options << '--line-numbers' << '--inline-source'
+
+  rdoc.options << '--webcvs=http://github.com/smtlaissezfaire/migration_collapser'
+
+  ["README.rdoc", "GPL_LICENSE", "MIT_LICENSE", "lib/**/*.rb"].each do |file|
+    rdoc.rdoc_files.include(file)
+  end
+end

data/rake_tasks/rspec.rb

@@ -0,0 +1,20 @@
+require 'spec/rake/spectask'
+require 'spec/rake/verify_rcov'
+
+desc 'Run the specs'
+Spec::Rake::SpecTask.new do |t|
+  t.warning = false
+  t.spec_opts = ["--color"]
+end
+
+desc "Create the html specdoc"
+Spec::Rake::SpecTask.new(:specdoc) do |t|
+  t.spec_opts = ["--format", "html:doc/specdoc.html"]
+end
+
+desc "Run all examples with RCov"
+Spec::Rake::SpecTask.new(:rcov) do |t|
+  t.rcov = true
+  t.rcov_opts = ['--exclude', 'spec', "--exclude", "gems"]
+  t.rcov_dir = "doc/rcov"
+end

data/rake_tasks/sloc.rb

@@ -0,0 +1,16 @@
+def sloc
+  `sloccount #{File.dirname(__FILE__)}/../lib #{File.dirname(__FILE__)}/../ext`
+end
+
+desc "Output sloccount report.  You'll need sloccount installed."
+task :sloc do
+  puts "Counting lines of code"
+  puts sloc
+end
+
+desc "Write sloccount report"
+task :output_sloc => :create_doc_directory do
+  File.open(File.dirname(__FILE__) + "/doc/lines_of_code.txt", "w") do |f|
+    f << sloc
+  end
+end

data/rake_tasks/tags.rb

@@ -0,0 +1,23 @@
+# Build the TAGS file for Emacs
+# Taken with slight modifications from
+# http://blog.lathi.net/articles/2007/11/07/navigating-your-projects-in-emacs
+#
+# Thanks Jim Weirich
+
+module Emacs
+  module Tags
+    def self.ruby_files
+      @ruby_files ||= FileList['**/*.rb'].exclude("pkg")
+    end
+  end
+end
+
+namespace :tags do
+  task :emacs do
+    puts "Making Emacs TAGS file"
+    sh "ctags -e #{Emacs::Tags.ruby_files}", :verbose => false
+  end
+end
+
+desc "Build the emacs tags file"
+task :tags => ["tags:emacs"]

data/spec/migration_collapser/file_replacer_spec.rb

@@ -0,0 +1,130 @@
+require "spec_helper"
+require "fakefs/spec_helpers"
+
+module MigrationCollapser
+  describe FileReplacer do
+    describe "replace" do
+      before do
+        @replacer = FileReplacer.new("revision")
+        @replacer.stub!(:delete_files)
+        @replacer.stub!(:create_migration)
+      end
+
+      it "should call delete_files" do
+        @replacer.should_receive(:delete_files)
+        @replacer.replace
+      end
+
+      it "should call create_migration" do
+        @replacer.should_receive(:create_migration)
+        @replacer.replace
+      end
+    end
+
+    def setup_rails
+      Dir.mkdir "/rails_root"
+      Dir.mkdir "/rails_root/db"
+      Dir.mkdir "/rails_root/db/migrate"
+
+      Dir.chdir "/rails_root"
+    end
+
+    describe "deleting files" do
+      include FakeFS::SpecHelpers
+
+      before do
+        setup_rails
+
+        @revision = 123456
+        @replacer = FileReplacer.new(@revision)
+      end
+
+      it "should raise an error if it has no files" do
+        lambda {
+          @replacer.delete_files
+        }.should raise_error("No files in db/migrate")
+      end
+
+      it "should delete the file if it it exists" do
+        FileUtils.touch "/rails_root/db/migrate/123456_my_migration.rb"
+        @replacer.delete_files
+        File.exists?("/rails_root/db/migrate/123456_my_migration.rb").should be_false
+      end
+
+      it "should delete the migration with any name" do
+        FileUtils.touch "/rails_root/db/migrate/123456_foo_bar.rb"
+        @replacer.delete_files
+        File.exists?("/rails_root/db/migrate/123456_foo_bar.rb").should be_false
+      end
+
+      it "should delete the migration with the correct number" do
+        @replacer.revision = 123
+        FileUtils.touch "/rails_root/db/migrate/123_foo_bar.rb"
+        @replacer.delete_files
+        File.exists?("/rails_root/db/migrate/123_foo_bar.rb").should be_false
+      end
+
+      it "should delete all the migrations before (and including) the given number" do
+        @replacer.revision = 002
+
+        FileUtils.touch "/rails_root/db/migrate/001_foo_bar.rb"
+        FileUtils.touch "/rails_root/db/migrate/002_foo_bar.rb"
+
+        @replacer.delete_files
+
+        File.exists?("/rails_root/db/migrate/001_foo_bar.rb").should be_false
+        File.exists?("/rails_root/db/migrate/002_foo_bar.rb").should be_false
+      end
+
+      it "should not delete migrations before the number" do
+        @replacer.revision = 2
+
+        FileUtils.touch "/rails_root/db/migrate/001_foo_bar.rb"
+        FileUtils.touch "/rails_root/db/migrate/002_foo_bar.rb"
+        FileUtils.touch "/rails_root/db/migrate/003_foo_bar.rb"
+
+        @replacer.delete_files
+
+        File.exists?("/rails_root/db/migrate/001_foo_bar.rb").should be_false
+        File.exists?("/rails_root/db/migrate/002_foo_bar.rb").should be_false
+        File.exists?("/rails_root/db/migrate/003_foo_bar.rb").should be_true
+      end
+
+      it "should raise an error if the file does not exist" do
+        FileUtils.touch "/rails_root/db/migrate/0001_my_migration.rb"
+
+        @replacer.revision = 123
+
+        lambda {
+          @replacer.delete_files
+        }.should raise_error
+      end
+    end
+
+    describe "replacing the migration" do
+      before do
+        @replacer = FileReplacer.new(1234)
+        @file = File.expand_path(File.dirname(__FILE__) + "/../../lib/migration_collapser/templates/initial_schema_migration.rb")
+      end
+
+      it "should copy the contents of the default migration" do
+        FileUtils.should_receive(:cp).with(@file, "db/migrate/999_initial_schema.rb")
+        @replacer.revision = 999
+
+        @replacer.create_migration
+      end
+    end
+
+    describe "revision" do
+      it "should have it as an int when assigned in the constructor" do
+        FileReplacer.new("1234").revision.should == 1234
+      end
+
+      it "should cast it to an int when provided through the accessor" do
+        replacer = FileReplacer.new("1234")
+        replacer.revision = "1111"
+        replacer.revision.should == 1111
+      end
+    end
+  end
+end