Code

update_view_title: format load seconds as part of the state string
[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 */
63 #define SIZEOF_REVGRAPH 19      /* Size of revision ancestry graphics. */
65 /* This color name can be used to refer to the default term colors. */
66 #define COLOR_DEFAULT   (-1)
68 #define ICONV_NONE      ((iconv_t) -1)
70 /* The format and size of the date column in the main view. */
71 #define DATE_FORMAT     "%Y-%m-%d %H:%M"
72 #define DATE_COLS       STRING_SIZE("2006-04-29 14:21 ")
74 #define AUTHOR_COLS     20
76 /* The default interval between line numbers. */
77 #define NUMBER_INTERVAL 1
79 #define TABSIZE         8
81 #define SCALE_SPLIT_VIEW(height)        ((height) * 2 / 3)
83 #define TIG_LS_REMOTE \
84         "git ls-remote . 2>/dev/null"
86 #define TIG_DIFF_CMD \
87         "git show --root --patch-with-stat --find-copies-harder -B -C %s 2>/dev/null"
89 #define TIG_LOG_CMD     \
90         "git log --cc --stat -n100 %s 2>/dev/null"
92 #define TIG_MAIN_CMD \
93         "git log --topo-order --pretty=raw %s 2>/dev/null"
95 #define TIG_TREE_CMD    \
96         "git ls-tree %s %s"
98 #define TIG_BLOB_CMD    \
99         "git cat-file blob %s"
101 /* XXX: Needs to be defined to the empty string. */
102 #define TIG_HELP_CMD    ""
103 #define TIG_PAGER_CMD   ""
105 /* Some ascii-shorthands fitted into the ncurses namespace. */
106 #define KEY_TAB         '\t'
107 #define KEY_RETURN      '\r'
108 #define KEY_ESC         27
111 struct ref {
112         char *name;             /* Ref name; tag or head names are shortened. */
113         char id[SIZEOF_REV];    /* Commit SHA1 ID */
114         unsigned int tag:1;     /* Is it a tag? */
115         unsigned int next:1;    /* For ref lists: are there more refs? */
116 };
118 static struct ref **get_refs(char *id);
120 struct int_map {
121         const char *name;
122         int namelen;
123         int value;
124 };
126 static int
127 set_from_int_map(struct int_map *map, size_t map_size,
128                  int *value, const char *name, int namelen)
131         int i;
133         for (i = 0; i < map_size; i++)
134                 if (namelen == map[i].namelen &&
135                     !strncasecmp(name, map[i].name, namelen)) {
136                         *value = map[i].value;
137                         return OK;
138                 }
140         return ERR;
144 /*
145  * String helpers
146  */
148 static inline void
149 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
151         if (srclen > dstlen - 1)
152                 srclen = dstlen - 1;
154         strncpy(dst, src, srclen);
155         dst[srclen] = 0;
158 /* Shorthands for safely copying into a fixed buffer. */
160 #define string_copy(dst, src) \
161         string_ncopy_do(dst, sizeof(dst), src, sizeof(dst))
163 #define string_ncopy(dst, src, srclen) \
164         string_ncopy_do(dst, sizeof(dst), src, srclen)
166 static char *
167 chomp_string(char *name)
169         int namelen;
171         while (isspace(*name))
172                 name++;
174         namelen = strlen(name) - 1;
175         while (namelen > 0 && isspace(name[namelen]))
176                 name[namelen--] = 0;
178         return name;
181 static bool
182 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
184         va_list args;
185         size_t pos = bufpos ? *bufpos : 0;
187         va_start(args, fmt);
188         pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
189         va_end(args);
191         if (bufpos)
192                 *bufpos = pos;
194         return pos >= bufsize ? FALSE : TRUE;
197 #define string_format(buf, fmt, args...) \
198         string_nformat(buf, sizeof(buf), NULL, fmt, args)
200 #define string_format_from(buf, from, fmt, args...) \
201         string_nformat(buf, sizeof(buf), from, fmt, args)
203 static int
204 string_enum_compare(const char *str1, const char *str2, int len)
206         size_t i;
208 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
210         /* Diff-Header == DIFF_HEADER */
211         for (i = 0; i < len; i++) {
212                 if (toupper(str1[i]) == toupper(str2[i]))
213                         continue;
215                 if (string_enum_sep(str1[i]) &&
216                     string_enum_sep(str2[i]))
217                         continue;
219                 return str1[i] - str2[i];
220         }
222         return 0;
225 /* Shell quoting
226  *
227  * NOTE: The following is a slightly modified copy of the git project's shell
228  * quoting routines found in the quote.c file.
229  *
230  * Help to copy the thing properly quoted for the shell safety.  any single
231  * quote is replaced with '\'', any exclamation point is replaced with '\!',
232  * and the whole thing is enclosed in a
233  *
234  * E.g.
235  *  original     sq_quote     result
236  *  name     ==> name      ==> 'name'
237  *  a b      ==> a b       ==> 'a b'
238  *  a'b      ==> a'\''b    ==> 'a'\''b'
239  *  a!b      ==> a'\!'b    ==> 'a'\!'b'
240  */
242 static size_t
243 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
245         char c;
247 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
249         BUFPUT('\'');
250         while ((c = *src++)) {
251                 if (c == '\'' || c == '!') {
252                         BUFPUT('\'');
253                         BUFPUT('\\');
254                         BUFPUT(c);
255                         BUFPUT('\'');
256                 } else {
257                         BUFPUT(c);
258                 }
259         }
260         BUFPUT('\'');
262         return bufsize;
266 /*
267  * User requests
268  */
270 #define REQ_INFO \
271         /* XXX: Keep the view request first and in sync with views[]. */ \
272         REQ_GROUP("View switching") \
273         REQ_(VIEW_MAIN,         "Show main view"), \
274         REQ_(VIEW_DIFF,         "Show diff view"), \
275         REQ_(VIEW_LOG,          "Show log view"), \
276         REQ_(VIEW_TREE,         "Show tree view"), \
277         REQ_(VIEW_BLOB,         "Show blob view"), \
278         REQ_(VIEW_HELP,         "Show help page"), \
279         REQ_(VIEW_PAGER,        "Show pager view"), \
280         \
281         REQ_GROUP("View manipulation") \
282         REQ_(ENTER,             "Enter current line and scroll"), \
283         REQ_(NEXT,              "Move to next"), \
284         REQ_(PREVIOUS,          "Move to previous"), \
285         REQ_(VIEW_NEXT,         "Move focus to next view"), \
286         REQ_(VIEW_CLOSE,        "Close the current view"), \
287         REQ_(QUIT,              "Close all views and quit"), \
288         \
289         REQ_GROUP("Cursor navigation") \
290         REQ_(MOVE_UP,           "Move cursor one line up"), \
291         REQ_(MOVE_DOWN,         "Move cursor one line down"), \
292         REQ_(MOVE_PAGE_DOWN,    "Move cursor one page down"), \
293         REQ_(MOVE_PAGE_UP,      "Move cursor one page up"), \
294         REQ_(MOVE_FIRST_LINE,   "Move cursor to first line"), \
295         REQ_(MOVE_LAST_LINE,    "Move cursor to last line"), \
296         \
297         REQ_GROUP("Scrolling") \
298         REQ_(SCROLL_LINE_UP,    "Scroll one line up"), \
299         REQ_(SCROLL_LINE_DOWN,  "Scroll one line down"), \
300         REQ_(SCROLL_PAGE_UP,    "Scroll one page up"), \
301         REQ_(SCROLL_PAGE_DOWN,  "Scroll one page down"), \
302         \
303         REQ_GROUP("Searching") \
304         REQ_(SEARCH,            "Search the view"), \
305         REQ_(SEARCH_BACK,       "Search backwards in the view"), \
306         REQ_(FIND_NEXT,         "Find next search match"), \
307         REQ_(FIND_PREV,         "Find previous search match"), \
308         \
309         REQ_GROUP("Misc") \
310         REQ_(NONE,              "Do nothing"), \
311         REQ_(PROMPT,            "Bring up the prompt"), \
312         REQ_(SCREEN_REDRAW,     "Redraw the screen"), \
313         REQ_(SCREEN_RESIZE,     "Resize the screen"), \
314         REQ_(SHOW_VERSION,      "Show version information"), \
315         REQ_(STOP_LOADING,      "Stop all loading views"), \
316         REQ_(TOGGLE_LINENO,     "Toggle line numbers"), \
317         REQ_(TOGGLE_REV_GRAPH,  "Toggle revision graph visualization")
320 /* User action requests. */
321 enum request {
322 #define REQ_GROUP(help)
323 #define REQ_(req, help) REQ_##req
325         /* Offset all requests to avoid conflicts with ncurses getch values. */
326         REQ_OFFSET = KEY_MAX + 1,
327         REQ_INFO,
328         REQ_UNKNOWN,
330 #undef  REQ_GROUP
331 #undef  REQ_
332 };
334 struct request_info {
335         enum request request;
336         char *name;
337         int namelen;
338         char *help;
339 };
341 static struct request_info req_info[] = {
342 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
343 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
344         REQ_INFO
345 #undef  REQ_GROUP
346 #undef  REQ_
347 };
349 static enum request
350 get_request(const char *name)
352         int namelen = strlen(name);
353         int i;
355         for (i = 0; i < ARRAY_SIZE(req_info); i++)
356                 if (req_info[i].namelen == namelen &&
357                     !string_enum_compare(req_info[i].name, name, namelen))
358                         return req_info[i].request;
360         return REQ_UNKNOWN;
364 /*
365  * Options
366  */
368 static const char usage[] =
369 VERSION " (" __DATE__ ")\n"
370 "\n"
371 "Usage: tig [options]\n"
372 "   or: tig [options] [--] [git log options]\n"
373 "   or: tig [options] log  [git log options]\n"
374 "   or: tig [options] diff [git diff options]\n"
375 "   or: tig [options] show [git show options]\n"
376 "   or: tig [options] <    [git command output]\n"
377 "\n"
378 "Options:\n"
379 "  -l                          Start up in log view\n"
380 "  -d                          Start up in diff view\n"
381 "  -n[I], --line-number[=I]    Show line numbers with given interval\n"
382 "  -b[N], --tab-size[=N]       Set number of spaces for tab expansion\n"
383 "  --                          Mark end of tig options\n"
384 "  -v, --version               Show version and exit\n"
385 "  -h, --help                  Show help message and exit\n";
387 /* Option and state variables. */
388 static bool opt_line_number             = FALSE;
389 static bool opt_rev_graph               = TRUE;
390 static int opt_num_interval             = NUMBER_INTERVAL;
391 static int opt_tab_size                 = TABSIZE;
392 static enum request opt_request         = REQ_VIEW_MAIN;
393 static char opt_cmd[SIZEOF_STR]         = "";
394 static char opt_path[SIZEOF_STR]        = "";
395 static FILE *opt_pipe                   = NULL;
396 static char opt_encoding[20]            = "UTF-8";
397 static bool opt_utf8                    = TRUE;
398 static char opt_codeset[20]             = "UTF-8";
399 static iconv_t opt_iconv                = ICONV_NONE;
400 static char opt_search[SIZEOF_STR]      = "";
402 enum option_type {
403         OPT_NONE,
404         OPT_INT,
405 };
407 static bool
408 check_option(char *opt, char short_name, char *name, enum option_type type, ...)
410         va_list args;
411         char *value = "";
412         int *number;
414         if (opt[0] != '-')
415                 return FALSE;
417         if (opt[1] == '-') {
418                 int namelen = strlen(name);
420                 opt += 2;
422                 if (strncmp(opt, name, namelen))
423                         return FALSE;
425                 if (opt[namelen] == '=')
426                         value = opt + namelen + 1;
428         } else {
429                 if (!short_name || opt[1] != short_name)
430                         return FALSE;
431                 value = opt + 2;
432         }
434         va_start(args, type);
435         if (type == OPT_INT) {
436                 number = va_arg(args, int *);
437                 if (isdigit(*value))
438                         *number = atoi(value);
439         }
440         va_end(args);
442         return TRUE;
445 /* Returns the index of log or diff command or -1 to exit. */
446 static bool
447 parse_options(int argc, char *argv[])
449         int i;
451         for (i = 1; i < argc; i++) {
452                 char *opt = argv[i];
454                 if (!strcmp(opt, "-l")) {
455                         opt_request = REQ_VIEW_LOG;
456                         continue;
457                 }
459                 if (!strcmp(opt, "-d")) {
460                         opt_request = REQ_VIEW_DIFF;
461                         continue;
462                 }
464                 if (check_option(opt, 'n', "line-number", OPT_INT, &opt_num_interval)) {
465                         opt_line_number = TRUE;
466                         continue;
467                 }
469                 if (check_option(opt, 'b', "tab-size", OPT_INT, &opt_tab_size)) {
470                         opt_tab_size = MIN(opt_tab_size, TABSIZE);
471                         continue;
472                 }
474                 if (check_option(opt, 'v', "version", OPT_NONE)) {
475                         printf("tig version %s\n", VERSION);
476                         return FALSE;
477                 }
479                 if (check_option(opt, 'h', "help", OPT_NONE)) {
480                         printf(usage);
481                         return FALSE;
482                 }
484                 if (!strcmp(opt, "--")) {
485                         i++;
486                         break;
487                 }
489                 if (!strcmp(opt, "log") ||
490                     !strcmp(opt, "diff") ||
491                     !strcmp(opt, "show")) {
492                         opt_request = opt[0] == 'l'
493                                     ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
494                         break;
495                 }
497                 if (opt[0] && opt[0] != '-')
498                         break;
500                 die("unknown option '%s'\n\n%s", opt, usage);
501         }
503         if (!isatty(STDIN_FILENO)) {
504                 opt_request = REQ_VIEW_PAGER;
505                 opt_pipe = stdin;
507         } else if (i < argc) {
508                 size_t buf_size;
510                 if (opt_request == REQ_VIEW_MAIN)
511                         /* XXX: This is vulnerable to the user overriding
512                          * options required for the main view parser. */
513                         string_copy(opt_cmd, "git log --stat --pretty=raw");
514                 else
515                         string_copy(opt_cmd, "git");
516                 buf_size = strlen(opt_cmd);
518                 while (buf_size < sizeof(opt_cmd) && i < argc) {
519                         opt_cmd[buf_size++] = ' ';
520                         buf_size = sq_quote(opt_cmd, buf_size, argv[i++]);
521                 }
523                 if (buf_size >= sizeof(opt_cmd))
524                         die("command too long");
526                 opt_cmd[buf_size] = 0;
528         }
530         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
531                 opt_utf8 = FALSE;
533         return TRUE;
537 /*
538  * Line-oriented content detection.
539  */
541 #define LINE_INFO \
542 LINE(DIFF_HEADER,  "diff --git ",       COLOR_YELLOW,   COLOR_DEFAULT,  0), \
543 LINE(DIFF_CHUNK,   "@@",                COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
544 LINE(DIFF_ADD,     "+",                 COLOR_GREEN,    COLOR_DEFAULT,  0), \
545 LINE(DIFF_DEL,     "-",                 COLOR_RED,      COLOR_DEFAULT,  0), \
546 LINE(DIFF_INDEX,        "index ",         COLOR_BLUE,   COLOR_DEFAULT,  0), \
547 LINE(DIFF_OLDMODE,      "old file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
548 LINE(DIFF_NEWMODE,      "new file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
549 LINE(DIFF_COPY_FROM,    "copy from",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
550 LINE(DIFF_COPY_TO,      "copy to",        COLOR_YELLOW, COLOR_DEFAULT,  0), \
551 LINE(DIFF_RENAME_FROM,  "rename from",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
552 LINE(DIFF_RENAME_TO,    "rename to",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
553 LINE(DIFF_SIMILARITY,   "similarity ",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
554 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
555 LINE(DIFF_TREE,         "diff-tree ",     COLOR_BLUE,   COLOR_DEFAULT,  0), \
556 LINE(PP_AUTHOR,    "Author: ",          COLOR_CYAN,     COLOR_DEFAULT,  0), \
557 LINE(PP_COMMIT,    "Commit: ",          COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
558 LINE(PP_MERGE,     "Merge: ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
559 LINE(PP_DATE,      "Date:   ",          COLOR_YELLOW,   COLOR_DEFAULT,  0), \
560 LINE(PP_ADATE,     "AuthorDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
561 LINE(PP_CDATE,     "CommitDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
562 LINE(PP_REFS,      "Refs: ",            COLOR_RED,      COLOR_DEFAULT,  0), \
563 LINE(COMMIT,       "commit ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
564 LINE(PARENT,       "parent ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
565 LINE(TREE,         "tree ",             COLOR_BLUE,     COLOR_DEFAULT,  0), \
566 LINE(AUTHOR,       "author ",           COLOR_CYAN,     COLOR_DEFAULT,  0), \
567 LINE(COMMITTER,    "committer ",        COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
568 LINE(SIGNOFF,      "    Signed-off-by", COLOR_YELLOW,   COLOR_DEFAULT,  0), \
569 LINE(ACKED,        "    Acked-by",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
570 LINE(DEFAULT,      "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
571 LINE(CURSOR,       "",                  COLOR_WHITE,    COLOR_GREEN,    A_BOLD), \
572 LINE(STATUS,       "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
573 LINE(TITLE_BLUR,   "",                  COLOR_WHITE,    COLOR_BLUE,     0), \
574 LINE(TITLE_FOCUS,  "",                  COLOR_WHITE,    COLOR_BLUE,     A_BOLD), \
575 LINE(MAIN_DATE,    "",                  COLOR_BLUE,     COLOR_DEFAULT,  0), \
576 LINE(MAIN_AUTHOR,  "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
577 LINE(MAIN_COMMIT,  "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
578 LINE(MAIN_DELIM,   "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
579 LINE(MAIN_TAG,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  A_BOLD), \
580 LINE(MAIN_REF,     "",                  COLOR_CYAN,     COLOR_DEFAULT,  A_BOLD), \
581 LINE(TREE_DIR,     "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
582 LINE(TREE_FILE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL)
584 enum line_type {
585 #define LINE(type, line, fg, bg, attr) \
586         LINE_##type
587         LINE_INFO
588 #undef  LINE
589 };
591 struct line_info {
592         const char *name;       /* Option name. */
593         int namelen;            /* Size of option name. */
594         const char *line;       /* The start of line to match. */
595         int linelen;            /* Size of string to match. */
596         int fg, bg, attr;       /* Color and text attributes for the lines. */
597 };
599 static struct line_info line_info[] = {
600 #define LINE(type, line, fg, bg, attr) \
601         { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
602         LINE_INFO
603 #undef  LINE
604 };
606 static enum line_type
607 get_line_type(char *line)
609         int linelen = strlen(line);
610         enum line_type type;
612         for (type = 0; type < ARRAY_SIZE(line_info); type++)
613                 /* Case insensitive search matches Signed-off-by lines better. */
614                 if (linelen >= line_info[type].linelen &&
615                     !strncasecmp(line_info[type].line, line, line_info[type].linelen))
616                         return type;
618         return LINE_DEFAULT;
621 static inline int
622 get_line_attr(enum line_type type)
624         assert(type < ARRAY_SIZE(line_info));
625         return COLOR_PAIR(type) | line_info[type].attr;
628 static struct line_info *
629 get_line_info(char *name, int namelen)
631         enum line_type type;
633         for (type = 0; type < ARRAY_SIZE(line_info); type++)
634                 if (namelen == line_info[type].namelen &&
635                     !string_enum_compare(line_info[type].name, name, namelen))
636                         return &line_info[type];
638         return NULL;
641 static void
642 init_colors(void)
644         int default_bg = COLOR_BLACK;
645         int default_fg = COLOR_WHITE;
646         enum line_type type;
648         start_color();
650         if (use_default_colors() != ERR) {
651                 default_bg = -1;
652                 default_fg = -1;
653         }
655         for (type = 0; type < ARRAY_SIZE(line_info); type++) {
656                 struct line_info *info = &line_info[type];
657                 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
658                 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
660                 init_pair(type, fg, bg);
661         }
664 struct line {
665         enum line_type type;
667         /* State flags */
668         unsigned int selected:1;
670         void *data;             /* User data */
671 };
674 /*
675  * Keys
676  */
678 struct keybinding {
679         int alias;
680         enum request request;
681         struct keybinding *next;
682 };
684 static struct keybinding default_keybindings[] = {
685         /* View switching */
686         { 'm',          REQ_VIEW_MAIN },
687         { 'd',          REQ_VIEW_DIFF },
688         { 'l',          REQ_VIEW_LOG },
689         { 't',          REQ_VIEW_TREE },
690         { 'f',          REQ_VIEW_BLOB },
691         { 'p',          REQ_VIEW_PAGER },
692         { 'h',          REQ_VIEW_HELP },
694         /* View manipulation */
695         { 'q',          REQ_VIEW_CLOSE },
696         { KEY_TAB,      REQ_VIEW_NEXT },
697         { KEY_RETURN,   REQ_ENTER },
698         { KEY_UP,       REQ_PREVIOUS },
699         { KEY_DOWN,     REQ_NEXT },
701         /* Cursor navigation */
702         { 'k',          REQ_MOVE_UP },
703         { 'j',          REQ_MOVE_DOWN },
704         { KEY_HOME,     REQ_MOVE_FIRST_LINE },
705         { KEY_END,      REQ_MOVE_LAST_LINE },
706         { KEY_NPAGE,    REQ_MOVE_PAGE_DOWN },
707         { ' ',          REQ_MOVE_PAGE_DOWN },
708         { KEY_PPAGE,    REQ_MOVE_PAGE_UP },
709         { 'b',          REQ_MOVE_PAGE_UP },
710         { '-',          REQ_MOVE_PAGE_UP },
712         /* Scrolling */
713         { KEY_IC,       REQ_SCROLL_LINE_UP },
714         { KEY_DC,       REQ_SCROLL_LINE_DOWN },
715         { 'w',          REQ_SCROLL_PAGE_UP },
716         { 's',          REQ_SCROLL_PAGE_DOWN },
718         /* Searching */
719         { '/',          REQ_SEARCH },
720         { '?',          REQ_SEARCH_BACK },
721         { 'n',          REQ_FIND_NEXT },
722         { 'N',          REQ_FIND_PREV },
724         /* Misc */
725         { 'Q',          REQ_QUIT },
726         { 'z',          REQ_STOP_LOADING },
727         { 'v',          REQ_SHOW_VERSION },
728         { 'r',          REQ_SCREEN_REDRAW },
729         { '.',          REQ_TOGGLE_LINENO },
730         { 'g',          REQ_TOGGLE_REV_GRAPH },
731         { ':',          REQ_PROMPT },
733         /* wgetch() with nodelay() enabled returns ERR when there's no input. */
734         { ERR,          REQ_NONE },
736         /* Using the ncurses SIGWINCH handler. */
737         { KEY_RESIZE,   REQ_SCREEN_RESIZE },
738 };
740 #define KEYMAP_INFO \
741         KEYMAP_(GENERIC), \
742         KEYMAP_(MAIN), \
743         KEYMAP_(DIFF), \
744         KEYMAP_(LOG), \
745         KEYMAP_(TREE), \
746         KEYMAP_(BLOB), \
747         KEYMAP_(PAGER), \
748         KEYMAP_(HELP) \
750 enum keymap {
751 #define KEYMAP_(name) KEYMAP_##name
752         KEYMAP_INFO
753 #undef  KEYMAP_
754 };
756 static struct int_map keymap_table[] = {
757 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
758         KEYMAP_INFO
759 #undef  KEYMAP_
760 };
762 #define set_keymap(map, name) \
763         set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
765 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
767 static void
768 add_keybinding(enum keymap keymap, enum request request, int key)
770         struct keybinding *keybinding;
772         keybinding = calloc(1, sizeof(*keybinding));
773         if (!keybinding)
774                 die("Failed to allocate keybinding");
776         keybinding->alias = key;
777         keybinding->request = request;
778         keybinding->next = keybindings[keymap];
779         keybindings[keymap] = keybinding;
782 /* Looks for a key binding first in the given map, then in the generic map, and
783  * lastly in the default keybindings. */
784 static enum request
785 get_keybinding(enum keymap keymap, int key)
787         struct keybinding *kbd;
788         int i;
790         for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
791                 if (kbd->alias == key)
792                         return kbd->request;
794         for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
795                 if (kbd->alias == key)
796                         return kbd->request;
798         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
799                 if (default_keybindings[i].alias == key)
800                         return default_keybindings[i].request;
802         return (enum request) key;
806 struct key {
807         char *name;
808         int value;
809 };
811 static struct key key_table[] = {
812         { "Enter",      KEY_RETURN },
813         { "Space",      ' ' },
814         { "Backspace",  KEY_BACKSPACE },
815         { "Tab",        KEY_TAB },
816         { "Escape",     KEY_ESC },
817         { "Left",       KEY_LEFT },
818         { "Right",      KEY_RIGHT },
819         { "Up",         KEY_UP },
820         { "Down",       KEY_DOWN },
821         { "Insert",     KEY_IC },
822         { "Delete",     KEY_DC },
823         { "Hash",       '#' },
824         { "Home",       KEY_HOME },
825         { "End",        KEY_END },
826         { "PageUp",     KEY_PPAGE },
827         { "PageDown",   KEY_NPAGE },
828         { "F1",         KEY_F(1) },
829         { "F2",         KEY_F(2) },
830         { "F3",         KEY_F(3) },
831         { "F4",         KEY_F(4) },
832         { "F5",         KEY_F(5) },
833         { "F6",         KEY_F(6) },
834         { "F7",         KEY_F(7) },
835         { "F8",         KEY_F(8) },
836         { "F9",         KEY_F(9) },
837         { "F10",        KEY_F(10) },
838         { "F11",        KEY_F(11) },
839         { "F12",        KEY_F(12) },
840 };
842 static int
843 get_key_value(const char *name)
845         int i;
847         for (i = 0; i < ARRAY_SIZE(key_table); i++)
848                 if (!strcasecmp(key_table[i].name, name))
849                         return key_table[i].value;
851         if (strlen(name) == 1 && isprint(*name))
852                 return (int) *name;
854         return ERR;
857 static char *
858 get_key(enum request request)
860         static char buf[BUFSIZ];
861         static char key_char[] = "'X'";
862         size_t pos = 0;
863         char *sep = "    ";
864         int i;
866         buf[pos] = 0;
868         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
869                 struct keybinding *keybinding = &default_keybindings[i];
870                 char *seq = NULL;
871                 int key;
873                 if (keybinding->request != request)
874                         continue;
876                 for (key = 0; key < ARRAY_SIZE(key_table); key++)
877                         if (key_table[key].value == keybinding->alias)
878                                 seq = key_table[key].name;
880                 if (seq == NULL &&
881                     keybinding->alias < 127 &&
882                     isprint(keybinding->alias)) {
883                         key_char[1] = (char) keybinding->alias;
884                         seq = key_char;
885                 }
887                 if (!seq)
888                         seq = "'?'";
890                 if (!string_format_from(buf, &pos, "%s%s", sep, seq))
891                         return "Too many keybindings!";
892                 sep = ", ";
893         }
895         return buf;
899 /*
900  * User config file handling.
901  */
903 static struct int_map color_map[] = {
904 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
905         COLOR_MAP(DEFAULT),
906         COLOR_MAP(BLACK),
907         COLOR_MAP(BLUE),
908         COLOR_MAP(CYAN),
909         COLOR_MAP(GREEN),
910         COLOR_MAP(MAGENTA),
911         COLOR_MAP(RED),
912         COLOR_MAP(WHITE),
913         COLOR_MAP(YELLOW),
914 };
916 #define set_color(color, name) \
917         set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
919 static struct int_map attr_map[] = {
920 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
921         ATTR_MAP(NORMAL),
922         ATTR_MAP(BLINK),
923         ATTR_MAP(BOLD),
924         ATTR_MAP(DIM),
925         ATTR_MAP(REVERSE),
926         ATTR_MAP(STANDOUT),
927         ATTR_MAP(UNDERLINE),
928 };
930 #define set_attribute(attr, name) \
931         set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
933 static int   config_lineno;
934 static bool  config_errors;
935 static char *config_msg;
937 /* Wants: object fgcolor bgcolor [attr] */
938 static int
939 option_color_command(int argc, char *argv[])
941         struct line_info *info;
943         if (argc != 3 && argc != 4) {
944                 config_msg = "Wrong number of arguments given to color command";
945                 return ERR;
946         }
948         info = get_line_info(argv[0], strlen(argv[0]));
949         if (!info) {
950                 config_msg = "Unknown color name";
951                 return ERR;
952         }
954         if (set_color(&info->fg, argv[1]) == ERR ||
955             set_color(&info->bg, argv[2]) == ERR) {
956                 config_msg = "Unknown color";
957                 return ERR;
958         }
960         if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
961                 config_msg = "Unknown attribute";
962                 return ERR;
963         }
965         return OK;
968 /* Wants: name = value */
969 static int
970 option_set_command(int argc, char *argv[])
972         if (argc != 3) {
973                 config_msg = "Wrong number of arguments given to set command";
974                 return ERR;
975         }
977         if (strcmp(argv[1], "=")) {
978                 config_msg = "No value assigned";
979                 return ERR;
980         }
982         if (!strcmp(argv[0], "show-rev-graph")) {
983                 opt_rev_graph = (!strcmp(argv[2], "1") ||
984                                  !strcmp(argv[2], "true") ||
985                                  !strcmp(argv[2], "yes"));
986                 return OK;
987         }
989         if (!strcmp(argv[0], "line-number-interval")) {
990                 opt_num_interval = atoi(argv[2]);
991                 return OK;
992         }
994         if (!strcmp(argv[0], "tab-size")) {
995                 opt_tab_size = atoi(argv[2]);
996                 return OK;
997         }
999         if (!strcmp(argv[0], "commit-encoding")) {
1000                 char *arg = argv[2];
1001                 int delimiter = *arg;
1002                 int i;
1004                 switch (delimiter) {
1005                 case '"':
1006                 case '\'':
1007                         for (arg++, i = 0; arg[i]; i++)
1008                                 if (arg[i] == delimiter) {
1009                                         arg[i] = 0;
1010                                         break;
1011                                 }
1012                 default:
1013                         string_copy(opt_encoding, arg);
1014                         return OK;
1015                 }
1016         }
1018         config_msg = "Unknown variable name";
1019         return ERR;
1022 /* Wants: mode request key */
1023 static int
1024 option_bind_command(int argc, char *argv[])
1026         enum request request;
1027         int keymap;
1028         int key;
1030         if (argc != 3) {
1031                 config_msg = "Wrong number of arguments given to bind command";
1032                 return ERR;
1033         }
1035         if (set_keymap(&keymap, argv[0]) == ERR) {
1036                 config_msg = "Unknown key map";
1037                 return ERR;
1038         }
1040         key = get_key_value(argv[1]);
1041         if (key == ERR) {
1042                 config_msg = "Unknown key";
1043                 return ERR;
1044         }
1046         request = get_request(argv[2]);
1047         if (request == REQ_UNKNOWN) {
1048                 config_msg = "Unknown request name";
1049                 return ERR;
1050         }
1052         add_keybinding(keymap, request, key);
1054         return OK;
1057 static int
1058 set_option(char *opt, char *value)
1060         char *argv[16];
1061         int valuelen;
1062         int argc = 0;
1064         /* Tokenize */
1065         while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1066                 argv[argc++] = value;
1068                 value += valuelen;
1069                 if (!*value)
1070                         break;
1072                 *value++ = 0;
1073                 while (isspace(*value))
1074                         value++;
1075         }
1077         if (!strcmp(opt, "color"))
1078                 return option_color_command(argc, argv);
1080         if (!strcmp(opt, "set"))
1081                 return option_set_command(argc, argv);
1083         if (!strcmp(opt, "bind"))
1084                 return option_bind_command(argc, argv);
1086         config_msg = "Unknown option command";
1087         return ERR;
1090 static int
1091 read_option(char *opt, int optlen, char *value, int valuelen)
1093         int status = OK;
1095         config_lineno++;
1096         config_msg = "Internal error";
1098         /* Check for comment markers, since read_properties() will
1099          * only ensure opt and value are split at first " \t". */
1100         optlen = strcspn(opt, "#");
1101         if (optlen == 0)
1102                 return OK;
1104         if (opt[optlen] != 0) {
1105                 config_msg = "No option value";
1106                 status = ERR;
1108         }  else {
1109                 /* Look for comment endings in the value. */
1110                 int len = strcspn(value, "#");
1112                 if (len < valuelen) {
1113                         valuelen = len;
1114                         value[valuelen] = 0;
1115                 }
1117                 status = set_option(opt, value);
1118         }
1120         if (status == ERR) {
1121                 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1122                         config_lineno, optlen, opt, config_msg);
1123                 config_errors = TRUE;
1124         }
1126         /* Always keep going if errors are encountered. */
1127         return OK;
1130 static int
1131 load_options(void)
1133         char *home = getenv("HOME");
1134         char buf[SIZEOF_STR];
1135         FILE *file;
1137         config_lineno = 0;
1138         config_errors = FALSE;
1140         if (!home || !string_format(buf, "%s/.tigrc", home))
1141                 return ERR;
1143         /* It's ok that the file doesn't exist. */
1144         file = fopen(buf, "r");
1145         if (!file)
1146                 return OK;
1148         if (read_properties(file, " \t", read_option) == ERR ||
1149             config_errors == TRUE)
1150                 fprintf(stderr, "Errors while loading %s.\n", buf);
1152         return OK;
1156 /*
1157  * The viewer
1158  */
1160 struct view;
1161 struct view_ops;
1163 /* The display array of active views and the index of the current view. */
1164 static struct view *display[2];
1165 static unsigned int current_view;
1167 #define foreach_displayed_view(view, i) \
1168         for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1170 #define displayed_views()       (display[1] != NULL ? 2 : 1)
1172 /* Current head and commit ID */
1173 static char ref_blob[SIZEOF_REF]        = "";
1174 static char ref_commit[SIZEOF_REF]      = "HEAD";
1175 static char ref_head[SIZEOF_REF]        = "HEAD";
1177 struct view {
1178         const char *name;       /* View name */
1179         const char *cmd_fmt;    /* Default command line format */
1180         const char *cmd_env;    /* Command line set via environment */
1181         const char *id;         /* Points to either of ref_{head,commit,blob} */
1183         struct view_ops *ops;   /* View operations */
1185         enum keymap keymap;     /* What keymap does this view have */
1187         char cmd[SIZEOF_STR];   /* Command buffer */
1188         char ref[SIZEOF_REF];   /* Hovered commit reference */
1189         char vid[SIZEOF_REF];   /* View ID. Set to id member when updating. */
1191         int height, width;      /* The width and height of the main window */
1192         WINDOW *win;            /* The main window */
1193         WINDOW *title;          /* The title window living below the main window */
1195         /* Navigation */
1196         unsigned long offset;   /* Offset of the window top */
1197         unsigned long lineno;   /* Current line number */
1199         /* Searching */
1200         char grep[SIZEOF_STR];  /* Search string */
1201         regex_t *regex;         /* Pre-compiled regex */
1203         /* If non-NULL, points to the view that opened this view. If this view
1204          * is closed tig will switch back to the parent view. */
1205         struct view *parent;
1207         /* Buffering */
1208         unsigned long lines;    /* Total number of lines */
1209         struct line *line;      /* Line index */
1210         unsigned long line_size;/* Total number of allocated lines */
1211         unsigned int digits;    /* Number of digits in the lines member. */
1213         /* Loading */
1214         FILE *pipe;
1215         time_t start_time;
1216 };
1218 struct view_ops {
1219         /* What type of content being displayed. Used in the title bar. */
1220         const char *type;
1221         /* Draw one line; @lineno must be < view->height. */
1222         bool (*draw)(struct view *view, struct line *line, unsigned int lineno, bool selected);
1223         /* Read one line; updates view->line. */
1224         bool (*read)(struct view *view, char *data);
1225         /* Depending on view, change display based on current line. */
1226         bool (*enter)(struct view *view, struct line *line);
1227         /* Search for regex in a line. */
1228         bool (*grep)(struct view *view, struct line *line);
1229         /* Select line */
1230         void (*select)(struct view *view, struct line *line);
1231 };
1233 static struct view_ops pager_ops;
1234 static struct view_ops main_ops;
1235 static struct view_ops tree_ops;
1236 static struct view_ops blob_ops;
1238 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1239         { name, cmd, #env, ref, ops, map}
1241 #define VIEW_(id, name, ops, ref) \
1242         VIEW_STR(name, TIG_##id##_CMD,  TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1245 static struct view views[] = {
1246         VIEW_(MAIN,  "main",  &main_ops,  ref_head),
1247         VIEW_(DIFF,  "diff",  &pager_ops, ref_commit),
1248         VIEW_(LOG,   "log",   &pager_ops, ref_head),
1249         VIEW_(TREE,  "tree",  &tree_ops,  ref_commit),
1250         VIEW_(BLOB,  "blob",  &blob_ops,  ref_blob),
1251         VIEW_(HELP,  "help",  &pager_ops, "static"),
1252         VIEW_(PAGER, "pager", &pager_ops, "static"),
1253 };
1255 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1257 #define foreach_view(view, i) \
1258         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1260 #define view_is_displayed(view) \
1261         (view == display[0] || view == display[1])
1263 static bool
1264 draw_view_line(struct view *view, unsigned int lineno)
1266         struct line *line;
1267         bool selected = (view->offset + lineno == view->lineno);
1268         bool draw_ok;
1270         assert(view_is_displayed(view));
1272         if (view->offset + lineno >= view->lines)
1273                 return FALSE;
1275         line = &view->line[view->offset + lineno];
1277         if (selected) {
1278                 line->selected = TRUE;
1279                 view->ops->select(view, line);
1280         } else if (line->selected) {
1281                 line->selected = FALSE;
1282                 wmove(view->win, lineno, 0);
1283                 wclrtoeol(view->win);
1284         }
1286         scrollok(view->win, FALSE);
1287         draw_ok = view->ops->draw(view, line, lineno, selected);
1288         scrollok(view->win, TRUE);
1290         return draw_ok;
1293 static void
1294 redraw_view_from(struct view *view, int lineno)
1296         assert(0 <= lineno && lineno < view->height);
1298         for (; lineno < view->height; lineno++) {
1299                 if (!draw_view_line(view, lineno))
1300                         break;
1301         }
1303         redrawwin(view->win);
1304         wrefresh(view->win);
1307 static void
1308 redraw_view(struct view *view)
1310         wclear(view->win);
1311         redraw_view_from(view, 0);
1315 static void
1316 update_view_title(struct view *view)
1318         char buf[SIZEOF_STR];
1319         size_t bufpos = 0;
1321         assert(view_is_displayed(view));
1323         string_format_from(buf, &bufpos, "[%s]", view->name);
1324         if (*view->ref)
1325                 string_format_from(buf, &bufpos, " %s", view->ref);
1327         if (view->lines || view->pipe) {
1328                 unsigned int view_lines = view->offset + view->height;
1329                 unsigned int lines = view->lines
1330                                    ? MIN(view_lines, view->lines) * 100 / view->lines
1331                                    : 0;
1333                 string_format_from(buf, &bufpos, " - %s %d of %d (%d%%)",
1334                                    view->ops->type,
1335                                    view->lineno + 1,
1336                                    view->lines,
1337                                    lines);
1339                 if (view->pipe) {
1340                         time_t secs = time(NULL) - view->start_time;
1342                         /* Three git seconds are a long time ... */
1343                         if (secs > 2)
1344                                 string_format_from(buf, &bufpos, " %lds", secs);
1345                 }
1346         }
1348         if (view == display[current_view])
1349                 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1350         else
1351                 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1353         werase(view->title);
1354         mvwaddnstr(view->title, 0, 0, buf, bufpos);
1355         wmove(view->title, 0, view->width - 1);
1356         wrefresh(view->title);
1359 static void
1360 resize_display(void)
1362         int offset, i;
1363         struct view *base = display[0];
1364         struct view *view = display[1] ? display[1] : display[0];
1366         /* Setup window dimensions */
1368         getmaxyx(stdscr, base->height, base->width);
1370         /* Make room for the status window. */
1371         base->height -= 1;
1373         if (view != base) {
1374                 /* Horizontal split. */
1375                 view->width   = base->width;
1376                 view->height  = SCALE_SPLIT_VIEW(base->height);
1377                 base->height -= view->height;
1379                 /* Make room for the title bar. */
1380                 view->height -= 1;
1381         }
1383         /* Make room for the title bar. */
1384         base->height -= 1;
1386         offset = 0;
1388         foreach_displayed_view (view, i) {
1389                 if (!view->win) {
1390                         view->win = newwin(view->height, 0, offset, 0);
1391                         if (!view->win)
1392                                 die("Failed to create %s view", view->name);
1394                         scrollok(view->win, TRUE);
1396                         view->title = newwin(1, 0, offset + view->height, 0);
1397                         if (!view->title)
1398                                 die("Failed to create title window");
1400                 } else {
1401                         wresize(view->win, view->height, view->width);
1402                         mvwin(view->win,   offset, 0);
1403                         mvwin(view->title, offset + view->height, 0);
1404                 }
1406                 offset += view->height + 1;
1407         }
1410 static void
1411 redraw_display(void)
1413         struct view *view;
1414         int i;
1416         foreach_displayed_view (view, i) {
1417                 redraw_view(view);
1418                 update_view_title(view);
1419         }
1422 static void
1423 update_display_cursor(struct view *view)
1425         /* Move the cursor to the right-most column of the cursor line.
1426          *
1427          * XXX: This could turn out to be a bit expensive, but it ensures that
1428          * the cursor does not jump around. */
1429         if (view->lines) {
1430                 wmove(view->win, view->lineno - view->offset, view->width - 1);
1431                 wrefresh(view->win);
1432         }
1435 /*
1436  * Navigation
1437  */
1439 /* Scrolling backend */
1440 static void
1441 do_scroll_view(struct view *view, int lines)
1443         bool redraw_current_line = FALSE;
1445         /* The rendering expects the new offset. */
1446         view->offset += lines;
1448         assert(0 <= view->offset && view->offset < view->lines);
1449         assert(lines);
1451         /* Move current line into the view. */
1452         if (view->lineno < view->offset) {
1453                 view->lineno = view->offset;
1454                 redraw_current_line = TRUE;
1455         } else if (view->lineno >= view->offset + view->height) {
1456                 view->lineno = view->offset + view->height - 1;
1457                 redraw_current_line = TRUE;
1458         }
1460         assert(view->offset <= view->lineno && view->lineno < view->lines);
1462         /* Redraw the whole screen if scrolling is pointless. */
1463         if (view->height < ABS(lines)) {
1464                 redraw_view(view);
1466         } else {
1467                 int line = lines > 0 ? view->height - lines : 0;
1468                 int end = line + ABS(lines);
1470                 wscrl(view->win, lines);
1472                 for (; line < end; line++) {
1473                         if (!draw_view_line(view, line))
1474                                 break;
1475                 }
1477                 if (redraw_current_line)
1478                         draw_view_line(view, view->lineno - view->offset);
1479         }
1481         redrawwin(view->win);
1482         wrefresh(view->win);
1483         report("");
1486 /* Scroll frontend */
1487 static void
1488 scroll_view(struct view *view, enum request request)
1490         int lines = 1;
1492         assert(view_is_displayed(view));
1494         switch (request) {
1495         case REQ_SCROLL_PAGE_DOWN:
1496                 lines = view->height;
1497         case REQ_SCROLL_LINE_DOWN:
1498                 if (view->offset + lines > view->lines)
1499                         lines = view->lines - view->offset;
1501                 if (lines == 0 || view->offset + view->height >= view->lines) {
1502                         report("Cannot scroll beyond the last line");
1503                         return;
1504                 }
1505                 break;
1507         case REQ_SCROLL_PAGE_UP:
1508                 lines = view->height;
1509         case REQ_SCROLL_LINE_UP:
1510                 if (lines > view->offset)
1511                         lines = view->offset;
1513                 if (lines == 0) {
1514                         report("Cannot scroll beyond the first line");
1515                         return;
1516                 }
1518                 lines = -lines;
1519                 break;
1521         default:
1522                 die("request %d not handled in switch", request);
1523         }
1525         do_scroll_view(view, lines);
1528 /* Cursor moving */
1529 static void
1530 move_view(struct view *view, enum request request)
1532         int scroll_steps = 0;
1533         int steps;
1535         switch (request) {
1536         case REQ_MOVE_FIRST_LINE:
1537                 steps = -view->lineno;
1538                 break;
1540         case REQ_MOVE_LAST_LINE:
1541                 steps = view->lines - view->lineno - 1;
1542                 break;
1544         case REQ_MOVE_PAGE_UP:
1545                 steps = view->height > view->lineno
1546                       ? -view->lineno : -view->height;
1547                 break;
1549         case REQ_MOVE_PAGE_DOWN:
1550                 steps = view->lineno + view->height >= view->lines
1551                       ? view->lines - view->lineno - 1 : view->height;
1552                 break;
1554         case REQ_MOVE_UP:
1555                 steps = -1;
1556                 break;
1558         case REQ_MOVE_DOWN:
1559                 steps = 1;
1560                 break;
1562         default:
1563                 die("request %d not handled in switch", request);
1564         }
1566         if (steps <= 0 && view->lineno == 0) {
1567                 report("Cannot move beyond the first line");
1568                 return;
1570         } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1571                 report("Cannot move beyond the last line");
1572                 return;
1573         }
1575         /* Move the current line */
1576         view->lineno += steps;
1577         assert(0 <= view->lineno && view->lineno < view->lines);
1579         /* Check whether the view needs to be scrolled */
1580         if (view->lineno < view->offset ||
1581             view->lineno >= view->offset + view->height) {
1582                 scroll_steps = steps;
1583                 if (steps < 0 && -steps > view->offset) {
1584                         scroll_steps = -view->offset;
1586                 } else if (steps > 0) {
1587                         if (view->lineno == view->lines - 1 &&
1588                             view->lines > view->height) {
1589                                 scroll_steps = view->lines - view->offset - 1;
1590                                 if (scroll_steps >= view->height)
1591                                         scroll_steps -= view->height - 1;
1592                         }
1593                 }
1594         }
1596         if (!view_is_displayed(view)) {
1597                 view->offset += steps;
1598                 view->ops->select(view, &view->line[view->lineno]);
1599                 return;
1600         }
1602         /* Repaint the old "current" line if we be scrolling */
1603         if (ABS(steps) < view->height)
1604                 draw_view_line(view, view->lineno - steps - view->offset);
1606         if (scroll_steps) {
1607                 do_scroll_view(view, scroll_steps);
1608                 return;
1609         }
1611         /* Draw the current line */
1612         draw_view_line(view, view->lineno - view->offset);
1614         redrawwin(view->win);
1615         wrefresh(view->win);
1616         report("");
1620 /*
1621  * Searching
1622  */
1624 static void search_view(struct view *view, enum request request);
1626 static bool
1627 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1629         assert(view_is_displayed(view));
1631         if (!view->ops->grep(view, line))
1632                 return FALSE;
1634         if (lineno - view->offset >= view->height) {
1635                 view->offset = lineno;
1636                 view->lineno = lineno;
1637                 redraw_view(view);
1639         } else {
1640                 unsigned long old_lineno = view->lineno - view->offset;
1642                 view->lineno = lineno;
1643                 draw_view_line(view, old_lineno);
1645                 draw_view_line(view, view->lineno - view->offset);
1646                 redrawwin(view->win);
1647                 wrefresh(view->win);
1648         }
1650         report("Line %ld matches '%s'", lineno + 1, view->grep);
1651         return TRUE;
1654 static void
1655 find_next(struct view *view, enum request request)
1657         unsigned long lineno = view->lineno;
1658         int direction;
1660         if (!*view->grep) {
1661                 if (!*opt_search)
1662                         report("No previous search");
1663                 else
1664                         search_view(view, request);
1665                 return;
1666         }
1668         switch (request) {
1669         case REQ_SEARCH:
1670         case REQ_FIND_NEXT:
1671                 direction = 1;
1672                 break;
1674         case REQ_SEARCH_BACK:
1675         case REQ_FIND_PREV:
1676                 direction = -1;
1677                 break;
1679         default:
1680                 return;
1681         }
1683         if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1684                 lineno += direction;
1686         /* Note, lineno is unsigned long so will wrap around in which case it
1687          * will become bigger than view->lines. */
1688         for (; lineno < view->lines; lineno += direction) {
1689                 struct line *line = &view->line[lineno];
1691                 if (find_next_line(view, lineno, line))
1692                         return;
1693         }
1695         report("No match found for '%s'", view->grep);
1698 static void
1699 search_view(struct view *view, enum request request)
1701         int regex_err;
1703         if (view->regex) {
1704                 regfree(view->regex);
1705                 *view->grep = 0;
1706         } else {
1707                 view->regex = calloc(1, sizeof(*view->regex));
1708                 if (!view->regex)
1709                         return;
1710         }
1712         regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
1713         if (regex_err != 0) {
1714                 char buf[SIZEOF_STR] = "unknown error";
1716                 regerror(regex_err, view->regex, buf, sizeof(buf));
1717                 report("Search failed: %s", buf);
1718                 return;
1719         }
1721         string_copy(view->grep, opt_search);
1723         find_next(view, request);
1726 /*
1727  * Incremental updating
1728  */
1730 static void
1731 end_update(struct view *view)
1733         if (!view->pipe)
1734                 return;
1735         set_nonblocking_input(FALSE);
1736         if (view->pipe == stdin)
1737                 fclose(view->pipe);
1738         else
1739                 pclose(view->pipe);
1740         view->pipe = NULL;
1743 static bool
1744 begin_update(struct view *view)
1746         const char *id = view->id;
1748         if (view->pipe)
1749                 end_update(view);
1751         if (opt_cmd[0]) {
1752                 string_copy(view->cmd, opt_cmd);
1753                 opt_cmd[0] = 0;
1754                 /* When running random commands, the view ref could have become
1755                  * invalid so clear it. */
1756                 view->ref[0] = 0;
1758         } else if (view == VIEW(REQ_VIEW_TREE)) {
1759                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1761                 if (strcmp(view->vid, view->id))
1762                         opt_path[0] = 0;
1764                 if (!string_format(view->cmd, format, id, opt_path))
1765                         return FALSE;
1767         } else {
1768                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1770                 if (!string_format(view->cmd, format, id, id, id, id, id))
1771                         return FALSE;
1772         }
1774         /* Special case for the pager view. */
1775         if (opt_pipe) {
1776                 view->pipe = opt_pipe;
1777                 opt_pipe = NULL;
1778         } else {
1779                 view->pipe = popen(view->cmd, "r");
1780         }
1782         if (!view->pipe)
1783                 return FALSE;
1785         set_nonblocking_input(TRUE);
1787         view->offset = 0;
1788         view->lines  = 0;
1789         view->lineno = 0;
1790         string_copy(view->vid, id);
1792         if (view->line) {
1793                 int i;
1795                 for (i = 0; i < view->lines; i++)
1796                         if (view->line[i].data)
1797                                 free(view->line[i].data);
1799                 free(view->line);
1800                 view->line = NULL;
1801         }
1803         view->start_time = time(NULL);
1805         return TRUE;
1808 static struct line *
1809 realloc_lines(struct view *view, size_t line_size)
1811         struct line *tmp = realloc(view->line, sizeof(*view->line) * line_size);
1813         if (!tmp)
1814                 return NULL;
1816         view->line = tmp;
1817         view->line_size = line_size;
1818         return view->line;
1821 static bool
1822 update_view(struct view *view)
1824         char in_buffer[BUFSIZ];
1825         char out_buffer[BUFSIZ * 2];
1826         char *line;
1827         /* The number of lines to read. If too low it will cause too much
1828          * redrawing (and possible flickering), if too high responsiveness
1829          * will suffer. */
1830         unsigned long lines = view->height;
1831         int redraw_from = -1;
1833         if (!view->pipe)
1834                 return TRUE;
1836         /* Only redraw if lines are visible. */
1837         if (view->offset + view->height >= view->lines)
1838                 redraw_from = view->lines - view->offset;
1840         /* FIXME: This is probably not perfect for backgrounded views. */
1841         if (!realloc_lines(view, view->lines + lines))
1842                 goto alloc_error;
1844         while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
1845                 size_t linelen = strlen(line);
1847                 if (linelen)
1848                         line[linelen - 1] = 0;
1850                 if (opt_iconv != ICONV_NONE) {
1851                         char *inbuf = line;
1852                         size_t inlen = linelen;
1854                         char *outbuf = out_buffer;
1855                         size_t outlen = sizeof(out_buffer);
1857                         size_t ret;
1859                         ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
1860                         if (ret != (size_t) -1) {
1861                                 line = out_buffer;
1862                                 linelen = strlen(out_buffer);
1863                         }
1864                 }
1866                 if (!view->ops->read(view, line))
1867                         goto alloc_error;
1869                 if (lines-- == 1)
1870                         break;
1871         }
1873         {
1874                 int digits;
1876                 lines = view->lines;
1877                 for (digits = 0; lines; digits++)
1878                         lines /= 10;
1880                 /* Keep the displayed view in sync with line number scaling. */
1881                 if (digits != view->digits) {
1882                         view->digits = digits;
1883                         redraw_from = 0;
1884                 }
1885         }
1887         if (!view_is_displayed(view))
1888                 goto check_pipe;
1890         if (view == VIEW(REQ_VIEW_TREE)) {
1891                 /* Clear the view and redraw everything since the tree sorting
1892                  * might have rearranged things. */
1893                 redraw_view(view);
1895         } else if (redraw_from >= 0) {
1896                 /* If this is an incremental update, redraw the previous line
1897                  * since for commits some members could have changed when
1898                  * loading the main view. */
1899                 if (redraw_from > 0)
1900                         redraw_from--;
1902                 /* Incrementally draw avoids flickering. */
1903                 redraw_view_from(view, redraw_from);
1904         }
1906         /* Update the title _after_ the redraw so that if the redraw picks up a
1907          * commit reference in view->ref it'll be available here. */
1908         update_view_title(view);
1910 check_pipe:
1911         if (ferror(view->pipe)) {
1912                 report("Failed to read: %s", strerror(errno));
1913                 goto end;
1915         } else if (feof(view->pipe)) {
1916                 report("");
1917                 goto end;
1918         }
1920         return TRUE;
1922 alloc_error:
1923         report("Allocation failure");
1925 end:
1926         end_update(view);
1927         return FALSE;
1931 /*
1932  * View opening
1933  */
1935 static void open_help_view(struct view *view)
1937         char buf[BUFSIZ];
1938         int lines = ARRAY_SIZE(req_info) + 2;
1939         int i;
1941         if (view->lines > 0)
1942                 return;
1944         for (i = 0; i < ARRAY_SIZE(req_info); i++)
1945                 if (!req_info[i].request)
1946                         lines++;
1948         view->line = calloc(lines, sizeof(*view->line));
1949         if (!view->line) {
1950                 report("Allocation failure");
1951                 return;
1952         }
1954         view->ops->read(view, "Quick reference for tig keybindings:");
1956         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
1957                 char *key;
1959                 if (!req_info[i].request) {
1960                         view->ops->read(view, "");
1961                         view->ops->read(view, req_info[i].help);
1962                         continue;
1963                 }
1965                 key = get_key(req_info[i].request);
1966                 if (!string_format(buf, "%-25s %s", key, req_info[i].help))
1967                         continue;
1969                 view->ops->read(view, buf);
1970         }
1973 enum open_flags {
1974         OPEN_DEFAULT = 0,       /* Use default view switching. */
1975         OPEN_SPLIT = 1,         /* Split current view. */
1976         OPEN_BACKGROUNDED = 2,  /* Backgrounded. */
1977         OPEN_RELOAD = 4,        /* Reload view even if it is the current. */
1978 };
1980 static void
1981 open_view(struct view *prev, enum request request, enum open_flags flags)
1983         bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
1984         bool split = !!(flags & OPEN_SPLIT);
1985         bool reload = !!(flags & OPEN_RELOAD);
1986         struct view *view = VIEW(request);
1987         int nviews = displayed_views();
1988         struct view *base_view = display[0];
1990         if (view == prev && nviews == 1 && !reload) {
1991                 report("Already in %s view", view->name);
1992                 return;
1993         }
1995         if (view == VIEW(REQ_VIEW_HELP)) {
1996                 open_help_view(view);
1998         } else if ((reload || strcmp(view->vid, view->id)) &&
1999                    !begin_update(view)) {
2000                 report("Failed to load %s view", view->name);
2001                 return;
2002         }
2004         if (split) {
2005                 display[1] = view;
2006                 if (!backgrounded)
2007                         current_view = 1;
2008         } else {
2009                 /* Maximize the current view. */
2010                 memset(display, 0, sizeof(display));
2011                 current_view = 0;
2012                 display[current_view] = view;
2013         }
2015         /* Resize the view when switching between split- and full-screen,
2016          * or when switching between two different full-screen views. */
2017         if (nviews != displayed_views() ||
2018             (nviews == 1 && base_view != display[0]))
2019                 resize_display();
2021         if (split && prev->lineno - prev->offset >= prev->height) {
2022                 /* Take the title line into account. */
2023                 int lines = prev->lineno - prev->offset - prev->height + 1;
2025                 /* Scroll the view that was split if the current line is
2026                  * outside the new limited view. */
2027                 do_scroll_view(prev, lines);
2028         }
2030         if (prev && view != prev) {
2031                 if (split && !backgrounded) {
2032                         /* "Blur" the previous view. */
2033                         update_view_title(prev);
2034                 }
2036                 view->parent = prev;
2037         }
2039         if (view->pipe && view->lines == 0) {
2040                 /* Clear the old view and let the incremental updating refill
2041                  * the screen. */
2042                 wclear(view->win);
2043                 report("");
2044         } else {
2045                 redraw_view(view);
2046                 report("");
2047         }
2049         /* If the view is backgrounded the above calls to report()
2050          * won't redraw the view title. */
2051         if (backgrounded)
2052                 update_view_title(view);
2056 /*
2057  * User request switch noodle
2058  */
2060 static int
2061 view_driver(struct view *view, enum request request)
2063         int i;
2065         switch (request) {
2066         case REQ_MOVE_UP:
2067         case REQ_MOVE_DOWN:
2068         case REQ_MOVE_PAGE_UP:
2069         case REQ_MOVE_PAGE_DOWN:
2070         case REQ_MOVE_FIRST_LINE:
2071         case REQ_MOVE_LAST_LINE:
2072                 move_view(view, request);
2073                 break;
2075         case REQ_SCROLL_LINE_DOWN:
2076         case REQ_SCROLL_LINE_UP:
2077         case REQ_SCROLL_PAGE_DOWN:
2078         case REQ_SCROLL_PAGE_UP:
2079                 scroll_view(view, request);
2080                 break;
2082         case REQ_VIEW_BLOB:
2083                 if (!ref_blob[0]) {
2084                         report("No file chosen, press 't' to open tree view");
2085                         break;
2086                 }
2087                 /* Fall-through */
2088         case REQ_VIEW_MAIN:
2089         case REQ_VIEW_DIFF:
2090         case REQ_VIEW_LOG:
2091         case REQ_VIEW_TREE:
2092         case REQ_VIEW_HELP:
2093         case REQ_VIEW_PAGER:
2094                 open_view(view, request, OPEN_DEFAULT);
2095                 break;
2097         case REQ_NEXT:
2098         case REQ_PREVIOUS:
2099                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2101                 if ((view == VIEW(REQ_VIEW_DIFF) &&
2102                      view->parent == VIEW(REQ_VIEW_MAIN)) ||
2103                    (view == VIEW(REQ_VIEW_BLOB) &&
2104                      view->parent == VIEW(REQ_VIEW_TREE))) {
2105                         view = view->parent;
2106                         move_view(view, request);
2107                         if (view_is_displayed(view))
2108                                 update_view_title(view);
2109                 } else {
2110                         move_view(view, request);
2111                         break;
2112                 }
2113                 /* Fall-through */
2115         case REQ_ENTER:
2116                 if (!view->lines) {
2117                         report("Nothing to enter");
2118                         break;
2119                 }
2120                 return view->ops->enter(view, &view->line[view->lineno]);
2122         case REQ_VIEW_NEXT:
2123         {
2124                 int nviews = displayed_views();
2125                 int next_view = (current_view + 1) % nviews;
2127                 if (next_view == current_view) {
2128                         report("Only one view is displayed");
2129                         break;
2130                 }
2132                 current_view = next_view;
2133                 /* Blur out the title of the previous view. */
2134                 update_view_title(view);
2135                 report("");
2136                 break;
2137         }
2138         case REQ_TOGGLE_LINENO:
2139                 opt_line_number = !opt_line_number;
2140                 redraw_display();
2141                 break;
2143         case REQ_TOGGLE_REV_GRAPH:
2144                 opt_rev_graph = !opt_rev_graph;
2145                 redraw_display();
2146                 break;
2148         case REQ_PROMPT:
2149                 /* Always reload^Wrerun commands from the prompt. */
2150                 open_view(view, opt_request, OPEN_RELOAD);
2151                 break;
2153         case REQ_SEARCH:
2154         case REQ_SEARCH_BACK:
2155                 search_view(view, request);
2156                 break;
2158         case REQ_FIND_NEXT:
2159         case REQ_FIND_PREV:
2160                 find_next(view, request);
2161                 break;
2163         case REQ_STOP_LOADING:
2164                 for (i = 0; i < ARRAY_SIZE(views); i++) {
2165                         view = &views[i];
2166                         if (view->pipe)
2167                                 report("Stopped loading the %s view", view->name),
2168                         end_update(view);
2169                 }
2170                 break;
2172         case REQ_SHOW_VERSION:
2173                 report("%s (built %s)", VERSION, __DATE__);
2174                 return TRUE;
2176         case REQ_SCREEN_RESIZE:
2177                 resize_display();
2178                 /* Fall-through */
2179         case REQ_SCREEN_REDRAW:
2180                 redraw_display();
2181                 break;
2183         case REQ_NONE:
2184                 doupdate();
2185                 return TRUE;
2187         case REQ_VIEW_CLOSE:
2188                 /* XXX: Mark closed views by letting view->parent point to the
2189                  * view itself. Parents to closed view should never be
2190                  * followed. */
2191                 if (view->parent &&
2192                     view->parent->parent != view->parent) {
2193                         memset(display, 0, sizeof(display));
2194                         current_view = 0;
2195                         display[current_view] = view->parent;
2196                         view->parent = view;
2197                         resize_display();
2198                         redraw_display();
2199                         break;
2200                 }
2201                 /* Fall-through */
2202         case REQ_QUIT:
2203                 return FALSE;
2205         default:
2206                 /* An unknown key will show most commonly used commands. */
2207                 report("Unknown key, press 'h' for help");
2208                 return TRUE;
2209         }
2211         return TRUE;
2215 /*
2216  * Pager backend
2217  */
2219 static bool
2220 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2222         char *text = line->data;
2223         enum line_type type = line->type;
2224         int textlen = strlen(text);
2225         int attr;
2227         wmove(view->win, lineno, 0);
2229         if (selected) {
2230                 type = LINE_CURSOR;
2231                 wchgat(view->win, -1, 0, type, NULL);
2232         }
2234         attr = get_line_attr(type);
2235         wattrset(view->win, attr);
2237         if (opt_line_number || opt_tab_size < TABSIZE) {
2238                 static char spaces[] = "                    ";
2239                 int col_offset = 0, col = 0;
2241                 if (opt_line_number) {
2242                         unsigned long real_lineno = view->offset + lineno + 1;
2244                         if (real_lineno == 1 ||
2245                             (real_lineno % opt_num_interval) == 0) {
2246                                 wprintw(view->win, "%.*d", view->digits, real_lineno);
2248                         } else {
2249                                 waddnstr(view->win, spaces,
2250                                          MIN(view->digits, STRING_SIZE(spaces)));
2251                         }
2252                         waddstr(view->win, ": ");
2253                         col_offset = view->digits + 2;
2254                 }
2256                 while (text && col_offset + col < view->width) {
2257                         int cols_max = view->width - col_offset - col;
2258                         char *pos = text;
2259                         int cols;
2261                         if (*text == '\t') {
2262                                 text++;
2263                                 assert(sizeof(spaces) > TABSIZE);
2264                                 pos = spaces;
2265                                 cols = opt_tab_size - (col % opt_tab_size);
2267                         } else {
2268                                 text = strchr(text, '\t');
2269                                 cols = line ? text - pos : strlen(pos);
2270                         }
2272                         waddnstr(view->win, pos, MIN(cols, cols_max));
2273                         col += cols;
2274                 }
2276         } else {
2277                 int col = 0, pos = 0;
2279                 for (; pos < textlen && col < view->width; pos++, col++)
2280                         if (text[pos] == '\t')
2281                                 col += TABSIZE - (col % TABSIZE) - 1;
2283                 waddnstr(view->win, text, pos);
2284         }
2286         return TRUE;
2289 static bool
2290 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2292         char refbuf[SIZEOF_STR];
2293         char *ref = NULL;
2294         FILE *pipe;
2296         if (!string_format(refbuf, "git describe %s", commit_id))
2297                 return TRUE;
2299         pipe = popen(refbuf, "r");
2300         if (!pipe)
2301                 return TRUE;
2303         if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2304                 ref = chomp_string(ref);
2305         pclose(pipe);
2307         if (!ref || !*ref)
2308                 return TRUE;
2310         /* This is the only fatal call, since it can "corrupt" the buffer. */
2311         if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2312                 return FALSE;
2314         return TRUE;
2317 static void
2318 add_pager_refs(struct view *view, struct line *line)
2320         char buf[SIZEOF_STR];
2321         char *commit_id = line->data + STRING_SIZE("commit ");
2322         struct ref **refs;
2323         size_t bufpos = 0, refpos = 0;
2324         const char *sep = "Refs: ";
2325         bool is_tag = FALSE;
2327         assert(line->type == LINE_COMMIT);
2329         refs = get_refs(commit_id);
2330         if (!refs) {
2331                 if (view == VIEW(REQ_VIEW_DIFF))
2332                         goto try_add_describe_ref;
2333                 return;
2334         }
2336         do {
2337                 struct ref *ref = refs[refpos];
2338                 char *fmt = ref->tag ? "%s[%s]" : "%s%s";
2340                 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2341                         return;
2342                 sep = ", ";
2343                 if (ref->tag)
2344                         is_tag = TRUE;
2345         } while (refs[refpos++]->next);
2347         if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2348 try_add_describe_ref:
2349                 /* Add <tag>-g<commit_id> "fake" reference. */
2350                 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2351                         return;
2352         }
2354         if (bufpos == 0)
2355                 return;
2357         if (!realloc_lines(view, view->line_size + 1))
2358                 return;
2360         line = &view->line[view->lines];
2361         line->data = strdup(buf);
2362         if (!line->data)
2363                 return;
2365         line->type = LINE_PP_REFS;
2366         view->lines++;
2369 static bool
2370 pager_read(struct view *view, char *data)
2372         struct line *line = &view->line[view->lines];
2374         line->data = strdup(data);
2375         if (!line->data)
2376                 return FALSE;
2378         line->type = get_line_type(line->data);
2379         view->lines++;
2381         if (line->type == LINE_COMMIT &&
2382             (view == VIEW(REQ_VIEW_DIFF) ||
2383              view == VIEW(REQ_VIEW_LOG)))
2384                 add_pager_refs(view, line);
2386         return TRUE;
2389 static bool
2390 pager_enter(struct view *view, struct line *line)
2392         int split = 0;
2394         if (line->type == LINE_COMMIT &&
2395            (view == VIEW(REQ_VIEW_LOG) ||
2396             view == VIEW(REQ_VIEW_PAGER))) {
2397                 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2398                 split = 1;
2399         }
2401         /* Always scroll the view even if it was split. That way
2402          * you can use Enter to scroll through the log view and
2403          * split open each commit diff. */
2404         scroll_view(view, REQ_SCROLL_LINE_DOWN);
2406         /* FIXME: A minor workaround. Scrolling the view will call report("")
2407          * but if we are scrolling a non-current view this won't properly
2408          * update the view title. */
2409         if (split)
2410                 update_view_title(view);
2412         return TRUE;
2415 static bool
2416 pager_grep(struct view *view, struct line *line)
2418         regmatch_t pmatch;
2419         char *text = line->data;
2421         if (!*text)
2422                 return FALSE;
2424         if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2425                 return FALSE;
2427         return TRUE;
2430 static void
2431 pager_select(struct view *view, struct line *line)
2433         if (line->type == LINE_COMMIT) {
2434                 char *text = line->data;
2436                 string_copy(view->ref, text + STRING_SIZE("commit "));
2437                 string_copy(ref_commit, view->ref);
2438         }
2441 static struct view_ops pager_ops = {
2442         "line",
2443         pager_draw,
2444         pager_read,
2445         pager_enter,
2446         pager_grep,
2447         pager_select,
2448 };
2451 /*
2452  * Tree backend
2453  */
2455 /* Parse output from git-ls-tree(1):
2456  *
2457  * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
2458  * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
2459  * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
2460  * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
2461  */
2463 #define SIZEOF_TREE_ATTR \
2464         STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
2466 #define TREE_UP_FORMAT "040000 tree %s\t.."
2468 static int
2469 tree_compare_entry(enum line_type type1, char *name1,
2470                    enum line_type type2, char *name2)
2472         if (type1 != type2) {
2473                 if (type1 == LINE_TREE_DIR)
2474                         return -1;
2475                 return 1;
2476         }
2478         return strcmp(name1, name2);
2481 static bool
2482 tree_read(struct view *view, char *text)
2484         size_t textlen = strlen(text);
2485         char buf[SIZEOF_STR];
2486         unsigned long pos;
2487         enum line_type type;
2488         bool first_read = view->lines == 0;
2490         if (textlen <= SIZEOF_TREE_ATTR)
2491                 return FALSE;
2493         type = text[STRING_SIZE("100644 ")] == 't'
2494              ? LINE_TREE_DIR : LINE_TREE_FILE;
2496         if (first_read) {
2497                 /* Add path info line */
2498                 if (string_format(buf, "Directory path /%s", opt_path) &&
2499                     realloc_lines(view, view->line_size + 1) &&
2500                     pager_read(view, buf))
2501                         view->line[view->lines - 1].type = LINE_DEFAULT;
2502                 else
2503                         return FALSE;
2505                 /* Insert "link" to parent directory. */
2506                 if (*opt_path &&
2507                     string_format(buf, TREE_UP_FORMAT, view->ref) &&
2508                     realloc_lines(view, view->line_size + 1) &&
2509                     pager_read(view, buf))
2510                         view->line[view->lines - 1].type = LINE_TREE_DIR;
2511                 else if (*opt_path)
2512                         return FALSE;
2513         }
2515         /* Strip the path part ... */
2516         if (*opt_path) {
2517                 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
2518                 size_t striplen = strlen(opt_path);
2519                 char *path = text + SIZEOF_TREE_ATTR;
2521                 if (pathlen > striplen)
2522                         memmove(path, path + striplen,
2523                                 pathlen - striplen + 1);
2524         }
2526         /* Skip "Directory ..." and ".." line. */
2527         for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
2528                 struct line *line = &view->line[pos];
2529                 char *path1 = ((char *) line->data) + SIZEOF_TREE_ATTR;
2530                 char *path2 = text + SIZEOF_TREE_ATTR;
2531                 int cmp = tree_compare_entry(line->type, path1, type, path2);
2533                 if (cmp <= 0)
2534                         continue;
2536                 text = strdup(text);
2537                 if (!text)
2538                         return FALSE;
2540                 if (view->lines > pos)
2541                         memmove(&view->line[pos + 1], &view->line[pos],
2542                                 (view->lines - pos) * sizeof(*line));
2544                 line = &view->line[pos];
2545                 line->data = text;
2546                 line->type = type;
2547                 view->lines++;
2548                 return TRUE;
2549         }
2551         if (!pager_read(view, text))
2552                 return FALSE;
2554         /* Move the current line to the first tree entry. */
2555         if (first_read)
2556                 view->lineno++;
2558         view->line[view->lines - 1].type = type;
2559         return TRUE;
2562 static bool
2563 tree_enter(struct view *view, struct line *line)
2565         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2566         enum request request;
2568         switch (line->type) {
2569         case LINE_TREE_DIR:
2570                 /* Depending on whether it is a subdir or parent (updir?) link
2571                  * mangle the path buffer. */
2572                 if (line == &view->line[1] && *opt_path) {
2573                         size_t path_len = strlen(opt_path);
2574                         char *dirsep = opt_path + path_len - 1;
2576                         while (dirsep > opt_path && dirsep[-1] != '/')
2577                                 dirsep--;
2579                         dirsep[0] = 0;
2581                 } else {
2582                         size_t pathlen = strlen(opt_path);
2583                         size_t origlen = pathlen;
2584                         char *data = line->data;
2585                         char *basename = data + SIZEOF_TREE_ATTR;
2587                         if (!string_format_from(opt_path, &pathlen, "%s/", basename)) {
2588                                 opt_path[origlen] = 0;
2589                                 return TRUE;
2590                         }
2591                 }
2593                 /* Trees and subtrees share the same ID, so they are not not
2594                  * unique like blobs. */
2595                 flags |= OPEN_RELOAD;
2596                 request = REQ_VIEW_TREE;
2597                 break;
2599         case LINE_TREE_FILE:
2600                 request = REQ_VIEW_BLOB;
2601                 break;
2603         default:
2604                 return TRUE;
2605         }
2607         open_view(view, request, flags);
2609         return TRUE;
2612 static void
2613 tree_select(struct view *view, struct line *line)
2615         char *text = line->data;
2617         text += STRING_SIZE("100644 blob ");
2619         if (line->type == LINE_TREE_FILE) {
2620                 string_ncopy(ref_blob, text, 40);
2621                 /* Also update the blob view's ref, since all there must always
2622                  * be in sync. */
2623                 string_copy(VIEW(REQ_VIEW_BLOB)->ref, ref_blob);
2625         } else if (line->type != LINE_TREE_DIR) {
2626                 return;
2627         }
2629         string_ncopy(view->ref, text, 40);
2632 static struct view_ops tree_ops = {
2633         "file",
2634         pager_draw,
2635         tree_read,
2636         tree_enter,
2637         pager_grep,
2638         tree_select,
2639 };
2641 static bool
2642 blob_read(struct view *view, char *line)
2644         bool state = pager_read(view, line);
2646         if (state == TRUE)
2647                 view->line[view->lines - 1].type = LINE_DEFAULT;
2649         return state;
2652 static struct view_ops blob_ops = {
2653         "line",
2654         pager_draw,
2655         blob_read,
2656         pager_enter,
2657         pager_grep,
2658         pager_select,
2659 };
2662 /*
2663  * Main view backend
2664  */
2666 struct commit {
2667         char id[SIZEOF_REV];            /* SHA1 ID. */
2668         char title[75];                 /* First line of the commit message. */
2669         char author[75];                /* Author of the commit. */
2670         struct tm time;                 /* Date from the author ident. */
2671         struct ref **refs;              /* Repository references. */
2672         chtype graph[SIZEOF_REVGRAPH];  /* Ancestry chain graphics. */
2673         size_t graph_size;              /* The width of the graph array. */
2674 };
2676 static bool
2677 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2679         char buf[DATE_COLS + 1];
2680         struct commit *commit = line->data;
2681         enum line_type type;
2682         int col = 0;
2683         size_t timelen;
2684         size_t authorlen;
2685         int trimmed = 1;
2687         if (!*commit->author)
2688                 return FALSE;
2690         wmove(view->win, lineno, col);
2692         if (selected) {
2693                 type = LINE_CURSOR;
2694                 wattrset(view->win, get_line_attr(type));
2695                 wchgat(view->win, -1, 0, type, NULL);
2697         } else {
2698                 type = LINE_MAIN_COMMIT;
2699                 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
2700         }
2702         timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
2703         waddnstr(view->win, buf, timelen);
2704         waddstr(view->win, " ");
2706         col += DATE_COLS;
2707         wmove(view->win, lineno, col);
2708         if (type != LINE_CURSOR)
2709                 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
2711         if (opt_utf8) {
2712                 authorlen = utf8_length(commit->author, AUTHOR_COLS - 2, &col, &trimmed);
2713         } else {
2714                 authorlen = strlen(commit->author);
2715                 if (authorlen > AUTHOR_COLS - 2) {
2716                         authorlen = AUTHOR_COLS - 2;
2717                         trimmed = 1;
2718                 }
2719         }
2721         if (trimmed) {
2722                 waddnstr(view->win, commit->author, authorlen);
2723                 if (type != LINE_CURSOR)
2724                         wattrset(view->win, get_line_attr(LINE_MAIN_DELIM));
2725                 waddch(view->win, '~');
2726         } else {
2727                 waddstr(view->win, commit->author);
2728         }
2730         col += AUTHOR_COLS;
2731         if (type != LINE_CURSOR)
2732                 wattrset(view->win, A_NORMAL);
2734         if (opt_rev_graph && commit->graph_size) {
2735                 size_t i;
2737                 wmove(view->win, lineno, col);
2738                 /* Using waddch() instead of waddnstr() ensures that
2739                  * they'll be rendered correctly for the cursor line. */
2740                 for (i = 0; i < commit->graph_size; i++)
2741                         waddch(view->win, commit->graph[i]);
2743                 col += commit->graph_size + 1;
2744         }
2746         wmove(view->win, lineno, col);
2748         if (commit->refs) {
2749                 size_t i = 0;
2751                 do {
2752                         if (type == LINE_CURSOR)
2753                                 ;
2754                         else if (commit->refs[i]->tag)
2755                                 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
2756                         else
2757                                 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
2758                         waddstr(view->win, "[");
2759                         waddstr(view->win, commit->refs[i]->name);
2760                         waddstr(view->win, "]");
2761                         if (type != LINE_CURSOR)
2762                                 wattrset(view->win, A_NORMAL);
2763                         waddstr(view->win, " ");
2764                         col += strlen(commit->refs[i]->name) + STRING_SIZE("[] ");
2765                 } while (commit->refs[i++]->next);
2766         }
2768         if (type != LINE_CURSOR)
2769                 wattrset(view->win, get_line_attr(type));
2771         {
2772                 int titlelen = strlen(commit->title);
2774                 if (col + titlelen > view->width)
2775                         titlelen = view->width - col;
2777                 waddnstr(view->win, commit->title, titlelen);
2778         }
2780         return TRUE;
2783 /* Reads git log --pretty=raw output and parses it into the commit struct. */
2784 static bool
2785 main_read(struct view *view, char *line)
2787         enum line_type type = get_line_type(line);
2788         struct commit *commit = view->lines
2789                               ? view->line[view->lines - 1].data : NULL;
2791         switch (type) {
2792         case LINE_COMMIT:
2793                 commit = calloc(1, sizeof(struct commit));
2794                 if (!commit)
2795                         return FALSE;
2797                 line += STRING_SIZE("commit ");
2799                 view->line[view->lines++].data = commit;
2800                 string_copy(commit->id, line);
2801                 commit->refs = get_refs(commit->id);
2802                 commit->graph[commit->graph_size++] = ACS_LTEE;
2803                 break;
2805         case LINE_AUTHOR:
2806         {
2807                 char *ident = line + STRING_SIZE("author ");
2808                 char *end = strchr(ident, '<');
2810                 if (!commit)
2811                         break;
2813                 if (end) {
2814                         char *email = end + 1;
2816                         for (; end > ident && isspace(end[-1]); end--) ;
2818                         if (end == ident && *email) {
2819                                 ident = email;
2820                                 end = strchr(ident, '>');
2821                                 for (; end > ident && isspace(end[-1]); end--) ;
2822                         }
2823                         *end = 0;
2824                 }
2826                 /* End is NULL or ident meaning there's no author. */
2827                 if (end <= ident)
2828                         ident = "Unknown";
2830                 string_copy(commit->author, ident);
2832                 /* Parse epoch and timezone */
2833                 if (end) {
2834                         char *secs = strchr(end + 1, '>');
2835                         char *zone;
2836                         time_t time;
2838                         if (!secs || secs[1] != ' ')
2839                                 break;
2841                         secs += 2;
2842                         time = (time_t) atol(secs);
2843                         zone = strchr(secs, ' ');
2844                         if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
2845                                 long tz;
2847                                 zone++;
2848                                 tz  = ('0' - zone[1]) * 60 * 60 * 10;
2849                                 tz += ('0' - zone[2]) * 60 * 60;
2850                                 tz += ('0' - zone[3]) * 60;
2851                                 tz += ('0' - zone[4]) * 60;
2853                                 if (zone[0] == '-')
2854                                         tz = -tz;
2856                                 time -= tz;
2857                         }
2858                         gmtime_r(&time, &commit->time);
2859                 }
2860                 break;
2861         }
2862         default:
2863                 if (!commit)
2864                         break;
2866                 /* Fill in the commit title if it has not already been set. */
2867                 if (commit->title[0])
2868                         break;
2870                 /* Require titles to start with a non-space character at the
2871                  * offset used by git log. */
2872                 /* FIXME: More gracefull handling of titles; append "..." to
2873                  * shortened titles, etc. */
2874                 if (strncmp(line, "    ", 4) ||
2875                     isspace(line[4]))
2876                         break;
2878                 string_copy(commit->title, line + 4);
2879         }
2881         return TRUE;
2884 static bool
2885 main_enter(struct view *view, struct line *line)
2887         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2889         open_view(view, REQ_VIEW_DIFF, flags);
2890         return TRUE;
2893 static bool
2894 main_grep(struct view *view, struct line *line)
2896         struct commit *commit = line->data;
2897         enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
2898         char buf[DATE_COLS + 1];
2899         regmatch_t pmatch;
2901         for (state = S_TITLE; state < S_END; state++) {
2902                 char *text;
2904                 switch (state) {
2905                 case S_TITLE:   text = commit->title;   break;
2906                 case S_AUTHOR:  text = commit->author;  break;
2907                 case S_DATE:
2908                         if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
2909                                 continue;
2910                         text = buf;
2911                         break;
2913                 default:
2914                         return FALSE;
2915                 }
2917                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
2918                         return TRUE;
2919         }
2921         return FALSE;
2924 static void
2925 main_select(struct view *view, struct line *line)
2927         struct commit *commit = line->data;
2929         string_copy(view->ref, commit->id);
2930         string_copy(ref_commit, view->ref);
2933 static struct view_ops main_ops = {
2934         "commit",
2935         main_draw,
2936         main_read,
2937         main_enter,
2938         main_grep,
2939         main_select,
2940 };
2943 /*
2944  * Unicode / UTF-8 handling
2945  *
2946  * NOTE: Much of the following code for dealing with unicode is derived from
2947  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
2948  * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
2949  */
2951 /* I've (over)annotated a lot of code snippets because I am not entirely
2952  * confident that the approach taken by this small UTF-8 interface is correct.
2953  * --jonas */
2955 static inline int
2956 unicode_width(unsigned long c)
2958         if (c >= 0x1100 &&
2959            (c <= 0x115f                         /* Hangul Jamo */
2960             || c == 0x2329
2961             || c == 0x232a
2962             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
2963                                                 /* CJK ... Yi */
2964             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
2965             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
2966             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
2967             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
2968             || (c >= 0xffe0  && c <= 0xffe6)
2969             || (c >= 0x20000 && c <= 0x2fffd)
2970             || (c >= 0x30000 && c <= 0x3fffd)))
2971                 return 2;
2973         return 1;
2976 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
2977  * Illegal bytes are set one. */
2978 static const unsigned char utf8_bytes[256] = {
2979         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,
2980         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,
2981         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,
2982         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,
2983         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,
2984         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,
2985         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,
2986         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,
2987 };
2989 /* Decode UTF-8 multi-byte representation into a unicode character. */
2990 static inline unsigned long
2991 utf8_to_unicode(const char *string, size_t length)
2993         unsigned long unicode;
2995         switch (length) {
2996         case 1:
2997                 unicode  =   string[0];
2998                 break;
2999         case 2:
3000                 unicode  =  (string[0] & 0x1f) << 6;
3001                 unicode +=  (string[1] & 0x3f);
3002                 break;
3003         case 3:
3004                 unicode  =  (string[0] & 0x0f) << 12;
3005                 unicode += ((string[1] & 0x3f) << 6);
3006                 unicode +=  (string[2] & 0x3f);
3007                 break;
3008         case 4:
3009                 unicode  =  (string[0] & 0x0f) << 18;
3010                 unicode += ((string[1] & 0x3f) << 12);
3011                 unicode += ((string[2] & 0x3f) << 6);
3012                 unicode +=  (string[3] & 0x3f);
3013                 break;
3014         case 5:
3015                 unicode  =  (string[0] & 0x0f) << 24;
3016                 unicode += ((string[1] & 0x3f) << 18);
3017                 unicode += ((string[2] & 0x3f) << 12);
3018                 unicode += ((string[3] & 0x3f) << 6);
3019                 unicode +=  (string[4] & 0x3f);
3020                 break;
3021         case 6:
3022                 unicode  =  (string[0] & 0x01) << 30;
3023                 unicode += ((string[1] & 0x3f) << 24);
3024                 unicode += ((string[2] & 0x3f) << 18);
3025                 unicode += ((string[3] & 0x3f) << 12);
3026                 unicode += ((string[4] & 0x3f) << 6);
3027                 unicode +=  (string[5] & 0x3f);
3028                 break;
3029         default:
3030                 die("Invalid unicode length");
3031         }
3033         /* Invalid characters could return the special 0xfffd value but NUL
3034          * should be just as good. */
3035         return unicode > 0xffff ? 0 : unicode;
3038 /* Calculates how much of string can be shown within the given maximum width
3039  * and sets trimmed parameter to non-zero value if all of string could not be
3040  * shown.
3041  *
3042  * Additionally, adds to coloffset how many many columns to move to align with
3043  * the expected position. Takes into account how multi-byte and double-width
3044  * characters will effect the cursor position.
3045  *
3046  * Returns the number of bytes to output from string to satisfy max_width. */
3047 static size_t
3048 utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed)
3050         const char *start = string;
3051         const char *end = strchr(string, '\0');
3052         size_t mbwidth = 0;
3053         size_t width = 0;
3055         *trimmed = 0;
3057         while (string < end) {
3058                 int c = *(unsigned char *) string;
3059                 unsigned char bytes = utf8_bytes[c];
3060                 size_t ucwidth;
3061                 unsigned long unicode;
3063                 if (string + bytes > end)
3064                         break;
3066                 /* Change representation to figure out whether
3067                  * it is a single- or double-width character. */
3069                 unicode = utf8_to_unicode(string, bytes);
3070                 /* FIXME: Graceful handling of invalid unicode character. */
3071                 if (!unicode)
3072                         break;
3074                 ucwidth = unicode_width(unicode);
3075                 width  += ucwidth;
3076                 if (width > max_width) {
3077                         *trimmed = 1;
3078                         break;
3079                 }
3081                 /* The column offset collects the differences between the
3082                  * number of bytes encoding a character and the number of
3083                  * columns will be used for rendering said character.
3084                  *
3085                  * So if some character A is encoded in 2 bytes, but will be
3086                  * represented on the screen using only 1 byte this will and up
3087                  * adding 1 to the multi-byte column offset.
3088                  *
3089                  * Assumes that no double-width character can be encoding in
3090                  * less than two bytes. */
3091                 if (bytes > ucwidth)
3092                         mbwidth += bytes - ucwidth;
3094                 string  += bytes;
3095         }
3097         *coloffset += mbwidth;
3099         return string - start;
3103 /*
3104  * Status management
3105  */
3107 /* Whether or not the curses interface has been initialized. */
3108 static bool cursed = FALSE;
3110 /* The status window is used for polling keystrokes. */
3111 static WINDOW *status_win;
3113 /* Update status and title window. */
3114 static void
3115 report(const char *msg, ...)
3117         static bool empty = TRUE;
3118         struct view *view = display[current_view];
3120         if (!empty || *msg) {
3121                 va_list args;
3123                 va_start(args, msg);
3125                 werase(status_win);
3126                 wmove(status_win, 0, 0);
3127                 if (*msg) {
3128                         vwprintw(status_win, msg, args);
3129                         empty = FALSE;
3130                 } else {
3131                         empty = TRUE;
3132                 }
3133                 wrefresh(status_win);
3135                 va_end(args);
3136         }
3138         update_view_title(view);
3139         update_display_cursor(view);
3142 /* Controls when nodelay should be in effect when polling user input. */
3143 static void
3144 set_nonblocking_input(bool loading)
3146         static unsigned int loading_views;
3148         if ((loading == FALSE && loading_views-- == 1) ||
3149             (loading == TRUE  && loading_views++ == 0))
3150                 nodelay(status_win, loading);
3153 static void
3154 init_display(void)
3156         int x, y;
3158         /* Initialize the curses library */
3159         if (isatty(STDIN_FILENO)) {
3160                 cursed = !!initscr();
3161         } else {
3162                 /* Leave stdin and stdout alone when acting as a pager. */
3163                 FILE *io = fopen("/dev/tty", "r+");
3165                 if (!io)
3166                         die("Failed to open /dev/tty");
3167                 cursed = !!newterm(NULL, io, io);
3168         }
3170         if (!cursed)
3171                 die("Failed to initialize curses");
3173         nonl();         /* Tell curses not to do NL->CR/NL on output */
3174         cbreak();       /* Take input chars one at a time, no wait for \n */
3175         noecho();       /* Don't echo input */
3176         leaveok(stdscr, TRUE);
3178         if (has_colors())
3179                 init_colors();
3181         getmaxyx(stdscr, y, x);
3182         status_win = newwin(1, 0, y - 1, 0);
3183         if (!status_win)
3184                 die("Failed to create status window");
3186         /* Enable keyboard mapping */
3187         keypad(status_win, TRUE);
3188         wbkgdset(status_win, get_line_attr(LINE_STATUS));
3191 static char *
3192 read_prompt(const char *prompt)
3194         enum { READING, STOP, CANCEL } status = READING;
3195         static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
3196         int pos = 0;
3198         while (status == READING) {
3199                 struct view *view;
3200                 int i, key;
3202                 foreach_view (view, i)
3203                         update_view(view);
3205                 report("%s%.*s", prompt, pos, buf);
3206                 /* Refresh, accept single keystroke of input */
3207                 key = wgetch(status_win);
3208                 switch (key) {
3209                 case KEY_RETURN:
3210                 case KEY_ENTER:
3211                 case '\n':
3212                         status = pos ? STOP : CANCEL;
3213                         break;
3215                 case KEY_BACKSPACE:
3216                         if (pos > 0)
3217                                 pos--;
3218                         else
3219                                 status = CANCEL;
3220                         break;
3222                 case KEY_ESC:
3223                         status = CANCEL;
3224                         break;
3226                 case ERR:
3227                         break;
3229                 default:
3230                         if (pos >= sizeof(buf)) {
3231                                 report("Input string too long");
3232                                 return NULL;
3233                         }
3235                         if (isprint(key))
3236                                 buf[pos++] = (char) key;
3237                 }
3238         }
3240         if (status == CANCEL) {
3241                 /* Clear the status window */
3242                 report("");
3243                 return NULL;
3244         }
3246         buf[pos++] = 0;
3248         return buf;
3251 /*
3252  * Repository references
3253  */
3255 static struct ref *refs;
3256 static size_t refs_size;
3258 /* Id <-> ref store */
3259 static struct ref ***id_refs;
3260 static size_t id_refs_size;
3262 static struct ref **
3263 get_refs(char *id)
3265         struct ref ***tmp_id_refs;
3266         struct ref **ref_list = NULL;
3267         size_t ref_list_size = 0;
3268         size_t i;
3270         for (i = 0; i < id_refs_size; i++)
3271                 if (!strcmp(id, id_refs[i][0]->id))
3272                         return id_refs[i];
3274         tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
3275         if (!tmp_id_refs)
3276                 return NULL;
3278         id_refs = tmp_id_refs;
3280         for (i = 0; i < refs_size; i++) {
3281                 struct ref **tmp;
3283                 if (strcmp(id, refs[i].id))
3284                         continue;
3286                 tmp = realloc(ref_list, (ref_list_size + 1) * sizeof(*ref_list));
3287                 if (!tmp) {
3288                         if (ref_list)
3289                                 free(ref_list);
3290                         return NULL;
3291                 }
3293                 ref_list = tmp;
3294                 if (ref_list_size > 0)
3295                         ref_list[ref_list_size - 1]->next = 1;
3296                 ref_list[ref_list_size] = &refs[i];
3298                 /* XXX: The properties of the commit chains ensures that we can
3299                  * safely modify the shared ref. The repo references will
3300                  * always be similar for the same id. */
3301                 ref_list[ref_list_size]->next = 0;
3302                 ref_list_size++;
3303         }
3305         if (ref_list)
3306                 id_refs[id_refs_size++] = ref_list;
3308         return ref_list;
3311 static int
3312 read_ref(char *id, int idlen, char *name, int namelen)
3314         struct ref *ref;
3315         bool tag = FALSE;
3317         if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
3318                 /* Commits referenced by tags has "^{}" appended. */
3319                 if (name[namelen - 1] != '}')
3320                         return OK;
3322                 while (namelen > 0 && name[namelen] != '^')
3323                         namelen--;
3325                 tag = TRUE;
3326                 namelen -= STRING_SIZE("refs/tags/");
3327                 name    += STRING_SIZE("refs/tags/");
3329         } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
3330                 namelen -= STRING_SIZE("refs/heads/");
3331                 name    += STRING_SIZE("refs/heads/");
3333         } else if (!strcmp(name, "HEAD")) {
3334                 return OK;
3335         }
3337         refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
3338         if (!refs)
3339                 return ERR;
3341         ref = &refs[refs_size++];
3342         ref->name = malloc(namelen + 1);
3343         if (!ref->name)
3344                 return ERR;
3346         strncpy(ref->name, name, namelen);
3347         ref->name[namelen] = 0;
3348         ref->tag = tag;
3349         string_copy(ref->id, id);
3351         return OK;
3354 static int
3355 load_refs(void)
3357         const char *cmd_env = getenv("TIG_LS_REMOTE");
3358         const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
3360         return read_properties(popen(cmd, "r"), "\t", read_ref);
3363 static int
3364 read_repo_config_option(char *name, int namelen, char *value, int valuelen)
3366         if (!strcmp(name, "i18n.commitencoding"))
3367                 string_copy(opt_encoding, value);
3369         return OK;
3372 static int
3373 load_repo_config(void)
3375         return read_properties(popen("git repo-config --list", "r"),
3376                                "=", read_repo_config_option);
3379 static int
3380 read_properties(FILE *pipe, const char *separators,
3381                 int (*read_property)(char *, int, char *, int))
3383         char buffer[BUFSIZ];
3384         char *name;
3385         int state = OK;
3387         if (!pipe)
3388                 return ERR;
3390         while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
3391                 char *value;
3392                 size_t namelen;
3393                 size_t valuelen;
3395                 name = chomp_string(name);
3396                 namelen = strcspn(name, separators);
3398                 if (name[namelen]) {
3399                         name[namelen] = 0;
3400                         value = chomp_string(name + namelen + 1);
3401                         valuelen = strlen(value);
3403                 } else {
3404                         value = "";
3405                         valuelen = 0;
3406                 }
3408                 state = read_property(name, namelen, value, valuelen);
3409         }
3411         if (state != ERR && ferror(pipe))
3412                 state = ERR;
3414         pclose(pipe);
3416         return state;
3420 /*
3421  * Main
3422  */
3424 static void __NORETURN
3425 quit(int sig)
3427         /* XXX: Restore tty modes and let the OS cleanup the rest! */
3428         if (cursed)
3429                 endwin();
3430         exit(0);
3433 static void __NORETURN
3434 die(const char *err, ...)
3436         va_list args;
3438         endwin();
3440         va_start(args, err);
3441         fputs("tig: ", stderr);
3442         vfprintf(stderr, err, args);
3443         fputs("\n", stderr);
3444         va_end(args);
3446         exit(1);
3449 int
3450 main(int argc, char *argv[])
3452         struct view *view;
3453         enum request request;
3454         size_t i;
3456         signal(SIGINT, quit);
3458         if (setlocale(LC_ALL, "")) {
3459                 string_copy(opt_codeset, nl_langinfo(CODESET));
3460         }
3462         if (load_options() == ERR)
3463                 die("Failed to load user config.");
3465         /* Load the repo config file so options can be overwritten from
3466          * the command line.  */
3467         if (load_repo_config() == ERR)
3468                 die("Failed to load repo config.");
3470         if (!parse_options(argc, argv))
3471                 return 0;
3473         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
3474                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
3475                 if (opt_iconv == ICONV_NONE)
3476                         die("Failed to initialize character set conversion");
3477         }
3479         if (load_refs() == ERR)
3480                 die("Failed to load refs.");
3482         /* Require a git repository unless when running in pager mode. */
3483         if (refs_size == 0 && opt_request != REQ_VIEW_PAGER)
3484                 die("Not a git repository");
3486         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
3487                 view->cmd_env = getenv(view->cmd_env);
3489         request = opt_request;
3491         init_display();
3493         while (view_driver(display[current_view], request)) {
3494                 int key;
3495                 int i;
3497                 foreach_view (view, i)
3498                         update_view(view);
3500                 /* Refresh, accept single keystroke of input */
3501                 key = wgetch(status_win);
3503                 request = get_keybinding(display[current_view]->keymap, key);
3505                 /* Some low-level request handling. This keeps access to
3506                  * status_win restricted. */
3507                 switch (request) {
3508                 case REQ_PROMPT:
3509                 {
3510                         char *cmd = read_prompt(":");
3512                         if (cmd && string_format(opt_cmd, "git %s", cmd)) {
3513                                 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
3514                                         opt_request = REQ_VIEW_DIFF;
3515                                 } else {
3516                                         opt_request = REQ_VIEW_PAGER;
3517                                 }
3518                                 break;
3519                         }
3521                         request = REQ_NONE;
3522                         break;
3523                 }
3524                 case REQ_SEARCH:
3525                 case REQ_SEARCH_BACK:
3526                 {
3527                         const char *prompt = request == REQ_SEARCH
3528                                            ? "/" : "?";
3529                         char *search = read_prompt(prompt);
3531                         if (search)
3532                                 string_copy(opt_search, search);
3533                         else
3534                                 request = REQ_NONE;
3535                         break;
3536                 }
3537                 case REQ_SCREEN_RESIZE:
3538                 {
3539                         int height, width;
3541                         getmaxyx(stdscr, height, width);
3543                         /* Resize the status view and let the view driver take
3544                          * care of resizing the displayed views. */
3545                         wresize(status_win, 1, width);
3546                         mvwin(status_win, height - 1, 0);
3547                         wrefresh(status_win);
3548                         break;
3549                 }
3550                 default:
3551                         break;
3552                 }
3553         }
3555         quit(0);
3557         return 0;