migrer 0.0.2 → 0.1.0

data/README.md

@@ -51,7 +51,7 @@ Create a data migration:
 
     $ bundle exec rails generate data_migration MyFirstDataMigration "optional description"
 
-This will create the file:  lib/tasks/data_migrations/<timestamp>_my_first_data_migration.rb
+This will create the file:  db/data_migrate/<timestamp>_my_first_data_migration.rb
 
 Open this file and replace "TODO" with your data migration code!
 
@@ -89,6 +89,10 @@ ActiveRecord migrations):**
 
     bundle exec rake data:unmark VERSION=<version>
 
+**View all unprocessed data migrations by filename:**
+
+    bundle exec rake data:pending
+
 ## Contributing
 
 1. Fork it

data/{lib/data_migration.rb → app/models/migrer/data_migration_version.rb}

@@ -1,13 +1,15 @@
-class DataMigration
-  def self.all
-    filenames = Dir.entries("#{Rails.root}/lib/tasks/data_migrations").select { |f| /^\d+.*\.rb$/ === f }
+class Migrer::DataMigrationVersion < ActiveRecord::Base
+
+  attr_accessible :version
+
+  def self.all_from_files
+    filenames = Dir.entries("#{Rails.root}/db/data_migrate").select { |f| /^\d+.*\.rb$/ === f }
     data_migrations = {}
 
     filenames.each do |f|
       match_data = /^(?<version>\d+)_(?<name>.+)\.rb$/.match(f)
 
