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