require 'pg'

class PostfixadminDb

  def initialize(db_host,
                 db_port,
                 db_opts,
                 db_tty,
                 db_name,
                 db_user,
                 db_pass)

    @db_host = db_host
    @db_port = db_port
    @db_opts = db_opts
    @db_tty  = db_tty
    @db_name = db_name
    @db_user = db_user
    @db_pass = db_pass
  end


  def get_domains_from_db()
    domains = []

    # Just assume PostgreSQL for now.
    begin
      connection = PGconn.connect(@db_host,
                                  @db_port,
                                  @db_opts,
                                  @db_tty,
                                  @db_name,
                                  @db_user,
                                  @db_pass)

      # 'ALL' is a magic domain, and we don't want it.
      sql_query = "SELECT domain FROM domain WHERE domain <> 'ALL';"
      connection.query(sql_query) do |result|
        domains = result.field_values('domain')
      end
      connection.close()
    rescue PGError => e
      # But pretend like we're database-agnostic in case we ever are.
      raise DatabaseError.new(e)
    end

    return domains
  end


  def get_accounts_from_db()
    accounts = []

    # Just assume PostgreSQL for now.
    begin
      connection = PGconn.connect(@db_host,
                                  @db_port,
                                  @db_opts,
                                  @db_tty,
                                  @db_name,
                                  @db_user,
                                  @db_pass)

      # If address = goto, then the alias basically says, "really
      # deliver to that address; it's not an alias."
      sql_query = 'SELECT username FROM mailbox;'
      connection.query(sql_query) do |result|
        accounts = result.field_values('username')
      end
      connection.close()
    rescue PGError => e
      # But pretend like we're database-agnostic in case we ever are.
      raise DatabaseError.new(e)
    end

    return accounts
  end

end