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