jonbell-mongrations 0.1.0

data/.gitignore

@@ -0,0 +1,2 @@
+*.swp
+*~

data/MIT-LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2010 [name of plugin creator]
+
+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.markdown

@@ -0,0 +1,56 @@
+Mongrations
+===========
+
+Mongrations aims to be similar to ActiveRecord's data migrations, except
+instead of worrying about schema changes, offering a way to change data when
+necessary. 
+
+Example: You decide after a few weeks of running an application that you want
+to add timestamps, and want to display the created_at time for an application.
+
+You could put a default timestamp into the model (and in this case that might
+be appropriate), but you might also just want to say, "everything needs to get
+a new timestamp that doesn't already have one, let's say when the US beat
+Canada in hockey in the Olympics."
+
+
+Example
+=======
+
+To generate a mongration, just do:
+
+`script/generate mongration whatever_you_want_your_mongration_to_be`
+
+To run it, do
+
+`rake db:mongrate`
+
+Other rake tasks added have been `db:mongrate:redo`, `db:mongrate:up`, `db:mongrate:down`, `db:mongrate:rollback`.
+
+
+Dependencies
+============
+
+You need Mongo and MongoMapper for this to be of any use.
+
+Also, this has only been tested on Rails 3.2.3.
+
+Disclaimer
+==========
+
+*This is not ready for production*
+
+I just adapted this, at 10:30PM, half watching the Olympics. I'm not responsible 
+for any damage to your data, your mongodb, your bongos, your cat, your wife, or 
+your kids as a result of installing this plugin.
+
+Give it a few days. Please report bugs. 
+
+Credit
+======
+Original code is from Rails 2.3.5 and ActiveRecord 2.3.5, now adapted to work
+with MongoMapper.
+
+License
+=======
+Released under the MIT license

data/Rakefile

@@ -0,0 +1,23 @@
+require 'rubygems'
+require 'rake'
+require 'rake/testtask'
+require 'rake/rdoctask'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "jonbell-mongrations"
+    gem.summary = %Q{Data migrating for MongoMapper}
+    gem.description = %Q{Mongrations aims to be similar to ActiveRecord's data migrations, except
+instead of worrying about schema changes, offering a way to change data when
+necessary}
+    gem.email = "jonbell@spamcop.net"
+    gem.homepage = "http://github.com/jonbell/mongrations"
+    gem.authors = ["jonbell"]
+    gem.add_dependency "mongo_mapper", ">= 0.11.0"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end

data/VERSION

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

data/generators/mongration/mongration_generator.rb

@@ -0,0 +1,15 @@
+Rails::Generator::Commands::Base.class_eval do
+  def next_migration_string(padding = 3)
+    Time.now.utc.strftime("%Y%m%d%H%M%S")
+  end
+end
+
+class MongrationGenerator < Rails::Generator::NamedBase
+
+	def manifest
+		record do |m|
+			m.directory File.join('db/mongrations')
+			m.migration_template 'mongration.rb', 'db/mongrations'
+		end
+	end
+end

data/generators/mongration/templates/mongration.rb

@@ -0,0 +1,7 @@
+class <%= class_name.underscore.camelize %> < MongoMapper::Mongration
+  def self.up
+  end
+
+  def self.down
+  end
+end

data/init.rb

@@ -0,0 +1,8 @@
+# Include hook code here
+require "mongo_mapper"
+
+require File.join(File.dirname(__FILE__), "lib", "mongration")
+require File.join(File.dirname(__FILE__), "lib", "mongo_mapper", "mongration")
+require File.join(File.dirname(__FILE__), "lib", "mongo_mapper", "migration_proxy")
+require File.join(File.dirname(__FILE__), "lib", "mongo_mapper", "migrator")
+require File.join(File.dirname(__FILE__), "lib", "mongo_mapper", "schema_migration")

data/install.rb

@@ -0,0 +1 @@
+# Install hook code here

data/lib/mongo_mapper/exceptions.rb

