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