diff options
Diffstat (limited to 'src/bin/psql/print.c')
| -rw-r--r-- | src/bin/psql/print.c | 1604 |
1 files changed, 844 insertions, 760 deletions
diff --git a/src/bin/psql/print.c b/src/bin/psql/print.c index c710f02f28b..5c6525b6f33 100644 --- a/src/bin/psql/print.c +++ b/src/bin/psql/print.c @@ -7,8 +7,8 @@ #include <math.h> #include <signal.h> #ifndef WIN32 -#include <unistd.h> /* for isatty() */ -#include <sys/ioctl.h> /* for ioctl() */ +#include <unistd.h> /* for isatty() */ +#include <sys/ioctl.h> /* for ioctl() */ #else #define popen(x,y) _popen(x,y) #define pclose(x) _pclose(x) @@ -16,386 +16,422 @@ #include <pqsignal.h> #include <libpq-fe.h> -#include <postgres_ext.h> /* for Oid type */ +#include <postgres_ext.h> /* for Oid type */ #define DEFAULT_PAGER "/bin/more" /*************************/ -/* Unaligned text */ +/* Unaligned text */ /*************************/ static void -print_unaligned_text(const char * title, char ** headers, char ** cells, char ** footers, - const char * opt_fieldsep, bool opt_barebones, - FILE * fout) +print_unaligned_text(const char *title, char **headers, char **cells, char **footers, + const char *opt_fieldsep, bool opt_barebones, + FILE *fout) { - unsigned int col_count = 0; - unsigned int i; - char ** ptr; - - if (!opt_fieldsep) - opt_fieldsep = ""; - - /* print title */ - if (!opt_barebones && title) - fprintf(fout, "%s\n", title); - - /* 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) - fputs("\n", fout); - - /* print cells */ - i = 0; - for (ptr = cells; *ptr; ptr++) { - fputs(*ptr, fout); - if ((i+1) % col_count) - fputs(opt_fieldsep, fout); - else - fputs("\n", fout); - i++; - } + unsigned int col_count = 0; + unsigned int i; + char **ptr; + + if (!opt_fieldsep) + opt_fieldsep = ""; + + /* print title */ + if (!opt_barebones && title) + fprintf(fout, "%s\n", title); + + /* 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) + fputs("\n", fout); + + /* print cells */ + i = 0; + for (ptr = cells; *ptr; ptr++) + { + fputs(*ptr, fout); + if ((i + 1) % col_count) + fputs(opt_fieldsep, fout); + else + fputs("\n", fout); + i++; + } - /* print footers */ + /* print footers */ - if (!opt_barebones && footers) - for (ptr = footers; *ptr; ptr++) - fprintf(fout, "%s\n", *ptr); + if (!opt_barebones && footers) + for (ptr = footers; *ptr; ptr++) + fprintf(fout, "%s\n", *ptr); } static void -print_unaligned_vertical(const char * title, char ** headers, char ** cells, char ** footers, - const char * opt_fieldsep, bool opt_barebones, - FILE * fout) +print_unaligned_vertical(const char *title, char **headers, char **cells, char **footers, + const char *opt_fieldsep, bool opt_barebones, + FILE *fout) { - unsigned int col_count = 0; - unsigned int i; - unsigned int record = 1; - char ** ptr; - - if (!opt_fieldsep) - opt_fieldsep = ""; - - /* print title */ - if (!opt_barebones && title) - fprintf(fout, "%s\n", title); - - /* count columns */ - for (ptr = headers; *ptr; ptr++) { - col_count++; - } - - /* print records */ - for (i=0, ptr = cells; *ptr; i++, ptr++) { - if (i % col_count == 0) { - if (!opt_barebones) - fprintf(fout, "-- RECORD %d\n", record++); - else - fputc('\n', fout); + unsigned int col_count = 0; + unsigned int i; + unsigned int record = 1; + char **ptr; + + if (!opt_fieldsep) + opt_fieldsep = ""; + + /* print title */ + if (!opt_barebones && title) + fprintf(fout, "%s\n", title); + + /* count columns */ + for (ptr = headers; *ptr; ptr++) + col_count++; + + /* print records */ + for (i = 0, ptr = cells; *ptr; i++, ptr++) + { + if (i % col_count == 0) + { + if (!opt_barebones) + fprintf(fout, "-- RECORD %d\n", record++); + else + fputc('\n', fout); + } + fprintf(fout, "%s%s%s\n", headers[i % col_count], opt_fieldsep, *ptr); } - fprintf(fout, "%s%s%s\n", headers[i%col_count], opt_fieldsep, *ptr); - } - /* print footers */ + /* print footers */ - if (!opt_barebones && footers) { - fputs("--- END ---\n", fout); - for (ptr = footers; *ptr; ptr++) - fprintf(fout, "%s\n", *ptr); - } + if (!opt_barebones && footers) + { + fputs("--- END ---\n", fout); + for (ptr = footers; *ptr; ptr++) + fprintf(fout, "%s\n", *ptr); + } } /********************/ -/* Aligned text */ +/* Aligned text */ /********************/ /* draw "line" */ static void -_print_horizontal_line(const unsigned int col_count, const unsigned int * widths, unsigned short border, FILE * fout) +_print_horizontal_line(const unsigned int col_count, const unsigned int *widths, unsigned short border, FILE *fout) { - 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); + 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, char ** headers, char ** cells, char ** footers, - const char * opt_align, bool opt_barebones, unsigned short int opt_border, - FILE * fout) +print_aligned_text(const char *title, char **headers, char **cells, char **footers, +const char *opt_align, bool opt_barebones, unsigned short int opt_border, + FILE *fout) { - unsigned int col_count = 0; - unsigned int i, tmp; - unsigned int * widths, total_w; - char ** ptr; - - /* 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 - 2; - else - 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); + unsigned int col_count = 0; + unsigned int i, + tmp; + unsigned int *widths, + total_w; + char **ptr; + + /* 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 - 2; else - fprintf(fout, "%-*s%s\n", (total_w-strlen(title))/2, "", title); - } - - /* print headers */ - if (!opt_barebones) { - if (opt_border==2) - _print_horizontal_line(col_count, widths, opt_border, fout); - - 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) , ""); - - if (i<col_count-1) { - if (opt_border==0) - fputc(' ', fout); + 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 - fputs(" | ", fout); - } + fprintf(fout, "%-*s%s\n", (total_w - strlen(title)) / 2, "", title); } - if (opt_border==2) - fputs(" |", fout); - else if (opt_border==1) - fputc(' ', fout);; - fputc('\n', fout); + /* print headers */ + if (!opt_barebones) + { + if (opt_border == 2) + _print_horizontal_line(col_count, widths, opt_border, fout); - _print_horizontal_line(col_count, widths, opt_border, fout); - } + 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), ""); + + if (i < col_count - 1) + { + if (opt_border == 0) + fputc(' ', fout); + else + fputs(" | ", fout); + } + } - /* 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); - } + if (opt_border == 2) + fputs(" |", fout); + else if (opt_border == 1) + fputc(' ', fout);; + fputc('\n', 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]); + _print_horizontal_line(col_count, widths, opt_border, fout); } - /* 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); + /* 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); + } } - } - if (opt_border==2) - _print_horizontal_line(col_count, widths, opt_border, fout); + if (opt_border == 2) + _print_horizontal_line(col_count, widths, opt_border, fout); - /* print footers */ - if (footers && !opt_barebones) - for (ptr = footers; *ptr; ptr++) - fprintf(fout, "%s\n", *ptr); + /* print footers */ + if (footers && !opt_barebones) + for (ptr = footers; *ptr; ptr++) + fprintf(fout, "%s\n", *ptr); - fputc('\n', fout); + fputc('\n', fout); - /* clean up */ - free(widths); + /* clean up */ + free(widths); } static void -print_aligned_vertical(const char * title, char ** headers, char ** cells, char ** footers, - bool opt_barebones, unsigned short int opt_border, - FILE * fout) +print_aligned_vertical(const char *title, char **headers, char **cells, char **footers, + bool opt_barebones, unsigned short int opt_border, + FILE *fout) { - unsigned int col_count = 0; - unsigned int record = 1; - char ** ptr; - unsigned int i, tmp, hwidth=0, dwidth=0; - char * divider; - - /* 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 */ - } - - /* 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); - } - divider[0] = '\0'; - if (opt_border == 2) - strcat(divider, "+-"); - for (i=0; i<hwidth; i++) strcat(divider, opt_border > 0 ? "-" : " "); - if (opt_border > 0) - strcat(divider, "-+-"); - else - 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 * div_copy = strdup(divider); - char * record_str = malloc(32); - size_t record_str_len; - - if (!div_copy || !record_str) { - perror("malloc"); - exit(EXIT_FAILURE); - } + unsigned int col_count = 0; + unsigned int record = 1; + char **ptr; + unsigned int i, + tmp, + hwidth = 0, + dwidth = 0; + char *divider; + + /* 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 */ + } - 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(div_copy)) { - void * new; - new = realloc(div_copy, record_str_len + opt_border); - if (!new) { - perror("realloc"); - exit(EXIT_FAILURE); - } - div_copy = new; - } - strncpy(div_copy + opt_border, record_str, record_str_len); - fprintf(fout, "%s\n", div_copy); - free(record_str); - free(div_copy); - } - else if (i != 0 && opt_border < 2) - fprintf(fout, "%s\n", divider); + /* 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); } + divider[0] = '\0'; if (opt_border == 2) - fputs("| ", fout); - fprintf(fout, "%-*s", hwidth, headers[i%col_count]); + strcat(divider, "+-"); + for (i = 0; i < hwidth; i++) + strcat(divider, opt_border > 0 ? "-" : " "); if (opt_border > 0) - fputs(" | ", fout); + strcat(divider, "-+-"); else - fputs(" ", fout); + 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 *div_copy = strdup(divider); + char *record_str = malloc(32); + size_t record_str_len; + + if (!div_copy || !record_str) + { + 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(div_copy)) + { + void *new; + + new = realloc(div_copy, record_str_len + opt_border); + if (!new) + { + perror("realloc"); + exit(EXIT_FAILURE); + } + div_copy = new; + } + strncpy(div_copy + opt_border, record_str, record_str_len); + fprintf(fout, "%s\n", div_copy); + free(record_str); + free(div_copy); + } + else if (i != 0 && opt_border < 2) + fprintf(fout, "%s\n", divider); + } + if (opt_border == 2) + fputs("| ", fout); + fprintf(fout, "%-*s", hwidth, headers[i % col_count]); + if (opt_border > 0) + fputs(" | ", fout); + else + fputs(" ", fout); - if (opt_border < 2) - fprintf(fout, "%s\n", *ptr); - else - fprintf(fout, "%-*s |\n", dwidth, *ptr); - } + if (opt_border < 2) + fprintf(fout, "%s\n", *ptr); + else + fprintf(fout, "%-*s |\n", dwidth, *ptr); + } - if (opt_border == 2) - fprintf(fout, "%s\n", divider); + if (opt_border == 2) + fprintf(fout, "%s\n", divider); - /* print footers */ + /* print footers */ - if (!opt_barebones && footers && *footers) { - if (opt_border < 2) - fputc('\n', fout); - for (ptr = footers; *ptr; ptr++) - fprintf(fout, "%s\n", *ptr); - } + if (!opt_barebones && footers && *footers) + { + if (opt_border < 2) + fputc('\n', fout); + for (ptr = footers; *ptr; ptr++) + fprintf(fout, "%s\n", *ptr); + } - fputc('\n', fout); - free(divider); + fputc('\n', fout); + free(divider); } @@ -408,351 +444,382 @@ print_aligned_vertical(const char * title, char ** headers, char ** cells, char static void -html_escaped_print(const char * in, FILE * fout) +html_escaped_print(const char *in, FILE *fout) { - const char * p; - for (p=in; *p; p++) - switch (*p) { - case '&': - fputs("&", fout); - break; - case '<': - fputs("<", fout); - break; - case '>': - fputs(">", fout); - break; - case '\n': - fputs("<br>", fout); - break; - default: - fputc(*p, fout); - } + const char *p; + + for (p = in; *p; p++) + switch (*p) + { + case '&': + fputs("&", fout); + break; + case '<': + fputs("<", fout); + break; + case '>': + fputs(">", fout); + break; + case '\n': + fputs("<br>", fout); + break; + default: + fputc(*p, fout); + } } static void -print_html_text(const char * title, char ** headers, char ** cells, char ** footers, - const char * opt_align, bool opt_barebones, unsigned short int opt_border, - char * opt_table_attr, - FILE * fout) +print_html_text(const char *title, char **headers, char **cells, char **footers, +const char *opt_align, bool opt_barebones, unsigned short int opt_border, + char *opt_table_attr, + FILE *fout) { - unsigned int col_count = 0; - unsigned int i; - char ** ptr; - - 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); - } - - /* 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(" ", fout); - else - html_escaped_print(*ptr, fout); - fputs("</td>\n", fout); + unsigned int col_count = 0; + unsigned int i; + char **ptr; + + 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); + } - if ( (i+1) % col_count == 0 ) - fputs(" </tr>\n", fout); - } + /* 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(" ", fout); + else + html_escaped_print(*ptr, fout); + fputs("</td>\n", fout); - fputs("</table>\n", fout); + if ((i + 1) % col_count == 0) + fputs(" </tr>\n", fout); + } - /* print footers */ + fputs("</table>\n", fout); - if (footers && !opt_barebones) - for (ptr = footers; *ptr; ptr++) { - html_escaped_print(*ptr, fout); - fputs("<br>\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); + fputc('\n', fout); } static void -print_html_vertical(const char * title, char ** headers, char ** cells, char ** footers, - const char * opt_align, bool opt_barebones, unsigned short int opt_border, - char * opt_table_attr, - FILE * fout) +print_html_vertical(const char *title, char **headers, char **cells, char **footers, +const char *opt_align, bool opt_barebones, unsigned short int opt_border, + char *opt_table_attr, + FILE *fout) { - unsigned int col_count = 0; - unsigned int i; - unsigned int record = 1; - char ** ptr; - - 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); - } - - /* count columns */ - for (ptr = headers; *ptr; ptr++) - col_count++; - - /* 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> </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(" ", fout); - else - html_escaped_print(*ptr, fout); - fputs("</td>\n </tr>\n", fout); - } + unsigned int col_count = 0; + unsigned int i; + unsigned int record = 1; + char **ptr; + + 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); + } - fputs("</table>\n", fout); + /* count columns */ + for (ptr = headers; *ptr; ptr++) + col_count++; - /* print footers */ - if (footers && !opt_barebones) - for (ptr = footers; *ptr; ptr++) { - html_escaped_print(*ptr, fout); - fputs("<br>\n", fout); + /* 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> </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(" ", fout); + else + html_escaped_print(*ptr, fout); + fputs("</td>\n </tr>\n", fout); } - fputc('\n', fout); + 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); } /*************************/ -/* LaTeX */ +/* LaTeX */ /*************************/ static void -latex_escaped_print(const char * in, FILE * fout) +latex_escaped_print(const char *in, FILE *fout) { - 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); - } + 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, char ** headers, char ** cells, char ** footers, - const char * opt_align, bool opt_barebones, unsigned short int opt_border, - FILE * fout) +print_latex_text(const char *title, char **headers, char **cells, char **footers, +const char *opt_align, bool opt_barebones, unsigned short int opt_border, + FILE *fout) { - unsigned int col_count = 0; - unsigned int i; - const char * cp; - char ** ptr; - - - /* print title */ - if (!opt_barebones && title) { - fputs("\begin{center}\n", fout); - latex_escaped_print(title, fout); - fputs("\n\end{center}\n\n", fout); - } - - /* 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); - - if (!opt_barebones && opt_border==2) - fputs("\\hline\n", fout); - - /* 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); + unsigned int col_count = 0; + unsigned int i; + const char *cp; + char **ptr; + + + /* print title */ + if (!opt_barebones && title) + { + fputs("\begin{center}\n", fout); + latex_escaped_print(title, fout); + fputs("\n\end{center}\n\n", fout); } - } - if (!opt_barebones) { - fputs(" \\\\\n", fout); - fputs("\\hline\n", fout); - } + /* 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); - /* print cells */ - for (i=0, ptr = cells; *ptr; i++, ptr++) { - latex_escaped_print(*ptr, fout); + if (!opt_barebones && opt_border == 2) + fputs("\\hline\n", fout); - if ( (i+1) % col_count == 0 ) - fputs(" \\\\\n", fout); - else - fputs(" & ", fout); - } + /* 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); - if (opt_border==2) - fputs("\\hline\n", fout); + if ((i + 1) % col_count == 0) + fputs(" \\\\\n", fout); + else + fputs(" & ", fout); + } - fputs("\\end{tabular}\n\n", fout); + 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); - } + /* print footers */ - fputc('\n', fout); + 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, char ** headers, char ** cells, char ** footers, - const char * opt_align, bool opt_barebones, unsigned short int opt_border, - FILE * fout) +print_latex_vertical(const char *title, char **headers, char **cells, char **footers, +const char *opt_align, bool opt_barebones, unsigned short int opt_border, + FILE *fout) { - unsigned int col_count = 0; - unsigned int i; - char ** ptr; - unsigned int record = 1; - - (void)opt_align; /* currently unused parameter */ - - /* print title */ - if (!opt_barebones && title) { - fputs("\begin{center}\n", fout); - latex_escaped_print(title, fout); - fputs("\n\end{center}\n\n", fout); - } - - /* 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); - - - /* count columns */ - for (ptr = headers; *ptr; ptr++) - col_count++; - - - /* 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); + unsigned int col_count = 0; + unsigned int i; + char **ptr; + unsigned int record = 1; + + (void) opt_align; /* currently unused parameter */ + + /* print title */ + if (!opt_barebones && title) + { + fputs("\begin{center}\n", fout); + latex_escaped_print(title, fout); + fputs("\n\end{center}\n\n", fout); } - latex_escaped_print(headers[i%col_count], fout); - fputs(" & ", fout); - latex_escaped_print(*ptr, fout); - fputs(" \\\\\n", fout); - } + /* 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); - if (opt_border==2) - fputs("\\hline\n", fout); - fputs("\\end{tabular}\n\n", fout); + /* count columns */ + for (ptr = headers; *ptr; ptr++) + col_count++; - /* print footers */ + /* 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); + } - if (footers && !opt_barebones) - for (ptr = footers; *ptr; ptr++) { - latex_escaped_print(*ptr, fout); - fputs(" \\\\\n", fout); + latex_escaped_print(headers[i % col_count], fout); + fputs(" & ", fout); + latex_escaped_print(*ptr, fout); + fputs(" \\\\\n", fout); } - fputc('\n', fout); + 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); } @@ -761,214 +828,231 @@ print_latex_vertical(const char * title, char ** headers, char ** cells, char ** /********************************/ -/* Public functions */ +/* Public functions */ /********************************/ void -printTable(const char * title, char ** headers, char ** cells, char ** footers, - const char * align, - const printTableOpt * opt, FILE * fout) +printTable(const char *title, char **headers, char **cells, char **footers, + const char *align, + const printTableOpt * opt, FILE *fout) { - char * default_footer[] = { NULL }; - unsigned short int border = opt->border; - FILE * pager = NULL, - * output; + char *default_footer[] = {NULL}; + unsigned short int border = opt->border; + FILE *pager = NULL, + *output; - if (opt->format == PRINT_NOTHING) - return; + if (opt->format == PRINT_NOTHING) + return; - if (!footers) - footers = default_footer; + if (!footers) + footers = default_footer; - if (opt->format != PRINT_HTML && border > 2) - border = 2; + if (opt->format != PRINT_HTML && border > 2) + border = 2; - /* check whether we need / can / are supposed to use pager */ - if (fout == stdout && opt->pager + /* check whether we need / can / are supposed to use pager */ + if (fout == stdout && opt->pager #ifndef WIN32 - && - isatty(fileno(stdin)) && - isatty(fileno(stdout)) + && + isatty(fileno(stdin)) && + isatty(fileno(stdout)) #endif - ) - { - const char * pagerprog; -#ifdef TIOCGWINSZ - unsigned int col_count=0, row_count=0, lines; - char ** ptr; - 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; + ) + { + const char *pagerprog; - if (opt->expanded) - lines = (col_count+1) * row_count; - else - lines = row_count+1; - if (!opt->tuples_only) - lines += 5; +#ifdef TIOCGWINSZ + unsigned int col_count = 0, + row_count = 0, + lines; + char **ptr; + 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; - result = ioctl(fileno(stdout), TIOCGWINSZ, &screen_size); - if (result==-1 || lines > screen_size.ws_row) { + result = ioctl(fileno(stdout), TIOCGWINSZ, &screen_size); + if (result == -1 || lines > screen_size.ws_row) + { #endif - pagerprog = getenv("PAGER"); - if (!pagerprog) pagerprog = DEFAULT_PAGER; - pager = popen(pagerprog, "w"); + pagerprog = getenv("PAGER"); + if (!pagerprog) + pagerprog = DEFAULT_PAGER; + pager = popen(pagerprog, "w"); #ifdef TIOCGWINSZ - } + } #endif - } - - if (pager) { - output = pager; - pqsignal(SIGPIPE, SIG_IGN); - } - else - output = fout; - - - /* print the stuff */ + } - switch (opt->format) { - case PRINT_UNALIGNED: - if (opt->expanded) - print_unaligned_vertical(title, headers, cells, footers, opt->fieldSep, opt->tuples_only, output); - else - print_unaligned_text(title, headers, cells, footers, opt->fieldSep, opt->tuples_only, output); - 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); + if (pager) + { + output = pager; + pqsignal(SIGPIPE, SIG_IGN); + } 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); - pqsignal(SIGPIPE, SIG_DFL); - } + output = fout; + + + /* print the stuff */ + + switch (opt->format) + { + case PRINT_UNALIGNED: + if (opt->expanded) + print_unaligned_vertical(title, headers, cells, footers, opt->fieldSep, opt->tuples_only, output); + else + print_unaligned_text(title, headers, cells, footers, opt->fieldSep, opt->tuples_only, output); + 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); + pqsignal(SIGPIPE, SIG_DFL); + } } void -printQuery(PGresult * result, const printQueryOpt * opt, FILE * fout) +printQuery(PGresult *result, const printQueryOpt * opt, FILE *fout) { - int nfields; - char ** headers; - char ** cells; - char ** footers; - char * align; - int i; + int nfields; + char **headers; + char **cells; + char **footers; + char *align; + int i; + + /* extract headers */ + + nfields = PQnfields(result); + + headers = calloc(nfields + 1, sizeof(*headers)); + if (!headers) + { + perror("calloc"); + exit(EXIT_FAILURE); + } - /* extract headers */ + for (i = 0; i < nfields; i++) + headers[i] = PQfname(result, i); - nfields = PQnfields(result); + /* set cells */ - headers = calloc(nfields+1, sizeof(*headers)); - if (!headers) { - perror("calloc"); - exit(EXIT_FAILURE); - } + cells = calloc(nfields * PQntuples(result) + 1, sizeof(*cells)); + if (!cells) + { + perror("calloc"); + exit(EXIT_FAILURE); + } - for (i=0; i<nfields; i++) - headers[i] = PQfname(result, i); + 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); + } - /* set cells */ + /* set footers */ - cells = calloc(nfields * PQntuples(result) + 1, sizeof(*cells)); - if (!cells) { - perror("calloc"); - exit(EXIT_FAILURE); - } + if (opt->footers) + footers = opt->footers; + else if (!opt->topt.expanded) + { + footers = calloc(2, sizeof(*footers)); + if (!footers) + { + perror("calloc"); + exit(EXIT_FAILURE); + } - for (i=0; i< nfields * PQntuples(result); i++) { - if (PQgetisnull(result, i / nfields, i % nfields)) - cells[i] = opt->nullPrint ? opt->nullPrint : ""; + footers[0] = malloc(100); + if (PQntuples(result) == 1) + strcpy(footers[0], "(1 row)"); + else + sprintf(footers[0], "(%d rows)", PQntuples(result)); + } else - cells[i] = PQgetvalue(result, i / nfields, i % nfields); - } + footers = NULL; - /* set footers */ + /* set alignment */ - if (opt->footers) - footers = opt->footers; - else if (!opt->topt.expanded) { - footers = calloc(2, sizeof(*footers)); - if (!footers) { - perror("calloc"); - exit(EXIT_FAILURE); + align = calloc(nfields + 1, sizeof(*align)); + if (!align) + { + perror("calloc"); + exit(EXIT_FAILURE); } - footers[0] = malloc(100); - if (PQntuples(result)==1) - strcpy(footers[0], "(1 row)"); - else - sprintf(footers[0], "(%d rows)", PQntuples(result)); - } - else - footers = NULL; - - /* set alignment */ - - align = calloc(nfields+1, sizeof(*align)); - if (!align) { - perror("calloc"); - exit(EXIT_FAILURE); - } - - 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'; - } + 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'; + } - /* call table printer */ + /* call table printer */ - printTable(opt->title, headers, cells, footers ? footers : opt->footers, align, - &opt->topt, fout); + printTable(opt->title, headers, cells, footers ? footers : opt->footers, align, + &opt->topt, fout); - free(headers); - free(cells); - if (footers) { - free(footers[0]); - free(footers); - } + free(headers); + free(cells); + if (footers) + { + free(footers[0]); + free(footers); + } } |