class PaperTrail::MigrationGenerator

def self.next_migration_number(dirname)
  ::ActiveRecord::Generators::Base.next_migration_number(dirname)
end

def add_paper_trail_migration(template, extra_options = {})
  migration_dir = File.expand_path("db/migrate")
  if self.class.migration_exists?(migration_dir, template)
    ::Kernel.warn "Migration already exists: #{template}"
  else
    migration_template(
      "#{template}.rb.erb",
      "db/migrate/#{template}.rb",
      { migration_version: migration_version }.merge(extra_options)
    )
  end
end

def migration_version
  format(
    "[%d.%d]",
    ActiveRecord::VERSION::MAJOR,
    ActiveRecord::VERSION::MINOR
  )
end