Code

Keep the cursor fixed while initial stage progress is reported
[tig.git] / 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  */
14 #ifdef HAVE_CONFIG_H
15 #include "config.h"
16 #endif
18 #ifndef TIG_VERSION
19 #define TIG_VERSION "unknown-version"
20 #endif
22 #ifndef DEBUG
23 #define NDEBUG
24 #endif
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>
42 #include <regex.h>
44 #include <locale.h>
45 #include <langinfo.h>
46 #include <iconv.h>
48 /* ncurses(3): Must be defined to have extended wide-character functions. */
49 #define _XOPEN_SOURCE_EXTENDED
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
61 #if __GNUC__ >= 3
62 #define __NORETURN __attribute__((__noreturn__))
63 #else
64 #define __NORETURN
65 #endif
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 int load_refs(void);
72 static size_t utf8_length(const char **string, size_t col, int *width, size_t max_width, int *trimmed, bool reserve);
74 #define ABS(x)          ((x) >= 0  ? (x) : -(x))
75 #define MIN(x, y)       ((x) < (y) ? (x) :  (y))
77 #define ARRAY_SIZE(x)   (sizeof(x) / sizeof(x[0]))
78 #define STRING_SIZE(x)  (sizeof(x) - 1)
80 #define SIZEOF_STR      1024    /* Default string size. */
81 #define SIZEOF_REF      256     /* Size of symbolic or SHA1 ID. */
82 #define SIZEOF_REV      41      /* Holds a SHA-1 and an ending NUL. */
83 #define SIZEOF_ARG      32      /* Default argument array size. */
85 /* Revision graph */
87 #define REVGRAPH_INIT   'I'
88 #define REVGRAPH_MERGE  'M'
89 #define REVGRAPH_BRANCH '+'
90 #define REVGRAPH_COMMIT '*'
91 #define REVGRAPH_BOUND  '^'
93 #define SIZEOF_REVGRAPH 19      /* Size of revision ancestry graphics. */
95 /* This color name can be used to refer to the default term colors. */
96 #define COLOR_DEFAULT   (-1)
98 #define ICONV_NONE      ((iconv_t) -1)
99 #ifndef ICONV_CONST
100 #define ICONV_CONST     /* nothing */
101 #endif
103 /* The format and size of the date column in the main view. */
104 #define DATE_FORMAT     "%Y-%m-%d %H:%M"
105 #define DATE_COLS       STRING_SIZE("2006-04-29 14:21 ")
107 #define AUTHOR_COLS     20
108 #define ID_COLS         8
110 /* The default interval between line numbers. */
111 #define NUMBER_INTERVAL 5
113 #define TAB_SIZE        8
115 #define SCALE_SPLIT_VIEW(height)        ((height) * 2 / 3)
117 #define NULL_ID         "0000000000000000000000000000000000000000"
119 #define S_ISGITLINK(mode) (((mode) & S_IFMT) == 0160000)
121 #ifndef GIT_CONFIG
122 #define GIT_CONFIG "config"
123 #endif
125 /* Some ASCII-shorthands fitted into the ncurses namespace. */
126 #define KEY_TAB         '\t'
127 #define KEY_RETURN      '\r'
128 #define KEY_ESC         27
131 struct ref {
132         char *name;             /* Ref name; tag or head names are shortened. */
133         char id[SIZEOF_REV];    /* Commit SHA1 ID */
134         unsigned int head:1;    /* Is it the current HEAD? */
135         unsigned int tag:1;     /* Is it a tag? */
136         unsigned int ltag:1;    /* If so, is the tag local? */
137         unsigned int remote:1;  /* Is it a remote ref? */
138         unsigned int tracked:1; /* Is it the remote for the current HEAD? */
139         unsigned int next:1;    /* For ref lists: are there more refs? */
140 };
142 static struct ref **get_refs(const char *id);
144 enum format_flags {
145         FORMAT_ALL,             /* Perform replacement in all arguments. */
146         FORMAT_DASH,            /* Perform replacement up until "--". */
147         FORMAT_NONE             /* No replacement should be performed. */
148 };
150 static bool format_argv(const char *dst[], const char *src[], enum format_flags flags);
152 enum input_status {
153         INPUT_OK,
154         INPUT_SKIP,
155         INPUT_STOP,
156         INPUT_CANCEL
157 };
159 typedef enum input_status (*input_handler)(void *data, char *buf, int c);
161 static char *prompt_input(const char *prompt, input_handler handler, void *data);
162 static bool prompt_yesno(const char *prompt);
164 /*
165  * String helpers
166  */
168 static inline void
169 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
171         if (srclen > dstlen - 1)
172                 srclen = dstlen - 1;
174         strncpy(dst, src, srclen);
175         dst[srclen] = 0;
178 /* Shorthands for safely copying into a fixed buffer. */
180 #define string_copy(dst, src) \
181         string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
183 #define string_ncopy(dst, src, srclen) \
184         string_ncopy_do(dst, sizeof(dst), src, srclen)
186 #define string_copy_rev(dst, src) \
187         string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
189 #define string_add(dst, from, src) \
190         string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
192 static void
193 string_expand(char *dst, size_t dstlen, const char *src, int tabsize)
195         size_t size, pos;
197         for (size = pos = 0; size < dstlen - 1 && src[pos]; pos++) {
198                 if (src[pos] == '\t') {
199                         size_t expanded = tabsize - (size % tabsize);
201                         if (expanded + size >= dstlen - 1)
202                                 expanded = dstlen - size - 1;
203                         memcpy(dst + size, "        ", expanded);
204                         size += expanded;
205                 } else {
206                         dst[size++] = src[pos];
207                 }
208         }
210         dst[size] = 0;
213 static char *
214 chomp_string(char *name)
216         int namelen;
218         while (isspace(*name))
219                 name++;
221         namelen = strlen(name) - 1;
222         while (namelen > 0 && isspace(name[namelen]))
223                 name[namelen--] = 0;
225         return name;
228 static bool
229 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
231         va_list args;
232         size_t pos = bufpos ? *bufpos : 0;
234         va_start(args, fmt);
235         pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
236         va_end(args);
238         if (bufpos)
239                 *bufpos = pos;
241         return pos >= bufsize ? FALSE : TRUE;
244 #define string_format(buf, fmt, args...) \
245         string_nformat(buf, sizeof(buf), NULL, fmt, args)
247 #define string_format_from(buf, from, fmt, args...) \
248         string_nformat(buf, sizeof(buf), from, fmt, args)
250 static int
251 string_enum_compare(const char *str1, const char *str2, int len)
253         size_t i;
255 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
257         /* Diff-Header == DIFF_HEADER */
258         for (i = 0; i < len; i++) {
259                 if (toupper(str1[i]) == toupper(str2[i]))
260                         continue;
262                 if (string_enum_sep(str1[i]) &&
263                     string_enum_sep(str2[i]))
264                         continue;
266                 return str1[i] - str2[i];
267         }
269         return 0;
272 struct enum_map {
273         const char *name;
274         int namelen;
275         int value;
276 };
278 #define ENUM_MAP(name, value) { name, STRING_SIZE(name), value }
280 static bool
281 map_enum_do(const struct enum_map *map, size_t map_size, int *value, const char *name)
283         size_t namelen = strlen(name);
284         int i;
286         for (i = 0; i < map_size; i++)
287                 if (namelen == map[i].namelen &&
288                     !string_enum_compare(name, map[i].name, namelen)) {
289                         *value = map[i].value;
290                         return TRUE;
291                 }
293         return FALSE;
296 #define map_enum(attr, map, name) \
297         map_enum_do(map, ARRAY_SIZE(map), attr, name)
299 #define prefixcmp(str1, str2) \
300         strncmp(str1, str2, STRING_SIZE(str2))
302 static inline int
303 suffixcmp(const char *str, int slen, const char *suffix)
305         size_t len = slen >= 0 ? slen : strlen(str);
306         size_t suffixlen = strlen(suffix);
308         return suffixlen < len ? strcmp(str + len - suffixlen, suffix) : -1;
312 static bool
313 argv_from_string(const char *argv[SIZEOF_ARG], int *argc, char *cmd)
315         int valuelen;
317         while (*cmd && *argc < SIZEOF_ARG && (valuelen = strcspn(cmd, " \t"))) {
318                 bool advance = cmd[valuelen] != 0;
320                 cmd[valuelen] = 0;
321                 argv[(*argc)++] = chomp_string(cmd);
322                 cmd = chomp_string(cmd + valuelen + advance);
323         }
325         if (*argc < SIZEOF_ARG)
326                 argv[*argc] = NULL;
327         return *argc < SIZEOF_ARG;
330 static void
331 argv_from_env(const char **argv, const char *name)
333         char *env = argv ? getenv(name) : NULL;
334         int argc = 0;
336         if (env && *env)
337                 env = strdup(env);
338         if (env && !argv_from_string(argv, &argc, env))
339                 die("Too many arguments in the `%s` environment variable", name);
343 /*
344  * Executing external commands.
345  */
347 enum io_type {
348         IO_FD,                  /* File descriptor based IO. */
349         IO_BG,                  /* Execute command in the background. */
350         IO_FG,                  /* Execute command with same std{in,out,err}. */
351         IO_RD,                  /* Read only fork+exec IO. */
352         IO_WR,                  /* Write only fork+exec IO. */
353         IO_AP,                  /* Append fork+exec output to file. */
354 };
356 struct io {
357         enum io_type type;      /* The requested type of pipe. */
358         const char *dir;        /* Directory from which to execute. */
359         pid_t pid;              /* Pipe for reading or writing. */
360         int pipe;               /* Pipe end for reading or writing. */
361         int error;              /* Error status. */
362         const char *argv[SIZEOF_ARG];   /* Shell command arguments. */
363         char *buf;              /* Read buffer. */
364         size_t bufalloc;        /* Allocated buffer size. */
365         size_t bufsize;         /* Buffer content size. */
366         char *bufpos;           /* Current buffer position. */
367         unsigned int eof:1;     /* Has end of file been reached. */
368 };
370 static void
371 reset_io(struct io *io)
373         io->pipe = -1;
374         io->pid = 0;
375         io->buf = io->bufpos = NULL;
376         io->bufalloc = io->bufsize = 0;
377         io->error = 0;
378         io->eof = 0;
381 static void
382 init_io(struct io *io, const char *dir, enum io_type type)
384         reset_io(io);
385         io->type = type;
386         io->dir = dir;
389 static bool
390 init_io_rd(struct io *io, const char *argv[], const char *dir,
391                 enum format_flags flags)
393         init_io(io, dir, IO_RD);
394         return format_argv(io->argv, argv, flags);
397 static bool
398 io_open(struct io *io, const char *name)
400         init_io(io, NULL, IO_FD);
401         io->pipe = *name ? open(name, O_RDONLY) : STDIN_FILENO;
402         if (io->pipe == -1)
403                 io->error = errno;
404         return io->pipe != -1;
407 static bool
408 kill_io(struct io *io)
410         return io->pid == 0 || kill(io->pid, SIGKILL) != -1;
413 static bool
414 done_io(struct io *io)
416         pid_t pid = io->pid;
418         if (io->pipe != -1)
419                 close(io->pipe);
420         free(io->buf);
421         reset_io(io);
423         while (pid > 0) {
424                 int status;
425                 pid_t waiting = waitpid(pid, &status, 0);
427                 if (waiting < 0) {
428                         if (errno == EINTR)
429                                 continue;
430                         report("waitpid failed (%s)", strerror(errno));
431                         return FALSE;
432                 }
434                 return waiting == pid &&
435                        !WIFSIGNALED(status) &&
436                        WIFEXITED(status) &&
437                        !WEXITSTATUS(status);
438         }
440         return TRUE;
443 static bool
444 start_io(struct io *io)
446         int pipefds[2] = { -1, -1 };
448         if (io->type == IO_FD)
449                 return TRUE;
451         if ((io->type == IO_RD || io->type == IO_WR) &&
452             pipe(pipefds) < 0)
453                 return FALSE;
454         else if (io->type == IO_AP)
455                 pipefds[1] = io->pipe;
457         if ((io->pid = fork())) {
458                 if (pipefds[!(io->type == IO_WR)] != -1)
459                         close(pipefds[!(io->type == IO_WR)]);
460                 if (io->pid != -1) {
461                         io->pipe = pipefds[!!(io->type == IO_WR)];
462                         return TRUE;
463                 }
465         } else {
466                 if (io->type != IO_FG) {
467                         int devnull = open("/dev/null", O_RDWR);
468                         int readfd  = io->type == IO_WR ? pipefds[0] : devnull;
469                         int writefd = (io->type == IO_RD || io->type == IO_AP)
470                                                         ? pipefds[1] : devnull;
472                         dup2(readfd,  STDIN_FILENO);
473                         dup2(writefd, STDOUT_FILENO);
474                         dup2(devnull, STDERR_FILENO);
476                         close(devnull);
477                         if (pipefds[0] != -1)
478                                 close(pipefds[0]);
479                         if (pipefds[1] != -1)
480                                 close(pipefds[1]);
481                 }
483                 if (io->dir && *io->dir && chdir(io->dir) == -1)
484                         die("Failed to change directory: %s", strerror(errno));
486                 execvp(io->argv[0], (char *const*) io->argv);
487                 die("Failed to execute program: %s", strerror(errno));
488         }
490         if (pipefds[!!(io->type == IO_WR)] != -1)
491                 close(pipefds[!!(io->type == IO_WR)]);
492         return FALSE;
495 static bool
496 run_io(struct io *io, const char **argv, const char *dir, enum io_type type)
498         init_io(io, dir, type);
499         if (!format_argv(io->argv, argv, FORMAT_NONE))
500                 return FALSE;
501         return start_io(io);
504 static int
505 run_io_do(struct io *io)
507         return start_io(io) && done_io(io);
510 static int
511 run_io_bg(const char **argv)
513         struct io io = {};
515         init_io(&io, NULL, IO_BG);
516         if (!format_argv(io.argv, argv, FORMAT_NONE))
517                 return FALSE;
518         return run_io_do(&io);
521 static bool
522 run_io_fg(const char **argv, const char *dir)
524         struct io io = {};
526         init_io(&io, dir, IO_FG);
527         if (!format_argv(io.argv, argv, FORMAT_NONE))
528                 return FALSE;
529         return run_io_do(&io);
532 static bool
533 run_io_append(const char **argv, enum format_flags flags, int fd)
535         struct io io = {};
537         init_io(&io, NULL, IO_AP);
538         io.pipe = fd;
539         if (format_argv(io.argv, argv, flags))
540                 return run_io_do(&io);
541         close(fd);
542         return FALSE;
545 static bool
546 run_io_rd(struct io *io, const char **argv, enum format_flags flags)
548         return init_io_rd(io, argv, NULL, flags) && start_io(io);
551 static bool
552 io_eof(struct io *io)
554         return io->eof;
557 static int
558 io_error(struct io *io)
560         return io->error;
563 static char *
564 io_strerror(struct io *io)
566         return strerror(io->error);
569 static bool
570 io_can_read(struct io *io)
572         struct timeval tv = { 0, 500 };
573         fd_set fds;
575         FD_ZERO(&fds);
576         FD_SET(io->pipe, &fds);
578         return select(io->pipe + 1, &fds, NULL, NULL, &tv) > 0;
581 static ssize_t
582 io_read(struct io *io, void *buf, size_t bufsize)
584         do {
585                 ssize_t readsize = read(io->pipe, buf, bufsize);
587                 if (readsize < 0 && (errno == EAGAIN || errno == EINTR))
588                         continue;
589                 else if (readsize == -1)
590                         io->error = errno;
591                 else if (readsize == 0)
592                         io->eof = 1;
593                 return readsize;
594         } while (1);
597 static char *
598 io_get(struct io *io, int c, bool can_read)
600         char *eol;
601         ssize_t readsize;
603         if (!io->buf) {
604                 io->buf = io->bufpos = malloc(BUFSIZ);
605                 if (!io->buf)
606                         return NULL;
607                 io->bufalloc = BUFSIZ;
608                 io->bufsize = 0;
609         }
611         while (TRUE) {
612                 if (io->bufsize > 0) {
613                         eol = memchr(io->bufpos, c, io->bufsize);
614                         if (eol) {
615                                 char *line = io->bufpos;
617                                 *eol = 0;
618                                 io->bufpos = eol + 1;
619                                 io->bufsize -= io->bufpos - line;
620                                 return line;
621                         }
622                 }
624                 if (io_eof(io)) {
625                         if (io->bufsize) {
626                                 io->bufpos[io->bufsize] = 0;
627                                 io->bufsize = 0;
628                                 return io->bufpos;
629                         }
630                         return NULL;
631                 }
633                 if (!can_read)
634                         return NULL;
636                 if (io->bufsize > 0 && io->bufpos > io->buf)
637                         memmove(io->buf, io->bufpos, io->bufsize);
639                 io->bufpos = io->buf;
640                 readsize = io_read(io, io->buf + io->bufsize, io->bufalloc - io->bufsize);
641                 if (io_error(io))
642                         return NULL;
643                 io->bufsize += readsize;
644         }
647 static bool
648 io_write(struct io *io, const void *buf, size_t bufsize)
650         size_t written = 0;
652         while (!io_error(io) && written < bufsize) {
653                 ssize_t size;
655                 size = write(io->pipe, buf + written, bufsize - written);
656                 if (size < 0 && (errno == EAGAIN || errno == EINTR))
657                         continue;
658                 else if (size == -1)
659                         io->error = errno;
660                 else
661                         written += size;
662         }
664         return written == bufsize;
667 static bool
668 io_read_buf(struct io *io, char buf[], size_t bufsize)
670         bool error;
672         io->buf = io->bufpos = buf;
673         io->bufalloc = bufsize;
674         error = !io_get(io, '\n', TRUE) && io_error(io);
675         io->buf = NULL;
677         return done_io(io) || error;
680 static bool
681 run_io_buf(const char **argv, char buf[], size_t bufsize)
683         struct io io = {};
685         return run_io_rd(&io, argv, FORMAT_NONE) && io_read_buf(&io, buf, bufsize);
688 static int
689 io_load(struct io *io, const char *separators,
690         int (*read_property)(char *, size_t, char *, size_t))
692         char *name;
693         int state = OK;
695         if (!start_io(io))
696                 return ERR;
698         while (state == OK && (name = io_get(io, '\n', TRUE))) {
699                 char *value;
700                 size_t namelen;
701                 size_t valuelen;
703                 name = chomp_string(name);
704                 namelen = strcspn(name, separators);
706                 if (name[namelen]) {
707                         name[namelen] = 0;
708                         value = chomp_string(name + namelen + 1);
709                         valuelen = strlen(value);
711                 } else {
712                         value = "";
713                         valuelen = 0;
714                 }
716                 state = read_property(name, namelen, value, valuelen);
717         }
719         if (state != ERR && io_error(io))
720                 state = ERR;
721         done_io(io);
723         return state;
726 static int
727 run_io_load(const char **argv, const char *separators,
728             int (*read_property)(char *, size_t, char *, size_t))
730         struct io io = {};
732         return init_io_rd(&io, argv, NULL, FORMAT_NONE)
733                 ? io_load(&io, separators, read_property) : ERR;
737 /*
738  * User requests
739  */
741 #define REQ_INFO \
742         /* XXX: Keep the view request first and in sync with views[]. */ \
743         REQ_GROUP("View switching") \
744         REQ_(VIEW_MAIN,         "Show main view"), \
745         REQ_(VIEW_DIFF,         "Show diff view"), \
746         REQ_(VIEW_LOG,          "Show log view"), \
747         REQ_(VIEW_TREE,         "Show tree view"), \
748         REQ_(VIEW_BLOB,         "Show blob view"), \
749         REQ_(VIEW_BLAME,        "Show blame view"), \
750         REQ_(VIEW_HELP,         "Show help page"), \
751         REQ_(VIEW_PAGER,        "Show pager view"), \
752         REQ_(VIEW_STATUS,       "Show status view"), \
753         REQ_(VIEW_STAGE,        "Show stage view"), \
754         \
755         REQ_GROUP("View manipulation") \
756         REQ_(ENTER,             "Enter current line and scroll"), \
757         REQ_(NEXT,              "Move to next"), \
758         REQ_(PREVIOUS,          "Move to previous"), \
759         REQ_(PARENT,            "Move to parent"), \
760         REQ_(VIEW_NEXT,         "Move focus to next view"), \
761         REQ_(REFRESH,           "Reload and refresh"), \
762         REQ_(MAXIMIZE,          "Maximize the current view"), \
763         REQ_(VIEW_CLOSE,        "Close the current view"), \
764         REQ_(QUIT,              "Close all views and quit"), \
765         \
766         REQ_GROUP("View specific requests") \
767         REQ_(STATUS_UPDATE,     "Update file status"), \
768         REQ_(STATUS_REVERT,     "Revert file changes"), \
769         REQ_(STATUS_MERGE,      "Merge file using external tool"), \
770         REQ_(STAGE_NEXT,        "Find next chunk to stage"), \
771         \
772         REQ_GROUP("Cursor navigation") \
773         REQ_(MOVE_UP,           "Move cursor one line up"), \
774         REQ_(MOVE_DOWN,         "Move cursor one line down"), \
775         REQ_(MOVE_PAGE_DOWN,    "Move cursor one page down"), \
776         REQ_(MOVE_PAGE_UP,      "Move cursor one page up"), \
777         REQ_(MOVE_FIRST_LINE,   "Move cursor to first line"), \
778         REQ_(MOVE_LAST_LINE,    "Move cursor to last line"), \
779         \
780         REQ_GROUP("Scrolling") \
781         REQ_(SCROLL_LEFT,       "Scroll two columns left"), \
782         REQ_(SCROLL_RIGHT,      "Scroll two columns right"), \
783         REQ_(SCROLL_LINE_UP,    "Scroll one line up"), \
784         REQ_(SCROLL_LINE_DOWN,  "Scroll one line down"), \
785         REQ_(SCROLL_PAGE_UP,    "Scroll one page up"), \
786         REQ_(SCROLL_PAGE_DOWN,  "Scroll one page down"), \
787         \
788         REQ_GROUP("Searching") \
789         REQ_(SEARCH,            "Search the view"), \
790         REQ_(SEARCH_BACK,       "Search backwards in the view"), \
791         REQ_(FIND_NEXT,         "Find next search match"), \
792         REQ_(FIND_PREV,         "Find previous search match"), \
793         \
794         REQ_GROUP("Option manipulation") \
795         REQ_(TOGGLE_LINENO,     "Toggle line numbers"), \
796         REQ_(TOGGLE_DATE,       "Toggle date display"), \
797         REQ_(TOGGLE_AUTHOR,     "Toggle author display"), \
798         REQ_(TOGGLE_REV_GRAPH,  "Toggle revision graph visualization"), \
799         REQ_(TOGGLE_REFS,       "Toggle reference display (tags/branches)"), \
800         \
801         REQ_GROUP("Misc") \
802         REQ_(PROMPT,            "Bring up the prompt"), \
803         REQ_(SCREEN_REDRAW,     "Redraw the screen"), \
804         REQ_(SHOW_VERSION,      "Show version information"), \
805         REQ_(STOP_LOADING,      "Stop all loading views"), \
806         REQ_(EDIT,              "Open in editor"), \
807         REQ_(NONE,              "Do nothing")
810 /* User action requests. */
811 enum request {
812 #define REQ_GROUP(help)
813 #define REQ_(req, help) REQ_##req
815         /* Offset all requests to avoid conflicts with ncurses getch values. */
816         REQ_OFFSET = KEY_MAX + 1,
817         REQ_INFO
819 #undef  REQ_GROUP
820 #undef  REQ_
821 };
823 struct request_info {
824         enum request request;
825         const char *name;
826         int namelen;
827         const char *help;
828 };
830 static const struct request_info req_info[] = {
831 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
832 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
833         REQ_INFO
834 #undef  REQ_GROUP
835 #undef  REQ_
836 };
838 static enum request
839 get_request(const char *name)
841         int namelen = strlen(name);
842         int i;
844         for (i = 0; i < ARRAY_SIZE(req_info); i++)
845                 if (req_info[i].namelen == namelen &&
846                     !string_enum_compare(req_info[i].name, name, namelen))
847                         return req_info[i].request;
849         return REQ_NONE;
853 /*
854  * Options
855  */
857 /* Option and state variables. */
858 static bool opt_date                    = TRUE;
859 static bool opt_author                  = TRUE;
860 static bool opt_line_number             = FALSE;
861 static bool opt_line_graphics           = TRUE;
862 static bool opt_rev_graph               = FALSE;
863 static bool opt_show_refs               = TRUE;
864 static int opt_num_interval             = NUMBER_INTERVAL;
865 static double opt_hscroll               = 0.50;
866 static int opt_tab_size                 = TAB_SIZE;
867 static int opt_author_cols              = AUTHOR_COLS-1;
868 static char opt_path[SIZEOF_STR]        = "";
869 static char opt_file[SIZEOF_STR]        = "";
870 static char opt_ref[SIZEOF_REF]         = "";
871 static char opt_head[SIZEOF_REF]        = "";
872 static char opt_head_rev[SIZEOF_REV]    = "";
873 static char opt_remote[SIZEOF_REF]      = "";
874 static char opt_encoding[20]            = "UTF-8";
875 static bool opt_utf8                    = TRUE;
876 static char opt_codeset[20]             = "UTF-8";
877 static iconv_t opt_iconv                = ICONV_NONE;
878 static char opt_search[SIZEOF_STR]      = "";
879 static char opt_cdup[SIZEOF_STR]        = "";
880 static char opt_prefix[SIZEOF_STR]      = "";
881 static char opt_git_dir[SIZEOF_STR]     = "";
882 static signed char opt_is_inside_work_tree      = -1; /* set to TRUE or FALSE */
883 static char opt_editor[SIZEOF_STR]      = "";
884 static FILE *opt_tty                    = NULL;
886 #define is_initial_commit()     (!*opt_head_rev)
887 #define is_head_commit(rev)     (!strcmp((rev), "HEAD") || !strcmp(opt_head_rev, (rev)))
890 /*
891  * Line-oriented content detection.
892  */
894 #define LINE_INFO \
895 LINE(DIFF_HEADER,  "diff --git ",       COLOR_YELLOW,   COLOR_DEFAULT,  0), \
896 LINE(DIFF_CHUNK,   "@@",                COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
897 LINE(DIFF_ADD,     "+",                 COLOR_GREEN,    COLOR_DEFAULT,  0), \
898 LINE(DIFF_DEL,     "-",                 COLOR_RED,      COLOR_DEFAULT,  0), \
899 LINE(DIFF_INDEX,        "index ",         COLOR_BLUE,   COLOR_DEFAULT,  0), \
900 LINE(DIFF_OLDMODE,      "old file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
901 LINE(DIFF_NEWMODE,      "new file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
902 LINE(DIFF_COPY_FROM,    "copy from",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
903 LINE(DIFF_COPY_TO,      "copy to",        COLOR_YELLOW, COLOR_DEFAULT,  0), \
904 LINE(DIFF_RENAME_FROM,  "rename from",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
905 LINE(DIFF_RENAME_TO,    "rename to",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
906 LINE(DIFF_SIMILARITY,   "similarity ",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
907 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
908 LINE(DIFF_TREE,         "diff-tree ",     COLOR_BLUE,   COLOR_DEFAULT,  0), \
909 LINE(PP_AUTHOR,    "Author: ",          COLOR_CYAN,     COLOR_DEFAULT,  0), \
910 LINE(PP_COMMIT,    "Commit: ",          COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
911 LINE(PP_MERGE,     "Merge: ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
912 LINE(PP_DATE,      "Date:   ",          COLOR_YELLOW,   COLOR_DEFAULT,  0), \
913 LINE(PP_ADATE,     "AuthorDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
914 LINE(PP_CDATE,     "CommitDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
915 LINE(PP_REFS,      "Refs: ",            COLOR_RED,      COLOR_DEFAULT,  0), \
916 LINE(COMMIT,       "commit ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
917 LINE(PARENT,       "parent ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
918 LINE(TREE,         "tree ",             COLOR_BLUE,     COLOR_DEFAULT,  0), \
919 LINE(AUTHOR,       "author ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
920 LINE(COMMITTER,    "committer ",        COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
921 LINE(SIGNOFF,      "    Signed-off-by", COLOR_YELLOW,   COLOR_DEFAULT,  0), \
922 LINE(ACKED,        "    Acked-by",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
923 LINE(DEFAULT,      "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
924 LINE(CURSOR,       "",                  COLOR_WHITE,    COLOR_GREEN,    A_BOLD), \
925 LINE(STATUS,       "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
926 LINE(DELIMITER,    "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
927 LINE(DATE,         "",                  COLOR_BLUE,     COLOR_DEFAULT,  0), \
928 LINE(MODE,         "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
929 LINE(LINE_NUMBER,  "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
930 LINE(TITLE_BLUR,   "",                  COLOR_WHITE,    COLOR_BLUE,     0), \
931 LINE(TITLE_FOCUS,  "",                  COLOR_WHITE,    COLOR_BLUE,     A_BOLD), \
932 LINE(MAIN_COMMIT,  "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
933 LINE(MAIN_TAG,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  A_BOLD), \
934 LINE(MAIN_LOCAL_TAG,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
935 LINE(MAIN_REMOTE,  "",                  COLOR_YELLOW,   COLOR_DEFAULT,  0), \
936 LINE(MAIN_TRACKED, "",                  COLOR_YELLOW,   COLOR_DEFAULT,  A_BOLD), \
937 LINE(MAIN_REF,     "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
938 LINE(MAIN_HEAD,    "",                  COLOR_CYAN,     COLOR_DEFAULT,  A_BOLD), \
939 LINE(MAIN_REVGRAPH,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
940 LINE(TREE_HEAD,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_BOLD), \
941 LINE(TREE_DIR,     "",                  COLOR_YELLOW,   COLOR_DEFAULT,  A_NORMAL), \
942 LINE(TREE_FILE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
943 LINE(STAT_HEAD,    "",                  COLOR_YELLOW,   COLOR_DEFAULT,  0), \
944 LINE(STAT_SECTION, "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
945 LINE(STAT_NONE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
946 LINE(STAT_STAGED,  "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
947 LINE(STAT_UNSTAGED,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
948 LINE(STAT_UNTRACKED,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
949 LINE(BLAME_ID,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0)
951 enum line_type {
952 #define LINE(type, line, fg, bg, attr) \
953         LINE_##type
954         LINE_INFO,
955         LINE_NONE
956 #undef  LINE
957 };
959 struct line_info {
960         const char *name;       /* Option name. */
961         int namelen;            /* Size of option name. */
962         const char *line;       /* The start of line to match. */
963         int linelen;            /* Size of string to match. */
964         int fg, bg, attr;       /* Color and text attributes for the lines. */
965 };
967 static struct line_info line_info[] = {
968 #define LINE(type, line, fg, bg, attr) \
969         { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
970         LINE_INFO
971 #undef  LINE
972 };
974 static enum line_type
975 get_line_type(const char *line)
977         int linelen = strlen(line);
978         enum line_type type;
980         for (type = 0; type < ARRAY_SIZE(line_info); type++)
981                 /* Case insensitive search matches Signed-off-by lines better. */
982                 if (linelen >= line_info[type].linelen &&
983                     !strncasecmp(line_info[type].line, line, line_info[type].linelen))
984                         return type;
986         return LINE_DEFAULT;
989 static inline int
990 get_line_attr(enum line_type type)
992         assert(type < ARRAY_SIZE(line_info));
993         return COLOR_PAIR(type) | line_info[type].attr;
996 static struct line_info *
997 get_line_info(const char *name)
999         size_t namelen = strlen(name);
1000         enum line_type type;
1002         for (type = 0; type < ARRAY_SIZE(line_info); type++)
1003                 if (namelen == line_info[type].namelen &&
1004                     !string_enum_compare(line_info[type].name, name, namelen))
1005                         return &line_info[type];
1007         return NULL;
1010 static void
1011 init_colors(void)
1013         int default_bg = line_info[LINE_DEFAULT].bg;
1014         int default_fg = line_info[LINE_DEFAULT].fg;
1015         enum line_type type;
1017         start_color();
1019         if (assume_default_colors(default_fg, default_bg) == ERR) {
1020                 default_bg = COLOR_BLACK;
1021                 default_fg = COLOR_WHITE;
1022         }
1024         for (type = 0; type < ARRAY_SIZE(line_info); type++) {
1025                 struct line_info *info = &line_info[type];
1026                 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
1027                 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
1029                 init_pair(type, fg, bg);
1030         }
1033 struct line {
1034         enum line_type type;
1036         /* State flags */
1037         unsigned int selected:1;
1038         unsigned int dirty:1;
1039         unsigned int cleareol:1;
1041         void *data;             /* User data */
1042 };
1045 /*
1046  * Keys
1047  */
1049 struct keybinding {
1050         int alias;
1051         enum request request;
1052 };
1054 static const struct keybinding default_keybindings[] = {
1055         /* View switching */
1056         { 'm',          REQ_VIEW_MAIN },
1057         { 'd',          REQ_VIEW_DIFF },
1058         { 'l',          REQ_VIEW_LOG },
1059         { 't',          REQ_VIEW_TREE },
1060         { 'f',          REQ_VIEW_BLOB },
1061         { 'B',          REQ_VIEW_BLAME },
1062         { 'p',          REQ_VIEW_PAGER },
1063         { 'h',          REQ_VIEW_HELP },
1064         { 'S',          REQ_VIEW_STATUS },
1065         { 'c',          REQ_VIEW_STAGE },
1067         /* View manipulation */
1068         { 'q',          REQ_VIEW_CLOSE },
1069         { KEY_TAB,      REQ_VIEW_NEXT },
1070         { KEY_RETURN,   REQ_ENTER },
1071         { KEY_UP,       REQ_PREVIOUS },
1072         { KEY_DOWN,     REQ_NEXT },
1073         { 'R',          REQ_REFRESH },
1074         { KEY_F(5),     REQ_REFRESH },
1075         { 'O',          REQ_MAXIMIZE },
1077         /* Cursor navigation */
1078         { 'k',          REQ_MOVE_UP },
1079         { 'j',          REQ_MOVE_DOWN },
1080         { KEY_HOME,     REQ_MOVE_FIRST_LINE },
1081         { KEY_END,      REQ_MOVE_LAST_LINE },
1082         { KEY_NPAGE,    REQ_MOVE_PAGE_DOWN },
1083         { ' ',          REQ_MOVE_PAGE_DOWN },
1084         { KEY_PPAGE,    REQ_MOVE_PAGE_UP },
1085         { 'b',          REQ_MOVE_PAGE_UP },
1086         { '-',          REQ_MOVE_PAGE_UP },
1088         /* Scrolling */
1089         { KEY_LEFT,     REQ_SCROLL_LEFT },
1090         { KEY_RIGHT,    REQ_SCROLL_RIGHT },
1091         { KEY_IC,       REQ_SCROLL_LINE_UP },
1092         { KEY_DC,       REQ_SCROLL_LINE_DOWN },
1093         { 'w',          REQ_SCROLL_PAGE_UP },
1094         { 's',          REQ_SCROLL_PAGE_DOWN },
1096         /* Searching */
1097         { '/',          REQ_SEARCH },
1098         { '?',          REQ_SEARCH_BACK },
1099         { 'n',          REQ_FIND_NEXT },
1100         { 'N',          REQ_FIND_PREV },
1102         /* Misc */
1103         { 'Q',          REQ_QUIT },
1104         { 'z',          REQ_STOP_LOADING },
1105         { 'v',          REQ_SHOW_VERSION },
1106         { 'r',          REQ_SCREEN_REDRAW },
1107         { '.',          REQ_TOGGLE_LINENO },
1108         { 'D',          REQ_TOGGLE_DATE },
1109         { 'A',          REQ_TOGGLE_AUTHOR },
1110         { 'g',          REQ_TOGGLE_REV_GRAPH },
1111         { 'F',          REQ_TOGGLE_REFS },
1112         { ':',          REQ_PROMPT },
1113         { 'u',          REQ_STATUS_UPDATE },
1114         { '!',          REQ_STATUS_REVERT },
1115         { 'M',          REQ_STATUS_MERGE },
1116         { '@',          REQ_STAGE_NEXT },
1117         { ',',          REQ_PARENT },
1118         { 'e',          REQ_EDIT },
1119 };
1121 #define KEYMAP_INFO \
1122         KEYMAP_(GENERIC), \
1123         KEYMAP_(MAIN), \
1124         KEYMAP_(DIFF), \
1125         KEYMAP_(LOG), \
1126         KEYMAP_(TREE), \
1127         KEYMAP_(BLOB), \
1128         KEYMAP_(BLAME), \
1129         KEYMAP_(PAGER), \
1130         KEYMAP_(HELP), \
1131         KEYMAP_(STATUS), \
1132         KEYMAP_(STAGE)
1134 enum keymap {
1135 #define KEYMAP_(name) KEYMAP_##name
1136         KEYMAP_INFO
1137 #undef  KEYMAP_
1138 };
1140 static const struct enum_map keymap_table[] = {
1141 #define KEYMAP_(name) ENUM_MAP(#name, KEYMAP_##name)
1142         KEYMAP_INFO
1143 #undef  KEYMAP_
1144 };
1146 #define set_keymap(map, name) map_enum(map, keymap_table, name)
1148 struct keybinding_table {
1149         struct keybinding *data;
1150         size_t size;
1151 };
1153 static struct keybinding_table keybindings[ARRAY_SIZE(keymap_table)];
1155 static void
1156 add_keybinding(enum keymap keymap, enum request request, int key)
1158         struct keybinding_table *table = &keybindings[keymap];
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;
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)
1172         size_t i;
1174         for (i = 0; i < keybindings[keymap].size; i++)
1175                 if (keybindings[keymap].data[i].alias == key)
1176                         return keybindings[keymap].data[i].request;
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;
1182         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
1183                 if (default_keybindings[i].alias == key)
1184                         return default_keybindings[i].request;
1186         return (enum request) key;
1190 struct key {
1191         const char *name;
1192         int value;
1193 };
1195 static const 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 };
1226 static int
1227 get_key_value(const char *name)
1229         int i;
1231         for (i = 0; i < ARRAY_SIZE(key_table); i++)
1232                 if (!strcasecmp(key_table[i].name, name))
1233                         return key_table[i].value;
1235         if (strlen(name) == 1 && isprint(*name))
1236                 return (int) *name;
1238         return ERR;
1241 static const char *
1242 get_key_name(int key_value)
1244         static char key_char[] = "'X'";
1245         const char *seq = NULL;
1246         int key;
1248         for (key = 0; key < ARRAY_SIZE(key_table); key++)
1249                 if (key_table[key].value == key_value)
1250                         seq = key_table[key].name;
1252         if (seq == NULL &&
1253             key_value < 127 &&
1254             isprint(key_value)) {
1255                 key_char[1] = (char) key_value;
1256                 seq = key_char;
1257         }
1259         return seq ? seq : "(no key)";
1262 static const char *
1263 get_key(enum request request)
1265         static char buf[BUFSIZ];
1266         size_t pos = 0;
1267         char *sep = "";
1268         int i;
1270         buf[pos] = 0;
1272         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
1273                 const struct keybinding *keybinding = &default_keybindings[i];
1275                 if (keybinding->request != request)
1276                         continue;
1278                 if (!string_format_from(buf, &pos, "%s%s", sep,
1279                                         get_key_name(keybinding->alias)))
1280                         return "Too many keybindings!";
1281                 sep = ", ";
1282         }
1284         return buf;
1287 struct run_request {
1288         enum keymap keymap;
1289         int key;
1290         const char *argv[SIZEOF_ARG];
1291 };
1293 static struct run_request *run_request;
1294 static size_t run_requests;
1296 static enum request
1297 add_run_request(enum keymap keymap, int key, int argc, const char **argv)
1299         struct run_request *req;
1301         if (argc >= ARRAY_SIZE(req->argv) - 1)
1302                 return REQ_NONE;
1304         req = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
1305         if (!req)
1306                 return REQ_NONE;
1308         run_request = req;
1309         req = &run_request[run_requests];
1310         req->keymap = keymap;
1311         req->key = key;
1312         req->argv[0] = NULL;
1314         if (!format_argv(req->argv, argv, FORMAT_NONE))
1315                 return REQ_NONE;
1317         return REQ_NONE + ++run_requests;
1320 static struct run_request *
1321 get_run_request(enum request request)
1323         if (request <= REQ_NONE)
1324                 return NULL;
1325         return &run_request[request - REQ_NONE - 1];
1328 static void
1329 add_builtin_run_requests(void)
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;
1344         for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1345                 enum request req;
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         }
1353 /*
1354  * User config file handling.
1355  */
1357 static int   config_lineno;
1358 static bool  config_errors;
1359 static const char *config_msg;
1361 static const struct enum_map color_map[] = {
1362 #define COLOR_MAP(name) ENUM_MAP(#name, COLOR_##name)
1363         COLOR_MAP(DEFAULT),
1364         COLOR_MAP(BLACK),
1365         COLOR_MAP(BLUE),
1366         COLOR_MAP(CYAN),
1367         COLOR_MAP(GREEN),
1368         COLOR_MAP(MAGENTA),
1369         COLOR_MAP(RED),
1370         COLOR_MAP(WHITE),
1371         COLOR_MAP(YELLOW),
1372 };
1374 static const struct enum_map attr_map[] = {
1375 #define ATTR_MAP(name) ENUM_MAP(#name, A_##name)
1376         ATTR_MAP(NORMAL),
1377         ATTR_MAP(BLINK),
1378         ATTR_MAP(BOLD),
1379         ATTR_MAP(DIM),
1380         ATTR_MAP(REVERSE),
1381         ATTR_MAP(STANDOUT),
1382         ATTR_MAP(UNDERLINE),
1383 };
1385 #define set_attribute(attr, name)       map_enum(attr, attr_map, name)
1387 static int parse_step(double *opt, const char *arg)
1389         *opt = atoi(arg);
1390         if (!strchr(arg, '%'))
1391                 return OK;
1393         /* "Shift down" so 100% and 1 does not conflict. */
1394         *opt = (*opt - 1) / 100;
1395         if (*opt >= 1.0) {
1396                 *opt = 0.99;
1397                 config_msg = "Step value larger than 100%";
1398                 return ERR;
1399         }
1400         if (*opt < 0.0) {
1401                 *opt = 1;
1402                 config_msg = "Invalid step value";
1403                 return ERR;
1404         }
1405         return OK;
1408 static int
1409 parse_int(int *opt, const char *arg, int min, int max)
1411         int value = atoi(arg);
1413         if (min <= value && value <= max) {
1414                 *opt = value;
1415                 return OK;
1416         }
1418         config_msg = "Integer value out of bound";
1419         return ERR;
1422 static bool
1423 set_color(int *color, const char *name)
1425         if (map_enum(color, color_map, name))
1426                 return TRUE;
1427         if (!prefixcmp(name, "color"))
1428                 return parse_int(color, name + 5, 0, 255) == OK;
1429         return FALSE;
1432 /* Wants: object fgcolor bgcolor [attribute] */
1433 static int
1434 option_color_command(int argc, const char *argv[])
1436         struct line_info *info;
1438         if (argc != 3 && argc != 4) {
1439                 config_msg = "Wrong number of arguments given to color command";
1440                 return ERR;
1441         }
1443         info = get_line_info(argv[0]);
1444         if (!info) {
1445                 static const struct enum_map obsolete[] = {
1446                         ENUM_MAP("main-delim",  LINE_DELIMITER),
1447                         ENUM_MAP("main-date",   LINE_DATE),
1448                         ENUM_MAP("main-author", LINE_AUTHOR),
1449                 };
1450                 int index;
1452                 if (!map_enum(&index, obsolete, argv[0])) {
1453                         config_msg = "Unknown color name";
1454                         return ERR;
1455                 }
1456                 info = &line_info[index];
1457         }
1459         if (!set_color(&info->fg, argv[1]) ||
1460             !set_color(&info->bg, argv[2])) {
1461                 config_msg = "Unknown color";
1462                 return ERR;
1463         }
1465         if (argc == 4 && !set_attribute(&info->attr, argv[3])) {
1466                 config_msg = "Unknown attribute";
1467                 return ERR;
1468         }
1470         return OK;
1473 static int parse_bool(bool *opt, const char *arg)
1475         *opt = (!strcmp(arg, "1") || !strcmp(arg, "true") || !strcmp(arg, "yes"))
1476                 ? TRUE : FALSE;
1477         return OK;
1480 static int
1481 parse_string(char *opt, const char *arg, size_t optsize)
1483         int arglen = strlen(arg);
1485         switch (arg[0]) {
1486         case '\"':
1487         case '\'':
1488                 if (arglen == 1 || arg[arglen - 1] != arg[0]) {
1489                         config_msg = "Unmatched quotation";
1490                         return ERR;
1491                 }
1492                 arg += 1; arglen -= 2;
1493         default:
1494                 string_ncopy_do(opt, optsize, arg, arglen);
1495                 return OK;
1496         }
1499 /* Wants: name = value */
1500 static int
1501 option_set_command(int argc, const char *argv[])
1503         if (argc != 3) {
1504                 config_msg = "Wrong number of arguments given to set command";
1505                 return ERR;
1506         }
1508         if (strcmp(argv[1], "=")) {
1509                 config_msg = "No value assigned";
1510                 return ERR;
1511         }
1513         if (!strcmp(argv[0], "show-author"))
1514                 return parse_bool(&opt_author, argv[2]);
1516         if (!strcmp(argv[0], "show-date"))
1517                 return parse_bool(&opt_date, argv[2]);
1519         if (!strcmp(argv[0], "show-rev-graph"))
1520                 return parse_bool(&opt_rev_graph, argv[2]);
1522         if (!strcmp(argv[0], "show-refs"))
1523                 return parse_bool(&opt_show_refs, argv[2]);
1525         if (!strcmp(argv[0], "show-line-numbers"))
1526                 return parse_bool(&opt_line_number, argv[2]);
1528         if (!strcmp(argv[0], "line-graphics"))
1529                 return parse_bool(&opt_line_graphics, argv[2]);
1531         if (!strcmp(argv[0], "line-number-interval"))
1532                 return parse_int(&opt_num_interval, argv[2], 1, 1024);
1534         if (!strcmp(argv[0], "author-width"))
1535                 return parse_int(&opt_author_cols, argv[2], 0, 1024);
1537         if (!strcmp(argv[0], "horizontal-scroll"))
1538                 return parse_step(&opt_hscroll, argv[2]);
1540         if (!strcmp(argv[0], "tab-size"))
1541                 return parse_int(&opt_tab_size, argv[2], 1, 1024);
1543         if (!strcmp(argv[0], "commit-encoding"))
1544                 return parse_string(opt_encoding, argv[2], sizeof(opt_encoding));
1546         config_msg = "Unknown variable name";
1547         return ERR;
1550 /* Wants: mode request key */
1551 static int
1552 option_bind_command(int argc, const char *argv[])
1554         enum request request;
1555         int keymap;
1556         int key;
1558         if (argc < 3) {
1559                 config_msg = "Wrong number of arguments given to bind command";
1560                 return ERR;
1561         }
1563         if (set_keymap(&keymap, argv[0]) == ERR) {
1564                 config_msg = "Unknown key map";
1565                 return ERR;
1566         }
1568         key = get_key_value(argv[1]);
1569         if (key == ERR) {
1570                 config_msg = "Unknown key";
1571                 return ERR;
1572         }
1574         request = get_request(argv[2]);
1575         if (request == REQ_NONE) {
1576                 static const struct enum_map obsolete[] = {
1577                         ENUM_MAP("cherry-pick",         REQ_NONE),
1578                         ENUM_MAP("screen-resize",       REQ_NONE),
1579                         ENUM_MAP("tree-parent",         REQ_PARENT),
1580                 };
1581                 int alias;
1583                 if (map_enum(&alias, obsolete, argv[2])) {
1584                         if (alias != REQ_NONE)
1585                                 add_keybinding(keymap, alias, key);
1586                         config_msg = "Obsolete request name";
1587                         return ERR;
1588                 }
1589         }
1590         if (request == REQ_NONE && *argv[2]++ == '!')
1591                 request = add_run_request(keymap, key, argc - 2, argv + 2);
1592         if (request == REQ_NONE) {
1593                 config_msg = "Unknown request name";
1594                 return ERR;
1595         }
1597         add_keybinding(keymap, request, key);
1599         return OK;
1602 static int
1603 set_option(const char *opt, char *value)
1605         const char *argv[SIZEOF_ARG];
1606         int argc = 0;
1608         if (!argv_from_string(argv, &argc, value)) {
1609                 config_msg = "Too many option arguments";
1610                 return ERR;
1611         }
1613         if (!strcmp(opt, "color"))
1614                 return option_color_command(argc, argv);
1616         if (!strcmp(opt, "set"))
1617                 return option_set_command(argc, argv);
1619         if (!strcmp(opt, "bind"))
1620                 return option_bind_command(argc, argv);
1622         config_msg = "Unknown option command";
1623         return ERR;
1626 static int
1627 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1629         int status = OK;
1631         config_lineno++;
1632         config_msg = "Internal error";
1634         /* Check for comment markers, since read_properties() will
1635          * only ensure opt and value are split at first " \t". */
1636         optlen = strcspn(opt, "#");
1637         if (optlen == 0)
1638                 return OK;
1640         if (opt[optlen] != 0) {
1641                 config_msg = "No option value";
1642                 status = ERR;
1644         }  else {
1645                 /* Look for comment endings in the value. */
1646                 size_t len = strcspn(value, "#");
1648                 if (len < valuelen) {
1649                         valuelen = len;
1650                         value[valuelen] = 0;
1651                 }
1653                 status = set_option(opt, value);
1654         }
1656         if (status == ERR) {
1657                 warn("Error on line %d, near '%.*s': %s",
1658                      config_lineno, (int) optlen, opt, config_msg);
1659                 config_errors = TRUE;
1660         }
1662         /* Always keep going if errors are encountered. */
1663         return OK;
1666 static void
1667 load_option_file(const char *path)
1669         struct io io = {};
1671         /* It's OK that the file doesn't exist. */
1672         if (!io_open(&io, path))
1673                 return;
1675         config_lineno = 0;
1676         config_errors = FALSE;
1678         if (io_load(&io, " \t", read_option) == ERR ||
1679             config_errors == TRUE)
1680                 warn("Errors while loading %s.", path);
1683 static int
1684 load_options(void)
1686         const char *home = getenv("HOME");
1687         const char *tigrc_user = getenv("TIGRC_USER");
1688         const char *tigrc_system = getenv("TIGRC_SYSTEM");
1689         char buf[SIZEOF_STR];
1691         add_builtin_run_requests();
1693         if (!tigrc_system)
1694                 tigrc_system = SYSCONFDIR "/tigrc";
1695         load_option_file(tigrc_system);
1697         if (!tigrc_user) {
1698                 if (!home || !string_format(buf, "%s/.tigrc", home))
1699                         return ERR;
1700                 tigrc_user = buf;
1701         }
1702         load_option_file(tigrc_user);
1704         return OK;
1708 /*
1709  * The viewer
1710  */
1712 struct view;
1713 struct view_ops;
1715 /* The display array of active views and the index of the current view. */
1716 static struct view *display[2];
1717 static unsigned int current_view;
1719 #define foreach_displayed_view(view, i) \
1720         for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1722 #define displayed_views()       (display[1] != NULL ? 2 : 1)
1724 /* Current head and commit ID */
1725 static char ref_blob[SIZEOF_REF]        = "";
1726 static char ref_commit[SIZEOF_REF]      = "HEAD";
1727 static char ref_head[SIZEOF_REF]        = "HEAD";
1729 struct view {
1730         const char *name;       /* View name */
1731         const char *cmd_env;    /* Command line set via environment */
1732         const char *id;         /* Points to either of ref_{head,commit,blob} */
1734         struct view_ops *ops;   /* View operations */
1736         enum keymap keymap;     /* What keymap does this view have */
1737         bool git_dir;           /* Whether the view requires a git directory. */
1739         char ref[SIZEOF_REF];   /* Hovered commit reference */
1740         char vid[SIZEOF_REF];   /* View ID. Set to id member when updating. */
1742         int height, width;      /* The width and height of the main window */
1743         WINDOW *win;            /* The main window */
1744         WINDOW *title;          /* The title window living below the main window */
1746         /* Navigation */
1747         unsigned long offset;   /* Offset of the window top */
1748         unsigned long yoffset;  /* Offset from the window side. */
1749         unsigned long lineno;   /* Current line number */
1750         unsigned long p_offset; /* Previous offset of the window top */
1751         unsigned long p_yoffset;/* Previous offset from the window side */
1752         unsigned long p_lineno; /* Previous current line number */
1753         bool p_restore;         /* Should the previous position be restored. */
1755         /* Searching */
1756         char grep[SIZEOF_STR];  /* Search string */
1757         regex_t *regex;         /* Pre-compiled regexp */
1759         /* If non-NULL, points to the view that opened this view. If this view
1760          * is closed tig will switch back to the parent view. */
1761         struct view *parent;
1763         /* Buffering */
1764         size_t lines;           /* Total number of lines */
1765         struct line *line;      /* Line index */
1766         size_t line_alloc;      /* Total number of allocated lines */
1767         unsigned int digits;    /* Number of digits in the lines member. */
1769         /* Drawing */
1770         struct line *curline;   /* Line currently being drawn. */
1771         enum line_type curtype; /* Attribute currently used for drawing. */
1772         unsigned long col;      /* Column when drawing. */
1773         bool has_scrolled;      /* View was scrolled. */
1775         /* Loading */
1776         struct io io;
1777         struct io *pipe;
1778         time_t start_time;
1779         time_t update_secs;
1780 };
1782 struct view_ops {
1783         /* What type of content being displayed. Used in the title bar. */
1784         const char *type;
1785         /* Default command arguments. */
1786         const char **argv;
1787         /* Open and reads in all view content. */
1788         bool (*open)(struct view *view);
1789         /* Read one line; updates view->line. */
1790         bool (*read)(struct view *view, char *data);
1791         /* Draw one line; @lineno must be < view->height. */
1792         bool (*draw)(struct view *view, struct line *line, unsigned int lineno);
1793         /* Depending on view handle a special requests. */
1794         enum request (*request)(struct view *view, enum request request, struct line *line);
1795         /* Search for regexp in a line. */
1796         bool (*grep)(struct view *view, struct line *line);
1797         /* Select line */
1798         void (*select)(struct view *view, struct line *line);
1799 };
1801 static struct view_ops blame_ops;
1802 static struct view_ops blob_ops;
1803 static struct view_ops diff_ops;
1804 static struct view_ops help_ops;
1805 static struct view_ops log_ops;
1806 static struct view_ops main_ops;
1807 static struct view_ops pager_ops;
1808 static struct view_ops stage_ops;
1809 static struct view_ops status_ops;
1810 static struct view_ops tree_ops;
1812 #define VIEW_STR(name, env, ref, ops, map, git) \
1813         { name, #env, ref, ops, map, git }
1815 #define VIEW_(id, name, ops, git, ref) \
1816         VIEW_STR(name, TIG_##id##_CMD, ref, ops, KEYMAP_##id, git)
1819 static struct view views[] = {
1820         VIEW_(MAIN,   "main",   &main_ops,   TRUE,  ref_head),
1821         VIEW_(DIFF,   "diff",   &diff_ops,   TRUE,  ref_commit),
1822         VIEW_(LOG,    "log",    &log_ops,    TRUE,  ref_head),
1823         VIEW_(TREE,   "tree",   &tree_ops,   TRUE,  ref_commit),
1824         VIEW_(BLOB,   "blob",   &blob_ops,   TRUE,  ref_blob),
1825         VIEW_(BLAME,  "blame",  &blame_ops,  TRUE,  ref_commit),
1826         VIEW_(HELP,   "help",   &help_ops,   FALSE, ""),
1827         VIEW_(PAGER,  "pager",  &pager_ops,  FALSE, "stdin"),
1828         VIEW_(STATUS, "status", &status_ops, TRUE,  ""),
1829         VIEW_(STAGE,  "stage",  &stage_ops,  TRUE,  ""),
1830 };
1832 #define VIEW(req)       (&views[(req) - REQ_OFFSET - 1])
1833 #define VIEW_REQ(view)  ((view) - views + REQ_OFFSET + 1)
1835 #define foreach_view(view, i) \
1836         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1838 #define view_is_displayed(view) \
1839         (view == display[0] || view == display[1])
1842 enum line_graphic {
1843         LINE_GRAPHIC_VLINE
1844 };
1846 static chtype line_graphics[] = {
1847         /* LINE_GRAPHIC_VLINE: */ '|'
1848 };
1850 static inline void
1851 set_view_attr(struct view *view, enum line_type type)
1853         if (!view->curline->selected && view->curtype != type) {
1854                 wattrset(view->win, get_line_attr(type));
1855                 wchgat(view->win, -1, 0, type, NULL);
1856                 view->curtype = type;
1857         }
1860 static int
1861 draw_chars(struct view *view, enum line_type type, const char *string,
1862            int max_len, bool use_tilde)
1864         int len = 0;
1865         int col = 0;
1866         int trimmed = FALSE;
1867         size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
1869         if (max_len <= 0)
1870                 return 0;
1872         if (opt_utf8) {
1873                 len = utf8_length(&string, skip, &col, max_len, &trimmed, use_tilde);
1874         } else {
1875                 col = len = strlen(string);
1876                 if (len > max_len) {
1877                         if (use_tilde) {
1878                                 max_len -= 1;
1879                         }
1880                         col = len = max_len;
1881                         trimmed = TRUE;
1882                 }
1883         }
1885         set_view_attr(view, type);
1886         if (len > 0)
1887                 waddnstr(view->win, string, len);
1888         if (trimmed && use_tilde) {
1889                 set_view_attr(view, LINE_DELIMITER);
1890                 waddch(view->win, '~');
1891                 col++;
1892         }
1894         return col;
1897 static int
1898 draw_space(struct view *view, enum line_type type, int max, int spaces)
1900         static char space[] = "                    ";
1901         int col = 0;
1903         spaces = MIN(max, spaces);
1905         while (spaces > 0) {
1906                 int len = MIN(spaces, sizeof(space) - 1);
1908                 col += draw_chars(view, type, space, len, FALSE);
1909                 spaces -= len;
1910         }
1912         return col;
1915 static bool
1916 draw_text(struct view *view, enum line_type type, const char *string, bool trim)
1918         view->col += draw_chars(view, type, string, view->width + view->yoffset - view->col, trim);
1919         return view->width + view->yoffset <= view->col;
1922 static bool
1923 draw_graphic(struct view *view, enum line_type type, chtype graphic[], size_t size)
1925         size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
1926         int max = view->width + view->yoffset - view->col;
1927         int i;
1929         if (max < size)
1930                 size = max;
1932         set_view_attr(view, type);
1933         /* Using waddch() instead of waddnstr() ensures that
1934          * they'll be rendered correctly for the cursor line. */
1935         for (i = skip; i < size; i++)
1936                 waddch(view->win, graphic[i]);
1938         view->col += size;
1939         if (size < max && skip <= size)
1940                 waddch(view->win, ' ');
1941         view->col++;
1943         return view->width + view->yoffset <= view->col;
1946 static bool
1947 draw_field(struct view *view, enum line_type type, const char *text, int len, bool trim)
1949         int max = MIN(view->width + view->yoffset - view->col, len);
1950         int col;
1952         if (text)
1953                 col = draw_chars(view, type, text, max - 1, trim);
1954         else
1955                 col = draw_space(view, type, max - 1, max - 1);
1957         view->col += col;
1958         view->col += draw_space(view, LINE_DEFAULT, max - col, max - col);
1959         return view->width + view->yoffset <= view->col;
1962 static bool
1963 draw_date(struct view *view, struct tm *time)
1965         char buf[DATE_COLS];
1966         char *date;
1967         int timelen = 0;
1969         if (time)
1970                 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, time);
1971         date = timelen ? buf : NULL;
1973         return draw_field(view, LINE_DATE, date, DATE_COLS, FALSE);
1976 static bool
1977 draw_author(struct view *view, const char *author)
1979         bool trim = opt_author_cols == 0 || opt_author_cols > 5 || !author;
1981         if (!trim) {
1982                 static char initials[10];
1983                 size_t pos;
1985 #define is_initial_sep(c) (isspace(c) || ispunct(c) || (c) == '@')
1987                 memset(initials, 0, sizeof(initials));
1988                 for (pos = 0; *author && pos < opt_author_cols - 1; author++, pos++) {
1989                         while (is_initial_sep(*author))
1990                                 author++;
1991                         strncpy(&initials[pos], author, sizeof(initials) - 1 - pos);
1992                         while (*author && !is_initial_sep(author[1]))
1993                                 author++;
1994                 }
1996                 author = initials;
1997         }
1999         return draw_field(view, LINE_AUTHOR, author, opt_author_cols, trim);
2002 static bool
2003 draw_mode(struct view *view, mode_t mode)
2005         const char *str;
2007         if (S_ISDIR(mode))
2008                 str = "drwxr-xr-x";
2009         else if (S_ISLNK(mode))
2010                 str = "lrwxrwxrwx";
2011         else if (S_ISGITLINK(mode))
2012                 str = "m---------";
2013         else if (S_ISREG(mode) && mode & S_IXUSR)
2014                 str = "-rwxr-xr-x";
2015         else if (S_ISREG(mode))
2016                 str = "-rw-r--r--";
2017         else
2018                 str = "----------";
2020         return draw_field(view, LINE_MODE, str, STRING_SIZE("-rw-r--r-- "), FALSE);
2023 static bool
2024 draw_lineno(struct view *view, unsigned int lineno)
2026         char number[10];
2027         int digits3 = view->digits < 3 ? 3 : view->digits;
2028         int max = MIN(view->width + view->yoffset - view->col, digits3);
2029         char *text = NULL;
2031         lineno += view->offset + 1;
2032         if (lineno == 1 || (lineno % opt_num_interval) == 0) {
2033                 static char fmt[] = "%1ld";
2035                 fmt[1] = '0' + (view->digits <= 9 ? digits3 : 1);
2036                 if (string_format(number, fmt, lineno))
2037                         text = number;
2038         }
2039         if (text)
2040                 view->col += draw_chars(view, LINE_LINE_NUMBER, text, max, TRUE);
2041         else
2042                 view->col += draw_space(view, LINE_LINE_NUMBER, max, digits3);
2043         return draw_graphic(view, LINE_DEFAULT, &line_graphics[LINE_GRAPHIC_VLINE], 1);
2046 static bool
2047 draw_view_line(struct view *view, unsigned int lineno)
2049         struct line *line;
2050         bool selected = (view->offset + lineno == view->lineno);
2052         assert(view_is_displayed(view));
2054         if (view->offset + lineno >= view->lines)
2055                 return FALSE;
2057         line = &view->line[view->offset + lineno];
2059         wmove(view->win, lineno, 0);
2060         if (line->cleareol)
2061                 wclrtoeol(view->win);
2062         view->col = 0;
2063         view->curline = line;
2064         view->curtype = LINE_NONE;
2065         line->selected = FALSE;
2066         line->dirty = line->cleareol = 0;
2068         if (selected) {
2069                 set_view_attr(view, LINE_CURSOR);
2070                 line->selected = TRUE;
2071                 view->ops->select(view, line);
2072         }
2074         return view->ops->draw(view, line, lineno);
2077 static void
2078 redraw_view_dirty(struct view *view)
2080         bool dirty = FALSE;
2081         int lineno;
2083         for (lineno = 0; lineno < view->height; lineno++) {
2084                 if (view->offset + lineno >= view->lines)
2085                         break;
2086                 if (!view->line[view->offset + lineno].dirty)
2087                         continue;
2088                 dirty = TRUE;
2089                 if (!draw_view_line(view, lineno))
2090                         break;
2091         }
2093         if (!dirty)
2094                 return;
2095         wnoutrefresh(view->win);
2098 static void
2099 redraw_view_from(struct view *view, int lineno)
2101         assert(0 <= lineno && lineno < view->height);
2103         for (; lineno < view->height; lineno++) {
2104                 if (!draw_view_line(view, lineno))
2105                         break;
2106         }
2108         wnoutrefresh(view->win);
2111 static void
2112 redraw_view(struct view *view)
2114         werase(view->win);
2115         redraw_view_from(view, 0);
2119 static void
2120 update_view_title(struct view *view)
2122         char buf[SIZEOF_STR];
2123         char state[SIZEOF_STR];
2124         size_t bufpos = 0, statelen = 0;
2126         assert(view_is_displayed(view));
2128         if (view != VIEW(REQ_VIEW_STATUS) && view->lines) {
2129                 unsigned int view_lines = view->offset + view->height;
2130                 unsigned int lines = view->lines
2131                                    ? MIN(view_lines, view->lines) * 100 / view->lines
2132                                    : 0;
2134                 string_format_from(state, &statelen, " - %s %d of %d (%d%%)",
2135                                    view->ops->type,
2136                                    view->lineno + 1,
2137                                    view->lines,
2138                                    lines);
2140         }
2142         if (view->pipe) {
2143                 time_t secs = time(NULL) - view->start_time;
2145                 /* Three git seconds are a long time ... */
2146                 if (secs > 2)
2147                         string_format_from(state, &statelen, " loading %lds", secs);
2148         }
2150         string_format_from(buf, &bufpos, "[%s]", view->name);
2151         if (*view->ref && bufpos < view->width) {
2152                 size_t refsize = strlen(view->ref);
2153                 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
2155                 if (minsize < view->width)
2156                         refsize = view->width - minsize + 7;
2157                 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
2158         }
2160         if (statelen && bufpos < view->width) {
2161                 string_format_from(buf, &bufpos, "%s", state);
2162         }
2164         if (view == display[current_view])
2165                 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
2166         else
2167                 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
2169         mvwaddnstr(view->title, 0, 0, buf, bufpos);
2170         wclrtoeol(view->title);
2171         wnoutrefresh(view->title);
2174 static void
2175 resize_display(void)
2177         int offset, i;
2178         struct view *base = display[0];
2179         struct view *view = display[1] ? display[1] : display[0];
2181         /* Setup window dimensions */
2183         getmaxyx(stdscr, base->height, base->width);
2185         /* Make room for the status window. */
2186         base->height -= 1;
2188         if (view != base) {
2189                 /* Horizontal split. */
2190                 view->width   = base->width;
2191                 view->height  = SCALE_SPLIT_VIEW(base->height);
2192                 base->height -= view->height;
2194                 /* Make room for the title bar. */
2195                 view->height -= 1;
2196         }
2198         /* Make room for the title bar. */
2199         base->height -= 1;
2201         offset = 0;
2203         foreach_displayed_view (view, i) {
2204                 if (!view->win) {
2205                         view->win = newwin(view->height, 0, offset, 0);
2206                         if (!view->win)
2207                                 die("Failed to create %s view", view->name);
2209                         scrollok(view->win, FALSE);
2211                         view->title = newwin(1, 0, offset + view->height, 0);
2212                         if (!view->title)
2213                                 die("Failed to create title window");
2215                 } else {
2216                         wresize(view->win, view->height, view->width);
2217                         mvwin(view->win,   offset, 0);
2218                         mvwin(view->title, offset + view->height, 0);
2219                 }
2221                 offset += view->height + 1;
2222         }
2225 static void
2226 redraw_display(bool clear)
2228         struct view *view;
2229         int i;
2231         foreach_displayed_view (view, i) {
2232                 if (clear)
2233                         wclear(view->win);
2234                 redraw_view(view);
2235                 update_view_title(view);
2236         }
2239 static void
2240 toggle_view_option(bool *option, const char *help)
2242         *option = !*option;
2243         redraw_display(FALSE);
2244         report("%sabling %s", *option ? "En" : "Dis", help);
2247 static void
2248 maximize_view(struct view *view)
2250         memset(display, 0, sizeof(display));
2251         current_view = 0;
2252         display[current_view] = view;
2253         resize_display();
2254         redraw_display(FALSE);
2255         report("");
2259 /*
2260  * Navigation
2261  */
2263 static bool
2264 goto_view_line(struct view *view, unsigned long offset, unsigned long lineno)
2266         if (lineno >= view->lines)
2267                 lineno = view->lines > 0 ? view->lines - 1 : 0;
2269         if (offset > lineno || offset + view->height <= lineno) {
2270                 unsigned long half = view->height / 2;
2272                 if (lineno > half)
2273                         offset = lineno - half;
2274                 else
2275                         offset = 0;
2276         }
2278         if (offset != view->offset || lineno != view->lineno) {
2279                 view->offset = offset;
2280                 view->lineno = lineno;
2281                 return TRUE;
2282         }
2284         return FALSE;
2287 static int
2288 apply_step(double step, int value)
2290         if (step >= 1)
2291                 return (int) step;
2292         value *= step + 0.01;
2293         return value ? value : 1;
2296 /* Scrolling backend */
2297 static void
2298 do_scroll_view(struct view *view, int lines)
2300         bool redraw_current_line = FALSE;
2302         /* The rendering expects the new offset. */
2303         view->offset += lines;
2305         assert(0 <= view->offset && view->offset < view->lines);
2306         assert(lines);
2308         /* Move current line into the view. */
2309         if (view->lineno < view->offset) {
2310                 view->lineno = view->offset;
2311                 redraw_current_line = TRUE;
2312         } else if (view->lineno >= view->offset + view->height) {
2313                 view->lineno = view->offset + view->height - 1;
2314                 redraw_current_line = TRUE;
2315         }
2317         assert(view->offset <= view->lineno && view->lineno < view->lines);
2319         /* Redraw the whole screen if scrolling is pointless. */
2320         if (view->height < ABS(lines)) {
2321                 redraw_view(view);
2323         } else {
2324                 int line = lines > 0 ? view->height - lines : 0;
2325                 int end = line + ABS(lines);
2327                 scrollok(view->win, TRUE);
2328                 wscrl(view->win, lines);
2329                 scrollok(view->win, FALSE);
2331                 while (line < end && draw_view_line(view, line))
2332                         line++;
2334                 if (redraw_current_line)
2335                         draw_view_line(view, view->lineno - view->offset);
2336                 wnoutrefresh(view->win);
2337         }
2339         view->has_scrolled = TRUE;
2340         report("");
2343 /* Scroll frontend */
2344 static void
2345 scroll_view(struct view *view, enum request request)
2347         int lines = 1;
2349         assert(view_is_displayed(view));
2351         switch (request) {
2352         case REQ_SCROLL_LEFT:
2353                 if (view->yoffset == 0) {
2354                         report("Cannot scroll beyond the first column");
2355                         return;
2356                 }
2357                 if (view->yoffset <= apply_step(opt_hscroll, view->width))
2358                         view->yoffset = 0;
2359                 else
2360                         view->yoffset -= apply_step(opt_hscroll, view->width);
2361                 redraw_view_from(view, 0);
2362                 report("");
2363                 return;
2364         case REQ_SCROLL_RIGHT:
2365                 view->yoffset += apply_step(opt_hscroll, view->width);
2366                 redraw_view(view);
2367                 report("");
2368                 return;
2369         case REQ_SCROLL_PAGE_DOWN:
2370                 lines = view->height;
2371         case REQ_SCROLL_LINE_DOWN:
2372                 if (view->offset + lines > view->lines)
2373                         lines = view->lines - view->offset;
2375                 if (lines == 0 || view->offset + view->height >= view->lines) {
2376                         report("Cannot scroll beyond the last line");
2377                         return;
2378                 }
2379                 break;
2381         case REQ_SCROLL_PAGE_UP:
2382                 lines = view->height;
2383         case REQ_SCROLL_LINE_UP:
2384                 if (lines > view->offset)
2385                         lines = view->offset;
2387                 if (lines == 0) {
2388                         report("Cannot scroll beyond the first line");
2389                         return;
2390                 }
2392                 lines = -lines;
2393                 break;
2395         default:
2396                 die("request %d not handled in switch", request);
2397         }
2399         do_scroll_view(view, lines);
2402 /* Cursor moving */
2403 static void
2404 move_view(struct view *view, enum request request)
2406         int scroll_steps = 0;
2407         int steps;
2409         switch (request) {
2410         case REQ_MOVE_FIRST_LINE:
2411                 steps = -view->lineno;
2412                 break;
2414         case REQ_MOVE_LAST_LINE:
2415                 steps = view->lines - view->lineno - 1;
2416                 break;
2418         case REQ_MOVE_PAGE_UP:
2419                 steps = view->height > view->lineno
2420                       ? -view->lineno : -view->height;
2421                 break;
2423         case REQ_MOVE_PAGE_DOWN:
2424                 steps = view->lineno + view->height >= view->lines
2425                       ? view->lines - view->lineno - 1 : view->height;
2426                 break;
2428         case REQ_MOVE_UP:
2429                 steps = -1;
2430                 break;
2432         case REQ_MOVE_DOWN:
2433                 steps = 1;
2434                 break;
2436         default:
2437                 die("request %d not handled in switch", request);
2438         }
2440         if (steps <= 0 && view->lineno == 0) {
2441                 report("Cannot move beyond the first line");
2442                 return;
2444         } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
2445                 report("Cannot move beyond the last line");
2446                 return;
2447         }
2449         /* Move the current line */
2450         view->lineno += steps;
2451         assert(0 <= view->lineno && view->lineno < view->lines);
2453         /* Check whether the view needs to be scrolled */
2454         if (view->lineno < view->offset ||
2455             view->lineno >= view->offset + view->height) {
2456                 scroll_steps = steps;
2457                 if (steps < 0 && -steps > view->offset) {
2458                         scroll_steps = -view->offset;
2460                 } else if (steps > 0) {
2461                         if (view->lineno == view->lines - 1 &&
2462                             view->lines > view->height) {
2463                                 scroll_steps = view->lines - view->offset - 1;
2464                                 if (scroll_steps >= view->height)
2465                                         scroll_steps -= view->height - 1;
2466                         }
2467                 }
2468         }
2470         if (!view_is_displayed(view)) {
2471                 view->offset += scroll_steps;
2472                 assert(0 <= view->offset && view->offset < view->lines);
2473                 view->ops->select(view, &view->line[view->lineno]);
2474                 return;
2475         }
2477         /* Repaint the old "current" line if we be scrolling */
2478         if (ABS(steps) < view->height)
2479                 draw_view_line(view, view->lineno - steps - view->offset);
2481         if (scroll_steps) {
2482                 do_scroll_view(view, scroll_steps);
2483                 return;
2484         }
2486         /* Draw the current line */
2487         draw_view_line(view, view->lineno - view->offset);
2489         wnoutrefresh(view->win);
2490         report("");
2494 /*
2495  * Searching
2496  */
2498 static void search_view(struct view *view, enum request request);
2500 static void
2501 select_view_line(struct view *view, unsigned long lineno)
2503         unsigned long old_lineno = view->lineno;
2504         unsigned long old_offset = view->offset;
2506         if (goto_view_line(view, view->offset, lineno)) {
2507                 if (view_is_displayed(view)) {
2508                         if (old_offset != view->offset) {
2509                                 redraw_view(view);
2510                         } else {
2511                                 draw_view_line(view, old_lineno - view->offset);
2512                                 draw_view_line(view, view->lineno - view->offset);
2513                                 wnoutrefresh(view->win);
2514                         }
2515                 } else {
2516                         view->ops->select(view, &view->line[view->lineno]);
2517                 }
2518         }
2521 static void
2522 find_next(struct view *view, enum request request)
2524         unsigned long lineno = view->lineno;
2525         int direction;
2527         if (!*view->grep) {
2528                 if (!*opt_search)
2529                         report("No previous search");
2530                 else
2531                         search_view(view, request);
2532                 return;
2533         }
2535         switch (request) {
2536         case REQ_SEARCH:
2537         case REQ_FIND_NEXT:
2538                 direction = 1;
2539                 break;
2541         case REQ_SEARCH_BACK:
2542         case REQ_FIND_PREV:
2543                 direction = -1;
2544                 break;
2546         default:
2547                 return;
2548         }
2550         if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
2551                 lineno += direction;
2553         /* Note, lineno is unsigned long so will wrap around in which case it
2554          * will become bigger than view->lines. */
2555         for (; lineno < view->lines; lineno += direction) {
2556                 if (view->ops->grep(view, &view->line[lineno])) {
2557                         select_view_line(view, lineno);
2558                         report("Line %ld matches '%s'", lineno + 1, view->grep);
2559                         return;
2560                 }
2561         }
2563         report("No match found for '%s'", view->grep);
2566 static void
2567 search_view(struct view *view, enum request request)
2569         int regex_err;
2571         if (view->regex) {
2572                 regfree(view->regex);
2573                 *view->grep = 0;
2574         } else {
2575                 view->regex = calloc(1, sizeof(*view->regex));
2576                 if (!view->regex)
2577                         return;
2578         }
2580         regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
2581         if (regex_err != 0) {
2582                 char buf[SIZEOF_STR] = "unknown error";
2584                 regerror(regex_err, view->regex, buf, sizeof(buf));
2585                 report("Search failed: %s", buf);
2586                 return;
2587         }
2589         string_copy(view->grep, opt_search);
2591         find_next(view, request);
2594 /*
2595  * Incremental updating
2596  */
2598 static void
2599 reset_view(struct view *view)
2601         int i;
2603         for (i = 0; i < view->lines; i++)
2604                 free(view->line[i].data);
2605         free(view->line);
2607         view->p_offset = view->offset;
2608         view->p_yoffset = view->yoffset;
2609         view->p_lineno = view->lineno;
2611         view->line = NULL;
2612         view->offset = 0;
2613         view->yoffset = 0;
2614         view->lines  = 0;
2615         view->lineno = 0;
2616         view->line_alloc = 0;
2617         view->vid[0] = 0;
2618         view->update_secs = 0;
2621 static void
2622 free_argv(const char *argv[])
2624         int argc;
2626         for (argc = 0; argv[argc]; argc++)
2627                 free((void *) argv[argc]);
2630 static bool
2631 format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags flags)
2633         char buf[SIZEOF_STR];
2634         int argc;
2635         bool noreplace = flags == FORMAT_NONE;
2637         free_argv(dst_argv);
2639         for (argc = 0; src_argv[argc]; argc++) {
2640                 const char *arg = src_argv[argc];
2641                 size_t bufpos = 0;
2643                 while (arg) {
2644                         char *next = strstr(arg, "%(");
2645                         int len = next - arg;
2646                         const char *value;
2648                         if (!next || noreplace) {
2649                                 if (flags == FORMAT_DASH && !strcmp(arg, "--"))
2650                                         noreplace = TRUE;
2651                                 len = strlen(arg);
2652                                 value = "";
2654                         } else if (!prefixcmp(next, "%(directory)")) {
2655                                 value = opt_path;
2657                         } else if (!prefixcmp(next, "%(file)")) {
2658                                 value = opt_file;
2660                         } else if (!prefixcmp(next, "%(ref)")) {
2661                                 value = *opt_ref ? opt_ref : "HEAD";
2663                         } else if (!prefixcmp(next, "%(head)")) {
2664                                 value = ref_head;
2666                         } else if (!prefixcmp(next, "%(commit)")) {
2667                                 value = ref_commit;
2669                         } else if (!prefixcmp(next, "%(blob)")) {
2670                                 value = ref_blob;
2672                         } else {
2673                                 report("Unknown replacement: `%s`", next);
2674                                 return FALSE;
2675                         }
2677                         if (!string_format_from(buf, &bufpos, "%.*s%s", len, arg, value))
2678                                 return FALSE;
2680                         arg = next && !noreplace ? strchr(next, ')') + 1 : NULL;
2681                 }
2683                 dst_argv[argc] = strdup(buf);
2684                 if (!dst_argv[argc])
2685                         break;
2686         }
2688         dst_argv[argc] = NULL;
2690         return src_argv[argc] == NULL;
2693 static bool
2694 restore_view_position(struct view *view)
2696         if (!view->p_restore || (view->pipe && view->lines <= view->p_lineno))
2697                 return FALSE;
2699         /* Changing the view position cancels the restoring. */
2700         /* FIXME: Changing back to the first line is not detected. */
2701         if (view->offset != 0 || view->lineno != 0) {
2702                 view->p_restore = FALSE;
2703                 return FALSE;
2704         }
2706         if (goto_view_line(view, view->p_offset, view->p_lineno) &&
2707             view_is_displayed(view))
2708                 werase(view->win);
2710         view->yoffset = view->p_yoffset;
2711         view->p_restore = FALSE;
2713         return TRUE;
2716 static void
2717 end_update(struct view *view, bool force)
2719         if (!view->pipe)
2720                 return;
2721         while (!view->ops->read(view, NULL))
2722                 if (!force)
2723                         return;
2724         set_nonblocking_input(FALSE);
2725         if (force)
2726                 kill_io(view->pipe);
2727         done_io(view->pipe);
2728         view->pipe = NULL;
2731 static void
2732 setup_update(struct view *view, const char *vid)
2734         set_nonblocking_input(TRUE);
2735         reset_view(view);
2736         string_copy_rev(view->vid, vid);
2737         view->pipe = &view->io;
2738         view->start_time = time(NULL);
2741 static bool
2742 prepare_update(struct view *view, const char *argv[], const char *dir,
2743                enum format_flags flags)
2745         if (view->pipe)
2746                 end_update(view, TRUE);
2747         return init_io_rd(&view->io, argv, dir, flags);
2750 static bool
2751 prepare_update_file(struct view *view, const char *name)
2753         if (view->pipe)
2754                 end_update(view, TRUE);
2755         return io_open(&view->io, name);
2758 static bool
2759 begin_update(struct view *view, bool refresh)
2761         if (view->pipe)
2762                 end_update(view, TRUE);
2764         if (refresh) {
2765                 if (!start_io(&view->io))
2766                         return FALSE;
2768         } else {
2769                 if (view == VIEW(REQ_VIEW_TREE) && strcmp(view->vid, view->id))
2770                         opt_path[0] = 0;
2772                 if (!run_io_rd(&view->io, view->ops->argv, FORMAT_ALL))
2773                         return FALSE;
2775                 /* Put the current ref_* value to the view title ref
2776                  * member. This is needed by the blob view. Most other
2777                  * views sets it automatically after loading because the
2778                  * first line is a commit line. */
2779                 string_copy_rev(view->ref, view->id);
2780         }
2782         setup_update(view, view->id);
2784         return TRUE;
2787 #define ITEM_CHUNK_SIZE 256
2788 static void *
2789 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2791         size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2792         size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2794         if (mem == NULL || num_chunks != num_chunks_new) {
2795                 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2796                 mem = realloc(mem, *size * item_size);
2797         }
2799         return mem;
2802 static struct line *
2803 realloc_lines(struct view *view, size_t line_size)
2805         size_t alloc = view->line_alloc;
2806         struct line *tmp = realloc_items(view->line, &alloc, line_size,
2807                                          sizeof(*view->line));
2809         if (!tmp)
2810                 return NULL;
2812         view->line = tmp;
2813         view->line_alloc = alloc;
2814         return view->line;
2817 static bool
2818 update_view(struct view *view)
2820         char out_buffer[BUFSIZ * 2];
2821         char *line;
2822         /* Clear the view and redraw everything since the tree sorting
2823          * might have rearranged things. */
2824         bool redraw = view->lines == 0;
2825         bool can_read = TRUE;
2827         if (!view->pipe)
2828                 return TRUE;
2830         if (!io_can_read(view->pipe)) {
2831                 if (view->lines == 0 && view_is_displayed(view)) {
2832                         time_t secs = time(NULL) - view->start_time;
2834                         if (secs > 1 && secs > view->update_secs) {
2835                                 if (view->update_secs == 0)
2836                                         redraw_view(view);
2837                                 update_view_title(view);
2838                                 view->update_secs = secs;
2839                         }
2840                 }
2841                 return TRUE;
2842         }
2844         for (; (line = io_get(view->pipe, '\n', can_read)); can_read = FALSE) {
2845                 if (opt_iconv != ICONV_NONE) {
2846                         ICONV_CONST char *inbuf = line;
2847                         size_t inlen = strlen(line) + 1;
2849                         char *outbuf = out_buffer;
2850                         size_t outlen = sizeof(out_buffer);
2852                         size_t ret;
2854                         ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2855                         if (ret != (size_t) -1)
2856                                 line = out_buffer;
2857                 }
2859                 if (!view->ops->read(view, line)) {
2860                         report("Allocation failure");
2861                         end_update(view, TRUE);
2862                         return FALSE;
2863                 }
2864         }
2866         {
2867                 unsigned long lines = view->lines;
2868                 int digits;
2870                 for (digits = 0; lines; digits++)
2871                         lines /= 10;
2873                 /* Keep the displayed view in sync with line number scaling. */
2874                 if (digits != view->digits) {
2875                         view->digits = digits;
2876                         if (opt_line_number || view == VIEW(REQ_VIEW_BLAME))
2877                                 redraw = TRUE;
2878                 }
2879         }
2881         if (io_error(view->pipe)) {
2882                 report("Failed to read: %s", io_strerror(view->pipe));
2883                 end_update(view, TRUE);
2885         } else if (io_eof(view->pipe)) {
2886                 report("");
2887                 end_update(view, FALSE);
2888         }
2890         if (restore_view_position(view))
2891                 redraw = TRUE;
2893         if (!view_is_displayed(view))
2894                 return TRUE;
2896         if (redraw)
2897                 redraw_view_from(view, 0);
2898         else
2899                 redraw_view_dirty(view);
2901         /* Update the title _after_ the redraw so that if the redraw picks up a
2902          * commit reference in view->ref it'll be available here. */
2903         update_view_title(view);
2904         return TRUE;
2907 static struct line *
2908 add_line_data(struct view *view, void *data, enum line_type type)
2910         struct line *line;
2912         if (!realloc_lines(view, view->lines + 1))
2913                 return NULL;
2915         line = &view->line[view->lines++];
2916         memset(line, 0, sizeof(*line));
2917         line->type = type;
2918         line->data = data;
2919         line->dirty = 1;
2921         return line;
2924 static struct line *
2925 add_line_text(struct view *view, const char *text, enum line_type type)
2927         char *data = text ? strdup(text) : NULL;
2929         return data ? add_line_data(view, data, type) : NULL;
2932 static struct line *
2933 add_line_format(struct view *view, enum line_type type, const char *fmt, ...)
2935         char buf[SIZEOF_STR];
2936         va_list args;
2938         va_start(args, fmt);
2939         if (vsnprintf(buf, sizeof(buf), fmt, args) >= sizeof(buf))
2940                 buf[0] = 0;
2941         va_end(args);
2943         return buf[0] ? add_line_text(view, buf, type) : NULL;
2946 /*
2947  * View opening
2948  */
2950 enum open_flags {
2951         OPEN_DEFAULT = 0,       /* Use default view switching. */
2952         OPEN_SPLIT = 1,         /* Split current view. */
2953         OPEN_RELOAD = 4,        /* Reload view even if it is the current. */
2954         OPEN_REFRESH = 16,      /* Refresh view using previous command. */
2955         OPEN_PREPARED = 32,     /* Open already prepared command. */
2956 };
2958 static void
2959 open_view(struct view *prev, enum request request, enum open_flags flags)
2961         bool split = !!(flags & OPEN_SPLIT);
2962         bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH | OPEN_PREPARED));
2963         bool nomaximize = !!(flags & OPEN_REFRESH);
2964         struct view *view = VIEW(request);
2965         int nviews = displayed_views();
2966         struct view *base_view = display[0];
2968         if (view == prev && nviews == 1 && !reload) {
2969                 report("Already in %s view", view->name);
2970                 return;
2971         }
2973         if (view->git_dir && !opt_git_dir[0]) {
2974                 report("The %s view is disabled in pager view", view->name);
2975                 return;
2976         }
2978         if (split) {
2979                 display[1] = view;
2980                 current_view = 1;
2981         } else if (!nomaximize) {
2982                 /* Maximize the current view. */
2983                 memset(display, 0, sizeof(display));
2984                 current_view = 0;
2985                 display[current_view] = view;
2986         }
2988         /* Resize the view when switching between split- and full-screen,
2989          * or when switching between two different full-screen views. */
2990         if (nviews != displayed_views() ||
2991             (nviews == 1 && base_view != display[0]))
2992                 resize_display();
2994         if (view->ops->open) {
2995                 if (view->pipe)
2996                         end_update(view, TRUE);
2997                 if (!view->ops->open(view)) {
2998                         report("Failed to load %s view", view->name);
2999                         return;
3000                 }
3001                 restore_view_position(view);
3003         } else if ((reload || strcmp(view->vid, view->id)) &&
3004                    !begin_update(view, flags & (OPEN_REFRESH | OPEN_PREPARED))) {
3005                 report("Failed to load %s view", view->name);
3006                 return;
3007         }
3009         if (split && prev->lineno - prev->offset >= prev->height) {
3010                 /* Take the title line into account. */
3011                 int lines = prev->lineno - prev->offset - prev->height + 1;
3013                 /* Scroll the view that was split if the current line is
3014                  * outside the new limited view. */
3015                 do_scroll_view(prev, lines);
3016         }
3018         if (prev && view != prev) {
3019                 if (split) {
3020                         /* "Blur" the previous view. */
3021                         update_view_title(prev);
3022                 }
3024                 view->parent = prev;
3025         }
3027         if (view->pipe && view->lines == 0) {
3028                 /* Clear the old view and let the incremental updating refill
3029                  * the screen. */
3030                 werase(view->win);
3031                 view->p_restore = flags & (OPEN_RELOAD | OPEN_REFRESH);
3032                 report("");
3033         } else if (view_is_displayed(view)) {
3034                 redraw_view(view);
3035                 report("");
3036         }
3039 static void
3040 open_external_viewer(const char *argv[], const char *dir)
3042         def_prog_mode();           /* save current tty modes */
3043         endwin();                  /* restore original tty modes */
3044         run_io_fg(argv, dir);
3045         fprintf(stderr, "Press Enter to continue");
3046         getc(opt_tty);
3047         reset_prog_mode();
3048         redraw_display(TRUE);
3051 static void
3052 open_mergetool(const char *file)
3054         const char *mergetool_argv[] = { "git", "mergetool", file, NULL };
3056         open_external_viewer(mergetool_argv, opt_cdup);
3059 static void
3060 open_editor(bool from_root, const char *file)
3062         const char *editor_argv[] = { "vi", file, NULL };
3063         const char *editor;
3065         editor = getenv("GIT_EDITOR");
3066         if (!editor && *opt_editor)
3067                 editor = opt_editor;
3068         if (!editor)
3069                 editor = getenv("VISUAL");
3070         if (!editor)
3071                 editor = getenv("EDITOR");
3072         if (!editor)
3073                 editor = "vi";
3075         editor_argv[0] = editor;
3076         open_external_viewer(editor_argv, from_root ? opt_cdup : NULL);
3079 static void
3080 open_run_request(enum request request)
3082         struct run_request *req = get_run_request(request);
3083         const char *argv[ARRAY_SIZE(req->argv)] = { NULL };
3085         if (!req) {
3086                 report("Unknown run request");
3087                 return;
3088         }
3090         if (format_argv(argv, req->argv, FORMAT_ALL))
3091                 open_external_viewer(argv, NULL);
3092         free_argv(argv);
3095 /*
3096  * User request switch noodle
3097  */
3099 static int
3100 view_driver(struct view *view, enum request request)
3102         int i;
3104         if (request == REQ_NONE) {
3105                 doupdate();
3106                 return TRUE;
3107         }
3109         if (request > REQ_NONE) {
3110                 open_run_request(request);
3111                 /* FIXME: When all views can refresh always do this. */
3112                 if (view == VIEW(REQ_VIEW_STATUS) ||
3113                     view == VIEW(REQ_VIEW_MAIN) ||
3114                     view == VIEW(REQ_VIEW_LOG) ||
3115                     view == VIEW(REQ_VIEW_STAGE))
3116                         request = REQ_REFRESH;
3117                 else
3118                         return TRUE;
3119         }
3121         if (view && view->lines) {
3122                 request = view->ops->request(view, request, &view->line[view->lineno]);
3123                 if (request == REQ_NONE)
3124                         return TRUE;
3125         }
3127         switch (request) {
3128         case REQ_MOVE_UP:
3129         case REQ_MOVE_DOWN:
3130         case REQ_MOVE_PAGE_UP:
3131         case REQ_MOVE_PAGE_DOWN:
3132         case REQ_MOVE_FIRST_LINE:
3133         case REQ_MOVE_LAST_LINE:
3134                 move_view(view, request);
3135                 break;
3137         case REQ_SCROLL_LEFT:
3138         case REQ_SCROLL_RIGHT:
3139         case REQ_SCROLL_LINE_DOWN:
3140         case REQ_SCROLL_LINE_UP:
3141         case REQ_SCROLL_PAGE_DOWN:
3142         case REQ_SCROLL_PAGE_UP:
3143                 scroll_view(view, request);
3144                 break;
3146         case REQ_VIEW_BLAME:
3147                 if (!opt_file[0]) {
3148                         report("No file chosen, press %s to open tree view",
3149                                get_key(REQ_VIEW_TREE));
3150                         break;
3151                 }
3152                 open_view(view, request, OPEN_DEFAULT);
3153                 break;
3155         case REQ_VIEW_BLOB:
3156                 if (!ref_blob[0]) {
3157                         report("No file chosen, press %s to open tree view",
3158                                get_key(REQ_VIEW_TREE));
3159                         break;
3160                 }
3161                 open_view(view, request, OPEN_DEFAULT);
3162                 break;
3164         case REQ_VIEW_PAGER:
3165                 if (!VIEW(REQ_VIEW_PAGER)->pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
3166                         report("No pager content, press %s to run command from prompt",
3167                                get_key(REQ_PROMPT));
3168                         break;
3169                 }
3170                 open_view(view, request, OPEN_DEFAULT);
3171                 break;
3173         case REQ_VIEW_STAGE:
3174                 if (!VIEW(REQ_VIEW_STAGE)->lines) {
3175                         report("No stage content, press %s to open the status view and choose file",
3176                                get_key(REQ_VIEW_STATUS));
3177                         break;
3178                 }
3179                 open_view(view, request, OPEN_DEFAULT);
3180                 break;
3182         case REQ_VIEW_STATUS:
3183                 if (opt_is_inside_work_tree == FALSE) {
3184                         report("The status view requires a working tree");
3185                         break;
3186                 }
3187                 open_view(view, request, OPEN_DEFAULT);
3188                 break;
3190         case REQ_VIEW_MAIN:
3191         case REQ_VIEW_DIFF:
3192         case REQ_VIEW_LOG:
3193         case REQ_VIEW_TREE:
3194         case REQ_VIEW_HELP:
3195                 open_view(view, request, OPEN_DEFAULT);
3196                 break;
3198         case REQ_NEXT:
3199         case REQ_PREVIOUS:
3200                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
3202                 if ((view == VIEW(REQ_VIEW_DIFF) &&
3203                      view->parent == VIEW(REQ_VIEW_MAIN)) ||
3204                    (view == VIEW(REQ_VIEW_DIFF) &&
3205                      view->parent == VIEW(REQ_VIEW_BLAME)) ||
3206                    (view == VIEW(REQ_VIEW_STAGE) &&
3207                      view->parent == VIEW(REQ_VIEW_STATUS)) ||
3208                    (view == VIEW(REQ_VIEW_BLOB) &&
3209                      view->parent == VIEW(REQ_VIEW_TREE))) {
3210                         int line;
3212                         view = view->parent;
3213                         line = view->lineno;
3214                         move_view(view, request);
3215                         if (view_is_displayed(view))
3216                                 update_view_title(view);
3217                         if (line != view->lineno)
3218                                 view->ops->request(view, REQ_ENTER,
3219                                                    &view->line[view->lineno]);
3221                 } else {
3222                         move_view(view, request);
3223                 }
3224                 break;
3226         case REQ_VIEW_NEXT:
3227         {
3228                 int nviews = displayed_views();
3229                 int next_view = (current_view + 1) % nviews;
3231                 if (next_view == current_view) {
3232                         report("Only one view is displayed");
3233                         break;
3234                 }
3236                 current_view = next_view;
3237                 /* Blur out the title of the previous view. */
3238                 update_view_title(view);
3239                 report("");
3240                 break;
3241         }
3242         case REQ_REFRESH:
3243                 report("Refreshing is not yet supported for the %s view", view->name);
3244                 break;
3246         case REQ_MAXIMIZE:
3247                 if (displayed_views() == 2)
3248                         maximize_view(view);
3249                 break;
3251         case REQ_TOGGLE_LINENO:
3252                 toggle_view_option(&opt_line_number, "line numbers");
3253                 break;
3255         case REQ_TOGGLE_DATE:
3256                 toggle_view_option(&opt_date, "date display");
3257                 break;
3259         case REQ_TOGGLE_AUTHOR:
3260                 toggle_view_option(&opt_author, "author display");
3261                 break;
3263         case REQ_TOGGLE_REV_GRAPH:
3264                 toggle_view_option(&opt_rev_graph, "revision graph display");
3265                 break;
3267         case REQ_TOGGLE_REFS:
3268                 toggle_view_option(&opt_show_refs, "reference display");
3269                 break;
3271         case REQ_SEARCH:
3272         case REQ_SEARCH_BACK:
3273                 search_view(view, request);
3274                 break;
3276         case REQ_FIND_NEXT:
3277         case REQ_FIND_PREV:
3278                 find_next(view, request);
3279                 break;
3281         case REQ_STOP_LOADING:
3282                 for (i = 0; i < ARRAY_SIZE(views); i++) {
3283                         view = &views[i];
3284                         if (view->pipe)
3285                                 report("Stopped loading the %s view", view->name),
3286                         end_update(view, TRUE);
3287                 }
3288                 break;
3290         case REQ_SHOW_VERSION:
3291                 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
3292                 return TRUE;
3294         case REQ_SCREEN_REDRAW:
3295                 redraw_display(TRUE);
3296                 break;
3298         case REQ_EDIT:
3299                 report("Nothing to edit");
3300                 break;
3302         case REQ_ENTER:
3303                 report("Nothing to enter");
3304                 break;
3306         case REQ_VIEW_CLOSE:
3307                 /* XXX: Mark closed views by letting view->parent point to the
3308                  * view itself. Parents to closed view should never be
3309                  * followed. */
3310                 if (view->parent &&
3311                     view->parent->parent != view->parent) {
3312                         maximize_view(view->parent);
3313                         view->parent = view;
3314                         break;
3315                 }
3316                 /* Fall-through */
3317         case REQ_QUIT:
3318                 return FALSE;
3320         default:
3321                 report("Unknown key, press 'h' for help");
3322                 return TRUE;
3323         }
3325         return TRUE;
3329 /*
3330  * View backend utilities
3331  */
3333 static void
3334 parse_timezone(time_t *time, const char *zone)
3336         long tz;
3338         tz  = ('0' - zone[1]) * 60 * 60 * 10;
3339         tz += ('0' - zone[2]) * 60 * 60;
3340         tz += ('0' - zone[3]) * 60;
3341         tz += ('0' - zone[4]);
3343         if (zone[0] == '-')
3344                 tz = -tz;
3346         *time -= tz;
3349 /* Parse author lines where the name may be empty:
3350  *      author  <email@address.tld> 1138474660 +0100
3351  */
3352 static void
3353 parse_author_line(char *ident, char *author, size_t authorsize, struct tm *tm)
3355         char *nameend = strchr(ident, '<');
3356         char *emailend = strchr(ident, '>');
3358         if (nameend && emailend)
3359                 *nameend = *emailend = 0;
3360         ident = chomp_string(ident);
3361         if (!*ident) {
3362                 if (nameend)
3363                         ident = chomp_string(nameend + 1);
3364                 if (!*ident)
3365                         ident = "Unknown";
3366         }
3368         string_ncopy_do(author, authorsize, ident, strlen(ident));
3370         /* Parse epoch and timezone */
3371         if (emailend && emailend[1] == ' ') {
3372                 char *secs = emailend + 2;
3373                 char *zone = strchr(secs, ' ');
3374                 time_t time = (time_t) atol(secs);
3376                 if (zone && strlen(zone) == STRING_SIZE(" +0700"))
3377                         parse_timezone(&time, zone + 1);
3379                 gmtime_r(&time, tm);
3380         }
3383 static enum input_status
3384 select_commit_parent_handler(void *data, char *buf, int c)
3386         size_t parents = *(size_t *) data;
3387         int parent = 0;
3389         if (!isdigit(c))
3390                 return INPUT_SKIP;
3392         if (*buf)
3393                 parent = atoi(buf) * 10;
3394         parent += c - '0';
3396         if (parent > parents)
3397                 return INPUT_SKIP;
3398         return INPUT_OK;
3401 static bool
3402 select_commit_parent(const char *id, char rev[SIZEOF_REV], const char *path)
3404         char buf[SIZEOF_STR * 4];
3405         const char *revlist_argv[] = {
3406                 "git", "rev-list", "-1", "--parents", id, "--", path, NULL
3407         };
3408         int parents;
3410         if (!run_io_buf(revlist_argv, buf, sizeof(buf)) ||
3411             !*chomp_string(buf) ||
3412             (parents = (strlen(buf) / 40) - 1) < 0) {
3413                 report("Failed to get parent information");
3414                 return FALSE;
3416         } else if (parents == 0) {
3417                 if (path)
3418                         report("Path '%s' does not exist in the parent", path);
3419                 else
3420                         report("The selected commit has no parents");
3421                 return FALSE;
3422         }
3424         if (parents > 1) {
3425                 char prompt[SIZEOF_STR];
3426                 char *result;
3428                 if (!string_format(prompt, "Which parent? [1..%d] ", parents))
3429                         return FALSE;
3430                 result = prompt_input(prompt, select_commit_parent_handler, &parents);
3431                 if (!result)
3432                         return FALSE;
3433                 parents = atoi(result);
3434         }
3436         string_copy_rev(rev, &buf[41 * parents]);
3437         return TRUE;
3440 /*
3441  * Pager backend
3442  */
3444 static bool
3445 pager_draw(struct view *view, struct line *line, unsigned int lineno)
3447         char text[SIZEOF_STR];
3449         if (opt_line_number && draw_lineno(view, lineno))
3450                 return TRUE;
3452         string_expand(text, sizeof(text), line->data, opt_tab_size);
3453         draw_text(view, line->type, text, TRUE);
3454         return TRUE;
3457 static bool
3458 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
3460         const char *describe_argv[] = { "git", "describe", commit_id, NULL };
3461         char refbuf[SIZEOF_STR];
3462         char *ref = NULL;
3464         if (run_io_buf(describe_argv, refbuf, sizeof(refbuf)))
3465                 ref = chomp_string(refbuf);
3467         if (!ref || !*ref)
3468                 return TRUE;
3470         /* This is the only fatal call, since it can "corrupt" the buffer. */
3471         if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
3472                 return FALSE;
3474         return TRUE;
3477 static void
3478 add_pager_refs(struct view *view, struct line *line)
3480         char buf[SIZEOF_STR];
3481         char *commit_id = (char *)line->data + STRING_SIZE("commit ");
3482         struct ref **refs;
3483         size_t bufpos = 0, refpos = 0;
3484         const char *sep = "Refs: ";
3485         bool is_tag = FALSE;
3487         assert(line->type == LINE_COMMIT);
3489         refs = get_refs(commit_id);
3490         if (!refs) {
3491                 if (view == VIEW(REQ_VIEW_DIFF))
3492                         goto try_add_describe_ref;
3493                 return;
3494         }
3496         do {
3497                 struct ref *ref = refs[refpos];
3498                 const char *fmt = ref->tag    ? "%s[%s]" :
3499                                   ref->remote ? "%s<%s>" : "%s%s";
3501                 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
3502                         return;
3503                 sep = ", ";
3504                 if (ref->tag)
3505                         is_tag = TRUE;
3506         } while (refs[refpos++]->next);
3508         if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
3509 try_add_describe_ref:
3510                 /* Add <tag>-g<commit_id> "fake" reference. */
3511                 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
3512                         return;
3513         }
3515         if (bufpos == 0)
3516                 return;
3518         add_line_text(view, buf, LINE_PP_REFS);
3521 static bool
3522 pager_read(struct view *view, char *data)
3524         struct line *line;
3526         if (!data)
3527                 return TRUE;
3529         line = add_line_text(view, data, get_line_type(data));
3530         if (!line)
3531                 return FALSE;
3533         if (line->type == LINE_COMMIT &&
3534             (view == VIEW(REQ_VIEW_DIFF) ||
3535              view == VIEW(REQ_VIEW_LOG)))
3536                 add_pager_refs(view, line);
3538         return TRUE;
3541 static enum request
3542 pager_request(struct view *view, enum request request, struct line *line)
3544         int split = 0;
3546         if (request != REQ_ENTER)
3547                 return request;
3549         if (line->type == LINE_COMMIT &&
3550            (view == VIEW(REQ_VIEW_LOG) ||
3551             view == VIEW(REQ_VIEW_PAGER))) {
3552                 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
3553                 split = 1;
3554         }
3556         /* Always scroll the view even if it was split. That way
3557          * you can use Enter to scroll through the log view and
3558          * split open each commit diff. */
3559         scroll_view(view, REQ_SCROLL_LINE_DOWN);
3561         /* FIXME: A minor workaround. Scrolling the view will call report("")
3562          * but if we are scrolling a non-current view this won't properly
3563          * update the view title. */
3564         if (split)
3565                 update_view_title(view);
3567         return REQ_NONE;
3570 static bool
3571 pager_grep(struct view *view, struct line *line)
3573         regmatch_t pmatch;
3574         char *text = line->data;
3576         if (!*text)
3577                 return FALSE;
3579         if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
3580                 return FALSE;
3582         return TRUE;
3585 static void
3586 pager_select(struct view *view, struct line *line)
3588         if (line->type == LINE_COMMIT) {
3589                 char *text = (char *)line->data + STRING_SIZE("commit ");
3591                 if (view != VIEW(REQ_VIEW_PAGER))
3592                         string_copy_rev(view->ref, text);
3593                 string_copy_rev(ref_commit, text);
3594         }
3597 static struct view_ops pager_ops = {
3598         "line",
3599         NULL,
3600         NULL,
3601         pager_read,
3602         pager_draw,
3603         pager_request,
3604         pager_grep,
3605         pager_select,
3606 };
3608 static const char *log_argv[SIZEOF_ARG] = {
3609         "git", "log", "--no-color", "--cc", "--stat", "-n100", "%(head)", NULL
3610 };
3612 static enum request
3613 log_request(struct view *view, enum request request, struct line *line)
3615         switch (request) {
3616         case REQ_REFRESH:
3617                 load_refs();
3618                 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
3619                 return REQ_NONE;
3620         default:
3621                 return pager_request(view, request, line);
3622         }
3625 static struct view_ops log_ops = {
3626         "line",
3627         log_argv,
3628         NULL,
3629         pager_read,
3630         pager_draw,
3631         log_request,
3632         pager_grep,
3633         pager_select,
3634 };
3636 static const char *diff_argv[SIZEOF_ARG] = {
3637         "git", "show", "--pretty=fuller", "--no-color", "--root",
3638                 "--patch-with-stat", "--find-copies-harder", "-C", "%(commit)", NULL
3639 };
3641 static struct view_ops diff_ops = {
3642         "line",
3643         diff_argv,
3644         NULL,
3645         pager_read,
3646         pager_draw,
3647         pager_request,
3648         pager_grep,
3649         pager_select,
3650 };
3652 /*
3653  * Help backend
3654  */
3656 static bool
3657 help_open(struct view *view)
3659         char buf[SIZEOF_STR];
3660         size_t bufpos;
3661         int i;
3663         if (view->lines > 0)
3664                 return TRUE;
3666         add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3668         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3669                 const char *key;
3671                 if (req_info[i].request == REQ_NONE)
3672                         continue;
3674                 if (!req_info[i].request) {
3675                         add_line_text(view, "", LINE_DEFAULT);
3676                         add_line_text(view, req_info[i].help, LINE_DEFAULT);
3677                         continue;
3678                 }
3680                 key = get_key(req_info[i].request);
3681                 if (!*key)
3682                         key = "(no key defined)";
3684                 for (bufpos = 0; bufpos <= req_info[i].namelen; bufpos++) {
3685                         buf[bufpos] = tolower(req_info[i].name[bufpos]);
3686                         if (buf[bufpos] == '_')
3687                                 buf[bufpos] = '-';
3688                 }
3690                 add_line_format(view, LINE_DEFAULT, "    %-25s %-20s %s",
3691                                 key, buf, req_info[i].help);
3692         }
3694         if (run_requests) {
3695                 add_line_text(view, "", LINE_DEFAULT);
3696                 add_line_text(view, "External commands:", LINE_DEFAULT);
3697         }
3699         for (i = 0; i < run_requests; i++) {
3700                 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3701                 const char *key;
3702                 int argc;
3704                 if (!req)
3705                         continue;
3707                 key = get_key_name(req->key);
3708                 if (!*key)
3709                         key = "(no key defined)";
3711                 for (bufpos = 0, argc = 0; req->argv[argc]; argc++)
3712                         if (!string_format_from(buf, &bufpos, "%s%s",
3713                                                 argc ? " " : "", req->argv[argc]))
3714                                 return REQ_NONE;
3716                 add_line_format(view, LINE_DEFAULT, "    %-10s %-14s `%s`",
3717                                 keymap_table[req->keymap].name, key, buf);
3718         }
3720         return TRUE;
3723 static struct view_ops help_ops = {
3724         "line",
3725         NULL,
3726         help_open,
3727         NULL,
3728         pager_draw,
3729         pager_request,
3730         pager_grep,
3731         pager_select,
3732 };
3735 /*
3736  * Tree backend
3737  */
3739 struct tree_stack_entry {
3740         struct tree_stack_entry *prev;  /* Entry below this in the stack */
3741         unsigned long lineno;           /* Line number to restore */
3742         char *name;                     /* Position of name in opt_path */
3743 };
3745 /* The top of the path stack. */
3746 static struct tree_stack_entry *tree_stack = NULL;
3747 unsigned long tree_lineno = 0;
3749 static void
3750 pop_tree_stack_entry(void)
3752         struct tree_stack_entry *entry = tree_stack;
3754         tree_lineno = entry->lineno;
3755         entry->name[0] = 0;
3756         tree_stack = entry->prev;
3757         free(entry);
3760 static void
3761 push_tree_stack_entry(const char *name, unsigned long lineno)
3763         struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3764         size_t pathlen = strlen(opt_path);
3766         if (!entry)
3767                 return;
3769         entry->prev = tree_stack;
3770         entry->name = opt_path + pathlen;
3771         tree_stack = entry;
3773         if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3774                 pop_tree_stack_entry();
3775                 return;
3776         }
3778         /* Move the current line to the first tree entry. */
3779         tree_lineno = 1;
3780         entry->lineno = lineno;
3783 /* Parse output from git-ls-tree(1):
3784  *
3785  * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3786  */
3788 #define SIZEOF_TREE_ATTR \
3789         STRING_SIZE("100644 blob f931e1d229c3e185caad4449bf5b66ed72462657\t")
3791 #define SIZEOF_TREE_MODE \
3792         STRING_SIZE("100644 ")
3794 #define TREE_ID_OFFSET \
3795         STRING_SIZE("100644 blob ")
3797 struct tree_entry {
3798         char id[SIZEOF_REV];
3799         mode_t mode;
3800         struct tm time;                 /* Date from the author ident. */
3801         char author[75];                /* Author of the commit. */
3802         char name[1];
3803 };
3805 static const char *
3806 tree_path(struct line *line)
3808         return ((struct tree_entry *) line->data)->name;
3812 static int
3813 tree_compare_entry(struct line *line1, struct line *line2)
3815         if (line1->type != line2->type)
3816                 return line1->type == LINE_TREE_DIR ? -1 : 1;
3817         return strcmp(tree_path(line1), tree_path(line2));
3820 static struct line *
3821 tree_entry(struct view *view, enum line_type type, const char *path,
3822            const char *mode, const char *id)
3824         struct tree_entry *entry = calloc(1, sizeof(*entry) + strlen(path));
3825         struct line *line = entry ? add_line_data(view, entry, type) : NULL;
3827         if (!entry || !line) {
3828                 free(entry);
3829                 return NULL;
3830         }
3832         strncpy(entry->name, path, strlen(path));
3833         if (mode)
3834                 entry->mode = strtoul(mode, NULL, 8);
3835         if (id)
3836                 string_copy_rev(entry->id, id);
3838         return line;
3841 static bool
3842 tree_read_date(struct view *view, char *text, bool *read_date)
3844         static char author_name[SIZEOF_STR];
3845         static struct tm author_time;
3847         if (!text && *read_date) {
3848                 *read_date = FALSE;
3849                 return TRUE;
3851         } else if (!text) {
3852                 char *path = *opt_path ? opt_path : ".";
3853                 /* Find next entry to process */
3854                 const char *log_file[] = {
3855                         "git", "log", "--no-color", "--pretty=raw",
3856                                 "--cc", "--raw", view->id, "--", path, NULL
3857                 };
3858                 struct io io = {};
3860                 if (!view->lines) {
3861                         tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL);
3862                         report("Tree is empty");
3863                         return TRUE;
3864                 }
3866                 if (!run_io_rd(&io, log_file, FORMAT_NONE)) {
3867                         report("Failed to load tree data");
3868                         return TRUE;
3869                 }
3871                 done_io(view->pipe);
3872                 view->io = io;
3873                 *read_date = TRUE;
3874                 return FALSE;
3876         } else if (*text == 'a' && get_line_type(text) == LINE_AUTHOR) {
3877                 parse_author_line(text + STRING_SIZE("author "),
3878                                   author_name, sizeof(author_name), &author_time);
3880         } else if (*text == ':') {
3881                 char *pos;
3882                 size_t annotated = 1;
3883                 size_t i;
3885                 pos = strchr(text, '\t');
3886                 if (!pos)
3887                         return TRUE;
3888                 text = pos + 1;
3889                 if (*opt_prefix && !strncmp(text, opt_prefix, strlen(opt_prefix)))
3890                         text += strlen(opt_prefix);
3891                 if (*opt_path && !strncmp(text, opt_path, strlen(opt_path)))
3892                         text += strlen(opt_path);
3893                 pos = strchr(text, '/');
3894                 if (pos)
3895                         *pos = 0;
3897                 for (i = 1; i < view->lines; i++) {
3898                         struct line *line = &view->line[i];
3899                         struct tree_entry *entry = line->data;
3901                         annotated += !!*entry->author;
3902                         if (*entry->author || strcmp(entry->name, text))
3903                                 continue;
3905                         string_copy(entry->author, author_name);
3906                         memcpy(&entry->time, &author_time, sizeof(entry->time));
3907                         line->dirty = 1;
3908                         break;
3909                 }
3911                 if (annotated == view->lines)
3912                         kill_io(view->pipe);
3913         }
3914         return TRUE;
3917 static bool
3918 tree_read(struct view *view, char *text)
3920         static bool read_date = FALSE;
3921         struct tree_entry *data;
3922         struct line *entry, *line;
3923         enum line_type type;
3924         size_t textlen = text ? strlen(text) : 0;
3925         char *path = text + SIZEOF_TREE_ATTR;
3927         if (read_date || !text)
3928                 return tree_read_date(view, text, &read_date);
3930         if (textlen <= SIZEOF_TREE_ATTR)
3931                 return FALSE;
3932         if (view->lines == 0 &&
3933             !tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL))
3934                 return FALSE;
3936         /* Strip the path part ... */
3937         if (*opt_path) {
3938                 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3939                 size_t striplen = strlen(opt_path);
3941                 if (pathlen > striplen)
3942                         memmove(path, path + striplen,
3943                                 pathlen - striplen + 1);
3945                 /* Insert "link" to parent directory. */
3946                 if (view->lines == 1 &&
3947                     !tree_entry(view, LINE_TREE_DIR, "..", "040000", view->ref))
3948                         return FALSE;
3949         }
3951         type = text[SIZEOF_TREE_MODE] == 't' ? LINE_TREE_DIR : LINE_TREE_FILE;
3952         entry = tree_entry(view, type, path, text, text + TREE_ID_OFFSET);
3953         if (!entry)
3954                 return FALSE;
3955         data = entry->data;
3957         /* Skip "Directory ..." and ".." line. */
3958         for (line = &view->line[1 + !!*opt_path]; line < entry; line++) {
3959                 if (tree_compare_entry(line, entry) <= 0)
3960                         continue;
3962                 memmove(line + 1, line, (entry - line) * sizeof(*entry));
3964                 line->data = data;
3965                 line->type = type;
3966                 for (; line <= entry; line++)
3967                         line->dirty = line->cleareol = 1;
3968                 return TRUE;
3969         }
3971         if (tree_lineno > view->lineno) {
3972                 view->lineno = tree_lineno;
3973                 tree_lineno = 0;
3974         }
3976         return TRUE;
3979 static bool
3980 tree_draw(struct view *view, struct line *line, unsigned int lineno)
3982         struct tree_entry *entry = line->data;
3984         if (line->type == LINE_TREE_HEAD) {
3985                 if (draw_text(view, line->type, "Directory path /", TRUE))
3986                         return TRUE;
3987         } else {
3988                 if (draw_mode(view, entry->mode))
3989                         return TRUE;
3991                 if (opt_author && draw_author(view, entry->author))
3992                         return TRUE;
3994                 if (opt_date && draw_date(view, *entry->author ? &entry->time : NULL))
3995                         return TRUE;
3996         }
3997         if (draw_text(view, line->type, entry->name, TRUE))
3998                 return TRUE;
3999         return TRUE;
4002 static void
4003 open_blob_editor()
4005         char file[SIZEOF_STR] = "/tmp/tigblob.XXXXXX";
4006         int fd = mkstemp(file);
4008         if (fd == -1)
4009                 report("Failed to create temporary file");
4010         else if (!run_io_append(blob_ops.argv, FORMAT_ALL, fd))
4011                 report("Failed to save blob data to file");
4012         else
4013                 open_editor(FALSE, file);
4014         if (fd != -1)
4015                 unlink(file);
4018 static enum request
4019 tree_request(struct view *view, enum request request, struct line *line)
4021         enum open_flags flags;
4023         switch (request) {
4024         case REQ_VIEW_BLAME:
4025                 if (line->type != LINE_TREE_FILE) {
4026                         report("Blame only supported for files");
4027                         return REQ_NONE;
4028                 }
4030                 string_copy(opt_ref, view->vid);
4031                 return request;
4033         case REQ_EDIT:
4034                 if (line->type != LINE_TREE_FILE) {
4035                         report("Edit only supported for files");
4036                 } else if (!is_head_commit(view->vid)) {
4037                         open_blob_editor();
4038                 } else {
4039                         open_editor(TRUE, opt_file);
4040                 }
4041                 return REQ_NONE;
4043         case REQ_PARENT:
4044                 if (!*opt_path) {
4045                         /* quit view if at top of tree */
4046                         return REQ_VIEW_CLOSE;
4047                 }
4048                 /* fake 'cd  ..' */
4049                 line = &view->line[1];
4050                 break;
4052         case REQ_ENTER:
4053                 break;
4055         default:
4056                 return request;
4057         }
4059         /* Cleanup the stack if the tree view is at a different tree. */
4060         while (!*opt_path && tree_stack)
4061                 pop_tree_stack_entry();
4063         switch (line->type) {
4064         case LINE_TREE_DIR:
4065                 /* Depending on whether it is a subdirectory or parent link
4066                  * mangle the path buffer. */
4067                 if (line == &view->line[1] && *opt_path) {
4068                         pop_tree_stack_entry();
4070                 } else {
4071                         const char *basename = tree_path(line);
4073                         push_tree_stack_entry(basename, view->lineno);
4074                 }
4076                 /* Trees and subtrees share the same ID, so they are not not
4077                  * unique like blobs. */
4078                 flags = OPEN_RELOAD;
4079                 request = REQ_VIEW_TREE;
4080                 break;
4082         case LINE_TREE_FILE:
4083                 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4084                 request = REQ_VIEW_BLOB;
4085                 break;
4087         default:
4088                 return REQ_NONE;
4089         }
4091         open_view(view, request, flags);
4092         if (request == REQ_VIEW_TREE)
4093                 view->lineno = tree_lineno;
4095         return REQ_NONE;
4098 static void
4099 tree_select(struct view *view, struct line *line)
4101         struct tree_entry *entry = line->data;
4103         if (line->type == LINE_TREE_FILE) {
4104                 string_copy_rev(ref_blob, entry->id);
4105                 string_format(opt_file, "%s%s", opt_path, tree_path(line));
4107         } else if (line->type != LINE_TREE_DIR) {
4108                 return;
4109         }
4111         string_copy_rev(view->ref, entry->id);
4114 static const char *tree_argv[SIZEOF_ARG] = {
4115         "git", "ls-tree", "%(commit)", "%(directory)", NULL
4116 };
4118 static struct view_ops tree_ops = {
4119         "file",
4120         tree_argv,
4121         NULL,
4122         tree_read,
4123         tree_draw,
4124         tree_request,
4125         pager_grep,
4126         tree_select,
4127 };
4129 static bool
4130 blob_read(struct view *view, char *line)
4132         if (!line)
4133                 return TRUE;
4134         return add_line_text(view, line, LINE_DEFAULT) != NULL;
4137 static enum request
4138 blob_request(struct view *view, enum request request, struct line *line)
4140         switch (request) {
4141         case REQ_EDIT:
4142                 open_blob_editor();
4143                 return REQ_NONE;
4144         default:
4145                 return pager_request(view, request, line);
4146         }
4149 static const char *blob_argv[SIZEOF_ARG] = {
4150         "git", "cat-file", "blob", "%(blob)", NULL
4151 };
4153 static struct view_ops blob_ops = {
4154         "line",
4155         blob_argv,
4156         NULL,
4157         blob_read,
4158         pager_draw,
4159         blob_request,
4160         pager_grep,
4161         pager_select,
4162 };
4164 /*
4165  * Blame backend
4166  *
4167  * Loading the blame view is a two phase job:
4168  *
4169  *  1. File content is read either using opt_file from the
4170  *     filesystem or using git-cat-file.
4171  *  2. Then blame information is incrementally added by
4172  *     reading output from git-blame.
4173  */
4175 static const char *blame_head_argv[] = {
4176         "git", "blame", "--incremental", "--", "%(file)", NULL
4177 };
4179 static const char *blame_ref_argv[] = {
4180         "git", "blame", "--incremental", "%(ref)", "--", "%(file)", NULL
4181 };
4183 static const char *blame_cat_file_argv[] = {
4184         "git", "cat-file", "blob", "%(ref):%(file)", NULL
4185 };
4187 struct blame_commit {
4188         char id[SIZEOF_REV];            /* SHA1 ID. */
4189         char title[128];                /* First line of the commit message. */
4190         char author[75];                /* Author of the commit. */
4191         struct tm time;                 /* Date from the author ident. */
4192         char filename[128];             /* Name of file. */
4193         bool has_previous;              /* Was a "previous" line detected. */
4194 };
4196 struct blame {
4197         struct blame_commit *commit;
4198         unsigned long lineno;
4199         char text[1];
4200 };
4202 static bool
4203 blame_open(struct view *view)
4205         if (*opt_ref || !io_open(&view->io, opt_file)) {
4206                 if (!run_io_rd(&view->io, blame_cat_file_argv, FORMAT_ALL))
4207                         return FALSE;
4208         }
4210         setup_update(view, opt_file);
4211         string_format(view->ref, "%s ...", opt_file);
4213         return TRUE;
4216 static struct blame_commit *
4217 get_blame_commit(struct view *view, const char *id)
4219         size_t i;
4221         for (i = 0; i < view->lines; i++) {
4222                 struct blame *blame = view->line[i].data;
4224                 if (!blame->commit)
4225                         continue;
4227                 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
4228                         return blame->commit;
4229         }
4231         {
4232                 struct blame_commit *commit = calloc(1, sizeof(*commit));
4234                 if (commit)
4235                         string_ncopy(commit->id, id, SIZEOF_REV);
4236                 return commit;
4237         }
4240 static bool
4241 parse_number(const char **posref, size_t *number, size_t min, size_t max)
4243         const char *pos = *posref;
4245         *posref = NULL;
4246         pos = strchr(pos + 1, ' ');
4247         if (!pos || !isdigit(pos[1]))
4248                 return FALSE;
4249         *number = atoi(pos + 1);
4250         if (*number < min || *number > max)
4251                 return FALSE;
4253         *posref = pos;
4254         return TRUE;
4257 static struct blame_commit *
4258 parse_blame_commit(struct view *view, const char *text, int *blamed)
4260         struct blame_commit *commit;
4261         struct blame *blame;
4262         const char *pos = text + SIZEOF_REV - 2;
4263         size_t orig_lineno = 0;
4264         size_t lineno;
4265         size_t group;
4267         if (strlen(text) <= SIZEOF_REV || pos[1] != ' ')
4268                 return NULL;
4270         if (!parse_number(&pos, &orig_lineno, 1, 9999999) ||
4271             !parse_number(&pos, &lineno, 1, view->lines) ||
4272             !parse_number(&pos, &group, 1, view->lines - lineno + 1))
4273                 return NULL;
4275         commit = get_blame_commit(view, text);
4276         if (!commit)
4277                 return NULL;
4279         *blamed += group;
4280         while (group--) {
4281                 struct line *line = &view->line[lineno + group - 1];
4283                 blame = line->data;
4284                 blame->commit = commit;
4285                 blame->lineno = orig_lineno + group - 1;
4286                 line->dirty = 1;
4287         }
4289         return commit;
4292 static bool
4293 blame_read_file(struct view *view, const char *line, bool *read_file)
4295         if (!line) {
4296                 const char **argv = *opt_ref ? blame_ref_argv : blame_head_argv;
4297                 struct io io = {};
4299                 if (view->lines == 0 && !view->parent)
4300                         die("No blame exist for %s", view->vid);
4302                 if (view->lines == 0 || !run_io_rd(&io, argv, FORMAT_ALL)) {
4303                         report("Failed to load blame data");
4304                         return TRUE;
4305                 }
4307                 done_io(view->pipe);
4308                 view->io = io;
4309                 *read_file = FALSE;
4310                 return FALSE;
4312         } else {
4313                 size_t linelen = strlen(line);
4314                 struct blame *blame = malloc(sizeof(*blame) + linelen);
4316                 if (!blame)
4317                         return FALSE;
4319                 blame->commit = NULL;
4320                 strncpy(blame->text, line, linelen);
4321                 blame->text[linelen] = 0;
4322                 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
4323         }
4326 static bool
4327 match_blame_header(const char *name, char **line)
4329         size_t namelen = strlen(name);
4330         bool matched = !strncmp(name, *line, namelen);
4332         if (matched)
4333                 *line += namelen;
4335         return matched;
4338 static bool
4339 blame_read(struct view *view, char *line)
4341         static struct blame_commit *commit = NULL;
4342         static int blamed = 0;
4343         static time_t author_time;
4344         static bool read_file = TRUE;
4346         if (read_file)
4347                 return blame_read_file(view, line, &read_file);
4349         if (!line) {
4350                 /* Reset all! */
4351                 commit = NULL;
4352                 blamed = 0;
4353                 read_file = TRUE;
4354                 string_format(view->ref, "%s", view->vid);
4355                 if (view_is_displayed(view)) {
4356                         update_view_title(view);
4357                         redraw_view_from(view, 0);
4358                 }
4359                 return TRUE;
4360         }
4362         if (!commit) {
4363                 commit = parse_blame_commit(view, line, &blamed);
4364                 string_format(view->ref, "%s %2d%%", view->vid,
4365                               view->lines ? blamed * 100 / view->lines : 0);
4367         } else if (match_blame_header("author ", &line)) {
4368                 string_ncopy(commit->author, line, strlen(line));
4370         } else if (match_blame_header("author-time ", &line)) {
4371                 author_time = (time_t) atol(line);
4373         } else if (match_blame_header("author-tz ", &line)) {
4374                 parse_timezone(&author_time, line);
4375                 gmtime_r(&author_time, &commit->time);
4377         } else if (match_blame_header("summary ", &line)) {
4378                 string_ncopy(commit->title, line, strlen(line));
4380         } else if (match_blame_header("previous ", &line)) {
4381                 commit->has_previous = TRUE;
4383         } else if (match_blame_header("filename ", &line)) {
4384                 string_ncopy(commit->filename, line, strlen(line));
4385                 commit = NULL;
4386         }
4388         return TRUE;
4391 static bool
4392 blame_draw(struct view *view, struct line *line, unsigned int lineno)
4394         struct blame *blame = line->data;
4395         struct tm *time = NULL;
4396         const char *id = NULL, *author = NULL;
4397         char text[SIZEOF_STR];
4399         if (blame->commit && *blame->commit->filename) {
4400                 id = blame->commit->id;
4401                 author = blame->commit->author;
4402                 time = &blame->commit->time;
4403         }
4405         if (opt_date && draw_date(view, time))
4406                 return TRUE;
4408         if (opt_author && draw_author(view, author))
4409                 return TRUE;
4411         if (draw_field(view, LINE_BLAME_ID, id, ID_COLS, FALSE))
4412                 return TRUE;
4414         if (draw_lineno(view, lineno))
4415                 return TRUE;
4417         string_expand(text, sizeof(text), blame->text, opt_tab_size);
4418         draw_text(view, LINE_DEFAULT, text, TRUE);
4419         return TRUE;
4422 static bool
4423 check_blame_commit(struct blame *blame, bool check_null_id)
4425         if (!blame->commit)
4426                 report("Commit data not loaded yet");
4427         else if (check_null_id && !strcmp(blame->commit->id, NULL_ID))
4428                 report("No commit exist for the selected line");
4429         else
4430                 return TRUE;
4431         return FALSE;
4434 static void
4435 setup_blame_parent_line(struct view *view, struct blame *blame)
4437         const char *diff_tree_argv[] = {
4438                 "git", "diff-tree", "-U0", blame->commit->id,
4439                         "--", blame->commit->filename, NULL
4440         };
4441         struct io io = {};
4442         int parent_lineno = -1;
4443         int blamed_lineno = -1;
4444         char *line;
4446         if (!run_io(&io, diff_tree_argv, NULL, IO_RD))
4447                 return;
4449         while ((line = io_get(&io, '\n', TRUE))) {
4450                 if (*line == '@') {
4451                         char *pos = strchr(line, '+');
4453                         parent_lineno = atoi(line + 4);
4454                         if (pos)
4455                                 blamed_lineno = atoi(pos + 1);
4457                 } else if (*line == '+' && parent_lineno != -1) {
4458                         if (blame->lineno == blamed_lineno - 1 &&
4459                             !strcmp(blame->text, line + 1)) {
4460                                 view->lineno = parent_lineno ? parent_lineno - 1 : 0;
4461                                 break;
4462                         }
4463                         blamed_lineno++;
4464                 }
4465         }
4467         done_io(&io);
4470 static enum request
4471 blame_request(struct view *view, enum request request, struct line *line)
4473         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4474         struct blame *blame = line->data;
4476         switch (request) {
4477         case REQ_VIEW_BLAME:
4478                 if (check_blame_commit(blame, TRUE)) {
4479                         string_copy(opt_ref, blame->commit->id);
4480                         string_copy(opt_file, blame->commit->filename);
4481                         if (blame->lineno)
4482                                 view->lineno = blame->lineno;
4483                         open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4484                 }
4485                 break;
4487         case REQ_PARENT:
4488                 if (check_blame_commit(blame, TRUE) &&
4489                     select_commit_parent(blame->commit->id, opt_ref,
4490                                          blame->commit->filename)) {
4491                         string_copy(opt_file, blame->commit->filename);
4492                         setup_blame_parent_line(view, blame);
4493                         open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4494                 }
4495                 break;
4497         case REQ_ENTER:
4498                 if (!check_blame_commit(blame, FALSE))
4499                         break;
4501                 if (view_is_displayed(VIEW(REQ_VIEW_DIFF)) &&
4502                     !strcmp(blame->commit->id, VIEW(REQ_VIEW_DIFF)->ref))
4503                         break;
4505                 if (!strcmp(blame->commit->id, NULL_ID)) {
4506                         struct view *diff = VIEW(REQ_VIEW_DIFF);
4507                         const char *diff_index_argv[] = {
4508                                 "git", "diff-index", "--root", "--patch-with-stat",
4509                                         "-C", "-M", "HEAD", "--", view->vid, NULL
4510                         };
4512                         if (!blame->commit->has_previous) {
4513                                 diff_index_argv[1] = "diff";
4514                                 diff_index_argv[2] = "--no-color";
4515                                 diff_index_argv[6] = "--";
4516                                 diff_index_argv[7] = "/dev/null";
4517                         }
4519                         if (!prepare_update(diff, diff_index_argv, NULL, FORMAT_DASH)) {
4520                                 report("Failed to allocate diff command");
4521                                 break;
4522                         }
4523                         flags |= OPEN_PREPARED;
4524                 }
4526                 open_view(view, REQ_VIEW_DIFF, flags);
4527                 if (VIEW(REQ_VIEW_DIFF)->pipe && !strcmp(blame->commit->id, NULL_ID))
4528                         string_copy_rev(VIEW(REQ_VIEW_DIFF)->ref, NULL_ID);
4529                 break;
4531         default:
4532                 return request;
4533         }
4535         return REQ_NONE;
4538 static bool
4539 blame_grep(struct view *view, struct line *line)
4541         struct blame *blame = line->data;
4542         struct blame_commit *commit = blame->commit;
4543         regmatch_t pmatch;
4545 #define MATCH(text, on)                                                 \
4546         (on && *text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4548         if (commit) {
4549                 char buf[DATE_COLS + 1];
4551                 if (MATCH(commit->title, 1) ||
4552                     MATCH(commit->author, opt_author) ||
4553                     MATCH(commit->id, opt_date))
4554                         return TRUE;
4556                 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
4557                     MATCH(buf, 1))
4558                         return TRUE;
4559         }
4561         return MATCH(blame->text, 1);
4563 #undef MATCH
4566 static void
4567 blame_select(struct view *view, struct line *line)
4569         struct blame *blame = line->data;
4570         struct blame_commit *commit = blame->commit;
4572         if (!commit)
4573                 return;
4575         if (!strcmp(commit->id, NULL_ID))
4576                 string_ncopy(ref_commit, "HEAD", 4);
4577         else
4578                 string_copy_rev(ref_commit, commit->id);
4581 static struct view_ops blame_ops = {
4582         "line",
4583         NULL,
4584         blame_open,
4585         blame_read,
4586         blame_draw,
4587         blame_request,
4588         blame_grep,
4589         blame_select,
4590 };
4592 /*
4593  * Status backend
4594  */
4596 struct status {
4597         char status;
4598         struct {
4599                 mode_t mode;
4600                 char rev[SIZEOF_REV];
4601                 char name[SIZEOF_STR];
4602         } old;
4603         struct {
4604                 mode_t mode;
4605                 char rev[SIZEOF_REV];
4606                 char name[SIZEOF_STR];
4607         } new;
4608 };
4610 static char status_onbranch[SIZEOF_STR];
4611 static struct status stage_status;
4612 static enum line_type stage_line_type;
4613 static size_t stage_chunks;
4614 static int *stage_chunk;
4616 /* This should work even for the "On branch" line. */
4617 static inline bool
4618 status_has_none(struct view *view, struct line *line)
4620         return line < view->line + view->lines && !line[1].data;
4623 /* Get fields from the diff line:
4624  * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
4625  */
4626 static inline bool
4627 status_get_diff(struct status *file, const char *buf, size_t bufsize)
4629         const char *old_mode = buf +  1;
4630         const char *new_mode = buf +  8;
4631         const char *old_rev  = buf + 15;
4632         const char *new_rev  = buf + 56;
4633         const char *status   = buf + 97;
4635         if (bufsize < 98 ||
4636             old_mode[-1] != ':' ||
4637             new_mode[-1] != ' ' ||
4638             old_rev[-1]  != ' ' ||
4639             new_rev[-1]  != ' ' ||
4640             status[-1]   != ' ')
4641                 return FALSE;
4643         file->status = *status;
4645         string_copy_rev(file->old.rev, old_rev);
4646         string_copy_rev(file->new.rev, new_rev);
4648         file->old.mode = strtoul(old_mode, NULL, 8);
4649         file->new.mode = strtoul(new_mode, NULL, 8);
4651         file->old.name[0] = file->new.name[0] = 0;
4653         return TRUE;
4656 static bool
4657 status_run(struct view *view, const char *argv[], char status, enum line_type type)
4659         struct status *unmerged = NULL;
4660         char *buf;
4661         struct io io = {};
4663         if (!run_io(&io, argv, NULL, IO_RD))
4664                 return FALSE;
4666         add_line_data(view, NULL, type);
4668         while ((buf = io_get(&io, 0, TRUE))) {
4669                 struct status *file = unmerged;
4671                 if (!file) {
4672                         file = calloc(1, sizeof(*file));
4673                         if (!file || !add_line_data(view, file, type))
4674                                 goto error_out;
4675                 }
4677                 /* Parse diff info part. */
4678                 if (status) {
4679                         file->status = status;
4680                         if (status == 'A')
4681                                 string_copy(file->old.rev, NULL_ID);
4683                 } else if (!file->status || file == unmerged) {
4684                         if (!status_get_diff(file, buf, strlen(buf)))
4685                                 goto error_out;
4687                         buf = io_get(&io, 0, TRUE);
4688                         if (!buf)
4689                                 break;
4691                         /* Collapse all modified entries that follow an
4692                          * associated unmerged entry. */
4693                         if (unmerged == file) {
4694                                 unmerged->status = 'U';
4695                                 unmerged = NULL;
4696                         } else if (file->status == 'U') {
4697                                 unmerged = file;
4698                         }
4699                 }
4701                 /* Grab the old name for rename/copy. */
4702                 if (!*file->old.name &&
4703                     (file->status == 'R' || file->status == 'C')) {
4704                         string_ncopy(file->old.name, buf, strlen(buf));
4706                         buf = io_get(&io, 0, TRUE);
4707                         if (!buf)
4708                                 break;
4709                 }
4711                 /* git-ls-files just delivers a NUL separated list of
4712                  * file names similar to the second half of the
4713                  * git-diff-* output. */
4714                 string_ncopy(file->new.name, buf, strlen(buf));
4715                 if (!*file->old.name)
4716                         string_copy(file->old.name, file->new.name);
4717                 file = NULL;
4718         }
4720         if (io_error(&io)) {
4721 error_out:
4722                 done_io(&io);
4723                 return FALSE;
4724         }
4726         if (!view->line[view->lines - 1].data)
4727                 add_line_data(view, NULL, LINE_STAT_NONE);
4729         done_io(&io);
4730         return TRUE;
4733 /* Don't show unmerged entries in the staged section. */
4734 static const char *status_diff_index_argv[] = {
4735         "git", "diff-index", "-z", "--diff-filter=ACDMRTXB",
4736                              "--cached", "-M", "HEAD", NULL
4737 };
4739 static const char *status_diff_files_argv[] = {
4740         "git", "diff-files", "-z", NULL
4741 };
4743 static const char *status_list_other_argv[] = {
4744         "git", "ls-files", "-z", "--others", "--exclude-standard", NULL
4745 };
4747 static const char *status_list_no_head_argv[] = {
4748         "git", "ls-files", "-z", "--cached", "--exclude-standard", NULL
4749 };
4751 static const char *update_index_argv[] = {
4752         "git", "update-index", "-q", "--unmerged", "--refresh", NULL
4753 };
4755 /* Restore the previous line number to stay in the context or select a
4756  * line with something that can be updated. */
4757 static void
4758 status_restore(struct view *view)
4760         if (view->p_lineno >= view->lines)
4761                 view->p_lineno = view->lines - 1;
4762         while (view->p_lineno < view->lines && !view->line[view->p_lineno].data)
4763                 view->p_lineno++;
4764         while (view->p_lineno > 0 && !view->line[view->p_lineno].data)
4765                 view->p_lineno--;
4767         /* If the above fails, always skip the "On branch" line. */
4768         if (view->p_lineno < view->lines)
4769                 view->lineno = view->p_lineno;
4770         else
4771                 view->lineno = 1;
4773         if (view->lineno < view->offset)
4774                 view->offset = view->lineno;
4775         else if (view->offset + view->height <= view->lineno)
4776                 view->offset = view->lineno - view->height + 1;
4778         view->p_restore = FALSE;
4781 static void
4782 status_update_onbranch(void)
4784         static const char *paths[][2] = {
4785                 { "rebase-apply/rebasing",      "Rebasing" },
4786                 { "rebase-apply/applying",      "Applying mailbox" },
4787                 { "rebase-apply/",              "Rebasing mailbox" },
4788                 { "rebase-merge/interactive",   "Interactive rebase" },
4789                 { "rebase-merge/",              "Rebase merge" },
4790                 { "MERGE_HEAD",                 "Merging" },
4791                 { "BISECT_LOG",                 "Bisecting" },
4792                 { "HEAD",                       "On branch" },
4793         };
4794         char buf[SIZEOF_STR];
4795         struct stat stat;
4796         int i;
4798         if (is_initial_commit()) {
4799                 string_copy(status_onbranch, "Initial commit");
4800                 return;
4801         }
4803         for (i = 0; i < ARRAY_SIZE(paths); i++) {
4804                 char *head = opt_head;
4806                 if (!string_format(buf, "%s/%s", opt_git_dir, paths[i][0]) ||
4807                     lstat(buf, &stat) < 0)
4808                         continue;
4810                 if (!*opt_head) {
4811                         struct io io = {};
4813                         if (string_format(buf, "%s/rebase-merge/head-name", opt_git_dir) &&
4814                             io_open(&io, buf) &&
4815                             io_read_buf(&io, buf, sizeof(buf))) {
4816                                 head = chomp_string(buf);
4817                                 if (!prefixcmp(head, "refs/heads/"))
4818                                         head += STRING_SIZE("refs/heads/");
4819                         }
4820                 }
4822                 if (!string_format(status_onbranch, "%s %s", paths[i][1], head))
4823                         string_copy(status_onbranch, opt_head);
4824                 return;
4825         }
4827         string_copy(status_onbranch, "Not currently on any branch");
4830 /* First parse staged info using git-diff-index(1), then parse unstaged
4831  * info using git-diff-files(1), and finally untracked files using
4832  * git-ls-files(1). */
4833 static bool
4834 status_open(struct view *view)
4836         reset_view(view);
4838         add_line_data(view, NULL, LINE_STAT_HEAD);
4839         status_update_onbranch();
4841         run_io_bg(update_index_argv);
4843         if (is_initial_commit()) {
4844                 if (!status_run(view, status_list_no_head_argv, 'A', LINE_STAT_STAGED))
4845                         return FALSE;
4846         } else if (!status_run(view, status_diff_index_argv, 0, LINE_STAT_STAGED)) {
4847                 return FALSE;
4848         }
4850         if (!status_run(view, status_diff_files_argv, 0, LINE_STAT_UNSTAGED) ||
4851             !status_run(view, status_list_other_argv, '?', LINE_STAT_UNTRACKED))
4852                 return FALSE;
4854         /* Restore the exact position or use the specialized restore
4855          * mode? */
4856         if (!view->p_restore)
4857                 status_restore(view);
4858         return TRUE;
4861 static bool
4862 status_draw(struct view *view, struct line *line, unsigned int lineno)
4864         struct status *status = line->data;
4865         enum line_type type;
4866         const char *text;
4868         if (!status) {
4869                 switch (line->type) {
4870                 case LINE_STAT_STAGED:
4871                         type = LINE_STAT_SECTION;
4872                         text = "Changes to be committed:";
4873                         break;
4875                 case LINE_STAT_UNSTAGED:
4876                         type = LINE_STAT_SECTION;
4877                         text = "Changed but not updated:";
4878                         break;
4880                 case LINE_STAT_UNTRACKED:
4881                         type = LINE_STAT_SECTION;
4882                         text = "Untracked files:";
4883                         break;
4885                 case LINE_STAT_NONE:
4886                         type = LINE_DEFAULT;
4887                         text = "  (no files)";
4888                         break;
4890                 case LINE_STAT_HEAD:
4891                         type = LINE_STAT_HEAD;
4892                         text = status_onbranch;
4893                         break;
4895                 default:
4896                         return FALSE;
4897                 }
4898         } else {
4899                 static char buf[] = { '?', ' ', ' ', ' ', 0 };
4901                 buf[0] = status->status;
4902                 if (draw_text(view, line->type, buf, TRUE))
4903                         return TRUE;
4904                 type = LINE_DEFAULT;
4905                 text = status->new.name;
4906         }
4908         draw_text(view, type, text, TRUE);
4909         return TRUE;
4912 static enum request
4913 status_load_error(struct view *view, struct view *stage, const char *path)
4915         if (displayed_views() == 2 || display[current_view] != view)
4916                 maximize_view(view);
4917         report("Failed to load '%s': %s", path, io_strerror(&stage->io));
4918         return REQ_NONE;
4921 static enum request
4922 status_enter(struct view *view, struct line *line)
4924         struct status *status = line->data;
4925         const char *oldpath = status ? status->old.name : NULL;
4926         /* Diffs for unmerged entries are empty when passing the new
4927          * path, so leave it empty. */
4928         const char *newpath = status && status->status != 'U' ? status->new.name : NULL;
4929         const char *info;
4930         enum open_flags split;
4931         struct view *stage = VIEW(REQ_VIEW_STAGE);
4933         if (line->type == LINE_STAT_NONE ||
4934             (!status && line[1].type == LINE_STAT_NONE)) {
4935                 report("No file to diff");
4936                 return REQ_NONE;
4937         }
4939         switch (line->type) {
4940         case LINE_STAT_STAGED:
4941                 if (is_initial_commit()) {
4942                         const char *no_head_diff_argv[] = {
4943                                 "git", "diff", "--no-color", "--patch-with-stat",
4944                                         "--", "/dev/null", newpath, NULL
4945                         };
4947                         if (!prepare_update(stage, no_head_diff_argv, opt_cdup, FORMAT_DASH))
4948                                 return status_load_error(view, stage, newpath);
4949                 } else {
4950                         const char *index_show_argv[] = {
4951                                 "git", "diff-index", "--root", "--patch-with-stat",
4952                                         "-C", "-M", "--cached", "HEAD", "--",
4953                                         oldpath, newpath, NULL
4954                         };
4956                         if (!prepare_update(stage, index_show_argv, opt_cdup, FORMAT_DASH))
4957                                 return status_load_error(view, stage, newpath);
4958                 }
4960                 if (status)
4961                         info = "Staged changes to %s";
4962                 else
4963                         info = "Staged changes";
4964                 break;
4966         case LINE_STAT_UNSTAGED:
4967         {
4968                 const char *files_show_argv[] = {
4969                         "git", "diff-files", "--root", "--patch-with-stat",
4970                                 "-C", "-M", "--", oldpath, newpath, NULL
4971                 };
4973                 if (!prepare_update(stage, files_show_argv, opt_cdup, FORMAT_DASH))
4974                         return status_load_error(view, stage, newpath);
4975                 if (status)
4976                         info = "Unstaged changes to %s";
4977                 else
4978                         info = "Unstaged changes";
4979                 break;
4980         }
4981         case LINE_STAT_UNTRACKED:
4982                 if (!newpath) {
4983                         report("No file to show");
4984                         return REQ_NONE;
4985                 }
4987                 if (!suffixcmp(status->new.name, -1, "/")) {
4988                         report("Cannot display a directory");
4989                         return REQ_NONE;
4990                 }
4992                 if (!prepare_update_file(stage, newpath))
4993                         return status_load_error(view, stage, newpath);
4994                 info = "Untracked file %s";
4995                 break;
4997         case LINE_STAT_HEAD:
4998                 return REQ_NONE;
5000         default:
5001                 die("line type %d not handled in switch", line->type);
5002         }
5004         split = view_is_displayed(view) ? OPEN_SPLIT : 0;
5005         open_view(view, REQ_VIEW_STAGE, OPEN_PREPARED | split);
5006         if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
5007                 if (status) {
5008                         stage_status = *status;
5009                 } else {
5010                         memset(&stage_status, 0, sizeof(stage_status));
5011                 }
5013                 stage_line_type = line->type;
5014                 stage_chunks = 0;
5015                 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
5016         }
5018         return REQ_NONE;
5021 static bool
5022 status_exists(struct status *status, enum line_type type)
5024         struct view *view = VIEW(REQ_VIEW_STATUS);
5025         unsigned long lineno;
5027         for (lineno = 0; lineno < view->lines; lineno++) {
5028                 struct line *line = &view->line[lineno];
5029                 struct status *pos = line->data;
5031                 if (line->type != type)
5032                         continue;
5033                 if (!pos && (!status || !status->status) && line[1].data) {
5034                         select_view_line(view, lineno);
5035                         return TRUE;
5036                 }
5037                 if (pos && !strcmp(status->new.name, pos->new.name)) {
5038                         select_view_line(view, lineno);
5039                         return TRUE;
5040                 }
5041         }
5043         return FALSE;
5047 static bool
5048 status_update_prepare(struct io *io, enum line_type type)
5050         const char *staged_argv[] = {
5051                 "git", "update-index", "-z", "--index-info", NULL
5052         };
5053         const char *others_argv[] = {
5054                 "git", "update-index", "-z", "--add", "--remove", "--stdin", NULL
5055         };
5057         switch (type) {
5058         case LINE_STAT_STAGED:
5059                 return run_io(io, staged_argv, opt_cdup, IO_WR);
5061         case LINE_STAT_UNSTAGED:
5062                 return run_io(io, others_argv, opt_cdup, IO_WR);
5064         case LINE_STAT_UNTRACKED:
5065                 return run_io(io, others_argv, NULL, IO_WR);
5067         default:
5068                 die("line type %d not handled in switch", type);
5069                 return FALSE;
5070         }
5073 static bool
5074 status_update_write(struct io *io, struct status *status, enum line_type type)
5076         char buf[SIZEOF_STR];
5077         size_t bufsize = 0;
5079         switch (type) {
5080         case LINE_STAT_STAGED:
5081                 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
5082                                         status->old.mode,
5083                                         status->old.rev,
5084                                         status->old.name, 0))
5085                         return FALSE;
5086                 break;
5088         case LINE_STAT_UNSTAGED:
5089         case LINE_STAT_UNTRACKED:
5090                 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
5091                         return FALSE;
5092                 break;
5094         default:
5095                 die("line type %d not handled in switch", type);
5096         }
5098         return io_write(io, buf, bufsize);
5101 static bool
5102 status_update_file(struct status *status, enum line_type type)
5104         struct io io = {};
5105         bool result;
5107         if (!status_update_prepare(&io, type))
5108                 return FALSE;
5110         result = status_update_write(&io, status, type);
5111         return done_io(&io) && result;
5114 static bool
5115 status_update_files(struct view *view, struct line *line)
5117         char buf[sizeof(view->ref)];
5118         struct io io = {};
5119         bool result = TRUE;
5120         struct line *pos = view->line + view->lines;
5121         int files = 0;
5122         int file, done;
5123         int cursor_y, cursor_x;
5125         if (!status_update_prepare(&io, line->type))
5126                 return FALSE;
5128         for (pos = line; pos < view->line + view->lines && pos->data; pos++)
5129                 files++;
5131         string_copy(buf, view->ref);
5132         getsyx(cursor_y, cursor_x);
5133         for (file = 0, done = 5; result && file < files; line++, file++) {
5134                 int almost_done = file * 100 / files;
5136                 if (almost_done > done) {
5137                         done = almost_done;
5138                         string_format(view->ref, "updating file %u of %u (%d%% done)",
5139                                       file, files, done);
5140                         update_view_title(view);
5141                         setsyx(cursor_y, cursor_x);
5142                         doupdate();
5143                 }
5144                 result = status_update_write(&io, line->data, line->type);
5145         }
5146         string_copy(view->ref, buf);
5148         return done_io(&io) && result;
5151 static bool
5152 status_update(struct view *view)
5154         struct line *line = &view->line[view->lineno];
5156         assert(view->lines);
5158         if (!line->data) {
5159                 /* This should work even for the "On branch" line. */
5160                 if (line < view->line + view->lines && !line[1].data) {
5161                         report("Nothing to update");
5162                         return FALSE;
5163                 }
5165                 if (!status_update_files(view, line + 1)) {
5166                         report("Failed to update file status");
5167                         return FALSE;
5168                 }
5170         } else if (!status_update_file(line->data, line->type)) {
5171                 report("Failed to update file status");
5172                 return FALSE;
5173         }
5175         return TRUE;
5178 static bool
5179 status_revert(struct status *status, enum line_type type, bool has_none)
5181         if (!status || type != LINE_STAT_UNSTAGED) {
5182                 if (type == LINE_STAT_STAGED) {
5183                         report("Cannot revert changes to staged files");
5184                 } else if (type == LINE_STAT_UNTRACKED) {
5185                         report("Cannot revert changes to untracked files");
5186                 } else if (has_none) {
5187                         report("Nothing to revert");
5188                 } else {
5189                         report("Cannot revert changes to multiple files");
5190                 }
5191                 return FALSE;
5193         } else {
5194                 char mode[10] = "100644";
5195                 const char *reset_argv[] = {
5196                         "git", "update-index", "--cacheinfo", mode,
5197                                 status->old.rev, status->old.name, NULL
5198                 };
5199                 const char *checkout_argv[] = {
5200                         "git", "checkout", "--", status->old.name, NULL
5201                 };
5203                 if (!prompt_yesno("Are you sure you want to overwrite any changes?"))
5204                         return FALSE;
5205                 string_format(mode, "%o", status->old.mode);
5206                 return (status->status != 'U' || run_io_fg(reset_argv, opt_cdup)) &&
5207                         run_io_fg(checkout_argv, opt_cdup);
5208         }
5211 static enum request
5212 status_request(struct view *view, enum request request, struct line *line)
5214         struct status *status = line->data;
5216         switch (request) {
5217         case REQ_STATUS_UPDATE:
5218                 if (!status_update(view))
5219                         return REQ_NONE;
5220                 break;
5222         case REQ_STATUS_REVERT:
5223                 if (!status_revert(status, line->type, status_has_none(view, line)))
5224                         return REQ_NONE;
5225                 break;
5227         case REQ_STATUS_MERGE:
5228                 if (!status || status->status != 'U') {
5229                         report("Merging only possible for files with unmerged status ('U').");
5230                         return REQ_NONE;
5231                 }
5232                 open_mergetool(status->new.name);
5233                 break;
5235         case REQ_EDIT:
5236                 if (!status)
5237                         return request;
5238                 if (status->status == 'D') {
5239                         report("File has been deleted.");
5240                         return REQ_NONE;
5241                 }
5243                 open_editor(status->status != '?', status->new.name);
5244                 break;
5246         case REQ_VIEW_BLAME:
5247                 if (status) {
5248                         string_copy(opt_file, status->new.name);
5249                         opt_ref[0] = 0;
5250                 }
5251                 return request;
5253         case REQ_ENTER:
5254                 /* After returning the status view has been split to
5255                  * show the stage view. No further reloading is
5256                  * necessary. */
5257                 return status_enter(view, line);
5259         case REQ_REFRESH:
5260                 /* Simply reload the view. */
5261                 break;
5263         default:
5264                 return request;
5265         }
5267         open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
5269         return REQ_NONE;
5272 static void
5273 status_select(struct view *view, struct line *line)
5275         struct status *status = line->data;
5276         char file[SIZEOF_STR] = "all files";
5277         const char *text;
5278         const char *key;
5280         if (status && !string_format(file, "'%s'", status->new.name))
5281                 return;
5283         if (!status && line[1].type == LINE_STAT_NONE)
5284                 line++;
5286         switch (line->type) {
5287         case LINE_STAT_STAGED:
5288                 text = "Press %s to unstage %s for commit";
5289                 break;
5291         case LINE_STAT_UNSTAGED:
5292                 text = "Press %s to stage %s for commit";
5293                 break;
5295         case LINE_STAT_UNTRACKED:
5296                 text = "Press %s to stage %s for addition";
5297                 break;
5299         case LINE_STAT_HEAD:
5300         case LINE_STAT_NONE:
5301                 text = "Nothing to update";
5302                 break;
5304         default:
5305                 die("line type %d not handled in switch", line->type);
5306         }
5308         if (status && status->status == 'U') {
5309                 text = "Press %s to resolve conflict in %s";
5310                 key = get_key(REQ_STATUS_MERGE);
5312         } else {
5313                 key = get_key(REQ_STATUS_UPDATE);
5314         }
5316         string_format(view->ref, text, key, file);
5319 static bool
5320 status_grep(struct view *view, struct line *line)
5322         struct status *status = line->data;
5323         enum { S_STATUS, S_NAME, S_END } state;
5324         char buf[2] = "?";
5325         regmatch_t pmatch;
5327         if (!status)
5328                 return FALSE;
5330         for (state = S_STATUS; state < S_END; state++) {
5331                 const char *text;
5333                 switch (state) {
5334                 case S_NAME:    text = status->new.name;        break;
5335                 case S_STATUS:
5336                         buf[0] = status->status;
5337                         text = buf;
5338                         break;
5340                 default:
5341                         return FALSE;
5342                 }
5344                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5345                         return TRUE;
5346         }
5348         return FALSE;
5351 static struct view_ops status_ops = {
5352         "file",
5353         NULL,
5354         status_open,
5355         NULL,
5356         status_draw,
5357         status_request,
5358         status_grep,
5359         status_select,
5360 };
5363 static bool
5364 stage_diff_write(struct io *io, struct line *line, struct line *end)
5366         while (line < end) {
5367                 if (!io_write(io, line->data, strlen(line->data)) ||
5368                     !io_write(io, "\n", 1))
5369                         return FALSE;
5370                 line++;
5371                 if (line->type == LINE_DIFF_CHUNK ||
5372                     line->type == LINE_DIFF_HEADER)
5373                         break;
5374         }
5376         return TRUE;
5379 static struct line *
5380 stage_diff_find(struct view *view, struct line *line, enum line_type type)
5382         for (; view->line < line; line--)
5383                 if (line->type == type)
5384                         return line;
5386         return NULL;
5389 static bool
5390 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
5392         const char *apply_argv[SIZEOF_ARG] = {
5393                 "git", "apply", "--whitespace=nowarn", NULL
5394         };
5395         struct line *diff_hdr;
5396         struct io io = {};
5397         int argc = 3;
5399         diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
5400         if (!diff_hdr)
5401                 return FALSE;
5403         if (!revert)
5404                 apply_argv[argc++] = "--cached";
5405         if (revert || stage_line_type == LINE_STAT_STAGED)
5406                 apply_argv[argc++] = "-R";
5407         apply_argv[argc++] = "-";
5408         apply_argv[argc++] = NULL;
5409         if (!run_io(&io, apply_argv, opt_cdup, IO_WR))
5410                 return FALSE;
5412         if (!stage_diff_write(&io, diff_hdr, chunk) ||
5413             !stage_diff_write(&io, chunk, view->line + view->lines))
5414                 chunk = NULL;
5416         done_io(&io);
5417         run_io_bg(update_index_argv);
5419         return chunk ? TRUE : FALSE;
5422 static bool
5423 stage_update(struct view *view, struct line *line)
5425         struct line *chunk = NULL;
5427         if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
5428                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5430         if (chunk) {
5431                 if (!stage_apply_chunk(view, chunk, FALSE)) {
5432                         report("Failed to apply chunk");
5433                         return FALSE;
5434                 }
5436         } else if (!stage_status.status) {
5437                 view = VIEW(REQ_VIEW_STATUS);
5439                 for (line = view->line; line < view->line + view->lines; line++)
5440                         if (line->type == stage_line_type)
5441                                 break;
5443                 if (!status_update_files(view, line + 1)) {
5444                         report("Failed to update files");
5445                         return FALSE;
5446                 }
5448         } else if (!status_update_file(&stage_status, stage_line_type)) {
5449                 report("Failed to update file");
5450                 return FALSE;
5451         }
5453         return TRUE;
5456 static bool
5457 stage_revert(struct view *view, struct line *line)
5459         struct line *chunk = NULL;
5461         if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
5462                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5464         if (chunk) {
5465                 if (!prompt_yesno("Are you sure you want to revert changes?"))
5466                         return FALSE;
5468                 if (!stage_apply_chunk(view, chunk, TRUE)) {
5469                         report("Failed to revert chunk");
5470                         return FALSE;
5471                 }
5472                 return TRUE;
5474         } else {
5475                 return status_revert(stage_status.status ? &stage_status : NULL,
5476                                      stage_line_type, FALSE);
5477         }
5481 static void
5482 stage_next(struct view *view, struct line *line)
5484         int i;
5486         if (!stage_chunks) {
5487                 static size_t alloc = 0;
5488                 int *tmp;
5490                 for (line = view->line; line < view->line + view->lines; line++) {
5491                         if (line->type != LINE_DIFF_CHUNK)
5492                                 continue;
5494                         tmp = realloc_items(stage_chunk, &alloc,
5495                                             stage_chunks, sizeof(*tmp));
5496                         if (!tmp) {
5497                                 report("Allocation failure");
5498                                 return;
5499                         }
5501                         stage_chunk = tmp;
5502                         stage_chunk[stage_chunks++] = line - view->line;
5503                 }
5504         }
5506         for (i = 0; i < stage_chunks; i++) {
5507                 if (stage_chunk[i] > view->lineno) {
5508                         do_scroll_view(view, stage_chunk[i] - view->lineno);
5509                         report("Chunk %d of %d", i + 1, stage_chunks);
5510                         return;
5511                 }
5512         }
5514         report("No next chunk found");
5517 static enum request
5518 stage_request(struct view *view, enum request request, struct line *line)
5520         switch (request) {
5521         case REQ_STATUS_UPDATE:
5522                 if (!stage_update(view, line))
5523                         return REQ_NONE;
5524                 break;
5526         case REQ_STATUS_REVERT:
5527                 if (!stage_revert(view, line))
5528                         return REQ_NONE;
5529                 break;
5531         case REQ_STAGE_NEXT:
5532                 if (stage_line_type == LINE_STAT_UNTRACKED) {
5533                         report("File is untracked; press %s to add",
5534                                get_key(REQ_STATUS_UPDATE));
5535                         return REQ_NONE;
5536                 }
5537                 stage_next(view, line);
5538                 return REQ_NONE;
5540         case REQ_EDIT:
5541                 if (!stage_status.new.name[0])
5542                         return request;
5543                 if (stage_status.status == 'D') {
5544                         report("File has been deleted.");
5545                         return REQ_NONE;
5546                 }
5548                 open_editor(stage_status.status != '?', stage_status.new.name);
5549                 break;
5551         case REQ_REFRESH:
5552                 /* Reload everything ... */
5553                 break;
5555         case REQ_VIEW_BLAME:
5556                 if (stage_status.new.name[0]) {
5557                         string_copy(opt_file, stage_status.new.name);
5558                         opt_ref[0] = 0;
5559                 }
5560                 return request;
5562         case REQ_ENTER:
5563                 return pager_request(view, request, line);
5565         default:
5566                 return request;
5567         }
5569         VIEW(REQ_VIEW_STATUS)->p_restore = TRUE;
5570         open_view(view, REQ_VIEW_STATUS, OPEN_REFRESH);
5572         /* Check whether the staged entry still exists, and close the
5573          * stage view if it doesn't. */
5574         if (!status_exists(&stage_status, stage_line_type)) {
5575                 status_restore(VIEW(REQ_VIEW_STATUS));
5576                 return REQ_VIEW_CLOSE;
5577         }
5579         if (stage_line_type == LINE_STAT_UNTRACKED) {
5580                 if (!suffixcmp(stage_status.new.name, -1, "/")) {
5581                         report("Cannot display a directory");
5582                         return REQ_NONE;
5583                 }
5585                 if (!prepare_update_file(view, stage_status.new.name)) {
5586                         report("Failed to open file: %s", strerror(errno));
5587                         return REQ_NONE;
5588                 }
5589         }
5590         open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
5592         return REQ_NONE;
5595 static struct view_ops stage_ops = {
5596         "line",
5597         NULL,
5598         NULL,
5599         pager_read,
5600         pager_draw,
5601         stage_request,
5602         pager_grep,
5603         pager_select,
5604 };
5607 /*
5608  * Revision graph
5609  */
5611 struct commit {
5612         char id[SIZEOF_REV];            /* SHA1 ID. */
5613         char title[128];                /* First line of the commit message. */
5614         char author[75];                /* Author of the commit. */
5615         struct tm time;                 /* Date from the author ident. */
5616         struct ref **refs;              /* Repository references. */
5617         chtype graph[SIZEOF_REVGRAPH];  /* Ancestry chain graphics. */
5618         size_t graph_size;              /* The width of the graph array. */
5619         bool has_parents;               /* Rewritten --parents seen. */
5620 };
5622 /* Size of rev graph with no  "padding" columns */
5623 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
5625 struct rev_graph {
5626         struct rev_graph *prev, *next, *parents;
5627         char rev[SIZEOF_REVITEMS][SIZEOF_REV];
5628         size_t size;
5629         struct commit *commit;
5630         size_t pos;
5631         unsigned int boundary:1;
5632 };
5634 /* Parents of the commit being visualized. */
5635 static struct rev_graph graph_parents[4];
5637 /* The current stack of revisions on the graph. */
5638 static struct rev_graph graph_stacks[4] = {
5639         { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
5640         { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
5641         { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
5642         { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
5643 };
5645 static inline bool
5646 graph_parent_is_merge(struct rev_graph *graph)
5648         return graph->parents->size > 1;
5651 static inline void
5652 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
5654         struct commit *commit = graph->commit;
5656         if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
5657                 commit->graph[commit->graph_size++] = symbol;
5660 static void
5661 clear_rev_graph(struct rev_graph *graph)
5663         graph->boundary = 0;
5664         graph->size = graph->pos = 0;
5665         graph->commit = NULL;
5666         memset(graph->parents, 0, sizeof(*graph->parents));
5669 static void
5670 done_rev_graph(struct rev_graph *graph)
5672         if (graph_parent_is_merge(graph) &&
5673             graph->pos < graph->size - 1 &&
5674             graph->next->size == graph->size + graph->parents->size - 1) {
5675                 size_t i = graph->pos + graph->parents->size - 1;
5677                 graph->commit->graph_size = i * 2;
5678                 while (i < graph->next->size - 1) {
5679                         append_to_rev_graph(graph, ' ');
5680                         append_to_rev_graph(graph, '\\');
5681                         i++;
5682                 }
5683         }
5685         clear_rev_graph(graph);
5688 static void
5689 push_rev_graph(struct rev_graph *graph, const char *parent)
5691         int i;
5693         /* "Collapse" duplicate parents lines.
5694          *
5695          * FIXME: This needs to also update update the drawn graph but
5696          * for now it just serves as a method for pruning graph lines. */
5697         for (i = 0; i < graph->size; i++)
5698                 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
5699                         return;
5701         if (graph->size < SIZEOF_REVITEMS) {
5702                 string_copy_rev(graph->rev[graph->size++], parent);
5703         }
5706 static chtype
5707 get_rev_graph_symbol(struct rev_graph *graph)
5709         chtype symbol;
5711         if (graph->boundary)
5712                 symbol = REVGRAPH_BOUND;
5713         else if (graph->parents->size == 0)
5714                 symbol = REVGRAPH_INIT;
5715         else if (graph_parent_is_merge(graph))
5716                 symbol = REVGRAPH_MERGE;
5717         else if (graph->pos >= graph->size)
5718                 symbol = REVGRAPH_BRANCH;
5719         else
5720                 symbol = REVGRAPH_COMMIT;
5722         return symbol;
5725 static void
5726 draw_rev_graph(struct rev_graph *graph)
5728         struct rev_filler {
5729                 chtype separator, line;
5730         };
5731         enum { DEFAULT, RSHARP, RDIAG, LDIAG };
5732         static struct rev_filler fillers[] = {
5733                 { ' ',  '|' },
5734                 { '`',  '.' },
5735                 { '\'', ' ' },
5736                 { '/',  ' ' },
5737         };
5738         chtype symbol = get_rev_graph_symbol(graph);
5739         struct rev_filler *filler;
5740         size_t i;
5742         if (opt_line_graphics)
5743                 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
5745         filler = &fillers[DEFAULT];
5747         for (i = 0; i < graph->pos; i++) {
5748                 append_to_rev_graph(graph, filler->line);
5749                 if (graph_parent_is_merge(graph->prev) &&
5750                     graph->prev->pos == i)
5751                         filler = &fillers[RSHARP];
5753                 append_to_rev_graph(graph, filler->separator);
5754         }
5756         /* Place the symbol for this revision. */
5757         append_to_rev_graph(graph, symbol);
5759         if (graph->prev->size > graph->size)
5760                 filler = &fillers[RDIAG];
5761         else
5762                 filler = &fillers[DEFAULT];
5764         i++;
5766         for (; i < graph->size; i++) {
5767                 append_to_rev_graph(graph, filler->separator);
5768                 append_to_rev_graph(graph, filler->line);
5769                 if (graph_parent_is_merge(graph->prev) &&
5770                     i < graph->prev->pos + graph->parents->size)
5771                         filler = &fillers[RSHARP];
5772                 if (graph->prev->size > graph->size)
5773                         filler = &fillers[LDIAG];
5774         }
5776         if (graph->prev->size > graph->size) {
5777                 append_to_rev_graph(graph, filler->separator);
5778                 if (filler->line != ' ')
5779                         append_to_rev_graph(graph, filler->line);
5780         }
5783 /* Prepare the next rev graph */
5784 static void
5785 prepare_rev_graph(struct rev_graph *graph)
5787         size_t i;
5789         /* First, traverse all lines of revisions up to the active one. */
5790         for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5791                 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5792                         break;
5794                 push_rev_graph(graph->next, graph->rev[graph->pos]);
5795         }
5797         /* Interleave the new revision parent(s). */
5798         for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5799                 push_rev_graph(graph->next, graph->parents->rev[i]);
5801         /* Lastly, put any remaining revisions. */
5802         for (i = graph->pos + 1; i < graph->size; i++)
5803                 push_rev_graph(graph->next, graph->rev[i]);
5806 static void
5807 update_rev_graph(struct view *view, struct rev_graph *graph)
5809         /* If this is the finalizing update ... */
5810         if (graph->commit)
5811                 prepare_rev_graph(graph);
5813         /* Graph visualization needs a one rev look-ahead,
5814          * so the first update doesn't visualize anything. */
5815         if (!graph->prev->commit)
5816                 return;
5818         if (view->lines > 2)
5819                 view->line[view->lines - 3].dirty = 1;
5820         if (view->lines > 1)
5821                 view->line[view->lines - 2].dirty = 1;
5822         draw_rev_graph(graph->prev);
5823         done_rev_graph(graph->prev->prev);
5827 /*
5828  * Main view backend
5829  */
5831 static const char *main_argv[SIZEOF_ARG] = {
5832         "git", "log", "--no-color", "--pretty=raw", "--parents",
5833                       "--topo-order", "%(head)", NULL
5834 };
5836 static bool
5837 main_draw(struct view *view, struct line *line, unsigned int lineno)
5839         struct commit *commit = line->data;
5841         if (!*commit->author)
5842                 return FALSE;
5844         if (opt_date && draw_date(view, &commit->time))
5845                 return TRUE;
5847         if (opt_author && draw_author(view, commit->author))
5848                 return TRUE;
5850         if (opt_rev_graph && commit->graph_size &&
5851             draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5852                 return TRUE;
5854         if (opt_show_refs && commit->refs) {
5855                 size_t i = 0;
5857                 do {
5858                         enum line_type type;
5860                         if (commit->refs[i]->head)
5861                                 type = LINE_MAIN_HEAD;
5862                         else if (commit->refs[i]->ltag)
5863                                 type = LINE_MAIN_LOCAL_TAG;
5864                         else if (commit->refs[i]->tag)
5865                                 type = LINE_MAIN_TAG;
5866                         else if (commit->refs[i]->tracked)
5867                                 type = LINE_MAIN_TRACKED;
5868                         else if (commit->refs[i]->remote)
5869                                 type = LINE_MAIN_REMOTE;
5870                         else
5871                                 type = LINE_MAIN_REF;
5873                         if (draw_text(view, type, "[", TRUE) ||
5874                             draw_text(view, type, commit->refs[i]->name, TRUE) ||
5875                             draw_text(view, type, "]", TRUE))
5876                                 return TRUE;
5878                         if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5879                                 return TRUE;
5880                 } while (commit->refs[i++]->next);
5881         }
5883         draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5884         return TRUE;
5887 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5888 static bool
5889 main_read(struct view *view, char *line)
5891         static struct rev_graph *graph = graph_stacks;
5892         enum line_type type;
5893         struct commit *commit;
5895         if (!line) {
5896                 int i;
5898                 if (!view->lines && !view->parent)
5899                         die("No revisions match the given arguments.");
5900                 if (view->lines > 0) {
5901                         commit = view->line[view->lines - 1].data;
5902                         view->line[view->lines - 1].dirty = 1;
5903                         if (!*commit->author) {
5904                                 view->lines--;
5905                                 free(commit);
5906                                 graph->commit = NULL;
5907                         }
5908                 }
5909                 update_rev_graph(view, graph);
5911                 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5912                         clear_rev_graph(&graph_stacks[i]);
5913                 return TRUE;
5914         }
5916         type = get_line_type(line);
5917         if (type == LINE_COMMIT) {
5918                 commit = calloc(1, sizeof(struct commit));
5919                 if (!commit)
5920                         return FALSE;
5922                 line += STRING_SIZE("commit ");
5923                 if (*line == '-') {
5924                         graph->boundary = 1;
5925                         line++;
5926                 }
5928                 string_copy_rev(commit->id, line);
5929                 commit->refs = get_refs(commit->id);
5930                 graph->commit = commit;
5931                 add_line_data(view, commit, LINE_MAIN_COMMIT);
5933                 while ((line = strchr(line, ' '))) {
5934                         line++;
5935                         push_rev_graph(graph->parents, line);
5936                         commit->has_parents = TRUE;
5937                 }
5938                 return TRUE;
5939         }
5941         if (!view->lines)
5942                 return TRUE;
5943         commit = view->line[view->lines - 1].data;
5945         switch (type) {
5946         case LINE_PARENT:
5947                 if (commit->has_parents)
5948                         break;
5949                 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5950                 break;
5952         case LINE_AUTHOR:
5953                 parse_author_line(line + STRING_SIZE("author "),
5954                                   commit->author, sizeof(commit->author),
5955                                   &commit->time);
5956                 update_rev_graph(view, graph);
5957                 graph = graph->next;
5958                 break;
5960         default:
5961                 /* Fill in the commit title if it has not already been set. */
5962                 if (commit->title[0])
5963                         break;
5965                 /* Require titles to start with a non-space character at the
5966                  * offset used by git log. */
5967                 if (strncmp(line, "    ", 4))
5968                         break;
5969                 line += 4;
5970                 /* Well, if the title starts with a whitespace character,
5971                  * try to be forgiving.  Otherwise we end up with no title. */
5972                 while (isspace(*line))
5973                         line++;
5974                 if (*line == '\0')
5975                         break;
5976                 /* FIXME: More graceful handling of titles; append "..." to
5977                  * shortened titles, etc. */
5979                 string_expand(commit->title, sizeof(commit->title), line, 1);
5980                 view->line[view->lines - 1].dirty = 1;
5981         }
5983         return TRUE;
5986 static enum request
5987 main_request(struct view *view, enum request request, struct line *line)
5989         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5991         switch (request) {
5992         case REQ_ENTER:
5993                 open_view(view, REQ_VIEW_DIFF, flags);
5994                 break;
5995         case REQ_REFRESH:
5996                 load_refs();
5997                 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
5998                 break;
5999         default:
6000                 return request;
6001         }
6003         return REQ_NONE;
6006 static bool
6007 grep_refs(struct ref **refs, regex_t *regex)
6009         regmatch_t pmatch;
6010         size_t i = 0;
6012         if (!refs)
6013                 return FALSE;
6014         do {
6015                 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
6016                         return TRUE;
6017         } while (refs[i++]->next);
6019         return FALSE;
6022 static bool
6023 main_grep(struct view *view, struct line *line)
6025         struct commit *commit = line->data;
6026         enum { S_TITLE, S_AUTHOR, S_DATE, S_REFS, S_END } state;
6027         char buf[DATE_COLS + 1];
6028         regmatch_t pmatch;
6030         for (state = S_TITLE; state < S_END; state++) {
6031                 char *text;
6033                 switch (state) {
6034                 case S_TITLE:   text = commit->title;   break;
6035                 case S_AUTHOR:
6036                         if (!opt_author)
6037                                 continue;
6038                         text = commit->author;
6039                         break;
6040                 case S_DATE:
6041                         if (!opt_date)
6042                                 continue;
6043                         if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
6044                                 continue;
6045                         text = buf;
6046                         break;
6047                 case S_REFS:
6048                         if (!opt_show_refs)
6049                                 continue;
6050                         if (grep_refs(commit->refs, view->regex) == TRUE)
6051                                 return TRUE;
6052                         continue;
6053                 default:
6054                         return FALSE;
6055                 }
6057                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
6058                         return TRUE;
6059         }
6061         return FALSE;
6064 static void
6065 main_select(struct view *view, struct line *line)
6067         struct commit *commit = line->data;
6069         string_copy_rev(view->ref, commit->id);
6070         string_copy_rev(ref_commit, view->ref);
6073 static struct view_ops main_ops = {
6074         "commit",
6075         main_argv,
6076         NULL,
6077         main_read,
6078         main_draw,
6079         main_request,
6080         main_grep,
6081         main_select,
6082 };
6085 /*
6086  * Unicode / UTF-8 handling
6087  *
6088  * NOTE: Much of the following code for dealing with Unicode is derived from
6089  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
6090  * src/intl/charset.c from the UTF-8 branch commit elinks-0.11.0-g31f2c28.
6091  */
6093 static inline int
6094 unicode_width(unsigned long c)
6096         if (c >= 0x1100 &&
6097            (c <= 0x115f                         /* Hangul Jamo */
6098             || c == 0x2329
6099             || c == 0x232a
6100             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
6101                                                 /* CJK ... Yi */
6102             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
6103             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
6104             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
6105             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
6106             || (c >= 0xffe0  && c <= 0xffe6)
6107             || (c >= 0x20000 && c <= 0x2fffd)
6108             || (c >= 0x30000 && c <= 0x3fffd)))
6109                 return 2;
6111         if (c == '\t')
6112                 return opt_tab_size;
6114         return 1;
6117 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
6118  * Illegal bytes are set one. */
6119 static const unsigned char utf8_bytes[256] = {
6120         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,
6121         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,
6122         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,
6123         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,
6124         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,
6125         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,
6126         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,
6127         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,
6128 };
6130 /* Decode UTF-8 multi-byte representation into a Unicode character. */
6131 static inline unsigned long
6132 utf8_to_unicode(const char *string, size_t length)
6134         unsigned long unicode;
6136         switch (length) {
6137         case 1:
6138                 unicode  =   string[0];
6139                 break;
6140         case 2:
6141                 unicode  =  (string[0] & 0x1f) << 6;
6142                 unicode +=  (string[1] & 0x3f);
6143                 break;
6144         case 3:
6145                 unicode  =  (string[0] & 0x0f) << 12;
6146                 unicode += ((string[1] & 0x3f) << 6);
6147                 unicode +=  (string[2] & 0x3f);
6148                 break;
6149         case 4:
6150                 unicode  =  (string[0] & 0x0f) << 18;
6151                 unicode += ((string[1] & 0x3f) << 12);
6152                 unicode += ((string[2] & 0x3f) << 6);
6153                 unicode +=  (string[3] & 0x3f);
6154                 break;
6155         case 5:
6156                 unicode  =  (string[0] & 0x0f) << 24;
6157                 unicode += ((string[1] & 0x3f) << 18);
6158                 unicode += ((string[2] & 0x3f) << 12);
6159                 unicode += ((string[3] & 0x3f) << 6);
6160                 unicode +=  (string[4] & 0x3f);
6161                 break;
6162         case 6:
6163                 unicode  =  (string[0] & 0x01) << 30;
6164                 unicode += ((string[1] & 0x3f) << 24);
6165                 unicode += ((string[2] & 0x3f) << 18);
6166                 unicode += ((string[3] & 0x3f) << 12);
6167                 unicode += ((string[4] & 0x3f) << 6);
6168                 unicode +=  (string[5] & 0x3f);
6169                 break;
6170         default:
6171                 die("Invalid Unicode length");
6172         }
6174         /* Invalid characters could return the special 0xfffd value but NUL
6175          * should be just as good. */
6176         return unicode > 0xffff ? 0 : unicode;
6179 /* Calculates how much of string can be shown within the given maximum width
6180  * and sets trimmed parameter to non-zero value if all of string could not be
6181  * shown. If the reserve flag is TRUE, it will reserve at least one
6182  * trailing character, which can be useful when drawing a delimiter.
6183  *
6184  * Returns the number of bytes to output from string to satisfy max_width. */
6185 static size_t
6186 utf8_length(const char **start, size_t skip, int *width, size_t max_width, int *trimmed, bool reserve)
6188         const char *string = *start;
6189         const char *end = strchr(string, '\0');
6190         unsigned char last_bytes = 0;
6191         size_t last_ucwidth = 0;
6193         *width = 0;
6194         *trimmed = 0;
6196         while (string < end) {
6197                 int c = *(unsigned char *) string;
6198                 unsigned char bytes = utf8_bytes[c];
6199                 size_t ucwidth;
6200                 unsigned long unicode;
6202                 if (string + bytes > end)
6203                         break;
6205                 /* Change representation to figure out whether
6206                  * it is a single- or double-width character. */
6208                 unicode = utf8_to_unicode(string, bytes);
6209                 /* FIXME: Graceful handling of invalid Unicode character. */
6210                 if (!unicode)
6211                         break;
6213                 ucwidth = unicode_width(unicode);
6214                 if (skip > 0) {
6215                         skip -= ucwidth <= skip ? ucwidth : skip;
6216                         *start += bytes;
6217                 }
6218                 *width  += ucwidth;
6219                 if (*width > max_width) {
6220                         *trimmed = 1;
6221                         *width -= ucwidth;
6222                         if (reserve && *width == max_width) {
6223                                 string -= last_bytes;
6224                                 *width -= last_ucwidth;
6225                         }
6226                         break;
6227                 }
6229                 string  += bytes;
6230                 last_bytes = ucwidth ? bytes : 0;
6231                 last_ucwidth = ucwidth;
6232         }
6234         return string - *start;
6238 /*
6239  * Status management
6240  */
6242 /* Whether or not the curses interface has been initialized. */
6243 static bool cursed = FALSE;
6245 /* Terminal hacks and workarounds. */
6246 static bool use_scroll_redrawwin;
6247 static bool use_scroll_status_wclear;
6249 /* The status window is used for polling keystrokes. */
6250 static WINDOW *status_win;
6252 /* Reading from the prompt? */
6253 static bool input_mode = FALSE;
6255 static bool status_empty = FALSE;
6257 /* Update status and title window. */
6258 static void
6259 report(const char *msg, ...)
6261         struct view *view = display[current_view];
6263         if (input_mode)
6264                 return;
6266         if (!view) {
6267                 char buf[SIZEOF_STR];
6268                 va_list args;
6270                 va_start(args, msg);
6271                 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
6272                         buf[sizeof(buf) - 1] = 0;
6273                         buf[sizeof(buf) - 2] = '.';
6274                         buf[sizeof(buf) - 3] = '.';
6275                         buf[sizeof(buf) - 4] = '.';
6276                 }
6277                 va_end(args);
6278                 die("%s", buf);
6279         }
6281         if (!status_empty || *msg) {
6282                 va_list args;
6284                 va_start(args, msg);
6286                 wmove(status_win, 0, 0);
6287                 if (view->has_scrolled && use_scroll_status_wclear)
6288                         wclear(status_win);
6289                 if (*msg) {
6290                         vwprintw(status_win, msg, args);
6291                         status_empty = FALSE;
6292                 } else {
6293                         status_empty = TRUE;
6294                 }
6295                 wclrtoeol(status_win);
6296                 wnoutrefresh(status_win);
6298                 va_end(args);
6299         }
6301         update_view_title(view);
6304 /* Controls when nodelay should be in effect when polling user input. */
6305 static void
6306 set_nonblocking_input(bool loading)
6308         static unsigned int loading_views;
6310         if ((loading == FALSE && loading_views-- == 1) ||
6311             (loading == TRUE  && loading_views++ == 0))
6312                 nodelay(status_win, loading);
6315 static void
6316 init_display(void)
6318         const char *term;
6319         int x, y;
6321         /* Initialize the curses library */
6322         if (isatty(STDIN_FILENO)) {
6323                 cursed = !!initscr();
6324                 opt_tty = stdin;
6325         } else {
6326                 /* Leave stdin and stdout alone when acting as a pager. */
6327                 opt_tty = fopen("/dev/tty", "r+");
6328                 if (!opt_tty)
6329                         die("Failed to open /dev/tty");
6330                 cursed = !!newterm(NULL, opt_tty, opt_tty);
6331         }
6333         if (!cursed)
6334                 die("Failed to initialize curses");
6336         nonl();         /* Disable conversion and detect newlines from input. */
6337         cbreak();       /* Take input chars one at a time, no wait for \n */
6338         noecho();       /* Don't echo input */
6339         leaveok(stdscr, FALSE);
6341         if (has_colors())
6342                 init_colors();
6344         getmaxyx(stdscr, y, x);
6345         status_win = newwin(1, 0, y - 1, 0);
6346         if (!status_win)
6347                 die("Failed to create status window");
6349         /* Enable keyboard mapping */
6350         keypad(status_win, TRUE);
6351         wbkgdset(status_win, get_line_attr(LINE_STATUS));
6353         TABSIZE = opt_tab_size;
6354         if (opt_line_graphics) {
6355                 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
6356         }
6358         term = getenv("XTERM_VERSION") ? NULL : getenv("COLORTERM");
6359         if (term && !strcmp(term, "gnome-terminal")) {
6360                 /* In the gnome-terminal-emulator, the message from
6361                  * scrolling up one line when impossible followed by
6362                  * scrolling down one line causes corruption of the
6363                  * status line. This is fixed by calling wclear. */
6364                 use_scroll_status_wclear = TRUE;
6365                 use_scroll_redrawwin = FALSE;
6367         } else if (term && !strcmp(term, "xrvt-xpm")) {
6368                 /* No problems with full optimizations in xrvt-(unicode)
6369                  * and aterm. */
6370                 use_scroll_status_wclear = use_scroll_redrawwin = FALSE;
6372         } else {
6373                 /* When scrolling in (u)xterm the last line in the
6374                  * scrolling direction will update slowly. */
6375                 use_scroll_redrawwin = TRUE;
6376                 use_scroll_status_wclear = FALSE;
6377         }
6380 static int
6381 get_input(int prompt_position)
6383         struct view *view;
6384         int i, key, cursor_y, cursor_x;
6386         if (prompt_position)
6387                 input_mode = TRUE;
6389         while (TRUE) {
6390                 foreach_view (view, i) {
6391                         update_view(view);
6392                         if (view_is_displayed(view) && view->has_scrolled &&
6393                             use_scroll_redrawwin)
6394                                 redrawwin(view->win);
6395                         view->has_scrolled = FALSE;
6396                 }
6398                 /* Update the cursor position. */
6399                 if (prompt_position) {
6400                         getbegyx(status_win, cursor_y, cursor_x);
6401                         cursor_x = prompt_position;
6402                 } else {
6403                         view = display[current_view];
6404                         getbegyx(view->win, cursor_y, cursor_x);
6405                         cursor_x = view->width - 1;
6406                         cursor_y += view->lineno - view->offset;
6407                 }
6408                 setsyx(cursor_y, cursor_x);
6410                 /* Refresh, accept single keystroke of input */
6411                 doupdate();
6412                 key = wgetch(status_win);
6414                 /* wgetch() with nodelay() enabled returns ERR when
6415                  * there's no input. */
6416                 if (key == ERR) {
6418                 } else if (key == KEY_RESIZE) {
6419                         int height, width;
6421                         getmaxyx(stdscr, height, width);
6423                         wresize(status_win, 1, width);
6424                         mvwin(status_win, height - 1, 0);
6425                         wnoutrefresh(status_win);
6426                         resize_display();
6427                         redraw_display(TRUE);
6429                 } else {
6430                         input_mode = FALSE;
6431                         return key;
6432                 }
6433         }
6436 static char *
6437 prompt_input(const char *prompt, input_handler handler, void *data)
6439         enum input_status status = INPUT_OK;
6440         static char buf[SIZEOF_STR];
6441         size_t pos = 0;
6443         buf[pos] = 0;
6445         while (status == INPUT_OK || status == INPUT_SKIP) {
6446                 int key;
6448                 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
6449                 wclrtoeol(status_win);
6451                 key = get_input(pos + 1);
6452                 switch (key) {
6453                 case KEY_RETURN:
6454                 case KEY_ENTER:
6455                 case '\n':
6456                         status = pos ? INPUT_STOP : INPUT_CANCEL;
6457                         break;
6459                 case KEY_BACKSPACE:
6460                         if (pos > 0)
6461                                 buf[--pos] = 0;
6462                         else
6463                                 status = INPUT_CANCEL;
6464                         break;
6466                 case KEY_ESC:
6467                         status = INPUT_CANCEL;
6468                         break;
6470                 default:
6471                         if (pos >= sizeof(buf)) {
6472                                 report("Input string too long");
6473                                 return NULL;
6474                         }
6476                         status = handler(data, buf, key);
6477                         if (status == INPUT_OK)
6478                                 buf[pos++] = (char) key;
6479                 }
6480         }
6482         /* Clear the status window */
6483         status_empty = FALSE;
6484         report("");
6486         if (status == INPUT_CANCEL)
6487                 return NULL;
6489         buf[pos++] = 0;
6491         return buf;
6494 static enum input_status
6495 prompt_yesno_handler(void *data, char *buf, int c)
6497         if (c == 'y' || c == 'Y')
6498                 return INPUT_STOP;
6499         if (c == 'n' || c == 'N')
6500                 return INPUT_CANCEL;
6501         return INPUT_SKIP;
6504 static bool
6505 prompt_yesno(const char *prompt)
6507         char prompt2[SIZEOF_STR];
6509         if (!string_format(prompt2, "%s [Yy/Nn]", prompt))
6510                 return FALSE;
6512         return !!prompt_input(prompt2, prompt_yesno_handler, NULL);
6515 static enum input_status
6516 read_prompt_handler(void *data, char *buf, int c)
6518         return isprint(c) ? INPUT_OK : INPUT_SKIP;
6521 static char *
6522 read_prompt(const char *prompt)
6524         return prompt_input(prompt, read_prompt_handler, NULL);
6527 /*
6528  * Repository properties
6529  */
6531 static struct ref *refs = NULL;
6532 static size_t refs_alloc = 0;
6533 static size_t refs_size = 0;
6535 /* Id <-> ref store */
6536 static struct ref ***id_refs = NULL;
6537 static size_t id_refs_alloc = 0;
6538 static size_t id_refs_size = 0;
6540 static int
6541 compare_refs(const void *ref1_, const void *ref2_)
6543         const struct ref *ref1 = *(const struct ref **)ref1_;
6544         const struct ref *ref2 = *(const struct ref **)ref2_;
6546         if (ref1->tag != ref2->tag)
6547                 return ref2->tag - ref1->tag;
6548         if (ref1->ltag != ref2->ltag)
6549                 return ref2->ltag - ref2->ltag;
6550         if (ref1->head != ref2->head)
6551                 return ref2->head - ref1->head;
6552         if (ref1->tracked != ref2->tracked)
6553                 return ref2->tracked - ref1->tracked;
6554         if (ref1->remote != ref2->remote)
6555                 return ref2->remote - ref1->remote;
6556         return strcmp(ref1->name, ref2->name);
6559 static struct ref **
6560 get_refs(const char *id)
6562         struct ref ***tmp_id_refs;
6563         struct ref **ref_list = NULL;
6564         size_t ref_list_alloc = 0;
6565         size_t ref_list_size = 0;
6566         size_t i;
6568         for (i = 0; i < id_refs_size; i++)
6569                 if (!strcmp(id, id_refs[i][0]->id))
6570                         return id_refs[i];
6572         tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
6573                                     sizeof(*id_refs));
6574         if (!tmp_id_refs)
6575                 return NULL;
6577         id_refs = tmp_id_refs;
6579         for (i = 0; i < refs_size; i++) {
6580                 struct ref **tmp;
6582                 if (strcmp(id, refs[i].id))
6583                         continue;
6585                 tmp = realloc_items(ref_list, &ref_list_alloc,
6586                                     ref_list_size + 1, sizeof(*ref_list));
6587                 if (!tmp) {
6588                         if (ref_list)
6589                                 free(ref_list);
6590                         return NULL;
6591                 }
6593                 ref_list = tmp;
6594                 ref_list[ref_list_size] = &refs[i];
6595                 /* XXX: The properties of the commit chains ensures that we can
6596                  * safely modify the shared ref. The repo references will
6597                  * always be similar for the same id. */
6598                 ref_list[ref_list_size]->next = 1;
6600                 ref_list_size++;
6601         }
6603         if (ref_list) {
6604                 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
6605                 ref_list[ref_list_size - 1]->next = 0;
6606                 id_refs[id_refs_size++] = ref_list;
6607         }
6609         return ref_list;
6612 static int
6613 read_ref(char *id, size_t idlen, char *name, size_t namelen)
6615         struct ref *ref;
6616         bool tag = FALSE;
6617         bool ltag = FALSE;
6618         bool remote = FALSE;
6619         bool tracked = FALSE;
6620         bool check_replace = FALSE;
6621         bool head = FALSE;
6623         if (!prefixcmp(name, "refs/tags/")) {
6624                 if (!suffixcmp(name, namelen, "^{}")) {
6625                         namelen -= 3;
6626                         name[namelen] = 0;
6627                         if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
6628                                 check_replace = TRUE;
6629                 } else {
6630                         ltag = TRUE;
6631                 }
6633                 tag = TRUE;
6634                 namelen -= STRING_SIZE("refs/tags/");
6635                 name    += STRING_SIZE("refs/tags/");
6637         } else if (!prefixcmp(name, "refs/remotes/")) {
6638                 remote = TRUE;
6639                 namelen -= STRING_SIZE("refs/remotes/");
6640                 name    += STRING_SIZE("refs/remotes/");
6641                 tracked  = !strcmp(opt_remote, name);
6643         } else if (!prefixcmp(name, "refs/heads/")) {
6644                 namelen -= STRING_SIZE("refs/heads/");
6645                 name    += STRING_SIZE("refs/heads/");
6646                 head     = !strncmp(opt_head, name, namelen);
6648         } else if (!strcmp(name, "HEAD")) {
6649                 string_ncopy(opt_head_rev, id, idlen);
6650                 return OK;
6651         }
6653         if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
6654                 /* it's an annotated tag, replace the previous SHA1 with the
6655                  * resolved commit id; relies on the fact git-ls-remote lists
6656                  * the commit id of an annotated tag right before the commit id
6657                  * it points to. */
6658                 refs[refs_size - 1].ltag = ltag;
6659                 string_copy_rev(refs[refs_size - 1].id, id);
6661                 return OK;
6662         }
6663         refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
6664         if (!refs)
6665                 return ERR;
6667         ref = &refs[refs_size++];
6668         ref->name = malloc(namelen + 1);
6669         if (!ref->name)
6670                 return ERR;
6672         strncpy(ref->name, name, namelen);
6673         ref->name[namelen] = 0;
6674         ref->head = head;
6675         ref->tag = tag;
6676         ref->ltag = ltag;
6677         ref->remote = remote;
6678         ref->tracked = tracked;
6679         string_copy_rev(ref->id, id);
6681         return OK;
6684 static int
6685 load_refs(void)
6687         static const char *ls_remote_argv[SIZEOF_ARG] = {
6688                 "git", "ls-remote", opt_git_dir, NULL
6689         };
6690         static bool init = FALSE;
6692         if (!init) {
6693                 argv_from_env(ls_remote_argv, "TIG_LS_REMOTE");
6694                 init = TRUE;
6695         }
6697         if (!*opt_git_dir)
6698                 return OK;
6700         while (refs_size > 0)
6701                 free(refs[--refs_size].name);
6702         while (id_refs_size > 0)
6703                 free(id_refs[--id_refs_size]);
6705         return run_io_load(ls_remote_argv, "\t", read_ref);
6708 static void
6709 set_remote_branch(const char *name, const char *value, size_t valuelen)
6711         if (!strcmp(name, ".remote")) {
6712                 string_ncopy(opt_remote, value, valuelen);
6714         } else if (*opt_remote && !strcmp(name, ".merge")) {
6715                 size_t from = strlen(opt_remote);
6717                 if (!prefixcmp(value, "refs/heads/"))
6718                         value += STRING_SIZE("refs/heads/");
6720                 if (!string_format_from(opt_remote, &from, "/%s", value))
6721                         opt_remote[0] = 0;
6722         }
6725 static void
6726 set_repo_config_option(char *name, char *value, int (*cmd)(int, const char **))
6728         const char *argv[SIZEOF_ARG] = { name, "=" };
6729         int argc = 1 + (cmd == option_set_command);
6730         int error = ERR;
6732         if (!argv_from_string(argv, &argc, value))
6733                 config_msg = "Too many option arguments";
6734         else
6735                 error = cmd(argc, argv);
6737         if (error == ERR)
6738                 warn("Option 'tig.%s': %s", name, config_msg);
6741 static bool
6742 set_environment_variable(const char *name, const char *value)
6744         size_t len = strlen(name) + 1 + strlen(value) + 1;
6745         char *env = malloc(len);
6747         if (env &&
6748             string_nformat(env, len, NULL, "%s=%s", name, value) &&
6749             putenv(env) == 0)
6750                 return TRUE;
6751         free(env);
6752         return FALSE;
6755 static void
6756 set_work_tree(const char *value)
6758         char cwd[SIZEOF_STR];
6760         if (!getcwd(cwd, sizeof(cwd)))
6761                 die("Failed to get cwd path: %s", strerror(errno));
6762         if (chdir(opt_git_dir) < 0)
6763                 die("Failed to chdir(%s): %s", strerror(errno));
6764         if (!getcwd(opt_git_dir, sizeof(opt_git_dir)))
6765                 die("Failed to get git path: %s", strerror(errno));
6766         if (chdir(cwd) < 0)
6767                 die("Failed to chdir(%s): %s", cwd, strerror(errno));
6768         if (chdir(value) < 0)
6769                 die("Failed to chdir(%s): %s", value, strerror(errno));
6770         if (!getcwd(cwd, sizeof(cwd)))
6771                 die("Failed to get cwd path: %s", strerror(errno));
6772         if (!set_environment_variable("GIT_WORK_TREE", cwd))
6773                 die("Failed to set GIT_WORK_TREE to '%s'", cwd);
6774         if (!set_environment_variable("GIT_DIR", opt_git_dir))
6775                 die("Failed to set GIT_DIR to '%s'", opt_git_dir);
6776         opt_is_inside_work_tree = TRUE;
6779 static int
6780 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
6782         if (!strcmp(name, "i18n.commitencoding"))
6783                 string_ncopy(opt_encoding, value, valuelen);
6785         else if (!strcmp(name, "core.editor"))
6786                 string_ncopy(opt_editor, value, valuelen);
6788         else if (!strcmp(name, "core.worktree"))
6789                 set_work_tree(value);
6791         else if (!prefixcmp(name, "tig.color."))
6792                 set_repo_config_option(name + 10, value, option_color_command);
6794         else if (!prefixcmp(name, "tig.bind."))
6795                 set_repo_config_option(name + 9, value, option_bind_command);
6797         else if (!prefixcmp(name, "tig."))
6798                 set_repo_config_option(name + 4, value, option_set_command);
6800         else if (*opt_head && !prefixcmp(name, "branch.") &&
6801                  !strncmp(name + 7, opt_head, strlen(opt_head)))
6802                 set_remote_branch(name + 7 + strlen(opt_head), value, valuelen);
6804         return OK;
6807 static int
6808 load_git_config(void)
6810         const char *config_list_argv[] = { "git", GIT_CONFIG, "--list", NULL };
6812         return run_io_load(config_list_argv, "=", read_repo_config_option);
6815 static int
6816 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
6818         if (!opt_git_dir[0]) {
6819                 string_ncopy(opt_git_dir, name, namelen);
6821         } else if (opt_is_inside_work_tree == -1) {
6822                 /* This can be 3 different values depending on the
6823                  * version of git being used. If git-rev-parse does not
6824                  * understand --is-inside-work-tree it will simply echo
6825                  * the option else either "true" or "false" is printed.
6826                  * Default to true for the unknown case. */
6827                 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
6829         } else if (*name == '.') {
6830                 string_ncopy(opt_cdup, name, namelen);
6832         } else {
6833                 string_ncopy(opt_prefix, name, namelen);
6834         }
6836         return OK;
6839 static int
6840 load_repo_info(void)
6842         const char *head_argv[] = {
6843                 "git", "symbolic-ref", "HEAD", NULL
6844         };
6845         const char *rev_parse_argv[] = {
6846                 "git", "rev-parse", "--git-dir", "--is-inside-work-tree",
6847                         "--show-cdup", "--show-prefix", NULL
6848         };
6850         if (run_io_buf(head_argv, opt_head, sizeof(opt_head))) {
6851                 chomp_string(opt_head);
6852                 if (!prefixcmp(opt_head, "refs/heads/")) {
6853                         char *offset = opt_head + STRING_SIZE("refs/heads/");
6855                         memmove(opt_head, offset, strlen(offset) + 1);
6856                 }
6857         }
6859         return run_io_load(rev_parse_argv, "=", read_repo_info);
6863 /*
6864  * Main
6865  */
6867 static const char usage[] =
6868 "tig " TIG_VERSION " (" __DATE__ ")\n"
6869 "\n"
6870 "Usage: tig        [options] [revs] [--] [paths]\n"
6871 "   or: tig show   [options] [revs] [--] [paths]\n"
6872 "   or: tig blame  [rev] path\n"
6873 "   or: tig status\n"
6874 "   or: tig <      [git command output]\n"
6875 "\n"
6876 "Options:\n"
6877 "  -v, --version   Show version and exit\n"
6878 "  -h, --help      Show help message and exit";
6880 static void __NORETURN
6881 quit(int sig)
6883         /* XXX: Restore tty modes and let the OS cleanup the rest! */
6884         if (cursed)
6885                 endwin();
6886         exit(0);
6889 static void __NORETURN
6890 die(const char *err, ...)
6892         va_list args;
6894         endwin();
6896         va_start(args, err);
6897         fputs("tig: ", stderr);
6898         vfprintf(stderr, err, args);
6899         fputs("\n", stderr);
6900         va_end(args);
6902         exit(1);
6905 static void
6906 warn(const char *msg, ...)
6908         va_list args;
6910         va_start(args, msg);
6911         fputs("tig warning: ", stderr);
6912         vfprintf(stderr, msg, args);
6913         fputs("\n", stderr);
6914         va_end(args);
6917 static enum request
6918 parse_options(int argc, const char *argv[])
6920         enum request request = REQ_VIEW_MAIN;
6921         const char *subcommand;
6922         bool seen_dashdash = FALSE;
6923         /* XXX: This is vulnerable to the user overriding options
6924          * required for the main view parser. */
6925         const char *custom_argv[SIZEOF_ARG] = {
6926                 "git", "log", "--no-color", "--pretty=raw", "--parents",
6927                         "--topo-order", NULL
6928         };
6929         int i, j = 6;
6931         if (!isatty(STDIN_FILENO)) {
6932                 io_open(&VIEW(REQ_VIEW_PAGER)->io, "");
6933                 return REQ_VIEW_PAGER;
6934         }
6936         if (argc <= 1)
6937                 return REQ_NONE;
6939         subcommand = argv[1];
6940         if (!strcmp(subcommand, "status")) {
6941                 if (argc > 2)
6942                         warn("ignoring arguments after `%s'", subcommand);
6943                 return REQ_VIEW_STATUS;
6945         } else if (!strcmp(subcommand, "blame")) {
6946                 if (argc <= 2 || argc > 4)
6947                         die("invalid number of options to blame\n\n%s", usage);
6949                 i = 2;
6950                 if (argc == 4) {
6951                         string_ncopy(opt_ref, argv[i], strlen(argv[i]));
6952                         i++;
6953                 }
6955                 string_ncopy(opt_file, argv[i], strlen(argv[i]));
6956                 return REQ_VIEW_BLAME;
6958         } else if (!strcmp(subcommand, "show")) {
6959                 request = REQ_VIEW_DIFF;
6961         } else {
6962                 subcommand = NULL;
6963         }
6965         if (subcommand) {
6966                 custom_argv[1] = subcommand;
6967                 j = 2;
6968         }
6970         for (i = 1 + !!subcommand; i < argc; i++) {
6971                 const char *opt = argv[i];
6973                 if (seen_dashdash || !strcmp(opt, "--")) {
6974                         seen_dashdash = TRUE;
6976                 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
6977                         printf("tig version %s\n", TIG_VERSION);
6978                         quit(0);
6980                 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
6981                         printf("%s\n", usage);
6982                         quit(0);
6983                 }
6985                 custom_argv[j++] = opt;
6986                 if (j >= ARRAY_SIZE(custom_argv))
6987                         die("command too long");
6988         }
6990         if (!prepare_update(VIEW(request), custom_argv, NULL, FORMAT_NONE))                                                                        
6991                 die("Failed to format arguments"); 
6993         return request;
6996 int
6997 main(int argc, const char *argv[])
6999         enum request request = parse_options(argc, argv);
7000         struct view *view;
7001         size_t i;
7003         signal(SIGINT, quit);
7004         signal(SIGPIPE, SIG_IGN);
7006         if (setlocale(LC_ALL, "")) {
7007                 char *codeset = nl_langinfo(CODESET);
7009                 string_ncopy(opt_codeset, codeset, strlen(codeset));
7010         }
7012         if (load_repo_info() == ERR)
7013                 die("Failed to load repo info.");
7015         if (load_options() == ERR)
7016                 die("Failed to load user config.");
7018         if (load_git_config() == ERR)
7019                 die("Failed to load repo config.");
7021         /* Require a git repository unless when running in pager mode. */
7022         if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
7023                 die("Not a git repository");
7025         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
7026                 opt_utf8 = FALSE;
7028         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
7029                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
7030                 if (opt_iconv == ICONV_NONE)
7031                         die("Failed to initialize character set conversion");
7032         }
7034         if (load_refs() == ERR)
7035                 die("Failed to load refs.");
7037         foreach_view (view, i)
7038                 argv_from_env(view->ops->argv, view->cmd_env);
7040         init_display();
7042         if (request != REQ_NONE)
7043                 open_view(NULL, request, OPEN_PREPARED);
7044         request = request == REQ_NONE ? REQ_VIEW_MAIN : REQ_NONE;
7046         while (view_driver(display[current_view], request)) {
7047                 int key = get_input(0);
7049                 view = display[current_view];
7050                 request = get_keybinding(view->keymap, key);
7052                 /* Some low-level request handling. This keeps access to
7053                  * status_win restricted. */
7054                 switch (request) {
7055                 case REQ_PROMPT:
7056                 {
7057                         char *cmd = read_prompt(":");
7059                         if (cmd && isdigit(*cmd)) {
7060                                 int lineno = view->lineno + 1;
7062                                 if (parse_int(&lineno, cmd, 1, view->lines + 1) == OK) {
7063                                         select_view_line(view, lineno - 1);
7064                                         report("");
7065                                 } else {
7066                                         report("Unable to parse '%s' as a line number", cmd);
7067                                 }
7069                         } else if (cmd) {
7070                                 struct view *next = VIEW(REQ_VIEW_PAGER);
7071                                 const char *argv[SIZEOF_ARG] = { "git" };
7072                                 int argc = 1;
7074                                 /* When running random commands, initially show the
7075                                  * command in the title. However, it maybe later be
7076                                  * overwritten if a commit line is selected. */
7077                                 string_ncopy(next->ref, cmd, strlen(cmd));
7079                                 if (!argv_from_string(argv, &argc, cmd)) {
7080                                         report("Too many arguments");
7081                                 } else if (!prepare_update(next, argv, NULL, FORMAT_DASH)) {
7082                                         report("Failed to format command");
7083                                 } else {
7084                                         open_view(view, REQ_VIEW_PAGER, OPEN_PREPARED);
7085                                 }
7086                         }
7088                         request = REQ_NONE;
7089                         break;
7090                 }
7091                 case REQ_SEARCH:
7092                 case REQ_SEARCH_BACK:
7093                 {
7094                         const char *prompt = request == REQ_SEARCH ? "/" : "?";
7095                         char *search = read_prompt(prompt);
7097                         if (search)
7098                                 string_ncopy(opt_search, search, strlen(search));
7099                         else if (*opt_search)
7100                                 request = request == REQ_SEARCH ?
7101                                         REQ_FIND_NEXT :
7102                                         REQ_FIND_PREV;
7103                         else
7104                                 request = REQ_NONE;
7105                         break;
7106                 }
7107                 default:
7108                         break;
7109                 }
7110         }
7112         quit(0);
7114         return 0;