manual: remove section on porcelains
[tig] / tig.c
1 /* Copyright (c) 2006-2007 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/stat.h>
36 #include <unistd.h>
37 #include <time.h>
38
39 #include <regex.h>
40
41 #include <locale.h>
42 #include <langinfo.h>
43 #include <iconv.h>
44
45 #include <curses.h>
46
47 #if __GNUC__ >= 3
48 #define __NORETURN __attribute__((__noreturn__))
49 #else
50 #define __NORETURN
51 #endif
52
53 static void __NORETURN die(const char *err, ...);
54 static void report(const char *msg, ...);
55 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, size_t, char *, size_t));
56 static void set_nonblocking_input(bool loading);
57 static size_t utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed);
58
59 #define ABS(x) ((x) >= 0 ? (x) : -(x))
60 #define MIN(x, y) ((x) < (y) ? (x) : (y))
61
62 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
63 #define STRING_SIZE(x) (sizeof(x) - 1)
64
65 #define SIZEOF_STR 1024 /* Default string size. */
66 #define SIZEOF_REF 256 /* Size of symbolic or SHA1 ID. */
67 #define SIZEOF_REV 41 /* Holds a SHA-1 and an ending NUL */
68
69 /* Revision graph */
70
71 #define REVGRAPH_INIT 'I'
72 #define REVGRAPH_MERGE 'M'
73 #define REVGRAPH_BRANCH '+'
74 #define REVGRAPH_COMMIT '*'
75 #define REVGRAPH_LINE '|'
76
77 #define SIZEOF_REVGRAPH 19 /* Size of revision ancestry graphics. */
78
79 /* This color name can be used to refer to the default term colors. */
80 #define COLOR_DEFAULT (-1)
81
82 #define ICONV_NONE ((iconv_t) -1)
83 #ifndef ICONV_CONST
84 #define ICONV_CONST /* nothing */
85 #endif
86
87 /* The format and size of the date column in the main view. */
88 #define DATE_FORMAT "%Y-%m-%d %H:%M"
89 #define DATE_COLS STRING_SIZE("2006-04-29 14:21 ")
90
91 #define AUTHOR_COLS 20
92
93 /* The default interval between line numbers. */
94 #define NUMBER_INTERVAL 1
95
96 #define TABSIZE 8
97
98 #define SCALE_SPLIT_VIEW(height) ((height) * 2 / 3)
99
100 #ifndef GIT_CONFIG
101 #define GIT_CONFIG "git config"
102 #endif
103
104 #define TIG_LS_REMOTE \
105 "git ls-remote $(git rev-parse --git-dir) 2>/dev/null"
106
107 #define TIG_DIFF_CMD \
108 "git show --no-color --root --patch-with-stat --find-copies-harder -B -C %s 2>/dev/null"
109
110 #define TIG_LOG_CMD \
111 "git log --no-color --cc --stat -n100 %s 2>/dev/null"
112
113 #define TIG_MAIN_CMD \
114 "git log --no-color --topo-order --pretty=raw %s 2>/dev/null"
115
116 #define TIG_TREE_CMD \
117 "git ls-tree %s %s"
118
119 #define TIG_BLOB_CMD \
120 "git cat-file blob %s"
121
122 /* XXX: Needs to be defined to the empty string. */
123 #define TIG_HELP_CMD ""
124 #define TIG_PAGER_CMD ""
125 #define TIG_STATUS_CMD ""
126 #define TIG_STAGE_CMD ""
127
128 /* Some ascii-shorthands fitted into the ncurses namespace. */
129 #define KEY_TAB '\t'
130 #define KEY_RETURN '\r'
131 #define KEY_ESC 27
132
133
134 struct ref {
135 char *name; /* Ref name; tag or head names are shortened. */
136 char id[SIZEOF_REV]; /* Commit SHA1 ID */
137 unsigned int tag:1; /* Is it a tag? */
138 unsigned int remote:1; /* Is it a remote ref? */
139 unsigned int next:1; /* For ref lists: are there more refs? */
140 };
141
142 static struct ref **get_refs(char *id);
143
144 struct int_map {
145 const char *name;
146 int namelen;
147 int value;
148 };
149
150 static int
151 set_from_int_map(struct int_map *map, size_t map_size,
152 int *value, const char *name, int namelen)
153 {
154
155 int i;
156
157 for (i = 0; i < map_size; i++)
158 if (namelen == map[i].namelen &&
159 !strncasecmp(name, map[i].name, namelen)) {
160 *value = map[i].value;
161 return OK;
162 }
163
164 return ERR;
165 }
166
167
168 /*
169 * String helpers
170 */
171
172 static inline void
173 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
174 {
175 if (srclen > dstlen - 1)
176 srclen = dstlen - 1;
177
178 strncpy(dst, src, srclen);
179 dst[srclen] = 0;
180 }
181
182 /* Shorthands for safely copying into a fixed buffer. */
183
184 #define string_copy(dst, src) \
185 string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
186
187 #define string_ncopy(dst, src, srclen) \
188 string_ncopy_do(dst, sizeof(dst), src, srclen)
189
190 #define string_copy_rev(dst, src) \
191 string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
192
193 #define string_add(dst, from, src) \
194 string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
195
196 static char *
197 chomp_string(char *name)
198 {
199 int namelen;
200
201 while (isspace(*name))
202 name++;
203
204 namelen = strlen(name) - 1;
205 while (namelen > 0 && isspace(name[namelen]))
206 name[namelen--] = 0;
207
208 return name;
209 }
210
211 static bool
212 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
213 {
214 va_list args;
215 size_t pos = bufpos ? *bufpos : 0;
216
217 va_start(args, fmt);
218 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
219 va_end(args);
220
221 if (bufpos)
222 *bufpos = pos;
223
224 return pos >= bufsize ? FALSE : TRUE;
225 }
226
227 #define string_format(buf, fmt, args...) \
228 string_nformat(buf, sizeof(buf), NULL, fmt, args)
229
230 #define string_format_from(buf, from, fmt, args...) \
231 string_nformat(buf, sizeof(buf), from, fmt, args)
232
233 static int
234 string_enum_compare(const char *str1, const char *str2, int len)
235 {
236 size_t i;
237
238 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
239
240 /* Diff-Header == DIFF_HEADER */
241 for (i = 0; i < len; i++) {
242 if (toupper(str1[i]) == toupper(str2[i]))
243 continue;
244
245 if (string_enum_sep(str1[i]) &&
246 string_enum_sep(str2[i]))
247 continue;
248
249 return str1[i] - str2[i];
250 }
251
252 return 0;
253 }
254
255 /* Shell quoting
256 *
257 * NOTE: The following is a slightly modified copy of the git project's shell
258 * quoting routines found in the quote.c file.
259 *
260 * Help to copy the thing properly quoted for the shell safety. any single
261 * quote is replaced with '\'', any exclamation point is replaced with '\!',
262 * and the whole thing is enclosed in a
263 *
264 * E.g.
265 * original sq_quote result
266 * name ==> name ==> 'name'
267 * a b ==> a b ==> 'a b'
268 * a'b ==> a'\''b ==> 'a'\''b'
269 * a!b ==> a'\!'b ==> 'a'\!'b'
270 */
271
272 static size_t
273 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
274 {
275 char c;
276
277 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
278
279 BUFPUT('\'');
280 while ((c = *src++)) {
281 if (c == '\'' || c == '!') {
282 BUFPUT('\'');
283 BUFPUT('\\');
284 BUFPUT(c);
285 BUFPUT('\'');
286 } else {
287 BUFPUT(c);
288 }
289 }
290 BUFPUT('\'');
291
292 if (bufsize < SIZEOF_STR)
293 buf[bufsize] = 0;
294
295 return bufsize;
296 }
297
298
299 /*
300 * User requests
301 */
302
303 #define REQ_INFO \
304 /* XXX: Keep the view request first and in sync with views[]. */ \
305 REQ_GROUP("View switching") \
306 REQ_(VIEW_MAIN, "Show main view"), \
307 REQ_(VIEW_DIFF, "Show diff view"), \
308 REQ_(VIEW_LOG, "Show log view"), \
309 REQ_(VIEW_TREE, "Show tree view"), \
310 REQ_(VIEW_BLOB, "Show blob view"), \
311 REQ_(VIEW_HELP, "Show help page"), \
312 REQ_(VIEW_PAGER, "Show pager view"), \
313 REQ_(VIEW_STATUS, "Show status view"), \
314 REQ_(VIEW_STAGE, "Show stage view"), \
315 \
316 REQ_GROUP("View manipulation") \
317 REQ_(ENTER, "Enter current line and scroll"), \
318 REQ_(NEXT, "Move to next"), \
319 REQ_(PREVIOUS, "Move to previous"), \
320 REQ_(VIEW_NEXT, "Move focus to next view"), \
321 REQ_(REFRESH, "Reload and refresh"), \
322 REQ_(VIEW_CLOSE, "Close the current view"), \
323 REQ_(QUIT, "Close all views and quit"), \
324 \
325 REQ_GROUP("Cursor navigation") \
326 REQ_(MOVE_UP, "Move cursor one line up"), \
327 REQ_(MOVE_DOWN, "Move cursor one line down"), \
328 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
329 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
330 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
331 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
332 \
333 REQ_GROUP("Scrolling") \
334 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
335 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
336 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
337 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
338 \
339 REQ_GROUP("Searching") \
340 REQ_(SEARCH, "Search the view"), \
341 REQ_(SEARCH_BACK, "Search backwards in the view"), \
342 REQ_(FIND_NEXT, "Find next search match"), \
343 REQ_(FIND_PREV, "Find previous search match"), \
344 \
345 REQ_GROUP("Misc") \
346 REQ_(PROMPT, "Bring up the prompt"), \
347 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
348 REQ_(SCREEN_RESIZE, "Resize the screen"), \
349 REQ_(SHOW_VERSION, "Show version information"), \
350 REQ_(STOP_LOADING, "Stop all loading views"), \
351 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
352 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization"), \
353 REQ_(STATUS_UPDATE, "Update file status"), \
354 REQ_(STATUS_MERGE, "Merge file using external tool"), \
355 REQ_(EDIT, "Open in editor"), \
356 REQ_(NONE, "Do nothing")
357
358
359 /* User action requests. */
360 enum request {
361 #define REQ_GROUP(help)
362 #define REQ_(req, help) REQ_##req
363
364 /* Offset all requests to avoid conflicts with ncurses getch values. */
365 REQ_OFFSET = KEY_MAX + 1,
366 REQ_INFO
367
368 #undef REQ_GROUP
369 #undef REQ_
370 };
371
372 struct request_info {
373 enum request request;
374 char *name;
375 int namelen;
376 char *help;
377 };
378
379 static struct request_info req_info[] = {
380 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
381 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
382 REQ_INFO
383 #undef REQ_GROUP
384 #undef REQ_
385 };
386
387 static enum request
388 get_request(const char *name)
389 {
390 int namelen = strlen(name);
391 int i;
392
393 for (i = 0; i < ARRAY_SIZE(req_info); i++)
394 if (req_info[i].namelen == namelen &&
395 !string_enum_compare(req_info[i].name, name, namelen))
396 return req_info[i].request;
397
398 return REQ_NONE;
399 }
400
401
402 /*
403 * Options
404 */
405
406 static const char usage[] =
407 "tig " TIG_VERSION " (" __DATE__ ")\n"
408 "\n"
409 "Usage: tig [options]\n"
410 " or: tig [options] [--] [git log options]\n"
411 " or: tig [options] log [git log options]\n"
412 " or: tig [options] diff [git diff options]\n"
413 " or: tig [options] show [git show options]\n"
414 " or: tig [options] < [git command output]\n"
415 "\n"
416 "Options:\n"
417 " -l Start up in log view\n"
418 " -d Start up in diff view\n"
419 " -S Start up in status view\n"
420 " -n[I], --line-number[=I] Show line numbers with given interval\n"
421 " -b[N], --tab-size[=N] Set number of spaces for tab expansion\n"
422 " -- Mark end of tig options\n"
423 " -v, --version Show version and exit\n"
424 " -h, --help Show help message and exit\n";
425
426 /* Option and state variables. */
427 static bool opt_line_number = FALSE;
428 static bool opt_rev_graph = FALSE;
429 static int opt_num_interval = NUMBER_INTERVAL;
430 static int opt_tab_size = TABSIZE;
431 static enum request opt_request = REQ_VIEW_MAIN;
432 static char opt_cmd[SIZEOF_STR] = "";
433 static char opt_path[SIZEOF_STR] = "";
434 static FILE *opt_pipe = NULL;
435 static char opt_encoding[20] = "UTF-8";
436 static bool opt_utf8 = TRUE;
437 static char opt_codeset[20] = "UTF-8";
438 static iconv_t opt_iconv = ICONV_NONE;
439 static char opt_search[SIZEOF_STR] = "";
440 static char opt_cdup[SIZEOF_STR] = "";
441 static char opt_git_dir[SIZEOF_STR] = "";
442 static char opt_is_inside_work_tree = -1; /* set to TRUE or FALSE */
443 static char opt_editor[SIZEOF_STR] = "";
444
445 enum option_type {
446 OPT_NONE,
447 OPT_INT,
448 };
449
450 static bool
451 check_option(char *opt, char short_name, char *name, enum option_type type, ...)
452 {
453 va_list args;
454 char *value = "";
455 int *number;
456
457 if (opt[0] != '-')
458 return FALSE;
459
460 if (opt[1] == '-') {
461 int namelen = strlen(name);
462
463 opt += 2;
464
465 if (strncmp(opt, name, namelen))
466 return FALSE;
467
468 if (opt[namelen] == '=')
469 value = opt + namelen + 1;
470
471 } else {
472 if (!short_name || opt[1] != short_name)
473 return FALSE;
474 value = opt + 2;
475 }
476
477 va_start(args, type);
478 if (type == OPT_INT) {
479 number = va_arg(args, int *);
480 if (isdigit(*value))
481 *number = atoi(value);
482 }
483 va_end(args);
484
485 return TRUE;
486 }
487
488 /* Returns the index of log or diff command or -1 to exit. */
489 static bool
490 parse_options(int argc, char *argv[])
491 {
492 int i;
493
494 for (i = 1; i < argc; i++) {
495 char *opt = argv[i];
496
497 if (!strcmp(opt, "log") ||
498 !strcmp(opt, "diff") ||
499 !strcmp(opt, "show")) {
500 opt_request = opt[0] == 'l'
501 ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
502 break;
503 }
504
505 if (opt[0] && opt[0] != '-')
506 break;
507
508 if (!strcmp(opt, "--")) {
509 i++;
510 break;
511 }
512
513 if (check_option(opt, 'v', "version", OPT_NONE)) {
514 printf("tig version %s\n", TIG_VERSION);
515 return FALSE;
516 }
517
518 if (check_option(opt, 'h', "help", OPT_NONE)) {
519 printf(usage);
520 return FALSE;
521 }
522
523 if (!strcmp(opt, "-S")) {
524 opt_request = REQ_VIEW_STATUS;
525 continue;
526 }
527
528 if (!strcmp(opt, "-l")) {
529 opt_request = REQ_VIEW_LOG;
530 continue;
531 }
532
533 if (!strcmp(opt, "-d")) {
534 opt_request = REQ_VIEW_DIFF;
535 continue;
536 }
537
538 if (check_option(opt, 'n', "line-number", OPT_INT, &opt_num_interval)) {
539 opt_line_number = TRUE;
540 continue;
541 }
542
543 if (check_option(opt, 'b', "tab-size", OPT_INT, &opt_tab_size)) {
544 opt_tab_size = MIN(opt_tab_size, TABSIZE);
545 continue;
546 }
547
548 die("unknown option '%s'\n\n%s", opt, usage);
549 }
550
551 if (!isatty(STDIN_FILENO)) {
552 opt_request = REQ_VIEW_PAGER;
553 opt_pipe = stdin;
554
555 } else if (i < argc) {
556 size_t buf_size;
557
558 if (opt_request == REQ_VIEW_MAIN)
559 /* XXX: This is vulnerable to the user overriding
560 * options required for the main view parser. */
561 string_copy(opt_cmd, "git log --no-color --pretty=raw");
562 else
563 string_copy(opt_cmd, "git");
564 buf_size = strlen(opt_cmd);
565
566 while (buf_size < sizeof(opt_cmd) && i < argc) {
567 opt_cmd[buf_size++] = ' ';
568 buf_size = sq_quote(opt_cmd, buf_size, argv[i++]);
569 }
570
571 if (buf_size >= sizeof(opt_cmd))
572 die("command too long");
573
574 opt_cmd[buf_size] = 0;
575 }
576
577 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
578 opt_utf8 = FALSE;
579
580 return TRUE;
581 }
582
583
584 /*
585 * Line-oriented content detection.
586 */
587
588 #define LINE_INFO \
589 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
590 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
591 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
592 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
593 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
594 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
595 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
596 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
597 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
598 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
599 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
600 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
601 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
602 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
603 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
604 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
605 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
606 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
607 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
608 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
609 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
610 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
611 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
612 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
613 LINE(AUTHOR, "author ", COLOR_CYAN, COLOR_DEFAULT, 0), \
614 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
615 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
616 LINE(ACKED, " Acked-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
617 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
618 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
619 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
620 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
621 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
622 LINE(MAIN_DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
623 LINE(MAIN_AUTHOR, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
624 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
625 LINE(MAIN_DELIM, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
626 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
627 LINE(MAIN_REMOTE, "", COLOR_YELLOW, COLOR_DEFAULT, A_BOLD), \
628 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
629 LINE(TREE_DIR, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
630 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
631 LINE(STAT_SECTION, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
632 LINE(STAT_NONE, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
633 LINE(STAT_STAGED, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
634 LINE(STAT_UNSTAGED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
635 LINE(STAT_UNTRACKED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0)
636
637 enum line_type {
638 #define LINE(type, line, fg, bg, attr) \
639 LINE_##type
640 LINE_INFO
641 #undef LINE
642 };
643
644 struct line_info {
645 const char *name; /* Option name. */
646 int namelen; /* Size of option name. */
647 const char *line; /* The start of line to match. */
648 int linelen; /* Size of string to match. */
649 int fg, bg, attr; /* Color and text attributes for the lines. */
650 };
651
652 static struct line_info line_info[] = {
653 #define LINE(type, line, fg, bg, attr) \
654 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
655 LINE_INFO
656 #undef LINE
657 };
658
659 static enum line_type
660 get_line_type(char *line)
661 {
662 int linelen = strlen(line);
663 enum line_type type;
664
665 for (type = 0; type < ARRAY_SIZE(line_info); type++)
666 /* Case insensitive search matches Signed-off-by lines better. */
667 if (linelen >= line_info[type].linelen &&
668 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
669 return type;
670
671 return LINE_DEFAULT;
672 }
673
674 static inline int
675 get_line_attr(enum line_type type)
676 {
677 assert(type < ARRAY_SIZE(line_info));
678 return COLOR_PAIR(type) | line_info[type].attr;
679 }
680
681 static struct line_info *
682 get_line_info(char *name, int namelen)
683 {
684 enum line_type type;
685
686 for (type = 0; type < ARRAY_SIZE(line_info); type++)
687 if (namelen == line_info[type].namelen &&
688 !string_enum_compare(line_info[type].name, name, namelen))
689 return &line_info[type];
690
691 return NULL;
692 }
693
694 static void
695 init_colors(void)
696 {
697 int default_bg = COLOR_BLACK;
698 int default_fg = COLOR_WHITE;
699 enum line_type type;
700
701 start_color();
702
703 if (use_default_colors() != ERR) {
704 default_bg = -1;
705 default_fg = -1;
706 }
707
708 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
709 struct line_info *info = &line_info[type];
710 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
711 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
712
713 init_pair(type, fg, bg);
714 }
715 }
716
717 struct line {
718 enum line_type type;
719
720 /* State flags */
721 unsigned int selected:1;
722
723 void *data; /* User data */
724 };
725
726
727 /*
728 * Keys
729 */
730
731 struct keybinding {
732 int alias;
733 enum request request;
734 struct keybinding *next;
735 };
736
737 static struct keybinding default_keybindings[] = {
738 /* View switching */
739 { 'm', REQ_VIEW_MAIN },
740 { 'd', REQ_VIEW_DIFF },
741 { 'l', REQ_VIEW_LOG },
742 { 't', REQ_VIEW_TREE },
743 { 'f', REQ_VIEW_BLOB },
744 { 'p', REQ_VIEW_PAGER },
745 { 'h', REQ_VIEW_HELP },
746 { 'S', REQ_VIEW_STATUS },
747 { 'c', REQ_VIEW_STAGE },
748
749 /* View manipulation */
750 { 'q', REQ_VIEW_CLOSE },
751 { KEY_TAB, REQ_VIEW_NEXT },
752 { KEY_RETURN, REQ_ENTER },
753 { KEY_UP, REQ_PREVIOUS },
754 { KEY_DOWN, REQ_NEXT },
755 { 'R', REQ_REFRESH },
756
757 /* Cursor navigation */
758 { 'k', REQ_MOVE_UP },
759 { 'j', REQ_MOVE_DOWN },
760 { KEY_HOME, REQ_MOVE_FIRST_LINE },
761 { KEY_END, REQ_MOVE_LAST_LINE },
762 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
763 { ' ', REQ_MOVE_PAGE_DOWN },
764 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
765 { 'b', REQ_MOVE_PAGE_UP },
766 { '-', REQ_MOVE_PAGE_UP },
767
768 /* Scrolling */
769 { KEY_IC, REQ_SCROLL_LINE_UP },
770 { KEY_DC, REQ_SCROLL_LINE_DOWN },
771 { 'w', REQ_SCROLL_PAGE_UP },
772 { 's', REQ_SCROLL_PAGE_DOWN },
773
774 /* Searching */
775 { '/', REQ_SEARCH },
776 { '?', REQ_SEARCH_BACK },
777 { 'n', REQ_FIND_NEXT },
778 { 'N', REQ_FIND_PREV },
779
780 /* Misc */
781 { 'Q', REQ_QUIT },
782 { 'z', REQ_STOP_LOADING },
783 { 'v', REQ_SHOW_VERSION },
784 { 'r', REQ_SCREEN_REDRAW },
785 { '.', REQ_TOGGLE_LINENO },
786 { 'g', REQ_TOGGLE_REV_GRAPH },
787 { ':', REQ_PROMPT },
788 { 'u', REQ_STATUS_UPDATE },
789 { 'M', REQ_STATUS_MERGE },
790 { 'e', REQ_EDIT },
791
792 /* Using the ncurses SIGWINCH handler. */
793 { KEY_RESIZE, REQ_SCREEN_RESIZE },
794 };
795
796 #define KEYMAP_INFO \
797 KEYMAP_(GENERIC), \
798 KEYMAP_(MAIN), \
799 KEYMAP_(DIFF), \
800 KEYMAP_(LOG), \
801 KEYMAP_(TREE), \
802 KEYMAP_(BLOB), \
803 KEYMAP_(PAGER), \
804 KEYMAP_(HELP), \
805 KEYMAP_(STATUS), \
806 KEYMAP_(STAGE)
807
808 enum keymap {
809 #define KEYMAP_(name) KEYMAP_##name
810 KEYMAP_INFO
811 #undef KEYMAP_
812 };
813
814 static struct int_map keymap_table[] = {
815 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
816 KEYMAP_INFO
817 #undef KEYMAP_
818 };
819
820 #define set_keymap(map, name) \
821 set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
822
823 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
824
825 static void
826 add_keybinding(enum keymap keymap, enum request request, int key)
827 {
828 struct keybinding *keybinding;
829
830 keybinding = calloc(1, sizeof(*keybinding));
831 if (!keybinding)
832 die("Failed to allocate keybinding");
833
834 keybinding->alias = key;
835 keybinding->request = request;
836 keybinding->next = keybindings[keymap];
837 keybindings[keymap] = keybinding;
838 }
839
840 /* Looks for a key binding first in the given map, then in the generic map, and
841 * lastly in the default keybindings. */
842 static enum request
843 get_keybinding(enum keymap keymap, int key)
844 {
845 struct keybinding *kbd;
846 int i;
847
848 for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
849 if (kbd->alias == key)
850 return kbd->request;
851
852 for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
853 if (kbd->alias == key)
854 return kbd->request;
855
856 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
857 if (default_keybindings[i].alias == key)
858 return default_keybindings[i].request;
859
860 return (enum request) key;
861 }
862
863
864 struct key {
865 char *name;
866 int value;
867 };
868
869 static struct key key_table[] = {
870 { "Enter", KEY_RETURN },
871 { "Space", ' ' },
872 { "Backspace", KEY_BACKSPACE },
873 { "Tab", KEY_TAB },
874 { "Escape", KEY_ESC },
875 { "Left", KEY_LEFT },
876 { "Right", KEY_RIGHT },
877 { "Up", KEY_UP },
878 { "Down", KEY_DOWN },
879 { "Insert", KEY_IC },
880 { "Delete", KEY_DC },
881 { "Hash", '#' },
882 { "Home", KEY_HOME },
883 { "End", KEY_END },
884 { "PageUp", KEY_PPAGE },
885 { "PageDown", KEY_NPAGE },
886 { "F1", KEY_F(1) },
887 { "F2", KEY_F(2) },
888 { "F3", KEY_F(3) },
889 { "F4", KEY_F(4) },
890 { "F5", KEY_F(5) },
891 { "F6", KEY_F(6) },
892 { "F7", KEY_F(7) },
893 { "F8", KEY_F(8) },
894 { "F9", KEY_F(9) },
895 { "F10", KEY_F(10) },
896 { "F11", KEY_F(11) },
897 { "F12", KEY_F(12) },
898 };
899
900 static int
901 get_key_value(const char *name)
902 {
903 int i;
904
905 for (i = 0; i < ARRAY_SIZE(key_table); i++)
906 if (!strcasecmp(key_table[i].name, name))
907 return key_table[i].value;
908
909 if (strlen(name) == 1 && isprint(*name))
910 return (int) *name;
911
912 return ERR;
913 }
914
915 static char *
916 get_key_name(int key_value)
917 {
918 static char key_char[] = "'X'";
919 char *seq = NULL;
920 int key;
921
922 for (key = 0; key < ARRAY_SIZE(key_table); key++)
923 if (key_table[key].value == key_value)
924 seq = key_table[key].name;
925
926 if (seq == NULL &&
927 key_value < 127 &&
928 isprint(key_value)) {
929 key_char[1] = (char) key_value;
930 seq = key_char;
931 }
932
933 return seq ? seq : "'?'";
934 }
935
936 static char *
937 get_key(enum request request)
938 {
939 static char buf[BUFSIZ];
940 size_t pos = 0;
941 char *sep = "";
942 int i;
943
944 buf[pos] = 0;
945
946 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
947 struct keybinding *keybinding = &default_keybindings[i];
948
949 if (keybinding->request != request)
950 continue;
951
952 if (!string_format_from(buf, &pos, "%s%s", sep,
953 get_key_name(keybinding->alias)))
954 return "Too many keybindings!";
955 sep = ", ";
956 }
957
958 return buf;
959 }
960
961 struct run_request {
962 enum keymap keymap;
963 int key;
964 char cmd[SIZEOF_STR];
965 };
966
967 static struct run_request *run_request;
968 static size_t run_requests;
969
970 static enum request
971 add_run_request(enum keymap keymap, int key, int argc, char **argv)
972 {
973 struct run_request *tmp;
974 struct run_request req = { keymap, key };
975 size_t bufpos;
976
977 for (bufpos = 0; argc > 0; argc--, argv++)
978 if (!string_format_from(req.cmd, &bufpos, "%s ", *argv))
979 return REQ_NONE;
980
981 req.cmd[bufpos - 1] = 0;
982
983 tmp = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
984 if (!tmp)
985 return REQ_NONE;
986
987 run_request = tmp;
988 run_request[run_requests++] = req;
989
990 return REQ_NONE + run_requests;
991 }
992
993 static struct run_request *
994 get_run_request(enum request request)
995 {
996 if (request <= REQ_NONE)
997 return NULL;
998 return &run_request[request - REQ_NONE - 1];
999 }
1000
1001 static void
1002 add_builtin_run_requests(void)
1003 {
1004 struct {
1005 enum keymap keymap;
1006 int key;
1007 char *argv[1];
1008 } reqs[] = {
1009 { KEYMAP_MAIN, 'C', { "git cherry-pick %(commit)" } },
1010 { KEYMAP_GENERIC, 'G', { "git gc" } },
1011 };
1012 int i;
1013
1014 for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1015 enum request req;
1016
1017 req = add_run_request(reqs[i].keymap, reqs[i].key, 1, reqs[i].argv);
1018 if (req != REQ_NONE)
1019 add_keybinding(reqs[i].keymap, req, reqs[i].key);
1020 }
1021 }
1022
1023 /*
1024 * User config file handling.
1025 */
1026
1027 static struct int_map color_map[] = {
1028 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
1029 COLOR_MAP(DEFAULT),
1030 COLOR_MAP(BLACK),
1031 COLOR_MAP(BLUE),
1032 COLOR_MAP(CYAN),
1033 COLOR_MAP(GREEN),
1034 COLOR_MAP(MAGENTA),
1035 COLOR_MAP(RED),
1036 COLOR_MAP(WHITE),
1037 COLOR_MAP(YELLOW),
1038 };
1039
1040 #define set_color(color, name) \
1041 set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
1042
1043 static struct int_map attr_map[] = {
1044 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
1045 ATTR_MAP(NORMAL),
1046 ATTR_MAP(BLINK),
1047 ATTR_MAP(BOLD),
1048 ATTR_MAP(DIM),
1049 ATTR_MAP(REVERSE),
1050 ATTR_MAP(STANDOUT),
1051 ATTR_MAP(UNDERLINE),
1052 };
1053
1054 #define set_attribute(attr, name) \
1055 set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
1056
1057 static int config_lineno;
1058 static bool config_errors;
1059 static char *config_msg;
1060
1061 /* Wants: object fgcolor bgcolor [attr] */
1062 static int
1063 option_color_command(int argc, char *argv[])
1064 {
1065 struct line_info *info;
1066
1067 if (argc != 3 && argc != 4) {
1068 config_msg = "Wrong number of arguments given to color command";
1069 return ERR;
1070 }
1071
1072 info = get_line_info(argv[0], strlen(argv[0]));
1073 if (!info) {
1074 config_msg = "Unknown color name";
1075 return ERR;
1076 }
1077
1078 if (set_color(&info->fg, argv[1]) == ERR ||
1079 set_color(&info->bg, argv[2]) == ERR) {
1080 config_msg = "Unknown color";
1081 return ERR;
1082 }
1083
1084 if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
1085 config_msg = "Unknown attribute";
1086 return ERR;
1087 }
1088
1089 return OK;
1090 }
1091
1092 /* Wants: name = value */
1093 static int
1094 option_set_command(int argc, char *argv[])
1095 {
1096 if (argc != 3) {
1097 config_msg = "Wrong number of arguments given to set command";
1098 return ERR;
1099 }
1100
1101 if (strcmp(argv[1], "=")) {
1102 config_msg = "No value assigned";
1103 return ERR;
1104 }
1105
1106 if (!strcmp(argv[0], "show-rev-graph")) {
1107 opt_rev_graph = (!strcmp(argv[2], "1") ||
1108 !strcmp(argv[2], "true") ||
1109 !strcmp(argv[2], "yes"));
1110 return OK;
1111 }
1112
1113 if (!strcmp(argv[0], "line-number-interval")) {
1114 opt_num_interval = atoi(argv[2]);
1115 return OK;
1116 }
1117
1118 if (!strcmp(argv[0], "tab-size")) {
1119 opt_tab_size = atoi(argv[2]);
1120 return OK;
1121 }
1122
1123 if (!strcmp(argv[0], "commit-encoding")) {
1124 char *arg = argv[2];
1125 int delimiter = *arg;
1126 int i;
1127
1128 switch (delimiter) {
1129 case '"':
1130 case '\'':
1131 for (arg++, i = 0; arg[i]; i++)
1132 if (arg[i] == delimiter) {
1133 arg[i] = 0;
1134 break;
1135 }
1136 default:
1137 string_ncopy(opt_encoding, arg, strlen(arg));
1138 return OK;
1139 }
1140 }
1141
1142 config_msg = "Unknown variable name";
1143 return ERR;
1144 }
1145
1146 /* Wants: mode request key */
1147 static int
1148 option_bind_command(int argc, char *argv[])
1149 {
1150 enum request request;
1151 int keymap;
1152 int key;
1153
1154 if (argc < 3) {
1155 config_msg = "Wrong number of arguments given to bind command";
1156 return ERR;
1157 }
1158
1159 if (set_keymap(&keymap, argv[0]) == ERR) {
1160 config_msg = "Unknown key map";
1161 return ERR;
1162 }
1163
1164 key = get_key_value(argv[1]);
1165 if (key == ERR) {
1166 config_msg = "Unknown key";
1167 return ERR;
1168 }
1169
1170 request = get_request(argv[2]);
1171 if (request == REQ_NONE) {
1172 const char *obsolete[] = { "cherry-pick" };
1173 size_t namelen = strlen(argv[2]);
1174 int i;
1175
1176 for (i = 0; i < ARRAY_SIZE(obsolete); i++) {
1177 if (namelen == strlen(obsolete[i]) &&
1178 !string_enum_compare(obsolete[i], argv[2], namelen)) {
1179 config_msg = "Obsolete request name";
1180 return ERR;
1181 }
1182 }
1183 }
1184 if (request == REQ_NONE && *argv[2]++ == '!')
1185 request = add_run_request(keymap, key, argc - 2, argv + 2);
1186 if (request == REQ_NONE) {
1187 config_msg = "Unknown request name";
1188 return ERR;
1189 }
1190
1191 add_keybinding(keymap, request, key);
1192
1193 return OK;
1194 }
1195
1196 static int
1197 set_option(char *opt, char *value)
1198 {
1199 char *argv[16];
1200 int valuelen;
1201 int argc = 0;
1202
1203 /* Tokenize */
1204 while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1205 argv[argc++] = value;
1206 value += valuelen;
1207
1208 /* Nothing more to tokenize or last available token. */
1209 if (!*value || argc >= ARRAY_SIZE(argv))
1210 break;
1211
1212 *value++ = 0;
1213 while (isspace(*value))
1214 value++;
1215 }
1216
1217 if (!strcmp(opt, "color"))
1218 return option_color_command(argc, argv);
1219
1220 if (!strcmp(opt, "set"))
1221 return option_set_command(argc, argv);
1222
1223 if (!strcmp(opt, "bind"))
1224 return option_bind_command(argc, argv);
1225
1226 config_msg = "Unknown option command";
1227 return ERR;
1228 }
1229
1230 static int
1231 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1232 {
1233 int status = OK;
1234
1235 config_lineno++;
1236 config_msg = "Internal error";
1237
1238 /* Check for comment markers, since read_properties() will
1239 * only ensure opt and value are split at first " \t". */
1240 optlen = strcspn(opt, "#");
1241 if (optlen == 0)
1242 return OK;
1243
1244 if (opt[optlen] != 0) {
1245 config_msg = "No option value";
1246 status = ERR;
1247
1248 } else {
1249 /* Look for comment endings in the value. */
1250 size_t len = strcspn(value, "#");
1251
1252 if (len < valuelen) {
1253 valuelen = len;
1254 value[valuelen] = 0;
1255 }
1256
1257 status = set_option(opt, value);
1258 }
1259
1260 if (status == ERR) {
1261 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1262 config_lineno, (int) optlen, opt, config_msg);
1263 config_errors = TRUE;
1264 }
1265
1266 /* Always keep going if errors are encountered. */
1267 return OK;
1268 }
1269
1270 static int
1271 load_options(void)
1272 {
1273 char *home = getenv("HOME");
1274 char buf[SIZEOF_STR];
1275 FILE *file;
1276
1277 config_lineno = 0;
1278 config_errors = FALSE;
1279
1280 add_builtin_run_requests();
1281
1282 if (!home || !string_format(buf, "%s/.tigrc", home))
1283 return ERR;
1284
1285 /* It's ok that the file doesn't exist. */
1286 file = fopen(buf, "r");
1287 if (!file)
1288 return OK;
1289
1290 if (read_properties(file, " \t", read_option) == ERR ||
1291 config_errors == TRUE)
1292 fprintf(stderr, "Errors while loading %s.\n", buf);
1293
1294 return OK;
1295 }
1296
1297
1298 /*
1299 * The viewer
1300 */
1301
1302 struct view;
1303 struct view_ops;
1304
1305 /* The display array of active views and the index of the current view. */
1306 static struct view *display[2];
1307 static unsigned int current_view;
1308
1309 /* Reading from the prompt? */
1310 static bool input_mode = FALSE;
1311
1312 #define foreach_displayed_view(view, i) \
1313 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1314
1315 #define displayed_views() (display[1] != NULL ? 2 : 1)
1316
1317 /* Current head and commit ID */
1318 static char ref_blob[SIZEOF_REF] = "";
1319 static char ref_commit[SIZEOF_REF] = "HEAD";
1320 static char ref_head[SIZEOF_REF] = "HEAD";
1321
1322 struct view {
1323 const char *name; /* View name */
1324 const char *cmd_fmt; /* Default command line format */
1325 const char *cmd_env; /* Command line set via environment */
1326 const char *id; /* Points to either of ref_{head,commit,blob} */
1327
1328 struct view_ops *ops; /* View operations */
1329
1330 enum keymap keymap; /* What keymap does this view have */
1331
1332 char cmd[SIZEOF_STR]; /* Command buffer */
1333 char ref[SIZEOF_REF]; /* Hovered commit reference */
1334 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
1335
1336 int height, width; /* The width and height of the main window */
1337 WINDOW *win; /* The main window */
1338 WINDOW *title; /* The title window living below the main window */
1339
1340 /* Navigation */
1341 unsigned long offset; /* Offset of the window top */
1342 unsigned long lineno; /* Current line number */
1343
1344 /* Searching */
1345 char grep[SIZEOF_STR]; /* Search string */
1346 regex_t *regex; /* Pre-compiled regex */
1347
1348 /* If non-NULL, points to the view that opened this view. If this view
1349 * is closed tig will switch back to the parent view. */
1350 struct view *parent;
1351
1352 /* Buffering */
1353 unsigned long lines; /* Total number of lines */
1354 struct line *line; /* Line index */
1355 unsigned long line_size;/* Total number of allocated lines */
1356 unsigned int digits; /* Number of digits in the lines member. */
1357
1358 /* Loading */
1359 FILE *pipe;
1360 time_t start_time;
1361 };
1362
1363 struct view_ops {
1364 /* What type of content being displayed. Used in the title bar. */
1365 const char *type;
1366 /* Open and reads in all view content. */
1367 bool (*open)(struct view *view);
1368 /* Read one line; updates view->line. */
1369 bool (*read)(struct view *view, char *data);
1370 /* Draw one line; @lineno must be < view->height. */
1371 bool (*draw)(struct view *view, struct line *line, unsigned int lineno, bool selected);
1372 /* Depending on view handle a special requests. */
1373 enum request (*request)(struct view *view, enum request request, struct line *line);
1374 /* Search for regex in a line. */
1375 bool (*grep)(struct view *view, struct line *line);
1376 /* Select line */
1377 void (*select)(struct view *view, struct line *line);
1378 };
1379
1380 static struct view_ops pager_ops;
1381 static struct view_ops main_ops;
1382 static struct view_ops tree_ops;
1383 static struct view_ops blob_ops;
1384 static struct view_ops help_ops;
1385 static struct view_ops status_ops;
1386 static struct view_ops stage_ops;
1387
1388 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1389 { name, cmd, #env, ref, ops, map}
1390
1391 #define VIEW_(id, name, ops, ref) \
1392 VIEW_STR(name, TIG_##id##_CMD, TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1393
1394
1395 static struct view views[] = {
1396 VIEW_(MAIN, "main", &main_ops, ref_head),
1397 VIEW_(DIFF, "diff", &pager_ops, ref_commit),
1398 VIEW_(LOG, "log", &pager_ops, ref_head),
1399 VIEW_(TREE, "tree", &tree_ops, ref_commit),
1400 VIEW_(BLOB, "blob", &blob_ops, ref_blob),
1401 VIEW_(HELP, "help", &help_ops, ""),
1402 VIEW_(PAGER, "pager", &pager_ops, "stdin"),
1403 VIEW_(STATUS, "status", &status_ops, ""),
1404 VIEW_(STAGE, "stage", &stage_ops, ""),
1405 };
1406
1407 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1408
1409 #define foreach_view(view, i) \
1410 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1411
1412 #define view_is_displayed(view) \
1413 (view == display[0] || view == display[1])
1414
1415 static bool
1416 draw_view_line(struct view *view, unsigned int lineno)
1417 {
1418 struct line *line;
1419 bool selected = (view->offset + lineno == view->lineno);
1420 bool draw_ok;
1421
1422 assert(view_is_displayed(view));
1423
1424 if (view->offset + lineno >= view->lines)
1425 return FALSE;
1426
1427 line = &view->line[view->offset + lineno];
1428
1429 if (selected) {
1430 line->selected = TRUE;
1431 view->ops->select(view, line);
1432 } else if (line->selected) {
1433 line->selected = FALSE;
1434 wmove(view->win, lineno, 0);
1435 wclrtoeol(view->win);
1436 }
1437
1438 scrollok(view->win, FALSE);
1439 draw_ok = view->ops->draw(view, line, lineno, selected);
1440 scrollok(view->win, TRUE);
1441
1442 return draw_ok;
1443 }
1444
1445 static void
1446 redraw_view_from(struct view *view, int lineno)
1447 {
1448 assert(0 <= lineno && lineno < view->height);
1449
1450 for (; lineno < view->height; lineno++) {
1451 if (!draw_view_line(view, lineno))
1452 break;
1453 }
1454
1455 redrawwin(view->win);
1456 if (input_mode)
1457 wnoutrefresh(view->win);
1458 else
1459 wrefresh(view->win);
1460 }
1461
1462 static void
1463 redraw_view(struct view *view)
1464 {
1465 wclear(view->win);
1466 redraw_view_from(view, 0);
1467 }
1468
1469
1470 static void
1471 update_view_title(struct view *view)
1472 {
1473 char buf[SIZEOF_STR];
1474 char state[SIZEOF_STR];
1475 size_t bufpos = 0, statelen = 0;
1476
1477 assert(view_is_displayed(view));
1478
1479 if (view != VIEW(REQ_VIEW_STATUS) && (view->lines || view->pipe)) {
1480 unsigned int view_lines = view->offset + view->height;
1481 unsigned int lines = view->lines
1482 ? MIN(view_lines, view->lines) * 100 / view->lines
1483 : 0;
1484
1485 string_format_from(state, &statelen, "- %s %d of %d (%d%%)",
1486 view->ops->type,
1487 view->lineno + 1,
1488 view->lines,
1489 lines);
1490
1491 if (view->pipe) {
1492 time_t secs = time(NULL) - view->start_time;
1493
1494 /* Three git seconds are a long time ... */
1495 if (secs > 2)
1496 string_format_from(state, &statelen, " %lds", secs);
1497 }
1498 }
1499
1500 string_format_from(buf, &bufpos, "[%s]", view->name);
1501 if (*view->ref && bufpos < view->width) {
1502 size_t refsize = strlen(view->ref);
1503 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
1504
1505 if (minsize < view->width)
1506 refsize = view->width - minsize + 7;
1507 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
1508 }
1509
1510 if (statelen && bufpos < view->width) {
1511 string_format_from(buf, &bufpos, " %s", state);
1512 }
1513
1514 if (view == display[current_view])
1515 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1516 else
1517 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1518
1519 mvwaddnstr(view->title, 0, 0, buf, bufpos);
1520 wclrtoeol(view->title);
1521 wmove(view->title, 0, view->width - 1);
1522
1523 if (input_mode)
1524 wnoutrefresh(view->title);
1525 else
1526 wrefresh(view->title);
1527 }
1528
1529 static void
1530 resize_display(void)
1531 {
1532 int offset, i;
1533 struct view *base = display[0];
1534 struct view *view = display[1] ? display[1] : display[0];
1535
1536 /* Setup window dimensions */
1537
1538 getmaxyx(stdscr, base->height, base->width);
1539
1540 /* Make room for the status window. */
1541 base->height -= 1;
1542
1543 if (view != base) {
1544 /* Horizontal split. */
1545 view->width = base->width;
1546 view->height = SCALE_SPLIT_VIEW(base->height);
1547 base->height -= view->height;
1548
1549 /* Make room for the title bar. */
1550 view->height -= 1;
1551 }
1552
1553 /* Make room for the title bar. */
1554 base->height -= 1;
1555
1556 offset = 0;
1557
1558 foreach_displayed_view (view, i) {
1559 if (!view->win) {
1560 view->win = newwin(view->height, 0, offset, 0);
1561 if (!view->win)
1562 die("Failed to create %s view", view->name);
1563
1564 scrollok(view->win, TRUE);
1565
1566 view->title = newwin(1, 0, offset + view->height, 0);
1567 if (!view->title)
1568 die("Failed to create title window");
1569
1570 } else {
1571 wresize(view->win, view->height, view->width);
1572 mvwin(view->win, offset, 0);
1573 mvwin(view->title, offset + view->height, 0);
1574 }
1575
1576 offset += view->height + 1;
1577 }
1578 }
1579
1580 static void
1581 redraw_display(void)
1582 {
1583 struct view *view;
1584 int i;
1585
1586 foreach_displayed_view (view, i) {
1587 redraw_view(view);
1588 update_view_title(view);
1589 }
1590 }
1591
1592 static void
1593 update_display_cursor(struct view *view)
1594 {
1595 /* Move the cursor to the right-most column of the cursor line.
1596 *
1597 * XXX: This could turn out to be a bit expensive, but it ensures that
1598 * the cursor does not jump around. */
1599 if (view->lines) {
1600 wmove(view->win, view->lineno - view->offset, view->width - 1);
1601 wrefresh(view->win);
1602 }
1603 }
1604
1605 /*
1606 * Navigation
1607 */
1608
1609 /* Scrolling backend */
1610 static void
1611 do_scroll_view(struct view *view, int lines)
1612 {
1613 bool redraw_current_line = FALSE;
1614
1615 /* The rendering expects the new offset. */
1616 view->offset += lines;
1617
1618 assert(0 <= view->offset && view->offset < view->lines);
1619 assert(lines);
1620
1621 /* Move current line into the view. */
1622 if (view->lineno < view->offset) {
1623 view->lineno = view->offset;
1624 redraw_current_line = TRUE;
1625 } else if (view->lineno >= view->offset + view->height) {
1626 view->lineno = view->offset + view->height - 1;
1627 redraw_current_line = TRUE;
1628 }
1629
1630 assert(view->offset <= view->lineno && view->lineno < view->lines);
1631
1632 /* Redraw the whole screen if scrolling is pointless. */
1633 if (view->height < ABS(lines)) {
1634 redraw_view(view);
1635
1636 } else {
1637 int line = lines > 0 ? view->height - lines : 0;
1638 int end = line + ABS(lines);
1639
1640 wscrl(view->win, lines);
1641
1642 for (; line < end; line++) {
1643 if (!draw_view_line(view, line))
1644 break;
1645 }
1646
1647 if (redraw_current_line)
1648 draw_view_line(view, view->lineno - view->offset);
1649 }
1650
1651 redrawwin(view->win);
1652 wrefresh(view->win);
1653 report("");
1654 }
1655
1656 /* Scroll frontend */
1657 static void
1658 scroll_view(struct view *view, enum request request)
1659 {
1660 int lines = 1;
1661
1662 assert(view_is_displayed(view));
1663
1664 switch (request) {
1665 case REQ_SCROLL_PAGE_DOWN:
1666 lines = view->height;
1667 case REQ_SCROLL_LINE_DOWN:
1668 if (view->offset + lines > view->lines)
1669 lines = view->lines - view->offset;
1670
1671 if (lines == 0 || view->offset + view->height >= view->lines) {
1672 report("Cannot scroll beyond the last line");
1673 return;
1674 }
1675 break;
1676
1677 case REQ_SCROLL_PAGE_UP:
1678 lines = view->height;
1679 case REQ_SCROLL_LINE_UP:
1680 if (lines > view->offset)
1681 lines = view->offset;
1682
1683 if (lines == 0) {
1684 report("Cannot scroll beyond the first line");
1685 return;
1686 }
1687
1688 lines = -lines;
1689 break;
1690
1691 default:
1692 die("request %d not handled in switch", request);
1693 }
1694
1695 do_scroll_view(view, lines);
1696 }
1697
1698 /* Cursor moving */
1699 static void
1700 move_view(struct view *view, enum request request)
1701 {
1702 int scroll_steps = 0;
1703 int steps;
1704
1705 switch (request) {
1706 case REQ_MOVE_FIRST_LINE:
1707 steps = -view->lineno;
1708 break;
1709
1710 case REQ_MOVE_LAST_LINE:
1711 steps = view->lines - view->lineno - 1;
1712 break;
1713
1714 case REQ_MOVE_PAGE_UP:
1715 steps = view->height > view->lineno
1716 ? -view->lineno : -view->height;
1717 break;
1718
1719 case REQ_MOVE_PAGE_DOWN:
1720 steps = view->lineno + view->height >= view->lines
1721 ? view->lines - view->lineno - 1 : view->height;
1722 break;
1723
1724 case REQ_MOVE_UP:
1725 steps = -1;
1726 break;
1727
1728 case REQ_MOVE_DOWN:
1729 steps = 1;
1730 break;
1731
1732 default:
1733 die("request %d not handled in switch", request);
1734 }
1735
1736 if (steps <= 0 && view->lineno == 0) {
1737 report("Cannot move beyond the first line");
1738 return;
1739
1740 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1741 report("Cannot move beyond the last line");
1742 return;
1743 }
1744
1745 /* Move the current line */
1746 view->lineno += steps;
1747 assert(0 <= view->lineno && view->lineno < view->lines);
1748
1749 /* Check whether the view needs to be scrolled */
1750 if (view->lineno < view->offset ||
1751 view->lineno >= view->offset + view->height) {
1752 scroll_steps = steps;
1753 if (steps < 0 && -steps > view->offset) {
1754 scroll_steps = -view->offset;
1755
1756 } else if (steps > 0) {
1757 if (view->lineno == view->lines - 1 &&
1758 view->lines > view->height) {
1759 scroll_steps = view->lines - view->offset - 1;
1760 if (scroll_steps >= view->height)
1761 scroll_steps -= view->height - 1;
1762 }
1763 }
1764 }
1765
1766 if (!view_is_displayed(view)) {
1767 view->offset += scroll_steps;
1768 assert(0 <= view->offset && view->offset < view->lines);
1769 view->ops->select(view, &view->line[view->lineno]);
1770 return;
1771 }
1772
1773 /* Repaint the old "current" line if we be scrolling */
1774 if (ABS(steps) < view->height)
1775 draw_view_line(view, view->lineno - steps - view->offset);
1776
1777 if (scroll_steps) {
1778 do_scroll_view(view, scroll_steps);
1779 return;
1780 }
1781
1782 /* Draw the current line */
1783 draw_view_line(view, view->lineno - view->offset);
1784
1785 redrawwin(view->win);
1786 wrefresh(view->win);
1787 report("");
1788 }
1789
1790
1791 /*
1792 * Searching
1793 */
1794
1795 static void search_view(struct view *view, enum request request);
1796
1797 static bool
1798 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1799 {
1800 assert(view_is_displayed(view));
1801
1802 if (!view->ops->grep(view, line))
1803 return FALSE;
1804
1805 if (lineno - view->offset >= view->height) {
1806 view->offset = lineno;
1807 view->lineno = lineno;
1808 redraw_view(view);
1809
1810 } else {
1811 unsigned long old_lineno = view->lineno - view->offset;
1812
1813 view->lineno = lineno;
1814 draw_view_line(view, old_lineno);
1815
1816 draw_view_line(view, view->lineno - view->offset);
1817 redrawwin(view->win);
1818 wrefresh(view->win);
1819 }
1820
1821 report("Line %ld matches '%s'", lineno + 1, view->grep);
1822 return TRUE;
1823 }
1824
1825 static void
1826 find_next(struct view *view, enum request request)
1827 {
1828 unsigned long lineno = view->lineno;
1829 int direction;
1830
1831 if (!*view->grep) {
1832 if (!*opt_search)
1833 report("No previous search");
1834 else
1835 search_view(view, request);
1836 return;
1837 }
1838
1839 switch (request) {
1840 case REQ_SEARCH:
1841 case REQ_FIND_NEXT:
1842 direction = 1;
1843 break;
1844
1845 case REQ_SEARCH_BACK:
1846 case REQ_FIND_PREV:
1847 direction = -1;
1848 break;
1849
1850 default:
1851 return;
1852 }
1853
1854 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1855 lineno += direction;
1856
1857 /* Note, lineno is unsigned long so will wrap around in which case it
1858 * will become bigger than view->lines. */
1859 for (; lineno < view->lines; lineno += direction) {
1860 struct line *line = &view->line[lineno];
1861
1862 if (find_next_line(view, lineno, line))
1863 return;
1864 }
1865
1866 report("No match found for '%s'", view->grep);
1867 }
1868
1869 static void
1870 search_view(struct view *view, enum request request)
1871 {
1872 int regex_err;
1873
1874 if (view->regex) {
1875 regfree(view->regex);
1876 *view->grep = 0;
1877 } else {
1878 view->regex = calloc(1, sizeof(*view->regex));
1879 if (!view->regex)
1880 return;
1881 }
1882
1883 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
1884 if (regex_err != 0) {
1885 char buf[SIZEOF_STR] = "unknown error";
1886
1887 regerror(regex_err, view->regex, buf, sizeof(buf));
1888 report("Search failed: %s", buf);
1889 return;
1890 }
1891
1892 string_copy(view->grep, opt_search);
1893
1894 find_next(view, request);
1895 }
1896
1897 /*
1898 * Incremental updating
1899 */
1900
1901 static void
1902 end_update(struct view *view)
1903 {
1904 if (!view->pipe)
1905 return;
1906 set_nonblocking_input(FALSE);
1907 if (view->pipe == stdin)
1908 fclose(view->pipe);
1909 else
1910 pclose(view->pipe);
1911 view->pipe = NULL;
1912 }
1913
1914 static bool
1915 begin_update(struct view *view)
1916 {
1917 if (view->pipe)
1918 end_update(view);
1919
1920 if (opt_cmd[0]) {
1921 string_copy(view->cmd, opt_cmd);
1922 opt_cmd[0] = 0;
1923 /* When running random commands, initially show the
1924 * command in the title. However, it maybe later be
1925 * overwritten if a commit line is selected. */
1926 if (view == VIEW(REQ_VIEW_PAGER))
1927 string_copy(view->ref, view->cmd);
1928 else
1929 view->ref[0] = 0;
1930
1931 } else if (view == VIEW(REQ_VIEW_TREE)) {
1932 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1933 char path[SIZEOF_STR];
1934
1935 if (strcmp(view->vid, view->id))
1936 opt_path[0] = path[0] = 0;
1937 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
1938 return FALSE;
1939
1940 if (!string_format(view->cmd, format, view->id, path))
1941 return FALSE;
1942
1943 } else {
1944 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1945 const char *id = view->id;
1946
1947 if (!string_format(view->cmd, format, id, id, id, id, id))
1948 return FALSE;
1949
1950 /* Put the current ref_* value to the view title ref
1951 * member. This is needed by the blob view. Most other
1952 * views sets it automatically after loading because the
1953 * first line is a commit line. */
1954 string_copy_rev(view->ref, view->id);
1955 }
1956
1957 /* Special case for the pager view. */
1958 if (opt_pipe) {
1959 view->pipe = opt_pipe;
1960 opt_pipe = NULL;
1961 } else {
1962 view->pipe = popen(view->cmd, "r");
1963 }
1964
1965 if (!view->pipe)
1966 return FALSE;
1967
1968 set_nonblocking_input(TRUE);
1969
1970 view->offset = 0;
1971 view->lines = 0;
1972 view->lineno = 0;
1973 string_copy_rev(view->vid, view->id);
1974
1975 if (view->line) {
1976 int i;
1977
1978 for (i = 0; i < view->lines; i++)
1979 if (view->line[i].data)
1980 free(view->line[i].data);
1981
1982 free(view->line);
1983 view->line = NULL;
1984 }
1985
1986 view->start_time = time(NULL);
1987
1988 return TRUE;
1989 }
1990
1991 static struct line *
1992 realloc_lines(struct view *view, size_t line_size)
1993 {
1994 struct line *tmp = realloc(view->line, sizeof(*view->line) * line_size);
1995
1996 if (!tmp)
1997 return NULL;
1998
1999 view->line = tmp;
2000 view->line_size = line_size;
2001 return view->line;
2002 }
2003
2004 static bool
2005 update_view(struct view *view)
2006 {
2007 char in_buffer[BUFSIZ];
2008 char out_buffer[BUFSIZ * 2];
2009 char *line;
2010 /* The number of lines to read. If too low it will cause too much
2011 * redrawing (and possible flickering), if too high responsiveness
2012 * will suffer. */
2013 unsigned long lines = view->height;
2014 int redraw_from = -1;
2015
2016 if (!view->pipe)
2017 return TRUE;
2018
2019 /* Only redraw if lines are visible. */
2020 if (view->offset + view->height >= view->lines)
2021 redraw_from = view->lines - view->offset;
2022
2023 /* FIXME: This is probably not perfect for backgrounded views. */
2024 if (!realloc_lines(view, view->lines + lines))
2025 goto alloc_error;
2026
2027 while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
2028 size_t linelen = strlen(line);
2029
2030 if (linelen)
2031 line[linelen - 1] = 0;
2032
2033 if (opt_iconv != ICONV_NONE) {
2034 ICONV_CONST char *inbuf = line;
2035 size_t inlen = linelen;
2036
2037 char *outbuf = out_buffer;
2038 size_t outlen = sizeof(out_buffer);
2039
2040 size_t ret;
2041
2042 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2043 if (ret != (size_t) -1) {
2044 line = out_buffer;
2045 linelen = strlen(out_buffer);
2046 }
2047 }
2048
2049 if (!view->ops->read(view, line))
2050 goto alloc_error;
2051
2052 if (lines-- == 1)
2053 break;
2054 }
2055
2056 {
2057 int digits;
2058
2059 lines = view->lines;
2060 for (digits = 0; lines; digits++)
2061 lines /= 10;
2062
2063 /* Keep the displayed view in sync with line number scaling. */
2064 if (digits != view->digits) {
2065 view->digits = digits;
2066 redraw_from = 0;
2067 }
2068 }
2069
2070 if (!view_is_displayed(view))
2071 goto check_pipe;
2072
2073 if (view == VIEW(REQ_VIEW_TREE)) {
2074 /* Clear the view and redraw everything since the tree sorting
2075 * might have rearranged things. */
2076 redraw_view(view);
2077
2078 } else if (redraw_from >= 0) {
2079 /* If this is an incremental update, redraw the previous line
2080 * since for commits some members could have changed when
2081 * loading the main view. */
2082 if (redraw_from > 0)
2083 redraw_from--;
2084
2085 /* Since revision graph visualization requires knowledge
2086 * about the parent commit, it causes a further one-off
2087 * needed to be redrawn for incremental updates. */
2088 if (redraw_from > 0 && opt_rev_graph)
2089 redraw_from--;
2090
2091 /* Incrementally draw avoids flickering. */
2092 redraw_view_from(view, redraw_from);
2093 }
2094
2095 /* Update the title _after_ the redraw so that if the redraw picks up a
2096 * commit reference in view->ref it'll be available here. */
2097 update_view_title(view);
2098
2099 check_pipe:
2100 if (ferror(view->pipe)) {
2101 report("Failed to read: %s", strerror(errno));
2102 goto end;
2103
2104 } else if (feof(view->pipe)) {
2105 report("");
2106 goto end;
2107 }
2108
2109 return TRUE;
2110
2111 alloc_error:
2112 report("Allocation failure");
2113
2114 end:
2115 view->ops->read(view, NULL);
2116 end_update(view);
2117 return FALSE;
2118 }
2119
2120 static struct line *
2121 add_line_data(struct view *view, void *data, enum line_type type)
2122 {
2123 struct line *line = &view->line[view->lines++];
2124
2125 memset(line, 0, sizeof(*line));
2126 line->type = type;
2127 line->data = data;
2128
2129 return line;
2130 }
2131
2132 static struct line *
2133 add_line_text(struct view *view, char *data, enum line_type type)
2134 {
2135 if (data)
2136 data = strdup(data);
2137
2138 return data ? add_line_data(view, data, type) : NULL;
2139 }
2140
2141
2142 /*
2143 * View opening
2144 */
2145
2146 enum open_flags {
2147 OPEN_DEFAULT = 0, /* Use default view switching. */
2148 OPEN_SPLIT = 1, /* Split current view. */
2149 OPEN_BACKGROUNDED = 2, /* Backgrounded. */
2150 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
2151 };
2152
2153 static void
2154 open_view(struct view *prev, enum request request, enum open_flags flags)
2155 {
2156 bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2157 bool split = !!(flags & OPEN_SPLIT);
2158 bool reload = !!(flags & OPEN_RELOAD);
2159 struct view *view = VIEW(request);
2160 int nviews = displayed_views();
2161 struct view *base_view = display[0];
2162
2163 if (view == prev && nviews == 1 && !reload) {
2164 report("Already in %s view", view->name);
2165 return;
2166 }
2167
2168 if (view->ops->open) {
2169 if (!view->ops->open(view)) {
2170 report("Failed to load %s view", view->name);
2171 return;
2172 }
2173
2174 } else if ((reload || strcmp(view->vid, view->id)) &&
2175 !begin_update(view)) {
2176 report("Failed to load %s view", view->name);
2177 return;
2178 }
2179
2180 if (split) {
2181 display[1] = view;
2182 if (!backgrounded)
2183 current_view = 1;
2184 } else {
2185 /* Maximize the current view. */
2186 memset(display, 0, sizeof(display));
2187 current_view = 0;
2188 display[current_view] = view;
2189 }
2190
2191 /* Resize the view when switching between split- and full-screen,
2192 * or when switching between two different full-screen views. */
2193 if (nviews != displayed_views() ||
2194 (nviews == 1 && base_view != display[0]))
2195 resize_display();
2196
2197 if (split && prev->lineno - prev->offset >= prev->height) {
2198 /* Take the title line into account. */
2199 int lines = prev->lineno - prev->offset - prev->height + 1;
2200
2201 /* Scroll the view that was split if the current line is
2202 * outside the new limited view. */
2203 do_scroll_view(prev, lines);
2204 }
2205
2206 if (prev && view != prev) {
2207 if (split && !backgrounded) {
2208 /* "Blur" the previous view. */
2209 update_view_title(prev);
2210 }
2211
2212 view->parent = prev;
2213 }
2214
2215 if (view->pipe && view->lines == 0) {
2216 /* Clear the old view and let the incremental updating refill
2217 * the screen. */
2218 wclear(view->win);
2219 report("");
2220 } else {
2221 redraw_view(view);
2222 report("");
2223 }
2224
2225 /* If the view is backgrounded the above calls to report()
2226 * won't redraw the view title. */
2227 if (backgrounded)
2228 update_view_title(view);
2229 }
2230
2231 static void
2232 open_external_viewer(const char *cmd)
2233 {
2234 def_prog_mode(); /* save current tty modes */
2235 endwin(); /* restore original tty modes */
2236 system(cmd);
2237 fprintf(stderr, "Press Enter to continue");
2238 getc(stdin);
2239 reset_prog_mode();
2240 redraw_display();
2241 }
2242
2243 static void
2244 open_mergetool(const char *file)
2245 {
2246 char cmd[SIZEOF_STR];
2247 char file_sq[SIZEOF_STR];
2248
2249 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2250 string_format(cmd, "git mergetool %s", file_sq)) {
2251 open_external_viewer(cmd);
2252 }
2253 }
2254
2255 static void
2256 open_editor(bool from_root, const char *file)
2257 {
2258 char cmd[SIZEOF_STR];
2259 char file_sq[SIZEOF_STR];
2260 char *editor;
2261 char *prefix = from_root ? opt_cdup : "";
2262
2263 editor = getenv("GIT_EDITOR");
2264 if (!editor && *opt_editor)
2265 editor = opt_editor;
2266 if (!editor)
2267 editor = getenv("VISUAL");
2268 if (!editor)
2269 editor = getenv("EDITOR");
2270 if (!editor)
2271 editor = "vi";
2272
2273 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2274 string_format(cmd, "%s %s%s", editor, prefix, file_sq)) {
2275 open_external_viewer(cmd);
2276 }
2277 }
2278
2279 static void
2280 open_run_request(enum request request)
2281 {
2282 struct run_request *req = get_run_request(request);
2283 char buf[SIZEOF_STR * 2];
2284 size_t bufpos;
2285 char *cmd;
2286
2287 if (!req) {
2288 report("Unknown run request");
2289 return;
2290 }
2291
2292 bufpos = 0;
2293 cmd = req->cmd;
2294
2295 while (cmd) {
2296 char *next = strstr(cmd, "%(");
2297 int len = next - cmd;
2298 char *value;
2299
2300 if (!next) {
2301 len = strlen(cmd);
2302 value = "";
2303
2304 } else if (!strncmp(next, "%(head)", 7)) {
2305 value = ref_head;
2306
2307 } else if (!strncmp(next, "%(commit)", 9)) {
2308 value = ref_commit;
2309
2310 } else if (!strncmp(next, "%(blob)", 7)) {
2311 value = ref_blob;
2312
2313 } else {
2314 report("Unknown replacement in run request: `%s`", req->cmd);
2315 return;
2316 }
2317
2318 if (!string_format_from(buf, &bufpos, "%.*s%s", len, cmd, value))
2319 return;
2320
2321 if (next)
2322 next = strchr(next, ')') + 1;
2323 cmd = next;
2324 }
2325
2326 open_external_viewer(buf);
2327 }
2328
2329 /*
2330 * User request switch noodle
2331 */
2332
2333 static int
2334 view_driver(struct view *view, enum request request)
2335 {
2336 int i;
2337
2338 if (request == REQ_NONE) {
2339 doupdate();
2340 return TRUE;
2341 }
2342
2343 if (request > REQ_NONE) {
2344 open_run_request(request);
2345 return TRUE;
2346 }
2347
2348 if (view && view->lines) {
2349 request = view->ops->request(view, request, &view->line[view->lineno]);
2350 if (request == REQ_NONE)
2351 return TRUE;
2352 }
2353
2354 switch (request) {
2355 case REQ_MOVE_UP:
2356 case REQ_MOVE_DOWN:
2357 case REQ_MOVE_PAGE_UP:
2358 case REQ_MOVE_PAGE_DOWN:
2359 case REQ_MOVE_FIRST_LINE:
2360 case REQ_MOVE_LAST_LINE:
2361 move_view(view, request);
2362 break;
2363
2364 case REQ_SCROLL_LINE_DOWN:
2365 case REQ_SCROLL_LINE_UP:
2366 case REQ_SCROLL_PAGE_DOWN:
2367 case REQ_SCROLL_PAGE_UP:
2368 scroll_view(view, request);
2369 break;
2370
2371 case REQ_VIEW_BLOB:
2372 if (!ref_blob[0]) {
2373 report("No file chosen, press %s to open tree view",
2374 get_key(REQ_VIEW_TREE));
2375 break;
2376 }
2377 open_view(view, request, OPEN_DEFAULT);
2378 break;
2379
2380 case REQ_VIEW_PAGER:
2381 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2382 report("No pager content, press %s to run command from prompt",
2383 get_key(REQ_PROMPT));
2384 break;
2385 }
2386 open_view(view, request, OPEN_DEFAULT);
2387 break;
2388
2389 case REQ_VIEW_STAGE:
2390 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2391 report("No stage content, press %s to open the status view and choose file",
2392 get_key(REQ_VIEW_STATUS));
2393 break;
2394 }
2395 open_view(view, request, OPEN_DEFAULT);
2396 break;
2397
2398 case REQ_VIEW_STATUS:
2399 if (opt_is_inside_work_tree == FALSE) {
2400 report("The status view requires a working tree");
2401 break;
2402 }
2403 open_view(view, request, OPEN_DEFAULT);
2404 break;
2405
2406 case REQ_VIEW_MAIN:
2407 case REQ_VIEW_DIFF:
2408 case REQ_VIEW_LOG:
2409 case REQ_VIEW_TREE:
2410 case REQ_VIEW_HELP:
2411 open_view(view, request, OPEN_DEFAULT);
2412 break;
2413
2414 case REQ_NEXT:
2415 case REQ_PREVIOUS:
2416 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2417
2418 if ((view == VIEW(REQ_VIEW_DIFF) &&
2419 view->parent == VIEW(REQ_VIEW_MAIN)) ||
2420 (view == VIEW(REQ_VIEW_STAGE) &&
2421 view->parent == VIEW(REQ_VIEW_STATUS)) ||
2422 (view == VIEW(REQ_VIEW_BLOB) &&
2423 view->parent == VIEW(REQ_VIEW_TREE))) {
2424 int line;
2425
2426 view = view->parent;
2427 line = view->lineno;
2428 move_view(view, request);
2429 if (view_is_displayed(view))
2430 update_view_title(view);
2431 if (line != view->lineno)
2432 view->ops->request(view, REQ_ENTER,
2433 &view->line[view->lineno]);
2434
2435 } else {
2436 move_view(view, request);
2437 }
2438 break;
2439
2440 case REQ_VIEW_NEXT:
2441 {
2442 int nviews = displayed_views();
2443 int next_view = (current_view + 1) % nviews;
2444
2445 if (next_view == current_view) {
2446 report("Only one view is displayed");
2447 break;
2448 }
2449
2450 current_view = next_view;
2451 /* Blur out the title of the previous view. */
2452 update_view_title(view);
2453 report("");
2454 break;
2455 }
2456 case REQ_REFRESH:
2457 report("Refreshing is not yet supported for the %s view", view->name);
2458 break;
2459
2460 case REQ_TOGGLE_LINENO:
2461 opt_line_number = !opt_line_number;
2462 redraw_display();
2463 break;
2464
2465 case REQ_TOGGLE_REV_GRAPH:
2466 opt_rev_graph = !opt_rev_graph;
2467 redraw_display();
2468 break;
2469
2470 case REQ_PROMPT:
2471 /* Always reload^Wrerun commands from the prompt. */
2472 open_view(view, opt_request, OPEN_RELOAD);
2473 break;
2474
2475 case REQ_SEARCH:
2476 case REQ_SEARCH_BACK:
2477 search_view(view, request);
2478 break;
2479
2480 case REQ_FIND_NEXT:
2481 case REQ_FIND_PREV:
2482 find_next(view, request);
2483 break;
2484
2485 case REQ_STOP_LOADING:
2486 for (i = 0; i < ARRAY_SIZE(views); i++) {
2487 view = &views[i];
2488 if (view->pipe)
2489 report("Stopped loading the %s view", view->name),
2490 end_update(view);
2491 }
2492 break;
2493
2494 case REQ_SHOW_VERSION:
2495 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2496 return TRUE;
2497
2498 case REQ_SCREEN_RESIZE:
2499 resize_display();
2500 /* Fall-through */
2501 case REQ_SCREEN_REDRAW:
2502 redraw_display();
2503 break;
2504
2505 case REQ_EDIT:
2506 report("Nothing to edit");
2507 break;
2508
2509
2510 case REQ_ENTER:
2511 report("Nothing to enter");
2512 break;
2513
2514
2515 case REQ_VIEW_CLOSE:
2516 /* XXX: Mark closed views by letting view->parent point to the
2517 * view itself. Parents to closed view should never be
2518 * followed. */
2519 if (view->parent &&
2520 view->parent->parent != view->parent) {
2521 memset(display, 0, sizeof(display));
2522 current_view = 0;
2523 display[current_view] = view->parent;
2524 view->parent = view;
2525 resize_display();
2526 redraw_display();
2527 break;
2528 }
2529 /* Fall-through */
2530 case REQ_QUIT:
2531 return FALSE;
2532
2533 default:
2534 /* An unknown key will show most commonly used commands. */
2535 report("Unknown key, press 'h' for help");
2536 return TRUE;
2537 }
2538
2539 return TRUE;
2540 }
2541
2542
2543 /*
2544 * Pager backend
2545 */
2546
2547 static bool
2548 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2549 {
2550 char *text = line->data;
2551 enum line_type type = line->type;
2552 int textlen = strlen(text);
2553 int attr;
2554
2555 wmove(view->win, lineno, 0);
2556
2557 if (selected) {
2558 type = LINE_CURSOR;
2559 wchgat(view->win, -1, 0, type, NULL);
2560 }
2561
2562 attr = get_line_attr(type);
2563 wattrset(view->win, attr);
2564
2565 if (opt_line_number || opt_tab_size < TABSIZE) {
2566 static char spaces[] = " ";
2567 int col_offset = 0, col = 0;
2568
2569 if (opt_line_number) {
2570 unsigned long real_lineno = view->offset + lineno + 1;
2571
2572 if (real_lineno == 1 ||
2573 (real_lineno % opt_num_interval) == 0) {
2574 wprintw(view->win, "%.*d", view->digits, real_lineno);
2575
2576 } else {
2577 waddnstr(view->win, spaces,
2578 MIN(view->digits, STRING_SIZE(spaces)));
2579 }
2580 waddstr(view->win, ": ");
2581 col_offset = view->digits + 2;
2582 }
2583
2584 while (text && col_offset + col < view->width) {
2585 int cols_max = view->width - col_offset - col;
2586 char *pos = text;
2587 int cols;
2588
2589 if (*text == '\t') {
2590 text++;
2591 assert(sizeof(spaces) > TABSIZE);
2592 pos = spaces;
2593 cols = opt_tab_size - (col % opt_tab_size);
2594
2595 } else {
2596 text = strchr(text, '\t');
2597 cols = line ? text - pos : strlen(pos);
2598 }
2599
2600 waddnstr(view->win, pos, MIN(cols, cols_max));
2601 col += cols;
2602 }
2603
2604 } else {
2605 int col = 0, pos = 0;
2606
2607 for (; pos < textlen && col < view->width; pos++, col++)
2608 if (text[pos] == '\t')
2609 col += TABSIZE - (col % TABSIZE) - 1;
2610
2611 waddnstr(view->win, text, pos);
2612 }
2613
2614 return TRUE;
2615 }
2616
2617 static bool
2618 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2619 {
2620 char refbuf[SIZEOF_STR];
2621 char *ref = NULL;
2622 FILE *pipe;
2623
2624 if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2625 return TRUE;
2626
2627 pipe = popen(refbuf, "r");
2628 if (!pipe)
2629 return TRUE;
2630
2631 if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2632 ref = chomp_string(ref);
2633 pclose(pipe);
2634
2635 if (!ref || !*ref)
2636 return TRUE;
2637
2638 /* This is the only fatal call, since it can "corrupt" the buffer. */
2639 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2640 return FALSE;
2641
2642 return TRUE;
2643 }
2644
2645 static void
2646 add_pager_refs(struct view *view, struct line *line)
2647 {
2648 char buf[SIZEOF_STR];
2649 char *commit_id = line->data + STRING_SIZE("commit ");
2650 struct ref **refs;
2651 size_t bufpos = 0, refpos = 0;
2652 const char *sep = "Refs: ";
2653 bool is_tag = FALSE;
2654
2655 assert(line->type == LINE_COMMIT);
2656
2657 refs = get_refs(commit_id);
2658 if (!refs) {
2659 if (view == VIEW(REQ_VIEW_DIFF))
2660 goto try_add_describe_ref;
2661 return;
2662 }
2663
2664 do {
2665 struct ref *ref = refs[refpos];
2666 char *fmt = ref->tag ? "%s[%s]" :
2667 ref->remote ? "%s<%s>" : "%s%s";
2668
2669 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2670 return;
2671 sep = ", ";
2672 if (ref->tag)
2673 is_tag = TRUE;
2674 } while (refs[refpos++]->next);
2675
2676 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2677 try_add_describe_ref:
2678 /* Add <tag>-g<commit_id> "fake" reference. */
2679 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2680 return;
2681 }
2682
2683 if (bufpos == 0)
2684 return;
2685
2686 if (!realloc_lines(view, view->line_size + 1))
2687 return;
2688
2689 add_line_text(view, buf, LINE_PP_REFS);
2690 }
2691
2692 static bool
2693 pager_read(struct view *view, char *data)
2694 {
2695 struct line *line;
2696
2697 if (!data)
2698 return TRUE;
2699
2700 line = add_line_text(view, data, get_line_type(data));
2701 if (!line)
2702 return FALSE;
2703
2704 if (line->type == LINE_COMMIT &&
2705 (view == VIEW(REQ_VIEW_DIFF) ||
2706 view == VIEW(REQ_VIEW_LOG)))
2707 add_pager_refs(view, line);
2708
2709 return TRUE;
2710 }
2711
2712 static enum request
2713 pager_request(struct view *view, enum request request, struct line *line)
2714 {
2715 int split = 0;
2716
2717 if (request != REQ_ENTER)
2718 return request;
2719
2720 if (line->type == LINE_COMMIT &&
2721 (view == VIEW(REQ_VIEW_LOG) ||
2722 view == VIEW(REQ_VIEW_PAGER))) {
2723 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2724 split = 1;
2725 }
2726
2727 /* Always scroll the view even if it was split. That way
2728 * you can use Enter to scroll through the log view and
2729 * split open each commit diff. */
2730 scroll_view(view, REQ_SCROLL_LINE_DOWN);
2731
2732 /* FIXME: A minor workaround. Scrolling the view will call report("")
2733 * but if we are scrolling a non-current view this won't properly
2734 * update the view title. */
2735 if (split)
2736 update_view_title(view);
2737
2738 return REQ_NONE;
2739 }
2740
2741 static bool
2742 pager_grep(struct view *view, struct line *line)
2743 {
2744 regmatch_t pmatch;
2745 char *text = line->data;
2746
2747 if (!*text)
2748 return FALSE;
2749
2750 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2751 return FALSE;
2752
2753 return TRUE;
2754 }
2755
2756 static void
2757 pager_select(struct view *view, struct line *line)
2758 {
2759 if (line->type == LINE_COMMIT) {
2760 char *text = line->data + STRING_SIZE("commit ");
2761
2762 if (view != VIEW(REQ_VIEW_PAGER))
2763 string_copy_rev(view->ref, text);
2764 string_copy_rev(ref_commit, text);
2765 }
2766 }
2767
2768 static struct view_ops pager_ops = {
2769 "line",
2770 NULL,
2771 pager_read,
2772 pager_draw,
2773 pager_request,
2774 pager_grep,
2775 pager_select,
2776 };
2777
2778
2779 /*
2780 * Help backend
2781 */
2782
2783 static bool
2784 help_open(struct view *view)
2785 {
2786 char buf[BUFSIZ];
2787 int lines = ARRAY_SIZE(req_info) + 2;
2788 int i;
2789
2790 if (view->lines > 0)
2791 return TRUE;
2792
2793 for (i = 0; i < ARRAY_SIZE(req_info); i++)
2794 if (!req_info[i].request)
2795 lines++;
2796
2797 lines += run_requests + 1;
2798
2799 view->line = calloc(lines, sizeof(*view->line));
2800 if (!view->line)
2801 return FALSE;
2802
2803 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
2804
2805 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
2806 char *key;
2807
2808 if (req_info[i].request == REQ_NONE)
2809 continue;
2810
2811 if (!req_info[i].request) {
2812 add_line_text(view, "", LINE_DEFAULT);
2813 add_line_text(view, req_info[i].help, LINE_DEFAULT);
2814 continue;
2815 }
2816
2817 key = get_key(req_info[i].request);
2818 if (!*key)
2819 key = "(no key defined)";
2820
2821 if (!string_format(buf, " %-25s %s", key, req_info[i].help))
2822 continue;
2823
2824 add_line_text(view, buf, LINE_DEFAULT);
2825 }
2826
2827 if (run_requests) {
2828 add_line_text(view, "", LINE_DEFAULT);
2829 add_line_text(view, "External commands:", LINE_DEFAULT);
2830 }
2831
2832 for (i = 0; i < run_requests; i++) {
2833 struct run_request *req = get_run_request(REQ_NONE + i + 1);
2834 char *key;
2835
2836 if (!req)
2837 continue;
2838
2839 key = get_key_name(req->key);
2840 if (!*key)
2841 key = "(no key defined)";
2842
2843 if (!string_format(buf, " %-10s %-14s `%s`",
2844 keymap_table[req->keymap].name,
2845 key, req->cmd))
2846 continue;
2847
2848 add_line_text(view, buf, LINE_DEFAULT);
2849 }
2850
2851 return TRUE;
2852 }
2853
2854 static struct view_ops help_ops = {
2855 "line",
2856 help_open,
2857 NULL,
2858 pager_draw,
2859 pager_request,
2860 pager_grep,
2861 pager_select,
2862 };
2863
2864
2865 /*
2866 * Tree backend
2867 */
2868
2869 struct tree_stack_entry {
2870 struct tree_stack_entry *prev; /* Entry below this in the stack */
2871 unsigned long lineno; /* Line number to restore */
2872 char *name; /* Position of name in opt_path */
2873 };
2874
2875 /* The top of the path stack. */
2876 static struct tree_stack_entry *tree_stack = NULL;
2877 unsigned long tree_lineno = 0;
2878
2879 static void
2880 pop_tree_stack_entry(void)
2881 {
2882 struct tree_stack_entry *entry = tree_stack;
2883
2884 tree_lineno = entry->lineno;
2885 entry->name[0] = 0;
2886 tree_stack = entry->prev;
2887 free(entry);
2888 }
2889
2890 static void
2891 push_tree_stack_entry(char *name, unsigned long lineno)
2892 {
2893 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
2894 size_t pathlen = strlen(opt_path);
2895
2896 if (!entry)
2897 return;
2898
2899 entry->prev = tree_stack;
2900 entry->name = opt_path + pathlen;
2901 tree_stack = entry;
2902
2903 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
2904 pop_tree_stack_entry();
2905 return;
2906 }
2907
2908 /* Move the current line to the first tree entry. */
2909 tree_lineno = 1;
2910 entry->lineno = lineno;
2911 }
2912
2913 /* Parse output from git-ls-tree(1):
2914 *
2915 * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
2916 * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
2917 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
2918 * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
2919 */
2920
2921 #define SIZEOF_TREE_ATTR \
2922 STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
2923
2924 #define TREE_UP_FORMAT "040000 tree %s\t.."
2925
2926 static int
2927 tree_compare_entry(enum line_type type1, char *name1,
2928 enum line_type type2, char *name2)
2929 {
2930 if (type1 != type2) {
2931 if (type1 == LINE_TREE_DIR)
2932 return -1;
2933 return 1;
2934 }
2935
2936 return strcmp(name1, name2);
2937 }
2938
2939 static bool
2940 tree_read(struct view *view, char *text)
2941 {
2942 size_t textlen = text ? strlen(text) : 0;
2943 char buf[SIZEOF_STR];
2944 unsigned long pos;
2945 enum line_type type;
2946 bool first_read = view->lines == 0;
2947
2948 if (textlen <= SIZEOF_TREE_ATTR)
2949 return FALSE;
2950
2951 type = text[STRING_SIZE("100644 ")] == 't'
2952 ? LINE_TREE_DIR : LINE_TREE_FILE;
2953
2954 if (first_read) {
2955 /* Add path info line */
2956 if (!string_format(buf, "Directory path /%s", opt_path) ||
2957 !realloc_lines(view, view->line_size + 1) ||
2958 !add_line_text(view, buf, LINE_DEFAULT))
2959 return FALSE;
2960
2961 /* Insert "link" to parent directory. */
2962 if (*opt_path) {
2963 if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
2964 !realloc_lines(view, view->line_size + 1) ||
2965 !add_line_text(view, buf, LINE_TREE_DIR))
2966 return FALSE;
2967 }
2968 }
2969
2970 /* Strip the path part ... */
2971 if (*opt_path) {
2972 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
2973 size_t striplen = strlen(opt_path);
2974 char *path = text + SIZEOF_TREE_ATTR;
2975
2976 if (pathlen > striplen)
2977 memmove(path, path + striplen,
2978 pathlen - striplen + 1);
2979 }
2980
2981 /* Skip "Directory ..." and ".." line. */
2982 for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
2983 struct line *line = &view->line[pos];
2984 char *path1 = ((char *) line->data) + SIZEOF_TREE_ATTR;
2985 char *path2 = text + SIZEOF_TREE_ATTR;
2986 int cmp = tree_compare_entry(line->type, path1, type, path2);
2987
2988 if (cmp <= 0)
2989 continue;
2990
2991 text = strdup(text);
2992 if (!text)
2993 return FALSE;
2994
2995 if (view->lines > pos)
2996 memmove(&view->line[pos + 1], &view->line[pos],
2997 (view->lines - pos) * sizeof(*line));
2998
2999 line = &view->line[pos];
3000 line->data = text;
3001 line->type = type;
3002 view->lines++;
3003 return TRUE;
3004 }
3005
3006 if (!add_line_text(view, text, type))
3007 return FALSE;
3008
3009 if (tree_lineno > view->lineno) {
3010 view->lineno = tree_lineno;
3011 tree_lineno = 0;
3012 }
3013
3014 return TRUE;
3015 }
3016
3017 static enum request
3018 tree_request(struct view *view, enum request request, struct line *line)
3019 {
3020 enum open_flags flags;
3021
3022 if (request != REQ_ENTER)
3023 return request;
3024
3025 /* Cleanup the stack if the tree view is at a different tree. */
3026 while (!*opt_path && tree_stack)
3027 pop_tree_stack_entry();
3028
3029 switch (line->type) {
3030 case LINE_TREE_DIR:
3031 /* Depending on whether it is a subdir or parent (updir?) link
3032 * mangle the path buffer. */
3033 if (line == &view->line[1] && *opt_path) {
3034 pop_tree_stack_entry();
3035
3036 } else {
3037 char *data = line->data;
3038 char *basename = data + SIZEOF_TREE_ATTR;
3039
3040 push_tree_stack_entry(basename, view->lineno);
3041 }
3042
3043 /* Trees and subtrees share the same ID, so they are not not
3044 * unique like blobs. */
3045 flags = OPEN_RELOAD;
3046 request = REQ_VIEW_TREE;
3047 break;
3048
3049 case LINE_TREE_FILE:
3050 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3051 request = REQ_VIEW_BLOB;
3052 break;
3053
3054 default:
3055 return TRUE;
3056 }
3057
3058 open_view(view, request, flags);
3059 if (request == REQ_VIEW_TREE) {
3060 view->lineno = tree_lineno;
3061 }
3062
3063 return REQ_NONE;
3064 }
3065
3066 static void
3067 tree_select(struct view *view, struct line *line)
3068 {
3069 char *text = line->data + STRING_SIZE("100644 blob ");
3070
3071 if (line->type == LINE_TREE_FILE) {
3072 string_copy_rev(ref_blob, text);
3073
3074 } else if (line->type != LINE_TREE_DIR) {
3075 return;
3076 }
3077
3078 string_copy_rev(view->ref, text);
3079 }
3080
3081 static struct view_ops tree_ops = {
3082 "file",
3083 NULL,
3084 tree_read,
3085 pager_draw,
3086 tree_request,
3087 pager_grep,
3088 tree_select,
3089 };
3090
3091 static bool
3092 blob_read(struct view *view, char *line)
3093 {
3094 return add_line_text(view, line, LINE_DEFAULT) != NULL;
3095 }
3096
3097 static struct view_ops blob_ops = {
3098 "line",
3099 NULL,
3100 blob_read,
3101 pager_draw,
3102 pager_request,
3103 pager_grep,
3104 pager_select,
3105 };
3106
3107
3108 /*
3109 * Status backend
3110 */
3111
3112 struct status {
3113 char status;
3114 struct {
3115 mode_t mode;
3116 char rev[SIZEOF_REV];
3117 } old;
3118 struct {
3119 mode_t mode;
3120 char rev[SIZEOF_REV];
3121 } new;
3122 char name[SIZEOF_STR];
3123 };
3124
3125 static struct status stage_status;
3126 static enum line_type stage_line_type;
3127
3128 /* Get fields from the diff line:
3129 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
3130 */
3131 static inline bool
3132 status_get_diff(struct status *file, char *buf, size_t bufsize)
3133 {
3134 char *old_mode = buf + 1;
3135 char *new_mode = buf + 8;
3136 char *old_rev = buf + 15;
3137 char *new_rev = buf + 56;
3138 char *status = buf + 97;
3139
3140 if (bufsize != 99 ||
3141 old_mode[-1] != ':' ||
3142 new_mode[-1] != ' ' ||
3143 old_rev[-1] != ' ' ||
3144 new_rev[-1] != ' ' ||
3145 status[-1] != ' ')
3146 return FALSE;
3147
3148 file->status = *status;
3149
3150 string_copy_rev(file->old.rev, old_rev);
3151 string_copy_rev(file->new.rev, new_rev);
3152
3153 file->old.mode = strtoul(old_mode, NULL, 8);
3154 file->new.mode = strtoul(new_mode, NULL, 8);
3155
3156 file->name[0] = 0;
3157
3158 return TRUE;
3159 }
3160
3161 static bool
3162 status_run(struct view *view, const char cmd[], bool diff, enum line_type type)
3163 {
3164 struct status *file = NULL;
3165 struct status *unmerged = NULL;
3166 char buf[SIZEOF_STR * 4];
3167 size_t bufsize = 0;
3168 FILE *pipe;
3169
3170 pipe = popen(cmd, "r");
3171 if (!pipe)
3172 return FALSE;
3173
3174 add_line_data(view, NULL, type);
3175
3176 while (!feof(pipe) && !ferror(pipe)) {
3177 char *sep;
3178 size_t readsize;
3179
3180 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
3181 if (!readsize)
3182 break;
3183 bufsize += readsize;
3184
3185 /* Process while we have NUL chars. */
3186 while ((sep = memchr(buf, 0, bufsize))) {
3187 size_t sepsize = sep - buf + 1;
3188
3189 if (!file) {
3190 if (!realloc_lines(view, view->line_size + 1))
3191 goto error_out;
3192
3193 file = calloc(1, sizeof(*file));
3194 if (!file)
3195 goto error_out;
3196
3197 add_line_data(view, file, type);
3198 }
3199
3200 /* Parse diff info part. */
3201 if (!diff) {
3202 file->status = '?';
3203
3204 } else if (!file->status) {
3205 if (!status_get_diff(file, buf, sepsize))
3206 goto error_out;
3207
3208 bufsize -= sepsize;
3209 memmove(buf, sep + 1, bufsize);
3210
3211 sep = memchr(buf, 0, bufsize);
3212 if (!sep)
3213 break;
3214 sepsize = sep - buf + 1;
3215
3216 /* Collapse all 'M'odified entries that
3217 * follow a associated 'U'nmerged entry.
3218 */
3219 if (file->status == 'U') {
3220 unmerged = file;
3221
3222 } else if (unmerged) {
3223 int collapse = !strcmp(buf, unmerged->name);
3224
3225 unmerged = NULL;
3226 if (collapse) {
3227 free(file);
3228 view->lines--;
3229 continue;
3230 }
3231 }
3232 }
3233
3234 /* git-ls-files just delivers a NUL separated
3235 * list of file names similar to the second half
3236 * of the git-diff-* output. */
3237 string_ncopy(file->name, buf, sepsize);
3238 bufsize -= sepsize;
3239 memmove(buf, sep + 1, bufsize);
3240 file = NULL;
3241 }
3242 }
3243
3244 if (ferror(pipe)) {
3245 error_out:
3246 pclose(pipe);
3247 return FALSE;
3248 }
3249
3250 if (!view->line[view->lines - 1].data)
3251 add_line_data(view, NULL, LINE_STAT_NONE);
3252
3253 pclose(pipe);
3254 return TRUE;
3255 }
3256
3257 /* Don't show unmerged entries in the staged section. */
3258 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --diff-filter=ACDMRTXB --cached HEAD"
3259 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
3260 #define STATUS_LIST_OTHER_CMD \
3261 "git ls-files -z --others --exclude-per-directory=.gitignore"
3262
3263 #define STATUS_DIFF_INDEX_SHOW_CMD \
3264 "git diff-index --root --patch-with-stat --find-copies-harder -B -C --cached HEAD -- %s 2>/dev/null"
3265
3266 #define STATUS_DIFF_FILES_SHOW_CMD \
3267 "git diff-files --root --patch-with-stat --find-copies-harder -B -C -- %s 2>/dev/null"
3268
3269 /* First parse staged info using git-diff-index(1), then parse unstaged
3270 * info using git-diff-files(1), and finally untracked files using
3271 * git-ls-files(1). */
3272 static bool
3273 status_open(struct view *view)
3274 {
3275 struct stat statbuf;
3276 char exclude[SIZEOF_STR];
3277 char cmd[SIZEOF_STR];
3278 unsigned long prev_lineno = view->lineno;
3279 size_t i;
3280
3281 for (i = 0; i < view->lines; i++)
3282 free(view->line[i].data);
3283 free(view->line);
3284 view->lines = view->line_size = view->lineno = 0;
3285 view->line = NULL;
3286
3287 if (!realloc_lines(view, view->line_size + 6))
3288 return FALSE;
3289
3290 if (!string_format(exclude, "%s/info/exclude", opt_git_dir))
3291 return FALSE;
3292
3293 string_copy(cmd, STATUS_LIST_OTHER_CMD);
3294
3295 if (stat(exclude, &statbuf) >= 0) {
3296 size_t cmdsize = strlen(cmd);
3297
3298 if (!string_format_from(cmd, &cmdsize, " %s", "--exclude-from=") ||
3299 sq_quote(cmd, cmdsize, exclude) >= sizeof(cmd))
3300 return FALSE;
3301 }
3302
3303 if (!status_run(view, STATUS_DIFF_INDEX_CMD, TRUE, LINE_STAT_STAGED) ||
3304 !status_run(view, STATUS_DIFF_FILES_CMD, TRUE, LINE_STAT_UNSTAGED) ||
3305 !status_run(view, cmd, FALSE, LINE_STAT_UNTRACKED))
3306 return FALSE;
3307
3308 /* If all went well restore the previous line number to stay in
3309 * the context. */
3310 if (prev_lineno < view->lines)
3311 view->lineno = prev_lineno;
3312 else
3313 view->lineno = view->lines - 1;
3314
3315 return TRUE;
3316 }
3317
3318 static bool
3319 status_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3320 {
3321 struct status *status = line->data;
3322
3323 wmove(view->win, lineno, 0);
3324
3325 if (selected) {
3326 wattrset(view->win, get_line_attr(LINE_CURSOR));
3327 wchgat(view->win, -1, 0, LINE_CURSOR, NULL);
3328
3329 } else if (!status && line->type != LINE_STAT_NONE) {
3330 wattrset(view->win, get_line_attr(LINE_STAT_SECTION));
3331 wchgat(view->win, -1, 0, LINE_STAT_SECTION, NULL);
3332
3333 } else {
3334 wattrset(view->win, get_line_attr(line->type));
3335 }
3336
3337 if (!status) {
3338 char *text;
3339
3340 switch (line->type) {
3341 case LINE_STAT_STAGED:
3342 text = "Changes to be committed:";
3343 break;
3344
3345 case LINE_STAT_UNSTAGED:
3346 text = "Changed but not updated:";
3347 break;
3348
3349 case LINE_STAT_UNTRACKED:
3350 text = "Untracked files:";
3351 break;
3352
3353 case LINE_STAT_NONE:
3354 text = " (no files)";
3355 break;
3356
3357 default:
3358 return FALSE;
3359 }
3360
3361 waddstr(view->win, text);
3362 return TRUE;
3363 }
3364
3365 waddch(view->win, status->status);
3366 if (!selected)
3367 wattrset(view->win, A_NORMAL);
3368 wmove(view->win, lineno, 4);
3369 waddstr(view->win, status->name);
3370
3371 return TRUE;
3372 }
3373
3374 static enum request
3375 status_enter(struct view *view, struct line *line)
3376 {
3377 struct status *status = line->data;
3378 char path[SIZEOF_STR] = "";
3379 char *info;
3380 size_t cmdsize = 0;
3381
3382 if (line->type == LINE_STAT_NONE ||
3383 (!status && line[1].type == LINE_STAT_NONE)) {
3384 report("No file to diff");
3385 return REQ_NONE;
3386 }
3387
3388 if (status && sq_quote(path, 0, status->name) >= sizeof(path))
3389 return REQ_QUIT;
3390
3391 if (opt_cdup[0] &&
3392 line->type != LINE_STAT_UNTRACKED &&
3393 !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
3394 return REQ_QUIT;
3395
3396 switch (line->type) {
3397 case LINE_STAT_STAGED:
3398 if (!string_format_from(opt_cmd, &cmdsize,
3399 STATUS_DIFF_INDEX_SHOW_CMD, path))
3400 return REQ_QUIT;
3401 if (status)
3402 info = "Staged changes to %s";
3403 else
3404 info = "Staged changes";
3405 break;
3406
3407 case LINE_STAT_UNSTAGED:
3408 if (!string_format_from(opt_cmd, &cmdsize,
3409 STATUS_DIFF_FILES_SHOW_CMD, path))
3410 return REQ_QUIT;
3411 if (status)
3412 info = "Unstaged changes to %s";
3413 else
3414 info = "Unstaged changes";
3415 break;
3416
3417 case LINE_STAT_UNTRACKED:
3418 if (opt_pipe)
3419 return REQ_QUIT;
3420
3421
3422 if (!status) {
3423 report("No file to show");
3424 return REQ_NONE;
3425 }
3426
3427 opt_pipe = fopen(status->name, "r");
3428 info = "Untracked file %s";
3429 break;
3430
3431 default:
3432 die("line type %d not handled in switch", line->type);
3433 }
3434
3435 open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | OPEN_SPLIT);
3436 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
3437 if (status) {
3438 stage_status = *status;
3439 } else {
3440 memset(&stage_status, 0, sizeof(stage_status));
3441 }
3442
3443 stage_line_type = line->type;
3444 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.name);
3445 }
3446
3447 return REQ_NONE;
3448 }
3449
3450
3451 static bool
3452 status_update_file(struct view *view, struct status *status, enum line_type type)
3453 {
3454 char cmd[SIZEOF_STR];
3455 char buf[SIZEOF_STR];
3456 size_t cmdsize = 0;
3457 size_t bufsize = 0;
3458 size_t written = 0;
3459 FILE *pipe;
3460
3461 if (opt_cdup[0] &&
3462 type != LINE_STAT_UNTRACKED &&
3463 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
3464 return FALSE;
3465
3466 switch (type) {
3467 case LINE_STAT_STAGED:
3468 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
3469 status->old.mode,
3470 status->old.rev,
3471 status->name, 0))
3472 return FALSE;
3473
3474 string_add(cmd, cmdsize, "git update-index -z --index-info");
3475 break;
3476
3477 case LINE_STAT_UNSTAGED:
3478 case LINE_STAT_UNTRACKED:
3479 if (!string_format_from(buf, &bufsize, "%s%c", status->name, 0))
3480 return FALSE;
3481
3482 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
3483 break;
3484
3485 default:
3486 die("line type %d not handled in switch", type);
3487 }
3488
3489 pipe = popen(cmd, "w");
3490 if (!pipe)
3491 return FALSE;
3492
3493 while (!ferror(pipe) && written < bufsize) {
3494 written += fwrite(buf + written, 1, bufsize - written, pipe);
3495 }
3496
3497 pclose(pipe);
3498
3499 if (written != bufsize)
3500 return FALSE;
3501
3502 return TRUE;
3503 }
3504
3505 static void
3506 status_update(struct view *view)
3507 {
3508 struct line *line = &view->line[view->lineno];
3509
3510 assert(view->lines);
3511
3512 if (!line->data) {
3513 while (++line < view->line + view->lines && line->data) {
3514 if (!status_update_file(view, line->data, line->type))
3515 report("Failed to update file status");
3516 }
3517
3518 if (!line[-1].data) {
3519 report("Nothing to update");
3520 return;
3521 }
3522
3523 } else if (!status_update_file(view, line->data, line->type)) {
3524 report("Failed to update file status");
3525 }
3526 }
3527
3528 static enum request
3529 status_request(struct view *view, enum request request, struct line *line)
3530 {
3531 struct status *status = line->data;
3532
3533 switch (request) {
3534 case REQ_STATUS_UPDATE:
3535 status_update(view);
3536 break;
3537
3538 case REQ_STATUS_MERGE:
3539 if (!status || status->status != 'U') {
3540 report("Merging only possible for files with unmerged status ('U').");
3541 return REQ_NONE;
3542 }
3543 open_mergetool(status->name);
3544 break;
3545
3546 case REQ_EDIT:
3547 if (!status)
3548 return request;
3549
3550 open_editor(status->status != '?', status->name);
3551 break;
3552
3553 case REQ_ENTER:
3554 /* After returning the status view has been split to
3555 * show the stage view. No further reloading is
3556 * necessary. */
3557 status_enter(view, line);
3558 return REQ_NONE;
3559
3560 case REQ_REFRESH:
3561 /* Simply reload the view. */
3562 break;
3563
3564 default:
3565 return request;
3566 }
3567
3568 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
3569
3570 return REQ_NONE;
3571 }
3572
3573 static void
3574 status_select(struct view *view, struct line *line)
3575 {
3576 struct status *status = line->data;
3577 char file[SIZEOF_STR] = "all files";
3578 char *text;
3579 char *key;
3580
3581 if (status && !string_format(file, "'%s'", status->name))
3582 return;
3583
3584 if (!status && line[1].type == LINE_STAT_NONE)
3585 line++;
3586
3587 switch (line->type) {
3588 case LINE_STAT_STAGED:
3589 text = "Press %s to unstage %s for commit";
3590 break;
3591
3592 case LINE_STAT_UNSTAGED:
3593 text = "Press %s to stage %s for commit";
3594 break;
3595
3596 case LINE_STAT_UNTRACKED:
3597 text = "Press %s to stage %s for addition";
3598 break;
3599
3600 case LINE_STAT_NONE:
3601 text = "Nothing to update";
3602 break;
3603
3604 default:
3605 die("line type %d not handled in switch", line->type);
3606 }
3607
3608 if (status && status->status == 'U') {
3609 text = "Press %s to resolve conflict in %s";
3610 key = get_key(REQ_STATUS_MERGE);
3611
3612 } else {
3613 key = get_key(REQ_STATUS_UPDATE);
3614 }
3615
3616 string_format(view->ref, text, key, file);
3617 }
3618
3619 static bool
3620 status_grep(struct view *view, struct line *line)
3621 {
3622 struct status *status = line->data;
3623 enum { S_STATUS, S_NAME, S_END } state;
3624 char buf[2] = "?";
3625 regmatch_t pmatch;
3626
3627 if (!status)
3628 return FALSE;
3629
3630 for (state = S_STATUS; state < S_END; state++) {
3631 char *text;
3632
3633 switch (state) {
3634 case S_NAME: text = status->name; break;
3635 case S_STATUS:
3636 buf[0] = status->status;
3637 text = buf;
3638 break;
3639
3640 default:
3641 return FALSE;
3642 }
3643
3644 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3645 return TRUE;
3646 }
3647
3648 return FALSE;
3649 }
3650
3651 static struct view_ops status_ops = {
3652 "file",
3653 status_open,
3654 NULL,
3655 status_draw,
3656 status_request,
3657 status_grep,
3658 status_select,
3659 };
3660
3661
3662 static bool
3663 stage_diff_line(FILE *pipe, struct line *line)
3664 {
3665 char *buf = line->data;
3666 size_t bufsize = strlen(buf);
3667 size_t written = 0;
3668
3669 while (!ferror(pipe) && written < bufsize) {
3670 written += fwrite(buf + written, 1, bufsize - written, pipe);
3671 }
3672
3673 fputc('\n', pipe);
3674
3675 return written == bufsize;
3676 }
3677
3678 static struct line *
3679 stage_diff_hdr(struct view *view, struct line *line)
3680 {
3681 int diff_hdr_dir = line->type == LINE_DIFF_CHUNK ? -1 : 1;
3682 struct line *diff_hdr;
3683
3684 if (line->type == LINE_DIFF_CHUNK)
3685 diff_hdr = line - 1;
3686 else
3687 diff_hdr = view->line + 1;
3688
3689 while (diff_hdr > view->line && diff_hdr < view->line + view->lines) {
3690 if (diff_hdr->type == LINE_DIFF_HEADER)
3691 return diff_hdr;
3692
3693 diff_hdr += diff_hdr_dir;
3694 }
3695
3696 return NULL;
3697 }
3698
3699 static bool
3700 stage_update_chunk(struct view *view, struct line *line)
3701 {
3702 char cmd[SIZEOF_STR];
3703 size_t cmdsize = 0;
3704 struct line *diff_hdr, *diff_chunk, *diff_end;
3705 FILE *pipe;
3706
3707 diff_hdr = stage_diff_hdr(view, line);
3708 if (!diff_hdr)
3709 return FALSE;
3710
3711 if (opt_cdup[0] &&
3712 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
3713 return FALSE;
3714
3715 if (!string_format_from(cmd, &cmdsize,
3716 "git apply --cached %s - && "
3717 "git update-index -q --unmerged --refresh 2>/dev/null",
3718 stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
3719 return FALSE;
3720
3721 pipe = popen(cmd, "w");
3722 if (!pipe)
3723 return FALSE;
3724
3725 diff_end = view->line + view->lines;
3726 if (line->type != LINE_DIFF_CHUNK) {
3727 diff_chunk = diff_hdr;
3728
3729 } else {
3730 for (diff_chunk = line + 1; diff_chunk < diff_end; diff_chunk++)
3731 if (diff_chunk->type == LINE_DIFF_CHUNK ||
3732 diff_chunk->type == LINE_DIFF_HEADER)
3733 diff_end = diff_chunk;
3734
3735 diff_chunk = line;
3736
3737 while (diff_hdr->type != LINE_DIFF_CHUNK) {
3738 switch (diff_hdr->type) {
3739 case LINE_DIFF_HEADER:
3740 case LINE_DIFF_INDEX:
3741 case LINE_DIFF_ADD:
3742 case LINE_DIFF_DEL:
3743 break;
3744
3745 default:
3746 diff_hdr++;
3747 continue;
3748 }
3749
3750 if (!stage_diff_line(pipe, diff_hdr++)) {
3751 pclose(pipe);
3752 return FALSE;
3753 }
3754 }
3755 }
3756
3757 while (diff_chunk < diff_end && stage_diff_line(pipe, diff_chunk))
3758 diff_chunk++;
3759
3760 pclose(pipe);
3761
3762 if (diff_chunk != diff_end)
3763 return FALSE;
3764
3765 return TRUE;
3766 }
3767
3768 static void
3769 stage_update(struct view *view, struct line *line)
3770 {
3771 if (stage_line_type != LINE_STAT_UNTRACKED &&
3772 (line->type == LINE_DIFF_CHUNK || !stage_status.status)) {
3773 if (!stage_update_chunk(view, line)) {
3774 report("Failed to apply chunk");
3775 return;
3776 }
3777
3778 } else if (!status_update_file(view, &stage_status, stage_line_type)) {
3779 report("Failed to update file");
3780 return;
3781 }
3782
3783 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
3784
3785 view = VIEW(REQ_VIEW_STATUS);
3786 if (view_is_displayed(view))
3787 status_enter(view, &view->line[view->lineno]);
3788 }
3789
3790 static enum request
3791 stage_request(struct view *view, enum request request, struct line *line)
3792 {
3793 switch (request) {
3794 case REQ_STATUS_UPDATE:
3795 stage_update(view, line);
3796 break;
3797
3798 case REQ_EDIT:
3799 if (!stage_status.name[0])
3800 return request;
3801
3802 open_editor(stage_status.status != '?', stage_status.name);
3803 break;
3804
3805 case REQ_ENTER:
3806 pager_request(view, request, line);
3807 break;
3808
3809 default:
3810 return request;
3811 }
3812
3813 return REQ_NONE;
3814 }
3815
3816 static struct view_ops stage_ops = {
3817 "line",
3818 NULL,
3819 pager_read,
3820 pager_draw,
3821 stage_request,
3822 pager_grep,
3823 pager_select,
3824 };
3825
3826
3827 /*
3828 * Revision graph
3829 */
3830
3831 struct commit {
3832 char id[SIZEOF_REV]; /* SHA1 ID. */
3833 char title[128]; /* First line of the commit message. */
3834 char author[75]; /* Author of the commit. */
3835 struct tm time; /* Date from the author ident. */
3836 struct ref **refs; /* Repository references. */
3837 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
3838 size_t graph_size; /* The width of the graph array. */
3839 };
3840
3841 /* Size of rev graph with no "padding" columns */
3842 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
3843
3844 struct rev_graph {
3845 struct rev_graph *prev, *next, *parents;
3846 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
3847 size_t size;
3848 struct commit *commit;
3849 size_t pos;
3850 };
3851
3852 /* Parents of the commit being visualized. */
3853 static struct rev_graph graph_parents[4];
3854
3855 /* The current stack of revisions on the graph. */
3856 static struct rev_graph graph_stacks[4] = {
3857 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
3858 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
3859 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
3860 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
3861 };
3862
3863 static inline bool
3864 graph_parent_is_merge(struct rev_graph *graph)
3865 {
3866 return graph->parents->size > 1;
3867 }
3868
3869 static inline void
3870 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
3871 {
3872 struct commit *commit = graph->commit;
3873
3874 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
3875 commit->graph[commit->graph_size++] = symbol;
3876 }
3877
3878 static void
3879 done_rev_graph(struct rev_graph *graph)
3880 {
3881 if (graph_parent_is_merge(graph) &&
3882 graph->pos < graph->size - 1 &&
3883 graph->next->size == graph->size + graph->parents->size - 1) {
3884 size_t i = graph->pos + graph->parents->size - 1;
3885
3886 graph->commit->graph_size = i * 2;
3887 while (i < graph->next->size - 1) {
3888 append_to_rev_graph(graph, ' ');
3889 append_to_rev_graph(graph, '\\');
3890 i++;
3891 }
3892 }
3893
3894 graph->size = graph->pos = 0;
3895 graph->commit = NULL;
3896 memset(graph->parents, 0, sizeof(*graph->parents));
3897 }
3898
3899 static void
3900 push_rev_graph(struct rev_graph *graph, char *parent)
3901 {
3902 int i;
3903
3904 /* "Collapse" duplicate parents lines.
3905 *
3906 * FIXME: This needs to also update update the drawn graph but
3907 * for now it just serves as a method for pruning graph lines. */
3908 for (i = 0; i < graph->size; i++)
3909 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
3910 return;
3911
3912 if (graph->size < SIZEOF_REVITEMS) {
3913 string_copy_rev(graph->rev[graph->size++], parent);
3914 }
3915 }
3916
3917 static chtype
3918 get_rev_graph_symbol(struct rev_graph *graph)
3919 {
3920 chtype symbol;
3921
3922 if (graph->parents->size == 0)
3923 symbol = REVGRAPH_INIT;
3924 else if (graph_parent_is_merge(graph))
3925 symbol = REVGRAPH_MERGE;
3926 else if (graph->pos >= graph->size)
3927 symbol = REVGRAPH_BRANCH;
3928 else
3929 symbol = REVGRAPH_COMMIT;
3930
3931 return symbol;
3932 }
3933
3934 static void
3935 draw_rev_graph(struct rev_graph *graph)
3936 {
3937 struct rev_filler {
3938 chtype separator, line;
3939 };
3940 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
3941 static struct rev_filler fillers[] = {
3942 { ' ', REVGRAPH_LINE },
3943 { '`', '.' },
3944 { '\'', ' ' },
3945 { '/', ' ' },
3946 };
3947 chtype symbol = get_rev_graph_symbol(graph);
3948 struct rev_filler *filler;
3949 size_t i;
3950
3951 filler = &fillers[DEFAULT];
3952
3953 for (i = 0; i < graph->pos; i++) {
3954 append_to_rev_graph(graph, filler->line);
3955 if (graph_parent_is_merge(graph->prev) &&
3956 graph->prev->pos == i)
3957 filler = &fillers[RSHARP];
3958
3959 append_to_rev_graph(graph, filler->separator);
3960 }
3961
3962 /* Place the symbol for this revision. */
3963 append_to_rev_graph(graph, symbol);
3964
3965 if (graph->prev->size > graph->size)
3966 filler = &fillers[RDIAG];
3967 else
3968 filler = &fillers[DEFAULT];
3969
3970 i++;
3971
3972 for (; i < graph->size; i++) {
3973 append_to_rev_graph(graph, filler->separator);
3974 append_to_rev_graph(graph, filler->line);
3975 if (graph_parent_is_merge(graph->prev) &&
3976 i < graph->prev->pos + graph->parents->size)
3977 filler = &fillers[RSHARP];
3978 if (graph->prev->size > graph->size)
3979 filler = &fillers[LDIAG];
3980 }
3981
3982 if (graph->prev->size > graph->size) {
3983 append_to_rev_graph(graph, filler->separator);
3984 if (filler->line != ' ')
3985 append_to_rev_graph(graph, filler->line);
3986 }
3987 }
3988
3989 /* Prepare the next rev graph */
3990 static void
3991 prepare_rev_graph(struct rev_graph *graph)
3992 {
3993 size_t i;
3994
3995 /* First, traverse all lines of revisions up to the active one. */
3996 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
3997 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
3998 break;
3999
4000 push_rev_graph(graph->next, graph->rev[graph->pos]);
4001 }
4002
4003 /* Interleave the new revision parent(s). */
4004 for (i = 0; i < graph->parents->size; i++)
4005 push_rev_graph(graph->next, graph->parents->rev[i]);
4006
4007 /* Lastly, put any remaining revisions. */
4008 for (i = graph->pos + 1; i < graph->size; i++)
4009 push_rev_graph(graph->next, graph->rev[i]);
4010 }
4011
4012 static void
4013 update_rev_graph(struct rev_graph *graph)
4014 {
4015 /* If this is the finalizing update ... */
4016 if (graph->commit)
4017 prepare_rev_graph(graph);
4018
4019 /* Graph visualization needs a one rev look-ahead,
4020 * so the first update doesn't visualize anything. */
4021 if (!graph->prev->commit)
4022 return;
4023
4024 draw_rev_graph(graph->prev);
4025 done_rev_graph(graph->prev->prev);
4026 }
4027
4028
4029 /*
4030 * Main view backend
4031 */
4032
4033 static bool
4034 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
4035 {
4036 char buf[DATE_COLS + 1];
4037 struct commit *commit = line->data;
4038 enum line_type type;
4039 int col = 0;
4040 size_t timelen;
4041 size_t authorlen;
4042 int trimmed = 1;
4043
4044 if (!*commit->author)
4045 return FALSE;
4046
4047 wmove(view->win, lineno, col);
4048
4049 if (selected) {
4050 type = LINE_CURSOR;
4051 wattrset(view->win, get_line_attr(type));
4052 wchgat(view->win, -1, 0, type, NULL);
4053
4054 } else {
4055 type = LINE_MAIN_COMMIT;
4056 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
4057 }
4058
4059 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
4060 waddnstr(view->win, buf, timelen);
4061 waddstr(view->win, " ");
4062
4063 col += DATE_COLS;
4064 wmove(view->win, lineno, col);
4065 if (type != LINE_CURSOR)
4066 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
4067
4068 if (opt_utf8) {
4069 authorlen = utf8_length(commit->author, AUTHOR_COLS - 2, &col, &trimmed);
4070 } else {
4071 authorlen = strlen(commit->author);
4072 if (authorlen > AUTHOR_COLS - 2) {
4073 authorlen = AUTHOR_COLS - 2;
4074 trimmed = 1;
4075 }
4076 }
4077
4078 if (trimmed) {
4079 waddnstr(view->win, commit->author, authorlen);
4080 if (type != LINE_CURSOR)
4081 wattrset(view->win, get_line_attr(LINE_MAIN_DELIM));
4082 waddch(view->win, '~');
4083 } else {
4084 waddstr(view->win, commit->author);
4085 }
4086
4087 col += AUTHOR_COLS;
4088 if (type != LINE_CURSOR)
4089 wattrset(view->win, A_NORMAL);
4090
4091 if (opt_rev_graph && commit->graph_size) {
4092 size_t i;
4093
4094 wmove(view->win, lineno, col);
4095 /* Using waddch() instead of waddnstr() ensures that
4096 * they'll be rendered correctly for the cursor line. */
4097 for (i = 0; i < commit->graph_size; i++)
4098 waddch(view->win, commit->graph[i]);
4099
4100 waddch(view->win, ' ');
4101 col += commit->graph_size + 1;
4102 }
4103
4104 wmove(view->win, lineno, col);
4105
4106 if (commit->refs) {
4107 size_t i = 0;
4108
4109 do {
4110 if (type == LINE_CURSOR)
4111 ;
4112 else if (commit->refs[i]->tag)
4113 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
4114 else if (commit->refs[i]->remote)
4115 wattrset(view->win, get_line_attr(LINE_MAIN_REMOTE));
4116 else
4117 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
4118 waddstr(view->win, "[");
4119 waddstr(view->win, commit->refs[i]->name);
4120 waddstr(view->win, "]");
4121 if (type != LINE_CURSOR)
4122 wattrset(view->win, A_NORMAL);
4123 waddstr(view->win, " ");
4124 col += strlen(commit->refs[i]->name) + STRING_SIZE("[] ");
4125 } while (commit->refs[i++]->next);
4126 }
4127
4128 if (type != LINE_CURSOR)
4129 wattrset(view->win, get_line_attr(type));
4130
4131 {
4132 int titlelen = strlen(commit->title);
4133
4134 if (col + titlelen > view->width)
4135 titlelen = view->width - col;
4136
4137 waddnstr(view->win, commit->title, titlelen);
4138 }
4139
4140 return TRUE;
4141 }
4142
4143 /* Reads git log --pretty=raw output and parses it into the commit struct. */
4144 static bool
4145 main_read(struct view *view, char *line)
4146 {
4147 static struct rev_graph *graph = graph_stacks;
4148 enum line_type type;
4149 struct commit *commit;
4150
4151 if (!line) {
4152 update_rev_graph(graph);
4153 return TRUE;
4154 }
4155
4156 type = get_line_type(line);
4157 if (type == LINE_COMMIT) {
4158 commit = calloc(1, sizeof(struct commit));
4159 if (!commit)
4160 return FALSE;
4161
4162 string_copy_rev(commit->id, line + STRING_SIZE("commit "));
4163 commit->refs = get_refs(commit->id);
4164 graph->commit = commit;
4165 add_line_data(view, commit, LINE_MAIN_COMMIT);
4166 return TRUE;
4167 }
4168
4169 if (!view->lines)
4170 return TRUE;
4171 commit = view->line[view->lines - 1].data;
4172
4173 switch (type) {
4174 case LINE_PARENT:
4175 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
4176 break;
4177
4178 case LINE_AUTHOR:
4179 {
4180 /* Parse author lines where the name may be empty:
4181 * author <email@address.tld> 1138474660 +0100
4182 */
4183 char *ident = line + STRING_SIZE("author ");
4184 char *nameend = strchr(ident, '<');
4185 char *emailend = strchr(ident, '>');
4186
4187 if (!nameend || !emailend)
4188 break;
4189
4190 update_rev_graph(graph);
4191 graph = graph->next;
4192
4193 *nameend = *emailend = 0;
4194 ident = chomp_string(ident);
4195 if (!*ident) {
4196 ident = chomp_string(nameend + 1);
4197 if (!*ident)
4198 ident = "Unknown";
4199 }
4200
4201 string_ncopy(commit->author, ident, strlen(ident));
4202
4203 /* Parse epoch and timezone */
4204 if (emailend[1] == ' ') {
4205 char *secs = emailend + 2;
4206 char *zone = strchr(secs, ' ');
4207 time_t time = (time_t) atol(secs);
4208
4209 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
4210 long tz;
4211
4212 zone++;
4213 tz = ('0' - zone[1]) * 60 * 60 * 10;
4214 tz += ('0' - zone[2]) * 60 * 60;
4215 tz += ('0' - zone[3]) * 60;
4216 tz += ('0' - zone[4]) * 60;
4217
4218 if (zone[0] == '-')
4219 tz = -tz;
4220
4221 time -= tz;
4222 }
4223
4224 gmtime_r(&time, &commit->time);
4225 }
4226 break;
4227 }
4228 default:
4229 /* Fill in the commit title if it has not already been set. */
4230 if (commit->title[0])
4231 break;
4232
4233 /* Require titles to start with a non-space character at the
4234 * offset used by git log. */
4235 if (strncmp(line, " ", 4))
4236 break;
4237 line += 4;
4238 /* Well, if the title starts with a whitespace character,
4239 * try to be forgiving. Otherwise we end up with no title. */
4240 while (isspace(*line))
4241 line++;
4242 if (*line == '\0')
4243 break;
4244 /* FIXME: More graceful handling of titles; append "..." to
4245 * shortened titles, etc. */
4246
4247 string_ncopy(commit->title, line, strlen(line));
4248 }
4249
4250 return TRUE;
4251 }
4252
4253 static enum request
4254 main_request(struct view *view, enum request request, struct line *line)
4255 {
4256 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4257
4258 if (request == REQ_ENTER)
4259 open_view(view, REQ_VIEW_DIFF, flags);
4260 else
4261 return request;
4262
4263 return REQ_NONE;
4264 }
4265
4266 static bool
4267 main_grep(struct view *view, struct line *line)
4268 {
4269 struct commit *commit = line->data;
4270 enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
4271 char buf[DATE_COLS + 1];
4272 regmatch_t pmatch;
4273
4274 for (state = S_TITLE; state < S_END; state++) {
4275 char *text;
4276
4277 switch (state) {
4278 case S_TITLE: text = commit->title; break;
4279 case S_AUTHOR: text = commit->author; break;
4280 case S_DATE:
4281 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
4282 continue;
4283 text = buf;
4284 break;
4285
4286 default:
4287 return FALSE;
4288 }
4289
4290 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4291 return TRUE;
4292 }
4293
4294 return FALSE;
4295 }
4296
4297 static void
4298 main_select(struct view *view, struct line *line)
4299 {
4300 struct commit *commit = line->data;
4301
4302 string_copy_rev(view->ref, commit->id);
4303 string_copy_rev(ref_commit, view->ref);
4304 }
4305
4306 static struct view_ops main_ops = {
4307 "commit",
4308 NULL,
4309 main_read,
4310 main_draw,
4311 main_request,
4312 main_grep,
4313 main_select,
4314 };
4315
4316
4317 /*
4318 * Unicode / UTF-8 handling
4319 *
4320 * NOTE: Much of the following code for dealing with unicode is derived from
4321 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
4322 * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
4323 */
4324
4325 /* I've (over)annotated a lot of code snippets because I am not entirely
4326 * confident that the approach taken by this small UTF-8 interface is correct.
4327 * --jonas */
4328
4329 static inline int
4330 unicode_width(unsigned long c)
4331 {
4332 if (c >= 0x1100 &&
4333 (c <= 0x115f /* Hangul Jamo */
4334 || c == 0x2329
4335 || c == 0x232a
4336 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
4337 /* CJK ... Yi */
4338 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
4339 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
4340 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
4341 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
4342 || (c >= 0xffe0 && c <= 0xffe6)
4343 || (c >= 0x20000 && c <= 0x2fffd)
4344 || (c >= 0x30000 && c <= 0x3fffd)))
4345 return 2;
4346
4347 return 1;
4348 }
4349
4350 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
4351 * Illegal bytes are set one. */
4352 static const unsigned char utf8_bytes[256] = {
4353 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,
4354 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,
4355 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,
4356 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,
4357 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,
4358 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,
4359 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,
4360 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,
4361 };
4362
4363 /* Decode UTF-8 multi-byte representation into a unicode character. */
4364 static inline unsigned long
4365 utf8_to_unicode(const char *string, size_t length)
4366 {
4367 unsigned long unicode;
4368
4369 switch (length) {
4370 case 1:
4371 unicode = string[0];
4372 break;
4373 case 2:
4374 unicode = (string[0] & 0x1f) << 6;
4375 unicode += (string[1] & 0x3f);
4376 break;
4377 case 3:
4378 unicode = (string[0] & 0x0f) << 12;
4379 unicode += ((string[1] & 0x3f) << 6);
4380 unicode += (string[2] & 0x3f);
4381 break;
4382 case 4:
4383 unicode = (string[0] & 0x0f) << 18;
4384 unicode += ((string[1] & 0x3f) << 12);
4385 unicode += ((string[2] & 0x3f) << 6);
4386 unicode += (string[3] & 0x3f);
4387 break;
4388 case 5:
4389 unicode = (string[0] & 0x0f) << 24;
4390 unicode += ((string[1] & 0x3f) << 18);
4391 unicode += ((string[2] & 0x3f) << 12);
4392 unicode += ((string[3] & 0x3f) << 6);
4393 unicode += (string[4] & 0x3f);
4394 break;
4395 case 6:
4396 unicode = (string[0] & 0x01) << 30;
4397 unicode += ((string[1] & 0x3f) << 24);
4398 unicode += ((string[2] & 0x3f) << 18);
4399 unicode += ((string[3] & 0x3f) << 12);
4400 unicode += ((string[4] & 0x3f) << 6);
4401 unicode += (string[5] & 0x3f);
4402 break;
4403 default:
4404 die("Invalid unicode length");
4405 }
4406
4407 /* Invalid characters could return the special 0xfffd value but NUL
4408 * should be just as good. */
4409 return unicode > 0xffff ? 0 : unicode;
4410 }
4411
4412 /* Calculates how much of string can be shown within the given maximum width
4413 * and sets trimmed parameter to non-zero value if all of string could not be
4414 * shown.
4415 *
4416 * Additionally, adds to coloffset how many many columns to move to align with
4417 * the expected position. Takes into account how multi-byte and double-width
4418 * characters will effect the cursor position.
4419 *
4420 * Returns the number of bytes to output from string to satisfy max_width. */
4421 static size_t
4422 utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed)
4423 {
4424 const char *start = string;
4425 const char *end = strchr(string, '\0');
4426 size_t mbwidth = 0;
4427 size_t width = 0;
4428
4429 *trimmed = 0;
4430
4431 while (string < end) {
4432 int c = *(unsigned char *) string;
4433 unsigned char bytes = utf8_bytes[c];
4434 size_t ucwidth;
4435 unsigned long unicode;
4436
4437 if (string + bytes > end)
4438 break;
4439
4440 /* Change representation to figure out whether
4441 * it is a single- or double-width character. */
4442
4443 unicode = utf8_to_unicode(string, bytes);
4444 /* FIXME: Graceful handling of invalid unicode character. */
4445 if (!unicode)
4446 break;
4447
4448 ucwidth = unicode_width(unicode);
4449 width += ucwidth;
4450 if (width > max_width) {
4451 *trimmed = 1;
4452 break;
4453 }
4454
4455 /* The column offset collects the differences between the
4456 * number of bytes encoding a character and the number of
4457 * columns will be used for rendering said character.
4458 *
4459 * So if some character A is encoded in 2 bytes, but will be
4460 * represented on the screen using only 1 byte this will and up
4461 * adding 1 to the multi-byte column offset.
4462 *
4463 * Assumes that no double-width character can be encoding in
4464 * less than two bytes. */
4465 if (bytes > ucwidth)
4466 mbwidth += bytes - ucwidth;
4467
4468 string += bytes;
4469 }
4470
4471 *coloffset += mbwidth;
4472
4473 return string - start;
4474 }
4475
4476
4477 /*
4478 * Status management
4479 */
4480
4481 /* Whether or not the curses interface has been initialized. */
4482 static bool cursed = FALSE;
4483
4484 /* The status window is used for polling keystrokes. */
4485 static WINDOW *status_win;
4486
4487 static bool status_empty = TRUE;
4488
4489 /* Update status and title window. */
4490 static void
4491 report(const char *msg, ...)
4492 {
4493 struct view *view = display[current_view];
4494
4495 if (input_mode)
4496 return;
4497
4498 if (!view) {
4499 char buf[SIZEOF_STR];
4500 va_list args;
4501
4502 va_start(args, msg);
4503 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
4504 buf[sizeof(buf) - 1] = 0;
4505 buf[sizeof(buf) - 2] = '.';
4506 buf[sizeof(buf) - 3] = '.';
4507 buf[sizeof(buf) - 4] = '.';
4508 }
4509 va_end(args);
4510 die("%s", buf);
4511 }
4512
4513 if (!status_empty || *msg) {
4514 va_list args;
4515
4516 va_start(args, msg);
4517
4518 wmove(status_win, 0, 0);
4519 if (*msg) {
4520 vwprintw(status_win, msg, args);
4521 status_empty = FALSE;
4522 } else {
4523 status_empty = TRUE;
4524 }
4525 wclrtoeol(status_win);
4526 wrefresh(status_win);
4527
4528 va_end(args);
4529 }
4530
4531 update_view_title(view);
4532 update_display_cursor(view);
4533 }
4534
4535 /* Controls when nodelay should be in effect when polling user input. */
4536 static void
4537 set_nonblocking_input(bool loading)
4538 {
4539 static unsigned int loading_views;
4540
4541 if ((loading == FALSE && loading_views-- == 1) ||
4542 (loading == TRUE && loading_views++ == 0))
4543 nodelay(status_win, loading);
4544 }
4545
4546 static void
4547 init_display(void)
4548 {
4549 int x, y;
4550
4551 /* Initialize the curses library */
4552 if (isatty(STDIN_FILENO)) {
4553 cursed = !!initscr();
4554 } else {
4555 /* Leave stdin and stdout alone when acting as a pager. */
4556 FILE *io = fopen("/dev/tty", "r+");
4557
4558 if (!io)
4559 die("Failed to open /dev/tty");
4560 cursed = !!newterm(NULL, io, io);
4561 }
4562
4563 if (!cursed)
4564 die("Failed to initialize curses");
4565
4566 nonl(); /* Tell curses not to do NL->CR/NL on output */
4567 cbreak(); /* Take input chars one at a time, no wait for \n */
4568 noecho(); /* Don't echo input */
4569 leaveok(stdscr, TRUE);
4570
4571 if (has_colors())
4572 init_colors();
4573
4574 getmaxyx(stdscr, y, x);
4575 status_win = newwin(1, 0, y - 1, 0);
4576 if (!status_win)
4577 die("Failed to create status window");
4578
4579 /* Enable keyboard mapping */
4580 keypad(status_win, TRUE);
4581 wbkgdset(status_win, get_line_attr(LINE_STATUS));
4582 }
4583
4584 static char *
4585 read_prompt(const char *prompt)
4586 {
4587 enum { READING, STOP, CANCEL } status = READING;
4588 static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
4589 int pos = 0;
4590
4591 while (status == READING) {
4592 struct view *view;
4593 int i, key;
4594
4595 input_mode = TRUE;
4596
4597 foreach_view (view, i)
4598 update_view(view);
4599
4600 input_mode = FALSE;
4601
4602 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
4603 wclrtoeol(status_win);
4604
4605 /* Refresh, accept single keystroke of input */
4606 key = wgetch(status_win);
4607 switch (key) {
4608 case KEY_RETURN:
4609 case KEY_ENTER:
4610 case '\n':
4611 status = pos ? STOP : CANCEL;
4612 break;
4613
4614 case KEY_BACKSPACE:
4615 if (pos > 0)
4616 pos--;
4617 else
4618 status = CANCEL;
4619 break;
4620
4621 case KEY_ESC:
4622 status = CANCEL;
4623 break;
4624
4625 case ERR:
4626 break;
4627
4628 default:
4629 if (pos >= sizeof(buf)) {
4630 report("Input string too long");
4631 return NULL;
4632 }
4633
4634 if (isprint(key))
4635 buf[pos++] = (char) key;
4636 }
4637 }
4638
4639 /* Clear the status window */
4640 status_empty = FALSE;
4641 report("");
4642
4643 if (status == CANCEL)
4644 return NULL;
4645
4646 buf[pos++] = 0;
4647
4648 return buf;
4649 }
4650
4651 /*
4652 * Repository references
4653 */
4654
4655 static struct ref *refs;
4656 static size_t refs_size;
4657
4658 /* Id <-> ref store */
4659 static struct ref ***id_refs;
4660 static size_t id_refs_size;
4661
4662 static struct ref **
4663 get_refs(char *id)
4664 {
4665 struct ref ***tmp_id_refs;
4666 struct ref **ref_list = NULL;
4667 size_t ref_list_size = 0;
4668 size_t i;
4669
4670 for (i = 0; i < id_refs_size; i++)
4671 if (!strcmp(id, id_refs[i][0]->id))
4672 return id_refs[i];
4673
4674 tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
4675 if (!tmp_id_refs)
4676 return NULL;
4677
4678 id_refs = tmp_id_refs;
4679
4680 for (i = 0; i < refs_size; i++) {
4681 struct ref **tmp;
4682
4683 if (strcmp(id, refs[i].id))
4684 continue;
4685
4686 tmp = realloc(ref_list, (ref_list_size + 1) * sizeof(*ref_list));
4687 if (!tmp) {
4688 if (ref_list)
4689 free(ref_list);
4690 return NULL;
4691 }
4692
4693 ref_list = tmp;
4694 if (ref_list_size > 0)
4695 ref_list[ref_list_size - 1]->next = 1;
4696 ref_list[ref_list_size] = &refs[i];
4697
4698 /* XXX: The properties of the commit chains ensures that we can
4699 * safely modify the shared ref. The repo references will
4700 * always be similar for the same id. */
4701 ref_list[ref_list_size]->next = 0;
4702 ref_list_size++;
4703 }
4704
4705 if (ref_list)
4706 id_refs[id_refs_size++] = ref_list;
4707
4708 return ref_list;
4709 }
4710
4711 static int
4712 read_ref(char *id, size_t idlen, char *name, size_t namelen)
4713 {
4714 struct ref *ref;
4715 bool tag = FALSE;
4716 bool remote = FALSE;
4717
4718 if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
4719 /* Commits referenced by tags has "^{}" appended. */
4720 if (name[namelen - 1] != '}')
4721 return OK;
4722
4723 while (namelen > 0 && name[namelen] != '^')
4724 namelen--;
4725
4726 tag = TRUE;
4727 namelen -= STRING_SIZE("refs/tags/");
4728 name += STRING_SIZE("refs/tags/");
4729
4730 } else if (!strncmp(name, "refs/remotes/", STRING_SIZE("refs/remotes/"))) {
4731 remote = TRUE;
4732 namelen -= STRING_SIZE("refs/remotes/");
4733 name += STRING_SIZE("refs/remotes/");
4734
4735 } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
4736 namelen -= STRING_SIZE("refs/heads/");
4737 name += STRING_SIZE("refs/heads/");
4738
4739 } else if (!strcmp(name, "HEAD")) {
4740 return OK;
4741 }
4742
4743 refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
4744 if (!refs)
4745 return ERR;
4746
4747 ref = &refs[refs_size++];
4748 ref->name = malloc(namelen + 1);
4749 if (!ref->name)
4750 return ERR;
4751
4752 strncpy(ref->name, name, namelen);
4753 ref->name[namelen] = 0;
4754 ref->tag = tag;
4755 ref->remote = remote;
4756 string_copy_rev(ref->id, id);
4757
4758 return OK;
4759 }
4760
4761 static int
4762 load_refs(void)
4763 {
4764 const char *cmd_env = getenv("TIG_LS_REMOTE");
4765 const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
4766
4767 return read_properties(popen(cmd, "r"), "\t", read_ref);
4768 }
4769
4770 static int
4771 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
4772 {
4773 if (!strcmp(name, "i18n.commitencoding"))
4774 string_ncopy(opt_encoding, value, valuelen);
4775
4776 if (!strcmp(name, "core.editor"))
4777 string_ncopy(opt_editor, value, valuelen);
4778
4779 return OK;
4780 }
4781
4782 static int
4783 load_repo_config(void)
4784 {
4785 return read_properties(popen(GIT_CONFIG " --list", "r"),
4786 "=", read_repo_config_option);
4787 }
4788
4789 static int
4790 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
4791 {
4792 if (!opt_git_dir[0]) {
4793 string_ncopy(opt_git_dir, name, namelen);
4794
4795 } else if (opt_is_inside_work_tree == -1) {
4796 /* This can be 3 different values depending on the
4797 * version of git being used. If git-rev-parse does not
4798 * understand --is-inside-work-tree it will simply echo
4799 * the option else either "true" or "false" is printed.
4800 * Default to true for the unknown case. */
4801 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
4802
4803 } else {
4804 string_ncopy(opt_cdup, name, namelen);
4805 }
4806
4807 return OK;
4808 }
4809
4810 /* XXX: The line outputted by "--show-cdup" can be empty so the option
4811 * must be the last one! */
4812 static int
4813 load_repo_info(void)
4814 {
4815 return read_properties(popen("git rev-parse --git-dir --is-inside-work-tree --show-cdup 2>/dev/null", "r"),
4816 "=", read_repo_info);
4817 }
4818
4819 static int
4820 read_properties(FILE *pipe, const char *separators,
4821 int (*read_property)(char *, size_t, char *, size_t))
4822 {
4823 char buffer[BUFSIZ];
4824 char *name;
4825 int state = OK;
4826
4827 if (!pipe)
4828 return ERR;
4829
4830 while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
4831 char *value;
4832 size_t namelen;
4833 size_t valuelen;
4834
4835 name = chomp_string(name);
4836 namelen = strcspn(name, separators);
4837
4838 if (name[namelen]) {
4839 name[namelen] = 0;
4840 value = chomp_string(name + namelen + 1);
4841 valuelen = strlen(value);
4842
4843 } else {
4844 value = "";
4845 valuelen = 0;
4846 }
4847
4848 state = read_property(name, namelen, value, valuelen);
4849 }
4850
4851 if (state != ERR && ferror(pipe))
4852 state = ERR;
4853
4854 pclose(pipe);
4855
4856 return state;
4857 }
4858
4859
4860 /*
4861 * Main
4862 */
4863
4864 static void __NORETURN
4865 quit(int sig)
4866 {
4867 /* XXX: Restore tty modes and let the OS cleanup the rest! */
4868 if (cursed)
4869 endwin();
4870 exit(0);
4871 }
4872
4873 static void __NORETURN
4874 die(const char *err, ...)
4875 {
4876 va_list args;
4877
4878 endwin();
4879
4880 va_start(args, err);
4881 fputs("tig: ", stderr);
4882 vfprintf(stderr, err, args);
4883 fputs("\n", stderr);
4884 va_end(args);
4885
4886 exit(1);
4887 }
4888
4889 int
4890 main(int argc, char *argv[])
4891 {
4892 struct view *view;
4893 enum request request;
4894 size_t i;
4895
4896 signal(SIGINT, quit);
4897
4898 if (setlocale(LC_ALL, "")) {
4899 char *codeset = nl_langinfo(CODESET);
4900
4901 string_ncopy(opt_codeset, codeset, strlen(codeset));
4902 }
4903
4904 if (load_repo_info() == ERR)
4905 die("Failed to load repo info.");
4906
4907 if (load_options() == ERR)
4908 die("Failed to load user config.");
4909
4910 /* Load the repo config file so options can be overwritten from
4911 * the command line. */
4912 if (load_repo_config() == ERR)
4913 die("Failed to load repo config.");
4914
4915 if (!parse_options(argc, argv))
4916 return 0;
4917
4918 /* Require a git repository unless when running in pager mode. */
4919 if (!opt_git_dir[0])
4920 die("Not a git repository");
4921
4922 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
4923 opt_iconv = iconv_open(opt_codeset, opt_encoding);
4924 if (opt_iconv == ICONV_NONE)
4925 die("Failed to initialize character set conversion");
4926 }
4927
4928 if (load_refs() == ERR)
4929 die("Failed to load refs.");
4930
4931 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
4932 view->cmd_env = getenv(view->cmd_env);
4933
4934 request = opt_request;
4935
4936 init_display();
4937
4938 while (view_driver(display[current_view], request)) {
4939 int key;
4940 int i;
4941
4942 foreach_view (view, i)
4943 update_view(view);
4944
4945 /* Refresh, accept single keystroke of input */
4946 key = wgetch(status_win);
4947
4948 /* wgetch() with nodelay() enabled returns ERR when there's no
4949 * input. */
4950 if (key == ERR) {
4951 request = REQ_NONE;
4952 continue;
4953 }
4954
4955 request = get_keybinding(display[current_view]->keymap, key);
4956
4957 /* Some low-level request handling. This keeps access to
4958 * status_win restricted. */
4959 switch (request) {
4960 case REQ_PROMPT:
4961 {
4962 char *cmd = read_prompt(":");
4963
4964 if (cmd && string_format(opt_cmd, "git %s", cmd)) {
4965 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
4966 opt_request = REQ_VIEW_DIFF;
4967 } else {
4968 opt_request = REQ_VIEW_PAGER;
4969 }
4970 break;
4971 }
4972
4973 request = REQ_NONE;
4974 break;
4975 }
4976 case REQ_SEARCH:
4977 case REQ_SEARCH_BACK:
4978 {
4979 const char *prompt = request == REQ_SEARCH
4980 ? "/" : "?";
4981 char *search = read_prompt(prompt);
4982
4983 if (search)
4984 string_ncopy(opt_search, search, strlen(search));
4985 else
4986 request = REQ_NONE;
4987 break;
4988 }
4989 case REQ_SCREEN_RESIZE:
4990 {
4991 int height, width;
4992
4993 getmaxyx(stdscr, height, width);
4994
4995 /* Resize the status view and let the view driver take
4996 * care of resizing the displayed views. */
4997 wresize(status_win, 1, width);
4998 mvwin(status_win, height - 1, 0);
4999 wrefresh(status_win);
5000 break;
5001 }
5002 default:
5003 break;
5004 }
5005 }
5006
5007 quit(0);
5008
5009 return 0;
5010 }