Geografic-Information-System/db/drivers/postgres/listdb.c

/*!
  \file db/driver/postgres/listdb.c
  
  \brief DBMI - Low Level PostgreSQL database driver - list databases
  
  This program is free software under the GNU General Public License
  (>=v2). Read the file COPYING that comes with GRASS for details.
  
  \author Radim Blazek
  \author Updated for GRASS 7 by Martin Landa <landa.martin gmail.com>
 */

#include <grass/dbmi.h>
#include <grass/glocale.h>
#include "globals.h"
#include "proto.h"

int db__driver_list_databases(dbString * dbpath, int npaths,
			      dbHandle ** dblist, int *dbcount)
{
    int i;
    const char *user, *passwd, *host, *port;
    PGCONN pgconn;
    PGresult *res;
    int rec_num = 0;
    dbHandle *list;

    *dblist = NULL;
    *dbcount = 0;

    /* TODO: the solution below is not good as user usually does not
     * have permissions for "template1" */

    if (npaths < 1) {
        db_d_append_error(_("No path given"));
        db_d_report_error();
        return DB_FAILED;
    }

    if (parse_conn(db_get_string(dbpath), &pgconn) == DB_FAILED) {
        db_d_report_error();
        return DB_FAILED;
    }
    
    G_debug(1, "db = %s, user = %s, pass = %s, "
            "host = %s, port = %s, options = %s, tty = %s",
	    pgconn.dbname, pgconn.user, pgconn.password, pgconn.host,
            pgconn.port, pgconn.options, pgconn.tty);

    db_get_login2("pg", NULL, &user, &passwd, &host, &port);
    G_debug(1, "user = %s, passwd = %s", user, passwd ? "xxx" : "");

    if (user || passwd) {
        pg_conn = PQsetdbLogin(host, port, pgconn.options, pgconn.tty,
                               "template1", user, passwd);
    }
    else {
        pg_conn =
            PQsetdb(host, port, pgconn.options, pgconn.tty,
                    "template1");
    }

    if (PQstatus(pg_conn) == CONNECTION_BAD) {
	db_d_append_error("%s\n%s",
			  _("Unable to connect to Postgres:"),
			  PQerrorMessage(pg_conn));
	db_d_report_error();
	PQfinish(pg_conn);
	return DB_FAILED;
    }

    res = PQexec(pg_conn, "select datname from pg_database");

    if (!res || PQresultStatus(res) != PGRES_TUPLES_OK) {
	db_d_append_error("%s\n%s",
			  _("Unable to select from Postgres:"),
			  PQerrorMessage(pg_conn));
	db_d_report_error();
	PQclear(res);
	PQfinish(pg_conn);
	return DB_FAILED;
    }

    rec_num = PQntuples(res);

    list = db_alloc_handle_array(rec_num);
    if (list == NULL) {
	db_d_append_error(_("Out of memory"));
	db_d_report_error();
	return DB_FAILED;
    }

    for (i = 0; i < rec_num; i++) {
	db_init_handle(&list[i]);
	if (db_set_handle(&list[i], PQgetvalue(res, i, 0), NULL) != DB_OK) {
	    db_d_append_error(_("Unable to set handle"));
	    db_d_report_error();
	    db_free_handle_array(list, rec_num);
	    return DB_FAILED;
	}
    }

    PQclear(res);
    PQfinish(pg_conn);

    *dblist = list;
    *dbcount = rec_num;

    return DB_OK;
}