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