postgresql/src/bin/psql/print.c

1127 lines
22 KiB
C
Raw Normal View History

2000-01-19 00:30:24 +01:00
/*
* psql - the PostgreSQL interactive terminal
*
* Copyright 2000 by PostgreSQL Global Development Group
2000-01-19 00:30:24 +01:00
*
* $Header: /cvsroot/pgsql/src/bin/psql/print.c,v 1.21 2001/08/01 18:44:54 momjian Exp $
2000-01-19 00:30:24 +01:00
*/
#include "postgres_fe.h"
#include "print.h"
#include <math.h>
#include <signal.h>
#ifndef WIN32
1999-11-05 00:14:30 +01:00
#include <unistd.h> /* for isatty() */
#include <sys/ioctl.h> /* for ioctl() */
#else
#define popen(x,y) _popen(x,y)
#define pclose(x) _pclose(x)
#endif
2000-02-16 14:15:26 +01:00
#include "pqsignal.h"
#include "libpq-fe.h"
2001-02-27 09:13:31 +01:00
#include "settings.h"
#ifndef __CYGWIN__
#define DEFAULT_PAGER "more"
#else
#define DEFAULT_PAGER "less"
#endif
#ifdef HAVE_TERMIOS_H
#include <termios.h>
#endif
/*************************/
1999-11-05 00:14:30 +01:00
/* Unaligned text */
/*************************/
static void
print_unaligned_text(const char *title, const char *const * headers,
const char *const * cells, const char *const * footers,
const char *opt_fieldsep, const char *opt_recordsep, bool opt_barebones,
FILE *fout)
{
1999-11-05 00:14:30 +01:00
unsigned int col_count = 0;
unsigned int i;
const char *const * ptr;
bool need_recordsep = false;
1999-11-05 00:14:30 +01:00
if (!opt_fieldsep)
opt_fieldsep = "";
if (!opt_recordsep)
opt_recordsep = "";
1999-11-05 00:14:30 +01:00
/* print title */
if (!opt_barebones && title)
2000-01-19 00:30:24 +01:00
fprintf(fout, "%s%s", title, opt_recordsep);
1999-11-05 00:14:30 +01:00
/* print headers and count columns */
for (ptr = headers; *ptr; ptr++)
{
col_count++;
if (!opt_barebones)
{
if (col_count > 1)
fputs(opt_fieldsep, fout);
fputs(*ptr, fout);
}
}
if (!opt_barebones)
need_recordsep = true;
1999-11-05 00:14:30 +01:00
/* print cells */
i = 0;
for (ptr = cells; *ptr; ptr++)
{
if (need_recordsep)
{
2000-01-19 00:30:24 +01:00
fputs(opt_recordsep, fout);
need_recordsep = false;
}
1999-11-05 00:14:30 +01:00
fputs(*ptr, fout);
if ((i + 1) % col_count)
fputs(opt_fieldsep, fout);
else
need_recordsep = true;
1999-11-05 00:14:30 +01:00
i++;
}
1999-11-05 00:14:30 +01:00
/* print footers */
1999-11-05 00:14:30 +01:00
if (!opt_barebones && footers)
for (ptr = footers; *ptr; ptr++)
{
if (need_recordsep)
{
fputs(opt_recordsep, fout);
need_recordsep = false;
}
2000-01-19 00:30:24 +01:00
fputs(*ptr, fout);
need_recordsep = true;
}
/* the last record needs to be concluded with a newline */
if (need_recordsep)
fputc('\n', fout);
}
static void
print_unaligned_vertical(const char *title, const char *const * headers,
const char *const * cells, const char *const * footers,
const char *opt_fieldsep, const char *opt_recordsep, bool opt_barebones,
FILE *fout)
{
1999-11-05 00:14:30 +01:00
unsigned int col_count = 0;
unsigned int i;
const char *const * ptr;
1999-11-05 00:14:30 +01:00
if (!opt_fieldsep)
opt_fieldsep = "";
if (!opt_recordsep)
opt_recordsep = "";
1999-11-05 00:14:30 +01:00
/* print title */
if (!opt_barebones && title)
fputs(title, fout);
1999-11-05 00:14:30 +01:00
/* count columns */
for (ptr = headers; *ptr; ptr++)
col_count++;
/* print records */
for (i = 0, ptr = cells; *ptr; i++, ptr++)
{
if (i != 0 || (!opt_barebones && title))
{
fputs(opt_recordsep, fout);
if (i % col_count == 0)
fputs(opt_recordsep, fout); /* another one */
}
fputs(headers[i % col_count], fout);
fputs(opt_fieldsep, fout);
fputs(*ptr, fout);
}
1999-11-05 00:14:30 +01:00
/* print footers */
2000-01-19 00:30:24 +01:00
if (!opt_barebones && footers && *footers)
1999-11-05 00:14:30 +01:00
{
fputs(opt_recordsep, fout);
1999-11-05 00:14:30 +01:00
for (ptr = footers; *ptr; ptr++)
{
fputs(opt_recordsep, fout);
fputs(*ptr, fout);
}
1999-11-05 00:14:30 +01:00
}
2000-01-19 00:30:24 +01:00
fputc('\n', fout);
}
/********************/
1999-11-05 00:14:30 +01:00
/* Aligned text */
/********************/
/* draw "line" */
static void
1999-11-05 00:14:30 +01:00
_print_horizontal_line(const unsigned int col_count, const unsigned int *widths, unsigned short border, FILE *fout)
{
1999-11-05 00:14:30 +01:00
unsigned int i,
j;
if (border == 1)
fputc('-', fout);
else if (border == 2)
fputs("+-", fout);
for (i = 0; i < col_count; i++)
{
for (j = 0; j < widths[i]; j++)
fputc('-', fout);
if (i < col_count - 1)
{
if (border == 0)
fputc(' ', fout);
else
fputs("-+-", fout);
}
}
if (border == 2)
fputs("-+", fout);
else if (border == 1)
fputc('-', fout);
fputc('\n', fout);
}
static void
print_aligned_text(const char *title, const char *const * headers,
const char *const * cells, const char *const * footers,
const char *opt_align, bool opt_barebones, unsigned short int opt_border,
FILE *fout)
{
1999-11-05 00:14:30 +01:00
unsigned int col_count = 0;
unsigned int i,
tmp;
unsigned int *widths,
total_w;
const char *const * ptr;
1999-11-05 00:14:30 +01:00
/* count columns */
for (ptr = headers; *ptr; ptr++)
col_count++;
widths = calloc(col_count, sizeof(*widths));
if (!widths)
{
perror("calloc");
exit(EXIT_FAILURE);
}
/* calc column widths */
for (i = 0; i < col_count; i++)
if ((tmp = strlen(headers[i])) > widths[i])
widths[i] = tmp; /* don't wanna call strlen twice */
for (i = 0, ptr = cells; *ptr; ptr++, i++)
if ((tmp = strlen(*ptr)) > widths[i % col_count])
widths[i % col_count] = tmp;
if (opt_border == 0)
total_w = col_count - 1;
else if (opt_border == 1)
total_w = col_count * 3 - 1;
else
1999-11-05 00:14:30 +01:00
total_w = col_count * 3 + 1;
for (i = 0; i < col_count; i++)
total_w += widths[i];
/* print title */
if (title && !opt_barebones)
{
if (strlen(title) >= total_w)
fprintf(fout, "%s\n", title);
else
2001-03-22 05:01:46 +01:00
fprintf(fout, "%-*s%s\n", (int) (total_w - strlen(title)) / 2, "", title);
}
1999-11-05 00:14:30 +01:00
/* print headers */
if (!opt_barebones)
{
if (opt_border == 2)
_print_horizontal_line(col_count, widths, opt_border, fout);
1999-11-05 00:14:30 +01:00
if (opt_border == 2)
fputs("| ", fout);
else if (opt_border == 1)
fputc(' ', fout);
for (i = 0; i < col_count; i++)
{
/* centered */
fprintf(fout, "%-*s%s%-*s",
(int) floor((widths[i] - strlen(headers[i])) / 2.0), "",
headers[i], (int) ceil((widths[i] - strlen(headers[i])) / 2.0), "");
1999-11-05 00:14:30 +01:00
if (i < col_count - 1)
{
if (opt_border == 0)
fputc(' ', fout);
else
fputs(" | ", fout);
}
}
1999-11-05 00:14:30 +01:00
if (opt_border == 2)
fputs(" |", fout);
else if (opt_border == 1)
fputc(' ', fout);;
fputc('\n', fout);
1999-11-05 00:14:30 +01:00
_print_horizontal_line(col_count, widths, opt_border, fout);
}
1999-11-05 00:14:30 +01:00
/* print cells */
for (i = 0, ptr = cells; *ptr; i++, ptr++)
{
/* beginning of line */
if (i % col_count == 0)
{
if (opt_border == 2)
fputs("| ", fout);
else if (opt_border == 1)
fputc(' ', fout);
}
/* content */
if (opt_align[(i) % col_count] == 'r')
fprintf(fout, "%*s", widths[i % col_count], cells[i]);
else
{
if ((i + 1) % col_count == 0 && opt_border != 2)
fputs(cells[i], fout);
else
fprintf(fout, "%-*s", widths[i % col_count], cells[i]);
}
/* divider */
if ((i + 1) % col_count)
{
if (opt_border == 0)
fputc(' ', fout);
else
fputs(" | ", fout);
}
/* end of line */
else
{
if (opt_border == 2)
fputs(" |", fout);
fputc('\n', fout);
}
}
1999-11-05 00:14:30 +01:00
if (opt_border == 2)
_print_horizontal_line(col_count, widths, opt_border, fout);
1999-11-05 00:14:30 +01:00
/* print footers */
if (footers && !opt_barebones)
for (ptr = footers; *ptr; ptr++)
fprintf(fout, "%s\n", *ptr);
1999-11-05 00:14:30 +01:00
fputc('\n', fout);
1999-11-05 00:14:30 +01:00
/* clean up */
free(widths);
}
static void
print_aligned_vertical(const char *title, const char *const * headers,
const char *const * cells, const char *const * footers,
bool opt_barebones, unsigned short int opt_border,
FILE *fout)
{
1999-11-05 00:14:30 +01:00
unsigned int col_count = 0;
unsigned int record = 1;
const char *const * ptr;
1999-11-05 00:14:30 +01:00
unsigned int i,
tmp,
hwidth = 0,
dwidth = 0;
char *divider;
if (cells[0] == NULL)
{
puts(gettext("(No rows)\n"));
return;
}
1999-11-05 00:14:30 +01:00
/* count columns and find longest header */
for (ptr = headers; *ptr; ptr++)
{
col_count++;
if ((tmp = strlen(*ptr)) > hwidth)
hwidth = tmp; /* don't wanna call strlen twice */
}
1999-11-05 00:14:30 +01:00
/* find longest data cell */
for (ptr = cells; *ptr; ptr++)
if ((tmp = strlen(*ptr)) > dwidth)
dwidth = tmp;
/* print title */
if (!opt_barebones && title)
fprintf(fout, "%s\n", title);
/* make horizontal border */
divider = malloc(hwidth + dwidth + 10);
if (!divider)
{
perror("malloc");
exit(EXIT_FAILURE);
}
1999-11-05 00:14:30 +01:00
divider[0] = '\0';
if (opt_border == 2)
1999-11-05 00:14:30 +01:00
strcat(divider, "+-");
for (i = 0; i < hwidth; i++)
strcat(divider, opt_border > 0 ? "-" : " ");
if (opt_border > 0)
1999-11-05 00:14:30 +01:00
strcat(divider, "-+-");
else
1999-11-05 00:14:30 +01:00
strcat(divider, " ");
for (i = 0; i < dwidth; i++)
strcat(divider, opt_border > 0 ? "-" : " ");
if (opt_border == 2)
strcat(divider, "-+");
/* print records */
for (i = 0, ptr = cells; *ptr; i++, ptr++)
{
if (i % col_count == 0)
{
if (!opt_barebones)
{
char *record_str = malloc(32);
size_t record_str_len;
if (!record_str)
1999-11-05 00:14:30 +01:00
{
perror("malloc");
exit(EXIT_FAILURE);
}
if (opt_border == 0)
sprintf(record_str, "* Record %d", record++);
else
sprintf(record_str, "[ RECORD %d ]", record++);
record_str_len = strlen(record_str);
if (record_str_len + opt_border > strlen(divider))
fprintf(fout, "%.*s%s\n", opt_border, divider, record_str);
else
{
char *div_copy = strdup(divider);
if (!div_copy)
{
perror("malloc");
exit(EXIT_FAILURE);
}
strncpy(div_copy + opt_border, record_str, record_str_len);
fprintf(fout, "%s\n", div_copy);
free(div_copy);
}
1999-11-05 00:14:30 +01:00
free(record_str);
}
else if (i != 0 || opt_border == 2)
1999-11-05 00:14:30 +01:00
fprintf(fout, "%s\n", divider);
}
1999-11-05 00:14:30 +01:00
if (opt_border == 2)
fputs("| ", fout);
fprintf(fout, "%-*s", hwidth, headers[i % col_count]);
1999-11-05 00:14:30 +01:00
if (opt_border > 0)
fputs(" | ", fout);
else
fputs(" ", fout);
1999-11-05 00:14:30 +01:00
if (opt_border < 2)
fprintf(fout, "%s\n", *ptr);
else
fprintf(fout, "%-*s |\n", dwidth, *ptr);
}
1999-11-05 00:14:30 +01:00
if (opt_border == 2)
fprintf(fout, "%s\n", divider);
1999-11-05 00:14:30 +01:00
/* print footers */
1999-11-05 00:14:30 +01:00
if (!opt_barebones && footers && *footers)
{
if (opt_border < 2)
fputc('\n', fout);
for (ptr = footers; *ptr; ptr++)
fprintf(fout, "%s\n", *ptr);
}
1999-11-05 00:14:30 +01:00
fputc('\n', fout);
free(divider);
}
/**********************/
/* HTML printing ******/
/**********************/
static void
1999-11-05 00:14:30 +01:00
html_escaped_print(const char *in, FILE *fout)
{
1999-11-05 00:14:30 +01:00
const char *p;
for (p = in; *p; p++)
switch (*p)
{
case '&':
fputs("&amp;", fout);
break;
case '<':
fputs("&lt;", fout);
break;
case '>':
fputs("&gt;", fout);
break;
case '\n':
fputs("<br>", fout);
break;
default:
fputc(*p, fout);
}
}
static void
print_html_text(const char *title, const char *const * headers,
const char *const * cells, const char *const * footers,
const char *opt_align, bool opt_barebones, unsigned short int opt_border,
const char *opt_table_attr,
FILE *fout)
{
1999-11-05 00:14:30 +01:00
unsigned int col_count = 0;
unsigned int i;
const char *const * ptr;
1999-11-05 00:14:30 +01:00
fprintf(fout, "<table border=%d", opt_border);
if (opt_table_attr)
fprintf(fout, " %s", opt_table_attr);
fputs(">\n", fout);
/* print title */
if (!opt_barebones && title)
{
fputs(" <caption>", fout);
html_escaped_print(title, fout);
fputs("</caption>\n", fout);
}
1999-11-05 00:14:30 +01:00
/* print headers and count columns */
if (!opt_barebones)
fputs(" <tr>\n", fout);
for (i = 0, ptr = headers; *ptr; i++, ptr++)
{
col_count++;
if (!opt_barebones)
{
fputs(" <th align=center>", fout);
html_escaped_print(*ptr, fout);
fputs("</th>\n", fout);
}
}
if (!opt_barebones)
fputs(" </tr>\n", fout);
/* print cells */
for (i = 0, ptr = cells; *ptr; i++, ptr++)
{
if (i % col_count == 0)
fputs(" <tr valign=top>\n", fout);
fprintf(fout, " <td align=%s>", opt_align[(i) % col_count] == 'r' ? "right" : "left");
if ((*ptr)[strspn(*ptr, " \t")] == '\0') /* is string only
* whitespace? */
fputs("&nbsp;", fout);
else
html_escaped_print(*ptr, fout);
fputs("</td>\n", fout);
1999-11-05 00:14:30 +01:00
if ((i + 1) % col_count == 0)
fputs(" </tr>\n", fout);
}
1999-11-05 00:14:30 +01:00
fputs("</table>\n", fout);
1999-11-05 00:14:30 +01:00
/* print footers */
if (footers && !opt_barebones)
for (ptr = footers; *ptr; ptr++)
{
html_escaped_print(*ptr, fout);
fputs("<br>\n", fout);
}
1999-11-05 00:14:30 +01:00
fputc('\n', fout);
}
static void
print_html_vertical(const char *title, const char *const * headers,
const char *const * cells, const char *const * footers,
const char *opt_align, bool opt_barebones, unsigned short int opt_border,
const char *opt_table_attr,
1999-11-05 00:14:30 +01:00
FILE *fout)
{
1999-11-05 00:14:30 +01:00
unsigned int col_count = 0;
unsigned int i;
unsigned int record = 1;
const char *const * ptr;
1999-11-05 00:14:30 +01:00
fprintf(fout, "<table border=%d", opt_border);
if (opt_table_attr)
fprintf(fout, " %s", opt_table_attr);
fputs(">\n", fout);
/* print title */
if (!opt_barebones && title)
{
fputs(" <caption>", fout);
html_escaped_print(title, fout);
fputs("</caption>\n", fout);
}
1999-11-05 00:14:30 +01:00
/* count columns */
for (ptr = headers; *ptr; ptr++)
col_count++;
1999-11-05 00:14:30 +01:00
/* print records */
for (i = 0, ptr = cells; *ptr; i++, ptr++)
{
if (i % col_count == 0)
{
if (!opt_barebones)
fprintf(fout, "\n <tr><td colspan=2 align=center>Record %d</td></tr>\n", record++);
else
fputs("\n <tr><td colspan=2>&nbsp;</td></tr>\n", fout);
}
fputs(" <tr valign=top>\n"
" <th>", fout);
html_escaped_print(headers[i % col_count], fout);
fputs("</th>\n", fout);
fprintf(fout, " <td align=%s>", opt_align[i % col_count] == 'r' ? "right" : "left");
if ((*ptr)[strspn(*ptr, " \t")] == '\0') /* is string only
* whitespace? */
fputs("&nbsp;", fout);
else
html_escaped_print(*ptr, fout);
fputs("</td>\n </tr>\n", fout);
}
1999-11-05 00:14:30 +01:00
fputs("</table>\n", fout);
/* print footers */
if (footers && !opt_barebones)
for (ptr = footers; *ptr; ptr++)
{
html_escaped_print(*ptr, fout);
fputs("<br>\n", fout);
}
fputc('\n', fout);
}
/*************************/
1999-11-05 00:14:30 +01:00
/* LaTeX */
/*************************/
static void
1999-11-05 00:14:30 +01:00
latex_escaped_print(const char *in, FILE *fout)
{
1999-11-05 00:14:30 +01:00
const char *p;
for (p = in; *p; p++)
switch (*p)
{
case '&':
fputs("\\&", fout);
break;
case '%':
fputs("\\%", fout);
break;
case '$':
fputs("\\$", fout);
break;
case '{':
fputs("\\{", fout);
break;
case '}':
fputs("\\}", fout);
break;
case '\\':
fputs("\\backslash", fout);
break;
case '\n':
fputs("\\\\", fout);
break;
default:
fputc(*p, fout);
}
}
static void
print_latex_text(const char *title, const char *const * headers,
const char *const * cells, const char *const * footers,
const char *opt_align, bool opt_barebones, unsigned short int opt_border,
FILE *fout)
{
1999-11-05 00:14:30 +01:00
unsigned int col_count = 0;
unsigned int i;
const char *cp;
const char *const * ptr;
1999-11-05 00:14:30 +01:00
/* print title */
if (!opt_barebones && title)
{
2000-01-19 00:30:24 +01:00
fputs("\\begin{center}\n", fout);
1999-11-05 00:14:30 +01:00
latex_escaped_print(title, fout);
2000-01-19 00:30:24 +01:00
fputs("\n\\end{center}\n\n", fout);
}
1999-11-05 00:14:30 +01:00
/* begin environment and set alignments and borders */
fputs("\\begin{tabular}{", fout);
if (opt_border == 0)
fputs(opt_align, fout);
else if (opt_border == 1)
{
for (cp = opt_align; *cp; cp++)
{
if (cp != opt_align)
fputc('|', fout);
fputc(*cp, fout);
}
}
else if (opt_border == 2)
{
for (cp = opt_align; *cp; cp++)
{
fputc('|', fout);
fputc(*cp, fout);
}
fputc('|', fout);
}
fputs("}\n", fout);
1999-11-05 00:14:30 +01:00
if (!opt_barebones && opt_border == 2)
fputs("\\hline\n", fout);
1999-11-05 00:14:30 +01:00
/* print headers and count columns */
for (i = 0, ptr = headers; *ptr; i++, ptr++)
{
col_count++;
if (!opt_barebones)
{
if (i != 0)
fputs(" & ", fout);
latex_escaped_print(*ptr, fout);
}
}
if (!opt_barebones)
{
fputs(" \\\\\n", fout);
fputs("\\hline\n", fout);
}
/* print cells */
for (i = 0, ptr = cells; *ptr; i++, ptr++)
{
latex_escaped_print(*ptr, fout);
1999-11-05 00:14:30 +01:00
if ((i + 1) % col_count == 0)
fputs(" \\\\\n", fout);
else
fputs(" & ", fout);
}
1999-11-05 00:14:30 +01:00
if (opt_border == 2)
fputs("\\hline\n", fout);
1999-11-05 00:14:30 +01:00
fputs("\\end{tabular}\n\n", fout);
1999-11-05 00:14:30 +01:00
/* print footers */
1999-11-05 00:14:30 +01:00
if (footers && !opt_barebones)
for (ptr = footers; *ptr; ptr++)
{
latex_escaped_print(*ptr, fout);
fputs(" \\\\\n", fout);
}
fputc('\n', fout);
}
static void
print_latex_vertical(const char *title, const char *const * headers,
const char *const * cells, const char *const * footers,
const char *opt_align, bool opt_barebones, unsigned short int opt_border,
FILE *fout)
{
1999-11-05 00:14:30 +01:00
unsigned int col_count = 0;
unsigned int i;
const char *const * ptr;
1999-11-05 00:14:30 +01:00
unsigned int record = 1;
(void) opt_align; /* currently unused parameter */
/* print title */
if (!opt_barebones && title)
{
2000-01-19 00:30:24 +01:00
fputs("\\begin{center}\n", fout);
1999-11-05 00:14:30 +01:00
latex_escaped_print(title, fout);
2000-01-19 00:30:24 +01:00
fputs("\n\\end{center}\n\n", fout);
}
1999-11-05 00:14:30 +01:00
/* begin environment and set alignments and borders */
fputs("\\begin{tabular}{", fout);
if (opt_border == 0)
fputs("cl", fout);
else if (opt_border == 1)
fputs("c|l", fout);
else if (opt_border == 2)
fputs("|c|l|", fout);
fputs("}\n", fout);
1999-11-05 00:14:30 +01:00
/* count columns */
for (ptr = headers; *ptr; ptr++)
col_count++;
1999-11-05 00:14:30 +01:00
/* print records */
for (i = 0, ptr = cells; *ptr; i++, ptr++)
{
/* new record */
if (i % col_count == 0)
{
if (!opt_barebones)
{
if (opt_border == 2)
fputs("\\hline\n", fout);
fprintf(fout, "\\multicolumn{2}{c}{Record %d} \\\\\n", record++);
}
if (opt_border >= 1)
fputs("\\hline\n", fout);
}
1999-11-05 00:14:30 +01:00
latex_escaped_print(headers[i % col_count], fout);
fputs(" & ", fout);
latex_escaped_print(*ptr, fout);
fputs(" \\\\\n", fout);
}
1999-11-05 00:14:30 +01:00
if (opt_border == 2)
fputs("\\hline\n", fout);
fputs("\\end{tabular}\n\n", fout);
/* print footers */
if (footers && !opt_barebones)
for (ptr = footers; *ptr; ptr++)
{
latex_escaped_print(*ptr, fout);
fputs(" \\\\\n", fout);
}
fputc('\n', fout);
}
/********************************/
/* Public functions */
/********************************/
void
printTable(const char *title,
const char *const * headers,
const char *const * cells,
const char *const * footers,
const char *align,
const printTableOpt *opt, FILE *fout)
{
const char *default_footer[] = {NULL};
1999-11-05 00:14:30 +01:00
unsigned short int border = opt->border;
FILE *pager = NULL,
*output;
1999-11-05 00:14:30 +01:00
if (opt->format == PRINT_NOTHING)
return;
1999-11-05 00:14:30 +01:00
if (!footers)
footers = default_footer;
1999-11-05 00:14:30 +01:00
if (opt->format != PRINT_HTML && border > 2)
border = 2;
1999-11-05 00:14:30 +01:00
/* check whether we need / can / are supposed to use pager */
if (fout == stdout && opt->pager
#ifndef WIN32
1999-11-05 00:14:30 +01:00
&&
isatty(fileno(stdin)) &&
isatty(fileno(stdout))
#endif
1999-11-05 00:14:30 +01:00
)
{
const char *pagerprog;
1999-11-05 00:14:30 +01:00
#ifdef TIOCGWINSZ
unsigned int col_count = 0,
row_count = 0,
lines;
const char *const * ptr;
1999-11-05 00:14:30 +01:00
int result;
struct winsize screen_size;
/* rough estimate of columns and rows */
if (headers)
for (ptr = headers; *ptr; ptr++)
col_count++;
if (cells)
for (ptr = cells; *ptr; ptr++)
row_count++;
row_count /= col_count;
if (opt->expanded)
lines = (col_count + 1) * row_count;
else
lines = row_count + 1;
if (!opt->tuples_only)
lines += 5;
1999-11-05 00:14:30 +01:00
result = ioctl(fileno(stdout), TIOCGWINSZ, &screen_size);
if (result == -1 || lines > screen_size.ws_row)
{
#endif
1999-11-05 00:14:30 +01:00
pagerprog = getenv("PAGER");
if (!pagerprog)
pagerprog = DEFAULT_PAGER;
pager = popen(pagerprog, "w");
#ifdef TIOCGWINSZ
1999-11-05 00:14:30 +01:00
}
#endif
1999-11-05 00:14:30 +01:00
}
1999-11-05 00:14:30 +01:00
if (pager)
{
output = pager;
#ifndef WIN32
1999-11-05 00:14:30 +01:00
pqsignal(SIGPIPE, SIG_IGN);
#endif
1999-11-05 00:14:30 +01:00
}
else
1999-11-05 00:14:30 +01:00
output = fout;
/* print the stuff */
switch (opt->format)
{
case PRINT_UNALIGNED:
if (opt->expanded)
2000-01-19 00:30:24 +01:00
print_unaligned_vertical(title, headers, cells, footers, opt->fieldSep, opt->recordSep, opt->tuples_only, output);
1999-11-05 00:14:30 +01:00
else
2000-01-19 00:30:24 +01:00
print_unaligned_text(title, headers, cells, footers, opt->fieldSep, opt->recordSep, opt->tuples_only, output);
1999-11-05 00:14:30 +01:00
break;
case PRINT_ALIGNED:
if (opt->expanded)
print_aligned_vertical(title, headers, cells, footers, opt->tuples_only, border, output);
else
print_aligned_text(title, headers, cells, footers, align, opt->tuples_only, border, output);
break;
case PRINT_HTML:
if (opt->expanded)
print_html_vertical(title, headers, cells, footers, align, opt->tuples_only, border, opt->tableAttr, output);
else
print_html_text(title, headers, cells, footers, align, opt->tuples_only, border, opt->tableAttr, output);
break;
case PRINT_LATEX:
if (opt->expanded)
print_latex_vertical(title, headers, cells, footers, align, opt->tuples_only, border, output);
else
print_latex_text(title, headers, cells, footers, align, opt->tuples_only, border, output);
break;
default:
fprintf(stderr, "+ Oops, you shouldn't see this!\n");
}
if (pager)
{
pclose(pager);
#ifndef WIN32
1999-11-05 00:14:30 +01:00
pqsignal(SIGPIPE, SIG_DFL);
#endif
1999-11-05 00:14:30 +01:00
}
}
void
printQuery(const PGresult *result, const printQueryOpt *opt, FILE *fout)
{
1999-11-05 00:14:30 +01:00
int nfields;
const char **headers;
const char **cells;
char **footers;
char *align;
1999-11-05 00:14:30 +01:00
int i;
/* extract headers */
nfields = PQnfields(result);
headers = calloc(nfields + 1, sizeof(*headers));
if (!headers)
{
perror("calloc");
exit(EXIT_FAILURE);
}
1999-11-05 00:14:30 +01:00
for (i = 0; i < nfields; i++)
headers[i] = PQfname(result, i);
1999-11-05 00:14:30 +01:00
/* set cells */
1999-11-05 00:14:30 +01:00
cells = calloc(nfields * PQntuples(result) + 1, sizeof(*cells));
if (!cells)
{
perror("calloc");
exit(EXIT_FAILURE);
}
1999-11-05 00:14:30 +01:00
for (i = 0; i < nfields * PQntuples(result); i++)
{
if (PQgetisnull(result, i / nfields, i % nfields))
cells[i] = opt->nullPrint ? opt->nullPrint : "";
else
cells[i] = PQgetvalue(result, i / nfields, i % nfields);
}
1999-11-05 00:14:30 +01:00
/* set footers */
1999-11-05 00:14:30 +01:00
if (opt->footers)
footers = opt->footers;
else if (!opt->topt.expanded && opt->default_footer)
1999-11-05 00:14:30 +01:00
{
footers = calloc(2, sizeof(*footers));
if (!footers)
{
perror("calloc");
exit(EXIT_FAILURE);
}
1999-11-05 00:14:30 +01:00
footers[0] = malloc(100);
if (PQntuples(result) == 1)
snprintf(footers[0], 100, gettext("(1 row)"));
1999-11-05 00:14:30 +01:00
else
snprintf(footers[0], 100, gettext("(%d rows)"), PQntuples(result));
1999-11-05 00:14:30 +01:00
}
else
1999-11-05 00:14:30 +01:00
footers = NULL;
1999-11-05 00:14:30 +01:00
/* set alignment */
1999-11-05 00:14:30 +01:00
align = calloc(nfields + 1, sizeof(*align));
if (!align)
{
perror("calloc");
exit(EXIT_FAILURE);
}
1999-11-05 00:14:30 +01:00
for (i = 0; i < nfields; i++)
{
Oid ftype = PQftype(result, i);
if (ftype == 20 || /* int8 */
ftype == 21 || /* int2 */
ftype == 23 || /* int4 */
(ftype >= 26 && ftype <= 30) || /* ?id */
ftype == 700 || /* float4 */
ftype == 701 || /* float8 */
ftype == 790 || /* money */
ftype == 1700 /* numeric */
)
align[i] = 'r';
else
align[i] = 'l';
}
1999-11-05 00:14:30 +01:00
/* call table printer */
printTable(opt->title, headers, cells,
footers ? (const char *const *) footers : (const char *const *) (opt->footers),
align, &opt->topt, fout);
1999-11-05 00:14:30 +01:00
free(headers);
free(cells);
if (footers)
{
free(footers[0]);
free(footers);
}
2001-03-01 19:52:50 +01:00
free(align);
}
/* the end */