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