Fixes the migration generators

This commit is contained in:
Chris Nicola 2013-02-04 16:31:05 -08:00
parent 1017159fa0
commit 121cdff544
1 changed files with 5 additions and 2 deletions

View File

@ -1,11 +1,9 @@
require 'rails/generators'
require 'rails/generators/migration'
require 'rails/generators/active_record/migration'
module PaperTrail
class InstallGenerator < Rails::Generators::Base
include Rails::Generators::Migration
extend ActiveRecord::Generators::Migration
source_root File.expand_path('../templates', __FILE__)
class_option :with_changes, :type => :boolean, :default => false, :desc => "Store changeset (diff) with each version"
@ -16,5 +14,10 @@ module PaperTrail
migration_template 'create_versions.rb', 'db/migrate/create_versions.rb'
migration_template 'add_object_changes_column_to_versions.rb', 'db/migrate/add_object_changes_column_to_versions.rb' if options.with_changes?
end
def self.next_migration_number(dirname)
next_migration_number = current_migration_number(dirname) + 1
ActiveRecord::Migration.next_migration_number(next_migration_number)
end
end
end