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