module Masshtab
  class DatabaseGenerator < Rails::Generators::Base
    source_root File.expand_path('../templates', __FILE__)

    class_option :application_name, type: :string, default: Rails.application.class.parent_name.underscore, desc: 'Rails application name'

    def mysql2_gem
      gem 'mysql2'
    end

    def database_yml
      template 'database.yml', Rails.root.join('config', 'database.yml')
    end

  end
end