#include "cache.h" #include "column.h" #include "string-list.h" #include "parse-options.h" #include "run-command.h" #include "utf8.h" #define XY2LINEAR(d, x, y) (COL_LAYOUT((d)->colopts) == COL_COLUMN ? \ (x) * (d)->rows + (y) : \ (y) * (d)->cols + (x)) struct column_data { const struct string_list *list; unsigned int colopts; struct column_options opts; int rows, cols; int *len; /* cell length */ int *width; /* index to the longest row in column */ }; /* return length of 's' in letters, ANSI escapes stripped */ static int item_length(unsigned int colopts, const char *s) { int len, i = 0; struct strbuf str = STRBUF_INIT; strbuf_addstr(&str, s); while ((s = strstr(str.buf + i, "\033[")) != NULL) { int len = strspn(s + 2, "0123456789;"); i = s - str.buf; strbuf_remove(&str, i, len + 3); /* \033[<len><func char> */ } len = utf8_strwidth(str.buf); strbuf_release(&str); return len; } /* * Calculate cell width, rows and cols for a table of equal cells, given * table width and how many spaces between cells. */ static void layout(struct column_data *data, int *width) { int i; *width = 0; for (i = 0; i < data->list->nr; i++) if (*width < data->len[i]) *width = data->len[i]; *width += data->opts.padding; data->cols = (data->opts.width - strlen(data->opts.indent)) / *width; if (data->cols == 0) data->cols = 1; data->rows = DIV_ROUND_UP(data->list->nr, data->cols); } static void compute_column_width(struct column_data *data) { int i, x, y; for (x = 0; x < data->cols; x++) { data->width[x] = XY2LINEAR(data, x, 0); for (y = 0; y < data->rows; y++) { i = XY2LINEAR(data, x, y); if (i < data->list->nr && data->len[data->width[x]] < data->len[i]) data->width[x] = i; } } } /* * Shrink all columns by shortening them one row each time (and adding * more columns along the way). Hopefully the longest cell will be * moved to the next column, column is shrunk so we have more space * for new columns. The process ends when the whole thing no longer * fits in data->total_width. */ static void shrink_columns(struct column_data *data) { REALLOC_ARRAY(data->width, data->cols); while (data->rows > 1) { int x, total_width, cols, rows; rows = data->rows; cols = data->cols; data->rows--; data->cols = DIV_ROUND_UP(data->list->nr, data->rows); if (data->cols != cols) REALLOC_ARRAY(data->width, data->cols); compute_column_width(data); total_width = strlen(data->opts.indent); for (x = 0; x < data->cols; x++) { total_width += data->len[data->width[x]]; total_width += data->opts.padding; } if (total_width > data->opts.width) { data->rows = rows; data->cols = cols; break; } } compute_column_width(data); } /* Display without layout when not enabled */ static void display_plain(const struct string_list *list, const char *indent, const char *nl) { int i; for (i = 0; i < list->nr; i++) printf("%s%s%s", indent, list->items[i].string, nl); } /* Print a cell to stdout with all necessary leading/traling space */ static int display_cell(struct column_data *data, int initial_width, const char *empty_cell, int x, int y) { int i, len, newline; i = XY2LINEAR(data, x, y); if (i >= data->list->nr) return -1; len = data->len[i]; if (data->width && data->len[data->width[x]] < initial_width) { /* * empty_cell has initial_width chars, if real column * is narrower, increase len a bit so we fill less * space. */ len += initial_width - data->len[data->width[x]]; len -= data->opts.padding; } if (COL_LAYOUT(data->colopts) == COL_COLUMN) newline = i + data->rows >= data->list->nr; else newline = x == data->cols - 1 || i == data->list->nr - 1; printf("%s%s%s", x == 0 ? data->opts.indent : "", data->list->items[i].string, newline ? data->opts.nl : empty_cell + len); return 0; } /* Display COL_COLUMN or COL_ROW */ static void display_table(const struct string_list *list, unsigned int colopts, const struct column_options *opts) { struct column_data data; int x, y, i, initial_width; char *empty_cell; memset(&data, 0, sizeof(data)); data.list = list; data.colopts = colopts; data.opts = *opts; data.len = xmalloc(sizeof(*data.len) * list->nr); for (i = 0; i < list->nr; i++) data.len[i] = item_length(colopts, list->items[i].string); layout(&data, &initial_width); if (colopts & COL_DENSE) shrink_columns(&data); empty_cell = xmalloc(initial_width + 1); memset(empty_cell, ' ', initial_width); empty_cell[initial_width] = '\0'; for (y = 0; y < data.rows; y++) { for (x = 0; x < data.cols; x++) if (display_cell(&data, initial_width, empty_cell, x, y)) break; } free(data.len); free(data.width); free(empty_cell); } void print_columns(const struct string_list *list, unsigned int colopts, const struct column_options *opts) { struct column_options nopts; if (!list->nr) return; assert((colopts & COL_ENABLE_MASK) != COL_AUTO); memset(&nopts, 0, sizeof(nopts)); nopts.indent = opts && opts->indent ? opts->indent : ""; nopts.nl = opts && opts->nl ? opts->nl : "\n"; nopts.padding = opts ? opts->padding : 1; nopts.width = opts && opts->width ? opts->width : term_columns() - 1; if (!column_active(colopts)) { display_plain(list, "", "\n"); return; } switch (COL_LAYOUT(colopts)) { case COL_PLAIN: display_plain(list, nopts.indent, nopts.nl); break; case COL_ROW: case COL_COLUMN: display_table(list, colopts, &nopts); break; default: die("BUG: invalid layout mode %d", COL_LAYOUT(colopts)); } } int finalize_colopts(unsigned int *colopts, int stdout_is_tty) { if ((*colopts & COL_ENABLE_MASK) == COL_AUTO) { if (stdout_is_tty < 0) stdout_is_tty = isatty(1); *colopts &= ~COL_ENABLE_MASK; if (stdout_is_tty) *colopts |= COL_ENABLED; } return 0; } struct colopt { const char *name; unsigned int value; unsigned int mask; }; #define LAYOUT_SET 1 #define ENABLE_SET 2 static int parse_option(const char *arg, int len, unsigned int *colopts, int *group_set) { struct colopt opts[] = { { "always", COL_ENABLED, COL_ENABLE_MASK }, { "never", COL_DISABLED, COL_ENABLE_MASK }, { "auto", COL_AUTO, COL_ENABLE_MASK }, { "plain", COL_PLAIN, COL_LAYOUT_MASK }, { "column", COL_COLUMN, COL_LAYOUT_MASK }, { "row", COL_ROW, COL_LAYOUT_MASK }, { "dense", COL_DENSE, 0 }, }; int i; for (i = 0; i < ARRAY_SIZE(opts); i++) { int set = 1, arg_len = len, name_len; const char *arg_str = arg; if (!opts[i].mask) { if (arg_len > 2 && !strncmp(arg_str, "no", 2)) { arg_str += 2; arg_len -= 2; set = 0; } } name_len = strlen(opts[i].name); if (arg_len != name_len || strncmp(arg_str, opts[i].name, name_len)) continue; switch (opts[i].mask) { case COL_ENABLE_MASK: *group_set |= ENABLE_SET; break; case COL_LAYOUT_MASK: *group_set |= LAYOUT_SET; break; } if (opts[i].mask) *colopts = (*colopts & ~opts[i].mask) | opts[i].value; else { if (set) *colopts |= opts[i].value; else *colopts &= ~opts[i].value; } return 0; } return error("unsupported option '%s'", arg); } static int parse_config(unsigned int *colopts, const char *value) { const char *sep = " ,"; int group_set = 0; while (*value) { int len = strcspn(value, sep); if (len) { if (parse_option(value, len, colopts, &group_set)) return -1; value += len; } value += strspn(value, sep); } /* * If none of "always", "never", and "auto" is specified, then setting * layout implies "always". * * Current value in COL_ENABLE_MASK is disregarded. This means if * you set column.ui = auto and pass --column=row, then "auto" * will become "always". */ if ((group_set & LAYOUT_SET) && !(group_set & ENABLE_SET)) *colopts = (*colopts & ~COL_ENABLE_MASK) | COL_ENABLED; return 0; } static int column_config(const char *var, const char *value, const char *key, unsigned int *colopts) { if (!value) return config_error_nonbool(var); if (parse_config(colopts, value)) return error("invalid column.%s mode %s", key, value); return 0; } int git_column_config(const char *var, const char *value, const char *command, unsigned int *colopts) { const char *it; if (!skip_prefix(var, "column.", &it)) return 0; if (!strcmp(it, "ui")) return column_config(var, value, "ui", colopts); if (command && !strcmp(it, command)) return column_config(var, value, it, colopts); return 0; } int parseopt_column_callback(const struct option *opt, const char *arg, int unset) { unsigned int *colopts = opt->value; *colopts |= COL_PARSEOPT; *colopts &= ~COL_ENABLE_MASK; if (unset) /* --no-column == never */ return 0; /* --column == always unless "arg" states otherwise */ *colopts |= COL_ENABLED; if (arg) return parse_config(colopts, arg); return 0; } static int fd_out = -1; static struct child_process column_process = CHILD_PROCESS_INIT; int run_column_filter(int colopts, const struct column_options *opts) { struct argv_array *argv; if (fd_out != -1) return -1; child_process_init(&column_process); argv = &column_process.args; argv_array_push(argv, "column"); argv_array_pushf(argv, "--raw-mode=%d", colopts); if (opts && opts->width) argv_array_pushf(argv, "--width=%d", opts->width); if (opts && opts->indent) argv_array_pushf(argv, "--indent=%s", opts->indent); if (opts && opts->padding) argv_array_pushf(argv, "--padding=%d", opts->padding); fflush(stdout); column_process.in = -1; column_process.out = dup(1); column_process.git_cmd = 1; if (start_command(&column_process)) return -2; fd_out = dup(1); close(1); dup2(column_process.in, 1); close(column_process.in); return 0; } int stop_column_filter(void) { if (fd_out == -1) return -1; fflush(stdout); close(1); finish_command(&column_process); dup2(fd_out, 1); close(fd_out); fd_out = -1; return 0; }