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