Refactor author drawing into draw_author
[tig] / tig.c
1 /* Copyright (c) 2006-2009 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/wait.h>
36 #include <sys/stat.h>
37 #include <sys/select.h>
38 #include <unistd.h>
39 #include <time.h>
40 #include <fcntl.h>
41
42 #include <regex.h>
43
44 #include <locale.h>
45 #include <langinfo.h>
46 #include <iconv.h>
47
48 /* ncurses(3): Must be defined to have extended wide-character functions. */
49 #define _XOPEN_SOURCE_EXTENDED
50
51 #ifdef HAVE_NCURSESW_NCURSES_H
52 #include <ncursesw/ncurses.h>
53 #else
54 #ifdef HAVE_NCURSES_NCURSES_H
55 #include <ncurses/ncurses.h>
56 #else
57 #include <ncurses.h>
58 #endif
59 #endif
60
61 #if __GNUC__ >= 3
62 #define __NORETURN __attribute__((__noreturn__))
63 #else
64 #define __NORETURN
65 #endif
66
67 static void __NORETURN die(const char *err, ...);
68 static void warn(const char *msg, ...);
69 static void report(const char *msg, ...);
70 static void set_nonblocking_input(bool loading);
71 static size_t utf8_length(const char *string, int *width, size_t max_width, int *trimmed, bool reserve);
72 static int load_refs(void);
73
74 #define ABS(x)          ((x) >= 0  ? (x) : -(x))
75 #define MIN(x, y)       ((x) < (y) ? (x) :  (y))
76
77 #define ARRAY_SIZE(x)   (sizeof(x) / sizeof(x[0]))
78 #define STRING_SIZE(x)  (sizeof(x) - 1)
79
80 #define SIZEOF_STR      1024    /* Default string size. */
81 #define SIZEOF_REF      256     /* Size of symbolic or SHA1 ID. */
82 #define SIZEOF_REV      41      /* Holds a SHA-1 and an ending NUL. */
83 #define SIZEOF_ARG      32      /* Default argument array size. */
84
85 /* Revision graph */
86
87 #define REVGRAPH_INIT   'I'
88 #define REVGRAPH_MERGE  'M'
89 #define REVGRAPH_BRANCH '+'
90 #define REVGRAPH_COMMIT '*'
91 #define REVGRAPH_BOUND  '^'
92
93 #define SIZEOF_REVGRAPH 19      /* Size of revision ancestry graphics. */
94
95 /* This color name can be used to refer to the default term colors. */
96 #define COLOR_DEFAULT   (-1)
97
98 #define ICONV_NONE      ((iconv_t) -1)
99 #ifndef ICONV_CONST
100 #define ICONV_CONST     /* nothing */
101 #endif
102
103 /* The format and size of the date column in the main view. */
104 #define DATE_FORMAT     "%Y-%m-%d %H:%M"
105 #define DATE_COLS       STRING_SIZE("2006-04-29 14:21 ")
106
107 #define AUTHOR_COLS     20
108 #define ID_COLS         8
109
110 /* The default interval between line numbers. */
111 #define NUMBER_INTERVAL 5
112
113 #define TAB_SIZE        8
114
115 #define SCALE_SPLIT_VIEW(height)        ((height) * 2 / 3)
116
117 #define NULL_ID         "0000000000000000000000000000000000000000"
118
119 #ifndef GIT_CONFIG
120 #define GIT_CONFIG "config"
121 #endif
122
123 /* Some ascii-shorthands fitted into the ncurses namespace. */
124 #define KEY_TAB         '\t'
125 #define KEY_RETURN      '\r'
126 #define KEY_ESC         27
127
128
129 struct ref {
130         char *name;             /* Ref name; tag or head names are shortened. */
131         char id[SIZEOF_REV];    /* Commit SHA1 ID */
132         unsigned int head:1;    /* Is it the current HEAD? */
133         unsigned int tag:1;     /* Is it a tag? */
134         unsigned int ltag:1;    /* If so, is the tag local? */
135         unsigned int remote:1;  /* Is it a remote ref? */
136         unsigned int tracked:1; /* Is it the remote for the current HEAD? */
137         unsigned int next:1;    /* For ref lists: are there more refs? */
138 };
139
140 static struct ref **get_refs(const char *id);
141
142 enum format_flags {
143         FORMAT_ALL,             /* Perform replacement in all arguments. */
144         FORMAT_DASH,            /* Perform replacement up until "--". */
145         FORMAT_NONE             /* No replacement should be performed. */
146 };
147
148 static bool format_argv(const char *dst[], const char *src[], enum format_flags flags);
149
150 struct int_map {
151         const char *name;
152         int namelen;
153         int value;
154 };
155
156 static int
157 set_from_int_map(struct int_map *map, size_t map_size,
158                  int *value, const char *name, int namelen)
159 {
160
161         int i;
162
163         for (i = 0; i < map_size; i++)
164                 if (namelen == map[i].namelen &&
165                     !strncasecmp(name, map[i].name, namelen)) {
166                         *value = map[i].value;
167                         return OK;
168                 }
169
170         return ERR;
171 }
172
173 enum input_status {
174         INPUT_OK,
175         INPUT_SKIP,
176         INPUT_STOP,
177         INPUT_CANCEL
178 };
179
180 typedef enum input_status (*input_handler)(void *data, char *buf, int c);
181
182 static char *prompt_input(const char *prompt, input_handler handler, void *data);
183 static bool prompt_yesno(const char *prompt);
184
185 /*
186  * String helpers
187  */
188
189 static inline void
190 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
191 {
192         if (srclen > dstlen - 1)
193                 srclen = dstlen - 1;
194
195         strncpy(dst, src, srclen);
196         dst[srclen] = 0;
197 }
198
199 /* Shorthands for safely copying into a fixed buffer. */
200
201 #define string_copy(dst, src) \
202         string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
203
204 #define string_ncopy(dst, src, srclen) \
205         string_ncopy_do(dst, sizeof(dst), src, srclen)
206
207 #define string_copy_rev(dst, src) \
208         string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
209
210 #define string_add(dst, from, src) \
211         string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
212
213 static char *
214 chomp_string(char *name)
215 {
216         int namelen;
217
218         while (isspace(*name))
219                 name++;
220
221         namelen = strlen(name) - 1;
222         while (namelen > 0 && isspace(name[namelen]))
223                 name[namelen--] = 0;
224
225         return name;
226 }
227
228 static bool
229 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
230 {
231         va_list args;
232         size_t pos = bufpos ? *bufpos : 0;
233
234         va_start(args, fmt);
235         pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
236         va_end(args);
237
238         if (bufpos)
239                 *bufpos = pos;
240
241         return pos >= bufsize ? FALSE : TRUE;
242 }
243
244 #define string_format(buf, fmt, args...) \
245         string_nformat(buf, sizeof(buf), NULL, fmt, args)
246
247 #define string_format_from(buf, from, fmt, args...) \
248         string_nformat(buf, sizeof(buf), from, fmt, args)
249
250 static int
251 string_enum_compare(const char *str1, const char *str2, int len)
252 {
253         size_t i;
254
255 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
256
257         /* Diff-Header == DIFF_HEADER */
258         for (i = 0; i < len; i++) {
259                 if (toupper(str1[i]) == toupper(str2[i]))
260                         continue;
261
262                 if (string_enum_sep(str1[i]) &&
263                     string_enum_sep(str2[i]))
264                         continue;
265
266                 return str1[i] - str2[i];
267         }
268
269         return 0;
270 }
271
272 #define prefixcmp(str1, str2) \
273         strncmp(str1, str2, STRING_SIZE(str2))
274
275 static inline int
276 suffixcmp(const char *str, int slen, const char *suffix)
277 {
278         size_t len = slen >= 0 ? slen : strlen(str);
279         size_t suffixlen = strlen(suffix);
280
281         return suffixlen < len ? strcmp(str + len - suffixlen, suffix) : -1;
282 }
283
284
285 static bool
286 argv_from_string(const char *argv[SIZEOF_ARG], int *argc, char *cmd)
287 {
288         int valuelen;
289
290         while (*cmd && *argc < SIZEOF_ARG && (valuelen = strcspn(cmd, " \t"))) {
291                 bool advance = cmd[valuelen] != 0;
292
293                 cmd[valuelen] = 0;
294                 argv[(*argc)++] = chomp_string(cmd);
295                 cmd = chomp_string(cmd + valuelen + advance);
296         }
297
298         if (*argc < SIZEOF_ARG)
299                 argv[*argc] = NULL;
300         return *argc < SIZEOF_ARG;
301 }
302
303 static void
304 argv_from_env(const char **argv, const char *name)
305 {
306         char *env = argv ? getenv(name) : NULL;
307         int argc = 0;
308
309         if (env && *env)
310                 env = strdup(env);
311         if (env && !argv_from_string(argv, &argc, env))
312                 die("Too many arguments in the `%s` environment variable", name);
313 }
314
315
316 /*
317  * Executing external commands.
318  */
319
320 enum io_type {
321         IO_FD,                  /* File descriptor based IO. */
322         IO_BG,                  /* Execute command in the background. */
323         IO_FG,                  /* Execute command with same std{in,out,err}. */
324         IO_RD,                  /* Read only fork+exec IO. */
325         IO_WR,                  /* Write only fork+exec IO. */
326         IO_AP,                  /* Append fork+exec output to file. */
327 };
328
329 struct io {
330         enum io_type type;      /* The requested type of pipe. */
331         const char *dir;        /* Directory from which to execute. */
332         pid_t pid;              /* Pipe for reading or writing. */
333         int pipe;               /* Pipe end for reading or writing. */
334         int error;              /* Error status. */
335         const char *argv[SIZEOF_ARG];   /* Shell command arguments. */
336         char *buf;              /* Read buffer. */
337         size_t bufalloc;        /* Allocated buffer size. */
338         size_t bufsize;         /* Buffer content size. */
339         char *bufpos;           /* Current buffer position. */
340         unsigned int eof:1;     /* Has end of file been reached. */
341 };
342
343 static void
344 reset_io(struct io *io)
345 {
346         io->pipe = -1;
347         io->pid = 0;
348         io->buf = io->bufpos = NULL;
349         io->bufalloc = io->bufsize = 0;
350         io->error = 0;
351         io->eof = 0;
352 }
353
354 static void
355 init_io(struct io *io, const char *dir, enum io_type type)
356 {
357         reset_io(io);
358         io->type = type;
359         io->dir = dir;
360 }
361
362 static bool
363 init_io_rd(struct io *io, const char *argv[], const char *dir,
364                 enum format_flags flags)
365 {
366         init_io(io, dir, IO_RD);
367         return format_argv(io->argv, argv, flags);
368 }
369
370 static bool
371 io_open(struct io *io, const char *name)
372 {
373         init_io(io, NULL, IO_FD);
374         io->pipe = *name ? open(name, O_RDONLY) : STDIN_FILENO;
375         return io->pipe != -1;
376 }
377
378 static bool
379 kill_io(struct io *io)
380 {
381         return io->pid == 0 || kill(io->pid, SIGKILL) != -1;
382 }
383
384 static bool
385 done_io(struct io *io)
386 {
387         pid_t pid = io->pid;
388
389         if (io->pipe != -1)
390                 close(io->pipe);
391         free(io->buf);
392         reset_io(io);
393
394         while (pid > 0) {
395                 int status;
396                 pid_t waiting = waitpid(pid, &status, 0);
397
398                 if (waiting < 0) {
399                         if (errno == EINTR)
400                                 continue;
401                         report("waitpid failed (%s)", strerror(errno));
402                         return FALSE;
403                 }
404
405                 return waiting == pid &&
406                        !WIFSIGNALED(status) &&
407                        WIFEXITED(status) &&
408                        !WEXITSTATUS(status);
409         }
410
411         return TRUE;
412 }
413
414 static bool
415 start_io(struct io *io)
416 {
417         int pipefds[2] = { -1, -1 };
418
419         if (io->type == IO_FD)
420                 return TRUE;
421
422         if ((io->type == IO_RD || io->type == IO_WR) &&
423             pipe(pipefds) < 0)
424                 return FALSE;
425         else if (io->type == IO_AP)
426                 pipefds[1] = io->pipe;
427
428         if ((io->pid = fork())) {
429                 if (pipefds[!(io->type == IO_WR)] != -1)
430                         close(pipefds[!(io->type == IO_WR)]);
431                 if (io->pid != -1) {
432                         io->pipe = pipefds[!!(io->type == IO_WR)];
433                         return TRUE;
434                 }
435
436         } else {
437                 if (io->type != IO_FG) {
438                         int devnull = open("/dev/null", O_RDWR);
439                         int readfd  = io->type == IO_WR ? pipefds[0] : devnull;
440                         int writefd = (io->type == IO_RD || io->type == IO_AP)
441                                                         ? pipefds[1] : devnull;
442
443                         dup2(readfd,  STDIN_FILENO);
444                         dup2(writefd, STDOUT_FILENO);
445                         dup2(devnull, STDERR_FILENO);
446
447                         close(devnull);
448                         if (pipefds[0] != -1)
449                                 close(pipefds[0]);
450                         if (pipefds[1] != -1)
451                                 close(pipefds[1]);
452                 }
453
454                 if (io->dir && *io->dir && chdir(io->dir) == -1)
455                         die("Failed to change directory: %s", strerror(errno));
456
457                 execvp(io->argv[0], (char *const*) io->argv);
458                 die("Failed to execute program: %s", strerror(errno));
459         }
460
461         if (pipefds[!!(io->type == IO_WR)] != -1)
462                 close(pipefds[!!(io->type == IO_WR)]);
463         return FALSE;
464 }
465
466 static bool
467 run_io(struct io *io, const char **argv, const char *dir, enum io_type type)
468 {
469         init_io(io, dir, type);
470         if (!format_argv(io->argv, argv, FORMAT_NONE))
471                 return FALSE;
472         return start_io(io);
473 }
474
475 static int
476 run_io_do(struct io *io)
477 {
478         return start_io(io) && done_io(io);
479 }
480
481 static int
482 run_io_bg(const char **argv)
483 {
484         struct io io = {};
485
486         init_io(&io, NULL, IO_BG);
487         if (!format_argv(io.argv, argv, FORMAT_NONE))
488                 return FALSE;
489         return run_io_do(&io);
490 }
491
492 static bool
493 run_io_fg(const char **argv, const char *dir)
494 {
495         struct io io = {};
496
497         init_io(&io, dir, IO_FG);
498         if (!format_argv(io.argv, argv, FORMAT_NONE))
499                 return FALSE;
500         return run_io_do(&io);
501 }
502
503 static bool
504 run_io_append(const char **argv, enum format_flags flags, int fd)
505 {
506         struct io io = {};
507
508         init_io(&io, NULL, IO_AP);
509         io.pipe = fd;
510         if (format_argv(io.argv, argv, flags))
511                 return run_io_do(&io);
512         close(fd);
513         return FALSE;
514 }
515
516 static bool
517 run_io_rd(struct io *io, const char **argv, enum format_flags flags)
518 {
519         return init_io_rd(io, argv, NULL, flags) && start_io(io);
520 }
521
522 static bool
523 io_eof(struct io *io)
524 {
525         return io->eof;
526 }
527
528 static int
529 io_error(struct io *io)
530 {
531         return io->error;
532 }
533
534 static bool
535 io_strerror(struct io *io)
536 {
537         return strerror(io->error);
538 }
539
540 static bool
541 io_can_read(struct io *io)
542 {
543         struct timeval tv = { 0, 500 };
544         fd_set fds;
545
546         FD_ZERO(&fds);
547         FD_SET(io->pipe, &fds);
548
549         return select(io->pipe + 1, &fds, NULL, NULL, &tv) > 0;
550 }
551
552 static ssize_t
553 io_read(struct io *io, void *buf, size_t bufsize)
554 {
555         do {
556                 ssize_t readsize = read(io->pipe, buf, bufsize);
557
558                 if (readsize < 0 && (errno == EAGAIN || errno == EINTR))
559                         continue;
560                 else if (readsize == -1)
561                         io->error = errno;
562                 else if (readsize == 0)
563                         io->eof = 1;
564                 return readsize;
565         } while (1);
566 }
567
568 static char *
569 io_get(struct io *io, int c, bool can_read)
570 {
571         char *eol;
572         ssize_t readsize;
573
574         if (!io->buf) {
575                 io->buf = io->bufpos = malloc(BUFSIZ);
576                 if (!io->buf)
577                         return NULL;
578                 io->bufalloc = BUFSIZ;
579                 io->bufsize = 0;
580         }
581
582         while (TRUE) {
583                 if (io->bufsize > 0) {
584                         eol = memchr(io->bufpos, c, io->bufsize);
585                         if (eol) {
586                                 char *line = io->bufpos;
587
588                                 *eol = 0;
589                                 io->bufpos = eol + 1;
590                                 io->bufsize -= io->bufpos - line;
591                                 return line;
592                         }
593                 }
594
595                 if (io_eof(io)) {
596                         if (io->bufsize) {
597                                 io->bufpos[io->bufsize] = 0;
598                                 io->bufsize = 0;
599                                 return io->bufpos;
600                         }
601                         return NULL;
602                 }
603
604                 if (!can_read)
605                         return NULL;
606
607                 if (io->bufsize > 0 && io->bufpos > io->buf)
608                         memmove(io->buf, io->bufpos, io->bufsize);
609
610                 io->bufpos = io->buf;
611                 readsize = io_read(io, io->buf + io->bufsize, io->bufalloc - io->bufsize);
612                 if (io_error(io))
613                         return NULL;
614                 io->bufsize += readsize;
615         }
616 }
617
618 static bool
619 io_write(struct io *io, const void *buf, size_t bufsize)
620 {
621         size_t written = 0;
622
623         while (!io_error(io) && written < bufsize) {
624                 ssize_t size;
625
626                 size = write(io->pipe, buf + written, bufsize - written);
627                 if (size < 0 && (errno == EAGAIN || errno == EINTR))
628                         continue;
629                 else if (size == -1)
630                         io->error = errno;
631                 else
632                         written += size;
633         }
634
635         return written == bufsize;
636 }
637
638 static bool
639 run_io_buf(const char **argv, char buf[], size_t bufsize)
640 {
641         struct io io = {};
642         bool error;
643
644         if (!run_io_rd(&io, argv, FORMAT_NONE))
645                 return FALSE;
646
647         io.buf = io.bufpos = buf;
648         io.bufalloc = bufsize;
649         error = !io_get(&io, '\n', TRUE) && io_error(&io);
650         io.buf = NULL;
651
652         return done_io(&io) || error;
653 }
654
655 static int read_properties(struct io *io, const char *separators, int (*read)(char *, size_t, char *, size_t));
656
657 /*
658  * User requests
659  */
660
661 #define REQ_INFO \
662         /* XXX: Keep the view request first and in sync with views[]. */ \
663         REQ_GROUP("View switching") \
664         REQ_(VIEW_MAIN,         "Show main view"), \
665         REQ_(VIEW_DIFF,         "Show diff view"), \
666         REQ_(VIEW_LOG,          "Show log view"), \
667         REQ_(VIEW_TREE,         "Show tree view"), \
668         REQ_(VIEW_BLOB,         "Show blob view"), \
669         REQ_(VIEW_BLAME,        "Show blame view"), \
670         REQ_(VIEW_HELP,         "Show help page"), \
671         REQ_(VIEW_PAGER,        "Show pager view"), \
672         REQ_(VIEW_STATUS,       "Show status view"), \
673         REQ_(VIEW_STAGE,        "Show stage view"), \
674         \
675         REQ_GROUP("View manipulation") \
676         REQ_(ENTER,             "Enter current line and scroll"), \
677         REQ_(NEXT,              "Move to next"), \
678         REQ_(PREVIOUS,          "Move to previous"), \
679         REQ_(PARENT,            "Move to parent"), \
680         REQ_(VIEW_NEXT,         "Move focus to next view"), \
681         REQ_(REFRESH,           "Reload and refresh"), \
682         REQ_(MAXIMIZE,          "Maximize the current view"), \
683         REQ_(VIEW_CLOSE,        "Close the current view"), \
684         REQ_(QUIT,              "Close all views and quit"), \
685         \
686         REQ_GROUP("View specific requests") \
687         REQ_(STATUS_UPDATE,     "Update file status"), \
688         REQ_(STATUS_REVERT,     "Revert file changes"), \
689         REQ_(STATUS_MERGE,      "Merge file using external tool"), \
690         REQ_(STAGE_NEXT,        "Find next chunk to stage"), \
691         \
692         REQ_GROUP("Cursor navigation") \
693         REQ_(MOVE_UP,           "Move cursor one line up"), \
694         REQ_(MOVE_DOWN,         "Move cursor one line down"), \
695         REQ_(MOVE_PAGE_DOWN,    "Move cursor one page down"), \
696         REQ_(MOVE_PAGE_UP,      "Move cursor one page up"), \
697         REQ_(MOVE_FIRST_LINE,   "Move cursor to first line"), \
698         REQ_(MOVE_LAST_LINE,    "Move cursor to last line"), \
699         \
700         REQ_GROUP("Scrolling") \
701         REQ_(SCROLL_LINE_UP,    "Scroll one line up"), \
702         REQ_(SCROLL_LINE_DOWN,  "Scroll one line down"), \
703         REQ_(SCROLL_PAGE_UP,    "Scroll one page up"), \
704         REQ_(SCROLL_PAGE_DOWN,  "Scroll one page down"), \
705         \
706         REQ_GROUP("Searching") \
707         REQ_(SEARCH,            "Search the view"), \
708         REQ_(SEARCH_BACK,       "Search backwards in the view"), \
709         REQ_(FIND_NEXT,         "Find next search match"), \
710         REQ_(FIND_PREV,         "Find previous search match"), \
711         \
712         REQ_GROUP("Option manipulation") \
713         REQ_(TOGGLE_LINENO,     "Toggle line numbers"), \
714         REQ_(TOGGLE_DATE,       "Toggle date display"), \
715         REQ_(TOGGLE_AUTHOR,     "Toggle author display"), \
716         REQ_(TOGGLE_REV_GRAPH,  "Toggle revision graph visualization"), \
717         REQ_(TOGGLE_REFS,       "Toggle reference display (tags/branches)"), \
718         \
719         REQ_GROUP("Misc") \
720         REQ_(PROMPT,            "Bring up the prompt"), \
721         REQ_(SCREEN_REDRAW,     "Redraw the screen"), \
722         REQ_(SHOW_VERSION,      "Show version information"), \
723         REQ_(STOP_LOADING,      "Stop all loading views"), \
724         REQ_(EDIT,              "Open in editor"), \
725         REQ_(NONE,              "Do nothing")
726
727
728 /* User action requests. */
729 enum request {
730 #define REQ_GROUP(help)
731 #define REQ_(req, help) REQ_##req
732
733         /* Offset all requests to avoid conflicts with ncurses getch values. */
734         REQ_OFFSET = KEY_MAX + 1,
735         REQ_INFO
736
737 #undef  REQ_GROUP
738 #undef  REQ_
739 };
740
741 struct request_info {
742         enum request request;
743         const char *name;
744         int namelen;
745         const char *help;
746 };
747
748 static struct request_info req_info[] = {
749 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
750 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
751         REQ_INFO
752 #undef  REQ_GROUP
753 #undef  REQ_
754 };
755
756 static enum request
757 get_request(const char *name)
758 {
759         int namelen = strlen(name);
760         int i;
761
762         for (i = 0; i < ARRAY_SIZE(req_info); i++)
763                 if (req_info[i].namelen == namelen &&
764                     !string_enum_compare(req_info[i].name, name, namelen))
765                         return req_info[i].request;
766
767         return REQ_NONE;
768 }
769
770
771 /*
772  * Options
773  */
774
775 static const char usage[] =
776 "tig " TIG_VERSION " (" __DATE__ ")\n"
777 "\n"
778 "Usage: tig        [options] [revs] [--] [paths]\n"
779 "   or: tig show   [options] [revs] [--] [paths]\n"
780 "   or: tig blame  [rev] path\n"
781 "   or: tig status\n"
782 "   or: tig <      [git command output]\n"
783 "\n"
784 "Options:\n"
785 "  -v, --version   Show version and exit\n"
786 "  -h, --help      Show help message and exit";
787
788 /* Option and state variables. */
789 static bool opt_date                    = TRUE;
790 static bool opt_author                  = TRUE;
791 static bool opt_line_number             = FALSE;
792 static bool opt_line_graphics           = TRUE;
793 static bool opt_rev_graph               = FALSE;
794 static bool opt_show_refs               = TRUE;
795 static int opt_num_interval             = NUMBER_INTERVAL;
796 static int opt_tab_size                 = TAB_SIZE;
797 static int opt_author_cols              = AUTHOR_COLS-1;
798 static char opt_path[SIZEOF_STR]        = "";
799 static char opt_file[SIZEOF_STR]        = "";
800 static char opt_ref[SIZEOF_REF]         = "";
801 static char opt_head[SIZEOF_REF]        = "";
802 static char opt_head_rev[SIZEOF_REV]    = "";
803 static char opt_remote[SIZEOF_REF]      = "";
804 static char opt_encoding[20]            = "UTF-8";
805 static bool opt_utf8                    = TRUE;
806 static char opt_codeset[20]             = "UTF-8";
807 static iconv_t opt_iconv                = ICONV_NONE;
808 static char opt_search[SIZEOF_STR]      = "";
809 static char opt_cdup[SIZEOF_STR]        = "";
810 static char opt_prefix[SIZEOF_STR]      = "";
811 static char opt_git_dir[SIZEOF_STR]     = "";
812 static signed char opt_is_inside_work_tree      = -1; /* set to TRUE or FALSE */
813 static char opt_editor[SIZEOF_STR]      = "";
814 static FILE *opt_tty                    = NULL;
815
816 #define is_initial_commit()     (!*opt_head_rev)
817 #define is_head_commit(rev)     (!strcmp((rev), "HEAD") || !strcmp(opt_head_rev, (rev)))
818
819 static enum request
820 parse_options(int argc, const char *argv[], const char ***run_argv)
821 {
822         enum request request = REQ_VIEW_MAIN;
823         const char *subcommand;
824         bool seen_dashdash = FALSE;
825         /* XXX: This is vulnerable to the user overriding options
826          * required for the main view parser. */
827         static const char *custom_argv[SIZEOF_ARG] = {
828                 "git", "log", "--no-color", "--pretty=raw", "--parents",
829                         "--topo-order", NULL
830         };
831         int i, j = 6;
832
833         if (!isatty(STDIN_FILENO))
834                 return REQ_VIEW_PAGER;
835
836         if (argc <= 1)
837                 return REQ_VIEW_MAIN;
838
839         subcommand = argv[1];
840         if (!strcmp(subcommand, "status") || !strcmp(subcommand, "-S")) {
841                 if (!strcmp(subcommand, "-S"))
842                         warn("`-S' has been deprecated; use `tig status' instead");
843                 if (argc > 2)
844                         warn("ignoring arguments after `%s'", subcommand);
845                 return REQ_VIEW_STATUS;
846
847         } else if (!strcmp(subcommand, "blame")) {
848                 if (argc <= 2 || argc > 4)
849                         die("invalid number of options to blame\n\n%s", usage);
850
851                 i = 2;
852                 if (argc == 4) {
853                         string_ncopy(opt_ref, argv[i], strlen(argv[i]));
854                         i++;
855                 }
856
857                 string_ncopy(opt_file, argv[i], strlen(argv[i]));
858                 return REQ_VIEW_BLAME;
859
860         } else if (!strcmp(subcommand, "show")) {
861                 request = REQ_VIEW_DIFF;
862
863         } else if (!strcmp(subcommand, "log") || !strcmp(subcommand, "diff")) {
864                 request = subcommand[0] == 'l' ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
865                 warn("`tig %s' has been deprecated", subcommand);
866
867         } else {
868                 subcommand = NULL;
869         }
870
871         if (subcommand) {
872                 custom_argv[1] = subcommand;
873                 j = 2;
874         }
875
876         for (i = 1 + !!subcommand; i < argc; i++) {
877                 const char *opt = argv[i];
878
879                 if (seen_dashdash || !strcmp(opt, "--")) {
880                         seen_dashdash = TRUE;
881
882                 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
883                         printf("tig version %s\n", TIG_VERSION);
884                         return REQ_NONE;
885
886                 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
887                         printf("%s\n", usage);
888                         return REQ_NONE;
889                 }
890
891                 custom_argv[j++] = opt;
892                 if (j >= ARRAY_SIZE(custom_argv))
893                         die("command too long");
894         }
895
896         custom_argv[j] = NULL;
897         *run_argv = custom_argv;
898
899         return request;
900 }
901
902
903 /*
904  * Line-oriented content detection.
905  */
906
907 #define LINE_INFO \
908 LINE(DIFF_HEADER,  "diff --git ",       COLOR_YELLOW,   COLOR_DEFAULT,  0), \
909 LINE(DIFF_CHUNK,   "@@",                COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
910 LINE(DIFF_ADD,     "+",                 COLOR_GREEN,    COLOR_DEFAULT,  0), \
911 LINE(DIFF_DEL,     "-",                 COLOR_RED,      COLOR_DEFAULT,  0), \
912 LINE(DIFF_INDEX,        "index ",         COLOR_BLUE,   COLOR_DEFAULT,  0), \
913 LINE(DIFF_OLDMODE,      "old file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
914 LINE(DIFF_NEWMODE,      "new file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
915 LINE(DIFF_COPY_FROM,    "copy from",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
916 LINE(DIFF_COPY_TO,      "copy to",        COLOR_YELLOW, COLOR_DEFAULT,  0), \
917 LINE(DIFF_RENAME_FROM,  "rename from",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
918 LINE(DIFF_RENAME_TO,    "rename to",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
919 LINE(DIFF_SIMILARITY,   "similarity ",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
920 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
921 LINE(DIFF_TREE,         "diff-tree ",     COLOR_BLUE,   COLOR_DEFAULT,  0), \
922 LINE(PP_AUTHOR,    "Author: ",          COLOR_CYAN,     COLOR_DEFAULT,  0), \
923 LINE(PP_COMMIT,    "Commit: ",          COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
924 LINE(PP_MERGE,     "Merge: ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
925 LINE(PP_DATE,      "Date:   ",          COLOR_YELLOW,   COLOR_DEFAULT,  0), \
926 LINE(PP_ADATE,     "AuthorDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
927 LINE(PP_CDATE,     "CommitDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
928 LINE(PP_REFS,      "Refs: ",            COLOR_RED,      COLOR_DEFAULT,  0), \
929 LINE(COMMIT,       "commit ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
930 LINE(PARENT,       "parent ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
931 LINE(TREE,         "tree ",             COLOR_BLUE,     COLOR_DEFAULT,  0), \
932 LINE(AUTHOR,       "author ",           COLOR_CYAN,     COLOR_DEFAULT,  0), \
933 LINE(COMMITTER,    "committer ",        COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
934 LINE(SIGNOFF,      "    Signed-off-by", COLOR_YELLOW,   COLOR_DEFAULT,  0), \
935 LINE(ACKED,        "    Acked-by",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
936 LINE(DEFAULT,      "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
937 LINE(CURSOR,       "",                  COLOR_WHITE,    COLOR_GREEN,    A_BOLD), \
938 LINE(STATUS,       "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
939 LINE(DELIMITER,    "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
940 LINE(DATE,         "",                  COLOR_BLUE,     COLOR_DEFAULT,  0), \
941 LINE(LINE_NUMBER,  "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
942 LINE(TITLE_BLUR,   "",                  COLOR_WHITE,    COLOR_BLUE,     0), \
943 LINE(TITLE_FOCUS,  "",                  COLOR_WHITE,    COLOR_BLUE,     A_BOLD), \
944 LINE(MAIN_AUTHOR,  "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
945 LINE(MAIN_COMMIT,  "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
946 LINE(MAIN_TAG,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  A_BOLD), \
947 LINE(MAIN_LOCAL_TAG,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
948 LINE(MAIN_REMOTE,  "",                  COLOR_YELLOW,   COLOR_DEFAULT,  0), \
949 LINE(MAIN_TRACKED, "",                  COLOR_YELLOW,   COLOR_DEFAULT,  A_BOLD), \
950 LINE(MAIN_REF,     "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
951 LINE(MAIN_HEAD,    "",                  COLOR_CYAN,     COLOR_DEFAULT,  A_BOLD), \
952 LINE(MAIN_REVGRAPH,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
953 LINE(TREE_PARENT,  "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_BOLD), \
954 LINE(TREE_MODE,    "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
955 LINE(TREE_DIR,     "",                  COLOR_YELLOW,   COLOR_DEFAULT,  A_NORMAL), \
956 LINE(TREE_FILE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
957 LINE(STAT_HEAD,    "",                  COLOR_YELLOW,   COLOR_DEFAULT,  0), \
958 LINE(STAT_SECTION, "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
959 LINE(STAT_NONE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
960 LINE(STAT_STAGED,  "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
961 LINE(STAT_UNSTAGED,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
962 LINE(STAT_UNTRACKED,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
963 LINE(BLAME_ID,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0)
964
965 enum line_type {
966 #define LINE(type, line, fg, bg, attr) \
967         LINE_##type
968         LINE_INFO,
969         LINE_NONE
970 #undef  LINE
971 };
972
973 struct line_info {
974         const char *name;       /* Option name. */
975         int namelen;            /* Size of option name. */
976         const char *line;       /* The start of line to match. */
977         int linelen;            /* Size of string to match. */
978         int fg, bg, attr;       /* Color and text attributes for the lines. */
979 };
980
981 static struct line_info line_info[] = {
982 #define LINE(type, line, fg, bg, attr) \
983         { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
984         LINE_INFO
985 #undef  LINE
986 };
987
988 static enum line_type
989 get_line_type(const char *line)
990 {
991         int linelen = strlen(line);
992         enum line_type type;
993
994         for (type = 0; type < ARRAY_SIZE(line_info); type++)
995                 /* Case insensitive search matches Signed-off-by lines better. */
996                 if (linelen >= line_info[type].linelen &&
997                     !strncasecmp(line_info[type].line, line, line_info[type].linelen))
998                         return type;
999
1000         return LINE_DEFAULT;
1001 }
1002
1003 static inline int
1004 get_line_attr(enum line_type type)
1005 {
1006         assert(type < ARRAY_SIZE(line_info));
1007         return COLOR_PAIR(type) | line_info[type].attr;
1008 }
1009
1010 static struct line_info *
1011 get_line_info(const char *name)
1012 {
1013         size_t namelen = strlen(name);
1014         enum line_type type;
1015
1016         for (type = 0; type < ARRAY_SIZE(line_info); type++)
1017                 if (namelen == line_info[type].namelen &&
1018                     !string_enum_compare(line_info[type].name, name, namelen))
1019                         return &line_info[type];
1020
1021         return NULL;
1022 }
1023
1024 static void
1025 init_colors(void)
1026 {
1027         int default_bg = line_info[LINE_DEFAULT].bg;
1028         int default_fg = line_info[LINE_DEFAULT].fg;
1029         enum line_type type;
1030
1031         start_color();
1032
1033         if (assume_default_colors(default_fg, default_bg) == ERR) {
1034                 default_bg = COLOR_BLACK;
1035                 default_fg = COLOR_WHITE;
1036         }
1037
1038         for (type = 0; type < ARRAY_SIZE(line_info); type++) {
1039                 struct line_info *info = &line_info[type];
1040                 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
1041                 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
1042
1043                 init_pair(type, fg, bg);
1044         }
1045 }
1046
1047 struct line {
1048         enum line_type type;
1049
1050         /* State flags */
1051         unsigned int selected:1;
1052         unsigned int dirty:1;
1053         unsigned int cleareol:1;
1054
1055         void *data;             /* User data */
1056 };
1057
1058
1059 /*
1060  * Keys
1061  */
1062
1063 struct keybinding {
1064         int alias;
1065         enum request request;
1066 };
1067
1068 static struct keybinding default_keybindings[] = {
1069         /* View switching */
1070         { 'm',          REQ_VIEW_MAIN },
1071         { 'd',          REQ_VIEW_DIFF },
1072         { 'l',          REQ_VIEW_LOG },
1073         { 't',          REQ_VIEW_TREE },
1074         { 'f',          REQ_VIEW_BLOB },
1075         { 'B',          REQ_VIEW_BLAME },
1076         { 'p',          REQ_VIEW_PAGER },
1077         { 'h',          REQ_VIEW_HELP },
1078         { 'S',          REQ_VIEW_STATUS },
1079         { 'c',          REQ_VIEW_STAGE },
1080
1081         /* View manipulation */
1082         { 'q',          REQ_VIEW_CLOSE },
1083         { KEY_TAB,      REQ_VIEW_NEXT },
1084         { KEY_RETURN,   REQ_ENTER },
1085         { KEY_UP,       REQ_PREVIOUS },
1086         { KEY_DOWN,     REQ_NEXT },
1087         { 'R',          REQ_REFRESH },
1088         { KEY_F(5),     REQ_REFRESH },
1089         { 'O',          REQ_MAXIMIZE },
1090
1091         /* Cursor navigation */
1092         { 'k',          REQ_MOVE_UP },
1093         { 'j',          REQ_MOVE_DOWN },
1094         { KEY_HOME,     REQ_MOVE_FIRST_LINE },
1095         { KEY_END,      REQ_MOVE_LAST_LINE },
1096         { KEY_NPAGE,    REQ_MOVE_PAGE_DOWN },
1097         { ' ',          REQ_MOVE_PAGE_DOWN },
1098         { KEY_PPAGE,    REQ_MOVE_PAGE_UP },
1099         { 'b',          REQ_MOVE_PAGE_UP },
1100         { '-',          REQ_MOVE_PAGE_UP },
1101
1102         /* Scrolling */
1103         { KEY_IC,       REQ_SCROLL_LINE_UP },
1104         { KEY_DC,       REQ_SCROLL_LINE_DOWN },
1105         { 'w',          REQ_SCROLL_PAGE_UP },
1106         { 's',          REQ_SCROLL_PAGE_DOWN },
1107
1108         /* Searching */
1109         { '/',          REQ_SEARCH },
1110         { '?',          REQ_SEARCH_BACK },
1111         { 'n',          REQ_FIND_NEXT },
1112         { 'N',          REQ_FIND_PREV },
1113
1114         /* Misc */
1115         { 'Q',          REQ_QUIT },
1116         { 'z',          REQ_STOP_LOADING },
1117         { 'v',          REQ_SHOW_VERSION },
1118         { 'r',          REQ_SCREEN_REDRAW },
1119         { '.',          REQ_TOGGLE_LINENO },
1120         { 'D',          REQ_TOGGLE_DATE },
1121         { 'A',          REQ_TOGGLE_AUTHOR },
1122         { 'g',          REQ_TOGGLE_REV_GRAPH },
1123         { 'F',          REQ_TOGGLE_REFS },
1124         { ':',          REQ_PROMPT },
1125         { 'u',          REQ_STATUS_UPDATE },
1126         { '!',          REQ_STATUS_REVERT },
1127         { 'M',          REQ_STATUS_MERGE },
1128         { '@',          REQ_STAGE_NEXT },
1129         { ',',          REQ_PARENT },
1130         { 'e',          REQ_EDIT },
1131 };
1132
1133 #define KEYMAP_INFO \
1134         KEYMAP_(GENERIC), \
1135         KEYMAP_(MAIN), \
1136         KEYMAP_(DIFF), \
1137         KEYMAP_(LOG), \
1138         KEYMAP_(TREE), \
1139         KEYMAP_(BLOB), \
1140         KEYMAP_(BLAME), \
1141         KEYMAP_(PAGER), \
1142         KEYMAP_(HELP), \
1143         KEYMAP_(STATUS), \
1144         KEYMAP_(STAGE)
1145
1146 enum keymap {
1147 #define KEYMAP_(name) KEYMAP_##name
1148         KEYMAP_INFO
1149 #undef  KEYMAP_
1150 };
1151
1152 static struct int_map keymap_table[] = {
1153 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
1154         KEYMAP_INFO
1155 #undef  KEYMAP_
1156 };
1157
1158 #define set_keymap(map, name) \
1159         set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
1160
1161 struct keybinding_table {
1162         struct keybinding *data;
1163         size_t size;
1164 };
1165
1166 static struct keybinding_table keybindings[ARRAY_SIZE(keymap_table)];
1167
1168 static void
1169 add_keybinding(enum keymap keymap, enum request request, int key)
1170 {
1171         struct keybinding_table *table = &keybindings[keymap];
1172
1173         table->data = realloc(table->data, (table->size + 1) * sizeof(*table->data));
1174         if (!table->data)
1175                 die("Failed to allocate keybinding");
1176         table->data[table->size].alias = key;
1177         table->data[table->size++].request = request;
1178 }
1179
1180 /* Looks for a key binding first in the given map, then in the generic map, and
1181  * lastly in the default keybindings. */
1182 static enum request
1183 get_keybinding(enum keymap keymap, int key)
1184 {
1185         size_t i;
1186
1187         for (i = 0; i < keybindings[keymap].size; i++)
1188                 if (keybindings[keymap].data[i].alias == key)
1189                         return keybindings[keymap].data[i].request;
1190
1191         for (i = 0; i < keybindings[KEYMAP_GENERIC].size; i++)
1192                 if (keybindings[KEYMAP_GENERIC].data[i].alias == key)
1193                         return keybindings[KEYMAP_GENERIC].data[i].request;
1194
1195         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
1196                 if (default_keybindings[i].alias == key)
1197                         return default_keybindings[i].request;
1198
1199         return (enum request) key;
1200 }
1201
1202
1203 struct key {
1204         const char *name;
1205         int value;
1206 };
1207
1208 static struct key key_table[] = {
1209         { "Enter",      KEY_RETURN },
1210         { "Space",      ' ' },
1211         { "Backspace",  KEY_BACKSPACE },
1212         { "Tab",        KEY_TAB },
1213         { "Escape",     KEY_ESC },
1214         { "Left",       KEY_LEFT },
1215         { "Right",      KEY_RIGHT },
1216         { "Up",         KEY_UP },
1217         { "Down",       KEY_DOWN },
1218         { "Insert",     KEY_IC },
1219         { "Delete",     KEY_DC },
1220         { "Hash",       '#' },
1221         { "Home",       KEY_HOME },
1222         { "End",        KEY_END },
1223         { "PageUp",     KEY_PPAGE },
1224         { "PageDown",   KEY_NPAGE },
1225         { "F1",         KEY_F(1) },
1226         { "F2",         KEY_F(2) },
1227         { "F3",         KEY_F(3) },
1228         { "F4",         KEY_F(4) },
1229         { "F5",         KEY_F(5) },
1230         { "F6",         KEY_F(6) },
1231         { "F7",         KEY_F(7) },
1232         { "F8",         KEY_F(8) },
1233         { "F9",         KEY_F(9) },
1234         { "F10",        KEY_F(10) },
1235         { "F11",        KEY_F(11) },
1236         { "F12",        KEY_F(12) },
1237 };
1238
1239 static int
1240 get_key_value(const char *name)
1241 {
1242         int i;
1243
1244         for (i = 0; i < ARRAY_SIZE(key_table); i++)
1245                 if (!strcasecmp(key_table[i].name, name))
1246                         return key_table[i].value;
1247
1248         if (strlen(name) == 1 && isprint(*name))
1249                 return (int) *name;
1250
1251         return ERR;
1252 }
1253
1254 static const char *
1255 get_key_name(int key_value)
1256 {
1257         static char key_char[] = "'X'";
1258         const char *seq = NULL;
1259         int key;
1260
1261         for (key = 0; key < ARRAY_SIZE(key_table); key++)
1262                 if (key_table[key].value == key_value)
1263                         seq = key_table[key].name;
1264
1265         if (seq == NULL &&
1266             key_value < 127 &&
1267             isprint(key_value)) {
1268                 key_char[1] = (char) key_value;
1269                 seq = key_char;
1270         }
1271
1272         return seq ? seq : "(no key)";
1273 }
1274
1275 static const char *
1276 get_key(enum request request)
1277 {
1278         static char buf[BUFSIZ];
1279         size_t pos = 0;
1280         char *sep = "";
1281         int i;
1282
1283         buf[pos] = 0;
1284
1285         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
1286                 struct keybinding *keybinding = &default_keybindings[i];
1287
1288                 if (keybinding->request != request)
1289                         continue;
1290
1291                 if (!string_format_from(buf, &pos, "%s%s", sep,
1292                                         get_key_name(keybinding->alias)))
1293                         return "Too many keybindings!";
1294                 sep = ", ";
1295         }
1296
1297         return buf;
1298 }
1299
1300 struct run_request {
1301         enum keymap keymap;
1302         int key;
1303         const char *argv[SIZEOF_ARG];
1304 };
1305
1306 static struct run_request *run_request;
1307 static size_t run_requests;
1308
1309 static enum request
1310 add_run_request(enum keymap keymap, int key, int argc, const char **argv)
1311 {
1312         struct run_request *req;
1313
1314         if (argc >= ARRAY_SIZE(req->argv) - 1)
1315                 return REQ_NONE;
1316
1317         req = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
1318         if (!req)
1319                 return REQ_NONE;
1320
1321         run_request = req;
1322         req = &run_request[run_requests];
1323         req->keymap = keymap;
1324         req->key = key;
1325         req->argv[0] = NULL;
1326
1327         if (!format_argv(req->argv, argv, FORMAT_NONE))
1328                 return REQ_NONE;
1329
1330         return REQ_NONE + ++run_requests;
1331 }
1332
1333 static struct run_request *
1334 get_run_request(enum request request)
1335 {
1336         if (request <= REQ_NONE)
1337                 return NULL;
1338         return &run_request[request - REQ_NONE - 1];
1339 }
1340
1341 static void
1342 add_builtin_run_requests(void)
1343 {
1344         const char *cherry_pick[] = { "git", "cherry-pick", "%(commit)", NULL };
1345         const char *gc[] = { "git", "gc", NULL };
1346         struct {
1347                 enum keymap keymap;
1348                 int key;
1349                 int argc;
1350                 const char **argv;
1351         } reqs[] = {
1352                 { KEYMAP_MAIN,    'C', ARRAY_SIZE(cherry_pick) - 1, cherry_pick },
1353                 { KEYMAP_GENERIC, 'G', ARRAY_SIZE(gc) - 1, gc },
1354         };
1355         int i;
1356
1357         for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1358                 enum request req;
1359
1360                 req = add_run_request(reqs[i].keymap, reqs[i].key, reqs[i].argc, reqs[i].argv);
1361                 if (req != REQ_NONE)
1362                         add_keybinding(reqs[i].keymap, req, reqs[i].key);
1363         }
1364 }
1365
1366 /*
1367  * User config file handling.
1368  */
1369
1370 static struct int_map color_map[] = {
1371 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
1372         COLOR_MAP(DEFAULT),
1373         COLOR_MAP(BLACK),
1374         COLOR_MAP(BLUE),
1375         COLOR_MAP(CYAN),
1376         COLOR_MAP(GREEN),
1377         COLOR_MAP(MAGENTA),
1378         COLOR_MAP(RED),
1379         COLOR_MAP(WHITE),
1380         COLOR_MAP(YELLOW),
1381 };
1382
1383 #define set_color(color, name) \
1384         set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
1385
1386 static struct int_map attr_map[] = {
1387 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
1388         ATTR_MAP(NORMAL),
1389         ATTR_MAP(BLINK),
1390         ATTR_MAP(BOLD),
1391         ATTR_MAP(DIM),
1392         ATTR_MAP(REVERSE),
1393         ATTR_MAP(STANDOUT),
1394         ATTR_MAP(UNDERLINE),
1395 };
1396
1397 #define set_attribute(attr, name) \
1398         set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
1399
1400 static int   config_lineno;
1401 static bool  config_errors;
1402 static const char *config_msg;
1403
1404 /* Wants: object fgcolor bgcolor [attr] */
1405 static int
1406 option_color_command(int argc, const char *argv[])
1407 {
1408         struct line_info *info;
1409
1410         if (argc != 3 && argc != 4) {
1411                 config_msg = "Wrong number of arguments given to color command";
1412                 return ERR;
1413         }
1414
1415         info = get_line_info(argv[0]);
1416         if (!info) {
1417                 if (!string_enum_compare(argv[0], "main-delim", strlen("main-delim"))) {
1418                         info = get_line_info("delimiter");
1419
1420                 } else if (!string_enum_compare(argv[0], "main-date", strlen("main-date"))) {
1421                         info = get_line_info("date");
1422
1423                 } else {
1424                         config_msg = "Unknown color name";
1425                         return ERR;
1426                 }
1427         }
1428
1429         if (set_color(&info->fg, argv[1]) == ERR ||
1430             set_color(&info->bg, argv[2]) == ERR) {
1431                 config_msg = "Unknown color";
1432                 return ERR;
1433         }
1434
1435         if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
1436                 config_msg = "Unknown attribute";
1437                 return ERR;
1438         }
1439
1440         return OK;
1441 }
1442
1443 static bool parse_bool(const char *s)
1444 {
1445         return (!strcmp(s, "1") || !strcmp(s, "true") ||
1446                 !strcmp(s, "yes")) ? TRUE : FALSE;
1447 }
1448
1449 static int
1450 parse_int(const char *s, int default_value, int min, int max)
1451 {
1452         int value = atoi(s);
1453
1454         return (value < min || value > max) ? default_value : value;
1455 }
1456
1457 /* Wants: name = value */
1458 static int
1459 option_set_command(int argc, const char *argv[])
1460 {
1461         if (argc != 3) {
1462                 config_msg = "Wrong number of arguments given to set command";
1463                 return ERR;
1464         }
1465
1466         if (strcmp(argv[1], "=")) {
1467                 config_msg = "No value assigned";
1468                 return ERR;
1469         }
1470
1471         if (!strcmp(argv[0], "show-author")) {
1472                 opt_author = parse_bool(argv[2]);
1473                 return OK;
1474         }
1475
1476         if (!strcmp(argv[0], "show-date")) {
1477                 opt_date = parse_bool(argv[2]);
1478                 return OK;
1479         }
1480
1481         if (!strcmp(argv[0], "show-rev-graph")) {
1482                 opt_rev_graph = parse_bool(argv[2]);
1483                 return OK;
1484         }
1485
1486         if (!strcmp(argv[0], "show-refs")) {
1487                 opt_show_refs = parse_bool(argv[2]);
1488                 return OK;
1489         }
1490
1491         if (!strcmp(argv[0], "show-line-numbers")) {
1492                 opt_line_number = parse_bool(argv[2]);
1493                 return OK;
1494         }
1495
1496         if (!strcmp(argv[0], "line-graphics")) {
1497                 opt_line_graphics = parse_bool(argv[2]);
1498                 return OK;
1499         }
1500
1501         if (!strcmp(argv[0], "line-number-interval")) {
1502                 opt_num_interval = parse_int(argv[2], opt_num_interval, 1, 1024);
1503                 return OK;
1504         }
1505
1506         if (!strcmp(argv[0], "author-width")) {
1507                 opt_author_cols = parse_int(argv[2], opt_author_cols, 0, 1024);
1508                 return OK;
1509         }
1510
1511         if (!strcmp(argv[0], "tab-size")) {
1512                 opt_tab_size = parse_int(argv[2], opt_tab_size, 1, 1024);
1513                 return OK;
1514         }
1515
1516         if (!strcmp(argv[0], "commit-encoding")) {
1517                 const char *arg = argv[2];
1518                 int arglen = strlen(arg);
1519
1520                 switch (arg[0]) {
1521                 case '"':
1522                 case '\'':
1523                         if (arglen == 1 || arg[arglen - 1] != arg[0]) {
1524                                 config_msg = "Unmatched quotation";
1525                                 return ERR;
1526                         }
1527                         arg += 1; arglen -= 2;
1528                 default:
1529                         string_ncopy(opt_encoding, arg, strlen(arg));
1530                         return OK;
1531                 }
1532         }
1533
1534         config_msg = "Unknown variable name";
1535         return ERR;
1536 }
1537
1538 /* Wants: mode request key */
1539 static int
1540 option_bind_command(int argc, const char *argv[])
1541 {
1542         enum request request;
1543         int keymap;
1544         int key;
1545
1546         if (argc < 3) {
1547                 config_msg = "Wrong number of arguments given to bind command";
1548                 return ERR;
1549         }
1550
1551         if (set_keymap(&keymap, argv[0]) == ERR) {
1552                 config_msg = "Unknown key map";
1553                 return ERR;
1554         }
1555
1556         key = get_key_value(argv[1]);
1557         if (key == ERR) {
1558                 config_msg = "Unknown key";
1559                 return ERR;
1560         }
1561
1562         request = get_request(argv[2]);
1563         if (request == REQ_NONE) {
1564                 struct {
1565                         const char *name;
1566                         enum request request;
1567                 } obsolete[] = {
1568                         { "cherry-pick",        REQ_NONE },
1569                         { "screen-resize",      REQ_NONE },
1570                         { "tree-parent",        REQ_PARENT },
1571                 };
1572                 size_t namelen = strlen(argv[2]);
1573                 int i;
1574
1575                 for (i = 0; i < ARRAY_SIZE(obsolete); i++) {
1576                         if (namelen != strlen(obsolete[i].name) ||
1577                             string_enum_compare(obsolete[i].name, argv[2], namelen))
1578                                 continue;
1579                         if (obsolete[i].request != REQ_NONE)
1580                                 add_keybinding(keymap, obsolete[i].request, key);
1581                         config_msg = "Obsolete request name";
1582                         return ERR;
1583                 }
1584         }
1585         if (request == REQ_NONE && *argv[2]++ == '!')
1586                 request = add_run_request(keymap, key, argc - 2, argv + 2);
1587         if (request == REQ_NONE) {
1588                 config_msg = "Unknown request name";
1589                 return ERR;
1590         }
1591
1592         add_keybinding(keymap, request, key);
1593
1594         return OK;
1595 }
1596
1597 static int
1598 set_option(const char *opt, char *value)
1599 {
1600         const char *argv[SIZEOF_ARG];
1601         int argc = 0;
1602
1603         if (!argv_from_string(argv, &argc, value)) {
1604                 config_msg = "Too many option arguments";
1605                 return ERR;
1606         }
1607
1608         if (!strcmp(opt, "color"))
1609                 return option_color_command(argc, argv);
1610
1611         if (!strcmp(opt, "set"))
1612                 return option_set_command(argc, argv);
1613
1614         if (!strcmp(opt, "bind"))
1615                 return option_bind_command(argc, argv);
1616
1617         config_msg = "Unknown option command";
1618         return ERR;
1619 }
1620
1621 static int
1622 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1623 {
1624         int status = OK;
1625
1626         config_lineno++;
1627         config_msg = "Internal error";
1628
1629         /* Check for comment markers, since read_properties() will
1630          * only ensure opt and value are split at first " \t". */
1631         optlen = strcspn(opt, "#");
1632         if (optlen == 0)
1633                 return OK;
1634
1635         if (opt[optlen] != 0) {
1636                 config_msg = "No option value";
1637                 status = ERR;
1638
1639         }  else {
1640                 /* Look for comment endings in the value. */
1641                 size_t len = strcspn(value, "#");
1642
1643                 if (len < valuelen) {
1644                         valuelen = len;
1645                         value[valuelen] = 0;
1646                 }
1647
1648                 status = set_option(opt, value);
1649         }
1650
1651         if (status == ERR) {
1652                 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1653                         config_lineno, (int) optlen, opt, config_msg);
1654                 config_errors = TRUE;
1655         }
1656
1657         /* Always keep going if errors are encountered. */
1658         return OK;
1659 }
1660
1661 static void
1662 load_option_file(const char *path)
1663 {
1664         struct io io = {};
1665
1666         /* It's ok that the file doesn't exist. */
1667         if (!io_open(&io, path))
1668                 return;
1669
1670         config_lineno = 0;
1671         config_errors = FALSE;
1672
1673         if (read_properties(&io, " \t", read_option) == ERR ||
1674             config_errors == TRUE)
1675                 fprintf(stderr, "Errors while loading %s.\n", path);
1676 }
1677
1678 static int
1679 load_options(void)
1680 {
1681         const char *home = getenv("HOME");
1682         const char *tigrc_user = getenv("TIGRC_USER");
1683         const char *tigrc_system = getenv("TIGRC_SYSTEM");
1684         char buf[SIZEOF_STR];
1685
1686         add_builtin_run_requests();
1687
1688         if (!tigrc_system) {
1689                 if (!string_format(buf, "%s/tigrc", SYSCONFDIR))
1690                         return ERR;
1691                 tigrc_system = buf;
1692         }
1693         load_option_file(tigrc_system);
1694
1695         if (!tigrc_user) {
1696                 if (!home || !string_format(buf, "%s/.tigrc", home))
1697                         return ERR;
1698                 tigrc_user = buf;
1699         }
1700         load_option_file(tigrc_user);
1701
1702         return OK;
1703 }
1704
1705
1706 /*
1707  * The viewer
1708  */
1709
1710 struct view;
1711 struct view_ops;
1712
1713 /* The display array of active views and the index of the current view. */
1714 static struct view *display[2];
1715 static unsigned int current_view;
1716
1717 /* Reading from the prompt? */
1718 static bool input_mode = FALSE;
1719
1720 #define foreach_displayed_view(view, i) \
1721         for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1722
1723 #define displayed_views()       (display[1] != NULL ? 2 : 1)
1724
1725 /* Current head and commit ID */
1726 static char ref_blob[SIZEOF_REF]        = "";
1727 static char ref_commit[SIZEOF_REF]      = "HEAD";
1728 static char ref_head[SIZEOF_REF]        = "HEAD";
1729
1730 struct view {
1731         const char *name;       /* View name */
1732         const char *cmd_env;    /* Command line set via environment */
1733         const char *id;         /* Points to either of ref_{head,commit,blob} */
1734
1735         struct view_ops *ops;   /* View operations */
1736
1737         enum keymap keymap;     /* What keymap does this view have */
1738         bool git_dir;           /* Whether the view requires a git directory. */
1739
1740         char ref[SIZEOF_REF];   /* Hovered commit reference */
1741         char vid[SIZEOF_REF];   /* View ID. Set to id member when updating. */
1742
1743         int height, width;      /* The width and height of the main window */
1744         WINDOW *win;            /* The main window */
1745         WINDOW *title;          /* The title window living below the main window */
1746
1747         /* Navigation */
1748         unsigned long offset;   /* Offset of the window top */
1749         unsigned long lineno;   /* Current line number */
1750         unsigned long p_offset; /* Previous offset of the window top */
1751         unsigned long p_lineno; /* Previous current line number */
1752         bool p_restore;         /* Should the previous position be restored. */
1753
1754         /* Searching */
1755         char grep[SIZEOF_STR];  /* Search string */
1756         regex_t *regex;         /* Pre-compiled regex */
1757
1758         /* If non-NULL, points to the view that opened this view. If this view
1759          * is closed tig will switch back to the parent view. */
1760         struct view *parent;
1761
1762         /* Buffering */
1763         size_t lines;           /* Total number of lines */
1764         struct line *line;      /* Line index */
1765         size_t line_alloc;      /* Total number of allocated lines */
1766         unsigned int digits;    /* Number of digits in the lines member. */
1767
1768         /* Drawing */
1769         struct line *curline;   /* Line currently being drawn. */
1770         enum line_type curtype; /* Attribute currently used for drawing. */
1771         unsigned long col;      /* Column when drawing. */
1772
1773         /* Loading */
1774         struct io io;
1775         struct io *pipe;
1776         time_t start_time;
1777         time_t update_secs;
1778 };
1779
1780 struct view_ops {
1781         /* What type of content being displayed. Used in the title bar. */
1782         const char *type;
1783         /* Default command arguments. */
1784         const char **argv;
1785         /* Open and reads in all view content. */
1786         bool (*open)(struct view *view);
1787         /* Read one line; updates view->line. */
1788         bool (*read)(struct view *view, char *data);
1789         /* Draw one line; @lineno must be < view->height. */
1790         bool (*draw)(struct view *view, struct line *line, unsigned int lineno);
1791         /* Depending on view handle a special requests. */
1792         enum request (*request)(struct view *view, enum request request, struct line *line);
1793         /* Search for regex in a line. */
1794         bool (*grep)(struct view *view, struct line *line);
1795         /* Select line */
1796         void (*select)(struct view *view, struct line *line);
1797 };
1798
1799 static struct view_ops blame_ops;
1800 static struct view_ops blob_ops;
1801 static struct view_ops diff_ops;
1802 static struct view_ops help_ops;
1803 static struct view_ops log_ops;
1804 static struct view_ops main_ops;
1805 static struct view_ops pager_ops;
1806 static struct view_ops stage_ops;
1807 static struct view_ops status_ops;
1808 static struct view_ops tree_ops;
1809
1810 #define VIEW_STR(name, env, ref, ops, map, git) \
1811         { name, #env, ref, ops, map, git }
1812
1813 #define VIEW_(id, name, ops, git, ref) \
1814         VIEW_STR(name, TIG_##id##_CMD, ref, ops, KEYMAP_##id, git)
1815
1816
1817 static struct view views[] = {
1818         VIEW_(MAIN,   "main",   &main_ops,   TRUE,  ref_head),
1819         VIEW_(DIFF,   "diff",   &diff_ops,   TRUE,  ref_commit),
1820         VIEW_(LOG,    "log",    &log_ops,    TRUE,  ref_head),
1821         VIEW_(TREE,   "tree",   &tree_ops,   TRUE,  ref_commit),
1822         VIEW_(BLOB,   "blob",   &blob_ops,   TRUE,  ref_blob),
1823         VIEW_(BLAME,  "blame",  &blame_ops,  TRUE,  ref_commit),
1824         VIEW_(HELP,   "help",   &help_ops,   FALSE, ""),
1825         VIEW_(PAGER,  "pager",  &pager_ops,  FALSE, "stdin"),
1826         VIEW_(STATUS, "status", &status_ops, TRUE,  ""),
1827         VIEW_(STAGE,  "stage",  &stage_ops,  TRUE,  ""),
1828 };
1829
1830 #define VIEW(req)       (&views[(req) - REQ_OFFSET - 1])
1831 #define VIEW_REQ(view)  ((view) - views + REQ_OFFSET + 1)
1832
1833 #define foreach_view(view, i) \
1834         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1835
1836 #define view_is_displayed(view) \
1837         (view == display[0] || view == display[1])
1838
1839
1840 enum line_graphic {
1841         LINE_GRAPHIC_VLINE
1842 };
1843
1844 static int line_graphics[] = {
1845         /* LINE_GRAPHIC_VLINE: */ '|'
1846 };
1847
1848 static inline void
1849 set_view_attr(struct view *view, enum line_type type)
1850 {
1851         if (!view->curline->selected && view->curtype != type) {
1852                 wattrset(view->win, get_line_attr(type));
1853                 wchgat(view->win, -1, 0, type, NULL);
1854                 view->curtype = type;
1855         }
1856 }
1857
1858 static int
1859 draw_chars(struct view *view, enum line_type type, const char *string,
1860            int max_len, bool use_tilde)
1861 {
1862         int len = 0;
1863         int col = 0;
1864         int trimmed = FALSE;
1865
1866         if (max_len <= 0)
1867                 return 0;
1868
1869         if (opt_utf8) {
1870                 len = utf8_length(string, &col, max_len, &trimmed, use_tilde);
1871         } else {
1872                 col = len = strlen(string);
1873                 if (len > max_len) {
1874                         if (use_tilde) {
1875                                 max_len -= 1;
1876                         }
1877                         col = len = max_len;
1878                         trimmed = TRUE;
1879                 }
1880         }
1881
1882         set_view_attr(view, type);
1883         waddnstr(view->win, string, len);
1884         if (trimmed && use_tilde) {
1885                 set_view_attr(view, LINE_DELIMITER);
1886                 waddch(view->win, '~');
1887                 col++;
1888         }
1889
1890         return col;
1891 }
1892
1893 static int
1894 draw_space(struct view *view, enum line_type type, int max, int spaces)
1895 {
1896         static char space[] = "                    ";
1897         int col = 0;
1898
1899         spaces = MIN(max, spaces);
1900
1901         while (spaces > 0) {
1902                 int len = MIN(spaces, sizeof(space) - 1);
1903
1904                 col += draw_chars(view, type, space, spaces, FALSE);
1905                 spaces -= len;
1906         }
1907
1908         return col;
1909 }
1910
1911 static bool
1912 draw_lineno(struct view *view, unsigned int lineno)
1913 {
1914         char number[10];
1915         int digits3 = view->digits < 3 ? 3 : view->digits;
1916         int max_number = MIN(digits3, STRING_SIZE(number));
1917         int max = view->width - view->col;
1918         int col;
1919
1920         if (max < max_number)
1921                 max_number = max;
1922
1923         lineno += view->offset + 1;
1924         if (lineno == 1 || (lineno % opt_num_interval) == 0) {
1925                 static char fmt[] = "%1ld";
1926
1927                 if (view->digits <= 9)
1928                         fmt[1] = '0' + digits3;
1929
1930                 if (!string_format(number, fmt, lineno))
1931                         number[0] = 0;
1932                 col = draw_chars(view, LINE_LINE_NUMBER, number, max_number, TRUE);
1933         } else {
1934                 col = draw_space(view, LINE_LINE_NUMBER, max_number, max_number);
1935         }
1936
1937         if (col < max) {
1938                 set_view_attr(view, LINE_DEFAULT);
1939                 waddch(view->win, line_graphics[LINE_GRAPHIC_VLINE]);
1940                 col++;
1941         }
1942
1943         if (col < max)
1944                 col += draw_space(view, LINE_DEFAULT, max - col, 1);
1945         view->col += col;
1946
1947         return view->width - view->col <= 0;
1948 }
1949
1950 static bool
1951 draw_text(struct view *view, enum line_type type, const char *string, bool trim)
1952 {
1953         view->col += draw_chars(view, type, string, view->width - view->col, trim);
1954         return view->width - view->col <= 0;
1955 }
1956
1957 static bool
1958 draw_graphic(struct view *view, enum line_type type, chtype graphic[], size_t size)
1959 {
1960         int max = view->width - view->col;
1961         int i;
1962
1963         if (max < size)
1964                 size = max;
1965
1966         set_view_attr(view, type);
1967         /* Using waddch() instead of waddnstr() ensures that
1968          * they'll be rendered correctly for the cursor line. */
1969         for (i = 0; i < size; i++)
1970                 waddch(view->win, graphic[i]);
1971
1972         view->col += size;
1973         if (size < max) {
1974                 waddch(view->win, ' ');
1975                 view->col++;
1976         }
1977
1978         return view->width - view->col <= 0;
1979 }
1980
1981 static bool
1982 draw_field(struct view *view, enum line_type type, const char *text, int len, bool trim)
1983 {
1984         int max = MIN(view->width - view->col, len);
1985         int col;
1986
1987         if (text)
1988                 col = draw_chars(view, type, text, max - 1, trim);
1989         else
1990                 col = draw_space(view, type, max - 1, max - 1);
1991
1992         view->col += col + draw_space(view, LINE_DEFAULT, max - col, max - col);
1993         return view->width - view->col <= 0;
1994 }
1995
1996 static bool
1997 draw_date(struct view *view, struct tm *time)
1998 {
1999         char buf[DATE_COLS];
2000         char *date;
2001         int timelen = 0;
2002
2003         if (time)
2004                 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, time);
2005         date = timelen ? buf : NULL;
2006
2007         return draw_field(view, LINE_DATE, date, DATE_COLS, FALSE);
2008 }
2009
2010 static bool
2011 draw_author(struct view *view, const char *author)
2012 {
2013         return draw_field(view, LINE_MAIN_AUTHOR, author, opt_author_cols, TRUE);
2014 }
2015
2016 static bool
2017 draw_view_line(struct view *view, unsigned int lineno)
2018 {
2019         struct line *line;
2020         bool selected = (view->offset + lineno == view->lineno);
2021
2022         assert(view_is_displayed(view));
2023
2024         if (view->offset + lineno >= view->lines)
2025                 return FALSE;
2026
2027         line = &view->line[view->offset + lineno];
2028
2029         wmove(view->win, lineno, 0);
2030         if (line->cleareol)
2031                 wclrtoeol(view->win);
2032         view->col = 0;
2033         view->curline = line;
2034         view->curtype = LINE_NONE;
2035         line->selected = FALSE;
2036         line->dirty = line->cleareol = 0;
2037
2038         if (selected) {
2039                 set_view_attr(view, LINE_CURSOR);
2040                 line->selected = TRUE;
2041                 view->ops->select(view, line);
2042         }
2043
2044         return view->ops->draw(view, line, lineno);
2045 }
2046
2047 static void
2048 redraw_view_dirty(struct view *view)
2049 {
2050         bool dirty = FALSE;
2051         int lineno;
2052
2053         for (lineno = 0; lineno < view->height; lineno++) {
2054                 if (view->offset + lineno >= view->lines)
2055                         break;
2056                 if (!view->line[view->offset + lineno].dirty)
2057                         continue;
2058                 dirty = TRUE;
2059                 if (!draw_view_line(view, lineno))
2060                         break;
2061         }
2062
2063         if (!dirty)
2064                 return;
2065         if (input_mode)
2066                 wnoutrefresh(view->win);
2067         else
2068                 wrefresh(view->win);
2069 }
2070
2071 static void
2072 redraw_view_from(struct view *view, int lineno)
2073 {
2074         assert(0 <= lineno && lineno < view->height);
2075
2076         for (; lineno < view->height; lineno++) {
2077                 if (!draw_view_line(view, lineno))
2078                         break;
2079         }
2080
2081         if (input_mode)
2082                 wnoutrefresh(view->win);
2083         else
2084                 wrefresh(view->win);
2085 }
2086
2087 static void
2088 redraw_view(struct view *view)
2089 {
2090         werase(view->win);
2091         redraw_view_from(view, 0);
2092 }
2093
2094
2095 static void
2096 update_display_cursor(struct view *view)
2097 {
2098         /* Move the cursor to the right-most column of the cursor line.
2099          *
2100          * XXX: This could turn out to be a bit expensive, but it ensures that
2101          * the cursor does not jump around. */
2102         if (view->lines) {
2103                 wmove(view->win, view->lineno - view->offset, view->width - 1);
2104                 wrefresh(view->win);
2105         }
2106 }
2107
2108 static void
2109 update_view_title(struct view *view)
2110 {
2111         char buf[SIZEOF_STR];
2112         char state[SIZEOF_STR];
2113         size_t bufpos = 0, statelen = 0;
2114
2115         assert(view_is_displayed(view));
2116
2117         if (view != VIEW(REQ_VIEW_STATUS) && view->lines) {
2118                 unsigned int view_lines = view->offset + view->height;
2119                 unsigned int lines = view->lines
2120                                    ? MIN(view_lines, view->lines) * 100 / view->lines
2121                                    : 0;
2122
2123                 string_format_from(state, &statelen, " - %s %d of %d (%d%%)",
2124                                    view->ops->type,
2125                                    view->lineno + 1,
2126                                    view->lines,
2127                                    lines);
2128
2129         }
2130
2131         if (view->pipe) {
2132                 time_t secs = time(NULL) - view->start_time;
2133
2134                 /* Three git seconds are a long time ... */
2135                 if (secs > 2)
2136                         string_format_from(state, &statelen, " loading %lds", secs);
2137         }
2138
2139         string_format_from(buf, &bufpos, "[%s]", view->name);
2140         if (*view->ref && bufpos < view->width) {
2141                 size_t refsize = strlen(view->ref);
2142                 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
2143
2144                 if (minsize < view->width)
2145                         refsize = view->width - minsize + 7;
2146                 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
2147         }
2148
2149         if (statelen && bufpos < view->width) {
2150                 string_format_from(buf, &bufpos, "%s", state);
2151         }
2152
2153         if (view == display[current_view])
2154                 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
2155         else
2156                 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
2157
2158         mvwaddnstr(view->title, 0, 0, buf, bufpos);
2159         wclrtoeol(view->title);
2160         wmove(view->title, 0, view->width - 1);
2161
2162         if (input_mode)
2163                 wnoutrefresh(view->title);
2164         else
2165                 wrefresh(view->title);
2166 }
2167
2168 static void
2169 resize_display(void)
2170 {
2171         int offset, i;
2172         struct view *base = display[0];
2173         struct view *view = display[1] ? display[1] : display[0];
2174
2175         /* Setup window dimensions */
2176
2177         getmaxyx(stdscr, base->height, base->width);
2178
2179         /* Make room for the status window. */
2180         base->height -= 1;
2181
2182         if (view != base) {
2183                 /* Horizontal split. */
2184                 view->width   = base->width;
2185                 view->height  = SCALE_SPLIT_VIEW(base->height);
2186                 base->height -= view->height;
2187
2188                 /* Make room for the title bar. */
2189                 view->height -= 1;
2190         }
2191
2192         /* Make room for the title bar. */
2193         base->height -= 1;
2194
2195         offset = 0;
2196
2197         foreach_displayed_view (view, i) {
2198                 if (!view->win) {
2199                         view->win = newwin(view->height, 0, offset, 0);
2200                         if (!view->win)
2201                                 die("Failed to create %s view", view->name);
2202
2203                         scrollok(view->win, FALSE);
2204
2205                         view->title = newwin(1, 0, offset + view->height, 0);
2206                         if (!view->title)
2207                                 die("Failed to create title window");
2208
2209                 } else {
2210                         wresize(view->win, view->height, view->width);
2211                         mvwin(view->win,   offset, 0);
2212                         mvwin(view->title, offset + view->height, 0);
2213                 }
2214
2215                 offset += view->height + 1;
2216         }
2217 }
2218
2219 static void
2220 redraw_display(bool clear)
2221 {
2222         struct view *view;
2223         int i;
2224
2225         foreach_displayed_view (view, i) {
2226                 if (clear)
2227                         wclear(view->win);
2228                 redraw_view(view);
2229                 update_view_title(view);
2230         }
2231
2232         if (display[current_view])
2233                 update_display_cursor(display[current_view]);
2234 }
2235
2236 static void
2237 toggle_view_option(bool *option, const char *help)
2238 {
2239         *option = !*option;
2240         redraw_display(FALSE);
2241         report("%sabling %s", *option ? "En" : "Dis", help);
2242 }
2243
2244 /*
2245  * Navigation
2246  */
2247
2248 /* Scrolling backend */
2249 static void
2250 do_scroll_view(struct view *view, int lines)
2251 {
2252         bool redraw_current_line = FALSE;
2253
2254         /* The rendering expects the new offset. */
2255         view->offset += lines;
2256
2257         assert(0 <= view->offset && view->offset < view->lines);
2258         assert(lines);
2259
2260         /* Move current line into the view. */
2261         if (view->lineno < view->offset) {
2262                 view->lineno = view->offset;
2263                 redraw_current_line = TRUE;
2264         } else if (view->lineno >= view->offset + view->height) {
2265                 view->lineno = view->offset + view->height - 1;
2266                 redraw_current_line = TRUE;
2267         }
2268
2269         assert(view->offset <= view->lineno && view->lineno < view->lines);
2270
2271         /* Redraw the whole screen if scrolling is pointless. */
2272         if (view->height < ABS(lines)) {
2273                 redraw_view(view);
2274
2275         } else {
2276                 int line = lines > 0 ? view->height - lines : 0;
2277                 int end = line + ABS(lines);
2278
2279                 scrollok(view->win, TRUE);
2280                 wscrl(view->win, lines);
2281                 scrollok(view->win, FALSE);
2282
2283                 while (line < end && draw_view_line(view, line))
2284                         line++;
2285
2286                 if (redraw_current_line)
2287                         draw_view_line(view, view->lineno - view->offset);
2288                 /* FIXME: Stupid hack to workaround bug where the message from
2289                  * scrolling up one line when impossible followed by scrolling
2290                  * down one line is not removed by the next action. */
2291                 if (lines > 0)
2292                         report("");
2293                 wrefresh(view->win);
2294         }
2295
2296         report("");
2297 }
2298
2299 /* Scroll frontend */
2300 static void
2301 scroll_view(struct view *view, enum request request)
2302 {
2303         int lines = 1;
2304
2305         assert(view_is_displayed(view));
2306
2307         switch (request) {
2308         case REQ_SCROLL_PAGE_DOWN:
2309                 lines = view->height;
2310         case REQ_SCROLL_LINE_DOWN:
2311                 if (view->offset + lines > view->lines)
2312                         lines = view->lines - view->offset;
2313
2314                 if (lines == 0 || view->offset + view->height >= view->lines) {
2315                         report("Cannot scroll beyond the last line");
2316                         return;
2317                 }
2318                 break;
2319
2320         case REQ_SCROLL_PAGE_UP:
2321                 lines = view->height;
2322         case REQ_SCROLL_LINE_UP:
2323                 if (lines > view->offset)
2324                         lines = view->offset;
2325
2326                 if (lines == 0) {
2327                         report("Cannot scroll beyond the first line");
2328                         return;
2329                 }
2330
2331                 lines = -lines;
2332                 break;
2333
2334         default:
2335                 die("request %d not handled in switch", request);
2336         }
2337
2338         do_scroll_view(view, lines);
2339 }
2340
2341 /* Cursor moving */
2342 static void
2343 move_view(struct view *view, enum request request)
2344 {
2345         int scroll_steps = 0;
2346         int steps;
2347
2348         switch (request) {
2349         case REQ_MOVE_FIRST_LINE:
2350                 steps = -view->lineno;
2351                 break;
2352
2353         case REQ_MOVE_LAST_LINE:
2354                 steps = view->lines - view->lineno - 1;
2355                 break;
2356
2357         case REQ_MOVE_PAGE_UP:
2358                 steps = view->height > view->lineno
2359                       ? -view->lineno : -view->height;
2360                 break;
2361
2362         case REQ_MOVE_PAGE_DOWN:
2363                 steps = view->lineno + view->height >= view->lines
2364                       ? view->lines - view->lineno - 1 : view->height;
2365                 break;
2366
2367         case REQ_MOVE_UP:
2368                 steps = -1;
2369                 break;
2370
2371         case REQ_MOVE_DOWN:
2372                 steps = 1;
2373                 break;
2374
2375         default:
2376                 die("request %d not handled in switch", request);
2377         }
2378
2379         if (steps <= 0 && view->lineno == 0) {
2380                 report("Cannot move beyond the first line");
2381                 return;
2382
2383         } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
2384                 report("Cannot move beyond the last line");
2385                 return;
2386         }
2387
2388         /* Move the current line */
2389         view->lineno += steps;
2390         assert(0 <= view->lineno && view->lineno < view->lines);
2391
2392         /* Check whether the view needs to be scrolled */
2393         if (view->lineno < view->offset ||
2394             view->lineno >= view->offset + view->height) {
2395                 scroll_steps = steps;
2396                 if (steps < 0 && -steps > view->offset) {
2397                         scroll_steps = -view->offset;
2398
2399                 } else if (steps > 0) {
2400                         if (view->lineno == view->lines - 1 &&
2401                             view->lines > view->height) {
2402                                 scroll_steps = view->lines - view->offset - 1;
2403                                 if (scroll_steps >= view->height)
2404                                         scroll_steps -= view->height - 1;
2405                         }
2406                 }
2407         }
2408
2409         if (!view_is_displayed(view)) {
2410                 view->offset += scroll_steps;
2411                 assert(0 <= view->offset && view->offset < view->lines);
2412                 view->ops->select(view, &view->line[view->lineno]);
2413                 return;
2414         }
2415
2416         /* Repaint the old "current" line if we be scrolling */
2417         if (ABS(steps) < view->height)
2418                 draw_view_line(view, view->lineno - steps - view->offset);
2419
2420         if (scroll_steps) {
2421                 do_scroll_view(view, scroll_steps);
2422                 return;
2423         }
2424
2425         /* Draw the current line */
2426         draw_view_line(view, view->lineno - view->offset);
2427
2428         wrefresh(view->win);
2429         report("");
2430 }
2431
2432
2433 /*
2434  * Searching
2435  */
2436
2437 static void search_view(struct view *view, enum request request);
2438
2439 static void
2440 select_view_line(struct view *view, unsigned long lineno)
2441 {
2442         if (lineno - view->offset >= view->height) {
2443                 view->offset = lineno;
2444                 view->lineno = lineno;
2445                 if (view_is_displayed(view))
2446                         redraw_view(view);
2447
2448         } else {
2449                 unsigned long old_lineno = view->lineno - view->offset;
2450
2451                 view->lineno = lineno;
2452                 if (view_is_displayed(view)) {
2453                         draw_view_line(view, old_lineno);
2454                         draw_view_line(view, view->lineno - view->offset);
2455                         wrefresh(view->win);
2456                 } else {
2457                         view->ops->select(view, &view->line[view->lineno]);
2458                 }
2459         }
2460 }
2461
2462 static void
2463 find_next(struct view *view, enum request request)
2464 {
2465         unsigned long lineno = view->lineno;
2466         int direction;
2467
2468         if (!*view->grep) {
2469                 if (!*opt_search)
2470                         report("No previous search");
2471                 else
2472                         search_view(view, request);
2473                 return;
2474         }
2475
2476         switch (request) {
2477         case REQ_SEARCH:
2478         case REQ_FIND_NEXT:
2479                 direction = 1;
2480                 break;
2481
2482         case REQ_SEARCH_BACK:
2483         case REQ_FIND_PREV:
2484                 direction = -1;
2485                 break;
2486
2487         default:
2488                 return;
2489         }
2490
2491         if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
2492                 lineno += direction;
2493
2494         /* Note, lineno is unsigned long so will wrap around in which case it
2495          * will become bigger than view->lines. */
2496         for (; lineno < view->lines; lineno += direction) {
2497                 if (view->ops->grep(view, &view->line[lineno])) {
2498                         select_view_line(view, lineno);
2499                         report("Line %ld matches '%s'", lineno + 1, view->grep);
2500                         return;
2501                 }
2502         }
2503
2504         report("No match found for '%s'", view->grep);
2505 }
2506
2507 static void
2508 search_view(struct view *view, enum request request)
2509 {
2510         int regex_err;
2511
2512         if (view->regex) {
2513                 regfree(view->regex);
2514                 *view->grep = 0;
2515         } else {
2516                 view->regex = calloc(1, sizeof(*view->regex));
2517                 if (!view->regex)
2518                         return;
2519         }
2520
2521         regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
2522         if (regex_err != 0) {
2523                 char buf[SIZEOF_STR] = "unknown error";
2524
2525                 regerror(regex_err, view->regex, buf, sizeof(buf));
2526                 report("Search failed: %s", buf);
2527                 return;
2528         }
2529
2530         string_copy(view->grep, opt_search);
2531
2532         find_next(view, request);
2533 }
2534
2535 /*
2536  * Incremental updating
2537  */
2538
2539 static void
2540 reset_view(struct view *view)
2541 {
2542         int i;
2543
2544         for (i = 0; i < view->lines; i++)
2545                 free(view->line[i].data);
2546         free(view->line);
2547
2548         view->p_offset = view->offset;
2549         view->p_lineno = view->lineno;
2550
2551         view->line = NULL;
2552         view->offset = 0;
2553         view->lines  = 0;
2554         view->lineno = 0;
2555         view->line_alloc = 0;
2556         view->vid[0] = 0;
2557         view->update_secs = 0;
2558 }
2559
2560 static void
2561 free_argv(const char *argv[])
2562 {
2563         int argc;
2564
2565         for (argc = 0; argv[argc]; argc++)
2566                 free((void *) argv[argc]);
2567 }
2568
2569 static bool
2570 format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags flags)
2571 {
2572         char buf[SIZEOF_STR];
2573         int argc;
2574         bool noreplace = flags == FORMAT_NONE;
2575
2576         free_argv(dst_argv);
2577
2578         for (argc = 0; src_argv[argc]; argc++) {
2579                 const char *arg = src_argv[argc];
2580                 size_t bufpos = 0;
2581
2582                 while (arg) {
2583                         char *next = strstr(arg, "%(");
2584                         int len = next - arg;
2585                         const char *value;
2586
2587                         if (!next || noreplace) {
2588                                 if (flags == FORMAT_DASH && !strcmp(arg, "--"))
2589                                         noreplace = TRUE;
2590                                 len = strlen(arg);
2591                                 value = "";
2592
2593                         } else if (!prefixcmp(next, "%(directory)")) {
2594                                 value = opt_path;
2595
2596                         } else if (!prefixcmp(next, "%(file)")) {
2597                                 value = opt_file;
2598
2599                         } else if (!prefixcmp(next, "%(ref)")) {
2600                                 value = *opt_ref ? opt_ref : "HEAD";
2601
2602                         } else if (!prefixcmp(next, "%(head)")) {
2603                                 value = ref_head;
2604
2605                         } else if (!prefixcmp(next, "%(commit)")) {
2606                                 value = ref_commit;
2607
2608                         } else if (!prefixcmp(next, "%(blob)")) {
2609                                 value = ref_blob;
2610
2611                         } else {
2612                                 report("Unknown replacement: `%s`", next);
2613                                 return FALSE;
2614                         }
2615
2616                         if (!string_format_from(buf, &bufpos, "%.*s%s", len, arg, value))
2617                                 return FALSE;
2618
2619                         arg = next && !noreplace ? strchr(next, ')') + 1 : NULL;
2620                 }
2621
2622                 dst_argv[argc] = strdup(buf);
2623                 if (!dst_argv[argc])
2624                         break;
2625         }
2626
2627         dst_argv[argc] = NULL;
2628
2629         return src_argv[argc] == NULL;
2630 }
2631
2632 static bool
2633 restore_view_position(struct view *view)
2634 {
2635         if (!view->p_restore || (view->pipe && view->lines <= view->p_lineno))
2636                 return FALSE;
2637
2638         /* Changing the view position cancels the restoring. */
2639         /* FIXME: Changing back to the first line is not detected. */
2640         if (view->offset != 0 || view->lineno != 0) {
2641                 view->p_restore = FALSE;
2642                 return FALSE;
2643         }
2644
2645         if (view->p_lineno >= view->lines) {
2646                 view->p_lineno = view->lines > 0 ? view->lines - 1 : 0;
2647                 if (view->p_offset >= view->p_lineno) {
2648                         unsigned long half = view->height / 2;
2649
2650                         if (view->p_lineno > half)
2651                                 view->p_offset = view->p_lineno - half;
2652                         else
2653                                 view->p_offset = 0;
2654                 }
2655         }
2656
2657         if (view_is_displayed(view) &&
2658             view->offset != view->p_offset &&
2659             view->lineno != view->p_lineno)
2660                 werase(view->win);
2661
2662         view->offset = view->p_offset;
2663         view->lineno = view->p_lineno;
2664         view->p_restore = FALSE;
2665
2666         return TRUE;
2667 }
2668
2669 static void
2670 end_update(struct view *view, bool force)
2671 {
2672         if (!view->pipe)
2673                 return;
2674         while (!view->ops->read(view, NULL))
2675                 if (!force)
2676                         return;
2677         set_nonblocking_input(FALSE);
2678         if (force)
2679                 kill_io(view->pipe);
2680         done_io(view->pipe);
2681         view->pipe = NULL;
2682 }
2683
2684 static void
2685 setup_update(struct view *view, const char *vid)
2686 {
2687         set_nonblocking_input(TRUE);
2688         reset_view(view);
2689         string_copy_rev(view->vid, vid);
2690         view->pipe = &view->io;
2691         view->start_time = time(NULL);
2692 }
2693
2694 static bool
2695 prepare_update(struct view *view, const char *argv[], const char *dir,
2696                enum format_flags flags)
2697 {
2698         if (view->pipe)
2699                 end_update(view, TRUE);
2700         return init_io_rd(&view->io, argv, dir, flags);
2701 }
2702
2703 static bool
2704 prepare_update_file(struct view *view, const char *name)
2705 {
2706         if (view->pipe)
2707                 end_update(view, TRUE);
2708         return io_open(&view->io, name);
2709 }
2710
2711 static bool
2712 begin_update(struct view *view, bool refresh)
2713 {
2714         if (view->pipe)
2715                 end_update(view, TRUE);
2716
2717         if (refresh) {
2718                 if (!start_io(&view->io))
2719                         return FALSE;
2720
2721         } else {
2722                 if (view == VIEW(REQ_VIEW_TREE) && strcmp(view->vid, view->id))
2723                         opt_path[0] = 0;
2724
2725                 if (!run_io_rd(&view->io, view->ops->argv, FORMAT_ALL))
2726                         return FALSE;
2727
2728                 /* Put the current ref_* value to the view title ref
2729                  * member. This is needed by the blob view. Most other
2730                  * views sets it automatically after loading because the
2731                  * first line is a commit line. */
2732                 string_copy_rev(view->ref, view->id);
2733         }
2734
2735         setup_update(view, view->id);
2736
2737         return TRUE;
2738 }
2739
2740 #define ITEM_CHUNK_SIZE 256
2741 static void *
2742 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2743 {
2744         size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2745         size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2746
2747         if (mem == NULL || num_chunks != num_chunks_new) {
2748                 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2749                 mem = realloc(mem, *size * item_size);
2750         }
2751
2752         return mem;
2753 }
2754
2755 static struct line *
2756 realloc_lines(struct view *view, size_t line_size)
2757 {
2758         size_t alloc = view->line_alloc;
2759         struct line *tmp = realloc_items(view->line, &alloc, line_size,
2760                                          sizeof(*view->line));
2761
2762         if (!tmp)
2763                 return NULL;
2764
2765         view->line = tmp;
2766         view->line_alloc = alloc;
2767         return view->line;
2768 }
2769
2770 static bool
2771 update_view(struct view *view)
2772 {
2773         char out_buffer[BUFSIZ * 2];
2774         char *line;
2775         /* Clear the view and redraw everything since the tree sorting
2776          * might have rearranged things. */
2777         bool redraw = view->lines == 0;
2778         bool can_read = TRUE;
2779
2780         if (!view->pipe)
2781                 return TRUE;
2782
2783         if (!io_can_read(view->pipe)) {
2784                 if (view->lines == 0) {
2785                         time_t secs = time(NULL) - view->start_time;
2786
2787                         if (secs > view->update_secs) {
2788                                 if (view->update_secs == 0)
2789                                         redraw_view(view);
2790                                 update_view_title(view);
2791                                 view->update_secs = secs;
2792                         }
2793                 }
2794                 return TRUE;
2795         }
2796
2797         for (; (line = io_get(view->pipe, '\n', can_read)); can_read = FALSE) {
2798                 if (opt_iconv != ICONV_NONE) {
2799                         ICONV_CONST char *inbuf = line;
2800                         size_t inlen = strlen(line) + 1;
2801
2802                         char *outbuf = out_buffer;
2803                         size_t outlen = sizeof(out_buffer);
2804
2805                         size_t ret;
2806
2807                         ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2808                         if (ret != (size_t) -1)
2809                                 line = out_buffer;
2810                 }
2811
2812                 if (!view->ops->read(view, line)) {
2813                         report("Allocation failure");
2814                         end_update(view, TRUE);
2815                         return FALSE;
2816                 }
2817         }
2818
2819         {
2820                 unsigned long lines = view->lines;
2821                 int digits;
2822
2823                 for (digits = 0; lines; digits++)
2824                         lines /= 10;
2825
2826                 /* Keep the displayed view in sync with line number scaling. */
2827                 if (digits != view->digits) {
2828                         view->digits = digits;
2829                         if (opt_line_number || view == VIEW(REQ_VIEW_BLAME))
2830                                 redraw = TRUE;
2831                 }
2832         }
2833
2834         if (io_error(view->pipe)) {
2835                 report("Failed to read: %s", io_strerror(view->pipe));
2836                 end_update(view, TRUE);
2837
2838         } else if (io_eof(view->pipe)) {
2839                 report("");
2840                 end_update(view, FALSE);
2841         }
2842
2843         if (restore_view_position(view))
2844                 redraw = TRUE;
2845
2846         if (!view_is_displayed(view))
2847                 return TRUE;
2848
2849         if (redraw)
2850                 redraw_view_from(view, 0);
2851         else
2852                 redraw_view_dirty(view);
2853
2854         /* Update the title _after_ the redraw so that if the redraw picks up a
2855          * commit reference in view->ref it'll be available here. */
2856         update_view_title(view);
2857         update_display_cursor(view);
2858         return TRUE;
2859 }
2860
2861 static struct line *
2862 add_line_data(struct view *view, void *data, enum line_type type)
2863 {
2864         struct line *line;
2865
2866         if (!realloc_lines(view, view->lines + 1))
2867                 return NULL;
2868
2869         line = &view->line[view->lines++];
2870         memset(line, 0, sizeof(*line));
2871         line->type = type;
2872         line->data = data;
2873         line->dirty = 1;
2874
2875         return line;
2876 }
2877
2878 static struct line *
2879 add_line_text(struct view *view, const char *text, enum line_type type)
2880 {
2881         char *data = text ? strdup(text) : NULL;
2882
2883         return data ? add_line_data(view, data, type) : NULL;
2884 }
2885
2886 static struct line *
2887 add_line_format(struct view *view, enum line_type type, const char *fmt, ...)
2888 {
2889         char buf[SIZEOF_STR];
2890         va_list args;
2891
2892         va_start(args, fmt);
2893         if (vsnprintf(buf, sizeof(buf), fmt, args) >= sizeof(buf))
2894                 buf[0] = 0;
2895         va_end(args);
2896
2897         return buf[0] ? add_line_text(view, buf, type) : NULL;
2898 }
2899
2900 /*
2901  * View opening
2902  */
2903
2904 enum open_flags {
2905         OPEN_DEFAULT = 0,       /* Use default view switching. */
2906         OPEN_SPLIT = 1,         /* Split current view. */
2907         OPEN_BACKGROUNDED = 2,  /* Backgrounded. */
2908         OPEN_RELOAD = 4,        /* Reload view even if it is the current. */
2909         OPEN_NOMAXIMIZE = 8,    /* Do not maximize the current view. */
2910         OPEN_REFRESH = 16,      /* Refresh view using previous command. */
2911         OPEN_PREPARED = 32,     /* Open already prepared command. */
2912 };
2913
2914 static void
2915 open_view(struct view *prev, enum request request, enum open_flags flags)
2916 {
2917         bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2918         bool split = !!(flags & OPEN_SPLIT);
2919         bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH | OPEN_PREPARED));
2920         bool nomaximize = !!(flags & (OPEN_NOMAXIMIZE | OPEN_REFRESH));
2921         struct view *view = VIEW(request);
2922         int nviews = displayed_views();
2923         struct view *base_view = display[0];
2924
2925         if (view == prev && nviews == 1 && !reload) {
2926                 report("Already in %s view", view->name);
2927                 return;
2928         }
2929
2930         if (view->git_dir && !opt_git_dir[0]) {
2931                 report("The %s view is disabled in pager view", view->name);
2932                 return;
2933         }
2934
2935         if (split) {
2936                 display[1] = view;
2937                 if (!backgrounded)
2938                         current_view = 1;
2939         } else if (!nomaximize) {
2940                 /* Maximize the current view. */
2941                 memset(display, 0, sizeof(display));
2942                 current_view = 0;
2943                 display[current_view] = view;
2944         }
2945
2946         /* Resize the view when switching between split- and full-screen,
2947          * or when switching between two different full-screen views. */
2948         if (nviews != displayed_views() ||
2949             (nviews == 1 && base_view != display[0]))
2950                 resize_display();
2951
2952         if (view->ops->open) {
2953                 if (view->pipe)
2954                         end_update(view, TRUE);
2955                 if (!view->ops->open(view)) {
2956                         report("Failed to load %s view", view->name);
2957                         return;
2958                 }
2959                 restore_view_position(view);
2960
2961         } else if ((reload || strcmp(view->vid, view->id)) &&
2962                    !begin_update(view, flags & (OPEN_REFRESH | OPEN_PREPARED))) {
2963                 report("Failed to load %s view", view->name);
2964                 return;
2965         }
2966
2967         if (split && prev->lineno - prev->offset >= prev->height) {
2968                 /* Take the title line into account. */
2969                 int lines = prev->lineno - prev->offset - prev->height + 1;
2970
2971                 /* Scroll the view that was split if the current line is
2972                  * outside the new limited view. */
2973                 do_scroll_view(prev, lines);
2974         }
2975
2976         if (prev && view != prev) {
2977                 if (split && !backgrounded) {
2978                         /* "Blur" the previous view. */
2979                         update_view_title(prev);
2980                 }
2981
2982                 view->parent = prev;
2983         }
2984
2985         if (view->pipe && view->lines == 0) {
2986                 /* Clear the old view and let the incremental updating refill
2987                  * the screen. */
2988                 werase(view->win);
2989                 view->p_restore = flags & (OPEN_RELOAD | OPEN_REFRESH);
2990                 report("");
2991         } else if (view_is_displayed(view)) {
2992                 redraw_view(view);
2993                 report("");
2994         }
2995
2996         /* If the view is backgrounded the above calls to report()
2997          * won't redraw the view title. */
2998         if (backgrounded)
2999                 update_view_title(view);
3000 }
3001
3002 static void
3003 open_external_viewer(const char *argv[], const char *dir)
3004 {
3005         def_prog_mode();           /* save current tty modes */
3006         endwin();                  /* restore original tty modes */
3007         run_io_fg(argv, dir);
3008         fprintf(stderr, "Press Enter to continue");
3009         getc(opt_tty);
3010         reset_prog_mode();
3011         redraw_display(TRUE);
3012 }
3013
3014 static void
3015 open_mergetool(const char *file)
3016 {
3017         const char *mergetool_argv[] = { "git", "mergetool", file, NULL };
3018
3019         open_external_viewer(mergetool_argv, opt_cdup);
3020 }
3021
3022 static void
3023 open_editor(bool from_root, const char *file)
3024 {
3025         const char *editor_argv[] = { "vi", file, NULL };
3026         const char *editor;
3027
3028         editor = getenv("GIT_EDITOR");
3029         if (!editor && *opt_editor)
3030                 editor = opt_editor;
3031         if (!editor)
3032                 editor = getenv("VISUAL");
3033         if (!editor)
3034                 editor = getenv("EDITOR");
3035         if (!editor)
3036                 editor = "vi";
3037
3038         editor_argv[0] = editor;
3039         open_external_viewer(editor_argv, from_root ? opt_cdup : NULL);
3040 }
3041
3042 static void
3043 open_run_request(enum request request)
3044 {
3045         struct run_request *req = get_run_request(request);
3046         const char *argv[ARRAY_SIZE(req->argv)] = { NULL };
3047
3048         if (!req) {
3049                 report("Unknown run request");
3050                 return;
3051         }
3052
3053         if (format_argv(argv, req->argv, FORMAT_ALL))
3054                 open_external_viewer(argv, NULL);
3055         free_argv(argv);
3056 }
3057
3058 /*
3059  * User request switch noodle
3060  */
3061
3062 static int
3063 view_driver(struct view *view, enum request request)
3064 {
3065         int i;
3066
3067         if (request == REQ_NONE) {
3068                 doupdate();
3069                 return TRUE;
3070         }
3071
3072         if (request > REQ_NONE) {
3073                 open_run_request(request);
3074                 /* FIXME: When all views can refresh always do this. */
3075                 if (view == VIEW(REQ_VIEW_STATUS) ||
3076                     view == VIEW(REQ_VIEW_MAIN) ||
3077                     view == VIEW(REQ_VIEW_LOG) ||
3078                     view == VIEW(REQ_VIEW_STAGE))
3079                         request = REQ_REFRESH;
3080                 else
3081                         return TRUE;
3082         }
3083
3084         if (view && view->lines) {
3085                 request = view->ops->request(view, request, &view->line[view->lineno]);
3086                 if (request == REQ_NONE)
3087                         return TRUE;
3088         }
3089
3090         switch (request) {
3091         case REQ_MOVE_UP:
3092         case REQ_MOVE_DOWN:
3093         case REQ_MOVE_PAGE_UP:
3094         case REQ_MOVE_PAGE_DOWN:
3095         case REQ_MOVE_FIRST_LINE:
3096         case REQ_MOVE_LAST_LINE:
3097                 move_view(view, request);
3098                 break;
3099
3100         case REQ_SCROLL_LINE_DOWN:
3101         case REQ_SCROLL_LINE_UP:
3102         case REQ_SCROLL_PAGE_DOWN:
3103         case REQ_SCROLL_PAGE_UP:
3104                 scroll_view(view, request);
3105                 break;
3106
3107         case REQ_VIEW_BLAME:
3108                 if (!opt_file[0]) {
3109                         report("No file chosen, press %s to open tree view",
3110                                get_key(REQ_VIEW_TREE));
3111                         break;
3112                 }
3113                 open_view(view, request, OPEN_DEFAULT);
3114                 break;
3115
3116         case REQ_VIEW_BLOB:
3117                 if (!ref_blob[0]) {
3118                         report("No file chosen, press %s to open tree view",
3119                                get_key(REQ_VIEW_TREE));
3120                         break;
3121                 }
3122                 open_view(view, request, OPEN_DEFAULT);
3123                 break;
3124
3125         case REQ_VIEW_PAGER:
3126                 if (!VIEW(REQ_VIEW_PAGER)->pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
3127                         report("No pager content, press %s to run command from prompt",
3128                                get_key(REQ_PROMPT));
3129                         break;
3130                 }
3131                 open_view(view, request, OPEN_DEFAULT);
3132                 break;
3133
3134         case REQ_VIEW_STAGE:
3135                 if (!VIEW(REQ_VIEW_STAGE)->lines) {
3136                         report("No stage content, press %s to open the status view and choose file",
3137                                get_key(REQ_VIEW_STATUS));
3138                         break;
3139                 }
3140                 open_view(view, request, OPEN_DEFAULT);
3141                 break;
3142
3143         case REQ_VIEW_STATUS:
3144                 if (opt_is_inside_work_tree == FALSE) {
3145                         report("The status view requires a working tree");
3146                         break;
3147                 }
3148                 open_view(view, request, OPEN_DEFAULT);
3149                 break;
3150
3151         case REQ_VIEW_MAIN:
3152         case REQ_VIEW_DIFF:
3153         case REQ_VIEW_LOG:
3154         case REQ_VIEW_TREE:
3155         case REQ_VIEW_HELP:
3156                 open_view(view, request, OPEN_DEFAULT);
3157                 break;
3158
3159         case REQ_NEXT:
3160         case REQ_PREVIOUS:
3161                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
3162
3163                 if ((view == VIEW(REQ_VIEW_DIFF) &&
3164                      view->parent == VIEW(REQ_VIEW_MAIN)) ||
3165                    (view == VIEW(REQ_VIEW_DIFF) &&
3166                      view->parent == VIEW(REQ_VIEW_BLAME)) ||
3167                    (view == VIEW(REQ_VIEW_STAGE) &&
3168                      view->parent == VIEW(REQ_VIEW_STATUS)) ||
3169                    (view == VIEW(REQ_VIEW_BLOB) &&
3170                      view->parent == VIEW(REQ_VIEW_TREE))) {
3171                         int line;
3172
3173                         view = view->parent;
3174                         line = view->lineno;
3175                         move_view(view, request);
3176                         if (view_is_displayed(view))
3177                                 update_view_title(view);
3178                         if (line != view->lineno)
3179                                 view->ops->request(view, REQ_ENTER,
3180                                                    &view->line[view->lineno]);
3181
3182                 } else {
3183                         move_view(view, request);
3184                 }
3185                 break;
3186
3187         case REQ_VIEW_NEXT:
3188         {
3189                 int nviews = displayed_views();
3190                 int next_view = (current_view + 1) % nviews;
3191
3192                 if (next_view == current_view) {
3193                         report("Only one view is displayed");
3194                         break;
3195                 }
3196
3197                 current_view = next_view;
3198                 /* Blur out the title of the previous view. */
3199                 update_view_title(view);
3200                 report("");
3201                 break;
3202         }
3203         case REQ_REFRESH:
3204                 report("Refreshing is not yet supported for the %s view", view->name);
3205                 break;
3206
3207         case REQ_MAXIMIZE:
3208                 if (displayed_views() == 2)
3209                         open_view(view, VIEW_REQ(view), OPEN_DEFAULT);
3210                 break;
3211
3212         case REQ_TOGGLE_LINENO:
3213                 toggle_view_option(&opt_line_number, "line numbers");
3214                 break;
3215
3216         case REQ_TOGGLE_DATE:
3217                 toggle_view_option(&opt_date, "date display");
3218                 break;
3219
3220         case REQ_TOGGLE_AUTHOR:
3221                 toggle_view_option(&opt_author, "author display");
3222                 break;
3223
3224         case REQ_TOGGLE_REV_GRAPH:
3225                 toggle_view_option(&opt_rev_graph, "revision graph display");
3226                 break;
3227
3228         case REQ_TOGGLE_REFS:
3229                 toggle_view_option(&opt_show_refs, "reference display");
3230                 break;
3231
3232         case REQ_SEARCH:
3233         case REQ_SEARCH_BACK:
3234                 search_view(view, request);
3235                 break;
3236
3237         case REQ_FIND_NEXT:
3238         case REQ_FIND_PREV:
3239                 find_next(view, request);
3240                 break;
3241
3242         case REQ_STOP_LOADING:
3243                 for (i = 0; i < ARRAY_SIZE(views); i++) {
3244                         view = &views[i];
3245                         if (view->pipe)
3246                                 report("Stopped loading the %s view", view->name),
3247                         end_update(view, TRUE);
3248                 }
3249                 break;
3250
3251         case REQ_SHOW_VERSION:
3252                 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
3253                 return TRUE;
3254
3255         case REQ_SCREEN_REDRAW:
3256                 redraw_display(TRUE);
3257                 break;
3258
3259         case REQ_EDIT:
3260                 report("Nothing to edit");
3261                 break;
3262
3263         case REQ_ENTER:
3264                 report("Nothing to enter");
3265                 break;
3266
3267         case REQ_VIEW_CLOSE:
3268                 /* XXX: Mark closed views by letting view->parent point to the
3269                  * view itself. Parents to closed view should never be
3270                  * followed. */
3271                 if (view->parent &&
3272                     view->parent->parent != view->parent) {
3273                         memset(display, 0, sizeof(display));
3274                         current_view = 0;
3275                         display[current_view] = view->parent;
3276                         view->parent = view;
3277                         resize_display();
3278                         redraw_display(FALSE);
3279                         report("");
3280                         break;
3281                 }
3282                 /* Fall-through */
3283         case REQ_QUIT:
3284                 return FALSE;
3285
3286         default:
3287                 report("Unknown key, press 'h' for help");
3288                 return TRUE;
3289         }
3290
3291         return TRUE;
3292 }
3293
3294
3295 /*
3296  * View backend utilities
3297  */
3298
3299 /* Parse author lines where the name may be empty:
3300  *      author  <email@address.tld> 1138474660 +0100
3301  */
3302 static void
3303 parse_author_line(char *ident, char *author, size_t authorsize, struct tm *tm)
3304 {
3305         char *nameend = strchr(ident, '<');
3306         char *emailend = strchr(ident, '>');
3307
3308         if (nameend && emailend)
3309                 *nameend = *emailend = 0;
3310         ident = chomp_string(ident);
3311         if (!*ident) {
3312                 if (nameend)
3313                         ident = chomp_string(nameend + 1);
3314                 if (!*ident)
3315                         ident = "Unknown";
3316         }
3317
3318         string_ncopy_do(author, authorsize, ident, strlen(ident));
3319
3320         /* Parse epoch and timezone */
3321         if (emailend && emailend[1] == ' ') {
3322                 char *secs = emailend + 2;
3323                 char *zone = strchr(secs, ' ');
3324                 time_t time = (time_t) atol(secs);
3325
3326                 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
3327                         long tz;
3328
3329                         zone++;
3330                         tz  = ('0' - zone[1]) * 60 * 60 * 10;
3331                         tz += ('0' - zone[2]) * 60 * 60;
3332                         tz += ('0' - zone[3]) * 60;
3333                         tz += ('0' - zone[4]) * 60;
3334
3335                         if (zone[0] == '-')
3336                                 tz = -tz;
3337
3338                         time -= tz;
3339                 }
3340
3341                 gmtime_r(&time, tm);
3342         }
3343 }
3344
3345 static enum input_status
3346 select_commit_parent_handler(void *data, char *buf, int c)
3347 {
3348         size_t parents = *(size_t *) data;
3349         int parent = 0;
3350
3351         if (!isdigit(c))
3352                 return INPUT_SKIP;
3353
3354         if (*buf)
3355                 parent = atoi(buf) * 10;
3356         parent += c - '0';
3357
3358         if (parent > parents)
3359                 return INPUT_SKIP;
3360         return INPUT_OK;
3361 }
3362
3363 static bool
3364 select_commit_parent(const char *id, char rev[SIZEOF_REV])
3365 {
3366         char buf[SIZEOF_STR * 4];
3367         const char *revlist_argv[] = {
3368                 "git", "rev-list", "-1", "--parents", id, NULL
3369         };
3370         int parents;
3371
3372         if (!run_io_buf(revlist_argv, buf, sizeof(buf)) ||
3373             !*chomp_string(buf) ||
3374             (parents = (strlen(buf) / 40) - 1) < 0) {
3375                 report("Failed to get parent information");
3376                 return FALSE;
3377
3378         } else if (parents == 0) {
3379                 report("The selected commit has no parents");
3380                 return FALSE;
3381         }
3382
3383         if (parents > 1) {
3384                 char prompt[SIZEOF_STR];
3385                 char *result;
3386
3387                 if (!string_format(prompt, "Which parent? [1..%d] ", parents))
3388                         return FALSE;
3389                 result = prompt_input(prompt, select_commit_parent_handler, &parents);
3390                 if (!result)
3391                         return FALSE;
3392                 parents = atoi(result);
3393         }
3394
3395         string_copy_rev(rev, &buf[41 * parents]);
3396         return TRUE;
3397 }
3398
3399 /*
3400  * Pager backend
3401  */
3402
3403 static bool
3404 pager_draw(struct view *view, struct line *line, unsigned int lineno)
3405 {
3406         char *text = line->data;
3407
3408         if (opt_line_number && draw_lineno(view, lineno))
3409                 return TRUE;
3410
3411         draw_text(view, line->type, text, TRUE);
3412         return TRUE;
3413 }
3414
3415 static bool
3416 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
3417 {
3418         const char *describe_argv[] = { "git", "describe", commit_id, NULL };
3419         char refbuf[SIZEOF_STR];
3420         char *ref = NULL;
3421
3422         if (run_io_buf(describe_argv, refbuf, sizeof(refbuf)))
3423                 ref = chomp_string(refbuf);
3424
3425         if (!ref || !*ref)
3426                 return TRUE;
3427
3428         /* This is the only fatal call, since it can "corrupt" the buffer. */
3429         if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
3430                 return FALSE;
3431
3432         return TRUE;
3433 }
3434
3435 static void
3436 add_pager_refs(struct view *view, struct line *line)
3437 {
3438         char buf[SIZEOF_STR];
3439         char *commit_id = (char *)line->data + STRING_SIZE("commit ");
3440         struct ref **refs;
3441         size_t bufpos = 0, refpos = 0;
3442         const char *sep = "Refs: ";
3443         bool is_tag = FALSE;
3444
3445         assert(line->type == LINE_COMMIT);
3446
3447         refs = get_refs(commit_id);
3448         if (!refs) {
3449                 if (view == VIEW(REQ_VIEW_DIFF))
3450                         goto try_add_describe_ref;
3451                 return;
3452         }
3453
3454         do {
3455                 struct ref *ref = refs[refpos];
3456                 const char *fmt = ref->tag    ? "%s[%s]" :
3457                                   ref->remote ? "%s<%s>" : "%s%s";
3458
3459                 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
3460                         return;
3461                 sep = ", ";
3462                 if (ref->tag)
3463                         is_tag = TRUE;
3464         } while (refs[refpos++]->next);
3465
3466         if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
3467 try_add_describe_ref:
3468                 /* Add <tag>-g<commit_id> "fake" reference. */
3469                 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
3470                         return;
3471         }
3472
3473         if (bufpos == 0)
3474                 return;
3475
3476         add_line_text(view, buf, LINE_PP_REFS);
3477 }
3478
3479 static bool
3480 pager_read(struct view *view, char *data)
3481 {
3482         struct line *line;
3483
3484         if (!data)
3485                 return TRUE;
3486
3487         line = add_line_text(view, data, get_line_type(data));
3488         if (!line)
3489                 return FALSE;
3490
3491         if (line->type == LINE_COMMIT &&
3492             (view == VIEW(REQ_VIEW_DIFF) ||
3493              view == VIEW(REQ_VIEW_LOG)))
3494                 add_pager_refs(view, line);
3495
3496         return TRUE;
3497 }
3498
3499 static enum request
3500 pager_request(struct view *view, enum request request, struct line *line)
3501 {
3502         int split = 0;
3503
3504         if (request != REQ_ENTER)
3505                 return request;
3506
3507         if (line->type == LINE_COMMIT &&
3508            (view == VIEW(REQ_VIEW_LOG) ||
3509             view == VIEW(REQ_VIEW_PAGER))) {
3510                 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
3511                 split = 1;
3512         }
3513
3514         /* Always scroll the view even if it was split. That way
3515          * you can use Enter to scroll through the log view and
3516          * split open each commit diff. */
3517         scroll_view(view, REQ_SCROLL_LINE_DOWN);
3518
3519         /* FIXME: A minor workaround. Scrolling the view will call report("")
3520          * but if we are scrolling a non-current view this won't properly
3521          * update the view title. */
3522         if (split)
3523                 update_view_title(view);
3524
3525         return REQ_NONE;
3526 }
3527
3528 static bool
3529 pager_grep(struct view *view, struct line *line)
3530 {
3531         regmatch_t pmatch;
3532         char *text = line->data;
3533
3534         if (!*text)
3535                 return FALSE;
3536
3537         if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
3538                 return FALSE;
3539
3540         return TRUE;
3541 }
3542
3543 static void
3544 pager_select(struct view *view, struct line *line)
3545 {
3546         if (line->type == LINE_COMMIT) {
3547                 char *text = (char *)line->data + STRING_SIZE("commit ");
3548
3549                 if (view != VIEW(REQ_VIEW_PAGER))
3550                         string_copy_rev(view->ref, text);
3551                 string_copy_rev(ref_commit, text);
3552         }
3553 }
3554
3555 static struct view_ops pager_ops = {
3556         "line",
3557         NULL,
3558         NULL,
3559         pager_read,
3560         pager_draw,
3561         pager_request,
3562         pager_grep,
3563         pager_select,
3564 };
3565
3566 static const char *log_argv[SIZEOF_ARG] = {
3567         "git", "log", "--no-color", "--cc", "--stat", "-n100", "%(head)", NULL
3568 };
3569
3570 static enum request
3571 log_request(struct view *view, enum request request, struct line *line)
3572 {
3573         switch (request) {
3574         case REQ_REFRESH:
3575                 load_refs();
3576                 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
3577                 return REQ_NONE;
3578         default:
3579                 return pager_request(view, request, line);
3580         }
3581 }
3582
3583 static struct view_ops log_ops = {
3584         "line",
3585         log_argv,
3586         NULL,
3587         pager_read,
3588         pager_draw,
3589         log_request,
3590         pager_grep,
3591         pager_select,
3592 };
3593
3594 static const char *diff_argv[SIZEOF_ARG] = {
3595         "git", "show", "--pretty=fuller", "--no-color", "--root",
3596                 "--patch-with-stat", "--find-copies-harder", "-C", "%(commit)", NULL
3597 };
3598
3599 static struct view_ops diff_ops = {
3600         "line",
3601         diff_argv,
3602         NULL,
3603         pager_read,
3604         pager_draw,
3605         pager_request,
3606         pager_grep,
3607         pager_select,
3608 };
3609
3610 /*
3611  * Help backend
3612  */
3613
3614 static bool
3615 help_open(struct view *view)
3616 {
3617         char buf[SIZEOF_STR];
3618         size_t bufpos;
3619         int i;
3620
3621         if (view->lines > 0)
3622                 return TRUE;
3623
3624         add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3625
3626         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3627                 const char *key;
3628
3629                 if (req_info[i].request == REQ_NONE)
3630                         continue;
3631
3632                 if (!req_info[i].request) {
3633                         add_line_text(view, "", LINE_DEFAULT);
3634                         add_line_text(view, req_info[i].help, LINE_DEFAULT);
3635                         continue;
3636                 }
3637
3638                 key = get_key(req_info[i].request);
3639                 if (!*key)
3640                         key = "(no key defined)";
3641
3642                 for (bufpos = 0; bufpos <= req_info[i].namelen; bufpos++) {
3643                         buf[bufpos] = tolower(req_info[i].name[bufpos]);
3644                         if (buf[bufpos] == '_')
3645                                 buf[bufpos] = '-';
3646                 }
3647
3648                 add_line_format(view, LINE_DEFAULT, "    %-25s %-20s %s",
3649                                 key, buf, req_info[i].help);
3650         }
3651
3652         if (run_requests) {
3653                 add_line_text(view, "", LINE_DEFAULT);
3654                 add_line_text(view, "External commands:", LINE_DEFAULT);
3655         }
3656
3657         for (i = 0; i < run_requests; i++) {
3658                 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3659                 const char *key;
3660                 int argc;
3661
3662                 if (!req)
3663                         continue;
3664
3665                 key = get_key_name(req->key);
3666                 if (!*key)
3667                         key = "(no key defined)";
3668
3669                 for (bufpos = 0, argc = 0; req->argv[argc]; argc++)
3670                         if (!string_format_from(buf, &bufpos, "%s%s",
3671                                                 argc ? " " : "", req->argv[argc]))
3672                                 return REQ_NONE;
3673
3674                 add_line_format(view, LINE_DEFAULT, "    %-10s %-14s `%s`",
3675                                 keymap_table[req->keymap].name, key, buf);
3676         }
3677
3678         return TRUE;
3679 }
3680
3681 static struct view_ops help_ops = {
3682         "line",
3683         NULL,
3684         help_open,
3685         NULL,
3686         pager_draw,
3687         pager_request,
3688         pager_grep,
3689         pager_select,
3690 };
3691
3692
3693 /*
3694  * Tree backend
3695  */
3696
3697 struct tree_stack_entry {
3698         struct tree_stack_entry *prev;  /* Entry below this in the stack */
3699         unsigned long lineno;           /* Line number to restore */
3700         char *name;                     /* Position of name in opt_path */
3701 };
3702
3703 /* The top of the path stack. */
3704 static struct tree_stack_entry *tree_stack = NULL;
3705 unsigned long tree_lineno = 0;
3706
3707 static void
3708 pop_tree_stack_entry(void)
3709 {
3710         struct tree_stack_entry *entry = tree_stack;
3711
3712         tree_lineno = entry->lineno;
3713         entry->name[0] = 0;
3714         tree_stack = entry->prev;
3715         free(entry);
3716 }
3717
3718 static void
3719 push_tree_stack_entry(const char *name, unsigned long lineno)
3720 {
3721         struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3722         size_t pathlen = strlen(opt_path);
3723
3724         if (!entry)
3725                 return;
3726
3727         entry->prev = tree_stack;
3728         entry->name = opt_path + pathlen;
3729         tree_stack = entry;
3730
3731         if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3732                 pop_tree_stack_entry();
3733                 return;
3734         }
3735
3736         /* Move the current line to the first tree entry. */
3737         tree_lineno = 1;
3738         entry->lineno = lineno;
3739 }
3740
3741 /* Parse output from git-ls-tree(1):
3742  *
3743  * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
3744  * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
3745  * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3746  * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
3747  */
3748
3749 #define SIZEOF_TREE_ATTR \
3750         STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
3751
3752 #define SIZEOF_TREE_MODE \
3753         STRING_SIZE("100644 ")
3754
3755 #define TREE_ID_OFFSET \
3756         STRING_SIZE("100644 blob ")
3757
3758 struct tree_entry {
3759         char id[SIZEOF_REV];
3760         mode_t mode;
3761         struct tm time;                 /* Date from the author ident. */
3762         char author[75];                /* Author of the commit. */
3763         char name[1];
3764 };
3765
3766 static const char *
3767 tree_path(struct line *line)
3768 {
3769         return ((struct tree_entry *) line->data)->name;
3770 }
3771
3772
3773 static int
3774 tree_compare_entry(struct line *line1, struct line *line2)
3775 {
3776         if (line1->type != line2->type)
3777                 return line1->type == LINE_TREE_DIR ? -1 : 1;
3778         return strcmp(tree_path(line1), tree_path(line2));
3779 }
3780
3781 static struct line *
3782 tree_entry(struct view *view, enum line_type type, const char *path,
3783            const char *mode, const char *id)
3784 {
3785         struct tree_entry *entry = calloc(1, sizeof(*entry) + strlen(path));
3786         struct line *line = entry ? add_line_data(view, entry, type) : NULL;
3787
3788         if (!entry || !line) {
3789                 free(entry);
3790                 return NULL;
3791         }
3792
3793         strncpy(entry->name, path, strlen(path));
3794         if (mode)
3795                 entry->mode = strtoul(mode, NULL, 8);
3796         if (id)
3797                 string_copy_rev(entry->id, id);
3798
3799         return line;
3800 }
3801
3802 static bool
3803 tree_read_date(struct view *view, char *text, bool *read_date)
3804 {
3805         static char author_name[SIZEOF_STR];
3806         static struct tm author_time;
3807
3808         if (!text && *read_date) {
3809                 *read_date = FALSE;
3810                 return TRUE;
3811
3812         } else if (!text) {
3813                 char *path = *opt_path ? opt_path : ".";
3814                 /* Find next entry to process */
3815                 const char *log_file[] = {
3816                         "git", "log", "--no-color", "--pretty=raw",
3817                                 "--cc", "--raw", view->id, "--", path, NULL
3818                 };
3819                 struct io io = {};
3820
3821                 if (!run_io_rd(&io, log_file, FORMAT_NONE)) {
3822                         report("Failed to load tree data");
3823                         return TRUE;
3824                 }
3825
3826                 done_io(view->pipe);
3827                 view->io = io;
3828                 *read_date = TRUE;
3829                 return FALSE;
3830
3831         } else if (*text == 'a' && get_line_type(text) == LINE_AUTHOR) {
3832                 parse_author_line(text + STRING_SIZE("author "),
3833                                   author_name, sizeof(author_name), &author_time);
3834
3835         } else if (*text == ':') {
3836                 char *pos;
3837                 size_t annotated = 1;
3838                 size_t i;
3839
3840                 pos = strchr(text, '\t');
3841                 if (!pos)
3842                         return TRUE;
3843                 text = pos + 1;
3844                 if (*opt_prefix && !strncmp(text, opt_prefix, strlen(opt_prefix)))
3845                         text += strlen(opt_prefix);
3846                 if (*opt_path && !strncmp(text, opt_path, strlen(opt_path)))
3847                         text += strlen(opt_path);
3848                 pos = strchr(text, '/');
3849                 if (pos)
3850                         *pos = 0;
3851
3852                 for (i = 1; i < view->lines; i++) {
3853                         struct line *line = &view->line[i];
3854                         struct tree_entry *entry = line->data;
3855
3856                         annotated += !!*entry->author;
3857                         if (*entry->author || strcmp(entry->name, text))
3858                                 continue;
3859
3860                         string_copy(entry->author, author_name);
3861                         memcpy(&entry->time, &author_time, sizeof(entry->time));
3862                         line->dirty = 1;
3863                         break;
3864                 }
3865
3866                 if (annotated == view->lines)
3867                         kill_io(view->pipe);
3868         }
3869         return TRUE;
3870 }
3871
3872 static bool
3873 tree_read(struct view *view, char *text)
3874 {
3875         static bool read_date = FALSE;
3876         struct tree_entry *data;
3877         struct line *entry, *line;
3878         enum line_type type;
3879         size_t textlen = text ? strlen(text) : 0;
3880         char *path = text + SIZEOF_TREE_ATTR;
3881
3882         if (read_date || !text)
3883                 return tree_read_date(view, text, &read_date);
3884
3885         if (textlen <= SIZEOF_TREE_ATTR)
3886                 return FALSE;
3887         if (view->lines == 0 &&
3888             !tree_entry(view, LINE_TREE_PARENT, opt_path, NULL, NULL))
3889                 return FALSE;
3890
3891         /* Strip the path part ... */
3892         if (*opt_path) {
3893                 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3894                 size_t striplen = strlen(opt_path);
3895
3896                 if (pathlen > striplen)
3897                         memmove(path, path + striplen,
3898                                 pathlen - striplen + 1);
3899
3900                 /* Insert "link" to parent directory. */
3901                 if (view->lines == 1 &&
3902                     !tree_entry(view, LINE_TREE_DIR, "..", "040000", view->ref))
3903                         return FALSE;
3904         }
3905
3906         type = text[SIZEOF_TREE_MODE] == 't' ? LINE_TREE_DIR : LINE_TREE_FILE;
3907         entry = tree_entry(view, type, path, text, text + TREE_ID_OFFSET);
3908         if (!entry)
3909                 return FALSE;
3910         data = entry->data;
3911
3912         /* Skip "Directory ..." and ".." line. */
3913         for (line = &view->line[1 + !!*opt_path]; line < entry; line++) {
3914                 if (tree_compare_entry(line, entry) <= 0)
3915                         continue;
3916
3917                 memmove(line + 1, line, (entry - line) * sizeof(*entry));
3918
3919                 line->data = data;
3920                 line->type = type;
3921                 for (; line <= entry; line++)
3922                         line->dirty = line->cleareol = 1;
3923                 return TRUE;
3924         }
3925
3926         if (tree_lineno > view->lineno) {
3927                 view->lineno = tree_lineno;
3928                 tree_lineno = 0;
3929         }
3930
3931         return TRUE;
3932 }
3933
3934 static bool
3935 tree_draw(struct view *view, struct line *line, unsigned int lineno)
3936 {
3937         struct tree_entry *entry = line->data;
3938
3939         if (line->type == LINE_TREE_PARENT) {
3940                 if (draw_text(view, line->type, "Directory path /", TRUE))
3941                         return TRUE;
3942         } else {
3943                 char mode[11] = "-r--r--r--";
3944
3945                 if (S_ISDIR(entry->mode)) {
3946                         mode[3] = mode[6] = mode[9] = 'x';
3947                         mode[0] = 'd';
3948                 }
3949                 if (S_ISLNK(entry->mode))
3950                         mode[0] = 'l';
3951                 if (entry->mode & S_IWUSR)
3952                         mode[2] = 'w';
3953                 if (entry->mode & S_IXUSR)
3954                         mode[3] = 'x';
3955                 if (entry->mode & S_IXGRP)
3956                         mode[6] = 'x';
3957                 if (entry->mode & S_IXOTH)
3958                         mode[9] = 'x';
3959                 if (draw_field(view, LINE_TREE_MODE, mode, 11, TRUE))
3960                         return TRUE;
3961
3962                 if (opt_author && draw_author(view, entry->author))
3963                         return TRUE;
3964
3965                 if (opt_date && draw_date(view, *entry->author ? &entry->time : NULL))
3966                         return TRUE;
3967         }
3968         if (draw_text(view, line->type, entry->name, TRUE))
3969                 return TRUE;
3970         return TRUE;
3971 }
3972
3973 static void
3974 open_blob_editor()
3975 {
3976         char file[SIZEOF_STR] = "/tmp/tigblob.XXXXXX";
3977         int fd = mkstemp(file);
3978
3979         if (fd == -1)
3980                 report("Failed to create temporary file");
3981         else if (!run_io_append(blob_ops.argv, FORMAT_ALL, fd))
3982                 report("Failed to save blob data to file");
3983         else
3984                 open_editor(FALSE, file);
3985         if (fd != -1)
3986                 unlink(file);
3987 }
3988
3989 static enum request
3990 tree_request(struct view *view, enum request request, struct line *line)
3991 {
3992         enum open_flags flags;
3993
3994         switch (request) {
3995         case REQ_VIEW_BLAME:
3996                 if (line->type != LINE_TREE_FILE) {
3997                         report("Blame only supported for files");
3998                         return REQ_NONE;
3999                 }
4000
4001                 string_copy(opt_ref, view->vid);
4002                 return request;
4003
4004         case REQ_EDIT:
4005                 if (line->type != LINE_TREE_FILE) {
4006                         report("Edit only supported for files");
4007                 } else if (!is_head_commit(view->vid)) {
4008                         open_blob_editor();
4009                 } else {
4010                         open_editor(TRUE, opt_file);
4011                 }
4012                 return REQ_NONE;
4013
4014         case REQ_PARENT:
4015                 if (!*opt_path) {
4016                         /* quit view if at top of tree */
4017                         return REQ_VIEW_CLOSE;
4018                 }
4019                 /* fake 'cd  ..' */
4020                 line = &view->line[1];
4021                 break;
4022
4023         case REQ_ENTER:
4024                 break;
4025
4026         default:
4027                 return request;
4028         }
4029
4030         /* Cleanup the stack if the tree view is at a different tree. */
4031         while (!*opt_path && tree_stack)
4032                 pop_tree_stack_entry();
4033
4034         switch (line->type) {
4035         case LINE_TREE_DIR:
4036                 /* Depending on whether it is a subdir or parent (updir?) link
4037                  * mangle the path buffer. */
4038                 if (line == &view->line[1] && *opt_path) {
4039                         pop_tree_stack_entry();
4040
4041                 } else {
4042                         const char *basename = tree_path(line);
4043
4044                         push_tree_stack_entry(basename, view->lineno);
4045                 }
4046
4047                 /* Trees and subtrees share the same ID, so they are not not
4048                  * unique like blobs. */
4049                 flags = OPEN_RELOAD;
4050                 request = REQ_VIEW_TREE;
4051                 break;
4052
4053         case LINE_TREE_FILE:
4054                 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4055                 request = REQ_VIEW_BLOB;
4056                 break;
4057
4058         default:
4059                 return REQ_NONE;
4060         }
4061
4062         open_view(view, request, flags);
4063         if (request == REQ_VIEW_TREE)
4064                 view->lineno = tree_lineno;
4065
4066         return REQ_NONE;
4067 }
4068
4069 static void
4070 tree_select(struct view *view, struct line *line)
4071 {
4072         struct tree_entry *entry = line->data;
4073
4074         if (line->type == LINE_TREE_FILE) {
4075                 string_copy_rev(ref_blob, entry->id);
4076                 string_format(opt_file, "%s%s", opt_path, tree_path(line));
4077
4078         } else if (line->type != LINE_TREE_DIR) {
4079                 return;
4080         }
4081
4082         string_copy_rev(view->ref, entry->id);
4083 }
4084
4085 static const char *tree_argv[SIZEOF_ARG] = {
4086         "git", "ls-tree", "%(commit)", "%(directory)", NULL
4087 };
4088
4089 static struct view_ops tree_ops = {
4090         "file",
4091         tree_argv,
4092         NULL,
4093         tree_read,
4094         tree_draw,
4095         tree_request,
4096         pager_grep,
4097         tree_select,
4098 };
4099
4100 static bool
4101 blob_read(struct view *view, char *line)
4102 {
4103         if (!line)
4104                 return TRUE;
4105         return add_line_text(view, line, LINE_DEFAULT) != NULL;
4106 }
4107
4108 static enum request
4109 blob_request(struct view *view, enum request request, struct line *line)
4110 {
4111         switch (request) {
4112         case REQ_EDIT:
4113                 open_blob_editor();
4114                 return REQ_NONE;
4115         default:
4116                 return pager_request(view, request, line);
4117         }
4118 }
4119
4120 static const char *blob_argv[SIZEOF_ARG] = {
4121         "git", "cat-file", "blob", "%(blob)", NULL
4122 };
4123
4124 static struct view_ops blob_ops = {
4125         "line",
4126         blob_argv,
4127         NULL,
4128         blob_read,
4129         pager_draw,
4130         blob_request,
4131         pager_grep,
4132         pager_select,
4133 };
4134
4135 /*
4136  * Blame backend
4137  *
4138  * Loading the blame view is a two phase job:
4139  *
4140  *  1. File content is read either using opt_file from the
4141  *     filesystem or using git-cat-file.
4142  *  2. Then blame information is incrementally added by
4143  *     reading output from git-blame.
4144  */
4145
4146 static const char *blame_head_argv[] = {
4147         "git", "blame", "--incremental", "--", "%(file)", NULL
4148 };
4149
4150 static const char *blame_ref_argv[] = {
4151         "git", "blame", "--incremental", "%(ref)", "--", "%(file)", NULL
4152 };
4153
4154 static const char *blame_cat_file_argv[] = {
4155         "git", "cat-file", "blob", "%(ref):%(file)", NULL
4156 };
4157
4158 struct blame_commit {
4159         char id[SIZEOF_REV];            /* SHA1 ID. */
4160         char title[128];                /* First line of the commit message. */
4161         char author[75];                /* Author of the commit. */
4162         struct tm time;                 /* Date from the author ident. */
4163         char filename[128];             /* Name of file. */
4164         bool has_previous;              /* Was a "previous" line detected. */
4165 };
4166
4167 struct blame {
4168         struct blame_commit *commit;
4169         char text[1];
4170 };
4171
4172 static bool
4173 blame_open(struct view *view)
4174 {
4175         if (*opt_ref || !io_open(&view->io, opt_file)) {
4176                 if (!run_io_rd(&view->io, blame_cat_file_argv, FORMAT_ALL))
4177                         return FALSE;
4178         }
4179
4180         setup_update(view, opt_file);
4181         string_format(view->ref, "%s ...", opt_file);
4182
4183         return TRUE;
4184 }
4185
4186 static struct blame_commit *
4187 get_blame_commit(struct view *view, const char *id)
4188 {
4189         size_t i;
4190
4191         for (i = 0; i < view->lines; i++) {
4192                 struct blame *blame = view->line[i].data;
4193
4194                 if (!blame->commit)
4195                         continue;
4196
4197                 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
4198                         return blame->commit;
4199         }
4200
4201         {
4202                 struct blame_commit *commit = calloc(1, sizeof(*commit));
4203
4204                 if (commit)
4205                         string_ncopy(commit->id, id, SIZEOF_REV);
4206                 return commit;
4207         }
4208 }
4209
4210 static bool
4211 parse_number(const char **posref, size_t *number, size_t min, size_t max)
4212 {
4213         const char *pos = *posref;
4214
4215         *posref = NULL;
4216         pos = strchr(pos + 1, ' ');
4217         if (!pos || !isdigit(pos[1]))
4218                 return FALSE;
4219         *number = atoi(pos + 1);
4220         if (*number < min || *number > max)
4221                 return FALSE;
4222
4223         *posref = pos;
4224         return TRUE;
4225 }
4226
4227 static struct blame_commit *
4228 parse_blame_commit(struct view *view, const char *text, int *blamed)
4229 {
4230         struct blame_commit *commit;
4231         struct blame *blame;
4232         const char *pos = text + SIZEOF_REV - 1;
4233         size_t lineno;
4234         size_t group;
4235
4236         if (strlen(text) <= SIZEOF_REV || *pos != ' ')
4237                 return NULL;
4238
4239         if (!parse_number(&pos, &lineno, 1, view->lines) ||
4240             !parse_number(&pos, &group, 1, view->lines - lineno + 1))
4241                 return NULL;
4242
4243         commit = get_blame_commit(view, text);
4244         if (!commit)
4245                 return NULL;
4246
4247         *blamed += group;
4248         while (group--) {
4249                 struct line *line = &view->line[lineno + group - 1];
4250
4251                 blame = line->data;
4252                 blame->commit = commit;
4253                 line->dirty = 1;
4254         }
4255
4256         return commit;
4257 }
4258
4259 static bool
4260 blame_read_file(struct view *view, const char *line, bool *read_file)
4261 {
4262         if (!line) {
4263                 const char **argv = *opt_ref ? blame_ref_argv : blame_head_argv;
4264                 struct io io = {};
4265
4266                 if (view->lines == 0 && !view->parent)
4267                         die("No blame exist for %s", view->vid);
4268
4269                 if (view->lines == 0 || !run_io_rd(&io, argv, FORMAT_ALL)) {
4270                         report("Failed to load blame data");
4271                         return TRUE;
4272                 }
4273
4274                 done_io(view->pipe);
4275                 view->io = io;
4276                 *read_file = FALSE;
4277                 return FALSE;
4278
4279         } else {
4280                 size_t linelen = strlen(line);
4281                 struct blame *blame = malloc(sizeof(*blame) + linelen);
4282
4283                 blame->commit = NULL;
4284                 strncpy(blame->text, line, linelen);
4285                 blame->text[linelen] = 0;
4286                 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
4287         }
4288 }
4289
4290 static bool
4291 match_blame_header(const char *name, char **line)
4292 {
4293         size_t namelen = strlen(name);
4294         bool matched = !strncmp(name, *line, namelen);
4295
4296         if (matched)
4297                 *line += namelen;
4298
4299         return matched;
4300 }
4301
4302 static bool
4303 blame_read(struct view *view, char *line)
4304 {
4305         static struct blame_commit *commit = NULL;
4306         static int blamed = 0;
4307         static time_t author_time;
4308         static bool read_file = TRUE;
4309
4310         if (read_file)
4311                 return blame_read_file(view, line, &read_file);
4312
4313         if (!line) {
4314                 /* Reset all! */
4315                 commit = NULL;
4316                 blamed = 0;
4317                 read_file = TRUE;
4318                 string_format(view->ref, "%s", view->vid);
4319                 if (view_is_displayed(view)) {
4320                         update_view_title(view);
4321                         redraw_view_from(view, 0);
4322                 }
4323                 return TRUE;
4324         }
4325
4326         if (!commit) {
4327                 commit = parse_blame_commit(view, line, &blamed);
4328                 string_format(view->ref, "%s %2d%%", view->vid,
4329                               view->lines ? blamed * 100 / view->lines : 0);
4330
4331         } else if (match_blame_header("author ", &line)) {
4332                 string_ncopy(commit->author, line, strlen(line));
4333
4334         } else if (match_blame_header("author-time ", &line)) {
4335                 author_time = (time_t) atol(line);
4336
4337         } else if (match_blame_header("author-tz ", &line)) {
4338                 long tz;
4339
4340                 tz  = ('0' - line[1]) * 60 * 60 * 10;
4341                 tz += ('0' - line[2]) * 60 * 60;
4342                 tz += ('0' - line[3]) * 60;
4343                 tz += ('0' - line[4]) * 60;
4344
4345                 if (line[0] == '-')
4346                         tz = -tz;
4347
4348                 author_time -= tz;
4349                 gmtime_r(&author_time, &commit->time);
4350
4351         } else if (match_blame_header("summary ", &line)) {
4352                 string_ncopy(commit->title, line, strlen(line));
4353
4354         } else if (match_blame_header("previous ", &line)) {
4355                 commit->has_previous = TRUE;
4356
4357         } else if (match_blame_header("filename ", &line)) {
4358                 string_ncopy(commit->filename, line, strlen(line));
4359                 commit = NULL;
4360         }
4361
4362         return TRUE;
4363 }
4364
4365 static bool
4366 blame_draw(struct view *view, struct line *line, unsigned int lineno)
4367 {
4368         struct blame *blame = line->data;
4369         struct tm *time = NULL;
4370         const char *id = NULL, *author = NULL;
4371
4372         if (blame->commit && *blame->commit->filename) {
4373                 id = blame->commit->id;
4374                 author = blame->commit->author;
4375                 time = &blame->commit->time;
4376         }
4377
4378         if (opt_date && draw_date(view, time))
4379                 return TRUE;
4380
4381         if (opt_author && draw_author(view, author))
4382                 return TRUE;
4383
4384         if (draw_field(view, LINE_BLAME_ID, id, ID_COLS, FALSE))
4385                 return TRUE;
4386
4387         if (draw_lineno(view, lineno))
4388                 return TRUE;
4389
4390         draw_text(view, LINE_DEFAULT, blame->text, TRUE);
4391         return TRUE;
4392 }
4393
4394 static bool
4395 check_blame_commit(struct blame *blame)
4396 {
4397         if (!blame->commit)
4398                 report("Commit data not loaded yet");
4399         else if (!strcmp(blame->commit->id, NULL_ID))
4400                 report("No commit exist for the selected line");
4401         else
4402                 return TRUE;
4403         return FALSE;
4404 }
4405
4406 static enum request
4407 blame_request(struct view *view, enum request request, struct line *line)
4408 {
4409         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4410         struct blame *blame = line->data;
4411
4412         switch (request) {
4413         case REQ_VIEW_BLAME:
4414                 if (check_blame_commit(blame)) {
4415                         string_copy(opt_ref, blame->commit->id);
4416                         open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4417                 }
4418                 break;
4419
4420         case REQ_PARENT:
4421                 if (check_blame_commit(blame) &&
4422                     select_commit_parent(blame->commit->id, opt_ref))
4423                         open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4424                 break;
4425
4426         case REQ_ENTER:
4427                 if (!blame->commit) {
4428                         report("No commit loaded yet");
4429                         break;
4430                 }
4431
4432                 if (view_is_displayed(VIEW(REQ_VIEW_DIFF)) &&
4433                     !strcmp(blame->commit->id, VIEW(REQ_VIEW_DIFF)->ref))
4434                         break;
4435
4436                 if (!strcmp(blame->commit->id, NULL_ID)) {
4437                         struct view *diff = VIEW(REQ_VIEW_DIFF);
4438                         const char *diff_index_argv[] = {
4439                                 "git", "diff-index", "--root", "--patch-with-stat",
4440                                         "-C", "-M", "HEAD", "--", view->vid, NULL
4441                         };
4442
4443                         if (!blame->commit->has_previous) {
4444                                 diff_index_argv[1] = "diff";
4445                                 diff_index_argv[2] = "--no-color";
4446                                 diff_index_argv[6] = "--";
4447                                 diff_index_argv[7] = "/dev/null";
4448                         }
4449
4450                         if (!prepare_update(diff, diff_index_argv, NULL, FORMAT_DASH)) {
4451                                 report("Failed to allocate diff command");
4452                                 break;
4453                         }
4454                         flags |= OPEN_PREPARED;
4455                 }
4456
4457                 open_view(view, REQ_VIEW_DIFF, flags);
4458                 if (VIEW(REQ_VIEW_DIFF)->pipe && !strcmp(blame->commit->id, NULL_ID))
4459                         string_copy_rev(VIEW(REQ_VIEW_DIFF)->ref, NULL_ID);
4460                 break;
4461
4462         default:
4463                 return request;
4464         }
4465
4466         return REQ_NONE;
4467 }
4468
4469 static bool
4470 blame_grep(struct view *view, struct line *line)
4471 {
4472         struct blame *blame = line->data;
4473         struct blame_commit *commit = blame->commit;
4474         regmatch_t pmatch;
4475
4476 #define MATCH(text, on)                                                 \
4477         (on && *text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4478
4479         if (commit) {
4480                 char buf[DATE_COLS + 1];
4481
4482                 if (MATCH(commit->title, 1) ||
4483                     MATCH(commit->author, opt_author) ||
4484                     MATCH(commit->id, opt_date))
4485                         return TRUE;
4486
4487                 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
4488                     MATCH(buf, 1))
4489                         return TRUE;
4490         }
4491
4492         return MATCH(blame->text, 1);
4493
4494 #undef MATCH
4495 }
4496
4497 static void
4498 blame_select(struct view *view, struct line *line)
4499 {
4500         struct blame *blame = line->data;
4501         struct blame_commit *commit = blame->commit;
4502
4503         if (!commit)
4504                 return;
4505
4506         if (!strcmp(commit->id, NULL_ID))
4507                 string_ncopy(ref_commit, "HEAD", 4);
4508         else
4509                 string_copy_rev(ref_commit, commit->id);
4510 }
4511
4512 static struct view_ops blame_ops = {
4513         "line",
4514         NULL,
4515         blame_open,
4516         blame_read,
4517         blame_draw,
4518         blame_request,
4519         blame_grep,
4520         blame_select,
4521 };
4522
4523 /*
4524  * Status backend
4525  */
4526
4527 struct status {
4528         char status;
4529         struct {
4530                 mode_t mode;
4531                 char rev[SIZEOF_REV];
4532                 char name[SIZEOF_STR];
4533         } old;
4534         struct {
4535                 mode_t mode;
4536                 char rev[SIZEOF_REV];
4537                 char name[SIZEOF_STR];
4538         } new;
4539 };
4540
4541 static char status_onbranch[SIZEOF_STR];
4542 static struct status stage_status;
4543 static enum line_type stage_line_type;
4544 static size_t stage_chunks;
4545 static int *stage_chunk;
4546
4547 /* This should work even for the "On branch" line. */
4548 static inline bool
4549 status_has_none(struct view *view, struct line *line)
4550 {
4551         return line < view->line + view->lines && !line[1].data;
4552 }
4553
4554 /* Get fields from the diff line:
4555  * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
4556  */
4557 static inline bool
4558 status_get_diff(struct status *file, const char *buf, size_t bufsize)
4559 {
4560         const char *old_mode = buf +  1;
4561         const char *new_mode = buf +  8;
4562         const char *old_rev  = buf + 15;
4563         const char *new_rev  = buf + 56;
4564         const char *status   = buf + 97;
4565
4566         if (bufsize < 98 ||
4567             old_mode[-1] != ':' ||
4568             new_mode[-1] != ' ' ||
4569             old_rev[-1]  != ' ' ||
4570             new_rev[-1]  != ' ' ||
4571             status[-1]   != ' ')
4572                 return FALSE;
4573
4574         file->status = *status;
4575
4576         string_copy_rev(file->old.rev, old_rev);
4577         string_copy_rev(file->new.rev, new_rev);
4578
4579         file->old.mode = strtoul(old_mode, NULL, 8);
4580         file->new.mode = strtoul(new_mode, NULL, 8);
4581
4582         file->old.name[0] = file->new.name[0] = 0;
4583
4584         return TRUE;
4585 }
4586
4587 static bool
4588 status_run(struct view *view, const char *argv[], char status, enum line_type type)
4589 {
4590         struct status *file = NULL;
4591         struct status *unmerged = NULL;
4592         char *buf;
4593         struct io io = {};
4594
4595         if (!run_io(&io, argv, NULL, IO_RD))
4596                 return FALSE;
4597
4598         add_line_data(view, NULL, type);
4599
4600         while ((buf = io_get(&io, 0, TRUE))) {
4601                 if (!file) {
4602                         file = calloc(1, sizeof(*file));
4603                         if (!file || !add_line_data(view, file, type))
4604                                 goto error_out;
4605                 }
4606
4607                 /* Parse diff info part. */
4608                 if (status) {
4609                         file->status = status;
4610                         if (status == 'A')
4611                                 string_copy(file->old.rev, NULL_ID);
4612
4613                 } else if (!file->status) {
4614                         if (!status_get_diff(file, buf, strlen(buf)))
4615                                 goto error_out;
4616
4617                         buf = io_get(&io, 0, TRUE);
4618                         if (!buf)
4619                                 break;
4620
4621                         /* Collapse all 'M'odified entries that follow a
4622                          * associated 'U'nmerged entry. */
4623                         if (file->status == 'U') {
4624                                 unmerged = file;
4625
4626                         } else if (unmerged) {
4627                                 int collapse = !strcmp(buf, unmerged->new.name);
4628
4629                                 unmerged = NULL;
4630                                 if (collapse) {
4631                                         free(file);
4632                                         file = NULL;
4633                                         view->lines--;
4634                                         continue;
4635                                 }
4636                         }
4637                 }
4638
4639                 /* Grab the old name for rename/copy. */
4640                 if (!*file->old.name &&
4641                     (file->status == 'R' || file->status == 'C')) {
4642                         string_ncopy(file->old.name, buf, strlen(buf));
4643
4644                         buf = io_get(&io, 0, TRUE);
4645                         if (!buf)
4646                                 break;
4647                 }
4648
4649                 /* git-ls-files just delivers a NUL separated list of
4650                  * file names similar to the second half of the
4651                  * git-diff-* output. */
4652                 string_ncopy(file->new.name, buf, strlen(buf));
4653                 if (!*file->old.name)
4654                         string_copy(file->old.name, file->new.name);
4655                 file = NULL;
4656         }
4657
4658         if (io_error(&io)) {
4659 error_out:
4660                 done_io(&io);
4661                 return FALSE;
4662         }
4663
4664         if (!view->line[view->lines - 1].data)
4665                 add_line_data(view, NULL, LINE_STAT_NONE);
4666
4667         done_io(&io);
4668         return TRUE;
4669 }
4670
4671 /* Don't show unmerged entries in the staged section. */
4672 static const char *status_diff_index_argv[] = {
4673         "git", "diff-index", "-z", "--diff-filter=ACDMRTXB",
4674                              "--cached", "-M", "HEAD", NULL
4675 };
4676
4677 static const char *status_diff_files_argv[] = {
4678         "git", "diff-files", "-z", NULL
4679 };
4680
4681 static const char *status_list_other_argv[] = {
4682         "git", "ls-files", "-z", "--others", "--exclude-standard", NULL
4683 };
4684
4685 static const char *status_list_no_head_argv[] = {
4686         "git", "ls-files", "-z", "--cached", "--exclude-standard", NULL
4687 };
4688
4689 static const char *update_index_argv[] = {
4690         "git", "update-index", "-q", "--unmerged", "--refresh", NULL
4691 };
4692
4693 /* Restore the previous line number to stay in the context or select a
4694  * line with something that can be updated. */
4695 static void
4696 status_restore(struct view *view)
4697 {
4698         if (view->p_lineno >= view->lines)
4699                 view->p_lineno = view->lines - 1;
4700         while (view->p_lineno < view->lines && !view->line[view->p_lineno].data)
4701                 view->p_lineno++;
4702         while (view->p_lineno > 0 && !view->line[view->p_lineno].data)
4703                 view->p_lineno--;
4704
4705         /* If the above fails, always skip the "On branch" line. */
4706         if (view->p_lineno < view->lines)
4707                 view->lineno = view->p_lineno;
4708         else
4709                 view->lineno = 1;
4710
4711         if (view->lineno < view->offset)
4712                 view->offset = view->lineno;
4713         else if (view->offset + view->height <= view->lineno)
4714                 view->offset = view->lineno - view->height + 1;
4715
4716         view->p_restore = FALSE;
4717 }
4718
4719 /* First parse staged info using git-diff-index(1), then parse unstaged
4720  * info using git-diff-files(1), and finally untracked files using
4721  * git-ls-files(1). */
4722 static bool
4723 status_open(struct view *view)
4724 {
4725         reset_view(view);
4726
4727         add_line_data(view, NULL, LINE_STAT_HEAD);
4728         if (is_initial_commit())
4729                 string_copy(status_onbranch, "Initial commit");
4730         else if (!*opt_head)
4731                 string_copy(status_onbranch, "Not currently on any branch");
4732         else if (!string_format(status_onbranch, "On branch %s", opt_head))
4733                 return FALSE;
4734
4735         run_io_bg(update_index_argv);
4736
4737         if (is_initial_commit()) {
4738                 if (!status_run(view, status_list_no_head_argv, 'A', LINE_STAT_STAGED))
4739                         return FALSE;
4740         } else if (!status_run(view, status_diff_index_argv, 0, LINE_STAT_STAGED)) {
4741                 return FALSE;
4742         }
4743
4744         if (!status_run(view, status_diff_files_argv, 0, LINE_STAT_UNSTAGED) ||
4745             !status_run(view, status_list_other_argv, '?', LINE_STAT_UNTRACKED))
4746                 return FALSE;
4747
4748         /* Restore the exact position or use the specialized restore
4749          * mode? */
4750         if (!view->p_restore)
4751                 status_restore(view);
4752         return TRUE;
4753 }
4754
4755 static bool
4756 status_draw(struct view *view, struct line *line, unsigned int lineno)
4757 {
4758         struct status *status = line->data;
4759         enum line_type type;
4760         const char *text;
4761
4762         if (!status) {
4763                 switch (line->type) {
4764                 case LINE_STAT_STAGED:
4765                         type = LINE_STAT_SECTION;
4766                         text = "Changes to be committed:";
4767                         break;
4768
4769                 case LINE_STAT_UNSTAGED:
4770                         type = LINE_STAT_SECTION;
4771                         text = "Changed but not updated:";
4772                         break;
4773
4774                 case LINE_STAT_UNTRACKED:
4775                         type = LINE_STAT_SECTION;
4776                         text = "Untracked files:";
4777                         break;
4778
4779                 case LINE_STAT_NONE:
4780                         type = LINE_DEFAULT;
4781                         text = "    (no files)";
4782                         break;
4783
4784                 case LINE_STAT_HEAD:
4785                         type = LINE_STAT_HEAD;
4786                         text = status_onbranch;
4787                         break;
4788
4789                 default:
4790                         return FALSE;
4791  &nb