module Sinatra::ActiveRecordExtension

def database=(spec)
  if spec.is_a?(Hash) and spec.symbolize_keys[environment.to_sym]
    ActiveRecord::Base.configurations = spec.stringify_keys
    ActiveRecord::Base.establish_connection(environment.to_sym)
  elsif spec.is_a?(Hash)     
    ActiveRecord::Base.configurations = {
      environment.to_s => spec.stringify_keys
    }
    ActiveRecord::Base.establish_connection(spec.stringify_keys)
  else
    if Gem.loaded_specs["activerecord"].version >= Gem::Version.create('6.0')
      ActiveRecord::Base.configurations ||= ActiveRecord::DatabaseConfigurations.new({}).resolve(spec)
    else
      ActiveRecord::Base.configurations ||= {}
      ActiveRecord::Base.configurations[environment.to_s] = ActiveRecord::ConnectionAdapters::ConnectionSpecification::ConnectionUrlResolver.new(spec).to_hash
    end
    ActiveRecord::Base.establish_connection(spec)
  end
end