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

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

int db__driver_list_tables (dbString **tlist, int *tcount, int system)

{
    int i, nrows, trows, vrows, ncols, tablecol, tschemacol, viewcol, vschemacol;
    dbString *list;
    PGresult *rest, *resv;
    char buf[1000];

    init_error();
    *tlist = NULL;
    *tcount = 0;


    /* Get table names */
    rest = PQexec(pg_conn, "select * from pg_tables where tablename !~ 'pg_*' order by tablename");
    
    if (!rest || PQresultStatus(rest) != PGRES_TUPLES_OK) {
	append_error ( "Cannot select table names\n" );
	append_error ( PQerrorMessage(pg_conn) );
	report_error();
	PQclear(rest);
	return DB_FAILED;
    }

    /* Find table and schema col */
    ncols = PQnfields(rest);
    tschemacol = -1;
    for (i = 0; i < ncols; i++) {
	if ( strcmp(PQfname(rest, i),"tablename") == 0 )
	    tablecol = i;

	if ( strcmp(PQfname(rest, i),"schemaname") == 0 )
	    tschemacol = i;
    }


    /* Get view names */
    resv = PQexec(pg_conn, "SELECT * FROM pg_views WHERE schemaname NOT IN ('pg_catalog','information_schema') AND viewname !~ '^pg_'");
    
    if (!resv || PQresultStatus(resv) != PGRES_TUPLES_OK) {
	append_error ( "Cannot select view names\n" );
	append_error ( PQerrorMessage(pg_conn) );
	report_error();
	PQclear(resv);
	return DB_FAILED;
    }

    /* Find viewname and schema col */
    ncols = PQnfields(resv);
    vschemacol = -1;
    for (i = 0; i < ncols; i++) {
	if ( strcmp(PQfname(resv, i),"viewname") == 0 )
	    viewcol = i;

	if ( strcmp(PQfname(resv, i),"schemaname") == 0 )
	    vschemacol = i;
    }



    trows = PQntuples(rest);
    vrows = PQntuples(resv);
    nrows = trows + vrows;
    
    list = db_alloc_string_array(nrows);
    
    if (list == NULL ) {
	append_error ( "Cannot db_alloc_string_array()");
	report_error();
	return DB_FAILED;
    }

    for (i = 0; i < trows; i++) {
	if ( tschemacol >= 0 ) {
	   sprintf ( buf, "%s.%s", (char *) PQgetvalue(rest, i, tschemacol), 
		                   (char *) PQgetvalue(rest, i, tablecol) );
	} else {
	   sprintf ( buf, "%s", (char *) PQgetvalue(rest, i, tablecol) );
	}
	db_set_string(&list[i], buf );
    }

    PQclear(rest);


    for (i = 0; i < vrows; i++) {
	if ( vschemacol >= 0 ) {
	   sprintf ( buf, "%s.%s", (char *) PQgetvalue(resv, i, vschemacol), 
		                   (char *) PQgetvalue(resv, i, viewcol) );
	} else {
	   sprintf ( buf, "%s", (char *) PQgetvalue(resv, i, viewcol) );
	}
	db_set_string(&list[i], buf );
    }

    PQclear(resv);


    *tlist = list;
    *tcount = nrows;
    return DB_OK;
}