#include <unistd.h>
#include "libpq-fe.h"

char *hostname = "";
char *port     = "55432";
char *dbname   = "alvherre";
char *username = "alvherre";
char *password = "";

/* establece conexión con la BD. */
PGconn *
sql_conn(void)
{
	PGconn     *conn;

	/* conectar */
	conn = PQsetdbLogin(hostname,
			port,
			NULL,  /* options */
			NULL,  /* tty */
			dbname,
			username,
			password);

	/* verificar resultado */
	if (PQstatus(conn) != CONNECTION_OK)
	{
		fprintf(stderr, "la conexión a %s falló.\n", dbname);
		fprintf(stderr, "%s", PQerrorMessage(conn));

		PQfinish(conn);
		exit(1);
	}

	/* si la conexión está buena, retornar */
	return conn;
}


/*
 * Recibe una conexión y una sentencia SQL, la ejecuta y despliega
 * el resultado en la pantalla.
 */
int
sql_exec(PGconn *conn, const char *sql)
{
	PGresult   *res;

	int         nfields;
	int         nrows;
	int         i, j, l;
	int        *length;
	char       *pad;

	/* Ejecuta la consulta */
	res = PQexec(conn, sql);

	/* Verifica el código de retorno */
	if (!res || PQresultStatus(res) > 2)
	{
		fprintf(stderr, "falló la consulta: %s\n", PQerrorMessage(conn));
		fprintf(stderr, "la consulta era: %s\n", sql);

		PQclear(res);
		PQfinish(conn);
		exit(-1);
	}

	/* Obtiene el número de campos */
	nrows = PQntuples(res);
	nfields = PQnfields(res);

	/* Para cada columna, obtiene el ancho necesario */
	length = (int *) malloc(sizeof(int) * nfields);
	for (j = 0; j < nfields; j++)
		length[j] = strlen(PQfname(res, j));

	for (i = 0; i < nrows; i++)
	{
		for (j = 0; j < nfields; j++)
		{
			l = strlen(PQgetvalue(res, i, j));
			if (l > length[j])
				length[j] = strlen(PQgetvalue(res, i, j));
		}
	}

	/* imprime un encabezado */
	for (j = 0, l = 0; j < nfields; j++)
	{
		fprintf(stdout, "%*s", length[j] + 2, PQfname(res, j));
		l += length[j] + 2;
	}
	fprintf(stdout, "\n");
	pad = (char *) malloc(l + 1);
	memset(pad, '-', l);
	pad[l] = '\0';
	fprintf(stdout, "%s\n", pad);
	free(pad);

	/* imprime cada fila */
	for (i = 0; i < nrows; i++)
	{
		for (j = 0; j < nfields; j++)
			fprintf(stdout, "%*s", length[j] + 2, PQgetvalue(res, i, j));
		fprintf(stdout, "\n");
	}

	/* libera memoria */
	PQclear(res);
	free(length);

	return 0;
}


/*
 * Recibe una consulta SQL en el primer argumento, la ejecuta
 * y muestra el resultado.
 */
int main(int argc, char **argv)
{
	PGconn *pgconn;

	pgconn = sql_conn();

	sql_exec(pgconn, argv[1]);
}