Slightly reduce memory usage for keybindings
[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 bool
2238 begin_update(struct view *view, bool refresh)
2239 {
2240         if (opt_cmd[0]) {
2241                 string_copy(view->cmd, opt_cmd);
2242                 opt_cmd[0] = 0;
2243                 /* When running random commands, initially show the
2244                  * command in the title. However, it maybe later be
2245                  * overwritten if a commit line is selected. */
2246                 if (view == VIEW(REQ_VIEW_PAGER))
2247                         string_copy(view->ref, view->cmd);
2248                 else
2249                         view->ref[0] = 0;
2250
2251         } else if (view == VIEW(REQ_VIEW_TREE)) {
2252                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2253                 char path[SIZEOF_STR];
2254
2255                 if (strcmp(view->vid, view->id))
2256                         opt_path[0] = path[0] = 0;
2257                 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
2258                         return FALSE;
2259
2260                 if (!string_format(view->cmd, format, view->id, path))
2261                         return FALSE;
2262
2263         } else if (!refresh) {
2264                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2265                 const char *id = view->id;
2266
2267                 if (!string_format(view->cmd, format, id, id, id, id, id))
2268                         return FALSE;
2269
2270                 /* Put the current ref_* value to the view title ref
2271                  * member. This is needed by the blob view. Most other
2272                  * views sets it automatically after loading because the
2273                  * first line is a commit line. */
2274                 string_copy_rev(view->ref, view->id);
2275         }
2276
2277         /* Special case for the pager view. */
2278         if (opt_pipe) {
2279                 view->pipe = opt_pipe;
2280                 opt_pipe = NULL;
2281         } else {
2282                 view->pipe = popen(view->cmd, "r");
2283         }
2284
2285         if (!view->pipe)
2286                 return FALSE;
2287
2288         set_nonblocking_input(TRUE);
2289         reset_view(view);
2290         string_copy_rev(view->vid, view->id);
2291
2292         view->start_time = time(NULL);
2293
2294         return TRUE;
2295 }
2296
2297 #define ITEM_CHUNK_SIZE 256
2298 static void *
2299 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2300 {
2301         size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2302         size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2303
2304         if (mem == NULL || num_chunks != num_chunks_new) {
2305                 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2306                 mem = realloc(mem, *size * item_size);
2307         }
2308
2309         return mem;
2310 }
2311
2312 static struct line *
2313 realloc_lines(struct view *view, size_t line_size)
2314 {
2315         size_t alloc = view->line_alloc;
2316         struct line *tmp = realloc_items(view->line, &alloc, line_size,
2317                                          sizeof(*view->line));
2318
2319         if (!tmp)
2320                 return NULL;
2321
2322         view->line = tmp;
2323         view->line_alloc = alloc;
2324         view->line_size = line_size;
2325         return view->line;
2326 }
2327
2328 static bool
2329 update_view(struct view *view)
2330 {
2331         char in_buffer[BUFSIZ];
2332         char out_buffer[BUFSIZ * 2];
2333         char *line;
2334         /* The number of lines to read. If too low it will cause too much
2335          * redrawing (and possible flickering), if too high responsiveness
2336          * will suffer. */
2337         unsigned long lines = view->height;
2338         int redraw_from = -1;
2339
2340         if (!view->pipe)
2341                 return TRUE;
2342
2343         /* Only redraw if lines are visible. */
2344         if (view->offset + view->height >= view->lines)
2345                 redraw_from = view->lines - view->offset;
2346
2347         /* FIXME: This is probably not perfect for backgrounded views. */
2348         if (!realloc_lines(view, view->lines + lines))
2349                 goto alloc_error;
2350
2351         while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
2352                 size_t linelen = strlen(line);
2353
2354                 if (linelen)
2355                         line[linelen - 1] = 0;
2356
2357                 if (opt_iconv != ICONV_NONE) {
2358                         ICONV_CONST char *inbuf = line;
2359                         size_t inlen = linelen;
2360
2361                         char *outbuf = out_buffer;
2362                         size_t outlen = sizeof(out_buffer);
2363
2364                         size_t ret;
2365
2366                         ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2367                         if (ret != (size_t) -1) {
2368                                 line = out_buffer;
2369                                 linelen = strlen(out_buffer);
2370                         }
2371                 }
2372
2373                 if (!view->ops->read(view, line))
2374                         goto alloc_error;
2375
2376                 if (lines-- == 1)
2377                         break;
2378         }
2379
2380         {
2381                 int digits;
2382
2383                 lines = view->lines;
2384                 for (digits = 0; lines; digits++)
2385                         lines /= 10;
2386
2387                 /* Keep the displayed view in sync with line number scaling. */
2388                 if (digits != view->digits) {
2389                         view->digits = digits;
2390                         redraw_from = 0;
2391                 }
2392         }
2393
2394         if (ferror(view->pipe) && errno != 0) {
2395                 report("Failed to read: %s", strerror(errno));
2396                 end_update(view, TRUE);
2397
2398         } else if (feof(view->pipe)) {
2399                 report("");
2400                 end_update(view, FALSE);
2401         }
2402
2403         if (!view_is_displayed(view))
2404                 return TRUE;
2405
2406         if (view == VIEW(REQ_VIEW_TREE)) {
2407                 /* Clear the view and redraw everything since the tree sorting
2408                  * might have rearranged things. */
2409                 redraw_view(view);
2410
2411         } else if (redraw_from >= 0) {
2412                 /* If this is an incremental update, redraw the previous line
2413                  * since for commits some members could have changed when
2414                  * loading the main view. */
2415                 if (redraw_from > 0)
2416                         redraw_from--;
2417
2418                 /* Since revision graph visualization requires knowledge
2419                  * about the parent commit, it causes a further one-off
2420                  * needed to be redrawn for incremental updates. */
2421                 if (redraw_from > 0 && opt_rev_graph)
2422                         redraw_from--;
2423
2424                 /* Incrementally draw avoids flickering. */
2425                 redraw_view_from(view, redraw_from);
2426         }
2427
2428         if (view == VIEW(REQ_VIEW_BLAME))
2429                 redraw_view_dirty(view);
2430
2431         /* Update the title _after_ the redraw so that if the redraw picks up a
2432          * commit reference in view->ref it'll be available here. */
2433         update_view_title(view);
2434         return TRUE;
2435
2436 alloc_error:
2437         report("Allocation failure");
2438         end_update(view, TRUE);
2439         return FALSE;
2440 }
2441
2442 static struct line *
2443 add_line_data(struct view *view, void *data, enum line_type type)
2444 {
2445         struct line *line = &view->line[view->lines++];
2446
2447         memset(line, 0, sizeof(*line));
2448         line->type = type;
2449         line->data = data;
2450
2451         return line;
2452 }
2453
2454 static struct line *
2455 add_line_text(struct view *view, const char *text, enum line_type type)
2456 {
2457         char *data = text ? strdup(text) : NULL;
2458
2459         return data ? add_line_data(view, data, type) : NULL;
2460 }
2461
2462
2463 /*
2464  * View opening
2465  */
2466
2467 enum open_flags {
2468         OPEN_DEFAULT = 0,       /* Use default view switching. */
2469         OPEN_SPLIT = 1,         /* Split current view. */
2470         OPEN_BACKGROUNDED = 2,  /* Backgrounded. */
2471         OPEN_RELOAD = 4,        /* Reload view even if it is the current. */
2472         OPEN_NOMAXIMIZE = 8,    /* Do not maximize the current view. */
2473         OPEN_REFRESH = 16,      /* Refresh view using previous command. */
2474 };
2475
2476 static void
2477 open_view(struct view *prev, enum request request, enum open_flags flags)
2478 {
2479         bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2480         bool split = !!(flags & OPEN_SPLIT);
2481         bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH));
2482         bool nomaximize = !!(flags & (OPEN_NOMAXIMIZE | OPEN_REFRESH));
2483         struct view *view = VIEW(request);
2484         int nviews = displayed_views();
2485         struct view *base_view = display[0];
2486
2487         if (view == prev && nviews == 1 && !reload) {
2488                 report("Already in %s view", view->name);
2489                 return;
2490         }
2491
2492         if (view->git_dir && !opt_git_dir[0]) {
2493                 report("The %s view is disabled in pager view", view->name);
2494                 return;
2495         }
2496
2497         if (split) {
2498                 display[1] = view;
2499                 if (!backgrounded)
2500                         current_view = 1;
2501         } else if (!nomaximize) {
2502                 /* Maximize the current view. */
2503                 memset(display, 0, sizeof(display));
2504                 current_view = 0;
2505                 display[current_view] = view;
2506         }
2507
2508         /* Resize the view when switching between split- and full-screen,
2509          * or when switching between two different full-screen views. */
2510         if (nviews != displayed_views() ||
2511             (nviews == 1 && base_view != display[0]))
2512                 resize_display();
2513
2514         if (view->pipe)
2515                 end_update(view, TRUE);
2516
2517         if (view->ops->open) {
2518                 if (!view->ops->open(view)) {
2519                         report("Failed to load %s view", view->name);
2520                         return;
2521                 }
2522
2523         } else if ((reload || strcmp(view->vid, view->id)) &&
2524                    !begin_update(view, flags & OPEN_REFRESH)) {
2525                 report("Failed to load %s view", view->name);
2526                 return;
2527         }
2528
2529         if (split && prev->lineno - prev->offset >= prev->height) {
2530                 /* Take the title line into account. */
2531                 int lines = prev->lineno - prev->offset - prev->height + 1;
2532
2533                 /* Scroll the view that was split if the current line is
2534                  * outside the new limited view. */
2535                 do_scroll_view(prev, lines);
2536         }
2537
2538         if (prev && view != prev) {
2539                 if (split && !backgrounded) {
2540                         /* "Blur" the previous view. */
2541                         update_view_title(prev);
2542                 }
2543
2544                 view->parent = prev;
2545         }
2546
2547         if (view->pipe && view->lines == 0) {
2548                 /* Clear the old view and let the incremental updating refill
2549                  * the screen. */
2550                 werase(view->win);
2551                 report("");
2552         } else if (view_is_displayed(view)) {
2553                 redraw_view(view);
2554                 report("");
2555         }
2556
2557         /* If the view is backgrounded the above calls to report()
2558          * won't redraw the view title. */
2559         if (backgrounded)
2560                 update_view_title(view);
2561 }
2562
2563 static bool
2564 run_confirm(const char *cmd, const char *prompt)
2565 {
2566         bool confirmation = prompt_yesno(prompt);
2567
2568         if (confirmation)
2569                 system(cmd);
2570
2571         return confirmation;
2572 }
2573
2574 static void
2575 open_external_viewer(const char *cmd)
2576 {
2577         def_prog_mode();           /* save current tty modes */
2578         endwin();                  /* restore original tty modes */
2579         system(cmd);
2580         fprintf(stderr, "Press Enter to continue");
2581         getc(opt_tty);
2582         reset_prog_mode();
2583         redraw_display();
2584 }
2585
2586 static void
2587 open_mergetool(const char *file)
2588 {
2589         char cmd[SIZEOF_STR];
2590         char file_sq[SIZEOF_STR];
2591
2592         if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2593             string_format(cmd, "git mergetool %s", file_sq)) {
2594                 open_external_viewer(cmd);
2595         }
2596 }
2597
2598 static void
2599 open_editor(bool from_root, const char *file)
2600 {
2601         char cmd[SIZEOF_STR];
2602         char file_sq[SIZEOF_STR];
2603         const char *editor;
2604         char *prefix = from_root ? opt_cdup : "";
2605
2606         editor = getenv("GIT_EDITOR");
2607         if (!editor && *opt_editor)
2608                 editor = opt_editor;
2609         if (!editor)
2610                 editor = getenv("VISUAL");
2611         if (!editor)
2612                 editor = getenv("EDITOR");
2613         if (!editor)
2614                 editor = "vi";
2615
2616         if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2617             string_format(cmd, "%s %s%s", editor, prefix, file_sq)) {
2618                 open_external_viewer(cmd);
2619         }
2620 }
2621
2622 static void
2623 open_run_request(enum request request)
2624 {
2625         struct run_request *req = get_run_request(request);
2626         char buf[SIZEOF_STR * 2];
2627         size_t bufpos;
2628         char *cmd;
2629
2630         if (!req) {
2631                 report("Unknown run request");
2632                 return;
2633         }
2634
2635         bufpos = 0;
2636         cmd = req->cmd;
2637
2638         while (cmd) {
2639                 char *next = strstr(cmd, "%(");
2640                 int len = next - cmd;
2641                 char *value;
2642
2643                 if (!next) {
2644                         len = strlen(cmd);
2645                         value = "";
2646
2647                 } else if (!strncmp(next, "%(head)", 7)) {
2648                         value = ref_head;
2649
2650                 } else if (!strncmp(next, "%(commit)", 9)) {
2651                         value = ref_commit;
2652
2653                 } else if (!strncmp(next, "%(blob)", 7)) {
2654                         value = ref_blob;
2655
2656                 } else {
2657                         report("Unknown replacement in run request: `%s`", req->cmd);
2658                         return;
2659                 }
2660
2661                 if (!string_format_from(buf, &bufpos, "%.*s%s", len, cmd, value))
2662                         return;
2663
2664                 if (next)
2665                         next = strchr(next, ')') + 1;
2666                 cmd = next;
2667         }
2668
2669         open_external_viewer(buf);
2670 }
2671
2672 /*
2673  * User request switch noodle
2674  */
2675
2676 static int
2677 view_driver(struct view *view, enum request request)
2678 {
2679         int i;
2680
2681         if (request == REQ_NONE) {
2682                 doupdate();
2683                 return TRUE;
2684         }
2685
2686         if (request > REQ_NONE) {
2687                 open_run_request(request);
2688                 /* FIXME: When all views can refresh always do this. */
2689                 if (view == VIEW(REQ_VIEW_STATUS) ||
2690                     view == VIEW(REQ_VIEW_MAIN) ||
2691                     view == VIEW(REQ_VIEW_LOG) ||
2692                     view == VIEW(REQ_VIEW_STAGE))
2693                         request = REQ_REFRESH;
2694                 else
2695                         return TRUE;
2696         }
2697
2698         if (view && view->lines) {
2699                 request = view->ops->request(view, request, &view->line[view->lineno]);
2700                 if (request == REQ_NONE)
2701                         return TRUE;
2702         }
2703
2704         switch (request) {
2705         case REQ_MOVE_UP:
2706         case REQ_MOVE_DOWN:
2707         case REQ_MOVE_PAGE_UP:
2708         case REQ_MOVE_PAGE_DOWN:
2709         case REQ_MOVE_FIRST_LINE:
2710         case REQ_MOVE_LAST_LINE:
2711                 move_view(view, request);
2712                 break;
2713
2714         case REQ_SCROLL_LINE_DOWN:
2715         case REQ_SCROLL_LINE_UP:
2716         case REQ_SCROLL_PAGE_DOWN:
2717         case REQ_SCROLL_PAGE_UP:
2718                 scroll_view(view, request);
2719                 break;
2720
2721         case REQ_VIEW_BLAME:
2722                 if (!opt_file[0]) {
2723                         report("No file chosen, press %s to open tree view",
2724                                get_key(REQ_VIEW_TREE));
2725                         break;
2726                 }
2727                 open_view(view, request, OPEN_DEFAULT);
2728                 break;
2729
2730         case REQ_VIEW_BLOB:
2731                 if (!ref_blob[0]) {
2732                         report("No file chosen, press %s to open tree view",
2733                                get_key(REQ_VIEW_TREE));
2734                         break;
2735                 }
2736                 open_view(view, request, OPEN_DEFAULT);
2737                 break;
2738
2739         case REQ_VIEW_PAGER:
2740                 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2741                         report("No pager content, press %s to run command from prompt",
2742                                get_key(REQ_PROMPT));
2743                         break;
2744                 }
2745                 open_view(view, request, OPEN_DEFAULT);
2746                 break;
2747
2748         case REQ_VIEW_STAGE:
2749                 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2750                         report("No stage content, press %s to open the status view and choose file",
2751                                get_key(REQ_VIEW_STATUS));
2752                         break;
2753                 }
2754                 open_view(view, request, OPEN_DEFAULT);
2755                 break;
2756
2757         case REQ_VIEW_STATUS:
2758                 if (opt_is_inside_work_tree == FALSE) {
2759                         report("The status view requires a working tree");
2760                         break;
2761                 }
2762                 open_view(view, request, OPEN_DEFAULT);
2763                 break;
2764
2765         case REQ_VIEW_MAIN:
2766         case REQ_VIEW_DIFF:
2767         case REQ_VIEW_LOG:
2768         case REQ_VIEW_TREE:
2769         case REQ_VIEW_HELP:
2770                 open_view(view, request, OPEN_DEFAULT);
2771                 break;
2772
2773         case REQ_NEXT:
2774         case REQ_PREVIOUS:
2775                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2776
2777                 if ((view == VIEW(REQ_VIEW_DIFF) &&
2778                      view->parent == VIEW(REQ_VIEW_MAIN)) ||
2779                    (view == VIEW(REQ_VIEW_DIFF) &&
2780                      view->parent == VIEW(REQ_VIEW_BLAME)) ||
2781                    (view == VIEW(REQ_VIEW_STAGE) &&
2782                      view->parent == VIEW(REQ_VIEW_STATUS)) ||
2783                    (view == VIEW(REQ_VIEW_BLOB) &&
2784                      view->parent == VIEW(REQ_VIEW_TREE))) {
2785                         int line;
2786
2787                         view = view->parent;
2788                         line = view->lineno;
2789                         move_view(view, request);
2790                         if (view_is_displayed(view))
2791                                 update_view_title(view);
2792                         if (line != view->lineno)
2793                                 view->ops->request(view, REQ_ENTER,
2794                                                    &view->line[view->lineno]);
2795
2796                 } else {
2797                         move_view(view, request);
2798                 }
2799                 break;
2800
2801         case REQ_VIEW_NEXT:
2802         {
2803                 int nviews = displayed_views();
2804                 int next_view = (current_view + 1) % nviews;
2805
2806                 if (next_view == current_view) {
2807                         report("Only one view is displayed");
2808                         break;
2809                 }
2810
2811                 current_view = next_view;
2812                 /* Blur out the title of the previous view. */
2813                 update_view_title(view);
2814                 report("");
2815                 break;
2816         }
2817         case REQ_REFRESH:
2818                 report("Refreshing is not yet supported for the %s view", view->name);
2819                 break;
2820
2821         case REQ_MAXIMIZE:
2822                 if (displayed_views() == 2)
2823                         open_view(view, VIEW_REQ(view), OPEN_DEFAULT);
2824                 break;
2825
2826         case REQ_TOGGLE_LINENO:
2827                 opt_line_number = !opt_line_number;
2828                 redraw_display();
2829                 break;
2830
2831         case REQ_TOGGLE_DATE:
2832                 opt_date = !opt_date;
2833                 redraw_display();
2834                 break;
2835
2836         case REQ_TOGGLE_AUTHOR:
2837                 opt_author = !opt_author;
2838                 redraw_display();
2839                 break;
2840
2841         case REQ_TOGGLE_REV_GRAPH:
2842                 opt_rev_graph = !opt_rev_graph;
2843                 redraw_display();
2844                 break;
2845
2846         case REQ_TOGGLE_REFS:
2847                 opt_show_refs = !opt_show_refs;
2848                 redraw_display();
2849                 break;
2850
2851         case REQ_SEARCH:
2852         case REQ_SEARCH_BACK:
2853                 search_view(view, request);
2854                 break;
2855
2856         case REQ_FIND_NEXT:
2857         case REQ_FIND_PREV:
2858                 find_next(view, request);
2859                 break;
2860
2861         case REQ_STOP_LOADING:
2862                 for (i = 0; i < ARRAY_SIZE(views); i++) {
2863                         view = &views[i];
2864                         if (view->pipe)
2865                                 report("Stopped loading the %s view", view->name),
2866                         end_update(view, TRUE);
2867                 }
2868                 break;
2869
2870         case REQ_SHOW_VERSION:
2871                 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2872                 return TRUE;
2873
2874         case REQ_SCREEN_RESIZE:
2875                 resize_display();
2876                 /* Fall-through */
2877         case REQ_SCREEN_REDRAW:
2878                 redraw_display();
2879                 break;
2880
2881         case REQ_EDIT:
2882                 report("Nothing to edit");
2883                 break;
2884
2885         case REQ_ENTER:
2886                 report("Nothing to enter");
2887                 break;
2888
2889         case REQ_VIEW_CLOSE:
2890                 /* XXX: Mark closed views by letting view->parent point to the
2891                  * view itself. Parents to closed view should never be
2892                  * followed. */
2893                 if (view->parent &&
2894                     view->parent->parent != view->parent) {
2895                         memset(display, 0, sizeof(display));
2896                         current_view = 0;
2897                         display[current_view] = view->parent;
2898                         view->parent = view;
2899                         resize_display();
2900                         redraw_display();
2901                         report("");
2902                         break;
2903                 }
2904                 /* Fall-through */
2905         case REQ_QUIT:
2906                 return FALSE;
2907
2908         default:
2909                 report("Unknown key, press 'h' for help");
2910                 return TRUE;
2911         }
2912
2913         return TRUE;
2914 }
2915
2916
2917 /*
2918  * Pager backend
2919  */
2920
2921 static bool
2922 pager_draw(struct view *view, struct line *line, unsigned int lineno)
2923 {
2924         char *text = line->data;
2925
2926         if (opt_line_number && draw_lineno(view, lineno))
2927                 return TRUE;
2928
2929         draw_text(view, line->type, text, TRUE);
2930         return TRUE;
2931 }
2932
2933 static bool
2934 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
2935 {
2936         char refbuf[SIZEOF_STR];
2937         char *ref = NULL;
2938         FILE *pipe;
2939
2940         if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2941                 return TRUE;
2942
2943         pipe = popen(refbuf, "r");
2944         if (!pipe)
2945                 return TRUE;
2946
2947         if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2948                 ref = chomp_string(ref);
2949         pclose(pipe);
2950
2951         if (!ref || !*ref)
2952                 return TRUE;
2953
2954         /* This is the only fatal call, since it can "corrupt" the buffer. */
2955         if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2956                 return FALSE;
2957
2958         return TRUE;
2959 }
2960
2961 static void
2962 add_pager_refs(struct view *view, struct line *line)
2963 {
2964         char buf[SIZEOF_STR];
2965         char *commit_id = (char *)line->data + STRING_SIZE("commit ");
2966         struct ref **refs;
2967         size_t bufpos = 0, refpos = 0;
2968         const char *sep = "Refs: ";
2969         bool is_tag = FALSE;
2970
2971         assert(line->type == LINE_COMMIT);
2972
2973         refs = get_refs(commit_id);
2974         if (!refs) {
2975                 if (view == VIEW(REQ_VIEW_DIFF))
2976                         goto try_add_describe_ref;
2977                 return;
2978         }
2979
2980         do {
2981                 struct ref *ref = refs[refpos];
2982                 const char *fmt = ref->tag    ? "%s[%s]" :
2983                                   ref->remote ? "%s<%s>" : "%s%s";
2984
2985                 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2986                         return;
2987                 sep = ", ";
2988                 if (ref->tag)
2989                         is_tag = TRUE;
2990         } while (refs[refpos++]->next);
2991
2992         if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2993 try_add_describe_ref:
2994                 /* Add <tag>-g<commit_id> "fake" reference. */
2995                 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2996                         return;
2997         }
2998
2999         if (bufpos == 0)
3000                 return;
3001
3002         if (!realloc_lines(view, view->line_size + 1))
3003                 return;
3004
3005         add_line_text(view, buf, LINE_PP_REFS);
3006 }
3007
3008 static bool
3009 pager_read(struct view *view, char *data)
3010 {
3011         struct line *line;
3012
3013         if (!data)
3014                 return TRUE;
3015
3016         line = add_line_text(view, data, get_line_type(data));
3017         if (!line)
3018                 return FALSE;
3019
3020         if (line->type == LINE_COMMIT &&
3021             (view == VIEW(REQ_VIEW_DIFF) ||
3022              view == VIEW(REQ_VIEW_LOG)))
3023                 add_pager_refs(view, line);
3024
3025         return TRUE;
3026 }
3027
3028 static enum request
3029 pager_request(struct view *view, enum request request, struct line *line)
3030 {
3031         int split = 0;
3032
3033         if (request != REQ_ENTER)
3034                 return request;
3035
3036         if (line->type == LINE_COMMIT &&
3037            (view == VIEW(REQ_VIEW_LOG) ||
3038             view == VIEW(REQ_VIEW_PAGER))) {
3039                 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
3040                 split = 1;
3041         }
3042
3043         /* Always scroll the view even if it was split. That way
3044          * you can use Enter to scroll through the log view and
3045          * split open each commit diff. */
3046         scroll_view(view, REQ_SCROLL_LINE_DOWN);
3047
3048         /* FIXME: A minor workaround. Scrolling the view will call report("")
3049          * but if we are scrolling a non-current view this won't properly
3050          * update the view title. */
3051         if (split)
3052                 update_view_title(view);
3053
3054         return REQ_NONE;
3055 }
3056
3057 static bool
3058 pager_grep(struct view *view, struct line *line)
3059 {
3060         regmatch_t pmatch;
3061         char *text = line->data;
3062
3063         if (!*text)
3064                 return FALSE;
3065
3066         if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
3067                 return FALSE;
3068
3069         return TRUE;
3070 }
3071
3072 static void
3073 pager_select(struct view *view, struct line *line)
3074 {
3075         if (line->type == LINE_COMMIT) {
3076                 char *text = (char *)line->data + STRING_SIZE("commit ");
3077
3078                 if (view != VIEW(REQ_VIEW_PAGER))
3079                         string_copy_rev(view->ref, text);
3080                 string_copy_rev(ref_commit, text);
3081         }
3082 }
3083
3084 static struct view_ops pager_ops = {
3085         "line",
3086         NULL,
3087         pager_read,
3088         pager_draw,
3089         pager_request,
3090         pager_grep,
3091         pager_select,
3092 };
3093
3094 static enum request
3095 log_request(struct view *view, enum request request, struct line *line)
3096 {
3097         switch (request) {
3098         case REQ_REFRESH:
3099                 load_refs();
3100                 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
3101                 return REQ_NONE;
3102         default:
3103                 return pager_request(view, request, line);
3104         }
3105 }
3106
3107 static struct view_ops log_ops = {
3108         "line",
3109         NULL,
3110         pager_read,
3111         pager_draw,
3112         log_request,
3113         pager_grep,
3114         pager_select,
3115 };
3116
3117
3118 /*
3119  * Help backend
3120  */
3121
3122 static bool
3123 help_open(struct view *view)
3124 {
3125         char buf[BUFSIZ];
3126         int lines = ARRAY_SIZE(req_info) + 2;
3127         int i;
3128
3129         if (view->lines > 0)
3130                 return TRUE;
3131
3132         for (i = 0; i < ARRAY_SIZE(req_info); i++)
3133                 if (!req_info[i].request)
3134                         lines++;
3135
3136         lines += run_requests + 1;
3137
3138         view->line = calloc(lines, sizeof(*view->line));
3139         if (!view->line)
3140                 return FALSE;
3141
3142         add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3143
3144         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3145                 const char *key;
3146
3147                 if (req_info[i].request == REQ_NONE)
3148                         continue;
3149
3150                 if (!req_info[i].request) {
3151                         add_line_text(view, "", LINE_DEFAULT);
3152                         add_line_text(view, req_info[i].help, LINE_DEFAULT);
3153                         continue;
3154                 }
3155
3156                 key = get_key(req_info[i].request);
3157                 if (!*key)
3158                         key = "(no key defined)";
3159
3160                 if (!string_format(buf, "    %-25s %s", key, req_info[i].help))
3161                         continue;
3162
3163                 add_line_text(view, buf, LINE_DEFAULT);
3164         }
3165
3166         if (run_requests) {
3167                 add_line_text(view, "", LINE_DEFAULT);
3168                 add_line_text(view, "External commands:", LINE_DEFAULT);
3169         }
3170
3171         for (i = 0; i < run_requests; i++) {
3172                 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3173                 const char *key;
3174
3175                 if (!req)
3176                         continue;
3177
3178                 key = get_key_name(req->key);
3179                 if (!*key)
3180                         key = "(no key defined)";
3181
3182                 if (!string_format(buf, "    %-10s %-14s `%s`",
3183                                    keymap_table[req->keymap].name,
3184                                    key, req->cmd))
3185                         continue;
3186
3187                 add_line_text(view, buf, LINE_DEFAULT);
3188         }
3189
3190         return TRUE;
3191 }
3192
3193 static struct view_ops help_ops = {
3194         "line",
3195         help_open,
3196         NULL,
3197         pager_draw,
3198         pager_request,
3199         pager_grep,
3200         pager_select,
3201 };
3202
3203
3204 /*
3205  * Tree backend
3206  */
3207
3208 struct tree_stack_entry {
3209         struct tree_stack_entry *prev;  /* Entry below this in the stack */
3210         unsigned long lineno;           /* Line number to restore */
3211         char *name;                     /* Position of name in opt_path */
3212 };
3213
3214 /* The top of the path stack. */
3215 static struct tree_stack_entry *tree_stack = NULL;
3216 unsigned long tree_lineno = 0;
3217
3218 static void
3219 pop_tree_stack_entry(void)
3220 {
3221         struct tree_stack_entry *entry = tree_stack;
3222
3223         tree_lineno = entry->lineno;
3224         entry->name[0] = 0;
3225         tree_stack = entry->prev;
3226         free(entry);
3227 }
3228
3229 static void
3230 push_tree_stack_entry(const char *name, unsigned long lineno)
3231 {
3232         struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3233         size_t pathlen = strlen(opt_path);
3234
3235         if (!entry)
3236                 return;
3237
3238         entry->prev = tree_stack;
3239         entry->name = opt_path + pathlen;
3240         tree_stack = entry;
3241
3242         if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3243                 pop_tree_stack_entry();
3244                 return;
3245         }
3246
3247         /* Move the current line to the first tree entry. */
3248         tree_lineno = 1;
3249         entry->lineno = lineno;
3250 }
3251
3252 /* Parse output from git-ls-tree(1):
3253  *
3254  * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
3255  * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
3256  * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3257  * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
3258  */
3259
3260 #define SIZEOF_TREE_ATTR \
3261         STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
3262
3263 #define TREE_UP_FORMAT "040000 tree %s\t.."
3264
3265 static int
3266 tree_compare_entry(enum line_type type1, const char *name1,
3267                    enum line_type type2, const char *name2)
3268 {
3269         if (type1 != type2) {
3270                 if (type1 == LINE_TREE_DIR)
3271                         return -1;
3272                 return 1;
3273         }
3274
3275         return strcmp(name1, name2);
3276 }
3277
3278 static const char *
3279 tree_path(struct line *line)
3280 {
3281         const char *path = line->data;
3282
3283         return path + SIZEOF_TREE_ATTR;
3284 }
3285
3286 static bool
3287 tree_read(struct view *view, char *text)
3288 {
3289         size_t textlen = text ? strlen(text) : 0;
3290         char buf[SIZEOF_STR];
3291         unsigned long pos;
3292         enum line_type type;
3293         bool first_read = view->lines == 0;
3294
3295         if (!text)
3296                 return TRUE;
3297         if (textlen <= SIZEOF_TREE_ATTR)
3298                 return FALSE;
3299
3300         type = text[STRING_SIZE("100644 ")] == 't'
3301              ? LINE_TREE_DIR : LINE_TREE_FILE;
3302
3303         if (first_read) {
3304                 /* Add path info line */
3305                 if (!string_format(buf, "Directory path /%s", opt_path) ||
3306                     !realloc_lines(view, view->line_size + 1) ||
3307                     !add_line_text(view, buf, LINE_DEFAULT))
3308                         return FALSE;
3309
3310                 /* Insert "link" to parent directory. */
3311                 if (*opt_path) {
3312                         if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
3313                             !realloc_lines(view, view->line_size + 1) ||
3314                             !add_line_text(view, buf, LINE_TREE_DIR))
3315                                 return FALSE;
3316                 }
3317         }
3318
3319         /* Strip the path part ... */
3320         if (*opt_path) {
3321                 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3322                 size_t striplen = strlen(opt_path);
3323                 char *path = text + SIZEOF_TREE_ATTR;
3324
3325                 if (pathlen > striplen)
3326                         memmove(path, path + striplen,
3327                                 pathlen - striplen + 1);
3328         }
3329
3330         /* Skip "Directory ..." and ".." line. */
3331         for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
3332                 struct line *line = &view->line[pos];
3333                 const char *path1 = tree_path(line);
3334                 char *path2 = text + SIZEOF_TREE_ATTR;
3335                 int cmp = tree_compare_entry(line->type, path1, type, path2);
3336
3337                 if (cmp <= 0)
3338                         continue;
3339
3340                 text = strdup(text);
3341                 if (!text)
3342                         return FALSE;
3343
3344                 if (view->lines > pos)
3345                         memmove(&view->line[pos + 1], &view->line[pos],
3346                                 (view->lines - pos) * sizeof(*line));
3347
3348                 line = &view->line[pos];
3349                 line->data = text;
3350                 line->type = type;
3351                 view->lines++;
3352                 return TRUE;
3353         }
3354
3355         if (!add_line_text(view, text, type))
3356                 return FALSE;
3357
3358         if (tree_lineno > view->lineno) {
3359                 view->lineno = tree_lineno;
3360                 tree_lineno = 0;
3361         }
3362
3363         return TRUE;
3364 }
3365
3366 static enum request
3367 tree_request(struct view *view, enum request request, struct line *line)
3368 {
3369         enum open_flags flags;
3370
3371         switch (request) {
3372         case REQ_VIEW_BLAME:
3373                 if (line->type != LINE_TREE_FILE) {
3374                         report("Blame only supported for files");
3375                         return REQ_NONE;
3376                 }
3377
3378                 string_copy(opt_ref, view->vid);
3379                 return request;
3380
3381         case REQ_EDIT:
3382                 if (line->type != LINE_TREE_FILE) {
3383                         report("Edit only supported for files");
3384                 } else if (!is_head_commit(view->vid)) {
3385                         report("Edit only supported for files in the current work tree");
3386                 } else {
3387                         open_editor(TRUE, opt_file);
3388                 }
3389                 return REQ_NONE;
3390
3391         case REQ_TREE_PARENT:
3392                 if (!*opt_path) {
3393                         /* quit view if at top of tree */
3394                         return REQ_VIEW_CLOSE;
3395                 }
3396                 /* fake 'cd  ..' */
3397                 line = &view->line[1];
3398                 break;
3399
3400         case REQ_ENTER:
3401                 break;
3402
3403         default:
3404                 return request;
3405         }
3406
3407         /* Cleanup the stack if the tree view is at a different tree. */
3408         while (!*opt_path && tree_stack)
3409                 pop_tree_stack_entry();
3410
3411         switch (line->type) {
3412         case LINE_TREE_DIR:
3413                 /* Depending on whether it is a subdir or parent (updir?) link
3414                  * mangle the path buffer. */
3415                 if (line == &view->line[1] && *opt_path) {
3416                         pop_tree_stack_entry();
3417
3418                 } else {
3419                         const char *basename = tree_path(line);
3420
3421                         push_tree_stack_entry(basename, view->lineno);
3422                 }
3423
3424                 /* Trees and subtrees share the same ID, so they are not not
3425                  * unique like blobs. */
3426                 flags = OPEN_RELOAD;
3427                 request = REQ_VIEW_TREE;
3428                 break;
3429
3430         case LINE_TREE_FILE:
3431                 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3432                 request = REQ_VIEW_BLOB;
3433                 break;
3434
3435         default:
3436                 return TRUE;
3437         }
3438
3439         open_view(view, request, flags);
3440         if (request == REQ_VIEW_TREE) {
3441                 view->lineno = tree_lineno;
3442         }
3443
3444         return REQ_NONE;
3445 }
3446
3447 static void
3448 tree_select(struct view *view, struct line *line)
3449 {
3450         char *text = (char *)line->data + STRING_SIZE("100644 blob ");
3451
3452         if (line->type == LINE_TREE_FILE) {
3453                 string_copy_rev(ref_blob, text);
3454                 string_format(opt_file, "%s%s", opt_path, tree_path(line));
3455
3456         } else if (line->type != LINE_TREE_DIR) {
3457                 return;
3458         }
3459
3460         string_copy_rev(view->ref, text);
3461 }
3462
3463 static struct view_ops tree_ops = {
3464         "file",
3465         NULL,
3466         tree_read,
3467         pager_draw,
3468         tree_request,
3469         pager_grep,
3470         tree_select,
3471 };
3472
3473 static bool
3474 blob_read(struct view *view, char *line)
3475 {
3476         if (!line)
3477                 return TRUE;
3478         return add_line_text(view, line, LINE_DEFAULT) != NULL;
3479 }
3480
3481 static struct view_ops blob_ops = {
3482         "line",
3483         NULL,
3484         blob_read,
3485         pager_draw,
3486         pager_request,
3487         pager_grep,
3488         pager_select,
3489 };
3490
3491 /*
3492  * Blame backend
3493  *
3494  * Loading the blame view is a two phase job:
3495  *
3496  *  1. File content is read either using opt_file from the
3497  *     filesystem or using git-cat-file.
3498  *  2. Then blame information is incrementally added by
3499  *     reading output from git-blame.
3500  */
3501
3502 struct blame_commit {
3503         char id[SIZEOF_REV];            /* SHA1 ID. */
3504         char title[128];                /* First line of the commit message. */
3505         char author[75];                /* Author of the commit. */
3506         struct tm time;                 /* Date from the author ident. */
3507         char filename[128];             /* Name of file. */
3508 };
3509
3510 struct blame {
3511         struct blame_commit *commit;
3512         unsigned int header:1;
3513         char text[1];
3514 };
3515
3516 #define BLAME_CAT_FILE_CMD "git cat-file blob %s:%s"
3517 #define BLAME_INCREMENTAL_CMD "git blame --incremental %s -- %s"
3518
3519 static bool
3520 blame_open(struct view *view)
3521 {
3522         char path[SIZEOF_STR];
3523         char ref[SIZEOF_STR] = "";
3524
3525         if (sq_quote(path, 0, opt_file) >= sizeof(path))
3526                 return FALSE;
3527
3528         if (*opt_ref && sq_quote(ref, 0, opt_ref) >= sizeof(ref))
3529                 return FALSE;
3530
3531         if (*opt_ref || !(view->pipe = fopen(opt_file, "r"))) {
3532                 const char *id = *opt_ref ? ref : "HEAD";
3533
3534                 if (!string_format(view->cmd, BLAME_CAT_FILE_CMD, id, path) ||
3535                     !(view->pipe = popen(view->cmd, "r")))
3536                         return FALSE;
3537         }
3538
3539         reset_view(view);
3540         string_format(view->ref, "%s ...", opt_file);
3541         string_copy_rev(view->vid, opt_file);
3542         set_nonblocking_input(TRUE);
3543         view->start_time = time(NULL);
3544
3545         return TRUE;
3546 }
3547
3548 static struct blame_commit *
3549 get_blame_commit(struct view *view, const char *id)
3550 {
3551         size_t i;
3552
3553         for (i = 0; i < view->lines; i++) {
3554                 struct blame *blame = view->line[i].data;
3555
3556                 if (!blame->commit)
3557                         continue;
3558
3559                 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
3560                         return blame->commit;
3561         }
3562
3563         {
3564                 struct blame_commit *commit = calloc(1, sizeof(*commit));
3565
3566                 if (commit)
3567                         string_ncopy(commit->id, id, SIZEOF_REV);
3568                 return commit;
3569         }
3570 }
3571
3572 static bool
3573 parse_number(const char **posref, size_t *number, size_t min, size_t max)
3574 {
3575         const char *pos = *posref;
3576
3577         *posref = NULL;
3578         pos = strchr(pos + 1, ' ');
3579         if (!pos || !isdigit(pos[1]))
3580                 return FALSE;
3581         *number = atoi(pos + 1);
3582         if (*number < min || *number > max)
3583                 return FALSE;
3584
3585         *posref = pos;
3586         return TRUE;
3587 }
3588
3589 static struct blame_commit *
3590 parse_blame_commit(struct view *view, const char *text, int *blamed)
3591 {
3592         struct blame_commit *commit;
3593         struct blame *blame;
3594         const char *pos = text + SIZEOF_REV - 1;
3595         size_t lineno;
3596         size_t group;
3597
3598         if (strlen(text) <= SIZEOF_REV || *pos != ' ')
3599                 return NULL;
3600
3601         if (!parse_number(&pos, &lineno, 1, view->lines) ||
3602             !parse_number(&pos, &group, 1, view->lines - lineno + 1))
3603                 return NULL;
3604
3605         commit = get_blame_commit(view, text);
3606         if (!commit)
3607                 return NULL;
3608
3609         *blamed += group;
3610         while (group--) {
3611                 struct line *line = &view->line[lineno + group - 1];
3612
3613                 blame = line->data;
3614                 blame->commit = commit;
3615                 blame->header = !group;
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_ENTER:
3768                 if (!blame->commit) {
3769                         report("No commit loaded yet");
3770                         break;
3771                 }
3772
3773                 if (!strcmp(blame->commit->id, NULL_ID)) {
3774                         char path[SIZEOF_STR];
3775
3776                         if (sq_quote(path, 0, view->vid) >= sizeof(path))
3777                                 break;
3778                         string_format(opt_cmd, "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s 2>/dev/null", path);
3779                 }
3780
3781                 open_view(view, REQ_VIEW_DIFF, flags);
3782                 break;
3783
3784         default:
3785                 return request;
3786         }
3787
3788         return REQ_NONE;
3789 }
3790
3791 static bool
3792 blame_grep(struct view *view, struct line *line)
3793 {
3794         struct blame *blame = line->data;
3795         struct blame_commit *commit = blame->commit;
3796         regmatch_t pmatch;
3797
3798 #define MATCH(text, on)                                                 \
3799         (on && *text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3800
3801         if (commit) {
3802                 char buf[DATE_COLS + 1];
3803
3804                 if (MATCH(commit->title, 1) ||
3805                     MATCH(commit->author, opt_author) ||
3806                     MATCH(commit->id, opt_date))
3807                         return TRUE;
3808
3809                 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
3810                     MATCH(buf, 1))
3811                         return TRUE;
3812         }
3813
3814         return MATCH(blame->text, 1);
3815
3816 #undef MATCH
3817 }
3818
3819 static void
3820 blame_select(struct view *view, struct line *line)
3821 {
3822         struct blame *blame = line->data;
3823         struct blame_commit *commit = blame->commit;
3824
3825         if (!commit)
3826                 return;
3827
3828         if (!strcmp(commit->id, NULL_ID))
3829                 string_ncopy(ref_commit, "HEAD", 4);
3830         else
3831                 string_copy_rev(ref_commit, commit->id);
3832 }
3833
3834 static struct view_ops blame_ops = {
3835         "line",
3836         blame_open,
3837         blame_read,
3838         blame_draw,
3839         blame_request,
3840         blame_grep,
3841         blame_select,
3842 };
3843
3844 /*
3845  * Status backend
3846  */
3847
3848 struct status {
3849         char status;
3850         struct {
3851                 mode_t mode;
3852                 char rev[SIZEOF_REV];
3853                 char name[SIZEOF_STR];
3854         } old;
3855         struct {
3856                 mode_t mode;
3857                 char rev[SIZEOF_REV];
3858                 char name[SIZEOF_STR];
3859         } new;
3860 };
3861
3862 static char status_onbranch[SIZEOF_STR];
3863 static struct status stage_status;
3864 static enum line_type stage_line_type;
3865 static size_t stage_chunks;
3866 static int *stage_chunk;
3867
3868 /* This should work even for the "On branch" line. */
3869 static inline bool
3870 status_has_none(struct view *view, struct line *line)
3871 {
3872         return line < view->line + view->lines && !line[1].data;
3873 }
3874
3875 /* Get fields from the diff line:
3876  * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
3877  */
3878 static inline bool
3879 status_get_diff(struct status *file, const char *buf, size_t bufsize)
3880 {
3881         const char *old_mode = buf +  1;
3882         const char *new_mode = buf +  8;
3883         const char *old_rev  = buf + 15;
3884         const char *new_rev  = buf + 56;
3885         const char *status   = buf + 97;
3886
3887         if (bufsize < 99 ||
3888             old_mode[-1] != ':' ||
3889             new_mode[-1] != ' ' ||
3890             old_rev[-1]  != ' ' ||
3891             new_rev[-1]  != ' ' ||
3892             status[-1]   != ' ')
3893                 return FALSE;
3894
3895         file->status = *status;
3896
3897         string_copy_rev(file->old.rev, old_rev);
3898         string_copy_rev(file->new.rev, new_rev);
3899
3900         file->old.mode = strtoul(old_mode, NULL, 8);
3901         file->new.mode = strtoul(new_mode, NULL, 8);
3902
3903         file->old.name[0] = file->new.name[0] = 0;
3904
3905         return TRUE;
3906 }
3907
3908 static bool
3909 status_run(struct view *view, const char cmd[], char status, enum line_type type)
3910 {
3911         struct status *file = NULL;
3912         struct status *unmerged = NULL;
3913         char buf[SIZEOF_STR * 4];
3914         size_t bufsize = 0;
3915         FILE *pipe;
3916
3917         pipe = popen(cmd, "r");
3918         if (!pipe)
3919                 return FALSE;
3920
3921         add_line_data(view, NULL, type);
3922
3923         while (!feof(pipe) && !ferror(pipe)) {
3924                 char *sep;
3925                 size_t readsize;
3926
3927                 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
3928                 if (!readsize)
3929                         break;
3930                 bufsize += readsize;
3931
3932                 /* Process while we have NUL chars. */
3933                 while ((sep = memchr(buf, 0, bufsize))) {
3934                         size_t sepsize = sep - buf + 1;
3935
3936                         if (!file) {
3937                                 if (!realloc_lines(view, view->line_size + 1))
3938                                         goto error_out;
3939
3940                                 file = calloc(1, sizeof(*file));
3941                                 if (!file)
3942                                         goto error_out;
3943
3944                                 add_line_data(view, file, type);
3945                         }
3946
3947                         /* Parse diff info part. */
3948                         if (status) {
3949                                 file->status = status;
3950                                 if (status == 'A')
3951                                         string_copy(file->old.rev, NULL_ID);
3952
3953                         } else if (!file->status) {
3954                                 if (!status_get_diff(file, buf, sepsize))
3955                                         goto error_out;
3956
3957                                 bufsize -= sepsize;
3958                                 memmove(buf, sep + 1, bufsize);
3959
3960                                 sep = memchr(buf, 0, bufsize);
3961                                 if (!sep)
3962                                         break;
3963                                 sepsize = sep - buf + 1;
3964
3965                                 /* Collapse all 'M'odified entries that
3966                                  * follow a associated 'U'nmerged entry.
3967                                  */
3968                                 if (file->status == 'U') {
3969                                         unmerged = file;
3970
3971                                 } else if (unmerged) {
3972                                         int collapse = !strcmp(buf, unmerged->new.name);
3973
3974                                         unmerged = NULL;
3975                                         if (collapse) {
3976                                                 free(file);
3977                                                 view->lines--;
3978                                                 continue;
3979                                         }
3980                                 }
3981                         }
3982
3983                         /* Grab the old name for rename/copy. */
3984                         if (!*file->old.name &&
3985                             (file->status == 'R' || file->status == 'C')) {
3986                                 sepsize = sep - buf + 1;
3987                                 string_ncopy(file->old.name, buf, sepsize);
3988                                 bufsize -= sepsize;
3989                                 memmove(buf, sep + 1, bufsize);
3990
3991                                 sep = memchr(buf, 0, bufsize);
3992                                 if (!sep)
3993                                         break;
3994                                 sepsize = sep - buf + 1;
3995                         }
3996
3997                         /* git-ls-files just delivers a NUL separated
3998                          * list of file names similar to the second half
3999                          * of the git-diff-* output. */
4000                         string_ncopy(file->new.name, buf, sepsize);
4001                         if (!*file->old.name)
4002                                 string_copy(file->old.name, file->new.name);
4003                         bufsize -= sepsize;
4004                         memmove(buf, sep + 1, bufsize);
4005                         file = NULL;
4006                 }
4007         }
4008
4009         if (ferror(pipe)) {
4010 error_out:
4011                 pclose(pipe);
4012                 return FALSE;
4013         }
4014
4015         if (!view->line[view->lines - 1].data)
4016                 add_line_data(view, NULL, LINE_STAT_NONE);
4017
4018         pclose(pipe);
4019         return TRUE;
4020 }
4021
4022 /* Don't show unmerged entries in the staged section. */
4023 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --diff-filter=ACDMRTXB --cached -M HEAD"
4024 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
4025 #define STATUS_LIST_OTHER_CMD \
4026         "git ls-files -z --others --exclude-standard"
4027 #define STATUS_LIST_NO_HEAD_CMD \
4028         "git ls-files -z --cached --exclude-standard"
4029
4030 #define STATUS_DIFF_INDEX_SHOW_CMD \
4031         "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s %s 2>/dev/null"
4032
4033 #define STATUS_DIFF_FILES_SHOW_CMD \
4034         "git diff-files --root --patch-with-stat -C -M -- %s %s 2>/dev/null"
4035
4036 #define STATUS_DIFF_NO_HEAD_SHOW_CMD \
4037         "git diff --no-color --patch-with-stat /dev/null %s 2>/dev/null"
4038
4039 /* First parse staged info using git-diff-index(1), then parse unstaged
4040  * info using git-diff-files(1), and finally untracked files using
4041  * git-ls-files(1). */
4042 static bool
4043 status_open(struct view *view)
4044 {
4045         unsigned long prev_lineno = view->lineno;
4046
4047         reset_view(view);
4048
4049         if (!realloc_lines(view, view->line_size + 7))
4050                 return FALSE;
4051
4052         add_line_data(view, NULL, LINE_STAT_HEAD);
4053         if (is_initial_commit())
4054                 string_copy(status_onbranch, "Initial commit");
4055         else if (!*opt_head)
4056                 string_copy(status_onbranch, "Not currently on any branch");
4057         else if (!string_format(status_onbranch, "On branch %s", opt_head))
4058                 return FALSE;
4059
4060         system("git update-index -q --refresh >/dev/null 2>/dev/null");
4061
4062         if (is_initial_commit()) {
4063                 if (!status_run(view, STATUS_LIST_NO_HEAD_CMD, 'A', LINE_STAT_STAGED))
4064                         return FALSE;
4065         } else if (!status_run(view, STATUS_DIFF_INDEX_CMD, 0, LINE_STAT_STAGED)) {
4066                 return FALSE;
4067         }
4068
4069         if (!status_run(view, STATUS_DIFF_FILES_CMD, 0, LINE_STAT_UNSTAGED) ||
4070             !status_run(view, STATUS_LIST_OTHER_CMD, '?', LINE_STAT_UNTRACKED))
4071                 return FALSE;
4072
4073         /* If all went well restore the previous line number to stay in
4074          * the context or select a line with something that can be
4075          * updated. */
4076         if (prev_lineno >= view->lines)
4077                 prev_lineno = view->lines - 1;
4078         while (prev_lineno < view->lines && !view->line[prev_lineno].data)
4079                 prev_lineno++;
4080         while (prev_lineno > 0 && !view->line[prev_lineno].data)
4081                 prev_lineno--;
4082
4083         /* If the above fails, always skip the "On branch" line. */
4084         if (prev_lineno < view->lines)
4085                 view->lineno = prev_lineno;
4086         else
4087                 view->lineno = 1;
4088
4089         if (view->lineno < view->offset)
4090                 view->offset = view->lineno;
4091         else if (view->offset + view->height <= view->lineno)
4092                 view->offset = view->lineno - view->height + 1;
4093
4094         return TRUE;
4095 }
4096
4097 static bool
4098 status_draw(struct view *view, struct line *line, unsigned int lineno)
4099 {
4100         struct status *status = line->data;
4101         enum line_type type;
4102         const char *text;
4103
4104         if (!status) {
4105                 switch (line->type) {
4106                 case LINE_STAT_STAGED:
4107                         type = LINE_STAT_SECTION;
4108                         text = "Changes to be committed:";
4109                         break;
4110
4111                 case LINE_STAT_UNSTAGED:
4112                         type = LINE_STAT_SECTION;
4113                         text = "Changed but not updated:";
4114                         break;
4115
4116                 case LINE_STAT_UNTRACKED:
4117                         type = LINE_STAT_SECTION;
4118                         text = "Untracked files:";
4119                         break;
4120
4121                 case LINE_STAT_NONE:
4122                         type = LINE_DEFAULT;
4123                         text = "    (no files)";
4124                         break;
4125
4126                 case LINE_STAT_HEAD:
4127                         type = LINE_STAT_HEAD;
4128                         text = status_onbranch;
4129                         break;
4130
4131                 default:
4132                         return FALSE;
4133                 }
4134         } else {
4135                 static char buf[] = { '?', ' ', ' ', ' ', 0 };
4136
4137                 buf[0] = status->status;
4138                 if (draw_text(view, line->type, buf, TRUE))
4139                         return TRUE;
4140                 type = LINE_DEFAULT;
4141                 text = status->new.name;
4142         }
4143
4144         draw_text(view, type, text, TRUE);
4145         return TRUE;
4146 }
4147
4148 static enum request
4149 status_enter(struct view *view, struct line *line)
4150 {
4151         struct status *status = line->data;
4152         char oldpath[SIZEOF_STR] = "";
4153         char newpath[SIZEOF_STR] = "";
4154         const char *info;
4155         size_t cmdsize = 0;
4156         enum open_flags split;
4157
4158         if (line->type == LINE_STAT_NONE ||
4159             (!status && line[1].type == LINE_STAT_NONE)) {
4160                 report("No file to diff");
4161                 return REQ_NONE;
4162         }
4163
4164         if (status) {
4165                 if (sq_quote(oldpath, 0, status->old.name) >= sizeof(oldpath))
4166                         return REQ_QUIT;
4167                 /* Diffs for unmerged entries are empty when pasing the
4168                  * new path, so leave it empty. */
4169                 if (status->status != 'U' &&
4170                     sq_quote(newpath, 0, status->new.name) >= sizeof(newpath))
4171                         return REQ_QUIT;
4172         }
4173
4174         if (opt_cdup[0] &&
4175             line->type != LINE_STAT_UNTRACKED &&
4176             !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
4177                 return REQ_QUIT;
4178
4179         switch (line->type) {
4180         case LINE_STAT_STAGED:
4181                 if (is_initial_commit()) {
4182                         if (!string_format_from(opt_cmd, &cmdsize,
4183                                                 STATUS_DIFF_NO_HEAD_SHOW_CMD,
4184                                                 newpath))
4185                                 return REQ_QUIT;
4186                 } else {
4187                         if (!string_format_from(opt_cmd, &cmdsize,
4188                                                 STATUS_DIFF_INDEX_SHOW_CMD,
4189                                                 oldpath, newpath))
4190                                 return REQ_QUIT;
4191                 }
4192
4193                 if (status)
4194                         info = "Staged changes to %s";
4195                 else
4196                         info = "Staged changes";
4197                 break;
4198
4199         case LINE_STAT_UNSTAGED:
4200                 if (!string_format_from(opt_cmd, &cmdsize,
4201                                         STATUS_DIFF_FILES_SHOW_CMD, oldpath, newpath))
4202                         return REQ_QUIT;
4203                 if (status)
4204                         info = "Unstaged changes to %s";
4205                 else
4206                         info = "Unstaged changes";
4207                 break;
4208
4209         case LINE_STAT_UNTRACKED:
4210                 if (opt_pipe)
4211                         return REQ_QUIT;
4212
4213                 if (!status) {
4214                         report("No file to show");
4215                         return REQ_NONE;
4216                 }
4217
4218                 if (!suffixcmp(status->new.name, -1, "/")) {
4219                         report("Cannot display a directory");
4220                         return REQ_NONE;
4221                 }
4222
4223                 opt_pipe = fopen(status->new.name, "r");
4224                 info = "Untracked file %s";
4225                 break;
4226
4227         case LINE_STAT_HEAD:
4228                 return REQ_NONE;
4229
4230         default:
4231                 die("line type %d not handled in switch", line->type);
4232         }
4233
4234         split = view_is_displayed(view) ? OPEN_SPLIT : 0;
4235         open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | split);
4236         if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
4237                 if (status) {
4238                         stage_status = *status;
4239                 } else {
4240                         memset(&stage_status, 0, sizeof(stage_status));
4241                 }
4242
4243                 stage_line_type = line->type;
4244                 stage_chunks = 0;
4245                 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
4246         }
4247
4248         return REQ_NONE;
4249 }
4250
4251 static bool
4252 status_exists(struct status *status, enum line_type type)
4253 {
4254         struct view *view = VIEW(REQ_VIEW_STATUS);
4255         struct line *line;
4256
4257         for (line = view->line; line < view->line + view->lines; line++) {
4258                 struct status *pos = line->data;
4259
4260                 if (line->type == type && pos &&
4261                     !strcmp(status->new.name, pos->new.name))
4262                         return TRUE;
4263         }
4264
4265         return FALSE;
4266 }
4267
4268
4269 static FILE *
4270 status_update_prepare(enum line_type type)
4271 {
4272         char cmd[SIZEOF_STR];
4273         size_t cmdsize = 0;
4274
4275         if (opt_cdup[0] &&
4276             type != LINE_STAT_UNTRACKED &&
4277             !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4278                 return NULL;
4279
4280         switch (type) {
4281         case LINE_STAT_STAGED:
4282                 string_add(cmd, cmdsize, "git update-index -z --index-info");
4283                 break;
4284
4285         case LINE_STAT_UNSTAGED:
4286         case LINE_STAT_UNTRACKED:
4287                 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
4288                 break;
4289
4290         default:
4291                 die("line type %d not handled in switch", type);
4292         }
4293
4294         return popen(cmd, "w");
4295 }
4296
4297 static bool
4298 status_update_write(FILE *pipe, struct status *status, enum line_type type)
4299 {
4300         char buf[SIZEOF_STR];
4301         size_t bufsize = 0;
4302         size_t written = 0;
4303
4304         switch (type) {
4305         case LINE_STAT_STAGED:
4306                 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
4307                                         status->old.mode,
4308                                         status->old.rev,
4309                                         status->old.name, 0))
4310                         return FALSE;
4311                 break;
4312
4313         case LINE_STAT_UNSTAGED:
4314         case LINE_STAT_UNTRACKED:
4315                 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
4316                         return FALSE;
4317                 break;
4318
4319         default:
4320                 die("line type %d not handled in switch", type);
4321         }
4322
4323         while (!ferror(pipe) && written < bufsize) {
4324                 written += fwrite(buf + written, 1, bufsize - written, pipe);
4325         }
4326
4327         return written == bufsize;
4328 }
4329
4330 static bool
4331 status_update_file(struct status *status, enum line_type type)
4332 {
4333         FILE *pipe = status_update_prepare(type);
4334         bool result;
4335
4336         if (!pipe)
4337                 return FALSE;
4338
4339         result = status_update_write(pipe, status, type);
4340         pclose(pipe);
4341         return result;
4342 }
4343
4344 static bool
4345 status_update_files(struct view *view, struct line *line)
4346 {
4347         FILE *pipe = status_update_prepare(line->type);
4348         bool result = TRUE;
4349         struct line *pos = view->line + view->lines;
4350         int files = 0;
4351         int file, done;
4352
4353         if (!pipe)
4354                 return FALSE;
4355
4356         for (pos = line; pos < view->line + view->lines && pos->data; pos++)
4357                 files++;
4358
4359         for (file = 0, done = 0; result && file < files; line++, file++) {
4360                 int almost_done = file * 100 / files;
4361
4362                 if (almost_done > done) {
4363                         done = almost_done;
4364                         string_format(view->ref, "updating file %u of %u (%d%% done)",
4365                                       file, files, done);
4366                         update_view_title(view);
4367                 }
4368                 result = status_update_write(pipe, line->data, line->type);
4369         }
4370
4371         pclose(pipe);
4372         return result;
4373 }
4374
4375 static bool
4376 status_update(struct view *view)
4377 {
4378         struct line *line = &view->line[view->lineno];
4379
4380         assert(view->lines);
4381
4382         if (!line->data) {
4383                 /* This should work even for the "On branch" line. */
4384                 if (line < view->line + view->lines && !line[1].data) {
4385                         report("Nothing to update");
4386                         return FALSE;
4387                 }
4388
4389                 if (!status_update_files(view, line + 1)) {
4390                         report("Failed to update file status");
4391                         return FALSE;
4392                 }
4393
4394         } else if (!status_update_file(line->data, line->type)) {
4395                 report("Failed to update file status");
4396                 return FALSE;
4397         }
4398
4399         return TRUE;
4400 }
4401
4402 static bool
4403 status_revert(struct status *status, enum line_type type, bool has_none)
4404 {
4405         if (!status || type != LINE_STAT_UNSTAGED) {
4406                 if (type == LINE_STAT_STAGED) {
4407                         report("Cannot revert changes to staged files");
4408                 } else if (type == LINE_STAT_UNTRACKED) {
4409                         report("Cannot revert changes to untracked files");
4410                 } else if (has_none) {
4411                         report("Nothing to revert");
4412                 } else {
4413                         report("Cannot revert changes to multiple files");
4414                 }
4415                 return FALSE;
4416
4417         } else {
4418                 char cmd[SIZEOF_STR];
4419                 char file_sq[SIZEOF_STR];
4420
4421                 if (sq_quote(file_sq, 0, status->old.name) >= sizeof(file_sq) ||
4422                     !string_format(cmd, "git checkout -- %s%s", opt_cdup, file_sq))
4423                         return FALSE;
4424
4425                 return run_confirm(cmd, "Are you sure you want to overwrite any changes?");
4426         }
4427 }
4428
4429 static enum request
4430 status_request(struct view *view, enum request request, struct line *line)
4431 {
4432         struct status *status = line->data;
4433
4434         switch (request) {
4435         case REQ_STATUS_UPDATE:
4436                 if (!status_update(view))
4437                         return REQ_NONE;
4438                 break;
4439
4440         case REQ_STATUS_REVERT:
4441                 if (!status_revert(status, line->type, status_has_none(view, line)))
4442                         return REQ_NONE;
4443                 break;
4444
4445         case REQ_STATUS_MERGE:
4446                 if (!status || status->status != 'U') {
4447                         report("Merging only possible for files with unmerged status ('U').");
4448                         return REQ_NONE;
4449                 }
4450                 open_mergetool(status->new.name);
4451                 break;
4452
4453         case REQ_EDIT:
4454                 if (!status)
4455                         return request;
4456                 if (status->status == 'D') {
4457                         report("File has been deleted.");
4458                         return REQ_NONE;
4459                 }
4460
4461                 open_editor(status->status != '?', status->new.name);
4462                 break;
4463
4464         case REQ_VIEW_BLAME:
4465                 if (status) {
4466                         string_copy(opt_file, status->new.name);
4467                         opt_ref[0] = 0;
4468                 }
4469                 return request;
4470
4471         case REQ_ENTER:
4472                 /* After returning the status view has been split to
4473                  * show the stage view. No further reloading is
4474                  * necessary. */
4475                 status_enter(view, line);
4476                 return REQ_NONE;
4477
4478         case REQ_REFRESH:
4479                 /* Simply reload the view. */
4480                 break;
4481
4482         default:
4483                 return request;
4484         }
4485
4486         open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
4487
4488         return REQ_NONE;
4489 }
4490
4491 static void
4492 status_select(struct view *view, struct line *line)
4493 {
4494         struct status *status = line->data;
4495         char file[SIZEOF_STR] = "all files";
4496         const char *text;
4497         const char *key;
4498
4499         if (status && !string_format(file, "'%s'", status->new.name))
4500                 return;
4501
4502         if (!status && line[1].type == LINE_STAT_NONE)
4503                 line++;
4504
4505         switch (line->type) {
4506         case LINE_STAT_STAGED:
4507                 text = "Press %s to unstage %s for commit";
4508                 break;
4509
4510         case LINE_STAT_UNSTAGED:
4511                 text = "Press %s to stage %s for commit";
4512                 break;
4513
4514         case LINE_STAT_UNTRACKED:
4515                 text = "Press %s to stage %s for addition";
4516                 break;
4517
4518         case LINE_STAT_HEAD:
4519         case LINE_STAT_NONE:
4520                 text = "Nothing to update";
4521                 break;
4522
4523         default:
4524                 die("line type %d not handled in switch", line->type);
4525         }
4526
4527         if (status && status->status == 'U') {
4528                 text = "Press %s to resolve conflict in %s";
4529                 key = get_key(REQ_STATUS_MERGE);
4530
4531         } else {
4532                 key = get_key(REQ_STATUS_UPDATE);
4533         }
4534
4535         string_format(view->ref, text, key, file);
4536 }
4537
4538 static bool
4539 status_grep(struct view *view, struct line *line)
4540 {
4541         struct status *status = line->data;
4542         enum { S_STATUS, S_NAME, S_END } state;
4543         char buf[2] = "?";
4544         regmatch_t pmatch;
4545
4546         if (!status)
4547                 return FALSE;
4548
4549         for (state = S_STATUS; state < S_END; state++) {
4550                 const char *text;
4551
4552                 switch (state) {
4553                 case S_NAME:    text = status->new.name;        break;
4554                 case S_STATUS:
4555                         buf[0] = status->status;
4556                         text = buf;
4557                         break;
4558
4559                 default:
4560                         return FALSE;
4561                 }
4562
4563                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4564                         return TRUE;
4565         }
4566
4567         return FALSE;
4568 }
4569
4570 static struct view_ops status_ops = {
4571         "file",
4572         status_open,
4573         NULL,
4574         status_draw,
4575         status_request,
4576         status_grep,
4577         status_select,
4578 };
4579
4580
4581 static bool
4582 stage_diff_line(FILE *pipe, struct line *line)
4583 {
4584         const char *buf = line->data;
4585         size_t bufsize = strlen(buf);
4586         size_t written = 0;
4587
4588         while (!ferror(pipe) && written < bufsize) {
4589                 written += fwrite(buf + written, 1, bufsize - written, pipe);
4590         }
4591
4592         fputc('\n', pipe);
4593
4594         return written == bufsize;
4595 }
4596
4597 static bool
4598 stage_diff_write(FILE *pipe, struct line *line, struct line *end)
4599 {
4600         while (line < end) {
4601                 if (!stage_diff_line(pipe, line++))
4602                         return FALSE;
4603                 if (line->type == LINE_DIFF_CHUNK ||
4604                     line->type == LINE_DIFF_HEADER)
4605                         break;
4606         }
4607
4608         return TRUE;
4609 }
4610
4611 static struct line *
4612 stage_diff_find(struct view *view, struct line *line, enum line_type type)
4613 {
4614         for (; view->line < line; line--)
4615                 if (line->type == type)
4616                         return line;
4617
4618         return NULL;
4619 }
4620
4621 static bool
4622 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
4623 {
4624         char cmd[SIZEOF_STR];
4625         size_t cmdsize = 0;
4626         struct line *diff_hdr;
4627         FILE *pipe;
4628
4629         diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
4630         if (!diff_hdr)
4631                 return FALSE;
4632
4633         if (opt_cdup[0] &&
4634             !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4635                 return FALSE;
4636
4637         if (!string_format_from(cmd, &cmdsize,
4638                                 "git apply --whitespace=nowarn %s %s - && "
4639                                 "git update-index -q --unmerged --refresh 2>/dev/null",
4640                                 revert ? "" : "--cached",
4641                                 revert || stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
4642                 return FALSE;
4643
4644         pipe = popen(cmd, "w");
4645         if (!pipe)
4646                 return FALSE;
4647
4648         if (!stage_diff_write(pipe, diff_hdr, chunk) ||
4649             !stage_diff_write(pipe, chunk, view->line + view->lines))
4650                 chunk = NULL;
4651
4652         pclose(pipe);
4653
4654         return chunk ? TRUE : FALSE;
4655 }
4656
4657 static bool
4658 stage_update(struct view *view, struct line *line)
4659 {
4660         struct line *chunk = NULL;
4661
4662         if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
4663                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4664
4665         if (chunk) {
4666                 if (!stage_apply_chunk(view, chunk, FALSE)) {
4667                         report("Failed to apply chunk");
4668                         return FALSE;
4669                 }
4670
4671         } else if (!stage_status.status) {
4672                 view = VIEW(REQ_VIEW_STATUS);
4673
4674                 for (line = view->line; line < view->line + view->lines; line++)
4675                         if (line->type == stage_line_type)
4676                                 break;
4677
4678                 if (!status_update_files(view, line + 1)) {
4679                         report("Failed to update files");
4680                         return FALSE;
4681                 }
4682
4683         } else if (!status_update_file(&stage_status, stage_line_type)) {
4684                 report("Failed to update file");
4685                 return FALSE;
4686         }
4687
4688         return TRUE;
4689 }
4690
4691 static bool
4692 stage_revert(struct view *view, struct line *line)
4693 {
4694         struct line *chunk = NULL;
4695
4696         if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
4697                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4698
4699         if (chunk) {
4700                 if (!prompt_yesno("Are you sure you want to revert changes?"))
4701                         return FALSE;
4702
4703                 if (!stage_apply_chunk(view, chunk, TRUE)) {
4704                         report("Failed to revert chunk");
4705                         return FALSE;
4706                 }
4707                 return TRUE;
4708
4709         } else {
4710                 return status_revert(stage_status.status ? &stage_status : NULL,
4711                                      stage_line_type, FALSE);
4712         }
4713 }
4714
4715
4716 static void
4717 stage_next(struct view *view, struct line *line)
4718 {
4719         int i;
4720
4721         if (!stage_chunks) {
4722                 static size_t alloc = 0;
4723                 int *tmp;
4724
4725                 for (line = view->line; line < view->line + view->lines; line++) {
4726                         if (line->type != LINE_DIFF_CHUNK)
4727                                 continue;
4728
4729                         tmp = realloc_items(stage_chunk, &alloc,
4730                                             stage_chunks, sizeof(*tmp));
4731                         if (!tmp) {
4732                                 report("Allocation failure");
4733                                 return;
4734                         }
4735
4736                         stage_chunk = tmp;
4737                         stage_chunk[stage_chunks++] = line - view->line;
4738                 }
4739         }
4740
4741         for (i = 0; i < stage_chunks; i++) {
4742                 if (stage_chunk[i] > view->lineno) {
4743                         do_scroll_view(view, stage_chunk[i] - view->lineno);
4744                         report("Chunk %d of %d", i + 1, stage_chunks);
4745                         return;
4746                 }
4747         }
4748
4749         report("No next chunk found");
4750 }
4751
4752 static enum request
4753 stage_request(struct view *view, enum request request, struct line *line)
4754 {
4755         switch (request) {
4756         case REQ_STATUS_UPDATE:
4757                 if (!stage_update(view, line))
4758                         return REQ_NONE;
4759                 break;
4760
4761         case REQ_STATUS_REVERT:
4762                 if (!stage_revert(view, line))
4763                         return REQ_NONE;
4764                 break;
4765
4766         case REQ_STAGE_NEXT:
4767                 if (stage_line_type == LINE_STAT_UNTRACKED) {
4768                         report("File is untracked; press %s to add",
4769                                get_key(REQ_STATUS_UPDATE));
4770                         return REQ_NONE;
4771                 }
4772                 stage_next(view, line);
4773                 return REQ_NONE;
4774
4775         case REQ_EDIT:
4776                 if (!stage_status.new.name[0])
4777                         return request;
4778                 if (stage_status.status == 'D') {
4779                         report("File has been deleted.");
4780                         return REQ_NONE;
4781                 }
4782
4783                 open_editor(stage_status.status != '?', stage_status.new.name);
4784                 break;
4785
4786         case REQ_REFRESH:
4787                 /* Reload everything ... */
4788                 break;
4789
4790         case REQ_VIEW_BLAME:
4791                 if (stage_status.new.name[0]) {
4792                         string_copy(opt_file, stage_status.new.name);
4793                         opt_ref[0] = 0;
4794                 }
4795                 return request;
4796
4797         case REQ_ENTER:
4798                 return pager_request(view, request, line);
4799
4800         default:
4801                 return request;
4802         }
4803
4804         open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD | OPEN_NOMAXIMIZE);
4805
4806         /* Check whether the staged entry still exists, and close the
4807          * stage view if it doesn't. */
4808         if (!status_exists(&stage_status, stage_line_type))
4809                 return REQ_VIEW_CLOSE;
4810
4811         if (stage_line_type == LINE_STAT_UNTRACKED) {
4812                 if (!suffixcmp(stage_status.new.name, -1, "/")) {
4813                         report("Cannot display a directory");
4814                         return REQ_NONE;
4815                 }
4816
4817                 opt_pipe = fopen(stage_status.new.name, "r");
4818         }
4819         open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
4820
4821         return REQ_NONE;
4822 }
4823
4824 static struct view_ops stage_ops = {
4825         "line",
4826         NULL,
4827         pager_read,
4828         pager_draw,
4829         stage_request,
4830         pager_grep,
4831         pager_select,
4832 };
4833
4834
4835 /*
4836  * Revision graph
4837  */
4838
4839 struct commit {
4840         char id[SIZEOF_REV];            /* SHA1 ID. */
4841         char title[128];                /* First line of the commit message. */
4842         char author[75];                /* Author of the commit. */
4843         struct tm time;                 /* Date from the author ident. */
4844         struct ref **refs;              /* Repository references. */
4845         chtype graph[SIZEOF_REVGRAPH];  /* Ancestry chain graphics. */
4846         size_t graph_size;              /* The width of the graph array. */
4847         bool has_parents;               /* Rewritten --parents seen. */
4848 };
4849
4850 /* Size of rev graph with no  "padding" columns */
4851 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
4852
4853 struct rev_graph {
4854         struct rev_graph *prev, *next, *parents;
4855         char rev[SIZEOF_REVITEMS][SIZEOF_REV];
4856         size_t size;
4857         struct commit *commit;
4858         size_t pos;
4859         unsigned int boundary:1;
4860 };
4861
4862 /* Parents of the commit being visualized. */
4863 static struct rev_graph graph_parents[4];
4864
4865 /* The current stack of revisions on the graph. */
4866 static struct rev_graph graph_stacks[4] = {
4867         { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
4868         { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
4869         { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
4870         { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
4871 };
4872
4873 static inline bool
4874 graph_parent_is_merge(struct rev_graph *graph)
4875 {
4876         return graph->parents->size > 1;
4877 }
4878
4879 static inline void
4880 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
4881 {
4882         struct commit *commit = graph->commit;
4883
4884         if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
4885                 commit->graph[commit->graph_size++] = symbol;
4886 }
4887
4888 static void
4889 clear_rev_graph(struct rev_graph *graph)
4890 {
4891         graph->boundary = 0;
4892         graph->size = graph->pos = 0;
4893         graph->commit = NULL;
4894         memset(graph->parents, 0, sizeof(*graph->parents));
4895 }
4896
4897 static void
4898 done_rev_graph(struct rev_graph *graph)
4899 {
4900         if (graph_parent_is_merge(graph) &&
4901             graph->pos < graph->size - 1 &&
4902             graph->next->size == graph->size + graph->parents->size - 1) {
4903                 size_t i = graph->pos + graph->parents->size - 1;
4904
4905                 graph->commit->graph_size = i * 2;
4906                 while (i < graph->next->size - 1) {
4907                         append_to_rev_graph(graph, ' ');
4908                         append_to_rev_graph(graph, '\\');
4909                         i++;
4910                 }
4911         }
4912
4913         clear_rev_graph(graph);
4914 }
4915
4916 static void
4917 push_rev_graph(struct rev_graph *graph, const char *parent)
4918 {
4919         int i;
4920
4921         /* "Collapse" duplicate parents lines.
4922          *
4923          * FIXME: This needs to also update update the drawn graph but
4924          * for now it just serves as a method for pruning graph lines. */
4925         for (i = 0; i < graph->size; i++)
4926                 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
4927                         return;
4928
4929         if (graph->size < SIZEOF_REVITEMS) {
4930                 string_copy_rev(graph->rev[graph->size++], parent);
4931         }
4932 }
4933
4934 static chtype
4935 get_rev_graph_symbol(struct rev_graph *graph)
4936 {
4937         chtype symbol;
4938
4939         if (graph->boundary)
4940                 symbol = REVGRAPH_BOUND;
4941         else if (graph->parents->size == 0)
4942                 symbol = REVGRAPH_INIT;
4943         else if (graph_parent_is_merge(graph))
4944                 symbol = REVGRAPH_MERGE;
4945         else if (graph->pos >= graph->size)
4946                 symbol = REVGRAPH_BRANCH;
4947         else
4948                 symbol = REVGRAPH_COMMIT;
4949
4950         return symbol;
4951 }
4952
4953 static void
4954 draw_rev_graph(struct rev_graph *graph)
4955 {
4956         struct rev_filler {
4957                 chtype separator, line;
4958         };
4959         enum { DEFAULT, RSHARP, RDIAG, LDIAG };
4960         static struct rev_filler fillers[] = {
4961                 { ' ',  '|' },
4962                 { '`',  '.' },
4963                 { '\'', ' ' },
4964                 { '/',  ' ' },
4965         };
4966         chtype symbol = get_rev_graph_symbol(graph);
4967         struct rev_filler *filler;
4968         size_t i;
4969
4970         if (opt_line_graphics)
4971                 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
4972
4973         filler = &fillers[DEFAULT];
4974
4975         for (i = 0; i < graph->pos; i++) {
4976                 append_to_rev_graph(graph, filler->line);
4977                 if (graph_parent_is_merge(graph->prev) &&
4978                     graph->prev->pos == i)
4979                         filler = &fillers[RSHARP];
4980
4981                 append_to_rev_graph(graph, filler->separator);
4982         }
4983
4984         /* Place the symbol for this revision. */
4985         append_to_rev_graph(graph, symbol);
4986
4987         if (graph->prev->size > graph->size)
4988                 filler = &fillers[RDIAG];
4989         else
4990                 filler = &fillers[DEFAULT];
4991
4992         i++;
4993
4994         for (; i < graph->size; i++) {
4995                 append_to_rev_graph(graph, filler->separator);
4996                 append_to_rev_graph(graph, filler->line);
4997                 if (graph_parent_is_merge(graph->prev) &&
4998                     i < graph->prev->pos + graph->parents->size)
4999                         filler = &fillers[RSHARP];
5000                 if (graph->prev->size > graph->size)
5001                         filler = &fillers[LDIAG];
5002         }
5003
5004         if (graph->prev->size > graph->size) {
5005                 append_to_rev_graph(graph, filler->separator);
5006                 if (filler->line != ' ')
5007                         append_to_rev_graph(graph, filler->line);
5008         }
5009 }
5010
5011 /* Prepare the next rev graph */
5012 static void
5013 prepare_rev_graph(struct rev_graph *graph)
5014 {
5015         size_t i;
5016
5017         /* First, traverse all lines of revisions up to the active one. */
5018         for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5019                 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5020                         break;
5021
5022                 push_rev_graph(graph->next, graph->rev[graph->pos]);
5023         }
5024
5025         /* Interleave the new revision parent(s). */
5026         for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5027                 push_rev_graph(graph->next, graph->parents->rev[i]);
5028
5029         /* Lastly, put any remaining revisions. */
5030         for (i = graph->pos + 1; i < graph->size; i++)
5031                 push_rev_graph(graph->next, graph->rev[i]);
5032 }
5033
5034 static void
5035 update_rev_graph(struct rev_graph *graph)
5036 {
5037         /* If this is the finalizing update ... */
5038         if (graph->commit)
5039                 prepare_rev_graph(graph);
5040
5041         /* Graph visualization needs a one rev look-ahead,
5042          * so the first update doesn't visualize anything. */
5043         if (!graph->prev->commit)
5044                 return;
5045
5046         draw_rev_graph(graph->prev);
5047         done_rev_graph(graph->prev->prev);
5048 }
5049
5050
5051 /*
5052  * Main view backend
5053  */
5054
5055 static bool
5056 main_draw(struct view *view, struct line *line, unsigned int lineno)
5057 {
5058         struct commit *commit = line->data;
5059
5060         if (!*commit->author)
5061                 return FALSE;
5062
5063         if (opt_date && draw_date(view, &commit->time))
5064                 return TRUE;
5065
5066         if (opt_author &&
5067             draw_field(view, LINE_MAIN_AUTHOR, commit->author, opt_author_cols, TRUE))
5068                 return TRUE;
5069
5070         if (opt_rev_graph && commit->graph_size &&
5071             draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5072                 return TRUE;
5073
5074         if (opt_show_refs && commit->refs) {
5075                 size_t i = 0;
5076
5077                 do {
5078                         enum line_type type;
5079
5080                         if (commit->refs[i]->head)
5081                                 type = LINE_MAIN_HEAD;
5082                         else if (commit->refs[i]->ltag)
5083                                 type = LINE_MAIN_LOCAL_TAG;
5084                         else if (commit->refs[i]->tag)
5085                                 type = LINE_MAIN_TAG;
5086                         else if (commit->refs[i]->tracked)
5087                                 type = LINE_MAIN_TRACKED;
5088                         else if (commit->refs[i]->remote)
5089                                 type = LINE_MAIN_REMOTE;
5090                         else
5091                                 type = LINE_MAIN_REF;
5092
5093                         if (draw_text(view, type, "[", TRUE) ||
5094                             draw_text(view, type, commit->refs[i]->name, TRUE) ||
5095                             draw_text(view, type, "]", TRUE))
5096                                 return TRUE;
5097
5098                         if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5099                                 return TRUE;
5100                 } while (commit->refs[i++]->next);
5101         }
5102
5103         draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5104         return TRUE;
5105 }
5106
5107 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5108 static bool
5109 main_read(struct view *view, char *line)
5110 {
5111         static struct rev_graph *graph = graph_stacks;
5112         enum line_type type;
5113         struct commit *commit;
5114
5115         if (!line) {
5116                 int i;
5117
5118                 if (!view->lines && !view->parent)
5119                         die("No revisions match the given arguments.");
5120                 if (view->lines > 0) {
5121                         commit = view->line[view->lines - 1].data;
5122                         if (!*commit->author) {
5123                                 view->lines--;
5124                                 free(commit);
5125                                 graph->commit = NULL;
5126                         }
5127                 }
5128                 update_rev_graph(graph);
5129
5130                 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5131                         clear_rev_graph(&graph_stacks[i]);
5132                 return TRUE;
5133         }
5134
5135         type = get_line_type(line);
5136         if (type == LINE_COMMIT) {
5137                 commit = calloc(1, sizeof(struct commit));
5138                 if (!commit)
5139                         return FALSE;
5140
5141                 line += STRING_SIZE("commit ");
5142                 if (*line == '-') {
5143                         graph->boundary = 1;
5144                         line++;
5145                 }
5146
5147                 string_copy_rev(commit->id, line);
5148                 commit->refs = get_refs(commit->id);
5149                 graph->commit = commit;
5150                 add_line_data(view, commit, LINE_MAIN_COMMIT);
5151
5152                 while ((line = strchr(line, ' '))) {
5153                         line++;
5154                         push_rev_graph(graph->parents, line);
5155                         commit->has_parents = TRUE;
5156                 }
5157                 return TRUE;
5158         }
5159
5160         if (!view->lines)
5161                 return TRUE;
5162         commit = view->line[view->lines - 1].data;
5163
5164         switch (type) {
5165         case LINE_PARENT:
5166                 if (commit->has_parents)
5167                         break;
5168                 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5169                 break;
5170
5171         case LINE_AUTHOR:
5172         {
5173                 /* Parse author lines where the name may be empty:
5174                  *      author  <email@address.tld> 1138474660 +0100
5175                  */
5176                 char *ident = line + STRING_SIZE("author ");
5177                 char *nameend = strchr(ident, '<');
5178                 char *emailend = strchr(ident, '>');
5179
5180                 if (!nameend || !emailend)
5181                         break;
5182
5183                 update_rev_graph(graph);
5184                 graph = graph->next;
5185
5186                 *nameend = *emailend = 0;
5187                 ident = chomp_string(ident);
5188                 if (!*ident) {
5189                         ident = chomp_string(nameend + 1);
5190                         if (!*ident)
5191                                 ident = "Unknown";
5192                 }
5193
5194                 string_ncopy(commit->author, ident, strlen(ident));
5195
5196                 /* Parse epoch and timezone */
5197                 if (emailend[1] == ' ') {
5198                         char *secs = emailend + 2;
5199                         char *zone = strchr(secs, ' ');
5200                         time_t time = (time_t) atol(secs);
5201
5202                         if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
5203                                 long tz;
5204
5205                                 zone++;
5206                                 tz  = ('0' - zone[1]) * 60 * 60 * 10;
5207                                 tz += ('0' - zone[2]) * 60 * 60;
5208                                 tz += ('0' - zone[3]) * 60;
5209                                 tz += ('0' - zone[4]) * 60;
5210
5211                                 if (zone[0] == '-')
5212                                         tz = -tz;
5213
5214                                 time -= tz;
5215                         }
5216
5217                         gmtime_r(&time, &commit->time);
5218                 }
5219                 break;
5220         }
5221         default:
5222                 /* Fill in the commit title if it has not already been set. */
5223                 if (commit->title[0])
5224                         break;
5225
5226                 /* Require titles to start with a non-space character at the
5227                  * offset used by git log. */
5228                 if (strncmp(line, "    ", 4))
5229                         break;
5230                 line += 4;
5231                 /* Well, if the title starts with a whitespace character,
5232                  * try to be forgiving.  Otherwise we end up with no title. */
5233                 while (isspace(*line))
5234                         line++;
5235                 if (*line == '\0')
5236                         break;
5237                 /* FIXME: More graceful handling of titles; append "..." to
5238                  * shortened titles, etc. */
5239
5240                 string_ncopy(commit->title, line, strlen(line));
5241         }
5242
5243         return TRUE;
5244 }
5245
5246 static enum request
5247 main_request(struct view *view, enum request request, struct line *line)
5248 {
5249         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5250
5251         switch (request) {
5252         case REQ_ENTER:
5253                 open_view(view, REQ_VIEW_DIFF, flags);
5254                 break;
5255         case REQ_REFRESH:
5256                 load_refs();
5257                 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
5258                 break;
5259         default:
5260                 return request;
5261         }
5262
5263         return REQ_NONE;
5264 }
5265
5266 static bool
5267 grep_refs(struct ref **refs, regex_t *regex)
5268 {
5269         regmatch_t pmatch;
5270         size_t i = 0;
5271
5272         if (!refs)
5273                 return FALSE;
5274         do {
5275                 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
5276                         return TRUE;
5277         } while (refs[i++]->next);
5278
5279         return FALSE;
5280 }
5281
5282 static bool
5283 main_grep(struct view *view, struct line *line)
5284 {
5285         struct commit *commit = line->data;
5286         enum { S_TITLE, S_AUTHOR, S_DATE, S_REFS, S_END } state;
5287         char buf[DATE_COLS + 1];
5288         regmatch_t pmatch;
5289
5290         for (state = S_TITLE; state < S_END; state++) {
5291                 char *text;
5292
5293                 switch (state) {
5294                 case S_TITLE:   text = commit->title;   break;
5295                 case S_AUTHOR:
5296                         if (!opt_author)
5297                                 continue;
5298                         text = commit->author;
5299                         break;
5300                 case S_DATE:
5301                         if (!opt_date)
5302                                 continue;
5303                         if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
5304                                 continue;
5305                         text = buf;
5306                         break;
5307                 case S_REFS:
5308                         if (!opt_show_refs)
5309                                 continue;
5310                         if (grep_refs(commit->refs, view->regex) == TRUE)
5311                                 return TRUE;
5312                         continue;
5313                 default:
5314                         return FALSE;
5315                 }
5316
5317                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5318                         return TRUE;
5319         }
5320
5321         return FALSE;
5322 }
5323
5324 static void
5325 main_select(struct view *view, struct line *line)
5326 {
5327         struct commit *commit = line->data;
5328
5329         string_copy_rev(view->ref, commit->id);
5330         string_copy_rev(ref_commit, view->ref);
5331 }
5332
5333 static struct view_ops main_ops = {
5334         "commit",
5335         NULL,
5336         main_read,
5337         main_draw,
5338         main_request,
5339         main_grep,
5340         main_select,
5341 };
5342
5343
5344 /*
5345  * Unicode / UTF-8 handling
5346  *
5347  * NOTE: Much of the following code for dealing with unicode is derived from
5348  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
5349  * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
5350  */
5351
5352 /* I've (over)annotated a lot of code snippets because I am not entirely
5353  * confident that the approach taken by this small UTF-8 interface is correct.
5354  * --jonas */
5355
5356 static inline int
5357 unicode_width(unsigned long c)
5358 {
5359         if (c >= 0x1100 &&
5360            (c <= 0x115f                         /* Hangul Jamo */
5361             || c == 0x2329
5362             || c == 0x232a
5363             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
5364                                                 /* CJK ... Yi */
5365             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
5366             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
5367             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
5368             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
5369             || (c >= 0xffe0  && c <= 0xffe6)
5370             || (c >= 0x20000 && c <= 0x2fffd)
5371             || (c >= 0x30000 && c <= 0x3fffd)))
5372                 return 2;
5373
5374         if (c == '\t')
5375                 return opt_tab_size;
5376
5377         return 1;
5378 }
5379
5380 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
5381  * Illegal bytes are set one. */
5382 static const unsigned char utf8_bytes[256] = {
5383         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,
5384         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,
5385         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,
5386         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,
5387         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,
5388         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,
5389         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,
5390         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,
5391 };
5392
5393 /* Decode UTF-8 multi-byte representation into a unicode character. */
5394 static inline unsigned long
5395 utf8_to_unicode(const char *string, size_t length)
5396 {
5397         unsigned long unicode;
5398
5399         switch (length) {
5400         case 1:
5401                 unicode  =   string[0];
5402                 break;
5403         case 2:
5404                 unicode  =  (string[0] & 0x1f) << 6;
5405                 unicode +=  (string[1] & 0x3f);
5406                 break;
5407         case 3:
5408                 unicode  =  (string[0] & 0x0f) << 12;
5409                 unicode += ((string[1] & 0x3f) << 6);
5410                 unicode +=  (string[2] & 0x3f);
5411                 break;
5412         case 4:
5413                 unicode  =  (string[0] & 0x0f) << 18;
5414                 unicode += ((string[1] & 0x3f) << 12);
5415                 unicode += ((string[2] & 0x3f) << 6);
5416                 unicode +=  (string[3] & 0x3f);
5417                 break;
5418         case 5:
5419                 unicode  =  (string[0] & 0x0f) << 24;
5420                 unicode += ((string[1] & 0x3f) << 18);
5421                 unicode += ((string[2] & 0x3f) << 12);
5422                 unicode += ((string[3] & 0x3f) << 6);
5423                 unicode +=  (string[4] & 0x3f);
5424                 break;
5425         case 6:
5426                 unicode  =  (string[0] & 0x01) << 30;
5427                 unicode += ((string[1] & 0x3f) << 24);
5428                 unicode += ((string[2] & 0x3f) << 18);
5429                 unicode += ((string[3] & 0x3f) << 12);
5430                 unicode += ((string[4] & 0x3f) << 6);
5431                 unicode +=  (string[5] & 0x3f);
5432                 break;
5433         default:
5434                 die("Invalid unicode length");
5435         }
5436
5437         /* Invalid characters could return the special 0xfffd value but NUL
5438          * should be just as good. */
5439         return unicode > 0xffff ? 0 : unicode;
5440 }
5441
5442 /* Calculates how much of string can be shown within the given maximum width
5443  * and sets trimmed parameter to non-zero value if all of string could not be
5444  * shown. If the reserve flag is TRUE, it will reserve at least one
5445  * trailing character, which can be useful when drawing a delimiter.
5446  *
5447  * Returns the number of bytes to output from string to satisfy max_width. */
5448 static size_t
5449 utf8_length(const char *string, int *width, size_t max_width, int *trimmed, bool reserve)
5450 {
5451         const char *start = string;
5452         const char *end = strchr(string, '\0');
5453         unsigned char last_bytes = 0;
5454         size_t last_ucwidth = 0;
5455
5456         *width = 0;
5457         *trimmed = 0;
5458
5459         while (string < end) {
5460                 int c = *(unsigned char *) string;
5461                 unsigned char bytes = utf8_bytes[c];
5462                 size_t ucwidth;
5463                 unsigned long unicode;
5464
5465                 if (string + bytes > end)
5466                         break;
5467
5468                 /* Change representation to figure out whether
5469                  * it is a single- or double-width character. */
5470
5471                 unicode = utf8_to_unicode(string, bytes);
5472                 /* FIXME: Graceful handling of invalid unicode character. */
5473                 if (!unicode)
5474                         break;
5475
5476                 ucwidth = unicode_width(unicode);
5477                 *width  += ucwidth;
5478                 if (*width > max_width) {
5479                         *trimmed = 1;
5480                         *width -= ucwidth;
5481                         if (reserve && *width == max_width) {
5482                                 string -= last_bytes;
5483                                 *width -= last_ucwidth;
5484                         }
5485                         break;
5486                 }
5487
5488                 string  += bytes;
5489                 last_bytes = bytes;
5490                 last_ucwidth = ucwidth;
5491         }
5492
5493         return string - start;
5494 }
5495
5496
5497 /*
5498  * Status management
5499  */
5500
5501 /* Whether or not the curses interface has been initialized. */
5502 static bool cursed = FALSE;
5503
5504 /* The status window is used for polling keystrokes. */
5505 static WINDOW *status_win;
5506
5507 static bool status_empty = TRUE;
5508
5509 /* Update status and title window. */
5510 static void
5511 report(const char *msg, ...)
5512 {
5513         struct view *view = display[current_view];
5514
5515         if (input_mode)
5516                 return;
5517
5518         if (!view) {
5519                 char buf[SIZEOF_STR];
5520                 va_list args;
5521
5522                 va_start(args, msg);
5523                 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
5524                         buf[sizeof(buf) - 1] = 0;
5525                         buf[sizeof(buf) - 2] = '.';
5526                         buf[sizeof(buf) - 3] = '.';
5527                         buf[sizeof(buf) - 4] = '.';
5528                 }
5529                 va_end(args);
5530                 die("%s", buf);
5531         }
5532
5533         if (!status_empty || *msg) {
5534                 va_list args;
5535
5536                 va_start(args, msg);
5537
5538                 wmove(status_win, 0, 0);
5539                 if (*msg) {
5540                         vwprintw(status_win, msg, args);
5541                         status_empty = FALSE;
5542                 } else {
5543                         status_empty = TRUE;
5544                 }
5545                 wclrtoeol(status_win);
5546                 wrefresh(status_win);
5547
5548                 va_end(args);
5549         }
5550
5551         update_view_title(view);
5552         update_display_cursor(view);
5553 }
5554
5555 /* Controls when nodelay should be in effect when polling user input. */
5556 static void
5557 set_nonblocking_input(bool loading)
5558 {
5559         static unsigned int loading_views;
5560
5561         if ((loading == FALSE && loading_views-- == 1) ||
5562             (loading == TRUE  && loading_views++ == 0))
5563                 nodelay(status_win, loading);
5564 }
5565
5566 static void
5567 init_display(void)
5568 {
5569         int x, y;
5570
5571         /* Initialize the curses library */
5572         if (isatty(STDIN_FILENO)) {
5573                 cursed = !!initscr();
5574                 opt_tty = stdin;
5575         } else {
5576                 /* Leave stdin and stdout alone when acting as a pager. */
5577                 opt_tty = fopen("/dev/tty", "r+");
5578                 if (!opt_tty)
5579                         die("Failed to open /dev/tty");
5580                 cursed = !!newterm(NULL, opt_tty, opt_tty);
5581         }
5582
5583         if (!cursed)
5584                 die("Failed to initialize curses");
5585
5586         nonl();         /* Tell curses not to do NL->CR/NL on output */
5587         cbreak();       /* Take input chars one at a time, no wait for \n */
5588         noecho();       /* Don't echo input */
5589         leaveok(stdscr, TRUE);
5590
5591         if (has_colors())
5592                 init_colors();
5593
5594         getmaxyx(stdscr, y, x);
5595         status_win = newwin(1, 0, y - 1, 0);
5596         if (!status_win)
5597                 die("Failed to create status window");
5598
5599         /* Enable keyboard mapping */
5600         keypad(status_win, TRUE);
5601         wbkgdset(status_win, get_line_attr(LINE_STATUS));
5602
5603         TABSIZE = opt_tab_size;
5604         if (opt_line_graphics) {
5605                 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
5606         }
5607 }
5608
5609 static bool
5610 prompt_yesno(const char *prompt)
5611 {
5612         enum { WAIT, STOP, CANCEL  } status = WAIT;
5613         bool answer = FALSE;
5614
5615         while (status == WAIT) {
5616                 struct view *view;
5617                 int i, key;
5618
5619                 input_mode = TRUE;
5620
5621                 foreach_view (view, i)
5622                         update_view(view);
5623
5624                 input_mode = FALSE;
5625
5626                 mvwprintw(status_win, 0, 0, "%s [Yy]/[Nn]", prompt);
5627                 wclrtoeol(status_win);
5628
5629                 /* Refresh, accept single keystroke of input */
5630                 key = wgetch(status_win);
5631                 switch (key) {
5632                 case ERR:
5633                         break;
5634
5635                 case 'y':
5636                 case 'Y':
5637                         answer = TRUE;
5638                         status = STOP;
5639                         break;
5640
5641                 case KEY_ESC:
5642                 case KEY_RETURN:
5643                 case KEY_ENTER:
5644                 case KEY_BACKSPACE:
5645                 case 'n':
5646                 case 'N':
5647                 case '\n':
5648                 default:
5649                         answer = FALSE;
5650                         status = CANCEL;
5651                 }
5652         }
5653
5654         /* Clear the status window */
5655         status_empty = FALSE;
5656         report("");
5657
5658         return answer;
5659 }
5660
5661 static char *
5662 read_prompt(const char *prompt)
5663 {
5664         enum { READING, STOP, CANCEL } status = READING;
5665         static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
5666         int pos = 0;
5667
5668         while (status == READING) {
5669                 struct view *view;
5670                 int i, key;
5671
5672                 input_mode = TRUE;
5673
5674                 foreach_view (view, i)
5675                         update_view(view);
5676
5677                 input_mode = FALSE;
5678
5679                 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
5680                 wclrtoeol(status_win);
5681
5682                 /* Refresh, accept single keystroke of input */
5683                 key = wgetch(status_win);
5684                 switch (key) {
5685                 case KEY_RETURN:
5686                 case KEY_ENTER:
5687                 case '\n':
5688                         status = pos ? STOP : CANCEL;
5689                         break;
5690
5691                 case KEY_BACKSPACE:
5692                         if (pos > 0)
5693                                 pos--;
5694                         else
5695                                 status = CANCEL;
5696                         break;
5697
5698                 case KEY_ESC:
5699                         status = CANCEL;
5700                         break;
5701
5702                 case ERR:
5703                         break;
5704
5705                 default:
5706                         if (pos >= sizeof(buf)) {
5707                                 report("Input string too long");
5708                                 return NULL;
5709                         }
5710
5711                         if (isprint(key))
5712                                 buf[pos++] = (char) key;
5713                 }
5714         }
5715
5716         /* Clear the status window */
5717         status_empty = FALSE;
5718         report("");
5719
5720         if (status == CANCEL)
5721                 return NULL;
5722
5723         buf[pos++] = 0;
5724
5725         return buf;
5726 }
5727
5728 /*
5729  * Repository references
5730  */
5731
5732 static struct ref *refs = NULL;
5733 static size_t refs_alloc = 0;
5734 static size_t refs_size = 0;
5735
5736 /* Id <-> ref store */
5737 static struct ref ***id_refs = NULL;
5738 static size_t id_refs_alloc = 0;
5739 static size_t id_refs_size = 0;
5740
5741 static int
5742 compare_refs(const void *ref1_, const void *ref2_)
5743 {
5744         const struct ref *ref1 = *(const struct ref **)ref1_;
5745         const struct ref *ref2 = *(const struct ref **)ref2_;
5746
5747         if (ref1->tag != ref2->tag)
5748                 return ref2->tag - ref1->tag;
5749         if (ref1->ltag != ref2->ltag)
5750                 return ref2->ltag - ref2->ltag;
5751         if (ref1->head != ref2->head)
5752                 return ref2->head - ref1->head;
5753         if (ref1->tracked != ref2->tracked)
5754                 return ref2->tracked - ref1->tracked;
5755         if (ref1->remote != ref2->remote)
5756                 return ref2->remote - ref1->remote;
5757         return strcmp(ref1->name, ref2->name);
5758 }
5759
5760 static struct ref **
5761 get_refs(const char *id)
5762 {
5763         struct ref ***tmp_id_refs;
5764         struct ref **ref_list = NULL;
5765         size_t ref_list_alloc = 0;
5766         size_t ref_list_size = 0;
5767         size_t i;
5768
5769         for (i = 0; i < id_refs_size; i++)
5770                 if (!strcmp(id, id_refs[i][0]->id))
5771                         return id_refs[i];
5772
5773         tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
5774                                     sizeof(*id_refs));
5775         if (!tmp_id_refs)
5776                 return NULL;
5777
5778         id_refs = tmp_id_refs;
5779
5780         for (i = 0; i < refs_size; i++) {
5781                 struct ref **tmp;
5782
5783                 if (strcmp(id, refs[i].id))
5784                         continue;
5785
5786                 tmp = realloc_items(ref_list, &ref_list_alloc,
5787                                     ref_list_size + 1, sizeof(*ref_list));
5788                 if (!tmp) {
5789                         if (ref_list)
5790                                 free(ref_list);
5791                         return NULL;
5792                 }
5793
5794                 ref_list = tmp;
5795                 ref_list[ref_list_size] = &refs[i];
5796                 /* XXX: The properties of the commit chains ensures that we can
5797                  * safely modify the shared ref. The repo references will
5798                  * always be similar for the same id. */
5799                 ref_list[ref_list_size]->next = 1;
5800
5801                 ref_list_size++;
5802         }
5803
5804         if (ref_list) {
5805                 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
5806                 ref_list[ref_list_size - 1]->next = 0;
5807                 id_refs[id_refs_size++] = ref_list;
5808         }
5809
5810         return ref_list;
5811 }
5812
5813 static int
5814 read_ref(char *id, size_t idlen, char *name, size_t namelen)
5815 {
5816         struct ref *ref;
5817         bool tag = FALSE;
5818         bool ltag = FALSE;
5819         bool remote = FALSE;
5820         bool tracked = FALSE;
5821         bool check_replace = FALSE;
5822         bool head = FALSE;
5823
5824         if (!prefixcmp(name, "refs/tags/")) {
5825                 if (!suffixcmp(name, namelen, "^{}")) {
5826                         namelen -= 3;
5827                         name[namelen] = 0;
5828                         if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
5829                                 check_replace = TRUE;
5830                 } else {
5831                         ltag = TRUE;
5832                 }
5833
5834                 tag = TRUE;
5835                 namelen -= STRING_SIZE("refs/tags/");
5836                 name    += STRING_SIZE("refs/tags/");
5837
5838         } else if (!prefixcmp(name, "refs/remotes/")) {
5839                 remote = TRUE;
5840                 namelen -= STRING_SIZE("refs/remotes/");
5841                 name    += STRING_SIZE("refs/remotes/");
5842                 tracked  = !strcmp(opt_remote, name);
5843
5844         } else if (!prefixcmp(name, "refs/heads/")) {
5845                 namelen -= STRING_SIZE("refs/heads/");
5846                 name    += STRING_SIZE("refs/heads/");
5847                 head     = !strncmp(opt_head, name, namelen);
5848
5849         } else if (!strcmp(name, "HEAD")) {
5850                 string_ncopy(opt_head_rev, id, idlen);
5851                 return OK;
5852         }
5853
5854         if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
5855                 /* it's an annotated tag, replace the previous sha1 with the
5856                  * resolved commit id; relies on the fact git-ls-remote lists
5857                  * the commit id of an annotated tag right before the commit id
5858                  * it points to. */
5859                 refs[refs_size - 1].ltag = ltag;
5860                 string_copy_rev(refs[refs_size - 1].id, id);
5861
5862                 return OK;
5863         }
5864         refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
5865         if (!refs)
5866                 return ERR;
5867
5868         ref = &refs[refs_size++];
5869         ref->name = malloc(namelen + 1);
5870         if (!ref->name)
5871                 return ERR;
5872
5873         strncpy(ref->name, name, namelen);
5874         ref->name[namelen] = 0;
5875         ref->head = head;
5876         ref->tag = tag;
5877         ref->ltag = ltag;
5878         ref->remote = remote;
5879         ref->tracked = tracked;
5880         string_copy_rev(ref->id, id);
5881
5882         return OK;
5883 }
5884
5885 static int
5886 load_refs(void)
5887 {
5888         const char *cmd_env = getenv("TIG_LS_REMOTE");
5889         const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
5890
5891         if (!*opt_git_dir)
5892                 return OK;
5893
5894         while (refs_size > 0)
5895                 free(refs[--refs_size].name);
5896         while (id_refs_size > 0)
5897                 free(id_refs[--id_refs_size]);
5898
5899         return read_properties(popen(cmd, "r"), "\t", read_ref);
5900 }
5901
5902 static int
5903 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
5904 {
5905         if (!strcmp(name, "i18n.commitencoding"))
5906                 string_ncopy(opt_encoding, value, valuelen);
5907
5908         if (!strcmp(name, "core.editor"))
5909                 string_ncopy(opt_editor, value, valuelen);
5910
5911         /* branch.<head>.remote */
5912         if (*opt_head &&
5913             !strncmp(name, "branch.", 7) &&
5914             !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5915             !strcmp(name + 7 + strlen(opt_head), ".remote"))
5916                 string_ncopy(opt_remote, value, valuelen);
5917
5918         if (*opt_head && *opt_remote &&
5919             !strncmp(name, "branch.", 7) &&
5920             !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5921             !strcmp(name + 7 + strlen(opt_head), ".merge")) {
5922                 size_t from = strlen(opt_remote);
5923
5924                 if (!prefixcmp(value, "refs/heads/")) {
5925                         value += STRING_SIZE("refs/heads/");
5926                         valuelen -= STRING_SIZE("refs/heads/");
5927                 }
5928
5929                 if (!string_format_from(opt_remote, &from, "/%s", value))
5930                         opt_remote[0] = 0;
5931         }
5932
5933         return OK;
5934 }
5935
5936 static int
5937 load_git_config(void)
5938 {
5939         return read_properties(popen("git " GIT_CONFIG " --list", "r"),
5940                                "=", read_repo_config_option);
5941 }
5942
5943 static int
5944 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
5945 {
5946         if (!opt_git_dir[0]) {
5947                 string_ncopy(opt_git_dir, name, namelen);
5948
5949         } else if (opt_is_inside_work_tree == -1) {
5950                 /* This can be 3 different values depending on the
5951                  * version of git being used. If git-rev-parse does not
5952                  * understand --is-inside-work-tree it will simply echo
5953                  * the option else either "true" or "false" is printed.
5954                  * Default to true for the unknown case. */
5955                 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
5956
5957         } else if (opt_cdup[0] == ' ') {
5958                 string_ncopy(opt_cdup, name, namelen);
5959         } else {
5960                 if (!prefixcmp(name, "refs/heads/")) {
5961                         namelen -= STRING_SIZE("refs/heads/");
5962                         name    += STRING_SIZE("refs/heads/");
5963                         string_ncopy(opt_head, name, namelen);
5964                 }
5965         }
5966
5967         return OK;
5968 }
5969
5970 static int
5971 load_repo_info(void)
5972 {
5973         int result;
5974         FILE *pipe = popen("(git rev-parse --git-dir --is-inside-work-tree "
5975                            " --show-cdup; git symbolic-ref HEAD) 2>/dev/null", "r");
5976
5977         /* XXX: The line outputted by "--show-cdup" can be empty so
5978          * initialize it to something invalid to make it possible to
5979          * detect whether it has been set or not. */
5980         opt_cdup[0] = ' ';
5981
5982         result = read_properties(pipe, "=", read_repo_info);
5983         if (opt_cdup[0] == ' ')
5984                 opt_cdup[0] = 0;
5985
5986         return result;
5987 }
5988
5989 static int
5990 read_properties(FILE *pipe, const char *separators,
5991                 int (*read_property)(char *, size_t, char *, size_t))
5992 {
5993         char buffer[BUFSIZ];
5994         char *name;
5995         int state = OK;
5996
5997         if (!pipe)
5998                 return ERR;
5999
6000         while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
6001                 char *value;
6002                 size_t namelen;
6003                 size_t valuelen;
6004
6005                 name = chomp_string(name);
6006                 namelen = strcspn(name, separators);
6007
6008                 if (name[namelen]) {
6009                         name[namelen] = 0;
6010                         value = chomp_string(name + namelen + 1);
6011                         valuelen = strlen(value);
6012
6013                 } else {
6014                         value = "";
6015                         valuelen = 0;
6016                 }
6017
6018                 state = read_property(name, namelen, value, valuelen);
6019         }
6020
6021         if (state != ERR && ferror(pipe))
6022                 state = ERR;
6023
6024         pclose(pipe);
6025
6026         return state;
6027 }
6028
6029
6030 /*
6031  * Main
6032  */
6033
6034 static void __NORETURN
6035 quit(int sig)
6036 {
6037         /* XXX: Restore tty modes and let the OS cleanup the rest! */
6038         if (cursed)
6039                 endwin();
6040         exit(0);
6041 }
6042
6043 static void __NORETURN
6044 die(const char *err, ...)
6045 {
6046         va_list args;
6047
6048         endwin();
6049
6050         va_start(args, err);
6051         fputs("tig: ", stderr);
6052         vfprintf(stderr, err, args);
6053         fputs("\n", stderr);
6054         va_end(args);
6055
6056         exit(1);
6057 }
6058
6059 static void
6060 warn(const char *msg, ...)
6061 {
6062         va_list args;
6063
6064         va_start(args, msg);
6065         fputs("tig warning: ", stderr);
6066         vfprintf(stderr, msg, args);
6067         fputs("\n", stderr);
6068         va_end(args);
6069 }
6070
6071 int
6072 main(int argc, const char *argv[])
6073 {
6074         struct view *view;
6075         enum request request;
6076         size_t i;
6077
6078         signal(SIGINT, quit);
6079
6080         if (setlocale(LC_ALL, "")) {
6081                 char *codeset = nl_langinfo(CODESET);
6082
6083                 string_ncopy(opt_codeset, codeset, strlen(codeset));
6084         }
6085
6086         if (load_repo_info() == ERR)
6087                 die("Failed to load repo info.");
6088
6089         if (load_options() == ERR)
6090                 die("Failed to load user config.");
6091
6092         if (load_git_config() == ERR)
6093                 die("Failed to load repo config.");
6094
6095         request = parse_options(argc, argv);
6096         if (request == REQ_NONE)
6097                 return 0;
6098
6099         /* Require a git repository unless when running in pager mode. */
6100         if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
6101                 die("Not a git repository");
6102
6103         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
6104                 opt_utf8 = FALSE;
6105
6106         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
6107                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
6108                 if (opt_iconv == ICONV_NONE)
6109                         die("Failed to initialize character set conversion");
6110         }
6111
6112         if (load_refs() == ERR)
6113                 die("Failed to load refs.");
6114
6115         foreach_view (view, i)
6116                 view->cmd_env = getenv(view->cmd_env);
6117
6118         init_display();
6119
6120         while (view_driver(display[current_view], request)) {
6121                 int key;
6122                 int i;
6123
6124                 foreach_view (view, i)
6125                         update_view(view);
6126                 view = display[current_view];
6127
6128                 /* Refresh, accept single keystroke of input */
6129                 key = wgetch(status_win);
6130
6131                 /* wgetch() with nodelay() enabled returns ERR when there's no
6132                  * input. */
6133                 if (key == ERR) {
6134                         request = REQ_NONE;
6135                         continue;
6136                 }
6137
6138                 request = get_keybinding(view->keymap, key);
6139
6140                 /* Some low-level request handling. This keeps access to
6141                  * status_win restricted. */
6142                 switch (request) {
6143                 case REQ_PROMPT:
6144                 {
6145                         char *cmd = read_prompt(":");
6146
6147                         if (cmd && string_format(opt_cmd, "git %s", cmd)) {
6148                                 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
6149                                         request = REQ_VIEW_DIFF;
6150                                 } else {
6151                                         request = REQ_VIEW_PAGER;
6152                                 }
6153
6154                                 /* Always reload^Wrerun commands from the prompt. */
6155                                 open_view(view, request, OPEN_RELOAD);
6156                         }
6157
6158                         request = REQ_NONE;
6159                         break;
6160                 }
6161                 case REQ_SEARCH:
6162                 case REQ_SEARCH_BACK:
6163                 {
6164                         const char *prompt = request == REQ_SEARCH ? "/" : "?";
6165                         char *search = read_prompt(prompt);
6166
6167                         if (search)
6168                                 string_ncopy(opt_search, search, strlen(search));
6169                         else
6170                                 request = REQ_NONE;
6171                         break;
6172                 }
6173                 case REQ_SCREEN_RESIZE:
6174                 {
6175                         int height, width;
6176
6177                         getmaxyx(stdscr, height, width);
6178
6179                         /* Resize the status view and let the view driver take
6180                          * care of resizing the displayed views. */
6181                         wresize(status_win, 1, width);
6182                         mvwin(status_win, height - 1, 0);
6183                         wrefresh(status_win);
6184                         break;
6185                 }
6186                 default:
6187                         break;
6188                 }
6189         }
6190
6191         quit(0);
6192
6193         return 0;
6194 }