class Rails::DBConsole

def self.start(*args)
  new(*args).start
end

def adapter_class
  ActiveRecord::ConnectionAdapters.resolve(db_config.adapter)
rescue LoadError
  ActiveRecord::ConnectionAdapters::AbstractAdapter
end

def configurations # :doc:
  require APP_PATH
  ActiveRecord::Base.configurations = Rails.application.config.database_configuration
  ActiveRecord::Base.configurations
end

def database
  @options[:database]
end

def db_config
  @db_config ||= begin
    # If the user provided a database, use that. Otherwise find
    # the first config in the database.yml
    config = if database
      @db_config = configurations.configs_for(env_name: environment, name: database, include_hidden: true)
    else
      @db_config = configurations.find_db_config(environment)
    end
    unless config
      missing_db = database ? "'#{database}' database is not" : "No databases are"
      raise ActiveRecord::AdapterNotSpecified,
        "#{missing_db} configured for '#{environment}'. Available configuration: #{configurations.inspect}"
    end
    config.validate!
    config
  end
end

def environment
  @options[:environment]
end

def initialize(options = {})
  @options = options
  @options[:environment] ||= Rails::Command.environment
end

def start
  adapter_class.dbconsole(db_config, @options)
rescue NotImplementedError, ActiveRecord::AdapterNotFound, LoadError => error
  abort error.message
end