1 /* Copyright (c) 2006-2008 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 /* ncurses(3): Must be defined to have extended wide-character functions. */
46 #define _XOPEN_SOURCE_EXTENDED
48 #ifdef HAVE_NCURSESW_NCURSES_H
49 #include <ncursesw/ncurses.h>
50 #else
51 #ifdef HAVE_NCURSES_NCURSES_H
52 #include <ncurses/ncurses.h>
53 #else
54 #include <ncurses.h>
55 #endif
56 #endif
58 #if __GNUC__ >= 3
59 #define __NORETURN __attribute__((__noreturn__))
60 #else
61 #define __NORETURN
62 #endif
64 static void __NORETURN die(const char *err, ...);
65 static void warn(const char *msg, ...);
66 static void report(const char *msg, ...);
67 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, size_t, char *, size_t));
68 static void set_nonblocking_input(bool loading);
69 static size_t utf8_length(const char *string, int *width, size_t max_width, int *trimmed, bool reserve);
70 static bool prompt_yesno(const char *prompt);
71 static int load_refs(void);
73 #define ABS(x) ((x) >= 0 ? (x) : -(x))
74 #define MIN(x, y) ((x) < (y) ? (x) : (y))
76 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
77 #define STRING_SIZE(x) (sizeof(x) - 1)
79 #define SIZEOF_STR 1024 /* Default string size. */
80 #define SIZEOF_REF 256 /* Size of symbolic or SHA1 ID. */
81 #define SIZEOF_REV 41 /* Holds a SHA-1 and an ending NUL. */
82 #define SIZEOF_ARG 32 /* Default argument array size. */
84 /* Revision graph */
86 #define REVGRAPH_INIT 'I'
87 #define REVGRAPH_MERGE 'M'
88 #define REVGRAPH_BRANCH '+'
89 #define REVGRAPH_COMMIT '*'
90 #define REVGRAPH_BOUND '^'
92 #define SIZEOF_REVGRAPH 19 /* Size of revision ancestry graphics. */
94 /* This color name can be used to refer to the default term colors. */
95 #define COLOR_DEFAULT (-1)
97 #define ICONV_NONE ((iconv_t) -1)
98 #ifndef ICONV_CONST
99 #define ICONV_CONST /* nothing */
100 #endif
102 /* The format and size of the date column in the main view. */
103 #define DATE_FORMAT "%Y-%m-%d %H:%M"
104 #define DATE_COLS STRING_SIZE("2006-04-29 14:21 ")
106 #define AUTHOR_COLS 20
107 #define ID_COLS 8
109 /* The default interval between line numbers. */
110 #define NUMBER_INTERVAL 5
112 #define TAB_SIZE 8
114 #define SCALE_SPLIT_VIEW(height) ((height) * 2 / 3)
116 #define NULL_ID "0000000000000000000000000000000000000000"
118 #ifndef GIT_CONFIG
119 #define GIT_CONFIG "config"
120 #endif
122 #define TIG_LS_REMOTE \
123 "git ls-remote . 2>/dev/null"
125 #define TIG_DIFF_CMD \
126 "git show --pretty=fuller --no-color --root --patch-with-stat --find-copies-harder -C %s 2>/dev/null"
128 #define TIG_LOG_CMD \
129 "git log --no-color --cc --stat -n100 %s 2>/dev/null"
131 #define TIG_MAIN_BASE \
132 "git log --no-color --pretty=raw --parents --topo-order"
134 #define TIG_MAIN_CMD \
135 TIG_MAIN_BASE " %s 2>/dev/null"
137 #define TIG_TREE_CMD \
138 "git ls-tree %s %s"
140 #define TIG_BLOB_CMD \
141 "git cat-file blob %s"
143 /* XXX: Needs to be defined to the empty string. */
144 #define TIG_HELP_CMD ""
145 #define TIG_PAGER_CMD ""
146 #define TIG_STATUS_CMD ""
147 #define TIG_STAGE_CMD ""
148 #define TIG_BLAME_CMD ""
150 /* Some ascii-shorthands fitted into the ncurses namespace. */
151 #define KEY_TAB '\t'
152 #define KEY_RETURN '\r'
153 #define KEY_ESC 27
156 struct ref {
157 char *name; /* Ref name; tag or head names are shortened. */
158 char id[SIZEOF_REV]; /* Commit SHA1 ID */
159 unsigned int head:1; /* Is it the current HEAD? */
160 unsigned int tag:1; /* Is it a tag? */
161 unsigned int ltag:1; /* If so, is the tag local? */
162 unsigned int remote:1; /* Is it a remote ref? */
163 unsigned int tracked:1; /* Is it the remote for the current HEAD? */
164 unsigned int next:1; /* For ref lists: are there more refs? */
165 };
167 static struct ref **get_refs(const char *id);
169 struct int_map {
170 const char *name;
171 int namelen;
172 int value;
173 };
175 static int
176 set_from_int_map(struct int_map *map, size_t map_size,
177 int *value, const char *name, int namelen)
178 {
180 int i;
182 for (i = 0; i < map_size; i++)
183 if (namelen == map[i].namelen &&
184 !strncasecmp(name, map[i].name, namelen)) {
185 *value = map[i].value;
186 return OK;
187 }
189 return ERR;
190 }
193 /*
194 * String helpers
195 */
197 static inline void
198 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
199 {
200 if (srclen > dstlen - 1)
201 srclen = dstlen - 1;
203 strncpy(dst, src, srclen);
204 dst[srclen] = 0;
205 }
207 /* Shorthands for safely copying into a fixed buffer. */
209 #define string_copy(dst, src) \
210 string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
212 #define string_ncopy(dst, src, srclen) \
213 string_ncopy_do(dst, sizeof(dst), src, srclen)
215 #define string_copy_rev(dst, src) \
216 string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
218 #define string_add(dst, from, src) \
219 string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
221 static char *
222 chomp_string(char *name)
223 {
224 int namelen;
226 while (isspace(*name))
227 name++;
229 namelen = strlen(name) - 1;
230 while (namelen > 0 && isspace(name[namelen]))
231 name[namelen--] = 0;
233 return name;
234 }
236 static bool
237 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
238 {
239 va_list args;
240 size_t pos = bufpos ? *bufpos : 0;
242 va_start(args, fmt);
243 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
244 va_end(args);
246 if (bufpos)
247 *bufpos = pos;
249 return pos >= bufsize ? FALSE : TRUE;
250 }
252 #define string_format(buf, fmt, args...) \
253 string_nformat(buf, sizeof(buf), NULL, fmt, args)
255 #define string_format_from(buf, from, fmt, args...) \
256 string_nformat(buf, sizeof(buf), from, fmt, args)
258 static int
259 string_enum_compare(const char *str1, const char *str2, int len)
260 {
261 size_t i;
263 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
265 /* Diff-Header == DIFF_HEADER */
266 for (i = 0; i < len; i++) {
267 if (toupper(str1[i]) == toupper(str2[i]))
268 continue;
270 if (string_enum_sep(str1[i]) &&
271 string_enum_sep(str2[i]))
272 continue;
274 return str1[i] - str2[i];
275 }
277 return 0;
278 }
280 #define prefixcmp(str1, str2) \
281 strncmp(str1, str2, STRING_SIZE(str2))
283 static inline int
284 suffixcmp(const char *str, int slen, const char *suffix)
285 {
286 size_t len = slen >= 0 ? slen : strlen(str);
287 size_t suffixlen = strlen(suffix);
289 return suffixlen < len ? strcmp(str + len - suffixlen, suffix) : -1;
290 }
292 /* Shell quoting
293 *
294 * NOTE: The following is a slightly modified copy of the git project's shell
295 * quoting routines found in the quote.c file.
296 *
297 * Help to copy the thing properly quoted for the shell safety. any single
298 * quote is replaced with '\'', any exclamation point is replaced with '\!',
299 * and the whole thing is enclosed in a
300 *
301 * E.g.
302 * original sq_quote result
303 * name ==> name ==> 'name'
304 * a b ==> a b ==> 'a b'
305 * a'b ==> a'\''b ==> 'a'\''b'
306 * a!b ==> a'\!'b ==> 'a'\!'b'
307 */
309 static size_t
310 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
311 {
312 char c;
314 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
316 BUFPUT('\'');
317 while ((c = *src++)) {
318 if (c == '\'' || c == '!') {
319 BUFPUT('\'');
320 BUFPUT('\\');
321 BUFPUT(c);
322 BUFPUT('\'');
323 } else {
324 BUFPUT(c);
325 }
326 }
327 BUFPUT('\'');
329 if (bufsize < SIZEOF_STR)
330 buf[bufsize] = 0;
332 return bufsize;
333 }
336 /*
337 * User requests
338 */
340 #define REQ_INFO \
341 /* XXX: Keep the view request first and in sync with views[]. */ \
342 REQ_GROUP("View switching") \
343 REQ_(VIEW_MAIN, "Show main view"), \
344 REQ_(VIEW_DIFF, "Show diff view"), \
345 REQ_(VIEW_LOG, "Show log view"), \
346 REQ_(VIEW_TREE, "Show tree view"), \
347 REQ_(VIEW_BLOB, "Show blob view"), \
348 REQ_(VIEW_BLAME, "Show blame view"), \
349 REQ_(VIEW_HELP, "Show help page"), \
350 REQ_(VIEW_PAGER, "Show pager view"), \
351 REQ_(VIEW_STATUS, "Show status view"), \
352 REQ_(VIEW_STAGE, "Show stage view"), \
353 \
354 REQ_GROUP("View manipulation") \
355 REQ_(ENTER, "Enter current line and scroll"), \
356 REQ_(NEXT, "Move to next"), \
357 REQ_(PREVIOUS, "Move to previous"), \
358 REQ_(VIEW_NEXT, "Move focus to next view"), \
359 REQ_(REFRESH, "Reload and refresh"), \
360 REQ_(MAXIMIZE, "Maximize the current view"), \
361 REQ_(VIEW_CLOSE, "Close the current view"), \
362 REQ_(QUIT, "Close all views and quit"), \
363 \
364 REQ_GROUP("View specific requests") \
365 REQ_(STATUS_UPDATE, "Update file status"), \
366 REQ_(STATUS_REVERT, "Revert file changes"), \
367 REQ_(STATUS_MERGE, "Merge file using external tool"), \
368 REQ_(STAGE_NEXT, "Find next chunk to stage"), \
369 REQ_(TREE_PARENT, "Switch to parent directory in tree view"), \
370 \
371 REQ_GROUP("Cursor navigation") \
372 REQ_(MOVE_UP, "Move cursor one line up"), \
373 REQ_(MOVE_DOWN, "Move cursor one line down"), \
374 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
375 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
376 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
377 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
378 \
379 REQ_GROUP("Scrolling") \
380 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
381 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
382 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
383 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
384 \
385 REQ_GROUP("Searching") \
386 REQ_(SEARCH, "Search the view"), \
387 REQ_(SEARCH_BACK, "Search backwards in the view"), \
388 REQ_(FIND_NEXT, "Find next search match"), \
389 REQ_(FIND_PREV, "Find previous search match"), \
390 \
391 REQ_GROUP("Option manipulation") \
392 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
393 REQ_(TOGGLE_DATE, "Toggle date display"), \
394 REQ_(TOGGLE_AUTHOR, "Toggle author display"), \
395 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization"), \
396 REQ_(TOGGLE_REFS, "Toggle reference display (tags/branches)"), \
397 \
398 REQ_GROUP("Misc") \
399 REQ_(PROMPT, "Bring up the prompt"), \
400 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
401 REQ_(SCREEN_RESIZE, "Resize the screen"), \
402 REQ_(SHOW_VERSION, "Show version information"), \
403 REQ_(STOP_LOADING, "Stop all loading views"), \
404 REQ_(EDIT, "Open in editor"), \
405 REQ_(NONE, "Do nothing")
408 /* User action requests. */
409 enum request {
410 #define REQ_GROUP(help)
411 #define REQ_(req, help) REQ_##req
413 /* Offset all requests to avoid conflicts with ncurses getch values. */
414 REQ_OFFSET = KEY_MAX + 1,
415 REQ_INFO
417 #undef REQ_GROUP
418 #undef REQ_
419 };
421 struct request_info {
422 enum request request;
423 const char *name;
424 int namelen;
425 const char *help;
426 };
428 static struct request_info req_info[] = {
429 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
430 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
431 REQ_INFO
432 #undef REQ_GROUP
433 #undef REQ_
434 };
436 static enum request
437 get_request(const char *name)
438 {
439 int namelen = strlen(name);
440 int i;
442 for (i = 0; i < ARRAY_SIZE(req_info); i++)
443 if (req_info[i].namelen == namelen &&
444 !string_enum_compare(req_info[i].name, name, namelen))
445 return req_info[i].request;
447 return REQ_NONE;
448 }
451 /*
452 * Options
453 */
455 static const char usage[] =
456 "tig " TIG_VERSION " (" __DATE__ ")\n"
457 "\n"
458 "Usage: tig [options] [revs] [--] [paths]\n"
459 " or: tig show [options] [revs] [--] [paths]\n"
460 " or: tig blame [rev] path\n"
461 " or: tig status\n"
462 " or: tig < [git command output]\n"
463 "\n"
464 "Options:\n"
465 " -v, --version Show version and exit\n"
466 " -h, --help Show help message and exit";
468 /* Option and state variables. */
469 static bool opt_date = TRUE;
470 static bool opt_author = TRUE;
471 static bool opt_line_number = FALSE;
472 static bool opt_line_graphics = TRUE;
473 static bool opt_rev_graph = FALSE;
474 static bool opt_show_refs = TRUE;
475 static int opt_num_interval = NUMBER_INTERVAL;
476 static int opt_tab_size = TAB_SIZE;
477 static int opt_author_cols = AUTHOR_COLS-1;
478 static char opt_cmd[SIZEOF_STR] = "";
479 static char opt_path[SIZEOF_STR] = "";
480 static char opt_file[SIZEOF_STR] = "";
481 static char opt_ref[SIZEOF_REF] = "";
482 static char opt_head[SIZEOF_REF] = "";
483 static char opt_head_rev[SIZEOF_REV] = "";
484 static char opt_remote[SIZEOF_REF] = "";
485 static FILE *opt_pipe = NULL;
486 static char opt_encoding[20] = "UTF-8";
487 static bool opt_utf8 = TRUE;
488 static char opt_codeset[20] = "UTF-8";
489 static iconv_t opt_iconv = ICONV_NONE;
490 static char opt_search[SIZEOF_STR] = "";
491 static char opt_cdup[SIZEOF_STR] = "";
492 static char opt_git_dir[SIZEOF_STR] = "";
493 static signed char opt_is_inside_work_tree = -1; /* set to TRUE or FALSE */
494 static char opt_editor[SIZEOF_STR] = "";
495 static FILE *opt_tty = NULL;
497 #define is_initial_commit() (!*opt_head_rev)
498 #define is_head_commit(rev) (!strcmp((rev), "HEAD") || !strcmp(opt_head_rev, (rev)))
500 static enum request
501 parse_options(int argc, const char *argv[])
502 {
503 enum request request = REQ_VIEW_MAIN;
504 size_t buf_size;
505 const char *subcommand;
506 bool seen_dashdash = FALSE;
507 int i;
509 if (!isatty(STDIN_FILENO)) {
510 opt_pipe = stdin;
511 return REQ_VIEW_PAGER;
512 }
514 if (argc <= 1)
515 return REQ_VIEW_MAIN;
517 subcommand = argv[1];
518 if (!strcmp(subcommand, "status") || !strcmp(subcommand, "-S")) {
519 if (!strcmp(subcommand, "-S"))
520 warn("`-S' has been deprecated; use `tig status' instead");
521 if (argc > 2)
522 warn("ignoring arguments after `%s'", subcommand);
523 return REQ_VIEW_STATUS;
525 } else if (!strcmp(subcommand, "blame")) {
526 if (argc <= 2 || argc > 4)
527 die("invalid number of options to blame\n\n%s", usage);
529 i = 2;
530 if (argc == 4) {
531 string_ncopy(opt_ref, argv[i], strlen(argv[i]));
532 i++;
533 }
535 string_ncopy(opt_file, argv[i], strlen(argv[i]));
536 return REQ_VIEW_BLAME;
538 } else if (!strcmp(subcommand, "show")) {
539 request = REQ_VIEW_DIFF;
541 } else if (!strcmp(subcommand, "log") || !strcmp(subcommand, "diff")) {
542 request = subcommand[0] == 'l' ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
543 warn("`tig %s' has been deprecated", subcommand);
545 } else {
546 subcommand = NULL;
547 }
549 if (!subcommand)
550 /* XXX: This is vulnerable to the user overriding
551 * options required for the main view parser. */
552 string_copy(opt_cmd, TIG_MAIN_BASE);
553 else
554 string_format(opt_cmd, "git %s", subcommand);
556 buf_size = strlen(opt_cmd);
558 for (i = 1 + !!subcommand; i < argc; i++) {
559 const char *opt = argv[i];
561 if (seen_dashdash || !strcmp(opt, "--")) {
562 seen_dashdash = TRUE;
564 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
565 printf("tig version %s\n", TIG_VERSION);
566 return REQ_NONE;
568 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
569 printf("%s\n", usage);
570 return REQ_NONE;
571 }
573 opt_cmd[buf_size++] = ' ';
574 buf_size = sq_quote(opt_cmd, buf_size, opt);
575 if (buf_size >= sizeof(opt_cmd))
576 die("command too long");
577 }
579 opt_cmd[buf_size] = 0;
581 return request;
582 }
585 /*
586 * Line-oriented content detection.
587 */
589 #define LINE_INFO \
590 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
591 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
592 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
593 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
594 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
595 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
596 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
597 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
598 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
599 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
600 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
601 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
602 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
603 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
604 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
605 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
606 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
607 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
608 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
609 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
610 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
611 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
612 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
613 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
614 LINE(AUTHOR, "author ", COLOR_CYAN, COLOR_DEFAULT, 0), \
615 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
616 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
617 LINE(ACKED, " Acked-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
618 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
619 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
620 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
621 LINE(DELIMITER, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
622 LINE(DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
623 LINE(LINE_NUMBER, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
624 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
625 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
626 LINE(MAIN_AUTHOR, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
627 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
628 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
629 LINE(MAIN_LOCAL_TAG,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
630 LINE(MAIN_REMOTE, "", COLOR_YELLOW, COLOR_DEFAULT, 0), \
631 LINE(MAIN_TRACKED, "", COLOR_YELLOW, COLOR_DEFAULT, A_BOLD), \
632 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
633 LINE(MAIN_HEAD, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
634 LINE(MAIN_REVGRAPH,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
635 LINE(TREE_DIR, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
636 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
637 LINE(STAT_HEAD, "", COLOR_YELLOW, COLOR_DEFAULT, 0), \
638 LINE(STAT_SECTION, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
639 LINE(STAT_NONE, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
640 LINE(STAT_STAGED, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
641 LINE(STAT_UNSTAGED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
642 LINE(STAT_UNTRACKED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
643 LINE(BLAME_ID, "", COLOR_MAGENTA, COLOR_DEFAULT, 0)
645 enum line_type {
646 #define LINE(type, line, fg, bg, attr) \
647 LINE_##type
648 LINE_INFO,
649 LINE_NONE
650 #undef LINE
651 };
653 struct line_info {
654 const char *name; /* Option name. */
655 int namelen; /* Size of option name. */
656 const char *line; /* The start of line to match. */
657 int linelen; /* Size of string to match. */
658 int fg, bg, attr; /* Color and text attributes for the lines. */
659 };
661 static struct line_info line_info[] = {
662 #define LINE(type, line, fg, bg, attr) \
663 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
664 LINE_INFO
665 #undef LINE
666 };
668 static enum line_type
669 get_line_type(const char *line)
670 {
671 int linelen = strlen(line);
672 enum line_type type;
674 for (type = 0; type < ARRAY_SIZE(line_info); type++)
675 /* Case insensitive search matches Signed-off-by lines better. */
676 if (linelen >= line_info[type].linelen &&
677 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
678 return type;
680 return LINE_DEFAULT;
681 }
683 static inline int
684 get_line_attr(enum line_type type)
685 {
686 assert(type < ARRAY_SIZE(line_info));
687 return COLOR_PAIR(type) | line_info[type].attr;
688 }
690 static struct line_info *
691 get_line_info(const char *name)
692 {
693 size_t namelen = strlen(name);
694 enum line_type type;
696 for (type = 0; type < ARRAY_SIZE(line_info); type++)
697 if (namelen == line_info[type].namelen &&
698 !string_enum_compare(line_info[type].name, name, namelen))
699 return &line_info[type];
701 return NULL;
702 }
704 static void
705 init_colors(void)
706 {
707 int default_bg = line_info[LINE_DEFAULT].bg;
708 int default_fg = line_info[LINE_DEFAULT].fg;
709 enum line_type type;
711 start_color();
713 if (assume_default_colors(default_fg, default_bg) == ERR) {
714 default_bg = COLOR_BLACK;
715 default_fg = COLOR_WHITE;
716 }
718 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
719 struct line_info *info = &line_info[type];
720 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
721 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
723 init_pair(type, fg, bg);
724 }
725 }
727 struct line {
728 enum line_type type;
730 /* State flags */
731 unsigned int selected:1;
732 unsigned int dirty:1;
734 void *data; /* User data */
735 };
738 /*
739 * Keys
740 */
742 struct keybinding {
743 int alias;
744 enum request request;
745 };
747 static struct keybinding default_keybindings[] = {
748 /* View switching */
749 { 'm', REQ_VIEW_MAIN },
750 { 'd', REQ_VIEW_DIFF },
751 { 'l', REQ_VIEW_LOG },
752 { 't', REQ_VIEW_TREE },
753 { 'f', REQ_VIEW_BLOB },
754 { 'B', REQ_VIEW_BLAME },
755 { 'p', REQ_VIEW_PAGER },
756 { 'h', REQ_VIEW_HELP },
757 { 'S', REQ_VIEW_STATUS },
758 { 'c', REQ_VIEW_STAGE },
760 /* View manipulation */
761 { 'q', REQ_VIEW_CLOSE },
762 { KEY_TAB, REQ_VIEW_NEXT },
763 { KEY_RETURN, REQ_ENTER },
764 { KEY_UP, REQ_PREVIOUS },
765 { KEY_DOWN, REQ_NEXT },
766 { 'R', REQ_REFRESH },
767 { KEY_F(5), REQ_REFRESH },
768 { 'O', REQ_MAXIMIZE },
770 /* Cursor navigation */
771 { 'k', REQ_MOVE_UP },
772 { 'j', REQ_MOVE_DOWN },
773 { KEY_HOME, REQ_MOVE_FIRST_LINE },
774 { KEY_END, REQ_MOVE_LAST_LINE },
775 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
776 { ' ', REQ_MOVE_PAGE_DOWN },
777 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
778 { 'b', REQ_MOVE_PAGE_UP },
779 { '-', REQ_MOVE_PAGE_UP },
781 /* Scrolling */
782 { KEY_IC, REQ_SCROLL_LINE_UP },
783 { KEY_DC, REQ_SCROLL_LINE_DOWN },
784 { 'w', REQ_SCROLL_PAGE_UP },
785 { 's', REQ_SCROLL_PAGE_DOWN },
787 /* Searching */
788 { '/', REQ_SEARCH },
789 { '?', REQ_SEARCH_BACK },
790 { 'n', REQ_FIND_NEXT },
791 { 'N', REQ_FIND_PREV },
793 /* Misc */
794 { 'Q', REQ_QUIT },
795 { 'z', REQ_STOP_LOADING },
796 { 'v', REQ_SHOW_VERSION },
797 { 'r', REQ_SCREEN_REDRAW },
798 { '.', REQ_TOGGLE_LINENO },
799 { 'D', REQ_TOGGLE_DATE },
800 { 'A', REQ_TOGGLE_AUTHOR },
801 { 'g', REQ_TOGGLE_REV_GRAPH },
802 { 'F', REQ_TOGGLE_REFS },
803 { ':', REQ_PROMPT },
804 { 'u', REQ_STATUS_UPDATE },
805 { '!', REQ_STATUS_REVERT },
806 { 'M', REQ_STATUS_MERGE },
807 { '@', REQ_STAGE_NEXT },
808 { ',', REQ_TREE_PARENT },
809 { 'e', REQ_EDIT },
811 /* Using the ncurses SIGWINCH handler. */
812 { KEY_RESIZE, REQ_SCREEN_RESIZE },
813 };
815 #define KEYMAP_INFO \
816 KEYMAP_(GENERIC), \
817 KEYMAP_(MAIN), \
818 KEYMAP_(DIFF), \
819 KEYMAP_(LOG), \
820 KEYMAP_(TREE), \
821 KEYMAP_(BLOB), \
822 KEYMAP_(BLAME), \
823 KEYMAP_(PAGER), \
824 KEYMAP_(HELP), \
825 KEYMAP_(STATUS), \
826 KEYMAP_(STAGE)
828 enum keymap {
829 #define KEYMAP_(name) KEYMAP_##name
830 KEYMAP_INFO
831 #undef KEYMAP_
832 };
834 static struct int_map keymap_table[] = {
835 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
836 KEYMAP_INFO
837 #undef KEYMAP_
838 };
840 #define set_keymap(map, name) \
841 set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
843 struct keybinding_table {
844 struct keybinding *data;
845 size_t size;
846 };
848 static struct keybinding_table keybindings[ARRAY_SIZE(keymap_table)];
850 static void
851 add_keybinding(enum keymap keymap, enum request request, int key)
852 {
853 struct keybinding_table *table = &keybindings[keymap];
855 table->data = realloc(table->data, (table->size + 1) * sizeof(*table->data));
856 if (!table->data)
857 die("Failed to allocate keybinding");
858 table->data[table->size].alias = key;
859 table->data[table->size++].request = request;
860 }
862 /* Looks for a key binding first in the given map, then in the generic map, and
863 * lastly in the default keybindings. */
864 static enum request
865 get_keybinding(enum keymap keymap, int key)
866 {
867 size_t i;
869 for (i = 0; i < keybindings[keymap].size; i++)
870 if (keybindings[keymap].data[i].alias == key)
871 return keybindings[keymap].data[i].request;
873 for (i = 0; i < keybindings[KEYMAP_GENERIC].size; i++)
874 if (keybindings[KEYMAP_GENERIC].data[i].alias == key)
875 return keybindings[KEYMAP_GENERIC].data[i].request;
877 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
878 if (default_keybindings[i].alias == key)
879 return default_keybindings[i].request;
881 return (enum request) key;
882 }
885 struct key {
886 const char *name;
887 int value;
888 };
890 static struct key key_table[] = {
891 { "Enter", KEY_RETURN },
892 { "Space", ' ' },
893 { "Backspace", KEY_BACKSPACE },
894 { "Tab", KEY_TAB },
895 { "Escape", KEY_ESC },
896 { "Left", KEY_LEFT },
897 { "Right", KEY_RIGHT },
898 { "Up", KEY_UP },
899 { "Down", KEY_DOWN },
900 { "Insert", KEY_IC },
901 { "Delete", KEY_DC },
902 { "Hash", '#' },
903 { "Home", KEY_HOME },
904 { "End", KEY_END },
905 { "PageUp", KEY_PPAGE },
906 { "PageDown", KEY_NPAGE },
907 { "F1", KEY_F(1) },
908 { "F2", KEY_F(2) },
909 { "F3", KEY_F(3) },
910 { "F4", KEY_F(4) },
911 { "F5", KEY_F(5) },
912 { "F6", KEY_F(6) },
913 { "F7", KEY_F(7) },
914 { "F8", KEY_F(8) },
915 { "F9", KEY_F(9) },
916 { "F10", KEY_F(10) },
917 { "F11", KEY_F(11) },
918 { "F12", KEY_F(12) },
919 };
921 static int
922 get_key_value(const char *name)
923 {
924 int i;
926 for (i = 0; i < ARRAY_SIZE(key_table); i++)
927 if (!strcasecmp(key_table[i].name, name))
928 return key_table[i].value;
930 if (strlen(name) == 1 && isprint(*name))
931 return (int) *name;
933 return ERR;
934 }
936 static const char *
937 get_key_name(int key_value)
938 {
939 static char key_char[] = "'X'";
940 const char *seq = NULL;
941 int key;
943 for (key = 0; key < ARRAY_SIZE(key_table); key++)
944 if (key_table[key].value == key_value)
945 seq = key_table[key].name;
947 if (seq == NULL &&
948 key_value < 127 &&
949 isprint(key_value)) {
950 key_char[1] = (char) key_value;
951 seq = key_char;
952 }
954 return seq ? seq : "(no key)";
955 }
957 static const char *
958 get_key(enum request request)
959 {
960 static char buf[BUFSIZ];
961 size_t pos = 0;
962 char *sep = "";
963 int i;
965 buf[pos] = 0;
967 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
968 struct keybinding *keybinding = &default_keybindings[i];
970 if (keybinding->request != request)
971 continue;
973 if (!string_format_from(buf, &pos, "%s%s", sep,
974 get_key_name(keybinding->alias)))
975 return "Too many keybindings!";
976 sep = ", ";
977 }
979 return buf;
980 }
982 struct run_request {
983 enum keymap keymap;
984 int key;
985 char cmd[SIZEOF_STR];
986 };
988 static struct run_request *run_request;
989 static size_t run_requests;
991 static enum request
992 add_run_request(enum keymap keymap, int key, int argc, const char **argv)
993 {
994 struct run_request *req;
995 char cmd[SIZEOF_STR];
996 size_t bufpos;
998 for (bufpos = 0; argc > 0; argc--, argv++)
999 if (!string_format_from(cmd, &bufpos, "%s ", *argv))
1000 return REQ_NONE;
1002 req = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
1003 if (!req)
1004 return REQ_NONE;
1006 run_request = req;
1007 req = &run_request[run_requests++];
1008 string_copy(req->cmd, cmd);
1009 req->keymap = keymap;
1010 req->key = key;
1012 return REQ_NONE + run_requests;
1013 }
1015 static struct run_request *
1016 get_run_request(enum request request)
1017 {
1018 if (request <= REQ_NONE)
1019 return NULL;
1020 return &run_request[request - REQ_NONE - 1];
1021 }
1023 static void
1024 add_builtin_run_requests(void)
1025 {
1026 struct {
1027 enum keymap keymap;
1028 int key;
1029 const char *argv[1];
1030 } reqs[] = {
1031 { KEYMAP_MAIN, 'C', { "git cherry-pick %(commit)" } },
1032 { KEYMAP_GENERIC, 'G', { "git gc" } },
1033 };
1034 int i;
1036 for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1037 enum request req;
1039 req = add_run_request(reqs[i].keymap, reqs[i].key, 1, reqs[i].argv);
1040 if (req != REQ_NONE)
1041 add_keybinding(reqs[i].keymap, req, reqs[i].key);
1042 }
1043 }
1045 /*
1046 * User config file handling.
1047 */
1049 static struct int_map color_map[] = {
1050 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
1051 COLOR_MAP(DEFAULT),
1052 COLOR_MAP(BLACK),
1053 COLOR_MAP(BLUE),
1054 COLOR_MAP(CYAN),
1055 COLOR_MAP(GREEN),
1056 COLOR_MAP(MAGENTA),
1057 COLOR_MAP(RED),
1058 COLOR_MAP(WHITE),
1059 COLOR_MAP(YELLOW),
1060 };
1062 #define set_color(color, name) \
1063 set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
1065 static struct int_map attr_map[] = {
1066 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
1067 ATTR_MAP(NORMAL),
1068 ATTR_MAP(BLINK),
1069 ATTR_MAP(BOLD),
1070 ATTR_MAP(DIM),
1071 ATTR_MAP(REVERSE),
1072 ATTR_MAP(STANDOUT),
1073 ATTR_MAP(UNDERLINE),
1074 };
1076 #define set_attribute(attr, name) \
1077 set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
1079 static int config_lineno;
1080 static bool config_errors;
1081 static const char *config_msg;
1083 /* Wants: object fgcolor bgcolor [attr] */
1084 static int
1085 option_color_command(int argc, const char *argv[])
1086 {
1087 struct line_info *info;
1089 if (argc != 3 && argc != 4) {
1090 config_msg = "Wrong number of arguments given to color command";
1091 return ERR;
1092 }
1094 info = get_line_info(argv[0]);
1095 if (!info) {
1096 if (!string_enum_compare(argv[0], "main-delim", strlen("main-delim"))) {
1097 info = get_line_info("delimiter");
1099 } else if (!string_enum_compare(argv[0], "main-date", strlen("main-date"))) {
1100 info = get_line_info("date");
1102 } else {
1103 config_msg = "Unknown color name";
1104 return ERR;
1105 }
1106 }
1108 if (set_color(&info->fg, argv[1]) == ERR ||
1109 set_color(&info->bg, argv[2]) == ERR) {
1110 config_msg = "Unknown color";
1111 return ERR;
1112 }
1114 if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
1115 config_msg = "Unknown attribute";
1116 return ERR;
1117 }
1119 return OK;
1120 }
1122 static bool parse_bool(const char *s)
1123 {
1124 return (!strcmp(s, "1") || !strcmp(s, "true") ||
1125 !strcmp(s, "yes")) ? TRUE : FALSE;
1126 }
1128 static int
1129 parse_int(const char *s, int default_value, int min, int max)
1130 {
1131 int value = atoi(s);
1133 return (value < min || value > max) ? default_value : value;
1134 }
1136 /* Wants: name = value */
1137 static int
1138 option_set_command(int argc, const char *argv[])
1139 {
1140 if (argc != 3) {
1141 config_msg = "Wrong number of arguments given to set command";
1142 return ERR;
1143 }
1145 if (strcmp(argv[1], "=")) {
1146 config_msg = "No value assigned";
1147 return ERR;
1148 }
1150 if (!strcmp(argv[0], "show-author")) {
1151 opt_author = parse_bool(argv[2]);
1152 return OK;
1153 }
1155 if (!strcmp(argv[0], "show-date")) {
1156 opt_date = parse_bool(argv[2]);
1157 return OK;
1158 }
1160 if (!strcmp(argv[0], "show-rev-graph")) {
1161 opt_rev_graph = parse_bool(argv[2]);
1162 return OK;
1163 }
1165 if (!strcmp(argv[0], "show-refs")) {
1166 opt_show_refs = parse_bool(argv[2]);
1167 return OK;
1168 }
1170 if (!strcmp(argv[0], "show-line-numbers")) {
1171 opt_line_number = parse_bool(argv[2]);
1172 return OK;
1173 }
1175 if (!strcmp(argv[0], "line-graphics")) {
1176 opt_line_graphics = parse_bool(argv[2]);
1177 return OK;
1178 }
1180 if (!strcmp(argv[0], "line-number-interval")) {
1181 opt_num_interval = parse_int(argv[2], opt_num_interval, 1, 1024);
1182 return OK;
1183 }
1185 if (!strcmp(argv[0], "author-width")) {
1186 opt_author_cols = parse_int(argv[2], opt_author_cols, 0, 1024);
1187 return OK;
1188 }
1190 if (!strcmp(argv[0], "tab-size")) {
1191 opt_tab_size = parse_int(argv[2], opt_tab_size, 1, 1024);
1192 return OK;
1193 }
1195 if (!strcmp(argv[0], "commit-encoding")) {
1196 const char *arg = argv[2];
1197 int arglen = strlen(arg);
1199 switch (arg[0]) {
1200 case '"':
1201 case '\'':
1202 if (arglen == 1 || arg[arglen - 1] != arg[0]) {
1203 config_msg = "Unmatched quotation";
1204 return ERR;
1205 }
1206 arg += 1; arglen -= 2;
1207 default:
1208 string_ncopy(opt_encoding, arg, strlen(arg));
1209 return OK;
1210 }
1211 }
1213 config_msg = "Unknown variable name";
1214 return ERR;
1215 }
1217 /* Wants: mode request key */
1218 static int
1219 option_bind_command(int argc, const char *argv[])
1220 {
1221 enum request request;
1222 int keymap;
1223 int key;
1225 if (argc < 3) {
1226 config_msg = "Wrong number of arguments given to bind command";
1227 return ERR;
1228 }
1230 if (set_keymap(&keymap, argv[0]) == ERR) {
1231 config_msg = "Unknown key map";
1232 return ERR;
1233 }
1235 key = get_key_value(argv[1]);
1236 if (key == ERR) {
1237 config_msg = "Unknown key";
1238 return ERR;
1239 }
1241 request = get_request(argv[2]);
1242 if (request == REQ_NONE) {
1243 const char *obsolete[] = { "cherry-pick" };
1244 size_t namelen = strlen(argv[2]);
1245 int i;
1247 for (i = 0; i < ARRAY_SIZE(obsolete); i++) {
1248 if (namelen == strlen(obsolete[i]) &&
1249 !string_enum_compare(obsolete[i], argv[2], namelen)) {
1250 config_msg = "Obsolete request name";
1251 return ERR;
1252 }
1253 }
1254 }
1255 if (request == REQ_NONE && *argv[2]++ == '!')
1256 request = add_run_request(keymap, key, argc - 2, argv + 2);
1257 if (request == REQ_NONE) {
1258 config_msg = "Unknown request name";
1259 return ERR;
1260 }
1262 add_keybinding(keymap, request, key);
1264 return OK;
1265 }
1267 static int
1268 set_option(const char *opt, char *value)
1269 {
1270 const char *argv[SIZEOF_ARG];
1271 int valuelen;
1272 int argc = 0;
1274 /* Tokenize */
1275 while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1276 argv[argc++] = value;
1277 value += valuelen;
1279 /* Nothing more to tokenize or last available token. */
1280 if (!*value || argc >= ARRAY_SIZE(argv))
1281 break;
1283 *value++ = 0;
1284 while (isspace(*value))
1285 value++;
1286 }
1288 if (!strcmp(opt, "color"))
1289 return option_color_command(argc, argv);
1291 if (!strcmp(opt, "set"))
1292 return option_set_command(argc, argv);
1294 if (!strcmp(opt, "bind"))
1295 return option_bind_command(argc, argv);
1297 config_msg = "Unknown option command";
1298 return ERR;
1299 }
1301 static int
1302 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1303 {
1304 int status = OK;
1306 config_lineno++;
1307 config_msg = "Internal error";
1309 /* Check for comment markers, since read_properties() will
1310 * only ensure opt and value are split at first " \t". */
1311 optlen = strcspn(opt, "#");
1312 if (optlen == 0)
1313 return OK;
1315 if (opt[optlen] != 0) {
1316 config_msg = "No option value";
1317 status = ERR;
1319 } else {
1320 /* Look for comment endings in the value. */
1321 size_t len = strcspn(value, "#");
1323 if (len < valuelen) {
1324 valuelen = len;
1325 value[valuelen] = 0;
1326 }
1328 status = set_option(opt, value);
1329 }
1331 if (status == ERR) {
1332 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1333 config_lineno, (int) optlen, opt, config_msg);
1334 config_errors = TRUE;
1335 }
1337 /* Always keep going if errors are encountered. */
1338 return OK;
1339 }
1341 static void
1342 load_option_file(const char *path)
1343 {
1344 FILE *file;
1346 /* It's ok that the file doesn't exist. */
1347 file = fopen(path, "r");
1348 if (!file)
1349 return;
1351 config_lineno = 0;
1352 config_errors = FALSE;
1354 if (read_properties(file, " \t", read_option) == ERR ||
1355 config_errors == TRUE)
1356 fprintf(stderr, "Errors while loading %s.\n", path);
1357 }
1359 static int
1360 load_options(void)
1361 {
1362 const char *home = getenv("HOME");
1363 const char *tigrc_user = getenv("TIGRC_USER");
1364 const char *tigrc_system = getenv("TIGRC_SYSTEM");
1365 char buf[SIZEOF_STR];
1367 add_builtin_run_requests();
1369 if (!tigrc_system) {
1370 if (!string_format(buf, "%s/tigrc", SYSCONFDIR))
1371 return ERR;
1372 tigrc_system = buf;
1373 }
1374 load_option_file(tigrc_system);
1376 if (!tigrc_user) {
1377 if (!home || !string_format(buf, "%s/.tigrc", home))
1378 return ERR;
1379 tigrc_user = buf;
1380 }
1381 load_option_file(tigrc_user);
1383 return OK;
1384 }
1387 /*
1388 * The viewer
1389 */
1391 struct view;
1392 struct view_ops;
1394 /* The display array of active views and the index of the current view. */
1395 static struct view *display[2];
1396 static unsigned int current_view;
1398 /* Reading from the prompt? */
1399 static bool input_mode = FALSE;
1401 #define foreach_displayed_view(view, i) \
1402 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1404 #define displayed_views() (display[1] != NULL ? 2 : 1)
1406 /* Current head and commit ID */
1407 static char ref_blob[SIZEOF_REF] = "";
1408 static char ref_commit[SIZEOF_REF] = "HEAD";
1409 static char ref_head[SIZEOF_REF] = "HEAD";
1411 struct view {
1412 const char *name; /* View name */
1413 const char *cmd_fmt; /* Default command line format */
1414 const char *cmd_env; /* Command line set via environment */
1415 const char *id; /* Points to either of ref_{head,commit,blob} */
1417 struct view_ops *ops; /* View operations */
1419 enum keymap keymap; /* What keymap does this view have */
1420 bool git_dir; /* Whether the view requires a git directory. */
1422 char cmd[SIZEOF_STR]; /* Command buffer */
1423 char ref[SIZEOF_REF]; /* Hovered commit reference */
1424 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
1426 int height, width; /* The width and height of the main window */
1427 WINDOW *win; /* The main window */
1428 WINDOW *title; /* The title window living below the main window */
1430 /* Navigation */
1431 unsigned long offset; /* Offset of the window top */
1432 unsigned long lineno; /* Current line number */
1434 /* Searching */
1435 char grep[SIZEOF_STR]; /* Search string */
1436 regex_t *regex; /* Pre-compiled regex */
1438 /* If non-NULL, points to the view that opened this view. If this view
1439 * is closed tig will switch back to the parent view. */
1440 struct view *parent;
1442 /* Buffering */
1443 size_t lines; /* Total number of lines */
1444 struct line *line; /* Line index */
1445 size_t line_alloc; /* Total number of allocated lines */
1446 size_t line_size; /* Total number of used lines */
1447 unsigned int digits; /* Number of digits in the lines member. */
1449 /* Drawing */
1450 struct line *curline; /* Line currently being drawn. */
1451 enum line_type curtype; /* Attribute currently used for drawing. */
1452 unsigned long col; /* Column when drawing. */
1454 /* Loading */
1455 FILE *pipe;
1456 time_t start_time;
1457 };
1459 struct view_ops {
1460 /* What type of content being displayed. Used in the title bar. */
1461 const char *type;
1462 /* Open and reads in all view content. */
1463 bool (*open)(struct view *view);
1464 /* Read one line; updates view->line. */
1465 bool (*read)(struct view *view, char *data);
1466 /* Draw one line; @lineno must be < view->height. */
1467 bool (*draw)(struct view *view, struct line *line, unsigned int lineno);
1468 /* Depending on view handle a special requests. */
1469 enum request (*request)(struct view *view, enum request request, struct line *line);
1470 /* Search for regex in a line. */
1471 bool (*grep)(struct view *view, struct line *line);
1472 /* Select line */
1473 void (*select)(struct view *view, struct line *line);
1474 };
1476 static struct view_ops blame_ops;
1477 static struct view_ops blob_ops;
1478 static struct view_ops help_ops;
1479 static struct view_ops log_ops;
1480 static struct view_ops main_ops;
1481 static struct view_ops pager_ops;
1482 static struct view_ops stage_ops;
1483 static struct view_ops status_ops;
1484 static struct view_ops tree_ops;
1486 #define VIEW_STR(name, cmd, env, ref, ops, map, git) \
1487 { name, cmd, #env, ref, ops, map, git }
1489 #define VIEW_(id, name, ops, git, ref) \
1490 VIEW_STR(name, TIG_##id##_CMD, TIG_##id##_CMD, ref, ops, KEYMAP_##id, git)
1493 static struct view views[] = {
1494 VIEW_(MAIN, "main", &main_ops, TRUE, ref_head),
1495 VIEW_(DIFF, "diff", &pager_ops, TRUE, ref_commit),
1496 VIEW_(LOG, "log", &log_ops, TRUE, ref_head),
1497 VIEW_(TREE, "tree", &tree_ops, TRUE, ref_commit),
1498 VIEW_(BLOB, "blob", &blob_ops, TRUE, ref_blob),
1499 VIEW_(BLAME, "blame", &blame_ops, TRUE, ref_commit),
1500 VIEW_(HELP, "help", &help_ops, FALSE, ""),
1501 VIEW_(PAGER, "pager", &pager_ops, FALSE, "stdin"),
1502 VIEW_(STATUS, "status", &status_ops, TRUE, ""),
1503 VIEW_(STAGE, "stage", &stage_ops, TRUE, ""),
1504 };
1506 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1507 #define VIEW_REQ(view) ((view) - views + REQ_OFFSET + 1)
1509 #define foreach_view(view, i) \
1510 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1512 #define view_is_displayed(view) \
1513 (view == display[0] || view == display[1])
1516 enum line_graphic {
1517 LINE_GRAPHIC_VLINE
1518 };
1520 static int line_graphics[] = {
1521 /* LINE_GRAPHIC_VLINE: */ '|'
1522 };
1524 static inline void
1525 set_view_attr(struct view *view, enum line_type type)
1526 {
1527 if (!view->curline->selected && view->curtype != type) {
1528 wattrset(view->win, get_line_attr(type));
1529 wchgat(view->win, -1, 0, type, NULL);
1530 view->curtype = type;
1531 }
1532 }
1534 static int
1535 draw_chars(struct view *view, enum line_type type, const char *string,
1536 int max_len, bool use_tilde)
1537 {
1538 int len = 0;
1539 int col = 0;
1540 int trimmed = FALSE;
1542 if (max_len <= 0)
1543 return 0;
1545 if (opt_utf8) {
1546 len = utf8_length(string, &col, max_len, &trimmed, use_tilde);
1547 } else {
1548 col = len = strlen(string);
1549 if (len > max_len) {
1550 if (use_tilde) {
1551 max_len -= 1;
1552 }
1553 col = len = max_len;
1554 trimmed = TRUE;
1555 }
1556 }
1558 set_view_attr(view, type);
1559 waddnstr(view->win, string, len);
1560 if (trimmed && use_tilde) {
1561 set_view_attr(view, LINE_DELIMITER);
1562 waddch(view->win, '~');
1563 col++;
1564 }
1566 return col;
1567 }
1569 static int
1570 draw_space(struct view *view, enum line_type type, int max, int spaces)
1571 {
1572 static char space[] = " ";
1573 int col = 0;
1575 spaces = MIN(max, spaces);
1577 while (spaces > 0) {
1578 int len = MIN(spaces, sizeof(space) - 1);
1580 col += draw_chars(view, type, space, spaces, FALSE);
1581 spaces -= len;
1582 }
1584 return col;
1585 }
1587 static bool
1588 draw_lineno(struct view *view, unsigned int lineno)
1589 {
1590 char number[10];
1591 int digits3 = view->digits < 3 ? 3 : view->digits;
1592 int max_number = MIN(digits3, STRING_SIZE(number));
1593 int max = view->width - view->col;
1594 int col;
1596 if (max < max_number)
1597 max_number = max;
1599 lineno += view->offset + 1;
1600 if (lineno == 1 || (lineno % opt_num_interval) == 0) {
1601 static char fmt[] = "%1ld";
1603 if (view->digits <= 9)
1604 fmt[1] = '0' + digits3;
1606 if (!string_format(number, fmt, lineno))
1607 number[0] = 0;
1608 col = draw_chars(view, LINE_LINE_NUMBER, number, max_number, TRUE);
1609 } else {
1610 col = draw_space(view, LINE_LINE_NUMBER, max_number, max_number);
1611 }
1613 if (col < max) {
1614 set_view_attr(view, LINE_DEFAULT);
1615 waddch(view->win, line_graphics[LINE_GRAPHIC_VLINE]);
1616 col++;
1617 }
1619 if (col < max)
1620 col += draw_space(view, LINE_DEFAULT, max - col, 1);
1621 view->col += col;
1623 return view->width - view->col <= 0;
1624 }
1626 static bool
1627 draw_text(struct view *view, enum line_type type, const char *string, bool trim)
1628 {
1629 view->col += draw_chars(view, type, string, view->width - view->col, trim);
1630 return view->width - view->col <= 0;
1631 }
1633 static bool
1634 draw_graphic(struct view *view, enum line_type type, chtype graphic[], size_t size)
1635 {
1636 int max = view->width - view->col;
1637 int i;
1639 if (max < size)
1640 size = max;
1642 set_view_attr(view, type);
1643 /* Using waddch() instead of waddnstr() ensures that
1644 * they'll be rendered correctly for the cursor line. */
1645 for (i = 0; i < size; i++)
1646 waddch(view->win, graphic[i]);
1648 view->col += size;
1649 if (size < max) {
1650 waddch(view->win, ' ');
1651 view->col++;
1652 }
1654 return view->width - view->col <= 0;
1655 }
1657 static bool
1658 draw_field(struct view *view, enum line_type type, const char *text, int len, bool trim)
1659 {
1660 int max = MIN(view->width - view->col, len);
1661 int col;
1663 if (text)
1664 col = draw_chars(view, type, text, max - 1, trim);
1665 else
1666 col = draw_space(view, type, max - 1, max - 1);
1668 view->col += col + draw_space(view, LINE_DEFAULT, max - col, max - col);
1669 return view->width - view->col <= 0;
1670 }
1672 static bool
1673 draw_date(struct view *view, struct tm *time)
1674 {
1675 char buf[DATE_COLS];
1676 char *date;
1677 int timelen = 0;
1679 if (time)
1680 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, time);
1681 date = timelen ? buf : NULL;
1683 return draw_field(view, LINE_DATE, date, DATE_COLS, FALSE);
1684 }
1686 static bool
1687 draw_view_line(struct view *view, unsigned int lineno)
1688 {
1689 struct line *line;
1690 bool selected = (view->offset + lineno == view->lineno);
1691 bool draw_ok;
1693 assert(view_is_displayed(view));
1695 if (view->offset + lineno >= view->lines)
1696 return FALSE;
1698 line = &view->line[view->offset + lineno];
1700 wmove(view->win, lineno, 0);
1701 view->col = 0;
1702 view->curline = line;
1703 view->curtype = LINE_NONE;
1704 line->selected = FALSE;
1706 if (selected) {
1707 set_view_attr(view, LINE_CURSOR);
1708 line->selected = TRUE;
1709 view->ops->select(view, line);
1710 } else if (line->selected) {
1711 wclrtoeol(view->win);
1712 }
1714 scrollok(view->win, FALSE);
1715 draw_ok = view->ops->draw(view, line, lineno);
1716 scrollok(view->win, TRUE);
1718 return draw_ok;
1719 }
1721 static void
1722 redraw_view_dirty(struct view *view)
1723 {
1724 bool dirty = FALSE;
1725 int lineno;
1727 for (lineno = 0; lineno < view->height; lineno++) {
1728 struct line *line = &view->line[view->offset + lineno];
1730 if (!line->dirty)
1731 continue;
1732 line->dirty = 0;
1733 dirty = TRUE;
1734 if (!draw_view_line(view, lineno))
1735 break;
1736 }
1738 if (!dirty)
1739 return;
1740 redrawwin(view->win);
1741 if (input_mode)
1742 wnoutrefresh(view->win);
1743 else
1744 wrefresh(view->win);
1745 }
1747 static void
1748 redraw_view_from(struct view *view, int lineno)
1749 {
1750 assert(0 <= lineno && lineno < view->height);
1752 for (; lineno < view->height; lineno++) {
1753 if (!draw_view_line(view, lineno))
1754 break;
1755 }
1757 redrawwin(view->win);
1758 if (input_mode)
1759 wnoutrefresh(view->win);
1760 else
1761 wrefresh(view->win);
1762 }
1764 static void
1765 redraw_view(struct view *view)
1766 {
1767 wclear(view->win);
1768 redraw_view_from(view, 0);
1769 }
1772 static void
1773 update_view_title(struct view *view)
1774 {
1775 char buf[SIZEOF_STR];
1776 char state[SIZEOF_STR];
1777 size_t bufpos = 0, statelen = 0;
1779 assert(view_is_displayed(view));
1781 if (view != VIEW(REQ_VIEW_STATUS) && (view->lines || view->pipe)) {
1782 unsigned int view_lines = view->offset + view->height;
1783 unsigned int lines = view->lines
1784 ? MIN(view_lines, view->lines) * 100 / view->lines
1785 : 0;
1787 string_format_from(state, &statelen, "- %s %d of %d (%d%%)",
1788 view->ops->type,
1789 view->lineno + 1,
1790 view->lines,
1791 lines);
1793 if (view->pipe) {
1794 time_t secs = time(NULL) - view->start_time;
1796 /* Three git seconds are a long time ... */
1797 if (secs > 2)
1798 string_format_from(state, &statelen, " %lds", secs);
1799 }
1800 }
1802 string_format_from(buf, &bufpos, "[%s]", view->name);
1803 if (*view->ref && bufpos < view->width) {
1804 size_t refsize = strlen(view->ref);
1805 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
1807 if (minsize < view->width)
1808 refsize = view->width - minsize + 7;
1809 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
1810 }
1812 if (statelen && bufpos < view->width) {
1813 string_format_from(buf, &bufpos, " %s", state);
1814 }
1816 if (view == display[current_view])
1817 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1818 else
1819 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1821 mvwaddnstr(view->title, 0, 0, buf, bufpos);
1822 wclrtoeol(view->title);
1823 wmove(view->title, 0, view->width - 1);
1825 if (input_mode)
1826 wnoutrefresh(view->title);
1827 else
1828 wrefresh(view->title);
1829 }
1831 static void
1832 resize_display(void)
1833 {
1834 int offset, i;
1835 struct view *base = display[0];
1836 struct view *view = display[1] ? display[1] : display[0];
1838 /* Setup window dimensions */
1840 getmaxyx(stdscr, base->height, base->width);
1842 /* Make room for the status window. */
1843 base->height -= 1;
1845 if (view != base) {
1846 /* Horizontal split. */
1847 view->width = base->width;
1848 view->height = SCALE_SPLIT_VIEW(base->height);
1849 base->height -= view->height;
1851 /* Make room for the title bar. */
1852 view->height -= 1;
1853 }
1855 /* Make room for the title bar. */
1856 base->height -= 1;
1858 offset = 0;
1860 foreach_displayed_view (view, i) {
1861 if (!view->win) {
1862 view->win = newwin(view->height, 0, offset, 0);
1863 if (!view->win)
1864 die("Failed to create %s view", view->name);
1866 scrollok(view->win, TRUE);
1868 view->title = newwin(1, 0, offset + view->height, 0);
1869 if (!view->title)
1870 die("Failed to create title window");
1872 } else {
1873 wresize(view->win, view->height, view->width);
1874 mvwin(view->win, offset, 0);
1875 mvwin(view->title, offset + view->height, 0);
1876 }
1878 offset += view->height + 1;
1879 }
1880 }
1882 static void
1883 redraw_display(void)
1884 {
1885 struct view *view;
1886 int i;
1888 foreach_displayed_view (view, i) {
1889 redraw_view(view);
1890 update_view_title(view);
1891 }
1892 }
1894 static void
1895 update_display_cursor(struct view *view)
1896 {
1897 /* Move the cursor to the right-most column of the cursor line.
1898 *
1899 * XXX: This could turn out to be a bit expensive, but it ensures that
1900 * the cursor does not jump around. */
1901 if (view->lines) {
1902 wmove(view->win, view->lineno - view->offset, view->width - 1);
1903 wrefresh(view->win);
1904 }
1905 }
1907 /*
1908 * Navigation
1909 */
1911 /* Scrolling backend */
1912 static void
1913 do_scroll_view(struct view *view, int lines)
1914 {
1915 bool redraw_current_line = FALSE;
1917 /* The rendering expects the new offset. */
1918 view->offset += lines;
1920 assert(0 <= view->offset && view->offset < view->lines);
1921 assert(lines);
1923 /* Move current line into the view. */
1924 if (view->lineno < view->offset) {
1925 view->lineno = view->offset;
1926 redraw_current_line = TRUE;
1927 } else if (view->lineno >= view->offset + view->height) {
1928 view->lineno = view->offset + view->height - 1;
1929 redraw_current_line = TRUE;
1930 }
1932 assert(view->offset <= view->lineno && view->lineno < view->lines);
1934 /* Redraw the whole screen if scrolling is pointless. */
1935 if (view->height < ABS(lines)) {
1936 redraw_view(view);
1938 } else {
1939 int line = lines > 0 ? view->height - lines : 0;
1940 int end = line + ABS(lines);
1942 wscrl(view->win, lines);
1944 for (; line < end; line++) {
1945 if (!draw_view_line(view, line))
1946 break;
1947 }
1949 if (redraw_current_line)
1950 draw_view_line(view, view->lineno - view->offset);
1951 }
1953 redrawwin(view->win);
1954 wrefresh(view->win);
1955 report("");
1956 }
1958 /* Scroll frontend */
1959 static void
1960 scroll_view(struct view *view, enum request request)
1961 {
1962 int lines = 1;
1964 assert(view_is_displayed(view));
1966 switch (request) {
1967 case REQ_SCROLL_PAGE_DOWN:
1968 lines = view->height;
1969 case REQ_SCROLL_LINE_DOWN:
1970 if (view->offset + lines > view->lines)
1971 lines = view->lines - view->offset;
1973 if (lines == 0 || view->offset + view->height >= view->lines) {
1974 report("Cannot scroll beyond the last line");
1975 return;
1976 }
1977 break;
1979 case REQ_SCROLL_PAGE_UP:
1980 lines = view->height;
1981 case REQ_SCROLL_LINE_UP:
1982 if (lines > view->offset)
1983 lines = view->offset;
1985 if (lines == 0) {
1986 report("Cannot scroll beyond the first line");
1987 return;
1988 }
1990 lines = -lines;
1991 break;
1993 default:
1994 die("request %d not handled in switch", request);
1995 }
1997 do_scroll_view(view, lines);
1998 }
2000 /* Cursor moving */
2001 static void
2002 move_view(struct view *view, enum request request)
2003 {
2004 int scroll_steps = 0;
2005 int steps;
2007 switch (request) {
2008 case REQ_MOVE_FIRST_LINE:
2009 steps = -view->lineno;
2010 break;
2012 case REQ_MOVE_LAST_LINE:
2013 steps = view->lines - view->lineno - 1;
2014 break;
2016 case REQ_MOVE_PAGE_UP:
2017 steps = view->height > view->lineno
2018 ? -view->lineno : -view->height;
2019 break;
2021 case REQ_MOVE_PAGE_DOWN:
2022 steps = view->lineno + view->height >= view->lines
2023 ? view->lines - view->lineno - 1 : view->height;
2024 break;
2026 case REQ_MOVE_UP:
2027 steps = -1;
2028 break;
2030 case REQ_MOVE_DOWN:
2031 steps = 1;
2032 break;
2034 default:
2035 die("request %d not handled in switch", request);
2036 }
2038 if (steps <= 0 && view->lineno == 0) {
2039 report("Cannot move beyond the first line");
2040 return;
2042 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
2043 report("Cannot move beyond the last line");
2044 return;
2045 }
2047 /* Move the current line */
2048 view->lineno += steps;
2049 assert(0 <= view->lineno && view->lineno < view->lines);
2051 /* Check whether the view needs to be scrolled */
2052 if (view->lineno < view->offset ||
2053 view->lineno >= view->offset + view->height) {
2054 scroll_steps = steps;
2055 if (steps < 0 && -steps > view->offset) {
2056 scroll_steps = -view->offset;
2058 } else if (steps > 0) {
2059 if (view->lineno == view->lines - 1 &&
2060 view->lines > view->height) {
2061 scroll_steps = view->lines - view->offset - 1;
2062 if (scroll_steps >= view->height)
2063 scroll_steps -= view->height - 1;
2064 }
2065 }
2066 }
2068 if (!view_is_displayed(view)) {
2069 view->offset += scroll_steps;
2070 assert(0 <= view->offset && view->offset < view->lines);
2071 view->ops->select(view, &view->line[view->lineno]);
2072 return;
2073 }
2075 /* Repaint the old "current" line if we be scrolling */
2076 if (ABS(steps) < view->height)
2077 draw_view_line(view, view->lineno - steps - view->offset);
2079 if (scroll_steps) {
2080 do_scroll_view(view, scroll_steps);
2081 return;
2082 }
2084 /* Draw the current line */
2085 draw_view_line(view, view->lineno - view->offset);
2087 redrawwin(view->win);
2088 wrefresh(view->win);
2089 report("");
2090 }
2093 /*
2094 * Searching
2095 */
2097 static void search_view(struct view *view, enum request request);
2099 static bool
2100 find_next_line(struct view *view, unsigned long lineno, struct line *line)
2101 {
2102 assert(view_is_displayed(view));
2104 if (!view->ops->grep(view, line))
2105 return FALSE;
2107 if (lineno - view->offset >= view->height) {
2108 view->offset = lineno;
2109 view->lineno = lineno;
2110 redraw_view(view);
2112 } else {
2113 unsigned long old_lineno = view->lineno - view->offset;
2115 view->lineno = lineno;
2116 draw_view_line(view, old_lineno);
2118 draw_view_line(view, view->lineno - view->offset);
2119 redrawwin(view->win);
2120 wrefresh(view->win);
2121 }
2123 report("Line %ld matches '%s'", lineno + 1, view->grep);
2124 return TRUE;
2125 }
2127 static void
2128 find_next(struct view *view, enum request request)
2129 {
2130 unsigned long lineno = view->lineno;
2131 int direction;
2133 if (!*view->grep) {
2134 if (!*opt_search)
2135 report("No previous search");
2136 else
2137 search_view(view, request);
2138 return;
2139 }
2141 switch (request) {
2142 case REQ_SEARCH:
2143 case REQ_FIND_NEXT:
2144 direction = 1;
2145 break;
2147 case REQ_SEARCH_BACK:
2148 case REQ_FIND_PREV:
2149 direction = -1;
2150 break;
2152 default:
2153 return;
2154 }
2156 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
2157 lineno += direction;
2159 /* Note, lineno is unsigned long so will wrap around in which case it
2160 * will become bigger than view->lines. */
2161 for (; lineno < view->lines; lineno += direction) {
2162 struct line *line = &view->line[lineno];
2164 if (find_next_line(view, lineno, line))
2165 return;
2166 }
2168 report("No match found for '%s'", view->grep);
2169 }
2171 static void
2172 search_view(struct view *view, enum request request)
2173 {
2174 int regex_err;
2176 if (view->regex) {
2177 regfree(view->regex);
2178 *view->grep = 0;
2179 } else {
2180 view->regex = calloc(1, sizeof(*view->regex));
2181 if (!view->regex)
2182 return;
2183 }
2185 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
2186 if (regex_err != 0) {
2187 char buf[SIZEOF_STR] = "unknown error";
2189 regerror(regex_err, view->regex, buf, sizeof(buf));
2190 report("Search failed: %s", buf);
2191 return;
2192 }
2194 string_copy(view->grep, opt_search);
2196 find_next(view, request);
2197 }
2199 /*
2200 * Incremental updating
2201 */
2203 static void
2204 reset_view(struct view *view)
2205 {
2206 int i;
2208 for (i = 0; i < view->lines; i++)
2209 free(view->line[i].data);
2210 free(view->line);
2212 view->line = NULL;
2213 view->offset = 0;
2214 view->lines = 0;
2215 view->lineno = 0;
2216 view->line_size = 0;
2217 view->line_alloc = 0;
2218 view->vid[0] = 0;
2219 }
2221 static void
2222 end_update(struct view *view, bool force)
2223 {
2224 if (!view->pipe)
2225 return;
2226 while (!view->ops->read(view, NULL))
2227 if (!force)
2228 return;
2229 set_nonblocking_input(FALSE);
2230 if (view->pipe == stdin)
2231 fclose(view->pipe);
2232 else
2233 pclose(view->pipe);
2234 view->pipe = NULL;
2235 }
2237 static void
2238 setup_update(struct view *view, const char *vid)
2239 {
2240 set_nonblocking_input(TRUE);
2241 reset_view(view);
2242 string_copy_rev(view->vid, vid);
2243 view->start_time = time(NULL);
2244 }
2246 static bool
2247 begin_update(struct view *view, bool refresh)
2248 {
2249 if (opt_cmd[0]) {
2250 string_copy(view->cmd, opt_cmd);
2251 opt_cmd[0] = 0;
2252 /* When running random commands, initially show the
2253 * command in the title. However, it maybe later be
2254 * overwritten if a commit line is selected. */
2255 if (view == VIEW(REQ_VIEW_PAGER))
2256 string_copy(view->ref, view->cmd);
2257 else
2258 view->ref[0] = 0;
2260 } else if (view == VIEW(REQ_VIEW_TREE)) {
2261 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2262 char path[SIZEOF_STR];
2264 if (strcmp(view->vid, view->id))
2265 opt_path[0] = path[0] = 0;
2266 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
2267 return FALSE;
2269 if (!string_format(view->cmd, format, view->id, path))
2270 return FALSE;
2272 } else if (!refresh) {
2273 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2274 const char *id = view->id;
2276 if (!string_format(view->cmd, format, id, id, id, id, id))
2277 return FALSE;
2279 /* Put the current ref_* value to the view title ref
2280 * member. This is needed by the blob view. Most other
2281 * views sets it automatically after loading because the
2282 * first line is a commit line. */
2283 string_copy_rev(view->ref, view->id);
2284 }
2286 /* Special case for the pager view. */
2287 if (opt_pipe) {
2288 view->pipe = opt_pipe;
2289 opt_pipe = NULL;
2290 } else {
2291 view->pipe = popen(view->cmd, "r");
2292 }
2294 if (!view->pipe)
2295 return FALSE;
2297 setup_update(view, view->id);
2299 return TRUE;
2300 }
2302 #define ITEM_CHUNK_SIZE 256
2303 static void *
2304 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2305 {
2306 size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2307 size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2309 if (mem == NULL || num_chunks != num_chunks_new) {
2310 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2311 mem = realloc(mem, *size * item_size);
2312 }
2314 return mem;
2315 }
2317 static struct line *
2318 realloc_lines(struct view *view, size_t line_size)
2319 {
2320 size_t alloc = view->line_alloc;
2321 struct line *tmp = realloc_items(view->line, &alloc, line_size,
2322 sizeof(*view->line));
2324 if (!tmp)
2325 return NULL;
2327 view->line = tmp;
2328 view->line_alloc = alloc;
2329 view->line_size = line_size;
2330 return view->line;
2331 }
2333 static bool
2334 update_view(struct view *view)
2335 {
2336 char in_buffer[BUFSIZ];
2337 char out_buffer[BUFSIZ * 2];
2338 char *line;
2339 /* The number of lines to read. If too low it will cause too much
2340 * redrawing (and possible flickering), if too high responsiveness
2341 * will suffer. */
2342 unsigned long lines = view->height;
2343 int redraw_from = -1;
2345 if (!view->pipe)
2346 return TRUE;
2348 /* Only redraw if lines are visible. */
2349 if (view->offset + view->height >= view->lines)
2350 redraw_from = view->lines - view->offset;
2352 /* FIXME: This is probably not perfect for backgrounded views. */
2353 if (!realloc_lines(view, view->lines + lines))
2354 goto alloc_error;
2356 while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
2357 size_t linelen = strlen(line);
2359 if (linelen)
2360 line[linelen - 1] = 0;
2362 if (opt_iconv != ICONV_NONE) {
2363 ICONV_CONST char *inbuf = line;
2364 size_t inlen = linelen;
2366 char *outbuf = out_buffer;
2367 size_t outlen = sizeof(out_buffer);
2369 size_t ret;
2371 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2372 if (ret != (size_t) -1) {
2373 line = out_buffer;
2374 linelen = strlen(out_buffer);
2375 }
2376 }
2378 if (!view->ops->read(view, line))
2379 goto alloc_error;
2381 if (lines-- == 1)
2382 break;
2383 }
2385 {
2386 int digits;
2388 lines = view->lines;
2389 for (digits = 0; lines; digits++)
2390 lines /= 10;
2392 /* Keep the displayed view in sync with line number scaling. */
2393 if (digits != view->digits) {
2394 view->digits = digits;
2395 redraw_from = 0;
2396 }
2397 }
2399 if (ferror(view->pipe) && errno != 0) {
2400 report("Failed to read: %s", strerror(errno));
2401 end_update(view, TRUE);
2403 } else if (feof(view->pipe)) {
2404 report("");
2405 end_update(view, FALSE);
2406 }
2408 if (!view_is_displayed(view))
2409 return TRUE;
2411 if (view == VIEW(REQ_VIEW_TREE)) {
2412 /* Clear the view and redraw everything since the tree sorting
2413 * might have rearranged things. */
2414 redraw_view(view);
2416 } else if (redraw_from >= 0) {
2417 /* If this is an incremental update, redraw the previous line
2418 * since for commits some members could have changed when
2419 * loading the main view. */
2420 if (redraw_from > 0)
2421 redraw_from--;
2423 /* Since revision graph visualization requires knowledge
2424 * about the parent commit, it causes a further one-off
2425 * needed to be redrawn for incremental updates. */
2426 if (redraw_from > 0 && opt_rev_graph)
2427 redraw_from--;
2429 /* Incrementally draw avoids flickering. */
2430 redraw_view_from(view, redraw_from);
2431 }
2433 if (view == VIEW(REQ_VIEW_BLAME))
2434 redraw_view_dirty(view);
2436 /* Update the title _after_ the redraw so that if the redraw picks up a
2437 * commit reference in view->ref it'll be available here. */
2438 update_view_title(view);
2439 return TRUE;
2441 alloc_error:
2442 report("Allocation failure");
2443 end_update(view, TRUE);
2444 return FALSE;
2445 }
2447 static struct line *
2448 add_line_data(struct view *view, void *data, enum line_type type)
2449 {
2450 struct line *line = &view->line[view->lines++];
2452 memset(line, 0, sizeof(*line));
2453 line->type = type;
2454 line->data = data;
2456 return line;
2457 }
2459 static struct line *
2460 add_line_text(struct view *view, const char *text, enum line_type type)
2461 {
2462 char *data = text ? strdup(text) : NULL;
2464 return data ? add_line_data(view, data, type) : NULL;
2465 }
2468 /*
2469 * View opening
2470 */
2472 enum open_flags {
2473 OPEN_DEFAULT = 0, /* Use default view switching. */
2474 OPEN_SPLIT = 1, /* Split current view. */
2475 OPEN_BACKGROUNDED = 2, /* Backgrounded. */
2476 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
2477 OPEN_NOMAXIMIZE = 8, /* Do not maximize the current view. */
2478 OPEN_REFRESH = 16, /* Refresh view using previous command. */
2479 };
2481 static void
2482 open_view(struct view *prev, enum request request, enum open_flags flags)
2483 {
2484 bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2485 bool split = !!(flags & OPEN_SPLIT);
2486 bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH));
2487 bool nomaximize = !!(flags & (OPEN_NOMAXIMIZE | OPEN_REFRESH));
2488 struct view *view = VIEW(request);
2489 int nviews = displayed_views();
2490 struct view *base_view = display[0];
2492 if (view == prev && nviews == 1 && !reload) {
2493 report("Already in %s view", view->name);
2494 return;
2495 }
2497 if (view->git_dir && !opt_git_dir[0]) {
2498 report("The %s view is disabled in pager view", view->name);
2499 return;
2500 }
2502 if (split) {
2503 display[1] = view;
2504 if (!backgrounded)
2505 current_view = 1;
2506 } else if (!nomaximize) {
2507 /* Maximize the current view. */
2508 memset(display, 0, sizeof(display));
2509 current_view = 0;
2510 display[current_view] = view;
2511 }
2513 /* Resize the view when switching between split- and full-screen,
2514 * or when switching between two different full-screen views. */
2515 if (nviews != displayed_views() ||
2516 (nviews == 1 && base_view != display[0]))
2517 resize_display();
2519 if (view->pipe)
2520 end_update(view, TRUE);
2522 if (view->ops->open) {
2523 if (!view->ops->open(view)) {
2524 report("Failed to load %s view", view->name);
2525 return;
2526 }
2528 } else if ((reload || strcmp(view->vid, view->id)) &&
2529 !begin_update(view, flags & OPEN_REFRESH)) {
2530 report("Failed to load %s view", view->name);
2531 return;
2532 }
2534 if (split && prev->lineno - prev->offset >= prev->height) {
2535 /* Take the title line into account. */
2536 int lines = prev->lineno - prev->offset - prev->height + 1;
2538 /* Scroll the view that was split if the current line is
2539 * outside the new limited view. */
2540 do_scroll_view(prev, lines);
2541 }
2543 if (prev && view != prev) {
2544 if (split && !backgrounded) {
2545 /* "Blur" the previous view. */
2546 update_view_title(prev);
2547 }
2549 view->parent = prev;
2550 }
2552 if (view->pipe && view->lines == 0) {
2553 /* Clear the old view and let the incremental updating refill
2554 * the screen. */
2555 werase(view->win);
2556 report("");
2557 } else if (view_is_displayed(view)) {
2558 redraw_view(view);
2559 report("");
2560 }
2562 /* If the view is backgrounded the above calls to report()
2563 * won't redraw the view title. */
2564 if (backgrounded)
2565 update_view_title(view);
2566 }
2568 static bool
2569 run_confirm(const char *cmd, const char *prompt)
2570 {
2571 bool confirmation = prompt_yesno(prompt);
2573 if (confirmation)
2574 system(cmd);
2576 return confirmation;
2577 }
2579 static void
2580 open_external_viewer(const char *cmd)
2581 {
2582 def_prog_mode(); /* save current tty modes */
2583 endwin(); /* restore original tty modes */
2584 system(cmd);
2585 fprintf(stderr, "Press Enter to continue");
2586 getc(opt_tty);
2587 reset_prog_mode();
2588 redraw_display();
2589 }
2591 static void
2592 open_mergetool(const char *file)
2593 {
2594 char cmd[SIZEOF_STR];
2595 char file_sq[SIZEOF_STR];
2597 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2598 string_format(cmd, "git mergetool %s", file_sq)) {
2599 open_external_viewer(cmd);
2600 }
2601 }
2603 static void
2604 open_editor(bool from_root, const char *file)
2605 {
2606 char cmd[SIZEOF_STR];
2607 char file_sq[SIZEOF_STR];
2608 const char *editor;
2609 char *prefix = from_root ? opt_cdup : "";
2611 editor = getenv("GIT_EDITOR");
2612 if (!editor && *opt_editor)
2613 editor = opt_editor;
2614 if (!editor)
2615 editor = getenv("VISUAL");
2616 if (!editor)
2617 editor = getenv("EDITOR");
2618 if (!editor)
2619 editor = "vi";
2621 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2622 string_format(cmd, "%s %s%s", editor, prefix, file_sq)) {
2623 open_external_viewer(cmd);
2624 }
2625 }
2627 static void
2628 open_run_request(enum request request)
2629 {
2630 struct run_request *req = get_run_request(request);
2631 char buf[SIZEOF_STR * 2];
2632 size_t bufpos;
2633 char *cmd;
2635 if (!req) {
2636 report("Unknown run request");
2637 return;
2638 }
2640 bufpos = 0;
2641 cmd = req->cmd;
2643 while (cmd) {
2644 char *next = strstr(cmd, "%(");
2645 int len = next - cmd;
2646 char *value;
2648 if (!next) {
2649 len = strlen(cmd);
2650 value = "";
2652 } else if (!strncmp(next, "%(head)", 7)) {
2653 value = ref_head;
2655 } else if (!strncmp(next, "%(commit)", 9)) {
2656 value = ref_commit;
2658 } else if (!strncmp(next, "%(blob)", 7)) {
2659 value = ref_blob;
2661 } else {
2662 report("Unknown replacement in run request: `%s`", req->cmd);
2663 return;
2664 }
2666 if (!string_format_from(buf, &bufpos, "%.*s%s", len, cmd, value))
2667 return;
2669 if (next)
2670 next = strchr(next, ')') + 1;
2671 cmd = next;
2672 }
2674 open_external_viewer(buf);
2675 }
2677 /*
2678 * User request switch noodle
2679 */
2681 static int
2682 view_driver(struct view *view, enum request request)
2683 {
2684 int i;
2686 if (request == REQ_NONE) {
2687 doupdate();
2688 return TRUE;
2689 }
2691 if (request > REQ_NONE) {
2692 open_run_request(request);
2693 /* FIXME: When all views can refresh always do this. */
2694 if (view == VIEW(REQ_VIEW_STATUS) ||
2695 view == VIEW(REQ_VIEW_MAIN) ||
2696 view == VIEW(REQ_VIEW_LOG) ||
2697 view == VIEW(REQ_VIEW_STAGE))
2698 request = REQ_REFRESH;
2699 else
2700 return TRUE;
2701 }
2703 if (view && view->lines) {
2704 request = view->ops->request(view, request, &view->line[view->lineno]);
2705 if (request == REQ_NONE)
2706 return TRUE;
2707 }
2709 switch (request) {
2710 case REQ_MOVE_UP:
2711 case REQ_MOVE_DOWN:
2712 case REQ_MOVE_PAGE_UP:
2713 case REQ_MOVE_PAGE_DOWN:
2714 case REQ_MOVE_FIRST_LINE:
2715 case REQ_MOVE_LAST_LINE:
2716 move_view(view, request);
2717 break;
2719 case REQ_SCROLL_LINE_DOWN:
2720 case REQ_SCROLL_LINE_UP:
2721 case REQ_SCROLL_PAGE_DOWN:
2722 case REQ_SCROLL_PAGE_UP:
2723 scroll_view(view, request);
2724 break;
2726 case REQ_VIEW_BLAME:
2727 if (!opt_file[0]) {
2728 report("No file chosen, press %s to open tree view",
2729 get_key(REQ_VIEW_TREE));
2730 break;
2731 }
2732 open_view(view, request, OPEN_DEFAULT);
2733 break;
2735 case REQ_VIEW_BLOB:
2736 if (!ref_blob[0]) {
2737 report("No file chosen, press %s to open tree view",
2738 get_key(REQ_VIEW_TREE));
2739 break;
2740 }
2741 open_view(view, request, OPEN_DEFAULT);
2742 break;
2744 case REQ_VIEW_PAGER:
2745 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2746 report("No pager content, press %s to run command from prompt",
2747 get_key(REQ_PROMPT));
2748 break;
2749 }
2750 open_view(view, request, OPEN_DEFAULT);
2751 break;
2753 case REQ_VIEW_STAGE:
2754 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2755 report("No stage content, press %s to open the status view and choose file",
2756 get_key(REQ_VIEW_STATUS));
2757 break;
2758 }
2759 open_view(view, request, OPEN_DEFAULT);
2760 break;
2762 case REQ_VIEW_STATUS:
2763 if (opt_is_inside_work_tree == FALSE) {
2764 report("The status view requires a working tree");
2765 break;
2766 }
2767 open_view(view, request, OPEN_DEFAULT);
2768 break;
2770 case REQ_VIEW_MAIN:
2771 case REQ_VIEW_DIFF:
2772 case REQ_VIEW_LOG:
2773 case REQ_VIEW_TREE:
2774 case REQ_VIEW_HELP:
2775 open_view(view, request, OPEN_DEFAULT);
2776 break;
2778 case REQ_NEXT:
2779 case REQ_PREVIOUS:
2780 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2782 if ((view == VIEW(REQ_VIEW_DIFF) &&
2783 view->parent == VIEW(REQ_VIEW_MAIN)) ||
2784 (view == VIEW(REQ_VIEW_DIFF) &&
2785 view->parent == VIEW(REQ_VIEW_BLAME)) ||
2786 (view == VIEW(REQ_VIEW_STAGE) &&
2787 view->parent == VIEW(REQ_VIEW_STATUS)) ||
2788 (view == VIEW(REQ_VIEW_BLOB) &&
2789 view->parent == VIEW(REQ_VIEW_TREE))) {
2790 int line;
2792 view = view->parent;
2793 line = view->lineno;
2794 move_view(view, request);
2795 if (view_is_displayed(view))
2796 update_view_title(view);
2797 if (line != view->lineno)
2798 view->ops->request(view, REQ_ENTER,
2799 &view->line[view->lineno]);
2801 } else {
2802 move_view(view, request);
2803 }
2804 break;
2806 case REQ_VIEW_NEXT:
2807 {
2808 int nviews = displayed_views();
2809 int next_view = (current_view + 1) % nviews;
2811 if (next_view == current_view) {
2812 report("Only one view is displayed");
2813 break;
2814 }
2816 current_view = next_view;
2817 /* Blur out the title of the previous view. */
2818 update_view_title(view);
2819 report("");
2820 break;
2821 }
2822 case REQ_REFRESH:
2823 report("Refreshing is not yet supported for the %s view", view->name);
2824 break;
2826 case REQ_MAXIMIZE:
2827 if (displayed_views() == 2)
2828 open_view(view, VIEW_REQ(view), OPEN_DEFAULT);
2829 break;
2831 case REQ_TOGGLE_LINENO:
2832 opt_line_number = !opt_line_number;
2833 redraw_display();
2834 break;
2836 case REQ_TOGGLE_DATE:
2837 opt_date = !opt_date;
2838 redraw_display();
2839 break;
2841 case REQ_TOGGLE_AUTHOR:
2842 opt_author = !opt_author;
2843 redraw_display();
2844 break;
2846 case REQ_TOGGLE_REV_GRAPH:
2847 opt_rev_graph = !opt_rev_graph;
2848 redraw_display();
2849 break;
2851 case REQ_TOGGLE_REFS:
2852 opt_show_refs = !opt_show_refs;
2853 redraw_display();
2854 break;
2856 case REQ_SEARCH:
2857 case REQ_SEARCH_BACK:
2858 search_view(view, request);
2859 break;
2861 case REQ_FIND_NEXT:
2862 case REQ_FIND_PREV:
2863 find_next(view, request);
2864 break;
2866 case REQ_STOP_LOADING:
2867 for (i = 0; i < ARRAY_SIZE(views); i++) {
2868 view = &views[i];
2869 if (view->pipe)
2870 report("Stopped loading the %s view", view->name),
2871 end_update(view, TRUE);
2872 }
2873 break;
2875 case REQ_SHOW_VERSION:
2876 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2877 return TRUE;
2879 case REQ_SCREEN_RESIZE:
2880 resize_display();
2881 /* Fall-through */
2882 case REQ_SCREEN_REDRAW:
2883 redraw_display();
2884 break;
2886 case REQ_EDIT:
2887 report("Nothing to edit");
2888 break;
2890 case REQ_ENTER:
2891 report("Nothing to enter");
2892 break;
2894 case REQ_VIEW_CLOSE:
2895 /* XXX: Mark closed views by letting view->parent point to the
2896 * view itself. Parents to closed view should never be
2897 * followed. */
2898 if (view->parent &&
2899 view->parent->parent != view->parent) {
2900 memset(display, 0, sizeof(display));
2901 current_view = 0;
2902 display[current_view] = view->parent;
2903 view->parent = view;
2904 resize_display();
2905 redraw_display();
2906 report("");
2907 break;
2908 }
2909 /* Fall-through */
2910 case REQ_QUIT:
2911 return FALSE;
2913 default:
2914 report("Unknown key, press 'h' for help");
2915 return TRUE;
2916 }
2918 return TRUE;
2919 }
2922 /*
2923 * Pager backend
2924 */
2926 static bool
2927 pager_draw(struct view *view, struct line *line, unsigned int lineno)
2928 {
2929 char *text = line->data;
2931 if (opt_line_number && draw_lineno(view, lineno))
2932 return TRUE;
2934 draw_text(view, line->type, text, TRUE);
2935 return TRUE;
2936 }
2938 static bool
2939 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
2940 {
2941 char refbuf[SIZEOF_STR];
2942 char *ref = NULL;
2943 FILE *pipe;
2945 if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2946 return TRUE;
2948 pipe = popen(refbuf, "r");
2949 if (!pipe)
2950 return TRUE;
2952 if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2953 ref = chomp_string(ref);
2954 pclose(pipe);
2956 if (!ref || !*ref)
2957 return TRUE;
2959 /* This is the only fatal call, since it can "corrupt" the buffer. */
2960 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2961 return FALSE;
2963 return TRUE;
2964 }
2966 static void
2967 add_pager_refs(struct view *view, struct line *line)
2968 {
2969 char buf[SIZEOF_STR];
2970 char *commit_id = (char *)line->data + STRING_SIZE("commit ");
2971 struct ref **refs;
2972 size_t bufpos = 0, refpos = 0;
2973 const char *sep = "Refs: ";
2974 bool is_tag = FALSE;
2976 assert(line->type == LINE_COMMIT);
2978 refs = get_refs(commit_id);
2979 if (!refs) {
2980 if (view == VIEW(REQ_VIEW_DIFF))
2981 goto try_add_describe_ref;
2982 return;
2983 }
2985 do {
2986 struct ref *ref = refs[refpos];
2987 const char *fmt = ref->tag ? "%s[%s]" :
2988 ref->remote ? "%s<%s>" : "%s%s";
2990 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2991 return;
2992 sep = ", ";
2993 if (ref->tag)
2994 is_tag = TRUE;
2995 } while (refs[refpos++]->next);
2997 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2998 try_add_describe_ref:
2999 /* Add <tag>-g<commit_id> "fake" reference. */
3000 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
3001 return;
3002 }
3004 if (bufpos == 0)
3005 return;
3007 if (!realloc_lines(view, view->line_size + 1))
3008 return;
3010 add_line_text(view, buf, LINE_PP_REFS);
3011 }
3013 static bool
3014 pager_read(struct view *view, char *data)
3015 {
3016 struct line *line;
3018 if (!data)
3019 return TRUE;
3021 line = add_line_text(view, data, get_line_type(data));
3022 if (!line)
3023 return FALSE;
3025 if (line->type == LINE_COMMIT &&
3026 (view == VIEW(REQ_VIEW_DIFF) ||
3027 view == VIEW(REQ_VIEW_LOG)))
3028 add_pager_refs(view, line);
3030 return TRUE;
3031 }
3033 static enum request
3034 pager_request(struct view *view, enum request request, struct line *line)
3035 {
3036 int split = 0;
3038 if (request != REQ_ENTER)
3039 return request;
3041 if (line->type == LINE_COMMIT &&
3042 (view == VIEW(REQ_VIEW_LOG) ||
3043 view == VIEW(REQ_VIEW_PAGER))) {
3044 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
3045 split = 1;
3046 }
3048 /* Always scroll the view even if it was split. That way
3049 * you can use Enter to scroll through the log view and
3050 * split open each commit diff. */
3051 scroll_view(view, REQ_SCROLL_LINE_DOWN);
3053 /* FIXME: A minor workaround. Scrolling the view will call report("")
3054 * but if we are scrolling a non-current view this won't properly
3055 * update the view title. */
3056 if (split)
3057 update_view_title(view);
3059 return REQ_NONE;
3060 }
3062 static bool
3063 pager_grep(struct view *view, struct line *line)
3064 {
3065 regmatch_t pmatch;
3066 char *text = line->data;
3068 if (!*text)
3069 return FALSE;
3071 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
3072 return FALSE;
3074 return TRUE;
3075 }
3077 static void
3078 pager_select(struct view *view, struct line *line)
3079 {
3080 if (line->type == LINE_COMMIT) {
3081 char *text = (char *)line->data + STRING_SIZE("commit ");
3083 if (view != VIEW(REQ_VIEW_PAGER))
3084 string_copy_rev(view->ref, text);
3085 string_copy_rev(ref_commit, text);
3086 }
3087 }
3089 static struct view_ops pager_ops = {
3090 "line",
3091 NULL,
3092 pager_read,
3093 pager_draw,
3094 pager_request,
3095 pager_grep,
3096 pager_select,
3097 };
3099 static enum request
3100 log_request(struct view *view, enum request request, struct line *line)
3101 {
3102 switch (request) {
3103 case REQ_REFRESH:
3104 load_refs();
3105 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
3106 return REQ_NONE;
3107 default:
3108 return pager_request(view, request, line);
3109 }
3110 }
3112 static struct view_ops log_ops = {
3113 "line",
3114 NULL,
3115 pager_read,
3116 pager_draw,
3117 log_request,
3118 pager_grep,
3119 pager_select,
3120 };
3123 /*
3124 * Help backend
3125 */
3127 static bool
3128 help_open(struct view *view)
3129 {
3130 char buf[BUFSIZ];
3131 int lines = ARRAY_SIZE(req_info) + 2;
3132 int i;
3134 if (view->lines > 0)
3135 return TRUE;
3137 for (i = 0; i < ARRAY_SIZE(req_info); i++)
3138 if (!req_info[i].request)
3139 lines++;
3141 lines += run_requests + 1;
3143 view->line = calloc(lines, sizeof(*view->line));
3144 if (!view->line)
3145 return FALSE;
3147 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3149 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3150 const char *key;
3152 if (req_info[i].request == REQ_NONE)
3153 continue;
3155 if (!req_info[i].request) {
3156 add_line_text(view, "", LINE_DEFAULT);
3157 add_line_text(view, req_info[i].help, LINE_DEFAULT);
3158 continue;
3159 }
3161 key = get_key(req_info[i].request);
3162 if (!*key)
3163 key = "(no key defined)";
3165 if (!string_format(buf, " %-25s %s", key, req_info[i].help))
3166 continue;
3168 add_line_text(view, buf, LINE_DEFAULT);
3169 }
3171 if (run_requests) {
3172 add_line_text(view, "", LINE_DEFAULT);
3173 add_line_text(view, "External commands:", LINE_DEFAULT);
3174 }
3176 for (i = 0; i < run_requests; i++) {
3177 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3178 const char *key;
3180 if (!req)
3181 continue;
3183 key = get_key_name(req->key);
3184 if (!*key)
3185 key = "(no key defined)";
3187 if (!string_format(buf, " %-10s %-14s `%s`",
3188 keymap_table[req->keymap].name,
3189 key, req->cmd))
3190 continue;
3192 add_line_text(view, buf, LINE_DEFAULT);
3193 }
3195 return TRUE;
3196 }
3198 static struct view_ops help_ops = {
3199 "line",
3200 help_open,
3201 NULL,
3202 pager_draw,
3203 pager_request,
3204 pager_grep,
3205 pager_select,
3206 };
3209 /*
3210 * Tree backend
3211 */
3213 struct tree_stack_entry {
3214 struct tree_stack_entry *prev; /* Entry below this in the stack */
3215 unsigned long lineno; /* Line number to restore */
3216 char *name; /* Position of name in opt_path */
3217 };
3219 /* The top of the path stack. */
3220 static struct tree_stack_entry *tree_stack = NULL;
3221 unsigned long tree_lineno = 0;
3223 static void
3224 pop_tree_stack_entry(void)
3225 {
3226 struct tree_stack_entry *entry = tree_stack;
3228 tree_lineno = entry->lineno;
3229 entry->name[0] = 0;
3230 tree_stack = entry->prev;
3231 free(entry);
3232 }
3234 static void
3235 push_tree_stack_entry(const char *name, unsigned long lineno)
3236 {
3237 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3238 size_t pathlen = strlen(opt_path);
3240 if (!entry)
3241 return;
3243 entry->prev = tree_stack;
3244 entry->name = opt_path + pathlen;
3245 tree_stack = entry;
3247 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3248 pop_tree_stack_entry();
3249 return;
3250 }
3252 /* Move the current line to the first tree entry. */
3253 tree_lineno = 1;
3254 entry->lineno = lineno;
3255 }
3257 /* Parse output from git-ls-tree(1):
3258 *
3259 * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
3260 * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
3261 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3262 * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
3263 */
3265 #define SIZEOF_TREE_ATTR \
3266 STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
3268 #define TREE_UP_FORMAT "040000 tree %s\t.."
3270 static int
3271 tree_compare_entry(enum line_type type1, const char *name1,
3272 enum line_type type2, const char *name2)
3273 {
3274 if (type1 != type2) {
3275 if (type1 == LINE_TREE_DIR)
3276 return -1;
3277 return 1;
3278 }
3280 return strcmp(name1, name2);
3281 }
3283 static const char *
3284 tree_path(struct line *line)
3285 {
3286 const char *path = line->data;
3288 return path + SIZEOF_TREE_ATTR;
3289 }
3291 static bool
3292 tree_read(struct view *view, char *text)
3293 {
3294 size_t textlen = text ? strlen(text) : 0;
3295 char buf[SIZEOF_STR];
3296 unsigned long pos;
3297 enum line_type type;
3298 bool first_read = view->lines == 0;
3300 if (!text)
3301 return TRUE;
3302 if (textlen <= SIZEOF_TREE_ATTR)
3303 return FALSE;
3305 type = text[STRING_SIZE("100644 ")] == 't'
3306 ? LINE_TREE_DIR : LINE_TREE_FILE;
3308 if (first_read) {
3309 /* Add path info line */
3310 if (!string_format(buf, "Directory path /%s", opt_path) ||
3311 !realloc_lines(view, view->line_size + 1) ||
3312 !add_line_text(view, buf, LINE_DEFAULT))
3313 return FALSE;
3315 /* Insert "link" to parent directory. */
3316 if (*opt_path) {
3317 if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
3318 !realloc_lines(view, view->line_size + 1) ||
3319 !add_line_text(view, buf, LINE_TREE_DIR))
3320 return FALSE;
3321 }
3322 }
3324 /* Strip the path part ... */
3325 if (*opt_path) {
3326 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3327 size_t striplen = strlen(opt_path);
3328 char *path = text + SIZEOF_TREE_ATTR;
3330 if (pathlen > striplen)
3331 memmove(path, path + striplen,
3332 pathlen - striplen + 1);
3333 }
3335 /* Skip "Directory ..." and ".." line. */
3336 for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
3337 struct line *line = &view->line[pos];
3338 const char *path1 = tree_path(line);
3339 char *path2 = text + SIZEOF_TREE_ATTR;
3340 int cmp = tree_compare_entry(line->type, path1, type, path2);
3342 if (cmp <= 0)
3343 continue;
3345 text = strdup(text);
3346 if (!text)
3347 return FALSE;
3349 if (view->lines > pos)
3350 memmove(&view->line[pos + 1], &view->line[pos],
3351 (view->lines - pos) * sizeof(*line));
3353 line = &view->line[pos];
3354 line->data = text;
3355 line->type = type;
3356 view->lines++;
3357 return TRUE;
3358 }
3360 if (!add_line_text(view, text, type))
3361 return FALSE;
3363 if (tree_lineno > view->lineno) {
3364 view->lineno = tree_lineno;
3365 tree_lineno = 0;
3366 }
3368 return TRUE;
3369 }
3371 static enum request
3372 tree_request(struct view *view, enum request request, struct line *line)
3373 {
3374 enum open_flags flags;
3376 switch (request) {
3377 case REQ_VIEW_BLAME:
3378 if (line->type != LINE_TREE_FILE) {
3379 report("Blame only supported for files");
3380 return REQ_NONE;
3381 }
3383 string_copy(opt_ref, view->vid);
3384 return request;
3386 case REQ_EDIT:
3387 if (line->type != LINE_TREE_FILE) {
3388 report("Edit only supported for files");
3389 } else if (!is_head_commit(view->vid)) {
3390 report("Edit only supported for files in the current work tree");
3391 } else {
3392 open_editor(TRUE, opt_file);
3393 }
3394 return REQ_NONE;
3396 case REQ_TREE_PARENT:
3397 if (!*opt_path) {
3398 /* quit view if at top of tree */
3399 return REQ_VIEW_CLOSE;
3400 }
3401 /* fake 'cd ..' */
3402 line = &view->line[1];
3403 break;
3405 case REQ_ENTER:
3406 break;
3408 default:
3409 return request;
3410 }
3412 /* Cleanup the stack if the tree view is at a different tree. */
3413 while (!*opt_path && tree_stack)
3414 pop_tree_stack_entry();
3416 switch (line->type) {
3417 case LINE_TREE_DIR:
3418 /* Depending on whether it is a subdir or parent (updir?) link
3419 * mangle the path buffer. */
3420 if (line == &view->line[1] && *opt_path) {
3421 pop_tree_stack_entry();
3423 } else {
3424 const char *basename = tree_path(line);
3426 push_tree_stack_entry(basename, view->lineno);
3427 }
3429 /* Trees and subtrees share the same ID, so they are not not
3430 * unique like blobs. */
3431 flags = OPEN_RELOAD;
3432 request = REQ_VIEW_TREE;
3433 break;
3435 case LINE_TREE_FILE:
3436 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3437 request = REQ_VIEW_BLOB;
3438 break;
3440 default:
3441 return TRUE;
3442 }
3444 open_view(view, request, flags);
3445 if (request == REQ_VIEW_TREE) {
3446 view->lineno = tree_lineno;
3447 }
3449 return REQ_NONE;
3450 }
3452 static void
3453 tree_select(struct view *view, struct line *line)
3454 {
3455 char *text = (char *)line->data + STRING_SIZE("100644 blob ");
3457 if (line->type == LINE_TREE_FILE) {
3458 string_copy_rev(ref_blob, text);
3459 string_format(opt_file, "%s%s", opt_path, tree_path(line));
3461 } else if (line->type != LINE_TREE_DIR) {
3462 return;
3463 }
3465 string_copy_rev(view->ref, text);
3466 }
3468 static struct view_ops tree_ops = {
3469 "file",
3470 NULL,
3471 tree_read,
3472 pager_draw,
3473 tree_request,
3474 pager_grep,
3475 tree_select,
3476 };
3478 static bool
3479 blob_read(struct view *view, char *line)
3480 {
3481 if (!line)
3482 return TRUE;
3483 return add_line_text(view, line, LINE_DEFAULT) != NULL;
3484 }
3486 static struct view_ops blob_ops = {
3487 "line",
3488 NULL,
3489 blob_read,
3490 pager_draw,
3491 pager_request,
3492 pager_grep,
3493 pager_select,
3494 };
3496 /*
3497 * Blame backend
3498 *
3499 * Loading the blame view is a two phase job:
3500 *
3501 * 1. File content is read either using opt_file from the
3502 * filesystem or using git-cat-file.
3503 * 2. Then blame information is incrementally added by
3504 * reading output from git-blame.
3505 */
3507 struct blame_commit {
3508 char id[SIZEOF_REV]; /* SHA1 ID. */
3509 char title[128]; /* First line of the commit message. */
3510 char author[75]; /* Author of the commit. */
3511 struct tm time; /* Date from the author ident. */
3512 char filename[128]; /* Name of file. */
3513 };
3515 struct blame {
3516 struct blame_commit *commit;
3517 char text[1];
3518 };
3520 #define BLAME_CAT_FILE_CMD "git cat-file blob %s:%s"
3521 #define BLAME_INCREMENTAL_CMD "git blame --incremental %s -- %s"
3523 static bool
3524 blame_open(struct view *view)
3525 {
3526 char path[SIZEOF_STR];
3527 char ref[SIZEOF_STR] = "";
3529 if (sq_quote(path, 0, opt_file) >= sizeof(path))
3530 return FALSE;
3532 if (*opt_ref && sq_quote(ref, 0, opt_ref) >= sizeof(ref))
3533 return FALSE;
3535 if (*opt_ref || !(view->pipe = fopen(opt_file, "r"))) {
3536 const char *id = *opt_ref ? ref : "HEAD";
3538 if (!string_format(view->cmd, BLAME_CAT_FILE_CMD, id, path) ||
3539 !(view->pipe = popen(view->cmd, "r")))
3540 return FALSE;
3541 }
3543 setup_update(view, opt_file);
3544 string_format(view->ref, "%s ...", opt_file);
3546 return TRUE;
3547 }
3549 static struct blame_commit *
3550 get_blame_commit(struct view *view, const char *id)
3551 {
3552 size_t i;
3554 for (i = 0; i < view->lines; i++) {
3555 struct blame *blame = view->line[i].data;
3557 if (!blame->commit)
3558 continue;
3560 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
3561 return blame->commit;
3562 }
3564 {
3565 struct blame_commit *commit = calloc(1, sizeof(*commit));
3567 if (commit)
3568 string_ncopy(commit->id, id, SIZEOF_REV);
3569 return commit;
3570 }
3571 }
3573 static bool
3574 parse_number(const char **posref, size_t *number, size_t min, size_t max)
3575 {
3576 const char *pos = *posref;
3578 *posref = NULL;
3579 pos = strchr(pos + 1, ' ');
3580 if (!pos || !isdigit(pos[1]))
3581 return FALSE;
3582 *number = atoi(pos + 1);
3583 if (*number < min || *number > max)
3584 return FALSE;
3586 *posref = pos;
3587 return TRUE;
3588 }
3590 static struct blame_commit *
3591 parse_blame_commit(struct view *view, const char *text, int *blamed)
3592 {
3593 struct blame_commit *commit;
3594 struct blame *blame;
3595 const char *pos = text + SIZEOF_REV - 1;
3596 size_t lineno;
3597 size_t group;
3599 if (strlen(text) <= SIZEOF_REV || *pos != ' ')
3600 return NULL;
3602 if (!parse_number(&pos, &lineno, 1, view->lines) ||
3603 !parse_number(&pos, &group, 1, view->lines - lineno + 1))
3604 return NULL;
3606 commit = get_blame_commit(view, text);
3607 if (!commit)
3608 return NULL;
3610 *blamed += group;
3611 while (group--) {
3612 struct line *line = &view->line[lineno + group - 1];
3614 blame = line->data;
3615 blame->commit = commit;
3616 line->dirty = 1;
3617 }
3619 return commit;
3620 }
3622 static bool
3623 blame_read_file(struct view *view, const char *line, bool *read_file)
3624 {
3625 if (!line) {
3626 char ref[SIZEOF_STR] = "";
3627 char path[SIZEOF_STR];
3628 FILE *pipe = NULL;
3630 if (view->lines == 0 && !view->parent)
3631 die("No blame exist for %s", view->vid);
3633 if (view->lines == 0 ||
3634 sq_quote(path, 0, opt_file) >= sizeof(path) ||
3635 (*opt_ref && sq_quote(ref, 0, opt_ref) >= sizeof(ref)) ||
3636 !string_format(view->cmd, BLAME_INCREMENTAL_CMD, ref, path) ||
3637 !(pipe = popen(view->cmd, "r"))) {
3638 report("Failed to load blame data");
3639 return TRUE;
3640 }
3642 fclose(view->pipe);
3643 view->pipe = pipe;
3644 *read_file = FALSE;
3645 return FALSE;
3647 } else {
3648 size_t linelen = strlen(line);
3649 struct blame *blame = malloc(sizeof(*blame) + linelen);
3651 blame->commit = NULL;
3652 strncpy(blame->text, line, linelen);
3653 blame->text[linelen] = 0;
3654 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
3655 }
3656 }
3658 static bool
3659 match_blame_header(const char *name, char **line)
3660 {
3661 size_t namelen = strlen(name);
3662 bool matched = !strncmp(name, *line, namelen);
3664 if (matched)
3665 *line += namelen;
3667 return matched;
3668 }
3670 static bool
3671 blame_read(struct view *view, char *line)
3672 {
3673 static struct blame_commit *commit = NULL;
3674 static int blamed = 0;
3675 static time_t author_time;
3676 static bool read_file = TRUE;
3678 if (read_file)
3679 return blame_read_file(view, line, &read_file);
3681 if (!line) {
3682 /* Reset all! */
3683 commit = NULL;
3684 blamed = 0;
3685 read_file = TRUE;
3686 string_format(view->ref, "%s", view->vid);
3687 if (view_is_displayed(view)) {
3688 update_view_title(view);
3689 redraw_view_from(view, 0);
3690 }
3691 return TRUE;
3692 }
3694 if (!commit) {
3695 commit = parse_blame_commit(view, line, &blamed);
3696 string_format(view->ref, "%s %2d%%", view->vid,
3697 blamed * 100 / view->lines);
3699 } else if (match_blame_header("author ", &line)) {
3700 string_ncopy(commit->author, line, strlen(line));
3702 } else if (match_blame_header("author-time ", &line)) {
3703 author_time = (time_t) atol(line);
3705 } else if (match_blame_header("author-tz ", &line)) {
3706 long tz;
3708 tz = ('0' - line[1]) * 60 * 60 * 10;
3709 tz += ('0' - line[2]) * 60 * 60;
3710 tz += ('0' - line[3]) * 60;
3711 tz += ('0' - line[4]) * 60;
3713 if (line[0] == '-')
3714 tz = -tz;
3716 author_time -= tz;
3717 gmtime_r(&author_time, &commit->time);
3719 } else if (match_blame_header("summary ", &line)) {
3720 string_ncopy(commit->title, line, strlen(line));
3722 } else if (match_blame_header("filename ", &line)) {
3723 string_ncopy(commit->filename, line, strlen(line));
3724 commit = NULL;
3725 }
3727 return TRUE;
3728 }
3730 static bool
3731 blame_draw(struct view *view, struct line *line, unsigned int lineno)
3732 {
3733 struct blame *blame = line->data;
3734 struct tm *time = NULL;
3735 const char *id = NULL, *author = NULL;
3737 if (blame->commit && *blame->commit->filename) {
3738 id = blame->commit->id;
3739 author = blame->commit->author;
3740 time = &blame->commit->time;
3741 }
3743 if (opt_date && draw_date(view, time))
3744 return TRUE;
3746 if (opt_author &&
3747 draw_field(view, LINE_MAIN_AUTHOR, author, opt_author_cols, TRUE))
3748 return TRUE;
3750 if (draw_field(view, LINE_BLAME_ID, id, ID_COLS, FALSE))
3751 return TRUE;
3753 if (draw_lineno(view, lineno))
3754 return TRUE;
3756 draw_text(view, LINE_DEFAULT, blame->text, TRUE);
3757 return TRUE;
3758 }
3760 static enum request
3761 blame_request(struct view *view, enum request request, struct line *line)
3762 {
3763 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3764 struct blame *blame = line->data;
3766 switch (request) {
3767 case REQ_VIEW_BLAME:
3768 if (!blame->commit || !strcmp(blame->commit->id, NULL_ID)) {
3769 report("Commit ID unknown");
3770 break;
3771 }
3772 string_copy(opt_ref, blame->commit->id);
3773 open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
3774 return request;
3776 case REQ_ENTER:
3777 if (!blame->commit) {
3778 report("No commit loaded yet");
3779 break;
3780 }
3782 if (view_is_displayed(VIEW(REQ_VIEW_DIFF)) &&
3783 !strcmp(blame->commit->id, VIEW(REQ_VIEW_DIFF)->ref))
3784 break;
3786 if (!strcmp(blame->commit->id, NULL_ID)) {
3787 char path[SIZEOF_STR];
3789 if (sq_quote(path, 0, view->vid) >= sizeof(path))
3790 break;
3791 string_format(opt_cmd, "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s 2>/dev/null", path);
3792 }
3794 open_view(view, REQ_VIEW_DIFF, flags);
3795 break;
3797 default:
3798 return request;
3799 }
3801 return REQ_NONE;
3802 }
3804 static bool
3805 blame_grep(struct view *view, struct line *line)
3806 {
3807 struct blame *blame = line->data;
3808 struct blame_commit *commit = blame->commit;
3809 regmatch_t pmatch;
3811 #define MATCH(text, on) \
3812 (on && *text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3814 if (commit) {
3815 char buf[DATE_COLS + 1];
3817 if (MATCH(commit->title, 1) ||
3818 MATCH(commit->author, opt_author) ||
3819 MATCH(commit->id, opt_date))
3820 return TRUE;
3822 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
3823 MATCH(buf, 1))
3824 return TRUE;
3825 }
3827 return MATCH(blame->text, 1);
3829 #undef MATCH
3830 }
3832 static void
3833 blame_select(struct view *view, struct line *line)
3834 {
3835 struct blame *blame = line->data;
3836 struct blame_commit *commit = blame->commit;
3838 if (!commit)
3839 return;
3841 if (!strcmp(commit->id, NULL_ID))
3842 string_ncopy(ref_commit, "HEAD", 4);
3843 else
3844 string_copy_rev(ref_commit, commit->id);
3845 }
3847 static struct view_ops blame_ops = {
3848 "line",
3849 blame_open,
3850 blame_read,
3851 blame_draw,
3852 blame_request,
3853 blame_grep,
3854 blame_select,
3855 };
3857 /*
3858 * Status backend
3859 */
3861 struct status {
3862 char status;
3863 struct {
3864 mode_t mode;
3865 char rev[SIZEOF_REV];
3866 char name[SIZEOF_STR];
3867 } old;
3868 struct {
3869 mode_t mode;
3870 char rev[SIZEOF_REV];
3871 char name[SIZEOF_STR];
3872 } new;
3873 };
3875 static char status_onbranch[SIZEOF_STR];
3876 static struct status stage_status;
3877 static enum line_type stage_line_type;
3878 static size_t stage_chunks;
3879 static int *stage_chunk;
3881 /* This should work even for the "On branch" line. */
3882 static inline bool
3883 status_has_none(struct view *view, struct line *line)
3884 {
3885 return line < view->line + view->lines && !line[1].data;
3886 }
3888 /* Get fields from the diff line:
3889 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
3890 */
3891 static inline bool
3892 status_get_diff(struct status *file, const char *buf, size_t bufsize)
3893 {
3894 const char *old_mode = buf + 1;
3895 const char *new_mode = buf + 8;
3896 const char *old_rev = buf + 15;
3897 const char *new_rev = buf + 56;
3898 const char *status = buf + 97;
3900 if (bufsize < 99 ||
3901 old_mode[-1] != ':' ||
3902 new_mode[-1] != ' ' ||
3903 old_rev[-1] != ' ' ||
3904 new_rev[-1] != ' ' ||
3905 status[-1] != ' ')
3906 return FALSE;
3908 file->status = *status;
3910 string_copy_rev(file->old.rev, old_rev);
3911 string_copy_rev(file->new.rev, new_rev);
3913 file->old.mode = strtoul(old_mode, NULL, 8);
3914 file->new.mode = strtoul(new_mode, NULL, 8);
3916 file->old.name[0] = file->new.name[0] = 0;
3918 return TRUE;
3919 }
3921 static bool
3922 status_run(struct view *view, const char cmd[], char status, enum line_type type)
3923 {
3924 struct status *file = NULL;
3925 struct status *unmerged = NULL;
3926 char buf[SIZEOF_STR * 4];
3927 size_t bufsize = 0;
3928 FILE *pipe;
3930 pipe = popen(cmd, "r");
3931 if (!pipe)
3932 return FALSE;
3934 add_line_data(view, NULL, type);
3936 while (!feof(pipe) && !ferror(pipe)) {
3937 char *sep;
3938 size_t readsize;
3940 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
3941 if (!readsize)
3942 break;
3943 bufsize += readsize;
3945 /* Process while we have NUL chars. */
3946 while ((sep = memchr(buf, 0, bufsize))) {
3947 size_t sepsize = sep - buf + 1;
3949 if (!file) {
3950 if (!realloc_lines(view, view->line_size + 1))
3951 goto error_out;
3953 file = calloc(1, sizeof(*file));
3954 if (!file)
3955 goto error_out;
3957 add_line_data(view, file, type);
3958 }
3960 /* Parse diff info part. */
3961 if (status) {
3962 file->status = status;
3963 if (status == 'A')
3964 string_copy(file->old.rev, NULL_ID);
3966 } else if (!file->status) {
3967 if (!status_get_diff(file, buf, sepsize))
3968 goto error_out;
3970 bufsize -= sepsize;
3971 memmove(buf, sep + 1, bufsize);
3973 sep = memchr(buf, 0, bufsize);
3974 if (!sep)
3975 break;
3976 sepsize = sep - buf + 1;
3978 /* Collapse all 'M'odified entries that
3979 * follow a associated 'U'nmerged entry.
3980 */
3981 if (file->status == 'U') {
3982 unmerged = file;
3984 } else if (unmerged) {
3985 int collapse = !strcmp(buf, unmerged->new.name);
3987 unmerged = NULL;
3988 if (collapse) {
3989 free(file);
3990 view->lines--;
3991 continue;
3992 }
3993 }
3994 }
3996 /* Grab the old name for rename/copy. */
3997 if (!*file->old.name &&
3998 (file->status == 'R' || file->status == 'C')) {
3999 sepsize = sep - buf + 1;
4000 string_ncopy(file->old.name, buf, sepsize);
4001 bufsize -= sepsize;
4002 memmove(buf, sep + 1, bufsize);
4004 sep = memchr(buf, 0, bufsize);
4005 if (!sep)
4006 break;
4007 sepsize = sep - buf + 1;
4008 }
4010 /* git-ls-files just delivers a NUL separated
4011 * list of file names similar to the second half
4012 * of the git-diff-* output. */
4013 string_ncopy(file->new.name, buf, sepsize);
4014 if (!*file->old.name)
4015 string_copy(file->old.name, file->new.name);
4016 bufsize -= sepsize;
4017 memmove(buf, sep + 1, bufsize);
4018 file = NULL;
4019 }
4020 }
4022 if (ferror(pipe)) {
4023 error_out:
4024 pclose(pipe);
4025 return FALSE;
4026 }
4028 if (!view->line[view->lines - 1].data)
4029 add_line_data(view, NULL, LINE_STAT_NONE);
4031 pclose(pipe);
4032 return TRUE;
4033 }
4035 /* Don't show unmerged entries in the staged section. */
4036 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --diff-filter=ACDMRTXB --cached -M HEAD"
4037 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
4038 #define STATUS_LIST_OTHER_CMD \
4039 "git ls-files -z --others --exclude-standard"
4040 #define STATUS_LIST_NO_HEAD_CMD \
4041 "git ls-files -z --cached --exclude-standard"
4043 #define STATUS_DIFF_INDEX_SHOW_CMD \
4044 "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s %s 2>/dev/null"
4046 #define STATUS_DIFF_FILES_SHOW_CMD \
4047 "git diff-files --root --patch-with-stat -C -M -- %s %s 2>/dev/null"
4049 #define STATUS_DIFF_NO_HEAD_SHOW_CMD \
4050 "git diff --no-color --patch-with-stat /dev/null %s 2>/dev/null"
4052 /* First parse staged info using git-diff-index(1), then parse unstaged
4053 * info using git-diff-files(1), and finally untracked files using
4054 * git-ls-files(1). */
4055 static bool
4056 status_open(struct view *view)
4057 {
4058 unsigned long prev_lineno = view->lineno;
4060 reset_view(view);
4062 if (!realloc_lines(view, view->line_size + 7))
4063 return FALSE;
4065 add_line_data(view, NULL, LINE_STAT_HEAD);
4066 if (is_initial_commit())
4067 string_copy(status_onbranch, "Initial commit");
4068 else if (!*opt_head)
4069 string_copy(status_onbranch, "Not currently on any branch");
4070 else if (!string_format(status_onbranch, "On branch %s", opt_head))
4071 return FALSE;
4073 system("git update-index -q --refresh >/dev/null 2>/dev/null");
4075 if (is_initial_commit()) {
4076 if (!status_run(view, STATUS_LIST_NO_HEAD_CMD, 'A', LINE_STAT_STAGED))
4077 return FALSE;
4078 } else if (!status_run(view, STATUS_DIFF_INDEX_CMD, 0, LINE_STAT_STAGED)) {
4079 return FALSE;
4080 }
4082 if (!status_run(view, STATUS_DIFF_FILES_CMD, 0, LINE_STAT_UNSTAGED) ||
4083 !status_run(view, STATUS_LIST_OTHER_CMD, '?', LINE_STAT_UNTRACKED))
4084 return FALSE;
4086 /* If all went well restore the previous line number to stay in
4087 * the context or select a line with something that can be
4088 * updated. */
4089 if (prev_lineno >= view->lines)
4090 prev_lineno = view->lines - 1;
4091 while (prev_lineno < view->lines && !view->line[prev_lineno].data)
4092 prev_lineno++;
4093 while (prev_lineno > 0 && !view->line[prev_lineno].data)
4094 prev_lineno--;
4096 /* If the above fails, always skip the "On branch" line. */
4097 if (prev_lineno < view->lines)
4098 view->lineno = prev_lineno;
4099 else
4100 view->lineno = 1;
4102 if (view->lineno < view->offset)
4103 view->offset = view->lineno;
4104 else if (view->offset + view->height <= view->lineno)
4105 view->offset = view->lineno - view->height + 1;
4107 return TRUE;
4108 }
4110 static bool
4111 status_draw(struct view *view, struct line *line, unsigned int lineno)
4112 {
4113 struct status *status = line->data;
4114 enum line_type type;
4115 const char *text;
4117 if (!status) {
4118 switch (line->type) {
4119 case LINE_STAT_STAGED:
4120 type = LINE_STAT_SECTION;
4121 text = "Changes to be committed:";
4122 break;
4124 case LINE_STAT_UNSTAGED:
4125 type = LINE_STAT_SECTION;
4126 text = "Changed but not updated:";
4127 break;
4129 case LINE_STAT_UNTRACKED:
4130 type = LINE_STAT_SECTION;
4131 text = "Untracked files:";
4132 break;
4134 case LINE_STAT_NONE:
4135 type = LINE_DEFAULT;
4136 text = " (no files)";
4137 break;
4139 case LINE_STAT_HEAD:
4140 type = LINE_STAT_HEAD;
4141 text = status_onbranch;
4142 break;
4144 default:
4145 return FALSE;
4146 }
4147 } else {
4148 static char buf[] = { '?', ' ', ' ', ' ', 0 };
4150 buf[0] = status->status;
4151 if (draw_text(view, line->type, buf, TRUE))
4152 return TRUE;
4153 type = LINE_DEFAULT;
4154 text = status->new.name;
4155 }
4157 draw_text(view, type, text, TRUE);
4158 return TRUE;
4159 }
4161 static enum request
4162 status_enter(struct view *view, struct line *line)
4163 {
4164 struct status *status = line->data;
4165 char oldpath[SIZEOF_STR] = "";
4166 char newpath[SIZEOF_STR] = "";
4167 const char *info;
4168 size_t cmdsize = 0;
4169 enum open_flags split;
4171 if (line->type == LINE_STAT_NONE ||
4172 (!status && line[1].type == LINE_STAT_NONE)) {
4173 report("No file to diff");
4174 return REQ_NONE;
4175 }
4177 if (status) {
4178 if (sq_quote(oldpath, 0, status->old.name) >= sizeof(oldpath))
4179 return REQ_QUIT;
4180 /* Diffs for unmerged entries are empty when pasing the
4181 * new path, so leave it empty. */
4182 if (status->status != 'U' &&
4183 sq_quote(newpath, 0, status->new.name) >= sizeof(newpath))
4184 return REQ_QUIT;
4185 }
4187 if (opt_cdup[0] &&
4188 line->type != LINE_STAT_UNTRACKED &&
4189 !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
4190 return REQ_QUIT;
4192 switch (line->type) {
4193 case LINE_STAT_STAGED:
4194 if (is_initial_commit()) {
4195 if (!string_format_from(opt_cmd, &cmdsize,
4196 STATUS_DIFF_NO_HEAD_SHOW_CMD,
4197 newpath))
4198 return REQ_QUIT;
4199 } else {
4200 if (!string_format_from(opt_cmd, &cmdsize,
4201 STATUS_DIFF_INDEX_SHOW_CMD,
4202 oldpath, newpath))
4203 return REQ_QUIT;
4204 }
4206 if (status)
4207 info = "Staged changes to %s";
4208 else
4209 info = "Staged changes";
4210 break;
4212 case LINE_STAT_UNSTAGED:
4213 if (!string_format_from(opt_cmd, &cmdsize,
4214 STATUS_DIFF_FILES_SHOW_CMD, oldpath, newpath))
4215 return REQ_QUIT;
4216 if (status)
4217 info = "Unstaged changes to %s";
4218 else
4219 info = "Unstaged changes";
4220 break;
4222 case LINE_STAT_UNTRACKED:
4223 if (opt_pipe)
4224 return REQ_QUIT;
4226 if (!status) {
4227 report("No file to show");
4228 return REQ_NONE;
4229 }
4231 if (!suffixcmp(status->new.name, -1, "/")) {
4232 report("Cannot display a directory");
4233 return REQ_NONE;
4234 }
4236 opt_pipe = fopen(status->new.name, "r");
4237 info = "Untracked file %s";
4238 break;
4240 case LINE_STAT_HEAD:
4241 return REQ_NONE;
4243 default:
4244 die("line type %d not handled in switch", line->type);
4245 }
4247 split = view_is_displayed(view) ? OPEN_SPLIT : 0;
4248 open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | split);
4249 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
4250 if (status) {
4251 stage_status = *status;
4252 } else {
4253 memset(&stage_status, 0, sizeof(stage_status));
4254 }
4256 stage_line_type = line->type;
4257 stage_chunks = 0;
4258 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
4259 }
4261 return REQ_NONE;
4262 }
4264 static bool
4265 status_exists(struct status *status, enum line_type type)
4266 {
4267 struct view *view = VIEW(REQ_VIEW_STATUS);
4268 struct line *line;
4270 for (line = view->line; line < view->line + view->lines; line++) {
4271 struct status *pos = line->data;
4273 if (line->type == type && pos &&
4274 !strcmp(status->new.name, pos->new.name))
4275 return TRUE;
4276 }
4278 return FALSE;
4279 }
4282 static FILE *
4283 status_update_prepare(enum line_type type)
4284 {
4285 char cmd[SIZEOF_STR];
4286 size_t cmdsize = 0;
4288 if (opt_cdup[0] &&
4289 type != LINE_STAT_UNTRACKED &&
4290 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4291 return NULL;
4293 switch (type) {
4294 case LINE_STAT_STAGED:
4295 string_add(cmd, cmdsize, "git update-index -z --index-info");
4296 break;
4298 case LINE_STAT_UNSTAGED:
4299 case LINE_STAT_UNTRACKED:
4300 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
4301 break;
4303 default:
4304 die("line type %d not handled in switch", type);
4305 }
4307 return popen(cmd, "w");
4308 }
4310 static bool
4311 status_update_write(FILE *pipe, struct status *status, enum line_type type)
4312 {
4313 char buf[SIZEOF_STR];
4314 size_t bufsize = 0;
4315 size_t written = 0;
4317 switch (type) {
4318 case LINE_STAT_STAGED:
4319 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
4320 status->old.mode,
4321 status->old.rev,
4322 status->old.name, 0))
4323 return FALSE;
4324 break;
4326 case LINE_STAT_UNSTAGED:
4327 case LINE_STAT_UNTRACKED:
4328 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
4329 return FALSE;
4330 break;
4332 default:
4333 die("line type %d not handled in switch", type);
4334 }
4336 while (!ferror(pipe) && written < bufsize) {
4337 written += fwrite(buf + written, 1, bufsize - written, pipe);
4338 }
4340 return written == bufsize;
4341 }
4343 static bool
4344 status_update_file(struct status *status, enum line_type type)
4345 {
4346 FILE *pipe = status_update_prepare(type);
4347 bool result;
4349 if (!pipe)
4350 return FALSE;
4352 result = status_update_write(pipe, status, type);
4353 pclose(pipe);
4354 return result;
4355 }
4357 static bool
4358 status_update_files(struct view *view, struct line *line)
4359 {
4360 FILE *pipe = status_update_prepare(line->type);
4361 bool result = TRUE;
4362 struct line *pos = view->line + view->lines;
4363 int files = 0;
4364 int file, done;
4366 if (!pipe)
4367 return FALSE;
4369 for (pos = line; pos < view->line + view->lines && pos->data; pos++)
4370 files++;
4372 for (file = 0, done = 0; result && file < files; line++, file++) {
4373 int almost_done = file * 100 / files;
4375 if (almost_done > done) {
4376 done = almost_done;
4377 string_format(view->ref, "updating file %u of %u (%d%% done)",
4378 file, files, done);
4379 update_view_title(view);
4380 }
4381 result = status_update_write(pipe, line->data, line->type);
4382 }
4384 pclose(pipe);
4385 return result;
4386 }
4388 static bool
4389 status_update(struct view *view)
4390 {
4391 struct line *line = &view->line[view->lineno];
4393 assert(view->lines);
4395 if (!line->data) {
4396 /* This should work even for the "On branch" line. */
4397 if (line < view->line + view->lines && !line[1].data) {
4398 report("Nothing to update");
4399 return FALSE;
4400 }
4402 if (!status_update_files(view, line + 1)) {
4403 report("Failed to update file status");
4404 return FALSE;
4405 }
4407 } else if (!status_update_file(line->data, line->type)) {
4408 report("Failed to update file status");
4409 return FALSE;
4410 }
4412 return TRUE;
4413 }
4415 static bool
4416 status_revert(struct status *status, enum line_type type, bool has_none)
4417 {
4418 if (!status || type != LINE_STAT_UNSTAGED) {
4419 if (type == LINE_STAT_STAGED) {
4420 report("Cannot revert changes to staged files");
4421 } else if (type == LINE_STAT_UNTRACKED) {
4422 report("Cannot revert changes to untracked files");
4423 } else if (has_none) {
4424 report("Nothing to revert");
4425 } else {
4426 report("Cannot revert changes to multiple files");
4427 }
4428 return FALSE;
4430 } else {
4431 char cmd[SIZEOF_STR];
4432 char file_sq[SIZEOF_STR];
4434 if (sq_quote(file_sq, 0, status->old.name) >= sizeof(file_sq) ||
4435 !string_format(cmd, "git checkout -- %s%s", opt_cdup, file_sq))
4436 return FALSE;
4438 return run_confirm(cmd, "Are you sure you want to overwrite any changes?");
4439 }
4440 }
4442 static enum request
4443 status_request(struct view *view, enum request request, struct line *line)
4444 {
4445 struct status *status = line->data;
4447 switch (request) {
4448 case REQ_STATUS_UPDATE:
4449 if (!status_update(view))
4450 return REQ_NONE;
4451 break;
4453 case REQ_STATUS_REVERT:
4454 if (!status_revert(status, line->type, status_has_none(view, line)))
4455 return REQ_NONE;
4456 break;
4458 case REQ_STATUS_MERGE:
4459 if (!status || status->status != 'U') {
4460 report("Merging only possible for files with unmerged status ('U').");
4461 return REQ_NONE;
4462 }
4463 open_mergetool(status->new.name);
4464 break;
4466 case REQ_EDIT:
4467 if (!status)
4468 return request;
4469 if (status->status == 'D') {
4470 report("File has been deleted.");
4471 return REQ_NONE;
4472 }
4474 open_editor(status->status != '?', status->new.name);
4475 break;
4477 case REQ_VIEW_BLAME:
4478 if (status) {
4479 string_copy(opt_file, status->new.name);
4480 opt_ref[0] = 0;
4481 }
4482 return request;
4484 case REQ_ENTER:
4485 /* After returning the status view has been split to
4486 * show the stage view. No further reloading is
4487 * necessary. */
4488 status_enter(view, line);
4489 return REQ_NONE;
4491 case REQ_REFRESH:
4492 /* Simply reload the view. */
4493 break;
4495 default:
4496 return request;
4497 }
4499 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
4501 return REQ_NONE;
4502 }
4504 static void
4505 status_select(struct view *view, struct line *line)
4506 {
4507 struct status *status = line->data;
4508 char file[SIZEOF_STR] = "all files";
4509 const char *text;
4510 const char *key;
4512 if (status && !string_format(file, "'%s'", status->new.name))
4513 return;
4515 if (!status && line[1].type == LINE_STAT_NONE)
4516 line++;
4518 switch (line->type) {
4519 case LINE_STAT_STAGED:
4520 text = "Press %s to unstage %s for commit";
4521 break;
4523 case LINE_STAT_UNSTAGED:
4524 text = "Press %s to stage %s for commit";
4525 break;
4527 case LINE_STAT_UNTRACKED:
4528 text = "Press %s to stage %s for addition";
4529 break;
4531 case LINE_STAT_HEAD:
4532 case LINE_STAT_NONE:
4533 text = "Nothing to update";
4534 break;
4536 default:
4537 die("line type %d not handled in switch", line->type);
4538 }
4540 if (status && status->status == 'U') {
4541 text = "Press %s to resolve conflict in %s";
4542 key = get_key(REQ_STATUS_MERGE);
4544 } else {
4545 key = get_key(REQ_STATUS_UPDATE);
4546 }
4548 string_format(view->ref, text, key, file);
4549 }
4551 static bool
4552 status_grep(struct view *view, struct line *line)
4553 {
4554 struct status *status = line->data;
4555 enum { S_STATUS, S_NAME, S_END } state;
4556 char buf[2] = "?";
4557 regmatch_t pmatch;
4559 if (!status)
4560 return FALSE;
4562 for (state = S_STATUS; state < S_END; state++) {
4563 const char *text;
4565 switch (state) {
4566 case S_NAME: text = status->new.name; break;
4567 case S_STATUS:
4568 buf[0] = status->status;
4569 text = buf;
4570 break;
4572 default:
4573 return FALSE;
4574 }
4576 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4577 return TRUE;
4578 }
4580 return FALSE;
4581 }
4583 static struct view_ops status_ops = {
4584 "file",
4585 status_open,
4586 NULL,
4587 status_draw,
4588 status_request,
4589 status_grep,
4590 status_select,
4591 };
4594 static bool
4595 stage_diff_line(FILE *pipe, struct line *line)
4596 {
4597 const char *buf = line->data;
4598 size_t bufsize = strlen(buf);
4599 size_t written = 0;
4601 while (!ferror(pipe) && written < bufsize) {
4602 written += fwrite(buf + written, 1, bufsize - written, pipe);
4603 }
4605 fputc('\n', pipe);
4607 return written == bufsize;
4608 }
4610 static bool
4611 stage_diff_write(FILE *pipe, struct line *line, struct line *end)
4612 {
4613 while (line < end) {
4614 if (!stage_diff_line(pipe, line++))
4615 return FALSE;
4616 if (line->type == LINE_DIFF_CHUNK ||
4617 line->type == LINE_DIFF_HEADER)
4618 break;
4619 }
4621 return TRUE;
4622 }
4624 static struct line *
4625 stage_diff_find(struct view *view, struct line *line, enum line_type type)
4626 {
4627 for (; view->line < line; line--)
4628 if (line->type == type)
4629 return line;
4631 return NULL;
4632 }
4634 static bool
4635 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
4636 {
4637 char cmd[SIZEOF_STR];
4638 size_t cmdsize = 0;
4639 struct line *diff_hdr;
4640 FILE *pipe;
4642 diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
4643 if (!diff_hdr)
4644 return FALSE;
4646 if (opt_cdup[0] &&
4647 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4648 return FALSE;
4650 if (!string_format_from(cmd, &cmdsize,
4651 "git apply --whitespace=nowarn %s %s - && "
4652 "git update-index -q --unmerged --refresh 2>/dev/null",
4653 revert ? "" : "--cached",
4654 revert || stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
4655 return FALSE;
4657 pipe = popen(cmd, "w");
4658 if (!pipe)
4659 return FALSE;
4661 if (!stage_diff_write(pipe, diff_hdr, chunk) ||
4662 !stage_diff_write(pipe, chunk, view->line + view->lines))
4663 chunk = NULL;
4665 pclose(pipe);
4667 return chunk ? TRUE : FALSE;
4668 }
4670 static bool
4671 stage_update(struct view *view, struct line *line)
4672 {
4673 struct line *chunk = NULL;
4675 if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
4676 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4678 if (chunk) {
4679 if (!stage_apply_chunk(view, chunk, FALSE)) {
4680 report("Failed to apply chunk");
4681 return FALSE;
4682 }
4684 } else if (!stage_status.status) {
4685 view = VIEW(REQ_VIEW_STATUS);
4687 for (line = view->line; line < view->line + view->lines; line++)
4688 if (line->type == stage_line_type)
4689 break;
4691 if (!status_update_files(view, line + 1)) {
4692 report("Failed to update files");
4693 return FALSE;
4694 }
4696 } else if (!status_update_file(&stage_status, stage_line_type)) {
4697 report("Failed to update file");
4698 return FALSE;
4699 }
4701 return TRUE;
4702 }
4704 static bool
4705 stage_revert(struct view *view, struct line *line)
4706 {
4707 struct line *chunk = NULL;
4709 if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
4710 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4712 if (chunk) {
4713 if (!prompt_yesno("Are you sure you want to revert changes?"))
4714 return FALSE;
4716 if (!stage_apply_chunk(view, chunk, TRUE)) {
4717 report("Failed to revert chunk");
4718 return FALSE;
4719 }
4720 return TRUE;
4722 } else {
4723 return status_revert(stage_status.status ? &stage_status : NULL,
4724 stage_line_type, FALSE);
4725 }
4726 }
4729 static void
4730 stage_next(struct view *view, struct line *line)
4731 {
4732 int i;
4734 if (!stage_chunks) {
4735 static size_t alloc = 0;
4736 int *tmp;
4738 for (line = view->line; line < view->line + view->lines; line++) {
4739 if (line->type != LINE_DIFF_CHUNK)
4740 continue;
4742 tmp = realloc_items(stage_chunk, &alloc,
4743 stage_chunks, sizeof(*tmp));
4744 if (!tmp) {
4745 report("Allocation failure");
4746 return;
4747 }
4749 stage_chunk = tmp;
4750 stage_chunk[stage_chunks++] = line - view->line;
4751 }
4752 }
4754 for (i = 0; i < stage_chunks; i++) {
4755 if (stage_chunk[i] > view->lineno) {
4756 do_scroll_view(view, stage_chunk[i] - view->lineno);
4757 report("Chunk %d of %d", i + 1, stage_chunks);
4758 return;
4759 }
4760 }
4762 report("No next chunk found");
4763 }
4765 static enum request
4766 stage_request(struct view *view, enum request request, struct line *line)
4767 {
4768 switch (request) {
4769 case REQ_STATUS_UPDATE:
4770 if (!stage_update(view, line))
4771 return REQ_NONE;
4772 break;
4774 case REQ_STATUS_REVERT:
4775 if (!stage_revert(view, line))
4776 return REQ_NONE;
4777 break;
4779 case REQ_STAGE_NEXT:
4780 if (stage_line_type == LINE_STAT_UNTRACKED) {
4781 report("File is untracked; press %s to add",
4782 get_key(REQ_STATUS_UPDATE));
4783 return REQ_NONE;
4784 }
4785 stage_next(view, line);
4786 return REQ_NONE;
4788 case REQ_EDIT:
4789 if (!stage_status.new.name[0])
4790 return request;
4791 if (stage_status.status == 'D') {
4792 report("File has been deleted.");
4793 return REQ_NONE;
4794 }
4796 open_editor(stage_status.status != '?', stage_status.new.name);
4797 break;
4799 case REQ_REFRESH:
4800 /* Reload everything ... */
4801 break;
4803 case REQ_VIEW_BLAME:
4804 if (stage_status.new.name[0]) {
4805 string_copy(opt_file, stage_status.new.name);
4806 opt_ref[0] = 0;
4807 }
4808 return request;
4810 case REQ_ENTER:
4811 return pager_request(view, request, line);
4813 default:
4814 return request;
4815 }
4817 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD | OPEN_NOMAXIMIZE);
4819 /* Check whether the staged entry still exists, and close the
4820 * stage view if it doesn't. */
4821 if (!status_exists(&stage_status, stage_line_type))
4822 return REQ_VIEW_CLOSE;
4824 if (stage_line_type == LINE_STAT_UNTRACKED) {
4825 if (!suffixcmp(stage_status.new.name, -1, "/")) {
4826 report("Cannot display a directory");
4827 return REQ_NONE;
4828 }
4830 opt_pipe = fopen(stage_status.new.name, "r");
4831 }
4832 open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
4834 return REQ_NONE;
4835 }
4837 static struct view_ops stage_ops = {
4838 "line",
4839 NULL,
4840 pager_read,
4841 pager_draw,
4842 stage_request,
4843 pager_grep,
4844 pager_select,
4845 };
4848 /*
4849 * Revision graph
4850 */
4852 struct commit {
4853 char id[SIZEOF_REV]; /* SHA1 ID. */
4854 char title[128]; /* First line of the commit message. */
4855 char author[75]; /* Author of the commit. */
4856 struct tm time; /* Date from the author ident. */
4857 struct ref **refs; /* Repository references. */
4858 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
4859 size_t graph_size; /* The width of the graph array. */
4860 bool has_parents; /* Rewritten --parents seen. */
4861 };
4863 /* Size of rev graph with no "padding" columns */
4864 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
4866 struct rev_graph {
4867 struct rev_graph *prev, *next, *parents;
4868 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
4869 size_t size;
4870 struct commit *commit;
4871 size_t pos;
4872 unsigned int boundary:1;
4873 };
4875 /* Parents of the commit being visualized. */
4876 static struct rev_graph graph_parents[4];
4878 /* The current stack of revisions on the graph. */
4879 static struct rev_graph graph_stacks[4] = {
4880 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
4881 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
4882 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
4883 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
4884 };
4886 static inline bool
4887 graph_parent_is_merge(struct rev_graph *graph)
4888 {
4889 return graph->parents->size > 1;
4890 }
4892 static inline void
4893 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
4894 {
4895 struct commit *commit = graph->commit;
4897 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
4898 commit->graph[commit->graph_size++] = symbol;
4899 }
4901 static void
4902 clear_rev_graph(struct rev_graph *graph)
4903 {
4904 graph->boundary = 0;
4905 graph->size = graph->pos = 0;
4906 graph->commit = NULL;
4907 memset(graph->parents, 0, sizeof(*graph->parents));
4908 }
4910 static void
4911 done_rev_graph(struct rev_graph *graph)
4912 {
4913 if (graph_parent_is_merge(graph) &&
4914 graph->pos < graph->size - 1 &&
4915 graph->next->size == graph->size + graph->parents->size - 1) {
4916 size_t i = graph->pos + graph->parents->size - 1;
4918 graph->commit->graph_size = i * 2;
4919 while (i < graph->next->size - 1) {
4920 append_to_rev_graph(graph, ' ');
4921 append_to_rev_graph(graph, '\\');
4922 i++;
4923 }
4924 }
4926 clear_rev_graph(graph);
4927 }
4929 static void
4930 push_rev_graph(struct rev_graph *graph, const char *parent)
4931 {
4932 int i;
4934 /* "Collapse" duplicate parents lines.
4935 *
4936 * FIXME: This needs to also update update the drawn graph but
4937 * for now it just serves as a method for pruning graph lines. */
4938 for (i = 0; i < graph->size; i++)
4939 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
4940 return;
4942 if (graph->size < SIZEOF_REVITEMS) {
4943 string_copy_rev(graph->rev[graph->size++], parent);
4944 }
4945 }
4947 static chtype
4948 get_rev_graph_symbol(struct rev_graph *graph)
4949 {
4950 chtype symbol;
4952 if (graph->boundary)
4953 symbol = REVGRAPH_BOUND;
4954 else if (graph->parents->size == 0)
4955 symbol = REVGRAPH_INIT;
4956 else if (graph_parent_is_merge(graph))
4957 symbol = REVGRAPH_MERGE;
4958 else if (graph->pos >= graph->size)
4959 symbol = REVGRAPH_BRANCH;
4960 else
4961 symbol = REVGRAPH_COMMIT;
4963 return symbol;
4964 }
4966 static void
4967 draw_rev_graph(struct rev_graph *graph)
4968 {
4969 struct rev_filler {
4970 chtype separator, line;
4971 };
4972 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
4973 static struct rev_filler fillers[] = {
4974 { ' ', '|' },
4975 { '`', '.' },
4976 { '\'', ' ' },
4977 { '/', ' ' },
4978 };
4979 chtype symbol = get_rev_graph_symbol(graph);
4980 struct rev_filler *filler;
4981 size_t i;
4983 if (opt_line_graphics)
4984 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
4986 filler = &fillers[DEFAULT];
4988 for (i = 0; i < graph->pos; i++) {
4989 append_to_rev_graph(graph, filler->line);
4990 if (graph_parent_is_merge(graph->prev) &&
4991 graph->prev->pos == i)
4992 filler = &fillers[RSHARP];
4994 append_to_rev_graph(graph, filler->separator);
4995 }
4997 /* Place the symbol for this revision. */
4998 append_to_rev_graph(graph, symbol);
5000 if (graph->prev->size > graph->size)
5001 filler = &fillers[RDIAG];
5002 else
5003 filler = &fillers[DEFAULT];
5005 i++;
5007 for (; i < graph->size; i++) {
5008 append_to_rev_graph(graph, filler->separator);
5009 append_to_rev_graph(graph, filler->line);
5010 if (graph_parent_is_merge(graph->prev) &&
5011 i < graph->prev->pos + graph->parents->size)
5012 filler = &fillers[RSHARP];
5013 if (graph->prev->size > graph->size)
5014 filler = &fillers[LDIAG];
5015 }
5017 if (graph->prev->size > graph->size) {
5018 append_to_rev_graph(graph, filler->separator);
5019 if (filler->line != ' ')
5020 append_to_rev_graph(graph, filler->line);
5021 }
5022 }
5024 /* Prepare the next rev graph */
5025 static void
5026 prepare_rev_graph(struct rev_graph *graph)
5027 {
5028 size_t i;
5030 /* First, traverse all lines of revisions up to the active one. */
5031 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5032 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5033 break;
5035 push_rev_graph(graph->next, graph->rev[graph->pos]);
5036 }
5038 /* Interleave the new revision parent(s). */
5039 for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5040 push_rev_graph(graph->next, graph->parents->rev[i]);
5042 /* Lastly, put any remaining revisions. */
5043 for (i = graph->pos + 1; i < graph->size; i++)
5044 push_rev_graph(graph->next, graph->rev[i]);
5045 }
5047 static void
5048 update_rev_graph(struct rev_graph *graph)
5049 {
5050 /* If this is the finalizing update ... */
5051 if (graph->commit)
5052 prepare_rev_graph(graph);
5054 /* Graph visualization needs a one rev look-ahead,
5055 * so the first update doesn't visualize anything. */
5056 if (!graph->prev->commit)
5057 return;
5059 draw_rev_graph(graph->prev);
5060 done_rev_graph(graph->prev->prev);
5061 }
5064 /*
5065 * Main view backend
5066 */
5068 static bool
5069 main_draw(struct view *view, struct line *line, unsigned int lineno)
5070 {
5071 struct commit *commit = line->data;
5073 if (!*commit->author)
5074 return FALSE;
5076 if (opt_date && draw_date(view, &commit->time))
5077 return TRUE;
5079 if (opt_author &&
5080 draw_field(view, LINE_MAIN_AUTHOR, commit->author, opt_author_cols, TRUE))
5081 return TRUE;
5083 if (opt_rev_graph && commit->graph_size &&
5084 draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5085 return TRUE;
5087 if (opt_show_refs && commit->refs) {
5088 size_t i = 0;
5090 do {
5091 enum line_type type;
5093 if (commit->refs[i]->head)
5094 type = LINE_MAIN_HEAD;
5095 else if (commit->refs[i]->ltag)
5096 type = LINE_MAIN_LOCAL_TAG;
5097 else if (commit->refs[i]->tag)
5098 type = LINE_MAIN_TAG;
5099 else if (commit->refs[i]->tracked)
5100 type = LINE_MAIN_TRACKED;
5101 else if (commit->refs[i]->remote)
5102 type = LINE_MAIN_REMOTE;
5103 else
5104 type = LINE_MAIN_REF;
5106 if (draw_text(view, type, "[", TRUE) ||
5107 draw_text(view, type, commit->refs[i]->name, TRUE) ||
5108 draw_text(view, type, "]", TRUE))
5109 return TRUE;
5111 if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5112 return TRUE;
5113 } while (commit->refs[i++]->next);
5114 }
5116 draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5117 return TRUE;
5118 }
5120 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5121 static bool
5122 main_read(struct view *view, char *line)
5123 {
5124 static struct rev_graph *graph = graph_stacks;
5125 enum line_type type;
5126 struct commit *commit;
5128 if (!line) {
5129 int i;
5131 if (!view->lines && !view->parent)
5132 die("No revisions match the given arguments.");
5133 if (view->lines > 0) {
5134 commit = view->line[view->lines - 1].data;
5135 if (!*commit->author) {
5136 view->lines--;
5137 free(commit);
5138 graph->commit = NULL;
5139 }
5140 }
5141 update_rev_graph(graph);
5143 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5144 clear_rev_graph(&graph_stacks[i]);
5145 return TRUE;
5146 }
5148 type = get_line_type(line);
5149 if (type == LINE_COMMIT) {
5150 commit = calloc(1, sizeof(struct commit));
5151 if (!commit)
5152 return FALSE;
5154 line += STRING_SIZE("commit ");
5155 if (*line == '-') {
5156 graph->boundary = 1;
5157 line++;
5158 }
5160 string_copy_rev(commit->id, line);
5161 commit->refs = get_refs(commit->id);
5162 graph->commit = commit;
5163 add_line_data(view, commit, LINE_MAIN_COMMIT);
5165 while ((line = strchr(line, ' '))) {
5166 line++;
5167 push_rev_graph(graph->parents, line);
5168 commit->has_parents = TRUE;
5169 }
5170 return TRUE;
5171 }
5173 if (!view->lines)
5174 return TRUE;
5175 commit = view->line[view->lines - 1].data;
5177 switch (type) {
5178 case LINE_PARENT:
5179 if (commit->has_parents)
5180 break;
5181 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5182 break;
5184 case LINE_AUTHOR:
5185 {
5186 /* Parse author lines where the name may be empty:
5187 * author <email@address.tld> 1138474660 +0100
5188 */
5189 char *ident = line + STRING_SIZE("author ");
5190 char *nameend = strchr(ident, '<');
5191 char *emailend = strchr(ident, '>');
5193 if (!nameend || !emailend)
5194 break;
5196 update_rev_graph(graph);
5197 graph = graph->next;
5199 *nameend = *emailend = 0;
5200 ident = chomp_string(ident);
5201 if (!*ident) {
5202 ident = chomp_string(nameend + 1);
5203 if (!*ident)
5204 ident = "Unknown";
5205 }
5207 string_ncopy(commit->author, ident, strlen(ident));
5209 /* Parse epoch and timezone */
5210 if (emailend[1] == ' ') {
5211 char *secs = emailend + 2;
5212 char *zone = strchr(secs, ' ');
5213 time_t time = (time_t) atol(secs);
5215 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
5216 long tz;
5218 zone++;
5219 tz = ('0' - zone[1]) * 60 * 60 * 10;
5220 tz += ('0' - zone[2]) * 60 * 60;
5221 tz += ('0' - zone[3]) * 60;
5222 tz += ('0' - zone[4]) * 60;
5224 if (zone[0] == '-')
5225 tz = -tz;
5227 time -= tz;
5228 }
5230 gmtime_r(&time, &commit->time);
5231 }
5232 break;
5233 }
5234 default:
5235 /* Fill in the commit title if it has not already been set. */
5236 if (commit->title[0])
5237 break;
5239 /* Require titles to start with a non-space character at the
5240 * offset used by git log. */
5241 if (strncmp(line, " ", 4))
5242 break;
5243 line += 4;
5244 /* Well, if the title starts with a whitespace character,
5245 * try to be forgiving. Otherwise we end up with no title. */
5246 while (isspace(*line))
5247 line++;
5248 if (*line == '\0')
5249 break;
5250 /* FIXME: More graceful handling of titles; append "..." to
5251 * shortened titles, etc. */
5253 string_ncopy(commit->title, line, strlen(line));
5254 }
5256 return TRUE;
5257 }
5259 static enum request
5260 main_request(struct view *view, enum request request, struct line *line)
5261 {
5262 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5264 switch (request) {
5265 case REQ_ENTER:
5266 open_view(view, REQ_VIEW_DIFF, flags);
5267 break;
5268 case REQ_REFRESH:
5269 load_refs();
5270 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
5271 break;
5272 default:
5273 return request;
5274 }
5276 return REQ_NONE;
5277 }
5279 static bool
5280 grep_refs(struct ref **refs, regex_t *regex)
5281 {
5282 regmatch_t pmatch;
5283 size_t i = 0;
5285 if (!refs)
5286 return FALSE;
5287 do {
5288 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
5289 return TRUE;
5290 } while (refs[i++]->next);
5292 return FALSE;
5293 }
5295 static bool
5296 main_grep(struct view *view, struct line *line)
5297 {
5298 struct commit *commit = line->data;
5299 enum { S_TITLE, S_AUTHOR, S_DATE, S_REFS, S_END } state;
5300 char buf[DATE_COLS + 1];
5301 regmatch_t pmatch;
5303 for (state = S_TITLE; state < S_END; state++) {
5304 char *text;
5306 switch (state) {
5307 case S_TITLE: text = commit->title; break;
5308 case S_AUTHOR:
5309 if (!opt_author)
5310 continue;
5311 text = commit->author;
5312 break;
5313 case S_DATE:
5314 if (!opt_date)
5315 continue;
5316 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
5317 continue;
5318 text = buf;
5319 break;
5320 case S_REFS:
5321 if (!opt_show_refs)
5322 continue;
5323 if (grep_refs(commit->refs, view->regex) == TRUE)
5324 return TRUE;
5325 continue;
5326 default:
5327 return FALSE;
5328 }
5330 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5331 return TRUE;
5332 }
5334 return FALSE;
5335 }
5337 static void
5338 main_select(struct view *view, struct line *line)
5339 {
5340 struct commit *commit = line->data;
5342 string_copy_rev(view->ref, commit->id);
5343 string_copy_rev(ref_commit, view->ref);
5344 }
5346 static struct view_ops main_ops = {
5347 "commit",
5348 NULL,
5349 main_read,
5350 main_draw,
5351 main_request,
5352 main_grep,
5353 main_select,
5354 };
5357 /*
5358 * Unicode / UTF-8 handling
5359 *
5360 * NOTE: Much of the following code for dealing with unicode is derived from
5361 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
5362 * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
5363 */
5365 /* I've (over)annotated a lot of code snippets because I am not entirely
5366 * confident that the approach taken by this small UTF-8 interface is correct.
5367 * --jonas */
5369 static inline int
5370 unicode_width(unsigned long c)
5371 {
5372 if (c >= 0x1100 &&
5373 (c <= 0x115f /* Hangul Jamo */
5374 || c == 0x2329
5375 || c == 0x232a
5376 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
5377 /* CJK ... Yi */
5378 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
5379 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
5380 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
5381 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
5382 || (c >= 0xffe0 && c <= 0xffe6)
5383 || (c >= 0x20000 && c <= 0x2fffd)
5384 || (c >= 0x30000 && c <= 0x3fffd)))
5385 return 2;
5387 if (c == '\t')
5388 return opt_tab_size;
5390 return 1;
5391 }
5393 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
5394 * Illegal bytes are set one. */
5395 static const unsigned char utf8_bytes[256] = {
5396 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,
5397 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,
5398 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,
5399 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,
5400 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,
5401 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,
5402 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,
5403 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,
5404 };
5406 /* Decode UTF-8 multi-byte representation into a unicode character. */
5407 static inline unsigned long
5408 utf8_to_unicode(const char *string, size_t length)
5409 {
5410 unsigned long unicode;
5412 switch (length) {
5413 case 1:
5414 unicode = string[0];
5415 break;
5416 case 2:
5417 unicode = (string[0] & 0x1f) << 6;
5418 unicode += (string[1] & 0x3f);
5419 break;
5420 case 3:
5421 unicode = (string[0] & 0x0f) << 12;
5422 unicode += ((string[1] & 0x3f) << 6);
5423 unicode += (string[2] & 0x3f);
5424 break;
5425 case 4:
5426 unicode = (string[0] & 0x0f) << 18;
5427 unicode += ((string[1] & 0x3f) << 12);
5428 unicode += ((string[2] & 0x3f) << 6);
5429 unicode += (string[3] & 0x3f);
5430 break;
5431 case 5:
5432 unicode = (string[0] & 0x0f) << 24;
5433 unicode += ((string[1] & 0x3f) << 18);
5434 unicode += ((string[2] & 0x3f) << 12);
5435 unicode += ((string[3] & 0x3f) << 6);
5436 unicode += (string[4] & 0x3f);
5437 break;
5438 case 6:
5439 unicode = (string[0] & 0x01) << 30;
5440 unicode += ((string[1] & 0x3f) << 24);
5441 unicode += ((string[2] & 0x3f) << 18);
5442 unicode += ((string[3] & 0x3f) << 12);
5443 unicode += ((string[4] & 0x3f) << 6);
5444 unicode += (string[5] & 0x3f);
5445 break;
5446 default:
5447 die("Invalid unicode length");
5448 }
5450 /* Invalid characters could return the special 0xfffd value but NUL
5451 * should be just as good. */
5452 return unicode > 0xffff ? 0 : unicode;
5453 }
5455 /* Calculates how much of string can be shown within the given maximum width
5456 * and sets trimmed parameter to non-zero value if all of string could not be
5457 * shown. If the reserve flag is TRUE, it will reserve at least one
5458 * trailing character, which can be useful when drawing a delimiter.
5459 *
5460 * Returns the number of bytes to output from string to satisfy max_width. */
5461 static size_t
5462 utf8_length(const char *string, int *width, size_t max_width, int *trimmed, bool reserve)
5463 {
5464 const char *start = string;
5465 const char *end = strchr(string, '\0');
5466 unsigned char last_bytes = 0;
5467 size_t last_ucwidth = 0;
5469 *width = 0;
5470 *trimmed = 0;
5472 while (string < end) {
5473 int c = *(unsigned char *) string;
5474 unsigned char bytes = utf8_bytes[c];
5475 size_t ucwidth;
5476 unsigned long unicode;
5478 if (string + bytes > end)
5479 break;
5481 /* Change representation to figure out whether
5482 * it is a single- or double-width character. */
5484 unicode = utf8_to_unicode(string, bytes);
5485 /* FIXME: Graceful handling of invalid unicode character. */
5486 if (!unicode)
5487 break;
5489 ucwidth = unicode_width(unicode);
5490 *width += ucwidth;
5491 if (*width > max_width) {
5492 *trimmed = 1;
5493 *width -= ucwidth;
5494 if (reserve && *width == max_width) {
5495 string -= last_bytes;
5496 *width -= last_ucwidth;
5497 }
5498 break;
5499 }
5501 string += bytes;
5502 last_bytes = bytes;
5503 last_ucwidth = ucwidth;
5504 }
5506 return string - start;
5507 }
5510 /*
5511 * Status management
5512 */
5514 /* Whether or not the curses interface has been initialized. */
5515 static bool cursed = FALSE;
5517 /* The status window is used for polling keystrokes. */
5518 static WINDOW *status_win;
5520 static bool status_empty = TRUE;
5522 /* Update status and title window. */
5523 static void
5524 report(const char *msg, ...)
5525 {
5526 struct view *view = display[current_view];
5528 if (input_mode)
5529 return;
5531 if (!view) {
5532 char buf[SIZEOF_STR];
5533 va_list args;
5535 va_start(args, msg);
5536 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
5537 buf[sizeof(buf) - 1] = 0;
5538 buf[sizeof(buf) - 2] = '.';
5539 buf[sizeof(buf) - 3] = '.';
5540 buf[sizeof(buf) - 4] = '.';
5541 }
5542 va_end(args);
5543 die("%s", buf);
5544 }
5546 if (!status_empty || *msg) {
5547 va_list args;
5549 va_start(args, msg);
5551 wmove(status_win, 0, 0);
5552 if (*msg) {
5553 vwprintw(status_win, msg, args);
5554 status_empty = FALSE;
5555 } else {
5556 status_empty = TRUE;
5557 }
5558 wclrtoeol(status_win);
5559 wrefresh(status_win);
5561 va_end(args);
5562 }
5564 update_view_title(view);
5565 update_display_cursor(view);
5566 }
5568 /* Controls when nodelay should be in effect when polling user input. */
5569 static void
5570 set_nonblocking_input(bool loading)
5571 {
5572 static unsigned int loading_views;
5574 if ((loading == FALSE && loading_views-- == 1) ||
5575 (loading == TRUE && loading_views++ == 0))
5576 nodelay(status_win, loading);
5577 }
5579 static void
5580 init_display(void)
5581 {
5582 int x, y;
5584 /* Initialize the curses library */
5585 if (isatty(STDIN_FILENO)) {
5586 cursed = !!initscr();
5587 opt_tty = stdin;
5588 } else {
5589 /* Leave stdin and stdout alone when acting as a pager. */
5590 opt_tty = fopen("/dev/tty", "r+");
5591 if (!opt_tty)
5592 die("Failed to open /dev/tty");
5593 cursed = !!newterm(NULL, opt_tty, opt_tty);
5594 }
5596 if (!cursed)
5597 die("Failed to initialize curses");
5599 nonl(); /* Tell curses not to do NL->CR/NL on output */
5600 cbreak(); /* Take input chars one at a time, no wait for \n */
5601 noecho(); /* Don't echo input */
5602 leaveok(stdscr, TRUE);
5604 if (has_colors())
5605 init_colors();
5607 getmaxyx(stdscr, y, x);
5608 status_win = newwin(1, 0, y - 1, 0);
5609 if (!status_win)
5610 die("Failed to create status window");
5612 /* Enable keyboard mapping */
5613 keypad(status_win, TRUE);
5614 wbkgdset(status_win, get_line_attr(LINE_STATUS));
5616 TABSIZE = opt_tab_size;
5617 if (opt_line_graphics) {
5618 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
5619 }
5620 }
5622 static bool
5623 prompt_yesno(const char *prompt)
5624 {
5625 enum { WAIT, STOP, CANCEL } status = WAIT;
5626 bool answer = FALSE;
5628 while (status == WAIT) {
5629 struct view *view;
5630 int i, key;
5632 input_mode = TRUE;
5634 foreach_view (view, i)
5635 update_view(view);
5637 input_mode = FALSE;
5639 mvwprintw(status_win, 0, 0, "%s [Yy]/[Nn]", prompt);
5640 wclrtoeol(status_win);
5642 /* Refresh, accept single keystroke of input */
5643 key = wgetch(status_win);
5644 switch (key) {
5645 case ERR:
5646 break;
5648 case 'y':
5649 case 'Y':
5650 answer = TRUE;
5651 status = STOP;
5652 break;
5654 case KEY_ESC:
5655 case KEY_RETURN:
5656 case KEY_ENTER:
5657 case KEY_BACKSPACE:
5658 case 'n':
5659 case 'N':
5660 case '\n':
5661 default:
5662 answer = FALSE;
5663 status = CANCEL;
5664 }
5665 }
5667 /* Clear the status window */
5668 status_empty = FALSE;
5669 report("");
5671 return answer;
5672 }
5674 static char *
5675 read_prompt(const char *prompt)
5676 {
5677 enum { READING, STOP, CANCEL } status = READING;
5678 static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
5679 int pos = 0;
5681 while (status == READING) {
5682 struct view *view;
5683 int i, key;
5685 input_mode = TRUE;
5687 foreach_view (view, i)
5688 update_view(view);
5690 input_mode = FALSE;
5692 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
5693 wclrtoeol(status_win);
5695 /* Refresh, accept single keystroke of input */
5696 key = wgetch(status_win);
5697 switch (key) {
5698 case KEY_RETURN:
5699 case KEY_ENTER:
5700 case '\n':
5701 status = pos ? STOP : CANCEL;
5702 break;
5704 case KEY_BACKSPACE:
5705 if (pos > 0)
5706 pos--;
5707 else
5708 status = CANCEL;
5709 break;
5711 case KEY_ESC:
5712 status = CANCEL;
5713 break;
5715 case ERR:
5716 break;
5718 default:
5719 if (pos >= sizeof(buf)) {
5720 report("Input string too long");
5721 return NULL;
5722 }
5724 if (isprint(key))
5725 buf[pos++] = (char) key;
5726 }
5727 }
5729 /* Clear the status window */
5730 status_empty = FALSE;
5731 report("");
5733 if (status == CANCEL)
5734 return NULL;
5736 buf[pos++] = 0;
5738 return buf;
5739 }
5741 /*
5742 * Repository references
5743 */
5745 static struct ref *refs = NULL;
5746 static size_t refs_alloc = 0;
5747 static size_t refs_size = 0;
5749 /* Id <-> ref store */
5750 static struct ref ***id_refs = NULL;
5751 static size_t id_refs_alloc = 0;
5752 static size_t id_refs_size = 0;
5754 static int
5755 compare_refs(const void *ref1_, const void *ref2_)
5756 {
5757 const struct ref *ref1 = *(const struct ref **)ref1_;
5758 const struct ref *ref2 = *(const struct ref **)ref2_;
5760 if (ref1->tag != ref2->tag)
5761 return ref2->tag - ref1->tag;
5762 if (ref1->ltag != ref2->ltag)
5763 return ref2->ltag - ref2->ltag;
5764 if (ref1->head != ref2->head)
5765 return ref2->head - ref1->head;
5766 if (ref1->tracked != ref2->tracked)
5767 return ref2->tracked - ref1->tracked;
5768 if (ref1->remote != ref2->remote)
5769 return ref2->remote - ref1->remote;
5770 return strcmp(ref1->name, ref2->name);
5771 }
5773 static struct ref **
5774 get_refs(const char *id)
5775 {
5776 struct ref ***tmp_id_refs;
5777 struct ref **ref_list = NULL;
5778 size_t ref_list_alloc = 0;
5779 size_t ref_list_size = 0;
5780 size_t i;
5782 for (i = 0; i < id_refs_size; i++)
5783 if (!strcmp(id, id_refs[i][0]->id))
5784 return id_refs[i];
5786 tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
5787 sizeof(*id_refs));
5788 if (!tmp_id_refs)
5789 return NULL;
5791 id_refs = tmp_id_refs;
5793 for (i = 0; i < refs_size; i++) {
5794 struct ref **tmp;
5796 if (strcmp(id, refs[i].id))
5797 continue;
5799 tmp = realloc_items(ref_list, &ref_list_alloc,
5800 ref_list_size + 1, sizeof(*ref_list));
5801 if (!tmp) {
5802 if (ref_list)
5803 free(ref_list);
5804 return NULL;
5805 }
5807 ref_list = tmp;
5808 ref_list[ref_list_size] = &refs[i];
5809 /* XXX: The properties of the commit chains ensures that we can
5810 * safely modify the shared ref. The repo references will
5811 * always be similar for the same id. */
5812 ref_list[ref_list_size]->next = 1;
5814 ref_list_size++;
5815 }
5817 if (ref_list) {
5818 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
5819 ref_list[ref_list_size - 1]->next = 0;
5820 id_refs[id_refs_size++] = ref_list;
5821 }
5823 return ref_list;
5824 }
5826 static int
5827 read_ref(char *id, size_t idlen, char *name, size_t namelen)
5828 {
5829 struct ref *ref;
5830 bool tag = FALSE;
5831 bool ltag = FALSE;
5832 bool remote = FALSE;
5833 bool tracked = FALSE;
5834 bool check_replace = FALSE;
5835 bool head = FALSE;
5837 if (!prefixcmp(name, "refs/tags/")) {
5838 if (!suffixcmp(name, namelen, "^{}")) {
5839 namelen -= 3;
5840 name[namelen] = 0;
5841 if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
5842 check_replace = TRUE;
5843 } else {
5844 ltag = TRUE;
5845 }
5847 tag = TRUE;
5848 namelen -= STRING_SIZE("refs/tags/");
5849 name += STRING_SIZE("refs/tags/");
5851 } else if (!prefixcmp(name, "refs/remotes/")) {
5852 remote = TRUE;
5853 namelen -= STRING_SIZE("refs/remotes/");
5854 name += STRING_SIZE("refs/remotes/");
5855 tracked = !strcmp(opt_remote, name);
5857 } else if (!prefixcmp(name, "refs/heads/")) {
5858 namelen -= STRING_SIZE("refs/heads/");
5859 name += STRING_SIZE("refs/heads/");
5860 head = !strncmp(opt_head, name, namelen);
5862 } else if (!strcmp(name, "HEAD")) {
5863 string_ncopy(opt_head_rev, id, idlen);
5864 return OK;
5865 }
5867 if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
5868 /* it's an annotated tag, replace the previous sha1 with the
5869 * resolved commit id; relies on the fact git-ls-remote lists
5870 * the commit id of an annotated tag right before the commit id
5871 * it points to. */
5872 refs[refs_size - 1].ltag = ltag;
5873 string_copy_rev(refs[refs_size - 1].id, id);
5875 return OK;
5876 }
5877 refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
5878 if (!refs)
5879 return ERR;
5881 ref = &refs[refs_size++];
5882 ref->name = malloc(namelen + 1);
5883 if (!ref->name)
5884 return ERR;
5886 strncpy(ref->name, name, namelen);
5887 ref->name[namelen] = 0;
5888 ref->head = head;
5889 ref->tag = tag;
5890 ref->ltag = ltag;
5891 ref->remote = remote;
5892 ref->tracked = tracked;
5893 string_copy_rev(ref->id, id);
5895 return OK;
5896 }
5898 static int
5899 load_refs(void)
5900 {
5901 const char *cmd_env = getenv("TIG_LS_REMOTE");
5902 const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
5904 if (!*opt_git_dir)
5905 return OK;
5907 while (refs_size > 0)
5908 free(refs[--refs_size].name);
5909 while (id_refs_size > 0)
5910 free(id_refs[--id_refs_size]);
5912 return read_properties(popen(cmd, "r"), "\t", read_ref);
5913 }
5915 static int
5916 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
5917 {
5918 if (!strcmp(name, "i18n.commitencoding"))
5919 string_ncopy(opt_encoding, value, valuelen);
5921 if (!strcmp(name, "core.editor"))
5922 string_ncopy(opt_editor, value, valuelen);
5924 /* branch.<head>.remote */
5925 if (*opt_head &&
5926 !strncmp(name, "branch.", 7) &&
5927 !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5928 !strcmp(name + 7 + strlen(opt_head), ".remote"))
5929 string_ncopy(opt_remote, value, valuelen);
5931 if (*opt_head && *opt_remote &&
5932 !strncmp(name, "branch.", 7) &&
5933 !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5934 !strcmp(name + 7 + strlen(opt_head), ".merge")) {
5935 size_t from = strlen(opt_remote);
5937 if (!prefixcmp(value, "refs/heads/")) {
5938 value += STRING_SIZE("refs/heads/");
5939 valuelen -= STRING_SIZE("refs/heads/");
5940 }
5942 if (!string_format_from(opt_remote, &from, "/%s", value))
5943 opt_remote[0] = 0;
5944 }
5946 return OK;
5947 }
5949 static int
5950 load_git_config(void)
5951 {
5952 return read_properties(popen("git " GIT_CONFIG " --list", "r"),
5953 "=", read_repo_config_option);
5954 }
5956 static int
5957 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
5958 {
5959 if (!opt_git_dir[0]) {
5960 string_ncopy(opt_git_dir, name, namelen);
5962 } else if (opt_is_inside_work_tree == -1) {
5963 /* This can be 3 different values depending on the
5964 * version of git being used. If git-rev-parse does not
5965 * understand --is-inside-work-tree it will simply echo
5966 * the option else either "true" or "false" is printed.
5967 * Default to true for the unknown case. */
5968 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
5970 } else if (opt_cdup[0] == ' ') {
5971 string_ncopy(opt_cdup, name, namelen);
5972 } else {
5973 if (!prefixcmp(name, "refs/heads/")) {
5974 namelen -= STRING_SIZE("refs/heads/");
5975 name += STRING_SIZE("refs/heads/");
5976 string_ncopy(opt_head, name, namelen);
5977 }
5978 }
5980 return OK;
5981 }
5983 static int
5984 load_repo_info(void)
5985 {
5986 int result;
5987 FILE *pipe = popen("(git rev-parse --git-dir --is-inside-work-tree "
5988 " --show-cdup; git symbolic-ref HEAD) 2>/dev/null", "r");
5990 /* XXX: The line outputted by "--show-cdup" can be empty so
5991 * initialize it to something invalid to make it possible to
5992 * detect whether it has been set or not. */
5993 opt_cdup[0] = ' ';
5995 result = read_properties(pipe, "=", read_repo_info);
5996 if (opt_cdup[0] == ' ')
5997 opt_cdup[0] = 0;
5999 return result;
6000 }
6002 static int
6003 read_properties(FILE *pipe, const char *separators,
6004 int (*read_property)(char *, size_t, char *, size_t))
6005 {
6006 char buffer[BUFSIZ];
6007 char *name;
6008 int state = OK;
6010 if (!pipe)
6011 return ERR;
6013 while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
6014 char *value;
6015 size_t namelen;
6016 size_t valuelen;
6018 name = chomp_string(name);
6019 namelen = strcspn(name, separators);
6021 if (name[namelen]) {
6022 name[namelen] = 0;
6023 value = chomp_string(name + namelen + 1);
6024 valuelen = strlen(value);
6026 } else {
6027 value = "";
6028 valuelen = 0;
6029 }
6031 state = read_property(name, namelen, value, valuelen);
6032 }
6034 if (state != ERR && ferror(pipe))
6035 state = ERR;
6037 pclose(pipe);
6039 return state;
6040 }
6043 /*
6044 * Main
6045 */
6047 static void __NORETURN
6048 quit(int sig)
6049 {
6050 /* XXX: Restore tty modes and let the OS cleanup the rest! */
6051 if (cursed)
6052 endwin();
6053 exit(0);
6054 }
6056 static void __NORETURN
6057 die(const char *err, ...)
6058 {
6059 va_list args;
6061 endwin();
6063 va_start(args, err);
6064 fputs("tig: ", stderr);
6065 vfprintf(stderr, err, args);
6066 fputs("\n", stderr);
6067 va_end(args);
6069 exit(1);
6070 }
6072 static void
6073 warn(const char *msg, ...)
6074 {
6075 va_list args;
6077 va_start(args, msg);
6078 fputs("tig warning: ", stderr);
6079 vfprintf(stderr, msg, args);
6080 fputs("\n", stderr);
6081 va_end(args);
6082 }
6084 int
6085 main(int argc, const char *argv[])
6086 {
6087 struct view *view;
6088 enum request request;
6089 size_t i;
6091 signal(SIGINT, quit);
6093 if (setlocale(LC_ALL, "")) {
6094 char *codeset = nl_langinfo(CODESET);
6096 string_ncopy(opt_codeset, codeset, strlen(codeset));
6097 }
6099 if (load_repo_info() == ERR)
6100 die("Failed to load repo info.");
6102 if (load_options() == ERR)
6103 die("Failed to load user config.");
6105 if (load_git_config() == ERR)
6106 die("Failed to load repo config.");
6108 request = parse_options(argc, argv);
6109 if (request == REQ_NONE)
6110 return 0;
6112 /* Require a git repository unless when running in pager mode. */
6113 if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
6114 die("Not a git repository");
6116 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
6117 opt_utf8 = FALSE;
6119 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
6120 opt_iconv = iconv_open(opt_codeset, opt_encoding);
6121 if (opt_iconv == ICONV_NONE)
6122 die("Failed to initialize character set conversion");
6123 }
6125 if (load_refs() == ERR)
6126 die("Failed to load refs.");
6128 foreach_view (view, i)
6129 view->cmd_env = getenv(view->cmd_env);
6131 init_display();
6133 while (view_driver(display[current_view], request)) {
6134 int key;
6135 int i;
6137 foreach_view (view, i)
6138 update_view(view);
6139 view = display[current_view];
6141 /* Refresh, accept single keystroke of input */
6142 key = wgetch(status_win);
6144 /* wgetch() with nodelay() enabled returns ERR when there's no
6145 * input. */
6146 if (key == ERR) {
6147 request = REQ_NONE;
6148 continue;
6149 }
6151 request = get_keybinding(view->keymap, key);
6153 /* Some low-level request handling. This keeps access to
6154 * status_win restricted. */
6155 switch (request) {
6156 case REQ_PROMPT:
6157 {
6158 char *cmd = read_prompt(":");
6160 if (cmd && string_format(opt_cmd, "git %s", cmd)) {
6161 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
6162 request = REQ_VIEW_DIFF;
6163 } else {
6164 request = REQ_VIEW_PAGER;
6165 }
6167 /* Always reload^Wrerun commands from the prompt. */
6168 open_view(view, request, OPEN_RELOAD);
6169 }
6171 request = REQ_NONE;
6172 break;
6173 }
6174 case REQ_SEARCH:
6175 case REQ_SEARCH_BACK:
6176 {
6177 const char *prompt = request == REQ_SEARCH ? "/" : "?";
6178 char *search = read_prompt(prompt);
6180 if (search)
6181 string_ncopy(opt_search, search, strlen(search));
6182 else
6183 request = REQ_NONE;
6184 break;
6185 }
6186 case REQ_SCREEN_RESIZE:
6187 {
6188 int height, width;
6190 getmaxyx(stdscr, height, width);
6192 /* Resize the status view and let the view driver take
6193 * care of resizing the displayed views. */
6194 wresize(status_win, 1, width);
6195 mvwin(status_win, height - 1, 0);
6196 wrefresh(status_win);
6197 break;
6198 }
6199 default:
6200 break;
6201 }
6202 }
6204 quit(0);
6206 return 0;
6207 }