1 /* Copyright (c) 2006-2010 Jonas Fonseca <fonseca@diku.dk>
2 *
3 * This program is free software; you can redistribute it and/or
4 * modify it under the terms of the GNU General Public License as
5 * published by the Free Software Foundation; either version 2 of
6 * the License, or (at your option) any later version.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 */
14 #ifdef HAVE_CONFIG_H
15 #include "config.h"
16 #endif
18 #ifndef TIG_VERSION
19 #define TIG_VERSION "unknown-version"
20 #endif
22 #ifndef DEBUG
23 #define NDEBUG
24 #endif
26 #include <assert.h>
27 #include <errno.h>
28 #include <ctype.h>
29 #include <signal.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #include <sys/types.h>
35 #include <sys/wait.h>
36 #include <sys/stat.h>
37 #include <sys/select.h>
38 #include <unistd.h>
39 #include <sys/time.h>
40 #include <time.h>
41 #include <fcntl.h>
43 #include <regex.h>
45 #include <locale.h>
46 #include <langinfo.h>
47 #include <iconv.h>
49 /* ncurses(3): Must be defined to have extended wide-character functions. */
50 #define _XOPEN_SOURCE_EXTENDED
52 #ifdef HAVE_NCURSESW_NCURSES_H
53 #include <ncursesw/ncurses.h>
54 #else
55 #ifdef HAVE_NCURSES_NCURSES_H
56 #include <ncurses/ncurses.h>
57 #else
58 #include <ncurses.h>
59 #endif
60 #endif
62 #if __GNUC__ >= 3
63 #define __NORETURN __attribute__((__noreturn__))
64 #else
65 #define __NORETURN
66 #endif
68 static void __NORETURN die(const char *err, ...);
69 static void warn(const char *msg, ...);
70 static void report(const char *msg, ...);
72 #define ABS(x) ((x) >= 0 ? (x) : -(x))
73 #define MIN(x, y) ((x) < (y) ? (x) : (y))
74 #define MAX(x, y) ((x) > (y) ? (x) : (y))
76 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
77 #define STRING_SIZE(x) (sizeof(x) - 1)
79 #define SIZEOF_STR 1024 /* Default string size. */
80 #define SIZEOF_REF 256 /* Size of symbolic or SHA1 ID. */
81 #define SIZEOF_REV 41 /* Holds a SHA-1 and an ending NUL. */
82 #define SIZEOF_ARG 32 /* Default argument array size. */
84 /* Revision graph */
86 #define REVGRAPH_INIT 'I'
87 #define REVGRAPH_MERGE 'M'
88 #define REVGRAPH_BRANCH '+'
89 #define REVGRAPH_COMMIT '*'
90 #define REVGRAPH_BOUND '^'
92 #define SIZEOF_REVGRAPH 19 /* Size of revision ancestry graphics. */
94 /* This color name can be used to refer to the default term colors. */
95 #define COLOR_DEFAULT (-1)
97 #define ICONV_NONE ((iconv_t) -1)
98 #ifndef ICONV_CONST
99 #define ICONV_CONST /* nothing */
100 #endif
102 /* The format and size of the date column in the main view. */
103 #define DATE_FORMAT "%Y-%m-%d %H:%M"
104 #define DATE_COLS STRING_SIZE("2006-04-29 14:21 ")
105 #define DATE_SHORT_COLS STRING_SIZE("2006-04-29 ")
107 #define ID_COLS 8
108 #define AUTHOR_COLS 19
110 #define MIN_VIEW_HEIGHT 4
112 #define NULL_ID "0000000000000000000000000000000000000000"
114 #define S_ISGITLINK(mode) (((mode) & S_IFMT) == 0160000)
116 /* Some ASCII-shorthands fitted into the ncurses namespace. */
117 #define KEY_TAB '\t'
118 #define KEY_RETURN '\r'
119 #define KEY_ESC 27
122 struct ref {
123 char id[SIZEOF_REV]; /* Commit SHA1 ID */
124 unsigned int head:1; /* Is it the current HEAD? */
125 unsigned int tag:1; /* Is it a tag? */
126 unsigned int ltag:1; /* If so, is the tag local? */
127 unsigned int remote:1; /* Is it a remote ref? */
128 unsigned int tracked:1; /* Is it the remote for the current HEAD? */
129 char name[1]; /* Ref name; tag or head names are shortened. */
130 };
132 struct ref_list {
133 char id[SIZEOF_REV]; /* Commit SHA1 ID */
134 size_t size; /* Number of refs. */
135 struct ref **refs; /* References for this ID. */
136 };
138 static struct ref *get_ref_head();
139 static struct ref_list *get_ref_list(const char *id);
140 static void foreach_ref(bool (*visitor)(void *data, const struct ref *ref), void *data);
141 static int load_refs(void);
143 enum input_status {
144 INPUT_OK,
145 INPUT_SKIP,
146 INPUT_STOP,
147 INPUT_CANCEL
148 };
150 typedef enum input_status (*input_handler)(void *data, char *buf, int c);
152 static char *prompt_input(const char *prompt, input_handler handler, void *data);
153 static bool prompt_yesno(const char *prompt);
155 struct menu_item {
156 int hotkey;
157 const char *text;
158 void *data;
159 };
161 static bool prompt_menu(const char *prompt, const struct menu_item *items, int *selected);
163 /*
164 * Allocation helpers ... Entering macro hell to never be seen again.
165 */
167 #define DEFINE_ALLOCATOR(name, type, chunk_size) \
168 static type * \
169 name(type **mem, size_t size, size_t increase) \
170 { \
171 size_t num_chunks = (size + chunk_size - 1) / chunk_size; \
172 size_t num_chunks_new = (size + increase + chunk_size - 1) / chunk_size;\
173 type *tmp = *mem; \
174 \
175 if (mem == NULL || num_chunks != num_chunks_new) { \
176 tmp = realloc(tmp, num_chunks_new * chunk_size * sizeof(type)); \
177 if (tmp) \
178 *mem = tmp; \
179 } \
180 \
181 return tmp; \
182 }
184 /*
185 * String helpers
186 */
188 static inline void
189 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
190 {
191 if (srclen > dstlen - 1)
192 srclen = dstlen - 1;
194 strncpy(dst, src, srclen);
195 dst[srclen] = 0;
196 }
198 /* Shorthands for safely copying into a fixed buffer. */
200 #define string_copy(dst, src) \
201 string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
203 #define string_ncopy(dst, src, srclen) \
204 string_ncopy_do(dst, sizeof(dst), src, srclen)
206 #define string_copy_rev(dst, src) \
207 string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
209 #define string_add(dst, from, src) \
210 string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
212 static void
213 string_expand(char *dst, size_t dstlen, const char *src, int tabsize)
214 {
215 size_t size, pos;
217 for (size = pos = 0; size < dstlen - 1 && src[pos]; pos++) {
218 if (src[pos] == '\t') {
219 size_t expanded = tabsize - (size % tabsize);
221 if (expanded + size >= dstlen - 1)
222 expanded = dstlen - size - 1;
223 memcpy(dst + size, " ", expanded);
224 size += expanded;
225 } else {
226 dst[size++] = src[pos];
227 }
228 }
230 dst[size] = 0;
231 }
233 static char *
234 chomp_string(char *name)
235 {
236 int namelen;
238 while (isspace(*name))
239 name++;
241 namelen = strlen(name) - 1;
242 while (namelen > 0 && isspace(name[namelen]))
243 name[namelen--] = 0;
245 return name;
246 }
248 static bool
249 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
250 {
251 va_list args;
252 size_t pos = bufpos ? *bufpos : 0;
254 va_start(args, fmt);
255 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
256 va_end(args);
258 if (bufpos)
259 *bufpos = pos;
261 return pos >= bufsize ? FALSE : TRUE;
262 }
264 #define string_format(buf, fmt, args...) \
265 string_nformat(buf, sizeof(buf), NULL, fmt, args)
267 #define string_format_from(buf, from, fmt, args...) \
268 string_nformat(buf, sizeof(buf), from, fmt, args)
270 static int
271 string_enum_compare(const char *str1, const char *str2, int len)
272 {
273 size_t i;
275 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
277 /* Diff-Header == DIFF_HEADER */
278 for (i = 0; i < len; i++) {
279 if (toupper(str1[i]) == toupper(str2[i]))
280 continue;
282 if (string_enum_sep(str1[i]) &&
283 string_enum_sep(str2[i]))
284 continue;
286 return str1[i] - str2[i];
287 }
289 return 0;
290 }
292 #define enum_equals(entry, str, len) \
293 ((entry).namelen == (len) && !string_enum_compare((entry).name, str, len))
295 struct enum_map {
296 const char *name;
297 int namelen;
298 int value;
299 };
301 #define ENUM_MAP(name, value) { name, STRING_SIZE(name), value }
303 static char *
304 enum_map_name(const char *name, size_t namelen)
305 {
306 static char buf[SIZEOF_STR];
307 int bufpos;
309 for (bufpos = 0; bufpos <= namelen; bufpos++) {
310 buf[bufpos] = tolower(name[bufpos]);
311 if (buf[bufpos] == '_')
312 buf[bufpos] = '-';
313 }
315 buf[bufpos] = 0;
316 return buf;
317 }
319 #define enum_name(entry) enum_map_name((entry).name, (entry).namelen)
321 static bool
322 map_enum_do(const struct enum_map *map, size_t map_size, int *value, const char *name)
323 {
324 size_t namelen = strlen(name);
325 int i;
327 for (i = 0; i < map_size; i++)
328 if (enum_equals(map[i], name, namelen)) {
329 *value = map[i].value;
330 return TRUE;
331 }
333 return FALSE;
334 }
336 #define map_enum(attr, map, name) \
337 map_enum_do(map, ARRAY_SIZE(map), attr, name)
339 #define prefixcmp(str1, str2) \
340 strncmp(str1, str2, STRING_SIZE(str2))
342 static inline int
343 suffixcmp(const char *str, int slen, const char *suffix)
344 {
345 size_t len = slen >= 0 ? slen : strlen(str);
346 size_t suffixlen = strlen(suffix);
348 return suffixlen < len ? strcmp(str + len - suffixlen, suffix) : -1;
349 }
352 /*
353 * Unicode / UTF-8 handling
354 *
355 * NOTE: Much of the following code for dealing with Unicode is derived from
356 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
357 * src/intl/charset.c from the UTF-8 branch commit elinks-0.11.0-g31f2c28.
358 */
360 static inline int
361 unicode_width(unsigned long c, int tab_size)
362 {
363 if (c >= 0x1100 &&
364 (c <= 0x115f /* Hangul Jamo */
365 || c == 0x2329
366 || c == 0x232a
367 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
368 /* CJK ... Yi */
369 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
370 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
371 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
372 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
373 || (c >= 0xffe0 && c <= 0xffe6)
374 || (c >= 0x20000 && c <= 0x2fffd)
375 || (c >= 0x30000 && c <= 0x3fffd)))
376 return 2;
378 if (c == '\t')
379 return tab_size;
381 return 1;
382 }
384 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
385 * Illegal bytes are set one. */
386 static const unsigned char utf8_bytes[256] = {
387 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,
388 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,
389 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,
390 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,
391 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,
392 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,
393 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,
394 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,
395 };
397 static inline unsigned char
398 utf8_char_length(const char *string, const char *end)
399 {
400 int c = *(unsigned char *) string;
402 return utf8_bytes[c];
403 }
405 /* Decode UTF-8 multi-byte representation into a Unicode character. */
406 static inline unsigned long
407 utf8_to_unicode(const char *string, size_t length)
408 {
409 unsigned long unicode;
411 switch (length) {
412 case 1:
413 unicode = string[0];
414 break;
415 case 2:
416 unicode = (string[0] & 0x1f) << 6;
417 unicode += (string[1] & 0x3f);
418 break;
419 case 3:
420 unicode = (string[0] & 0x0f) << 12;
421 unicode += ((string[1] & 0x3f) << 6);
422 unicode += (string[2] & 0x3f);
423 break;
424 case 4:
425 unicode = (string[0] & 0x0f) << 18;
426 unicode += ((string[1] & 0x3f) << 12);
427 unicode += ((string[2] & 0x3f) << 6);
428 unicode += (string[3] & 0x3f);
429 break;
430 case 5:
431 unicode = (string[0] & 0x0f) << 24;
432 unicode += ((string[1] & 0x3f) << 18);
433 unicode += ((string[2] & 0x3f) << 12);
434 unicode += ((string[3] & 0x3f) << 6);
435 unicode += (string[4] & 0x3f);
436 break;
437 case 6:
438 unicode = (string[0] & 0x01) << 30;
439 unicode += ((string[1] & 0x3f) << 24);
440 unicode += ((string[2] & 0x3f) << 18);
441 unicode += ((string[3] & 0x3f) << 12);
442 unicode += ((string[4] & 0x3f) << 6);
443 unicode += (string[5] & 0x3f);
444 break;
445 default:
446 return 0;
447 }
449 /* Invalid characters could return the special 0xfffd value but NUL
450 * should be just as good. */
451 return unicode > 0xffff ? 0 : unicode;
452 }
454 /* Calculates how much of string can be shown within the given maximum width
455 * and sets trimmed parameter to non-zero value if all of string could not be
456 * shown. If the reserve flag is TRUE, it will reserve at least one
457 * trailing character, which can be useful when drawing a delimiter.
458 *
459 * Returns the number of bytes to output from string to satisfy max_width. */
460 static size_t
461 utf8_length(const char **start, size_t skip, int *width, size_t max_width, int *trimmed, bool reserve, int tab_size)
462 {
463 const char *string = *start;
464 const char *end = strchr(string, '\0');
465 unsigned char last_bytes = 0;
466 size_t last_ucwidth = 0;
468 *width = 0;
469 *trimmed = 0;
471 while (string < end) {
472 unsigned char bytes = utf8_char_length(string, end);
473 size_t ucwidth;
474 unsigned long unicode;
476 if (string + bytes > end)
477 break;
479 /* Change representation to figure out whether
480 * it is a single- or double-width character. */
482 unicode = utf8_to_unicode(string, bytes);
483 /* FIXME: Graceful handling of invalid Unicode character. */
484 if (!unicode)
485 break;
487 ucwidth = unicode_width(unicode, tab_size);
488 if (skip > 0) {
489 skip -= ucwidth <= skip ? ucwidth : skip;
490 *start += bytes;
491 }
492 *width += ucwidth;
493 if (*width > max_width) {
494 *trimmed = 1;
495 *width -= ucwidth;
496 if (reserve && *width == max_width) {
497 string -= last_bytes;
498 *width -= last_ucwidth;
499 }
500 break;
501 }
503 string += bytes;
504 last_bytes = ucwidth ? bytes : 0;
505 last_ucwidth = ucwidth;
506 }
508 return string - *start;
509 }
512 #define DATE_INFO \
513 DATE_(NO), \
514 DATE_(DEFAULT), \
515 DATE_(LOCAL), \
516 DATE_(RELATIVE), \
517 DATE_(SHORT)
519 enum date {
520 #define DATE_(name) DATE_##name
521 DATE_INFO
522 #undef DATE_
523 };
525 static const struct enum_map date_map[] = {
526 #define DATE_(name) ENUM_MAP(#name, DATE_##name)
527 DATE_INFO
528 #undef DATE_
529 };
531 struct time {
532 time_t sec;
533 int tz;
534 };
536 static inline int timecmp(const struct time *t1, const struct time *t2)
537 {
538 return t1->sec - t2->sec;
539 }
541 static const char *
542 mkdate(const struct time *time, enum date date)
543 {
544 static char buf[DATE_COLS + 1];
545 static const struct enum_map reldate[] = {
546 { "second", 1, 60 * 2 },
547 { "minute", 60, 60 * 60 * 2 },
548 { "hour", 60 * 60, 60 * 60 * 24 * 2 },
549 { "day", 60 * 60 * 24, 60 * 60 * 24 * 7 * 2 },
550 { "week", 60 * 60 * 24 * 7, 60 * 60 * 24 * 7 * 5 },
551 { "month", 60 * 60 * 24 * 30, 60 * 60 * 24 * 30 * 12 },
552 };
553 struct tm tm;
555 if (!date || !time || !time->sec)
556 return "";
558 if (date == DATE_RELATIVE) {
559 struct timeval now;
560 time_t date = time->sec + time->tz;
561 time_t seconds;
562 int i;
564 gettimeofday(&now, NULL);
565 seconds = now.tv_sec < date ? date - now.tv_sec : now.tv_sec - date;
566 for (i = 0; i < ARRAY_SIZE(reldate); i++) {
567 if (seconds >= reldate[i].value)
568 continue;
570 seconds /= reldate[i].namelen;
571 if (!string_format(buf, "%ld %s%s %s",
572 seconds, reldate[i].name,
573 seconds > 1 ? "s" : "",
574 now.tv_sec >= date ? "ago" : "ahead"))
575 break;
576 return buf;
577 }
578 }
580 if (date == DATE_LOCAL) {
581 time_t date = time->sec + time->tz;
582 localtime_r(&date, &tm);
583 }
584 else {
585 gmtime_r(&time->sec, &tm);
586 }
587 return strftime(buf, sizeof(buf), DATE_FORMAT, &tm) ? buf : NULL;
588 }
591 #define AUTHOR_VALUES \
592 AUTHOR_(NO), \
593 AUTHOR_(FULL), \
594 AUTHOR_(ABBREVIATED)
596 enum author {
597 #define AUTHOR_(name) AUTHOR_##name
598 AUTHOR_VALUES,
599 #undef AUTHOR_
600 AUTHOR_DEFAULT = AUTHOR_FULL
601 };
603 static const struct enum_map author_map[] = {
604 #define AUTHOR_(name) ENUM_MAP(#name, AUTHOR_##name)
605 AUTHOR_VALUES
606 #undef AUTHOR_
607 };
609 static const char *
610 get_author_initials(const char *author)
611 {
612 static char initials[AUTHOR_COLS * 6 + 1];
613 size_t pos = 0;
614 const char *end = strchr(author, '\0');
616 #define is_initial_sep(c) (isspace(c) || ispunct(c) || (c) == '@' || (c) == '-')
618 memset(initials, 0, sizeof(initials));
619 while (author < end) {
620 unsigned char bytes;
621 size_t i;
623 while (is_initial_sep(*author))
624 author++;
626 bytes = utf8_char_length(author, end);
627 if (bytes < sizeof(initials) - 1 - pos) {
628 while (bytes--) {
629 initials[pos++] = *author++;
630 }
631 }
633 for (i = pos; author < end && !is_initial_sep(*author); author++) {
634 if (i < sizeof(initials) - 1)
635 initials[i++] = *author;
636 }
638 initials[i++] = 0;
639 }
641 return initials;
642 }
645 static bool
646 argv_from_string(const char *argv[SIZEOF_ARG], int *argc, char *cmd)
647 {
648 int valuelen;
650 while (*cmd && *argc < SIZEOF_ARG && (valuelen = strcspn(cmd, " \t"))) {
651 bool advance = cmd[valuelen] != 0;
653 cmd[valuelen] = 0;
654 argv[(*argc)++] = chomp_string(cmd);
655 cmd = chomp_string(cmd + valuelen + advance);
656 }
658 if (*argc < SIZEOF_ARG)
659 argv[*argc] = NULL;
660 return *argc < SIZEOF_ARG;
661 }
663 static bool
664 argv_from_env(const char **argv, const char *name)
665 {
666 char *env = argv ? getenv(name) : NULL;
667 int argc = 0;
669 if (env && *env)
670 env = strdup(env);
671 return !env || argv_from_string(argv, &argc, env);
672 }
674 static void
675 argv_free(const char *argv[])
676 {
677 int argc;
679 if (!argv)
680 return;
681 for (argc = 0; argv[argc]; argc++)
682 free((void *) argv[argc]);
683 argv[0] = NULL;
684 }
686 DEFINE_ALLOCATOR(argv_realloc, const char *, SIZEOF_ARG)
688 static bool
689 argv_append(const char ***argv, const char *arg)
690 {
691 int argc = 0;
693 while (*argv && (*argv)[argc])
694 argc++;
696 if (!argv_realloc(argv, argc, 2))
697 return FALSE;
699 (*argv)[argc++] = strdup(arg);
700 (*argv)[argc] = NULL;
701 return TRUE;
702 }
704 static bool
705 argv_copy(const char ***dst, const char *src[])
706 {
707 int argc;
709 for (argc = 0; src[argc]; argc++)
710 if (!argv_append(dst, src[argc]))
711 return FALSE;
712 return TRUE;
713 }
716 /*
717 * Executing external commands.
718 */
720 enum io_type {
721 IO_FD, /* File descriptor based IO. */
722 IO_BG, /* Execute command in the background. */
723 IO_FG, /* Execute command with same std{in,out,err}. */
724 IO_RD, /* Read only fork+exec IO. */
725 IO_WR, /* Write only fork+exec IO. */
726 IO_AP, /* Append fork+exec output to file. */
727 };
729 struct io {
730 int pipe; /* Pipe end for reading or writing. */
731 pid_t pid; /* PID of spawned process. */
732 int error; /* Error status. */
733 char *buf; /* Read buffer. */
734 size_t bufalloc; /* Allocated buffer size. */
735 size_t bufsize; /* Buffer content size. */
736 char *bufpos; /* Current buffer position. */
737 unsigned int eof:1; /* Has end of file been reached. */
738 };
740 static void
741 io_init(struct io *io)
742 {
743 memset(io, 0, sizeof(*io));
744 io->pipe = -1;
745 }
747 static bool
748 io_open(struct io *io, const char *fmt, ...)
749 {
750 char name[SIZEOF_STR] = "";
751 bool fits;
752 va_list args;
754 io_init(io);
756 va_start(args, fmt);
757 fits = vsnprintf(name, sizeof(name), fmt, args) < sizeof(name);
758 va_end(args);
760 if (!fits) {
761 io->error = ENAMETOOLONG;
762 return FALSE;
763 }
764 io->pipe = *name ? open(name, O_RDONLY) : STDIN_FILENO;
765 if (io->pipe == -1)
766 io->error = errno;
767 return io->pipe != -1;
768 }
770 static bool
771 io_kill(struct io *io)
772 {
773 return io->pid == 0 || kill(io->pid, SIGKILL) != -1;
774 }
776 static bool
777 io_done(struct io *io)
778 {
779 pid_t pid = io->pid;
781 if (io->pipe != -1)
782 close(io->pipe);
783 free(io->buf);
784 io_init(io);
786 while (pid > 0) {
787 int status;
788 pid_t waiting = waitpid(pid, &status, 0);
790 if (waiting < 0) {
791 if (errno == EINTR)
792 continue;
793 io->error = errno;
794 return FALSE;
795 }
797 return waiting == pid &&
798 !WIFSIGNALED(status) &&
799 WIFEXITED(status) &&
800 !WEXITSTATUS(status);
801 }
803 return TRUE;
804 }
806 static bool
807 io_run(struct io *io, enum io_type type, const char *dir, const char *argv[], ...)
808 {
809 int pipefds[2] = { -1, -1 };
810 va_list args;
812 io_init(io);
814 if ((type == IO_RD || type == IO_WR) && pipe(pipefds) < 0) {
815 io->error = errno;
816 return FALSE;
817 } else if (type == IO_AP) {
818 va_start(args, argv);
819 pipefds[1] = va_arg(args, int);
820 va_end(args);
821 }
823 if ((io->pid = fork())) {
824 if (io->pid == -1)
825 io->error = errno;
826 if (pipefds[!(type == IO_WR)] != -1)
827 close(pipefds[!(type == IO_WR)]);
828 if (io->pid != -1) {
829 io->pipe = pipefds[!!(type == IO_WR)];
830 return TRUE;
831 }
833 } else {
834 if (type != IO_FG) {
835 int devnull = open("/dev/null", O_RDWR);
836 int readfd = type == IO_WR ? pipefds[0] : devnull;
837 int writefd = (type == IO_RD || type == IO_AP)
838 ? pipefds[1] : devnull;
840 dup2(readfd, STDIN_FILENO);
841 dup2(writefd, STDOUT_FILENO);
842 dup2(devnull, STDERR_FILENO);
844 close(devnull);
845 if (pipefds[0] != -1)
846 close(pipefds[0]);
847 if (pipefds[1] != -1)
848 close(pipefds[1]);
849 }
851 if (dir && *dir && chdir(dir) == -1)
852 exit(errno);
854 execvp(argv[0], (char *const*) argv);
855 exit(errno);
856 }
858 if (pipefds[!!(type == IO_WR)] != -1)
859 close(pipefds[!!(type == IO_WR)]);
860 return FALSE;
861 }
863 static bool
864 io_complete(enum io_type type, const char **argv, const char *dir, int fd)
865 {
866 struct io io;
868 return io_run(&io, type, dir, argv, fd) && io_done(&io);
869 }
871 static bool
872 io_run_bg(const char **argv)
873 {
874 return io_complete(IO_BG, argv, NULL, -1);
875 }
877 static bool
878 io_run_fg(const char **argv, const char *dir)
879 {
880 return io_complete(IO_FG, argv, dir, -1);
881 }
883 static bool
884 io_run_append(const char **argv, int fd)
885 {
886 return io_complete(IO_AP, argv, NULL, fd);
887 }
889 static bool
890 io_eof(struct io *io)
891 {
892 return io->eof;
893 }
895 static int
896 io_error(struct io *io)
897 {
898 return io->error;
899 }
901 static char *
902 io_strerror(struct io *io)
903 {
904 return strerror(io->error);
905 }
907 static bool
908 io_can_read(struct io *io)
909 {
910 struct timeval tv = { 0, 500 };
911 fd_set fds;
913 FD_ZERO(&fds);
914 FD_SET(io->pipe, &fds);
916 return select(io->pipe + 1, &fds, NULL, NULL, &tv) > 0;
917 }
919 static ssize_t
920 io_read(struct io *io, void *buf, size_t bufsize)
921 {
922 do {
923 ssize_t readsize = read(io->pipe, buf, bufsize);
925 if (readsize < 0 && (errno == EAGAIN || errno == EINTR))
926 continue;
927 else if (readsize == -1)
928 io->error = errno;
929 else if (readsize == 0)
930 io->eof = 1;
931 return readsize;
932 } while (1);
933 }
935 DEFINE_ALLOCATOR(io_realloc_buf, char, BUFSIZ)
937 static char *
938 io_get(struct io *io, int c, bool can_read)
939 {
940 char *eol;
941 ssize_t readsize;
943 while (TRUE) {
944 if (io->bufsize > 0) {
945 eol = memchr(io->bufpos, c, io->bufsize);
946 if (eol) {
947 char *line = io->bufpos;
949 *eol = 0;
950 io->bufpos = eol + 1;
951 io->bufsize -= io->bufpos - line;
952 return line;
953 }
954 }
956 if (io_eof(io)) {
957 if (io->bufsize) {
958 io->bufpos[io->bufsize] = 0;
959 io->bufsize = 0;
960 return io->bufpos;
961 }
962 return NULL;
963 }
965 if (!can_read)
966 return NULL;
968 if (io->bufsize > 0 && io->bufpos > io->buf)
969 memmove(io->buf, io->bufpos, io->bufsize);
971 if (io->bufalloc == io->bufsize) {
972 if (!io_realloc_buf(&io->buf, io->bufalloc, BUFSIZ))
973 return NULL;
974 io->bufalloc += BUFSIZ;
975 }
977 io->bufpos = io->buf;
978 readsize = io_read(io, io->buf + io->bufsize, io->bufalloc - io->bufsize);
979 if (io_error(io))
980 return NULL;
981 io->bufsize += readsize;
982 }
983 }
985 static bool
986 io_write(struct io *io, const void *buf, size_t bufsize)
987 {
988 size_t written = 0;
990 while (!io_error(io) && written < bufsize) {
991 ssize_t size;
993 size = write(io->pipe, buf + written, bufsize - written);
994 if (size < 0 && (errno == EAGAIN || errno == EINTR))
995 continue;
996 else if (size == -1)
997 io->error = errno;
998 else
999 written += size;
1000 }
1002 return written == bufsize;
1003 }
1005 static bool
1006 io_read_buf(struct io *io, char buf[], size_t bufsize)
1007 {
1008 char *result = io_get(io, '\n', TRUE);
1010 if (result) {
1011 result = chomp_string(result);
1012 string_ncopy_do(buf, bufsize, result, strlen(result));
1013 }
1015 return io_done(io) && result;
1016 }
1018 static bool
1019 io_run_buf(const char **argv, char buf[], size_t bufsize)
1020 {
1021 struct io io;
1023 return io_run(&io, IO_RD, NULL, argv) && io_read_buf(&io, buf, bufsize);
1024 }
1026 static int
1027 io_load(struct io *io, const char *separators,
1028 int (*read_property)(char *, size_t, char *, size_t))
1029 {
1030 char *name;
1031 int state = OK;
1033 while (state == OK && (name = io_get(io, '\n', TRUE))) {
1034 char *value;
1035 size_t namelen;
1036 size_t valuelen;
1038 name = chomp_string(name);
1039 namelen = strcspn(name, separators);
1041 if (name[namelen]) {
1042 name[namelen] = 0;
1043 value = chomp_string(name + namelen + 1);
1044 valuelen = strlen(value);
1046 } else {
1047 value = "";
1048 valuelen = 0;
1049 }
1051 state = read_property(name, namelen, value, valuelen);
1052 }
1054 if (state != ERR && io_error(io))
1055 state = ERR;
1056 io_done(io);
1058 return state;
1059 }
1061 static int
1062 io_run_load(const char **argv, const char *separators,
1063 int (*read_property)(char *, size_t, char *, size_t))
1064 {
1065 struct io io;
1067 if (!io_run(&io, IO_RD, NULL, argv))
1068 return ERR;
1069 return io_load(&io, separators, read_property);
1070 }
1073 /*
1074 * User requests
1075 */
1077 #define REQ_INFO \
1078 /* XXX: Keep the view request first and in sync with views[]. */ \
1079 REQ_GROUP("View switching") \
1080 REQ_(VIEW_MAIN, "Show main view"), \
1081 REQ_(VIEW_DIFF, "Show diff view"), \
1082 REQ_(VIEW_LOG, "Show log view"), \
1083 REQ_(VIEW_TREE, "Show tree view"), \
1084 REQ_(VIEW_BLOB, "Show blob view"), \
1085 REQ_(VIEW_BLAME, "Show blame view"), \
1086 REQ_(VIEW_BRANCH, "Show branch view"), \
1087 REQ_(VIEW_HELP, "Show help page"), \
1088 REQ_(VIEW_PAGER, "Show pager view"), \
1089 REQ_(VIEW_STATUS, "Show status view"), \
1090 REQ_(VIEW_STAGE, "Show stage view"), \
1091 \
1092 REQ_GROUP("View manipulation") \
1093 REQ_(ENTER, "Enter current line and scroll"), \
1094 REQ_(NEXT, "Move to next"), \
1095 REQ_(PREVIOUS, "Move to previous"), \
1096 REQ_(PARENT, "Move to parent"), \
1097 REQ_(VIEW_NEXT, "Move focus to next view"), \
1098 REQ_(REFRESH, "Reload and refresh"), \
1099 REQ_(MAXIMIZE, "Maximize the current view"), \
1100 REQ_(VIEW_CLOSE, "Close the current view"), \
1101 REQ_(QUIT, "Close all views and quit"), \
1102 \
1103 REQ_GROUP("View specific requests") \
1104 REQ_(STATUS_UPDATE, "Update file status"), \
1105 REQ_(STATUS_REVERT, "Revert file changes"), \
1106 REQ_(STATUS_MERGE, "Merge file using external tool"), \
1107 REQ_(STAGE_NEXT, "Find next chunk to stage"), \
1108 \
1109 REQ_GROUP("Cursor navigation") \
1110 REQ_(MOVE_UP, "Move cursor one line up"), \
1111 REQ_(MOVE_DOWN, "Move cursor one line down"), \
1112 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
1113 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
1114 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
1115 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
1116 \
1117 REQ_GROUP("Scrolling") \
1118 REQ_(SCROLL_LEFT, "Scroll two columns left"), \
1119 REQ_(SCROLL_RIGHT, "Scroll two columns right"), \
1120 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
1121 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
1122 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
1123 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
1124 \
1125 REQ_GROUP("Searching") \
1126 REQ_(SEARCH, "Search the view"), \
1127 REQ_(SEARCH_BACK, "Search backwards in the view"), \
1128 REQ_(FIND_NEXT, "Find next search match"), \
1129 REQ_(FIND_PREV, "Find previous search match"), \
1130 \
1131 REQ_GROUP("Option manipulation") \
1132 REQ_(OPTIONS, "Open option menu"), \
1133 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
1134 REQ_(TOGGLE_DATE, "Toggle date display"), \
1135 REQ_(TOGGLE_DATE_SHORT, "Toggle short (date-only) dates"), \
1136 REQ_(TOGGLE_AUTHOR, "Toggle author display"), \
1137 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization"), \
1138 REQ_(TOGGLE_REFS, "Toggle reference display (tags/branches)"), \
1139 REQ_(TOGGLE_SORT_ORDER, "Toggle ascending/descending sort order"), \
1140 REQ_(TOGGLE_SORT_FIELD, "Toggle field to sort by"), \
1141 \
1142 REQ_GROUP("Misc") \
1143 REQ_(PROMPT, "Bring up the prompt"), \
1144 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
1145 REQ_(SHOW_VERSION, "Show version information"), \
1146 REQ_(STOP_LOADING, "Stop all loading views"), \
1147 REQ_(EDIT, "Open in editor"), \
1148 REQ_(NONE, "Do nothing")
1151 /* User action requests. */
1152 enum request {
1153 #define REQ_GROUP(help)
1154 #define REQ_(req, help) REQ_##req
1156 /* Offset all requests to avoid conflicts with ncurses getch values. */
1157 REQ_UNKNOWN = KEY_MAX + 1,
1158 REQ_OFFSET,
1159 REQ_INFO
1161 #undef REQ_GROUP
1162 #undef REQ_
1163 };
1165 struct request_info {
1166 enum request request;
1167 const char *name;
1168 int namelen;
1169 const char *help;
1170 };
1172 static const struct request_info req_info[] = {
1173 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
1174 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
1175 REQ_INFO
1176 #undef REQ_GROUP
1177 #undef REQ_
1178 };
1180 static enum request
1181 get_request(const char *name)
1182 {
1183 int namelen = strlen(name);
1184 int i;
1186 for (i = 0; i < ARRAY_SIZE(req_info); i++)
1187 if (enum_equals(req_info[i], name, namelen))
1188 return req_info[i].request;
1190 return REQ_UNKNOWN;
1191 }
1194 /*
1195 * Options
1196 */
1198 /* Option and state variables. */
1199 static enum date opt_date = DATE_DEFAULT;
1200 static enum author opt_author = AUTHOR_DEFAULT;
1201 static bool opt_line_number = FALSE;
1202 static bool opt_line_graphics = TRUE;
1203 static bool opt_rev_graph = FALSE;
1204 static bool opt_show_refs = TRUE;
1205 static int opt_num_interval = 5;
1206 static double opt_hscroll = 0.50;
1207 static double opt_scale_split_view = 2.0 / 3.0;
1208 static int opt_tab_size = 8;
1209 static int opt_author_cols = AUTHOR_COLS;
1210 static char opt_path[SIZEOF_STR] = "";
1211 static char opt_file[SIZEOF_STR] = "";
1212 static char opt_ref[SIZEOF_REF] = "";
1213 static char opt_head[SIZEOF_REF] = "";
1214 static char opt_remote[SIZEOF_REF] = "";
1215 static char opt_encoding[20] = "UTF-8";
1216 static iconv_t opt_iconv_in = ICONV_NONE;
1217 static iconv_t opt_iconv_out = ICONV_NONE;
1218 static char opt_search[SIZEOF_STR] = "";
1219 static char opt_cdup[SIZEOF_STR] = "";
1220 static char opt_prefix[SIZEOF_STR] = "";
1221 static char opt_git_dir[SIZEOF_STR] = "";
1222 static signed char opt_is_inside_work_tree = -1; /* set to TRUE or FALSE */
1223 static char opt_editor[SIZEOF_STR] = "";
1224 static FILE *opt_tty = NULL;
1226 #define is_initial_commit() (!get_ref_head())
1227 #define is_head_commit(rev) (!strcmp((rev), "HEAD") || (get_ref_head() && !strcmp(rev, get_ref_head()->id)))
1230 /*
1231 * Line-oriented content detection.
1232 */
1234 #define LINE_INFO \
1235 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1236 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1237 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
1238 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
1239 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
1240 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1241 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1242 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1243 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1244 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1245 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1246 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1247 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1248 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
1249 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
1250 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1251 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
1252 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1253 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1254 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1255 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
1256 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
1257 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
1258 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
1259 LINE(AUTHOR, "author ", COLOR_GREEN, COLOR_DEFAULT, 0), \
1260 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1261 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1262 LINE(ACKED, " Acked-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1263 LINE(TESTED, " Tested-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1264 LINE(REVIEWED, " Reviewed-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1265 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
1266 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
1267 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
1268 LINE(DELIMITER, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1269 LINE(DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
1270 LINE(MODE, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
1271 LINE(LINE_NUMBER, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
1272 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
1273 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
1274 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
1275 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
1276 LINE(MAIN_LOCAL_TAG,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1277 LINE(MAIN_REMOTE, "", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1278 LINE(MAIN_TRACKED, "", COLOR_YELLOW, COLOR_DEFAULT, A_BOLD), \
1279 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
1280 LINE(MAIN_HEAD, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
1281 LINE(MAIN_REVGRAPH,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1282 LINE(TREE_HEAD, "", COLOR_DEFAULT, COLOR_DEFAULT, A_BOLD), \
1283 LINE(TREE_DIR, "", COLOR_YELLOW, COLOR_DEFAULT, A_NORMAL), \
1284 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
1285 LINE(STAT_HEAD, "", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1286 LINE(STAT_SECTION, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
1287 LINE(STAT_NONE, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
1288 LINE(STAT_STAGED, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1289 LINE(STAT_UNSTAGED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1290 LINE(STAT_UNTRACKED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1291 LINE(HELP_KEYMAP, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
1292 LINE(HELP_GROUP, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
1293 LINE(BLAME_ID, "", COLOR_MAGENTA, COLOR_DEFAULT, 0)
1295 enum line_type {
1296 #define LINE(type, line, fg, bg, attr) \
1297 LINE_##type
1298 LINE_INFO,
1299 LINE_NONE
1300 #undef LINE
1301 };
1303 struct line_info {
1304 const char *name; /* Option name. */
1305 int namelen; /* Size of option name. */
1306 const char *line; /* The start of line to match. */
1307 int linelen; /* Size of string to match. */
1308 int fg, bg, attr; /* Color and text attributes for the lines. */
1309 };
1311 static struct line_info line_info[] = {
1312 #define LINE(type, line, fg, bg, attr) \
1313 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
1314 LINE_INFO
1315 #undef LINE
1316 };
1318 static enum line_type
1319 get_line_type(const char *line)
1320 {
1321 int linelen = strlen(line);
1322 enum line_type type;
1324 for (type = 0; type < ARRAY_SIZE(line_info); type++)
1325 /* Case insensitive search matches Signed-off-by lines better. */
1326 if (linelen >= line_info[type].linelen &&
1327 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
1328 return type;
1330 return LINE_DEFAULT;
1331 }
1333 static inline int
1334 get_line_attr(enum line_type type)
1335 {
1336 assert(type < ARRAY_SIZE(line_info));
1337 return COLOR_PAIR(type) | line_info[type].attr;
1338 }
1340 static struct line_info *
1341 get_line_info(const char *name)
1342 {
1343 size_t namelen = strlen(name);
1344 enum line_type type;
1346 for (type = 0; type < ARRAY_SIZE(line_info); type++)
1347 if (enum_equals(line_info[type], name, namelen))
1348 return &line_info[type];
1350 return NULL;
1351 }
1353 static void
1354 init_colors(void)
1355 {
1356 int default_bg = line_info[LINE_DEFAULT].bg;
1357 int default_fg = line_info[LINE_DEFAULT].fg;
1358 enum line_type type;
1360 start_color();
1362 if (assume_default_colors(default_fg, default_bg) == ERR) {
1363 default_bg = COLOR_BLACK;
1364 default_fg = COLOR_WHITE;
1365 }
1367 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
1368 struct line_info *info = &line_info[type];
1369 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
1370 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
1372 init_pair(type, fg, bg);
1373 }
1374 }
1376 struct line {
1377 enum line_type type;
1379 /* State flags */
1380 unsigned int selected:1;
1381 unsigned int dirty:1;
1382 unsigned int cleareol:1;
1383 unsigned int other:16;
1385 void *data; /* User data */
1386 };
1389 /*
1390 * Keys
1391 */
1393 struct keybinding {
1394 int alias;
1395 enum request request;
1396 };
1398 static struct keybinding default_keybindings[] = {
1399 /* View switching */
1400 { 'm', REQ_VIEW_MAIN },
1401 { 'd', REQ_VIEW_DIFF },
1402 { 'l', REQ_VIEW_LOG },
1403 { 't', REQ_VIEW_TREE },
1404 { 'f', REQ_VIEW_BLOB },
1405 { 'B', REQ_VIEW_BLAME },
1406 { 'H', REQ_VIEW_BRANCH },
1407 { 'p', REQ_VIEW_PAGER },
1408 { 'h', REQ_VIEW_HELP },
1409 { 'S', REQ_VIEW_STATUS },
1410 { 'c', REQ_VIEW_STAGE },
1412 /* View manipulation */
1413 { 'q', REQ_VIEW_CLOSE },
1414 { KEY_TAB, REQ_VIEW_NEXT },
1415 { KEY_RETURN, REQ_ENTER },
1416 { KEY_UP, REQ_PREVIOUS },
1417 { KEY_DOWN, REQ_NEXT },
1418 { 'R', REQ_REFRESH },
1419 { KEY_F(5), REQ_REFRESH },
1420 { 'O', REQ_MAXIMIZE },
1422 /* Cursor navigation */
1423 { 'k', REQ_MOVE_UP },
1424 { 'j', REQ_MOVE_DOWN },
1425 { KEY_HOME, REQ_MOVE_FIRST_LINE },
1426 { KEY_END, REQ_MOVE_LAST_LINE },
1427 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
1428 { ' ', REQ_MOVE_PAGE_DOWN },
1429 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
1430 { 'b', REQ_MOVE_PAGE_UP },
1431 { '-', REQ_MOVE_PAGE_UP },
1433 /* Scrolling */
1434 { KEY_LEFT, REQ_SCROLL_LEFT },
1435 { KEY_RIGHT, REQ_SCROLL_RIGHT },
1436 { KEY_IC, REQ_SCROLL_LINE_UP },
1437 { KEY_DC, REQ_SCROLL_LINE_DOWN },
1438 { 'w', REQ_SCROLL_PAGE_UP },
1439 { 's', REQ_SCROLL_PAGE_DOWN },
1441 /* Searching */
1442 { '/', REQ_SEARCH },
1443 { '?', REQ_SEARCH_BACK },
1444 { 'n', REQ_FIND_NEXT },
1445 { 'N', REQ_FIND_PREV },
1447 /* Misc */
1448 { 'Q', REQ_QUIT },
1449 { 'z', REQ_STOP_LOADING },
1450 { 'v', REQ_SHOW_VERSION },
1451 { 'r', REQ_SCREEN_REDRAW },
1452 { 'o', REQ_OPTIONS },
1453 { '.', REQ_TOGGLE_LINENO },
1454 { 'D', REQ_TOGGLE_DATE },
1455 { 'A', REQ_TOGGLE_AUTHOR },
1456 { 'g', REQ_TOGGLE_REV_GRAPH },
1457 { 'F', REQ_TOGGLE_REFS },
1458 { 'I', REQ_TOGGLE_SORT_ORDER },
1459 { 'i', REQ_TOGGLE_SORT_FIELD },
1460 { ':', REQ_PROMPT },
1461 { 'u', REQ_STATUS_UPDATE },
1462 { '!', REQ_STATUS_REVERT },
1463 { 'M', REQ_STATUS_MERGE },
1464 { '@', REQ_STAGE_NEXT },
1465 { ',', REQ_PARENT },
1466 { 'e', REQ_EDIT },
1467 };
1469 #define KEYMAP_INFO \
1470 KEYMAP_(GENERIC), \
1471 KEYMAP_(MAIN), \
1472 KEYMAP_(DIFF), \
1473 KEYMAP_(LOG), \
1474 KEYMAP_(TREE), \
1475 KEYMAP_(BLOB), \
1476 KEYMAP_(BLAME), \
1477 KEYMAP_(BRANCH), \
1478 KEYMAP_(PAGER), \
1479 KEYMAP_(HELP), \
1480 KEYMAP_(STATUS), \
1481 KEYMAP_(STAGE)
1483 enum keymap {
1484 #define KEYMAP_(name) KEYMAP_##name
1485 KEYMAP_INFO
1486 #undef KEYMAP_
1487 };
1489 static const struct enum_map keymap_table[] = {
1490 #define KEYMAP_(name) ENUM_MAP(#name, KEYMAP_##name)
1491 KEYMAP_INFO
1492 #undef KEYMAP_
1493 };
1495 #define set_keymap(map, name) map_enum(map, keymap_table, name)
1497 struct keybinding_table {
1498 struct keybinding *data;
1499 size_t size;
1500 };
1502 static struct keybinding_table keybindings[ARRAY_SIZE(keymap_table)];
1504 static void
1505 add_keybinding(enum keymap keymap, enum request request, int key)
1506 {
1507 struct keybinding_table *table = &keybindings[keymap];
1508 size_t i;
1510 for (i = 0; i < keybindings[keymap].size; i++) {
1511 if (keybindings[keymap].data[i].alias == key) {
1512 keybindings[keymap].data[i].request = request;
1513 return;
1514 }
1515 }
1517 table->data = realloc(table->data, (table->size + 1) * sizeof(*table->data));
1518 if (!table->data)
1519 die("Failed to allocate keybinding");
1520 table->data[table->size].alias = key;
1521 table->data[table->size++].request = request;
1523 if (request == REQ_NONE && keymap == KEYMAP_GENERIC) {
1524 int i;
1526 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
1527 if (default_keybindings[i].alias == key)
1528 default_keybindings[i].request = REQ_NONE;
1529 }
1530 }
1532 /* Looks for a key binding first in the given map, then in the generic map, and
1533 * lastly in the default keybindings. */
1534 static enum request
1535 get_keybinding(enum keymap keymap, int key)
1536 {
1537 size_t i;
1539 for (i = 0; i < keybindings[keymap].size; i++)
1540 if (keybindings[keymap].data[i].alias == key)
1541 return keybindings[keymap].data[i].request;
1543 for (i = 0; i < keybindings[KEYMAP_GENERIC].size; i++)
1544 if (keybindings[KEYMAP_GENERIC].data[i].alias == key)
1545 return keybindings[KEYMAP_GENERIC].data[i].request;
1547 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
1548 if (default_keybindings[i].alias == key)
1549 return default_keybindings[i].request;
1551 return (enum request) key;
1552 }
1555 struct key {
1556 const char *name;
1557 int value;
1558 };
1560 static const struct key key_table[] = {
1561 { "Enter", KEY_RETURN },
1562 { "Space", ' ' },
1563 { "Backspace", KEY_BACKSPACE },
1564 { "Tab", KEY_TAB },
1565 { "Escape", KEY_ESC },
1566 { "Left", KEY_LEFT },
1567 { "Right", KEY_RIGHT },
1568 { "Up", KEY_UP },
1569 { "Down", KEY_DOWN },
1570 { "Insert", KEY_IC },
1571 { "Delete", KEY_DC },
1572 { "Hash", '#' },
1573 { "Home", KEY_HOME },
1574 { "End", KEY_END },
1575 { "PageUp", KEY_PPAGE },
1576 { "PageDown", KEY_NPAGE },
1577 { "F1", KEY_F(1) },
1578 { "F2", KEY_F(2) },
1579 { "F3", KEY_F(3) },
1580 { "F4", KEY_F(4) },
1581 { "F5", KEY_F(5) },
1582 { "F6", KEY_F(6) },
1583 { "F7", KEY_F(7) },
1584 { "F8", KEY_F(8) },
1585 { "F9", KEY_F(9) },
1586 { "F10", KEY_F(10) },
1587 { "F11", KEY_F(11) },
1588 { "F12", KEY_F(12) },
1589 };
1591 static int
1592 get_key_value(const char *name)
1593 {
1594 int i;
1596 for (i = 0; i < ARRAY_SIZE(key_table); i++)
1597 if (!strcasecmp(key_table[i].name, name))
1598 return key_table[i].value;
1600 if (strlen(name) == 1 && isprint(*name))
1601 return (int) *name;
1603 return ERR;
1604 }
1606 static const char *
1607 get_key_name(int key_value)
1608 {
1609 static char key_char[] = "'X'";
1610 const char *seq = NULL;
1611 int key;
1613 for (key = 0; key < ARRAY_SIZE(key_table); key++)
1614 if (key_table[key].value == key_value)
1615 seq = key_table[key].name;
1617 if (seq == NULL &&
1618 key_value < 127 &&
1619 isprint(key_value)) {
1620 key_char[1] = (char) key_value;
1621 seq = key_char;
1622 }
1624 return seq ? seq : "(no key)";
1625 }
1627 static bool
1628 append_key(char *buf, size_t *pos, const struct keybinding *keybinding)
1629 {
1630 const char *sep = *pos > 0 ? ", " : "";
1631 const char *keyname = get_key_name(keybinding->alias);
1633 return string_nformat(buf, BUFSIZ, pos, "%s%s", sep, keyname);
1634 }
1636 static bool
1637 append_keymap_request_keys(char *buf, size_t *pos, enum request request,
1638 enum keymap keymap, bool all)
1639 {
1640 int i;
1642 for (i = 0; i < keybindings[keymap].size; i++) {
1643 if (keybindings[keymap].data[i].request == request) {
1644 if (!append_key(buf, pos, &keybindings[keymap].data[i]))
1645 return FALSE;
1646 if (!all)
1647 break;
1648 }
1649 }
1651 return TRUE;
1652 }
1654 #define get_key(keymap, request) get_keys(keymap, request, FALSE)
1656 static const char *
1657 get_keys(enum keymap keymap, enum request request, bool all)
1658 {
1659 static char buf[BUFSIZ];
1660 size_t pos = 0;
1661 int i;
1663 buf[pos] = 0;
1665 if (!append_keymap_request_keys(buf, &pos, request, keymap, all))
1666 return "Too many keybindings!";
1667 if (pos > 0 && !all)
1668 return buf;
1670 if (keymap != KEYMAP_GENERIC) {
1671 /* Only the generic keymap includes the default keybindings when
1672 * listing all keys. */
1673 if (all)
1674 return buf;
1676 if (!append_keymap_request_keys(buf, &pos, request, KEYMAP_GENERIC, all))
1677 return "Too many keybindings!";
1678 if (pos)
1679 return buf;
1680 }
1682 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
1683 if (default_keybindings[i].request == request) {
1684 if (!append_key(buf, &pos, &default_keybindings[i]))
1685 return "Too many keybindings!";
1686 if (!all)
1687 return buf;
1688 }
1689 }
1691 return buf;
1692 }
1694 struct run_request {
1695 enum keymap keymap;
1696 int key;
1697 const char **argv;
1698 };
1700 static struct run_request *run_request;
1701 static size_t run_requests;
1703 DEFINE_ALLOCATOR(realloc_run_requests, struct run_request, 8)
1705 static enum request
1706 add_run_request(enum keymap keymap, int key, int argc, const char **argv)
1707 {
1708 struct run_request *req;
1710 if (!realloc_run_requests(&run_request, run_requests, 1))
1711 return REQ_NONE;
1713 req = &run_request[run_requests];
1714 req->keymap = keymap;
1715 req->key = key;
1716 req->argv = NULL;
1718 if (!argv_copy(&req->argv, argv))
1719 return REQ_NONE;
1721 return REQ_NONE + ++run_requests;
1722 }
1724 static struct run_request *
1725 get_run_request(enum request request)
1726 {
1727 if (request <= REQ_NONE)
1728 return NULL;
1729 return &run_request[request - REQ_NONE - 1];
1730 }
1732 static void
1733 add_builtin_run_requests(void)
1734 {
1735 const char *cherry_pick[] = { "git", "cherry-pick", "%(commit)", NULL };
1736 const char *checkout[] = { "git", "checkout", "%(branch)", NULL };
1737 const char *commit[] = { "git", "commit", NULL };
1738 const char *gc[] = { "git", "gc", NULL };
1739 struct {
1740 enum keymap keymap;
1741 int key;
1742 int argc;
1743 const char **argv;
1744 } reqs[] = {
1745 { KEYMAP_MAIN, 'C', ARRAY_SIZE(cherry_pick) - 1, cherry_pick },
1746 { KEYMAP_STATUS, 'C', ARRAY_SIZE(commit) - 1, commit },
1747 { KEYMAP_BRANCH, 'C', ARRAY_SIZE(checkout) - 1, checkout },
1748 { KEYMAP_GENERIC, 'G', ARRAY_SIZE(gc) - 1, gc },
1749 };
1750 int i;
1752 for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1753 enum request req = get_keybinding(reqs[i].keymap, reqs[i].key);
1755 if (req != reqs[i].key)
1756 continue;
1757 req = add_run_request(reqs[i].keymap, reqs[i].key, reqs[i].argc, reqs[i].argv);
1758 if (req != REQ_NONE)
1759 add_keybinding(reqs[i].keymap, req, reqs[i].key);
1760 }
1761 }
1763 /*
1764 * User config file handling.
1765 */
1767 static int config_lineno;
1768 static bool config_errors;
1769 static const char *config_msg;
1771 static const struct enum_map color_map[] = {
1772 #define COLOR_MAP(name) ENUM_MAP(#name, COLOR_##name)
1773 COLOR_MAP(DEFAULT),
1774 COLOR_MAP(BLACK),
1775 COLOR_MAP(BLUE),
1776 COLOR_MAP(CYAN),
1777 COLOR_MAP(GREEN),
1778 COLOR_MAP(MAGENTA),
1779 COLOR_MAP(RED),
1780 COLOR_MAP(WHITE),
1781 COLOR_MAP(YELLOW),
1782 };
1784 static const struct enum_map attr_map[] = {
1785 #define ATTR_MAP(name) ENUM_MAP(#name, A_##name)
1786 ATTR_MAP(NORMAL),
1787 ATTR_MAP(BLINK),
1788 ATTR_MAP(BOLD),
1789 ATTR_MAP(DIM),
1790 ATTR_MAP(REVERSE),
1791 ATTR_MAP(STANDOUT),
1792 ATTR_MAP(UNDERLINE),
1793 };
1795 #define set_attribute(attr, name) map_enum(attr, attr_map, name)
1797 static int parse_step(double *opt, const char *arg)
1798 {
1799 *opt = atoi(arg);
1800 if (!strchr(arg, '%'))
1801 return OK;
1803 /* "Shift down" so 100% and 1 does not conflict. */
1804 *opt = (*opt - 1) / 100;
1805 if (*opt >= 1.0) {
1806 *opt = 0.99;
1807 config_msg = "Step value larger than 100%";
1808 return ERR;
1809 }
1810 if (*opt < 0.0) {
1811 *opt = 1;
1812 config_msg = "Invalid step value";
1813 return ERR;
1814 }
1815 return OK;
1816 }
1818 static int
1819 parse_int(int *opt, const char *arg, int min, int max)
1820 {
1821 int value = atoi(arg);
1823 if (min <= value && value <= max) {
1824 *opt = value;
1825 return OK;
1826 }
1828 config_msg = "Integer value out of bound";
1829 return ERR;
1830 }
1832 static bool
1833 set_color(int *color, const char *name)
1834 {
1835 if (map_enum(color, color_map, name))
1836 return TRUE;
1837 if (!prefixcmp(name, "color"))
1838 return parse_int(color, name + 5, 0, 255) == OK;
1839 return FALSE;
1840 }
1842 /* Wants: object fgcolor bgcolor [attribute] */
1843 static int
1844 option_color_command(int argc, const char *argv[])
1845 {
1846 struct line_info *info;
1848 if (argc < 3) {
1849 config_msg = "Wrong number of arguments given to color command";
1850 return ERR;
1851 }
1853 info = get_line_info(argv[0]);
1854 if (!info) {
1855 static const struct enum_map obsolete[] = {
1856 ENUM_MAP("main-delim", LINE_DELIMITER),
1857 ENUM_MAP("main-date", LINE_DATE),
1858 ENUM_MAP("main-author", LINE_AUTHOR),
1859 };
1860 int index;
1862 if (!map_enum(&index, obsolete, argv[0])) {
1863 config_msg = "Unknown color name";
1864 return ERR;
1865 }
1866 info = &line_info[index];
1867 }
1869 if (!set_color(&info->fg, argv[1]) ||
1870 !set_color(&info->bg, argv[2])) {
1871 config_msg = "Unknown color";
1872 return ERR;
1873 }
1875 info->attr = 0;
1876 while (argc-- > 3) {
1877 int attr;
1879 if (!set_attribute(&attr, argv[argc])) {
1880 config_msg = "Unknown attribute";
1881 return ERR;
1882 }
1883 info->attr |= attr;
1884 }
1886 return OK;
1887 }
1889 static int parse_bool(bool *opt, const char *arg)
1890 {
1891 *opt = (!strcmp(arg, "1") || !strcmp(arg, "true") || !strcmp(arg, "yes"))
1892 ? TRUE : FALSE;
1893 return OK;
1894 }
1896 static int parse_enum_do(unsigned int *opt, const char *arg,
1897 const struct enum_map *map, size_t map_size)
1898 {
1899 bool is_true;
1901 assert(map_size > 1);
1903 if (map_enum_do(map, map_size, (int *) opt, arg))
1904 return OK;
1906 if (parse_bool(&is_true, arg) != OK)
1907 return ERR;
1909 *opt = is_true ? map[1].value : map[0].value;
1910 return OK;
1911 }
1913 #define parse_enum(opt, arg, map) \
1914 parse_enum_do(opt, arg, map, ARRAY_SIZE(map))
1916 static int
1917 parse_string(char *opt, const char *arg, size_t optsize)
1918 {
1919 int arglen = strlen(arg);
1921 switch (arg[0]) {
1922 case '\"':
1923 case '\'':
1924 if (arglen == 1 || arg[arglen - 1] != arg[0]) {
1925 config_msg = "Unmatched quotation";
1926 return ERR;
1927 }
1928 arg += 1; arglen -= 2;
1929 default:
1930 string_ncopy_do(opt, optsize, arg, arglen);
1931 return OK;
1932 }
1933 }
1935 /* Wants: name = value */
1936 static int
1937 option_set_command(int argc, const char *argv[])
1938 {
1939 if (argc != 3) {
1940 config_msg = "Wrong number of arguments given to set command";
1941 return ERR;
1942 }
1944 if (strcmp(argv[1], "=")) {
1945 config_msg = "No value assigned";
1946 return ERR;
1947 }
1949 if (!strcmp(argv[0], "show-author"))
1950 return parse_enum(&opt_author, argv[2], author_map);
1952 if (!strcmp(argv[0], "show-date"))
1953 return parse_enum(&opt_date, argv[2], date_map);
1955 if (!strcmp(argv[0], "show-rev-graph"))
1956 return parse_bool(&opt_rev_graph, argv[2]);
1958 if (!strcmp(argv[0], "show-refs"))
1959 return parse_bool(&opt_show_refs, argv[2]);
1961 if (!strcmp(argv[0], "show-line-numbers"))
1962 return parse_bool(&opt_line_number, argv[2]);
1964 if (!strcmp(argv[0], "line-graphics"))
1965 return parse_bool(&opt_line_graphics, argv[2]);
1967 if (!strcmp(argv[0], "line-number-interval"))
1968 return parse_int(&opt_num_interval, argv[2], 1, 1024);
1970 if (!strcmp(argv[0], "author-width"))
1971 return parse_int(&opt_author_cols, argv[2], 0, 1024);
1973 if (!strcmp(argv[0], "horizontal-scroll"))
1974 return parse_step(&opt_hscroll, argv[2]);
1976 if (!strcmp(argv[0], "split-view-height"))
1977 return parse_step(&opt_scale_split_view, argv[2]);
1979 if (!strcmp(argv[0], "tab-size"))
1980 return parse_int(&opt_tab_size, argv[2], 1, 1024);
1982 if (!strcmp(argv[0], "commit-encoding"))
1983 return parse_string(opt_encoding, argv[2], sizeof(opt_encoding));
1985 config_msg = "Unknown variable name";
1986 return ERR;
1987 }
1989 /* Wants: mode request key */
1990 static int
1991 option_bind_command(int argc, const char *argv[])
1992 {
1993 enum request request;
1994 int keymap = -1;
1995 int key;
1997 if (argc < 3) {
1998 config_msg = "Wrong number of arguments given to bind command";
1999 return ERR;
2000 }
2002 if (!set_keymap(&keymap, argv[0])) {
2003 config_msg = "Unknown key map";
2004 return ERR;
2005 }
2007 key = get_key_value(argv[1]);
2008 if (key == ERR) {
2009 config_msg = "Unknown key";
2010 return ERR;
2011 }
2013 request = get_request(argv[2]);
2014 if (request == REQ_UNKNOWN) {
2015 static const struct enum_map obsolete[] = {
2016 ENUM_MAP("cherry-pick", REQ_NONE),
2017 ENUM_MAP("screen-resize", REQ_NONE),
2018 ENUM_MAP("tree-parent", REQ_PARENT),
2019 };
2020 int alias;
2022 if (map_enum(&alias, obsolete, argv[2])) {
2023 if (alias != REQ_NONE)
2024 add_keybinding(keymap, alias, key);
2025 config_msg = "Obsolete request name";
2026 return ERR;
2027 }
2028 }
2029 if (request == REQ_UNKNOWN && *argv[2]++ == '!')
2030 request = add_run_request(keymap, key, argc - 2, argv + 2);
2031 if (request == REQ_UNKNOWN) {
2032 config_msg = "Unknown request name";
2033 return ERR;
2034 }
2036 add_keybinding(keymap, request, key);
2038 return OK;
2039 }
2041 static int
2042 set_option(const char *opt, char *value)
2043 {
2044 const char *argv[SIZEOF_ARG];
2045 int argc = 0;
2047 if (!argv_from_string(argv, &argc, value)) {
2048 config_msg = "Too many option arguments";
2049 return ERR;
2050 }
2052 if (!strcmp(opt, "color"))
2053 return option_color_command(argc, argv);
2055 if (!strcmp(opt, "set"))
2056 return option_set_command(argc, argv);
2058 if (!strcmp(opt, "bind"))
2059 return option_bind_command(argc, argv);
2061 config_msg = "Unknown option command";
2062 return ERR;
2063 }
2065 static int
2066 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
2067 {
2068 int status = OK;
2070 config_lineno++;
2071 config_msg = "Internal error";
2073 /* Check for comment markers, since read_properties() will
2074 * only ensure opt and value are split at first " \t". */
2075 optlen = strcspn(opt, "#");
2076 if (optlen == 0)
2077 return OK;
2079 if (opt[optlen] != 0) {
2080 config_msg = "No option value";
2081 status = ERR;
2083 } else {
2084 /* Look for comment endings in the value. */
2085 size_t len = strcspn(value, "#");
2087 if (len < valuelen) {
2088 valuelen = len;
2089 value[valuelen] = 0;
2090 }
2092 status = set_option(opt, value);
2093 }
2095 if (status == ERR) {
2096 warn("Error on line %d, near '%.*s': %s",
2097 config_lineno, (int) optlen, opt, config_msg);
2098 config_errors = TRUE;
2099 }
2101 /* Always keep going if errors are encountered. */
2102 return OK;
2103 }
2105 static void
2106 load_option_file(const char *path)
2107 {
2108 struct io io;
2110 /* It's OK that the file doesn't exist. */
2111 if (!io_open(&io, "%s", path))
2112 return;
2114 config_lineno = 0;
2115 config_errors = FALSE;
2117 if (io_load(&io, " \t", read_option) == ERR ||
2118 config_errors == TRUE)
2119 warn("Errors while loading %s.", path);
2120 }
2122 static int
2123 load_options(void)
2124 {
2125 const char *home = getenv("HOME");
2126 const char *tigrc_user = getenv("TIGRC_USER");
2127 const char *tigrc_system = getenv("TIGRC_SYSTEM");
2128 char buf[SIZEOF_STR];
2130 if (!tigrc_system)
2131 tigrc_system = SYSCONFDIR "/tigrc";
2132 load_option_file(tigrc_system);
2134 if (!tigrc_user) {
2135 if (!home || !string_format(buf, "%s/.tigrc", home))
2136 return ERR;
2137 tigrc_user = buf;
2138 }
2139 load_option_file(tigrc_user);
2141 /* Add _after_ loading config files to avoid adding run requests
2142 * that conflict with keybindings. */
2143 add_builtin_run_requests();
2145 return OK;
2146 }
2149 /*
2150 * The viewer
2151 */
2153 struct view;
2154 struct view_ops;
2156 /* The display array of active views and the index of the current view. */
2157 static struct view *display[2];
2158 static unsigned int current_view;
2160 #define foreach_displayed_view(view, i) \
2161 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
2163 #define displayed_views() (display[1] != NULL ? 2 : 1)
2165 /* Current head and commit ID */
2166 static char ref_blob[SIZEOF_REF] = "";
2167 static char ref_commit[SIZEOF_REF] = "HEAD";
2168 static char ref_head[SIZEOF_REF] = "HEAD";
2169 static char ref_branch[SIZEOF_REF] = "";
2171 enum view_type {
2172 VIEW_MAIN,
2173 VIEW_DIFF,
2174 VIEW_LOG,
2175 VIEW_TREE,
2176 VIEW_BLOB,
2177 VIEW_BLAME,
2178 VIEW_BRANCH,
2179 VIEW_HELP,
2180 VIEW_PAGER,
2181 VIEW_STATUS,
2182 VIEW_STAGE,
2183 };
2185 struct view {
2186 enum view_type type; /* View type */
2187 const char *name; /* View name */
2188 const char *cmd_env; /* Command line set via environment */
2189 const char *id; /* Points to either of ref_{head,commit,blob} */
2191 struct view_ops *ops; /* View operations */
2193 enum keymap keymap; /* What keymap does this view have */
2194 bool git_dir; /* Whether the view requires a git directory. */
2196 char ref[SIZEOF_REF]; /* Hovered commit reference */
2197 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
2199 int height, width; /* The width and height of the main window */
2200 WINDOW *win; /* The main window */
2201 WINDOW *title; /* The title window living below the main window */
2203 /* Navigation */
2204 unsigned long offset; /* Offset of the window top */
2205 unsigned long yoffset; /* Offset from the window side. */
2206 unsigned long lineno; /* Current line number */
2207 unsigned long p_offset; /* Previous offset of the window top */
2208 unsigned long p_yoffset;/* Previous offset from the window side */
2209 unsigned long p_lineno; /* Previous current line number */
2210 bool p_restore; /* Should the previous position be restored. */
2212 /* Searching */
2213 char grep[SIZEOF_STR]; /* Search string */
2214 regex_t *regex; /* Pre-compiled regexp */
2216 /* If non-NULL, points to the view that opened this view. If this view
2217 * is closed tig will switch back to the parent view. */
2218 struct view *parent;
2219 struct view *prev;
2221 /* Buffering */
2222 size_t lines; /* Total number of lines */
2223 struct line *line; /* Line index */
2224 unsigned int digits; /* Number of digits in the lines member. */
2226 /* Drawing */
2227 struct line *curline; /* Line currently being drawn. */
2228 enum line_type curtype; /* Attribute currently used for drawing. */
2229 unsigned long col; /* Column when drawing. */
2230 bool has_scrolled; /* View was scrolled. */
2232 /* Loading */
2233 const char **argv; /* Shell command arguments. */
2234 const char *dir; /* Directory from which to execute. */
2235 struct io io;
2236 struct io *pipe;
2237 time_t start_time;
2238 time_t update_secs;
2239 };
2241 struct view_ops {
2242 /* What type of content being displayed. Used in the title bar. */
2243 const char *type;
2244 /* Default command arguments. */
2245 const char **argv;
2246 /* Open and reads in all view content. */
2247 bool (*open)(struct view *view);
2248 /* Read one line; updates view->line. */
2249 bool (*read)(struct view *view, char *data);
2250 /* Draw one line; @lineno must be < view->height. */
2251 bool (*draw)(struct view *view, struct line *line, unsigned int lineno);
2252 /* Depending on view handle a special requests. */
2253 enum request (*request)(struct view *view, enum request request, struct line *line);
2254 /* Search for regexp in a line. */
2255 bool (*grep)(struct view *view, struct line *line);
2256 /* Select line */
2257 void (*select)(struct view *view, struct line *line);
2258 /* Prepare view for loading */
2259 bool (*prepare)(struct view *view);
2260 };
2262 static struct view_ops blame_ops;
2263 static struct view_ops blob_ops;
2264 static struct view_ops diff_ops;
2265 static struct view_ops help_ops;
2266 static struct view_ops log_ops;
2267 static struct view_ops main_ops;
2268 static struct view_ops pager_ops;
2269 static struct view_ops stage_ops;
2270 static struct view_ops status_ops;
2271 static struct view_ops tree_ops;
2272 static struct view_ops branch_ops;
2274 #define VIEW_STR(type, name, env, ref, ops, map, git) \
2275 { type, name, #env, ref, ops, map, git }
2277 #define VIEW_(id, name, ops, git, ref) \
2278 VIEW_STR(VIEW_##id, name, TIG_##id##_CMD, ref, ops, KEYMAP_##id, git)
2280 static struct view views[] = {
2281 VIEW_(MAIN, "main", &main_ops, TRUE, ref_head),
2282 VIEW_(DIFF, "diff", &diff_ops, TRUE, ref_commit),
2283 VIEW_(LOG, "log", &log_ops, TRUE, ref_head),
2284 VIEW_(TREE, "tree", &tree_ops, TRUE, ref_commit),
2285 VIEW_(BLOB, "blob", &blob_ops, TRUE, ref_blob),
2286 VIEW_(BLAME, "blame", &blame_ops, TRUE, ref_commit),
2287 VIEW_(BRANCH, "branch", &branch_ops, TRUE, ref_head),
2288 VIEW_(HELP, "help", &help_ops, FALSE, ""),
2289 VIEW_(PAGER, "pager", &pager_ops, FALSE, "stdin"),
2290 VIEW_(STATUS, "status", &status_ops, TRUE, ""),
2291 VIEW_(STAGE, "stage", &stage_ops, TRUE, ""),
2292 };
2294 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
2296 #define foreach_view(view, i) \
2297 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
2299 #define view_is_displayed(view) \
2300 (view == display[0] || view == display[1])
2302 static enum request
2303 view_request(struct view *view, enum request request)
2304 {
2305 if (!view || !view->lines)
2306 return request;
2307 return view->ops->request(view, request, &view->line[view->lineno]);
2308 }
2311 /*
2312 * View drawing.
2313 */
2315 static inline void
2316 set_view_attr(struct view *view, enum line_type type)
2317 {
2318 if (!view->curline->selected && view->curtype != type) {
2319 (void) wattrset(view->win, get_line_attr(type));
2320 wchgat(view->win, -1, 0, type, NULL);
2321 view->curtype = type;
2322 }
2323 }
2325 static int
2326 draw_chars(struct view *view, enum line_type type, const char *string,
2327 int max_len, bool use_tilde)
2328 {
2329 static char out_buffer[BUFSIZ * 2];
2330 int len = 0;
2331 int col = 0;
2332 int trimmed = FALSE;
2333 size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
2335 if (max_len <= 0)
2336 return 0;
2338 len = utf8_length(&string, skip, &col, max_len, &trimmed, use_tilde, opt_tab_size);
2340 set_view_attr(view, type);
2341 if (len > 0) {
2342 if (opt_iconv_out != ICONV_NONE) {
2343 ICONV_CONST char *inbuf = (ICONV_CONST char *) string;
2344 size_t inlen = len + 1;
2346 char *outbuf = out_buffer;
2347 size_t outlen = sizeof(out_buffer);
2349 size_t ret;
2351 ret = iconv(opt_iconv_out, &inbuf, &inlen, &outbuf, &outlen);
2352 if (ret != (size_t) -1) {
2353 string = out_buffer;
2354 len = sizeof(out_buffer) - outlen;
2355 }
2356 }
2358 waddnstr(view->win, string, len);
2359 }
2360 if (trimmed && use_tilde) {
2361 set_view_attr(view, LINE_DELIMITER);
2362 waddch(view->win, '~');
2363 col++;
2364 }
2366 return col;
2367 }
2369 static int
2370 draw_space(struct view *view, enum line_type type, int max, int spaces)
2371 {
2372 static char space[] = " ";
2373 int col = 0;
2375 spaces = MIN(max, spaces);
2377 while (spaces > 0) {
2378 int len = MIN(spaces, sizeof(space) - 1);
2380 col += draw_chars(view, type, space, len, FALSE);
2381 spaces -= len;
2382 }
2384 return col;
2385 }
2387 static bool
2388 draw_text(struct view *view, enum line_type type, const char *string, bool trim)
2389 {
2390 view->col += draw_chars(view, type, string, view->width + view->yoffset - view->col, trim);
2391 return view->width + view->yoffset <= view->col;
2392 }
2394 static bool
2395 draw_graphic(struct view *view, enum line_type type, chtype graphic[], size_t size)
2396 {
2397 size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
2398 int max = view->width + view->yoffset - view->col;
2399 int i;
2401 if (max < size)
2402 size = max;
2404 set_view_attr(view, type);
2405 /* Using waddch() instead of waddnstr() ensures that
2406 * they'll be rendered correctly for the cursor line. */
2407 for (i = skip; i < size; i++)
2408 waddch(view->win, graphic[i]);
2410 view->col += size;
2411 if (size < max && skip <= size)
2412 waddch(view->win, ' ');
2413 view->col++;
2415 return view->width + view->yoffset <= view->col;
2416 }
2418 static bool
2419 draw_field(struct view *view, enum line_type type, const char *text, int len, bool trim)
2420 {
2421 int max = MIN(view->width + view->yoffset - view->col, len);
2422 int col;
2424 if (text)
2425 col = draw_chars(view, type, text, max - 1, trim);
2426 else
2427 col = draw_space(view, type, max - 1, max - 1);
2429 view->col += col;
2430 view->col += draw_space(view, LINE_DEFAULT, max - col, max - col);
2431 return view->width + view->yoffset <= view->col;
2432 }
2434 static bool
2435 draw_date(struct view *view, struct time *time)
2436 {
2437 const char *date = mkdate(time, opt_date);
2438 int cols = opt_date == DATE_SHORT ? DATE_SHORT_COLS : DATE_COLS;
2440 return draw_field(view, LINE_DATE, date, cols, FALSE);
2441 }
2443 static bool
2444 draw_author(struct view *view, const char *author)
2445 {
2446 bool trim = opt_author_cols == 0 || opt_author_cols > 5;
2447 bool abbreviate = opt_author == AUTHOR_ABBREVIATED || !trim;
2449 if (abbreviate && author)
2450 author = get_author_initials(author);
2452 return draw_field(view, LINE_AUTHOR, author, opt_author_cols, trim);
2453 }
2455 static bool
2456 draw_mode(struct view *view, mode_t mode)
2457 {
2458 const char *str;
2460 if (S_ISDIR(mode))
2461 str = "drwxr-xr-x";
2462 else if (S_ISLNK(mode))
2463 str = "lrwxrwxrwx";
2464 else if (S_ISGITLINK(mode))
2465 str = "m---------";
2466 else if (S_ISREG(mode) && mode & S_IXUSR)
2467 str = "-rwxr-xr-x";
2468 else if (S_ISREG(mode))
2469 str = "-rw-r--r--";
2470 else
2471 str = "----------";
2473 return draw_field(view, LINE_MODE, str, STRING_SIZE("-rw-r--r-- "), FALSE);
2474 }
2476 static bool
2477 draw_lineno(struct view *view, unsigned int lineno)
2478 {
2479 char number[10];
2480 int digits3 = view->digits < 3 ? 3 : view->digits;
2481 int max = MIN(view->width + view->yoffset - view->col, digits3);
2482 char *text = NULL;
2483 chtype separator = opt_line_graphics ? ACS_VLINE : '|';
2485 lineno += view->offset + 1;
2486 if (lineno == 1 || (lineno % opt_num_interval) == 0) {
2487 static char fmt[] = "%1ld";
2489 fmt[1] = '0' + (view->digits <= 9 ? digits3 : 1);
2490 if (string_format(number, fmt, lineno))
2491 text = number;
2492 }
2493 if (text)
2494 view->col += draw_chars(view, LINE_LINE_NUMBER, text, max, TRUE);
2495 else
2496 view->col += draw_space(view, LINE_LINE_NUMBER, max, digits3);
2497 return draw_graphic(view, LINE_DEFAULT, &separator, 1);
2498 }
2500 static bool
2501 draw_view_line(struct view *view, unsigned int lineno)
2502 {
2503 struct line *line;
2504 bool selected = (view->offset + lineno == view->lineno);
2506 assert(view_is_displayed(view));
2508 if (view->offset + lineno >= view->lines)
2509 return FALSE;
2511 line = &view->line[view->offset + lineno];
2513 wmove(view->win, lineno, 0);
2514 if (line->cleareol)
2515 wclrtoeol(view->win);
2516 view->col = 0;
2517 view->curline = line;
2518 view->curtype = LINE_NONE;
2519 line->selected = FALSE;
2520 line->dirty = line->cleareol = 0;
2522 if (selected) {
2523 set_view_attr(view, LINE_CURSOR);
2524 line->selected = TRUE;
2525 view->ops->select(view, line);
2526 }
2528 return view->ops->draw(view, line, lineno);
2529 }
2531 static void
2532 redraw_view_dirty(struct view *view)
2533 {
2534 bool dirty = FALSE;
2535 int lineno;
2537 for (lineno = 0; lineno < view->height; lineno++) {
2538 if (view->offset + lineno >= view->lines)
2539 break;
2540 if (!view->line[view->offset + lineno].dirty)
2541 continue;
2542 dirty = TRUE;
2543 if (!draw_view_line(view, lineno))
2544 break;
2545 }
2547 if (!dirty)
2548 return;
2549 wnoutrefresh(view->win);
2550 }
2552 static void
2553 redraw_view_from(struct view *view, int lineno)
2554 {
2555 assert(0 <= lineno && lineno < view->height);
2557 for (; lineno < view->height; lineno++) {
2558 if (!draw_view_line(view, lineno))
2559 break;
2560 }
2562 wnoutrefresh(view->win);
2563 }
2565 static void
2566 redraw_view(struct view *view)
2567 {
2568 werase(view->win);
2569 redraw_view_from(view, 0);
2570 }
2573 static void
2574 update_view_title(struct view *view)
2575 {
2576 char buf[SIZEOF_STR];
2577 char state[SIZEOF_STR];
2578 size_t bufpos = 0, statelen = 0;
2580 assert(view_is_displayed(view));
2582 if (view->type != VIEW_STATUS && view->lines) {
2583 unsigned int view_lines = view->offset + view->height;
2584 unsigned int lines = view->lines
2585 ? MIN(view_lines, view->lines) * 100 / view->lines
2586 : 0;
2588 string_format_from(state, &statelen, " - %s %d of %d (%d%%)",
2589 view->ops->type,
2590 view->lineno + 1,
2591 view->lines,
2592 lines);
2594 }
2596 if (view->pipe) {
2597 time_t secs = time(NULL) - view->start_time;
2599 /* Three git seconds are a long time ... */
2600 if (secs > 2)
2601 string_format_from(state, &statelen, " loading %lds", secs);
2602 }
2604 string_format_from(buf, &bufpos, "[%s]", view->name);
2605 if (*view->ref && bufpos < view->width) {
2606 size_t refsize = strlen(view->ref);
2607 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
2609 if (minsize < view->width)
2610 refsize = view->width - minsize + 7;
2611 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
2612 }
2614 if (statelen && bufpos < view->width) {
2615 string_format_from(buf, &bufpos, "%s", state);
2616 }
2618 if (view == display[current_view])
2619 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
2620 else
2621 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
2623 mvwaddnstr(view->title, 0, 0, buf, bufpos);
2624 wclrtoeol(view->title);
2625 wnoutrefresh(view->title);
2626 }
2628 static int
2629 apply_step(double step, int value)
2630 {
2631 if (step >= 1)
2632 return (int) step;
2633 value *= step + 0.01;
2634 return value ? value : 1;
2635 }
2637 static void
2638 resize_display(void)
2639 {
2640 int offset, i;
2641 struct view *base = display[0];
2642 struct view *view = display[1] ? display[1] : display[0];
2644 /* Setup window dimensions */
2646 getmaxyx(stdscr, base->height, base->width);
2648 /* Make room for the status window. */
2649 base->height -= 1;
2651 if (view != base) {
2652 /* Horizontal split. */
2653 view->width = base->width;
2654 view->height = apply_step(opt_scale_split_view, base->height);
2655 view->height = MAX(view->height, MIN_VIEW_HEIGHT);
2656 view->height = MIN(view->height, base->height - MIN_VIEW_HEIGHT);
2657 base->height -= view->height;
2659 /* Make room for the title bar. */
2660 view->height -= 1;
2661 }
2663 /* Make room for the title bar. */
2664 base->height -= 1;
2666 offset = 0;
2668 foreach_displayed_view (view, i) {
2669 if (!view->win) {
2670 view->win = newwin(view->height, 0, offset, 0);
2671 if (!view->win)
2672 die("Failed to create %s view", view->name);
2674 scrollok(view->win, FALSE);
2676 view->title = newwin(1, 0, offset + view->height, 0);
2677 if (!view->title)
2678 die("Failed to create title window");
2680 } else {
2681 wresize(view->win, view->height, view->width);
2682 mvwin(view->win, offset, 0);
2683 mvwin(view->title, offset + view->height, 0);
2684 }
2686 offset += view->height + 1;
2687 }
2688 }
2690 static void
2691 redraw_display(bool clear)
2692 {
2693 struct view *view;
2694 int i;
2696 foreach_displayed_view (view, i) {
2697 if (clear)
2698 wclear(view->win);
2699 redraw_view(view);
2700 update_view_title(view);
2701 }
2702 }
2705 /*
2706 * Option management
2707 */
2709 static void
2710 toggle_enum_option_do(unsigned int *opt, const char *help,
2711 const struct enum_map *map, size_t size)
2712 {
2713 *opt = (*opt + 1) % size;
2714 redraw_display(FALSE);
2715 report("Displaying %s %s", enum_name(map[*opt]), help);
2716 }
2718 #define toggle_enum_option(opt, help, map) \
2719 toggle_enum_option_do(opt, help, map, ARRAY_SIZE(map))
2721 #define toggle_date() toggle_enum_option(&opt_date, "dates", date_map)
2722 #define toggle_author() toggle_enum_option(&opt_author, "author names", author_map)
2724 static void
2725 toggle_view_option(bool *option, const char *help)
2726 {
2727 *option = !*option;
2728 redraw_display(FALSE);
2729 report("%sabling %s", *option ? "En" : "Dis", help);
2730 }
2732 static void
2733 open_option_menu(void)
2734 {
2735 const struct menu_item menu[] = {
2736 { '.', "line numbers", &opt_line_number },
2737 { 'D', "date display", &opt_date },
2738 { 'A', "author display", &opt_author },
2739 { 'g', "revision graph display", &opt_rev_graph },
2740 { 'F', "reference display", &opt_show_refs },
2741 { 0 }
2742 };
2743 int selected = 0;
2745 if (prompt_menu("Toggle option", menu, &selected)) {
2746 if (menu[selected].data == &opt_date)
2747 toggle_date();
2748 else if (menu[selected].data == &opt_author)
2749 toggle_author();
2750 else
2751 toggle_view_option(menu[selected].data, menu[selected].text);
2752 }
2753 }
2755 static void
2756 maximize_view(struct view *view)
2757 {
2758 memset(display, 0, sizeof(display));
2759 current_view = 0;
2760 display[current_view] = view;
2761 resize_display();
2762 redraw_display(FALSE);
2763 report("");
2764 }
2767 /*
2768 * Navigation
2769 */
2771 static bool
2772 goto_view_line(struct view *view, unsigned long offset, unsigned long lineno)
2773 {
2774 if (lineno >= view->lines)
2775 lineno = view->lines > 0 ? view->lines - 1 : 0;
2777 if (offset > lineno || offset + view->height <= lineno) {
2778 unsigned long half = view->height / 2;
2780 if (lineno > half)
2781 offset = lineno - half;
2782 else
2783 offset = 0;
2784 }
2786 if (offset != view->offset || lineno != view->lineno) {
2787 view->offset = offset;
2788 view->lineno = lineno;
2789 return TRUE;
2790 }
2792 return FALSE;
2793 }
2795 /* Scrolling backend */
2796 static void
2797 do_scroll_view(struct view *view, int lines)
2798 {
2799 bool redraw_current_line = FALSE;
2801 /* The rendering expects the new offset. */
2802 view->offset += lines;
2804 assert(0 <= view->offset && view->offset < view->lines);
2805 assert(lines);
2807 /* Move current line into the view. */
2808 if (view->lineno < view->offset) {
2809 view->lineno = view->offset;
2810 redraw_current_line = TRUE;
2811 } else if (view->lineno >= view->offset + view->height) {
2812 view->lineno = view->offset + view->height - 1;
2813 redraw_current_line = TRUE;
2814 }
2816 assert(view->offset <= view->lineno && view->lineno < view->lines);
2818 /* Redraw the whole screen if scrolling is pointless. */
2819 if (view->height < ABS(lines)) {
2820 redraw_view(view);
2822 } else {
2823 int line = lines > 0 ? view->height - lines : 0;
2824 int end = line + ABS(lines);
2826 scrollok(view->win, TRUE);
2827 wscrl(view->win, lines);
2828 scrollok(view->win, FALSE);
2830 while (line < end && draw_view_line(view, line))
2831 line++;
2833 if (redraw_current_line)
2834 draw_view_line(view, view->lineno - view->offset);
2835 wnoutrefresh(view->win);
2836 }
2838 view->has_scrolled = TRUE;
2839 report("");
2840 }
2842 /* Scroll frontend */
2843 static void
2844 scroll_view(struct view *view, enum request request)
2845 {
2846 int lines = 1;
2848 assert(view_is_displayed(view));
2850 switch (request) {
2851 case REQ_SCROLL_LEFT:
2852 if (view->yoffset == 0) {
2853 report("Cannot scroll beyond the first column");
2854 return;
2855 }
2856 if (view->yoffset <= apply_step(opt_hscroll, view->width))
2857 view->yoffset = 0;
2858 else
2859 view->yoffset -= apply_step(opt_hscroll, view->width);
2860 redraw_view_from(view, 0);
2861 report("");
2862 return;
2863 case REQ_SCROLL_RIGHT:
2864 view->yoffset += apply_step(opt_hscroll, view->width);
2865 redraw_view(view);
2866 report("");
2867 return;
2868 case REQ_SCROLL_PAGE_DOWN:
2869 lines = view->height;
2870 case REQ_SCROLL_LINE_DOWN:
2871 if (view->offset + lines > view->lines)
2872 lines = view->lines - view->offset;
2874 if (lines == 0 || view->offset + view->height >= view->lines) {
2875 report("Cannot scroll beyond the last line");
2876 return;
2877 }
2878 break;
2880 case REQ_SCROLL_PAGE_UP:
2881 lines = view->height;
2882 case REQ_SCROLL_LINE_UP:
2883 if (lines > view->offset)
2884 lines = view->offset;
2886 if (lines == 0) {
2887 report("Cannot scroll beyond the first line");
2888 return;
2889 }
2891 lines = -lines;
2892 break;
2894 default:
2895 die("request %d not handled in switch", request);
2896 }
2898 do_scroll_view(view, lines);
2899 }
2901 /* Cursor moving */
2902 static void
2903 move_view(struct view *view, enum request request)
2904 {
2905 int scroll_steps = 0;
2906 int steps;
2908 switch (request) {
2909 case REQ_MOVE_FIRST_LINE:
2910 steps = -view->lineno;
2911 break;
2913 case REQ_MOVE_LAST_LINE:
2914 steps = view->lines - view->lineno - 1;
2915 break;
2917 case REQ_MOVE_PAGE_UP:
2918 steps = view->height > view->lineno
2919 ? -view->lineno : -view->height;
2920 break;
2922 case REQ_MOVE_PAGE_DOWN:
2923 steps = view->lineno + view->height >= view->lines
2924 ? view->lines - view->lineno - 1 : view->height;
2925 break;
2927 case REQ_MOVE_UP:
2928 steps = -1;
2929 break;
2931 case REQ_MOVE_DOWN:
2932 steps = 1;
2933 break;
2935 default:
2936 die("request %d not handled in switch", request);
2937 }
2939 if (steps <= 0 && view->lineno == 0) {
2940 report("Cannot move beyond the first line");
2941 return;
2943 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
2944 report("Cannot move beyond the last line");
2945 return;
2946 }
2948 /* Move the current line */
2949 view->lineno += steps;
2950 assert(0 <= view->lineno && view->lineno < view->lines);
2952 /* Check whether the view needs to be scrolled */
2953 if (view->lineno < view->offset ||
2954 view->lineno >= view->offset + view->height) {
2955 scroll_steps = steps;
2956 if (steps < 0 && -steps > view->offset) {
2957 scroll_steps = -view->offset;
2959 } else if (steps > 0) {
2960 if (view->lineno == view->lines - 1 &&
2961 view->lines > view->height) {
2962 scroll_steps = view->lines - view->offset - 1;
2963 if (scroll_steps >= view->height)
2964 scroll_steps -= view->height - 1;
2965 }
2966 }
2967 }
2969 if (!view_is_displayed(view)) {
2970 view->offset += scroll_steps;
2971 assert(0 <= view->offset && view->offset < view->lines);
2972 view->ops->select(view, &view->line[view->lineno]);
2973 return;
2974 }
2976 /* Repaint the old "current" line if we be scrolling */
2977 if (ABS(steps) < view->height)
2978 draw_view_line(view, view->lineno - steps - view->offset);
2980 if (scroll_steps) {
2981 do_scroll_view(view, scroll_steps);
2982 return;
2983 }
2985 /* Draw the current line */
2986 draw_view_line(view, view->lineno - view->offset);
2988 wnoutrefresh(view->win);
2989 report("");
2990 }
2993 /*
2994 * Searching
2995 */
2997 static void search_view(struct view *view, enum request request);
2999 static bool
3000 grep_text(struct view *view, const char *text[])
3001 {
3002 regmatch_t pmatch;
3003 size_t i;
3005 for (i = 0; text[i]; i++)
3006 if (*text[i] &&
3007 regexec(view->regex, text[i], 1, &pmatch, 0) != REG_NOMATCH)
3008 return TRUE;
3009 return FALSE;
3010 }
3012 static void
3013 select_view_line(struct view *view, unsigned long lineno)
3014 {
3015 unsigned long old_lineno = view->lineno;
3016 unsigned long old_offset = view->offset;
3018 if (goto_view_line(view, view->offset, lineno)) {
3019 if (view_is_displayed(view)) {
3020 if (old_offset != view->offset) {
3021 redraw_view(view);
3022 } else {
3023 draw_view_line(view, old_lineno - view->offset);
3024 draw_view_line(view, view->lineno - view->offset);
3025 wnoutrefresh(view->win);
3026 }
3027 } else {
3028 view->ops->select(view, &view->line[view->lineno]);
3029 }
3030 }
3031 }
3033 static void
3034 find_next(struct view *view, enum request request)
3035 {
3036 unsigned long lineno = view->lineno;
3037 int direction;
3039 if (!*view->grep) {
3040 if (!*opt_search)
3041 report("No previous search");
3042 else
3043 search_view(view, request);
3044 return;
3045 }
3047 switch (request) {
3048 case REQ_SEARCH:
3049 case REQ_FIND_NEXT:
3050 direction = 1;
3051 break;
3053 case REQ_SEARCH_BACK:
3054 case REQ_FIND_PREV:
3055 direction = -1;
3056 break;
3058 default:
3059 return;
3060 }
3062 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
3063 lineno += direction;
3065 /* Note, lineno is unsigned long so will wrap around in which case it
3066 * will become bigger than view->lines. */
3067 for (; lineno < view->lines; lineno += direction) {
3068 if (view->ops->grep(view, &view->line[lineno])) {
3069 select_view_line(view, lineno);
3070 report("Line %ld matches '%s'", lineno + 1, view->grep);
3071 return;
3072 }
3073 }
3075 report("No match found for '%s'", view->grep);
3076 }
3078 static void
3079 search_view(struct view *view, enum request request)
3080 {
3081 int regex_err;
3083 if (view->regex) {
3084 regfree(view->regex);
3085 *view->grep = 0;
3086 } else {
3087 view->regex = calloc(1, sizeof(*view->regex));
3088 if (!view->regex)
3089 return;
3090 }
3092 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
3093 if (regex_err != 0) {
3094 char buf[SIZEOF_STR] = "unknown error";
3096 regerror(regex_err, view->regex, buf, sizeof(buf));
3097 report("Search failed: %s", buf);
3098 return;
3099 }
3101 string_copy(view->grep, opt_search);
3103 find_next(view, request);
3104 }
3106 /*
3107 * Incremental updating
3108 */
3110 static void
3111 reset_view(struct view *view)
3112 {
3113 int i;
3115 for (i = 0; i < view->lines; i++)
3116 free(view->line[i].data);
3117 free(view->line);
3119 view->p_offset = view->offset;
3120 view->p_yoffset = view->yoffset;
3121 view->p_lineno = view->lineno;
3123 view->line = NULL;
3124 view->offset = 0;
3125 view->yoffset = 0;
3126 view->lines = 0;
3127 view->lineno = 0;
3128 view->vid[0] = 0;
3129 view->update_secs = 0;
3130 }
3132 static const char *
3133 format_arg(const char *name)
3134 {
3135 static struct {
3136 const char *name;
3137 size_t namelen;
3138 const char *value;
3139 const char *value_if_empty;
3140 } vars[] = {
3141 #define FORMAT_VAR(name, value, value_if_empty) \
3142 { name, STRING_SIZE(name), value, value_if_empty }
3143 FORMAT_VAR("%(directory)", opt_path, ""),
3144 FORMAT_VAR("%(file)", opt_file, ""),
3145 FORMAT_VAR("%(ref)", opt_ref, "HEAD"),
3146 FORMAT_VAR("%(head)", ref_head, ""),
3147 FORMAT_VAR("%(commit)", ref_commit, ""),
3148 FORMAT_VAR("%(blob)", ref_blob, ""),
3149 FORMAT_VAR("%(branch)", ref_branch, ""),
3150 };
3151 int i;
3153 for (i = 0; i < ARRAY_SIZE(vars); i++)
3154 if (!strncmp(name, vars[i].name, vars[i].namelen))
3155 return *vars[i].value ? vars[i].value : vars[i].value_if_empty;
3157 report("Unknown replacement: `%s`", name);
3158 return NULL;
3159 }
3161 static bool
3162 format_argv(const char ***dst_argv, const char *src_argv[], bool replace)
3163 {
3164 char buf[SIZEOF_STR];
3165 int argc;
3167 argv_free(*dst_argv);
3169 for (argc = 0; src_argv[argc]; argc++) {
3170 const char *arg = src_argv[argc];
3171 size_t bufpos = 0;
3173 while (arg) {
3174 char *next = strstr(arg, "%(");
3175 int len = next - arg;
3176 const char *value;
3178 if (!next || !replace) {
3179 len = strlen(arg);
3180 value = "";
3182 } else {
3183 value = format_arg(next);
3185 if (!value) {
3186 return FALSE;
3187 }
3188 }
3190 if (!string_format_from(buf, &bufpos, "%.*s%s", len, arg, value))
3191 return FALSE;
3193 arg = next && replace ? strchr(next, ')') + 1 : NULL;
3194 }
3196 if (!argv_append(dst_argv, buf))
3197 break;
3198 }
3200 return src_argv[argc] == NULL;
3201 }
3203 static bool
3204 restore_view_position(struct view *view)
3205 {
3206 if (!view->p_restore || (view->pipe && view->lines <= view->p_lineno))
3207 return FALSE;
3209 /* Changing the view position cancels the restoring. */
3210 /* FIXME: Changing back to the first line is not detected. */
3211 if (view->offset != 0 || view->lineno != 0) {
3212 view->p_restore = FALSE;
3213 return FALSE;
3214 }
3216 if (goto_view_line(view, view->p_offset, view->p_lineno) &&
3217 view_is_displayed(view))
3218 werase(view->win);
3220 view->yoffset = view->p_yoffset;
3221 view->p_restore = FALSE;
3223 return TRUE;
3224 }
3226 static void
3227 end_update(struct view *view, bool force)
3228 {
3229 if (!view->pipe)
3230 return;
3231 while (!view->ops->read(view, NULL))
3232 if (!force)
3233 return;
3234 if (force)
3235 io_kill(view->pipe);
3236 io_done(view->pipe);
3237 view->pipe = NULL;
3238 }
3240 static void
3241 setup_update(struct view *view, const char *vid)
3242 {
3243 reset_view(view);
3244 string_copy_rev(view->vid, vid);
3245 view->pipe = &view->io;
3246 view->start_time = time(NULL);
3247 }
3249 static bool
3250 prepare_io(struct view *view, const char *dir, const char *argv[], bool replace)
3251 {
3252 view->dir = dir;
3253 return format_argv(&view->argv, argv, replace);
3254 }
3256 static bool
3257 prepare_update(struct view *view, const char *argv[], const char *dir)
3258 {
3259 if (view->pipe)
3260 end_update(view, TRUE);
3261 return prepare_io(view, dir, argv, FALSE);
3262 }
3264 static bool
3265 start_update(struct view *view, const char **argv, const char *dir)
3266 {
3267 if (view->pipe)
3268 io_done(view->pipe);
3269 return prepare_io(view, dir, argv, FALSE) &&
3270 io_run(&view->io, IO_RD, dir, view->argv);
3271 }
3273 static bool
3274 prepare_update_file(struct view *view, const char *name)
3275 {
3276 if (view->pipe)
3277 end_update(view, TRUE);
3278 argv_free(view->argv);
3279 return io_open(&view->io, "%s/%s", opt_cdup[0] ? opt_cdup : ".", name);
3280 }
3282 static bool
3283 begin_update(struct view *view, bool refresh)
3284 {
3285 if (view->pipe)
3286 end_update(view, TRUE);
3288 if (!refresh) {
3289 if (view->ops->prepare) {
3290 if (!view->ops->prepare(view))
3291 return FALSE;
3292 } else if (!prepare_io(view, NULL, view->ops->argv, TRUE)) {
3293 return FALSE;
3294 }
3296 /* Put the current ref_* value to the view title ref
3297 * member. This is needed by the blob view. Most other
3298 * views sets it automatically after loading because the
3299 * first line is a commit line. */
3300 string_copy_rev(view->ref, view->id);
3301 }
3303 if (view->argv && view->argv[0] &&
3304 !io_run(&view->io, IO_RD, view->dir, view->argv))
3305 return FALSE;
3307 setup_update(view, view->id);
3309 return TRUE;
3310 }
3312 static bool
3313 update_view(struct view *view)
3314 {
3315 char out_buffer[BUFSIZ * 2];
3316 char *line;
3317 /* Clear the view and redraw everything since the tree sorting
3318 * might have rearranged things. */
3319 bool redraw = view->lines == 0;
3320 bool can_read = TRUE;
3322 if (!view->pipe)
3323 return TRUE;
3325 if (!io_can_read(view->pipe)) {
3326 if (view->lines == 0 && view_is_displayed(view)) {
3327 time_t secs = time(NULL) - view->start_time;
3329 if (secs > 1 && secs > view->update_secs) {
3330 if (view->update_secs == 0)
3331 redraw_view(view);
3332 update_view_title(view);
3333 view->update_secs = secs;
3334 }
3335 }
3336 return TRUE;
3337 }
3339 for (; (line = io_get(view->pipe, '\n', can_read)); can_read = FALSE) {
3340 if (opt_iconv_in != ICONV_NONE) {
3341 ICONV_CONST char *inbuf = line;
3342 size_t inlen = strlen(line) + 1;
3344 char *outbuf = out_buffer;
3345 size_t outlen = sizeof(out_buffer);
3347 size_t ret;
3349 ret = iconv(opt_iconv_in, &inbuf, &inlen, &outbuf, &outlen);
3350 if (ret != (size_t) -1)
3351 line = out_buffer;
3352 }
3354 if (!view->ops->read(view, line)) {
3355 report("Allocation failure");
3356 end_update(view, TRUE);
3357 return FALSE;
3358 }
3359 }
3361 {
3362 unsigned long lines = view->lines;
3363 int digits;
3365 for (digits = 0; lines; digits++)
3366 lines /= 10;
3368 /* Keep the displayed view in sync with line number scaling. */
3369 if (digits != view->digits) {
3370 view->digits = digits;
3371 if (opt_line_number || view->type == VIEW_BLAME)
3372 redraw = TRUE;
3373 }
3374 }
3376 if (io_error(view->pipe)) {
3377 report("Failed to read: %s", io_strerror(view->pipe));
3378 end_update(view, TRUE);
3380 } else if (io_eof(view->pipe)) {
3381 if (view_is_displayed(view))
3382 report("");
3383 end_update(view, FALSE);
3384 }
3386 if (restore_view_position(view))
3387 redraw = TRUE;
3389 if (!view_is_displayed(view))
3390 return TRUE;
3392 if (redraw)
3393 redraw_view_from(view, 0);
3394 else
3395 redraw_view_dirty(view);
3397 /* Update the title _after_ the redraw so that if the redraw picks up a
3398 * commit reference in view->ref it'll be available here. */
3399 update_view_title(view);
3400 return TRUE;
3401 }
3403 DEFINE_ALLOCATOR(realloc_lines, struct line, 256)
3405 static struct line *
3406 add_line_data(struct view *view, void *data, enum line_type type)
3407 {
3408 struct line *line;
3410 if (!realloc_lines(&view->line, view->lines, 1))
3411 return NULL;
3413 line = &view->line[view->lines++];
3414 memset(line, 0, sizeof(*line));
3415 line->type = type;
3416 line->data = data;
3417 line->dirty = 1;
3419 return line;
3420 }
3422 static struct line *
3423 add_line_text(struct view *view, const char *text, enum line_type type)
3424 {
3425 char *data = text ? strdup(text) : NULL;
3427 return data ? add_line_data(view, data, type) : NULL;
3428 }
3430 static struct line *
3431 add_line_format(struct view *view, enum line_type type, const char *fmt, ...)
3432 {
3433 char buf[SIZEOF_STR];
3434 va_list args;
3436 va_start(args, fmt);
3437 if (vsnprintf(buf, sizeof(buf), fmt, args) >= sizeof(buf))
3438 buf[0] = 0;
3439 va_end(args);
3441 return buf[0] ? add_line_text(view, buf, type) : NULL;
3442 }
3444 /*
3445 * View opening
3446 */
3448 enum open_flags {
3449 OPEN_DEFAULT = 0, /* Use default view switching. */
3450 OPEN_SPLIT = 1, /* Split current view. */
3451 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
3452 OPEN_REFRESH = 16, /* Refresh view using previous command. */
3453 OPEN_PREPARED = 32, /* Open already prepared command. */
3454 };
3456 static void
3457 open_view(struct view *prev, enum request request, enum open_flags flags)
3458 {
3459 bool split = !!(flags & OPEN_SPLIT);
3460 bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH | OPEN_PREPARED));
3461 bool nomaximize = !!(flags & OPEN_REFRESH);
3462 struct view *view = VIEW(request);
3463 int nviews = displayed_views();
3464 struct view *base_view = display[0];
3466 if (view == prev && nviews == 1 && !reload) {
3467 report("Already in %s view", view->name);
3468 return;
3469 }
3471 if (view->git_dir && !opt_git_dir[0]) {
3472 report("The %s view is disabled in pager view", view->name);
3473 return;
3474 }
3476 if (split) {
3477 display[1] = view;
3478 current_view = 1;
3479 view->parent = prev;
3480 } else if (!nomaximize) {
3481 /* Maximize the current view. */
3482 memset(display, 0, sizeof(display));
3483 current_view = 0;
3484 display[current_view] = view;
3485 }
3487 /* No prev signals that this is the first loaded view. */
3488 if (prev && view != prev) {
3489 view->prev = prev;
3490 }
3492 /* Resize the view when switching between split- and full-screen,
3493 * or when switching between two different full-screen views. */
3494 if (nviews != displayed_views() ||
3495 (nviews == 1 && base_view != display[0]))
3496 resize_display();
3498 if (view->ops->open) {
3499 if (view->pipe)
3500 end_update(view, TRUE);
3501 if (!view->ops->open(view)) {
3502 report("Failed to load %s view", view->name);
3503 return;
3504 }
3505 restore_view_position(view);
3507 } else if ((reload || strcmp(view->vid, view->id)) &&
3508 !begin_update(view, flags & (OPEN_REFRESH | OPEN_PREPARED))) {
3509 report("Failed to load %s view", view->name);
3510 return;
3511 }
3513 if (split && prev->lineno - prev->offset >= prev->height) {
3514 /* Take the title line into account. */
3515 int lines = prev->lineno - prev->offset - prev->height + 1;
3517 /* Scroll the view that was split if the current line is
3518 * outside the new limited view. */
3519 do_scroll_view(prev, lines);
3520 }
3522 if (prev && view != prev && split && view_is_displayed(prev)) {
3523 /* "Blur" the previous view. */
3524 update_view_title(prev);
3525 }
3527 if (view->pipe && view->lines == 0) {
3528 /* Clear the old view and let the incremental updating refill
3529 * the screen. */
3530 werase(view->win);
3531 view->p_restore = flags & (OPEN_RELOAD | OPEN_REFRESH);
3532 report("");
3533 } else if (view_is_displayed(view)) {
3534 redraw_view(view);
3535 report("");
3536 }
3537 }
3539 static void
3540 open_external_viewer(const char *argv[], const char *dir)
3541 {
3542 def_prog_mode(); /* save current tty modes */
3543 endwin(); /* restore original tty modes */
3544 io_run_fg(argv, dir);
3545 fprintf(stderr, "Press Enter to continue");
3546 getc(opt_tty);
3547 reset_prog_mode();
3548 redraw_display(TRUE);
3549 }
3551 static void
3552 open_mergetool(const char *file)
3553 {
3554 const char *mergetool_argv[] = { "git", "mergetool", file, NULL };
3556 open_external_viewer(mergetool_argv, opt_cdup);
3557 }
3559 static void
3560 open_editor(const char *file)
3561 {
3562 const char *editor_argv[] = { "vi", file, NULL };
3563 const char *editor;
3565 editor = getenv("GIT_EDITOR");
3566 if (!editor && *opt_editor)
3567 editor = opt_editor;
3568 if (!editor)
3569 editor = getenv("VISUAL");
3570 if (!editor)
3571 editor = getenv("EDITOR");
3572 if (!editor)
3573 editor = "vi";
3575 editor_argv[0] = editor;
3576 open_external_viewer(editor_argv, opt_cdup);
3577 }
3579 static void
3580 open_run_request(enum request request)
3581 {
3582 struct run_request *req = get_run_request(request);
3583 const char **argv = NULL;
3585 if (!req) {
3586 report("Unknown run request");
3587 return;
3588 }
3590 if (format_argv(&argv, req->argv, TRUE))
3591 open_external_viewer(argv, NULL);
3592 if (argv)
3593 argv_free(argv);
3594 free(argv);
3595 }
3597 /*
3598 * User request switch noodle
3599 */
3601 static int
3602 view_driver(struct view *view, enum request request)
3603 {
3604 int i;
3606 if (request == REQ_NONE)
3607 return TRUE;
3609 if (request > REQ_NONE) {
3610 open_run_request(request);
3611 view_request(view, REQ_REFRESH);
3612 return TRUE;
3613 }
3615 request = view_request(view, request);
3616 if (request == REQ_NONE)
3617 return TRUE;
3619 switch (request) {
3620 case REQ_MOVE_UP:
3621 case REQ_MOVE_DOWN:
3622 case REQ_MOVE_PAGE_UP:
3623 case REQ_MOVE_PAGE_DOWN:
3624 case REQ_MOVE_FIRST_LINE:
3625 case REQ_MOVE_LAST_LINE:
3626 move_view(view, request);
3627 break;
3629 case REQ_SCROLL_LEFT:
3630 case REQ_SCROLL_RIGHT:
3631 case REQ_SCROLL_LINE_DOWN:
3632 case REQ_SCROLL_LINE_UP:
3633 case REQ_SCROLL_PAGE_DOWN:
3634 case REQ_SCROLL_PAGE_UP:
3635 scroll_view(view, request);
3636 break;
3638 case REQ_VIEW_BLAME:
3639 if (!opt_file[0]) {
3640 report("No file chosen, press %s to open tree view",
3641 get_key(view->keymap, REQ_VIEW_TREE));
3642 break;
3643 }
3644 open_view(view, request, OPEN_DEFAULT);
3645 break;
3647 case REQ_VIEW_BLOB:
3648 if (!ref_blob[0]) {
3649 report("No file chosen, press %s to open tree view",
3650 get_key(view->keymap, REQ_VIEW_TREE));
3651 break;
3652 }
3653 open_view(view, request, OPEN_DEFAULT);
3654 break;
3656 case REQ_VIEW_PAGER:
3657 if (!VIEW(REQ_VIEW_PAGER)->pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
3658 report("No pager content, press %s to run command from prompt",
3659 get_key(view->keymap, REQ_PROMPT));
3660 break;
3661 }
3662 open_view(view, request, OPEN_DEFAULT);
3663 break;
3665 case REQ_VIEW_STAGE:
3666 if (!VIEW(REQ_VIEW_STAGE)->lines) {
3667 report("No stage content, press %s to open the status view and choose file",
3668 get_key(view->keymap, REQ_VIEW_STATUS));
3669 break;
3670 }
3671 open_view(view, request, OPEN_DEFAULT);
3672 break;
3674 case REQ_VIEW_STATUS:
3675 if (opt_is_inside_work_tree == FALSE) {
3676 report("The status view requires a working tree");
3677 break;
3678 }
3679 open_view(view, request, OPEN_DEFAULT);
3680 break;
3682 case REQ_VIEW_MAIN:
3683 case REQ_VIEW_DIFF:
3684 case REQ_VIEW_LOG:
3685 case REQ_VIEW_TREE:
3686 case REQ_VIEW_HELP:
3687 case REQ_VIEW_BRANCH:
3688 open_view(view, request, OPEN_DEFAULT);
3689 break;
3691 case REQ_NEXT:
3692 case REQ_PREVIOUS:
3693 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
3695 if (view->parent) {
3696 int line;
3698 view = view->parent;
3699 line = view->lineno;
3700 move_view(view, request);
3701 if (view_is_displayed(view))
3702 update_view_title(view);
3703 if (line != view->lineno)
3704 view_request(view, REQ_ENTER);
3705 } else {
3706 move_view(view, request);
3707 }
3708 break;
3710 case REQ_VIEW_NEXT:
3711 {
3712 int nviews = displayed_views();
3713 int next_view = (current_view + 1) % nviews;
3715 if (next_view == current_view) {
3716 report("Only one view is displayed");
3717 break;
3718 }
3720 current_view = next_view;
3721 /* Blur out the title of the previous view. */
3722 update_view_title(view);
3723 report("");
3724 break;
3725 }
3726 case REQ_REFRESH:
3727 report("Refreshing is not yet supported for the %s view", view->name);
3728 break;
3730 case REQ_MAXIMIZE:
3731 if (displayed_views() == 2)
3732 maximize_view(view);
3733 break;
3735 case REQ_OPTIONS:
3736 open_option_menu();
3737 break;
3739 case REQ_TOGGLE_LINENO:
3740 toggle_view_option(&opt_line_number, "line numbers");
3741 break;
3743 case REQ_TOGGLE_DATE:
3744 toggle_date();
3745 break;
3747 case REQ_TOGGLE_AUTHOR:
3748 toggle_author();
3749 break;
3751 case REQ_TOGGLE_REV_GRAPH:
3752 toggle_view_option(&opt_rev_graph, "revision graph display");
3753 break;
3755 case REQ_TOGGLE_REFS:
3756 toggle_view_option(&opt_show_refs, "reference display");
3757 break;
3759 case REQ_TOGGLE_SORT_FIELD:
3760 case REQ_TOGGLE_SORT_ORDER:
3761 report("Sorting is not yet supported for the %s view", view->name);
3762 break;
3764 case REQ_SEARCH:
3765 case REQ_SEARCH_BACK:
3766 search_view(view, request);
3767 break;
3769 case REQ_FIND_NEXT:
3770 case REQ_FIND_PREV:
3771 find_next(view, request);
3772 break;
3774 case REQ_STOP_LOADING:
3775 foreach_view(view, i) {
3776 if (view->pipe)
3777 report("Stopped loading the %s view", view->name),
3778 end_update(view, TRUE);
3779 }
3780 break;
3782 case REQ_SHOW_VERSION:
3783 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
3784 return TRUE;
3786 case REQ_SCREEN_REDRAW:
3787 redraw_display(TRUE);
3788 break;
3790 case REQ_EDIT:
3791 report("Nothing to edit");
3792 break;
3794 case REQ_ENTER:
3795 report("Nothing to enter");
3796 break;
3798 case REQ_VIEW_CLOSE:
3799 /* XXX: Mark closed views by letting view->prev point to the
3800 * view itself. Parents to closed view should never be
3801 * followed. */
3802 if (view->prev && view->prev != view) {
3803 maximize_view(view->prev);
3804 view->prev = view;
3805 break;
3806 }
3807 /* Fall-through */
3808 case REQ_QUIT:
3809 return FALSE;
3811 default:
3812 report("Unknown key, press %s for help",
3813 get_key(view->keymap, REQ_VIEW_HELP));
3814 return TRUE;
3815 }
3817 return TRUE;
3818 }
3821 /*
3822 * View backend utilities
3823 */
3825 enum sort_field {
3826 ORDERBY_NAME,
3827 ORDERBY_DATE,
3828 ORDERBY_AUTHOR,
3829 };
3831 struct sort_state {
3832 const enum sort_field *fields;
3833 size_t size, current;
3834 bool reverse;
3835 };
3837 #define SORT_STATE(fields) { fields, ARRAY_SIZE(fields), 0 }
3838 #define get_sort_field(state) ((state).fields[(state).current])
3839 #define sort_order(state, result) ((state).reverse ? -(result) : (result))
3841 static void
3842 sort_view(struct view *view, enum request request, struct sort_state *state,
3843 int (*compare)(const void *, const void *))
3844 {
3845 switch (request) {
3846 case REQ_TOGGLE_SORT_FIELD:
3847 state->current = (state->current + 1) % state->size;
3848 break;
3850 case REQ_TOGGLE_SORT_ORDER:
3851 state->reverse = !state->reverse;
3852 break;
3853 default:
3854 die("Not a sort request");
3855 }
3857 qsort(view->line, view->lines, sizeof(*view->line), compare);
3858 redraw_view(view);
3859 }
3861 DEFINE_ALLOCATOR(realloc_authors, const char *, 256)
3863 /* Small author cache to reduce memory consumption. It uses binary
3864 * search to lookup or find place to position new entries. No entries
3865 * are ever freed. */
3866 static const char *
3867 get_author(const char *name)
3868 {
3869 static const char **authors;
3870 static size_t authors_size;
3871 int from = 0, to = authors_size - 1;
3873 while (from <= to) {
3874 size_t pos = (to + from) / 2;
3875 int cmp = strcmp(name, authors[pos]);
3877 if (!cmp)
3878 return authors[pos];
3880 if (cmp < 0)
3881 to = pos - 1;
3882 else
3883 from = pos + 1;
3884 }
3886 if (!realloc_authors(&authors, authors_size, 1))
3887 return NULL;
3888 name = strdup(name);
3889 if (!name)
3890 return NULL;
3892 memmove(authors + from + 1, authors + from, (authors_size - from) * sizeof(*authors));
3893 authors[from] = name;
3894 authors_size++;
3896 return name;
3897 }
3899 static void
3900 parse_timesec(struct time *time, const char *sec)
3901 {
3902 time->sec = (time_t) atol(sec);
3903 }
3905 static void
3906 parse_timezone(struct time *time, const char *zone)
3907 {
3908 long tz;
3910 tz = ('0' - zone[1]) * 60 * 60 * 10;
3911 tz += ('0' - zone[2]) * 60 * 60;
3912 tz += ('0' - zone[3]) * 60 * 10;
3913 tz += ('0' - zone[4]) * 60;
3915 if (zone[0] == '-')
3916 tz = -tz;
3918 time->tz = tz;
3919 time->sec -= tz;
3920 }
3922 /* Parse author lines where the name may be empty:
3923 * author <email@address.tld> 1138474660 +0100
3924 */
3925 static void
3926 parse_author_line(char *ident, const char **author, struct time *time)
3927 {
3928 char *nameend = strchr(ident, '<');
3929 char *emailend = strchr(ident, '>');
3931 if (nameend && emailend)
3932 *nameend = *emailend = 0;
3933 ident = chomp_string(ident);
3934 if (!*ident) {
3935 if (nameend)
3936 ident = chomp_string(nameend + 1);
3937 if (!*ident)
3938 ident = "Unknown";
3939 }
3941 *author = get_author(ident);
3943 /* Parse epoch and timezone */
3944 if (emailend && emailend[1] == ' ') {
3945 char *secs = emailend + 2;
3946 char *zone = strchr(secs, ' ');
3948 parse_timesec(time, secs);
3950 if (zone && strlen(zone) == STRING_SIZE(" +0700"))
3951 parse_timezone(time, zone + 1);
3952 }
3953 }
3955 static bool
3956 open_commit_parent_menu(char buf[SIZEOF_STR], int *parents)
3957 {
3958 char rev[SIZEOF_REV];
3959 const char *revlist_argv[] = {
3960 "git", "log", "--no-color", "-1", "--pretty=format:%s", rev, NULL
3961 };
3962 struct menu_item *items;
3963 char text[SIZEOF_STR];
3964 bool ok = TRUE;
3965 int i;
3967 items = calloc(*parents + 1, sizeof(*items));
3968 if (!items)
3969 return FALSE;
3971 for (i = 0; i < *parents; i++) {
3972 string_copy_rev(rev, &buf[SIZEOF_REV * i]);
3973 if (!io_run_buf(revlist_argv, text, sizeof(text)) ||
3974 !(items[i].text = strdup(text))) {
3975 ok = FALSE;
3976 break;
3977 }
3978 }
3980 if (ok) {
3981 *parents = 0;
3982 ok = prompt_menu("Select parent", items, parents);
3983 }
3984 for (i = 0; items[i].text; i++)
3985 free((char *) items[i].text);
3986 free(items);
3987 return ok;
3988 }
3990 static bool
3991 select_commit_parent(const char *id, char rev[SIZEOF_REV], const char *path)
3992 {
3993 char buf[SIZEOF_STR * 4];
3994 const char *revlist_argv[] = {
3995 "git", "log", "--no-color", "-1",
3996 "--pretty=format:%P", id, "--", path, NULL
3997 };
3998 int parents;
4000 if (!io_run_buf(revlist_argv, buf, sizeof(buf)) ||
4001 (parents = strlen(buf) / 40) < 0) {
4002 report("Failed to get parent information");
4003 return FALSE;
4005 } else if (parents == 0) {
4006 if (path)
4007 report("Path '%s' does not exist in the parent", path);
4008 else
4009 report("The selected commit has no parents");
4010 return FALSE;
4011 }
4013 if (parents == 1)
4014 parents = 0;
4015 else if (!open_commit_parent_menu(buf, &parents))
4016 return FALSE;
4018 string_copy_rev(rev, &buf[41 * parents]);
4019 return TRUE;
4020 }
4022 /*
4023 * Pager backend
4024 */
4026 static bool
4027 pager_draw(struct view *view, struct line *line, unsigned int lineno)
4028 {
4029 char text[SIZEOF_STR];
4031 if (opt_line_number && draw_lineno(view, lineno))
4032 return TRUE;
4034 string_expand(text, sizeof(text), line->data, opt_tab_size);
4035 draw_text(view, line->type, text, TRUE);
4036 return TRUE;
4037 }
4039 static bool
4040 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
4041 {
4042 const char *describe_argv[] = { "git", "describe", commit_id, NULL };
4043 char ref[SIZEOF_STR];
4045 if (!io_run_buf(describe_argv, ref, sizeof(ref)) || !*ref)
4046 return TRUE;
4048 /* This is the only fatal call, since it can "corrupt" the buffer. */
4049 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
4050 return FALSE;
4052 return TRUE;
4053 }
4055 static void
4056 add_pager_refs(struct view *view, struct line *line)
4057 {
4058 char buf[SIZEOF_STR];
4059 char *commit_id = (char *)line->data + STRING_SIZE("commit ");
4060 struct ref_list *list;
4061 size_t bufpos = 0, i;
4062 const char *sep = "Refs: ";
4063 bool is_tag = FALSE;
4065 assert(line->type == LINE_COMMIT);
4067 list = get_ref_list(commit_id);
4068 if (!list) {
4069 if (view->type == VIEW_DIFF)
4070 goto try_add_describe_ref;
4071 return;
4072 }
4074 for (i = 0; i < list->size; i++) {
4075 struct ref *ref = list->refs[i];
4076 const char *fmt = ref->tag ? "%s[%s]" :
4077 ref->remote ? "%s<%s>" : "%s%s";
4079 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
4080 return;
4081 sep = ", ";
4082 if (ref->tag)
4083 is_tag = TRUE;
4084 }
4086 if (!is_tag && view->type == VIEW_DIFF) {
4087 try_add_describe_ref:
4088 /* Add <tag>-g<commit_id> "fake" reference. */
4089 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
4090 return;
4091 }
4093 if (bufpos == 0)
4094 return;
4096 add_line_text(view, buf, LINE_PP_REFS);
4097 }
4099 static bool
4100 pager_read(struct view *view, char *data)
4101 {
4102 struct line *line;
4104 if (!data)
4105 return TRUE;
4107 line = add_line_text(view, data, get_line_type(data));
4108 if (!line)
4109 return FALSE;
4111 if (line->type == LINE_COMMIT &&
4112 (view->type == VIEW_DIFF ||
4113 view->type == VIEW_LOG))
4114 add_pager_refs(view, line);
4116 return TRUE;
4117 }
4119 static enum request
4120 pager_request(struct view *view, enum request request, struct line *line)
4121 {
4122 int split = 0;
4124 if (request != REQ_ENTER)
4125 return request;
4127 if (line->type == LINE_COMMIT &&
4128 (view->type == VIEW_LOG ||
4129 view->type == VIEW_PAGER)) {
4130 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
4131 split = 1;
4132 }
4134 /* Always scroll the view even if it was split. That way
4135 * you can use Enter to scroll through the log view and
4136 * split open each commit diff. */
4137 scroll_view(view, REQ_SCROLL_LINE_DOWN);
4139 /* FIXME: A minor workaround. Scrolling the view will call report("")
4140 * but if we are scrolling a non-current view this won't properly
4141 * update the view title. */
4142 if (split)
4143 update_view_title(view);
4145 return REQ_NONE;
4146 }
4148 static bool
4149 pager_grep(struct view *view, struct line *line)
4150 {
4151 const char *text[] = { line->data, NULL };
4153 return grep_text(view, text);
4154 }
4156 static void
4157 pager_select(struct view *view, struct line *line)
4158 {
4159 if (line->type == LINE_COMMIT) {
4160 char *text = (char *)line->data + STRING_SIZE("commit ");
4162 if (view->type != VIEW_PAGER)
4163 string_copy_rev(view->ref, text);
4164 string_copy_rev(ref_commit, text);
4165 }
4166 }
4168 static struct view_ops pager_ops = {
4169 "line",
4170 NULL,
4171 NULL,
4172 pager_read,
4173 pager_draw,
4174 pager_request,
4175 pager_grep,
4176 pager_select,
4177 };
4179 static const char *log_argv[SIZEOF_ARG] = {
4180 "git", "log", "--no-color", "--cc", "--stat", "-n100", "%(head)", NULL
4181 };
4183 static enum request
4184 log_request(struct view *view, enum request request, struct line *line)
4185 {
4186 switch (request) {
4187 case REQ_REFRESH:
4188 load_refs();
4189 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
4190 return REQ_NONE;
4191 default:
4192 return pager_request(view, request, line);
4193 }
4194 }
4196 static struct view_ops log_ops = {
4197 "line",
4198 log_argv,
4199 NULL,
4200 pager_read,
4201 pager_draw,
4202 log_request,
4203 pager_grep,
4204 pager_select,
4205 };
4207 static const char *diff_argv[SIZEOF_ARG] = {
4208 "git", "show", "--pretty=fuller", "--no-color", "--root",
4209 "--patch-with-stat", "--find-copies-harder", "-C", "%(commit)", NULL
4210 };
4212 static struct view_ops diff_ops = {
4213 "line",
4214 diff_argv,
4215 NULL,
4216 pager_read,
4217 pager_draw,
4218 pager_request,
4219 pager_grep,
4220 pager_select,
4221 };
4223 /*
4224 * Help backend
4225 */
4227 static bool help_keymap_hidden[ARRAY_SIZE(keymap_table)];
4229 static bool
4230 help_open_keymap_title(struct view *view, enum keymap keymap)
4231 {
4232 struct line *line;
4234 line = add_line_format(view, LINE_HELP_KEYMAP, "[%c] %s bindings",
4235 help_keymap_hidden[keymap] ? '+' : '-',
4236 enum_name(keymap_table[keymap]));
4237 if (line)
4238 line->other = keymap;
4240 return help_keymap_hidden[keymap];
4241 }
4243 static void
4244 help_open_keymap(struct view *view, enum keymap keymap)
4245 {
4246 const char *group = NULL;
4247 char buf[SIZEOF_STR];
4248 size_t bufpos;
4249 bool add_title = TRUE;
4250 int i;
4252 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
4253 const char *key = NULL;
4255 if (req_info[i].request == REQ_NONE)
4256 continue;
4258 if (!req_info[i].request) {
4259 group = req_info[i].help;
4260 continue;
4261 }
4263 key = get_keys(keymap, req_info[i].request, TRUE);
4264 if (!key || !*key)
4265 continue;
4267 if (add_title && help_open_keymap_title(view, keymap))
4268 return;
4269 add_title = FALSE;
4271 if (group) {
4272 add_line_text(view, group, LINE_HELP_GROUP);
4273 group = NULL;
4274 }
4276 add_line_format(view, LINE_DEFAULT, " %-25s %-20s %s", key,
4277 enum_name(req_info[i]), req_info[i].help);
4278 }
4280 group = "External commands:";
4282 for (i = 0; i < run_requests; i++) {
4283 struct run_request *req = get_run_request(REQ_NONE + i + 1);
4284 const char *key;
4285 int argc;
4287 if (!req || req->keymap != keymap)
4288 continue;
4290 key = get_key_name(req->key);
4291 if (!*key)
4292 key = "(no key defined)";
4294 if (add_title && help_open_keymap_title(view, keymap))
4295 return;
4296 if (group) {
4297 add_line_text(view, group, LINE_HELP_GROUP);
4298 group = NULL;
4299 }
4301 for (bufpos = 0, argc = 0; req->argv[argc]; argc++)
4302 if (!string_format_from(buf, &bufpos, "%s%s",
4303 argc ? " " : "", req->argv[argc]))
4304 return;
4306 add_line_format(view, LINE_DEFAULT, " %-25s `%s`", key, buf);
4307 }
4308 }
4310 static bool
4311 help_open(struct view *view)
4312 {
4313 enum keymap keymap;
4315 reset_view(view);
4316 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
4317 add_line_text(view, "", LINE_DEFAULT);
4319 for (keymap = 0; keymap < ARRAY_SIZE(keymap_table); keymap++)
4320 help_open_keymap(view, keymap);
4322 return TRUE;
4323 }
4325 static enum request
4326 help_request(struct view *view, enum request request, struct line *line)
4327 {
4328 switch (request) {
4329 case REQ_ENTER:
4330 if (line->type == LINE_HELP_KEYMAP) {
4331 help_keymap_hidden[line->other] =
4332 !help_keymap_hidden[line->other];
4333 view->p_restore = TRUE;
4334 open_view(view, REQ_VIEW_HELP, OPEN_REFRESH);
4335 }
4337 return REQ_NONE;
4338 default:
4339 return pager_request(view, request, line);
4340 }
4341 }
4343 static struct view_ops help_ops = {
4344 "line",
4345 NULL,
4346 help_open,
4347 NULL,
4348 pager_draw,
4349 help_request,
4350 pager_grep,
4351 pager_select,
4352 };
4355 /*
4356 * Tree backend
4357 */
4359 struct tree_stack_entry {
4360 struct tree_stack_entry *prev; /* Entry below this in the stack */
4361 unsigned long lineno; /* Line number to restore */
4362 char *name; /* Position of name in opt_path */
4363 };
4365 /* The top of the path stack. */
4366 static struct tree_stack_entry *tree_stack = NULL;
4367 unsigned long tree_lineno = 0;
4369 static void
4370 pop_tree_stack_entry(void)
4371 {
4372 struct tree_stack_entry *entry = tree_stack;
4374 tree_lineno = entry->lineno;
4375 entry->name[0] = 0;
4376 tree_stack = entry->prev;
4377 free(entry);
4378 }
4380 static void
4381 push_tree_stack_entry(const char *name, unsigned long lineno)
4382 {
4383 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
4384 size_t pathlen = strlen(opt_path);
4386 if (!entry)
4387 return;
4389 entry->prev = tree_stack;
4390 entry->name = opt_path + pathlen;
4391 tree_stack = entry;
4393 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
4394 pop_tree_stack_entry();
4395 return;
4396 }
4398 /* Move the current line to the first tree entry. */
4399 tree_lineno = 1;
4400 entry->lineno = lineno;
4401 }
4403 /* Parse output from git-ls-tree(1):
4404 *
4405 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
4406 */
4408 #define SIZEOF_TREE_ATTR \
4409 STRING_SIZE("100644 blob f931e1d229c3e185caad4449bf5b66ed72462657\t")
4411 #define SIZEOF_TREE_MODE \
4412 STRING_SIZE("100644 ")
4414 #define TREE_ID_OFFSET \
4415 STRING_SIZE("100644 blob ")
4417 struct tree_entry {
4418 char id[SIZEOF_REV];
4419 mode_t mode;
4420 struct time time; /* Date from the author ident. */
4421 const char *author; /* Author of the commit. */
4422 char name[1];
4423 };
4425 static const char *
4426 tree_path(const struct line *line)
4427 {
4428 return ((struct tree_entry *) line->data)->name;
4429 }
4431 static int
4432 tree_compare_entry(const struct line *line1, const struct line *line2)
4433 {
4434 if (line1->type != line2->type)
4435 return line1->type == LINE_TREE_DIR ? -1 : 1;
4436 return strcmp(tree_path(line1), tree_path(line2));
4437 }
4439 static const enum sort_field tree_sort_fields[] = {
4440 ORDERBY_NAME, ORDERBY_DATE, ORDERBY_AUTHOR
4441 };
4442 static struct sort_state tree_sort_state = SORT_STATE(tree_sort_fields);
4444 static int
4445 tree_compare(const void *l1, const void *l2)
4446 {
4447 const struct line *line1 = (const struct line *) l1;
4448 const struct line *line2 = (const struct line *) l2;
4449 const struct tree_entry *entry1 = ((const struct line *) l1)->data;
4450 const struct tree_entry *entry2 = ((const struct line *) l2)->data;
4452 if (line1->type == LINE_TREE_HEAD)
4453 return -1;
4454 if (line2->type == LINE_TREE_HEAD)
4455 return 1;
4457 switch (get_sort_field(tree_sort_state)) {
4458 case ORDERBY_DATE:
4459 return sort_order(tree_sort_state, timecmp(&entry1->time, &entry2->time));
4461 case ORDERBY_AUTHOR:
4462 return sort_order(tree_sort_state, strcmp(entry1->author, entry2->author));
4464 case ORDERBY_NAME:
4465 default:
4466 return sort_order(tree_sort_state, tree_compare_entry(line1, line2));
4467 }
4468 }
4471 static struct line *
4472 tree_entry(struct view *view, enum line_type type, const char *path,
4473 const char *mode, const char *id)
4474 {
4475 struct tree_entry *entry = calloc(1, sizeof(*entry) + strlen(path));
4476 struct line *line = entry ? add_line_data(view, entry, type) : NULL;
4478 if (!entry || !line) {
4479 free(entry);
4480 return NULL;
4481 }
4483 strncpy(entry->name, path, strlen(path));
4484 if (mode)
4485 entry->mode = strtoul(mode, NULL, 8);
4486 if (id)
4487 string_copy_rev(entry->id, id);
4489 return line;
4490 }
4492 static bool
4493 tree_read_date(struct view *view, char *text, bool *read_date)
4494 {
4495 static const char *author_name;
4496 static struct time author_time;
4498 if (!text && *read_date) {
4499 *read_date = FALSE;
4500 return TRUE;
4502 } else if (!text) {
4503 char *path = *opt_path ? opt_path : ".";
4504 /* Find next entry to process */
4505 const char *log_file[] = {
4506 "git", "log", "--no-color", "--pretty=raw",
4507 "--cc", "--raw", view->id, "--", path, NULL
4508 };
4510 if (!view->lines) {
4511 tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL);
4512 report("Tree is empty");
4513 return TRUE;
4514 }
4516 if (!start_update(view, log_file, opt_cdup)) {
4517 report("Failed to load tree data");
4518 return TRUE;
4519 }
4521 *read_date = TRUE;
4522 return FALSE;
4524 } else if (*text == 'a' && get_line_type(text) == LINE_AUTHOR) {
4525 parse_author_line(text + STRING_SIZE("author "),
4526 &author_name, &author_time);
4528 } else if (*text == ':') {
4529 char *pos;
4530 size_t annotated = 1;
4531 size_t i;
4533 pos = strchr(text, '\t');
4534 if (!pos)
4535 return TRUE;
4536 text = pos + 1;
4537 if (*opt_path && !strncmp(text, opt_path, strlen(opt_path)))
4538 text += strlen(opt_path);
4539 pos = strchr(text, '/');
4540 if (pos)
4541 *pos = 0;
4543 for (i = 1; i < view->lines; i++) {
4544 struct line *line = &view->line[i];
4545 struct tree_entry *entry = line->data;
4547 annotated += !!entry->author;
4548 if (entry->author || strcmp(entry->name, text))
4549 continue;
4551 entry->author = author_name;
4552 entry->time = author_time;
4553 line->dirty = 1;
4554 break;
4555 }
4557 if (annotated == view->lines)
4558 io_kill(view->pipe);
4559 }
4560 return TRUE;
4561 }
4563 static bool
4564 tree_read(struct view *view, char *text)
4565 {
4566 static bool read_date = FALSE;
4567 struct tree_entry *data;
4568 struct line *entry, *line;
4569 enum line_type type;
4570 size_t textlen = text ? strlen(text) : 0;
4571 char *path = text + SIZEOF_TREE_ATTR;
4573 if (read_date || !text)
4574 return tree_read_date(view, text, &read_date);
4576 if (textlen <= SIZEOF_TREE_ATTR)
4577 return FALSE;
4578 if (view->lines == 0 &&
4579 !tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL))
4580 return FALSE;
4582 /* Strip the path part ... */
4583 if (*opt_path) {
4584 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
4585 size_t striplen = strlen(opt_path);
4587 if (pathlen > striplen)
4588 memmove(path, path + striplen,
4589 pathlen - striplen + 1);
4591 /* Insert "link" to parent directory. */
4592 if (view->lines == 1 &&
4593 !tree_entry(view, LINE_TREE_DIR, "..", "040000", view->ref))
4594 return FALSE;
4595 }
4597 type = text[SIZEOF_TREE_MODE] == 't' ? LINE_TREE_DIR : LINE_TREE_FILE;
4598 entry = tree_entry(view, type, path, text, text + TREE_ID_OFFSET);
4599 if (!entry)
4600 return FALSE;
4601 data = entry->data;
4603 /* Skip "Directory ..." and ".." line. */
4604 for (line = &view->line[1 + !!*opt_path]; line < entry; line++) {
4605 if (tree_compare_entry(line, entry) <= 0)
4606 continue;
4608 memmove(line + 1, line, (entry - line) * sizeof(*entry));
4610 line->data = data;
4611 line->type = type;
4612 for (; line <= entry; line++)
4613 line->dirty = line->cleareol = 1;
4614 return TRUE;
4615 }
4617 if (tree_lineno > view->lineno) {
4618 view->lineno = tree_lineno;
4619 tree_lineno = 0;
4620 }
4622 return TRUE;
4623 }
4625 static bool
4626 tree_draw(struct view *view, struct line *line, unsigned int lineno)
4627 {
4628 struct tree_entry *entry = line->data;
4630 if (line->type == LINE_TREE_HEAD) {
4631 if (draw_text(view, line->type, "Directory path /", TRUE))
4632 return TRUE;
4633 } else {
4634 if (draw_mode(view, entry->mode))
4635 return TRUE;
4637 if (opt_author && draw_author(view, entry->author))
4638 return TRUE;
4640 if (opt_date && draw_date(view, &entry->time))
4641 return TRUE;
4642 }
4643 if (draw_text(view, line->type, entry->name, TRUE))
4644 return TRUE;
4645 return TRUE;
4646 }
4648 static void
4649 open_blob_editor(const char *id)
4650 {
4651 const char *blob_argv[] = { "git", "cat-file", "blob", id, NULL };
4652 char file[SIZEOF_STR] = "/tmp/tigblob.XXXXXX";
4653 int fd = mkstemp(file);
4655 if (fd == -1)
4656 report("Failed to create temporary file");
4657 else if (!io_run_append(blob_argv, fd))
4658 report("Failed to save blob data to file");
4659 else
4660 open_editor(file);
4661 if (fd != -1)
4662 unlink(file);
4663 }
4665 static enum request
4666 tree_request(struct view *view, enum request request, struct line *line)
4667 {
4668 enum open_flags flags;
4669 struct tree_entry *entry = line->data;
4671 switch (request) {
4672 case REQ_VIEW_BLAME:
4673 if (line->type != LINE_TREE_FILE) {
4674 report("Blame only supported for files");
4675 return REQ_NONE;
4676 }
4678 string_copy(opt_ref, view->vid);
4679 return request;
4681 case REQ_EDIT:
4682 if (line->type != LINE_TREE_FILE) {
4683 report("Edit only supported for files");
4684 } else if (!is_head_commit(view->vid)) {
4685 open_blob_editor(entry->id);
4686 } else {
4687 open_editor(opt_file);
4688 }
4689 return REQ_NONE;
4691 case REQ_TOGGLE_SORT_FIELD:
4692 case REQ_TOGGLE_SORT_ORDER:
4693 sort_view(view, request, &tree_sort_state, tree_compare);
4694 return REQ_NONE;
4696 case REQ_PARENT:
4697 if (!*opt_path) {
4698 /* quit view if at top of tree */
4699 return REQ_VIEW_CLOSE;
4700 }
4701 /* fake 'cd ..' */
4702 line = &view->line[1];
4703 break;
4705 case REQ_ENTER:
4706 break;
4708 default:
4709 return request;
4710 }
4712 /* Cleanup the stack if the tree view is at a different tree. */
4713 while (!*opt_path && tree_stack)
4714 pop_tree_stack_entry();
4716 switch (line->type) {
4717 case LINE_TREE_DIR:
4718 /* Depending on whether it is a subdirectory or parent link
4719 * mangle the path buffer. */
4720 if (line == &view->line[1] && *opt_path) {
4721 pop_tree_stack_entry();
4723 } else {
4724 const char *basename = tree_path(line);
4726 push_tree_stack_entry(basename, view->lineno);
4727 }
4729 /* Trees and subtrees share the same ID, so they are not not
4730 * unique like blobs. */
4731 flags = OPEN_RELOAD;
4732 request = REQ_VIEW_TREE;
4733 break;
4735 case LINE_TREE_FILE:
4736 flags = view_is_displayed(view) ? OPEN_SPLIT : OPEN_DEFAULT;
4737 request = REQ_VIEW_BLOB;
4738 break;
4740 default:
4741 return REQ_NONE;
4742 }
4744 open_view(view, request, flags);
4745 if (request == REQ_VIEW_TREE)
4746 view->lineno = tree_lineno;
4748 return REQ_NONE;
4749 }
4751 static bool
4752 tree_grep(struct view *view, struct line *line)
4753 {
4754 struct tree_entry *entry = line->data;
4755 const char *text[] = {
4756 entry->name,
4757 opt_author ? entry->author : "",
4758 mkdate(&entry->time, opt_date),
4759 NULL
4760 };
4762 return grep_text(view, text);
4763 }
4765 static void
4766 tree_select(struct view *view, struct line *line)
4767 {
4768 struct tree_entry *entry = line->data;
4770 if (line->type == LINE_TREE_FILE) {
4771 string_copy_rev(ref_blob, entry->id);
4772 string_format(opt_file, "%s%s", opt_path, tree_path(line));
4774 } else if (line->type != LINE_TREE_DIR) {
4775 return;
4776 }
4778 string_copy_rev(view->ref, entry->id);
4779 }
4781 static bool
4782 tree_prepare(struct view *view)
4783 {
4784 if (view->lines == 0 && opt_prefix[0]) {
4785 char *pos = opt_prefix;
4787 while (pos && *pos) {
4788 char *end = strchr(pos, '/');
4790 if (end)
4791 *end = 0;
4792 push_tree_stack_entry(pos, 0);
4793 pos = end;
4794 if (end) {
4795 *end = '/';
4796 pos++;
4797 }
4798 }
4800 } else if (strcmp(view->vid, view->id)) {
4801 opt_path[0] = 0;
4802 }
4804 return prepare_io(view, opt_cdup, view->ops->argv, TRUE);
4805 }
4807 static const char *tree_argv[SIZEOF_ARG] = {
4808 "git", "ls-tree", "%(commit)", "%(directory)", NULL
4809 };
4811 static struct view_ops tree_ops = {
4812 "file",
4813 tree_argv,
4814 NULL,
4815 tree_read,
4816 tree_draw,
4817 tree_request,
4818 tree_grep,
4819 tree_select,
4820 tree_prepare,
4821 };
4823 static bool
4824 blob_read(struct view *view, char *line)
4825 {
4826 if (!line)
4827 return TRUE;
4828 return add_line_text(view, line, LINE_DEFAULT) != NULL;
4829 }
4831 static enum request
4832 blob_request(struct view *view, enum request request, struct line *line)
4833 {
4834 switch (request) {
4835 case REQ_EDIT:
4836 open_blob_editor(view->vid);
4837 return REQ_NONE;
4838 default:
4839 return pager_request(view, request, line);
4840 }
4841 }
4843 static const char *blob_argv[SIZEOF_ARG] = {
4844 "git", "cat-file", "blob", "%(blob)", NULL
4845 };
4847 static struct view_ops blob_ops = {
4848 "line",
4849 blob_argv,
4850 NULL,
4851 blob_read,
4852 pager_draw,
4853 blob_request,
4854 pager_grep,
4855 pager_select,
4856 };
4858 /*
4859 * Blame backend
4860 *
4861 * Loading the blame view is a two phase job:
4862 *
4863 * 1. File content is read either using opt_file from the
4864 * filesystem or using git-cat-file.
4865 * 2. Then blame information is incrementally added by
4866 * reading output from git-blame.
4867 */
4869 struct blame_commit {
4870 char id[SIZEOF_REV]; /* SHA1 ID. */
4871 char title[128]; /* First line of the commit message. */
4872 const char *author; /* Author of the commit. */
4873 struct time time; /* Date from the author ident. */
4874 char filename[128]; /* Name of file. */
4875 bool has_previous; /* Was a "previous" line detected. */
4876 };
4878 struct blame {
4879 struct blame_commit *commit;
4880 unsigned long lineno;
4881 char text[1];
4882 };
4884 static bool
4885 blame_open(struct view *view)
4886 {
4887 char path[SIZEOF_STR];
4889 if (!view->prev && *opt_prefix) {
4890 string_copy(path, opt_file);
4891 if (!string_format(opt_file, "%s%s", opt_prefix, path))
4892 return FALSE;
4893 }
4895 if (*opt_ref || !io_open(&view->io, "%s%s", opt_cdup, opt_file)) {
4896 const char *blame_cat_file_argv[] = {
4897 "git", "cat-file", "blob", path, NULL
4898 };
4900 if (!string_format(path, "%s:%s", opt_ref, opt_file) ||
4901 !start_update(view, blame_cat_file_argv, opt_cdup))
4902 return FALSE;
4903 }
4905 setup_update(view, opt_file);
4906 string_format(view->ref, "%s ...", opt_file);
4908 return TRUE;
4909 }
4911 static struct blame_commit *
4912 get_blame_commit(struct view *view, const char *id)
4913 {
4914 size_t i;
4916 for (i = 0; i < view->lines; i++) {
4917 struct blame *blame = view->line[i].data;
4919 if (!blame->commit)
4920 continue;
4922 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
4923 return blame->commit;
4924 }
4926 {
4927 struct blame_commit *commit = calloc(1, sizeof(*commit));
4929 if (commit)
4930 string_ncopy(commit->id, id, SIZEOF_REV);
4931 return commit;
4932 }
4933 }
4935 static bool
4936 parse_number(const char **posref, size_t *number, size_t min, size_t max)
4937 {
4938 const char *pos = *posref;
4940 *posref = NULL;
4941 pos = strchr(pos + 1, ' ');
4942 if (!pos || !isdigit(pos[1]))
4943 return FALSE;
4944 *number = atoi(pos + 1);
4945 if (*number < min || *number > max)
4946 return FALSE;
4948 *posref = pos;
4949 return TRUE;
4950 }
4952 static struct blame_commit *
4953 parse_blame_commit(struct view *view, const char *text, int *blamed)
4954 {
4955 struct blame_commit *commit;
4956 struct blame *blame;
4957 const char *pos = text + SIZEOF_REV - 2;
4958 size_t orig_lineno = 0;
4959 size_t lineno;
4960 size_t group;
4962 if (strlen(text) <= SIZEOF_REV || pos[1] != ' ')
4963 return NULL;
4965 if (!parse_number(&pos, &orig_lineno, 1, 9999999) ||
4966 !parse_number(&pos, &lineno, 1, view->lines) ||
4967 !parse_number(&pos, &group, 1, view->lines - lineno + 1))
4968 return NULL;
4970 commit = get_blame_commit(view, text);
4971 if (!commit)
4972 return NULL;
4974 *blamed += group;
4975 while (group--) {
4976 struct line *line = &view->line[lineno + group - 1];
4978 blame = line->data;
4979 blame->commit = commit;
4980 blame->lineno = orig_lineno + group - 1;
4981 line->dirty = 1;
4982 }
4984 return commit;
4985 }
4987 static bool
4988 blame_read_file(struct view *view, const char *line, bool *read_file)
4989 {
4990 if (!line) {
4991 const char *blame_argv[] = {
4992 "git", "blame", "--incremental",
4993 *opt_ref ? opt_ref : "--incremental", "--", opt_file, NULL
4994 };
4996 if (view->lines == 0 && !view->prev)
4997 die("No blame exist for %s", view->vid);
4999 if (view->lines == 0 || !start_update(view, blame_argv, opt_cdup)) {
5000 report("Failed to load blame data");
5001 return TRUE;
5002 }
5004 *read_file = FALSE;
5005 return FALSE;
5007 } else {
5008 size_t linelen = strlen(line);
5009 struct blame *blame = malloc(sizeof(*blame) + linelen);
5011 if (!blame)
5012 return FALSE;
5014 blame->commit = NULL;
5015 strncpy(blame->text, line, linelen);
5016 blame->text[linelen] = 0;
5017 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
5018 }
5019 }
5021 static bool
5022 match_blame_header(const char *name, char **line)
5023 {
5024 size_t namelen = strlen(name);
5025 bool matched = !strncmp(name, *line, namelen);
5027 if (matched)
5028 *line += namelen;
5030 return matched;
5031 }
5033 static bool
5034 blame_read(struct view *view, char *line)
5035 {
5036 static struct blame_commit *commit = NULL;
5037 static int blamed = 0;
5038 static bool read_file = TRUE;
5040 if (read_file)
5041 return blame_read_file(view, line, &read_file);
5043 if (!line) {
5044 /* Reset all! */
5045 commit = NULL;
5046 blamed = 0;
5047 read_file = TRUE;
5048 string_format(view->ref, "%s", view->vid);
5049 if (view_is_displayed(view)) {
5050 update_view_title(view);
5051 redraw_view_from(view, 0);
5052 }
5053 return TRUE;
5054 }
5056 if (!commit) {
5057 commit = parse_blame_commit(view, line, &blamed);
5058 string_format(view->ref, "%s %2d%%", view->vid,
5059 view->lines ? blamed * 100 / view->lines : 0);
5061 } else if (match_blame_header("author ", &line)) {
5062 commit->author = get_author(line);
5064 } else if (match_blame_header("author-time ", &line)) {
5065 parse_timesec(&commit->time, line);
5067 } else if (match_blame_header("author-tz ", &line)) {
5068 parse_timezone(&commit->time, line);
5070 } else if (match_blame_header("summary ", &line)) {
5071 string_ncopy(commit->title, line, strlen(line));
5073 } else if (match_blame_header("previous ", &line)) {
5074 commit->has_previous = TRUE;
5076 } else if (match_blame_header("filename ", &line)) {
5077 string_ncopy(commit->filename, line, strlen(line));
5078 commit = NULL;
5079 }
5081 return TRUE;
5082 }
5084 static bool
5085 blame_draw(struct view *view, struct line *line, unsigned int lineno)
5086 {
5087 struct blame *blame = line->data;
5088 struct time *time = NULL;
5089 const char *id = NULL, *author = NULL;
5090 char text[SIZEOF_STR];
5092 if (blame->commit && *blame->commit->filename) {
5093 id = blame->commit->id;
5094 author = blame->commit->author;
5095 time = &blame->commit->time;
5096 }
5098 if (opt_date && draw_date(view, time))
5099 return TRUE;
5101 if (opt_author && draw_author(view, author))
5102 return TRUE;
5104 if (draw_field(view, LINE_BLAME_ID, id, ID_COLS, FALSE))
5105 return TRUE;
5107 if (draw_lineno(view, lineno))
5108 return TRUE;
5110 string_expand(text, sizeof(text), blame->text, opt_tab_size);
5111 draw_text(view, LINE_DEFAULT, text, TRUE);
5112 return TRUE;
5113 }
5115 static bool
5116 check_blame_commit(struct blame *blame, bool check_null_id)
5117 {
5118 if (!blame->commit)
5119 report("Commit data not loaded yet");
5120 else if (check_null_id && !strcmp(blame->commit->id, NULL_ID))
5121 report("No commit exist for the selected line");
5122 else
5123 return TRUE;
5124 return FALSE;
5125 }
5127 static void
5128 setup_blame_parent_line(struct view *view, struct blame *blame)
5129 {
5130 const char *diff_tree_argv[] = {
5131 "git", "diff-tree", "-U0", blame->commit->id,
5132 "--", blame->commit->filename, NULL
5133 };
5134 struct io io;
5135 int parent_lineno = -1;
5136 int blamed_lineno = -1;
5137 char *line;
5139 if (!io_run(&io, IO_RD, NULL, diff_tree_argv))
5140 return;
5142 while ((line = io_get(&io, '\n', TRUE))) {
5143 if (*line == '@') {
5144 char *pos = strchr(line, '+');
5146 parent_lineno = atoi(line + 4);
5147 if (pos)
5148 blamed_lineno = atoi(pos + 1);
5150 } else if (*line == '+' && parent_lineno != -1) {
5151 if (blame->lineno == blamed_lineno - 1 &&
5152 !strcmp(blame->text, line + 1)) {
5153 view->lineno = parent_lineno ? parent_lineno - 1 : 0;
5154 break;
5155 }
5156 blamed_lineno++;
5157 }
5158 }
5160 io_done(&io);
5161 }
5163 static enum request
5164 blame_request(struct view *view, enum request request, struct line *line)
5165 {
5166 enum open_flags flags = view_is_displayed(view) ? OPEN_SPLIT : OPEN_DEFAULT;
5167 struct blame *blame = line->data;
5169 switch (request) {
5170 case REQ_VIEW_BLAME:
5171 if (check_blame_commit(blame, TRUE)) {
5172 string_copy(opt_ref, blame->commit->id);
5173 string_copy(opt_file, blame->commit->filename);
5174 if (blame->lineno)
5175 view->lineno = blame->lineno;
5176 open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
5177 }
5178 break;
5180 case REQ_PARENT:
5181 if (check_blame_commit(blame, TRUE) &&
5182 select_commit_parent(blame->commit->id, opt_ref,
5183 blame->commit->filename)) {
5184 string_copy(opt_file, blame->commit->filename);
5185 setup_blame_parent_line(view, blame);
5186 open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
5187 }
5188 break;
5190 case REQ_ENTER:
5191 if (!check_blame_commit(blame, FALSE))
5192 break;
5194 if (view_is_displayed(VIEW(REQ_VIEW_DIFF)) &&
5195 !strcmp(blame->commit->id, VIEW(REQ_VIEW_DIFF)->ref))
5196 break;
5198 if (!strcmp(blame->commit->id, NULL_ID)) {
5199 struct view *diff = VIEW(REQ_VIEW_DIFF);
5200 const char *diff_index_argv[] = {
5201 "git", "diff-index", "--root", "--patch-with-stat",
5202 "-C", "-M", "HEAD", "--", view->vid, NULL
5203 };
5205 if (!blame->commit->has_previous) {
5206 diff_index_argv[1] = "diff";
5207 diff_index_argv[2] = "--no-color";
5208 diff_index_argv[6] = "--";
5209 diff_index_argv[7] = "/dev/null";
5210 }
5212 if (!prepare_update(diff, diff_index_argv, NULL)) {
5213 report("Failed to allocate diff command");
5214 break;
5215 }
5216 flags |= OPEN_PREPARED;
5217 }
5219 open_view(view, REQ_VIEW_DIFF, flags);
5220 if (VIEW(REQ_VIEW_DIFF)->pipe && !strcmp(blame->commit->id, NULL_ID))
5221 string_copy_rev(VIEW(REQ_VIEW_DIFF)->ref, NULL_ID);
5222 break;
5224 default:
5225 return request;
5226 }
5228 return REQ_NONE;
5229 }
5231 static bool
5232 blame_grep(struct view *view, struct line *line)
5233 {
5234 struct blame *blame = line->data;
5235 struct blame_commit *commit = blame->commit;
5236 const char *text[] = {
5237 blame->text,
5238 commit ? commit->title : "",
5239 commit ? commit->id : "",
5240 commit && opt_author ? commit->author : "",
5241 commit ? mkdate(&commit->time, opt_date) : "",
5242 NULL
5243 };
5245 return grep_text(view, text);
5246 }
5248 static void
5249 blame_select(struct view *view, struct line *line)
5250 {
5251 struct blame *blame = line->data;
5252 struct blame_commit *commit = blame->commit;
5254 if (!commit)
5255 return;
5257 if (!strcmp(commit->id, NULL_ID))
5258 string_ncopy(ref_commit, "HEAD", 4);
5259 else
5260 string_copy_rev(ref_commit, commit->id);
5261 }
5263 static struct view_ops blame_ops = {
5264 "line",
5265 NULL,
5266 blame_open,
5267 blame_read,
5268 blame_draw,
5269 blame_request,
5270 blame_grep,
5271 blame_select,
5272 };
5274 /*
5275 * Branch backend
5276 */
5278 struct branch {
5279 const char *author; /* Author of the last commit. */
5280 struct time time; /* Date of the last activity. */
5281 const struct ref *ref; /* Name and commit ID information. */
5282 };
5284 static const struct ref branch_all;
5286 static const enum sort_field branch_sort_fields[] = {
5287 ORDERBY_NAME, ORDERBY_DATE, ORDERBY_AUTHOR
5288 };
5289 static struct sort_state branch_sort_state = SORT_STATE(branch_sort_fields);
5291 static int
5292 branch_compare(const void *l1, const void *l2)
5293 {
5294 const struct branch *branch1 = ((const struct line *) l1)->data;
5295 const struct branch *branch2 = ((const struct line *) l2)->data;
5297 switch (get_sort_field(branch_sort_state)) {
5298 case ORDERBY_DATE:
5299 return sort_order(branch_sort_state, timecmp(&branch1->time, &branch2->time));
5301 case ORDERBY_AUTHOR:
5302 return sort_order(branch_sort_state, strcmp(branch1->author, branch2->author));
5304 case ORDERBY_NAME:
5305 default:
5306 return sort_order(branch_sort_state, strcmp(branch1->ref->name, branch2->ref->name));
5307 }
5308 }
5310 static bool
5311 branch_draw(struct view *view, struct line *line, unsigned int lineno)
5312 {
5313 struct branch *branch = line->data;
5314 enum line_type type = branch->ref->head ? LINE_MAIN_HEAD : LINE_DEFAULT;
5316 if (opt_date && draw_date(view, &branch->time))
5317 return TRUE;
5319 if (opt_author && draw_author(view, branch->author))
5320 return TRUE;
5322 draw_text(view, type, branch->ref == &branch_all ? "All branches" : branch->ref->name, TRUE);
5323 return TRUE;
5324 }
5326 static enum request
5327 branch_request(struct view *view, enum request request, struct line *line)
5328 {
5329 struct branch *branch = line->data;
5331 switch (request) {
5332 case REQ_REFRESH:
5333 load_refs();
5334 open_view(view, REQ_VIEW_BRANCH, OPEN_REFRESH);
5335 return REQ_NONE;
5337 case REQ_TOGGLE_SORT_FIELD:
5338 case REQ_TOGGLE_SORT_ORDER:
5339 sort_view(view, request, &branch_sort_state, branch_compare);
5340 return REQ_NONE;
5342 case REQ_ENTER:
5343 if (branch->ref == &branch_all) {
5344 const char *all_branches_argv[] = {
5345 "git", "log", "--no-color", "--pretty=raw", "--parents",
5346 "--topo-order", "--all", NULL
5347 };
5348 struct view *main_view = VIEW(REQ_VIEW_MAIN);
5350 if (!prepare_update(main_view, all_branches_argv, NULL)) {
5351 report("Failed to load view of all branches");
5352 return REQ_NONE;
5353 }
5354 open_view(view, REQ_VIEW_MAIN, OPEN_PREPARED | OPEN_SPLIT);
5355 } else {
5356 open_view(view, REQ_VIEW_MAIN, OPEN_SPLIT);
5357 }
5358 return REQ_NONE;
5360 default:
5361 return request;
5362 }
5363 }
5365 static bool
5366 branch_read(struct view *view, char *line)
5367 {
5368 static char id[SIZEOF_REV];
5369 struct branch *reference;
5370 size_t i;
5372 if (!line)
5373 return TRUE;
5375 switch (get_line_type(line)) {
5376 case LINE_COMMIT:
5377 string_copy_rev(id, line + STRING_SIZE("commit "));
5378 return TRUE;
5380 case LINE_AUTHOR:
5381 for (i = 0, reference = NULL; i < view->lines; i++) {
5382 struct branch *branch = view->line[i].data;
5384 if (strcmp(branch->ref->id, id))
5385 continue;
5387 view->line[i].dirty = TRUE;
5388 if (reference) {
5389 branch->author = reference->author;
5390 branch->time = reference->time;
5391 continue;
5392 }
5394 parse_author_line(line + STRING_SIZE("author "),
5395 &branch->author, &branch->time);
5396 reference = branch;
5397 }
5398 return TRUE;
5400 default:
5401 return TRUE;
5402 }
5404 }
5406 static bool
5407 branch_open_visitor(void *data, const struct ref *ref)
5408 {
5409 struct view *view = data;
5410 struct branch *branch;
5412 if (ref->tag || ref->ltag || ref->remote)
5413 return TRUE;
5415 branch = calloc(1, sizeof(*branch));
5416 if (!branch)
5417 return FALSE;
5419 branch->ref = ref;
5420 return !!add_line_data(view, branch, LINE_DEFAULT);
5421 }
5423 static bool
5424 branch_open(struct view *view)
5425 {
5426 const char *branch_log[] = {
5427 "git", "log", "--no-color", "--pretty=raw",
5428 "--simplify-by-decoration", "--all", NULL
5429 };
5431 if (!start_update(view, branch_log, NULL)) {
5432 report("Failed to load branch data");
5433 return TRUE;
5434 }
5436 setup_update(view, view->id);
5437 branch_open_visitor(view, &branch_all);
5438 foreach_ref(branch_open_visitor, view);
5439 view->p_restore = TRUE;
5441 return TRUE;
5442 }
5444 static bool
5445 branch_grep(struct view *view, struct line *line)
5446 {
5447 struct branch *branch = line->data;
5448 const char *text[] = {
5449 branch->ref->name,
5450 branch->author,
5451 NULL
5452 };
5454 return grep_text(view, text);
5455 }
5457 static void
5458 branch_select(struct view *view, struct line *line)
5459 {
5460 struct branch *branch = line->data;
5462 string_copy_rev(view->ref, branch->ref->id);
5463 string_copy_rev(ref_commit, branch->ref->id);
5464 string_copy_rev(ref_head, branch->ref->id);
5465 string_copy_rev(ref_branch, branch->ref->name);
5466 }
5468 static struct view_ops branch_ops = {
5469 "branch",
5470 NULL,
5471 branch_open,
5472 branch_read,
5473 branch_draw,
5474 branch_request,
5475 branch_grep,
5476 branch_select,
5477 };
5479 /*
5480 * Status backend
5481 */
5483 struct status {
5484 char status;
5485 struct {
5486 mode_t mode;
5487 char rev[SIZEOF_REV];
5488 char name[SIZEOF_STR];
5489 } old;
5490 struct {
5491 mode_t mode;
5492 char rev[SIZEOF_REV];
5493 char name[SIZEOF_STR];
5494 } new;
5495 };
5497 static char status_onbranch[SIZEOF_STR];
5498 static struct status stage_status;
5499 static enum line_type stage_line_type;
5500 static size_t stage_chunks;
5501 static int *stage_chunk;
5503 DEFINE_ALLOCATOR(realloc_ints, int, 32)
5505 /* This should work even for the "On branch" line. */
5506 static inline bool
5507 status_has_none(struct view *view, struct line *line)
5508 {
5509 return line < view->line + view->lines && !line[1].data;
5510 }
5512 /* Get fields from the diff line:
5513 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
5514 */
5515 static inline bool
5516 status_get_diff(struct status *file, const char *buf, size_t bufsize)
5517 {
5518 const char *old_mode = buf + 1;
5519 const char *new_mode = buf + 8;
5520 const char *old_rev = buf + 15;
5521 const char *new_rev = buf + 56;
5522 const char *status = buf + 97;
5524 if (bufsize < 98 ||
5525 old_mode[-1] != ':' ||
5526 new_mode[-1] != ' ' ||
5527 old_rev[-1] != ' ' ||
5528 new_rev[-1] != ' ' ||
5529 status[-1] != ' ')
5530 return FALSE;
5532 file->status = *status;
5534 string_copy_rev(file->old.rev, old_rev);
5535 string_copy_rev(file->new.rev, new_rev);
5537 file->old.mode = strtoul(old_mode, NULL, 8);
5538 file->new.mode = strtoul(new_mode, NULL, 8);
5540 file->old.name[0] = file->new.name[0] = 0;
5542 return TRUE;
5543 }
5545 static bool
5546 status_run(struct view *view, const char *argv[], char status, enum line_type type)
5547 {
5548 struct status *unmerged = NULL;
5549 char *buf;
5550 struct io io;
5552 if (!io_run(&io, IO_RD, opt_cdup, argv))
5553 return FALSE;
5555 add_line_data(view, NULL, type);
5557 while ((buf = io_get(&io, 0, TRUE))) {
5558 struct status *file = unmerged;
5560 if (!file) {
5561 file = calloc(1, sizeof(*file));
5562 if (!file || !add_line_data(view, file, type))
5563 goto error_out;
5564 }
5566 /* Parse diff info part. */
5567 if (status) {
5568 file->status = status;
5569 if (status == 'A')
5570 string_copy(file->old.rev, NULL_ID);
5572 } else if (!file->status || file == unmerged) {
5573 if (!status_get_diff(file, buf, strlen(buf)))
5574 goto error_out;
5576 buf = io_get(&io, 0, TRUE);
5577 if (!buf)
5578 break;
5580 /* Collapse all modified entries that follow an
5581 * associated unmerged entry. */
5582 if (unmerged == file) {
5583 unmerged->status = 'U';
5584 unmerged = NULL;
5585 } else if (file->status == 'U') {
5586 unmerged = file;
5587 }
5588 }
5590 /* Grab the old name for rename/copy. */
5591 if (!*file->old.name &&
5592 (file->status == 'R' || file->status == 'C')) {
5593 string_ncopy(file->old.name, buf, strlen(buf));
5595 buf = io_get(&io, 0, TRUE);
5596 if (!buf)
5597 break;
5598 }
5600 /* git-ls-files just delivers a NUL separated list of
5601 * file names similar to the second half of the
5602 * git-diff-* output. */
5603 string_ncopy(file->new.name, buf, strlen(buf));
5604 if (!*file->old.name)
5605 string_copy(file->old.name, file->new.name);
5606 file = NULL;
5607 }
5609 if (io_error(&io)) {
5610 error_out:
5611 io_done(&io);
5612 return FALSE;
5613 }
5615 if (!view->line[view->lines - 1].data)
5616 add_line_data(view, NULL, LINE_STAT_NONE);
5618 io_done(&io);
5619 return TRUE;
5620 }
5622 /* Don't show unmerged entries in the staged section. */
5623 static const char *status_diff_index_argv[] = {
5624 "git", "diff-index", "-z", "--diff-filter=ACDMRTXB",
5625 "--cached", "-M", "HEAD", NULL
5626 };
5628 static const char *status_diff_files_argv[] = {
5629 "git", "diff-files", "-z", NULL
5630 };
5632 static const char *status_list_other_argv[] = {
5633 "git", "ls-files", "-z", "--others", "--exclude-standard", opt_prefix, NULL
5634 };
5636 static const char *status_list_no_head_argv[] = {
5637 "git", "ls-files", "-z", "--cached", "--exclude-standard", NULL
5638 };
5640 static const char *update_index_argv[] = {
5641 "git", "update-index", "-q", "--unmerged", "--refresh", NULL
5642 };
5644 /* Restore the previous line number to stay in the context or select a
5645 * line with something that can be updated. */
5646 static void
5647 status_restore(struct view *view)
5648 {
5649 if (view->p_lineno >= view->lines)
5650 view->p_lineno = view->lines - 1;
5651 while (view->p_lineno < view->lines && !view->line[view->p_lineno].data)
5652 view->p_lineno++;
5653 while (view->p_lineno > 0 && !view->line[view->p_lineno].data)
5654 view->p_lineno--;
5656 /* If the above fails, always skip the "On branch" line. */
5657 if (view->p_lineno < view->lines)
5658 view->lineno = view->p_lineno;
5659 else
5660 view->lineno = 1;
5662 if (view->lineno < view->offset)
5663 view->offset = view->lineno;
5664 else if (view->offset + view->height <= view->lineno)
5665 view->offset = view->lineno - view->height + 1;
5667 view->p_restore = FALSE;
5668 }
5670 static void
5671 status_update_onbranch(void)
5672 {
5673 static const char *paths[][2] = {
5674 { "rebase-apply/rebasing", "Rebasing" },
5675 { "rebase-apply/applying", "Applying mailbox" },
5676 { "rebase-apply/", "Rebasing mailbox" },
5677 { "rebase-merge/interactive", "Interactive rebase" },
5678 { "rebase-merge/", "Rebase merge" },
5679 { "MERGE_HEAD", "Merging" },
5680 { "BISECT_LOG", "Bisecting" },
5681 { "HEAD", "On branch" },
5682 };
5683 char buf[SIZEOF_STR];
5684 struct stat stat;
5685 int i;
5687 if (is_initial_commit()) {
5688 string_copy(status_onbranch, "Initial commit");
5689 return;
5690 }
5692 for (i = 0; i < ARRAY_SIZE(paths); i++) {
5693 char *head = opt_head;
5695 if (!string_format(buf, "%s/%s", opt_git_dir, paths[i][0]) ||
5696 lstat(buf, &stat) < 0)
5697 continue;
5699 if (!*opt_head) {
5700 struct io io;
5702 if (io_open(&io, "%s/rebase-merge/head-name", opt_git_dir) &&
5703 io_read_buf(&io, buf, sizeof(buf))) {
5704 head = buf;
5705 if (!prefixcmp(head, "refs/heads/"))
5706 head += STRING_SIZE("refs/heads/");
5707 }
5708 }
5710 if (!string_format(status_onbranch, "%s %s", paths[i][1], head))
5711 string_copy(status_onbranch, opt_head);
5712 return;
5713 }
5715 string_copy(status_onbranch, "Not currently on any branch");
5716 }
5718 /* First parse staged info using git-diff-index(1), then parse unstaged
5719 * info using git-diff-files(1), and finally untracked files using
5720 * git-ls-files(1). */
5721 static bool
5722 status_open(struct view *view)
5723 {
5724 reset_view(view);
5726 add_line_data(view, NULL, LINE_STAT_HEAD);
5727 status_update_onbranch();
5729 io_run_bg(update_index_argv);
5731 if (is_initial_commit()) {
5732 if (!status_run(view, status_list_no_head_argv, 'A', LINE_STAT_STAGED))
5733 return FALSE;
5734 } else if (!status_run(view, status_diff_index_argv, 0, LINE_STAT_STAGED)) {
5735 return FALSE;
5736 }
5738 if (!status_run(view, status_diff_files_argv, 0, LINE_STAT_UNSTAGED) ||
5739 !status_run(view, status_list_other_argv, '?', LINE_STAT_UNTRACKED))
5740 return FALSE;
5742 /* Restore the exact position or use the specialized restore
5743 * mode? */
5744 if (!view->p_restore)
5745 status_restore(view);
5746 return TRUE;
5747 }
5749 static bool
5750 status_draw(struct view *view, struct line *line, unsigned int lineno)
5751 {
5752 struct status *status = line->data;
5753 enum line_type type;
5754 const char *text;
5756 if (!status) {
5757 switch (line->type) {
5758 case LINE_STAT_STAGED:
5759 type = LINE_STAT_SECTION;
5760 text = "Changes to be committed:";
5761 break;
5763 case LINE_STAT_UNSTAGED:
5764 type = LINE_STAT_SECTION;
5765 text = "Changed but not updated:";
5766 break;
5768 case LINE_STAT_UNTRACKED:
5769 type = LINE_STAT_SECTION;
5770 text = "Untracked files:";
5771 break;
5773 case LINE_STAT_NONE:
5774 type = LINE_DEFAULT;
5775 text = " (no files)";
5776 break;
5778 case LINE_STAT_HEAD:
5779 type = LINE_STAT_HEAD;
5780 text = status_onbranch;
5781 break;
5783 default:
5784 return FALSE;
5785 }
5786 } else {
5787 static char buf[] = { '?', ' ', ' ', ' ', 0 };
5789 buf[0] = status->status;
5790 if (draw_text(view, line->type, buf, TRUE))
5791 return TRUE;
5792 type = LINE_DEFAULT;
5793 text = status->new.name;
5794 }
5796 draw_text(view, type, text, TRUE);
5797 return TRUE;
5798 }
5800 static enum request
5801 status_load_error(struct view *view, struct view *stage, const char *path)
5802 {
5803 if (displayed_views() == 2 || display[current_view] != view)
5804 maximize_view(view);
5805 report("Failed to load '%s': %s", path, io_strerror(&stage->io));
5806 return REQ_NONE;
5807 }
5809 static enum request
5810 status_enter(struct view *view, struct line *line)
5811 {
5812 struct status *status = line->data;
5813 const char *oldpath = status ? status->old.name : NULL;
5814 /* Diffs for unmerged entries are empty when passing the new
5815 * path, so leave it empty. */
5816 const char *newpath = status && status->status != 'U' ? status->new.name : NULL;
5817 const char *info;
5818 enum open_flags split;
5819 struct view *stage = VIEW(REQ_VIEW_STAGE);
5821 if (line->type == LINE_STAT_NONE ||
5822 (!status && line[1].type == LINE_STAT_NONE)) {
5823 report("No file to diff");
5824 return REQ_NONE;
5825 }
5827 switch (line->type) {
5828 case LINE_STAT_STAGED:
5829 if (is_initial_commit()) {
5830 const char *no_head_diff_argv[] = {
5831 "git", "diff", "--no-color", "--patch-with-stat",
5832 "--", "/dev/null", newpath, NULL
5833 };
5835 if (!prepare_update(stage, no_head_diff_argv, opt_cdup))
5836 return status_load_error(view, stage, newpath);
5837 } else {
5838 const char *index_show_argv[] = {
5839 "git", "diff-index", "--root", "--patch-with-stat",
5840 "-C", "-M", "--cached", "HEAD", "--",
5841 oldpath, newpath, NULL
5842 };
5844 if (!prepare_update(stage, index_show_argv, opt_cdup))
5845 return status_load_error(view, stage, newpath);
5846 }
5848 if (status)
5849 info = "Staged changes to %s";
5850 else
5851 info = "Staged changes";
5852 break;
5854 case LINE_STAT_UNSTAGED:
5855 {
5856 const char *files_show_argv[] = {
5857 "git", "diff-files", "--root", "--patch-with-stat",
5858 "-C", "-M", "--", oldpath, newpath, NULL
5859 };
5861 if (!prepare_update(stage, files_show_argv, opt_cdup))
5862 return status_load_error(view, stage, newpath);
5863 if (status)
5864 info = "Unstaged changes to %s";
5865 else
5866 info = "Unstaged changes";
5867 break;
5868 }
5869 case LINE_STAT_UNTRACKED:
5870 if (!newpath) {
5871 report("No file to show");
5872 return REQ_NONE;
5873 }
5875 if (!suffixcmp(status->new.name, -1, "/")) {
5876 report("Cannot display a directory");
5877 return REQ_NONE;
5878 }
5880 if (!prepare_update_file(stage, newpath))
5881 return status_load_error(view, stage, newpath);
5882 info = "Untracked file %s";
5883 break;
5885 case LINE_STAT_HEAD:
5886 return REQ_NONE;
5888 default:
5889 die("line type %d not handled in switch", line->type);
5890 }
5892 split = view_is_displayed(view) ? OPEN_SPLIT : OPEN_DEFAULT;
5893 open_view(view, REQ_VIEW_STAGE, OPEN_PREPARED | split);
5894 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
5895 if (status) {
5896 stage_status = *status;
5897 } else {
5898 memset(&stage_status, 0, sizeof(stage_status));
5899 }
5901 stage_line_type = line->type;
5902 stage_chunks = 0;
5903 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
5904 }
5906 return REQ_NONE;
5907 }
5909 static bool
5910 status_exists(struct status *status, enum line_type type)
5911 {
5912 struct view *view = VIEW(REQ_VIEW_STATUS);
5913 unsigned long lineno;
5915 for (lineno = 0; lineno < view->lines; lineno++) {
5916 struct line *line = &view->line[lineno];
5917 struct status *pos = line->data;
5919 if (line->type != type)
5920 continue;
5921 if (!pos && (!status || !status->status) && line[1].data) {
5922 select_view_line(view, lineno);
5923 return TRUE;
5924 }
5925 if (pos && !strcmp(status->new.name, pos->new.name)) {
5926 select_view_line(view, lineno);
5927 return TRUE;
5928 }
5929 }
5931 return FALSE;
5932 }
5935 static bool
5936 status_update_prepare(struct io *io, enum line_type type)
5937 {
5938 const char *staged_argv[] = {
5939 "git", "update-index", "-z", "--index-info", NULL
5940 };
5941 const char *others_argv[] = {
5942 "git", "update-index", "-z", "--add", "--remove", "--stdin", NULL
5943 };
5945 switch (type) {
5946 case LINE_STAT_STAGED:
5947 return io_run(io, IO_WR, opt_cdup, staged_argv);
5949 case LINE_STAT_UNSTAGED:
5950 case LINE_STAT_UNTRACKED:
5951 return io_run(io, IO_WR, opt_cdup, others_argv);
5953 default:
5954 die("line type %d not handled in switch", type);
5955 return FALSE;
5956 }
5957 }
5959 static bool
5960 status_update_write(struct io *io, struct status *status, enum line_type type)
5961 {
5962 char buf[SIZEOF_STR];
5963 size_t bufsize = 0;
5965 switch (type) {
5966 case LINE_STAT_STAGED:
5967 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
5968 status->old.mode,
5969 status->old.rev,
5970 status->old.name, 0))
5971 return FALSE;
5972 break;
5974 case LINE_STAT_UNSTAGED:
5975 case LINE_STAT_UNTRACKED:
5976 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
5977 return FALSE;
5978 break;
5980 default:
5981 die("line type %d not handled in switch", type);
5982 }
5984 return io_write(io, buf, bufsize);
5985 }
5987 static bool
5988 status_update_file(struct status *status, enum line_type type)
5989 {
5990 struct io io;
5991 bool result;
5993 if (!status_update_prepare(&io, type))
5994 return FALSE;
5996 result = status_update_write(&io, status, type);
5997 return io_done(&io) && result;
5998 }
6000 static bool
6001 status_update_files(struct view *view, struct line *line)
6002 {
6003 char buf[sizeof(view->ref)];
6004 struct io io;
6005 bool result = TRUE;
6006 struct line *pos = view->line + view->lines;
6007 int files = 0;
6008 int file, done;
6009 int cursor_y = -1, cursor_x = -1;
6011 if (!status_update_prepare(&io, line->type))
6012 return FALSE;
6014 for (pos = line; pos < view->line + view->lines && pos->data; pos++)
6015 files++;
6017 string_copy(buf, view->ref);
6018 getsyx(cursor_y, cursor_x);
6019 for (file = 0, done = 5; result && file < files; line++, file++) {
6020 int almost_done = file * 100 / files;
6022 if (almost_done > done) {
6023 done = almost_done;
6024 string_format(view->ref, "updating file %u of %u (%d%% done)",
6025 file, files, done);
6026 update_view_title(view);
6027 setsyx(cursor_y, cursor_x);
6028 doupdate();
6029 }
6030 result = status_update_write(&io, line->data, line->type);
6031 }
6032 string_copy(view->ref, buf);
6034 return io_done(&io) && result;
6035 }
6037 static bool
6038 status_update(struct view *view)
6039 {
6040 struct line *line = &view->line[view->lineno];
6042 assert(view->lines);
6044 if (!line->data) {
6045 /* This should work even for the "On branch" line. */
6046 if (line < view->line + view->lines && !line[1].data) {
6047 report("Nothing to update");
6048 return FALSE;
6049 }
6051 if (!status_update_files(view, line + 1)) {
6052 report("Failed to update file status");
6053 return FALSE;
6054 }
6056 } else if (!status_update_file(line->data, line->type)) {
6057 report("Failed to update file status");
6058 return FALSE;
6059 }
6061 return TRUE;
6062 }
6064 static bool
6065 status_revert(struct status *status, enum line_type type, bool has_none)
6066 {
6067 if (!status || type != LINE_STAT_UNSTAGED) {
6068 if (type == LINE_STAT_STAGED) {
6069 report("Cannot revert changes to staged files");
6070 } else if (type == LINE_STAT_UNTRACKED) {
6071 report("Cannot revert changes to untracked files");
6072 } else if (has_none) {
6073 report("Nothing to revert");
6074 } else {
6075 report("Cannot revert changes to multiple files");
6076 }
6078 } else if (prompt_yesno("Are you sure you want to revert changes?")) {
6079 char mode[10] = "100644";
6080 const char *reset_argv[] = {
6081 "git", "update-index", "--cacheinfo", mode,
6082 status->old.rev, status->old.name, NULL
6083 };
6084 const char *checkout_argv[] = {
6085 "git", "checkout", "--", status->old.name, NULL
6086 };
6088 if (status->status == 'U') {
6089 string_format(mode, "%5o", status->old.mode);
6091 if (status->old.mode == 0 && status->new.mode == 0) {
6092 reset_argv[2] = "--force-remove";
6093 reset_argv[3] = status->old.name;
6094 reset_argv[4] = NULL;
6095 }
6097 if (!io_run_fg(reset_argv, opt_cdup))
6098 return FALSE;
6099 if (status->old.mode == 0 && status->new.mode == 0)
6100 return TRUE;
6101 }
6103 return io_run_fg(checkout_argv, opt_cdup);
6104 }
6106 return FALSE;
6107 }
6109 static enum request
6110 status_request(struct view *view, enum request request, struct line *line)
6111 {
6112 struct status *status = line->data;
6114 switch (request) {
6115 case REQ_STATUS_UPDATE:
6116 if (!status_update(view))
6117 return REQ_NONE;
6118 break;
6120 case REQ_STATUS_REVERT:
6121 if (!status_revert(status, line->type, status_has_none(view, line)))
6122 return REQ_NONE;
6123 break;
6125 case REQ_STATUS_MERGE:
6126 if (!status || status->status != 'U') {
6127 report("Merging only possible for files with unmerged status ('U').");
6128 return REQ_NONE;
6129 }
6130 open_mergetool(status->new.name);
6131 break;
6133 case REQ_EDIT:
6134 if (!status)
6135 return request;
6136 if (status->status == 'D') {
6137 report("File has been deleted.");
6138 return REQ_NONE;
6139 }
6141 open_editor(status->new.name);
6142 break;
6144 case REQ_VIEW_BLAME:
6145 if (status)
6146 opt_ref[0] = 0;
6147 return request;
6149 case REQ_ENTER:
6150 /* After returning the status view has been split to
6151 * show the stage view. No further reloading is
6152 * necessary. */
6153 return status_enter(view, line);
6155 case REQ_REFRESH:
6156 /* Simply reload the view. */
6157 break;
6159 default:
6160 return request;
6161 }
6163 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
6165 return REQ_NONE;
6166 }
6168 static void
6169 status_select(struct view *view, struct line *line)
6170 {
6171 struct status *status = line->data;
6172 char file[SIZEOF_STR] = "all files";
6173 const char *text;
6174 const char *key;
6176 if (status && !string_format(file, "'%s'", status->new.name))
6177 return;
6179 if (!status && line[1].type == LINE_STAT_NONE)
6180 line++;
6182 switch (line->type) {
6183 case LINE_STAT_STAGED:
6184 text = "Press %s to unstage %s for commit";
6185 break;
6187 case LINE_STAT_UNSTAGED:
6188 text = "Press %s to stage %s for commit";
6189 break;
6191 case LINE_STAT_UNTRACKED:
6192 text = "Press %s to stage %s for addition";
6193 break;
6195 case LINE_STAT_HEAD:
6196 case LINE_STAT_NONE:
6197 text = "Nothing to update";
6198 break;
6200 default:
6201 die("line type %d not handled in switch", line->type);
6202 }
6204 if (status && status->status == 'U') {
6205 text = "Press %s to resolve conflict in %s";
6206 key = get_key(KEYMAP_STATUS, REQ_STATUS_MERGE);
6208 } else {
6209 key = get_key(KEYMAP_STATUS, REQ_STATUS_UPDATE);
6210 }
6212 string_format(view->ref, text, key, file);
6213 if (status)
6214 string_copy(opt_file, status->new.name);
6215 }
6217 static bool
6218 status_grep(struct view *view, struct line *line)
6219 {
6220 struct status *status = line->data;
6222 if (status) {
6223 const char buf[2] = { status->status, 0 };
6224 const char *text[] = { status->new.name, buf, NULL };
6226 return grep_text(view, text);
6227 }
6229 return FALSE;
6230 }
6232 static struct view_ops status_ops = {
6233 "file",
6234 NULL,
6235 status_open,
6236 NULL,
6237 status_draw,
6238 status_request,
6239 status_grep,
6240 status_select,
6241 };
6244 static bool
6245 stage_diff_write(struct io *io, struct line *line, struct line *end)
6246 {
6247 while (line < end) {
6248 if (!io_write(io, line->data, strlen(line->data)) ||
6249 !io_write(io, "\n", 1))
6250 return FALSE;
6251 line++;
6252 if (line->type == LINE_DIFF_CHUNK ||
6253 line->type == LINE_DIFF_HEADER)
6254 break;
6255 }
6257 return TRUE;
6258 }
6260 static struct line *
6261 stage_diff_find(struct view *view, struct line *line, enum line_type type)
6262 {
6263 for (; view->line < line; line--)
6264 if (line->type == type)
6265 return line;
6267 return NULL;
6268 }
6270 static bool
6271 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
6272 {
6273 const char *apply_argv[SIZEOF_ARG] = {
6274 "git", "apply", "--whitespace=nowarn", NULL
6275 };
6276 struct line *diff_hdr;
6277 struct io io;
6278 int argc = 3;
6280 diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
6281 if (!diff_hdr)
6282 return FALSE;
6284 if (!revert)
6285 apply_argv[argc++] = "--cached";
6286 if (revert || stage_line_type == LINE_STAT_STAGED)
6287 apply_argv[argc++] = "-R";
6288 apply_argv[argc++] = "-";
6289 apply_argv[argc++] = NULL;
6290 if (!io_run(&io, IO_WR, opt_cdup, apply_argv))
6291 return FALSE;
6293 if (!stage_diff_write(&io, diff_hdr, chunk) ||
6294 !stage_diff_write(&io, chunk, view->line + view->lines))
6295 chunk = NULL;
6297 io_done(&io);
6298 io_run_bg(update_index_argv);
6300 return chunk ? TRUE : FALSE;
6301 }
6303 static bool
6304 stage_update(struct view *view, struct line *line)
6305 {
6306 struct line *chunk = NULL;
6308 if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
6309 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
6311 if (chunk) {
6312 if (!stage_apply_chunk(view, chunk, FALSE)) {
6313 report("Failed to apply chunk");
6314 return FALSE;
6315 }
6317 } else if (!stage_status.status) {
6318 view = VIEW(REQ_VIEW_STATUS);
6320 for (line = view->line; line < view->line + view->lines; line++)
6321 if (line->type == stage_line_type)
6322 break;
6324 if (!status_update_files(view, line + 1)) {
6325 report("Failed to update files");
6326 return FALSE;
6327 }
6329 } else if (!status_update_file(&stage_status, stage_line_type)) {
6330 report("Failed to update file");
6331 return FALSE;
6332 }
6334 return TRUE;
6335 }
6337 static bool
6338 stage_revert(struct view *view, struct line *line)
6339 {
6340 struct line *chunk = NULL;
6342 if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
6343 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
6345 if (chunk) {
6346 if (!prompt_yesno("Are you sure you want to revert changes?"))
6347 return FALSE;
6349 if (!stage_apply_chunk(view, chunk, TRUE)) {
6350 report("Failed to revert chunk");
6351 return FALSE;
6352 }
6353 return TRUE;
6355 } else {
6356 return status_revert(stage_status.status ? &stage_status : NULL,
6357 stage_line_type, FALSE);
6358 }
6359 }
6362 static void
6363 stage_next(struct view *view, struct line *line)
6364 {
6365 int i;
6367 if (!stage_chunks) {
6368 for (line = view->line; line < view->line + view->lines; line++) {
6369 if (line->type != LINE_DIFF_CHUNK)
6370 continue;
6372 if (!realloc_ints(&stage_chunk, stage_chunks, 1)) {
6373 report("Allocation failure");
6374 return;
6375 }
6377 stage_chunk[stage_chunks++] = line - view->line;
6378 }
6379 }
6381 for (i = 0; i < stage_chunks; i++) {
6382 if (stage_chunk[i] > view->lineno) {
6383 do_scroll_view(view, stage_chunk[i] - view->lineno);
6384 report("Chunk %d of %d", i + 1, stage_chunks);
6385 return;
6386 }
6387 }
6389 report("No next chunk found");
6390 }
6392 static enum request
6393 stage_request(struct view *view, enum request request, struct line *line)
6394 {
6395 switch (request) {
6396 case REQ_STATUS_UPDATE:
6397 if (!stage_update(view, line))
6398 return REQ_NONE;
6399 break;
6401 case REQ_STATUS_REVERT:
6402 if (!stage_revert(view, line))
6403 return REQ_NONE;
6404 break;
6406 case REQ_STAGE_NEXT:
6407 if (stage_line_type == LINE_STAT_UNTRACKED) {
6408 report("File is untracked; press %s to add",
6409 get_key(KEYMAP_STAGE, REQ_STATUS_UPDATE));
6410 return REQ_NONE;
6411 }
6412 stage_next(view, line);
6413 return REQ_NONE;
6415 case REQ_EDIT:
6416 if (!stage_status.new.name[0])
6417 return request;
6418 if (stage_status.status == 'D') {
6419 report("File has been deleted.");
6420 return REQ_NONE;
6421 }
6423 open_editor(stage_status.new.name);
6424 break;
6426 case REQ_REFRESH:
6427 /* Reload everything ... */
6428 break;
6430 case REQ_VIEW_BLAME:
6431 if (stage_status.new.name[0]) {
6432 string_copy(opt_file, stage_status.new.name);
6433 opt_ref[0] = 0;
6434 }
6435 return request;
6437 case REQ_ENTER:
6438 return pager_request(view, request, line);
6440 default:
6441 return request;
6442 }
6444 VIEW(REQ_VIEW_STATUS)->p_restore = TRUE;
6445 open_view(view, REQ_VIEW_STATUS, OPEN_REFRESH);
6447 /* Check whether the staged entry still exists, and close the
6448 * stage view if it doesn't. */
6449 if (!status_exists(&stage_status, stage_line_type)) {
6450 status_restore(VIEW(REQ_VIEW_STATUS));
6451 return REQ_VIEW_CLOSE;
6452 }
6454 if (stage_line_type == LINE_STAT_UNTRACKED) {
6455 if (!suffixcmp(stage_status.new.name, -1, "/")) {
6456 report("Cannot display a directory");
6457 return REQ_NONE;
6458 }
6460 if (!prepare_update_file(view, stage_status.new.name)) {
6461 report("Failed to open file: %s", strerror(errno));
6462 return REQ_NONE;
6463 }
6464 }
6465 open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
6467 return REQ_NONE;
6468 }
6470 static struct view_ops stage_ops = {
6471 "line",
6472 NULL,
6473 NULL,
6474 pager_read,
6475 pager_draw,
6476 stage_request,
6477 pager_grep,
6478 pager_select,
6479 };
6482 /*
6483 * Revision graph
6484 */
6486 struct commit {
6487 char id[SIZEOF_REV]; /* SHA1 ID. */
6488 char title[128]; /* First line of the commit message. */
6489 const char *author; /* Author of the commit. */
6490 struct time time; /* Date from the author ident. */
6491 struct ref_list *refs; /* Repository references. */
6492 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
6493 size_t graph_size; /* The width of the graph array. */
6494 bool has_parents; /* Rewritten --parents seen. */
6495 };
6497 /* Size of rev graph with no "padding" columns */
6498 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
6500 struct rev_graph {
6501 struct rev_graph *prev, *next, *parents;
6502 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
6503 size_t size;
6504 struct commit *commit;
6505 size_t pos;
6506 unsigned int boundary:1;
6507 };
6509 /* Parents of the commit being visualized. */
6510 static struct rev_graph graph_parents[4];
6512 /* The current stack of revisions on the graph. */
6513 static struct rev_graph graph_stacks[4] = {
6514 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
6515 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
6516 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
6517 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
6518 };
6520 static inline bool
6521 graph_parent_is_merge(struct rev_graph *graph)
6522 {
6523 return graph->parents->size > 1;
6524 }
6526 static inline void
6527 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
6528 {
6529 struct commit *commit = graph->commit;
6531 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
6532 commit->graph[commit->graph_size++] = symbol;
6533 }
6535 static void
6536 clear_rev_graph(struct rev_graph *graph)
6537 {
6538 graph->boundary = 0;
6539 graph->size = graph->pos = 0;
6540 graph->commit = NULL;
6541 memset(graph->parents, 0, sizeof(*graph->parents));
6542 }
6544 static void
6545 done_rev_graph(struct rev_graph *graph)
6546 {
6547 if (graph_parent_is_merge(graph) &&
6548 graph->pos < graph->size - 1 &&
6549 graph->next->size == graph->size + graph->parents->size - 1) {
6550 size_t i = graph->pos + graph->parents->size - 1;
6552 graph->commit->graph_size = i * 2;
6553 while (i < graph->next->size - 1) {
6554 append_to_rev_graph(graph, ' ');
6555 append_to_rev_graph(graph, '\\');
6556 i++;
6557 }
6558 }
6560 clear_rev_graph(graph);
6561 }
6563 static void
6564 push_rev_graph(struct rev_graph *graph, const char *parent)
6565 {
6566 int i;
6568 /* "Collapse" duplicate parents lines.
6569 *
6570 * FIXME: This needs to also update update the drawn graph but
6571 * for now it just serves as a method for pruning graph lines. */
6572 for (i = 0; i < graph->size; i++)
6573 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
6574 return;
6576 if (graph->size < SIZEOF_REVITEMS) {
6577 string_copy_rev(graph->rev[graph->size++], parent);
6578 }
6579 }
6581 static chtype
6582 get_rev_graph_symbol(struct rev_graph *graph)
6583 {
6584 chtype symbol;
6586 if (graph->boundary)
6587 symbol = REVGRAPH_BOUND;
6588 else if (graph->parents->size == 0)
6589 symbol = REVGRAPH_INIT;
6590 else if (graph_parent_is_merge(graph))
6591 symbol = REVGRAPH_MERGE;
6592 else if (graph->pos >= graph->size)
6593 symbol = REVGRAPH_BRANCH;
6594 else
6595 symbol = REVGRAPH_COMMIT;
6597 return symbol;
6598 }
6600 static void
6601 draw_rev_graph(struct rev_graph *graph)
6602 {
6603 struct rev_filler {
6604 chtype separator, line;
6605 };
6606 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
6607 static struct rev_filler fillers[] = {
6608 { ' ', '|' },
6609 { '`', '.' },
6610 { '\'', ' ' },
6611 { '/', ' ' },
6612 };
6613 chtype symbol = get_rev_graph_symbol(graph);
6614 struct rev_filler *filler;
6615 size_t i;
6617 fillers[DEFAULT].line = opt_line_graphics ? ACS_VLINE : '|';
6618 filler = &fillers[DEFAULT];
6620 for (i = 0; i < graph->pos; i++) {
6621 append_to_rev_graph(graph, filler->line);
6622 if (graph_parent_is_merge(graph->prev) &&
6623 graph->prev->pos == i)
6624 filler = &fillers[RSHARP];
6626 append_to_rev_graph(graph, filler->separator);
6627 }
6629 /* Place the symbol for this revision. */
6630 append_to_rev_graph(graph, symbol);
6632 if (graph->prev->size > graph->size)
6633 filler = &fillers[RDIAG];
6634 else
6635 filler = &fillers[DEFAULT];
6637 i++;
6639 for (; i < graph->size; i++) {
6640 append_to_rev_graph(graph, filler->separator);
6641 append_to_rev_graph(graph, filler->line);
6642 if (graph_parent_is_merge(graph->prev) &&
6643 i < graph->prev->pos + graph->parents->size)
6644 filler = &fillers[RSHARP];
6645 if (graph->prev->size > graph->size)
6646 filler = &fillers[LDIAG];
6647 }
6649 if (graph->prev->size > graph->size) {
6650 append_to_rev_graph(graph, filler->separator);
6651 if (filler->line != ' ')
6652 append_to_rev_graph(graph, filler->line);
6653 }
6654 }
6656 /* Prepare the next rev graph */
6657 static void
6658 prepare_rev_graph(struct rev_graph *graph)
6659 {
6660 size_t i;
6662 /* First, traverse all lines of revisions up to the active one. */
6663 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
6664 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
6665 break;
6667 push_rev_graph(graph->next, graph->rev[graph->pos]);
6668 }
6670 /* Interleave the new revision parent(s). */
6671 for (i = 0; !graph->boundary && i < graph->parents->size; i++)
6672 push_rev_graph(graph->next, graph->parents->rev[i]);
6674 /* Lastly, put any remaining revisions. */
6675 for (i = graph->pos + 1; i < graph->size; i++)
6676 push_rev_graph(graph->next, graph->rev[i]);
6677 }
6679 static void
6680 update_rev_graph(struct view *view, struct rev_graph *graph)
6681 {
6682 /* If this is the finalizing update ... */
6683 if (graph->commit)
6684 prepare_rev_graph(graph);
6686 /* Graph visualization needs a one rev look-ahead,
6687 * so the first update doesn't visualize anything. */
6688 if (!graph->prev->commit)
6689 return;
6691 if (view->lines > 2)
6692 view->line[view->lines - 3].dirty = 1;
6693 if (view->lines > 1)
6694 view->line[view->lines - 2].dirty = 1;
6695 draw_rev_graph(graph->prev);
6696 done_rev_graph(graph->prev->prev);
6697 }
6700 /*
6701 * Main view backend
6702 */
6704 static const char *main_argv[SIZEOF_ARG] = {
6705 "git", "log", "--no-color", "--pretty=raw", "--parents",
6706 "--topo-order", "%(head)", NULL
6707 };
6709 static bool
6710 main_draw(struct view *view, struct line *line, unsigned int lineno)
6711 {
6712 struct commit *commit = line->data;
6714 if (!commit->author)
6715 return FALSE;
6717 if (opt_date && draw_date(view, &commit->time))
6718 return TRUE;
6720 if (opt_author && draw_author(view, commit->author))
6721 return TRUE;
6723 if (opt_rev_graph && commit->graph_size &&
6724 draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
6725 return TRUE;
6727 if (opt_show_refs && commit->refs) {
6728 size_t i;
6730 for (i = 0; i < commit->refs->size; i++) {
6731 struct ref *ref = commit->refs->refs[i];
6732 enum line_type type;
6734 if (ref->head)
6735 type = LINE_MAIN_HEAD;
6736 else if (ref->ltag)
6737 type = LINE_MAIN_LOCAL_TAG;
6738 else if (ref->tag)
6739 type = LINE_MAIN_TAG;
6740 else if (ref->tracked)
6741 type = LINE_MAIN_TRACKED;
6742 else if (ref->remote)
6743 type = LINE_MAIN_REMOTE;
6744 else
6745 type = LINE_MAIN_REF;
6747 if (draw_text(view, type, "[", TRUE) ||
6748 draw_text(view, type, ref->name, TRUE) ||
6749 draw_text(view, type, "]", TRUE))
6750 return TRUE;
6752 if (draw_text(view, LINE_DEFAULT, " ", TRUE))
6753 return TRUE;
6754 }
6755 }
6757 draw_text(view, LINE_DEFAULT, commit->title, TRUE);
6758 return TRUE;
6759 }
6761 /* Reads git log --pretty=raw output and parses it into the commit struct. */
6762 static bool
6763 main_read(struct view *view, char *line)
6764 {
6765 static struct rev_graph *graph = graph_stacks;
6766 enum line_type type;
6767 struct commit *commit;
6769 if (!line) {
6770 int i;
6772 if (!view->lines && !view->prev)
6773 die("No revisions match the given arguments.");
6774 if (view->lines > 0) {
6775 commit = view->line[view->lines - 1].data;
6776 view->line[view->lines - 1].dirty = 1;
6777 if (!commit->author) {
6778 view->lines--;
6779 free(commit);
6780 graph->commit = NULL;
6781 }
6782 }
6783 update_rev_graph(view, graph);
6785 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
6786 clear_rev_graph(&graph_stacks[i]);
6787 return TRUE;
6788 }
6790 type = get_line_type(line);
6791 if (type == LINE_COMMIT) {
6792 commit = calloc(1, sizeof(struct commit));
6793 if (!commit)
6794 return FALSE;
6796 line += STRING_SIZE("commit ");
6797 if (*line == '-') {
6798 graph->boundary = 1;
6799 line++;
6800 }
6802 string_copy_rev(commit->id, line);
6803 commit->refs = get_ref_list(commit->id);
6804 graph->commit = commit;
6805 add_line_data(view, commit, LINE_MAIN_COMMIT);
6807 while ((line = strchr(line, ' '))) {
6808 line++;
6809 push_rev_graph(graph->parents, line);
6810 commit->has_parents = TRUE;
6811 }
6812 return TRUE;
6813 }
6815 if (!view->lines)
6816 return TRUE;
6817 commit = view->line[view->lines - 1].data;
6819 switch (type) {
6820 case LINE_PARENT:
6821 if (commit->has_parents)
6822 break;
6823 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
6824 break;
6826 case LINE_AUTHOR:
6827 parse_author_line(line + STRING_SIZE("author "),
6828 &commit->author, &commit->time);
6829 update_rev_graph(view, graph);
6830 graph = graph->next;
6831 break;
6833 default:
6834 /* Fill in the commit title if it has not already been set. */
6835 if (commit->title[0])
6836 break;
6838 /* Require titles to start with a non-space character at the
6839 * offset used by git log. */
6840 if (strncmp(line, " ", 4))
6841 break;
6842 line += 4;
6843 /* Well, if the title starts with a whitespace character,
6844 * try to be forgiving. Otherwise we end up with no title. */
6845 while (isspace(*line))
6846 line++;
6847 if (*line == '\0')
6848 break;
6849 /* FIXME: More graceful handling of titles; append "..." to
6850 * shortened titles, etc. */
6852 string_expand(commit->title, sizeof(commit->title), line, 1);
6853 view->line[view->lines - 1].dirty = 1;
6854 }
6856 return TRUE;
6857 }
6859 static enum request
6860 main_request(struct view *view, enum request request, struct line *line)
6861 {
6862 enum open_flags flags = view_is_displayed(view) ? OPEN_SPLIT : OPEN_DEFAULT;
6864 switch (request) {
6865 case REQ_ENTER:
6866 open_view(view, REQ_VIEW_DIFF, flags);
6867 break;
6868 case REQ_REFRESH:
6869 load_refs();
6870 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
6871 break;
6872 default:
6873 return request;
6874 }
6876 return REQ_NONE;
6877 }
6879 static bool
6880 grep_refs(struct ref_list *list, regex_t *regex)
6881 {
6882 regmatch_t pmatch;
6883 size_t i;
6885 if (!opt_show_refs || !list)
6886 return FALSE;
6888 for (i = 0; i < list->size; i++) {
6889 if (regexec(regex, list->refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
6890 return TRUE;
6891 }
6893 return FALSE;
6894 }
6896 static bool
6897 main_grep(struct view *view, struct line *line)
6898 {
6899 struct commit *commit = line->data;
6900 const char *text[] = {
6901 commit->title,
6902 opt_author ? commit->author : "",
6903 mkdate(&commit->time, opt_date),
6904 NULL
6905 };
6907 return grep_text(view, text) || grep_refs(commit->refs, view->regex);
6908 }
6910 static void
6911 main_select(struct view *view, struct line *line)
6912 {
6913 struct commit *commit = line->data;
6915 string_copy_rev(view->ref, commit->id);
6916 string_copy_rev(ref_commit, view->ref);
6917 }
6919 static struct view_ops main_ops = {
6920 "commit",
6921 main_argv,
6922 NULL,
6923 main_read,
6924 main_draw,
6925 main_request,
6926 main_grep,
6927 main_select,
6928 };
6931 /*
6932 * Status management
6933 */
6935 /* Whether or not the curses interface has been initialized. */
6936 static bool cursed = FALSE;
6938 /* Terminal hacks and workarounds. */
6939 static bool use_scroll_redrawwin;
6940 static bool use_scroll_status_wclear;
6942 /* The status window is used for polling keystrokes. */
6943 static WINDOW *status_win;
6945 /* Reading from the prompt? */
6946 static bool input_mode = FALSE;
6948 static bool status_empty = FALSE;
6950 /* Update status and title window. */
6951 static void
6952 report(const char *msg, ...)
6953 {
6954 struct view *view = display[current_view];
6956 if (input_mode)
6957 return;
6959 if (!view) {
6960 char buf[SIZEOF_STR];
6961 va_list args;
6963 va_start(args, msg);
6964 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
6965 buf[sizeof(buf) - 1] = 0;
6966 buf[sizeof(buf) - 2] = '.';
6967 buf[sizeof(buf) - 3] = '.';
6968 buf[sizeof(buf) - 4] = '.';
6969 }
6970 va_end(args);
6971 die("%s", buf);
6972 }
6974 if (!status_empty || *msg) {
6975 va_list args;
6977 va_start(args, msg);
6979 wmove(status_win, 0, 0);
6980 if (view->has_scrolled && use_scroll_status_wclear)
6981 wclear(status_win);
6982 if (*msg) {
6983 vwprintw(status_win, msg, args);
6984 status_empty = FALSE;
6985 } else {
6986 status_empty = TRUE;
6987 }
6988 wclrtoeol(status_win);
6989 wnoutrefresh(status_win);
6991 va_end(args);
6992 }
6994 update_view_title(view);
6995 }
6997 static void
6998 init_display(void)
6999 {
7000 const char *term;
7001 int x, y;
7003 /* Initialize the curses library */
7004 if (isatty(STDIN_FILENO)) {
7005 cursed = !!initscr();
7006 opt_tty = stdin;
7007 } else {
7008 /* Leave stdin and stdout alone when acting as a pager. */
7009 opt_tty = fopen("/dev/tty", "r+");
7010 if (!opt_tty)
7011 die("Failed to open /dev/tty");
7012 cursed = !!newterm(NULL, opt_tty, opt_tty);
7013 }
7015 if (!cursed)
7016 die("Failed to initialize curses");
7018 nonl(); /* Disable conversion and detect newlines from input. */
7019 cbreak(); /* Take input chars one at a time, no wait for \n */
7020 noecho(); /* Don't echo input */
7021 leaveok(stdscr, FALSE);
7023 if (has_colors())
7024 init_colors();
7026 getmaxyx(stdscr, y, x);
7027 status_win = newwin(1, 0, y - 1, 0);
7028 if (!status_win)
7029 die("Failed to create status window");
7031 /* Enable keyboard mapping */
7032 keypad(status_win, TRUE);
7033 wbkgdset(status_win, get_line_attr(LINE_STATUS));
7035 TABSIZE = opt_tab_size;
7037 term = getenv("XTERM_VERSION") ? NULL : getenv("COLORTERM");
7038 if (term && !strcmp(term, "gnome-terminal")) {
7039 /* In the gnome-terminal-emulator, the message from
7040 * scrolling up one line when impossible followed by
7041 * scrolling down one line causes corruption of the
7042 * status line. This is fixed by calling wclear. */
7043 use_scroll_status_wclear = TRUE;
7044 use_scroll_redrawwin = FALSE;
7046 } else if (term && !strcmp(term, "xrvt-xpm")) {
7047 /* No problems with full optimizations in xrvt-(unicode)
7048 * and aterm. */
7049 use_scroll_status_wclear = use_scroll_redrawwin = FALSE;
7051 } else {
7052 /* When scrolling in (u)xterm the last line in the
7053 * scrolling direction will update slowly. */
7054 use_scroll_redrawwin = TRUE;
7055 use_scroll_status_wclear = FALSE;
7056 }
7057 }
7059 static int
7060 get_input(int prompt_position)
7061 {
7062 struct view *view;
7063 int i, key, cursor_y, cursor_x;
7064 bool loading = FALSE;
7066 if (prompt_position)
7067 input_mode = TRUE;
7069 while (TRUE) {
7070 foreach_view (view, i) {
7071 update_view(view);
7072 if (view_is_displayed(view) && view->has_scrolled &&
7073 use_scroll_redrawwin)
7074 redrawwin(view->win);
7075 view->has_scrolled = FALSE;
7076 if (view->pipe)
7077 loading = TRUE;
7078 }
7080 /* Update the cursor position. */
7081 if (prompt_position) {
7082 getbegyx(status_win, cursor_y, cursor_x);
7083 cursor_x = prompt_position;
7084 } else {
7085 view = display[current_view];
7086 getbegyx(view->win, cursor_y, cursor_x);
7087 cursor_x = view->width - 1;
7088 cursor_y += view->lineno - view->offset;
7089 }
7090 setsyx(cursor_y, cursor_x);
7092 /* Refresh, accept single keystroke of input */
7093 doupdate();
7094 nodelay(status_win, loading);
7095 key = wgetch(status_win);
7097 /* wgetch() with nodelay() enabled returns ERR when
7098 * there's no input. */
7099 if (key == ERR) {
7101 } else if (key == KEY_RESIZE) {
7102 int height, width;
7104 getmaxyx(stdscr, height, width);
7106 wresize(status_win, 1, width);
7107 mvwin(status_win, height - 1, 0);
7108 wnoutrefresh(status_win);
7109 resize_display();
7110 redraw_display(TRUE);
7112 } else {
7113 input_mode = FALSE;
7114 return key;
7115 }
7116 }
7117 }
7119 static char *
7120 prompt_input(const char *prompt, input_handler handler, void *data)
7121 {
7122 enum input_status status = INPUT_OK;
7123 static char buf[SIZEOF_STR];
7124 size_t pos = 0;
7126 buf[pos] = 0;
7128 while (status == INPUT_OK || status == INPUT_SKIP) {
7129 int key;
7131 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
7132 wclrtoeol(status_win);
7134 key = get_input(pos + 1);
7135 switch (key) {
7136 case KEY_RETURN:
7137 case KEY_ENTER:
7138 case '\n':
7139 status = pos ? INPUT_STOP : INPUT_CANCEL;
7140 break;
7142 case KEY_BACKSPACE:
7143 if (pos > 0)
7144 buf[--pos] = 0;
7145 else
7146 status = INPUT_CANCEL;
7147 break;
7149 case KEY_ESC:
7150 status = INPUT_CANCEL;
7151 break;
7153 default:
7154 if (pos >= sizeof(buf)) {
7155 report("Input string too long");
7156 return NULL;
7157 }
7159 status = handler(data, buf, key);
7160 if (status == INPUT_OK)
7161 buf[pos++] = (char) key;
7162 }
7163 }
7165 /* Clear the status window */
7166 status_empty = FALSE;
7167 report("");
7169 if (status == INPUT_CANCEL)
7170 return NULL;
7172 buf[pos++] = 0;
7174 return buf;
7175 }
7177 static enum input_status
7178 prompt_yesno_handler(void *data, char *buf, int c)
7179 {
7180 if (c == 'y' || c == 'Y')
7181 return INPUT_STOP;
7182 if (c == 'n' || c == 'N')
7183 return INPUT_CANCEL;
7184 return INPUT_SKIP;
7185 }
7187 static bool
7188 prompt_yesno(const char *prompt)
7189 {
7190 char prompt2[SIZEOF_STR];
7192 if (!string_format(prompt2, "%s [Yy/Nn]", prompt))
7193 return FALSE;
7195 return !!prompt_input(prompt2, prompt_yesno_handler, NULL);
7196 }
7198 static enum input_status
7199 read_prompt_handler(void *data, char *buf, int c)
7200 {
7201 return isprint(c) ? INPUT_OK : INPUT_SKIP;
7202 }
7204 static char *
7205 read_prompt(const char *prompt)
7206 {
7207 return prompt_input(prompt, read_prompt_handler, NULL);
7208 }
7210 static bool prompt_menu(const char *prompt, const struct menu_item *items, int *selected)
7211 {
7212 enum input_status status = INPUT_OK;
7213 int size = 0;
7215 while (items[size].text)
7216 size++;
7218 while (status == INPUT_OK) {
7219 const struct menu_item *item = &items[*selected];
7220 int key;
7221 int i;
7223 mvwprintw(status_win, 0, 0, "%s (%d of %d) ",
7224 prompt, *selected + 1, size);
7225 if (item->hotkey)
7226 wprintw(status_win, "[%c] ", (char) item->hotkey);
7227 wprintw(status_win, "%s", item->text);
7228 wclrtoeol(status_win);
7230 key = get_input(COLS - 1);
7231 switch (key) {
7232 case KEY_RETURN:
7233 case KEY_ENTER:
7234 case '\n':
7235 status = INPUT_STOP;
7236 break;
7238 case KEY_LEFT:
7239 case KEY_UP:
7240 *selected = *selected - 1;
7241 if (*selected < 0)
7242 *selected = size - 1;
7243 break;
7245 case KEY_RIGHT:
7246 case KEY_DOWN:
7247 *selected = (*selected + 1) % size;
7248 break;
7250 case KEY_ESC:
7251 status = INPUT_CANCEL;
7252 break;
7254 default:
7255 for (i = 0; items[i].text; i++)
7256 if (items[i].hotkey == key) {
7257 *selected = i;
7258 status = INPUT_STOP;
7259 break;
7260 }
7261 }
7262 }
7264 /* Clear the status window */
7265 status_empty = FALSE;
7266 report("");
7268 return status != INPUT_CANCEL;
7269 }
7271 /*
7272 * Repository properties
7273 */
7275 static struct ref **refs = NULL;
7276 static size_t refs_size = 0;
7277 static struct ref *refs_head = NULL;
7279 static struct ref_list **ref_lists = NULL;
7280 static size_t ref_lists_size = 0;
7282 DEFINE_ALLOCATOR(realloc_refs, struct ref *, 256)
7283 DEFINE_ALLOCATOR(realloc_refs_list, struct ref *, 8)
7284 DEFINE_ALLOCATOR(realloc_ref_lists, struct ref_list *, 8)
7286 static int
7287 compare_refs(const void *ref1_, const void *ref2_)
7288 {
7289 const struct ref *ref1 = *(const struct ref **)ref1_;
7290 const struct ref *ref2 = *(const struct ref **)ref2_;
7292 if (ref1->tag != ref2->tag)
7293 return ref2->tag - ref1->tag;
7294 if (ref1->ltag != ref2->ltag)
7295 return ref2->ltag - ref2->ltag;
7296 if (ref1->head != ref2->head)
7297 return ref2->head - ref1->head;
7298 if (ref1->tracked != ref2->tracked)
7299 return ref2->tracked - ref1->tracked;
7300 if (ref1->remote != ref2->remote)
7301 return ref2->remote - ref1->remote;
7302 return strcmp(ref1->name, ref2->name);
7303 }
7305 static void
7306 foreach_ref(bool (*visitor)(void *data, const struct ref *ref), void *data)
7307 {
7308 size_t i;
7310 for (i = 0; i < refs_size; i++)
7311 if (!visitor(data, refs[i]))
7312 break;
7313 }
7315 static struct ref *
7316 get_ref_head()
7317 {
7318 return refs_head;
7319 }
7321 static struct ref_list *
7322 get_ref_list(const char *id)
7323 {
7324 struct ref_list *list;
7325 size_t i;
7327 for (i = 0; i < ref_lists_size; i++)
7328 if (!strcmp(id, ref_lists[i]->id))
7329 return ref_lists[i];
7331 if (!realloc_ref_lists(&ref_lists, ref_lists_size, 1))
7332 return NULL;
7333 list = calloc(1, sizeof(*list));
7334 if (!list)
7335 return NULL;
7337 for (i = 0; i < refs_size; i++) {
7338 if (!strcmp(id, refs[i]->id) &&
7339 realloc_refs_list(&list->refs, list->size, 1))
7340 list->refs[list->size++] = refs[i];
7341 }
7343 if (!list->refs) {
7344 free(list);
7345 return NULL;
7346 }
7348 qsort(list->refs, list->size, sizeof(*list->refs), compare_refs);
7349 ref_lists[ref_lists_size++] = list;
7350 return list;
7351 }
7353 static int
7354 read_ref(char *id, size_t idlen, char *name, size_t namelen)
7355 {
7356 struct ref *ref = NULL;
7357 bool tag = FALSE;
7358 bool ltag = FALSE;
7359 bool remote = FALSE;
7360 bool tracked = FALSE;
7361 bool head = FALSE;
7362 int from = 0, to = refs_size - 1;
7364 if (!prefixcmp(name, "refs/tags/")) {
7365 if (!suffixcmp(name, namelen, "^{}")) {
7366 namelen -= 3;
7367 name[namelen] = 0;
7368 } else {
7369 ltag = TRUE;
7370 }
7372 tag = TRUE;
7373 namelen -= STRING_SIZE("refs/tags/");
7374 name += STRING_SIZE("refs/tags/");
7376 } else if (!prefixcmp(name, "refs/remotes/")) {
7377 remote = TRUE;
7378 namelen -= STRING_SIZE("refs/remotes/");
7379 name += STRING_SIZE("refs/remotes/");
7380 tracked = !strcmp(opt_remote, name);
7382 } else if (!prefixcmp(name, "refs/heads/")) {
7383 namelen -= STRING_SIZE("refs/heads/");
7384 name += STRING_SIZE("refs/heads/");
7385 if (!strncmp(opt_head, name, namelen))
7386 return OK;
7388 } else if (!strcmp(name, "HEAD")) {
7389 head = TRUE;
7390 if (*opt_head) {
7391 namelen = strlen(opt_head);
7392 name = opt_head;
7393 }
7394 }
7396 /* If we are reloading or it's an annotated tag, replace the
7397 * previous SHA1 with the resolved commit id; relies on the fact
7398 * git-ls-remote lists the commit id of an annotated tag right
7399 * before the commit id it points to. */
7400 while (from <= to) {
7401 size_t pos = (to + from) / 2;
7402 int cmp = strcmp(name, refs[pos]->name);
7404 if (!cmp) {
7405 ref = refs[pos];
7406 break;
7407 }
7409 if (cmp < 0)
7410 to = pos - 1;
7411 else
7412 from = pos + 1;
7413 }
7415 if (!ref) {
7416 if (!realloc_refs(&refs, refs_size, 1))
7417 return ERR;
7418 ref = calloc(1, sizeof(*ref) + namelen);
7419 if (!ref)
7420 return ERR;
7421 memmove(refs + from + 1, refs + from,
7422 (refs_size - from) * sizeof(*refs));
7423 refs[from] = ref;
7424 strncpy(ref->name, name, namelen);
7425 refs_size++;
7426 }
7428 ref->head = head;
7429 ref->tag = tag;
7430 ref->ltag = ltag;
7431 ref->remote = remote;
7432 ref->tracked = tracked;
7433 string_copy_rev(ref->id, id);
7435 if (head)
7436 refs_head = ref;
7437 return OK;
7438 }
7440 static int
7441 load_refs(void)
7442 {
7443 const char *head_argv[] = {
7444 "git", "symbolic-ref", "HEAD", NULL
7445 };
7446 static const char *ls_remote_argv[SIZEOF_ARG] = {
7447 "git", "ls-remote", opt_git_dir, NULL
7448 };
7449 static bool init = FALSE;
7450 size_t i;
7452 if (!init) {
7453 if (!argv_from_env(ls_remote_argv, "TIG_LS_REMOTE"))
7454 die("TIG_LS_REMOTE contains too many arguments");
7455 init = TRUE;
7456 }
7458 if (!*opt_git_dir)
7459 return OK;
7461 if (io_run_buf(head_argv, opt_head, sizeof(opt_head)) &&
7462 !prefixcmp(opt_head, "refs/heads/")) {
7463 char *offset = opt_head + STRING_SIZE("refs/heads/");
7465 memmove(opt_head, offset, strlen(offset) + 1);
7466 }
7468 refs_head = NULL;
7469 for (i = 0; i < refs_size; i++)
7470 refs[i]->id[0] = 0;
7472 if (io_run_load(ls_remote_argv, "\t", read_ref) == ERR)
7473 return ERR;
7475 /* Update the ref lists to reflect changes. */
7476 for (i = 0; i < ref_lists_size; i++) {
7477 struct ref_list *list = ref_lists[i];
7478 size_t old, new;
7480 for (old = new = 0; old < list->size; old++)
7481 if (!strcmp(list->id, list->refs[old]->id))
7482 list->refs[new++] = list->refs[old];
7483 list->size = new;
7484 }
7486 return OK;
7487 }
7489 static void
7490 set_remote_branch(const char *name, const char *value, size_t valuelen)
7491 {
7492 if (!strcmp(name, ".remote")) {
7493 string_ncopy(opt_remote, value, valuelen);
7495 } else if (*opt_remote && !strcmp(name, ".merge")) {
7496 size_t from = strlen(opt_remote);
7498 if (!prefixcmp(value, "refs/heads/"))
7499 value += STRING_SIZE("refs/heads/");
7501 if (!string_format_from(opt_remote, &from, "/%s", value))
7502 opt_remote[0] = 0;
7503 }
7504 }
7506 static void
7507 set_repo_config_option(char *name, char *value, int (*cmd)(int, const char **))
7508 {
7509 const char *argv[SIZEOF_ARG] = { name, "=" };
7510 int argc = 1 + (cmd == option_set_command);
7511 int error = ERR;
7513 if (!argv_from_string(argv, &argc, value))
7514 config_msg = "Too many option arguments";
7515 else
7516 error = cmd(argc, argv);
7518 if (error == ERR)
7519 warn("Option 'tig.%s': %s", name, config_msg);
7520 }
7522 static bool
7523 set_environment_variable(const char *name, const char *value)
7524 {
7525 size_t len = strlen(name) + 1 + strlen(value) + 1;
7526 char *env = malloc(len);
7528 if (env &&
7529 string_nformat(env, len, NULL, "%s=%s", name, value) &&
7530 putenv(env) == 0)
7531 return TRUE;
7532 free(env);
7533 return FALSE;
7534 }
7536 static void
7537 set_work_tree(const char *value)
7538 {
7539 char cwd[SIZEOF_STR];
7541 if (!getcwd(cwd, sizeof(cwd)))
7542 die("Failed to get cwd path: %s", strerror(errno));
7543 if (chdir(opt_git_dir) < 0)
7544 die("Failed to chdir(%s): %s", strerror(errno));
7545 if (!getcwd(opt_git_dir, sizeof(opt_git_dir)))
7546 die("Failed to get git path: %s", strerror(errno));
7547 if (chdir(cwd) < 0)
7548 die("Failed to chdir(%s): %s", cwd, strerror(errno));
7549 if (chdir(value) < 0)
7550 die("Failed to chdir(%s): %s", value, strerror(errno));
7551 if (!getcwd(cwd, sizeof(cwd)))
7552 die("Failed to get cwd path: %s", strerror(errno));
7553 if (!set_environment_variable("GIT_WORK_TREE", cwd))
7554 die("Failed to set GIT_WORK_TREE to '%s'", cwd);
7555 if (!set_environment_variable("GIT_DIR", opt_git_dir))
7556 die("Failed to set GIT_DIR to '%s'", opt_git_dir);
7557 opt_is_inside_work_tree = TRUE;
7558 }
7560 static int
7561 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
7562 {
7563 if (!strcmp(name, "i18n.commitencoding"))
7564 string_ncopy(opt_encoding, value, valuelen);
7566 else if (!strcmp(name, "core.editor"))
7567 string_ncopy(opt_editor, value, valuelen);
7569 else if (!strcmp(name, "core.worktree"))
7570 set_work_tree(value);
7572 else if (!prefixcmp(name, "tig.color."))
7573 set_repo_config_option(name + 10, value, option_color_command);
7575 else if (!prefixcmp(name, "tig.bind."))
7576 set_repo_config_option(name + 9, value, option_bind_command);
7578 else if (!prefixcmp(name, "tig."))
7579 set_repo_config_option(name + 4, value, option_set_command);
7581 else if (*opt_head && !prefixcmp(name, "branch.") &&
7582 !strncmp(name + 7, opt_head, strlen(opt_head)))
7583 set_remote_branch(name + 7 + strlen(opt_head), value, valuelen);
7585 return OK;
7586 }
7588 static int
7589 load_git_config(void)
7590 {
7591 const char *config_list_argv[] = { "git", "config", "--list", NULL };
7593 return io_run_load(config_list_argv, "=", read_repo_config_option);
7594 }
7596 static int
7597 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
7598 {
7599 if (!opt_git_dir[0]) {
7600 string_ncopy(opt_git_dir, name, namelen);
7602 } else if (opt_is_inside_work_tree == -1) {
7603 /* This can be 3 different values depending on the
7604 * version of git being used. If git-rev-parse does not
7605 * understand --is-inside-work-tree it will simply echo
7606 * the option else either "true" or "false" is printed.
7607 * Default to true for the unknown case. */
7608 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
7610 } else if (*name == '.') {
7611 string_ncopy(opt_cdup, name, namelen);
7613 } else {
7614 string_ncopy(opt_prefix, name, namelen);
7615 }
7617 return OK;
7618 }
7620 static int
7621 load_repo_info(void)
7622 {
7623 const char *rev_parse_argv[] = {
7624 "git", "rev-parse", "--git-dir", "--is-inside-work-tree",
7625 "--show-cdup", "--show-prefix", NULL
7626 };
7628 return io_run_load(rev_parse_argv, "=", read_repo_info);
7629 }
7632 /*
7633 * Main
7634 */
7636 static const char usage[] =
7637 "tig " TIG_VERSION " (" __DATE__ ")\n"
7638 "\n"
7639 "Usage: tig [options] [revs] [--] [paths]\n"
7640 " or: tig show [options] [revs] [--] [paths]\n"
7641 " or: tig blame [rev] path\n"
7642 " or: tig status\n"
7643 " or: tig < [git command output]\n"
7644 "\n"
7645 "Options:\n"
7646 " -v, --version Show version and exit\n"
7647 " -h, --help Show help message and exit";
7649 static void __NORETURN
7650 quit(int sig)
7651 {
7652 /* XXX: Restore tty modes and let the OS cleanup the rest! */
7653 if (cursed)
7654 endwin();
7655 exit(0);
7656 }
7658 static void __NORETURN
7659 die(const char *err, ...)
7660 {
7661 va_list args;
7663 endwin();
7665 va_start(args, err);
7666 fputs("tig: ", stderr);
7667 vfprintf(stderr, err, args);
7668 fputs("\n", stderr);
7669 va_end(args);
7671 exit(1);
7672 }
7674 static void
7675 warn(const char *msg, ...)
7676 {
7677 va_list args;
7679 va_start(args, msg);
7680 fputs("tig warning: ", stderr);
7681 vfprintf(stderr, msg, args);
7682 fputs("\n", stderr);
7683 va_end(args);
7684 }
7686 static enum request
7687 parse_options(int argc, const char *argv[])
7688 {
7689 enum request request = REQ_VIEW_MAIN;
7690 const char *subcommand;
7691 bool seen_dashdash = FALSE;
7692 /* XXX: This is vulnerable to the user overriding options
7693 * required for the main view parser. */
7694 const char *custom_argv[SIZEOF_ARG] = {
7695 "git", "log", "--no-color", "--pretty=raw", "--parents",
7696 "--topo-order", NULL
7697 };
7698 int i, j = 6;
7700 if (!isatty(STDIN_FILENO)) {
7701 io_open(&VIEW(REQ_VIEW_PAGER)->io, "");
7702 return REQ_VIEW_PAGER;
7703 }
7705 if (argc <= 1)
7706 return REQ_NONE;
7708 subcommand = argv[1];
7709 if (!strcmp(subcommand, "status")) {
7710 if (argc > 2)
7711 warn("ignoring arguments after `%s'", subcommand);
7712 return REQ_VIEW_STATUS;
7714 } else if (!strcmp(subcommand, "blame")) {
7715 if (argc <= 2 || argc > 4)
7716 die("invalid number of options to blame\n\n%s", usage);
7718 i = 2;
7719 if (argc == 4) {
7720 string_ncopy(opt_ref, argv[i], strlen(argv[i]));
7721 i++;
7722 }
7724 string_ncopy(opt_file, argv[i], strlen(argv[i]));
7725 return REQ_VIEW_BLAME;
7727 } else if (!strcmp(subcommand, "show")) {
7728 request = REQ_VIEW_DIFF;
7730 } else {
7731 subcommand = NULL;
7732 }
7734 if (subcommand) {
7735 custom_argv[1] = subcommand;
7736 j = 2;
7737 }
7739 for (i = 1 + !!subcommand; i < argc; i++) {
7740 const char *opt = argv[i];
7742 if (seen_dashdash || !strcmp(opt, "--")) {
7743 seen_dashdash = TRUE;
7745 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
7746 printf("tig version %s\n", TIG_VERSION);
7747 quit(0);
7749 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
7750 printf("%s\n", usage);
7751 quit(0);
7752 }
7754 custom_argv[j++] = opt;
7755 if (j >= ARRAY_SIZE(custom_argv))
7756 die("command too long");
7757 }
7759 if (!prepare_update(VIEW(request), custom_argv, NULL))
7760 die("Failed to format arguments");
7762 return request;
7763 }
7765 int
7766 main(int argc, const char *argv[])
7767 {
7768 const char *codeset = "UTF-8";
7769 enum request request = parse_options(argc, argv);
7770 struct view *view;
7771 size_t i;
7773 signal(SIGINT, quit);
7774 signal(SIGPIPE, SIG_IGN);
7776 if (setlocale(LC_ALL, "")) {
7777 codeset = nl_langinfo(CODESET);
7778 }
7780 if (load_repo_info() == ERR)
7781 die("Failed to load repo info.");
7783 if (load_options() == ERR)
7784 die("Failed to load user config.");
7786 if (load_git_config() == ERR)
7787 die("Failed to load repo config.");
7789 /* Require a git repository unless when running in pager mode. */
7790 if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
7791 die("Not a git repository");
7793 if (*opt_encoding && strcmp(codeset, "UTF-8")) {
7794 opt_iconv_in = iconv_open("UTF-8", opt_encoding);
7795 if (opt_iconv_in == ICONV_NONE)
7796 die("Failed to initialize character set conversion");
7797 }
7799 if (codeset && strcmp(codeset, "UTF-8")) {
7800 opt_iconv_out = iconv_open(codeset, "UTF-8");
7801 if (opt_iconv_out == ICONV_NONE)
7802 die("Failed to initialize character set conversion");
7803 }
7805 if (load_refs() == ERR)
7806 die("Failed to load refs.");
7808 foreach_view (view, i)
7809 if (!argv_from_env(view->ops->argv, view->cmd_env))
7810 die("Too many arguments in the `%s` environment variable",
7811 view->cmd_env);
7813 init_display();
7815 if (request != REQ_NONE)
7816 open_view(NULL, request, OPEN_PREPARED);
7817 request = request == REQ_NONE ? REQ_VIEW_MAIN : REQ_NONE;
7819 while (view_driver(display[current_view], request)) {
7820 int key = get_input(0);
7822 view = display[current_view];
7823 request = get_keybinding(view->keymap, key);
7825 /* Some low-level request handling. This keeps access to
7826 * status_win restricted. */
7827 switch (request) {
7828 case REQ_NONE:
7829 report("Unknown key, press %s for help",
7830 get_key(view->keymap, REQ_VIEW_HELP));
7831 break;
7832 case REQ_PROMPT:
7833 {
7834 char *cmd = read_prompt(":");
7836 if (cmd && isdigit(*cmd)) {
7837 int lineno = view->lineno + 1;
7839 if (parse_int(&lineno, cmd, 1, view->lines + 1) == OK) {
7840 select_view_line(view, lineno - 1);
7841 report("");
7842 } else {
7843 report("Unable to parse '%s' as a line number", cmd);
7844 }
7846 } else if (cmd) {
7847 struct view *next = VIEW(REQ_VIEW_PAGER);
7848 const char *argv[SIZEOF_ARG] = { "git" };
7849 int argc = 1;
7851 /* When running random commands, initially show the
7852 * command in the title. However, it maybe later be
7853 * overwritten if a commit line is selected. */
7854 string_ncopy(next->ref, cmd, strlen(cmd));
7856 if (!argv_from_string(argv, &argc, cmd)) {
7857 report("Too many arguments");
7858 } else if (!prepare_update(next, argv, NULL)) {
7859 report("Failed to format command");
7860 } else {
7861 open_view(view, REQ_VIEW_PAGER, OPEN_PREPARED);
7862 }
7863 }
7865 request = REQ_NONE;
7866 break;
7867 }
7868 case REQ_SEARCH:
7869 case REQ_SEARCH_BACK:
7870 {
7871 const char *prompt = request == REQ_SEARCH ? "/" : "?";
7872 char *search = read_prompt(prompt);
7874 if (search)
7875 string_ncopy(opt_search, search, strlen(search));
7876 else if (*opt_search)
7877 request = request == REQ_SEARCH ?
7878 REQ_FIND_NEXT :
7879 REQ_FIND_PREV;
7880 else
7881 request = REQ_NONE;
7882 break;
7883 }
7884 default:
7885 break;
7886 }
7887 }
7889 quit(0);
7891 return 0;
7892 }