Blame: do to not reload the diff view for same commits
[tig] / tig.c
1 /* Copyright (c) 2006-2008 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 #ifdef HAVE_CONFIG_H
15 #include "config.h"
16 #endif
17
18 #ifndef TIG_VERSION
19 #define TIG_VERSION "unknown-version"
20 #endif
21
22 #ifndef DEBUG
23 #define NDEBUG
24 #endif
25
26 #include <assert.h>
27 #include <errno.h>
28 #include <ctype.h>
29 #include <signal.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #include <sys/types.h>
35 #include <sys/stat.h>
36 #include <unistd.h>
37 #include <time.h>
38
39 #include <regex.h>
40
41 #include <locale.h>
42 #include <langinfo.h>
43 #include <iconv.h>
44
45 /* ncurses(3): Must be defined to have extended wide-character functions. */
46 #define _XOPEN_SOURCE_EXTENDED
47
48 #ifdef HAVE_NCURSESW_NCURSES_H
49 #include <ncursesw/ncurses.h>
50 #else
51 #ifdef HAVE_NCURSES_NCURSES_H
52 #include <ncurses/ncurses.h>
53 #else
54 #include <ncurses.h>
55 #endif
56 #endif
57
58 #if __GNUC__ >= 3
59 #define __NORETURN __attribute__((__noreturn__))
60 #else
61 #define __NORETURN
62 #endif
63
64 static void __NORETURN die(const char *err, ...);
65 static void warn(const char *msg, ...);
66 static void report(const char *msg, ...);
67 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, size_t, char *, size_t));
68 static void set_nonblocking_input(bool loading);
69 static size_t utf8_length(const char *string, int *width, size_t max_width, int *trimmed, bool reserve);
70 static bool prompt_yesno(const char *prompt);
71 static int load_refs(void);
72
73 #define ABS(x)          ((x) >= 0  ? (x) : -(x))
74 #define MIN(x, y)       ((x) < (y) ? (x) :  (y))
75
76 #define ARRAY_SIZE(x)   (sizeof(x) / sizeof(x[0]))
77 #define STRING_SIZE(x)  (sizeof(x) - 1)
78
79 #define SIZEOF_STR      1024    /* Default string size. */
80 #define SIZEOF_REF      256     /* Size of symbolic or SHA1 ID. */
81 #define SIZEOF_REV      41      /* Holds a SHA-1 and an ending NUL. */
82 #define SIZEOF_ARG      32      /* Default argument array size. */
83
84 /* Revision graph */
85
86 #define REVGRAPH_INIT   'I'
87 #define REVGRAPH_MERGE  'M'
88 #define REVGRAPH_BRANCH '+'
89 #define REVGRAPH_COMMIT '*'
90 #define REVGRAPH_BOUND  '^'
91
92 #define SIZEOF_REVGRAPH 19      /* Size of revision ancestry graphics. */
93
94 /* This color name can be used to refer to the default term colors. */
95 #define COLOR_DEFAULT   (-1)
96
97 #define ICONV_NONE      ((iconv_t) -1)
98 #ifndef ICONV_CONST
99 #define ICONV_CONST     /* nothing */
100 #endif
101
102 /* The format and size of the date column in the main view. */
103 #define DATE_FORMAT     "%Y-%m-%d %H:%M"
104 #define DATE_COLS       STRING_SIZE("2006-04-29 14:21 ")
105
106 #define AUTHOR_COLS     20
107 #define ID_COLS         8
108
109 /* The default interval between line numbers. */
110 #define NUMBER_INTERVAL 5
111
112 #define TAB_SIZE        8
113
114 #define SCALE_SPLIT_VIEW(height)        ((height) * 2 / 3)
115
116 #define NULL_ID         "0000000000000000000000000000000000000000"
117
118 #ifndef GIT_CONFIG
119 #define GIT_CONFIG "config"
120 #endif
121
122 #define TIG_LS_REMOTE \
123         "git ls-remote . 2>/dev/null"
124
125 #define TIG_DIFF_CMD \
126         "git show --pretty=fuller --no-color --root --patch-with-stat --find-copies-harder -C %s 2>/dev/null"
127
128 #define TIG_LOG_CMD     \
129         "git log --no-color --cc --stat -n100 %s 2>/dev/null"
130
131 #define TIG_MAIN_BASE \
132         "git log --no-color --pretty=raw --parents --topo-order"
133
134 #define TIG_MAIN_CMD \
135         TIG_MAIN_BASE " %s 2>/dev/null"
136
137 #define TIG_TREE_CMD    \
138         "git ls-tree %s %s"
139
140 #define TIG_BLOB_CMD    \
141         "git cat-file blob %s"
142
143 /* XXX: Needs to be defined to the empty string. */
144 #define TIG_HELP_CMD    ""
145 #define TIG_PAGER_CMD   ""
146 #define TIG_STATUS_CMD  ""
147 #define TIG_STAGE_CMD   ""
148 #define TIG_BLAME_CMD   ""
149
150 /* Some ascii-shorthands fitted into the ncurses namespace. */
151 #define KEY_TAB         '\t'
152 #define KEY_RETURN      '\r'
153 #define KEY_ESC         27
154
155
156 struct ref {
157         char *name;             /* Ref name; tag or head names are shortened. */
158         char id[SIZEOF_REV];    /* Commit SHA1 ID */
159         unsigned int head:1;    /* Is it the current HEAD? */
160         unsigned int tag:1;     /* Is it a tag? */
161         unsigned int ltag:1;    /* If so, is the tag local? */
162         unsigned int remote:1;  /* Is it a remote ref? */
163         unsigned int tracked:1; /* Is it the remote for the current HEAD? */
164         unsigned int next:1;    /* For ref lists: are there more refs? */
165 };
166
167 static struct ref **get_refs(const char *id);
168
169 struct int_map {
170         const char *name;
171         int namelen;
172         int value;
173 };
174
175 static int
176 set_from_int_map(struct int_map *map, size_t map_size,
177                  int *value, const char *name, int namelen)
178 {
179
180         int i;
181
182         for (i = 0; i < map_size; i++)
183                 if (namelen == map[i].namelen &&
184                     !strncasecmp(name, map[i].name, namelen)) {
185                         *value = map[i].value;
186                         return OK;
187                 }
188
189         return ERR;
190 }
191
192
193 /*
194  * String helpers
195  */
196
197 static inline void
198 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
199 {
200         if (srclen > dstlen - 1)
201                 srclen = dstlen - 1;
202
203         strncpy(dst, src, srclen);
204         dst[srclen] = 0;
205 }
206
207 /* Shorthands for safely copying into a fixed buffer. */
208
209 #define string_copy(dst, src) \
210         string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
211
212 #define string_ncopy(dst, src, srclen) \
213         string_ncopy_do(dst, sizeof(dst), src, srclen)
214
215 #define string_copy_rev(dst, src) \
216         string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
217
218 #define string_add(dst, from, src) \
219         string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
220
221 static char *
222 chomp_string(char *name)
223 {
224         int namelen;
225
226         while (isspace(*name))
227                 name++;
228
229         namelen = strlen(name) - 1;
230         while (namelen > 0 && isspace(name[namelen]))
231                 name[namelen--] = 0;
232
233         return name;
234 }
235
236 static bool
237 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
238 {
239         va_list args;
240         size_t pos = bufpos ? *bufpos : 0;
241
242         va_start(args, fmt);
243         pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
244         va_end(args);
245
246         if (bufpos)
247                 *bufpos = pos;
248
249         return pos >= bufsize ? FALSE : TRUE;
250 }
251
252 #define string_format(buf, fmt, args...) \
253         string_nformat(buf, sizeof(buf), NULL, fmt, args)
254
255 #define string_format_from(buf, from, fmt, args...) \
256         string_nformat(buf, sizeof(buf), from, fmt, args)
257
258 static int
259 string_enum_compare(const char *str1, const char *str2, int len)
260 {
261         size_t i;
262
263 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
264
265         /* Diff-Header == DIFF_HEADER */
266         for (i = 0; i < len; i++) {
267                 if (toupper(str1[i]) == toupper(str2[i]))
268                         continue;
269
270                 if (string_enum_sep(str1[i]) &&
271                     string_enum_sep(str2[i]))
272                         continue;
273
274                 return str1[i] - str2[i];
275         }
276
277         return 0;
278 }
279
280 #define prefixcmp(str1, str2) \
281         strncmp(str1, str2, STRING_SIZE(str2))
282
283 static inline int
284 suffixcmp(const char *str, int slen, const char *suffix)
285 {
286         size_t len = slen >= 0 ? slen : strlen(str);
287         size_t suffixlen = strlen(suffix);
288
289         return suffixlen < len ? strcmp(str + len - suffixlen, suffix) : -1;
290 }
291
292 /* Shell quoting
293  *
294  * NOTE: The following is a slightly modified copy of the git project's shell
295  * quoting routines found in the quote.c file.
296  *
297  * Help to copy the thing properly quoted for the shell safety.  any single
298  * quote is replaced with '\'', any exclamation point is replaced with '\!',
299  * and the whole thing is enclosed in a
300  *
301  * E.g.
302  *  original     sq_quote     result
303  *  name     ==> name      ==> 'name'
304  *  a b      ==> a b       ==> 'a b'
305  *  a'b      ==> a'\''b    ==> 'a'\''b'
306  *  a!b      ==> a'\!'b    ==> 'a'\!'b'
307  */
308
309 static size_t
310 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
311 {
312         char c;
313
314 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
315
316         BUFPUT('\'');
317         while ((c = *src++)) {
318                 if (c == '\'' || c == '!') {
319                         BUFPUT('\'');
320                         BUFPUT('\\');
321                         BUFPUT(c);
322                         BUFPUT('\'');
323                 } else {
324                         BUFPUT(c);
325                 }
326         }
327         BUFPUT('\'');
328
329         if (bufsize < SIZEOF_STR)
330                 buf[bufsize] = 0;
331
332         return bufsize;
333 }
334
335
336 /*
337  * User requests
338  */
339
340 #define REQ_INFO \
341         /* XXX: Keep the view request first and in sync with views[]. */ \
342         REQ_GROUP("View switching") \
343         REQ_(VIEW_MAIN,         "Show main view"), \
344         REQ_(VIEW_DIFF,         "Show diff view"), \
345         REQ_(VIEW_LOG,          "Show log view"), \
346         REQ_(VIEW_TREE,         "Show tree view"), \
347         REQ_(VIEW_BLOB,         "Show blob view"), \
348         REQ_(VIEW_BLAME,        "Show blame view"), \
349         REQ_(VIEW_HELP,         "Show help page"), \
350         REQ_(VIEW_PAGER,        "Show pager view"), \
351         REQ_(VIEW_STATUS,       "Show status view"), \
352         REQ_(VIEW_STAGE,        "Show stage view"), \
353         \
354         REQ_GROUP("View manipulation") \
355         REQ_(ENTER,             "Enter current line and scroll"), \
356         REQ_(NEXT,              "Move to next"), \
357         REQ_(PREVIOUS,          "Move to previous"), \
358         REQ_(VIEW_NEXT,         "Move focus to next view"), \
359         REQ_(REFRESH,           "Reload and refresh"), \
360         REQ_(MAXIMIZE,          "Maximize the current view"), \
361         REQ_(VIEW_CLOSE,        "Close the current view"), \
362         REQ_(QUIT,              "Close all views and quit"), \
363         \
364         REQ_GROUP("View specific requests") \
365         REQ_(STATUS_UPDATE,     "Update file status"), \
366         REQ_(STATUS_REVERT,     "Revert file changes"), \
367         REQ_(STATUS_MERGE,      "Merge file using external tool"), \
368         REQ_(STAGE_NEXT,        "Find next chunk to stage"), \
369         REQ_(TREE_PARENT,       "Switch to parent directory in tree view"), \
370         \
371         REQ_GROUP("Cursor navigation") \
372         REQ_(MOVE_UP,           "Move cursor one line up"), \
373         REQ_(MOVE_DOWN,         "Move cursor one line down"), \
374         REQ_(MOVE_PAGE_DOWN,    "Move cursor one page down"), \
375         REQ_(MOVE_PAGE_UP,      "Move cursor one page up"), \
376         REQ_(MOVE_FIRST_LINE,   "Move cursor to first line"), \
377         REQ_(MOVE_LAST_LINE,    "Move cursor to last line"), \
378         \
379         REQ_GROUP("Scrolling") \
380         REQ_(SCROLL_LINE_UP,    "Scroll one line up"), \
381         REQ_(SCROLL_LINE_DOWN,  "Scroll one line down"), \
382         REQ_(SCROLL_PAGE_UP,    "Scroll one page up"), \
383         REQ_(SCROLL_PAGE_DOWN,  "Scroll one page down"), \
384         \
385         REQ_GROUP("Searching") \
386         REQ_(SEARCH,            "Search the view"), \
387         REQ_(SEARCH_BACK,       "Search backwards in the view"), \
388         REQ_(FIND_NEXT,         "Find next search match"), \
389         REQ_(FIND_PREV,         "Find previous search match"), \
390         \
391         REQ_GROUP("Option manipulation") \
392         REQ_(TOGGLE_LINENO,     "Toggle line numbers"), \
393         REQ_(TOGGLE_DATE,       "Toggle date display"), \
394         REQ_(TOGGLE_AUTHOR,     "Toggle author display"), \
395         REQ_(TOGGLE_REV_GRAPH,  "Toggle revision graph visualization"), \
396         REQ_(TOGGLE_REFS,       "Toggle reference display (tags/branches)"), \
397         \
398         REQ_GROUP("Misc") \
399         REQ_(PROMPT,            "Bring up the prompt"), \
400         REQ_(SCREEN_REDRAW,     "Redraw the screen"), \
401         REQ_(SCREEN_RESIZE,     "Resize the screen"), \
402         REQ_(SHOW_VERSION,      "Show version information"), \
403         REQ_(STOP_LOADING,      "Stop all loading views"), \
404         REQ_(EDIT,              "Open in editor"), \
405         REQ_(NONE,              "Do nothing")
406
407
408 /* User action requests. */
409 enum request {
410 #define REQ_GROUP(help)
411 #define REQ_(req, help) REQ_##req
412
413         /* Offset all requests to avoid conflicts with ncurses getch values. */
414         REQ_OFFSET = KEY_MAX + 1,
415         REQ_INFO
416
417 #undef  REQ_GROUP
418 #undef  REQ_
419 };
420
421 struct request_info {
422         enum request request;
423         const char *name;
424         int namelen;
425         const char *help;
426 };
427
428 static struct request_info req_info[] = {
429 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
430 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
431         REQ_INFO
432 #undef  REQ_GROUP
433 #undef  REQ_
434 };
435
436 static enum request
437 get_request(const char *name)
438 {
439         int namelen = strlen(name);
440         int i;
441
442         for (i = 0; i < ARRAY_SIZE(req_info); i++)
443                 if (req_info[i].namelen == namelen &&
444                     !string_enum_compare(req_info[i].name, name, namelen))
445                         return req_info[i].request;
446
447         return REQ_NONE;
448 }
449
450
451 /*
452  * Options
453  */
454
455 static const char usage[] =
456 "tig " TIG_VERSION " (" __DATE__ ")\n"
457 "\n"
458 "Usage: tig        [options] [revs] [--] [paths]\n"
459 "   or: tig show   [options] [revs] [--] [paths]\n"
460 "   or: tig blame  [rev] path\n"
461 "   or: tig status\n"
462 "   or: tig <      [git command output]\n"
463 "\n"
464 "Options:\n"
465 "  -v, --version   Show version and exit\n"
466 "  -h, --help      Show help message and exit";
467
468 /* Option and state variables. */
469 static bool opt_date                    = TRUE;
470 static bool opt_author                  = TRUE;
471 static bool opt_line_number             = FALSE;
472 static bool opt_line_graphics           = TRUE;
473 static bool opt_rev_graph               = FALSE;
474 static bool opt_show_refs               = TRUE;
475 static int opt_num_interval             = NUMBER_INTERVAL;
476 static int opt_tab_size                 = TAB_SIZE;
477 static int opt_author_cols              = AUTHOR_COLS-1;
478 static char opt_cmd[SIZEOF_STR]         = "";
479 static char opt_path[SIZEOF_STR]        = "";
480 static char opt_file[SIZEOF_STR]        = "";
481 static char opt_ref[SIZEOF_REF]         = "";
482 static char opt_head[SIZEOF_REF]        = "";
483 static char opt_head_rev[SIZEOF_REV]    = "";
484 static char opt_remote[SIZEOF_REF]      = "";
485 static FILE *opt_pipe                   = NULL;
486 static char opt_encoding[20]            = "UTF-8";
487 static bool opt_utf8                    = TRUE;
488 static char opt_codeset[20]             = "UTF-8";
489 static iconv_t opt_iconv                = ICONV_NONE;
490 static char opt_search[SIZEOF_STR]      = "";
491 static char opt_cdup[SIZEOF_STR]        = "";
492 static char opt_git_dir[SIZEOF_STR]     = "";
493 static signed char opt_is_inside_work_tree      = -1; /* set to TRUE or FALSE */
494 static char opt_editor[SIZEOF_STR]      = "";
495 static FILE *opt_tty                    = NULL;
496
497 #define is_initial_commit()     (!*opt_head_rev)
498 #define is_head_commit(rev)     (!strcmp((rev), "HEAD") || !strcmp(opt_head_rev, (rev)))
499
500 static enum request
501 parse_options(int argc, const char *argv[])
502 {
503         enum request request = REQ_VIEW_MAIN;
504         size_t buf_size;
505         const char *subcommand;
506         bool seen_dashdash = FALSE;
507         int i;
508
509         if (!isatty(STDIN_FILENO)) {
510                 opt_pipe = stdin;
511                 return REQ_VIEW_PAGER;
512         }
513
514         if (argc <= 1)
515                 return REQ_VIEW_MAIN;
516
517         subcommand = argv[1];
518         if (!strcmp(subcommand, "status") || !strcmp(subcommand, "-S")) {
519                 if (!strcmp(subcommand, "-S"))
520                         warn("`-S' has been deprecated; use `tig status' instead");
521                 if (argc > 2)
522                         warn("ignoring arguments after `%s'", subcommand);
523                 return REQ_VIEW_STATUS;
524
525         } else if (!strcmp(subcommand, "blame")) {
526                 if (argc <= 2 || argc > 4)
527                         die("invalid number of options to blame\n\n%s", usage);
528
529                 i = 2;
530                 if (argc == 4) {
531                         string_ncopy(opt_ref, argv[i], strlen(argv[i]));
532                         i++;
533                 }
534
535                 string_ncopy(opt_file, argv[i], strlen(argv[i]));
536                 return REQ_VIEW_BLAME;
537
538         } else if (!strcmp(subcommand, "show")) {
539                 request = REQ_VIEW_DIFF;
540
541         } else if (!strcmp(subcommand, "log") || !strcmp(subcommand, "diff")) {
542                 request = subcommand[0] == 'l' ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
543                 warn("`tig %s' has been deprecated", subcommand);
544
545         } else {
546                 subcommand = NULL;
547         }
548
549         if (!subcommand)
550                 /* XXX: This is vulnerable to the user overriding
551                  * options required for the main view parser. */
552                 string_copy(opt_cmd, TIG_MAIN_BASE);
553         else
554                 string_format(opt_cmd, "git %s", subcommand);
555
556         buf_size = strlen(opt_cmd);
557
558         for (i = 1 + !!subcommand; i < argc; i++) {
559                 const char *opt = argv[i];
560
561                 if (seen_dashdash || !strcmp(opt, "--")) {
562                         seen_dashdash = TRUE;
563
564                 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
565                         printf("tig version %s\n", TIG_VERSION);
566                         return REQ_NONE;
567
568                 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
569                         printf("%s\n", usage);
570                         return REQ_NONE;
571                 }
572
573                 opt_cmd[buf_size++] = ' ';
574                 buf_size = sq_quote(opt_cmd, buf_size, opt);
575                 if (buf_size >= sizeof(opt_cmd))
576                         die("command too long");
577         }
578
579         opt_cmd[buf_size] = 0;
580
581         return request;
582 }
583
584
585 /*
586  * Line-oriented content detection.
587  */
588
589 #define LINE_INFO \
590 LINE(DIFF_HEADER,  "diff --git ",       COLOR_YELLOW,   COLOR_DEFAULT,  0), \
591 LINE(DIFF_CHUNK,   "@@",                COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
592 LINE(DIFF_ADD,     "+",                 COLOR_GREEN,    COLOR_DEFAULT,  0), \
593 LINE(DIFF_DEL,     "-",                 COLOR_RED,      COLOR_DEFAULT,  0), \
594 LINE(DIFF_INDEX,        "index ",         COLOR_BLUE,   COLOR_DEFAULT,  0), \
595 LINE(DIFF_OLDMODE,      "old file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
596 LINE(DIFF_NEWMODE,      "new file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
597 LINE(DIFF_COPY_FROM,    "copy from",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
598 LINE(DIFF_COPY_TO,      "copy to",        COLOR_YELLOW, COLOR_DEFAULT,  0), \
599 LINE(DIFF_RENAME_FROM,  "rename from",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
600 LINE(DIFF_RENAME_TO,    "rename to",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
601 LINE(DIFF_SIMILARITY,   "similarity ",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
602 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
603 LINE(DIFF_TREE,         "diff-tree ",     COLOR_BLUE,   COLOR_DEFAULT,  0), \
604 LINE(PP_AUTHOR,    "Author: ",          COLOR_CYAN,     COLOR_DEFAULT,  0), \
605 LINE(PP_COMMIT,    "Commit: ",          COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
606 LINE(PP_MERGE,     "Merge: ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
607 LINE(PP_DATE,      "Date:   ",          COLOR_YELLOW,   COLOR_DEFAULT,  0), \
608 LINE(PP_ADATE,     "AuthorDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
609 LINE(PP_CDATE,     "CommitDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
610 LINE(PP_REFS,      "Refs: ",            COLOR_RED,      COLOR_DEFAULT,  0), \
611 LINE(COMMIT,       "commit ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
612 LINE(PARENT,       "parent ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
613 LINE(TREE,         "tree ",             COLOR_BLUE,     COLOR_DEFAULT,  0), \
614 LINE(AUTHOR,       "author ",           COLOR_CYAN,     COLOR_DEFAULT,  0), \
615 LINE(COMMITTER,    "committer ",        COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
616 LINE(SIGNOFF,      "    Signed-off-by", COLOR_YELLOW,   COLOR_DEFAULT,  0), \
617 LINE(ACKED,        "    Acked-by",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
618 LINE(DEFAULT,      "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
619 LINE(CURSOR,       "",                  COLOR_WHITE,    COLOR_GREEN,    A_BOLD), \
620 LINE(STATUS,       "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
621 LINE(DELIMITER,    "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
622 LINE(DATE,         "",                  COLOR_BLUE,     COLOR_DEFAULT,  0), \
623 LINE(LINE_NUMBER,  "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
624 LINE(TITLE_BLUR,   "",                  COLOR_WHITE,    COLOR_BLUE,     0), \
625 LINE(TITLE_FOCUS,  "",                  COLOR_WHITE,    COLOR_BLUE,     A_BOLD), \
626 LINE(MAIN_AUTHOR,  "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
627 LINE(MAIN_COMMIT,  "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
628 LINE(MAIN_TAG,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  A_BOLD), \
629 LINE(MAIN_LOCAL_TAG,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
630 LINE(MAIN_REMOTE,  "",                  COLOR_YELLOW,   COLOR_DEFAULT,  0), \
631 LINE(MAIN_TRACKED, "",                  COLOR_YELLOW,   COLOR_DEFAULT,  A_BOLD), \
632 LINE(MAIN_REF,     "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
633 LINE(MAIN_HEAD,    "",                  COLOR_CYAN,     COLOR_DEFAULT,  A_BOLD), \
634 LINE(MAIN_REVGRAPH,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
635 LINE(TREE_DIR,     "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
636 LINE(TREE_FILE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
637 LINE(STAT_HEAD,    "",                  COLOR_YELLOW,   COLOR_DEFAULT,  0), \
638 LINE(STAT_SECTION, "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
639 LINE(STAT_NONE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
640 LINE(STAT_STAGED,  "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
641 LINE(STAT_UNSTAGED,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
642 LINE(STAT_UNTRACKED,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
643 LINE(BLAME_ID,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0)
644
645 enum line_type {
646 #define LINE(type, line, fg, bg, attr) \
647         LINE_##type
648         LINE_INFO,
649         LINE_NONE
650 #undef  LINE
651 };
652
653 struct line_info {
654         const char *name;       /* Option name. */
655         int namelen;            /* Size of option name. */
656         const char *line;       /* The start of line to match. */
657         int linelen;            /* Size of string to match. */
658         int fg, bg, attr;       /* Color and text attributes for the lines. */
659 };
660
661 static struct line_info line_info[] = {
662 #define LINE(type, line, fg, bg, attr) \
663         { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
664         LINE_INFO
665 #undef  LINE
666 };
667
668 static enum line_type
669 get_line_type(const char *line)
670 {
671         int linelen = strlen(line);
672         enum line_type type;
673
674         for (type = 0; type < ARRAY_SIZE(line_info); type++)
675                 /* Case insensitive search matches Signed-off-by lines better. */
676                 if (linelen >= line_info[type].linelen &&
677                     !strncasecmp(line_info[type].line, line, line_info[type].linelen))
678                         return type;
679
680         return LINE_DEFAULT;
681 }
682
683 static inline int
684 get_line_attr(enum line_type type)
685 {
686         assert(type < ARRAY_SIZE(line_info));
687         return COLOR_PAIR(type) | line_info[type].attr;
688 }
689
690 static struct line_info *
691 get_line_info(const char *name)
692 {
693         size_t namelen = strlen(name);
694         enum line_type type;
695
696         for (type = 0; type < ARRAY_SIZE(line_info); type++)
697                 if (namelen == line_info[type].namelen &&
698                     !string_enum_compare(line_info[type].name, name, namelen))
699                         return &line_info[type];
700
701         return NULL;
702 }
703
704 static void
705 init_colors(void)
706 {
707         int default_bg = line_info[LINE_DEFAULT].bg;
708         int default_fg = line_info[LINE_DEFAULT].fg;
709         enum line_type type;
710
711         start_color();
712
713         if (assume_default_colors(default_fg, default_bg) == ERR) {
714                 default_bg = COLOR_BLACK;
715                 default_fg = COLOR_WHITE;
716         }
717
718         for (type = 0; type < ARRAY_SIZE(line_info); type++) {
719                 struct line_info *info = &line_info[type];
720                 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
721                 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
722
723                 init_pair(type, fg, bg);
724         }
725 }
726
727 struct line {
728         enum line_type type;
729
730         /* State flags */
731         unsigned int selected:1;
732         unsigned int dirty:1;
733
734         void *data;             /* User data */
735 };
736
737
738 /*
739  * Keys
740  */
741
742 struct keybinding {
743         int alias;
744         enum request request;
745 };
746
747 static struct keybinding default_keybindings[] = {
748         /* View switching */
749         { 'm',          REQ_VIEW_MAIN },
750         { 'd',          REQ_VIEW_DIFF },
751         { 'l',          REQ_VIEW_LOG },
752         { 't',          REQ_VIEW_TREE },
753         { 'f',          REQ_VIEW_BLOB },
754         { 'B',          REQ_VIEW_BLAME },
755         { 'p',          REQ_VIEW_PAGER },
756         { 'h',          REQ_VIEW_HELP },
757         { 'S',          REQ_VIEW_STATUS },
758         { 'c',          REQ_VIEW_STAGE },
759
760         /* View manipulation */
761         { 'q',          REQ_VIEW_CLOSE },
762         { KEY_TAB,      REQ_VIEW_NEXT },
763         { KEY_RETURN,   REQ_ENTER },
764         { KEY_UP,       REQ_PREVIOUS },
765         { KEY_DOWN,     REQ_NEXT },
766         { 'R',          REQ_REFRESH },
767         { KEY_F(5),     REQ_REFRESH },
768         { 'O',          REQ_MAXIMIZE },
769
770         /* Cursor navigation */
771         { 'k',          REQ_MOVE_UP },
772         { 'j',          REQ_MOVE_DOWN },
773         { KEY_HOME,     REQ_MOVE_FIRST_LINE },
774         { KEY_END,      REQ_MOVE_LAST_LINE },
775         { KEY_NPAGE,    REQ_MOVE_PAGE_DOWN },
776         { ' ',          REQ_MOVE_PAGE_DOWN },
777         { KEY_PPAGE,    REQ_MOVE_PAGE_UP },
778         { 'b',          REQ_MOVE_PAGE_UP },
779         { '-',          REQ_MOVE_PAGE_UP },
780
781         /* Scrolling */
782         { KEY_IC,       REQ_SCROLL_LINE_UP },
783         { KEY_DC,       REQ_SCROLL_LINE_DOWN },
784         { 'w',          REQ_SCROLL_PAGE_UP },
785         { 's',          REQ_SCROLL_PAGE_DOWN },
786
787         /* Searching */
788         { '/',          REQ_SEARCH },
789         { '?',          REQ_SEARCH_BACK },
790         { 'n',          REQ_FIND_NEXT },
791         { 'N',          REQ_FIND_PREV },
792
793         /* Misc */
794         { 'Q',          REQ_QUIT },
795         { 'z',          REQ_STOP_LOADING },
796         { 'v',          REQ_SHOW_VERSION },
797         { 'r',          REQ_SCREEN_REDRAW },
798         { '.',          REQ_TOGGLE_LINENO },
799         { 'D',          REQ_TOGGLE_DATE },
800         { 'A',          REQ_TOGGLE_AUTHOR },
801         { 'g',          REQ_TOGGLE_REV_GRAPH },
802         { 'F',          REQ_TOGGLE_REFS },
803         { ':',          REQ_PROMPT },
804         { 'u',          REQ_STATUS_UPDATE },
805         { '!',          REQ_STATUS_REVERT },
806         { 'M',          REQ_STATUS_MERGE },
807         { '@',          REQ_STAGE_NEXT },
808         { ',',          REQ_TREE_PARENT },
809         { 'e',          REQ_EDIT },
810
811         /* Using the ncurses SIGWINCH handler. */
812         { KEY_RESIZE,   REQ_SCREEN_RESIZE },
813 };
814
815 #define KEYMAP_INFO \
816         KEYMAP_(GENERIC), \
817         KEYMAP_(MAIN), \
818         KEYMAP_(DIFF), \
819         KEYMAP_(LOG), \
820         KEYMAP_(TREE), \
821         KEYMAP_(BLOB), \
822         KEYMAP_(BLAME), \
823         KEYMAP_(PAGER), \
824         KEYMAP_(HELP), \
825         KEYMAP_(STATUS), \
826         KEYMAP_(STAGE)
827
828 enum keymap {
829 #define KEYMAP_(name) KEYMAP_##name
830         KEYMAP_INFO
831 #undef  KEYMAP_
832 };
833
834 static struct int_map keymap_table[] = {
835 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
836         KEYMAP_INFO
837 #undef  KEYMAP_
838 };
839
840 #define set_keymap(map, name) \
841         set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
842
843 struct keybinding_table {
844         struct keybinding *data;
845         size_t size;
846 };
847
848 static struct keybinding_table keybindings[ARRAY_SIZE(keymap_table)];
849
850 static void
851 add_keybinding(enum keymap keymap, enum request request, int key)
852 {
853         struct keybinding_table *table = &keybindings[keymap];
854
855         table->data = realloc(table->data, (table->size + 1) * sizeof(*table->data));
856         if (!table->data)
857                 die("Failed to allocate keybinding");
858         table->data[table->size].alias = key;
859         table->data[table->size++].request = request;
860 }
861
862 /* Looks for a key binding first in the given map, then in the generic map, and
863  * lastly in the default keybindings. */
864 static enum request
865 get_keybinding(enum keymap keymap, int key)
866 {
867         size_t i;
868
869         for (i = 0; i < keybindings[keymap].size; i++)
870                 if (keybindings[keymap].data[i].alias == key)
871                         return keybindings[keymap].data[i].request;
872
873         for (i = 0; i < keybindings[KEYMAP_GENERIC].size; i++)
874                 if (keybindings[KEYMAP_GENERIC].data[i].alias == key)
875                         return keybindings[KEYMAP_GENERIC].data[i].request;
876
877         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
878                 if (default_keybindings[i].alias == key)
879                         return default_keybindings[i].request;
880
881         return (enum request) key;
882 }
883
884
885 struct key {
886         const char *name;
887         int value;
888 };
889
890 static struct key key_table[] = {
891         { "Enter",      KEY_RETURN },
892         { "Space",      ' ' },
893         { "Backspace",  KEY_BACKSPACE },
894         { "Tab",        KEY_TAB },
895         { "Escape",     KEY_ESC },
896         { "Left",       KEY_LEFT },
897         { "Right",      KEY_RIGHT },
898         { "Up",         KEY_UP },
899         { "Down",       KEY_DOWN },
900         { "Insert",     KEY_IC },
901         { "Delete",     KEY_DC },
902         { "Hash",       '#' },
903         { "Home",       KEY_HOME },
904         { "End",        KEY_END },
905         { "PageUp",     KEY_PPAGE },
906         { "PageDown",   KEY_NPAGE },
907         { "F1",         KEY_F(1) },
908         { "F2",         KEY_F(2) },
909         { "F3",         KEY_F(3) },
910         { "F4",         KEY_F(4) },
911         { "F5",         KEY_F(5) },
912         { "F6",         KEY_F(6) },
913         { "F7",         KEY_F(7) },
914         { "F8",         KEY_F(8) },
915         { "F9",         KEY_F(9) },
916         { "F10",        KEY_F(10) },
917         { "F11",        KEY_F(11) },
918         { "F12",        KEY_F(12) },
919 };
920
921 static int
922 get_key_value(const char *name)
923 {
924         int i;
925
926         for (i = 0; i < ARRAY_SIZE(key_table); i++)
927                 if (!strcasecmp(key_table[i].name, name))
928                         return key_table[i].value;
929
930         if (strlen(name) == 1 && isprint(*name))
931                 return (int) *name;
932
933         return ERR;
934 }
935
936 static const char *
937 get_key_name(int key_value)
938 {
939         static char key_char[] = "'X'";
940         const char *seq = NULL;
941         int key;
942
943         for (key = 0; key < ARRAY_SIZE(key_table); key++)
944                 if (key_table[key].value == key_value)
945                         seq = key_table[key].name;
946
947         if (seq == NULL &&
948             key_value < 127 &&
949             isprint(key_value)) {
950                 key_char[1] = (char) key_value;
951                 seq = key_char;
952         }
953
954         return seq ? seq : "(no key)";
955 }
956
957 static const char *
958 get_key(enum request request)
959 {
960         static char buf[BUFSIZ];
961         size_t pos = 0;
962         char *sep = "";
963         int i;
964
965         buf[pos] = 0;
966
967         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
968                 struct keybinding *keybinding = &default_keybindings[i];
969
970                 if (keybinding->request != request)
971                         continue;
972
973                 if (!string_format_from(buf, &pos, "%s%s", sep,
974                                         get_key_name(keybinding->alias)))
975                         return "Too many keybindings!";
976                 sep = ", ";
977         }
978
979         return buf;
980 }
981
982 struct run_request {
983         enum keymap keymap;
984         int key;
985         char cmd[SIZEOF_STR];
986 };
987
988 static struct run_request *run_request;
989 static size_t run_requests;
990
991 static enum request
992 add_run_request(enum keymap keymap, int key, int argc, const char **argv)
993 {
994         struct run_request *req;
995         char cmd[SIZEOF_STR];
996         size_t bufpos;
997
998         for (bufpos = 0; argc > 0; argc--, argv++)
999                 if (!string_format_from(cmd, &bufpos, "%s ", *argv))
1000                         return REQ_NONE;
1001
1002         req = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
1003         if (!req)
1004                 return REQ_NONE;
1005
1006         run_request = req;
1007         req = &run_request[run_requests++];
1008         string_copy(req->cmd, cmd);
1009         req->keymap = keymap;
1010         req->key = key;
1011
1012         return REQ_NONE + run_requests;
1013 }
1014
1015 static struct run_request *
1016 get_run_request(enum request request)
1017 {
1018         if (request <= REQ_NONE)
1019                 return NULL;
1020         return &run_request[request - REQ_NONE - 1];
1021 }
1022
1023 static void
1024 add_builtin_run_requests(void)
1025 {
1026         struct {
1027                 enum keymap keymap;
1028                 int key;
1029                 const char *argv[1];
1030         } reqs[] = {
1031                 { KEYMAP_MAIN,    'C', { "git cherry-pick %(commit)" } },
1032                 { KEYMAP_GENERIC, 'G', { "git gc" } },
1033         };
1034         int i;
1035
1036         for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1037                 enum request req;
1038
1039                 req = add_run_request(reqs[i].keymap, reqs[i].key, 1, reqs[i].argv);
1040                 if (req != REQ_NONE)
1041                         add_keybinding(reqs[i].keymap, req, reqs[i].key);
1042         }
1043 }
1044
1045 /*
1046  * User config file handling.
1047  */
1048
1049 static struct int_map color_map[] = {
1050 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
1051         COLOR_MAP(DEFAULT),
1052         COLOR_MAP(BLACK),
1053         COLOR_MAP(BLUE),
1054         COLOR_MAP(CYAN),
1055         COLOR_MAP(GREEN),
1056         COLOR_MAP(MAGENTA),
1057         COLOR_MAP(RED),
1058         COLOR_MAP(WHITE),
1059         COLOR_MAP(YELLOW),
1060 };
1061
1062 #define set_color(color, name) \
1063         set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
1064
1065 static struct int_map attr_map[] = {
1066 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
1067         ATTR_MAP(NORMAL),
1068         ATTR_MAP(BLINK),
1069         ATTR_MAP(BOLD),
1070         ATTR_MAP(DIM),
1071         ATTR_MAP(REVERSE),
1072         ATTR_MAP(STANDOUT),
1073         ATTR_MAP(UNDERLINE),
1074 };
1075
1076 #define set_attribute(attr, name) \
1077         set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
1078
1079 static int   config_lineno;
1080 static bool  config_errors;
1081 static const char *config_msg;
1082
1083 /* Wants: object fgcolor bgcolor [attr] */
1084 static int
1085 option_color_command(int argc, const char *argv[])
1086 {
1087         struct line_info *info;
1088
1089         if (argc != 3 && argc != 4) {
1090                 config_msg = "Wrong number of arguments given to color command";
1091                 return ERR;
1092         }
1093
1094         info = get_line_info(argv[0]);
1095         if (!info) {
1096                 if (!string_enum_compare(argv[0], "main-delim", strlen("main-delim"))) {
1097                         info = get_line_info("delimiter");
1098
1099                 } else if (!string_enum_compare(argv[0], "main-date", strlen("main-date"))) {
1100                         info = get_line_info("date");
1101
1102                 } else {
1103                         config_msg = "Unknown color name";
1104                         return ERR;
1105                 }
1106         }
1107
1108         if (set_color(&info->fg, argv[1]) == ERR ||
1109             set_color(&info->bg, argv[2]) == ERR) {
1110                 config_msg = "Unknown color";
1111                 return ERR;
1112         }
1113
1114         if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
1115                 config_msg = "Unknown attribute";
1116                 return ERR;
1117         }
1118
1119         return OK;
1120 }
1121
1122 static bool parse_bool(const char *s)
1123 {
1124         return (!strcmp(s, "1") || !strcmp(s, "true") ||
1125                 !strcmp(s, "yes")) ? TRUE : FALSE;
1126 }
1127
1128 static int
1129 parse_int(const char *s, int default_value, int min, int max)
1130 {
1131         int value = atoi(s);
1132
1133         return (value < min || value > max) ? default_value : value;
1134 }
1135
1136 /* Wants: name = value */
1137 static int
1138 option_set_command(int argc, const char *argv[])
1139 {
1140         if (argc != 3) {
1141                 config_msg = "Wrong number of arguments given to set command";
1142                 return ERR;
1143         }
1144
1145         if (strcmp(argv[1], "=")) {
1146                 config_msg = "No value assigned";
1147                 return ERR;
1148         }
1149
1150         if (!strcmp(argv[0], "show-author")) {
1151                 opt_author = parse_bool(argv[2]);
1152                 return OK;
1153         }
1154
1155         if (!strcmp(argv[0], "show-date")) {
1156                 opt_date = parse_bool(argv[2]);
1157                 return OK;
1158         }
1159
1160         if (!strcmp(argv[0], "show-rev-graph")) {
1161                 opt_rev_graph = parse_bool(argv[2]);
1162                 return OK;
1163         }
1164
1165         if (!strcmp(argv[0], "show-refs")) {
1166                 opt_show_refs = parse_bool(argv[2]);
1167                 return OK;
1168         }
1169
1170         if (!strcmp(argv[0], "show-line-numbers")) {
1171                 opt_line_number = parse_bool(argv[2]);
1172                 return OK;
1173         }
1174
1175         if (!strcmp(argv[0], "line-graphics")) {
1176                 opt_line_graphics = parse_bool(argv[2]);
1177                 return OK;
1178         }
1179
1180         if (!strcmp(argv[0], "line-number-interval")) {
1181                 opt_num_interval = parse_int(argv[2], opt_num_interval, 1, 1024);
1182                 return OK;
1183         }
1184
1185         if (!strcmp(argv[0], "author-width")) {
1186                 opt_author_cols = parse_int(argv[2], opt_author_cols, 0, 1024);
1187                 return OK;
1188         }
1189
1190         if (!strcmp(argv[0], "tab-size")) {
1191                 opt_tab_size = parse_int(argv[2], opt_tab_size, 1, 1024);
1192                 return OK;
1193         }
1194
1195         if (!strcmp(argv[0], "commit-encoding")) {
1196                 const char *arg = argv[2];
1197                 int arglen = strlen(arg);
1198
1199                 switch (arg[0]) {
1200                 case '"':
1201                 case '\'':
1202                         if (arglen == 1 || arg[arglen - 1] != arg[0]) {
1203                                 config_msg = "Unmatched quotation";
1204                                 return ERR;
1205                         }
1206                         arg += 1; arglen -= 2;
1207                 default:
1208                         string_ncopy(opt_encoding, arg, strlen(arg));
1209                         return OK;
1210                 }
1211         }
1212
1213         config_msg = "Unknown variable name";
1214         return ERR;
1215 }
1216
1217 /* Wants: mode request key */
1218 static int
1219 option_bind_command(int argc, const char *argv[])
1220 {
1221         enum request request;
1222         int keymap;
1223         int key;
1224
1225         if (argc < 3) {
1226                 config_msg = "Wrong number of arguments given to bind command";
1227                 return ERR;
1228         }
1229
1230         if (set_keymap(&keymap, argv[0]) == ERR) {
1231                 config_msg = "Unknown key map";
1232                 return ERR;
1233         }
1234
1235         key = get_key_value(argv[1]);
1236         if (key == ERR) {
1237                 config_msg = "Unknown key";
1238                 return ERR;
1239         }
1240
1241         request = get_request(argv[2]);
1242         if (request == REQ_NONE) {
1243                 const char *obsolete[] = { "cherry-pick" };
1244                 size_t namelen = strlen(argv[2]);
1245                 int i;
1246
1247                 for (i = 0; i < ARRAY_SIZE(obsolete); i++) {
1248                         if (namelen == strlen(obsolete[i]) &&
1249                             !string_enum_compare(obsolete[i], argv[2], namelen)) {
1250                                 config_msg = "Obsolete request name";
1251                                 return ERR;
1252                         }
1253                 }
1254         }
1255         if (request == REQ_NONE && *argv[2]++ == '!')
1256                 request = add_run_request(keymap, key, argc - 2, argv + 2);
1257         if (request == REQ_NONE) {
1258                 config_msg = "Unknown request name";
1259                 return ERR;
1260         }
1261
1262         add_keybinding(keymap, request, key);
1263
1264         return OK;
1265 }
1266
1267 static int
1268 set_option(const char *opt, char *value)
1269 {
1270         const char *argv[SIZEOF_ARG];
1271         int valuelen;
1272         int argc = 0;
1273
1274         /* Tokenize */
1275         while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1276                 argv[argc++] = value;
1277                 value += valuelen;
1278
1279                 /* Nothing more to tokenize or last available token. */
1280                 if (!*value || argc >= ARRAY_SIZE(argv))
1281                         break;
1282
1283                 *value++ = 0;
1284                 while (isspace(*value))
1285                         value++;
1286         }
1287
1288         if (!strcmp(opt, "color"))
1289                 return option_color_command(argc, argv);
1290
1291         if (!strcmp(opt, "set"))
1292                 return option_set_command(argc, argv);
1293
1294         if (!strcmp(opt, "bind"))
1295                 return option_bind_command(argc, argv);
1296
1297         config_msg = "Unknown option command";
1298         return ERR;
1299 }
1300
1301 static int
1302 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1303 {
1304         int status = OK;
1305
1306         config_lineno++;
1307         config_msg = "Internal error";
1308
1309         /* Check for comment markers, since read_properties() will
1310          * only ensure opt and value are split at first " \t". */
1311         optlen = strcspn(opt, "#");
1312         if (optlen == 0)
1313                 return OK;
1314
1315         if (opt[optlen] != 0) {
1316                 config_msg = "No option value";
1317                 status = ERR;
1318
1319         }  else {
1320                 /* Look for comment endings in the value. */
1321                 size_t len = strcspn(value, "#");
1322
1323                 if (len < valuelen) {
1324                         valuelen = len;
1325                         value[valuelen] = 0;
1326                 }
1327
1328                 status = set_option(opt, value);
1329         }
1330
1331         if (status == ERR) {
1332                 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1333                         config_lineno, (int) optlen, opt, config_msg);
1334                 config_errors = TRUE;
1335         }
1336
1337         /* Always keep going if errors are encountered. */
1338         return OK;
1339 }
1340
1341 static void
1342 load_option_file(const char *path)
1343 {
1344         FILE *file;
1345
1346         /* It's ok that the file doesn't exist. */
1347         file = fopen(path, "r");
1348         if (!file)
1349                 return;
1350
1351         config_lineno = 0;
1352         config_errors = FALSE;
1353
1354         if (read_properties(file, " \t", read_option) == ERR ||
1355             config_errors == TRUE)
1356                 fprintf(stderr, "Errors while loading %s.\n", path);
1357 }
1358
1359 static int
1360 load_options(void)
1361 {
1362         const char *home = getenv("HOME");
1363         const char *tigrc_user = getenv("TIGRC_USER");
1364         const char *tigrc_system = getenv("TIGRC_SYSTEM");
1365         char buf[SIZEOF_STR];
1366
1367         add_builtin_run_requests();
1368
1369         if (!tigrc_system) {
1370                 if (!string_format(buf, "%s/tigrc", SYSCONFDIR))
1371                         return ERR;
1372                 tigrc_system = buf;
1373         }
1374         load_option_file(tigrc_system);
1375
1376         if (!tigrc_user) {
1377                 if (!home || !string_format(buf, "%s/.tigrc", home))
1378                         return ERR;
1379                 tigrc_user = buf;
1380         }
1381         load_option_file(tigrc_user);
1382
1383         return OK;
1384 }
1385
1386
1387 /*
1388  * The viewer
1389  */
1390
1391 struct view;
1392 struct view_ops;
1393
1394 /* The display array of active views and the index of the current view. */
1395 static struct view *display[2];
1396 static unsigned int current_view;
1397
1398 /* Reading from the prompt? */
1399 static bool input_mode = FALSE;
1400
1401 #define foreach_displayed_view(view, i) \
1402         for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1403
1404 #define displayed_views()       (display[1] != NULL ? 2 : 1)
1405
1406 /* Current head and commit ID */
1407 static char ref_blob[SIZEOF_REF]        = "";
1408 static char ref_commit[SIZEOF_REF]      = "HEAD";
1409 static char ref_head[SIZEOF_REF]        = "HEAD";
1410
1411 struct view {
1412         const char *name;       /* View name */
1413         const char *cmd_fmt;    /* Default command line format */
1414         const char *cmd_env;    /* Command line set via environment */
1415         const char *id;         /* Points to either of ref_{head,commit,blob} */
1416
1417         struct view_ops *ops;   /* View operations */
1418
1419         enum keymap keymap;     /* What keymap does this view have */
1420         bool git_dir;           /* Whether the view requires a git directory. */
1421
1422         char cmd[SIZEOF_STR];   /* Command buffer */
1423         char ref[SIZEOF_REF];   /* Hovered commit reference */
1424         char vid[SIZEOF_REF];   /* View ID. Set to id member when updating. */
1425
1426         int height, width;      /* The width and height of the main window */
1427         WINDOW *win;            /* The main window */
1428         WINDOW *title;          /* The title window living below the main window */
1429
1430         /* Navigation */
1431         unsigned long offset;   /* Offset of the window top */
1432         unsigned long lineno;   /* Current line number */
1433
1434         /* Searching */
1435         char grep[SIZEOF_STR];  /* Search string */
1436         regex_t *regex;         /* Pre-compiled regex */
1437
1438         /* If non-NULL, points to the view that opened this view. If this view
1439          * is closed tig will switch back to the parent view. */
1440         struct view *parent;
1441
1442         /* Buffering */
1443         size_t lines;           /* Total number of lines */
1444         struct line *line;      /* Line index */
1445         size_t line_alloc;      /* Total number of allocated lines */
1446         size_t line_size;       /* Total number of used lines */
1447         unsigned int digits;    /* Number of digits in the lines member. */
1448
1449         /* Drawing */
1450         struct line *curline;   /* Line currently being drawn. */
1451         enum line_type curtype; /* Attribute currently used for drawing. */
1452         unsigned long col;      /* Column when drawing. */
1453
1454         /* Loading */
1455         FILE *pipe;
1456         time_t start_time;
1457 };
1458
1459 struct view_ops {
1460         /* What type of content being displayed. Used in the title bar. */
1461         const char *type;
1462         /* Open and reads in all view content. */
1463         bool (*open)(struct view *view);
1464         /* Read one line; updates view->line. */
1465         bool (*read)(struct view *view, char *data);
1466         /* Draw one line; @lineno must be < view->height. */
1467         bool (*draw)(struct view *view, struct line *line, unsigned int lineno);
1468         /* Depending on view handle a special requests. */
1469         enum request (*request)(struct view *view, enum request request, struct line *line);
1470         /* Search for regex in a line. */
1471         bool (*grep)(struct view *view, struct line *line);
1472         /* Select line */
1473         void (*select)(struct view *view, struct line *line);
1474 };
1475
1476 static struct view_ops blame_ops;
1477 static struct view_ops blob_ops;
1478 static struct view_ops help_ops;
1479 static struct view_ops log_ops;
1480 static struct view_ops main_ops;
1481 static struct view_ops pager_ops;
1482 static struct view_ops stage_ops;
1483 static struct view_ops status_ops;
1484 static struct view_ops tree_ops;
1485
1486 #define VIEW_STR(name, cmd, env, ref, ops, map, git) \
1487         { name, cmd, #env, ref, ops, map, git }
1488
1489 #define VIEW_(id, name, ops, git, ref) \
1490         VIEW_STR(name, TIG_##id##_CMD,  TIG_##id##_CMD, ref, ops, KEYMAP_##id, git)
1491
1492
1493 static struct view views[] = {
1494         VIEW_(MAIN,   "main",   &main_ops,   TRUE,  ref_head),
1495         VIEW_(DIFF,   "diff",   &pager_ops,  TRUE,  ref_commit),
1496         VIEW_(LOG,    "log",    &log_ops,    TRUE,  ref_head),
1497         VIEW_(TREE,   "tree",   &tree_ops,   TRUE,  ref_commit),
1498         VIEW_(BLOB,   "blob",   &blob_ops,   TRUE,  ref_blob),
1499         VIEW_(BLAME,  "blame",  &blame_ops,  TRUE,  ref_commit),
1500         VIEW_(HELP,   "help",   &help_ops,   FALSE, ""),
1501         VIEW_(PAGER,  "pager",  &pager_ops,  FALSE, "stdin"),
1502         VIEW_(STATUS, "status", &status_ops, TRUE,  ""),
1503         VIEW_(STAGE,  "stage",  &stage_ops,  TRUE,  ""),
1504 };
1505
1506 #define VIEW(req)       (&views[(req) - REQ_OFFSET - 1])
1507 #define VIEW_REQ(view)  ((view) - views + REQ_OFFSET + 1)
1508
1509 #define foreach_view(view, i) \
1510         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1511
1512 #define view_is_displayed(view) \
1513         (view == display[0] || view == display[1])
1514
1515
1516 enum line_graphic {
1517         LINE_GRAPHIC_VLINE
1518 };
1519
1520 static int line_graphics[] = {
1521         /* LINE_GRAPHIC_VLINE: */ '|'
1522 };
1523
1524 static inline void
1525 set_view_attr(struct view *view, enum line_type type)
1526 {
1527         if (!view->curline->selected && view->curtype != type) {
1528                 wattrset(view->win, get_line_attr(type));
1529                 wchgat(view->win, -1, 0, type, NULL);
1530                 view->curtype = type;
1531         }
1532 }
1533
1534 static int
1535 draw_chars(struct view *view, enum line_type type, const char *string,
1536            int max_len, bool use_tilde)
1537 {
1538         int len = 0;
1539         int col = 0;
1540         int trimmed = FALSE;
1541
1542         if (max_len <= 0)
1543                 return 0;
1544
1545         if (opt_utf8) {
1546                 len = utf8_length(string, &col, max_len, &trimmed, use_tilde);
1547         } else {
1548                 col = len = strlen(string);
1549                 if (len > max_len) {
1550                         if (use_tilde) {
1551                                 max_len -= 1;
1552                         }
1553                         col = len = max_len;
1554                         trimmed = TRUE;
1555                 }
1556         }
1557
1558         set_view_attr(view, type);
1559         waddnstr(view->win, string, len);
1560         if (trimmed && use_tilde) {
1561                 set_view_attr(view, LINE_DELIMITER);
1562                 waddch(view->win, '~');
1563                 col++;
1564         }
1565
1566         return col;
1567 }
1568
1569 static int
1570 draw_space(struct view *view, enum line_type type, int max, int spaces)
1571 {
1572         static char space[] = "                    ";
1573         int col = 0;
1574
1575         spaces = MIN(max, spaces);
1576
1577         while (spaces > 0) {
1578                 int len = MIN(spaces, sizeof(space) - 1);
1579
1580                 col += draw_chars(view, type, space, spaces, FALSE);
1581                 spaces -= len;
1582         }
1583
1584         return col;
1585 }
1586
1587 static bool
1588 draw_lineno(struct view *view, unsigned int lineno)
1589 {
1590         char number[10];
1591         int digits3 = view->digits < 3 ? 3 : view->digits;
1592         int max_number = MIN(digits3, STRING_SIZE(number));
1593         int max = view->width - view->col;
1594         int col;
1595
1596         if (max < max_number)
1597                 max_number = max;
1598
1599         lineno += view->offset + 1;
1600         if (lineno == 1 || (lineno % opt_num_interval) == 0) {
1601                 static char fmt[] = "%1ld";
1602
1603                 if (view->digits <= 9)
1604                         fmt[1] = '0' + digits3;
1605
1606                 if (!string_format(number, fmt, lineno))
1607                         number[0] = 0;
1608                 col = draw_chars(view, LINE_LINE_NUMBER, number, max_number, TRUE);
1609         } else {
1610                 col = draw_space(view, LINE_LINE_NUMBER, max_number, max_number);
1611         }
1612
1613         if (col < max) {
1614                 set_view_attr(view, LINE_DEFAULT);
1615                 waddch(view->win, line_graphics[LINE_GRAPHIC_VLINE]);
1616                 col++;
1617         }
1618
1619         if (col < max)
1620                 col += draw_space(view, LINE_DEFAULT, max - col, 1);
1621         view->col += col;
1622
1623         return view->width - view->col <= 0;
1624 }
1625
1626 static bool
1627 draw_text(struct view *view, enum line_type type, const char *string, bool trim)
1628 {
1629         view->col += draw_chars(view, type, string, view->width - view->col, trim);
1630         return view->width - view->col <= 0;
1631 }
1632
1633 static bool
1634 draw_graphic(struct view *view, enum line_type type, chtype graphic[], size_t size)
1635 {
1636         int max = view->width - view->col;
1637         int i;
1638
1639         if (max < size)
1640                 size = max;
1641
1642         set_view_attr(view, type);
1643         /* Using waddch() instead of waddnstr() ensures that
1644          * they'll be rendered correctly for the cursor line. */
1645         for (i = 0; i < size; i++)
1646                 waddch(view->win, graphic[i]);
1647
1648         view->col += size;
1649         if (size < max) {
1650                 waddch(view->win, ' ');
1651                 view->col++;
1652         }
1653
1654         return view->width - view->col <= 0;
1655 }
1656
1657 static bool
1658 draw_field(struct view *view, enum line_type type, const char *text, int len, bool trim)
1659 {
1660         int max = MIN(view->width - view->col, len);
1661         int col;
1662
1663         if (text)
1664                 col = draw_chars(view, type, text, max - 1, trim);
1665         else
1666                 col = draw_space(view, type, max - 1, max - 1);
1667
1668         view->col += col + draw_space(view, LINE_DEFAULT, max - col, max - col);
1669         return view->width - view->col <= 0;
1670 }
1671
1672 static bool
1673 draw_date(struct view *view, struct tm *time)
1674 {
1675         char buf[DATE_COLS];
1676         char *date;
1677         int timelen = 0;
1678
1679         if (time)
1680                 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, time);
1681         date = timelen ? buf : NULL;
1682
1683         return draw_field(view, LINE_DATE, date, DATE_COLS, FALSE);
1684 }
1685
1686 static bool
1687 draw_view_line(struct view *view, unsigned int lineno)
1688 {
1689         struct line *line;
1690         bool selected = (view->offset + lineno == view->lineno);
1691         bool draw_ok;
1692
1693         assert(view_is_displayed(view));
1694
1695         if (view->offset + lineno >= view->lines)
1696                 return FALSE;
1697
1698         line = &view->line[view->offset + lineno];
1699
1700         wmove(view->win, lineno, 0);
1701         view->col = 0;
1702         view->curline = line;
1703         view->curtype = LINE_NONE;
1704         line->selected = FALSE;
1705
1706         if (selected) {
1707                 set_view_attr(view, LINE_CURSOR);
1708                 line->selected = TRUE;
1709                 view->ops->select(view, line);
1710         } else if (line->selected) {
1711                 wclrtoeol(view->win);
1712         }
1713
1714         scrollok(view->win, FALSE);
1715         draw_ok = view->ops->draw(view, line, lineno);
1716         scrollok(view->win, TRUE);
1717
1718         return draw_ok;
1719 }
1720
1721 static void
1722 redraw_view_dirty(struct view *view)
1723 {
1724         bool dirty = FALSE;
1725         int lineno;
1726
1727         for (lineno = 0; lineno < view->height; lineno++) {
1728                 struct line *line = &view->line[view->offset + lineno];
1729
1730                 if (!line->dirty)
1731                         continue;
1732                 line->dirty = 0;
1733                 dirty = TRUE;
1734                 if (!draw_view_line(view, lineno))
1735                         break;
1736         }
1737
1738         if (!dirty)
1739                 return;
1740         redrawwin(view->win);
1741         if (input_mode)
1742                 wnoutrefresh(view->win);
1743         else
1744                 wrefresh(view->win);
1745 }
1746
1747 static void
1748 redraw_view_from(struct view *view, int lineno)
1749 {
1750         assert(0 <= lineno && lineno < view->height);
1751
1752         for (; lineno < view->height; lineno++) {
1753                 if (!draw_view_line(view, lineno))
1754                         break;
1755         }
1756
1757         redrawwin(view->win);
1758         if (input_mode)
1759                 wnoutrefresh(view->win);
1760         else
1761                 wrefresh(view->win);
1762 }
1763
1764 static void
1765 redraw_view(struct view *view)
1766 {
1767         wclear(view->win);
1768         redraw_view_from(view, 0);
1769 }
1770
1771
1772 static void
1773 update_view_title(struct view *view)
1774 {
1775         char buf[SIZEOF_STR];
1776         char state[SIZEOF_STR];
1777         size_t bufpos = 0, statelen = 0;
1778
1779         assert(view_is_displayed(view));
1780
1781         if (view != VIEW(REQ_VIEW_STATUS) && (view->lines || view->pipe)) {
1782                 unsigned int view_lines = view->offset + view->height;
1783                 unsigned int lines = view->lines
1784                                    ? MIN(view_lines, view->lines) * 100 / view->lines
1785                                    : 0;
1786
1787                 string_format_from(state, &statelen, "- %s %d of %d (%d%%)",
1788                                    view->ops->type,
1789                                    view->lineno + 1,
1790                                    view->lines,
1791                                    lines);
1792
1793                 if (view->pipe) {
1794                         time_t secs = time(NULL) - view->start_time;
1795
1796                         /* Three git seconds are a long time ... */
1797                         if (secs > 2)
1798                                 string_format_from(state, &statelen, " %lds", secs);
1799                 }
1800         }
1801
1802         string_format_from(buf, &bufpos, "[%s]", view->name);
1803         if (*view->ref && bufpos < view->width) {
1804                 size_t refsize = strlen(view->ref);
1805                 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
1806
1807                 if (minsize < view->width)
1808                         refsize = view->width - minsize + 7;
1809                 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
1810         }
1811
1812         if (statelen && bufpos < view->width) {
1813                 string_format_from(buf, &bufpos, " %s", state);
1814         }
1815
1816         if (view == display[current_view])
1817                 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1818         else
1819                 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1820
1821         mvwaddnstr(view->title, 0, 0, buf, bufpos);
1822         wclrtoeol(view->title);
1823         wmove(view->title, 0, view->width - 1);
1824
1825         if (input_mode)
1826                 wnoutrefresh(view->title);
1827         else
1828                 wrefresh(view->title);
1829 }
1830
1831 static void
1832 resize_display(void)
1833 {
1834         int offset, i;
1835         struct view *base = display[0];
1836         struct view *view = display[1] ? display[1] : display[0];
1837
1838         /* Setup window dimensions */
1839
1840         getmaxyx(stdscr, base->height, base->width);
1841
1842         /* Make room for the status window. */
1843         base->height -= 1;
1844
1845         if (view != base) {
1846                 /* Horizontal split. */
1847                 view->width   = base->width;
1848                 view->height  = SCALE_SPLIT_VIEW(base->height);
1849                 base->height -= view->height;
1850
1851                 /* Make room for the title bar. */
1852                 view->height -= 1;
1853         }
1854
1855         /* Make room for the title bar. */
1856         base->height -= 1;
1857
1858         offset = 0;
1859
1860         foreach_displayed_view (view, i) {
1861                 if (!view->win) {
1862                         view->win = newwin(view->height, 0, offset, 0);
1863                         if (!view->win)
1864                                 die("Failed to create %s view", view->name);
1865
1866                         scrollok(view->win, TRUE);
1867
1868                         view->title = newwin(1, 0, offset + view->height, 0);
1869                         if (!view->title)
1870                                 die("Failed to create title window");
1871
1872                 } else {
1873                         wresize(view->win, view->height, view->width);
1874                         mvwin(view->win,   offset, 0);
1875                         mvwin(view->title, offset + view->height, 0);
1876                 }
1877
1878                 offset += view->height + 1;
1879         }
1880 }
1881
1882 static void
1883 redraw_display(void)
1884 {
1885         struct view *view;
1886         int i;
1887
1888         foreach_displayed_view (view, i) {
1889                 redraw_view(view);
1890                 update_view_title(view);
1891         }
1892 }
1893
1894 static void
1895 update_display_cursor(struct view *view)
1896 {
1897         /* Move the cursor to the right-most column of the cursor line.
1898          *
1899          * XXX: This could turn out to be a bit expensive, but it ensures that
1900          * the cursor does not jump around. */
1901         if (view->lines) {
1902                 wmove(view->win, view->lineno - view->offset, view->width - 1);
1903                 wrefresh(view->win);
1904         }
1905 }
1906
1907 /*
1908  * Navigation
1909  */
1910
1911 /* Scrolling backend */
1912 static void
1913 do_scroll_view(struct view *view, int lines)
1914 {
1915         bool redraw_current_line = FALSE;
1916
1917         /* The rendering expects the new offset. */
1918         view->offset += lines;
1919
1920         assert(0 <= view->offset && view->offset < view->lines);
1921         assert(lines);
1922
1923         /* Move current line into the view. */
1924         if (view->lineno < view->offset) {
1925                 view->lineno = view->offset;
1926                 redraw_current_line = TRUE;
1927         } else if (view->lineno >= view->offset + view->height) {
1928                 view->lineno = view->offset + view->height - 1;
1929                 redraw_current_line = TRUE;
1930         }
1931
1932         assert(view->offset <= view->lineno && view->lineno < view->lines);
1933
1934         /* Redraw the whole screen if scrolling is pointless. */
1935         if (view->height < ABS(lines)) {
1936                 redraw_view(view);
1937
1938         } else {
1939                 int line = lines > 0 ? view->height - lines : 0;
1940                 int end = line + ABS(lines);
1941
1942                 wscrl(view->win, lines);
1943
1944                 for (; line < end; line++) {
1945                         if (!draw_view_line(view, line))
1946                                 break;
1947                 }
1948
1949                 if (redraw_current_line)
1950                         draw_view_line(view, view->lineno - view->offset);
1951         }
1952
1953         redrawwin(view->win);
1954         wrefresh(view->win);
1955         report("");
1956 }
1957
1958 /* Scroll frontend */
1959 static void
1960 scroll_view(struct view *view, enum request request)
1961 {
1962         int lines = 1;
1963
1964         assert(view_is_displayed(view));
1965
1966         switch (request) {
1967         case REQ_SCROLL_PAGE_DOWN:
1968                 lines = view->height;
1969         case REQ_SCROLL_LINE_DOWN:
1970                 if (view->offset + lines > view->lines)
1971                         lines = view->lines - view->offset;
1972
1973                 if (lines == 0 || view->offset + view->height >= view->lines) {
1974                         report("Cannot scroll beyond the last line");
1975                         return;
1976                 }
1977                 break;
1978
1979         case REQ_SCROLL_PAGE_UP:
1980                 lines = view->height;
1981         case REQ_SCROLL_LINE_UP:
1982                 if (lines > view->offset)
1983                         lines = view->offset;
1984
1985                 if (lines == 0) {
1986                         report("Cannot scroll beyond the first line");
1987                         return;
1988                 }
1989
1990                 lines = -lines;
1991                 break;
1992
1993         default:
1994                 die("request %d not handled in switch", request);
1995         }
1996
1997         do_scroll_view(view, lines);
1998 }
1999
2000 /* Cursor moving */
2001 static void
2002 move_view(struct view *view, enum request request)
2003 {
2004         int scroll_steps = 0;
2005         int steps;
2006
2007         switch (request) {
2008         case REQ_MOVE_FIRST_LINE:
2009                 steps = -view->lineno;
2010                 break;
2011
2012         case REQ_MOVE_LAST_LINE:
2013                 steps = view->lines - view->lineno - 1;
2014                 break;
2015
2016         case REQ_MOVE_PAGE_UP:
2017                 steps = view->height > view->lineno
2018                       ? -view->lineno : -view->height;
2019                 break;
2020
2021         case REQ_MOVE_PAGE_DOWN:
2022                 steps = view->lineno + view->height >= view->lines
2023                       ? view->lines - view->lineno - 1 : view->height;
2024                 break;
2025
2026         case REQ_MOVE_UP:
2027                 steps = -1;
2028                 break;
2029
2030         case REQ_MOVE_DOWN:
2031                 steps = 1;
2032                 break;
2033
2034         default:
2035                 die("request %d not handled in switch", request);
2036         }
2037
2038         if (steps <= 0 && view->lineno == 0) {
2039                 report("Cannot move beyond the first line");
2040                 return;
2041
2042         } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
2043                 report("Cannot move beyond the last line");
2044                 return;
2045         }
2046
2047         /* Move the current line */
2048         view->lineno += steps;
2049         assert(0 <= view->lineno && view->lineno < view->lines);
2050
2051         /* Check whether the view needs to be scrolled */
2052         if (view->lineno < view->offset ||
2053             view->lineno >= view->offset + view->height) {
2054                 scroll_steps = steps;
2055                 if (steps < 0 && -steps > view->offset) {
2056                         scroll_steps = -view->offset;
2057
2058                 } else if (steps > 0) {
2059                         if (view->lineno == view->lines - 1 &&
2060                             view->lines > view->height) {
2061                                 scroll_steps = view->lines - view->offset - 1;
2062                                 if (scroll_steps >= view->height)
2063                                         scroll_steps -= view->height - 1;
2064                         }
2065                 }
2066         }
2067
2068         if (!view_is_displayed(view)) {
2069                 view->offset += scroll_steps;
2070                 assert(0 <= view->offset && view->offset < view->lines);
2071                 view->ops->select(view, &view->line[view->lineno]);
2072                 return;
2073         }
2074
2075         /* Repaint the old "current" line if we be scrolling */
2076         if (ABS(steps) < view->height)
2077                 draw_view_line(view, view->lineno - steps - view->offset);
2078
2079         if (scroll_steps) {
2080                 do_scroll_view(view, scroll_steps);
2081                 return;
2082         }
2083
2084         /* Draw the current line */
2085         draw_view_line(view, view->lineno - view->offset);
2086
2087         redrawwin(view->win);
2088         wrefresh(view->win);
2089         report("");
2090 }
2091
2092
2093 /*
2094  * Searching
2095  */
2096
2097 static void search_view(struct view *view, enum request request);
2098
2099 static bool
2100 find_next_line(struct view *view, unsigned long lineno, struct line *line)
2101 {
2102         assert(view_is_displayed(view));
2103
2104         if (!view->ops->grep(view, line))
2105                 return FALSE;
2106
2107         if (lineno - view->offset >= view->height) {
2108                 view->offset = lineno;
2109                 view->lineno = lineno;
2110                 redraw_view(view);
2111
2112         } else {
2113                 unsigned long old_lineno = view->lineno - view->offset;
2114
2115                 view->lineno = lineno;
2116                 draw_view_line(view, old_lineno);
2117
2118                 draw_view_line(view, view->lineno - view->offset);
2119                 redrawwin(view->win);
2120                 wrefresh(view->win);
2121         }
2122
2123         report("Line %ld matches '%s'", lineno + 1, view->grep);
2124         return TRUE;
2125 }
2126
2127 static void
2128 find_next(struct view *view, enum request request)
2129 {
2130         unsigned long lineno = view->lineno;
2131         int direction;
2132
2133         if (!*view->grep) {
2134                 if (!*opt_search)
2135                         report("No previous search");
2136                 else
2137                         search_view(view, request);
2138                 return;
2139         }
2140
2141         switch (request) {
2142         case REQ_SEARCH:
2143         case REQ_FIND_NEXT:
2144                 direction = 1;
2145                 break;
2146
2147         case REQ_SEARCH_BACK:
2148         case REQ_FIND_PREV:
2149                 direction = -1;
2150                 break;
2151
2152         default:
2153                 return;
2154         }
2155
2156         if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
2157                 lineno += direction;
2158
2159         /* Note, lineno is unsigned long so will wrap around in which case it
2160          * will become bigger than view->lines. */
2161         for (; lineno < view->lines; lineno += direction) {
2162                 struct line *line = &view->line[lineno];
2163
2164                 if (find_next_line(view, lineno, line))
2165                         return;
2166         }
2167
2168         report("No match found for '%s'", view->grep);
2169 }
2170
2171 static void
2172 search_view(struct view *view, enum request request)
2173 {
2174         int regex_err;
2175
2176         if (view->regex) {
2177                 regfree(view->regex);
2178                 *view->grep = 0;
2179         } else {
2180                 view->regex = calloc(1, sizeof(*view->regex));
2181                 if (!view->regex)
2182                         return;
2183         }
2184
2185         regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
2186         if (regex_err != 0) {
2187                 char buf[SIZEOF_STR] = "unknown error";
2188
2189                 regerror(regex_err, view->regex, buf, sizeof(buf));
2190                 report("Search failed: %s", buf);
2191                 return;
2192         }
2193
2194         string_copy(view->grep, opt_search);
2195
2196         find_next(view, request);
2197 }
2198
2199 /*
2200  * Incremental updating
2201  */
2202
2203 static void
2204 reset_view(struct view *view)
2205 {
2206         int i;
2207
2208         for (i = 0; i < view->lines; i++)
2209                 free(view->line[i].data);
2210         free(view->line);
2211
2212         view->line = NULL;
2213         view->offset = 0;
2214         view->lines  = 0;
2215         view->lineno = 0;
2216         view->line_size = 0;
2217         view->line_alloc = 0;
2218         view->vid[0] = 0;
2219 }
2220
2221 static void
2222 end_update(struct view *view, bool force)
2223 {
2224         if (!view->pipe)
2225                 return;
2226         while (!view->ops->read(view, NULL))
2227                 if (!force)
2228                         return;
2229         set_nonblocking_input(FALSE);
2230         if (view->pipe == stdin)
2231                 fclose(view->pipe);
2232         else
2233                 pclose(view->pipe);
2234         view->pipe = NULL;
2235 }
2236
2237 static void
2238 setup_update(struct view *view, const char *vid)
2239 {
2240         set_nonblocking_input(TRUE);
2241         reset_view(view);
2242         string_copy_rev(view->vid, vid);
2243         view->start_time = time(NULL);
2244 }
2245
2246 static bool
2247 begin_update(struct view *view, bool refresh)
2248 {
2249         if (opt_cmd[0]) {
2250                 string_copy(view->cmd, opt_cmd);
2251                 opt_cmd[0] = 0;
2252                 /* When running random commands, initially show the
2253                  * command in the title. However, it maybe later be
2254                  * overwritten if a commit line is selected. */
2255                 if (view == VIEW(REQ_VIEW_PAGER))
2256                         string_copy(view->ref, view->cmd);
2257                 else
2258                         view->ref[0] = 0;
2259
2260         } else if (view == VIEW(REQ_VIEW_TREE)) {
2261                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2262                 char path[SIZEOF_STR];
2263
2264                 if (strcmp(view->vid, view->id))
2265                         opt_path[0] = path[0] = 0;
2266                 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
2267                         return FALSE;
2268
2269                 if (!string_format(view->cmd, format, view->id, path))
2270                         return FALSE;
2271
2272         } else if (!refresh) {
2273                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2274                 const char *id = view->id;
2275
2276                 if (!string_format(view->cmd, format, id, id, id, id, id))
2277                         return FALSE;
2278
2279                 /* Put the current ref_* value to the view title ref
2280                  * member. This is needed by the blob view. Most other
2281                  * views sets it automatically after loading because the
2282                  * first line is a commit line. */
2283                 string_copy_rev(view->ref, view->id);
2284         }
2285
2286         /* Special case for the pager view. */
2287         if (opt_pipe) {
2288                 view->pipe = opt_pipe;
2289                 opt_pipe = NULL;
2290         } else {
2291                 view->pipe = popen(view->cmd, "r");
2292         }
2293
2294         if (!view->pipe)
2295                 return FALSE;
2296
2297         setup_update(view, view->id);
2298
2299         return TRUE;
2300 }
2301
2302 #define ITEM_CHUNK_SIZE 256
2303 static void *
2304 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2305 {
2306         size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2307         size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2308
2309         if (mem == NULL || num_chunks != num_chunks_new) {
2310                 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2311                 mem = realloc(mem, *size * item_size);
2312         }
2313
2314         return mem;
2315 }
2316
2317 static struct line *
2318 realloc_lines(struct view *view, size_t line_size)
2319 {
2320         size_t alloc = view->line_alloc;
2321         struct line *tmp = realloc_items(view->line, &alloc, line_size,
2322                                          sizeof(*view->line));
2323
2324         if (!tmp)
2325                 return NULL;
2326
2327         view->line = tmp;
2328         view->line_alloc = alloc;
2329         view->line_size = line_size;
2330         return view->line;
2331 }
2332
2333 static bool
2334 update_view(struct view *view)
2335 {
2336         char in_buffer[BUFSIZ];
2337         char out_buffer[BUFSIZ * 2];
2338         char *line;
2339         /* The number of lines to read. If too low it will cause too much
2340          * redrawing (and possible flickering), if too high responsiveness
2341          * will suffer. */
2342         unsigned long lines = view->height;
2343         int redraw_from = -1;
2344
2345         if (!view->pipe)
2346                 return TRUE;
2347
2348         /* Only redraw if lines are visible. */
2349         if (view->offset + view->height >= view->lines)
2350                 redraw_from = view->lines - view->offset;
2351
2352         /* FIXME: This is probably not perfect for backgrounded views. */
2353         if (!realloc_lines(view, view->lines + lines))
2354                 goto alloc_error;
2355
2356         while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
2357                 size_t linelen = strlen(line);
2358
2359                 if (linelen)
2360                         line[linelen - 1] = 0;
2361
2362                 if (opt_iconv != ICONV_NONE) {
2363                         ICONV_CONST char *inbuf = line;
2364                         size_t inlen = linelen;
2365
2366                         char *outbuf = out_buffer;
2367                         size_t outlen = sizeof(out_buffer);
2368
2369                         size_t ret;
2370
2371                         ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2372                         if (ret != (size_t) -1) {
2373                                 line = out_buffer;
2374                                 linelen = strlen(out_buffer);
2375                         }
2376                 }
2377
2378                 if (!view->ops->read(view, line))
2379                         goto alloc_error;
2380
2381                 if (lines-- == 1)
2382                         break;
2383         }
2384
2385         {
2386                 int digits;
2387
2388                 lines = view->lines;
2389                 for (digits = 0; lines; digits++)
2390                         lines /= 10;
2391
2392                 /* Keep the displayed view in sync with line number scaling. */
2393                 if (digits != view->digits) {
2394                         view->digits = digits;
2395                         redraw_from = 0;
2396                 }
2397         }
2398
2399         if (ferror(view->pipe) && errno != 0) {
2400                 report("Failed to read: %s", strerror(errno));
2401                 end_update(view, TRUE);
2402
2403         } else if (feof(view->pipe)) {
2404                 report("");
2405                 end_update(view, FALSE);
2406         }
2407
2408         if (!view_is_displayed(view))
2409                 return TRUE;
2410
2411         if (view == VIEW(REQ_VIEW_TREE)) {
2412                 /* Clear the view and redraw everything since the tree sorting
2413                  * might have rearranged things. */
2414                 redraw_view(view);
2415
2416         } else if (redraw_from >= 0) {
2417                 /* If this is an incremental update, redraw the previous line
2418                  * since for commits some members could have changed when
2419                  * loading the main view. */
2420                 if (redraw_from > 0)
2421                         redraw_from--;
2422
2423                 /* Since revision graph visualization requires knowledge
2424                  * about the parent commit, it causes a further one-off
2425                  * needed to be redrawn for incremental updates. */
2426                 if (redraw_from > 0 && opt_rev_graph)
2427                         redraw_from--;
2428
2429                 /* Incrementally draw avoids flickering. */
2430                 redraw_view_from(view, redraw_from);
2431         }
2432
2433         if (view == VIEW(REQ_VIEW_BLAME))
2434                 redraw_view_dirty(view);
2435
2436         /* Update the title _after_ the redraw so that if the redraw picks up a
2437          * commit reference in view->ref it'll be available here. */
2438         update_view_title(view);
2439         return TRUE;
2440
2441 alloc_error:
2442         report("Allocation failure");
2443         end_update(view, TRUE);
2444         return FALSE;
2445 }
2446
2447 static struct line *
2448 add_line_data(struct view *view, void *data, enum line_type type)
2449 {
2450         struct line *line = &view->line[view->lines++];
2451
2452         memset(line, 0, sizeof(*line));
2453         line->type = type;
2454         line->data = data;
2455
2456         return line;
2457 }
2458
2459 static struct line *
2460 add_line_text(struct view *view, const char *text, enum line_type type)
2461 {
2462         char *data = text ? strdup(text) : NULL;
2463
2464         return data ? add_line_data(view, data, type) : NULL;
2465 }
2466
2467
2468 /*
2469  * View opening
2470  */
2471
2472 enum open_flags {
2473         OPEN_DEFAULT = 0,       /* Use default view switching. */
2474         OPEN_SPLIT = 1,         /* Split current view. */
2475         OPEN_BACKGROUNDED = 2,  /* Backgrounded. */
2476         OPEN_RELOAD = 4,        /* Reload view even if it is the current. */
2477         OPEN_NOMAXIMIZE = 8,    /* Do not maximize the current view. */
2478         OPEN_REFRESH = 16,      /* Refresh view using previous command. */
2479 };
2480
2481 static void
2482 open_view(struct view *prev, enum request request, enum open_flags flags)
2483 {
2484         bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2485         bool split = !!(flags & OPEN_SPLIT);
2486         bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH));
2487         bool nomaximize = !!(flags & (OPEN_NOMAXIMIZE | OPEN_REFRESH));
2488         struct view *view = VIEW(request);
2489         int nviews = displayed_views();
2490         struct view *base_view = display[0];
2491
2492         if (view == prev && nviews == 1 && !reload) {
2493                 report("Already in %s view", view->name);
2494                 return;
2495         }
2496
2497         if (view->git_dir && !opt_git_dir[0]) {
2498                 report("The %s view is disabled in pager view", view->name);
2499                 return;
2500         }
2501
2502         if (split) {
2503                 display[1] = view;
2504                 if (!backgrounded)
2505                         current_view = 1;
2506         } else if (!nomaximize) {
2507                 /* Maximize the current view. */
2508                 memset(display, 0, sizeof(display));
2509                 current_view = 0;
2510                 display[current_view] = view;
2511         }
2512
2513         /* Resize the view when switching between split- and full-screen,
2514          * or when switching between two different full-screen views. */
2515         if (nviews != displayed_views() ||
2516             (nviews == 1 && base_view != display[0]))
2517                 resize_display();
2518
2519         if (view->pipe)
2520                 end_update(view, TRUE);
2521
2522         if (view->ops->open) {
2523                 if (!view->ops->open(view)) {
2524                         report("Failed to load %s view", view->name);
2525                         return;
2526                 }
2527
2528         } else if ((reload || strcmp(view->vid, view->id)) &&
2529                    !begin_update(view, flags & OPEN_REFRESH)) {
2530                 report("Failed to load %s view", view->name);
2531                 return;
2532         }
2533
2534         if (split && prev->lineno - prev->offset >= prev->height) {
2535                 /* Take the title line into account. */
2536                 int lines = prev->lineno - prev->offset - prev->height + 1;
2537
2538                 /* Scroll the view that was split if the current line is
2539                  * outside the new limited view. */
2540                 do_scroll_view(prev, lines);
2541         }
2542
2543         if (prev && view != prev) {
2544                 if (split && !backgrounded) {
2545                         /* "Blur" the previous view. */
2546                         update_view_title(prev);
2547                 }
2548
2549                 view->parent = prev;
2550         }
2551
2552         if (view->pipe && view->lines == 0) {
2553                 /* Clear the old view and let the incremental updating refill
2554                  * the screen. */
2555                 werase(view->win);
2556                 report("");
2557         } else if (view_is_displayed(view)) {
2558                 redraw_view(view);
2559                 report("");
2560         }
2561
2562         /* If the view is backgrounded the above calls to report()
2563          * won't redraw the view title. */
2564         if (backgrounded)
2565                 update_view_title(view);
2566 }
2567
2568 static bool
2569 run_confirm(const char *cmd, const char *prompt)
2570 {
2571         bool confirmation = prompt_yesno(prompt);
2572
2573         if (confirmation)
2574                 system(cmd);
2575
2576         return confirmation;
2577 }
2578
2579 static void
2580 open_external_viewer(const char *cmd)
2581 {
2582         def_prog_mode();           /* save current tty modes */
2583         endwin();                  /* restore original tty modes */
2584         system(cmd);
2585         fprintf(stderr, "Press Enter to continue");
2586         getc(opt_tty);
2587         reset_prog_mode();
2588         redraw_display();
2589 }
2590
2591 static void
2592 open_mergetool(const char *file)
2593 {
2594         char cmd[SIZEOF_STR];
2595         char file_sq[SIZEOF_STR];
2596
2597         if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2598             string_format(cmd, "git mergetool %s", file_sq)) {
2599                 open_external_viewer(cmd);
2600         }
2601 }
2602
2603 static void
2604 open_editor(bool from_root, const char *file)
2605 {
2606         char cmd[SIZEOF_STR];
2607         char file_sq[SIZEOF_STR];
2608         const char *editor;
2609         char *prefix = from_root ? opt_cdup : "";
2610
2611         editor = getenv("GIT_EDITOR");
2612         if (!editor && *opt_editor)
2613                 editor = opt_editor;
2614         if (!editor)
2615                 editor = getenv("VISUAL");
2616         if (!editor)
2617                 editor = getenv("EDITOR");
2618         if (!editor)
2619                 editor = "vi";
2620
2621         if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2622             string_format(cmd, "%s %s%s", editor, prefix, file_sq)) {
2623                 open_external_viewer(cmd);
2624         }
2625 }
2626
2627 static void
2628 open_run_request(enum request request)
2629 {
2630         struct run_request *req = get_run_request(request);
2631         char buf[SIZEOF_STR * 2];
2632         size_t bufpos;
2633         char *cmd;
2634
2635         if (!req) {
2636                 report("Unknown run request");
2637                 return;
2638         }
2639
2640         bufpos = 0;
2641         cmd = req->cmd;
2642
2643         while (cmd) {
2644                 char *next = strstr(cmd, "%(");
2645                 int len = next - cmd;
2646                 char *value;
2647
2648                 if (!next) {
2649                         len = strlen(cmd);
2650                         value = "";
2651
2652                 } else if (!strncmp(next, "%(head)", 7)) {
2653                         value = ref_head;
2654
2655                 } else if (!strncmp(next, "%(commit)", 9)) {
2656                         value = ref_commit;
2657
2658                 } else if (!strncmp(next, "%(blob)", 7)) {
2659                         value = ref_blob;
2660
2661                 } else {
2662                         report("Unknown replacement in run request: `%s`", req->cmd);
2663                         return;
2664                 }
2665
2666                 if (!string_format_from(buf, &bufpos, "%.*s%s", len, cmd, value))
2667                         return;
2668
2669                 if (next)
2670                         next = strchr(next, ')') + 1;
2671                 cmd = next;
2672         }
2673
2674         open_external_viewer(buf);
2675 }
2676
2677 /*
2678  * User request switch noodle
2679  */
2680
2681 static int
2682 view_driver(struct view *view, enum request request)
2683 {
2684         int i;
2685
2686         if (request == REQ_NONE) {
2687                 doupdate();
2688                 return TRUE;
2689         }
2690
2691         if (request > REQ_NONE) {
2692                 open_run_request(request);
2693                 /* FIXME: When all views can refresh always do this. */
2694                 if (view == VIEW(REQ_VIEW_STATUS) ||
2695                     view == VIEW(REQ_VIEW_MAIN) ||
2696                     view == VIEW(REQ_VIEW_LOG) ||
2697                     view == VIEW(REQ_VIEW_STAGE))
2698                         request = REQ_REFRESH;
2699                 else
2700                         return TRUE;
2701         }
2702
2703         if (view && view->lines) {
2704                 request = view->ops->request(view, request, &view->line[view->lineno]);
2705                 if (request == REQ_NONE)
2706                         return TRUE;
2707         }
2708
2709         switch (request) {
2710         case REQ_MOVE_UP:
2711         case REQ_MOVE_DOWN:
2712         case REQ_MOVE_PAGE_UP:
2713         case REQ_MOVE_PAGE_DOWN:
2714         case REQ_MOVE_FIRST_LINE:
2715         case REQ_MOVE_LAST_LINE:
2716                 move_view(view, request);
2717                 break;
2718
2719         case REQ_SCROLL_LINE_DOWN:
2720         case REQ_SCROLL_LINE_UP:
2721         case REQ_SCROLL_PAGE_DOWN:
2722         case REQ_SCROLL_PAGE_UP:
2723                 scroll_view(view, request);
2724                 break;
2725
2726         case REQ_VIEW_BLAME:
2727                 if (!opt_file[0]) {
2728                         report("No file chosen, press %s to open tree view",
2729                                get_key(REQ_VIEW_TREE));
2730                         break;
2731                 }
2732                 open_view(view, request, OPEN_DEFAULT);
2733                 break;
2734
2735         case REQ_VIEW_BLOB:
2736                 if (!ref_blob[0]) {
2737                         report("No file chosen, press %s to open tree view",
2738                                get_key(REQ_VIEW_TREE));
2739                         break;
2740                 }
2741                 open_view(view, request, OPEN_DEFAULT);
2742                 break;
2743
2744         case REQ_VIEW_PAGER:
2745                 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2746                         report("No pager content, press %s to run command from prompt",
2747                                get_key(REQ_PROMPT));
2748                         break;
2749                 }
2750                 open_view(view, request, OPEN_DEFAULT);
2751                 break;
2752
2753         case REQ_VIEW_STAGE:
2754                 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2755                         report("No stage content, press %s to open the status view and choose file",
2756                                get_key(REQ_VIEW_STATUS));
2757                         break;
2758                 }
2759                 open_view(view, request, OPEN_DEFAULT);
2760                 break;
2761
2762         case REQ_VIEW_STATUS:
2763                 if (opt_is_inside_work_tree == FALSE) {
2764                         report("The status view requires a working tree");
2765                         break;
2766                 }
2767                 open_view(view, request, OPEN_DEFAULT);
2768                 break;
2769
2770         case REQ_VIEW_MAIN:
2771         case REQ_VIEW_DIFF:
2772         case REQ_VIEW_LOG:
2773         case REQ_VIEW_TREE:
2774         case REQ_VIEW_HELP:
2775                 open_view(view, request, OPEN_DEFAULT);
2776                 break;
2777
2778         case REQ_NEXT:
2779         case REQ_PREVIOUS:
2780                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2781
2782                 if ((view == VIEW(REQ_VIEW_DIFF) &&
2783                      view->parent == VIEW(REQ_VIEW_MAIN)) ||
2784                    (view == VIEW(REQ_VIEW_DIFF) &&
2785                      view->parent == VIEW(REQ_VIEW_BLAME)) ||
2786                    (view == VIEW(REQ_VIEW_STAGE) &&
2787                      view->parent == VIEW(REQ_VIEW_STATUS)) ||
2788                    (view == VIEW(REQ_VIEW_BLOB) &&
2789                      view->parent == VIEW(REQ_VIEW_TREE))) {
2790                         int line;
2791
2792                         view = view->parent;
2793                         line = view->lineno;
2794                         move_view(view, request);
2795                         if (view_is_displayed(view))
2796                                 update_view_title(view);
2797                         if (line != view->lineno)
2798                                 view->ops->request(view, REQ_ENTER,
2799                                                    &view->line[view->lineno]);
2800
2801                 } else {
2802                         move_view(view, request);
2803                 }
2804                 break;
2805
2806         case REQ_VIEW_NEXT:
2807         {
2808                 int nviews = displayed_views();
2809                 int next_view = (current_view + 1) % nviews;
2810
2811                 if (next_view == current_view) {
2812                         report("Only one view is displayed");
2813                         break;
2814                 }
2815
2816                 current_view = next_view;
2817                 /* Blur out the title of the previous view. */
2818                 update_view_title(view);
2819                 report("");
2820                 break;
2821         }
2822         case REQ_REFRESH:
2823                 report("Refreshing is not yet supported for the %s view", view->name);
2824                 break;
2825
2826         case REQ_MAXIMIZE:
2827                 if (displayed_views() == 2)
2828                         open_view(view, VIEW_REQ(view), OPEN_DEFAULT);
2829                 break;
2830
2831         case REQ_TOGGLE_LINENO:
2832                 opt_line_number = !opt_line_number;
2833                 redraw_display();
2834                 break;
2835
2836         case REQ_TOGGLE_DATE:
2837                 opt_date = !opt_date;
2838                 redraw_display();
2839                 break;
2840
2841         case REQ_TOGGLE_AUTHOR:
2842                 opt_author = !opt_author;
2843                 redraw_display();
2844                 break;
2845
2846         case REQ_TOGGLE_REV_GRAPH:
2847                 opt_rev_graph = !opt_rev_graph;
2848                 redraw_display();
2849                 break;
2850
2851         case REQ_TOGGLE_REFS:
2852                 opt_show_refs = !opt_show_refs;
2853                 redraw_display();
2854                 break;
2855
2856         case REQ_SEARCH:
2857         case REQ_SEARCH_BACK:
2858                 search_view(view, request);
2859                 break;
2860
2861         case REQ_FIND_NEXT:
2862         case REQ_FIND_PREV:
2863                 find_next(view, request);
2864                 break;
2865
2866         case REQ_STOP_LOADING:
2867                 for (i = 0; i < ARRAY_SIZE(views); i++) {
2868                         view = &views[i];
2869                         if (view->pipe)
2870                                 report("Stopped loading the %s view", view->name),
2871                         end_update(view, TRUE);
2872                 }
2873                 break;
2874
2875         case REQ_SHOW_VERSION:
2876                 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2877                 return TRUE;
2878
2879         case REQ_SCREEN_RESIZE:
2880                 resize_display();
2881                 /* Fall-through */
2882         case REQ_SCREEN_REDRAW:
2883                 redraw_display();
2884                 break;
2885
2886         case REQ_EDIT:
2887                 report("Nothing to edit");
2888                 break;
2889
2890         case REQ_ENTER:
2891                 report("Nothing to enter");
2892                 break;
2893
2894         case REQ_VIEW_CLOSE:
2895                 /* XXX: Mark closed views by letting view->parent point to the
2896                  * view itself. Parents to closed view should never be
2897                  * followed. */
2898                 if (view->parent &&
2899                     view->parent->parent != view->parent) {
2900                         memset(display, 0, sizeof(display));
2901                         current_view = 0;
2902                         display[current_view] = view->parent;
2903                         view->parent = view;
2904                         resize_display();
2905                         redraw_display();
2906                         report("");
2907                         break;
2908                 }
2909                 /* Fall-through */
2910         case REQ_QUIT:
2911                 return FALSE;
2912
2913         default:
2914                 report("Unknown key, press 'h' for help");
2915                 return TRUE;
2916         }
2917
2918         return TRUE;
2919 }
2920
2921
2922 /*
2923  * Pager backend
2924  */
2925
2926 static bool
2927 pager_draw(struct view *view, struct line *line, unsigned int lineno)
2928 {
2929         char *text = line->data;
2930
2931         if (opt_line_number && draw_lineno(view, lineno))
2932                 return TRUE;
2933
2934         draw_text(view, line->type, text, TRUE);
2935         return TRUE;
2936 }
2937
2938 static bool
2939 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
2940 {
2941         char refbuf[SIZEOF_STR];
2942         char *ref = NULL;
2943         FILE *pipe;
2944
2945         if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2946                 return TRUE;
2947
2948         pipe = popen(refbuf, "r");
2949         if (!pipe)
2950                 return TRUE;
2951
2952         if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2953                 ref = chomp_string(ref);
2954         pclose(pipe);
2955
2956         if (!ref || !*ref)
2957                 return TRUE;
2958
2959         /* This is the only fatal call, since it can "corrupt" the buffer. */
2960         if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2961                 return FALSE;
2962
2963         return TRUE;
2964 }
2965
2966 static void
2967 add_pager_refs(struct view *view, struct line *line)
2968 {
2969         char buf[SIZEOF_STR];
2970         char *commit_id = (char *)line->data + STRING_SIZE("commit ");
2971         struct ref **refs;
2972         size_t bufpos = 0, refpos = 0;
2973         const char *sep = "Refs: ";
2974         bool is_tag = FALSE;
2975
2976         assert(line->type == LINE_COMMIT);
2977
2978         refs = get_refs(commit_id);
2979         if (!refs) {
2980                 if (view == VIEW(REQ_VIEW_DIFF))
2981                         goto try_add_describe_ref;
2982                 return;
2983         }
2984
2985         do {
2986                 struct ref *ref = refs[refpos];
2987                 const char *fmt = ref->tag    ? "%s[%s]" :
2988                                   ref->remote ? "%s<%s>" : "%s%s";
2989
2990                 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2991                         return;
2992                 sep = ", ";
2993                 if (ref->tag)
2994                         is_tag = TRUE;
2995         } while (refs[refpos++]->next);
2996
2997         if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2998 try_add_describe_ref:
2999                 /* Add <tag>-g<commit_id> "fake" reference. */
3000                 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
3001                         return;
3002         }
3003
3004         if (bufpos == 0)
3005                 return;
3006
3007         if (!realloc_lines(view, view->line_size + 1))
3008                 return;
3009
3010         add_line_text(view, buf, LINE_PP_REFS);
3011 }
3012
3013 static bool
3014 pager_read(struct view *view, char *data)
3015 {
3016         struct line *line;
3017
3018         if (!data)
3019                 return TRUE;
3020
3021         line = add_line_text(view, data, get_line_type(data));
3022         if (!line)
3023                 return FALSE;
3024
3025         if (line->type == LINE_COMMIT &&
3026             (view == VIEW(REQ_VIEW_DIFF) ||
3027              view == VIEW(REQ_VIEW_LOG)))
3028                 add_pager_refs(view, line);
3029
3030         return TRUE;
3031 }
3032
3033 static enum request
3034 pager_request(struct view *view, enum request request, struct line *line)
3035 {
3036         int split = 0;
3037
3038         if (request != REQ_ENTER)
3039                 return request;
3040
3041         if (line->type == LINE_COMMIT &&
3042            (view == VIEW(REQ_VIEW_LOG) ||
3043             view == VIEW(REQ_VIEW_PAGER))) {
3044                 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
3045                 split = 1;
3046         }
3047
3048         /* Always scroll the view even if it was split. That way
3049          * you can use Enter to scroll through the log view and
3050          * split open each commit diff. */
3051         scroll_view(view, REQ_SCROLL_LINE_DOWN);
3052
3053         /* FIXME: A minor workaround. Scrolling the view will call report("")
3054          * but if we are scrolling a non-current view this won't properly
3055          * update the view title. */
3056         if (split)
3057                 update_view_title(view);
3058
3059         return REQ_NONE;
3060 }
3061
3062 static bool
3063 pager_grep(struct view *view, struct line *line)
3064 {
3065         regmatch_t pmatch;
3066         char *text = line->data;
3067
3068         if (!*text)
3069                 return FALSE;
3070
3071         if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
3072                 return FALSE;
3073
3074         return TRUE;
3075 }
3076
3077 static void
3078 pager_select(struct view *view, struct line *line)
3079 {
3080         if (line->type == LINE_COMMIT) {
3081                 char *text = (char *)line->data + STRING_SIZE("commit ");
3082
3083                 if (view != VIEW(REQ_VIEW_PAGER))
3084                         string_copy_rev(view->ref, text);
3085                 string_copy_rev(ref_commit, text);
3086         }
3087 }
3088
3089 static struct view_ops pager_ops = {
3090         "line",
3091         NULL,
3092         pager_read,
3093         pager_draw,
3094         pager_request,
3095         pager_grep,
3096         pager_select,
3097 };
3098
3099 static enum request
3100 log_request(struct view *view, enum request request, struct line *line)
3101 {
3102         switch (request) {
3103         case REQ_REFRESH:
3104                 load_refs();
3105                 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
3106                 return REQ_NONE;
3107         default:
3108                 return pager_request(view, request, line);
3109         }
3110 }
3111
3112 static struct view_ops log_ops = {
3113         "line",
3114         NULL,
3115         pager_read,
3116         pager_draw,
3117         log_request,
3118         pager_grep,
3119         pager_select,
3120 };
3121
3122
3123 /*
3124  * Help backend
3125  */
3126
3127 static bool
3128 help_open(struct view *view)
3129 {
3130         char buf[BUFSIZ];
3131         int lines = ARRAY_SIZE(req_info) + 2;
3132         int i;
3133
3134         if (view->lines > 0)
3135                 return TRUE;
3136
3137         for (i = 0; i < ARRAY_SIZE(req_info); i++)
3138                 if (!req_info[i].request)
3139                         lines++;
3140
3141         lines += run_requests + 1;
3142
3143         view->line = calloc(lines, sizeof(*view->line));
3144         if (!view->line)
3145                 return FALSE;
3146
3147         add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3148
3149         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3150                 const char *key;
3151
3152                 if (req_info[i].request == REQ_NONE)
3153                         continue;
3154
3155                 if (!req_info[i].request) {
3156                         add_line_text(view, "", LINE_DEFAULT);
3157                         add_line_text(view, req_info[i].help, LINE_DEFAULT);
3158                         continue;
3159                 }
3160
3161                 key = get_key(req_info[i].request);
3162                 if (!*key)
3163                         key = "(no key defined)";
3164
3165                 if (!string_format(buf, "    %-25s %s", key, req_info[i].help))
3166                         continue;
3167
3168                 add_line_text(view, buf, LINE_DEFAULT);
3169         }
3170
3171         if (run_requests) {
3172                 add_line_text(view, "", LINE_DEFAULT);
3173                 add_line_text(view, "External commands:", LINE_DEFAULT);
3174         }
3175
3176         for (i = 0; i < run_requests; i++) {
3177                 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3178                 const char *key;
3179
3180                 if (!req)
3181                         continue;
3182
3183                 key = get_key_name(req->key);
3184                 if (!*key)
3185                         key = "(no key defined)";
3186
3187                 if (!string_format(buf, "    %-10s %-14s `%s`",
3188                                    keymap_table[req->keymap].name,
3189                                    key, req->cmd))
3190                         continue;
3191
3192                 add_line_text(view, buf, LINE_DEFAULT);
3193         }
3194
3195         return TRUE;
3196 }
3197
3198 static struct view_ops help_ops = {
3199         "line",
3200         help_open,
3201         NULL,
3202         pager_draw,
3203         pager_request,
3204         pager_grep,
3205         pager_select,
3206 };
3207
3208
3209 /*
3210  * Tree backend
3211  */
3212
3213 struct tree_stack_entry {
3214         struct tree_stack_entry *prev;  /* Entry below this in the stack */
3215         unsigned long lineno;           /* Line number to restore */
3216         char *name;                     /* Position of name in opt_path */
3217 };
3218
3219 /* The top of the path stack. */
3220 static struct tree_stack_entry *tree_stack = NULL;
3221 unsigned long tree_lineno = 0;
3222
3223 static void
3224 pop_tree_stack_entry(void)
3225 {
3226         struct tree_stack_entry *entry = tree_stack;
3227
3228         tree_lineno = entry->lineno;
3229         entry->name[0] = 0;
3230         tree_stack = entry->prev;
3231         free(entry);
3232 }
3233
3234 static void
3235 push_tree_stack_entry(const char *name, unsigned long lineno)
3236 {
3237         struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3238         size_t pathlen = strlen(opt_path);
3239
3240         if (!entry)
3241                 return;
3242
3243         entry->prev = tree_stack;
3244         entry->name = opt_path + pathlen;
3245         tree_stack = entry;
3246
3247         if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3248                 pop_tree_stack_entry();
3249                 return;
3250         }
3251
3252         /* Move the current line to the first tree entry. */
3253         tree_lineno = 1;
3254         entry->lineno = lineno;
3255 }
3256
3257 /* Parse output from git-ls-tree(1):
3258  *
3259  * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
3260  * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
3261  * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3262  * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
3263  */
3264
3265 #define SIZEOF_TREE_ATTR \
3266         STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
3267
3268 #define TREE_UP_FORMAT "040000 tree %s\t.."
3269
3270 static int
3271 tree_compare_entry(enum line_type type1, const char *name1,
3272                    enum line_type type2, const char *name2)
3273 {
3274         if (type1 != type2) {
3275                 if (type1 == LINE_TREE_DIR)
3276                         return -1;
3277                 return 1;
3278         }
3279
3280         return strcmp(name1, name2);
3281 }
3282
3283 static const char *
3284 tree_path(struct line *line)
3285 {
3286         const char *path = line->data;
3287
3288         return path + SIZEOF_TREE_ATTR;
3289 }
3290
3291 static bool
3292 tree_read(struct view *view, char *text)
3293 {
3294         size_t textlen = text ? strlen(text) : 0;
3295         char buf[SIZEOF_STR];
3296         unsigned long pos;
3297         enum line_type type;
3298         bool first_read = view->lines == 0;
3299
3300         if (!text)
3301                 return TRUE;
3302         if (textlen <= SIZEOF_TREE_ATTR)
3303                 return FALSE;
3304
3305         type = text[STRING_SIZE("100644 ")] == 't'
3306              ? LINE_TREE_DIR : LINE_TREE_FILE;
3307
3308         if (first_read) {
3309                 /* Add path info line */
3310                 if (!string_format(buf, "Directory path /%s", opt_path) ||
3311                     !realloc_lines(view, view->line_size + 1) ||
3312                     !add_line_text(view, buf, LINE_DEFAULT))
3313                         return FALSE;
3314
3315                 /* Insert "link" to parent directory. */
3316                 if (*opt_path) {
3317                         if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
3318                             !realloc_lines(view, view->line_size + 1) ||
3319                             !add_line_text(view, buf, LINE_TREE_DIR))
3320                                 return FALSE;
3321                 }
3322         }
3323
3324         /* Strip the path part ... */
3325         if (*opt_path) {
3326                 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3327                 size_t striplen = strlen(opt_path);
3328                 char *path = text + SIZEOF_TREE_ATTR;
3329
3330                 if (pathlen > striplen)
3331                         memmove(path, path + striplen,
3332                                 pathlen - striplen + 1);
3333         }
3334
3335         /* Skip "Directory ..." and ".." line. */
3336         for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
3337                 struct line *line = &view->line[pos];
3338                 const char *path1 = tree_path(line);
3339                 char *path2 = text + SIZEOF_TREE_ATTR;
3340                 int cmp = tree_compare_entry(line->type, path1, type, path2);
3341
3342                 if (cmp <= 0)
3343                         continue;
3344
3345                 text = strdup(text);
3346                 if (!text)
3347                         return FALSE;
3348
3349                 if (view->lines > pos)
3350                         memmove(&view->line[pos + 1], &view->line[pos],
3351                                 (view->lines - pos) * sizeof(*line));
3352
3353                 line = &view->line[pos];
3354                 line->data = text;
3355                 line->type = type;
3356                 view->lines++;
3357                 return TRUE;
3358         }
3359
3360         if (!add_line_text(view, text, type))
3361                 return FALSE;
3362
3363         if (tree_lineno > view->lineno) {
3364                 view->lineno = tree_lineno;
3365                 tree_lineno = 0;
3366         }
3367
3368         return TRUE;
3369 }
3370
3371 static enum request
3372 tree_request(struct view *view, enum request request, struct line *line)
3373 {
3374         enum open_flags flags;
3375
3376         switch (request) {
3377         case REQ_VIEW_BLAME:
3378                 if (line->type != LINE_TREE_FILE) {
3379                         report("Blame only supported for files");
3380                         return REQ_NONE;
3381                 }
3382
3383                 string_copy(opt_ref, view->vid);
3384                 return request;
3385
3386         case REQ_EDIT:
3387                 if (line->type != LINE_TREE_FILE) {
3388                         report("Edit only supported for files");
3389                 } else if (!is_head_commit(view->vid)) {
3390                         report("Edit only supported for files in the current work tree");
3391                 } else {
3392                         open_editor(TRUE, opt_file);
3393                 }
3394                 return REQ_NONE;
3395
3396         case REQ_TREE_PARENT:
3397                 if (!*opt_path) {
3398                         /* quit view if at top of tree */
3399                         return REQ_VIEW_CLOSE;
3400                 }
3401                 /* fake 'cd  ..' */
3402                 line = &view->line[1];
3403                 break;
3404
3405         case REQ_ENTER:
3406                 break;
3407
3408         default:
3409                 return request;
3410         }
3411
3412         /* Cleanup the stack if the tree view is at a different tree. */
3413         while (!*opt_path && tree_stack)
3414                 pop_tree_stack_entry();
3415
3416         switch (line->type) {
3417         case LINE_TREE_DIR:
3418                 /* Depending on whether it is a subdir or parent (updir?) link
3419                  * mangle the path buffer. */
3420                 if (line == &view->line[1] && *opt_path) {
3421                         pop_tree_stack_entry();
3422
3423                 } else {
3424                         const char *basename = tree_path(line);
3425
3426                         push_tree_stack_entry(basename, view->lineno);
3427                 }
3428
3429                 /* Trees and subtrees share the same ID, so they are not not
3430                  * unique like blobs. */
3431                 flags = OPEN_RELOAD;
3432                 request = REQ_VIEW_TREE;
3433                 break;
3434
3435         case LINE_TREE_FILE:
3436                 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3437                 request = REQ_VIEW_BLOB;
3438                 break;
3439
3440         default:
3441                 return TRUE;
3442         }
3443
3444         open_view(view, request, flags);
3445         if (request == REQ_VIEW_TREE) {
3446                 view->lineno = tree_lineno;
3447         }
3448
3449         return REQ_NONE;
3450 }
3451
3452 static void
3453 tree_select(struct view *view, struct line *line)
3454 {
3455         char *text = (char *)line->data + STRING_SIZE("100644 blob ");
3456
3457         if (line->type == LINE_TREE_FILE) {
3458                 string_copy_rev(ref_blob, text);
3459                 string_format(opt_file, "%s%s", opt_path, tree_path(line));
3460
3461         } else if (line->type != LINE_TREE_DIR) {
3462                 return;
3463         }
3464
3465         string_copy_rev(view->ref, text);
3466 }
3467
3468 static struct view_ops tree_ops = {
3469         "file",
3470         NULL,
3471         tree_read,
3472         pager_draw,
3473         tree_request,
3474         pager_grep,
3475         tree_select,
3476 };
3477
3478 static bool
3479 blob_read(struct view *view, char *line)
3480 {
3481         if (!line)
3482                 return TRUE;
3483         return add_line_text(view, line, LINE_DEFAULT) != NULL;
3484 }
3485
3486 static struct view_ops blob_ops = {
3487         "line",
3488         NULL,
3489         blob_read,
3490         pager_draw,
3491         pager_request,
3492         pager_grep,
3493         pager_select,
3494 };
3495
3496 /*
3497  * Blame backend
3498  *
3499  * Loading the blame view is a two phase job:
3500  *
3501  *  1. File content is read either using opt_file from the
3502  *     filesystem or using git-cat-file.
3503  *  2. Then blame information is incrementally added by
3504  *     reading output from git-blame.
3505  */
3506
3507 struct blame_commit {
3508         char id[SIZEOF_REV];            /* SHA1 ID. */
3509         char title[128];                /* First line of the commit message. */
3510         char author[75];                /* Author of the commit. */
3511         struct tm time;                 /* Date from the author ident. */
3512         char filename[128];             /* Name of file. */
3513 };
3514
3515 struct blame {
3516         struct blame_commit *commit;
3517         char text[1];
3518 };
3519
3520 #define BLAME_CAT_FILE_CMD "git cat-file blob %s:%s"
3521 #define BLAME_INCREMENTAL_CMD "git blame --incremental %s -- %s"
3522
3523 static bool
3524 blame_open(struct view *view)
3525 {
3526         char path[SIZEOF_STR];
3527         char ref[SIZEOF_STR] = "";
3528
3529         if (sq_quote(path, 0, opt_file) >= sizeof(path))
3530                 return FALSE;
3531
3532         if (*opt_ref && sq_quote(ref, 0, opt_ref) >= sizeof(ref))
3533                 return FALSE;
3534
3535         if (*opt_ref || !(view->pipe = fopen(opt_file, "r"))) {
3536                 const char *id = *opt_ref ? ref : "HEAD";
3537
3538                 if (!string_format(view->cmd, BLAME_CAT_FILE_CMD, id, path) ||
3539                     !(view->pipe = popen(view->cmd, "r")))
3540                         return FALSE;
3541         }
3542
3543         setup_update(view, opt_file);
3544         string_format(view->ref, "%s ...", opt_file);
3545
3546         return TRUE;
3547 }
3548
3549 static struct blame_commit *
3550 get_blame_commit(struct view *view, const char *id)
3551 {
3552         size_t i;
3553
3554         for (i = 0; i < view->lines; i++) {
3555                 struct blame *blame = view->line[i].data;
3556
3557                 if (!blame->commit)
3558                         continue;
3559
3560                 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
3561                         return blame->commit;
3562         }
3563
3564         {
3565                 struct blame_commit *commit = calloc(1, sizeof(*commit));
3566
3567                 if (commit)
3568                         string_ncopy(commit->id, id, SIZEOF_REV);
3569                 return commit;
3570         }
3571 }
3572
3573 static bool
3574 parse_number(const char **posref, size_t *number, size_t min, size_t max)
3575 {
3576         const char *pos = *posref;
3577
3578         *posref = NULL;
3579         pos = strchr(pos + 1, ' ');
3580         if (!pos || !isdigit(pos[1]))
3581                 return FALSE;
3582         *number = atoi(pos + 1);
3583         if (*number < min || *number > max)
3584                 return FALSE;
3585
3586         *posref = pos;
3587         return TRUE;
3588 }
3589
3590 static struct blame_commit *
3591 parse_blame_commit(struct view *view, const char *text, int *blamed)
3592 {
3593         struct blame_commit *commit;
3594         struct blame *blame;
3595         const char *pos = text + SIZEOF_REV - 1;
3596         size_t lineno;
3597         size_t group;
3598
3599         if (strlen(text) <= SIZEOF_REV || *pos != ' ')
3600                 return NULL;
3601
3602         if (!parse_number(&pos, &lineno, 1, view->lines) ||
3603             !parse_number(&pos, &group, 1, view->lines - lineno + 1))
3604                 return NULL;
3605
3606         commit = get_blame_commit(view, text);
3607         if (!commit)
3608                 return NULL;
3609
3610         *blamed += group;
3611         while (group--) {
3612                 struct line *line = &view->line[lineno + group - 1];
3613
3614                 blame = line->data;
3615                 blame->commit = commit;
3616                 line->dirty = 1;
3617         }
3618
3619         return commit;
3620 }
3621
3622 static bool
3623 blame_read_file(struct view *view, const char *line, bool *read_file)
3624 {
3625         if (!line) {
3626                 char ref[SIZEOF_STR] = "";
3627                 char path[SIZEOF_STR];
3628                 FILE *pipe = NULL;
3629
3630                 if (view->lines == 0 && !view->parent)
3631                         die("No blame exist for %s", view->vid);
3632
3633                 if (view->lines == 0 ||
3634                     sq_quote(path, 0, opt_file) >= sizeof(path) ||
3635                     (*opt_ref && sq_quote(ref, 0, opt_ref) >= sizeof(ref)) ||
3636                     !string_format(view->cmd, BLAME_INCREMENTAL_CMD, ref, path) ||
3637                     !(pipe = popen(view->cmd, "r"))) {
3638                         report("Failed to load blame data");
3639                         return TRUE;
3640                 }
3641
3642                 fclose(view->pipe);
3643                 view->pipe = pipe;
3644                 *read_file = FALSE;
3645                 return FALSE;
3646
3647         } else {
3648                 size_t linelen = strlen(line);
3649                 struct blame *blame = malloc(sizeof(*blame) + linelen);
3650
3651                 blame->commit = NULL;
3652                 strncpy(blame->text, line, linelen);
3653                 blame->text[linelen] = 0;
3654                 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
3655         }
3656 }
3657
3658 static bool
3659 match_blame_header(const char *name, char **line)
3660 {
3661         size_t namelen = strlen(name);
3662         bool matched = !strncmp(name, *line, namelen);
3663
3664         if (matched)
3665                 *line += namelen;
3666
3667         return matched;
3668 }
3669
3670 static bool
3671 blame_read(struct view *view, char *line)
3672 {
3673         static struct blame_commit *commit = NULL;
3674         static int blamed = 0;
3675         static time_t author_time;
3676         static bool read_file = TRUE;
3677
3678         if (read_file)
3679                 return blame_read_file(view, line, &read_file);
3680
3681         if (!line) {
3682                 /* Reset all! */
3683                 commit = NULL;
3684                 blamed = 0;
3685                 read_file = TRUE;
3686                 string_format(view->ref, "%s", view->vid);
3687                 if (view_is_displayed(view)) {
3688                         update_view_title(view);
3689                         redraw_view_from(view, 0);
3690                 }
3691                 return TRUE;
3692         }
3693
3694         if (!commit) {
3695                 commit = parse_blame_commit(view, line, &blamed);
3696                 string_format(view->ref, "%s %2d%%", view->vid,
3697                               blamed * 100 / view->lines);
3698
3699         } else if (match_blame_header("author ", &line)) {
3700                 string_ncopy(commit->author, line, strlen(line));
3701
3702         } else if (match_blame_header("author-time ", &line)) {
3703                 author_time = (time_t) atol(line);
3704
3705         } else if (match_blame_header("author-tz ", &line)) {
3706                 long tz;
3707
3708                 tz  = ('0' - line[1]) * 60 * 60 * 10;
3709                 tz += ('0' - line[2]) * 60 * 60;
3710                 tz += ('0' - line[3]) * 60;
3711                 tz += ('0' - line[4]) * 60;
3712
3713                 if (line[0] == '-')
3714                         tz = -tz;
3715
3716                 author_time -= tz;
3717                 gmtime_r(&author_time, &commit->time);
3718
3719         } else if (match_blame_header("summary ", &line)) {
3720                 string_ncopy(commit->title, line, strlen(line));
3721
3722         } else if (match_blame_header("filename ", &line)) {
3723                 string_ncopy(commit->filename, line, strlen(line));
3724                 commit = NULL;
3725         }
3726
3727         return TRUE;
3728 }
3729
3730 static bool
3731 blame_draw(struct view *view, struct line *line, unsigned int lineno)
3732 {
3733         struct blame *blame = line->data;
3734         struct tm *time = NULL;
3735         const char *id = NULL, *author = NULL;
3736
3737         if (blame->commit && *blame->commit->filename) {
3738                 id = blame->commit->id;
3739                 author = blame->commit->author;
3740                 time = &blame->commit->time;
3741         }
3742
3743         if (opt_date && draw_date(view, time))
3744                 return TRUE;
3745
3746         if (opt_author &&
3747             draw_field(view, LINE_MAIN_AUTHOR, author, opt_author_cols, TRUE))
3748                 return TRUE;
3749
3750         if (draw_field(view, LINE_BLAME_ID, id, ID_COLS, FALSE))
3751                 return TRUE;
3752
3753         if (draw_lineno(view, lineno))
3754                 return TRUE;
3755
3756         draw_text(view, LINE_DEFAULT, blame->text, TRUE);
3757         return TRUE;
3758 }
3759
3760 static enum request
3761 blame_request(struct view *view, enum request request, struct line *line)
3762 {
3763         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3764         struct blame *blame = line->data;
3765
3766         switch (request) {
3767         case REQ_VIEW_BLAME:
3768                 if (!blame->commit || !strcmp(blame->commit->id, NULL_ID)) {
3769                         report("Commit ID unknown");
3770                         break;
3771                 }
3772                 string_copy(opt_ref, blame->commit->id);
3773                 open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
3774                 return request;
3775
3776         case REQ_ENTER:
3777                 if (!blame->commit) {
3778                         report("No commit loaded yet");
3779                         break;
3780                 }
3781
3782                 if (view_is_displayed(VIEW(REQ_VIEW_DIFF)) &&
3783                     !strcmp(blame->commit->id, VIEW(REQ_VIEW_DIFF)->ref))
3784                         break;
3785
3786                 if (!strcmp(blame->commit->id, NULL_ID)) {
3787                         char path[SIZEOF_STR];
3788
3789                         if (sq_quote(path, 0, view->vid) >= sizeof(path))
3790                                 break;
3791                         string_format(opt_cmd, "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s 2>/dev/null", path);
3792                 }
3793
3794                 open_view(view, REQ_VIEW_DIFF, flags);
3795                 break;
3796
3797         default:
3798                 return request;
3799         }
3800
3801         return REQ_NONE;
3802 }
3803
3804 static bool
3805 blame_grep(struct view *view, struct line *line)
3806 {
3807         struct blame *blame = line->data;
3808         struct blame_commit *commit = blame->commit;
3809         regmatch_t pmatch;
3810
3811 #define MATCH(text, on)                                                 \
3812         (on && *text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3813
3814         if (commit) {
3815                 char buf[DATE_COLS + 1];
3816
3817                 if (MATCH(commit->title, 1) ||
3818                     MATCH(commit->author, opt_author) ||
3819                     MATCH(commit->id, opt_date))
3820                         return TRUE;
3821
3822                 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
3823                     MATCH(buf, 1))
3824                         return TRUE;
3825         }
3826
3827         return MATCH(blame->text, 1);
3828
3829 #undef MATCH
3830 }
3831
3832 static void
3833 blame_select(struct view *view, struct line *line)
3834 {
3835         struct blame *blame = line->data;
3836         struct blame_commit *commit = blame->commit;
3837
3838         if (!commit)
3839                 return;
3840
3841         if (!strcmp(commit->id, NULL_ID))
3842                 string_ncopy(ref_commit, "HEAD", 4);
3843         else
3844                 string_copy_rev(ref_commit, commit->id);
3845 }
3846
3847 static struct view_ops blame_ops = {
3848         "line",
3849         blame_open,
3850         blame_read,
3851         blame_draw,
3852         blame_request,
3853         blame_grep,
3854         blame_select,
3855 };
3856
3857 /*
3858  * Status backend
3859  */
3860
3861 struct status {
3862         char status;
3863         struct {
3864                 mode_t mode;
3865                 char rev[SIZEOF_REV];
3866                 char name[SIZEOF_STR];
3867         } old;
3868         struct {
3869                 mode_t mode;
3870                 char rev[SIZEOF_REV];
3871                 char name[SIZEOF_STR];
3872         } new;
3873 };
3874
3875 static char status_onbranch[SIZEOF_STR];
3876 static struct status stage_status;
3877 static enum line_type stage_line_type;
3878 static size_t stage_chunks;
3879 static int *stage_chunk;
3880
3881 /* This should work even for the "On branch" line. */
3882 static inline bool
3883 status_has_none(struct view *view, struct line *line)
3884 {
3885         return line < view->line + view->lines && !line[1].data;
3886 }
3887
3888 /* Get fields from the diff line:
3889  * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
3890  */
3891 static inline bool
3892 status_get_diff(struct status *file, const char *buf, size_t bufsize)
3893 {
3894         const char *old_mode = buf +  1;
3895         const char *new_mode = buf +  8;
3896         const char *old_rev  = buf + 15;
3897         const char *new_rev  = buf + 56;
3898         const char *status   = buf + 97;
3899
3900         if (bufsize < 99 ||
3901             old_mode[-1] != ':' ||
3902             new_mode[-1] != ' ' ||
3903             old_rev[-1]  != ' ' ||
3904             new_rev[-1]  != ' ' ||
3905             status[-1]   != ' ')
3906                 return FALSE;
3907
3908         file->status = *status;
3909
3910         string_copy_rev(file->old.rev, old_rev);
3911         string_copy_rev(file->new.rev, new_rev);
3912
3913         file->old.mode = strtoul(old_mode, NULL, 8);
3914         file->new.mode = strtoul(new_mode, NULL, 8);
3915
3916         file->old.name[0] = file->new.name[0] = 0;
3917
3918         return TRUE;
3919 }
3920
3921 static bool
3922 status_run(struct view *view, const char cmd[], char status, enum line_type type)
3923 {
3924         struct status *file = NULL;
3925         struct status *unmerged = NULL;
3926         char buf[SIZEOF_STR * 4];
3927         size_t bufsize = 0;
3928         FILE *pipe;
3929
3930         pipe = popen(cmd, "r");
3931         if (!pipe)
3932                 return FALSE;
3933
3934         add_line_data(view, NULL, type);
3935
3936         while (!feof(pipe) && !ferror(pipe)) {
3937                 char *sep;
3938                 size_t readsize;
3939
3940                 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
3941                 if (!readsize)
3942                         break;
3943                 bufsize += readsize;
3944
3945                 /* Process while we have NUL chars. */
3946                 while ((sep = memchr(buf, 0, bufsize))) {
3947                         size_t sepsize = sep - buf + 1;
3948
3949                         if (!file) {
3950                                 if (!realloc_lines(view, view->line_size + 1))
3951                                         goto error_out;
3952
3953                                 file = calloc(1, sizeof(*file));
3954                                 if (!file)
3955                                         goto error_out;
3956
3957                                 add_line_data(view, file, type);
3958                         }
3959
3960                         /* Parse diff info part. */
3961                         if (status) {
3962                                 file->status = status;
3963                                 if (status == 'A')
3964                                         string_copy(file->old.rev, NULL_ID);
3965
3966                         } else if (!file->status) {
3967                                 if (!status_get_diff(file, buf, sepsize))
3968                                         goto error_out;
3969
3970                                 bufsize -= sepsize;
3971                                 memmove(buf, sep + 1, bufsize);
3972
3973                                 sep = memchr(buf, 0, bufsize);
3974                                 if (!sep)
3975                                         break;
3976                                 sepsize = sep - buf + 1;
3977
3978                                 /* Collapse all 'M'odified entries that
3979                                  * follow a associated 'U'nmerged entry.
3980                                  */
3981                                 if (file->status == 'U') {
3982                                         unmerged = file;
3983
3984                                 } else if (unmerged) {
3985                                         int collapse = !strcmp(buf, unmerged->new.name);
3986
3987                                         unmerged = NULL;
3988                                         if (collapse) {
3989                                                 free(file);
3990                                                 view->lines--;
3991                                                 continue;
3992                                         }
3993                                 }
3994                         }
3995
3996                         /* Grab the old name for rename/copy. */
3997                         if (!*file->old.name &&
3998                             (file->status == 'R' || file->status == 'C')) {
3999                                 sepsize = sep - buf + 1;
4000                                 string_ncopy(file->old.name, buf, sepsize);
4001                                 bufsize -= sepsize;
4002                                 memmove(buf, sep + 1, bufsize);
4003
4004                                 sep = memchr(buf, 0, bufsize);
4005                                 if (!sep)
4006                                         break;
4007                                 sepsize = sep - buf + 1;
4008                         }
4009
4010                         /* git-ls-files just delivers a NUL separated
4011                          * list of file names similar to the second half
4012                          * of the git-diff-* output. */
4013                         string_ncopy(file->new.name, buf, sepsize);
4014                         if (!*file->old.name)
4015                                 string_copy(file->old.name, file->new.name);
4016                         bufsize -= sepsize;
4017                         memmove(buf, sep + 1, bufsize);
4018                         file = NULL;
4019                 }
4020         }
4021
4022         if (ferror(pipe)) {
4023 error_out:
4024                 pclose(pipe);
4025                 return FALSE;
4026         }
4027
4028         if (!view->line[view->lines - 1].data)
4029                 add_line_data(view, NULL, LINE_STAT_NONE);
4030
4031         pclose(pipe);
4032         return TRUE;
4033 }
4034
4035 /* Don't show unmerged entries in the staged section. */
4036 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --diff-filter=ACDMRTXB --cached -M HEAD"
4037 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
4038 #define STATUS_LIST_OTHER_CMD \
4039         "git ls-files -z --others --exclude-standard"
4040 #define STATUS_LIST_NO_HEAD_CMD \
4041         "git ls-files -z --cached --exclude-standard"
4042
4043 #define STATUS_DIFF_INDEX_SHOW_CMD \
4044         "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s %s 2>/dev/null"
4045
4046 #define STATUS_DIFF_FILES_SHOW_CMD \
4047         "git diff-files --root --patch-with-stat -C -M -- %s %s 2>/dev/null"
4048
4049 #define STATUS_DIFF_NO_HEAD_SHOW_CMD \
4050         "git diff --no-color --patch-with-stat /dev/null %s 2>/dev/null"
4051
4052 /* First parse staged info using git-diff-index(1), then parse unstaged
4053  * info using git-diff-files(1), and finally untracked files using
4054  * git-ls-files(1). */
4055 static bool
4056 status_open(struct view *view)
4057 {
4058         unsigned long prev_lineno = view->lineno;
4059
4060         reset_view(view);
4061
4062         if (!realloc_lines(view, view->line_size + 7))
4063                 return FALSE;
4064
4065         add_line_data(view, NULL, LINE_STAT_HEAD);
4066         if (is_initial_commit())
4067                 string_copy(status_onbranch, "Initial commit");
4068         else if (!*opt_head)
4069                 string_copy(status_onbranch, "Not currently on any branch");
4070         else if (!string_format(status_onbranch, "On branch %s", opt_head))
4071                 return FALSE;
4072
4073         system("git update-index -q --refresh >/dev/null 2>/dev/null");
4074
4075         if (is_initial_commit()) {
4076                 if (!status_run(view, STATUS_LIST_NO_HEAD_CMD, 'A', LINE_STAT_STAGED))
4077                         return FALSE;
4078         } else if (!status_run(view, STATUS_DIFF_INDEX_CMD, 0, LINE_STAT_STAGED)) {
4079                 return FALSE;
4080         }
4081
4082         if (!status_run(view, STATUS_DIFF_FILES_CMD, 0, LINE_STAT_UNSTAGED) ||
4083             !status_run(view, STATUS_LIST_OTHER_CMD, '?', LINE_STAT_UNTRACKED))
4084                 return FALSE;
4085
4086         /* If all went well restore the previous line number to stay in
4087          * the context or select a line with something that can be
4088          * updated. */
4089         if (prev_lineno >= view->lines)
4090                 prev_lineno = view->lines - 1;
4091         while (prev_lineno < view->lines && !view->line[prev_lineno].data)
4092                 prev_lineno++;
4093         while (prev_lineno > 0 && !view->line[prev_lineno].data)
4094                 prev_lineno--;
4095
4096         /* If the above fails, always skip the "On branch" line. */
4097         if (prev_lineno < view->lines)
4098                 view->lineno = prev_lineno;
4099         else
4100                 view->lineno = 1;
4101
4102         if (view->lineno < view->offset)
4103                 view->offset = view->lineno;
4104         else if (view->offset + view->height <= view->lineno)
4105                 view->offset = view->lineno - view->height + 1;
4106
4107         return TRUE;
4108 }
4109
4110 static bool
4111 status_draw(struct view *view, struct line *line, unsigned int lineno)
4112 {
4113         struct status *status = line->data;
4114         enum line_type type;
4115         const char *text;
4116
4117         if (!status) {
4118                 switch (line->type) {
4119                 case LINE_STAT_STAGED:
4120                         type = LINE_STAT_SECTION;
4121                         text = "Changes to be committed:";
4122                         break;
4123
4124                 case LINE_STAT_UNSTAGED:
4125                         type = LINE_STAT_SECTION;
4126                         text = "Changed but not updated:";
4127                         break;
4128
4129                 case LINE_STAT_UNTRACKED:
4130                         type = LINE_STAT_SECTION;
4131                         text = "Untracked files:";
4132                         break;
4133
4134                 case LINE_STAT_NONE:
4135                         type = LINE_DEFAULT;
4136                         text = "    (no files)";
4137                         break;
4138
4139                 case LINE_STAT_HEAD:
4140                         type = LINE_STAT_HEAD;
4141                         text = status_onbranch;
4142                         break;
4143
4144                 default:
4145                         return FALSE;
4146                 }
4147         } else {
4148                 static char buf[] = { '?', ' ', ' ', ' ', 0 };
4149
4150                 buf[0] = status->status;
4151                 if (draw_text(view, line->type, buf, TRUE))
4152                         return TRUE;
4153                 type = LINE_DEFAULT;
4154                 text = status->new.name;
4155         }
4156
4157         draw_text(view, type, text, TRUE);
4158         return TRUE;
4159 }
4160
4161 static enum request
4162 status_enter(struct view *view, struct line *line)
4163 {
4164         struct status *status = line->data;
4165         char oldpath[SIZEOF_STR] = "";
4166         char newpath[SIZEOF_STR] = "";
4167         const char *info;
4168         size_t cmdsize = 0;
4169         enum open_flags split;
4170
4171         if (line->type == LINE_STAT_NONE ||
4172             (!status && line[1].type == LINE_STAT_NONE)) {
4173                 report("No file to diff");
4174                 return REQ_NONE;
4175         }
4176
4177         if (status) {
4178                 if (sq_quote(oldpath, 0, status->old.name) >= sizeof(oldpath))
4179                         return REQ_QUIT;
4180                 /* Diffs for unmerged entries are empty when pasing the
4181                  * new path, so leave it empty. */
4182                 if (status->status != 'U' &&
4183                     sq_quote(newpath, 0, status->new.name) >= sizeof(newpath))
4184                         return REQ_QUIT;
4185         }
4186
4187         if (opt_cdup[0] &&
4188             line->type != LINE_STAT_UNTRACKED &&
4189             !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
4190                 return REQ_QUIT;
4191
4192         switch (line->type) {
4193         case LINE_STAT_STAGED:
4194                 if (is_initial_commit()) {
4195                         if (!string_format_from(opt_cmd, &cmdsize,
4196                                                 STATUS_DIFF_NO_HEAD_SHOW_CMD,
4197                                                 newpath))
4198                                 return REQ_QUIT;
4199                 } else {
4200                         if (!string_format_from(opt_cmd, &cmdsize,
4201                                                 STATUS_DIFF_INDEX_SHOW_CMD,
4202                                                 oldpath, newpath))
4203                                 return REQ_QUIT;
4204                 }
4205
4206                 if (status)
4207                         info = "Staged changes to %s";
4208                 else
4209                         info = "Staged changes";
4210                 break;
4211
4212         case LINE_STAT_UNSTAGED:
4213                 if (!string_format_from(opt_cmd, &cmdsize,
4214                                         STATUS_DIFF_FILES_SHOW_CMD, oldpath, newpath))
4215                         return REQ_QUIT;
4216                 if (status)
4217                         info = "Unstaged changes to %s";
4218                 else
4219                         info = "Unstaged changes";
4220                 break;
4221
4222         case LINE_STAT_UNTRACKED:
4223                 if (opt_pipe)
4224                         return REQ_QUIT;
4225
4226                 if (!status) {
4227                         report("No file to show");
4228                         return REQ_NONE;
4229                 }
4230
4231                 if (!suffixcmp(status->new.name, -1, "/")) {
4232                         report("Cannot display a directory");
4233                         return REQ_NONE;
4234                 }
4235
4236                 opt_pipe = fopen(status->new.name, "r");
4237                 info = "Untracked file %s";
4238                 break;
4239
4240         case LINE_STAT_HEAD:
4241                 return REQ_NONE;
4242
4243         default:
4244                 die("line type %d not handled in switch", line->type);
4245         }
4246
4247         split = view_is_displayed(view) ? OPEN_SPLIT : 0;
4248         open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | split);
4249         if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
4250                 if (status) {
4251                         stage_status = *status;
4252                 } else {
4253                         memset(&stage_status, 0, sizeof(stage_status));
4254                 }
4255
4256                 stage_line_type = line->type;
4257                 stage_chunks = 0;
4258                 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
4259         }
4260
4261         return REQ_NONE;
4262 }
4263
4264 static bool
4265 status_exists(struct status *status, enum line_type type)
4266 {
4267         struct view *view = VIEW(REQ_VIEW_STATUS);
4268         struct line *line;
4269
4270         for (line = view->line; line < view->line + view->lines; line++) {
4271                 struct status *pos = line->data;
4272
4273                 if (line->type == type && pos &&
4274                     !strcmp(status->new.name, pos->new.name))
4275                         return TRUE;
4276         }
4277
4278         return FALSE;
4279 }
4280
4281
4282 static FILE *
4283 status_update_prepare(enum line_type type)
4284 {
4285         char cmd[SIZEOF_STR];
4286         size_t cmdsize = 0;
4287
4288         if (opt_cdup[0] &&
4289             type != LINE_STAT_UNTRACKED &&
4290             !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4291                 return NULL;
4292
4293         switch (type) {
4294         case LINE_STAT_STAGED:
4295                 string_add(cmd, cmdsize, "git update-index -z --index-info");
4296                 break;
4297
4298         case LINE_STAT_UNSTAGED:
4299         case LINE_STAT_UNTRACKED:
4300                 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
4301                 break;
4302
4303         default:
4304                 die("line type %d not handled in switch", type);
4305         }
4306
4307         return popen(cmd, "w");
4308 }
4309
4310 static bool
4311 status_update_write(FILE *pipe, struct status *status, enum line_type type)
4312 {
4313         char buf[SIZEOF_STR];
4314         size_t bufsize = 0;
4315         size_t written = 0;
4316
4317         switch (type) {
4318         case LINE_STAT_STAGED:
4319                 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
4320                                         status->old.mode,
4321                                         status->old.rev,
4322                                         status->old.name, 0))
4323                         return FALSE;
4324                 break;
4325
4326         case LINE_STAT_UNSTAGED:
4327         case LINE_STAT_UNTRACKED:
4328                 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
4329                         return FALSE;
4330                 break;
4331
4332         default:
4333                 die("line type %d not handled in switch", type);
4334         }
4335
4336         while (!ferror(pipe) && written < bufsize) {
4337                 written += fwrite(buf + written, 1, bufsize - written, pipe);
4338         }
4339
4340         return written == bufsize;
4341 }
4342
4343 static bool
4344 status_update_file(struct status *status, enum line_type type)
4345 {
4346         FILE *pipe = status_update_prepare(type);
4347         bool result;
4348
4349         if (!pipe)
4350                 return FALSE;
4351
4352         result = status_update_write(pipe, status, type);
4353         pclose(pipe);
4354         return result;
4355 }
4356
4357 static bool
4358 status_update_files(struct view *view, struct line *line)
4359 {
4360         FILE *pipe = status_update_prepare(line->type);
4361         bool result = TRUE;
4362         struct line *pos = view->line + view->lines;
4363         int files = 0;
4364         int file, done;
4365
4366         if (!pipe)
4367                 return FALSE;
4368
4369         for (pos = line; pos < view->line + view->lines && pos->data; pos++)
4370                 files++;
4371
4372         for (file = 0, done = 0; result && file < files; line++, file++) {
4373                 int almost_done = file * 100 / files;
4374
4375                 if (almost_done > done) {
4376                         done = almost_done;
4377                         string_format(view->ref, "updating file %u of %u (%d%% done)",
4378                                       file, files, done);
4379                         update_view_title(view);
4380                 }
4381                 result = status_update_write(pipe, line->data, line->type);
4382         }
4383
4384         pclose(pipe);
4385         return result;
4386 }
4387
4388 static bool
4389 status_update(struct view *view)
4390 {
4391         struct line *line = &view->line[view->lineno];
4392
4393         assert(view->lines);
4394
4395         if (!line->data) {
4396                 /* This should work even for the "On branch" line. */
4397                 if (line < view->line + view->lines && !line[1].data) {
4398                         report("Nothing to update");
4399                         return FALSE;
4400                 }
4401
4402                 if (!status_update_files(view, line + 1)) {
4403                         report("Failed to update file status");
4404                         return FALSE;
4405                 }
4406
4407         } else if (!status_update_file(line->data, line->type)) {
4408                 report("Failed to update file status");
4409                 return FALSE;
4410         }
4411
4412         return TRUE;
4413 }
4414
4415 static bool
4416 status_revert(struct status *status, enum line_type type, bool has_none)
4417 {
4418         if (!status || type != LINE_STAT_UNSTAGED) {
4419                 if (type == LINE_STAT_STAGED) {
4420                         report("Cannot revert changes to staged files");
4421                 } else if (type == LINE_STAT_UNTRACKED) {
4422                         report("Cannot revert changes to untracked files");
4423                 } else if (has_none) {
4424                         report("Nothing to revert");
4425                 } else {
4426                         report("Cannot revert changes to multiple files");
4427                 }
4428                 return FALSE;
4429
4430         } else {
4431                 char cmd[SIZEOF_STR];
4432                 char file_sq[SIZEOF_STR];
4433
4434                 if (sq_quote(file_sq, 0, status->old.name) >= sizeof(file_sq) ||
4435                     !string_format(cmd, "git checkout -- %s%s", opt_cdup, file_sq))
4436                         return FALSE;
4437
4438                 return run_confirm(cmd, "Are you sure you want to overwrite any changes?");
4439         }
4440 }
4441
4442 static enum request
4443 status_request(struct view *view, enum request request, struct line *line)
4444 {
4445         struct status *status = line->data;
4446
4447         switch (request) {
4448         case REQ_STATUS_UPDATE:
4449                 if (!status_update(view))
4450                         return REQ_NONE;
4451                 break;
4452
4453         case REQ_STATUS_REVERT:
4454                 if (!status_revert(status, line->type, status_has_none(view, line)))
4455                         return REQ_NONE;
4456                 break;
4457
4458         case REQ_STATUS_MERGE:
4459                 if (!status || status->status != 'U') {
4460                         report("Merging only possible for files with unmerged status ('U').");
4461                         return REQ_NONE;
4462                 }
4463                 open_mergetool(status->new.name);
4464                 break;
4465
4466         case REQ_EDIT:
4467                 if (!status)
4468                         return request;
4469                 if (status->status == 'D') {
4470                         report("File has been deleted.");
4471                         return REQ_NONE;
4472                 }
4473
4474                 open_editor(status->status != '?', status->new.name);
4475                 break;
4476
4477         case REQ_VIEW_BLAME:
4478                 if (status) {
4479                         string_copy(opt_file, status->new.name);
4480                         opt_ref[0] = 0;
4481                 }
4482                 return request;
4483
4484         case REQ_ENTER:
4485                 /* After returning the status view has been split to
4486                  * show the stage view. No further reloading is
4487                  * necessary. */
4488                 status_enter(view, line);
4489                 return REQ_NONE;
4490
4491         case REQ_REFRESH:
4492                 /* Simply reload the view. */
4493                 break;
4494
4495         default:
4496                 return request;
4497         }
4498
4499         open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
4500
4501         return REQ_NONE;
4502 }
4503
4504 static void
4505 status_select(struct view *view, struct line *line)
4506 {
4507         struct status *status = line->data;
4508         char file[SIZEOF_STR] = "all files";
4509         const char *text;
4510         const char *key;
4511
4512         if (status && !string_format(file, "'%s'", status->new.name))
4513                 return;
4514
4515         if (!status && line[1].type == LINE_STAT_NONE)
4516                 line++;
4517
4518         switch (line->type) {
4519         case LINE_STAT_STAGED:
4520                 text = "Press %s to unstage %s for commit";
4521                 break;
4522
4523         case LINE_STAT_UNSTAGED:
4524                 text = "Press %s to stage %s for commit";
4525                 break;
4526
4527         case LINE_STAT_UNTRACKED:
4528                 text = "Press %s to stage %s for addition";
4529                 break;
4530
4531         case LINE_STAT_HEAD:
4532         case LINE_STAT_NONE:
4533                 text = "Nothing to update";
4534                 break;
4535
4536         default:
4537                 die("line type %d not handled in switch", line->type);
4538         }
4539
4540         if (status && status->status == 'U') {
4541                 text = "Press %s to resolve conflict in %s";
4542                 key = get_key(REQ_STATUS_MERGE);
4543
4544         } else {
4545                 key = get_key(REQ_STATUS_UPDATE);
4546         }
4547
4548         string_format(view->ref, text, key, file);
4549 }
4550
4551 static bool
4552 status_grep(struct view *view, struct line *line)
4553 {
4554         struct status *status = line->data;
4555         enum { S_STATUS, S_NAME, S_END } state;
4556         char buf[2] = "?";
4557         regmatch_t pmatch;
4558
4559         if (!status)
4560                 return FALSE;
4561
4562         for (state = S_STATUS; state < S_END; state++) {
4563                 const char *text;
4564
4565                 switch (state) {
4566                 case S_NAME:    text = status->new.name;        break;
4567                 case S_STATUS:
4568                         buf[0] = status->status;
4569                         text = buf;
4570                         break;
4571
4572                 default:
4573                         return FALSE;
4574                 }
4575
4576                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4577                         return TRUE;
4578         }
4579
4580         return FALSE;
4581 }
4582
4583 static struct view_ops status_ops = {
4584         "file",
4585         status_open,
4586         NULL,
4587         status_draw,
4588         status_request,
4589         status_grep,
4590         status_select,
4591 };
4592
4593
4594 static bool
4595 stage_diff_line(FILE *pipe, struct line *line)
4596 {
4597         const char *buf = line->data;
4598         size_t bufsize = strlen(buf);
4599         size_t written = 0;
4600
4601         while (!ferror(pipe) && written < bufsize) {
4602                 written += fwrite(buf + written, 1, bufsize - written, pipe);
4603         }
4604
4605         fputc('\n', pipe);
4606
4607         return written == bufsize;
4608 }
4609
4610 static bool
4611 stage_diff_write(FILE *pipe, struct line *line, struct line *end)
4612 {
4613         while (line < end) {
4614                 if (!stage_diff_line(pipe, line++))
4615                         return FALSE;
4616                 if (line->type == LINE_DIFF_CHUNK ||
4617                     line->type == LINE_DIFF_HEADER)
4618                         break;
4619         }
4620
4621         return TRUE;
4622 }
4623
4624 static struct line *
4625 stage_diff_find(struct view *view, struct line *line, enum line_type type)
4626 {
4627         for (; view->line < line; line--)
4628                 if (line->type == type)
4629                         return line;
4630
4631         return NULL;
4632 }
4633
4634 static bool
4635 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
4636 {
4637         char cmd[SIZEOF_STR];
4638         size_t cmdsize = 0;
4639         struct line *diff_hdr;
4640         FILE *pipe;
4641
4642         diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
4643         if (!diff_hdr)
4644                 return FALSE;
4645
4646         if (opt_cdup[0] &&
4647             !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4648                 return FALSE;
4649
4650         if (!string_format_from(cmd, &cmdsize,
4651                                 "git apply --whitespace=nowarn %s %s - && "
4652                                 "git update-index -q --unmerged --refresh 2>/dev/null",
4653                                 revert ? "" : "--cached",
4654                                 revert || stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
4655                 return FALSE;
4656
4657         pipe = popen(cmd, "w");
4658         if (!pipe)
4659                 return FALSE;
4660
4661         if (!stage_diff_write(pipe, diff_hdr, chunk) ||
4662             !stage_diff_write(pipe, chunk, view->line + view->lines))
4663                 chunk = NULL;
4664
4665         pclose(pipe);
4666
4667         return chunk ? TRUE : FALSE;
4668 }
4669
4670 static bool
4671 stage_update(struct view *view, struct line *line)
4672 {
4673         struct line *chunk = NULL;
4674
4675         if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
4676                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4677
4678         if (chunk) {
4679                 if (!stage_apply_chunk(view, chunk, FALSE)) {
4680                         report("Failed to apply chunk");
4681                         return FALSE;
4682                 }
4683
4684         } else if (!stage_status.status) {
4685                 view = VIEW(REQ_VIEW_STATUS);
4686
4687                 for (line = view->line; line < view->line + view->lines; line++)
4688                         if (line->type == stage_line_type)
4689                                 break;
4690
4691                 if (!status_update_files(view, line + 1)) {
4692                         report("Failed to update files");
4693                         return FALSE;
4694                 }
4695
4696         } else if (!status_update_file(&stage_status, stage_line_type)) {
4697                 report("Failed to update file");
4698                 return FALSE;
4699         }
4700
4701         return TRUE;
4702 }
4703
4704 static bool
4705 stage_revert(struct view *view, struct line *line)
4706 {
4707         struct line *chunk = NULL;
4708
4709         if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
4710                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4711
4712         if (chunk) {
4713                 if (!prompt_yesno("Are you sure you want to revert changes?"))
4714                         return FALSE;
4715
4716                 if (!stage_apply_chunk(view, chunk, TRUE)) {
4717                         report("Failed to revert chunk");
4718                         return FALSE;
4719                 }
4720                 return TRUE;
4721
4722         } else {
4723                 return status_revert(stage_status.status ? &stage_status : NULL,
4724                                      stage_line_type, FALSE);
4725         }
4726 }
4727
4728
4729 static void
4730 stage_next(struct view *view, struct line *line)
4731 {
4732         int i;
4733
4734         if (!stage_chunks) {
4735                 static size_t alloc = 0;
4736                 int *tmp;
4737
4738                 for (line = view->line; line < view->line + view->lines; line++) {
4739                         if (line->type != LINE_DIFF_CHUNK)
4740                                 continue;
4741
4742                         tmp = realloc_items(stage_chunk, &alloc,
4743                                             stage_chunks, sizeof(*tmp));
4744                         if (!tmp) {
4745                                 report("Allocation failure");
4746                                 return;
4747                         }
4748
4749                         stage_chunk = tmp;
4750                         stage_chunk[stage_chunks++] = line - view->line;
4751                 }
4752         }
4753
4754         for (i = 0; i < stage_chunks; i++) {
4755                 if (stage_chunk[i] > view->lineno) {
4756                         do_scroll_view(view, stage_chunk[i] - view->lineno);
4757                         report("Chunk %d of %d", i + 1, stage_chunks);
4758                         return;
4759                 }
4760         }
4761
4762         report("No next chunk found");
4763 }
4764
4765 static enum request
4766 stage_request(struct view *view, enum request request, struct line *line)
4767 {
4768         switch (request) {
4769         case REQ_STATUS_UPDATE:
4770                 if (!stage_update(view, line))
4771                         return REQ_NONE;
4772                 break;
4773
4774         case REQ_STATUS_REVERT:
4775                 if (!stage_revert(view, line))
4776                         return REQ_NONE;
4777                 break;
4778
4779         case REQ_STAGE_NEXT:
4780                 if (stage_line_type == LINE_STAT_UNTRACKED) {
4781                         report("File is untracked; press %s to add",
4782                                get_key(REQ_STATUS_UPDATE));
4783                         return REQ_NONE;
4784                 }
4785                 stage_next(view, line);
4786                 return REQ_NONE;
4787
4788         case REQ_EDIT:
4789                 if (!stage_status.new.name[0])
4790                         return request;
4791                 if (stage_status.status == 'D') {
4792                         report("File has been deleted.");
4793                         return REQ_NONE;
4794                 }
4795
4796                 open_editor(stage_status.status != '?', stage_status.new.name);
4797                 break;
4798
4799         case REQ_REFRESH:
4800                 /* Reload everything ... */
4801                 break;
4802
4803         case REQ_VIEW_BLAME:
4804                 if (stage_status.new.name[0]) {
4805                         string_copy(opt_file, stage_status.new.name);
4806                         opt_ref[0] = 0;
4807                 }
4808                 return request;
4809
4810         case REQ_ENTER:
4811                 return pager_request(view, request, line);
4812
4813         default:
4814                 return request;
4815         }
4816
4817         open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD | OPEN_NOMAXIMIZE);
4818
4819         /* Check whether the staged entry still exists, and close the
4820          * stage view if it doesn't. */
4821         if (!status_exists(&stage_status, stage_line_type))
4822                 return REQ_VIEW_CLOSE;
4823
4824         if (stage_line_type == LINE_STAT_UNTRACKED) {
4825                 if (!suffixcmp(stage_status.new.name, -1, "/")) {
4826                         report("Cannot display a directory");
4827                         return REQ_NONE;
4828                 }
4829
4830                 opt_pipe = fopen(stage_status.new.name, "r");
4831         }
4832         open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
4833
4834         return REQ_NONE;
4835 }
4836
4837 static struct view_ops stage_ops = {
4838         "line",
4839         NULL,
4840         pager_read,
4841         pager_draw,
4842         stage_request,
4843         pager_grep,
4844         pager_select,
4845 };
4846
4847
4848 /*
4849  * Revision graph
4850  */
4851
4852 struct commit {
4853         char id[SIZEOF_REV];            /* SHA1 ID. */
4854         char title[128];                /* First line of the commit message. */
4855         char author[75];                /* Author of the commit. */
4856         struct tm time;                 /* Date from the author ident. */
4857         struct ref **refs;              /* Repository references. */
4858         chtype graph[SIZEOF_REVGRAPH];  /* Ancestry chain graphics. */
4859         size_t graph_size;              /* The width of the graph array. */
4860         bool has_parents;               /* Rewritten --parents seen. */
4861 };
4862
4863 /* Size of rev graph with no  "padding" columns */
4864 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
4865
4866 struct rev_graph {
4867         struct rev_graph *prev, *next, *parents;
4868         char rev[SIZEOF_REVITEMS][SIZEOF_REV];
4869         size_t size;
4870         struct commit *commit;
4871         size_t pos;
4872         unsigned int boundary:1;
4873 };
4874
4875 /* Parents of the commit being visualized. */
4876 static struct rev_graph graph_parents[4];
4877
4878 /* The current stack of revisions on the graph. */
4879 static struct rev_graph graph_stacks[4] = {
4880         { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
4881         { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
4882         { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
4883         { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
4884 };
4885
4886 static inline bool
4887 graph_parent_is_merge(struct rev_graph *graph)
4888 {
4889         return graph->parents->size > 1;
4890 }
4891
4892 static inline void
4893 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
4894 {
4895         struct commit *commit = graph->commit;
4896
4897         if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
4898                 commit->graph[commit->graph_size++] = symbol;
4899 }
4900
4901 static void
4902 clear_rev_graph(struct rev_graph *graph)
4903 {
4904         graph->boundary = 0;
4905         graph->size = graph->pos = 0;
4906         graph->commit = NULL;
4907         memset(graph->parents, 0, sizeof(*graph->parents));
4908 }
4909
4910 static void
4911 done_rev_graph(struct rev_graph *graph)
4912 {
4913         if (graph_parent_is_merge(graph) &&
4914             graph->pos < graph->size - 1 &&
4915             graph->next->size == graph->size + graph->parents->size - 1) {
4916                 size_t i = graph->pos + graph->parents->size - 1;
4917
4918                 graph->commit->graph_size = i * 2;
4919                 while (i < graph->next->size - 1) {
4920                         append_to_rev_graph(graph, ' ');
4921                         append_to_rev_graph(graph, '\\');
4922                         i++;
4923                 }
4924         }
4925
4926         clear_rev_graph(graph);
4927 }
4928
4929 static void
4930 push_rev_graph(struct rev_graph *graph, const char *parent)
4931 {
4932         int i;
4933
4934         /* "Collapse" duplicate parents lines.
4935          *
4936          * FIXME: This needs to also update update the drawn graph but
4937          * for now it just serves as a method for pruning graph lines. */
4938         for (i = 0; i < graph->size; i++)
4939                 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
4940                         return;
4941
4942         if (graph->size < SIZEOF_REVITEMS) {
4943                 string_copy_rev(graph->rev[graph->size++], parent);
4944         }
4945 }
4946
4947 static chtype
4948 get_rev_graph_symbol(struct rev_graph *graph)
4949 {
4950         chtype symbol;
4951
4952         if (graph->boundary)
4953                 symbol = REVGRAPH_BOUND;
4954         else if (graph->parents->size == 0)
4955                 symbol = REVGRAPH_INIT;
4956         else if (graph_parent_is_merge(graph))
4957                 symbol = REVGRAPH_MERGE;
4958         else if (graph->pos >= graph->size)
4959                 symbol = REVGRAPH_BRANCH;
4960         else
4961                 symbol = REVGRAPH_COMMIT;
4962
4963         return symbol;
4964 }
4965
4966 static void
4967 draw_rev_graph(struct rev_graph *graph)
4968 {
4969         struct rev_filler {
4970                 chtype separator, line;
4971         };
4972         enum { DEFAULT, RSHARP, RDIAG, LDIAG };
4973         static struct rev_filler fillers[] = {
4974                 { ' ',  '|' },
4975                 { '`',  '.' },
4976                 { '\'', ' ' },
4977                 { '/',  ' ' },
4978         };
4979         chtype symbol = get_rev_graph_symbol(graph);
4980         struct rev_filler *filler;
4981         size_t i;
4982
4983         if (opt_line_graphics)
4984                 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
4985
4986         filler = &fillers[DEFAULT];
4987
4988         for (i = 0; i < graph->pos; i++) {
4989                 append_to_rev_graph(graph, filler->line);
4990                 if (graph_parent_is_merge(graph->prev) &&
4991                     graph->prev->pos == i)
4992                         filler = &fillers[RSHARP];
4993
4994                 append_to_rev_graph(graph, filler->separator);
4995         }
4996
4997         /* Place the symbol for this revision. */
4998         append_to_rev_graph(graph, symbol);
4999
5000         if (graph->prev->size > graph->size)
5001                 filler = &fillers[RDIAG];
5002         else
5003                 filler = &fillers[DEFAULT];
5004
5005         i++;
5006
5007         for (; i < graph->size; i++) {
5008                 append_to_rev_graph(graph, filler->separator);
5009                 append_to_rev_graph(graph, filler->line);
5010                 if (graph_parent_is_merge(graph->prev) &&
5011                     i < graph->prev->pos + graph->parents->size)
5012                         filler = &fillers[RSHARP];
5013                 if (graph->prev->size > graph->size)
5014                         filler = &fillers[LDIAG];
5015         }
5016
5017         if (graph->prev->size > graph->size) {
5018                 append_to_rev_graph(graph, filler->separator);
5019                 if (filler->line != ' ')
5020                         append_to_rev_graph(graph, filler->line);
5021         }
5022 }
5023
5024 /* Prepare the next rev graph */
5025 static void
5026 prepare_rev_graph(struct rev_graph *graph)
5027 {
5028         size_t i;
5029
5030         /* First, traverse all lines of revisions up to the active one. */
5031         for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5032                 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5033                         break;
5034
5035                 push_rev_graph(graph->next, graph->rev[graph->pos]);
5036         }
5037
5038         /* Interleave the new revision parent(s). */
5039         for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5040                 push_rev_graph(graph->next, graph->parents->rev[i]);
5041
5042         /* Lastly, put any remaining revisions. */
5043         for (i = graph->pos + 1; i < graph->size; i++)
5044                 push_rev_graph(graph->next, graph->rev[i]);
5045 }
5046
5047 static void
5048 update_rev_graph(struct rev_graph *graph)
5049 {
5050         /* If this is the finalizing update ... */
5051         if (graph->commit)
5052                 prepare_rev_graph(graph);
5053
5054         /* Graph visualization needs a one rev look-ahead,
5055          * so the first update doesn't visualize anything. */
5056         if (!graph->prev->commit)
5057                 return;
5058
5059         draw_rev_graph(graph->prev);
5060         done_rev_graph(graph->prev->prev);
5061 }
5062
5063
5064 /*
5065  * Main view backend
5066  */
5067
5068 static bool
5069 main_draw(struct view *view, struct line *line, unsigned int lineno)
5070 {
5071         struct commit *commit = line->data;
5072
5073         if (!*commit->author)
5074                 return FALSE;
5075
5076         if (opt_date && draw_date(view, &commit->time))
5077                 return TRUE;
5078
5079         if (opt_author &&
5080             draw_field(view, LINE_MAIN_AUTHOR, commit->author, opt_author_cols, TRUE))
5081                 return TRUE;
5082
5083         if (opt_rev_graph && commit->graph_size &&
5084             draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5085                 return TRUE;
5086
5087         if (opt_show_refs && commit->refs) {
5088                 size_t i = 0;
5089
5090                 do {
5091                         enum line_type type;
5092
5093                         if (commit->refs[i]->head)
5094                                 type = LINE_MAIN_HEAD;
5095                         else if (commit->refs[i]->ltag)
5096                                 type = LINE_MAIN_LOCAL_TAG;
5097                         else if (commit->refs[i]->tag)
5098                                 type = LINE_MAIN_TAG;
5099                         else if (commit->refs[i]->tracked)
5100                                 type = LINE_MAIN_TRACKED;
5101                         else if (commit->refs[i]->remote)
5102                                 type = LINE_MAIN_REMOTE;
5103                         else
5104                                 type = LINE_MAIN_REF;
5105
5106                         if (draw_text(view, type, "[", TRUE) ||
5107                             draw_text(view, type, commit->refs[i]->name, TRUE) ||
5108                             draw_text(view, type, "]", TRUE))
5109                                 return TRUE;
5110
5111                         if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5112                                 return TRUE;
5113                 } while (commit->refs[i++]->next);
5114         }
5115
5116         draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5117         return TRUE;
5118 }
5119
5120 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5121 static bool
5122 main_read(struct view *view, char *line)
5123 {
5124         static struct rev_graph *graph = graph_stacks;
5125         enum line_type type;
5126         struct commit *commit;
5127
5128         if (!line) {
5129                 int i;
5130
5131                 if (!view->lines && !view->parent)
5132                         die("No revisions match the given arguments.");
5133                 if (view->lines > 0) {
5134                         commit = view->line[view->lines - 1].data;
5135                         if (!*commit->author) {
5136                                 view->lines--;
5137                                 free(commit);
5138                                 graph->commit = NULL;
5139                         }
5140                 }
5141                 update_rev_graph(graph);
5142
5143                 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5144                         clear_rev_graph(&graph_stacks[i]);
5145                 return TRUE;
5146         }
5147
5148         type = get_line_type(line);
5149         if (type == LINE_COMMIT) {
5150                 commit = calloc(1, sizeof(struct commit));
5151                 if (!commit)
5152                         return FALSE;
5153
5154                 line += STRING_SIZE("commit ");
5155                 if (*line == '-') {
5156                         graph->boundary = 1;
5157                         line++;
5158                 }
5159
5160                 string_copy_rev(commit->id, line);
5161                 commit->refs = get_refs(commit->id);
5162                 graph->commit = commit;
5163                 add_line_data(view, commit, LINE_MAIN_COMMIT);
5164
5165                 while ((line = strchr(line, ' '))) {
5166                         line++;
5167                         push_rev_graph(graph->parents, line);
5168                         commit->has_parents = TRUE;
5169                 }
5170                 return TRUE;
5171         }
5172
5173         if (!view->lines)
5174                 return TRUE;
5175         commit = view->line[view->lines - 1].data;
5176
5177         switch (type) {
5178         case LINE_PARENT:
5179                 if (commit->has_parents)
5180                         break;
5181                 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5182                 break;
5183
5184         case LINE_AUTHOR:
5185         {
5186                 /* Parse author lines where the name may be empty:
5187                  *      author  <email@address.tld> 1138474660 +0100
5188                  */
5189                 char *ident = line + STRING_SIZE("author ");
5190                 char *nameend = strchr(ident, '<');
5191                 char *emailend = strchr(ident, '>');
5192
5193                 if (!nameend || !emailend)
5194                         break;
5195
5196                 update_rev_graph(graph);
5197                 graph = graph->next;
5198
5199                 *nameend = *emailend = 0;
5200                 ident = chomp_string(ident);
5201                 if (!*ident) {
5202                         ident = chomp_string(nameend + 1);
5203                         if (!*ident)
5204                                 ident = "Unknown";
5205                 }
5206
5207                 string_ncopy(commit->author, ident, strlen(ident));
5208
5209                 /* Parse epoch and timezone */
5210                 if (emailend[1] == ' ') {
5211                         char *secs = emailend + 2;
5212                         char *zone = strchr(secs, ' ');
5213                         time_t time = (time_t) atol(secs);
5214
5215                         if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
5216                                 long tz;
5217
5218                                 zone++;
5219                                 tz  = ('0' - zone[1]) * 60 * 60 * 10;
5220                                 tz += ('0' - zone[2]) * 60 * 60;
5221                                 tz += ('0' - zone[3]) * 60;
5222                                 tz += ('0' - zone[4]) * 60;
5223
5224                                 if (zone[0] == '-')
5225                                         tz = -tz;
5226
5227                                 time -= tz;
5228                         }
5229
5230                         gmtime_r(&time, &commit->time);
5231                 }
5232                 break;
5233         }
5234         default:
5235                 /* Fill in the commit title if it has not already been set. */
5236                 if (commit->title[0])
5237                         break;
5238
5239                 /* Require titles to start with a non-space character at the
5240                  * offset used by git log. */
5241                 if (strncmp(line, "    ", 4))
5242                         break;
5243                 line += 4;
5244                 /* Well, if the title starts with a whitespace character,
5245                  * try to be forgiving.  Otherwise we end up with no title. */
5246                 while (isspace(*line))
5247                         line++;
5248                 if (*line == '\0')
5249                         break;
5250                 /* FIXME: More graceful handling of titles; append "..." to
5251                  * shortened titles, etc. */
5252
5253                 string_ncopy(commit->title, line, strlen(line));
5254         }
5255
5256         return TRUE;
5257 }
5258
5259 static enum request
5260 main_request(struct view *view, enum request request, struct line *line)
5261 {
5262         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5263
5264         switch (request) {
5265         case REQ_ENTER:
5266                 open_view(view, REQ_VIEW_DIFF, flags);
5267                 break;
5268         case REQ_REFRESH:
5269                 load_refs();
5270                 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
5271                 break;
5272         default:
5273                 return request;
5274         }
5275
5276         return REQ_NONE;
5277 }
5278
5279 static bool
5280 grep_refs(struct ref **refs, regex_t *regex)
5281 {
5282         regmatch_t pmatch;
5283         size_t i = 0;
5284
5285         if (!refs)
5286                 return FALSE;
5287         do {
5288                 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
5289                         return TRUE;
5290         } while (refs[i++]->next);
5291
5292         return FALSE;
5293 }
5294
5295 static bool
5296 main_grep(struct view *view, struct line *line)
5297 {
5298         struct commit *commit = line->data;
5299         enum { S_TITLE, S_AUTHOR, S_DATE, S_REFS, S_END } state;
5300         char buf[DATE_COLS + 1];
5301         regmatch_t pmatch;
5302
5303         for (state = S_TITLE; state < S_END; state++) {
5304                 char *text;
5305
5306                 switch (state) {
5307                 case S_TITLE:   text = commit->title;   break;
5308                 case S_AUTHOR:
5309                         if (!opt_author)
5310                                 continue;
5311                         text = commit->author;
5312                         break;
5313                 case S_DATE:
5314                         if (!opt_date)
5315                                 continue;
5316                         if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
5317                                 continue;
5318                         text = buf;
5319                         break;
5320                 case S_REFS:
5321                         if (!opt_show_refs)
5322                                 continue;
5323                         if (grep_refs(commit->refs, view->regex) == TRUE)
5324                                 return TRUE;
5325                         continue;
5326                 default:
5327                         return FALSE;
5328                 }
5329
5330                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5331                         return TRUE;
5332         }
5333
5334         return FALSE;
5335 }
5336
5337 static void
5338 main_select(struct view *view, struct line *line)
5339 {
5340         struct commit *commit = line->data;
5341
5342         string_copy_rev(view->ref, commit->id);
5343         string_copy_rev(ref_commit, view->ref);
5344 }
5345
5346 static struct view_ops main_ops = {
5347         "commit",
5348         NULL,
5349         main_read,
5350         main_draw,
5351         main_request,
5352         main_grep,
5353         main_select,
5354 };
5355
5356
5357 /*
5358  * Unicode / UTF-8 handling
5359  *
5360  * NOTE: Much of the following code for dealing with unicode is derived from
5361  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
5362  * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
5363  */
5364
5365 /* I've (over)annotated a lot of code snippets because I am not entirely
5366  * confident that the approach taken by this small UTF-8 interface is correct.
5367  * --jonas */
5368
5369 static inline int
5370 unicode_width(unsigned long c)
5371 {
5372         if (c >= 0x1100 &&
5373            (c <= 0x115f                         /* Hangul Jamo */
5374             || c == 0x2329
5375             || c == 0x232a
5376             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
5377                                                 /* CJK ... Yi */
5378             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
5379             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
5380             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
5381             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
5382             || (c >= 0xffe0  && c <= 0xffe6)
5383             || (c >= 0x20000 && c <= 0x2fffd)
5384             || (c >= 0x30000 && c <= 0x3fffd)))
5385                 return 2;
5386
5387         if (c == '\t')
5388                 return opt_tab_size;
5389
5390         return 1;
5391 }
5392
5393 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
5394  * Illegal bytes are set one. */
5395 static const unsigned char utf8_bytes[256] = {
5396         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,
5397         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,
5398         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,
5399         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,
5400         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,
5401         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,
5402         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,
5403         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,
5404 };
5405
5406 /* Decode UTF-8 multi-byte representation into a unicode character. */
5407 static inline unsigned long
5408 utf8_to_unicode(const char *string, size_t length)
5409 {
5410         unsigned long unicode;
5411
5412         switch (length) {
5413         case 1:
5414                 unicode  =   string[0];
5415                 break;
5416         case 2:
5417                 unicode  =  (string[0] & 0x1f) << 6;
5418                 unicode +=  (string[1] & 0x3f);
5419                 break;
5420         case 3:
5421                 unicode  =  (string[0] & 0x0f) << 12;
5422                 unicode += ((string[1] & 0x3f) << 6);
5423                 unicode +=  (string[2] & 0x3f);
5424                 break;
5425         case 4:
5426                 unicode  =  (string[0] & 0x0f) << 18;
5427                 unicode += ((string[1] & 0x3f) << 12);
5428                 unicode += ((string[2] & 0x3f) << 6);
5429                 unicode +=  (string[3] & 0x3f);
5430                 break;
5431         case 5:
5432                 unicode  =  (string[0] & 0x0f) << 24;
5433                 unicode += ((string[1] & 0x3f) << 18);
5434                 unicode += ((string[2] & 0x3f) << 12);
5435                 unicode += ((string[3] & 0x3f) << 6);
5436                 unicode +=  (string[4] & 0x3f);
5437                 break;
5438         case 6:
5439                 unicode  =  (string[0] & 0x01) << 30;
5440                 unicode += ((string[1] & 0x3f) << 24);
5441                 unicode += ((string[2] & 0x3f) << 18);
5442                 unicode += ((string[3] & 0x3f) << 12);
5443                 unicode += ((string[4] & 0x3f) << 6);
5444                 unicode +=  (string[5] & 0x3f);
5445                 break;
5446         default:
5447                 die("Invalid unicode length");
5448         }
5449
5450         /* Invalid characters could return the special 0xfffd value but NUL
5451          * should be just as good. */
5452         return unicode > 0xffff ? 0 : unicode;
5453 }
5454
5455 /* Calculates how much of string can be shown within the given maximum width
5456  * and sets trimmed parameter to non-zero value if all of string could not be
5457  * shown. If the reserve flag is TRUE, it will reserve at least one
5458  * trailing character, which can be useful when drawing a delimiter.
5459  *
5460  * Returns the number of bytes to output from string to satisfy max_width. */
5461 static size_t
5462 utf8_length(const char *string, int *width, size_t max_width, int *trimmed, bool reserve)
5463 {
5464         const char *start = string;
5465         const char *end = strchr(string, '\0');
5466         unsigned char last_bytes = 0;
5467         size_t last_ucwidth = 0;
5468
5469         *width = 0;
5470         *trimmed = 0;
5471
5472         while (string < end) {
5473                 int c = *(unsigned char *) string;
5474                 unsigned char bytes = utf8_bytes[c];
5475                 size_t ucwidth;
5476                 unsigned long unicode;
5477
5478                 if (string + bytes > end)
5479                         break;
5480
5481                 /* Change representation to figure out whether
5482                  * it is a single- or double-width character. */
5483
5484                 unicode = utf8_to_unicode(string, bytes);
5485                 /* FIXME: Graceful handling of invalid unicode character. */
5486                 if (!unicode)
5487                         break;
5488
5489                 ucwidth = unicode_width(unicode);
5490                 *width  += ucwidth;
5491                 if (*width > max_width) {
5492                         *trimmed = 1;
5493                         *width -= ucwidth;
5494                         if (reserve && *width == max_width) {
5495                                 string -= last_bytes;
5496                                 *width -= last_ucwidth;
5497                         }
5498                         break;
5499                 }
5500
5501                 string  += bytes;
5502                 last_bytes = bytes;
5503                 last_ucwidth = ucwidth;
5504         }
5505
5506         return string - start;
5507 }
5508
5509
5510 /*
5511  * Status management
5512  */
5513
5514 /* Whether or not the curses interface has been initialized. */
5515 static bool cursed = FALSE;
5516
5517 /* The status window is used for polling keystrokes. */
5518 static WINDOW *status_win;
5519
5520 static bool status_empty = TRUE;
5521
5522 /* Update status and title window. */
5523 static void
5524 report(const char *msg, ...)
5525 {
5526         struct view *view = display[current_view];
5527
5528         if (input_mode)
5529                 return;
5530
5531         if (!view) {
5532                 char buf[SIZEOF_STR];
5533                 va_list args;
5534
5535                 va_start(args, msg);
5536                 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
5537                         buf[sizeof(buf) - 1] = 0;
5538                         buf[sizeof(buf) - 2] = '.';
5539                         buf[sizeof(buf) - 3] = '.';
5540                         buf[sizeof(buf) - 4] = '.';
5541                 }
5542                 va_end(args);
5543                 die("%s", buf);
5544         }
5545
5546         if (!status_empty || *msg) {
5547                 va_list args;
5548
5549                 va_start(args, msg);
5550
5551                 wmove(status_win, 0, 0);
5552                 if (*msg) {
5553                         vwprintw(status_win, msg, args);
5554                         status_empty = FALSE;
5555                 } else {
5556                         status_empty = TRUE;
5557                 }
5558                 wclrtoeol(status_win);
5559                 wrefresh(status_win);
5560
5561                 va_end(args);
5562         }
5563
5564         update_view_title(view);
5565         update_display_cursor(view);
5566 }
5567
5568 /* Controls when nodelay should be in effect when polling user input. */
5569 static void
5570 set_nonblocking_input(bool loading)
5571 {
5572         static unsigned int loading_views;
5573
5574         if ((loading == FALSE && loading_views-- == 1) ||
5575             (loading == TRUE  && loading_views++ == 0))
5576                 nodelay(status_win, loading);
5577 }
5578
5579 static void
5580 init_display(void)
5581 {
5582         int x, y;
5583
5584         /* Initialize the curses library */
5585         if (isatty(STDIN_FILENO)) {
5586                 cursed = !!initscr();
5587                 opt_tty = stdin;
5588         } else {
5589                 /* Leave stdin and stdout alone when acting as a pager. */
5590                 opt_tty = fopen("/dev/tty", "r+");
5591                 if (!opt_tty)
5592                         die("Failed to open /dev/tty");
5593                 cursed = !!newterm(NULL, opt_tty, opt_tty);
5594         }
5595
5596         if (!cursed)
5597                 die("Failed to initialize curses");
5598
5599         nonl();         /* Tell curses not to do NL->CR/NL on output */
5600         cbreak();       /* Take input chars one at a time, no wait for \n */
5601         noecho();       /* Don't echo input */
5602         leaveok(stdscr, TRUE);
5603
5604         if (has_colors())
5605                 init_colors();
5606
5607         getmaxyx(stdscr, y, x);
5608         status_win = newwin(1, 0, y - 1, 0);
5609         if (!status_win)
5610                 die("Failed to create status window");
5611
5612         /* Enable keyboard mapping */
5613         keypad(status_win, TRUE);
5614         wbkgdset(status_win, get_line_attr(LINE_STATUS));
5615
5616         TABSIZE = opt_tab_size;
5617         if (opt_line_graphics) {
5618                 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
5619         }
5620 }
5621
5622 static bool
5623 prompt_yesno(const char *prompt)
5624 {
5625         enum { WAIT, STOP, CANCEL  } status = WAIT;
5626         bool answer = FALSE;
5627
5628         while (status == WAIT) {
5629                 struct view *view;
5630                 int i, key;
5631
5632                 input_mode = TRUE;
5633
5634                 foreach_view (view, i)
5635                         update_view(view);
5636
5637                 input_mode = FALSE;
5638
5639                 mvwprintw(status_win, 0, 0, "%s [Yy]/[Nn]", prompt);
5640                 wclrtoeol(status_win);
5641
5642                 /* Refresh, accept single keystroke of input */
5643                 key = wgetch(status_win);
5644                 switch (key) {
5645                 case ERR:
5646                         break;
5647
5648                 case 'y':
5649                 case 'Y':
5650                         answer = TRUE;
5651                         status = STOP;
5652                         break;
5653
5654                 case KEY_ESC:
5655                 case KEY_RETURN:
5656                 case KEY_ENTER:
5657                 case KEY_BACKSPACE:
5658                 case 'n':
5659                 case 'N':
5660                 case '\n':
5661                 default:
5662                         answer = FALSE;
5663                         status = CANCEL;
5664                 }
5665         }
5666
5667         /* Clear the status window */
5668         status_empty = FALSE;
5669         report("");
5670
5671         return answer;
5672 }
5673
5674 static char *
5675 read_prompt(const char *prompt)
5676 {
5677         enum { READING, STOP, CANCEL } status = READING;
5678         static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
5679         int pos = 0;
5680
5681         while (status == READING) {
5682                 struct view *view;
5683                 int i, key;
5684
5685                 input_mode = TRUE;
5686
5687                 foreach_view (view, i)
5688                         update_view(view);
5689
5690                 input_mode = FALSE;
5691
5692                 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
5693                 wclrtoeol(status_win);
5694
5695                 /* Refresh, accept single keystroke of input */
5696                 key = wgetch(status_win);
5697                 switch (key) {
5698                 case KEY_RETURN:
5699                 case KEY_ENTER:
5700                 case '\n':
5701                         status = pos ? STOP : CANCEL;
5702                         break;
5703
5704                 case KEY_BACKSPACE:
5705                         if (pos > 0)
5706                                 pos--;
5707                         else
5708                                 status = CANCEL;
5709                         break;
5710
5711                 case KEY_ESC:
5712                         status = CANCEL;
5713                         break;
5714
5715                 case ERR:
5716                         break;
5717
5718                 default:
5719                         if (pos >= sizeof(buf)) {
5720                                 report("Input string too long");
5721                                 return NULL;
5722                         }
5723
5724                         if (isprint(key))
5725                                 buf[pos++] = (char) key;
5726                 }
5727         }
5728
5729         /* Clear the status window */
5730         status_empty = FALSE;
5731         report("");
5732
5733         if (status == CANCEL)
5734                 return NULL;
5735
5736         buf[pos++] = 0;
5737
5738         return buf;
5739 }
5740
5741 /*
5742  * Repository references
5743  */
5744
5745 static struct ref *refs = NULL;
5746 static size_t refs_alloc = 0;
5747 static size_t refs_size = 0;
5748
5749 /* Id <-> ref store */
5750 static struct ref ***id_refs = NULL;
5751 static size_t id_refs_alloc = 0;
5752 static size_t id_refs_size = 0;
5753
5754 static int
5755 compare_refs(const void *ref1_, const void *ref2_)
5756 {
5757         const struct ref *ref1 = *(const struct ref **)ref1_;
5758         const struct ref *ref2 = *(const struct ref **)ref2_;
5759
5760         if (ref1->tag != ref2->tag)
5761                 return ref2->tag - ref1->tag;
5762         if (ref1->ltag != ref2->ltag)
5763                 return ref2->ltag - ref2->ltag;
5764         if (ref1->head != ref2->head)
5765                 return ref2->head - ref1->head;
5766         if (ref1->tracked != ref2->tracked)
5767                 return ref2->tracked - ref1->tracked;
5768         if (ref1->remote != ref2->remote)
5769                 return ref2->remote - ref1->remote;
5770         return strcmp(ref1->name, ref2->name);
5771 }
5772
5773 static struct ref **
5774 get_refs(const char *id)
5775 {
5776         struct ref ***tmp_id_refs;
5777         struct ref **ref_list = NULL;
5778         size_t ref_list_alloc = 0;
5779         size_t ref_list_size = 0;
5780         size_t i;
5781
5782         for (i = 0; i < id_refs_size; i++)
5783                 if (!strcmp(id, id_refs[i][0]->id))
5784                         return id_refs[i];
5785
5786         tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
5787                                     sizeof(*id_refs));
5788         if (!tmp_id_refs)
5789                 return NULL;
5790
5791         id_refs = tmp_id_refs;
5792
5793         for (i = 0; i < refs_size; i++) {
5794                 struct ref **tmp;
5795
5796                 if (strcmp(id, refs[i].id))
5797                         continue;
5798
5799                 tmp = realloc_items(ref_list, &ref_list_alloc,
5800                                     ref_list_size + 1, sizeof(*ref_list));
5801                 if (!tmp) {
5802                         if (ref_list)
5803                                 free(ref_list);
5804                         return NULL;
5805                 }
5806
5807                 ref_list = tmp;
5808                 ref_list[ref_list_size] = &refs[i];
5809                 /* XXX: The properties of the commit chains ensures that we can
5810                  * safely modify the shared ref. The repo references will
5811                  * always be similar for the same id. */
5812                 ref_list[ref_list_size]->next = 1;
5813
5814                 ref_list_size++;
5815         }
5816
5817         if (ref_list) {
5818                 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
5819                 ref_list[ref_list_size - 1]->next = 0;
5820                 id_refs[id_refs_size++] = ref_list;
5821         }
5822
5823         return ref_list;
5824 }
5825
5826 static int
5827 read_ref(char *id, size_t idlen, char *name, size_t namelen)
5828 {
5829         struct ref *ref;
5830         bool tag = FALSE;
5831         bool ltag = FALSE;
5832         bool remote = FALSE;
5833         bool tracked = FALSE;
5834         bool check_replace = FALSE;
5835         bool head = FALSE;
5836
5837         if (!prefixcmp(name, "refs/tags/")) {
5838                 if (!suffixcmp(name, namelen, "^{}")) {
5839                         namelen -= 3;
5840                         name[namelen] = 0;
5841                         if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
5842                                 check_replace = TRUE;
5843                 } else {
5844                         ltag = TRUE;
5845                 }
5846
5847                 tag = TRUE;
5848                 namelen -= STRING_SIZE("refs/tags/");
5849                 name    += STRING_SIZE("refs/tags/");
5850
5851         } else if (!prefixcmp(name, "refs/remotes/")) {
5852                 remote = TRUE;
5853                 namelen -= STRING_SIZE("refs/remotes/");
5854                 name    += STRING_SIZE("refs/remotes/");
5855                 tracked  = !strcmp(opt_remote, name);
5856
5857         } else if (!prefixcmp(name, "refs/heads/")) {
5858                 namelen -= STRING_SIZE("refs/heads/");
5859                 name    += STRING_SIZE("refs/heads/");
5860                 head     = !strncmp(opt_head, name, namelen);
5861
5862         } else if (!strcmp(name, "HEAD")) {
5863                 string_ncopy(opt_head_rev, id, idlen);
5864                 return OK;
5865         }
5866
5867         if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
5868                 /* it's an annotated tag, replace the previous sha1 with the
5869                  * resolved commit id; relies on the fact git-ls-remote lists
5870                  * the commit id of an annotated tag right before the commit id
5871                  * it points to. */
5872                 refs[refs_size - 1].ltag = ltag;
5873                 string_copy_rev(refs[refs_size - 1].id, id);
5874
5875                 return OK;
5876         }
5877         refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
5878         if (!refs)
5879                 return ERR;
5880
5881         ref = &refs[refs_size++];
5882         ref->name = malloc(namelen + 1);
5883         if (!ref->name)
5884                 return ERR;
5885
5886         strncpy(ref->name, name, namelen);
5887         ref->name[namelen] = 0;
5888         ref->head = head;
5889         ref->tag = tag;
5890         ref->ltag = ltag;
5891         ref->remote = remote;
5892         ref->tracked = tracked;
5893         string_copy_rev(ref->id, id);
5894
5895         return OK;
5896 }
5897
5898 static int
5899 load_refs(void)
5900 {
5901         const char *cmd_env = getenv("TIG_LS_REMOTE");
5902         const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
5903
5904         if (!*opt_git_dir)
5905                 return OK;
5906
5907         while (refs_size > 0)
5908                 free(refs[--refs_size].name);
5909         while (id_refs_size > 0)
5910                 free(id_refs[--id_refs_size]);
5911
5912         return read_properties(popen(cmd, "r"), "\t", read_ref);
5913 }
5914
5915 static int
5916 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
5917 {
5918         if (!strcmp(name, "i18n.commitencoding"))
5919                 string_ncopy(opt_encoding, value, valuelen);
5920
5921         if (!strcmp(name, "core.editor"))
5922                 string_ncopy(opt_editor, value, valuelen);
5923
5924         /* branch.<head>.remote */
5925         if (*opt_head &&
5926             !strncmp(name, "branch.", 7) &&
5927             !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5928             !strcmp(name + 7 + strlen(opt_head), ".remote"))
5929                 string_ncopy(opt_remote, value, valuelen);
5930
5931         if (*opt_head && *opt_remote &&
5932             !strncmp(name, "branch.", 7) &&
5933             !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5934             !strcmp(name + 7 + strlen(opt_head), ".merge")) {
5935                 size_t from = strlen(opt_remote);
5936
5937                 if (!prefixcmp(value, "refs/heads/")) {
5938                         value += STRING_SIZE("refs/heads/");
5939                         valuelen -= STRING_SIZE("refs/heads/");
5940                 }
5941
5942                 if (!string_format_from(opt_remote, &from, "/%s", value))
5943                         opt_remote[0] = 0;
5944         }
5945
5946         return OK;
5947 }
5948
5949 static int
5950 load_git_config(void)
5951 {
5952         return read_properties(popen("git " GIT_CONFIG " --list", "r"),
5953                                "=", read_repo_config_option);
5954 }
5955
5956 static int
5957 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
5958 {
5959         if (!opt_git_dir[0]) {
5960                 string_ncopy(opt_git_dir, name, namelen);
5961
5962         } else if (opt_is_inside_work_tree == -1) {
5963                 /* This can be 3 different values depending on the
5964                  * version of git being used. If git-rev-parse does not
5965                  * understand --is-inside-work-tree it will simply echo
5966                  * the option else either "true" or "false" is printed.
5967                  * Default to true for the unknown case. */
5968                 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
5969
5970         } else if (opt_cdup[0] == ' ') {
5971                 string_ncopy(opt_cdup, name, namelen);
5972         } else {
5973                 if (!prefixcmp(name, "refs/heads/")) {
5974                         namelen -= STRING_SIZE("refs/heads/");
5975                         name    += STRING_SIZE("refs/heads/");
5976                         string_ncopy(opt_head, name, namelen);
5977                 }
5978         }
5979
5980         return OK;
5981 }
5982
5983 static int
5984 load_repo_info(void)
5985 {
5986         int result;
5987         FILE *pipe = popen("(git rev-parse --git-dir --is-inside-work-tree "
5988                            " --show-cdup; git symbolic-ref HEAD) 2>/dev/null", "r");
5989
5990         /* XXX: The line outputted by "--show-cdup" can be empty so
5991          * initialize it to something invalid to make it possible to
5992          * detect whether it has been set or not. */
5993         opt_cdup[0] = ' ';
5994
5995         result = read_properties(pipe, "=", read_repo_info);
5996         if (opt_cdup[0] == ' ')
5997                 opt_cdup[0] = 0;
5998
5999         return result;
6000 }
6001
6002 static int
6003 read_properties(FILE *pipe, const char *separators,
6004                 int (*read_property)(char *, size_t, char *, size_t))
6005 {
6006         char buffer[BUFSIZ];
6007         char *name;
6008         int state = OK;
6009
6010         if (!pipe)
6011                 return ERR;
6012
6013         while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
6014                 char *value;
6015                 size_t namelen;
6016                 size_t valuelen;
6017
6018                 name = chomp_string(name);
6019                 namelen = strcspn(name, separators);
6020
6021                 if (name[namelen]) {
6022                         name[namelen] = 0;
6023                         value = chomp_string(name + namelen + 1);
6024                         valuelen = strlen(value);
6025
6026                 } else {
6027                         value = "";
6028                         valuelen = 0;
6029                 }
6030
6031                 state = read_property(name, namelen, value, valuelen);
6032         }
6033
6034         if (state != ERR && ferror(pipe))
6035                 state = ERR;
6036
6037         pclose(pipe);
6038
6039         return state;
6040 }
6041
6042
6043 /*
6044  * Main
6045  */
6046
6047 static void __NORETURN
6048 quit(int sig)
6049 {
6050         /* XXX: Restore tty modes and let the OS cleanup the rest! */
6051         if (cursed)
6052                 endwin();
6053         exit(0);
6054 }
6055
6056 static void __NORETURN
6057 die(const char *err, ...)
6058 {
6059         va_list args;
6060
6061         endwin();
6062
6063         va_start(args, err);
6064         fputs("tig: ", stderr);
6065         vfprintf(stderr, err, args);
6066         fputs("\n", stderr);
6067         va_end(args);
6068
6069         exit(1);
6070 }
6071
6072 static void
6073 warn(const char *msg, ...)
6074 {
6075         va_list args;
6076
6077         va_start(args, msg);
6078         fputs("tig warning: ", stderr);
6079         vfprintf(stderr, msg, args);
6080         fputs("\n", stderr);
6081         va_end(args);
6082 }
6083
6084 int
6085 main(int argc, const char *argv[])
6086 {
6087         struct view *view;
6088         enum request request;
6089         size_t i;
6090
6091         signal(SIGINT, quit);
6092
6093         if (setlocale(LC_ALL, "")) {
6094                 char *codeset = nl_langinfo(CODESET);
6095
6096                 string_ncopy(opt_codeset, codeset, strlen(codeset));
6097         }
6098
6099         if (load_repo_info() == ERR)
6100                 die("Failed to load repo info.");
6101
6102         if (load_options() == ERR)
6103                 die("Failed to load user config.");
6104
6105         if (load_git_config() == ERR)
6106                 die("Failed to load repo config.");
6107
6108         request = parse_options(argc, argv);
6109         if (request == REQ_NONE)
6110                 return 0;
6111
6112         /* Require a git repository unless when running in pager mode. */
6113         if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
6114                 die("Not a git repository");
6115
6116         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
6117                 opt_utf8 = FALSE;
6118
6119         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
6120                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
6121                 if (opt_iconv == ICONV_NONE)
6122                         die("Failed to initialize character set conversion");
6123         }
6124
6125         if (load_refs() == ERR)
6126                 die("Failed to load refs.");
6127
6128         foreach_view (view, i)
6129                 view->cmd_env = getenv(view->cmd_env);
6130
6131         init_display();
6132
6133         while (view_driver(display[current_view], request)) {
6134                 int key;
6135                 int i;
6136
6137                 foreach_view (view, i)
6138                         update_view(view);
6139                 view = display[current_view];
6140
6141                 /* Refresh, accept single keystroke of input */
6142                 key = wgetch(status_win);
6143
6144                 /* wgetch() with nodelay() enabled returns ERR when there's no
6145                  * input. */
6146                 if (key == ERR) {
6147                         request = REQ_NONE;
6148                         continue;
6149                 }
6150
6151                 request = get_keybinding(view->keymap, key);
6152
6153                 /* Some low-level request handling. This keeps access to
6154                  * status_win restricted. */
6155                 switch (request) {
6156                 case REQ_PROMPT:
6157                 {
6158                         char *cmd = read_prompt(":");
6159
6160                         if (cmd && string_format(opt_cmd, "git %s", cmd)) {
6161                                 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
6162                                         request = REQ_VIEW_DIFF;
6163                                 } else {
6164                                         request = REQ_VIEW_PAGER;
6165                                 }
6166
6167                                 /* Always reload^Wrerun commands from the prompt. */
6168                                 open_view(view, request, OPEN_RELOAD);
6169                         }
6170
6171                         request = REQ_NONE;
6172                         break;
6173                 }
6174                 case REQ_SEARCH:
6175                 case REQ_SEARCH_BACK:
6176                 {
6177                         const char *prompt = request == REQ_SEARCH ? "/" : "?";
6178                         char *search = read_prompt(prompt);
6179
6180                         if (search)
6181                                 string_ncopy(opt_search, search, strlen(search));
6182                         else
6183                                 request = REQ_NONE;
6184                         break;
6185                 }
6186                 case REQ_SCREEN_RESIZE:
6187                 {
6188                         int height, width;
6189
6190                         getmaxyx(stdscr, height, width);
6191
6192                         /* Resize the status view and let the view driver take
6193                          * care of resizing the displayed views. */
6194                         wresize(status_win, 1, width);
6195                         mvwin(status_win, height - 1, 0);
6196                         wrefresh(status_win);
6197                         break;
6198                 }
6199                 default:
6200                         break;
6201                 }
6202         }
6203
6204         quit(0);
6205
6206         return 0;
6207 }