# frozen_string_literal: true

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

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

  class_option :migrate_data, type: :boolean, default: false

  desc 'Creates migration files required by the active_recall spaced repetition gem.'

  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_active_recall_tables'
    create_migration_file_if_not_exist 'add_active_recall_item_answer_counts'
    if options['migrate_data']
      create_migration_file_if_not_exist 'migrate_okubo_to_active_recall'
    end
  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