Code

Make the rev graph visualization have a one rev look-ahead
[tig.git] / tig.c
1 /* Copyright (c) 2006 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 #ifndef VERSION
15 #define VERSION "tig-0.4.git"
16 #endif
18 #ifndef DEBUG
19 #define NDEBUG
20 #endif
22 #include <assert.h>
23 #include <errno.h>
24 #include <ctype.h>
25 #include <signal.h>
26 #include <stdarg.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <unistd.h>
31 #include <time.h>
33 #include <sys/types.h>
34 #include <regex.h>
36 #include <locale.h>
37 #include <langinfo.h>
38 #include <iconv.h>
40 #include <curses.h>
42 #if __GNUC__ >= 3
43 #define __NORETURN __attribute__((__noreturn__))
44 #else
45 #define __NORETURN
46 #endif
48 static void __NORETURN die(const char *err, ...);
49 static void report(const char *msg, ...);
50 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, int, char *, int));
51 static void set_nonblocking_input(bool loading);
52 static size_t utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed);
54 #define ABS(x)          ((x) >= 0  ? (x) : -(x))
55 #define MIN(x, y)       ((x) < (y) ? (x) :  (y))
57 #define ARRAY_SIZE(x)   (sizeof(x) / sizeof(x[0]))
58 #define STRING_SIZE(x)  (sizeof(x) - 1)
60 #define SIZEOF_STR      1024    /* Default string size. */
61 #define SIZEOF_REF      256     /* Size of symbolic or SHA1 ID. */
62 #define SIZEOF_REV      41      /* Holds a SHA-1 and an ending NUL */
64 /* Revision graph */
66 #define REVGRAPH_INIT   'I'
67 #define REVGRAPH_MERGE  'M'
68 #define REVGRAPH_BRANCH '+'
69 #define REVGRAPH_COMMIT '*'
70 #define REVGRAPH_LINE   '|'
72 #define SIZEOF_REVGRAPH 19      /* Size of revision ancestry graphics. */
74 /* This color name can be used to refer to the default term colors. */
75 #define COLOR_DEFAULT   (-1)
77 #define ICONV_NONE      ((iconv_t) -1)
79 /* The format and size of the date column in the main view. */
80 #define DATE_FORMAT     "%Y-%m-%d %H:%M"
81 #define DATE_COLS       STRING_SIZE("2006-04-29 14:21 ")
83 #define AUTHOR_COLS     20
85 /* The default interval between line numbers. */
86 #define NUMBER_INTERVAL 1
88 #define TABSIZE         8
90 #define SCALE_SPLIT_VIEW(height)        ((height) * 2 / 3)
92 #define TIG_LS_REMOTE \
93         "git ls-remote . 2>/dev/null"
95 #define TIG_DIFF_CMD \
96         "git show --root --patch-with-stat --find-copies-harder -B -C %s 2>/dev/null"
98 #define TIG_LOG_CMD     \
99         "git log --cc --stat -n100 %s 2>/dev/null"
101 #define TIG_MAIN_CMD \
102         "git log --topo-order --pretty=raw %s 2>/dev/null"
104 #define TIG_TREE_CMD    \
105         "git ls-tree %s %s"
107 #define TIG_BLOB_CMD    \
108         "git cat-file blob %s"
110 /* XXX: Needs to be defined to the empty string. */
111 #define TIG_HELP_CMD    ""
112 #define TIG_PAGER_CMD   ""
114 /* Some ascii-shorthands fitted into the ncurses namespace. */
115 #define KEY_TAB         '\t'
116 #define KEY_RETURN      '\r'
117 #define KEY_ESC         27
120 struct ref {
121         char *name;             /* Ref name; tag or head names are shortened. */
122         char id[SIZEOF_REV];    /* Commit SHA1 ID */
123         unsigned int tag:1;     /* Is it a tag? */
124         unsigned int next:1;    /* For ref lists: are there more refs? */
125 };
127 static struct ref **get_refs(char *id);
129 struct int_map {
130         const char *name;
131         int namelen;
132         int value;
133 };
135 static int
136 set_from_int_map(struct int_map *map, size_t map_size,
137                  int *value, const char *name, int namelen)
140         int i;
142         for (i = 0; i < map_size; i++)
143                 if (namelen == map[i].namelen &&
144                     !strncasecmp(name, map[i].name, namelen)) {
145                         *value = map[i].value;
146                         return OK;
147                 }
149         return ERR;
153 /*
154  * String helpers
155  */
157 static inline void
158 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
160         if (srclen > dstlen - 1)
161                 srclen = dstlen - 1;
163         strncpy(dst, src, srclen);
164         dst[srclen] = 0;
167 /* Shorthands for safely copying into a fixed buffer. */
169 #define string_copy(dst, src) \
170         string_ncopy_do(dst, sizeof(dst), src, sizeof(dst))
172 #define string_ncopy(dst, src, srclen) \
173         string_ncopy_do(dst, sizeof(dst), src, srclen)
175 static char *
176 chomp_string(char *name)
178         int namelen;
180         while (isspace(*name))
181                 name++;
183         namelen = strlen(name) - 1;
184         while (namelen > 0 && isspace(name[namelen]))
185                 name[namelen--] = 0;
187         return name;
190 static bool
191 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
193         va_list args;
194         size_t pos = bufpos ? *bufpos : 0;
196         va_start(args, fmt);
197         pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
198         va_end(args);
200         if (bufpos)
201                 *bufpos = pos;
203         return pos >= bufsize ? FALSE : TRUE;
206 #define string_format(buf, fmt, args...) \
207         string_nformat(buf, sizeof(buf), NULL, fmt, args)
209 #define string_format_from(buf, from, fmt, args...) \
210         string_nformat(buf, sizeof(buf), from, fmt, args)
212 static int
213 string_enum_compare(const char *str1, const char *str2, int len)
215         size_t i;
217 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
219         /* Diff-Header == DIFF_HEADER */
220         for (i = 0; i < len; i++) {
221                 if (toupper(str1[i]) == toupper(str2[i]))
222                         continue;
224                 if (string_enum_sep(str1[i]) &&
225                     string_enum_sep(str2[i]))
226                         continue;
228                 return str1[i] - str2[i];
229         }
231         return 0;
234 /* Shell quoting
235  *
236  * NOTE: The following is a slightly modified copy of the git project's shell
237  * quoting routines found in the quote.c file.
238  *
239  * Help to copy the thing properly quoted for the shell safety.  any single
240  * quote is replaced with '\'', any exclamation point is replaced with '\!',
241  * and the whole thing is enclosed in a
242  *
243  * E.g.
244  *  original     sq_quote     result
245  *  name     ==> name      ==> 'name'
246  *  a b      ==> a b       ==> 'a b'
247  *  a'b      ==> a'\''b    ==> 'a'\''b'
248  *  a!b      ==> a'\!'b    ==> 'a'\!'b'
249  */
251 static size_t
252 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
254         char c;
256 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
258         BUFPUT('\'');
259         while ((c = *src++)) {
260                 if (c == '\'' || c == '!') {
261                         BUFPUT('\'');
262                         BUFPUT('\\');
263                         BUFPUT(c);
264                         BUFPUT('\'');
265                 } else {
266                         BUFPUT(c);
267                 }
268         }
269         BUFPUT('\'');
271         return bufsize;
275 /*
276  * User requests
277  */
279 #define REQ_INFO \
280         /* XXX: Keep the view request first and in sync with views[]. */ \
281         REQ_GROUP("View switching") \
282         REQ_(VIEW_MAIN,         "Show main view"), \
283         REQ_(VIEW_DIFF,         "Show diff view"), \
284         REQ_(VIEW_LOG,          "Show log view"), \
285         REQ_(VIEW_TREE,         "Show tree view"), \
286         REQ_(VIEW_BLOB,         "Show blob view"), \
287         REQ_(VIEW_HELP,         "Show help page"), \
288         REQ_(VIEW_PAGER,        "Show pager view"), \
289         \
290         REQ_GROUP("View manipulation") \
291         REQ_(ENTER,             "Enter current line and scroll"), \
292         REQ_(NEXT,              "Move to next"), \
293         REQ_(PREVIOUS,          "Move to previous"), \
294         REQ_(VIEW_NEXT,         "Move focus to next view"), \
295         REQ_(VIEW_CLOSE,        "Close the current view"), \
296         REQ_(QUIT,              "Close all views and quit"), \
297         \
298         REQ_GROUP("Cursor navigation") \
299         REQ_(MOVE_UP,           "Move cursor one line up"), \
300         REQ_(MOVE_DOWN,         "Move cursor one line down"), \
301         REQ_(MOVE_PAGE_DOWN,    "Move cursor one page down"), \
302         REQ_(MOVE_PAGE_UP,      "Move cursor one page up"), \
303         REQ_(MOVE_FIRST_LINE,   "Move cursor to first line"), \
304         REQ_(MOVE_LAST_LINE,    "Move cursor to last line"), \
305         \
306         REQ_GROUP("Scrolling") \
307         REQ_(SCROLL_LINE_UP,    "Scroll one line up"), \
308         REQ_(SCROLL_LINE_DOWN,  "Scroll one line down"), \
309         REQ_(SCROLL_PAGE_UP,    "Scroll one page up"), \
310         REQ_(SCROLL_PAGE_DOWN,  "Scroll one page down"), \
311         \
312         REQ_GROUP("Searching") \
313         REQ_(SEARCH,            "Search the view"), \
314         REQ_(SEARCH_BACK,       "Search backwards in the view"), \
315         REQ_(FIND_NEXT,         "Find next search match"), \
316         REQ_(FIND_PREV,         "Find previous search match"), \
317         \
318         REQ_GROUP("Misc") \
319         REQ_(NONE,              "Do nothing"), \
320         REQ_(PROMPT,            "Bring up the prompt"), \
321         REQ_(SCREEN_REDRAW,     "Redraw the screen"), \
322         REQ_(SCREEN_RESIZE,     "Resize the screen"), \
323         REQ_(SHOW_VERSION,      "Show version information"), \
324         REQ_(STOP_LOADING,      "Stop all loading views"), \
325         REQ_(TOGGLE_LINENO,     "Toggle line numbers"), \
326         REQ_(TOGGLE_REV_GRAPH,  "Toggle revision graph visualization")
329 /* User action requests. */
330 enum request {
331 #define REQ_GROUP(help)
332 #define REQ_(req, help) REQ_##req
334         /* Offset all requests to avoid conflicts with ncurses getch values. */
335         REQ_OFFSET = KEY_MAX + 1,
336         REQ_INFO,
337         REQ_UNKNOWN,
339 #undef  REQ_GROUP
340 #undef  REQ_
341 };
343 struct request_info {
344         enum request request;
345         char *name;
346         int namelen;
347         char *help;
348 };
350 static struct request_info req_info[] = {
351 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
352 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
353         REQ_INFO
354 #undef  REQ_GROUP
355 #undef  REQ_
356 };
358 static enum request
359 get_request(const char *name)
361         int namelen = strlen(name);
362         int i;
364         for (i = 0; i < ARRAY_SIZE(req_info); i++)
365                 if (req_info[i].namelen == namelen &&
366                     !string_enum_compare(req_info[i].name, name, namelen))
367                         return req_info[i].request;
369         return REQ_UNKNOWN;
373 /*
374  * Options
375  */
377 static const char usage[] =
378 VERSION " (" __DATE__ ")\n"
379 "\n"
380 "Usage: tig [options]\n"
381 "   or: tig [options] [--] [git log options]\n"
382 "   or: tig [options] log  [git log options]\n"
383 "   or: tig [options] diff [git diff options]\n"
384 "   or: tig [options] show [git show options]\n"
385 "   or: tig [options] <    [git command output]\n"
386 "\n"
387 "Options:\n"
388 "  -l                          Start up in log view\n"
389 "  -d                          Start up in diff view\n"
390 "  -n[I], --line-number[=I]    Show line numbers with given interval\n"
391 "  -b[N], --tab-size[=N]       Set number of spaces for tab expansion\n"
392 "  --                          Mark end of tig options\n"
393 "  -v, --version               Show version and exit\n"
394 "  -h, --help                  Show help message and exit\n";
396 /* Option and state variables. */
397 static bool opt_line_number             = FALSE;
398 static bool opt_rev_graph               = TRUE;
399 static int opt_num_interval             = NUMBER_INTERVAL;
400 static int opt_tab_size                 = TABSIZE;
401 static enum request opt_request         = REQ_VIEW_MAIN;
402 static char opt_cmd[SIZEOF_STR]         = "";
403 static char opt_path[SIZEOF_STR]        = "";
404 static FILE *opt_pipe                   = NULL;
405 static char opt_encoding[20]            = "UTF-8";
406 static bool opt_utf8                    = TRUE;
407 static char opt_codeset[20]             = "UTF-8";
408 static iconv_t opt_iconv                = ICONV_NONE;
409 static char opt_search[SIZEOF_STR]      = "";
411 enum option_type {
412         OPT_NONE,
413         OPT_INT,
414 };
416 static bool
417 check_option(char *opt, char short_name, char *name, enum option_type type, ...)
419         va_list args;
420         char *value = "";
421         int *number;
423         if (opt[0] != '-')
424                 return FALSE;
426         if (opt[1] == '-') {
427                 int namelen = strlen(name);
429                 opt += 2;
431                 if (strncmp(opt, name, namelen))
432                         return FALSE;
434                 if (opt[namelen] == '=')
435                         value = opt + namelen + 1;
437         } else {
438                 if (!short_name || opt[1] != short_name)
439                         return FALSE;
440                 value = opt + 2;
441         }
443         va_start(args, type);
444         if (type == OPT_INT) {
445                 number = va_arg(args, int *);
446                 if (isdigit(*value))
447                         *number = atoi(value);
448         }
449         va_end(args);
451         return TRUE;
454 /* Returns the index of log or diff command or -1 to exit. */
455 static bool
456 parse_options(int argc, char *argv[])
458         int i;
460         for (i = 1; i < argc; i++) {
461                 char *opt = argv[i];
463                 if (!strcmp(opt, "-l")) {
464                         opt_request = REQ_VIEW_LOG;
465                         continue;
466                 }
468                 if (!strcmp(opt, "-d")) {
469                         opt_request = REQ_VIEW_DIFF;
470                         continue;
471                 }
473                 if (check_option(opt, 'n', "line-number", OPT_INT, &opt_num_interval)) {
474                         opt_line_number = TRUE;
475                         continue;
476                 }
478                 if (check_option(opt, 'b', "tab-size", OPT_INT, &opt_tab_size)) {
479                         opt_tab_size = MIN(opt_tab_size, TABSIZE);
480                         continue;
481                 }
483                 if (check_option(opt, 'v', "version", OPT_NONE)) {
484                         printf("tig version %s\n", VERSION);
485                         return FALSE;
486                 }
488                 if (check_option(opt, 'h', "help", OPT_NONE)) {
489                         printf(usage);
490                         return FALSE;
491                 }
493                 if (!strcmp(opt, "--")) {
494                         i++;
495                         break;
496                 }
498                 if (!strcmp(opt, "log") ||
499                     !strcmp(opt, "diff") ||
500                     !strcmp(opt, "show")) {
501                         opt_request = opt[0] == 'l'
502                                     ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
503                         break;
504                 }
506                 if (opt[0] && opt[0] != '-')
507                         break;
509                 die("unknown option '%s'\n\n%s", opt, usage);
510         }
512         if (!isatty(STDIN_FILENO)) {
513                 opt_request = REQ_VIEW_PAGER;
514                 opt_pipe = stdin;
516         } else if (i < argc) {
517                 size_t buf_size;
519                 if (opt_request == REQ_VIEW_MAIN)
520                         /* XXX: This is vulnerable to the user overriding
521                          * options required for the main view parser. */
522                         string_copy(opt_cmd, "git log --stat --pretty=raw");
523                 else
524                         string_copy(opt_cmd, "git");
525                 buf_size = strlen(opt_cmd);
527                 while (buf_size < sizeof(opt_cmd) && i < argc) {
528                         opt_cmd[buf_size++] = ' ';
529                         buf_size = sq_quote(opt_cmd, buf_size, argv[i++]);
530                 }
532                 if (buf_size >= sizeof(opt_cmd))
533                         die("command too long");
535                 opt_cmd[buf_size] = 0;
537         }
539         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
540                 opt_utf8 = FALSE;
542         return TRUE;
546 /*
547  * Line-oriented content detection.
548  */
550 #define LINE_INFO \
551 LINE(DIFF_HEADER,  "diff --git ",       COLOR_YELLOW,   COLOR_DEFAULT,  0), \
552 LINE(DIFF_CHUNK,   "@@",                COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
553 LINE(DIFF_ADD,     "+",                 COLOR_GREEN,    COLOR_DEFAULT,  0), \
554 LINE(DIFF_DEL,     "-",                 COLOR_RED,      COLOR_DEFAULT,  0), \
555 LINE(DIFF_INDEX,        "index ",         COLOR_BLUE,   COLOR_DEFAULT,  0), \
556 LINE(DIFF_OLDMODE,      "old file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
557 LINE(DIFF_NEWMODE,      "new file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
558 LINE(DIFF_COPY_FROM,    "copy from",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
559 LINE(DIFF_COPY_TO,      "copy to",        COLOR_YELLOW, COLOR_DEFAULT,  0), \
560 LINE(DIFF_RENAME_FROM,  "rename from",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
561 LINE(DIFF_RENAME_TO,    "rename to",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
562 LINE(DIFF_SIMILARITY,   "similarity ",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
563 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
564 LINE(DIFF_TREE,         "diff-tree ",     COLOR_BLUE,   COLOR_DEFAULT,  0), \
565 LINE(PP_AUTHOR,    "Author: ",          COLOR_CYAN,     COLOR_DEFAULT,  0), \
566 LINE(PP_COMMIT,    "Commit: ",          COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
567 LINE(PP_MERGE,     "Merge: ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
568 LINE(PP_DATE,      "Date:   ",          COLOR_YELLOW,   COLOR_DEFAULT,  0), \
569 LINE(PP_ADATE,     "AuthorDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
570 LINE(PP_CDATE,     "CommitDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
571 LINE(PP_REFS,      "Refs: ",            COLOR_RED,      COLOR_DEFAULT,  0), \
572 LINE(COMMIT,       "commit ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
573 LINE(PARENT,       "parent ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
574 LINE(TREE,         "tree ",             COLOR_BLUE,     COLOR_DEFAULT,  0), \
575 LINE(AUTHOR,       "author ",           COLOR_CYAN,     COLOR_DEFAULT,  0), \
576 LINE(COMMITTER,    "committer ",        COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
577 LINE(SIGNOFF,      "    Signed-off-by", COLOR_YELLOW,   COLOR_DEFAULT,  0), \
578 LINE(ACKED,        "    Acked-by",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
579 LINE(DEFAULT,      "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
580 LINE(CURSOR,       "",                  COLOR_WHITE,    COLOR_GREEN,    A_BOLD), \
581 LINE(STATUS,       "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
582 LINE(TITLE_BLUR,   "",                  COLOR_WHITE,    COLOR_BLUE,     0), \
583 LINE(TITLE_FOCUS,  "",                  COLOR_WHITE,    COLOR_BLUE,     A_BOLD), \
584 LINE(MAIN_DATE,    "",                  COLOR_BLUE,     COLOR_DEFAULT,  0), \
585 LINE(MAIN_AUTHOR,  "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
586 LINE(MAIN_COMMIT,  "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
587 LINE(MAIN_DELIM,   "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
588 LINE(MAIN_TAG,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  A_BOLD), \
589 LINE(MAIN_REF,     "",                  COLOR_CYAN,     COLOR_DEFAULT,  A_BOLD), \
590 LINE(TREE_DIR,     "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
591 LINE(TREE_FILE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL)
593 enum line_type {
594 #define LINE(type, line, fg, bg, attr) \
595         LINE_##type
596         LINE_INFO
597 #undef  LINE
598 };
600 struct line_info {
601         const char *name;       /* Option name. */
602         int namelen;            /* Size of option name. */
603         const char *line;       /* The start of line to match. */
604         int linelen;            /* Size of string to match. */
605         int fg, bg, attr;       /* Color and text attributes for the lines. */
606 };
608 static struct line_info line_info[] = {
609 #define LINE(type, line, fg, bg, attr) \
610         { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
611         LINE_INFO
612 #undef  LINE
613 };
615 static enum line_type
616 get_line_type(char *line)
618         int linelen = strlen(line);
619         enum line_type type;
621         for (type = 0; type < ARRAY_SIZE(line_info); type++)
622                 /* Case insensitive search matches Signed-off-by lines better. */
623                 if (linelen >= line_info[type].linelen &&
624                     !strncasecmp(line_info[type].line, line, line_info[type].linelen))
625                         return type;
627         return LINE_DEFAULT;
630 static inline int
631 get_line_attr(enum line_type type)
633         assert(type < ARRAY_SIZE(line_info));
634         return COLOR_PAIR(type) | line_info[type].attr;
637 static struct line_info *
638 get_line_info(char *name, int namelen)
640         enum line_type type;
642         for (type = 0; type < ARRAY_SIZE(line_info); type++)
643                 if (namelen == line_info[type].namelen &&
644                     !string_enum_compare(line_info[type].name, name, namelen))
645                         return &line_info[type];
647         return NULL;
650 static void
651 init_colors(void)
653         int default_bg = COLOR_BLACK;
654         int default_fg = COLOR_WHITE;
655         enum line_type type;
657         start_color();
659         if (use_default_colors() != ERR) {
660                 default_bg = -1;
661                 default_fg = -1;
662         }
664         for (type = 0; type < ARRAY_SIZE(line_info); type++) {
665                 struct line_info *info = &line_info[type];
666                 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
667                 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
669                 init_pair(type, fg, bg);
670         }
673 struct line {
674         enum line_type type;
676         /* State flags */
677         unsigned int selected:1;
679         void *data;             /* User data */
680 };
683 /*
684  * Keys
685  */
687 struct keybinding {
688         int alias;
689         enum request request;
690         struct keybinding *next;
691 };
693 static struct keybinding default_keybindings[] = {
694         /* View switching */
695         { 'm',          REQ_VIEW_MAIN },
696         { 'd',          REQ_VIEW_DIFF },
697         { 'l',          REQ_VIEW_LOG },
698         { 't',          REQ_VIEW_TREE },
699         { 'f',          REQ_VIEW_BLOB },
700         { 'p',          REQ_VIEW_PAGER },
701         { 'h',          REQ_VIEW_HELP },
703         /* View manipulation */
704         { 'q',          REQ_VIEW_CLOSE },
705         { KEY_TAB,      REQ_VIEW_NEXT },
706         { KEY_RETURN,   REQ_ENTER },
707         { KEY_UP,       REQ_PREVIOUS },
708         { KEY_DOWN,     REQ_NEXT },
710         /* Cursor navigation */
711         { 'k',          REQ_MOVE_UP },
712         { 'j',          REQ_MOVE_DOWN },
713         { KEY_HOME,     REQ_MOVE_FIRST_LINE },
714         { KEY_END,      REQ_MOVE_LAST_LINE },
715         { KEY_NPAGE,    REQ_MOVE_PAGE_DOWN },
716         { ' ',          REQ_MOVE_PAGE_DOWN },
717         { KEY_PPAGE,    REQ_MOVE_PAGE_UP },
718         { 'b',          REQ_MOVE_PAGE_UP },
719         { '-',          REQ_MOVE_PAGE_UP },
721         /* Scrolling */
722         { KEY_IC,       REQ_SCROLL_LINE_UP },
723         { KEY_DC,       REQ_SCROLL_LINE_DOWN },
724         { 'w',          REQ_SCROLL_PAGE_UP },
725         { 's',          REQ_SCROLL_PAGE_DOWN },
727         /* Searching */
728         { '/',          REQ_SEARCH },
729         { '?',          REQ_SEARCH_BACK },
730         { 'n',          REQ_FIND_NEXT },
731         { 'N',          REQ_FIND_PREV },
733         /* Misc */
734         { 'Q',          REQ_QUIT },
735         { 'z',          REQ_STOP_LOADING },
736         { 'v',          REQ_SHOW_VERSION },
737         { 'r',          REQ_SCREEN_REDRAW },
738         { '.',          REQ_TOGGLE_LINENO },
739         { 'g',          REQ_TOGGLE_REV_GRAPH },
740         { ':',          REQ_PROMPT },
742         /* wgetch() with nodelay() enabled returns ERR when there's no input. */
743         { ERR,          REQ_NONE },
745         /* Using the ncurses SIGWINCH handler. */
746         { KEY_RESIZE,   REQ_SCREEN_RESIZE },
747 };
749 #define KEYMAP_INFO \
750         KEYMAP_(GENERIC), \
751         KEYMAP_(MAIN), \
752         KEYMAP_(DIFF), \
753         KEYMAP_(LOG), \
754         KEYMAP_(TREE), \
755         KEYMAP_(BLOB), \
756         KEYMAP_(PAGER), \
757         KEYMAP_(HELP) \
759 enum keymap {
760 #define KEYMAP_(name) KEYMAP_##name
761         KEYMAP_INFO
762 #undef  KEYMAP_
763 };
765 static struct int_map keymap_table[] = {
766 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
767         KEYMAP_INFO
768 #undef  KEYMAP_
769 };
771 #define set_keymap(map, name) \
772         set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
774 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
776 static void
777 add_keybinding(enum keymap keymap, enum request request, int key)
779         struct keybinding *keybinding;
781         keybinding = calloc(1, sizeof(*keybinding));
782         if (!keybinding)
783                 die("Failed to allocate keybinding");
785         keybinding->alias = key;
786         keybinding->request = request;
787         keybinding->next = keybindings[keymap];
788         keybindings[keymap] = keybinding;
791 /* Looks for a key binding first in the given map, then in the generic map, and
792  * lastly in the default keybindings. */
793 static enum request
794 get_keybinding(enum keymap keymap, int key)
796         struct keybinding *kbd;
797         int i;
799         for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
800                 if (kbd->alias == key)
801                         return kbd->request;
803         for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
804                 if (kbd->alias == key)
805                         return kbd->request;
807         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
808                 if (default_keybindings[i].alias == key)
809                         return default_keybindings[i].request;
811         return (enum request) key;
815 struct key {
816         char *name;
817         int value;
818 };
820 static struct key key_table[] = {
821         { "Enter",      KEY_RETURN },
822         { "Space",      ' ' },
823         { "Backspace",  KEY_BACKSPACE },
824         { "Tab",        KEY_TAB },
825         { "Escape",     KEY_ESC },
826         { "Left",       KEY_LEFT },
827         { "Right",      KEY_RIGHT },
828         { "Up",         KEY_UP },
829         { "Down",       KEY_DOWN },
830         { "Insert",     KEY_IC },
831         { "Delete",     KEY_DC },
832         { "Hash",       '#' },
833         { "Home",       KEY_HOME },
834         { "End",        KEY_END },
835         { "PageUp",     KEY_PPAGE },
836         { "PageDown",   KEY_NPAGE },
837         { "F1",         KEY_F(1) },
838         { "F2",         KEY_F(2) },
839         { "F3",         KEY_F(3) },
840         { "F4",         KEY_F(4) },
841         { "F5",         KEY_F(5) },
842         { "F6",         KEY_F(6) },
843         { "F7",         KEY_F(7) },
844         { "F8",         KEY_F(8) },
845         { "F9",         KEY_F(9) },
846         { "F10",        KEY_F(10) },
847         { "F11",        KEY_F(11) },
848         { "F12",        KEY_F(12) },
849 };
851 static int
852 get_key_value(const char *name)
854         int i;
856         for (i = 0; i < ARRAY_SIZE(key_table); i++)
857                 if (!strcasecmp(key_table[i].name, name))
858                         return key_table[i].value;
860         if (strlen(name) == 1 && isprint(*name))
861                 return (int) *name;
863         return ERR;
866 static char *
867 get_key(enum request request)
869         static char buf[BUFSIZ];
870         static char key_char[] = "'X'";
871         size_t pos = 0;
872         char *sep = "    ";
873         int i;
875         buf[pos] = 0;
877         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
878                 struct keybinding *keybinding = &default_keybindings[i];
879                 char *seq = NULL;
880                 int key;
882                 if (keybinding->request != request)
883                         continue;
885                 for (key = 0; key < ARRAY_SIZE(key_table); key++)
886                         if (key_table[key].value == keybinding->alias)
887                                 seq = key_table[key].name;
889                 if (seq == NULL &&
890                     keybinding->alias < 127 &&
891                     isprint(keybinding->alias)) {
892                         key_char[1] = (char) keybinding->alias;
893                         seq = key_char;
894                 }
896                 if (!seq)
897                         seq = "'?'";
899                 if (!string_format_from(buf, &pos, "%s%s", sep, seq))
900                         return "Too many keybindings!";
901                 sep = ", ";
902         }
904         return buf;
908 /*
909  * User config file handling.
910  */
912 static struct int_map color_map[] = {
913 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
914         COLOR_MAP(DEFAULT),
915         COLOR_MAP(BLACK),
916         COLOR_MAP(BLUE),
917         COLOR_MAP(CYAN),
918         COLOR_MAP(GREEN),
919         COLOR_MAP(MAGENTA),
920         COLOR_MAP(RED),
921         COLOR_MAP(WHITE),
922         COLOR_MAP(YELLOW),
923 };
925 #define set_color(color, name) \
926         set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
928 static struct int_map attr_map[] = {
929 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
930         ATTR_MAP(NORMAL),
931         ATTR_MAP(BLINK),
932         ATTR_MAP(BOLD),
933         ATTR_MAP(DIM),
934         ATTR_MAP(REVERSE),
935         ATTR_MAP(STANDOUT),
936         ATTR_MAP(UNDERLINE),
937 };
939 #define set_attribute(attr, name) \
940         set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
942 static int   config_lineno;
943 static bool  config_errors;
944 static char *config_msg;
946 /* Wants: object fgcolor bgcolor [attr] */
947 static int
948 option_color_command(int argc, char *argv[])
950         struct line_info *info;
952         if (argc != 3 && argc != 4) {
953                 config_msg = "Wrong number of arguments given to color command";
954                 return ERR;
955         }
957         info = get_line_info(argv[0], strlen(argv[0]));
958         if (!info) {
959                 config_msg = "Unknown color name";
960                 return ERR;
961         }
963         if (set_color(&info->fg, argv[1]) == ERR ||
964             set_color(&info->bg, argv[2]) == ERR) {
965                 config_msg = "Unknown color";
966                 return ERR;
967         }
969         if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
970                 config_msg = "Unknown attribute";
971                 return ERR;
972         }
974         return OK;
977 /* Wants: name = value */
978 static int
979 option_set_command(int argc, char *argv[])
981         if (argc != 3) {
982                 config_msg = "Wrong number of arguments given to set command";
983                 return ERR;
984         }
986         if (strcmp(argv[1], "=")) {
987                 config_msg = "No value assigned";
988                 return ERR;
989         }
991         if (!strcmp(argv[0], "show-rev-graph")) {
992                 opt_rev_graph = (!strcmp(argv[2], "1") ||
993                                  !strcmp(argv[2], "true") ||
994                                  !strcmp(argv[2], "yes"));
995                 return OK;
996         }
998         if (!strcmp(argv[0], "line-number-interval")) {
999                 opt_num_interval = atoi(argv[2]);
1000                 return OK;
1001         }
1003         if (!strcmp(argv[0], "tab-size")) {
1004                 opt_tab_size = atoi(argv[2]);
1005                 return OK;
1006         }
1008         if (!strcmp(argv[0], "commit-encoding")) {
1009                 char *arg = argv[2];
1010                 int delimiter = *arg;
1011                 int i;
1013                 switch (delimiter) {
1014                 case '"':
1015                 case '\'':
1016                         for (arg++, i = 0; arg[i]; i++)
1017                                 if (arg[i] == delimiter) {
1018                                         arg[i] = 0;
1019                                         break;
1020                                 }
1021                 default:
1022                         string_copy(opt_encoding, arg);
1023                         return OK;
1024                 }
1025         }
1027         config_msg = "Unknown variable name";
1028         return ERR;
1031 /* Wants: mode request key */
1032 static int
1033 option_bind_command(int argc, char *argv[])
1035         enum request request;
1036         int keymap;
1037         int key;
1039         if (argc != 3) {
1040                 config_msg = "Wrong number of arguments given to bind command";
1041                 return ERR;
1042         }
1044         if (set_keymap(&keymap, argv[0]) == ERR) {
1045                 config_msg = "Unknown key map";
1046                 return ERR;
1047         }
1049         key = get_key_value(argv[1]);
1050         if (key == ERR) {
1051                 config_msg = "Unknown key";
1052                 return ERR;
1053         }
1055         request = get_request(argv[2]);
1056         if (request == REQ_UNKNOWN) {
1057                 config_msg = "Unknown request name";
1058                 return ERR;
1059         }
1061         add_keybinding(keymap, request, key);
1063         return OK;
1066 static int
1067 set_option(char *opt, char *value)
1069         char *argv[16];
1070         int valuelen;
1071         int argc = 0;
1073         /* Tokenize */
1074         while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1075                 argv[argc++] = value;
1077                 value += valuelen;
1078                 if (!*value)
1079                         break;
1081                 *value++ = 0;
1082                 while (isspace(*value))
1083                         value++;
1084         }
1086         if (!strcmp(opt, "color"))
1087                 return option_color_command(argc, argv);
1089         if (!strcmp(opt, "set"))
1090                 return option_set_command(argc, argv);
1092         if (!strcmp(opt, "bind"))
1093                 return option_bind_command(argc, argv);
1095         config_msg = "Unknown option command";
1096         return ERR;
1099 static int
1100 read_option(char *opt, int optlen, char *value, int valuelen)
1102         int status = OK;
1104         config_lineno++;
1105         config_msg = "Internal error";
1107         /* Check for comment markers, since read_properties() will
1108          * only ensure opt and value are split at first " \t". */
1109         optlen = strcspn(opt, "#");
1110         if (optlen == 0)
1111                 return OK;
1113         if (opt[optlen] != 0) {
1114                 config_msg = "No option value";
1115                 status = ERR;
1117         }  else {
1118                 /* Look for comment endings in the value. */
1119                 int len = strcspn(value, "#");
1121                 if (len < valuelen) {
1122                         valuelen = len;
1123                         value[valuelen] = 0;
1124                 }
1126                 status = set_option(opt, value);
1127         }
1129         if (status == ERR) {
1130                 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1131                         config_lineno, optlen, opt, config_msg);
1132                 config_errors = TRUE;
1133         }
1135         /* Always keep going if errors are encountered. */
1136         return OK;
1139 static int
1140 load_options(void)
1142         char *home = getenv("HOME");
1143         char buf[SIZEOF_STR];
1144         FILE *file;
1146         config_lineno = 0;
1147         config_errors = FALSE;
1149         if (!home || !string_format(buf, "%s/.tigrc", home))
1150                 return ERR;
1152         /* It's ok that the file doesn't exist. */
1153         file = fopen(buf, "r");
1154         if (!file)
1155                 return OK;
1157         if (read_properties(file, " \t", read_option) == ERR ||
1158             config_errors == TRUE)
1159                 fprintf(stderr, "Errors while loading %s.\n", buf);
1161         return OK;
1165 /*
1166  * The viewer
1167  */
1169 struct view;
1170 struct view_ops;
1172 /* The display array of active views and the index of the current view. */
1173 static struct view *display[2];
1174 static unsigned int current_view;
1176 #define foreach_displayed_view(view, i) \
1177         for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1179 #define displayed_views()       (display[1] != NULL ? 2 : 1)
1181 /* Current head and commit ID */
1182 static char ref_blob[SIZEOF_REF]        = "";
1183 static char ref_commit[SIZEOF_REF]      = "HEAD";
1184 static char ref_head[SIZEOF_REF]        = "HEAD";
1186 struct view {
1187         const char *name;       /* View name */
1188         const char *cmd_fmt;    /* Default command line format */
1189         const char *cmd_env;    /* Command line set via environment */
1190         const char *id;         /* Points to either of ref_{head,commit,blob} */
1192         struct view_ops *ops;   /* View operations */
1194         enum keymap keymap;     /* What keymap does this view have */
1196         char cmd[SIZEOF_STR];   /* Command buffer */
1197         char ref[SIZEOF_REF];   /* Hovered commit reference */
1198         char vid[SIZEOF_REF];   /* View ID. Set to id member when updating. */
1200         int height, width;      /* The width and height of the main window */
1201         WINDOW *win;            /* The main window */
1202         WINDOW *title;          /* The title window living below the main window */
1204         /* Navigation */
1205         unsigned long offset;   /* Offset of the window top */
1206         unsigned long lineno;   /* Current line number */
1208         /* Searching */
1209         char grep[SIZEOF_STR];  /* Search string */
1210         regex_t *regex;         /* Pre-compiled regex */
1212         /* If non-NULL, points to the view that opened this view. If this view
1213          * is closed tig will switch back to the parent view. */
1214         struct view *parent;
1216         /* Buffering */
1217         unsigned long lines;    /* Total number of lines */
1218         struct line *line;      /* Line index */
1219         unsigned long line_size;/* Total number of allocated lines */
1220         unsigned int digits;    /* Number of digits in the lines member. */
1222         /* Loading */
1223         FILE *pipe;
1224         time_t start_time;
1225 };
1227 struct view_ops {
1228         /* What type of content being displayed. Used in the title bar. */
1229         const char *type;
1230         /* Draw one line; @lineno must be < view->height. */
1231         bool (*draw)(struct view *view, struct line *line, unsigned int lineno, bool selected);
1232         /* Read one line; updates view->line. */
1233         bool (*read)(struct view *view, char *data);
1234         /* Depending on view, change display based on current line. */
1235         bool (*enter)(struct view *view, struct line *line);
1236         /* Search for regex in a line. */
1237         bool (*grep)(struct view *view, struct line *line);
1238         /* Select line */
1239         void (*select)(struct view *view, struct line *line);
1240 };
1242 static struct view_ops pager_ops;
1243 static struct view_ops main_ops;
1244 static struct view_ops tree_ops;
1245 static struct view_ops blob_ops;
1247 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1248         { name, cmd, #env, ref, ops, map}
1250 #define VIEW_(id, name, ops, ref) \
1251         VIEW_STR(name, TIG_##id##_CMD,  TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1254 static struct view views[] = {
1255         VIEW_(MAIN,  "main",  &main_ops,  ref_head),
1256         VIEW_(DIFF,  "diff",  &pager_ops, ref_commit),
1257         VIEW_(LOG,   "log",   &pager_ops, ref_head),
1258         VIEW_(TREE,  "tree",  &tree_ops,  ref_commit),
1259         VIEW_(BLOB,  "blob",  &blob_ops,  ref_blob),
1260         VIEW_(HELP,  "help",  &pager_ops, "static"),
1261         VIEW_(PAGER, "pager", &pager_ops, "static"),
1262 };
1264 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1266 #define foreach_view(view, i) \
1267         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1269 #define view_is_displayed(view) \
1270         (view == display[0] || view == display[1])
1272 static bool
1273 draw_view_line(struct view *view, unsigned int lineno)
1275         struct line *line;
1276         bool selected = (view->offset + lineno == view->lineno);
1278         assert(view_is_displayed(view));
1280         if (view->offset + lineno >= view->lines)
1281                 return FALSE;
1283         line = &view->line[view->offset + lineno];
1285         if (selected) {
1286                 line->selected = TRUE;
1287                 view->ops->select(view, line);
1288         } else if (line->selected) {
1289                 line->selected = FALSE;
1290                 wmove(view->win, lineno, 0);
1291                 wclrtoeol(view->win);
1292         }
1294         return view->ops->draw(view, line, lineno, selected);
1297 static void
1298 redraw_view_from(struct view *view, int lineno)
1300         assert(0 <= lineno && lineno < view->height);
1302         for (; lineno < view->height; lineno++) {
1303                 if (!draw_view_line(view, lineno))
1304                         break;
1305         }
1307         redrawwin(view->win);
1308         wrefresh(view->win);
1311 static void
1312 redraw_view(struct view *view)
1314         wclear(view->win);
1315         redraw_view_from(view, 0);
1319 static void
1320 update_view_title(struct view *view)
1322         assert(view_is_displayed(view));
1324         if (view == display[current_view])
1325                 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1326         else
1327                 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1329         werase(view->title);
1330         wmove(view->title, 0, 0);
1332         if (*view->ref)
1333                 wprintw(view->title, "[%s] %s", view->name, view->ref);
1334         else
1335                 wprintw(view->title, "[%s]", view->name);
1337         if (view->lines || view->pipe) {
1338                 unsigned int view_lines = view->offset + view->height;
1339                 unsigned int lines = view->lines
1340                                    ? MIN(view_lines, view->lines) * 100 / view->lines
1341                                    : 0;
1343                 wprintw(view->title, " - %s %d of %d (%d%%)",
1344                         view->ops->type,
1345                         view->lineno + 1,
1346                         view->lines,
1347                         lines);
1348         }
1350         if (view->pipe) {
1351                 time_t secs = time(NULL) - view->start_time;
1353                 /* Three git seconds are a long time ... */
1354                 if (secs > 2)
1355                         wprintw(view->title, " %lds", secs);
1356         }
1358         wmove(view->title, 0, view->width - 1);
1359         wrefresh(view->title);
1362 static void
1363 resize_display(void)
1365         int offset, i;
1366         struct view *base = display[0];
1367         struct view *view = display[1] ? display[1] : display[0];
1369         /* Setup window dimensions */
1371         getmaxyx(stdscr, base->height, base->width);
1373         /* Make room for the status window. */
1374         base->height -= 1;
1376         if (view != base) {
1377                 /* Horizontal split. */
1378                 view->width   = base->width;
1379                 view->height  = SCALE_SPLIT_VIEW(base->height);
1380                 base->height -= view->height;
1382                 /* Make room for the title bar. */
1383                 view->height -= 1;
1384         }
1386         /* Make room for the title bar. */
1387         base->height -= 1;
1389         offset = 0;
1391         foreach_displayed_view (view, i) {
1392                 if (!view->win) {
1393                         view->win = newwin(view->height, 0, offset, 0);
1394                         if (!view->win)
1395                                 die("Failed to create %s view", view->name);
1397                         scrollok(view->win, TRUE);
1399                         view->title = newwin(1, 0, offset + view->height, 0);
1400                         if (!view->title)
1401                                 die("Failed to create title window");
1403                 } else {
1404                         wresize(view->win, view->height, view->width);
1405                         mvwin(view->win,   offset, 0);
1406                         mvwin(view->title, offset + view->height, 0);
1407                 }
1409                 offset += view->height + 1;
1410         }
1413 static void
1414 redraw_display(void)
1416         struct view *view;
1417         int i;
1419         foreach_displayed_view (view, i) {
1420                 redraw_view(view);
1421                 update_view_title(view);
1422         }
1425 static void
1426 update_display_cursor(void)
1428         struct view *view = display[current_view];
1430         /* Move the cursor to the right-most column of the cursor line.
1431          *
1432          * XXX: This could turn out to be a bit expensive, but it ensures that
1433          * the cursor does not jump around. */
1434         if (view->lines) {
1435                 wmove(view->win, view->lineno - view->offset, view->width - 1);
1436                 wrefresh(view->win);
1437         }
1440 /*
1441  * Navigation
1442  */
1444 /* Scrolling backend */
1445 static void
1446 do_scroll_view(struct view *view, int lines)
1448         bool redraw_current_line = FALSE;
1450         /* The rendering expects the new offset. */
1451         view->offset += lines;
1453         assert(0 <= view->offset && view->offset < view->lines);
1454         assert(lines);
1456         /* Move current line into the view. */
1457         if (view->lineno < view->offset) {
1458                 view->lineno = view->offset;
1459                 redraw_current_line = TRUE;
1460         } else if (view->lineno >= view->offset + view->height) {
1461                 view->lineno = view->offset + view->height - 1;
1462                 redraw_current_line = TRUE;
1463         }
1465         assert(view->offset <= view->lineno && view->lineno < view->lines);
1467         /* Redraw the whole screen if scrolling is pointless. */
1468         if (view->height < ABS(lines)) {
1469                 redraw_view(view);
1471         } else {
1472                 int line = lines > 0 ? view->height - lines : 0;
1473                 int end = line + ABS(lines);
1475                 wscrl(view->win, lines);
1477                 for (; line < end; line++) {
1478                         if (!draw_view_line(view, line))
1479                                 break;
1480                 }
1482                 if (redraw_current_line)
1483                         draw_view_line(view, view->lineno - view->offset);
1484         }
1486         redrawwin(view->win);
1487         wrefresh(view->win);
1488         report("");
1491 /* Scroll frontend */
1492 static void
1493 scroll_view(struct view *view, enum request request)
1495         int lines = 1;
1497         assert(view_is_displayed(view));
1499         switch (request) {
1500         case REQ_SCROLL_PAGE_DOWN:
1501                 lines = view->height;
1502         case REQ_SCROLL_LINE_DOWN:
1503                 if (view->offset + lines > view->lines)
1504                         lines = view->lines - view->offset;
1506                 if (lines == 0 || view->offset + view->height >= view->lines) {
1507                         report("Cannot scroll beyond the last line");
1508                         return;
1509                 }
1510                 break;
1512         case REQ_SCROLL_PAGE_UP:
1513                 lines = view->height;
1514         case REQ_SCROLL_LINE_UP:
1515                 if (lines > view->offset)
1516                         lines = view->offset;
1518                 if (lines == 0) {
1519                         report("Cannot scroll beyond the first line");
1520                         return;
1521                 }
1523                 lines = -lines;
1524                 break;
1526         default:
1527                 die("request %d not handled in switch", request);
1528         }
1530         do_scroll_view(view, lines);
1533 /* Cursor moving */
1534 static void
1535 move_view(struct view *view, enum request request)
1537         int scroll_steps = 0;
1538         int steps;
1540         switch (request) {
1541         case REQ_MOVE_FIRST_LINE:
1542                 steps = -view->lineno;
1543                 break;
1545         case REQ_MOVE_LAST_LINE:
1546                 steps = view->lines - view->lineno - 1;
1547                 break;
1549         case REQ_MOVE_PAGE_UP:
1550                 steps = view->height > view->lineno
1551                       ? -view->lineno : -view->height;
1552                 break;
1554         case REQ_MOVE_PAGE_DOWN:
1555                 steps = view->lineno + view->height >= view->lines
1556                       ? view->lines - view->lineno - 1 : view->height;
1557                 break;
1559         case REQ_MOVE_UP:
1560                 steps = -1;
1561                 break;
1563         case REQ_MOVE_DOWN:
1564                 steps = 1;
1565                 break;
1567         default:
1568                 die("request %d not handled in switch", request);
1569         }
1571         if (steps <= 0 && view->lineno == 0) {
1572                 report("Cannot move beyond the first line");
1573                 return;
1575         } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1576                 report("Cannot move beyond the last line");
1577                 return;
1578         }
1580         /* Move the current line */
1581         view->lineno += steps;
1582         assert(0 <= view->lineno && view->lineno < view->lines);
1584         /* Check whether the view needs to be scrolled */
1585         if (view->lineno < view->offset ||
1586             view->lineno >= view->offset + view->height) {
1587                 scroll_steps = steps;
1588                 if (steps < 0 && -steps > view->offset) {
1589                         scroll_steps = -view->offset;
1591                 } else if (steps > 0) {
1592                         if (view->lineno == view->lines - 1 &&
1593                             view->lines > view->height) {
1594                                 scroll_steps = view->lines - view->offset - 1;
1595                                 if (scroll_steps >= view->height)
1596                                         scroll_steps -= view->height - 1;
1597                         }
1598                 }
1599         }
1601         if (!view_is_displayed(view)) {
1602                 view->offset += steps;
1603                 view->ops->select(view, &view->line[view->lineno]);
1604                 return;
1605         }
1607         /* Repaint the old "current" line if we be scrolling */
1608         if (ABS(steps) < view->height)
1609                 draw_view_line(view, view->lineno - steps - view->offset);
1611         if (scroll_steps) {
1612                 do_scroll_view(view, scroll_steps);
1613                 return;
1614         }
1616         /* Draw the current line */
1617         draw_view_line(view, view->lineno - view->offset);
1619         redrawwin(view->win);
1620         wrefresh(view->win);
1621         report("");
1625 /*
1626  * Searching
1627  */
1629 static void search_view(struct view *view, enum request request);
1631 static bool
1632 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1634         assert(view_is_displayed(view));
1636         if (!view->ops->grep(view, line))
1637                 return FALSE;
1639         if (lineno - view->offset >= view->height) {
1640                 view->offset = lineno;
1641                 view->lineno = lineno;
1642                 redraw_view(view);
1644         } else {
1645                 unsigned long old_lineno = view->lineno - view->offset;
1647                 view->lineno = lineno;
1648                 draw_view_line(view, old_lineno);
1650                 draw_view_line(view, view->lineno - view->offset);
1651                 redrawwin(view->win);
1652                 wrefresh(view->win);
1653         }
1655         report("Line %ld matches '%s'", lineno + 1, view->grep);
1656         return TRUE;
1659 static void
1660 find_next(struct view *view, enum request request)
1662         unsigned long lineno = view->lineno;
1663         int direction;
1665         if (!*view->grep) {
1666                 if (!*opt_search)
1667                         report("No previous search");
1668                 else
1669                         search_view(view, request);
1670                 return;
1671         }
1673         switch (request) {
1674         case REQ_SEARCH:
1675         case REQ_FIND_NEXT:
1676                 direction = 1;
1677                 break;
1679         case REQ_SEARCH_BACK:
1680         case REQ_FIND_PREV:
1681                 direction = -1;
1682                 break;
1684         default:
1685                 return;
1686         }
1688         if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1689                 lineno += direction;
1691         /* Note, lineno is unsigned long so will wrap around in which case it
1692          * will become bigger than view->lines. */
1693         for (; lineno < view->lines; lineno += direction) {
1694                 struct line *line = &view->line[lineno];
1696                 if (find_next_line(view, lineno, line))
1697                         return;
1698         }
1700         report("No match found for '%s'", view->grep);
1703 static void
1704 search_view(struct view *view, enum request request)
1706         int regex_err;
1708         if (view->regex) {
1709                 regfree(view->regex);
1710                 *view->grep = 0;
1711         } else {
1712                 view->regex = calloc(1, sizeof(*view->regex));
1713                 if (!view->regex)
1714                         return;
1715         }
1717         regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
1718         if (regex_err != 0) {
1719                 char buf[SIZEOF_STR] = "unknown error";
1721                 regerror(regex_err, view->regex, buf, sizeof(buf));
1722                 report("Search failed: %s", buf);
1723                 return;
1724         }
1726         string_copy(view->grep, opt_search);
1728         find_next(view, request);
1731 /*
1732  * Incremental updating
1733  */
1735 static void
1736 end_update(struct view *view)
1738         if (!view->pipe)
1739                 return;
1740         set_nonblocking_input(FALSE);
1741         if (view->pipe == stdin)
1742                 fclose(view->pipe);
1743         else
1744                 pclose(view->pipe);
1745         view->pipe = NULL;
1748 static bool
1749 begin_update(struct view *view)
1751         const char *id = view->id;
1753         if (view->pipe)
1754                 end_update(view);
1756         if (opt_cmd[0]) {
1757                 string_copy(view->cmd, opt_cmd);
1758                 opt_cmd[0] = 0;
1759                 /* When running random commands, the view ref could have become
1760                  * invalid so clear it. */
1761                 view->ref[0] = 0;
1763         } else if (view == VIEW(REQ_VIEW_TREE)) {
1764                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1766                 if (strcmp(view->vid, view->id))
1767                         opt_path[0] = 0;
1769                 if (!string_format(view->cmd, format, id, opt_path))
1770                         return FALSE;
1772         } else {
1773                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1775                 if (!string_format(view->cmd, format, id, id, id, id, id))
1776                         return FALSE;
1777         }
1779         /* Special case for the pager view. */
1780         if (opt_pipe) {
1781                 view->pipe = opt_pipe;
1782                 opt_pipe = NULL;
1783         } else {
1784                 view->pipe = popen(view->cmd, "r");
1785         }
1787         if (!view->pipe)
1788                 return FALSE;
1790         set_nonblocking_input(TRUE);
1792         view->offset = 0;
1793         view->lines  = 0;
1794         view->lineno = 0;
1795         string_copy(view->vid, id);
1797         if (view->line) {
1798                 int i;
1800                 for (i = 0; i < view->lines; i++)
1801                         if (view->line[i].data)
1802                                 free(view->line[i].data);
1804                 free(view->line);
1805                 view->line = NULL;
1806         }
1808         view->start_time = time(NULL);
1810         return TRUE;
1813 static struct line *
1814 realloc_lines(struct view *view, size_t line_size)
1816         struct line *tmp = realloc(view->line, sizeof(*view->line) * line_size);
1818         if (!tmp)
1819                 return NULL;
1821         view->line = tmp;
1822         view->line_size = line_size;
1823         return view->line;
1826 static bool
1827 update_view(struct view *view)
1829         char in_buffer[BUFSIZ];
1830         char out_buffer[BUFSIZ * 2];
1831         char *line;
1832         /* The number of lines to read. If too low it will cause too much
1833          * redrawing (and possible flickering), if too high responsiveness
1834          * will suffer. */
1835         unsigned long lines = view->height;
1836         int redraw_from = -1;
1838         if (!view->pipe)
1839                 return TRUE;
1841         /* Only redraw if lines are visible. */
1842         if (view->offset + view->height >= view->lines)
1843                 redraw_from = view->lines - view->offset;
1845         /* FIXME: This is probably not perfect for backgrounded views. */
1846         if (!realloc_lines(view, view->lines + lines))
1847                 goto alloc_error;
1849         while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
1850                 size_t linelen = strlen(line);
1852                 if (linelen)
1853                         line[linelen - 1] = 0;
1855                 if (opt_iconv != ICONV_NONE) {
1856                         char *inbuf = line;
1857                         size_t inlen = linelen;
1859                         char *outbuf = out_buffer;
1860                         size_t outlen = sizeof(out_buffer);
1862                         size_t ret;
1864                         ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
1865                         if (ret != (size_t) -1) {
1866                                 line = out_buffer;
1867                                 linelen = strlen(out_buffer);
1868                         }
1869                 }
1871                 if (!view->ops->read(view, line))
1872                         goto alloc_error;
1874                 if (lines-- == 1)
1875                         break;
1876         }
1878         {
1879                 int digits;
1881                 lines = view->lines;
1882                 for (digits = 0; lines; digits++)
1883                         lines /= 10;
1885                 /* Keep the displayed view in sync with line number scaling. */
1886                 if (digits != view->digits) {
1887                         view->digits = digits;
1888                         redraw_from = 0;
1889                 }
1890         }
1892         if (!view_is_displayed(view))
1893                 goto check_pipe;
1895         if (view == VIEW(REQ_VIEW_TREE)) {
1896                 /* Clear the view and redraw everything since the tree sorting
1897                  * might have rearranged things. */
1898                 redraw_view(view);
1900         } else if (redraw_from >= 0) {
1901                 /* If this is an incremental update, redraw the previous line
1902                  * since for commits some members could have changed when
1903                  * loading the main view. */
1904                 if (redraw_from > 0)
1905                         redraw_from--;
1907                 /* Incrementally draw avoids flickering. */
1908                 redraw_view_from(view, redraw_from);
1909         }
1911         /* Update the title _after_ the redraw so that if the redraw picks up a
1912          * commit reference in view->ref it'll be available here. */
1913         update_view_title(view);
1915 check_pipe:
1916         if (ferror(view->pipe)) {
1917                 report("Failed to read: %s", strerror(errno));
1918                 goto end;
1920         } else if (feof(view->pipe)) {
1921                 report("");
1922                 goto end;
1923         }
1925         return TRUE;
1927 alloc_error:
1928         report("Allocation failure");
1930 end:
1931         view->ops->read(view, NULL);
1932         end_update(view);
1933         return FALSE;
1937 /*
1938  * View opening
1939  */
1941 static void open_help_view(struct view *view)
1943         char buf[BUFSIZ];
1944         int lines = ARRAY_SIZE(req_info) + 2;
1945         int i;
1947         if (view->lines > 0)
1948                 return;
1950         for (i = 0; i < ARRAY_SIZE(req_info); i++)
1951                 if (!req_info[i].request)
1952                         lines++;
1954         view->line = calloc(lines, sizeof(*view->line));
1955         if (!view->line) {
1956                 report("Allocation failure");
1957                 return;
1958         }
1960         view->ops->read(view, "Quick reference for tig keybindings:");
1962         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
1963                 char *key;
1965                 if (!req_info[i].request) {
1966                         view->ops->read(view, "");
1967                         view->ops->read(view, req_info[i].help);
1968                         continue;
1969                 }
1971                 key = get_key(req_info[i].request);
1972                 if (!string_format(buf, "%-25s %s", key, req_info[i].help))
1973                         continue;
1975                 view->ops->read(view, buf);
1976         }
1979 enum open_flags {
1980         OPEN_DEFAULT = 0,       /* Use default view switching. */
1981         OPEN_SPLIT = 1,         /* Split current view. */
1982         OPEN_BACKGROUNDED = 2,  /* Backgrounded. */
1983         OPEN_RELOAD = 4,        /* Reload view even if it is the current. */
1984 };
1986 static void
1987 open_view(struct view *prev, enum request request, enum open_flags flags)
1989         bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
1990         bool split = !!(flags & OPEN_SPLIT);
1991         bool reload = !!(flags & OPEN_RELOAD);
1992         struct view *view = VIEW(request);
1993         int nviews = displayed_views();
1994         struct view *base_view = display[0];
1996         if (view == prev && nviews == 1 && !reload) {
1997                 report("Already in %s view", view->name);
1998                 return;
1999         }
2001         if (view == VIEW(REQ_VIEW_HELP)) {
2002                 open_help_view(view);
2004         } else if ((reload || strcmp(view->vid, view->id)) &&
2005                    !begin_update(view)) {
2006                 report("Failed to load %s view", view->name);
2007                 return;
2008         }
2010         if (split) {
2011                 display[1] = view;
2012                 if (!backgrounded)
2013                         current_view = 1;
2014         } else {
2015                 /* Maximize the current view. */
2016                 memset(display, 0, sizeof(display));
2017                 current_view = 0;
2018                 display[current_view] = view;
2019         }
2021         /* Resize the view when switching between split- and full-screen,
2022          * or when switching between two different full-screen views. */
2023         if (nviews != displayed_views() ||
2024             (nviews == 1 && base_view != display[0]))
2025                 resize_display();
2027         if (split && prev->lineno - prev->offset >= prev->height) {
2028                 /* Take the title line into account. */
2029                 int lines = prev->lineno - prev->offset - prev->height + 1;
2031                 /* Scroll the view that was split if the current line is
2032                  * outside the new limited view. */
2033                 do_scroll_view(prev, lines);
2034         }
2036         if (prev && view != prev) {
2037                 if (split && !backgrounded) {
2038                         /* "Blur" the previous view. */
2039                         update_view_title(prev);
2040                 }
2042                 view->parent = prev;
2043         }
2045         if (view->pipe && view->lines == 0) {
2046                 /* Clear the old view and let the incremental updating refill
2047                  * the screen. */
2048                 wclear(view->win);
2049                 report("");
2050         } else {
2051                 redraw_view(view);
2052                 report("");
2053         }
2055         /* If the view is backgrounded the above calls to report()
2056          * won't redraw the view title. */
2057         if (backgrounded)
2058                 update_view_title(view);
2062 /*
2063  * User request switch noodle
2064  */
2066 static int
2067 view_driver(struct view *view, enum request request)
2069         int i;
2071         switch (request) {
2072         case REQ_MOVE_UP:
2073         case REQ_MOVE_DOWN:
2074         case REQ_MOVE_PAGE_UP:
2075         case REQ_MOVE_PAGE_DOWN:
2076         case REQ_MOVE_FIRST_LINE:
2077         case REQ_MOVE_LAST_LINE:
2078                 move_view(view, request);
2079                 break;
2081         case REQ_SCROLL_LINE_DOWN:
2082         case REQ_SCROLL_LINE_UP:
2083         case REQ_SCROLL_PAGE_DOWN:
2084         case REQ_SCROLL_PAGE_UP:
2085                 scroll_view(view, request);
2086                 break;
2088         case REQ_VIEW_BLOB:
2089                 if (!ref_blob[0]) {
2090                         report("No file chosen, press 't' to open tree view");
2091                         break;
2092                 }
2093                 /* Fall-through */
2094         case REQ_VIEW_MAIN:
2095         case REQ_VIEW_DIFF:
2096         case REQ_VIEW_LOG:
2097         case REQ_VIEW_TREE:
2098         case REQ_VIEW_HELP:
2099         case REQ_VIEW_PAGER:
2100                 open_view(view, request, OPEN_DEFAULT);
2101                 break;
2103         case REQ_NEXT:
2104         case REQ_PREVIOUS:
2105                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2107                 if ((view == VIEW(REQ_VIEW_DIFF) &&
2108                      view->parent == VIEW(REQ_VIEW_MAIN)) ||
2109                    (view == VIEW(REQ_VIEW_BLOB) &&
2110                      view->parent == VIEW(REQ_VIEW_TREE))) {
2111                         view = view->parent;
2112                         move_view(view, request);
2113                         if (view_is_displayed(view))
2114                                 update_view_title(view);
2115                 } else {
2116                         move_view(view, request);
2117                         break;
2118                 }
2119                 /* Fall-through */
2121         case REQ_ENTER:
2122                 if (!view->lines) {
2123                         report("Nothing to enter");
2124                         break;
2125                 }
2126                 return view->ops->enter(view, &view->line[view->lineno]);
2128         case REQ_VIEW_NEXT:
2129         {
2130                 int nviews = displayed_views();
2131                 int next_view = (current_view + 1) % nviews;
2133                 if (next_view == current_view) {
2134                         report("Only one view is displayed");
2135                         break;
2136                 }
2138                 current_view = next_view;
2139                 /* Blur out the title of the previous view. */
2140                 update_view_title(view);
2141                 report("");
2142                 break;
2143         }
2144         case REQ_TOGGLE_LINENO:
2145                 opt_line_number = !opt_line_number;
2146                 redraw_display();
2147                 break;
2149         case REQ_TOGGLE_REV_GRAPH:
2150                 opt_rev_graph = !opt_rev_graph;
2151                 redraw_display();
2152                 break;
2154         case REQ_PROMPT:
2155                 /* Always reload^Wrerun commands from the prompt. */
2156                 open_view(view, opt_request, OPEN_RELOAD);
2157                 break;
2159         case REQ_SEARCH:
2160         case REQ_SEARCH_BACK:
2161                 search_view(view, request);
2162                 break;
2164         case REQ_FIND_NEXT:
2165         case REQ_FIND_PREV:
2166                 find_next(view, request);
2167                 break;
2169         case REQ_STOP_LOADING:
2170                 for (i = 0; i < ARRAY_SIZE(views); i++) {
2171                         view = &views[i];
2172                         if (view->pipe)
2173                                 report("Stopped loading the %s view", view->name),
2174                         end_update(view);
2175                 }
2176                 break;
2178         case REQ_SHOW_VERSION:
2179                 report("%s (built %s)", VERSION, __DATE__);
2180                 return TRUE;
2182         case REQ_SCREEN_RESIZE:
2183                 resize_display();
2184                 /* Fall-through */
2185         case REQ_SCREEN_REDRAW:
2186                 redraw_display();
2187                 break;
2189         case REQ_NONE:
2190                 doupdate();
2191                 return TRUE;
2193         case REQ_VIEW_CLOSE:
2194                 /* XXX: Mark closed views by letting view->parent point to the
2195                  * view itself. Parents to closed view should never be
2196                  * followed. */
2197                 if (view->parent &&
2198                     view->parent->parent != view->parent) {
2199                         memset(display, 0, sizeof(display));
2200                         current_view = 0;
2201                         display[current_view] = view->parent;
2202                         view->parent = view;
2203                         resize_display();
2204                         redraw_display();
2205                         break;
2206                 }
2207                 /* Fall-through */
2208         case REQ_QUIT:
2209                 return FALSE;
2211         default:
2212                 /* An unknown key will show most commonly used commands. */
2213                 report("Unknown key, press 'h' for help");
2214                 return TRUE;
2215         }
2217         return TRUE;
2221 /*
2222  * Pager backend
2223  */
2225 static bool
2226 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2228         char *text = line->data;
2229         enum line_type type = line->type;
2230         int textlen = strlen(text);
2231         int attr;
2233         wmove(view->win, lineno, 0);
2235         if (selected) {
2236                 type = LINE_CURSOR;
2237                 wchgat(view->win, -1, 0, type, NULL);
2238         }
2240         attr = get_line_attr(type);
2241         wattrset(view->win, attr);
2243         if (opt_line_number || opt_tab_size < TABSIZE) {
2244                 static char spaces[] = "                    ";
2245                 int col_offset = 0, col = 0;
2247                 if (opt_line_number) {
2248                         unsigned long real_lineno = view->offset + lineno + 1;
2250                         if (real_lineno == 1 ||
2251                             (real_lineno % opt_num_interval) == 0) {
2252                                 wprintw(view->win, "%.*d", view->digits, real_lineno);
2254                         } else {
2255                                 waddnstr(view->win, spaces,
2256                                          MIN(view->digits, STRING_SIZE(spaces)));
2257                         }
2258                         waddstr(view->win, ": ");
2259                         col_offset = view->digits + 2;
2260                 }
2262                 while (text && col_offset + col < view->width) {
2263                         int cols_max = view->width - col_offset - col;
2264                         char *pos = text;
2265                         int cols;
2267                         if (*text == '\t') {
2268                                 text++;
2269                                 assert(sizeof(spaces) > TABSIZE);
2270                                 pos = spaces;
2271                                 cols = opt_tab_size - (col % opt_tab_size);
2273                         } else {
2274                                 text = strchr(text, '\t');
2275                                 cols = line ? text - pos : strlen(pos);
2276                         }
2278                         waddnstr(view->win, pos, MIN(cols, cols_max));
2279                         col += cols;
2280                 }
2282         } else {
2283                 int col = 0, pos = 0;
2285                 for (; pos < textlen && col < view->width; pos++, col++)
2286                         if (text[pos] == '\t')
2287                                 col += TABSIZE - (col % TABSIZE) - 1;
2289                 waddnstr(view->win, text, pos);
2290         }
2292         return TRUE;
2295 static bool
2296 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2298         char refbuf[SIZEOF_STR];
2299         char *ref = NULL;
2300         FILE *pipe;
2302         if (!string_format(refbuf, "git describe %s", commit_id))
2303                 return TRUE;
2305         pipe = popen(refbuf, "r");
2306         if (!pipe)
2307                 return TRUE;
2309         if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2310                 ref = chomp_string(ref);
2311         pclose(pipe);
2313         if (!ref || !*ref)
2314                 return TRUE;
2316         /* This is the only fatal call, since it can "corrupt" the buffer. */
2317         if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2318                 return FALSE;
2320         return TRUE;
2323 static void
2324 add_pager_refs(struct view *view, struct line *line)
2326         char buf[SIZEOF_STR];
2327         char *commit_id = line->data + STRING_SIZE("commit ");
2328         struct ref **refs;
2329         size_t bufpos = 0, refpos = 0;
2330         const char *sep = "Refs: ";
2331         bool is_tag = FALSE;
2333         assert(line->type == LINE_COMMIT);
2335         refs = get_refs(commit_id);
2336         if (!refs) {
2337                 if (view == VIEW(REQ_VIEW_DIFF))
2338                         goto try_add_describe_ref;
2339                 return;
2340         }
2342         do {
2343                 struct ref *ref = refs[refpos];
2344                 char *fmt = ref->tag ? "%s[%s]" : "%s%s";
2346                 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2347                         return;
2348                 sep = ", ";
2349                 if (ref->tag)
2350                         is_tag = TRUE;
2351         } while (refs[refpos++]->next);
2353         if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2354 try_add_describe_ref:
2355                 /* Add <tag>-g<commit_id> "fake" reference. */
2356                 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2357                         return;
2358         }
2360         if (bufpos == 0)
2361                 return;
2363         if (!realloc_lines(view, view->line_size + 1))
2364                 return;
2366         line = &view->line[view->lines];
2367         line->data = strdup(buf);
2368         if (!line->data)
2369                 return;
2371         line->type = LINE_PP_REFS;
2372         view->lines++;
2375 static bool
2376 pager_read(struct view *view, char *data)
2378         struct line *line = &view->line[view->lines];
2380         if (!data)
2381                 return TRUE;
2383         line->data = strdup(data);
2384         if (!line->data)
2385                 return FALSE;
2387         line->type = get_line_type(line->data);
2388         view->lines++;
2390         if (line->type == LINE_COMMIT &&
2391             (view == VIEW(REQ_VIEW_DIFF) ||
2392              view == VIEW(REQ_VIEW_LOG)))
2393                 add_pager_refs(view, line);
2395         return TRUE;
2398 static bool
2399 pager_enter(struct view *view, struct line *line)
2401         int split = 0;
2403         if (line->type == LINE_COMMIT &&
2404            (view == VIEW(REQ_VIEW_LOG) ||
2405             view == VIEW(REQ_VIEW_PAGER))) {
2406                 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2407                 split = 1;
2408         }
2410         /* Always scroll the view even if it was split. That way
2411          * you can use Enter to scroll through the log view and
2412          * split open each commit diff. */
2413         scroll_view(view, REQ_SCROLL_LINE_DOWN);
2415         /* FIXME: A minor workaround. Scrolling the view will call report("")
2416          * but if we are scrolling a non-current view this won't properly
2417          * update the view title. */
2418         if (split)
2419                 update_view_title(view);
2421         return TRUE;
2424 static bool
2425 pager_grep(struct view *view, struct line *line)
2427         regmatch_t pmatch;
2428         char *text = line->data;
2430         if (!*text)
2431                 return FALSE;
2433         if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2434                 return FALSE;
2436         return TRUE;
2439 static void
2440 pager_select(struct view *view, struct line *line)
2442         if (line->type == LINE_COMMIT) {
2443                 char *text = line->data;
2445                 string_copy(view->ref, text + STRING_SIZE("commit "));
2446                 string_copy(ref_commit, view->ref);
2447         }
2450 static struct view_ops pager_ops = {
2451         "line",
2452         pager_draw,
2453         pager_read,
2454         pager_enter,
2455         pager_grep,
2456         pager_select,
2457 };
2460 /*
2461  * Tree backend
2462  */
2464 /* Parse output from git-ls-tree(1):
2465  *
2466  * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
2467  * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
2468  * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
2469  * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
2470  */
2472 #define SIZEOF_TREE_ATTR \
2473         STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
2475 #define TREE_UP_FORMAT "040000 tree %s\t.."
2477 static int
2478 tree_compare_entry(enum line_type type1, char *name1,
2479                    enum line_type type2, char *name2)
2481         if (type1 != type2) {
2482                 if (type1 == LINE_TREE_DIR)
2483                         return -1;
2484                 return 1;
2485         }
2487         return strcmp(name1, name2);
2490 static bool
2491 tree_read(struct view *view, char *text)
2493         size_t textlen = text ? strlen(text) : 0;
2494         char buf[SIZEOF_STR];
2495         unsigned long pos;
2496         enum line_type type;
2497         bool first_read = view->lines == 0;
2499         if (textlen <= SIZEOF_TREE_ATTR)
2500                 return FALSE;
2502         type = text[STRING_SIZE("100644 ")] == 't'
2503              ? LINE_TREE_DIR : LINE_TREE_FILE;
2505         if (first_read) {
2506                 /* Add path info line */
2507                 if (string_format(buf, "Directory path /%s", opt_path) &&
2508                     realloc_lines(view, view->line_size + 1) &&
2509                     pager_read(view, buf))
2510                         view->line[view->lines - 1].type = LINE_DEFAULT;
2511                 else
2512                         return FALSE;
2514                 /* Insert "link" to parent directory. */
2515                 if (*opt_path &&
2516                     string_format(buf, TREE_UP_FORMAT, view->ref) &&
2517                     realloc_lines(view, view->line_size + 1) &&
2518                     pager_read(view, buf))
2519                         view->line[view->lines - 1].type = LINE_TREE_DIR;
2520                 else if (*opt_path)
2521                         return FALSE;
2522         }
2524         /* Strip the path part ... */
2525         if (*opt_path) {
2526                 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
2527                 size_t striplen = strlen(opt_path);
2528                 char *path = text + SIZEOF_TREE_ATTR;
2530                 if (pathlen > striplen)
2531                         memmove(path, path + striplen,
2532                                 pathlen - striplen + 1);
2533         }
2535         /* Skip "Directory ..." and ".." line. */
2536         for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
2537                 struct line *line = &view->line[pos];
2538                 char *path1 = ((char *) line->data) + SIZEOF_TREE_ATTR;
2539                 char *path2 = text + SIZEOF_TREE_ATTR;
2540                 int cmp = tree_compare_entry(line->type, path1, type, path2);
2542                 if (cmp <= 0)
2543                         continue;
2545                 text = strdup(text);
2546                 if (!text)
2547                         return FALSE;
2549                 if (view->lines > pos)
2550                         memmove(&view->line[pos + 1], &view->line[pos],
2551                                 (view->lines - pos) * sizeof(*line));
2553                 line = &view->line[pos];
2554                 line->data = text;
2555                 line->type = type;
2556                 view->lines++;
2557                 return TRUE;
2558         }
2560         if (!pager_read(view, text))
2561                 return FALSE;
2563         /* Move the current line to the first tree entry. */
2564         if (first_read)
2565                 view->lineno++;
2567         view->line[view->lines - 1].type = type;
2568         return TRUE;
2571 static bool
2572 tree_enter(struct view *view, struct line *line)
2574         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2575         enum request request;
2577         switch (line->type) {
2578         case LINE_TREE_DIR:
2579                 /* Depending on whether it is a subdir or parent (updir?) link
2580                  * mangle the path buffer. */
2581                 if (line == &view->line[1] && *opt_path) {
2582                         size_t path_len = strlen(opt_path);
2583                         char *dirsep = opt_path + path_len - 1;
2585                         while (dirsep > opt_path && dirsep[-1] != '/')
2586                                 dirsep--;
2588                         dirsep[0] = 0;
2590                 } else {
2591                         size_t pathlen = strlen(opt_path);
2592                         size_t origlen = pathlen;
2593                         char *data = line->data;
2594                         char *basename = data + SIZEOF_TREE_ATTR;
2596                         if (!string_format_from(opt_path, &pathlen, "%s/", basename)) {
2597                                 opt_path[origlen] = 0;
2598                                 return TRUE;
2599                         }
2600                 }
2602                 /* Trees and subtrees share the same ID, so they are not not
2603                  * unique like blobs. */
2604                 flags |= OPEN_RELOAD;
2605                 request = REQ_VIEW_TREE;
2606                 break;
2608         case LINE_TREE_FILE:
2609                 request = REQ_VIEW_BLOB;
2610                 break;
2612         default:
2613                 return TRUE;
2614         }
2616         open_view(view, request, flags);
2618         return TRUE;
2621 static void
2622 tree_select(struct view *view, struct line *line)
2624         char *text = line->data;
2626         text += STRING_SIZE("100644 blob ");
2628         if (line->type == LINE_TREE_FILE) {
2629                 string_ncopy(ref_blob, text, 40);
2630                 /* Also update the blob view's ref, since all there must always
2631                  * be in sync. */
2632                 string_copy(VIEW(REQ_VIEW_BLOB)->ref, ref_blob);
2634         } else if (line->type != LINE_TREE_DIR) {
2635                 return;
2636         }
2638         string_ncopy(view->ref, text, 40);
2641 static struct view_ops tree_ops = {
2642         "file",
2643         pager_draw,
2644         tree_read,
2645         tree_enter,
2646         pager_grep,
2647         tree_select,
2648 };
2650 static bool
2651 blob_read(struct view *view, char *line)
2653         bool state = pager_read(view, line);
2655         if (state == TRUE)
2656                 view->line[view->lines - 1].type = LINE_DEFAULT;
2658         return state;
2661 static struct view_ops blob_ops = {
2662         "line",
2663         pager_draw,
2664         blob_read,
2665         pager_enter,
2666         pager_grep,
2667         pager_select,
2668 };
2671 /*
2672  * Revision graph
2673  */
2675 struct commit {
2676         char id[SIZEOF_REV];            /* SHA1 ID. */
2677         char title[75];                 /* First line of the commit message. */
2678         char author[75];                /* Author of the commit. */
2679         struct tm time;                 /* Date from the author ident. */
2680         struct ref **refs;              /* Repository references. */
2681         chtype graph[SIZEOF_REVGRAPH];  /* Ancestry chain graphics. */
2682         size_t graph_size;              /* The width of the graph array. */
2683 };
2685 /* Size of rev graph with no  "padding" columns */
2686 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
2688 struct rev_graph {
2689         struct rev_graph *prev, *next, *parents;
2690         char rev[SIZEOF_REVITEMS][SIZEOF_REV];
2691         size_t size;
2692         struct commit *commit;
2693         size_t pos;
2694 };
2696 /* Parents of the commit being visualized. */
2697 static struct rev_graph graph_parents[4];
2699 /* The current stack of revisions on the graph. */
2700 static struct rev_graph graph_stacks[4] = {
2701         { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
2702         { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
2703         { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
2704         { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
2705 };
2707 static inline bool
2708 graph_parent_is_merge(struct rev_graph *graph)
2710         return graph->parents->size > 1;
2713 static inline void
2714 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
2716         struct commit *commit = graph->commit;
2718         if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
2719                 commit->graph[commit->graph_size++] = symbol;
2722 static void
2723 done_rev_graph(struct rev_graph *graph)
2725         if (graph_parent_is_merge(graph) &&
2726             graph->pos < graph->size - 1 &&
2727             graph->next->size == graph->size + graph->parents->size - 1) {
2728                 size_t i = graph->pos + graph->parents->size - 1;
2730                 graph->commit->graph_size = i * 2;
2731                 while (i < graph->next->size - 1) {
2732                         append_to_rev_graph(graph, ' ');
2733                         append_to_rev_graph(graph, '\\');
2734                         i++;
2735                 }
2736         }
2738         graph->size = graph->pos = 0;
2739         graph->commit = NULL;
2740         memset(graph->parents, 0, sizeof(*graph->parents));
2743 static void
2744 push_rev_graph(struct rev_graph *graph, char *parent)
2746         /* Combine duplicate parents lines. */
2747         if (graph->size > 0 &&
2748             !strncmp(graph->rev[graph->size - 1], parent, SIZEOF_REV))
2749                 return;
2751         if (graph->size < SIZEOF_REVITEMS) {
2752                 string_ncopy(graph->rev[graph->size++], parent, SIZEOF_REV);
2753         }
2756 static chtype
2757 get_rev_graph_symbol(struct rev_graph *graph)
2759         chtype symbol;
2761         if (graph->parents->size == 0)
2762                 symbol = REVGRAPH_INIT;
2763         else if (graph_parent_is_merge(graph))
2764                 symbol = REVGRAPH_MERGE;
2765         else if (graph->pos >= graph->size)
2766                 symbol = REVGRAPH_BRANCH;
2767         else
2768                 symbol = REVGRAPH_COMMIT;
2770         return symbol;
2773 static void
2774 draw_rev_graph(struct rev_graph *graph)
2776         struct rev_filler {
2777                 chtype separator, line;
2778         };
2779         enum { DEFAULT, RSHARP, RDIAG, LDIAG };
2780         static struct rev_filler fillers[] = {
2781                 { ' ',  REVGRAPH_LINE },
2782                 { '`',  '.' },
2783                 { '\'', ' ' },
2784                 { '/',  ' ' },
2785         };
2786         chtype symbol = get_rev_graph_symbol(graph);
2787         struct rev_filler *filler;
2788         size_t i;
2790         filler = &fillers[DEFAULT];
2792         for (i = 0; i < graph->pos; i++) {
2793                 append_to_rev_graph(graph, filler->line);
2794                 if (graph_parent_is_merge(graph->prev) &&
2795                     graph->prev->pos == i)
2796                         filler = &fillers[RSHARP];
2798                 append_to_rev_graph(graph, filler->separator);
2799         }
2801         /* Place the symbol for this revision. */
2802         append_to_rev_graph(graph, symbol);
2804         if (graph->prev->size > graph->size)
2805                 filler = &fillers[RDIAG];
2806         else
2807                 filler = &fillers[DEFAULT];
2809         i++;
2811         for (; i < graph->size; i++) {
2812                 append_to_rev_graph(graph, filler->separator);
2813                 append_to_rev_graph(graph, filler->line);
2814                 if (graph_parent_is_merge(graph->prev) &&
2815                     i < graph->prev->pos + graph->parents->size)
2816                         filler = &fillers[RSHARP];
2817                 if (graph->prev->size > graph->size)
2818                         filler = &fillers[LDIAG];
2819         }
2821         if (graph->prev->size > graph->size) {
2822                 append_to_rev_graph(graph, filler->separator);
2823                 if (filler->line != ' ')
2824                         append_to_rev_graph(graph, filler->line);
2825         }
2828 /* Prepare the next rev graph */
2829 static void
2830 prepare_rev_graph(struct rev_graph *graph)
2832         size_t i;
2834         /* First, traverse all lines of revisions up to the active one. */
2835         for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
2836                 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
2837                         break;
2839                 push_rev_graph(graph->next, graph->rev[graph->pos]);
2840         }
2842         /* Interleave the new revision parent(s). */
2843         for (i = 0; i < graph->parents->size; i++)
2844                 push_rev_graph(graph->next, graph->parents->rev[i]);
2846         /* Lastly, put any remaining revisions. */
2847         for (i = graph->pos + 1; i < graph->size; i++)
2848                 push_rev_graph(graph->next, graph->rev[i]);
2851 static void
2852 update_rev_graph(struct rev_graph *graph)
2854         /* If this is the finalizing update ... */
2855         if (graph->commit)
2856                 prepare_rev_graph(graph);
2858         /* Graph visualization needs a one rev look-ahead,
2859          * so the first update doesn't visualize anything. */
2860         if (!graph->prev->commit)
2861                 return;
2863         draw_rev_graph(graph->prev);
2864         done_rev_graph(graph->prev->prev);
2868 /*
2869  * Main view backend
2870  */
2872 static bool
2873 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2875         char buf[DATE_COLS + 1];
2876         struct commit *commit = line->data;
2877         enum line_type type;
2878         int col = 0;
2879         size_t timelen;
2880         size_t authorlen;
2881         int trimmed = 1;
2883         if (!*commit->author)
2884                 return FALSE;
2886         wmove(view->win, lineno, col);
2888         if (selected) {
2889                 type = LINE_CURSOR;
2890                 wattrset(view->win, get_line_attr(type));
2891                 wchgat(view->win, -1, 0, type, NULL);
2893         } else {
2894                 type = LINE_MAIN_COMMIT;
2895                 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
2896         }
2898         timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
2899         waddnstr(view->win, buf, timelen);
2900         waddstr(view->win, " ");
2902         col += DATE_COLS;
2903         wmove(view->win, lineno, col);
2904         if (type != LINE_CURSOR)
2905                 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
2907         if (opt_utf8) {
2908                 authorlen = utf8_length(commit->author, AUTHOR_COLS - 2, &col, &trimmed);
2909         } else {
2910                 authorlen = strlen(commit->author);
2911                 if (authorlen > AUTHOR_COLS - 2) {
2912                         authorlen = AUTHOR_COLS - 2;
2913                         trimmed = 1;
2914                 }
2915         }
2917         if (trimmed) {
2918                 waddnstr(view->win, commit->author, authorlen);
2919                 if (type != LINE_CURSOR)
2920                         wattrset(view->win, get_line_attr(LINE_MAIN_DELIM));
2921                 waddch(view->win, '~');
2922         } else {
2923                 waddstr(view->win, commit->author);
2924         }
2926         col += AUTHOR_COLS;
2927         if (type != LINE_CURSOR)
2928                 wattrset(view->win, A_NORMAL);
2930         if (opt_rev_graph && commit->graph_size) {
2931                 size_t i;
2933                 wmove(view->win, lineno, col);
2934                 /* Using waddch() instead of waddnstr() ensures that
2935                  * they'll be rendered correctly for the cursor line. */
2936                 for (i = 0; i < commit->graph_size; i++)
2937                         waddch(view->win, commit->graph[i]);
2939                 waddch(view->win, ' ');
2940                 col += commit->graph_size + 1;
2941         }
2943         wmove(view->win, lineno, col);
2945         if (commit->refs) {
2946                 size_t i = 0;
2948                 do {
2949                         if (type == LINE_CURSOR)
2950                                 ;
2951                         else if (commit->refs[i]->tag)
2952                                 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
2953                         else
2954                                 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
2955                         waddstr(view->win, "[");
2956                         waddstr(view->win, commit->refs[i]->name);
2957                         waddstr(view->win, "]");
2958                         if (type != LINE_CURSOR)
2959                                 wattrset(view->win, A_NORMAL);
2960                         waddstr(view->win, " ");
2961                         col += strlen(commit->refs[i]->name) + STRING_SIZE("[] ");
2962                 } while (commit->refs[i++]->next);
2963         }
2965         if (type != LINE_CURSOR)
2966                 wattrset(view->win, get_line_attr(type));
2968         {
2969                 int titlelen = strlen(commit->title);
2971                 if (col + titlelen > view->width)
2972                         titlelen = view->width - col;
2974                 waddnstr(view->win, commit->title, titlelen);
2975         }
2977         return TRUE;
2980 /* Reads git log --pretty=raw output and parses it into the commit struct. */
2981 static bool
2982 main_read(struct view *view, char *line)
2984         static struct rev_graph *graph = graph_stacks;
2985         enum line_type type;
2986         struct commit *commit = view->lines
2987                               ? view->line[view->lines - 1].data : NULL;
2989         if (!line) {
2990                 update_rev_graph(graph);
2991                 return TRUE;
2992         }
2994         type = get_line_type(line);
2996         switch (type) {
2997         case LINE_COMMIT:
2998                 commit = calloc(1, sizeof(struct commit));
2999                 if (!commit)
3000                         return FALSE;
3002                 line += STRING_SIZE("commit ");
3004                 view->line[view->lines++].data = commit;
3005                 string_copy(commit->id, line);
3006                 commit->refs = get_refs(commit->id);
3007                 graph->commit = commit;
3008                 break;
3010         case LINE_PARENT:
3011                 if (commit) {
3012                         line += STRING_SIZE("parent ");
3013                         push_rev_graph(graph->parents, line);
3014                 }
3015                 break;
3017         case LINE_AUTHOR:
3018         {
3019                 char *ident = line + STRING_SIZE("author ");
3020                 char *end = strchr(ident, '<');
3022                 if (!commit)
3023                         break;
3025                 update_rev_graph(graph);
3026                 graph = graph->next;
3028                 if (end) {
3029                         char *email = end + 1;
3031                         for (; end > ident && isspace(end[-1]); end--) ;
3033                         if (end == ident && *email) {
3034                                 ident = email;
3035                                 end = strchr(ident, '>');
3036                                 for (; end > ident && isspace(end[-1]); end--) ;
3037                         }
3038                         *end = 0;
3039                 }
3041                 /* End is NULL or ident meaning there's no author. */
3042                 if (end <= ident)
3043                         ident = "Unknown";
3045                 string_copy(commit->author, ident);
3047                 /* Parse epoch and timezone */
3048                 if (end) {
3049                         char *secs = strchr(end + 1, '>');
3050                         char *zone;
3051                         time_t time;
3053                         if (!secs || secs[1] != ' ')
3054                                 break;
3056                         secs += 2;
3057                         time = (time_t) atol(secs);
3058                         zone = strchr(secs, ' ');
3059                         if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
3060                                 long tz;
3062                                 zone++;
3063                                 tz  = ('0' - zone[1]) * 60 * 60 * 10;
3064                                 tz += ('0' - zone[2]) * 60 * 60;
3065                                 tz += ('0' - zone[3]) * 60;
3066                                 tz += ('0' - zone[4]) * 60;
3068                                 if (zone[0] == '-')
3069                                         tz = -tz;
3071                                 time -= tz;
3072                         }
3073                         gmtime_r(&time, &commit->time);
3074                 }
3075                 break;
3076         }
3077         default:
3078                 if (!commit)
3079                         break;
3081                 /* Fill in the commit title if it has not already been set. */
3082                 if (commit->title[0])
3083                         break;
3085                 /* Require titles to start with a non-space character at the
3086                  * offset used by git log. */
3087                 /* FIXME: More gracefull handling of titles; append "..." to
3088                  * shortened titles, etc. */
3089                 if (strncmp(line, "    ", 4) ||
3090                     isspace(line[4]))
3091                         break;
3093                 string_copy(commit->title, line + 4);
3094         }
3096         return TRUE;
3099 static bool
3100 main_enter(struct view *view, struct line *line)
3102         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3104         open_view(view, REQ_VIEW_DIFF, flags);
3105         return TRUE;
3108 static bool
3109 main_grep(struct view *view, struct line *line)
3111         struct commit *commit = line->data;
3112         enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
3113         char buf[DATE_COLS + 1];
3114         regmatch_t pmatch;
3116         for (state = S_TITLE; state < S_END; state++) {
3117                 char *text;
3119                 switch (state) {
3120                 case S_TITLE:   text = commit->title;   break;
3121                 case S_AUTHOR:  text = commit->author;  break;
3122                 case S_DATE:
3123                         if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
3124                                 continue;
3125                         text = buf;
3126                         break;
3128                 default:
3129                         return FALSE;
3130                 }
3132                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3133                         return TRUE;
3134         }
3136         return FALSE;
3139 static void
3140 main_select(struct view *view, struct line *line)
3142         struct commit *commit = line->data;
3144         string_copy(view->ref, commit->id);
3145         string_copy(ref_commit, view->ref);
3148 static struct view_ops main_ops = {
3149         "commit",
3150         main_draw,
3151         main_read,
3152         main_enter,
3153         main_grep,
3154         main_select,
3155 };
3158 /*
3159  * Unicode / UTF-8 handling
3160  *
3161  * NOTE: Much of the following code for dealing with unicode is derived from
3162  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
3163  * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
3164  */
3166 /* I've (over)annotated a lot of code snippets because I am not entirely
3167  * confident that the approach taken by this small UTF-8 interface is correct.
3168  * --jonas */
3170 static inline int
3171 unicode_width(unsigned long c)
3173         if (c >= 0x1100 &&
3174            (c <= 0x115f                         /* Hangul Jamo */
3175             || c == 0x2329
3176             || c == 0x232a
3177             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
3178                                                 /* CJK ... Yi */
3179             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
3180             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
3181             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
3182             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
3183             || (c >= 0xffe0  && c <= 0xffe6)
3184             || (c >= 0x20000 && c <= 0x2fffd)
3185             || (c >= 0x30000 && c <= 0x3fffd)))
3186                 return 2;
3188         return 1;
3191 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
3192  * Illegal bytes are set one. */
3193 static const unsigned char utf8_bytes[256] = {
3194         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,
3195         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,
3196         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,
3197         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,
3198         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,
3199         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,
3200         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,
3201         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,
3202 };
3204 /* Decode UTF-8 multi-byte representation into a unicode character. */
3205 static inline unsigned long
3206 utf8_to_unicode(const char *string, size_t length)
3208         unsigned long unicode;
3210         switch (length) {
3211         case 1:
3212                 unicode  =   string[0];
3213                 break;
3214         case 2:
3215                 unicode  =  (string[0] & 0x1f) << 6;
3216                 unicode +=  (string[1] & 0x3f);
3217                 break;
3218         case 3:
3219                 unicode  =  (string[0] & 0x0f) << 12;
3220                 unicode += ((string[1] & 0x3f) << 6);
3221                 unicode +=  (string[2] & 0x3f);
3222                 break;
3223         case 4:
3224                 unicode  =  (string[0] & 0x0f) << 18;
3225                 unicode += ((string[1] & 0x3f) << 12);
3226                 unicode += ((string[2] & 0x3f) << 6);
3227                 unicode +=  (string[3] & 0x3f);
3228                 break;
3229         case 5:
3230                 unicode  =  (string[0] & 0x0f) << 24;
3231                 unicode += ((string[1] & 0x3f) << 18);
3232                 unicode += ((string[2] & 0x3f) << 12);
3233                 unicode += ((string[3] & 0x3f) << 6);
3234                 unicode +=  (string[4] & 0x3f);
3235                 break;
3236         case 6:
3237                 unicode  =  (string[0] & 0x01) << 30;
3238                 unicode += ((string[1] & 0x3f) << 24);
3239                 unicode += ((string[2] & 0x3f) << 18);
3240                 unicode += ((string[3] & 0x3f) << 12);
3241                 unicode += ((string[4] & 0x3f) << 6);
3242                 unicode +=  (string[5] & 0x3f);
3243                 break;
3244         default:
3245                 die("Invalid unicode length");
3246         }
3248         /* Invalid characters could return the special 0xfffd value but NUL
3249          * should be just as good. */
3250         return unicode > 0xffff ? 0 : unicode;
3253 /* Calculates how much of string can be shown within the given maximum width
3254  * and sets trimmed parameter to non-zero value if all of string could not be
3255  * shown.
3256  *
3257  * Additionally, adds to coloffset how many many columns to move to align with
3258  * the expected position. Takes into account how multi-byte and double-width
3259  * characters will effect the cursor position.
3260  *
3261  * Returns the number of bytes to output from string to satisfy max_width. */
3262 static size_t
3263 utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed)
3265         const char *start = string;
3266         const char *end = strchr(string, '\0');
3267         size_t mbwidth = 0;
3268         size_t width = 0;
3270         *trimmed = 0;
3272         while (string < end) {
3273                 int c = *(unsigned char *) string;
3274                 unsigned char bytes = utf8_bytes[c];
3275                 size_t ucwidth;
3276                 unsigned long unicode;
3278                 if (string + bytes > end)
3279                         break;
3281                 /* Change representation to figure out whether
3282                  * it is a single- or double-width character. */
3284                 unicode = utf8_to_unicode(string, bytes);
3285                 /* FIXME: Graceful handling of invalid unicode character. */
3286                 if (!unicode)
3287                         break;
3289                 ucwidth = unicode_width(unicode);
3290                 width  += ucwidth;
3291                 if (width > max_width) {
3292                         *trimmed = 1;
3293                         break;
3294                 }
3296                 /* The column offset collects the differences between the
3297                  * number of bytes encoding a character and the number of
3298                  * columns will be used for rendering said character.
3299                  *
3300                  * So if some character A is encoded in 2 bytes, but will be
3301                  * represented on the screen using only 1 byte this will and up
3302                  * adding 1 to the multi-byte column offset.
3303                  *
3304                  * Assumes that no double-width character can be encoding in
3305                  * less than two bytes. */
3306                 if (bytes > ucwidth)
3307                         mbwidth += bytes - ucwidth;
3309                 string  += bytes;
3310         }
3312         *coloffset += mbwidth;
3314         return string - start;
3318 /*
3319  * Status management
3320  */
3322 /* Whether or not the curses interface has been initialized. */
3323 static bool cursed = FALSE;
3325 /* The status window is used for polling keystrokes. */
3326 static WINDOW *status_win;
3328 /* Update status and title window. */
3329 static void
3330 report(const char *msg, ...)
3332         static bool empty = TRUE;
3333         struct view *view = display[current_view];
3335         if (!empty || *msg) {
3336                 va_list args;
3338                 va_start(args, msg);
3340                 werase(status_win);
3341                 wmove(status_win, 0, 0);
3342                 if (*msg) {
3343                         vwprintw(status_win, msg, args);
3344                         empty = FALSE;
3345                 } else {
3346                         empty = TRUE;
3347                 }
3348                 wrefresh(status_win);
3350                 va_end(args);
3351         }
3353         update_view_title(view);
3354         update_display_cursor();
3357 /* Controls when nodelay should be in effect when polling user input. */
3358 static void
3359 set_nonblocking_input(bool loading)
3361         static unsigned int loading_views;
3363         if ((loading == FALSE && loading_views-- == 1) ||
3364             (loading == TRUE  && loading_views++ == 0))
3365                 nodelay(status_win, loading);
3368 static void
3369 init_display(void)
3371         int x, y;
3373         /* Initialize the curses library */
3374         if (isatty(STDIN_FILENO)) {
3375                 cursed = !!initscr();
3376         } else {
3377                 /* Leave stdin and stdout alone when acting as a pager. */
3378                 FILE *io = fopen("/dev/tty", "r+");
3380                 if (!io)
3381                         die("Failed to open /dev/tty");
3382                 cursed = !!newterm(NULL, io, io);
3383         }
3385         if (!cursed)
3386                 die("Failed to initialize curses");
3388         nonl();         /* Tell curses not to do NL->CR/NL on output */
3389         cbreak();       /* Take input chars one at a time, no wait for \n */
3390         noecho();       /* Don't echo input */
3391         leaveok(stdscr, TRUE);
3393         if (has_colors())
3394                 init_colors();
3396         getmaxyx(stdscr, y, x);
3397         status_win = newwin(1, 0, y - 1, 0);
3398         if (!status_win)
3399                 die("Failed to create status window");
3401         /* Enable keyboard mapping */
3402         keypad(status_win, TRUE);
3403         wbkgdset(status_win, get_line_attr(LINE_STATUS));
3406 static char *
3407 read_prompt(const char *prompt)
3409         enum { READING, STOP, CANCEL } status = READING;
3410         static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
3411         int pos = 0;
3413         while (status == READING) {
3414                 struct view *view;
3415                 int i, key;
3417                 foreach_view (view, i)
3418                         update_view(view);
3420                 report("%s%.*s", prompt, pos, buf);
3421                 /* Refresh, accept single keystroke of input */
3422                 key = wgetch(status_win);
3423                 switch (key) {
3424                 case KEY_RETURN:
3425                 case KEY_ENTER:
3426                 case '\n':
3427                         status = pos ? STOP : CANCEL;
3428                         break;
3430                 case KEY_BACKSPACE:
3431                         if (pos > 0)
3432                                 pos--;
3433                         else
3434                                 status = CANCEL;
3435                         break;
3437                 case KEY_ESC:
3438                         status = CANCEL;
3439                         break;
3441                 case ERR:
3442                         break;
3444                 default:
3445                         if (pos >= sizeof(buf)) {
3446                                 report("Input string too long");
3447                                 return NULL;
3448                         }
3450                         if (isprint(key))
3451                                 buf[pos++] = (char) key;
3452                 }
3453         }
3455         if (status == CANCEL) {
3456                 /* Clear the status window */
3457                 report("");
3458                 return NULL;
3459         }
3461         buf[pos++] = 0;
3463         return buf;
3466 /*
3467  * Repository references
3468  */
3470 static struct ref *refs;
3471 static size_t refs_size;
3473 /* Id <-> ref store */
3474 static struct ref ***id_refs;
3475 static size_t id_refs_size;
3477 static struct ref **
3478 get_refs(char *id)
3480         struct ref ***tmp_id_refs;
3481         struct ref **ref_list = NULL;
3482         size_t ref_list_size = 0;
3483         size_t i;
3485         for (i = 0; i < id_refs_size; i++)
3486                 if (!strcmp(id, id_refs[i][0]->id))
3487                         return id_refs[i];
3489         tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
3490         if (!tmp_id_refs)
3491                 return NULL;
3493         id_refs = tmp_id_refs;
3495         for (i = 0; i < refs_size; i++) {
3496                 struct ref **tmp;
3498                 if (strcmp(id, refs[i].id))
3499                         continue;
3501                 tmp = realloc(ref_list, (ref_list_size + 1) * sizeof(*ref_list));
3502                 if (!tmp) {
3503                         if (ref_list)
3504                                 free(ref_list);
3505                         return NULL;
3506                 }
3508                 ref_list = tmp;
3509                 if (ref_list_size > 0)
3510                         ref_list[ref_list_size - 1]->next = 1;
3511                 ref_list[ref_list_size] = &refs[i];
3513                 /* XXX: The properties of the commit chains ensures that we can
3514                  * safely modify the shared ref. The repo references will
3515                  * always be similar for the same id. */
3516                 ref_list[ref_list_size]->next = 0;
3517                 ref_list_size++;
3518         }
3520         if (ref_list)
3521                 id_refs[id_refs_size++] = ref_list;
3523         return ref_list;
3526 static int
3527 read_ref(char *id, int idlen, char *name, int namelen)
3529         struct ref *ref;
3530         bool tag = FALSE;
3532         if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
3533                 /* Commits referenced by tags has "^{}" appended. */
3534                 if (name[namelen - 1] != '}')
3535                         return OK;
3537                 while (namelen > 0 && name[namelen] != '^')
3538                         namelen--;
3540                 tag = TRUE;
3541                 namelen -= STRING_SIZE("refs/tags/");
3542                 name    += STRING_SIZE("refs/tags/");
3544         } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
3545                 namelen -= STRING_SIZE("refs/heads/");
3546                 name    += STRING_SIZE("refs/heads/");
3548         } else if (!strcmp(name, "HEAD")) {
3549                 return OK;
3550         }
3552         refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
3553         if (!refs)
3554                 return ERR;
3556         ref = &refs[refs_size++];
3557         ref->name = malloc(namelen + 1);
3558         if (!ref->name)
3559                 return ERR;
3561         strncpy(ref->name, name, namelen);
3562         ref->name[namelen] = 0;
3563         ref->tag = tag;
3564         string_copy(ref->id, id);
3566         return OK;
3569 static int
3570 load_refs(void)
3572         const char *cmd_env = getenv("TIG_LS_REMOTE");
3573         const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
3575         return read_properties(popen(cmd, "r"), "\t", read_ref);
3578 static int
3579 read_repo_config_option(char *name, int namelen, char *value, int valuelen)
3581         if (!strcmp(name, "i18n.commitencoding"))
3582                 string_copy(opt_encoding, value);
3584         return OK;
3587 static int
3588 load_repo_config(void)
3590         return read_properties(popen("git repo-config --list", "r"),
3591                                "=", read_repo_config_option);
3594 static int
3595 read_properties(FILE *pipe, const char *separators,
3596                 int (*read_property)(char *, int, char *, int))
3598         char buffer[BUFSIZ];
3599         char *name;
3600         int state = OK;
3602         if (!pipe)
3603                 return ERR;
3605         while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
3606                 char *value;
3607                 size_t namelen;
3608                 size_t valuelen;
3610                 name = chomp_string(name);
3611                 namelen = strcspn(name, separators);
3613                 if (name[namelen]) {
3614                         name[namelen] = 0;
3615                         value = chomp_string(name + namelen + 1);
3616                         valuelen = strlen(value);
3618                 } else {
3619                         value = "";
3620                         valuelen = 0;
3621                 }
3623                 state = read_property(name, namelen, value, valuelen);
3624         }
3626         if (state != ERR && ferror(pipe))
3627                 state = ERR;
3629         pclose(pipe);
3631         return state;
3635 /*
3636  * Main
3637  */
3639 static void __NORETURN
3640 quit(int sig)
3642         /* XXX: Restore tty modes and let the OS cleanup the rest! */
3643         if (cursed)
3644                 endwin();
3645         exit(0);
3648 static void __NORETURN
3649 die(const char *err, ...)
3651         va_list args;
3653         endwin();
3655         va_start(args, err);
3656         fputs("tig: ", stderr);
3657         vfprintf(stderr, err, args);
3658         fputs("\n", stderr);
3659         va_end(args);
3661         exit(1);
3664 int
3665 main(int argc, char *argv[])
3667         struct view *view;
3668         enum request request;
3669         size_t i;
3671         signal(SIGINT, quit);
3673         if (setlocale(LC_ALL, "")) {
3674                 string_copy(opt_codeset, nl_langinfo(CODESET));
3675         }
3677         if (load_options() == ERR)
3678                 die("Failed to load user config.");
3680         /* Load the repo config file so options can be overwritten from
3681          * the command line.  */
3682         if (load_repo_config() == ERR)
3683                 die("Failed to load repo config.");
3685         if (!parse_options(argc, argv))
3686                 return 0;
3688         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
3689                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
3690                 if (opt_iconv == ICONV_NONE)
3691                         die("Failed to initialize character set conversion");
3692         }
3694         if (load_refs() == ERR)
3695                 die("Failed to load refs.");
3697         /* Require a git repository unless when running in pager mode. */
3698         if (refs_size == 0 && opt_request != REQ_VIEW_PAGER)
3699                 die("Not a git repository");
3701         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
3702                 view->cmd_env = getenv(view->cmd_env);
3704         request = opt_request;
3706         init_display();
3708         while (view_driver(display[current_view], request)) {
3709                 int key;
3710                 int i;
3712                 foreach_view (view, i)
3713                         update_view(view);
3715                 /* Refresh, accept single keystroke of input */
3716                 key = wgetch(status_win);
3718                 request = get_keybinding(display[current_view]->keymap, key);
3720                 /* Some low-level request handling. This keeps access to
3721                  * status_win restricted. */
3722                 switch (request) {
3723                 case REQ_PROMPT:
3724                 {
3725                         char *cmd = read_prompt(":");
3727                         if (cmd && string_format(opt_cmd, "git %s", cmd)) {
3728                                 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
3729                                         opt_request = REQ_VIEW_DIFF;
3730                                 } else {
3731                                         opt_request = REQ_VIEW_PAGER;
3732                                 }
3733                                 break;
3734                         }
3736                         request = REQ_NONE;
3737                         break;
3738                 }
3739                 case REQ_SEARCH:
3740                 case REQ_SEARCH_BACK:
3741                 {
3742                         const char *prompt = request == REQ_SEARCH
3743                                            ? "/" : "?";
3744                         char *search = read_prompt(prompt);
3746                         if (search)
3747                                 string_copy(opt_search, search);
3748                         else
3749                                 request = REQ_NONE;
3750                         break;
3751                 }
3752                 case REQ_SCREEN_RESIZE:
3753                 {
3754                         int height, width;
3756                         getmaxyx(stdscr, height, width);
3758                         /* Resize the status view and let the view driver take
3759                          * care of resizing the displayed views. */
3760                         wresize(status_win, 1, width);
3761                         mvwin(status_win, height - 1, 0);
3762                         wrefresh(status_win);
3763                         break;
3764                 }
3765                 default:
3766                         break;
3767                 }
3768         }
3770         quit(0);
3772         return 0;