require_relative "../base"

module Rsg
  module Orm
    class ActiveRecordGenerator < Generators::Base
      include Rails::Generators::Database

      SUPPORTED_DBS = %w(sqlite3 mysql postgres)
      class_option :database, type: :string, default: nil, desc: "Configure for selected database (options: #{SUPPORTED_DBS.join("/")})"

      def self.source_root
        Pathname.new(__FILE__).dirname.join("templates").expand_path
      end

      def banner
        say "Configuring active record"
      end

      def enable_active_record
        enable_railtie "active_record"
      end

      def configure_driver
        database = options.fetch(:database) { ask("Which database you'd like to use?", limited_to: SUPPORTED_DBS, default: "sqlite3") }
        template "active_record/#{database}.yml", "config/database.yml"

        name, version = gem_for_database(database)
        append_gem name, version: version
      end

      def write_db_sample
        copy_file "db.rake", "lib/tasks/db.rake"
        copy_file "seeds.rb", "db/seeds.rb"
        copy_file "samples.rb", "db/samples.rb"
      end

      def migrate
        return if no?("Would you like to create the database schema?")

        rake "db:create db:migrate"
      end
    end
  end
end