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