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