| #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; |
| |
| ALLOC_ARRAY(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 = xmallocz(initial_width); |
| memset(empty_cell, ' ', initial_width); |
| 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; |
| } |