Code

74be272a673cd1edb1a17643f4acb377231441e6
[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 const char *
313 mkdate(const time_t *time)
315         static char buf[DATE_COLS + 1];
316         struct tm tm;
318         gmtime_r(time, &tm);
319         return strftime(buf, sizeof(buf), DATE_FORMAT, &tm) ? buf : NULL;
323 static bool
324 argv_from_string(const char *argv[SIZEOF_ARG], int *argc, char *cmd)
326         int valuelen;
328         while (*cmd && *argc < SIZEOF_ARG && (valuelen = strcspn(cmd, " \t"))) {
329                 bool advance = cmd[valuelen] != 0;
331                 cmd[valuelen] = 0;
332                 argv[(*argc)++] = chomp_string(cmd);
333                 cmd = chomp_string(cmd + valuelen + advance);
334         }
336         if (*argc < SIZEOF_ARG)
337                 argv[*argc] = NULL;
338         return *argc < SIZEOF_ARG;
341 static void
342 argv_from_env(const char **argv, const char *name)
344         char *env = argv ? getenv(name) : NULL;
345         int argc = 0;
347         if (env && *env)
348                 env = strdup(env);
349         if (env && !argv_from_string(argv, &argc, env))
350                 die("Too many arguments in the `%s` environment variable", name);
354 /*
355  * Executing external commands.
356  */
358 enum io_type {
359         IO_FD,                  /* File descriptor based IO. */
360         IO_BG,                  /* Execute command in the background. */
361         IO_FG,                  /* Execute command with same std{in,out,err}. */
362         IO_RD,                  /* Read only fork+exec IO. */
363         IO_WR,                  /* Write only fork+exec IO. */
364         IO_AP,                  /* Append fork+exec output to file. */
365 };
367 struct io {
368         enum io_type type;      /* The requested type of pipe. */
369         const char *dir;        /* Directory from which to execute. */
370         pid_t pid;              /* Pipe for reading or writing. */
371         int pipe;               /* Pipe end for reading or writing. */
372         int error;              /* Error status. */
373         const char *argv[SIZEOF_ARG];   /* Shell command arguments. */
374         char *buf;              /* Read buffer. */
375         size_t bufalloc;        /* Allocated buffer size. */
376         size_t bufsize;         /* Buffer content size. */
377         char *bufpos;           /* Current buffer position. */
378         unsigned int eof:1;     /* Has end of file been reached. */
379 };
381 static void
382 reset_io(struct io *io)
384         io->pipe = -1;
385         io->pid = 0;
386         io->buf = io->bufpos = NULL;
387         io->bufalloc = io->bufsize = 0;
388         io->error = 0;
389         io->eof = 0;
392 static void
393 init_io(struct io *io, const char *dir, enum io_type type)
395         reset_io(io);
396         io->type = type;
397         io->dir = dir;
400 static bool
401 init_io_rd(struct io *io, const char *argv[], const char *dir,
402                 enum format_flags flags)
404         init_io(io, dir, IO_RD);
405         return format_argv(io->argv, argv, flags);
408 static bool
409 io_open(struct io *io, const char *name)
411         init_io(io, NULL, IO_FD);
412         io->pipe = *name ? open(name, O_RDONLY) : STDIN_FILENO;
413         if (io->pipe == -1)
414                 io->error = errno;
415         return io->pipe != -1;
418 static bool
419 kill_io(struct io *io)
421         return io->pid == 0 || kill(io->pid, SIGKILL) != -1;
424 static bool
425 done_io(struct io *io)
427         pid_t pid = io->pid;
429         if (io->pipe != -1)
430                 close(io->pipe);
431         free(io->buf);
432         reset_io(io);
434         while (pid > 0) {
435                 int status;
436                 pid_t waiting = waitpid(pid, &status, 0);
438                 if (waiting < 0) {
439                         if (errno == EINTR)
440                                 continue;
441                         report("waitpid failed (%s)", strerror(errno));
442                         return FALSE;
443                 }
445                 return waiting == pid &&
446                        !WIFSIGNALED(status) &&
447                        WIFEXITED(status) &&
448                        !WEXITSTATUS(status);
449         }
451         return TRUE;
454 static bool
455 start_io(struct io *io)
457         int pipefds[2] = { -1, -1 };
459         if (io->type == IO_FD)
460                 return TRUE;
462         if ((io->type == IO_RD || io->type == IO_WR) &&
463             pipe(pipefds) < 0)
464                 return FALSE;
465         else if (io->type == IO_AP)
466                 pipefds[1] = io->pipe;
468         if ((io->pid = fork())) {
469                 if (pipefds[!(io->type == IO_WR)] != -1)
470                         close(pipefds[!(io->type == IO_WR)]);
471                 if (io->pid != -1) {
472                         io->pipe = pipefds[!!(io->type == IO_WR)];
473                         return TRUE;
474                 }
476         } else {
477                 if (io->type != IO_FG) {
478                         int devnull = open("/dev/null", O_RDWR);
479                         int readfd  = io->type == IO_WR ? pipefds[0] : devnull;
480                         int writefd = (io->type == IO_RD || io->type == IO_AP)
481                                                         ? pipefds[1] : devnull;
483                         dup2(readfd,  STDIN_FILENO);
484                         dup2(writefd, STDOUT_FILENO);
485                         dup2(devnull, STDERR_FILENO);
487                         close(devnull);
488                         if (pipefds[0] != -1)
489                                 close(pipefds[0]);
490                         if (pipefds[1] != -1)
491                                 close(pipefds[1]);
492                 }
494                 if (io->dir && *io->dir && chdir(io->dir) == -1)
495                         die("Failed to change directory: %s", strerror(errno));
497                 execvp(io->argv[0], (char *const*) io->argv);
498                 die("Failed to execute program: %s", strerror(errno));
499         }
501         if (pipefds[!!(io->type == IO_WR)] != -1)
502                 close(pipefds[!!(io->type == IO_WR)]);
503         return FALSE;
506 static bool
507 run_io(struct io *io, const char **argv, const char *dir, enum io_type type)
509         init_io(io, dir, type);
510         if (!format_argv(io->argv, argv, FORMAT_NONE))
511                 return FALSE;
512         return start_io(io);
515 static int
516 run_io_do(struct io *io)
518         return start_io(io) && done_io(io);
521 static int
522 run_io_bg(const char **argv)
524         struct io io = {};
526         init_io(&io, NULL, IO_BG);
527         if (!format_argv(io.argv, argv, FORMAT_NONE))
528                 return FALSE;
529         return run_io_do(&io);
532 static bool
533 run_io_fg(const char **argv, const char *dir)
535         struct io io = {};
537         init_io(&io, dir, IO_FG);
538         if (!format_argv(io.argv, argv, FORMAT_NONE))
539                 return FALSE;
540         return run_io_do(&io);
543 static bool
544 run_io_append(const char **argv, enum format_flags flags, int fd)
546         struct io io = {};
548         init_io(&io, NULL, IO_AP);
549         io.pipe = fd;
550         if (format_argv(io.argv, argv, flags))
551                 return run_io_do(&io);
552         close(fd);
553         return FALSE;
556 static bool
557 run_io_rd(struct io *io, const char **argv, enum format_flags flags)
559         return init_io_rd(io, argv, NULL, flags) && start_io(io);
562 static bool
563 io_eof(struct io *io)
565         return io->eof;
568 static int
569 io_error(struct io *io)
571         return io->error;
574 static char *
575 io_strerror(struct io *io)
577         return strerror(io->error);
580 static bool
581 io_can_read(struct io *io)
583         struct timeval tv = { 0, 500 };
584         fd_set fds;
586         FD_ZERO(&fds);
587         FD_SET(io->pipe, &fds);
589         return select(io->pipe + 1, &fds, NULL, NULL, &tv) > 0;
592 static ssize_t
593 io_read(struct io *io, void *buf, size_t bufsize)
595         do {
596                 ssize_t readsize = read(io->pipe, buf, bufsize);
598                 if (readsize < 0 && (errno == EAGAIN || errno == EINTR))
599                         continue;
600                 else if (readsize == -1)
601                         io->error = errno;
602                 else if (readsize == 0)
603                         io->eof = 1;
604                 return readsize;
605         } while (1);
608 static char *
609 io_get(struct io *io, int c, bool can_read)
611         char *eol;
612         ssize_t readsize;
614         if (!io->buf) {
615                 io->buf = io->bufpos = malloc(BUFSIZ);
616                 if (!io->buf)
617                         return NULL;
618                 io->bufalloc = BUFSIZ;
619                 io->bufsize = 0;
620         }
622         while (TRUE) {
623                 if (io->bufsize > 0) {
624                         eol = memchr(io->bufpos, c, io->bufsize);
625                         if (eol) {
626                                 char *line = io->bufpos;
628                                 *eol = 0;
629                                 io->bufpos = eol + 1;
630                                 io->bufsize -= io->bufpos - line;
631                                 return line;
632                         }
633                 }
635                 if (io_eof(io)) {
636                         if (io->bufsize) {
637                                 io->bufpos[io->bufsize] = 0;
638                                 io->bufsize = 0;
639                                 return io->bufpos;
640                         }
641                         return NULL;
642                 }
644                 if (!can_read)
645                         return NULL;
647                 if (io->bufsize > 0 && io->bufpos > io->buf)
648                         memmove(io->buf, io->bufpos, io->bufsize);
650                 io->bufpos = io->buf;
651                 readsize = io_read(io, io->buf + io->bufsize, io->bufalloc - io->bufsize);
652                 if (io_error(io))
653                         return NULL;
654                 io->bufsize += readsize;
655         }
658 static bool
659 io_write(struct io *io, const void *buf, size_t bufsize)
661         size_t written = 0;
663         while (!io_error(io) && written < bufsize) {
664                 ssize_t size;
666                 size = write(io->pipe, buf + written, bufsize - written);
667                 if (size < 0 && (errno == EAGAIN || errno == EINTR))
668                         continue;
669                 else if (size == -1)
670                         io->error = errno;
671                 else
672                         written += size;
673         }
675         return written == bufsize;
678 static bool
679 io_read_buf(struct io *io, char buf[], size_t bufsize)
681         bool error;
683         io->buf = io->bufpos = buf;
684         io->bufalloc = bufsize;
685         error = !io_get(io, '\n', TRUE) && io_error(io);
686         io->buf = NULL;
688         return done_io(io) || error;
691 static bool
692 run_io_buf(const char **argv, char buf[], size_t bufsize)
694         struct io io = {};
696         return run_io_rd(&io, argv, FORMAT_NONE) && io_read_buf(&io, buf, bufsize);
699 static int
700 io_load(struct io *io, const char *separators,
701         int (*read_property)(char *, size_t, char *, size_t))
703         char *name;
704         int state = OK;
706         if (!start_io(io))
707                 return ERR;
709         while (state == OK && (name = io_get(io, '\n', TRUE))) {
710                 char *value;
711                 size_t namelen;
712                 size_t valuelen;
714                 name = chomp_string(name);
715                 namelen = strcspn(name, separators);
717                 if (name[namelen]) {
718                         name[namelen] = 0;
719                         value = chomp_string(name + namelen + 1);
720                         valuelen = strlen(value);
722                 } else {
723                         value = "";
724                         valuelen = 0;
725                 }
727                 state = read_property(name, namelen, value, valuelen);
728         }
730         if (state != ERR && io_error(io))
731                 state = ERR;
732         done_io(io);
734         return state;
737 static int
738 run_io_load(const char **argv, const char *separators,
739             int (*read_property)(char *, size_t, char *, size_t))
741         struct io io = {};
743         return init_io_rd(&io, argv, NULL, FORMAT_NONE)
744                 ? io_load(&io, separators, read_property) : ERR;
748 /*
749  * User requests
750  */
752 #define REQ_INFO \
753         /* XXX: Keep the view request first and in sync with views[]. */ \
754         REQ_GROUP("View switching") \
755         REQ_(VIEW_MAIN,         "Show main view"), \
756         REQ_(VIEW_DIFF,         "Show diff view"), \
757         REQ_(VIEW_LOG,          "Show log view"), \
758         REQ_(VIEW_TREE,         "Show tree view"), \
759         REQ_(VIEW_BLOB,         "Show blob view"), \
760         REQ_(VIEW_BLAME,        "Show blame view"), \
761         REQ_(VIEW_HELP,         "Show help page"), \
762         REQ_(VIEW_PAGER,        "Show pager view"), \
763         REQ_(VIEW_STATUS,       "Show status view"), \
764         REQ_(VIEW_STAGE,        "Show stage view"), \
765         \
766         REQ_GROUP("View manipulation") \
767         REQ_(ENTER,             "Enter current line and scroll"), \
768         REQ_(NEXT,              "Move to next"), \
769         REQ_(PREVIOUS,          "Move to previous"), \
770         REQ_(PARENT,            "Move to parent"), \
771         REQ_(VIEW_NEXT,         "Move focus to next view"), \
772         REQ_(REFRESH,           "Reload and refresh"), \
773         REQ_(MAXIMIZE,          "Maximize the current view"), \
774         REQ_(VIEW_CLOSE,        "Close the current view"), \
775         REQ_(QUIT,              "Close all views and quit"), \
776         \
777         REQ_GROUP("View specific requests") \
778         REQ_(STATUS_UPDATE,     "Update file status"), \
779         REQ_(STATUS_REVERT,     "Revert file changes"), \
780         REQ_(STATUS_MERGE,      "Merge file using external tool"), \
781         REQ_(STAGE_NEXT,        "Find next chunk to stage"), \
782         \
783         REQ_GROUP("Cursor navigation") \
784         REQ_(MOVE_UP,           "Move cursor one line up"), \
785         REQ_(MOVE_DOWN,         "Move cursor one line down"), \
786         REQ_(MOVE_PAGE_DOWN,    "Move cursor one page down"), \
787         REQ_(MOVE_PAGE_UP,      "Move cursor one page up"), \
788         REQ_(MOVE_FIRST_LINE,   "Move cursor to first line"), \
789         REQ_(MOVE_LAST_LINE,    "Move cursor to last line"), \
790         \
791         REQ_GROUP("Scrolling") \
792         REQ_(SCROLL_LEFT,       "Scroll two columns left"), \
793         REQ_(SCROLL_RIGHT,      "Scroll two columns right"), \
794         REQ_(SCROLL_LINE_UP,    "Scroll one line up"), \
795         REQ_(SCROLL_LINE_DOWN,  "Scroll one line down"), \
796         REQ_(SCROLL_PAGE_UP,    "Scroll one page up"), \
797         REQ_(SCROLL_PAGE_DOWN,  "Scroll one page down"), \
798         \
799         REQ_GROUP("Searching") \
800         REQ_(SEARCH,            "Search the view"), \
801         REQ_(SEARCH_BACK,       "Search backwards in the view"), \
802         REQ_(FIND_NEXT,         "Find next search match"), \
803         REQ_(FIND_PREV,         "Find previous search match"), \
804         \
805         REQ_GROUP("Option manipulation") \
806         REQ_(TOGGLE_LINENO,     "Toggle line numbers"), \
807         REQ_(TOGGLE_DATE,       "Toggle date display"), \
808         REQ_(TOGGLE_AUTHOR,     "Toggle author display"), \
809         REQ_(TOGGLE_REV_GRAPH,  "Toggle revision graph visualization"), \
810         REQ_(TOGGLE_REFS,       "Toggle reference display (tags/branches)"), \
811         \
812         REQ_GROUP("Misc") \
813         REQ_(PROMPT,            "Bring up the prompt"), \
814         REQ_(SCREEN_REDRAW,     "Redraw the screen"), \
815         REQ_(SHOW_VERSION,      "Show version information"), \
816         REQ_(STOP_LOADING,      "Stop all loading views"), \
817         REQ_(EDIT,              "Open in editor"), \
818         REQ_(NONE,              "Do nothing")
821 /* User action requests. */
822 enum request {
823 #define REQ_GROUP(help)
824 #define REQ_(req, help) REQ_##req
826         /* Offset all requests to avoid conflicts with ncurses getch values. */
827         REQ_OFFSET = KEY_MAX + 1,
828         REQ_INFO
830 #undef  REQ_GROUP
831 #undef  REQ_
832 };
834 struct request_info {
835         enum request request;
836         const char *name;
837         int namelen;
838         const char *help;
839 };
841 static const struct request_info req_info[] = {
842 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
843 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
844         REQ_INFO
845 #undef  REQ_GROUP
846 #undef  REQ_
847 };
849 static enum request
850 get_request(const char *name)
852         int namelen = strlen(name);
853         int i;
855         for (i = 0; i < ARRAY_SIZE(req_info); i++)
856                 if (req_info[i].namelen == namelen &&
857                     !string_enum_compare(req_info[i].name, name, namelen))
858                         return req_info[i].request;
860         return REQ_NONE;
864 /*
865  * Options
866  */
868 /* Option and state variables. */
869 static bool opt_date                    = TRUE;
870 static bool opt_author                  = TRUE;
871 static bool opt_line_number             = FALSE;
872 static bool opt_line_graphics           = TRUE;
873 static bool opt_rev_graph               = FALSE;
874 static bool opt_show_refs               = TRUE;
875 static int opt_num_interval             = NUMBER_INTERVAL;
876 static double opt_hscroll               = 0.50;
877 static int opt_tab_size                 = TAB_SIZE;
878 static int opt_author_cols              = AUTHOR_COLS-1;
879 static char opt_path[SIZEOF_STR]        = "";
880 static char opt_file[SIZEOF_STR]        = "";
881 static char opt_ref[SIZEOF_REF]         = "";
882 static char opt_head[SIZEOF_REF]        = "";
883 static char opt_head_rev[SIZEOF_REV]    = "";
884 static char opt_remote[SIZEOF_REF]      = "";
885 static char opt_encoding[20]            = "UTF-8";
886 static bool opt_utf8                    = TRUE;
887 static char opt_codeset[20]             = "UTF-8";
888 static iconv_t opt_iconv                = ICONV_NONE;
889 static char opt_search[SIZEOF_STR]      = "";
890 static char opt_cdup[SIZEOF_STR]        = "";
891 static char opt_prefix[SIZEOF_STR]      = "";
892 static char opt_git_dir[SIZEOF_STR]     = "";
893 static signed char opt_is_inside_work_tree      = -1; /* set to TRUE or FALSE */
894 static char opt_editor[SIZEOF_STR]      = "";
895 static FILE *opt_tty                    = NULL;
897 #define is_initial_commit()     (!*opt_head_rev)
898 #define is_head_commit(rev)     (!strcmp((rev), "HEAD") || !strcmp(opt_head_rev, (rev)))
901 /*
902  * Line-oriented content detection.
903  */
905 #define LINE_INFO \
906 LINE(DIFF_HEADER,  "diff --git ",       COLOR_YELLOW,   COLOR_DEFAULT,  0), \
907 LINE(DIFF_CHUNK,   "@@",                COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
908 LINE(DIFF_ADD,     "+",                 COLOR_GREEN,    COLOR_DEFAULT,  0), \
909 LINE(DIFF_DEL,     "-",                 COLOR_RED,      COLOR_DEFAULT,  0), \
910 LINE(DIFF_INDEX,        "index ",         COLOR_BLUE,   COLOR_DEFAULT,  0), \
911 LINE(DIFF_OLDMODE,      "old file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
912 LINE(DIFF_NEWMODE,      "new file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
913 LINE(DIFF_COPY_FROM,    "copy from",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
914 LINE(DIFF_COPY_TO,      "copy to",        COLOR_YELLOW, COLOR_DEFAULT,  0), \
915 LINE(DIFF_RENAME_FROM,  "rename from",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
916 LINE(DIFF_RENAME_TO,    "rename to",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
917 LINE(DIFF_SIMILARITY,   "similarity ",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
918 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
919 LINE(DIFF_TREE,         "diff-tree ",     COLOR_BLUE,   COLOR_DEFAULT,  0), \
920 LINE(PP_AUTHOR,    "Author: ",          COLOR_CYAN,     COLOR_DEFAULT,  0), \
921 LINE(PP_COMMIT,    "Commit: ",          COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
922 LINE(PP_MERGE,     "Merge: ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
923 LINE(PP_DATE,      "Date:   ",          COLOR_YELLOW,   COLOR_DEFAULT,  0), \
924 LINE(PP_ADATE,     "AuthorDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
925 LINE(PP_CDATE,     "CommitDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
926 LINE(PP_REFS,      "Refs: ",            COLOR_RED,      COLOR_DEFAULT,  0), \
927 LINE(COMMIT,       "commit ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
928 LINE(PARENT,       "parent ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
929 LINE(TREE,         "tree ",             COLOR_BLUE,     COLOR_DEFAULT,  0), \
930 LINE(AUTHOR,       "author ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
931 LINE(COMMITTER,    "committer ",        COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
932 LINE(SIGNOFF,      "    Signed-off-by", COLOR_YELLOW,   COLOR_DEFAULT,  0), \
933 LINE(ACKED,        "    Acked-by",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
934 LINE(DEFAULT,      "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
935 LINE(CURSOR,       "",                  COLOR_WHITE,    COLOR_GREEN,    A_BOLD), \
936 LINE(STATUS,       "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
937 LINE(DELIMITER,    "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
938 LINE(DATE,         "",                  COLOR_BLUE,     COLOR_DEFAULT,  0), \
939 LINE(MODE,         "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
940 LINE(LINE_NUMBER,  "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
941 LINE(TITLE_BLUR,   "",                  COLOR_WHITE,    COLOR_BLUE,     0), \
942 LINE(TITLE_FOCUS,  "",                  COLOR_WHITE,    COLOR_BLUE,     A_BOLD), \
943 LINE(MAIN_COMMIT,  "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
944 LINE(MAIN_TAG,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  A_BOLD), \
945 LINE(MAIN_LOCAL_TAG,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
946 LINE(MAIN_REMOTE,  "",                  COLOR_YELLOW,   COLOR_DEFAULT,  0), \
947 LINE(MAIN_TRACKED, "",                  COLOR_YELLOW,   COLOR_DEFAULT,  A_BOLD), \
948 LINE(MAIN_REF,     "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
949 LINE(MAIN_HEAD,    "",                  COLOR_CYAN,     COLOR_DEFAULT,  A_BOLD), \
950 LINE(MAIN_REVGRAPH,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
951 LINE(TREE_HEAD,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_BOLD), \
952 LINE(TREE_DIR,     "",                  COLOR_YELLOW,   COLOR_DEFAULT,  A_NORMAL), \
953 LINE(TREE_FILE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
954 LINE(STAT_HEAD,    "",                  COLOR_YELLOW,   COLOR_DEFAULT,  0), \
955 LINE(STAT_SECTION, "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
956 LINE(STAT_NONE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
957 LINE(STAT_STAGED,  "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
958 LINE(STAT_UNSTAGED,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
959 LINE(STAT_UNTRACKED,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
960 LINE(BLAME_ID,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0)
962 enum line_type {
963 #define LINE(type, line, fg, bg, attr) \
964         LINE_##type
965         LINE_INFO,
966         LINE_NONE
967 #undef  LINE
968 };
970 struct line_info {
971         const char *name;       /* Option name. */
972         int namelen;            /* Size of option name. */
973         const char *line;       /* The start of line to match. */
974         int linelen;            /* Size of string to match. */
975         int fg, bg, attr;       /* Color and text attributes for the lines. */
976 };
978 static struct line_info line_info[] = {
979 #define LINE(type, line, fg, bg, attr) \
980         { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
981         LINE_INFO
982 #undef  LINE
983 };
985 static enum line_type
986 get_line_type(const char *line)
988         int linelen = strlen(line);
989         enum line_type type;
991         for (type = 0; type < ARRAY_SIZE(line_info); type++)
992                 /* Case insensitive search matches Signed-off-by lines better. */
993                 if (linelen >= line_info[type].linelen &&
994                     !strncasecmp(line_info[type].line, line, line_info[type].linelen))
995                         return type;
997         return LINE_DEFAULT;
1000 static inline int
1001 get_line_attr(enum line_type type)
1003         assert(type < ARRAY_SIZE(line_info));
1004         return COLOR_PAIR(type) | line_info[type].attr;
1007 static struct line_info *
1008 get_line_info(const char *name)
1010         size_t namelen = strlen(name);
1011         enum line_type type;
1013         for (type = 0; type < ARRAY_SIZE(line_info); type++)
1014                 if (namelen == line_info[type].namelen &&
1015                     !string_enum_compare(line_info[type].name, name, namelen))
1016                         return &line_info[type];
1018         return NULL;
1021 static void
1022 init_colors(void)
1024         int default_bg = line_info[LINE_DEFAULT].bg;
1025         int default_fg = line_info[LINE_DEFAULT].fg;
1026         enum line_type type;
1028         start_color();
1030         if (assume_default_colors(default_fg, default_bg) == ERR) {
1031                 default_bg = COLOR_BLACK;
1032                 default_fg = COLOR_WHITE;
1033         }
1035         for (type = 0; type < ARRAY_SIZE(line_info); type++) {
1036                 struct line_info *info = &line_info[type];
1037                 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
1038                 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
1040                 init_pair(type, fg, bg);
1041         }
1044 struct line {
1045         enum line_type type;
1047         /* State flags */
1048         unsigned int selected:1;
1049         unsigned int dirty:1;
1050         unsigned int cleareol:1;
1052         void *data;             /* User data */
1053 };
1056 /*
1057  * Keys
1058  */
1060 struct keybinding {
1061         int alias;
1062         enum request request;
1063 };
1065 static const struct keybinding default_keybindings[] = {
1066         /* View switching */
1067         { 'm',          REQ_VIEW_MAIN },
1068         { 'd',          REQ_VIEW_DIFF },
1069         { 'l',          REQ_VIEW_LOG },
1070         { 't',          REQ_VIEW_TREE },
1071         { 'f',          REQ_VIEW_BLOB },
1072         { 'B',          REQ_VIEW_BLAME },
1073         { 'p',          REQ_VIEW_PAGER },
1074         { 'h',          REQ_VIEW_HELP },
1075         { 'S',          REQ_VIEW_STATUS },
1076         { 'c',          REQ_VIEW_STAGE },
1078         /* View manipulation */
1079         { 'q',          REQ_VIEW_CLOSE },
1080         { KEY_TAB,      REQ_VIEW_NEXT },
1081         { KEY_RETURN,   REQ_ENTER },
1082         { KEY_UP,       REQ_PREVIOUS },
1083         { KEY_DOWN,     REQ_NEXT },
1084         { 'R',          REQ_REFRESH },
1085         { KEY_F(5),     REQ_REFRESH },
1086         { 'O',          REQ_MAXIMIZE },
1088         /* Cursor navigation */
1089         { 'k',          REQ_MOVE_UP },
1090         { 'j',          REQ_MOVE_DOWN },
1091         { KEY_HOME,     REQ_MOVE_FIRST_LINE },
1092         { KEY_END,      REQ_MOVE_LAST_LINE },
1093         { KEY_NPAGE,    REQ_MOVE_PAGE_DOWN },
1094         { ' ',          REQ_MOVE_PAGE_DOWN },
1095         { KEY_PPAGE,    REQ_MOVE_PAGE_UP },
1096         { 'b',          REQ_MOVE_PAGE_UP },
1097         { '-',          REQ_MOVE_PAGE_UP },
1099         /* Scrolling */
1100         { KEY_LEFT,     REQ_SCROLL_LEFT },
1101         { KEY_RIGHT,    REQ_SCROLL_RIGHT },
1102         { KEY_IC,       REQ_SCROLL_LINE_UP },
1103         { KEY_DC,       REQ_SCROLL_LINE_DOWN },
1104         { 'w',          REQ_SCROLL_PAGE_UP },
1105         { 's',          REQ_SCROLL_PAGE_DOWN },
1107         /* Searching */
1108         { '/',          REQ_SEARCH },
1109         { '?',          REQ_SEARCH_BACK },
1110         { 'n',          REQ_FIND_NEXT },
1111         { 'N',          REQ_FIND_PREV },
1113         /* Misc */
1114         { 'Q',          REQ_QUIT },
1115         { 'z',          REQ_STOP_LOADING },
1116         { 'v',          REQ_SHOW_VERSION },
1117         { 'r',          REQ_SCREEN_REDRAW },
1118         { '.',          REQ_TOGGLE_LINENO },
1119         { 'D',          REQ_TOGGLE_DATE },
1120         { 'A',          REQ_TOGGLE_AUTHOR },
1121         { 'g',          REQ_TOGGLE_REV_GRAPH },
1122         { 'F',          REQ_TOGGLE_REFS },
1123         { ':',          REQ_PROMPT },
1124         { 'u',          REQ_STATUS_UPDATE },
1125         { '!',          REQ_STATUS_REVERT },
1126         { 'M',          REQ_STATUS_MERGE },
1127         { '@',          REQ_STAGE_NEXT },
1128         { ',',          REQ_PARENT },
1129         { 'e',          REQ_EDIT },
1130 };
1132 #define KEYMAP_INFO \
1133         KEYMAP_(GENERIC), \
1134         KEYMAP_(MAIN), \
1135         KEYMAP_(DIFF), \
1136         KEYMAP_(LOG), \
1137         KEYMAP_(TREE), \
1138         KEYMAP_(BLOB), \
1139         KEYMAP_(BLAME), \
1140         KEYMAP_(PAGER), \
1141         KEYMAP_(HELP), \
1142         KEYMAP_(STATUS), \
1143         KEYMAP_(STAGE)
1145 enum keymap {
1146 #define KEYMAP_(name) KEYMAP_##name
1147         KEYMAP_INFO
1148 #undef  KEYMAP_
1149 };
1151 static const struct enum_map keymap_table[] = {
1152 #define KEYMAP_(name) ENUM_MAP(#name, KEYMAP_##name)
1153         KEYMAP_INFO
1154 #undef  KEYMAP_
1155 };
1157 #define set_keymap(map, name) map_enum(map, keymap_table, name)
1159 struct keybinding_table {
1160         struct keybinding *data;
1161         size_t size;
1162 };
1164 static struct keybinding_table keybindings[ARRAY_SIZE(keymap_table)];
1166 static void
1167 add_keybinding(enum keymap keymap, enum request request, int key)
1169         struct keybinding_table *table = &keybindings[keymap];
1171         table->data = realloc(table->data, (table->size + 1) * sizeof(*table->data));
1172         if (!table->data)
1173                 die("Failed to allocate keybinding");
1174         table->data[table->size].alias = key;
1175         table->data[table->size++].request = request;
1178 /* Looks for a key binding first in the given map, then in the generic map, and
1179  * lastly in the default keybindings. */
1180 static enum request
1181 get_keybinding(enum keymap keymap, int key)
1183         size_t i;
1185         for (i = 0; i < keybindings[keymap].size; i++)
1186                 if (keybindings[keymap].data[i].alias == key)
1187                         return keybindings[keymap].data[i].request;
1189         for (i = 0; i < keybindings[KEYMAP_GENERIC].size; i++)
1190                 if (keybindings[KEYMAP_GENERIC].data[i].alias == key)
1191                         return keybindings[KEYMAP_GENERIC].data[i].request;
1193         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
1194                 if (default_keybindings[i].alias == key)
1195                         return default_keybindings[i].request;
1197         return (enum request) key;
1201 struct key {
1202         const char *name;
1203         int value;
1204 };
1206 static const struct key key_table[] = {
1207         { "Enter",      KEY_RETURN },
1208         { "Space",      ' ' },
1209         { "Backspace",  KEY_BACKSPACE },
1210         { "Tab",        KEY_TAB },
1211         { "Escape",     KEY_ESC },
1212         { "Left",       KEY_LEFT },
1213         { "Right",      KEY_RIGHT },
1214         { "Up",         KEY_UP },
1215         { "Down",       KEY_DOWN },
1216         { "Insert",     KEY_IC },
1217         { "Delete",     KEY_DC },
1218         { "Hash",       '#' },
1219         { "Home",       KEY_HOME },
1220         { "End",        KEY_END },
1221         { "PageUp",     KEY_PPAGE },
1222         { "PageDown",   KEY_NPAGE },
1223         { "F1",         KEY_F(1) },
1224         { "F2",         KEY_F(2) },
1225         { "F3",         KEY_F(3) },
1226         { "F4",         KEY_F(4) },
1227         { "F5",         KEY_F(5) },
1228         { "F6",         KEY_F(6) },
1229         { "F7",         KEY_F(7) },
1230         { "F8",         KEY_F(8) },
1231         { "F9",         KEY_F(9) },
1232         { "F10",        KEY_F(10) },
1233         { "F11",        KEY_F(11) },
1234         { "F12",        KEY_F(12) },
1235 };
1237 static int
1238 get_key_value(const char *name)
1240         int i;
1242         for (i = 0; i < ARRAY_SIZE(key_table); i++)
1243                 if (!strcasecmp(key_table[i].name, name))
1244                         return key_table[i].value;
1246         if (strlen(name) == 1 && isprint(*name))
1247                 return (int) *name;
1249         return ERR;
1252 static const char *
1253 get_key_name(int key_value)
1255         static char key_char[] = "'X'";
1256         const char *seq = NULL;
1257         int key;
1259         for (key = 0; key < ARRAY_SIZE(key_table); key++)
1260                 if (key_table[key].value == key_value)
1261                         seq = key_table[key].name;
1263         if (seq == NULL &&
1264             key_value < 127 &&
1265             isprint(key_value)) {
1266                 key_char[1] = (char) key_value;
1267                 seq = key_char;
1268         }
1270         return seq ? seq : "(no key)";
1273 static const char *
1274 get_key(enum request request)
1276         static char buf[BUFSIZ];
1277         size_t pos = 0;
1278         char *sep = "";
1279         int i;
1281         buf[pos] = 0;
1283         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
1284                 const struct keybinding *keybinding = &default_keybindings[i];
1286                 if (keybinding->request != request)
1287                         continue;
1289                 if (!string_format_from(buf, &pos, "%s%s", sep,
1290                                         get_key_name(keybinding->alias)))
1291                         return "Too many keybindings!";
1292                 sep = ", ";
1293         }
1295         return buf;
1298 struct run_request {
1299         enum keymap keymap;
1300         int key;
1301         const char *argv[SIZEOF_ARG];
1302 };
1304 static struct run_request *run_request;
1305 static size_t run_requests;
1307 static enum request
1308 add_run_request(enum keymap keymap, int key, int argc, const char **argv)
1310         struct run_request *req;
1312         if (argc >= ARRAY_SIZE(req->argv) - 1)
1313                 return REQ_NONE;
1315         req = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
1316         if (!req)
1317                 return REQ_NONE;
1319         run_request = req;
1320         req = &run_request[run_requests];
1321         req->keymap = keymap;
1322         req->key = key;
1323         req->argv[0] = NULL;
1325         if (!format_argv(req->argv, argv, FORMAT_NONE))
1326                 return REQ_NONE;
1328         return REQ_NONE + ++run_requests;
1331 static struct run_request *
1332 get_run_request(enum request request)
1334         if (request <= REQ_NONE)
1335                 return NULL;
1336         return &run_request[request - REQ_NONE - 1];
1339 static void
1340 add_builtin_run_requests(void)
1342         const char *cherry_pick[] = { "git", "cherry-pick", "%(commit)", NULL };
1343         const char *gc[] = { "git", "gc", NULL };
1344         struct {
1345                 enum keymap keymap;
1346                 int key;
1347                 int argc;
1348                 const char **argv;
1349         } reqs[] = {
1350                 { KEYMAP_MAIN,    'C', ARRAY_SIZE(cherry_pick) - 1, cherry_pick },
1351                 { KEYMAP_GENERIC, 'G', ARRAY_SIZE(gc) - 1, gc },
1352         };
1353         int i;
1355         for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1356                 enum request req;
1358                 req = add_run_request(reqs[i].keymap, reqs[i].key, reqs[i].argc, reqs[i].argv);
1359                 if (req != REQ_NONE)
1360                         add_keybinding(reqs[i].keymap, req, reqs[i].key);
1361         }
1364 /*
1365  * User config file handling.
1366  */
1368 static int   config_lineno;
1369 static bool  config_errors;
1370 static const char *config_msg;
1372 static const struct enum_map color_map[] = {
1373 #define COLOR_MAP(name) ENUM_MAP(#name, COLOR_##name)
1374         COLOR_MAP(DEFAULT),
1375         COLOR_MAP(BLACK),
1376         COLOR_MAP(BLUE),
1377         COLOR_MAP(CYAN),
1378         COLOR_MAP(GREEN),
1379         COLOR_MAP(MAGENTA),
1380         COLOR_MAP(RED),
1381         COLOR_MAP(WHITE),
1382         COLOR_MAP(YELLOW),
1383 };
1385 static const struct enum_map attr_map[] = {
1386 #define ATTR_MAP(name) ENUM_MAP(#name, A_##name)
1387         ATTR_MAP(NORMAL),
1388         ATTR_MAP(BLINK),
1389         ATTR_MAP(BOLD),
1390         ATTR_MAP(DIM),
1391         ATTR_MAP(REVERSE),
1392         ATTR_MAP(STANDOUT),
1393         ATTR_MAP(UNDERLINE),
1394 };
1396 #define set_attribute(attr, name)       map_enum(attr, attr_map, name)
1398 static int parse_step(double *opt, const char *arg)
1400         *opt = atoi(arg);
1401         if (!strchr(arg, '%'))
1402                 return OK;
1404         /* "Shift down" so 100% and 1 does not conflict. */
1405         *opt = (*opt - 1) / 100;
1406         if (*opt >= 1.0) {
1407                 *opt = 0.99;
1408                 config_msg = "Step value larger than 100%";
1409                 return ERR;
1410         }
1411         if (*opt < 0.0) {
1412                 *opt = 1;
1413                 config_msg = "Invalid step value";
1414                 return ERR;
1415         }
1416         return OK;
1419 static int
1420 parse_int(int *opt, const char *arg, int min, int max)
1422         int value = atoi(arg);
1424         if (min <= value && value <= max) {
1425                 *opt = value;
1426                 return OK;
1427         }
1429         config_msg = "Integer value out of bound";
1430         return ERR;
1433 static bool
1434 set_color(int *color, const char *name)
1436         if (map_enum(color, color_map, name))
1437                 return TRUE;
1438         if (!prefixcmp(name, "color"))
1439                 return parse_int(color, name + 5, 0, 255) == OK;
1440         return FALSE;
1443 /* Wants: object fgcolor bgcolor [attribute] */
1444 static int
1445 option_color_command(int argc, const char *argv[])
1447         struct line_info *info;
1449         if (argc != 3 && argc != 4) {
1450                 config_msg = "Wrong number of arguments given to color command";
1451                 return ERR;
1452         }
1454         info = get_line_info(argv[0]);
1455         if (!info) {
1456                 static const struct enum_map obsolete[] = {
1457                         ENUM_MAP("main-delim",  LINE_DELIMITER),
1458                         ENUM_MAP("main-date",   LINE_DATE),
1459                         ENUM_MAP("main-author", LINE_AUTHOR),
1460                 };
1461                 int index;
1463                 if (!map_enum(&index, obsolete, argv[0])) {
1464                         config_msg = "Unknown color name";
1465                         return ERR;
1466                 }
1467                 info = &line_info[index];
1468         }
1470         if (!set_color(&info->fg, argv[1]) ||
1471             !set_color(&info->bg, argv[2])) {
1472                 config_msg = "Unknown color";
1473                 return ERR;
1474         }
1476         if (argc == 4 && !set_attribute(&info->attr, argv[3])) {
1477                 config_msg = "Unknown attribute";
1478                 return ERR;
1479         }
1481         return OK;
1484 static int parse_bool(bool *opt, const char *arg)
1486         *opt = (!strcmp(arg, "1") || !strcmp(arg, "true") || !strcmp(arg, "yes"))
1487                 ? TRUE : FALSE;
1488         return OK;
1491 static int
1492 parse_string(char *opt, const char *arg, size_t optsize)
1494         int arglen = strlen(arg);
1496         switch (arg[0]) {
1497         case '\"':
1498         case '\'':
1499                 if (arglen == 1 || arg[arglen - 1] != arg[0]) {
1500                         config_msg = "Unmatched quotation";
1501                         return ERR;
1502                 }
1503                 arg += 1; arglen -= 2;
1504         default:
1505                 string_ncopy_do(opt, optsize, arg, arglen);
1506                 return OK;
1507         }
1510 /* Wants: name = value */
1511 static int
1512 option_set_command(int argc, const char *argv[])
1514         if (argc != 3) {
1515                 config_msg = "Wrong number of arguments given to set command";
1516                 return ERR;
1517         }
1519         if (strcmp(argv[1], "=")) {
1520                 config_msg = "No value assigned";
1521                 return ERR;
1522         }
1524         if (!strcmp(argv[0], "show-author"))
1525                 return parse_bool(&opt_author, argv[2]);
1527         if (!strcmp(argv[0], "show-date"))
1528                 return parse_bool(&opt_date, argv[2]);
1530         if (!strcmp(argv[0], "show-rev-graph"))
1531                 return parse_bool(&opt_rev_graph, argv[2]);
1533         if (!strcmp(argv[0], "show-refs"))
1534                 return parse_bool(&opt_show_refs, argv[2]);
1536         if (!strcmp(argv[0], "show-line-numbers"))
1537                 return parse_bool(&opt_line_number, argv[2]);
1539         if (!strcmp(argv[0], "line-graphics"))
1540                 return parse_bool(&opt_line_graphics, argv[2]);
1542         if (!strcmp(argv[0], "line-number-interval"))
1543                 return parse_int(&opt_num_interval, argv[2], 1, 1024);
1545         if (!strcmp(argv[0], "author-width"))
1546                 return parse_int(&opt_author_cols, argv[2], 0, 1024);
1548         if (!strcmp(argv[0], "horizontal-scroll"))
1549                 return parse_step(&opt_hscroll, argv[2]);
1551         if (!strcmp(argv[0], "tab-size"))
1552                 return parse_int(&opt_tab_size, argv[2], 1, 1024);
1554         if (!strcmp(argv[0], "commit-encoding"))
1555                 return parse_string(opt_encoding, argv[2], sizeof(opt_encoding));
1557         config_msg = "Unknown variable name";
1558         return ERR;
1561 /* Wants: mode request key */
1562 static int
1563 option_bind_command(int argc, const char *argv[])
1565         enum request request;
1566         int keymap;
1567         int key;
1569         if (argc < 3) {
1570                 config_msg = "Wrong number of arguments given to bind command";
1571                 return ERR;
1572         }
1574         if (set_keymap(&keymap, argv[0]) == ERR) {
1575                 config_msg = "Unknown key map";
1576                 return ERR;
1577         }
1579         key = get_key_value(argv[1]);
1580         if (key == ERR) {
1581                 config_msg = "Unknown key";
1582                 return ERR;
1583         }
1585         request = get_request(argv[2]);
1586         if (request == REQ_NONE) {
1587                 static const struct enum_map obsolete[] = {
1588                         ENUM_MAP("cherry-pick",         REQ_NONE),
1589                         ENUM_MAP("screen-resize",       REQ_NONE),
1590                         ENUM_MAP("tree-parent",         REQ_PARENT),
1591                 };
1592                 int alias;
1594                 if (map_enum(&alias, obsolete, argv[2])) {
1595                         if (alias != REQ_NONE)
1596                                 add_keybinding(keymap, alias, key);
1597                         config_msg = "Obsolete request name";
1598                         return ERR;
1599                 }
1600         }
1601         if (request == REQ_NONE && *argv[2]++ == '!')
1602                 request = add_run_request(keymap, key, argc - 2, argv + 2);
1603         if (request == REQ_NONE) {
1604                 config_msg = "Unknown request name";
1605                 return ERR;
1606         }
1608         add_keybinding(keymap, request, key);
1610         return OK;
1613 static int
1614 set_option(const char *opt, char *value)
1616         const char *argv[SIZEOF_ARG];
1617         int argc = 0;
1619         if (!argv_from_string(argv, &argc, value)) {
1620                 config_msg = "Too many option arguments";
1621                 return ERR;
1622         }
1624         if (!strcmp(opt, "color"))
1625                 return option_color_command(argc, argv);
1627         if (!strcmp(opt, "set"))
1628                 return option_set_command(argc, argv);
1630         if (!strcmp(opt, "bind"))
1631                 return option_bind_command(argc, argv);
1633         config_msg = "Unknown option command";
1634         return ERR;
1637 static int
1638 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1640         int status = OK;
1642         config_lineno++;
1643         config_msg = "Internal error";
1645         /* Check for comment markers, since read_properties() will
1646          * only ensure opt and value are split at first " \t". */
1647         optlen = strcspn(opt, "#");
1648         if (optlen == 0)
1649                 return OK;
1651         if (opt[optlen] != 0) {
1652                 config_msg = "No option value";
1653                 status = ERR;
1655         }  else {
1656                 /* Look for comment endings in the value. */
1657                 size_t len = strcspn(value, "#");
1659                 if (len < valuelen) {
1660                         valuelen = len;
1661                         value[valuelen] = 0;
1662                 }
1664                 status = set_option(opt, value);
1665         }
1667         if (status == ERR) {
1668                 warn("Error on line %d, near '%.*s': %s",
1669                      config_lineno, (int) optlen, opt, config_msg);
1670                 config_errors = TRUE;
1671         }
1673         /* Always keep going if errors are encountered. */
1674         return OK;
1677 static void
1678 load_option_file(const char *path)
1680         struct io io = {};
1682         /* It's OK that the file doesn't exist. */
1683         if (!io_open(&io, path))
1684                 return;
1686         config_lineno = 0;
1687         config_errors = FALSE;
1689         if (io_load(&io, " \t", read_option) == ERR ||
1690             config_errors == TRUE)
1691                 warn("Errors while loading %s.", path);
1694 static int
1695 load_options(void)
1697         const char *home = getenv("HOME");
1698         const char *tigrc_user = getenv("TIGRC_USER");
1699         const char *tigrc_system = getenv("TIGRC_SYSTEM");
1700         char buf[SIZEOF_STR];
1702         add_builtin_run_requests();
1704         if (!tigrc_system)
1705                 tigrc_system = SYSCONFDIR "/tigrc";
1706         load_option_file(tigrc_system);
1708         if (!tigrc_user) {
1709                 if (!home || !string_format(buf, "%s/.tigrc", home))
1710                         return ERR;
1711                 tigrc_user = buf;
1712         }
1713         load_option_file(tigrc_user);
1715         return OK;
1719 /*
1720  * The viewer
1721  */
1723 struct view;
1724 struct view_ops;
1726 /* The display array of active views and the index of the current view. */
1727 static struct view *display[2];
1728 static unsigned int current_view;
1730 #define foreach_displayed_view(view, i) \
1731         for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1733 #define displayed_views()       (display[1] != NULL ? 2 : 1)
1735 /* Current head and commit ID */
1736 static char ref_blob[SIZEOF_REF]        = "";
1737 static char ref_commit[SIZEOF_REF]      = "HEAD";
1738 static char ref_head[SIZEOF_REF]        = "HEAD";
1740 struct view {
1741         const char *name;       /* View name */
1742         const char *cmd_env;    /* Command line set via environment */
1743         const char *id;         /* Points to either of ref_{head,commit,blob} */
1745         struct view_ops *ops;   /* View operations */
1747         enum keymap keymap;     /* What keymap does this view have */
1748         bool git_dir;           /* Whether the view requires a git directory. */
1750         char ref[SIZEOF_REF];   /* Hovered commit reference */
1751         char vid[SIZEOF_REF];   /* View ID. Set to id member when updating. */
1753         int height, width;      /* The width and height of the main window */
1754         WINDOW *win;            /* The main window */
1755         WINDOW *title;          /* The title window living below the main window */
1757         /* Navigation */
1758         unsigned long offset;   /* Offset of the window top */
1759         unsigned long yoffset;  /* Offset from the window side. */
1760         unsigned long lineno;   /* Current line number */
1761         unsigned long p_offset; /* Previous offset of the window top */
1762         unsigned long p_yoffset;/* Previous offset from the window side */
1763         unsigned long p_lineno; /* Previous current line number */
1764         bool p_restore;         /* Should the previous position be restored. */
1766         /* Searching */
1767         char grep[SIZEOF_STR];  /* Search string */
1768         regex_t *regex;         /* Pre-compiled regexp */
1770         /* If non-NULL, points to the view that opened this view. If this view
1771          * is closed tig will switch back to the parent view. */
1772         struct view *parent;
1774         /* Buffering */
1775         size_t lines;           /* Total number of lines */
1776         struct line *line;      /* Line index */
1777         size_t line_alloc;      /* Total number of allocated lines */
1778         unsigned int digits;    /* Number of digits in the lines member. */
1780         /* Drawing */
1781         struct line *curline;   /* Line currently being drawn. */
1782         enum line_type curtype; /* Attribute currently used for drawing. */
1783         unsigned long col;      /* Column when drawing. */
1784         bool has_scrolled;      /* View was scrolled. */
1786         /* Loading */
1787         struct io io;
1788         struct io *pipe;
1789         time_t start_time;
1790         time_t update_secs;
1791 };
1793 struct view_ops {
1794         /* What type of content being displayed. Used in the title bar. */
1795         const char *type;
1796         /* Default command arguments. */
1797         const char **argv;
1798         /* Open and reads in all view content. */
1799         bool (*open)(struct view *view);
1800         /* Read one line; updates view->line. */
1801         bool (*read)(struct view *view, char *data);
1802         /* Draw one line; @lineno must be < view->height. */
1803         bool (*draw)(struct view *view, struct line *line, unsigned int lineno);
1804         /* Depending on view handle a special requests. */
1805         enum request (*request)(struct view *view, enum request request, struct line *line);
1806         /* Search for regexp in a line. */
1807         bool (*grep)(struct view *view, struct line *line);
1808         /* Select line */
1809         void (*select)(struct view *view, struct line *line);
1810 };
1812 static struct view_ops blame_ops;
1813 static struct view_ops blob_ops;
1814 static struct view_ops diff_ops;
1815 static struct view_ops help_ops;
1816 static struct view_ops log_ops;
1817 static struct view_ops main_ops;
1818 static struct view_ops pager_ops;
1819 static struct view_ops stage_ops;
1820 static struct view_ops status_ops;
1821 static struct view_ops tree_ops;
1823 #define VIEW_STR(name, env, ref, ops, map, git) \
1824         { name, #env, ref, ops, map, git }
1826 #define VIEW_(id, name, ops, git, ref) \
1827         VIEW_STR(name, TIG_##id##_CMD, ref, ops, KEYMAP_##id, git)
1830 static struct view views[] = {
1831         VIEW_(MAIN,   "main",   &main_ops,   TRUE,  ref_head),
1832         VIEW_(DIFF,   "diff",   &diff_ops,   TRUE,  ref_commit),
1833         VIEW_(LOG,    "log",    &log_ops,    TRUE,  ref_head),
1834         VIEW_(TREE,   "tree",   &tree_ops,   TRUE,  ref_commit),
1835         VIEW_(BLOB,   "blob",   &blob_ops,   TRUE,  ref_blob),
1836         VIEW_(BLAME,  "blame",  &blame_ops,  TRUE,  ref_commit),
1837         VIEW_(HELP,   "help",   &help_ops,   FALSE, ""),
1838         VIEW_(PAGER,  "pager",  &pager_ops,  FALSE, "stdin"),
1839         VIEW_(STATUS, "status", &status_ops, TRUE,  ""),
1840         VIEW_(STAGE,  "stage",  &stage_ops,  TRUE,  ""),
1841 };
1843 #define VIEW(req)       (&views[(req) - REQ_OFFSET - 1])
1844 #define VIEW_REQ(view)  ((view) - views + REQ_OFFSET + 1)
1846 #define foreach_view(view, i) \
1847         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1849 #define view_is_displayed(view) \
1850         (view == display[0] || view == display[1])
1853 enum line_graphic {
1854         LINE_GRAPHIC_VLINE
1855 };
1857 static chtype line_graphics[] = {
1858         /* LINE_GRAPHIC_VLINE: */ '|'
1859 };
1861 static inline void
1862 set_view_attr(struct view *view, enum line_type type)
1864         if (!view->curline->selected && view->curtype != type) {
1865                 wattrset(view->win, get_line_attr(type));
1866                 wchgat(view->win, -1, 0, type, NULL);
1867                 view->curtype = type;
1868         }
1871 static int
1872 draw_chars(struct view *view, enum line_type type, const char *string,
1873            int max_len, bool use_tilde)
1875         int len = 0;
1876         int col = 0;
1877         int trimmed = FALSE;
1878         size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
1880         if (max_len <= 0)
1881                 return 0;
1883         if (opt_utf8) {
1884                 len = utf8_length(&string, skip, &col, max_len, &trimmed, use_tilde);
1885         } else {
1886                 col = len = strlen(string);
1887                 if (len > max_len) {
1888                         if (use_tilde) {
1889                                 max_len -= 1;
1890                         }
1891                         col = len = max_len;
1892                         trimmed = TRUE;
1893                 }
1894         }
1896         set_view_attr(view, type);
1897         if (len > 0)
1898                 waddnstr(view->win, string, len);
1899         if (trimmed && use_tilde) {
1900                 set_view_attr(view, LINE_DELIMITER);
1901                 waddch(view->win, '~');
1902                 col++;
1903         }
1905         return col;
1908 static int
1909 draw_space(struct view *view, enum line_type type, int max, int spaces)
1911         static char space[] = "                    ";
1912         int col = 0;
1914         spaces = MIN(max, spaces);
1916         while (spaces > 0) {
1917                 int len = MIN(spaces, sizeof(space) - 1);
1919                 col += draw_chars(view, type, space, len, FALSE);
1920                 spaces -= len;
1921         }
1923         return col;
1926 static bool
1927 draw_text(struct view *view, enum line_type type, const char *string, bool trim)
1929         view->col += draw_chars(view, type, string, view->width + view->yoffset - view->col, trim);
1930         return view->width + view->yoffset <= view->col;
1933 static bool
1934 draw_graphic(struct view *view, enum line_type type, chtype graphic[], size_t size)
1936         size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
1937         int max = view->width + view->yoffset - view->col;
1938         int i;
1940         if (max < size)
1941                 size = max;
1943         set_view_attr(view, type);
1944         /* Using waddch() instead of waddnstr() ensures that
1945          * they'll be rendered correctly for the cursor line. */
1946         for (i = skip; i < size; i++)
1947                 waddch(view->win, graphic[i]);
1949         view->col += size;
1950         if (size < max && skip <= size)
1951                 waddch(view->win, ' ');
1952         view->col++;
1954         return view->width + view->yoffset <= view->col;
1957 static bool
1958 draw_field(struct view *view, enum line_type type, const char *text, int len, bool trim)
1960         int max = MIN(view->width + view->yoffset - view->col, len);
1961         int col;
1963         if (text)
1964                 col = draw_chars(view, type, text, max - 1, trim);
1965         else
1966                 col = draw_space(view, type, max - 1, max - 1);
1968         view->col += col;
1969         view->col += draw_space(view, LINE_DEFAULT, max - col, max - col);
1970         return view->width + view->yoffset <= view->col;
1973 static bool
1974 draw_date(struct view *view, time_t *time)
1976         const char *date = mkdate(time);
1978         return draw_field(view, LINE_DATE, date, DATE_COLS, FALSE);
1981 static bool
1982 draw_author(struct view *view, const char *author)
1984         bool trim = opt_author_cols == 0 || opt_author_cols > 5 || !author;
1986         if (!trim) {
1987                 static char initials[10];
1988                 size_t pos;
1990 #define is_initial_sep(c) (isspace(c) || ispunct(c) || (c) == '@')
1992                 memset(initials, 0, sizeof(initials));
1993                 for (pos = 0; *author && pos < opt_author_cols - 1; author++, pos++) {
1994                         while (is_initial_sep(*author))
1995                                 author++;
1996                         strncpy(&initials[pos], author, sizeof(initials) - 1 - pos);
1997                         while (*author && !is_initial_sep(author[1]))
1998                                 author++;
1999                 }
2001                 author = initials;
2002         }
2004         return draw_field(view, LINE_AUTHOR, author, opt_author_cols, trim);
2007 static bool
2008 draw_mode(struct view *view, mode_t mode)
2010         const char *str;
2012         if (S_ISDIR(mode))
2013                 str = "drwxr-xr-x";
2014         else if (S_ISLNK(mode))
2015                 str = "lrwxrwxrwx";
2016         else if (S_ISGITLINK(mode))
2017                 str = "m---------";
2018         else if (S_ISREG(mode) && mode & S_IXUSR)
2019                 str = "-rwxr-xr-x";
2020         else if (S_ISREG(mode))
2021                 str = "-rw-r--r--";
2022         else
2023                 str = "----------";
2025         return draw_field(view, LINE_MODE, str, STRING_SIZE("-rw-r--r-- "), FALSE);
2028 static bool
2029 draw_lineno(struct view *view, unsigned int lineno)
2031         char number[10];
2032         int digits3 = view->digits < 3 ? 3 : view->digits;
2033         int max = MIN(view->width + view->yoffset - view->col, digits3);
2034         char *text = NULL;
2036         lineno += view->offset + 1;
2037         if (lineno == 1 || (lineno % opt_num_interval) == 0) {
2038                 static char fmt[] = "%1ld";
2040                 fmt[1] = '0' + (view->digits <= 9 ? digits3 : 1);
2041                 if (string_format(number, fmt, lineno))
2042                         text = number;
2043         }
2044         if (text)
2045                 view->col += draw_chars(view, LINE_LINE_NUMBER, text, max, TRUE);
2046         else
2047                 view->col += draw_space(view, LINE_LINE_NUMBER, max, digits3);
2048         return draw_graphic(view, LINE_DEFAULT, &line_graphics[LINE_GRAPHIC_VLINE], 1);
2051 static bool
2052 draw_view_line(struct view *view, unsigned int lineno)
2054         struct line *line;
2055         bool selected = (view->offset + lineno == view->lineno);
2057         assert(view_is_displayed(view));
2059         if (view->offset + lineno >= view->lines)
2060                 return FALSE;
2062         line = &view->line[view->offset + lineno];
2064         wmove(view->win, lineno, 0);
2065         if (line->cleareol)
2066                 wclrtoeol(view->win);
2067         view->col = 0;
2068         view->curline = line;
2069         view->curtype = LINE_NONE;
2070         line->selected = FALSE;
2071         line->dirty = line->cleareol = 0;
2073         if (selected) {
2074                 set_view_attr(view, LINE_CURSOR);
2075                 line->selected = TRUE;
2076                 view->ops->select(view, line);
2077         }
2079         return view->ops->draw(view, line, lineno);
2082 static void
2083 redraw_view_dirty(struct view *view)
2085         bool dirty = FALSE;
2086         int lineno;
2088         for (lineno = 0; lineno < view->height; lineno++) {
2089                 if (view->offset + lineno >= view->lines)
2090                         break;
2091                 if (!view->line[view->offset + lineno].dirty)
2092                         continue;
2093                 dirty = TRUE;
2094                 if (!draw_view_line(view, lineno))
2095                         break;
2096         }
2098         if (!dirty)
2099                 return;
2100         wnoutrefresh(view->win);
2103 static void
2104 redraw_view_from(struct view *view, int lineno)
2106         assert(0 <= lineno && lineno < view->height);
2108         for (; lineno < view->height; lineno++) {
2109                 if (!draw_view_line(view, lineno))
2110                         break;
2111         }
2113         wnoutrefresh(view->win);
2116 static void
2117 redraw_view(struct view *view)
2119         werase(view->win);
2120         redraw_view_from(view, 0);
2124 static void
2125 update_view_title(struct view *view)
2127         char buf[SIZEOF_STR];
2128         char state[SIZEOF_STR];
2129         size_t bufpos = 0, statelen = 0;
2131         assert(view_is_displayed(view));
2133         if (view != VIEW(REQ_VIEW_STATUS) && view->lines) {
2134                 unsigned int view_lines = view->offset + view->height;
2135                 unsigned int lines = view->lines
2136                                    ? MIN(view_lines, view->lines) * 100 / view->lines
2137                                    : 0;
2139                 string_format_from(state, &statelen, " - %s %d of %d (%d%%)",
2140                                    view->ops->type,
2141                                    view->lineno + 1,
2142                                    view->lines,
2143                                    lines);
2145         }
2147         if (view->pipe) {
2148                 time_t secs = time(NULL) - view->start_time;
2150                 /* Three git seconds are a long time ... */
2151                 if (secs > 2)
2152                         string_format_from(state, &statelen, " loading %lds", secs);
2153         }
2155         string_format_from(buf, &bufpos, "[%s]", view->name);
2156         if (*view->ref && bufpos < view->width) {
2157                 size_t refsize = strlen(view->ref);
2158                 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
2160                 if (minsize < view->width)
2161                         refsize = view->width - minsize + 7;
2162                 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
2163         }
2165         if (statelen && bufpos < view->width) {
2166                 string_format_from(buf, &bufpos, "%s", state);
2167         }
2169         if (view == display[current_view])
2170                 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
2171         else
2172                 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
2174         mvwaddnstr(view->title, 0, 0, buf, bufpos);
2175         wclrtoeol(view->title);
2176         wnoutrefresh(view->title);
2179 static void
2180 resize_display(void)
2182         int offset, i;
2183         struct view *base = display[0];
2184         struct view *view = display[1] ? display[1] : display[0];
2186         /* Setup window dimensions */
2188         getmaxyx(stdscr, base->height, base->width);
2190         /* Make room for the status window. */
2191         base->height -= 1;
2193         if (view != base) {
2194                 /* Horizontal split. */
2195                 view->width   = base->width;
2196                 view->height  = SCALE_SPLIT_VIEW(base->height);
2197                 base->height -= view->height;
2199                 /* Make room for the title bar. */
2200                 view->height -= 1;
2201         }
2203         /* Make room for the title bar. */
2204         base->height -= 1;
2206         offset = 0;
2208         foreach_displayed_view (view, i) {
2209                 if (!view->win) {
2210                         view->win = newwin(view->height, 0, offset, 0);
2211                         if (!view->win)
2212                                 die("Failed to create %s view", view->name);
2214                         scrollok(view->win, FALSE);
2216                         view->title = newwin(1, 0, offset + view->height, 0);
2217                         if (!view->title)
2218                                 die("Failed to create title window");
2220                 } else {
2221                         wresize(view->win, view->height, view->width);
2222                         mvwin(view->win,   offset, 0);
2223                         mvwin(view->title, offset + view->height, 0);
2224                 }
2226                 offset += view->height + 1;
2227         }
2230 static void
2231 redraw_display(bool clear)
2233         struct view *view;
2234         int i;
2236         foreach_displayed_view (view, i) {
2237                 if (clear)
2238                         wclear(view->win);
2239                 redraw_view(view);
2240                 update_view_title(view);
2241         }
2244 static void
2245 toggle_view_option(bool *option, const char *help)
2247         *option = !*option;
2248         redraw_display(FALSE);
2249         report("%sabling %s", *option ? "En" : "Dis", help);
2252 static void
2253 maximize_view(struct view *view)
2255         memset(display, 0, sizeof(display));
2256         current_view = 0;
2257         display[current_view] = view;
2258         resize_display();
2259         redraw_display(FALSE);
2260         report("");
2264 /*
2265  * Navigation
2266  */
2268 static bool
2269 goto_view_line(struct view *view, unsigned long offset, unsigned long lineno)
2271         if (lineno >= view->lines)
2272                 lineno = view->lines > 0 ? view->lines - 1 : 0;
2274         if (offset > lineno || offset + view->height <= lineno) {
2275                 unsigned long half = view->height / 2;
2277                 if (lineno > half)
2278                         offset = lineno - half;
2279                 else
2280                         offset = 0;
2281         }
2283         if (offset != view->offset || lineno != view->lineno) {
2284                 view->offset = offset;
2285                 view->lineno = lineno;
2286                 return TRUE;
2287         }
2289         return FALSE;
2292 static int
2293 apply_step(double step, int value)
2295         if (step >= 1)
2296                 return (int) step;
2297         value *= step + 0.01;
2298         return value ? value : 1;
2301 /* Scrolling backend */
2302 static void
2303 do_scroll_view(struct view *view, int lines)
2305         bool redraw_current_line = FALSE;
2307         /* The rendering expects the new offset. */
2308         view->offset += lines;
2310         assert(0 <= view->offset && view->offset < view->lines);
2311         assert(lines);
2313         /* Move current line into the view. */
2314         if (view->lineno < view->offset) {
2315                 view->lineno = view->offset;
2316                 redraw_current_line = TRUE;
2317         } else if (view->lineno >= view->offset + view->height) {
2318                 view->lineno = view->offset + view->height - 1;
2319                 redraw_current_line = TRUE;
2320         }
2322         assert(view->offset <= view->lineno && view->lineno < view->lines);
2324         /* Redraw the whole screen if scrolling is pointless. */
2325         if (view->height < ABS(lines)) {
2326                 redraw_view(view);
2328         } else {
2329                 int line = lines > 0 ? view->height - lines : 0;
2330                 int end = line + ABS(lines);
2332                 scrollok(view->win, TRUE);
2333                 wscrl(view->win, lines);
2334                 scrollok(view->win, FALSE);
2336                 while (line < end && draw_view_line(view, line))
2337                         line++;
2339                 if (redraw_current_line)
2340                         draw_view_line(view, view->lineno - view->offset);
2341                 wnoutrefresh(view->win);
2342         }
2344         view->has_scrolled = TRUE;
2345         report("");
2348 /* Scroll frontend */
2349 static void
2350 scroll_view(struct view *view, enum request request)
2352         int lines = 1;
2354         assert(view_is_displayed(view));
2356         switch (request) {
2357         case REQ_SCROLL_LEFT:
2358                 if (view->yoffset == 0) {
2359                         report("Cannot scroll beyond the first column");
2360                         return;
2361                 }
2362                 if (view->yoffset <= apply_step(opt_hscroll, view->width))
2363                         view->yoffset = 0;
2364                 else
2365                         view->yoffset -= apply_step(opt_hscroll, view->width);
2366                 redraw_view_from(view, 0);
2367                 report("");
2368                 return;
2369         case REQ_SCROLL_RIGHT:
2370                 view->yoffset += apply_step(opt_hscroll, view->width);
2371                 redraw_view(view);
2372                 report("");
2373                 return;
2374         case REQ_SCROLL_PAGE_DOWN:
2375                 lines = view->height;
2376         case REQ_SCROLL_LINE_DOWN:
2377                 if (view->offset + lines > view->lines)
2378                         lines = view->lines - view->offset;
2380                 if (lines == 0 || view->offset + view->height >= view->lines) {
2381                         report("Cannot scroll beyond the last line");
2382                         return;
2383                 }
2384                 break;
2386         case REQ_SCROLL_PAGE_UP:
2387                 lines = view->height;
2388         case REQ_SCROLL_LINE_UP:
2389                 if (lines > view->offset)
2390                         lines = view->offset;
2392                 if (lines == 0) {
2393                         report("Cannot scroll beyond the first line");
2394                         return;
2395                 }
2397                 lines = -lines;
2398                 break;
2400         default:
2401                 die("request %d not handled in switch", request);
2402         }
2404         do_scroll_view(view, lines);
2407 /* Cursor moving */
2408 static void
2409 move_view(struct view *view, enum request request)
2411         int scroll_steps = 0;
2412         int steps;
2414         switch (request) {
2415         case REQ_MOVE_FIRST_LINE:
2416                 steps = -view->lineno;
2417                 break;
2419         case REQ_MOVE_LAST_LINE:
2420                 steps = view->lines - view->lineno - 1;
2421                 break;
2423         case REQ_MOVE_PAGE_UP:
2424                 steps = view->height > view->lineno
2425                       ? -view->lineno : -view->height;
2426                 break;
2428         case REQ_MOVE_PAGE_DOWN:
2429                 steps = view->lineno + view->height >= view->lines
2430                       ? view->lines - view->lineno - 1 : view->height;
2431                 break;
2433         case REQ_MOVE_UP:
2434                 steps = -1;
2435                 break;
2437         case REQ_MOVE_DOWN:
2438                 steps = 1;
2439                 break;
2441         default:
2442                 die("request %d not handled in switch", request);
2443         }
2445         if (steps <= 0 && view->lineno == 0) {
2446                 report("Cannot move beyond the first line");
2447                 return;
2449         } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
2450                 report("Cannot move beyond the last line");
2451                 return;
2452         }
2454         /* Move the current line */
2455         view->lineno += steps;
2456         assert(0 <= view->lineno && view->lineno < view->lines);
2458         /* Check whether the view needs to be scrolled */
2459         if (view->lineno < view->offset ||
2460             view->lineno >= view->offset + view->height) {
2461                 scroll_steps = steps;
2462                 if (steps < 0 && -steps > view->offset) {
2463                         scroll_steps = -view->offset;
2465                 } else if (steps > 0) {
2466                         if (view->lineno == view->lines - 1 &&
2467                             view->lines > view->height) {
2468                                 scroll_steps = view->lines - view->offset - 1;
2469                                 if (scroll_steps >= view->height)
2470                                         scroll_steps -= view->height - 1;
2471                         }
2472                 }
2473         }
2475         if (!view_is_displayed(view)) {
2476                 view->offset += scroll_steps;
2477                 assert(0 <= view->offset && view->offset < view->lines);
2478                 view->ops->select(view, &view->line[view->lineno]);
2479                 return;
2480         }
2482         /* Repaint the old "current" line if we be scrolling */
2483         if (ABS(steps) < view->height)
2484                 draw_view_line(view, view->lineno - steps - view->offset);
2486         if (scroll_steps) {
2487                 do_scroll_view(view, scroll_steps);
2488                 return;
2489         }
2491         /* Draw the current line */
2492         draw_view_line(view, view->lineno - view->offset);
2494         wnoutrefresh(view->win);
2495         report("");
2499 /*
2500  * Searching
2501  */
2503 static void search_view(struct view *view, enum request request);
2505 static bool
2506 grep_text(struct view *view, const char *text[])
2508         regmatch_t pmatch;
2509         size_t i;
2511         for (i = 0; text[i]; i++)
2512                 if (*text[i] &&
2513                     regexec(view->regex, text[i], 1, &pmatch, 0) != REG_NOMATCH)
2514                         return TRUE;
2515         return FALSE;
2518 static void
2519 select_view_line(struct view *view, unsigned long lineno)
2521         unsigned long old_lineno = view->lineno;
2522         unsigned long old_offset = view->offset;
2524         if (goto_view_line(view, view->offset, lineno)) {
2525                 if (view_is_displayed(view)) {
2526                         if (old_offset != view->offset) {
2527                                 redraw_view(view);
2528                         } else {
2529                                 draw_view_line(view, old_lineno - view->offset);
2530                                 draw_view_line(view, view->lineno - view->offset);
2531                                 wnoutrefresh(view->win);
2532                         }
2533                 } else {
2534                         view->ops->select(view, &view->line[view->lineno]);
2535                 }
2536         }
2539 static void
2540 find_next(struct view *view, enum request request)
2542         unsigned long lineno = view->lineno;
2543         int direction;
2545         if (!*view->grep) {
2546                 if (!*opt_search)
2547                         report("No previous search");
2548                 else
2549                         search_view(view, request);
2550                 return;
2551         }
2553         switch (request) {
2554         case REQ_SEARCH:
2555         case REQ_FIND_NEXT:
2556                 direction = 1;
2557                 break;
2559         case REQ_SEARCH_BACK:
2560         case REQ_FIND_PREV:
2561                 direction = -1;
2562                 break;
2564         default:
2565                 return;
2566         }
2568         if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
2569                 lineno += direction;
2571         /* Note, lineno is unsigned long so will wrap around in which case it
2572          * will become bigger than view->lines. */
2573         for (; lineno < view->lines; lineno += direction) {
2574                 if (view->ops->grep(view, &view->line[lineno])) {
2575                         select_view_line(view, lineno);
2576                         report("Line %ld matches '%s'", lineno + 1, view->grep);
2577                         return;
2578                 }
2579         }
2581         report("No match found for '%s'", view->grep);
2584 static void
2585 search_view(struct view *view, enum request request)
2587         int regex_err;
2589         if (view->regex) {
2590                 regfree(view->regex);
2591                 *view->grep = 0;
2592         } else {
2593                 view->regex = calloc(1, sizeof(*view->regex));
2594                 if (!view->regex)
2595                         return;
2596         }
2598         regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
2599         if (regex_err != 0) {
2600                 char buf[SIZEOF_STR] = "unknown error";
2602                 regerror(regex_err, view->regex, buf, sizeof(buf));
2603                 report("Search failed: %s", buf);
2604                 return;
2605         }
2607         string_copy(view->grep, opt_search);
2609         find_next(view, request);
2612 /*
2613  * Incremental updating
2614  */
2616 static void
2617 reset_view(struct view *view)
2619         int i;
2621         for (i = 0; i < view->lines; i++)
2622                 free(view->line[i].data);
2623         free(view->line);
2625         view->p_offset = view->offset;
2626         view->p_yoffset = view->yoffset;
2627         view->p_lineno = view->lineno;
2629         view->line = NULL;
2630         view->offset = 0;
2631         view->yoffset = 0;
2632         view->lines  = 0;
2633         view->lineno = 0;
2634         view->line_alloc = 0;
2635         view->vid[0] = 0;
2636         view->update_secs = 0;
2639 static void
2640 free_argv(const char *argv[])
2642         int argc;
2644         for (argc = 0; argv[argc]; argc++)
2645                 free((void *) argv[argc]);
2648 static bool
2649 format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags flags)
2651         char buf[SIZEOF_STR];
2652         int argc;
2653         bool noreplace = flags == FORMAT_NONE;
2655         free_argv(dst_argv);
2657         for (argc = 0; src_argv[argc]; argc++) {
2658                 const char *arg = src_argv[argc];
2659                 size_t bufpos = 0;
2661                 while (arg) {
2662                         char *next = strstr(arg, "%(");
2663                         int len = next - arg;
2664                         const char *value;
2666                         if (!next || noreplace) {
2667                                 if (flags == FORMAT_DASH && !strcmp(arg, "--"))
2668                                         noreplace = TRUE;
2669                                 len = strlen(arg);
2670                                 value = "";
2672                         } else if (!prefixcmp(next, "%(directory)")) {
2673                                 value = opt_path;
2675                         } else if (!prefixcmp(next, "%(file)")) {
2676                                 value = opt_file;
2678                         } else if (!prefixcmp(next, "%(ref)")) {
2679                                 value = *opt_ref ? opt_ref : "HEAD";
2681                         } else if (!prefixcmp(next, "%(head)")) {
2682                                 value = ref_head;
2684                         } else if (!prefixcmp(next, "%(commit)")) {
2685                                 value = ref_commit;
2687                         } else if (!prefixcmp(next, "%(blob)")) {
2688                                 value = ref_blob;
2690                         } else {
2691                                 report("Unknown replacement: `%s`", next);
2692                                 return FALSE;
2693                         }
2695                         if (!string_format_from(buf, &bufpos, "%.*s%s", len, arg, value))
2696                                 return FALSE;
2698                         arg = next && !noreplace ? strchr(next, ')') + 1 : NULL;
2699                 }
2701                 dst_argv[argc] = strdup(buf);
2702                 if (!dst_argv[argc])
2703                         break;
2704         }
2706         dst_argv[argc] = NULL;
2708         return src_argv[argc] == NULL;
2711 static bool
2712 restore_view_position(struct view *view)
2714         if (!view->p_restore || (view->pipe && view->lines <= view->p_lineno))
2715                 return FALSE;
2717         /* Changing the view position cancels the restoring. */
2718         /* FIXME: Changing back to the first line is not detected. */
2719         if (view->offset != 0 || view->lineno != 0) {
2720                 view->p_restore = FALSE;
2721                 return FALSE;
2722         }
2724         if (goto_view_line(view, view->p_offset, view->p_lineno) &&
2725             view_is_displayed(view))
2726                 werase(view->win);
2728         view->yoffset = view->p_yoffset;
2729         view->p_restore = FALSE;
2731         return TRUE;
2734 static void
2735 end_update(struct view *view, bool force)
2737         if (!view->pipe)
2738                 return;
2739         while (!view->ops->read(view, NULL))
2740                 if (!force)
2741                         return;
2742         set_nonblocking_input(FALSE);
2743         if (force)
2744                 kill_io(view->pipe);
2745         done_io(view->pipe);
2746         view->pipe = NULL;
2749 static void
2750 setup_update(struct view *view, const char *vid)
2752         set_nonblocking_input(TRUE);
2753         reset_view(view);
2754         string_copy_rev(view->vid, vid);
2755         view->pipe = &view->io;
2756         view->start_time = time(NULL);
2759 static bool
2760 prepare_update(struct view *view, const char *argv[], const char *dir,
2761                enum format_flags flags)
2763         if (view->pipe)
2764                 end_update(view, TRUE);
2765         return init_io_rd(&view->io, argv, dir, flags);
2768 static bool
2769 prepare_update_file(struct view *view, const char *name)
2771         if (view->pipe)
2772                 end_update(view, TRUE);
2773         return io_open(&view->io, name);
2776 static bool
2777 begin_update(struct view *view, bool refresh)
2779         if (view->pipe)
2780                 end_update(view, TRUE);
2782         if (refresh) {
2783                 if (!start_io(&view->io))
2784                         return FALSE;
2786         } else {
2787                 if (view == VIEW(REQ_VIEW_TREE) && strcmp(view->vid, view->id))
2788                         opt_path[0] = 0;
2790                 if (!run_io_rd(&view->io, view->ops->argv, FORMAT_ALL))
2791                         return FALSE;
2793                 /* Put the current ref_* value to the view title ref
2794                  * member. This is needed by the blob view. Most other
2795                  * views sets it automatically after loading because the
2796                  * first line is a commit line. */
2797                 string_copy_rev(view->ref, view->id);
2798         }
2800         setup_update(view, view->id);
2802         return TRUE;
2805 #define ITEM_CHUNK_SIZE 256
2806 static void *
2807 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2809         size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2810         size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2812         if (mem == NULL || num_chunks != num_chunks_new) {
2813                 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2814                 mem = realloc(mem, *size * item_size);
2815         }
2817         return mem;
2820 static struct line *
2821 realloc_lines(struct view *view, size_t line_size)
2823         size_t alloc = view->line_alloc;
2824         struct line *tmp = realloc_items(view->line, &alloc, line_size,
2825                                          sizeof(*view->line));
2827         if (!tmp)
2828                 return NULL;
2830         view->line = tmp;
2831         view->line_alloc = alloc;
2832         return view->line;
2835 static bool
2836 update_view(struct view *view)
2838         char out_buffer[BUFSIZ * 2];
2839         char *line;
2840         /* Clear the view and redraw everything since the tree sorting
2841          * might have rearranged things. */
2842         bool redraw = view->lines == 0;
2843         bool can_read = TRUE;
2845         if (!view->pipe)
2846                 return TRUE;
2848         if (!io_can_read(view->pipe)) {
2849                 if (view->lines == 0 && view_is_displayed(view)) {
2850                         time_t secs = time(NULL) - view->start_time;
2852                         if (secs > 1 && secs > view->update_secs) {
2853                                 if (view->update_secs == 0)
2854                                         redraw_view(view);
2855                                 update_view_title(view);
2856                                 view->update_secs = secs;
2857                         }
2858                 }
2859                 return TRUE;
2860         }
2862         for (; (line = io_get(view->pipe, '\n', can_read)); can_read = FALSE) {
2863                 if (opt_iconv != ICONV_NONE) {
2864                         ICONV_CONST char *inbuf = line;
2865                         size_t inlen = strlen(line) + 1;
2867                         char *outbuf = out_buffer;
2868                         size_t outlen = sizeof(out_buffer);
2870                         size_t ret;
2872                         ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2873                         if (ret != (size_t) -1)
2874                                 line = out_buffer;
2875                 }
2877                 if (!view->ops->read(view, line)) {
2878                         report("Allocation failure");
2879                         end_update(view, TRUE);
2880                         return FALSE;
2881                 }
2882         }
2884         {
2885                 unsigned long lines = view->lines;
2886                 int digits;
2888                 for (digits = 0; lines; digits++)
2889                         lines /= 10;
2891                 /* Keep the displayed view in sync with line number scaling. */
2892                 if (digits != view->digits) {
2893                         view->digits = digits;
2894                         if (opt_line_number || view == VIEW(REQ_VIEW_BLAME))
2895                                 redraw = TRUE;
2896                 }
2897         }
2899         if (io_error(view->pipe)) {
2900                 report("Failed to read: %s", io_strerror(view->pipe));
2901                 end_update(view, TRUE);
2903         } else if (io_eof(view->pipe)) {
2904                 report("");
2905                 end_update(view, FALSE);
2906         }
2908         if (restore_view_position(view))
2909                 redraw = TRUE;
2911         if (!view_is_displayed(view))
2912                 return TRUE;
2914         if (redraw)
2915                 redraw_view_from(view, 0);
2916         else
2917                 redraw_view_dirty(view);
2919         /* Update the title _after_ the redraw so that if the redraw picks up a
2920          * commit reference in view->ref it'll be available here. */
2921         update_view_title(view);
2922         return TRUE;
2925 static struct line *
2926 add_line_data(struct view *view, void *data, enum line_type type)
2928         struct line *line;
2930         if (!realloc_lines(view, view->lines + 1))
2931                 return NULL;
2933         line = &view->line[view->lines++];
2934         memset(line, 0, sizeof(*line));
2935         line->type = type;
2936         line->data = data;
2937         line->dirty = 1;
2939         return line;
2942 static struct line *
2943 add_line_text(struct view *view, const char *text, enum line_type type)
2945         char *data = text ? strdup(text) : NULL;
2947         return data ? add_line_data(view, data, type) : NULL;
2950 static struct line *
2951 add_line_format(struct view *view, enum line_type type, const char *fmt, ...)
2953         char buf[SIZEOF_STR];
2954         va_list args;
2956         va_start(args, fmt);
2957         if (vsnprintf(buf, sizeof(buf), fmt, args) >= sizeof(buf))
2958                 buf[0] = 0;
2959         va_end(args);
2961         return buf[0] ? add_line_text(view, buf, type) : NULL;
2964 /*
2965  * View opening
2966  */
2968 enum open_flags {
2969         OPEN_DEFAULT = 0,       /* Use default view switching. */
2970         OPEN_SPLIT = 1,         /* Split current view. */
2971         OPEN_RELOAD = 4,        /* Reload view even if it is the current. */
2972         OPEN_REFRESH = 16,      /* Refresh view using previous command. */
2973         OPEN_PREPARED = 32,     /* Open already prepared command. */
2974 };
2976 static void
2977 open_view(struct view *prev, enum request request, enum open_flags flags)
2979         bool split = !!(flags & OPEN_SPLIT);
2980         bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH | OPEN_PREPARED));
2981         bool nomaximize = !!(flags & OPEN_REFRESH);
2982         struct view *view = VIEW(request);
2983         int nviews = displayed_views();
2984         struct view *base_view = display[0];
2986         if (view == prev && nviews == 1 && !reload) {
2987                 report("Already in %s view", view->name);
2988                 return;
2989         }
2991         if (view->git_dir && !opt_git_dir[0]) {
2992                 report("The %s view is disabled in pager view", view->name);
2993                 return;
2994         }
2996         if (split) {
2997                 display[1] = view;
2998                 current_view = 1;
2999         } else if (!nomaximize) {
3000                 /* Maximize the current view. */
3001                 memset(display, 0, sizeof(display));
3002                 current_view = 0;
3003                 display[current_view] = view;
3004         }
3006         /* Resize the view when switching between split- and full-screen,
3007          * or when switching between two different full-screen views. */
3008         if (nviews != displayed_views() ||
3009             (nviews == 1 && base_view != display[0]))
3010                 resize_display();
3012         if (view->ops->open) {
3013                 if (view->pipe)
3014                         end_update(view, TRUE);
3015                 if (!view->ops->open(view)) {
3016                         report("Failed to load %s view", view->name);
3017                         return;
3018                 }
3019                 restore_view_position(view);
3021         } else if ((reload || strcmp(view->vid, view->id)) &&
3022                    !begin_update(view, flags & (OPEN_REFRESH | OPEN_PREPARED))) {
3023                 report("Failed to load %s view", view->name);
3024                 return;
3025         }
3027         if (split && prev->lineno - prev->offset >= prev->height) {
3028                 /* Take the title line into account. */
3029                 int lines = prev->lineno - prev->offset - prev->height + 1;
3031                 /* Scroll the view that was split if the current line is
3032                  * outside the new limited view. */
3033                 do_scroll_view(prev, lines);
3034         }
3036         if (prev && view != prev) {
3037                 if (split) {
3038                         /* "Blur" the previous view. */
3039                         update_view_title(prev);
3040                 }
3042                 view->parent = prev;
3043         }
3045         if (view->pipe && view->lines == 0) {
3046                 /* Clear the old view and let the incremental updating refill
3047                  * the screen. */
3048                 werase(view->win);
3049                 view->p_restore = flags & (OPEN_RELOAD | OPEN_REFRESH);
3050                 report("");
3051         } else if (view_is_displayed(view)) {
3052                 redraw_view(view);
3053                 report("");
3054         }
3057 static void
3058 open_external_viewer(const char *argv[], const char *dir)
3060         def_prog_mode();           /* save current tty modes */
3061         endwin();                  /* restore original tty modes */
3062         run_io_fg(argv, dir);
3063         fprintf(stderr, "Press Enter to continue");
3064         getc(opt_tty);
3065         reset_prog_mode();
3066         redraw_display(TRUE);
3069 static void
3070 open_mergetool(const char *file)
3072         const char *mergetool_argv[] = { "git", "mergetool", file, NULL };
3074         open_external_viewer(mergetool_argv, opt_cdup);
3077 static void
3078 open_editor(bool from_root, const char *file)
3080         const char *editor_argv[] = { "vi", file, NULL };
3081         const char *editor;
3083         editor = getenv("GIT_EDITOR");
3084         if (!editor && *opt_editor)
3085                 editor = opt_editor;
3086         if (!editor)
3087                 editor = getenv("VISUAL");
3088         if (!editor)
3089                 editor = getenv("EDITOR");
3090         if (!editor)
3091                 editor = "vi";
3093         editor_argv[0] = editor;
3094         open_external_viewer(editor_argv, from_root ? opt_cdup : NULL);
3097 static void
3098 open_run_request(enum request request)
3100         struct run_request *req = get_run_request(request);
3101         const char *argv[ARRAY_SIZE(req->argv)] = { NULL };
3103         if (!req) {
3104                 report("Unknown run request");
3105                 return;
3106         }
3108         if (format_argv(argv, req->argv, FORMAT_ALL))
3109                 open_external_viewer(argv, NULL);
3110         free_argv(argv);
3113 /*
3114  * User request switch noodle
3115  */
3117 static int
3118 view_driver(struct view *view, enum request request)
3120         int i;
3122         if (request == REQ_NONE)
3123                 return TRUE;
3125         if (request > REQ_NONE) {
3126                 open_run_request(request);
3127                 /* FIXME: When all views can refresh always do this. */
3128                 if (view == VIEW(REQ_VIEW_STATUS) ||
3129                     view == VIEW(REQ_VIEW_MAIN) ||
3130                     view == VIEW(REQ_VIEW_LOG) ||
3131                     view == VIEW(REQ_VIEW_STAGE))
3132                         request = REQ_REFRESH;
3133                 else
3134                         return TRUE;
3135         }
3137         if (view && view->lines) {
3138                 request = view->ops->request(view, request, &view->line[view->lineno]);
3139                 if (request == REQ_NONE)
3140                         return TRUE;
3141         }
3143         switch (request) {
3144         case REQ_MOVE_UP:
3145         case REQ_MOVE_DOWN:
3146         case REQ_MOVE_PAGE_UP:
3147         case REQ_MOVE_PAGE_DOWN:
3148         case REQ_MOVE_FIRST_LINE:
3149         case REQ_MOVE_LAST_LINE:
3150                 move_view(view, request);
3151                 break;
3153         case REQ_SCROLL_LEFT:
3154         case REQ_SCROLL_RIGHT:
3155         case REQ_SCROLL_LINE_DOWN:
3156         case REQ_SCROLL_LINE_UP:
3157         case REQ_SCROLL_PAGE_DOWN:
3158         case REQ_SCROLL_PAGE_UP:
3159                 scroll_view(view, request);
3160                 break;
3162         case REQ_VIEW_BLAME:
3163                 if (!opt_file[0]) {
3164                         report("No file chosen, press %s to open tree view",
3165                                get_key(REQ_VIEW_TREE));
3166                         break;
3167                 }
3168                 open_view(view, request, OPEN_DEFAULT);
3169                 break;
3171         case REQ_VIEW_BLOB:
3172                 if (!ref_blob[0]) {
3173                         report("No file chosen, press %s to open tree view",
3174                                get_key(REQ_VIEW_TREE));
3175                         break;
3176                 }
3177                 open_view(view, request, OPEN_DEFAULT);
3178                 break;
3180         case REQ_VIEW_PAGER:
3181                 if (!VIEW(REQ_VIEW_PAGER)->pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
3182                         report("No pager content, press %s to run command from prompt",
3183                                get_key(REQ_PROMPT));
3184                         break;
3185                 }
3186                 open_view(view, request, OPEN_DEFAULT);
3187                 break;
3189         case REQ_VIEW_STAGE:
3190                 if (!VIEW(REQ_VIEW_STAGE)->lines) {
3191                         report("No stage content, press %s to open the status view and choose file",
3192                                get_key(REQ_VIEW_STATUS));
3193                         break;
3194                 }
3195                 open_view(view, request, OPEN_DEFAULT);
3196                 break;
3198         case REQ_VIEW_STATUS:
3199                 if (opt_is_inside_work_tree == FALSE) {
3200                         report("The status view requires a working tree");
3201                         break;
3202                 }
3203                 open_view(view, request, OPEN_DEFAULT);
3204                 break;
3206         case REQ_VIEW_MAIN:
3207         case REQ_VIEW_DIFF:
3208         case REQ_VIEW_LOG:
3209         case REQ_VIEW_TREE:
3210         case REQ_VIEW_HELP:
3211                 open_view(view, request, OPEN_DEFAULT);
3212                 break;
3214         case REQ_NEXT:
3215         case REQ_PREVIOUS:
3216                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
3218                 if ((view == VIEW(REQ_VIEW_DIFF) &&
3219                      view->parent == VIEW(REQ_VIEW_MAIN)) ||
3220                    (view == VIEW(REQ_VIEW_DIFF) &&
3221                      view->parent == VIEW(REQ_VIEW_BLAME)) ||
3222                    (view == VIEW(REQ_VIEW_STAGE) &&
3223                      view->parent == VIEW(REQ_VIEW_STATUS)) ||
3224                    (view == VIEW(REQ_VIEW_BLOB) &&
3225                      view->parent == VIEW(REQ_VIEW_TREE))) {
3226                         int line;
3228                         view = view->parent;
3229                         line = view->lineno;
3230                         move_view(view, request);
3231                         if (view_is_displayed(view))
3232                                 update_view_title(view);
3233                         if (line != view->lineno)
3234                                 view->ops->request(view, REQ_ENTER,
3235                                                    &view->line[view->lineno]);
3237                 } else {
3238                         move_view(view, request);
3239                 }
3240                 break;
3242         case REQ_VIEW_NEXT:
3243         {
3244                 int nviews = displayed_views();
3245                 int next_view = (current_view + 1) % nviews;
3247                 if (next_view == current_view) {
3248                         report("Only one view is displayed");
3249                         break;
3250                 }
3252                 current_view = next_view;
3253                 /* Blur out the title of the previous view. */
3254                 update_view_title(view);
3255                 report("");
3256                 break;
3257         }
3258         case REQ_REFRESH:
3259                 report("Refreshing is not yet supported for the %s view", view->name);
3260                 break;
3262         case REQ_MAXIMIZE:
3263                 if (displayed_views() == 2)
3264                         maximize_view(view);
3265                 break;
3267         case REQ_TOGGLE_LINENO:
3268                 toggle_view_option(&opt_line_number, "line numbers");
3269                 break;
3271         case REQ_TOGGLE_DATE:
3272                 toggle_view_option(&opt_date, "date display");
3273                 break;
3275         case REQ_TOGGLE_AUTHOR:
3276                 toggle_view_option(&opt_author, "author display");
3277                 break;
3279         case REQ_TOGGLE_REV_GRAPH:
3280                 toggle_view_option(&opt_rev_graph, "revision graph display");
3281                 break;
3283         case REQ_TOGGLE_REFS:
3284                 toggle_view_option(&opt_show_refs, "reference display");
3285                 break;
3287         case REQ_SEARCH:
3288         case REQ_SEARCH_BACK:
3289                 search_view(view, request);
3290                 break;
3292         case REQ_FIND_NEXT:
3293         case REQ_FIND_PREV:
3294                 find_next(view, request);
3295                 break;
3297         case REQ_STOP_LOADING:
3298                 for (i = 0; i < ARRAY_SIZE(views); i++) {
3299                         view = &views[i];
3300                         if (view->pipe)
3301                                 report("Stopped loading the %s view", view->name),
3302                         end_update(view, TRUE);
3303                 }
3304                 break;
3306         case REQ_SHOW_VERSION:
3307                 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
3308                 return TRUE;
3310         case REQ_SCREEN_REDRAW:
3311                 redraw_display(TRUE);
3312                 break;
3314         case REQ_EDIT:
3315                 report("Nothing to edit");
3316                 break;
3318         case REQ_ENTER:
3319                 report("Nothing to enter");
3320                 break;
3322         case REQ_VIEW_CLOSE:
3323                 /* XXX: Mark closed views by letting view->parent point to the
3324                  * view itself. Parents to closed view should never be
3325                  * followed. */
3326                 if (view->parent &&
3327                     view->parent->parent != view->parent) {
3328                         maximize_view(view->parent);
3329                         view->parent = view;
3330                         break;
3331                 }
3332                 /* Fall-through */
3333         case REQ_QUIT:
3334                 return FALSE;
3336         default:
3337                 report("Unknown key, press 'h' for help");
3338                 return TRUE;
3339         }
3341         return TRUE;
3345 /*
3346  * View backend utilities
3347  */
3349 /* Small author cache to reduce memory consumption. It uses binary
3350  * search to lookup or find place to position new entries. No entries
3351  * are ever freed. */
3352 static const char *
3353 get_author(const char *name)
3355         static const char **authors;
3356         static size_t authors_alloc;
3357         static size_t authors_size;
3358         const char **tmp;
3359         int from = 0, to = authors_size - 1;
3361         while (from <= to) {
3362                 size_t pos = (to + from) / 2;
3363                 int cmp = strcmp(name, authors[pos]);
3365                 if (!cmp)
3366                         return authors[pos];
3368                 if (cmp < 0)
3369                         to = pos - 1;
3370                 else
3371                         from = pos + 1;
3372         }
3374         tmp = realloc_items(authors, &authors_alloc, authors_size + 1, sizeof(*authors));
3375         if (!tmp)
3376                 return NULL;
3377         name = strdup(name);
3378         if (!name)
3379                 return NULL;
3381         authors = tmp;
3382         memmove(authors + from + 1, authors + from, (authors_size - from) * sizeof(*authors));
3383         authors[from] = name;
3384         authors_size++;
3386         return name;
3389 static void
3390 parse_timezone(time_t *time, const char *zone)
3392         long tz;
3394         tz  = ('0' - zone[1]) * 60 * 60 * 10;
3395         tz += ('0' - zone[2]) * 60 * 60;
3396         tz += ('0' - zone[3]) * 60;
3397         tz += ('0' - zone[4]);
3399         if (zone[0] == '-')
3400                 tz = -tz;
3402         *time -= tz;
3405 /* Parse author lines where the name may be empty:
3406  *      author  <email@address.tld> 1138474660 +0100
3407  */
3408 static void
3409 parse_author_line(char *ident, const char **author, time_t *time)
3411         char *nameend = strchr(ident, '<');
3412         char *emailend = strchr(ident, '>');
3414         if (nameend && emailend)
3415                 *nameend = *emailend = 0;
3416         ident = chomp_string(ident);
3417         if (!*ident) {
3418                 if (nameend)
3419                         ident = chomp_string(nameend + 1);
3420                 if (!*ident)
3421                         ident = "Unknown";
3422         }
3424         *author = get_author(ident);
3426         /* Parse epoch and timezone */
3427         if (emailend && emailend[1] == ' ') {
3428                 char *secs = emailend + 2;
3429                 char *zone = strchr(secs, ' ');
3431                 *time = (time_t) atol(secs);
3433                 if (zone && strlen(zone) == STRING_SIZE(" +0700"))
3434                         parse_timezone(time, zone + 1);
3435         }
3438 static enum input_status
3439 select_commit_parent_handler(void *data, char *buf, int c)
3441         size_t parents = *(size_t *) data;
3442         int parent = 0;
3444         if (!isdigit(c))
3445                 return INPUT_SKIP;
3447         if (*buf)
3448                 parent = atoi(buf) * 10;
3449         parent += c - '0';
3451         if (parent > parents)
3452                 return INPUT_SKIP;
3453         return INPUT_OK;
3456 static bool
3457 select_commit_parent(const char *id, char rev[SIZEOF_REV], const char *path)
3459         char buf[SIZEOF_STR * 4];
3460         const char *revlist_argv[] = {
3461                 "git", "rev-list", "-1", "--parents", id, "--", path, NULL
3462         };
3463         int parents;
3465         if (!run_io_buf(revlist_argv, buf, sizeof(buf)) ||
3466             !*chomp_string(buf) ||
3467             (parents = (strlen(buf) / 40) - 1) < 0) {
3468                 report("Failed to get parent information");
3469                 return FALSE;
3471         } else if (parents == 0) {
3472                 if (path)
3473                         report("Path '%s' does not exist in the parent", path);
3474                 else
3475                         report("The selected commit has no parents");
3476                 return FALSE;
3477         }
3479         if (parents > 1) {
3480                 char prompt[SIZEOF_STR];
3481                 char *result;
3483                 if (!string_format(prompt, "Which parent? [1..%d] ", parents))
3484                         return FALSE;
3485                 result = prompt_input(prompt, select_commit_parent_handler, &parents);
3486                 if (!result)
3487                         return FALSE;
3488                 parents = atoi(result);
3489         }
3491         string_copy_rev(rev, &buf[41 * parents]);
3492         return TRUE;
3495 /*
3496  * Pager backend
3497  */
3499 static bool
3500 pager_draw(struct view *view, struct line *line, unsigned int lineno)
3502         char text[SIZEOF_STR];
3504         if (opt_line_number && draw_lineno(view, lineno))
3505                 return TRUE;
3507         string_expand(text, sizeof(text), line->data, opt_tab_size);
3508         draw_text(view, line->type, text, TRUE);
3509         return TRUE;
3512 static bool
3513 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
3515         const char *describe_argv[] = { "git", "describe", commit_id, NULL };
3516         char refbuf[SIZEOF_STR];
3517         char *ref = NULL;
3519         if (run_io_buf(describe_argv, refbuf, sizeof(refbuf)))
3520                 ref = chomp_string(refbuf);
3522         if (!ref || !*ref)
3523                 return TRUE;
3525         /* This is the only fatal call, since it can "corrupt" the buffer. */
3526         if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
3527                 return FALSE;
3529         return TRUE;
3532 static void
3533 add_pager_refs(struct view *view, struct line *line)
3535         char buf[SIZEOF_STR];
3536         char *commit_id = (char *)line->data + STRING_SIZE("commit ");
3537         struct ref **refs;
3538         size_t bufpos = 0, refpos = 0;
3539         const char *sep = "Refs: ";
3540         bool is_tag = FALSE;
3542         assert(line->type == LINE_COMMIT);
3544         refs = get_refs(commit_id);
3545         if (!refs) {
3546                 if (view == VIEW(REQ_VIEW_DIFF))
3547                         goto try_add_describe_ref;
3548                 return;
3549         }
3551         do {
3552                 struct ref *ref = refs[refpos];
3553                 const char *fmt = ref->tag    ? "%s[%s]" :
3554                                   ref->remote ? "%s<%s>" : "%s%s";
3556                 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
3557                         return;
3558                 sep = ", ";
3559                 if (ref->tag)
3560                         is_tag = TRUE;
3561         } while (refs[refpos++]->next);
3563         if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
3564 try_add_describe_ref:
3565                 /* Add <tag>-g<commit_id> "fake" reference. */
3566                 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
3567                         return;
3568         }
3570         if (bufpos == 0)
3571                 return;
3573         add_line_text(view, buf, LINE_PP_REFS);
3576 static bool
3577 pager_read(struct view *view, char *data)
3579         struct line *line;
3581         if (!data)
3582                 return TRUE;
3584         line = add_line_text(view, data, get_line_type(data));
3585         if (!line)
3586                 return FALSE;
3588         if (line->type == LINE_COMMIT &&
3589             (view == VIEW(REQ_VIEW_DIFF) ||
3590              view == VIEW(REQ_VIEW_LOG)))
3591                 add_pager_refs(view, line);
3593         return TRUE;
3596 static enum request
3597 pager_request(struct view *view, enum request request, struct line *line)
3599         int split = 0;
3601         if (request != REQ_ENTER)
3602                 return request;
3604         if (line->type == LINE_COMMIT &&
3605            (view == VIEW(REQ_VIEW_LOG) ||
3606             view == VIEW(REQ_VIEW_PAGER))) {
3607                 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
3608                 split = 1;
3609         }
3611         /* Always scroll the view even if it was split. That way
3612          * you can use Enter to scroll through the log view and
3613          * split open each commit diff. */
3614         scroll_view(view, REQ_SCROLL_LINE_DOWN);
3616         /* FIXME: A minor workaround. Scrolling the view will call report("")
3617          * but if we are scrolling a non-current view this won't properly
3618          * update the view title. */
3619         if (split)
3620                 update_view_title(view);
3622         return REQ_NONE;
3625 static bool
3626 pager_grep(struct view *view, struct line *line)
3628         const char *text[] = { line->data, NULL };
3630         return grep_text(view, text);
3633 static void
3634 pager_select(struct view *view, struct line *line)
3636         if (line->type == LINE_COMMIT) {
3637                 char *text = (char *)line->data + STRING_SIZE("commit ");
3639                 if (view != VIEW(REQ_VIEW_PAGER))
3640                         string_copy_rev(view->ref, text);
3641                 string_copy_rev(ref_commit, text);
3642         }
3645 static struct view_ops pager_ops = {
3646         "line",
3647         NULL,
3648         NULL,
3649         pager_read,
3650         pager_draw,
3651         pager_request,
3652         pager_grep,
3653         pager_select,
3654 };
3656 static const char *log_argv[SIZEOF_ARG] = {
3657         "git", "log", "--no-color", "--cc", "--stat", "-n100", "%(head)", NULL
3658 };
3660 static enum request
3661 log_request(struct view *view, enum request request, struct line *line)
3663         switch (request) {
3664         case REQ_REFRESH:
3665                 load_refs();
3666                 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
3667                 return REQ_NONE;
3668         default:
3669                 return pager_request(view, request, line);
3670         }
3673 static struct view_ops log_ops = {
3674         "line",
3675         log_argv,
3676         NULL,
3677         pager_read,
3678         pager_draw,
3679         log_request,
3680         pager_grep,
3681         pager_select,
3682 };
3684 static const char *diff_argv[SIZEOF_ARG] = {
3685         "git", "show", "--pretty=fuller", "--no-color", "--root",
3686                 "--patch-with-stat", "--find-copies-harder", "-C", "%(commit)", NULL
3687 };
3689 static struct view_ops diff_ops = {
3690         "line",
3691         diff_argv,
3692         NULL,
3693         pager_read,
3694         pager_draw,
3695         pager_request,
3696         pager_grep,
3697         pager_select,
3698 };
3700 /*
3701  * Help backend
3702  */
3704 static bool
3705 help_open(struct view *view)
3707         char buf[SIZEOF_STR];
3708         size_t bufpos;
3709         int i;
3711         if (view->lines > 0)
3712                 return TRUE;
3714         add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3716         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3717                 const char *key;
3719                 if (req_info[i].request == REQ_NONE)
3720                         continue;
3722                 if (!req_info[i].request) {
3723                         add_line_text(view, "", LINE_DEFAULT);
3724                         add_line_text(view, req_info[i].help, LINE_DEFAULT);
3725                         continue;
3726                 }
3728                 key = get_key(req_info[i].request);
3729                 if (!*key)
3730                         key = "(no key defined)";
3732                 for (bufpos = 0; bufpos <= req_info[i].namelen; bufpos++) {
3733                         buf[bufpos] = tolower(req_info[i].name[bufpos]);
3734                         if (buf[bufpos] == '_')
3735                                 buf[bufpos] = '-';
3736                 }
3738                 add_line_format(view, LINE_DEFAULT, "    %-25s %-20s %s",
3739                                 key, buf, req_info[i].help);
3740         }
3742         if (run_requests) {
3743                 add_line_text(view, "", LINE_DEFAULT);
3744                 add_line_text(view, "External commands:", LINE_DEFAULT);
3745         }
3747         for (i = 0; i < run_requests; i++) {
3748                 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3749                 const char *key;
3750                 int argc;
3752                 if (!req)
3753                         continue;
3755                 key = get_key_name(req->key);
3756                 if (!*key)
3757                         key = "(no key defined)";
3759                 for (bufpos = 0, argc = 0; req->argv[argc]; argc++)
3760                         if (!string_format_from(buf, &bufpos, "%s%s",
3761                                                 argc ? " " : "", req->argv[argc]))
3762                                 return REQ_NONE;
3764                 add_line_format(view, LINE_DEFAULT, "    %-10s %-14s `%s`",
3765                                 keymap_table[req->keymap].name, key, buf);
3766         }
3768         return TRUE;
3771 static struct view_ops help_ops = {
3772         "line",
3773         NULL,
3774         help_open,
3775         NULL,
3776         pager_draw,
3777         pager_request,
3778         pager_grep,
3779         pager_select,
3780 };
3783 /*
3784  * Tree backend
3785  */
3787 struct tree_stack_entry {
3788         struct tree_stack_entry *prev;  /* Entry below this in the stack */
3789         unsigned long lineno;           /* Line number to restore */
3790         char *name;                     /* Position of name in opt_path */
3791 };
3793 /* The top of the path stack. */
3794 static struct tree_stack_entry *tree_stack = NULL;
3795 unsigned long tree_lineno = 0;
3797 static void
3798 pop_tree_stack_entry(void)
3800         struct tree_stack_entry *entry = tree_stack;
3802         tree_lineno = entry->lineno;
3803         entry->name[0] = 0;
3804         tree_stack = entry->prev;
3805         free(entry);
3808 static void
3809 push_tree_stack_entry(const char *name, unsigned long lineno)
3811         struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3812         size_t pathlen = strlen(opt_path);
3814         if (!entry)
3815                 return;
3817         entry->prev = tree_stack;
3818         entry->name = opt_path + pathlen;
3819         tree_stack = entry;
3821         if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3822                 pop_tree_stack_entry();
3823                 return;
3824         }
3826         /* Move the current line to the first tree entry. */
3827         tree_lineno = 1;
3828         entry->lineno = lineno;
3831 /* Parse output from git-ls-tree(1):
3832  *
3833  * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3834  */
3836 #define SIZEOF_TREE_ATTR \
3837         STRING_SIZE("100644 blob f931e1d229c3e185caad4449bf5b66ed72462657\t")
3839 #define SIZEOF_TREE_MODE \
3840         STRING_SIZE("100644 ")
3842 #define TREE_ID_OFFSET \
3843         STRING_SIZE("100644 blob ")
3845 struct tree_entry {
3846         char id[SIZEOF_REV];
3847         mode_t mode;
3848         time_t time;                    /* Date from the author ident. */
3849         const char *author;             /* Author of the commit. */
3850         char name[1];
3851 };
3853 static const char *
3854 tree_path(struct line *line)
3856         return ((struct tree_entry *) line->data)->name;
3860 static int
3861 tree_compare_entry(struct line *line1, struct line *line2)
3863         if (line1->type != line2->type)
3864                 return line1->type == LINE_TREE_DIR ? -1 : 1;
3865         return strcmp(tree_path(line1), tree_path(line2));
3868 static struct line *
3869 tree_entry(struct view *view, enum line_type type, const char *path,
3870            const char *mode, const char *id)
3872         struct tree_entry *entry = calloc(1, sizeof(*entry) + strlen(path));
3873         struct line *line = entry ? add_line_data(view, entry, type) : NULL;
3875         if (!entry || !line) {
3876                 free(entry);
3877                 return NULL;
3878         }
3880         strncpy(entry->name, path, strlen(path));
3881         if (mode)
3882                 entry->mode = strtoul(mode, NULL, 8);
3883         if (id)
3884                 string_copy_rev(entry->id, id);
3886         return line;
3889 static bool
3890 tree_read_date(struct view *view, char *text, bool *read_date)
3892         static const char *author_name;
3893         static time_t author_time;
3895         if (!text && *read_date) {
3896                 *read_date = FALSE;
3897                 return TRUE;
3899         } else if (!text) {
3900                 char *path = *opt_path ? opt_path : ".";
3901                 /* Find next entry to process */
3902                 const char *log_file[] = {
3903                         "git", "log", "--no-color", "--pretty=raw",
3904                                 "--cc", "--raw", view->id, "--", path, NULL
3905                 };
3906                 struct io io = {};
3908                 if (!view->lines) {
3909                         tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL);
3910                         report("Tree is empty");
3911                         return TRUE;
3912                 }
3914                 if (!run_io_rd(&io, log_file, FORMAT_NONE)) {
3915                         report("Failed to load tree data");
3916                         return TRUE;
3917                 }
3919                 done_io(view->pipe);
3920                 view->io = io;
3921                 *read_date = TRUE;
3922                 return FALSE;
3924         } else if (*text == 'a' && get_line_type(text) == LINE_AUTHOR) {
3925                 parse_author_line(text + STRING_SIZE("author "),
3926                                   &author_name, &author_time);
3928         } else if (*text == ':') {
3929                 char *pos;
3930                 size_t annotated = 1;
3931                 size_t i;
3933                 pos = strchr(text, '\t');
3934                 if (!pos)
3935                         return TRUE;
3936                 text = pos + 1;
3937                 if (*opt_prefix && !strncmp(text, opt_prefix, strlen(opt_prefix)))
3938                         text += strlen(opt_prefix);
3939                 if (*opt_path && !strncmp(text, opt_path, strlen(opt_path)))
3940                         text += strlen(opt_path);
3941                 pos = strchr(text, '/');
3942                 if (pos)
3943                         *pos = 0;
3945                 for (i = 1; i < view->lines; i++) {
3946                         struct line *line = &view->line[i];
3947                         struct tree_entry *entry = line->data;
3949                         annotated += !!entry->author;
3950                         if (entry->author || strcmp(entry->name, text))
3951                                 continue;
3953                         entry->author = author_name;
3954                         entry->time = author_time;
3955                         line->dirty = 1;
3956                         break;
3957                 }
3959                 if (annotated == view->lines)
3960                         kill_io(view->pipe);
3961         }
3962         return TRUE;
3965 static bool
3966 tree_read(struct view *view, char *text)
3968         static bool read_date = FALSE;
3969         struct tree_entry *data;
3970         struct line *entry, *line;
3971         enum line_type type;
3972         size_t textlen = text ? strlen(text) : 0;
3973         char *path = text + SIZEOF_TREE_ATTR;
3975         if (read_date || !text)
3976                 return tree_read_date(view, text, &read_date);
3978         if (textlen <= SIZEOF_TREE_ATTR)
3979                 return FALSE;
3980         if (view->lines == 0 &&
3981             !tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL))
3982                 return FALSE;
3984         /* Strip the path part ... */
3985         if (*opt_path) {
3986                 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3987                 size_t striplen = strlen(opt_path);
3989                 if (pathlen > striplen)
3990                         memmove(path, path + striplen,
3991                                 pathlen - striplen + 1);
3993                 /* Insert "link" to parent directory. */
3994                 if (view->lines == 1 &&
3995                     !tree_entry(view, LINE_TREE_DIR, "..", "040000", view->ref))
3996                         return FALSE;
3997         }
3999         type = text[SIZEOF_TREE_MODE] == 't' ? LINE_TREE_DIR : LINE_TREE_FILE;
4000         entry = tree_entry(view, type, path, text, text + TREE_ID_OFFSET);
4001         if (!entry)
4002                 return FALSE;
4003         data = entry->data;
4005         /* Skip "Directory ..." and ".." line. */
4006         for (line = &view->line[1 + !!*opt_path]; line < entry; line++) {
4007                 if (tree_compare_entry(line, entry) <= 0)
4008                         continue;
4010                 memmove(line + 1, line, (entry - line) * sizeof(*entry));
4012                 line->data = data;
4013                 line->type = type;
4014                 for (; line <= entry; line++)
4015                         line->dirty = line->cleareol = 1;
4016                 return TRUE;
4017         }
4019         if (tree_lineno > view->lineno) {
4020                 view->lineno = tree_lineno;
4021                 tree_lineno = 0;
4022         }
4024         return TRUE;
4027 static bool
4028 tree_draw(struct view *view, struct line *line, unsigned int lineno)
4030         struct tree_entry *entry = line->data;
4032         if (line->type == LINE_TREE_HEAD) {
4033                 if (draw_text(view, line->type, "Directory path /", TRUE))
4034                         return TRUE;
4035         } else {
4036                 if (draw_mode(view, entry->mode))
4037                         return TRUE;
4039                 if (opt_author && draw_author(view, entry->author))
4040                         return TRUE;
4042                 if (opt_date && draw_date(view, entry->author ? &entry->time : NULL))
4043                         return TRUE;
4044         }
4045         if (draw_text(view, line->type, entry->name, TRUE))
4046                 return TRUE;
4047         return TRUE;
4050 static void
4051 open_blob_editor()
4053         char file[SIZEOF_STR] = "/tmp/tigblob.XXXXXX";
4054         int fd = mkstemp(file);
4056         if (fd == -1)
4057                 report("Failed to create temporary file");
4058         else if (!run_io_append(blob_ops.argv, FORMAT_ALL, fd))
4059                 report("Failed to save blob data to file");
4060         else
4061                 open_editor(FALSE, file);
4062         if (fd != -1)
4063                 unlink(file);
4066 static enum request
4067 tree_request(struct view *view, enum request request, struct line *line)
4069         enum open_flags flags;
4071         switch (request) {
4072         case REQ_VIEW_BLAME:
4073                 if (line->type != LINE_TREE_FILE) {
4074                         report("Blame only supported for files");
4075                         return REQ_NONE;
4076                 }
4078                 string_copy(opt_ref, view->vid);
4079                 return request;
4081         case REQ_EDIT:
4082                 if (line->type != LINE_TREE_FILE) {
4083                         report("Edit only supported for files");
4084                 } else if (!is_head_commit(view->vid)) {
4085                         open_blob_editor();
4086                 } else {
4087                         open_editor(TRUE, opt_file);
4088                 }
4089                 return REQ_NONE;
4091         case REQ_PARENT:
4092                 if (!*opt_path) {
4093                         /* quit view if at top of tree */
4094                         return REQ_VIEW_CLOSE;
4095                 }
4096                 /* fake 'cd  ..' */
4097                 line = &view->line[1];
4098                 break;
4100         case REQ_ENTER:
4101                 break;
4103         default:
4104                 return request;
4105         }
4107         /* Cleanup the stack if the tree view is at a different tree. */
4108         while (!*opt_path && tree_stack)
4109                 pop_tree_stack_entry();
4111         switch (line->type) {
4112         case LINE_TREE_DIR:
4113                 /* Depending on whether it is a subdirectory or parent link
4114                  * mangle the path buffer. */
4115                 if (line == &view->line[1] && *opt_path) {
4116                         pop_tree_stack_entry();
4118                 } else {
4119                         const char *basename = tree_path(line);
4121                         push_tree_stack_entry(basename, view->lineno);
4122                 }
4124                 /* Trees and subtrees share the same ID, so they are not not
4125                  * unique like blobs. */
4126                 flags = OPEN_RELOAD;
4127                 request = REQ_VIEW_TREE;
4128                 break;
4130         case LINE_TREE_FILE:
4131                 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4132                 request = REQ_VIEW_BLOB;
4133                 break;
4135         default:
4136                 return REQ_NONE;
4137         }
4139         open_view(view, request, flags);
4140         if (request == REQ_VIEW_TREE)
4141                 view->lineno = tree_lineno;
4143         return REQ_NONE;
4146 static bool
4147 tree_grep(struct view *view, struct line *line)
4149         struct tree_entry *entry = line->data;
4150         const char *text[] = {
4151                 entry->name,
4152                 opt_author ? entry->author : "",
4153                 opt_date ? mkdate(&entry->time) : "",
4154                 NULL
4155         };
4157         return grep_text(view, text);
4160 static void
4161 tree_select(struct view *view, struct line *line)
4163         struct tree_entry *entry = line->data;
4165         if (line->type == LINE_TREE_FILE) {
4166                 string_copy_rev(ref_blob, entry->id);
4167                 string_format(opt_file, "%s%s", opt_path, tree_path(line));
4169         } else if (line->type != LINE_TREE_DIR) {
4170                 return;
4171         }
4173         string_copy_rev(view->ref, entry->id);
4176 static const char *tree_argv[SIZEOF_ARG] = {
4177         "git", "ls-tree", "%(commit)", "%(directory)", NULL
4178 };
4180 static struct view_ops tree_ops = {
4181         "file",
4182         tree_argv,
4183         NULL,
4184         tree_read,
4185         tree_draw,
4186         tree_request,
4187         tree_grep,
4188         tree_select,
4189 };
4191 static bool
4192 blob_read(struct view *view, char *line)
4194         if (!line)
4195                 return TRUE;
4196         return add_line_text(view, line, LINE_DEFAULT) != NULL;
4199 static enum request
4200 blob_request(struct view *view, enum request request, struct line *line)
4202         switch (request) {
4203         case REQ_EDIT:
4204                 open_blob_editor();
4205                 return REQ_NONE;
4206         default:
4207                 return pager_request(view, request, line);
4208         }
4211 static const char *blob_argv[SIZEOF_ARG] = {
4212         "git", "cat-file", "blob", "%(blob)", NULL
4213 };
4215 static struct view_ops blob_ops = {
4216         "line",
4217         blob_argv,
4218         NULL,
4219         blob_read,
4220         pager_draw,
4221         blob_request,
4222         pager_grep,
4223         pager_select,
4224 };
4226 /*
4227  * Blame backend
4228  *
4229  * Loading the blame view is a two phase job:
4230  *
4231  *  1. File content is read either using opt_file from the
4232  *     filesystem or using git-cat-file.
4233  *  2. Then blame information is incrementally added by
4234  *     reading output from git-blame.
4235  */
4237 static const char *blame_head_argv[] = {
4238         "git", "blame", "--incremental", "--", "%(file)", NULL
4239 };
4241 static const char *blame_ref_argv[] = {
4242         "git", "blame", "--incremental", "%(ref)", "--", "%(file)", NULL
4243 };
4245 static const char *blame_cat_file_argv[] = {
4246         "git", "cat-file", "blob", "%(ref):%(file)", NULL
4247 };
4249 struct blame_commit {
4250         char id[SIZEOF_REV];            /* SHA1 ID. */
4251         char title[128];                /* First line of the commit message. */
4252         const char *author;             /* Author of the commit. */
4253         time_t time;                    /* Date from the author ident. */
4254         char filename[128];             /* Name of file. */
4255         bool has_previous;              /* Was a "previous" line detected. */
4256 };
4258 struct blame {
4259         struct blame_commit *commit;
4260         unsigned long lineno;
4261         char text[1];
4262 };
4264 static bool
4265 blame_open(struct view *view)
4267         if (*opt_ref || !io_open(&view->io, opt_file)) {
4268                 if (!run_io_rd(&view->io, blame_cat_file_argv, FORMAT_ALL))
4269                         return FALSE;
4270         }
4272         setup_update(view, opt_file);
4273         string_format(view->ref, "%s ...", opt_file);
4275         return TRUE;
4278 static struct blame_commit *
4279 get_blame_commit(struct view *view, const char *id)
4281         size_t i;
4283         for (i = 0; i < view->lines; i++) {
4284                 struct blame *blame = view->line[i].data;
4286                 if (!blame->commit)
4287                         continue;
4289                 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
4290                         return blame->commit;
4291         }
4293         {
4294                 struct blame_commit *commit = calloc(1, sizeof(*commit));
4296                 if (commit)
4297                         string_ncopy(commit->id, id, SIZEOF_REV);
4298                 return commit;
4299         }
4302 static bool
4303 parse_number(const char **posref, size_t *number, size_t min, size_t max)
4305         const char *pos = *posref;
4307         *posref = NULL;
4308         pos = strchr(pos + 1, ' ');
4309         if (!pos || !isdigit(pos[1]))
4310                 return FALSE;
4311         *number = atoi(pos + 1);
4312         if (*number < min || *number > max)
4313                 return FALSE;
4315         *posref = pos;
4316         return TRUE;
4319 static struct blame_commit *
4320 parse_blame_commit(struct view *view, const char *text, int *blamed)
4322         struct blame_commit *commit;
4323         struct blame *blame;
4324         const char *pos = text + SIZEOF_REV - 2;
4325         size_t orig_lineno = 0;
4326         size_t lineno;
4327         size_t group;
4329         if (strlen(text) <= SIZEOF_REV || pos[1] != ' ')
4330                 return NULL;
4332         if (!parse_number(&pos, &orig_lineno, 1, 9999999) ||
4333             !parse_number(&pos, &lineno, 1, view->lines) ||
4334             !parse_number(&pos, &group, 1, view->lines - lineno + 1))
4335                 return NULL;
4337         commit = get_blame_commit(view, text);
4338         if (!commit)
4339                 return NULL;
4341         *blamed += group;
4342         while (group--) {
4343                 struct line *line = &view->line[lineno + group - 1];
4345                 blame = line->data;
4346                 blame->commit = commit;
4347                 blame->lineno = orig_lineno + group - 1;
4348                 line->dirty = 1;
4349         }
4351         return commit;
4354 static bool
4355 blame_read_file(struct view *view, const char *line, bool *read_file)
4357         if (!line) {
4358                 const char **argv = *opt_ref ? blame_ref_argv : blame_head_argv;
4359                 struct io io = {};
4361                 if (view->lines == 0 && !view->parent)
4362                         die("No blame exist for %s", view->vid);
4364                 if (view->lines == 0 || !run_io_rd(&io, argv, FORMAT_ALL)) {
4365                         report("Failed to load blame data");
4366                         return TRUE;
4367                 }
4369                 done_io(view->pipe);
4370                 view->io = io;
4371                 *read_file = FALSE;
4372                 return FALSE;
4374         } else {
4375                 size_t linelen = strlen(line);
4376                 struct blame *blame = malloc(sizeof(*blame) + linelen);
4378                 if (!blame)
4379                         return FALSE;
4381                 blame->commit = NULL;
4382                 strncpy(blame->text, line, linelen);
4383                 blame->text[linelen] = 0;
4384                 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
4385         }
4388 static bool
4389 match_blame_header(const char *name, char **line)
4391         size_t namelen = strlen(name);
4392         bool matched = !strncmp(name, *line, namelen);
4394         if (matched)
4395                 *line += namelen;
4397         return matched;
4400 static bool
4401 blame_read(struct view *view, char *line)
4403         static struct blame_commit *commit = NULL;
4404         static int blamed = 0;
4405         static bool read_file = TRUE;
4407         if (read_file)
4408                 return blame_read_file(view, line, &read_file);
4410         if (!line) {
4411                 /* Reset all! */
4412                 commit = NULL;
4413                 blamed = 0;
4414                 read_file = TRUE;
4415                 string_format(view->ref, "%s", view->vid);
4416                 if (view_is_displayed(view)) {
4417                         update_view_title(view);
4418                         redraw_view_from(view, 0);
4419                 }
4420                 return TRUE;
4421         }
4423         if (!commit) {
4424                 commit = parse_blame_commit(view, line, &blamed);
4425                 string_format(view->ref, "%s %2d%%", view->vid,
4426                               view->lines ? blamed * 100 / view->lines : 0);
4428         } else if (match_blame_header("author ", &line)) {
4429                 commit->author = get_author(line);
4431         } else if (match_blame_header("author-time ", &line)) {
4432                 commit->time = (time_t) atol(line);
4434         } else if (match_blame_header("author-tz ", &line)) {
4435                 parse_timezone(&commit->time, line);
4437         } else if (match_blame_header("summary ", &line)) {
4438                 string_ncopy(commit->title, line, strlen(line));
4440         } else if (match_blame_header("previous ", &line)) {
4441                 commit->has_previous = TRUE;
4443         } else if (match_blame_header("filename ", &line)) {
4444                 string_ncopy(commit->filename, line, strlen(line));
4445                 commit = NULL;
4446         }
4448         return TRUE;
4451 static bool
4452 blame_draw(struct view *view, struct line *line, unsigned int lineno)
4454         struct blame *blame = line->data;
4455         time_t *time = NULL;
4456         const char *id = NULL, *author = NULL;
4457         char text[SIZEOF_STR];
4459         if (blame->commit && *blame->commit->filename) {
4460                 id = blame->commit->id;
4461                 author = blame->commit->author;
4462                 time = &blame->commit->time;
4463         }
4465         if (opt_date && draw_date(view, time))
4466                 return TRUE;
4468         if (opt_author && draw_author(view, author))
4469                 return TRUE;
4471         if (draw_field(view, LINE_BLAME_ID, id, ID_COLS, FALSE))
4472                 return TRUE;
4474         if (draw_lineno(view, lineno))
4475                 return TRUE;
4477         string_expand(text, sizeof(text), blame->text, opt_tab_size);
4478         draw_text(view, LINE_DEFAULT, text, TRUE);
4479         return TRUE;
4482 static bool
4483 check_blame_commit(struct blame *blame, bool check_null_id)
4485         if (!blame->commit)
4486                 report("Commit data not loaded yet");
4487         else if (check_null_id && !strcmp(blame->commit->id, NULL_ID))
4488                 report("No commit exist for the selected line");
4489         else
4490                 return TRUE;
4491         return FALSE;
4494 static void
4495 setup_blame_parent_line(struct view *view, struct blame *blame)
4497         const char *diff_tree_argv[] = {
4498                 "git", "diff-tree", "-U0", blame->commit->id,
4499                         "--", blame->commit->filename, NULL
4500         };
4501         struct io io = {};
4502         int parent_lineno = -1;
4503         int blamed_lineno = -1;
4504         char *line;
4506         if (!run_io(&io, diff_tree_argv, NULL, IO_RD))
4507                 return;
4509         while ((line = io_get(&io, '\n', TRUE))) {
4510                 if (*line == '@') {
4511                         char *pos = strchr(line, '+');
4513                         parent_lineno = atoi(line + 4);
4514                         if (pos)
4515                                 blamed_lineno = atoi(pos + 1);
4517                 } else if (*line == '+' && parent_lineno != -1) {
4518                         if (blame->lineno == blamed_lineno - 1 &&
4519                             !strcmp(blame->text, line + 1)) {
4520                                 view->lineno = parent_lineno ? parent_lineno - 1 : 0;
4521                                 break;
4522                         }
4523                         blamed_lineno++;
4524                 }
4525         }
4527         done_io(&io);
4530 static enum request
4531 blame_request(struct view *view, enum request request, struct line *line)
4533         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4534         struct blame *blame = line->data;
4536         switch (request) {
4537         case REQ_VIEW_BLAME:
4538                 if (check_blame_commit(blame, TRUE)) {
4539                         string_copy(opt_ref, blame->commit->id);
4540                         string_copy(opt_file, blame->commit->filename);
4541                         if (blame->lineno)
4542                                 view->lineno = blame->lineno;
4543                         open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4544                 }
4545                 break;
4547         case REQ_PARENT:
4548                 if (check_blame_commit(blame, TRUE) &&
4549                     select_commit_parent(blame->commit->id, opt_ref,
4550                                          blame->commit->filename)) {
4551                         string_copy(opt_file, blame->commit->filename);
4552                         setup_blame_parent_line(view, blame);
4553                         open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4554                 }
4555                 break;
4557         case REQ_ENTER:
4558                 if (!check_blame_commit(blame, FALSE))
4559                         break;
4561                 if (view_is_displayed(VIEW(REQ_VIEW_DIFF)) &&
4562                     !strcmp(blame->commit->id, VIEW(REQ_VIEW_DIFF)->ref))
4563                         break;
4565                 if (!strcmp(blame->commit->id, NULL_ID)) {
4566                         struct view *diff = VIEW(REQ_VIEW_DIFF);
4567                         const char *diff_index_argv[] = {
4568                                 "git", "diff-index", "--root", "--patch-with-stat",
4569                                         "-C", "-M", "HEAD", "--", view->vid, NULL
4570                         };
4572                         if (!blame->commit->has_previous) {
4573                                 diff_index_argv[1] = "diff";
4574                                 diff_index_argv[2] = "--no-color";
4575                                 diff_index_argv[6] = "--";
4576                                 diff_index_argv[7] = "/dev/null";
4577                         }
4579                         if (!prepare_update(diff, diff_index_argv, NULL, FORMAT_DASH)) {
4580                                 report("Failed to allocate diff command");
4581                                 break;
4582                         }
4583                         flags |= OPEN_PREPARED;
4584                 }
4586                 open_view(view, REQ_VIEW_DIFF, flags);
4587                 if (VIEW(REQ_VIEW_DIFF)->pipe && !strcmp(blame->commit->id, NULL_ID))
4588                         string_copy_rev(VIEW(REQ_VIEW_DIFF)->ref, NULL_ID);
4589                 break;
4591         default:
4592                 return request;
4593         }
4595         return REQ_NONE;
4598 static bool
4599 blame_grep(struct view *view, struct line *line)
4601         struct blame *blame = line->data;
4602         struct blame_commit *commit = blame->commit;
4603         const char *text[] = {
4604                 blame->text,
4605                 commit ? commit->title : "",
4606                 commit ? commit->id : "",
4607                 commit && opt_author ? commit->author : "",
4608                 commit && opt_date ? mkdate(&commit->time) : "",
4609         };
4611         return grep_text(view, text);
4614 static void
4615 blame_select(struct view *view, struct line *line)
4617         struct blame *blame = line->data;
4618         struct blame_commit *commit = blame->commit;
4620         if (!commit)
4621                 return;
4623         if (!strcmp(commit->id, NULL_ID))
4624                 string_ncopy(ref_commit, "HEAD", 4);
4625         else
4626                 string_copy_rev(ref_commit, commit->id);
4629 static struct view_ops blame_ops = {
4630         "line",
4631         NULL,
4632         blame_open,
4633         blame_read,
4634         blame_draw,
4635         blame_request,
4636         blame_grep,
4637         blame_select,
4638 };
4640 /*
4641  * Status backend
4642  */
4644 struct status {
4645         char status;
4646         struct {
4647                 mode_t mode;
4648                 char rev[SIZEOF_REV];
4649                 char name[SIZEOF_STR];
4650         } old;
4651         struct {
4652                 mode_t mode;
4653                 char rev[SIZEOF_REV];
4654                 char name[SIZEOF_STR];
4655         } new;
4656 };
4658 static char status_onbranch[SIZEOF_STR];
4659 static struct status stage_status;
4660 static enum line_type stage_line_type;
4661 static size_t stage_chunks;
4662 static int *stage_chunk;
4664 /* This should work even for the "On branch" line. */
4665 static inline bool
4666 status_has_none(struct view *view, struct line *line)
4668         return line < view->line + view->lines && !line[1].data;
4671 /* Get fields from the diff line:
4672  * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
4673  */
4674 static inline bool
4675 status_get_diff(struct status *file, const char *buf, size_t bufsize)
4677         const char *old_mode = buf +  1;
4678         const char *new_mode = buf +  8;
4679         const char *old_rev  = buf + 15;
4680         const char *new_rev  = buf + 56;
4681         const char *status   = buf + 97;
4683         if (bufsize < 98 ||
4684             old_mode[-1] != ':' ||
4685             new_mode[-1] != ' ' ||
4686             old_rev[-1]  != ' ' ||
4687             new_rev[-1]  != ' ' ||
4688             status[-1]   != ' ')
4689                 return FALSE;
4691         file->status = *status;
4693         string_copy_rev(file->old.rev, old_rev);
4694         string_copy_rev(file->new.rev, new_rev);
4696         file->old.mode = strtoul(old_mode, NULL, 8);
4697         file->new.mode = strtoul(new_mode, NULL, 8);
4699         file->old.name[0] = file->new.name[0] = 0;
4701         return TRUE;
4704 static bool
4705 status_run(struct view *view, const char *argv[], char status, enum line_type type)
4707         struct status *unmerged = NULL;
4708         char *buf;
4709         struct io io = {};
4711         if (!run_io(&io, argv, NULL, IO_RD))
4712                 return FALSE;
4714         add_line_data(view, NULL, type);
4716         while ((buf = io_get(&io, 0, TRUE))) {
4717                 struct status *file = unmerged;
4719                 if (!file) {
4720                         file = calloc(1, sizeof(*file));
4721                         if (!file || !add_line_data(view, file, type))
4722                                 goto error_out;
4723                 }
4725                 /* Parse diff info part. */
4726                 if (status) {
4727                         file->status = status;
4728                         if (status == 'A')
4729                                 string_copy(file->old.rev, NULL_ID);
4731                 } else if (!file->status || file == unmerged) {
4732                         if (!status_get_diff(file, buf, strlen(buf)))
4733                                 goto error_out;
4735                         buf = io_get(&io, 0, TRUE);
4736                         if (!buf)
4737                                 break;
4739                         /* Collapse all modified entries that follow an
4740                          * associated unmerged entry. */
4741                         if (unmerged == file) {
4742                                 unmerged->status = 'U';
4743                                 unmerged = NULL;
4744                         } else if (file->status == 'U') {
4745                                 unmerged = file;
4746                         }
4747                 }
4749                 /* Grab the old name for rename/copy. */
4750                 if (!*file->old.name &&
4751                     (file->status == 'R' || file->status == 'C')) {
4752                         string_ncopy(file->old.name, buf, strlen(buf));
4754                         buf = io_get(&io, 0, TRUE);
4755                         if (!buf)
4756                                 break;
4757                 }
4759                 /* git-ls-files just delivers a NUL separated list of
4760                  * file names similar to the second half of the
4761                  * git-diff-* output. */
4762                 string_ncopy(file->new.name, buf, strlen(buf));
4763                 if (!*file->old.name)
4764                         string_copy(file->old.name, file->new.name);
4765                 file = NULL;
4766         }
4768         if (io_error(&io)) {
4769 error_out:
4770                 done_io(&io);
4771                 return FALSE;
4772         }
4774         if (!view->line[view->lines - 1].data)
4775                 add_line_data(view, NULL, LINE_STAT_NONE);
4777         done_io(&io);
4778         return TRUE;
4781 /* Don't show unmerged entries in the staged section. */
4782 static const char *status_diff_index_argv[] = {
4783         "git", "diff-index", "-z", "--diff-filter=ACDMRTXB",
4784                              "--cached", "-M", "HEAD", NULL
4785 };
4787 static const char *status_diff_files_argv[] = {
4788         "git", "diff-files", "-z", NULL
4789 };
4791 static const char *status_list_other_argv[] = {
4792         "git", "ls-files", "-z", "--others", "--exclude-standard", NULL
4793 };
4795 static const char *status_list_no_head_argv[] = {
4796         "git", "ls-files", "-z", "--cached", "--exclude-standard", NULL
4797 };
4799 static const char *update_index_argv[] = {
4800         "git", "update-index", "-q", "--unmerged", "--refresh", NULL
4801 };
4803 /* Restore the previous line number to stay in the context or select a
4804  * line with something that can be updated. */
4805 static void
4806 status_restore(struct view *view)
4808         if (view->p_lineno >= view->lines)
4809                 view->p_lineno = view->lines - 1;
4810         while (view->p_lineno < view->lines && !view->line[view->p_lineno].data)
4811                 view->p_lineno++;
4812         while (view->p_lineno > 0 && !view->line[view->p_lineno].data)
4813                 view->p_lineno--;
4815         /* If the above fails, always skip the "On branch" line. */
4816         if (view->p_lineno < view->lines)
4817                 view->lineno = view->p_lineno;
4818         else
4819                 view->lineno = 1;
4821         if (view->lineno < view->offset)
4822                 view->offset = view->lineno;
4823         else if (view->offset + view->height <= view->lineno)
4824                 view->offset = view->lineno - view->height + 1;
4826         view->p_restore = FALSE;
4829 static void
4830 status_update_onbranch(void)
4832         static const char *paths[][2] = {
4833                 { "rebase-apply/rebasing",      "Rebasing" },
4834                 { "rebase-apply/applying",      "Applying mailbox" },
4835                 { "rebase-apply/",              "Rebasing mailbox" },
4836                 { "rebase-merge/interactive",   "Interactive rebase" },
4837                 { "rebase-merge/",              "Rebase merge" },
4838                 { "MERGE_HEAD",                 "Merging" },
4839                 { "BISECT_LOG",                 "Bisecting" },
4840                 { "HEAD",                       "On branch" },
4841         };
4842         char buf[SIZEOF_STR];
4843         struct stat stat;
4844         int i;
4846         if (is_initial_commit()) {
4847                 string_copy(status_onbranch, "Initial commit");
4848                 return;
4849         }
4851         for (i = 0; i < ARRAY_SIZE(paths); i++) {
4852                 char *head = opt_head;
4854                 if (!string_format(buf, "%s/%s", opt_git_dir, paths[i][0]) ||
4855                     lstat(buf, &stat) < 0)
4856                         continue;
4858                 if (!*opt_head) {
4859                         struct io io = {};
4861                         if (string_format(buf, "%s/rebase-merge/head-name", opt_git_dir) &&
4862                             io_open(&io, buf) &&
4863                             io_read_buf(&io, buf, sizeof(buf))) {
4864                                 head = chomp_string(buf);
4865                                 if (!prefixcmp(head, "refs/heads/"))
4866                                         head += STRING_SIZE("refs/heads/");
4867                         }
4868                 }
4870                 if (!string_format(status_onbranch, "%s %s", paths[i][1], head))
4871                         string_copy(status_onbranch, opt_head);
4872                 return;
4873         }
4875         string_copy(status_onbranch, "Not currently on any branch");
4878 /* First parse staged info using git-diff-index(1), then parse unstaged
4879  * info using git-diff-files(1), and finally untracked files using
4880  * git-ls-files(1). */
4881 static bool
4882 status_open(struct view *view)
4884         reset_view(view);
4886         add_line_data(view, NULL, LINE_STAT_HEAD);
4887         status_update_onbranch();
4889         run_io_bg(update_index_argv);
4891         if (is_initial_commit()) {
4892                 if (!status_run(view, status_list_no_head_argv, 'A', LINE_STAT_STAGED))
4893                         return FALSE;
4894         } else if (!status_run(view, status_diff_index_argv, 0, LINE_STAT_STAGED)) {
4895                 return FALSE;
4896         }
4898         if (!status_run(view, status_diff_files_argv, 0, LINE_STAT_UNSTAGED) ||
4899             !status_run(view, status_list_other_argv, '?', LINE_STAT_UNTRACKED))
4900                 return FALSE;
4902         /* Restore the exact position or use the specialized restore
4903          * mode? */
4904         if (!view->p_restore)
4905                 status_restore(view);
4906         return TRUE;
4909 static bool
4910 status_draw(struct view *view, struct line *line, unsigned int lineno)
4912         struct status *status = line->data;
4913         enum line_type type;
4914         const char *text;
4916         if (!status) {
4917                 switch (line->type) {
4918                 case LINE_STAT_STAGED:
4919                         type = LINE_STAT_SECTION;
4920                         text = "Changes to be committed:";
4921                         break;
4923                 case LINE_STAT_UNSTAGED:
4924                         type = LINE_STAT_SECTION;
4925                         text = "Changed but not updated:";
4926                         break;
4928                 case LINE_STAT_UNTRACKED:
4929                         type = LINE_STAT_SECTION;
4930                         text = "Untracked files:";
4931                         break;
4933                 case LINE_STAT_NONE:
4934                         type = LINE_DEFAULT;
4935                         text = "  (no files)";
4936                         break;
4938                 case LINE_STAT_HEAD:
4939                         type = LINE_STAT_HEAD;
4940                         text = status_onbranch;
4941                         break;
4943                 default:
4944                         return FALSE;
4945                 }
4946         } else {
4947                 static char buf[] = { '?', ' ', ' ', ' ', 0 };
4949                 buf[0] = status->status;
4950                 if (draw_text(view, line->type, buf, TRUE))
4951                         return TRUE;
4952                 type = LINE_DEFAULT;
4953                 text = status->new.name;
4954         }
4956         draw_text(view, type, text, TRUE);
4957         return TRUE;
4960 static enum request
4961 status_load_error(struct view *view, struct view *stage, const char *path)
4963         if (displayed_views() == 2 || display[current_view] != view)
4964                 maximize_view(view);
4965         report("Failed to load '%s': %s", path, io_strerror(&stage->io));
4966         return REQ_NONE;
4969 static enum request
4970 status_enter(struct view *view, struct line *line)
4972         struct status *status = line->data;
4973         const char *oldpath = status ? status->old.name : NULL;
4974         /* Diffs for unmerged entries are empty when passing the new
4975          * path, so leave it empty. */
4976         const char *newpath = status && status->status != 'U' ? status->new.name : NULL;
4977         const char *info;
4978         enum open_flags split;
4979         struct view *stage = VIEW(REQ_VIEW_STAGE);
4981         if (line->type == LINE_STAT_NONE ||
4982             (!status && line[1].type == LINE_STAT_NONE)) {
4983                 report("No file to diff");
4984                 return REQ_NONE;
4985         }
4987         switch (line->type) {
4988         case LINE_STAT_STAGED:
4989                 if (is_initial_commit()) {
4990                         const char *no_head_diff_argv[] = {
4991                                 "git", "diff", "--no-color", "--patch-with-stat",
4992                                         "--", "/dev/null", newpath, NULL
4993                         };
4995                         if (!prepare_update(stage, no_head_diff_argv, opt_cdup, FORMAT_DASH))
4996                                 return status_load_error(view, stage, newpath);
4997                 } else {
4998                         const char *index_show_argv[] = {
4999                                 "git", "diff-index", "--root", "--patch-with-stat",
5000                                         "-C", "-M", "--cached", "HEAD", "--",
5001                                         oldpath, newpath, NULL
5002                         };
5004                         if (!prepare_update(stage, index_show_argv, opt_cdup, FORMAT_DASH))
5005                                 return status_load_error(view, stage, newpath);
5006                 }
5008                 if (status)
5009                         info = "Staged changes to %s";
5010                 else
5011                         info = "Staged changes";
5012                 break;
5014         case LINE_STAT_UNSTAGED:
5015         {
5016                 const char *files_show_argv[] = {
5017                         "git", "diff-files", "--root", "--patch-with-stat",
5018                                 "-C", "-M", "--", oldpath, newpath, NULL
5019                 };
5021                 if (!prepare_update(stage, files_show_argv, opt_cdup, FORMAT_DASH))
5022                         return status_load_error(view, stage, newpath);
5023                 if (status)
5024                         info = "Unstaged changes to %s";
5025                 else
5026                         info = "Unstaged changes";
5027                 break;
5028         }
5029         case LINE_STAT_UNTRACKED:
5030                 if (!newpath) {
5031                         report("No file to show");
5032                         return REQ_NONE;
5033                 }
5035                 if (!suffixcmp(status->new.name, -1, "/")) {
5036                         report("Cannot display a directory");
5037                         return REQ_NONE;
5038                 }
5040                 if (!prepare_update_file(stage, newpath))
5041                         return status_load_error(view, stage, newpath);
5042                 info = "Untracked file %s";
5043                 break;
5045         case LINE_STAT_HEAD:
5046                 return REQ_NONE;
5048         default:
5049                 die("line type %d not handled in switch", line->type);
5050         }
5052         split = view_is_displayed(view) ? OPEN_SPLIT : 0;
5053         open_view(view, REQ_VIEW_STAGE, OPEN_PREPARED | split);
5054         if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
5055                 if (status) {
5056                         stage_status = *status;
5057                 } else {
5058                         memset(&stage_status, 0, sizeof(stage_status));
5059                 }
5061                 stage_line_type = line->type;
5062                 stage_chunks = 0;
5063                 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
5064         }
5066         return REQ_NONE;
5069 static bool
5070 status_exists(struct status *status, enum line_type type)
5072         struct view *view = VIEW(REQ_VIEW_STATUS);
5073         unsigned long lineno;
5075         for (lineno = 0; lineno < view->lines; lineno++) {
5076                 struct line *line = &view->line[lineno];
5077                 struct status *pos = line->data;
5079                 if (line->type != type)
5080                         continue;
5081                 if (!pos && (!status || !status->status) && line[1].data) {
5082                         select_view_line(view, lineno);
5083                         return TRUE;
5084                 }
5085                 if (pos && !strcmp(status->new.name, pos->new.name)) {
5086                         select_view_line(view, lineno);
5087                         return TRUE;
5088                 }
5089         }
5091         return FALSE;
5095 static bool
5096 status_update_prepare(struct io *io, enum line_type type)
5098         const char *staged_argv[] = {
5099                 "git", "update-index", "-z", "--index-info", NULL
5100         };
5101         const char *others_argv[] = {
5102                 "git", "update-index", "-z", "--add", "--remove", "--stdin", NULL
5103         };
5105         switch (type) {
5106         case LINE_STAT_STAGED:
5107                 return run_io(io, staged_argv, opt_cdup, IO_WR);
5109         case LINE_STAT_UNSTAGED:
5110                 return run_io(io, others_argv, opt_cdup, IO_WR);
5112         case LINE_STAT_UNTRACKED:
5113                 return run_io(io, others_argv, NULL, IO_WR);
5115         default:
5116                 die("line type %d not handled in switch", type);
5117                 return FALSE;
5118         }
5121 static bool
5122 status_update_write(struct io *io, struct status *status, enum line_type type)
5124         char buf[SIZEOF_STR];
5125         size_t bufsize = 0;
5127         switch (type) {
5128         case LINE_STAT_STAGED:
5129                 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
5130                                         status->old.mode,
5131                                         status->old.rev,
5132                                         status->old.name, 0))
5133                         return FALSE;
5134                 break;
5136         case LINE_STAT_UNSTAGED:
5137         case LINE_STAT_UNTRACKED:
5138                 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
5139                         return FALSE;
5140                 break;
5142         default:
5143                 die("line type %d not handled in switch", type);
5144         }
5146         return io_write(io, buf, bufsize);
5149 static bool
5150 status_update_file(struct status *status, enum line_type type)
5152         struct io io = {};
5153         bool result;
5155         if (!status_update_prepare(&io, type))
5156                 return FALSE;
5158         result = status_update_write(&io, status, type);
5159         return done_io(&io) && result;
5162 static bool
5163 status_update_files(struct view *view, struct line *line)
5165         char buf[sizeof(view->ref)];
5166         struct io io = {};
5167         bool result = TRUE;
5168         struct line *pos = view->line + view->lines;
5169         int files = 0;
5170         int file, done;
5171         int cursor_y, cursor_x;
5173         if (!status_update_prepare(&io, line->type))
5174                 return FALSE;
5176         for (pos = line; pos < view->line + view->lines && pos->data; pos++)
5177                 files++;
5179         string_copy(buf, view->ref);
5180         getsyx(cursor_y, cursor_x);
5181         for (file = 0, done = 5; result && file < files; line++, file++) {
5182                 int almost_done = file * 100 / files;
5184                 if (almost_done > done) {
5185                         done = almost_done;
5186                         string_format(view->ref, "updating file %u of %u (%d%% done)",
5187                                       file, files, done);
5188                         update_view_title(view);
5189                         setsyx(cursor_y, cursor_x);
5190                         doupdate();
5191                 }
5192                 result = status_update_write(&io, line->data, line->type);
5193         }
5194         string_copy(view->ref, buf);
5196         return done_io(&io) && result;
5199 static bool
5200 status_update(struct view *view)
5202         struct line *line = &view->line[view->lineno];
5204         assert(view->lines);
5206         if (!line->data) {
5207                 /* This should work even for the "On branch" line. */
5208                 if (line < view->line + view->lines && !line[1].data) {
5209                         report("Nothing to update");
5210                         return FALSE;
5211                 }
5213                 if (!status_update_files(view, line + 1)) {
5214                         report("Failed to update file status");
5215                         return FALSE;
5216                 }
5218         } else if (!status_update_file(line->data, line->type)) {
5219                 report("Failed to update file status");
5220                 return FALSE;
5221         }
5223         return TRUE;
5226 static bool
5227 status_revert(struct status *status, enum line_type type, bool has_none)
5229         if (!status || type != LINE_STAT_UNSTAGED) {
5230                 if (type == LINE_STAT_STAGED) {
5231                         report("Cannot revert changes to staged files");
5232                 } else if (type == LINE_STAT_UNTRACKED) {
5233                         report("Cannot revert changes to untracked files");
5234                 } else if (has_none) {
5235                         report("Nothing to revert");
5236                 } else {
5237                         report("Cannot revert changes to multiple files");
5238                 }
5239                 return FALSE;
5241         } else {
5242                 char mode[10] = "100644";
5243                 const char *reset_argv[] = {
5244                         "git", "update-index", "--cacheinfo", mode,
5245                                 status->old.rev, status->old.name, NULL
5246                 };
5247                 const char *checkout_argv[] = {
5248                         "git", "checkout", "--", status->old.name, NULL
5249                 };
5251                 if (!prompt_yesno("Are you sure you want to overwrite any changes?"))
5252                         return FALSE;
5253                 string_format(mode, "%o", status->old.mode);
5254                 return (status->status != 'U' || run_io_fg(reset_argv, opt_cdup)) &&
5255                         run_io_fg(checkout_argv, opt_cdup);
5256         }
5259 static enum request
5260 status_request(struct view *view, enum request request, struct line *line)
5262         struct status *status = line->data;
5264         switch (request) {
5265         case REQ_STATUS_UPDATE:
5266                 if (!status_update(view))
5267                         return REQ_NONE;
5268                 break;
5270         case REQ_STATUS_REVERT:
5271                 if (!status_revert(status, line->type, status_has_none(view, line)))
5272                         return REQ_NONE;
5273                 break;
5275         case REQ_STATUS_MERGE:
5276                 if (!status || status->status != 'U') {
5277                         report("Merging only possible for files with unmerged status ('U').");
5278                         return REQ_NONE;
5279                 }
5280                 open_mergetool(status->new.name);
5281                 break;
5283         case REQ_EDIT:
5284                 if (!status)
5285                         return request;
5286                 if (status->status == 'D') {
5287                         report("File has been deleted.");
5288                         return REQ_NONE;
5289                 }
5291                 open_editor(status->status != '?', status->new.name);
5292                 break;
5294         case REQ_VIEW_BLAME:
5295                 if (status) {
5296                         string_copy(opt_file, status->new.name);
5297                         opt_ref[0] = 0;
5298                 }
5299                 return request;
5301         case REQ_ENTER:
5302                 /* After returning the status view has been split to
5303                  * show the stage view. No further reloading is
5304                  * necessary. */
5305                 return status_enter(view, line);
5307         case REQ_REFRESH:
5308                 /* Simply reload the view. */
5309                 break;
5311         default:
5312                 return request;
5313         }
5315         open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
5317         return REQ_NONE;
5320 static void
5321 status_select(struct view *view, struct line *line)
5323         struct status *status = line->data;
5324         char file[SIZEOF_STR] = "all files";
5325         const char *text;
5326         const char *key;
5328         if (status && !string_format(file, "'%s'", status->new.name))
5329                 return;
5331         if (!status && line[1].type == LINE_STAT_NONE)
5332                 line++;
5334         switch (line->type) {
5335         case LINE_STAT_STAGED:
5336                 text = "Press %s to unstage %s for commit";
5337                 break;
5339         case LINE_STAT_UNSTAGED:
5340                 text = "Press %s to stage %s for commit";
5341                 break;
5343         case LINE_STAT_UNTRACKED:
5344                 text = "Press %s to stage %s for addition";
5345                 break;
5347         case LINE_STAT_HEAD:
5348         case LINE_STAT_NONE:
5349                 text = "Nothing to update";
5350                 break;
5352         default:
5353                 die("line type %d not handled in switch", line->type);
5354         }
5356         if (status && status->status == 'U') {
5357                 text = "Press %s to resolve conflict in %s";
5358                 key = get_key(REQ_STATUS_MERGE);
5360         } else {
5361                 key = get_key(REQ_STATUS_UPDATE);
5362         }
5364         string_format(view->ref, text, key, file);
5367 static bool
5368 status_grep(struct view *view, struct line *line)
5370         struct status *status = line->data;
5372         if (status) {
5373                 const char buf[2] = { status->status, 0 };
5374                 const char *text[] = { status->new.name, buf, NULL };
5376                 return grep_text(view, text);
5377         }
5379         return FALSE;
5382 static struct view_ops status_ops = {
5383         "file",
5384         NULL,
5385         status_open,
5386         NULL,
5387         status_draw,
5388         status_request,
5389         status_grep,
5390         status_select,
5391 };
5394 static bool
5395 stage_diff_write(struct io *io, struct line *line, struct line *end)
5397         while (line < end) {
5398                 if (!io_write(io, line->data, strlen(line->data)) ||
5399                     !io_write(io, "\n", 1))
5400                         return FALSE;
5401                 line++;
5402                 if (line->type == LINE_DIFF_CHUNK ||
5403                     line->type == LINE_DIFF_HEADER)
5404                         break;
5405         }
5407         return TRUE;
5410 static struct line *
5411 stage_diff_find(struct view *view, struct line *line, enum line_type type)
5413         for (; view->line < line; line--)
5414                 if (line->type == type)
5415                         return line;
5417         return NULL;
5420 static bool
5421 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
5423         const char *apply_argv[SIZEOF_ARG] = {
5424                 "git", "apply", "--whitespace=nowarn", NULL
5425         };
5426         struct line *diff_hdr;
5427         struct io io = {};
5428         int argc = 3;
5430         diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
5431         if (!diff_hdr)
5432                 return FALSE;
5434         if (!revert)
5435                 apply_argv[argc++] = "--cached";
5436         if (revert || stage_line_type == LINE_STAT_STAGED)
5437                 apply_argv[argc++] = "-R";
5438         apply_argv[argc++] = "-";
5439         apply_argv[argc++] = NULL;
5440         if (!run_io(&io, apply_argv, opt_cdup, IO_WR))
5441                 return FALSE;
5443         if (!stage_diff_write(&io, diff_hdr, chunk) ||
5444             !stage_diff_write(&io, chunk, view->line + view->lines))
5445                 chunk = NULL;
5447         done_io(&io);
5448         run_io_bg(update_index_argv);
5450         return chunk ? TRUE : FALSE;
5453 static bool
5454 stage_update(struct view *view, struct line *line)
5456         struct line *chunk = NULL;
5458         if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
5459                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5461         if (chunk) {
5462                 if (!stage_apply_chunk(view, chunk, FALSE)) {
5463                         report("Failed to apply chunk");
5464                         return FALSE;
5465                 }
5467         } else if (!stage_status.status) {
5468                 view = VIEW(REQ_VIEW_STATUS);
5470                 for (line = view->line; line < view->line + view->lines; line++)
5471                         if (line->type == stage_line_type)
5472                                 break;
5474                 if (!status_update_files(view, line + 1)) {
5475                         report("Failed to update files");
5476                         return FALSE;
5477                 }
5479         } else if (!status_update_file(&stage_status, stage_line_type)) {
5480                 report("Failed to update file");
5481                 return FALSE;
5482         }
5484         return TRUE;
5487 static bool
5488 stage_revert(struct view *view, struct line *line)
5490         struct line *chunk = NULL;
5492         if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
5493                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5495         if (chunk) {
5496                 if (!prompt_yesno("Are you sure you want to revert changes?"))
5497                         return FALSE;
5499                 if (!stage_apply_chunk(view, chunk, TRUE)) {
5500                         report("Failed to revert chunk");
5501                         return FALSE;
5502                 }
5503                 return TRUE;
5505         } else {
5506                 return status_revert(stage_status.status ? &stage_status : NULL,
5507                                      stage_line_type, FALSE);
5508         }
5512 static void
5513 stage_next(struct view *view, struct line *line)
5515         int i;
5517         if (!stage_chunks) {
5518                 static size_t alloc = 0;
5519                 int *tmp;
5521                 for (line = view->line; line < view->line + view->lines; line++) {
5522                         if (line->type != LINE_DIFF_CHUNK)
5523                                 continue;
5525                         tmp = realloc_items(stage_chunk, &alloc,
5526                                             stage_chunks, sizeof(*tmp));
5527                         if (!tmp) {
5528                                 report("Allocation failure");
5529                                 return;
5530                         }
5532                         stage_chunk = tmp;
5533                         stage_chunk[stage_chunks++] = line - view->line;
5534                 }
5535         }
5537         for (i = 0; i < stage_chunks; i++) {
5538                 if (stage_chunk[i] > view->lineno) {
5539                         do_scroll_view(view, stage_chunk[i] - view->lineno);
5540                         report("Chunk %d of %d", i + 1, stage_chunks);
5541                         return;
5542                 }
5543         }
5545         report("No next chunk found");
5548 static enum request
5549 stage_request(struct view *view, enum request request, struct line *line)
5551         switch (request) {
5552         case REQ_STATUS_UPDATE:
5553                 if (!stage_update(view, line))
5554                         return REQ_NONE;
5555                 break;
5557         case REQ_STATUS_REVERT:
5558                 if (!stage_revert(view, line))
5559                         return REQ_NONE;
5560                 break;
5562         case REQ_STAGE_NEXT:
5563                 if (stage_line_type == LINE_STAT_UNTRACKED) {
5564                         report("File is untracked; press %s to add",
5565                                get_key(REQ_STATUS_UPDATE));
5566                         return REQ_NONE;
5567                 }
5568                 stage_next(view, line);
5569                 return REQ_NONE;
5571         case REQ_EDIT:
5572                 if (!stage_status.new.name[0])
5573                         return request;
5574                 if (stage_status.status == 'D') {
5575                         report("File has been deleted.");
5576                         return REQ_NONE;
5577                 }
5579                 open_editor(stage_status.status != '?', stage_status.new.name);
5580                 break;
5582         case REQ_REFRESH:
5583                 /* Reload everything ... */
5584                 break;
5586         case REQ_VIEW_BLAME:
5587                 if (stage_status.new.name[0]) {
5588                         string_copy(opt_file, stage_status.new.name);
5589                         opt_ref[0] = 0;
5590                 }
5591                 return request;
5593         case REQ_ENTER:
5594                 return pager_request(view, request, line);
5596         default:
5597                 return request;
5598         }
5600         VIEW(REQ_VIEW_STATUS)->p_restore = TRUE;
5601         open_view(view, REQ_VIEW_STATUS, OPEN_REFRESH);
5603         /* Check whether the staged entry still exists, and close the
5604          * stage view if it doesn't. */
5605         if (!status_exists(&stage_status, stage_line_type)) {
5606                 status_restore(VIEW(REQ_VIEW_STATUS));
5607                 return REQ_VIEW_CLOSE;
5608         }
5610         if (stage_line_type == LINE_STAT_UNTRACKED) {
5611                 if (!suffixcmp(stage_status.new.name, -1, "/")) {
5612                         report("Cannot display a directory");
5613                         return REQ_NONE;
5614                 }
5616                 if (!prepare_update_file(view, stage_status.new.name)) {
5617                         report("Failed to open file: %s", strerror(errno));
5618                         return REQ_NONE;
5619                 }
5620         }
5621         open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
5623         return REQ_NONE;
5626 static struct view_ops stage_ops = {
5627         "line",
5628         NULL,
5629         NULL,
5630         pager_read,
5631         pager_draw,
5632         stage_request,
5633         pager_grep,
5634         pager_select,
5635 };
5638 /*
5639  * Revision graph
5640  */
5642 struct commit {
5643         char id[SIZEOF_REV];            /* SHA1 ID. */
5644         char title[128];                /* First line of the commit message. */
5645         const char *author;             /* Author of the commit. */
5646         time_t time;                    /* Date from the author ident. */
5647         struct ref **refs;              /* Repository references. */
5648         chtype graph[SIZEOF_REVGRAPH];  /* Ancestry chain graphics. */
5649         size_t graph_size;              /* The width of the graph array. */
5650         bool has_parents;               /* Rewritten --parents seen. */
5651 };
5653 /* Size of rev graph with no  "padding" columns */
5654 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
5656 struct rev_graph {
5657         struct rev_graph *prev, *next, *parents;
5658         char rev[SIZEOF_REVITEMS][SIZEOF_REV];
5659         size_t size;
5660         struct commit *commit;
5661         size_t pos;
5662         unsigned int boundary:1;
5663 };
5665 /* Parents of the commit being visualized. */
5666 static struct rev_graph graph_parents[4];
5668 /* The current stack of revisions on the graph. */
5669 static struct rev_graph graph_stacks[4] = {
5670         { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
5671         { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
5672         { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
5673         { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
5674 };
5676 static inline bool
5677 graph_parent_is_merge(struct rev_graph *graph)
5679         return graph->parents->size > 1;
5682 static inline void
5683 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
5685         struct commit *commit = graph->commit;
5687         if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
5688                 commit->graph[commit->graph_size++] = symbol;
5691 static void
5692 clear_rev_graph(struct rev_graph *graph)
5694         graph->boundary = 0;
5695         graph->size = graph->pos = 0;
5696         graph->commit = NULL;
5697         memset(graph->parents, 0, sizeof(*graph->parents));
5700 static void
5701 done_rev_graph(struct rev_graph *graph)
5703         if (graph_parent_is_merge(graph) &&
5704             graph->pos < graph->size - 1 &&
5705             graph->next->size == graph->size + graph->parents->size - 1) {
5706                 size_t i = graph->pos + graph->parents->size - 1;
5708                 graph->commit->graph_size = i * 2;
5709                 while (i < graph->next->size - 1) {
5710                         append_to_rev_graph(graph, ' ');
5711                         append_to_rev_graph(graph, '\\');
5712                         i++;
5713                 }
5714         }
5716         clear_rev_graph(graph);
5719 static void
5720 push_rev_graph(struct rev_graph *graph, const char *parent)
5722         int i;
5724         /* "Collapse" duplicate parents lines.
5725          *
5726          * FIXME: This needs to also update update the drawn graph but
5727          * for now it just serves as a method for pruning graph lines. */
5728         for (i = 0; i < graph->size; i++)
5729                 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
5730                         return;
5732         if (graph->size < SIZEOF_REVITEMS) {
5733                 string_copy_rev(graph->rev[graph->size++], parent);
5734         }
5737 static chtype
5738 get_rev_graph_symbol(struct rev_graph *graph)
5740         chtype symbol;
5742         if (graph->boundary)
5743                 symbol = REVGRAPH_BOUND;
5744         else if (graph->parents->size == 0)
5745                 symbol = REVGRAPH_INIT;
5746         else if (graph_parent_is_merge(graph))
5747                 symbol = REVGRAPH_MERGE;
5748         else if (graph->pos >= graph->size)
5749                 symbol = REVGRAPH_BRANCH;
5750         else
5751                 symbol = REVGRAPH_COMMIT;
5753         return symbol;
5756 static void
5757 draw_rev_graph(struct rev_graph *graph)
5759         struct rev_filler {
5760                 chtype separator, line;
5761         };
5762         enum { DEFAULT, RSHARP, RDIAG, LDIAG };
5763         static struct rev_filler fillers[] = {
5764                 { ' ',  '|' },
5765                 { '`',  '.' },
5766                 { '\'', ' ' },
5767                 { '/',  ' ' },
5768         };
5769         chtype symbol = get_rev_graph_symbol(graph);
5770         struct rev_filler *filler;
5771         size_t i;
5773         if (opt_line_graphics)
5774                 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
5776         filler = &fillers[DEFAULT];
5778         for (i = 0; i < graph->pos; i++) {
5779                 append_to_rev_graph(graph, filler->line);
5780                 if (graph_parent_is_merge(graph->prev) &&
5781                     graph->prev->pos == i)
5782                         filler = &fillers[RSHARP];
5784                 append_to_rev_graph(graph, filler->separator);
5785         }
5787         /* Place the symbol for this revision. */
5788         append_to_rev_graph(graph, symbol);
5790         if (graph->prev->size > graph->size)
5791                 filler = &fillers[RDIAG];
5792         else
5793                 filler = &fillers[DEFAULT];
5795         i++;
5797         for (; i < graph->size; i++) {
5798                 append_to_rev_graph(graph, filler->separator);
5799                 append_to_rev_graph(graph, filler->line);
5800                 if (graph_parent_is_merge(graph->prev) &&
5801                     i < graph->prev->pos + graph->parents->size)
5802                         filler = &fillers[RSHARP];
5803                 if (graph->prev->size > graph->size)
5804                         filler = &fillers[LDIAG];
5805         }
5807         if (graph->prev->size > graph->size) {
5808                 append_to_rev_graph(graph, filler->separator);
5809                 if (filler->line != ' ')
5810                         append_to_rev_graph(graph, filler->line);
5811         }
5814 /* Prepare the next rev graph */
5815 static void
5816 prepare_rev_graph(struct rev_graph *graph)
5818         size_t i;
5820         /* First, traverse all lines of revisions up to the active one. */
5821         for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5822                 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5823                         break;
5825                 push_rev_graph(graph->next, graph->rev[graph->pos]);
5826         }
5828         /* Interleave the new revision parent(s). */
5829         for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5830                 push_rev_graph(graph->next, graph->parents->rev[i]);
5832         /* Lastly, put any remaining revisions. */
5833         for (i = graph->pos + 1; i < graph->size; i++)
5834                 push_rev_graph(graph->next, graph->rev[i]);
5837 static void
5838 update_rev_graph(struct view *view, struct rev_graph *graph)
5840         /* If this is the finalizing update ... */
5841         if (graph->commit)
5842                 prepare_rev_graph(graph);
5844         /* Graph visualization needs a one rev look-ahead,
5845          * so the first update doesn't visualize anything. */
5846         if (!graph->prev->commit)
5847                 return;
5849         if (view->lines > 2)
5850                 view->line[view->lines - 3].dirty = 1;
5851         if (view->lines > 1)
5852                 view->line[view->lines - 2].dirty = 1;
5853         draw_rev_graph(graph->prev);
5854         done_rev_graph(graph->prev->prev);
5858 /*
5859  * Main view backend
5860  */
5862 static const char *main_argv[SIZEOF_ARG] = {
5863         "git", "log", "--no-color", "--pretty=raw", "--parents",
5864                       "--topo-order", "%(head)", NULL
5865 };
5867 static bool
5868 main_draw(struct view *view, struct line *line, unsigned int lineno)
5870         struct commit *commit = line->data;
5872         if (!commit->author)
5873                 return FALSE;
5875         if (opt_date && draw_date(view, &commit->time))
5876                 return TRUE;
5878         if (opt_author && draw_author(view, commit->author))
5879                 return TRUE;
5881         if (opt_rev_graph && commit->graph_size &&
5882             draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5883                 return TRUE;
5885         if (opt_show_refs && commit->refs) {
5886                 size_t i = 0;
5888                 do {
5889                         enum line_type type;
5891                         if (commit->refs[i]->head)
5892                                 type = LINE_MAIN_HEAD;
5893                         else if (commit->refs[i]->ltag)
5894                                 type = LINE_MAIN_LOCAL_TAG;
5895                         else if (commit->refs[i]->tag)
5896                                 type = LINE_MAIN_TAG;
5897                         else if (commit->refs[i]->tracked)
5898                                 type = LINE_MAIN_TRACKED;
5899                         else if (commit->refs[i]->remote)
5900                                 type = LINE_MAIN_REMOTE;
5901                         else
5902                                 type = LINE_MAIN_REF;
5904                         if (draw_text(view, type, "[", TRUE) ||
5905                             draw_text(view, type, commit->refs[i]->name, TRUE) ||
5906                             draw_text(view, type, "]", TRUE))
5907                                 return TRUE;
5909                         if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5910                                 return TRUE;
5911                 } while (commit->refs[i++]->next);
5912         }
5914         draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5915         return TRUE;
5918 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5919 static bool
5920 main_read(struct view *view, char *line)
5922         static struct rev_graph *graph = graph_stacks;
5923         enum line_type type;
5924         struct commit *commit;
5926         if (!line) {
5927                 int i;
5929                 if (!view->lines && !view->parent)
5930                         die("No revisions match the given arguments.");
5931                 if (view->lines > 0) {
5932                         commit = view->line[view->lines - 1].data;
5933                         view->line[view->lines - 1].dirty = 1;
5934                         if (!commit->author) {
5935                                 view->lines--;
5936                                 free(commit);
5937                                 graph->commit = NULL;
5938                         }
5939                 }
5940                 update_rev_graph(view, graph);
5942                 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5943                         clear_rev_graph(&graph_stacks[i]);
5944                 return TRUE;
5945         }
5947         type = get_line_type(line);
5948         if (type == LINE_COMMIT) {
5949                 commit = calloc(1, sizeof(struct commit));
5950                 if (!commit)
5951                         return FALSE;
5953                 line += STRING_SIZE("commit ");
5954                 if (*line == '-') {
5955                         graph->boundary = 1;
5956                         line++;
5957                 }
5959                 string_copy_rev(commit->id, line);
5960                 commit->refs = get_refs(commit->id);
5961                 graph->commit = commit;
5962                 add_line_data(view, commit, LINE_MAIN_COMMIT);
5964                 while ((line = strchr(line, ' '))) {
5965                         line++;
5966                         push_rev_graph(graph->parents, line);
5967                         commit->has_parents = TRUE;
5968                 }
5969                 return TRUE;
5970         }
5972         if (!view->lines)
5973                 return TRUE;
5974         commit = view->line[view->lines - 1].data;
5976         switch (type) {
5977         case LINE_PARENT:
5978                 if (commit->has_parents)
5979                         break;
5980                 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5981                 break;
5983         case LINE_AUTHOR:
5984                 parse_author_line(line + STRING_SIZE("author "),
5985                                   &commit->author, &commit->time);
5986                 update_rev_graph(view, graph);
5987                 graph = graph->next;
5988                 break;
5990         default:
5991                 /* Fill in the commit title if it has not already been set. */
5992                 if (commit->title[0])
5993                         break;
5995                 /* Require titles to start with a non-space character at the
5996                  * offset used by git log. */
5997                 if (strncmp(line, "    ", 4))
5998                         break;
5999                 line += 4;
6000                 /* Well, if the title starts with a whitespace character,
6001                  * try to be forgiving.  Otherwise we end up with no title. */
6002                 while (isspace(*line))
6003                         line++;
6004                 if (*line == '\0')
6005                         break;
6006                 /* FIXME: More graceful handling of titles; append "..." to
6007                  * shortened titles, etc. */
6009                 string_expand(commit->title, sizeof(commit->title), line, 1);
6010                 view->line[view->lines - 1].dirty = 1;
6011         }
6013         return TRUE;
6016 static enum request
6017 main_request(struct view *view, enum request request, struct line *line)
6019         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
6021         switch (request) {
6022         case REQ_ENTER:
6023                 open_view(view, REQ_VIEW_DIFF, flags);
6024                 break;
6025         case REQ_REFRESH:
6026                 load_refs();
6027                 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
6028                 break;
6029         default:
6030                 return request;
6031         }
6033         return REQ_NONE;
6036 static bool
6037 grep_refs(struct ref **refs, regex_t *regex)
6039         regmatch_t pmatch;
6040         size_t i = 0;
6042         if (!opt_show_refs || !refs)
6043                 return FALSE;
6044         do {
6045                 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
6046                         return TRUE;
6047         } while (refs[i++]->next);
6049         return FALSE;
6052 static bool
6053 main_grep(struct view *view, struct line *line)
6055         struct commit *commit = line->data;
6056         const char *text[] = {
6057                 commit->title,
6058                 opt_author ? commit->author : "",
6059                 opt_date ? mkdate(&commit->time) : "",
6060                 NULL
6061         };
6063         return grep_text(view, text) || grep_refs(commit->refs, view->regex);
6066 static void
6067 main_select(struct view *view, struct line *line)
6069         struct commit *commit = line->data;
6071         string_copy_rev(view->ref, commit->id);
6072         string_copy_rev(ref_commit, view->ref);
6075 static struct view_ops main_ops = {
6076         "commit",
6077         main_argv,
6078         NULL,
6079         main_read,
6080         main_draw,
6081         main_request,
6082         main_grep,
6083         main_select,
6084 };
6087 /*
6088  * Unicode / UTF-8 handling
6089  *
6090  * NOTE: Much of the following code for dealing with Unicode is derived from
6091  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
6092  * src/intl/charset.c from the UTF-8 branch commit elinks-0.11.0-g31f2c28.
6093  */
6095 static inline int
6096 unicode_width(unsigned long c)
6098         if (c >= 0x1100 &&
6099            (c <= 0x115f                         /* Hangul Jamo */
6100             || c == 0x2329
6101             || c == 0x232a
6102             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
6103                                                 /* CJK ... Yi */
6104             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
6105             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
6106             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
6107             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
6108             || (c >= 0xffe0  && c <= 0xffe6)
6109             || (c >= 0x20000 && c <= 0x2fffd)
6110             || (c >= 0x30000 && c <= 0x3fffd)))
6111                 return 2;
6113         if (c == '\t')
6114                 return opt_tab_size;
6116         return 1;
6119 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
6120  * Illegal bytes are set one. */
6121 static const unsigned char utf8_bytes[256] = {
6122         1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
6123         1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
6124         1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
6125         1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
6126         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,
6127         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,
6128         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,
6129         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,
6130 };
6132 /* Decode UTF-8 multi-byte representation into a Unicode character. */
6133 static inline unsigned long
6134 utf8_to_unicode(const char *string, size_t length)
6136         unsigned long unicode;
6138         switch (length) {
6139         case 1:
6140                 unicode  =   string[0];
6141                 break;
6142         case 2:
6143                 unicode  =  (string[0] & 0x1f) << 6;
6144                 unicode +=  (string[1] & 0x3f);
6145                 break;
6146         case 3:
6147                 unicode  =  (string[0] & 0x0f) << 12;
6148                 unicode += ((string[1] & 0x3f) << 6);
6149                 unicode +=  (string[2] & 0x3f);
6150                 break;
6151         case 4:
6152                 unicode  =  (string[0] & 0x0f) << 18;
6153                 unicode += ((string[1] & 0x3f) << 12);
6154                 unicode += ((string[2] & 0x3f) << 6);
6155                 unicode +=  (string[3] & 0x3f);
6156                 break;
6157         case 5:
6158                 unicode  =  (string[0] & 0x0f) << 24;
6159                 unicode += ((string[1] & 0x3f) << 18);
6160                 unicode += ((string[2] & 0x3f) << 12);
6161                 unicode += ((string[3] & 0x3f) << 6);
6162                 unicode +=  (string[4] & 0x3f);
6163                 break;
6164         case 6:
6165                 unicode  =  (string[0] & 0x01) << 30;
6166                 unicode += ((string[1] & 0x3f) << 24);
6167                 unicode += ((string[2] & 0x3f) << 18);
6168                 unicode += ((string[3] & 0x3f) << 12);
6169                 unicode += ((string[4] & 0x3f) << 6);
6170                 unicode +=  (string[5] & 0x3f);
6171                 break;
6172         default:
6173                 die("Invalid Unicode length");
6174         }
6176         /* Invalid characters could return the special 0xfffd value but NUL
6177          * should be just as good. */
6178         return unicode > 0xffff ? 0 : unicode;
6181 /* Calculates how much of string can be shown within the given maximum width
6182  * and sets trimmed parameter to non-zero value if all of string could not be
6183  * shown. If the reserve flag is TRUE, it will reserve at least one
6184  * trailing character, which can be useful when drawing a delimiter.
6185  *
6186  * Returns the number of bytes to output from string to satisfy max_width. */
6187 static size_t
6188 utf8_length(const char **start, size_t skip, int *width, size_t max_width, int *trimmed, bool reserve)
6190         const char *string = *start;
6191         const char *end = strchr(string, '\0');
6192         unsigned char last_bytes = 0;
6193         size_t last_ucwidth = 0;
6195         *width = 0;
6196         *trimmed = 0;
6198         while (string < end) {
6199                 int c = *(unsigned char *) string;
6200                 unsigned char bytes = utf8_bytes[c];
6201                 size_t ucwidth;
6202                 unsigned long unicode;
6204                 if (string + bytes > end)
6205                         break;
6207                 /* Change representation to figure out whether
6208                  * it is a single- or double-width character. */
6210                 unicode = utf8_to_unicode(string, bytes);
6211                 /* FIXME: Graceful handling of invalid Unicode character. */
6212                 if (!unicode)
6213                         break;
6215                 ucwidth = unicode_width(unicode);
6216                 if (skip > 0) {
6217                         skip -= ucwidth <= skip ? ucwidth : skip;
6218                         *start += bytes;
6219                 }
6220                 *width  += ucwidth;
6221                 if (*width > max_width) {
6222                         *trimmed = 1;
6223                         *width -= ucwidth;
6224                         if (reserve && *width == max_width) {
6225                                 string -= last_bytes;
6226                                 *width -= last_ucwidth;
6227                         }
6228                         break;
6229                 }
6231                 string  += bytes;
6232                 last_bytes = ucwidth ? bytes : 0;
6233                 last_ucwidth = ucwidth;
6234         }
6236         return string - *start;
6240 /*
6241  * Status management
6242  */
6244 /* Whether or not the curses interface has been initialized. */
6245 static bool cursed = FALSE;
6247 /* Terminal hacks and workarounds. */
6248 static bool use_scroll_redrawwin;
6249 static bool use_scroll_status_wclear;
6251 /* The status window is used for polling keystrokes. */
6252 static WINDOW *status_win;
6254 /* Reading from the prompt? */
6255 static bool input_mode = FALSE;
6257 static bool status_empty = FALSE;
6259 /* Update status and title window. */
6260 static void
6261 report(const char *msg, ...)
6263         struct view *view = display[current_view];
6265         if (input_mode)
6266                 return;
6268         if (!view) {
6269                 char buf[SIZEOF_STR];
6270                 va_list args;
6272                 va_start(args, msg);
6273                 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
6274                         buf[sizeof(buf) - 1] = 0;
6275                         buf[sizeof(buf) - 2] = '.';
6276                         buf[sizeof(buf) - 3] = '.';
6277                         buf[sizeof(buf) - 4] = '.';
6278                 }
6279                 va_end(args);
6280                 die("%s", buf);
6281         }
6283         if (!status_empty || *msg) {
6284                 va_list args;
6286                 va_start(args, msg);
6288                 wmove(status_win, 0, 0);
6289                 if (view->has_scrolled && use_scroll_status_wclear)
6290                         wclear(status_win);
6291                 if (*msg) {
6292                         vwprintw(status_win, msg, args);
6293                         status_empty = FALSE;
6294                 } else {
6295                         status_empty = TRUE;
6296                 }
6297                 wclrtoeol(status_win);
6298                 wnoutrefresh(status_win);
6300                 va_end(args);
6301         }
6303         update_view_title(view);
6306 /* Controls when nodelay should be in effect when polling user input. */
6307 static void
6308 set_nonblocking_input(bool loading)
6310         static unsigned int loading_views;
6312         if ((loading == FALSE && loading_views-- == 1) ||
6313             (loading == TRUE  && loading_views++ == 0))
6314                 nodelay(status_win, loading);
6317 static void
6318 init_display(void)
6320         const char *term;
6321         int x, y;
6323         /* Initialize the curses library */
6324         if (isatty(STDIN_FILENO)) {
6325                 cursed = !!initscr();
6326                 opt_tty = stdin;
6327         } else {
6328                 /* Leave stdin and stdout alone when acting as a pager. */
6329                 opt_tty = fopen("/dev/tty", "r+");
6330                 if (!opt_tty)
6331                         die("Failed to open /dev/tty");
6332                 cursed = !!newterm(NULL, opt_tty, opt_tty);
6333         }
6335         if (!cursed)
6336                 die("Failed to initialize curses");
6338         nonl();         /* Disable conversion and detect newlines from input. */
6339         cbreak();       /* Take input chars one at a time, no wait for \n */
6340         noecho();       /* Don't echo input */
6341         leaveok(stdscr, FALSE);
6343         if (has_colors())
6344                 init_colors();
6346         getmaxyx(stdscr, y, x);
6347         status_win = newwin(1, 0, y - 1, 0);
6348         if (!status_win)
6349                 die("Failed to create status window");
6351         /* Enable keyboard mapping */
6352         keypad(status_win, TRUE);
6353         wbkgdset(status_win, get_line_attr(LINE_STATUS));
6355         TABSIZE = opt_tab_size;
6356         if (opt_line_graphics) {
6357                 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
6358         }
6360         term = getenv("XTERM_VERSION") ? NULL : getenv("COLORTERM");
6361         if (term && !strcmp(term, "gnome-terminal")) {
6362                 /* In the gnome-terminal-emulator, the message from
6363                  * scrolling up one line when impossible followed by
6364                  * scrolling down one line causes corruption of the
6365                  * status line. This is fixed by calling wclear. */
6366                 use_scroll_status_wclear = TRUE;
6367                 use_scroll_redrawwin = FALSE;
6369         } else if (term && !strcmp(term, "xrvt-xpm")) {
6370                 /* No problems with full optimizations in xrvt-(unicode)
6371                  * and aterm. */
6372                 use_scroll_status_wclear = use_scroll_redrawwin = FALSE;
6374         } else {
6375                 /* When scrolling in (u)xterm the last line in the
6376                  * scrolling direction will update slowly. */
6377                 use_scroll_redrawwin = TRUE;
6378                 use_scroll_status_wclear = FALSE;
6379         }
6382 static int
6383 get_input(int prompt_position)
6385         struct view *view;
6386         int i, key, cursor_y, cursor_x;
6388         if (prompt_position)
6389                 input_mode = TRUE;
6391         while (TRUE) {
6392                 foreach_view (view, i) {
6393                         update_view(view);
6394                         if (view_is_displayed(view) && view->has_scrolled &&
6395                             use_scroll_redrawwin)
6396                                 redrawwin(view->win);
6397                         view->has_scrolled = FALSE;
6398                 }
6400                 /* Update the cursor position. */
6401                 if (prompt_position) {
6402                         getbegyx(status_win, cursor_y, cursor_x);
6403                         cursor_x = prompt_position;
6404                 } else {
6405                         view = display[current_view];
6406                         getbegyx(view->win, cursor_y, cursor_x);
6407                         cursor_x = view->width - 1;
6408                         cursor_y += view->lineno - view->offset;
6409                 }
6410                 setsyx(cursor_y, cursor_x);
6412                 /* Refresh, accept single keystroke of input */
6413                 doupdate();
6414                 key = wgetch(status_win);
6416                 /* wgetch() with nodelay() enabled returns ERR when
6417                  * there's no input. */
6418                 if (key == ERR) {
6420                 } else if (key == KEY_RESIZE) {
6421                         int height, width;
6423                         getmaxyx(stdscr, height, width);
6425                         wresize(status_win, 1, width);
6426                         mvwin(status_win, height - 1, 0);
6427                         wnoutrefresh(status_win);
6428                         resize_display();
6429                         redraw_display(TRUE);
6431                 } else {
6432                         input_mode = FALSE;
6433                         return key;
6434                 }
6435         }
6438 static char *
6439 prompt_input(const char *prompt, input_handler handler, void *data)
6441         enum input_status status = INPUT_OK;
6442         static char buf[SIZEOF_STR];
6443         size_t pos = 0;
6445         buf[pos] = 0;
6447         while (status == INPUT_OK || status == INPUT_SKIP) {
6448                 int key;
6450                 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
6451                 wclrtoeol(status_win);
6453                 key = get_input(pos + 1);
6454                 switch (key) {
6455                 case KEY_RETURN:
6456                 case KEY_ENTER:
6457                 case '\n':
6458                         status = pos ? INPUT_STOP : INPUT_CANCEL;
6459                         break;
6461                 case KEY_BACKSPACE:
6462                         if (pos > 0)
6463                                 buf[--pos] = 0;
6464                         else
6465                                 status = INPUT_CANCEL;
6466                         break;
6468                 case KEY_ESC:
6469                         status = INPUT_CANCEL;
6470                         break;
6472                 default:
6473                         if (pos >= sizeof(buf)) {
6474                                 report("Input string too long");
6475                                 return NULL;
6476                         }
6478                         status = handler(data, buf, key);
6479                         if (status == INPUT_OK)
6480                                 buf[pos++] = (char) key;
6481                 }
6482         }
6484         /* Clear the status window */
6485         status_empty = FALSE;
6486         report("");
6488         if (status == INPUT_CANCEL)
6489                 return NULL;
6491         buf[pos++] = 0;
6493         return buf;
6496 static enum input_status
6497 prompt_yesno_handler(void *data, char *buf, int c)
6499         if (c == 'y' || c == 'Y')
6500                 return INPUT_STOP;
6501         if (c == 'n' || c == 'N')
6502                 return INPUT_CANCEL;
6503         return INPUT_SKIP;
6506 static bool
6507 prompt_yesno(const char *prompt)
6509         char prompt2[SIZEOF_STR];
6511         if (!string_format(prompt2, "%s [Yy/Nn]", prompt))
6512                 return FALSE;
6514         return !!prompt_input(prompt2, prompt_yesno_handler, NULL);
6517 static enum input_status
6518 read_prompt_handler(void *data, char *buf, int c)
6520         return isprint(c) ? INPUT_OK : INPUT_SKIP;
6523 static char *
6524 read_prompt(const char *prompt)
6526         return prompt_input(prompt, read_prompt_handler, NULL);
6529 /*
6530  * Repository properties
6531  */
6533 static struct ref *refs = NULL;
6534 static size_t refs_alloc = 0;
6535 static size_t refs_size = 0;
6537 /* Id <-> ref store */
6538 static struct ref ***id_refs = NULL;
6539 static size_t id_refs_alloc = 0;
6540 static size_t id_refs_size = 0;
6542 static int
6543 compare_refs(const void *ref1_, const void *ref2_)
6545         const struct ref *ref1 = *(const struct ref **)ref1_;
6546         const struct ref *ref2 = *(const struct ref **)ref2_;
6548         if (ref1->tag != ref2->tag)
6549                 return ref2->tag - ref1->tag;
6550         if (ref1->ltag != ref2->ltag)
6551                 return ref2->ltag - ref2->ltag;
6552         if (ref1->head != ref2->head)
6553                 return ref2->head - ref1->head;
6554         if (ref1->tracked != ref2->tracked)
6555                 return ref2->tracked - ref1->tracked;
6556         if (ref1->remote != ref2->remote)
6557                 return ref2->remote - ref1->remote;
6558         return strcmp(ref1->name, ref2->name);
6561 static struct ref **
6562 get_refs(const char *id)
6564         struct ref ***tmp_id_refs;
6565         struct ref **ref_list = NULL;
6566         size_t ref_list_alloc = 0;
6567         size_t ref_list_size = 0;
6568         size_t i;
6570         for (i = 0; i < id_refs_size; i++)
6571                 if (!strcmp(id, id_refs[i][0]->id))
6572                         return id_refs[i];
6574         tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
6575                                     sizeof(*id_refs));
6576         if (!tmp_id_refs)
6577                 return NULL;
6579         id_refs = tmp_id_refs;
6581         for (i = 0; i < refs_size; i++) {
6582                 struct ref **tmp;
6584                 if (strcmp(id, refs[i].id))
6585                         continue;
6587                 tmp = realloc_items(ref_list, &ref_list_alloc,
6588                                     ref_list_size + 1, sizeof(*ref_list));
6589                 if (!tmp) {
6590                         if (ref_list)
6591                                 free(ref_list);
6592                         return NULL;
6593                 }
6595                 ref_list = tmp;
6596                 ref_list[ref_list_size] = &refs[i];
6597                 /* XXX: The properties of the commit chains ensures that we can
6598                  * safely modify the shared ref. The repo references will
6599                  * always be similar for the same id. */
6600                 ref_list[ref_list_size]->next = 1;
6602                 ref_list_size++;
6603         }
6605         if (ref_list) {
6606                 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
6607                 ref_list[ref_list_size - 1]->next = 0;
6608                 id_refs[id_refs_size++] = ref_list;
6609         }
6611         return ref_list;
6614 static int
6615 read_ref(char *id, size_t idlen, char *name, size_t namelen)
6617         struct ref *ref;
6618         bool tag = FALSE;
6619         bool ltag = FALSE;
6620         bool remote = FALSE;
6621         bool tracked = FALSE;
6622         bool check_replace = FALSE;
6623         bool head = FALSE;
6625         if (!prefixcmp(name, "refs/tags/")) {
6626                 if (!suffixcmp(name, namelen, "^{}")) {
6627                         namelen -= 3;
6628                         name[namelen] = 0;
6629                         if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
6630                                 check_replace = TRUE;
6631                 } else {
6632                         ltag = TRUE;
6633                 }
6635                 tag = TRUE;
6636                 namelen -= STRING_SIZE("refs/tags/");
6637                 name    += STRING_SIZE("refs/tags/");
6639         } else if (!prefixcmp(name, "refs/remotes/")) {
6640                 remote = TRUE;
6641                 namelen -= STRING_SIZE("refs/remotes/");
6642                 name    += STRING_SIZE("refs/remotes/");
6643                 tracked  = !strcmp(opt_remote, name);
6645         } else if (!prefixcmp(name, "refs/heads/")) {
6646                 namelen -= STRING_SIZE("refs/heads/");
6647                 name    += STRING_SIZE("refs/heads/");
6648                 head     = !strncmp(opt_head, name, namelen);
6650         } else if (!strcmp(name, "HEAD")) {
6651                 string_ncopy(opt_head_rev, id, idlen);
6652                 return OK;
6653         }
6655         if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
6656                 /* it's an annotated tag, replace the previous SHA1 with the
6657                  * resolved commit id; relies on the fact git-ls-remote lists
6658                  * the commit id of an annotated tag right before the commit id
6659                  * it points to. */
6660                 refs[refs_size - 1].ltag = ltag;
6661                 string_copy_rev(refs[refs_size - 1].id, id);
6663                 return OK;
6664         }
6665         refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
6666         if (!refs)
6667                 return ERR;
6669         ref = &refs[refs_size++];
6670         ref->name = malloc(namelen + 1);
6671         if (!ref->name)
6672                 return ERR;
6674         strncpy(ref->name, name, namelen);
6675         ref->name[namelen] = 0;
6676         ref->head = head;
6677         ref->tag = tag;
6678         ref->ltag = ltag;
6679         ref->remote = remote;
6680         ref->tracked = tracked;
6681         string_copy_rev(ref->id, id);
6683         return OK;
6686 static int
6687 load_refs(void)
6689         static const char *ls_remote_argv[SIZEOF_ARG] = {
6690                 "git", "ls-remote", opt_git_dir, NULL
6691         };
6692         static bool init = FALSE;
6694         if (!init) {
6695                 argv_from_env(ls_remote_argv, "TIG_LS_REMOTE");
6696                 init = TRUE;
6697         }
6699         if (!*opt_git_dir)
6700                 return OK;
6702         while (refs_size > 0)
6703                 free(refs[--refs_size].name);
6704         while (id_refs_size > 0)
6705                 free(id_refs[--id_refs_size]);
6707         return run_io_load(ls_remote_argv, "\t", read_ref);
6710 static void
6711 set_remote_branch(const char *name, const char *value, size_t valuelen)
6713         if (!strcmp(name, ".remote")) {
6714                 string_ncopy(opt_remote, value, valuelen);
6716         } else if (*opt_remote && !strcmp(name, ".merge")) {
6717                 size_t from = strlen(opt_remote);
6719                 if (!prefixcmp(value, "refs/heads/"))
6720                         value += STRING_SIZE("refs/heads/");
6722                 if (!string_format_from(opt_remote, &from, "/%s", value))
6723                         opt_remote[0] = 0;
6724         }
6727 static void
6728 set_repo_config_option(char *name, char *value, int (*cmd)(int, const char **))
6730         const char *argv[SIZEOF_ARG] = { name, "=" };
6731         int argc = 1 + (cmd == option_set_command);
6732         int error = ERR;
6734         if (!argv_from_string(argv, &argc, value))
6735                 config_msg = "Too many option arguments";
6736         else
6737                 error = cmd(argc, argv);
6739         if (error == ERR)
6740                 warn("Option 'tig.%s': %s", name, config_msg);
6743 static bool
6744 set_environment_variable(const char *name, const char *value)
6746         size_t len = strlen(name) + 1 + strlen(value) + 1;
6747         char *env = malloc(len);
6749         if (env &&
6750             string_nformat(env, len, NULL, "%s=%s", name, value) &&
6751             putenv(env) == 0)
6752                 return TRUE;
6753         free(env);
6754         return FALSE;
6757 static void
6758 set_work_tree(const char *value)
6760         char cwd[SIZEOF_STR];
6762         if (!getcwd(cwd, sizeof(cwd)))
6763                 die("Failed to get cwd path: %s", strerror(errno));
6764         if (chdir(opt_git_dir) < 0)
6765                 die("Failed to chdir(%s): %s", strerror(errno));
6766         if (!getcwd(opt_git_dir, sizeof(opt_git_dir)))
6767                 die("Failed to get git path: %s", strerror(errno));
6768         if (chdir(cwd) < 0)
6769                 die("Failed to chdir(%s): %s", cwd, strerror(errno));
6770         if (chdir(value) < 0)
6771                 die("Failed to chdir(%s): %s", value, strerror(errno));
6772         if (!getcwd(cwd, sizeof(cwd)))
6773                 die("Failed to get cwd path: %s", strerror(errno));
6774         if (!set_environment_variable("GIT_WORK_TREE", cwd))
6775                 die("Failed to set GIT_WORK_TREE to '%s'", cwd);
6776         if (!set_environment_variable("GIT_DIR", opt_git_dir))
6777                 die("Failed to set GIT_DIR to '%s'", opt_git_dir);
6778         opt_is_inside_work_tree = TRUE;
6781 static int
6782 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
6784         if (!strcmp(name, "i18n.commitencoding"))
6785                 string_ncopy(opt_encoding, value, valuelen);
6787         else if (!strcmp(name, "core.editor"))
6788                 string_ncopy(opt_editor, value, valuelen);
6790         else if (!strcmp(name, "core.worktree"))
6791                 set_work_tree(value);
6793         else if (!prefixcmp(name, "tig.color."))
6794                 set_repo_config_option(name + 10, value, option_color_command);
6796         else if (!prefixcmp(name, "tig.bind."))
6797                 set_repo_config_option(name + 9, value, option_bind_command);
6799         else if (!prefixcmp(name, "tig."))
6800                 set_repo_config_option(name + 4, value, option_set_command);
6802         else if (*opt_head && !prefixcmp(name, "branch.") &&
6803                  !strncmp(name + 7, opt_head, strlen(opt_head)))
6804                 set_remote_branch(name + 7 + strlen(opt_head), value, valuelen);
6806         return OK;
6809 static int
6810 load_git_config(void)
6812         const char *config_list_argv[] = { "git", GIT_CONFIG, "--list", NULL };
6814         return run_io_load(config_list_argv, "=", read_repo_config_option);
6817 static int
6818 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
6820         if (!opt_git_dir[0]) {
6821                 string_ncopy(opt_git_dir, name, namelen);
6823         } else if (opt_is_inside_work_tree == -1) {
6824                 /* This can be 3 different values depending on the
6825                  * version of git being used. If git-rev-parse does not
6826                  * understand --is-inside-work-tree it will simply echo
6827                  * the option else either "true" or "false" is printed.
6828                  * Default to true for the unknown case. */
6829                 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
6831         } else if (*name == '.') {
6832                 string_ncopy(opt_cdup, name, namelen);
6834         } else {
6835                 string_ncopy(opt_prefix, name, namelen);
6836         }
6838         return OK;
6841 static int
6842 load_repo_info(void)
6844         const char *head_argv[] = {
6845                 "git", "symbolic-ref", "HEAD", NULL
6846         };
6847         const char *rev_parse_argv[] = {
6848                 "git", "rev-parse", "--git-dir", "--is-inside-work-tree",
6849                         "--show-cdup", "--show-prefix", NULL
6850         };
6852         if (run_io_buf(head_argv, opt_head, sizeof(opt_head))) {
6853                 chomp_string(opt_head);
6854                 if (!prefixcmp(opt_head, "refs/heads/")) {
6855                         char *offset = opt_head + STRING_SIZE("refs/heads/");
6857                         memmove(opt_head, offset, strlen(offset) + 1);
6858                 }
6859         }
6861         return run_io_load(rev_parse_argv, "=", read_repo_info);
6865 /*
6866  * Main
6867  */
6869 static const char usage[] =
6870 "tig " TIG_VERSION " (" __DATE__ ")\n"
6871 "\n"
6872 "Usage: tig        [options] [revs] [--] [paths]\n"
6873 "   or: tig show   [options] [revs] [--] [paths]\n"
6874 "   or: tig blame  [rev] path\n"
6875 "   or: tig status\n"
6876 "   or: tig <      [git command output]\n"
6877 "\n"
6878 "Options:\n"
6879 "  -v, --version   Show version and exit\n"
6880 "  -h, --help      Show help message and exit";
6882 static void __NORETURN
6883 quit(int sig)
6885         /* XXX: Restore tty modes and let the OS cleanup the rest! */
6886         if (cursed)
6887                 endwin();
6888         exit(0);
6891 static void __NORETURN
6892 die(const char *err, ...)
6894         va_list args;
6896         endwin();
6898         va_start(args, err);
6899         fputs("tig: ", stderr);
6900         vfprintf(stderr, err, args);
6901         fputs("\n", stderr);
6902         va_end(args);
6904         exit(1);
6907 static void
6908 warn(const char *msg, ...)
6910         va_list args;
6912         va_start(args, msg);
6913         fputs("tig warning: ", stderr);
6914         vfprintf(stderr, msg, args);
6915         fputs("\n", stderr);
6916         va_end(args);
6919 static enum request
6920 parse_options(int argc, const char *argv[])
6922         enum request request = REQ_VIEW_MAIN;
6923         const char *subcommand;
6924         bool seen_dashdash = FALSE;
6925         /* XXX: This is vulnerable to the user overriding options
6926          * required for the main view parser. */
6927         const char *custom_argv[SIZEOF_ARG] = {
6928                 "git", "log", "--no-color", "--pretty=raw", "--parents",
6929                         "--topo-order", NULL
6930         };
6931         int i, j = 6;
6933         if (!isatty(STDIN_FILENO)) {
6934                 io_open(&VIEW(REQ_VIEW_PAGER)->io, "");
6935                 return REQ_VIEW_PAGER;
6936         }
6938         if (argc <= 1)
6939                 return REQ_NONE;
6941         subcommand = argv[1];
6942         if (!strcmp(subcommand, "status")) {
6943                 if (argc > 2)
6944                         warn("ignoring arguments after `%s'", subcommand);
6945                 return REQ_VIEW_STATUS;
6947         } else if (!strcmp(subcommand, "blame")) {
6948                 if (argc <= 2 || argc > 4)
6949                         die("invalid number of options to blame\n\n%s", usage);
6951                 i = 2;
6952                 if (argc == 4) {
6953                         string_ncopy(opt_ref, argv[i], strlen(argv[i]));
6954                         i++;
6955                 }
6957                 string_ncopy(opt_file, argv[i], strlen(argv[i]));
6958                 return REQ_VIEW_BLAME;
6960         } else if (!strcmp(subcommand, "show")) {
6961                 request = REQ_VIEW_DIFF;
6963         } else {
6964                 subcommand = NULL;
6965         }
6967         if (subcommand) {
6968                 custom_argv[1] = subcommand;
6969                 j = 2;
6970         }
6972         for (i = 1 + !!subcommand; i < argc; i++) {
6973                 const char *opt = argv[i];
6975                 if (seen_dashdash || !strcmp(opt, "--")) {
6976                         seen_dashdash = TRUE;
6978                 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
6979                         printf("tig version %s\n", TIG_VERSION);
6980                         quit(0);
6982                 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
6983                         printf("%s\n", usage);
6984                         quit(0);
6985                 }
6987                 custom_argv[j++] = opt;
6988                 if (j >= ARRAY_SIZE(custom_argv))
6989                         die("command too long");
6990         }
6992         if (!prepare_update(VIEW(request), custom_argv, NULL, FORMAT_NONE))                                                                        
6993                 die("Failed to format arguments"); 
6995         return request;
6998 int
6999 main(int argc, const char *argv[])
7001         enum request request = parse_options(argc, argv);
7002         struct view *view;
7003         size_t i;
7005         signal(SIGINT, quit);
7006         signal(SIGPIPE, SIG_IGN);
7008         if (setlocale(LC_ALL, "")) {
7009                 char *codeset = nl_langinfo(CODESET);
7011                 string_ncopy(opt_codeset, codeset, strlen(codeset));
7012         }
7014         if (load_repo_info() == ERR)
7015                 die("Failed to load repo info.");
7017         if (load_options() == ERR)
7018                 die("Failed to load user config.");
7020         if (load_git_config() == ERR)
7021                 die("Failed to load repo config.");
7023         /* Require a git repository unless when running in pager mode. */
7024         if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
7025                 die("Not a git repository");
7027         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
7028                 opt_utf8 = FALSE;
7030         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
7031                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
7032                 if (opt_iconv == ICONV_NONE)
7033                         die("Failed to initialize character set conversion");
7034         }
7036         if (load_refs() == ERR)
7037                 die("Failed to load refs.");
7039         foreach_view (view, i)
7040                 argv_from_env(view->ops->argv, view->cmd_env);
7042         init_display();
7044         if (request != REQ_NONE)
7045                 open_view(NULL, request, OPEN_PREPARED);
7046         request = request == REQ_NONE ? REQ_VIEW_MAIN : REQ_NONE;
7048         while (view_driver(display[current_view], request)) {
7049                 int key = get_input(0);
7051                 view = display[current_view];
7052                 request = get_keybinding(view->keymap, key);
7054                 /* Some low-level request handling. This keeps access to
7055                  * status_win restricted. */
7056                 switch (request) {
7057                 case REQ_PROMPT:
7058                 {
7059                         char *cmd = read_prompt(":");
7061                         if (cmd && isdigit(*cmd)) {
7062                                 int lineno = view->lineno + 1;
7064                                 if (parse_int(&lineno, cmd, 1, view->lines + 1) == OK) {
7065                                         select_view_line(view, lineno - 1);
7066                                         report("");
7067                                 } else {
7068                                         report("Unable to parse '%s' as a line number", cmd);
7069                                 }
7071                         } else if (cmd) {
7072                                 struct view *next = VIEW(REQ_VIEW_PAGER);
7073                                 const char *argv[SIZEOF_ARG] = { "git" };
7074                                 int argc = 1;
7076                                 /* When running random commands, initially show the
7077                                  * command in the title. However, it maybe later be
7078                                  * overwritten if a commit line is selected. */
7079                                 string_ncopy(next->ref, cmd, strlen(cmd));
7081                                 if (!argv_from_string(argv, &argc, cmd)) {
7082                                         report("Too many arguments");
7083                                 } else if (!prepare_update(next, argv, NULL, FORMAT_DASH)) {
7084                                         report("Failed to format command");
7085                                 } else {
7086                                         open_view(view, REQ_VIEW_PAGER, OPEN_PREPARED);
7087                                 }
7088                         }
7090                         request = REQ_NONE;
7091                         break;
7092                 }
7093                 case REQ_SEARCH:
7094                 case REQ_SEARCH_BACK:
7095                 {
7096                         const char *prompt = request == REQ_SEARCH ? "/" : "?";
7097                         char *search = read_prompt(prompt);
7099                         if (search)
7100                                 string_ncopy(opt_search, search, strlen(search));
7101                         else if (*opt_search)
7102                                 request = request == REQ_SEARCH ?
7103                                         REQ_FIND_NEXT :
7104                                         REQ_FIND_PREV;
7105                         else
7106                                 request = REQ_NONE;
7107                         break;
7108                 }
7109                 default:
7110                         break;
7111                 }
7112         }
7114         quit(0);
7116         return 0;