@@ -0,0 +1,30 @@
+module MongoMapper
+  class MongoMapperError < StandardError; end
+
+  class IrreversibleMigration < MongoMapperError#:nodoc:
+  end
+
+  class DuplicateMigrationVersionError < MongoMapperError#:nodoc:
+    def initialize(version)
+      super("Multiple migrations have the version number #{version}")
+    end
+  end
+
+  class DuplicateMigrationNameError < MongoMapperError#:nodoc:
+    def initialize(name)
+      super("Multiple migrations have the name #{name}")
+    end
+  end
+
+  class UnknownMigrationVersionError < MongoMapperError #:nodoc:
+    def initialize(version)
+      super("No migration with version number #{version}")
+    end
+  end
+
+  class IllegalMigrationNameError < MongoMapperError#:nodoc:
+    def initialize(name)
+      super("Illegal name for migration file: #{name}\n\t(only lower case letters, numbers, and '_' allowed)")
+    end
+  end
+end

data/lib/mongo_mapper/migration_proxy.rb

@@ -0,0 +1,22 @@
+# MigrationProxy is used to defer loading of the actual migration classes
+# until they are needed
+module MongoMapper
+  class MigrationProxy
+
+    attr_accessor :name, :version, :filename
+
+    delegate :migrate, :announce, :write, :to=>:migration
+
+    private
+
+      def migration
+        @migration ||= load_migration
+      end
+
+      def load_migration
+        load(filename)
+        name.constantize
+      end
+
+  end
+end

data/lib/mongo_mapper/migrator.rb

@@ -0,0 +1,169 @@
+module MongoMapper
+  class Migrator#:nodoc:
+    class << self
+      def migrate(migrations_path, target_version = nil)
+        case
+          when target_version.nil?              then up(migrations_path, target_version)
+          when current_version > target_version then down(migrations_path, target_version)
+          else                                       up(migrations_path, target_version)
+        end
+      end
+
+      def rollback(migrations_path, steps=1)
+        migrator = self.new(:down, migrations_path)
+        start_index = migrator.migrations.index(migrator.current_migration)
+        
+        return unless start_index
+        
+        finish = migrator.migrations[start_index + steps]
+        down(migrations_path, finish ? finish.version : 0)
+      end
+
+      def up(migrations_path, target_version = nil)
+        self.new(:up, migrations_path, target_version).migrate
+      end
+
+      def down(migrations_path, target_version = nil)
+        self.new(:down, migrations_path, target_version).migrate
+      end
+      
+      def run(direction, migrations_path, target_version)
+        self.new(direction, migrations_path, target_version).run
+      end
+
+      def get_all_versions
+        MongoMapper::SchemaMigration.all.map{|sm| sm.version.to_i}.sort
+      end
+
+      def current_version
+        get_all_versions.max || 0
+      end
+
+      def proper_table_name(name)
+        # Use the Active Record objects own table_name, or pre/suffix from MongoMapper::Base if name is a symbol/string
+        name.table_name rescue "#{MongoMapper::Base.table_name_prefix}#{name}#{MongoMapper::Base.table_name_suffix}"
+      end
+    end
+
+    def initialize(direction, migrations_path, target_version = nil)
+      @direction, @migrations_path, @target_version = direction, migrations_path, target_version      
+    end
+
+    def current_version
+      migrated.last || 0
+    end
+    
+    def current_migration
+      migrations.detect { |m| m.version == current_version }
+    end
+    
+    def run
+      target = migrations.detect { |m| m.version == @target_version }
+      raise UnknownMigrationVersionError.new(@target_version) if target.nil?
+      unless (up? && migrated.include?(target.version.to_i)) || (down? && !migrated.include?(target.version.to_i))
+        target.migrate(@direction)
+        record_version_state_after_migrating(target.version)
+      end
+    end
+
+    def migrate
+      current = migrations.detect { |m| m.version == current_version }
+      target = migrations.detect { |m| m.version == @target_version }
+
+      if target.nil? && !@target_version.nil? && @target_version > 0
+        raise UnknownMigrationVersionError.new(@target_version)
+      end
+      
+      start = up? ? 0 : (migrations.index(current) || 0)
+      finish = migrations.index(target) || migrations.size - 1
+      runnable = migrations[start..finish]
+      
+      # skip the last migration if we're headed down, but not ALL the way down
+      runnable.pop if down? && !target.nil?
+      
+      runnable.each do |migration|
+        # On our way up, we skip migrating the ones we've already migrated
+        next if up? && migrated.include?(migration.version.to_i)
+
+        # On our way down, we skip reverting the ones we've never migrated
+        if down? && !migrated.include?(migration.version.to_i)
+          migration.announce 'never migrated, skipping'; migration.write
+          next
+        end
+
+        migration.migrate(@direction)
+        record_version_state_after_migrating(migration.version)
+      end
+    end
+
+    def migrations
+      @migrations ||= begin
+        files = Dir["#{@migrations_path}/[0-9]*_*.rb"]
+        
+        migrations = files.inject([]) do |klasses, file|
+          version, name = file.scan(/([0-9]+)_([_a-z0-9]*).rb/).first
+          
+          raise IllegalMigrationNameError.new(file) unless version
+          version = version.to_i
+          
+          if klasses.detect { |m| m.version == version }
+            raise DuplicateMigrationVersionError.new(version) 
+          end
+
+          if klasses.detect { |m| m.name == name.camelize }
+            raise DuplicateMigrationNameError.new(name.camelize) 
+          end
+          
+          klasses << returning(MigrationProxy.new) do |migration|
+            migration.name     = name.camelize
+            migration.version  = version
+            migration.filename = file
+          end
+        end
+        
+        migrations = migrations.sort_by(&:version)
+        down? ? migrations.reverse : migrations
+      end
+    end
+
+    def pending_migrations
+      already_migrated = migrated
+      migrations.reject { |m| already_migrated.include?(m.version.to_i) }
+    end
+
+    def migrated
+      @migrated_versions ||= self.class.get_all_versions
+    end
+
+    private
+      def record_version_state_after_migrating(version)
+        @migrated_versions ||= []
+        if down?
+          @migrated_versions.delete(version.to_i)
+          sm = MongoMapper::SchemaMigration.find_by_version(version.to_s)
+          sm && sm.delete
+        else
+          @migrated_versions.push(version.to_i).sort!
+          MongoMapper::SchemaMigration.create(:version => version)
+        end
+      end
+
+      def up?
+        @direction == :up
+      end
+
+      def down?
+        @direction == :down
+      end
+
+      # Wrap the migration in a transaction only if supported by the adapter.
+      def ddl_transaction(&block)
+        if Base.connection.supports_ddl_transactions?
+          Base.transaction { block.call }
+        else
+          block.call
+        end
+      end
+  end
+  
+end

