Code

Fix draw_lineno to always set the static format buffer
[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;
5124         if (!status_update_prepare(&io, line->type))
5125                 return FALSE;
5127         for (pos = line; pos < view->line + view->lines && pos->data; pos++)
5128                 files++;
5130         string_copy(buf, view->ref);
5131         for (file = 0, done = 5; result && file < files; line++, file++) {
5132                 int almost_done = file * 100 / files;
5134                 if (almost_done > done) {
5135                         done = almost_done;
5136                         string_format(view->ref, "updating file %u of %u (%d%% done)",
5137                                       file, files, done);
5138                         update_view_title(view);
5139                         doupdate();
5140                 }
5141                 result = status_update_write(&io, line->data, line->type);
5142         }
5143         string_copy(view->ref, buf);
5145         return done_io(&io) && result;
5148 static bool
5149 status_update(struct view *view)
5151         struct line *line = &view->line[view->lineno];
5153         assert(view->lines);
5155         if (!line->data) {
5156                 /* This should work even for the "On branch" line. */
5157                 if (line < view->line + view->lines && !line[1].data) {
5158                         report("Nothing to update");
5159                         return FALSE;
5160                 }
5162                 if (!status_update_files(view, line + 1)) {
5163                         report("Failed to update file status");
5164                         return FALSE;
5165                 }
5167         } else if (!status_update_file(line->data, line->type)) {
5168                 report("Failed to update file status");
5169                 return FALSE;
5170         }
5172         return TRUE;
5175 static bool
5176 status_revert(struct status *status, enum line_type type, bool has_none)
5178         if (!status || type != LINE_STAT_UNSTAGED) {
5179                 if (type == LINE_STAT_STAGED) {
5180                         report("Cannot revert changes to staged files");
5181                 } else if (type == LINE_STAT_UNTRACKED) {
5182                         report("Cannot revert changes to untracked files");
5183                 } else if (has_none) {
5184                         report("Nothing to revert");
5185                 } else {
5186                         report("Cannot revert changes to multiple files");
5187                 }
5188                 return FALSE;
5190         } else {
5191                 char mode[10] = "100644";
5192                 const char *reset_argv[] = {
5193                         "git", "update-index", "--cacheinfo", mode,
5194                                 status->old.rev, status->old.name, NULL
5195                 };
5196                 const char *checkout_argv[] = {
5197                         "git", "checkout", "--", status->old.name, NULL
5198                 };
5200                 if (!prompt_yesno("Are you sure you want to overwrite any changes?"))
5201                         return FALSE;
5202                 string_format(mode, "%o", status->old.mode);
5203                 return (status->status != 'U' || run_io_fg(reset_argv, opt_cdup)) &&
5204                         run_io_fg(checkout_argv, opt_cdup);
5205         }
5208 static enum request
5209 status_request(struct view *view, enum request request, struct line *line)
5211         struct status *status = line->data;
5213         switch (request) {
5214         case REQ_STATUS_UPDATE:
5215                 if (!status_update(view))
5216                         return REQ_NONE;
5217                 break;
5219         case REQ_STATUS_REVERT:
5220                 if (!status_revert(status, line->type, status_has_none(view, line)))
5221                         return REQ_NONE;
5222                 break;
5224         case REQ_STATUS_MERGE:
5225                 if (!status || status->status != 'U') {
5226                         report("Merging only possible for files with unmerged status ('U').");
5227                         return REQ_NONE;
5228                 }
5229                 open_mergetool(status->new.name);
5230                 break;
5232         case REQ_EDIT:
5233                 if (!status)
5234                         return request;
5235                 if (status->status == 'D') {
5236                         report("File has been deleted.");
5237                         return REQ_NONE;
5238                 }
5240                 open_editor(status->status != '?', status->new.name);
5241                 break;
5243         case REQ_VIEW_BLAME:
5244                 if (status) {
5245                         string_copy(opt_file, status->new.name);
5246                         opt_ref[0] = 0;
5247                 }
5248                 return request;
5250         case REQ_ENTER:
5251                 /* After returning the status view has been split to
5252                  * show the stage view. No further reloading is
5253                  * necessary. */
5254                 return status_enter(view, line);
5256         case REQ_REFRESH:
5257                 /* Simply reload the view. */
5258                 break;
5260         default:
5261                 return request;
5262         }
5264         open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
5266         return REQ_NONE;
5269 static void
5270 status_select(struct view *view, struct line *line)
5272         struct status *status = line->data;
5273         char file[SIZEOF_STR] = "all files";
5274         const char *text;
5275         const char *key;
5277         if (status && !string_format(file, "'%s'", status->new.name))
5278                 return;
5280         if (!status && line[1].type == LINE_STAT_NONE)
5281                 line++;
5283         switch (line->type) {
5284         case LINE_STAT_STAGED:
5285                 text = "Press %s to unstage %s for commit";
5286                 break;
5288         case LINE_STAT_UNSTAGED:
5289                 text = "Press %s to stage %s for commit";
5290                 break;
5292         case LINE_STAT_UNTRACKED:
5293                 text = "Press %s to stage %s for addition";
5294                 break;
5296         case LINE_STAT_HEAD:
5297         case LINE_STAT_NONE:
5298                 text = "Nothing to update";
5299                 break;
5301         default:
5302                 die("line type %d not handled in switch", line->type);
5303         }
5305         if (status && status->status == 'U') {
5306                 text = "Press %s to resolve conflict in %s";
5307                 key = get_key(REQ_STATUS_MERGE);
5309         } else {
5310                 key = get_key(REQ_STATUS_UPDATE);
5311         }
5313         string_format(view->ref, text, key, file);
5316 static bool
5317 status_grep(struct view *view, struct line *line)
5319         struct status *status = line->data;
5320         enum { S_STATUS, S_NAME, S_END } state;
5321         char buf[2] = "?";
5322         regmatch_t pmatch;
5324         if (!status)
5325                 return FALSE;
5327         for (state = S_STATUS; state < S_END; state++) {
5328                 const char *text;
5330                 switch (state) {
5331                 case S_NAME:    text = status->new.name;        break;
5332                 case S_STATUS:
5333                         buf[0] = status->status;
5334                         text = buf;
5335                         break;
5337                 default:
5338                         return FALSE;
5339                 }
5341                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5342                         return TRUE;
5343         }
5345         return FALSE;
5348 static struct view_ops status_ops = {
5349         "file",
5350         NULL,
5351         status_open,
5352         NULL,
5353         status_draw,
5354         status_request,
5355         status_grep,
5356         status_select,
5357 };
5360 static bool
5361 stage_diff_write(struct io *io, struct line *line, struct line *end)
5363         while (line < end) {
5364                 if (!io_write(io, line->data, strlen(line->data)) ||
5365                     !io_write(io, "\n", 1))
5366                         return FALSE;
5367                 line++;
5368                 if (line->type == LINE_DIFF_CHUNK ||
5369                     line->type == LINE_DIFF_HEADER)
5370                         break;
5371         }
5373         return TRUE;
5376 static struct line *
5377 stage_diff_find(struct view *view, struct line *line, enum line_type type)
5379         for (; view->line < line; line--)
5380                 if (line->type == type)
5381                         return line;
5383         return NULL;
5386 static bool
5387 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
5389         const char *apply_argv[SIZEOF_ARG] = {
5390                 "git", "apply", "--whitespace=nowarn", NULL
5391         };
5392         struct line *diff_hdr;
5393         struct io io = {};
5394         int argc = 3;
5396         diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
5397         if (!diff_hdr)
5398                 return FALSE;
5400         if (!revert)
5401                 apply_argv[argc++] = "--cached";
5402         if (revert || stage_line_type == LINE_STAT_STAGED)
5403                 apply_argv[argc++] = "-R";
5404         apply_argv[argc++] = "-";
5405         apply_argv[argc++] = NULL;
5406         if (!run_io(&io, apply_argv, opt_cdup, IO_WR))
5407                 return FALSE;
5409         if (!stage_diff_write(&io, diff_hdr, chunk) ||
5410             !stage_diff_write(&io, chunk, view->line + view->lines))
5411                 chunk = NULL;
5413         done_io(&io);
5414         run_io_bg(update_index_argv);
5416         return chunk ? TRUE : FALSE;
5419 static bool
5420 stage_update(struct view *view, struct line *line)
5422         struct line *chunk = NULL;
5424         if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
5425                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5427         if (chunk) {
5428                 if (!stage_apply_chunk(view, chunk, FALSE)) {
5429                         report("Failed to apply chunk");
5430                         return FALSE;
5431                 }
5433         } else if (!stage_status.status) {
5434                 view = VIEW(REQ_VIEW_STATUS);
5436                 for (line = view->line; line < view->line + view->lines; line++)
5437                         if (line->type == stage_line_type)
5438                                 break;
5440                 if (!status_update_files(view, line + 1)) {
5441                         report("Failed to update files");
5442                         return FALSE;
5443                 }
5445         } else if (!status_update_file(&stage_status, stage_line_type)) {
5446                 report("Failed to update file");
5447                 return FALSE;
5448         }
5450         return TRUE;
5453 static bool
5454 stage_revert(struct view *view, struct line *line)
5456         struct line *chunk = NULL;
5458         if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
5459                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5461         if (chunk) {
5462                 if (!prompt_yesno("Are you sure you want to revert changes?"))
5463                         return FALSE;
5465                 if (!stage_apply_chunk(view, chunk, TRUE)) {
5466                         report("Failed to revert chunk");
5467                         return FALSE;
5468                 }
5469                 return TRUE;
5471         } else {
5472                 return status_revert(stage_status.status ? &stage_status : NULL,
5473                                      stage_line_type, FALSE);
5474         }
5478 static void
5479 stage_next(struct view *view, struct line *line)
5481         int i;
5483         if (!stage_chunks) {
5484                 static size_t alloc = 0;
5485                 int *tmp;
5487                 for (line = view->line; line < view->line + view->lines; line++) {
5488                         if (line->type != LINE_DIFF_CHUNK)
5489                                 continue;
5491                         tmp = realloc_items(stage_chunk, &alloc,
5492                                             stage_chunks, sizeof(*tmp));
5493                         if (!tmp) {
5494                                 report("Allocation failure");
5495                                 return;
5496                         }
5498                         stage_chunk = tmp;
5499                         stage_chunk[stage_chunks++] = line - view->line;
5500                 }
5501         }
5503         for (i = 0; i < stage_chunks; i++) {
5504                 if (stage_chunk[i] > view->lineno) {
5505                         do_scroll_view(view, stage_chunk[i] - view->lineno);
5506                         report("Chunk %d of %d", i + 1, stage_chunks);
5507                         return;
5508                 }
5509         }
5511         report("No next chunk found");
5514 static enum request
5515 stage_request(struct view *view, enum request request, struct line *line)
5517         switch (request) {
5518         case REQ_STATUS_UPDATE:
5519                 if (!stage_update(view, line))
5520                         return REQ_NONE;
5521                 break;
5523         case REQ_STATUS_REVERT:
5524                 if (!stage_revert(view, line))
5525                         return REQ_NONE;
5526                 break;
5528         case REQ_STAGE_NEXT:
5529                 if (stage_line_type == LINE_STAT_UNTRACKED) {
5530                         report("File is untracked; press %s to add",
5531                                get_key(REQ_STATUS_UPDATE));
5532                         return REQ_NONE;
5533                 }
5534                 stage_next(view, line);
5535                 return REQ_NONE;
5537         case REQ_EDIT:
5538                 if (!stage_status.new.name[0])
5539                         return request;
5540                 if (stage_status.status == 'D') {
5541                         report("File has been deleted.");
5542                         return REQ_NONE;
5543                 }
5545                 open_editor(stage_status.status != '?', stage_status.new.name);
5546                 break;
5548         case REQ_REFRESH:
5549                 /* Reload everything ... */
5550                 break;
5552         case REQ_VIEW_BLAME:
5553                 if (stage_status.new.name[0]) {
5554                         string_copy(opt_file, stage_status.new.name);
5555                         opt_ref[0] = 0;
5556                 }
5557                 return request;
5559         case REQ_ENTER:
5560                 return pager_request(view, request, line);
5562         default:
5563                 return request;
5564         }
5566         VIEW(REQ_VIEW_STATUS)->p_restore = TRUE;
5567         open_view(view, REQ_VIEW_STATUS, OPEN_REFRESH);
5569         /* Check whether the staged entry still exists, and close the
5570          * stage view if it doesn't. */
5571         if (!status_exists(&stage_status, stage_line_type)) {
5572                 status_restore(VIEW(REQ_VIEW_STATUS));
5573                 return REQ_VIEW_CLOSE;
5574         }
5576         if (stage_line_type == LINE_STAT_UNTRACKED) {
5577                 if (!suffixcmp(stage_status.new.name, -1, "/")) {
5578                         report("Cannot display a directory");
5579                         return REQ_NONE;
5580                 }
5582                 if (!prepare_update_file(view, stage_status.new.name)) {
5583                         report("Failed to open file: %s", strerror(errno));
5584                         return REQ_NONE;
5585                 }
5586         }
5587         open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
5589         return REQ_NONE;
5592 static struct view_ops stage_ops = {
5593         "line",
5594         NULL,
5595         NULL,
5596         pager_read,
5597         pager_draw,
5598         stage_request,
5599         pager_grep,
5600         pager_select,
5601 };
5604 /*
5605  * Revision graph
5606  */
5608 struct commit {
5609         char id[SIZEOF_REV];            /* SHA1 ID. */
5610         char title[128];                /* First line of the commit message. */
5611         char author[75];                /* Author of the commit. */
5612         struct tm time;                 /* Date from the author ident. */
5613         struct ref **refs;              /* Repository references. */
5614         chtype graph[SIZEOF_REVGRAPH];  /* Ancestry chain graphics. */
5615         size_t graph_size;              /* The width of the graph array. */
5616         bool has_parents;               /* Rewritten --parents seen. */
5617 };
5619 /* Size of rev graph with no  "padding" columns */
5620 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
5622 struct rev_graph {
5623         struct rev_graph *prev, *next, *parents;
5624         char rev[SIZEOF_REVITEMS][SIZEOF_REV];
5625         size_t size;
5626         struct commit *commit;
5627         size_t pos;
5628         unsigned int boundary:1;
5629 };
5631 /* Parents of the commit being visualized. */
5632 static struct rev_graph graph_parents[4];
5634 /* The current stack of revisions on the graph. */
5635 static struct rev_graph graph_stacks[4] = {
5636         { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
5637         { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
5638         { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
5639         { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
5640 };
5642 static inline bool
5643 graph_parent_is_merge(struct rev_graph *graph)
5645         return graph->parents->size > 1;
5648 static inline void
5649 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
5651         struct commit *commit = graph->commit;
5653         if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
5654                 commit->graph[commit->graph_size++] = symbol;
5657 static void
5658 clear_rev_graph(struct rev_graph *graph)
5660         graph->boundary = 0;
5661         graph->size = graph->pos = 0;
5662         graph->commit = NULL;
5663         memset(graph->parents, 0, sizeof(*graph->parents));
5666 static void
5667 done_rev_graph(struct rev_graph *graph)
5669         if (graph_parent_is_merge(graph) &&
5670             graph->pos < graph->size - 1 &&
5671             graph->next->size == graph->size + graph->parents->size - 1) {
5672                 size_t i = graph->pos + graph->parents->size - 1;
5674                 graph->commit->graph_size = i * 2;
5675                 while (i < graph->next->size - 1) {
5676                         append_to_rev_graph(graph, ' ');
5677                         append_to_rev_graph(graph, '\\');
5678                         i++;
5679                 }
5680         }
5682         clear_rev_graph(graph);
5685 static void
5686 push_rev_graph(struct rev_graph *graph, const char *parent)
5688         int i;
5690         /* "Collapse" duplicate parents lines.
5691          *
5692          * FIXME: This needs to also update update the drawn graph but
5693          * for now it just serves as a method for pruning graph lines. */
5694         for (i = 0; i < graph->size; i++)
5695                 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
5696                         return;
5698         if (graph->size < SIZEOF_REVITEMS) {
5699                 string_copy_rev(graph->rev[graph->size++], parent);
5700         }
5703 static chtype
5704 get_rev_graph_symbol(struct rev_graph *graph)
5706         chtype symbol;
5708         if (graph->boundary)
5709                 symbol = REVGRAPH_BOUND;
5710         else if (graph->parents->size == 0)
5711                 symbol = REVGRAPH_INIT;
5712         else if (graph_parent_is_merge(graph))
5713                 symbol = REVGRAPH_MERGE;
5714         else if (graph->pos >= graph->size)
5715                 symbol = REVGRAPH_BRANCH;
5716         else
5717                 symbol = REVGRAPH_COMMIT;
5719         return symbol;
5722 static void
5723 draw_rev_graph(struct rev_graph *graph)
5725         struct rev_filler {
5726                 chtype separator, line;
5727         };
5728         enum { DEFAULT, RSHARP, RDIAG, LDIAG };
5729         static struct rev_filler fillers[] = {
5730                 { ' ',  '|' },
5731                 { '`',  '.' },
5732                 { '\'', ' ' },
5733                 { '/',  ' ' },
5734         };
5735         chtype symbol = get_rev_graph_symbol(graph);
5736         struct rev_filler *filler;
5737         size_t i;
5739         if (opt_line_graphics)
5740                 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
5742         filler = &fillers[DEFAULT];
5744         for (i = 0; i < graph->pos; i++) {
5745                 append_to_rev_graph(graph, filler->line);
5746                 if (graph_parent_is_merge(graph->prev) &&
5747                     graph->prev->pos == i)
5748                         filler = &fillers[RSHARP];
5750                 append_to_rev_graph(graph, filler->separator);
5751         }
5753         /* Place the symbol for this revision. */
5754         append_to_rev_graph(graph, symbol);
5756         if (graph->prev->size > graph->size)
5757                 filler = &fillers[RDIAG];
5758         else
5759                 filler = &fillers[DEFAULT];
5761         i++;
5763         for (; i < graph->size; i++) {
5764                 append_to_rev_graph(graph, filler->separator);
5765                 append_to_rev_graph(graph, filler->line);
5766                 if (graph_parent_is_merge(graph->prev) &&
5767                     i < graph->prev->pos + graph->parents->size)
5768                         filler = &fillers[RSHARP];
5769                 if (graph->prev->size > graph->size)
5770                         filler = &fillers[LDIAG];
5771         }
5773         if (graph->prev->size > graph->size) {
5774                 append_to_rev_graph(graph, filler->separator);
5775                 if (filler->line != ' ')
5776                         append_to_rev_graph(graph, filler->line);
5777         }
5780 /* Prepare the next rev graph */
5781 static void
5782 prepare_rev_graph(struct rev_graph *graph)
5784         size_t i;
5786         /* First, traverse all lines of revisions up to the active one. */
5787         for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5788                 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5789                         break;
5791                 push_rev_graph(graph->next, graph->rev[graph->pos]);
5792         }
5794         /* Interleave the new revision parent(s). */
5795         for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5796                 push_rev_graph(graph->next, graph->parents->rev[i]);
5798         /* Lastly, put any remaining revisions. */
5799         for (i = graph->pos + 1; i < graph->size; i++)
5800                 push_rev_graph(graph->next, graph->rev[i]);
5803 static void
5804 update_rev_graph(struct view *view, struct rev_graph *graph)
5806         /* If this is the finalizing update ... */
5807         if (graph->commit)
5808                 prepare_rev_graph(graph);
5810         /* Graph visualization needs a one rev look-ahead,
5811          * so the first update doesn't visualize anything. */
5812         if (!graph->prev->commit)
5813                 return;
5815         if (view->lines > 2)
5816                 view->line[view->lines - 3].dirty = 1;
5817         if (view->lines > 1)
5818                 view->line[view->lines - 2].dirty = 1;
5819         draw_rev_graph(graph->prev);
5820         done_rev_graph(graph->prev->prev);
5824 /*
5825  * Main view backend
5826  */
5828 static const char *main_argv[SIZEOF_ARG] = {
5829         "git", "log", "--no-color", "--pretty=raw", "--parents",
5830                       "--topo-order", "%(head)", NULL
5831 };
5833 static bool
5834 main_draw(struct view *view, struct line *line, unsigned int lineno)
5836         struct commit *commit = line->data;
5838         if (!*commit->author)
5839                 return FALSE;
5841         if (opt_date && draw_date(view, &commit->time))
5842                 return TRUE;
5844         if (opt_author && draw_author(view, commit->author))
5845                 return TRUE;
5847         if (opt_rev_graph && commit->graph_size &&
5848             draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5849                 return TRUE;
5851         if (opt_show_refs && commit->refs) {
5852                 size_t i = 0;
5854                 do {
5855                         enum line_type type;
5857                         if (commit->refs[i]->head)
5858                                 type = LINE_MAIN_HEAD;
5859                         else if (commit->refs[i]->ltag)
5860                                 type = LINE_MAIN_LOCAL_TAG;
5861                         else if (commit->refs[i]->tag)
5862                                 type = LINE_MAIN_TAG;
5863                         else if (commit->refs[i]->tracked)
5864                                 type = LINE_MAIN_TRACKED;
5865                         else if (commit->refs[i]->remote)
5866                                 type = LINE_MAIN_REMOTE;
5867                         else
5868                                 type = LINE_MAIN_REF;
5870                         if (draw_text(view, type, "[", TRUE) ||
5871                             draw_text(view, type, commit->refs[i]->name, TRUE) ||
5872                             draw_text(view, type, "]", TRUE))
5873                                 return TRUE;
5875                         if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5876                                 return TRUE;
5877                 } while (commit->refs[i++]->next);
5878         }
5880         draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5881         return TRUE;
5884 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5885 static bool
5886 main_read(struct view *view, char *line)
5888         static struct rev_graph *graph = graph_stacks;
5889         enum line_type type;
5890         struct commit *commit;
5892         if (!line) {
5893                 int i;
5895                 if (!view->lines && !view->parent)
5896                         die("No revisions match the given arguments.");
5897                 if (view->lines > 0) {
5898                         commit = view->line[view->lines - 1].data;
5899                         view->line[view->lines - 1].dirty = 1;
5900                         if (!*commit->author) {
5901                                 view->lines--;
5902                                 free(commit);
5903                                 graph->commit = NULL;
5904                         }
5905                 }
5906                 update_rev_graph(view, graph);
5908                 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5909                         clear_rev_graph(&graph_stacks[i]);
5910                 return TRUE;
5911         }
5913         type = get_line_type(line);
5914         if (type == LINE_COMMIT) {
5915                 commit = calloc(1, sizeof(struct commit));
5916                 if (!commit)
5917                         return FALSE;
5919                 line += STRING_SIZE("commit ");
5920                 if (*line == '-') {
5921                         graph->boundary = 1;
5922                         line++;
5923                 }
5925                 string_copy_rev(commit->id, line);
5926                 commit->refs = get_refs(commit->id);
5927                 graph->commit = commit;
5928                 add_line_data(view, commit, LINE_MAIN_COMMIT);
5930                 while ((line = strchr(line, ' '))) {
5931                         line++;
5932                         push_rev_graph(graph->parents, line);
5933                         commit->has_parents = TRUE;
5934                 }
5935                 return TRUE;
5936         }
5938         if (!view->lines)
5939                 return TRUE;
5940         commit = view->line[view->lines - 1].data;
5942         switch (type) {
5943         case LINE_PARENT:
5944                 if (commit->has_parents)
5945                         break;
5946                 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5947                 break;
5949         case LINE_AUTHOR:
5950                 parse_author_line(line + STRING_SIZE("author "),
5951                                   commit->author, sizeof(commit->author),
5952                                   &commit->time);
5953                 update_rev_graph(view, graph);
5954                 graph = graph->next;
5955                 break;
5957         default:
5958                 /* Fill in the commit title if it has not already been set. */
5959                 if (commit->title[0])
5960                         break;
5962                 /* Require titles to start with a non-space character at the
5963                  * offset used by git log. */
5964                 if (strncmp(line, "    ", 4))
5965                         break;
5966                 line += 4;
5967                 /* Well, if the title starts with a whitespace character,
5968                  * try to be forgiving.  Otherwise we end up with no title. */
5969                 while (isspace(*line))
5970                         line++;
5971                 if (*line == '\0')
5972                         break;
5973                 /* FIXME: More graceful handling of titles; append "..." to
5974                  * shortened titles, etc. */
5976                 string_expand(commit->title, sizeof(commit->title), line, 1);
5977                 view->line[view->lines - 1].dirty = 1;
5978         }
5980         return TRUE;
5983 static enum request
5984 main_request(struct view *view, enum request request, struct line *line)
5986         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5988         switch (request) {
5989         case REQ_ENTER:
5990                 open_view(view, REQ_VIEW_DIFF, flags);
5991                 break;
5992         case REQ_REFRESH:
5993                 load_refs();
5994                 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
5995                 break;
5996         default:
5997                 return request;
5998         }
6000         return REQ_NONE;
6003 static bool
6004 grep_refs(struct ref **refs, regex_t *regex)
6006         regmatch_t pmatch;
6007         size_t i = 0;
6009         if (!refs)
6010                 return FALSE;
6011         do {
6012                 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
6013                         return TRUE;
6014         } while (refs[i++]->next);
6016         return FALSE;
6019 static bool
6020 main_grep(struct view *view, struct line *line)
6022         struct commit *commit = line->data;
6023         enum { S_TITLE, S_AUTHOR, S_DATE, S_REFS, S_END } state;
6024         char buf[DATE_COLS + 1];
6025         regmatch_t pmatch;
6027         for (state = S_TITLE; state < S_END; state++) {
6028                 char *text;
6030                 switch (state) {
6031                 case S_TITLE:   text = commit->title;   break;
6032                 case S_AUTHOR:
6033                         if (!opt_author)
6034                                 continue;
6035                         text = commit->author;
6036                         break;
6037                 case S_DATE:
6038                         if (!opt_date)
6039                                 continue;
6040                         if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
6041                                 continue;
6042                         text = buf;
6043                         break;
6044                 case S_REFS:
6045                         if (!opt_show_refs)
6046                                 continue;
6047                         if (grep_refs(commit->refs, view->regex) == TRUE)
6048                                 return TRUE;
6049                         continue;
6050                 default:
6051                         return FALSE;
6052                 }
6054                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
6055                         return TRUE;
6056         }
6058         return FALSE;
6061 static void
6062 main_select(struct view *view, struct line *line)
6064         struct commit *commit = line->data;
6066         string_copy_rev(view->ref, commit->id);
6067         string_copy_rev(ref_commit, view->ref);
6070 static struct view_ops main_ops = {
6071         "commit",
6072         main_argv,
6073         NULL,
6074         main_read,
6075         main_draw,
6076         main_request,
6077         main_grep,
6078         main_select,
6079 };
6082 /*
6083  * Unicode / UTF-8 handling
6084  *
6085  * NOTE: Much of the following code for dealing with Unicode is derived from
6086  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
6087  * src/intl/charset.c from the UTF-8 branch commit elinks-0.11.0-g31f2c28.
6088  */
6090 static inline int
6091 unicode_width(unsigned long c)
6093         if (c >= 0x1100 &&
6094            (c <= 0x115f                         /* Hangul Jamo */
6095             || c == 0x2329
6096             || c == 0x232a
6097             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
6098                                                 /* CJK ... Yi */
6099             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
6100             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
6101             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
6102             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
6103             || (c >= 0xffe0  && c <= 0xffe6)
6104             || (c >= 0x20000 && c <= 0x2fffd)
6105             || (c >= 0x30000 && c <= 0x3fffd)))
6106                 return 2;
6108         if (c == '\t')
6109                 return opt_tab_size;
6111         return 1;
6114 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
6115  * Illegal bytes are set one. */
6116 static const unsigned char utf8_bytes[256] = {
6117         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,
6118         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,
6119         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,
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         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,
6124         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,
6125 };
6127 /* Decode UTF-8 multi-byte representation into a Unicode character. */
6128 static inline unsigned long
6129 utf8_to_unicode(const char *string, size_t length)
6131         unsigned long unicode;
6133         switch (length) {
6134         case 1:
6135                 unicode  =   string[0];
6136                 break;
6137         case 2:
6138                 unicode  =  (string[0] & 0x1f) << 6;
6139                 unicode +=  (string[1] & 0x3f);
6140                 break;
6141         case 3:
6142                 unicode  =  (string[0] & 0x0f) << 12;
6143                 unicode += ((string[1] & 0x3f) << 6);
6144                 unicode +=  (string[2] & 0x3f);
6145                 break;
6146         case 4:
6147                 unicode  =  (string[0] & 0x0f) << 18;
6148                 unicode += ((string[1] & 0x3f) << 12);
6149                 unicode += ((string[2] & 0x3f) << 6);
6150                 unicode +=  (string[3] & 0x3f);
6151                 break;
6152         case 5:
6153                 unicode  =  (string[0] & 0x0f) << 24;
6154                 unicode += ((string[1] & 0x3f) << 18);
6155                 unicode += ((string[2] & 0x3f) << 12);
6156                 unicode += ((string[3] & 0x3f) << 6);
6157                 unicode +=  (string[4] & 0x3f);
6158                 break;
6159         case 6:
6160                 unicode  =  (string[0] & 0x01) << 30;
6161                 unicode += ((string[1] & 0x3f) << 24);
6162                 unicode += ((string[2] & 0x3f) << 18);
6163                 unicode += ((string[3] & 0x3f) << 12);
6164                 unicode += ((string[4] & 0x3f) << 6);
6165                 unicode +=  (string[5] & 0x3f);
6166                 break;
6167         default:
6168                 die("Invalid Unicode length");
6169         }
6171         /* Invalid characters could return the special 0xfffd value but NUL
6172          * should be just as good. */
6173         return unicode > 0xffff ? 0 : unicode;
6176 /* Calculates how much of string can be shown within the given maximum width
6177  * and sets trimmed parameter to non-zero value if all of string could not be
6178  * shown. If the reserve flag is TRUE, it will reserve at least one
6179  * trailing character, which can be useful when drawing a delimiter.
6180  *
6181  * Returns the number of bytes to output from string to satisfy max_width. */
6182 static size_t
6183 utf8_length(const char **start, size_t skip, int *width, size_t max_width, int *trimmed, bool reserve)
6185         const char *string = *start;
6186         const char *end = strchr(string, '\0');
6187         unsigned char last_bytes = 0;
6188         size_t last_ucwidth = 0;
6190         *width = 0;
6191         *trimmed = 0;
6193         while (string < end) {
6194                 int c = *(unsigned char *) string;
6195                 unsigned char bytes = utf8_bytes[c];
6196                 size_t ucwidth;
6197                 unsigned long unicode;
6199                 if (string + bytes > end)
6200                         break;
6202                 /* Change representation to figure out whether
6203                  * it is a single- or double-width character. */
6205                 unicode = utf8_to_unicode(string, bytes);
6206                 /* FIXME: Graceful handling of invalid Unicode character. */
6207                 if (!unicode)
6208                         break;
6210                 ucwidth = unicode_width(unicode);
6211                 if (skip > 0) {
6212                         skip -= ucwidth <= skip ? ucwidth : skip;
6213                         *start += bytes;
6214                 }
6215                 *width  += ucwidth;
6216                 if (*width > max_width) {
6217                         *trimmed = 1;
6218                         *width -= ucwidth;
6219                         if (reserve && *width == max_width) {
6220                                 string -= last_bytes;
6221                                 *width -= last_ucwidth;
6222                         }
6223                         break;
6224                 }
6226                 string  += bytes;
6227                 last_bytes = ucwidth ? bytes : 0;
6228                 last_ucwidth = ucwidth;
6229         }
6231         return string - *start;
6235 /*
6236  * Status management
6237  */
6239 /* Whether or not the curses interface has been initialized. */
6240 static bool cursed = FALSE;
6242 /* Terminal hacks and workarounds. */
6243 static bool use_scroll_redrawwin;
6244 static bool use_scroll_status_wclear;
6246 /* The status window is used for polling keystrokes. */
6247 static WINDOW *status_win;
6249 /* Reading from the prompt? */
6250 static bool input_mode = FALSE;
6252 static bool status_empty = FALSE;
6254 /* Update status and title window. */
6255 static void
6256 report(const char *msg, ...)
6258         struct view *view = display[current_view];
6260         if (input_mode)
6261                 return;
6263         if (!view) {
6264                 char buf[SIZEOF_STR];
6265                 va_list args;
6267                 va_start(args, msg);
6268                 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
6269                         buf[sizeof(buf) - 1] = 0;
6270                         buf[sizeof(buf) - 2] = '.';
6271                         buf[sizeof(buf) - 3] = '.';
6272                         buf[sizeof(buf) - 4] = '.';
6273                 }
6274                 va_end(args);
6275                 die("%s", buf);
6276         }
6278         if (!status_empty || *msg) {
6279                 va_list args;
6281                 va_start(args, msg);
6283                 wmove(status_win, 0, 0);
6284                 if (view->has_scrolled && use_scroll_status_wclear)
6285                         wclear(status_win);
6286                 if (*msg) {
6287                         vwprintw(status_win, msg, args);
6288                         status_empty = FALSE;
6289                 } else {
6290                         status_empty = TRUE;
6291                 }
6292                 wclrtoeol(status_win);
6293                 wnoutrefresh(status_win);
6295                 va_end(args);
6296         }
6298         update_view_title(view);
6301 /* Controls when nodelay should be in effect when polling user input. */
6302 static void
6303 set_nonblocking_input(bool loading)
6305         static unsigned int loading_views;
6307         if ((loading == FALSE && loading_views-- == 1) ||
6308             (loading == TRUE  && loading_views++ == 0))
6309                 nodelay(status_win, loading);
6312 static void
6313 init_display(void)
6315         const char *term;
6316         int x, y;
6318         /* Initialize the curses library */
6319         if (isatty(STDIN_FILENO)) {
6320                 cursed = !!initscr();
6321                 opt_tty = stdin;
6322         } else {
6323                 /* Leave stdin and stdout alone when acting as a pager. */
6324                 opt_tty = fopen("/dev/tty", "r+");
6325                 if (!opt_tty)
6326                         die("Failed to open /dev/tty");
6327                 cursed = !!newterm(NULL, opt_tty, opt_tty);
6328         }
6330         if (!cursed)
6331                 die("Failed to initialize curses");
6333         nonl();         /* Disable conversion and detect newlines from input. */
6334         cbreak();       /* Take input chars one at a time, no wait for \n */
6335         noecho();       /* Don't echo input */
6336         leaveok(stdscr, FALSE);
6338         if (has_colors())
6339                 init_colors();
6341         getmaxyx(stdscr, y, x);
6342         status_win = newwin(1, 0, y - 1, 0);
6343         if (!status_win)
6344                 die("Failed to create status window");
6346         /* Enable keyboard mapping */
6347         keypad(status_win, TRUE);
6348         wbkgdset(status_win, get_line_attr(LINE_STATUS));
6350         TABSIZE = opt_tab_size;
6351         if (opt_line_graphics) {
6352                 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
6353         }
6355         term = getenv("XTERM_VERSION") ? NULL : getenv("COLORTERM");
6356         if (term && !strcmp(term, "gnome-terminal")) {
6357                 /* In the gnome-terminal-emulator, the message from
6358                  * scrolling up one line when impossible followed by
6359                  * scrolling down one line causes corruption of the
6360                  * status line. This is fixed by calling wclear. */
6361                 use_scroll_status_wclear = TRUE;
6362                 use_scroll_redrawwin = FALSE;
6364         } else if (term && !strcmp(term, "xrvt-xpm")) {
6365                 /* No problems with full optimizations in xrvt-(unicode)
6366                  * and aterm. */
6367                 use_scroll_status_wclear = use_scroll_redrawwin = FALSE;
6369         } else {
6370                 /* When scrolling in (u)xterm the last line in the
6371                  * scrolling direction will update slowly. */
6372                 use_scroll_redrawwin = TRUE;
6373                 use_scroll_status_wclear = FALSE;
6374         }
6377 static int
6378 get_input(int prompt_position)
6380         struct view *view;
6381         int i, key, cursor_y, cursor_x;
6383         if (prompt_position)
6384                 input_mode = TRUE;
6386         while (TRUE) {
6387                 foreach_view (view, i) {
6388                         update_view(view);
6389                         if (view_is_displayed(view) && view->has_scrolled &&
6390                             use_scroll_redrawwin)
6391                                 redrawwin(view->win);
6392                         view->has_scrolled = FALSE;
6393                 }
6395                 /* Update the cursor position. */
6396                 if (prompt_position) {
6397                         getbegyx(status_win, cursor_y, cursor_x);
6398                         cursor_x = prompt_position;
6399                 } else {
6400                         view = display[current_view];
6401                         getbegyx(view->win, cursor_y, cursor_x);
6402                         cursor_x = view->width - 1;
6403                         cursor_y += view->lineno - view->offset;
6404                 }
6405                 setsyx(cursor_y, cursor_x);
6407                 /* Refresh, accept single keystroke of input */
6408                 doupdate();
6409                 key = wgetch(status_win);
6411                 /* wgetch() with nodelay() enabled returns ERR when
6412                  * there's no input. */
6413                 if (key == ERR) {
6415                 } else if (key == KEY_RESIZE) {
6416                         int height, width;
6418                         getmaxyx(stdscr, height, width);
6420                         wresize(status_win, 1, width);
6421                         mvwin(status_win, height - 1, 0);
6422                         wnoutrefresh(status_win);
6423                         resize_display();
6424                         redraw_display(TRUE);
6426                 } else {
6427                         input_mode = FALSE;
6428                         return key;
6429                 }
6430         }
6433 static char *
6434 prompt_input(const char *prompt, input_handler handler, void *data)
6436         enum input_status status = INPUT_OK;
6437         static char buf[SIZEOF_STR];
6438         size_t pos = 0;
6440         buf[pos] = 0;
6442         while (status == INPUT_OK || status == INPUT_SKIP) {
6443                 int key;
6445                 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
6446                 wclrtoeol(status_win);
6448                 key = get_input(pos + 1);
6449                 switch (key) {
6450                 case KEY_RETURN:
6451                 case KEY_ENTER:
6452                 case '\n':
6453                         status = pos ? INPUT_STOP : INPUT_CANCEL;
6454                         break;
6456                 case KEY_BACKSPACE:
6457                         if (pos > 0)
6458                                 buf[--pos] = 0;
6459                         else
6460                                 status = INPUT_CANCEL;
6461                         break;
6463                 case KEY_ESC:
6464                         status = INPUT_CANCEL;
6465                         break;
6467                 default:
6468                         if (pos >= sizeof(buf)) {
6469                                 report("Input string too long");
6470                                 return NULL;
6471                         }
6473                         status = handler(data, buf, key);
6474                         if (status == INPUT_OK)
6475                                 buf[pos++] = (char) key;
6476                 }
6477         }
6479         /* Clear the status window */
6480         status_empty = FALSE;
6481         report("");
6483         if (status == INPUT_CANCEL)
6484                 return NULL;
6486         buf[pos++] = 0;
6488         return buf;
6491 static enum input_status
6492 prompt_yesno_handler(void *data, char *buf, int c)
6494         if (c == 'y' || c == 'Y')
6495                 return INPUT_STOP;
6496         if (c == 'n' || c == 'N')
6497                 return INPUT_CANCEL;
6498         return INPUT_SKIP;
6501 static bool
6502 prompt_yesno(const char *prompt)
6504         char prompt2[SIZEOF_STR];
6506         if (!string_format(prompt2, "%s [Yy/Nn]", prompt))
6507                 return FALSE;
6509         return !!prompt_input(prompt2, prompt_yesno_handler, NULL);
6512 static enum input_status
6513 read_prompt_handler(void *data, char *buf, int c)
6515         return isprint(c) ? INPUT_OK : INPUT_SKIP;
6518 static char *
6519 read_prompt(const char *prompt)
6521         return prompt_input(prompt, read_prompt_handler, NULL);
6524 /*
6525  * Repository properties
6526  */
6528 static struct ref *refs = NULL;
6529 static size_t refs_alloc = 0;
6530 static size_t refs_size = 0;
6532 /* Id <-> ref store */
6533 static struct ref ***id_refs = NULL;
6534 static size_t id_refs_alloc = 0;
6535 static size_t id_refs_size = 0;
6537 static int
6538 compare_refs(const void *ref1_, const void *ref2_)
6540         const struct ref *ref1 = *(const struct ref **)ref1_;
6541         const struct ref *ref2 = *(const struct ref **)ref2_;
6543         if (ref1->tag != ref2->tag)
6544                 return ref2->tag - ref1->tag;
6545         if (ref1->ltag != ref2->ltag)
6546                 return ref2->ltag - ref2->ltag;
6547         if (ref1->head != ref2->head)
6548                 return ref2->head - ref1->head;
6549         if (ref1->tracked != ref2->tracked)
6550                 return ref2->tracked - ref1->tracked;
6551         if (ref1->remote != ref2->remote)
6552                 return ref2->remote - ref1->remote;
6553         return strcmp(ref1->name, ref2->name);
6556 static struct ref **
6557 get_refs(const char *id)
6559         struct ref ***tmp_id_refs;
6560         struct ref **ref_list = NULL;
6561         size_t ref_list_alloc = 0;
6562         size_t ref_list_size = 0;
6563         size_t i;
6565         for (i = 0; i < id_refs_size; i++)
6566                 if (!strcmp(id, id_refs[i][0]->id))
6567                         return id_refs[i];
6569         tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
6570                                     sizeof(*id_refs));
6571         if (!tmp_id_refs)
6572                 return NULL;
6574         id_refs = tmp_id_refs;
6576         for (i = 0; i < refs_size; i++) {
6577                 struct ref **tmp;
6579                 if (strcmp(id, refs[i].id))
6580                         continue;
6582                 tmp = realloc_items(ref_list, &ref_list_alloc,
6583                                     ref_list_size + 1, sizeof(*ref_list));
6584                 if (!tmp) {
6585                         if (ref_list)
6586                                 free(ref_list);
6587                         return NULL;
6588                 }
6590                 ref_list = tmp;
6591                 ref_list[ref_list_size] = &refs[i];
6592                 /* XXX: The properties of the commit chains ensures that we can
6593                  * safely modify the shared ref. The repo references will
6594                  * always be similar for the same id. */
6595                 ref_list[ref_list_size]->next = 1;
6597                 ref_list_size++;
6598         }
6600         if (ref_list) {
6601                 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
6602                 ref_list[ref_list_size - 1]->next = 0;
6603                 id_refs[id_refs_size++] = ref_list;
6604         }
6606         return ref_list;
6609 static int
6610 read_ref(char *id, size_t idlen, char *name, size_t namelen)
6612         struct ref *ref;
6613         bool tag = FALSE;
6614         bool ltag = FALSE;
6615         bool remote = FALSE;
6616         bool tracked = FALSE;
6617         bool check_replace = FALSE;
6618         bool head = FALSE;
6620         if (!prefixcmp(name, "refs/tags/")) {
6621                 if (!suffixcmp(name, namelen, "^{}")) {
6622                         namelen -= 3;
6623                         name[namelen] = 0;
6624                         if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
6625                                 check_replace = TRUE;
6626                 } else {
6627                         ltag = TRUE;
6628                 }
6630                 tag = TRUE;
6631                 namelen -= STRING_SIZE("refs/tags/");
6632                 name    += STRING_SIZE("refs/tags/");
6634         } else if (!prefixcmp(name, "refs/remotes/")) {
6635                 remote = TRUE;
6636                 namelen -= STRING_SIZE("refs/remotes/");
6637                 name    += STRING_SIZE("refs/remotes/");
6638                 tracked  = !strcmp(opt_remote, name);
6640         } else if (!prefixcmp(name, "refs/heads/")) {
6641                 namelen -= STRING_SIZE("refs/heads/");
6642                 name    += STRING_SIZE("refs/heads/");
6643                 head     = !strncmp(opt_head, name, namelen);
6645         } else if (!strcmp(name, "HEAD")) {
6646                 string_ncopy(opt_head_rev, id, idlen);
6647                 return OK;
6648         }
6650         if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
6651                 /* it's an annotated tag, replace the previous SHA1 with the
6652                  * resolved commit id; relies on the fact git-ls-remote lists
6653                  * the commit id of an annotated tag right before the commit id
6654                  * it points to. */
6655                 refs[refs_size - 1].ltag = ltag;
6656                 string_copy_rev(refs[refs_size - 1].id, id);
6658                 return OK;
6659         }
6660         refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
6661         if (!refs)
6662                 return ERR;
6664         ref = &refs[refs_size++];
6665         ref->name = malloc(namelen + 1);
6666         if (!ref->name)
6667                 return ERR;
6669         strncpy(ref->name, name, namelen);
6670         ref->name[namelen] = 0;
6671         ref->head = head;
6672         ref->tag = tag;
6673         ref->ltag = ltag;
6674         ref->remote = remote;
6675         ref->tracked = tracked;
6676         string_copy_rev(ref->id, id);
6678         return OK;
6681 static int
6682 load_refs(void)
6684         static const char *ls_remote_argv[SIZEOF_ARG] = {
6685                 "git", "ls-remote", opt_git_dir, NULL
6686         };
6687         static bool init = FALSE;
6689         if (!init) {
6690                 argv_from_env(ls_remote_argv, "TIG_LS_REMOTE");
6691                 init = TRUE;
6692         }
6694         if (!*opt_git_dir)
6695                 return OK;
6697         while (refs_size > 0)
6698                 free(refs[--refs_size].name);
6699         while (id_refs_size > 0)
6700                 free(id_refs[--id_refs_size]);
6702         return run_io_load(ls_remote_argv, "\t", read_ref);
6705 static void
6706 set_remote_branch(const char *name, const char *value, size_t valuelen)
6708         if (!strcmp(name, ".remote")) {
6709                 string_ncopy(opt_remote, value, valuelen);
6711         } else if (*opt_remote && !strcmp(name, ".merge")) {
6712                 size_t from = strlen(opt_remote);
6714                 if (!prefixcmp(value, "refs/heads/"))
6715                         value += STRING_SIZE("refs/heads/");
6717                 if (!string_format_from(opt_remote, &from, "/%s", value))
6718                         opt_remote[0] = 0;
6719         }
6722 static void
6723 set_repo_config_option(char *name, char *value, int (*cmd)(int, const char **))
6725         const char *argv[SIZEOF_ARG] = { name, "=" };
6726         int argc = 1 + (cmd == option_set_command);
6727         int error = ERR;
6729         if (!argv_from_string(argv, &argc, value))
6730                 config_msg = "Too many option arguments";
6731         else
6732                 error = cmd(argc, argv);
6734         if (error == ERR)
6735                 warn("Option 'tig.%s': %s", name, config_msg);
6738 static bool
6739 set_environment_variable(const char *name, const char *value)
6741         size_t len = strlen(name) + 1 + strlen(value) + 1;
6742         char *env = malloc(len);
6744         if (env &&
6745             string_nformat(env, len, NULL, "%s=%s", name, value) &&
6746             putenv(env) == 0)
6747                 return TRUE;
6748         free(env);
6749         return FALSE;
6752 static void
6753 set_work_tree(const char *value)
6755         char cwd[SIZEOF_STR];
6757         if (!getcwd(cwd, sizeof(cwd)))
6758                 die("Failed to get cwd path: %s", strerror(errno));
6759         if (chdir(opt_git_dir) < 0)
6760                 die("Failed to chdir(%s): %s", strerror(errno));
6761         if (!getcwd(opt_git_dir, sizeof(opt_git_dir)))
6762                 die("Failed to get git path: %s", strerror(errno));
6763         if (chdir(cwd) < 0)
6764                 die("Failed to chdir(%s): %s", cwd, strerror(errno));
6765         if (chdir(value) < 0)
6766                 die("Failed to chdir(%s): %s", value, strerror(errno));
6767         if (!getcwd(cwd, sizeof(cwd)))
6768                 die("Failed to get cwd path: %s", strerror(errno));
6769         if (!set_environment_variable("GIT_WORK_TREE", cwd))
6770                 die("Failed to set GIT_WORK_TREE to '%s'", cwd);
6771         if (!set_environment_variable("GIT_DIR", opt_git_dir))
6772                 die("Failed to set GIT_DIR to '%s'", opt_git_dir);
6773         opt_is_inside_work_tree = TRUE;
6776 static int
6777 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
6779         if (!strcmp(name, "i18n.commitencoding"))
6780                 string_ncopy(opt_encoding, value, valuelen);
6782         else if (!strcmp(name, "core.editor"))
6783                 string_ncopy(opt_editor, value, valuelen);
6785         else if (!strcmp(name, "core.worktree"))
6786                 set_work_tree(value);
6788         else if (!prefixcmp(name, "tig.color."))
6789                 set_repo_config_option(name + 10, value, option_color_command);
6791         else if (!prefixcmp(name, "tig.bind."))
6792                 set_repo_config_option(name + 9, value, option_bind_command);
6794         else if (!prefixcmp(name, "tig."))
6795                 set_repo_config_option(name + 4, value, option_set_command);
6797         else if (*opt_head && !prefixcmp(name, "branch.") &&
6798                  !strncmp(name + 7, opt_head, strlen(opt_head)))
6799                 set_remote_branch(name + 7 + strlen(opt_head), value, valuelen);
6801         return OK;
6804 static int
6805 load_git_config(void)
6807         const char *config_list_argv[] = { "git", GIT_CONFIG, "--list", NULL };
6809         return run_io_load(config_list_argv, "=", read_repo_config_option);
6812 static int
6813 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
6815         if (!opt_git_dir[0]) {
6816                 string_ncopy(opt_git_dir, name, namelen);
6818         } else if (opt_is_inside_work_tree == -1) {
6819                 /* This can be 3 different values depending on the
6820                  * version of git being used. If git-rev-parse does not
6821                  * understand --is-inside-work-tree it will simply echo
6822                  * the option else either "true" or "false" is printed.
6823                  * Default to true for the unknown case. */
6824                 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
6826         } else if (*name == '.') {
6827                 string_ncopy(opt_cdup, name, namelen);
6829         } else {
6830                 string_ncopy(opt_prefix, name, namelen);
6831         }
6833         return OK;
6836 static int
6837 load_repo_info(void)
6839         const char *head_argv[] = {
6840                 "git", "symbolic-ref", "HEAD", NULL
6841         };
6842         const char *rev_parse_argv[] = {
6843                 "git", "rev-parse", "--git-dir", "--is-inside-work-tree",
6844                         "--show-cdup", "--show-prefix", NULL
6845         };
6847         if (run_io_buf(head_argv, opt_head, sizeof(opt_head))) {
6848                 chomp_string(opt_head);
6849                 if (!prefixcmp(opt_head, "refs/heads/")) {
6850                         char *offset = opt_head + STRING_SIZE("refs/heads/");
6852                         memmove(opt_head, offset, strlen(offset) + 1);
6853                 }
6854         }
6856         return run_io_load(rev_parse_argv, "=", read_repo_info);
6860 /*
6861  * Main
6862  */
6864 static const char usage[] =
6865 "tig " TIG_VERSION " (" __DATE__ ")\n"
6866 "\n"
6867 "Usage: tig        [options] [revs] [--] [paths]\n"
6868 "   or: tig show   [options] [revs] [--] [paths]\n"
6869 "   or: tig blame  [rev] path\n"
6870 "   or: tig status\n"
6871 "   or: tig <      [git command output]\n"
6872 "\n"
6873 "Options:\n"
6874 "  -v, --version   Show version and exit\n"
6875 "  -h, --help      Show help message and exit";
6877 static void __NORETURN
6878 quit(int sig)
6880         /* XXX: Restore tty modes and let the OS cleanup the rest! */
6881         if (cursed)
6882                 endwin();
6883         exit(0);
6886 static void __NORETURN
6887 die(const char *err, ...)
6889         va_list args;
6891         endwin();
6893         va_start(args, err);
6894         fputs("tig: ", stderr);
6895         vfprintf(stderr, err, args);
6896         fputs("\n", stderr);
6897         va_end(args);
6899         exit(1);
6902 static void
6903 warn(const char *msg, ...)
6905         va_list args;
6907         va_start(args, msg);
6908         fputs("tig warning: ", stderr);
6909         vfprintf(stderr, msg, args);
6910         fputs("\n", stderr);
6911         va_end(args);
6914 static enum request
6915 parse_options(int argc, const char *argv[])
6917         enum request request = REQ_VIEW_MAIN;
6918         const char *subcommand;
6919         bool seen_dashdash = FALSE;
6920         /* XXX: This is vulnerable to the user overriding options
6921          * required for the main view parser. */
6922         const char *custom_argv[SIZEOF_ARG] = {
6923                 "git", "log", "--no-color", "--pretty=raw", "--parents",
6924                         "--topo-order", NULL
6925         };
6926         int i, j = 6;
6928         if (!isatty(STDIN_FILENO)) {
6929                 io_open(&VIEW(REQ_VIEW_PAGER)->io, "");
6930                 return REQ_VIEW_PAGER;
6931         }
6933         if (argc <= 1)
6934                 return REQ_NONE;
6936         subcommand = argv[1];
6937         if (!strcmp(subcommand, "status")) {
6938                 if (argc > 2)
6939                         warn("ignoring arguments after `%s'", subcommand);
6940                 return REQ_VIEW_STATUS;
6942         } else if (!strcmp(subcommand, "blame")) {
6943                 if (argc <= 2 || argc > 4)
6944                         die("invalid number of options to blame\n\n%s", usage);
6946                 i = 2;
6947                 if (argc == 4) {
6948                         string_ncopy(opt_ref, argv[i], strlen(argv[i]));
6949                         i++;
6950                 }
6952                 string_ncopy(opt_file, argv[i], strlen(argv[i]));
6953                 return REQ_VIEW_BLAME;
6955         } else if (!strcmp(subcommand, "show")) {
6956                 request = REQ_VIEW_DIFF;
6958         } else {
6959                 subcommand = NULL;
6960         }
6962         if (subcommand) {
6963                 custom_argv[1] = subcommand;
6964                 j = 2;
6965         }
6967         for (i = 1 + !!subcommand; i < argc; i++) {
6968                 const char *opt = argv[i];
6970                 if (seen_dashdash || !strcmp(opt, "--")) {
6971                         seen_dashdash = TRUE;
6973                 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
6974                         printf("tig version %s\n", TIG_VERSION);
6975                         quit(0);
6977                 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
6978                         printf("%s\n", usage);
6979                         quit(0);
6980                 }
6982                 custom_argv[j++] = opt;
6983                 if (j >= ARRAY_SIZE(custom_argv))
6984                         die("command too long");
6985         }
6987         if (!prepare_update(VIEW(request), custom_argv, NULL, FORMAT_NONE))                                                                        
6988                 die("Failed to format arguments"); 
6990         return request;
6993 int
6994 main(int argc, const char *argv[])
6996         enum request request = parse_options(argc, argv);
6997         struct view *view;
6998         size_t i;
7000         signal(SIGINT, quit);
7001         signal(SIGPIPE, SIG_IGN);
7003         if (setlocale(LC_ALL, "")) {
7004                 char *codeset = nl_langinfo(CODESET);
7006                 string_ncopy(opt_codeset, codeset, strlen(codeset));
7007         }
7009         if (load_repo_info() == ERR)
7010                 die("Failed to load repo info.");
7012         if (load_options() == ERR)
7013                 die("Failed to load user config.");
7015         if (load_git_config() == ERR)
7016                 die("Failed to load repo config.");
7018         /* Require a git repository unless when running in pager mode. */
7019         if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
7020                 die("Not a git repository");
7022         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
7023                 opt_utf8 = FALSE;
7025         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
7026                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
7027                 if (opt_iconv == ICONV_NONE)
7028                         die("Failed to initialize character set conversion");
7029         }
7031         if (load_refs() == ERR)
7032                 die("Failed to load refs.");
7034         foreach_view (view, i)
7035                 argv_from_env(view->ops->argv, view->cmd_env);
7037         init_display();
7039         if (request != REQ_NONE)
7040                 open_view(NULL, request, OPEN_PREPARED);
7041         request = request == REQ_NONE ? REQ_VIEW_MAIN : REQ_NONE;
7043         while (view_driver(display[current_view], request)) {
7044                 int key = get_input(0);
7046                 view = display[current_view];
7047                 request = get_keybinding(view->keymap, key);
7049                 /* Some low-level request handling. This keeps access to
7050                  * status_win restricted. */
7051                 switch (request) {
7052                 case REQ_PROMPT:
7053                 {
7054                         char *cmd = read_prompt(":");
7056                         if (cmd && isdigit(*cmd)) {
7057                                 int lineno = view->lineno + 1;
7059                                 if (parse_int(&lineno, cmd, 1, view->lines + 1) == OK) {
7060                                         select_view_line(view, lineno - 1);
7061                                         report("");
7062                                 } else {
7063                                         report("Unable to parse '%s' as a line number", cmd);
7064                                 }
7066                         } else if (cmd) {
7067                                 struct view *next = VIEW(REQ_VIEW_PAGER);
7068                                 const char *argv[SIZEOF_ARG] = { "git" };
7069                                 int argc = 1;
7071                                 /* When running random commands, initially show the
7072                                  * command in the title. However, it maybe later be
7073                                  * overwritten if a commit line is selected. */
7074                                 string_ncopy(next->ref, cmd, strlen(cmd));
7076                                 if (!argv_from_string(argv, &argc, cmd)) {
7077                                         report("Too many arguments");
7078                                 } else if (!prepare_update(next, argv, NULL, FORMAT_DASH)) {
7079                                         report("Failed to format command");
7080                                 } else {
7081                                         open_view(view, REQ_VIEW_PAGER, OPEN_PREPARED);
7082                                 }
7083                         }
7085                         request = REQ_NONE;
7086                         break;
7087                 }
7088                 case REQ_SEARCH:
7089                 case REQ_SEARCH_BACK:
7090                 {
7091                         const char *prompt = request == REQ_SEARCH ? "/" : "?";
7092                         char *search = read_prompt(prompt);
7094                         if (search)
7095                                 string_ncopy(opt_search, search, strlen(search));
7096                         else if (*opt_search)
7097                                 request = request == REQ_SEARCH ?
7098                                         REQ_FIND_NEXT :
7099                                         REQ_FIND_PREV;
7100                         else
7101                                 request = REQ_NONE;
7102                         break;
7103                 }
7104                 default:
7105                         break;
7106                 }
7107         }
7109         quit(0);
7111         return 0;