rgravina/okubo

require 'rails/generators'
require 'rails/generators/migration'
require 'rails/generators/active_record'

class OkuboGenerator < Rails::Generators::Base
  include Rails::Generators::Migration

  desc "Creates migration files required by the okubo spaced repetition gem."

  self.source_paths << File.join(File.dirname(__FILE__), 'templates')

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

  def create_migration_files
    create_migration_file_if_not_exist 'create_okubo_tables'
    create_migration_file_if_not_exist 'add_okubo_item_answer_counts'
  end

  private

  def create_migration_file_if_not_exist(file_name)
    unless self.class.migration_exists?(File.dirname(File.expand_path("db/migrate/#{file_name}")), file_name)
      migration_template "#{file_name}.rb", "db/migrate/#{file_name}.rb"
    end
  end
end