![[tokkee]](http://tokkee.org/images/avatar.png){kind=avatar}
1 /* Copyright (c) 2006 Jonas Fonseca <fonseca@diku.dk>
2 *
3 * This program is free software; you can redistribute it and/or
4 * modify it under the terms of the GNU General Public License as
5 * published by the Free Software Foundation; either version 2 of
6 * the License, or (at your option) any later version.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 */
14 #ifndef TIG_VERSION
15 #define TIG_VERSION "unknown-version"
16 #endif
18 #ifndef DEBUG
19 #define NDEBUG
20 #endif
22 #include <assert.h>
23 #include <errno.h>
24 #include <ctype.h>
25 #include <signal.h>
26 #include <stdarg.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <sys/types.h>
31 #include <sys/stat.h>
32 #include <unistd.h>
33 #include <time.h>
35 #include <regex.h>
37 #include <locale.h>
38 #include <langinfo.h>
39 #include <iconv.h>
41 #include <curses.h>
43 #include "config.h"
45 #if __GNUC__ >= 3
46 #define __NORETURN __attribute__((__noreturn__))
47 #else
48 #define __NORETURN
49 #endif
51 static void __NORETURN die(const char *err, ...);
52 static void report(const char *msg, ...);
53 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, size_t, char *, size_t));
54 static void set_nonblocking_input(bool loading);
55 static size_t utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed);
57 #define ABS(x) ((x) >= 0 ? (x) : -(x))
58 #define MIN(x, y) ((x) < (y) ? (x) : (y))
60 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
61 #define STRING_SIZE(x) (sizeof(x) - 1)
63 #define SIZEOF_STR 1024 /* Default string size. */
64 #define SIZEOF_REF 256 /* Size of symbolic or SHA1 ID. */
65 #define SIZEOF_REV 41 /* Holds a SHA-1 and an ending NUL */
67 /* Revision graph */
69 #define REVGRAPH_INIT 'I'
70 #define REVGRAPH_MERGE 'M'
71 #define REVGRAPH_BRANCH '+'
72 #define REVGRAPH_COMMIT '*'
73 #define REVGRAPH_LINE '|'
75 #define SIZEOF_REVGRAPH 19 /* Size of revision ancestry graphics. */
77 /* This color name can be used to refer to the default term colors. */
78 #define COLOR_DEFAULT (-1)
80 #define ICONV_NONE ((iconv_t) -1)
82 /* The format and size of the date column in the main view. */
83 #define DATE_FORMAT "%Y-%m-%d %H:%M"
84 #define DATE_COLS STRING_SIZE("2006-04-29 14:21 ")
86 #define AUTHOR_COLS 20
88 /* The default interval between line numbers. */
89 #define NUMBER_INTERVAL 1
91 #define TABSIZE 8
93 #define SCALE_SPLIT_VIEW(height) ((height) * 2 / 3)
95 #ifndef GIT_CONFIG
96 #define "git config"
97 #endif
99 #define TIG_LS_REMOTE \
100 "git ls-remote $(git rev-parse --git-dir) 2>/dev/null"
102 #define TIG_DIFF_CMD \
103 "git show --root --patch-with-stat --find-copies-harder -B -C %s 2>/dev/null"
105 #define TIG_LOG_CMD \
106 "git log --cc --stat -n100 %s 2>/dev/null"
108 #define TIG_MAIN_CMD \
109 "git log --topo-order --pretty=raw %s 2>/dev/null"
111 #define TIG_TREE_CMD \
112 "git ls-tree %s %s"
114 #define TIG_BLOB_CMD \
115 "git cat-file blob %s"
117 /* XXX: Needs to be defined to the empty string. */
118 #define TIG_HELP_CMD ""
119 #define TIG_PAGER_CMD ""
120 #define TIG_STATUS_CMD ""
121 #define TIG_STAGE_CMD ""
123 /* Some ascii-shorthands fitted into the ncurses namespace. */
124 #define KEY_TAB '\t'
125 #define KEY_RETURN '\r'
126 #define KEY_ESC 27
129 struct ref {
130 char *name; /* Ref name; tag or head names are shortened. */
131 char id[SIZEOF_REV]; /* Commit SHA1 ID */
132 unsigned int tag:1; /* Is it a tag? */
133 unsigned int remote:1; /* Is it a remote ref? */
134 unsigned int next:1; /* For ref lists: are there more refs? */
135 };
137 static struct ref **get_refs(char *id);
139 struct int_map {
140 const char *name;
141 int namelen;
142 int value;
143 };
145 static int
146 set_from_int_map(struct int_map *map, size_t map_size,
147 int *value, const char *name, int namelen)
148 {
150 int i;
152 for (i = 0; i < map_size; i++)
153 if (namelen == map[i].namelen &&
154 !strncasecmp(name, map[i].name, namelen)) {
155 *value = map[i].value;
156 return OK;
157 }
159 return ERR;
160 }
163 /*
164 * String helpers
165 */
167 static inline void
168 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
169 {
170 if (srclen > dstlen - 1)
171 srclen = dstlen - 1;
173 strncpy(dst, src, srclen);
174 dst[srclen] = 0;
175 }
177 /* Shorthands for safely copying into a fixed buffer. */
179 #define string_copy(dst, src) \
180 string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
182 #define string_ncopy(dst, src, srclen) \
183 string_ncopy_do(dst, sizeof(dst), src, srclen)
185 #define string_copy_rev(dst, src) \
186 string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
188 #define string_add(dst, from, src) \
189 string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
191 static char *
192 chomp_string(char *name)
193 {
194 int namelen;
196 while (isspace(*name))
197 name++;
199 namelen = strlen(name) - 1;
200 while (namelen > 0 && isspace(name[namelen]))
201 name[namelen--] = 0;
203 return name;
204 }
206 static bool
207 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
208 {
209 va_list args;
210 size_t pos = bufpos ? *bufpos : 0;
212 va_start(args, fmt);
213 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
214 va_end(args);
216 if (bufpos)
217 *bufpos = pos;
219 return pos >= bufsize ? FALSE : TRUE;
220 }
222 #define string_format(buf, fmt, args...) \
223 string_nformat(buf, sizeof(buf), NULL, fmt, args)
225 #define string_format_from(buf, from, fmt, args...) \
226 string_nformat(buf, sizeof(buf), from, fmt, args)
228 static int
229 string_enum_compare(const char *str1, const char *str2, int len)
230 {
231 size_t i;
233 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
235 /* Diff-Header == DIFF_HEADER */
236 for (i = 0; i < len; i++) {
237 if (toupper(str1[i]) == toupper(str2[i]))
238 continue;
240 if (string_enum_sep(str1[i]) &&
241 string_enum_sep(str2[i]))
242 continue;
244 return str1[i] - str2[i];
245 }
247 return 0;
248 }
250 /* Shell quoting
251 *
252 * NOTE: The following is a slightly modified copy of the git project's shell
253 * quoting routines found in the quote.c file.
254 *
255 * Help to copy the thing properly quoted for the shell safety. any single
256 * quote is replaced with '\'', any exclamation point is replaced with '\!',
257 * and the whole thing is enclosed in a
258 *
259 * E.g.
260 * original sq_quote result
261 * name ==> name ==> 'name'
262 * a b ==> a b ==> 'a b'
263 * a'b ==> a'\''b ==> 'a'\''b'
264 * a!b ==> a'\!'b ==> 'a'\!'b'
265 */
267 static size_t
268 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
269 {
270 char c;
272 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
274 BUFPUT('\'');
275 while ((c = *src++)) {
276 if (c == '\'' || c == '!') {
277 BUFPUT('\'');
278 BUFPUT('\\');
279 BUFPUT(c);
280 BUFPUT('\'');
281 } else {
282 BUFPUT(c);
283 }
284 }
285 BUFPUT('\'');
287 if (bufsize < SIZEOF_STR)
288 buf[bufsize] = 0;
290 return bufsize;
291 }
294 /*
295 * User requests
296 */
298 #define REQ_INFO \
299 /* XXX: Keep the view request first and in sync with views[]. */ \
300 REQ_GROUP("View switching") \
301 REQ_(VIEW_MAIN, "Show main view"), \
302 REQ_(VIEW_DIFF, "Show diff view"), \
303 REQ_(VIEW_LOG, "Show log view"), \
304 REQ_(VIEW_TREE, "Show tree view"), \
305 REQ_(VIEW_BLOB, "Show blob view"), \
306 REQ_(VIEW_HELP, "Show help page"), \
307 REQ_(VIEW_PAGER, "Show pager view"), \
308 REQ_(VIEW_STATUS, "Show status view"), \
309 REQ_(VIEW_STAGE, "Show stage view"), \
310 \
311 REQ_GROUP("View manipulation") \
312 REQ_(ENTER, "Enter current line and scroll"), \
313 REQ_(NEXT, "Move to next"), \
314 REQ_(PREVIOUS, "Move to previous"), \
315 REQ_(VIEW_NEXT, "Move focus to next view"), \
316 REQ_(VIEW_CLOSE, "Close the current view"), \
317 REQ_(QUIT, "Close all views and quit"), \
318 \
319 REQ_GROUP("Cursor navigation") \
320 REQ_(MOVE_UP, "Move cursor one line up"), \
321 REQ_(MOVE_DOWN, "Move cursor one line down"), \
322 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
323 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
324 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
325 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
326 \
327 REQ_GROUP("Scrolling") \
328 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
329 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
330 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
331 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
332 \
333 REQ_GROUP("Searching") \
334 REQ_(SEARCH, "Search the view"), \
335 REQ_(SEARCH_BACK, "Search backwards in the view"), \
336 REQ_(FIND_NEXT, "Find next search match"), \
337 REQ_(FIND_PREV, "Find previous search match"), \
338 \
339 REQ_GROUP("Misc") \
340 REQ_(NONE, "Do nothing"), \
341 REQ_(PROMPT, "Bring up the prompt"), \
342 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
343 REQ_(SCREEN_RESIZE, "Resize the screen"), \
344 REQ_(SHOW_VERSION, "Show version information"), \
345 REQ_(STOP_LOADING, "Stop all loading views"), \
346 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
347 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization"), \
348 REQ_(STATUS_UPDATE, "Update file status"), \
349 REQ_(EDIT, "Open in editor")
352 /* User action requests. */
353 enum request {
354 #define REQ_GROUP(help)
355 #define REQ_(req, help) REQ_##req
357 /* Offset all requests to avoid conflicts with ncurses getch values. */
358 REQ_OFFSET = KEY_MAX + 1,
359 REQ_INFO,
360 REQ_UNKNOWN,
362 #undef REQ_GROUP
363 #undef REQ_
364 };
366 struct request_info {
367 enum request request;
368 char *name;
369 int namelen;
370 char *help;
371 };
373 static struct request_info req_info[] = {
374 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
375 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
376 REQ_INFO
377 #undef REQ_GROUP
378 #undef REQ_
379 };
381 static enum request
382 get_request(const char *name)
383 {
384 int namelen = strlen(name);
385 int i;
387 for (i = 0; i < ARRAY_SIZE(req_info); i++)
388 if (req_info[i].namelen == namelen &&
389 !string_enum_compare(req_info[i].name, name, namelen))
390 return req_info[i].request;
392 return REQ_UNKNOWN;
393 }
396 /*
397 * Options
398 */
400 static const char usage[] =
401 "tig " TIG_VERSION " (" __DATE__ ")\n"
402 "\n"
403 "Usage: tig [options]\n"
404 " or: tig [options] [--] [git log options]\n"
405 " or: tig [options] log [git log options]\n"
406 " or: tig [options] diff [git diff options]\n"
407 " or: tig [options] show [git show options]\n"
408 " or: tig [options] < [git command output]\n"
409 "\n"
410 "Options:\n"
411 " -l Start up in log view\n"
412 " -d Start up in diff view\n"
413 " -S Start up in status view\n"
414 " -n[I], --line-number[=I] Show line numbers with given interval\n"
415 " -b[N], --tab-size[=N] Set number of spaces for tab expansion\n"
416 " -- Mark end of tig options\n"
417 " -v, --version Show version and exit\n"
418 " -h, --help Show help message and exit\n";
420 /* Option and state variables. */
421 static bool opt_line_number = FALSE;
422 static bool opt_rev_graph = FALSE;
423 static int opt_num_interval = NUMBER_INTERVAL;
424 static int opt_tab_size = TABSIZE;
425 static enum request opt_request = REQ_VIEW_MAIN;
426 static char opt_cmd[SIZEOF_STR] = "";
427 static char opt_path[SIZEOF_STR] = "";
428 static FILE *opt_pipe = NULL;
429 static char opt_encoding[20] = "UTF-8";
430 static bool opt_utf8 = TRUE;
431 static char opt_codeset[20] = "UTF-8";
432 static iconv_t opt_iconv = ICONV_NONE;
433 static char opt_search[SIZEOF_STR] = "";
434 static char opt_cdup[SIZEOF_STR] = "";
435 static char opt_git_dir[SIZEOF_STR] = "";
436 static char opt_editor[SIZEOF_STR] = "";
438 enum option_type {
439 OPT_NONE,
440 OPT_INT,
441 };
443 static bool
444 check_option(char *opt, char short_name, char *name, enum option_type type, ...)
445 {
446 va_list args;
447 char *value = "";
448 int *number;
450 if (opt[0] != '-')
451 return FALSE;
453 if (opt[1] == '-') {
454 int namelen = strlen(name);
456 opt += 2;
458 if (strncmp(opt, name, namelen))
459 return FALSE;
461 if (opt[namelen] == '=')
462 value = opt + namelen + 1;
464 } else {
465 if (!short_name || opt[1] != short_name)
466 return FALSE;
467 value = opt + 2;
468 }
470 va_start(args, type);
471 if (type == OPT_INT) {
472 number = va_arg(args, int *);
473 if (isdigit(*value))
474 *number = atoi(value);
475 }
476 va_end(args);
478 return TRUE;
479 }
481 /* Returns the index of log or diff command or -1 to exit. */
482 static bool
483 parse_options(int argc, char *argv[])
484 {
485 int i;
487 for (i = 1; i < argc; i++) {
488 char *opt = argv[i];
490 if (!strcmp(opt, "log") ||
491 !strcmp(opt, "diff") ||
492 !strcmp(opt, "show")) {
493 opt_request = opt[0] == 'l'
494 ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
495 break;
496 }
498 if (opt[0] && opt[0] != '-')
499 break;
501 if (!strcmp(opt, "-l")) {
502 opt_request = REQ_VIEW_LOG;
503 continue;
504 }
506 if (!strcmp(opt, "-d")) {
507 opt_request = REQ_VIEW_DIFF;
508 continue;
509 }
511 if (!strcmp(opt, "-S")) {
512 opt_request = REQ_VIEW_STATUS;
513 continue;
514 }
516 if (check_option(opt, 'n', "line-number", OPT_INT, &opt_num_interval)) {
517 opt_line_number = TRUE;
518 continue;
519 }
521 if (check_option(opt, 'b', "tab-size", OPT_INT, &opt_tab_size)) {
522 opt_tab_size = MIN(opt_tab_size, TABSIZE);
523 continue;
524 }
526 if (check_option(opt, 'v', "version", OPT_NONE)) {
527 printf("tig version %s\n", TIG_VERSION);
528 return FALSE;
529 }
531 if (check_option(opt, 'h', "help", OPT_NONE)) {
532 printf(usage);
533 return FALSE;
534 }
536 if (!strcmp(opt, "--")) {
537 i++;
538 break;
539 }
541 die("unknown option '%s'\n\n%s", opt, usage);
542 }
544 if (!isatty(STDIN_FILENO)) {
545 opt_request = REQ_VIEW_PAGER;
546 opt_pipe = stdin;
548 } else if (i < argc) {
549 size_t buf_size;
551 if (opt_request == REQ_VIEW_MAIN)
552 /* XXX: This is vulnerable to the user overriding
553 * options required for the main view parser. */
554 string_copy(opt_cmd, "git log --pretty=raw");
555 else
556 string_copy(opt_cmd, "git");
557 buf_size = strlen(opt_cmd);
559 while (buf_size < sizeof(opt_cmd) && i < argc) {
560 opt_cmd[buf_size++] = ' ';
561 buf_size = sq_quote(opt_cmd, buf_size, argv[i++]);
562 }
564 if (buf_size >= sizeof(opt_cmd))
565 die("command too long");
567 opt_cmd[buf_size] = 0;
568 }
570 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
571 opt_utf8 = FALSE;
573 return TRUE;
574 }
577 /*
578 * Line-oriented content detection.
579 */
581 #define LINE_INFO \
582 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
583 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
584 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
585 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
586 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
587 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
588 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
589 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
590 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
591 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
592 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
593 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
594 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
595 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
596 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
597 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
598 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
599 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
600 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
601 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
602 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
603 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
604 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
605 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
606 LINE(AUTHOR, "author ", COLOR_CYAN, COLOR_DEFAULT, 0), \
607 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
608 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
609 LINE(ACKED, " Acked-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
610 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
611 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
612 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
613 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
614 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
615 LINE(MAIN_DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
616 LINE(MAIN_AUTHOR, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
617 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
618 LINE(MAIN_DELIM, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
619 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
620 LINE(MAIN_REMOTE, "", COLOR_YELLOW, COLOR_DEFAULT, A_BOLD), \
621 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
622 LINE(TREE_DIR, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
623 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
624 LINE(STAT_SECTION, "", COLOR_DEFAULT, COLOR_BLUE, A_BOLD), \
625 LINE(STAT_NONE, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
626 LINE(STAT_STAGED, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
627 LINE(STAT_UNSTAGED,"", COLOR_YELLOW, COLOR_DEFAULT, 0), \
628 LINE(STAT_UNTRACKED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0)
630 enum line_type {
631 #define LINE(type, line, fg, bg, attr) \
632 LINE_##type
633 LINE_INFO
634 #undef LINE
635 };
637 struct line_info {
638 const char *name; /* Option name. */
639 int namelen; /* Size of option name. */
640 const char *line; /* The start of line to match. */
641 int linelen; /* Size of string to match. */
642 int fg, bg, attr; /* Color and text attributes for the lines. */
643 };
645 static struct line_info line_info[] = {
646 #define LINE(type, line, fg, bg, attr) \
647 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
648 LINE_INFO
649 #undef LINE
650 };
652 static enum line_type
653 get_line_type(char *line)
654 {
655 int linelen = strlen(line);
656 enum line_type type;
658 for (type = 0; type < ARRAY_SIZE(line_info); type++)
659 /* Case insensitive search matches Signed-off-by lines better. */
660 if (linelen >= line_info[type].linelen &&
661 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
662 return type;
664 return LINE_DEFAULT;
665 }
667 static inline int
668 get_line_attr(enum line_type type)
669 {
670 assert(type < ARRAY_SIZE(line_info));
671 return COLOR_PAIR(type) | line_info[type].attr;
672 }
674 static struct line_info *
675 get_line_info(char *name, int namelen)
676 {
677 enum line_type type;
679 for (type = 0; type < ARRAY_SIZE(line_info); type++)
680 if (namelen == line_info[type].namelen &&
681 !string_enum_compare(line_info[type].name, name, namelen))
682 return &line_info[type];
684 return NULL;
685 }
687 static void
688 init_colors(void)
689 {
690 int default_bg = COLOR_BLACK;
691 int default_fg = COLOR_WHITE;
692 enum line_type type;
694 start_color();
696 if (use_default_colors() != ERR) {
697 default_bg = -1;
698 default_fg = -1;
699 }
701 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
702 struct line_info *info = &line_info[type];
703 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
704 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
706 init_pair(type, fg, bg);
707 }
708 }
710 struct line {
711 enum line_type type;
713 /* State flags */
714 unsigned int selected:1;
716 void *data; /* User data */
717 };
720 /*
721 * Keys
722 */
724 struct keybinding {
725 int alias;
726 enum request request;
727 struct keybinding *next;
728 };
730 static struct keybinding default_keybindings[] = {
731 /* View switching */
732 { 'm', REQ_VIEW_MAIN },
733 { 'd', REQ_VIEW_DIFF },
734 { 'l', REQ_VIEW_LOG },
735 { 't', REQ_VIEW_TREE },
736 { 'f', REQ_VIEW_BLOB },
737 { 'p', REQ_VIEW_PAGER },
738 { 'h', REQ_VIEW_HELP },
739 { 'S', REQ_VIEW_STATUS },
740 { 'c', REQ_VIEW_STAGE },
742 /* View manipulation */
743 { 'q', REQ_VIEW_CLOSE },
744 { KEY_TAB, REQ_VIEW_NEXT },
745 { KEY_RETURN, REQ_ENTER },
746 { KEY_UP, REQ_PREVIOUS },
747 { KEY_DOWN, REQ_NEXT },
749 /* Cursor navigation */
750 { 'k', REQ_MOVE_UP },
751 { 'j', REQ_MOVE_DOWN },
752 { KEY_HOME, REQ_MOVE_FIRST_LINE },
753 { KEY_END, REQ_MOVE_LAST_LINE },
754 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
755 { ' ', REQ_MOVE_PAGE_DOWN },
756 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
757 { 'b', REQ_MOVE_PAGE_UP },
758 { '-', REQ_MOVE_PAGE_UP },
760 /* Scrolling */
761 { KEY_IC, REQ_SCROLL_LINE_UP },
762 { KEY_DC, REQ_SCROLL_LINE_DOWN },
763 { 'w', REQ_SCROLL_PAGE_UP },
764 { 's', REQ_SCROLL_PAGE_DOWN },
766 /* Searching */
767 { '/', REQ_SEARCH },
768 { '?', REQ_SEARCH_BACK },
769 { 'n', REQ_FIND_NEXT },
770 { 'N', REQ_FIND_PREV },
772 /* Misc */
773 { 'Q', REQ_QUIT },
774 { 'z', REQ_STOP_LOADING },
775 { 'v', REQ_SHOW_VERSION },
776 { 'r', REQ_SCREEN_REDRAW },
777 { '.', REQ_TOGGLE_LINENO },
778 { 'g', REQ_TOGGLE_REV_GRAPH },
779 { ':', REQ_PROMPT },
780 { 'u', REQ_STATUS_UPDATE },
781 { 'e', REQ_EDIT },
783 /* Using the ncurses SIGWINCH handler. */
784 { KEY_RESIZE, REQ_SCREEN_RESIZE },
785 };
787 #define KEYMAP_INFO \
788 KEYMAP_(GENERIC), \
789 KEYMAP_(MAIN), \
790 KEYMAP_(DIFF), \
791 KEYMAP_(LOG), \
792 KEYMAP_(TREE), \
793 KEYMAP_(BLOB), \
794 KEYMAP_(PAGER), \
795 KEYMAP_(HELP), \
796 KEYMAP_(STATUS), \
797 KEYMAP_(STAGE)
799 enum keymap {
800 #define KEYMAP_(name) KEYMAP_##name
801 KEYMAP_INFO
802 #undef KEYMAP_
803 };
805 static struct int_map keymap_table[] = {
806 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
807 KEYMAP_INFO
808 #undef KEYMAP_
809 };
811 #define set_keymap(map, name) \
812 set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
814 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
816 static void
817 add_keybinding(enum keymap keymap, enum request request, int key)
818 {
819 struct keybinding *keybinding;
821 keybinding = calloc(1, sizeof(*keybinding));
822 if (!keybinding)
823 die("Failed to allocate keybinding");
825 keybinding->alias = key;
826 keybinding->request = request;
827 keybinding->next = keybindings[keymap];
828 keybindings[keymap] = keybinding;
829 }
831 /* Looks for a key binding first in the given map, then in the generic map, and
832 * lastly in the default keybindings. */
833 static enum request
834 get_keybinding(enum keymap keymap, int key)
835 {
836 struct keybinding *kbd;
837 int i;
839 for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
840 if (kbd->alias == key)
841 return kbd->request;
843 for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
844 if (kbd->alias == key)
845 return kbd->request;
847 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
848 if (default_keybindings[i].alias == key)
849 return default_keybindings[i].request;
851 return (enum request) key;
852 }
855 struct key {
856 char *name;
857 int value;
858 };
860 static struct key key_table[] = {
861 { "Enter", KEY_RETURN },
862 { "Space", ' ' },
863 { "Backspace", KEY_BACKSPACE },
864 { "Tab", KEY_TAB },
865 { "Escape", KEY_ESC },
866 { "Left", KEY_LEFT },
867 { "Right", KEY_RIGHT },
868 { "Up", KEY_UP },
869 { "Down", KEY_DOWN },
870 { "Insert", KEY_IC },
871 { "Delete", KEY_DC },
872 { "Hash", '#' },
873 { "Home", KEY_HOME },
874 { "End", KEY_END },
875 { "PageUp", KEY_PPAGE },
876 { "PageDown", KEY_NPAGE },
877 { "F1", KEY_F(1) },
878 { "F2", KEY_F(2) },
879 { "F3", KEY_F(3) },
880 { "F4", KEY_F(4) },
881 { "F5", KEY_F(5) },
882 { "F6", KEY_F(6) },
883 { "F7", KEY_F(7) },
884 { "F8", KEY_F(8) },
885 { "F9", KEY_F(9) },
886 { "F10", KEY_F(10) },
887 { "F11", KEY_F(11) },
888 { "F12", KEY_F(12) },
889 };
891 static int
892 get_key_value(const char *name)
893 {
894 int i;
896 for (i = 0; i < ARRAY_SIZE(key_table); i++)
897 if (!strcasecmp(key_table[i].name, name))
898 return key_table[i].value;
900 if (strlen(name) == 1 && isprint(*name))
901 return (int) *name;
903 return ERR;
904 }
906 static char *
907 get_key(enum request request)
908 {
909 static char buf[BUFSIZ];
910 static char key_char[] = "'X'";
911 size_t pos = 0;
912 char *sep = "";
913 int i;
915 buf[pos] = 0;
917 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
918 struct keybinding *keybinding = &default_keybindings[i];
919 char *seq = NULL;
920 int key;
922 if (keybinding->request != request)
923 continue;
925 for (key = 0; key < ARRAY_SIZE(key_table); key++)
926 if (key_table[key].value == keybinding->alias)
927 seq = key_table[key].name;
929 if (seq == NULL &&
930 keybinding->alias < 127 &&
931 isprint(keybinding->alias)) {
932 key_char[1] = (char) keybinding->alias;
933 seq = key_char;
934 }
936 if (!seq)
937 seq = "'?'";
939 if (!string_format_from(buf, &pos, "%s%s", sep, seq))
940 return "Too many keybindings!";
941 sep = ", ";
942 }
944 return buf;
945 }
948 /*
949 * User config file handling.
950 */
952 static struct int_map color_map[] = {
953 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
954 COLOR_MAP(DEFAULT),
955 COLOR_MAP(BLACK),
956 COLOR_MAP(BLUE),
957 COLOR_MAP(CYAN),
958 COLOR_MAP(GREEN),
959 COLOR_MAP(MAGENTA),
960 COLOR_MAP(RED),
961 COLOR_MAP(WHITE),
962 COLOR_MAP(YELLOW),
963 };
965 #define set_color(color, name) \
966 set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
968 static struct int_map attr_map[] = {
969 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
970 ATTR_MAP(NORMAL),
971 ATTR_MAP(BLINK),
972 ATTR_MAP(BOLD),
973 ATTR_MAP(DIM),
974 ATTR_MAP(REVERSE),
975 ATTR_MAP(STANDOUT),
976 ATTR_MAP(UNDERLINE),
977 };
979 #define set_attribute(attr, name) \
980 set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
982 static int config_lineno;
983 static bool config_errors;
984 static char *config_msg;
986 /* Wants: object fgcolor bgcolor [attr] */
987 static int
988 option_color_command(int argc, char *argv[])
989 {
990 struct line_info *info;
992 if (argc != 3 && argc != 4) {
993 config_msg = "Wrong number of arguments given to color command";
994 return ERR;
995 }
997 info = get_line_info(argv[0], strlen(argv[0]));
998 if (!info) {
999 config_msg = "Unknown color name";
1000 return ERR;
1001 }
1003 if (set_color(&info->fg, argv[1]) == ERR ||
1004 set_color(&info->bg, argv[2]) == ERR) {
1005 config_msg = "Unknown color";
1006 return ERR;
1007 }
1009 if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
1010 config_msg = "Unknown attribute";
1011 return ERR;
1012 }
1014 return OK;
1015 }
1017 /* Wants: name = value */
1018 static int
1019 option_set_command(int argc, char *argv[])
1020 {
1021 if (argc != 3) {
1022 config_msg = "Wrong number of arguments given to set command";
1023 return ERR;
1024 }
1026 if (strcmp(argv[1], "=")) {
1027 config_msg = "No value assigned";
1028 return ERR;
1029 }
1031 if (!strcmp(argv[0], "show-rev-graph")) {
1032 opt_rev_graph = (!strcmp(argv[2], "1") ||
1033 !strcmp(argv[2], "true") ||
1034 !strcmp(argv[2], "yes"));
1035 return OK;
1036 }
1038 if (!strcmp(argv[0], "line-number-interval")) {
1039 opt_num_interval = atoi(argv[2]);
1040 return OK;
1041 }
1043 if (!strcmp(argv[0], "tab-size")) {
1044 opt_tab_size = atoi(argv[2]);
1045 return OK;
1046 }
1048 if (!strcmp(argv[0], "commit-encoding")) {
1049 char *arg = argv[2];
1050 int delimiter = *arg;
1051 int i;
1053 switch (delimiter) {
1054 case '"':
1055 case '\'':
1056 for (arg++, i = 0; arg[i]; i++)
1057 if (arg[i] == delimiter) {
1058 arg[i] = 0;
1059 break;
1060 }
1061 default:
1062 string_ncopy(opt_encoding, arg, strlen(arg));
1063 return OK;
1064 }
1065 }
1067 config_msg = "Unknown variable name";
1068 return ERR;
1069 }
1071 /* Wants: mode request key */
1072 static int
1073 option_bind_command(int argc, char *argv[])
1074 {
1075 enum request request;
1076 int keymap;
1077 int key;
1079 if (argc != 3) {
1080 config_msg = "Wrong number of arguments given to bind command";
1081 return ERR;
1082 }
1084 if (set_keymap(&keymap, argv[0]) == ERR) {
1085 config_msg = "Unknown key map";
1086 return ERR;
1087 }
1089 key = get_key_value(argv[1]);
1090 if (key == ERR) {
1091 config_msg = "Unknown key";
1092 return ERR;
1093 }
1095 request = get_request(argv[2]);
1096 if (request == REQ_UNKNOWN) {
1097 config_msg = "Unknown request name";
1098 return ERR;
1099 }
1101 add_keybinding(keymap, request, key);
1103 return OK;
1104 }
1106 static int
1107 set_option(char *opt, char *value)
1108 {
1109 char *argv[16];
1110 int valuelen;
1111 int argc = 0;
1113 /* Tokenize */
1114 while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1115 argv[argc++] = value;
1117 value += valuelen;
1118 if (!*value)
1119 break;
1121 *value++ = 0;
1122 while (isspace(*value))
1123 value++;
1124 }
1126 if (!strcmp(opt, "color"))
1127 return option_color_command(argc, argv);
1129 if (!strcmp(opt, "set"))
1130 return option_set_command(argc, argv);
1132 if (!strcmp(opt, "bind"))
1133 return option_bind_command(argc, argv);
1135 config_msg = "Unknown option command";
1136 return ERR;
1137 }
1139 static int
1140 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1141 {
1142 int status = OK;
1144 config_lineno++;
1145 config_msg = "Internal error";
1147 /* Check for comment markers, since read_properties() will
1148 * only ensure opt and value are split at first " \t". */
1149 optlen = strcspn(opt, "#");
1150 if (optlen == 0)
1151 return OK;
1153 if (opt[optlen] != 0) {
1154 config_msg = "No option value";
1155 status = ERR;
1157 } else {
1158 /* Look for comment endings in the value. */
1159 size_t len = strcspn(value, "#");
1161 if (len < valuelen) {
1162 valuelen = len;
1163 value[valuelen] = 0;
1164 }
1166 status = set_option(opt, value);
1167 }
1169 if (status == ERR) {
1170 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1171 config_lineno, (int) optlen, opt, config_msg);
1172 config_errors = TRUE;
1173 }
1175 /* Always keep going if errors are encountered. */
1176 return OK;
1177 }
1179 static int
1180 load_options(void)
1181 {
1182 char *home = getenv("HOME");
1183 char buf[SIZEOF_STR];
1184 FILE *file;
1186 config_lineno = 0;
1187 config_errors = FALSE;
1189 if (!home || !string_format(buf, "%s/.tigrc", home))
1190 return ERR;
1192 /* It's ok that the file doesn't exist. */
1193 file = fopen(buf, "r");
1194 if (!file)
1195 return OK;
1197 if (read_properties(file, " \t", read_option) == ERR ||
1198 config_errors == TRUE)
1199 fprintf(stderr, "Errors while loading %s.\n", buf);
1201 return OK;
1202 }
1205 /*
1206 * The viewer
1207 */
1209 struct view;
1210 struct view_ops;
1212 /* The display array of active views and the index of the current view. */
1213 static struct view *display[2];
1214 static unsigned int current_view;
1216 /* Reading from the prompt? */
1217 static bool input_mode = FALSE;
1219 #define foreach_displayed_view(view, i) \
1220 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1222 #define displayed_views() (display[1] != NULL ? 2 : 1)
1224 /* Current head and commit ID */
1225 static char ref_blob[SIZEOF_REF] = "";
1226 static char ref_commit[SIZEOF_REF] = "HEAD";
1227 static char ref_head[SIZEOF_REF] = "HEAD";
1229 struct view {
1230 const char *name; /* View name */
1231 const char *cmd_fmt; /* Default command line format */
1232 const char *cmd_env; /* Command line set via environment */
1233 const char *id; /* Points to either of ref_{head,commit,blob} */
1235 struct view_ops *ops; /* View operations */
1237 enum keymap keymap; /* What keymap does this view have */
1239 char cmd[SIZEOF_STR]; /* Command buffer */
1240 char ref[SIZEOF_REF]; /* Hovered commit reference */
1241 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
1243 int height, width; /* The width and height of the main window */
1244 WINDOW *win; /* The main window */
1245 WINDOW *title; /* The title window living below the main window */
1247 /* Navigation */
1248 unsigned long offset; /* Offset of the window top */
1249 unsigned long lineno; /* Current line number */
1251 /* Searching */
1252 char grep[SIZEOF_STR]; /* Search string */
1253 regex_t *regex; /* Pre-compiled regex */
1255 /* If non-NULL, points to the view that opened this view. If this view
1256 * is closed tig will switch back to the parent view. */
1257 struct view *parent;
1259 /* Buffering */
1260 unsigned long lines; /* Total number of lines */
1261 struct line *line; /* Line index */
1262 unsigned long line_size;/* Total number of allocated lines */
1263 unsigned int digits; /* Number of digits in the lines member. */
1265 /* Loading */
1266 FILE *pipe;
1267 time_t start_time;
1268 };
1270 struct view_ops {
1271 /* What type of content being displayed. Used in the title bar. */
1272 const char *type;
1273 /* Open and reads in all view content. */
1274 bool (*open)(struct view *view);
1275 /* Read one line; updates view->line. */
1276 bool (*read)(struct view *view, char *data);
1277 /* Draw one line; @lineno must be < view->height. */
1278 bool (*draw)(struct view *view, struct line *line, unsigned int lineno, bool selected);
1279 /* Depending on view handle a special requests. */
1280 enum request (*request)(struct view *view, enum request request, struct line *line);
1281 /* Search for regex in a line. */
1282 bool (*grep)(struct view *view, struct line *line);
1283 /* Select line */
1284 void (*select)(struct view *view, struct line *line);
1285 };
1287 static struct view_ops pager_ops;
1288 static struct view_ops main_ops;
1289 static struct view_ops tree_ops;
1290 static struct view_ops blob_ops;
1291 static struct view_ops help_ops;
1292 static struct view_ops status_ops;
1293 static struct view_ops stage_ops;
1295 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1296 { name, cmd, #env, ref, ops, map}
1298 #define VIEW_(id, name, ops, ref) \
1299 VIEW_STR(name, TIG_##id##_CMD, TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1302 static struct view views[] = {
1303 VIEW_(MAIN, "main", &main_ops, ref_head),
1304 VIEW_(DIFF, "diff", &pager_ops, ref_commit),
1305 VIEW_(LOG, "log", &pager_ops, ref_head),
1306 VIEW_(TREE, "tree", &tree_ops, ref_commit),
1307 VIEW_(BLOB, "blob", &blob_ops, ref_blob),
1308 VIEW_(HELP, "help", &help_ops, ""),
1309 VIEW_(PAGER, "pager", &pager_ops, "stdin"),
1310 VIEW_(STATUS, "status", &status_ops, ""),
1311 VIEW_(STAGE, "stage", &stage_ops, ""),
1312 };
1314 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1316 #define foreach_view(view, i) \
1317 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1319 #define view_is_displayed(view) \
1320 (view == display[0] || view == display[1])
1322 static bool
1323 draw_view_line(struct view *view, unsigned int lineno)
1324 {
1325 struct line *line;
1326 bool selected = (view->offset + lineno == view->lineno);
1327 bool draw_ok;
1329 assert(view_is_displayed(view));
1331 if (view->offset + lineno >= view->lines)
1332 return FALSE;
1334 line = &view->line[view->offset + lineno];
1336 if (selected) {
1337 line->selected = TRUE;
1338 view->ops->select(view, line);
1339 } else if (line->selected) {
1340 line->selected = FALSE;
1341 wmove(view->win, lineno, 0);
1342 wclrtoeol(view->win);
1343 }
1345 scrollok(view->win, FALSE);
1346 draw_ok = view->ops->draw(view, line, lineno, selected);
1347 scrollok(view->win, TRUE);
1349 return draw_ok;
1350 }
1352 static void
1353 redraw_view_from(struct view *view, int lineno)
1354 {
1355 assert(0 <= lineno && lineno < view->height);
1357 for (; lineno < view->height; lineno++) {
1358 if (!draw_view_line(view, lineno))
1359 break;
1360 }
1362 redrawwin(view->win);
1363 if (input_mode)
1364 wnoutrefresh(view->win);
1365 else
1366 wrefresh(view->win);
1367 }
1369 static void
1370 redraw_view(struct view *view)
1371 {
1372 wclear(view->win);
1373 redraw_view_from(view, 0);
1374 }
1377 static void
1378 update_view_title(struct view *view)
1379 {
1380 char buf[SIZEOF_STR];
1381 char state[SIZEOF_STR];
1382 size_t bufpos = 0, statelen = 0;
1384 assert(view_is_displayed(view));
1386 if (view != VIEW(REQ_VIEW_STATUS) && (view->lines || view->pipe)) {
1387 unsigned int view_lines = view->offset + view->height;
1388 unsigned int lines = view->lines
1389 ? MIN(view_lines, view->lines) * 100 / view->lines
1390 : 0;
1392 string_format_from(state, &statelen, "- %s %d of %d (%d%%)",
1393 view->ops->type,
1394 view->lineno + 1,
1395 view->lines,
1396 lines);
1398 if (view->pipe) {
1399 time_t secs = time(NULL) - view->start_time;
1401 /* Three git seconds are a long time ... */
1402 if (secs > 2)
1403 string_format_from(state, &statelen, " %lds", secs);
1404 }
1405 }
1407 string_format_from(buf, &bufpos, "[%s]", view->name);
1408 if (*view->ref && bufpos < view->width) {
1409 size_t refsize = strlen(view->ref);
1410 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
1412 if (minsize < view->width)
1413 refsize = view->width - minsize + 7;
1414 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
1415 }
1417 if (statelen && bufpos < view->width) {
1418 string_format_from(buf, &bufpos, " %s", state);
1419 }
1421 if (view == display[current_view])
1422 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1423 else
1424 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1426 mvwaddnstr(view->title, 0, 0, buf, bufpos);
1427 wclrtoeol(view->title);
1428 wmove(view->title, 0, view->width - 1);
1430 if (input_mode)
1431 wnoutrefresh(view->title);
1432 else
1433 wrefresh(view->title);
1434 }
1436 static void
1437 resize_display(void)
1438 {
1439 int offset, i;
1440 struct view *base = display[0];
1441 struct view *view = display[1] ? display[1] : display[0];
1443 /* Setup window dimensions */
1445 getmaxyx(stdscr, base->height, base->width);
1447 /* Make room for the status window. */
1448 base->height -= 1;
1450 if (view != base) {
1451 /* Horizontal split. */
1452 view->width = base->width;
1453 view->height = SCALE_SPLIT_VIEW(base->height);
1454 base->height -= view->height;
1456 /* Make room for the title bar. */
1457 view->height -= 1;
1458 }
1460 /* Make room for the title bar. */
1461 base->height -= 1;
1463 offset = 0;
1465 foreach_displayed_view (view, i) {
1466 if (!view->win) {
1467 view->win = newwin(view->height, 0, offset, 0);
1468 if (!view->win)
1469 die("Failed to create %s view", view->name);
1471 scrollok(view->win, TRUE);
1473 view->title = newwin(1, 0, offset + view->height, 0);
1474 if (!view->title)
1475 die("Failed to create title window");
1477 } else {
1478 wresize(view->win, view->height, view->width);
1479 mvwin(view->win, offset, 0);
1480 mvwin(view->title, offset + view->height, 0);
1481 }
1483 offset += view->height + 1;
1484 }
1485 }
1487 static void
1488 redraw_display(void)
1489 {
1490 struct view *view;
1491 int i;
1493 foreach_displayed_view (view, i) {
1494 redraw_view(view);
1495 update_view_title(view);
1496 }
1497 }
1499 static void
1500 update_display_cursor(struct view *view)
1501 {
1502 /* Move the cursor to the right-most column of the cursor line.
1503 *
1504 * XXX: This could turn out to be a bit expensive, but it ensures that
1505 * the cursor does not jump around. */
1506 if (view->lines) {
1507 wmove(view->win, view->lineno - view->offset, view->width - 1);
1508 wrefresh(view->win);
1509 }
1510 }
1512 /*
1513 * Navigation
1514 */
1516 /* Scrolling backend */
1517 static void
1518 do_scroll_view(struct view *view, int lines)
1519 {
1520 bool redraw_current_line = FALSE;
1522 /* The rendering expects the new offset. */
1523 view->offset += lines;
1525 assert(0 <= view->offset && view->offset < view->lines);
1526 assert(lines);
1528 /* Move current line into the view. */
1529 if (view->lineno < view->offset) {
1530 view->lineno = view->offset;
1531 redraw_current_line = TRUE;
1532 } else if (view->lineno >= view->offset + view->height) {
1533 view->lineno = view->offset + view->height - 1;
1534 redraw_current_line = TRUE;
1535 }
1537 assert(view->offset <= view->lineno && view->lineno < view->lines);
1539 /* Redraw the whole screen if scrolling is pointless. */
1540 if (view->height < ABS(lines)) {
1541 redraw_view(view);
1543 } else {
1544 int line = lines > 0 ? view->height - lines : 0;
1545 int end = line + ABS(lines);
1547 wscrl(view->win, lines);
1549 for (; line < end; line++) {
1550 if (!draw_view_line(view, line))
1551 break;
1552 }
1554 if (redraw_current_line)
1555 draw_view_line(view, view->lineno - view->offset);
1556 }
1558 redrawwin(view->win);
1559 wrefresh(view->win);
1560 report("");
1561 }
1563 /* Scroll frontend */
1564 static void
1565 scroll_view(struct view *view, enum request request)
1566 {
1567 int lines = 1;
1569 assert(view_is_displayed(view));
1571 switch (request) {
1572 case REQ_SCROLL_PAGE_DOWN:
1573 lines = view->height;
1574 case REQ_SCROLL_LINE_DOWN:
1575 if (view->offset + lines > view->lines)
1576 lines = view->lines - view->offset;
1578 if (lines == 0 || view->offset + view->height >= view->lines) {
1579 report("Cannot scroll beyond the last line");
1580 return;
1581 }
1582 break;
1584 case REQ_SCROLL_PAGE_UP:
1585 lines = view->height;
1586 case REQ_SCROLL_LINE_UP:
1587 if (lines > view->offset)
1588 lines = view->offset;
1590 if (lines == 0) {
1591 report("Cannot scroll beyond the first line");
1592 return;
1593 }
1595 lines = -lines;
1596 break;
1598 default:
1599 die("request %d not handled in switch", request);
1600 }
1602 do_scroll_view(view, lines);
1603 }
1605 /* Cursor moving */
1606 static void
1607 move_view(struct view *view, enum request request)
1608 {
1609 int scroll_steps = 0;
1610 int steps;
1612 switch (request) {
1613 case REQ_MOVE_FIRST_LINE:
1614 steps = -view->lineno;
1615 break;
1617 case REQ_MOVE_LAST_LINE:
1618 steps = view->lines - view->lineno - 1;
1619 break;
1621 case REQ_MOVE_PAGE_UP:
1622 steps = view->height > view->lineno
1623 ? -view->lineno : -view->height;
1624 break;
1626 case REQ_MOVE_PAGE_DOWN:
1627 steps = view->lineno + view->height >= view->lines
1628 ? view->lines - view->lineno - 1 : view->height;
1629 break;
1631 case REQ_MOVE_UP:
1632 steps = -1;
1633 break;
1635 case REQ_MOVE_DOWN:
1636 steps = 1;
1637 break;
1639 default:
1640 die("request %d not handled in switch", request);
1641 }
1643 if (steps <= 0 && view->lineno == 0) {
1644 report("Cannot move beyond the first line");
1645 return;
1647 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1648 report("Cannot move beyond the last line");
1649 return;
1650 }
1652 /* Move the current line */
1653 view->lineno += steps;
1654 assert(0 <= view->lineno && view->lineno < view->lines);
1656 /* Check whether the view needs to be scrolled */
1657 if (view->lineno < view->offset ||
1658 view->lineno >= view->offset + view->height) {
1659 scroll_steps = steps;
1660 if (steps < 0 && -steps > view->offset) {
1661 scroll_steps = -view->offset;
1663 } else if (steps > 0) {
1664 if (view->lineno == view->lines - 1 &&
1665 view->lines > view->height) {
1666 scroll_steps = view->lines - view->offset - 1;
1667 if (scroll_steps >= view->height)
1668 scroll_steps -= view->height - 1;
1669 }
1670 }
1671 }
1673 if (!view_is_displayed(view)) {
1674 view->offset += scroll_steps;
1675 assert(0 <= view->offset && view->offset < view->lines);
1676 view->ops->select(view, &view->line[view->lineno]);
1677 return;
1678 }
1680 /* Repaint the old "current" line if we be scrolling */
1681 if (ABS(steps) < view->height)
1682 draw_view_line(view, view->lineno - steps - view->offset);
1684 if (scroll_steps) {
1685 do_scroll_view(view, scroll_steps);
1686 return;
1687 }
1689 /* Draw the current line */
1690 draw_view_line(view, view->lineno - view->offset);
1692 redrawwin(view->win);
1693 wrefresh(view->win);
1694 report("");
1695 }
1698 /*
1699 * Searching
1700 */
1702 static void search_view(struct view *view, enum request request);
1704 static bool
1705 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1706 {
1707 assert(view_is_displayed(view));
1709 if (!view->ops->grep(view, line))
1710 return FALSE;
1712 if (lineno - view->offset >= view->height) {
1713 view->offset = lineno;
1714 view->lineno = lineno;
1715 redraw_view(view);
1717 } else {
1718 unsigned long old_lineno = view->lineno - view->offset;
1720 view->lineno = lineno;
1721 draw_view_line(view, old_lineno);
1723 draw_view_line(view, view->lineno - view->offset);
1724 redrawwin(view->win);
1725 wrefresh(view->win);
1726 }
1728 report("Line %ld matches '%s'", lineno + 1, view->grep);
1729 return TRUE;
1730 }
1732 static void
1733 find_next(struct view *view, enum request request)
1734 {
1735 unsigned long lineno = view->lineno;
1736 int direction;
1738 if (!*view->grep) {
1739 if (!*opt_search)
1740 report("No previous search");
1741 else
1742 search_view(view, request);
1743 return;
1744 }
1746 switch (request) {
1747 case REQ_SEARCH:
1748 case REQ_FIND_NEXT:
1749 direction = 1;
1750 break;
1752 case REQ_SEARCH_BACK:
1753 case REQ_FIND_PREV:
1754 direction = -1;
1755 break;
1757 default:
1758 return;
1759 }
1761 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1762 lineno += direction;
1764 /* Note, lineno is unsigned long so will wrap around in which case it
1765 * will become bigger than view->lines. */
1766 for (; lineno < view->lines; lineno += direction) {
1767 struct line *line = &view->line[lineno];
1769 if (find_next_line(view, lineno, line))
1770 return;
1771 }
1773 report("No match found for '%s'", view->grep);
1774 }
1776 static void
1777 search_view(struct view *view, enum request request)
1778 {
1779 int regex_err;
1781 if (view->regex) {
1782 regfree(view->regex);
1783 *view->grep = 0;
1784 } else {
1785 view->regex = calloc(1, sizeof(*view->regex));
1786 if (!view->regex)
1787 return;
1788 }
1790 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
1791 if (regex_err != 0) {
1792 char buf[SIZEOF_STR] = "unknown error";
1794 regerror(regex_err, view->regex, buf, sizeof(buf));
1795 report("Search failed: %s", buf);
1796 return;
1797 }
1799 string_copy(view->grep, opt_search);
1801 find_next(view, request);
1802 }
1804 /*
1805 * Incremental updating
1806 */
1808 static void
1809 end_update(struct view *view)
1810 {
1811 if (!view->pipe)
1812 return;
1813 set_nonblocking_input(FALSE);
1814 if (view->pipe == stdin)
1815 fclose(view->pipe);
1816 else
1817 pclose(view->pipe);
1818 view->pipe = NULL;
1819 }
1821 static bool
1822 begin_update(struct view *view)
1823 {
1824 if (view->pipe)
1825 end_update(view);
1827 if (opt_cmd[0]) {
1828 string_copy(view->cmd, opt_cmd);
1829 opt_cmd[0] = 0;
1830 /* When running random commands, initially show the
1831 * command in the title. However, it maybe later be
1832 * overwritten if a commit line is selected. */
1833 if (view == VIEW(REQ_VIEW_PAGER))
1834 string_copy(view->ref, view->cmd);
1835 else
1836 view->ref[0] = 0;
1838 } else if (view == VIEW(REQ_VIEW_TREE)) {
1839 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1840 char path[SIZEOF_STR];
1842 if (strcmp(view->vid, view->id))
1843 opt_path[0] = path[0] = 0;
1844 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
1845 return FALSE;
1847 if (!string_format(view->cmd, format, view->id, path))
1848 return FALSE;
1850 } else {
1851 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1852 const char *id = view->id;
1854 if (!string_format(view->cmd, format, id, id, id, id, id))
1855 return FALSE;
1857 /* Put the current ref_* value to the view title ref
1858 * member. This is needed by the blob view. Most other
1859 * views sets it automatically after loading because the
1860 * first line is a commit line. */
1861 string_copy_rev(view->ref, view->id);
1862 }
1864 /* Special case for the pager view. */
1865 if (opt_pipe) {
1866 view->pipe = opt_pipe;
1867 opt_pipe = NULL;
1868 } else {
1869 view->pipe = popen(view->cmd, "r");
1870 }
1872 if (!view->pipe)
1873 return FALSE;
1875 set_nonblocking_input(TRUE);
1877 view->offset = 0;
1878 view->lines = 0;
1879 view->lineno = 0;
1880 string_copy_rev(view->vid, view->id);
1882 if (view->line) {
1883 int i;
1885 for (i = 0; i < view->lines; i++)
1886 if (view->line[i].data)
1887 free(view->line[i].data);
1889 free(view->line);
1890 view->line = NULL;
1891 }
1893 view->start_time = time(NULL);
1895 return TRUE;
1896 }
1898 static struct line *
1899 realloc_lines(struct view *view, size_t line_size)
1900 {
1901 struct line *tmp = realloc(view->line, sizeof(*view->line) * line_size);
1903 if (!tmp)
1904 return NULL;
1906 view->line = tmp;
1907 view->line_size = line_size;
1908 return view->line;
1909 }
1911 static bool
1912 update_view(struct view *view)
1913 {
1914 char in_buffer[BUFSIZ];
1915 char out_buffer[BUFSIZ * 2];
1916 char *line;
1917 /* The number of lines to read. If too low it will cause too much
1918 * redrawing (and possible flickering), if too high responsiveness
1919 * will suffer. */
1920 unsigned long lines = view->height;
1921 int redraw_from = -1;
1923 if (!view->pipe)
1924 return TRUE;
1926 /* Only redraw if lines are visible. */
1927 if (view->offset + view->height >= view->lines)
1928 redraw_from = view->lines - view->offset;
1930 /* FIXME: This is probably not perfect for backgrounded views. */
1931 if (!realloc_lines(view, view->lines + lines))
1932 goto alloc_error;
1934 while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
1935 size_t linelen = strlen(line);
1937 if (linelen)
1938 line[linelen - 1] = 0;
1940 if (opt_iconv != ICONV_NONE) {
1941 ICONV_INBUF_TYPE inbuf = line;
1942 size_t inlen = linelen;
1944 char *outbuf = out_buffer;
1945 size_t outlen = sizeof(out_buffer);
1947 size_t ret;
1949 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
1950 if (ret != (size_t) -1) {
1951 line = out_buffer;
1952 linelen = strlen(out_buffer);
1953 }
1954 }
1956 if (!view->ops->read(view, line))
1957 goto alloc_error;
1959 if (lines-- == 1)
1960 break;
1961 }
1963 {
1964 int digits;
1966 lines = view->lines;
1967 for (digits = 0; lines; digits++)
1968 lines /= 10;
1970 /* Keep the displayed view in sync with line number scaling. */
1971 if (digits != view->digits) {
1972 view->digits = digits;
1973 redraw_from = 0;
1974 }
1975 }
1977 if (!view_is_displayed(view))
1978 goto check_pipe;
1980 if (view == VIEW(REQ_VIEW_TREE)) {
1981 /* Clear the view and redraw everything since the tree sorting
1982 * might have rearranged things. */
1983 redraw_view(view);
1985 } else if (redraw_from >= 0) {
1986 /* If this is an incremental update, redraw the previous line
1987 * since for commits some members could have changed when
1988 * loading the main view. */
1989 if (redraw_from > 0)
1990 redraw_from--;
1992 /* Since revision graph visualization requires knowledge
1993 * about the parent commit, it causes a further one-off
1994 * needed to be redrawn for incremental updates. */
1995 if (redraw_from > 0 && opt_rev_graph)
1996 redraw_from--;
1998 /* Incrementally draw avoids flickering. */
1999 redraw_view_from(view, redraw_from);
2000 }
2002 /* Update the title _after_ the redraw so that if the redraw picks up a
2003 * commit reference in view->ref it'll be available here. */
2004 update_view_title(view);
2006 check_pipe:
2007 if (ferror(view->pipe)) {
2008 report("Failed to read: %s", strerror(errno));
2009 goto end;
2011 } else if (feof(view->pipe)) {
2012 report("");
2013 goto end;
2014 }
2016 return TRUE;
2018 alloc_error:
2019 report("Allocation failure");
2021 end:
2022 view->ops->read(view, NULL);
2023 end_update(view);
2024 return FALSE;
2025 }
2027 static struct line *
2028 add_line_data(struct view *view, void *data, enum line_type type)
2029 {
2030 struct line *line = &view->line[view->lines++];
2032 memset(line, 0, sizeof(*line));
2033 line->type = type;
2034 line->data = data;
2036 return line;
2037 }
2039 static struct line *
2040 add_line_text(struct view *view, char *data, enum line_type type)
2041 {
2042 if (data)
2043 data = strdup(data);
2045 return data ? add_line_data(view, data, type) : NULL;
2046 }
2049 /*
2050 * View opening
2051 */
2053 enum open_flags {
2054 OPEN_DEFAULT = 0, /* Use default view switching. */
2055 OPEN_SPLIT = 1, /* Split current view. */
2056 OPEN_BACKGROUNDED = 2, /* Backgrounded. */
2057 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
2058 };
2060 static void
2061 open_view(struct view *prev, enum request request, enum open_flags flags)
2062 {
2063 bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2064 bool split = !!(flags & OPEN_SPLIT);
2065 bool reload = !!(flags & OPEN_RELOAD);
2066 struct view *view = VIEW(request);
2067 int nviews = displayed_views();
2068 struct view *base_view = display[0];
2070 if (view == prev && nviews == 1 && !reload) {
2071 report("Already in %s view", view->name);
2072 return;
2073 }
2075 if (view->ops->open) {
2076 if (!view->ops->open(view)) {
2077 report("Failed to load %s view", view->name);
2078 return;
2079 }
2081 } else if ((reload || strcmp(view->vid, view->id)) &&
2082 !begin_update(view)) {
2083 report("Failed to load %s view", view->name);
2084 return;
2085 }
2087 if (split) {
2088 display[1] = view;
2089 if (!backgrounded)
2090 current_view = 1;
2091 } else {
2092 /* Maximize the current view. */
2093 memset(display, 0, sizeof(display));
2094 current_view = 0;
2095 display[current_view] = view;
2096 }
2098 /* Resize the view when switching between split- and full-screen,
2099 * or when switching between two different full-screen views. */
2100 if (nviews != displayed_views() ||
2101 (nviews == 1 && base_view != display[0]))
2102 resize_display();
2104 if (split && prev->lineno - prev->offset >= prev->height) {
2105 /* Take the title line into account. */
2106 int lines = prev->lineno - prev->offset - prev->height + 1;
2108 /* Scroll the view that was split if the current line is
2109 * outside the new limited view. */
2110 do_scroll_view(prev, lines);
2111 }
2113 if (prev && view != prev) {
2114 if (split && !backgrounded) {
2115 /* "Blur" the previous view. */
2116 update_view_title(prev);
2117 }
2119 view->parent = prev;
2120 }
2122 if (view->pipe && view->lines == 0) {
2123 /* Clear the old view and let the incremental updating refill
2124 * the screen. */
2125 wclear(view->win);
2126 report("");
2127 } else {
2128 redraw_view(view);
2129 report("");
2130 }
2132 /* If the view is backgrounded the above calls to report()
2133 * won't redraw the view title. */
2134 if (backgrounded)
2135 update_view_title(view);
2136 }
2138 static void
2139 open_editor(struct view *view, char *file)
2140 {
2141 char cmd[SIZEOF_STR];
2142 char file_sq[SIZEOF_STR];
2143 char *editor;
2145 editor = getenv("GIT_EDITOR");
2146 if (!editor && *opt_editor)
2147 editor = opt_editor;
2148 if (!editor)
2149 editor = getenv("VISUAL");
2150 if (!editor)
2151 editor = getenv("EDITOR");
2152 if (!editor)
2153 editor = "vi";
2155 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2156 string_format(cmd, "%s %s", editor, file_sq)) {
2157 def_prog_mode(); /* save current tty modes */
2158 endwin(); /* restore original tty modes */
2159 system(cmd);
2160 reset_prog_mode();
2161 redraw_display();
2162 }
2163 }
2165 /*
2166 * User request switch noodle
2167 */
2169 static int
2170 view_driver(struct view *view, enum request request)
2171 {
2172 int i;
2174 if (view && view->lines) {
2175 request = view->ops->request(view, request, &view->line[view->lineno]);
2176 if (request == REQ_NONE)
2177 return TRUE;
2178 }
2180 switch (request) {
2181 case REQ_MOVE_UP:
2182 case REQ_MOVE_DOWN:
2183 case REQ_MOVE_PAGE_UP:
2184 case REQ_MOVE_PAGE_DOWN:
2185 case REQ_MOVE_FIRST_LINE:
2186 case REQ_MOVE_LAST_LINE:
2187 move_view(view, request);
2188 break;
2190 case REQ_SCROLL_LINE_DOWN:
2191 case REQ_SCROLL_LINE_UP:
2192 case REQ_SCROLL_PAGE_DOWN:
2193 case REQ_SCROLL_PAGE_UP:
2194 scroll_view(view, request);
2195 break;
2197 case REQ_VIEW_BLOB:
2198 if (!ref_blob[0]) {
2199 report("No file chosen, press %s to open tree view",
2200 get_key(REQ_VIEW_TREE));
2201 break;
2202 }
2203 open_view(view, request, OPEN_DEFAULT);
2204 break;
2206 case REQ_VIEW_PAGER:
2207 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2208 report("No pager content, press %s to run command from prompt",
2209 get_key(REQ_PROMPT));
2210 break;
2211 }
2212 open_view(view, request, OPEN_DEFAULT);
2213 break;
2215 case REQ_VIEW_STAGE:
2216 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2217 report("No stage content, press %s to open the status view and choose file",
2218 get_key(REQ_VIEW_STATUS));
2219 break;
2220 }
2221 open_view(view, request, OPEN_DEFAULT);
2222 break;
2224 case REQ_VIEW_MAIN:
2225 case REQ_VIEW_DIFF:
2226 case REQ_VIEW_LOG:
2227 case REQ_VIEW_TREE:
2228 case REQ_VIEW_HELP:
2229 case REQ_VIEW_STATUS:
2230 open_view(view, request, OPEN_DEFAULT);
2231 break;
2233 case REQ_NEXT:
2234 case REQ_PREVIOUS:
2235 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2237 if ((view == VIEW(REQ_VIEW_DIFF) &&
2238 view->parent == VIEW(REQ_VIEW_MAIN)) ||
2239 (view == VIEW(REQ_VIEW_STAGE) &&
2240 view->parent == VIEW(REQ_VIEW_STATUS)) ||
2241 (view == VIEW(REQ_VIEW_BLOB) &&
2242 view->parent == VIEW(REQ_VIEW_TREE))) {
2243 int line;
2245 view = view->parent;
2246 line = view->lineno;
2247 move_view(view, request);
2248 if (view_is_displayed(view))
2249 update_view_title(view);
2250 if (line != view->lineno)
2251 view->ops->request(view, REQ_ENTER,
2252 &view->line[view->lineno]);
2254 } else {
2255 move_view(view, request);
2256 }
2257 break;
2259 case REQ_VIEW_NEXT:
2260 {
2261 int nviews = displayed_views();
2262 int next_view = (current_view + 1) % nviews;
2264 if (next_view == current_view) {
2265 report("Only one view is displayed");
2266 break;
2267 }
2269 current_view = next_view;
2270 /* Blur out the title of the previous view. */
2271 update_view_title(view);
2272 report("");
2273 break;
2274 }
2275 case REQ_TOGGLE_LINENO:
2276 opt_line_number = !opt_line_number;
2277 redraw_display();
2278 break;
2280 case REQ_TOGGLE_REV_GRAPH:
2281 opt_rev_graph = !opt_rev_graph;
2282 redraw_display();
2283 break;
2285 case REQ_PROMPT:
2286 /* Always reload^Wrerun commands from the prompt. */
2287 open_view(view, opt_request, OPEN_RELOAD);
2288 break;
2290 case REQ_SEARCH:
2291 case REQ_SEARCH_BACK:
2292 search_view(view, request);
2293 break;
2295 case REQ_FIND_NEXT:
2296 case REQ_FIND_PREV:
2297 find_next(view, request);
2298 break;
2300 case REQ_STOP_LOADING:
2301 for (i = 0; i < ARRAY_SIZE(views); i++) {
2302 view = &views[i];
2303 if (view->pipe)
2304 report("Stopped loading the %s view", view->name),
2305 end_update(view);
2306 }
2307 break;
2309 case REQ_SHOW_VERSION:
2310 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2311 return TRUE;
2313 case REQ_SCREEN_RESIZE:
2314 resize_display();
2315 /* Fall-through */
2316 case REQ_SCREEN_REDRAW:
2317 redraw_display();
2318 break;
2320 case REQ_EDIT:
2321 report("Nothing to edit");
2322 break;
2324 case REQ_ENTER:
2325 report("Nothing to enter");
2326 break;
2328 case REQ_NONE:
2329 doupdate();
2330 return TRUE;
2332 case REQ_VIEW_CLOSE:
2333 /* XXX: Mark closed views by letting view->parent point to the
2334 * view itself. Parents to closed view should never be
2335 * followed. */
2336 if (view->parent &&
2337 view->parent->parent != view->parent) {
2338 memset(display, 0, sizeof(display));
2339 current_view = 0;
2340 display[current_view] = view->parent;
2341 view->parent = view;
2342 resize_display();
2343 redraw_display();
2344 break;
2345 }
2346 /* Fall-through */
2347 case REQ_QUIT:
2348 return FALSE;
2350 default:
2351 /* An unknown key will show most commonly used commands. */
2352 report("Unknown key, press 'h' for help");
2353 return TRUE;
2354 }
2356 return TRUE;
2357 }
2360 /*
2361 * Pager backend
2362 */
2364 static bool
2365 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2366 {
2367 char *text = line->data;
2368 enum line_type type = line->type;
2369 int textlen = strlen(text);
2370 int attr;
2372 wmove(view->win, lineno, 0);
2374 if (selected) {
2375 type = LINE_CURSOR;
2376 wchgat(view->win, -1, 0, type, NULL);
2377 }
2379 attr = get_line_attr(type);
2380 wattrset(view->win, attr);
2382 if (opt_line_number || opt_tab_size < TABSIZE) {
2383 static char spaces[] = " ";
2384 int col_offset = 0, col = 0;
2386 if (opt_line_number) {
2387 unsigned long real_lineno = view->offset + lineno + 1;
2389 if (real_lineno == 1 ||
2390 (real_lineno % opt_num_interval) == 0) {
2391 wprintw(view->win, "%.*d", view->digits, real_lineno);
2393 } else {
2394 waddnstr(view->win, spaces,
2395 MIN(view->digits, STRING_SIZE(spaces)));
2396 }
2397 waddstr(view->win, ": ");
2398 col_offset = view->digits + 2;
2399 }
2401 while (text && col_offset + col < view->width) {
2402 int cols_max = view->width - col_offset - col;
2403 char *pos = text;
2404 int cols;
2406 if (*text == '\t') {
2407 text++;
2408 assert(sizeof(spaces) > TABSIZE);
2409 pos = spaces;
2410 cols = opt_tab_size - (col % opt_tab_size);
2412 } else {
2413 text = strchr(text, '\t');
2414 cols = line ? text - pos : strlen(pos);
2415 }
2417 waddnstr(view->win, pos, MIN(cols, cols_max));
2418 col += cols;
2419 }
2421 } else {
2422 int col = 0, pos = 0;
2424 for (; pos < textlen && col < view->width; pos++, col++)
2425 if (text[pos] == '\t')
2426 col += TABSIZE - (col % TABSIZE) - 1;
2428 waddnstr(view->win, text, pos);
2429 }
2431 return TRUE;
2432 }
2434 static bool
2435 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2436 {
2437 char refbuf[SIZEOF_STR];
2438 char *ref = NULL;
2439 FILE *pipe;
2441 if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2442 return TRUE;
2444 pipe = popen(refbuf, "r");
2445 if (!pipe)
2446 return TRUE;
2448 if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2449 ref = chomp_string(ref);
2450 pclose(pipe);
2452 if (!ref || !*ref)
2453 return TRUE;
2455 /* This is the only fatal call, since it can "corrupt" the buffer. */
2456 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2457 return FALSE;
2459 return TRUE;
2460 }
2462 static void
2463 add_pager_refs(struct view *view, struct line *line)
2464 {
2465 char buf[SIZEOF_STR];
2466 char *commit_id = line->data + STRING_SIZE("commit ");
2467 struct ref **refs;
2468 size_t bufpos = 0, refpos = 0;
2469 const char *sep = "Refs: ";
2470 bool is_tag = FALSE;
2472 assert(line->type == LINE_COMMIT);
2474 refs = get_refs(commit_id);
2475 if (!refs) {
2476 if (view == VIEW(REQ_VIEW_DIFF))
2477 goto try_add_describe_ref;
2478 return;
2479 }
2481 do {
2482 struct ref *ref = refs[refpos];
2483 char *fmt = ref->tag ? "%s[%s]" :
2484 ref->remote ? "%s<%s>" : "%s%s";
2486 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2487 return;
2488 sep = ", ";
2489 if (ref->tag)
2490 is_tag = TRUE;
2491 } while (refs[refpos++]->next);
2493 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2494 try_add_describe_ref:
2495 /* Add <tag>-g<commit_id> "fake" reference. */
2496 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2497 return;
2498 }
2500 if (bufpos == 0)
2501 return;
2503 if (!realloc_lines(view, view->line_size + 1))
2504 return;
2506 add_line_text(view, buf, LINE_PP_REFS);
2507 }
2509 static bool
2510 pager_read(struct view *view, char *data)
2511 {
2512 struct line *line;
2514 if (!data)
2515 return TRUE;
2517 line = add_line_text(view, data, get_line_type(data));
2518 if (!line)
2519 return FALSE;
2521 if (line->type == LINE_COMMIT &&
2522 (view == VIEW(REQ_VIEW_DIFF) ||
2523 view == VIEW(REQ_VIEW_LOG)))
2524 add_pager_refs(view, line);
2526 return TRUE;
2527 }
2529 static enum request
2530 pager_request(struct view *view, enum request request, struct line *line)
2531 {
2532 int split = 0;
2534 if (request != REQ_ENTER)
2535 return request;
2537 if (line->type == LINE_COMMIT &&
2538 (view == VIEW(REQ_VIEW_LOG) ||
2539 view == VIEW(REQ_VIEW_PAGER))) {
2540 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2541 split = 1;
2542 }
2544 /* Always scroll the view even if it was split. That way
2545 * you can use Enter to scroll through the log view and
2546 * split open each commit diff. */
2547 scroll_view(view, REQ_SCROLL_LINE_DOWN);
2549 /* FIXME: A minor workaround. Scrolling the view will call report("")
2550 * but if we are scrolling a non-current view this won't properly
2551 * update the view title. */
2552 if (split)
2553 update_view_title(view);
2555 return REQ_NONE;
2556 }
2558 static bool
2559 pager_grep(struct view *view, struct line *line)
2560 {
2561 regmatch_t pmatch;
2562 char *text = line->data;
2564 if (!*text)
2565 return FALSE;
2567 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2568 return FALSE;
2570 return TRUE;
2571 }
2573 static void
2574 pager_select(struct view *view, struct line *line)
2575 {
2576 if (line->type == LINE_COMMIT) {
2577 char *text = line->data + STRING_SIZE("commit ");
2579 if (view != VIEW(REQ_VIEW_PAGER))
2580 string_copy_rev(view->ref, text);
2581 string_copy_rev(ref_commit, text);
2582 }
2583 }
2585 static struct view_ops pager_ops = {
2586 "line",
2587 NULL,
2588 pager_read,
2589 pager_draw,
2590 pager_request,
2591 pager_grep,
2592 pager_select,
2593 };
2596 /*
2597 * Help backend
2598 */
2600 static bool
2601 help_open(struct view *view)
2602 {
2603 char buf[BUFSIZ];
2604 int lines = ARRAY_SIZE(req_info) + 2;
2605 int i;
2607 if (view->lines > 0)
2608 return TRUE;
2610 for (i = 0; i < ARRAY_SIZE(req_info); i++)
2611 if (!req_info[i].request)
2612 lines++;
2614 view->line = calloc(lines, sizeof(*view->line));
2615 if (!view->line)
2616 return FALSE;
2618 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
2620 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
2621 char *key;
2623 if (!req_info[i].request) {
2624 add_line_text(view, "", LINE_DEFAULT);
2625 add_line_text(view, req_info[i].help, LINE_DEFAULT);
2626 continue;
2627 }
2629 key = get_key(req_info[i].request);
2630 if (!string_format(buf, " %-25s %s", key, req_info[i].help))
2631 continue;
2633 add_line_text(view, buf, LINE_DEFAULT);
2634 }
2636 return TRUE;
2637 }
2639 static struct view_ops help_ops = {
2640 "line",
2641 help_open,
2642 NULL,
2643 pager_draw,
2644 pager_request,
2645 pager_grep,
2646 pager_select,
2647 };
2650 /*
2651 * Tree backend
2652 */
2654 struct tree_stack_entry {
2655 struct tree_stack_entry *prev; /* Entry below this in the stack */
2656 unsigned long lineno; /* Line number to restore */
2657 char *name; /* Position of name in opt_path */
2658 };
2660 /* The top of the path stack. */
2661 static struct tree_stack_entry *tree_stack = NULL;
2662 unsigned long tree_lineno = 0;
2664 static void
2665 pop_tree_stack_entry(void)
2666 {
2667 struct tree_stack_entry *entry = tree_stack;
2669 tree_lineno = entry->lineno;
2670 entry->name[0] = 0;
2671 tree_stack = entry->prev;
2672 free(entry);
2673 }
2675 static void
2676 push_tree_stack_entry(char *name, unsigned long lineno)
2677 {
2678 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
2679 size_t pathlen = strlen(opt_path);
2681 if (!entry)
2682 return;
2684 entry->prev = tree_stack;
2685 entry->name = opt_path + pathlen;
2686 tree_stack = entry;
2688 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
2689 pop_tree_stack_entry();
2690 return;
2691 }
2693 /* Move the current line to the first tree entry. */
2694 tree_lineno = 1;
2695 entry->lineno = lineno;
2696 }
2698 /* Parse output from git-ls-tree(1):
2699 *
2700 * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
2701 * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
2702 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
2703 * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
2704 */
2706 #define SIZEOF_TREE_ATTR \
2707 STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
2709 #define TREE_UP_FORMAT "040000 tree %s\t.."
2711 static int
2712 tree_compare_entry(enum line_type type1, char *name1,
2713 enum line_type type2, char *name2)
2714 {
2715 if (type1 != type2) {
2716 if (type1 == LINE_TREE_DIR)
2717 return -1;
2718 return 1;
2719 }
2721 return strcmp(name1, name2);
2722 }
2724 static bool
2725 tree_read(struct view *view, char *text)
2726 {
2727 size_t textlen = text ? strlen(text) : 0;
2728 char buf[SIZEOF_STR];
2729 unsigned long pos;
2730 enum line_type type;
2731 bool first_read = view->lines == 0;
2733 if (textlen <= SIZEOF_TREE_ATTR)
2734 return FALSE;
2736 type = text[STRING_SIZE("100644 ")] == 't'
2737 ? LINE_TREE_DIR : LINE_TREE_FILE;
2739 if (first_read) {
2740 /* Add path info line */
2741 if (!string_format(buf, "Directory path /%s", opt_path) ||
2742 !realloc_lines(view, view->line_size + 1) ||
2743 !add_line_text(view, buf, LINE_DEFAULT))
2744 return FALSE;
2746 /* Insert "link" to parent directory. */
2747 if (*opt_path) {
2748 if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
2749 !realloc_lines(view, view->line_size + 1) ||
2750 !add_line_text(view, buf, LINE_TREE_DIR))
2751 return FALSE;
2752 }
2753 }
2755 /* Strip the path part ... */
2756 if (*opt_path) {
2757 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
2758 size_t striplen = strlen(opt_path);
2759 char *path = text + SIZEOF_TREE_ATTR;
2761 if (pathlen > striplen)
2762 memmove(path, path + striplen,
2763 pathlen - striplen + 1);
2764 }
2766 /* Skip "Directory ..." and ".." line. */
2767 for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
2768 struct line *line = &view->line[pos];
2769 char *path1 = ((char *) line->data) + SIZEOF_TREE_ATTR;
2770 char *path2 = text + SIZEOF_TREE_ATTR;
2771 int cmp = tree_compare_entry(line->type, path1, type, path2);
2773 if (cmp <= 0)
2774 continue;
2776 text = strdup(text);
2777 if (!text)
2778 return FALSE;
2780 if (view->lines > pos)
2781 memmove(&view->line[pos + 1], &view->line[pos],
2782 (view->lines - pos) * sizeof(*line));
2784 line = &view->line[pos];
2785 line->data = text;
2786 line->type = type;
2787 view->lines++;
2788 return TRUE;
2789 }
2791 if (!add_line_text(view, text, type))
2792 return FALSE;
2794 if (tree_lineno > view->lineno) {
2795 view->lineno = tree_lineno;
2796 tree_lineno = 0;
2797 }
2799 return TRUE;
2800 }
2802 static enum request
2803 tree_request(struct view *view, enum request request, struct line *line)
2804 {
2805 enum open_flags flags;
2807 if (request != REQ_ENTER)
2808 return request;
2810 /* Cleanup the stack if the tree view is at a different tree. */
2811 while (!*opt_path && tree_stack)
2812 pop_tree_stack_entry();
2814 switch (line->type) {
2815 case LINE_TREE_DIR:
2816 /* Depending on whether it is a subdir or parent (updir?) link
2817 * mangle the path buffer. */
2818 if (line == &view->line[1] && *opt_path) {
2819 pop_tree_stack_entry();
2821 } else {
2822 char *data = line->data;
2823 char *basename = data + SIZEOF_TREE_ATTR;
2825 push_tree_stack_entry(basename, view->lineno);
2826 }
2828 /* Trees and subtrees share the same ID, so they are not not
2829 * unique like blobs. */
2830 flags = OPEN_RELOAD;
2831 request = REQ_VIEW_TREE;
2832 break;
2834 case LINE_TREE_FILE:
2835 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2836 request = REQ_VIEW_BLOB;
2837 break;
2839 default:
2840 return TRUE;
2841 }
2843 open_view(view, request, flags);
2844 if (request == REQ_VIEW_TREE) {
2845 view->lineno = tree_lineno;
2846 }
2848 return REQ_NONE;
2849 }
2851 static void
2852 tree_select(struct view *view, struct line *line)
2853 {
2854 char *text = line->data + STRING_SIZE("100644 blob ");
2856 if (line->type == LINE_TREE_FILE) {
2857 string_copy_rev(ref_blob, text);
2859 } else if (line->type != LINE_TREE_DIR) {
2860 return;
2861 }
2863 string_copy_rev(view->ref, text);
2864 }
2866 static struct view_ops tree_ops = {
2867 "file",
2868 NULL,
2869 tree_read,
2870 pager_draw,
2871 tree_request,
2872 pager_grep,
2873 tree_select,
2874 };
2876 static bool
2877 blob_read(struct view *view, char *line)
2878 {
2879 return add_line_text(view, line, LINE_DEFAULT);
2880 }
2882 static struct view_ops blob_ops = {
2883 "line",
2884 NULL,
2885 blob_read,
2886 pager_draw,
2887 pager_request,
2888 pager_grep,
2889 pager_select,
2890 };
2893 /*
2894 * Status backend
2895 */
2897 struct status {
2898 char status;
2899 struct {
2900 mode_t mode;
2901 char rev[SIZEOF_REV];
2902 } old;
2903 struct {
2904 mode_t mode;
2905 char rev[SIZEOF_REV];
2906 } new;
2907 char name[SIZEOF_STR];
2908 };
2910 static struct status stage_status;
2911 static enum line_type stage_line_type;
2913 /* Get fields from the diff line:
2914 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
2915 */
2916 static inline bool
2917 status_get_diff(struct status *file, char *buf, size_t bufsize)
2918 {
2919 char *old_mode = buf + 1;
2920 char *new_mode = buf + 8;
2921 char *old_rev = buf + 15;
2922 char *new_rev = buf + 56;
2923 char *status = buf + 97;
2925 if (bufsize != 99 ||
2926 old_mode[-1] != ':' ||
2927 new_mode[-1] != ' ' ||
2928 old_rev[-1] != ' ' ||
2929 new_rev[-1] != ' ' ||
2930 status[-1] != ' ')
2931 return FALSE;
2933 file->status = *status;
2935 string_copy_rev(file->old.rev, old_rev);
2936 string_copy_rev(file->new.rev, new_rev);
2938 file->old.mode = strtoul(old_mode, NULL, 8);
2939 file->new.mode = strtoul(new_mode, NULL, 8);
2941 file->name[0] = 0;
2943 return TRUE;
2944 }
2946 static bool
2947 status_run(struct view *view, const char cmd[], bool diff, enum line_type type)
2948 {
2949 struct status *file = NULL;
2950 char buf[SIZEOF_STR * 4];
2951 size_t bufsize = 0;
2952 FILE *pipe;
2954 pipe = popen(cmd, "r");
2955 if (!pipe)
2956 return FALSE;
2958 add_line_data(view, NULL, type);
2960 while (!feof(pipe) && !ferror(pipe)) {
2961 char *sep;
2962 size_t readsize;
2964 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
2965 if (!readsize)
2966 break;
2967 bufsize += readsize;
2969 /* Process while we have NUL chars. */
2970 while ((sep = memchr(buf, 0, bufsize))) {
2971 size_t sepsize = sep - buf + 1;
2973 if (!file) {
2974 if (!realloc_lines(view, view->line_size + 1))
2975 goto error_out;
2977 file = calloc(1, sizeof(*file));
2978 if (!file)
2979 goto error_out;
2981 add_line_data(view, file, type);
2982 }
2984 /* Parse diff info part. */
2985 if (!diff) {
2986 file->status = '?';
2988 } else if (!file->status) {
2989 if (!status_get_diff(file, buf, sepsize))
2990 goto error_out;
2992 bufsize -= sepsize;
2993 memmove(buf, sep + 1, bufsize);
2995 sep = memchr(buf, 0, bufsize);
2996 if (!sep)
2997 break;
2998 sepsize = sep - buf + 1;
2999 }
3001 /* git-ls-files just delivers a NUL separated
3002 * list of file names similar to the second half
3003 * of the git-diff-* output. */
3004 string_ncopy(file->name, buf, sepsize);
3005 bufsize -= sepsize;
3006 memmove(buf, sep + 1, bufsize);
3007 file = NULL;
3008 }
3009 }
3011 if (ferror(pipe)) {
3012 error_out:
3013 pclose(pipe);
3014 return FALSE;
3015 }
3017 if (!view->line[view->lines - 1].data)
3018 add_line_data(view, NULL, LINE_STAT_NONE);
3020 pclose(pipe);
3021 return TRUE;
3022 }
3024 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --cached HEAD"
3025 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
3026 #define STATUS_LIST_OTHER_CMD \
3027 "git ls-files -z --others --exclude-per-directory=.gitignore"
3029 #define STATUS_DIFF_SHOW_CMD \
3030 "git diff --root --patch-with-stat --find-copies-harder -B -C %s -- %s 2>/dev/null"
3032 /* First parse staged info using git-diff-index(1), then parse unstaged
3033 * info using git-diff-files(1), and finally untracked files using
3034 * git-ls-files(1). */
3035 static bool
3036 status_open(struct view *view)
3037 {
3038 struct stat statbuf;
3039 char exclude[SIZEOF_STR];
3040 char cmd[SIZEOF_STR];
3041 size_t i;
3043 for (i = 0; i < view->lines; i++)
3044 free(view->line[i].data);
3045 free(view->line);
3046 view->lines = view->line_size = 0;
3047 view->line = NULL;
3049 if (!realloc_lines(view, view->line_size + 6))
3050 return FALSE;
3052 if (!string_format(exclude, "%s/info/exclude", opt_git_dir))
3053 return FALSE;
3055 string_copy(cmd, STATUS_LIST_OTHER_CMD);
3057 if (stat(exclude, &statbuf) >= 0) {
3058 size_t cmdsize = strlen(cmd);
3060 if (!string_format_from(cmd, &cmdsize, " %s", "--exclude-from=") ||
3061 sq_quote(cmd, cmdsize, exclude) >= sizeof(cmd))
3062 return FALSE;
3063 }
3065 if (!status_run(view, STATUS_DIFF_INDEX_CMD, TRUE, LINE_STAT_STAGED) ||
3066 !status_run(view, STATUS_DIFF_FILES_CMD, TRUE, LINE_STAT_UNSTAGED) ||
3067 !status_run(view, cmd, FALSE, LINE_STAT_UNTRACKED))
3068 return FALSE;
3070 return TRUE;
3071 }
3073 static bool
3074 status_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3075 {
3076 struct status *status = line->data;
3078 wmove(view->win, lineno, 0);
3080 if (selected) {
3081 wattrset(view->win, get_line_attr(LINE_CURSOR));
3082 wchgat(view->win, -1, 0, LINE_CURSOR, NULL);
3084 } else if (!status && line->type != LINE_STAT_NONE) {
3085 wattrset(view->win, get_line_attr(LINE_STAT_SECTION));
3086 wchgat(view->win, -1, 0, LINE_STAT_SECTION, NULL);
3088 } else {
3089 wattrset(view->win, get_line_attr(line->type));
3090 }
3092 if (!status) {
3093 char *text;
3095 switch (line->type) {
3096 case LINE_STAT_STAGED:
3097 text = "Changes to be committed:";
3098 break;
3100 case LINE_STAT_UNSTAGED:
3101 text = "Changed but not updated:";
3102 break;
3104 case LINE_STAT_UNTRACKED:
3105 text = "Untracked files:";
3106 break;
3108 case LINE_STAT_NONE:
3109 text = " (no files)";
3110 break;
3112 default:
3113 return FALSE;
3114 }
3116 waddstr(view->win, text);
3117 return TRUE;
3118 }
3120 waddch(view->win, status->status);
3121 if (!selected)
3122 wattrset(view->win, A_NORMAL);
3123 wmove(view->win, lineno, 4);
3124 waddstr(view->win, status->name);
3126 return TRUE;
3127 }
3129 static enum request
3130 status_enter(struct view *view, struct line *line)
3131 {
3132 struct status *status = line->data;
3133 char path[SIZEOF_STR] = "";
3134 char *info;
3135 size_t cmdsize = 0;
3137 if (line->type == LINE_STAT_NONE ||
3138 (!status && line[1].type == LINE_STAT_NONE)) {
3139 report("No file to diff");
3140 return REQ_NONE;
3141 }
3143 if (status && sq_quote(path, 0, status->name) >= sizeof(path))
3144 return REQ_QUIT;
3146 if (opt_cdup[0] &&
3147 line->type != LINE_STAT_UNTRACKED &&
3148 !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
3149 return REQ_QUIT;
3151 switch (line->type) {
3152 case LINE_STAT_STAGED:
3153 if (!string_format_from(opt_cmd, &cmdsize, STATUS_DIFF_SHOW_CMD,
3154 "--cached", path))
3155 return REQ_QUIT;
3156 if (status)
3157 info = "Staged changes to %s";
3158 else
3159 info = "Staged changes";
3160 break;
3162 case LINE_STAT_UNSTAGED:
3163 if (!string_format_from(opt_cmd, &cmdsize, STATUS_DIFF_SHOW_CMD,
3164 "", path))
3165 return REQ_QUIT;
3166 if (status)
3167 info = "Unstaged changes to %s";
3168 else
3169 info = "Unstaged changes";
3170 break;
3172 case LINE_STAT_UNTRACKED:
3173 if (opt_pipe)
3174 return REQ_QUIT;
3177 if (!status) {
3178 report("No file to show");
3179 return REQ_NONE;
3180 }
3182 opt_pipe = fopen(status->name, "r");
3183 info = "Untracked file %s";
3184 break;
3186 default:
3187 die("w00t");
3188 }
3190 open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | OPEN_SPLIT);
3191 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
3192 if (status) {
3193 stage_status = *status;
3194 } else {
3195 memset(&stage_status, 0, sizeof(stage_status));
3196 }
3198 stage_line_type = line->type;
3199 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.name);
3200 }
3202 return REQ_NONE;
3203 }
3206 static bool
3207 status_update_file(struct view *view, struct status *status, enum line_type type)
3208 {
3209 char cmd[SIZEOF_STR];
3210 char buf[SIZEOF_STR];
3211 size_t cmdsize = 0;
3212 size_t bufsize = 0;
3213 size_t written = 0;
3214 FILE *pipe;
3216 if (opt_cdup[0] &&
3217 type != LINE_STAT_UNTRACKED &&
3218 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
3219 return FALSE;
3221 switch (type) {
3222 case LINE_STAT_STAGED:
3223 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
3224 status->old.mode,
3225 status->old.rev,
3226 status->name, 0))
3227 return FALSE;
3229 string_add(cmd, cmdsize, "git update-index -z --index-info");
3230 break;
3232 case LINE_STAT_UNSTAGED:
3233 case LINE_STAT_UNTRACKED:
3234 if (!string_format_from(buf, &bufsize, "%s%c", status->name, 0))
3235 return FALSE;
3237 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
3238 break;
3240 default:
3241 die("w00t");
3242 }
3244 pipe = popen(cmd, "w");
3245 if (!pipe)
3246 return FALSE;
3248 while (!ferror(pipe) && written < bufsize) {
3249 written += fwrite(buf + written, 1, bufsize - written, pipe);
3250 }
3252 pclose(pipe);
3254 if (written != bufsize)
3255 return FALSE;
3257 return TRUE;
3258 }
3260 static void
3261 status_update(struct view *view)
3262 {
3263 struct line *line = &view->line[view->lineno];
3265 assert(view->lines);
3267 if (!line->data) {
3268 while (++line < view->line + view->lines && line->data) {
3269 if (!status_update_file(view, line->data, line->type))
3270 report("Failed to update file status");
3271 }
3273 if (!line[-1].data) {
3274 report("Nothing to update");
3275 return;
3276 }
3278 } else if (!status_update_file(view, line->data, line->type)) {
3279 report("Failed to update file status");
3280 }
3282 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
3283 }
3285 static enum request
3286 status_request(struct view *view, enum request request, struct line *line)
3287 {
3288 struct status *status = line->data;
3290 switch (request) {
3291 case REQ_STATUS_UPDATE:
3292 status_update(view);
3293 break;
3295 case REQ_EDIT:
3296 if (!status)
3297 return request;
3299 open_editor(view, status->name);
3300 break;
3302 case REQ_ENTER:
3303 status_enter(view, line);
3304 break;
3306 default:
3307 return request;
3308 }
3310 return REQ_NONE;
3311 }
3313 static void
3314 status_select(struct view *view, struct line *line)
3315 {
3316 struct status *status = line->data;
3317 char file[SIZEOF_STR] = "all files";
3318 char *text;
3320 if (status && !string_format(file, "'%s'", status->name))
3321 return;
3323 if (!status && line[1].type == LINE_STAT_NONE)
3324 line++;
3326 switch (line->type) {
3327 case LINE_STAT_STAGED:
3328 text = "Press %s to unstage %s for commit";
3329 break;
3331 case LINE_STAT_UNSTAGED:
3332 text = "Press %s to stage %s for commit";
3333 break;
3335 case LINE_STAT_UNTRACKED:
3336 text = "Press %s to stage %s for addition";
3337 break;
3339 case LINE_STAT_NONE:
3340 text = "Nothing to update";
3341 break;
3343 default:
3344 die("w00t");
3345 }
3347 string_format(view->ref, text, get_key(REQ_STATUS_UPDATE), file);
3348 }
3350 static bool
3351 status_grep(struct view *view, struct line *line)
3352 {
3353 struct status *status = line->data;
3354 enum { S_STATUS, S_NAME, S_END } state;
3355 char buf[2] = "?";
3356 regmatch_t pmatch;
3358 if (!status)
3359 return FALSE;
3361 for (state = S_STATUS; state < S_END; state++) {
3362 char *text;
3364 switch (state) {
3365 case S_NAME: text = status->name; break;
3366 case S_STATUS:
3367 buf[0] = status->status;
3368 text = buf;
3369 break;
3371 default:
3372 return FALSE;
3373 }
3375 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3376 return TRUE;
3377 }
3379 return FALSE;
3380 }
3382 static struct view_ops status_ops = {
3383 "file",
3384 status_open,
3385 NULL,
3386 status_draw,
3387 status_request,
3388 status_grep,
3389 status_select,
3390 };
3393 static bool
3394 stage_diff_line(FILE *pipe, struct line *line)
3395 {
3396 char *buf = line->data;
3397 size_t bufsize = strlen(buf);
3398 size_t written = 0;
3400 while (!ferror(pipe) && written < bufsize) {
3401 written += fwrite(buf + written, 1, bufsize - written, pipe);
3402 }
3404 fputc('\n', pipe);
3406 return written == bufsize;
3407 }
3409 static struct line *
3410 stage_diff_hdr(struct view *view, struct line *line)
3411 {
3412 int diff_hdr_dir = line->type == LINE_DIFF_CHUNK ? -1 : 1;
3413 struct line *diff_hdr;
3415 if (line->type == LINE_DIFF_CHUNK)
3416 diff_hdr = line - 1;
3417 else
3418 diff_hdr = view->line + 1;
3420 while (diff_hdr > view->line && diff_hdr < view->line + view->lines) {
3421 if (diff_hdr->type == LINE_DIFF_HEADER)
3422 return diff_hdr;
3424 diff_hdr += diff_hdr_dir;
3425 }
3427 return NULL;
3428 }
3430 static bool
3431 stage_update_chunk(struct view *view, struct line *line)
3432 {
3433 char cmd[SIZEOF_STR];
3434 size_t cmdsize = 0;
3435 struct line *diff_hdr, *diff_chunk, *diff_end;
3436 FILE *pipe;
3438 diff_hdr = stage_diff_hdr(view, line);
3439 if (!diff_hdr)
3440 return FALSE;
3442 if (opt_cdup[0] &&
3443 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
3444 return FALSE;
3446 if (!string_format_from(cmd, &cmdsize,
3447 "git apply --cached %s - && "
3448 "git update-index -q --unmerged --refresh 2>/dev/null",
3449 stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
3450 return FALSE;
3452 pipe = popen(cmd, "w");
3453 if (!pipe)
3454 return FALSE;
3456 diff_end = view->line + view->lines;
3457 if (line->type != LINE_DIFF_CHUNK) {
3458 diff_chunk = diff_hdr;
3460 } else {
3461 for (diff_chunk = line + 1; diff_chunk < diff_end; diff_chunk++)
3462 if (diff_chunk->type == LINE_DIFF_CHUNK ||
3463 diff_chunk->type == LINE_DIFF_HEADER)
3464 diff_end = diff_chunk;
3466 diff_chunk = line;
3468 while (diff_hdr->type != LINE_DIFF_CHUNK) {
3469 switch (diff_hdr->type) {
3470 case LINE_DIFF_HEADER:
3471 case LINE_DIFF_INDEX:
3472 case LINE_DIFF_ADD:
3473 case LINE_DIFF_DEL:
3474 break;
3476 default:
3477 diff_hdr++;
3478 continue;
3479 }
3481 if (!stage_diff_line(pipe, diff_hdr++)) {
3482 pclose(pipe);
3483 return FALSE;
3484 }
3485 }
3486 }
3488 while (diff_chunk < diff_end && stage_diff_line(pipe, diff_chunk))
3489 diff_chunk++;
3491 pclose(pipe);
3493 if (diff_chunk != diff_end)
3494 return FALSE;
3496 return TRUE;
3497 }
3499 static void
3500 stage_update(struct view *view, struct line *line)
3501 {
3502 if (stage_line_type != LINE_STAT_UNTRACKED &&
3503 (line->type == LINE_DIFF_CHUNK || !stage_status.status)) {
3504 if (!stage_update_chunk(view, line)) {
3505 report("Failed to apply chunk");
3506 return;
3507 }
3509 } else if (!status_update_file(view, &stage_status, stage_line_type)) {
3510 report("Failed to update file");
3511 return;
3512 }
3514 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
3516 view = VIEW(REQ_VIEW_STATUS);
3517 if (view_is_displayed(view))
3518 status_enter(view, &view->line[view->lineno]);
3519 }
3521 static enum request
3522 stage_request(struct view *view, enum request request, struct line *line)
3523 {
3524 switch (request) {
3525 case REQ_STATUS_UPDATE:
3526 stage_update(view, line);
3527 break;
3529 case REQ_EDIT:
3530 if (!stage_status.name[0])
3531 return request;
3533 open_editor(view, stage_status.name);
3534 break;
3536 case REQ_ENTER:
3537 pager_request(view, request, line);
3538 break;
3540 default:
3541 return request;
3542 }
3544 return REQ_NONE;
3545 }
3547 static struct view_ops stage_ops = {
3548 "line",
3549 NULL,
3550 pager_read,
3551 pager_draw,
3552 stage_request,
3553 pager_grep,
3554 pager_select,
3555 };
3558 /*
3559 * Revision graph
3560 */
3562 struct commit {
3563 char id[SIZEOF_REV]; /* SHA1 ID. */
3564 char title[128]; /* First line of the commit message. */
3565 char author[75]; /* Author of the commit. */
3566 struct tm time; /* Date from the author ident. */
3567 struct ref **refs; /* Repository references. */
3568 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
3569 size_t graph_size; /* The width of the graph array. */
3570 };
3572 /* Size of rev graph with no "padding" columns */
3573 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
3575 struct rev_graph {
3576 struct rev_graph *prev, *next, *parents;
3577 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
3578 size_t size;
3579 struct commit *commit;
3580 size_t pos;
3581 };
3583 /* Parents of the commit being visualized. */
3584 static struct rev_graph graph_parents[4];
3586 /* The current stack of revisions on the graph. */
3587 static struct rev_graph graph_stacks[4] = {
3588 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
3589 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
3590 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
3591 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
3592 };
3594 static inline bool
3595 graph_parent_is_merge(struct rev_graph *graph)
3596 {
3597 return graph->parents->size > 1;
3598 }
3600 static inline void
3601 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
3602 {
3603 struct commit *commit = graph->commit;
3605 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
3606 commit->graph[commit->graph_size++] = symbol;
3607 }
3609 static void
3610 done_rev_graph(struct rev_graph *graph)
3611 {
3612 if (graph_parent_is_merge(graph) &&
3613 graph->pos < graph->size - 1 &&
3614 graph->next->size == graph->size + graph->parents->size - 1) {
3615 size_t i = graph->pos + graph->parents->size - 1;
3617 graph->commit->graph_size = i * 2;
3618 while (i < graph->next->size - 1) {
3619 append_to_rev_graph(graph, ' ');
3620 append_to_rev_graph(graph, '\\');
3621 i++;
3622 }
3623 }
3625 graph->size = graph->pos = 0;
3626 graph->commit = NULL;
3627 memset(graph->parents, 0, sizeof(*graph->parents));
3628 }
3630 static void
3631 push_rev_graph(struct rev_graph *graph, char *parent)
3632 {
3633 int i;
3635 /* "Collapse" duplicate parents lines.
3636 *
3637 * FIXME: This needs to also update update the drawn graph but
3638 * for now it just serves as a method for pruning graph lines. */
3639 for (i = 0; i < graph->size; i++)
3640 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
3641 return;
3643 if (graph->size < SIZEOF_REVITEMS) {
3644 string_copy_rev(graph->rev[graph->size++], parent);
3645 }
3646 }
3648 static chtype
3649 get_rev_graph_symbol(struct rev_graph *graph)
3650 {
3651 chtype symbol;
3653 if (graph->parents->size == 0)
3654 symbol = REVGRAPH_INIT;
3655 else if (graph_parent_is_merge(graph))
3656 symbol = REVGRAPH_MERGE;
3657 else if (graph->pos >= graph->size)
3658 symbol = REVGRAPH_BRANCH;
3659 else
3660 symbol = REVGRAPH_COMMIT;
3662 return symbol;
3663 }
3665 static void
3666 draw_rev_graph(struct rev_graph *graph)
3667 {
3668 struct rev_filler {
3669 chtype separator, line;
3670 };
3671 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
3672 static struct rev_filler fillers[] = {
3673 { ' ', REVGRAPH_LINE },
3674 { '`', '.' },
3675 { '\'', ' ' },
3676 { '/', ' ' },
3677 };
3678 chtype symbol = get_rev_graph_symbol(graph);
3679 struct rev_filler *filler;
3680 size_t i;
3682 filler = &fillers[DEFAULT];
3684 for (i = 0; i < graph->pos; i++) {
3685 append_to_rev_graph(graph, filler->line);
3686 if (graph_parent_is_merge(graph->prev) &&
3687 graph->prev->pos == i)
3688 filler = &fillers[RSHARP];
3690 append_to_rev_graph(graph, filler->separator);
3691 }
3693 /* Place the symbol for this revision. */
3694 append_to_rev_graph(graph, symbol);
3696 if (graph->prev->size > graph->size)
3697 filler = &fillers[RDIAG];
3698 else
3699 filler = &fillers[DEFAULT];
3701 i++;
3703 for (; i < graph->size; i++) {
3704 append_to_rev_graph(graph, filler->separator);
3705 append_to_rev_graph(graph, filler->line);
3706 if (graph_parent_is_merge(graph->prev) &&
3707 i < graph->prev->pos + graph->parents->size)
3708 filler = &fillers[RSHARP];
3709 if (graph->prev->size > graph->size)
3710 filler = &fillers[LDIAG];
3711 }
3713 if (graph->prev->size > graph->size) {
3714 append_to_rev_graph(graph, filler->separator);
3715 if (filler->line != ' ')
3716 append_to_rev_graph(graph, filler->line);
3717 }
3718 }
3720 /* Prepare the next rev graph */
3721 static void
3722 prepare_rev_graph(struct rev_graph *graph)
3723 {
3724 size_t i;
3726 /* First, traverse all lines of revisions up to the active one. */
3727 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
3728 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
3729 break;
3731 push_rev_graph(graph->next, graph->rev[graph->pos]);
3732 }
3734 /* Interleave the new revision parent(s). */
3735 for (i = 0; i < graph->parents->size; i++)
3736 push_rev_graph(graph->next, graph->parents->rev[i]);
3738 /* Lastly, put any remaining revisions. */
3739 for (i = graph->pos + 1; i < graph->size; i++)
3740 push_rev_graph(graph->next, graph->rev[i]);
3741 }
3743 static void
3744 update_rev_graph(struct rev_graph *graph)
3745 {
3746 /* If this is the finalizing update ... */
3747 if (graph->commit)
3748 prepare_rev_graph(graph);
3750 /* Graph visualization needs a one rev look-ahead,
3751 * so the first update doesn't visualize anything. */
3752 if (!graph->prev->commit)
3753 return;
3755 draw_rev_graph(graph->prev);
3756 done_rev_graph(graph->prev->prev);
3757 }
3760 /*
3761 * Main view backend
3762 */
3764 static bool
3765 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3766 {
3767 char buf[DATE_COLS + 1];
3768 struct commit *commit = line->data;
3769 enum line_type type;
3770 int col = 0;
3771 size_t timelen;
3772 size_t authorlen;
3773 int trimmed = 1;
3775 if (!*commit->author)
3776 return FALSE;
3778 wmove(view->win, lineno, col);
3780 if (selected) {
3781 type = LINE_CURSOR;
3782 wattrset(view->win, get_line_attr(type));
3783 wchgat(view->win, -1, 0, type, NULL);
3785 } else {
3786 type = LINE_MAIN_COMMIT;
3787 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
3788 }
3790 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
3791 waddnstr(view->win, buf, timelen);
3792 waddstr(view->win, " ");
3794 col += DATE_COLS;
3795 wmove(view->win, lineno, col);
3796 if (type != LINE_CURSOR)
3797 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
3799 if (opt_utf8) {
3800 authorlen = utf8_length(commit->author, AUTHOR_COLS - 2, &col, &trimmed);
3801 } else {
3802 authorlen = strlen(commit->author);
3803 if (authorlen > AUTHOR_COLS - 2) {
3804 authorlen = AUTHOR_COLS - 2;
3805 trimmed = 1;
3806 }
3807 }
3809 if (trimmed) {
3810 waddnstr(view->win, commit->author, authorlen);
3811 if (type != LINE_CURSOR)
3812 wattrset(view->win, get_line_attr(LINE_MAIN_DELIM));
3813 waddch(view->win, '~');
3814 } else {
3815 waddstr(view->win, commit->author);
3816 }
3818 col += AUTHOR_COLS;
3819 if (type != LINE_CURSOR)
3820 wattrset(view->win, A_NORMAL);
3822 if (opt_rev_graph && commit->graph_size) {
3823 size_t i;
3825 wmove(view->win, lineno, col);
3826 /* Using waddch() instead of waddnstr() ensures that
3827 * they'll be rendered correctly for the cursor line. */
3828 for (i = 0; i < commit->graph_size; i++)
3829 waddch(view->win, commit->graph[i]);
3831 waddch(view->win, ' ');
3832 col += commit->graph_size + 1;
3833 }
3835 wmove(view->win, lineno, col);
3837 if (commit->refs) {
3838 size_t i = 0;
3840 do {
3841 if (type == LINE_CURSOR)
3842 ;
3843 else if (commit->refs[i]->tag)
3844 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
3845 else if (commit->refs[i]->remote)
3846 wattrset(view->win, get_line_attr(LINE_MAIN_REMOTE));
3847 else
3848 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
3849 waddstr(view->win, "[");
3850 waddstr(view->win, commit->refs[i]->name);
3851 waddstr(view->win, "]");
3852 if (type != LINE_CURSOR)
3853 wattrset(view->win, A_NORMAL);
3854 waddstr(view->win, " ");
3855 col += strlen(commit->refs[i]->name) + STRING_SIZE("[] ");
3856 } while (commit->refs[i++]->next);
3857 }
3859 if (type != LINE_CURSOR)
3860 wattrset(view->win, get_line_attr(type));
3862 {
3863 int titlelen = strlen(commit->title);
3865 if (col + titlelen > view->width)
3866 titlelen = view->width - col;
3868 waddnstr(view->win, commit->title, titlelen);
3869 }
3871 return TRUE;
3872 }
3874 /* Reads git log --pretty=raw output and parses it into the commit struct. */
3875 static bool
3876 main_read(struct view *view, char *line)
3877 {
3878 static struct rev_graph *graph = graph_stacks;
3879 enum line_type type;
3880 struct commit *commit;
3882 if (!line) {
3883 update_rev_graph(graph);
3884 return TRUE;
3885 }
3887 type = get_line_type(line);
3888 if (type == LINE_COMMIT) {
3889 commit = calloc(1, sizeof(struct commit));
3890 if (!commit)
3891 return FALSE;
3893 string_copy_rev(commit->id, line + STRING_SIZE("commit "));
3894 commit->refs = get_refs(commit->id);
3895 graph->commit = commit;
3896 add_line_data(view, commit, LINE_MAIN_COMMIT);
3897 return TRUE;
3898 }
3900 if (!view->lines)
3901 return TRUE;
3902 commit = view->line[view->lines - 1].data;
3904 switch (type) {
3905 case LINE_PARENT:
3906 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
3907 break;
3909 case LINE_AUTHOR:
3910 {
3911 /* Parse author lines where the name may be empty:
3912 * author <email@address.tld> 1138474660 +0100
3913 */
3914 char *ident = line + STRING_SIZE("author ");
3915 char *nameend = strchr(ident, '<');
3916 char *emailend = strchr(ident, '>');
3918 if (!nameend || !emailend)
3919 break;
3921 update_rev_graph(graph);
3922 graph = graph->next;
3924 *nameend = *emailend = 0;
3925 ident = chomp_string(ident);
3926 if (!*ident) {
3927 ident = chomp_string(nameend + 1);
3928 if (!*ident)
3929 ident = "Unknown";
3930 }
3932 string_ncopy(commit->author, ident, strlen(ident));
3934 /* Parse epoch and timezone */
3935 if (emailend[1] == ' ') {
3936 char *secs = emailend + 2;
3937 char *zone = strchr(secs, ' ');
3938 time_t time = (time_t) atol(secs);
3940 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
3941 long tz;
3943 zone++;
3944 tz = ('0' - zone[1]) * 60 * 60 * 10;
3945 tz += ('0' - zone[2]) * 60 * 60;
3946 tz += ('0' - zone[3]) * 60;
3947 tz += ('0' - zone[4]) * 60;
3949 if (zone[0] == '-')
3950 tz = -tz;
3952 time -= tz;
3953 }
3955 gmtime_r(&time, &commit->time);
3956 }
3957 break;
3958 }
3959 default:
3960 /* Fill in the commit title if it has not already been set. */
3961 if (commit->title[0])
3962 break;
3964 /* Require titles to start with a non-space character at the
3965 * offset used by git log. */
3966 if (strncmp(line, " ", 4))
3967 break;
3968 line += 4;
3969 /* Well, if the title starts with a whitespace character,
3970 * try to be forgiving. Otherwise we end up with no title. */
3971 while (isspace(*line))
3972 line++;
3973 if (*line == '\0')
3974 break;
3975 /* FIXME: More graceful handling of titles; append "..." to
3976 * shortened titles, etc. */
3978 string_ncopy(commit->title, line, strlen(line));
3979 }
3981 return TRUE;
3982 }
3984 static enum request
3985 main_request(struct view *view, enum request request, struct line *line)
3986 {
3987 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3989 if (request == REQ_ENTER)
3990 open_view(view, REQ_VIEW_DIFF, flags);
3991 else
3992 return request;
3994 return REQ_NONE;
3995 }
3997 static bool
3998 main_grep(struct view *view, struct line *line)
3999 {
4000 struct commit *commit = line->data;
4001 enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
4002 char buf[DATE_COLS + 1];
4003 regmatch_t pmatch;
4005 for (state = S_TITLE; state < S_END; state++) {
4006 char *text;
4008 switch (state) {
4009 case S_TITLE: text = commit->title; break;
4010 case S_AUTHOR: text = commit->author; break;
4011 case S_DATE:
4012 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
4013 continue;
4014 text = buf;
4015 break;
4017 default:
4018 return FALSE;
4019 }
4021 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4022 return TRUE;
4023 }
4025 return FALSE;
4026 }
4028 static void
4029 main_select(struct view *view, struct line *line)
4030 {
4031 struct commit *commit = line->data;
4033 string_copy_rev(view->ref, commit->id);
4034 string_copy_rev(ref_commit, view->ref);
4035 }
4037 static struct view_ops main_ops = {
4038 "commit",
4039 NULL,
4040 main_read,
4041 main_draw,
4042 main_request,
4043 main_grep,
4044 main_select,
4045 };
4048 /*
4049 * Unicode / UTF-8 handling
4050 *
4051 * NOTE: Much of the following code for dealing with unicode is derived from
4052 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
4053 * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
4054 */
4056 /* I've (over)annotated a lot of code snippets because I am not entirely
4057 * confident that the approach taken by this small UTF-8 interface is correct.
4058 * --jonas */
4060 static inline int
4061 unicode_width(unsigned long c)
4062 {
4063 if (c >= 0x1100 &&
4064 (c <= 0x115f /* Hangul Jamo */
4065 || c == 0x2329
4066 || c == 0x232a
4067 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
4068 /* CJK ... Yi */
4069 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
4070 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
4071 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
4072 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
4073 || (c >= 0xffe0 && c <= 0xffe6)
4074 || (c >= 0x20000 && c <= 0x2fffd)
4075 || (c >= 0x30000 && c <= 0x3fffd)))
4076 return 2;
4078 return 1;
4079 }
4081 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
4082 * Illegal bytes are set one. */
4083 static const unsigned char utf8_bytes[256] = {
4084 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,
4085 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,
4086 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,
4087 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,
4088 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,
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 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,
4091 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,
4092 };
4094 /* Decode UTF-8 multi-byte representation into a unicode character. */
4095 static inline unsigned long
4096 utf8_to_unicode(const char *string, size_t length)
4097 {
4098 unsigned long unicode;
4100 switch (length) {
4101 case 1:
4102 unicode = string[0];
4103 break;
4104 case 2:
4105 unicode = (string[0] & 0x1f) << 6;
4106 unicode += (string[1] & 0x3f);
4107 break;
4108 case 3:
4109 unicode = (string[0] & 0x0f) << 12;
4110 unicode += ((string[1] & 0x3f) << 6);
4111 unicode += (string[2] & 0x3f);
4112 break;
4113 case 4:
4114 unicode = (string[0] & 0x0f) << 18;
4115 unicode += ((string[1] & 0x3f) << 12);
4116 unicode += ((string[2] & 0x3f) << 6);
4117 unicode += (string[3] & 0x3f);
4118 break;
4119 case 5:
4120 unicode = (string[0] & 0x0f) << 24;
4121 unicode += ((string[1] & 0x3f) << 18);
4122 unicode += ((string[2] & 0x3f) << 12);
4123 unicode += ((string[3] & 0x3f) << 6);
4124 unicode += (string[4] & 0x3f);
4125 break;
4126 case 6:
4127 unicode = (string[0] & 0x01) << 30;
4128 unicode += ((string[1] & 0x3f) << 24);
4129 unicode += ((string[2] & 0x3f) << 18);
4130 unicode += ((string[3] & 0x3f) << 12);
4131 unicode += ((string[4] & 0x3f) << 6);
4132 unicode += (string[5] & 0x3f);
4133 break;
4134 default:
4135 die("Invalid unicode length");
4136 }
4138 /* Invalid characters could return the special 0xfffd value but NUL
4139 * should be just as good. */
4140 return unicode > 0xffff ? 0 : unicode;
4141 }
4143 /* Calculates how much of string can be shown within the given maximum width
4144 * and sets trimmed parameter to non-zero value if all of string could not be
4145 * shown.
4146 *
4147 * Additionally, adds to coloffset how many many columns to move to align with
4148 * the expected position. Takes into account how multi-byte and double-width
4149 * characters will effect the cursor position.
4150 *
4151 * Returns the number of bytes to output from string to satisfy max_width. */
4152 static size_t
4153 utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed)
4154 {
4155 const char *start = string;
4156 const char *end = strchr(string, '\0');
4157 size_t mbwidth = 0;
4158 size_t width = 0;
4160 *trimmed = 0;
4162 while (string < end) {
4163 int c = *(unsigned char *) string;
4164 unsigned char bytes = utf8_bytes[c];
4165 size_t ucwidth;
4166 unsigned long unicode;
4168 if (string + bytes > end)
4169 break;
4171 /* Change representation to figure out whether
4172 * it is a single- or double-width character. */
4174 unicode = utf8_to_unicode(string, bytes);
4175 /* FIXME: Graceful handling of invalid unicode character. */
4176 if (!unicode)
4177 break;
4179 ucwidth = unicode_width(unicode);
4180 width += ucwidth;
4181 if (width > max_width) {
4182 *trimmed = 1;
4183 break;
4184 }
4186 /* The column offset collects the differences between the
4187 * number of bytes encoding a character and the number of
4188 * columns will be used for rendering said character.
4189 *
4190 * So if some character A is encoded in 2 bytes, but will be
4191 * represented on the screen using only 1 byte this will and up
4192 * adding 1 to the multi-byte column offset.
4193 *
4194 * Assumes that no double-width character can be encoding in
4195 * less than two bytes. */
4196 if (bytes > ucwidth)
4197 mbwidth += bytes - ucwidth;
4199 string += bytes;
4200 }
4202 *coloffset += mbwidth;
4204 return string - start;
4205 }
4208 /*
4209 * Status management
4210 */
4212 /* Whether or not the curses interface has been initialized. */
4213 static bool cursed = FALSE;
4215 /* The status window is used for polling keystrokes. */
4216 static WINDOW *status_win;
4218 static bool status_empty = TRUE;
4220 /* Update status and title window. */
4221 static void
4222 report(const char *msg, ...)
4223 {
4224 struct view *view = display[current_view];
4226 if (input_mode)
4227 return;
4229 if (!status_empty || *msg) {
4230 va_list args;
4232 va_start(args, msg);
4234 wmove(status_win, 0, 0);
4235 if (*msg) {
4236 vwprintw(status_win, msg, args);
4237 status_empty = FALSE;
4238 } else {
4239 status_empty = TRUE;
4240 }
4241 wclrtoeol(status_win);
4242 wrefresh(status_win);
4244 va_end(args);
4245 }
4247 update_view_title(view);
4248 update_display_cursor(view);
4249 }
4251 /* Controls when nodelay should be in effect when polling user input. */
4252 static void
4253 set_nonblocking_input(bool loading)
4254 {
4255 static unsigned int loading_views;
4257 if ((loading == FALSE && loading_views-- == 1) ||
4258 (loading == TRUE && loading_views++ == 0))
4259 nodelay(status_win, loading);
4260 }
4262 static void
4263 init_display(void)
4264 {
4265 int x, y;
4267 /* Initialize the curses library */
4268 if (isatty(STDIN_FILENO)) {
4269 cursed = !!initscr();
4270 } else {
4271 /* Leave stdin and stdout alone when acting as a pager. */
4272 FILE *io = fopen("/dev/tty", "r+");
4274 if (!io)
4275 die("Failed to open /dev/tty");
4276 cursed = !!newterm(NULL, io, io);
4277 }
4279 if (!cursed)
4280 die("Failed to initialize curses");
4282 nonl(); /* Tell curses not to do NL->CR/NL on output */
4283 cbreak(); /* Take input chars one at a time, no wait for \n */
4284 noecho(); /* Don't echo input */
4285 leaveok(stdscr, TRUE);
4287 if (has_colors())
4288 init_colors();
4290 getmaxyx(stdscr, y, x);
4291 status_win = newwin(1, 0, y - 1, 0);
4292 if (!status_win)
4293 die("Failed to create status window");
4295 /* Enable keyboard mapping */
4296 keypad(status_win, TRUE);
4297 wbkgdset(status_win, get_line_attr(LINE_STATUS));
4298 }
4300 static char *
4301 read_prompt(const char *prompt)
4302 {
4303 enum { READING, STOP, CANCEL } status = READING;
4304 static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
4305 int pos = 0;
4307 while (status == READING) {
4308 struct view *view;
4309 int i, key;
4311 input_mode = TRUE;
4313 foreach_view (view, i)
4314 update_view(view);
4316 input_mode = FALSE;
4318 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
4319 wclrtoeol(status_win);
4321 /* Refresh, accept single keystroke of input */
4322 key = wgetch(status_win);
4323 switch (key) {
4324 case KEY_RETURN:
4325 case KEY_ENTER:
4326 case '\n':
4327 status = pos ? STOP : CANCEL;
4328 break;
4330 case KEY_BACKSPACE:
4331 if (pos > 0)
4332 pos--;
4333 else
4334 status = CANCEL;
4335 break;
4337 case KEY_ESC:
4338 status = CANCEL;
4339 break;
4341 case ERR:
4342 break;
4344 default:
4345 if (pos >= sizeof(buf)) {
4346 report("Input string too long");
4347 return NULL;
4348 }
4350 if (isprint(key))
4351 buf[pos++] = (char) key;
4352 }
4353 }
4355 /* Clear the status window */
4356 status_empty = FALSE;
4357 report("");
4359 if (status == CANCEL)
4360 return NULL;
4362 buf[pos++] = 0;
4364 return buf;
4365 }
4367 /*
4368 * Repository references
4369 */
4371 static struct ref *refs;
4372 static size_t refs_size;
4374 /* Id <-> ref store */
4375 static struct ref ***id_refs;
4376 static size_t id_refs_size;
4378 static struct ref **
4379 get_refs(char *id)
4380 {
4381 struct ref ***tmp_id_refs;
4382 struct ref **ref_list = NULL;
4383 size_t ref_list_size = 0;
4384 size_t i;
4386 for (i = 0; i < id_refs_size; i++)
4387 if (!strcmp(id, id_refs[i][0]->id))
4388 return id_refs[i];
4390 tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
4391 if (!tmp_id_refs)
4392 return NULL;
4394 id_refs = tmp_id_refs;
4396 for (i = 0; i < refs_size; i++) {
4397 struct ref **tmp;
4399 if (strcmp(id, refs[i].id))
4400 continue;
4402 tmp = realloc(ref_list, (ref_list_size + 1) * sizeof(*ref_list));
4403 if (!tmp) {
4404 if (ref_list)
4405 free(ref_list);
4406 return NULL;
4407 }
4409 ref_list = tmp;
4410 if (ref_list_size > 0)
4411 ref_list[ref_list_size - 1]->next = 1;
4412 ref_list[ref_list_size] = &refs[i];
4414 /* XXX: The properties of the commit chains ensures that we can
4415 * safely modify the shared ref. The repo references will
4416 * always be similar for the same id. */
4417 ref_list[ref_list_size]->next = 0;
4418 ref_list_size++;
4419 }
4421 if (ref_list)
4422 id_refs[id_refs_size++] = ref_list;
4424 return ref_list;
4425 }
4427 static int
4428 read_ref(char *id, size_t idlen, char *name, size_t namelen)
4429 {
4430 struct ref *ref;
4431 bool tag = FALSE;
4432 bool remote = FALSE;
4434 if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
4435 /* Commits referenced by tags has "^{}" appended. */
4436 if (name[namelen - 1] != '}')
4437 return OK;
4439 while (namelen > 0 && name[namelen] != '^')
4440 namelen--;
4442 tag = TRUE;
4443 namelen -= STRING_SIZE("refs/tags/");
4444 name += STRING_SIZE("refs/tags/");
4446 } else if (!strncmp(name, "refs/remotes/", STRING_SIZE("refs/remotes/"))) {
4447 remote = TRUE;
4448 namelen -= STRING_SIZE("refs/remotes/");
4449 name += STRING_SIZE("refs/remotes/");
4451 } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
4452 namelen -= STRING_SIZE("refs/heads/");
4453 name += STRING_SIZE("refs/heads/");
4455 } else if (!strcmp(name, "HEAD")) {
4456 return OK;
4457 }
4459 refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
4460 if (!refs)
4461 return ERR;
4463 ref = &refs[refs_size++];
4464 ref->name = malloc(namelen + 1);
4465 if (!ref->name)
4466 return ERR;
4468 strncpy(ref->name, name, namelen);
4469 ref->name[namelen] = 0;
4470 ref->tag = tag;
4471 ref->remote = remote;
4472 string_copy_rev(ref->id, id);
4474 return OK;
4475 }
4477 static int
4478 load_refs(void)
4479 {
4480 const char *cmd_env = getenv("TIG_LS_REMOTE");
4481 const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
4483 return read_properties(popen(cmd, "r"), "\t", read_ref);
4484 }
4486 static int
4487 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
4488 {
4489 if (!strcmp(name, "i18n.commitencoding"))
4490 string_ncopy(opt_encoding, value, valuelen);
4492 if (!strcmp(name, "core.editor"))
4493 string_ncopy(opt_editor, value, valuelen);
4495 return OK;
4496 }
4498 static int
4499 load_repo_config(void)
4500 {
4501 return read_properties(popen(GIT_CONFIG " --list", "r"),
4502 "=", read_repo_config_option);
4503 }
4505 static int
4506 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
4507 {
4508 if (!opt_git_dir[0])
4509 string_ncopy(opt_git_dir, name, namelen);
4510 else
4511 string_ncopy(opt_cdup, name, namelen);
4512 return OK;
4513 }
4515 /* XXX: The line outputted by "--show-cdup" can be empty so the option
4516 * must be the last one! */
4517 static int
4518 load_repo_info(void)
4519 {
4520 return read_properties(popen("git rev-parse --git-dir --show-cdup 2>/dev/null", "r"),
4521 "=", read_repo_info);
4522 }
4524 static int
4525 read_properties(FILE *pipe, const char *separators,
4526 int (*read_property)(char *, size_t, char *, size_t))
4527 {
4528 char buffer[BUFSIZ];
4529 char *name;
4530 int state = OK;
4532 if (!pipe)
4533 return ERR;
4535 while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
4536 char *value;
4537 size_t namelen;
4538 size_t valuelen;
4540 name = chomp_string(name);
4541 namelen = strcspn(name, separators);
4543 if (name[namelen]) {
4544 name[namelen] = 0;
4545 value = chomp_string(name + namelen + 1);
4546 valuelen = strlen(value);
4548 } else {
4549 value = "";
4550 valuelen = 0;
4551 }
4553 state = read_property(name, namelen, value, valuelen);
4554 }
4556 if (state != ERR && ferror(pipe))
4557 state = ERR;
4559 pclose(pipe);
4561 return state;
4562 }
4565 /*
4566 * Main
4567 */
4569 static void __NORETURN
4570 quit(int sig)
4571 {
4572 /* XXX: Restore tty modes and let the OS cleanup the rest! */
4573 if (cursed)
4574 endwin();
4575 exit(0);
4576 }
4578 static void __NORETURN
4579 die(const char *err, ...)
4580 {
4581 va_list args;
4583 endwin();
4585 va_start(args, err);
4586 fputs("tig: ", stderr);
4587 vfprintf(stderr, err, args);
4588 fputs("\n", stderr);
4589 va_end(args);
4591 exit(1);
4592 }
4594 int
4595 main(int argc, char *argv[])
4596 {
4597 struct view *view;
4598 enum request request;
4599 size_t i;
4601 signal(SIGINT, quit);
4603 if (setlocale(LC_ALL, "")) {
4604 char *codeset = nl_langinfo(CODESET);
4606 string_ncopy(opt_codeset, codeset, strlen(codeset));
4607 }
4609 if (load_repo_info() == ERR)
4610 die("Failed to load repo info.");
4612 /* Require a git repository unless when running in pager mode. */
4613 if (!opt_git_dir[0])
4614 die("Not a git repository");
4616 if (load_options() == ERR)
4617 die("Failed to load user config.");
4619 /* Load the repo config file so options can be overwritten from
4620 * the command line. */
4621 if (load_repo_config() == ERR)
4622 die("Failed to load repo config.");
4624 if (!parse_options(argc, argv))
4625 return 0;
4627 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
4628 opt_iconv = iconv_open(opt_codeset, opt_encoding);
4629 if (opt_iconv == ICONV_NONE)
4630 die("Failed to initialize character set conversion");
4631 }
4633 if (load_refs() == ERR)
4634 die("Failed to load refs.");
4636 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
4637 view->cmd_env = getenv(view->cmd_env);
4639 request = opt_request;
4641 init_display();
4643 while (view_driver(display[current_view], request)) {
4644 int key;
4645 int i;
4647 foreach_view (view, i)
4648 update_view(view);
4650 /* Refresh, accept single keystroke of input */
4651 key = wgetch(status_win);
4653 /* wgetch() with nodelay() enabled returns ERR when there's no
4654 * input. */
4655 if (key == ERR) {
4656 request = REQ_NONE;
4657 continue;
4658 }
4660 request = get_keybinding(display[current_view]->keymap, key);
4662 /* Some low-level request handling. This keeps access to
4663 * status_win restricted. */
4664 switch (request) {
4665 case REQ_PROMPT:
4666 {
4667 char *cmd = read_prompt(":");
4669 if (cmd && string_format(opt_cmd, "git %s", cmd)) {
4670 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
4671 opt_request = REQ_VIEW_DIFF;
4672 } else {
4673 opt_request = REQ_VIEW_PAGER;
4674 }
4675 break;
4676 }
4678 request = REQ_NONE;
4679 break;
4680 }
4681 case REQ_SEARCH:
4682 case REQ_SEARCH_BACK:
4683 {
4684 const char *prompt = request == REQ_SEARCH
4685 ? "/" : "?";
4686 char *search = read_prompt(prompt);
4688 if (search)
4689 string_ncopy(opt_search, search, strlen(search));
4690 else
4691 request = REQ_NONE;
4692 break;
4693 }
4694 case REQ_SCREEN_RESIZE:
4695 {
4696 int height, width;
4698 getmaxyx(stdscr, height, width);
4700 /* Resize the status view and let the view driver take
4701 * care of resizing the displayed views. */
4702 wresize(status_win, 1, width);
4703 mvwin(status_win, height - 1, 0);
4704 wrefresh(status_win);
4705 break;
4706 }
4707 default:
4708 break;
4709 }
4710 }
4712 quit(0);
4714 return 0;
4715 }