sensu-plugins/sensu-plugins-postgres

# frozen_string_literal: true

module Pgpass
  def read_pgpass(pg_pass_file)
    File.readlines(pg_pass_file).each do |line|
      return line.strip.split(':') unless line.start_with?('#')

      next
    end
  end

  def pgpass
    if File.file?(config[:pgpass])
      pgpass = Hash[%i[hostname port database user password].zip(read_pgpass(config[:pgpass]))]
      pgpass[:database] = nil if pgpass[:database] == '*'
      pgpass.each do |k, v|
        config[k] ||= v
      end
    else
      config[:hostname] ||= ENV['PGHOST']     || 'localhost'
      config[:port]     ||= ENV['PGPORT']     || 5432
      config[:database] ||= ENV['PGDATABASE'] || 'postgres'
      config[:user]     ||= ENV['PGUSER']     || 'postgres'
      config[:password] ||= ENV['PGPASSWORD']
    end
  end
end