Fix tree viewing again; introduced by string safety patch
[tig] / tig.c
1 /* Copyright (c) 2006 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 #ifndef VERSION
15 #define VERSION "tig-0.4.git"
16 #endif
17
18 #ifndef DEBUG
19 #define NDEBUG
20 #endif
21
22 #include <assert.h>
23 #include <errno.h>
24 #include <ctype.h>
25 #include <signal.h>
26 #include <stdarg.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <unistd.h>
31 #include <time.h>
32
33 #include <sys/types.h>
34 #include <regex.h>
35
36 #include <locale.h>
37 #include <langinfo.h>
38 #include <iconv.h>
39
40 #include <curses.h>
41
42 #if __GNUC__ >= 3
43 #define __NORETURN __attribute__((__noreturn__))
44 #else
45 #define __NORETURN
46 #endif
47
48 static void __NORETURN die(const char *err, ...);
49 static void report(const char *msg, ...);
50 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, int, char *, int));
51 static void set_nonblocking_input(bool loading);
52 static size_t utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed);
53
54 #define ABS(x) ((x) >= 0 ? (x) : -(x))
55 #define MIN(x, y) ((x) < (y) ? (x) : (y))
56
57 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
58 #define STRING_SIZE(x) (sizeof(x) - 1)
59
60 #define SIZEOF_STR 1024 /* Default string size. */
61 #define SIZEOF_REF 256 /* Size of symbolic or SHA1 ID. */
62 #define SIZEOF_REVGRAPH 19 /* Size of revision ancestry graphics. */
63
64 /* This color name can be used to refer to the default term colors. */
65 #define COLOR_DEFAULT (-1)
66
67 #define ICONV_NONE ((iconv_t) -1)
68
69 /* The format and size of the date column in the main view. */
70 #define DATE_FORMAT "%Y-%m-%d %H:%M"
71 #define DATE_COLS STRING_SIZE("2006-04-29 14:21 ")
72
73 #define AUTHOR_COLS 20
74
75 /* The default interval between line numbers. */
76 #define NUMBER_INTERVAL 1
77
78 #define TABSIZE 8
79
80 #define SCALE_SPLIT_VIEW(height) ((height) * 2 / 3)
81
82 #define TIG_LS_REMOTE \
83 "git ls-remote . 2>/dev/null"
84
85 #define TIG_DIFF_CMD \
86 "git show --root --patch-with-stat --find-copies-harder -B -C %s 2>/dev/null"
87
88 #define TIG_LOG_CMD \
89 "git log --cc --stat -n100 %s 2>/dev/null"
90
91 #define TIG_MAIN_CMD \
92 "git log --topo-order --pretty=raw %s 2>/dev/null"
93
94 #define TIG_TREE_CMD \
95 "git ls-tree %s %s"
96
97 #define TIG_BLOB_CMD \
98 "git cat-file blob %s"
99
100 /* XXX: Needs to be defined to the empty string. */
101 #define TIG_HELP_CMD ""
102 #define TIG_PAGER_CMD ""
103
104 /* Some ascii-shorthands fitted into the ncurses namespace. */
105 #define KEY_TAB '\t'
106 #define KEY_RETURN '\r'
107 #define KEY_ESC 27
108
109
110 struct ref {
111 char *name; /* Ref name; tag or head names are shortened. */
112 char id[41]; /* Commit SHA1 ID */
113 unsigned int tag:1; /* Is it a tag? */
114 unsigned int next:1; /* For ref lists: are there more refs? */
115 };
116
117 static struct ref **get_refs(char *id);
118
119 struct int_map {
120 const char *name;
121 int namelen;
122 int value;
123 };
124
125 static int
126 set_from_int_map(struct int_map *map, size_t map_size,
127 int *value, const char *name, int namelen)
128 {
129
130 int i;
131
132 for (i = 0; i < map_size; i++)
133 if (namelen == map[i].namelen &&
134 !strncasecmp(name, map[i].name, namelen)) {
135 *value = map[i].value;
136 return OK;
137 }
138
139 return ERR;
140 }
141
142
143 /*
144 * String helpers
145 */
146
147 static inline void
148 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
149 {
150 if (srclen > dstlen - 1)
151 srclen = dstlen - 1;
152
153 strncpy(dst, src, srclen);
154 dst[srclen] = 0;
155 }
156
157 /* Shorthands for safely copying into a fixed buffer. */
158
159 #define string_copy(dst, src) \
160 string_ncopy_do(dst, sizeof(dst), src, sizeof(dst))
161
162 #define string_ncopy(dst, src, srclen) \
163 string_ncopy_do(dst, sizeof(dst), src, srclen)
164
165 static char *
166 chomp_string(char *name)
167 {
168 int namelen;
169
170 while (isspace(*name))
171 name++;
172
173 namelen = strlen(name) - 1;
174 while (namelen > 0 && isspace(name[namelen]))
175 name[namelen--] = 0;
176
177 return name;
178 }
179
180 static bool
181 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
182 {
183 va_list args;
184 size_t pos = bufpos ? *bufpos : 0;
185
186 va_start(args, fmt);
187 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
188 va_end(args);
189
190 if (bufpos)
191 *bufpos = pos;
192
193 return pos >= bufsize ? FALSE : TRUE;
194 }
195
196 #define string_format(buf, fmt, args...) \
197 string_nformat(buf, sizeof(buf), NULL, fmt, args)
198
199 #define string_format_from(buf, from, fmt, args...) \
200 string_nformat(buf, sizeof(buf), from, fmt, args)
201
202 static int
203 string_enum_compare(const char *str1, const char *str2, int len)
204 {
205 size_t i;
206
207 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
208
209 /* Diff-Header == DIFF_HEADER */
210 for (i = 0; i < len; i++) {
211 if (toupper(str1[i]) == toupper(str2[i]))
212 continue;
213
214 if (string_enum_sep(str1[i]) &&
215 string_enum_sep(str2[i]))
216 continue;
217
218 return str1[i] - str2[i];
219 }
220
221 return 0;
222 }
223
224 /* Shell quoting
225 *
226 * NOTE: The following is a slightly modified copy of the git project's shell
227 * quoting routines found in the quote.c file.
228 *
229 * Help to copy the thing properly quoted for the shell safety. any single
230 * quote is replaced with '\'', any exclamation point is replaced with '\!',
231 * and the whole thing is enclosed in a
232 *
233 * E.g.
234 * original sq_quote result
235 * name ==> name ==> 'name'
236 * a b ==> a b ==> 'a b'
237 * a'b ==> a'\''b ==> 'a'\''b'
238 * a!b ==> a'\!'b ==> 'a'\!'b'
239 */
240
241 static size_t
242 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
243 {
244 char c;
245
246 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
247
248 BUFPUT('\'');
249 while ((c = *src++)) {
250 if (c == '\'' || c == '!') {
251 BUFPUT('\'');
252 BUFPUT('\\');
253 BUFPUT(c);
254 BUFPUT('\'');
255 } else {
256 BUFPUT(c);
257 }
258 }
259 BUFPUT('\'');
260
261 return bufsize;
262 }
263
264
265 /*
266 * User requests
267 */
268
269 #define REQ_INFO \
270 /* XXX: Keep the view request first and in sync with views[]. */ \
271 REQ_GROUP("View switching") \
272 REQ_(VIEW_MAIN, "Show main view"), \
273 REQ_(VIEW_DIFF, "Show diff view"), \
274 REQ_(VIEW_LOG, "Show log view"), \
275 REQ_(VIEW_TREE, "Show tree view"), \
276 REQ_(VIEW_BLOB, "Show blob view"), \
277 REQ_(VIEW_HELP, "Show help page"), \
278 REQ_(VIEW_PAGER, "Show pager view"), \
279 \
280 REQ_GROUP("View manipulation") \
281 REQ_(ENTER, "Enter current line and scroll"), \
282 REQ_(NEXT, "Move to next"), \
283 REQ_(PREVIOUS, "Move to previous"), \
284 REQ_(VIEW_NEXT, "Move focus to next view"), \
285 REQ_(VIEW_CLOSE, "Close the current view"), \
286 REQ_(QUIT, "Close all views and quit"), \
287 \
288 REQ_GROUP("Cursor navigation") \
289 REQ_(MOVE_UP, "Move cursor one line up"), \
290 REQ_(MOVE_DOWN, "Move cursor one line down"), \
291 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
292 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
293 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
294 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
295 \
296 REQ_GROUP("Scrolling") \
297 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
298 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
299 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
300 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
301 \
302 REQ_GROUP("Searching") \
303 REQ_(SEARCH, "Search the view"), \
304 REQ_(SEARCH_BACK, "Search backwards in the view"), \
305 REQ_(FIND_NEXT, "Find next search match"), \
306 REQ_(FIND_PREV, "Find previous search match"), \
307 \
308 REQ_GROUP("Misc") \
309 REQ_(NONE, "Do nothing"), \
310 REQ_(PROMPT, "Bring up the prompt"), \
311 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
312 REQ_(SCREEN_RESIZE, "Resize the screen"), \
313 REQ_(SHOW_VERSION, "Show version information"), \
314 REQ_(STOP_LOADING, "Stop all loading views"), \
315 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
316 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization")
317
318
319 /* User action requests. */
320 enum request {
321 #define REQ_GROUP(help)
322 #define REQ_(req, help) REQ_##req
323
324 /* Offset all requests to avoid conflicts with ncurses getch values. */
325 REQ_OFFSET = KEY_MAX + 1,
326 REQ_INFO,
327 REQ_UNKNOWN,
328
329 #undef REQ_GROUP
330 #undef REQ_
331 };
332
333 struct request_info {
334 enum request request;
335 char *name;
336 int namelen;
337 char *help;
338 };
339
340 static struct request_info req_info[] = {
341 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
342 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
343 REQ_INFO
344 #undef REQ_GROUP
345 #undef REQ_
346 };
347
348 static enum request
349 get_request(const char *name)
350 {
351 int namelen = strlen(name);
352 int i;
353
354 for (i = 0; i < ARRAY_SIZE(req_info); i++)
355 if (req_info[i].namelen == namelen &&
356 !string_enum_compare(req_info[i].name, name, namelen))
357 return req_info[i].request;
358
359 return REQ_UNKNOWN;
360 }
361
362
363 /*
364 * Options
365 */
366
367 static const char usage[] =
368 VERSION " (" __DATE__ ")\n"
369 "\n"
370 "Usage: tig [options]\n"
371 " or: tig [options] [--] [git log options]\n"
372 " or: tig [options] log [git log options]\n"
373 " or: tig [options] diff [git diff options]\n"
374 " or: tig [options] show [git show options]\n"
375 " or: tig [options] < [git command output]\n"
376 "\n"
377 "Options:\n"
378 " -l Start up in log view\n"
379 " -d Start up in diff view\n"
380 " -n[I], --line-number[=I] Show line numbers with given interval\n"
381 " -b[N], --tab-size[=N] Set number of spaces for tab expansion\n"
382 " -- Mark end of tig options\n"
383 " -v, --version Show version and exit\n"
384 " -h, --help Show help message and exit\n";
385
386 /* Option and state variables. */
387 static bool opt_line_number = FALSE;
388 static bool opt_rev_graph = TRUE;
389 static int opt_num_interval = NUMBER_INTERVAL;
390 static int opt_tab_size = TABSIZE;
391 static enum request opt_request = REQ_VIEW_MAIN;
392 static char opt_cmd[SIZEOF_STR] = "";
393 static char opt_path[SIZEOF_STR] = "";
394 static FILE *opt_pipe = NULL;
395 static char opt_encoding[20] = "UTF-8";
396 static bool opt_utf8 = TRUE;
397 static char opt_codeset[20] = "UTF-8";
398 static iconv_t opt_iconv = ICONV_NONE;
399 static char opt_search[SIZEOF_STR] = "";
400
401 enum option_type {
402 OPT_NONE,
403 OPT_INT,
404 };
405
406 static bool
407 check_option(char *opt, char short_name, char *name, enum option_type type, ...)
408 {
409 va_list args;
410 char *value = "";
411 int *number;
412
413 if (opt[0] != '-')
414 return FALSE;
415
416 if (opt[1] == '-') {
417 int namelen = strlen(name);
418
419 opt += 2;
420
421 if (strncmp(opt, name, namelen))
422 return FALSE;
423
424 if (opt[namelen] == '=')
425 value = opt + namelen + 1;
426
427 } else {
428 if (!short_name || opt[1] != short_name)
429 return FALSE;
430 value = opt + 2;
431 }
432
433 va_start(args, type);
434 if (type == OPT_INT) {
435 number = va_arg(args, int *);
436 if (isdigit(*value))
437 *number = atoi(value);
438 }
439 va_end(args);
440
441 return TRUE;
442 }
443
444 /* Returns the index of log or diff command or -1 to exit. */
445 static bool
446 parse_options(int argc, char *argv[])
447 {
448 int i;
449
450 for (i = 1; i < argc; i++) {
451 char *opt = argv[i];
452
453 if (!strcmp(opt, "-l")) {
454 opt_request = REQ_VIEW_LOG;
455 continue;
456 }
457
458 if (!strcmp(opt, "-d")) {
459 opt_request = REQ_VIEW_DIFF;
460 continue;
461 }
462
463 if (check_option(opt, 'n', "line-number", OPT_INT, &opt_num_interval)) {
464 opt_line_number = TRUE;
465 continue;
466 }
467
468 if (check_option(opt, 'b', "tab-size", OPT_INT, &opt_tab_size)) {
469 opt_tab_size = MIN(opt_tab_size, TABSIZE);
470 continue;
471 }
472
473 if (check_option(opt, 'v', "version", OPT_NONE)) {
474 printf("tig version %s\n", VERSION);
475 return FALSE;
476 }
477
478 if (check_option(opt, 'h', "help", OPT_NONE)) {
479 printf(usage);
480 return FALSE;
481 }
482
483 if (!strcmp(opt, "--")) {
484 i++;
485 break;
486 }
487
488 if (!strcmp(opt, "log") ||
489 !strcmp(opt, "diff") ||
490 !strcmp(opt, "show")) {
491 opt_request = opt[0] == 'l'
492 ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
493 break;
494 }
495
496 if (opt[0] && opt[0] != '-')
497 break;
498
499 die("unknown option '%s'\n\n%s", opt, usage);
500 }
501
502 if (!isatty(STDIN_FILENO)) {
503 opt_request = REQ_VIEW_PAGER;
504 opt_pipe = stdin;
505
506 } else if (i < argc) {
507 size_t buf_size;
508
509 if (opt_request == REQ_VIEW_MAIN)
510 /* XXX: This is vulnerable to the user overriding
511 * options required for the main view parser. */
512 string_copy(opt_cmd, "git log --stat --pretty=raw");
513 else
514 string_copy(opt_cmd, "git");
515 buf_size = strlen(opt_cmd);
516
517 while (buf_size < sizeof(opt_cmd) && i < argc) {
518 opt_cmd[buf_size++] = ' ';
519 buf_size = sq_quote(opt_cmd, buf_size, argv[i++]);
520 }
521
522 if (buf_size >= sizeof(opt_cmd))
523 die("command too long");
524
525 opt_cmd[buf_size] = 0;
526
527 }
528
529 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
530 opt_utf8 = FALSE;
531
532 return TRUE;
533 }
534
535
536 /*
537 * Line-oriented content detection.
538 */
539
540 #define LINE_INFO \
541 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
542 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
543 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
544 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
545 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
546 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
547 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
548 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
549 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
550 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
551 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
552 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
553 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
554 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
555 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
556 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
557 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
558 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
559 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
560 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
561 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
562 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
563 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
564 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
565 LINE(AUTHOR, "author ", COLOR_CYAN, COLOR_DEFAULT, 0), \
566 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
567 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
568 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
569 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
570 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
571 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
572 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
573 LINE(MAIN_DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
574 LINE(MAIN_AUTHOR, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
575 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
576 LINE(MAIN_DELIM, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
577 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
578 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
579 LINE(TREE_DIR, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
580 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL)
581
582 enum line_type {
583 #define LINE(type, line, fg, bg, attr) \
584 LINE_##type
585 LINE_INFO
586 #undef LINE
587 };
588
589 struct line_info {
590 const char *name; /* Option name. */
591 int namelen; /* Size of option name. */
592 const char *line; /* The start of line to match. */
593 int linelen; /* Size of string to match. */
594 int fg, bg, attr; /* Color and text attributes for the lines. */
595 };
596
597 static struct line_info line_info[] = {
598 #define LINE(type, line, fg, bg, attr) \
599 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
600 LINE_INFO
601 #undef LINE
602 };
603
604 static enum line_type
605 get_line_type(char *line)
606 {
607 int linelen = strlen(line);
608 enum line_type type;
609
610 for (type = 0; type < ARRAY_SIZE(line_info); type++)
611 /* Case insensitive search matches Signed-off-by lines better. */
612 if (linelen >= line_info[type].linelen &&
613 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
614 return type;
615
616 return LINE_DEFAULT;
617 }
618
619 static inline int
620 get_line_attr(enum line_type type)
621 {
622 assert(type < ARRAY_SIZE(line_info));
623 return COLOR_PAIR(type) | line_info[type].attr;
624 }
625
626 static struct line_info *
627 get_line_info(char *name, int namelen)
628 {
629 enum line_type type;
630
631 for (type = 0; type < ARRAY_SIZE(line_info); type++)
632 if (namelen == line_info[type].namelen &&
633 !string_enum_compare(line_info[type].name, name, namelen))
634 return &line_info[type];
635
636 return NULL;
637 }
638
639 static void
640 init_colors(void)
641 {
642 int default_bg = COLOR_BLACK;
643 int default_fg = COLOR_WHITE;
644 enum line_type type;
645
646 start_color();
647
648 if (use_default_colors() != ERR) {
649 default_bg = -1;
650 default_fg = -1;
651 }
652
653 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
654 struct line_info *info = &line_info[type];
655 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
656 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
657
658 init_pair(type, fg, bg);
659 }
660 }
661
662 struct line {
663 enum line_type type;
664 void *data; /* User data */
665 };
666
667
668 /*
669 * Keys
670 */
671
672 struct keybinding {
673 int alias;
674 enum request request;
675 struct keybinding *next;
676 };
677
678 static struct keybinding default_keybindings[] = {
679 /* View switching */
680 { 'm', REQ_VIEW_MAIN },
681 { 'd', REQ_VIEW_DIFF },
682 { 'l', REQ_VIEW_LOG },
683 { 't', REQ_VIEW_TREE },
684 { 'f', REQ_VIEW_BLOB },
685 { 'p', REQ_VIEW_PAGER },
686 { 'h', REQ_VIEW_HELP },
687
688 /* View manipulation */
689 { 'q', REQ_VIEW_CLOSE },
690 { KEY_TAB, REQ_VIEW_NEXT },
691 { KEY_RETURN, REQ_ENTER },
692 { KEY_UP, REQ_PREVIOUS },
693 { KEY_DOWN, REQ_NEXT },
694
695 /* Cursor navigation */
696 { 'k', REQ_MOVE_UP },
697 { 'j', REQ_MOVE_DOWN },
698 { KEY_HOME, REQ_MOVE_FIRST_LINE },
699 { KEY_END, REQ_MOVE_LAST_LINE },
700 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
701 { ' ', REQ_MOVE_PAGE_DOWN },
702 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
703 { 'b', REQ_MOVE_PAGE_UP },
704 { '-', REQ_MOVE_PAGE_UP },
705
706 /* Scrolling */
707 { KEY_IC, REQ_SCROLL_LINE_UP },
708 { KEY_DC, REQ_SCROLL_LINE_DOWN },
709 { 'w', REQ_SCROLL_PAGE_UP },
710 { 's', REQ_SCROLL_PAGE_DOWN },
711
712 /* Searching */
713 { '/', REQ_SEARCH },
714 { '?', REQ_SEARCH_BACK },
715 { 'n', REQ_FIND_NEXT },
716 { 'N', REQ_FIND_PREV },
717
718 /* Misc */
719 { 'Q', REQ_QUIT },
720 { 'z', REQ_STOP_LOADING },
721 { 'v', REQ_SHOW_VERSION },
722 { 'r', REQ_SCREEN_REDRAW },
723 { '.', REQ_TOGGLE_LINENO },
724 { 'g', REQ_TOGGLE_REV_GRAPH },
725 { ':', REQ_PROMPT },
726
727 /* wgetch() with nodelay() enabled returns ERR when there's no input. */
728 { ERR, REQ_NONE },
729
730 /* Using the ncurses SIGWINCH handler. */
731 { KEY_RESIZE, REQ_SCREEN_RESIZE },
732 };
733
734 #define KEYMAP_INFO \
735 KEYMAP_(GENERIC), \
736 KEYMAP_(MAIN), \
737 KEYMAP_(DIFF), \
738 KEYMAP_(LOG), \
739 KEYMAP_(TREE), \
740 KEYMAP_(BLOB), \
741 KEYMAP_(PAGER), \
742 KEYMAP_(HELP) \
743
744 enum keymap {
745 #define KEYMAP_(name) KEYMAP_##name
746 KEYMAP_INFO
747 #undef KEYMAP_
748 };
749
750 static struct int_map keymap_table[] = {
751 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
752 KEYMAP_INFO
753 #undef KEYMAP_
754 };
755
756 #define set_keymap(map, name) \
757 set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
758
759 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
760
761 static void
762 add_keybinding(enum keymap keymap, enum request request, int key)
763 {
764 struct keybinding *keybinding;
765
766 keybinding = calloc(1, sizeof(*keybinding));
767 if (!keybinding)
768 die("Failed to allocate keybinding");
769
770 keybinding->alias = key;
771 keybinding->request = request;
772 keybinding->next = keybindings[keymap];
773 keybindings[keymap] = keybinding;
774 }
775
776 /* Looks for a key binding first in the given map, then in the generic map, and
777 * lastly in the default keybindings. */
778 static enum request
779 get_keybinding(enum keymap keymap, int key)
780 {
781 struct keybinding *kbd;
782 int i;
783
784 for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
785 if (kbd->alias == key)
786 return kbd->request;
787
788 for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
789 if (kbd->alias == key)
790 return kbd->request;
791
792 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
793 if (default_keybindings[i].alias == key)
794 return default_keybindings[i].request;
795
796 return (enum request) key;
797 }
798
799
800 struct key {
801 char *name;
802 int value;
803 };
804
805 static struct key key_table[] = {
806 { "Enter", KEY_RETURN },
807 { "Space", ' ' },
808 { "Backspace", KEY_BACKSPACE },
809 { "Tab", KEY_TAB },
810 { "Escape", KEY_ESC },
811 { "Left", KEY_LEFT },
812 { "Right", KEY_RIGHT },
813 { "Up", KEY_UP },
814 { "Down", KEY_DOWN },
815 { "Insert", KEY_IC },
816 { "Delete", KEY_DC },
817 { "Hash", '#' },
818 { "Home", KEY_HOME },
819 { "End", KEY_END },
820 { "PageUp", KEY_PPAGE },
821 { "PageDown", KEY_NPAGE },
822 { "F1", KEY_F(1) },
823 { "F2", KEY_F(2) },
824 { "F3", KEY_F(3) },
825 { "F4", KEY_F(4) },
826 { "F5", KEY_F(5) },
827 { "F6", KEY_F(6) },
828 { "F7", KEY_F(7) },
829 { "F8", KEY_F(8) },
830 { "F9", KEY_F(9) },
831 { "F10", KEY_F(10) },
832 { "F11", KEY_F(11) },
833 { "F12", KEY_F(12) },
834 };
835
836 static int
837 get_key_value(const char *name)
838 {
839 int i;
840
841 for (i = 0; i < ARRAY_SIZE(key_table); i++)
842 if (!strcasecmp(key_table[i].name, name))
843 return key_table[i].value;
844
845 if (strlen(name) == 1 && isprint(*name))
846 return (int) *name;
847
848 return ERR;
849 }
850
851 static char *
852 get_key(enum request request)
853 {
854 static char buf[BUFSIZ];
855 static char key_char[] = "'X'";
856 size_t pos = 0;
857 char *sep = " ";
858 int i;
859
860 buf[pos] = 0;
861
862 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
863 struct keybinding *keybinding = &default_keybindings[i];
864 char *seq = NULL;
865 int key;
866
867 if (keybinding->request != request)
868 continue;
869
870 for (key = 0; key < ARRAY_SIZE(key_table); key++)
871 if (key_table[key].value == keybinding->alias)
872 seq = key_table[key].name;
873
874 if (seq == NULL &&
875 keybinding->alias < 127 &&
876 isprint(keybinding->alias)) {
877 key_char[1] = (char) keybinding->alias;
878 seq = key_char;
879 }
880
881 if (!seq)
882 seq = "'?'";
883
884 if (!string_format_from(buf, &pos, "%s%s", sep, seq))
885 return "Too many keybindings!";
886 sep = ", ";
887 }
888
889 return buf;
890 }
891
892
893 /*
894 * User config file handling.
895 */
896
897 static struct int_map color_map[] = {
898 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
899 COLOR_MAP(DEFAULT),
900 COLOR_MAP(BLACK),
901 COLOR_MAP(BLUE),
902 COLOR_MAP(CYAN),
903 COLOR_MAP(GREEN),
904 COLOR_MAP(MAGENTA),
905 COLOR_MAP(RED),
906 COLOR_MAP(WHITE),
907 COLOR_MAP(YELLOW),
908 };
909
910 #define set_color(color, name) \
911 set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
912
913 static struct int_map attr_map[] = {
914 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
915 ATTR_MAP(NORMAL),
916 ATTR_MAP(BLINK),
917 ATTR_MAP(BOLD),
918 ATTR_MAP(DIM),
919 ATTR_MAP(REVERSE),
920 ATTR_MAP(STANDOUT),
921 ATTR_MAP(UNDERLINE),
922 };
923
924 #define set_attribute(attr, name) \
925 set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
926
927 static int config_lineno;
928 static bool config_errors;
929 static char *config_msg;
930
931 /* Wants: object fgcolor bgcolor [attr] */
932 static int
933 option_color_command(int argc, char *argv[])
934 {
935 struct line_info *info;
936
937 if (argc != 3 && argc != 4) {
938 config_msg = "Wrong number of arguments given to color command";
939 return ERR;
940 }
941
942 info = get_line_info(argv[0], strlen(argv[0]));
943 if (!info) {
944 config_msg = "Unknown color name";
945 return ERR;
946 }
947
948 if (set_color(&info->fg, argv[1]) == ERR ||
949 set_color(&info->bg, argv[2]) == ERR) {
950 config_msg = "Unknown color";
951 return ERR;
952 }
953
954 if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
955 config_msg = "Unknown attribute";
956 return ERR;
957 }
958
959 return OK;
960 }
961
962 /* Wants: name = value */
963 static int
964 option_set_command(int argc, char *argv[])
965 {
966 if (argc != 3) {
967 config_msg = "Wrong number of arguments given to set command";
968 return ERR;
969 }
970
971 if (strcmp(argv[1], "=")) {
972 config_msg = "No value assigned";
973 return ERR;
974 }
975
976 if (!strcmp(argv[0], "show-rev-graph")) {
977 opt_rev_graph = (!strcmp(argv[2], "1") ||
978 !strcmp(argv[2], "true") ||
979 !strcmp(argv[2], "yes"));
980 return OK;
981 }
982
983 if (!strcmp(argv[0], "line-number-interval")) {
984 opt_num_interval = atoi(argv[2]);
985 return OK;
986 }
987
988 if (!strcmp(argv[0], "tab-size")) {
989 opt_tab_size = atoi(argv[2]);
990 return OK;
991 }
992
993 if (!strcmp(argv[0], "commit-encoding")) {
994 char *arg = argv[2];
995 int delimiter = *arg;
996 int i;
997
998 switch (delimiter) {
999 case '"':
1000 case '\'':
1001 for (arg++, i = 0; arg[i]; i++)
1002 if (arg[i] == delimiter) {
1003 arg[i] = 0;
1004 break;
1005 }
1006 default:
1007 string_copy(opt_encoding, arg);
1008 return OK;
1009 }
1010 }
1011
1012 config_msg = "Unknown variable name";
1013 return ERR;
1014 }
1015
1016 /* Wants: mode request key */
1017 static int
1018 option_bind_command(int argc, char *argv[])
1019 {
1020 enum request request;
1021 int keymap;
1022 int key;
1023
1024 if (argc != 3) {
1025 config_msg = "Wrong number of arguments given to bind command";
1026 return ERR;
1027 }
1028
1029 if (set_keymap(&keymap, argv[0]) == ERR) {
1030 config_msg = "Unknown key map";
1031 return ERR;
1032 }
1033
1034 key = get_key_value(argv[1]);
1035 if (key == ERR) {
1036 config_msg = "Unknown key";
1037 return ERR;
1038 }
1039
1040 request = get_request(argv[2]);
1041 if (request == REQ_UNKNOWN) {
1042 config_msg = "Unknown request name";
1043 return ERR;
1044 }
1045
1046 add_keybinding(keymap, request, key);
1047
1048 return OK;
1049 }
1050
1051 static int
1052 set_option(char *opt, char *value)
1053 {
1054 char *argv[16];
1055 int valuelen;
1056 int argc = 0;
1057
1058 /* Tokenize */
1059 while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1060 argv[argc++] = value;
1061
1062 value += valuelen;
1063 if (!*value)
1064 break;
1065
1066 *value++ = 0;
1067 while (isspace(*value))
1068 value++;
1069 }
1070
1071 if (!strcmp(opt, "color"))
1072 return option_color_command(argc, argv);
1073
1074 if (!strcmp(opt, "set"))
1075 return option_set_command(argc, argv);
1076
1077 if (!strcmp(opt, "bind"))
1078 return option_bind_command(argc, argv);
1079
1080 config_msg = "Unknown option command";
1081 return ERR;
1082 }
1083
1084 static int
1085 read_option(char *opt, int optlen, char *value, int valuelen)
1086 {
1087 int status = OK;
1088
1089 config_lineno++;
1090 config_msg = "Internal error";
1091
1092 /* Check for comment markers, since read_properties() will
1093 * only ensure opt and value are split at first " \t". */
1094 optlen = strcspn(opt, "#");
1095 if (optlen == 0)
1096 return OK;
1097
1098 if (opt[optlen] != 0) {
1099 config_msg = "No option value";
1100 status = ERR;
1101
1102 } else {
1103 /* Look for comment endings in the value. */
1104 int len = strcspn(value, "#");
1105
1106 if (len < valuelen) {
1107 valuelen = len;
1108 value[valuelen] = 0;
1109 }
1110
1111 status = set_option(opt, value);
1112 }
1113
1114 if (status == ERR) {
1115 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1116 config_lineno, optlen, opt, config_msg);
1117 config_errors = TRUE;
1118 }
1119
1120 /* Always keep going if errors are encountered. */
1121 return OK;
1122 }
1123
1124 static int
1125 load_options(void)
1126 {
1127 char *home = getenv("HOME");
1128 char buf[SIZEOF_STR];
1129 FILE *file;
1130
1131 config_lineno = 0;
1132 config_errors = FALSE;
1133
1134 if (!home || !string_format(buf, "%s/.tigrc", home))
1135 return ERR;
1136
1137 /* It's ok that the file doesn't exist. */
1138 file = fopen(buf, "r");
1139 if (!file)
1140 return OK;
1141
1142 if (read_properties(file, " \t", read_option) == ERR ||
1143 config_errors == TRUE)
1144 fprintf(stderr, "Errors while loading %s.\n", buf);
1145
1146 return OK;
1147 }
1148
1149
1150 /*
1151 * The viewer
1152 */
1153
1154 struct view;
1155 struct view_ops;
1156
1157 /* The display array of active views and the index of the current view. */
1158 static struct view *display[2];
1159 static unsigned int current_view;
1160
1161 #define foreach_displayed_view(view, i) \
1162 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1163
1164 #define displayed_views() (display[1] != NULL ? 2 : 1)
1165
1166 /* Current head and commit ID */
1167 static char ref_blob[SIZEOF_REF] = "";
1168 static char ref_commit[SIZEOF_REF] = "HEAD";
1169 static char ref_head[SIZEOF_REF] = "HEAD";
1170
1171 struct view {
1172 const char *name; /* View name */
1173 const char *cmd_fmt; /* Default command line format */
1174 const char *cmd_env; /* Command line set via environment */
1175 const char *id; /* Points to either of ref_{head,commit,blob} */
1176
1177 struct view_ops *ops; /* View operations */
1178
1179 enum keymap keymap; /* What keymap does this view have */
1180
1181 char cmd[SIZEOF_STR]; /* Command buffer */
1182 char ref[SIZEOF_REF]; /* Hovered commit reference */
1183 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
1184
1185 int height, width; /* The width and height of the main window */
1186 WINDOW *win; /* The main window */
1187 WINDOW *title; /* The title window living below the main window */
1188
1189 /* Navigation */
1190 unsigned long offset; /* Offset of the window top */
1191 unsigned long lineno; /* Current line number */
1192
1193 /* Searching */
1194 char grep[SIZEOF_STR]; /* Search string */
1195 regex_t *regex; /* Pre-compiled regex */
1196
1197 /* If non-NULL, points to the view that opened this view. If this view
1198 * is closed tig will switch back to the parent view. */
1199 struct view *parent;
1200
1201 /* Buffering */
1202 unsigned long lines; /* Total number of lines */
1203 struct line *line; /* Line index */
1204 unsigned long line_size;/* Total number of allocated lines */
1205 unsigned int digits; /* Number of digits in the lines member. */
1206
1207 /* Loading */
1208 FILE *pipe;
1209 time_t start_time;
1210 };
1211
1212 struct view_ops {
1213 /* What type of content being displayed. Used in the title bar. */
1214 const char *type;
1215 /* Draw one line; @lineno must be < view->height. */
1216 bool (*draw)(struct view *view, struct line *line, unsigned int lineno);
1217 /* Read one line; updates view->line. */
1218 bool (*read)(struct view *view, char *data);
1219 /* Depending on view, change display based on current line. */
1220 bool (*enter)(struct view *view, struct line *line);
1221 /* Search for regex in a line. */
1222 bool (*grep)(struct view *view, struct line *line);
1223 };
1224
1225 static struct view_ops pager_ops;
1226 static struct view_ops main_ops;
1227 static struct view_ops tree_ops;
1228 static struct view_ops blob_ops;
1229
1230 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1231 { name, cmd, #env, ref, ops, map}
1232
1233 #define VIEW_(id, name, ops, ref) \
1234 VIEW_STR(name, TIG_##id##_CMD, TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1235
1236
1237 static struct view views[] = {
1238 VIEW_(MAIN, "main", &main_ops, ref_head),
1239 VIEW_(DIFF, "diff", &pager_ops, ref_commit),
1240 VIEW_(LOG, "log", &pager_ops, ref_head),
1241 VIEW_(TREE, "tree", &tree_ops, ref_commit),
1242 VIEW_(BLOB, "blob", &blob_ops, ref_blob),
1243 VIEW_(HELP, "help", &pager_ops, "static"),
1244 VIEW_(PAGER, "pager", &pager_ops, "static"),
1245 };
1246
1247 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1248
1249 #define foreach_view(view, i) \
1250 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1251
1252 #define view_is_displayed(view) \
1253 (view == display[0] || view == display[1])
1254
1255 static bool
1256 draw_view_line(struct view *view, unsigned int lineno)
1257 {
1258 assert(view_is_displayed(view));
1259
1260 if (view->offset + lineno >= view->lines)
1261 return FALSE;
1262
1263 return view->ops->draw(view, &view->line[view->offset + lineno], lineno);
1264 }
1265
1266 static void
1267 redraw_view_from(struct view *view, int lineno)
1268 {
1269 assert(0 <= lineno && lineno < view->height);
1270
1271 for (; lineno < view->height; lineno++) {
1272 if (!draw_view_line(view, lineno))
1273 break;
1274 }
1275
1276 redrawwin(view->win);
1277 wrefresh(view->win);
1278 }
1279
1280 static void
1281 redraw_view(struct view *view)
1282 {
1283 wclear(view->win);
1284 redraw_view_from(view, 0);
1285 }
1286
1287
1288 static void
1289 update_view_title(struct view *view)
1290 {
1291 assert(view_is_displayed(view));
1292
1293 if (view == display[current_view])
1294 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1295 else
1296 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1297
1298 werase(view->title);
1299 wmove(view->title, 0, 0);
1300
1301 if (*view->ref)
1302 wprintw(view->title, "[%s] %s", view->name, view->ref);
1303 else
1304 wprintw(view->title, "[%s]", view->name);
1305
1306 if (view->lines || view->pipe) {
1307 unsigned int view_lines = view->offset + view->height;
1308 unsigned int lines = view->lines
1309 ? MIN(view_lines, view->lines) * 100 / view->lines
1310 : 0;
1311
1312 wprintw(view->title, " - %s %d of %d (%d%%)",
1313 view->ops->type,
1314 view->lineno + 1,
1315 view->lines,
1316 lines);
1317 }
1318
1319 if (view->pipe) {
1320 time_t secs = time(NULL) - view->start_time;
1321
1322 /* Three git seconds are a long time ... */
1323 if (secs > 2)
1324 wprintw(view->title, " %lds", secs);
1325 }
1326
1327 wmove(view->title, 0, view->width - 1);
1328 wrefresh(view->title);
1329 }
1330
1331 static void
1332 resize_display(void)
1333 {
1334 int offset, i;
1335 struct view *base = display[0];
1336 struct view *view = display[1] ? display[1] : display[0];
1337
1338 /* Setup window dimensions */
1339
1340 getmaxyx(stdscr, base->height, base->width);
1341
1342 /* Make room for the status window. */
1343 base->height -= 1;
1344
1345 if (view != base) {
1346 /* Horizontal split. */
1347 view->width = base->width;
1348 view->height = SCALE_SPLIT_VIEW(base->height);
1349 base->height -= view->height;
1350
1351 /* Make room for the title bar. */
1352 view->height -= 1;
1353 }
1354
1355 /* Make room for the title bar. */
1356 base->height -= 1;
1357
1358 offset = 0;
1359
1360 foreach_displayed_view (view, i) {
1361 if (!view->win) {
1362 view->win = newwin(view->height, 0, offset, 0);
1363 if (!view->win)
1364 die("Failed to create %s view", view->name);
1365
1366 scrollok(view->win, TRUE);
1367
1368 view->title = newwin(1, 0, offset + view->height, 0);
1369 if (!view->title)
1370 die("Failed to create title window");
1371
1372 } else {
1373 wresize(view->win, view->height, view->width);
1374 mvwin(view->win, offset, 0);
1375 mvwin(view->title, offset + view->height, 0);
1376 }
1377
1378 offset += view->height + 1;
1379 }
1380 }
1381
1382 static void
1383 redraw_display(void)
1384 {
1385 struct view *view;
1386 int i;
1387
1388 foreach_displayed_view (view, i) {
1389 redraw_view(view);
1390 update_view_title(view);
1391 }
1392 }
1393
1394 static void
1395 update_display_cursor(void)
1396 {
1397 struct view *view = display[current_view];
1398
1399 /* Move the cursor to the right-most column of the cursor line.
1400 *
1401 * XXX: This could turn out to be a bit expensive, but it ensures that
1402 * the cursor does not jump around. */
1403 if (view->lines) {
1404 wmove(view->win, view->lineno - view->offset, view->width - 1);
1405 wrefresh(view->win);
1406 }
1407 }
1408
1409 /*
1410 * Navigation
1411 */
1412
1413 /* Scrolling backend */
1414 static void
1415 do_scroll_view(struct view *view, int lines, bool redraw)
1416 {
1417 assert(view_is_displayed(view));
1418
1419 /* The rendering expects the new offset. */
1420 view->offset += lines;
1421
1422 assert(0 <= view->offset && view->offset < view->lines);
1423 assert(lines);
1424
1425 /* Redraw the whole screen if scrolling is pointless. */
1426 if (view->height < ABS(lines)) {
1427 redraw_view(view);
1428
1429 } else {
1430 int line = lines > 0 ? view->height - lines : 0;
1431 int end = line + ABS(lines);
1432
1433 wscrl(view->win, lines);
1434
1435 for (; line < end; line++) {
1436 if (!draw_view_line(view, line))
1437 break;
1438 }
1439 }
1440
1441 /* Move current line into the view. */
1442 if (view->lineno < view->offset) {
1443 view->lineno = view->offset;
1444 draw_view_line(view, 0);
1445
1446 } else if (view->lineno >= view->offset + view->height) {
1447 if (view->lineno == view->offset + view->height) {
1448 /* Clear the hidden line so it doesn't show if the view
1449 * is scrolled up. */
1450 wmove(view->win, view->height, 0);
1451 wclrtoeol(view->win);
1452 }
1453 view->lineno = view->offset + view->height - 1;
1454 draw_view_line(view, view->lineno - view->offset);
1455 }
1456
1457 assert(view->offset <= view->lineno && view->lineno < view->lines);
1458
1459 if (!redraw)
1460 return;
1461
1462 redrawwin(view->win);
1463 wrefresh(view->win);
1464 report("");
1465 }
1466
1467 /* Scroll frontend */
1468 static void
1469 scroll_view(struct view *view, enum request request)
1470 {
1471 int lines = 1;
1472
1473 switch (request) {
1474 case REQ_SCROLL_PAGE_DOWN:
1475 lines = view->height;
1476 case REQ_SCROLL_LINE_DOWN:
1477 if (view->offset + lines > view->lines)
1478 lines = view->lines - view->offset;
1479
1480 if (lines == 0 || view->offset + view->height >= view->lines) {
1481 report("Cannot scroll beyond the last line");
1482 return;
1483 }
1484 break;
1485
1486 case REQ_SCROLL_PAGE_UP:
1487 lines = view->height;
1488 case REQ_SCROLL_LINE_UP:
1489 if (lines > view->offset)
1490 lines = view->offset;
1491
1492 if (lines == 0) {
1493 report("Cannot scroll beyond the first line");
1494 return;
1495 }
1496
1497 lines = -lines;
1498 break;
1499
1500 default:
1501 die("request %d not handled in switch", request);
1502 }
1503
1504 do_scroll_view(view, lines, TRUE);
1505 }
1506
1507 /* Cursor moving */
1508 static void
1509 move_view(struct view *view, enum request request, bool redraw)
1510 {
1511 int steps;
1512
1513 switch (request) {
1514 case REQ_MOVE_FIRST_LINE:
1515 steps = -view->lineno;
1516 break;
1517
1518 case REQ_MOVE_LAST_LINE:
1519 steps = view->lines - view->lineno - 1;
1520 break;
1521
1522 case REQ_MOVE_PAGE_UP:
1523 steps = view->height > view->lineno
1524 ? -view->lineno : -view->height;
1525 break;
1526
1527 case REQ_MOVE_PAGE_DOWN:
1528 steps = view->lineno + view->height >= view->lines
1529 ? view->lines - view->lineno - 1 : view->height;
1530 break;
1531
1532 case REQ_MOVE_UP:
1533 steps = -1;
1534 break;
1535
1536 case REQ_MOVE_DOWN:
1537 steps = 1;
1538 break;
1539
1540 default:
1541 die("request %d not handled in switch", request);
1542 }
1543
1544 if (steps <= 0 && view->lineno == 0) {
1545 report("Cannot move beyond the first line");
1546 return;
1547
1548 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1549 report("Cannot move beyond the last line");
1550 return;
1551 }
1552
1553 /* Move the current line */
1554 view->lineno += steps;
1555 assert(0 <= view->lineno && view->lineno < view->lines);
1556
1557 /* Repaint the old "current" line if we be scrolling */
1558 if (ABS(steps) < view->height) {
1559 int prev_lineno = view->lineno - steps - view->offset;
1560
1561 wmove(view->win, prev_lineno, 0);
1562 wclrtoeol(view->win);
1563 draw_view_line(view, prev_lineno);
1564 }
1565
1566 /* Check whether the view needs to be scrolled */
1567 if (view->lineno < view->offset ||
1568 view->lineno >= view->offset + view->height) {
1569 if (steps < 0 && -steps > view->offset) {
1570 steps = -view->offset;
1571
1572 } else if (steps > 0) {
1573 if (view->lineno == view->lines - 1 &&
1574 view->lines > view->height) {
1575 steps = view->lines - view->offset - 1;
1576 if (steps >= view->height)
1577 steps -= view->height - 1;
1578 }
1579 }
1580
1581 do_scroll_view(view, steps, redraw);
1582 return;
1583 }
1584
1585 /* Draw the current line */
1586 draw_view_line(view, view->lineno - view->offset);
1587
1588 if (!redraw)
1589 return;
1590
1591 redrawwin(view->win);
1592 wrefresh(view->win);
1593 report("");
1594 }
1595
1596
1597 /*
1598 * Searching
1599 */
1600
1601 static void search_view(struct view *view, enum request request, const char *search);
1602
1603 static bool
1604 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1605 {
1606 assert(view_is_displayed(view));
1607
1608 if (!view->ops->grep(view, line))
1609 return FALSE;
1610
1611 if (lineno - view->offset >= view->height) {
1612 view->offset = lineno;
1613 view->lineno = lineno;
1614 redraw_view(view);
1615
1616 } else {
1617 unsigned long old_lineno = view->lineno - view->offset;
1618
1619 view->lineno = lineno;
1620
1621 wmove(view->win, old_lineno, 0);
1622 wclrtoeol(view->win);
1623 draw_view_line(view, old_lineno);
1624
1625 draw_view_line(view, view->lineno - view->offset);
1626 redrawwin(view->win);
1627 wrefresh(view->win);
1628 }
1629
1630 report("Line %ld matches '%s'", lineno + 1, view->grep);
1631 return TRUE;
1632 }
1633
1634 static void
1635 find_next(struct view *view, enum request request)
1636 {
1637 unsigned long lineno = view->lineno;
1638 int direction;
1639
1640 if (!*view->grep) {
1641 if (!*opt_search)
1642 report("No previous search");
1643 else
1644 search_view(view, request, opt_search);
1645 return;
1646 }
1647
1648 switch (request) {
1649 case REQ_SEARCH:
1650 case REQ_FIND_NEXT:
1651 direction = 1;
1652 break;
1653
1654 case REQ_SEARCH_BACK:
1655 case REQ_FIND_PREV:
1656 direction = -1;
1657 break;
1658
1659 default:
1660 return;
1661 }
1662
1663 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1664 lineno += direction;
1665
1666 /* Note, lineno is unsigned long so will wrap around in which case it
1667 * will become bigger than view->lines. */
1668 for (; lineno < view->lines; lineno += direction) {
1669 struct line *line = &view->line[lineno];
1670
1671 if (find_next_line(view, lineno, line))
1672 return;
1673 }
1674
1675 report("No match found for '%s'", view->grep);
1676 }
1677
1678 static void
1679 search_view(struct view *view, enum request request, const char *search)
1680 {
1681 int regex_err;
1682
1683 if (view->regex) {
1684 regfree(view->regex);
1685 *view->grep = 0;
1686 } else {
1687 view->regex = calloc(1, sizeof(*view->regex));
1688 if (!view->regex)
1689 return;
1690 }
1691
1692 regex_err = regcomp(view->regex, search, REG_EXTENDED);
1693 if (regex_err != 0) {
1694 char buf[SIZEOF_STR] = "unknown error";
1695
1696 regerror(regex_err, view->regex, buf, sizeof(buf));
1697 report("Search failed: %s", buf);
1698 return;
1699 }
1700
1701 string_copy(view->grep, search);
1702
1703 find_next(view, request);
1704 }
1705
1706 /*
1707 * Incremental updating
1708 */
1709
1710 static void
1711 end_update(struct view *view)
1712 {
1713 if (!view->pipe)
1714 return;
1715 set_nonblocking_input(FALSE);
1716 if (view->pipe == stdin)
1717 fclose(view->pipe);
1718 else
1719 pclose(view->pipe);
1720 view->pipe = NULL;
1721 }
1722
1723 static bool
1724 begin_update(struct view *view)
1725 {
1726 const char *id = view->id;
1727
1728 if (view->pipe)
1729 end_update(view);
1730
1731 if (opt_cmd[0]) {
1732 string_copy(view->cmd, opt_cmd);
1733 opt_cmd[0] = 0;
1734 /* When running random commands, the view ref could have become
1735 * invalid so clear it. */
1736 view->ref[0] = 0;
1737
1738 } else if (view == VIEW(REQ_VIEW_TREE)) {
1739 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1740
1741 if (strcmp(view->vid, view->id))
1742 opt_path[0] = 0;
1743
1744 if (!string_format(view->cmd, format, id, opt_path))
1745 return FALSE;
1746
1747 } else {
1748 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1749
1750 if (!string_format(view->cmd, format, id, id, id, id, id))
1751 return FALSE;
1752 }
1753
1754 /* Special case for the pager view. */
1755 if (opt_pipe) {
1756 view->pipe = opt_pipe;
1757 opt_pipe = NULL;
1758 } else {
1759 view->pipe = popen(view->cmd, "r");
1760 }
1761
1762 if (!view->pipe)
1763 return FALSE;
1764
1765 set_nonblocking_input(TRUE);
1766
1767 view->offset = 0;
1768 view->lines = 0;
1769 view->lineno = 0;
1770 string_copy(view->vid, id);
1771
1772 if (view->line) {
1773 int i;
1774
1775 for (i = 0; i < view->lines; i++)
1776 if (view->line[i].data)
1777 free(view->line[i].data);
1778
1779 free(view->line);
1780 view->line = NULL;
1781 }
1782
1783 view->start_time = time(NULL);
1784
1785 return TRUE;
1786 }
1787
1788 static struct line *
1789 realloc_lines(struct view *view, size_t line_size)
1790 {
1791 struct line *tmp = realloc(view->line, sizeof(*view->line) * line_size);
1792
1793 if (!tmp)
1794 return NULL;
1795
1796 view->line = tmp;
1797 view->line_size = line_size;
1798 return view->line;
1799 }
1800
1801 static bool
1802 update_view(struct view *view)
1803 {
1804 char in_buffer[BUFSIZ];
1805 char out_buffer[BUFSIZ * 2];
1806 char *line;
1807 /* The number of lines to read. If too low it will cause too much
1808 * redrawing (and possible flickering), if too high responsiveness
1809 * will suffer. */
1810 unsigned long lines = view->height;
1811 int redraw_from = -1;
1812
1813 if (!view->pipe)
1814 return TRUE;
1815
1816 /* Only redraw if lines are visible. */
1817 if (view->offset + view->height >= view->lines)
1818 redraw_from = view->lines - view->offset;
1819
1820 /* FIXME: This is probably not perfect for backgrounded views. */
1821 if (!realloc_lines(view, view->lines + lines))
1822 goto alloc_error;
1823
1824 while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
1825 size_t linelen = strlen(line);
1826
1827 if (linelen)
1828 line[linelen - 1] = 0;
1829
1830 if (opt_iconv != ICONV_NONE) {
1831 char *inbuf = line;
1832 size_t inlen = linelen;
1833
1834 char *outbuf = out_buffer;
1835 size_t outlen = sizeof(out_buffer);
1836
1837 size_t ret;
1838
1839 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
1840 if (ret != (size_t) -1) {
1841 line = out_buffer;
1842 linelen = strlen(out_buffer);
1843 }
1844 }
1845
1846 if (!view->ops->read(view, line))
1847 goto alloc_error;
1848
1849 if (lines-- == 1)
1850 break;
1851 }
1852
1853 {
1854 int digits;
1855
1856 lines = view->lines;
1857 for (digits = 0; lines; digits++)
1858 lines /= 10;
1859
1860 /* Keep the displayed view in sync with line number scaling. */
1861 if (digits != view->digits) {
1862 view->digits = digits;
1863 redraw_from = 0;
1864 }
1865 }
1866
1867 if (!view_is_displayed(view))
1868 goto check_pipe;
1869
1870 if (view == VIEW(REQ_VIEW_TREE)) {
1871 /* Clear the view and redraw everything since the tree sorting
1872 * might have rearranged things. */
1873 redraw_view(view);
1874
1875 } else if (redraw_from >= 0) {
1876 /* If this is an incremental update, redraw the previous line
1877 * since for commits some members could have changed when
1878 * loading the main view. */
1879 if (redraw_from > 0)
1880 redraw_from--;
1881
1882 /* Incrementally draw avoids flickering. */
1883 redraw_view_from(view, redraw_from);
1884 }
1885
1886 /* Update the title _after_ the redraw so that if the redraw picks up a
1887 * commit reference in view->ref it'll be available here. */
1888 update_view_title(view);
1889
1890 check_pipe:
1891 if (ferror(view->pipe)) {
1892 report("Failed to read: %s", strerror(errno));
1893 goto end;
1894
1895 } else if (feof(view->pipe)) {
1896 report("");
1897 goto end;
1898 }
1899
1900 return TRUE;
1901
1902 alloc_error:
1903 report("Allocation failure");
1904
1905 end:
1906 end_update(view);
1907 return FALSE;
1908 }
1909
1910
1911 /*
1912 * View opening
1913 */
1914
1915 static void open_help_view(struct view *view)
1916 {
1917 char buf[BUFSIZ];
1918 int lines = ARRAY_SIZE(req_info) + 2;
1919 int i;
1920
1921 if (view->lines > 0)
1922 return;
1923
1924 for (i = 0; i < ARRAY_SIZE(req_info); i++)
1925 if (!req_info[i].request)
1926 lines++;
1927
1928 view->line = calloc(lines, sizeof(*view->line));
1929 if (!view->line) {
1930 report("Allocation failure");
1931 return;
1932 }
1933
1934 view->ops->read(view, "Quick reference for tig keybindings:");
1935
1936 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
1937 char *key;
1938
1939 if (!req_info[i].request) {
1940 view->ops->read(view, "");
1941 view->ops->read(view, req_info[i].help);
1942 continue;
1943 }
1944
1945 key = get_key(req_info[i].request);
1946 if (!string_format(buf, "%-25s %s", key, req_info[i].help))
1947 continue;
1948
1949 view->ops->read(view, buf);
1950 }
1951 }
1952
1953 enum open_flags {
1954 OPEN_DEFAULT = 0, /* Use default view switching. */
1955 OPEN_SPLIT = 1, /* Split current view. */
1956 OPEN_BACKGROUNDED = 2, /* Backgrounded. */
1957 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
1958 };
1959
1960 static void
1961 open_view(struct view *prev, enum request request, enum open_flags flags)
1962 {
1963 bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
1964 bool split = !!(flags & OPEN_SPLIT);
1965 bool reload = !!(flags & OPEN_RELOAD);
1966 struct view *view = VIEW(request);
1967 int nviews = displayed_views();
1968 struct view *base_view = display[0];
1969
1970 if (view == prev && nviews == 1 && !reload) {
1971 report("Already in %s view", view->name);
1972 return;
1973 }
1974
1975 if (view == VIEW(REQ_VIEW_HELP)) {
1976 open_help_view(view);
1977
1978 } else if ((reload || strcmp(view->vid, view->id)) &&
1979 !begin_update(view)) {
1980 report("Failed to load %s view", view->name);
1981 return;
1982 }
1983
1984 if (split) {
1985 display[1] = view;
1986 if (!backgrounded)
1987 current_view = 1;
1988 } else {
1989 /* Maximize the current view. */
1990 memset(display, 0, sizeof(display));
1991 current_view = 0;
1992 display[current_view] = view;
1993 }
1994
1995 /* Resize the view when switching between split- and full-screen,
1996 * or when switching between two different full-screen views. */
1997 if (nviews != displayed_views() ||
1998 (nviews == 1 && base_view != display[0]))
1999 resize_display();
2000
2001 if (split && prev->lineno - prev->offset >= prev->height) {
2002 /* Take the title line into account. */
2003 int lines = prev->lineno - prev->offset - prev->height + 1;
2004
2005 /* Scroll the view that was split if the current line is
2006 * outside the new limited view. */
2007 do_scroll_view(prev, lines, TRUE);
2008 }
2009
2010 if (prev && view != prev) {
2011 if (split && !backgrounded) {
2012 /* "Blur" the previous view. */
2013 update_view_title(prev);
2014 }
2015
2016 view->parent = prev;
2017 }
2018
2019 if (view->pipe && view->lines == 0) {
2020 /* Clear the old view and let the incremental updating refill
2021 * the screen. */
2022 wclear(view->win);
2023 report("");
2024 } else {
2025 redraw_view(view);
2026 report("");
2027 }
2028
2029 /* If the view is backgrounded the above calls to report()
2030 * won't redraw the view title. */
2031 if (backgrounded)
2032 update_view_title(view);
2033 }
2034
2035
2036 /*
2037 * User request switch noodle
2038 */
2039
2040 static int
2041 view_driver(struct view *view, enum request request)
2042 {
2043 int i;
2044
2045 switch (request) {
2046 case REQ_MOVE_UP:
2047 case REQ_MOVE_DOWN:
2048 case REQ_MOVE_PAGE_UP:
2049 case REQ_MOVE_PAGE_DOWN:
2050 case REQ_MOVE_FIRST_LINE:
2051 case REQ_MOVE_LAST_LINE:
2052 move_view(view, request, TRUE);
2053 break;
2054
2055 case REQ_SCROLL_LINE_DOWN:
2056 case REQ_SCROLL_LINE_UP:
2057 case REQ_SCROLL_PAGE_DOWN:
2058 case REQ_SCROLL_PAGE_UP:
2059 scroll_view(view, request);
2060 break;
2061
2062 case REQ_VIEW_BLOB:
2063 if (!ref_blob[0]) {
2064 report("No file chosen, press 't' to open tree view");
2065 break;
2066 }
2067 /* Fall-through */
2068 case REQ_VIEW_MAIN:
2069 case REQ_VIEW_DIFF:
2070 case REQ_VIEW_LOG:
2071 case REQ_VIEW_TREE:
2072 case REQ_VIEW_HELP:
2073 case REQ_VIEW_PAGER:
2074 open_view(view, request, OPEN_DEFAULT);
2075 break;
2076
2077 case REQ_NEXT:
2078 case REQ_PREVIOUS:
2079 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2080
2081 if ((view == VIEW(REQ_VIEW_DIFF) &&
2082 view->parent == VIEW(REQ_VIEW_MAIN)) ||
2083 (view == VIEW(REQ_VIEW_BLOB) &&
2084 view->parent == VIEW(REQ_VIEW_TREE))) {
2085 bool redraw = display[1] == view;
2086
2087 view = view->parent;
2088 move_view(view, request, redraw);
2089 if (redraw)
2090 update_view_title(view);
2091 } else {
2092 move_view(view, request, TRUE);
2093 break;
2094 }
2095 /* Fall-through */
2096
2097 case REQ_ENTER:
2098 if (!view->lines) {
2099 report("Nothing to enter");
2100 break;
2101 }
2102 return view->ops->enter(view, &view->line[view->lineno]);
2103
2104 case REQ_VIEW_NEXT:
2105 {
2106 int nviews = displayed_views();
2107 int next_view = (current_view + 1) % nviews;
2108
2109 if (next_view == current_view) {
2110 report("Only one view is displayed");
2111 break;
2112 }
2113
2114 current_view = next_view;
2115 /* Blur out the title of the previous view. */
2116 update_view_title(view);
2117 report("");
2118 break;
2119 }
2120 case REQ_TOGGLE_LINENO:
2121 opt_line_number = !opt_line_number;
2122 redraw_display();
2123 break;
2124
2125 case REQ_TOGGLE_REV_GRAPH:
2126 opt_rev_graph = !opt_rev_graph;
2127 redraw_display();
2128 break;
2129
2130 case REQ_PROMPT:
2131 /* Always reload^Wrerun commands from the prompt. */
2132 open_view(view, opt_request, OPEN_RELOAD);
2133 break;
2134
2135 case REQ_SEARCH:
2136 case REQ_SEARCH_BACK:
2137 search_view(view, request, opt_search);
2138 break;
2139
2140 case REQ_FIND_NEXT:
2141 case REQ_FIND_PREV:
2142 find_next(view, request);
2143 break;
2144
2145 case REQ_STOP_LOADING:
2146 for (i = 0; i < ARRAY_SIZE(views); i++) {
2147 view = &views[i];
2148 if (view->pipe)
2149 report("Stopped loading the %s view", view->name),
2150 end_update(view);
2151 }
2152 break;
2153
2154 case REQ_SHOW_VERSION:
2155 report("%s (built %s)", VERSION, __DATE__);
2156 return TRUE;
2157
2158 case REQ_SCREEN_RESIZE:
2159 resize_display();
2160 /* Fall-through */
2161 case REQ_SCREEN_REDRAW:
2162 redraw_display();
2163 break;
2164
2165 case REQ_NONE:
2166 doupdate();
2167 return TRUE;
2168
2169 case REQ_VIEW_CLOSE:
2170 /* XXX: Mark closed views by letting view->parent point to the
2171 * view itself. Parents to closed view should never be
2172 * followed. */
2173 if (view->parent &&
2174 view->parent->parent != view->parent) {
2175 memset(display, 0, sizeof(display));
2176 current_view = 0;
2177 display[current_view] = view->parent;
2178 view->parent = view;
2179 resize_display();
2180 redraw_display();
2181 break;
2182 }
2183 /* Fall-through */
2184 case REQ_QUIT:
2185 return FALSE;
2186
2187 default:
2188 /* An unknown key will show most commonly used commands. */
2189 report("Unknown key, press 'h' for help");
2190 return TRUE;
2191 }
2192
2193 return TRUE;
2194 }
2195
2196
2197 /*
2198 * Pager backend
2199 */
2200
2201 static bool
2202 pager_draw(struct view *view, struct line *line, unsigned int lineno)
2203 {
2204 char *text = line->data;
2205 enum line_type type = line->type;
2206 int textlen = strlen(text);
2207 int attr;
2208
2209 wmove(view->win, lineno, 0);
2210
2211 if (view->offset + lineno == view->lineno) {
2212 if (type == LINE_COMMIT) {
2213 string_copy(view->ref, text + 7);
2214 string_copy(ref_commit, view->ref);
2215
2216 } else if (type == LINE_TREE_DIR || type == LINE_TREE_FILE) {
2217 string_ncopy(view->ref, text + STRING_SIZE("100644 blob "), 40);
2218 string_copy(ref_blob, view->ref);
2219 }
2220
2221 type = LINE_CURSOR;
2222 wchgat(view->win, -1, 0, type, NULL);
2223 }
2224
2225 attr = get_line_attr(type);
2226 wattrset(view->win, attr);
2227
2228 if (opt_line_number || opt_tab_size < TABSIZE) {
2229 static char spaces[] = " ";
2230 int col_offset = 0, col = 0;
2231
2232 if (opt_line_number) {
2233 unsigned long real_lineno = view->offset + lineno + 1;
2234
2235 if (real_lineno == 1 ||
2236 (real_lineno % opt_num_interval) == 0) {
2237 wprintw(view->win, "%.*d", view->digits, real_lineno);
2238
2239 } else {
2240 waddnstr(view->win, spaces,
2241 MIN(view->digits, STRING_SIZE(spaces)));
2242 }
2243 waddstr(view->win, ": ");
2244 col_offset = view->digits + 2;
2245 }
2246
2247 while (text && col_offset + col < view->width) {
2248 int cols_max = view->width - col_offset - col;
2249 char *pos = text;
2250 int cols;
2251
2252 if (*text == '\t') {
2253 text++;
2254 assert(sizeof(spaces) > TABSIZE);
2255 pos = spaces;
2256 cols = opt_tab_size - (col % opt_tab_size);
2257
2258 } else {
2259 text = strchr(text, '\t');
2260 cols = line ? text - pos : strlen(pos);
2261 }
2262
2263 waddnstr(view->win, pos, MIN(cols, cols_max));
2264 col += cols;
2265 }
2266
2267 } else {
2268 int col = 0, pos = 0;
2269
2270 for (; pos < textlen && col < view->width; pos++, col++)
2271 if (text[pos] == '\t')
2272 col += TABSIZE - (col % TABSIZE) - 1;
2273
2274 waddnstr(view->win, text, pos);
2275 }
2276
2277 return TRUE;
2278 }
2279
2280 static bool
2281 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2282 {
2283 char refbuf[SIZEOF_STR];
2284 char *ref = NULL;
2285 FILE *pipe;
2286
2287 if (!string_format(refbuf, "git describe %s", commit_id))
2288 return TRUE;
2289
2290 pipe = popen(refbuf, "r");
2291 if (!pipe)
2292 return TRUE;
2293
2294 if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2295 ref = chomp_string(ref);
2296 pclose(pipe);
2297
2298 if (!ref || !*ref)
2299 return TRUE;
2300
2301 /* This is the only fatal call, since it can "corrupt" the buffer. */
2302 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2303 return FALSE;
2304
2305 return TRUE;
2306 }
2307
2308 static void
2309 add_pager_refs(struct view *view, struct line *line)
2310 {
2311 char buf[SIZEOF_STR];
2312 char *commit_id = line->data + STRING_SIZE("commit ");
2313 struct ref **refs;
2314 size_t bufpos = 0, refpos = 0;
2315 const char *sep = "Refs: ";
2316 bool is_tag = FALSE;
2317
2318 assert(line->type == LINE_COMMIT);
2319
2320 refs = get_refs(commit_id);
2321 if (!refs) {
2322 if (view == VIEW(REQ_VIEW_DIFF))
2323 goto try_add_describe_ref;
2324 return;
2325 }
2326
2327 do {
2328 struct ref *ref = refs[refpos];
2329 char *fmt = ref->tag ? "%s[%s]" : "%s%s";
2330
2331 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2332 return;
2333 sep = ", ";
2334 if (ref->tag)
2335 is_tag = TRUE;
2336 } while (refs[refpos++]->next);
2337
2338 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2339 try_add_describe_ref:
2340 /* Add <tag>-g<commit_id> "fake" reference. */
2341 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2342 return;
2343 }
2344
2345 if (bufpos == 0)
2346 return;
2347
2348 if (!realloc_lines(view, view->line_size + 1))
2349 return;
2350
2351 line = &view->line[view->lines];
2352 line->data = strdup(buf);
2353 if (!line->data)
2354 return;
2355
2356 line->type = LINE_PP_REFS;
2357 view->lines++;
2358 }
2359
2360 static bool
2361 pager_read(struct view *view, char *data)
2362 {
2363 struct line *line = &view->line[view->lines];
2364
2365 line->data = strdup(data);
2366 if (!line->data)
2367 return FALSE;
2368
2369 line->type = get_line_type(line->data);
2370 view->lines++;
2371
2372 if (line->type == LINE_COMMIT &&
2373 (view == VIEW(REQ_VIEW_DIFF) ||
2374 view == VIEW(REQ_VIEW_LOG)))
2375 add_pager_refs(view, line);
2376
2377 return TRUE;
2378 }
2379
2380 static bool
2381 pager_enter(struct view *view, struct line *line)
2382 {
2383 int split = 0;
2384
2385 if (line->type == LINE_COMMIT &&
2386 (view == VIEW(REQ_VIEW_LOG) ||
2387 view == VIEW(REQ_VIEW_PAGER))) {
2388 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2389 split = 1;
2390 }
2391
2392 /* Always scroll the view even if it was split. That way
2393 * you can use Enter to scroll through the log view and
2394 * split open each commit diff. */
2395 scroll_view(view, REQ_SCROLL_LINE_DOWN);
2396
2397 /* FIXME: A minor workaround. Scrolling the view will call report("")
2398 * but if we are scrolling a non-current view this won't properly
2399 * update the view title. */
2400 if (split)
2401 update_view_title(view);
2402
2403 return TRUE;
2404 }
2405
2406 static bool
2407 pager_grep(struct view *view, struct line *line)
2408 {
2409 regmatch_t pmatch;
2410 char *text = line->data;
2411
2412 if (!*text)
2413 return FALSE;
2414
2415 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2416 return FALSE;
2417
2418 return TRUE;
2419 }
2420
2421 static struct view_ops pager_ops = {
2422 "line",
2423 pager_draw,
2424 pager_read,
2425 pager_enter,
2426 pager_grep,
2427 };
2428
2429
2430 /*
2431 * Tree backend
2432 */
2433
2434 /* Parse output from git ls-tree:
2435 *
2436 * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
2437 * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
2438 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
2439 * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
2440 */
2441
2442 #define SIZEOF_TREE_ATTR \
2443 STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
2444
2445 #define TREE_UP_FORMAT "040000 tree %s\t.."
2446
2447 static int
2448 tree_compare_entry(enum line_type type1, char *name1,
2449 enum line_type type2, char *name2)
2450 {
2451 if (type1 != type2) {
2452 if (type1 == LINE_TREE_DIR)
2453 return -1;
2454 return 1;
2455 }
2456
2457 return strcmp(name1, name2);
2458 }
2459
2460 static bool
2461 tree_read(struct view *view, char *text)
2462 {
2463 size_t textlen = strlen(text);
2464 char buf[SIZEOF_STR];
2465 unsigned long pos;
2466 enum line_type type;
2467 bool first_read = view->lines == 0;
2468
2469 if (textlen <= SIZEOF_TREE_ATTR)
2470 return FALSE;
2471
2472 type = text[STRING_SIZE("100644 ")] == 't'
2473 ? LINE_TREE_DIR : LINE_TREE_FILE;
2474
2475 if (first_read) {
2476 /* Add path info line */
2477 if (string_format(buf, "Directory path /%s", opt_path) &&
2478 realloc_lines(view, view->line_size + 1) &&
2479 pager_read(view, buf))
2480 view->line[view->lines - 1].type = LINE_DEFAULT;
2481 else
2482 return FALSE;
2483
2484 /* Insert "link" to parent directory. */
2485 if (*opt_path &&
2486 string_format(buf, TREE_UP_FORMAT, view->ref) &&
2487 realloc_lines(view, view->line_size + 1) &&
2488 pager_read(view, buf))
2489 view->line[view->lines - 1].type = LINE_TREE_DIR;
2490 else if (*opt_path)
2491 return FALSE;
2492 }
2493
2494 /* Strip the path part ... */
2495 if (*opt_path) {
2496 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
2497 size_t striplen = strlen(opt_path);
2498 char *path = text + SIZEOF_TREE_ATTR;
2499
2500 if (pathlen > striplen)
2501 memmove(path, path + striplen,
2502 pathlen - striplen + 1);
2503 }
2504
2505 /* Skip "Directory ..." and ".." line. */
2506 for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
2507 struct line *line = &view->line[pos];
2508 char *path1 = ((char *) line->data) + SIZEOF_TREE_ATTR;
2509 char *path2 = text + SIZEOF_TREE_ATTR;
2510 int cmp = tree_compare_entry(line->type, path1, type, path2);
2511
2512 if (cmp <= 0)
2513 continue;
2514
2515 text = strdup(text);
2516 if (!text)
2517 return FALSE;
2518
2519 if (view->lines > pos)
2520 memmove(&view->line[pos + 1], &view->line[pos],
2521 (view->lines - pos) * sizeof(*line));
2522
2523 line = &view->line[pos];
2524 line->data = text;
2525 line->type = type;
2526 view->lines++;
2527 return TRUE;
2528 }
2529
2530 if (!pager_read(view, text))
2531 return FALSE;
2532
2533 /* Move the current line to the first tree entry. */
2534 if (first_read)
2535 view->lineno++;
2536
2537 view->line[view->lines - 1].type = type;
2538 return TRUE;
2539 }
2540
2541 static bool
2542 tree_enter(struct view *view, struct line *line)
2543 {
2544 enum open_flags flags = OPEN_DEFAULT;
2545 char *data = line->data;
2546 enum request request;
2547
2548 switch (line->type) {
2549 case LINE_TREE_DIR:
2550 /* Depending on whether it is a subdir or parent (updir?) link
2551 * mangle the path buffer. */
2552 if (line == &view->line[1] && *opt_path) {
2553 size_t path_len = strlen(opt_path);
2554 char *dirsep = opt_path + path_len - 1;
2555
2556 while (dirsep > opt_path && dirsep[-1] != '/')
2557 dirsep--;
2558
2559 dirsep[0] = 0;
2560
2561 } else {
2562 size_t pathlen = strlen(opt_path);
2563 size_t origlen = pathlen;
2564 char *basename = data + SIZEOF_TREE_ATTR;
2565
2566 if (!string_format_from(opt_path, &pathlen, "%s/", basename)) {
2567 opt_path[origlen] = 0;
2568 return TRUE;
2569 }
2570 }
2571
2572 /* Trees and subtrees share the same ID, so they are not not
2573 * unique like blobs. */
2574 flags |= OPEN_RELOAD;
2575 request = REQ_VIEW_TREE;
2576 break;
2577
2578 case LINE_TREE_FILE:
2579 /* This causes the blob view to become split, and not having it
2580 * in the tree dir case will make the blob view automatically
2581 * disappear when moving to a different directory. */
2582 flags |= OPEN_SPLIT;
2583 request = REQ_VIEW_BLOB;
2584 break;
2585
2586 default:
2587 return TRUE;
2588 }
2589
2590 open_view(view, request, flags);
2591
2592 if (!VIEW(request)->pipe)
2593 return TRUE;
2594
2595 /* For tree views insert the path to the parent as the first line. */
2596 if (request == REQ_VIEW_BLOB) {
2597 /* Mirror what is showed in the title bar. */
2598 string_ncopy(ref_blob, data + STRING_SIZE("100644 blob "), 40);
2599 string_copy(VIEW(REQ_VIEW_BLOB)->ref, ref_blob);
2600 return TRUE;
2601 }
2602
2603 return TRUE;
2604 }
2605
2606 static struct view_ops tree_ops = {
2607 "file",
2608 pager_draw,
2609 tree_read,
2610 tree_enter,
2611 pager_grep,
2612 };
2613
2614 static bool
2615 blob_read(struct view *view, char *line)
2616 {
2617 bool state = pager_read(view, line);
2618
2619 if (state == TRUE)
2620 view->line[view->lines - 1].type = LINE_DEFAULT;
2621
2622 return state;
2623 }
2624
2625 static struct view_ops blob_ops = {
2626 "line",
2627 pager_draw,
2628 blob_read,
2629 pager_enter,
2630 pager_grep,
2631 };
2632
2633
2634 /*
2635 * Main view backend
2636 */
2637
2638 struct commit {
2639 char id[41]; /* SHA1 ID. */
2640 char title[75]; /* First line of the commit message. */
2641 char author[75]; /* Author of the commit. */
2642 struct tm time; /* Date from the author ident. */
2643 struct ref **refs; /* Repository references. */
2644 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
2645 size_t graph_size; /* The width of the graph array. */
2646 };
2647
2648 static bool
2649 main_draw(struct view *view, struct line *line, unsigned int lineno)
2650 {
2651 char buf[DATE_COLS + 1];
2652 struct commit *commit = line->data;
2653 enum line_type type;
2654 int col = 0;
2655 size_t timelen;
2656 size_t authorlen;
2657 int trimmed = 1;
2658
2659 if (!*commit->author)
2660 return FALSE;
2661
2662 wmove(view->win, lineno, col);
2663
2664 if (view->offset + lineno == view->lineno) {
2665 string_copy(view->ref, commit->id);
2666 string_copy(ref_commit, view->ref);
2667 type = LINE_CURSOR;
2668 wattrset(view->win, get_line_attr(type));
2669 wchgat(view->win, -1, 0, type, NULL);
2670
2671 } else {
2672 type = LINE_MAIN_COMMIT;
2673 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
2674 }
2675
2676 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
2677 waddnstr(view->win, buf, timelen);
2678 waddstr(view->win, " ");
2679
2680 col += DATE_COLS;
2681 wmove(view->win, lineno, col);
2682 if (type != LINE_CURSOR)
2683 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
2684
2685 if (opt_utf8) {
2686 authorlen = utf8_length(commit->author, AUTHOR_COLS - 2, &col, &trimmed);
2687 } else {
2688 authorlen = strlen(commit->author);
2689 if (authorlen > AUTHOR_COLS - 2) {
2690 authorlen = AUTHOR_COLS - 2;
2691 trimmed = 1;
2692 }
2693 }
2694
2695 if (trimmed) {
2696 waddnstr(view->win, commit->author, authorlen);
2697 if (type != LINE_CURSOR)
2698 wattrset(view->win, get_line_attr(LINE_MAIN_DELIM));
2699 waddch(view->win, '~');
2700 } else {
2701 waddstr(view->win, commit->author);
2702 }
2703
2704 col += AUTHOR_COLS;
2705 if (type != LINE_CURSOR)
2706 wattrset(view->win, A_NORMAL);
2707
2708 if (opt_rev_graph && commit->graph_size) {
2709 size_t i;
2710
2711 wmove(view->win, lineno, col);
2712 /* Using waddch() instead of waddnstr() ensures that
2713 * they'll be rendered correctly for the cursor line. */
2714 for (i = 0; i < commit->graph_size; i++)
2715 waddch(view->win, commit->graph[i]);
2716
2717 col += commit->graph_size + 1;
2718 }
2719
2720 wmove(view->win, lineno, col);
2721
2722 if (commit->refs) {
2723 size_t i = 0;
2724
2725 do {
2726 if (type == LINE_CURSOR)
2727 ;
2728 else if (commit->refs[i]->tag)
2729 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
2730 else
2731 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
2732 waddstr(view->win, "[");
2733 waddstr(view->win, commit->refs[i]->name);
2734 waddstr(view->win, "]");
2735 if (type != LINE_CURSOR)
2736 wattrset(view->win, A_NORMAL);
2737 waddstr(view->win, " ");
2738 col += strlen(commit->refs[i]->name) + STRING_SIZE("[] ");
2739 } while (commit->refs[i++]->next);
2740 }
2741
2742 if (type != LINE_CURSOR)
2743 wattrset(view->win, get_line_attr(type));
2744
2745 {
2746 int titlelen = strlen(commit->title);
2747
2748 if (col + titlelen > view->width)
2749 titlelen = view->width - col;
2750
2751 waddnstr(view->win, commit->title, titlelen);
2752 }
2753
2754 return TRUE;
2755 }
2756
2757 /* Reads git log --pretty=raw output and parses it into the commit struct. */
2758 static bool
2759 main_read(struct view *view, char *line)
2760 {
2761 enum line_type type = get_line_type(line);
2762 struct commit *commit = view->lines
2763 ? view->line[view->lines - 1].data : NULL;
2764
2765 switch (type) {
2766 case LINE_COMMIT:
2767 commit = calloc(1, sizeof(struct commit));
2768 if (!commit)
2769 return FALSE;
2770
2771 line += STRING_SIZE("commit ");
2772
2773 view->line[view->lines++].data = commit;
2774 string_copy(commit->id, line);
2775 commit->refs = get_refs(commit->id);
2776 commit->graph[commit->graph_size++] = ACS_LTEE;
2777 break;
2778
2779 case LINE_AUTHOR:
2780 {
2781 char *ident = line + STRING_SIZE("author ");
2782 char *end = strchr(ident, '<');
2783
2784 if (!commit)
2785 break;
2786
2787 if (end) {
2788 char *email = end + 1;
2789
2790 for (; end > ident && isspace(end[-1]); end--) ;
2791
2792 if (end == ident && *email) {
2793 ident = email;
2794 end = strchr(ident, '>');
2795 for (; end > ident && isspace(end[-1]); end--) ;
2796 }
2797 *end = 0;
2798 }
2799
2800 /* End is NULL or ident meaning there's no author. */
2801 if (end <= ident)
2802 ident = "Unknown";
2803
2804 string_copy(commit->author, ident);
2805
2806 /* Parse epoch and timezone */
2807 if (end) {
2808 char *secs = strchr(end + 1, '>');
2809 char *zone;
2810 time_t time;
2811
2812 if (!secs || secs[1] != ' ')
2813 break;
2814
2815 secs += 2;
2816 time = (time_t) atol(secs);
2817 zone = strchr(secs, ' ');
2818 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
2819 long tz;
2820
2821 zone++;
2822 tz = ('0' - zone[1]) * 60 * 60 * 10;
2823 tz += ('0' - zone[2]) * 60 * 60;
2824 tz += ('0' - zone[3]) * 60;
2825 tz += ('0' - zone[4]) * 60;
2826
2827 if (zone[0] == '-')
2828 tz = -tz;
2829
2830 time -= tz;
2831 }
2832 gmtime_r(&time, &commit->time);
2833 }
2834 break;
2835 }
2836 default:
2837 if (!commit)
2838 break;
2839
2840 /* Fill in the commit title if it has not already been set. */
2841 if (commit->title[0])
2842 break;
2843
2844 /* Require titles to start with a non-space character at the
2845 * offset used by git log. */
2846 /* FIXME: More gracefull handling of titles; append "..." to
2847 * shortened titles, etc. */
2848 if (strncmp(line, " ", 4) ||
2849 isspace(line[4]))
2850 break;
2851
2852 string_copy(commit->title, line + 4);
2853 }
2854
2855 return TRUE;
2856 }
2857
2858 static bool
2859 main_enter(struct view *view, struct line *line)
2860 {
2861 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2862
2863 open_view(view, REQ_VIEW_DIFF, flags);
2864 return TRUE;
2865 }
2866
2867 static bool
2868 main_grep(struct view *view, struct line *line)
2869 {
2870 struct commit *commit = line->data;
2871 enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
2872 char buf[DATE_COLS + 1];
2873 regmatch_t pmatch;
2874
2875 for (state = S_TITLE; state < S_END; state++) {
2876 char *text;
2877
2878 switch (state) {
2879 case S_TITLE: text = commit->title; break;
2880 case S_AUTHOR: text = commit->author; break;
2881 case S_DATE:
2882 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
2883 continue;
2884 text = buf;
2885 break;
2886
2887 default:
2888 return FALSE;
2889 }
2890
2891 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
2892 return TRUE;
2893 }
2894
2895 return FALSE;
2896 }
2897
2898 static struct view_ops main_ops = {
2899 "commit",
2900 main_draw,
2901 main_read,
2902 main_enter,
2903 main_grep,
2904 };
2905
2906
2907 /*
2908 * Unicode / UTF-8 handling
2909 *
2910 * NOTE: Much of the following code for dealing with unicode is derived from
2911 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
2912 * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
2913 */
2914
2915 /* I've (over)annotated a lot of code snippets because I am not entirely
2916 * confident that the approach taken by this small UTF-8 interface is correct.
2917 * --jonas */
2918
2919 static inline int
2920 unicode_width(unsigned long c)
2921 {
2922 if (c >= 0x1100 &&
2923 (c <= 0x115f /* Hangul Jamo */
2924 || c == 0x2329
2925 || c == 0x232a
2926 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
2927 /* CJK ... Yi */
2928 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
2929 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
2930 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
2931 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
2932 || (c >= 0xffe0 && c <= 0xffe6)
2933 || (c >= 0x20000 && c <= 0x2fffd)
2934 || (c >= 0x30000 && c <= 0x3fffd)))
2935 return 2;
2936
2937 return 1;
2938 }
2939
2940 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
2941 * Illegal bytes are set one. */
2942 static const unsigned char utf8_bytes[256] = {
2943 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,
2944 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,
2945 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,
2946 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,
2947 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,
2948 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,
2949 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,
2950 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,
2951 };
2952
2953 /* Decode UTF-8 multi-byte representation into a unicode character. */
2954 static inline unsigned long
2955 utf8_to_unicode(const char *string, size_t length)
2956 {
2957 unsigned long unicode;
2958
2959 switch (length) {
2960 case 1:
2961 unicode = string[0];
2962 break;
2963 case 2:
2964 unicode = (string[0] & 0x1f) << 6;
2965 unicode += (string[1] & 0x3f);
2966 break;
2967 case 3:
2968 unicode = (string[0] & 0x0f) << 12;
2969 unicode += ((string[1] & 0x3f) << 6);
2970 unicode += (string[2] & 0x3f);
2971 break;
2972 case 4:
2973 unicode = (string[0] & 0x0f) << 18;
2974 unicode += ((string[1] & 0x3f) << 12);
2975 unicode += ((string[2] & 0x3f) << 6);
2976 unicode += (string[3] & 0x3f);
2977 break;
2978 case 5:
2979 unicode = (string[0] & 0x0f) << 24;
2980 unicode += ((string[1] & 0x3f) << 18);
2981 unicode += ((string[2] & 0x3f) << 12);
2982 unicode += ((string[3] & 0x3f) << 6);
2983 unicode += (string[4] & 0x3f);
2984 break;
2985 case 6:
2986 unicode = (string[0] & 0x01) << 30;
2987 unicode += ((string[1] & 0x3f) << 24);
2988 unicode += ((string[2] & 0x3f) << 18);
2989 unicode += ((string[3] & 0x3f) << 12);
2990 unicode += ((string[4] & 0x3f) << 6);
2991 unicode += (string[5] & 0x3f);
2992 break;
2993 default:
2994 die("Invalid unicode length");
2995 }
2996
2997 /* Invalid characters could return the special 0xfffd value but NUL
2998 * should be just as good. */
2999 return unicode > 0xffff ? 0 : unicode;
3000 }
3001
3002 /* Calculates how much of string can be shown within the given maximum width
3003 * and sets trimmed parameter to non-zero value if all of string could not be
3004 * shown.
3005 *
3006 * Additionally, adds to coloffset how many many columns to move to align with
3007 * the expected position. Takes into account how multi-byte and double-width
3008 * characters will effect the cursor position.
3009 *
3010 * Returns the number of bytes to output from string to satisfy max_width. */
3011 static size_t
3012 utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed)
3013 {
3014 const char *start = string;
3015 const char *end = strchr(string, '\0');
3016 size_t mbwidth = 0;
3017 size_t width = 0;
3018
3019 *trimmed = 0;
3020
3021 while (string < end) {
3022 int c = *(unsigned char *) string;
3023 unsigned char bytes = utf8_bytes[c];
3024 size_t ucwidth;
3025 unsigned long unicode;
3026
3027 if (string + bytes > end)
3028 break;
3029
3030 /* Change representation to figure out whether
3031 * it is a single- or double-width character. */
3032
3033 unicode = utf8_to_unicode(string, bytes);
3034 /* FIXME: Graceful handling of invalid unicode character. */
3035 if (!unicode)
3036 break;
3037
3038 ucwidth = unicode_width(unicode);
3039 width += ucwidth;
3040 if (width > max_width) {
3041 *trimmed = 1;
3042 break;
3043 }
3044
3045 /* The column offset collects the differences between the
3046 * number of bytes encoding a character and the number of
3047 * columns will be used for rendering said character.
3048 *
3049 * So if some character A is encoded in 2 bytes, but will be
3050 * represented on the screen using only 1 byte this will and up
3051 * adding 1 to the multi-byte column offset.
3052 *
3053 * Assumes that no double-width character can be encoding in
3054 * less than two bytes. */
3055 if (bytes > ucwidth)
3056 mbwidth += bytes - ucwidth;
3057
3058 string += bytes;
3059 }
3060
3061 *coloffset += mbwidth;
3062
3063 return string - start;
3064 }
3065
3066
3067 /*
3068 * Status management
3069 */
3070
3071 /* Whether or not the curses interface has been initialized. */
3072 static bool cursed = FALSE;
3073
3074 /* The status window is used for polling keystrokes. */
3075 static WINDOW *status_win;
3076
3077 /* Update status and title window. */
3078 static void
3079 report(const char *msg, ...)
3080 {
3081 static bool empty = TRUE;
3082 struct view *view = display[current_view];
3083
3084 if (!empty || *msg) {
3085 va_list args;
3086
3087 va_start(args, msg);
3088
3089 werase(status_win);
3090 wmove(status_win, 0, 0);
3091 if (*msg) {
3092 vwprintw(status_win, msg, args);
3093 empty = FALSE;
3094 } else {
3095 empty = TRUE;
3096 }
3097 wrefresh(status_win);
3098
3099 va_end(args);
3100 }
3101
3102 update_view_title(view);
3103 update_display_cursor();
3104 }
3105
3106 /* Controls when nodelay should be in effect when polling user input. */
3107 static void
3108 set_nonblocking_input(bool loading)
3109 {
3110 static unsigned int loading_views;
3111
3112 if ((loading == FALSE && loading_views-- == 1) ||
3113 (loading == TRUE && loading_views++ == 0))
3114 nodelay(status_win, loading);
3115 }
3116
3117 static void
3118 init_display(void)
3119 {
3120 int x, y;
3121
3122 /* Initialize the curses library */
3123 if (isatty(STDIN_FILENO)) {
3124 cursed = !!initscr();
3125 } else {
3126 /* Leave stdin and stdout alone when acting as a pager. */
3127 FILE *io = fopen("/dev/tty", "r+");
3128
3129 if (!io)
3130 die("Failed to open /dev/tty");
3131 cursed = !!newterm(NULL, io, io);
3132 }
3133
3134 if (!cursed)
3135 die("Failed to initialize curses");
3136
3137 nonl(); /* Tell curses not to do NL->CR/NL on output */
3138 cbreak(); /* Take input chars one at a time, no wait for \n */
3139 noecho(); /* Don't echo input */
3140 leaveok(stdscr, TRUE);
3141
3142 if (has_colors())
3143 init_colors();
3144
3145 getmaxyx(stdscr, y, x);
3146 status_win = newwin(1, 0, y - 1, 0);
3147 if (!status_win)
3148 die("Failed to create status window");
3149
3150 /* Enable keyboard mapping */
3151 keypad(status_win, TRUE);
3152 wbkgdset(status_win, get_line_attr(LINE_STATUS));
3153 }
3154
3155 static char *
3156 read_prompt(const char *prompt)
3157 {
3158 enum { READING, STOP, CANCEL } status = READING;
3159 static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
3160 int pos = 0;
3161
3162 while (status == READING) {
3163 struct view *view;
3164 int i, key;
3165
3166 foreach_view (view, i)
3167 update_view(view);
3168
3169 report("%s%.*s", prompt, pos, buf);
3170 /* Refresh, accept single keystroke of input */
3171 key = wgetch(status_win);
3172 switch (key) {
3173 case KEY_RETURN:
3174 case KEY_ENTER:
3175 case '\n':
3176 status = pos ? STOP : CANCEL;
3177 break;
3178
3179 case KEY_BACKSPACE:
3180 if (pos > 0)
3181 pos--;
3182 else
3183 status = CANCEL;
3184 break;
3185
3186 case KEY_ESC:
3187 status = CANCEL;
3188 break;
3189
3190 case ERR:
3191 break;
3192
3193 default:
3194 if (pos >= sizeof(buf)) {
3195 report("Input string too long");
3196 return NULL;
3197 }
3198
3199 if (isprint(key))
3200 buf[pos++] = (char) key;
3201 }
3202 }
3203
3204 if (status == CANCEL) {
3205 /* Clear the status window */
3206 report("");
3207 return NULL;
3208 }
3209
3210 buf[pos++] = 0;
3211
3212 return buf;
3213 }
3214
3215 /*
3216 * Repository references
3217 */
3218
3219 static struct ref *refs;
3220 static size_t refs_size;
3221
3222 /* Id <-> ref store */
3223 static struct ref ***id_refs;
3224 static size_t id_refs_size;
3225
3226 static struct ref **
3227 get_refs(char *id)
3228 {
3229 struct ref ***tmp_id_refs;
3230 struct ref **ref_list = NULL;
3231 size_t ref_list_size = 0;
3232 size_t i;
3233
3234 for (i = 0; i < id_refs_size; i++)
3235 if (!strcmp(id, id_refs[i][0]->id))
3236 return id_refs[i];
3237
3238 tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
3239 if (!tmp_id_refs)
3240 return NULL;
3241
3242 id_refs = tmp_id_refs;
3243
3244 for (i = 0; i < refs_size; i++) {
3245 struct ref **tmp;
3246
3247 if (strcmp(id, refs[i].id))
3248 continue;
3249
3250 tmp = realloc(ref_list, (ref_list_size + 1) * sizeof(*ref_list));
3251 if (!tmp) {
3252 if (ref_list)
3253 free(ref_list);
3254 return NULL;
3255 }
3256
3257 ref_list = tmp;
3258 if (ref_list_size > 0)
3259 ref_list[ref_list_size - 1]->next = 1;
3260 ref_list[ref_list_size] = &refs[i];
3261
3262 /* XXX: The properties of the commit chains ensures that we can
3263 * safely modify the shared ref. The repo references will
3264 * always be similar for the same id. */
3265 ref_list[ref_list_size]->next = 0;
3266 ref_list_size++;
3267 }
3268
3269 if (ref_list)
3270 id_refs[id_refs_size++] = ref_list;
3271
3272 return ref_list;
3273 }
3274
3275 static int
3276 read_ref(char *id, int idlen, char *name, int namelen)
3277 {
3278 struct ref *ref;
3279 bool tag = FALSE;
3280
3281 if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
3282 /* Commits referenced by tags has "^{}" appended. */
3283 if (name[namelen - 1] != '}')
3284 return OK;
3285
3286 while (namelen > 0 && name[namelen] != '^')
3287 namelen--;
3288
3289 tag = TRUE;
3290 namelen -= STRING_SIZE("refs/tags/");
3291 name += STRING_SIZE("refs/tags/");
3292
3293 } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
3294 namelen -= STRING_SIZE("refs/heads/");
3295 name += STRING_SIZE("refs/heads/");
3296
3297 } else if (!strcmp(name, "HEAD")) {
3298 return OK;
3299 }
3300
3301 refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
3302 if (!refs)
3303 return ERR;
3304
3305 ref = &refs[refs_size++];
3306 ref->name = malloc(namelen + 1);
3307 if (!ref->name)
3308 return ERR;
3309
3310 strncpy(ref->name, name, namelen);
3311 ref->name[namelen] = 0;
3312 ref->tag = tag;
3313 string_copy(ref->id, id);
3314
3315 return OK;
3316 }
3317
3318 static int
3319 load_refs(void)
3320 {
3321 const char *cmd_env = getenv("TIG_LS_REMOTE");
3322 const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
3323
3324 return read_properties(popen(cmd, "r"), "\t", read_ref);
3325 }
3326
3327 static int
3328 read_repo_config_option(char *name, int namelen, char *value, int valuelen)
3329 {
3330 if (!strcmp(name, "i18n.commitencoding"))
3331 string_copy(opt_encoding, value);
3332
3333 return OK;
3334 }
3335
3336 static int
3337 load_repo_config(void)
3338 {
3339 return read_properties(popen("git repo-config --list", "r"),
3340 "=", read_repo_config_option);
3341 }
3342
3343 static int
3344 read_properties(FILE *pipe, const char *separators,
3345 int (*read_property)(char *, int, char *, int))
3346 {
3347 char buffer[BUFSIZ];
3348 char *name;
3349 int state = OK;
3350
3351 if (!pipe)
3352 return ERR;
3353
3354 while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
3355 char *value;
3356 size_t namelen;
3357 size_t valuelen;
3358
3359 name = chomp_string(name);
3360 namelen = strcspn(name, separators);
3361
3362 if (name[namelen]) {
3363 name[namelen] = 0;
3364 value = chomp_string(name + namelen + 1);
3365 valuelen = strlen(value);
3366
3367 } else {
3368 value = "";
3369 valuelen = 0;
3370 }
3371
3372 state = read_property(name, namelen, value, valuelen);
3373 }
3374
3375 if (state != ERR && ferror(pipe))
3376 state = ERR;
3377
3378 pclose(pipe);
3379
3380 return state;
3381 }
3382
3383
3384 /*
3385 * Main
3386 */
3387
3388 static void __NORETURN
3389 quit(int sig)
3390 {
3391 /* XXX: Restore tty modes and let the OS cleanup the rest! */
3392 if (cursed)
3393 endwin();
3394 exit(0);
3395 }
3396
3397 static void __NORETURN
3398 die(const char *err, ...)
3399 {
3400 va_list args;
3401
3402 endwin();
3403
3404 va_start(args, err);
3405 fputs("tig: ", stderr);
3406 vfprintf(stderr, err, args);
3407 fputs("\n", stderr);
3408 va_end(args);
3409
3410 exit(1);
3411 }
3412
3413 int
3414 main(int argc, char *argv[])
3415 {
3416 struct view *view;
3417 enum request request;
3418 size_t i;
3419
3420 signal(SIGINT, quit);
3421
3422 if (setlocale(LC_ALL, "")) {
3423 string_copy(opt_codeset, nl_langinfo(CODESET));
3424 }
3425
3426 if (load_options() == ERR)
3427 die("Failed to load user config.");
3428
3429 /* Load the repo config file so options can be overwritten from
3430 * the command line. */
3431 if (load_repo_config() == ERR)
3432 die("Failed to load repo config.");
3433
3434 if (!parse_options(argc, argv))
3435 return 0;
3436
3437 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
3438 opt_iconv = iconv_open(opt_codeset, opt_encoding);
3439 if (opt_iconv == ICONV_NONE)
3440 die("Failed to initialize character set conversion");
3441 }
3442
3443 if (load_refs() == ERR)
3444 die("Failed to load refs.");
3445
3446 /* Require a git repository unless when running in pager mode. */
3447 if (refs_size == 0 && opt_request != REQ_VIEW_PAGER)
3448 die("Not a git repository");
3449
3450 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
3451 view->cmd_env = getenv(view->cmd_env);
3452
3453 request = opt_request;
3454
3455 init_display();
3456
3457 while (view_driver(display[current_view], request)) {
3458 int key;
3459 int i;
3460
3461 foreach_view (view, i)
3462 update_view(view);
3463
3464 /* Refresh, accept single keystroke of input */
3465 key = wgetch(status_win);
3466
3467 request = get_keybinding(display[current_view]->keymap, key);
3468
3469 /* Some low-level request handling. This keeps access to
3470 * status_win restricted. */
3471 switch (request) {
3472 case REQ_PROMPT:
3473 {
3474 char *cmd = read_prompt(":");
3475
3476 if (cmd && string_format(opt_cmd, "git %s", cmd)) {
3477 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
3478 opt_request = REQ_VIEW_DIFF;
3479 } else {
3480 opt_request = REQ_VIEW_PAGER;
3481 }
3482 break;
3483 }
3484
3485 request = REQ_NONE;
3486 break;
3487 }
3488 case REQ_SEARCH:
3489 case REQ_SEARCH_BACK:
3490 {
3491 const char *prompt = request == REQ_SEARCH
3492 ? "/" : "?";
3493 char *search = read_prompt(prompt);
3494
3495 if (search)
3496 string_copy(opt_search, search);
3497 else
3498 request = REQ_NONE;
3499 break;
3500 }
3501 case REQ_SCREEN_RESIZE:
3502 {
3503 int height, width;
3504
3505 getmaxyx(stdscr, height, width);
3506
3507 /* Resize the status view and let the view driver take
3508 * care of resizing the displayed views. */
3509 wresize(status_win, 1, width);
3510 mvwin(status_win, height - 1, 0);
3511 wrefresh(status_win);
3512 break;
3513 }
3514 default:
3515 break;
3516 }
3517 }
3518
3519 quit(0);
3520
3521 return 0;
3522 }