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