1 /* Copyright (c) 2006-2008 Jonas Fonseca <fonseca@diku.dk>
2 *
3 * This program is free software; you can redistribute it and/or
4 * modify it under the terms of the GNU General Public License as
5 * published by the Free Software Foundation; either version 2 of
6 * the License, or (at your option) any later version.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 */
14 #ifdef HAVE_CONFIG_H
15 #include "config.h"
16 #endif
18 #ifndef TIG_VERSION
19 #define TIG_VERSION "unknown-version"
20 #endif
22 #ifndef DEBUG
23 #define NDEBUG
24 #endif
26 #include <assert.h>
27 #include <errno.h>
28 #include <ctype.h>
29 #include <signal.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #include <sys/types.h>
35 #include <sys/stat.h>
36 #include <unistd.h>
37 #include <time.h>
39 #include <regex.h>
41 #include <locale.h>
42 #include <langinfo.h>
43 #include <iconv.h>
45 /* ncurses(3): Must be defined to have extended wide-character functions. */
46 #define _XOPEN_SOURCE_EXTENDED
48 #include <curses.h>
50 #if __GNUC__ >= 3
51 #define __NORETURN __attribute__((__noreturn__))
52 #else
53 #define __NORETURN
54 #endif
56 static void __NORETURN die(const char *err, ...);
57 static void warn(const char *msg, ...);
58 static void report(const char *msg, ...);
59 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, size_t, char *, size_t));
60 static void set_nonblocking_input(bool loading);
61 static size_t utf8_length(const char *string, size_t max_width, int *trimmed, bool reserve);
63 #define ABS(x) ((x) >= 0 ? (x) : -(x))
64 #define MIN(x, y) ((x) < (y) ? (x) : (y))
66 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
67 #define STRING_SIZE(x) (sizeof(x) - 1)
69 #define SIZEOF_STR 1024 /* Default string size. */
70 #define SIZEOF_REF 256 /* Size of symbolic or SHA1 ID. */
71 #define SIZEOF_REV 41 /* Holds a SHA-1 and an ending NUL */
73 /* Revision graph */
75 #define REVGRAPH_INIT 'I'
76 #define REVGRAPH_MERGE 'M'
77 #define REVGRAPH_BRANCH '+'
78 #define REVGRAPH_COMMIT '*'
79 #define REVGRAPH_BOUND '^'
80 #define REVGRAPH_LINE '|'
82 #define SIZEOF_REVGRAPH 19 /* Size of revision ancestry graphics. */
84 /* This color name can be used to refer to the default term colors. */
85 #define COLOR_DEFAULT (-1)
87 #define ICONV_NONE ((iconv_t) -1)
88 #ifndef ICONV_CONST
89 #define ICONV_CONST /* nothing */
90 #endif
92 /* The format and size of the date column in the main view. */
93 #define DATE_FORMAT "%Y-%m-%d %H:%M"
94 #define DATE_COLS STRING_SIZE("2006-04-29 14:21 ")
96 #define AUTHOR_COLS 20
97 #define ID_COLS 8
99 /* The default interval between line numbers. */
100 #define NUMBER_INTERVAL 5
102 #define TABSIZE 8
104 #define SCALE_SPLIT_VIEW(height) ((height) * 2 / 3)
106 #define NULL_ID "0000000000000000000000000000000000000000"
108 #ifndef GIT_CONFIG
109 #define GIT_CONFIG "git config"
110 #endif
112 #define TIG_LS_REMOTE \
113 "git ls-remote $(git rev-parse --git-dir) 2>/dev/null"
115 #define TIG_DIFF_CMD \
116 "git show --pretty=fuller --no-color --root --patch-with-stat --find-copies-harder -C %s 2>/dev/null"
118 #define TIG_LOG_CMD \
119 "git log --no-color --cc --stat -n100 %s 2>/dev/null"
121 #define TIG_MAIN_CMD \
122 "git log --no-color --topo-order --parents --boundary --pretty=raw %s 2>/dev/null"
124 #define TIG_TREE_CMD \
125 "git ls-tree %s %s"
127 #define TIG_BLOB_CMD \
128 "git cat-file blob %s"
130 /* XXX: Needs to be defined to the empty string. */
131 #define TIG_HELP_CMD ""
132 #define TIG_PAGER_CMD ""
133 #define TIG_STATUS_CMD ""
134 #define TIG_STAGE_CMD ""
135 #define TIG_BLAME_CMD ""
137 /* Some ascii-shorthands fitted into the ncurses namespace. */
138 #define KEY_TAB '\t'
139 #define KEY_RETURN '\r'
140 #define KEY_ESC 27
143 struct ref {
144 char *name; /* Ref name; tag or head names are shortened. */
145 char id[SIZEOF_REV]; /* Commit SHA1 ID */
146 unsigned int head:1; /* Is it the current HEAD? */
147 unsigned int tag:1; /* Is it a tag? */
148 unsigned int ltag:1; /* If so, is the tag local? */
149 unsigned int remote:1; /* Is it a remote ref? */
150 unsigned int tracked:1; /* Is it the remote for the current HEAD? */
151 unsigned int next:1; /* For ref lists: are there more refs? */
152 };
154 static struct ref **get_refs(char *id);
156 struct int_map {
157 const char *name;
158 int namelen;
159 int value;
160 };
162 static int
163 set_from_int_map(struct int_map *map, size_t map_size,
164 int *value, const char *name, int namelen)
165 {
167 int i;
169 for (i = 0; i < map_size; i++)
170 if (namelen == map[i].namelen &&
171 !strncasecmp(name, map[i].name, namelen)) {
172 *value = map[i].value;
173 return OK;
174 }
176 return ERR;
177 }
180 /*
181 * String helpers
182 */
184 static inline void
185 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
186 {
187 if (srclen > dstlen - 1)
188 srclen = dstlen - 1;
190 strncpy(dst, src, srclen);
191 dst[srclen] = 0;
192 }
194 /* Shorthands for safely copying into a fixed buffer. */
196 #define string_copy(dst, src) \
197 string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
199 #define string_ncopy(dst, src, srclen) \
200 string_ncopy_do(dst, sizeof(dst), src, srclen)
202 #define string_copy_rev(dst, src) \
203 string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
205 #define string_add(dst, from, src) \
206 string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
208 static char *
209 chomp_string(char *name)
210 {
211 int namelen;
213 while (isspace(*name))
214 name++;
216 namelen = strlen(name) - 1;
217 while (namelen > 0 && isspace(name[namelen]))
218 name[namelen--] = 0;
220 return name;
221 }
223 static bool
224 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
225 {
226 va_list args;
227 size_t pos = bufpos ? *bufpos : 0;
229 va_start(args, fmt);
230 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
231 va_end(args);
233 if (bufpos)
234 *bufpos = pos;
236 return pos >= bufsize ? FALSE : TRUE;
237 }
239 #define string_format(buf, fmt, args...) \
240 string_nformat(buf, sizeof(buf), NULL, fmt, args)
242 #define string_format_from(buf, from, fmt, args...) \
243 string_nformat(buf, sizeof(buf), from, fmt, args)
245 static int
246 string_enum_compare(const char *str1, const char *str2, int len)
247 {
248 size_t i;
250 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
252 /* Diff-Header == DIFF_HEADER */
253 for (i = 0; i < len; i++) {
254 if (toupper(str1[i]) == toupper(str2[i]))
255 continue;
257 if (string_enum_sep(str1[i]) &&
258 string_enum_sep(str2[i]))
259 continue;
261 return str1[i] - str2[i];
262 }
264 return 0;
265 }
267 /* Shell quoting
268 *
269 * NOTE: The following is a slightly modified copy of the git project's shell
270 * quoting routines found in the quote.c file.
271 *
272 * Help to copy the thing properly quoted for the shell safety. any single
273 * quote is replaced with '\'', any exclamation point is replaced with '\!',
274 * and the whole thing is enclosed in a
275 *
276 * E.g.
277 * original sq_quote result
278 * name ==> name ==> 'name'
279 * a b ==> a b ==> 'a b'
280 * a'b ==> a'\''b ==> 'a'\''b'
281 * a!b ==> a'\!'b ==> 'a'\!'b'
282 */
284 static size_t
285 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
286 {
287 char c;
289 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
291 BUFPUT('\'');
292 while ((c = *src++)) {
293 if (c == '\'' || c == '!') {
294 BUFPUT('\'');
295 BUFPUT('\\');
296 BUFPUT(c);
297 BUFPUT('\'');
298 } else {
299 BUFPUT(c);
300 }
301 }
302 BUFPUT('\'');
304 if (bufsize < SIZEOF_STR)
305 buf[bufsize] = 0;
307 return bufsize;
308 }
311 /*
312 * User requests
313 */
315 #define REQ_INFO \
316 /* XXX: Keep the view request first and in sync with views[]. */ \
317 REQ_GROUP("View switching") \
318 REQ_(VIEW_MAIN, "Show main view"), \
319 REQ_(VIEW_DIFF, "Show diff view"), \
320 REQ_(VIEW_LOG, "Show log view"), \
321 REQ_(VIEW_TREE, "Show tree view"), \
322 REQ_(VIEW_BLOB, "Show blob view"), \
323 REQ_(VIEW_BLAME, "Show blame view"), \
324 REQ_(VIEW_HELP, "Show help page"), \
325 REQ_(VIEW_PAGER, "Show pager view"), \
326 REQ_(VIEW_STATUS, "Show status view"), \
327 REQ_(VIEW_STAGE, "Show stage view"), \
328 \
329 REQ_GROUP("View manipulation") \
330 REQ_(ENTER, "Enter current line and scroll"), \
331 REQ_(NEXT, "Move to next"), \
332 REQ_(PREVIOUS, "Move to previous"), \
333 REQ_(VIEW_NEXT, "Move focus to next view"), \
334 REQ_(REFRESH, "Reload and refresh"), \
335 REQ_(MAXIMIZE, "Maximize the current view"), \
336 REQ_(VIEW_CLOSE, "Close the current view"), \
337 REQ_(QUIT, "Close all views and quit"), \
338 \
339 REQ_GROUP("Cursor navigation") \
340 REQ_(MOVE_UP, "Move cursor one line up"), \
341 REQ_(MOVE_DOWN, "Move cursor one line down"), \
342 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
343 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
344 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
345 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
346 \
347 REQ_GROUP("Scrolling") \
348 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
349 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
350 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
351 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
352 \
353 REQ_GROUP("Searching") \
354 REQ_(SEARCH, "Search the view"), \
355 REQ_(SEARCH_BACK, "Search backwards in the view"), \
356 REQ_(FIND_NEXT, "Find next search match"), \
357 REQ_(FIND_PREV, "Find previous search match"), \
358 \
359 REQ_GROUP("Misc") \
360 REQ_(PROMPT, "Bring up the prompt"), \
361 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
362 REQ_(SCREEN_RESIZE, "Resize the screen"), \
363 REQ_(SHOW_VERSION, "Show version information"), \
364 REQ_(STOP_LOADING, "Stop all loading views"), \
365 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
366 REQ_(TOGGLE_DATE, "Toggle date display"), \
367 REQ_(TOGGLE_AUTHOR, "Toggle author display"), \
368 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization"), \
369 REQ_(TOGGLE_REFS, "Toggle reference display (tags/branches)"), \
370 REQ_(STATUS_UPDATE, "Update file status"), \
371 REQ_(STATUS_MERGE, "Merge file using external tool"), \
372 REQ_(TREE_PARENT, "Switch to parent directory in tree view"), \
373 REQ_(EDIT, "Open in editor"), \
374 REQ_(NONE, "Do nothing")
377 /* User action requests. */
378 enum request {
379 #define REQ_GROUP(help)
380 #define REQ_(req, help) REQ_##req
382 /* Offset all requests to avoid conflicts with ncurses getch values. */
383 REQ_OFFSET = KEY_MAX + 1,
384 REQ_INFO
386 #undef REQ_GROUP
387 #undef REQ_
388 };
390 struct request_info {
391 enum request request;
392 char *name;
393 int namelen;
394 char *help;
395 };
397 static struct request_info req_info[] = {
398 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
399 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
400 REQ_INFO
401 #undef REQ_GROUP
402 #undef REQ_
403 };
405 static enum request
406 get_request(const char *name)
407 {
408 int namelen = strlen(name);
409 int i;
411 for (i = 0; i < ARRAY_SIZE(req_info); i++)
412 if (req_info[i].namelen == namelen &&
413 !string_enum_compare(req_info[i].name, name, namelen))
414 return req_info[i].request;
416 return REQ_NONE;
417 }
420 /*
421 * Options
422 */
424 static const char usage[] =
425 "tig " TIG_VERSION " (" __DATE__ ")\n"
426 "\n"
427 "Usage: tig [options] [revs] [--] [paths]\n"
428 " or: tig show [options] [revs] [--] [paths]\n"
429 " or: tig blame [rev] path\n"
430 " or: tig status\n"
431 " or: tig < [git command output]\n"
432 "\n"
433 "Options:\n"
434 " -v, --version Show version and exit\n"
435 " -h, --help Show help message and exit";
437 /* Option and state variables. */
438 static bool opt_date = TRUE;
439 static bool opt_author = TRUE;
440 static bool opt_line_number = FALSE;
441 static bool opt_rev_graph = FALSE;
442 static bool opt_show_refs = TRUE;
443 static int opt_num_interval = NUMBER_INTERVAL;
444 static int opt_tab_size = TABSIZE;
445 static enum request opt_request = REQ_VIEW_MAIN;
446 static char opt_cmd[SIZEOF_STR] = "";
447 static char opt_path[SIZEOF_STR] = "";
448 static char opt_file[SIZEOF_STR] = "";
449 static char opt_ref[SIZEOF_REF] = "";
450 static char opt_head[SIZEOF_REF] = "";
451 static char opt_remote[SIZEOF_REF] = "";
452 static bool opt_no_head = TRUE;
453 static FILE *opt_pipe = NULL;
454 static char opt_encoding[20] = "UTF-8";
455 static bool opt_utf8 = TRUE;
456 static char opt_codeset[20] = "UTF-8";
457 static iconv_t opt_iconv = ICONV_NONE;
458 static char opt_search[SIZEOF_STR] = "";
459 static char opt_cdup[SIZEOF_STR] = "";
460 static char opt_git_dir[SIZEOF_STR] = "";
461 static signed char opt_is_inside_work_tree = -1; /* set to TRUE or FALSE */
462 static char opt_editor[SIZEOF_STR] = "";
464 static bool
465 parse_options(int argc, char *argv[])
466 {
467 size_t buf_size;
468 char *subcommand;
469 bool seen_dashdash = FALSE;
470 int i;
472 if (!isatty(STDIN_FILENO)) {
473 opt_request = REQ_VIEW_PAGER;
474 opt_pipe = stdin;
475 return TRUE;
476 }
478 if (argc <= 1)
479 return TRUE;
481 subcommand = argv[1];
482 if (!strcmp(subcommand, "status") || !strcmp(subcommand, "-S")) {
483 opt_request = REQ_VIEW_STATUS;
484 if (!strcmp(subcommand, "-S"))
485 warn("`-S' has been deprecated; use `tig status' instead");
486 if (argc > 2)
487 warn("ignoring arguments after `%s'", subcommand);
488 return TRUE;
490 } else if (!strcmp(subcommand, "blame")) {
491 opt_request = REQ_VIEW_BLAME;
492 if (argc <= 2 || argc > 4)
493 die("invalid number of options to blame\n\n%s", usage);
495 i = 2;
496 if (argc == 4) {
497 string_ncopy(opt_ref, argv[i], strlen(argv[i]));
498 i++;
499 }
501 string_ncopy(opt_file, argv[i], strlen(argv[i]));
502 return TRUE;
504 } else if (!strcmp(subcommand, "show")) {
505 opt_request = REQ_VIEW_DIFF;
507 } else if (!strcmp(subcommand, "log") || !strcmp(subcommand, "diff")) {
508 opt_request = subcommand[0] == 'l'
509 ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
510 warn("`tig %s' has been deprecated", subcommand);
512 } else {
513 subcommand = NULL;
514 }
516 if (!subcommand)
517 /* XXX: This is vulnerable to the user overriding
518 * options required for the main view parser. */
519 string_copy(opt_cmd, "git log --no-color --pretty=raw --boundary --parents");
520 else
521 string_format(opt_cmd, "git %s", subcommand);
523 buf_size = strlen(opt_cmd);
525 for (i = 1 + !!subcommand; i < argc; i++) {
526 char *opt = argv[i];
528 if (seen_dashdash || !strcmp(opt, "--")) {
529 seen_dashdash = TRUE;
531 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
532 printf("tig version %s\n", TIG_VERSION);
533 return FALSE;
535 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
536 printf("%s\n", usage);
537 return FALSE;
538 }
540 opt_cmd[buf_size++] = ' ';
541 buf_size = sq_quote(opt_cmd, buf_size, opt);
542 if (buf_size >= sizeof(opt_cmd))
543 die("command too long");
544 }
546 opt_cmd[buf_size] = 0;
548 return TRUE;
549 }
552 /*
553 * Line-oriented content detection.
554 */
556 #define LINE_INFO \
557 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
558 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
559 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
560 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
561 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
562 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
563 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
564 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
565 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
566 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
567 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
568 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
569 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
570 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
571 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
572 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
573 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
574 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
575 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
576 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
577 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
578 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
579 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
580 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
581 LINE(AUTHOR, "author ", COLOR_CYAN, COLOR_DEFAULT, 0), \
582 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
583 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
584 LINE(ACKED, " Acked-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
585 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
586 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
587 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
588 LINE(DELIMITER, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
589 LINE(DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
590 LINE(LINE_NUMBER, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
591 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
592 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
593 LINE(MAIN_AUTHOR, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
594 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
595 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
596 LINE(MAIN_LOCAL_TAG,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
597 LINE(MAIN_REMOTE, "", COLOR_YELLOW, COLOR_DEFAULT, 0), \
598 LINE(MAIN_TRACKED, "", COLOR_YELLOW, COLOR_DEFAULT, A_BOLD), \
599 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
600 LINE(MAIN_HEAD, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
601 LINE(MAIN_REVGRAPH,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
602 LINE(TREE_DIR, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
603 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
604 LINE(STAT_HEAD, "", COLOR_YELLOW, COLOR_DEFAULT, 0), \
605 LINE(STAT_SECTION, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
606 LINE(STAT_NONE, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
607 LINE(STAT_STAGED, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
608 LINE(STAT_UNSTAGED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
609 LINE(STAT_UNTRACKED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
610 LINE(BLAME_AUTHOR, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
611 LINE(BLAME_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
612 LINE(BLAME_ID, "", COLOR_MAGENTA, COLOR_DEFAULT, 0)
614 enum line_type {
615 #define LINE(type, line, fg, bg, attr) \
616 LINE_##type
617 LINE_INFO
618 #undef LINE
619 };
621 struct line_info {
622 const char *name; /* Option name. */
623 int namelen; /* Size of option name. */
624 const char *line; /* The start of line to match. */
625 int linelen; /* Size of string to match. */
626 int fg, bg, attr; /* Color and text attributes for the lines. */
627 };
629 static struct line_info line_info[] = {
630 #define LINE(type, line, fg, bg, attr) \
631 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
632 LINE_INFO
633 #undef LINE
634 };
636 static enum line_type
637 get_line_type(char *line)
638 {
639 int linelen = strlen(line);
640 enum line_type type;
642 for (type = 0; type < ARRAY_SIZE(line_info); type++)
643 /* Case insensitive search matches Signed-off-by lines better. */
644 if (linelen >= line_info[type].linelen &&
645 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
646 return type;
648 return LINE_DEFAULT;
649 }
651 static inline int
652 get_line_attr(enum line_type type)
653 {
654 assert(type < ARRAY_SIZE(line_info));
655 return COLOR_PAIR(type) | line_info[type].attr;
656 }
658 static struct line_info *
659 get_line_info(char *name)
660 {
661 size_t namelen = strlen(name);
662 enum line_type type;
664 for (type = 0; type < ARRAY_SIZE(line_info); type++)
665 if (namelen == line_info[type].namelen &&
666 !string_enum_compare(line_info[type].name, name, namelen))
667 return &line_info[type];
669 return NULL;
670 }
672 static void
673 init_colors(void)
674 {
675 int default_bg = line_info[LINE_DEFAULT].bg;
676 int default_fg = line_info[LINE_DEFAULT].fg;
677 enum line_type type;
679 start_color();
681 if (assume_default_colors(default_fg, default_bg) == ERR) {
682 default_bg = COLOR_BLACK;
683 default_fg = COLOR_WHITE;
684 }
686 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
687 struct line_info *info = &line_info[type];
688 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
689 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
691 init_pair(type, fg, bg);
692 }
693 }
695 struct line {
696 enum line_type type;
698 /* State flags */
699 unsigned int selected:1;
700 unsigned int dirty:1;
702 void *data; /* User data */
703 };
706 /*
707 * Keys
708 */
710 struct keybinding {
711 int alias;
712 enum request request;
713 struct keybinding *next;
714 };
716 static struct keybinding default_keybindings[] = {
717 /* View switching */
718 { 'm', REQ_VIEW_MAIN },
719 { 'd', REQ_VIEW_DIFF },
720 { 'l', REQ_VIEW_LOG },
721 { 't', REQ_VIEW_TREE },
722 { 'f', REQ_VIEW_BLOB },
723 { 'B', REQ_VIEW_BLAME },
724 { 'p', REQ_VIEW_PAGER },
725 { 'h', REQ_VIEW_HELP },
726 { 'S', REQ_VIEW_STATUS },
727 { 'c', REQ_VIEW_STAGE },
729 /* View manipulation */
730 { 'q', REQ_VIEW_CLOSE },
731 { KEY_TAB, REQ_VIEW_NEXT },
732 { KEY_RETURN, REQ_ENTER },
733 { KEY_UP, REQ_PREVIOUS },
734 { KEY_DOWN, REQ_NEXT },
735 { 'R', REQ_REFRESH },
736 { 'M', REQ_MAXIMIZE },
738 /* Cursor navigation */
739 { 'k', REQ_MOVE_UP },
740 { 'j', REQ_MOVE_DOWN },
741 { KEY_HOME, REQ_MOVE_FIRST_LINE },
742 { KEY_END, REQ_MOVE_LAST_LINE },
743 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
744 { ' ', REQ_MOVE_PAGE_DOWN },
745 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
746 { 'b', REQ_MOVE_PAGE_UP },
747 { '-', REQ_MOVE_PAGE_UP },
749 /* Scrolling */
750 { KEY_IC, REQ_SCROLL_LINE_UP },
751 { KEY_DC, REQ_SCROLL_LINE_DOWN },
752 { 'w', REQ_SCROLL_PAGE_UP },
753 { 's', REQ_SCROLL_PAGE_DOWN },
755 /* Searching */
756 { '/', REQ_SEARCH },
757 { '?', REQ_SEARCH_BACK },
758 { 'n', REQ_FIND_NEXT },
759 { 'N', REQ_FIND_PREV },
761 /* Misc */
762 { 'Q', REQ_QUIT },
763 { 'z', REQ_STOP_LOADING },
764 { 'v', REQ_SHOW_VERSION },
765 { 'r', REQ_SCREEN_REDRAW },
766 { '.', REQ_TOGGLE_LINENO },
767 { 'D', REQ_TOGGLE_DATE },
768 { 'A', REQ_TOGGLE_AUTHOR },
769 { 'g', REQ_TOGGLE_REV_GRAPH },
770 { 'F', REQ_TOGGLE_REFS },
771 { ':', REQ_PROMPT },
772 { 'u', REQ_STATUS_UPDATE },
773 { 'M', REQ_STATUS_MERGE },
774 { ',', REQ_TREE_PARENT },
775 { 'e', REQ_EDIT },
777 /* Using the ncurses SIGWINCH handler. */
778 { KEY_RESIZE, REQ_SCREEN_RESIZE },
779 };
781 #define KEYMAP_INFO \
782 KEYMAP_(GENERIC), \
783 KEYMAP_(MAIN), \
784 KEYMAP_(DIFF), \
785 KEYMAP_(LOG), \
786 KEYMAP_(TREE), \
787 KEYMAP_(BLOB), \
788 KEYMAP_(BLAME), \
789 KEYMAP_(PAGER), \
790 KEYMAP_(HELP), \
791 KEYMAP_(STATUS), \
792 KEYMAP_(STAGE)
794 enum keymap {
795 #define KEYMAP_(name) KEYMAP_##name
796 KEYMAP_INFO
797 #undef KEYMAP_
798 };
800 static struct int_map keymap_table[] = {
801 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
802 KEYMAP_INFO
803 #undef KEYMAP_
804 };
806 #define set_keymap(map, name) \
807 set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
809 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
811 static void
812 add_keybinding(enum keymap keymap, enum request request, int key)
813 {
814 struct keybinding *keybinding;
816 keybinding = calloc(1, sizeof(*keybinding));
817 if (!keybinding)
818 die("Failed to allocate keybinding");
820 keybinding->alias = key;
821 keybinding->request = request;
822 keybinding->next = keybindings[keymap];
823 keybindings[keymap] = keybinding;
824 }
826 /* Looks for a key binding first in the given map, then in the generic map, and
827 * lastly in the default keybindings. */
828 static enum request
829 get_keybinding(enum keymap keymap, int key)
830 {
831 struct keybinding *kbd;
832 int i;
834 for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
835 if (kbd->alias == key)
836 return kbd->request;
838 for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
839 if (kbd->alias == key)
840 return kbd->request;
842 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
843 if (default_keybindings[i].alias == key)
844 return default_keybindings[i].request;
846 return (enum request) key;
847 }
850 struct key {
851 char *name;
852 int value;
853 };
855 static struct key key_table[] = {
856 { "Enter", KEY_RETURN },
857 { "Space", ' ' },
858 { "Backspace", KEY_BACKSPACE },
859 { "Tab", KEY_TAB },
860 { "Escape", KEY_ESC },
861 { "Left", KEY_LEFT },
862 { "Right", KEY_RIGHT },
863 { "Up", KEY_UP },
864 { "Down", KEY_DOWN },
865 { "Insert", KEY_IC },
866 { "Delete", KEY_DC },
867 { "Hash", '#' },
868 { "Home", KEY_HOME },
869 { "End", KEY_END },
870 { "PageUp", KEY_PPAGE },
871 { "PageDown", KEY_NPAGE },
872 { "F1", KEY_F(1) },
873 { "F2", KEY_F(2) },
874 { "F3", KEY_F(3) },
875 { "F4", KEY_F(4) },
876 { "F5", KEY_F(5) },
877 { "F6", KEY_F(6) },
878 { "F7", KEY_F(7) },
879 { "F8", KEY_F(8) },
880 { "F9", KEY_F(9) },
881 { "F10", KEY_F(10) },
882 { "F11", KEY_F(11) },
883 { "F12", KEY_F(12) },
884 };
886 static int
887 get_key_value(const char *name)
888 {
889 int i;
891 for (i = 0; i < ARRAY_SIZE(key_table); i++)
892 if (!strcasecmp(key_table[i].name, name))
893 return key_table[i].value;
895 if (strlen(name) == 1 && isprint(*name))
896 return (int) *name;
898 return ERR;
899 }
901 static char *
902 get_key_name(int key_value)
903 {
904 static char key_char[] = "'X'";
905 char *seq = NULL;
906 int key;
908 for (key = 0; key < ARRAY_SIZE(key_table); key++)
909 if (key_table[key].value == key_value)
910 seq = key_table[key].name;
912 if (seq == NULL &&
913 key_value < 127 &&
914 isprint(key_value)) {
915 key_char[1] = (char) key_value;
916 seq = key_char;
917 }
919 return seq ? seq : "'?'";
920 }
922 static char *
923 get_key(enum request request)
924 {
925 static char buf[BUFSIZ];
926 size_t pos = 0;
927 char *sep = "";
928 int i;
930 buf[pos] = 0;
932 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
933 struct keybinding *keybinding = &default_keybindings[i];
935 if (keybinding->request != request)
936 continue;
938 if (!string_format_from(buf, &pos, "%s%s", sep,
939 get_key_name(keybinding->alias)))
940 return "Too many keybindings!";
941 sep = ", ";
942 }
944 return buf;
945 }
947 struct run_request {
948 enum keymap keymap;
949 int key;
950 char cmd[SIZEOF_STR];
951 };
953 static struct run_request *run_request;
954 static size_t run_requests;
956 static enum request
957 add_run_request(enum keymap keymap, int key, int argc, char **argv)
958 {
959 struct run_request *tmp;
960 struct run_request req = { keymap, key };
961 size_t bufpos;
963 for (bufpos = 0; argc > 0; argc--, argv++)
964 if (!string_format_from(req.cmd, &bufpos, "%s ", *argv))
965 return REQ_NONE;
967 req.cmd[bufpos - 1] = 0;
969 tmp = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
970 if (!tmp)
971 return REQ_NONE;
973 run_request = tmp;
974 run_request[run_requests++] = req;
976 return REQ_NONE + run_requests;
977 }
979 static struct run_request *
980 get_run_request(enum request request)
981 {
982 if (request <= REQ_NONE)
983 return NULL;
984 return &run_request[request - REQ_NONE - 1];
985 }
987 static void
988 add_builtin_run_requests(void)
989 {
990 struct {
991 enum keymap keymap;
992 int key;
993 char *argv[1];
994 } reqs[] = {
995 { KEYMAP_MAIN, 'C', { "git cherry-pick %(commit)" } },
996 { KEYMAP_GENERIC, 'G', { "git gc" } },
997 };
998 int i;
1000 for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1001 enum request req;
1003 req = add_run_request(reqs[i].keymap, reqs[i].key, 1, reqs[i].argv);
1004 if (req != REQ_NONE)
1005 add_keybinding(reqs[i].keymap, req, reqs[i].key);
1006 }
1007 }
1009 /*
1010 * User config file handling.
1011 */
1013 static struct int_map color_map[] = {
1014 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
1015 COLOR_MAP(DEFAULT),
1016 COLOR_MAP(BLACK),
1017 COLOR_MAP(BLUE),
1018 COLOR_MAP(CYAN),
1019 COLOR_MAP(GREEN),
1020 COLOR_MAP(MAGENTA),
1021 COLOR_MAP(RED),
1022 COLOR_MAP(WHITE),
1023 COLOR_MAP(YELLOW),
1024 };
1026 #define set_color(color, name) \
1027 set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
1029 static struct int_map attr_map[] = {
1030 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
1031 ATTR_MAP(NORMAL),
1032 ATTR_MAP(BLINK),
1033 ATTR_MAP(BOLD),
1034 ATTR_MAP(DIM),
1035 ATTR_MAP(REVERSE),
1036 ATTR_MAP(STANDOUT),
1037 ATTR_MAP(UNDERLINE),
1038 };
1040 #define set_attribute(attr, name) \
1041 set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
1043 static int config_lineno;
1044 static bool config_errors;
1045 static char *config_msg;
1047 /* Wants: object fgcolor bgcolor [attr] */
1048 static int
1049 option_color_command(int argc, char *argv[])
1050 {
1051 struct line_info *info;
1053 if (argc != 3 && argc != 4) {
1054 config_msg = "Wrong number of arguments given to color command";
1055 return ERR;
1056 }
1058 info = get_line_info(argv[0]);
1059 if (!info) {
1060 if (!string_enum_compare(argv[0], "main-delim", strlen("main-delim"))) {
1061 info = get_line_info("delimiter");
1063 } else if (!string_enum_compare(argv[0], "main-date", strlen("main-date"))) {
1064 info = get_line_info("date");
1066 } else {
1067 config_msg = "Unknown color name";
1068 return ERR;
1069 }
1070 }
1072 if (set_color(&info->fg, argv[1]) == ERR ||
1073 set_color(&info->bg, argv[2]) == ERR) {
1074 config_msg = "Unknown color";
1075 return ERR;
1076 }
1078 if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
1079 config_msg = "Unknown attribute";
1080 return ERR;
1081 }
1083 return OK;
1084 }
1086 static bool parse_bool(const char *s)
1087 {
1088 return (!strcmp(s, "1") || !strcmp(s, "true") ||
1089 !strcmp(s, "yes")) ? TRUE : FALSE;
1090 }
1092 /* Wants: name = value */
1093 static int
1094 option_set_command(int argc, char *argv[])
1095 {
1096 if (argc != 3) {
1097 config_msg = "Wrong number of arguments given to set command";
1098 return ERR;
1099 }
1101 if (strcmp(argv[1], "=")) {
1102 config_msg = "No value assigned";
1103 return ERR;
1104 }
1106 if (!strcmp(argv[0], "show-author")) {
1107 opt_author = parse_bool(argv[2]);
1108 return OK;
1109 }
1111 if (!strcmp(argv[0], "show-date")) {
1112 opt_date = parse_bool(argv[2]);
1113 return OK;
1114 }
1116 if (!strcmp(argv[0], "show-rev-graph")) {
1117 opt_rev_graph = parse_bool(argv[2]);
1118 return OK;
1119 }
1121 if (!strcmp(argv[0], "show-refs")) {
1122 opt_show_refs = parse_bool(argv[2]);
1123 return OK;
1124 }
1126 if (!strcmp(argv[0], "show-line-numbers")) {
1127 opt_line_number = parse_bool(argv[2]);
1128 return OK;
1129 }
1131 if (!strcmp(argv[0], "line-number-interval")) {
1132 opt_num_interval = atoi(argv[2]);
1133 return OK;
1134 }
1136 if (!strcmp(argv[0], "tab-size")) {
1137 opt_tab_size = atoi(argv[2]);
1138 return OK;
1139 }
1141 if (!strcmp(argv[0], "commit-encoding")) {
1142 char *arg = argv[2];
1143 int delimiter = *arg;
1144 int i;
1146 switch (delimiter) {
1147 case '"':
1148 case '\'':
1149 for (arg++, i = 0; arg[i]; i++)
1150 if (arg[i] == delimiter) {
1151 arg[i] = 0;
1152 break;
1153 }
1154 default:
1155 string_ncopy(opt_encoding, arg, strlen(arg));
1156 return OK;
1157 }
1158 }
1160 config_msg = "Unknown variable name";
1161 return ERR;
1162 }
1164 /* Wants: mode request key */
1165 static int
1166 option_bind_command(int argc, char *argv[])
1167 {
1168 enum request request;
1169 int keymap;
1170 int key;
1172 if (argc < 3) {
1173 config_msg = "Wrong number of arguments given to bind command";
1174 return ERR;
1175 }
1177 if (set_keymap(&keymap, argv[0]) == ERR) {
1178 config_msg = "Unknown key map";
1179 return ERR;
1180 }
1182 key = get_key_value(argv[1]);
1183 if (key == ERR) {
1184 config_msg = "Unknown key";
1185 return ERR;
1186 }
1188 request = get_request(argv[2]);
1189 if (request == REQ_NONE) {
1190 const char *obsolete[] = { "cherry-pick" };
1191 size_t namelen = strlen(argv[2]);
1192 int i;
1194 for (i = 0; i < ARRAY_SIZE(obsolete); i++) {
1195 if (namelen == strlen(obsolete[i]) &&
1196 !string_enum_compare(obsolete[i], argv[2], namelen)) {
1197 config_msg = "Obsolete request name";
1198 return ERR;
1199 }
1200 }
1201 }
1202 if (request == REQ_NONE && *argv[2]++ == '!')
1203 request = add_run_request(keymap, key, argc - 2, argv + 2);
1204 if (request == REQ_NONE) {
1205 config_msg = "Unknown request name";
1206 return ERR;
1207 }
1209 add_keybinding(keymap, request, key);
1211 return OK;
1212 }
1214 static int
1215 set_option(char *opt, char *value)
1216 {
1217 char *argv[16];
1218 int valuelen;
1219 int argc = 0;
1221 /* Tokenize */
1222 while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1223 argv[argc++] = value;
1224 value += valuelen;
1226 /* Nothing more to tokenize or last available token. */
1227 if (!*value || argc >= ARRAY_SIZE(argv))
1228 break;
1230 *value++ = 0;
1231 while (isspace(*value))
1232 value++;
1233 }
1235 if (!strcmp(opt, "color"))
1236 return option_color_command(argc, argv);
1238 if (!strcmp(opt, "set"))
1239 return option_set_command(argc, argv);
1241 if (!strcmp(opt, "bind"))
1242 return option_bind_command(argc, argv);
1244 config_msg = "Unknown option command";
1245 return ERR;
1246 }
1248 static int
1249 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1250 {
1251 int status = OK;
1253 config_lineno++;
1254 config_msg = "Internal error";
1256 /* Check for comment markers, since read_properties() will
1257 * only ensure opt and value are split at first " \t". */
1258 optlen = strcspn(opt, "#");
1259 if (optlen == 0)
1260 return OK;
1262 if (opt[optlen] != 0) {
1263 config_msg = "No option value";
1264 status = ERR;
1266 } else {
1267 /* Look for comment endings in the value. */
1268 size_t len = strcspn(value, "#");
1270 if (len < valuelen) {
1271 valuelen = len;
1272 value[valuelen] = 0;
1273 }
1275 status = set_option(opt, value);
1276 }
1278 if (status == ERR) {
1279 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1280 config_lineno, (int) optlen, opt, config_msg);
1281 config_errors = TRUE;
1282 }
1284 /* Always keep going if errors are encountered. */
1285 return OK;
1286 }
1288 static void
1289 load_option_file(const char *path)
1290 {
1291 FILE *file;
1293 /* It's ok that the file doesn't exist. */
1294 file = fopen(path, "r");
1295 if (!file)
1296 return;
1298 config_lineno = 0;
1299 config_errors = FALSE;
1301 if (read_properties(file, " \t", read_option) == ERR ||
1302 config_errors == TRUE)
1303 fprintf(stderr, "Errors while loading %s.\n", path);
1304 }
1306 static int
1307 load_options(void)
1308 {
1309 char *home = getenv("HOME");
1310 char *tigrc_user = getenv("TIGRC_USER");
1311 char *tigrc_system = getenv("TIGRC_SYSTEM");
1312 char buf[SIZEOF_STR];
1314 add_builtin_run_requests();
1316 if (!tigrc_system) {
1317 if (!string_format(buf, "%s/tigrc", SYSCONFDIR))
1318 return ERR;
1319 tigrc_system = buf;
1320 }
1321 load_option_file(tigrc_system);
1323 if (!tigrc_user) {
1324 if (!home || !string_format(buf, "%s/.tigrc", home))
1325 return ERR;
1326 tigrc_user = buf;
1327 }
1328 load_option_file(tigrc_user);
1330 return OK;
1331 }
1334 /*
1335 * The viewer
1336 */
1338 struct view;
1339 struct view_ops;
1341 /* The display array of active views and the index of the current view. */
1342 static struct view *display[2];
1343 static unsigned int current_view;
1345 /* Reading from the prompt? */
1346 static bool input_mode = FALSE;
1348 #define foreach_displayed_view(view, i) \
1349 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1351 #define displayed_views() (display[1] != NULL ? 2 : 1)
1353 /* Current head and commit ID */
1354 static char ref_blob[SIZEOF_REF] = "";
1355 static char ref_commit[SIZEOF_REF] = "HEAD";
1356 static char ref_head[SIZEOF_REF] = "HEAD";
1358 struct view {
1359 const char *name; /* View name */
1360 const char *cmd_fmt; /* Default command line format */
1361 const char *cmd_env; /* Command line set via environment */
1362 const char *id; /* Points to either of ref_{head,commit,blob} */
1364 struct view_ops *ops; /* View operations */
1366 enum keymap keymap; /* What keymap does this view have */
1367 bool git_dir; /* Whether the view requires a git directory. */
1369 char cmd[SIZEOF_STR]; /* Command buffer */
1370 char ref[SIZEOF_REF]; /* Hovered commit reference */
1371 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
1373 int height, width; /* The width and height of the main window */
1374 WINDOW *win; /* The main window */
1375 WINDOW *title; /* The title window living below the main window */
1377 /* Navigation */
1378 unsigned long offset; /* Offset of the window top */
1379 unsigned long lineno; /* Current line number */
1381 /* Searching */
1382 char grep[SIZEOF_STR]; /* Search string */
1383 regex_t *regex; /* Pre-compiled regex */
1385 /* If non-NULL, points to the view that opened this view. If this view
1386 * is closed tig will switch back to the parent view. */
1387 struct view *parent;
1389 /* Buffering */
1390 size_t lines; /* Total number of lines */
1391 struct line *line; /* Line index */
1392 size_t line_alloc; /* Total number of allocated lines */
1393 size_t line_size; /* Total number of used lines */
1394 unsigned int digits; /* Number of digits in the lines member. */
1396 /* Loading */
1397 FILE *pipe;
1398 time_t start_time;
1399 };
1401 struct view_ops {
1402 /* What type of content being displayed. Used in the title bar. */
1403 const char *type;
1404 /* Open and reads in all view content. */
1405 bool (*open)(struct view *view);
1406 /* Read one line; updates view->line. */
1407 bool (*read)(struct view *view, char *data);
1408 /* Draw one line; @lineno must be < view->height. */
1409 bool (*draw)(struct view *view, struct line *line, unsigned int lineno, bool selected);
1410 /* Depending on view handle a special requests. */
1411 enum request (*request)(struct view *view, enum request request, struct line *line);
1412 /* Search for regex in a line. */
1413 bool (*grep)(struct view *view, struct line *line);
1414 /* Select line */
1415 void (*select)(struct view *view, struct line *line);
1416 };
1418 static struct view_ops pager_ops;
1419 static struct view_ops main_ops;
1420 static struct view_ops tree_ops;
1421 static struct view_ops blob_ops;
1422 static struct view_ops blame_ops;
1423 static struct view_ops help_ops;
1424 static struct view_ops status_ops;
1425 static struct view_ops stage_ops;
1427 #define VIEW_STR(name, cmd, env, ref, ops, map, git) \
1428 { name, cmd, #env, ref, ops, map, git }
1430 #define VIEW_(id, name, ops, git, ref) \
1431 VIEW_STR(name, TIG_##id##_CMD, TIG_##id##_CMD, ref, ops, KEYMAP_##id, git)
1434 static struct view views[] = {
1435 VIEW_(MAIN, "main", &main_ops, TRUE, ref_head),
1436 VIEW_(DIFF, "diff", &pager_ops, TRUE, ref_commit),
1437 VIEW_(LOG, "log", &pager_ops, TRUE, ref_head),
1438 VIEW_(TREE, "tree", &tree_ops, TRUE, ref_commit),
1439 VIEW_(BLOB, "blob", &blob_ops, TRUE, ref_blob),
1440 VIEW_(BLAME, "blame", &blame_ops, TRUE, ref_commit),
1441 VIEW_(HELP, "help", &help_ops, FALSE, ""),
1442 VIEW_(PAGER, "pager", &pager_ops, FALSE, "stdin"),
1443 VIEW_(STATUS, "status", &status_ops, TRUE, ""),
1444 VIEW_(STAGE, "stage", &stage_ops, TRUE, ""),
1445 };
1447 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1448 #define VIEW_REQ(view) ((view) - views + REQ_OFFSET + 1)
1450 #define foreach_view(view, i) \
1451 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1453 #define view_is_displayed(view) \
1454 (view == display[0] || view == display[1])
1456 static int
1457 draw_text(struct view *view, const char *string, int max_len,
1458 bool use_tilde, bool selected)
1459 {
1460 int len = 0;
1461 int trimmed = FALSE;
1463 if (max_len <= 0)
1464 return 0;
1466 if (opt_utf8) {
1467 len = utf8_length(string, max_len, &trimmed, use_tilde);
1468 } else {
1469 len = strlen(string);
1470 if (len > max_len) {
1471 if (use_tilde) {
1472 max_len -= 1;
1473 }
1474 len = max_len;
1475 trimmed = TRUE;
1476 }
1477 }
1479 waddnstr(view->win, string, len);
1480 if (trimmed && use_tilde) {
1481 if (!selected)
1482 wattrset(view->win, get_line_attr(LINE_DELIMITER));
1483 waddch(view->win, '~');
1484 len++;
1485 }
1487 return len;
1488 }
1490 static int
1491 draw_lineno(struct view *view, unsigned int lineno, int max, bool selected)
1492 {
1493 static char fmt[] = "%1ld";
1494 char number[10] = " ";
1495 int digits3 = view->digits < 3 ? 3 : view->digits;
1496 int max_number = MIN(digits3, STRING_SIZE(number));
1497 bool showtrimmed = FALSE;
1498 int col;
1500 lineno += view->offset + 1;
1501 if (lineno == 1 || (lineno % opt_num_interval) == 0) {
1502 if (view->digits <= 9)
1503 fmt[1] = '0' + digits3;
1505 if (!string_format(number, fmt, lineno))
1506 number[0] = 0;
1507 showtrimmed = TRUE;
1508 }
1510 if (max < max_number)
1511 max_number = max;
1513 if (!selected)
1514 wattrset(view->win, get_line_attr(LINE_LINE_NUMBER));
1515 col = draw_text(view, number, max_number, showtrimmed, selected);
1516 if (col < max) {
1517 if (!selected)
1518 wattrset(view->win, A_NORMAL);
1519 waddch(view->win, ACS_VLINE);
1520 col++;
1521 }
1522 if (col < max) {
1523 waddch(view->win, ' ');
1524 col++;
1525 }
1527 return col;
1528 }
1530 static int
1531 draw_date(struct view *view, struct tm *time, int max, bool selected)
1532 {
1533 char buf[DATE_COLS];
1534 int col;
1535 int timelen = 0;
1537 if (max > DATE_COLS)
1538 max = DATE_COLS;
1539 if (time)
1540 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, time);
1541 if (!timelen) {
1542 memset(buf, ' ', sizeof(buf) - 1);
1543 buf[sizeof(buf) - 1] = 0;
1544 }
1546 if (!selected)
1547 wattrset(view->win, get_line_attr(LINE_DATE));
1548 col = draw_text(view, buf, max, FALSE, selected);
1549 if (col < max) {
1550 if (!selected)
1551 wattrset(view->win, get_line_attr(LINE_DEFAULT));
1552 waddch(view->win, ' ');
1553 col++;
1554 }
1556 return col;
1557 }
1559 static bool
1560 draw_view_line(struct view *view, unsigned int lineno)
1561 {
1562 struct line *line;
1563 bool selected = (view->offset + lineno == view->lineno);
1564 bool draw_ok;
1566 assert(view_is_displayed(view));
1568 if (view->offset + lineno >= view->lines)
1569 return FALSE;
1571 line = &view->line[view->offset + lineno];
1573 wmove(view->win, lineno, 0);
1575 if (selected) {
1576 line->selected = TRUE;
1577 view->ops->select(view, line);
1578 wchgat(view->win, -1, 0, LINE_CURSOR, NULL);
1579 wattrset(view->win, get_line_attr(LINE_CURSOR));
1580 } else if (line->selected) {
1581 line->selected = FALSE;
1582 wclrtoeol(view->win);
1583 }
1585 scrollok(view->win, FALSE);
1586 draw_ok = view->ops->draw(view, line, lineno, selected);
1587 scrollok(view->win, TRUE);
1589 return draw_ok;
1590 }
1592 static void
1593 redraw_view_dirty(struct view *view)
1594 {
1595 bool dirty = FALSE;
1596 int lineno;
1598 for (lineno = 0; lineno < view->height; lineno++) {
1599 struct line *line = &view->line[view->offset + lineno];
1601 if (!line->dirty)
1602 continue;
1603 line->dirty = 0;
1604 dirty = TRUE;
1605 if (!draw_view_line(view, lineno))
1606 break;
1607 }
1609 if (!dirty)
1610 return;
1611 redrawwin(view->win);
1612 if (input_mode)
1613 wnoutrefresh(view->win);
1614 else
1615 wrefresh(view->win);
1616 }
1618 static void
1619 redraw_view_from(struct view *view, int lineno)
1620 {
1621 assert(0 <= lineno && lineno < view->height);
1623 for (; lineno < view->height; lineno++) {
1624 if (!draw_view_line(view, lineno))
1625 break;
1626 }
1628 redrawwin(view->win);
1629 if (input_mode)
1630 wnoutrefresh(view->win);
1631 else
1632 wrefresh(view->win);
1633 }
1635 static void
1636 redraw_view(struct view *view)
1637 {
1638 wclear(view->win);
1639 redraw_view_from(view, 0);
1640 }
1643 static void
1644 update_view_title(struct view *view)
1645 {
1646 char buf[SIZEOF_STR];
1647 char state[SIZEOF_STR];
1648 size_t bufpos = 0, statelen = 0;
1650 assert(view_is_displayed(view));
1652 if (view != VIEW(REQ_VIEW_STATUS) && (view->lines || view->pipe)) {
1653 unsigned int view_lines = view->offset + view->height;
1654 unsigned int lines = view->lines
1655 ? MIN(view_lines, view->lines) * 100 / view->lines
1656 : 0;
1658 string_format_from(state, &statelen, "- %s %d of %d (%d%%)",
1659 view->ops->type,
1660 view->lineno + 1,
1661 view->lines,
1662 lines);
1664 if (view->pipe) {
1665 time_t secs = time(NULL) - view->start_time;
1667 /* Three git seconds are a long time ... */
1668 if (secs > 2)
1669 string_format_from(state, &statelen, " %lds", secs);
1670 }
1671 }
1673 string_format_from(buf, &bufpos, "[%s]", view->name);
1674 if (*view->ref && bufpos < view->width) {
1675 size_t refsize = strlen(view->ref);
1676 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
1678 if (minsize < view->width)
1679 refsize = view->width - minsize + 7;
1680 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
1681 }
1683 if (statelen && bufpos < view->width) {
1684 string_format_from(buf, &bufpos, " %s", state);
1685 }
1687 if (view == display[current_view])
1688 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1689 else
1690 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1692 mvwaddnstr(view->title, 0, 0, buf, bufpos);
1693 wclrtoeol(view->title);
1694 wmove(view->title, 0, view->width - 1);
1696 if (input_mode)
1697 wnoutrefresh(view->title);
1698 else
1699 wrefresh(view->title);
1700 }
1702 static void
1703 resize_display(void)
1704 {
1705 int offset, i;
1706 struct view *base = display[0];
1707 struct view *view = display[1] ? display[1] : display[0];
1709 /* Setup window dimensions */
1711 getmaxyx(stdscr, base->height, base->width);
1713 /* Make room for the status window. */
1714 base->height -= 1;
1716 if (view != base) {
1717 /* Horizontal split. */
1718 view->width = base->width;
1719 view->height = SCALE_SPLIT_VIEW(base->height);
1720 base->height -= view->height;
1722 /* Make room for the title bar. */
1723 view->height -= 1;
1724 }
1726 /* Make room for the title bar. */
1727 base->height -= 1;
1729 offset = 0;
1731 foreach_displayed_view (view, i) {
1732 if (!view->win) {
1733 view->win = newwin(view->height, 0, offset, 0);
1734 if (!view->win)
1735 die("Failed to create %s view", view->name);
1737 scrollok(view->win, TRUE);
1739 view->title = newwin(1, 0, offset + view->height, 0);
1740 if (!view->title)
1741 die("Failed to create title window");
1743 } else {
1744 wresize(view->win, view->height, view->width);
1745 mvwin(view->win, offset, 0);
1746 mvwin(view->title, offset + view->height, 0);
1747 }
1749 offset += view->height + 1;
1750 }
1751 }
1753 static void
1754 redraw_display(void)
1755 {
1756 struct view *view;
1757 int i;
1759 foreach_displayed_view (view, i) {
1760 redraw_view(view);
1761 update_view_title(view);
1762 }
1763 }
1765 static void
1766 update_display_cursor(struct view *view)
1767 {
1768 /* Move the cursor to the right-most column of the cursor line.
1769 *
1770 * XXX: This could turn out to be a bit expensive, but it ensures that
1771 * the cursor does not jump around. */
1772 if (view->lines) {
1773 wmove(view->win, view->lineno - view->offset, view->width - 1);
1774 wrefresh(view->win);
1775 }
1776 }
1778 /*
1779 * Navigation
1780 */
1782 /* Scrolling backend */
1783 static void
1784 do_scroll_view(struct view *view, int lines)
1785 {
1786 bool redraw_current_line = FALSE;
1788 /* The rendering expects the new offset. */
1789 view->offset += lines;
1791 assert(0 <= view->offset && view->offset < view->lines);
1792 assert(lines);
1794 /* Move current line into the view. */
1795 if (view->lineno < view->offset) {
1796 view->lineno = view->offset;
1797 redraw_current_line = TRUE;
1798 } else if (view->lineno >= view->offset + view->height) {
1799 view->lineno = view->offset + view->height - 1;
1800 redraw_current_line = TRUE;
1801 }
1803 assert(view->offset <= view->lineno && view->lineno < view->lines);
1805 /* Redraw the whole screen if scrolling is pointless. */
1806 if (view->height < ABS(lines)) {
1807 redraw_view(view);
1809 } else {
1810 int line = lines > 0 ? view->height - lines : 0;
1811 int end = line + ABS(lines);
1813 wscrl(view->win, lines);
1815 for (; line < end; line++) {
1816 if (!draw_view_line(view, line))
1817 break;
1818 }
1820 if (redraw_current_line)
1821 draw_view_line(view, view->lineno - view->offset);
1822 }
1824 redrawwin(view->win);
1825 wrefresh(view->win);
1826 report("");
1827 }
1829 /* Scroll frontend */
1830 static void
1831 scroll_view(struct view *view, enum request request)
1832 {
1833 int lines = 1;
1835 assert(view_is_displayed(view));
1837 switch (request) {
1838 case REQ_SCROLL_PAGE_DOWN:
1839 lines = view->height;
1840 case REQ_SCROLL_LINE_DOWN:
1841 if (view->offset + lines > view->lines)
1842 lines = view->lines - view->offset;
1844 if (lines == 0 || view->offset + view->height >= view->lines) {
1845 report("Cannot scroll beyond the last line");
1846 return;
1847 }
1848 break;
1850 case REQ_SCROLL_PAGE_UP:
1851 lines = view->height;
1852 case REQ_SCROLL_LINE_UP:
1853 if (lines > view->offset)
1854 lines = view->offset;
1856 if (lines == 0) {
1857 report("Cannot scroll beyond the first line");
1858 return;
1859 }
1861 lines = -lines;
1862 break;
1864 default:
1865 die("request %d not handled in switch", request);
1866 }
1868 do_scroll_view(view, lines);
1869 }
1871 /* Cursor moving */
1872 static void
1873 move_view(struct view *view, enum request request)
1874 {
1875 int scroll_steps = 0;
1876 int steps;
1878 switch (request) {
1879 case REQ_MOVE_FIRST_LINE:
1880 steps = -view->lineno;
1881 break;
1883 case REQ_MOVE_LAST_LINE:
1884 steps = view->lines - view->lineno - 1;
1885 break;
1887 case REQ_MOVE_PAGE_UP:
1888 steps = view->height > view->lineno
1889 ? -view->lineno : -view->height;
1890 break;
1892 case REQ_MOVE_PAGE_DOWN:
1893 steps = view->lineno + view->height >= view->lines
1894 ? view->lines - view->lineno - 1 : view->height;
1895 break;
1897 case REQ_MOVE_UP:
1898 steps = -1;
1899 break;
1901 case REQ_MOVE_DOWN:
1902 steps = 1;
1903 break;
1905 default:
1906 die("request %d not handled in switch", request);
1907 }
1909 if (steps <= 0 && view->lineno == 0) {
1910 report("Cannot move beyond the first line");
1911 return;
1913 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1914 report("Cannot move beyond the last line");
1915 return;
1916 }
1918 /* Move the current line */
1919 view->lineno += steps;
1920 assert(0 <= view->lineno && view->lineno < view->lines);
1922 /* Check whether the view needs to be scrolled */
1923 if (view->lineno < view->offset ||
1924 view->lineno >= view->offset + view->height) {
1925 scroll_steps = steps;
1926 if (steps < 0 && -steps > view->offset) {
1927 scroll_steps = -view->offset;
1929 } else if (steps > 0) {
1930 if (view->lineno == view->lines - 1 &&
1931 view->lines > view->height) {
1932 scroll_steps = view->lines - view->offset - 1;
1933 if (scroll_steps >= view->height)
1934 scroll_steps -= view->height - 1;
1935 }
1936 }
1937 }
1939 if (!view_is_displayed(view)) {
1940 view->offset += scroll_steps;
1941 assert(0 <= view->offset && view->offset < view->lines);
1942 view->ops->select(view, &view->line[view->lineno]);
1943 return;
1944 }
1946 /* Repaint the old "current" line if we be scrolling */
1947 if (ABS(steps) < view->height)
1948 draw_view_line(view, view->lineno - steps - view->offset);
1950 if (scroll_steps) {
1951 do_scroll_view(view, scroll_steps);
1952 return;
1953 }
1955 /* Draw the current line */
1956 draw_view_line(view, view->lineno - view->offset);
1958 redrawwin(view->win);
1959 wrefresh(view->win);
1960 report("");
1961 }
1964 /*
1965 * Searching
1966 */
1968 static void search_view(struct view *view, enum request request);
1970 static bool
1971 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1972 {
1973 assert(view_is_displayed(view));
1975 if (!view->ops->grep(view, line))
1976 return FALSE;
1978 if (lineno - view->offset >= view->height) {
1979 view->offset = lineno;
1980 view->lineno = lineno;
1981 redraw_view(view);
1983 } else {
1984 unsigned long old_lineno = view->lineno - view->offset;
1986 view->lineno = lineno;
1987 draw_view_line(view, old_lineno);
1989 draw_view_line(view, view->lineno - view->offset);
1990 redrawwin(view->win);
1991 wrefresh(view->win);
1992 }
1994 report("Line %ld matches '%s'", lineno + 1, view->grep);
1995 return TRUE;
1996 }
1998 static void
1999 find_next(struct view *view, enum request request)
2000 {
2001 unsigned long lineno = view->lineno;
2002 int direction;
2004 if (!*view->grep) {
2005 if (!*opt_search)
2006 report("No previous search");
2007 else
2008 search_view(view, request);
2009 return;
2010 }
2012 switch (request) {
2013 case REQ_SEARCH:
2014 case REQ_FIND_NEXT:
2015 direction = 1;
2016 break;
2018 case REQ_SEARCH_BACK:
2019 case REQ_FIND_PREV:
2020 direction = -1;
2021 break;
2023 default:
2024 return;
2025 }
2027 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
2028 lineno += direction;
2030 /* Note, lineno is unsigned long so will wrap around in which case it
2031 * will become bigger than view->lines. */
2032 for (; lineno < view->lines; lineno += direction) {
2033 struct line *line = &view->line[lineno];
2035 if (find_next_line(view, lineno, line))
2036 return;
2037 }
2039 report("No match found for '%s'", view->grep);
2040 }
2042 static void
2043 search_view(struct view *view, enum request request)
2044 {
2045 int regex_err;
2047 if (view->regex) {
2048 regfree(view->regex);
2049 *view->grep = 0;
2050 } else {
2051 view->regex = calloc(1, sizeof(*view->regex));
2052 if (!view->regex)
2053 return;
2054 }
2056 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
2057 if (regex_err != 0) {
2058 char buf[SIZEOF_STR] = "unknown error";
2060 regerror(regex_err, view->regex, buf, sizeof(buf));
2061 report("Search failed: %s", buf);
2062 return;
2063 }
2065 string_copy(view->grep, opt_search);
2067 find_next(view, request);
2068 }
2070 /*
2071 * Incremental updating
2072 */
2074 static void
2075 end_update(struct view *view)
2076 {
2077 if (!view->pipe)
2078 return;
2079 set_nonblocking_input(FALSE);
2080 if (view->pipe == stdin)
2081 fclose(view->pipe);
2082 else
2083 pclose(view->pipe);
2084 view->pipe = NULL;
2085 }
2087 static bool
2088 begin_update(struct view *view)
2089 {
2090 if (view->pipe)
2091 end_update(view);
2093 if (opt_cmd[0]) {
2094 string_copy(view->cmd, opt_cmd);
2095 opt_cmd[0] = 0;
2096 /* When running random commands, initially show the
2097 * command in the title. However, it maybe later be
2098 * overwritten if a commit line is selected. */
2099 if (view == VIEW(REQ_VIEW_PAGER))
2100 string_copy(view->ref, view->cmd);
2101 else
2102 view->ref[0] = 0;
2104 } else if (view == VIEW(REQ_VIEW_TREE)) {
2105 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2106 char path[SIZEOF_STR];
2108 if (strcmp(view->vid, view->id))
2109 opt_path[0] = path[0] = 0;
2110 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
2111 return FALSE;
2113 if (!string_format(view->cmd, format, view->id, path))
2114 return FALSE;
2116 } else {
2117 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2118 const char *id = view->id;
2120 if (!string_format(view->cmd, format, id, id, id, id, id))
2121 return FALSE;
2123 /* Put the current ref_* value to the view title ref
2124 * member. This is needed by the blob view. Most other
2125 * views sets it automatically after loading because the
2126 * first line is a commit line. */
2127 string_copy_rev(view->ref, view->id);
2128 }
2130 /* Special case for the pager view. */
2131 if (opt_pipe) {
2132 view->pipe = opt_pipe;
2133 opt_pipe = NULL;
2134 } else {
2135 view->pipe = popen(view->cmd, "r");
2136 }
2138 if (!view->pipe)
2139 return FALSE;
2141 set_nonblocking_input(TRUE);
2143 view->offset = 0;
2144 view->lines = 0;
2145 view->lineno = 0;
2146 string_copy_rev(view->vid, view->id);
2148 if (view->line) {
2149 int i;
2151 for (i = 0; i < view->lines; i++)
2152 if (view->line[i].data)
2153 free(view->line[i].data);
2155 free(view->line);
2156 view->line = NULL;
2157 }
2159 view->start_time = time(NULL);
2161 return TRUE;
2162 }
2164 #define ITEM_CHUNK_SIZE 256
2165 static void *
2166 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2167 {
2168 size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2169 size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2171 if (mem == NULL || num_chunks != num_chunks_new) {
2172 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2173 mem = realloc(mem, *size * item_size);
2174 }
2176 return mem;
2177 }
2179 static struct line *
2180 realloc_lines(struct view *view, size_t line_size)
2181 {
2182 size_t alloc = view->line_alloc;
2183 struct line *tmp = realloc_items(view->line, &alloc, line_size,
2184 sizeof(*view->line));
2186 if (!tmp)
2187 return NULL;
2189 view->line = tmp;
2190 view->line_alloc = alloc;
2191 view->line_size = line_size;
2192 return view->line;
2193 }
2195 static bool
2196 update_view(struct view *view)
2197 {
2198 char in_buffer[BUFSIZ];
2199 char out_buffer[BUFSIZ * 2];
2200 char *line;
2201 /* The number of lines to read. If too low it will cause too much
2202 * redrawing (and possible flickering), if too high responsiveness
2203 * will suffer. */
2204 unsigned long lines = view->height;
2205 int redraw_from = -1;
2207 if (!view->pipe)
2208 return TRUE;
2210 /* Only redraw if lines are visible. */
2211 if (view->offset + view->height >= view->lines)
2212 redraw_from = view->lines - view->offset;
2214 /* FIXME: This is probably not perfect for backgrounded views. */
2215 if (!realloc_lines(view, view->lines + lines))
2216 goto alloc_error;
2218 while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
2219 size_t linelen = strlen(line);
2221 if (linelen)
2222 line[linelen - 1] = 0;
2224 if (opt_iconv != ICONV_NONE) {
2225 ICONV_CONST char *inbuf = line;
2226 size_t inlen = linelen;
2228 char *outbuf = out_buffer;
2229 size_t outlen = sizeof(out_buffer);
2231 size_t ret;
2233 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2234 if (ret != (size_t) -1) {
2235 line = out_buffer;
2236 linelen = strlen(out_buffer);
2237 }
2238 }
2240 if (!view->ops->read(view, line))
2241 goto alloc_error;
2243 if (lines-- == 1)
2244 break;
2245 }
2247 {
2248 int digits;
2250 lines = view->lines;
2251 for (digits = 0; lines; digits++)
2252 lines /= 10;
2254 /* Keep the displayed view in sync with line number scaling. */
2255 if (digits != view->digits) {
2256 view->digits = digits;
2257 redraw_from = 0;
2258 }
2259 }
2261 if (!view_is_displayed(view))
2262 goto check_pipe;
2264 if (view == VIEW(REQ_VIEW_TREE)) {
2265 /* Clear the view and redraw everything since the tree sorting
2266 * might have rearranged things. */
2267 redraw_view(view);
2269 } else if (redraw_from >= 0) {
2270 /* If this is an incremental update, redraw the previous line
2271 * since for commits some members could have changed when
2272 * loading the main view. */
2273 if (redraw_from > 0)
2274 redraw_from--;
2276 /* Since revision graph visualization requires knowledge
2277 * about the parent commit, it causes a further one-off
2278 * needed to be redrawn for incremental updates. */
2279 if (redraw_from > 0 && opt_rev_graph)
2280 redraw_from--;
2282 /* Incrementally draw avoids flickering. */
2283 redraw_view_from(view, redraw_from);
2284 }
2286 if (view == VIEW(REQ_VIEW_BLAME))
2287 redraw_view_dirty(view);
2289 /* Update the title _after_ the redraw so that if the redraw picks up a
2290 * commit reference in view->ref it'll be available here. */
2291 update_view_title(view);
2293 check_pipe:
2294 if (ferror(view->pipe)) {
2295 report("Failed to read: %s", strerror(errno));
2296 goto end;
2298 } else if (feof(view->pipe)) {
2299 report("");
2300 goto end;
2301 }
2303 return TRUE;
2305 alloc_error:
2306 report("Allocation failure");
2308 end:
2309 if (view->ops->read(view, NULL))
2310 end_update(view);
2311 return FALSE;
2312 }
2314 static struct line *
2315 add_line_data(struct view *view, void *data, enum line_type type)
2316 {
2317 struct line *line = &view->line[view->lines++];
2319 memset(line, 0, sizeof(*line));
2320 line->type = type;
2321 line->data = data;
2323 return line;
2324 }
2326 static struct line *
2327 add_line_text(struct view *view, char *data, enum line_type type)
2328 {
2329 if (data)
2330 data = strdup(data);
2332 return data ? add_line_data(view, data, type) : NULL;
2333 }
2336 /*
2337 * View opening
2338 */
2340 enum open_flags {
2341 OPEN_DEFAULT = 0, /* Use default view switching. */
2342 OPEN_SPLIT = 1, /* Split current view. */
2343 OPEN_BACKGROUNDED = 2, /* Backgrounded. */
2344 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
2345 OPEN_NOMAXIMIZE = 8, /* Do not maximize the current view. */
2346 };
2348 static void
2349 open_view(struct view *prev, enum request request, enum open_flags flags)
2350 {
2351 bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2352 bool split = !!(flags & OPEN_SPLIT);
2353 bool reload = !!(flags & OPEN_RELOAD);
2354 bool nomaximize = !!(flags & OPEN_NOMAXIMIZE);
2355 struct view *view = VIEW(request);
2356 int nviews = displayed_views();
2357 struct view *base_view = display[0];
2359 if (view == prev && nviews == 1 && !reload) {
2360 report("Already in %s view", view->name);
2361 return;
2362 }
2364 if (view->git_dir && !opt_git_dir[0]) {
2365 report("The %s view is disabled in pager view", view->name);
2366 return;
2367 }
2369 if (split) {
2370 display[1] = view;
2371 if (!backgrounded)
2372 current_view = 1;
2373 } else if (!nomaximize) {
2374 /* Maximize the current view. */
2375 memset(display, 0, sizeof(display));
2376 current_view = 0;
2377 display[current_view] = view;
2378 }
2380 /* Resize the view when switching between split- and full-screen,
2381 * or when switching between two different full-screen views. */
2382 if (nviews != displayed_views() ||
2383 (nviews == 1 && base_view != display[0]))
2384 resize_display();
2386 if (view->ops->open) {
2387 if (!view->ops->open(view)) {
2388 report("Failed to load %s view", view->name);
2389 return;
2390 }
2392 } else if ((reload || strcmp(view->vid, view->id)) &&
2393 !begin_update(view)) {
2394 report("Failed to load %s view", view->name);
2395 return;
2396 }
2398 if (split && prev->lineno - prev->offset >= prev->height) {
2399 /* Take the title line into account. */
2400 int lines = prev->lineno - prev->offset - prev->height + 1;
2402 /* Scroll the view that was split if the current line is
2403 * outside the new limited view. */
2404 do_scroll_view(prev, lines);
2405 }
2407 if (prev && view != prev) {
2408 if (split && !backgrounded) {
2409 /* "Blur" the previous view. */
2410 update_view_title(prev);
2411 }
2413 view->parent = prev;
2414 }
2416 if (view->pipe && view->lines == 0) {
2417 /* Clear the old view and let the incremental updating refill
2418 * the screen. */
2419 werase(view->win);
2420 report("");
2421 } else {
2422 redraw_view(view);
2423 report("");
2424 }
2426 /* If the view is backgrounded the above calls to report()
2427 * won't redraw the view title. */
2428 if (backgrounded)
2429 update_view_title(view);
2430 }
2432 static void
2433 open_external_viewer(const char *cmd)
2434 {
2435 def_prog_mode(); /* save current tty modes */
2436 endwin(); /* restore original tty modes */
2437 system(cmd);
2438 fprintf(stderr, "Press Enter to continue");
2439 getc(stdin);
2440 reset_prog_mode();
2441 redraw_display();
2442 }
2444 static void
2445 open_mergetool(const char *file)
2446 {
2447 char cmd[SIZEOF_STR];
2448 char file_sq[SIZEOF_STR];
2450 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2451 string_format(cmd, "git mergetool %s", file_sq)) {
2452 open_external_viewer(cmd);
2453 }
2454 }
2456 static void
2457 open_editor(bool from_root, const char *file)
2458 {
2459 char cmd[SIZEOF_STR];
2460 char file_sq[SIZEOF_STR];
2461 char *editor;
2462 char *prefix = from_root ? opt_cdup : "";
2464 editor = getenv("GIT_EDITOR");
2465 if (!editor && *opt_editor)
2466 editor = opt_editor;
2467 if (!editor)
2468 editor = getenv("VISUAL");
2469 if (!editor)
2470 editor = getenv("EDITOR");
2471 if (!editor)
2472 editor = "vi";
2474 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2475 string_format(cmd, "%s %s%s", editor, prefix, file_sq)) {
2476 open_external_viewer(cmd);
2477 }
2478 }
2480 static void
2481 open_run_request(enum request request)
2482 {
2483 struct run_request *req = get_run_request(request);
2484 char buf[SIZEOF_STR * 2];
2485 size_t bufpos;
2486 char *cmd;
2488 if (!req) {
2489 report("Unknown run request");
2490 return;
2491 }
2493 bufpos = 0;
2494 cmd = req->cmd;
2496 while (cmd) {
2497 char *next = strstr(cmd, "%(");
2498 int len = next - cmd;
2499 char *value;
2501 if (!next) {
2502 len = strlen(cmd);
2503 value = "";
2505 } else if (!strncmp(next, "%(head)", 7)) {
2506 value = ref_head;
2508 } else if (!strncmp(next, "%(commit)", 9)) {
2509 value = ref_commit;
2511 } else if (!strncmp(next, "%(blob)", 7)) {
2512 value = ref_blob;
2514 } else {
2515 report("Unknown replacement in run request: `%s`", req->cmd);
2516 return;
2517 }
2519 if (!string_format_from(buf, &bufpos, "%.*s%s", len, cmd, value))
2520 return;
2522 if (next)
2523 next = strchr(next, ')') + 1;
2524 cmd = next;
2525 }
2527 open_external_viewer(buf);
2528 }
2530 /*
2531 * User request switch noodle
2532 */
2534 static int
2535 view_driver(struct view *view, enum request request)
2536 {
2537 int i;
2539 if (request == REQ_NONE) {
2540 doupdate();
2541 return TRUE;
2542 }
2544 if (request > REQ_NONE) {
2545 open_run_request(request);
2546 /* FIXME: When all views can refresh always do this. */
2547 if (view == VIEW(REQ_VIEW_STATUS) ||
2548 view == VIEW(REQ_VIEW_STAGE))
2549 request = REQ_REFRESH;
2550 else
2551 return TRUE;
2552 }
2554 if (view && view->lines) {
2555 request = view->ops->request(view, request, &view->line[view->lineno]);
2556 if (request == REQ_NONE)
2557 return TRUE;
2558 }
2560 switch (request) {
2561 case REQ_MOVE_UP:
2562 case REQ_MOVE_DOWN:
2563 case REQ_MOVE_PAGE_UP:
2564 case REQ_MOVE_PAGE_DOWN:
2565 case REQ_MOVE_FIRST_LINE:
2566 case REQ_MOVE_LAST_LINE:
2567 move_view(view, request);
2568 break;
2570 case REQ_SCROLL_LINE_DOWN:
2571 case REQ_SCROLL_LINE_UP:
2572 case REQ_SCROLL_PAGE_DOWN:
2573 case REQ_SCROLL_PAGE_UP:
2574 scroll_view(view, request);
2575 break;
2577 case REQ_VIEW_BLAME:
2578 if (!opt_file[0]) {
2579 report("No file chosen, press %s to open tree view",
2580 get_key(REQ_VIEW_TREE));
2581 break;
2582 }
2583 open_view(view, request, OPEN_DEFAULT);
2584 break;
2586 case REQ_VIEW_BLOB:
2587 if (!ref_blob[0]) {
2588 report("No file chosen, press %s to open tree view",
2589 get_key(REQ_VIEW_TREE));
2590 break;
2591 }
2592 open_view(view, request, OPEN_DEFAULT);
2593 break;
2595 case REQ_VIEW_PAGER:
2596 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2597 report("No pager content, press %s to run command from prompt",
2598 get_key(REQ_PROMPT));
2599 break;
2600 }
2601 open_view(view, request, OPEN_DEFAULT);
2602 break;
2604 case REQ_VIEW_STAGE:
2605 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2606 report("No stage content, press %s to open the status view and choose file",
2607 get_key(REQ_VIEW_STATUS));
2608 break;
2609 }
2610 open_view(view, request, OPEN_DEFAULT);
2611 break;
2613 case REQ_VIEW_STATUS:
2614 if (opt_is_inside_work_tree == FALSE) {
2615 report("The status view requires a working tree");
2616 break;
2617 }
2618 open_view(view, request, OPEN_DEFAULT);
2619 break;
2621 case REQ_VIEW_MAIN:
2622 case REQ_VIEW_DIFF:
2623 case REQ_VIEW_LOG:
2624 case REQ_VIEW_TREE:
2625 case REQ_VIEW_HELP:
2626 open_view(view, request, OPEN_DEFAULT);
2627 break;
2629 case REQ_NEXT:
2630 case REQ_PREVIOUS:
2631 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2633 if ((view == VIEW(REQ_VIEW_DIFF) &&
2634 view->parent == VIEW(REQ_VIEW_MAIN)) ||
2635 (view == VIEW(REQ_VIEW_DIFF) &&
2636 view->parent == VIEW(REQ_VIEW_BLAME)) ||
2637 (view == VIEW(REQ_VIEW_STAGE) &&
2638 view->parent == VIEW(REQ_VIEW_STATUS)) ||
2639 (view == VIEW(REQ_VIEW_BLOB) &&
2640 view->parent == VIEW(REQ_VIEW_TREE))) {
2641 int line;
2643 view = view->parent;
2644 line = view->lineno;
2645 move_view(view, request);
2646 if (view_is_displayed(view))
2647 update_view_title(view);
2648 if (line != view->lineno)
2649 view->ops->request(view, REQ_ENTER,
2650 &view->line[view->lineno]);
2652 } else {
2653 move_view(view, request);
2654 }
2655 break;
2657 case REQ_VIEW_NEXT:
2658 {
2659 int nviews = displayed_views();
2660 int next_view = (current_view + 1) % nviews;
2662 if (next_view == current_view) {
2663 report("Only one view is displayed");
2664 break;
2665 }
2667 current_view = next_view;
2668 /* Blur out the title of the previous view. */
2669 update_view_title(view);
2670 report("");
2671 break;
2672 }
2673 case REQ_REFRESH:
2674 report("Refreshing is not yet supported for the %s view", view->name);
2675 break;
2677 case REQ_MAXIMIZE:
2678 if (displayed_views() == 2)
2679 open_view(view, VIEW_REQ(view), OPEN_DEFAULT);
2680 break;
2682 case REQ_TOGGLE_LINENO:
2683 opt_line_number = !opt_line_number;
2684 redraw_display();
2685 break;
2687 case REQ_TOGGLE_DATE:
2688 opt_date = !opt_date;
2689 redraw_display();
2690 break;
2692 case REQ_TOGGLE_AUTHOR:
2693 opt_author = !opt_author;
2694 redraw_display();
2695 break;
2697 case REQ_TOGGLE_REV_GRAPH:
2698 opt_rev_graph = !opt_rev_graph;
2699 redraw_display();
2700 break;
2702 case REQ_TOGGLE_REFS:
2703 opt_show_refs = !opt_show_refs;
2704 redraw_display();
2705 break;
2707 case REQ_PROMPT:
2708 /* Always reload^Wrerun commands from the prompt. */
2709 open_view(view, opt_request, OPEN_RELOAD);
2710 break;
2712 case REQ_SEARCH:
2713 case REQ_SEARCH_BACK:
2714 search_view(view, request);
2715 break;
2717 case REQ_FIND_NEXT:
2718 case REQ_FIND_PREV:
2719 find_next(view, request);
2720 break;
2722 case REQ_STOP_LOADING:
2723 for (i = 0; i < ARRAY_SIZE(views); i++) {
2724 view = &views[i];
2725 if (view->pipe)
2726 report("Stopped loading the %s view", view->name),
2727 end_update(view);
2728 }
2729 break;
2731 case REQ_SHOW_VERSION:
2732 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2733 return TRUE;
2735 case REQ_SCREEN_RESIZE:
2736 resize_display();
2737 /* Fall-through */
2738 case REQ_SCREEN_REDRAW:
2739 redraw_display();
2740 break;
2742 case REQ_EDIT:
2743 report("Nothing to edit");
2744 break;
2747 case REQ_ENTER:
2748 report("Nothing to enter");
2749 break;
2752 case REQ_VIEW_CLOSE:
2753 /* XXX: Mark closed views by letting view->parent point to the
2754 * view itself. Parents to closed view should never be
2755 * followed. */
2756 if (view->parent &&
2757 view->parent->parent != view->parent) {
2758 memset(display, 0, sizeof(display));
2759 current_view = 0;
2760 display[current_view] = view->parent;
2761 view->parent = view;
2762 resize_display();
2763 redraw_display();
2764 break;
2765 }
2766 /* Fall-through */
2767 case REQ_QUIT:
2768 return FALSE;
2770 default:
2771 /* An unknown key will show most commonly used commands. */
2772 report("Unknown key, press 'h' for help");
2773 return TRUE;
2774 }
2776 return TRUE;
2777 }
2780 /*
2781 * Pager backend
2782 */
2784 static bool
2785 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2786 {
2787 static char spaces[] = " ";
2788 char *text = line->data;
2789 int col = 0;
2791 if (opt_line_number) {
2792 col += draw_lineno(view, lineno, view->width, selected);
2793 if (col >= view->width)
2794 return TRUE;
2795 }
2797 if (!selected)
2798 wattrset(view->win, get_line_attr(line->type));
2800 if (opt_tab_size < TABSIZE) {
2801 int col_offset = col;
2803 col = 0;
2804 while (text && col_offset + col < view->width) {
2805 int cols_max = view->width - col_offset - col;
2806 char *pos = text;
2807 int cols;
2809 if (*text == '\t') {
2810 text++;
2811 assert(sizeof(spaces) > TABSIZE);
2812 pos = spaces;
2813 cols = opt_tab_size - (col % opt_tab_size);
2815 } else {
2816 text = strchr(text, '\t');
2817 cols = line ? text - pos : strlen(pos);
2818 }
2820 waddnstr(view->win, pos, MIN(cols, cols_max));
2821 col += cols;
2822 }
2824 } else {
2825 draw_text(view, text, view->width - col, TRUE, selected);
2826 }
2828 return TRUE;
2829 }
2831 static bool
2832 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2833 {
2834 char refbuf[SIZEOF_STR];
2835 char *ref = NULL;
2836 FILE *pipe;
2838 if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2839 return TRUE;
2841 pipe = popen(refbuf, "r");
2842 if (!pipe)
2843 return TRUE;
2845 if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2846 ref = chomp_string(ref);
2847 pclose(pipe);
2849 if (!ref || !*ref)
2850 return TRUE;
2852 /* This is the only fatal call, since it can "corrupt" the buffer. */
2853 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2854 return FALSE;
2856 return TRUE;
2857 }
2859 static void
2860 add_pager_refs(struct view *view, struct line *line)
2861 {
2862 char buf[SIZEOF_STR];
2863 char *commit_id = (char *)line->data + STRING_SIZE("commit ");
2864 struct ref **refs;
2865 size_t bufpos = 0, refpos = 0;
2866 const char *sep = "Refs: ";
2867 bool is_tag = FALSE;
2869 assert(line->type == LINE_COMMIT);
2871 refs = get_refs(commit_id);
2872 if (!refs) {
2873 if (view == VIEW(REQ_VIEW_DIFF))
2874 goto try_add_describe_ref;
2875 return;
2876 }
2878 do {
2879 struct ref *ref = refs[refpos];
2880 char *fmt = ref->tag ? "%s[%s]" :
2881 ref->remote ? "%s<%s>" : "%s%s";
2883 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2884 return;
2885 sep = ", ";
2886 if (ref->tag)
2887 is_tag = TRUE;
2888 } while (refs[refpos++]->next);
2890 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2891 try_add_describe_ref:
2892 /* Add <tag>-g<commit_id> "fake" reference. */
2893 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2894 return;
2895 }
2897 if (bufpos == 0)
2898 return;
2900 if (!realloc_lines(view, view->line_size + 1))
2901 return;
2903 add_line_text(view, buf, LINE_PP_REFS);
2904 }
2906 static bool
2907 pager_read(struct view *view, char *data)
2908 {
2909 struct line *line;
2911 if (!data)
2912 return TRUE;
2914 line = add_line_text(view, data, get_line_type(data));
2915 if (!line)
2916 return FALSE;
2918 if (line->type == LINE_COMMIT &&
2919 (view == VIEW(REQ_VIEW_DIFF) ||
2920 view == VIEW(REQ_VIEW_LOG)))
2921 add_pager_refs(view, line);
2923 return TRUE;
2924 }
2926 static enum request
2927 pager_request(struct view *view, enum request request, struct line *line)
2928 {
2929 int split = 0;
2931 if (request != REQ_ENTER)
2932 return request;
2934 if (line->type == LINE_COMMIT &&
2935 (view == VIEW(REQ_VIEW_LOG) ||
2936 view == VIEW(REQ_VIEW_PAGER))) {
2937 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2938 split = 1;
2939 }
2941 /* Always scroll the view even if it was split. That way
2942 * you can use Enter to scroll through the log view and
2943 * split open each commit diff. */
2944 scroll_view(view, REQ_SCROLL_LINE_DOWN);
2946 /* FIXME: A minor workaround. Scrolling the view will call report("")
2947 * but if we are scrolling a non-current view this won't properly
2948 * update the view title. */
2949 if (split)
2950 update_view_title(view);
2952 return REQ_NONE;
2953 }
2955 static bool
2956 pager_grep(struct view *view, struct line *line)
2957 {
2958 regmatch_t pmatch;
2959 char *text = line->data;
2961 if (!*text)
2962 return FALSE;
2964 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2965 return FALSE;
2967 return TRUE;
2968 }
2970 static void
2971 pager_select(struct view *view, struct line *line)
2972 {
2973 if (line->type == LINE_COMMIT) {
2974 char *text = (char *)line->data + STRING_SIZE("commit ");
2976 if (view != VIEW(REQ_VIEW_PAGER))
2977 string_copy_rev(view->ref, text);
2978 string_copy_rev(ref_commit, text);
2979 }
2980 }
2982 static struct view_ops pager_ops = {
2983 "line",
2984 NULL,
2985 pager_read,
2986 pager_draw,
2987 pager_request,
2988 pager_grep,
2989 pager_select,
2990 };
2993 /*
2994 * Help backend
2995 */
2997 static bool
2998 help_open(struct view *view)
2999 {
3000 char buf[BUFSIZ];
3001 int lines = ARRAY_SIZE(req_info) + 2;
3002 int i;
3004 if (view->lines > 0)
3005 return TRUE;
3007 for (i = 0; i < ARRAY_SIZE(req_info); i++)
3008 if (!req_info[i].request)
3009 lines++;
3011 lines += run_requests + 1;
3013 view->line = calloc(lines, sizeof(*view->line));
3014 if (!view->line)
3015 return FALSE;
3017 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3019 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3020 char *key;
3022 if (req_info[i].request == REQ_NONE)
3023 continue;
3025 if (!req_info[i].request) {
3026 add_line_text(view, "", LINE_DEFAULT);
3027 add_line_text(view, req_info[i].help, LINE_DEFAULT);
3028 continue;
3029 }
3031 key = get_key(req_info[i].request);
3032 if (!*key)
3033 key = "(no key defined)";
3035 if (!string_format(buf, " %-25s %s", key, req_info[i].help))
3036 continue;
3038 add_line_text(view, buf, LINE_DEFAULT);
3039 }
3041 if (run_requests) {
3042 add_line_text(view, "", LINE_DEFAULT);
3043 add_line_text(view, "External commands:", LINE_DEFAULT);
3044 }
3046 for (i = 0; i < run_requests; i++) {
3047 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3048 char *key;
3050 if (!req)
3051 continue;
3053 key = get_key_name(req->key);
3054 if (!*key)
3055 key = "(no key defined)";
3057 if (!string_format(buf, " %-10s %-14s `%s`",
3058 keymap_table[req->keymap].name,
3059 key, req->cmd))
3060 continue;
3062 add_line_text(view, buf, LINE_DEFAULT);
3063 }
3065 return TRUE;
3066 }
3068 static struct view_ops help_ops = {
3069 "line",
3070 help_open,
3071 NULL,
3072 pager_draw,
3073 pager_request,
3074 pager_grep,
3075 pager_select,
3076 };
3079 /*
3080 * Tree backend
3081 */
3083 struct tree_stack_entry {
3084 struct tree_stack_entry *prev; /* Entry below this in the stack */
3085 unsigned long lineno; /* Line number to restore */
3086 char *name; /* Position of name in opt_path */
3087 };
3089 /* The top of the path stack. */
3090 static struct tree_stack_entry *tree_stack = NULL;
3091 unsigned long tree_lineno = 0;
3093 static void
3094 pop_tree_stack_entry(void)
3095 {
3096 struct tree_stack_entry *entry = tree_stack;
3098 tree_lineno = entry->lineno;
3099 entry->name[0] = 0;
3100 tree_stack = entry->prev;
3101 free(entry);
3102 }
3104 static void
3105 push_tree_stack_entry(char *name, unsigned long lineno)
3106 {
3107 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3108 size_t pathlen = strlen(opt_path);
3110 if (!entry)
3111 return;
3113 entry->prev = tree_stack;
3114 entry->name = opt_path + pathlen;
3115 tree_stack = entry;
3117 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3118 pop_tree_stack_entry();
3119 return;
3120 }
3122 /* Move the current line to the first tree entry. */
3123 tree_lineno = 1;
3124 entry->lineno = lineno;
3125 }
3127 /* Parse output from git-ls-tree(1):
3128 *
3129 * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
3130 * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
3131 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3132 * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
3133 */
3135 #define SIZEOF_TREE_ATTR \
3136 STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
3138 #define TREE_UP_FORMAT "040000 tree %s\t.."
3140 static int
3141 tree_compare_entry(enum line_type type1, char *name1,
3142 enum line_type type2, char *name2)
3143 {
3144 if (type1 != type2) {
3145 if (type1 == LINE_TREE_DIR)
3146 return -1;
3147 return 1;
3148 }
3150 return strcmp(name1, name2);
3151 }
3153 static char *
3154 tree_path(struct line *line)
3155 {
3156 char *path = line->data;
3158 return path + SIZEOF_TREE_ATTR;
3159 }
3161 static bool
3162 tree_read(struct view *view, char *text)
3163 {
3164 size_t textlen = text ? strlen(text) : 0;
3165 char buf[SIZEOF_STR];
3166 unsigned long pos;
3167 enum line_type type;
3168 bool first_read = view->lines == 0;
3170 if (!text)
3171 return TRUE;
3172 if (textlen <= SIZEOF_TREE_ATTR)
3173 return FALSE;
3175 type = text[STRING_SIZE("100644 ")] == 't'
3176 ? LINE_TREE_DIR : LINE_TREE_FILE;
3178 if (first_read) {
3179 /* Add path info line */
3180 if (!string_format(buf, "Directory path /%s", opt_path) ||
3181 !realloc_lines(view, view->line_size + 1) ||
3182 !add_line_text(view, buf, LINE_DEFAULT))
3183 return FALSE;
3185 /* Insert "link" to parent directory. */
3186 if (*opt_path) {
3187 if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
3188 !realloc_lines(view, view->line_size + 1) ||
3189 !add_line_text(view, buf, LINE_TREE_DIR))
3190 return FALSE;
3191 }
3192 }
3194 /* Strip the path part ... */
3195 if (*opt_path) {
3196 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3197 size_t striplen = strlen(opt_path);
3198 char *path = text + SIZEOF_TREE_ATTR;
3200 if (pathlen > striplen)
3201 memmove(path, path + striplen,
3202 pathlen - striplen + 1);
3203 }
3205 /* Skip "Directory ..." and ".." line. */
3206 for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
3207 struct line *line = &view->line[pos];
3208 char *path1 = tree_path(line);
3209 char *path2 = text + SIZEOF_TREE_ATTR;
3210 int cmp = tree_compare_entry(line->type, path1, type, path2);
3212 if (cmp <= 0)
3213 continue;
3215 text = strdup(text);
3216 if (!text)
3217 return FALSE;
3219 if (view->lines > pos)
3220 memmove(&view->line[pos + 1], &view->line[pos],
3221 (view->lines - pos) * sizeof(*line));
3223 line = &view->line[pos];
3224 line->data = text;
3225 line->type = type;
3226 view->lines++;
3227 return TRUE;
3228 }
3230 if (!add_line_text(view, text, type))
3231 return FALSE;
3233 if (tree_lineno > view->lineno) {
3234 view->lineno = tree_lineno;
3235 tree_lineno = 0;
3236 }
3238 return TRUE;
3239 }
3241 static enum request
3242 tree_request(struct view *view, enum request request, struct line *line)
3243 {
3244 enum open_flags flags;
3246 if (request == REQ_VIEW_BLAME) {
3247 char *filename = tree_path(line);
3249 if (line->type == LINE_TREE_DIR) {
3250 report("Cannot show blame for directory %s", opt_path);
3251 return REQ_NONE;
3252 }
3254 string_copy(opt_ref, view->vid);
3255 string_format(opt_file, "%s%s", opt_path, filename);
3256 return request;
3257 }
3258 if (request == REQ_TREE_PARENT) {
3259 if (*opt_path) {
3260 /* fake 'cd ..' */
3261 request = REQ_ENTER;
3262 line = &view->line[1];
3263 } else {
3264 /* quit view if at top of tree */
3265 return REQ_VIEW_CLOSE;
3266 }
3267 }
3268 if (request != REQ_ENTER)
3269 return request;
3271 /* Cleanup the stack if the tree view is at a different tree. */
3272 while (!*opt_path && tree_stack)
3273 pop_tree_stack_entry();
3275 switch (line->type) {
3276 case LINE_TREE_DIR:
3277 /* Depending on whether it is a subdir or parent (updir?) link
3278 * mangle the path buffer. */
3279 if (line == &view->line[1] && *opt_path) {
3280 pop_tree_stack_entry();
3282 } else {
3283 char *basename = tree_path(line);
3285 push_tree_stack_entry(basename, view->lineno);
3286 }
3288 /* Trees and subtrees share the same ID, so they are not not
3289 * unique like blobs. */
3290 flags = OPEN_RELOAD;
3291 request = REQ_VIEW_TREE;
3292 break;
3294 case LINE_TREE_FILE:
3295 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3296 request = REQ_VIEW_BLOB;
3297 break;
3299 default:
3300 return TRUE;
3301 }
3303 open_view(view, request, flags);
3304 if (request == REQ_VIEW_TREE) {
3305 view->lineno = tree_lineno;
3306 }
3308 return REQ_NONE;
3309 }
3311 static void
3312 tree_select(struct view *view, struct line *line)
3313 {
3314 char *text = (char *)line->data + STRING_SIZE("100644 blob ");
3316 if (line->type == LINE_TREE_FILE) {
3317 string_copy_rev(ref_blob, text);
3319 } else if (line->type != LINE_TREE_DIR) {
3320 return;
3321 }
3323 string_copy_rev(view->ref, text);
3324 }
3326 static struct view_ops tree_ops = {
3327 "file",
3328 NULL,
3329 tree_read,
3330 pager_draw,
3331 tree_request,
3332 pager_grep,
3333 tree_select,
3334 };
3336 static bool
3337 blob_read(struct view *view, char *line)
3338 {
3339 if (!line)
3340 return TRUE;
3341 return add_line_text(view, line, LINE_DEFAULT) != NULL;
3342 }
3344 static struct view_ops blob_ops = {
3345 "line",
3346 NULL,
3347 blob_read,
3348 pager_draw,
3349 pager_request,
3350 pager_grep,
3351 pager_select,
3352 };
3354 /*
3355 * Blame backend
3356 *
3357 * Loading the blame view is a two phase job:
3358 *
3359 * 1. File content is read either using opt_file from the
3360 * filesystem or using git-cat-file.
3361 * 2. Then blame information is incrementally added by
3362 * reading output from git-blame.
3363 */
3365 struct blame_commit {
3366 char id[SIZEOF_REV]; /* SHA1 ID. */
3367 char title[128]; /* First line of the commit message. */
3368 char author[75]; /* Author of the commit. */
3369 struct tm time; /* Date from the author ident. */
3370 char filename[128]; /* Name of file. */
3371 };
3373 struct blame {
3374 struct blame_commit *commit;
3375 unsigned int header:1;
3376 char text[1];
3377 };
3379 #define BLAME_CAT_FILE_CMD "git cat-file blob %s:%s"
3380 #define BLAME_INCREMENTAL_CMD "git blame --incremental %s %s"
3382 static bool
3383 blame_open(struct view *view)
3384 {
3385 char path[SIZEOF_STR];
3386 char ref[SIZEOF_STR] = "";
3388 if (sq_quote(path, 0, opt_file) >= sizeof(path))
3389 return FALSE;
3391 if (*opt_ref && sq_quote(ref, 0, opt_ref) >= sizeof(ref))
3392 return FALSE;
3394 if (*opt_ref) {
3395 if (!string_format(view->cmd, BLAME_CAT_FILE_CMD, ref, path))
3396 return FALSE;
3397 } else {
3398 view->pipe = fopen(opt_file, "r");
3399 if (!view->pipe &&
3400 !string_format(view->cmd, BLAME_CAT_FILE_CMD, "HEAD", path))
3401 return FALSE;
3402 }
3404 if (!view->pipe)
3405 view->pipe = popen(view->cmd, "r");
3406 if (!view->pipe)
3407 return FALSE;
3409 if (!string_format(view->cmd, BLAME_INCREMENTAL_CMD, ref, path))
3410 return FALSE;
3412 string_format(view->ref, "%s ...", opt_file);
3413 string_copy_rev(view->vid, opt_file);
3414 set_nonblocking_input(TRUE);
3416 if (view->line) {
3417 int i;
3419 for (i = 0; i < view->lines; i++)
3420 free(view->line[i].data);
3421 free(view->line);
3422 }
3424 view->lines = view->line_alloc = view->line_size = view->lineno = 0;
3425 view->offset = view->lines = view->lineno = 0;
3426 view->line = NULL;
3427 view->start_time = time(NULL);
3429 return TRUE;
3430 }
3432 static struct blame_commit *
3433 get_blame_commit(struct view *view, const char *id)
3434 {
3435 size_t i;
3437 for (i = 0; i < view->lines; i++) {
3438 struct blame *blame = view->line[i].data;
3440 if (!blame->commit)
3441 continue;
3443 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
3444 return blame->commit;
3445 }
3447 {
3448 struct blame_commit *commit = calloc(1, sizeof(*commit));
3450 if (commit)
3451 string_ncopy(commit->id, id, SIZEOF_REV);
3452 return commit;
3453 }
3454 }
3456 static bool
3457 parse_number(char **posref, size_t *number, size_t min, size_t max)
3458 {
3459 char *pos = *posref;
3461 *posref = NULL;
3462 pos = strchr(pos + 1, ' ');
3463 if (!pos || !isdigit(pos[1]))
3464 return FALSE;
3465 *number = atoi(pos + 1);
3466 if (*number < min || *number > max)
3467 return FALSE;
3469 *posref = pos;
3470 return TRUE;
3471 }
3473 static struct blame_commit *
3474 parse_blame_commit(struct view *view, char *text, int *blamed)
3475 {
3476 struct blame_commit *commit;
3477 struct blame *blame;
3478 char *pos = text + SIZEOF_REV - 1;
3479 size_t lineno;
3480 size_t group;
3482 if (strlen(text) <= SIZEOF_REV || *pos != ' ')
3483 return NULL;
3485 if (!parse_number(&pos, &lineno, 1, view->lines) ||
3486 !parse_number(&pos, &group, 1, view->lines - lineno + 1))
3487 return NULL;
3489 commit = get_blame_commit(view, text);
3490 if (!commit)
3491 return NULL;
3493 *blamed += group;
3494 while (group--) {
3495 struct line *line = &view->line[lineno + group - 1];
3497 blame = line->data;
3498 blame->commit = commit;
3499 blame->header = !group;
3500 line->dirty = 1;
3501 }
3503 return commit;
3504 }
3506 static bool
3507 blame_read_file(struct view *view, char *line)
3508 {
3509 if (!line) {
3510 FILE *pipe = NULL;
3512 if (view->lines > 0)
3513 pipe = popen(view->cmd, "r");
3514 else if (!view->parent)
3515 die("No blame exist for %s", view->vid);
3516 view->cmd[0] = 0;
3517 if (!pipe) {
3518 report("Failed to load blame data");
3519 return TRUE;
3520 }
3522 fclose(view->pipe);
3523 view->pipe = pipe;
3524 return FALSE;
3526 } else {
3527 size_t linelen = strlen(line);
3528 struct blame *blame = malloc(sizeof(*blame) + linelen);
3530 if (!line)
3531 return FALSE;
3533 blame->commit = NULL;
3534 strncpy(blame->text, line, linelen);
3535 blame->text[linelen] = 0;
3536 return add_line_data(view, blame, LINE_BLAME_COMMIT) != NULL;
3537 }
3538 }
3540 static bool
3541 match_blame_header(const char *name, char **line)
3542 {
3543 size_t namelen = strlen(name);
3544 bool matched = !strncmp(name, *line, namelen);
3546 if (matched)
3547 *line += namelen;
3549 return matched;
3550 }
3552 static bool
3553 blame_read(struct view *view, char *line)
3554 {
3555 static struct blame_commit *commit = NULL;
3556 static int blamed = 0;
3557 static time_t author_time;
3559 if (*view->cmd)
3560 return blame_read_file(view, line);
3562 if (!line) {
3563 /* Reset all! */
3564 commit = NULL;
3565 blamed = 0;
3566 string_format(view->ref, "%s", view->vid);
3567 if (view_is_displayed(view)) {
3568 update_view_title(view);
3569 redraw_view_from(view, 0);
3570 }
3571 return TRUE;
3572 }
3574 if (!commit) {
3575 commit = parse_blame_commit(view, line, &blamed);
3576 string_format(view->ref, "%s %2d%%", view->vid,
3577 blamed * 100 / view->lines);
3579 } else if (match_blame_header("author ", &line)) {
3580 string_ncopy(commit->author, line, strlen(line));
3582 } else if (match_blame_header("author-time ", &line)) {
3583 author_time = (time_t) atol(line);
3585 } else if (match_blame_header("author-tz ", &line)) {
3586 long tz;
3588 tz = ('0' - line[1]) * 60 * 60 * 10;
3589 tz += ('0' - line[2]) * 60 * 60;
3590 tz += ('0' - line[3]) * 60;
3591 tz += ('0' - line[4]) * 60;
3593 if (line[0] == '-')
3594 tz = -tz;
3596 author_time -= tz;
3597 gmtime_r(&author_time, &commit->time);
3599 } else if (match_blame_header("summary ", &line)) {
3600 string_ncopy(commit->title, line, strlen(line));
3602 } else if (match_blame_header("filename ", &line)) {
3603 string_ncopy(commit->filename, line, strlen(line));
3604 commit = NULL;
3605 }
3607 return TRUE;
3608 }
3610 static bool
3611 blame_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3612 {
3613 struct blame *blame = line->data;
3614 int col = 0;
3616 if (opt_date) {
3617 struct tm *time = blame->commit && *blame->commit->filename
3618 ? &blame->commit->time : NULL;
3620 col += draw_date(view, time, view->width, selected);
3621 if (col >= view->width)
3622 return TRUE;
3623 }
3625 if (opt_author) {
3626 int max = MIN(AUTHOR_COLS - 1, view->width - col);
3628 if (!selected)
3629 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
3630 if (blame->commit)
3631 draw_text(view, blame->commit->author, max, TRUE, selected);
3632 col += AUTHOR_COLS;
3633 if (col >= view->width)
3634 return TRUE;
3635 wmove(view->win, lineno, col);
3636 }
3638 {
3639 int max = MIN(ID_COLS - 1, view->width - col);
3641 if (!selected)
3642 wattrset(view->win, get_line_attr(LINE_BLAME_ID));
3643 if (blame->commit)
3644 draw_text(view, blame->commit->id, max, FALSE, -1);
3645 col += ID_COLS;
3646 if (col >= view->width)
3647 return TRUE;
3648 wmove(view->win, lineno, col);
3649 }
3651 {
3652 col += draw_lineno(view, lineno, view->width - col, selected);
3653 if (col >= view->width)
3654 return TRUE;
3655 }
3657 col += draw_text(view, blame->text, view->width - col, TRUE, selected);
3659 return TRUE;
3660 }
3662 static enum request
3663 blame_request(struct view *view, enum request request, struct line *line)
3664 {
3665 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3666 struct blame *blame = line->data;
3668 switch (request) {
3669 case REQ_ENTER:
3670 if (!blame->commit) {
3671 report("No commit loaded yet");
3672 break;
3673 }
3675 if (!strcmp(blame->commit->id, NULL_ID)) {
3676 char path[SIZEOF_STR];
3678 if (sq_quote(path, 0, view->vid) >= sizeof(path))
3679 break;
3680 string_format(opt_cmd, "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s 2>/dev/null", path);
3681 }
3683 open_view(view, REQ_VIEW_DIFF, flags);
3684 break;
3686 default:
3687 return request;
3688 }
3690 return REQ_NONE;
3691 }
3693 static bool
3694 blame_grep(struct view *view, struct line *line)
3695 {
3696 struct blame *blame = line->data;
3697 struct blame_commit *commit = blame->commit;
3698 regmatch_t pmatch;
3700 #define MATCH(text) \
3701 (*text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3703 if (commit) {
3704 char buf[DATE_COLS + 1];
3706 if (MATCH(commit->title) ||
3707 MATCH(commit->author) ||
3708 MATCH(commit->id))
3709 return TRUE;
3711 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
3712 MATCH(buf))
3713 return TRUE;
3714 }
3716 return MATCH(blame->text);
3718 #undef MATCH
3719 }
3721 static void
3722 blame_select(struct view *view, struct line *line)
3723 {
3724 struct blame *blame = line->data;
3725 struct blame_commit *commit = blame->commit;
3727 if (!commit)
3728 return;
3730 if (!strcmp(commit->id, NULL_ID))
3731 string_ncopy(ref_commit, "HEAD", 4);
3732 else
3733 string_copy_rev(ref_commit, commit->id);
3734 }
3736 static struct view_ops blame_ops = {
3737 "line",
3738 blame_open,
3739 blame_read,
3740 blame_draw,
3741 blame_request,
3742 blame_grep,
3743 blame_select,
3744 };
3746 /*
3747 * Status backend
3748 */
3750 struct status {
3751 char status;
3752 struct {
3753 mode_t mode;
3754 char rev[SIZEOF_REV];
3755 char name[SIZEOF_STR];
3756 } old;
3757 struct {
3758 mode_t mode;
3759 char rev[SIZEOF_REV];
3760 char name[SIZEOF_STR];
3761 } new;
3762 };
3764 static char status_onbranch[SIZEOF_STR];
3765 static struct status stage_status;
3766 static enum line_type stage_line_type;
3768 /* Get fields from the diff line:
3769 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
3770 */
3771 static inline bool
3772 status_get_diff(struct status *file, char *buf, size_t bufsize)
3773 {
3774 char *old_mode = buf + 1;
3775 char *new_mode = buf + 8;
3776 char *old_rev = buf + 15;
3777 char *new_rev = buf + 56;
3778 char *status = buf + 97;
3780 if (bufsize < 99 ||
3781 old_mode[-1] != ':' ||
3782 new_mode[-1] != ' ' ||
3783 old_rev[-1] != ' ' ||
3784 new_rev[-1] != ' ' ||
3785 status[-1] != ' ')
3786 return FALSE;
3788 file->status = *status;
3790 string_copy_rev(file->old.rev, old_rev);
3791 string_copy_rev(file->new.rev, new_rev);
3793 file->old.mode = strtoul(old_mode, NULL, 8);
3794 file->new.mode = strtoul(new_mode, NULL, 8);
3796 file->old.name[0] = file->new.name[0] = 0;
3798 return TRUE;
3799 }
3801 static bool
3802 status_run(struct view *view, const char cmd[], char status, enum line_type type)
3803 {
3804 struct status *file = NULL;
3805 struct status *unmerged = NULL;
3806 char buf[SIZEOF_STR * 4];
3807 size_t bufsize = 0;
3808 FILE *pipe;
3810 pipe = popen(cmd, "r");
3811 if (!pipe)
3812 return FALSE;
3814 add_line_data(view, NULL, type);
3816 while (!feof(pipe) && !ferror(pipe)) {
3817 char *sep;
3818 size_t readsize;
3820 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
3821 if (!readsize)
3822 break;
3823 bufsize += readsize;
3825 /* Process while we have NUL chars. */
3826 while ((sep = memchr(buf, 0, bufsize))) {
3827 size_t sepsize = sep - buf + 1;
3829 if (!file) {
3830 if (!realloc_lines(view, view->line_size + 1))
3831 goto error_out;
3833 file = calloc(1, sizeof(*file));
3834 if (!file)
3835 goto error_out;
3837 add_line_data(view, file, type);
3838 }
3840 /* Parse diff info part. */
3841 if (status) {
3842 file->status = status;
3843 if (status == 'A')
3844 string_copy(file->old.rev, NULL_ID);
3846 } else if (!file->status) {
3847 if (!status_get_diff(file, buf, sepsize))
3848 goto error_out;
3850 bufsize -= sepsize;
3851 memmove(buf, sep + 1, bufsize);
3853 sep = memchr(buf, 0, bufsize);
3854 if (!sep)
3855 break;
3856 sepsize = sep - buf + 1;
3858 /* Collapse all 'M'odified entries that
3859 * follow a associated 'U'nmerged entry.
3860 */
3861 if (file->status == 'U') {
3862 unmerged = file;
3864 } else if (unmerged) {
3865 int collapse = !strcmp(buf, unmerged->new.name);
3867 unmerged = NULL;
3868 if (collapse) {
3869 free(file);
3870 view->lines--;
3871 continue;
3872 }
3873 }
3874 }
3876 /* Grab the old name for rename/copy. */
3877 if (!*file->old.name &&
3878 (file->status == 'R' || file->status == 'C')) {
3879 sepsize = sep - buf + 1;
3880 string_ncopy(file->old.name, buf, sepsize);
3881 bufsize -= sepsize;
3882 memmove(buf, sep + 1, bufsize);
3884 sep = memchr(buf, 0, bufsize);
3885 if (!sep)
3886 break;
3887 sepsize = sep - buf + 1;
3888 }
3890 /* git-ls-files just delivers a NUL separated
3891 * list of file names similar to the second half
3892 * of the git-diff-* output. */
3893 string_ncopy(file->new.name, buf, sepsize);
3894 if (!*file->old.name)
3895 string_copy(file->old.name, file->new.name);
3896 bufsize -= sepsize;
3897 memmove(buf, sep + 1, bufsize);
3898 file = NULL;
3899 }
3900 }
3902 if (ferror(pipe)) {
3903 error_out:
3904 pclose(pipe);
3905 return FALSE;
3906 }
3908 if (!view->line[view->lines - 1].data)
3909 add_line_data(view, NULL, LINE_STAT_NONE);
3911 pclose(pipe);
3912 return TRUE;
3913 }
3915 /* Don't show unmerged entries in the staged section. */
3916 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --diff-filter=ACDMRTXB --cached -M HEAD"
3917 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
3918 #define STATUS_LIST_OTHER_CMD \
3919 "git ls-files -z --others --exclude-per-directory=.gitignore"
3920 #define STATUS_LIST_NO_HEAD_CMD \
3921 "git ls-files -z --cached --exclude-per-directory=.gitignore"
3923 #define STATUS_DIFF_INDEX_SHOW_CMD \
3924 "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s %s 2>/dev/null"
3926 #define STATUS_DIFF_FILES_SHOW_CMD \
3927 "git diff-files --root --patch-with-stat -C -M -- %s %s 2>/dev/null"
3929 #define STATUS_DIFF_NO_HEAD_SHOW_CMD \
3930 "git diff --no-color --patch-with-stat /dev/null %s 2>/dev/null"
3932 /* First parse staged info using git-diff-index(1), then parse unstaged
3933 * info using git-diff-files(1), and finally untracked files using
3934 * git-ls-files(1). */
3935 static bool
3936 status_open(struct view *view)
3937 {
3938 struct stat statbuf;
3939 char exclude[SIZEOF_STR];
3940 char indexcmd[SIZEOF_STR] = STATUS_DIFF_INDEX_CMD;
3941 char othercmd[SIZEOF_STR] = STATUS_LIST_OTHER_CMD;
3942 unsigned long prev_lineno = view->lineno;
3943 char indexstatus = 0;
3944 size_t i;
3946 for (i = 0; i < view->lines; i++)
3947 free(view->line[i].data);
3948 free(view->line);
3949 view->lines = view->line_alloc = view->line_size = view->lineno = 0;
3950 view->line = NULL;
3952 if (!realloc_lines(view, view->line_size + 7))
3953 return FALSE;
3955 add_line_data(view, NULL, LINE_STAT_HEAD);
3956 if (opt_no_head)
3957 string_copy(status_onbranch, "Initial commit");
3958 else if (!*opt_head)
3959 string_copy(status_onbranch, "Not currently on any branch");
3960 else if (!string_format(status_onbranch, "On branch %s", opt_head))
3961 return FALSE;
3963 if (opt_no_head) {
3964 string_copy(indexcmd, STATUS_LIST_NO_HEAD_CMD);
3965 indexstatus = 'A';
3966 }
3968 if (!string_format(exclude, "%s/info/exclude", opt_git_dir))
3969 return FALSE;
3971 if (stat(exclude, &statbuf) >= 0) {
3972 size_t cmdsize = strlen(othercmd);
3974 if (!string_format_from(othercmd, &cmdsize, " %s", "--exclude-from=") ||
3975 sq_quote(othercmd, cmdsize, exclude) >= sizeof(othercmd))
3976 return FALSE;
3978 cmdsize = strlen(indexcmd);
3979 if (opt_no_head &&
3980 (!string_format_from(indexcmd, &cmdsize, " %s", "--exclude-from=") ||
3981 sq_quote(indexcmd, cmdsize, exclude) >= sizeof(indexcmd)))
3982 return FALSE;
3983 }
3985 system("git update-index -q --refresh 2>/dev/null");
3987 if (!status_run(view, indexcmd, indexstatus, LINE_STAT_STAGED) ||
3988 !status_run(view, STATUS_DIFF_FILES_CMD, 0, LINE_STAT_UNSTAGED) ||
3989 !status_run(view, othercmd, '?', LINE_STAT_UNTRACKED))
3990 return FALSE;
3992 /* If all went well restore the previous line number to stay in
3993 * the context or select a line with something that can be
3994 * updated. */
3995 if (prev_lineno >= view->lines)
3996 prev_lineno = view->lines - 1;
3997 while (prev_lineno < view->lines && !view->line[prev_lineno].data)
3998 prev_lineno++;
3999 while (prev_lineno > 0 && !view->line[prev_lineno].data)
4000 prev_lineno--;
4002 /* If the above fails, always skip the "On branch" line. */
4003 if (prev_lineno < view->lines)
4004 view->lineno = prev_lineno;
4005 else
4006 view->lineno = 1;
4008 if (view->lineno < view->offset)
4009 view->offset = view->lineno;
4010 else if (view->offset + view->height <= view->lineno)
4011 view->offset = view->lineno - view->height + 1;
4013 return TRUE;
4014 }
4016 static bool
4017 status_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
4018 {
4019 struct status *status = line->data;
4020 char *text;
4021 int col = 0;
4023 if (selected) {
4024 /* No attributes. */
4026 } else if (line->type == LINE_STAT_HEAD) {
4027 wattrset(view->win, get_line_attr(LINE_STAT_HEAD));
4028 wchgat(view->win, -1, 0, LINE_STAT_HEAD, NULL);
4030 } else if (!status && line->type != LINE_STAT_NONE) {
4031 wattrset(view->win, get_line_attr(LINE_STAT_SECTION));
4032 wchgat(view->win, -1, 0, LINE_STAT_SECTION, NULL);
4034 } else {
4035 wattrset(view->win, get_line_attr(line->type));
4036 }
4038 if (!status) {
4039 switch (line->type) {
4040 case LINE_STAT_STAGED:
4041 text = "Changes to be committed:";
4042 break;
4044 case LINE_STAT_UNSTAGED:
4045 text = "Changed but not updated:";
4046 break;
4048 case LINE_STAT_UNTRACKED:
4049 text = "Untracked files:";
4050 break;
4052 case LINE_STAT_NONE:
4053 text = " (no files)";
4054 break;
4056 case LINE_STAT_HEAD:
4057 text = status_onbranch;
4058 break;
4060 default:
4061 return FALSE;
4062 }
4063 } else {
4064 char buf[] = { status->status, ' ', ' ', ' ', 0 };
4066 col += draw_text(view, buf, view->width, TRUE, selected);
4067 if (!selected)
4068 wattrset(view->win, A_NORMAL);
4069 text = status->new.name;
4070 }
4072 draw_text(view, text, view->width - col, TRUE, selected);
4073 return TRUE;
4074 }
4076 static enum request
4077 status_enter(struct view *view, struct line *line)
4078 {
4079 struct status *status = line->data;
4080 char oldpath[SIZEOF_STR] = "";
4081 char newpath[SIZEOF_STR] = "";
4082 char *info;
4083 size_t cmdsize = 0;
4084 enum open_flags split;
4086 if (line->type == LINE_STAT_NONE ||
4087 (!status && line[1].type == LINE_STAT_NONE)) {
4088 report("No file to diff");
4089 return REQ_NONE;
4090 }
4092 if (status) {
4093 if (sq_quote(oldpath, 0, status->old.name) >= sizeof(oldpath))
4094 return REQ_QUIT;
4095 /* Diffs for unmerged entries are empty when pasing the
4096 * new path, so leave it empty. */
4097 if (status->status != 'U' &&
4098 sq_quote(newpath, 0, status->new.name) >= sizeof(newpath))
4099 return REQ_QUIT;
4100 }
4102 if (opt_cdup[0] &&
4103 line->type != LINE_STAT_UNTRACKED &&
4104 !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
4105 return REQ_QUIT;
4107 switch (line->type) {
4108 case LINE_STAT_STAGED:
4109 if (opt_no_head) {
4110 if (!string_format_from(opt_cmd, &cmdsize,
4111 STATUS_DIFF_NO_HEAD_SHOW_CMD,
4112 newpath))
4113 return REQ_QUIT;
4114 } else {
4115 if (!string_format_from(opt_cmd, &cmdsize,
4116 STATUS_DIFF_INDEX_SHOW_CMD,
4117 oldpath, newpath))
4118 return REQ_QUIT;
4119 }
4121 if (status)
4122 info = "Staged changes to %s";
4123 else
4124 info = "Staged changes";
4125 break;
4127 case LINE_STAT_UNSTAGED:
4128 if (!string_format_from(opt_cmd, &cmdsize,
4129 STATUS_DIFF_FILES_SHOW_CMD, oldpath, newpath))
4130 return REQ_QUIT;
4131 if (status)
4132 info = "Unstaged changes to %s";
4133 else
4134 info = "Unstaged changes";
4135 break;
4137 case LINE_STAT_UNTRACKED:
4138 if (opt_pipe)
4139 return REQ_QUIT;
4141 if (!status) {
4142 report("No file to show");
4143 return REQ_NONE;
4144 }
4146 opt_pipe = fopen(status->new.name, "r");
4147 info = "Untracked file %s";
4148 break;
4150 case LINE_STAT_HEAD:
4151 return REQ_NONE;
4153 default:
4154 die("line type %d not handled in switch", line->type);
4155 }
4157 split = view_is_displayed(view) ? OPEN_SPLIT : 0;
4158 open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | split);
4159 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
4160 if (status) {
4161 stage_status = *status;
4162 } else {
4163 memset(&stage_status, 0, sizeof(stage_status));
4164 }
4166 stage_line_type = line->type;
4167 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
4168 }
4170 return REQ_NONE;
4171 }
4173 static bool
4174 status_exists(struct status *status, enum line_type type)
4175 {
4176 struct view *view = VIEW(REQ_VIEW_STATUS);
4177 struct line *line;
4179 for (line = view->line; line < view->line + view->lines; line++) {
4180 struct status *pos = line->data;
4182 if (line->type == type && pos &&
4183 !strcmp(status->new.name, pos->new.name))
4184 return TRUE;
4185 }
4187 return FALSE;
4188 }
4191 static FILE *
4192 status_update_prepare(enum line_type type)
4193 {
4194 char cmd[SIZEOF_STR];
4195 size_t cmdsize = 0;
4197 if (opt_cdup[0] &&
4198 type != LINE_STAT_UNTRACKED &&
4199 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4200 return NULL;
4202 switch (type) {
4203 case LINE_STAT_STAGED:
4204 string_add(cmd, cmdsize, "git update-index -z --index-info");
4205 break;
4207 case LINE_STAT_UNSTAGED:
4208 case LINE_STAT_UNTRACKED:
4209 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
4210 break;
4212 default:
4213 die("line type %d not handled in switch", type);
4214 }
4216 return popen(cmd, "w");
4217 }
4219 static bool
4220 status_update_write(FILE *pipe, struct status *status, enum line_type type)
4221 {
4222 char buf[SIZEOF_STR];
4223 size_t bufsize = 0;
4224 size_t written = 0;
4226 switch (type) {
4227 case LINE_STAT_STAGED:
4228 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
4229 status->old.mode,
4230 status->old.rev,
4231 status->old.name, 0))
4232 return FALSE;
4233 break;
4235 case LINE_STAT_UNSTAGED:
4236 case LINE_STAT_UNTRACKED:
4237 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
4238 return FALSE;
4239 break;
4241 default:
4242 die("line type %d not handled in switch", type);
4243 }
4245 while (!ferror(pipe) && written < bufsize) {
4246 written += fwrite(buf + written, 1, bufsize - written, pipe);
4247 }
4249 return written == bufsize;
4250 }
4252 static bool
4253 status_update_file(struct status *status, enum line_type type)
4254 {
4255 FILE *pipe = status_update_prepare(type);
4256 bool result;
4258 if (!pipe)
4259 return FALSE;
4261 result = status_update_write(pipe, status, type);
4262 pclose(pipe);
4263 return result;
4264 }
4266 static bool
4267 status_update_files(struct view *view, struct line *line)
4268 {
4269 FILE *pipe = status_update_prepare(line->type);
4270 bool result = TRUE;
4271 struct line *pos = view->line + view->lines;
4272 int files = 0;
4273 int file, done;
4275 if (!pipe)
4276 return FALSE;
4278 for (pos = line; pos < view->line + view->lines && pos->data; pos++)
4279 files++;
4281 for (file = 0, done = 0; result && file < files; line++, file++) {
4282 int almost_done = file * 100 / files;
4284 if (almost_done > done) {
4285 done = almost_done;
4286 string_format(view->ref, "updating file %u of %u (%d%% done)",
4287 file, files, done);
4288 update_view_title(view);
4289 }
4290 result = status_update_write(pipe, line->data, line->type);
4291 }
4293 pclose(pipe);
4294 return result;
4295 }
4297 static bool
4298 status_update(struct view *view)
4299 {
4300 struct line *line = &view->line[view->lineno];
4302 assert(view->lines);
4304 if (!line->data) {
4305 /* This should work even for the "On branch" line. */
4306 if (line < view->line + view->lines && !line[1].data) {
4307 report("Nothing to update");
4308 return FALSE;
4309 }
4311 if (!status_update_files(view, line + 1))
4312 report("Failed to update file status");
4314 } else if (!status_update_file(line->data, line->type)) {
4315 report("Failed to update file status");
4316 }
4318 return TRUE;
4319 }
4321 static enum request
4322 status_request(struct view *view, enum request request, struct line *line)
4323 {
4324 struct status *status = line->data;
4326 switch (request) {
4327 case REQ_STATUS_UPDATE:
4328 if (!status_update(view))
4329 return REQ_NONE;
4330 break;
4332 case REQ_STATUS_MERGE:
4333 if (!status || status->status != 'U') {
4334 report("Merging only possible for files with unmerged status ('U').");
4335 return REQ_NONE;
4336 }
4337 open_mergetool(status->new.name);
4338 break;
4340 case REQ_EDIT:
4341 if (!status)
4342 return request;
4344 open_editor(status->status != '?', status->new.name);
4345 break;
4347 case REQ_VIEW_BLAME:
4348 if (status) {
4349 string_copy(opt_file, status->new.name);
4350 opt_ref[0] = 0;
4351 }
4352 return request;
4354 case REQ_ENTER:
4355 /* After returning the status view has been split to
4356 * show the stage view. No further reloading is
4357 * necessary. */
4358 status_enter(view, line);
4359 return REQ_NONE;
4361 case REQ_REFRESH:
4362 /* Simply reload the view. */
4363 break;
4365 default:
4366 return request;
4367 }
4369 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
4371 return REQ_NONE;
4372 }
4374 static void
4375 status_select(struct view *view, struct line *line)
4376 {
4377 struct status *status = line->data;
4378 char file[SIZEOF_STR] = "all files";
4379 char *text;
4380 char *key;
4382 if (status && !string_format(file, "'%s'", status->new.name))
4383 return;
4385 if (!status && line[1].type == LINE_STAT_NONE)
4386 line++;
4388 switch (line->type) {
4389 case LINE_STAT_STAGED:
4390 text = "Press %s to unstage %s for commit";
4391 break;
4393 case LINE_STAT_UNSTAGED:
4394 text = "Press %s to stage %s for commit";
4395 break;
4397 case LINE_STAT_UNTRACKED:
4398 text = "Press %s to stage %s for addition";
4399 break;
4401 case LINE_STAT_HEAD:
4402 case LINE_STAT_NONE:
4403 text = "Nothing to update";
4404 break;
4406 default:
4407 die("line type %d not handled in switch", line->type);
4408 }
4410 if (status && status->status == 'U') {
4411 text = "Press %s to resolve conflict in %s";
4412 key = get_key(REQ_STATUS_MERGE);
4414 } else {
4415 key = get_key(REQ_STATUS_UPDATE);
4416 }
4418 string_format(view->ref, text, key, file);
4419 }
4421 static bool
4422 status_grep(struct view *view, struct line *line)
4423 {
4424 struct status *status = line->data;
4425 enum { S_STATUS, S_NAME, S_END } state;
4426 char buf[2] = "?";
4427 regmatch_t pmatch;
4429 if (!status)
4430 return FALSE;
4432 for (state = S_STATUS; state < S_END; state++) {
4433 char *text;
4435 switch (state) {
4436 case S_NAME: text = status->new.name; break;
4437 case S_STATUS:
4438 buf[0] = status->status;
4439 text = buf;
4440 break;
4442 default:
4443 return FALSE;
4444 }
4446 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4447 return TRUE;
4448 }
4450 return FALSE;
4451 }
4453 static struct view_ops status_ops = {
4454 "file",
4455 status_open,
4456 NULL,
4457 status_draw,
4458 status_request,
4459 status_grep,
4460 status_select,
4461 };
4464 static bool
4465 stage_diff_line(FILE *pipe, struct line *line)
4466 {
4467 char *buf = line->data;
4468 size_t bufsize = strlen(buf);
4469 size_t written = 0;
4471 while (!ferror(pipe) && written < bufsize) {
4472 written += fwrite(buf + written, 1, bufsize - written, pipe);
4473 }
4475 fputc('\n', pipe);
4477 return written == bufsize;
4478 }
4480 static bool
4481 stage_diff_write(FILE *pipe, struct line *line, struct line *end)
4482 {
4483 while (line < end) {
4484 if (!stage_diff_line(pipe, line++))
4485 return FALSE;
4486 if (line->type == LINE_DIFF_CHUNK ||
4487 line->type == LINE_DIFF_HEADER)
4488 break;
4489 }
4491 return TRUE;
4492 }
4494 static struct line *
4495 stage_diff_find(struct view *view, struct line *line, enum line_type type)
4496 {
4497 for (; view->line < line; line--)
4498 if (line->type == type)
4499 return line;
4501 return NULL;
4502 }
4504 static bool
4505 stage_update_chunk(struct view *view, struct line *chunk)
4506 {
4507 char cmd[SIZEOF_STR];
4508 size_t cmdsize = 0;
4509 struct line *diff_hdr;
4510 FILE *pipe;
4512 diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
4513 if (!diff_hdr)
4514 return FALSE;
4516 if (opt_cdup[0] &&
4517 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4518 return FALSE;
4520 if (!string_format_from(cmd, &cmdsize,
4521 "git apply --whitespace=nowarn --cached %s - && "
4522 "git update-index -q --unmerged --refresh 2>/dev/null",
4523 stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
4524 return FALSE;
4526 pipe = popen(cmd, "w");
4527 if (!pipe)
4528 return FALSE;
4530 if (!stage_diff_write(pipe, diff_hdr, chunk) ||
4531 !stage_diff_write(pipe, chunk, view->line + view->lines))
4532 chunk = NULL;
4534 pclose(pipe);
4536 return chunk ? TRUE : FALSE;
4537 }
4539 static bool
4540 stage_update(struct view *view, struct line *line)
4541 {
4542 struct line *chunk = NULL;
4544 if (!opt_no_head && stage_line_type != LINE_STAT_UNTRACKED)
4545 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4547 if (chunk) {
4548 if (!stage_update_chunk(view, chunk)) {
4549 report("Failed to apply chunk");
4550 return FALSE;
4551 }
4553 } else if (!status_update_file(&stage_status, stage_line_type)) {
4554 report("Failed to update file");
4555 return FALSE;
4556 }
4558 return TRUE;
4559 }
4561 static enum request
4562 stage_request(struct view *view, enum request request, struct line *line)
4563 {
4564 switch (request) {
4565 case REQ_STATUS_UPDATE:
4566 stage_update(view, line);
4567 break;
4569 case REQ_EDIT:
4570 if (!stage_status.new.name[0])
4571 return request;
4573 open_editor(stage_status.status != '?', stage_status.new.name);
4574 break;
4576 case REQ_REFRESH:
4577 /* Reload everything ... */
4578 break;
4580 case REQ_VIEW_BLAME:
4581 if (stage_status.new.name[0]) {
4582 string_copy(opt_file, stage_status.new.name);
4583 opt_ref[0] = 0;
4584 }
4585 return request;
4587 case REQ_ENTER:
4588 return pager_request(view, request, line);
4590 default:
4591 return request;
4592 }
4594 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD | OPEN_NOMAXIMIZE);
4596 /* Check whether the staged entry still exists, and close the
4597 * stage view if it doesn't. */
4598 if (!status_exists(&stage_status, stage_line_type))
4599 return REQ_VIEW_CLOSE;
4601 if (stage_line_type == LINE_STAT_UNTRACKED)
4602 opt_pipe = fopen(stage_status.new.name, "r");
4603 else
4604 string_copy(opt_cmd, view->cmd);
4605 open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | OPEN_NOMAXIMIZE);
4607 return REQ_NONE;
4608 }
4610 static struct view_ops stage_ops = {
4611 "line",
4612 NULL,
4613 pager_read,
4614 pager_draw,
4615 stage_request,
4616 pager_grep,
4617 pager_select,
4618 };
4621 /*
4622 * Revision graph
4623 */
4625 struct commit {
4626 char id[SIZEOF_REV]; /* SHA1 ID. */
4627 char title[128]; /* First line of the commit message. */
4628 char author[75]; /* Author of the commit. */
4629 struct tm time; /* Date from the author ident. */
4630 struct ref **refs; /* Repository references. */
4631 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
4632 size_t graph_size; /* The width of the graph array. */
4633 bool has_parents; /* Rewritten --parents seen. */
4634 };
4636 /* Size of rev graph with no "padding" columns */
4637 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
4639 struct rev_graph {
4640 struct rev_graph *prev, *next, *parents;
4641 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
4642 size_t size;
4643 struct commit *commit;
4644 size_t pos;
4645 unsigned int boundary:1;
4646 };
4648 /* Parents of the commit being visualized. */
4649 static struct rev_graph graph_parents[4];
4651 /* The current stack of revisions on the graph. */
4652 static struct rev_graph graph_stacks[4] = {
4653 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
4654 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
4655 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
4656 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
4657 };
4659 static inline bool
4660 graph_parent_is_merge(struct rev_graph *graph)
4661 {
4662 return graph->parents->size > 1;
4663 }
4665 static inline void
4666 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
4667 {
4668 struct commit *commit = graph->commit;
4670 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
4671 commit->graph[commit->graph_size++] = symbol;
4672 }
4674 static void
4675 done_rev_graph(struct rev_graph *graph)
4676 {
4677 if (graph_parent_is_merge(graph) &&
4678 graph->pos < graph->size - 1 &&
4679 graph->next->size == graph->size + graph->parents->size - 1) {
4680 size_t i = graph->pos + graph->parents->size - 1;
4682 graph->commit->graph_size = i * 2;
4683 while (i < graph->next->size - 1) {
4684 append_to_rev_graph(graph, ' ');
4685 append_to_rev_graph(graph, '\\');
4686 i++;
4687 }
4688 }
4690 graph->size = graph->pos = 0;
4691 graph->commit = NULL;
4692 memset(graph->parents, 0, sizeof(*graph->parents));
4693 }
4695 static void
4696 push_rev_graph(struct rev_graph *graph, char *parent)
4697 {
4698 int i;
4700 /* "Collapse" duplicate parents lines.
4701 *
4702 * FIXME: This needs to also update update the drawn graph but
4703 * for now it just serves as a method for pruning graph lines. */
4704 for (i = 0; i < graph->size; i++)
4705 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
4706 return;
4708 if (graph->size < SIZEOF_REVITEMS) {
4709 string_copy_rev(graph->rev[graph->size++], parent);
4710 }
4711 }
4713 static chtype
4714 get_rev_graph_symbol(struct rev_graph *graph)
4715 {
4716 chtype symbol;
4718 if (graph->boundary)
4719 symbol = REVGRAPH_BOUND;
4720 else if (graph->parents->size == 0)
4721 symbol = REVGRAPH_INIT;
4722 else if (graph_parent_is_merge(graph))
4723 symbol = REVGRAPH_MERGE;
4724 else if (graph->pos >= graph->size)
4725 symbol = REVGRAPH_BRANCH;
4726 else
4727 symbol = REVGRAPH_COMMIT;
4729 return symbol;
4730 }
4732 static void
4733 draw_rev_graph(struct rev_graph *graph)
4734 {
4735 struct rev_filler {
4736 chtype separator, line;
4737 };
4738 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
4739 static struct rev_filler fillers[] = {
4740 { ' ', REVGRAPH_LINE },
4741 { '`', '.' },
4742 { '\'', ' ' },
4743 { '/', ' ' },
4744 };
4745 chtype symbol = get_rev_graph_symbol(graph);
4746 struct rev_filler *filler;
4747 size_t i;
4749 filler = &fillers[DEFAULT];
4751 for (i = 0; i < graph->pos; i++) {
4752 append_to_rev_graph(graph, filler->line);
4753 if (graph_parent_is_merge(graph->prev) &&
4754 graph->prev->pos == i)
4755 filler = &fillers[RSHARP];
4757 append_to_rev_graph(graph, filler->separator);
4758 }
4760 /* Place the symbol for this revision. */
4761 append_to_rev_graph(graph, symbol);
4763 if (graph->prev->size > graph->size)
4764 filler = &fillers[RDIAG];
4765 else
4766 filler = &fillers[DEFAULT];
4768 i++;
4770 for (; i < graph->size; i++) {
4771 append_to_rev_graph(graph, filler->separator);
4772 append_to_rev_graph(graph, filler->line);
4773 if (graph_parent_is_merge(graph->prev) &&
4774 i < graph->prev->pos + graph->parents->size)
4775 filler = &fillers[RSHARP];
4776 if (graph->prev->size > graph->size)
4777 filler = &fillers[LDIAG];
4778 }
4780 if (graph->prev->size > graph->size) {
4781 append_to_rev_graph(graph, filler->separator);
4782 if (filler->line != ' ')
4783 append_to_rev_graph(graph, filler->line);
4784 }
4785 }
4787 /* Prepare the next rev graph */
4788 static void
4789 prepare_rev_graph(struct rev_graph *graph)
4790 {
4791 size_t i;
4793 /* First, traverse all lines of revisions up to the active one. */
4794 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
4795 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
4796 break;
4798 push_rev_graph(graph->next, graph->rev[graph->pos]);
4799 }
4801 /* Interleave the new revision parent(s). */
4802 for (i = 0; !graph->boundary && i < graph->parents->size; i++)
4803 push_rev_graph(graph->next, graph->parents->rev[i]);
4805 /* Lastly, put any remaining revisions. */
4806 for (i = graph->pos + 1; i < graph->size; i++)
4807 push_rev_graph(graph->next, graph->rev[i]);
4808 }
4810 static void
4811 update_rev_graph(struct rev_graph *graph)
4812 {
4813 /* If this is the finalizing update ... */
4814 if (graph->commit)
4815 prepare_rev_graph(graph);
4817 /* Graph visualization needs a one rev look-ahead,
4818 * so the first update doesn't visualize anything. */
4819 if (!graph->prev->commit)
4820 return;
4822 draw_rev_graph(graph->prev);
4823 done_rev_graph(graph->prev->prev);
4824 }
4827 /*
4828 * Main view backend
4829 */
4831 static bool
4832 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
4833 {
4834 struct commit *commit = line->data;
4835 enum line_type type;
4836 int col = 0;
4838 if (!*commit->author)
4839 return FALSE;
4841 if (selected) {
4842 type = LINE_CURSOR;
4843 } else {
4844 type = LINE_MAIN_COMMIT;
4845 }
4847 if (opt_date) {
4848 col += draw_date(view, &commit->time, view->width, selected);
4849 if (col >= view->width)
4850 return TRUE;
4851 }
4852 if (type != LINE_CURSOR)
4853 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
4855 if (opt_author) {
4856 int max_len;
4858 max_len = view->width - col;
4859 if (max_len > AUTHOR_COLS - 1)
4860 max_len = AUTHOR_COLS - 1;
4861 draw_text(view, commit->author, max_len, TRUE, selected);
4862 col += AUTHOR_COLS;
4863 if (col >= view->width)
4864 return TRUE;
4865 }
4867 if (opt_rev_graph && commit->graph_size) {
4868 size_t graph_size = view->width - col;
4869 size_t i;
4871 if (type != LINE_CURSOR)
4872 wattrset(view->win, get_line_attr(LINE_MAIN_REVGRAPH));
4873 wmove(view->win, lineno, col);
4874 if (graph_size > commit->graph_size)
4875 graph_size = commit->graph_size;
4876 /* Using waddch() instead of waddnstr() ensures that
4877 * they'll be rendered correctly for the cursor line. */
4878 for (i = 0; i < graph_size; i++)
4879 waddch(view->win, commit->graph[i]);
4881 col += commit->graph_size + 1;
4882 if (col >= view->width)
4883 return TRUE;
4884 waddch(view->win, ' ');
4885 }
4886 if (type != LINE_CURSOR)
4887 wattrset(view->win, A_NORMAL);
4889 wmove(view->win, lineno, col);
4891 if (opt_show_refs && commit->refs) {
4892 size_t i = 0;
4894 do {
4895 if (type == LINE_CURSOR)
4896 ;
4897 else if (commit->refs[i]->head)
4898 wattrset(view->win, get_line_attr(LINE_MAIN_HEAD));
4899 else if (commit->refs[i]->ltag)
4900 wattrset(view->win, get_line_attr(LINE_MAIN_LOCAL_TAG));
4901 else if (commit->refs[i]->tag)
4902 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
4903 else if (commit->refs[i]->tracked)
4904 wattrset(view->win, get_line_attr(LINE_MAIN_TRACKED));
4905 else if (commit->refs[i]->remote)
4906 wattrset(view->win, get_line_attr(LINE_MAIN_REMOTE));
4907 else
4908 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
4910 col += draw_text(view, "[", view->width - col, TRUE, selected);
4911 col += draw_text(view, commit->refs[i]->name, view->width - col,
4912 TRUE, selected);
4913 col += draw_text(view, "]", view->width - col, TRUE, selected);
4914 if (type != LINE_CURSOR)
4915 wattrset(view->win, A_NORMAL);
4916 col += draw_text(view, " ", view->width - col, TRUE, selected);
4917 if (col >= view->width)
4918 return TRUE;
4919 } while (commit->refs[i++]->next);
4920 }
4922 if (type != LINE_CURSOR)
4923 wattrset(view->win, get_line_attr(type));
4925 draw_text(view, commit->title, view->width - col, TRUE, selected);
4926 return TRUE;
4927 }
4929 /* Reads git log --pretty=raw output and parses it into the commit struct. */
4930 static bool
4931 main_read(struct view *view, char *line)
4932 {
4933 static struct rev_graph *graph = graph_stacks;
4934 enum line_type type;
4935 struct commit *commit;
4937 if (!line) {
4938 if (!view->lines && !view->parent)
4939 die("No revisions match the given arguments.");
4940 update_rev_graph(graph);
4941 return TRUE;
4942 }
4944 type = get_line_type(line);
4945 if (type == LINE_COMMIT) {
4946 commit = calloc(1, sizeof(struct commit));
4947 if (!commit)
4948 return FALSE;
4950 line += STRING_SIZE("commit ");
4951 if (*line == '-') {
4952 graph->boundary = 1;
4953 line++;
4954 }
4956 string_copy_rev(commit->id, line);
4957 commit->refs = get_refs(commit->id);
4958 graph->commit = commit;
4959 add_line_data(view, commit, LINE_MAIN_COMMIT);
4961 while ((line = strchr(line, ' '))) {
4962 line++;
4963 push_rev_graph(graph->parents, line);
4964 commit->has_parents = TRUE;
4965 }
4966 return TRUE;
4967 }
4969 if (!view->lines)
4970 return TRUE;
4971 commit = view->line[view->lines - 1].data;
4973 switch (type) {
4974 case LINE_PARENT:
4975 if (commit->has_parents)
4976 break;
4977 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
4978 break;
4980 case LINE_AUTHOR:
4981 {
4982 /* Parse author lines where the name may be empty:
4983 * author <email@address.tld> 1138474660 +0100
4984 */
4985 char *ident = line + STRING_SIZE("author ");
4986 char *nameend = strchr(ident, '<');
4987 char *emailend = strchr(ident, '>');
4989 if (!nameend || !emailend)
4990 break;
4992 update_rev_graph(graph);
4993 graph = graph->next;
4995 *nameend = *emailend = 0;
4996 ident = chomp_string(ident);
4997 if (!*ident) {
4998 ident = chomp_string(nameend + 1);
4999 if (!*ident)
5000 ident = "Unknown";
5001 }
5003 string_ncopy(commit->author, ident, strlen(ident));
5005 /* Parse epoch and timezone */
5006 if (emailend[1] == ' ') {
5007 char *secs = emailend + 2;
5008 char *zone = strchr(secs, ' ');
5009 time_t time = (time_t) atol(secs);
5011 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
5012 long tz;
5014 zone++;
5015 tz = ('0' - zone[1]) * 60 * 60 * 10;
5016 tz += ('0' - zone[2]) * 60 * 60;
5017 tz += ('0' - zone[3]) * 60;
5018 tz += ('0' - zone[4]) * 60;
5020 if (zone[0] == '-')
5021 tz = -tz;
5023 time -= tz;
5024 }
5026 gmtime_r(&time, &commit->time);
5027 }
5028 break;
5029 }
5030 default:
5031 /* Fill in the commit title if it has not already been set. */
5032 if (commit->title[0])
5033 break;
5035 /* Require titles to start with a non-space character at the
5036 * offset used by git log. */
5037 if (strncmp(line, " ", 4))
5038 break;
5039 line += 4;
5040 /* Well, if the title starts with a whitespace character,
5041 * try to be forgiving. Otherwise we end up with no title. */
5042 while (isspace(*line))
5043 line++;
5044 if (*line == '\0')
5045 break;
5046 /* FIXME: More graceful handling of titles; append "..." to
5047 * shortened titles, etc. */
5049 string_ncopy(commit->title, line, strlen(line));
5050 }
5052 return TRUE;
5053 }
5055 static enum request
5056 main_request(struct view *view, enum request request, struct line *line)
5057 {
5058 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5060 if (request == REQ_ENTER)
5061 open_view(view, REQ_VIEW_DIFF, flags);
5062 else
5063 return request;
5065 return REQ_NONE;
5066 }
5068 static bool
5069 main_grep(struct view *view, struct line *line)
5070 {
5071 struct commit *commit = line->data;
5072 enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
5073 char buf[DATE_COLS + 1];
5074 regmatch_t pmatch;
5076 for (state = S_TITLE; state < S_END; state++) {
5077 char *text;
5079 switch (state) {
5080 case S_TITLE: text = commit->title; break;
5081 case S_AUTHOR: text = commit->author; break;
5082 case S_DATE:
5083 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
5084 continue;
5085 text = buf;
5086 break;
5088 default:
5089 return FALSE;
5090 }
5092 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5093 return TRUE;
5094 }
5096 return FALSE;
5097 }
5099 static void
5100 main_select(struct view *view, struct line *line)
5101 {
5102 struct commit *commit = line->data;
5104 string_copy_rev(view->ref, commit->id);
5105 string_copy_rev(ref_commit, view->ref);
5106 }
5108 static struct view_ops main_ops = {
5109 "commit",
5110 NULL,
5111 main_read,
5112 main_draw,
5113 main_request,
5114 main_grep,
5115 main_select,
5116 };
5119 /*
5120 * Unicode / UTF-8 handling
5121 *
5122 * NOTE: Much of the following code for dealing with unicode is derived from
5123 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
5124 * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
5125 */
5127 /* I've (over)annotated a lot of code snippets because I am not entirely
5128 * confident that the approach taken by this small UTF-8 interface is correct.
5129 * --jonas */
5131 static inline int
5132 unicode_width(unsigned long c)
5133 {
5134 if (c >= 0x1100 &&
5135 (c <= 0x115f /* Hangul Jamo */
5136 || c == 0x2329
5137 || c == 0x232a
5138 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
5139 /* CJK ... Yi */
5140 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
5141 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
5142 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
5143 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
5144 || (c >= 0xffe0 && c <= 0xffe6)
5145 || (c >= 0x20000 && c <= 0x2fffd)
5146 || (c >= 0x30000 && c <= 0x3fffd)))
5147 return 2;
5149 if (c == '\t')
5150 return opt_tab_size;
5152 return 1;
5153 }
5155 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
5156 * Illegal bytes are set one. */
5157 static const unsigned char utf8_bytes[256] = {
5158 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
5159 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
5160 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
5161 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
5162 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
5163 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
5164 2,2,2,2,2,2,2,2, 2,2,2,2,2,2,2,2, 2,2,2,2,2,2,2,2, 2,2,2,2,2,2,2,2,
5165 3,3,3,3,3,3,3,3, 3,3,3,3,3,3,3,3, 4,4,4,4,4,4,4,4, 5,5,5,5,6,6,1,1,
5166 };
5168 /* Decode UTF-8 multi-byte representation into a unicode character. */
5169 static inline unsigned long
5170 utf8_to_unicode(const char *string, size_t length)
5171 {
5172 unsigned long unicode;
5174 switch (length) {
5175 case 1:
5176 unicode = string[0];
5177 break;
5178 case 2:
5179 unicode = (string[0] & 0x1f) << 6;
5180 unicode += (string[1] & 0x3f);
5181 break;
5182 case 3:
5183 unicode = (string[0] & 0x0f) << 12;
5184 unicode += ((string[1] & 0x3f) << 6);
5185 unicode += (string[2] & 0x3f);
5186 break;
5187 case 4:
5188 unicode = (string[0] & 0x0f) << 18;
5189 unicode += ((string[1] & 0x3f) << 12);
5190 unicode += ((string[2] & 0x3f) << 6);
5191 unicode += (string[3] & 0x3f);
5192 break;
5193 case 5:
5194 unicode = (string[0] & 0x0f) << 24;
5195 unicode += ((string[1] & 0x3f) << 18);
5196 unicode += ((string[2] & 0x3f) << 12);
5197 unicode += ((string[3] & 0x3f) << 6);
5198 unicode += (string[4] & 0x3f);
5199 break;
5200 case 6:
5201 unicode = (string[0] & 0x01) << 30;
5202 unicode += ((string[1] & 0x3f) << 24);
5203 unicode += ((string[2] & 0x3f) << 18);
5204 unicode += ((string[3] & 0x3f) << 12);
5205 unicode += ((string[4] & 0x3f) << 6);
5206 unicode += (string[5] & 0x3f);
5207 break;
5208 default:
5209 die("Invalid unicode length");
5210 }
5212 /* Invalid characters could return the special 0xfffd value but NUL
5213 * should be just as good. */
5214 return unicode > 0xffff ? 0 : unicode;
5215 }
5217 /* Calculates how much of string can be shown within the given maximum width
5218 * and sets trimmed parameter to non-zero value if all of string could not be
5219 * shown. If the reserve flag is TRUE, it will reserve at least one
5220 * trailing character, which can be useful when drawing a delimiter.
5221 *
5222 * Returns the number of bytes to output from string to satisfy max_width. */
5223 static size_t
5224 utf8_length(const char *string, size_t max_width, int *trimmed, bool reserve)
5225 {
5226 const char *start = string;
5227 const char *end = strchr(string, '\0');
5228 unsigned char last_bytes = 0;
5229 size_t width = 0;
5231 *trimmed = 0;
5233 while (string < end) {
5234 int c = *(unsigned char *) string;
5235 unsigned char bytes = utf8_bytes[c];
5236 size_t ucwidth;
5237 unsigned long unicode;
5239 if (string + bytes > end)
5240 break;
5242 /* Change representation to figure out whether
5243 * it is a single- or double-width character. */
5245 unicode = utf8_to_unicode(string, bytes);
5246 /* FIXME: Graceful handling of invalid unicode character. */
5247 if (!unicode)
5248 break;
5250 ucwidth = unicode_width(unicode);
5251 width += ucwidth;
5252 if (width > max_width) {
5253 *trimmed = 1;
5254 if (reserve && width - ucwidth == max_width) {
5255 string -= last_bytes;
5256 }
5257 break;
5258 }
5260 string += bytes;
5261 last_bytes = bytes;
5262 }
5264 return string - start;
5265 }
5268 /*
5269 * Status management
5270 */
5272 /* Whether or not the curses interface has been initialized. */
5273 static bool cursed = FALSE;
5275 /* The status window is used for polling keystrokes. */
5276 static WINDOW *status_win;
5278 static bool status_empty = TRUE;
5280 /* Update status and title window. */
5281 static void
5282 report(const char *msg, ...)
5283 {
5284 struct view *view = display[current_view];
5286 if (input_mode)
5287 return;
5289 if (!view) {
5290 char buf[SIZEOF_STR];
5291 va_list args;
5293 va_start(args, msg);
5294 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
5295 buf[sizeof(buf) - 1] = 0;
5296 buf[sizeof(buf) - 2] = '.';
5297 buf[sizeof(buf) - 3] = '.';
5298 buf[sizeof(buf) - 4] = '.';
5299 }
5300 va_end(args);
5301 die("%s", buf);
5302 }
5304 if (!status_empty || *msg) {
5305 va_list args;
5307 va_start(args, msg);
5309 wmove(status_win, 0, 0);
5310 if (*msg) {
5311 vwprintw(status_win, msg, args);
5312 status_empty = FALSE;
5313 } else {
5314 status_empty = TRUE;
5315 }
5316 wclrtoeol(status_win);
5317 wrefresh(status_win);
5319 va_end(args);
5320 }
5322 update_view_title(view);
5323 update_display_cursor(view);
5324 }
5326 /* Controls when nodelay should be in effect when polling user input. */
5327 static void
5328 set_nonblocking_input(bool loading)
5329 {
5330 static unsigned int loading_views;
5332 if ((loading == FALSE && loading_views-- == 1) ||
5333 (loading == TRUE && loading_views++ == 0))
5334 nodelay(status_win, loading);
5335 }
5337 static void
5338 init_display(void)
5339 {
5340 int x, y;
5342 /* Initialize the curses library */
5343 if (isatty(STDIN_FILENO)) {
5344 cursed = !!initscr();
5345 } else {
5346 /* Leave stdin and stdout alone when acting as a pager. */
5347 FILE *io = fopen("/dev/tty", "r+");
5349 if (!io)
5350 die("Failed to open /dev/tty");
5351 cursed = !!newterm(NULL, io, io);
5352 }
5354 if (!cursed)
5355 die("Failed to initialize curses");
5357 nonl(); /* Tell curses not to do NL->CR/NL on output */
5358 cbreak(); /* Take input chars one at a time, no wait for \n */
5359 noecho(); /* Don't echo input */
5360 leaveok(stdscr, TRUE);
5362 if (has_colors())
5363 init_colors();
5365 getmaxyx(stdscr, y, x);
5366 status_win = newwin(1, 0, y - 1, 0);
5367 if (!status_win)
5368 die("Failed to create status window");
5370 /* Enable keyboard mapping */
5371 keypad(status_win, TRUE);
5372 wbkgdset(status_win, get_line_attr(LINE_STATUS));
5373 }
5375 static char *
5376 read_prompt(const char *prompt)
5377 {
5378 enum { READING, STOP, CANCEL } status = READING;
5379 static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
5380 int pos = 0;
5382 while (status == READING) {
5383 struct view *view;
5384 int i, key;
5386 input_mode = TRUE;
5388 foreach_view (view, i)
5389 update_view(view);
5391 input_mode = FALSE;
5393 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
5394 wclrtoeol(status_win);
5396 /* Refresh, accept single keystroke of input */
5397 key = wgetch(status_win);
5398 switch (key) {
5399 case KEY_RETURN:
5400 case KEY_ENTER:
5401 case '\n':
5402 status = pos ? STOP : CANCEL;
5403 break;
5405 case KEY_BACKSPACE:
5406 if (pos > 0)
5407 pos--;
5408 else
5409 status = CANCEL;
5410 break;
5412 case KEY_ESC:
5413 status = CANCEL;
5414 break;
5416 case ERR:
5417 break;
5419 default:
5420 if (pos >= sizeof(buf)) {
5421 report("Input string too long");
5422 return NULL;
5423 }
5425 if (isprint(key))
5426 buf[pos++] = (char) key;
5427 }
5428 }
5430 /* Clear the status window */
5431 status_empty = FALSE;
5432 report("");
5434 if (status == CANCEL)
5435 return NULL;
5437 buf[pos++] = 0;
5439 return buf;
5440 }
5442 /*
5443 * Repository references
5444 */
5446 static struct ref *refs = NULL;
5447 static size_t refs_alloc = 0;
5448 static size_t refs_size = 0;
5450 /* Id <-> ref store */
5451 static struct ref ***id_refs = NULL;
5452 static size_t id_refs_alloc = 0;
5453 static size_t id_refs_size = 0;
5455 static struct ref **
5456 get_refs(char *id)
5457 {
5458 struct ref ***tmp_id_refs;
5459 struct ref **ref_list = NULL;
5460 size_t ref_list_alloc = 0;
5461 size_t ref_list_size = 0;
5462 size_t i;
5464 for (i = 0; i < id_refs_size; i++)
5465 if (!strcmp(id, id_refs[i][0]->id))
5466 return id_refs[i];
5468 tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
5469 sizeof(*id_refs));
5470 if (!tmp_id_refs)
5471 return NULL;
5473 id_refs = tmp_id_refs;
5475 for (i = 0; i < refs_size; i++) {
5476 struct ref **tmp;
5478 if (strcmp(id, refs[i].id))
5479 continue;
5481 tmp = realloc_items(ref_list, &ref_list_alloc,
5482 ref_list_size + 1, sizeof(*ref_list));
5483 if (!tmp) {
5484 if (ref_list)
5485 free(ref_list);
5486 return NULL;
5487 }
5489 ref_list = tmp;
5490 if (ref_list_size > 0)
5491 ref_list[ref_list_size - 1]->next = 1;
5492 ref_list[ref_list_size] = &refs[i];
5494 /* XXX: The properties of the commit chains ensures that we can
5495 * safely modify the shared ref. The repo references will
5496 * always be similar for the same id. */
5497 ref_list[ref_list_size]->next = 0;
5498 ref_list_size++;
5499 }
5501 if (ref_list)
5502 id_refs[id_refs_size++] = ref_list;
5504 return ref_list;
5505 }
5507 static int
5508 read_ref(char *id, size_t idlen, char *name, size_t namelen)
5509 {
5510 struct ref *ref;
5511 bool tag = FALSE;
5512 bool ltag = FALSE;
5513 bool remote = FALSE;
5514 bool tracked = FALSE;
5515 bool check_replace = FALSE;
5516 bool head = FALSE;
5518 if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
5519 if (!strcmp(name + namelen - 3, "^{}")) {
5520 namelen -= 3;
5521 name[namelen] = 0;
5522 if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
5523 check_replace = TRUE;
5524 } else {
5525 ltag = TRUE;
5526 }
5528 tag = TRUE;
5529 namelen -= STRING_SIZE("refs/tags/");
5530 name += STRING_SIZE("refs/tags/");
5532 } else if (!strncmp(name, "refs/remotes/", STRING_SIZE("refs/remotes/"))) {
5533 remote = TRUE;
5534 namelen -= STRING_SIZE("refs/remotes/");
5535 name += STRING_SIZE("refs/remotes/");
5536 tracked = !strcmp(opt_remote, name);
5538 } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
5539 namelen -= STRING_SIZE("refs/heads/");
5540 name += STRING_SIZE("refs/heads/");
5541 head = !strncmp(opt_head, name, namelen);
5543 } else if (!strcmp(name, "HEAD")) {
5544 opt_no_head = FALSE;
5545 return OK;
5546 }
5548 if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
5549 /* it's an annotated tag, replace the previous sha1 with the
5550 * resolved commit id; relies on the fact git-ls-remote lists
5551 * the commit id of an annotated tag right beofre the commit id
5552 * it points to. */
5553 refs[refs_size - 1].ltag = ltag;
5554 string_copy_rev(refs[refs_size - 1].id, id);
5556 return OK;
5557 }
5558 refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
5559 if (!refs)
5560 return ERR;
5562 ref = &refs[refs_size++];
5563 ref->name = malloc(namelen + 1);
5564 if (!ref->name)
5565 return ERR;
5567 strncpy(ref->name, name, namelen);
5568 ref->name[namelen] = 0;
5569 ref->head = head;
5570 ref->tag = tag;
5571 ref->ltag = ltag;
5572 ref->remote = remote;
5573 ref->tracked = tracked;
5574 string_copy_rev(ref->id, id);
5576 return OK;
5577 }
5579 static int
5580 load_refs(void)
5581 {
5582 const char *cmd_env = getenv("TIG_LS_REMOTE");
5583 const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
5585 return read_properties(popen(cmd, "r"), "\t", read_ref);
5586 }
5588 static int
5589 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
5590 {
5591 if (!strcmp(name, "i18n.commitencoding"))
5592 string_ncopy(opt_encoding, value, valuelen);
5594 if (!strcmp(name, "core.editor"))
5595 string_ncopy(opt_editor, value, valuelen);
5597 /* branch.<head>.remote */
5598 if (*opt_head &&
5599 !strncmp(name, "branch.", 7) &&
5600 !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5601 !strcmp(name + 7 + strlen(opt_head), ".remote"))
5602 string_ncopy(opt_remote, value, valuelen);
5604 if (*opt_head && *opt_remote &&
5605 !strncmp(name, "branch.", 7) &&
5606 !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5607 !strcmp(name + 7 + strlen(opt_head), ".merge")) {
5608 size_t from = strlen(opt_remote);
5610 if (!strncmp(value, "refs/heads/", STRING_SIZE("refs/heads/"))) {
5611 value += STRING_SIZE("refs/heads/");
5612 valuelen -= STRING_SIZE("refs/heads/");
5613 }
5615 if (!string_format_from(opt_remote, &from, "/%s", value))
5616 opt_remote[0] = 0;
5617 }
5619 return OK;
5620 }
5622 static int
5623 load_git_config(void)
5624 {
5625 return read_properties(popen(GIT_CONFIG " --list", "r"),
5626 "=", read_repo_config_option);
5627 }
5629 static int
5630 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
5631 {
5632 if (!opt_git_dir[0]) {
5633 string_ncopy(opt_git_dir, name, namelen);
5635 } else if (opt_is_inside_work_tree == -1) {
5636 /* This can be 3 different values depending on the
5637 * version of git being used. If git-rev-parse does not
5638 * understand --is-inside-work-tree it will simply echo
5639 * the option else either "true" or "false" is printed.
5640 * Default to true for the unknown case. */
5641 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
5643 } else if (opt_cdup[0] == ' ') {
5644 string_ncopy(opt_cdup, name, namelen);
5645 } else {
5646 if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
5647 namelen -= STRING_SIZE("refs/heads/");
5648 name += STRING_SIZE("refs/heads/");
5649 string_ncopy(opt_head, name, namelen);
5650 }
5651 }
5653 return OK;
5654 }
5656 static int
5657 load_repo_info(void)
5658 {
5659 int result;
5660 FILE *pipe = popen("(git rev-parse --git-dir --is-inside-work-tree "
5661 " --show-cdup; git symbolic-ref HEAD) 2>/dev/null", "r");
5663 /* XXX: The line outputted by "--show-cdup" can be empty so
5664 * initialize it to something invalid to make it possible to
5665 * detect whether it has been set or not. */
5666 opt_cdup[0] = ' ';
5668 result = read_properties(pipe, "=", read_repo_info);
5669 if (opt_cdup[0] == ' ')
5670 opt_cdup[0] = 0;
5672 return result;
5673 }
5675 static int
5676 read_properties(FILE *pipe, const char *separators,
5677 int (*read_property)(char *, size_t, char *, size_t))
5678 {
5679 char buffer[BUFSIZ];
5680 char *name;
5681 int state = OK;
5683 if (!pipe)
5684 return ERR;
5686 while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
5687 char *value;
5688 size_t namelen;
5689 size_t valuelen;
5691 name = chomp_string(name);
5692 namelen = strcspn(name, separators);
5694 if (name[namelen]) {
5695 name[namelen] = 0;
5696 value = chomp_string(name + namelen + 1);
5697 valuelen = strlen(value);
5699 } else {
5700 value = "";
5701 valuelen = 0;
5702 }
5704 state = read_property(name, namelen, value, valuelen);
5705 }
5707 if (state != ERR && ferror(pipe))
5708 state = ERR;
5710 pclose(pipe);
5712 return state;
5713 }
5716 /*
5717 * Main
5718 */
5720 static void __NORETURN
5721 quit(int sig)
5722 {
5723 /* XXX: Restore tty modes and let the OS cleanup the rest! */
5724 if (cursed)
5725 endwin();
5726 exit(0);
5727 }
5729 static void __NORETURN
5730 die(const char *err, ...)
5731 {
5732 va_list args;
5734 endwin();
5736 va_start(args, err);
5737 fputs("tig: ", stderr);
5738 vfprintf(stderr, err, args);
5739 fputs("\n", stderr);
5740 va_end(args);
5742 exit(1);
5743 }
5745 static void
5746 warn(const char *msg, ...)
5747 {
5748 va_list args;
5750 va_start(args, msg);
5751 fputs("tig warning: ", stderr);
5752 vfprintf(stderr, msg, args);
5753 fputs("\n", stderr);
5754 va_end(args);
5755 }
5757 int
5758 main(int argc, char *argv[])
5759 {
5760 struct view *view;
5761 enum request request;
5762 size_t i;
5764 signal(SIGINT, quit);
5766 if (setlocale(LC_ALL, "")) {
5767 char *codeset = nl_langinfo(CODESET);
5769 string_ncopy(opt_codeset, codeset, strlen(codeset));
5770 }
5772 if (load_repo_info() == ERR)
5773 die("Failed to load repo info.");
5775 if (load_options() == ERR)
5776 die("Failed to load user config.");
5778 if (load_git_config() == ERR)
5779 die("Failed to load repo config.");
5781 if (!parse_options(argc, argv))
5782 return 0;
5784 /* Require a git repository unless when running in pager mode. */
5785 if (!opt_git_dir[0] && opt_request != REQ_VIEW_PAGER)
5786 die("Not a git repository");
5788 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
5789 opt_utf8 = FALSE;
5791 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
5792 opt_iconv = iconv_open(opt_codeset, opt_encoding);
5793 if (opt_iconv == ICONV_NONE)
5794 die("Failed to initialize character set conversion");
5795 }
5797 if (*opt_git_dir && load_refs() == ERR)
5798 die("Failed to load refs.");
5800 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
5801 view->cmd_env = getenv(view->cmd_env);
5803 request = opt_request;
5805 init_display();
5807 while (view_driver(display[current_view], request)) {
5808 int key;
5809 int i;
5811 foreach_view (view, i)
5812 update_view(view);
5814 /* Refresh, accept single keystroke of input */
5815 key = wgetch(status_win);
5817 /* wgetch() with nodelay() enabled returns ERR when there's no
5818 * input. */
5819 if (key == ERR) {
5820 request = REQ_NONE;
5821 continue;
5822 }
5824 request = get_keybinding(display[current_view]->keymap, key);
5826 /* Some low-level request handling. This keeps access to
5827 * status_win restricted. */
5828 switch (request) {
5829 case REQ_PROMPT:
5830 {
5831 char *cmd = read_prompt(":");
5833 if (cmd && string_format(opt_cmd, "git %s", cmd)) {
5834 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
5835 opt_request = REQ_VIEW_DIFF;
5836 } else {
5837 opt_request = REQ_VIEW_PAGER;
5838 }
5839 break;
5840 }
5842 request = REQ_NONE;
5843 break;
5844 }
5845 case REQ_SEARCH:
5846 case REQ_SEARCH_BACK:
5847 {
5848 const char *prompt = request == REQ_SEARCH
5849 ? "/" : "?";
5850 char *search = read_prompt(prompt);
5852 if (search)
5853 string_ncopy(opt_search, search, strlen(search));
5854 else
5855 request = REQ_NONE;
5856 break;
5857 }
5858 case REQ_SCREEN_RESIZE:
5859 {
5860 int height, width;
5862 getmaxyx(stdscr, height, width);
5864 /* Resize the status view and let the view driver take
5865 * care of resizing the displayed views. */
5866 wresize(status_win, 1, width);
5867 mvwin(status_win, height - 1, 0);
5868 wrefresh(status_win);
5869 break;
5870 }
5871 default:
5872 break;
5873 }
5874 }
5876 quit(0);
5878 return 0;
5879 }