data/lib/mongo_mapper/mongration.rb

@@ -0,0 +1,96 @@
+module MongoMapper
+  class Mongration
+    @@verbose = true
+    cattr_accessor :verbose
+
+    class << self
+      def up_with_benchmarks #:nodoc:
+        migrate(:up)
+      end
+
+      # Execute this migration in the named direction
+      def migrate(direction)
+        return unless respond_to?(direction)
+
+        case direction
+          when :up   then announce "migrating"
+          when :down then announce "reverting"
+        end
+
+        result = nil
+        time = Benchmark.measure { result = send("#{direction}_without_benchmarks") }
+
+        case direction
+          when :up   then announce "migrated (%.4fs)" % time.real; write
+          when :down then announce "reverted (%.4fs)" % time.real; write
+        end
+
+        result
+      end
+
+      # Because the method added may do an alias_method, it can be invoked
+      # recursively. We use @ignore_new_methods as a guard to indicate whether
+      # it is safe for the call to proceed.
+      def singleton_method_added(sym) #:nodoc:
+        return if defined?(@ignore_new_methods) && @ignore_new_methods
+
+        begin
+          @ignore_new_methods = true
+
+          case sym
+            when :up, :down
+              klass = (class << self; self; end)
+              klass.send(:alias_method_chain, sym, "benchmarks")
+          end
+        ensure
+          @ignore_new_methods = false
+        end
+      end
+
+      def write(text="")
+        puts(text) if verbose
+      end
+
+      def announce(message)
+        text = "#{@version} #{name}: #{message}"
+        length = [0, 75 - text.length].max
+        write "== %s %s" % [text, "=" * length]
+      end
+
+      def say(message, subitem=false)
+        write "#{subitem ? "   ->" : "--"} #{message}"
+      end
+
+      def say_with_time(message)
+        say(message)
+        result = nil
+        time = Benchmark.measure { result = yield }
+        say "%.4fs" % time.real, :subitem
+        say("#{result} rows", :subitem) if result.is_a?(Integer)
+        result
+      end
+
+      def suppress_messages
+        save, self.verbose = verbose, false
+        yield
+      ensure
+        self.verbose = save
+      end
+
+      def connection
+        MongoMapper.connection
+      end
+
+      def method_missing(method, *arguments, &block)
+        arg_list = arguments.map(&:inspect) * ', '
+
+        say_with_time "#{method}(#{arg_list})" do
+          unless arguments.empty? || method == :execute
+            arguments[0] = Migrator.proper_table_name(arguments.first)
+          end
+          connection.send(method, *arguments, &block)
+        end
+      end
+    end
+  end
+end

