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