?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Refactor draw_lineno to use draw_graphic
[tig.git] / tig.c
blob148b6b60b1147c130ea7778745011f9cd4646daa
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, spaces, 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 if (view->digits <= 9)
2036 fmt[1] = '0' + digits3;
2037
2038 if (string_format(number, fmt, lineno))
2039 text = number;
2040 }
2041 if (text)
2042 view->col += draw_chars(view, LINE_LINE_NUMBER, text, max, TRUE);
2043 else
2044 view->col += draw_space(view, LINE_LINE_NUMBER, max, digits3);
2045 return draw_graphic(view, LINE_DEFAULT, &line_graphics[LINE_GRAPHIC_VLINE], 1);
2046 }
2047
2048 static bool
2049 draw_view_line(struct view *view, unsigned int lineno)
2050 {
2051 struct line *line;
2052 bool selected = (view->offset + lineno == view->lineno);
2053
2054 assert(view_is_displayed(view));
2055
2056 if (view->offset + lineno >= view->lines)
2057 return FALSE;
2058
2059 line = &view->line[view->offset + lineno];
2060
2061 wmove(view->win, lineno, 0);
2062 if (line->cleareol)
2063 wclrtoeol(view->win);
2064 view->col = 0;
2065 view->curline = line;
2066 view->curtype = LINE_NONE;
2067 line->selected = FALSE;
2068 line->dirty = line->cleareol = 0;
2069
2070 if (selected) {
2071 set_view_attr(view, LINE_CURSOR);
2072 line->selected = TRUE;
2073 view->ops->select(view, line);
2074 }
2075
2076 return view->ops->draw(view, line, lineno);
2077 }
2078
2079 static void
2080 redraw_view_dirty(struct view *view)
2081 {
2082 bool dirty = FALSE;
2083 int lineno;
2084
2085 for (lineno = 0; lineno < view->height; lineno++) {
2086 if (view->offset + lineno >= view->lines)
2087 break;
2088 if (!view->line[view->offset + lineno].dirty)
2089 continue;
2090 dirty = TRUE;
2091 if (!draw_view_line(view, lineno))
2092 break;
2093 }
2094
2095 if (!dirty)
2096 return;
2097 wnoutrefresh(view->win);
2098 }
2099
2100 static void
2101 redraw_view_from(struct view *view, int lineno)
2102 {
2103 assert(0 <= lineno && lineno < view->height);
2104
2105 for (; lineno < view->height; lineno++) {
2106 if (!draw_view_line(view, lineno))
2107 break;
2108 }
2109
2110 wnoutrefresh(view->win);
2111 }
2112
2113 static void
2114 redraw_view(struct view *view)
2115 {
2116 werase(view->win);
2117 redraw_view_from(view, 0);
2118 }
2119
2120
2121 static void
2122 update_view_title(struct view *view)
2123 {
2124 char buf[SIZEOF_STR];
2125 char state[SIZEOF_STR];
2126 size_t bufpos = 0, statelen = 0;
2127
2128 assert(view_is_displayed(view));
2129
2130 if (view != VIEW(REQ_VIEW_STATUS) && view->lines) {
2131 unsigned int view_lines = view->offset + view->height;
2132 unsigned int lines = view->lines
2133 ? MIN(view_lines, view->lines) * 100 / view->lines
2134 : 0;
2135
2136 string_format_from(state, &statelen, " - %s %d of %d (%d%%)",
2137 view->ops->type,
2138 view->lineno + 1,
2139 view->lines,
2140 lines);
2141
2142 }
2143
2144 if (view->pipe) {
2145 time_t secs = time(NULL) - view->start_time;
2146
2147 /* Three git seconds are a long time ... */
2148 if (secs > 2)
2149 string_format_from(state, &statelen, " loading %lds", secs);
2150 }
2151
2152 string_format_from(buf, &bufpos, "[%s]", view->name);
2153 if (*view->ref && bufpos < view->width) {
2154 size_t refsize = strlen(view->ref);
2155 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
2156
2157 if (minsize < view->width)
2158 refsize = view->width - minsize + 7;
2159 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
2160 }
2161
2162 if (statelen && bufpos < view->width) {
2163 string_format_from(buf, &bufpos, "%s", state);
2164 }
2165
2166 if (view == display[current_view])
2167 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
2168 else
2169 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
2170
2171 mvwaddnstr(view->title, 0, 0, buf, bufpos);
2172 wclrtoeol(view->title);
2173 wnoutrefresh(view->title);
2174 }
2175
2176 static void
2177 resize_display(void)
2178 {
2179 int offset, i;
2180 struct view *base = display[0];
2181 struct view *view = display[1] ? display[1] : display[0];
2182
2183 /* Setup window dimensions */
2184
2185 getmaxyx(stdscr, base->height, base->width);
2186
2187 /* Make room for the status window. */
2188 base->height -= 1;
2189
2190 if (view != base) {
2191 /* Horizontal split. */
2192 view->width = base->width;
2193 view->height = SCALE_SPLIT_VIEW(base->height);
2194 base->height -= view->height;
2195
2196 /* Make room for the title bar. */
2197 view->height -= 1;
2198 }
2199
2200 /* Make room for the title bar. */
2201 base->height -= 1;
2202
2203 offset = 0;
2204
2205 foreach_displayed_view (view, i) {
2206 if (!view->win) {
2207 view->win = newwin(view->height, 0, offset, 0);
2208 if (!view->win)
2209 die("Failed to create %s view", view->name);
2210
2211 scrollok(view->win, FALSE);
2212
2213 view->title = newwin(1, 0, offset + view->height, 0);
2214 if (!view->title)
2215 die("Failed to create title window");
2216
2217 } else {
2218 wresize(view->win, view->height, view->width);
2219 mvwin(view->win, offset, 0);
2220 mvwin(view->title, offset + view->height, 0);
2221 }
2222
2223 offset += view->height + 1;
2224 }
2225 }
2226
2227 static void
2228 redraw_display(bool clear)
2229 {
2230 struct view *view;
2231 int i;
2232
2233 foreach_displayed_view (view, i) {
2234 if (clear)
2235 wclear(view->win);
2236 redraw_view(view);
2237 update_view_title(view);
2238 }
2239 }
2240
2241 static void
2242 toggle_view_option(bool *option, const char *help)
2243 {
2244 *option = !*option;
2245 redraw_display(FALSE);
2246 report("%sabling %s", *option ? "En" : "Dis", help);
2247 }
2248
2249 static void
2250 maximize_view(struct view *view)
2251 {
2252 memset(display, 0, sizeof(display));
2253 current_view = 0;
2254 display[current_view] = view;
2255 resize_display();
2256 redraw_display(FALSE);
2257 report("");
2258 }
2259
2260
2261 /*
2262 * Navigation
2263 */
2264
2265 static bool
2266 goto_view_line(struct view *view, unsigned long offset, unsigned long lineno)
2267 {
2268 if (lineno >= view->lines)
2269 lineno = view->lines > 0 ? view->lines - 1 : 0;
2270
2271 if (offset > lineno || offset + view->height <= lineno) {
2272 unsigned long half = view->height / 2;
2273
2274 if (lineno > half)
2275 offset = lineno - half;
2276 else
2277 offset = 0;
2278 }
2279
2280 if (offset != view->offset || lineno != view->lineno) {
2281 view->offset = offset;
2282 view->lineno = lineno;
2283 return TRUE;
2284 }
2285
2286 return FALSE;
2287 }
2288
2289 static int
2290 apply_step(double step, int value)
2291 {
2292 if (step >= 1)
2293 return (int) step;
2294 value *= step + 0.01;
2295 return value ? value : 1;
2296 }
2297
2298 /* Scrolling backend */
2299 static void
2300 do_scroll_view(struct view *view, int lines)
2301 {
2302 bool redraw_current_line = FALSE;
2303
2304 /* The rendering expects the new offset. */
2305 view->offset += lines;
2306
2307 assert(0 <= view->offset && view->offset < view->lines);
2308 assert(lines);
2309
2310 /* Move current line into the view. */
2311 if (view->lineno < view->offset) {
2312 view->lineno = view->offset;
2313 redraw_current_line = TRUE;
2314 } else if (view->lineno >= view->offset + view->height) {
2315 view->lineno = view->offset + view->height - 1;
2316 redraw_current_line = TRUE;
2317 }
2318
2319 assert(view->offset <= view->lineno && view->lineno < view->lines);
2320
2321 /* Redraw the whole screen if scrolling is pointless. */
2322 if (view->height < ABS(lines)) {
2323 redraw_view(view);
2324
2325 } else {
2326 int line = lines > 0 ? view->height - lines : 0;
2327 int end = line + ABS(lines);
2328
2329 scrollok(view->win, TRUE);
2330 wscrl(view->win, lines);
2331 scrollok(view->win, FALSE);
2332
2333 while (line < end && draw_view_line(view, line))
2334 line++;
2335
2336 if (redraw_current_line)
2337 draw_view_line(view, view->lineno - view->offset);
2338 wnoutrefresh(view->win);
2339 }
2340
2341 view->has_scrolled = TRUE;
2342 report("");
2343 }
2344
2345 /* Scroll frontend */
2346 static void
2347 scroll_view(struct view *view, enum request request)
2348 {
2349 int lines = 1;
2350
2351 assert(view_is_displayed(view));
2352
2353 switch (request) {
2354 case REQ_SCROLL_LEFT:
2355 if (view->yoffset == 0) {
2356 report("Cannot scroll beyond the first column");
2357 return;
2358 }
2359 if (view->yoffset <= apply_step(opt_hscroll, view->width))
2360 view->yoffset = 0;
2361 else
2362 view->yoffset -= apply_step(opt_hscroll, view->width);
2363 redraw_view_from(view, 0);
2364 report("");
2365 return;
2366 case REQ_SCROLL_RIGHT:
2367 view->yoffset += apply_step(opt_hscroll, view->width);
2368 redraw_view(view);
2369 report("");
2370 return;
2371 case REQ_SCROLL_PAGE_DOWN:
2372 lines = view->height;
2373 case REQ_SCROLL_LINE_DOWN:
2374 if (view->offset + lines > view->lines)
2375 lines = view->lines - view->offset;
2376
2377 if (lines == 0 || view->offset + view->height >= view->lines) {
2378 report("Cannot scroll beyond the last line");
2379 return;
2380 }
2381 break;
2382
2383 case REQ_SCROLL_PAGE_UP:
2384 lines = view->height;
2385 case REQ_SCROLL_LINE_UP:
2386 if (lines > view->offset)
2387 lines = view->offset;
2388
2389 if (lines == 0) {
2390 report("Cannot scroll beyond the first line");
2391 return;
2392 }
2393
2394 lines = -lines;
2395 break;
2396
2397 default:
2398 die("request %d not handled in switch", request);
2399 }
2400
2401 do_scroll_view(view, lines);
2402 }
2403
2404 /* Cursor moving */
2405 static void
2406 move_view(struct view *view, enum request request)
2407 {
2408 int scroll_steps = 0;
2409 int steps;
2410
2411 switch (request) {
2412 case REQ_MOVE_FIRST_LINE:
2413 steps = -view->lineno;
2414 break;
2415
2416 case REQ_MOVE_LAST_LINE:
2417 steps = view->lines - view->lineno - 1;
2418 break;
2419
2420 case REQ_MOVE_PAGE_UP:
2421 steps = view->height > view->lineno
2422 ? -view->lineno : -view->height;
2423 break;
2424
2425 case REQ_MOVE_PAGE_DOWN:
2426 steps = view->lineno + view->height >= view->lines
2427 ? view->lines - view->lineno - 1 : view->height;
2428 break;
2429
2430 case REQ_MOVE_UP:
2431 steps = -1;
2432 break;
2433
2434 case REQ_MOVE_DOWN:
2435 steps = 1;
2436 break;
2437
2438 default:
2439 die("request %d not handled in switch", request);
2440 }
2441
2442 if (steps <= 0 && view->lineno == 0) {
2443 report("Cannot move beyond the first line");
2444 return;
2445
2446 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
2447 report("Cannot move beyond the last line");
2448 return;
2449 }
2450
2451 /* Move the current line */
2452 view->lineno += steps;
2453 assert(0 <= view->lineno && view->lineno < view->lines);
2454
2455 /* Check whether the view needs to be scrolled */
2456 if (view->lineno < view->offset ||
2457 view->lineno >= view->offset + view->height) {
2458 scroll_steps = steps;
2459 if (steps < 0 && -steps > view->offset) {
2460 scroll_steps = -view->offset;
2461
2462 } else if (steps > 0) {
2463 if (view->lineno == view->lines - 1 &&
2464 view->lines > view->height) {
2465 scroll_steps = view->lines - view->offset - 1;
2466 if (scroll_steps >= view->height)
2467 scroll_steps -= view->height - 1;
2468 }
2469 }
2470 }
2471
2472 if (!view_is_displayed(view)) {
2473 view->offset += scroll_steps;
2474 assert(0 <= view->offset && view->offset < view->lines);
2475 view->ops->select(view, &view->line[view->lineno]);
2476 return;
2477 }
2478
2479 /* Repaint the old "current" line if we be scrolling */
2480 if (ABS(steps) < view->height)
2481 draw_view_line(view, view->lineno - steps - view->offset);
2482
2483 if (scroll_steps) {
2484 do_scroll_view(view, scroll_steps);
2485 return;
2486 }
2487
2488 /* Draw the current line */
2489 draw_view_line(view, view->lineno - view->offset);
2490
2491 wnoutrefresh(view->win);
2492 report("");
2493 }
2494
2495
2496 /*
2497 * Searching
2498 */
2499
2500 static void search_view(struct view *view, enum request request);
2501
2502 static void
2503 select_view_line(struct view *view, unsigned long lineno)
2504 {
2505 unsigned long old_lineno = view->lineno;
2506 unsigned long old_offset = view->offset;
2507
2508 if (goto_view_line(view, view->offset, lineno)) {
2509 if (view_is_displayed(view)) {
2510 if (old_offset != view->offset) {
2511 redraw_view(view);
2512 } else {
2513 draw_view_line(view, old_lineno - view->offset);
2514 draw_view_line(view, view->lineno - view->offset);
2515 wnoutrefresh(view->win);
2516 }
2517 } else {
2518 view->ops->select(view, &view->line[view->lineno]);
2519 }
2520 }
2521 }
2522
2523 static void
2524 find_next(struct view *view, enum request request)
2525 {
2526 unsigned long lineno = view->lineno;
2527 int direction;
2528
2529 if (!*view->grep) {
2530 if (!*opt_search)
2531 report("No previous search");
2532 else
2533 search_view(view, request);
2534 return;
2535 }
2536
2537 switch (request) {
2538 case REQ_SEARCH:
2539 case REQ_FIND_NEXT:
2540 direction = 1;
2541 break;
2542
2543 case REQ_SEARCH_BACK:
2544 case REQ_FIND_PREV:
2545 direction = -1;
2546 break;
2547
2548 default:
2549 return;
2550 }
2551
2552 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
2553 lineno += direction;
2554
2555 /* Note, lineno is unsigned long so will wrap around in which case it
2556 * will become bigger than view->lines. */
2557 for (; lineno < view->lines; lineno += direction) {
2558 if (view->ops->grep(view, &view->line[lineno])) {
2559 select_view_line(view, lineno);
2560 report("Line %ld matches '%s'", lineno + 1, view->grep);
2561 return;
2562 }
2563 }
2564
2565 report("No match found for '%s'", view->grep);
2566 }
2567
2568 static void
2569 search_view(struct view *view, enum request request)
2570 {
2571 int regex_err;
2572
2573 if (view->regex) {
2574 regfree(view->regex);
2575 *view->grep = 0;
2576 } else {
2577 view->regex = calloc(1, sizeof(*view->regex));
2578 if (!view->regex)
2579 return;
2580 }
2581
2582 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
2583 if (regex_err != 0) {
2584 char buf[SIZEOF_STR] = "unknown error";
2585
2586 regerror(regex_err, view->regex, buf, sizeof(buf));
2587 report("Search failed: %s", buf);
2588 return;
2589 }
2590
2591 string_copy(view->grep, opt_search);
2592
2593 find_next(view, request);
2594 }
2595
2596 /*
2597 * Incremental updating
2598 */
2599
2600 static void
2601 reset_view(struct view *view)
2602 {
2603 int i;
2604
2605 for (i = 0; i < view->lines; i++)
2606 free(view->line[i].data);
2607 free(view->line);
2608
2609 view->p_offset = view->offset;
2610 view->p_yoffset = view->yoffset;
2611 view->p_lineno = view->lineno;
2612
2613 view->line = NULL;
2614 view->offset = 0;
2615 view->yoffset = 0;
2616 view->lines = 0;
2617 view->lineno = 0;
2618 view->line_alloc = 0;
2619 view->vid[0] = 0;
2620 view->update_secs = 0;
2621 }
2622
2623 static void
2624 free_argv(const char *argv[])
2625 {
2626 int argc;
2627
2628 for (argc = 0; argv[argc]; argc++)
2629 free((void *) argv[argc]);
2630 }
2631
2632 static bool
2633 format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags flags)
2634 {
2635 char buf[SIZEOF_STR];
2636 int argc;
2637 bool noreplace = flags == FORMAT_NONE;
2638
2639 free_argv(dst_argv);
2640
2641 for (argc = 0; src_argv[argc]; argc++) {
2642 const char *arg = src_argv[argc];
2643 size_t bufpos = 0;
2644
2645 while (arg) {
2646 char *next = strstr(arg, "%(");
2647 int len = next - arg;
2648 const char *value;
2649
2650 if (!next || noreplace) {
2651 if (flags == FORMAT_DASH && !strcmp(arg, "--"))
2652 noreplace = TRUE;
2653 len = strlen(arg);
2654 value = "";
2655
2656 } else if (!prefixcmp(next, "%(directory)")) {
2657 value = opt_path;
2658
2659 } else if (!prefixcmp(next, "%(file)")) {
2660 value = opt_file;
2661
2662 } else if (!prefixcmp(next, "%(ref)")) {
2663 value = *opt_ref ? opt_ref : "HEAD";
2664
2665 } else if (!prefixcmp(next, "%(head)")) {
2666 value = ref_head;
2667
2668 } else if (!prefixcmp(next, "%(commit)")) {
2669 value = ref_commit;
2670
2671 } else if (!prefixcmp(next, "%(blob)")) {
2672 value = ref_blob;
2673
2674 } else {
2675 report("Unknown replacement: `%s`", next);
2676 return FALSE;
2677 }
2678
2679 if (!string_format_from(buf, &bufpos, "%.*s%s", len, arg, value))
2680 return FALSE;
2681
2682 arg = next && !noreplace ? strchr(next, ')') + 1 : NULL;
2683 }
2684
2685 dst_argv[argc] = strdup(buf);
2686 if (!dst_argv[argc])
2687 break;
2688 }
2689
2690 dst_argv[argc] = NULL;
2691
2692 return src_argv[argc] == NULL;
2693 }
2694
2695 static bool
2696 restore_view_position(struct view *view)
2697 {
2698 if (!view->p_restore || (view->pipe && view->lines <= view->p_lineno))
2699 return FALSE;
2700
2701 /* Changing the view position cancels the restoring. */
2702 /* FIXME: Changing back to the first line is not detected. */
2703 if (view->offset != 0 || view->lineno != 0) {
2704 view->p_restore = FALSE;
2705 return FALSE;
2706 }
2707
2708 if (goto_view_line(view, view->p_offset, view->p_lineno) &&
2709 view_is_displayed(view))
2710 werase(view->win);
2711
2712 view->yoffset = view->p_yoffset;
2713 view->p_restore = FALSE;
2714
2715 return TRUE;
2716 }
2717
2718 static void
2719 end_update(struct view *view, bool force)
2720 {
2721 if (!view->pipe)
2722 return;
2723 while (!view->ops->read(view, NULL))
2724 if (!force)
2725 return;
2726 set_nonblocking_input(FALSE);
2727 if (force)
2728 kill_io(view->pipe);
2729 done_io(view->pipe);
2730 view->pipe = NULL;
2731 }
2732
2733 static void
2734 setup_update(struct view *view, const char *vid)
2735 {
2736 set_nonblocking_input(TRUE);
2737 reset_view(view);
2738 string_copy_rev(view->vid, vid);
2739 view->pipe = &view->io;
2740 view->start_time = time(NULL);
2741 }
2742
2743 static bool
2744 prepare_update(struct view *view, const char *argv[], const char *dir,
2745 enum format_flags flags)
2746 {
2747 if (view->pipe)
2748 end_update(view, TRUE);
2749 return init_io_rd(&view->io, argv, dir, flags);
2750 }
2751
2752 static bool
2753 prepare_update_file(struct view *view, const char *name)
2754 {
2755 if (view->pipe)
2756 end_update(view, TRUE);
2757 return io_open(&view->io, name);
2758 }
2759
2760 static bool
2761 begin_update(struct view *view, bool refresh)
2762 {
2763 if (view->pipe)
2764 end_update(view, TRUE);
2765
2766 if (refresh) {
2767 if (!start_io(&view->io))
2768 return FALSE;
2769
2770 } else {
2771 if (view == VIEW(REQ_VIEW_TREE) && strcmp(view->vid, view->id))
2772 opt_path[0] = 0;
2773
2774 if (!run_io_rd(&view->io, view->ops->argv, FORMAT_ALL))
2775 return FALSE;
2776
2777 /* Put the current ref_* value to the view title ref
2778 * member. This is needed by the blob view. Most other
2779 * views sets it automatically after loading because the
2780 * first line is a commit line. */
2781 string_copy_rev(view->ref, view->id);
2782 }
2783
2784 setup_update(view, view->id);
2785
2786 return TRUE;
2787 }
2788
2789 #define ITEM_CHUNK_SIZE 256
2790 static void *
2791 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2792 {
2793 size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2794 size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2795
2796 if (mem == NULL || num_chunks != num_chunks_new) {
2797 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2798 mem = realloc(mem, *size * item_size);
2799 }
2800
2801 return mem;
2802 }
2803
2804 static struct line *
2805 realloc_lines(struct view *view, size_t line_size)
2806 {
2807 size_t alloc = view->line_alloc;
2808 struct line *tmp = realloc_items(view->line, &alloc, line_size,
2809 sizeof(*view->line));
2810
2811 if (!tmp)
2812 return NULL;
2813
2814 view->line = tmp;
2815 view->line_alloc = alloc;
2816 return view->line;
2817 }
2818
2819 static bool
2820 update_view(struct view *view)
2821 {
2822 char out_buffer[BUFSIZ * 2];
2823 char *line;
2824 /* Clear the view and redraw everything since the tree sorting
2825 * might have rearranged things. */
2826 bool redraw = view->lines == 0;
2827 bool can_read = TRUE;
2828
2829 if (!view->pipe)
2830 return TRUE;
2831
2832 if (!io_can_read(view->pipe)) {
2833 if (view->lines == 0 && view_is_displayed(view)) {
2834 time_t secs = time(NULL) - view->start_time;
2835
2836 if (secs > 1 && secs > view->update_secs) {
2837 if (view->update_secs == 0)
2838 redraw_view(view);
2839 update_view_title(view);
2840 view->update_secs = secs;
2841 }
2842 }
2843 return TRUE;
2844 }
2845
2846 for (; (line = io_get(view->pipe, '\n', can_read)); can_read = FALSE) {
2847 if (opt_iconv != ICONV_NONE) {
2848 ICONV_CONST char *inbuf = line;
2849 size_t inlen = strlen(line) + 1;
2850
2851 char *outbuf = out_buffer;
2852 size_t outlen = sizeof(out_buffer);
2853
2854 size_t ret;
2855
2856 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2857 if (ret != (size_t) -1)
2858 line = out_buffer;
2859 }
2860
2861 if (!view->ops->read(view, line)) {
2862 report("Allocation failure");
2863 end_update(view, TRUE);
2864 return FALSE;
2865 }
2866 }
2867
2868 {
2869 unsigned long lines = view->lines;
2870 int digits;
2871
2872 for (digits = 0; lines; digits++)
2873 lines /= 10;
2874
2875 /* Keep the displayed view in sync with line number scaling. */
2876 if (digits != view->digits) {
2877 view->digits = digits;
2878 if (opt_line_number || view == VIEW(REQ_VIEW_BLAME))
2879 redraw = TRUE;
2880 }
2881 }
2882
2883 if (io_error(view->pipe)) {
2884 report("Failed to read: %s", io_strerror(view->pipe));
2885 end_update(view, TRUE);
2886
2887 } else if (io_eof(view->pipe)) {
2888 report("");
2889 end_update(view, FALSE);
2890 }
2891
2892 if (restore_view_position(view))
2893 redraw = TRUE;
2894
2895 if (!view_is_displayed(view))
2896 return TRUE;
2897
2898 if (redraw)
2899 redraw_view_from(view, 0);
2900 else
2901 redraw_view_dirty(view);
2902
2903 /* Update the title _after_ the redraw so that if the redraw picks up a
2904 * commit reference in view->ref it'll be available here. */
2905 update_view_title(view);
2906 return TRUE;
2907 }
2908
2909 static struct line *
2910 add_line_data(struct view *view, void *data, enum line_type type)
2911 {
2912 struct line *line;
2913
2914 if (!realloc_lines(view, view->lines + 1))
2915 return NULL;
2916
2917 line = &view->line[view->lines++];
2918 memset(line, 0, sizeof(*line));
2919 line->type = type;
2920 line->data = data;
2921 line->dirty = 1;
2922
2923 return line;
2924 }
2925
2926 static struct line *
2927 add_line_text(struct view *view, const char *text, enum line_type type)
2928 {
2929 char *data = text ? strdup(text) : NULL;
2930
2931 return data ? add_line_data(view, data, type) : NULL;
2932 }
2933
2934 static struct line *
2935 add_line_format(struct view *view, enum line_type type, const char *fmt, ...)
2936 {
2937 char buf[SIZEOF_STR];
2938 va_list args;
2939
2940 va_start(args, fmt);
2941 if (vsnprintf(buf, sizeof(buf), fmt, args) >= sizeof(buf))
2942 buf[0] = 0;
2943 va_end(args);
2944
2945 return buf[0] ? add_line_text(view, buf, type) : NULL;
2946 }
2947
2948 /*
2949 * View opening
2950 */
2951
2952 enum open_flags {
2953 OPEN_DEFAULT = 0, /* Use default view switching. */
2954 OPEN_SPLIT = 1, /* Split current view. */
2955 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
2956 OPEN_REFRESH = 16, /* Refresh view using previous command. */
2957 OPEN_PREPARED = 32, /* Open already prepared command. */
2958 };
2959
2960 static void
2961 open_view(struct view *prev, enum request request, enum open_flags flags)
2962 {
2963 bool split = !!(flags & OPEN_SPLIT);
2964 bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH | OPEN_PREPARED));
2965 bool nomaximize = !!(flags & OPEN_REFRESH);
2966 struct view *view = VIEW(request);
2967 int nviews = displayed_views();
2968 struct view *base_view = display[0];
2969
2970 if (view == prev && nviews == 1 && !reload) {
2971 report("Already in %s view", view->name);
2972 return;
2973 }
2974
2975 if (view->git_dir && !opt_git_dir[0]) {
2976 report("The %s view is disabled in pager view", view->name);
2977 return;
2978 }
2979
2980 if (split) {
2981 display[1] = view;
2982 current_view = 1;
2983 } else if (!nomaximize) {
2984 /* Maximize the current view. */
2985 memset(display, 0, sizeof(display));
2986 current_view = 0;
2987 display[current_view] = view;
2988 }
2989
2990 /* Resize the view when switching between split- and full-screen,
2991 * or when switching between two different full-screen views. */
2992 if (nviews != displayed_views() ||
2993 (nviews == 1 && base_view != display[0]))
2994 resize_display();
2995
2996 if (view->ops->open) {
2997 if (view->pipe)
2998 end_update(view, TRUE);
2999 if (!view->ops->open(view)) {
3000 report("Failed to load %s view", view->name);
3001 return;
3002 }
3003 restore_view_position(view);
3004
3005 } else if ((reload || strcmp(view->vid, view->id)) &&
3006 !begin_update(view, flags & (OPEN_REFRESH | OPEN_PREPARED))) {
3007 report("Failed to load %s view", view->name);
3008 return;
3009 }
3010
3011 if (split && prev->lineno - prev->offset >= prev->height) {
3012 /* Take the title line into account. */
3013 int lines = prev->lineno - prev->offset - prev->height + 1;
3014
3015 /* Scroll the view that was split if the current line is
3016 * outside the new limited view. */
3017 do_scroll_view(prev, lines);
3018 }
3019
3020 if (prev && view != prev) {
3021 if (split) {
3022 /* "Blur" the previous view. */
3023 update_view_title(prev);
3024 }
3025
3026 view->parent = prev;
3027 }
3028
3029 if (view->pipe && view->lines == 0) {
3030 /* Clear the old view and let the incremental updating refill
3031 * the screen. */
3032 werase(view->win);
3033 view->p_restore = flags & (OPEN_RELOAD | OPEN_REFRESH);
3034 report("");
3035 } else if (view_is_displayed(view)) {
3036 redraw_view(view);
3037 report("");
3038 }
3039 }
3040
3041 static void
3042 open_external_viewer(const char *argv[], const char *dir)
3043 {
3044 def_prog_mode(); /* save current tty modes */
3045 endwin(); /* restore original tty modes */
3046 run_io_fg(argv, dir);
3047 fprintf(stderr, "Press Enter to continue");
3048 getc(opt_tty);
3049 reset_prog_mode();
3050 redraw_display(TRUE);
3051 }
3052
3053 static void
3054 open_mergetool(const char *file)
3055 {
3056 const char *mergetool_argv[] = { "git", "mergetool", file, NULL };
3057
3058 open_external_viewer(mergetool_argv, opt_cdup);
3059 }
3060
3061 static void
3062 open_editor(bool from_root, const char *file)
3063 {
3064 const char *editor_argv[] = { "vi", file, NULL };
3065 const char *editor;
3066
3067 editor = getenv("GIT_EDITOR");
3068 if (!editor && *opt_editor)
3069 editor = opt_editor;
3070 if (!editor)
3071 editor = getenv("VISUAL");
3072 if (!editor)
3073 editor = getenv("EDITOR");
3074 if (!editor)
3075 editor = "vi";
3076
3077 editor_argv[0] = editor;
3078 open_external_viewer(editor_argv, from_root ? opt_cdup : NULL);
3079 }
3080
3081 static void
3082 open_run_request(enum request request)
3083 {
3084 struct run_request *req = get_run_request(request);
3085 const char *argv[ARRAY_SIZE(req->argv)] = { NULL };
3086
3087 if (!req) {
3088 report("Unknown run request");
3089 return;
3090 }
3091
3092 if (format_argv(argv, req->argv, FORMAT_ALL))
3093 open_external_viewer(argv, NULL);
3094 free_argv(argv);
3095 }
3096
3097 /*
3098 * User request switch noodle
3099 */
3100
3101 static int
3102 view_driver(struct view *view, enum request request)
3103 {
3104 int i;
3105
3106 if (request == REQ_NONE) {
3107 doupdate();
3108 return TRUE;
3109 }
3110
3111 if (request > REQ_NONE) {
3112 open_run_request(request);
3113 /* FIXME: When all views can refresh always do this. */
3114 if (view == VIEW(REQ_VIEW_STATUS) ||
3115 view == VIEW(REQ_VIEW_MAIN) ||
3116 view == VIEW(REQ_VIEW_LOG) ||
3117 view == VIEW(REQ_VIEW_STAGE))
3118 request = REQ_REFRESH;
3119 else
3120 return TRUE;
3121 }
3122
3123 if (view && view->lines) {
3124 request = view->ops->request(view, request, &view->line[view->lineno]);
3125 if (request == REQ_NONE)
3126 return TRUE;
3127 }
3128
3129 switch (request) {
3130 case REQ_MOVE_UP:
3131 case REQ_MOVE_DOWN:
3132 case REQ_MOVE_PAGE_UP:
3133 case REQ_MOVE_PAGE_DOWN:
3134 case REQ_MOVE_FIRST_LINE:
3135 case REQ_MOVE_LAST_LINE:
3136 move_view(view, request);
3137 break;
3138
3139 case REQ_SCROLL_LEFT:
3140 case REQ_SCROLL_RIGHT:
3141 case REQ_SCROLL_LINE_DOWN:
3142 case REQ_SCROLL_LINE_UP:
3143 case REQ_SCROLL_PAGE_DOWN:
3144 case REQ_SCROLL_PAGE_UP:
3145 scroll_view(view, request);
3146 break;
3147
3148 case REQ_VIEW_BLAME:
3149 if (!opt_file[0]) {
3150 report("No file chosen, press %s to open tree view",
3151 get_key(REQ_VIEW_TREE));
3152 break;
3153 }
3154 open_view(view, request, OPEN_DEFAULT);
3155 break;
3156
3157 case REQ_VIEW_BLOB:
3158 if (!ref_blob[0]) {
3159 report("No file chosen, press %s to open tree view",
3160 get_key(REQ_VIEW_TREE));
3161 break;
3162 }
3163 open_view(view, request, OPEN_DEFAULT);
3164 break;
3165
3166 case REQ_VIEW_PAGER:
3167 if (!VIEW(REQ_VIEW_PAGER)->pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
3168 report("No pager content, press %s to run command from prompt",
3169 get_key(REQ_PROMPT));
3170 break;
3171 }
3172 open_view(view, request, OPEN_DEFAULT);
3173 break;
3174
3175 case REQ_VIEW_STAGE:
3176 if (!VIEW(REQ_VIEW_STAGE)->lines) {
3177 report("No stage content, press %s to open the status view and choose file",
3178 get_key(REQ_VIEW_STATUS));
3179 break;
3180 }
3181 open_view(view, request, OPEN_DEFAULT);
3182 break;
3183
3184 case REQ_VIEW_STATUS:
3185 if (opt_is_inside_work_tree == FALSE) {
3186 report("The status view requires a working tree");
3187 break;
3188 }
3189 open_view(view, request, OPEN_DEFAULT);
3190 break;
3191
3192 case REQ_VIEW_MAIN:
3193 case REQ_VIEW_DIFF:
3194 case REQ_VIEW_LOG:
3195 case REQ_VIEW_TREE:
3196 case REQ_VIEW_HELP:
3197 open_view(view, request, OPEN_DEFAULT);
3198 break;
3199
3200 case REQ_NEXT:
3201 case REQ_PREVIOUS:
3202 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
3203
3204 if ((view == VIEW(REQ_VIEW_DIFF) &&
3205 view->parent == VIEW(REQ_VIEW_MAIN)) ||
3206 (view == VIEW(REQ_VIEW_DIFF) &&
3207 view->parent == VIEW(REQ_VIEW_BLAME)) ||
3208 (view == VIEW(REQ_VIEW_STAGE) &&
3209 view->parent == VIEW(REQ_VIEW_STATUS)) ||
3210 (view == VIEW(REQ_VIEW_BLOB) &&
3211 view->parent == VIEW(REQ_VIEW_TREE))) {
3212 int line;
3213
3214 view = view->parent;
3215 line = view->lineno;
3216 move_view(view, request);
3217 if (view_is_displayed(view))
3218 update_view_title(view);
3219 if (line != view->lineno)
3220 view->ops->request(view, REQ_ENTER,
3221 &view->line[view->lineno]);
3222
3223 } else {
3224 move_view(view, request);
3225 }
3226 break;
3227
3228 case REQ_VIEW_NEXT:
3229 {
3230 int nviews = displayed_views();
3231 int next_view = (current_view + 1) % nviews;
3232
3233 if (next_view == current_view) {
3234 report("Only one view is displayed");
3235 break;
3236 }
3237
3238 current_view = next_view;
3239 /* Blur out the title of the previous view. */
3240 update_view_title(view);
3241 report("");
3242 break;
3243 }
3244 case REQ_REFRESH:
3245 report("Refreshing is not yet supported for the %s view", view->name);
3246 break;
3247
3248 case REQ_MAXIMIZE:
3249 if (displayed_views() == 2)
3250 maximize_view(view);
3251 break;
3252
3253 case REQ_TOGGLE_LINENO:
3254 toggle_view_option(&opt_line_number, "line numbers");
3255 break;
3256
3257 case REQ_TOGGLE_DATE:
3258 toggle_view_option(&opt_date, "date display");
3259 break;
3260
3261 case REQ_TOGGLE_AUTHOR:
3262 toggle_view_option(&opt_author, "author display");
3263 break;
3264
3265 case REQ_TOGGLE_REV_GRAPH:
3266 toggle_view_option(&opt_rev_graph, "revision graph display");
3267 break;
3268
3269 case REQ_TOGGLE_REFS:
3270 toggle_view_option(&opt_show_refs, "reference display");
3271 break;
3272
3273 case REQ_SEARCH:
3274 case REQ_SEARCH_BACK:
3275 search_view(view, request);
3276 break;
3277
3278 case REQ_FIND_NEXT:
3279 case REQ_FIND_PREV:
3280 find_next(view, request);
3281 break;
3282
3283 case REQ_STOP_LOADING:
3284 for (i = 0; i < ARRAY_SIZE(views); i++) {
3285 view = &views[i];
3286 if (view->pipe)
3287 report("Stopped loading the %s view", view->name),
3288 end_update(view, TRUE);
3289 }
3290 break;
3291
3292 case REQ_SHOW_VERSION:
3293 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
3294 return TRUE;
3295
3296 case REQ_SCREEN_REDRAW:
3297 redraw_display(TRUE);
3298 break;
3299
3300 case REQ_EDIT:
3301 report("Nothing to edit");
3302 break;
3303
3304 case REQ_ENTER:
3305 report("Nothing to enter");
3306 break;
3307
3308 case REQ_VIEW_CLOSE:
3309 /* XXX: Mark closed views by letting view->parent point to the
3310 * view itself. Parents to closed view should never be
3311 * followed. */
3312 if (view->parent &&
3313 view->parent->parent != view->parent) {
3314 maximize_view(view->parent);
3315 view->parent = view;
3316 break;
3317 }
3318 /* Fall-through */
3319 case REQ_QUIT:
3320 return FALSE;
3321
3322 default:
3323 report("Unknown key, press 'h' for help");
3324 return TRUE;
3325 }
3326
3327 return TRUE;
3328 }
3329
3330
3331 /*
3332 * View backend utilities
3333 */
3334
3335 static void
3336 parse_timezone(time_t *time, const char *zone)
3337 {
3338 long tz;
3339
3340 tz = ('0' - zone[1]) * 60 * 60 * 10;
3341 tz += ('0' - zone[2]) * 60 * 60;
3342 tz += ('0' - zone[3]) * 60;
3343 tz += ('0' - zone[4]);
3344
3345 if (zone[0] == '-')
3346 tz = -tz;
3347
3348 *time -= tz;
3349 }
3350
3351 /* Parse author lines where the name may be empty:
3352 * author <email@address.tld> 1138474660 +0100
3353 */
3354 static void
3355 parse_author_line(char *ident, char *author, size_t authorsize, struct tm *tm)
3356 {
3357 char *nameend = strchr(ident, '<');
3358 char *emailend = strchr(ident, '>');
3359
3360 if (nameend && emailend)
3361 *nameend = *emailend = 0;
3362 ident = chomp_string(ident);
3363 if (!*ident) {
3364 if (nameend)
3365 ident = chomp_string(nameend + 1);
3366 if (!*ident)
3367 ident = "Unknown";
3368 }
3369
3370 string_ncopy_do(author, authorsize, ident, strlen(ident));
3371
3372 /* Parse epoch and timezone */
3373 if (emailend && emailend[1] == ' ') {
3374 char *secs = emailend + 2;
3375 char *zone = strchr(secs, ' ');
3376 time_t time = (time_t) atol(secs);
3377
3378 if (zone && strlen(zone) == STRING_SIZE(" +0700"))
3379 parse_timezone(&time, zone + 1);
3380
3381 gmtime_r(&time, tm);
3382 }
3383 }
3384
3385 static enum input_status
3386 select_commit_parent_handler(void *data, char *buf, int c)
3387 {
3388 size_t parents = *(size_t *) data;
3389 int parent = 0;
3390
3391 if (!isdigit(c))
3392 return INPUT_SKIP;
3393
3394 if (*buf)
3395 parent = atoi(buf) * 10;
3396 parent += c - '0';
3397
3398 if (parent > parents)
3399 return INPUT_SKIP;
3400 return INPUT_OK;
3401 }
3402
3403 static bool
3404 select_commit_parent(const char *id, char rev[SIZEOF_REV], const char *path)
3405 {
3406 char buf[SIZEOF_STR * 4];
3407 const char *revlist_argv[] = {
3408 "git", "rev-list", "-1", "--parents", id, "--", path, NULL
3409 };
3410 int parents;
3411
3412 if (!run_io_buf(revlist_argv, buf, sizeof(buf)) ||
3413 !*chomp_string(buf) ||
3414 (parents = (strlen(buf) / 40) - 1) < 0) {
3415 report("Failed to get parent information");
3416 return FALSE;
3417
3418 } else if (parents == 0) {
3419 if (path)
3420 report("Path '%s' does not exist in the parent", path);
3421 else
3422 report("The selected commit has no parents");
3423 return FALSE;
3424 }
3425
3426 if (parents > 1) {
3427 char prompt[SIZEOF_STR];
3428 char *result;
3429
3430 if (!string_format(prompt, "Which parent? [1..%d] ", parents))
3431 return FALSE;
3432 result = prompt_input(prompt, select_commit_parent_handler, &parents);
3433 if (!result)
3434 return FALSE;
3435 parents = atoi(result);
3436 }
3437
3438 string_copy_rev(rev, &buf[41 * parents]);
3439 return TRUE;
3440 }
3441
3442 /*
3443 * Pager backend
3444 */
3445
3446 static bool
3447 pager_draw(struct view *view, struct line *line, unsigned int lineno)
3448 {
3449 char text[SIZEOF_STR];
3450
3451 if (opt_line_number && draw_lineno(view, lineno))
3452 return TRUE;
3453
3454 string_expand(text, sizeof(text), line->data, opt_tab_size);
3455 draw_text(view, line->type, text, TRUE);
3456 return TRUE;
3457 }
3458
3459 static bool
3460 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
3461 {
3462 const char *describe_argv[] = { "git", "describe", commit_id, NULL };
3463 char refbuf[SIZEOF_STR];
3464 char *ref = NULL;
3465
3466 if (run_io_buf(describe_argv, refbuf, sizeof(refbuf)))
3467 ref = chomp_string(refbuf);
3468
3469 if (!ref || !*ref)
3470 return TRUE;
3471
3472 /* This is the only fatal call, since it can "corrupt" the buffer. */
3473 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
3474 return FALSE;
3475
3476 return TRUE;
3477 }
3478
3479 static void
3480 add_pager_refs(struct view *view, struct line *line)
3481 {
3482 char buf[SIZEOF_STR];
3483 char *commit_id = (char *)line->data + STRING_SIZE("commit ");
3484 struct ref **refs;
3485 size_t bufpos = 0, refpos = 0;
3486 const char *sep = "Refs: ";
3487 bool is_tag = FALSE;
3488
3489 assert(line->type == LINE_COMMIT);
3490
3491 refs = get_refs(commit_id);
3492 if (!refs) {
3493 if (view == VIEW(REQ_VIEW_DIFF))
3494 goto try_add_describe_ref;
3495 return;
3496 }
3497
3498 do {
3499 struct ref *ref = refs[refpos];
3500 const char *fmt = ref->tag ? "%s[%s]" :
3501 ref->remote ? "%s<%s>" : "%s%s";
3502
3503 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
3504 return;
3505 sep = ", ";
3506 if (ref->tag)
3507 is_tag = TRUE;
3508 } while (refs[refpos++]->next);
3509
3510 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
3511 try_add_describe_ref:
3512 /* Add <tag>-g<commit_id> "fake" reference. */
3513 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
3514 return;
3515 }
3516
3517 if (bufpos == 0)
3518 return;
3519
3520 add_line_text(view, buf, LINE_PP_REFS);
3521 }
3522
3523 static bool
3524 pager_read(struct view *view, char *data)
3525 {
3526 struct line *line;
3527
3528 if (!data)
3529 return TRUE;
3530
3531 line = add_line_text(view, data, get_line_type(data));
3532 if (!line)
3533 return FALSE;
3534
3535 if (line->type == LINE_COMMIT &&
3536 (view == VIEW(REQ_VIEW_DIFF) ||
3537 view == VIEW(REQ_VIEW_LOG)))
3538 add_pager_refs(view, line);
3539
3540 return TRUE;
3541 }
3542
3543 static enum request
3544 pager_request(struct view *view, enum request request, struct line *line)
3545 {
3546 int split = 0;
3547
3548 if (request != REQ_ENTER)
3549 return request;
3550
3551 if (line->type == LINE_COMMIT &&
3552 (view == VIEW(REQ_VIEW_LOG) ||
3553 view == VIEW(REQ_VIEW_PAGER))) {
3554 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
3555 split = 1;
3556 }
3557
3558 /* Always scroll the view even if it was split. That way
3559 * you can use Enter to scroll through the log view and
3560 * split open each commit diff. */
3561 scroll_view(view, REQ_SCROLL_LINE_DOWN);
3562
3563 /* FIXME: A minor workaround. Scrolling the view will call report("")
3564 * but if we are scrolling a non-current view this won't properly
3565 * update the view title. */
3566 if (split)
3567 update_view_title(view);
3568
3569 return REQ_NONE;
3570 }
3571
3572 static bool
3573 pager_grep(struct view *view, struct line *line)
3574 {
3575 regmatch_t pmatch;
3576 char *text = line->data;
3577
3578 if (!*text)
3579 return FALSE;
3580
3581 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
3582 return FALSE;
3583
3584 return TRUE;
3585 }
3586
3587 static void
3588 pager_select(struct view *view, struct line *line)
3589 {
3590 if (line->type == LINE_COMMIT) {
3591 char *text = (char *)line->data + STRING_SIZE("commit ");
3592
3593 if (view != VIEW(REQ_VIEW_PAGER))
3594 string_copy_rev(view->ref, text);
3595 string_copy_rev(ref_commit, text);
3596 }
3597 }
3598
3599 static struct view_ops pager_ops = {
3600 "line",
3601 NULL,
3602 NULL,
3603 pager_read,
3604 pager_draw,
3605 pager_request,
3606 pager_grep,
3607 pager_select,
3608 };
3609
3610 static const char *log_argv[SIZEOF_ARG] = {
3611 "git", "log", "--no-color", "--cc", "--stat", "-n100", "%(head)", NULL
3612 };
3613
3614 static enum request
3615 log_request(struct view *view, enum request request, struct line *line)
3616 {
3617 switch (request) {
3618 case REQ_REFRESH:
3619 load_refs();
3620 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
3621 return REQ_NONE;
3622 default:
3623 return pager_request(view, request, line);
3624 }
3625 }
3626
3627 static struct view_ops log_ops = {
3628 "line",
3629 log_argv,
3630 NULL,
3631 pager_read,
3632 pager_draw,
3633 log_request,
3634 pager_grep,
3635 pager_select,
3636 };
3637
3638 static const char *diff_argv[SIZEOF_ARG] = {
3639 "git", "show", "--pretty=fuller", "--no-color", "--root",
3640 "--patch-with-stat", "--find-copies-harder", "-C", "%(commit)", NULL
3641 };
3642
3643 static struct view_ops diff_ops = {
3644 "line",
3645 diff_argv,
3646 NULL,
3647 pager_read,
3648 pager_draw,
3649 pager_request,
3650 pager_grep,
3651 pager_select,
3652 };
3653
3654 /*
3655 * Help backend
3656 */
3657
3658 static bool
3659 help_open(struct view *view)
3660 {
3661 char buf[SIZEOF_STR];
3662 size_t bufpos;
3663 int i;
3664
3665 if (view->lines > 0)
3666 return TRUE;
3667
3668 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3669
3670 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3671 const char *key;
3672
3673 if (req_info[i].request == REQ_NONE)
3674 continue;
3675
3676 if (!req_info[i].request) {
3677 add_line_text(view, "", LINE_DEFAULT);
3678 add_line_text(view, req_info[i].help, LINE_DEFAULT);
3679 continue;
3680 }
3681
3682 key = get_key(req_info[i].request);
3683 if (!*key)
3684 key = "(no key defined)";
3685
3686 for (bufpos = 0; bufpos <= req_info[i].namelen; bufpos++) {
3687 buf[bufpos] = tolower(req_info[i].name[bufpos]);
3688 if (buf[bufpos] == '_')
3689 buf[bufpos] = '-';
3690 }
3691
3692 add_line_format(view, LINE_DEFAULT, " %-25s %-20s %s",
3693 key, buf, req_info[i].help);
3694 }
3695
3696 if (run_requests) {
3697 add_line_text(view, "", LINE_DEFAULT);
3698 add_line_text(view, "External commands:", LINE_DEFAULT);
3699 }
3700
3701 for (i = 0; i < run_requests; i++) {
3702 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3703 const char *key;
3704 int argc;
3705
3706 if (!req)
3707 continue;
3708
3709 key = get_key_name(req->key);
3710 if (!*key)
3711 key = "(no key defined)";
3712
3713 for (bufpos = 0, argc = 0; req->argv[argc]; argc++)
3714 if (!string_format_from(buf, &bufpos, "%s%s",
3715 argc ? " " : "", req->argv[argc]))
3716 return REQ_NONE;
3717
3718 add_line_format(view, LINE_DEFAULT, " %-10s %-14s `%s`",
3719 keymap_table[req->keymap].name, key, buf);
3720 }
3721
3722 return TRUE;
3723 }
3724
3725 static struct view_ops help_ops = {
3726 "line",
3727 NULL,
3728 help_open,
3729 NULL,
3730 pager_draw,
3731 pager_request,
3732 pager_grep,
3733 pager_select,
3734 };
3735
3736
3737 /*
3738 * Tree backend
3739 */
3740
3741 struct tree_stack_entry {
3742 struct tree_stack_entry *prev; /* Entry below this in the stack */
3743 unsigned long lineno; /* Line number to restore */
3744 char *name; /* Position of name in opt_path */
3745 };
3746
3747 /* The top of the path stack. */
3748 static struct tree_stack_entry *tree_stack = NULL;
3749 unsigned long tree_lineno = 0;
3750
3751 static void
3752 pop_tree_stack_entry(void)
3753 {
3754 struct tree_stack_entry *entry = tree_stack;
3755
3756 tree_lineno = entry->lineno;
3757 entry->name[0] = 0;
3758 tree_stack = entry->prev;
3759 free(entry);
3760 }
3761
3762 static void
3763 push_tree_stack_entry(const char *name, unsigned long lineno)
3764 {
3765 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3766 size_t pathlen = strlen(opt_path);
3767
3768 if (!entry)
3769 return;
3770
3771 entry->prev = tree_stack;
3772 entry->name = opt_path + pathlen;
3773 tree_stack = entry;
3774
3775 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3776 pop_tree_stack_entry();
3777 return;
3778 }
3779
3780 /* Move the current line to the first tree entry. */
3781 tree_lineno = 1;
3782 entry->lineno = lineno;
3783 }
3784
3785 /* Parse output from git-ls-tree(1):
3786 *
3787 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3788 */
3789
3790 #define SIZEOF_TREE_ATTR \
3791 STRING_SIZE("100644 blob f931e1d229c3e185caad4449bf5b66ed72462657\t")
3792
3793 #define SIZEOF_TREE_MODE \
3794 STRING_SIZE("100644 ")
3795
3796 #define TREE_ID_OFFSET \
3797 STRING_SIZE("100644 blob ")
3798
3799 struct tree_entry {
3800 char id[SIZEOF_REV];
3801 mode_t mode;
3802 struct tm time; /* Date from the author ident. */
3803 char author[75]; /* Author of the commit. */
3804 char name[1];
3805 };
3806
3807 static const char *
3808 tree_path(struct line *line)
3809 {
3810 return ((struct tree_entry *) line->data)->name;
3811 }
3812
3813
3814 static int
3815 tree_compare_entry(struct line *line1, struct line *line2)
3816 {
3817 if (line1->type != line2->type)
3818 return line1->type == LINE_TREE_DIR ? -1 : 1;
3819 return strcmp(tree_path(line1), tree_path(line2));
3820 }
3821
3822 static struct line *
3823 tree_entry(struct view *view, enum line_type type, const char *path,
3824 const char *mode, const char *id)
3825 {
3826 struct tree_entry *entry = calloc(1, sizeof(*entry) + strlen(path));
3827 struct line *line = entry ? add_line_data(view, entry, type) : NULL;
3828
3829 if (!entry || !line) {
3830 free(entry);
3831 return NULL;
3832 }
3833
3834 strncpy(entry->name, path, strlen(path));
3835 if (mode)
3836 entry->mode = strtoul(mode, NULL, 8);
3837 if (id)
3838 string_copy_rev(entry->id, id);
3839
3840 return line;
3841 }
3842
3843 static bool
3844 tree_read_date(struct view *view, char *text, bool *read_date)
3845 {
3846 static char author_name[SIZEOF_STR];
3847 static struct tm author_time;
3848
3849 if (!text && *read_date) {
3850 *read_date = FALSE;
3851 return TRUE;
3852
3853 } else if (!text) {
3854 char *path = *opt_path ? opt_path : ".";
3855 /* Find next entry to process */
3856 const char *log_file[] = {
3857 "git", "log", "--no-color", "--pretty=raw",
3858 "--cc", "--raw", view->id, "--", path, NULL
3859 };
3860 struct io io = {};
3861
3862 if (!view->lines) {
3863 tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL);
3864 report("Tree is empty");
3865 return TRUE;
3866 }
3867
3868 if (!run_io_rd(&io, log_file, FORMAT_NONE)) {
3869 report("Failed to load tree data");
3870 return TRUE;
3871 }
3872
3873 done_io(view->pipe);
3874 view->io = io;
3875 *read_date = TRUE;
3876 return FALSE;
3877
3878 } else if (*text == 'a' && get_line_type(text) == LINE_AUTHOR) {
3879 parse_author_line(text + STRING_SIZE("author "),
3880 author_name, sizeof(author_name), &author_time);
3881
3882 } else if (*text == ':') {
3883 char *pos;
3884 size_t annotated = 1;
3885 size_t i;
3886
3887 pos = strchr(text, '\t');
3888 if (!pos)
3889 return TRUE;
3890 text = pos + 1;
3891 if (*opt_prefix && !strncmp(text, opt_prefix, strlen(opt_prefix)))
3892 text += strlen(opt_prefix);
3893 if (*opt_path && !strncmp(text, opt_path, strlen(opt_path)))
3894 text += strlen(opt_path);
3895 pos = strchr(text, '/');
3896 if (pos)
3897 *pos = 0;
3898
3899 for (i = 1; i < view->lines; i++) {
3900 struct line *line = &view->line[i];
3901 struct tree_entry *entry = line->data;
3902
3903 annotated += !!*entry->author;
3904 if (*entry->author || strcmp(entry->name, text))
3905 continue;
3906
3907 string_copy(entry->author, author_name);
3908 memcpy(&entry->time, &author_time, sizeof(entry->time));
3909 line->dirty = 1;
3910 break;
3911 }
3912
3913 if (annotated == view->lines)
3914 kill_io(view->pipe);
3915 }
3916 return TRUE;
3917 }
3918
3919 static bool
3920 tree_read(struct view *view, char *text)
3921 {
3922 static bool read_date = FALSE;
3923 struct tree_entry *data;
3924 struct line *entry, *line;
3925 enum line_type type;
3926 size_t textlen = text ? strlen(text) : 0;
3927 char *path = text + SIZEOF_TREE_ATTR;
3928
3929 if (read_date || !text)
3930 return tree_read_date(view, text, &read_date);
3931
3932 if (textlen <= SIZEOF_TREE_ATTR)
3933 return FALSE;
3934 if (view->lines == 0 &&
3935 !tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL))
3936 return FALSE;
3937
3938 /* Strip the path part ... */
3939 if (*opt_path) {
3940 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3941 size_t striplen = strlen(opt_path);
3942
3943 if (pathlen > striplen)
3944 memmove(path, path + striplen,
3945 pathlen - striplen + 1);
3946
3947 /* Insert "link" to parent directory. */
3948 if (view->lines == 1 &&
3949 !tree_entry(view, LINE_TREE_DIR, "..", "040000", view->ref))
3950 return FALSE;
3951 }
3952
3953 type = text[SIZEOF_TREE_MODE] == 't' ? LINE_TREE_DIR : LINE_TREE_FILE;
3954 entry = tree_entry(view, type, path, text, text + TREE_ID_OFFSET);
3955 if (!entry)
3956 return FALSE;
3957 data = entry->data;
3958
3959 /* Skip "Directory ..." and ".." line. */
3960 for (line = &view->line[1 + !!*opt_path]; line < entry; line++) {
3961 if (tree_compare_entry(line, entry) <= 0)
3962 continue;
3963
3964 memmove(line + 1, line, (entry - line) * sizeof(*entry));
3965
3966 line->data = data;
3967 line->type = type;
3968 for (; line <= entry; line++)
3969 line->dirty = line->cleareol = 1;
3970 return TRUE;
3971 }
3972
3973 if (tree_lineno > view->lineno) {
3974 view->lineno = tree_lineno;
3975 tree_lineno = 0;
3976 }
3977
3978 return TRUE;
3979 }
3980
3981 static bool
3982 tree_draw(struct view *view, struct line *line, unsigned int lineno)
3983 {
3984 struct tree_entry *entry = line->data;
3985
3986 if (line->type == LINE_TREE_HEAD) {
3987 if (draw_text(view, line->type, "Directory path /", TRUE))
3988 return TRUE;
3989 } else {
3990 if (draw_mode(view, entry->mode))
3991 return TRUE;
3992
3993 if (opt_author && draw_author(view, entry->author))
3994 return TRUE;
3995
3996 if (opt_date && draw_date(view, *entry->author ? &entry->time : NULL))
3997 return TRUE;
3998 }
3999 if (draw_text(view, line->type, entry->name, TRUE))
4000 return TRUE;
4001 return TRUE;
4002 }
4003
4004 static void
4005 open_blob_editor()
4006 {
4007 char file[SIZEOF_STR] = "/tmp/tigblob.XXXXXX";
4008 int fd = mkstemp(file);
4009
4010 if (fd == -1)
4011 report("Failed to create temporary file");
4012 else if (!run_io_append(blob_ops.argv, FORMAT_ALL, fd))
4013 report("Failed to save blob data to file");
4014 else
4015 open_editor(FALSE, file);
4016 if (fd != -1)
4017 unlink(file);
4018 }
4019
4020 static enum request
4021 tree_request(struct view *view, enum request request, struct line *line)
4022 {
4023 enum open_flags flags;
4024
4025 switch (request) {
4026 case REQ_VIEW_BLAME:
4027 if (line->type != LINE_TREE_FILE) {
4028 report("Blame only supported for files");
4029 return REQ_NONE;
4030 }
4031
4032 string_copy(opt_ref, view->vid);
4033 return request;
4034
4035 case REQ_EDIT:
4036 if (line->type != LINE_TREE_FILE) {
4037 report("Edit only supported for files");
4038 } else if (!is_head_commit(view->vid)) {
4039 open_blob_editor();
4040 } else {
4041 open_editor(TRUE, opt_file);
4042 }
4043 return REQ_NONE;
4044
4045 case REQ_PARENT:
4046 if (!*opt_path) {
4047 /* quit view if at top of tree */
4048 return REQ_VIEW_CLOSE;
4049 }
4050 /* fake 'cd ..' */
4051 line = &view->line[1];
4052 break;
4053
4054 case REQ_ENTER:
4055 break;
4056
4057 default:
4058 return request;
4059 }
4060
4061 /* Cleanup the stack if the tree view is at a different tree. */
4062 while (!*opt_path && tree_stack)
4063 pop_tree_stack_entry();
4064
4065 switch (line->type) {
4066 case LINE_TREE_DIR:
4067 /* Depending on whether it is a subdirectory or parent link
4068 * mangle the path buffer. */
4069 if (line == &view->line[1] && *opt_path) {
4070 pop_tree_stack_entry();
4071
4072 } else {
4073 const char *basename = tree_path(line);
4074
4075 push_tree_stack_entry(basename, view->lineno);
4076 }
4077
4078 /* Trees and subtrees share the same ID, so they are not not
4079 * unique like blobs. */
4080 flags = OPEN_RELOAD;
4081 request = REQ_VIEW_TREE;
4082 break;
4083
4084 case LINE_TREE_FILE:
4085 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4086 request = REQ_VIEW_BLOB;
4087 break;
4088
4089 default:
4090 return REQ_NONE;
4091 }
4092
4093 open_view(view, request, flags);
4094 if (request == REQ_VIEW_TREE)
4095 view->lineno = tree_lineno;
4096
4097 return REQ_NONE;
4098 }
4099
4100 static void
4101 tree_select(struct view *view, struct line *line)
4102 {
4103 struct tree_entry *entry = line->data;
4104
4105 if (line->type == LINE_TREE_FILE) {
4106 string_copy_rev(ref_blob, entry->id);
4107 string_format(opt_file, "%s%s", opt_path, tree_path(line));
4108
4109 } else if (line->type != LINE_TREE_DIR) {
4110 return;
4111 }
4112
4113 string_copy_rev(view->ref, entry->id);
4114 }
4115
4116 static const char *tree_argv[SIZEOF_ARG] = {
4117 "git", "ls-tree", "%(commit)", "%(directory)", NULL
4118 };
4119
4120 static struct view_ops tree_ops = {
4121 "file",
4122 tree_argv,
4123 NULL,
4124 tree_read,
4125 tree_draw,
4126 tree_request,
4127 pager_grep,
4128 tree_select,
4129 };
4130
4131 static bool
4132 blob_read(struct view *view, char *line)
4133 {
4134 if (!line)
4135 return TRUE;
4136 return add_line_text(view, line, LINE_DEFAULT) != NULL;
4137 }
4138
4139 static enum request
4140 blob_request(struct view *view, enum request request, struct line *line)
4141 {
4142 switch (request) {
4143 case REQ_EDIT:
4144 open_blob_editor();
4145 return REQ_NONE;
4146 default:
4147 return pager_request(view, request, line);
4148 }
4149 }
4150
4151 static const char *blob_argv[SIZEOF_ARG] = {
4152 "git", "cat-file", "blob", "%(blob)", NULL
4153 };
4154
4155 static struct view_ops blob_ops = {
4156 "line",
4157 blob_argv,
4158 NULL,
4159 blob_read,
4160 pager_draw,
4161 blob_request,
4162 pager_grep,
4163 pager_select,
4164 };
4165
4166 /*
4167 * Blame backend
4168 *
4169 * Loading the blame view is a two phase job:
4170 *
4171 * 1. File content is read either using opt_file from the
4172 * filesystem or using git-cat-file.
4173 * 2. Then blame information is incrementally added by
4174 * reading output from git-blame.
4175 */
4176
4177 static const char *blame_head_argv[] = {
4178 "git", "blame", "--incremental", "--", "%(file)", NULL
4179 };
4180
4181 static const char *blame_ref_argv[] = {
4182 "git", "blame", "--incremental", "%(ref)", "--", "%(file)", NULL
4183 };
4184
4185 static const char *blame_cat_file_argv[] = {
4186 "git", "cat-file", "blob", "%(ref):%(file)", NULL
4187 };
4188
4189 struct blame_commit {
4190 char id[SIZEOF_REV]; /* SHA1 ID. */
4191 char title[128]; /* First line of the commit message. */
4192 char author[75]; /* Author of the commit. */
4193 struct tm time; /* Date from the author ident. */
4194 char filename[128]; /* Name of file. */
4195 bool has_previous; /* Was a "previous" line detected. */
4196 };
4197
4198 struct blame {
4199 struct blame_commit *commit;
4200 unsigned long lineno;
4201 char text[1];
4202 };
4203
4204 static bool
4205 blame_open(struct view *view)
4206 {
4207 if (*opt_ref || !io_open(&view->io, opt_file)) {
4208 if (!run_io_rd(&view->io, blame_cat_file_argv, FORMAT_ALL))
4209 return FALSE;
4210 }
4211
4212 setup_update(view, opt_file);
4213 string_format(view->ref, "%s ...", opt_file);
4214
4215 return TRUE;
4216 }
4217
4218 static struct blame_commit *
4219 get_blame_commit(struct view *view, const char *id)
4220 {
4221 size_t i;
4222
4223 for (i = 0; i < view->lines; i++) {
4224 struct blame *blame = view->line[i].data;
4225
4226 if (!blame->commit)
4227 continue;
4228
4229 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
4230 return blame->commit;
4231 }
4232
4233 {
4234 struct blame_commit *commit = calloc(1, sizeof(*commit));
4235
4236 if (commit)
4237 string_ncopy(commit->id, id, SIZEOF_REV);
4238 return commit;
4239 }
4240 }
4241
4242 static bool
4243 parse_number(const char **posref, size_t *number, size_t min, size_t max)
4244 {
4245 const char *pos = *posref;
4246
4247 *posref = NULL;
4248 pos = strchr(pos + 1, ' ');
4249 if (!pos || !isdigit(pos[1]))
4250 return FALSE;
4251 *number = atoi(pos + 1);
4252 if (*number < min || *number > max)
4253 return FALSE;
4254
4255 *posref = pos;
4256 return TRUE;
4257 }
4258
4259 static struct blame_commit *
4260 parse_blame_commit(struct view *view, const char *text, int *blamed)
4261 {
4262 struct blame_commit *commit;
4263 struct blame *blame;
4264 const char *pos = text + SIZEOF_REV - 2;
4265 size_t orig_lineno = 0;
4266 size_t lineno;
4267 size_t group;
4268
4269 if (strlen(text) <= SIZEOF_REV || pos[1] != ' ')
4270 return NULL;
4271
4272 if (!parse_number(&pos, &orig_lineno, 1, 9999999) ||
4273 !parse_number(&pos, &lineno, 1, view->lines) ||
4274 !parse_number(&pos, &group, 1, view->lines - lineno + 1))
4275 return NULL;
4276
4277 commit = get_blame_commit(view, text);
4278 if (!commit)
4279 return NULL;
4280
4281 *blamed += group;
4282 while (group--) {
4283 struct line *line = &view->line[lineno + group - 1];
4284
4285 blame = line->data;
4286 blame->commit = commit;
4287 blame->lineno = orig_lineno + group - 1;
4288 line->dirty = 1;
4289 }
4290
4291 return commit;
4292 }
4293
4294 static bool
4295 blame_read_file(struct view *view, const char *line, bool *read_file)
4296 {
4297 if (!line) {
4298 const char **argv = *opt_ref ? blame_ref_argv : blame_head_argv;
4299 struct io io = {};
4300
4301 if (view->lines == 0 && !view->parent)
4302 die("No blame exist for %s", view->vid);
4303
4304 if (view->lines == 0 || !run_io_rd(&io, argv, FORMAT_ALL)) {
4305 report("Failed to load blame data");
4306 return TRUE;
4307 }
4308
4309 done_io(view->pipe);
4310 view->io = io;
4311 *read_file = FALSE;
4312 return FALSE;
4313
4314 } else {
4315 size_t linelen = strlen(line);
4316 struct blame *blame = malloc(sizeof(*blame) + linelen);
4317
4318 if (!blame)
4319 return FALSE;
4320
4321 blame->commit = NULL;
4322 strncpy(blame->text, line, linelen);
4323 blame->text[linelen] = 0;
4324 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
4325 }
4326 }
4327
4328 static bool
4329 match_blame_header(const char *name, char **line)
4330 {
4331 size_t namelen = strlen(name);
4332 bool matched = !strncmp(name, *line, namelen);
4333
4334 if (matched)
4335 *line += namelen;
4336
4337 return matched;
4338 }
4339
4340 static bool
4341 blame_read(struct view *view, char *line)
4342 {
4343 static struct blame_commit *commit = NULL;
4344 static int blamed = 0;
4345 static time_t author_time;
4346 static bool read_file = TRUE;
4347
4348 if (read_file)
4349 return blame_read_file(view, line, &read_file);
4350
4351 if (!line) {
4352 /* Reset all! */
4353 commit = NULL;
4354 blamed = 0;
4355 read_file = TRUE;
4356 string_format(view->ref, "%s", view->vid);
4357 if (view_is_displayed(view)) {
4358 update_view_title(view);
4359 redraw_view_from(view, 0);
4360 }
4361 return TRUE;
4362 }
4363
4364 if (!commit) {
4365 commit = parse_blame_commit(view, line, &blamed);
4366 string_format(view->ref, "%s %2d%%", view->vid,
4367 view->lines ? blamed * 100 / view->lines : 0);
4368
4369 } else if (match_blame_header("author ", &line)) {
4370 string_ncopy(commit->author, line, strlen(line));
4371
4372 } else if (match_blame_header("author-time ", &line)) {
4373 author_time = (time_t) atol(line);
4374
4375 } else if (match_blame_header("author-tz ", &line)) {
4376 parse_timezone(&author_time, line);
4377 gmtime_r(&author_time, &commit->time);
4378
4379 } else if (match_blame_header("summary ", &line)) {
4380 string_ncopy(commit->title, line, strlen(line));
4381
4382 } else if (match_blame_header("previous ", &line)) {
4383 commit->has_previous = TRUE;
4384
4385 } else if (match_blame_header("filename ", &line)) {
4386 string_ncopy(commit->filename, line, strlen(line));
4387 commit = NULL;
4388 }
4389
4390 return TRUE;
4391 }
4392
4393 static bool
4394 blame_draw(struct view *view, struct line *line, unsigned int lineno)
4395 {
4396 struct blame *blame = line->data;
4397 struct tm *time = NULL;
4398 const char *id = NULL, *author = NULL;
4399 char text[SIZEOF_STR];
4400
4401 if (blame->commit && *blame->commit->filename) {
4402 id = blame->commit->id;
4403 author = blame->commit->author;
4404 time = &blame->commit->time;
4405 }
4406
4407 if (opt_date && draw_date(view, time))
4408 return TRUE;
4409
4410 if (opt_author && draw_author(view, author))
4411 return TRUE;
4412
4413 if (draw_field(view, LINE_BLAME_ID, id, ID_COLS, FALSE))
4414 return TRUE;
4415
4416 if (draw_lineno(view, lineno))
4417 return TRUE;
4418
4419 string_expand(text, sizeof(text), blame->text, opt_tab_size);
4420 draw_text(view, LINE_DEFAULT, text, TRUE);
4421 return TRUE;
4422 }
4423
4424 static bool
4425 check_blame_commit(struct blame *blame, bool check_null_id)
4426 {
4427 if (!blame->commit)
4428 report("Commit data not loaded yet");
4429 else if (check_null_id && !strcmp(blame->commit->id, NULL_ID))
4430 report("No commit exist for the selected line");
4431 else
4432 return TRUE;
4433 return FALSE;
4434 }
4435
4436 static void
4437 setup_blame_parent_line(struct view *view, struct blame *blame)
4438 {
4439 const char *diff_tree_argv[] = {
4440 "git", "diff-tree", "-U0", blame->commit->id,
4441 "--", blame->commit->filename, NULL
4442 };
4443 struct io io = {};
4444 int parent_lineno = -1;
4445 int blamed_lineno = -1;
4446 char *line;
4447
4448 if (!run_io(&io, diff_tree_argv, NULL, IO_RD))
4449 return;
4450
4451 while ((line = io_get(&io, '\n', TRUE))) {
4452 if (*line == '@') {
4453 char *pos = strchr(line, '+');
4454
4455 parent_lineno = atoi(line + 4);
4456 if (pos)
4457 blamed_lineno = atoi(pos + 1);
4458
4459 } else if (*line == '+' && parent_lineno != -1) {
4460 if (blame->lineno == blamed_lineno - 1 &&
4461 !strcmp(blame->text, line + 1)) {
4462 view->lineno = parent_lineno ? parent_lineno - 1 : 0;
4463 break;
4464 }
4465 blamed_lineno++;
4466 }
4467 }
4468
4469 done_io(&io);
4470 }
4471
4472 static enum request
4473 blame_request(struct view *view, enum request request, struct line *line)
4474 {
4475 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4476 struct blame *blame = line->data;
4477
4478 switch (request) {
4479 case REQ_VIEW_BLAME:
4480 if (check_blame_commit(blame, TRUE)) {
4481 string_copy(opt_ref, blame->commit->id);
4482 string_copy(opt_file, blame->commit->filename);
4483 if (blame->lineno)
4484 view->lineno = blame->lineno;
4485 open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4486 }
4487 break;
4488
4489 case REQ_PARENT:
4490 if (check_blame_commit(blame, TRUE) &&
4491 select_commit_parent(blame->commit->id, opt_ref,
4492 blame->commit->filename)) {
4493 string_copy(opt_file, blame->commit->filename);
4494 setup_blame_parent_line(view, blame);
4495 open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4496 }
4497 break;
4498
4499 case REQ_ENTER:
4500 if (!check_blame_commit(blame, FALSE))
4501 break;
4502
4503 if (view_is_displayed(VIEW(REQ_VIEW_DIFF)) &&
4504 !strcmp(blame->commit->id, VIEW(REQ_VIEW_DIFF)->ref))
4505 break;
4506
4507 if (!strcmp(blame->commit->id, NULL_ID)) {
4508 struct view *diff = VIEW(REQ_VIEW_DIFF);
4509 const char *diff_index_argv[] = {
4510 "git", "diff-index", "--root", "--patch-with-stat",
4511 "-C", "-M", "HEAD", "--", view->vid, NULL
4512 };
4513
4514 if (!blame->commit->has_previous) {
4515 diff_index_argv[1] = "diff";
4516 diff_index_argv[2] = "--no-color";
4517 diff_index_argv[6] = "--";
4518 diff_index_argv[7] = "/dev/null";
4519 }
4520
4521 if (!prepare_update(diff, diff_index_argv, NULL, FORMAT_DASH)) {
4522 report("Failed to allocate diff command");
4523 break;
4524 }
4525 flags |= OPEN_PREPARED;
4526 }
4527
4528 open_view(view, REQ_VIEW_DIFF, flags);
4529 if (VIEW(REQ_VIEW_DIFF)->pipe && !strcmp(blame->commit->id, NULL_ID))
4530 string_copy_rev(VIEW(REQ_VIEW_DIFF)->ref, NULL_ID);
4531 break;
4532
4533 default:
4534 return request;
4535 }
4536
4537 return REQ_NONE;
4538 }
4539
4540 static bool
4541 blame_grep(struct view *view, struct line *line)
4542 {
4543 struct blame *blame = line->data;
4544 struct blame_commit *commit = blame->commit;
4545 regmatch_t pmatch;
4546
4547 #define MATCH(text, on) \
4548 (on && *text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4549
4550 if (commit) {
4551 char buf[DATE_COLS + 1];
4552
4553 if (MATCH(commit->title, 1) ||
4554 MATCH(commit->author, opt_author) ||
4555 MATCH(commit->id, opt_date))
4556 return TRUE;
4557
4558 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
4559 MATCH(buf, 1))
4560 return TRUE;
4561 }
4562
4563 return MATCH(blame->text, 1);
4564
4565 #undef MATCH
4566 }
4567
4568 static void
4569 blame_select(struct view *view, struct line *line)
4570 {
4571 struct blame *blame = line->data;
4572 struct blame_commit *commit = blame->commit;
4573
4574 if (!commit)
4575 return;
4576
4577 if (!strcmp(commit->id, NULL_ID))
4578 string_ncopy(ref_commit, "HEAD", 4);
4579 else
4580 string_copy_rev(ref_commit, commit->id);
4581 }
4582
4583 static struct view_ops blame_ops = {
4584 "line",
4585 NULL,
4586 blame_open,
4587 blame_read,
4588 blame_draw,
4589 blame_request,
4590 blame_grep,
4591 blame_select,
4592 };
4593
4594 /*
4595 * Status backend
4596 */
4597
4598 struct status {
4599 char status;
4600 struct {
4601 mode_t mode;
4602 char rev[SIZEOF_REV];
4603 char name[SIZEOF_STR];
4604 } old;
4605 struct {
4606 mode_t mode;
4607 char rev[SIZEOF_REV];
4608 char name[SIZEOF_STR];
4609 } new;
4610 };
4611
4612 static char status_onbranch[SIZEOF_STR];
4613 static struct status stage_status;
4614 static enum line_type stage_line_type;
4615 static size_t stage_chunks;
4616 static int *stage_chunk;
4617
4618 /* This should work even for the "On branch" line. */
4619 static inline bool
4620 status_has_none(struct view *view, struct line *line)
4621 {
4622 return line < view->line + view->lines && !line[1].data;
4623 }
4624
4625 /* Get fields from the diff line:
4626 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
4627 */
4628 static inline bool
4629 status_get_diff(struct status *file, const char *buf, size_t bufsize)
4630 {
4631 const char *old_mode = buf + 1;
4632 const char *new_mode = buf + 8;
4633 const char *old_rev = buf + 15;
4634 const char *new_rev = buf + 56;
4635 const char *status = buf + 97;
4636
4637 if (bufsize < 98 ||
4638 old_mode[-1] != ':' ||
4639 new_mode[-1] != ' ' ||
4640 old_rev[-1] != ' ' ||
4641 new_rev[-1] != ' ' ||
4642 status[-1] != ' ')
4643 return FALSE;
4644
4645 file->status = *status;
4646
4647 string_copy_rev(file->old.rev, old_rev);
4648 string_copy_rev(file->new.rev, new_rev);
4649
4650 file->old.mode = strtoul(old_mode, NULL, 8);
4651 file->new.mode = strtoul(new_mode, NULL, 8);
4652
4653 file->old.name[0] = file->new.name[0] = 0;
4654
4655 return TRUE;
4656 }
4657
4658 static bool
4659 status_run(struct view *view, const char *argv[], char status, enum line_type type)
4660 {
4661 struct status *unmerged = NULL;
4662 char *buf;
4663 struct io io = {};
4664
4665 if (!run_io(&io, argv, NULL, IO_RD))
4666 return FALSE;
4667
4668 add_line_data(view, NULL, type);
4669
4670 while ((buf = io_get(&io, 0, TRUE))) {
4671 struct status *file = unmerged;
4672
4673 if (!file) {
4674 file = calloc(1, sizeof(*file));
4675 if (!file || !add_line_data(view, file, type))
4676 goto error_out;
4677 }
4678
4679 /* Parse diff info part. */
4680 if (status) {
4681 file->status = status;
4682 if (status == 'A')
4683 string_copy(file->old.rev, NULL_ID);
4684
4685 } else if (!file->status || file == unmerged) {
4686 if (!status_get_diff(file, buf, strlen(buf)))
4687 goto error_out;
4688
4689 buf = io_get(&io, 0, TRUE);
4690 if (!buf)
4691 break;
4692
4693 /* Collapse all modified entries that follow an
4694 * associated unmerged entry. */
4695 if (unmerged == file) {
4696 unmerged->status = 'U';
4697 unmerged = NULL;
4698 } else if (file->status == 'U') {
4699 unmerged = file;
4700 }
4701 }
4702
4703 /* Grab the old name for rename/copy. */
4704 if (!*file->old.name &&
4705 (file->status == 'R' || file->status == 'C')) {
4706 string_ncopy(file->old.name, buf, strlen(buf));
4707
4708 buf = io_get(&io, 0, TRUE);
4709 if (!buf)
4710 break;
4711 }
4712
4713 /* git-ls-files just delivers a NUL separated list of
4714 * file names similar to the second half of the
4715 * git-diff-* output. */
4716 string_ncopy(file->new.name, buf, strlen(buf));
4717 if (!*file->old.name)
4718 string_copy(file->old.name, file->new.name);
4719 file = NULL;
4720 }
4721
4722 if (io_error(&io)) {
4723 error_out:
4724 done_io(&io);
4725 return FALSE;
4726 }
4727
4728 if (!view->line[view->lines - 1].data)
4729 add_line_data(view, NULL, LINE_STAT_NONE);
4730
4731 done_io(&io);
4732 return TRUE;
4733 }
4734
4735 /* Don't show unmerged entries in the staged section. */
4736 static const char *status_diff_index_argv[] = {
4737 "git", "diff-index", "-z", "--diff-filter=ACDMRTXB",
4738 "--cached", "-M", "HEAD", NULL
4739 };
4740
4741 static const char *status_diff_files_argv[] = {
4742 "git", "diff-files", "-z", NULL
4743 };
4744
4745 static const char *status_list_other_argv[] = {
4746 "git", "ls-files", "-z", "--others", "--exclude-standard", NULL
4747 };
4748
4749 static const char *status_list_no_head_argv[] = {
4750 "git", "ls-files", "-z", "--cached", "--exclude-standard", NULL
4751 };
4752
4753 static const char *update_index_argv[] = {
4754 "git", "update-index", "-q", "--unmerged", "--refresh", NULL
4755 };
4756
4757 /* Restore the previous line number to stay in the context or select a
4758 * line with something that can be updated. */
4759 static void
4760 status_restore(struct view *view)
4761 {
4762 if (view->p_lineno >= view->lines)
4763 view->p_lineno = view->lines - 1;
4764 while (view->p_lineno < view->lines && !view->line[view->p_lineno].data)
4765 view->p_lineno++;
4766 while (view->p_lineno > 0 && !view->line[view->p_lineno].data)
4767 view->p_lineno--;
4768
4769 /* If the above fails, always skip the "On branch" line. */
4770 if (view->p_lineno < view->lines)
4771 view->lineno = view->p_lineno;
4772 else
4773 view->lineno = 1;
4774
4775 if (view->lineno < view->offset)
4776 view->offset = view->lineno;
4777 else if (view->offset + view->height <= view->lineno)
4778 view->offset = view->lineno - view->height + 1;
4779
4780 view->p_restore = FALSE;
4781 }
4782
4783 static void
4784 status_update_onbranch(void)
4785 {
4786 static const char *paths[][2] = {
4787 { "rebase-apply/rebasing", "Rebasing" },
4788 { "rebase-apply/applying", "Applying mailbox" },
4789 { "rebase-apply/", "Rebasing mailbox" },
4790 { "rebase-merge/interactive", "Interactive rebase" },
4791 { "rebase-merge/", "Rebase merge" },
4792 { "MERGE_HEAD", "Merging" },
4793 { "BISECT_LOG", "Bisecting" },
4794 { "HEAD", "On branch" },
4795 };
4796 char buf[SIZEOF_STR];
4797 struct stat stat;
4798 int i;
4799
4800 if (is_initial_commit()) {
4801 string_copy(status_onbranch, "Initial commit");
4802 return;
4803 }
4804
4805 for (i = 0; i < ARRAY_SIZE(paths); i++) {
4806 char *head = opt_head;
4807
4808 if (!string_format(buf, "%s/%s", opt_git_dir, paths[i][0]) ||
4809 lstat(buf, &stat) < 0)
4810 continue;
4811
4812 if (!*opt_head) {
4813 struct io io = {};
4814
4815 if (string_format(buf, "%s/rebase-merge/head-name", opt_git_dir) &&
4816 io_open(&io, buf) &&
4817 io_read_buf(&io, buf, sizeof(buf))) {
4818 head = chomp_string(buf);
4819 if (!prefixcmp(head, "refs/heads/"))
4820 head += STRING_SIZE("refs/heads/");
4821 }
4822 }
4823
4824 if (!string_format(status_onbranch, "%s %s", paths[i][1], head))
4825 string_copy(status_onbranch, opt_head);
4826 return;
4827 }
4828
4829 string_copy(status_onbranch, "Not currently on any branch");
4830 }
4831
4832 /* First parse staged info using git-diff-index(1), then parse unstaged
4833 * info using git-diff-files(1), and finally untracked files using
4834 * git-ls-files(1). */
4835 static bool
4836 status_open(struct view *view)
4837 {
4838 reset_view(view);
4839
4840 add_line_data(view, NULL, LINE_STAT_HEAD);
4841 status_update_onbranch();
4842
4843 run_io_bg(update_index_argv);
4844
4845 if (is_initial_commit()) {
4846 if (!status_run(view, status_list_no_head_argv, 'A', LINE_STAT_STAGED))
4847 return FALSE;
4848 } else if (!status_run(view, status_diff_index_argv, 0, LINE_STAT_STAGED)) {
4849 return FALSE;
4850 }
4851
4852 if (!status_run(view, status_diff_files_argv, 0, LINE_STAT_UNSTAGED) ||
4853 !status_run(view, status_list_other_argv, '?', LINE_STAT_UNTRACKED))
4854 return FALSE;
4855
4856 /* Restore the exact position or use the specialized restore
4857 * mode? */
4858 if (!view->p_restore)
4859 status_restore(view);
4860 return TRUE;
4861 }
4862
4863 static bool
4864 status_draw(struct view *view, struct line *line, unsigned int lineno)
4865 {
4866 struct status *status = line->data;
4867 enum line_type type;
4868 const char *text;
4869
4870 if (!status) {
4871 switch (line->type) {
4872 case LINE_STAT_STAGED:
4873 type = LINE_STAT_SECTION;
4874 text = "Changes to be committed:";
4875 break;
4876
4877 case LINE_STAT_UNSTAGED:
4878 type = LINE_STAT_SECTION;
4879 text = "Changed but not updated:";
4880 break;
4881
4882 case LINE_STAT_UNTRACKED:
4883 type = LINE_STAT_SECTION;
4884 text = "Untracked files:";
4885 break;
4886
4887 case LINE_STAT_NONE:
4888 type = LINE_DEFAULT;
4889 text = " (no files)";
4890 break;
4891
4892 case LINE_STAT_HEAD:
4893 type = LINE_STAT_HEAD;
4894 text = status_onbranch;
4895 break;
4896
4897 default:
4898 return FALSE;
4899 }
4900 } else {
4901 static char buf[] = { '?', ' ', ' ', ' ', 0 };
4902
4903 buf[0] = status->status;
4904 if (draw_text(view, line->type, buf, TRUE))
4905 return TRUE;
4906 type = LINE_DEFAULT;
4907 text = status->new.name;
4908 }
4909
4910 draw_text(view, type, text, TRUE);
4911 return TRUE;
4912 }
4913
4914 static enum request
4915 status_load_error(struct view *view, struct view *stage, const char *path)
4916 {
4917 if (displayed_views() == 2 || display[current_view] != view)
4918 maximize_view(view);
4919 report("Failed to load '%s': %s", path, io_strerror(&stage->io));
4920 return REQ_NONE;
4921 }
4922
4923 static enum request
4924 status_enter(struct view *view, struct line *line)
4925 {
4926 struct status *status = line->data;
4927 const char *oldpath = status ? status->old.name : NULL;
4928 /* Diffs for unmerged entries are empty when passing the new
4929 * path, so leave it empty. */
4930 const char *newpath = status && status->status != 'U' ? status->new.name : NULL;
4931 const char *info;
4932 enum open_flags split;
4933 struct view *stage = VIEW(REQ_VIEW_STAGE);
4934
4935 if (line->type == LINE_STAT_NONE ||
4936 (!status && line[1].type == LINE_STAT_NONE)) {
4937 report("No file to diff");
4938 return REQ_NONE;
4939 }
4940
4941 switch (line->type) {
4942 case LINE_STAT_STAGED:
4943 if (is_initial_commit()) {
4944 const char *no_head_diff_argv[] = {
4945 "git", "diff", "--no-color", "--patch-with-stat",
4946 "--", "/dev/null", newpath, NULL
4947 };
4948
4949 if (!prepare_update(stage, no_head_diff_argv, opt_cdup, FORMAT_DASH))
4950 return status_load_error(view, stage, newpath);
4951 } else {
4952 const char *index_show_argv[] = {
4953 "git", "diff-index", "--root", "--patch-with-stat",
4954 "-C", "-M", "--cached", "HEAD", "--",
4955 oldpath, newpath, NULL
4956 };
4957
4958 if (!prepare_update(stage, index_show_argv, opt_cdup, FORMAT_DASH))
4959 return status_load_error(view, stage, newpath);
4960 }
4961
4962 if (status)
4963 info = "Staged changes to %s";
4964 else
4965 info = "Staged changes";
4966 break;
4967
4968 case LINE_STAT_UNSTAGED:
4969 {
4970 const char *files_show_argv[] = {
4971 "git", "diff-files", "--root", "--patch-with-stat",
4972 "-C", "-M", "--", oldpath, newpath, NULL
4973 };
4974
4975 if (!prepare_update(stage, files_show_argv, opt_cdup, FORMAT_DASH))
4976 return status_load_error(view, stage, newpath);
4977 if (status)
4978 info = "Unstaged changes to %s";
4979 else
4980 info = "Unstaged changes";
4981 break;
4982 }
4983 case LINE_STAT_UNTRACKED:
4984 if (!newpath) {
4985 report("No file to show");
4986 return REQ_NONE;
4987 }
4988
4989 if (!suffixcmp(status->new.name, -1, "/")) {
4990 report("Cannot display a directory");
4991 return REQ_NONE;
4992 }
4993
4994 if (!prepare_update_file(stage, newpath))
4995 return status_load_error(view, stage, newpath);
4996 info = "Untracked file %s";
4997 break;
4998
4999 case LINE_STAT_HEAD:
5000 return REQ_NONE;
5001
5002 default:
5003 die("line type %d not handled in switch", line->type);
5004 }
5005
5006 split = view_is_displayed(view) ? OPEN_SPLIT : 0;
5007 open_view(view, REQ_VIEW_STAGE, OPEN_PREPARED | split);
5008 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
5009 if (status) {
5010 stage_status = *status;
5011 } else {
5012 memset(&stage_status, 0, sizeof(stage_status));
5013 }
5014
5015 stage_line_type = line->type;
5016 stage_chunks = 0;
5017 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
5018 }
5019
5020 return REQ_NONE;
5021 }
5022
5023 static bool
5024 status_exists(struct status *status, enum line_type type)
5025 {
5026 struct view *view = VIEW(REQ_VIEW_STATUS);
5027 unsigned long lineno;
5028
5029 for (lineno = 0; lineno < view->lines; lineno++) {
5030 struct line *line = &view->line[lineno];
5031 struct status *pos = line->data;
5032
5033 if (line->type != type)
5034 continue;
5035 if (!pos && (!status || !status->status) && line[1].data) {
5036 select_view_line(view, lineno);
5037 return TRUE;
5038 }
5039 if (pos && !strcmp(status->new.name, pos->new.name)) {
5040 select_view_line(view, lineno);
5041 return TRUE;
5042 }
5043 }
5044
5045 return FALSE;
5046 }
5047
5048
5049 static bool
5050 status_update_prepare(struct io *io, enum line_type type)
5051 {
5052 const char *staged_argv[] = {
5053 "git", "update-index", "-z", "--index-info", NULL
5054 };
5055 const char *others_argv[] = {
5056 "git", "update-index", "-z", "--add", "--remove", "--stdin", NULL
5057 };
5058
5059 switch (type) {
5060 case LINE_STAT_STAGED:
5061 return run_io(io, staged_argv, opt_cdup, IO_WR);
5062
5063 case LINE_STAT_UNSTAGED:
5064 return run_io(io, others_argv, opt_cdup, IO_WR);
5065
5066 case LINE_STAT_UNTRACKED:
5067 return run_io(io, others_argv, NULL, IO_WR);
5068
5069 default:
5070 die("line type %d not handled in switch", type);
5071 return FALSE;
5072 }
5073 }
5074
5075 static bool
5076 status_update_write(struct io *io, struct status *status, enum line_type type)
5077 {
5078 char buf[SIZEOF_STR];
5079 size_t bufsize = 0;
5080
5081 switch (type) {
5082 case LINE_STAT_STAGED:
5083 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
5084 status->old.mode,
5085 status->old.rev,
5086 status->old.name, 0))
5087 return FALSE;
5088 break;
5089
5090 case LINE_STAT_UNSTAGED:
5091 case LINE_STAT_UNTRACKED:
5092 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
5093 return FALSE;
5094 break;
5095
5096 default:
5097 die("line type %d not handled in switch", type);
5098 }
5099
5100 return io_write(io, buf, bufsize);
5101 }
5102
5103 static bool
5104 status_update_file(struct status *status, enum line_type type)
5105 {
5106 struct io io = {};
5107 bool result;
5108
5109 if (!status_update_prepare(&io, type))
5110 return FALSE;
5111
5112 result = status_update_write(&io, status, type);
5113 return done_io(&io) && result;
5114 }
5115
5116 static bool
5117 status_update_files(struct view *view, struct line *line)
5118 {
5119 char buf[sizeof(view->ref)];
5120 struct io io = {};
5121 bool result = TRUE;
5122 struct line *pos = view->line + view->lines;
5123 int files = 0;
5124 int file, done;
5125
5126 if (!status_update_prepare(&io, line->type))
5127 return FALSE;
5128
5129 for (pos = line; pos < view->line + view->lines && pos->data; pos++)
5130 files++;
5131
5132 string_copy(buf, view->ref);
5133 for (file = 0, done = 5; result && file < files; line++, file++) {
5134 int almost_done = file * 100 / files;
5135
5136 if (almost_done > done) {
5137 done = almost_done;
5138 string_format(view->ref, "updating file %u of %u (%d%% done)",
5139 file, files, done);
5140 update_view_title(view);
5141 doupdate();
5142 }
5143 result = status_update_write(&io, line->data, line->type);
5144 }
5145 string_copy(view->ref, buf);
5146
5147 return done_io(&io) && result;
5148 }
5149
5150 static bool
5151 status_update(struct view *view)
5152 {
5153 struct line *line = &view->line[view->lineno];
5154
5155 assert(view->lines);
5156
5157 if (!line->data) {
5158 /* This should work even for the "On branch" line. */
5159 if (line < view->line + view->lines && !line[1].data) {
5160 report("Nothing to update");
5161 return FALSE;
5162 }
5163
5164 if (!status_update_files(view, line + 1)) {
5165 report("Failed to update file status");
5166 return FALSE;
5167 }
5168
5169 } else if (!status_update_file(line->data, line->type)) {
5170 report("Failed to update file status");
5171 return FALSE;
5172 }
5173
5174 return TRUE;
5175 }
5176
5177 static bool
5178 status_revert(struct status *status, enum line_type type, bool has_none)
5179 {
5180 if (!status || type != LINE_STAT_UNSTAGED) {
5181 if (type == LINE_STAT_STAGED) {
5182 report("Cannot revert changes to staged files");
5183 } else if (type == LINE_STAT_UNTRACKED) {
5184 report("Cannot revert changes to untracked files");
5185 } else if (has_none) {
5186 report("Nothing to revert");
5187 } else {
5188 report("Cannot revert changes to multiple files");
5189 }
5190 return FALSE;
5191
5192 } else {
5193 char mode[10] = "100644";
5194 const char *reset_argv[] = {
5195 "git", "update-index", "--cacheinfo", mode,
5196 status->old.rev, status->old.name, NULL
5197 };
5198 const char *checkout_argv[] = {
5199 "git", "checkout", "--", status->old.name, NULL
5200 };
5201
5202 if (!prompt_yesno("Are you sure you want to overwrite any changes?"))
5203 return FALSE;
5204 string_format(mode, "%o", status->old.mode);
5205 return (status->status != 'U' || run_io_fg(reset_argv, opt_cdup)) &&
5206 run_io_fg(checkout_argv, opt_cdup);
5207 }
5208 }
5209
5210 static enum request
5211 status_request(struct view *view, enum request request, struct line *line)
5212 {
5213 struct status *status = line->data;
5214
5215 switch (request) {
5216 case REQ_STATUS_UPDATE:
5217 if (!status_update(view))
5218 return REQ_NONE;
5219 break;
5220
5221 case REQ_STATUS_REVERT:
5222 if (!status_revert(status, line->type, status_has_none(view, line)))
5223 return REQ_NONE;
5224 break;
5225
5226 case REQ_STATUS_MERGE:
5227 if (!status || status->status != 'U') {
5228 report("Merging only possible for files with unmerged status ('U').");
5229 return REQ_NONE;
5230 }
5231 open_mergetool(status->new.name);
5232 break;
5233
5234 case REQ_EDIT:
5235 if (!status)
5236 return request;
5237 if (status->status == 'D') {
5238 report("File has been deleted.");
5239 return REQ_NONE;
5240 }
5241
5242 open_editor(status->status != '?', status->new.name);
5243 break;
5244
5245 case REQ_VIEW_BLAME:
5246 if (status) {
5247 string_copy(opt_file, status->new.name);
5248 opt_ref[0] = 0;
5249 }
5250 return request;
5251
5252 case REQ_ENTER:
5253 /* After returning the status view has been split to
5254 * show the stage view. No further reloading is
5255 * necessary. */
5256 return status_enter(view, line);
5257
5258 case REQ_REFRESH:
5259 /* Simply reload the view. */
5260 break;
5261
5262 default:
5263 return request;
5264 }
5265
5266 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
5267
5268 return REQ_NONE;
5269 }
5270
5271 static void
5272 status_select(struct view *view, struct line *line)
5273 {
5274 struct status *status = line->data;
5275 char file[SIZEOF_STR] = "all files";
5276 const char *text;
5277 const char *key;
5278
5279 if (status && !string_format(file, "'%s'", status->new.name))
5280 return;
5281
5282 if (!status && line[1].type == LINE_STAT_NONE)
5283 line++;
5284
5285 switch (line->type) {
5286 case LINE_STAT_STAGED:
5287 text = "Press %s to unstage %s for commit";
5288 break;
5289
5290 case LINE_STAT_UNSTAGED:
5291 text = "Press %s to stage %s for commit";
5292 break;
5293
5294 case LINE_STAT_UNTRACKED:
5295 text = "Press %s to stage %s for addition";
5296 break;
5297
5298 case LINE_STAT_HEAD:
5299 case LINE_STAT_NONE:
5300 text = "Nothing to update";
5301 break;
5302
5303 default:
5304 die("line type %d not handled in switch", line->type);
5305 }
5306
5307 if (status && status->status == 'U') {
5308 text = "Press %s to resolve conflict in %s";
5309 key = get_key(REQ_STATUS_MERGE);
5310
5311 } else {
5312 key = get_key(REQ_STATUS_UPDATE);
5313 }
5314
5315 string_format(view->ref, text, key, file);
5316 }
5317
5318 static bool
5319 status_grep(struct view *view, struct line *line)
5320 {
5321 struct status *status = line->data;
5322 enum { S_STATUS, S_NAME, S_END } state;
5323 char buf[2] = "?";
5324 regmatch_t pmatch;
5325
5326 if (!status)
5327 return FALSE;
5328
5329 for (state = S_STATUS; state < S_END; state++) {
5330 const char *text;
5331
5332 switch (state) {
5333 case S_NAME: text = status->new.name; break;
5334 case S_STATUS:
5335 buf[0] = status->status;
5336 text = buf;
5337 break;
5338
5339 default:
5340 return FALSE;
5341 }
5342
5343 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5344 return TRUE;
5345 }
5346
5347 return FALSE;
5348 }
5349
5350 static struct view_ops status_ops = {
5351 "file",
5352 NULL,
5353 status_open,
5354 NULL,
5355 status_draw,
5356 status_request,
5357 status_grep,
5358 status_select,
5359 };
5360
5361
5362 static bool
5363 stage_diff_write(struct io *io, struct line *line, struct line *end)
5364 {
5365 while (line < end) {
5366 if (!io_write(io, line->data, strlen(line->data)) ||
5367 !io_write(io, "\n", 1))
5368 return FALSE;
5369 line++;
5370 if (line->type == LINE_DIFF_CHUNK ||
5371 line->type == LINE_DIFF_HEADER)
5372 break;
5373 }
5374
5375 return TRUE;
5376 }
5377
5378 static struct line *
5379 stage_diff_find(struct view *view, struct line *line, enum line_type type)
5380 {
5381 for (; view->line < line; line--)
5382 if (line->type == type)
5383 return line;
5384
5385 return NULL;
5386 }
5387
5388 static bool
5389 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
5390 {
5391 const char *apply_argv[SIZEOF_ARG] = {
5392 "git", "apply", "--whitespace=nowarn", NULL
5393 };
5394 struct line *diff_hdr;
5395 struct io io = {};
5396 int argc = 3;
5397
5398 diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
5399 if (!diff_hdr)
5400 return FALSE;
5401
5402 if (!revert)
5403 apply_argv[argc++] = "--cached";
5404 if (revert || stage_line_type == LINE_STAT_STAGED)
5405 apply_argv[argc++] = "-R";
5406 apply_argv[argc++] = "-";
5407 apply_argv[argc++] = NULL;
5408 if (!run_io(&io, apply_argv, opt_cdup, IO_WR))
5409 return FALSE;
5410
5411 if (!stage_diff_write(&io, diff_hdr, chunk) ||
5412 !stage_diff_write(&io, chunk, view->line + view->lines))
5413 chunk = NULL;
5414
5415 done_io(&io);
5416 run_io_bg(update_index_argv);
5417
5418 return chunk ? TRUE : FALSE;
5419 }
5420
5421 static bool
5422 stage_update(struct view *view, struct line *line)
5423 {
5424 struct line *chunk = NULL;
5425
5426 if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
5427 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5428
5429 if (chunk) {
5430 if (!stage_apply_chunk(view, chunk, FALSE)) {
5431 report("Failed to apply chunk");
5432 return FALSE;
5433 }
5434
5435 } else if (!stage_status.status) {
5436 view = VIEW(REQ_VIEW_STATUS);
5437
5438 for (line = view->line; line < view->line + view->lines; line++)
5439 if (line->type == stage_line_type)
5440 break;
5441
5442 if (!status_update_files(view, line + 1)) {
5443 report("Failed to update files");
5444 return FALSE;
5445 }
5446
5447 } else if (!status_update_file(&stage_status, stage_line_type)) {
5448 report("Failed to update file");
5449 return FALSE;
5450 }
5451
5452 return TRUE;
5453 }
5454
5455 static bool
5456 stage_revert(struct view *view, struct line *line)
5457 {
5458 struct line *chunk = NULL;
5459
5460 if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
5461 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5462
5463 if (chunk) {
5464 if (!prompt_yesno("Are you sure you want to revert changes?"))
5465 return FALSE;
5466
5467 if (!stage_apply_chunk(view, chunk, TRUE)) {
5468 report("Failed to revert chunk");
5469 return FALSE;
5470 }
5471 return TRUE;
5472
5473 } else {
5474 return status_revert(stage_status.status ? &stage_status : NULL,
5475 stage_line_type, FALSE);
5476 }
5477 }
5478
5479
5480 static void
5481 stage_next(struct view *view, struct line *line)
5482 {
5483 int i;
5484
5485 if (!stage_chunks) {
5486 static size_t alloc = 0;
5487 int *tmp;
5488
5489 for (line = view->line; line < view->line + view->lines; line++) {
5490 if (line->type != LINE_DIFF_CHUNK)
5491 continue;
5492
5493 tmp = realloc_items(stage_chunk, &alloc,
5494 stage_chunks, sizeof(*tmp));
5495 if (!tmp) {
5496 report("Allocation failure");
5497 return;
5498 }
5499
5500 stage_chunk = tmp;
5501 stage_chunk[stage_chunks++] = line - view->line;
5502 }
5503 }
5504
5505 for (i = 0; i < stage_chunks; i++) {
5506 if (stage_chunk[i] > view->lineno) {
5507 do_scroll_view(view, stage_chunk[i] - view->lineno);
5508 report("Chunk %d of %d", i + 1, stage_chunks);
5509 return;
5510 }
5511 }
5512
5513 report("No next chunk found");
5514 }
5515
5516 static enum request
5517 stage_request(struct view *view, enum request request, struct line *line)
5518 {
5519 switch (request) {
5520 case REQ_STATUS_UPDATE:
5521 if (!stage_update(view, line))
5522 return REQ_NONE;
5523 break;
5524
5525 case REQ_STATUS_REVERT:
5526 if (!stage_revert(view, line))
5527 return REQ_NONE;
5528 break;
5529
5530 case REQ_STAGE_NEXT:
5531 if (stage_line_type == LINE_STAT_UNTRACKED) {
5532 report("File is untracked; press %s to add",
5533 get_key(REQ_STATUS_UPDATE));
5534 return REQ_NONE;
5535 }
5536 stage_next(view, line);
5537 return REQ_NONE;
5538
5539 case REQ_EDIT:
5540 if (!stage_status.new.name[0])
5541 return request;
5542 if (stage_status.status == 'D') {
5543 report("File has been deleted.");
5544 return REQ_NONE;
5545 }
5546
5547 open_editor(stage_status.status != '?', stage_status.new.name);
5548 break;
5549
5550 case REQ_REFRESH:
5551 /* Reload everything ... */
5552 break;
5553
5554 case REQ_VIEW_BLAME:
5555 if (stage_status.new.name[0]) {
5556 string_copy(opt_file, stage_status.new.name);
5557 opt_ref[0] = 0;
5558 }
5559 return request;
5560
5561 case REQ_ENTER:
5562 return pager_request(view, request, line);
5563
5564 default:
5565 return request;
5566 }
5567
5568 VIEW(REQ_VIEW_STATUS)->p_restore = TRUE;
5569 open_view(view, REQ_VIEW_STATUS, OPEN_REFRESH);
5570
5571 /* Check whether the staged entry still exists, and close the
5572 * stage view if it doesn't. */
5573 if (!status_exists(&stage_status, stage_line_type)) {
5574 status_restore(VIEW(REQ_VIEW_STATUS));
5575 return REQ_VIEW_CLOSE;
5576 }
5577
5578 if (stage_line_type == LINE_STAT_UNTRACKED) {
5579 if (!suffixcmp(stage_status.new.name, -1, "/")) {
5580 report("Cannot display a directory");
5581 return REQ_NONE;
5582 }
5583
5584 if (!prepare_update_file(view, stage_status.new.name)) {
5585 report("Failed to open file: %s", strerror(errno));
5586 return REQ_NONE;
5587 }
5588 }
5589 open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
5590
5591 return REQ_NONE;
5592 }
5593
5594 static struct view_ops stage_ops = {
5595 "line",
5596 NULL,
5597 NULL,
5598 pager_read,
5599 pager_draw,
5600 stage_request,
5601 pager_grep,
5602 pager_select,
5603 };
5604
5605
5606 /*
5607 * Revision graph
5608 */
5609
5610 struct commit {
5611 char id[SIZEOF_REV]; /* SHA1 ID. */
5612 char title[128]; /* First line of the commit message. */
5613 char author[75]; /* Author of the commit. */
5614 struct tm time; /* Date from the author ident. */
5615 struct ref **refs; /* Repository references. */
5616 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
5617 size_t graph_size; /* The width of the graph array. */
5618 bool has_parents; /* Rewritten --parents seen. */
5619 };
5620
5621 /* Size of rev graph with no "padding" columns */
5622 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
5623
5624 struct rev_graph {
5625 struct rev_graph *prev, *next, *parents;
5626 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
5627 size_t size;
5628 struct commit *commit;
5629 size_t pos;
5630 unsigned int boundary:1;
5631 };
5632
5633 /* Parents of the commit being visualized. */
5634 static struct rev_graph graph_parents[4];
5635
5636 /* The current stack of revisions on the graph. */
5637 static struct rev_graph graph_stacks[4] = {
5638 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
5639 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
5640 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
5641 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
5642 };
5643
5644 static inline bool
5645 graph_parent_is_merge(struct rev_graph *graph)
5646 {
5647 return graph->parents->size > 1;
5648 }
5649
5650 static inline void
5651 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
5652 {
5653 struct commit *commit = graph->commit;
5654
5655 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
5656 commit->graph[commit->graph_size++] = symbol;
5657 }
5658
5659 static void
5660 clear_rev_graph(struct rev_graph *graph)
5661 {
5662 graph->boundary = 0;
5663 graph->size = graph->pos = 0;
5664 graph->commit = NULL;
5665 memset(graph->parents, 0, sizeof(*graph->parents));
5666 }
5667
5668 static void
5669 done_rev_graph(struct rev_graph *graph)
5670 {
5671 if (graph_parent_is_merge(graph) &&
5672 graph->pos < graph->size - 1 &&
5673 graph->next->size == graph->size + graph->parents->size - 1) {
5674 size_t i = graph->pos + graph->parents->size - 1;
5675
5676 graph->commit->graph_size = i * 2;
5677 while (i < graph->next->size - 1) {
5678 append_to_rev_graph(graph, ' ');
5679 append_to_rev_graph(graph, '\\');
5680 i++;
5681 }
5682 }
5683
5684 clear_rev_graph(graph);
5685 }
5686
5687 static void
5688 push_rev_graph(struct rev_graph *graph, const char *parent)
5689 {
5690 int i;
5691
5692 /* "Collapse" duplicate parents lines.
5693 *
5694 * FIXME: This needs to also update update the drawn graph but
5695 * for now it just serves as a method for pruning graph lines. */
5696 for (i = 0; i < graph->size; i++)
5697 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
5698 return;
5699
5700 if (graph->size < SIZEOF_REVITEMS) {
5701 string_copy_rev(graph->rev[graph->size++], parent);
5702 }
5703 }
5704
5705 static chtype
5706 get_rev_graph_symbol(struct rev_graph *graph)
5707 {
5708 chtype symbol;
5709
5710 if (graph->boundary)
5711 symbol = REVGRAPH_BOUND;
5712 else if (graph->parents->size == 0)
5713 symbol = REVGRAPH_INIT;
5714 else if (graph_parent_is_merge(graph))
5715 symbol = REVGRAPH_MERGE;
5716 else if (graph->pos >= graph->size)
5717 symbol = REVGRAPH_BRANCH;
5718 else
5719 symbol = REVGRAPH_COMMIT;
5720
5721 return symbol;
5722 }
5723
5724 static void
5725 draw_rev_graph(struct rev_graph *graph)
5726 {
5727 struct rev_filler {
5728 chtype separator, line;
5729 };
5730 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
5731 static struct rev_filler fillers[] = {
5732 { ' ', '|' },
5733 { '`', '.' },
5734 { '\'', ' ' },
5735 { '/', ' ' },
5736 };
5737 chtype symbol = get_rev_graph_symbol(graph);
5738 struct rev_filler *filler;
5739 size_t i;
5740
5741 if (opt_line_graphics)
5742 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
5743
5744 filler = &fillers[DEFAULT];
5745
5746 for (i = 0; i < graph->pos; i++) {
5747 append_to_rev_graph(graph, filler->line);
5748 if (graph_parent_is_merge(graph->prev) &&
5749 graph->prev->pos == i)
5750 filler = &fillers[RSHARP];
5751
5752 append_to_rev_graph(graph, filler->separator);
5753 }
5754
5755 /* Place the symbol for this revision. */
5756 append_to_rev_graph(graph, symbol);
5757
5758 if (graph->prev->size > graph->size)
5759 filler = &fillers[RDIAG];
5760 else
5761 filler = &fillers[DEFAULT];
5762
5763 i++;
5764
5765 for (; i < graph->size; i++) {
5766 append_to_rev_graph(graph, filler->separator);
5767 append_to_rev_graph(graph, filler->line);
5768 if (graph_parent_is_merge(graph->prev) &&
5769 i < graph->prev->pos + graph->parents->size)
5770 filler = &fillers[RSHARP];
5771 if (graph->prev->size > graph->size)
5772 filler = &fillers[LDIAG];
5773 }
5774
5775 if (graph->prev->size > graph->size) {
5776 append_to_rev_graph(graph, filler->separator);
5777 if (filler->line != ' ')
5778 append_to_rev_graph(graph, filler->line);
5779 }
5780 }
5781
5782 /* Prepare the next rev graph */
5783 static void
5784 prepare_rev_graph(struct rev_graph *graph)
5785 {
5786 size_t i;
5787
5788 /* First, traverse all lines of revisions up to the active one. */
5789 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5790 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5791 break;
5792
5793 push_rev_graph(graph->next, graph->rev[graph->pos]);
5794 }
5795
5796 /* Interleave the new revision parent(s). */
5797 for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5798 push_rev_graph(graph->next, graph->parents->rev[i]);
5799
5800 /* Lastly, put any remaining revisions. */
5801 for (i = graph->pos + 1; i < graph->size; i++)
5802 push_rev_graph(graph->next, graph->rev[i]);
5803 }
5804
5805 static void
5806 update_rev_graph(struct view *view, struct rev_graph *graph)
5807 {
5808 /* If this is the finalizing update ... */
5809 if (graph->commit)
5810 prepare_rev_graph(graph);
5811
5812 /* Graph visualization needs a one rev look-ahead,
5813 * so the first update doesn't visualize anything. */
5814 if (!graph->prev->commit)
5815 return;
5816
5817 if (view->lines > 2)
5818 view->line[view->lines - 3].dirty = 1;
5819 if (view->lines > 1)
5820 view->line[view->lines - 2].dirty = 1;
5821 draw_rev_graph(graph->prev);
5822 done_rev_graph(graph->prev->prev);
5823 }
5824
5825
5826 /*
5827 * Main view backend
5828 */
5829
5830 static const char *main_argv[SIZEOF_ARG] = {
5831 "git", "log", "--no-color", "--pretty=raw", "--parents",
5832 "--topo-order", "%(head)", NULL
5833 };
5834
5835 static bool
5836 main_draw(struct view *view, struct line *line, unsigned int lineno)
5837 {
5838 struct commit *commit = line->data;
5839
5840 if (!*commit->author)
5841 return FALSE;
5842
5843 if (opt_date && draw_date(view, &commit->time))
5844 return TRUE;
5845
5846 if (opt_author && draw_author(view, commit->author))
5847 return TRUE;
5848
5849 if (opt_rev_graph && commit->graph_size &&
5850 draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5851 return TRUE;
5852
5853 if (opt_show_refs && commit->refs) {
5854 size_t i = 0;
5855
5856 do {
5857 enum line_type type;
5858
5859 if (commit->refs[i]->head)
5860 type = LINE_MAIN_HEAD;
5861 else if (commit->refs[i]->ltag)
5862 type = LINE_MAIN_LOCAL_TAG;
5863 else if (commit->refs[i]->tag)
5864 type = LINE_MAIN_TAG;
5865 else if (commit->refs[i]->tracked)
5866 type = LINE_MAIN_TRACKED;
5867 else if (commit->refs[i]->remote)
5868 type = LINE_MAIN_REMOTE;
5869 else
5870 type = LINE_MAIN_REF;
5871
5872 if (draw_text(view, type, "[", TRUE) ||
5873 draw_text(view, type, commit->refs[i]->name, TRUE) ||
5874 draw_text(view, type, "]", TRUE))
5875 return TRUE;
5876
5877 if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5878 return TRUE;
5879 } while (commit->refs[i++]->next);
5880 }
5881
5882 draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5883 return TRUE;
5884 }
5885
5886 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5887 static bool
5888 main_read(struct view *view, char *line)
5889 {
5890 static struct rev_graph *graph = graph_stacks;
5891 enum line_type type;
5892 struct commit *commit;
5893
5894 if (!line) {
5895 int i;
5896
5897 if (!view->lines && !view->parent)
5898 die("No revisions match the given arguments.");
5899 if (view->lines > 0) {
5900 commit = view->line[view->lines - 1].data;
5901 view->line[view->lines - 1].dirty = 1;
5902 if (!*commit->author) {
5903 view->lines--;
5904 free(commit);
5905 graph->commit = NULL;
5906 }
5907 }
5908 update_rev_graph(view, graph);
5909
5910 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5911 clear_rev_graph(&graph_stacks[i]);
5912 return TRUE;
5913 }
5914
5915 type = get_line_type(line);
5916 if (type == LINE_COMMIT) {
5917 commit = calloc(1, sizeof(struct commit));
5918 if (!commit)
5919 return FALSE;
5920
5921 line += STRING_SIZE("commit ");
5922 if (*line == '-') {
5923 graph->boundary = 1;
5924 line++;
5925 }
5926
5927 string_copy_rev(commit->id, line);
5928 commit->refs = get_refs(commit->id);
5929 graph->commit = commit;
5930 add_line_data(view, commit, LINE_MAIN_COMMIT);
5931
5932 while ((line = strchr(line, ' '))) {
5933 line++;
5934 push_rev_graph(graph->parents, line);
5935 commit->has_parents = TRUE;
5936 }
5937 return TRUE;
5938 }
5939
5940 if (!view->lines)
5941 return TRUE;
5942 commit = view->line[view->lines - 1].data;
5943
5944 switch (type) {
5945 case LINE_PARENT:
5946 if (commit->has_parents)
5947 break;
5948 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5949 break;
5950
5951 case LINE_AUTHOR:
5952 parse_author_line(line + STRING_SIZE("author "),
5953 commit->author, sizeof(commit->author),
5954 &commit->time);
5955 update_rev_graph(view, graph);
5956 graph = graph->next;
5957 break;
5958
5959 default:
5960 /* Fill in the commit title if it has not already been set. */
5961 if (commit->title[0])
5962 break;
5963
5964 /* Require titles to start with a non-space character at the
5965 * offset used by git log. */
5966 if (strncmp(line, " ", 4))
5967 break;
5968 line += 4;
5969 /* Well, if the title starts with a whitespace character,
5970 * try to be forgiving. Otherwise we end up with no title. */
5971 while (isspace(*line))
5972 line++;
5973 if (*line == '\0')
5974 break;
5975 /* FIXME: More graceful handling of titles; append "..." to
5976 * shortened titles, etc. */
5977
5978 string_expand(commit->title, sizeof(commit->title), line, 1);
5979 view->line[view->lines - 1].dirty = 1;
5980 }
5981
5982 return TRUE;
5983 }
5984
5985 static enum request
5986 main_request(struct view *view, enum request request, struct line *line)
5987 {
5988 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5989
5990 switch (request) {
5991 case REQ_ENTER:
5992 open_view(view, REQ_VIEW_DIFF, flags);
5993 break;
5994 case REQ_REFRESH:
5995 load_refs();
5996 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
5997 break;
5998 default:
5999 return request;
6000 }
6001
6002 return REQ_NONE;
6003 }
6004
6005 static bool
6006 grep_refs(struct ref **refs, regex_t *regex)
6007 {
6008 regmatch_t pmatch;
6009 size_t i = 0;
6010
6011 if (!refs)
6012 return FALSE;
6013 do {
6014 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
6015 return TRUE;
6016 } while (refs[i++]->next);
6017
6018 return FALSE;
6019 }
6020
6021 static bool
6022 main_grep(struct view *view, struct line *line)
6023 {
6024 struct commit *commit = line->data;
6025 enum { S_TITLE, S_AUTHOR, S_DATE, S_REFS, S_END } state;
6026 char buf[DATE_COLS + 1];
6027 regmatch_t pmatch;
6028
6029 for (state = S_TITLE; state < S_END; state++) {
6030 char *text;
6031
6032 switch (state) {
6033 case S_TITLE: text = commit->title; break;
6034 case S_AUTHOR:
6035 if (!opt_author)
6036 continue;
6037 text = commit->author;
6038 break;
6039 case S_DATE:
6040 if (!opt_date)
6041 continue;
6042 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
6043 continue;
6044 text = buf;
6045 break;
6046 case S_REFS:
6047 if (!opt_show_refs)
6048 continue;
6049 if (grep_refs(commit->refs, view->regex) == TRUE)
6050 return TRUE;
6051 continue;
6052 default:
6053 return FALSE;
6054 }
6055
6056 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
6057 return TRUE;
6058 }
6059
6060 return FALSE;
6061 }
6062
6063 static void
6064 main_select(struct view *view, struct line *line)
6065 {
6066 struct commit *commit = line->data;
6067
6068 string_copy_rev(view->ref, commit->id);
6069 string_copy_rev(ref_commit, view->ref);
6070 }
6071
6072 static struct view_ops main_ops = {
6073 "commit",
6074 main_argv,
6075 NULL,
6076 main_read,
6077 main_draw,
6078 main_request,
6079 main_grep,
6080 main_select,
6081 };
6082
6083
6084 /*
6085 * Unicode / UTF-8 handling
6086 *
6087 * NOTE: Much of the following code for dealing with Unicode is derived from
6088 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
6089 * src/intl/charset.c from the UTF-8 branch commit elinks-0.11.0-g31f2c28.
6090 */
6091
6092 static inline int
6093 unicode_width(unsigned long c)
6094 {
6095 if (c >= 0x1100 &&
6096 (c <= 0x115f /* Hangul Jamo */
6097 || c == 0x2329
6098 || c == 0x232a
6099 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
6100 /* CJK ... Yi */
6101 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
6102 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
6103 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
6104 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
6105 || (c >= 0xffe0 && c <= 0xffe6)
6106 || (c >= 0x20000 && c <= 0x2fffd)
6107 || (c >= 0x30000 && c <= 0x3fffd)))
6108 return 2;
6109
6110 if (c == '\t')
6111 return opt_tab_size;
6112
6113 return 1;
6114 }
6115
6116 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
6117 * Illegal bytes are set one. */
6118 static const unsigned char utf8_bytes[256] = {
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 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
6124 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1, 1,1,1,1,1,1,1,1,
6125 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,
6126 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,
6127 };
6128
6129 /* Decode UTF-8 multi-byte representation into a Unicode character. */
6130 static inline unsigned long
6131 utf8_to_unicode(const char *string, size_t length)
6132 {
6133 unsigned long unicode;
6134
6135 switch (length) {
6136 case 1:
6137 unicode = string[0];
6138 break;
6139 case 2:
6140 unicode = (string[0] & 0x1f) << 6;
6141 unicode += (string[1] & 0x3f);
6142 break;
6143 case 3:
6144 unicode = (string[0] & 0x0f) << 12;
6145 unicode += ((string[1] & 0x3f) << 6);
6146 unicode += (string[2] & 0x3f);
6147 break;
6148 case 4:
6149 unicode = (string[0] & 0x0f) << 18;
6150 unicode += ((string[1] & 0x3f) << 12);
6151 unicode += ((string[2] & 0x3f) << 6);
6152 unicode += (string[3] & 0x3f);
6153 break;
6154 case 5:
6155 unicode = (string[0] & 0x0f) << 24;
6156 unicode += ((string[1] & 0x3f) << 18);
6157 unicode += ((string[2] & 0x3f) << 12);
6158 unicode += ((string[3] & 0x3f) << 6);
6159 unicode += (string[4] & 0x3f);
6160 break;
6161 case 6:
6162 unicode = (string[0] & 0x01) << 30;
6163 unicode += ((string[1] & 0x3f) << 24);
6164 unicode += ((string[2] & 0x3f) << 18);
6165 unicode += ((string[3] & 0x3f) << 12);
6166 unicode += ((string[4] & 0x3f) << 6);
6167 unicode += (string[5] & 0x3f);
6168 break;
6169 default:
6170 die("Invalid Unicode length");
6171 }
6172
6173 /* Invalid characters could return the special 0xfffd value but NUL
6174 * should be just as good. */
6175 return unicode > 0xffff ? 0 : unicode;
6176 }
6177
6178 /* Calculates how much of string can be shown within the given maximum width
6179 * and sets trimmed parameter to non-zero value if all of string could not be
6180 * shown. If the reserve flag is TRUE, it will reserve at least one
6181 * trailing character, which can be useful when drawing a delimiter.
6182 *
6183 * Returns the number of bytes to output from string to satisfy max_width. */
6184 static size_t
6185 utf8_length(const char **start, size_t skip, int *width, size_t max_width, int *trimmed, bool reserve)
6186 {
6187 const char *string = *start;
6188 const char *end = strchr(string, '\0');
6189 unsigned char last_bytes = 0;
6190 size_t last_ucwidth = 0;
6191
6192 *width = 0;
6193 *trimmed = 0;
6194
6195 while (string < end) {
6196 int c = *(unsigned char *) string;
6197 unsigned char bytes = utf8_bytes[c];
6198 size_t ucwidth;
6199 unsigned long unicode;
6200
6201 if (string + bytes > end)
6202 break;
6203
6204 /* Change representation to figure out whether
6205 * it is a single- or double-width character. */
6206
6207 unicode = utf8_to_unicode(string, bytes);
6208 /* FIXME: Graceful handling of invalid Unicode character. */
6209 if (!unicode)
6210 break;
6211
6212 ucwidth = unicode_width(unicode);
6213 if (skip > 0) {
6214 skip -= ucwidth <= skip ? ucwidth : skip;
6215 *start += bytes;
6216 }
6217 *width += ucwidth;
6218 if (*width > max_width) {
6219 *trimmed = 1;
6220 *width -= ucwidth;
6221 if (reserve && *width == max_width) {
6222 string -= last_bytes;
6223 *width -= last_ucwidth;
6224 }
6225 break;
6226 }
6227
6228 string += bytes;
6229 last_bytes = ucwidth ? bytes : 0;
6230 last_ucwidth = ucwidth;
6231 }
6232
6233 return string - *start;
6234 }
6235
6236
6237 /*
6238 * Status management
6239 */
6240
6241 /* Whether or not the curses interface has been initialized. */
6242 static bool cursed = FALSE;
6243
6244 /* Terminal hacks and workarounds. */
6245 static bool use_scroll_redrawwin;
6246 static bool use_scroll_status_wclear;
6247
6248 /* The status window is used for polling keystrokes. */
6249 static WINDOW *status_win;
6250
6251 /* Reading from the prompt? */
6252 static bool input_mode = FALSE;
6253
6254 static bool status_empty = FALSE;
6255
6256 /* Update status and title window. */
6257 static void
6258 report(const char *msg, ...)
6259 {
6260 struct view *view = display[current_view];
6261
6262 if (input_mode)
6263 return;
6264
6265 if (!view) {
6266 char buf[SIZEOF_STR];
6267 va_list args;
6268
6269 va_start(args, msg);
6270 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
6271 buf[sizeof(buf) - 1] = 0;
6272 buf[sizeof(buf) - 2] = '.';
6273 buf[sizeof(buf) - 3] = '.';
6274 buf[sizeof(buf) - 4] = '.';
6275 }
6276 va_end(args);
6277 die("%s", buf);
6278 }
6279
6280 if (!status_empty || *msg) {
6281 va_list args;
6282
6283 va_start(args, msg);
6284
6285 wmove(status_win, 0, 0);
6286 if (view->has_scrolled && use_scroll_status_wclear)
6287 wclear(status_win);
6288 if (*msg) {
6289 vwprintw(status_win, msg, args);
6290 status_empty = FALSE;
6291 } else {
6292 status_empty = TRUE;
6293 }
6294 wclrtoeol(status_win);
6295 wnoutrefresh(status_win);
6296
6297 va_end(args);
6298 }
6299
6300 update_view_title(view);
6301 }
6302
6303 /* Controls when nodelay should be in effect when polling user input. */
6304 static void
6305 set_nonblocking_input(bool loading)
6306 {
6307 static unsigned int loading_views;
6308
6309 if ((loading == FALSE && loading_views-- == 1) ||
6310 (loading == TRUE && loading_views++ == 0))
6311 nodelay(status_win, loading);
6312 }
6313
6314 static void
6315 init_display(void)
6316 {
6317 const char *term;
6318 int x, y;
6319
6320 /* Initialize the curses library */
6321 if (isatty(STDIN_FILENO)) {
6322 cursed = !!initscr();
6323 opt_tty = stdin;
6324 } else {
6325 /* Leave stdin and stdout alone when acting as a pager. */
6326 opt_tty = fopen("/dev/tty", "r+");
6327 if (!opt_tty)
6328 die("Failed to open /dev/tty");
6329 cursed = !!newterm(NULL, opt_tty, opt_tty);
6330 }
6331
6332 if (!cursed)
6333 die("Failed to initialize curses");
6334
6335 nonl(); /* Disable conversion and detect newlines from input. */
6336 cbreak(); /* Take input chars one at a time, no wait for \n */
6337 noecho(); /* Don't echo input */
6338 leaveok(stdscr, FALSE);
6339
6340 if (has_colors())
6341 init_colors();
6342
6343 getmaxyx(stdscr, y, x);
6344 status_win = newwin(1, 0, y - 1, 0);
6345 if (!status_win)
6346 die("Failed to create status window");
6347
6348 /* Enable keyboard mapping */
6349 keypad(status_win, TRUE);
6350 wbkgdset(status_win, get_line_attr(LINE_STATUS));
6351
6352 TABSIZE = opt_tab_size;
6353 if (opt_line_graphics) {
6354 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
6355 }
6356
6357 term = getenv("XTERM_VERSION") ? NULL : getenv("COLORTERM");
6358 if (term && !strcmp(term, "gnome-terminal")) {
6359 /* In the gnome-terminal-emulator, the message from
6360 * scrolling up one line when impossible followed by
6361 * scrolling down one line causes corruption of the
6362 * status line. This is fixed by calling wclear. */
6363 use_scroll_status_wclear = TRUE;
6364 use_scroll_redrawwin = FALSE;
6365
6366 } else if (term && !strcmp(term, "xrvt-xpm")) {
6367 /* No problems with full optimizations in xrvt-(unicode)
6368 * and aterm. */
6369 use_scroll_status_wclear = use_scroll_redrawwin = FALSE;
6370
6371 } else {
6372 /* When scrolling in (u)xterm the last line in the
6373 * scrolling direction will update slowly. */
6374 use_scroll_redrawwin = TRUE;
6375 use_scroll_status_wclear = FALSE;
6376 }
6377 }
6378
6379 static int
6380 get_input(int prompt_position)
6381 {
6382 struct view *view;
6383 int i, key, cursor_y, cursor_x;
6384
6385 if (prompt_position)
6386 input_mode = TRUE;
6387
6388 while (TRUE) {
6389 foreach_view (view, i) {
6390 update_view(view);
6391 if (view_is_displayed(view) && view->has_scrolled &&
6392 use_scroll_redrawwin)
6393 redrawwin(view->win);
6394 view->has_scrolled = FALSE;
6395 }
6396
6397 /* Update the cursor position. */
6398 if (prompt_position) {
6399 getbegyx(status_win, cursor_y, cursor_x);
6400 cursor_x = prompt_position;
6401 } else {
6402 view = display[current_view];
6403 getbegyx(view->win, cursor_y, cursor_x);
6404 cursor_x = view->width - 1;
6405 cursor_y += view->lineno - view->offset;
6406 }
6407 setsyx(cursor_y, cursor_x);
6408
6409 /* Refresh, accept single keystroke of input */
6410 doupdate();
6411 key = wgetch(status_win);
6412
6413 /* wgetch() with nodelay() enabled returns ERR when
6414 * there's no input. */
6415 if (key == ERR) {
6416
6417 } else if (key == KEY_RESIZE) {
6418 int height, width;
6419
6420 getmaxyx(stdscr, height, width);
6421
6422 wresize(status_win, 1, width);
6423 mvwin(status_win, height - 1, 0);
6424 wnoutrefresh(status_win);
6425 resize_display();
6426 redraw_display(TRUE);
6427
6428 } else {
6429 input_mode = FALSE;
6430 return key;
6431 }
6432 }
6433 }
6434
6435 static char *
6436 prompt_input(const char *prompt, input_handler handler, void *data)
6437 {
6438 enum input_status status = INPUT_OK;
6439 static char buf[SIZEOF_STR];
6440 size_t pos = 0;
6441
6442 buf[pos] = 0;
6443
6444 while (status == INPUT_OK || status == INPUT_SKIP) {
6445 int key;
6446
6447 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
6448 wclrtoeol(status_win);
6449
6450 key = get_input(pos + 1);
6451 switch (key) {
6452 case KEY_RETURN:
6453 case KEY_ENTER:
6454 case '\n':
6455 status = pos ? INPUT_STOP : INPUT_CANCEL;
6456 break;
6457
6458 case KEY_BACKSPACE:
6459 if (pos > 0)
6460 buf[--pos] = 0;
6461 else
6462 status = INPUT_CANCEL;
6463 break;
6464
6465 case KEY_ESC:
6466 status = INPUT_CANCEL;
6467 break;
6468
6469 default:
6470 if (pos >= sizeof(buf)) {
6471 report("Input string too long");
6472 return NULL;
6473 }
6474
6475 status = handler(data, buf, key);
6476 if (status == INPUT_OK)
6477 buf[pos++] = (char) key;
6478 }
6479 }
6480
6481 /* Clear the status window */
6482 status_empty = FALSE;
6483 report("");
6484
6485 if (status == INPUT_CANCEL)
6486 return NULL;
6487
6488 buf[pos++] = 0;
6489
6490 return buf;
6491 }
6492
6493 static enum input_status
6494 prompt_yesno_handler(void *data, char *buf, int c)
6495 {
6496 if (c == 'y' || c == 'Y')
6497 return INPUT_STOP;
6498 if (c == 'n' || c == 'N')
6499 return INPUT_CANCEL;
6500 return INPUT_SKIP;
6501 }
6502
6503 static bool
6504 prompt_yesno(const char *prompt)
6505 {
6506 char prompt2[SIZEOF_STR];
6507
6508 if (!string_format(prompt2, "%s [Yy/Nn]", prompt))
6509 return FALSE;
6510
6511 return !!prompt_input(prompt2, prompt_yesno_handler, NULL);
6512 }
6513
6514 static enum input_status
6515 read_prompt_handler(void *data, char *buf, int c)
6516 {
6517 return isprint(c) ? INPUT_OK : INPUT_SKIP;
6518 }
6519
6520 static char *
6521 read_prompt(const char *prompt)
6522 {
6523 return prompt_input(prompt, read_prompt_handler, NULL);
6524 }
6525
6526 /*
6527 * Repository properties
6528 */
6529
6530 static struct ref *refs = NULL;
6531 static size_t refs_alloc = 0;
6532 static size_t refs_size = 0;
6533
6534 /* Id <-> ref store */
6535 static struct ref ***id_refs = NULL;
6536 static size_t id_refs_alloc = 0;
6537 static size_t id_refs_size = 0;
6538
6539 static int
6540 compare_refs(const void *ref1_, const void *ref2_)
6541 {
6542 const struct ref *ref1 = *(const struct ref **)ref1_;
6543 const struct ref *ref2 = *(const struct ref **)ref2_;
6544
6545 if (ref1->tag != ref2->tag)
6546 return ref2->tag - ref1->tag;
6547 if (ref1->ltag != ref2->ltag)
6548 return ref2->ltag - ref2->ltag;
6549 if (ref1->head != ref2->head)
6550 return ref2->head - ref1->head;
6551 if (ref1->tracked != ref2->tracked)
6552 return ref2->tracked - ref1->tracked;
6553 if (ref1->remote != ref2->remote)
6554 return ref2->remote - ref1->remote;
6555 return strcmp(ref1->name, ref2->name);
6556 }
6557
6558 static struct ref **
6559 get_refs(const char *id)
6560 {
6561 struct ref ***tmp_id_refs;
6562 struct ref **ref_list = NULL;
6563 size_t ref_list_alloc = 0;
6564 size_t ref_list_size = 0;
6565 size_t i;
6566
6567 for (i = 0; i < id_refs_size; i++)
6568 if (!strcmp(id, id_refs[i][0]->id))
6569 return id_refs[i];
6570
6571 tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
6572 sizeof(*id_refs));
6573 if (!tmp_id_refs)
6574 return NULL;
6575
6576 id_refs = tmp_id_refs;
6577
6578 for (i = 0; i < refs_size; i++) {
6579 struct ref **tmp;
6580
6581 if (strcmp(id, refs[i].id))
6582 continue;
6583
6584 tmp = realloc_items(ref_list, &ref_list_alloc,
6585 ref_list_size + 1, sizeof(*ref_list));
6586 if (!tmp) {
6587 if (ref_list)
6588 free(ref_list);
6589 return NULL;
6590 }
6591
6592 ref_list = tmp;
6593 ref_list[ref_list_size] = &refs[i];
6594 /* XXX: The properties of the commit chains ensures that we can
6595 * safely modify the shared ref. The repo references will
6596 * always be similar for the same id. */
6597 ref_list[ref_list_size]->next = 1;
6598
6599 ref_list_size++;
6600 }
6601
6602 if (ref_list) {
6603 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
6604 ref_list[ref_list_size - 1]->next = 0;
6605 id_refs[id_refs_size++] = ref_list;
6606 }
6607
6608 return ref_list;
6609 }
6610
6611 static int
6612 read_ref(char *id, size_t idlen, char *name, size_t namelen)
6613 {
6614 struct ref *ref;
6615 bool tag = FALSE;
6616 bool ltag = FALSE;
6617 bool remote = FALSE;
6618 bool tracked = FALSE;
6619 bool check_replace = FALSE;
6620 bool head = FALSE;
6621
6622 if (!prefixcmp(name, "refs/tags/")) {
6623 if (!suffixcmp(name, namelen, "^{}")) {
6624 namelen -= 3;
6625 name[namelen] = 0;
6626 if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
6627 check_replace = TRUE;
6628 } else {
6629 ltag = TRUE;
6630 }
6631
6632 tag = TRUE;
6633 namelen -= STRING_SIZE("refs/tags/");
6634 name += STRING_SIZE("refs/tags/");
6635
6636 } else if (!prefixcmp(name, "refs/remotes/")) {
6637 remote = TRUE;
6638 namelen -= STRING_SIZE("refs/remotes/");
6639 name += STRING_SIZE("refs/remotes/");
6640 tracked = !strcmp(opt_remote, name);
6641
6642 } else if (!prefixcmp(name, "refs/heads/")) {
6643 namelen -= STRING_SIZE("refs/heads/");
6644 name += STRING_SIZE("refs/heads/");
6645 head = !strncmp(opt_head, name, namelen);
6646
6647 } else if (!strcmp(name, "HEAD")) {
6648 string_ncopy(opt_head_rev, id, idlen);
6649 return OK;
6650 }
6651
6652 if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
6653 /* it's an annotated tag, replace the previous SHA1 with the
6654 * resolved commit id; relies on the fact git-ls-remote lists
6655 * the commit id of an annotated tag right before the commit id
6656 * it points to. */
6657 refs[refs_size - 1].ltag = ltag;
6658 string_copy_rev(refs[refs_size - 1].id, id);
6659
6660 return OK;
6661 }
6662 refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
6663 if (!refs)
6664 return ERR;
6665
6666 ref = &refs[refs_size++];
6667 ref->name = malloc(namelen + 1);
6668 if (!ref->name)
6669 return ERR;
6670
6671 strncpy(ref->name, name, namelen);
6672 ref->name[namelen] = 0;
6673 ref->head = head;
6674 ref->tag = tag;
6675 ref->ltag = ltag;
6676 ref->remote = remote;
6677 ref->tracked = tracked;
6678 string_copy_rev(ref->id, id);
6679
6680 return OK;
6681 }
6682
6683 static int
6684 load_refs(void)
6685 {
6686 static const char *ls_remote_argv[SIZEOF_ARG] = {
6687 "git", "ls-remote", opt_git_dir, NULL
6688 };
6689 static bool init = FALSE;
6690
6691 if (!init) {
6692 argv_from_env(ls_remote_argv, "TIG_LS_REMOTE");
6693 init = TRUE;
6694 }
6695
6696 if (!*opt_git_dir)
6697 return OK;
6698
6699 while (refs_size > 0)
6700 free(refs[--refs_size].name);
6701 while (id_refs_size > 0)
6702 free(id_refs[--id_refs_size]);
6703
6704 return run_io_load(ls_remote_argv, "\t", read_ref);
6705 }
6706
6707 static void
6708 set_remote_branch(const char *name, const char *value, size_t valuelen)
6709 {
6710 if (!strcmp(name, ".remote")) {
6711 string_ncopy(opt_remote, value, valuelen);
6712
6713 } else if (*opt_remote && !strcmp(name, ".merge")) {
6714 size_t from = strlen(opt_remote);
6715
6716 if (!prefixcmp(value, "refs/heads/"))
6717 value += STRING_SIZE("refs/heads/");
6718
6719 if (!string_format_from(opt_remote, &from, "/%s", value))
6720 opt_remote[0] = 0;
6721 }
6722 }
6723
6724 static void
6725 set_repo_config_option(char *name, char *value, int (*cmd)(int, const char **))
6726 {
6727 const char *argv[SIZEOF_ARG] = { name, "=" };
6728 int argc = 1 + (cmd == option_set_command);
6729 int error = ERR;
6730
6731 if (!argv_from_string(argv, &argc, value))
6732 config_msg = "Too many option arguments";
6733 else
6734 error = cmd(argc, argv);
6735
6736 if (error == ERR)
6737 warn("Option 'tig.%s': %s", name, config_msg);
6738 }
6739
6740 static bool
6741 set_environment_variable(const char *name, const char *value)
6742 {
6743 size_t len = strlen(name) + 1 + strlen(value) + 1;
6744 char *env = malloc(len);
6745
6746 if (env &&
6747 string_nformat(env, len, NULL, "%s=%s", name, value) &&
6748 putenv(env) == 0)
6749 return TRUE;
6750 free(env);
6751 return FALSE;
6752 }
6753
6754 static void
6755 set_work_tree(const char *value)
6756 {
6757 char cwd[SIZEOF_STR];
6758
6759 if (!getcwd(cwd, sizeof(cwd)))
6760 die("Failed to get cwd path: %s", strerror(errno));
6761 if (chdir(opt_git_dir) < 0)
6762 die("Failed to chdir(%s): %s", strerror(errno));
6763 if (!getcwd(opt_git_dir, sizeof(opt_git_dir)))
6764 die("Failed to get git path: %s", strerror(errno));
6765 if (chdir(cwd) < 0)
6766 die("Failed to chdir(%s): %s", cwd, strerror(errno));
6767 if (chdir(value) < 0)
6768 die("Failed to chdir(%s): %s", value, strerror(errno));
6769 if (!getcwd(cwd, sizeof(cwd)))
6770 die("Failed to get cwd path: %s", strerror(errno));
6771 if (!set_environment_variable("GIT_WORK_TREE", cwd))
6772 die("Failed to set GIT_WORK_TREE to '%s'", cwd);
6773 if (!set_environment_variable("GIT_DIR", opt_git_dir))
6774 die("Failed to set GIT_DIR to '%s'", opt_git_dir);
6775 opt_is_inside_work_tree = TRUE;
6776 }
6777
6778 static int
6779 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
6780 {
6781 if (!strcmp(name, "i18n.commitencoding"))
6782 string_ncopy(opt_encoding, value, valuelen);
6783
6784 else if (!strcmp(name, "core.editor"))
6785 string_ncopy(opt_editor, value, valuelen);
6786
6787 else if (!strcmp(name, "core.worktree"))
6788 set_work_tree(value);
6789
6790 else if (!prefixcmp(name, "tig.color."))
6791 set_repo_config_option(name + 10, value, option_color_command);
6792
6793 else if (!prefixcmp(name, "tig.bind."))
6794 set_repo_config_option(name + 9, value, option_bind_command);
6795
6796 else if (!prefixcmp(name, "tig."))
6797 set_repo_config_option(name + 4, value, option_set_command);
6798
6799 else if (*opt_head && !prefixcmp(name, "branch.") &&
6800 !strncmp(name + 7, opt_head, strlen(opt_head)))
6801 set_remote_branch(name + 7 + strlen(opt_head), value, valuelen);
6802
6803 return OK;
6804 }
6805
6806 static int
6807 load_git_config(void)
6808 {
6809 const char *config_list_argv[] = { "git", GIT_CONFIG, "--list", NULL };
6810
6811 return run_io_load(config_list_argv, "=", read_repo_config_option);
6812 }
6813
6814 static int
6815 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
6816 {
6817 if (!opt_git_dir[0]) {
6818 string_ncopy(opt_git_dir, name, namelen);
6819
6820 } else if (opt_is_inside_work_tree == -1) {
6821 /* This can be 3 different values depending on the
6822 * version of git being used. If git-rev-parse does not
6823 * understand --is-inside-work-tree it will simply echo
6824 * the option else either "true" or "false" is printed.
6825 * Default to true for the unknown case. */
6826 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
6827
6828 } else if (*name == '.') {
6829 string_ncopy(opt_cdup, name, namelen);
6830
6831 } else {
6832 string_ncopy(opt_prefix, name, namelen);
6833 }
6834
6835 return OK;
6836 }
6837
6838 static int
6839 load_repo_info(void)
6840 {
6841 const char *head_argv[] = {
6842 "git", "symbolic-ref", "HEAD", NULL
6843 };
6844 const char *rev_parse_argv[] = {
6845 "git", "rev-parse", "--git-dir", "--is-inside-work-tree",
6846 "--show-cdup", "--show-prefix", NULL
6847 };
6848
6849 if (run_io_buf(head_argv, opt_head, sizeof(opt_head))) {
6850 chomp_string(opt_head);
6851 if (!prefixcmp(opt_head, "refs/heads/")) {
6852 char *offset = opt_head + STRING_SIZE("refs/heads/");
6853
6854 memmove(opt_head, offset, strlen(offset) + 1);
6855 }
6856 }
6857
6858 return run_io_load(rev_parse_argv, "=", read_repo_info);
6859 }
6860
6861
6862 /*
6863 * Main
6864 */
6865
6866 static const char usage[] =
6867 "tig " TIG_VERSION " (" __DATE__ ")\n"
6868 "\n"
6869 "Usage: tig [options] [revs] [--] [paths]\n"
6870 " or: tig show [options] [revs] [--] [paths]\n"
6871 " or: tig blame [rev] path\n"
6872 " or: tig status\n"
6873 " or: tig < [git command output]\n"
6874 "\n"
6875 "Options:\n"
6876 " -v, --version Show version and exit\n"
6877 " -h, --help Show help message and exit";
6878
6879 static void __NORETURN
6880 quit(int sig)
6881 {
6882 /* XXX: Restore tty modes and let the OS cleanup the rest! */
6883 if (cursed)
6884 endwin();
6885 exit(0);
6886 }
6887
6888 static void __NORETURN
6889 die(const char *err, ...)
6890 {
6891 va_list args;
6892
6893 endwin();
6894
6895 va_start(args, err);
6896 fputs("tig: ", stderr);
6897 vfprintf(stderr, err, args);
6898 fputs("\n", stderr);
6899 va_end(args);
6900
6901 exit(1);
6902 }
6903
6904 static void
6905 warn(const char *msg, ...)
6906 {
6907 va_list args;
6908
6909 va_start(args, msg);
6910 fputs("tig warning: ", stderr);
6911 vfprintf(stderr, msg, args);
6912 fputs("\n", stderr);
6913 va_end(args);
6914 }
6915
6916 static enum request
6917 parse_options(int argc, const char *argv[])
6918 {
6919 enum request request = REQ_VIEW_MAIN;
6920 const char *subcommand;
6921 bool seen_dashdash = FALSE;
6922 /* XXX: This is vulnerable to the user overriding options
6923 * required for the main view parser. */
6924 const char *custom_argv[SIZEOF_ARG] = {
6925 "git", "log", "--no-color", "--pretty=raw", "--parents",
6926 "--topo-order", NULL
6927 };
6928 int i, j = 6;
6929
6930 if (!isatty(STDIN_FILENO)) {
6931 io_open(&VIEW(REQ_VIEW_PAGER)->io, "");
6932 return REQ_VIEW_PAGER;
6933 }
6934
6935 if (argc <= 1)
6936 return REQ_NONE;
6937
6938 subcommand = argv[1];
6939 if (!strcmp(subcommand, "status")) {
6940 if (argc > 2)
6941 warn("ignoring arguments after `%s'", subcommand);
6942 return REQ_VIEW_STATUS;
6943
6944 } else if (!strcmp(subcommand, "blame")) {
6945 if (argc <= 2 || argc > 4)
6946 die("invalid number of options to blame\n\n%s", usage);
6947
6948 i = 2;
6949 if (argc == 4) {
6950 string_ncopy(opt_ref, argv[i], strlen(argv[i]));
6951 i++;
6952 }
6953
6954 string_ncopy(opt_file, argv[i], strlen(argv[i]));
6955 return REQ_VIEW_BLAME;
6956
6957 } else if (!strcmp(subcommand, "show")) {
6958 request = REQ_VIEW_DIFF;
6959
6960 } else {
6961 subcommand = NULL;
6962 }
6963
6964 if (subcommand) {
6965 custom_argv[1] = subcommand;
6966 j = 2;
6967 }
6968
6969 for (i = 1 + !!subcommand; i < argc; i++) {
6970 const char *opt = argv[i];
6971
6972 if (seen_dashdash || !strcmp(opt, "--")) {
6973 seen_dashdash = TRUE;
6974
6975 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
6976 printf("tig version %s\n", TIG_VERSION);
6977 quit(0);
6978
6979 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
6980 printf("%s\n", usage);
6981 quit(0);
6982 }
6983
6984 custom_argv[j++] = opt;
6985 if (j >= ARRAY_SIZE(custom_argv))
6986 die("command too long");
6987 }
6988
6989 if (!prepare_update(VIEW(request), custom_argv, NULL, FORMAT_NONE))
6990 die("Failed to format arguments");
6991
6992 return request;
6993 }
6994
6995 int
6996 main(int argc, const char *argv[])
6997 {
6998 enum request request = parse_options(argc, argv);
6999 struct view *view;
7000 size_t i;
7001
7002 signal(SIGINT, quit);
7003 signal(SIGPIPE, SIG_IGN);
7004
7005 if (setlocale(LC_ALL, "")) {
7006 char *codeset = nl_langinfo(CODESET);
7007
7008 string_ncopy(opt_codeset, codeset, strlen(codeset));
7009 }
7010
7011 if (load_repo_info() == ERR)
7012 die("Failed to load repo info.");
7013
7014 if (load_options() == ERR)
7015 die("Failed to load user config.");
7016
7017 if (load_git_config() == ERR)
7018 die("Failed to load repo config.");
7019
7020 /* Require a git repository unless when running in pager mode. */
7021 if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
7022 die("Not a git repository");
7023
7024 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
7025 opt_utf8 = FALSE;
7026
7027 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
7028 opt_iconv = iconv_open(opt_codeset, opt_encoding);
7029 if (opt_iconv == ICONV_NONE)
7030 die("Failed to initialize character set conversion");
7031 }
7032
7033 if (load_refs() == ERR)
7034 die("Failed to load refs.");
7035
7036 foreach_view (view, i)
7037 argv_from_env(view->ops->argv, view->cmd_env);
7038
7039 init_display();
7040
7041 if (request != REQ_NONE)
7042 open_view(NULL, request, OPEN_PREPARED);
7043 request = request == REQ_NONE ? REQ_VIEW_MAIN : REQ_NONE;
7044
7045 while (view_driver(display[current_view], request)) {
7046 int key = get_input(0);
7047
7048 view = display[current_view];
7049 request = get_keybinding(view->keymap, key);
7050
7051 /* Some low-level request handling. This keeps access to
7052 * status_win restricted. */
7053 switch (request) {
7054 case REQ_PROMPT:
7055 {
7056 char *cmd = read_prompt(":");
7057
7058 if (cmd && isdigit(*cmd)) {
7059 int lineno = view->lineno + 1;
7060
7061 if (parse_int(&lineno, cmd, 1, view->lines + 1) == OK) {
7062 select_view_line(view, lineno - 1);
7063 report("");
7064 } else {
7065 report("Unable to parse '%s' as a line number", cmd);
7066 }
7067
7068 } else if (cmd) {
7069 struct view *next = VIEW(REQ_VIEW_PAGER);
7070 const char *argv[SIZEOF_ARG] = { "git" };
7071 int argc = 1;
7072
7073 /* When running random commands, initially show the
7074 * command in the title. However, it maybe later be
7075 * overwritten if a commit line is selected. */
7076 string_ncopy(next->ref, cmd, strlen(cmd));
7077
7078 if (!argv_from_string(argv, &argc, cmd)) {
7079 report("Too many arguments");
7080 } else if (!prepare_update(next, argv, NULL, FORMAT_DASH)) {
7081 report("Failed to format command");
7082 } else {
7083 open_view(view, REQ_VIEW_PAGER, OPEN_PREPARED);
7084 }
7085 }
7086
7087 request = REQ_NONE;
7088 break;
7089 }
7090 case REQ_SEARCH:
7091 case REQ_SEARCH_BACK:
7092 {
7093 const char *prompt = request == REQ_SEARCH ? "/" : "?";
7094 char *search = read_prompt(prompt);
7095
7096 if (search)
7097 string_ncopy(opt_search, search, strlen(search));
7098 else if (*opt_search)
7099 request = request == REQ_SEARCH ?
7100 REQ_FIND_NEXT :
7101 REQ_FIND_PREV;
7102 else
7103 request = REQ_NONE;
7104 break;
7105 }
7106 default:
7107 break;
7108 }
7109 }
7110
7111 quit(0);
7112
7113 return 0;
7114 }
Text-mode interface for git