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