module Puppet::Rails

# File lib/puppet/rails.rb, line 10
def self.connect
  # This global init does not work for testing, because we remove
  # the state dir on every test.
  return if ActiveRecord::Base.connected?

  Puppet.settings.use(:main, :rails, :master)

  ActiveRecord::Base.logger = Logger.new(Puppet[:railslog])
  begin
    loglevel = Logger.const_get(Puppet[:rails_loglevel].upcase)
    ActiveRecord::Base.logger.level = loglevel
  rescue => detail
    Puppet.warning "'#{Puppet[:rails_loglevel]}' is not a valid Rails log level; using debug"
    ActiveRecord::Base.logger.level = Logger::DEBUG
  end

  # As of ActiveRecord 2.2 allow_concurrency has been deprecated and no longer has any effect.
  ActiveRecord::Base.allow_concurrency = true if Puppet::Util.activerecord_version < 2.2

  ActiveRecord::Base.verify_active_connections!

  begin
    args = database_arguments
    Puppet.info "Connecting to #{args[:adapter]} database: #{args[:database]}"
    ActiveRecord::Base.establish_connection(args)
  rescue => detail
    message = "Could not connect to database: #{detail}"
    Puppet.log_exception(detail, message)
    raise Puppet::Error, message
  end
end

# File lib/puppet/rails.rb, line 43
def self.database_arguments
  adapter = Puppet[:dbadapter]

  args = {:adapter => adapter, :log_level => Puppet[:rails_loglevel]}

  case adapter
  when "sqlite3"
    args[:database] = Puppet[:dblocation]
  when "mysql", "mysql2", "postgresql"
    args[:host]     = Puppet[:dbserver] unless Puppet[:dbserver].to_s.empty?
    args[:port]     = Puppet[:dbport] unless Puppet[:dbport].to_s.empty?
    args[:username] = Puppet[:dbuser] unless Puppet[:dbuser].to_s.empty?
    args[:password] = Puppet[:dbpassword] unless Puppet[:dbpassword].to_s.empty?
    args[:pool]     = Puppet[:dbconnections].to_i unless Puppet[:dbconnections].to_i <= 0
    args[:database] = Puppet[:dbname]
    args[:reconnect]= true

    socket          = Puppet[:dbsocket]
    args[:socket]   = socket unless socket.to_s.empty?
  when "oracle_enhanced"
    args[:database] = Puppet[:dbname] unless Puppet[:dbname].to_s.empty?
    args[:username] = Puppet[:dbuser] unless Puppet[:dbuser].to_s.empty?
    args[:password] = Puppet[:dbpassword] unless Puppet[:dbpassword].to_s.empty?
    args[:pool]     = Puppet[:dbconnections].to_i unless Puppet[:dbconnections].to_i <= 0
  else
    raise ArgumentError, "Invalid db adapter #{adapter}"
  end
  args
end

# File lib/puppet/rails.rb, line 75
def self.init
  raise Puppet::DevError, "No activerecord, cannot init Puppet::Rails" unless Puppet.features.rails?

  connect

  unless ActiveRecord::Base.connection.tables.include?("resources")
    require 'puppet/rails/database/schema'
    Puppet::Rails::Schema.init
  end

  migrate if Puppet[:dbmigrate]
end

# File lib/puppet/rails.rb, line 89
def self.migrate
  dbdir = nil
  $LOAD_PATH.each { |d|
    tmp = File.join(d, "puppet/rails/database")
    if FileTest.directory?(tmp)
      dbdir = tmp
      break
    end
  }

  raise Puppet::Error, "Could not find Puppet::Rails database dir" unless dbdir

  raise Puppet::Error, "Database has problems, can't migrate." unless ActiveRecord::Base.connection.tables.include?("resources")

  Puppet.notice "Migrating"

  begin
    ActiveRecord::Migrator.migrate(dbdir)
  rescue => detail
    message = "Could not migrate database: #{detail}"
    Puppet.log_exception(detail, message)
    raise Puppet::Error, "Could not migrate database: #{detail}"
  end
end

# File lib/puppet/rails.rb, line 115
def self.teardown
  raise Puppet::DevError, "No activerecord, cannot init Puppet::Rails" unless Puppet.features.rails?

  Puppet.settings.use(:master, :rails)

  begin
    ActiveRecord::Base.establish_connection(database_arguments)
  rescue => detail
    Puppet.log_exception(detail)
    raise Puppet::Error, "Could not connect to database: #{detail}"
  end

  ActiveRecord::Base.connection.tables.each do |t|
    ActiveRecord::Base.connection.drop_table t
  end
end