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 format_flags {
144 FORMAT_ALL, /* Perform replacement in all arguments. */
145 FORMAT_NONE /* No replacement should be performed. */
146 };
148 static bool format_argv(const char *dst[], const char *src[], enum format_flags flags);
150 enum input_status {
151 INPUT_OK,
152 INPUT_SKIP,
153 INPUT_STOP,
154 INPUT_CANCEL
155 };
157 typedef enum input_status (*input_handler)(void *data, char *buf, int c);
159 static char *prompt_input(const char *prompt, input_handler handler, void *data);
160 static bool prompt_yesno(const char *prompt);
162 struct menu_item {
163 int hotkey;
164 const char *text;
165 void *data;
166 };
168 static bool prompt_menu(const char *prompt, const struct menu_item *items, int *selected);
170 /*
171 * Allocation helpers ... Entering macro hell to never be seen again.
172 */
174 #define DEFINE_ALLOCATOR(name, type, chunk_size) \
175 static type * \
176 name(type **mem, size_t size, size_t increase) \
177 { \
178 size_t num_chunks = (size + chunk_size - 1) / chunk_size; \
179 size_t num_chunks_new = (size + increase + chunk_size - 1) / chunk_size;\
180 type *tmp = *mem; \
181 \
182 if (mem == NULL || num_chunks != num_chunks_new) { \
183 tmp = realloc(tmp, num_chunks_new * chunk_size * sizeof(type)); \
184 if (tmp) \
185 *mem = tmp; \
186 } \
187 \
188 return tmp; \
189 }
191 /*
192 * String helpers
193 */
195 static inline void
196 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
197 {
198 if (srclen > dstlen - 1)
199 srclen = dstlen - 1;
201 strncpy(dst, src, srclen);
202 dst[srclen] = 0;
203 }
205 /* Shorthands for safely copying into a fixed buffer. */
207 #define string_copy(dst, src) \
208 string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
210 #define string_ncopy(dst, src, srclen) \
211 string_ncopy_do(dst, sizeof(dst), src, srclen)
213 #define string_copy_rev(dst, src) \
214 string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
216 #define string_add(dst, from, src) \
217 string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
219 static void
220 string_expand(char *dst, size_t dstlen, const char *src, int tabsize)
221 {
222 size_t size, pos;
224 for (size = pos = 0; size < dstlen - 1 && src[pos]; pos++) {
225 if (src[pos] == '\t') {
226 size_t expanded = tabsize - (size % tabsize);
228 if (expanded + size >= dstlen - 1)
229 expanded = dstlen - size - 1;
230 memcpy(dst + size, " ", expanded);
231 size += expanded;
232 } else {
233 dst[size++] = src[pos];
234 }
235 }
237 dst[size] = 0;
238 }
240 static char *
241 chomp_string(char *name)
242 {
243 int namelen;
245 while (isspace(*name))
246 name++;
248 namelen = strlen(name) - 1;
249 while (namelen > 0 && isspace(name[namelen]))
250 name[namelen--] = 0;
252 return name;
253 }
255 static bool
256 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
257 {
258 va_list args;
259 size_t pos = bufpos ? *bufpos : 0;
261 va_start(args, fmt);
262 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
263 va_end(args);
265 if (bufpos)
266 *bufpos = pos;
268 return pos >= bufsize ? FALSE : TRUE;
269 }
271 #define string_format(buf, fmt, args...) \
272 string_nformat(buf, sizeof(buf), NULL, fmt, args)
274 #define string_format_from(buf, from, fmt, args...) \
275 string_nformat(buf, sizeof(buf), from, fmt, args)
277 static int
278 string_enum_compare(const char *str1, const char *str2, int len)
279 {
280 size_t i;
282 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
284 /* Diff-Header == DIFF_HEADER */
285 for (i = 0; i < len; i++) {
286 if (toupper(str1[i]) == toupper(str2[i]))
287 continue;
289 if (string_enum_sep(str1[i]) &&
290 string_enum_sep(str2[i]))
291 continue;
293 return str1[i] - str2[i];
294 }
296 return 0;
297 }
299 #define enum_equals(entry, str, len) \
300 ((entry).namelen == (len) && !string_enum_compare((entry).name, str, len))
302 struct enum_map {
303 const char *name;
304 int namelen;
305 int value;
306 };
308 #define ENUM_MAP(name, value) { name, STRING_SIZE(name), value }
310 static char *
311 enum_map_name(const char *name, size_t namelen)
312 {
313 static char buf[SIZEOF_STR];
314 int bufpos;
316 for (bufpos = 0; bufpos <= namelen; bufpos++) {
317 buf[bufpos] = tolower(name[bufpos]);
318 if (buf[bufpos] == '_')
319 buf[bufpos] = '-';
320 }
322 buf[bufpos] = 0;
323 return buf;
324 }
326 #define enum_name(entry) enum_map_name((entry).name, (entry).namelen)
328 static bool
329 map_enum_do(const struct enum_map *map, size_t map_size, int *value, const char *name)
330 {
331 size_t namelen = strlen(name);
332 int i;
334 for (i = 0; i < map_size; i++)
335 if (enum_equals(map[i], name, namelen)) {
336 *value = map[i].value;
337 return TRUE;
338 }
340 return FALSE;
341 }
343 #define map_enum(attr, map, name) \
344 map_enum_do(map, ARRAY_SIZE(map), attr, name)
346 #define prefixcmp(str1, str2) \
347 strncmp(str1, str2, STRING_SIZE(str2))
349 static inline int
350 suffixcmp(const char *str, int slen, const char *suffix)
351 {
352 size_t len = slen >= 0 ? slen : strlen(str);
353 size_t suffixlen = strlen(suffix);
355 return suffixlen < len ? strcmp(str + len - suffixlen, suffix) : -1;
356 }
359 /*
360 * Unicode / UTF-8 handling
361 *
362 * NOTE: Much of the following code for dealing with Unicode is derived from
363 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
364 * src/intl/charset.c from the UTF-8 branch commit elinks-0.11.0-g31f2c28.
365 */
367 static inline int
368 unicode_width(unsigned long c, int tab_size)
369 {
370 if (c >= 0x1100 &&
371 (c <= 0x115f /* Hangul Jamo */
372 || c == 0x2329
373 || c == 0x232a
374 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
375 /* CJK ... Yi */
376 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
377 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
378 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
379 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
380 || (c >= 0xffe0 && c <= 0xffe6)
381 || (c >= 0x20000 && c <= 0x2fffd)
382 || (c >= 0x30000 && c <= 0x3fffd)))
383 return 2;
385 if (c == '\t')
386 return tab_size;
388 return 1;
389 }
391 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
392 * Illegal bytes are set one. */
393 static const unsigned char utf8_bytes[256] = {
394 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,
395 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,
396 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,
397 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,
398 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,
399 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,
400 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,
401 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,
402 };
404 static inline unsigned char
405 utf8_char_length(const char *string, const char *end)
406 {
407 int c = *(unsigned char *) string;
409 return utf8_bytes[c];
410 }
412 /* Decode UTF-8 multi-byte representation into a Unicode character. */
413 static inline unsigned long
414 utf8_to_unicode(const char *string, size_t length)
415 {
416 unsigned long unicode;
418 switch (length) {
419 case 1:
420 unicode = string[0];
421 break;
422 case 2:
423 unicode = (string[0] & 0x1f) << 6;
424 unicode += (string[1] & 0x3f);
425 break;
426 case 3:
427 unicode = (string[0] & 0x0f) << 12;
428 unicode += ((string[1] & 0x3f) << 6);
429 unicode += (string[2] & 0x3f);
430 break;
431 case 4:
432 unicode = (string[0] & 0x0f) << 18;
433 unicode += ((string[1] & 0x3f) << 12);
434 unicode += ((string[2] & 0x3f) << 6);
435 unicode += (string[3] & 0x3f);
436 break;
437 case 5:
438 unicode = (string[0] & 0x0f) << 24;
439 unicode += ((string[1] & 0x3f) << 18);
440 unicode += ((string[2] & 0x3f) << 12);
441 unicode += ((string[3] & 0x3f) << 6);
442 unicode += (string[4] & 0x3f);
443 break;
444 case 6:
445 unicode = (string[0] & 0x01) << 30;
446 unicode += ((string[1] & 0x3f) << 24);
447 unicode += ((string[2] & 0x3f) << 18);
448 unicode += ((string[3] & 0x3f) << 12);
449 unicode += ((string[4] & 0x3f) << 6);
450 unicode += (string[5] & 0x3f);
451 break;
452 default:
453 return 0;
454 }
456 /* Invalid characters could return the special 0xfffd value but NUL
457 * should be just as good. */
458 return unicode > 0xffff ? 0 : unicode;
459 }
461 /* Calculates how much of string can be shown within the given maximum width
462 * and sets trimmed parameter to non-zero value if all of string could not be
463 * shown. If the reserve flag is TRUE, it will reserve at least one
464 * trailing character, which can be useful when drawing a delimiter.
465 *
466 * Returns the number of bytes to output from string to satisfy max_width. */
467 static size_t
468 utf8_length(const char **start, size_t skip, int *width, size_t max_width, int *trimmed, bool reserve, int tab_size)
469 {
470 const char *string = *start;
471 const char *end = strchr(string, '\0');
472 unsigned char last_bytes = 0;
473 size_t last_ucwidth = 0;
475 *width = 0;
476 *trimmed = 0;
478 while (string < end) {
479 unsigned char bytes = utf8_char_length(string, end);
480 size_t ucwidth;
481 unsigned long unicode;
483 if (string + bytes > end)
484 break;
486 /* Change representation to figure out whether
487 * it is a single- or double-width character. */
489 unicode = utf8_to_unicode(string, bytes);
490 /* FIXME: Graceful handling of invalid Unicode character. */
491 if (!unicode)
492 break;
494 ucwidth = unicode_width(unicode, tab_size);
495 if (skip > 0) {
496 skip -= ucwidth <= skip ? ucwidth : skip;
497 *start += bytes;
498 }
499 *width += ucwidth;
500 if (*width > max_width) {
501 *trimmed = 1;
502 *width -= ucwidth;
503 if (reserve && *width == max_width) {
504 string -= last_bytes;
505 *width -= last_ucwidth;
506 }
507 break;
508 }
510 string += bytes;
511 last_bytes = ucwidth ? bytes : 0;
512 last_ucwidth = ucwidth;
513 }
515 return string - *start;
516 }
519 #define DATE_INFO \
520 DATE_(NO), \
521 DATE_(DEFAULT), \
522 DATE_(LOCAL), \
523 DATE_(RELATIVE), \
524 DATE_(SHORT)
526 enum date {
527 #define DATE_(name) DATE_##name
528 DATE_INFO
529 #undef DATE_
530 };
532 static const struct enum_map date_map[] = {
533 #define DATE_(name) ENUM_MAP(#name, DATE_##name)
534 DATE_INFO
535 #undef DATE_
536 };
538 struct time {
539 time_t sec;
540 int tz;
541 };
543 static inline int timecmp(const struct time *t1, const struct time *t2)
544 {
545 return t1->sec - t2->sec;
546 }
548 static const char *
549 mkdate(const struct time *time, enum date date)
550 {
551 static char buf[DATE_COLS + 1];
552 static const struct enum_map reldate[] = {
553 { "second", 1, 60 * 2 },
554 { "minute", 60, 60 * 60 * 2 },
555 { "hour", 60 * 60, 60 * 60 * 24 * 2 },
556 { "day", 60 * 60 * 24, 60 * 60 * 24 * 7 * 2 },
557 { "week", 60 * 60 * 24 * 7, 60 * 60 * 24 * 7 * 5 },
558 { "month", 60 * 60 * 24 * 30, 60 * 60 * 24 * 30 * 12 },
559 };
560 struct tm tm;
562 if (!date || !time || !time->sec)
563 return "";
565 if (date == DATE_RELATIVE) {
566 struct timeval now;
567 time_t date = time->sec + time->tz;
568 time_t seconds;
569 int i;
571 gettimeofday(&now, NULL);
572 seconds = now.tv_sec < date ? date - now.tv_sec : now.tv_sec - date;
573 for (i = 0; i < ARRAY_SIZE(reldate); i++) {
574 if (seconds >= reldate[i].value)
575 continue;
577 seconds /= reldate[i].namelen;
578 if (!string_format(buf, "%ld %s%s %s",
579 seconds, reldate[i].name,
580 seconds > 1 ? "s" : "",
581 now.tv_sec >= date ? "ago" : "ahead"))
582 break;
583 return buf;
584 }
585 }
587 if (date == DATE_LOCAL) {
588 time_t date = time->sec + time->tz;
589 localtime_r(&date, &tm);
590 }
591 else {
592 gmtime_r(&time->sec, &tm);
593 }
594 return strftime(buf, sizeof(buf), DATE_FORMAT, &tm) ? buf : NULL;
595 }
598 #define AUTHOR_VALUES \
599 AUTHOR_(NO), \
600 AUTHOR_(FULL), \
601 AUTHOR_(ABBREVIATED)
603 enum author {
604 #define AUTHOR_(name) AUTHOR_##name
605 AUTHOR_VALUES,
606 #undef AUTHOR_
607 AUTHOR_DEFAULT = AUTHOR_FULL
608 };
610 static const struct enum_map author_map[] = {
611 #define AUTHOR_(name) ENUM_MAP(#name, AUTHOR_##name)
612 AUTHOR_VALUES
613 #undef AUTHOR_
614 };
616 static const char *
617 get_author_initials(const char *author)
618 {
619 static char initials[AUTHOR_COLS * 6 + 1];
620 size_t pos = 0;
621 const char *end = strchr(author, '\0');
623 #define is_initial_sep(c) (isspace(c) || ispunct(c) || (c) == '@' || (c) == '-')
625 memset(initials, 0, sizeof(initials));
626 while (author < end) {
627 unsigned char bytes;
628 size_t i;
630 while (is_initial_sep(*author))
631 author++;
633 bytes = utf8_char_length(author, end);
634 if (bytes < sizeof(initials) - 1 - pos) {
635 while (bytes--) {
636 initials[pos++] = *author++;
637 }
638 }
640 for (i = pos; author < end && !is_initial_sep(*author); author++) {
641 if (i < sizeof(initials) - 1)
642 initials[i++] = *author;
643 }
645 initials[i++] = 0;
646 }
648 return initials;
649 }
652 static bool
653 argv_from_string(const char *argv[SIZEOF_ARG], int *argc, char *cmd)
654 {
655 int valuelen;
657 while (*cmd && *argc < SIZEOF_ARG && (valuelen = strcspn(cmd, " \t"))) {
658 bool advance = cmd[valuelen] != 0;
660 cmd[valuelen] = 0;
661 argv[(*argc)++] = chomp_string(cmd);
662 cmd = chomp_string(cmd + valuelen + advance);
663 }
665 if (*argc < SIZEOF_ARG)
666 argv[*argc] = NULL;
667 return *argc < SIZEOF_ARG;
668 }
670 static bool
671 argv_from_env(const char **argv, const char *name)
672 {
673 char *env = argv ? getenv(name) : NULL;
674 int argc = 0;
676 if (env && *env)
677 env = strdup(env);
678 return !env || argv_from_string(argv, &argc, env);
679 }
681 static void
682 argv_free(const char *argv[])
683 {
684 int argc;
686 for (argc = 0; argv[argc]; argc++)
687 free((void *) argv[argc]);
688 }
691 /*
692 * Executing external commands.
693 */
695 enum io_type {
696 IO_FD, /* File descriptor based IO. */
697 IO_BG, /* Execute command in the background. */
698 IO_FG, /* Execute command with same std{in,out,err}. */
699 IO_RD, /* Read only fork+exec IO. */
700 IO_WR, /* Write only fork+exec IO. */
701 IO_AP, /* Append fork+exec output to file. */
702 };
704 struct io {
705 enum io_type type; /* The requested type of pipe. */
706 const char *dir; /* Directory from which to execute. */
707 pid_t pid; /* PID of spawned process. */
708 int pipe; /* Pipe end for reading or writing. */
709 int error; /* Error status. */
710 const char *argv[SIZEOF_ARG]; /* Shell command arguments. */
711 char *buf; /* Read buffer. */
712 size_t bufalloc; /* Allocated buffer size. */
713 size_t bufsize; /* Buffer content size. */
714 char *bufpos; /* Current buffer position. */
715 unsigned int eof:1; /* Has end of file been reached. */
716 };
718 static void
719 io_reset(struct io *io)
720 {
721 io->pipe = -1;
722 io->pid = 0;
723 io->buf = io->bufpos = NULL;
724 io->bufalloc = io->bufsize = 0;
725 io->error = 0;
726 io->eof = 0;
727 }
729 static void
730 io_init(struct io *io, const char *dir, enum io_type type)
731 {
732 io_reset(io);
733 io->type = type;
734 io->dir = dir;
735 }
737 static bool
738 io_format(struct io *io, const char *dir, enum io_type type,
739 const char *argv[], enum format_flags flags)
740 {
741 io_init(io, dir, type);
742 return format_argv(io->argv, argv, flags);
743 }
745 static bool
746 io_open(struct io *io, const char *fmt, ...)
747 {
748 char name[SIZEOF_STR] = "";
749 bool fits;
750 va_list args;
752 io_init(io, NULL, IO_FD);
754 va_start(args, fmt);
755 fits = vsnprintf(name, sizeof(name), fmt, args) < sizeof(name);
756 va_end(args);
758 if (!fits) {
759 io->error = ENAMETOOLONG;
760 return FALSE;
761 }
762 io->pipe = *name ? open(name, O_RDONLY) : STDIN_FILENO;
763 if (io->pipe == -1)
764 io->error = errno;
765 return io->pipe != -1;
766 }
768 static bool
769 io_kill(struct io *io)
770 {
771 return io->pid == 0 || kill(io->pid, SIGKILL) != -1;
772 }
774 static bool
775 io_done(struct io *io)
776 {
777 pid_t pid = io->pid;
779 if (io->pipe != -1)
780 close(io->pipe);
781 free(io->buf);
782 io_reset(io);
784 while (pid > 0) {
785 int status;
786 pid_t waiting = waitpid(pid, &status, 0);
788 if (waiting < 0) {
789 if (errno == EINTR)
790 continue;
791 io->error = errno;
792 return FALSE;
793 }
795 return waiting == pid &&
796 !WIFSIGNALED(status) &&
797 WIFEXITED(status) &&
798 !WEXITSTATUS(status);
799 }
801 return TRUE;
802 }
804 static bool
805 io_start(struct io *io)
806 {
807 int pipefds[2] = { -1, -1 };
809 if (io->type == IO_FD)
810 return TRUE;
812 if ((io->type == IO_RD || io->type == IO_WR) && pipe(pipefds) < 0) {
813 io->error = errno;
814 return FALSE;
815 } else if (io->type == IO_AP) {
816 pipefds[1] = io->pipe;
817 }
819 if ((io->pid = fork())) {
820 if (io->pid == -1)
821 io->error = errno;
822 if (pipefds[!(io->type == IO_WR)] != -1)
823 close(pipefds[!(io->type == IO_WR)]);
824 if (io->pid != -1) {
825 io->pipe = pipefds[!!(io->type == IO_WR)];
826 return TRUE;
827 }
829 } else {
830 if (io->type != IO_FG) {
831 int devnull = open("/dev/null", O_RDWR);
832 int readfd = io->type == IO_WR ? pipefds[0] : devnull;
833 int writefd = (io->type == IO_RD || io->type == IO_AP)
834 ? pipefds[1] : devnull;
836 dup2(readfd, STDIN_FILENO);
837 dup2(writefd, STDOUT_FILENO);
838 dup2(devnull, STDERR_FILENO);
840 close(devnull);
841 if (pipefds[0] != -1)
842 close(pipefds[0]);
843 if (pipefds[1] != -1)
844 close(pipefds[1]);
845 }
847 if (io->dir && *io->dir && chdir(io->dir) == -1)
848 exit(errno);
850 execvp(io->argv[0], (char *const*) io->argv);
851 exit(errno);
852 }
854 if (pipefds[!!(io->type == IO_WR)] != -1)
855 close(pipefds[!!(io->type == IO_WR)]);
856 return FALSE;
857 }
859 static bool
860 io_run(struct io *io, const char **argv, const char *dir, enum io_type type)
861 {
862 io_init(io, dir, type);
863 if (!format_argv(io->argv, argv, FORMAT_NONE))
864 return FALSE;
865 return io_start(io);
866 }
868 static int
869 io_complete(struct io *io)
870 {
871 return io_start(io) && io_done(io);
872 }
874 static int
875 io_run_bg(const char **argv)
876 {
877 struct io io = {};
879 if (!io_format(&io, NULL, IO_BG, argv, FORMAT_NONE))
880 return FALSE;
881 return io_complete(&io);
882 }
884 static bool
885 io_run_fg(const char **argv, const char *dir)
886 {
887 struct io io = {};
889 if (!io_format(&io, dir, IO_FG, argv, FORMAT_NONE))
890 return FALSE;
891 return io_complete(&io);
892 }
894 static bool
895 io_run_append(const char **argv, enum format_flags flags, int fd)
896 {
897 struct io io = {};
899 if (!io_format(&io, NULL, IO_AP, argv, flags)) {
900 close(fd);
901 return FALSE;
902 }
904 io.pipe = fd;
905 return io_complete(&io);
906 }
908 static bool
909 io_run_rd(struct io *io, const char **argv, const char *dir, enum format_flags flags)
910 {
911 return io_format(io, dir, IO_RD, argv, flags) && io_start(io);
912 }
914 static bool
915 io_eof(struct io *io)
916 {
917 return io->eof;
918 }
920 static int
921 io_error(struct io *io)
922 {
923 return io->error;
924 }
926 static char *
927 io_strerror(struct io *io)
928 {
929 return strerror(io->error);
930 }
932 static bool
933 io_can_read(struct io *io)
934 {
935 struct timeval tv = { 0, 500 };
936 fd_set fds;
938 FD_ZERO(&fds);
939 FD_SET(io->pipe, &fds);
941 return select(io->pipe + 1, &fds, NULL, NULL, &tv) > 0;
942 }
944 static ssize_t
945 io_read(struct io *io, void *buf, size_t bufsize)
946 {
947 do {
948 ssize_t readsize = read(io->pipe, buf, bufsize);
950 if (readsize < 0 && (errno == EAGAIN || errno == EINTR))
951 continue;
952 else if (readsize == -1)
953 io->error = errno;
954 else if (readsize == 0)
955 io->eof = 1;
956 return readsize;
957 } while (1);
958 }
960 DEFINE_ALLOCATOR(io_realloc_buf, char, BUFSIZ)
962 static char *
963 io_get(struct io *io, int c, bool can_read)
964 {
965 char *eol;
966 ssize_t readsize;
968 while (TRUE) {
969 if (io->bufsize > 0) {
970 eol = memchr(io->bufpos, c, io->bufsize);
971 if (eol) {
972 char *line = io->bufpos;
974 *eol = 0;
975 io->bufpos = eol + 1;
976 io->bufsize -= io->bufpos - line;
977 return line;
978 }
979 }
981 if (io_eof(io)) {
982 if (io->bufsize) {
983 io->bufpos[io->bufsize] = 0;
984 io->bufsize = 0;
985 return io->bufpos;
986 }
987 return NULL;
988 }
990 if (!can_read)
991 return NULL;
993 if (io->bufsize > 0 && io->bufpos > io->buf)
994 memmove(io->buf, io->bufpos, io->bufsize);
996 if (io->bufalloc == io->bufsize) {
997 if (!io_realloc_buf(&io->buf, io->bufalloc, BUFSIZ))
998 return NULL;
999 io->bufalloc += BUFSIZ;
1000 }
1002 io->bufpos = io->buf;
1003 readsize = io_read(io, io->buf + io->bufsize, io->bufalloc - io->bufsize);
1004 if (io_error(io))
1005 return NULL;
1006 io->bufsize += readsize;
1007 }
1008 }
1010 static bool
1011 io_write(struct io *io, const void *buf, size_t bufsize)
1012 {
1013 size_t written = 0;
1015 while (!io_error(io) && written < bufsize) {
1016 ssize_t size;
1018 size = write(io->pipe, buf + written, bufsize - written);
1019 if (size < 0 && (errno == EAGAIN || errno == EINTR))
1020 continue;
1021 else if (size == -1)
1022 io->error = errno;
1023 else
1024 written += size;
1025 }
1027 return written == bufsize;
1028 }
1030 static bool
1031 io_read_buf(struct io *io, char buf[], size_t bufsize)
1032 {
1033 char *result = io_get(io, '\n', TRUE);
1035 if (result) {
1036 result = chomp_string(result);
1037 string_ncopy_do(buf, bufsize, result, strlen(result));
1038 }
1040 return io_done(io) && result;
1041 }
1043 static bool
1044 io_run_buf(const char **argv, char buf[], size_t bufsize)
1045 {
1046 struct io io = {};
1048 return io_run_rd(&io, argv, NULL, FORMAT_NONE)
1049 && io_read_buf(&io, buf, bufsize);
1050 }
1052 static int
1053 io_load(struct io *io, const char *separators,
1054 int (*read_property)(char *, size_t, char *, size_t))
1055 {
1056 char *name;
1057 int state = OK;
1059 if (!io_start(io))
1060 return ERR;
1062 while (state == OK && (name = io_get(io, '\n', TRUE))) {
1063 char *value;
1064 size_t namelen;
1065 size_t valuelen;
1067 name = chomp_string(name);
1068 namelen = strcspn(name, separators);
1070 if (name[namelen]) {
1071 name[namelen] = 0;
1072 value = chomp_string(name + namelen + 1);
1073 valuelen = strlen(value);
1075 } else {
1076 value = "";
1077 valuelen = 0;
1078 }
1080 state = read_property(name, namelen, value, valuelen);
1081 }
1083 if (state != ERR && io_error(io))
1084 state = ERR;
1085 io_done(io);
1087 return state;
1088 }
1090 static int
1091 io_run_load(const char **argv, const char *separators,
1092 int (*read_property)(char *, size_t, char *, size_t))
1093 {
1094 struct io io = {};
1096 return io_format(&io, NULL, IO_RD, argv, FORMAT_NONE)
1097 ? io_load(&io, separators, read_property) : ERR;
1098 }
1101 /*
1102 * User requests
1103 */
1105 #define REQ_INFO \
1106 /* XXX: Keep the view request first and in sync with views[]. */ \
1107 REQ_GROUP("View switching") \
1108 REQ_(VIEW_MAIN, "Show main view"), \
1109 REQ_(VIEW_DIFF, "Show diff view"), \
1110 REQ_(VIEW_LOG, "Show log view"), \
1111 REQ_(VIEW_TREE, "Show tree view"), \
1112 REQ_(VIEW_BLOB, "Show blob view"), \
1113 REQ_(VIEW_BLAME, "Show blame view"), \
1114 REQ_(VIEW_BRANCH, "Show branch view"), \
1115 REQ_(VIEW_HELP, "Show help page"), \
1116 REQ_(VIEW_PAGER, "Show pager view"), \
1117 REQ_(VIEW_STATUS, "Show status view"), \
1118 REQ_(VIEW_STAGE, "Show stage view"), \
1119 \
1120 REQ_GROUP("View manipulation") \
1121 REQ_(ENTER, "Enter current line and scroll"), \
1122 REQ_(NEXT, "Move to next"), \
1123 REQ_(PREVIOUS, "Move to previous"), \
1124 REQ_(PARENT, "Move to parent"), \
1125 REQ_(VIEW_NEXT, "Move focus to next view"), \
1126 REQ_(REFRESH, "Reload and refresh"), \
1127 REQ_(MAXIMIZE, "Maximize the current view"), \
1128 REQ_(VIEW_CLOSE, "Close the current view"), \
1129 REQ_(QUIT, "Close all views and quit"), \
1130 \
1131 REQ_GROUP("View specific requests") \
1132 REQ_(STATUS_UPDATE, "Update file status"), \
1133 REQ_(STATUS_REVERT, "Revert file changes"), \
1134 REQ_(STATUS_MERGE, "Merge file using external tool"), \
1135 REQ_(STAGE_NEXT, "Find next chunk to stage"), \
1136 \
1137 REQ_GROUP("Cursor navigation") \
1138 REQ_(MOVE_UP, "Move cursor one line up"), \
1139 REQ_(MOVE_DOWN, "Move cursor one line down"), \
1140 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
1141 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
1142 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
1143 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
1144 \
1145 REQ_GROUP("Scrolling") \
1146 REQ_(SCROLL_LEFT, "Scroll two columns left"), \
1147 REQ_(SCROLL_RIGHT, "Scroll two columns right"), \
1148 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
1149 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
1150 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
1151 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
1152 \
1153 REQ_GROUP("Searching") \
1154 REQ_(SEARCH, "Search the view"), \
1155 REQ_(SEARCH_BACK, "Search backwards in the view"), \
1156 REQ_(FIND_NEXT, "Find next search match"), \
1157 REQ_(FIND_PREV, "Find previous search match"), \
1158 \
1159 REQ_GROUP("Option manipulation") \
1160 REQ_(OPTIONS, "Open option menu"), \
1161 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
1162 REQ_(TOGGLE_DATE, "Toggle date display"), \
1163 REQ_(TOGGLE_DATE_SHORT, "Toggle short (date-only) dates"), \
1164 REQ_(TOGGLE_AUTHOR, "Toggle author display"), \
1165 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization"), \
1166 REQ_(TOGGLE_REFS, "Toggle reference display (tags/branches)"), \
1167 REQ_(TOGGLE_SORT_ORDER, "Toggle ascending/descending sort order"), \
1168 REQ_(TOGGLE_SORT_FIELD, "Toggle field to sort by"), \
1169 \
1170 REQ_GROUP("Misc") \
1171 REQ_(PROMPT, "Bring up the prompt"), \
1172 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
1173 REQ_(SHOW_VERSION, "Show version information"), \
1174 REQ_(STOP_LOADING, "Stop all loading views"), \
1175 REQ_(EDIT, "Open in editor"), \
1176 REQ_(NONE, "Do nothing")
1179 /* User action requests. */
1180 enum request {
1181 #define REQ_GROUP(help)
1182 #define REQ_(req, help) REQ_##req
1184 /* Offset all requests to avoid conflicts with ncurses getch values. */
1185 REQ_UNKNOWN = KEY_MAX + 1,
1186 REQ_OFFSET,
1187 REQ_INFO
1189 #undef REQ_GROUP
1190 #undef REQ_
1191 };
1193 struct request_info {
1194 enum request request;
1195 const char *name;
1196 int namelen;
1197 const char *help;
1198 };
1200 static const struct request_info req_info[] = {
1201 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
1202 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
1203 REQ_INFO
1204 #undef REQ_GROUP
1205 #undef REQ_
1206 };
1208 static enum request
1209 get_request(const char *name)
1210 {
1211 int namelen = strlen(name);
1212 int i;
1214 for (i = 0; i < ARRAY_SIZE(req_info); i++)
1215 if (enum_equals(req_info[i], name, namelen))
1216 return req_info[i].request;
1218 return REQ_UNKNOWN;
1219 }
1222 /*
1223 * Options
1224 */
1226 /* Option and state variables. */
1227 static enum date opt_date = DATE_DEFAULT;
1228 static enum author opt_author = AUTHOR_DEFAULT;
1229 static bool opt_line_number = FALSE;
1230 static bool opt_line_graphics = TRUE;
1231 static bool opt_rev_graph = FALSE;
1232 static bool opt_show_refs = TRUE;
1233 static int opt_num_interval = 5;
1234 static double opt_hscroll = 0.50;
1235 static double opt_scale_split_view = 2.0 / 3.0;
1236 static int opt_tab_size = 8;
1237 static int opt_author_cols = AUTHOR_COLS;
1238 static char opt_path[SIZEOF_STR] = "";
1239 static char opt_file[SIZEOF_STR] = "";
1240 static char opt_ref[SIZEOF_REF] = "";
1241 static char opt_head[SIZEOF_REF] = "";
1242 static char opt_remote[SIZEOF_REF] = "";
1243 static char opt_encoding[20] = "UTF-8";
1244 static iconv_t opt_iconv_in = ICONV_NONE;
1245 static iconv_t opt_iconv_out = ICONV_NONE;
1246 static char opt_search[SIZEOF_STR] = "";
1247 static char opt_cdup[SIZEOF_STR] = "";
1248 static char opt_prefix[SIZEOF_STR] = "";
1249 static char opt_git_dir[SIZEOF_STR] = "";
1250 static signed char opt_is_inside_work_tree = -1; /* set to TRUE or FALSE */
1251 static char opt_editor[SIZEOF_STR] = "";
1252 static FILE *opt_tty = NULL;
1254 #define is_initial_commit() (!get_ref_head())
1255 #define is_head_commit(rev) (!strcmp((rev), "HEAD") || (get_ref_head() && !strcmp(rev, get_ref_head()->id)))
1258 /*
1259 * Line-oriented content detection.
1260 */
1262 #define LINE_INFO \
1263 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1264 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1265 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
1266 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
1267 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
1268 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1269 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1270 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1271 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1272 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1273 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1274 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1275 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1276 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
1277 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
1278 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1279 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
1280 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1281 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1282 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1283 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
1284 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
1285 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
1286 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
1287 LINE(AUTHOR, "author ", COLOR_GREEN, COLOR_DEFAULT, 0), \
1288 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1289 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1290 LINE(ACKED, " Acked-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1291 LINE(TESTED, " Tested-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1292 LINE(REVIEWED, " Reviewed-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1293 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
1294 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
1295 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
1296 LINE(DELIMITER, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1297 LINE(DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
1298 LINE(MODE, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
1299 LINE(LINE_NUMBER, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
1300 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
1301 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
1302 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
1303 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
1304 LINE(MAIN_LOCAL_TAG,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1305 LINE(MAIN_REMOTE, "", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1306 LINE(MAIN_TRACKED, "", COLOR_YELLOW, COLOR_DEFAULT, A_BOLD), \
1307 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
1308 LINE(MAIN_HEAD, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
1309 LINE(MAIN_REVGRAPH,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1310 LINE(TREE_HEAD, "", COLOR_DEFAULT, COLOR_DEFAULT, A_BOLD), \
1311 LINE(TREE_DIR, "", COLOR_YELLOW, COLOR_DEFAULT, A_NORMAL), \
1312 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
1313 LINE(STAT_HEAD, "", COLOR_YELLOW, COLOR_DEFAULT, 0), \
1314 LINE(STAT_SECTION, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
1315 LINE(STAT_NONE, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
1316 LINE(STAT_STAGED, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1317 LINE(STAT_UNSTAGED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1318 LINE(STAT_UNTRACKED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
1319 LINE(HELP_KEYMAP, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
1320 LINE(HELP_GROUP, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
1321 LINE(BLAME_ID, "", COLOR_MAGENTA, COLOR_DEFAULT, 0)
1323 enum line_type {
1324 #define LINE(type, line, fg, bg, attr) \
1325 LINE_##type
1326 LINE_INFO,
1327 LINE_NONE
1328 #undef LINE
1329 };
1331 struct line_info {
1332 const char *name; /* Option name. */
1333 int namelen; /* Size of option name. */
1334 const char *line; /* The start of line to match. */
1335 int linelen; /* Size of string to match. */
1336 int fg, bg, attr; /* Color and text attributes for the lines. */
1337 };
1339 static struct line_info line_info[] = {
1340 #define LINE(type, line, fg, bg, attr) \
1341 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
1342 LINE_INFO
1343 #undef LINE
1344 };
1346 static enum line_type
1347 get_line_type(const char *line)
1348 {
1349 int linelen = strlen(line);
1350 enum line_type type;
1352 for (type = 0; type < ARRAY_SIZE(line_info); type++)
1353 /* Case insensitive search matches Signed-off-by lines better. */
1354 if (linelen >= line_info[type].linelen &&
1355 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
1356 return type;
1358 return LINE_DEFAULT;
1359 }
1361 static inline int
1362 get_line_attr(enum line_type type)
1363 {
1364 assert(type < ARRAY_SIZE(line_info));
1365 return COLOR_PAIR(type) | line_info[type].attr;
1366 }
1368 static struct line_info *
1369 get_line_info(const char *name)
1370 {
1371 size_t namelen = strlen(name);
1372 enum line_type type;
1374 for (type = 0; type < ARRAY_SIZE(line_info); type++)
1375 if (enum_equals(line_info[type], name, namelen))
1376 return &line_info[type];
1378 return NULL;
1379 }
1381 static void
1382 init_colors(void)
1383 {
1384 int default_bg = line_info[LINE_DEFAULT].bg;
1385 int default_fg = line_info[LINE_DEFAULT].fg;
1386 enum line_type type;
1388 start_color();
1390 if (assume_default_colors(default_fg, default_bg) == ERR) {
1391 default_bg = COLOR_BLACK;
1392 default_fg = COLOR_WHITE;
1393 }
1395 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
1396 struct line_info *info = &line_info[type];
1397 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
1398 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
1400 init_pair(type, fg, bg);
1401 }
1402 }
1404 struct line {
1405 enum line_type type;
1407 /* State flags */
1408 unsigned int selected:1;
1409 unsigned int dirty:1;
1410 unsigned int cleareol:1;
1411 unsigned int other:16;
1413 void *data; /* User data */
1414 };
1417 /*
1418 * Keys
1419 */
1421 struct keybinding {
1422 int alias;
1423 enum request request;
1424 };
1426 static struct keybinding default_keybindings[] = {
1427 /* View switching */
1428 { 'm', REQ_VIEW_MAIN },
1429 { 'd', REQ_VIEW_DIFF },
1430 { 'l', REQ_VIEW_LOG },
1431 { 't', REQ_VIEW_TREE },
1432 { 'f', REQ_VIEW_BLOB },
1433 { 'B', REQ_VIEW_BLAME },
1434 { 'H', REQ_VIEW_BRANCH },
1435 { 'p', REQ_VIEW_PAGER },
1436 { 'h', REQ_VIEW_HELP },
1437 { 'S', REQ_VIEW_STATUS },
1438 { 'c', REQ_VIEW_STAGE },
1440 /* View manipulation */
1441 { 'q', REQ_VIEW_CLOSE },
1442 { KEY_TAB, REQ_VIEW_NEXT },
1443 { KEY_RETURN, REQ_ENTER },
1444 { KEY_UP, REQ_PREVIOUS },
1445 { KEY_DOWN, REQ_NEXT },
1446 { 'R', REQ_REFRESH },
1447 { KEY_F(5), REQ_REFRESH },
1448 { 'O', REQ_MAXIMIZE },
1450 /* Cursor navigation */
1451 { 'k', REQ_MOVE_UP },
1452 { 'j', REQ_MOVE_DOWN },
1453 { KEY_HOME, REQ_MOVE_FIRST_LINE },
1454 { KEY_END, REQ_MOVE_LAST_LINE },
1455 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
1456 { ' ', REQ_MOVE_PAGE_DOWN },
1457 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
1458 { 'b', REQ_MOVE_PAGE_UP },
1459 { '-', REQ_MOVE_PAGE_UP },
1461 /* Scrolling */
1462 { KEY_LEFT, REQ_SCROLL_LEFT },
1463 { KEY_RIGHT, REQ_SCROLL_RIGHT },
1464 { KEY_IC, REQ_SCROLL_LINE_UP },
1465 { KEY_DC, REQ_SCROLL_LINE_DOWN },
1466 { 'w', REQ_SCROLL_PAGE_UP },
1467 { 's', REQ_SCROLL_PAGE_DOWN },
1469 /* Searching */
1470 { '/', REQ_SEARCH },
1471 { '?', REQ_SEARCH_BACK },
1472 { 'n', REQ_FIND_NEXT },
1473 { 'N', REQ_FIND_PREV },
1475 /* Misc */
1476 { 'Q', REQ_QUIT },
1477 { 'z', REQ_STOP_LOADING },
1478 { 'v', REQ_SHOW_VERSION },
1479 { 'r', REQ_SCREEN_REDRAW },
1480 { 'o', REQ_OPTIONS },
1481 { '.', REQ_TOGGLE_LINENO },
1482 { 'D', REQ_TOGGLE_DATE },
1483 { 'A', REQ_TOGGLE_AUTHOR },
1484 { 'g', REQ_TOGGLE_REV_GRAPH },
1485 { 'F', REQ_TOGGLE_REFS },
1486 { 'I', REQ_TOGGLE_SORT_ORDER },
1487 { 'i', REQ_TOGGLE_SORT_FIELD },
1488 { ':', REQ_PROMPT },
1489 { 'u', REQ_STATUS_UPDATE },
1490 { '!', REQ_STATUS_REVERT },
1491 { 'M', REQ_STATUS_MERGE },
1492 { '@', REQ_STAGE_NEXT },
1493 { ',', REQ_PARENT },
1494 { 'e', REQ_EDIT },
1495 };
1497 #define KEYMAP_INFO \
1498 KEYMAP_(GENERIC), \
1499 KEYMAP_(MAIN), \
1500 KEYMAP_(DIFF), \
1501 KEYMAP_(LOG), \
1502 KEYMAP_(TREE), \
1503 KEYMAP_(BLOB), \
1504 KEYMAP_(BLAME), \
1505 KEYMAP_(BRANCH), \
1506 KEYMAP_(PAGER), \
1507 KEYMAP_(HELP), \
1508 KEYMAP_(STATUS), \
1509 KEYMAP_(STAGE)
1511 enum keymap {
1512 #define KEYMAP_(name) KEYMAP_##name
1513 KEYMAP_INFO
1514 #undef KEYMAP_
1515 };
1517 static const struct enum_map keymap_table[] = {
1518 #define KEYMAP_(name) ENUM_MAP(#name, KEYMAP_##name)
1519 KEYMAP_INFO
1520 #undef KEYMAP_
1521 };
1523 #define set_keymap(map, name) map_enum(map, keymap_table, name)
1525 struct keybinding_table {
1526 struct keybinding *data;
1527 size_t size;
1528 };
1530 static struct keybinding_table keybindings[ARRAY_SIZE(keymap_table)];
1532 static void
1533 add_keybinding(enum keymap keymap, enum request request, int key)
1534 {
1535 struct keybinding_table *table = &keybindings[keymap];
1536 size_t i;
1538 for (i = 0; i < keybindings[keymap].size; i++) {
1539 if (keybindings[keymap].data[i].alias == key) {
1540 keybindings[keymap].data[i].request = request;
1541 return;
1542 }
1543 }
1545 table->data = realloc(table->data, (table->size + 1) * sizeof(*table->data));
1546 if (!table->data)
1547 die("Failed to allocate keybinding");
1548 table->data[table->size].alias = key;
1549 table->data[table->size++].request = request;
1551 if (request == REQ_NONE && keymap == KEYMAP_GENERIC) {
1552 int i;
1554 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
1555 if (default_keybindings[i].alias == key)
1556 default_keybindings[i].request = REQ_NONE;
1557 }
1558 }
1560 /* Looks for a key binding first in the given map, then in the generic map, and
1561 * lastly in the default keybindings. */
1562 static enum request
1563 get_keybinding(enum keymap keymap, int key)
1564 {
1565 size_t i;
1567 for (i = 0; i < keybindings[keymap].size; i++)
1568 if (keybindings[keymap].data[i].alias == key)
1569 return keybindings[keymap].data[i].request;
1571 for (i = 0; i < keybindings[KEYMAP_GENERIC].size; i++)
1572 if (keybindings[KEYMAP_GENERIC].data[i].alias == key)
1573 return keybindings[KEYMAP_GENERIC].data[i].request;
1575 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
1576 if (default_keybindings[i].alias == key)
1577 return default_keybindings[i].request;
1579 return (enum request) key;
1580 }
1583 struct key {
1584 const char *name;
1585 int value;
1586 };
1588 static const struct key key_table[] = {
1589 { "Enter", KEY_RETURN },
1590 { "Space", ' ' },
1591 { "Backspace", KEY_BACKSPACE },
1592 { "Tab", KEY_TAB },
1593 { "Escape", KEY_ESC },
1594 { "Left", KEY_LEFT },
1595 { "Right", KEY_RIGHT },
1596 { "Up", KEY_UP },
1597 { "Down", KEY_DOWN },
1598 { "Insert", KEY_IC },
1599 { "Delete", KEY_DC },
1600 { "Hash", '#' },
1601 { "Home", KEY_HOME },
1602 { "End", KEY_END },
1603 { "PageUp", KEY_PPAGE },
1604 { "PageDown", KEY_NPAGE },
1605 { "F1", KEY_F(1) },
1606 { "F2", KEY_F(2) },
1607 { "F3", KEY_F(3) },
1608 { "F4", KEY_F(4) },
1609 { "F5", KEY_F(5) },
1610 { "F6", KEY_F(6) },
1611 { "F7", KEY_F(7) },
1612 { "F8", KEY_F(8) },
1613 { "F9", KEY_F(9) },
1614 { "F10", KEY_F(10) },
1615 { "F11", KEY_F(11) },
1616 { "F12", KEY_F(12) },
1617 };
1619 static int
1620 get_key_value(const char *name)
1621 {
1622 int i;
1624 for (i = 0; i < ARRAY_SIZE(key_table); i++)
1625 if (!strcasecmp(key_table[i].name, name))
1626 return key_table[i].value;
1628 if (strlen(name) == 1 && isprint(*name))
1629 return (int) *name;
1631 return ERR;
1632 }
1634 static const char *
1635 get_key_name(int key_value)
1636 {
1637 static char key_char[] = "'X'";
1638 const char *seq = NULL;
1639 int key;
1641 for (key = 0; key < ARRAY_SIZE(key_table); key++)
1642 if (key_table[key].value == key_value)
1643 seq = key_table[key].name;
1645 if (seq == NULL &&
1646 key_value < 127 &&
1647 isprint(key_value)) {
1648 key_char[1] = (char) key_value;
1649 seq = key_char;
1650 }
1652 return seq ? seq : "(no key)";
1653 }
1655 static bool
1656 append_key(char *buf, size_t *pos, const struct keybinding *keybinding)
1657 {
1658 const char *sep = *pos > 0 ? ", " : "";
1659 const char *keyname = get_key_name(keybinding->alias);
1661 return string_nformat(buf, BUFSIZ, pos, "%s%s", sep, keyname);
1662 }
1664 static bool
1665 append_keymap_request_keys(char *buf, size_t *pos, enum request request,
1666 enum keymap keymap, bool all)
1667 {
1668 int i;
1670 for (i = 0; i < keybindings[keymap].size; i++) {
1671 if (keybindings[keymap].data[i].request == request) {
1672 if (!append_key(buf, pos, &keybindings[keymap].data[i]))
1673 return FALSE;
1674 if (!all)
1675 break;
1676 }
1677 }
1679 return TRUE;
1680 }
1682 #define get_key(keymap, request) get_keys(keymap, request, FALSE)
1684 static const char *
1685 get_keys(enum keymap keymap, enum request request, bool all)
1686 {
1687 static char buf[BUFSIZ];
1688 size_t pos = 0;
1689 int i;
1691 buf[pos] = 0;
1693 if (!append_keymap_request_keys(buf, &pos, request, keymap, all))
1694 return "Too many keybindings!";
1695 if (pos > 0 && !all)
1696 return buf;
1698 if (keymap != KEYMAP_GENERIC) {
1699 /* Only the generic keymap includes the default keybindings when
1700 * listing all keys. */
1701 if (all)
1702 return buf;
1704 if (!append_keymap_request_keys(buf, &pos, request, KEYMAP_GENERIC, all))
1705 return "Too many keybindings!";
1706 if (pos)
1707 return buf;
1708 }
1710 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
1711 if (default_keybindings[i].request == request) {
1712 if (!append_key(buf, &pos, &default_keybindings[i]))
1713 return "Too many keybindings!";
1714 if (!all)
1715 return buf;
1716 }
1717 }
1719 return buf;
1720 }
1722 struct run_request {
1723 enum keymap keymap;
1724 int key;
1725 const char *argv[SIZEOF_ARG];
1726 };
1728 static struct run_request *run_request;
1729 static size_t run_requests;
1731 DEFINE_ALLOCATOR(realloc_run_requests, struct run_request, 8)
1733 static enum request
1734 add_run_request(enum keymap keymap, int key, int argc, const char **argv)
1735 {
1736 struct run_request *req;
1738 if (argc >= ARRAY_SIZE(req->argv) - 1)
1739 return REQ_NONE;
1741 if (!realloc_run_requests(&run_request, run_requests, 1))
1742 return REQ_NONE;
1744 req = &run_request[run_requests];
1745 req->keymap = keymap;
1746 req->key = key;
1747 req->argv[0] = NULL;
1749 if (!format_argv(req->argv, argv, FORMAT_NONE))
1750 return REQ_NONE;
1752 return REQ_NONE + ++run_requests;
1753 }
1755 static struct run_request *
1756 get_run_request(enum request request)
1757 {
1758 if (request <= REQ_NONE)
1759 return NULL;
1760 return &run_request[request - REQ_NONE - 1];
1761 }
1763 static void
1764 add_builtin_run_requests(void)
1765 {
1766 const char *cherry_pick[] = { "git", "cherry-pick", "%(commit)", NULL };
1767 const char *checkout[] = { "git", "checkout", "%(branch)", NULL };
1768 const char *commit[] = { "git", "commit", NULL };
1769 const char *gc[] = { "git", "gc", NULL };
1770 struct {
1771 enum keymap keymap;
1772 int key;
1773 int argc;
1774 const char **argv;
1775 } reqs[] = {
1776 { KEYMAP_MAIN, 'C', ARRAY_SIZE(cherry_pick) - 1, cherry_pick },
1777 { KEYMAP_STATUS, 'C', ARRAY_SIZE(commit) - 1, commit },
1778 { KEYMAP_BRANCH, 'C', ARRAY_SIZE(checkout) - 1, checkout },
1779 { KEYMAP_GENERIC, 'G', ARRAY_SIZE(gc) - 1, gc },
1780 };
1781 int i;
1783 for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1784 enum request req = get_keybinding(reqs[i].keymap, reqs[i].key);
1786 if (req != reqs[i].key)
1787 continue;
1788 req = add_run_request(reqs[i].keymap, reqs[i].key, reqs[i].argc, reqs[i].argv);
1789 if (req != REQ_NONE)
1790 add_keybinding(reqs[i].keymap, req, reqs[i].key);
1791 }
1792 }
1794 /*
1795 * User config file handling.
1796 */
1798 static int config_lineno;
1799 static bool config_errors;
1800 static const char *config_msg;
1802 static const struct enum_map color_map[] = {
1803 #define COLOR_MAP(name) ENUM_MAP(#name, COLOR_##name)
1804 COLOR_MAP(DEFAULT),
1805 COLOR_MAP(BLACK),
1806 COLOR_MAP(BLUE),
1807 COLOR_MAP(CYAN),
1808 COLOR_MAP(GREEN),
1809 COLOR_MAP(MAGENTA),
1810 COLOR_MAP(RED),
1811 COLOR_MAP(WHITE),
1812 COLOR_MAP(YELLOW),
1813 };
1815 static const struct enum_map attr_map[] = {
1816 #define ATTR_MAP(name) ENUM_MAP(#name, A_##name)
1817 ATTR_MAP(NORMAL),
1818 ATTR_MAP(BLINK),
1819 ATTR_MAP(BOLD),
1820 ATTR_MAP(DIM),
1821 ATTR_MAP(REVERSE),
1822 ATTR_MAP(STANDOUT),
1823 ATTR_MAP(UNDERLINE),
1824 };
1826 #define set_attribute(attr, name) map_enum(attr, attr_map, name)
1828 static int parse_step(double *opt, const char *arg)
1829 {
1830 *opt = atoi(arg);
1831 if (!strchr(arg, '%'))
1832 return OK;
1834 /* "Shift down" so 100% and 1 does not conflict. */
1835 *opt = (*opt - 1) / 100;
1836 if (*opt >= 1.0) {
1837 *opt = 0.99;
1838 config_msg = "Step value larger than 100%";
1839 return ERR;
1840 }
1841 if (*opt < 0.0) {
1842 *opt = 1;
1843 config_msg = "Invalid step value";
1844 return ERR;
1845 }
1846 return OK;
1847 }
1849 static int
1850 parse_int(int *opt, const char *arg, int min, int max)
1851 {
1852 int value = atoi(arg);
1854 if (min <= value && value <= max) {
1855 *opt = value;
1856 return OK;
1857 }
1859 config_msg = "Integer value out of bound";
1860 return ERR;
1861 }
1863 static bool
1864 set_color(int *color, const char *name)
1865 {
1866 if (map_enum(color, color_map, name))
1867 return TRUE;
1868 if (!prefixcmp(name, "color"))
1869 return parse_int(color, name + 5, 0, 255) == OK;
1870 return FALSE;
1871 }
1873 /* Wants: object fgcolor bgcolor [attribute] */
1874 static int
1875 option_color_command(int argc, const char *argv[])
1876 {
1877 struct line_info *info;
1879 if (argc < 3) {
1880 config_msg = "Wrong number of arguments given to color command";
1881 return ERR;
1882 }
1884 info = get_line_info(argv[0]);
1885 if (!info) {
1886 static const struct enum_map obsolete[] = {
1887 ENUM_MAP("main-delim", LINE_DELIMITER),
1888 ENUM_MAP("main-date", LINE_DATE),
1889 ENUM_MAP("main-author", LINE_AUTHOR),
1890 };
1891 int index;
1893 if (!map_enum(&index, obsolete, argv[0])) {
1894 config_msg = "Unknown color name";
1895 return ERR;
1896 }
1897 info = &line_info[index];
1898 }
1900 if (!set_color(&info->fg, argv[1]) ||
1901 !set_color(&info->bg, argv[2])) {
1902 config_msg = "Unknown color";
1903 return ERR;
1904 }
1906 info->attr = 0;
1907 while (argc-- > 3) {
1908 int attr;
1910 if (!set_attribute(&attr, argv[argc])) {
1911 config_msg = "Unknown attribute";
1912 return ERR;
1913 }
1914 info->attr |= attr;
1915 }
1917 return OK;
1918 }
1920 static int parse_bool(bool *opt, const char *arg)
1921 {
1922 *opt = (!strcmp(arg, "1") || !strcmp(arg, "true") || !strcmp(arg, "yes"))
1923 ? TRUE : FALSE;
1924 return OK;
1925 }
1927 static int parse_enum_do(unsigned int *opt, const char *arg,
1928 const struct enum_map *map, size_t map_size)
1929 {
1930 bool is_true;
1932 assert(map_size > 1);
1934 if (map_enum_do(map, map_size, (int *) opt, arg))
1935 return OK;
1937 if (parse_bool(&is_true, arg) != OK)
1938 return ERR;
1940 *opt = is_true ? map[1].value : map[0].value;
1941 return OK;
1942 }
1944 #define parse_enum(opt, arg, map) \
1945 parse_enum_do(opt, arg, map, ARRAY_SIZE(map))
1947 static int
1948 parse_string(char *opt, const char *arg, size_t optsize)
1949 {
1950 int arglen = strlen(arg);
1952 switch (arg[0]) {
1953 case '\"':
1954 case '\'':
1955 if (arglen == 1 || arg[arglen - 1] != arg[0]) {
1956 config_msg = "Unmatched quotation";
1957 return ERR;
1958 }
1959 arg += 1; arglen -= 2;
1960 default:
1961 string_ncopy_do(opt, optsize, arg, arglen);
1962 return OK;
1963 }
1964 }
1966 /* Wants: name = value */
1967 static int
1968 option_set_command(int argc, const char *argv[])
1969 {
1970 if (argc != 3) {
1971 config_msg = "Wrong number of arguments given to set command";
1972 return ERR;
1973 }
1975 if (strcmp(argv[1], "=")) {
1976 config_msg = "No value assigned";
1977 return ERR;
1978 }
1980 if (!strcmp(argv[0], "show-author"))
1981 return parse_enum(&opt_author, argv[2], author_map);
1983 if (!strcmp(argv[0], "show-date"))
1984 return parse_enum(&opt_date, argv[2], date_map);
1986 if (!strcmp(argv[0], "show-rev-graph"))
1987 return parse_bool(&opt_rev_graph, argv[2]);
1989 if (!strcmp(argv[0], "show-refs"))
1990 return parse_bool(&opt_show_refs, argv[2]);
1992 if (!strcmp(argv[0], "show-line-numbers"))
1993 return parse_bool(&opt_line_number, argv[2]);
1995 if (!strcmp(argv[0], "line-graphics"))
1996 return parse_bool(&opt_line_graphics, argv[2]);
1998 if (!strcmp(argv[0], "line-number-interval"))
1999 return parse_int(&opt_num_interval, argv[2], 1, 1024);
2001 if (!strcmp(argv[0], "author-width"))
2002 return parse_int(&opt_author_cols, argv[2], 0, 1024);
2004 if (!strcmp(argv[0], "horizontal-scroll"))
2005 return parse_step(&opt_hscroll, argv[2]);
2007 if (!strcmp(argv[0], "split-view-height"))
2008 return parse_step(&opt_scale_split_view, argv[2]);
2010 if (!strcmp(argv[0], "tab-size"))
2011 return parse_int(&opt_tab_size, argv[2], 1, 1024);
2013 if (!strcmp(argv[0], "commit-encoding"))
2014 return parse_string(opt_encoding, argv[2], sizeof(opt_encoding));
2016 config_msg = "Unknown variable name";
2017 return ERR;
2018 }
2020 /* Wants: mode request key */
2021 static int
2022 option_bind_command(int argc, const char *argv[])
2023 {
2024 enum request request;
2025 int keymap = -1;
2026 int key;
2028 if (argc < 3) {
2029 config_msg = "Wrong number of arguments given to bind command";
2030 return ERR;
2031 }
2033 if (!set_keymap(&keymap, argv[0])) {
2034 config_msg = "Unknown key map";
2035 return ERR;
2036 }
2038 key = get_key_value(argv[1]);
2039 if (key == ERR) {
2040 config_msg = "Unknown key";
2041 return ERR;
2042 }
2044 request = get_request(argv[2]);
2045 if (request == REQ_UNKNOWN) {
2046 static const struct enum_map obsolete[] = {
2047 ENUM_MAP("cherry-pick", REQ_NONE),
2048 ENUM_MAP("screen-resize", REQ_NONE),
2049 ENUM_MAP("tree-parent", REQ_PARENT),
2050 };
2051 int alias;
2053 if (map_enum(&alias, obsolete, argv[2])) {
2054 if (alias != REQ_NONE)
2055 add_keybinding(keymap, alias, key);
2056 config_msg = "Obsolete request name";
2057 return ERR;
2058 }
2059 }
2060 if (request == REQ_UNKNOWN && *argv[2]++ == '!')
2061 request = add_run_request(keymap, key, argc - 2, argv + 2);
2062 if (request == REQ_UNKNOWN) {
2063 config_msg = "Unknown request name";
2064 return ERR;
2065 }
2067 add_keybinding(keymap, request, key);
2069 return OK;
2070 }
2072 static int
2073 set_option(const char *opt, char *value)
2074 {
2075 const char *argv[SIZEOF_ARG];
2076 int argc = 0;
2078 if (!argv_from_string(argv, &argc, value)) {
2079 config_msg = "Too many option arguments";
2080 return ERR;
2081 }
2083 if (!strcmp(opt, "color"))
2084 return option_color_command(argc, argv);
2086 if (!strcmp(opt, "set"))
2087 return option_set_command(argc, argv);
2089 if (!strcmp(opt, "bind"))
2090 return option_bind_command(argc, argv);
2092 config_msg = "Unknown option command";
2093 return ERR;
2094 }
2096 static int
2097 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
2098 {
2099 int status = OK;
2101 config_lineno++;
2102 config_msg = "Internal error";
2104 /* Check for comment markers, since read_properties() will
2105 * only ensure opt and value are split at first " \t". */
2106 optlen = strcspn(opt, "#");
2107 if (optlen == 0)
2108 return OK;
2110 if (opt[optlen] != 0) {
2111 config_msg = "No option value";
2112 status = ERR;
2114 } else {
2115 /* Look for comment endings in the value. */
2116 size_t len = strcspn(value, "#");
2118 if (len < valuelen) {
2119 valuelen = len;
2120 value[valuelen] = 0;
2121 }
2123 status = set_option(opt, value);
2124 }
2126 if (status == ERR) {
2127 warn("Error on line %d, near '%.*s': %s",
2128 config_lineno, (int) optlen, opt, config_msg);
2129 config_errors = TRUE;
2130 }
2132 /* Always keep going if errors are encountered. */
2133 return OK;
2134 }
2136 static void
2137 load_option_file(const char *path)
2138 {
2139 struct io io = {};
2141 /* It's OK that the file doesn't exist. */
2142 if (!io_open(&io, "%s", path))
2143 return;
2145 config_lineno = 0;
2146 config_errors = FALSE;
2148 if (io_load(&io, " \t", read_option) == ERR ||
2149 config_errors == TRUE)
2150 warn("Errors while loading %s.", path);
2151 }
2153 static int
2154 load_options(void)
2155 {
2156 const char *home = getenv("HOME");
2157 const char *tigrc_user = getenv("TIGRC_USER");
2158 const char *tigrc_system = getenv("TIGRC_SYSTEM");
2159 char buf[SIZEOF_STR];
2161 if (!tigrc_system)
2162 tigrc_system = SYSCONFDIR "/tigrc";
2163 load_option_file(tigrc_system);
2165 if (!tigrc_user) {
2166 if (!home || !string_format(buf, "%s/.tigrc", home))
2167 return ERR;
2168 tigrc_user = buf;
2169 }
2170 load_option_file(tigrc_user);
2172 /* Add _after_ loading config files to avoid adding run requests
2173 * that conflict with keybindings. */
2174 add_builtin_run_requests();
2176 return OK;
2177 }
2180 /*
2181 * The viewer
2182 */
2184 struct view;
2185 struct view_ops;
2187 /* The display array of active views and the index of the current view. */
2188 static struct view *display[2];
2189 static unsigned int current_view;
2191 #define foreach_displayed_view(view, i) \
2192 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
2194 #define displayed_views() (display[1] != NULL ? 2 : 1)
2196 /* Current head and commit ID */
2197 static char ref_blob[SIZEOF_REF] = "";
2198 static char ref_commit[SIZEOF_REF] = "HEAD";
2199 static char ref_head[SIZEOF_REF] = "HEAD";
2200 static char ref_branch[SIZEOF_REF] = "";
2202 enum view_type {
2203 VIEW_MAIN,
2204 VIEW_DIFF,
2205 VIEW_LOG,
2206 VIEW_TREE,
2207 VIEW_BLOB,
2208 VIEW_BLAME,
2209 VIEW_BRANCH,
2210 VIEW_HELP,
2211 VIEW_PAGER,
2212 VIEW_STATUS,
2213 VIEW_STAGE,
2214 };
2216 struct view {
2217 enum view_type type; /* View type */
2218 const char *name; /* View name */
2219 const char *cmd_env; /* Command line set via environment */
2220 const char *id; /* Points to either of ref_{head,commit,blob} */
2222 struct view_ops *ops; /* View operations */
2224 enum keymap keymap; /* What keymap does this view have */
2225 bool git_dir; /* Whether the view requires a git directory. */
2227 char ref[SIZEOF_REF]; /* Hovered commit reference */
2228 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
2230 int height, width; /* The width and height of the main window */
2231 WINDOW *win; /* The main window */
2232 WINDOW *title; /* The title window living below the main window */
2234 /* Navigation */
2235 unsigned long offset; /* Offset of the window top */
2236 unsigned long yoffset; /* Offset from the window side. */
2237 unsigned long lineno; /* Current line number */
2238 unsigned long p_offset; /* Previous offset of the window top */
2239 unsigned long p_yoffset;/* Previous offset from the window side */
2240 unsigned long p_lineno; /* Previous current line number */
2241 bool p_restore; /* Should the previous position be restored. */
2243 /* Searching */
2244 char grep[SIZEOF_STR]; /* Search string */
2245 regex_t *regex; /* Pre-compiled regexp */
2247 /* If non-NULL, points to the view that opened this view. If this view
2248 * is closed tig will switch back to the parent view. */
2249 struct view *parent;
2250 struct view *prev;
2252 /* Buffering */
2253 size_t lines; /* Total number of lines */
2254 struct line *line; /* Line index */
2255 unsigned int digits; /* Number of digits in the lines member. */
2257 /* Drawing */
2258 struct line *curline; /* Line currently being drawn. */
2259 enum line_type curtype; /* Attribute currently used for drawing. */
2260 unsigned long col; /* Column when drawing. */
2261 bool has_scrolled; /* View was scrolled. */
2263 /* Loading */
2264 struct io io;
2265 struct io *pipe;
2266 time_t start_time;
2267 time_t update_secs;
2268 };
2270 struct view_ops {
2271 /* What type of content being displayed. Used in the title bar. */
2272 const char *type;
2273 /* Default command arguments. */
2274 const char **argv;
2275 /* Open and reads in all view content. */
2276 bool (*open)(struct view *view);
2277 /* Read one line; updates view->line. */
2278 bool (*read)(struct view *view, char *data);
2279 /* Draw one line; @lineno must be < view->height. */
2280 bool (*draw)(struct view *view, struct line *line, unsigned int lineno);
2281 /* Depending on view handle a special requests. */
2282 enum request (*request)(struct view *view, enum request request, struct line *line);
2283 /* Search for regexp in a line. */
2284 bool (*grep)(struct view *view, struct line *line);
2285 /* Select line */
2286 void (*select)(struct view *view, struct line *line);
2287 /* Prepare view for loading */
2288 bool (*prepare)(struct view *view);
2289 };
2291 static struct view_ops blame_ops;
2292 static struct view_ops blob_ops;
2293 static struct view_ops diff_ops;
2294 static struct view_ops help_ops;
2295 static struct view_ops log_ops;
2296 static struct view_ops main_ops;
2297 static struct view_ops pager_ops;
2298 static struct view_ops stage_ops;
2299 static struct view_ops status_ops;
2300 static struct view_ops tree_ops;
2301 static struct view_ops branch_ops;
2303 #define VIEW_STR(type, name, env, ref, ops, map, git) \
2304 { type, name, #env, ref, ops, map, git }
2306 #define VIEW_(id, name, ops, git, ref) \
2307 VIEW_STR(VIEW_##id, name, TIG_##id##_CMD, ref, ops, KEYMAP_##id, git)
2309 static struct view views[] = {
2310 VIEW_(MAIN, "main", &main_ops, TRUE, ref_head),
2311 VIEW_(DIFF, "diff", &diff_ops, TRUE, ref_commit),
2312 VIEW_(LOG, "log", &log_ops, TRUE, ref_head),
2313 VIEW_(TREE, "tree", &tree_ops, TRUE, ref_commit),
2314 VIEW_(BLOB, "blob", &blob_ops, TRUE, ref_blob),
2315 VIEW_(BLAME, "blame", &blame_ops, TRUE, ref_commit),
2316 VIEW_(BRANCH, "branch", &branch_ops, TRUE, ref_head),
2317 VIEW_(HELP, "help", &help_ops, FALSE, ""),
2318 VIEW_(PAGER, "pager", &pager_ops, FALSE, "stdin"),
2319 VIEW_(STATUS, "status", &status_ops, TRUE, ""),
2320 VIEW_(STAGE, "stage", &stage_ops, TRUE, ""),
2321 };
2323 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
2325 #define foreach_view(view, i) \
2326 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
2328 #define view_is_displayed(view) \
2329 (view == display[0] || view == display[1])
2331 static enum request
2332 view_request(struct view *view, enum request request)
2333 {
2334 if (!view || !view->lines)
2335 return request;
2336 return view->ops->request(view, request, &view->line[view->lineno]);
2337 }
2340 /*
2341 * View drawing.
2342 */
2344 static inline void
2345 set_view_attr(struct view *view, enum line_type type)
2346 {
2347 if (!view->curline->selected && view->curtype != type) {
2348 (void) wattrset(view->win, get_line_attr(type));
2349 wchgat(view->win, -1, 0, type, NULL);
2350 view->curtype = type;
2351 }
2352 }
2354 static int
2355 draw_chars(struct view *view, enum line_type type, const char *string,
2356 int max_len, bool use_tilde)
2357 {
2358 static char out_buffer[BUFSIZ * 2];
2359 int len = 0;
2360 int col = 0;
2361 int trimmed = FALSE;
2362 size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
2364 if (max_len <= 0)
2365 return 0;
2367 len = utf8_length(&string, skip, &col, max_len, &trimmed, use_tilde, opt_tab_size);
2369 set_view_attr(view, type);
2370 if (len > 0) {
2371 if (opt_iconv_out != ICONV_NONE) {
2372 ICONV_CONST char *inbuf = (ICONV_CONST char *) string;
2373 size_t inlen = len + 1;
2375 char *outbuf = out_buffer;
2376 size_t outlen = sizeof(out_buffer);
2378 size_t ret;
2380 ret = iconv(opt_iconv_out, &inbuf, &inlen, &outbuf, &outlen);
2381 if (ret != (size_t) -1) {
2382 string = out_buffer;
2383 len = sizeof(out_buffer) - outlen;
2384 }
2385 }
2387 waddnstr(view->win, string, len);
2388 }
2389 if (trimmed && use_tilde) {
2390 set_view_attr(view, LINE_DELIMITER);
2391 waddch(view->win, '~');
2392 col++;
2393 }
2395 return col;
2396 }
2398 static int
2399 draw_space(struct view *view, enum line_type type, int max, int spaces)
2400 {
2401 static char space[] = " ";
2402 int col = 0;
2404 spaces = MIN(max, spaces);
2406 while (spaces > 0) {
2407 int len = MIN(spaces, sizeof(space) - 1);
2409 col += draw_chars(view, type, space, len, FALSE);
2410 spaces -= len;
2411 }
2413 return col;
2414 }
2416 static bool
2417 draw_text(struct view *view, enum line_type type, const char *string, bool trim)
2418 {
2419 view->col += draw_chars(view, type, string, view->width + view->yoffset - view->col, trim);
2420 return view->width + view->yoffset <= view->col;
2421 }
2423 static bool
2424 draw_graphic(struct view *view, enum line_type type, chtype graphic[], size_t size)
2425 {
2426 size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
2427 int max = view->width + view->yoffset - view->col;
2428 int i;
2430 if (max < size)
2431 size = max;
2433 set_view_attr(view, type);
2434 /* Using waddch() instead of waddnstr() ensures that
2435 * they'll be rendered correctly for the cursor line. */
2436 for (i = skip; i < size; i++)
2437 waddch(view->win, graphic[i]);
2439 view->col += size;
2440 if (size < max && skip <= size)
2441 waddch(view->win, ' ');
2442 view->col++;
2444 return view->width + view->yoffset <= view->col;
2445 }
2447 static bool
2448 draw_field(struct view *view, enum line_type type, const char *text, int len, bool trim)
2449 {
2450 int max = MIN(view->width + view->yoffset - view->col, len);
2451 int col;
2453 if (text)
2454 col = draw_chars(view, type, text, max - 1, trim);
2455 else
2456 col = draw_space(view, type, max - 1, max - 1);
2458 view->col += col;
2459 view->col += draw_space(view, LINE_DEFAULT, max - col, max - col);
2460 return view->width + view->yoffset <= view->col;
2461 }
2463 static bool
2464 draw_date(struct view *view, struct time *time)
2465 {
2466 const char *date = mkdate(time, opt_date);
2467 int cols = opt_date == DATE_SHORT ? DATE_SHORT_COLS : DATE_COLS;
2469 return draw_field(view, LINE_DATE, date, cols, FALSE);
2470 }
2472 static bool
2473 draw_author(struct view *view, const char *author)
2474 {
2475 bool trim = opt_author_cols == 0 || opt_author_cols > 5;
2476 bool abbreviate = opt_author == AUTHOR_ABBREVIATED || !trim;
2478 if (abbreviate && author)
2479 author = get_author_initials(author);
2481 return draw_field(view, LINE_AUTHOR, author, opt_author_cols, trim);
2482 }
2484 static bool
2485 draw_mode(struct view *view, mode_t mode)
2486 {
2487 const char *str;
2489 if (S_ISDIR(mode))
2490 str = "drwxr-xr-x";
2491 else if (S_ISLNK(mode))
2492 str = "lrwxrwxrwx";
2493 else if (S_ISGITLINK(mode))
2494 str = "m---------";
2495 else if (S_ISREG(mode) && mode & S_IXUSR)
2496 str = "-rwxr-xr-x";
2497 else if (S_ISREG(mode))
2498 str = "-rw-r--r--";
2499 else
2500 str = "----------";
2502 return draw_field(view, LINE_MODE, str, STRING_SIZE("-rw-r--r-- "), FALSE);
2503 }
2505 static bool
2506 draw_lineno(struct view *view, unsigned int lineno)
2507 {
2508 char number[10];
2509 int digits3 = view->digits < 3 ? 3 : view->digits;
2510 int max = MIN(view->width + view->yoffset - view->col, digits3);
2511 char *text = NULL;
2512 chtype separator = opt_line_graphics ? ACS_VLINE : '|';
2514 lineno += view->offset + 1;
2515 if (lineno == 1 || (lineno % opt_num_interval) == 0) {
2516 static char fmt[] = "%1ld";
2518 fmt[1] = '0' + (view->digits <= 9 ? digits3 : 1);
2519 if (string_format(number, fmt, lineno))
2520 text = number;
2521 }
2522 if (text)
2523 view->col += draw_chars(view, LINE_LINE_NUMBER, text, max, TRUE);
2524 else
2525 view->col += draw_space(view, LINE_LINE_NUMBER, max, digits3);
2526 return draw_graphic(view, LINE_DEFAULT, &separator, 1);
2527 }
2529 static bool
2530 draw_view_line(struct view *view, unsigned int lineno)
2531 {
2532 struct line *line;
2533 bool selected = (view->offset + lineno == view->lineno);
2535 assert(view_is_displayed(view));
2537 if (view->offset + lineno >= view->lines)
2538 return FALSE;
2540 line = &view->line[view->offset + lineno];
2542 wmove(view->win, lineno, 0);
2543 if (line->cleareol)
2544 wclrtoeol(view->win);
2545 view->col = 0;
2546 view->curline = line;
2547 view->curtype = LINE_NONE;
2548 line->selected = FALSE;
2549 line->dirty = line->cleareol = 0;
2551 if (selected) {
2552 set_view_attr(view, LINE_CURSOR);
2553 line->selected = TRUE;
2554 view->ops->select(view, line);
2555 }
2557 return view->ops->draw(view, line, lineno);
2558 }
2560 static void
2561 redraw_view_dirty(struct view *view)
2562 {
2563 bool dirty = FALSE;
2564 int lineno;
2566 for (lineno = 0; lineno < view->height; lineno++) {
2567 if (view->offset + lineno >= view->lines)
2568 break;
2569 if (!view->line[view->offset + lineno].dirty)
2570 continue;
2571 dirty = TRUE;
2572 if (!draw_view_line(view, lineno))
2573 break;
2574 }
2576 if (!dirty)
2577 return;
2578 wnoutrefresh(view->win);
2579 }
2581 static void
2582 redraw_view_from(struct view *view, int lineno)
2583 {
2584 assert(0 <= lineno && lineno < view->height);
2586 for (; lineno < view->height; lineno++) {
2587 if (!draw_view_line(view, lineno))
2588 break;
2589 }
2591 wnoutrefresh(view->win);
2592 }
2594 static void
2595 redraw_view(struct view *view)
2596 {
2597 werase(view->win);
2598 redraw_view_from(view, 0);
2599 }
2602 static void
2603 update_view_title(struct view *view)
2604 {
2605 char buf[SIZEOF_STR];
2606 char state[SIZEOF_STR];
2607 size_t bufpos = 0, statelen = 0;
2609 assert(view_is_displayed(view));
2611 if (view->type != VIEW_STATUS && view->lines) {
2612 unsigned int view_lines = view->offset + view->height;
2613 unsigned int lines = view->lines
2614 ? MIN(view_lines, view->lines) * 100 / view->lines
2615 : 0;
2617 string_format_from(state, &statelen, " - %s %d of %d (%d%%)",
2618 view->ops->type,
2619 view->lineno + 1,
2620 view->lines,
2621 lines);
2623 }
2625 if (view->pipe) {
2626 time_t secs = time(NULL) - view->start_time;
2628 /* Three git seconds are a long time ... */
2629 if (secs > 2)
2630 string_format_from(state, &statelen, " loading %lds", secs);
2631 }
2633 string_format_from(buf, &bufpos, "[%s]", view->name);
2634 if (*view->ref && bufpos < view->width) {
2635 size_t refsize = strlen(view->ref);
2636 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
2638 if (minsize < view->width)
2639 refsize = view->width - minsize + 7;
2640 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
2641 }
2643 if (statelen && bufpos < view->width) {
2644 string_format_from(buf, &bufpos, "%s", state);
2645 }
2647 if (view == display[current_view])
2648 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
2649 else
2650 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
2652 mvwaddnstr(view->title, 0, 0, buf, bufpos);
2653 wclrtoeol(view->title);
2654 wnoutrefresh(view->title);
2655 }
2657 static int
2658 apply_step(double step, int value)
2659 {
2660 if (step >= 1)
2661 return (int) step;
2662 value *= step + 0.01;
2663 return value ? value : 1;
2664 }
2666 static void
2667 resize_display(void)
2668 {
2669 int offset, i;
2670 struct view *base = display[0];
2671 struct view *view = display[1] ? display[1] : display[0];
2673 /* Setup window dimensions */
2675 getmaxyx(stdscr, base->height, base->width);
2677 /* Make room for the status window. */
2678 base->height -= 1;
2680 if (view != base) {
2681 /* Horizontal split. */
2682 view->width = base->width;
2683 view->height = apply_step(opt_scale_split_view, base->height);
2684 view->height = MAX(view->height, MIN_VIEW_HEIGHT);
2685 view->height = MIN(view->height, base->height - MIN_VIEW_HEIGHT);
2686 base->height -= view->height;
2688 /* Make room for the title bar. */
2689 view->height -= 1;
2690 }
2692 /* Make room for the title bar. */
2693 base->height -= 1;
2695 offset = 0;
2697 foreach_displayed_view (view, i) {
2698 if (!view->win) {
2699 view->win = newwin(view->height, 0, offset, 0);
2700 if (!view->win)
2701 die("Failed to create %s view", view->name);
2703 scrollok(view->win, FALSE);
2705 view->title = newwin(1, 0, offset + view->height, 0);
2706 if (!view->title)
2707 die("Failed to create title window");
2709 } else {
2710 wresize(view->win, view->height, view->width);
2711 mvwin(view->win, offset, 0);
2712 mvwin(view->title, offset + view->height, 0);
2713 }
2715 offset += view->height + 1;
2716 }
2717 }
2719 static void
2720 redraw_display(bool clear)
2721 {
2722 struct view *view;
2723 int i;
2725 foreach_displayed_view (view, i) {
2726 if (clear)
2727 wclear(view->win);
2728 redraw_view(view);
2729 update_view_title(view);
2730 }
2731 }
2734 /*
2735 * Option management
2736 */
2738 static void
2739 toggle_enum_option_do(unsigned int *opt, const char *help,
2740 const struct enum_map *map, size_t size)
2741 {
2742 *opt = (*opt + 1) % size;
2743 redraw_display(FALSE);
2744 report("Displaying %s %s", enum_name(map[*opt]), help);
2745 }
2747 #define toggle_enum_option(opt, help, map) \
2748 toggle_enum_option_do(opt, help, map, ARRAY_SIZE(map))
2750 #define toggle_date() toggle_enum_option(&opt_date, "dates", date_map)
2751 #define toggle_author() toggle_enum_option(&opt_author, "author names", author_map)
2753 static void
2754 toggle_view_option(bool *option, const char *help)
2755 {
2756 *option = !*option;
2757 redraw_display(FALSE);
2758 report("%sabling %s", *option ? "En" : "Dis", help);
2759 }
2761 static void
2762 open_option_menu(void)
2763 {
2764 const struct menu_item menu[] = {
2765 { '.', "line numbers", &opt_line_number },
2766 { 'D', "date display", &opt_date },
2767 { 'A', "author display", &opt_author },
2768 { 'g', "revision graph display", &opt_rev_graph },
2769 { 'F', "reference display", &opt_show_refs },
2770 { 0 }
2771 };
2772 int selected = 0;
2774 if (prompt_menu("Toggle option", menu, &selected)) {
2775 if (menu[selected].data == &opt_date)
2776 toggle_date();
2777 else if (menu[selected].data == &opt_author)
2778 toggle_author();
2779 else
2780 toggle_view_option(menu[selected].data, menu[selected].text);
2781 }
2782 }
2784 static void
2785 maximize_view(struct view *view)
2786 {
2787 memset(display, 0, sizeof(display));
2788 current_view = 0;
2789 display[current_view] = view;
2790 resize_display();
2791 redraw_display(FALSE);
2792 report("");
2793 }
2796 /*
2797 * Navigation
2798 */
2800 static bool
2801 goto_view_line(struct view *view, unsigned long offset, unsigned long lineno)
2802 {
2803 if (lineno >= view->lines)
2804 lineno = view->lines > 0 ? view->lines - 1 : 0;
2806 if (offset > lineno || offset + view->height <= lineno) {
2807 unsigned long half = view->height / 2;
2809 if (lineno > half)
2810 offset = lineno - half;
2811 else
2812 offset = 0;
2813 }
2815 if (offset != view->offset || lineno != view->lineno) {
2816 view->offset = offset;
2817 view->lineno = lineno;
2818 return TRUE;
2819 }
2821 return FALSE;
2822 }
2824 /* Scrolling backend */
2825 static void
2826 do_scroll_view(struct view *view, int lines)
2827 {
2828 bool redraw_current_line = FALSE;
2830 /* The rendering expects the new offset. */
2831 view->offset += lines;
2833 assert(0 <= view->offset && view->offset < view->lines);
2834 assert(lines);
2836 /* Move current line into the view. */
2837 if (view->lineno < view->offset) {
2838 view->lineno = view->offset;
2839 redraw_current_line = TRUE;
2840 } else if (view->lineno >= view->offset + view->height) {
2841 view->lineno = view->offset + view->height - 1;
2842 redraw_current_line = TRUE;
2843 }
2845 assert(view->offset <= view->lineno && view->lineno < view->lines);
2847 /* Redraw the whole screen if scrolling is pointless. */
2848 if (view->height < ABS(lines)) {
2849 redraw_view(view);
2851 } else {
2852 int line = lines > 0 ? view->height - lines : 0;
2853 int end = line + ABS(lines);
2855 scrollok(view->win, TRUE);
2856 wscrl(view->win, lines);
2857 scrollok(view->win, FALSE);
2859 while (line < end && draw_view_line(view, line))
2860 line++;
2862 if (redraw_current_line)
2863 draw_view_line(view, view->lineno - view->offset);
2864 wnoutrefresh(view->win);
2865 }
2867 view->has_scrolled = TRUE;
2868 report("");
2869 }
2871 /* Scroll frontend */
2872 static void
2873 scroll_view(struct view *view, enum request request)
2874 {
2875 int lines = 1;
2877 assert(view_is_displayed(view));
2879 switch (request) {
2880 case REQ_SCROLL_LEFT:
2881 if (view->yoffset == 0) {
2882 report("Cannot scroll beyond the first column");
2883 return;
2884 }
2885 if (view->yoffset <= apply_step(opt_hscroll, view->width))
2886 view->yoffset = 0;
2887 else
2888 view->yoffset -= apply_step(opt_hscroll, view->width);
2889 redraw_view_from(view, 0);
2890 report("");
2891 return;
2892 case REQ_SCROLL_RIGHT:
2893 view->yoffset += apply_step(opt_hscroll, view->width);
2894 redraw_view(view);
2895 report("");
2896 return;
2897 case REQ_SCROLL_PAGE_DOWN:
2898 lines = view->height;
2899 case REQ_SCROLL_LINE_DOWN:
2900 if (view->offset + lines > view->lines)
2901 lines = view->lines - view->offset;
2903 if (lines == 0 || view->offset + view->height >= view->lines) {
2904 report("Cannot scroll beyond the last line");
2905 return;
2906 }
2907 break;
2909 case REQ_SCROLL_PAGE_UP:
2910 lines = view->height;
2911 case REQ_SCROLL_LINE_UP:
2912 if (lines > view->offset)
2913 lines = view->offset;
2915 if (lines == 0) {
2916 report("Cannot scroll beyond the first line");
2917 return;
2918 }
2920 lines = -lines;
2921 break;
2923 default:
2924 die("request %d not handled in switch", request);
2925 }
2927 do_scroll_view(view, lines);
2928 }
2930 /* Cursor moving */
2931 static void
2932 move_view(struct view *view, enum request request)
2933 {
2934 int scroll_steps = 0;
2935 int steps;
2937 switch (request) {
2938 case REQ_MOVE_FIRST_LINE:
2939 steps = -view->lineno;
2940 break;
2942 case REQ_MOVE_LAST_LINE:
2943 steps = view->lines - view->lineno - 1;
2944 break;
2946 case REQ_MOVE_PAGE_UP:
2947 steps = view->height > view->lineno
2948 ? -view->lineno : -view->height;
2949 break;
2951 case REQ_MOVE_PAGE_DOWN:
2952 steps = view->lineno + view->height >= view->lines
2953 ? view->lines - view->lineno - 1 : view->height;
2954 break;
2956 case REQ_MOVE_UP:
2957 steps = -1;
2958 break;
2960 case REQ_MOVE_DOWN:
2961 steps = 1;
2962 break;
2964 default:
2965 die("request %d not handled in switch", request);
2966 }
2968 if (steps <= 0 && view->lineno == 0) {
2969 report("Cannot move beyond the first line");
2970 return;
2972 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
2973 report("Cannot move beyond the last line");
2974 return;
2975 }
2977 /* Move the current line */
2978 view->lineno += steps;
2979 assert(0 <= view->lineno && view->lineno < view->lines);
2981 /* Check whether the view needs to be scrolled */
2982 if (view->lineno < view->offset ||
2983 view->lineno >= view->offset + view->height) {
2984 scroll_steps = steps;
2985 if (steps < 0 && -steps > view->offset) {
2986 scroll_steps = -view->offset;
2988 } else if (steps > 0) {
2989 if (view->lineno == view->lines - 1 &&
2990 view->lines > view->height) {
2991 scroll_steps = view->lines - view->offset - 1;
2992 if (scroll_steps >= view->height)
2993 scroll_steps -= view->height - 1;
2994 }
2995 }
2996 }
2998 if (!view_is_displayed(view)) {
2999 view->offset += scroll_steps;
3000 assert(0 <= view->offset && view->offset < view->lines);
3001 view->ops->select(view, &view->line[view->lineno]);
3002 return;
3003 }
3005 /* Repaint the old "current" line if we be scrolling */
3006 if (ABS(steps) < view->height)
3007 draw_view_line(view, view->lineno - steps - view->offset);
3009 if (scroll_steps) {
3010 do_scroll_view(view, scroll_steps);
3011 return;
3012 }
3014 /* Draw the current line */
3015 draw_view_line(view, view->lineno - view->offset);
3017 wnoutrefresh(view->win);
3018 report("");
3019 }
3022 /*
3023 * Searching
3024 */
3026 static void search_view(struct view *view, enum request request);
3028 static bool
3029 grep_text(struct view *view, const char *text[])
3030 {
3031 regmatch_t pmatch;
3032 size_t i;
3034 for (i = 0; text[i]; i++)
3035 if (*text[i] &&
3036 regexec(view->regex, text[i], 1, &pmatch, 0) != REG_NOMATCH)
3037 return TRUE;
3038 return FALSE;
3039 }
3041 static void
3042 select_view_line(struct view *view, unsigned long lineno)
3043 {
3044 unsigned long old_lineno = view->lineno;
3045 unsigned long old_offset = view->offset;
3047 if (goto_view_line(view, view->offset, lineno)) {
3048 if (view_is_displayed(view)) {
3049 if (old_offset != view->offset) {
3050 redraw_view(view);
3051 } else {
3052 draw_view_line(view, old_lineno - view->offset);
3053 draw_view_line(view, view->lineno - view->offset);
3054 wnoutrefresh(view->win);
3055 }
3056 } else {
3057 view->ops->select(view, &view->line[view->lineno]);
3058 }
3059 }
3060 }
3062 static void
3063 find_next(struct view *view, enum request request)
3064 {
3065 unsigned long lineno = view->lineno;
3066 int direction;
3068 if (!*view->grep) {
3069 if (!*opt_search)
3070 report("No previous search");
3071 else
3072 search_view(view, request);
3073 return;
3074 }
3076 switch (request) {
3077 case REQ_SEARCH:
3078 case REQ_FIND_NEXT:
3079 direction = 1;
3080 break;
3082 case REQ_SEARCH_BACK:
3083 case REQ_FIND_PREV:
3084 direction = -1;
3085 break;
3087 default:
3088 return;
3089 }
3091 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
3092 lineno += direction;
3094 /* Note, lineno is unsigned long so will wrap around in which case it
3095 * will become bigger than view->lines. */
3096 for (; lineno < view->lines; lineno += direction) {
3097 if (view->ops->grep(view, &view->line[lineno])) {
3098 select_view_line(view, lineno);
3099 report("Line %ld matches '%s'", lineno + 1, view->grep);
3100 return;
3101 }
3102 }
3104 report("No match found for '%s'", view->grep);
3105 }
3107 static void
3108 search_view(struct view *view, enum request request)
3109 {
3110 int regex_err;
3112 if (view->regex) {
3113 regfree(view->regex);
3114 *view->grep = 0;
3115 } else {
3116 view->regex = calloc(1, sizeof(*view->regex));
3117 if (!view->regex)
3118 return;
3119 }
3121 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
3122 if (regex_err != 0) {
3123 char buf[SIZEOF_STR] = "unknown error";
3125 regerror(regex_err, view->regex, buf, sizeof(buf));
3126 report("Search failed: %s", buf);
3127 return;
3128 }
3130 string_copy(view->grep, opt_search);
3132 find_next(view, request);
3133 }
3135 /*
3136 * Incremental updating
3137 */
3139 static void
3140 reset_view(struct view *view)
3141 {
3142 int i;
3144 for (i = 0; i < view->lines; i++)
3145 free(view->line[i].data);
3146 free(view->line);
3148 view->p_offset = view->offset;
3149 view->p_yoffset = view->yoffset;
3150 view->p_lineno = view->lineno;
3152 view->line = NULL;
3153 view->offset = 0;
3154 view->yoffset = 0;
3155 view->lines = 0;
3156 view->lineno = 0;
3157 view->vid[0] = 0;
3158 view->update_secs = 0;
3159 }
3161 static const char *
3162 format_arg(const char *name)
3163 {
3164 static struct {
3165 const char *name;
3166 size_t namelen;
3167 const char *value;
3168 const char *value_if_empty;
3169 } vars[] = {
3170 #define FORMAT_VAR(name, value, value_if_empty) \
3171 { name, STRING_SIZE(name), value, value_if_empty }
3172 FORMAT_VAR("%(directory)", opt_path, ""),
3173 FORMAT_VAR("%(file)", opt_file, ""),
3174 FORMAT_VAR("%(ref)", opt_ref, "HEAD"),
3175 FORMAT_VAR("%(head)", ref_head, ""),
3176 FORMAT_VAR("%(commit)", ref_commit, ""),
3177 FORMAT_VAR("%(blob)", ref_blob, ""),
3178 FORMAT_VAR("%(branch)", ref_branch, ""),
3179 };
3180 int i;
3182 for (i = 0; i < ARRAY_SIZE(vars); i++)
3183 if (!strncmp(name, vars[i].name, vars[i].namelen))
3184 return *vars[i].value ? vars[i].value : vars[i].value_if_empty;
3186 report("Unknown replacement: `%s`", name);
3187 return NULL;
3188 }
3190 static bool
3191 format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags flags)
3192 {
3193 char buf[SIZEOF_STR];
3194 int argc;
3195 bool noreplace = flags == FORMAT_NONE;
3197 argv_free(dst_argv);
3199 for (argc = 0; src_argv[argc]; argc++) {
3200 const char *arg = src_argv[argc];
3201 size_t bufpos = 0;
3203 while (arg) {
3204 char *next = strstr(arg, "%(");
3205 int len = next - arg;
3206 const char *value;
3208 if (!next || noreplace) {
3209 len = strlen(arg);
3210 value = "";
3212 } else {
3213 value = format_arg(next);
3215 if (!value) {
3216 return FALSE;
3217 }
3218 }
3220 if (!string_format_from(buf, &bufpos, "%.*s%s", len, arg, value))
3221 return FALSE;
3223 arg = next && !noreplace ? strchr(next, ')') + 1 : NULL;
3224 }
3226 dst_argv[argc] = strdup(buf);
3227 if (!dst_argv[argc])
3228 break;
3229 }
3231 dst_argv[argc] = NULL;
3233 return src_argv[argc] == NULL;
3234 }
3236 static bool
3237 restore_view_position(struct view *view)
3238 {
3239 if (!view->p_restore || (view->pipe && view->lines <= view->p_lineno))
3240 return FALSE;
3242 /* Changing the view position cancels the restoring. */
3243 /* FIXME: Changing back to the first line is not detected. */
3244 if (view->offset != 0 || view->lineno != 0) {
3245 view->p_restore = FALSE;
3246 return FALSE;
3247 }
3249 if (goto_view_line(view, view->p_offset, view->p_lineno) &&
3250 view_is_displayed(view))
3251 werase(view->win);
3253 view->yoffset = view->p_yoffset;
3254 view->p_restore = FALSE;
3256 return TRUE;
3257 }
3259 static void
3260 end_update(struct view *view, bool force)
3261 {
3262 if (!view->pipe)
3263 return;
3264 while (!view->ops->read(view, NULL))
3265 if (!force)
3266 return;
3267 if (force)
3268 io_kill(view->pipe);
3269 io_done(view->pipe);
3270 view->pipe = NULL;
3271 }
3273 static void
3274 setup_update(struct view *view, const char *vid)
3275 {
3276 reset_view(view);
3277 string_copy_rev(view->vid, vid);
3278 view->pipe = &view->io;
3279 view->start_time = time(NULL);
3280 }
3282 static bool
3283 prepare_update(struct view *view, const char *argv[], const char *dir)
3284 {
3285 if (view->pipe)
3286 end_update(view, TRUE);
3287 return io_format(&view->io, dir, IO_RD, argv, FORMAT_NONE);
3288 }
3290 static bool
3291 prepare_update_file(struct view *view, const char *name)
3292 {
3293 if (view->pipe)
3294 end_update(view, TRUE);
3295 return io_open(&view->io, "%s/%s", opt_cdup[0] ? opt_cdup : ".", name);
3296 }
3298 static bool
3299 begin_update(struct view *view, bool refresh)
3300 {
3301 if (view->pipe)
3302 end_update(view, TRUE);
3304 if (!refresh) {
3305 if (view->ops->prepare) {
3306 if (!view->ops->prepare(view))
3307 return FALSE;
3308 } else if (!io_format(&view->io, NULL, IO_RD, view->ops->argv, FORMAT_ALL)) {
3309 return FALSE;
3310 }
3312 /* Put the current ref_* value to the view title ref
3313 * member. This is needed by the blob view. Most other
3314 * views sets it automatically after loading because the
3315 * first line is a commit line. */
3316 string_copy_rev(view->ref, view->id);
3317 }
3319 if (!io_start(&view->io))
3320 return FALSE;
3322 setup_update(view, view->id);
3324 return TRUE;
3325 }
3327 static bool
3328 update_view(struct view *view)
3329 {
3330 char out_buffer[BUFSIZ * 2];
3331 char *line;
3332 /* Clear the view and redraw everything since the tree sorting
3333 * might have rearranged things. */
3334 bool redraw = view->lines == 0;
3335 bool can_read = TRUE;
3337 if (!view->pipe)
3338 return TRUE;
3340 if (!io_can_read(view->pipe)) {
3341 if (view->lines == 0 && view_is_displayed(view)) {
3342 time_t secs = time(NULL) - view->start_time;
3344 if (secs > 1 && secs > view->update_secs) {
3345 if (view->update_secs == 0)
3346 redraw_view(view);
3347 update_view_title(view);
3348 view->update_secs = secs;
3349 }
3350 }
3351 return TRUE;
3352 }
3354 for (; (line = io_get(view->pipe, '\n', can_read)); can_read = FALSE) {
3355 if (opt_iconv_in != ICONV_NONE) {
3356 ICONV_CONST char *inbuf = line;
3357 size_t inlen = strlen(line) + 1;
3359 char *outbuf = out_buffer;
3360 size_t outlen = sizeof(out_buffer);
3362 size_t ret;
3364 ret = iconv(opt_iconv_in, &inbuf, &inlen, &outbuf, &outlen);
3365 if (ret != (size_t) -1)
3366 line = out_buffer;
3367 }
3369 if (!view->ops->read(view, line)) {
3370 report("Allocation failure");
3371 end_update(view, TRUE);
3372 return FALSE;
3373 }
3374 }
3376 {
3377 unsigned long lines = view->lines;
3378 int digits;
3380 for (digits = 0; lines; digits++)
3381 lines /= 10;
3383 /* Keep the displayed view in sync with line number scaling. */
3384 if (digits != view->digits) {
3385 view->digits = digits;
3386 if (opt_line_number || view->type == VIEW_BLAME)
3387 redraw = TRUE;
3388 }
3389 }
3391 if (io_error(view->pipe)) {
3392 report("Failed to read: %s", io_strerror(view->pipe));
3393 end_update(view, TRUE);
3395 } else if (io_eof(view->pipe)) {
3396 if (view_is_displayed(view))
3397 report("");
3398 end_update(view, FALSE);
3399 }
3401 if (restore_view_position(view))
3402 redraw = TRUE;
3404 if (!view_is_displayed(view))
3405 return TRUE;
3407 if (redraw)
3408 redraw_view_from(view, 0);
3409 else
3410 redraw_view_dirty(view);
3412 /* Update the title _after_ the redraw so that if the redraw picks up a
3413 * commit reference in view->ref it'll be available here. */
3414 update_view_title(view);
3415 return TRUE;
3416 }
3418 DEFINE_ALLOCATOR(realloc_lines, struct line, 256)
3420 static struct line *
3421 add_line_data(struct view *view, void *data, enum line_type type)
3422 {
3423 struct line *line;
3425 if (!realloc_lines(&view->line, view->lines, 1))
3426 return NULL;
3428 line = &view->line[view->lines++];
3429 memset(line, 0, sizeof(*line));
3430 line->type = type;
3431 line->data = data;
3432 line->dirty = 1;
3434 return line;
3435 }
3437 static struct line *
3438 add_line_text(struct view *view, const char *text, enum line_type type)
3439 {
3440 char *data = text ? strdup(text) : NULL;
3442 return data ? add_line_data(view, data, type) : NULL;
3443 }
3445 static struct line *
3446 add_line_format(struct view *view, enum line_type type, const char *fmt, ...)
3447 {
3448 char buf[SIZEOF_STR];
3449 va_list args;
3451 va_start(args, fmt);
3452 if (vsnprintf(buf, sizeof(buf), fmt, args) >= sizeof(buf))
3453 buf[0] = 0;
3454 va_end(args);
3456 return buf[0] ? add_line_text(view, buf, type) : NULL;
3457 }
3459 /*
3460 * View opening
3461 */
3463 enum open_flags {
3464 OPEN_DEFAULT = 0, /* Use default view switching. */
3465 OPEN_SPLIT = 1, /* Split current view. */
3466 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
3467 OPEN_REFRESH = 16, /* Refresh view using previous command. */
3468 OPEN_PREPARED = 32, /* Open already prepared command. */
3469 };
3471 static void
3472 open_view(struct view *prev, enum request request, enum open_flags flags)
3473 {
3474 bool split = !!(flags & OPEN_SPLIT);
3475 bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH | OPEN_PREPARED));
3476 bool nomaximize = !!(flags & OPEN_REFRESH);
3477 struct view *view = VIEW(request);
3478 int nviews = displayed_views();
3479 struct view *base_view = display[0];
3481 if (view == prev && nviews == 1 && !reload) {
3482 report("Already in %s view", view->name);
3483 return;
3484 }
3486 if (view->git_dir && !opt_git_dir[0]) {
3487 report("The %s view is disabled in pager view", view->name);
3488 return;
3489 }
3491 if (split) {
3492 display[1] = view;
3493 current_view = 1;
3494 view->parent = prev;
3495 } else if (!nomaximize) {
3496 /* Maximize the current view. */
3497 memset(display, 0, sizeof(display));
3498 current_view = 0;
3499 display[current_view] = view;
3500 }
3502 /* No prev signals that this is the first loaded view. */
3503 if (prev && view != prev) {
3504 view->prev = prev;
3505 }
3507 /* Resize the view when switching between split- and full-screen,
3508 * or when switching between two different full-screen views. */
3509 if (nviews != displayed_views() ||
3510 (nviews == 1 && base_view != display[0]))
3511 resize_display();
3513 if (view->ops->open) {
3514 if (view->pipe)
3515 end_update(view, TRUE);
3516 if (!view->ops->open(view)) {
3517 report("Failed to load %s view", view->name);
3518 return;
3519 }
3520 restore_view_position(view);
3522 } else if ((reload || strcmp(view->vid, view->id)) &&
3523 !begin_update(view, flags & (OPEN_REFRESH | OPEN_PREPARED))) {
3524 report("Failed to load %s view", view->name);
3525 return;
3526 }
3528 if (split && prev->lineno - prev->offset >= prev->height) {
3529 /* Take the title line into account. */
3530 int lines = prev->lineno - prev->offset - prev->height + 1;
3532 /* Scroll the view that was split if the current line is
3533 * outside the new limited view. */
3534 do_scroll_view(prev, lines);
3535 }
3537 if (prev && view != prev && split && view_is_displayed(prev)) {
3538 /* "Blur" the previous view. */
3539 update_view_title(prev);
3540 }
3542 if (view->pipe && view->lines == 0) {
3543 /* Clear the old view and let the incremental updating refill
3544 * the screen. */
3545 werase(view->win);
3546 view->p_restore = flags & (OPEN_RELOAD | OPEN_REFRESH);
3547 report("");
3548 } else if (view_is_displayed(view)) {
3549 redraw_view(view);
3550 report("");
3551 }
3552 }
3554 static void
3555 open_external_viewer(const char *argv[], const char *dir)
3556 {
3557 def_prog_mode(); /* save current tty modes */
3558 endwin(); /* restore original tty modes */
3559 io_run_fg(argv, dir);
3560 fprintf(stderr, "Press Enter to continue");
3561 getc(opt_tty);
3562 reset_prog_mode();
3563 redraw_display(TRUE);
3564 }
3566 static void
3567 open_mergetool(const char *file)
3568 {
3569 const char *mergetool_argv[] = { "git", "mergetool", file, NULL };
3571 open_external_viewer(mergetool_argv, opt_cdup);
3572 }
3574 static void
3575 open_editor(const char *file)
3576 {
3577 const char *editor_argv[] = { "vi", file, NULL };
3578 const char *editor;
3580 editor = getenv("GIT_EDITOR");
3581 if (!editor && *opt_editor)
3582 editor = opt_editor;
3583 if (!editor)
3584 editor = getenv("VISUAL");
3585 if (!editor)
3586 editor = getenv("EDITOR");
3587 if (!editor)
3588 editor = "vi";
3590 editor_argv[0] = editor;
3591 open_external_viewer(editor_argv, opt_cdup);
3592 }
3594 static void
3595 open_run_request(enum request request)
3596 {
3597 struct run_request *req = get_run_request(request);
3598 const char *argv[ARRAY_SIZE(req->argv)] = { NULL };
3600 if (!req) {
3601 report("Unknown run request");
3602 return;
3603 }
3605 if (format_argv(argv, req->argv, FORMAT_ALL))
3606 open_external_viewer(argv, NULL);
3607 argv_free(argv);
3608 }
3610 /*
3611 * User request switch noodle
3612 */
3614 static int
3615 view_driver(struct view *view, enum request request)
3616 {
3617 int i;
3619 if (request == REQ_NONE)
3620 return TRUE;
3622 if (request > REQ_NONE) {
3623 open_run_request(request);
3624 view_request(view, REQ_REFRESH);
3625 return TRUE;
3626 }
3628 request = view_request(view, request);
3629 if (request == REQ_NONE)
3630 return TRUE;
3632 switch (request) {
3633 case REQ_MOVE_UP:
3634 case REQ_MOVE_DOWN:
3635 case REQ_MOVE_PAGE_UP:
3636 case REQ_MOVE_PAGE_DOWN:
3637 case REQ_MOVE_FIRST_LINE:
3638 case REQ_MOVE_LAST_LINE:
3639 move_view(view, request);
3640 break;
3642 case REQ_SCROLL_LEFT:
3643 case REQ_SCROLL_RIGHT:
3644 case REQ_SCROLL_LINE_DOWN:
3645 case REQ_SCROLL_LINE_UP:
3646 case REQ_SCROLL_PAGE_DOWN:
3647 case REQ_SCROLL_PAGE_UP:
3648 scroll_view(view, request);
3649 break;
3651 case REQ_VIEW_BLAME:
3652 if (!opt_file[0]) {
3653 report("No file chosen, press %s to open tree view",
3654 get_key(view->keymap, REQ_VIEW_TREE));
3655 break;
3656 }
3657 open_view(view, request, OPEN_DEFAULT);
3658 break;
3660 case REQ_VIEW_BLOB:
3661 if (!ref_blob[0]) {
3662 report("No file chosen, press %s to open tree view",
3663 get_key(view->keymap, REQ_VIEW_TREE));
3664 break;
3665 }
3666 open_view(view, request, OPEN_DEFAULT);
3667 break;
3669 case REQ_VIEW_PAGER:
3670 if (!VIEW(REQ_VIEW_PAGER)->pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
3671 report("No pager content, press %s to run command from prompt",
3672 get_key(view->keymap, REQ_PROMPT));
3673 break;
3674 }
3675 open_view(view, request, OPEN_DEFAULT);
3676 break;
3678 case REQ_VIEW_STAGE:
3679 if (!VIEW(REQ_VIEW_STAGE)->lines) {
3680 report("No stage content, press %s to open the status view and choose file",
3681 get_key(view->keymap, REQ_VIEW_STATUS));
3682 break;
3683 }
3684 open_view(view, request, OPEN_DEFAULT);
3685 break;
3687 case REQ_VIEW_STATUS:
3688 if (opt_is_inside_work_tree == FALSE) {
3689 report("The status view requires a working tree");
3690 break;
3691 }
3692 open_view(view, request, OPEN_DEFAULT);
3693 break;
3695 case REQ_VIEW_MAIN:
3696 case REQ_VIEW_DIFF:
3697 case REQ_VIEW_LOG:
3698 case REQ_VIEW_TREE:
3699 case REQ_VIEW_HELP:
3700 case REQ_VIEW_BRANCH:
3701 open_view(view, request, OPEN_DEFAULT);
3702 break;
3704 case REQ_NEXT:
3705 case REQ_PREVIOUS:
3706 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
3708 if (view->parent) {
3709 int line;
3711 view = view->parent;
3712 line = view->lineno;
3713 move_view(view, request);
3714 if (view_is_displayed(view))
3715 update_view_title(view);
3716 if (line != view->lineno)
3717 view_request(view, REQ_ENTER);
3718 } else {
3719 move_view(view, request);
3720 }
3721 break;
3723 case REQ_VIEW_NEXT:
3724 {
3725 int nviews = displayed_views();
3726 int next_view = (current_view + 1) % nviews;
3728 if (next_view == current_view) {
3729 report("Only one view is displayed");
3730 break;
3731 }
3733 current_view = next_view;
3734 /* Blur out the title of the previous view. */
3735 update_view_title(view);
3736 report("");
3737 break;
3738 }
3739 case REQ_REFRESH:
3740 report("Refreshing is not yet supported for the %s view", view->name);
3741 break;
3743 case REQ_MAXIMIZE:
3744 if (displayed_views() == 2)
3745 maximize_view(view);
3746 break;
3748 case REQ_OPTIONS:
3749 open_option_menu();
3750 break;
3752 case REQ_TOGGLE_LINENO:
3753 toggle_view_option(&opt_line_number, "line numbers");
3754 break;
3756 case REQ_TOGGLE_DATE:
3757 toggle_date();
3758 break;
3760 case REQ_TOGGLE_AUTHOR:
3761 toggle_author();
3762 break;
3764 case REQ_TOGGLE_REV_GRAPH:
3765 toggle_view_option(&opt_rev_graph, "revision graph display");
3766 break;
3768 case REQ_TOGGLE_REFS:
3769 toggle_view_option(&opt_show_refs, "reference display");
3770 break;
3772 case REQ_TOGGLE_SORT_FIELD:
3773 case REQ_TOGGLE_SORT_ORDER:
3774 report("Sorting is not yet supported for the %s view", view->name);
3775 break;
3777 case REQ_SEARCH:
3778 case REQ_SEARCH_BACK:
3779 search_view(view, request);
3780 break;
3782 case REQ_FIND_NEXT:
3783 case REQ_FIND_PREV:
3784 find_next(view, request);
3785 break;
3787 case REQ_STOP_LOADING:
3788 foreach_view(view, i) {
3789 if (view->pipe)
3790 report("Stopped loading the %s view", view->name),
3791 end_update(view, TRUE);
3792 }
3793 break;
3795 case REQ_SHOW_VERSION:
3796 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
3797 return TRUE;
3799 case REQ_SCREEN_REDRAW:
3800 redraw_display(TRUE);
3801 break;
3803 case REQ_EDIT:
3804 report("Nothing to edit");
3805 break;
3807 case REQ_ENTER:
3808 report("Nothing to enter");
3809 break;
3811 case REQ_VIEW_CLOSE:
3812 /* XXX: Mark closed views by letting view->prev point to the
3813 * view itself. Parents to closed view should never be
3814 * followed. */
3815 if (view->prev && view->prev != view) {
3816 maximize_view(view->prev);
3817 view->prev = view;
3818 break;
3819 }
3820 /* Fall-through */
3821 case REQ_QUIT:
3822 return FALSE;
3824 default:
3825 report("Unknown key, press %s for help",
3826 get_key(view->keymap, REQ_VIEW_HELP));
3827 return TRUE;
3828 }
3830 return TRUE;
3831 }
3834 /*
3835 * View backend utilities
3836 */
3838 enum sort_field {
3839 ORDERBY_NAME,
3840 ORDERBY_DATE,
3841 ORDERBY_AUTHOR,
3842 };
3844 struct sort_state {
3845 const enum sort_field *fields;
3846 size_t size, current;
3847 bool reverse;
3848 };
3850 #define SORT_STATE(fields) { fields, ARRAY_SIZE(fields), 0 }
3851 #define get_sort_field(state) ((state).fields[(state).current])
3852 #define sort_order(state, result) ((state).reverse ? -(result) : (result))
3854 static void
3855 sort_view(struct view *view, enum request request, struct sort_state *state,
3856 int (*compare)(const void *, const void *))
3857 {
3858 switch (request) {
3859 case REQ_TOGGLE_SORT_FIELD:
3860 state->current = (state->current + 1) % state->size;
3861 break;
3863 case REQ_TOGGLE_SORT_ORDER:
3864 state->reverse = !state->reverse;
3865 break;
3866 default:
3867 die("Not a sort request");
3868 }
3870 qsort(view->line, view->lines, sizeof(*view->line), compare);
3871 redraw_view(view);
3872 }
3874 DEFINE_ALLOCATOR(realloc_authors, const char *, 256)
3876 /* Small author cache to reduce memory consumption. It uses binary
3877 * search to lookup or find place to position new entries. No entries
3878 * are ever freed. */
3879 static const char *
3880 get_author(const char *name)
3881 {
3882 static const char **authors;
3883 static size_t authors_size;
3884 int from = 0, to = authors_size - 1;
3886 while (from <= to) {
3887 size_t pos = (to + from) / 2;
3888 int cmp = strcmp(name, authors[pos]);
3890 if (!cmp)
3891 return authors[pos];
3893 if (cmp < 0)
3894 to = pos - 1;
3895 else
3896 from = pos + 1;
3897 }
3899 if (!realloc_authors(&authors, authors_size, 1))
3900 return NULL;
3901 name = strdup(name);
3902 if (!name)
3903 return NULL;
3905 memmove(authors + from + 1, authors + from, (authors_size - from) * sizeof(*authors));
3906 authors[from] = name;
3907 authors_size++;
3909 return name;
3910 }
3912 static void
3913 parse_timesec(struct time *time, const char *sec)
3914 {
3915 time->sec = (time_t) atol(sec);
3916 }
3918 static void
3919 parse_timezone(struct time *time, const char *zone)
3920 {
3921 long tz;
3923 tz = ('0' - zone[1]) * 60 * 60 * 10;
3924 tz += ('0' - zone[2]) * 60 * 60;
3925 tz += ('0' - zone[3]) * 60 * 10;
3926 tz += ('0' - zone[4]) * 60;
3928 if (zone[0] == '-')
3929 tz = -tz;
3931 time->tz = tz;
3932 time->sec -= tz;
3933 }
3935 /* Parse author lines where the name may be empty:
3936 * author <email@address.tld> 1138474660 +0100
3937 */
3938 static void
3939 parse_author_line(char *ident, const char **author, struct time *time)
3940 {
3941 char *nameend = strchr(ident, '<');
3942 char *emailend = strchr(ident, '>');
3944 if (nameend && emailend)
3945 *nameend = *emailend = 0;
3946 ident = chomp_string(ident);
3947 if (!*ident) {
3948 if (nameend)
3949 ident = chomp_string(nameend + 1);
3950 if (!*ident)
3951 ident = "Unknown";
3952 }
3954 *author = get_author(ident);
3956 /* Parse epoch and timezone */
3957 if (emailend && emailend[1] == ' ') {
3958 char *secs = emailend + 2;
3959 char *zone = strchr(secs, ' ');
3961 parse_timesec(time, secs);
3963 if (zone && strlen(zone) == STRING_SIZE(" +0700"))
3964 parse_timezone(time, zone + 1);
3965 }
3966 }
3968 static bool
3969 open_commit_parent_menu(char buf[SIZEOF_STR], int *parents)
3970 {
3971 char rev[SIZEOF_REV];
3972 const char *revlist_argv[] = {
3973 "git", "log", "--no-color", "-1", "--pretty=format:%s", rev, NULL
3974 };
3975 struct menu_item *items;
3976 char text[SIZEOF_STR];
3977 bool ok = TRUE;
3978 int i;
3980 items = calloc(*parents + 1, sizeof(*items));
3981 if (!items)
3982 return FALSE;
3984 for (i = 0; i < *parents; i++) {
3985 string_copy_rev(rev, &buf[SIZEOF_REV * i]);
3986 if (!io_run_buf(revlist_argv, text, sizeof(text)) ||
3987 !(items[i].text = strdup(text))) {
3988 ok = FALSE;
3989 break;
3990 }
3991 }
3993 if (ok) {
3994 *parents = 0;
3995 ok = prompt_menu("Select parent", items, parents);
3996 }
3997 for (i = 0; items[i].text; i++)
3998 free((char *) items[i].text);
3999 free(items);
4000 return ok;
4001 }
4003 static bool
4004 select_commit_parent(const char *id, char rev[SIZEOF_REV], const char *path)
4005 {
4006 char buf[SIZEOF_STR * 4];
4007 const char *revlist_argv[] = {
4008 "git", "log", "--no-color", "-1",
4009 "--pretty=format:%P", id, "--", path, NULL
4010 };
4011 int parents;
4013 if (!io_run_buf(revlist_argv, buf, sizeof(buf)) ||
4014 (parents = strlen(buf) / 40) < 0) {
4015 report("Failed to get parent information");
4016 return FALSE;
4018 } else if (parents == 0) {
4019 if (path)
4020 report("Path '%s' does not exist in the parent", path);
4021 else
4022 report("The selected commit has no parents");
4023 return FALSE;
4024 }
4026 if (parents == 1)
4027 parents = 0;
4028 else if (!open_commit_parent_menu(buf, &parents))
4029 return FALSE;
4031 string_copy_rev(rev, &buf[41 * parents]);
4032 return TRUE;
4033 }
4035 /*
4036 * Pager backend
4037 */
4039 static bool
4040 pager_draw(struct view *view, struct line *line, unsigned int lineno)
4041 {
4042 char text[SIZEOF_STR];
4044 if (opt_line_number && draw_lineno(view, lineno))
4045 return TRUE;
4047 string_expand(text, sizeof(text), line->data, opt_tab_size);
4048 draw_text(view, line->type, text, TRUE);
4049 return TRUE;
4050 }
4052 static bool
4053 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
4054 {
4055 const char *describe_argv[] = { "git", "describe", commit_id, NULL };
4056 char ref[SIZEOF_STR];
4058 if (!io_run_buf(describe_argv, ref, sizeof(ref)) || !*ref)
4059 return TRUE;
4061 /* This is the only fatal call, since it can "corrupt" the buffer. */
4062 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
4063 return FALSE;
4065 return TRUE;
4066 }
4068 static void
4069 add_pager_refs(struct view *view, struct line *line)
4070 {
4071 char buf[SIZEOF_STR];
4072 char *commit_id = (char *)line->data + STRING_SIZE("commit ");
4073 struct ref_list *list;
4074 size_t bufpos = 0, i;
4075 const char *sep = "Refs: ";
4076 bool is_tag = FALSE;
4078 assert(line->type == LINE_COMMIT);
4080 list = get_ref_list(commit_id);
4081 if (!list) {
4082 if (view->type == VIEW_DIFF)
4083 goto try_add_describe_ref;
4084 return;
4085 }
4087 for (i = 0; i < list->size; i++) {
4088 struct ref *ref = list->refs[i];
4089 const char *fmt = ref->tag ? "%s[%s]" :
4090 ref->remote ? "%s<%s>" : "%s%s";
4092 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
4093 return;
4094 sep = ", ";
4095 if (ref->tag)
4096 is_tag = TRUE;
4097 }
4099 if (!is_tag && view->type == VIEW_DIFF) {
4100 try_add_describe_ref:
4101 /* Add <tag>-g<commit_id> "fake" reference. */
4102 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
4103 return;
4104 }
4106 if (bufpos == 0)
4107 return;
4109 add_line_text(view, buf, LINE_PP_REFS);
4110 }
4112 static bool
4113 pager_read(struct view *view, char *data)
4114 {
4115 struct line *line;
4117 if (!data)
4118 return TRUE;
4120 line = add_line_text(view, data, get_line_type(data));
4121 if (!line)
4122 return FALSE;
4124 if (line->type == LINE_COMMIT &&
4125 (view->type == VIEW_DIFF ||
4126 view->type == VIEW_LOG))
4127 add_pager_refs(view, line);
4129 return TRUE;
4130 }
4132 static enum request
4133 pager_request(struct view *view, enum request request, struct line *line)
4134 {
4135 int split = 0;
4137 if (request != REQ_ENTER)
4138 return request;
4140 if (line->type == LINE_COMMIT &&
4141 (view->type == VIEW_LOG ||
4142 view->type == VIEW_PAGER)) {
4143 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
4144 split = 1;
4145 }
4147 /* Always scroll the view even if it was split. That way
4148 * you can use Enter to scroll through the log view and
4149 * split open each commit diff. */
4150 scroll_view(view, REQ_SCROLL_LINE_DOWN);
4152 /* FIXME: A minor workaround. Scrolling the view will call report("")
4153 * but if we are scrolling a non-current view this won't properly
4154 * update the view title. */
4155 if (split)
4156 update_view_title(view);
4158 return REQ_NONE;
4159 }
4161 static bool
4162 pager_grep(struct view *view, struct line *line)
4163 {
4164 const char *text[] = { line->data, NULL };
4166 return grep_text(view, text);
4167 }
4169 static void
4170 pager_select(struct view *view, struct line *line)
4171 {
4172 if (line->type == LINE_COMMIT) {
4173 char *text = (char *)line->data + STRING_SIZE("commit ");
4175 if (view->type != VIEW_PAGER)
4176 string_copy_rev(view->ref, text);
4177 string_copy_rev(ref_commit, text);
4178 }
4179 }
4181 static struct view_ops pager_ops = {
4182 "line",
4183 NULL,
4184 NULL,
4185 pager_read,
4186 pager_draw,
4187 pager_request,
4188 pager_grep,
4189 pager_select,
4190 };
4192 static const char *log_argv[SIZEOF_ARG] = {
4193 "git", "log", "--no-color", "--cc", "--stat", "-n100", "%(head)", NULL
4194 };
4196 static enum request
4197 log_request(struct view *view, enum request request, struct line *line)
4198 {
4199 switch (request) {
4200 case REQ_REFRESH:
4201 load_refs();
4202 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
4203 return REQ_NONE;
4204 default:
4205 return pager_request(view, request, line);
4206 }
4207 }
4209 static struct view_ops log_ops = {
4210 "line",
4211 log_argv,
4212 NULL,
4213 pager_read,
4214 pager_draw,
4215 log_request,
4216 pager_grep,
4217 pager_select,
4218 };
4220 static const char *diff_argv[SIZEOF_ARG] = {
4221 "git", "show", "--pretty=fuller", "--no-color", "--root",
4222 "--patch-with-stat", "--find-copies-harder", "-C", "%(commit)", NULL
4223 };
4225 static struct view_ops diff_ops = {
4226 "line",
4227 diff_argv,
4228 NULL,
4229 pager_read,
4230 pager_draw,
4231 pager_request,
4232 pager_grep,
4233 pager_select,
4234 };
4236 /*
4237 * Help backend
4238 */
4240 static bool help_keymap_hidden[ARRAY_SIZE(keymap_table)];
4242 static bool
4243 help_open_keymap_title(struct view *view, enum keymap keymap)
4244 {
4245 struct line *line;
4247 line = add_line_format(view, LINE_HELP_KEYMAP, "[%c] %s bindings",
4248 help_keymap_hidden[keymap] ? '+' : '-',
4249 enum_name(keymap_table[keymap]));
4250 if (line)
4251 line->other = keymap;
4253 return help_keymap_hidden[keymap];
4254 }
4256 static void
4257 help_open_keymap(struct view *view, enum keymap keymap)
4258 {
4259 const char *group = NULL;
4260 char buf[SIZEOF_STR];
4261 size_t bufpos;
4262 bool add_title = TRUE;
4263 int i;
4265 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
4266 const char *key = NULL;
4268 if (req_info[i].request == REQ_NONE)
4269 continue;
4271 if (!req_info[i].request) {
4272 group = req_info[i].help;
4273 continue;
4274 }
4276 key = get_keys(keymap, req_info[i].request, TRUE);
4277 if (!key || !*key)
4278 continue;
4280 if (add_title && help_open_keymap_title(view, keymap))
4281 return;
4282 add_title = FALSE;
4284 if (group) {
4285 add_line_text(view, group, LINE_HELP_GROUP);
4286 group = NULL;
4287 }
4289 add_line_format(view, LINE_DEFAULT, " %-25s %-20s %s", key,
4290 enum_name(req_info[i]), req_info[i].help);
4291 }
4293 group = "External commands:";
4295 for (i = 0; i < run_requests; i++) {
4296 struct run_request *req = get_run_request(REQ_NONE + i + 1);
4297 const char *key;
4298 int argc;
4300 if (!req || req->keymap != keymap)
4301 continue;
4303 key = get_key_name(req->key);
4304 if (!*key)
4305 key = "(no key defined)";
4307 if (add_title && help_open_keymap_title(view, keymap))
4308 return;
4309 if (group) {
4310 add_line_text(view, group, LINE_HELP_GROUP);
4311 group = NULL;
4312 }
4314 for (bufpos = 0, argc = 0; req->argv[argc]; argc++)
4315 if (!string_format_from(buf, &bufpos, "%s%s",
4316 argc ? " " : "", req->argv[argc]))
4317 return;
4319 add_line_format(view, LINE_DEFAULT, " %-25s `%s`", key, buf);
4320 }
4321 }
4323 static bool
4324 help_open(struct view *view)
4325 {
4326 enum keymap keymap;
4328 reset_view(view);
4329 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
4330 add_line_text(view, "", LINE_DEFAULT);
4332 for (keymap = 0; keymap < ARRAY_SIZE(keymap_table); keymap++)
4333 help_open_keymap(view, keymap);
4335 return TRUE;
4336 }
4338 static enum request
4339 help_request(struct view *view, enum request request, struct line *line)
4340 {
4341 switch (request) {
4342 case REQ_ENTER:
4343 if (line->type == LINE_HELP_KEYMAP) {
4344 help_keymap_hidden[line->other] =
4345 !help_keymap_hidden[line->other];
4346 view->p_restore = TRUE;
4347 open_view(view, REQ_VIEW_HELP, OPEN_REFRESH);
4348 }
4350 return REQ_NONE;
4351 default:
4352 return pager_request(view, request, line);
4353 }
4354 }
4356 static struct view_ops help_ops = {
4357 "line",
4358 NULL,
4359 help_open,
4360 NULL,
4361 pager_draw,
4362 help_request,
4363 pager_grep,
4364 pager_select,
4365 };
4368 /*
4369 * Tree backend
4370 */
4372 struct tree_stack_entry {
4373 struct tree_stack_entry *prev; /* Entry below this in the stack */
4374 unsigned long lineno; /* Line number to restore */
4375 char *name; /* Position of name in opt_path */
4376 };
4378 /* The top of the path stack. */
4379 static struct tree_stack_entry *tree_stack = NULL;
4380 unsigned long tree_lineno = 0;
4382 static void
4383 pop_tree_stack_entry(void)
4384 {
4385 struct tree_stack_entry *entry = tree_stack;
4387 tree_lineno = entry->lineno;
4388 entry->name[0] = 0;
4389 tree_stack = entry->prev;
4390 free(entry);
4391 }
4393 static void
4394 push_tree_stack_entry(const char *name, unsigned long lineno)
4395 {
4396 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
4397 size_t pathlen = strlen(opt_path);
4399 if (!entry)
4400 return;
4402 entry->prev = tree_stack;
4403 entry->name = opt_path + pathlen;
4404 tree_stack = entry;
4406 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
4407 pop_tree_stack_entry();
4408 return;
4409 }
4411 /* Move the current line to the first tree entry. */
4412 tree_lineno = 1;
4413 entry->lineno = lineno;
4414 }
4416 /* Parse output from git-ls-tree(1):
4417 *
4418 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
4419 */
4421 #define SIZEOF_TREE_ATTR \
4422 STRING_SIZE("100644 blob f931e1d229c3e185caad4449bf5b66ed72462657\t")
4424 #define SIZEOF_TREE_MODE \
4425 STRING_SIZE("100644 ")
4427 #define TREE_ID_OFFSET \
4428 STRING_SIZE("100644 blob ")
4430 struct tree_entry {
4431 char id[SIZEOF_REV];
4432 mode_t mode;
4433 struct time time; /* Date from the author ident. */
4434 const char *author; /* Author of the commit. */
4435 char name[1];
4436 };
4438 static const char *
4439 tree_path(const struct line *line)
4440 {
4441 return ((struct tree_entry *) line->data)->name;
4442 }
4444 static int
4445 tree_compare_entry(const struct line *line1, const struct line *line2)
4446 {
4447 if (line1->type != line2->type)
4448 return line1->type == LINE_TREE_DIR ? -1 : 1;
4449 return strcmp(tree_path(line1), tree_path(line2));
4450 }
4452 static const enum sort_field tree_sort_fields[] = {
4453 ORDERBY_NAME, ORDERBY_DATE, ORDERBY_AUTHOR
4454 };
4455 static struct sort_state tree_sort_state = SORT_STATE(tree_sort_fields);
4457 static int
4458 tree_compare(const void *l1, const void *l2)
4459 {
4460 const struct line *line1 = (const struct line *) l1;
4461 const struct line *line2 = (const struct line *) l2;
4462 const struct tree_entry *entry1 = ((const struct line *) l1)->data;
4463 const struct tree_entry *entry2 = ((const struct line *) l2)->data;
4465 if (line1->type == LINE_TREE_HEAD)
4466 return -1;
4467 if (line2->type == LINE_TREE_HEAD)
4468 return 1;
4470 switch (get_sort_field(tree_sort_state)) {
4471 case ORDERBY_DATE:
4472 return sort_order(tree_sort_state, timecmp(&entry1->time, &entry2->time));
4474 case ORDERBY_AUTHOR:
4475 return sort_order(tree_sort_state, strcmp(entry1->author, entry2->author));
4477 case ORDERBY_NAME:
4478 default:
4479 return sort_order(tree_sort_state, tree_compare_entry(line1, line2));
4480 }
4481 }
4484 static struct line *
4485 tree_entry(struct view *view, enum line_type type, const char *path,
4486 const char *mode, const char *id)
4487 {
4488 struct tree_entry *entry = calloc(1, sizeof(*entry) + strlen(path));
4489 struct line *line = entry ? add_line_data(view, entry, type) : NULL;
4491 if (!entry || !line) {
4492 free(entry);
4493 return NULL;
4494 }
4496 strncpy(entry->name, path, strlen(path));
4497 if (mode)
4498 entry->mode = strtoul(mode, NULL, 8);
4499 if (id)
4500 string_copy_rev(entry->id, id);
4502 return line;
4503 }
4505 static bool
4506 tree_read_date(struct view *view, char *text, bool *read_date)
4507 {
4508 static const char *author_name;
4509 static struct time author_time;
4511 if (!text && *read_date) {
4512 *read_date = FALSE;
4513 return TRUE;
4515 } else if (!text) {
4516 char *path = *opt_path ? opt_path : ".";
4517 /* Find next entry to process */
4518 const char *log_file[] = {
4519 "git", "log", "--no-color", "--pretty=raw",
4520 "--cc", "--raw", view->id, "--", path, NULL
4521 };
4522 struct io io = {};
4524 if (!view->lines) {
4525 tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL);
4526 report("Tree is empty");
4527 return TRUE;
4528 }
4530 if (!io_run_rd(&io, log_file, opt_cdup, FORMAT_NONE)) {
4531 report("Failed to load tree data");
4532 return TRUE;
4533 }
4535 io_done(view->pipe);
4536 view->io = io;
4537 *read_date = TRUE;
4538 return FALSE;
4540 } else if (*text == 'a' && get_line_type(text) == LINE_AUTHOR) {
4541 parse_author_line(text + STRING_SIZE("author "),
4542 &author_name, &author_time);
4544 } else if (*text == ':') {
4545 char *pos;
4546 size_t annotated = 1;
4547 size_t i;
4549 pos = strchr(text, '\t');
4550 if (!pos)
4551 return TRUE;
4552 text = pos + 1;
4553 if (*opt_path && !strncmp(text, opt_path, strlen(opt_path)))
4554 text += strlen(opt_path);
4555 pos = strchr(text, '/');
4556 if (pos)
4557 *pos = 0;
4559 for (i = 1; i < view->lines; i++) {
4560 struct line *line = &view->line[i];
4561 struct tree_entry *entry = line->data;
4563 annotated += !!entry->author;
4564 if (entry->author || strcmp(entry->name, text))
4565 continue;
4567 entry->author = author_name;
4568 entry->time = author_time;
4569 line->dirty = 1;
4570 break;
4571 }
4573 if (annotated == view->lines)
4574 io_kill(view->pipe);
4575 }
4576 return TRUE;
4577 }
4579 static bool
4580 tree_read(struct view *view, char *text)
4581 {
4582 static bool read_date = FALSE;
4583 struct tree_entry *data;
4584 struct line *entry, *line;
4585 enum line_type type;
4586 size_t textlen = text ? strlen(text) : 0;
4587 char *path = text + SIZEOF_TREE_ATTR;
4589 if (read_date || !text)
4590 return tree_read_date(view, text, &read_date);
4592 if (textlen <= SIZEOF_TREE_ATTR)
4593 return FALSE;
4594 if (view->lines == 0 &&
4595 !tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL))
4596 return FALSE;
4598 /* Strip the path part ... */
4599 if (*opt_path) {
4600 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
4601 size_t striplen = strlen(opt_path);
4603 if (pathlen > striplen)
4604 memmove(path, path + striplen,
4605 pathlen - striplen + 1);
4607 /* Insert "link" to parent directory. */
4608 if (view->lines == 1 &&
4609 !tree_entry(view, LINE_TREE_DIR, "..", "040000", view->ref))
4610 return FALSE;
4611 }
4613 type = text[SIZEOF_TREE_MODE] == 't' ? LINE_TREE_DIR : LINE_TREE_FILE;
4614 entry = tree_entry(view, type, path, text, text + TREE_ID_OFFSET);
4615 if (!entry)
4616 return FALSE;
4617 data = entry->data;
4619 /* Skip "Directory ..." and ".." line. */
4620 for (line = &view->line[1 + !!*opt_path]; line < entry; line++) {
4621 if (tree_compare_entry(line, entry) <= 0)
4622 continue;
4624 memmove(line + 1, line, (entry - line) * sizeof(*entry));
4626 line->data = data;
4627 line->type = type;
4628 for (; line <= entry; line++)
4629 line->dirty = line->cleareol = 1;
4630 return TRUE;
4631 }
4633 if (tree_lineno > view->lineno) {
4634 view->lineno = tree_lineno;
4635 tree_lineno = 0;
4636 }
4638 return TRUE;
4639 }
4641 static bool
4642 tree_draw(struct view *view, struct line *line, unsigned int lineno)
4643 {
4644 struct tree_entry *entry = line->data;
4646 if (line->type == LINE_TREE_HEAD) {
4647 if (draw_text(view, line->type, "Directory path /", TRUE))
4648 return TRUE;
4649 } else {
4650 if (draw_mode(view, entry->mode))
4651 return TRUE;
4653 if (opt_author && draw_author(view, entry->author))
4654 return TRUE;
4656 if (opt_date && draw_date(view, &entry->time))
4657 return TRUE;
4658 }
4659 if (draw_text(view, line->type, entry->name, TRUE))
4660 return TRUE;
4661 return TRUE;
4662 }
4664 static void
4665 open_blob_editor(const char *id)
4666 {
4667 const char *blob_argv[] = { "git", "cat-file", "blob", id, NULL };
4668 char file[SIZEOF_STR] = "/tmp/tigblob.XXXXXX";
4669 int fd = mkstemp(file);
4671 if (fd == -1)
4672 report("Failed to create temporary file");
4673 else if (!io_run_append(blob_argv, FORMAT_NONE, fd))
4674 report("Failed to save blob data to file");
4675 else
4676 open_editor(file);
4677 if (fd != -1)
4678 unlink(file);
4679 }
4681 static enum request
4682 tree_request(struct view *view, enum request request, struct line *line)
4683 {
4684 enum open_flags flags;
4685 struct tree_entry *entry = line->data;
4687 switch (request) {
4688 case REQ_VIEW_BLAME:
4689 if (line->type != LINE_TREE_FILE) {
4690 report("Blame only supported for files");
4691 return REQ_NONE;
4692 }
4694 string_copy(opt_ref, view->vid);
4695 return request;
4697 case REQ_EDIT:
4698 if (line->type != LINE_TREE_FILE) {
4699 report("Edit only supported for files");
4700 } else if (!is_head_commit(view->vid)) {
4701 open_blob_editor(entry->id);
4702 } else {
4703 open_editor(opt_file);
4704 }
4705 return REQ_NONE;
4707 case REQ_TOGGLE_SORT_FIELD:
4708 case REQ_TOGGLE_SORT_ORDER:
4709 sort_view(view, request, &tree_sort_state, tree_compare);
4710 return REQ_NONE;
4712 case REQ_PARENT:
4713 if (!*opt_path) {
4714 /* quit view if at top of tree */
4715 return REQ_VIEW_CLOSE;
4716 }
4717 /* fake 'cd ..' */
4718 line = &view->line[1];
4719 break;
4721 case REQ_ENTER:
4722 break;
4724 default:
4725 return request;
4726 }
4728 /* Cleanup the stack if the tree view is at a different tree. */
4729 while (!*opt_path && tree_stack)
4730 pop_tree_stack_entry();
4732 switch (line->type) {
4733 case LINE_TREE_DIR:
4734 /* Depending on whether it is a subdirectory or parent link
4735 * mangle the path buffer. */
4736 if (line == &view->line[1] && *opt_path) {
4737 pop_tree_stack_entry();
4739 } else {
4740 const char *basename = tree_path(line);
4742 push_tree_stack_entry(basename, view->lineno);
4743 }
4745 /* Trees and subtrees share the same ID, so they are not not
4746 * unique like blobs. */
4747 flags = OPEN_RELOAD;
4748 request = REQ_VIEW_TREE;
4749 break;
4751 case LINE_TREE_FILE:
4752 flags = view_is_displayed(view) ? OPEN_SPLIT : OPEN_DEFAULT;
4753 request = REQ_VIEW_BLOB;
4754 break;
4756 default:
4757 return REQ_NONE;
4758 }
4760 open_view(view, request, flags);
4761 if (request == REQ_VIEW_TREE)
4762 view->lineno = tree_lineno;
4764 return REQ_NONE;
4765 }
4767 static bool
4768 tree_grep(struct view *view, struct line *line)
4769 {
4770 struct tree_entry *entry = line->data;
4771 const char *text[] = {
4772 entry->name,
4773 opt_author ? entry->author : "",
4774 mkdate(&entry->time, opt_date),
4775 NULL
4776 };
4778 return grep_text(view, text);
4779 }
4781 static void
4782 tree_select(struct view *view, struct line *line)
4783 {
4784 struct tree_entry *entry = line->data;
4786 if (line->type == LINE_TREE_FILE) {
4787 string_copy_rev(ref_blob, entry->id);
4788 string_format(opt_file, "%s%s", opt_path, tree_path(line));
4790 } else if (line->type != LINE_TREE_DIR) {
4791 return;
4792 }
4794 string_copy_rev(view->ref, entry->id);
4795 }
4797 static bool
4798 tree_prepare(struct view *view)
4799 {
4800 if (view->lines == 0 && opt_prefix[0]) {
4801 char *pos = opt_prefix;
4803 while (pos && *pos) {
4804 char *end = strchr(pos, '/');
4806 if (end)
4807 *end = 0;
4808 push_tree_stack_entry(pos, 0);
4809 pos = end;
4810 if (end) {
4811 *end = '/';
4812 pos++;
4813 }
4814 }
4816 } else if (strcmp(view->vid, view->id)) {
4817 opt_path[0] = 0;
4818 }
4820 return io_format(&view->io, opt_cdup, IO_RD, view->ops->argv, FORMAT_ALL);
4821 }
4823 static const char *tree_argv[SIZEOF_ARG] = {
4824 "git", "ls-tree", "%(commit)", "%(directory)", NULL
4825 };
4827 static struct view_ops tree_ops = {
4828 "file",
4829 tree_argv,
4830 NULL,
4831 tree_read,
4832 tree_draw,
4833 tree_request,
4834 tree_grep,
4835 tree_select,
4836 tree_prepare,
4837 };
4839 static bool
4840 blob_read(struct view *view, char *line)
4841 {
4842 if (!line)
4843 return TRUE;
4844 return add_line_text(view, line, LINE_DEFAULT) != NULL;
4845 }
4847 static enum request
4848 blob_request(struct view *view, enum request request, struct line *line)
4849 {
4850 switch (request) {
4851 case REQ_EDIT:
4852 open_blob_editor(view->vid);
4853 return REQ_NONE;
4854 default:
4855 return pager_request(view, request, line);
4856 }
4857 }
4859 static const char *blob_argv[SIZEOF_ARG] = {
4860 "git", "cat-file", "blob", "%(blob)", NULL
4861 };
4863 static struct view_ops blob_ops = {
4864 "line",
4865 blob_argv,
4866 NULL,
4867 blob_read,
4868 pager_draw,
4869 blob_request,
4870 pager_grep,
4871 pager_select,
4872 };
4874 /*
4875 * Blame backend
4876 *
4877 * Loading the blame view is a two phase job:
4878 *
4879 * 1. File content is read either using opt_file from the
4880 * filesystem or using git-cat-file.
4881 * 2. Then blame information is incrementally added by
4882 * reading output from git-blame.
4883 */
4885 static const char *blame_head_argv[] = {
4886 "git", "blame", "--incremental", "--", "%(file)", NULL
4887 };
4889 static const char *blame_ref_argv[] = {
4890 "git", "blame", "--incremental", "%(ref)", "--", "%(file)", NULL
4891 };
4893 static const char *blame_cat_file_argv[] = {
4894 "git", "cat-file", "blob", "%(ref):%(file)", NULL
4895 };
4897 struct blame_commit {
4898 char id[SIZEOF_REV]; /* SHA1 ID. */
4899 char title[128]; /* First line of the commit message. */
4900 const char *author; /* Author of the commit. */
4901 struct time time; /* Date from the author ident. */
4902 char filename[128]; /* Name of file. */
4903 bool has_previous; /* Was a "previous" line detected. */
4904 };
4906 struct blame {
4907 struct blame_commit *commit;
4908 unsigned long lineno;
4909 char text[1];
4910 };
4912 static bool
4913 blame_open(struct view *view)
4914 {
4915 char path[SIZEOF_STR];
4917 if (!view->prev && *opt_prefix) {
4918 string_copy(path, opt_file);
4919 if (!string_format(opt_file, "%s%s", opt_prefix, path))
4920 return FALSE;
4921 }
4923 if (*opt_ref || !io_open(&view->io, "%s%s", opt_cdup, opt_file)) {
4924 if (!io_run_rd(&view->io, blame_cat_file_argv, opt_cdup, FORMAT_ALL))
4925 return FALSE;
4926 }
4928 setup_update(view, opt_file);
4929 string_format(view->ref, "%s ...", opt_file);
4931 return TRUE;
4932 }
4934 static struct blame_commit *
4935 get_blame_commit(struct view *view, const char *id)
4936 {
4937 size_t i;
4939 for (i = 0; i < view->lines; i++) {
4940 struct blame *blame = view->line[i].data;
4942 if (!blame->commit)
4943 continue;
4945 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
4946 return blame->commit;
4947 }
4949 {
4950 struct blame_commit *commit = calloc(1, sizeof(*commit));
4952 if (commit)
4953 string_ncopy(commit->id, id, SIZEOF_REV);
4954 return commit;
4955 }
4956 }
4958 static bool
4959 parse_number(const char **posref, size_t *number, size_t min, size_t max)
4960 {
4961 const char *pos = *posref;
4963 *posref = NULL;
4964 pos = strchr(pos + 1, ' ');
4965 if (!pos || !isdigit(pos[1]))
4966 return FALSE;
4967 *number = atoi(pos + 1);
4968 if (*number < min || *number > max)
4969 return FALSE;
4971 *posref = pos;
4972 return TRUE;
4973 }
4975 static struct blame_commit *
4976 parse_blame_commit(struct view *view, const char *text, int *blamed)
4977 {
4978 struct blame_commit *commit;
4979 struct blame *blame;
4980 const char *pos = text + SIZEOF_REV - 2;
4981 size_t orig_lineno = 0;
4982 size_t lineno;
4983 size_t group;
4985 if (strlen(text) <= SIZEOF_REV || pos[1] != ' ')
4986 return NULL;
4988 if (!parse_number(&pos, &orig_lineno, 1, 9999999) ||
4989 !parse_number(&pos, &lineno, 1, view->lines) ||
4990 !parse_number(&pos, &group, 1, view->lines - lineno + 1))
4991 return NULL;
4993 commit = get_blame_commit(view, text);
4994 if (!commit)
4995 return NULL;
4997 *blamed += group;
4998 while (group--) {
4999 struct line *line = &view->line[lineno + group - 1];
5001 blame = line->data;
5002 blame->commit = commit;
5003 blame->lineno = orig_lineno + group - 1;
5004 line->dirty = 1;
5005 }
5007 return commit;
5008 }
5010 static bool
5011 blame_read_file(struct view *view, const char *line, bool *read_file)
5012 {
5013 if (!line) {
5014 const char **argv = *opt_ref ? blame_ref_argv : blame_head_argv;
5015 struct io io = {};
5017 if (view->lines == 0 && !view->prev)
5018 die("No blame exist for %s", view->vid);
5020 if (view->lines == 0 || !io_run_rd(&io, argv, opt_cdup, FORMAT_ALL)) {
5021 report("Failed to load blame data");
5022 return TRUE;
5023 }
5025 io_done(view->pipe);
5026 view->io = io;
5027 *read_file = FALSE;
5028 return FALSE;
5030 } else {
5031 size_t linelen = strlen(line);
5032 struct blame *blame = malloc(sizeof(*blame) + linelen);
5034 if (!blame)
5035 return FALSE;
5037 blame->commit = NULL;
5038 strncpy(blame->text, line, linelen);
5039 blame->text[linelen] = 0;
5040 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
5041 }
5042 }
5044 static bool
5045 match_blame_header(const char *name, char **line)
5046 {
5047 size_t namelen = strlen(name);
5048 bool matched = !strncmp(name, *line, namelen);
5050 if (matched)
5051 *line += namelen;
5053 return matched;
5054 }
5056 static bool
5057 blame_read(struct view *view, char *line)
5058 {
5059 static struct blame_commit *commit = NULL;
5060 static int blamed = 0;
5061 static bool read_file = TRUE;
5063 if (read_file)
5064 return blame_read_file(view, line, &read_file);
5066 if (!line) {
5067 /* Reset all! */
5068 commit = NULL;
5069 blamed = 0;
5070 read_file = TRUE;
5071 string_format(view->ref, "%s", view->vid);
5072 if (view_is_displayed(view)) {
5073 update_view_title(view);
5074 redraw_view_from(view, 0);
5075 }
5076 return TRUE;
5077 }
5079 if (!commit) {
5080 commit = parse_blame_commit(view, line, &blamed);
5081 string_format(view->ref, "%s %2d%%", view->vid,
5082 view->lines ? blamed * 100 / view->lines : 0);
5084 } else if (match_blame_header("author ", &line)) {
5085 commit->author = get_author(line);
5087 } else if (match_blame_header("author-time ", &line)) {
5088 parse_timesec(&commit->time, line);
5090 } else if (match_blame_header("author-tz ", &line)) {
5091 parse_timezone(&commit->time, line);
5093 } else if (match_blame_header("summary ", &line)) {
5094 string_ncopy(commit->title, line, strlen(line));
5096 } else if (match_blame_header("previous ", &line)) {
5097 commit->has_previous = TRUE;
5099 } else if (match_blame_header("filename ", &line)) {
5100 string_ncopy(commit->filename, line, strlen(line));
5101 commit = NULL;
5102 }
5104 return TRUE;
5105 }
5107 static bool
5108 blame_draw(struct view *view, struct line *line, unsigned int lineno)
5109 {
5110 struct blame *blame = line->data;
5111 struct time *time = NULL;
5112 const char *id = NULL, *author = NULL;
5113 char text[SIZEOF_STR];
5115 if (blame->commit && *blame->commit->filename) {
5116 id = blame->commit->id;
5117 author = blame->commit->author;
5118 time = &blame->commit->time;
5119 }
5121 if (opt_date && draw_date(view, time))
5122 return TRUE;
5124 if (opt_author && draw_author(view, author))
5125 return TRUE;
5127 if (draw_field(view, LINE_BLAME_ID, id, ID_COLS, FALSE))
5128 return TRUE;
5130 if (draw_lineno(view, lineno))
5131 return TRUE;
5133 string_expand(text, sizeof(text), blame->text, opt_tab_size);
5134 draw_text(view, LINE_DEFAULT, text, TRUE);
5135 return TRUE;
5136 }
5138 static bool
5139 check_blame_commit(struct blame *blame, bool check_null_id)
5140 {
5141 if (!blame->commit)
5142 report("Commit data not loaded yet");
5143 else if (check_null_id && !strcmp(blame->commit->id, NULL_ID))
5144 report("No commit exist for the selected line");
5145 else
5146 return TRUE;
5147 return FALSE;
5148 }
5150 static void
5151 setup_blame_parent_line(struct view *view, struct blame *blame)
5152 {
5153 const char *diff_tree_argv[] = {
5154 "git", "diff-tree", "-U0", blame->commit->id,
5155 "--", blame->commit->filename, NULL
5156 };
5157 struct io io = {};
5158 int parent_lineno = -1;
5159 int blamed_lineno = -1;
5160 char *line;
5162 if (!io_run(&io, diff_tree_argv, NULL, IO_RD))
5163 return;
5165 while ((line = io_get(&io, '\n', TRUE))) {
5166 if (*line == '@') {
5167 char *pos = strchr(line, '+');
5169 parent_lineno = atoi(line + 4);
5170 if (pos)
5171 blamed_lineno = atoi(pos + 1);
5173 } else if (*line == '+' && parent_lineno != -1) {
5174 if (blame->lineno == blamed_lineno - 1 &&
5175 !strcmp(blame->text, line + 1)) {
5176 view->lineno = parent_lineno ? parent_lineno - 1 : 0;
5177 break;
5178 }
5179 blamed_lineno++;
5180 }
5181 }
5183 io_done(&io);
5184 }
5186 static enum request
5187 blame_request(struct view *view, enum request request, struct line *line)
5188 {
5189 enum open_flags flags = view_is_displayed(view) ? OPEN_SPLIT : OPEN_DEFAULT;
5190 struct blame *blame = line->data;
5192 switch (request) {
5193 case REQ_VIEW_BLAME:
5194 if (check_blame_commit(blame, TRUE)) {
5195 string_copy(opt_ref, blame->commit->id);
5196 string_copy(opt_file, blame->commit->filename);
5197 if (blame->lineno)
5198 view->lineno = blame->lineno;
5199 open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
5200 }
5201 break;
5203 case REQ_PARENT:
5204 if (check_blame_commit(blame, TRUE) &&
5205 select_commit_parent(blame->commit->id, opt_ref,
5206 blame->commit->filename)) {
5207 string_copy(opt_file, blame->commit->filename);
5208 setup_blame_parent_line(view, blame);
5209 open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
5210 }
5211 break;
5213 case REQ_ENTER:
5214 if (!check_blame_commit(blame, FALSE))
5215 break;
5217 if (view_is_displayed(VIEW(REQ_VIEW_DIFF)) &&
5218 !strcmp(blame->commit->id, VIEW(REQ_VIEW_DIFF)->ref))
5219 break;
5221 if (!strcmp(blame->commit->id, NULL_ID)) {
5222 struct view *diff = VIEW(REQ_VIEW_DIFF);
5223 const char *diff_index_argv[] = {
5224 "git", "diff-index", "--root", "--patch-with-stat",
5225 "-C", "-M", "HEAD", "--", view->vid, NULL
5226 };
5228 if (!blame->commit->has_previous) {
5229 diff_index_argv[1] = "diff";
5230 diff_index_argv[2] = "--no-color";
5231 diff_index_argv[6] = "--";
5232 diff_index_argv[7] = "/dev/null";
5233 }
5235 if (!prepare_update(diff, diff_index_argv, NULL)) {
5236 report("Failed to allocate diff command");
5237 break;
5238 }
5239 flags |= OPEN_PREPARED;
5240 }
5242 open_view(view, REQ_VIEW_DIFF, flags);
5243 if (VIEW(REQ_VIEW_DIFF)->pipe && !strcmp(blame->commit->id, NULL_ID))
5244 string_copy_rev(VIEW(REQ_VIEW_DIFF)->ref, NULL_ID);
5245 break;
5247 default:
5248 return request;
5249 }
5251 return REQ_NONE;
5252 }
5254 static bool
5255 blame_grep(struct view *view, struct line *line)
5256 {
5257 struct blame *blame = line->data;
5258 struct blame_commit *commit = blame->commit;
5259 const char *text[] = {
5260 blame->text,
5261 commit ? commit->title : "",
5262 commit ? commit->id : "",
5263 commit && opt_author ? commit->author : "",
5264 commit ? mkdate(&commit->time, opt_date) : "",
5265 NULL
5266 };
5268 return grep_text(view, text);
5269 }
5271 static void
5272 blame_select(struct view *view, struct line *line)
5273 {
5274 struct blame *blame = line->data;
5275 struct blame_commit *commit = blame->commit;
5277 if (!commit)
5278 return;
5280 if (!strcmp(commit->id, NULL_ID))
5281 string_ncopy(ref_commit, "HEAD", 4);
5282 else
5283 string_copy_rev(ref_commit, commit->id);
5284 }
5286 static struct view_ops blame_ops = {
5287 "line",
5288 NULL,
5289 blame_open,
5290 blame_read,
5291 blame_draw,
5292 blame_request,
5293 blame_grep,
5294 blame_select,
5295 };
5297 /*
5298 * Branch backend
5299 */
5301 struct branch {
5302 const char *author; /* Author of the last commit. */
5303 struct time time; /* Date of the last activity. */
5304 const struct ref *ref; /* Name and commit ID information. */
5305 };
5307 static const struct ref branch_all;
5309 static const enum sort_field branch_sort_fields[] = {
5310 ORDERBY_NAME, ORDERBY_DATE, ORDERBY_AUTHOR
5311 };
5312 static struct sort_state branch_sort_state = SORT_STATE(branch_sort_fields);
5314 static int
5315 branch_compare(const void *l1, const void *l2)
5316 {
5317 const struct branch *branch1 = ((const struct line *) l1)->data;
5318 const struct branch *branch2 = ((const struct line *) l2)->data;
5320 switch (get_sort_field(branch_sort_state)) {
5321 case ORDERBY_DATE:
5322 return sort_order(branch_sort_state, timecmp(&branch1->time, &branch2->time));
5324 case ORDERBY_AUTHOR:
5325 return sort_order(branch_sort_state, strcmp(branch1->author, branch2->author));
5327 case ORDERBY_NAME:
5328 default:
5329 return sort_order(branch_sort_state, strcmp(branch1->ref->name, branch2->ref->name));
5330 }
5331 }
5333 static bool
5334 branch_draw(struct view *view, struct line *line, unsigned int lineno)
5335 {
5336 struct branch *branch = line->data;
5337 enum line_type type = branch->ref->head ? LINE_MAIN_HEAD : LINE_DEFAULT;
5339 if (opt_date && draw_date(view, &branch->time))
5340 return TRUE;
5342 if (opt_author && draw_author(view, branch->author))
5343 return TRUE;
5345 draw_text(view, type, branch->ref == &branch_all ? "All branches" : branch->ref->name, TRUE);
5346 return TRUE;
5347 }
5349 static enum request
5350 branch_request(struct view *view, enum request request, struct line *line)
5351 {
5352 struct branch *branch = line->data;
5354 switch (request) {
5355 case REQ_REFRESH:
5356 load_refs();
5357 open_view(view, REQ_VIEW_BRANCH, OPEN_REFRESH);
5358 return REQ_NONE;
5360 case REQ_TOGGLE_SORT_FIELD:
5361 case REQ_TOGGLE_SORT_ORDER:
5362 sort_view(view, request, &branch_sort_state, branch_compare);
5363 return REQ_NONE;
5365 case REQ_ENTER:
5366 if (branch->ref == &branch_all) {
5367 const char *all_branches_argv[] = {
5368 "git", "log", "--no-color", "--pretty=raw", "--parents",
5369 "--topo-order", "--all", NULL
5370 };
5371 struct view *main_view = VIEW(REQ_VIEW_MAIN);
5373 if (!prepare_update(main_view, all_branches_argv, NULL)) {
5374 report("Failed to load view of all branches");
5375 return REQ_NONE;
5376 }
5377 open_view(view, REQ_VIEW_MAIN, OPEN_PREPARED | OPEN_SPLIT);
5378 } else {
5379 open_view(view, REQ_VIEW_MAIN, OPEN_SPLIT);
5380 }
5381 return REQ_NONE;
5383 default:
5384 return request;
5385 }
5386 }
5388 static bool
5389 branch_read(struct view *view, char *line)
5390 {
5391 static char id[SIZEOF_REV];
5392 struct branch *reference;
5393 size_t i;
5395 if (!line)
5396 return TRUE;
5398 switch (get_line_type(line)) {
5399 case LINE_COMMIT:
5400 string_copy_rev(id, line + STRING_SIZE("commit "));
5401 return TRUE;
5403 case LINE_AUTHOR:
5404 for (i = 0, reference = NULL; i < view->lines; i++) {
5405 struct branch *branch = view->line[i].data;
5407 if (strcmp(branch->ref->id, id))
5408 continue;
5410 view->line[i].dirty = TRUE;
5411 if (reference) {
5412 branch->author = reference->author;
5413 branch->time = reference->time;
5414 continue;
5415 }
5417 parse_author_line(line + STRING_SIZE("author "),
5418 &branch->author, &branch->time);
5419 reference = branch;
5420 }
5421 return TRUE;
5423 default:
5424 return TRUE;
5425 }
5427 }
5429 static bool
5430 branch_open_visitor(void *data, const struct ref *ref)
5431 {
5432 struct view *view = data;
5433 struct branch *branch;
5435 if (ref->tag || ref->ltag || ref->remote)
5436 return TRUE;
5438 branch = calloc(1, sizeof(*branch));
5439 if (!branch)
5440 return FALSE;
5442 branch->ref = ref;
5443 return !!add_line_data(view, branch, LINE_DEFAULT);
5444 }
5446 static bool
5447 branch_open(struct view *view)
5448 {
5449 const char *branch_log[] = {
5450 "git", "log", "--no-color", "--pretty=raw",
5451 "--simplify-by-decoration", "--all", NULL
5452 };
5454 if (!io_run_rd(&view->io, branch_log, NULL, FORMAT_NONE)) {
5455 report("Failed to load branch data");
5456 return TRUE;
5457 }
5459 setup_update(view, view->id);
5460 branch_open_visitor(view, &branch_all);
5461 foreach_ref(branch_open_visitor, view);
5462 view->p_restore = TRUE;
5464 return TRUE;
5465 }
5467 static bool
5468 branch_grep(struct view *view, struct line *line)
5469 {
5470 struct branch *branch = line->data;
5471 const char *text[] = {
5472 branch->ref->name,
5473 branch->author,
5474 NULL
5475 };
5477 return grep_text(view, text);
5478 }
5480 static void
5481 branch_select(struct view *view, struct line *line)
5482 {
5483 struct branch *branch = line->data;
5485 string_copy_rev(view->ref, branch->ref->id);
5486 string_copy_rev(ref_commit, branch->ref->id);
5487 string_copy_rev(ref_head, branch->ref->id);
5488 string_copy_rev(ref_branch, branch->ref->name);
5489 }
5491 static struct view_ops branch_ops = {
5492 "branch",
5493 NULL,
5494 branch_open,
5495 branch_read,
5496 branch_draw,
5497 branch_request,
5498 branch_grep,
5499 branch_select,
5500 };
5502 /*
5503 * Status backend
5504 */
5506 struct status {
5507 char status;
5508 struct {
5509 mode_t mode;
5510 char rev[SIZEOF_REV];
5511 char name[SIZEOF_STR];
5512 } old;
5513 struct {
5514 mode_t mode;
5515 char rev[SIZEOF_REV];
5516 char name[SIZEOF_STR];
5517 } new;
5518 };
5520 static char status_onbranch[SIZEOF_STR];
5521 static struct status stage_status;
5522 static enum line_type stage_line_type;
5523 static size_t stage_chunks;
5524 static int *stage_chunk;
5526 DEFINE_ALLOCATOR(realloc_ints, int, 32)
5528 /* This should work even for the "On branch" line. */
5529 static inline bool
5530 status_has_none(struct view *view, struct line *line)
5531 {
5532 return line < view->line + view->lines && !line[1].data;
5533 }
5535 /* Get fields from the diff line:
5536 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
5537 */
5538 static inline bool
5539 status_get_diff(struct status *file, const char *buf, size_t bufsize)
5540 {
5541 const char *old_mode = buf + 1;
5542 const char *new_mode = buf + 8;
5543 const char *old_rev = buf + 15;
5544 const char *new_rev = buf + 56;
5545 const char *status = buf + 97;
5547 if (bufsize < 98 ||
5548 old_mode[-1] != ':' ||
5549 new_mode[-1] != ' ' ||
5550 old_rev[-1] != ' ' ||
5551 new_rev[-1] != ' ' ||
5552 status[-1] != ' ')
5553 return FALSE;
5555 file->status = *status;
5557 string_copy_rev(file->old.rev, old_rev);
5558 string_copy_rev(file->new.rev, new_rev);
5560 file->old.mode = strtoul(old_mode, NULL, 8);
5561 file->new.mode = strtoul(new_mode, NULL, 8);
5563 file->old.name[0] = file->new.name[0] = 0;
5565 return TRUE;
5566 }
5568 static bool
5569 status_run(struct view *view, const char *argv[], char status, enum line_type type)
5570 {
5571 struct status *unmerged = NULL;
5572 char *buf;
5573 struct io io = {};
5575 if (!io_run(&io, argv, opt_cdup, IO_RD))
5576 return FALSE;
5578 add_line_data(view, NULL, type);
5580 while ((buf = io_get(&io, 0, TRUE))) {
5581 struct status *file = unmerged;
5583 if (!file) {
5584 file = calloc(1, sizeof(*file));
5585 if (!file || !add_line_data(view, file, type))
5586 goto error_out;
5587 }
5589 /* Parse diff info part. */
5590 if (status) {
5591 file->status = status;
5592 if (status == 'A')
5593 string_copy(file->old.rev, NULL_ID);
5595 } else if (!file->status || file == unmerged) {
5596 if (!status_get_diff(file, buf, strlen(buf)))
5597 goto error_out;
5599 buf = io_get(&io, 0, TRUE);
5600 if (!buf)
5601 break;
5603 /* Collapse all modified entries that follow an
5604 * associated unmerged entry. */
5605 if (unmerged == file) {
5606 unmerged->status = 'U';
5607 unmerged = NULL;
5608 } else if (file->status == 'U') {
5609 unmerged = file;
5610 }
5611 }
5613 /* Grab the old name for rename/copy. */
5614 if (!*file->old.name &&
5615 (file->status == 'R' || file->status == 'C')) {
5616 string_ncopy(file->old.name, buf, strlen(buf));
5618 buf = io_get(&io, 0, TRUE);
5619 if (!buf)
5620 break;
5621 }
5623 /* git-ls-files just delivers a NUL separated list of
5624 * file names similar to the second half of the
5625 * git-diff-* output. */
5626 string_ncopy(file->new.name, buf, strlen(buf));
5627 if (!*file->old.name)
5628 string_copy(file->old.name, file->new.name);
5629 file = NULL;
5630 }
5632 if (io_error(&io)) {
5633 error_out:
5634 io_done(&io);
5635 return FALSE;
5636 }
5638 if (!view->line[view->lines - 1].data)
5639 add_line_data(view, NULL, LINE_STAT_NONE);
5641 io_done(&io);
5642 return TRUE;
5643 }
5645 /* Don't show unmerged entries in the staged section. */
5646 static const char *status_diff_index_argv[] = {
5647 "git", "diff-index", "-z", "--diff-filter=ACDMRTXB",
5648 "--cached", "-M", "HEAD", NULL
5649 };
5651 static const char *status_diff_files_argv[] = {
5652 "git", "diff-files", "-z", NULL
5653 };
5655 static const char *status_list_other_argv[] = {
5656 "git", "ls-files", "-z", "--others", "--exclude-standard", opt_prefix, NULL
5657 };
5659 static const char *status_list_no_head_argv[] = {
5660 "git", "ls-files", "-z", "--cached", "--exclude-standard", NULL
5661 };
5663 static const char *update_index_argv[] = {
5664 "git", "update-index", "-q", "--unmerged", "--refresh", NULL
5665 };
5667 /* Restore the previous line number to stay in the context or select a
5668 * line with something that can be updated. */
5669 static void
5670 status_restore(struct view *view)
5671 {
5672 if (view->p_lineno >= view->lines)
5673 view->p_lineno = view->lines - 1;
5674 while (view->p_lineno < view->lines && !view->line[view->p_lineno].data)
5675 view->p_lineno++;
5676 while (view->p_lineno > 0 && !view->line[view->p_lineno].data)
5677 view->p_lineno--;
5679 /* If the above fails, always skip the "On branch" line. */
5680 if (view->p_lineno < view->lines)
5681 view->lineno = view->p_lineno;
5682 else
5683 view->lineno = 1;
5685 if (view->lineno < view->offset)
5686 view->offset = view->lineno;
5687 else if (view->offset + view->height <= view->lineno)
5688 view->offset = view->lineno - view->height + 1;
5690 view->p_restore = FALSE;
5691 }
5693 static void
5694 status_update_onbranch(void)
5695 {
5696 static const char *paths[][2] = {
5697 { "rebase-apply/rebasing", "Rebasing" },
5698 { "rebase-apply/applying", "Applying mailbox" },
5699 { "rebase-apply/", "Rebasing mailbox" },
5700 { "rebase-merge/interactive", "Interactive rebase" },
5701 { "rebase-merge/", "Rebase merge" },
5702 { "MERGE_HEAD", "Merging" },
5703 { "BISECT_LOG", "Bisecting" },
5704 { "HEAD", "On branch" },
5705 };
5706 char buf[SIZEOF_STR];
5707 struct stat stat;
5708 int i;
5710 if (is_initial_commit()) {
5711 string_copy(status_onbranch, "Initial commit");
5712 return;
5713 }
5715 for (i = 0; i < ARRAY_SIZE(paths); i++) {
5716 char *head = opt_head;
5718 if (!string_format(buf, "%s/%s", opt_git_dir, paths[i][0]) ||
5719 lstat(buf, &stat) < 0)
5720 continue;
5722 if (!*opt_head) {
5723 struct io io = {};
5725 if (io_open(&io, "%s/rebase-merge/head-name", opt_git_dir) &&
5726 io_read_buf(&io, buf, sizeof(buf))) {
5727 head = buf;
5728 if (!prefixcmp(head, "refs/heads/"))
5729 head += STRING_SIZE("refs/heads/");
5730 }
5731 }
5733 if (!string_format(status_onbranch, "%s %s", paths[i][1], head))
5734 string_copy(status_onbranch, opt_head);
5735 return;
5736 }
5738 string_copy(status_onbranch, "Not currently on any branch");
5739 }
5741 /* First parse staged info using git-diff-index(1), then parse unstaged
5742 * info using git-diff-files(1), and finally untracked files using
5743 * git-ls-files(1). */
5744 static bool
5745 status_open(struct view *view)
5746 {
5747 reset_view(view);
5749 add_line_data(view, NULL, LINE_STAT_HEAD);
5750 status_update_onbranch();
5752 io_run_bg(update_index_argv);
5754 if (is_initial_commit()) {
5755 if (!status_run(view, status_list_no_head_argv, 'A', LINE_STAT_STAGED))
5756 return FALSE;
5757 } else if (!status_run(view, status_diff_index_argv, 0, LINE_STAT_STAGED)) {
5758 return FALSE;
5759 }
5761 if (!status_run(view, status_diff_files_argv, 0, LINE_STAT_UNSTAGED) ||
5762 !status_run(view, status_list_other_argv, '?', LINE_STAT_UNTRACKED))
5763 return FALSE;
5765 /* Restore the exact position or use the specialized restore
5766 * mode? */
5767 if (!view->p_restore)
5768 status_restore(view);
5769 return TRUE;
5770 }
5772 static bool
5773 status_draw(struct view *view, struct line *line, unsigned int lineno)
5774 {
5775 struct status *status = line->data;
5776 enum line_type type;
5777 const char *text;
5779 if (!status) {
5780 switch (line->type) {
5781 case LINE_STAT_STAGED:
5782 type = LINE_STAT_SECTION;
5783 text = "Changes to be committed:";
5784 break;
5786 case LINE_STAT_UNSTAGED:
5787 type = LINE_STAT_SECTION;
5788 text = "Changed but not updated:";
5789 break;
5791 case LINE_STAT_UNTRACKED:
5792 type = LINE_STAT_SECTION;
5793 text = "Untracked files:";
5794 break;
5796 case LINE_STAT_NONE:
5797 type = LINE_DEFAULT;
5798 text = " (no files)";
5799 break;
5801 case LINE_STAT_HEAD:
5802 type = LINE_STAT_HEAD;
5803 text = status_onbranch;
5804 break;
5806 default:
5807 return FALSE;
5808 }
5809 } else {
5810 static char buf[] = { '?', ' ', ' ', ' ', 0 };
5812 buf[0] = status->status;
5813 if (draw_text(view, line->type, buf, TRUE))
5814 return TRUE;
5815 type = LINE_DEFAULT;
5816 text = status->new.name;
5817 }
5819 draw_text(view, type, text, TRUE);
5820 return TRUE;
5821 }
5823 static enum request
5824 status_load_error(struct view *view, struct view *stage, const char *path)
5825 {
5826 if (displayed_views() == 2 || display[current_view] != view)
5827 maximize_view(view);
5828 report("Failed to load '%s': %s", path, io_strerror(&stage->io));
5829 return REQ_NONE;
5830 }
5832 static enum request
5833 status_enter(struct view *view, struct line *line)
5834 {
5835 struct status *status = line->data;
5836 const char *oldpath = status ? status->old.name : NULL;
5837 /* Diffs for unmerged entries are empty when passing the new
5838 * path, so leave it empty. */
5839 const char *newpath = status && status->status != 'U' ? status->new.name : NULL;
5840 const char *info;
5841 enum open_flags split;
5842 struct view *stage = VIEW(REQ_VIEW_STAGE);
5844 if (line->type == LINE_STAT_NONE ||
5845 (!status && line[1].type == LINE_STAT_NONE)) {
5846 report("No file to diff");
5847 return REQ_NONE;
5848 }
5850 switch (line->type) {
5851 case LINE_STAT_STAGED:
5852 if (is_initial_commit()) {
5853 const char *no_head_diff_argv[] = {
5854 "git", "diff", "--no-color", "--patch-with-stat",
5855 "--", "/dev/null", newpath, NULL
5856 };
5858 if (!prepare_update(stage, no_head_diff_argv, opt_cdup))
5859 return status_load_error(view, stage, newpath);
5860 } else {
5861 const char *index_show_argv[] = {
5862 "git", "diff-index", "--root", "--patch-with-stat",
5863 "-C", "-M", "--cached", "HEAD", "--",
5864 oldpath, newpath, NULL
5865 };
5867 if (!prepare_update(stage, index_show_argv, opt_cdup))
5868 return status_load_error(view, stage, newpath);
5869 }
5871 if (status)
5872 info = "Staged changes to %s";
5873 else
5874 info = "Staged changes";
5875 break;
5877 case LINE_STAT_UNSTAGED:
5878 {
5879 const char *files_show_argv[] = {
5880 "git", "diff-files", "--root", "--patch-with-stat",
5881 "-C", "-M", "--", oldpath, newpath, NULL
5882 };
5884 if (!prepare_update(stage, files_show_argv, opt_cdup))
5885 return status_load_error(view, stage, newpath);
5886 if (status)
5887 info = "Unstaged changes to %s";
5888 else
5889 info = "Unstaged changes";
5890 break;
5891 }
5892 case LINE_STAT_UNTRACKED:
5893 if (!newpath) {
5894 report("No file to show");
5895 return REQ_NONE;
5896 }
5898 if (!suffixcmp(status->new.name, -1, "/")) {
5899 report("Cannot display a directory");
5900 return REQ_NONE;
5901 }
5903 if (!prepare_update_file(stage, newpath))
5904 return status_load_error(view, stage, newpath);
5905 info = "Untracked file %s";
5906 break;
5908 case LINE_STAT_HEAD:
5909 return REQ_NONE;
5911 default:
5912 die("line type %d not handled in switch", line->type);
5913 }
5915 split = view_is_displayed(view) ? OPEN_SPLIT : OPEN_DEFAULT;
5916 open_view(view, REQ_VIEW_STAGE, OPEN_PREPARED | split);
5917 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
5918 if (status) {
5919 stage_status = *status;
5920 } else {
5921 memset(&stage_status, 0, sizeof(stage_status));
5922 }
5924 stage_line_type = line->type;
5925 stage_chunks = 0;
5926 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
5927 }
5929 return REQ_NONE;
5930 }
5932 static bool
5933 status_exists(struct status *status, enum line_type type)
5934 {
5935 struct view *view = VIEW(REQ_VIEW_STATUS);
5936 unsigned long lineno;
5938 for (lineno = 0; lineno < view->lines; lineno++) {
5939 struct line *line = &view->line[lineno];
5940 struct status *pos = line->data;
5942 if (line->type != type)
5943 continue;
5944 if (!pos && (!status || !status->status) && line[1].data) {
5945 select_view_line(view, lineno);
5946 return TRUE;
5947 }
5948 if (pos && !strcmp(status->new.name, pos->new.name)) {
5949 select_view_line(view, lineno);
5950 return TRUE;
5951 }
5952 }
5954 return FALSE;
5955 }
5958 static bool
5959 status_update_prepare(struct io *io, enum line_type type)
5960 {
5961 const char *staged_argv[] = {
5962 "git", "update-index", "-z", "--index-info", NULL
5963 };
5964 const char *others_argv[] = {
5965 "git", "update-index", "-z", "--add", "--remove", "--stdin", NULL
5966 };
5968 switch (type) {
5969 case LINE_STAT_STAGED:
5970 return io_run(io, staged_argv, opt_cdup, IO_WR);
5972 case LINE_STAT_UNSTAGED:
5973 case LINE_STAT_UNTRACKED:
5974 return io_run(io, others_argv, opt_cdup, IO_WR);
5976 default:
5977 die("line type %d not handled in switch", type);
5978 return FALSE;
5979 }
5980 }
5982 static bool
5983 status_update_write(struct io *io, struct status *status, enum line_type type)
5984 {
5985 char buf[SIZEOF_STR];
5986 size_t bufsize = 0;
5988 switch (type) {
5989 case LINE_STAT_STAGED:
5990 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
5991 status->old.mode,
5992 status->old.rev,
5993 status->old.name, 0))
5994 return FALSE;
5995 break;
5997 case LINE_STAT_UNSTAGED:
5998 case LINE_STAT_UNTRACKED:
5999 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
6000 return FALSE;
6001 break;
6003 default:
6004 die("line type %d not handled in switch", type);
6005 }
6007 return io_write(io, buf, bufsize);
6008 }
6010 static bool
6011 status_update_file(struct status *status, enum line_type type)
6012 {
6013 struct io io = {};
6014 bool result;
6016 if (!status_update_prepare(&io, type))
6017 return FALSE;
6019 result = status_update_write(&io, status, type);
6020 return io_done(&io) && result;
6021 }
6023 static bool
6024 status_update_files(struct view *view, struct line *line)
6025 {
6026 char buf[sizeof(view->ref)];
6027 struct io io = {};
6028 bool result = TRUE;
6029 struct line *pos = view->line + view->lines;
6030 int files = 0;
6031 int file, done;
6032 int cursor_y = -1, cursor_x = -1;
6034 if (!status_update_prepare(&io, line->type))
6035 return FALSE;
6037 for (pos = line; pos < view->line + view->lines && pos->data; pos++)
6038 files++;
6040 string_copy(buf, view->ref);
6041 getsyx(cursor_y, cursor_x);
6042 for (file = 0, done = 5; result && file < files; line++, file++) {
6043 int almost_done = file * 100 / files;
6045 if (almost_done > done) {
6046 done = almost_done;
6047 string_format(view->ref, "updating file %u of %u (%d%% done)",
6048 file, files, done);
6049 update_view_title(view);
6050 setsyx(cursor_y, cursor_x);
6051 doupdate();
6052 }
6053 result = status_update_write(&io, line->data, line->type);
6054 }
6055 string_copy(view->ref, buf);
6057 return io_done(&io) && result;
6058 }
6060 static bool
6061 status_update(struct view *view)
6062 {
6063 struct line *line = &view->line[view->lineno];
6065 assert(view->lines);
6067 if (!line->data) {
6068 /* This should work even for the "On branch" line. */
6069 if (line < view->line + view->lines && !line[1].data) {
6070 report("Nothing to update");
6071 return FALSE;
6072 }
6074 if (!status_update_files(view, line + 1)) {
6075 report("Failed to update file status");
6076 return FALSE;
6077 }
6079 } else if (!status_update_file(line->data, line->type)) {
6080 report("Failed to update file status");
6081 return FALSE;
6082 }
6084 return TRUE;
6085 }
6087 static bool
6088 status_revert(struct status *status, enum line_type type, bool has_none)
6089 {
6090 if (!status || type != LINE_STAT_UNSTAGED) {
6091 if (type == LINE_STAT_STAGED) {
6092 report("Cannot revert changes to staged files");
6093 } else if (type == LINE_STAT_UNTRACKED) {
6094 report("Cannot revert changes to untracked files");
6095 } else if (has_none) {
6096 report("Nothing to revert");
6097 } else {
6098 report("Cannot revert changes to multiple files");
6099 }
6101 } else if (prompt_yesno("Are you sure you want to revert changes?")) {
6102 char mode[10] = "100644";
6103 const char *reset_argv[] = {
6104 "git", "update-index", "--cacheinfo", mode,
6105 status->old.rev, status->old.name, NULL
6106 };
6107 const char *checkout_argv[] = {
6108 "git", "checkout", "--", status->old.name, NULL
6109 };
6111 if (status->status == 'U') {
6112 string_format(mode, "%5o", status->old.mode);
6114 if (status->old.mode == 0 && status->new.mode == 0) {
6115 reset_argv[2] = "--force-remove";
6116 reset_argv[3] = status->old.name;
6117 reset_argv[4] = NULL;
6118 }
6120 if (!io_run_fg(reset_argv, opt_cdup))
6121 return FALSE;
6122 if (status->old.mode == 0 && status->new.mode == 0)
6123 return TRUE;
6124 }
6126 return io_run_fg(checkout_argv, opt_cdup);
6127 }
6129 return FALSE;
6130 }
6132 static enum request
6133 status_request(struct view *view, enum request request, struct line *line)
6134 {
6135 struct status *status = line->data;
6137 switch (request) {
6138 case REQ_STATUS_UPDATE:
6139 if (!status_update(view))
6140 return REQ_NONE;
6141 break;
6143 case REQ_STATUS_REVERT:
6144 if (!status_revert(status, line->type, status_has_none(view, line)))
6145 return REQ_NONE;
6146 break;
6148 case REQ_STATUS_MERGE:
6149 if (!status || status->status != 'U') {
6150 report("Merging only possible for files with unmerged status ('U').");
6151 return REQ_NONE;
6152 }
6153 open_mergetool(status->new.name);
6154 break;
6156 case REQ_EDIT:
6157 if (!status)
6158 return request;
6159 if (status->status == 'D') {
6160 report("File has been deleted.");
6161 return REQ_NONE;
6162 }
6164 open_editor(status->new.name);
6165 break;
6167 case REQ_VIEW_BLAME:
6168 if (status)
6169 opt_ref[0] = 0;
6170 return request;
6172 case REQ_ENTER:
6173 /* After returning the status view has been split to
6174 * show the stage view. No further reloading is
6175 * necessary. */
6176 return status_enter(view, line);
6178 case REQ_REFRESH:
6179 /* Simply reload the view. */
6180 break;
6182 default:
6183 return request;
6184 }
6186 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
6188 return REQ_NONE;
6189 }
6191 static void
6192 status_select(struct view *view, struct line *line)
6193 {
6194 struct status *status = line->data;
6195 char file[SIZEOF_STR] = "all files";
6196 const char *text;
6197 const char *key;
6199 if (status && !string_format(file, "'%s'", status->new.name))
6200 return;
6202 if (!status && line[1].type == LINE_STAT_NONE)
6203 line++;
6205 switch (line->type) {
6206 case LINE_STAT_STAGED:
6207 text = "Press %s to unstage %s for commit";
6208 break;
6210 case LINE_STAT_UNSTAGED:
6211 text = "Press %s to stage %s for commit";
6212 break;
6214 case LINE_STAT_UNTRACKED:
6215 text = "Press %s to stage %s for addition";
6216 break;
6218 case LINE_STAT_HEAD:
6219 case LINE_STAT_NONE:
6220 text = "Nothing to update";
6221 break;
6223 default:
6224 die("line type %d not handled in switch", line->type);
6225 }
6227 if (status && status->status == 'U') {
6228 text = "Press %s to resolve conflict in %s";
6229 key = get_key(KEYMAP_STATUS, REQ_STATUS_MERGE);
6231 } else {
6232 key = get_key(KEYMAP_STATUS, REQ_STATUS_UPDATE);
6233 }
6235 string_format(view->ref, text, key, file);
6236 if (status)
6237 string_copy(opt_file, status->new.name);
6238 }
6240 static bool
6241 status_grep(struct view *view, struct line *line)
6242 {
6243 struct status *status = line->data;
6245 if (status) {
6246 const char buf[2] = { status->status, 0 };
6247 const char *text[] = { status->new.name, buf, NULL };
6249 return grep_text(view, text);
6250 }
6252 return FALSE;
6253 }
6255 static struct view_ops status_ops = {
6256 "file",
6257 NULL,
6258 status_open,
6259 NULL,
6260 status_draw,
6261 status_request,
6262 status_grep,
6263 status_select,
6264 };
6267 static bool
6268 stage_diff_write(struct io *io, struct line *line, struct line *end)
6269 {
6270 while (line < end) {
6271 if (!io_write(io, line->data, strlen(line->data)) ||
6272 !io_write(io, "\n", 1))
6273 return FALSE;
6274 line++;
6275 if (line->type == LINE_DIFF_CHUNK ||
6276 line->type == LINE_DIFF_HEADER)
6277 break;
6278 }
6280 return TRUE;
6281 }
6283 static struct line *
6284 stage_diff_find(struct view *view, struct line *line, enum line_type type)
6285 {
6286 for (; view->line < line; line--)
6287 if (line->type == type)
6288 return line;
6290 return NULL;
6291 }
6293 static bool
6294 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
6295 {
6296 const char *apply_argv[SIZEOF_ARG] = {
6297 "git", "apply", "--whitespace=nowarn", NULL
6298 };
6299 struct line *diff_hdr;
6300 struct io io = {};
6301 int argc = 3;
6303 diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
6304 if (!diff_hdr)
6305 return FALSE;
6307 if (!revert)
6308 apply_argv[argc++] = "--cached";
6309 if (revert || stage_line_type == LINE_STAT_STAGED)
6310 apply_argv[argc++] = "-R";
6311 apply_argv[argc++] = "-";
6312 apply_argv[argc++] = NULL;
6313 if (!io_run(&io, apply_argv, opt_cdup, IO_WR))
6314 return FALSE;
6316 if (!stage_diff_write(&io, diff_hdr, chunk) ||
6317 !stage_diff_write(&io, chunk, view->line + view->lines))
6318 chunk = NULL;
6320 io_done(&io);
6321 io_run_bg(update_index_argv);
6323 return chunk ? TRUE : FALSE;
6324 }
6326 static bool
6327 stage_update(struct view *view, struct line *line)
6328 {
6329 struct line *chunk = NULL;
6331 if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
6332 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
6334 if (chunk) {
6335 if (!stage_apply_chunk(view, chunk, FALSE)) {
6336 report("Failed to apply chunk");
6337 return FALSE;
6338 }
6340 } else if (!stage_status.status) {
6341 view = VIEW(REQ_VIEW_STATUS);
6343 for (line = view->line; line < view->line + view->lines; line++)
6344 if (line->type == stage_line_type)
6345 break;
6347 if (!status_update_files(view, line + 1)) {
6348 report("Failed to update files");
6349 return FALSE;
6350 }
6352 } else if (!status_update_file(&stage_status, stage_line_type)) {
6353 report("Failed to update file");
6354 return FALSE;
6355 }
6357 return TRUE;
6358 }
6360 static bool
6361 stage_revert(struct view *view, struct line *line)
6362 {
6363 struct line *chunk = NULL;
6365 if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
6366 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
6368 if (chunk) {
6369 if (!prompt_yesno("Are you sure you want to revert changes?"))
6370 return FALSE;
6372 if (!stage_apply_chunk(view, chunk, TRUE)) {
6373 report("Failed to revert chunk");
6374 return FALSE;
6375 }
6376 return TRUE;
6378 } else {
6379 return status_revert(stage_status.status ? &stage_status : NULL,
6380 stage_line_type, FALSE);
6381 }
6382 }
6385 static void
6386 stage_next(struct view *view, struct line *line)
6387 {
6388 int i;
6390 if (!stage_chunks) {
6391 for (line = view->line; line < view->line + view->lines; line++) {
6392 if (line->type != LINE_DIFF_CHUNK)
6393 continue;
6395 if (!realloc_ints(&stage_chunk, stage_chunks, 1)) {
6396 report("Allocation failure");
6397 return;
6398 }
6400 stage_chunk[stage_chunks++] = line - view->line;
6401 }
6402 }
6404 for (i = 0; i < stage_chunks; i++) {
6405 if (stage_chunk[i] > view->lineno) {
6406 do_scroll_view(view, stage_chunk[i] - view->lineno);
6407 report("Chunk %d of %d", i + 1, stage_chunks);
6408 return;
6409 }
6410 }
6412 report("No next chunk found");
6413 }
6415 static enum request
6416 stage_request(struct view *view, enum request request, struct line *line)
6417 {
6418 switch (request) {
6419 case REQ_STATUS_UPDATE:
6420 if (!stage_update(view, line))
6421 return REQ_NONE;
6422 break;
6424 case REQ_STATUS_REVERT:
6425 if (!stage_revert(view, line))
6426 return REQ_NONE;
6427 break;
6429 case REQ_STAGE_NEXT:
6430 if (stage_line_type == LINE_STAT_UNTRACKED) {
6431 report("File is untracked; press %s to add",
6432 get_key(KEYMAP_STAGE, REQ_STATUS_UPDATE));
6433 return REQ_NONE;
6434 }
6435 stage_next(view, line);
6436 return REQ_NONE;
6438 case REQ_EDIT:
6439 if (!stage_status.new.name[0])
6440 return request;
6441 if (stage_status.status == 'D') {
6442 report("File has been deleted.");
6443 return REQ_NONE;
6444 }
6446 open_editor(stage_status.new.name);
6447 break;
6449 case REQ_REFRESH:
6450 /* Reload everything ... */
6451 break;
6453 case REQ_VIEW_BLAME:
6454 if (stage_status.new.name[0]) {
6455 string_copy(opt_file, stage_status.new.name);
6456 opt_ref[0] = 0;
6457 }
6458 return request;
6460 case REQ_ENTER:
6461 return pager_request(view, request, line);
6463 default:
6464 return request;
6465 }
6467 VIEW(REQ_VIEW_STATUS)->p_restore = TRUE;
6468 open_view(view, REQ_VIEW_STATUS, OPEN_REFRESH);
6470 /* Check whether the staged entry still exists, and close the
6471 * stage view if it doesn't. */
6472 if (!status_exists(&stage_status, stage_line_type)) {
6473 status_restore(VIEW(REQ_VIEW_STATUS));
6474 return REQ_VIEW_CLOSE;
6475 }
6477 if (stage_line_type == LINE_STAT_UNTRACKED) {
6478 if (!suffixcmp(stage_status.new.name, -1, "/")) {
6479 report("Cannot display a directory");
6480 return REQ_NONE;
6481 }
6483 if (!prepare_update_file(view, stage_status.new.name)) {
6484 report("Failed to open file: %s", strerror(errno));
6485 return REQ_NONE;
6486 }
6487 }
6488 open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
6490 return REQ_NONE;
6491 }
6493 static struct view_ops stage_ops = {
6494 "line",
6495 NULL,
6496 NULL,
6497 pager_read,
6498 pager_draw,
6499 stage_request,
6500 pager_grep,
6501 pager_select,
6502 };
6505 /*
6506 * Revision graph
6507 */
6509 struct commit {
6510 char id[SIZEOF_REV]; /* SHA1 ID. */
6511 char title[128]; /* First line of the commit message. */
6512 const char *author; /* Author of the commit. */
6513 struct time time; /* Date from the author ident. */
6514 struct ref_list *refs; /* Repository references. */
6515 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
6516 size_t graph_size; /* The width of the graph array. */
6517 bool has_parents; /* Rewritten --parents seen. */
6518 };
6520 /* Size of rev graph with no "padding" columns */
6521 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
6523 struct rev_graph {
6524 struct rev_graph *prev, *next, *parents;
6525 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
6526 size_t size;
6527 struct commit *commit;
6528 size_t pos;
6529 unsigned int boundary:1;
6530 };
6532 /* Parents of the commit being visualized. */
6533 static struct rev_graph graph_parents[4];
6535 /* The current stack of revisions on the graph. */
6536 static struct rev_graph graph_stacks[4] = {
6537 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
6538 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
6539 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
6540 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
6541 };
6543 static inline bool
6544 graph_parent_is_merge(struct rev_graph *graph)
6545 {
6546 return graph->parents->size > 1;
6547 }
6549 static inline void
6550 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
6551 {
6552 struct commit *commit = graph->commit;
6554 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
6555 commit->graph[commit->graph_size++] = symbol;
6556 }
6558 static void
6559 clear_rev_graph(struct rev_graph *graph)
6560 {
6561 graph->boundary = 0;
6562 graph->size = graph->pos = 0;
6563 graph->commit = NULL;
6564 memset(graph->parents, 0, sizeof(*graph->parents));
6565 }
6567 static void
6568 done_rev_graph(struct rev_graph *graph)
6569 {
6570 if (graph_parent_is_merge(graph) &&
6571 graph->pos < graph->size - 1 &&
6572 graph->next->size == graph->size + graph->parents->size - 1) {
6573 size_t i = graph->pos + graph->parents->size - 1;
6575 graph->commit->graph_size = i * 2;
6576 while (i < graph->next->size - 1) {
6577 append_to_rev_graph(graph, ' ');
6578 append_to_rev_graph(graph, '\\');
6579 i++;
6580 }
6581 }
6583 clear_rev_graph(graph);
6584 }
6586 static void
6587 push_rev_graph(struct rev_graph *graph, const char *parent)
6588 {
6589 int i;
6591 /* "Collapse" duplicate parents lines.
6592 *
6593 * FIXME: This needs to also update update the drawn graph but
6594 * for now it just serves as a method for pruning graph lines. */
6595 for (i = 0; i < graph->size; i++)
6596 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
6597 return;
6599 if (graph->size < SIZEOF_REVITEMS) {
6600 string_copy_rev(graph->rev[graph->size++], parent);
6601 }
6602 }
6604 static chtype
6605 get_rev_graph_symbol(struct rev_graph *graph)
6606 {
6607 chtype symbol;
6609 if (graph->boundary)
6610 symbol = REVGRAPH_BOUND;
6611 else if (graph->parents->size == 0)
6612 symbol = REVGRAPH_INIT;
6613 else if (graph_parent_is_merge(graph))
6614 symbol = REVGRAPH_MERGE;
6615 else if (graph->pos >= graph->size)
6616 symbol = REVGRAPH_BRANCH;
6617 else
6618 symbol = REVGRAPH_COMMIT;
6620 return symbol;
6621 }
6623 static void
6624 draw_rev_graph(struct rev_graph *graph)
6625 {
6626 struct rev_filler {
6627 chtype separator, line;
6628 };
6629 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
6630 static struct rev_filler fillers[] = {
6631 { ' ', '|' },
6632 { '`', '.' },
6633 { '\'', ' ' },
6634 { '/', ' ' },
6635 };
6636 chtype symbol = get_rev_graph_symbol(graph);
6637 struct rev_filler *filler;
6638 size_t i;
6640 fillers[DEFAULT].line = opt_line_graphics ? ACS_VLINE : '|';
6641 filler = &fillers[DEFAULT];
6643 for (i = 0; i < graph->pos; i++) {
6644 append_to_rev_graph(graph, filler->line);
6645 if (graph_parent_is_merge(graph->prev) &&
6646 graph->prev->pos == i)
6647 filler = &fillers[RSHARP];
6649 append_to_rev_graph(graph, filler->separator);
6650 }
6652 /* Place the symbol for this revision. */
6653 append_to_rev_graph(graph, symbol);
6655 if (graph->prev->size > graph->size)
6656 filler = &fillers[RDIAG];
6657 else
6658 filler = &fillers[DEFAULT];
6660 i++;
6662 for (; i < graph->size; i++) {
6663 append_to_rev_graph(graph, filler->separator);
6664 append_to_rev_graph(graph, filler->line);
6665 if (graph_parent_is_merge(graph->prev) &&
6666 i < graph->prev->pos + graph->parents->size)
6667 filler = &fillers[RSHARP];
6668 if (graph->prev->size > graph->size)
6669 filler = &fillers[LDIAG];
6670 }
6672 if (graph->prev->size > graph->size) {
6673 append_to_rev_graph(graph, filler->separator);
6674 if (filler->line != ' ')
6675 append_to_rev_graph(graph, filler->line);
6676 }
6677 }
6679 /* Prepare the next rev graph */
6680 static void
6681 prepare_rev_graph(struct rev_graph *graph)
6682 {
6683 size_t i;
6685 /* First, traverse all lines of revisions up to the active one. */
6686 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
6687 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
6688 break;
6690 push_rev_graph(graph->next, graph->rev[graph->pos]);
6691 }
6693 /* Interleave the new revision parent(s). */
6694 for (i = 0; !graph->boundary && i < graph->parents->size; i++)
6695 push_rev_graph(graph->next, graph->parents->rev[i]);
6697 /* Lastly, put any remaining revisions. */
6698 for (i = graph->pos + 1; i < graph->size; i++)
6699 push_rev_graph(graph->next, graph->rev[i]);
6700 }
6702 static void
6703 update_rev_graph(struct view *view, struct rev_graph *graph)
6704 {
6705 /* If this is the finalizing update ... */
6706 if (graph->commit)
6707 prepare_rev_graph(graph);
6709 /* Graph visualization needs a one rev look-ahead,
6710 * so the first update doesn't visualize anything. */
6711 if (!graph->prev->commit)
6712 return;
6714 if (view->lines > 2)
6715 view->line[view->lines - 3].dirty = 1;
6716 if (view->lines > 1)
6717 view->line[view->lines - 2].dirty = 1;
6718 draw_rev_graph(graph->prev);
6719 done_rev_graph(graph->prev->prev);
6720 }
6723 /*
6724 * Main view backend
6725 */
6727 static const char *main_argv[SIZEOF_ARG] = {
6728 "git", "log", "--no-color", "--pretty=raw", "--parents",
6729 "--topo-order", "%(head)", NULL
6730 };
6732 static bool
6733 main_draw(struct view *view, struct line *line, unsigned int lineno)
6734 {
6735 struct commit *commit = line->data;
6737 if (!commit->author)
6738 return FALSE;
6740 if (opt_date && draw_date(view, &commit->time))
6741 return TRUE;
6743 if (opt_author && draw_author(view, commit->author))
6744 return TRUE;
6746 if (opt_rev_graph && commit->graph_size &&
6747 draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
6748 return TRUE;
6750 if (opt_show_refs && commit->refs) {
6751 size_t i;
6753 for (i = 0; i < commit->refs->size; i++) {
6754 struct ref *ref = commit->refs->refs[i];
6755 enum line_type type;
6757 if (ref->head)
6758 type = LINE_MAIN_HEAD;
6759 else if (ref->ltag)
6760 type = LINE_MAIN_LOCAL_TAG;
6761 else if (ref->tag)
6762 type = LINE_MAIN_TAG;
6763 else if (ref->tracked)
6764 type = LINE_MAIN_TRACKED;
6765 else if (ref->remote)
6766 type = LINE_MAIN_REMOTE;
6767 else
6768 type = LINE_MAIN_REF;
6770 if (draw_text(view, type, "[", TRUE) ||
6771 draw_text(view, type, ref->name, TRUE) ||
6772 draw_text(view, type, "]", TRUE))
6773 return TRUE;
6775 if (draw_text(view, LINE_DEFAULT, " ", TRUE))
6776 return TRUE;
6777 }
6778 }
6780 draw_text(view, LINE_DEFAULT, commit->title, TRUE);
6781 return TRUE;
6782 }
6784 /* Reads git log --pretty=raw output and parses it into the commit struct. */
6785 static bool
6786 main_read(struct view *view, char *line)
6787 {
6788 static struct rev_graph *graph = graph_stacks;
6789 enum line_type type;
6790 struct commit *commit;
6792 if (!line) {
6793 int i;
6795 if (!view->lines && !view->prev)
6796 die("No revisions match the given arguments.");
6797 if (view->lines > 0) {
6798 commit = view->line[view->lines - 1].data;
6799 view->line[view->lines - 1].dirty = 1;
6800 if (!commit->author) {
6801 view->lines--;
6802 free(commit);
6803 graph->commit = NULL;
6804 }
6805 }
6806 update_rev_graph(view, graph);
6808 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
6809 clear_rev_graph(&graph_stacks[i]);
6810 return TRUE;
6811 }
6813 type = get_line_type(line);
6814 if (type == LINE_COMMIT) {
6815 commit = calloc(1, sizeof(struct commit));
6816 if (!commit)
6817 return FALSE;
6819 line += STRING_SIZE("commit ");
6820 if (*line == '-') {
6821 graph->boundary = 1;
6822 line++;
6823 }
6825 string_copy_rev(commit->id, line);
6826 commit->refs = get_ref_list(commit->id);
6827 graph->commit = commit;
6828 add_line_data(view, commit, LINE_MAIN_COMMIT);
6830 while ((line = strchr(line, ' '))) {
6831 line++;
6832 push_rev_graph(graph->parents, line);
6833 commit->has_parents = TRUE;
6834 }
6835 return TRUE;
6836 }
6838 if (!view->lines)
6839 return TRUE;
6840 commit = view->line[view->lines - 1].data;
6842 switch (type) {
6843 case LINE_PARENT:
6844 if (commit->has_parents)
6845 break;
6846 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
6847 break;
6849 case LINE_AUTHOR:
6850 parse_author_line(line + STRING_SIZE("author "),
6851 &commit->author, &commit->time);
6852 update_rev_graph(view, graph);
6853 graph = graph->next;
6854 break;
6856 default:
6857 /* Fill in the commit title if it has not already been set. */
6858 if (commit->title[0])
6859 break;
6861 /* Require titles to start with a non-space character at the
6862 * offset used by git log. */
6863 if (strncmp(line, " ", 4))
6864 break;
6865 line += 4;
6866 /* Well, if the title starts with a whitespace character,
6867 * try to be forgiving. Otherwise we end up with no title. */
6868 while (isspace(*line))
6869 line++;
6870 if (*line == '\0')
6871 break;
6872 /* FIXME: More graceful handling of titles; append "..." to
6873 * shortened titles, etc. */
6875 string_expand(commit->title, sizeof(commit->title), line, 1);
6876 view->line[view->lines - 1].dirty = 1;
6877 }
6879 return TRUE;
6880 }
6882 static enum request
6883 main_request(struct view *view, enum request request, struct line *line)
6884 {
6885 enum open_flags flags = view_is_displayed(view) ? OPEN_SPLIT : OPEN_DEFAULT;
6887 switch (request) {
6888 case REQ_ENTER:
6889 open_view(view, REQ_VIEW_DIFF, flags);
6890 break;
6891 case REQ_REFRESH:
6892 load_refs();
6893 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
6894 break;
6895 default:
6896 return request;
6897 }
6899 return REQ_NONE;
6900 }
6902 static bool
6903 grep_refs(struct ref_list *list, regex_t *regex)
6904 {
6905 regmatch_t pmatch;
6906 size_t i;
6908 if (!opt_show_refs || !list)
6909 return FALSE;
6911 for (i = 0; i < list->size; i++) {
6912 if (regexec(regex, list->refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
6913 return TRUE;
6914 }
6916 return FALSE;
6917 }
6919 static bool
6920 main_grep(struct view *view, struct line *line)
6921 {
6922 struct commit *commit = line->data;
6923 const char *text[] = {
6924 commit->title,
6925 opt_author ? commit->author : "",
6926 mkdate(&commit->time, opt_date),
6927 NULL
6928 };
6930 return grep_text(view, text) || grep_refs(commit->refs, view->regex);
6931 }
6933 static void
6934 main_select(struct view *view, struct line *line)
6935 {
6936 struct commit *commit = line->data;
6938 string_copy_rev(view->ref, commit->id);
6939 string_copy_rev(ref_commit, view->ref);
6940 }
6942 static struct view_ops main_ops = {
6943 "commit",
6944 main_argv,
6945 NULL,
6946 main_read,
6947 main_draw,
6948 main_request,
6949 main_grep,
6950 main_select,
6951 };
6954 /*
6955 * Status management
6956 */
6958 /* Whether or not the curses interface has been initialized. */
6959 static bool cursed = FALSE;
6961 /* Terminal hacks and workarounds. */
6962 static bool use_scroll_redrawwin;
6963 static bool use_scroll_status_wclear;
6965 /* The status window is used for polling keystrokes. */
6966 static WINDOW *status_win;
6968 /* Reading from the prompt? */
6969 static bool input_mode = FALSE;
6971 static bool status_empty = FALSE;
6973 /* Update status and title window. */
6974 static void
6975 report(const char *msg, ...)
6976 {
6977 struct view *view = display[current_view];
6979 if (input_mode)
6980 return;
6982 if (!view) {
6983 char buf[SIZEOF_STR];
6984 va_list args;
6986 va_start(args, msg);
6987 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
6988 buf[sizeof(buf) - 1] = 0;
6989 buf[sizeof(buf) - 2] = '.';
6990 buf[sizeof(buf) - 3] = '.';
6991 buf[sizeof(buf) - 4] = '.';
6992 }
6993 va_end(args);
6994 die("%s", buf);
6995 }
6997 if (!status_empty || *msg) {
6998 va_list args;
7000 va_start(args, msg);
7002 wmove(status_win, 0, 0);
7003 if (view->has_scrolled && use_scroll_status_wclear)
7004 wclear(status_win);
7005 if (*msg) {
7006 vwprintw(status_win, msg, args);
7007 status_empty = FALSE;
7008 } else {
7009 status_empty = TRUE;
7010 }
7011 wclrtoeol(status_win);
7012 wnoutrefresh(status_win);
7014 va_end(args);
7015 }
7017 update_view_title(view);
7018 }
7020 static void
7021 init_display(void)
7022 {
7023 const char *term;
7024 int x, y;
7026 /* Initialize the curses library */
7027 if (isatty(STDIN_FILENO)) {
7028 cursed = !!initscr();
7029 opt_tty = stdin;
7030 } else {
7031 /* Leave stdin and stdout alone when acting as a pager. */
7032 opt_tty = fopen("/dev/tty", "r+");
7033 if (!opt_tty)
7034 die("Failed to open /dev/tty");
7035 cursed = !!newterm(NULL, opt_tty, opt_tty);
7036 }
7038 if (!cursed)
7039 die("Failed to initialize curses");
7041 nonl(); /* Disable conversion and detect newlines from input. */
7042 cbreak(); /* Take input chars one at a time, no wait for \n */
7043 noecho(); /* Don't echo input */
7044 leaveok(stdscr, FALSE);
7046 if (has_colors())
7047 init_colors();
7049 getmaxyx(stdscr, y, x);
7050 status_win = newwin(1, 0, y - 1, 0);
7051 if (!status_win)
7052 die("Failed to create status window");
7054 /* Enable keyboard mapping */
7055 keypad(status_win, TRUE);
7056 wbkgdset(status_win, get_line_attr(LINE_STATUS));
7058 TABSIZE = opt_tab_size;
7060 term = getenv("XTERM_VERSION") ? NULL : getenv("COLORTERM");
7061 if (term && !strcmp(term, "gnome-terminal")) {
7062 /* In the gnome-terminal-emulator, the message from
7063 * scrolling up one line when impossible followed by
7064 * scrolling down one line causes corruption of the
7065 * status line. This is fixed by calling wclear. */
7066 use_scroll_status_wclear = TRUE;
7067 use_scroll_redrawwin = FALSE;
7069 } else if (term && !strcmp(term, "xrvt-xpm")) {
7070 /* No problems with full optimizations in xrvt-(unicode)
7071 * and aterm. */
7072 use_scroll_status_wclear = use_scroll_redrawwin = FALSE;
7074 } else {
7075 /* When scrolling in (u)xterm the last line in the
7076 * scrolling direction will update slowly. */
7077 use_scroll_redrawwin = TRUE;
7078 use_scroll_status_wclear = FALSE;
7079 }
7080 }
7082 static int
7083 get_input(int prompt_position)
7084 {
7085 struct view *view;
7086 int i, key, cursor_y, cursor_x;
7087 bool loading = FALSE;
7089 if (prompt_position)
7090 input_mode = TRUE;
7092 while (TRUE) {
7093 foreach_view (view, i) {
7094 update_view(view);
7095 if (view_is_displayed(view) && view->has_scrolled &&
7096 use_scroll_redrawwin)
7097 redrawwin(view->win);
7098 view->has_scrolled = FALSE;
7099 if (view->pipe)
7100 loading = TRUE;
7101 }
7103 /* Update the cursor position. */
7104 if (prompt_position) {
7105 getbegyx(status_win, cursor_y, cursor_x);
7106 cursor_x = prompt_position;
7107 } else {
7108 view = display[current_view];
7109 getbegyx(view->win, cursor_y, cursor_x);
7110 cursor_x = view->width - 1;
7111 cursor_y += view->lineno - view->offset;
7112 }
7113 setsyx(cursor_y, cursor_x);
7115 /* Refresh, accept single keystroke of input */
7116 doupdate();
7117 nodelay(status_win, loading);
7118 key = wgetch(status_win);
7120 /* wgetch() with nodelay() enabled returns ERR when
7121 * there's no input. */
7122 if (key == ERR) {
7124 } else if (key == KEY_RESIZE) {
7125 int height, width;
7127 getmaxyx(stdscr, height, width);
7129 wresize(status_win, 1, width);
7130 mvwin(status_win, height - 1, 0);
7131 wnoutrefresh(status_win);
7132 resize_display();
7133 redraw_display(TRUE);
7135 } else {
7136 input_mode = FALSE;
7137 return key;
7138 }
7139 }
7140 }
7142 static char *
7143 prompt_input(const char *prompt, input_handler handler, void *data)
7144 {
7145 enum input_status status = INPUT_OK;
7146 static char buf[SIZEOF_STR];
7147 size_t pos = 0;
7149 buf[pos] = 0;
7151 while (status == INPUT_OK || status == INPUT_SKIP) {
7152 int key;
7154 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
7155 wclrtoeol(status_win);
7157 key = get_input(pos + 1);
7158 switch (key) {
7159 case KEY_RETURN:
7160 case KEY_ENTER:
7161 case '\n':
7162 status = pos ? INPUT_STOP : INPUT_CANCEL;
7163 break;
7165 case KEY_BACKSPACE:
7166 if (pos > 0)
7167 buf[--pos] = 0;
7168 else
7169 status = INPUT_CANCEL;
7170 break;
7172 case KEY_ESC:
7173 status = INPUT_CANCEL;
7174 break;
7176 default:
7177 if (pos >= sizeof(buf)) {
7178 report("Input string too long");
7179 return NULL;
7180 }
7182 status = handler(data, buf, key);
7183 if (status == INPUT_OK)
7184 buf[pos++] = (char) key;
7185 }
7186 }
7188 /* Clear the status window */
7189 status_empty = FALSE;
7190 report("");
7192 if (status == INPUT_CANCEL)
7193 return NULL;
7195 buf[pos++] = 0;
7197 return buf;
7198 }
7200 static enum input_status
7201 prompt_yesno_handler(void *data, char *buf, int c)
7202 {
7203 if (c == 'y' || c == 'Y')
7204 return INPUT_STOP;
7205 if (c == 'n' || c == 'N')
7206 return INPUT_CANCEL;
7207 return INPUT_SKIP;
7208 }
7210 static bool
7211 prompt_yesno(const char *prompt)
7212 {
7213 char prompt2[SIZEOF_STR];
7215 if (!string_format(prompt2, "%s [Yy/Nn]", prompt))
7216 return FALSE;
7218 return !!prompt_input(prompt2, prompt_yesno_handler, NULL);
7219 }
7221 static enum input_status
7222 read_prompt_handler(void *data, char *buf, int c)
7223 {
7224 return isprint(c) ? INPUT_OK : INPUT_SKIP;
7225 }
7227 static char *
7228 read_prompt(const char *prompt)
7229 {
7230 return prompt_input(prompt, read_prompt_handler, NULL);
7231 }
7233 static bool prompt_menu(const char *prompt, const struct menu_item *items, int *selected)
7234 {
7235 enum input_status status = INPUT_OK;
7236 int size = 0;
7238 while (items[size].text)
7239 size++;
7241 while (status == INPUT_OK) {
7242 const struct menu_item *item = &items[*selected];
7243 int key;
7244 int i;
7246 mvwprintw(status_win, 0, 0, "%s (%d of %d) ",
7247 prompt, *selected + 1, size);
7248 if (item->hotkey)
7249 wprintw(status_win, "[%c] ", (char) item->hotkey);
7250 wprintw(status_win, "%s", item->text);
7251 wclrtoeol(status_win);
7253 key = get_input(COLS - 1);
7254 switch (key) {
7255 case KEY_RETURN:
7256 case KEY_ENTER:
7257 case '\n':
7258 status = INPUT_STOP;
7259 break;
7261 case KEY_LEFT:
7262 case KEY_UP:
7263 *selected = *selected - 1;
7264 if (*selected < 0)
7265 *selected = size - 1;
7266 break;
7268 case KEY_RIGHT:
7269 case KEY_DOWN:
7270 *selected = (*selected + 1) % size;
7271 break;
7273 case KEY_ESC:
7274 status = INPUT_CANCEL;
7275 break;
7277 default:
7278 for (i = 0; items[i].text; i++)
7279 if (items[i].hotkey == key) {
7280 *selected = i;
7281 status = INPUT_STOP;
7282 break;
7283 }
7284 }
7285 }
7287 /* Clear the status window */
7288 status_empty = FALSE;
7289 report("");
7291 return status != INPUT_CANCEL;
7292 }
7294 /*
7295 * Repository properties
7296 */
7298 static struct ref **refs = NULL;
7299 static size_t refs_size = 0;
7300 static struct ref *refs_head = NULL;
7302 static struct ref_list **ref_lists = NULL;
7303 static size_t ref_lists_size = 0;
7305 DEFINE_ALLOCATOR(realloc_refs, struct ref *, 256)
7306 DEFINE_ALLOCATOR(realloc_refs_list, struct ref *, 8)
7307 DEFINE_ALLOCATOR(realloc_ref_lists, struct ref_list *, 8)
7309 static int
7310 compare_refs(const void *ref1_, const void *ref2_)
7311 {
7312 const struct ref *ref1 = *(const struct ref **)ref1_;
7313 const struct ref *ref2 = *(const struct ref **)ref2_;
7315 if (ref1->tag != ref2->tag)
7316 return ref2->tag - ref1->tag;
7317 if (ref1->ltag != ref2->ltag)
7318 return ref2->ltag - ref2->ltag;
7319 if (ref1->head != ref2->head)
7320 return ref2->head - ref1->head;
7321 if (ref1->tracked != ref2->tracked)
7322 return ref2->tracked - ref1->tracked;
7323 if (ref1->remote != ref2->remote)
7324 return ref2->remote - ref1->remote;
7325 return strcmp(ref1->name, ref2->name);
7326 }
7328 static void
7329 foreach_ref(bool (*visitor)(void *data, const struct ref *ref), void *data)
7330 {
7331 size_t i;
7333 for (i = 0; i < refs_size; i++)
7334 if (!visitor(data, refs[i]))
7335 break;
7336 }
7338 static struct ref *
7339 get_ref_head()
7340 {
7341 return refs_head;
7342 }
7344 static struct ref_list *
7345 get_ref_list(const char *id)
7346 {
7347 struct ref_list *list;
7348 size_t i;
7350 for (i = 0; i < ref_lists_size; i++)
7351 if (!strcmp(id, ref_lists[i]->id))
7352 return ref_lists[i];
7354 if (!realloc_ref_lists(&ref_lists, ref_lists_size, 1))
7355 return NULL;
7356 list = calloc(1, sizeof(*list));
7357 if (!list)
7358 return NULL;
7360 for (i = 0; i < refs_size; i++) {
7361 if (!strcmp(id, refs[i]->id) &&
7362 realloc_refs_list(&list->refs, list->size, 1))
7363 list->refs[list->size++] = refs[i];
7364 }
7366 if (!list->refs) {
7367 free(list);
7368 return NULL;
7369 }
7371 qsort(list->refs, list->size, sizeof(*list->refs), compare_refs);
7372 ref_lists[ref_lists_size++] = list;
7373 return list;
7374 }
7376 static int
7377 read_ref(char *id, size_t idlen, char *name, size_t namelen)
7378 {
7379 struct ref *ref = NULL;
7380 bool tag = FALSE;
7381 bool ltag = FALSE;
7382 bool remote = FALSE;
7383 bool tracked = FALSE;
7384 bool head = FALSE;
7385 int from = 0, to = refs_size - 1;
7387 if (!prefixcmp(name, "refs/tags/")) {
7388 if (!suffixcmp(name, namelen, "^{}")) {
7389 namelen -= 3;
7390 name[namelen] = 0;
7391 } else {
7392 ltag = TRUE;
7393 }
7395 tag = TRUE;
7396 namelen -= STRING_SIZE("refs/tags/");
7397 name += STRING_SIZE("refs/tags/");
7399 } else if (!prefixcmp(name, "refs/remotes/")) {
7400 remote = TRUE;
7401 namelen -= STRING_SIZE("refs/remotes/");
7402 name += STRING_SIZE("refs/remotes/");
7403 tracked = !strcmp(opt_remote, name);
7405 } else if (!prefixcmp(name, "refs/heads/")) {
7406 namelen -= STRING_SIZE("refs/heads/");
7407 name += STRING_SIZE("refs/heads/");
7408 if (!strncmp(opt_head, name, namelen))
7409 return OK;
7411 } else if (!strcmp(name, "HEAD")) {
7412 head = TRUE;
7413 if (*opt_head) {
7414 namelen = strlen(opt_head);
7415 name = opt_head;
7416 }
7417 }
7419 /* If we are reloading or it's an annotated tag, replace the
7420 * previous SHA1 with the resolved commit id; relies on the fact
7421 * git-ls-remote lists the commit id of an annotated tag right
7422 * before the commit id it points to. */
7423 while (from <= to) {
7424 size_t pos = (to + from) / 2;
7425 int cmp = strcmp(name, refs[pos]->name);
7427 if (!cmp) {
7428 ref = refs[pos];
7429 break;
7430 }
7432 if (cmp < 0)
7433 to = pos - 1;
7434 else
7435 from = pos + 1;
7436 }
7438 if (!ref) {
7439 if (!realloc_refs(&refs, refs_size, 1))
7440 return ERR;
7441 ref = calloc(1, sizeof(*ref) + namelen);
7442 if (!ref)
7443 return ERR;
7444 memmove(refs + from + 1, refs + from,
7445 (refs_size - from) * sizeof(*refs));
7446 refs[from] = ref;
7447 strncpy(ref->name, name, namelen);
7448 refs_size++;
7449 }
7451 ref->head = head;
7452 ref->tag = tag;
7453 ref->ltag = ltag;
7454 ref->remote = remote;
7455 ref->tracked = tracked;
7456 string_copy_rev(ref->id, id);
7458 if (head)
7459 refs_head = ref;
7460 return OK;
7461 }
7463 static int
7464 load_refs(void)
7465 {
7466 const char *head_argv[] = {
7467 "git", "symbolic-ref", "HEAD", NULL
7468 };
7469 static const char *ls_remote_argv[SIZEOF_ARG] = {
7470 "git", "ls-remote", opt_git_dir, NULL
7471 };
7472 static bool init = FALSE;
7473 size_t i;
7475 if (!init) {
7476 if (!argv_from_env(ls_remote_argv, "TIG_LS_REMOTE"))
7477 die("TIG_LS_REMOTE contains too many arguments");
7478 init = TRUE;
7479 }
7481 if (!*opt_git_dir)
7482 return OK;
7484 if (io_run_buf(head_argv, opt_head, sizeof(opt_head)) &&
7485 !prefixcmp(opt_head, "refs/heads/")) {
7486 char *offset = opt_head + STRING_SIZE("refs/heads/");
7488 memmove(opt_head, offset, strlen(offset) + 1);
7489 }
7491 refs_head = NULL;
7492 for (i = 0; i < refs_size; i++)
7493 refs[i]->id[0] = 0;
7495 if (io_run_load(ls_remote_argv, "\t", read_ref) == ERR)
7496 return ERR;
7498 /* Update the ref lists to reflect changes. */
7499 for (i = 0; i < ref_lists_size; i++) {
7500 struct ref_list *list = ref_lists[i];
7501 size_t old, new;
7503 for (old = new = 0; old < list->size; old++)
7504 if (!strcmp(list->id, list->refs[old]->id))
7505 list->refs[new++] = list->refs[old];
7506 list->size = new;
7507 }
7509 return OK;
7510 }
7512 static void
7513 set_remote_branch(const char *name, const char *value, size_t valuelen)
7514 {
7515 if (!strcmp(name, ".remote")) {
7516 string_ncopy(opt_remote, value, valuelen);
7518 } else if (*opt_remote && !strcmp(name, ".merge")) {
7519 size_t from = strlen(opt_remote);
7521 if (!prefixcmp(value, "refs/heads/"))
7522 value += STRING_SIZE("refs/heads/");
7524 if (!string_format_from(opt_remote, &from, "/%s", value))
7525 opt_remote[0] = 0;
7526 }
7527 }
7529 static void
7530 set_repo_config_option(char *name, char *value, int (*cmd)(int, const char **))
7531 {
7532 const char *argv[SIZEOF_ARG] = { name, "=" };
7533 int argc = 1 + (cmd == option_set_command);
7534 int error = ERR;
7536 if (!argv_from_string(argv, &argc, value))
7537 config_msg = "Too many option arguments";
7538 else
7539 error = cmd(argc, argv);
7541 if (error == ERR)
7542 warn("Option 'tig.%s': %s", name, config_msg);
7543 }
7545 static bool
7546 set_environment_variable(const char *name, const char *value)
7547 {
7548 size_t len = strlen(name) + 1 + strlen(value) + 1;
7549 char *env = malloc(len);
7551 if (env &&
7552 string_nformat(env, len, NULL, "%s=%s", name, value) &&
7553 putenv(env) == 0)
7554 return TRUE;
7555 free(env);
7556 return FALSE;
7557 }
7559 static void
7560 set_work_tree(const char *value)
7561 {
7562 char cwd[SIZEOF_STR];
7564 if (!getcwd(cwd, sizeof(cwd)))
7565 die("Failed to get cwd path: %s", strerror(errno));
7566 if (chdir(opt_git_dir) < 0)
7567 die("Failed to chdir(%s): %s", strerror(errno));
7568 if (!getcwd(opt_git_dir, sizeof(opt_git_dir)))
7569 die("Failed to get git path: %s", strerror(errno));
7570 if (chdir(cwd) < 0)
7571 die("Failed to chdir(%s): %s", cwd, strerror(errno));
7572 if (chdir(value) < 0)
7573 die("Failed to chdir(%s): %s", value, strerror(errno));
7574 if (!getcwd(cwd, sizeof(cwd)))
7575 die("Failed to get cwd path: %s", strerror(errno));
7576 if (!set_environment_variable("GIT_WORK_TREE", cwd))
7577 die("Failed to set GIT_WORK_TREE to '%s'", cwd);
7578 if (!set_environment_variable("GIT_DIR", opt_git_dir))
7579 die("Failed to set GIT_DIR to '%s'", opt_git_dir);
7580 opt_is_inside_work_tree = TRUE;
7581 }
7583 static int
7584 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
7585 {
7586 if (!strcmp(name, "i18n.commitencoding"))
7587 string_ncopy(opt_encoding, value, valuelen);
7589 else if (!strcmp(name, "core.editor"))
7590 string_ncopy(opt_editor, value, valuelen);
7592 else if (!strcmp(name, "core.worktree"))
7593 set_work_tree(value);
7595 else if (!prefixcmp(name, "tig.color."))
7596 set_repo_config_option(name + 10, value, option_color_command);
7598 else if (!prefixcmp(name, "tig.bind."))
7599 set_repo_config_option(name + 9, value, option_bind_command);
7601 else if (!prefixcmp(name, "tig."))
7602 set_repo_config_option(name + 4, value, option_set_command);
7604 else if (*opt_head && !prefixcmp(name, "branch.") &&
7605 !strncmp(name + 7, opt_head, strlen(opt_head)))
7606 set_remote_branch(name + 7 + strlen(opt_head), value, valuelen);
7608 return OK;
7609 }
7611 static int
7612 load_git_config(void)
7613 {
7614 const char *config_list_argv[] = { "git", "config", "--list", NULL };
7616 return io_run_load(config_list_argv, "=", read_repo_config_option);
7617 }
7619 static int
7620 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
7621 {
7622 if (!opt_git_dir[0]) {
7623 string_ncopy(opt_git_dir, name, namelen);
7625 } else if (opt_is_inside_work_tree == -1) {
7626 /* This can be 3 different values depending on the
7627 * version of git being used. If git-rev-parse does not
7628 * understand --is-inside-work-tree it will simply echo
7629 * the option else either "true" or "false" is printed.
7630 * Default to true for the unknown case. */
7631 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
7633 } else if (*name == '.') {
7634 string_ncopy(opt_cdup, name, namelen);
7636 } else {
7637 string_ncopy(opt_prefix, name, namelen);
7638 }
7640 return OK;
7641 }
7643 static int
7644 load_repo_info(void)
7645 {
7646 const char *rev_parse_argv[] = {
7647 "git", "rev-parse", "--git-dir", "--is-inside-work-tree",
7648 "--show-cdup", "--show-prefix", NULL
7649 };
7651 return io_run_load(rev_parse_argv, "=", read_repo_info);
7652 }
7655 /*
7656 * Main
7657 */
7659 static const char usage[] =
7660 "tig " TIG_VERSION " (" __DATE__ ")\n"
7661 "\n"
7662 "Usage: tig [options] [revs] [--] [paths]\n"
7663 " or: tig show [options] [revs] [--] [paths]\n"
7664 " or: tig blame [rev] path\n"
7665 " or: tig status\n"
7666 " or: tig < [git command output]\n"
7667 "\n"
7668 "Options:\n"
7669 " -v, --version Show version and exit\n"
7670 " -h, --help Show help message and exit";
7672 static void __NORETURN
7673 quit(int sig)
7674 {
7675 /* XXX: Restore tty modes and let the OS cleanup the rest! */
7676 if (cursed)
7677 endwin();
7678 exit(0);
7679 }
7681 static void __NORETURN
7682 die(const char *err, ...)
7683 {
7684 va_list args;
7686 endwin();
7688 va_start(args, err);
7689 fputs("tig: ", stderr);
7690 vfprintf(stderr, err, args);
7691 fputs("\n", stderr);
7692 va_end(args);
7694 exit(1);
7695 }
7697 static void
7698 warn(const char *msg, ...)
7699 {
7700 va_list args;
7702 va_start(args, msg);
7703 fputs("tig warning: ", stderr);
7704 vfprintf(stderr, msg, args);
7705 fputs("\n", stderr);
7706 va_end(args);
7707 }
7709 static enum request
7710 parse_options(int argc, const char *argv[])
7711 {
7712 enum request request = REQ_VIEW_MAIN;
7713 const char *subcommand;
7714 bool seen_dashdash = FALSE;
7715 /* XXX: This is vulnerable to the user overriding options
7716 * required for the main view parser. */
7717 const char *custom_argv[SIZEOF_ARG] = {
7718 "git", "log", "--no-color", "--pretty=raw", "--parents",
7719 "--topo-order", NULL
7720 };
7721 int i, j = 6;
7723 if (!isatty(STDIN_FILENO)) {
7724 io_open(&VIEW(REQ_VIEW_PAGER)->io, "");
7725 return REQ_VIEW_PAGER;
7726 }
7728 if (argc <= 1)
7729 return REQ_NONE;
7731 subcommand = argv[1];
7732 if (!strcmp(subcommand, "status")) {
7733 if (argc > 2)
7734 warn("ignoring arguments after `%s'", subcommand);
7735 return REQ_VIEW_STATUS;
7737 } else if (!strcmp(subcommand, "blame")) {
7738 if (argc <= 2 || argc > 4)
7739 die("invalid number of options to blame\n\n%s", usage);
7741 i = 2;
7742 if (argc == 4) {
7743 string_ncopy(opt_ref, argv[i], strlen(argv[i]));
7744 i++;
7745 }
7747 string_ncopy(opt_file, argv[i], strlen(argv[i]));
7748 return REQ_VIEW_BLAME;
7750 } else if (!strcmp(subcommand, "show")) {
7751 request = REQ_VIEW_DIFF;
7753 } else {
7754 subcommand = NULL;
7755 }
7757 if (subcommand) {
7758 custom_argv[1] = subcommand;
7759 j = 2;
7760 }
7762 for (i = 1 + !!subcommand; i < argc; i++) {
7763 const char *opt = argv[i];
7765 if (seen_dashdash || !strcmp(opt, "--")) {
7766 seen_dashdash = TRUE;
7768 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
7769 printf("tig version %s\n", TIG_VERSION);
7770 quit(0);
7772 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
7773 printf("%s\n", usage);
7774 quit(0);
7775 }
7777 custom_argv[j++] = opt;
7778 if (j >= ARRAY_SIZE(custom_argv))
7779 die("command too long");
7780 }
7782 if (!prepare_update(VIEW(request), custom_argv, NULL))
7783 die("Failed to format arguments");
7785 return request;
7786 }
7788 int
7789 main(int argc, const char *argv[])
7790 {
7791 const char *codeset = "UTF-8";
7792 enum request request = parse_options(argc, argv);
7793 struct view *view;
7794 size_t i;
7796 signal(SIGINT, quit);
7797 signal(SIGPIPE, SIG_IGN);
7799 if (setlocale(LC_ALL, "")) {
7800 codeset = nl_langinfo(CODESET);
7801 }
7803 if (load_repo_info() == ERR)
7804 die("Failed to load repo info.");
7806 if (load_options() == ERR)
7807 die("Failed to load user config.");
7809 if (load_git_config() == ERR)
7810 die("Failed to load repo config.");
7812 /* Require a git repository unless when running in pager mode. */
7813 if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
7814 die("Not a git repository");
7816 if (*opt_encoding && strcmp(codeset, "UTF-8")) {
7817 opt_iconv_in = iconv_open("UTF-8", opt_encoding);
7818 if (opt_iconv_in == ICONV_NONE)
7819 die("Failed to initialize character set conversion");
7820 }
7822 if (codeset && strcmp(codeset, "UTF-8")) {
7823 opt_iconv_out = iconv_open(codeset, "UTF-8");
7824 if (opt_iconv_out == ICONV_NONE)
7825 die("Failed to initialize character set conversion");
7826 }
7828 if (load_refs() == ERR)
7829 die("Failed to load refs.");
7831 foreach_view (view, i)
7832 if (!argv_from_env(view->ops->argv, view->cmd_env))
7833 die("Too many arguments in the `%s` environment variable",
7834 view->cmd_env);
7836 init_display();
7838 if (request != REQ_NONE)
7839 open_view(NULL, request, OPEN_PREPARED);
7840 request = request == REQ_NONE ? REQ_VIEW_MAIN : REQ_NONE;
7842 while (view_driver(display[current_view], request)) {
7843 int key = get_input(0);
7845 view = display[current_view];
7846 request = get_keybinding(view->keymap, key);
7848 /* Some low-level request handling. This keeps access to
7849 * status_win restricted. */
7850 switch (request) {
7851 case REQ_NONE:
7852 report("Unknown key, press %s for help",
7853 get_key(view->keymap, REQ_VIEW_HELP));
7854 break;
7855 case REQ_PROMPT:
7856 {
7857 char *cmd = read_prompt(":");
7859 if (cmd && isdigit(*cmd)) {
7860 int lineno = view->lineno + 1;
7862 if (parse_int(&lineno, cmd, 1, view->lines + 1) == OK) {
7863 select_view_line(view, lineno - 1);
7864 report("");
7865 } else {
7866 report("Unable to parse '%s' as a line number", cmd);
7867 }
7869 } else if (cmd) {
7870 struct view *next = VIEW(REQ_VIEW_PAGER);
7871 const char *argv[SIZEOF_ARG] = { "git" };
7872 int argc = 1;
7874 /* When running random commands, initially show the
7875 * command in the title. However, it maybe later be
7876 * overwritten if a commit line is selected. */
7877 string_ncopy(next->ref, cmd, strlen(cmd));
7879 if (!argv_from_string(argv, &argc, cmd)) {
7880 report("Too many arguments");
7881 } else if (!prepare_update(next, argv, NULL)) {
7882 report("Failed to format command");
7883 } else {
7884 open_view(view, REQ_VIEW_PAGER, OPEN_PREPARED);
7885 }
7886 }
7888 request = REQ_NONE;
7889 break;
7890 }
7891 case REQ_SEARCH:
7892 case REQ_SEARCH_BACK:
7893 {
7894 const char *prompt = request == REQ_SEARCH ? "/" : "?";
7895 char *search = read_prompt(prompt);
7897 if (search)
7898 string_ncopy(opt_search, search, strlen(search));
7899 else if (*opt_search)
7900 request = request == REQ_SEARCH ?
7901 REQ_FIND_NEXT :
7902 REQ_FIND_PREV;
7903 else
7904 request = REQ_NONE;
7905 break;
7906 }
7907 default:
7908 break;
7909 }
7910 }
7912 quit(0);
7914 return 0;
7915 }