1 /* Copyright (c) 2006-2007 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 #ifdef HAVE_CONFIG_H
15 #include "config.h"
16 #endif
18 #ifndef TIG_VERSION
19 #define TIG_VERSION "unknown-version"
20 #endif
22 #ifndef DEBUG
23 #define NDEBUG
24 #endif
26 #include <assert.h>
27 #include <errno.h>
28 #include <ctype.h>
29 #include <signal.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #include <sys/types.h>
35 #include <sys/stat.h>
36 #include <unistd.h>
37 #include <time.h>
39 #include <regex.h>
41 #include <locale.h>
42 #include <langinfo.h>
43 #include <iconv.h>
45 #include <curses.h>
47 #if __GNUC__ >= 3
48 #define __NORETURN __attribute__((__noreturn__))
49 #else
50 #define __NORETURN
51 #endif
53 static void __NORETURN die(const char *err, ...);
54 static void report(const char *msg, ...);
55 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, size_t, char *, size_t));
56 static void set_nonblocking_input(bool loading);
57 static size_t utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed);
59 #define ABS(x) ((x) >= 0 ? (x) : -(x))
60 #define MIN(x, y) ((x) < (y) ? (x) : (y))
62 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
63 #define STRING_SIZE(x) (sizeof(x) - 1)
65 #define SIZEOF_STR 1024 /* Default string size. */
66 #define SIZEOF_REF 256 /* Size of symbolic or SHA1 ID. */
67 #define SIZEOF_REV 41 /* Holds a SHA-1 and an ending NUL */
69 /* Revision graph */
71 #define REVGRAPH_INIT 'I'
72 #define REVGRAPH_MERGE 'M'
73 #define REVGRAPH_BRANCH '+'
74 #define REVGRAPH_COMMIT '*'
75 #define REVGRAPH_LINE '|'
77 #define SIZEOF_REVGRAPH 19 /* Size of revision ancestry graphics. */
79 /* This color name can be used to refer to the default term colors. */
80 #define COLOR_DEFAULT (-1)
82 #define ICONV_NONE ((iconv_t) -1)
83 #ifndef ICONV_CONST
84 #define ICONV_CONST /* nothing */
85 #endif
87 /* The format and size of the date column in the main view. */
88 #define DATE_FORMAT "%Y-%m-%d %H:%M"
89 #define DATE_COLS STRING_SIZE("2006-04-29 14:21 ")
91 #define AUTHOR_COLS 20
93 /* The default interval between line numbers. */
94 #define NUMBER_INTERVAL 1
96 #define TABSIZE 8
98 #define SCALE_SPLIT_VIEW(height) ((height) * 2 / 3)
100 #ifndef GIT_CONFIG
101 #define GIT_CONFIG "git config"
102 #endif
104 #define TIG_LS_REMOTE \
105 "git ls-remote $(git rev-parse --git-dir) 2>/dev/null"
107 #define TIG_DIFF_CMD \
108 "git show --root --patch-with-stat --find-copies-harder -B -C %s 2>/dev/null"
110 #define TIG_LOG_CMD \
111 "git log --cc --stat -n100 %s 2>/dev/null"
113 #define TIG_MAIN_CMD \
114 "git log --topo-order --pretty=raw %s 2>/dev/null"
116 #define TIG_TREE_CMD \
117 "git ls-tree %s %s"
119 #define TIG_BLOB_CMD \
120 "git cat-file blob %s"
122 /* XXX: Needs to be defined to the empty string. */
123 #define TIG_HELP_CMD ""
124 #define TIG_PAGER_CMD ""
125 #define TIG_STATUS_CMD ""
126 #define TIG_STAGE_CMD ""
128 /* Some ascii-shorthands fitted into the ncurses namespace. */
129 #define KEY_TAB '\t'
130 #define KEY_RETURN '\r'
131 #define KEY_ESC 27
134 struct ref {
135 char *name; /* Ref name; tag or head names are shortened. */
136 char id[SIZEOF_REV]; /* Commit SHA1 ID */
137 unsigned int tag:1; /* Is it a tag? */
138 unsigned int remote:1; /* Is it a remote ref? */
139 unsigned int next:1; /* For ref lists: are there more refs? */
140 };
142 static struct ref **get_refs(char *id);
144 struct int_map {
145 const char *name;
146 int namelen;
147 int value;
148 };
150 static int
151 set_from_int_map(struct int_map *map, size_t map_size,
152 int *value, const char *name, int namelen)
153 {
155 int i;
157 for (i = 0; i < map_size; i++)
158 if (namelen == map[i].namelen &&
159 !strncasecmp(name, map[i].name, namelen)) {
160 *value = map[i].value;
161 return OK;
162 }
164 return ERR;
165 }
168 /*
169 * String helpers
170 */
172 static inline void
173 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
174 {
175 if (srclen > dstlen - 1)
176 srclen = dstlen - 1;
178 strncpy(dst, src, srclen);
179 dst[srclen] = 0;
180 }
182 /* Shorthands for safely copying into a fixed buffer. */
184 #define string_copy(dst, src) \
185 string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
187 #define string_ncopy(dst, src, srclen) \
188 string_ncopy_do(dst, sizeof(dst), src, srclen)
190 #define string_copy_rev(dst, src) \
191 string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
193 #define string_add(dst, from, src) \
194 string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
196 static char *
197 chomp_string(char *name)
198 {
199 int namelen;
201 while (isspace(*name))
202 name++;
204 namelen = strlen(name) - 1;
205 while (namelen > 0 && isspace(name[namelen]))
206 name[namelen--] = 0;
208 return name;
209 }
211 static bool
212 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
213 {
214 va_list args;
215 size_t pos = bufpos ? *bufpos : 0;
217 va_start(args, fmt);
218 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
219 va_end(args);
221 if (bufpos)
222 *bufpos = pos;
224 return pos >= bufsize ? FALSE : TRUE;
225 }
227 #define string_format(buf, fmt, args...) \
228 string_nformat(buf, sizeof(buf), NULL, fmt, args)
230 #define string_format_from(buf, from, fmt, args...) \
231 string_nformat(buf, sizeof(buf), from, fmt, args)
233 static int
234 string_enum_compare(const char *str1, const char *str2, int len)
235 {
236 size_t i;
238 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
240 /* Diff-Header == DIFF_HEADER */
241 for (i = 0; i < len; i++) {
242 if (toupper(str1[i]) == toupper(str2[i]))
243 continue;
245 if (string_enum_sep(str1[i]) &&
246 string_enum_sep(str2[i]))
247 continue;
249 return str1[i] - str2[i];
250 }
252 return 0;
253 }
255 /* Shell quoting
256 *
257 * NOTE: The following is a slightly modified copy of the git project's shell
258 * quoting routines found in the quote.c file.
259 *
260 * Help to copy the thing properly quoted for the shell safety. any single
261 * quote is replaced with '\'', any exclamation point is replaced with '\!',
262 * and the whole thing is enclosed in a
263 *
264 * E.g.
265 * original sq_quote result
266 * name ==> name ==> 'name'
267 * a b ==> a b ==> 'a b'
268 * a'b ==> a'\''b ==> 'a'\''b'
269 * a!b ==> a'\!'b ==> 'a'\!'b'
270 */
272 static size_t
273 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
274 {
275 char c;
277 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
279 BUFPUT('\'');
280 while ((c = *src++)) {
281 if (c == '\'' || c == '!') {
282 BUFPUT('\'');
283 BUFPUT('\\');
284 BUFPUT(c);
285 BUFPUT('\'');
286 } else {
287 BUFPUT(c);
288 }
289 }
290 BUFPUT('\'');
292 if (bufsize < SIZEOF_STR)
293 buf[bufsize] = 0;
295 return bufsize;
296 }
299 /*
300 * User requests
301 */
303 #define REQ_INFO \
304 /* XXX: Keep the view request first and in sync with views[]. */ \
305 REQ_GROUP("View switching") \
306 REQ_(VIEW_MAIN, "Show main view"), \
307 REQ_(VIEW_DIFF, "Show diff view"), \
308 REQ_(VIEW_LOG, "Show log view"), \
309 REQ_(VIEW_TREE, "Show tree view"), \
310 REQ_(VIEW_BLOB, "Show blob view"), \
311 REQ_(VIEW_HELP, "Show help page"), \
312 REQ_(VIEW_PAGER, "Show pager view"), \
313 REQ_(VIEW_STATUS, "Show status view"), \
314 REQ_(VIEW_STAGE, "Show stage view"), \
315 \
316 REQ_GROUP("View manipulation") \
317 REQ_(ENTER, "Enter current line and scroll"), \
318 REQ_(NEXT, "Move to next"), \
319 REQ_(PREVIOUS, "Move to previous"), \
320 REQ_(VIEW_NEXT, "Move focus to next view"), \
321 REQ_(VIEW_CLOSE, "Close the current view"), \
322 REQ_(QUIT, "Close all views and quit"), \
323 \
324 REQ_GROUP("Cursor navigation") \
325 REQ_(MOVE_UP, "Move cursor one line up"), \
326 REQ_(MOVE_DOWN, "Move cursor one line down"), \
327 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
328 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
329 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
330 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
331 \
332 REQ_GROUP("Scrolling") \
333 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
334 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
335 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
336 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
337 \
338 REQ_GROUP("Searching") \
339 REQ_(SEARCH, "Search the view"), \
340 REQ_(SEARCH_BACK, "Search backwards in the view"), \
341 REQ_(FIND_NEXT, "Find next search match"), \
342 REQ_(FIND_PREV, "Find previous search match"), \
343 \
344 REQ_GROUP("Misc") \
345 REQ_(NONE, "Do nothing"), \
346 REQ_(PROMPT, "Bring up the prompt"), \
347 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
348 REQ_(SCREEN_RESIZE, "Resize the screen"), \
349 REQ_(SHOW_VERSION, "Show version information"), \
350 REQ_(STOP_LOADING, "Stop all loading views"), \
351 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
352 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization"), \
353 REQ_(STATUS_UPDATE, "Update file status"), \
354 REQ_(EDIT, "Open in editor")
357 /* User action requests. */
358 enum request {
359 #define REQ_GROUP(help)
360 #define REQ_(req, help) REQ_##req
362 /* Offset all requests to avoid conflicts with ncurses getch values. */
363 REQ_OFFSET = KEY_MAX + 1,
364 REQ_INFO,
365 REQ_UNKNOWN,
367 #undef REQ_GROUP
368 #undef REQ_
369 };
371 struct request_info {
372 enum request request;
373 char *name;
374 int namelen;
375 char *help;
376 };
378 static struct request_info req_info[] = {
379 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
380 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
381 REQ_INFO
382 #undef REQ_GROUP
383 #undef REQ_
384 };
386 static enum request
387 get_request(const char *name)
388 {
389 int namelen = strlen(name);
390 int i;
392 for (i = 0; i < ARRAY_SIZE(req_info); i++)
393 if (req_info[i].namelen == namelen &&
394 !string_enum_compare(req_info[i].name, name, namelen))
395 return req_info[i].request;
397 return REQ_UNKNOWN;
398 }
401 /*
402 * Options
403 */
405 static const char usage[] =
406 "tig " TIG_VERSION " (" __DATE__ ")\n"
407 "\n"
408 "Usage: tig [options]\n"
409 " or: tig [options] [--] [git log options]\n"
410 " or: tig [options] log [git log options]\n"
411 " or: tig [options] diff [git diff options]\n"
412 " or: tig [options] show [git show options]\n"
413 " or: tig [options] < [git command output]\n"
414 "\n"
415 "Options:\n"
416 " -l Start up in log view\n"
417 " -d Start up in diff view\n"
418 " -S Start up in status view\n"
419 " -n[I], --line-number[=I] Show line numbers with given interval\n"
420 " -b[N], --tab-size[=N] Set number of spaces for tab expansion\n"
421 " -- Mark end of tig options\n"
422 " -v, --version Show version and exit\n"
423 " -h, --help Show help message and exit\n";
425 /* Option and state variables. */
426 static bool opt_line_number = FALSE;
427 static bool opt_rev_graph = FALSE;
428 static int opt_num_interval = NUMBER_INTERVAL;
429 static int opt_tab_size = TABSIZE;
430 static enum request opt_request = REQ_VIEW_MAIN;
431 static char opt_cmd[SIZEOF_STR] = "";
432 static char opt_path[SIZEOF_STR] = "";
433 static FILE *opt_pipe = NULL;
434 static char opt_encoding[20] = "UTF-8";
435 static bool opt_utf8 = TRUE;
436 static char opt_codeset[20] = "UTF-8";
437 static iconv_t opt_iconv = ICONV_NONE;
438 static char opt_search[SIZEOF_STR] = "";
439 static char opt_cdup[SIZEOF_STR] = "";
440 static char opt_git_dir[SIZEOF_STR] = "";
441 static char opt_editor[SIZEOF_STR] = "";
443 enum option_type {
444 OPT_NONE,
445 OPT_INT,
446 };
448 static bool
449 check_option(char *opt, char short_name, char *name, enum option_type type, ...)
450 {
451 va_list args;
452 char *value = "";
453 int *number;
455 if (opt[0] != '-')
456 return FALSE;
458 if (opt[1] == '-') {
459 int namelen = strlen(name);
461 opt += 2;
463 if (strncmp(opt, name, namelen))
464 return FALSE;
466 if (opt[namelen] == '=')
467 value = opt + namelen + 1;
469 } else {
470 if (!short_name || opt[1] != short_name)
471 return FALSE;
472 value = opt + 2;
473 }
475 va_start(args, type);
476 if (type == OPT_INT) {
477 number = va_arg(args, int *);
478 if (isdigit(*value))
479 *number = atoi(value);
480 }
481 va_end(args);
483 return TRUE;
484 }
486 /* Returns the index of log or diff command or -1 to exit. */
487 static bool
488 parse_options(int argc, char *argv[])
489 {
490 int i;
492 for (i = 1; i < argc; i++) {
493 char *opt = argv[i];
495 if (!strcmp(opt, "log") ||
496 !strcmp(opt, "diff") ||
497 !strcmp(opt, "show")) {
498 opt_request = opt[0] == 'l'
499 ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
500 break;
501 }
503 if (opt[0] && opt[0] != '-')
504 break;
506 if (!strcmp(opt, "-l")) {
507 opt_request = REQ_VIEW_LOG;
508 continue;
509 }
511 if (!strcmp(opt, "-d")) {
512 opt_request = REQ_VIEW_DIFF;
513 continue;
514 }
516 if (!strcmp(opt, "-S")) {
517 opt_request = REQ_VIEW_STATUS;
518 continue;
519 }
521 if (check_option(opt, 'n', "line-number", OPT_INT, &opt_num_interval)) {
522 opt_line_number = TRUE;
523 continue;
524 }
526 if (check_option(opt, 'b', "tab-size", OPT_INT, &opt_tab_size)) {
527 opt_tab_size = MIN(opt_tab_size, TABSIZE);
528 continue;
529 }
531 if (check_option(opt, 'v', "version", OPT_NONE)) {
532 printf("tig version %s\n", TIG_VERSION);
533 return FALSE;
534 }
536 if (check_option(opt, 'h', "help", OPT_NONE)) {
537 printf(usage);
538 return FALSE;
539 }
541 if (!strcmp(opt, "--")) {
542 i++;
543 break;
544 }
546 die("unknown option '%s'\n\n%s", opt, usage);
547 }
549 if (!isatty(STDIN_FILENO)) {
550 opt_request = REQ_VIEW_PAGER;
551 opt_pipe = stdin;
553 } else if (i < argc) {
554 size_t buf_size;
556 if (opt_request == REQ_VIEW_MAIN)
557 /* XXX: This is vulnerable to the user overriding
558 * options required for the main view parser. */
559 string_copy(opt_cmd, "git log --pretty=raw");
560 else
561 string_copy(opt_cmd, "git");
562 buf_size = strlen(opt_cmd);
564 while (buf_size < sizeof(opt_cmd) && i < argc) {
565 opt_cmd[buf_size++] = ' ';
566 buf_size = sq_quote(opt_cmd, buf_size, argv[i++]);
567 }
569 if (buf_size >= sizeof(opt_cmd))
570 die("command too long");
572 opt_cmd[buf_size] = 0;
573 }
575 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
576 opt_utf8 = FALSE;
578 return TRUE;
579 }
582 /*
583 * Line-oriented content detection.
584 */
586 #define LINE_INFO \
587 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
588 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
589 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
590 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
591 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
592 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
593 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
594 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
595 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
596 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
597 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
598 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
599 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
600 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
601 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
602 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
603 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
604 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
605 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
606 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
607 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
608 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
609 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
610 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
611 LINE(AUTHOR, "author ", COLOR_CYAN, COLOR_DEFAULT, 0), \
612 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
613 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
614 LINE(ACKED, " Acked-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
615 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
616 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
617 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
618 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
619 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
620 LINE(MAIN_DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
621 LINE(MAIN_AUTHOR, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
622 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
623 LINE(MAIN_DELIM, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
624 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
625 LINE(MAIN_REMOTE, "", COLOR_YELLOW, COLOR_DEFAULT, A_BOLD), \
626 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
627 LINE(TREE_DIR, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
628 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
629 LINE(STAT_SECTION, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
630 LINE(STAT_NONE, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
631 LINE(STAT_STAGED, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
632 LINE(STAT_UNSTAGED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
633 LINE(STAT_UNTRACKED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0)
635 enum line_type {
636 #define LINE(type, line, fg, bg, attr) \
637 LINE_##type
638 LINE_INFO
639 #undef LINE
640 };
642 struct line_info {
643 const char *name; /* Option name. */
644 int namelen; /* Size of option name. */
645 const char *line; /* The start of line to match. */
646 int linelen; /* Size of string to match. */
647 int fg, bg, attr; /* Color and text attributes for the lines. */
648 };
650 static struct line_info line_info[] = {
651 #define LINE(type, line, fg, bg, attr) \
652 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
653 LINE_INFO
654 #undef LINE
655 };
657 static enum line_type
658 get_line_type(char *line)
659 {
660 int linelen = strlen(line);
661 enum line_type type;
663 for (type = 0; type < ARRAY_SIZE(line_info); type++)
664 /* Case insensitive search matches Signed-off-by lines better. */
665 if (linelen >= line_info[type].linelen &&
666 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
667 return type;
669 return LINE_DEFAULT;
670 }
672 static inline int
673 get_line_attr(enum line_type type)
674 {
675 assert(type < ARRAY_SIZE(line_info));
676 return COLOR_PAIR(type) | line_info[type].attr;
677 }
679 static struct line_info *
680 get_line_info(char *name, int namelen)
681 {
682 enum line_type type;
684 for (type = 0; type < ARRAY_SIZE(line_info); type++)
685 if (namelen == line_info[type].namelen &&
686 !string_enum_compare(line_info[type].name, name, namelen))
687 return &line_info[type];
689 return NULL;
690 }
692 static void
693 init_colors(void)
694 {
695 int default_bg = COLOR_BLACK;
696 int default_fg = COLOR_WHITE;
697 enum line_type type;
699 start_color();
701 if (use_default_colors() != ERR) {
702 default_bg = -1;
703 default_fg = -1;
704 }
706 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
707 struct line_info *info = &line_info[type];
708 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
709 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
711 init_pair(type, fg, bg);
712 }
713 }
715 struct line {
716 enum line_type type;
718 /* State flags */
719 unsigned int selected:1;
721 void *data; /* User data */
722 };
725 /*
726 * Keys
727 */
729 struct keybinding {
730 int alias;
731 enum request request;
732 struct keybinding *next;
733 };
735 static struct keybinding default_keybindings[] = {
736 /* View switching */
737 { 'm', REQ_VIEW_MAIN },
738 { 'd', REQ_VIEW_DIFF },
739 { 'l', REQ_VIEW_LOG },
740 { 't', REQ_VIEW_TREE },
741 { 'f', REQ_VIEW_BLOB },
742 { 'p', REQ_VIEW_PAGER },
743 { 'h', REQ_VIEW_HELP },
744 { 'S', REQ_VIEW_STATUS },
745 { 'c', REQ_VIEW_STAGE },
747 /* View manipulation */
748 { 'q', REQ_VIEW_CLOSE },
749 { KEY_TAB, REQ_VIEW_NEXT },
750 { KEY_RETURN, REQ_ENTER },
751 { KEY_UP, REQ_PREVIOUS },
752 { KEY_DOWN, REQ_NEXT },
754 /* Cursor navigation */
755 { 'k', REQ_MOVE_UP },
756 { 'j', REQ_MOVE_DOWN },
757 { KEY_HOME, REQ_MOVE_FIRST_LINE },
758 { KEY_END, REQ_MOVE_LAST_LINE },
759 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
760 { ' ', REQ_MOVE_PAGE_DOWN },
761 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
762 { 'b', REQ_MOVE_PAGE_UP },
763 { '-', REQ_MOVE_PAGE_UP },
765 /* Scrolling */
766 { KEY_IC, REQ_SCROLL_LINE_UP },
767 { KEY_DC, REQ_SCROLL_LINE_DOWN },
768 { 'w', REQ_SCROLL_PAGE_UP },
769 { 's', REQ_SCROLL_PAGE_DOWN },
771 /* Searching */
772 { '/', REQ_SEARCH },
773 { '?', REQ_SEARCH_BACK },
774 { 'n', REQ_FIND_NEXT },
775 { 'N', REQ_FIND_PREV },
777 /* Misc */
778 { 'Q', REQ_QUIT },
779 { 'z', REQ_STOP_LOADING },
780 { 'v', REQ_SHOW_VERSION },
781 { 'r', REQ_SCREEN_REDRAW },
782 { '.', REQ_TOGGLE_LINENO },
783 { 'g', REQ_TOGGLE_REV_GRAPH },
784 { ':', REQ_PROMPT },
785 { 'u', REQ_STATUS_UPDATE },
786 { 'e', REQ_EDIT },
788 /* Using the ncurses SIGWINCH handler. */
789 { KEY_RESIZE, REQ_SCREEN_RESIZE },
790 };
792 #define KEYMAP_INFO \
793 KEYMAP_(GENERIC), \
794 KEYMAP_(MAIN), \
795 KEYMAP_(DIFF), \
796 KEYMAP_(LOG), \
797 KEYMAP_(TREE), \
798 KEYMAP_(BLOB), \
799 KEYMAP_(PAGER), \
800 KEYMAP_(HELP), \
801 KEYMAP_(STATUS), \
802 KEYMAP_(STAGE)
804 enum keymap {
805 #define KEYMAP_(name) KEYMAP_##name
806 KEYMAP_INFO
807 #undef KEYMAP_
808 };
810 static struct int_map keymap_table[] = {
811 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
812 KEYMAP_INFO
813 #undef KEYMAP_
814 };
816 #define set_keymap(map, name) \
817 set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
819 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
821 static void
822 add_keybinding(enum keymap keymap, enum request request, int key)
823 {
824 struct keybinding *keybinding;
826 keybinding = calloc(1, sizeof(*keybinding));
827 if (!keybinding)
828 die("Failed to allocate keybinding");
830 keybinding->alias = key;
831 keybinding->request = request;
832 keybinding->next = keybindings[keymap];
833 keybindings[keymap] = keybinding;
834 }
836 /* Looks for a key binding first in the given map, then in the generic map, and
837 * lastly in the default keybindings. */
838 static enum request
839 get_keybinding(enum keymap keymap, int key)
840 {
841 struct keybinding *kbd;
842 int i;
844 for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
845 if (kbd->alias == key)
846 return kbd->request;
848 for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
849 if (kbd->alias == key)
850 return kbd->request;
852 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
853 if (default_keybindings[i].alias == key)
854 return default_keybindings[i].request;
856 return (enum request) key;
857 }
860 struct key {
861 char *name;
862 int value;
863 };
865 static struct key key_table[] = {
866 { "Enter", KEY_RETURN },
867 { "Space", ' ' },
868 { "Backspace", KEY_BACKSPACE },
869 { "Tab", KEY_TAB },
870 { "Escape", KEY_ESC },
871 { "Left", KEY_LEFT },
872 { "Right", KEY_RIGHT },
873 { "Up", KEY_UP },
874 { "Down", KEY_DOWN },
875 { "Insert", KEY_IC },
876 { "Delete", KEY_DC },
877 { "Hash", '#' },
878 { "Home", KEY_HOME },
879 { "End", KEY_END },
880 { "PageUp", KEY_PPAGE },
881 { "PageDown", KEY_NPAGE },
882 { "F1", KEY_F(1) },
883 { "F2", KEY_F(2) },
884 { "F3", KEY_F(3) },
885 { "F4", KEY_F(4) },
886 { "F5", KEY_F(5) },
887 { "F6", KEY_F(6) },
888 { "F7", KEY_F(7) },
889 { "F8", KEY_F(8) },
890 { "F9", KEY_F(9) },
891 { "F10", KEY_F(10) },
892 { "F11", KEY_F(11) },
893 { "F12", KEY_F(12) },
894 };
896 static int
897 get_key_value(const char *name)
898 {
899 int i;
901 for (i = 0; i < ARRAY_SIZE(key_table); i++)
902 if (!strcasecmp(key_table[i].name, name))
903 return key_table[i].value;
905 if (strlen(name) == 1 && isprint(*name))
906 return (int) *name;
908 return ERR;
909 }
911 static char *
912 get_key(enum request request)
913 {
914 static char buf[BUFSIZ];
915 static char key_char[] = "'X'";
916 size_t pos = 0;
917 char *sep = "";
918 int i;
920 buf[pos] = 0;
922 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
923 struct keybinding *keybinding = &default_keybindings[i];
924 char *seq = NULL;
925 int key;
927 if (keybinding->request != request)
928 continue;
930 for (key = 0; key < ARRAY_SIZE(key_table); key++)
931 if (key_table[key].value == keybinding->alias)
932 seq = key_table[key].name;
934 if (seq == NULL &&
935 keybinding->alias < 127 &&
936 isprint(keybinding->alias)) {
937 key_char[1] = (char) keybinding->alias;
938 seq = key_char;
939 }
941 if (!seq)
942 seq = "'?'";
944 if (!string_format_from(buf, &pos, "%s%s", sep, seq))
945 return "Too many keybindings!";
946 sep = ", ";
947 }
949 return buf;
950 }
953 /*
954 * User config file handling.
955 */
957 static struct int_map color_map[] = {
958 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
959 COLOR_MAP(DEFAULT),
960 COLOR_MAP(BLACK),
961 COLOR_MAP(BLUE),
962 COLOR_MAP(CYAN),
963 COLOR_MAP(GREEN),
964 COLOR_MAP(MAGENTA),
965 COLOR_MAP(RED),
966 COLOR_MAP(WHITE),
967 COLOR_MAP(YELLOW),
968 };
970 #define set_color(color, name) \
971 set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
973 static struct int_map attr_map[] = {
974 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
975 ATTR_MAP(NORMAL),
976 ATTR_MAP(BLINK),
977 ATTR_MAP(BOLD),
978 ATTR_MAP(DIM),
979 ATTR_MAP(REVERSE),
980 ATTR_MAP(STANDOUT),
981 ATTR_MAP(UNDERLINE),
982 };
984 #define set_attribute(attr, name) \
985 set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
987 static int config_lineno;
988 static bool config_errors;
989 static char *config_msg;
991 /* Wants: object fgcolor bgcolor [attr] */
992 static int
993 option_color_command(int argc, char *argv[])
994 {
995 struct line_info *info;
997 if (argc != 3 && argc != 4) {
998 config_msg = "Wrong number of arguments given to color command";
999 return ERR;
1000 }
1002 info = get_line_info(argv[0], strlen(argv[0]));
1003 if (!info) {
1004 config_msg = "Unknown color name";
1005 return ERR;
1006 }
1008 if (set_color(&info->fg, argv[1]) == ERR ||
1009 set_color(&info->bg, argv[2]) == ERR) {
1010 config_msg = "Unknown color";
1011 return ERR;
1012 }
1014 if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
1015 config_msg = "Unknown attribute";
1016 return ERR;
1017 }
1019 return OK;
1020 }
1022 /* Wants: name = value */
1023 static int
1024 option_set_command(int argc, char *argv[])
1025 {
1026 if (argc != 3) {
1027 config_msg = "Wrong number of arguments given to set command";
1028 return ERR;
1029 }
1031 if (strcmp(argv[1], "=")) {
1032 config_msg = "No value assigned";
1033 return ERR;
1034 }
1036 if (!strcmp(argv[0], "show-rev-graph")) {
1037 opt_rev_graph = (!strcmp(argv[2], "1") ||
1038 !strcmp(argv[2], "true") ||
1039 !strcmp(argv[2], "yes"));
1040 return OK;
1041 }
1043 if (!strcmp(argv[0], "line-number-interval")) {
1044 opt_num_interval = atoi(argv[2]);
1045 return OK;
1046 }
1048 if (!strcmp(argv[0], "tab-size")) {
1049 opt_tab_size = atoi(argv[2]);
1050 return OK;
1051 }
1053 if (!strcmp(argv[0], "commit-encoding")) {
1054 char *arg = argv[2];
1055 int delimiter = *arg;
1056 int i;
1058 switch (delimiter) {
1059 case '"':
1060 case '\'':
1061 for (arg++, i = 0; arg[i]; i++)
1062 if (arg[i] == delimiter) {
1063 arg[i] = 0;
1064 break;
1065 }
1066 default:
1067 string_ncopy(opt_encoding, arg, strlen(arg));
1068 return OK;
1069 }
1070 }
1072 config_msg = "Unknown variable name";
1073 return ERR;
1074 }
1076 /* Wants: mode request key */
1077 static int
1078 option_bind_command(int argc, char *argv[])
1079 {
1080 enum request request;
1081 int keymap;
1082 int key;
1084 if (argc != 3) {
1085 config_msg = "Wrong number of arguments given to bind command";
1086 return ERR;
1087 }
1089 if (set_keymap(&keymap, argv[0]) == ERR) {
1090 config_msg = "Unknown key map";
1091 return ERR;
1092 }
1094 key = get_key_value(argv[1]);
1095 if (key == ERR) {
1096 config_msg = "Unknown key";
1097 return ERR;
1098 }
1100 request = get_request(argv[2]);
1101 if (request == REQ_UNKNOWN) {
1102 config_msg = "Unknown request name";
1103 return ERR;
1104 }
1106 add_keybinding(keymap, request, key);
1108 return OK;
1109 }
1111 static int
1112 set_option(char *opt, char *value)
1113 {
1114 char *argv[16];
1115 int valuelen;
1116 int argc = 0;
1118 /* Tokenize */
1119 while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1120 argv[argc++] = value;
1122 value += valuelen;
1123 if (!*value)
1124 break;
1126 *value++ = 0;
1127 while (isspace(*value))
1128 value++;
1129 }
1131 if (!strcmp(opt, "color"))
1132 return option_color_command(argc, argv);
1134 if (!strcmp(opt, "set"))
1135 return option_set_command(argc, argv);
1137 if (!strcmp(opt, "bind"))
1138 return option_bind_command(argc, argv);
1140 config_msg = "Unknown option command";
1141 return ERR;
1142 }
1144 static int
1145 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1146 {
1147 int status = OK;
1149 config_lineno++;
1150 config_msg = "Internal error";
1152 /* Check for comment markers, since read_properties() will
1153 * only ensure opt and value are split at first " \t". */
1154 optlen = strcspn(opt, "#");
1155 if (optlen == 0)
1156 return OK;
1158 if (opt[optlen] != 0) {
1159 config_msg = "No option value";
1160 status = ERR;
1162 } else {
1163 /* Look for comment endings in the value. */
1164 size_t len = strcspn(value, "#");
1166 if (len < valuelen) {
1167 valuelen = len;
1168 value[valuelen] = 0;
1169 }
1171 status = set_option(opt, value);
1172 }
1174 if (status == ERR) {
1175 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1176 config_lineno, (int) optlen, opt, config_msg);
1177 config_errors = TRUE;
1178 }
1180 /* Always keep going if errors are encountered. */
1181 return OK;
1182 }
1184 static int
1185 load_options(void)
1186 {
1187 char *home = getenv("HOME");
1188 char buf[SIZEOF_STR];
1189 FILE *file;
1191 config_lineno = 0;
1192 config_errors = FALSE;
1194 if (!home || !string_format(buf, "%s/.tigrc", home))
1195 return ERR;
1197 /* It's ok that the file doesn't exist. */
1198 file = fopen(buf, "r");
1199 if (!file)
1200 return OK;
1202 if (read_properties(file, " \t", read_option) == ERR ||
1203 config_errors == TRUE)
1204 fprintf(stderr, "Errors while loading %s.\n", buf);
1206 return OK;
1207 }
1210 /*
1211 * The viewer
1212 */
1214 struct view;
1215 struct view_ops;
1217 /* The display array of active views and the index of the current view. */
1218 static struct view *display[2];
1219 static unsigned int current_view;
1221 /* Reading from the prompt? */
1222 static bool input_mode = FALSE;
1224 #define foreach_displayed_view(view, i) \
1225 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1227 #define displayed_views() (display[1] != NULL ? 2 : 1)
1229 /* Current head and commit ID */
1230 static char ref_blob[SIZEOF_REF] = "";
1231 static char ref_commit[SIZEOF_REF] = "HEAD";
1232 static char ref_head[SIZEOF_REF] = "HEAD";
1234 struct view {
1235 const char *name; /* View name */
1236 const char *cmd_fmt; /* Default command line format */
1237 const char *cmd_env; /* Command line set via environment */
1238 const char *id; /* Points to either of ref_{head,commit,blob} */
1240 struct view_ops *ops; /* View operations */
1242 enum keymap keymap; /* What keymap does this view have */
1244 char cmd[SIZEOF_STR]; /* Command buffer */
1245 char ref[SIZEOF_REF]; /* Hovered commit reference */
1246 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
1248 int height, width; /* The width and height of the main window */
1249 WINDOW *win; /* The main window */
1250 WINDOW *title; /* The title window living below the main window */
1252 /* Navigation */
1253 unsigned long offset; /* Offset of the window top */
1254 unsigned long lineno; /* Current line number */
1256 /* Searching */
1257 char grep[SIZEOF_STR]; /* Search string */
1258 regex_t *regex; /* Pre-compiled regex */
1260 /* If non-NULL, points to the view that opened this view. If this view
1261 * is closed tig will switch back to the parent view. */
1262 struct view *parent;
1264 /* Buffering */
1265 unsigned long lines; /* Total number of lines */
1266 struct line *line; /* Line index */
1267 unsigned long line_size;/* Total number of allocated lines */
1268 unsigned int digits; /* Number of digits in the lines member. */
1270 /* Loading */
1271 FILE *pipe;
1272 time_t start_time;
1273 };
1275 struct view_ops {
1276 /* What type of content being displayed. Used in the title bar. */
1277 const char *type;
1278 /* Open and reads in all view content. */
1279 bool (*open)(struct view *view);
1280 /* Read one line; updates view->line. */
1281 bool (*read)(struct view *view, char *data);
1282 /* Draw one line; @lineno must be < view->height. */
1283 bool (*draw)(struct view *view, struct line *line, unsigned int lineno, bool selected);
1284 /* Depending on view handle a special requests. */
1285 enum request (*request)(struct view *view, enum request request, struct line *line);
1286 /* Search for regex in a line. */
1287 bool (*grep)(struct view *view, struct line *line);
1288 /* Select line */
1289 void (*select)(struct view *view, struct line *line);
1290 };
1292 static struct view_ops pager_ops;
1293 static struct view_ops main_ops;
1294 static struct view_ops tree_ops;
1295 static struct view_ops blob_ops;
1296 static struct view_ops help_ops;
1297 static struct view_ops status_ops;
1298 static struct view_ops stage_ops;
1300 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1301 { name, cmd, #env, ref, ops, map}
1303 #define VIEW_(id, name, ops, ref) \
1304 VIEW_STR(name, TIG_##id##_CMD, TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1307 static struct view views[] = {
1308 VIEW_(MAIN, "main", &main_ops, ref_head),
1309 VIEW_(DIFF, "diff", &pager_ops, ref_commit),
1310 VIEW_(LOG, "log", &pager_ops, ref_head),
1311 VIEW_(TREE, "tree", &tree_ops, ref_commit),
1312 VIEW_(BLOB, "blob", &blob_ops, ref_blob),
1313 VIEW_(HELP, "help", &help_ops, ""),
1314 VIEW_(PAGER, "pager", &pager_ops, "stdin"),
1315 VIEW_(STATUS, "status", &status_ops, ""),
1316 VIEW_(STAGE, "stage", &stage_ops, ""),
1317 };
1319 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1321 #define foreach_view(view, i) \
1322 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1324 #define view_is_displayed(view) \
1325 (view == display[0] || view == display[1])
1327 static bool
1328 draw_view_line(struct view *view, unsigned int lineno)
1329 {
1330 struct line *line;
1331 bool selected = (view->offset + lineno == view->lineno);
1332 bool draw_ok;
1334 assert(view_is_displayed(view));
1336 if (view->offset + lineno >= view->lines)
1337 return FALSE;
1339 line = &view->line[view->offset + lineno];
1341 if (selected) {
1342 line->selected = TRUE;
1343 view->ops->select(view, line);
1344 } else if (line->selected) {
1345 line->selected = FALSE;
1346 wmove(view->win, lineno, 0);
1347 wclrtoeol(view->win);
1348 }
1350 scrollok(view->win, FALSE);
1351 draw_ok = view->ops->draw(view, line, lineno, selected);
1352 scrollok(view->win, TRUE);
1354 return draw_ok;
1355 }
1357 static void
1358 redraw_view_from(struct view *view, int lineno)
1359 {
1360 assert(0 <= lineno && lineno < view->height);
1362 for (; lineno < view->height; lineno++) {
1363 if (!draw_view_line(view, lineno))
1364 break;
1365 }
1367 redrawwin(view->win);
1368 if (input_mode)
1369 wnoutrefresh(view->win);
1370 else
1371 wrefresh(view->win);
1372 }
1374 static void
1375 redraw_view(struct view *view)
1376 {
1377 wclear(view->win);
1378 redraw_view_from(view, 0);
1379 }
1382 static void
1383 update_view_title(struct view *view)
1384 {
1385 char buf[SIZEOF_STR];
1386 char state[SIZEOF_STR];
1387 size_t bufpos = 0, statelen = 0;
1389 assert(view_is_displayed(view));
1391 if (view != VIEW(REQ_VIEW_STATUS) && (view->lines || view->pipe)) {
1392 unsigned int view_lines = view->offset + view->height;
1393 unsigned int lines = view->lines
1394 ? MIN(view_lines, view->lines) * 100 / view->lines
1395 : 0;
1397 string_format_from(state, &statelen, "- %s %d of %d (%d%%)",
1398 view->ops->type,
1399 view->lineno + 1,
1400 view->lines,
1401 lines);
1403 if (view->pipe) {
1404 time_t secs = time(NULL) - view->start_time;
1406 /* Three git seconds are a long time ... */
1407 if (secs > 2)
1408 string_format_from(state, &statelen, " %lds", secs);
1409 }
1410 }
1412 string_format_from(buf, &bufpos, "[%s]", view->name);
1413 if (*view->ref && bufpos < view->width) {
1414 size_t refsize = strlen(view->ref);
1415 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
1417 if (minsize < view->width)
1418 refsize = view->width - minsize + 7;
1419 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
1420 }
1422 if (statelen && bufpos < view->width) {
1423 string_format_from(buf, &bufpos, " %s", state);
1424 }
1426 if (view == display[current_view])
1427 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1428 else
1429 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1431 mvwaddnstr(view->title, 0, 0, buf, bufpos);
1432 wclrtoeol(view->title);
1433 wmove(view->title, 0, view->width - 1);
1435 if (input_mode)
1436 wnoutrefresh(view->title);
1437 else
1438 wrefresh(view->title);
1439 }
1441 static void
1442 resize_display(void)
1443 {
1444 int offset, i;
1445 struct view *base = display[0];
1446 struct view *view = display[1] ? display[1] : display[0];
1448 /* Setup window dimensions */
1450 getmaxyx(stdscr, base->height, base->width);
1452 /* Make room for the status window. */
1453 base->height -= 1;
1455 if (view != base) {
1456 /* Horizontal split. */
1457 view->width = base->width;
1458 view->height = SCALE_SPLIT_VIEW(base->height);
1459 base->height -= view->height;
1461 /* Make room for the title bar. */
1462 view->height -= 1;
1463 }
1465 /* Make room for the title bar. */
1466 base->height -= 1;
1468 offset = 0;
1470 foreach_displayed_view (view, i) {
1471 if (!view->win) {
1472 view->win = newwin(view->height, 0, offset, 0);
1473 if (!view->win)
1474 die("Failed to create %s view", view->name);
1476 scrollok(view->win, TRUE);
1478 view->title = newwin(1, 0, offset + view->height, 0);
1479 if (!view->title)
1480 die("Failed to create title window");
1482 } else {
1483 wresize(view->win, view->height, view->width);
1484 mvwin(view->win, offset, 0);
1485 mvwin(view->title, offset + view->height, 0);
1486 }
1488 offset += view->height + 1;
1489 }
1490 }
1492 static void
1493 redraw_display(void)
1494 {
1495 struct view *view;
1496 int i;
1498 foreach_displayed_view (view, i) {
1499 redraw_view(view);
1500 update_view_title(view);
1501 }
1502 }
1504 static void
1505 update_display_cursor(struct view *view)
1506 {
1507 /* Move the cursor to the right-most column of the cursor line.
1508 *
1509 * XXX: This could turn out to be a bit expensive, but it ensures that
1510 * the cursor does not jump around. */
1511 if (view->lines) {
1512 wmove(view->win, view->lineno - view->offset, view->width - 1);
1513 wrefresh(view->win);
1514 }
1515 }
1517 /*
1518 * Navigation
1519 */
1521 /* Scrolling backend */
1522 static void
1523 do_scroll_view(struct view *view, int lines)
1524 {
1525 bool redraw_current_line = FALSE;
1527 /* The rendering expects the new offset. */
1528 view->offset += lines;
1530 assert(0 <= view->offset && view->offset < view->lines);
1531 assert(lines);
1533 /* Move current line into the view. */
1534 if (view->lineno < view->offset) {
1535 view->lineno = view->offset;
1536 redraw_current_line = TRUE;
1537 } else if (view->lineno >= view->offset + view->height) {
1538 view->lineno = view->offset + view->height - 1;
1539 redraw_current_line = TRUE;
1540 }
1542 assert(view->offset <= view->lineno && view->lineno < view->lines);
1544 /* Redraw the whole screen if scrolling is pointless. */
1545 if (view->height < ABS(lines)) {
1546 redraw_view(view);
1548 } else {
1549 int line = lines > 0 ? view->height - lines : 0;
1550 int end = line + ABS(lines);
1552 wscrl(view->win, lines);
1554 for (; line < end; line++) {
1555 if (!draw_view_line(view, line))
1556 break;
1557 }
1559 if (redraw_current_line)
1560 draw_view_line(view, view->lineno - view->offset);
1561 }
1563 redrawwin(view->win);
1564 wrefresh(view->win);
1565 report("");
1566 }
1568 /* Scroll frontend */
1569 static void
1570 scroll_view(struct view *view, enum request request)
1571 {
1572 int lines = 1;
1574 assert(view_is_displayed(view));
1576 switch (request) {
1577 case REQ_SCROLL_PAGE_DOWN:
1578 lines = view->height;
1579 case REQ_SCROLL_LINE_DOWN:
1580 if (view->offset + lines > view->lines)
1581 lines = view->lines - view->offset;
1583 if (lines == 0 || view->offset + view->height >= view->lines) {
1584 report("Cannot scroll beyond the last line");
1585 return;
1586 }
1587 break;
1589 case REQ_SCROLL_PAGE_UP:
1590 lines = view->height;
1591 case REQ_SCROLL_LINE_UP:
1592 if (lines > view->offset)
1593 lines = view->offset;
1595 if (lines == 0) {
1596 report("Cannot scroll beyond the first line");
1597 return;
1598 }
1600 lines = -lines;
1601 break;
1603 default:
1604 die("request %d not handled in switch", request);
1605 }
1607 do_scroll_view(view, lines);
1608 }
1610 /* Cursor moving */
1611 static void
1612 move_view(struct view *view, enum request request)
1613 {
1614 int scroll_steps = 0;
1615 int steps;
1617 switch (request) {
1618 case REQ_MOVE_FIRST_LINE:
1619 steps = -view->lineno;
1620 break;
1622 case REQ_MOVE_LAST_LINE:
1623 steps = view->lines - view->lineno - 1;
1624 break;
1626 case REQ_MOVE_PAGE_UP:
1627 steps = view->height > view->lineno
1628 ? -view->lineno : -view->height;
1629 break;
1631 case REQ_MOVE_PAGE_DOWN:
1632 steps = view->lineno + view->height >= view->lines
1633 ? view->lines - view->lineno - 1 : view->height;
1634 break;
1636 case REQ_MOVE_UP:
1637 steps = -1;
1638 break;
1640 case REQ_MOVE_DOWN:
1641 steps = 1;
1642 break;
1644 default:
1645 die("request %d not handled in switch", request);
1646 }
1648 if (steps <= 0 && view->lineno == 0) {
1649 report("Cannot move beyond the first line");
1650 return;
1652 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1653 report("Cannot move beyond the last line");
1654 return;
1655 }
1657 /* Move the current line */
1658 view->lineno += steps;
1659 assert(0 <= view->lineno && view->lineno < view->lines);
1661 /* Check whether the view needs to be scrolled */
1662 if (view->lineno < view->offset ||
1663 view->lineno >= view->offset + view->height) {
1664 scroll_steps = steps;
1665 if (steps < 0 && -steps > view->offset) {
1666 scroll_steps = -view->offset;
1668 } else if (steps > 0) {
1669 if (view->lineno == view->lines - 1 &&
1670 view->lines > view->height) {
1671 scroll_steps = view->lines - view->offset - 1;
1672 if (scroll_steps >= view->height)
1673 scroll_steps -= view->height - 1;
1674 }
1675 }
1676 }
1678 if (!view_is_displayed(view)) {
1679 view->offset += scroll_steps;
1680 assert(0 <= view->offset && view->offset < view->lines);
1681 view->ops->select(view, &view->line[view->lineno]);
1682 return;
1683 }
1685 /* Repaint the old "current" line if we be scrolling */
1686 if (ABS(steps) < view->height)
1687 draw_view_line(view, view->lineno - steps - view->offset);
1689 if (scroll_steps) {
1690 do_scroll_view(view, scroll_steps);
1691 return;
1692 }
1694 /* Draw the current line */
1695 draw_view_line(view, view->lineno - view->offset);
1697 redrawwin(view->win);
1698 wrefresh(view->win);
1699 report("");
1700 }
1703 /*
1704 * Searching
1705 */
1707 static void search_view(struct view *view, enum request request);
1709 static bool
1710 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1711 {
1712 assert(view_is_displayed(view));
1714 if (!view->ops->grep(view, line))
1715 return FALSE;
1717 if (lineno - view->offset >= view->height) {
1718 view->offset = lineno;
1719 view->lineno = lineno;
1720 redraw_view(view);
1722 } else {
1723 unsigned long old_lineno = view->lineno - view->offset;
1725 view->lineno = lineno;
1726 draw_view_line(view, old_lineno);
1728 draw_view_line(view, view->lineno - view->offset);
1729 redrawwin(view->win);
1730 wrefresh(view->win);
1731 }
1733 report("Line %ld matches '%s'", lineno + 1, view->grep);
1734 return TRUE;
1735 }
1737 static void
1738 find_next(struct view *view, enum request request)
1739 {
1740 unsigned long lineno = view->lineno;
1741 int direction;
1743 if (!*view->grep) {
1744 if (!*opt_search)
1745 report("No previous search");
1746 else
1747 search_view(view, request);
1748 return;
1749 }
1751 switch (request) {
1752 case REQ_SEARCH:
1753 case REQ_FIND_NEXT:
1754 direction = 1;
1755 break;
1757 case REQ_SEARCH_BACK:
1758 case REQ_FIND_PREV:
1759 direction = -1;
1760 break;
1762 default:
1763 return;
1764 }
1766 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1767 lineno += direction;
1769 /* Note, lineno is unsigned long so will wrap around in which case it
1770 * will become bigger than view->lines. */
1771 for (; lineno < view->lines; lineno += direction) {
1772 struct line *line = &view->line[lineno];
1774 if (find_next_line(view, lineno, line))
1775 return;
1776 }
1778 report("No match found for '%s'", view->grep);
1779 }
1781 static void
1782 search_view(struct view *view, enum request request)
1783 {
1784 int regex_err;
1786 if (view->regex) {
1787 regfree(view->regex);
1788 *view->grep = 0;
1789 } else {
1790 view->regex = calloc(1, sizeof(*view->regex));
1791 if (!view->regex)
1792 return;
1793 }
1795 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
1796 if (regex_err != 0) {
1797 char buf[SIZEOF_STR] = "unknown error";
1799 regerror(regex_err, view->regex, buf, sizeof(buf));
1800 report("Search failed: %s", buf);
1801 return;
1802 }
1804 string_copy(view->grep, opt_search);
1806 find_next(view, request);
1807 }
1809 /*
1810 * Incremental updating
1811 */
1813 static void
1814 end_update(struct view *view)
1815 {
1816 if (!view->pipe)
1817 return;
1818 set_nonblocking_input(FALSE);
1819 if (view->pipe == stdin)
1820 fclose(view->pipe);
1821 else
1822 pclose(view->pipe);
1823 view->pipe = NULL;
1824 }
1826 static bool
1827 begin_update(struct view *view)
1828 {
1829 if (view->pipe)
1830 end_update(view);
1832 if (opt_cmd[0]) {
1833 string_copy(view->cmd, opt_cmd);
1834 opt_cmd[0] = 0;
1835 /* When running random commands, initially show the
1836 * command in the title. However, it maybe later be
1837 * overwritten if a commit line is selected. */
1838 if (view == VIEW(REQ_VIEW_PAGER))
1839 string_copy(view->ref, view->cmd);
1840 else
1841 view->ref[0] = 0;
1843 } else if (view == VIEW(REQ_VIEW_TREE)) {
1844 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1845 char path[SIZEOF_STR];
1847 if (strcmp(view->vid, view->id))
1848 opt_path[0] = path[0] = 0;
1849 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
1850 return FALSE;
1852 if (!string_format(view->cmd, format, view->id, path))
1853 return FALSE;
1855 } else {
1856 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1857 const char *id = view->id;
1859 if (!string_format(view->cmd, format, id, id, id, id, id))
1860 return FALSE;
1862 /* Put the current ref_* value to the view title ref
1863 * member. This is needed by the blob view. Most other
1864 * views sets it automatically after loading because the
1865 * first line is a commit line. */
1866 string_copy_rev(view->ref, view->id);
1867 }
1869 /* Special case for the pager view. */
1870 if (opt_pipe) {
1871 view->pipe = opt_pipe;
1872 opt_pipe = NULL;
1873 } else {
1874 view->pipe = popen(view->cmd, "r");
1875 }
1877 if (!view->pipe)
1878 return FALSE;
1880 set_nonblocking_input(TRUE);
1882 view->offset = 0;
1883 view->lines = 0;
1884 view->lineno = 0;
1885 string_copy_rev(view->vid, view->id);
1887 if (view->line) {
1888 int i;
1890 for (i = 0; i < view->lines; i++)
1891 if (view->line[i].data)
1892 free(view->line[i].data);
1894 free(view->line);
1895 view->line = NULL;
1896 }
1898 view->start_time = time(NULL);
1900 return TRUE;
1901 }
1903 static struct line *
1904 realloc_lines(struct view *view, size_t line_size)
1905 {
1906 struct line *tmp = realloc(view->line, sizeof(*view->line) * line_size);
1908 if (!tmp)
1909 return NULL;
1911 view->line = tmp;
1912 view->line_size = line_size;
1913 return view->line;
1914 }
1916 static bool
1917 update_view(struct view *view)
1918 {
1919 char in_buffer[BUFSIZ];
1920 char out_buffer[BUFSIZ * 2];
1921 char *line;
1922 /* The number of lines to read. If too low it will cause too much
1923 * redrawing (and possible flickering), if too high responsiveness
1924 * will suffer. */
1925 unsigned long lines = view->height;
1926 int redraw_from = -1;
1928 if (!view->pipe)
1929 return TRUE;
1931 /* Only redraw if lines are visible. */
1932 if (view->offset + view->height >= view->lines)
1933 redraw_from = view->lines - view->offset;
1935 /* FIXME: This is probably not perfect for backgrounded views. */
1936 if (!realloc_lines(view, view->lines + lines))
1937 goto alloc_error;
1939 while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
1940 size_t linelen = strlen(line);
1942 if (linelen)
1943 line[linelen - 1] = 0;
1945 if (opt_iconv != ICONV_NONE) {
1946 ICONV_CONST char *inbuf = line;
1947 size_t inlen = linelen;
1949 char *outbuf = out_buffer;
1950 size_t outlen = sizeof(out_buffer);
1952 size_t ret;
1954 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
1955 if (ret != (size_t) -1) {
1956 line = out_buffer;
1957 linelen = strlen(out_buffer);
1958 }
1959 }
1961 if (!view->ops->read(view, line))
1962 goto alloc_error;
1964 if (lines-- == 1)
1965 break;
1966 }
1968 {
1969 int digits;
1971 lines = view->lines;
1972 for (digits = 0; lines; digits++)
1973 lines /= 10;
1975 /* Keep the displayed view in sync with line number scaling. */
1976 if (digits != view->digits) {
1977 view->digits = digits;
1978 redraw_from = 0;
1979 }
1980 }
1982 if (!view_is_displayed(view))
1983 goto check_pipe;
1985 if (view == VIEW(REQ_VIEW_TREE)) {
1986 /* Clear the view and redraw everything since the tree sorting
1987 * might have rearranged things. */
1988 redraw_view(view);
1990 } else if (redraw_from >= 0) {
1991 /* If this is an incremental update, redraw the previous line
1992 * since for commits some members could have changed when
1993 * loading the main view. */
1994 if (redraw_from > 0)
1995 redraw_from--;
1997 /* Since revision graph visualization requires knowledge
1998 * about the parent commit, it causes a further one-off
1999 * needed to be redrawn for incremental updates. */
2000 if (redraw_from > 0 && opt_rev_graph)
2001 redraw_from--;
2003 /* Incrementally draw avoids flickering. */
2004 redraw_view_from(view, redraw_from);
2005 }
2007 /* Update the title _after_ the redraw so that if the redraw picks up a
2008 * commit reference in view->ref it'll be available here. */
2009 update_view_title(view);
2011 check_pipe:
2012 if (ferror(view->pipe)) {
2013 report("Failed to read: %s", strerror(errno));
2014 goto end;
2016 } else if (feof(view->pipe)) {
2017 report("");
2018 goto end;
2019 }
2021 return TRUE;
2023 alloc_error:
2024 report("Allocation failure");
2026 end:
2027 view->ops->read(view, NULL);
2028 end_update(view);
2029 return FALSE;
2030 }
2032 static struct line *
2033 add_line_data(struct view *view, void *data, enum line_type type)
2034 {
2035 struct line *line = &view->line[view->lines++];
2037 memset(line, 0, sizeof(*line));
2038 line->type = type;
2039 line->data = data;
2041 return line;
2042 }
2044 static struct line *
2045 add_line_text(struct view *view, char *data, enum line_type type)
2046 {
2047 if (data)
2048 data = strdup(data);
2050 return data ? add_line_data(view, data, type) : NULL;
2051 }
2054 /*
2055 * View opening
2056 */
2058 enum open_flags {
2059 OPEN_DEFAULT = 0, /* Use default view switching. */
2060 OPEN_SPLIT = 1, /* Split current view. */
2061 OPEN_BACKGROUNDED = 2, /* Backgrounded. */
2062 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
2063 };
2065 static void
2066 open_view(struct view *prev, enum request request, enum open_flags flags)
2067 {
2068 bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2069 bool split = !!(flags & OPEN_SPLIT);
2070 bool reload = !!(flags & OPEN_RELOAD);
2071 struct view *view = VIEW(request);
2072 int nviews = displayed_views();
2073 struct view *base_view = display[0];
2075 if (view == prev && nviews == 1 && !reload) {
2076 report("Already in %s view", view->name);
2077 return;
2078 }
2080 if (view->ops->open) {
2081 if (!view->ops->open(view)) {
2082 report("Failed to load %s view", view->name);
2083 return;
2084 }
2086 } else if ((reload || strcmp(view->vid, view->id)) &&
2087 !begin_update(view)) {
2088 report("Failed to load %s view", view->name);
2089 return;
2090 }
2092 if (split) {
2093 display[1] = view;
2094 if (!backgrounded)
2095 current_view = 1;
2096 } else {
2097 /* Maximize the current view. */
2098 memset(display, 0, sizeof(display));
2099 current_view = 0;
2100 display[current_view] = view;
2101 }
2103 /* Resize the view when switching between split- and full-screen,
2104 * or when switching between two different full-screen views. */
2105 if (nviews != displayed_views() ||
2106 (nviews == 1 && base_view != display[0]))
2107 resize_display();
2109 if (split && prev->lineno - prev->offset >= prev->height) {
2110 /* Take the title line into account. */
2111 int lines = prev->lineno - prev->offset - prev->height + 1;
2113 /* Scroll the view that was split if the current line is
2114 * outside the new limited view. */
2115 do_scroll_view(prev, lines);
2116 }
2118 if (prev && view != prev) {
2119 if (split && !backgrounded) {
2120 /* "Blur" the previous view. */
2121 update_view_title(prev);
2122 }
2124 view->parent = prev;
2125 }
2127 if (view->pipe && view->lines == 0) {
2128 /* Clear the old view and let the incremental updating refill
2129 * the screen. */
2130 wclear(view->win);
2131 report("");
2132 } else {
2133 redraw_view(view);
2134 report("");
2135 }
2137 /* If the view is backgrounded the above calls to report()
2138 * won't redraw the view title. */
2139 if (backgrounded)
2140 update_view_title(view);
2141 }
2143 static void
2144 open_editor(struct view *view, char *file)
2145 {
2146 char cmd[SIZEOF_STR];
2147 char file_sq[SIZEOF_STR];
2148 char *editor;
2150 editor = getenv("GIT_EDITOR");
2151 if (!editor && *opt_editor)
2152 editor = opt_editor;
2153 if (!editor)
2154 editor = getenv("VISUAL");
2155 if (!editor)
2156 editor = getenv("EDITOR");
2157 if (!editor)
2158 editor = "vi";
2160 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2161 string_format(cmd, "%s %s", editor, file_sq)) {
2162 def_prog_mode(); /* save current tty modes */
2163 endwin(); /* restore original tty modes */
2164 system(cmd);
2165 reset_prog_mode();
2166 redraw_display();
2167 }
2168 }
2170 /*
2171 * User request switch noodle
2172 */
2174 static int
2175 view_driver(struct view *view, enum request request)
2176 {
2177 int i;
2179 if (view && view->lines) {
2180 request = view->ops->request(view, request, &view->line[view->lineno]);
2181 if (request == REQ_NONE)
2182 return TRUE;
2183 }
2185 switch (request) {
2186 case REQ_MOVE_UP:
2187 case REQ_MOVE_DOWN:
2188 case REQ_MOVE_PAGE_UP:
2189 case REQ_MOVE_PAGE_DOWN:
2190 case REQ_MOVE_FIRST_LINE:
2191 case REQ_MOVE_LAST_LINE:
2192 move_view(view, request);
2193 break;
2195 case REQ_SCROLL_LINE_DOWN:
2196 case REQ_SCROLL_LINE_UP:
2197 case REQ_SCROLL_PAGE_DOWN:
2198 case REQ_SCROLL_PAGE_UP:
2199 scroll_view(view, request);
2200 break;
2202 case REQ_VIEW_BLOB:
2203 if (!ref_blob[0]) {
2204 report("No file chosen, press %s to open tree view",
2205 get_key(REQ_VIEW_TREE));
2206 break;
2207 }
2208 open_view(view, request, OPEN_DEFAULT);
2209 break;
2211 case REQ_VIEW_PAGER:
2212 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2213 report("No pager content, press %s to run command from prompt",
2214 get_key(REQ_PROMPT));
2215 break;
2216 }
2217 open_view(view, request, OPEN_DEFAULT);
2218 break;
2220 case REQ_VIEW_STAGE:
2221 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2222 report("No stage content, press %s to open the status view and choose file",
2223 get_key(REQ_VIEW_STATUS));
2224 break;
2225 }
2226 open_view(view, request, OPEN_DEFAULT);
2227 break;
2229 case REQ_VIEW_MAIN:
2230 case REQ_VIEW_DIFF:
2231 case REQ_VIEW_LOG:
2232 case REQ_VIEW_TREE:
2233 case REQ_VIEW_HELP:
2234 case REQ_VIEW_STATUS:
2235 open_view(view, request, OPEN_DEFAULT);
2236 break;
2238 case REQ_NEXT:
2239 case REQ_PREVIOUS:
2240 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2242 if ((view == VIEW(REQ_VIEW_DIFF) &&
2243 view->parent == VIEW(REQ_VIEW_MAIN)) ||
2244 (view == VIEW(REQ_VIEW_STAGE) &&
2245 view->parent == VIEW(REQ_VIEW_STATUS)) ||
2246 (view == VIEW(REQ_VIEW_BLOB) &&
2247 view->parent == VIEW(REQ_VIEW_TREE))) {
2248 int line;
2250 view = view->parent;
2251 line = view->lineno;
2252 move_view(view, request);
2253 if (view_is_displayed(view))
2254 update_view_title(view);
2255 if (line != view->lineno)
2256 view->ops->request(view, REQ_ENTER,
2257 &view->line[view->lineno]);
2259 } else {
2260 move_view(view, request);
2261 }
2262 break;
2264 case REQ_VIEW_NEXT:
2265 {
2266 int nviews = displayed_views();
2267 int next_view = (current_view + 1) % nviews;
2269 if (next_view == current_view) {
2270 report("Only one view is displayed");
2271 break;
2272 }
2274 current_view = next_view;
2275 /* Blur out the title of the previous view. */
2276 update_view_title(view);
2277 report("");
2278 break;
2279 }
2280 case REQ_TOGGLE_LINENO:
2281 opt_line_number = !opt_line_number;
2282 redraw_display();
2283 break;
2285 case REQ_TOGGLE_REV_GRAPH:
2286 opt_rev_graph = !opt_rev_graph;
2287 redraw_display();
2288 break;
2290 case REQ_PROMPT:
2291 /* Always reload^Wrerun commands from the prompt. */
2292 open_view(view, opt_request, OPEN_RELOAD);
2293 break;
2295 case REQ_SEARCH:
2296 case REQ_SEARCH_BACK:
2297 search_view(view, request);
2298 break;
2300 case REQ_FIND_NEXT:
2301 case REQ_FIND_PREV:
2302 find_next(view, request);
2303 break;
2305 case REQ_STOP_LOADING:
2306 for (i = 0; i < ARRAY_SIZE(views); i++) {
2307 view = &views[i];
2308 if (view->pipe)
2309 report("Stopped loading the %s view", view->name),
2310 end_update(view);
2311 }
2312 break;
2314 case REQ_SHOW_VERSION:
2315 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2316 return TRUE;
2318 case REQ_SCREEN_RESIZE:
2319 resize_display();
2320 /* Fall-through */
2321 case REQ_SCREEN_REDRAW:
2322 redraw_display();
2323 break;
2325 case REQ_EDIT:
2326 report("Nothing to edit");
2327 break;
2329 case REQ_ENTER:
2330 report("Nothing to enter");
2331 break;
2333 case REQ_NONE:
2334 doupdate();
2335 return TRUE;
2337 case REQ_VIEW_CLOSE:
2338 /* XXX: Mark closed views by letting view->parent point to the
2339 * view itself. Parents to closed view should never be
2340 * followed. */
2341 if (view->parent &&
2342 view->parent->parent != view->parent) {
2343 memset(display, 0, sizeof(display));
2344 current_view = 0;
2345 display[current_view] = view->parent;
2346 view->parent = view;
2347 resize_display();
2348 redraw_display();
2349 break;
2350 }
2351 /* Fall-through */
2352 case REQ_QUIT:
2353 return FALSE;
2355 default:
2356 /* An unknown key will show most commonly used commands. */
2357 report("Unknown key, press 'h' for help");
2358 return TRUE;
2359 }
2361 return TRUE;
2362 }
2365 /*
2366 * Pager backend
2367 */
2369 static bool
2370 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2371 {
2372 char *text = line->data;
2373 enum line_type type = line->type;
2374 int textlen = strlen(text);
2375 int attr;
2377 wmove(view->win, lineno, 0);
2379 if (selected) {
2380 type = LINE_CURSOR;
2381 wchgat(view->win, -1, 0, type, NULL);
2382 }
2384 attr = get_line_attr(type);
2385 wattrset(view->win, attr);
2387 if (opt_line_number || opt_tab_size < TABSIZE) {
2388 static char spaces[] = " ";
2389 int col_offset = 0, col = 0;
2391 if (opt_line_number) {
2392 unsigned long real_lineno = view->offset + lineno + 1;
2394 if (real_lineno == 1 ||
2395 (real_lineno % opt_num_interval) == 0) {
2396 wprintw(view->win, "%.*d", view->digits, real_lineno);
2398 } else {
2399 waddnstr(view->win, spaces,
2400 MIN(view->digits, STRING_SIZE(spaces)));
2401 }
2402 waddstr(view->win, ": ");
2403 col_offset = view->digits + 2;
2404 }
2406 while (text && col_offset + col < view->width) {
2407 int cols_max = view->width - col_offset - col;
2408 char *pos = text;
2409 int cols;
2411 if (*text == '\t') {
2412 text++;
2413 assert(sizeof(spaces) > TABSIZE);
2414 pos = spaces;
2415 cols = opt_tab_size - (col % opt_tab_size);
2417 } else {
2418 text = strchr(text, '\t');
2419 cols = line ? text - pos : strlen(pos);
2420 }
2422 waddnstr(view->win, pos, MIN(cols, cols_max));
2423 col += cols;
2424 }
2426 } else {
2427 int col = 0, pos = 0;
2429 for (; pos < textlen && col < view->width; pos++, col++)
2430 if (text[pos] == '\t')
2431 col += TABSIZE - (col % TABSIZE) - 1;
2433 waddnstr(view->win, text, pos);
2434 }
2436 return TRUE;
2437 }
2439 static bool
2440 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2441 {
2442 char refbuf[SIZEOF_STR];
2443 char *ref = NULL;
2444 FILE *pipe;
2446 if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2447 return TRUE;
2449 pipe = popen(refbuf, "r");
2450 if (!pipe)
2451 return TRUE;
2453 if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2454 ref = chomp_string(ref);
2455 pclose(pipe);
2457 if (!ref || !*ref)
2458 return TRUE;
2460 /* This is the only fatal call, since it can "corrupt" the buffer. */
2461 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2462 return FALSE;
2464 return TRUE;
2465 }
2467 static void
2468 add_pager_refs(struct view *view, struct line *line)
2469 {
2470 char buf[SIZEOF_STR];
2471 char *commit_id = line->data + STRING_SIZE("commit ");
2472 struct ref **refs;
2473 size_t bufpos = 0, refpos = 0;
2474 const char *sep = "Refs: ";
2475 bool is_tag = FALSE;
2477 assert(line->type == LINE_COMMIT);
2479 refs = get_refs(commit_id);
2480 if (!refs) {
2481 if (view == VIEW(REQ_VIEW_DIFF))
2482 goto try_add_describe_ref;
2483 return;
2484 }
2486 do {
2487 struct ref *ref = refs[refpos];
2488 char *fmt = ref->tag ? "%s[%s]" :
2489 ref->remote ? "%s<%s>" : "%s%s";
2491 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2492 return;
2493 sep = ", ";
2494 if (ref->tag)
2495 is_tag = TRUE;
2496 } while (refs[refpos++]->next);
2498 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2499 try_add_describe_ref:
2500 /* Add <tag>-g<commit_id> "fake" reference. */
2501 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2502 return;
2503 }
2505 if (bufpos == 0)
2506 return;
2508 if (!realloc_lines(view, view->line_size + 1))
2509 return;
2511 add_line_text(view, buf, LINE_PP_REFS);
2512 }
2514 static bool
2515 pager_read(struct view *view, char *data)
2516 {
2517 struct line *line;
2519 if (!data)
2520 return TRUE;
2522 line = add_line_text(view, data, get_line_type(data));
2523 if (!line)
2524 return FALSE;
2526 if (line->type == LINE_COMMIT &&
2527 (view == VIEW(REQ_VIEW_DIFF) ||
2528 view == VIEW(REQ_VIEW_LOG)))
2529 add_pager_refs(view, line);
2531 return TRUE;
2532 }
2534 static enum request
2535 pager_request(struct view *view, enum request request, struct line *line)
2536 {
2537 int split = 0;
2539 if (request != REQ_ENTER)
2540 return request;
2542 if (line->type == LINE_COMMIT &&
2543 (view == VIEW(REQ_VIEW_LOG) ||
2544 view == VIEW(REQ_VIEW_PAGER))) {
2545 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2546 split = 1;
2547 }
2549 /* Always scroll the view even if it was split. That way
2550 * you can use Enter to scroll through the log view and
2551 * split open each commit diff. */
2552 scroll_view(view, REQ_SCROLL_LINE_DOWN);
2554 /* FIXME: A minor workaround. Scrolling the view will call report("")
2555 * but if we are scrolling a non-current view this won't properly
2556 * update the view title. */
2557 if (split)
2558 update_view_title(view);
2560 return REQ_NONE;
2561 }
2563 static bool
2564 pager_grep(struct view *view, struct line *line)
2565 {
2566 regmatch_t pmatch;
2567 char *text = line->data;
2569 if (!*text)
2570 return FALSE;
2572 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2573 return FALSE;
2575 return TRUE;
2576 }
2578 static void
2579 pager_select(struct view *view, struct line *line)
2580 {
2581 if (line->type == LINE_COMMIT) {
2582 char *text = line->data + STRING_SIZE("commit ");
2584 if (view != VIEW(REQ_VIEW_PAGER))
2585 string_copy_rev(view->ref, text);
2586 string_copy_rev(ref_commit, text);
2587 }
2588 }
2590 static struct view_ops pager_ops = {
2591 "line",
2592 NULL,
2593 pager_read,
2594 pager_draw,
2595 pager_request,
2596 pager_grep,
2597 pager_select,
2598 };
2601 /*
2602 * Help backend
2603 */
2605 static bool
2606 help_open(struct view *view)
2607 {
2608 char buf[BUFSIZ];
2609 int lines = ARRAY_SIZE(req_info) + 2;
2610 int i;
2612 if (view->lines > 0)
2613 return TRUE;
2615 for (i = 0; i < ARRAY_SIZE(req_info); i++)
2616 if (!req_info[i].request)
2617 lines++;
2619 view->line = calloc(lines, sizeof(*view->line));
2620 if (!view->line)
2621 return FALSE;
2623 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
2625 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
2626 char *key;
2628 if (!req_info[i].request) {
2629 add_line_text(view, "", LINE_DEFAULT);
2630 add_line_text(view, req_info[i].help, LINE_DEFAULT);
2631 continue;
2632 }
2634 key = get_key(req_info[i].request);
2635 if (!string_format(buf, " %-25s %s", key, req_info[i].help))
2636 continue;
2638 add_line_text(view, buf, LINE_DEFAULT);
2639 }
2641 return TRUE;
2642 }
2644 static struct view_ops help_ops = {
2645 "line",
2646 help_open,
2647 NULL,
2648 pager_draw,
2649 pager_request,
2650 pager_grep,
2651 pager_select,
2652 };
2655 /*
2656 * Tree backend
2657 */
2659 struct tree_stack_entry {
2660 struct tree_stack_entry *prev; /* Entry below this in the stack */
2661 unsigned long lineno; /* Line number to restore */
2662 char *name; /* Position of name in opt_path */
2663 };
2665 /* The top of the path stack. */
2666 static struct tree_stack_entry *tree_stack = NULL;
2667 unsigned long tree_lineno = 0;
2669 static void
2670 pop_tree_stack_entry(void)
2671 {
2672 struct tree_stack_entry *entry = tree_stack;
2674 tree_lineno = entry->lineno;
2675 entry->name[0] = 0;
2676 tree_stack = entry->prev;
2677 free(entry);
2678 }
2680 static void
2681 push_tree_stack_entry(char *name, unsigned long lineno)
2682 {
2683 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
2684 size_t pathlen = strlen(opt_path);
2686 if (!entry)
2687 return;
2689 entry->prev = tree_stack;
2690 entry->name = opt_path + pathlen;
2691 tree_stack = entry;
2693 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
2694 pop_tree_stack_entry();
2695 return;
2696 }
2698 /* Move the current line to the first tree entry. */
2699 tree_lineno = 1;
2700 entry->lineno = lineno;
2701 }
2703 /* Parse output from git-ls-tree(1):
2704 *
2705 * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
2706 * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
2707 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
2708 * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
2709 */
2711 #define SIZEOF_TREE_ATTR \
2712 STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
2714 #define TREE_UP_FORMAT "040000 tree %s\t.."
2716 static int
2717 tree_compare_entry(enum line_type type1, char *name1,
2718 enum line_type type2, char *name2)
2719 {
2720 if (type1 != type2) {
2721 if (type1 == LINE_TREE_DIR)
2722 return -1;
2723 return 1;
2724 }
2726 return strcmp(name1, name2);
2727 }
2729 static bool
2730 tree_read(struct view *view, char *text)
2731 {
2732 size_t textlen = text ? strlen(text) : 0;
2733 char buf[SIZEOF_STR];
2734 unsigned long pos;
2735 enum line_type type;
2736 bool first_read = view->lines == 0;
2738 if (textlen <= SIZEOF_TREE_ATTR)
2739 return FALSE;
2741 type = text[STRING_SIZE("100644 ")] == 't'
2742 ? LINE_TREE_DIR : LINE_TREE_FILE;
2744 if (first_read) {
2745 /* Add path info line */
2746 if (!string_format(buf, "Directory path /%s", opt_path) ||
2747 !realloc_lines(view, view->line_size + 1) ||
2748 !add_line_text(view, buf, LINE_DEFAULT))
2749 return FALSE;
2751 /* Insert "link" to parent directory. */
2752 if (*opt_path) {
2753 if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
2754 !realloc_lines(view, view->line_size + 1) ||
2755 !add_line_text(view, buf, LINE_TREE_DIR))
2756 return FALSE;
2757 }
2758 }
2760 /* Strip the path part ... */
2761 if (*opt_path) {
2762 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
2763 size_t striplen = strlen(opt_path);
2764 char *path = text + SIZEOF_TREE_ATTR;
2766 if (pathlen > striplen)
2767 memmove(path, path + striplen,
2768 pathlen - striplen + 1);
2769 }
2771 /* Skip "Directory ..." and ".." line. */
2772 for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
2773 struct line *line = &view->line[pos];
2774 char *path1 = ((char *) line->data) + SIZEOF_TREE_ATTR;
2775 char *path2 = text + SIZEOF_TREE_ATTR;
2776 int cmp = tree_compare_entry(line->type, path1, type, path2);
2778 if (cmp <= 0)
2779 continue;
2781 text = strdup(text);
2782 if (!text)
2783 return FALSE;
2785 if (view->lines > pos)
2786 memmove(&view->line[pos + 1], &view->line[pos],
2787 (view->lines - pos) * sizeof(*line));
2789 line = &view->line[pos];
2790 line->data = text;
2791 line->type = type;
2792 view->lines++;
2793 return TRUE;
2794 }
2796 if (!add_line_text(view, text, type))
2797 return FALSE;
2799 if (tree_lineno > view->lineno) {
2800 view->lineno = tree_lineno;
2801 tree_lineno = 0;
2802 }
2804 return TRUE;
2805 }
2807 static enum request
2808 tree_request(struct view *view, enum request request, struct line *line)
2809 {
2810 enum open_flags flags;
2812 if (request != REQ_ENTER)
2813 return request;
2815 /* Cleanup the stack if the tree view is at a different tree. */
2816 while (!*opt_path && tree_stack)
2817 pop_tree_stack_entry();
2819 switch (line->type) {
2820 case LINE_TREE_DIR:
2821 /* Depending on whether it is a subdir or parent (updir?) link
2822 * mangle the path buffer. */
2823 if (line == &view->line[1] && *opt_path) {
2824 pop_tree_stack_entry();
2826 } else {
2827 char *data = line->data;
2828 char *basename = data + SIZEOF_TREE_ATTR;
2830 push_tree_stack_entry(basename, view->lineno);
2831 }
2833 /* Trees and subtrees share the same ID, so they are not not
2834 * unique like blobs. */
2835 flags = OPEN_RELOAD;
2836 request = REQ_VIEW_TREE;
2837 break;
2839 case LINE_TREE_FILE:
2840 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2841 request = REQ_VIEW_BLOB;
2842 break;
2844 default:
2845 return TRUE;
2846 }
2848 open_view(view, request, flags);
2849 if (request == REQ_VIEW_TREE) {
2850 view->lineno = tree_lineno;
2851 }
2853 return REQ_NONE;
2854 }
2856 static void
2857 tree_select(struct view *view, struct line *line)
2858 {
2859 char *text = line->data + STRING_SIZE("100644 blob ");
2861 if (line->type == LINE_TREE_FILE) {
2862 string_copy_rev(ref_blob, text);
2864 } else if (line->type != LINE_TREE_DIR) {
2865 return;
2866 }
2868 string_copy_rev(view->ref, text);
2869 }
2871 static struct view_ops tree_ops = {
2872 "file",
2873 NULL,
2874 tree_read,
2875 pager_draw,
2876 tree_request,
2877 pager_grep,
2878 tree_select,
2879 };
2881 static bool
2882 blob_read(struct view *view, char *line)
2883 {
2884 return add_line_text(view, line, LINE_DEFAULT) != NULL;
2885 }
2887 static struct view_ops blob_ops = {
2888 "line",
2889 NULL,
2890 blob_read,
2891 pager_draw,
2892 pager_request,
2893 pager_grep,
2894 pager_select,
2895 };
2898 /*
2899 * Status backend
2900 */
2902 struct status {
2903 char status;
2904 struct {
2905 mode_t mode;
2906 char rev[SIZEOF_REV];
2907 } old;
2908 struct {
2909 mode_t mode;
2910 char rev[SIZEOF_REV];
2911 } new;
2912 char name[SIZEOF_STR];
2913 };
2915 static struct status stage_status;
2916 static enum line_type stage_line_type;
2918 /* Get fields from the diff line:
2919 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
2920 */
2921 static inline bool
2922 status_get_diff(struct status *file, char *buf, size_t bufsize)
2923 {
2924 char *old_mode = buf + 1;
2925 char *new_mode = buf + 8;
2926 char *old_rev = buf + 15;
2927 char *new_rev = buf + 56;
2928 char *status = buf + 97;
2930 if (bufsize != 99 ||
2931 old_mode[-1] != ':' ||
2932 new_mode[-1] != ' ' ||
2933 old_rev[-1] != ' ' ||
2934 new_rev[-1] != ' ' ||
2935 status[-1] != ' ')
2936 return FALSE;
2938 file->status = *status;
2940 string_copy_rev(file->old.rev, old_rev);
2941 string_copy_rev(file->new.rev, new_rev);
2943 file->old.mode = strtoul(old_mode, NULL, 8);
2944 file->new.mode = strtoul(new_mode, NULL, 8);
2946 file->name[0] = 0;
2948 return TRUE;
2949 }
2951 static bool
2952 status_run(struct view *view, const char cmd[], bool diff, enum line_type type)
2953 {
2954 struct status *file = NULL;
2955 char buf[SIZEOF_STR * 4];
2956 size_t bufsize = 0;
2957 FILE *pipe;
2959 pipe = popen(cmd, "r");
2960 if (!pipe)
2961 return FALSE;
2963 add_line_data(view, NULL, type);
2965 while (!feof(pipe) && !ferror(pipe)) {
2966 char *sep;
2967 size_t readsize;
2969 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
2970 if (!readsize)
2971 break;
2972 bufsize += readsize;
2974 /* Process while we have NUL chars. */
2975 while ((sep = memchr(buf, 0, bufsize))) {
2976 size_t sepsize = sep - buf + 1;
2978 if (!file) {
2979 if (!realloc_lines(view, view->line_size + 1))
2980 goto error_out;
2982 file = calloc(1, sizeof(*file));
2983 if (!file)
2984 goto error_out;
2986 add_line_data(view, file, type);
2987 }
2989 /* Parse diff info part. */
2990 if (!diff) {
2991 file->status = '?';
2993 } else if (!file->status) {
2994 if (!status_get_diff(file, buf, sepsize))
2995 goto error_out;
2997 bufsize -= sepsize;
2998 memmove(buf, sep + 1, bufsize);
3000 sep = memchr(buf, 0, bufsize);
3001 if (!sep)
3002 break;
3003 sepsize = sep - buf + 1;
3004 }
3006 /* git-ls-files just delivers a NUL separated
3007 * list of file names similar to the second half
3008 * of the git-diff-* output. */
3009 string_ncopy(file->name, buf, sepsize);
3010 bufsize -= sepsize;
3011 memmove(buf, sep + 1, bufsize);
3012 file = NULL;
3013 }
3014 }
3016 if (ferror(pipe)) {
3017 error_out:
3018 pclose(pipe);
3019 return FALSE;
3020 }
3022 if (!view->line[view->lines - 1].data)
3023 add_line_data(view, NULL, LINE_STAT_NONE);
3025 pclose(pipe);
3026 return TRUE;
3027 }
3029 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --cached HEAD"
3030 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
3031 #define STATUS_LIST_OTHER_CMD \
3032 "git ls-files -z --others --exclude-per-directory=.gitignore"
3034 #define STATUS_DIFF_SHOW_CMD \
3035 "git diff --root --patch-with-stat --find-copies-harder -B -C %s -- %s 2>/dev/null"
3037 /* First parse staged info using git-diff-index(1), then parse unstaged
3038 * info using git-diff-files(1), and finally untracked files using
3039 * git-ls-files(1). */
3040 static bool
3041 status_open(struct view *view)
3042 {
3043 struct stat statbuf;
3044 char exclude[SIZEOF_STR];
3045 char cmd[SIZEOF_STR];
3046 size_t i;
3048 for (i = 0; i < view->lines; i++)
3049 free(view->line[i].data);
3050 free(view->line);
3051 view->lines = view->line_size = 0;
3052 view->line = NULL;
3054 if (!realloc_lines(view, view->line_size + 6))
3055 return FALSE;
3057 if (!string_format(exclude, "%s/info/exclude", opt_git_dir))
3058 return FALSE;
3060 string_copy(cmd, STATUS_LIST_OTHER_CMD);
3062 if (stat(exclude, &statbuf) >= 0) {
3063 size_t cmdsize = strlen(cmd);
3065 if (!string_format_from(cmd, &cmdsize, " %s", "--exclude-from=") ||
3066 sq_quote(cmd, cmdsize, exclude) >= sizeof(cmd))
3067 return FALSE;
3068 }
3070 if (!status_run(view, STATUS_DIFF_INDEX_CMD, TRUE, LINE_STAT_STAGED) ||
3071 !status_run(view, STATUS_DIFF_FILES_CMD, TRUE, LINE_STAT_UNSTAGED) ||
3072 !status_run(view, cmd, FALSE, LINE_STAT_UNTRACKED))
3073 return FALSE;
3075 return TRUE;
3076 }
3078 static bool
3079 status_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3080 {
3081 struct status *status = line->data;
3083 wmove(view->win, lineno, 0);
3085 if (selected) {
3086 wattrset(view->win, get_line_attr(LINE_CURSOR));
3087 wchgat(view->win, -1, 0, LINE_CURSOR, NULL);
3089 } else if (!status && line->type != LINE_STAT_NONE) {
3090 wattrset(view->win, get_line_attr(LINE_STAT_SECTION));
3091 wchgat(view->win, -1, 0, LINE_STAT_SECTION, NULL);
3093 } else {
3094 wattrset(view->win, get_line_attr(line->type));
3095 }
3097 if (!status) {
3098 char *text;
3100 switch (line->type) {
3101 case LINE_STAT_STAGED:
3102 text = "Changes to be committed:";
3103 break;
3105 case LINE_STAT_UNSTAGED:
3106 text = "Changed but not updated:";
3107 break;
3109 case LINE_STAT_UNTRACKED:
3110 text = "Untracked files:";
3111 break;
3113 case LINE_STAT_NONE:
3114 text = " (no files)";
3115 break;
3117 default:
3118 return FALSE;
3119 }
3121 waddstr(view->win, text);
3122 return TRUE;
3123 }
3125 waddch(view->win, status->status);
3126 if (!selected)
3127 wattrset(view->win, A_NORMAL);
3128 wmove(view->win, lineno, 4);
3129 waddstr(view->win, status->name);
3131 return TRUE;
3132 }
3134 static enum request
3135 status_enter(struct view *view, struct line *line)
3136 {
3137 struct status *status = line->data;
3138 char path[SIZEOF_STR] = "";
3139 char *info;
3140 size_t cmdsize = 0;
3142 if (line->type == LINE_STAT_NONE ||
3143 (!status && line[1].type == LINE_STAT_NONE)) {
3144 report("No file to diff");
3145 return REQ_NONE;
3146 }
3148 if (status && sq_quote(path, 0, status->name) >= sizeof(path))
3149 return REQ_QUIT;
3151 if (opt_cdup[0] &&
3152 line->type != LINE_STAT_UNTRACKED &&
3153 !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
3154 return REQ_QUIT;
3156 switch (line->type) {
3157 case LINE_STAT_STAGED:
3158 if (!string_format_from(opt_cmd, &cmdsize, STATUS_DIFF_SHOW_CMD,
3159 "--cached", path))
3160 return REQ_QUIT;
3161 if (status)
3162 info = "Staged changes to %s";
3163 else
3164 info = "Staged changes";
3165 break;
3167 case LINE_STAT_UNSTAGED:
3168 if (!string_format_from(opt_cmd, &cmdsize, STATUS_DIFF_SHOW_CMD,
3169 "", path))
3170 return REQ_QUIT;
3171 if (status)
3172 info = "Unstaged changes to %s";
3173 else
3174 info = "Unstaged changes";
3175 break;
3177 case LINE_STAT_UNTRACKED:
3178 if (opt_pipe)
3179 return REQ_QUIT;
3182 if (!status) {
3183 report("No file to show");
3184 return REQ_NONE;
3185 }
3187 opt_pipe = fopen(status->name, "r");
3188 info = "Untracked file %s";
3189 break;
3191 default:
3192 die("w00t");
3193 }
3195 open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | OPEN_SPLIT);
3196 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
3197 if (status) {
3198 stage_status = *status;
3199 } else {
3200 memset(&stage_status, 0, sizeof(stage_status));
3201 }
3203 stage_line_type = line->type;
3204 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.name);
3205 }
3207 return REQ_NONE;
3208 }
3211 static bool
3212 status_update_file(struct view *view, struct status *status, enum line_type type)
3213 {
3214 char cmd[SIZEOF_STR];
3215 char buf[SIZEOF_STR];
3216 size_t cmdsize = 0;
3217 size_t bufsize = 0;
3218 size_t written = 0;
3219 FILE *pipe;
3221 if (opt_cdup[0] &&
3222 type != LINE_STAT_UNTRACKED &&
3223 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
3224 return FALSE;
3226 switch (type) {
3227 case LINE_STAT_STAGED:
3228 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
3229 status->old.mode,
3230 status->old.rev,
3231 status->name, 0))
3232 return FALSE;
3234 string_add(cmd, cmdsize, "git update-index -z --index-info");
3235 break;
3237 case LINE_STAT_UNSTAGED:
3238 case LINE_STAT_UNTRACKED:
3239 if (!string_format_from(buf, &bufsize, "%s%c", status->name, 0))
3240 return FALSE;
3242 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
3243 break;
3245 default:
3246 die("w00t");
3247 }
3249 pipe = popen(cmd, "w");
3250 if (!pipe)
3251 return FALSE;
3253 while (!ferror(pipe) && written < bufsize) {
3254 written += fwrite(buf + written, 1, bufsize - written, pipe);
3255 }
3257 pclose(pipe);
3259 if (written != bufsize)
3260 return FALSE;
3262 return TRUE;
3263 }
3265 static void
3266 status_update(struct view *view)
3267 {
3268 struct line *line = &view->line[view->lineno];
3270 assert(view->lines);
3272 if (!line->data) {
3273 while (++line < view->line + view->lines && line->data) {
3274 if (!status_update_file(view, line->data, line->type))
3275 report("Failed to update file status");
3276 }
3278 if (!line[-1].data) {
3279 report("Nothing to update");
3280 return;
3281 }
3283 } else if (!status_update_file(view, line->data, line->type)) {
3284 report("Failed to update file status");
3285 }
3287 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
3288 }
3290 static enum request
3291 status_request(struct view *view, enum request request, struct line *line)
3292 {
3293 struct status *status = line->data;
3295 switch (request) {
3296 case REQ_STATUS_UPDATE:
3297 status_update(view);
3298 break;
3300 case REQ_EDIT:
3301 if (!status)
3302 return request;
3304 open_editor(view, status->name);
3305 break;
3307 case REQ_ENTER:
3308 status_enter(view, line);
3309 break;
3311 default:
3312 return request;
3313 }
3315 return REQ_NONE;
3316 }
3318 static void
3319 status_select(struct view *view, struct line *line)
3320 {
3321 struct status *status = line->data;
3322 char file[SIZEOF_STR] = "all files";
3323 char *text;
3325 if (status && !string_format(file, "'%s'", status->name))
3326 return;
3328 if (!status && line[1].type == LINE_STAT_NONE)
3329 line++;
3331 switch (line->type) {
3332 case LINE_STAT_STAGED:
3333 text = "Press %s to unstage %s for commit";
3334 break;
3336 case LINE_STAT_UNSTAGED:
3337 text = "Press %s to stage %s for commit";
3338 break;
3340 case LINE_STAT_UNTRACKED:
3341 text = "Press %s to stage %s for addition";
3342 break;
3344 case LINE_STAT_NONE:
3345 text = "Nothing to update";
3346 break;
3348 default:
3349 die("w00t");
3350 }
3352 string_format(view->ref, text, get_key(REQ_STATUS_UPDATE), file);
3353 }
3355 static bool
3356 status_grep(struct view *view, struct line *line)
3357 {
3358 struct status *status = line->data;
3359 enum { S_STATUS, S_NAME, S_END } state;
3360 char buf[2] = "?";
3361 regmatch_t pmatch;
3363 if (!status)
3364 return FALSE;
3366 for (state = S_STATUS; state < S_END; state++) {
3367 char *text;
3369 switch (state) {
3370 case S_NAME: text = status->name; break;
3371 case S_STATUS:
3372 buf[0] = status->status;
3373 text = buf;
3374 break;
3376 default:
3377 return FALSE;
3378 }
3380 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3381 return TRUE;
3382 }
3384 return FALSE;
3385 }
3387 static struct view_ops status_ops = {
3388 "file",
3389 status_open,
3390 NULL,
3391 status_draw,
3392 status_request,
3393 status_grep,
3394 status_select,
3395 };
3398 static bool
3399 stage_diff_line(FILE *pipe, struct line *line)
3400 {
3401 char *buf = line->data;
3402 size_t bufsize = strlen(buf);
3403 size_t written = 0;
3405 while (!ferror(pipe) && written < bufsize) {
3406 written += fwrite(buf + written, 1, bufsize - written, pipe);
3407 }
3409 fputc('\n', pipe);
3411 return written == bufsize;
3412 }
3414 static struct line *
3415 stage_diff_hdr(struct view *view, struct line *line)
3416 {
3417 int diff_hdr_dir = line->type == LINE_DIFF_CHUNK ? -1 : 1;
3418 struct line *diff_hdr;
3420 if (line->type == LINE_DIFF_CHUNK)
3421 diff_hdr = line - 1;
3422 else
3423 diff_hdr = view->line + 1;
3425 while (diff_hdr > view->line && diff_hdr < view->line + view->lines) {
3426 if (diff_hdr->type == LINE_DIFF_HEADER)
3427 return diff_hdr;
3429 diff_hdr += diff_hdr_dir;
3430 }
3432 return NULL;
3433 }
3435 static bool
3436 stage_update_chunk(struct view *view, struct line *line)
3437 {
3438 char cmd[SIZEOF_STR];
3439 size_t cmdsize = 0;
3440 struct line *diff_hdr, *diff_chunk, *diff_end;
3441 FILE *pipe;
3443 diff_hdr = stage_diff_hdr(view, line);
3444 if (!diff_hdr)
3445 return FALSE;
3447 if (opt_cdup[0] &&
3448 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
3449 return FALSE;
3451 if (!string_format_from(cmd, &cmdsize,
3452 "git apply --cached %s - && "
3453 "git update-index -q --unmerged --refresh 2>/dev/null",
3454 stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
3455 return FALSE;
3457 pipe = popen(cmd, "w");
3458 if (!pipe)
3459 return FALSE;
3461 diff_end = view->line + view->lines;
3462 if (line->type != LINE_DIFF_CHUNK) {
3463 diff_chunk = diff_hdr;
3465 } else {
3466 for (diff_chunk = line + 1; diff_chunk < diff_end; diff_chunk++)
3467 if (diff_chunk->type == LINE_DIFF_CHUNK ||
3468 diff_chunk->type == LINE_DIFF_HEADER)
3469 diff_end = diff_chunk;
3471 diff_chunk = line;
3473 while (diff_hdr->type != LINE_DIFF_CHUNK) {
3474 switch (diff_hdr->type) {
3475 case LINE_DIFF_HEADER:
3476 case LINE_DIFF_INDEX:
3477 case LINE_DIFF_ADD:
3478 case LINE_DIFF_DEL:
3479 break;
3481 default:
3482 diff_hdr++;
3483 continue;
3484 }
3486 if (!stage_diff_line(pipe, diff_hdr++)) {
3487 pclose(pipe);
3488 return FALSE;
3489 }
3490 }
3491 }
3493 while (diff_chunk < diff_end && stage_diff_line(pipe, diff_chunk))
3494 diff_chunk++;
3496 pclose(pipe);
3498 if (diff_chunk != diff_end)
3499 return FALSE;
3501 return TRUE;
3502 }
3504 static void
3505 stage_update(struct view *view, struct line *line)
3506 {
3507 if (stage_line_type != LINE_STAT_UNTRACKED &&
3508 (line->type == LINE_DIFF_CHUNK || !stage_status.status)) {
3509 if (!stage_update_chunk(view, line)) {
3510 report("Failed to apply chunk");
3511 return;
3512 }
3514 } else if (!status_update_file(view, &stage_status, stage_line_type)) {
3515 report("Failed to update file");
3516 return;
3517 }
3519 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
3521 view = VIEW(REQ_VIEW_STATUS);
3522 if (view_is_displayed(view))
3523 status_enter(view, &view->line[view->lineno]);
3524 }
3526 static enum request
3527 stage_request(struct view *view, enum request request, struct line *line)
3528 {
3529 switch (request) {
3530 case REQ_STATUS_UPDATE:
3531 stage_update(view, line);
3532 break;
3534 case REQ_EDIT:
3535 if (!stage_status.name[0])
3536 return request;
3538 open_editor(view, stage_status.name);
3539 break;
3541 case REQ_ENTER:
3542 pager_request(view, request, line);
3543 break;
3545 default:
3546 return request;
3547 }
3549 return REQ_NONE;
3550 }
3552 static struct view_ops stage_ops = {
3553 "line",
3554 NULL,
3555 pager_read,
3556 pager_draw,
3557 stage_request,
3558 pager_grep,
3559 pager_select,
3560 };
3563 /*
3564 * Revision graph
3565 */
3567 struct commit {
3568 char id[SIZEOF_REV]; /* SHA1 ID. */
3569 char title[128]; /* First line of the commit message. */
3570 char author[75]; /* Author of the commit. */
3571 struct tm time; /* Date from the author ident. */
3572 struct ref **refs; /* Repository references. */
3573 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
3574 size_t graph_size; /* The width of the graph array. */
3575 };
3577 /* Size of rev graph with no "padding" columns */
3578 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
3580 struct rev_graph {
3581 struct rev_graph *prev, *next, *parents;
3582 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
3583 size_t size;
3584 struct commit *commit;
3585 size_t pos;
3586 };
3588 /* Parents of the commit being visualized. */
3589 static struct rev_graph graph_parents[4];
3591 /* The current stack of revisions on the graph. */
3592 static struct rev_graph graph_stacks[4] = {
3593 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
3594 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
3595 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
3596 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
3597 };
3599 static inline bool
3600 graph_parent_is_merge(struct rev_graph *graph)
3601 {
3602 return graph->parents->size > 1;
3603 }
3605 static inline void
3606 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
3607 {
3608 struct commit *commit = graph->commit;
3610 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
3611 commit->graph[commit->graph_size++] = symbol;
3612 }
3614 static void
3615 done_rev_graph(struct rev_graph *graph)
3616 {
3617 if (graph_parent_is_merge(graph) &&
3618 graph->pos < graph->size - 1 &&
3619 graph->next->size == graph->size + graph->parents->size - 1) {
3620 size_t i = graph->pos + graph->parents->size - 1;
3622 graph->commit->graph_size = i * 2;
3623 while (i < graph->next->size - 1) {
3624 append_to_rev_graph(graph, ' ');
3625 append_to_rev_graph(graph, '\\');
3626 i++;
3627 }
3628 }
3630 graph->size = graph->pos = 0;
3631 graph->commit = NULL;
3632 memset(graph->parents, 0, sizeof(*graph->parents));
3633 }
3635 static void
3636 push_rev_graph(struct rev_graph *graph, char *parent)
3637 {
3638 int i;
3640 /* "Collapse" duplicate parents lines.
3641 *
3642 * FIXME: This needs to also update update the drawn graph but
3643 * for now it just serves as a method for pruning graph lines. */
3644 for (i = 0; i < graph->size; i++)
3645 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
3646 return;
3648 if (graph->size < SIZEOF_REVITEMS) {
3649 string_copy_rev(graph->rev[graph->size++], parent);
3650 }
3651 }
3653 static chtype
3654 get_rev_graph_symbol(struct rev_graph *graph)
3655 {
3656 chtype symbol;
3658 if (graph->parents->size == 0)
3659 symbol = REVGRAPH_INIT;
3660 else if (graph_parent_is_merge(graph))
3661 symbol = REVGRAPH_MERGE;
3662 else if (graph->pos >= graph->size)
3663 symbol = REVGRAPH_BRANCH;
3664 else
3665 symbol = REVGRAPH_COMMIT;
3667 return symbol;
3668 }
3670 static void
3671 draw_rev_graph(struct rev_graph *graph)
3672 {
3673 struct rev_filler {
3674 chtype separator, line;
3675 };
3676 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
3677 static struct rev_filler fillers[] = {
3678 { ' ', REVGRAPH_LINE },
3679 { '`', '.' },
3680 { '\'', ' ' },
3681 { '/', ' ' },
3682 };
3683 chtype symbol = get_rev_graph_symbol(graph);
3684 struct rev_filler *filler;
3685 size_t i;
3687 filler = &fillers[DEFAULT];
3689 for (i = 0; i < graph->pos; i++) {
3690 append_to_rev_graph(graph, filler->line);
3691 if (graph_parent_is_merge(graph->prev) &&
3692 graph->prev->pos == i)
3693 filler = &fillers[RSHARP];
3695 append_to_rev_graph(graph, filler->separator);
3696 }
3698 /* Place the symbol for this revision. */
3699 append_to_rev_graph(graph, symbol);
3701 if (graph->prev->size > graph->size)
3702 filler = &fillers[RDIAG];
3703 else
3704 filler = &fillers[DEFAULT];
3706 i++;
3708 for (; i < graph->size; i++) {
3709 append_to_rev_graph(graph, filler->separator);
3710 append_to_rev_graph(graph, filler->line);
3711 if (graph_parent_is_merge(graph->prev) &&
3712 i < graph->prev->pos + graph->parents->size)
3713 filler = &fillers[RSHARP];
3714 if (graph->prev->size > graph->size)
3715 filler = &fillers[LDIAG];
3716 }
3718 if (graph->prev->size > graph->size) {
3719 append_to_rev_graph(graph, filler->separator);
3720 if (filler->line != ' ')
3721 append_to_rev_graph(graph, filler->line);
3722 }
3723 }
3725 /* Prepare the next rev graph */
3726 static void
3727 prepare_rev_graph(struct rev_graph *graph)
3728 {
3729 size_t i;
3731 /* First, traverse all lines of revisions up to the active one. */
3732 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
3733 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
3734 break;
3736 push_rev_graph(graph->next, graph->rev[graph->pos]);
3737 }
3739 /* Interleave the new revision parent(s). */
3740 for (i = 0; i < graph->parents->size; i++)
3741 push_rev_graph(graph->next, graph->parents->rev[i]);
3743 /* Lastly, put any remaining revisions. */
3744 for (i = graph->pos + 1; i < graph->size; i++)
3745 push_rev_graph(graph->next, graph->rev[i]);
3746 }
3748 static void
3749 update_rev_graph(struct rev_graph *graph)
3750 {
3751 /* If this is the finalizing update ... */
3752 if (graph->commit)
3753 prepare_rev_graph(graph);
3755 /* Graph visualization needs a one rev look-ahead,
3756 * so the first update doesn't visualize anything. */
3757 if (!graph->prev->commit)
3758 return;
3760 draw_rev_graph(graph->prev);
3761 done_rev_graph(graph->prev->prev);
3762 }
3765 /*
3766 * Main view backend
3767 */
3769 static bool
3770 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3771 {
3772 char buf[DATE_COLS + 1];
3773 struct commit *commit = line->data;
3774 enum line_type type;
3775 int col = 0;
3776 size_t timelen;
3777 size_t authorlen;
3778 int trimmed = 1;
3780 if (!*commit->author)
3781 return FALSE;
3783 wmove(view->win, lineno, col);
3785 if (selected) {
3786 type = LINE_CURSOR;
3787 wattrset(view->win, get_line_attr(type));
3788 wchgat(view->win, -1, 0, type, NULL);
3790 } else {
3791 type = LINE_MAIN_COMMIT;
3792 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
3793 }
3795 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
3796 waddnstr(view->win, buf, timelen);
3797 waddstr(view->win, " ");
3799 col += DATE_COLS;
3800 wmove(view->win, lineno, col);
3801 if (type != LINE_CURSOR)
3802 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
3804 if (opt_utf8) {
3805 authorlen = utf8_length(commit->author, AUTHOR_COLS - 2, &col, &trimmed);
3806 } else {
3807 authorlen = strlen(commit->author);
3808 if (authorlen > AUTHOR_COLS - 2) {
3809 authorlen = AUTHOR_COLS - 2;
3810 trimmed = 1;
3811 }
3812 }
3814 if (trimmed) {
3815 waddnstr(view->win, commit->author, authorlen);
3816 if (type != LINE_CURSOR)
3817 wattrset(view->win, get_line_attr(LINE_MAIN_DELIM));
3818 waddch(view->win, '~');
3819 } else {
3820 waddstr(view->win, commit->author);
3821 }
3823 col += AUTHOR_COLS;
3824 if (type != LINE_CURSOR)
3825 wattrset(view->win, A_NORMAL);
3827 if (opt_rev_graph && commit->graph_size) {
3828 size_t i;
3830 wmove(view->win, lineno, col);
3831 /* Using waddch() instead of waddnstr() ensures that
3832 * they'll be rendered correctly for the cursor line. */
3833 for (i = 0; i < commit->graph_size; i++)
3834 waddch(view->win, commit->graph[i]);
3836 waddch(view->win, ' ');
3837 col += commit->graph_size + 1;
3838 }
3840 wmove(view->win, lineno, col);
3842 if (commit->refs) {
3843 size_t i = 0;
3845 do {
3846 if (type == LINE_CURSOR)
3847 ;
3848 else if (commit->refs[i]->tag)
3849 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
3850 else if (commit->refs[i]->remote)
3851 wattrset(view->win, get_line_attr(LINE_MAIN_REMOTE));
3852 else
3853 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
3854 waddstr(view->win, "[");
3855 waddstr(view->win, commit->refs[i]->name);
3856 waddstr(view->win, "]");
3857 if (type != LINE_CURSOR)
3858 wattrset(view->win, A_NORMAL);
3859 waddstr(view->win, " ");
3860 col += strlen(commit->refs[i]->name) + STRING_SIZE("[] ");
3861 } while (commit->refs[i++]->next);
3862 }
3864 if (type != LINE_CURSOR)
3865 wattrset(view->win, get_line_attr(type));
3867 {
3868 int titlelen = strlen(commit->title);
3870 if (col + titlelen > view->width)
3871 titlelen = view->width - col;
3873 waddnstr(view->win, commit->title, titlelen);
3874 }
3876 return TRUE;
3877 }
3879 /* Reads git log --pretty=raw output and parses it into the commit struct. */
3880 static bool
3881 main_read(struct view *view, char *line)
3882 {
3883 static struct rev_graph *graph = graph_stacks;
3884 enum line_type type;
3885 struct commit *commit;
3887 if (!line) {
3888 update_rev_graph(graph);
3889 return TRUE;
3890 }
3892 type = get_line_type(line);
3893 if (type == LINE_COMMIT) {
3894 commit = calloc(1, sizeof(struct commit));
3895 if (!commit)
3896 return FALSE;
3898 string_copy_rev(commit->id, line + STRING_SIZE("commit "));
3899 commit->refs = get_refs(commit->id);
3900 graph->commit = commit;
3901 add_line_data(view, commit, LINE_MAIN_COMMIT);
3902 return TRUE;
3903 }
3905 if (!view->lines)
3906 return TRUE;
3907 commit = view->line[view->lines - 1].data;
3909 switch (type) {
3910 case LINE_PARENT:
3911 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
3912 break;
3914 case LINE_AUTHOR:
3915 {
3916 /* Parse author lines where the name may be empty:
3917 * author <email@address.tld> 1138474660 +0100
3918 */
3919 char *ident = line + STRING_SIZE("author ");
3920 char *nameend = strchr(ident, '<');
3921 char *emailend = strchr(ident, '>');
3923 if (!nameend || !emailend)
3924 break;
3926 update_rev_graph(graph);
3927 graph = graph->next;
3929 *nameend = *emailend = 0;
3930 ident = chomp_string(ident);
3931 if (!*ident) {
3932 ident = chomp_string(nameend + 1);
3933 if (!*ident)
3934 ident = "Unknown";
3935 }
3937 string_ncopy(commit->author, ident, strlen(ident));
3939 /* Parse epoch and timezone */
3940 if (emailend[1] == ' ') {
3941 char *secs = emailend + 2;
3942 char *zone = strchr(secs, ' ');
3943 time_t time = (time_t) atol(secs);
3945 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
3946 long tz;
3948 zone++;
3949 tz = ('0' - zone[1]) * 60 * 60 * 10;
3950 tz += ('0' - zone[2]) * 60 * 60;
3951 tz += ('0' - zone[3]) * 60;
3952 tz += ('0' - zone[4]) * 60;
3954 if (zone[0] == '-')
3955 tz = -tz;
3957 time -= tz;
3958 }
3960 gmtime_r(&time, &commit->time);
3961 }
3962 break;
3963 }
3964 default:
3965 /* Fill in the commit title if it has not already been set. */
3966 if (commit->title[0])
3967 break;
3969 /* Require titles to start with a non-space character at the
3970 * offset used by git log. */
3971 if (strncmp(line, " ", 4))
3972 break;
3973 line += 4;
3974 /* Well, if the title starts with a whitespace character,
3975 * try to be forgiving. Otherwise we end up with no title. */
3976 while (isspace(*line))
3977 line++;
3978 if (*line == '\0')
3979 break;
3980 /* FIXME: More graceful handling of titles; append "..." to
3981 * shortened titles, etc. */
3983 string_ncopy(commit->title, line, strlen(line));
3984 }
3986 return TRUE;
3987 }
3989 static enum request
3990 main_request(struct view *view, enum request request, struct line *line)
3991 {
3992 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3994 if (request == REQ_ENTER)
3995 open_view(view, REQ_VIEW_DIFF, flags);
3996 else
3997 return request;
3999 return REQ_NONE;
4000 }
4002 static bool
4003 main_grep(struct view *view, struct line *line)
4004 {
4005 struct commit *commit = line->data;
4006 enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
4007 char buf[DATE_COLS + 1];
4008 regmatch_t pmatch;
4010 for (state = S_TITLE; state < S_END; state++) {
4011 char *text;
4013 switch (state) {
4014 case S_TITLE: text = commit->title; break;
4015 case S_AUTHOR: text = commit->author; break;
4016 case S_DATE:
4017 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
4018 continue;
4019 text = buf;
4020 break;
4022 default:
4023 return FALSE;
4024 }
4026 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4027 return TRUE;
4028 }
4030 return FALSE;
4031 }
4033 static void
4034 main_select(struct view *view, struct line *line)
4035 {
4036 struct commit *commit = line->data;
4038 string_copy_rev(view->ref, commit->id);
4039 string_copy_rev(ref_commit, view->ref);
4040 }
4042 static struct view_ops main_ops = {
4043 "commit",
4044 NULL,
4045 main_read,
4046 main_draw,
4047 main_request,
4048 main_grep,
4049 main_select,
4050 };
4053 /*
4054 * Unicode / UTF-8 handling
4055 *
4056 * NOTE: Much of the following code for dealing with unicode is derived from
4057 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
4058 * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
4059 */
4061 /* I've (over)annotated a lot of code snippets because I am not entirely
4062 * confident that the approach taken by this small UTF-8 interface is correct.
4063 * --jonas */
4065 static inline int
4066 unicode_width(unsigned long c)
4067 {
4068 if (c >= 0x1100 &&
4069 (c <= 0x115f /* Hangul Jamo */
4070 || c == 0x2329
4071 || c == 0x232a
4072 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
4073 /* CJK ... Yi */
4074 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
4075 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
4076 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
4077 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
4078 || (c >= 0xffe0 && c <= 0xffe6)
4079 || (c >= 0x20000 && c <= 0x2fffd)
4080 || (c >= 0x30000 && c <= 0x3fffd)))
4081 return 2;
4083 return 1;
4084 }
4086 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
4087 * Illegal bytes are set one. */
4088 static const unsigned char utf8_bytes[256] = {
4089 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,
4090 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,
4091 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,
4092 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,
4093 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,
4094 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,
4095 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,
4096 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,
4097 };
4099 /* Decode UTF-8 multi-byte representation into a unicode character. */
4100 static inline unsigned long
4101 utf8_to_unicode(const char *string, size_t length)
4102 {
4103 unsigned long unicode;
4105 switch (length) {
4106 case 1:
4107 unicode = string[0];
4108 break;
4109 case 2:
4110 unicode = (string[0] & 0x1f) << 6;
4111 unicode += (string[1] & 0x3f);
4112 break;
4113 case 3:
4114 unicode = (string[0] & 0x0f) << 12;
4115 unicode += ((string[1] & 0x3f) << 6);
4116 unicode += (string[2] & 0x3f);
4117 break;
4118 case 4:
4119 unicode = (string[0] & 0x0f) << 18;
4120 unicode += ((string[1] & 0x3f) << 12);
4121 unicode += ((string[2] & 0x3f) << 6);
4122 unicode += (string[3] & 0x3f);
4123 break;
4124 case 5:
4125 unicode = (string[0] & 0x0f) << 24;
4126 unicode += ((string[1] & 0x3f) << 18);
4127 unicode += ((string[2] & 0x3f) << 12);
4128 unicode += ((string[3] & 0x3f) << 6);
4129 unicode += (string[4] & 0x3f);
4130 break;
4131 case 6:
4132 unicode = (string[0] & 0x01) << 30;
4133 unicode += ((string[1] & 0x3f) << 24);
4134 unicode += ((string[2] & 0x3f) << 18);
4135 unicode += ((string[3] & 0x3f) << 12);
4136 unicode += ((string[4] & 0x3f) << 6);
4137 unicode += (string[5] & 0x3f);
4138 break;
4139 default:
4140 die("Invalid unicode length");
4141 }
4143 /* Invalid characters could return the special 0xfffd value but NUL
4144 * should be just as good. */
4145 return unicode > 0xffff ? 0 : unicode;
4146 }
4148 /* Calculates how much of string can be shown within the given maximum width
4149 * and sets trimmed parameter to non-zero value if all of string could not be
4150 * shown.
4151 *
4152 * Additionally, adds to coloffset how many many columns to move to align with
4153 * the expected position. Takes into account how multi-byte and double-width
4154 * characters will effect the cursor position.
4155 *
4156 * Returns the number of bytes to output from string to satisfy max_width. */
4157 static size_t
4158 utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed)
4159 {
4160 const char *start = string;
4161 const char *end = strchr(string, '\0');
4162 size_t mbwidth = 0;
4163 size_t width = 0;
4165 *trimmed = 0;
4167 while (string < end) {
4168 int c = *(unsigned char *) string;
4169 unsigned char bytes = utf8_bytes[c];
4170 size_t ucwidth;
4171 unsigned long unicode;
4173 if (string + bytes > end)
4174 break;
4176 /* Change representation to figure out whether
4177 * it is a single- or double-width character. */
4179 unicode = utf8_to_unicode(string, bytes);
4180 /* FIXME: Graceful handling of invalid unicode character. */
4181 if (!unicode)
4182 break;
4184 ucwidth = unicode_width(unicode);
4185 width += ucwidth;
4186 if (width > max_width) {
4187 *trimmed = 1;
4188 break;
4189 }
4191 /* The column offset collects the differences between the
4192 * number of bytes encoding a character and the number of
4193 * columns will be used for rendering said character.
4194 *
4195 * So if some character A is encoded in 2 bytes, but will be
4196 * represented on the screen using only 1 byte this will and up
4197 * adding 1 to the multi-byte column offset.
4198 *
4199 * Assumes that no double-width character can be encoding in
4200 * less than two bytes. */
4201 if (bytes > ucwidth)
4202 mbwidth += bytes - ucwidth;
4204 string += bytes;
4205 }
4207 *coloffset += mbwidth;
4209 return string - start;
4210 }
4213 /*
4214 * Status management
4215 */
4217 /* Whether or not the curses interface has been initialized. */
4218 static bool cursed = FALSE;
4220 /* The status window is used for polling keystrokes. */
4221 static WINDOW *status_win;
4223 static bool status_empty = TRUE;
4225 /* Update status and title window. */
4226 static void
4227 report(const char *msg, ...)
4228 {
4229 struct view *view = display[current_view];
4231 if (input_mode)
4232 return;
4234 if (!status_empty || *msg) {
4235 va_list args;
4237 va_start(args, msg);
4239 wmove(status_win, 0, 0);
4240 if (*msg) {
4241 vwprintw(status_win, msg, args);
4242 status_empty = FALSE;
4243 } else {
4244 status_empty = TRUE;
4245 }
4246 wclrtoeol(status_win);
4247 wrefresh(status_win);
4249 va_end(args);
4250 }
4252 update_view_title(view);
4253 update_display_cursor(view);
4254 }
4256 /* Controls when nodelay should be in effect when polling user input. */
4257 static void
4258 set_nonblocking_input(bool loading)
4259 {
4260 static unsigned int loading_views;
4262 if ((loading == FALSE && loading_views-- == 1) ||
4263 (loading == TRUE && loading_views++ == 0))
4264 nodelay(status_win, loading);
4265 }
4267 static void
4268 init_display(void)
4269 {
4270 int x, y;
4272 /* Initialize the curses library */
4273 if (isatty(STDIN_FILENO)) {
4274 cursed = !!initscr();
4275 } else {
4276 /* Leave stdin and stdout alone when acting as a pager. */
4277 FILE *io = fopen("/dev/tty", "r+");
4279 if (!io)
4280 die("Failed to open /dev/tty");
4281 cursed = !!newterm(NULL, io, io);
4282 }
4284 if (!cursed)
4285 die("Failed to initialize curses");
4287 nonl(); /* Tell curses not to do NL->CR/NL on output */
4288 cbreak(); /* Take input chars one at a time, no wait for \n */
4289 noecho(); /* Don't echo input */
4290 leaveok(stdscr, TRUE);
4292 if (has_colors())
4293 init_colors();
4295 getmaxyx(stdscr, y, x);
4296 status_win = newwin(1, 0, y - 1, 0);
4297 if (!status_win)
4298 die("Failed to create status window");
4300 /* Enable keyboard mapping */
4301 keypad(status_win, TRUE);
4302 wbkgdset(status_win, get_line_attr(LINE_STATUS));
4303 }
4305 static char *
4306 read_prompt(const char *prompt)
4307 {
4308 enum { READING, STOP, CANCEL } status = READING;
4309 static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
4310 int pos = 0;
4312 while (status == READING) {
4313 struct view *view;
4314 int i, key;
4316 input_mode = TRUE;
4318 foreach_view (view, i)
4319 update_view(view);
4321 input_mode = FALSE;
4323 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
4324 wclrtoeol(status_win);
4326 /* Refresh, accept single keystroke of input */
4327 key = wgetch(status_win);
4328 switch (key) {
4329 case KEY_RETURN:
4330 case KEY_ENTER:
4331 case '\n':
4332 status = pos ? STOP : CANCEL;
4333 break;
4335 case KEY_BACKSPACE:
4336 if (pos > 0)
4337 pos--;
4338 else
4339 status = CANCEL;
4340 break;
4342 case KEY_ESC:
4343 status = CANCEL;
4344 break;
4346 case ERR:
4347 break;
4349 default:
4350 if (pos >= sizeof(buf)) {
4351 report("Input string too long");
4352 return NULL;
4353 }
4355 if (isprint(key))
4356 buf[pos++] = (char) key;
4357 }
4358 }
4360 /* Clear the status window */
4361 status_empty = FALSE;
4362 report("");
4364 if (status == CANCEL)
4365 return NULL;
4367 buf[pos++] = 0;
4369 return buf;
4370 }
4372 /*
4373 * Repository references
4374 */
4376 static struct ref *refs;
4377 static size_t refs_size;
4379 /* Id <-> ref store */
4380 static struct ref ***id_refs;
4381 static size_t id_refs_size;
4383 static struct ref **
4384 get_refs(char *id)
4385 {
4386 struct ref ***tmp_id_refs;
4387 struct ref **ref_list = NULL;
4388 size_t ref_list_size = 0;
4389 size_t i;
4391 for (i = 0; i < id_refs_size; i++)
4392 if (!strcmp(id, id_refs[i][0]->id))
4393 return id_refs[i];
4395 tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
4396 if (!tmp_id_refs)
4397 return NULL;
4399 id_refs = tmp_id_refs;
4401 for (i = 0; i < refs_size; i++) {
4402 struct ref **tmp;
4404 if (strcmp(id, refs[i].id))
4405 continue;
4407 tmp = realloc(ref_list, (ref_list_size + 1) * sizeof(*ref_list));
4408 if (!tmp) {
4409 if (ref_list)
4410 free(ref_list);
4411 return NULL;
4412 }
4414 ref_list = tmp;
4415 if (ref_list_size > 0)
4416 ref_list[ref_list_size - 1]->next = 1;
4417 ref_list[ref_list_size] = &refs[i];
4419 /* XXX: The properties of the commit chains ensures that we can
4420 * safely modify the shared ref. The repo references will
4421 * always be similar for the same id. */
4422 ref_list[ref_list_size]->next = 0;
4423 ref_list_size++;
4424 }
4426 if (ref_list)
4427 id_refs[id_refs_size++] = ref_list;
4429 return ref_list;
4430 }
4432 static int
4433 read_ref(char *id, size_t idlen, char *name, size_t namelen)
4434 {
4435 struct ref *ref;
4436 bool tag = FALSE;
4437 bool remote = FALSE;
4439 if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
4440 /* Commits referenced by tags has "^{}" appended. */
4441 if (name[namelen - 1] != '}')
4442 return OK;
4444 while (namelen > 0 && name[namelen] != '^')
4445 namelen--;
4447 tag = TRUE;
4448 namelen -= STRING_SIZE("refs/tags/");
4449 name += STRING_SIZE("refs/tags/");
4451 } else if (!strncmp(name, "refs/remotes/", STRING_SIZE("refs/remotes/"))) {
4452 remote = TRUE;
4453 namelen -= STRING_SIZE("refs/remotes/");
4454 name += STRING_SIZE("refs/remotes/");
4456 } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
4457 namelen -= STRING_SIZE("refs/heads/");
4458 name += STRING_SIZE("refs/heads/");
4460 } else if (!strcmp(name, "HEAD")) {
4461 return OK;
4462 }
4464 refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
4465 if (!refs)
4466 return ERR;
4468 ref = &refs[refs_size++];
4469 ref->name = malloc(namelen + 1);
4470 if (!ref->name)
4471 return ERR;
4473 strncpy(ref->name, name, namelen);
4474 ref->name[namelen] = 0;
4475 ref->tag = tag;
4476 ref->remote = remote;
4477 string_copy_rev(ref->id, id);
4479 return OK;
4480 }
4482 static int
4483 load_refs(void)
4484 {
4485 const char *cmd_env = getenv("TIG_LS_REMOTE");
4486 const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
4488 return read_properties(popen(cmd, "r"), "\t", read_ref);
4489 }
4491 static int
4492 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
4493 {
4494 if (!strcmp(name, "i18n.commitencoding"))
4495 string_ncopy(opt_encoding, value, valuelen);
4497 if (!strcmp(name, "core.editor"))
4498 string_ncopy(opt_editor, value, valuelen);
4500 return OK;
4501 }
4503 static int
4504 load_repo_config(void)
4505 {
4506 return read_properties(popen(GIT_CONFIG " --list", "r"),
4507 "=", read_repo_config_option);
4508 }
4510 static int
4511 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
4512 {
4513 if (!opt_git_dir[0])
4514 string_ncopy(opt_git_dir, name, namelen);
4515 else
4516 string_ncopy(opt_cdup, name, namelen);
4517 return OK;
4518 }
4520 /* XXX: The line outputted by "--show-cdup" can be empty so the option
4521 * must be the last one! */
4522 static int
4523 load_repo_info(void)
4524 {
4525 return read_properties(popen("git rev-parse --git-dir --show-cdup 2>/dev/null", "r"),
4526 "=", read_repo_info);
4527 }
4529 static int
4530 read_properties(FILE *pipe, const char *separators,
4531 int (*read_property)(char *, size_t, char *, size_t))
4532 {
4533 char buffer[BUFSIZ];
4534 char *name;
4535 int state = OK;
4537 if (!pipe)
4538 return ERR;
4540 while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
4541 char *value;
4542 size_t namelen;
4543 size_t valuelen;
4545 name = chomp_string(name);
4546 namelen = strcspn(name, separators);
4548 if (name[namelen]) {
4549 name[namelen] = 0;
4550 value = chomp_string(name + namelen + 1);
4551 valuelen = strlen(value);
4553 } else {
4554 value = "";
4555 valuelen = 0;
4556 }
4558 state = read_property(name, namelen, value, valuelen);
4559 }
4561 if (state != ERR && ferror(pipe))
4562 state = ERR;
4564 pclose(pipe);
4566 return state;
4567 }
4570 /*
4571 * Main
4572 */
4574 static void __NORETURN
4575 quit(int sig)
4576 {
4577 /* XXX: Restore tty modes and let the OS cleanup the rest! */
4578 if (cursed)
4579 endwin();
4580 exit(0);
4581 }
4583 static void __NORETURN
4584 die(const char *err, ...)
4585 {
4586 va_list args;
4588 endwin();
4590 va_start(args, err);
4591 fputs("tig: ", stderr);
4592 vfprintf(stderr, err, args);
4593 fputs("\n", stderr);
4594 va_end(args);
4596 exit(1);
4597 }
4599 int
4600 main(int argc, char *argv[])
4601 {
4602 struct view *view;
4603 enum request request;
4604 size_t i;
4606 signal(SIGINT, quit);
4608 if (setlocale(LC_ALL, "")) {
4609 char *codeset = nl_langinfo(CODESET);
4611 string_ncopy(opt_codeset, codeset, strlen(codeset));
4612 }
4614 if (load_repo_info() == ERR)
4615 die("Failed to load repo info.");
4617 if (load_options() == ERR)
4618 die("Failed to load user config.");
4620 /* Load the repo config file so options can be overwritten from
4621 * the command line. */
4622 if (load_repo_config() == ERR)
4623 die("Failed to load repo config.");
4625 if (!parse_options(argc, argv))
4626 return 0;
4628 /* Require a git repository unless when running in pager mode. */
4629 if (!opt_git_dir[0])
4630 die("Not a git repository");
4632 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
4633 opt_iconv = iconv_open(opt_codeset, opt_encoding);
4634 if (opt_iconv == ICONV_NONE)
4635 die("Failed to initialize character set conversion");
4636 }
4638 if (load_refs() == ERR)
4639 die("Failed to load refs.");
4641 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
4642 view->cmd_env = getenv(view->cmd_env);
4644 request = opt_request;
4646 init_display();
4648 while (view_driver(display[current_view], request)) {
4649 int key;
4650 int i;
4652 foreach_view (view, i)
4653 update_view(view);
4655 /* Refresh, accept single keystroke of input */
4656 key = wgetch(status_win);
4658 /* wgetch() with nodelay() enabled returns ERR when there's no
4659 * input. */
4660 if (key == ERR) {
4661 request = REQ_NONE;
4662 continue;
4663 }
4665 request = get_keybinding(display[current_view]->keymap, key);
4667 /* Some low-level request handling. This keeps access to
4668 * status_win restricted. */
4669 switch (request) {
4670 case REQ_PROMPT:
4671 {
4672 char *cmd = read_prompt(":");
4674 if (cmd && string_format(opt_cmd, "git %s", cmd)) {
4675 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
4676 opt_request = REQ_VIEW_DIFF;
4677 } else {
4678 opt_request = REQ_VIEW_PAGER;
4679 }
4680 break;
4681 }
4683 request = REQ_NONE;
4684 break;
4685 }
4686 case REQ_SEARCH:
4687 case REQ_SEARCH_BACK:
4688 {
4689 const char *prompt = request == REQ_SEARCH
4690 ? "/" : "?";
4691 char *search = read_prompt(prompt);
4693 if (search)
4694 string_ncopy(opt_search, search, strlen(search));
4695 else
4696 request = REQ_NONE;
4697 break;
4698 }
4699 case REQ_SCREEN_RESIZE:
4700 {
4701 int height, width;
4703 getmaxyx(stdscr, height, width);
4705 /* Resize the status view and let the view driver take
4706 * care of resizing the displayed views. */
4707 wresize(status_win, 1, width);
4708 mvwin(status_win, height - 1, 0);
4709 wrefresh(status_win);
4710 break;
4711 }
4712 default:
4713 break;
4714 }
4715 }
4717 quit(0);
4719 return 0;
4720 }