Code

update_rev_graph: move code to prepare_rev_graph
[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[3];
2699 /* The current stack of revisions on the graph. */
2700 static struct rev_graph graph_stacks[3] = {
2701         { &graph_stacks[2], &graph_stacks[1], &graph_parents[0] },
2702         { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
2703         { &graph_stacks[1], &graph_stacks[0], &graph_parents[2] },
2704 };
2706 static inline bool
2707 graph_parent_is_merge(struct rev_graph *graph)
2709         return graph->parents->size > 1;
2712 static inline void
2713 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
2715         struct commit *commit = graph->commit;
2717         if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
2718                 commit->graph[commit->graph_size++] = symbol;
2721 static void
2722 done_rev_graph(struct rev_graph *graph)
2724         if (graph_parent_is_merge(graph) &&
2725             graph->pos < graph->size - 1 &&
2726             graph->next->size == graph->size + graph->parents->size - 1) {
2727                 size_t i = graph->pos + graph->parents->size - 1;
2729                 graph->commit->graph_size = i * 2;
2730                 while (i < graph->next->size - 1) {
2731                         append_to_rev_graph(graph, ' ');
2732                         append_to_rev_graph(graph, '\\');
2733                         i++;
2734                 }
2735         }
2737         graph->size = graph->pos = 0;
2738         graph->commit = NULL;
2739         memset(graph->parents, 0, sizeof(*graph->parents));
2742 static void
2743 push_rev_graph(struct rev_graph *graph, char *parent)
2745         /* Combine duplicate parents lines. */
2746         if (graph->size > 0 &&
2747             !strncmp(graph->rev[graph->size - 1], parent, SIZEOF_REV))
2748                 return;
2750         if (graph->size < SIZEOF_REVITEMS) {
2751                 string_ncopy(graph->rev[graph->size++], parent, SIZEOF_REV);
2752         }
2755 static chtype
2756 get_rev_graph_symbol(struct rev_graph *graph)
2758         chtype symbol;
2760         if (graph->parents->size == 0)
2761                 symbol = REVGRAPH_INIT;
2762         else if (graph_parent_is_merge(graph))
2763                 symbol = REVGRAPH_MERGE;
2764         else if (graph->pos >= graph->size)
2765                 symbol = REVGRAPH_BRANCH;
2766         else
2767                 symbol = REVGRAPH_COMMIT;
2769         return symbol;
2772 static void
2773 draw_rev_graph(struct rev_graph *graph)
2775         struct rev_filler {
2776                 chtype separator, line;
2777         };
2778         enum { DEFAULT, RSHARP, RDIAG, LDIAG };
2779         static struct rev_filler fillers[] = {
2780                 { ' ',  REVGRAPH_LINE },
2781                 { '`',  '.' },
2782                 { '\'', ' ' },
2783                 { '/',  ' ' },
2784         };
2785         chtype symbol = get_rev_graph_symbol(graph);
2786         struct rev_filler *filler;
2787         size_t i;
2789         filler = &fillers[DEFAULT];
2791         for (i = 0; i < graph->pos; i++) {
2792                 append_to_rev_graph(graph, filler->line);
2793                 if (graph_parent_is_merge(graph->prev) &&
2794                     graph->prev->pos == i)
2795                         filler = &fillers[RSHARP];
2797                 append_to_rev_graph(graph, filler->separator);
2798         }
2800         /* Place the symbol for this revision. */
2801         append_to_rev_graph(graph, symbol);
2803         if (graph->prev->size > graph->size)
2804                 filler = &fillers[RDIAG];
2805         else
2806                 filler = &fillers[DEFAULT];
2808         i++;
2810         for (; i < graph->size; i++) {
2811                 append_to_rev_graph(graph, filler->separator);
2812                 append_to_rev_graph(graph, filler->line);
2813                 if (graph_parent_is_merge(graph->prev) &&
2814                     i < graph->prev->pos + graph->parents->size)
2815                         filler = &fillers[RSHARP];
2816                 if (graph->prev->size > graph->size)
2817                         filler = &fillers[LDIAG];
2818         }
2820         if (graph->prev->size > graph->size) {
2821                 append_to_rev_graph(graph, filler->separator);
2822                 if (filler->line != ' ')
2823                         append_to_rev_graph(graph, filler->line);
2824         }
2827 /* Prepare the next rev graph */
2828 static void
2829 prepare_rev_graph(struct rev_graph *graph)
2831         size_t i;
2833         /* First, traverse all lines of revisions up to the active one. */
2834         for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
2835                 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
2836                         break;
2838                 push_rev_graph(graph->next, graph->rev[graph->pos]);
2839         }
2841         /* Interleave the new revision parent(s). */
2842         for (i = 0; i < graph->parents->size; i++)
2843                 push_rev_graph(graph->next, graph->parents->rev[i]);
2845         /* Lastly, put any remaining revisions. */
2846         for (i = graph->pos + 1; i < graph->size; i++)
2847                 push_rev_graph(graph->next, graph->rev[i]);
2850 static void
2851 update_rev_graph(struct rev_graph *graph)
2853         prepare_rev_graph(graph);
2855         draw_rev_graph(graph->prev);
2856         done_rev_graph(graph->prev->prev);
2860 /*
2861  * Main view backend
2862  */
2864 static bool
2865 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2867         char buf[DATE_COLS + 1];
2868         struct commit *commit = line->data;
2869         enum line_type type;
2870         int col = 0;
2871         size_t timelen;
2872         size_t authorlen;
2873         int trimmed = 1;
2875         if (!*commit->author)
2876                 return FALSE;
2878         wmove(view->win, lineno, col);
2880         if (selected) {
2881                 type = LINE_CURSOR;
2882                 wattrset(view->win, get_line_attr(type));
2883                 wchgat(view->win, -1, 0, type, NULL);
2885         } else {
2886                 type = LINE_MAIN_COMMIT;
2887                 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
2888         }
2890         timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
2891         waddnstr(view->win, buf, timelen);
2892         waddstr(view->win, " ");
2894         col += DATE_COLS;
2895         wmove(view->win, lineno, col);
2896         if (type != LINE_CURSOR)
2897                 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
2899         if (opt_utf8) {
2900                 authorlen = utf8_length(commit->author, AUTHOR_COLS - 2, &col, &trimmed);
2901         } else {
2902                 authorlen = strlen(commit->author);
2903                 if (authorlen > AUTHOR_COLS - 2) {
2904                         authorlen = AUTHOR_COLS - 2;
2905                         trimmed = 1;
2906                 }
2907         }
2909         if (trimmed) {
2910                 waddnstr(view->win, commit->author, authorlen);
2911                 if (type != LINE_CURSOR)
2912                         wattrset(view->win, get_line_attr(LINE_MAIN_DELIM));
2913                 waddch(view->win, '~');
2914         } else {
2915                 waddstr(view->win, commit->author);
2916         }
2918         col += AUTHOR_COLS;
2919         if (type != LINE_CURSOR)
2920                 wattrset(view->win, A_NORMAL);
2922         if (opt_rev_graph && commit->graph_size) {
2923                 size_t i;
2925                 wmove(view->win, lineno, col);
2926                 /* Using waddch() instead of waddnstr() ensures that
2927                  * they'll be rendered correctly for the cursor line. */
2928                 for (i = 0; i < commit->graph_size; i++)
2929                         waddch(view->win, commit->graph[i]);
2931                 waddch(view->win, ' ');
2932                 col += commit->graph_size + 1;
2933         }
2935         wmove(view->win, lineno, col);
2937         if (commit->refs) {
2938                 size_t i = 0;
2940                 do {
2941                         if (type == LINE_CURSOR)
2942                                 ;
2943                         else if (commit->refs[i]->tag)
2944                                 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
2945                         else
2946                                 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
2947                         waddstr(view->win, "[");
2948                         waddstr(view->win, commit->refs[i]->name);
2949                         waddstr(view->win, "]");
2950                         if (type != LINE_CURSOR)
2951                                 wattrset(view->win, A_NORMAL);
2952                         waddstr(view->win, " ");
2953                         col += strlen(commit->refs[i]->name) + STRING_SIZE("[] ");
2954                 } while (commit->refs[i++]->next);
2955         }
2957         if (type != LINE_CURSOR)
2958                 wattrset(view->win, get_line_attr(type));
2960         {
2961                 int titlelen = strlen(commit->title);
2963                 if (col + titlelen > view->width)
2964                         titlelen = view->width - col;
2966                 waddnstr(view->win, commit->title, titlelen);
2967         }
2969         return TRUE;
2972 /* Reads git log --pretty=raw output and parses it into the commit struct. */
2973 static bool
2974 main_read(struct view *view, char *line)
2976         static struct rev_graph *graph = graph_stacks;
2977         enum line_type type;
2978         struct commit *commit = view->lines
2979                               ? view->line[view->lines - 1].data : NULL;
2981         if (!line) {
2982                 return TRUE;
2983         }
2985         type = get_line_type(line);
2987         switch (type) {
2988         case LINE_COMMIT:
2989                 commit = calloc(1, sizeof(struct commit));
2990                 if (!commit)
2991                         return FALSE;
2993                 line += STRING_SIZE("commit ");
2995                 view->line[view->lines++].data = commit;
2996                 string_copy(commit->id, line);
2997                 commit->refs = get_refs(commit->id);
2998                 graph->commit = commit;
2999                 break;
3001         case LINE_PARENT:
3002                 if (commit) {
3003                         line += STRING_SIZE("parent ");
3004                         push_rev_graph(graph->parents, line);
3005                 }
3006                 break;
3008         case LINE_AUTHOR:
3009         {
3010                 char *ident = line + STRING_SIZE("author ");
3011                 char *end = strchr(ident, '<');
3013                 if (!commit)
3014                         break;
3016                 update_rev_graph(graph);
3017                 graph = graph->next;
3019                 if (end) {
3020                         char *email = end + 1;
3022                         for (; end > ident && isspace(end[-1]); end--) ;
3024                         if (end == ident && *email) {
3025                                 ident = email;
3026                                 end = strchr(ident, '>');
3027                                 for (; end > ident && isspace(end[-1]); end--) ;
3028                         }
3029                         *end = 0;
3030                 }
3032                 /* End is NULL or ident meaning there's no author. */
3033                 if (end <= ident)
3034                         ident = "Unknown";
3036                 string_copy(commit->author, ident);
3038                 /* Parse epoch and timezone */
3039                 if (end) {
3040                         char *secs = strchr(end + 1, '>');
3041                         char *zone;
3042                         time_t time;
3044                         if (!secs || secs[1] != ' ')
3045                                 break;
3047                         secs += 2;
3048                         time = (time_t) atol(secs);
3049                         zone = strchr(secs, ' ');
3050                         if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
3051                                 long tz;
3053                                 zone++;
3054                                 tz  = ('0' - zone[1]) * 60 * 60 * 10;
3055                                 tz += ('0' - zone[2]) * 60 * 60;
3056                                 tz += ('0' - zone[3]) * 60;
3057                                 tz += ('0' - zone[4]) * 60;
3059                                 if (zone[0] == '-')
3060                                         tz = -tz;
3062                                 time -= tz;
3063                         }
3064                         gmtime_r(&time, &commit->time);
3065                 }
3066                 break;
3067         }
3068         default:
3069                 if (!commit)
3070                         break;
3072                 /* Fill in the commit title if it has not already been set. */
3073                 if (commit->title[0])
3074                         break;
3076                 /* Require titles to start with a non-space character at the
3077                  * offset used by git log. */
3078                 /* FIXME: More gracefull handling of titles; append "..." to
3079                  * shortened titles, etc. */
3080                 if (strncmp(line, "    ", 4) ||
3081                     isspace(line[4]))
3082                         break;
3084                 string_copy(commit->title, line + 4);
3085         }
3087         return TRUE;
3090 static bool
3091 main_enter(struct view *view, struct line *line)
3093         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3095         open_view(view, REQ_VIEW_DIFF, flags);
3096         return TRUE;
3099 static bool
3100 main_grep(struct view *view, struct line *line)
3102         struct commit *commit = line->data;
3103         enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
3104         char buf[DATE_COLS + 1];
3105         regmatch_t pmatch;
3107         for (state = S_TITLE; state < S_END; state++) {
3108                 char *text;
3110                 switch (state) {
3111                 case S_TITLE:   text = commit->title;   break;
3112                 case S_AUTHOR:  text = commit->author;  break;
3113                 case S_DATE:
3114                         if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
3115                                 continue;
3116                         text = buf;
3117                         break;
3119                 default:
3120                         return FALSE;
3121                 }
3123                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3124                         return TRUE;
3125         }
3127         return FALSE;
3130 static void
3131 main_select(struct view *view, struct line *line)
3133         struct commit *commit = line->data;
3135         string_copy(view->ref, commit->id);
3136         string_copy(ref_commit, view->ref);
3139 static struct view_ops main_ops = {
3140         "commit",
3141         main_draw,
3142         main_read,
3143         main_enter,
3144         main_grep,
3145         main_select,
3146 };
3149 /*
3150  * Unicode / UTF-8 handling
3151  *
3152  * NOTE: Much of the following code for dealing with unicode is derived from
3153  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
3154  * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
3155  */
3157 /* I've (over)annotated a lot of code snippets because I am not entirely
3158  * confident that the approach taken by this small UTF-8 interface is correct.
3159  * --jonas */
3161 static inline int
3162 unicode_width(unsigned long c)
3164         if (c >= 0x1100 &&
3165            (c <= 0x115f                         /* Hangul Jamo */
3166             || c == 0x2329
3167             || c == 0x232a
3168             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
3169                                                 /* CJK ... Yi */
3170             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
3171             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
3172             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
3173             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
3174             || (c >= 0xffe0  && c <= 0xffe6)
3175             || (c >= 0x20000 && c <= 0x2fffd)
3176             || (c >= 0x30000 && c <= 0x3fffd)))
3177                 return 2;
3179         return 1;
3182 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
3183  * Illegal bytes are set one. */
3184 static const unsigned char utf8_bytes[256] = {
3185         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,
3186         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,
3187         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,
3188         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,
3189         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,
3190         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,
3191         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,
3192         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,
3193 };
3195 /* Decode UTF-8 multi-byte representation into a unicode character. */
3196 static inline unsigned long
3197 utf8_to_unicode(const char *string, size_t length)
3199         unsigned long unicode;
3201         switch (length) {
3202         case 1:
3203                 unicode  =   string[0];
3204                 break;
3205         case 2:
3206                 unicode  =  (string[0] & 0x1f) << 6;
3207                 unicode +=  (string[1] & 0x3f);
3208                 break;
3209         case 3:
3210                 unicode  =  (string[0] & 0x0f) << 12;
3211                 unicode += ((string[1] & 0x3f) << 6);
3212                 unicode +=  (string[2] & 0x3f);
3213                 break;
3214         case 4:
3215                 unicode  =  (string[0] & 0x0f) << 18;
3216                 unicode += ((string[1] & 0x3f) << 12);
3217                 unicode += ((string[2] & 0x3f) << 6);
3218                 unicode +=  (string[3] & 0x3f);
3219                 break;
3220         case 5:
3221                 unicode  =  (string[0] & 0x0f) << 24;
3222                 unicode += ((string[1] & 0x3f) << 18);
3223                 unicode += ((string[2] & 0x3f) << 12);
3224                 unicode += ((string[3] & 0x3f) << 6);
3225                 unicode +=  (string[4] & 0x3f);
3226                 break;
3227         case 6:
3228                 unicode  =  (string[0] & 0x01) << 30;
3229                 unicode += ((string[1] & 0x3f) << 24);
3230                 unicode += ((string[2] & 0x3f) << 18);
3231                 unicode += ((string[3] & 0x3f) << 12);
3232                 unicode += ((string[4] & 0x3f) << 6);
3233                 unicode +=  (string[5] & 0x3f);
3234                 break;
3235         default:
3236                 die("Invalid unicode length");
3237         }
3239         /* Invalid characters could return the special 0xfffd value but NUL
3240          * should be just as good. */
3241         return unicode > 0xffff ? 0 : unicode;
3244 /* Calculates how much of string can be shown within the given maximum width
3245  * and sets trimmed parameter to non-zero value if all of string could not be
3246  * shown.
3247  *
3248  * Additionally, adds to coloffset how many many columns to move to align with
3249  * the expected position. Takes into account how multi-byte and double-width
3250  * characters will effect the cursor position.
3251  *
3252  * Returns the number of bytes to output from string to satisfy max_width. */
3253 static size_t
3254 utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed)
3256         const char *start = string;
3257         const char *end = strchr(string, '\0');
3258         size_t mbwidth = 0;
3259         size_t width = 0;
3261         *trimmed = 0;
3263         while (string < end) {
3264                 int c = *(unsigned char *) string;
3265                 unsigned char bytes = utf8_bytes[c];
3266                 size_t ucwidth;
3267                 unsigned long unicode;
3269                 if (string + bytes > end)
3270                         break;
3272                 /* Change representation to figure out whether
3273                  * it is a single- or double-width character. */
3275                 unicode = utf8_to_unicode(string, bytes);
3276                 /* FIXME: Graceful handling of invalid unicode character. */
3277                 if (!unicode)
3278                         break;
3280                 ucwidth = unicode_width(unicode);
3281                 width  += ucwidth;
3282                 if (width > max_width) {
3283                         *trimmed = 1;
3284                         break;
3285                 }
3287                 /* The column offset collects the differences between the
3288                  * number of bytes encoding a character and the number of
3289                  * columns will be used for rendering said character.
3290                  *
3291                  * So if some character A is encoded in 2 bytes, but will be
3292                  * represented on the screen using only 1 byte this will and up
3293                  * adding 1 to the multi-byte column offset.
3294                  *
3295                  * Assumes that no double-width character can be encoding in
3296                  * less than two bytes. */
3297                 if (bytes > ucwidth)
3298                         mbwidth += bytes - ucwidth;
3300                 string  += bytes;
3301         }
3303         *coloffset += mbwidth;
3305         return string - start;
3309 /*
3310  * Status management
3311  */
3313 /* Whether or not the curses interface has been initialized. */
3314 static bool cursed = FALSE;
3316 /* The status window is used for polling keystrokes. */
3317 static WINDOW *status_win;
3319 /* Update status and title window. */
3320 static void
3321 report(const char *msg, ...)
3323         static bool empty = TRUE;
3324         struct view *view = display[current_view];
3326         if (!empty || *msg) {
3327                 va_list args;
3329                 va_start(args, msg);
3331                 werase(status_win);
3332                 wmove(status_win, 0, 0);
3333                 if (*msg) {
3334                         vwprintw(status_win, msg, args);
3335                         empty = FALSE;
3336                 } else {
3337                         empty = TRUE;
3338                 }
3339                 wrefresh(status_win);
3341                 va_end(args);
3342         }
3344         update_view_title(view);
3345         update_display_cursor();
3348 /* Controls when nodelay should be in effect when polling user input. */
3349 static void
3350 set_nonblocking_input(bool loading)
3352         static unsigned int loading_views;
3354         if ((loading == FALSE && loading_views-- == 1) ||
3355             (loading == TRUE  && loading_views++ == 0))
3356                 nodelay(status_win, loading);
3359 static void
3360 init_display(void)
3362         int x, y;
3364         /* Initialize the curses library */
3365         if (isatty(STDIN_FILENO)) {
3366                 cursed = !!initscr();
3367         } else {
3368                 /* Leave stdin and stdout alone when acting as a pager. */
3369                 FILE *io = fopen("/dev/tty", "r+");
3371                 if (!io)
3372                         die("Failed to open /dev/tty");
3373                 cursed = !!newterm(NULL, io, io);
3374         }
3376         if (!cursed)
3377                 die("Failed to initialize curses");
3379         nonl();         /* Tell curses not to do NL->CR/NL on output */
3380         cbreak();       /* Take input chars one at a time, no wait for \n */
3381         noecho();       /* Don't echo input */
3382         leaveok(stdscr, TRUE);
3384         if (has_colors())
3385                 init_colors();
3387         getmaxyx(stdscr, y, x);
3388         status_win = newwin(1, 0, y - 1, 0);
3389         if (!status_win)
3390                 die("Failed to create status window");
3392         /* Enable keyboard mapping */
3393         keypad(status_win, TRUE);
3394         wbkgdset(status_win, get_line_attr(LINE_STATUS));
3397 static char *
3398 read_prompt(const char *prompt)
3400         enum { READING, STOP, CANCEL } status = READING;
3401         static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
3402         int pos = 0;
3404         while (status == READING) {
3405                 struct view *view;
3406                 int i, key;
3408                 foreach_view (view, i)
3409                         update_view(view);
3411                 report("%s%.*s", prompt, pos, buf);
3412                 /* Refresh, accept single keystroke of input */
3413                 key = wgetch(status_win);
3414                 switch (key) {
3415                 case KEY_RETURN:
3416                 case KEY_ENTER:
3417                 case '\n':
3418                         status = pos ? STOP : CANCEL;
3419                         break;
3421                 case KEY_BACKSPACE:
3422                         if (pos > 0)
3423                                 pos--;
3424                         else
3425                                 status = CANCEL;
3426                         break;
3428                 case KEY_ESC:
3429                         status = CANCEL;
3430                         break;
3432                 case ERR:
3433                         break;
3435                 default:
3436                         if (pos >= sizeof(buf)) {
3437                                 report("Input string too long");
3438                                 return NULL;
3439                         }
3441                         if (isprint(key))
3442                                 buf[pos++] = (char) key;
3443                 }
3444         }
3446         if (status == CANCEL) {
3447                 /* Clear the status window */
3448                 report("");
3449                 return NULL;
3450         }
3452         buf[pos++] = 0;
3454         return buf;
3457 /*
3458  * Repository references
3459  */
3461 static struct ref *refs;
3462 static size_t refs_size;
3464 /* Id <-> ref store */
3465 static struct ref ***id_refs;
3466 static size_t id_refs_size;
3468 static struct ref **
3469 get_refs(char *id)
3471         struct ref ***tmp_id_refs;
3472         struct ref **ref_list = NULL;
3473         size_t ref_list_size = 0;
3474         size_t i;
3476         for (i = 0; i < id_refs_size; i++)
3477                 if (!strcmp(id, id_refs[i][0]->id))
3478                         return id_refs[i];
3480         tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
3481         if (!tmp_id_refs)
3482                 return NULL;
3484         id_refs = tmp_id_refs;
3486         for (i = 0; i < refs_size; i++) {
3487                 struct ref **tmp;
3489                 if (strcmp(id, refs[i].id))
3490                         continue;
3492                 tmp = realloc(ref_list, (ref_list_size + 1) * sizeof(*ref_list));
3493                 if (!tmp) {
3494                         if (ref_list)
3495                                 free(ref_list);
3496                         return NULL;
3497                 }
3499                 ref_list = tmp;
3500                 if (ref_list_size > 0)
3501                         ref_list[ref_list_size - 1]->next = 1;
3502                 ref_list[ref_list_size] = &refs[i];
3504                 /* XXX: The properties of the commit chains ensures that we can
3505                  * safely modify the shared ref. The repo references will
3506                  * always be similar for the same id. */
3507                 ref_list[ref_list_size]->next = 0;
3508                 ref_list_size++;
3509         }
3511         if (ref_list)
3512                 id_refs[id_refs_size++] = ref_list;
3514         return ref_list;
3517 static int
3518 read_ref(char *id, int idlen, char *name, int namelen)
3520         struct ref *ref;
3521         bool tag = FALSE;
3523         if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
3524                 /* Commits referenced by tags has "^{}" appended. */
3525                 if (name[namelen - 1] != '}')
3526                         return OK;
3528                 while (namelen > 0 && name[namelen] != '^')
3529                         namelen--;
3531                 tag = TRUE;
3532                 namelen -= STRING_SIZE("refs/tags/");
3533                 name    += STRING_SIZE("refs/tags/");
3535         } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
3536                 namelen -= STRING_SIZE("refs/heads/");
3537                 name    += STRING_SIZE("refs/heads/");
3539         } else if (!strcmp(name, "HEAD")) {
3540                 return OK;
3541         }
3543         refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
3544         if (!refs)
3545                 return ERR;
3547         ref = &refs[refs_size++];
3548         ref->name = malloc(namelen + 1);
3549         if (!ref->name)
3550                 return ERR;
3552         strncpy(ref->name, name, namelen);
3553         ref->name[namelen] = 0;
3554         ref->tag = tag;
3555         string_copy(ref->id, id);
3557         return OK;
3560 static int
3561 load_refs(void)
3563         const char *cmd_env = getenv("TIG_LS_REMOTE");
3564         const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
3566         return read_properties(popen(cmd, "r"), "\t", read_ref);
3569 static int
3570 read_repo_config_option(char *name, int namelen, char *value, int valuelen)
3572         if (!strcmp(name, "i18n.commitencoding"))
3573                 string_copy(opt_encoding, value);
3575         return OK;
3578 static int
3579 load_repo_config(void)
3581         return read_properties(popen("git repo-config --list", "r"),
3582                                "=", read_repo_config_option);
3585 static int
3586 read_properties(FILE *pipe, const char *separators,
3587                 int (*read_property)(char *, int, char *, int))
3589         char buffer[BUFSIZ];
3590         char *name;
3591         int state = OK;
3593         if (!pipe)
3594                 return ERR;
3596         while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
3597                 char *value;
3598                 size_t namelen;
3599                 size_t valuelen;
3601                 name = chomp_string(name);
3602                 namelen = strcspn(name, separators);
3604                 if (name[namelen]) {
3605                         name[namelen] = 0;
3606                         value = chomp_string(name + namelen + 1);
3607                         valuelen = strlen(value);
3609                 } else {
3610                         value = "";
3611                         valuelen = 0;
3612                 }
3614                 state = read_property(name, namelen, value, valuelen);
3615         }
3617         if (state != ERR && ferror(pipe))
3618                 state = ERR;
3620         pclose(pipe);
3622         return state;
3626 /*
3627  * Main
3628  */
3630 static void __NORETURN
3631 quit(int sig)
3633         /* XXX: Restore tty modes and let the OS cleanup the rest! */
3634         if (cursed)
3635                 endwin();
3636         exit(0);
3639 static void __NORETURN
3640 die(const char *err, ...)
3642         va_list args;
3644         endwin();
3646         va_start(args, err);
3647         fputs("tig: ", stderr);
3648         vfprintf(stderr, err, args);
3649         fputs("\n", stderr);
3650         va_end(args);
3652         exit(1);
3655 int
3656 main(int argc, char *argv[])
3658         struct view *view;
3659         enum request request;
3660         size_t i;
3662         signal(SIGINT, quit);
3664         if (setlocale(LC_ALL, "")) {
3665                 string_copy(opt_codeset, nl_langinfo(CODESET));
3666         }
3668         if (load_options() == ERR)
3669                 die("Failed to load user config.");
3671         /* Load the repo config file so options can be overwritten from
3672          * the command line.  */
3673         if (load_repo_config() == ERR)
3674                 die("Failed to load repo config.");
3676         if (!parse_options(argc, argv))
3677                 return 0;
3679         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
3680                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
3681                 if (opt_iconv == ICONV_NONE)
3682                         die("Failed to initialize character set conversion");
3683         }
3685         if (load_refs() == ERR)
3686                 die("Failed to load refs.");
3688         /* Require a git repository unless when running in pager mode. */
3689         if (refs_size == 0 && opt_request != REQ_VIEW_PAGER)
3690                 die("Not a git repository");
3692         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
3693                 view->cmd_env = getenv(view->cmd_env);
3695         request = opt_request;
3697         init_display();
3699         while (view_driver(display[current_view], request)) {
3700                 int key;
3701                 int i;
3703                 foreach_view (view, i)
3704                         update_view(view);
3706                 /* Refresh, accept single keystroke of input */
3707                 key = wgetch(status_win);
3709                 request = get_keybinding(display[current_view]->keymap, key);
3711                 /* Some low-level request handling. This keeps access to
3712                  * status_win restricted. */
3713                 switch (request) {
3714                 case REQ_PROMPT:
3715                 {
3716                         char *cmd = read_prompt(":");
3718                         if (cmd && string_format(opt_cmd, "git %s", cmd)) {
3719                                 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
3720                                         opt_request = REQ_VIEW_DIFF;
3721                                 } else {
3722                                         opt_request = REQ_VIEW_PAGER;
3723                                 }
3724                                 break;
3725                         }
3727                         request = REQ_NONE;
3728                         break;
3729                 }
3730                 case REQ_SEARCH:
3731                 case REQ_SEARCH_BACK:
3732                 {
3733                         const char *prompt = request == REQ_SEARCH
3734                                            ? "/" : "?";
3735                         char *search = read_prompt(prompt);
3737                         if (search)
3738                                 string_copy(opt_search, search);
3739                         else
3740                                 request = REQ_NONE;
3741                         break;
3742                 }
3743                 case REQ_SCREEN_RESIZE:
3744                 {
3745                         int height, width;
3747                         getmaxyx(stdscr, height, width);
3749                         /* Resize the status view and let the view driver take
3750                          * care of resizing the displayed views. */
3751                         wresize(status_win, 1, width);
3752                         mvwin(status_win, height - 1, 0);
3753                         wrefresh(status_win);
3754                         break;
3755                 }
3756                 default:
3757                         break;
3758                 }
3759         }
3761         quit(0);
3763         return 0;