~mdw / tig / blob
? search:


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