![[tokkee]](http://tokkee.org/images/avatar.png){kind=avatar}
1 /* Copyright (c) 2006-2007 Jonas Fonseca <fonseca@diku.dk>
2 *
3 * This program is free software; you can redistribute it and/or
4 * modify it under the terms of the GNU General Public License as
5 * published by the Free Software Foundation; either version 2 of
6 * the License, or (at your option) any later version.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 */
14 #ifdef HAVE_CONFIG_H
15 #include "config.h"
16 #endif
18 #ifndef TIG_VERSION
19 #define TIG_VERSION "unknown-version"
20 #endif
22 #ifndef DEBUG
23 #define NDEBUG
24 #endif
26 #include <assert.h>
27 #include <errno.h>
28 #include <ctype.h>
29 #include <signal.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #include <sys/types.h>
35 #include <sys/stat.h>
36 #include <unistd.h>
37 #include <time.h>
39 #include <regex.h>
41 #include <locale.h>
42 #include <langinfo.h>
43 #include <iconv.h>
45 #include <curses.h>
47 #if __GNUC__ >= 3
48 #define __NORETURN __attribute__((__noreturn__))
49 #else
50 #define __NORETURN
51 #endif
53 static void __NORETURN die(const char *err, ...);
54 static void report(const char *msg, ...);
55 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, size_t, char *, size_t));
56 static void set_nonblocking_input(bool loading);
57 static size_t utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed);
59 #define ABS(x) ((x) >= 0 ? (x) : -(x))
60 #define MIN(x, y) ((x) < (y) ? (x) : (y))
62 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
63 #define STRING_SIZE(x) (sizeof(x) - 1)
65 #define SIZEOF_STR 1024 /* Default string size. */
66 #define SIZEOF_REF 256 /* Size of symbolic or SHA1 ID. */
67 #define SIZEOF_REV 41 /* Holds a SHA-1 and an ending NUL */
69 /* Revision graph */
71 #define REVGRAPH_INIT 'I'
72 #define REVGRAPH_MERGE 'M'
73 #define REVGRAPH_BRANCH '+'
74 #define REVGRAPH_COMMIT '*'
75 #define REVGRAPH_LINE '|'
77 #define SIZEOF_REVGRAPH 19 /* Size of revision ancestry graphics. */
79 /* This color name can be used to refer to the default term colors. */
80 #define COLOR_DEFAULT (-1)
82 #define ICONV_NONE ((iconv_t) -1)
83 #ifndef ICONV_CONST
84 #define ICONV_CONST /* nothing */
85 #endif
87 /* The format and size of the date column in the main view. */
88 #define DATE_FORMAT "%Y-%m-%d %H:%M"
89 #define DATE_COLS STRING_SIZE("2006-04-29 14:21 ")
91 #define AUTHOR_COLS 20
93 /* The default interval between line numbers. */
94 #define NUMBER_INTERVAL 1
96 #define TABSIZE 8
98 #define SCALE_SPLIT_VIEW(height) ((height) * 2 / 3)
100 #ifndef GIT_CONFIG
101 #define GIT_CONFIG "git config"
102 #endif
104 #define TIG_LS_REMOTE \
105 "git ls-remote $(git rev-parse --git-dir) 2>/dev/null"
107 #define TIG_DIFF_CMD \
108 "git show --root --patch-with-stat --find-copies-harder -B -C %s 2>/dev/null"
110 #define TIG_LOG_CMD \
111 "git log --cc --stat -n100 %s 2>/dev/null"
113 #define TIG_MAIN_CMD \
114 "git log --topo-order --pretty=raw %s 2>/dev/null"
116 #define TIG_TREE_CMD \
117 "git ls-tree %s %s"
119 #define TIG_BLOB_CMD \
120 "git cat-file blob %s"
122 /* XXX: Needs to be defined to the empty string. */
123 #define TIG_HELP_CMD ""
124 #define TIG_PAGER_CMD ""
125 #define TIG_STATUS_CMD ""
126 #define TIG_STAGE_CMD ""
128 /* Some ascii-shorthands fitted into the ncurses namespace. */
129 #define KEY_TAB '\t'
130 #define KEY_RETURN '\r'
131 #define KEY_ESC 27
134 struct ref {
135 char *name; /* Ref name; tag or head names are shortened. */
136 char id[SIZEOF_REV]; /* Commit SHA1 ID */
137 unsigned int tag:1; /* Is it a tag? */
138 unsigned int remote:1; /* Is it a remote ref? */
139 unsigned int next:1; /* For ref lists: are there more refs? */
140 };
142 static struct ref **get_refs(char *id);
144 struct int_map {
145 const char *name;
146 int namelen;
147 int value;
148 };
150 static int
151 set_from_int_map(struct int_map *map, size_t map_size,
152 int *value, const char *name, int namelen)
153 {
155 int i;
157 for (i = 0; i < map_size; i++)
158 if (namelen == map[i].namelen &&
159 !strncasecmp(name, map[i].name, namelen)) {
160 *value = map[i].value;
161 return OK;
162 }
164 return ERR;
165 }
168 /*
169 * String helpers
170 */
172 static inline void
173 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
174 {
175 if (srclen > dstlen - 1)
176 srclen = dstlen - 1;
178 strncpy(dst, src, srclen);
179 dst[srclen] = 0;
180 }
182 /* Shorthands for safely copying into a fixed buffer. */
184 #define string_copy(dst, src) \
185 string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
187 #define string_ncopy(dst, src, srclen) \
188 string_ncopy_do(dst, sizeof(dst), src, srclen)
190 #define string_copy_rev(dst, src) \
191 string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
193 #define string_add(dst, from, src) \
194 string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
196 static char *
197 chomp_string(char *name)
198 {
199 int namelen;
201 while (isspace(*name))
202 name++;
204 namelen = strlen(name) - 1;
205 while (namelen > 0 && isspace(name[namelen]))
206 name[namelen--] = 0;
208 return name;
209 }
211 static bool
212 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
213 {
214 va_list args;
215 size_t pos = bufpos ? *bufpos : 0;
217 va_start(args, fmt);
218 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
219 va_end(args);
221 if (bufpos)
222 *bufpos = pos;
224 return pos >= bufsize ? FALSE : TRUE;
225 }
227 #define string_format(buf, fmt, args...) \
228 string_nformat(buf, sizeof(buf), NULL, fmt, args)
230 #define string_format_from(buf, from, fmt, args...) \
231 string_nformat(buf, sizeof(buf), from, fmt, args)
233 static int
234 string_enum_compare(const char *str1, const char *str2, int len)
235 {
236 size_t i;
238 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
240 /* Diff-Header == DIFF_HEADER */
241 for (i = 0; i < len; i++) {
242 if (toupper(str1[i]) == toupper(str2[i]))
243 continue;
245 if (string_enum_sep(str1[i]) &&
246 string_enum_sep(str2[i]))
247 continue;
249 return str1[i] - str2[i];
250 }
252 return 0;
253 }
255 /* Shell quoting
256 *
257 * NOTE: The following is a slightly modified copy of the git project's shell
258 * quoting routines found in the quote.c file.
259 *
260 * Help to copy the thing properly quoted for the shell safety. any single
261 * quote is replaced with '\'', any exclamation point is replaced with '\!',
262 * and the whole thing is enclosed in a
263 *
264 * E.g.
265 * original sq_quote result
266 * name ==> name ==> 'name'
267 * a b ==> a b ==> 'a b'
268 * a'b ==> a'\''b ==> 'a'\''b'
269 * a!b ==> a'\!'b ==> 'a'\!'b'
270 */
272 static size_t
273 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
274 {
275 char c;
277 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
279 BUFPUT('\'');
280 while ((c = *src++)) {
281 if (c == '\'' || c == '!') {
282 BUFPUT('\'');
283 BUFPUT('\\');
284 BUFPUT(c);
285 BUFPUT('\'');
286 } else {
287 BUFPUT(c);
288 }
289 }
290 BUFPUT('\'');
292 if (bufsize < SIZEOF_STR)
293 buf[bufsize] = 0;
295 return bufsize;
296 }
299 /*
300 * User requests
301 */
303 #define REQ_INFO \
304 /* XXX: Keep the view request first and in sync with views[]. */ \
305 REQ_GROUP("View switching") \
306 REQ_(VIEW_MAIN, "Show main view"), \
307 REQ_(VIEW_DIFF, "Show diff view"), \
308 REQ_(VIEW_LOG, "Show log view"), \
309 REQ_(VIEW_TREE, "Show tree view"), \
310 REQ_(VIEW_BLOB, "Show blob view"), \
311 REQ_(VIEW_HELP, "Show help page"), \
312 REQ_(VIEW_PAGER, "Show pager view"), \
313 REQ_(VIEW_STATUS, "Show status view"), \
314 REQ_(VIEW_STAGE, "Show stage view"), \
315 \
316 REQ_GROUP("View manipulation") \
317 REQ_(ENTER, "Enter current line and scroll"), \
318 REQ_(NEXT, "Move to next"), \
319 REQ_(PREVIOUS, "Move to previous"), \
320 REQ_(VIEW_NEXT, "Move focus to next view"), \
321 REQ_(VIEW_CLOSE, "Close the current view"), \
322 REQ_(QUIT, "Close all views and quit"), \
323 \
324 REQ_GROUP("Cursor navigation") \
325 REQ_(MOVE_UP, "Move cursor one line up"), \
326 REQ_(MOVE_DOWN, "Move cursor one line down"), \
327 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
328 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
329 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
330 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
331 \
332 REQ_GROUP("Scrolling") \
333 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
334 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
335 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
336 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
337 \
338 REQ_GROUP("Searching") \
339 REQ_(SEARCH, "Search the view"), \
340 REQ_(SEARCH_BACK, "Search backwards in the view"), \
341 REQ_(FIND_NEXT, "Find next search match"), \
342 REQ_(FIND_PREV, "Find previous search match"), \
343 \
344 REQ_GROUP("Misc") \
345 REQ_(NONE, "Do nothing"), \
346 REQ_(PROMPT, "Bring up the prompt"), \
347 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
348 REQ_(SCREEN_RESIZE, "Resize the screen"), \
349 REQ_(SHOW_VERSION, "Show version information"), \
350 REQ_(STOP_LOADING, "Stop all loading views"), \
351 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
352 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization"), \
353 REQ_(STATUS_UPDATE, "Update file status"), \
354 REQ_(EDIT, "Open in editor"), \
355 REQ_(CHERRY_PICK, "Cherry-pick commit to current branch")
358 /* User action requests. */
359 enum request {
360 #define REQ_GROUP(help)
361 #define REQ_(req, help) REQ_##req
363 /* Offset all requests to avoid conflicts with ncurses getch values. */
364 REQ_OFFSET = KEY_MAX + 1,
365 REQ_INFO,
366 REQ_UNKNOWN,
368 #undef REQ_GROUP
369 #undef REQ_
370 };
372 struct request_info {
373 enum request request;
374 char *name;
375 int namelen;
376 char *help;
377 };
379 static struct request_info req_info[] = {
380 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
381 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
382 REQ_INFO
383 #undef REQ_GROUP
384 #undef REQ_
385 };
387 static enum request
388 get_request(const char *name)
389 {
390 int namelen = strlen(name);
391 int i;
393 for (i = 0; i < ARRAY_SIZE(req_info); i++)
394 if (req_info[i].namelen == namelen &&
395 !string_enum_compare(req_info[i].name, name, namelen))
396 return req_info[i].request;
398 return REQ_UNKNOWN;
399 }
402 /*
403 * Options
404 */
406 static const char usage[] =
407 "tig " TIG_VERSION " (" __DATE__ ")\n"
408 "\n"
409 "Usage: tig [options]\n"
410 " or: tig [options] [--] [git log options]\n"
411 " or: tig [options] log [git log options]\n"
412 " or: tig [options] diff [git diff options]\n"
413 " or: tig [options] show [git show options]\n"
414 " or: tig [options] < [git command output]\n"
415 "\n"
416 "Options:\n"
417 " -l Start up in log view\n"
418 " -d Start up in diff view\n"
419 " -S Start up in status view\n"
420 " -n[I], --line-number[=I] Show line numbers with given interval\n"
421 " -b[N], --tab-size[=N] Set number of spaces for tab expansion\n"
422 " -- Mark end of tig options\n"
423 " -v, --version Show version and exit\n"
424 " -h, --help Show help message and exit\n";
426 /* Option and state variables. */
427 static bool opt_line_number = FALSE;
428 static bool opt_rev_graph = FALSE;
429 static int opt_num_interval = NUMBER_INTERVAL;
430 static int opt_tab_size = TABSIZE;
431 static enum request opt_request = REQ_VIEW_MAIN;
432 static char opt_cmd[SIZEOF_STR] = "";
433 static char opt_path[SIZEOF_STR] = "";
434 static FILE *opt_pipe = NULL;
435 static char opt_encoding[20] = "UTF-8";
436 static bool opt_utf8 = TRUE;
437 static char opt_codeset[20] = "UTF-8";
438 static iconv_t opt_iconv = ICONV_NONE;
439 static char opt_search[SIZEOF_STR] = "";
440 static char opt_cdup[SIZEOF_STR] = "";
441 static char opt_git_dir[SIZEOF_STR] = "";
442 static char opt_editor[SIZEOF_STR] = "";
444 enum option_type {
445 OPT_NONE,
446 OPT_INT,
447 };
449 static bool
450 check_option(char *opt, char short_name, char *name, enum option_type type, ...)
451 {
452 va_list args;
453 char *value = "";
454 int *number;
456 if (opt[0] != '-')
457 return FALSE;
459 if (opt[1] == '-') {
460 int namelen = strlen(name);
462 opt += 2;
464 if (strncmp(opt, name, namelen))
465 return FALSE;
467 if (opt[namelen] == '=')
468 value = opt + namelen + 1;
470 } else {
471 if (!short_name || opt[1] != short_name)
472 return FALSE;
473 value = opt + 2;
474 }
476 va_start(args, type);
477 if (type == OPT_INT) {
478 number = va_arg(args, int *);
479 if (isdigit(*value))
480 *number = atoi(value);
481 }
482 va_end(args);
484 return TRUE;
485 }
487 /* Returns the index of log or diff command or -1 to exit. */
488 static bool
489 parse_options(int argc, char *argv[])
490 {
491 int i;
493 for (i = 1; i < argc; i++) {
494 char *opt = argv[i];
496 if (!strcmp(opt, "log") ||
497 !strcmp(opt, "diff") ||
498 !strcmp(opt, "show")) {
499 opt_request = opt[0] == 'l'
500 ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
501 break;
502 }
504 if (opt[0] && opt[0] != '-')
505 break;
507 if (!strcmp(opt, "-l")) {
508 opt_request = REQ_VIEW_LOG;
509 continue;
510 }
512 if (!strcmp(opt, "-d")) {
513 opt_request = REQ_VIEW_DIFF;
514 continue;
515 }
517 if (!strcmp(opt, "-S")) {
518 opt_request = REQ_VIEW_STATUS;
519 continue;
520 }
522 if (check_option(opt, 'n', "line-number", OPT_INT, &opt_num_interval)) {
523 opt_line_number = TRUE;
524 continue;
525 }
527 if (check_option(opt, 'b', "tab-size", OPT_INT, &opt_tab_size)) {
528 opt_tab_size = MIN(opt_tab_size, TABSIZE);
529 continue;
530 }
532 if (check_option(opt, 'v', "version", OPT_NONE)) {
533 printf("tig version %s\n", TIG_VERSION);
534 return FALSE;
535 }
537 if (check_option(opt, 'h', "help", OPT_NONE)) {
538 printf(usage);
539 return FALSE;
540 }
542 if (!strcmp(opt, "--")) {
543 i++;
544 break;
545 }
547 die("unknown option '%s'\n\n%s", opt, usage);
548 }
550 if (!isatty(STDIN_FILENO)) {
551 opt_request = REQ_VIEW_PAGER;
552 opt_pipe = stdin;
554 } else if (i < argc) {
555 size_t buf_size;
557 if (opt_request == REQ_VIEW_MAIN)
558 /* XXX: This is vulnerable to the user overriding
559 * options required for the main view parser. */
560 string_copy(opt_cmd, "git log --pretty=raw");
561 else
562 string_copy(opt_cmd, "git");
563 buf_size = strlen(opt_cmd);
565 while (buf_size < sizeof(opt_cmd) && i < argc) {
566 opt_cmd[buf_size++] = ' ';
567 buf_size = sq_quote(opt_cmd, buf_size, argv[i++]);
568 }
570 if (buf_size >= sizeof(opt_cmd))
571 die("command too long");
573 opt_cmd[buf_size] = 0;
574 }
576 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
577 opt_utf8 = FALSE;
579 return TRUE;
580 }
583 /*
584 * Line-oriented content detection.
585 */
587 #define LINE_INFO \
588 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
589 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
590 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
591 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
592 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
593 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
594 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
595 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
596 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
597 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
598 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
599 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
600 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
601 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
602 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
603 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
604 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
605 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
606 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
607 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
608 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
609 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
610 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
611 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
612 LINE(AUTHOR, "author ", COLOR_CYAN, COLOR_DEFAULT, 0), \
613 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
614 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
615 LINE(ACKED, " Acked-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
616 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
617 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
618 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
619 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
620 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
621 LINE(MAIN_DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
622 LINE(MAIN_AUTHOR, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
623 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
624 LINE(MAIN_DELIM, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
625 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
626 LINE(MAIN_REMOTE, "", COLOR_YELLOW, COLOR_DEFAULT, A_BOLD), \
627 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
628 LINE(TREE_DIR, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
629 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
630 LINE(STAT_SECTION, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
631 LINE(STAT_NONE, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
632 LINE(STAT_STAGED, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
633 LINE(STAT_UNSTAGED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
634 LINE(STAT_UNTRACKED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0)
636 enum line_type {
637 #define LINE(type, line, fg, bg, attr) \
638 LINE_##type
639 LINE_INFO
640 #undef LINE
641 };
643 struct line_info {
644 const char *name; /* Option name. */
645 int namelen; /* Size of option name. */
646 const char *line; /* The start of line to match. */
647 int linelen; /* Size of string to match. */
648 int fg, bg, attr; /* Color and text attributes for the lines. */
649 };
651 static struct line_info line_info[] = {
652 #define LINE(type, line, fg, bg, attr) \
653 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
654 LINE_INFO
655 #undef LINE
656 };
658 static enum line_type
659 get_line_type(char *line)
660 {
661 int linelen = strlen(line);
662 enum line_type type;
664 for (type = 0; type < ARRAY_SIZE(line_info); type++)
665 /* Case insensitive search matches Signed-off-by lines better. */
666 if (linelen >= line_info[type].linelen &&
667 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
668 return type;
670 return LINE_DEFAULT;
671 }
673 static inline int
674 get_line_attr(enum line_type type)
675 {
676 assert(type < ARRAY_SIZE(line_info));
677 return COLOR_PAIR(type) | line_info[type].attr;
678 }
680 static struct line_info *
681 get_line_info(char *name, int namelen)
682 {
683 enum line_type type;
685 for (type = 0; type < ARRAY_SIZE(line_info); type++)
686 if (namelen == line_info[type].namelen &&
687 !string_enum_compare(line_info[type].name, name, namelen))
688 return &line_info[type];
690 return NULL;
691 }
693 static void
694 init_colors(void)
695 {
696 int default_bg = COLOR_BLACK;
697 int default_fg = COLOR_WHITE;
698 enum line_type type;
700 start_color();
702 if (use_default_colors() != ERR) {
703 default_bg = -1;
704 default_fg = -1;
705 }
707 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
708 struct line_info *info = &line_info[type];
709 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
710 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
712 init_pair(type, fg, bg);
713 }
714 }
716 struct line {
717 enum line_type type;
719 /* State flags */
720 unsigned int selected:1;
722 void *data; /* User data */
723 };
726 /*
727 * Keys
728 */
730 struct keybinding {
731 int alias;
732 enum request request;
733 struct keybinding *next;
734 };
736 static struct keybinding default_keybindings[] = {
737 /* View switching */
738 { 'm', REQ_VIEW_MAIN },
739 { 'd', REQ_VIEW_DIFF },
740 { 'l', REQ_VIEW_LOG },
741 { 't', REQ_VIEW_TREE },
742 { 'f', REQ_VIEW_BLOB },
743 { 'p', REQ_VIEW_PAGER },
744 { 'h', REQ_VIEW_HELP },
745 { 'S', REQ_VIEW_STATUS },
746 { 'c', REQ_VIEW_STAGE },
748 /* View manipulation */
749 { 'q', REQ_VIEW_CLOSE },
750 { KEY_TAB, REQ_VIEW_NEXT },
751 { KEY_RETURN, REQ_ENTER },
752 { KEY_UP, REQ_PREVIOUS },
753 { KEY_DOWN, REQ_NEXT },
755 /* Cursor navigation */
756 { 'k', REQ_MOVE_UP },
757 { 'j', REQ_MOVE_DOWN },
758 { KEY_HOME, REQ_MOVE_FIRST_LINE },
759 { KEY_END, REQ_MOVE_LAST_LINE },
760 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
761 { ' ', REQ_MOVE_PAGE_DOWN },
762 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
763 { 'b', REQ_MOVE_PAGE_UP },
764 { '-', REQ_MOVE_PAGE_UP },
766 /* Scrolling */
767 { KEY_IC, REQ_SCROLL_LINE_UP },
768 { KEY_DC, REQ_SCROLL_LINE_DOWN },
769 { 'w', REQ_SCROLL_PAGE_UP },
770 { 's', REQ_SCROLL_PAGE_DOWN },
772 /* Searching */
773 { '/', REQ_SEARCH },
774 { '?', REQ_SEARCH_BACK },
775 { 'n', REQ_FIND_NEXT },
776 { 'N', REQ_FIND_PREV },
778 /* Misc */
779 { 'Q', REQ_QUIT },
780 { 'z', REQ_STOP_LOADING },
781 { 'v', REQ_SHOW_VERSION },
782 { 'r', REQ_SCREEN_REDRAW },
783 { '.', REQ_TOGGLE_LINENO },
784 { 'g', REQ_TOGGLE_REV_GRAPH },
785 { ':', REQ_PROMPT },
786 { 'u', REQ_STATUS_UPDATE },
787 { 'e', REQ_EDIT },
788 { 'C', REQ_CHERRY_PICK },
790 /* Using the ncurses SIGWINCH handler. */
791 { KEY_RESIZE, REQ_SCREEN_RESIZE },
792 };
794 #define KEYMAP_INFO \
795 KEYMAP_(GENERIC), \
796 KEYMAP_(MAIN), \
797 KEYMAP_(DIFF), \
798 KEYMAP_(LOG), \
799 KEYMAP_(TREE), \
800 KEYMAP_(BLOB), \
801 KEYMAP_(PAGER), \
802 KEYMAP_(HELP), \
803 KEYMAP_(STATUS), \
804 KEYMAP_(STAGE)
806 enum keymap {
807 #define KEYMAP_(name) KEYMAP_##name
808 KEYMAP_INFO
809 #undef KEYMAP_
810 };
812 static struct int_map keymap_table[] = {
813 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
814 KEYMAP_INFO
815 #undef KEYMAP_
816 };
818 #define set_keymap(map, name) \
819 set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
821 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
823 static void
824 add_keybinding(enum keymap keymap, enum request request, int key)
825 {
826 struct keybinding *keybinding;
828 keybinding = calloc(1, sizeof(*keybinding));
829 if (!keybinding)
830 die("Failed to allocate keybinding");
832 keybinding->alias = key;
833 keybinding->request = request;
834 keybinding->next = keybindings[keymap];
835 keybindings[keymap] = keybinding;
836 }
838 /* Looks for a key binding first in the given map, then in the generic map, and
839 * lastly in the default keybindings. */
840 static enum request
841 get_keybinding(enum keymap keymap, int key)
842 {
843 struct keybinding *kbd;
844 int i;
846 for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
847 if (kbd->alias == key)
848 return kbd->request;
850 for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
851 if (kbd->alias == key)
852 return kbd->request;
854 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
855 if (default_keybindings[i].alias == key)
856 return default_keybindings[i].request;
858 return (enum request) key;
859 }
862 struct key {
863 char *name;
864 int value;
865 };
867 static struct key key_table[] = {
868 { "Enter", KEY_RETURN },
869 { "Space", ' ' },
870 { "Backspace", KEY_BACKSPACE },
871 { "Tab", KEY_TAB },
872 { "Escape", KEY_ESC },
873 { "Left", KEY_LEFT },
874 { "Right", KEY_RIGHT },
875 { "Up", KEY_UP },
876 { "Down", KEY_DOWN },
877 { "Insert", KEY_IC },
878 { "Delete", KEY_DC },
879 { "Hash", '#' },
880 { "Home", KEY_HOME },
881 { "End", KEY_END },
882 { "PageUp", KEY_PPAGE },
883 { "PageDown", KEY_NPAGE },
884 { "F1", KEY_F(1) },
885 { "F2", KEY_F(2) },
886 { "F3", KEY_F(3) },
887 { "F4", KEY_F(4) },
888 { "F5", KEY_F(5) },
889 { "F6", KEY_F(6) },
890 { "F7", KEY_F(7) },
891 { "F8", KEY_F(8) },
892 { "F9", KEY_F(9) },
893 { "F10", KEY_F(10) },
894 { "F11", KEY_F(11) },
895 { "F12", KEY_F(12) },
896 };
898 static int
899 get_key_value(const char *name)
900 {
901 int i;
903 for (i = 0; i < ARRAY_SIZE(key_table); i++)
904 if (!strcasecmp(key_table[i].name, name))
905 return key_table[i].value;
907 if (strlen(name) == 1 && isprint(*name))
908 return (int) *name;
910 return ERR;
911 }
913 static char *
914 get_key(enum request request)
915 {
916 static char buf[BUFSIZ];
917 static char key_char[] = "'X'";
918 size_t pos = 0;
919 char *sep = "";
920 int i;
922 buf[pos] = 0;
924 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
925 struct keybinding *keybinding = &default_keybindings[i];
926 char *seq = NULL;
927 int key;
929 if (keybinding->request != request)
930 continue;
932 for (key = 0; key < ARRAY_SIZE(key_table); key++)
933 if (key_table[key].value == keybinding->alias)
934 seq = key_table[key].name;
936 if (seq == NULL &&
937 keybinding->alias < 127 &&
938 isprint(keybinding->alias)) {
939 key_char[1] = (char) keybinding->alias;
940 seq = key_char;
941 }
943 if (!seq)
944 seq = "'?'";
946 if (!string_format_from(buf, &pos, "%s%s", sep, seq))
947 return "Too many keybindings!";
948 sep = ", ";
949 }
951 return buf;
952 }
955 /*
956 * User config file handling.
957 */
959 static struct int_map color_map[] = {
960 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
961 COLOR_MAP(DEFAULT),
962 COLOR_MAP(BLACK),
963 COLOR_MAP(BLUE),
964 COLOR_MAP(CYAN),
965 COLOR_MAP(GREEN),
966 COLOR_MAP(MAGENTA),
967 COLOR_MAP(RED),
968 COLOR_MAP(WHITE),
969 COLOR_MAP(YELLOW),
970 };
972 #define set_color(color, name) \
973 set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
975 static struct int_map attr_map[] = {
976 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
977 ATTR_MAP(NORMAL),
978 ATTR_MAP(BLINK),
979 ATTR_MAP(BOLD),
980 ATTR_MAP(DIM),
981 ATTR_MAP(REVERSE),
982 ATTR_MAP(STANDOUT),
983 ATTR_MAP(UNDERLINE),
984 };
986 #define set_attribute(attr, name) \
987 set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
989 static int config_lineno;
990 static bool config_errors;
991 static char *config_msg;
993 /* Wants: object fgcolor bgcolor [attr] */
994 static int
995 option_color_command(int argc, char *argv[])
996 {
997 struct line_info *info;
999 if (argc != 3 && argc != 4) {
1000 config_msg = "Wrong number of arguments given to color command";
1001 return ERR;
1002 }
1004 info = get_line_info(argv[0], strlen(argv[0]));
1005 if (!info) {
1006 config_msg = "Unknown color name";
1007 return ERR;
1008 }
1010 if (set_color(&info->fg, argv[1]) == ERR ||
1011 set_color(&info->bg, argv[2]) == ERR) {
1012 config_msg = "Unknown color";
1013 return ERR;
1014 }
1016 if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
1017 config_msg = "Unknown attribute";
1018 return ERR;
1019 }
1021 return OK;
1022 }
1024 /* Wants: name = value */
1025 static int
1026 option_set_command(int argc, char *argv[])
1027 {
1028 if (argc != 3) {
1029 config_msg = "Wrong number of arguments given to set command";
1030 return ERR;
1031 }
1033 if (strcmp(argv[1], "=")) {
1034 config_msg = "No value assigned";
1035 return ERR;
1036 }
1038 if (!strcmp(argv[0], "show-rev-graph")) {
1039 opt_rev_graph = (!strcmp(argv[2], "1") ||
1040 !strcmp(argv[2], "true") ||
1041 !strcmp(argv[2], "yes"));
1042 return OK;
1043 }
1045 if (!strcmp(argv[0], "line-number-interval")) {
1046 opt_num_interval = atoi(argv[2]);
1047 return OK;
1048 }
1050 if (!strcmp(argv[0], "tab-size")) {
1051 opt_tab_size = atoi(argv[2]);
1052 return OK;
1053 }
1055 if (!strcmp(argv[0], "commit-encoding")) {
1056 char *arg = argv[2];
1057 int delimiter = *arg;
1058 int i;
1060 switch (delimiter) {
1061 case '"':
1062 case '\'':
1063 for (arg++, i = 0; arg[i]; i++)
1064 if (arg[i] == delimiter) {
1065 arg[i] = 0;
1066 break;
1067 }
1068 default:
1069 string_ncopy(opt_encoding, arg, strlen(arg));
1070 return OK;
1071 }
1072 }
1074 config_msg = "Unknown variable name";
1075 return ERR;
1076 }
1078 /* Wants: mode request key */
1079 static int
1080 option_bind_command(int argc, char *argv[])
1081 {
1082 enum request request;
1083 int keymap;
1084 int key;
1086 if (argc != 3) {
1087 config_msg = "Wrong number of arguments given to bind command";
1088 return ERR;
1089 }
1091 if (set_keymap(&keymap, argv[0]) == ERR) {
1092 config_msg = "Unknown key map";
1093 return ERR;
1094 }
1096 key = get_key_value(argv[1]);
1097 if (key == ERR) {
1098 config_msg = "Unknown key";
1099 return ERR;
1100 }
1102 request = get_request(argv[2]);
1103 if (request == REQ_UNKNOWN) {
1104 config_msg = "Unknown request name";
1105 return ERR;
1106 }
1108 add_keybinding(keymap, request, key);
1110 return OK;
1111 }
1113 static int
1114 set_option(char *opt, char *value)
1115 {
1116 char *argv[16];
1117 int valuelen;
1118 int argc = 0;
1120 /* Tokenize */
1121 while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1122 argv[argc++] = value;
1124 value += valuelen;
1125 if (!*value)
1126 break;
1128 *value++ = 0;
1129 while (isspace(*value))
1130 value++;
1131 }
1133 if (!strcmp(opt, "color"))
1134 return option_color_command(argc, argv);
1136 if (!strcmp(opt, "set"))
1137 return option_set_command(argc, argv);
1139 if (!strcmp(opt, "bind"))
1140 return option_bind_command(argc, argv);
1142 config_msg = "Unknown option command";
1143 return ERR;
1144 }
1146 static int
1147 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1148 {
1149 int status = OK;
1151 config_lineno++;
1152 config_msg = "Internal error";
1154 /* Check for comment markers, since read_properties() will
1155 * only ensure opt and value are split at first " \t". */
1156 optlen = strcspn(opt, "#");
1157 if (optlen == 0)
1158 return OK;
1160 if (opt[optlen] != 0) {
1161 config_msg = "No option value";
1162 status = ERR;
1164 } else {
1165 /* Look for comment endings in the value. */
1166 size_t len = strcspn(value, "#");
1168 if (len < valuelen) {
1169 valuelen = len;
1170 value[valuelen] = 0;
1171 }
1173 status = set_option(opt, value);
1174 }
1176 if (status == ERR) {
1177 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1178 config_lineno, (int) optlen, opt, config_msg);
1179 config_errors = TRUE;
1180 }
1182 /* Always keep going if errors are encountered. */
1183 return OK;
1184 }
1186 static int
1187 load_options(void)
1188 {
1189 char *home = getenv("HOME");
1190 char buf[SIZEOF_STR];
1191 FILE *file;
1193 config_lineno = 0;
1194 config_errors = FALSE;
1196 if (!home || !string_format(buf, "%s/.tigrc", home))
1197 return ERR;
1199 /* It's ok that the file doesn't exist. */
1200 file = fopen(buf, "r");
1201 if (!file)
1202 return OK;
1204 if (read_properties(file, " \t", read_option) == ERR ||
1205 config_errors == TRUE)
1206 fprintf(stderr, "Errors while loading %s.\n", buf);
1208 return OK;
1209 }
1212 /*
1213 * The viewer
1214 */
1216 struct view;
1217 struct view_ops;
1219 /* The display array of active views and the index of the current view. */
1220 static struct view *display[2];
1221 static unsigned int current_view;
1223 /* Reading from the prompt? */
1224 static bool input_mode = FALSE;
1226 #define foreach_displayed_view(view, i) \
1227 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1229 #define displayed_views() (display[1] != NULL ? 2 : 1)
1231 /* Current head and commit ID */
1232 static char ref_blob[SIZEOF_REF] = "";
1233 static char ref_commit[SIZEOF_REF] = "HEAD";
1234 static char ref_head[SIZEOF_REF] = "HEAD";
1236 struct view {
1237 const char *name; /* View name */
1238 const char *cmd_fmt; /* Default command line format */
1239 const char *cmd_env; /* Command line set via environment */
1240 const char *id; /* Points to either of ref_{head,commit,blob} */
1242 struct view_ops *ops; /* View operations */
1244 enum keymap keymap; /* What keymap does this view have */
1246 char cmd[SIZEOF_STR]; /* Command buffer */
1247 char ref[SIZEOF_REF]; /* Hovered commit reference */
1248 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
1250 int height, width; /* The width and height of the main window */
1251 WINDOW *win; /* The main window */
1252 WINDOW *title; /* The title window living below the main window */
1254 /* Navigation */
1255 unsigned long offset; /* Offset of the window top */
1256 unsigned long lineno; /* Current line number */
1258 /* Searching */
1259 char grep[SIZEOF_STR]; /* Search string */
1260 regex_t *regex; /* Pre-compiled regex */
1262 /* If non-NULL, points to the view that opened this view. If this view
1263 * is closed tig will switch back to the parent view. */
1264 struct view *parent;
1266 /* Buffering */
1267 unsigned long lines; /* Total number of lines */
1268 struct line *line; /* Line index */
1269 unsigned long line_size;/* Total number of allocated lines */
1270 unsigned int digits; /* Number of digits in the lines member. */
1272 /* Loading */
1273 FILE *pipe;
1274 time_t start_time;
1275 };
1277 struct view_ops {
1278 /* What type of content being displayed. Used in the title bar. */
1279 const char *type;
1280 /* Open and reads in all view content. */
1281 bool (*open)(struct view *view);
1282 /* Read one line; updates view->line. */
1283 bool (*read)(struct view *view, char *data);
1284 /* Draw one line; @lineno must be < view->height. */
1285 bool (*draw)(struct view *view, struct line *line, unsigned int lineno, bool selected);
1286 /* Depending on view handle a special requests. */
1287 enum request (*request)(struct view *view, enum request request, struct line *line);
1288 /* Search for regex in a line. */
1289 bool (*grep)(struct view *view, struct line *line);
1290 /* Select line */
1291 void (*select)(struct view *view, struct line *line);
1292 };
1294 static struct view_ops pager_ops;
1295 static struct view_ops main_ops;
1296 static struct view_ops tree_ops;
1297 static struct view_ops blob_ops;
1298 static struct view_ops help_ops;
1299 static struct view_ops status_ops;
1300 static struct view_ops stage_ops;
1302 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1303 { name, cmd, #env, ref, ops, map}
1305 #define VIEW_(id, name, ops, ref) \
1306 VIEW_STR(name, TIG_##id##_CMD, TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1309 static struct view views[] = {
1310 VIEW_(MAIN, "main", &main_ops, ref_head),
1311 VIEW_(DIFF, "diff", &pager_ops, ref_commit),
1312 VIEW_(LOG, "log", &pager_ops, ref_head),
1313 VIEW_(TREE, "tree", &tree_ops, ref_commit),
1314 VIEW_(BLOB, "blob", &blob_ops, ref_blob),
1315 VIEW_(HELP, "help", &help_ops, ""),
1316 VIEW_(PAGER, "pager", &pager_ops, "stdin"),
1317 VIEW_(STATUS, "status", &status_ops, ""),
1318 VIEW_(STAGE, "stage", &stage_ops, ""),
1319 };
1321 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1323 #define foreach_view(view, i) \
1324 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1326 #define view_is_displayed(view) \
1327 (view == display[0] || view == display[1])
1329 static bool
1330 draw_view_line(struct view *view, unsigned int lineno)
1331 {
1332 struct line *line;
1333 bool selected = (view->offset + lineno == view->lineno);
1334 bool draw_ok;
1336 assert(view_is_displayed(view));
1338 if (view->offset + lineno >= view->lines)
1339 return FALSE;
1341 line = &view->line[view->offset + lineno];
1343 if (selected) {
1344 line->selected = TRUE;
1345 view->ops->select(view, line);
1346 } else if (line->selected) {
1347 line->selected = FALSE;
1348 wmove(view->win, lineno, 0);
1349 wclrtoeol(view->win);
1350 }
1352 scrollok(view->win, FALSE);
1353 draw_ok = view->ops->draw(view, line, lineno, selected);
1354 scrollok(view->win, TRUE);
1356 return draw_ok;
1357 }
1359 static void
1360 redraw_view_from(struct view *view, int lineno)
1361 {
1362 assert(0 <= lineno && lineno < view->height);
1364 for (; lineno < view->height; lineno++) {
1365 if (!draw_view_line(view, lineno))
1366 break;
1367 }
1369 redrawwin(view->win);
1370 if (input_mode)
1371 wnoutrefresh(view->win);
1372 else
1373 wrefresh(view->win);
1374 }
1376 static void
1377 redraw_view(struct view *view)
1378 {
1379 wclear(view->win);
1380 redraw_view_from(view, 0);
1381 }
1384 static void
1385 update_view_title(struct view *view)
1386 {
1387 char buf[SIZEOF_STR];
1388 char state[SIZEOF_STR];
1389 size_t bufpos = 0, statelen = 0;
1391 assert(view_is_displayed(view));
1393 if (view != VIEW(REQ_VIEW_STATUS) && (view->lines || view->pipe)) {
1394 unsigned int view_lines = view->offset + view->height;
1395 unsigned int lines = view->lines
1396 ? MIN(view_lines, view->lines) * 100 / view->lines
1397 : 0;
1399 string_format_from(state, &statelen, "- %s %d of %d (%d%%)",
1400 view->ops->type,
1401 view->lineno + 1,
1402 view->lines,
1403 lines);
1405 if (view->pipe) {
1406 time_t secs = time(NULL) - view->start_time;
1408 /* Three git seconds are a long time ... */
1409 if (secs > 2)
1410 string_format_from(state, &statelen, " %lds", secs);
1411 }
1412 }
1414 string_format_from(buf, &bufpos, "[%s]", view->name);
1415 if (*view->ref && bufpos < view->width) {
1416 size_t refsize = strlen(view->ref);
1417 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
1419 if (minsize < view->width)
1420 refsize = view->width - minsize + 7;
1421 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
1422 }
1424 if (statelen && bufpos < view->width) {
1425 string_format_from(buf, &bufpos, " %s", state);
1426 }
1428 if (view == display[current_view])
1429 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1430 else
1431 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1433 mvwaddnstr(view->title, 0, 0, buf, bufpos);
1434 wclrtoeol(view->title);
1435 wmove(view->title, 0, view->width - 1);
1437 if (input_mode)
1438 wnoutrefresh(view->title);
1439 else
1440 wrefresh(view->title);
1441 }
1443 static void
1444 resize_display(void)
1445 {
1446 int offset, i;
1447 struct view *base = display[0];
1448 struct view *view = display[1] ? display[1] : display[0];
1450 /* Setup window dimensions */
1452 getmaxyx(stdscr, base->height, base->width);
1454 /* Make room for the status window. */
1455 base->height -= 1;
1457 if (view != base) {
1458 /* Horizontal split. */
1459 view->width = base->width;
1460 view->height = SCALE_SPLIT_VIEW(base->height);
1461 base->height -= view->height;
1463 /* Make room for the title bar. */
1464 view->height -= 1;
1465 }
1467 /* Make room for the title bar. */
1468 base->height -= 1;
1470 offset = 0;
1472 foreach_displayed_view (view, i) {
1473 if (!view->win) {
1474 view->win = newwin(view->height, 0, offset, 0);
1475 if (!view->win)
1476 die("Failed to create %s view", view->name);
1478 scrollok(view->win, TRUE);
1480 view->title = newwin(1, 0, offset + view->height, 0);
1481 if (!view->title)
1482 die("Failed to create title window");
1484 } else {
1485 wresize(view->win, view->height, view->width);
1486 mvwin(view->win, offset, 0);
1487 mvwin(view->title, offset + view->height, 0);
1488 }
1490 offset += view->height + 1;
1491 }
1492 }
1494 static void
1495 redraw_display(void)
1496 {
1497 struct view *view;
1498 int i;
1500 foreach_displayed_view (view, i) {
1501 redraw_view(view);
1502 update_view_title(view);
1503 }
1504 }
1506 static void
1507 update_display_cursor(struct view *view)
1508 {
1509 /* Move the cursor to the right-most column of the cursor line.
1510 *
1511 * XXX: This could turn out to be a bit expensive, but it ensures that
1512 * the cursor does not jump around. */
1513 if (view->lines) {
1514 wmove(view->win, view->lineno - view->offset, view->width - 1);
1515 wrefresh(view->win);
1516 }
1517 }
1519 /*
1520 * Navigation
1521 */
1523 /* Scrolling backend */
1524 static void
1525 do_scroll_view(struct view *view, int lines)
1526 {
1527 bool redraw_current_line = FALSE;
1529 /* The rendering expects the new offset. */
1530 view->offset += lines;
1532 assert(0 <= view->offset && view->offset < view->lines);
1533 assert(lines);
1535 /* Move current line into the view. */
1536 if (view->lineno < view->offset) {
1537 view->lineno = view->offset;
1538 redraw_current_line = TRUE;
1539 } else if (view->lineno >= view->offset + view->height) {
1540 view->lineno = view->offset + view->height - 1;
1541 redraw_current_line = TRUE;
1542 }
1544 assert(view->offset <= view->lineno && view->lineno < view->lines);
1546 /* Redraw the whole screen if scrolling is pointless. */
1547 if (view->height < ABS(lines)) {
1548 redraw_view(view);
1550 } else {
1551 int line = lines > 0 ? view->height - lines : 0;
1552 int end = line + ABS(lines);
1554 wscrl(view->win, lines);
1556 for (; line < end; line++) {
1557 if (!draw_view_line(view, line))
1558 break;
1559 }
1561 if (redraw_current_line)
1562 draw_view_line(view, view->lineno - view->offset);
1563 }
1565 redrawwin(view->win);
1566 wrefresh(view->win);
1567 report("");
1568 }
1570 /* Scroll frontend */
1571 static void
1572 scroll_view(struct view *view, enum request request)
1573 {
1574 int lines = 1;
1576 assert(view_is_displayed(view));
1578 switch (request) {
1579 case REQ_SCROLL_PAGE_DOWN:
1580 lines = view->height;
1581 case REQ_SCROLL_LINE_DOWN:
1582 if (view->offset + lines > view->lines)
1583 lines = view->lines - view->offset;
1585 if (lines == 0 || view->offset + view->height >= view->lines) {
1586 report("Cannot scroll beyond the last line");
1587 return;
1588 }
1589 break;
1591 case REQ_SCROLL_PAGE_UP:
1592 lines = view->height;
1593 case REQ_SCROLL_LINE_UP:
1594 if (lines > view->offset)
1595 lines = view->offset;
1597 if (lines == 0) {
1598 report("Cannot scroll beyond the first line");
1599 return;
1600 }
1602 lines = -lines;
1603 break;
1605 default:
1606 die("request %d not handled in switch", request);
1607 }
1609 do_scroll_view(view, lines);
1610 }
1612 /* Cursor moving */
1613 static void
1614 move_view(struct view *view, enum request request)
1615 {
1616 int scroll_steps = 0;
1617 int steps;
1619 switch (request) {
1620 case REQ_MOVE_FIRST_LINE:
1621 steps = -view->lineno;
1622 break;
1624 case REQ_MOVE_LAST_LINE:
1625 steps = view->lines - view->lineno - 1;
1626 break;
1628 case REQ_MOVE_PAGE_UP:
1629 steps = view->height > view->lineno
1630 ? -view->lineno : -view->height;
1631 break;
1633 case REQ_MOVE_PAGE_DOWN:
1634 steps = view->lineno + view->height >= view->lines
1635 ? view->lines - view->lineno - 1 : view->height;
1636 break;
1638 case REQ_MOVE_UP:
1639 steps = -1;
1640 break;
1642 case REQ_MOVE_DOWN:
1643 steps = 1;
1644 break;
1646 default:
1647 die("request %d not handled in switch", request);
1648 }
1650 if (steps <= 0 && view->lineno == 0) {
1651 report("Cannot move beyond the first line");
1652 return;
1654 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1655 report("Cannot move beyond the last line");
1656 return;
1657 }
1659 /* Move the current line */
1660 view->lineno += steps;
1661 assert(0 <= view->lineno && view->lineno < view->lines);
1663 /* Check whether the view needs to be scrolled */
1664 if (view->lineno < view->offset ||
1665 view->lineno >= view->offset + view->height) {
1666 scroll_steps = steps;
1667 if (steps < 0 && -steps > view->offset) {
1668 scroll_steps = -view->offset;
1670 } else if (steps > 0) {
1671 if (view->lineno == view->lines - 1 &&
1672 view->lines > view->height) {
1673 scroll_steps = view->lines - view->offset - 1;
1674 if (scroll_steps >= view->height)
1675 scroll_steps -= view->height - 1;
1676 }
1677 }
1678 }
1680 if (!view_is_displayed(view)) {
1681 view->offset += scroll_steps;
1682 assert(0 <= view->offset && view->offset < view->lines);
1683 view->ops->select(view, &view->line[view->lineno]);
1684 return;
1685 }
1687 /* Repaint the old "current" line if we be scrolling */
1688 if (ABS(steps) < view->height)
1689 draw_view_line(view, view->lineno - steps - view->offset);
1691 if (scroll_steps) {
1692 do_scroll_view(view, scroll_steps);
1693 return;
1694 }
1696 /* Draw the current line */
1697 draw_view_line(view, view->lineno - view->offset);
1699 redrawwin(view->win);
1700 wrefresh(view->win);
1701 report("");
1702 }
1705 /*
1706 * Searching
1707 */
1709 static void search_view(struct view *view, enum request request);
1711 static bool
1712 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1713 {
1714 assert(view_is_displayed(view));
1716 if (!view->ops->grep(view, line))
1717 return FALSE;
1719 if (lineno - view->offset >= view->height) {
1720 view->offset = lineno;
1721 view->lineno = lineno;
1722 redraw_view(view);
1724 } else {
1725 unsigned long old_lineno = view->lineno - view->offset;
1727 view->lineno = lineno;
1728 draw_view_line(view, old_lineno);
1730 draw_view_line(view, view->lineno - view->offset);
1731 redrawwin(view->win);
1732 wrefresh(view->win);
1733 }
1735 report("Line %ld matches '%s'", lineno + 1, view->grep);
1736 return TRUE;
1737 }
1739 static void
1740 find_next(struct view *view, enum request request)
1741 {
1742 unsigned long lineno = view->lineno;
1743 int direction;
1745 if (!*view->grep) {
1746 if (!*opt_search)
1747 report("No previous search");
1748 else
1749 search_view(view, request);
1750 return;
1751 }
1753 switch (request) {
1754 case REQ_SEARCH:
1755 case REQ_FIND_NEXT:
1756 direction = 1;
1757 break;
1759 case REQ_SEARCH_BACK:
1760 case REQ_FIND_PREV:
1761 direction = -1;
1762 break;
1764 default:
1765 return;
1766 }
1768 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1769 lineno += direction;
1771 /* Note, lineno is unsigned long so will wrap around in which case it
1772 * will become bigger than view->lines. */
1773 for (; lineno < view->lines; lineno += direction) {
1774 struct line *line = &view->line[lineno];
1776 if (find_next_line(view, lineno, line))
1777 return;
1778 }
1780 report("No match found for '%s'", view->grep);
1781 }
1783 static void
1784 search_view(struct view *view, enum request request)
1785 {
1786 int regex_err;
1788 if (view->regex) {
1789 regfree(view->regex);
1790 *view->grep = 0;
1791 } else {
1792 view->regex = calloc(1, sizeof(*view->regex));
1793 if (!view->regex)
1794 return;
1795 }
1797 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
1798 if (regex_err != 0) {
1799 char buf[SIZEOF_STR] = "unknown error";
1801 regerror(regex_err, view->regex, buf, sizeof(buf));
1802 report("Search failed: %s", buf);
1803 return;
1804 }
1806 string_copy(view->grep, opt_search);
1808 find_next(view, request);
1809 }
1811 /*
1812 * Incremental updating
1813 */
1815 static void
1816 end_update(struct view *view)
1817 {
1818 if (!view->pipe)
1819 return;
1820 set_nonblocking_input(FALSE);
1821 if (view->pipe == stdin)
1822 fclose(view->pipe);
1823 else
1824 pclose(view->pipe);
1825 view->pipe = NULL;
1826 }
1828 static bool
1829 begin_update(struct view *view)
1830 {
1831 if (view->pipe)
1832 end_update(view);
1834 if (opt_cmd[0]) {
1835 string_copy(view->cmd, opt_cmd);
1836 opt_cmd[0] = 0;
1837 /* When running random commands, initially show the
1838 * command in the title. However, it maybe later be
1839 * overwritten if a commit line is selected. */
1840 if (view == VIEW(REQ_VIEW_PAGER))
1841 string_copy(view->ref, view->cmd);
1842 else
1843 view->ref[0] = 0;
1845 } else if (view == VIEW(REQ_VIEW_TREE)) {
1846 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1847 char path[SIZEOF_STR];
1849 if (strcmp(view->vid, view->id))
1850 opt_path[0] = path[0] = 0;
1851 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
1852 return FALSE;
1854 if (!string_format(view->cmd, format, view->id, path))
1855 return FALSE;
1857 } else {
1858 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1859 const char *id = view->id;
1861 if (!string_format(view->cmd, format, id, id, id, id, id))
1862 return FALSE;
1864 /* Put the current ref_* value to the view title ref
1865 * member. This is needed by the blob view. Most other
1866 * views sets it automatically after loading because the
1867 * first line is a commit line. */
1868 string_copy_rev(view->ref, view->id);
1869 }
1871 /* Special case for the pager view. */
1872 if (opt_pipe) {
1873 view->pipe = opt_pipe;
1874 opt_pipe = NULL;
1875 } else {
1876 view->pipe = popen(view->cmd, "r");
1877 }
1879 if (!view->pipe)
1880 return FALSE;
1882 set_nonblocking_input(TRUE);
1884 view->offset = 0;
1885 view->lines = 0;
1886 view->lineno = 0;
1887 string_copy_rev(view->vid, view->id);
1889 if (view->line) {
1890 int i;
1892 for (i = 0; i < view->lines; i++)
1893 if (view->line[i].data)
1894 free(view->line[i].data);
1896 free(view->line);
1897 view->line = NULL;
1898 }
1900 view->start_time = time(NULL);
1902 return TRUE;
1903 }
1905 static struct line *
1906 realloc_lines(struct view *view, size_t line_size)
1907 {
1908 struct line *tmp = realloc(view->line, sizeof(*view->line) * line_size);
1910 if (!tmp)
1911 return NULL;
1913 view->line = tmp;
1914 view->line_size = line_size;
1915 return view->line;
1916 }
1918 static bool
1919 update_view(struct view *view)
1920 {
1921 char in_buffer[BUFSIZ];
1922 char out_buffer[BUFSIZ * 2];
1923 char *line;
1924 /* The number of lines to read. If too low it will cause too much
1925 * redrawing (and possible flickering), if too high responsiveness
1926 * will suffer. */
1927 unsigned long lines = view->height;
1928 int redraw_from = -1;
1930 if (!view->pipe)
1931 return TRUE;
1933 /* Only redraw if lines are visible. */
1934 if (view->offset + view->height >= view->lines)
1935 redraw_from = view->lines - view->offset;
1937 /* FIXME: This is probably not perfect for backgrounded views. */
1938 if (!realloc_lines(view, view->lines + lines))
1939 goto alloc_error;
1941 while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
1942 size_t linelen = strlen(line);
1944 if (linelen)
1945 line[linelen - 1] = 0;
1947 if (opt_iconv != ICONV_NONE) {
1948 ICONV_CONST char *inbuf = line;
1949 size_t inlen = linelen;
1951 char *outbuf = out_buffer;
1952 size_t outlen = sizeof(out_buffer);
1954 size_t ret;
1956 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
1957 if (ret != (size_t) -1) {
1958 line = out_buffer;
1959 linelen = strlen(out_buffer);
1960 }
1961 }
1963 if (!view->ops->read(view, line))
1964 goto alloc_error;
1966 if (lines-- == 1)
1967 break;
1968 }
1970 {
1971 int digits;
1973 lines = view->lines;
1974 for (digits = 0; lines; digits++)
1975 lines /= 10;
1977 /* Keep the displayed view in sync with line number scaling. */
1978 if (digits != view->digits) {
1979 view->digits = digits;
1980 redraw_from = 0;
1981 }
1982 }
1984 if (!view_is_displayed(view))
1985 goto check_pipe;
1987 if (view == VIEW(REQ_VIEW_TREE)) {
1988 /* Clear the view and redraw everything since the tree sorting
1989 * might have rearranged things. */
1990 redraw_view(view);
1992 } else if (redraw_from >= 0) {
1993 /* If this is an incremental update, redraw the previous line
1994 * since for commits some members could have changed when
1995 * loading the main view. */
1996 if (redraw_from > 0)
1997 redraw_from--;
1999 /* Since revision graph visualization requires knowledge
2000 * about the parent commit, it causes a further one-off
2001 * needed to be redrawn for incremental updates. */
2002 if (redraw_from > 0 && opt_rev_graph)
2003 redraw_from--;
2005 /* Incrementally draw avoids flickering. */
2006 redraw_view_from(view, redraw_from);
2007 }
2009 /* Update the title _after_ the redraw so that if the redraw picks up a
2010 * commit reference in view->ref it'll be available here. */
2011 update_view_title(view);
2013 check_pipe:
2014 if (ferror(view->pipe)) {
2015 report("Failed to read: %s", strerror(errno));
2016 goto end;
2018 } else if (feof(view->pipe)) {
2019 report("");
2020 goto end;
2021 }
2023 return TRUE;
2025 alloc_error:
2026 report("Allocation failure");
2028 end:
2029 view->ops->read(view, NULL);
2030 end_update(view);
2031 return FALSE;
2032 }
2034 static struct line *
2035 add_line_data(struct view *view, void *data, enum line_type type)
2036 {
2037 struct line *line = &view->line[view->lines++];
2039 memset(line, 0, sizeof(*line));
2040 line->type = type;
2041 line->data = data;
2043 return line;
2044 }
2046 static struct line *
2047 add_line_text(struct view *view, char *data, enum line_type type)
2048 {
2049 if (data)
2050 data = strdup(data);
2052 return data ? add_line_data(view, data, type) : NULL;
2053 }
2056 /*
2057 * View opening
2058 */
2060 enum open_flags {
2061 OPEN_DEFAULT = 0, /* Use default view switching. */
2062 OPEN_SPLIT = 1, /* Split current view. */
2063 OPEN_BACKGROUNDED = 2, /* Backgrounded. */
2064 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
2065 };
2067 static void
2068 open_view(struct view *prev, enum request request, enum open_flags flags)
2069 {
2070 bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2071 bool split = !!(flags & OPEN_SPLIT);
2072 bool reload = !!(flags & OPEN_RELOAD);
2073 struct view *view = VIEW(request);
2074 int nviews = displayed_views();
2075 struct view *base_view = display[0];
2077 if (view == prev && nviews == 1 && !reload) {
2078 report("Already in %s view", view->name);
2079 return;
2080 }
2082 if (view->ops->open) {
2083 if (!view->ops->open(view)) {
2084 report("Failed to load %s view", view->name);
2085 return;
2086 }
2088 } else if ((reload || strcmp(view->vid, view->id)) &&
2089 !begin_update(view)) {
2090 report("Failed to load %s view", view->name);
2091 return;
2092 }
2094 if (split) {
2095 display[1] = view;
2096 if (!backgrounded)
2097 current_view = 1;
2098 } else {
2099 /* Maximize the current view. */
2100 memset(display, 0, sizeof(display));
2101 current_view = 0;
2102 display[current_view] = view;
2103 }
2105 /* Resize the view when switching between split- and full-screen,
2106 * or when switching between two different full-screen views. */
2107 if (nviews != displayed_views() ||
2108 (nviews == 1 && base_view != display[0]))
2109 resize_display();
2111 if (split && prev->lineno - prev->offset >= prev->height) {
2112 /* Take the title line into account. */
2113 int lines = prev->lineno - prev->offset - prev->height + 1;
2115 /* Scroll the view that was split if the current line is
2116 * outside the new limited view. */
2117 do_scroll_view(prev, lines);
2118 }
2120 if (prev && view != prev) {
2121 if (split && !backgrounded) {
2122 /* "Blur" the previous view. */
2123 update_view_title(prev);
2124 }
2126 view->parent = prev;
2127 }
2129 if (view->pipe && view->lines == 0) {
2130 /* Clear the old view and let the incremental updating refill
2131 * the screen. */
2132 wclear(view->win);
2133 report("");
2134 } else {
2135 redraw_view(view);
2136 report("");
2137 }
2139 /* If the view is backgrounded the above calls to report()
2140 * won't redraw the view title. */
2141 if (backgrounded)
2142 update_view_title(view);
2143 }
2145 static void
2146 open_editor(struct view *view, bool from_root, char *file)
2147 {
2148 char cmd[SIZEOF_STR];
2149 char file_sq[SIZEOF_STR];
2150 char *editor;
2151 char *prefix = from_root ? opt_cdup : "";
2153 editor = getenv("GIT_EDITOR");
2154 if (!editor && *opt_editor)
2155 editor = opt_editor;
2156 if (!editor)
2157 editor = getenv("VISUAL");
2158 if (!editor)
2159 editor = getenv("EDITOR");
2160 if (!editor)
2161 editor = "vi";
2163 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2164 string_format(cmd, "%s %s%s", editor, prefix, file_sq)) {
2165 def_prog_mode(); /* save current tty modes */
2166 endwin(); /* restore original tty modes */
2167 system(cmd);
2168 reset_prog_mode();
2169 redraw_display();
2170 }
2171 }
2173 /*
2174 * User request switch noodle
2175 */
2177 static int
2178 view_driver(struct view *view, enum request request)
2179 {
2180 int i;
2182 if (request == REQ_NONE) {
2183 doupdate();
2184 return TRUE;
2185 }
2187 if (view && view->lines) {
2188 request = view->ops->request(view, request, &view->line[view->lineno]);
2189 if (request == REQ_NONE)
2190 return TRUE;
2191 }
2193 switch (request) {
2194 case REQ_MOVE_UP:
2195 case REQ_MOVE_DOWN:
2196 case REQ_MOVE_PAGE_UP:
2197 case REQ_MOVE_PAGE_DOWN:
2198 case REQ_MOVE_FIRST_LINE:
2199 case REQ_MOVE_LAST_LINE:
2200 move_view(view, request);
2201 break;
2203 case REQ_SCROLL_LINE_DOWN:
2204 case REQ_SCROLL_LINE_UP:
2205 case REQ_SCROLL_PAGE_DOWN:
2206 case REQ_SCROLL_PAGE_UP:
2207 scroll_view(view, request);
2208 break;
2210 case REQ_VIEW_BLOB:
2211 if (!ref_blob[0]) {
2212 report("No file chosen, press %s to open tree view",
2213 get_key(REQ_VIEW_TREE));
2214 break;
2215 }
2216 open_view(view, request, OPEN_DEFAULT);
2217 break;
2219 case REQ_VIEW_PAGER:
2220 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2221 report("No pager content, press %s to run command from prompt",
2222 get_key(REQ_PROMPT));
2223 break;
2224 }
2225 open_view(view, request, OPEN_DEFAULT);
2226 break;
2228 case REQ_VIEW_STAGE:
2229 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2230 report("No stage content, press %s to open the status view and choose file",
2231 get_key(REQ_VIEW_STATUS));
2232 break;
2233 }
2234 open_view(view, request, OPEN_DEFAULT);
2235 break;
2237 case REQ_VIEW_MAIN:
2238 case REQ_VIEW_DIFF:
2239 case REQ_VIEW_LOG:
2240 case REQ_VIEW_TREE:
2241 case REQ_VIEW_HELP:
2242 case REQ_VIEW_STATUS:
2243 open_view(view, request, OPEN_DEFAULT);
2244 break;
2246 case REQ_NEXT:
2247 case REQ_PREVIOUS:
2248 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2250 if ((view == VIEW(REQ_VIEW_DIFF) &&
2251 view->parent == VIEW(REQ_VIEW_MAIN)) ||
2252 (view == VIEW(REQ_VIEW_STAGE) &&
2253 view->parent == VIEW(REQ_VIEW_STATUS)) ||
2254 (view == VIEW(REQ_VIEW_BLOB) &&
2255 view->parent == VIEW(REQ_VIEW_TREE))) {
2256 int line;
2258 view = view->parent;
2259 line = view->lineno;
2260 move_view(view, request);
2261 if (view_is_displayed(view))
2262 update_view_title(view);
2263 if (line != view->lineno)
2264 view->ops->request(view, REQ_ENTER,
2265 &view->line[view->lineno]);
2267 } else {
2268 move_view(view, request);
2269 }
2270 break;
2272 case REQ_VIEW_NEXT:
2273 {
2274 int nviews = displayed_views();
2275 int next_view = (current_view + 1) % nviews;
2277 if (next_view == current_view) {
2278 report("Only one view is displayed");
2279 break;
2280 }
2282 current_view = next_view;
2283 /* Blur out the title of the previous view. */
2284 update_view_title(view);
2285 report("");
2286 break;
2287 }
2288 case REQ_TOGGLE_LINENO:
2289 opt_line_number = !opt_line_number;
2290 redraw_display();
2291 break;
2293 case REQ_TOGGLE_REV_GRAPH:
2294 opt_rev_graph = !opt_rev_graph;
2295 redraw_display();
2296 break;
2298 case REQ_PROMPT:
2299 /* Always reload^Wrerun commands from the prompt. */
2300 open_view(view, opt_request, OPEN_RELOAD);
2301 break;
2303 case REQ_SEARCH:
2304 case REQ_SEARCH_BACK:
2305 search_view(view, request);
2306 break;
2308 case REQ_FIND_NEXT:
2309 case REQ_FIND_PREV:
2310 find_next(view, request);
2311 break;
2313 case REQ_STOP_LOADING:
2314 for (i = 0; i < ARRAY_SIZE(views); i++) {
2315 view = &views[i];
2316 if (view->pipe)
2317 report("Stopped loading the %s view", view->name),
2318 end_update(view);
2319 }
2320 break;
2322 case REQ_SHOW_VERSION:
2323 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2324 return TRUE;
2326 case REQ_SCREEN_RESIZE:
2327 resize_display();
2328 /* Fall-through */
2329 case REQ_SCREEN_REDRAW:
2330 redraw_display();
2331 break;
2333 case REQ_EDIT:
2334 report("Nothing to edit");
2335 break;
2337 case REQ_CHERRY_PICK:
2338 report("Nothing to cherry-pick");
2339 break;
2341 case REQ_ENTER:
2342 report("Nothing to enter");
2343 break;
2346 case REQ_VIEW_CLOSE:
2347 /* XXX: Mark closed views by letting view->parent point to the
2348 * view itself. Parents to closed view should never be
2349 * followed. */
2350 if (view->parent &&
2351 view->parent->parent != view->parent) {
2352 memset(display, 0, sizeof(display));
2353 current_view = 0;
2354 display[current_view] = view->parent;
2355 view->parent = view;
2356 resize_display();
2357 redraw_display();
2358 break;
2359 }
2360 /* Fall-through */
2361 case REQ_QUIT:
2362 return FALSE;
2364 default:
2365 /* An unknown key will show most commonly used commands. */
2366 report("Unknown key, press 'h' for help");
2367 return TRUE;
2368 }
2370 return TRUE;
2371 }
2374 /*
2375 * Pager backend
2376 */
2378 static bool
2379 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2380 {
2381 char *text = line->data;
2382 enum line_type type = line->type;
2383 int textlen = strlen(text);
2384 int attr;
2386 wmove(view->win, lineno, 0);
2388 if (selected) {
2389 type = LINE_CURSOR;
2390 wchgat(view->win, -1, 0, type, NULL);
2391 }
2393 attr = get_line_attr(type);
2394 wattrset(view->win, attr);
2396 if (opt_line_number || opt_tab_size < TABSIZE) {
2397 static char spaces[] = " ";
2398 int col_offset = 0, col = 0;
2400 if (opt_line_number) {
2401 unsigned long real_lineno = view->offset + lineno + 1;
2403 if (real_lineno == 1 ||
2404 (real_lineno % opt_num_interval) == 0) {
2405 wprintw(view->win, "%.*d", view->digits, real_lineno);
2407 } else {
2408 waddnstr(view->win, spaces,
2409 MIN(view->digits, STRING_SIZE(spaces)));
2410 }
2411 waddstr(view->win, ": ");
2412 col_offset = view->digits + 2;
2413 }
2415 while (text && col_offset + col < view->width) {
2416 int cols_max = view->width - col_offset - col;
2417 char *pos = text;
2418 int cols;
2420 if (*text == '\t') {
2421 text++;
2422 assert(sizeof(spaces) > TABSIZE);
2423 pos = spaces;
2424 cols = opt_tab_size - (col % opt_tab_size);
2426 } else {
2427 text = strchr(text, '\t');
2428 cols = line ? text - pos : strlen(pos);
2429 }
2431 waddnstr(view->win, pos, MIN(cols, cols_max));
2432 col += cols;
2433 }
2435 } else {
2436 int col = 0, pos = 0;
2438 for (; pos < textlen && col < view->width; pos++, col++)
2439 if (text[pos] == '\t')
2440 col += TABSIZE - (col % TABSIZE) - 1;
2442 waddnstr(view->win, text, pos);
2443 }
2445 return TRUE;
2446 }
2448 static bool
2449 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2450 {
2451 char refbuf[SIZEOF_STR];
2452 char *ref = NULL;
2453 FILE *pipe;
2455 if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2456 return TRUE;
2458 pipe = popen(refbuf, "r");
2459 if (!pipe)
2460 return TRUE;
2462 if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2463 ref = chomp_string(ref);
2464 pclose(pipe);
2466 if (!ref || !*ref)
2467 return TRUE;
2469 /* This is the only fatal call, since it can "corrupt" the buffer. */
2470 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2471 return FALSE;
2473 return TRUE;
2474 }
2476 static void
2477 add_pager_refs(struct view *view, struct line *line)
2478 {
2479 char buf[SIZEOF_STR];
2480 char *commit_id = line->data + STRING_SIZE("commit ");
2481 struct ref **refs;
2482 size_t bufpos = 0, refpos = 0;
2483 const char *sep = "Refs: ";
2484 bool is_tag = FALSE;
2486 assert(line->type == LINE_COMMIT);
2488 refs = get_refs(commit_id);
2489 if (!refs) {
2490 if (view == VIEW(REQ_VIEW_DIFF))
2491 goto try_add_describe_ref;
2492 return;
2493 }
2495 do {
2496 struct ref *ref = refs[refpos];
2497 char *fmt = ref->tag ? "%s[%s]" :
2498 ref->remote ? "%s<%s>" : "%s%s";
2500 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2501 return;
2502 sep = ", ";
2503 if (ref->tag)
2504 is_tag = TRUE;
2505 } while (refs[refpos++]->next);
2507 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2508 try_add_describe_ref:
2509 /* Add <tag>-g<commit_id> "fake" reference. */
2510 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2511 return;
2512 }
2514 if (bufpos == 0)
2515 return;
2517 if (!realloc_lines(view, view->line_size + 1))
2518 return;
2520 add_line_text(view, buf, LINE_PP_REFS);
2521 }
2523 static bool
2524 pager_read(struct view *view, char *data)
2525 {
2526 struct line *line;
2528 if (!data)
2529 return TRUE;
2531 line = add_line_text(view, data, get_line_type(data));
2532 if (!line)
2533 return FALSE;
2535 if (line->type == LINE_COMMIT &&
2536 (view == VIEW(REQ_VIEW_DIFF) ||
2537 view == VIEW(REQ_VIEW_LOG)))
2538 add_pager_refs(view, line);
2540 return TRUE;
2541 }
2543 static enum request
2544 pager_request(struct view *view, enum request request, struct line *line)
2545 {
2546 int split = 0;
2548 if (request != REQ_ENTER)
2549 return request;
2551 if (line->type == LINE_COMMIT &&
2552 (view == VIEW(REQ_VIEW_LOG) ||
2553 view == VIEW(REQ_VIEW_PAGER))) {
2554 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2555 split = 1;
2556 }
2558 /* Always scroll the view even if it was split. That way
2559 * you can use Enter to scroll through the log view and
2560 * split open each commit diff. */
2561 scroll_view(view, REQ_SCROLL_LINE_DOWN);
2563 /* FIXME: A minor workaround. Scrolling the view will call report("")
2564 * but if we are scrolling a non-current view this won't properly
2565 * update the view title. */
2566 if (split)
2567 update_view_title(view);
2569 return REQ_NONE;
2570 }
2572 static bool
2573 pager_grep(struct view *view, struct line *line)
2574 {
2575 regmatch_t pmatch;
2576 char *text = line->data;
2578 if (!*text)
2579 return FALSE;
2581 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2582 return FALSE;
2584 return TRUE;
2585 }
2587 static void
2588 pager_select(struct view *view, struct line *line)
2589 {
2590 if (line->type == LINE_COMMIT) {
2591 char *text = line->data + STRING_SIZE("commit ");
2593 if (view != VIEW(REQ_VIEW_PAGER))
2594 string_copy_rev(view->ref, text);
2595 string_copy_rev(ref_commit, text);
2596 }
2597 }
2599 static struct view_ops pager_ops = {
2600 "line",
2601 NULL,
2602 pager_read,
2603 pager_draw,
2604 pager_request,
2605 pager_grep,
2606 pager_select,
2607 };
2610 /*
2611 * Help backend
2612 */
2614 static bool
2615 help_open(struct view *view)
2616 {
2617 char buf[BUFSIZ];
2618 int lines = ARRAY_SIZE(req_info) + 2;
2619 int i;
2621 if (view->lines > 0)
2622 return TRUE;
2624 for (i = 0; i < ARRAY_SIZE(req_info); i++)
2625 if (!req_info[i].request)
2626 lines++;
2628 view->line = calloc(lines, sizeof(*view->line));
2629 if (!view->line)
2630 return FALSE;
2632 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
2634 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
2635 char *key;
2637 if (!req_info[i].request) {
2638 add_line_text(view, "", LINE_DEFAULT);
2639 add_line_text(view, req_info[i].help, LINE_DEFAULT);
2640 continue;
2641 }
2643 key = get_key(req_info[i].request);
2644 if (!string_format(buf, " %-25s %s", key, req_info[i].help))
2645 continue;
2647 add_line_text(view, buf, LINE_DEFAULT);
2648 }
2650 return TRUE;
2651 }
2653 static struct view_ops help_ops = {
2654 "line",
2655 help_open,
2656 NULL,
2657 pager_draw,
2658 pager_request,
2659 pager_grep,
2660 pager_select,
2661 };
2664 /*
2665 * Tree backend
2666 */
2668 struct tree_stack_entry {
2669 struct tree_stack_entry *prev; /* Entry below this in the stack */
2670 unsigned long lineno; /* Line number to restore */
2671 char *name; /* Position of name in opt_path */
2672 };
2674 /* The top of the path stack. */
2675 static struct tree_stack_entry *tree_stack = NULL;
2676 unsigned long tree_lineno = 0;
2678 static void
2679 pop_tree_stack_entry(void)
2680 {
2681 struct tree_stack_entry *entry = tree_stack;
2683 tree_lineno = entry->lineno;
2684 entry->name[0] = 0;
2685 tree_stack = entry->prev;
2686 free(entry);
2687 }
2689 static void
2690 push_tree_stack_entry(char *name, unsigned long lineno)
2691 {
2692 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
2693 size_t pathlen = strlen(opt_path);
2695 if (!entry)
2696 return;
2698 entry->prev = tree_stack;
2699 entry->name = opt_path + pathlen;
2700 tree_stack = entry;
2702 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
2703 pop_tree_stack_entry();
2704 return;
2705 }
2707 /* Move the current line to the first tree entry. */
2708 tree_lineno = 1;
2709 entry->lineno = lineno;
2710 }
2712 /* Parse output from git-ls-tree(1):
2713 *
2714 * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
2715 * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
2716 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
2717 * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
2718 */
2720 #define SIZEOF_TREE_ATTR \
2721 STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
2723 #define TREE_UP_FORMAT "040000 tree %s\t.."
2725 static int
2726 tree_compare_entry(enum line_type type1, char *name1,
2727 enum line_type type2, char *name2)
2728 {
2729 if (type1 != type2) {
2730 if (type1 == LINE_TREE_DIR)
2731 return -1;
2732 return 1;
2733 }
2735 return strcmp(name1, name2);
2736 }
2738 static bool
2739 tree_read(struct view *view, char *text)
2740 {
2741 size_t textlen = text ? strlen(text) : 0;
2742 char buf[SIZEOF_STR];
2743 unsigned long pos;
2744 enum line_type type;
2745 bool first_read = view->lines == 0;
2747 if (textlen <= SIZEOF_TREE_ATTR)
2748 return FALSE;
2750 type = text[STRING_SIZE("100644 ")] == 't'
2751 ? LINE_TREE_DIR : LINE_TREE_FILE;
2753 if (first_read) {
2754 /* Add path info line */
2755 if (!string_format(buf, "Directory path /%s", opt_path) ||
2756 !realloc_lines(view, view->line_size + 1) ||
2757 !add_line_text(view, buf, LINE_DEFAULT))
2758 return FALSE;
2760 /* Insert "link" to parent directory. */
2761 if (*opt_path) {
2762 if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
2763 !realloc_lines(view, view->line_size + 1) ||
2764 !add_line_text(view, buf, LINE_TREE_DIR))
2765 return FALSE;
2766 }
2767 }
2769 /* Strip the path part ... */
2770 if (*opt_path) {
2771 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
2772 size_t striplen = strlen(opt_path);
2773 char *path = text + SIZEOF_TREE_ATTR;
2775 if (pathlen > striplen)
2776 memmove(path, path + striplen,
2777 pathlen - striplen + 1);
2778 }
2780 /* Skip "Directory ..." and ".." line. */
2781 for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
2782 struct line *line = &view->line[pos];
2783 char *path1 = ((char *) line->data) + SIZEOF_TREE_ATTR;
2784 char *path2 = text + SIZEOF_TREE_ATTR;
2785 int cmp = tree_compare_entry(line->type, path1, type, path2);
2787 if (cmp <= 0)
2788 continue;
2790 text = strdup(text);
2791 if (!text)
2792 return FALSE;
2794 if (view->lines > pos)
2795 memmove(&view->line[pos + 1], &view->line[pos],
2796 (view->lines - pos) * sizeof(*line));
2798 line = &view->line[pos];
2799 line->data = text;
2800 line->type = type;
2801 view->lines++;
2802 return TRUE;
2803 }
2805 if (!add_line_text(view, text, type))
2806 return FALSE;
2808 if (tree_lineno > view->lineno) {
2809 view->lineno = tree_lineno;
2810 tree_lineno = 0;
2811 }
2813 return TRUE;
2814 }
2816 static enum request
2817 tree_request(struct view *view, enum request request, struct line *line)
2818 {
2819 enum open_flags flags;
2821 if (request != REQ_ENTER)
2822 return request;
2824 /* Cleanup the stack if the tree view is at a different tree. */
2825 while (!*opt_path && tree_stack)
2826 pop_tree_stack_entry();
2828 switch (line->type) {
2829 case LINE_TREE_DIR:
2830 /* Depending on whether it is a subdir or parent (updir?) link
2831 * mangle the path buffer. */
2832 if (line == &view->line[1] && *opt_path) {
2833 pop_tree_stack_entry();
2835 } else {
2836 char *data = line->data;
2837 char *basename = data + SIZEOF_TREE_ATTR;
2839 push_tree_stack_entry(basename, view->lineno);
2840 }
2842 /* Trees and subtrees share the same ID, so they are not not
2843 * unique like blobs. */
2844 flags = OPEN_RELOAD;
2845 request = REQ_VIEW_TREE;
2846 break;
2848 case LINE_TREE_FILE:
2849 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2850 request = REQ_VIEW_BLOB;
2851 break;
2853 default:
2854 return TRUE;
2855 }
2857 open_view(view, request, flags);
2858 if (request == REQ_VIEW_TREE) {
2859 view->lineno = tree_lineno;
2860 }
2862 return REQ_NONE;
2863 }
2865 static void
2866 tree_select(struct view *view, struct line *line)
2867 {
2868 char *text = line->data + STRING_SIZE("100644 blob ");
2870 if (line->type == LINE_TREE_FILE) {
2871 string_copy_rev(ref_blob, text);
2873 } else if (line->type != LINE_TREE_DIR) {
2874 return;
2875 }
2877 string_copy_rev(view->ref, text);
2878 }
2880 static struct view_ops tree_ops = {
2881 "file",
2882 NULL,
2883 tree_read,
2884 pager_draw,
2885 tree_request,
2886 pager_grep,
2887 tree_select,
2888 };
2890 static bool
2891 blob_read(struct view *view, char *line)
2892 {
2893 return add_line_text(view, line, LINE_DEFAULT) != NULL;
2894 }
2896 static struct view_ops blob_ops = {
2897 "line",
2898 NULL,
2899 blob_read,
2900 pager_draw,
2901 pager_request,
2902 pager_grep,
2903 pager_select,
2904 };
2907 /*
2908 * Status backend
2909 */
2911 struct status {
2912 char status;
2913 struct {
2914 mode_t mode;
2915 char rev[SIZEOF_REV];
2916 } old;
2917 struct {
2918 mode_t mode;
2919 char rev[SIZEOF_REV];
2920 } new;
2921 char name[SIZEOF_STR];
2922 };
2924 static struct status stage_status;
2925 static enum line_type stage_line_type;
2927 /* Get fields from the diff line:
2928 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
2929 */
2930 static inline bool
2931 status_get_diff(struct status *file, char *buf, size_t bufsize)
2932 {
2933 char *old_mode = buf + 1;
2934 char *new_mode = buf + 8;
2935 char *old_rev = buf + 15;
2936 char *new_rev = buf + 56;
2937 char *status = buf + 97;
2939 if (bufsize != 99 ||
2940 old_mode[-1] != ':' ||
2941 new_mode[-1] != ' ' ||
2942 old_rev[-1] != ' ' ||
2943 new_rev[-1] != ' ' ||
2944 status[-1] != ' ')
2945 return FALSE;
2947 file->status = *status;
2949 string_copy_rev(file->old.rev, old_rev);
2950 string_copy_rev(file->new.rev, new_rev);
2952 file->old.mode = strtoul(old_mode, NULL, 8);
2953 file->new.mode = strtoul(new_mode, NULL, 8);
2955 file->name[0] = 0;
2957 return TRUE;
2958 }
2960 static bool
2961 status_run(struct view *view, const char cmd[], bool diff, enum line_type type)
2962 {
2963 struct status *file = NULL;
2964 char buf[SIZEOF_STR * 4];
2965 size_t bufsize = 0;
2966 FILE *pipe;
2968 pipe = popen(cmd, "r");
2969 if (!pipe)
2970 return FALSE;
2972 add_line_data(view, NULL, type);
2974 while (!feof(pipe) && !ferror(pipe)) {
2975 char *sep;
2976 size_t readsize;
2978 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
2979 if (!readsize)
2980 break;
2981 bufsize += readsize;
2983 /* Process while we have NUL chars. */
2984 while ((sep = memchr(buf, 0, bufsize))) {
2985 size_t sepsize = sep - buf + 1;
2987 if (!file) {
2988 if (!realloc_lines(view, view->line_size + 1))
2989 goto error_out;
2991 file = calloc(1, sizeof(*file));
2992 if (!file)
2993 goto error_out;
2995 add_line_data(view, file, type);
2996 }
2998 /* Parse diff info part. */
2999 if (!diff) {
3000 file->status = '?';
3002 } else if (!file->status) {
3003 if (!status_get_diff(file, buf, sepsize))
3004 goto error_out;
3006 bufsize -= sepsize;
3007 memmove(buf, sep + 1, bufsize);
3009 sep = memchr(buf, 0, bufsize);
3010 if (!sep)
3011 break;
3012 sepsize = sep - buf + 1;
3013 }
3015 /* git-ls-files just delivers a NUL separated
3016 * list of file names similar to the second half
3017 * of the git-diff-* output. */
3018 string_ncopy(file->name, buf, sepsize);
3019 bufsize -= sepsize;
3020 memmove(buf, sep + 1, bufsize);
3021 file = NULL;
3022 }
3023 }
3025 if (ferror(pipe)) {
3026 error_out:
3027 pclose(pipe);
3028 return FALSE;
3029 }
3031 if (!view->line[view->lines - 1].data)
3032 add_line_data(view, NULL, LINE_STAT_NONE);
3034 pclose(pipe);
3035 return TRUE;
3036 }
3038 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --cached HEAD"
3039 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
3040 #define STATUS_LIST_OTHER_CMD \
3041 "git ls-files -z --others --exclude-per-directory=.gitignore"
3043 #define STATUS_DIFF_SHOW_CMD \
3044 "git diff --root --patch-with-stat --find-copies-harder -B -C %s -- %s 2>/dev/null"
3046 /* First parse staged info using git-diff-index(1), then parse unstaged
3047 * info using git-diff-files(1), and finally untracked files using
3048 * git-ls-files(1). */
3049 static bool
3050 status_open(struct view *view)
3051 {
3052 struct stat statbuf;
3053 char exclude[SIZEOF_STR];
3054 char cmd[SIZEOF_STR];
3055 unsigned long prev_lineno = view->lineno;
3056 size_t i;
3059 for (i = 0; i < view->lines; i++)
3060 free(view->line[i].data);
3061 free(view->line);
3062 view->lines = view->line_size = view->lineno = 0;
3063 view->line = NULL;
3065 if (!realloc_lines(view, view->line_size + 6))
3066 return FALSE;
3068 if (!string_format(exclude, "%s/info/exclude", opt_git_dir))
3069 return FALSE;
3071 string_copy(cmd, STATUS_LIST_OTHER_CMD);
3073 if (stat(exclude, &statbuf) >= 0) {
3074 size_t cmdsize = strlen(cmd);
3076 if (!string_format_from(cmd, &cmdsize, " %s", "--exclude-from=") ||
3077 sq_quote(cmd, cmdsize, exclude) >= sizeof(cmd))
3078 return FALSE;
3079 }
3081 if (!status_run(view, STATUS_DIFF_INDEX_CMD, TRUE, LINE_STAT_STAGED) ||
3082 !status_run(view, STATUS_DIFF_FILES_CMD, TRUE, LINE_STAT_UNSTAGED) ||
3083 !status_run(view, cmd, FALSE, LINE_STAT_UNTRACKED))
3084 return FALSE;
3086 /* If all went well restore the previous line number to stay in
3087 * the context. */
3088 if (prev_lineno < view->lines)
3089 view->lineno = prev_lineno;
3090 else
3091 view->lineno = view->lines - 1;
3093 return TRUE;
3094 }
3096 static bool
3097 status_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3098 {
3099 struct status *status = line->data;
3101 wmove(view->win, lineno, 0);
3103 if (selected) {
3104 wattrset(view->win, get_line_attr(LINE_CURSOR));
3105 wchgat(view->win, -1, 0, LINE_CURSOR, NULL);
3107 } else if (!status && line->type != LINE_STAT_NONE) {
3108 wattrset(view->win, get_line_attr(LINE_STAT_SECTION));
3109 wchgat(view->win, -1, 0, LINE_STAT_SECTION, NULL);
3111 } else {
3112 wattrset(view->win, get_line_attr(line->type));
3113 }
3115 if (!status) {
3116 char *text;
3118 switch (line->type) {
3119 case LINE_STAT_STAGED:
3120 text = "Changes to be committed:";
3121 break;
3123 case LINE_STAT_UNSTAGED:
3124 text = "Changed but not updated:";
3125 break;
3127 case LINE_STAT_UNTRACKED:
3128 text = "Untracked files:";
3129 break;
3131 case LINE_STAT_NONE:
3132 text = " (no files)";
3133 break;
3135 default:
3136 return FALSE;
3137 }
3139 waddstr(view->win, text);
3140 return TRUE;
3141 }
3143 waddch(view->win, status->status);
3144 if (!selected)
3145 wattrset(view->win, A_NORMAL);
3146 wmove(view->win, lineno, 4);
3147 waddstr(view->win, status->name);
3149 return TRUE;
3150 }
3152 static enum request
3153 status_enter(struct view *view, struct line *line)
3154 {
3155 struct status *status = line->data;
3156 char path[SIZEOF_STR] = "";
3157 char *info;
3158 size_t cmdsize = 0;
3160 if (line->type == LINE_STAT_NONE ||
3161 (!status && line[1].type == LINE_STAT_NONE)) {
3162 report("No file to diff");
3163 return REQ_NONE;
3164 }
3166 if (status && sq_quote(path, 0, status->name) >= sizeof(path))
3167 return REQ_QUIT;
3169 if (opt_cdup[0] &&
3170 line->type != LINE_STAT_UNTRACKED &&
3171 !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
3172 return REQ_QUIT;
3174 switch (line->type) {
3175 case LINE_STAT_STAGED:
3176 if (!string_format_from(opt_cmd, &cmdsize, STATUS_DIFF_SHOW_CMD,
3177 "--cached", path))
3178 return REQ_QUIT;
3179 if (status)
3180 info = "Staged changes to %s";
3181 else
3182 info = "Staged changes";
3183 break;
3185 case LINE_STAT_UNSTAGED:
3186 if (!string_format_from(opt_cmd, &cmdsize, STATUS_DIFF_SHOW_CMD,
3187 "", path))
3188 return REQ_QUIT;
3189 if (status)
3190 info = "Unstaged changes to %s";
3191 else
3192 info = "Unstaged changes";
3193 break;
3195 case LINE_STAT_UNTRACKED:
3196 if (opt_pipe)
3197 return REQ_QUIT;
3200 if (!status) {
3201 report("No file to show");
3202 return REQ_NONE;
3203 }
3205 opt_pipe = fopen(status->name, "r");
3206 info = "Untracked file %s";
3207 break;
3209 default:
3210 die("w00t");
3211 }
3213 open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | OPEN_SPLIT);
3214 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
3215 if (status) {
3216 stage_status = *status;
3217 } else {
3218 memset(&stage_status, 0, sizeof(stage_status));
3219 }
3221 stage_line_type = line->type;
3222 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.name);
3223 }
3225 return REQ_NONE;
3226 }
3229 static bool
3230 status_update_file(struct view *view, struct status *status, enum line_type type)
3231 {
3232 char cmd[SIZEOF_STR];
3233 char buf[SIZEOF_STR];
3234 size_t cmdsize = 0;
3235 size_t bufsize = 0;
3236 size_t written = 0;
3237 FILE *pipe;
3239 if (opt_cdup[0] &&
3240 type != LINE_STAT_UNTRACKED &&
3241 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
3242 return FALSE;
3244 switch (type) {
3245 case LINE_STAT_STAGED:
3246 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
3247 status->old.mode,
3248 status->old.rev,
3249 status->name, 0))
3250 return FALSE;
3252 string_add(cmd, cmdsize, "git update-index -z --index-info");
3253 break;
3255 case LINE_STAT_UNSTAGED:
3256 case LINE_STAT_UNTRACKED:
3257 if (!string_format_from(buf, &bufsize, "%s%c", status->name, 0))
3258 return FALSE;
3260 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
3261 break;
3263 default:
3264 die("w00t");
3265 }
3267 pipe = popen(cmd, "w");
3268 if (!pipe)
3269 return FALSE;
3271 while (!ferror(pipe) && written < bufsize) {
3272 written += fwrite(buf + written, 1, bufsize - written, pipe);
3273 }
3275 pclose(pipe);
3277 if (written != bufsize)
3278 return FALSE;
3280 return TRUE;
3281 }
3283 static void
3284 status_update(struct view *view)
3285 {
3286 struct line *line = &view->line[view->lineno];
3288 assert(view->lines);
3290 if (!line->data) {
3291 while (++line < view->line + view->lines && line->data) {
3292 if (!status_update_file(view, line->data, line->type))
3293 report("Failed to update file status");
3294 }
3296 if (!line[-1].data) {
3297 report("Nothing to update");
3298 return;
3299 }
3301 } else if (!status_update_file(view, line->data, line->type)) {
3302 report("Failed to update file status");
3303 }
3305 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
3306 }
3308 static enum request
3309 status_request(struct view *view, enum request request, struct line *line)
3310 {
3311 struct status *status = line->data;
3313 switch (request) {
3314 case REQ_STATUS_UPDATE:
3315 status_update(view);
3316 break;
3318 case REQ_EDIT:
3319 if (!status)
3320 return request;
3322 open_editor(view, status->status != '?', status->name);
3323 break;
3325 case REQ_ENTER:
3326 status_enter(view, line);
3327 break;
3329 default:
3330 return request;
3331 }
3333 return REQ_NONE;
3334 }
3336 static void
3337 status_select(struct view *view, struct line *line)
3338 {
3339 struct status *status = line->data;
3340 char file[SIZEOF_STR] = "all files";
3341 char *text;
3343 if (status && !string_format(file, "'%s'", status->name))
3344 return;
3346 if (!status && line[1].type == LINE_STAT_NONE)
3347 line++;
3349 switch (line->type) {
3350 case LINE_STAT_STAGED:
3351 text = "Press %s to unstage %s for commit";
3352 break;
3354 case LINE_STAT_UNSTAGED:
3355 text = "Press %s to stage %s for commit";
3356 break;
3358 case LINE_STAT_UNTRACKED:
3359 text = "Press %s to stage %s for addition";
3360 break;
3362 case LINE_STAT_NONE:
3363 text = "Nothing to update";
3364 break;
3366 default:
3367 die("w00t");
3368 }
3370 string_format(view->ref, text, get_key(REQ_STATUS_UPDATE), file);
3371 }
3373 static bool
3374 status_grep(struct view *view, struct line *line)
3375 {
3376 struct status *status = line->data;
3377 enum { S_STATUS, S_NAME, S_END } state;
3378 char buf[2] = "?";
3379 regmatch_t pmatch;
3381 if (!status)
3382 return FALSE;
3384 for (state = S_STATUS; state < S_END; state++) {
3385 char *text;
3387 switch (state) {
3388 case S_NAME: text = status->name; break;
3389 case S_STATUS:
3390 buf[0] = status->status;
3391 text = buf;
3392 break;
3394 default:
3395 return FALSE;
3396 }
3398 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3399 return TRUE;
3400 }
3402 return FALSE;
3403 }
3405 static struct view_ops status_ops = {
3406 "file",
3407 status_open,
3408 NULL,
3409 status_draw,
3410 status_request,
3411 status_grep,
3412 status_select,
3413 };
3416 static bool
3417 stage_diff_line(FILE *pipe, struct line *line)
3418 {
3419 char *buf = line->data;
3420 size_t bufsize = strlen(buf);
3421 size_t written = 0;
3423 while (!ferror(pipe) && written < bufsize) {
3424 written += fwrite(buf + written, 1, bufsize - written, pipe);
3425 }
3427 fputc('\n', pipe);
3429 return written == bufsize;
3430 }
3432 static struct line *
3433 stage_diff_hdr(struct view *view, struct line *line)
3434 {
3435 int diff_hdr_dir = line->type == LINE_DIFF_CHUNK ? -1 : 1;
3436 struct line *diff_hdr;
3438 if (line->type == LINE_DIFF_CHUNK)
3439 diff_hdr = line - 1;
3440 else
3441 diff_hdr = view->line + 1;
3443 while (diff_hdr > view->line && diff_hdr < view->line + view->lines) {
3444 if (diff_hdr->type == LINE_DIFF_HEADER)
3445 return diff_hdr;
3447 diff_hdr += diff_hdr_dir;
3448 }
3450 return NULL;
3451 }
3453 static bool
3454 stage_update_chunk(struct view *view, struct line *line)
3455 {
3456 char cmd[SIZEOF_STR];
3457 size_t cmdsize = 0;
3458 struct line *diff_hdr, *diff_chunk, *diff_end;
3459 FILE *pipe;
3461 diff_hdr = stage_diff_hdr(view, line);
3462 if (!diff_hdr)
3463 return FALSE;
3465 if (opt_cdup[0] &&
3466 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
3467 return FALSE;
3469 if (!string_format_from(cmd, &cmdsize,
3470 "git apply --cached %s - && "
3471 "git update-index -q --unmerged --refresh 2>/dev/null",
3472 stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
3473 return FALSE;
3475 pipe = popen(cmd, "w");
3476 if (!pipe)
3477 return FALSE;
3479 diff_end = view->line + view->lines;
3480 if (line->type != LINE_DIFF_CHUNK) {
3481 diff_chunk = diff_hdr;
3483 } else {
3484 for (diff_chunk = line + 1; diff_chunk < diff_end; diff_chunk++)
3485 if (diff_chunk->type == LINE_DIFF_CHUNK ||
3486 diff_chunk->type == LINE_DIFF_HEADER)
3487 diff_end = diff_chunk;
3489 diff_chunk = line;
3491 while (diff_hdr->type != LINE_DIFF_CHUNK) {
3492 switch (diff_hdr->type) {
3493 case LINE_DIFF_HEADER:
3494 case LINE_DIFF_INDEX:
3495 case LINE_DIFF_ADD:
3496 case LINE_DIFF_DEL:
3497 break;
3499 default:
3500 diff_hdr++;
3501 continue;
3502 }
3504 if (!stage_diff_line(pipe, diff_hdr++)) {
3505 pclose(pipe);
3506 return FALSE;
3507 }
3508 }
3509 }
3511 while (diff_chunk < diff_end && stage_diff_line(pipe, diff_chunk))
3512 diff_chunk++;
3514 pclose(pipe);
3516 if (diff_chunk != diff_end)
3517 return FALSE;
3519 return TRUE;
3520 }
3522 static void
3523 stage_update(struct view *view, struct line *line)
3524 {
3525 if (stage_line_type != LINE_STAT_UNTRACKED &&
3526 (line->type == LINE_DIFF_CHUNK || !stage_status.status)) {
3527 if (!stage_update_chunk(view, line)) {
3528 report("Failed to apply chunk");
3529 return;
3530 }
3532 } else if (!status_update_file(view, &stage_status, stage_line_type)) {
3533 report("Failed to update file");
3534 return;
3535 }
3537 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
3539 view = VIEW(REQ_VIEW_STATUS);
3540 if (view_is_displayed(view))
3541 status_enter(view, &view->line[view->lineno]);
3542 }
3544 static enum request
3545 stage_request(struct view *view, enum request request, struct line *line)
3546 {
3547 switch (request) {
3548 case REQ_STATUS_UPDATE:
3549 stage_update(view, line);
3550 break;
3552 case REQ_EDIT:
3553 if (!stage_status.name[0])
3554 return request;
3556 open_editor(view, stage_status.status != '?', stage_status.name);
3557 break;
3559 case REQ_ENTER:
3560 pager_request(view, request, line);
3561 break;
3563 default:
3564 return request;
3565 }
3567 return REQ_NONE;
3568 }
3570 static struct view_ops stage_ops = {
3571 "line",
3572 NULL,
3573 pager_read,
3574 pager_draw,
3575 stage_request,
3576 pager_grep,
3577 pager_select,
3578 };
3581 /*
3582 * Revision graph
3583 */
3585 struct commit {
3586 char id[SIZEOF_REV]; /* SHA1 ID. */
3587 char title[128]; /* First line of the commit message. */
3588 char author[75]; /* Author of the commit. */
3589 struct tm time; /* Date from the author ident. */
3590 struct ref **refs; /* Repository references. */
3591 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
3592 size_t graph_size; /* The width of the graph array. */
3593 };
3595 /* Size of rev graph with no "padding" columns */
3596 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
3598 struct rev_graph {
3599 struct rev_graph *prev, *next, *parents;
3600 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
3601 size_t size;
3602 struct commit *commit;
3603 size_t pos;
3604 };
3606 /* Parents of the commit being visualized. */
3607 static struct rev_graph graph_parents[4];
3609 /* The current stack of revisions on the graph. */
3610 static struct rev_graph graph_stacks[4] = {
3611 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
3612 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
3613 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
3614 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
3615 };
3617 static inline bool
3618 graph_parent_is_merge(struct rev_graph *graph)
3619 {
3620 return graph->parents->size > 1;
3621 }
3623 static inline void
3624 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
3625 {
3626 struct commit *commit = graph->commit;
3628 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
3629 commit->graph[commit->graph_size++] = symbol;
3630 }
3632 static void
3633 done_rev_graph(struct rev_graph *graph)
3634 {
3635 if (graph_parent_is_merge(graph) &&
3636 graph->pos < graph->size - 1 &&
3637 graph->next->size == graph->size + graph->parents->size - 1) {
3638 size_t i = graph->pos + graph->parents->size - 1;
3640 graph->commit->graph_size = i * 2;
3641 while (i < graph->next->size - 1) {
3642 append_to_rev_graph(graph, ' ');
3643 append_to_rev_graph(graph, '\\');
3644 i++;
3645 }
3646 }
3648 graph->size = graph->pos = 0;
3649 graph->commit = NULL;
3650 memset(graph->parents, 0, sizeof(*graph->parents));
3651 }
3653 static void
3654 push_rev_graph(struct rev_graph *graph, char *parent)
3655 {
3656 int i;
3658 /* "Collapse" duplicate parents lines.
3659 *
3660 * FIXME: This needs to also update update the drawn graph but
3661 * for now it just serves as a method for pruning graph lines. */
3662 for (i = 0; i < graph->size; i++)
3663 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
3664 return;
3666 if (graph->size < SIZEOF_REVITEMS) {
3667 string_copy_rev(graph->rev[graph->size++], parent);
3668 }
3669 }
3671 static chtype
3672 get_rev_graph_symbol(struct rev_graph *graph)
3673 {
3674 chtype symbol;
3676 if (graph->parents->size == 0)
3677 symbol = REVGRAPH_INIT;
3678 else if (graph_parent_is_merge(graph))
3679 symbol = REVGRAPH_MERGE;
3680 else if (graph->pos >= graph->size)
3681 symbol = REVGRAPH_BRANCH;
3682 else
3683 symbol = REVGRAPH_COMMIT;
3685 return symbol;
3686 }
3688 static void
3689 draw_rev_graph(struct rev_graph *graph)
3690 {
3691 struct rev_filler {
3692 chtype separator, line;
3693 };
3694 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
3695 static struct rev_filler fillers[] = {
3696 { ' ', REVGRAPH_LINE },
3697 { '`', '.' },
3698 { '\'', ' ' },
3699 { '/', ' ' },
3700 };
3701 chtype symbol = get_rev_graph_symbol(graph);
3702 struct rev_filler *filler;
3703 size_t i;
3705 filler = &fillers[DEFAULT];
3707 for (i = 0; i < graph->pos; i++) {
3708 append_to_rev_graph(graph, filler->line);
3709 if (graph_parent_is_merge(graph->prev) &&
3710 graph->prev->pos == i)
3711 filler = &fillers[RSHARP];
3713 append_to_rev_graph(graph, filler->separator);
3714 }
3716 /* Place the symbol for this revision. */
3717 append_to_rev_graph(graph, symbol);
3719 if (graph->prev->size > graph->size)
3720 filler = &fillers[RDIAG];
3721 else
3722 filler = &fillers[DEFAULT];
3724 i++;
3726 for (; i < graph->size; i++) {
3727 append_to_rev_graph(graph, filler->separator);
3728 append_to_rev_graph(graph, filler->line);
3729 if (graph_parent_is_merge(graph->prev) &&
3730 i < graph->prev->pos + graph->parents->size)
3731 filler = &fillers[RSHARP];
3732 if (graph->prev->size > graph->size)
3733 filler = &fillers[LDIAG];
3734 }
3736 if (graph->prev->size > graph->size) {
3737 append_to_rev_graph(graph, filler->separator);
3738 if (filler->line != ' ')
3739 append_to_rev_graph(graph, filler->line);
3740 }
3741 }
3743 /* Prepare the next rev graph */
3744 static void
3745 prepare_rev_graph(struct rev_graph *graph)
3746 {
3747 size_t i;
3749 /* First, traverse all lines of revisions up to the active one. */
3750 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
3751 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
3752 break;
3754 push_rev_graph(graph->next, graph->rev[graph->pos]);
3755 }
3757 /* Interleave the new revision parent(s). */
3758 for (i = 0; i < graph->parents->size; i++)
3759 push_rev_graph(graph->next, graph->parents->rev[i]);
3761 /* Lastly, put any remaining revisions. */
3762 for (i = graph->pos + 1; i < graph->size; i++)
3763 push_rev_graph(graph->next, graph->rev[i]);
3764 }
3766 static void
3767 update_rev_graph(struct rev_graph *graph)
3768 {
3769 /* If this is the finalizing update ... */
3770 if (graph->commit)
3771 prepare_rev_graph(graph);
3773 /* Graph visualization needs a one rev look-ahead,
3774 * so the first update doesn't visualize anything. */
3775 if (!graph->prev->commit)
3776 return;
3778 draw_rev_graph(graph->prev);
3779 done_rev_graph(graph->prev->prev);
3780 }
3783 /*
3784 * Main view backend
3785 */
3787 static bool
3788 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3789 {
3790 char buf[DATE_COLS + 1];
3791 struct commit *commit = line->data;
3792 enum line_type type;
3793 int col = 0;
3794 size_t timelen;
3795 size_t authorlen;
3796 int trimmed = 1;
3798 if (!*commit->author)
3799 return FALSE;
3801 wmove(view->win, lineno, col);
3803 if (selected) {
3804 type = LINE_CURSOR;
3805 wattrset(view->win, get_line_attr(type));
3806 wchgat(view->win, -1, 0, type, NULL);
3808 } else {
3809 type = LINE_MAIN_COMMIT;
3810 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
3811 }
3813 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
3814 waddnstr(view->win, buf, timelen);
3815 waddstr(view->win, " ");
3817 col += DATE_COLS;
3818 wmove(view->win, lineno, col);
3819 if (type != LINE_CURSOR)
3820 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
3822 if (opt_utf8) {
3823 authorlen = utf8_length(commit->author, AUTHOR_COLS - 2, &col, &trimmed);
3824 } else {
3825 authorlen = strlen(commit->author);
3826 if (authorlen > AUTHOR_COLS - 2) {
3827 authorlen = AUTHOR_COLS - 2;
3828 trimmed = 1;
3829 }
3830 }
3832 if (trimmed) {
3833 waddnstr(view->win, commit->author, authorlen);
3834 if (type != LINE_CURSOR)
3835 wattrset(view->win, get_line_attr(LINE_MAIN_DELIM));
3836 waddch(view->win, '~');
3837 } else {
3838 waddstr(view->win, commit->author);
3839 }
3841 col += AUTHOR_COLS;
3842 if (type != LINE_CURSOR)
3843 wattrset(view->win, A_NORMAL);
3845 if (opt_rev_graph && commit->graph_size) {
3846 size_t i;
3848 wmove(view->win, lineno, col);
3849 /* Using waddch() instead of waddnstr() ensures that
3850 * they'll be rendered correctly for the cursor line. */
3851 for (i = 0; i < commit->graph_size; i++)
3852 waddch(view->win, commit->graph[i]);
3854 waddch(view->win, ' ');
3855 col += commit->graph_size + 1;
3856 }
3858 wmove(view->win, lineno, col);
3860 if (commit->refs) {
3861 size_t i = 0;
3863 do {
3864 if (type == LINE_CURSOR)
3865 ;
3866 else if (commit->refs[i]->tag)
3867 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
3868 else if (commit->refs[i]->remote)
3869 wattrset(view->win, get_line_attr(LINE_MAIN_REMOTE));
3870 else
3871 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
3872 waddstr(view->win, "[");
3873 waddstr(view->win, commit->refs[i]->name);
3874 waddstr(view->win, "]");
3875 if (type != LINE_CURSOR)
3876 wattrset(view->win, A_NORMAL);
3877 waddstr(view->win, " ");
3878 col += strlen(commit->refs[i]->name) + STRING_SIZE("[] ");
3879 } while (commit->refs[i++]->next);
3880 }
3882 if (type != LINE_CURSOR)
3883 wattrset(view->win, get_line_attr(type));
3885 {
3886 int titlelen = strlen(commit->title);
3888 if (col + titlelen > view->width)
3889 titlelen = view->width - col;
3891 waddnstr(view->win, commit->title, titlelen);
3892 }
3894 return TRUE;
3895 }
3897 /* Reads git log --pretty=raw output and parses it into the commit struct. */
3898 static bool
3899 main_read(struct view *view, char *line)
3900 {
3901 static struct rev_graph *graph = graph_stacks;
3902 enum line_type type;
3903 struct commit *commit;
3905 if (!line) {
3906 update_rev_graph(graph);
3907 return TRUE;
3908 }
3910 type = get_line_type(line);
3911 if (type == LINE_COMMIT) {
3912 commit = calloc(1, sizeof(struct commit));
3913 if (!commit)
3914 return FALSE;
3916 string_copy_rev(commit->id, line + STRING_SIZE("commit "));
3917 commit->refs = get_refs(commit->id);
3918 graph->commit = commit;
3919 add_line_data(view, commit, LINE_MAIN_COMMIT);
3920 return TRUE;
3921 }
3923 if (!view->lines)
3924 return TRUE;
3925 commit = view->line[view->lines - 1].data;
3927 switch (type) {
3928 case LINE_PARENT:
3929 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
3930 break;
3932 case LINE_AUTHOR:
3933 {
3934 /* Parse author lines where the name may be empty:
3935 * author <email@address.tld> 1138474660 +0100
3936 */
3937 char *ident = line + STRING_SIZE("author ");
3938 char *nameend = strchr(ident, '<');
3939 char *emailend = strchr(ident, '>');
3941 if (!nameend || !emailend)
3942 break;
3944 update_rev_graph(graph);
3945 graph = graph->next;
3947 *nameend = *emailend = 0;
3948 ident = chomp_string(ident);
3949 if (!*ident) {
3950 ident = chomp_string(nameend + 1);
3951 if (!*ident)
3952 ident = "Unknown";
3953 }
3955 string_ncopy(commit->author, ident, strlen(ident));
3957 /* Parse epoch and timezone */
3958 if (emailend[1] == ' ') {
3959 char *secs = emailend + 2;
3960 char *zone = strchr(secs, ' ');
3961 time_t time = (time_t) atol(secs);
3963 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
3964 long tz;
3966 zone++;
3967 tz = ('0' - zone[1]) * 60 * 60 * 10;
3968 tz += ('0' - zone[2]) * 60 * 60;
3969 tz += ('0' - zone[3]) * 60;
3970 tz += ('0' - zone[4]) * 60;
3972 if (zone[0] == '-')
3973 tz = -tz;
3975 time -= tz;
3976 }
3978 gmtime_r(&time, &commit->time);
3979 }
3980 break;
3981 }
3982 default:
3983 /* Fill in the commit title if it has not already been set. */
3984 if (commit->title[0])
3985 break;
3987 /* Require titles to start with a non-space character at the
3988 * offset used by git log. */
3989 if (strncmp(line, " ", 4))
3990 break;
3991 line += 4;
3992 /* Well, if the title starts with a whitespace character,
3993 * try to be forgiving. Otherwise we end up with no title. */
3994 while (isspace(*line))
3995 line++;
3996 if (*line == '\0')
3997 break;
3998 /* FIXME: More graceful handling of titles; append "..." to
3999 * shortened titles, etc. */
4001 string_ncopy(commit->title, line, strlen(line));
4002 }
4004 return TRUE;
4005 }
4007 static void
4008 cherry_pick_commit(struct commit *commit)
4009 {
4010 char cmd[SIZEOF_STR];
4011 char *cherry_pick = getenv("TIG_CHERRY_PICK");
4013 if (!cherry_pick)
4014 cherry_pick = "git cherry-pick";
4016 if (string_format(cmd, "%s %s", cherry_pick, commit->id)) {
4017 def_prog_mode(); /* save current tty modes */
4018 endwin(); /* restore original tty modes */
4019 system(cmd);
4020 fprintf(stderr, "Press Enter to continue");
4021 getc(stdin);
4022 reset_prog_mode();
4023 redraw_display();
4024 }
4025 }
4027 static enum request
4028 main_request(struct view *view, enum request request, struct line *line)
4029 {
4030 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4032 if (request == REQ_ENTER)
4033 open_view(view, REQ_VIEW_DIFF, flags);
4034 else if (request == REQ_CHERRY_PICK)
4035 cherry_pick_commit(line->data);
4036 else
4037 return request;
4039 return REQ_NONE;
4040 }
4042 static bool
4043 main_grep(struct view *view, struct line *line)
4044 {
4045 struct commit *commit = line->data;
4046 enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
4047 char buf[DATE_COLS + 1];
4048 regmatch_t pmatch;
4050 for (state = S_TITLE; state < S_END; state++) {
4051 char *text;
4053 switch (state) {
4054 case S_TITLE: text = commit->title; break;
4055 case S_AUTHOR: text = commit->author; break;
4056 case S_DATE:
4057 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
4058 continue;
4059 text = buf;
4060 break;
4062 default:
4063 return FALSE;
4064 }
4066 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4067 return TRUE;
4068 }
4070 return FALSE;
4071 }
4073 static void
4074 main_select(struct view *view, struct line *line)
4075 {
4076 struct commit *commit = line->data;
4078 string_copy_rev(view->ref, commit->id);
4079 string_copy_rev(ref_commit, view->ref);
4080 }
4082 static struct view_ops main_ops = {
4083 "commit",
4084 NULL,
4085 main_read,
4086 main_draw,
4087 main_request,
4088 main_grep,
4089 main_select,
4090 };
4093 /*
4094 * Unicode / UTF-8 handling
4095 *
4096 * NOTE: Much of the following code for dealing with unicode is derived from
4097 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
4098 * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
4099 */
4101 /* I've (over)annotated a lot of code snippets because I am not entirely
4102 * confident that the approach taken by this small UTF-8 interface is correct.
4103 * --jonas */
4105 static inline int
4106 unicode_width(unsigned long c)
4107 {
4108 if (c >= 0x1100 &&
4109 (c <= 0x115f /* Hangul Jamo */
4110 || c == 0x2329
4111 || c == 0x232a
4112 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
4113 /* CJK ... Yi */
4114 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
4115 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
4116 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
4117 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
4118 || (c >= 0xffe0 && c <= 0xffe6)
4119 || (c >= 0x20000 && c <= 0x2fffd)
4120 || (c >= 0x30000 && c <= 0x3fffd)))
4121 return 2;
4123 return 1;
4124 }
4126 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
4127 * Illegal bytes are set one. */
4128 static const unsigned char utf8_bytes[256] = {
4129 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,
4130 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,
4131 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,
4132 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,
4133 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,
4134 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,
4135 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,
4136 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,
4137 };
4139 /* Decode UTF-8 multi-byte representation into a unicode character. */
4140 static inline unsigned long
4141 utf8_to_unicode(const char *string, size_t length)
4142 {
4143 unsigned long unicode;
4145 switch (length) {
4146 case 1:
4147 unicode = string[0];
4148 break;
4149 case 2:
4150 unicode = (string[0] & 0x1f) << 6;
4151 unicode += (string[1] & 0x3f);
4152 break;
4153 case 3:
4154 unicode = (string[0] & 0x0f) << 12;
4155 unicode += ((string[1] & 0x3f) << 6);
4156 unicode += (string[2] & 0x3f);
4157 break;
4158 case 4:
4159 unicode = (string[0] & 0x0f) << 18;
4160 unicode += ((string[1] & 0x3f) << 12);
4161 unicode += ((string[2] & 0x3f) << 6);
4162 unicode += (string[3] & 0x3f);
4163 break;
4164 case 5:
4165 unicode = (string[0] & 0x0f) << 24;
4166 unicode += ((string[1] & 0x3f) << 18);
4167 unicode += ((string[2] & 0x3f) << 12);
4168 unicode += ((string[3] & 0x3f) << 6);
4169 unicode += (string[4] & 0x3f);
4170 break;
4171 case 6:
4172 unicode = (string[0] & 0x01) << 30;
4173 unicode += ((string[1] & 0x3f) << 24);
4174 unicode += ((string[2] & 0x3f) << 18);
4175 unicode += ((string[3] & 0x3f) << 12);
4176 unicode += ((string[4] & 0x3f) << 6);
4177 unicode += (string[5] & 0x3f);
4178 break;
4179 default:
4180 die("Invalid unicode length");
4181 }
4183 /* Invalid characters could return the special 0xfffd value but NUL
4184 * should be just as good. */
4185 return unicode > 0xffff ? 0 : unicode;
4186 }
4188 /* Calculates how much of string can be shown within the given maximum width
4189 * and sets trimmed parameter to non-zero value if all of string could not be
4190 * shown.
4191 *
4192 * Additionally, adds to coloffset how many many columns to move to align with
4193 * the expected position. Takes into account how multi-byte and double-width
4194 * characters will effect the cursor position.
4195 *
4196 * Returns the number of bytes to output from string to satisfy max_width. */
4197 static size_t
4198 utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed)
4199 {
4200 const char *start = string;
4201 const char *end = strchr(string, '\0');
4202 size_t mbwidth = 0;
4203 size_t width = 0;
4205 *trimmed = 0;
4207 while (string < end) {
4208 int c = *(unsigned char *) string;
4209 unsigned char bytes = utf8_bytes[c];
4210 size_t ucwidth;
4211 unsigned long unicode;
4213 if (string + bytes > end)
4214 break;
4216 /* Change representation to figure out whether
4217 * it is a single- or double-width character. */
4219 unicode = utf8_to_unicode(string, bytes);
4220 /* FIXME: Graceful handling of invalid unicode character. */
4221 if (!unicode)
4222 break;
4224 ucwidth = unicode_width(unicode);
4225 width += ucwidth;
4226 if (width > max_width) {
4227 *trimmed = 1;
4228 break;
4229 }
4231 /* The column offset collects the differences between the
4232 * number of bytes encoding a character and the number of
4233 * columns will be used for rendering said character.
4234 *
4235 * So if some character A is encoded in 2 bytes, but will be
4236 * represented on the screen using only 1 byte this will and up
4237 * adding 1 to the multi-byte column offset.
4238 *
4239 * Assumes that no double-width character can be encoding in
4240 * less than two bytes. */
4241 if (bytes > ucwidth)
4242 mbwidth += bytes - ucwidth;
4244 string += bytes;
4245 }
4247 *coloffset += mbwidth;
4249 return string - start;
4250 }
4253 /*
4254 * Status management
4255 */
4257 /* Whether or not the curses interface has been initialized. */
4258 static bool cursed = FALSE;
4260 /* The status window is used for polling keystrokes. */
4261 static WINDOW *status_win;
4263 static bool status_empty = TRUE;
4265 /* Update status and title window. */
4266 static void
4267 report(const char *msg, ...)
4268 {
4269 struct view *view = display[current_view];
4271 if (input_mode)
4272 return;
4274 if (!status_empty || *msg) {
4275 va_list args;
4277 va_start(args, msg);
4279 wmove(status_win, 0, 0);
4280 if (*msg) {
4281 vwprintw(status_win, msg, args);
4282 status_empty = FALSE;
4283 } else {
4284 status_empty = TRUE;
4285 }
4286 wclrtoeol(status_win);
4287 wrefresh(status_win);
4289 va_end(args);
4290 }
4292 update_view_title(view);
4293 update_display_cursor(view);
4294 }
4296 /* Controls when nodelay should be in effect when polling user input. */
4297 static void
4298 set_nonblocking_input(bool loading)
4299 {
4300 static unsigned int loading_views;
4302 if ((loading == FALSE && loading_views-- == 1) ||
4303 (loading == TRUE && loading_views++ == 0))
4304 nodelay(status_win, loading);
4305 }
4307 static void
4308 init_display(void)
4309 {
4310 int x, y;
4312 /* Initialize the curses library */
4313 if (isatty(STDIN_FILENO)) {
4314 cursed = !!initscr();
4315 } else {
4316 /* Leave stdin and stdout alone when acting as a pager. */
4317 FILE *io = fopen("/dev/tty", "r+");
4319 if (!io)
4320 die("Failed to open /dev/tty");
4321 cursed = !!newterm(NULL, io, io);
4322 }
4324 if (!cursed)
4325 die("Failed to initialize curses");
4327 nonl(); /* Tell curses not to do NL->CR/NL on output */
4328 cbreak(); /* Take input chars one at a time, no wait for \n */
4329 noecho(); /* Don't echo input */
4330 leaveok(stdscr, TRUE);
4332 if (has_colors())
4333 init_colors();
4335 getmaxyx(stdscr, y, x);
4336 status_win = newwin(1, 0, y - 1, 0);
4337 if (!status_win)
4338 die("Failed to create status window");
4340 /* Enable keyboard mapping */
4341 keypad(status_win, TRUE);
4342 wbkgdset(status_win, get_line_attr(LINE_STATUS));
4343 }
4345 static char *
4346 read_prompt(const char *prompt)
4347 {
4348 enum { READING, STOP, CANCEL } status = READING;
4349 static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
4350 int pos = 0;
4352 while (status == READING) {
4353 struct view *view;
4354 int i, key;
4356 input_mode = TRUE;
4358 foreach_view (view, i)
4359 update_view(view);
4361 input_mode = FALSE;
4363 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
4364 wclrtoeol(status_win);
4366 /* Refresh, accept single keystroke of input */
4367 key = wgetch(status_win);
4368 switch (key) {
4369 case KEY_RETURN:
4370 case KEY_ENTER:
4371 case '\n':
4372 status = pos ? STOP : CANCEL;
4373 break;
4375 case KEY_BACKSPACE:
4376 if (pos > 0)
4377 pos--;
4378 else
4379 status = CANCEL;
4380 break;
4382 case KEY_ESC:
4383 status = CANCEL;
4384 break;
4386 case ERR:
4387 break;
4389 default:
4390 if (pos >= sizeof(buf)) {
4391 report("Input string too long");
4392 return NULL;
4393 }
4395 if (isprint(key))
4396 buf[pos++] = (char) key;
4397 }
4398 }
4400 /* Clear the status window */
4401 status_empty = FALSE;
4402 report("");
4404 if (status == CANCEL)
4405 return NULL;
4407 buf[pos++] = 0;
4409 return buf;
4410 }
4412 /*
4413 * Repository references
4414 */
4416 static struct ref *refs;
4417 static size_t refs_size;
4419 /* Id <-> ref store */
4420 static struct ref ***id_refs;
4421 static size_t id_refs_size;
4423 static struct ref **
4424 get_refs(char *id)
4425 {
4426 struct ref ***tmp_id_refs;
4427 struct ref **ref_list = NULL;
4428 size_t ref_list_size = 0;
4429 size_t i;
4431 for (i = 0; i < id_refs_size; i++)
4432 if (!strcmp(id, id_refs[i][0]->id))
4433 return id_refs[i];
4435 tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
4436 if (!tmp_id_refs)
4437 return NULL;
4439 id_refs = tmp_id_refs;
4441 for (i = 0; i < refs_size; i++) {
4442 struct ref **tmp;
4444 if (strcmp(id, refs[i].id))
4445 continue;
4447 tmp = realloc(ref_list, (ref_list_size + 1) * sizeof(*ref_list));
4448 if (!tmp) {
4449 if (ref_list)
4450 free(ref_list);
4451 return NULL;
4452 }
4454 ref_list = tmp;
4455 if (ref_list_size > 0)
4456 ref_list[ref_list_size - 1]->next = 1;
4457 ref_list[ref_list_size] = &refs[i];
4459 /* XXX: The properties of the commit chains ensures that we can
4460 * safely modify the shared ref. The repo references will
4461 * always be similar for the same id. */
4462 ref_list[ref_list_size]->next = 0;
4463 ref_list_size++;
4464 }
4466 if (ref_list)
4467 id_refs[id_refs_size++] = ref_list;
4469 return ref_list;
4470 }
4472 static int
4473 read_ref(char *id, size_t idlen, char *name, size_t namelen)
4474 {
4475 struct ref *ref;
4476 bool tag = FALSE;
4477 bool remote = FALSE;
4479 if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
4480 /* Commits referenced by tags has "^{}" appended. */
4481 if (name[namelen - 1] != '}')
4482 return OK;
4484 while (namelen > 0 && name[namelen] != '^')
4485 namelen--;
4487 tag = TRUE;
4488 namelen -= STRING_SIZE("refs/tags/");
4489 name += STRING_SIZE("refs/tags/");
4491 } else if (!strncmp(name, "refs/remotes/", STRING_SIZE("refs/remotes/"))) {
4492 remote = TRUE;
4493 namelen -= STRING_SIZE("refs/remotes/");
4494 name += STRING_SIZE("refs/remotes/");
4496 } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
4497 namelen -= STRING_SIZE("refs/heads/");
4498 name += STRING_SIZE("refs/heads/");
4500 } else if (!strcmp(name, "HEAD")) {
4501 return OK;
4502 }
4504 refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
4505 if (!refs)
4506 return ERR;
4508 ref = &refs[refs_size++];
4509 ref->name = malloc(namelen + 1);
4510 if (!ref->name)
4511 return ERR;
4513 strncpy(ref->name, name, namelen);
4514 ref->name[namelen] = 0;
4515 ref->tag = tag;
4516 ref->remote = remote;
4517 string_copy_rev(ref->id, id);
4519 return OK;
4520 }
4522 static int
4523 load_refs(void)
4524 {
4525 const char *cmd_env = getenv("TIG_LS_REMOTE");
4526 const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
4528 return read_properties(popen(cmd, "r"), "\t", read_ref);
4529 }
4531 static int
4532 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
4533 {
4534 if (!strcmp(name, "i18n.commitencoding"))
4535 string_ncopy(opt_encoding, value, valuelen);
4537 if (!strcmp(name, "core.editor"))
4538 string_ncopy(opt_editor, value, valuelen);
4540 return OK;
4541 }
4543 static int
4544 load_repo_config(void)
4545 {
4546 return read_properties(popen(GIT_CONFIG " --list", "r"),
4547 "=", read_repo_config_option);
4548 }
4550 static int
4551 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
4552 {
4553 if (!opt_git_dir[0])
4554 string_ncopy(opt_git_dir, name, namelen);
4555 else
4556 string_ncopy(opt_cdup, name, namelen);
4557 return OK;
4558 }
4560 /* XXX: The line outputted by "--show-cdup" can be empty so the option
4561 * must be the last one! */
4562 static int
4563 load_repo_info(void)
4564 {
4565 return read_properties(popen("git rev-parse --git-dir --show-cdup 2>/dev/null", "r"),
4566 "=", read_repo_info);
4567 }
4569 static int
4570 read_properties(FILE *pipe, const char *separators,
4571 int (*read_property)(char *, size_t, char *, size_t))
4572 {
4573 char buffer[BUFSIZ];
4574 char *name;
4575 int state = OK;
4577 if (!pipe)
4578 return ERR;
4580 while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
4581 char *value;
4582 size_t namelen;
4583 size_t valuelen;
4585 name = chomp_string(name);
4586 namelen = strcspn(name, separators);
4588 if (name[namelen]) {
4589 name[namelen] = 0;
4590 value = chomp_string(name + namelen + 1);
4591 valuelen = strlen(value);
4593 } else {
4594 value = "";
4595 valuelen = 0;
4596 }
4598 state = read_property(name, namelen, value, valuelen);
4599 }
4601 if (state != ERR && ferror(pipe))
4602 state = ERR;
4604 pclose(pipe);
4606 return state;
4607 }
4610 /*
4611 * Main
4612 */
4614 static void __NORETURN
4615 quit(int sig)
4616 {
4617 /* XXX: Restore tty modes and let the OS cleanup the rest! */
4618 if (cursed)
4619 endwin();
4620 exit(0);
4621 }
4623 static void __NORETURN
4624 die(const char *err, ...)
4625 {
4626 va_list args;
4628 endwin();
4630 va_start(args, err);
4631 fputs("tig: ", stderr);
4632 vfprintf(stderr, err, args);
4633 fputs("\n", stderr);
4634 va_end(args);
4636 exit(1);
4637 }
4639 int
4640 main(int argc, char *argv[])
4641 {
4642 struct view *view;
4643 enum request request;
4644 size_t i;
4646 signal(SIGINT, quit);
4648 if (setlocale(LC_ALL, "")) {
4649 char *codeset = nl_langinfo(CODESET);
4651 string_ncopy(opt_codeset, codeset, strlen(codeset));
4652 }
4654 if (load_repo_info() == ERR)
4655 die("Failed to load repo info.");
4657 if (load_options() == ERR)
4658 die("Failed to load user config.");
4660 /* Load the repo config file so options can be overwritten from
4661 * the command line. */
4662 if (load_repo_config() == ERR)
4663 die("Failed to load repo config.");
4665 if (!parse_options(argc, argv))
4666 return 0;
4668 /* Require a git repository unless when running in pager mode. */
4669 if (!opt_git_dir[0])
4670 die("Not a git repository");
4672 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
4673 opt_iconv = iconv_open(opt_codeset, opt_encoding);
4674 if (opt_iconv == ICONV_NONE)
4675 die("Failed to initialize character set conversion");
4676 }
4678 if (load_refs() == ERR)
4679 die("Failed to load refs.");
4681 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
4682 view->cmd_env = getenv(view->cmd_env);
4684 request = opt_request;
4686 init_display();
4688 while (view_driver(display[current_view], request)) {
4689 int key;
4690 int i;
4692 foreach_view (view, i)
4693 update_view(view);
4695 /* Refresh, accept single keystroke of input */
4696 key = wgetch(status_win);
4698 /* wgetch() with nodelay() enabled returns ERR when there's no
4699 * input. */
4700 if (key == ERR) {
4701 request = REQ_NONE;
4702 continue;
4703 }
4705 request = get_keybinding(display[current_view]->keymap, key);
4707 /* Some low-level request handling. This keeps access to
4708 * status_win restricted. */
4709 switch (request) {
4710 case REQ_PROMPT:
4711 {
4712 char *cmd = read_prompt(":");
4714 if (cmd && string_format(opt_cmd, "git %s", cmd)) {
4715 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
4716 opt_request = REQ_VIEW_DIFF;
4717 } else {
4718 opt_request = REQ_VIEW_PAGER;
4719 }
4720 break;
4721 }
4723 request = REQ_NONE;
4724 break;
4725 }
4726 case REQ_SEARCH:
4727 case REQ_SEARCH_BACK:
4728 {
4729 const char *prompt = request == REQ_SEARCH
4730 ? "/" : "?";
4731 char *search = read_prompt(prompt);
4733 if (search)
4734 string_ncopy(opt_search, search, strlen(search));
4735 else
4736 request = REQ_NONE;
4737 break;
4738 }
4739 case REQ_SCREEN_RESIZE:
4740 {
4741 int height, width;
4743 getmaxyx(stdscr, height, width);
4745 /* Resize the status view and let the view driver take
4746 * care of resizing the displayed views. */
4747 wresize(status_win, 1, width);
4748 mvwin(status_win, height - 1, 0);
4749 wrefresh(status_win);
4750 break;
4751 }
4752 default:
4753 break;
4754 }
4755 }
4757 quit(0);
4759 return 0;
4760 }