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 (!strcmp(blame->commit->id, NULL_ID)) {
3783 char path[SIZEOF_STR];
3785 if (sq_quote(path, 0, view->vid) >= sizeof(path))
3786 break;
3787 string_format(opt_cmd, "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s 2>/dev/null", path);
3788 }
3790 open_view(view, REQ_VIEW_DIFF, flags);
3791 break;
3793 default:
3794 return request;
3795 }
3797 return REQ_NONE;
3798 }
3800 static bool
3801 blame_grep(struct view *view, struct line *line)
3802 {
3803 struct blame *blame = line->data;
3804 struct blame_commit *commit = blame->commit;
3805 regmatch_t pmatch;
3807 #define MATCH(text, on) \
3808 (on && *text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3810 if (commit) {
3811 char buf[DATE_COLS + 1];
3813 if (MATCH(commit->title, 1) ||
3814 MATCH(commit->author, opt_author) ||
3815 MATCH(commit->id, opt_date))
3816 return TRUE;
3818 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
3819 MATCH(buf, 1))
3820 return TRUE;
3821 }
3823 return MATCH(blame->text, 1);
3825 #undef MATCH
3826 }
3828 static void
3829 blame_select(struct view *view, struct line *line)
3830 {
3831 struct blame *blame = line->data;
3832 struct blame_commit *commit = blame->commit;
3834 if (!commit)
3835 return;
3837 if (!strcmp(commit->id, NULL_ID))
3838 string_ncopy(ref_commit, "HEAD", 4);
3839 else
3840 string_copy_rev(ref_commit, commit->id);
3841 }
3843 static struct view_ops blame_ops = {
3844 "line",
3845 blame_open,
3846 blame_read,
3847 blame_draw,
3848 blame_request,
3849 blame_grep,
3850 blame_select,
3851 };
3853 /*
3854 * Status backend
3855 */
3857 struct status {
3858 char status;
3859 struct {
3860 mode_t mode;
3861 char rev[SIZEOF_REV];
3862 char name[SIZEOF_STR];
3863 } old;
3864 struct {
3865 mode_t mode;
3866 char rev[SIZEOF_REV];
3867 char name[SIZEOF_STR];
3868 } new;
3869 };
3871 static char status_onbranch[SIZEOF_STR];
3872 static struct status stage_status;
3873 static enum line_type stage_line_type;
3874 static size_t stage_chunks;
3875 static int *stage_chunk;
3877 /* This should work even for the "On branch" line. */
3878 static inline bool
3879 status_has_none(struct view *view, struct line *line)
3880 {
3881 return line < view->line + view->lines && !line[1].data;
3882 }
3884 /* Get fields from the diff line:
3885 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
3886 */
3887 static inline bool
3888 status_get_diff(struct status *file, const char *buf, size_t bufsize)
3889 {
3890 const char *old_mode = buf + 1;
3891 const char *new_mode = buf + 8;
3892 const char *old_rev = buf + 15;
3893 const char *new_rev = buf + 56;
3894 const char *status = buf + 97;
3896 if (bufsize < 99 ||
3897 old_mode[-1] != ':' ||
3898 new_mode[-1] != ' ' ||
3899 old_rev[-1] != ' ' ||
3900 new_rev[-1] != ' ' ||
3901 status[-1] != ' ')
3902 return FALSE;
3904 file->status = *status;
3906 string_copy_rev(file->old.rev, old_rev);
3907 string_copy_rev(file->new.rev, new_rev);
3909 file->old.mode = strtoul(old_mode, NULL, 8);
3910 file->new.mode = strtoul(new_mode, NULL, 8);
3912 file->old.name[0] = file->new.name[0] = 0;
3914 return TRUE;
3915 }
3917 static bool
3918 status_run(struct view *view, const char cmd[], char status, enum line_type type)
3919 {
3920 struct status *file = NULL;
3921 struct status *unmerged = NULL;
3922 char buf[SIZEOF_STR * 4];
3923 size_t bufsize = 0;
3924 FILE *pipe;
3926 pipe = popen(cmd, "r");
3927 if (!pipe)
3928 return FALSE;
3930 add_line_data(view, NULL, type);
3932 while (!feof(pipe) && !ferror(pipe)) {
3933 char *sep;
3934 size_t readsize;
3936 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
3937 if (!readsize)
3938 break;
3939 bufsize += readsize;
3941 /* Process while we have NUL chars. */
3942 while ((sep = memchr(buf, 0, bufsize))) {
3943 size_t sepsize = sep - buf + 1;
3945 if (!file) {
3946 if (!realloc_lines(view, view->line_size + 1))
3947 goto error_out;
3949 file = calloc(1, sizeof(*file));
3950 if (!file)
3951 goto error_out;
3953 add_line_data(view, file, type);
3954 }
3956 /* Parse diff info part. */
3957 if (status) {
3958 file->status = status;
3959 if (status == 'A')
3960 string_copy(file->old.rev, NULL_ID);
3962 } else if (!file->status) {
3963 if (!status_get_diff(file, buf, sepsize))
3964 goto error_out;
3966 bufsize -= sepsize;
3967 memmove(buf, sep + 1, bufsize);
3969 sep = memchr(buf, 0, bufsize);
3970 if (!sep)
3971 break;
3972 sepsize = sep - buf + 1;
3974 /* Collapse all 'M'odified entries that
3975 * follow a associated 'U'nmerged entry.
3976 */
3977 if (file->status == 'U') {
3978 unmerged = file;
3980 } else if (unmerged) {
3981 int collapse = !strcmp(buf, unmerged->new.name);
3983 unmerged = NULL;
3984 if (collapse) {
3985 free(file);
3986 view->lines--;
3987 continue;
3988 }
3989 }
3990 }
3992 /* Grab the old name for rename/copy. */
3993 if (!*file->old.name &&
3994 (file->status == 'R' || file->status == 'C')) {
3995 sepsize = sep - buf + 1;
3996 string_ncopy(file->old.name, buf, sepsize);
3997 bufsize -= sepsize;
3998 memmove(buf, sep + 1, bufsize);
4000 sep = memchr(buf, 0, bufsize);
4001 if (!sep)
4002 break;
4003 sepsize = sep - buf + 1;
4004 }
4006 /* git-ls-files just delivers a NUL separated
4007 * list of file names similar to the second half
4008 * of the git-diff-* output. */
4009 string_ncopy(file->new.name, buf, sepsize);
4010 if (!*file->old.name)
4011 string_copy(file->old.name, file->new.name);
4012 bufsize -= sepsize;
4013 memmove(buf, sep + 1, bufsize);
4014 file = NULL;
4015 }
4016 }
4018 if (ferror(pipe)) {
4019 error_out:
4020 pclose(pipe);
4021 return FALSE;
4022 }
4024 if (!view->line[view->lines - 1].data)
4025 add_line_data(view, NULL, LINE_STAT_NONE);
4027 pclose(pipe);
4028 return TRUE;
4029 }
4031 /* Don't show unmerged entries in the staged section. */
4032 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --diff-filter=ACDMRTXB --cached -M HEAD"
4033 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
4034 #define STATUS_LIST_OTHER_CMD \
4035 "git ls-files -z --others --exclude-standard"
4036 #define STATUS_LIST_NO_HEAD_CMD \
4037 "git ls-files -z --cached --exclude-standard"
4039 #define STATUS_DIFF_INDEX_SHOW_CMD \
4040 "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s %s 2>/dev/null"
4042 #define STATUS_DIFF_FILES_SHOW_CMD \
4043 "git diff-files --root --patch-with-stat -C -M -- %s %s 2>/dev/null"
4045 #define STATUS_DIFF_NO_HEAD_SHOW_CMD \
4046 "git diff --no-color --patch-with-stat /dev/null %s 2>/dev/null"
4048 /* First parse staged info using git-diff-index(1), then parse unstaged
4049 * info using git-diff-files(1), and finally untracked files using
4050 * git-ls-files(1). */
4051 static bool
4052 status_open(struct view *view)
4053 {
4054 unsigned long prev_lineno = view->lineno;
4056 reset_view(view);
4058 if (!realloc_lines(view, view->line_size + 7))
4059 return FALSE;
4061 add_line_data(view, NULL, LINE_STAT_HEAD);
4062 if (is_initial_commit())
4063 string_copy(status_onbranch, "Initial commit");
4064 else if (!*opt_head)
4065 string_copy(status_onbranch, "Not currently on any branch");
4066 else if (!string_format(status_onbranch, "On branch %s", opt_head))
4067 return FALSE;
4069 system("git update-index -q --refresh >/dev/null 2>/dev/null");
4071 if (is_initial_commit()) {
4072 if (!status_run(view, STATUS_LIST_NO_HEAD_CMD, 'A', LINE_STAT_STAGED))
4073 return FALSE;
4074 } else if (!status_run(view, STATUS_DIFF_INDEX_CMD, 0, LINE_STAT_STAGED)) {
4075 return FALSE;
4076 }
4078 if (!status_run(view, STATUS_DIFF_FILES_CMD, 0, LINE_STAT_UNSTAGED) ||
4079 !status_run(view, STATUS_LIST_OTHER_CMD, '?', LINE_STAT_UNTRACKED))
4080 return FALSE;
4082 /* If all went well restore the previous line number to stay in
4083 * the context or select a line with something that can be
4084 * updated. */
4085 if (prev_lineno >= view->lines)
4086 prev_lineno = view->lines - 1;
4087 while (prev_lineno < view->lines && !view->line[prev_lineno].data)
4088 prev_lineno++;
4089 while (prev_lineno > 0 && !view->line[prev_lineno].data)
4090 prev_lineno--;
4092 /* If the above fails, always skip the "On branch" line. */
4093 if (prev_lineno < view->lines)
4094 view->lineno = prev_lineno;
4095 else
4096 view->lineno = 1;
4098 if (view->lineno < view->offset)
4099 view->offset = view->lineno;
4100 else if (view->offset + view->height <= view->lineno)
4101 view->offset = view->lineno - view->height + 1;
4103 return TRUE;
4104 }
4106 static bool
4107 status_draw(struct view *view, struct line *line, unsigned int lineno)
4108 {
4109 struct status *status = line->data;
4110 enum line_type type;
4111 const char *text;
4113 if (!status) {
4114 switch (line->type) {
4115 case LINE_STAT_STAGED:
4116 type = LINE_STAT_SECTION;
4117 text = "Changes to be committed:";
4118 break;
4120 case LINE_STAT_UNSTAGED:
4121 type = LINE_STAT_SECTION;
4122 text = "Changed but not updated:";
4123 break;
4125 case LINE_STAT_UNTRACKED:
4126 type = LINE_STAT_SECTION;
4127 text = "Untracked files:";
4128 break;
4130 case LINE_STAT_NONE:
4131 type = LINE_DEFAULT;
4132 text = " (no files)";
4133 break;
4135 case LINE_STAT_HEAD:
4136 type = LINE_STAT_HEAD;
4137 text = status_onbranch;
4138 break;
4140 default:
4141 return FALSE;
4142 }
4143 } else {
4144 static char buf[] = { '?', ' ', ' ', ' ', 0 };
4146 buf[0] = status->status;
4147 if (draw_text(view, line->type, buf, TRUE))
4148 return TRUE;
4149 type = LINE_DEFAULT;
4150 text = status->new.name;
4151 }
4153 draw_text(view, type, text, TRUE);
4154 return TRUE;
4155 }
4157 static enum request
4158 status_enter(struct view *view, struct line *line)
4159 {
4160 struct status *status = line->data;
4161 char oldpath[SIZEOF_STR] = "";
4162 char newpath[SIZEOF_STR] = "";
4163 const char *info;
4164 size_t cmdsize = 0;
4165 enum open_flags split;
4167 if (line->type == LINE_STAT_NONE ||
4168 (!status && line[1].type == LINE_STAT_NONE)) {
4169 report("No file to diff");
4170 return REQ_NONE;
4171 }
4173 if (status) {
4174 if (sq_quote(oldpath, 0, status->old.name) >= sizeof(oldpath))
4175 return REQ_QUIT;
4176 /* Diffs for unmerged entries are empty when pasing the
4177 * new path, so leave it empty. */
4178 if (status->status != 'U' &&
4179 sq_quote(newpath, 0, status->new.name) >= sizeof(newpath))
4180 return REQ_QUIT;
4181 }
4183 if (opt_cdup[0] &&
4184 line->type != LINE_STAT_UNTRACKED &&
4185 !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
4186 return REQ_QUIT;
4188 switch (line->type) {
4189 case LINE_STAT_STAGED:
4190 if (is_initial_commit()) {
4191 if (!string_format_from(opt_cmd, &cmdsize,
4192 STATUS_DIFF_NO_HEAD_SHOW_CMD,
4193 newpath))
4194 return REQ_QUIT;
4195 } else {
4196 if (!string_format_from(opt_cmd, &cmdsize,
4197 STATUS_DIFF_INDEX_SHOW_CMD,
4198 oldpath, newpath))
4199 return REQ_QUIT;
4200 }
4202 if (status)
4203 info = "Staged changes to %s";
4204 else
4205 info = "Staged changes";
4206 break;
4208 case LINE_STAT_UNSTAGED:
4209 if (!string_format_from(opt_cmd, &cmdsize,
4210 STATUS_DIFF_FILES_SHOW_CMD, oldpath, newpath))
4211 return REQ_QUIT;
4212 if (status)
4213 info = "Unstaged changes to %s";
4214 else
4215 info = "Unstaged changes";
4216 break;
4218 case LINE_STAT_UNTRACKED:
4219 if (opt_pipe)
4220 return REQ_QUIT;
4222 if (!status) {
4223 report("No file to show");
4224 return REQ_NONE;
4225 }
4227 if (!suffixcmp(status->new.name, -1, "/")) {
4228 report("Cannot display a directory");
4229 return REQ_NONE;
4230 }
4232 opt_pipe = fopen(status->new.name, "r");
4233 info = "Untracked file %s";
4234 break;
4236 case LINE_STAT_HEAD:
4237 return REQ_NONE;
4239 default:
4240 die("line type %d not handled in switch", line->type);
4241 }
4243 split = view_is_displayed(view) ? OPEN_SPLIT : 0;
4244 open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | split);
4245 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
4246 if (status) {
4247 stage_status = *status;
4248 } else {
4249 memset(&stage_status, 0, sizeof(stage_status));
4250 }
4252 stage_line_type = line->type;
4253 stage_chunks = 0;
4254 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
4255 }
4257 return REQ_NONE;
4258 }
4260 static bool
4261 status_exists(struct status *status, enum line_type type)
4262 {
4263 struct view *view = VIEW(REQ_VIEW_STATUS);
4264 struct line *line;
4266 for (line = view->line; line < view->line + view->lines; line++) {
4267 struct status *pos = line->data;
4269 if (line->type == type && pos &&
4270 !strcmp(status->new.name, pos->new.name))
4271 return TRUE;
4272 }
4274 return FALSE;
4275 }
4278 static FILE *
4279 status_update_prepare(enum line_type type)
4280 {
4281 char cmd[SIZEOF_STR];
4282 size_t cmdsize = 0;
4284 if (opt_cdup[0] &&
4285 type != LINE_STAT_UNTRACKED &&
4286 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4287 return NULL;
4289 switch (type) {
4290 case LINE_STAT_STAGED:
4291 string_add(cmd, cmdsize, "git update-index -z --index-info");
4292 break;
4294 case LINE_STAT_UNSTAGED:
4295 case LINE_STAT_UNTRACKED:
4296 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
4297 break;
4299 default:
4300 die("line type %d not handled in switch", type);
4301 }
4303 return popen(cmd, "w");
4304 }
4306 static bool
4307 status_update_write(FILE *pipe, struct status *status, enum line_type type)
4308 {
4309 char buf[SIZEOF_STR];
4310 size_t bufsize = 0;
4311 size_t written = 0;
4313 switch (type) {
4314 case LINE_STAT_STAGED:
4315 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
4316 status->old.mode,
4317 status->old.rev,
4318 status->old.name, 0))
4319 return FALSE;
4320 break;
4322 case LINE_STAT_UNSTAGED:
4323 case LINE_STAT_UNTRACKED:
4324 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
4325 return FALSE;
4326 break;
4328 default:
4329 die("line type %d not handled in switch", type);
4330 }
4332 while (!ferror(pipe) && written < bufsize) {
4333 written += fwrite(buf + written, 1, bufsize - written, pipe);
4334 }
4336 return written == bufsize;
4337 }
4339 static bool
4340 status_update_file(struct status *status, enum line_type type)
4341 {
4342 FILE *pipe = status_update_prepare(type);
4343 bool result;
4345 if (!pipe)
4346 return FALSE;
4348 result = status_update_write(pipe, status, type);
4349 pclose(pipe);
4350 return result;
4351 }
4353 static bool
4354 status_update_files(struct view *view, struct line *line)
4355 {
4356 FILE *pipe = status_update_prepare(line->type);
4357 bool result = TRUE;
4358 struct line *pos = view->line + view->lines;
4359 int files = 0;
4360 int file, done;
4362 if (!pipe)
4363 return FALSE;
4365 for (pos = line; pos < view->line + view->lines && pos->data; pos++)
4366 files++;
4368 for (file = 0, done = 0; result && file < files; line++, file++) {
4369 int almost_done = file * 100 / files;
4371 if (almost_done > done) {
4372 done = almost_done;
4373 string_format(view->ref, "updating file %u of %u (%d%% done)",
4374 file, files, done);
4375 update_view_title(view);
4376 }
4377 result = status_update_write(pipe, line->data, line->type);
4378 }
4380 pclose(pipe);
4381 return result;
4382 }
4384 static bool
4385 status_update(struct view *view)
4386 {
4387 struct line *line = &view->line[view->lineno];
4389 assert(view->lines);
4391 if (!line->data) {
4392 /* This should work even for the "On branch" line. */
4393 if (line < view->line + view->lines && !line[1].data) {
4394 report("Nothing to update");
4395 return FALSE;
4396 }
4398 if (!status_update_files(view, line + 1)) {
4399 report("Failed to update file status");
4400 return FALSE;
4401 }
4403 } else if (!status_update_file(line->data, line->type)) {
4404 report("Failed to update file status");
4405 return FALSE;
4406 }
4408 return TRUE;
4409 }
4411 static bool
4412 status_revert(struct status *status, enum line_type type, bool has_none)
4413 {
4414 if (!status || type != LINE_STAT_UNSTAGED) {
4415 if (type == LINE_STAT_STAGED) {
4416 report("Cannot revert changes to staged files");
4417 } else if (type == LINE_STAT_UNTRACKED) {
4418 report("Cannot revert changes to untracked files");
4419 } else if (has_none) {
4420 report("Nothing to revert");
4421 } else {
4422 report("Cannot revert changes to multiple files");
4423 }
4424 return FALSE;
4426 } else {
4427 char cmd[SIZEOF_STR];
4428 char file_sq[SIZEOF_STR];
4430 if (sq_quote(file_sq, 0, status->old.name) >= sizeof(file_sq) ||
4431 !string_format(cmd, "git checkout -- %s%s", opt_cdup, file_sq))
4432 return FALSE;
4434 return run_confirm(cmd, "Are you sure you want to overwrite any changes?");
4435 }
4436 }
4438 static enum request
4439 status_request(struct view *view, enum request request, struct line *line)
4440 {
4441 struct status *status = line->data;
4443 switch (request) {
4444 case REQ_STATUS_UPDATE:
4445 if (!status_update(view))
4446 return REQ_NONE;
4447 break;
4449 case REQ_STATUS_REVERT:
4450 if (!status_revert(status, line->type, status_has_none(view, line)))
4451 return REQ_NONE;
4452 break;
4454 case REQ_STATUS_MERGE:
4455 if (!status || status->status != 'U') {
4456 report("Merging only possible for files with unmerged status ('U').");
4457 return REQ_NONE;
4458 }
4459 open_mergetool(status->new.name);
4460 break;
4462 case REQ_EDIT:
4463 if (!status)
4464 return request;
4465 if (status->status == 'D') {
4466 report("File has been deleted.");
4467 return REQ_NONE;
4468 }
4470 open_editor(status->status != '?', status->new.name);
4471 break;
4473 case REQ_VIEW_BLAME:
4474 if (status) {
4475 string_copy(opt_file, status->new.name);
4476 opt_ref[0] = 0;
4477 }
4478 return request;
4480 case REQ_ENTER:
4481 /* After returning the status view has been split to
4482 * show the stage view. No further reloading is
4483 * necessary. */
4484 status_enter(view, line);
4485 return REQ_NONE;
4487 case REQ_REFRESH:
4488 /* Simply reload the view. */
4489 break;
4491 default:
4492 return request;
4493 }
4495 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
4497 return REQ_NONE;
4498 }
4500 static void
4501 status_select(struct view *view, struct line *line)
4502 {
4503 struct status *status = line->data;
4504 char file[SIZEOF_STR] = "all files";
4505 const char *text;
4506 const char *key;
4508 if (status && !string_format(file, "'%s'", status->new.name))
4509 return;
4511 if (!status && line[1].type == LINE_STAT_NONE)
4512 line++;
4514 switch (line->type) {
4515 case LINE_STAT_STAGED:
4516 text = "Press %s to unstage %s for commit";
4517 break;
4519 case LINE_STAT_UNSTAGED:
4520 text = "Press %s to stage %s for commit";
4521 break;
4523 case LINE_STAT_UNTRACKED:
4524 text = "Press %s to stage %s for addition";
4525 break;
4527 case LINE_STAT_HEAD:
4528 case LINE_STAT_NONE:
4529 text = "Nothing to update";
4530 break;
4532 default:
4533 die("line type %d not handled in switch", line->type);
4534 }
4536 if (status && status->status == 'U') {
4537 text = "Press %s to resolve conflict in %s";
4538 key = get_key(REQ_STATUS_MERGE);
4540 } else {
4541 key = get_key(REQ_STATUS_UPDATE);
4542 }
4544 string_format(view->ref, text, key, file);
4545 }
4547 static bool
4548 status_grep(struct view *view, struct line *line)
4549 {
4550 struct status *status = line->data;
4551 enum { S_STATUS, S_NAME, S_END } state;
4552 char buf[2] = "?";
4553 regmatch_t pmatch;
4555 if (!status)
4556 return FALSE;
4558 for (state = S_STATUS; state < S_END; state++) {
4559 const char *text;
4561 switch (state) {
4562 case S_NAME: text = status->new.name; break;
4563 case S_STATUS:
4564 buf[0] = status->status;
4565 text = buf;
4566 break;
4568 default:
4569 return FALSE;
4570 }
4572 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4573 return TRUE;
4574 }
4576 return FALSE;
4577 }
4579 static struct view_ops status_ops = {
4580 "file",
4581 status_open,
4582 NULL,
4583 status_draw,
4584 status_request,
4585 status_grep,
4586 status_select,
4587 };
4590 static bool
4591 stage_diff_line(FILE *pipe, struct line *line)
4592 {
4593 const char *buf = line->data;
4594 size_t bufsize = strlen(buf);
4595 size_t written = 0;
4597 while (!ferror(pipe) && written < bufsize) {
4598 written += fwrite(buf + written, 1, bufsize - written, pipe);
4599 }
4601 fputc('\n', pipe);
4603 return written == bufsize;
4604 }
4606 static bool
4607 stage_diff_write(FILE *pipe, struct line *line, struct line *end)
4608 {
4609 while (line < end) {
4610 if (!stage_diff_line(pipe, line++))
4611 return FALSE;
4612 if (line->type == LINE_DIFF_CHUNK ||
4613 line->type == LINE_DIFF_HEADER)
4614 break;
4615 }
4617 return TRUE;
4618 }
4620 static struct line *
4621 stage_diff_find(struct view *view, struct line *line, enum line_type type)
4622 {
4623 for (; view->line < line; line--)
4624 if (line->type == type)
4625 return line;
4627 return NULL;
4628 }
4630 static bool
4631 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
4632 {
4633 char cmd[SIZEOF_STR];
4634 size_t cmdsize = 0;
4635 struct line *diff_hdr;
4636 FILE *pipe;
4638 diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
4639 if (!diff_hdr)
4640 return FALSE;
4642 if (opt_cdup[0] &&
4643 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4644 return FALSE;
4646 if (!string_format_from(cmd, &cmdsize,
4647 "git apply --whitespace=nowarn %s %s - && "
4648 "git update-index -q --unmerged --refresh 2>/dev/null",
4649 revert ? "" : "--cached",
4650 revert || stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
4651 return FALSE;
4653 pipe = popen(cmd, "w");
4654 if (!pipe)
4655 return FALSE;
4657 if (!stage_diff_write(pipe, diff_hdr, chunk) ||
4658 !stage_diff_write(pipe, chunk, view->line + view->lines))
4659 chunk = NULL;
4661 pclose(pipe);
4663 return chunk ? TRUE : FALSE;
4664 }
4666 static bool
4667 stage_update(struct view *view, struct line *line)
4668 {
4669 struct line *chunk = NULL;
4671 if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
4672 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4674 if (chunk) {
4675 if (!stage_apply_chunk(view, chunk, FALSE)) {
4676 report("Failed to apply chunk");
4677 return FALSE;
4678 }
4680 } else if (!stage_status.status) {
4681 view = VIEW(REQ_VIEW_STATUS);
4683 for (line = view->line; line < view->line + view->lines; line++)
4684 if (line->type == stage_line_type)
4685 break;
4687 if (!status_update_files(view, line + 1)) {
4688 report("Failed to update files");
4689 return FALSE;
4690 }
4692 } else if (!status_update_file(&stage_status, stage_line_type)) {
4693 report("Failed to update file");
4694 return FALSE;
4695 }
4697 return TRUE;
4698 }
4700 static bool
4701 stage_revert(struct view *view, struct line *line)
4702 {
4703 struct line *chunk = NULL;
4705 if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
4706 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4708 if (chunk) {
4709 if (!prompt_yesno("Are you sure you want to revert changes?"))
4710 return FALSE;
4712 if (!stage_apply_chunk(view, chunk, TRUE)) {
4713 report("Failed to revert chunk");
4714 return FALSE;
4715 }
4716 return TRUE;
4718 } else {
4719 return status_revert(stage_status.status ? &stage_status : NULL,
4720 stage_line_type, FALSE);
4721 }
4722 }
4725 static void
4726 stage_next(struct view *view, struct line *line)
4727 {
4728 int i;
4730 if (!stage_chunks) {
4731 static size_t alloc = 0;
4732 int *tmp;
4734 for (line = view->line; line < view->line + view->lines; line++) {
4735 if (line->type != LINE_DIFF_CHUNK)
4736 continue;
4738 tmp = realloc_items(stage_chunk, &alloc,
4739 stage_chunks, sizeof(*tmp));
4740 if (!tmp) {
4741 report("Allocation failure");
4742 return;
4743 }
4745 stage_chunk = tmp;
4746 stage_chunk[stage_chunks++] = line - view->line;
4747 }
4748 }
4750 for (i = 0; i < stage_chunks; i++) {
4751 if (stage_chunk[i] > view->lineno) {
4752 do_scroll_view(view, stage_chunk[i] - view->lineno);
4753 report("Chunk %d of %d", i + 1, stage_chunks);
4754 return;
4755 }
4756 }
4758 report("No next chunk found");
4759 }
4761 static enum request
4762 stage_request(struct view *view, enum request request, struct line *line)
4763 {
4764 switch (request) {
4765 case REQ_STATUS_UPDATE:
4766 if (!stage_update(view, line))
4767 return REQ_NONE;
4768 break;
4770 case REQ_STATUS_REVERT:
4771 if (!stage_revert(view, line))
4772 return REQ_NONE;
4773 break;
4775 case REQ_STAGE_NEXT:
4776 if (stage_line_type == LINE_STAT_UNTRACKED) {
4777 report("File is untracked; press %s to add",
4778 get_key(REQ_STATUS_UPDATE));
4779 return REQ_NONE;
4780 }
4781 stage_next(view, line);
4782 return REQ_NONE;
4784 case REQ_EDIT:
4785 if (!stage_status.new.name[0])
4786 return request;
4787 if (stage_status.status == 'D') {
4788 report("File has been deleted.");
4789 return REQ_NONE;
4790 }
4792 open_editor(stage_status.status != '?', stage_status.new.name);
4793 break;
4795 case REQ_REFRESH:
4796 /* Reload everything ... */
4797 break;
4799 case REQ_VIEW_BLAME:
4800 if (stage_status.new.name[0]) {
4801 string_copy(opt_file, stage_status.new.name);
4802 opt_ref[0] = 0;
4803 }
4804 return request;
4806 case REQ_ENTER:
4807 return pager_request(view, request, line);
4809 default:
4810 return request;
4811 }
4813 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD | OPEN_NOMAXIMIZE);
4815 /* Check whether the staged entry still exists, and close the
4816 * stage view if it doesn't. */
4817 if (!status_exists(&stage_status, stage_line_type))
4818 return REQ_VIEW_CLOSE;
4820 if (stage_line_type == LINE_STAT_UNTRACKED) {
4821 if (!suffixcmp(stage_status.new.name, -1, "/")) {
4822 report("Cannot display a directory");
4823 return REQ_NONE;
4824 }
4826 opt_pipe = fopen(stage_status.new.name, "r");
4827 }
4828 open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
4830 return REQ_NONE;
4831 }
4833 static struct view_ops stage_ops = {
4834 "line",
4835 NULL,
4836 pager_read,
4837 pager_draw,
4838 stage_request,
4839 pager_grep,
4840 pager_select,
4841 };
4844 /*
4845 * Revision graph
4846 */
4848 struct commit {
4849 char id[SIZEOF_REV]; /* SHA1 ID. */
4850 char title[128]; /* First line of the commit message. */
4851 char author[75]; /* Author of the commit. */
4852 struct tm time; /* Date from the author ident. */
4853 struct ref **refs; /* Repository references. */
4854 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
4855 size_t graph_size; /* The width of the graph array. */
4856 bool has_parents; /* Rewritten --parents seen. */
4857 };
4859 /* Size of rev graph with no "padding" columns */
4860 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
4862 struct rev_graph {
4863 struct rev_graph *prev, *next, *parents;
4864 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
4865 size_t size;
4866 struct commit *commit;
4867 size_t pos;
4868 unsigned int boundary:1;
4869 };
4871 /* Parents of the commit being visualized. */
4872 static struct rev_graph graph_parents[4];
4874 /* The current stack of revisions on the graph. */
4875 static struct rev_graph graph_stacks[4] = {
4876 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
4877 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
4878 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
4879 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
4880 };
4882 static inline bool
4883 graph_parent_is_merge(struct rev_graph *graph)
4884 {
4885 return graph->parents->size > 1;
4886 }
4888 static inline void
4889 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
4890 {
4891 struct commit *commit = graph->commit;
4893 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
4894 commit->graph[commit->graph_size++] = symbol;
4895 }
4897 static void
4898 clear_rev_graph(struct rev_graph *graph)
4899 {
4900 graph->boundary = 0;
4901 graph->size = graph->pos = 0;
4902 graph->commit = NULL;
4903 memset(graph->parents, 0, sizeof(*graph->parents));
4904 }
4906 static void
4907 done_rev_graph(struct rev_graph *graph)
4908 {
4909 if (graph_parent_is_merge(graph) &&
4910 graph->pos < graph->size - 1 &&
4911 graph->next->size == graph->size + graph->parents->size - 1) {
4912 size_t i = graph->pos + graph->parents->size - 1;
4914 graph->commit->graph_size = i * 2;
4915 while (i < graph->next->size - 1) {
4916 append_to_rev_graph(graph, ' ');
4917 append_to_rev_graph(graph, '\\');
4918 i++;
4919 }
4920 }
4922 clear_rev_graph(graph);
4923 }
4925 static void
4926 push_rev_graph(struct rev_graph *graph, const char *parent)
4927 {
4928 int i;
4930 /* "Collapse" duplicate parents lines.
4931 *
4932 * FIXME: This needs to also update update the drawn graph but
4933 * for now it just serves as a method for pruning graph lines. */
4934 for (i = 0; i < graph->size; i++)
4935 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
4936 return;
4938 if (graph->size < SIZEOF_REVITEMS) {
4939 string_copy_rev(graph->rev[graph->size++], parent);
4940 }
4941 }
4943 static chtype
4944 get_rev_graph_symbol(struct rev_graph *graph)
4945 {
4946 chtype symbol;
4948 if (graph->boundary)
4949 symbol = REVGRAPH_BOUND;
4950 else if (graph->parents->size == 0)
4951 symbol = REVGRAPH_INIT;
4952 else if (graph_parent_is_merge(graph))
4953 symbol = REVGRAPH_MERGE;
4954 else if (graph->pos >= graph->size)
4955 symbol = REVGRAPH_BRANCH;
4956 else
4957 symbol = REVGRAPH_COMMIT;
4959 return symbol;
4960 }
4962 static void
4963 draw_rev_graph(struct rev_graph *graph)
4964 {
4965 struct rev_filler {
4966 chtype separator, line;
4967 };
4968 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
4969 static struct rev_filler fillers[] = {
4970 { ' ', '|' },
4971 { '`', '.' },
4972 { '\'', ' ' },
4973 { '/', ' ' },
4974 };
4975 chtype symbol = get_rev_graph_symbol(graph);
4976 struct rev_filler *filler;
4977 size_t i;
4979 if (opt_line_graphics)
4980 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
4982 filler = &fillers[DEFAULT];
4984 for (i = 0; i < graph->pos; i++) {
4985 append_to_rev_graph(graph, filler->line);
4986 if (graph_parent_is_merge(graph->prev) &&
4987 graph->prev->pos == i)
4988 filler = &fillers[RSHARP];
4990 append_to_rev_graph(graph, filler->separator);
4991 }
4993 /* Place the symbol for this revision. */
4994 append_to_rev_graph(graph, symbol);
4996 if (graph->prev->size > graph->size)
4997 filler = &fillers[RDIAG];
4998 else
4999 filler = &fillers[DEFAULT];
5001 i++;
5003 for (; i < graph->size; i++) {
5004 append_to_rev_graph(graph, filler->separator);
5005 append_to_rev_graph(graph, filler->line);
5006 if (graph_parent_is_merge(graph->prev) &&
5007 i < graph->prev->pos + graph->parents->size)
5008 filler = &fillers[RSHARP];
5009 if (graph->prev->size > graph->size)
5010 filler = &fillers[LDIAG];
5011 }
5013 if (graph->prev->size > graph->size) {
5014 append_to_rev_graph(graph, filler->separator);
5015 if (filler->line != ' ')
5016 append_to_rev_graph(graph, filler->line);
5017 }
5018 }
5020 /* Prepare the next rev graph */
5021 static void
5022 prepare_rev_graph(struct rev_graph *graph)
5023 {
5024 size_t i;
5026 /* First, traverse all lines of revisions up to the active one. */
5027 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5028 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5029 break;
5031 push_rev_graph(graph->next, graph->rev[graph->pos]);
5032 }
5034 /* Interleave the new revision parent(s). */
5035 for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5036 push_rev_graph(graph->next, graph->parents->rev[i]);
5038 /* Lastly, put any remaining revisions. */
5039 for (i = graph->pos + 1; i < graph->size; i++)
5040 push_rev_graph(graph->next, graph->rev[i]);
5041 }
5043 static void
5044 update_rev_graph(struct rev_graph *graph)
5045 {
5046 /* If this is the finalizing update ... */
5047 if (graph->commit)
5048 prepare_rev_graph(graph);
5050 /* Graph visualization needs a one rev look-ahead,
5051 * so the first update doesn't visualize anything. */
5052 if (!graph->prev->commit)
5053 return;
5055 draw_rev_graph(graph->prev);
5056 done_rev_graph(graph->prev->prev);
5057 }
5060 /*
5061 * Main view backend
5062 */
5064 static bool
5065 main_draw(struct view *view, struct line *line, unsigned int lineno)
5066 {
5067 struct commit *commit = line->data;
5069 if (!*commit->author)
5070 return FALSE;
5072 if (opt_date && draw_date(view, &commit->time))
5073 return TRUE;
5075 if (opt_author &&
5076 draw_field(view, LINE_MAIN_AUTHOR, commit->author, opt_author_cols, TRUE))
5077 return TRUE;
5079 if (opt_rev_graph && commit->graph_size &&
5080 draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5081 return TRUE;
5083 if (opt_show_refs && commit->refs) {
5084 size_t i = 0;
5086 do {
5087 enum line_type type;
5089 if (commit->refs[i]->head)
5090 type = LINE_MAIN_HEAD;
5091 else if (commit->refs[i]->ltag)
5092 type = LINE_MAIN_LOCAL_TAG;
5093 else if (commit->refs[i]->tag)
5094 type = LINE_MAIN_TAG;
5095 else if (commit->refs[i]->tracked)
5096 type = LINE_MAIN_TRACKED;
5097 else if (commit->refs[i]->remote)
5098 type = LINE_MAIN_REMOTE;
5099 else
5100 type = LINE_MAIN_REF;
5102 if (draw_text(view, type, "[", TRUE) ||
5103 draw_text(view, type, commit->refs[i]->name, TRUE) ||
5104 draw_text(view, type, "]", TRUE))
5105 return TRUE;
5107 if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5108 return TRUE;
5109 } while (commit->refs[i++]->next);
5110 }
5112 draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5113 return TRUE;
5114 }
5116 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5117 static bool
5118 main_read(struct view *view, char *line)
5119 {
5120 static struct rev_graph *graph = graph_stacks;
5121 enum line_type type;
5122 struct commit *commit;
5124 if (!line) {
5125 int i;
5127 if (!view->lines && !view->parent)
5128 die("No revisions match the given arguments.");
5129 if (view->lines > 0) {
5130 commit = view->line[view->lines - 1].data;
5131 if (!*commit->author) {
5132 view->lines--;
5133 free(commit);
5134 graph->commit = NULL;
5135 }
5136 }
5137 update_rev_graph(graph);
5139 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5140 clear_rev_graph(&graph_stacks[i]);
5141 return TRUE;
5142 }
5144 type = get_line_type(line);
5145 if (type == LINE_COMMIT) {
5146 commit = calloc(1, sizeof(struct commit));
5147 if (!commit)
5148 return FALSE;
5150 line += STRING_SIZE("commit ");
5151 if (*line == '-') {
5152 graph->boundary = 1;
5153 line++;
5154 }
5156 string_copy_rev(commit->id, line);
5157 commit->refs = get_refs(commit->id);
5158 graph->commit = commit;
5159 add_line_data(view, commit, LINE_MAIN_COMMIT);
5161 while ((line = strchr(line, ' '))) {
5162 line++;
5163 push_rev_graph(graph->parents, line);
5164 commit->has_parents = TRUE;
5165 }
5166 return TRUE;
5167 }
5169 if (!view->lines)
5170 return TRUE;
5171 commit = view->line[view->lines - 1].data;
5173 switch (type) {
5174 case LINE_PARENT:
5175 if (commit->has_parents)
5176 break;
5177 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5178 break;
5180 case LINE_AUTHOR:
5181 {
5182 /* Parse author lines where the name may be empty:
5183 * author <email@address.tld> 1138474660 +0100
5184 */
5185 char *ident = line + STRING_SIZE("author ");
5186 char *nameend = strchr(ident, '<');
5187 char *emailend = strchr(ident, '>');
5189 if (!nameend || !emailend)
5190 break;
5192 update_rev_graph(graph);
5193 graph = graph->next;
5195 *nameend = *emailend = 0;
5196 ident = chomp_string(ident);
5197 if (!*ident) {
5198 ident = chomp_string(nameend + 1);
5199 if (!*ident)
5200 ident = "Unknown";
5201 }
5203 string_ncopy(commit->author, ident, strlen(ident));
5205 /* Parse epoch and timezone */
5206 if (emailend[1] == ' ') {
5207 char *secs = emailend + 2;
5208 char *zone = strchr(secs, ' ');
5209 time_t time = (time_t) atol(secs);
5211 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
5212 long tz;
5214 zone++;
5215 tz = ('0' - zone[1]) * 60 * 60 * 10;
5216 tz += ('0' - zone[2]) * 60 * 60;
5217 tz += ('0' - zone[3]) * 60;
5218 tz += ('0' - zone[4]) * 60;
5220 if (zone[0] == '-')
5221 tz = -tz;
5223 time -= tz;
5224 }
5226 gmtime_r(&time, &commit->time);
5227 }
5228 break;
5229 }
5230 default:
5231 /* Fill in the commit title if it has not already been set. */
5232 if (commit->title[0])
5233 break;
5235 /* Require titles to start with a non-space character at the
5236 * offset used by git log. */
5237 if (strncmp(line, " ", 4))
5238 break;
5239 line += 4;
5240 /* Well, if the title starts with a whitespace character,
5241 * try to be forgiving. Otherwise we end up with no title. */
5242 while (isspace(*line))
5243 line++;
5244 if (*line == '\0')
5245 break;
5246 /* FIXME: More graceful handling of titles; append "..." to
5247 * shortened titles, etc. */
5249 string_ncopy(commit->title, line, strlen(line));
5250 }
5252 return TRUE;
5253 }
5255 static enum request
5256 main_request(struct view *view, enum request request, struct line *line)
5257 {
5258 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5260 switch (request) {
5261 case REQ_ENTER:
5262 open_view(view, REQ_VIEW_DIFF, flags);
5263 break;
5264 case REQ_REFRESH:
5265 load_refs();
5266 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
5267 break;
5268 default:
5269 return request;
5270 }
5272 return REQ_NONE;
5273 }
5275 static bool
5276 grep_refs(struct ref **refs, regex_t *regex)
5277 {
5278 regmatch_t pmatch;
5279 size_t i = 0;
5281 if (!refs)
5282 return FALSE;
5283 do {
5284 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
5285 return TRUE;
5286 } while (refs[i++]->next);
5288 return FALSE;
5289 }
5291 static bool
5292 main_grep(struct view *view, struct line *line)
5293 {
5294 struct commit *commit = line->data;
5295 enum { S_TITLE, S_AUTHOR, S_DATE, S_REFS, S_END } state;
5296 char buf[DATE_COLS + 1];
5297 regmatch_t pmatch;
5299 for (state = S_TITLE; state < S_END; state++) {
5300 char *text;
5302 switch (state) {
5303 case S_TITLE: text = commit->title; break;
5304 case S_AUTHOR:
5305 if (!opt_author)
5306 continue;
5307 text = commit->author;
5308 break;
5309 case S_DATE:
5310 if (!opt_date)
5311 continue;
5312 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
5313 continue;
5314 text = buf;
5315 break;
5316 case S_REFS:
5317 if (!opt_show_refs)
5318 continue;
5319 if (grep_refs(commit->refs, view->regex) == TRUE)
5320 return TRUE;
5321 continue;
5322 default:
5323 return FALSE;
5324 }
5326 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5327 return TRUE;
5328 }
5330 return FALSE;
5331 }
5333 static void
5334 main_select(struct view *view, struct line *line)
5335 {
5336 struct commit *commit = line->data;
5338 string_copy_rev(view->ref, commit->id);
5339 string_copy_rev(ref_commit, view->ref);
5340 }
5342 static struct view_ops main_ops = {
5343 "commit",
5344 NULL,
5345 main_read,
5346 main_draw,
5347 main_request,
5348 main_grep,
5349 main_select,
5350 };
5353 /*
5354 * Unicode / UTF-8 handling
5355 *
5356 * NOTE: Much of the following code for dealing with unicode is derived from
5357 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
5358 * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
5359 */
5361 /* I've (over)annotated a lot of code snippets because I am not entirely
5362 * confident that the approach taken by this small UTF-8 interface is correct.
5363 * --jonas */
5365 static inline int
5366 unicode_width(unsigned long c)
5367 {
5368 if (c >= 0x1100 &&
5369 (c <= 0x115f /* Hangul Jamo */
5370 || c == 0x2329
5371 || c == 0x232a
5372 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
5373 /* CJK ... Yi */
5374 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
5375 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
5376 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
5377 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
5378 || (c >= 0xffe0 && c <= 0xffe6)
5379 || (c >= 0x20000 && c <= 0x2fffd)
5380 || (c >= 0x30000 && c <= 0x3fffd)))
5381 return 2;
5383 if (c == '\t')
5384 return opt_tab_size;
5386 return 1;
5387 }
5389 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
5390 * Illegal bytes are set one. */
5391 static const unsigned char utf8_bytes[256] = {
5392 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,
5393 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,
5394 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,
5395 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,
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 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,
5399 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,
5400 };
5402 /* Decode UTF-8 multi-byte representation into a unicode character. */
5403 static inline unsigned long
5404 utf8_to_unicode(const char *string, size_t length)
5405 {
5406 unsigned long unicode;
5408 switch (length) {
5409 case 1:
5410 unicode = string[0];
5411 break;
5412 case 2:
5413 unicode = (string[0] & 0x1f) << 6;
5414 unicode += (string[1] & 0x3f);
5415 break;
5416 case 3:
5417 unicode = (string[0] & 0x0f) << 12;
5418 unicode += ((string[1] & 0x3f) << 6);
5419 unicode += (string[2] & 0x3f);
5420 break;
5421 case 4:
5422 unicode = (string[0] & 0x0f) << 18;
5423 unicode += ((string[1] & 0x3f) << 12);
5424 unicode += ((string[2] & 0x3f) << 6);
5425 unicode += (string[3] & 0x3f);
5426 break;
5427 case 5:
5428 unicode = (string[0] & 0x0f) << 24;
5429 unicode += ((string[1] & 0x3f) << 18);
5430 unicode += ((string[2] & 0x3f) << 12);
5431 unicode += ((string[3] & 0x3f) << 6);
5432 unicode += (string[4] & 0x3f);
5433 break;
5434 case 6:
5435 unicode = (string[0] & 0x01) << 30;
5436 unicode += ((string[1] & 0x3f) << 24);
5437 unicode += ((string[2] & 0x3f) << 18);
5438 unicode += ((string[3] & 0x3f) << 12);
5439 unicode += ((string[4] & 0x3f) << 6);
5440 unicode += (string[5] & 0x3f);
5441 break;
5442 default:
5443 die("Invalid unicode length");
5444 }
5446 /* Invalid characters could return the special 0xfffd value but NUL
5447 * should be just as good. */
5448 return unicode > 0xffff ? 0 : unicode;
5449 }
5451 /* Calculates how much of string can be shown within the given maximum width
5452 * and sets trimmed parameter to non-zero value if all of string could not be
5453 * shown. If the reserve flag is TRUE, it will reserve at least one
5454 * trailing character, which can be useful when drawing a delimiter.
5455 *
5456 * Returns the number of bytes to output from string to satisfy max_width. */
5457 static size_t
5458 utf8_length(const char *string, int *width, size_t max_width, int *trimmed, bool reserve)
5459 {
5460 const char *start = string;
5461 const char *end = strchr(string, '\0');
5462 unsigned char last_bytes = 0;
5463 size_t last_ucwidth = 0;
5465 *width = 0;
5466 *trimmed = 0;
5468 while (string < end) {
5469 int c = *(unsigned char *) string;
5470 unsigned char bytes = utf8_bytes[c];
5471 size_t ucwidth;
5472 unsigned long unicode;
5474 if (string + bytes > end)
5475 break;
5477 /* Change representation to figure out whether
5478 * it is a single- or double-width character. */
5480 unicode = utf8_to_unicode(string, bytes);
5481 /* FIXME: Graceful handling of invalid unicode character. */
5482 if (!unicode)
5483 break;
5485 ucwidth = unicode_width(unicode);
5486 *width += ucwidth;
5487 if (*width > max_width) {
5488 *trimmed = 1;
5489 *width -= ucwidth;
5490 if (reserve && *width == max_width) {
5491 string -= last_bytes;
5492 *width -= last_ucwidth;
5493 }
5494 break;
5495 }
5497 string += bytes;
5498 last_bytes = bytes;
5499 last_ucwidth = ucwidth;
5500 }
5502 return string - start;
5503 }
5506 /*
5507 * Status management
5508 */
5510 /* Whether or not the curses interface has been initialized. */
5511 static bool cursed = FALSE;
5513 /* The status window is used for polling keystrokes. */
5514 static WINDOW *status_win;
5516 static bool status_empty = TRUE;
5518 /* Update status and title window. */
5519 static void
5520 report(const char *msg, ...)
5521 {
5522 struct view *view = display[current_view];
5524 if (input_mode)
5525 return;
5527 if (!view) {
5528 char buf[SIZEOF_STR];
5529 va_list args;
5531 va_start(args, msg);
5532 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
5533 buf[sizeof(buf) - 1] = 0;
5534 buf[sizeof(buf) - 2] = '.';
5535 buf[sizeof(buf) - 3] = '.';
5536 buf[sizeof(buf) - 4] = '.';
5537 }
5538 va_end(args);
5539 die("%s", buf);
5540 }
5542 if (!status_empty || *msg) {
5543 va_list args;
5545 va_start(args, msg);
5547 wmove(status_win, 0, 0);
5548 if (*msg) {
5549 vwprintw(status_win, msg, args);
5550 status_empty = FALSE;
5551 } else {
5552 status_empty = TRUE;
5553 }
5554 wclrtoeol(status_win);
5555 wrefresh(status_win);
5557 va_end(args);
5558 }
5560 update_view_title(view);
5561 update_display_cursor(view);
5562 }
5564 /* Controls when nodelay should be in effect when polling user input. */
5565 static void
5566 set_nonblocking_input(bool loading)
5567 {
5568 static unsigned int loading_views;
5570 if ((loading == FALSE && loading_views-- == 1) ||
5571 (loading == TRUE && loading_views++ == 0))
5572 nodelay(status_win, loading);
5573 }
5575 static void
5576 init_display(void)
5577 {
5578 int x, y;
5580 /* Initialize the curses library */
5581 if (isatty(STDIN_FILENO)) {
5582 cursed = !!initscr();
5583 opt_tty = stdin;
5584 } else {
5585 /* Leave stdin and stdout alone when acting as a pager. */
5586 opt_tty = fopen("/dev/tty", "r+");
5587 if (!opt_tty)
5588 die("Failed to open /dev/tty");
5589 cursed = !!newterm(NULL, opt_tty, opt_tty);
5590 }
5592 if (!cursed)
5593 die("Failed to initialize curses");
5595 nonl(); /* Tell curses not to do NL->CR/NL on output */
5596 cbreak(); /* Take input chars one at a time, no wait for \n */
5597 noecho(); /* Don't echo input */
5598 leaveok(stdscr, TRUE);
5600 if (has_colors())
5601 init_colors();
5603 getmaxyx(stdscr, y, x);
5604 status_win = newwin(1, 0, y - 1, 0);
5605 if (!status_win)
5606 die("Failed to create status window");
5608 /* Enable keyboard mapping */
5609 keypad(status_win, TRUE);
5610 wbkgdset(status_win, get_line_attr(LINE_STATUS));
5612 TABSIZE = opt_tab_size;
5613 if (opt_line_graphics) {
5614 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
5615 }
5616 }
5618 static bool
5619 prompt_yesno(const char *prompt)
5620 {
5621 enum { WAIT, STOP, CANCEL } status = WAIT;
5622 bool answer = FALSE;
5624 while (status == WAIT) {
5625 struct view *view;
5626 int i, key;
5628 input_mode = TRUE;
5630 foreach_view (view, i)
5631 update_view(view);
5633 input_mode = FALSE;
5635 mvwprintw(status_win, 0, 0, "%s [Yy]/[Nn]", prompt);
5636 wclrtoeol(status_win);
5638 /* Refresh, accept single keystroke of input */
5639 key = wgetch(status_win);
5640 switch (key) {
5641 case ERR:
5642 break;
5644 case 'y':
5645 case 'Y':
5646 answer = TRUE;
5647 status = STOP;
5648 break;
5650 case KEY_ESC:
5651 case KEY_RETURN:
5652 case KEY_ENTER:
5653 case KEY_BACKSPACE:
5654 case 'n':
5655 case 'N':
5656 case '\n':
5657 default:
5658 answer = FALSE;
5659 status = CANCEL;
5660 }
5661 }
5663 /* Clear the status window */
5664 status_empty = FALSE;
5665 report("");
5667 return answer;
5668 }
5670 static char *
5671 read_prompt(const char *prompt)
5672 {
5673 enum { READING, STOP, CANCEL } status = READING;
5674 static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
5675 int pos = 0;
5677 while (status == READING) {
5678 struct view *view;
5679 int i, key;
5681 input_mode = TRUE;
5683 foreach_view (view, i)
5684 update_view(view);
5686 input_mode = FALSE;
5688 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
5689 wclrtoeol(status_win);
5691 /* Refresh, accept single keystroke of input */
5692 key = wgetch(status_win);
5693 switch (key) {
5694 case KEY_RETURN:
5695 case KEY_ENTER:
5696 case '\n':
5697 status = pos ? STOP : CANCEL;
5698 break;
5700 case KEY_BACKSPACE:
5701 if (pos > 0)
5702 pos--;
5703 else
5704 status = CANCEL;
5705 break;
5707 case KEY_ESC:
5708 status = CANCEL;
5709 break;
5711 case ERR:
5712 break;
5714 default:
5715 if (pos >= sizeof(buf)) {
5716 report("Input string too long");
5717 return NULL;
5718 }
5720 if (isprint(key))
5721 buf[pos++] = (char) key;
5722 }
5723 }
5725 /* Clear the status window */
5726 status_empty = FALSE;
5727 report("");
5729 if (status == CANCEL)
5730 return NULL;
5732 buf[pos++] = 0;
5734 return buf;
5735 }
5737 /*
5738 * Repository references
5739 */
5741 static struct ref *refs = NULL;
5742 static size_t refs_alloc = 0;
5743 static size_t refs_size = 0;
5745 /* Id <-> ref store */
5746 static struct ref ***id_refs = NULL;
5747 static size_t id_refs_alloc = 0;
5748 static size_t id_refs_size = 0;
5750 static int
5751 compare_refs(const void *ref1_, const void *ref2_)
5752 {
5753 const struct ref *ref1 = *(const struct ref **)ref1_;
5754 const struct ref *ref2 = *(const struct ref **)ref2_;
5756 if (ref1->tag != ref2->tag)
5757 return ref2->tag - ref1->tag;
5758 if (ref1->ltag != ref2->ltag)
5759 return ref2->ltag - ref2->ltag;
5760 if (ref1->head != ref2->head)
5761 return ref2->head - ref1->head;
5762 if (ref1->tracked != ref2->tracked)
5763 return ref2->tracked - ref1->tracked;
5764 if (ref1->remote != ref2->remote)
5765 return ref2->remote - ref1->remote;
5766 return strcmp(ref1->name, ref2->name);
5767 }
5769 static struct ref **
5770 get_refs(const char *id)
5771 {
5772 struct ref ***tmp_id_refs;
5773 struct ref **ref_list = NULL;
5774 size_t ref_list_alloc = 0;
5775 size_t ref_list_size = 0;
5776 size_t i;
5778 for (i = 0; i < id_refs_size; i++)
5779 if (!strcmp(id, id_refs[i][0]->id))
5780 return id_refs[i];
5782 tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
5783 sizeof(*id_refs));
5784 if (!tmp_id_refs)
5785 return NULL;
5787 id_refs = tmp_id_refs;
5789 for (i = 0; i < refs_size; i++) {
5790 struct ref **tmp;
5792 if (strcmp(id, refs[i].id))
5793 continue;
5795 tmp = realloc_items(ref_list, &ref_list_alloc,
5796 ref_list_size + 1, sizeof(*ref_list));
5797 if (!tmp) {
5798 if (ref_list)
5799 free(ref_list);
5800 return NULL;
5801 }
5803 ref_list = tmp;
5804 ref_list[ref_list_size] = &refs[i];
5805 /* XXX: The properties of the commit chains ensures that we can
5806 * safely modify the shared ref. The repo references will
5807 * always be similar for the same id. */
5808 ref_list[ref_list_size]->next = 1;
5810 ref_list_size++;
5811 }
5813 if (ref_list) {
5814 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
5815 ref_list[ref_list_size - 1]->next = 0;
5816 id_refs[id_refs_size++] = ref_list;
5817 }
5819 return ref_list;
5820 }
5822 static int
5823 read_ref(char *id, size_t idlen, char *name, size_t namelen)
5824 {
5825 struct ref *ref;
5826 bool tag = FALSE;
5827 bool ltag = FALSE;
5828 bool remote = FALSE;
5829 bool tracked = FALSE;
5830 bool check_replace = FALSE;
5831 bool head = FALSE;
5833 if (!prefixcmp(name, "refs/tags/")) {
5834 if (!suffixcmp(name, namelen, "^{}")) {
5835 namelen -= 3;
5836 name[namelen] = 0;
5837 if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
5838 check_replace = TRUE;
5839 } else {
5840 ltag = TRUE;
5841 }
5843 tag = TRUE;
5844 namelen -= STRING_SIZE("refs/tags/");
5845 name += STRING_SIZE("refs/tags/");
5847 } else if (!prefixcmp(name, "refs/remotes/")) {
5848 remote = TRUE;
5849 namelen -= STRING_SIZE("refs/remotes/");
5850 name += STRING_SIZE("refs/remotes/");
5851 tracked = !strcmp(opt_remote, name);
5853 } else if (!prefixcmp(name, "refs/heads/")) {
5854 namelen -= STRING_SIZE("refs/heads/");
5855 name += STRING_SIZE("refs/heads/");
5856 head = !strncmp(opt_head, name, namelen);
5858 } else if (!strcmp(name, "HEAD")) {
5859 string_ncopy(opt_head_rev, id, idlen);
5860 return OK;
5861 }
5863 if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
5864 /* it's an annotated tag, replace the previous sha1 with the
5865 * resolved commit id; relies on the fact git-ls-remote lists
5866 * the commit id of an annotated tag right before the commit id
5867 * it points to. */
5868 refs[refs_size - 1].ltag = ltag;
5869 string_copy_rev(refs[refs_size - 1].id, id);
5871 return OK;
5872 }
5873 refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
5874 if (!refs)
5875 return ERR;
5877 ref = &refs[refs_size++];
5878 ref->name = malloc(namelen + 1);
5879 if (!ref->name)
5880 return ERR;
5882 strncpy(ref->name, name, namelen);
5883 ref->name[namelen] = 0;
5884 ref->head = head;
5885 ref->tag = tag;
5886 ref->ltag = ltag;
5887 ref->remote = remote;
5888 ref->tracked = tracked;
5889 string_copy_rev(ref->id, id);
5891 return OK;
5892 }
5894 static int
5895 load_refs(void)
5896 {
5897 const char *cmd_env = getenv("TIG_LS_REMOTE");
5898 const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
5900 if (!*opt_git_dir)
5901 return OK;
5903 while (refs_size > 0)
5904 free(refs[--refs_size].name);
5905 while (id_refs_size > 0)
5906 free(id_refs[--id_refs_size]);
5908 return read_properties(popen(cmd, "r"), "\t", read_ref);
5909 }
5911 static int
5912 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
5913 {
5914 if (!strcmp(name, "i18n.commitencoding"))
5915 string_ncopy(opt_encoding, value, valuelen);
5917 if (!strcmp(name, "core.editor"))
5918 string_ncopy(opt_editor, value, valuelen);
5920 /* branch.<head>.remote */
5921 if (*opt_head &&
5922 !strncmp(name, "branch.", 7) &&
5923 !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5924 !strcmp(name + 7 + strlen(opt_head), ".remote"))
5925 string_ncopy(opt_remote, value, valuelen);
5927 if (*opt_head && *opt_remote &&
5928 !strncmp(name, "branch.", 7) &&
5929 !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5930 !strcmp(name + 7 + strlen(opt_head), ".merge")) {
5931 size_t from = strlen(opt_remote);
5933 if (!prefixcmp(value, "refs/heads/")) {
5934 value += STRING_SIZE("refs/heads/");
5935 valuelen -= STRING_SIZE("refs/heads/");
5936 }
5938 if (!string_format_from(opt_remote, &from, "/%s", value))
5939 opt_remote[0] = 0;
5940 }
5942 return OK;
5943 }
5945 static int
5946 load_git_config(void)
5947 {
5948 return read_properties(popen("git " GIT_CONFIG " --list", "r"),
5949 "=", read_repo_config_option);
5950 }
5952 static int
5953 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
5954 {
5955 if (!opt_git_dir[0]) {
5956 string_ncopy(opt_git_dir, name, namelen);
5958 } else if (opt_is_inside_work_tree == -1) {
5959 /* This can be 3 different values depending on the
5960 * version of git being used. If git-rev-parse does not
5961 * understand --is-inside-work-tree it will simply echo
5962 * the option else either "true" or "false" is printed.
5963 * Default to true for the unknown case. */
5964 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
5966 } else if (opt_cdup[0] == ' ') {
5967 string_ncopy(opt_cdup, name, namelen);
5968 } else {
5969 if (!prefixcmp(name, "refs/heads/")) {
5970 namelen -= STRING_SIZE("refs/heads/");
5971 name += STRING_SIZE("refs/heads/");
5972 string_ncopy(opt_head, name, namelen);
5973 }
5974 }
5976 return OK;
5977 }
5979 static int
5980 load_repo_info(void)
5981 {
5982 int result;
5983 FILE *pipe = popen("(git rev-parse --git-dir --is-inside-work-tree "
5984 " --show-cdup; git symbolic-ref HEAD) 2>/dev/null", "r");
5986 /* XXX: The line outputted by "--show-cdup" can be empty so
5987 * initialize it to something invalid to make it possible to
5988 * detect whether it has been set or not. */
5989 opt_cdup[0] = ' ';
5991 result = read_properties(pipe, "=", read_repo_info);
5992 if (opt_cdup[0] == ' ')
5993 opt_cdup[0] = 0;
5995 return result;
5996 }
5998 static int
5999 read_properties(FILE *pipe, const char *separators,
6000 int (*read_property)(char *, size_t, char *, size_t))
6001 {
6002 char buffer[BUFSIZ];
6003 char *name;
6004 int state = OK;
6006 if (!pipe)
6007 return ERR;
6009 while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
6010 char *value;
6011 size_t namelen;
6012 size_t valuelen;
6014 name = chomp_string(name);
6015 namelen = strcspn(name, separators);
6017 if (name[namelen]) {
6018 name[namelen] = 0;
6019 value = chomp_string(name + namelen + 1);
6020 valuelen = strlen(value);
6022 } else {
6023 value = "";
6024 valuelen = 0;
6025 }
6027 state = read_property(name, namelen, value, valuelen);
6028 }
6030 if (state != ERR && ferror(pipe))
6031 state = ERR;
6033 pclose(pipe);
6035 return state;
6036 }
6039 /*
6040 * Main
6041 */
6043 static void __NORETURN
6044 quit(int sig)
6045 {
6046 /* XXX: Restore tty modes and let the OS cleanup the rest! */
6047 if (cursed)
6048 endwin();
6049 exit(0);
6050 }
6052 static void __NORETURN
6053 die(const char *err, ...)
6054 {
6055 va_list args;
6057 endwin();
6059 va_start(args, err);
6060 fputs("tig: ", stderr);
6061 vfprintf(stderr, err, args);
6062 fputs("\n", stderr);
6063 va_end(args);
6065 exit(1);
6066 }
6068 static void
6069 warn(const char *msg, ...)
6070 {
6071 va_list args;
6073 va_start(args, msg);
6074 fputs("tig warning: ", stderr);
6075 vfprintf(stderr, msg, args);
6076 fputs("\n", stderr);
6077 va_end(args);
6078 }
6080 int
6081 main(int argc, const char *argv[])
6082 {
6083 struct view *view;
6084 enum request request;
6085 size_t i;
6087 signal(SIGINT, quit);
6089 if (setlocale(LC_ALL, "")) {
6090 char *codeset = nl_langinfo(CODESET);
6092 string_ncopy(opt_codeset, codeset, strlen(codeset));
6093 }
6095 if (load_repo_info() == ERR)
6096 die("Failed to load repo info.");
6098 if (load_options() == ERR)
6099 die("Failed to load user config.");
6101 if (load_git_config() == ERR)
6102 die("Failed to load repo config.");
6104 request = parse_options(argc, argv);
6105 if (request == REQ_NONE)
6106 return 0;
6108 /* Require a git repository unless when running in pager mode. */
6109 if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
6110 die("Not a git repository");
6112 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
6113 opt_utf8 = FALSE;
6115 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
6116 opt_iconv = iconv_open(opt_codeset, opt_encoding);
6117 if (opt_iconv == ICONV_NONE)
6118 die("Failed to initialize character set conversion");
6119 }
6121 if (load_refs() == ERR)
6122 die("Failed to load refs.");
6124 foreach_view (view, i)
6125 view->cmd_env = getenv(view->cmd_env);
6127 init_display();
6129 while (view_driver(display[current_view], request)) {
6130 int key;
6131 int i;
6133 foreach_view (view, i)
6134 update_view(view);
6135 view = display[current_view];
6137 /* Refresh, accept single keystroke of input */
6138 key = wgetch(status_win);
6140 /* wgetch() with nodelay() enabled returns ERR when there's no
6141 * input. */
6142 if (key == ERR) {
6143 request = REQ_NONE;
6144 continue;
6145 }
6147 request = get_keybinding(view->keymap, key);
6149 /* Some low-level request handling. This keeps access to
6150 * status_win restricted. */
6151 switch (request) {
6152 case REQ_PROMPT:
6153 {
6154 char *cmd = read_prompt(":");
6156 if (cmd && string_format(opt_cmd, "git %s", cmd)) {
6157 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
6158 request = REQ_VIEW_DIFF;
6159 } else {
6160 request = REQ_VIEW_PAGER;
6161 }
6163 /* Always reload^Wrerun commands from the prompt. */
6164 open_view(view, request, OPEN_RELOAD);
6165 }
6167 request = REQ_NONE;
6168 break;
6169 }
6170 case REQ_SEARCH:
6171 case REQ_SEARCH_BACK:
6172 {
6173 const char *prompt = request == REQ_SEARCH ? "/" : "?";
6174 char *search = read_prompt(prompt);
6176 if (search)
6177 string_ncopy(opt_search, search, strlen(search));
6178 else
6179 request = REQ_NONE;
6180 break;
6181 }
6182 case REQ_SCREEN_RESIZE:
6183 {
6184 int height, width;
6186 getmaxyx(stdscr, height, width);
6188 /* Resize the status view and let the view driver take
6189 * care of resizing the displayed views. */
6190 wresize(status_win, 1, width);
6191 mvwin(status_win, height - 1, 0);
6192 wrefresh(status_win);
6193 break;
6194 }
6195 default:
6196 break;
6197 }
6198 }
6200 quit(0);
6202 return 0;
6203 }