data/lib/mongo_mapper/schema_migration.rb

@@ -0,0 +1,8 @@
+module MongoMapper
+  class SchemaMigration
+    include MongoMapper::Document
+
+    key :version, String
+
+  end
+end

data/lib/mongrations.rb

@@ -0,0 +1,17 @@
+require 'mongo_mapper'
+
+module MongoMapper
+  autoload :MongoMapperError, 'mongo_mapper/exceptions'
+  autoload :IrreversibleMigration, 'mongo_mapper/exceptions'
+  autoload :DuplicateMigrationVersionError, 'mongo_mapper/exceptions'
+  autoload :DuplicateMigrationNameError, 'mongo_mapper/exceptions'
+  autoload :UnknownMigrationVersionError, 'mongo_mapper/exceptions'
+  autoload :IllegalMigrationNameError, 'mongo_mapper/exceptions'
+
+  autoload :MigrationProxy, 'mongo_mapper/migration_proxy'
+  autoload :Migrator, 'mongo_mapper/migrator'
+  autoload :Mongration, 'mongo_mapper/mongration'
+  autoload :SchemaMigration, 'mongo_mapper/schema_migration'
+end
+
+require 'mongrations/railtie' if defined?(Rails)

data/lib/mongrations/railtie.rb

@@ -0,0 +1,10 @@
+require 'rails'
+require File.join(File.dirname(__FILE__), '..', 'mongrations')
+
+module Mongrations
+  class Railtie < Rails::Railtie
+    rake_tasks do
+      load File.join(File.dirname(__FILE__), '..', '..', 'tasks', 'mongo.rake')
+    end
+  end
+end

data/lib/tasks/mongo.rake

@@ -0,0 +1,39 @@
+namespace :mongo do
+  desc "Migrate the database through scripts in db/mongrations. Target specific version with VERSION=x."
+  task :mongrate => :environment do
+    MongoMapper::Migrator.migrate("db/mongrations/", ENV["VERSION"] ? ENV["VERSION"].to_i : nil)
+  end
+  
+  namespace :mongrate do
+    desc  'Rollbacks the database one mongration and re migrate up. If you want to rollback more than one step, define STEP=x. Target specific version with VERSION=x.'
+    task :redo => :environment do
+      if ENV["VERSION"]
+        Rake::Task["mongo:mongrate:down"].invoke
+        Rake::Task["mongo:mongrate:up"].invoke
+      else
+        Rake::Task["mongo:mongo_rollback"].invoke
+        Rake::Task["mongo:mongrate"].invoke
+      end
+    end
+
+    desc 'Runs the "up" for a given mongration VERSION.'
+    task :up => :environment do
+      version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
+      raise "VERSION is required" unless version
+      MongoMapper::Migrator.run(:up, "db/mongrations/", version)
+    end
+
+    desc 'Runs the "down" for a given mongration VERSION.'
+    task :down => :environment do
+      version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
+      raise "VERSION is required" unless version
+      MongoMapper::Migrator.run(:down, "db/mongrations/", version)
+    end
+  end
+  
+  desc 'Rolls the schema back to the previous version. Specify the number of steps with STEP=n'
+  task :mongo_rollback => :environment do
+    step = ENV['STEP'] ? ENV['STEP'].to_i : 1
+    MongoMapper::Migrator.rollback('db/mongrations/', step)
+  end
+end