-      record = ActiveRecord::Base.connection.execute(
-          "SELECT * FROM data_migration_versions WHERE version = '#{match_data[:version]}'").first
+      record = Migrer::DataMigrationVersion.find_by_version(match_data[:version])
 
       data_migrations.merge!(
           match_data[:version] => {

data/lib/generators/data_migration/USAGE

@@ -8,4 +8,4 @@ Example:
     rails generate data_migration DataMigrationName "Optional description"
 
     This will create:
-        lib/tasks/data_migrations/<timestamp>_data_migration_name.rb
+        db/data_migrate/<timestamp>_data_migration_name.rb

data/lib/generators/data_migration/data_migration_generator.rb

@@ -6,11 +6,11 @@ class DataMigrationGenerator < Rails::Generators::NamedBase
 
   def generate_data_migration
     template "data_migration.rb",
-             "lib/tasks/data_migrations/#{file_name}"
+             "db/data_migrate/#{file_name}"
   end
 
   private
   def file_name
-    "#{ActiveRecord::Generators::Base.next_migration_number('lib/tasks/data_migrations')}_#{name.underscore}.rb"
+    "#{ActiveRecord::Generators::Base.next_migration_number('db/data_migrate')}_#{name.underscore}.rb"
   end
 end

data/lib/generators/data_migration/templates/data_migration.rb

@@ -1,8 +1,6 @@
-class <%= name %> < DataMigration
-
-<%= "# #{description}" %>
-
-    def self.run
-      #TODO: data_migration code
-    end
-end
+class <%= name %>
+<%= "\n  # #{description}\n" if description.present? %>
+  def self.run
+    #TODO: data_migration code
+  end
+end

data/lib/migrer/ansi_colors.rb

@@ -0,0 +1,13 @@
+class String
+  def migrer_green
+    "\033[32m#{self}\033[0m"
+  end
+
+  def migrer_red
+    "\033[31m#{self}\033[0m"
+  end
+
+  def migrer_yellow
+    "\033[33m#{self}\033[0m"
+  end
+end

data/lib/migrer/version.rb

@@ -1,3 +1,3 @@
 module Migrer
-  VERSION = "0.0.2"
+  VERSION = "0.1.0"
 end

data/lib/migrer.rb

@@ -1,6 +1,9 @@
 require "migrer/version"
 
 module Migrer
+  def self.table_name_prefix
+    ''
+  end
 end
 
 # Require our engine

data/lib/tasks/migrer.rake

@@ -1,19 +1,19 @@
-require 'data_migration'
+require 'migrer/ansi_colors'
 
 namespace :data do
-  desc "Data migration tasks"
 
+  desc 'Run all unprocessed data migrations or a single data migration if VERSION is specified.'
   task migrate: :environment do
-    data_migrations = DataMigration.all
+    data_migrations = Migrer::DataMigrationVersion.all_from_files
 
     if (version = ENV['VERSION'])
       data_migration = data_migrations[version]
 
       if data_migration.present?
         if data_migration[:processed]
-          puts "Data migration already processed. Do you want to run it anyway? (responses other than 'yes' will exit)"
+          puts "Data migration already processed. Do you want to run it anyway? (responses other than 'yes' will exit)".migrer_yellow
         else
-          puts "Starting data migration #{data_migration[:class_name]}. Do you wish to continue? (responses other than 'yes' will exit)"
+          puts "Starting data migration #{data_migration[:class_name]}. Do you wish to continue? (responses other than 'yes' will exit)".migrer_yellow
         end
 
         prompt = $stdin.gets.chomp
@@ -22,134 +22,142 @@ namespace :data do
           puts "#{data_migration[:class_name]}: migrating"
           t_start = Time.now
 
-          require "#{Rails.root}/lib/tasks/data_migrations/#{data_migration[:basefilename]}"
+          require "#{Rails.root}/db/data_migrate/#{data_migration[:basefilename]}"
           eval(data_migration[:class_name]).run
 
           t_end = Time.now
 
           unless data_migration[:processed]
-            ActiveRecord::Base.connection.execute(
-                "INSERT INTO data_migration_versions
-                 VALUES (NULL, '#{version}', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)" )
+            Migrer::DataMigrationVersion.create(version: version)
           end
 
-          puts "#{data_migration[:class_name]}: migrated (#{t_end - t_start}s)"
+          puts "#{data_migration[:class_name]}:" + " migrated (#{t_end - t_start}s)".migrer_green
         end
       else
-        puts "No data migration found matching version: #{version}"
+        puts "No data migration found matching version: #{version}".migrer_red
       end
     else
       data_migrations.each do |k, v|
         unless v[:processed]
-          puts "Starting data migration #{v[:class_name]}. Do you wish to continue? (responses other than 'yes' will exit)"
+          puts "Starting data migration #{v[:class_name]}. Do you wish to continue? (responses other than 'yes' will exit)".migrer_yellow
           prompt = $stdin.gets.chomp
           if prompt == "yes"
             puts "#{v[:class_name]}: migrating"
             t_start = Time.now
 
-            require "#{Rails.root}/lib/tasks/data_migrations/#{v[:basefilename]}"
+            require "#{Rails.root}/db/data_migrate/#{v[:basefilename]}"
             eval(v[:class_name]).run
 
             t_end = Time.now
 
-            ActiveRecord::Base.connection.execute(
-                "INSERT INTO data_migration_versions
-                 VALUES (NULL, '#{k}', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)" )
+            Migrer::DataMigrationVersion.create(version: k)
 
-            puts "#{v[:class_name]}: migrated (#{t_end - t_start}s)"
+            puts "#{v[:class_name]}:" + " migrated (#{t_end - t_start}s)".migrer_green
           end
         end
       end
     end
   end
 
+  desc 'Mark a single migration as already processed (requires VERSION).'
   task mark: :environment do
-    data_migrations = DataMigration.all
+    data_migrations = Migrer::DataMigrationVersion.all_from_files
 
     if (version = ENV['VERSION'])
       data_migration = data_migrations[version]
 
       if data_migration.present?
         if data_migration[:processed]
-          puts "Data migration already processed."
+          puts "Data migration already processed.".migrer_yellow
         else
-          puts "Data migration #{data_migration[:class_name]} will be marked as processed. Continue? (responses other than 'yes' will exit)"
+          puts "Data migration #{data_migration[:class_name]} will be marked as processed. Continue? (responses other than 'yes' will exit)".migrer_yellow
 
           prompt = $stdin.gets.chomp
 
           if prompt == "yes"
-            ActiveRecord::Base.connection.execute(
-                "INSERT INTO data_migration_versions
-                 VALUES (NULL, '#{version}', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)" )
-
-            puts "#{data_migration[:class_name]}: marked as migrated"
+            Migrer::DataMigrationVersion.create(version: version)
+            puts "#{data_migration[:class_name]}:" + " marked as migrated".migrer_green
           end
         end
       else
-        puts "No data migration found matching version: #{version}"
+        puts "No data migration found matching version: #{version}".migrer_red
       end
     else
-      puts "VERSION must be supplied."
+      puts "VERSION must be supplied.".migrer_red
     end
   end
 
+  desc 'Mark all data migrations as already processed.'
   task mark_all: :environment do
-    unprocessed_data_migrations = DataMigration.all.select { |k, v| !v[:processed] }
+    unprocessed_data_migrations = Migrer::DataMigrationVersion.all_from_files.select { |k, v| !v[:processed] }
 
-    puts "This will mark all data migrations as already processed. Continue? (responses other than 'yes' will exit)"
+    puts "This will mark all data migrations as already processed. Continue? (responses other than 'yes' will exit)".migrer_yellow
 
     prompt = $stdin.gets.chomp
     if prompt == "yes"
       unprocessed_data_migrations.each do |k, v|
-        ActiveRecord::Base.connection.execute(
-            "INSERT INTO data_migration_versions
-             VALUES (NULL, '#{k}', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)" )
-
-        puts "#{v[:class_name]}: marked as migrated"
+        Migrer::DataMigrationVersion.create(version: k)
+        puts "#{v[:class_name]}:" + " marked as migrated".migrer_green
       end
     end
   end
 
+  desc 'Revert a data migration to an unprocessed state (requires VERSION).'
   task unmark: :environment do
-    data_migrations = DataMigration.all
+    data_migrations = Migrer::DataMigrationVersion.all_from_files
 
     if (version = ENV['VERSION'])
       data_migration = data_migrations[version]
 
       if data_migration.present?
         if !data_migration[:processed]
-          puts "Data migration not yet processed."
+          puts "Data migration not yet processed.".migrer_yellow
         else
-          puts "Data migration #{data_migration[:class_name]} will be unmarked as processed. Continue? (responses other than 'yes' will exit)"
+          puts "Data migration #{data_migration[:class_name]} will be unmarked as processed. Continue? (responses other than 'yes' will exit)".migrer_yellow
 
           prompt = $stdin.gets.chomp
 
           if prompt == "yes"
-            ActiveRecord::Base.connection.execute(
-                "DELETE FROM data_migration_versions
-                 WHERE version = '#{version}'" )
-
-            puts "#{data_migration[:class_name]}: unmarked as migrated"
+            Migrer::DataMigrationVersion.find_by_version(version).destroy
+            puts "#{data_migration[:class_name]}:" + " unmarked as migrated".migrer_green
           end
         end
       else
-        puts "No data migration found matching version: #{version}"
+        puts "No data migration found matching version: #{version}".migrer_red
       end
     else
-      puts "VERSION must be supplied."
+      puts "VERSION must be supplied.".migrer_red
     end
   end
 
+  desc 'Revert all data migrations to an unprocessed state.'
   task unmark_all: :environment do
-    puts "All data migrations will be unmarked as processed. Continue? (responses other than 'yes' will exit)"
+    puts "All data migrations will be unmarked as processed. Continue? (responses other than 'yes' will exit)".migrer_yellow
 
     prompt = $stdin.gets.chomp
 
     if prompt == "yes"
-      ActiveRecord::Base.connection.execute(
-          "DELETE FROM data_migration_versions" )
+      Migrer::DataMigrationVersion.destroy_all
+      puts "Data migration records cleared".migrer_green
+    end
+  end
 
-      puts "Data migration records cleared"
+  desc 'View all unprocessed data migrations by filename.'
+  task pending: :environment do
+    data_migrations = Migrer::DataMigrationVersion.all_from_files
+    pending = data_migrations.reject {|k, v| v[:processed]}
+
+    if data_migrations.present?
+      if pending.present?
+        puts "The following migrations have not yet been processed:".migrer_yellow
+        pending.each do |k, v|
+          puts v[:filename]
+        end
+      else
+        puts "All existing data migrations have been processed.".migrer_yellow
+      end
+    else
+      puts "No data migrations found.".migrer_yellow
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: migrer
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-01-13 00:00:00.000000000 Z
+date: 2013-01-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -40,12 +40,13 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- app/models/migrer/data_migration_version.rb
 - db/migrate/20121229002105_create_migrer_table.rb
-- lib/data_migration.rb
 - lib/generators/data_migration/USAGE
 - lib/generators/data_migration/data_migration_generator.rb
 - lib/generators/data_migration/templates/data_migration.rb
 - lib/migrer.rb
+- lib/migrer/ansi_colors.rb
 - lib/migrer/engine.rb
 - lib/migrer/version.rb
 - lib/tasks/migrer.rake