update_view_title: use string_format_from instead of wprintw
[tig] / 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  */
13
14 #ifndef VERSION
15 #define VERSION "tig-0.4.git"
16 #endif
17
18 #ifndef DEBUG
19 #define NDEBUG
20 #endif
21
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>
32
33 #include <sys/types.h>
34 #include <regex.h>
35
36 #include <locale.h>
37 #include <langinfo.h>
38 #include <iconv.h>
39
40 #include <curses.h>
41
42 #if __GNUC__ >= 3
43 #define __NORETURN __attribute__((__noreturn__))
44 #else
45 #define __NORETURN
46 #endif
47
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);
53
54 #define ABS(x)          ((x) >= 0  ? (x) : -(x))
55 #define MIN(x, y)       ((x) < (y) ? (x) :  (y))
56
57 #define ARRAY_SIZE(x)   (sizeof(x) / sizeof(x[0]))
58 #define STRING_SIZE(x)  (sizeof(x) - 1)
59
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. */
64
65 /* This color name can be used to refer to the default term colors. */
66 #define COLOR_DEFAULT   (-1)
67
68 #define ICONV_NONE      ((iconv_t) -1)
69
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 ")
73
74 #define AUTHOR_COLS     20
75
76 /* The default interval between line numbers. */
77 #define NUMBER_INTERVAL 1
78
79 #define TABSIZE         8
80
81 #define SCALE_SPLIT_VIEW(height)        ((height) * 2 / 3)
82
83 #define TIG_LS_REMOTE \
84         "git ls-remote . 2>/dev/null"
85
86 #define TIG_DIFF_CMD \
87         "git show --root --patch-with-stat --find-copies-harder -B -C %s 2>/dev/null"
88
89 #define TIG_LOG_CMD     \
90         "git log --cc --stat -n100 %s 2>/dev/null"
91
92 #define TIG_MAIN_CMD \
93         "git log --topo-order --pretty=raw %s 2>/dev/null"
94
95 #define TIG_TREE_CMD    \
96         "git ls-tree %s %s"
97
98 #define TIG_BLOB_CMD    \
99         "git cat-file blob %s"
100
101 /* XXX: Needs to be defined to the empty string. */
102 #define TIG_HELP_CMD    ""
103 #define TIG_PAGER_CMD   ""
104
105 /* Some ascii-shorthands fitted into the ncurses namespace. */
106 #define KEY_TAB         '\t'
107 #define KEY_RETURN      '\r'
108 #define KEY_ESC         27
109
110
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 };
117
118 static struct ref **get_refs(char *id);
119
120 struct int_map {
121         const char *name;
122         int namelen;
123         int value;
124 };
125
126 static int
127 set_from_int_map(struct int_map *map, size_t map_size,
128                  int *value, const char *name, int namelen)
129 {
130
131         int i;
132
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                 }
139
140         return ERR;
141 }
142
143
144 /*
145  * String helpers
146  */
147
148 static inline void
149 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
150 {
151         if (srclen > dstlen - 1)
152                 srclen = dstlen - 1;
153
154         strncpy(dst, src, srclen);
155         dst[srclen] = 0;
156 }
157
158 /* Shorthands for safely copying into a fixed buffer. */
159
160 #define string_copy(dst, src) \
161         string_ncopy_do(dst, sizeof(dst), src, sizeof(dst))
162
163 #define string_ncopy(dst, src, srclen) \
164         string_ncopy_do(dst, sizeof(dst), src, srclen)
165
166 static char *
167 chomp_string(char *name)
168 {
169         int namelen;
170
171         while (isspace(*name))
172                 name++;
173
174         namelen = strlen(name) - 1;
175         while (namelen > 0 && isspace(name[namelen]))
176                 name[namelen--] = 0;
177
178         return name;
179 }
180
181 static bool
182 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
183 {
184         va_list args;
185         size_t pos = bufpos ? *bufpos : 0;
186
187         va_start(args, fmt);
188         pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
189         va_end(args);
190
191         if (bufpos)
192                 *bufpos = pos;
193
194         return pos >= bufsize ? FALSE : TRUE;
195 }
196
197 #define string_format(buf, fmt, args...) \
198         string_nformat(buf, sizeof(buf), NULL, fmt, args)
199
200 #define string_format_from(buf, from, fmt, args...) \
201         string_nformat(buf, sizeof(buf), from, fmt, args)
202
203 static int
204 string_enum_compare(const char *str1, const char *str2, int len)
205 {
206         size_t i;
207
208 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
209
210         /* Diff-Header == DIFF_HEADER */
211         for (i = 0; i < len; i++) {
212                 if (toupper(str1[i]) == toupper(str2[i]))
213                         continue;
214
215                 if (string_enum_sep(str1[i]) &&
216                     string_enum_sep(str2[i]))
217                         continue;
218
219                 return str1[i] - str2[i];
220         }
221
222         return 0;
223 }
224
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  */
241
242 static size_t
243 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
244 {
245         char c;
246
247 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
248
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('\'');
261
262         return bufsize;
263 }
264
265
266 /*
267  * User requests
268  */
269
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")
318
319
320 /* User action requests. */
321 enum request {
322 #define REQ_GROUP(help)
323 #define REQ_(req, help) REQ_##req
324
325         /* Offset all requests to avoid conflicts with ncurses getch values. */
326         REQ_OFFSET = KEY_MAX + 1,
327         REQ_INFO,
328         REQ_UNKNOWN,
329
330 #undef  REQ_GROUP
331 #undef  REQ_
332 };
333
334 struct request_info {
335         enum request request;
336         char *name;
337         int namelen;
338         char *help;
339 };
340
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 };
348
349 static enum request
350 get_request(const char *name)
351 {
352         int namelen = strlen(name);
353         int i;
354
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;
359
360         return REQ_UNKNOWN;
361 }
362
363
364 /*
365  * Options
366  */
367
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";
386
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]      = "";
401
402 enum option_type {
403         OPT_NONE,
404         OPT_INT,
405 };
406
407 static bool
408 check_option(char *opt, char short_name, char *name, enum option_type type, ...)
409 {
410         va_list args;
411         char *value = "";
412         int *number;
413
414         if (opt[0] != '-')
415                 return FALSE;
416
417         if (opt[1] == '-') {
418                 int namelen = strlen(name);
419
420                 opt += 2;
421
422                 if (strncmp(opt, name, namelen))
423                         return FALSE;
424
425                 if (opt[namelen] == '=')
426                         value = opt + namelen + 1;
427
428         } else {
429                 if (!short_name || opt[1] != short_name)
430                         return FALSE;
431                 value = opt + 2;
432         }
433
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);
441
442         return TRUE;
443 }
444
445 /* Returns the index of log or diff command or -1 to exit. */
446 static bool
447 parse_options(int argc, char *argv[])
448 {
449         int i;
450
451         for (i = 1; i < argc; i++) {
452                 char *opt = argv[i];
453
454                 if (!strcmp(opt, "-l")) {
455                         opt_request = REQ_VIEW_LOG;
456                         continue;
457                 }
458
459                 if (!strcmp(opt, "-d")) {
460                         opt_request = REQ_VIEW_DIFF;
461                         continue;
462                 }
463
464                 if (check_option(opt, 'n', "line-number", OPT_INT, &opt_num_interval)) {
465                         opt_line_number = TRUE;
466                         continue;
467                 }
468
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                 }
473
474                 if (check_option(opt, 'v', "version", OPT_NONE)) {
475                         printf("tig version %s\n", VERSION);
476                         return FALSE;
477                 }
478
479                 if (check_option(opt, 'h', "help", OPT_NONE)) {
480                         printf(usage);
481                         return FALSE;
482                 }
483
484                 if (!strcmp(opt, "--")) {
485                         i++;
486                         break;
487                 }
488
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                 }
496
497                 if (opt[0] && opt[0] != '-')
498                         break;
499
500                 die("unknown option '%s'\n\n%s", opt, usage);
501         }
502
503         if (!isatty(STDIN_FILENO)) {
504                 opt_request = REQ_VIEW_PAGER;
505                 opt_pipe = stdin;
506
507         } else if (i < argc) {
508                 size_t buf_size;
509
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);
517
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                 }
522
523                 if (buf_size >= sizeof(opt_cmd))
524                         die("command too long");
525
526                 opt_cmd[buf_size] = 0;
527
528         }
529
530         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
531                 opt_utf8 = FALSE;
532
533         return TRUE;
534 }
535
536
537 /*
538  * Line-oriented content detection.
539  */
540
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)
583
584 enum line_type {
585 #define LINE(type, line, fg, bg, attr) \
586         LINE_##type
587         LINE_INFO
588 #undef  LINE
589 };
590
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 };
598
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 };
605
606 static enum line_type
607 get_line_type(char *line)
608 {
609         int linelen = strlen(line);
610         enum line_type type;
611
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;
617
618         return LINE_DEFAULT;
619 }
620
621 static inline int
622 get_line_attr(enum line_type type)
623 {
624         assert(type < ARRAY_SIZE(line_info));
625         return COLOR_PAIR(type) | line_info[type].attr;
626 }
627
628 static struct line_info *
629 get_line_info(char *name, int namelen)
630 {
631         enum line_type type;
632
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];
637
638         return NULL;
639 }
640
641 static void
642 init_colors(void)
643 {
644         int default_bg = COLOR_BLACK;
645         int default_fg = COLOR_WHITE;
646         enum line_type type;
647
648         start_color();
649
650         if (use_default_colors() != ERR) {
651                 default_bg = -1;
652                 default_fg = -1;
653         }
654
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;
659
660                 init_pair(type, fg, bg);
661         }
662 }
663
664 struct line {
665         enum line_type type;
666
667         /* State flags */
668         unsigned int selected:1;
669
670         void *data;             /* User data */
671 };
672
673
674 /*
675  * Keys
676  */
677
678 struct keybinding {
679         int alias;
680         enum request request;
681         struct keybinding *next;
682 };
683
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 },
693
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 },
700
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 },
711
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 },
717
718         /* Searching */
719         { '/',          REQ_SEARCH },
720         { '?',          REQ_SEARCH_BACK },
721         { 'n',          REQ_FIND_NEXT },
722         { 'N',          REQ_FIND_PREV },
723
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 },
732
733         /* wgetch() with nodelay() enabled returns ERR when there's no input. */
734         { ERR,          REQ_NONE },
735
736         /* Using the ncurses SIGWINCH handler. */
737         { KEY_RESIZE,   REQ_SCREEN_RESIZE },
738 };
739
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) \
749
750 enum keymap {
751 #define KEYMAP_(name) KEYMAP_##name
752         KEYMAP_INFO
753 #undef  KEYMAP_
754 };
755
756 static struct int_map keymap_table[] = {
757 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
758         KEYMAP_INFO
759 #undef  KEYMAP_
760 };
761
762 #define set_keymap(map, name) \
763         set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
764
765 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
766
767 static void
768 add_keybinding(enum keymap keymap, enum request request, int key)
769 {
770         struct keybinding *keybinding;
771
772         keybinding = calloc(1, sizeof(*keybinding));
773         if (!keybinding)
774                 die("Failed to allocate keybinding");
775
776         keybinding->alias = key;
777         keybinding->request = request;
778         keybinding->next = keybindings[keymap];
779         keybindings[keymap] = keybinding;
780 }
781
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)
786 {
787         struct keybinding *kbd;
788         int i;
789
790         for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
791                 if (kbd->alias == key)
792                         return kbd->request;
793
794         for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
795                 if (kbd->alias == key)
796                         return kbd->request;
797
798         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
799                 if (default_keybindings[i].alias == key)
800                         return default_keybindings[i].request;
801
802         return (enum request) key;
803 }
804
805
806 struct key {
807         char *name;
808         int value;
809 };
810
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 };
841
842 static int
843 get_key_value(const char *name)
844 {
845         int i;
846
847         for (i = 0; i < ARRAY_SIZE(key_table); i++)
848                 if (!strcasecmp(key_table[i].name, name))
849                         return key_table[i].value;
850
851         if (strlen(name) == 1 && isprint(*name))
852                 return (int) *name;
853
854         return ERR;
855 }
856
857 static char *
858 get_key(enum request request)
859 {
860         static char buf[BUFSIZ];
861         static char key_char[] = "'X'";
862         size_t pos = 0;
863         char *sep = "    ";
864         int i;
865
866         buf[pos] = 0;
867
868         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
869                 struct keybinding *keybinding = &default_keybindings[i];
870                 char *seq = NULL;
871                 int key;
872
873                 if (keybinding->request != request)
874                         continue;
875
876                 for (key = 0; key < ARRAY_SIZE(key_table); key++)
877                         if (key_table[key].value == keybinding->alias)
878                                 seq = key_table[key].name;
879
880                 if (seq == NULL &&
881                     keybinding->alias < 127 &&
882                     isprint(keybinding->alias)) {
883                         key_char[1] = (char) keybinding->alias;
884                         seq = key_char;
885                 }
886
887                 if (!seq)
888                         seq = "'?'";
889
890                 if (!string_format_from(buf, &pos, "%s%s", sep, seq))
891                         return "Too many keybindings!";
892                 sep = ", ";
893         }
894
895         return buf;
896 }
897
898
899 /*
900  * User config file handling.
901  */
902
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 };
915
916 #define set_color(color, name) \
917         set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
918
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 };
929
930 #define set_attribute(attr, name) \
931         set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
932
933 static int   config_lineno;
934 static bool  config_errors;
935 static char *config_msg;
936
937 /* Wants: object fgcolor bgcolor [attr] */
938 static int
939 option_color_command(int argc, char *argv[])
940 {
941         struct line_info *info;
942
943         if (argc != 3 && argc != 4) {
944                 config_msg = "Wrong number of arguments given to color command";
945                 return ERR;
946         }
947
948         info = get_line_info(argv[0], strlen(argv[0]));
949         if (!info) {
950                 config_msg = "Unknown color name";
951                 return ERR;
952         }
953
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         }
959
960         if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
961                 config_msg = "Unknown attribute";
962                 return ERR;
963         }
964
965         return OK;
966 }
967
968 /* Wants: name = value */
969 static int
970 option_set_command(int argc, char *argv[])
971 {
972         if (argc != 3) {
973                 config_msg = "Wrong number of arguments given to set command";
974                 return ERR;
975         }
976
977         if (strcmp(argv[1], "=")) {
978                 config_msg = "No value assigned";
979                 return ERR;
980         }
981
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         }
988
989         if (!strcmp(argv[0], "line-number-interval")) {
990                 opt_num_interval = atoi(argv[2]);
991                 return OK;
992         }
993
994         if (!strcmp(argv[0], "tab-size")) {
995                 opt_tab_size = atoi(argv[2]);
996                 return OK;
997         }
998
999         if (!strcmp(argv[0], "commit-encoding")) {
1000                 char *arg = argv[2];
1001                 int delimiter = *arg;
1002                 int i;
1003
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         }
1017
1018         config_msg = "Unknown variable name";
1019         return ERR;
1020 }
1021
1022 /* Wants: mode request key */
1023 static int
1024 option_bind_command(int argc, char *argv[])
1025 {
1026         enum request request;
1027         int keymap;
1028         int key;
1029
1030         if (argc != 3) {
1031                 config_msg = "Wrong number of arguments given to bind command";
1032                 return ERR;
1033         }
1034
1035         if (set_keymap(&keymap, argv[0]) == ERR) {
1036                 config_msg = "Unknown key map";
1037                 return ERR;
1038         }
1039
1040         key = get_key_value(argv[1]);
1041         if (key == ERR) {
1042                 config_msg = "Unknown key";
1043                 return ERR;
1044         }
1045
1046         request = get_request(argv[2]);
1047         if (request == REQ_UNKNOWN) {
1048                 config_msg = "Unknown request name";
1049                 return ERR;
1050         }
1051
1052         add_keybinding(keymap, request, key);
1053
1054         return OK;
1055 }
1056
1057 static int
1058 set_option(char *opt, char *value)
1059 {
1060         char *argv[16];
1061         int valuelen;
1062         int argc = 0;
1063
1064         /* Tokenize */
1065         while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1066                 argv[argc++] = value;
1067
1068                 value += valuelen;
1069                 if (!*value)
1070                         break;
1071
1072                 *value++ = 0;
1073                 while (isspace(*value))
1074                         value++;
1075         }
1076
1077         if (!strcmp(opt, "color"))
1078                 return option_color_command(argc, argv);
1079
1080         if (!strcmp(opt, "set"))
1081                 return option_set_command(argc, argv);
1082
1083         if (!strcmp(opt, "bind"))
1084                 return option_bind_command(argc, argv);
1085
1086         config_msg = "Unknown option command";
1087         return ERR;
1088 }
1089
1090 static int
1091 read_option(char *opt, int optlen, char *value, int valuelen)
1092 {
1093         int status = OK;
1094
1095         config_lineno++;
1096         config_msg = "Internal error";
1097
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;
1103
1104         if (opt[optlen] != 0) {
1105                 config_msg = "No option value";
1106                 status = ERR;
1107
1108         }  else {
1109                 /* Look for comment endings in the value. */
1110                 int len = strcspn(value, "#");
1111
1112                 if (len < valuelen) {
1113                         valuelen = len;
1114                         value[valuelen] = 0;
1115                 }
1116
1117                 status = set_option(opt, value);
1118         }
1119
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         }
1125
1126         /* Always keep going if errors are encountered. */
1127         return OK;
1128 }
1129
1130 static int
1131 load_options(void)
1132 {
1133         char *home = getenv("HOME");
1134         char buf[SIZEOF_STR];
1135         FILE *file;
1136
1137         config_lineno = 0;
1138         config_errors = FALSE;
1139
1140         if (!home || !string_format(buf, "%s/.tigrc", home))
1141                 return ERR;
1142
1143         /* It's ok that the file doesn't exist. */
1144         file = fopen(buf, "r");
1145         if (!file)
1146                 return OK;
1147
1148         if (read_properties(file, " \t", read_option) == ERR ||
1149             config_errors == TRUE)
1150                 fprintf(stderr, "Errors while loading %s.\n", buf);
1151
1152         return OK;
1153 }
1154
1155
1156 /*
1157  * The viewer
1158  */
1159
1160 struct view;
1161 struct view_ops;
1162
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;
1166
1167 #define foreach_displayed_view(view, i) \
1168         for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1169
1170 #define displayed_views()       (display[1] != NULL ? 2 : 1)
1171
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";
1176
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} */
1182
1183         struct view_ops *ops;   /* View operations */
1184
1185         enum keymap keymap;     /* What keymap does this view have */
1186
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. */
1190
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 */
1194
1195         /* Navigation */
1196         unsigned long offset;   /* Offset of the window top */
1197         unsigned long lineno;   /* Current line number */
1198
1199         /* Searching */
1200         char grep[SIZEOF_STR];  /* Search string */
1201         regex_t *regex;         /* Pre-compiled regex */
1202
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;
1206
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. */
1212
1213         /* Loading */
1214         FILE *pipe;
1215         time_t start_time;
1216 };
1217
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 };
1232
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;
1237
1238 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1239         { name, cmd, #env, ref, ops, map}
1240
1241 #define VIEW_(id, name, ops, ref) \
1242         VIEW_STR(name, TIG_##id##_CMD,  TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1243
1244
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 };
1254
1255 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1256
1257 #define foreach_view(view, i) \
1258         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1259
1260 #define view_is_displayed(view) \
1261         (view == display[0] || view == display[1])
1262
1263 static bool
1264 draw_view_line(struct view *view, unsigned int lineno)
1265 {
1266         struct line *line;
1267         bool selected = (view->offset + lineno == view->lineno);
1268         bool draw_ok;
1269
1270         assert(view_is_displayed(view));
1271
1272         if (view->offset + lineno >= view->lines)
1273                 return FALSE;
1274
1275         line = &view->line[view->offset + lineno];
1276
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         }
1285
1286         scrollok(view->win, FALSE);
1287         draw_ok = view->ops->draw(view, line, lineno, selected);
1288         scrollok(view->win, TRUE);
1289
1290         return draw_ok;
1291 }
1292
1293 static void
1294 redraw_view_from(struct view *view, int lineno)
1295 {
1296         assert(0 <= lineno && lineno < view->height);
1297
1298         for (; lineno < view->height; lineno++) {
1299                 if (!draw_view_line(view, lineno))
1300                         break;
1301         }
1302
1303         redrawwin(view->win);
1304         wrefresh(view->win);
1305 }
1306
1307 static void
1308 redraw_view(struct view *view)
1309 {
1310         wclear(view->win);
1311         redraw_view_from(view, 0);
1312 }
1313
1314
1315 static void
1316 update_view_title(struct view *view)
1317 {
1318         char buf[SIZEOF_STR];
1319         size_t bufpos = 0;
1320
1321         assert(view_is_displayed(view));
1322
1323         string_format_from(buf, &bufpos, "[%s]", view->name);
1324         if (*view->ref)
1325                 string_format_from(buf, &bufpos, " %s", view->ref);
1326
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;
1332
1333                 string_format_from(buf, &bufpos, " - %s %d of %d (%d%%)",
1334                                    view->ops->type,
1335                                    view->lineno + 1,
1336                                    view->lines,
1337                                    lines);
1338         }
1339
1340         if (view->pipe) {
1341                 time_t secs = time(NULL) - view->start_time;
1342
1343                 /* Three git seconds are a long time ... */
1344                 if (secs > 2)
1345                         string_format_from(buf, &bufpos, " %lds", secs);
1346         }
1347
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));
1352
1353         werase(view->title);
1354         mvwaddnstr(view->title, 0, 0, buf, bufpos);
1355         wmove(view->title, 0, view->width - 1);
1356         wrefresh(view->title);
1357 }
1358
1359 static void
1360 resize_display(void)
1361 {
1362         int offset, i;
1363         struct view *base = display[0];
1364         struct view *view = display[1] ? display[1] : display[0];
1365
1366         /* Setup window dimensions */
1367
1368         getmaxyx(stdscr, base->height, base->width);
1369
1370         /* Make room for the status window. */
1371         base->height -= 1;
1372
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;
1378
1379                 /* Make room for the title bar. */
1380                 view->height -= 1;
1381         }
1382
1383         /* Make room for the title bar. */
1384         base->height -= 1;
1385
1386         offset = 0;
1387
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);
1393
1394                         scrollok(view->win, TRUE);
1395
1396                         view->title = newwin(1, 0, offset + view->height, 0);
1397                         if (!view->title)
1398                                 die("Failed to create title window");
1399
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                 }
1405
1406                 offset += view->height + 1;
1407         }
1408 }
1409
1410 static void
1411 redraw_display(void)
1412 {
1413         struct view *view;
1414         int i;
1415
1416         foreach_displayed_view (view, i) {
1417                 redraw_view(view);
1418                 update_view_title(view);
1419         }
1420 }
1421
1422 static void
1423 update_display_cursor(struct view *view)
1424 {
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         }
1433 }
1434
1435 /*
1436  * Navigation
1437  */
1438
1439 /* Scrolling backend */
1440 static void
1441 do_scroll_view(struct view *view, int lines)
1442 {
1443         bool redraw_current_line = FALSE;
1444
1445         /* The rendering expects the new offset. */
1446         view->offset += lines;
1447
1448         assert(0 <= view->offset && view->offset < view->lines);
1449         assert(lines);
1450
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         }
1459
1460         assert(view->offset <= view->lineno && view->lineno < view->lines);
1461
1462         /* Redraw the whole screen if scrolling is pointless. */
1463         if (view->height < ABS(lines)) {
1464                 redraw_view(view);
1465
1466         } else {
1467                 int line = lines > 0 ? view->height - lines : 0;
1468                 int end = line + ABS(lines);
1469
1470                 wscrl(view->win, lines);
1471
1472                 for (; line < end; line++) {
1473                         if (!draw_view_line(view, line))
1474                                 break;
1475                 }
1476
1477                 if (redraw_current_line)
1478                         draw_view_line(view, view->lineno - view->offset);
1479         }
1480
1481         redrawwin(view->win);
1482         wrefresh(view->win);
1483         report("");
1484 }
1485
1486 /* Scroll frontend */
1487 static void
1488 scroll_view(struct view *view, enum request request)
1489 {
1490         int lines = 1;
1491
1492         assert(view_is_displayed(view));
1493
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;
1500
1501                 if (lines == 0 || view->offset + view->height >= view->lines) {
1502                         report("Cannot scroll beyond the last line");
1503                         return;
1504                 }
1505                 break;
1506
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;
1512
1513                 if (lines == 0) {
1514                         report("Cannot scroll beyond the first line");
1515                         return;
1516                 }
1517
1518                 lines = -lines;
1519                 break;
1520
1521         default:
1522                 die("request %d not handled in switch", request);
1523         }
1524
1525         do_scroll_view(view, lines);
1526 }
1527
1528 /* Cursor moving */
1529 static void
1530 move_view(struct view *view, enum request request)
1531 {
1532         int scroll_steps = 0;
1533         int steps;
1534
1535         switch (request) {
1536         case REQ_MOVE_FIRST_LINE:
1537                 steps = -view->lineno;
1538                 break;
1539
1540         case REQ_MOVE_LAST_LINE:
1541                 steps = view->lines - view->lineno - 1;
1542                 break;
1543
1544         case REQ_MOVE_PAGE_UP:
1545                 steps = view->height > view->lineno
1546                       ? -view->lineno : -view->height;
1547                 break;
1548
1549         case REQ_MOVE_PAGE_DOWN:
1550                 steps = view->lineno + view->height >= view->lines
1551                       ? view->lines - view->lineno - 1 : view->height;
1552                 break;
1553
1554         case REQ_MOVE_UP:
1555                 steps = -1;
1556                 break;
1557
1558         case REQ_MOVE_DOWN:
1559                 steps = 1;
1560                 break;
1561
1562         default:
1563                 die("request %d not handled in switch", request);
1564         }
1565
1566         if (steps <= 0 && view->lineno == 0) {
1567                 report("Cannot move beyond the first line");
1568                 return;
1569
1570         } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1571                 report("Cannot move beyond the last line");
1572                 return;
1573         }
1574
1575         /* Move the current line */
1576         view->lineno += steps;
1577         assert(0 <= view->lineno && view->lineno < view->lines);
1578
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;
1585
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         }
1595
1596         if (!view_is_displayed(view)) {
1597                 view->offset += steps;
1598                 view->ops->select(view, &view->line[view->lineno]);
1599                 return;
1600         }
1601
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);
1605
1606         if (scroll_steps) {
1607                 do_scroll_view(view, scroll_steps);
1608                 return;
1609         }
1610
1611         /* Draw the current line */
1612         draw_view_line(view, view->lineno - view->offset);
1613
1614         redrawwin(view->win);
1615         wrefresh(view->win);
1616         report("");
1617 }
1618
1619
1620 /*
1621  * Searching
1622  */
1623
1624 static void search_view(struct view *view, enum request request);
1625
1626 static bool
1627 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1628 {
1629         assert(view_is_displayed(view));
1630
1631         if (!view->ops->grep(view, line))
1632                 return FALSE;
1633
1634         if (lineno - view->offset >= view->height) {
1635                 view->offset = lineno;
1636                 view->lineno = lineno;
1637                 redraw_view(view);
1638
1639         } else {
1640                 unsigned long old_lineno = view->lineno - view->offset;
1641
1642                 view->lineno = lineno;
1643                 draw_view_line(view, old_lineno);
1644
1645                 draw_view_line(view, view->lineno - view->offset);
1646                 redrawwin(view->win);
1647                 wrefresh(view->win);
1648         }
1649
1650         report("Line %ld matches '%s'", lineno + 1, view->grep);
1651         return TRUE;
1652 }
1653
1654 static void
1655 find_next(struct view *view, enum request request)
1656 {
1657         unsigned long lineno = view->lineno;
1658         int direction;
1659
1660         if (!*view->grep) {
1661                 if (!*opt_search)
1662                         report("No previous search");
1663                 else
1664                         search_view(view, request);
1665                 return;
1666         }
1667
1668         switch (request) {
1669         case REQ_SEARCH:
1670         case REQ_FIND_NEXT:
1671                 direction = 1;
1672                 break;
1673
1674         case REQ_SEARCH_BACK:
1675         case REQ_FIND_PREV:
1676                 direction = -1;
1677                 break;
1678
1679         default:
1680                 return;
1681         }
1682
1683         if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1684                 lineno += direction;
1685
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];
1690
1691                 if (find_next_line(view, lineno, line))
1692                         return;
1693         }
1694
1695         report("No match found for '%s'", view->grep);
1696 }
1697
1698 static void
1699 search_view(struct view *view, enum request request)
1700 {
1701         int regex_err;
1702
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         }
1711
1712         regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
1713         if (regex_err != 0) {
1714                 char buf[SIZEOF_STR] = "unknown error";
1715
1716                 regerror(regex_err, view->regex, buf, sizeof(buf));
1717                 report("Search failed: %s", buf);
1718                 return;
1719         }
1720
1721         string_copy(view->grep, opt_search);
1722
1723         find_next(view, request);
1724 }
1725
1726 /*
1727  * Incremental updating
1728  */
1729
1730 static void
1731 end_update(struct view *view)
1732 {
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;
1741 }
1742
1743 static bool
1744 begin_update(struct view *view)
1745 {
1746         const char *id = view->id;
1747
1748         if (view->pipe)
1749                 end_update(view);
1750
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;
1757
1758         } else if (view == VIEW(REQ_VIEW_TREE)) {
1759                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1760
1761                 if (strcmp(view->vid, view->id))
1762                         opt_path[0] = 0;
1763
1764                 if (!string_format(view->cmd, format, id, opt_path))
1765                         return FALSE;
1766
1767         } else {
1768                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1769
1770                 if (!string_format(view->cmd, format, id, id, id, id, id))
1771                         return FALSE;
1772         }
1773
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         }
1781
1782         if (!view->pipe)
1783                 return FALSE;
1784
1785         set_nonblocking_input(TRUE);
1786
1787         view->offset = 0;
1788         view->lines  = 0;
1789         view->lineno = 0;
1790         string_copy(view->vid, id);
1791
1792         if (view->line) {
1793                 int i;
1794
1795                 for (i = 0; i < view->lines; i++)
1796                         if (view->line[i].data)
1797                                 free(view->line[i].data);
1798
1799                 free(view->line);
1800                 view->line = NULL;
1801         }
1802
1803         view->start_time = time(NULL);
1804
1805         return TRUE;
1806 }
1807
1808 static struct line *
1809 realloc_lines(struct view *view, size_t line_size)
1810 {
1811         struct line *tmp = realloc(view->line, sizeof(*view->line) * line_size);
1812
1813         if (!tmp)
1814                 return NULL;
1815
1816         view->line = tmp;
1817         view->line_size = line_size;
1818         return view->line;
1819 }
1820
1821 static bool
1822 update_view(struct view *view)
1823 {
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;
1832
1833         if (!view->pipe)
1834                 return TRUE;
1835
1836         /* Only redraw if lines are visible. */
1837         if (view->offset + view->height >= view->lines)
1838                 redraw_from = view->lines - view->offset;
1839
1840         /* FIXME: This is probably not perfect for backgrounded views. */
1841         if (!realloc_lines(view, view->lines + lines))
1842                 goto alloc_error;
1843
1844         while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
1845                 size_t linelen = strlen(line);
1846
1847                 if (linelen)
1848                         line[linelen - 1] = 0;
1849
1850                 if (opt_iconv != ICONV_NONE) {
1851                         char *inbuf = line;
1852                         size_t inlen = linelen;
1853
1854                         char *outbuf = out_buffer;
1855                         size_t outlen = sizeof(out_buffer);
1856
1857                         size_t ret;
1858
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                 }
1865
1866                 if (!view->ops->read(view, line))
1867                         goto alloc_error;
1868
1869                 if (lines-- == 1)
1870                         break;
1871         }
1872
1873         {
1874                 int digits;
1875
1876                 lines = view->lines;
1877                 for (digits = 0; lines; digits++)
1878                         lines /= 10;
1879
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         }
1886
1887         if (!view_is_displayed(view))
1888                 goto check_pipe;
1889
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);
1894
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--;
1901
1902                 /* Incrementally draw avoids flickering. */
1903                 redraw_view_from(view, redraw_from);
1904         }
1905
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);
1909
1910 check_pipe:
1911         if (ferror(view->pipe)) {
1912                 report("Failed to read: %s", strerror(errno));
1913                 goto end;
1914
1915         } else if (feof(view->pipe)) {
1916                 report("");
1917                 goto end;
1918         }
1919
1920         return TRUE;
1921
1922 alloc_error:
1923         report("Allocation failure");
1924
1925 end:
1926         end_update(view);
1927         return FALSE;
1928 }
1929
1930
1931 /*
1932  * View opening
1933  */
1934
1935 static void open_help_view(struct view *view)
1936 {
1937         char buf[BUFSIZ];
1938         int lines = ARRAY_SIZE(req_info) + 2;
1939         int i;
1940
1941         if (view->lines > 0)
1942                 return;
1943
1944         for (i = 0; i < ARRAY_SIZE(req_info); i++)
1945                 if (!req_info[i].request)
1946                         lines++;
1947
1948         view->line = calloc(lines, sizeof(*view->line));
1949         if (!view->line) {
1950                 report("Allocation failure");
1951                 return;
1952         }
1953
1954         view->ops->read(view, "Quick reference for tig keybindings:");
1955
1956         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
1957                 char *key;
1958
1959                 if (!req_info[i].request) {
1960                         view->ops->read(view, "");
1961                         view->ops->read(view, req_info[i].help);
1962                         continue;
1963                 }
1964
1965                 key = get_key(req_info[i].request);
1966                 if (!string_format(buf, "%-25s %s", key, req_info[i].help))
1967                         continue;
1968
1969                 view->ops->read(view, buf);
1970         }
1971 }
1972
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 };
1979
1980 static void
1981 open_view(struct view *prev, enum request request, enum open_flags flags)
1982 {
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];
1989
1990         if (view == prev && nviews == 1 && !reload) {
1991                 report("Already in %s view", view->name);
1992                 return;
1993         }
1994
1995         if (view == VIEW(REQ_VIEW_HELP)) {
1996                 open_help_view(view);
1997
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         }
2003
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         }
2014
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();
2020
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;
2024
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         }
2029
2030         if (prev && view != prev) {
2031                 if (split && !backgrounded) {
2032                         /* "Blur" the previous view. */
2033                         update_view_title(prev);
2034                 }
2035
2036                 view->parent = prev;
2037         }
2038
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         }
2048
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);
2053 }
2054
2055
2056 /*
2057  * User request switch noodle
2058  */
2059
2060 static int
2061 view_driver(struct view *view, enum request request)
2062 {
2063         int i;
2064
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;
2074
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;
2081
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;
2096
2097         case REQ_NEXT:
2098         case REQ_PREVIOUS:
2099                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2100
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 */
2114
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]);
2121
2122         case REQ_VIEW_NEXT:
2123         {
2124                 int nviews = displayed_views();
2125                 int next_view = (current_view + 1) % nviews;
2126
2127                 if (next_view == current_view) {
2128                         report("Only one view is displayed");
2129                         break;
2130                 }
2131
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;
2142
2143         case REQ_TOGGLE_REV_GRAPH:
2144                 opt_rev_graph = !opt_rev_graph;
2145                 redraw_display();
2146                 break;
2147
2148         case REQ_PROMPT:
2149                 /* Always reload^Wrerun commands from the prompt. */
2150                 open_view(view, opt_request, OPEN_RELOAD);
2151                 break;
2152
2153         case REQ_SEARCH:
2154         case REQ_SEARCH_BACK:
2155                 search_view(view, request);
2156                 break;
2157
2158         case REQ_FIND_NEXT:
2159         case REQ_FIND_PREV:
2160                 find_next(view, request);
2161                 break;
2162
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;
2171
2172         case REQ_SHOW_VERSION:
2173                 report("%s (built %s)", VERSION, __DATE__);
2174                 return TRUE;
2175
2176         case REQ_SCREEN_RESIZE:
2177                 resize_display();
2178                 /* Fall-through */
2179         case REQ_SCREEN_REDRAW:
2180                 redraw_display();
2181                 break;
2182
2183         case REQ_NONE:
2184                 doupdate();
2185                 return TRUE;
2186
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;
2204
2205         default:
2206                 /* An unknown key will show most commonly used commands. */
2207                 report("Unknown key, press 'h' for help");
2208                 return TRUE;
2209         }
2210
2211         return TRUE;
2212 }
2213
2214
2215 /*
2216  * Pager backend
2217  */
2218
2219 static bool
2220 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2221 {
2222         char *text = line->data;
2223         enum line_type type = line->type;
2224         int textlen = strlen(text);
2225         int attr;
2226
2227         wmove(view->win, lineno, 0);
2228
2229         if (selected) {
2230                 type = LINE_CURSOR;
2231                 wchgat(view->win, -1, 0, type, NULL);
2232         }
2233
2234         attr = get_line_attr(type);
2235         wattrset(view->win, attr);
2236
2237         if (opt_line_number || opt_tab_size < TABSIZE) {
2238                 static char spaces[] = "                    ";
2239                 int col_offset = 0, col = 0;
2240
2241                 if (opt_line_number) {
2242                         unsigned long real_lineno = view->offset + lineno + 1;
2243
2244                         if (real_lineno == 1 ||
2245                             (real_lineno % opt_num_interval) == 0) {
2246                                 wprintw(view->win, "%.*d", view->digits, real_lineno);
2247
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                 }
2255
2256                 while (text && col_offset + col < view->width) {
2257                         int cols_max = view->width - col_offset - col;
2258                         char *pos = text;
2259                         int cols;
2260
2261                         if (*text == '\t') {
2262                                 text++;
2263                                 assert(sizeof(spaces) > TABSIZE);
2264                                 pos = spaces;
2265                                 cols = opt_tab_size - (col % opt_tab_size);
2266
2267                         } else {
2268                                 text = strchr(text, '\t');
2269                                 cols = line ? text - pos : strlen(pos);
2270                         }
2271
2272                         waddnstr(view->win, pos, MIN(cols, cols_max));
2273                         col += cols;
2274                 }
2275
2276         } else {
2277                 int col = 0, pos = 0;
2278
2279                 for (; pos < textlen && col < view->width; pos++, col++)
2280                         if (text[pos] == '\t')
2281                                 col += TABSIZE - (col % TABSIZE) - 1;
2282
2283                 waddnstr(view->win, text, pos);
2284         }
2285
2286         return TRUE;
2287 }
2288
2289 static bool
2290 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2291 {
2292         char refbuf[SIZEOF_STR];
2293         char *ref = NULL;
2294         FILE *pipe;
2295
2296         if (!string_format(refbuf, "git describe %s", commit_id))
2297                 return TRUE;
2298
2299         pipe = popen(refbuf, "r");
2300         if (!pipe)
2301                 return TRUE;
2302
2303         if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2304                 ref = chomp_string(ref);
2305         pclose(pipe);
2306
2307         if (!ref || !*ref)
2308                 return TRUE;
2309
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;
2313
2314         return TRUE;
2315 }
2316
2317 static void
2318 add_pager_refs(struct view *view, struct line *line)
2319 {
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;
2326
2327         assert(line->type == LINE_COMMIT);
2328
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         }
2335
2336         do {
2337                 struct ref *ref = refs[refpos];
2338                 char *fmt = ref->tag ? "%s[%s]" : "%s%s";
2339
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);
2346
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         }
2353
2354         if (bufpos == 0)
2355                 return;
2356
2357         if (!realloc_lines(view, view->line_size + 1))
2358                 return;
2359
2360         line = &view->line[view->lines];
2361         line->data = strdup(buf);
2362         if (!line->data)
2363                 return;
2364
2365         line->type = LINE_PP_REFS;
2366         view->lines++;
2367 }
2368
2369 static bool
2370 pager_read(struct view *view, char *data)
2371 {
2372         struct line *line = &view->line[view->lines];
2373
2374         line->data = strdup(data);
2375         if (!line->data)
2376                 return FALSE;
2377
2378         line->type = get_line_type(line->data);
2379         view->lines++;
2380
2381         if (line->type == LINE_COMMIT &&
2382             (view == VIEW(REQ_VIEW_DIFF) ||
2383              view == VIEW(REQ_VIEW_LOG)))
2384                 add_pager_refs(view, line);
2385
2386         return TRUE;
2387 }
2388
2389 static bool
2390 pager_enter(struct view *view, struct line *line)
2391 {
2392         int split = 0;
2393
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         }
2400
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);
2405
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);
2411
2412         return TRUE;
2413 }
2414
2415 static bool
2416 pager_grep(struct view *view, struct line *line)
2417 {
2418         regmatch_t pmatch;
2419         char *text = line->data;
2420
2421         if (!*text)
2422                 return FALSE;
2423
2424         if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2425                 return FALSE;
2426
2427         return TRUE;
2428 }
2429
2430 static void
2431 pager_select(struct view *view, struct line *line)
2432 {
2433         if (line->type == LINE_COMMIT) {
2434                 char *text = line->data;
2435
2436                 string_copy(view->ref, text + STRING_SIZE("commit "));
2437                 string_copy(ref_commit, view->ref);
2438         }
2439 }
2440
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 };
2449
2450
2451 /*
2452  * Tree backend
2453  */
2454
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  */
2462
2463 #define SIZEOF_TREE_ATTR \
2464         STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
2465
2466 #define TREE_UP_FORMAT "040000 tree %s\t.."
2467
2468 static int
2469 tree_compare_entry(enum line_type type1, char *name1,
2470                    enum line_type type2, char *name2)
2471 {
2472         if (type1 != type2) {
2473                 if (type1 == LINE_TREE_DIR)
2474                         return -1;
2475                 return 1;
2476         }
2477
2478         return strcmp(name1, name2);
2479 }
2480
2481 static bool
2482 tree_read(struct view *view, char *text)
2483 {
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;
2489
2490         if (textlen <= SIZEOF_TREE_ATTR)
2491                 return FALSE;
2492
2493         type = text[STRING_SIZE("100644 ")] == 't'
2494              ? LINE_TREE_DIR : LINE_TREE_FILE;
2495
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;
2504
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         }
2514
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;
2520
2521                 if (pathlen > striplen)
2522                         memmove(path, path + striplen,
2523                                 pathlen - striplen + 1);
2524         }
2525
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);
2532
2533                 if (cmp <= 0)
2534                         continue;
2535
2536                 text = strdup(text);
2537                 if (!text)
2538                         return FALSE;
2539
2540                 if (view->lines > pos)
2541                         memmove(&view->line[pos + 1], &view->line[pos],
2542                                 (view->lines - pos) * sizeof(*line));
2543
2544                 line = &view->line[pos];
2545                 line->data = text;
2546                 line->type = type;
2547                 view->lines++;
2548                 return TRUE;
2549         }
2550
2551         if (!pager_read(view, text))
2552                 return FALSE;
2553
2554         /* Move the current line to the first tree entry. */
2555         if (first_read)
2556                 view->lineno++;
2557
2558         view->line[view->lines - 1].type = type;
2559         return TRUE;
2560 }
2561
2562 static bool
2563 tree_enter(struct view *view, struct line *line)
2564 {
2565         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2566         enum request request;
2567
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;
2575
2576                         while (dirsep > opt_path && dirsep[-1] != '/')
2577                                 dirsep--;
2578
2579                         dirsep[0] = 0;
2580
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;
2586
2587                         if (!string_format_from(opt_path, &pathlen, "%s/", basename)) {
2588                                 opt_path[origlen] = 0;
2589                                 return TRUE;
2590                         }
2591                 }
2592
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;
2598
2599         case LINE_TREE_FILE:
2600                 request = REQ_VIEW_BLOB;
2601                 break;
2602
2603         default:
2604                 return TRUE;
2605         }
2606
2607         open_view(view, request, flags);
2608
2609         return TRUE;
2610 }
2611
2612 static void
2613 tree_select(struct view *view, struct line *line)
2614 {
2615         char *text = line->data;
2616
2617         text += STRING_SIZE("100644 blob ");
2618
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);
2624
2625         } else if (line->type != LINE_TREE_DIR) {
2626                 return;
2627         }
2628
2629         string_ncopy(view->ref, text, 40);
2630 }
2631
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 };
2640
2641 static bool
2642 blob_read(struct view *view, char *line)
2643 {
2644         bool state = pager_read(view, line);
2645
2646         if (state == TRUE)
2647                 view->line[view->lines - 1].type = LINE_DEFAULT;
2648
2649         return state;
2650 }
2651
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 };
2660
2661
2662 /*
2663  * Main view backend
2664  */
2665
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 };
2675
2676 static bool
2677 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2678 {
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;
2686
2687         if (!*commit->author)
2688                 return FALSE;
2689
2690         wmove(view->win, lineno, col);
2691
2692         if (selected) {
2693                 type = LINE_CURSOR;
2694                 wattrset(view->win, get_line_attr(type));
2695                 wchgat(view->win, -1, 0, type, NULL);
2696
2697         } else {
2698                 type = LINE_MAIN_COMMIT;
2699                 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
2700         }
2701
2702         timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
2703         waddnstr(view->win, buf, timelen);
2704         waddstr(view->win, " ");
2705
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));
2710
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         }
2720
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         }
2729
2730         col += AUTHOR_COLS;
2731         if (type != LINE_CURSOR)
2732                 wattrset(view->win, A_NORMAL);
2733
2734         if (opt_rev_graph && commit->graph_size) {
2735                 size_t i;
2736
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]);
2742
2743                 col += commit->graph_size + 1;
2744         }
2745
2746         wmove(view->win, lineno, col);
2747
2748         if (commit->refs) {
2749                 size_t i = 0;
2750
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         }
2767
2768         if (type != LINE_CURSOR)
2769                 wattrset(view->win, get_line_attr(type));
2770
2771         {
2772                 int titlelen = strlen(commit->title);
2773
2774                 if (col + titlelen > view->width)
2775                         titlelen = view->width - col;
2776
2777                 waddnstr(view->win, commit->title, titlelen);
2778         }
2779
2780         return TRUE;
2781 }
2782
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)
2786 {
2787         enum line_type type = get_line_type(line);
2788         struct commit *commit = view->lines
2789                               ? view->line[view->lines - 1].data : NULL;
2790
2791         switch (type) {
2792         case LINE_COMMIT:
2793                 commit = calloc(1, sizeof(struct commit));
2794                 if (!commit)
2795                         return FALSE;
2796
2797                 line += STRING_SIZE("commit ");
2798
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;
2804
2805         case LINE_AUTHOR:
2806         {
2807                 char *ident = line + STRING_SIZE("author ");
2808                 char *end = strchr(ident, '<');
2809
2810                 if (!commit)
2811                         break;
2812
2813                 if (end) {
2814                         char *email = end + 1;
2815
2816                         for (; end > ident && isspace(end[-1]); end--) ;
2817
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                 }
2825
2826                 /* End is NULL or ident meaning there's no author. */
2827                 if (end <= ident)
2828                         ident = "Unknown";
2829
2830                 string_copy(commit->author, ident);
2831
2832                 /* Parse epoch and timezone */
2833                 if (end) {
2834                         char *secs = strchr(end + 1, '>');
2835                         char *zone;
2836                         time_t time;
2837
2838                         if (!secs || secs[1] != ' ')
2839                                 break;
2840
2841                         secs += 2;
2842                         time = (time_t) atol(secs);
2843                         zone = strchr(secs, ' ');
2844                         if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
2845                                 long tz;
2846
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;
2852
2853                                 if (zone[0] == '-')
2854                                         tz = -tz;
2855
2856                                 time -= tz;
2857                         }
2858                         gmtime_r(&time, &commit->time);
2859                 }
2860                 break;
2861         }
2862         default:
2863                 if (!commit)
2864                         break;
2865
2866                 /* Fill in the commit title if it has not already been set. */
2867                 if (commit->title[0])
2868                         break;
2869
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;
2877
2878                 string_copy(commit->title, line + 4);
2879         }
2880
2881         return TRUE;
2882 }
2883
2884 static bool
2885 main_enter(struct view *view, struct line *line)
2886 {
2887         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2888
2889         open_view(view, REQ_VIEW_DIFF, flags);
2890         return TRUE;
2891 }
2892
2893 static bool
2894 main_grep(struct view *view, struct line *line)
2895 {
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;
2900
2901         for (state = S_TITLE; state < S_END; state++) {
2902                 char *text;
2903
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;
2912
2913                 default:
2914                         return FALSE;
2915                 }
2916
2917                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
2918                         return TRUE;
2919         }
2920
2921         return FALSE;
2922 }
2923
2924 static void
2925 main_select(struct view *view, struct line *line)
2926 {
2927         struct commit *commit = line->data;
2928
2929         string_copy(view->ref, commit->id);
2930         string_copy(ref_commit, view->ref);
2931 }
2932
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 };
2941
2942
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  */
2950
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 */
2954
2955 static inline int
2956 unicode_width(unsigned long c)
2957 {
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;
2972
2973         return 1;
2974 }
2975
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 };
2988
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)
2992 {
2993         unsigned long unicode;
2994
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         }
3032
3033         /* Invalid characters could return the special 0xfffd value but NUL
3034          * should be just as good. */
3035         return unicode > 0xffff ? 0 : unicode;
3036 }
3037
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)
3049 {
3050         const char *start = string;
3051         const char *end = strchr(string, '\0');
3052         size_t mbwidth = 0;
3053         size_t width = 0;
3054
3055         *trimmed = 0;
3056
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;
3062
3063                 if (string + bytes > end)
3064                         break;
3065
3066                 /* Change representation to figure out whether
3067                  * it is a single- or double-width character. */
3068
3069                 unicode = utf8_to_unicode(string, bytes);
3070                 /* FIXME: Graceful handling of invalid unicode character. */
3071                 if (!unicode)
3072                         break;
3073
3074                 ucwidth = unicode_width(unicode);
3075                 width  += ucwidth;
3076                 if (width > max_width) {
3077                         *trimmed = 1;
3078                         break;
3079                 }
3080
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;
3093
3094                 string  += bytes;
3095         }
3096
3097         *coloffset += mbwidth;
3098
3099         return string - start;
3100 }
3101
3102
3103 /*
3104  * Status management
3105  */
3106
3107 /* Whether or not the curses interface has been initialized. */
3108 static bool cursed = FALSE;
3109
3110 /* The status window is used for polling keystrokes. */
3111 static WINDOW *status_win;
3112
3113 /* Update status and title window. */
3114 static void
3115 report(const char *msg, ...)
3116 {
3117         static bool empty = TRUE;
3118         struct view *view = display[current_view];
3119
3120         if (!empty || *msg) {
3121                 va_list args;
3122
3123                 va_start(args, msg);
3124
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);
3134
3135                 va_end(args);
3136         }
3137
3138         update_view_title(view);
3139         update_display_cursor(view);
3140 }
3141
3142 /* Controls when nodelay should be in effect when polling user input. */
3143 static void
3144 set_nonblocking_input(bool loading)
3145 {
3146         static unsigned int loading_views;
3147
3148         if ((loading == FALSE && loading_views-- == 1) ||
3149             (loading == TRUE  && loading_views++ == 0))
3150                 nodelay(status_win, loading);
3151 }
3152
3153 static void
3154 init_display(void)
3155 {
3156         int x, y;
3157
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+");
3164
3165                 if (!io)
3166                         die("Failed to open /dev/tty");
3167                 cursed = !!newterm(NULL, io, io);
3168         }
3169
3170         if (!cursed)
3171                 die("Failed to initialize curses");
3172
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);
3177
3178         if (has_colors())
3179                 init_colors();
3180
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");
3185
3186         /* Enable keyboard mapping */
3187         keypad(status_win, TRUE);
3188         wbkgdset(status_win, get_line_attr(LINE_STATUS));
3189 }
3190
3191 static char *
3192 read_prompt(const char *prompt)
3193 {
3194         enum { READING, STOP, CANCEL } status = READING;
3195         static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
3196         int pos = 0;
3197
3198         while (status == READING) {
3199                 struct view *view;
3200                 int i, key;
3201
3202                 foreach_view (view, i)
3203                         update_view(view);
3204
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;
3214
3215                 case KEY_BACKSPACE:
3216                         if (pos > 0)
3217                                 pos--;
3218                         else
3219                                 status = CANCEL;
3220                         break;
3221
3222                 case KEY_ESC:
3223                         status = CANCEL;
3224                         break;
3225
3226                 case ERR:
3227                         break;
3228
3229                 default:
3230                         if (pos >= sizeof(buf)) {
3231                                 report("Input string too long");
3232                                 return NULL;
3233                         }
3234
3235                         if (isprint(key))
3236                                 buf[pos++] = (char) key;
3237                 }
3238         }
3239
3240         if (status == CANCEL) {
3241                 /* Clear the status window */
3242                 report("");
3243                 return NULL;
3244         }
3245
3246         buf[pos++] = 0;
3247
3248         return buf;
3249 }
3250
3251 /*
3252  * Repository references
3253  */
3254
3255 static struct ref *refs;
3256 static size_t refs_size;
3257
3258 /* Id <-> ref store */
3259 static struct ref ***id_refs;
3260 static size_t id_refs_size;
3261
3262 static struct ref **
3263 get_refs(char *id)
3264 {
3265         struct ref ***tmp_id_refs;
3266         struct ref **ref_list = NULL;
3267         size_t ref_list_size = 0;
3268         size_t i;
3269
3270         for (i = 0; i < id_refs_size; i++)
3271                 if (!strcmp(id, id_refs[i][0]->id))
3272                         return id_refs[i];
3273
3274         tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
3275         if (!tmp_id_refs)
3276                 return NULL;
3277
3278         id_refs = tmp_id_refs;
3279
3280         for (i = 0; i < refs_size; i++) {
3281                 struct ref **tmp;
3282
3283                 if (strcmp(id, refs[i].id))
3284                         continue;
3285
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                 }
3292
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];
3297
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         }
3304
3305         if (ref_list)
3306                 id_refs[id_refs_size++] = ref_list;
3307
3308         return ref_list;
3309 }
3310
3311 static int
3312 read_ref(char *id, int idlen, char *name, int namelen)
3313 {
3314         struct ref *ref;
3315         bool tag = FALSE;
3316
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;
3321
3322                 while (namelen > 0 && name[namelen] != '^')
3323                         namelen--;
3324
3325                 tag = TRUE;
3326                 namelen -= STRING_SIZE("refs/tags/");
3327                 name    += STRING_SIZE("refs/tags/");
3328
3329         } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
3330                 namelen -= STRING_SIZE("refs/heads/");
3331                 name    += STRING_SIZE("refs/heads/");
3332
3333         } else if (!strcmp(name, "HEAD")) {
3334                 return OK;
3335         }
3336
3337         refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
3338         if (!refs)
3339                 return ERR;
3340
3341         ref = &refs[refs_size++];
3342         ref->name = malloc(namelen + 1);
3343         if (!ref->name)
3344                 return ERR;
3345
3346         strncpy(ref->name, name, namelen);
3347         ref->name[namelen] = 0;
3348         ref->tag = tag;
3349         string_copy(ref->id, id);
3350
3351         return OK;
3352 }
3353
3354 static int
3355 load_refs(void)
3356 {
3357         const char *cmd_env = getenv("TIG_LS_REMOTE");
3358         const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
3359
3360         return read_properties(popen(cmd, "r"), "\t", read_ref);
3361 }
3362
3363 static int
3364 read_repo_config_option(char *name, int namelen, char *value, int valuelen)
3365 {
3366         if (!strcmp(name, "i18n.commitencoding"))
3367                 string_copy(opt_encoding, value);
3368
3369         return OK;
3370 }
3371
3372 static int
3373 load_repo_config(void)
3374 {
3375         return read_properties(popen("git repo-config --list", "r"),
3376                                "=", read_repo_config_option);
3377 }
3378
3379 static int
3380 read_properties(FILE *pipe, const char *separators,
3381                 int (*read_property)(char *, int, char *, int))
3382 {
3383         char buffer[BUFSIZ];
3384         char *name;
3385         int state = OK;
3386
3387         if (!pipe)
3388                 return ERR;
3389
3390         while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
3391                 char *value;
3392                 size_t namelen;
3393                 size_t valuelen;
3394
3395                 name = chomp_string(name);
3396                 namelen = strcspn(name, separators);
3397
3398                 if (name[namelen]) {
3399                         name[namelen] = 0;
3400                         value = chomp_string(name + namelen + 1);
3401                         valuelen = strlen(value);
3402
3403                 } else {
3404                         value = "";
3405                         valuelen = 0;
3406                 }
3407
3408                 state = read_property(name, namelen, value, valuelen);
3409         }
3410
3411         if (state != ERR && ferror(pipe))
3412                 state = ERR;
3413
3414         pclose(pipe);
3415
3416         return state;
3417 }
3418
3419
3420 /*
3421  * Main
3422  */
3423
3424 static void __NORETURN
3425 quit(int sig)
3426 {
3427         /* XXX: Restore tty modes and let the OS cleanup the rest! */
3428         if (cursed)
3429                 endwin();
3430         exit(0);
3431 }
3432
3433 static void __NORETURN
3434 die(const char *err, ...)
3435 {
3436         va_list args;
3437
3438         endwin();
3439
3440         va_start(args, err);
3441         fputs("tig: ", stderr);
3442         vfprintf(stderr, err, args);
3443         fputs("\n", stderr);
3444         va_end(args);
3445
3446         exit(1);
3447 }
3448
3449 int
3450 main(int argc, char *argv[])
3451 {
3452         struct view *view;
3453         enum request request;
3454         size_t i;
3455
3456         signal(SIGINT, quit);
3457
3458         if (setlocale(LC_ALL, "")) {
3459                 string_copy(opt_codeset, nl_langinfo(CODESET));
3460         }
3461
3462         if (load_options() == ERR)
3463                 die("Failed to load user config.");
3464
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.");
3469
3470         if (!parse_options(argc, argv))
3471                 return 0;
3472
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         }
3478
3479         if (load_refs() == ERR)
3480                 die("Failed to load refs.");
3481
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");
3485
3486         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
3487                 view->cmd_env = getenv(view->cmd_env);
3488
3489         request = opt_request;
3490
3491         init_display();
3492
3493         while (view_driver(display[current_view], request)) {
3494                 int key;
3495                 int i;
3496
3497                 foreach_view (view, i)
3498                         update_view(view);
3499
3500                 /* Refresh, accept single keystroke of input */
3501                 key = wgetch(status_win);
3502
3503                 request = get_keybinding(display[current_view]->keymap, key);
3504
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(":");
3511
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                         }
3520
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);
3530
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;
3540
3541                         getmaxyx(stdscr, height, width);
3542
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         }
3554
3555         quit(0);
3556
3557         return 0;
3558 }