data/tasks/mongo.rake

@@ -0,0 +1,39 @@
+namespace :mongo do
+  desc "Migrate the database through scripts in db/mongrations. Target specific version with VERSION=x."
+  task :mongrate => :environment do
+    MongoMapper::Migrator.migrate("db/mongrations/", ENV["VERSION"] ? ENV["VERSION"].to_i : nil)
+  end
+  
+  namespace :mongrate do
+    desc  'Rollbacks the database one mongration and re migrate up. If you want to rollback more than one step, define STEP=x. Target specific version with VERSION=x.'
+    task :redo => :environment do
+      if ENV["VERSION"]
+        Rake::Task["mongo:mongrate:down"].invoke
+        Rake::Task["mongo:mongrate:up"].invoke
+      else
+        Rake::Task["mongo:mongo_rollback"].invoke
+        Rake::Task["mongo:mongrate"].invoke
+      end
+    end
+
+    desc 'Runs the "up" for a given mongration VERSION.'
+    task :up => :environment do
+      version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
+      raise "VERSION is required" unless version
+      MongoMapper::Migrator.run(:up, "db/mongrations/", version)
+    end
+
+    desc 'Runs the "down" for a given mongration VERSION.'
+    task :down => :environment do
+      version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
+      raise "VERSION is required" unless version
+      MongoMapper::Migrator.run(:down, "db/mongrations/", version)
+    end
+  end
+  
+  desc 'Rolls the schema back to the previous version. Specify the number of steps with STEP=n'
+  task :mongo_rollback => :environment do
+    step = ENV['STEP'] ? ENV['STEP'].to_i : 1
+    MongoMapper::Migrator.rollback('db/mongrations/', step)
+  end
+end

data/test/mongrations_test.rb

@@ -0,0 +1,8 @@
+require 'test_helper'
+
+class MongrationsTest < ActiveSupport::TestCase
+  # Replace this with your real tests.
+  test "the truth" do
+    assert true
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,3 @@
+require 'rubygems'
+require 'active_support'
+require 'active_support/test_case'

data/uninstall.rb

@@ -0,0 +1 @@
+# Uninstall hook code here

metadata

@@ -0,0 +1,99 @@
+--- !ruby/object:Gem::Specification 
+name: jonbell-mongrations
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- jonbell
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-04-05 00:00:00 -05:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: mongo_mapper
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 11
+        - 0
+        version: 0.11.0
+  type: :runtime
+  version_requirements: *id001
+description: |-
+  Mongrations aims to be similar to ActiveRecord's data migrations, except
+  instead of worrying about schema changes, offering a way to change data when
+  necessary
+email: jonbell@spamcop.net
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.markdown
+files: 
+- .gitignore
+- MIT-LICENSE
+- README.markdown
+- Rakefile
+- VERSION
+- generators/mongration/mongration_generator.rb
+- generators/mongration/templates/mongration.rb
+- init.rb
+- install.rb
+- lib/mongo_mapper/exceptions.rb
+- lib/mongo_mapper/migration_proxy.rb
+- lib/mongo_mapper/migrator.rb
+- lib/mongo_mapper/mongration.rb
+- lib/mongo_mapper/schema_migration.rb
+- lib/mongrations.rb
+- lib/mongrations/railtie.rb
+- lib/tasks/mongo.rake
+- tasks/mongo.rake
+- test/mongrations_test.rb
+- test/test_helper.rb
+- uninstall.rb
+has_rdoc: true
+homepage: http://github.com/jonbell/mongrations
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: Data migrating for MongoMapper
+test_files: 
+- test/mongrations_test.rb
+- test/test_helper.rb