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