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