![[tokkee]](http://tokkee.org/images/avatar.png){kind=avatar}
4216d8700846bc48ba6ca7f95699dd4e0e59b6c3
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 struct keybinding *next;
746 };
748 static struct keybinding default_keybindings[] = {
749 /* View switching */
750 { 'm', REQ_VIEW_MAIN },
751 { 'd', REQ_VIEW_DIFF },
752 { 'l', REQ_VIEW_LOG },
753 { 't', REQ_VIEW_TREE },
754 { 'f', REQ_VIEW_BLOB },
755 { 'B', REQ_VIEW_BLAME },
756 { 'p', REQ_VIEW_PAGER },
757 { 'h', REQ_VIEW_HELP },
758 { 'S', REQ_VIEW_STATUS },
759 { 'c', REQ_VIEW_STAGE },
761 /* View manipulation */
762 { 'q', REQ_VIEW_CLOSE },
763 { KEY_TAB, REQ_VIEW_NEXT },
764 { KEY_RETURN, REQ_ENTER },
765 { KEY_UP, REQ_PREVIOUS },
766 { KEY_DOWN, REQ_NEXT },
767 { 'R', REQ_REFRESH },
768 { KEY_F(5), REQ_REFRESH },
769 { 'O', REQ_MAXIMIZE },
771 /* Cursor navigation */
772 { 'k', REQ_MOVE_UP },
773 { 'j', REQ_MOVE_DOWN },
774 { KEY_HOME, REQ_MOVE_FIRST_LINE },
775 { KEY_END, REQ_MOVE_LAST_LINE },
776 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
777 { ' ', REQ_MOVE_PAGE_DOWN },
778 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
779 { 'b', REQ_MOVE_PAGE_UP },
780 { '-', REQ_MOVE_PAGE_UP },
782 /* Scrolling */
783 { KEY_IC, REQ_SCROLL_LINE_UP },
784 { KEY_DC, REQ_SCROLL_LINE_DOWN },
785 { 'w', REQ_SCROLL_PAGE_UP },
786 { 's', REQ_SCROLL_PAGE_DOWN },
788 /* Searching */
789 { '/', REQ_SEARCH },
790 { '?', REQ_SEARCH_BACK },
791 { 'n', REQ_FIND_NEXT },
792 { 'N', REQ_FIND_PREV },
794 /* Misc */
795 { 'Q', REQ_QUIT },
796 { 'z', REQ_STOP_LOADING },
797 { 'v', REQ_SHOW_VERSION },
798 { 'r', REQ_SCREEN_REDRAW },
799 { '.', REQ_TOGGLE_LINENO },
800 { 'D', REQ_TOGGLE_DATE },
801 { 'A', REQ_TOGGLE_AUTHOR },
802 { 'g', REQ_TOGGLE_REV_GRAPH },
803 { 'F', REQ_TOGGLE_REFS },
804 { ':', REQ_PROMPT },
805 { 'u', REQ_STATUS_UPDATE },
806 { '!', REQ_STATUS_REVERT },
807 { 'M', REQ_STATUS_MERGE },
808 { '@', REQ_STAGE_NEXT },
809 { ',', REQ_TREE_PARENT },
810 { 'e', REQ_EDIT },
812 /* Using the ncurses SIGWINCH handler. */
813 { KEY_RESIZE, REQ_SCREEN_RESIZE },
814 };
816 #define KEYMAP_INFO \
817 KEYMAP_(GENERIC), \
818 KEYMAP_(MAIN), \
819 KEYMAP_(DIFF), \
820 KEYMAP_(LOG), \
821 KEYMAP_(TREE), \
822 KEYMAP_(BLOB), \
823 KEYMAP_(BLAME), \
824 KEYMAP_(PAGER), \
825 KEYMAP_(HELP), \
826 KEYMAP_(STATUS), \
827 KEYMAP_(STAGE)
829 enum keymap {
830 #define KEYMAP_(name) KEYMAP_##name
831 KEYMAP_INFO
832 #undef KEYMAP_
833 };
835 static struct int_map keymap_table[] = {
836 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
837 KEYMAP_INFO
838 #undef KEYMAP_
839 };
841 #define set_keymap(map, name) \
842 set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
844 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
846 static void
847 add_keybinding(enum keymap keymap, enum request request, int key)
848 {
849 struct keybinding *keybinding;
851 keybinding = calloc(1, sizeof(*keybinding));
852 if (!keybinding)
853 die("Failed to allocate keybinding");
855 keybinding->alias = key;
856 keybinding->request = request;
857 keybinding->next = keybindings[keymap];
858 keybindings[keymap] = keybinding;
859 }
861 /* Looks for a key binding first in the given map, then in the generic map, and
862 * lastly in the default keybindings. */
863 static enum request
864 get_keybinding(enum keymap keymap, int key)
865 {
866 struct keybinding *kbd;
867 int i;
869 for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
870 if (kbd->alias == key)
871 return kbd->request;
873 for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
874 if (kbd->alias == key)
875 return kbd->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 bool
2238 begin_update(struct view *view, bool refresh)
2239 {
2240 if (opt_cmd[0]) {
2241 string_copy(view->cmd, opt_cmd);
2242 opt_cmd[0] = 0;
2243 /* When running random commands, initially show the
2244 * command in the title. However, it maybe later be
2245 * overwritten if a commit line is selected. */
2246 if (view == VIEW(REQ_VIEW_PAGER))
2247 string_copy(view->ref, view->cmd);
2248 else
2249 view->ref[0] = 0;
2251 } else if (view == VIEW(REQ_VIEW_TREE)) {
2252 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2253 char path[SIZEOF_STR];
2255 if (strcmp(view->vid, view->id))
2256 opt_path[0] = path[0] = 0;
2257 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
2258 return FALSE;
2260 if (!string_format(view->cmd, format, view->id, path))
2261 return FALSE;
2263 } else if (!refresh) {
2264 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2265 const char *id = view->id;
2267 if (!string_format(view->cmd, format, id, id, id, id, id))
2268 return FALSE;
2270 /* Put the current ref_* value to the view title ref
2271 * member. This is needed by the blob view. Most other
2272 * views sets it automatically after loading because the
2273 * first line is a commit line. */
2274 string_copy_rev(view->ref, view->id);
2275 }
2277 /* Special case for the pager view. */
2278 if (opt_pipe) {
2279 view->pipe = opt_pipe;
2280 opt_pipe = NULL;
2281 } else {
2282 view->pipe = popen(view->cmd, "r");
2283 }
2285 if (!view->pipe)
2286 return FALSE;
2288 set_nonblocking_input(TRUE);
2289 reset_view(view);
2290 string_copy_rev(view->vid, view->id);
2292 view->start_time = time(NULL);
2294 return TRUE;
2295 }
2297 #define ITEM_CHUNK_SIZE 256
2298 static void *
2299 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2300 {
2301 size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2302 size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2304 if (mem == NULL || num_chunks != num_chunks_new) {
2305 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2306 mem = realloc(mem, *size * item_size);
2307 }
2309 return mem;
2310 }
2312 static struct line *
2313 realloc_lines(struct view *view, size_t line_size)
2314 {
2315 size_t alloc = view->line_alloc;
2316 struct line *tmp = realloc_items(view->line, &alloc, line_size,
2317 sizeof(*view->line));
2319 if (!tmp)
2320 return NULL;
2322 view->line = tmp;
2323 view->line_alloc = alloc;
2324 view->line_size = line_size;
2325 return view->line;
2326 }
2328 static bool
2329 update_view(struct view *view)
2330 {
2331 char in_buffer[BUFSIZ];
2332 char out_buffer[BUFSIZ * 2];
2333 char *line;
2334 /* The number of lines to read. If too low it will cause too much
2335 * redrawing (and possible flickering), if too high responsiveness
2336 * will suffer. */
2337 unsigned long lines = view->height;
2338 int redraw_from = -1;
2340 if (!view->pipe)
2341 return TRUE;
2343 /* Only redraw if lines are visible. */
2344 if (view->offset + view->height >= view->lines)
2345 redraw_from = view->lines - view->offset;
2347 /* FIXME: This is probably not perfect for backgrounded views. */
2348 if (!realloc_lines(view, view->lines + lines))
2349 goto alloc_error;
2351 while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
2352 size_t linelen = strlen(line);
2354 if (linelen)
2355 line[linelen - 1] = 0;
2357 if (opt_iconv != ICONV_NONE) {
2358 ICONV_CONST char *inbuf = line;
2359 size_t inlen = linelen;
2361 char *outbuf = out_buffer;
2362 size_t outlen = sizeof(out_buffer);
2364 size_t ret;
2366 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2367 if (ret != (size_t) -1) {
2368 line = out_buffer;
2369 linelen = strlen(out_buffer);
2370 }
2371 }
2373 if (!view->ops->read(view, line))
2374 goto alloc_error;
2376 if (lines-- == 1)
2377 break;
2378 }
2380 {
2381 int digits;
2383 lines = view->lines;
2384 for (digits = 0; lines; digits++)
2385 lines /= 10;
2387 /* Keep the displayed view in sync with line number scaling. */
2388 if (digits != view->digits) {
2389 view->digits = digits;
2390 redraw_from = 0;
2391 }
2392 }
2394 if (ferror(view->pipe) && errno != 0) {
2395 report("Failed to read: %s", strerror(errno));
2396 end_update(view, TRUE);
2398 } else if (feof(view->pipe)) {
2399 report("");
2400 end_update(view, FALSE);
2401 }
2403 if (view == VIEW(REQ_VIEW_TREE)) {
2404 /* Clear the view and redraw everything since the tree sorting
2405 * might have rearranged things. */
2406 redraw_view(view);
2408 } else if (redraw_from >= 0) {
2409 /* If this is an incremental update, redraw the previous line
2410 * since for commits some members could have changed when
2411 * loading the main view. */
2412 if (redraw_from > 0)
2413 redraw_from--;
2415 /* Since revision graph visualization requires knowledge
2416 * about the parent commit, it causes a further one-off
2417 * needed to be redrawn for incremental updates. */
2418 if (redraw_from > 0 && opt_rev_graph)
2419 redraw_from--;
2421 /* Incrementally draw avoids flickering. */
2422 redraw_view_from(view, redraw_from);
2423 }
2425 if (view == VIEW(REQ_VIEW_BLAME))
2426 redraw_view_dirty(view);
2428 /* Update the title _after_ the redraw so that if the redraw picks up a
2429 * commit reference in view->ref it'll be available here. */
2430 update_view_title(view);
2431 return TRUE;
2433 alloc_error:
2434 report("Allocation failure");
2435 end_update(view, TRUE);
2436 return FALSE;
2437 }
2439 static struct line *
2440 add_line_data(struct view *view, void *data, enum line_type type)
2441 {
2442 struct line *line = &view->line[view->lines++];
2444 memset(line, 0, sizeof(*line));
2445 line->type = type;
2446 line->data = data;
2448 return line;
2449 }
2451 static struct line *
2452 add_line_text(struct view *view, const char *text, enum line_type type)
2453 {
2454 char *data = text ? strdup(text) : NULL;
2456 return data ? add_line_data(view, data, type) : NULL;
2457 }
2460 /*
2461 * View opening
2462 */
2464 enum open_flags {
2465 OPEN_DEFAULT = 0, /* Use default view switching. */
2466 OPEN_SPLIT = 1, /* Split current view. */
2467 OPEN_BACKGROUNDED = 2, /* Backgrounded. */
2468 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
2469 OPEN_NOMAXIMIZE = 8, /* Do not maximize the current view. */
2470 OPEN_REFRESH = 16, /* Refresh view using previous command. */
2471 };
2473 static void
2474 open_view(struct view *prev, enum request request, enum open_flags flags)
2475 {
2476 bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2477 bool split = !!(flags & OPEN_SPLIT);
2478 bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH));
2479 bool nomaximize = !!(flags & (OPEN_NOMAXIMIZE | OPEN_REFRESH));
2480 struct view *view = VIEW(request);
2481 int nviews = displayed_views();
2482 struct view *base_view = display[0];
2484 if (view == prev && nviews == 1 && !reload) {
2485 report("Already in %s view", view->name);
2486 return;
2487 }
2489 if (view->git_dir && !opt_git_dir[0]) {
2490 report("The %s view is disabled in pager view", view->name);
2491 return;
2492 }
2494 if (split) {
2495 display[1] = view;
2496 if (!backgrounded)
2497 current_view = 1;
2498 } else if (!nomaximize) {
2499 /* Maximize the current view. */
2500 memset(display, 0, sizeof(display));
2501 current_view = 0;
2502 display[current_view] = view;
2503 }
2505 /* Resize the view when switching between split- and full-screen,
2506 * or when switching between two different full-screen views. */
2507 if (nviews != displayed_views() ||
2508 (nviews == 1 && base_view != display[0]))
2509 resize_display();
2511 if (view->pipe)
2512 end_update(view, TRUE);
2514 if (view->ops->open) {
2515 if (!view->ops->open(view)) {
2516 report("Failed to load %s view", view->name);
2517 return;
2518 }
2520 } else if ((reload || strcmp(view->vid, view->id)) &&
2521 !begin_update(view, flags & OPEN_REFRESH)) {
2522 report("Failed to load %s view", view->name);
2523 return;
2524 }
2526 if (split && prev->lineno - prev->offset >= prev->height) {
2527 /* Take the title line into account. */
2528 int lines = prev->lineno - prev->offset - prev->height + 1;
2530 /* Scroll the view that was split if the current line is
2531 * outside the new limited view. */
2532 do_scroll_view(prev, lines);
2533 }
2535 if (prev && view != prev) {
2536 if (split && !backgrounded) {
2537 /* "Blur" the previous view. */
2538 update_view_title(prev);
2539 }
2541 view->parent = prev;
2542 }
2544 if (view->pipe && view->lines == 0) {
2545 /* Clear the old view and let the incremental updating refill
2546 * the screen. */
2547 werase(view->win);
2548 report("");
2549 } else if (view_is_displayed(view)) {
2550 redraw_view(view);
2551 report("");
2552 }
2554 /* If the view is backgrounded the above calls to report()
2555 * won't redraw the view title. */
2556 if (backgrounded)
2557 update_view_title(view);
2558 }
2560 static bool
2561 run_confirm(const char *cmd, const char *prompt)
2562 {
2563 bool confirmation = prompt_yesno(prompt);
2565 if (confirmation)
2566 system(cmd);
2568 return confirmation;
2569 }
2571 static void
2572 open_external_viewer(const char *cmd)
2573 {
2574 def_prog_mode(); /* save current tty modes */
2575 endwin(); /* restore original tty modes */
2576 system(cmd);
2577 fprintf(stderr, "Press Enter to continue");
2578 getc(opt_tty);
2579 reset_prog_mode();
2580 redraw_display();
2581 }
2583 static void
2584 open_mergetool(const char *file)
2585 {
2586 char cmd[SIZEOF_STR];
2587 char file_sq[SIZEOF_STR];
2589 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2590 string_format(cmd, "git mergetool %s", file_sq)) {
2591 open_external_viewer(cmd);
2592 }
2593 }
2595 static void
2596 open_editor(bool from_root, const char *file)
2597 {
2598 char cmd[SIZEOF_STR];
2599 char file_sq[SIZEOF_STR];
2600 const char *editor;
2601 char *prefix = from_root ? opt_cdup : "";
2603 editor = getenv("GIT_EDITOR");
2604 if (!editor && *opt_editor)
2605 editor = opt_editor;
2606 if (!editor)
2607 editor = getenv("VISUAL");
2608 if (!editor)
2609 editor = getenv("EDITOR");
2610 if (!editor)
2611 editor = "vi";
2613 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2614 string_format(cmd, "%s %s%s", editor, prefix, file_sq)) {
2615 open_external_viewer(cmd);
2616 }
2617 }
2619 static void
2620 open_run_request(enum request request)
2621 {
2622 struct run_request *req = get_run_request(request);
2623 char buf[SIZEOF_STR * 2];
2624 size_t bufpos;
2625 char *cmd;
2627 if (!req) {
2628 report("Unknown run request");
2629 return;
2630 }
2632 bufpos = 0;
2633 cmd = req->cmd;
2635 while (cmd) {
2636 char *next = strstr(cmd, "%(");
2637 int len = next - cmd;
2638 char *value;
2640 if (!next) {
2641 len = strlen(cmd);
2642 value = "";
2644 } else if (!strncmp(next, "%(head)", 7)) {
2645 value = ref_head;
2647 } else if (!strncmp(next, "%(commit)", 9)) {
2648 value = ref_commit;
2650 } else if (!strncmp(next, "%(blob)", 7)) {
2651 value = ref_blob;
2653 } else {
2654 report("Unknown replacement in run request: `%s`", req->cmd);
2655 return;
2656 }
2658 if (!string_format_from(buf, &bufpos, "%.*s%s", len, cmd, value))
2659 return;
2661 if (next)
2662 next = strchr(next, ')') + 1;
2663 cmd = next;
2664 }
2666 open_external_viewer(buf);
2667 }
2669 /*
2670 * User request switch noodle
2671 */
2673 static int
2674 view_driver(struct view *view, enum request request)
2675 {
2676 int i;
2678 if (request == REQ_NONE) {
2679 doupdate();
2680 return TRUE;
2681 }
2683 if (request > REQ_NONE) {
2684 open_run_request(request);
2685 /* FIXME: When all views can refresh always do this. */
2686 if (view == VIEW(REQ_VIEW_STATUS) ||
2687 view == VIEW(REQ_VIEW_MAIN) ||
2688 view == VIEW(REQ_VIEW_LOG) ||
2689 view == VIEW(REQ_VIEW_STAGE))
2690 request = REQ_REFRESH;
2691 else
2692 return TRUE;
2693 }
2695 if (view && view->lines) {
2696 request = view->ops->request(view, request, &view->line[view->lineno]);
2697 if (request == REQ_NONE)
2698 return TRUE;
2699 }
2701 switch (request) {
2702 case REQ_MOVE_UP:
2703 case REQ_MOVE_DOWN:
2704 case REQ_MOVE_PAGE_UP:
2705 case REQ_MOVE_PAGE_DOWN:
2706 case REQ_MOVE_FIRST_LINE:
2707 case REQ_MOVE_LAST_LINE:
2708 move_view(view, request);
2709 break;
2711 case REQ_SCROLL_LINE_DOWN:
2712 case REQ_SCROLL_LINE_UP:
2713 case REQ_SCROLL_PAGE_DOWN:
2714 case REQ_SCROLL_PAGE_UP:
2715 scroll_view(view, request);
2716 break;
2718 case REQ_VIEW_BLAME:
2719 if (!opt_file[0]) {
2720 report("No file chosen, press %s to open tree view",
2721 get_key(REQ_VIEW_TREE));
2722 break;
2723 }
2724 open_view(view, request, OPEN_DEFAULT);
2725 break;
2727 case REQ_VIEW_BLOB:
2728 if (!ref_blob[0]) {
2729 report("No file chosen, press %s to open tree view",
2730 get_key(REQ_VIEW_TREE));
2731 break;
2732 }
2733 open_view(view, request, OPEN_DEFAULT);
2734 break;
2736 case REQ_VIEW_PAGER:
2737 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2738 report("No pager content, press %s to run command from prompt",
2739 get_key(REQ_PROMPT));
2740 break;
2741 }
2742 open_view(view, request, OPEN_DEFAULT);
2743 break;
2745 case REQ_VIEW_STAGE:
2746 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2747 report("No stage content, press %s to open the status view and choose file",
2748 get_key(REQ_VIEW_STATUS));
2749 break;
2750 }
2751 open_view(view, request, OPEN_DEFAULT);
2752 break;
2754 case REQ_VIEW_STATUS:
2755 if (opt_is_inside_work_tree == FALSE) {
2756 report("The status view requires a working tree");
2757 break;
2758 }
2759 open_view(view, request, OPEN_DEFAULT);
2760 break;
2762 case REQ_VIEW_MAIN:
2763 case REQ_VIEW_DIFF:
2764 case REQ_VIEW_LOG:
2765 case REQ_VIEW_TREE:
2766 case REQ_VIEW_HELP:
2767 open_view(view, request, OPEN_DEFAULT);
2768 break;
2770 case REQ_NEXT:
2771 case REQ_PREVIOUS:
2772 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2774 if ((view == VIEW(REQ_VIEW_DIFF) &&
2775 view->parent == VIEW(REQ_VIEW_MAIN)) ||
2776 (view == VIEW(REQ_VIEW_DIFF) &&
2777 view->parent == VIEW(REQ_VIEW_BLAME)) ||
2778 (view == VIEW(REQ_VIEW_STAGE) &&
2779 view->parent == VIEW(REQ_VIEW_STATUS)) ||
2780 (view == VIEW(REQ_VIEW_BLOB) &&
2781 view->parent == VIEW(REQ_VIEW_TREE))) {
2782 int line;
2784 view = view->parent;
2785 line = view->lineno;
2786 move_view(view, request);
2787 if (view_is_displayed(view))
2788 update_view_title(view);
2789 if (line != view->lineno)
2790 view->ops->request(view, REQ_ENTER,
2791 &view->line[view->lineno]);
2793 } else {
2794 move_view(view, request);
2795 }
2796 break;
2798 case REQ_VIEW_NEXT:
2799 {
2800 int nviews = displayed_views();
2801 int next_view = (current_view + 1) % nviews;
2803 if (next_view == current_view) {
2804 report("Only one view is displayed");
2805 break;
2806 }
2808 current_view = next_view;
2809 /* Blur out the title of the previous view. */
2810 update_view_title(view);
2811 report("");
2812 break;
2813 }
2814 case REQ_REFRESH:
2815 report("Refreshing is not yet supported for the %s view", view->name);
2816 break;
2818 case REQ_MAXIMIZE:
2819 if (displayed_views() == 2)
2820 open_view(view, VIEW_REQ(view), OPEN_DEFAULT);
2821 break;
2823 case REQ_TOGGLE_LINENO:
2824 opt_line_number = !opt_line_number;
2825 redraw_display();
2826 break;
2828 case REQ_TOGGLE_DATE:
2829 opt_date = !opt_date;
2830 redraw_display();
2831 break;
2833 case REQ_TOGGLE_AUTHOR:
2834 opt_author = !opt_author;
2835 redraw_display();
2836 break;
2838 case REQ_TOGGLE_REV_GRAPH:
2839 opt_rev_graph = !opt_rev_graph;
2840 redraw_display();
2841 break;
2843 case REQ_TOGGLE_REFS:
2844 opt_show_refs = !opt_show_refs;
2845 redraw_display();
2846 break;
2848 case REQ_SEARCH:
2849 case REQ_SEARCH_BACK:
2850 search_view(view, request);
2851 break;
2853 case REQ_FIND_NEXT:
2854 case REQ_FIND_PREV:
2855 find_next(view, request);
2856 break;
2858 case REQ_STOP_LOADING:
2859 for (i = 0; i < ARRAY_SIZE(views); i++) {
2860 view = &views[i];
2861 if (view->pipe)
2862 report("Stopped loading the %s view", view->name),
2863 end_update(view, TRUE);
2864 }
2865 break;
2867 case REQ_SHOW_VERSION:
2868 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2869 return TRUE;
2871 case REQ_SCREEN_RESIZE:
2872 resize_display();
2873 /* Fall-through */
2874 case REQ_SCREEN_REDRAW:
2875 redraw_display();
2876 break;
2878 case REQ_EDIT:
2879 report("Nothing to edit");
2880 break;
2882 case REQ_ENTER:
2883 report("Nothing to enter");
2884 break;
2886 case REQ_VIEW_CLOSE:
2887 /* XXX: Mark closed views by letting view->parent point to the
2888 * view itself. Parents to closed view should never be
2889 * followed. */
2890 if (view->parent &&
2891 view->parent->parent != view->parent) {
2892 memset(display, 0, sizeof(display));
2893 current_view = 0;
2894 display[current_view] = view->parent;
2895 view->parent = view;
2896 resize_display();
2897 redraw_display();
2898 report("");
2899 break;
2900 }
2901 /* Fall-through */
2902 case REQ_QUIT:
2903 return FALSE;
2905 default:
2906 report("Unknown key, press 'h' for help");
2907 return TRUE;
2908 }
2910 return TRUE;
2911 }
2914 /*
2915 * Pager backend
2916 */
2918 static bool
2919 pager_draw(struct view *view, struct line *line, unsigned int lineno)
2920 {
2921 char *text = line->data;
2923 if (opt_line_number && draw_lineno(view, lineno))
2924 return TRUE;
2926 draw_text(view, line->type, text, TRUE);
2927 return TRUE;
2928 }
2930 static bool
2931 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
2932 {
2933 char refbuf[SIZEOF_STR];
2934 char *ref = NULL;
2935 FILE *pipe;
2937 if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2938 return TRUE;
2940 pipe = popen(refbuf, "r");
2941 if (!pipe)
2942 return TRUE;
2944 if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2945 ref = chomp_string(ref);
2946 pclose(pipe);
2948 if (!ref || !*ref)
2949 return TRUE;
2951 /* This is the only fatal call, since it can "corrupt" the buffer. */
2952 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2953 return FALSE;
2955 return TRUE;
2956 }
2958 static void
2959 add_pager_refs(struct view *view, struct line *line)
2960 {
2961 char buf[SIZEOF_STR];
2962 char *commit_id = (char *)line->data + STRING_SIZE("commit ");
2963 struct ref **refs;
2964 size_t bufpos = 0, refpos = 0;
2965 const char *sep = "Refs: ";
2966 bool is_tag = FALSE;
2968 assert(line->type == LINE_COMMIT);
2970 refs = get_refs(commit_id);
2971 if (!refs) {
2972 if (view == VIEW(REQ_VIEW_DIFF))
2973 goto try_add_describe_ref;
2974 return;
2975 }
2977 do {
2978 struct ref *ref = refs[refpos];
2979 const char *fmt = ref->tag ? "%s[%s]" :
2980 ref->remote ? "%s<%s>" : "%s%s";
2982 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2983 return;
2984 sep = ", ";
2985 if (ref->tag)
2986 is_tag = TRUE;
2987 } while (refs[refpos++]->next);
2989 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2990 try_add_describe_ref:
2991 /* Add <tag>-g<commit_id> "fake" reference. */
2992 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2993 return;
2994 }
2996 if (bufpos == 0)
2997 return;
2999 if (!realloc_lines(view, view->line_size + 1))
3000 return;
3002 add_line_text(view, buf, LINE_PP_REFS);
3003 }
3005 static bool
3006 pager_read(struct view *view, char *data)
3007 {
3008 struct line *line;
3010 if (!data)
3011 return TRUE;
3013 line = add_line_text(view, data, get_line_type(data));
3014 if (!line)
3015 return FALSE;
3017 if (line->type == LINE_COMMIT &&
3018 (view == VIEW(REQ_VIEW_DIFF) ||
3019 view == VIEW(REQ_VIEW_LOG)))
3020 add_pager_refs(view, line);
3022 return TRUE;
3023 }
3025 static enum request
3026 pager_request(struct view *view, enum request request, struct line *line)
3027 {
3028 int split = 0;
3030 if (request != REQ_ENTER)
3031 return request;
3033 if (line->type == LINE_COMMIT &&
3034 (view == VIEW(REQ_VIEW_LOG) ||
3035 view == VIEW(REQ_VIEW_PAGER))) {
3036 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
3037 split = 1;
3038 }
3040 /* Always scroll the view even if it was split. That way
3041 * you can use Enter to scroll through the log view and
3042 * split open each commit diff. */
3043 scroll_view(view, REQ_SCROLL_LINE_DOWN);
3045 /* FIXME: A minor workaround. Scrolling the view will call report("")
3046 * but if we are scrolling a non-current view this won't properly
3047 * update the view title. */
3048 if (split)
3049 update_view_title(view);
3051 return REQ_NONE;
3052 }
3054 static bool
3055 pager_grep(struct view *view, struct line *line)
3056 {
3057 regmatch_t pmatch;
3058 char *text = line->data;
3060 if (!*text)
3061 return FALSE;
3063 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
3064 return FALSE;
3066 return TRUE;
3067 }
3069 static void
3070 pager_select(struct view *view, struct line *line)
3071 {
3072 if (line->type == LINE_COMMIT) {
3073 char *text = (char *)line->data + STRING_SIZE("commit ");
3075 if (view != VIEW(REQ_VIEW_PAGER))
3076 string_copy_rev(view->ref, text);
3077 string_copy_rev(ref_commit, text);
3078 }
3079 }
3081 static struct view_ops pager_ops = {
3082 "line",
3083 NULL,
3084 pager_read,
3085 pager_draw,
3086 pager_request,
3087 pager_grep,
3088 pager_select,
3089 };
3091 static enum request
3092 log_request(struct view *view, enum request request, struct line *line)
3093 {
3094 switch (request) {
3095 case REQ_REFRESH:
3096 load_refs();
3097 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
3098 return REQ_NONE;
3099 default:
3100 return pager_request(view, request, line);
3101 }
3102 }
3104 static struct view_ops log_ops = {
3105 "line",
3106 NULL,
3107 pager_read,
3108 pager_draw,
3109 log_request,
3110 pager_grep,
3111 pager_select,
3112 };
3115 /*
3116 * Help backend
3117 */
3119 static bool
3120 help_open(struct view *view)
3121 {
3122 char buf[BUFSIZ];
3123 int lines = ARRAY_SIZE(req_info) + 2;
3124 int i;
3126 if (view->lines > 0)
3127 return TRUE;
3129 for (i = 0; i < ARRAY_SIZE(req_info); i++)
3130 if (!req_info[i].request)
3131 lines++;
3133 lines += run_requests + 1;
3135 view->line = calloc(lines, sizeof(*view->line));
3136 if (!view->line)
3137 return FALSE;
3139 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3141 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3142 const char *key;
3144 if (req_info[i].request == REQ_NONE)
3145 continue;
3147 if (!req_info[i].request) {
3148 add_line_text(view, "", LINE_DEFAULT);
3149 add_line_text(view, req_info[i].help, LINE_DEFAULT);
3150 continue;
3151 }
3153 key = get_key(req_info[i].request);
3154 if (!*key)
3155 key = "(no key defined)";
3157 if (!string_format(buf, " %-25s %s", key, req_info[i].help))
3158 continue;
3160 add_line_text(view, buf, LINE_DEFAULT);
3161 }
3163 if (run_requests) {
3164 add_line_text(view, "", LINE_DEFAULT);
3165 add_line_text(view, "External commands:", LINE_DEFAULT);
3166 }
3168 for (i = 0; i < run_requests; i++) {
3169 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3170 const char *key;
3172 if (!req)
3173 continue;
3175 key = get_key_name(req->key);
3176 if (!*key)
3177 key = "(no key defined)";
3179 if (!string_format(buf, " %-10s %-14s `%s`",
3180 keymap_table[req->keymap].name,
3181 key, req->cmd))
3182 continue;
3184 add_line_text(view, buf, LINE_DEFAULT);
3185 }
3187 return TRUE;
3188 }
3190 static struct view_ops help_ops = {
3191 "line",
3192 help_open,
3193 NULL,
3194 pager_draw,
3195 pager_request,
3196 pager_grep,
3197 pager_select,
3198 };
3201 /*
3202 * Tree backend
3203 */
3205 struct tree_stack_entry {
3206 struct tree_stack_entry *prev; /* Entry below this in the stack */
3207 unsigned long lineno; /* Line number to restore */
3208 char *name; /* Position of name in opt_path */
3209 };
3211 /* The top of the path stack. */
3212 static struct tree_stack_entry *tree_stack = NULL;
3213 unsigned long tree_lineno = 0;
3215 static void
3216 pop_tree_stack_entry(void)
3217 {
3218 struct tree_stack_entry *entry = tree_stack;
3220 tree_lineno = entry->lineno;
3221 entry->name[0] = 0;
3222 tree_stack = entry->prev;
3223 free(entry);
3224 }
3226 static void
3227 push_tree_stack_entry(const char *name, unsigned long lineno)
3228 {
3229 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3230 size_t pathlen = strlen(opt_path);
3232 if (!entry)
3233 return;
3235 entry->prev = tree_stack;
3236 entry->name = opt_path + pathlen;
3237 tree_stack = entry;
3239 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3240 pop_tree_stack_entry();
3241 return;
3242 }
3244 /* Move the current line to the first tree entry. */
3245 tree_lineno = 1;
3246 entry->lineno = lineno;
3247 }
3249 /* Parse output from git-ls-tree(1):
3250 *
3251 * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
3252 * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
3253 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3254 * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
3255 */
3257 #define SIZEOF_TREE_ATTR \
3258 STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
3260 #define TREE_UP_FORMAT "040000 tree %s\t.."
3262 static int
3263 tree_compare_entry(enum line_type type1, const char *name1,
3264 enum line_type type2, const char *name2)
3265 {
3266 if (type1 != type2) {
3267 if (type1 == LINE_TREE_DIR)
3268 return -1;
3269 return 1;
3270 }
3272 return strcmp(name1, name2);
3273 }
3275 static const char *
3276 tree_path(struct line *line)
3277 {
3278 const char *path = line->data;
3280 return path + SIZEOF_TREE_ATTR;
3281 }
3283 static bool
3284 tree_read(struct view *view, char *text)
3285 {
3286 size_t textlen = text ? strlen(text) : 0;
3287 char buf[SIZEOF_STR];
3288 unsigned long pos;
3289 enum line_type type;
3290 bool first_read = view->lines == 0;
3292 if (!text)
3293 return TRUE;
3294 if (textlen <= SIZEOF_TREE_ATTR)
3295 return FALSE;
3297 type = text[STRING_SIZE("100644 ")] == 't'
3298 ? LINE_TREE_DIR : LINE_TREE_FILE;
3300 if (first_read) {
3301 /* Add path info line */
3302 if (!string_format(buf, "Directory path /%s", opt_path) ||
3303 !realloc_lines(view, view->line_size + 1) ||
3304 !add_line_text(view, buf, LINE_DEFAULT))
3305 return FALSE;
3307 /* Insert "link" to parent directory. */
3308 if (*opt_path) {
3309 if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
3310 !realloc_lines(view, view->line_size + 1) ||
3311 !add_line_text(view, buf, LINE_TREE_DIR))
3312 return FALSE;
3313 }
3314 }
3316 /* Strip the path part ... */
3317 if (*opt_path) {
3318 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3319 size_t striplen = strlen(opt_path);
3320 char *path = text + SIZEOF_TREE_ATTR;
3322 if (pathlen > striplen)
3323 memmove(path, path + striplen,
3324 pathlen - striplen + 1);
3325 }
3327 /* Skip "Directory ..." and ".." line. */
3328 for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
3329 struct line *line = &view->line[pos];
3330 const char *path1 = tree_path(line);
3331 char *path2 = text + SIZEOF_TREE_ATTR;
3332 int cmp = tree_compare_entry(line->type, path1, type, path2);
3334 if (cmp <= 0)
3335 continue;
3337 text = strdup(text);
3338 if (!text)
3339 return FALSE;
3341 if (view->lines > pos)
3342 memmove(&view->line[pos + 1], &view->line[pos],
3343 (view->lines - pos) * sizeof(*line));
3345 line = &view->line[pos];
3346 line->data = text;
3347 line->type = type;
3348 view->lines++;
3349 return TRUE;
3350 }
3352 if (!add_line_text(view, text, type))
3353 return FALSE;
3355 if (tree_lineno > view->lineno) {
3356 view->lineno = tree_lineno;
3357 tree_lineno = 0;
3358 }
3360 return TRUE;
3361 }
3363 static enum request
3364 tree_request(struct view *view, enum request request, struct line *line)
3365 {
3366 enum open_flags flags;
3368 switch (request) {
3369 case REQ_VIEW_BLAME:
3370 if (line->type != LINE_TREE_FILE) {
3371 report("Blame only supported for files");
3372 return REQ_NONE;
3373 }
3375 string_copy(opt_ref, view->vid);
3376 return request;
3378 case REQ_EDIT:
3379 if (line->type != LINE_TREE_FILE) {
3380 report("Edit only supported for files");
3381 } else if (!is_head_commit(view->vid)) {
3382 report("Edit only supported for files in the current work tree");
3383 } else {
3384 open_editor(TRUE, opt_file);
3385 }
3386 return REQ_NONE;
3388 case REQ_TREE_PARENT:
3389 if (!*opt_path) {
3390 /* quit view if at top of tree */
3391 return REQ_VIEW_CLOSE;
3392 }
3393 /* fake 'cd ..' */
3394 line = &view->line[1];
3395 break;
3397 case REQ_ENTER:
3398 break;
3400 default:
3401 return request;
3402 }
3404 /* Cleanup the stack if the tree view is at a different tree. */
3405 while (!*opt_path && tree_stack)
3406 pop_tree_stack_entry();
3408 switch (line->type) {
3409 case LINE_TREE_DIR:
3410 /* Depending on whether it is a subdir or parent (updir?) link
3411 * mangle the path buffer. */
3412 if (line == &view->line[1] && *opt_path) {
3413 pop_tree_stack_entry();
3415 } else {
3416 const char *basename = tree_path(line);
3418 push_tree_stack_entry(basename, view->lineno);
3419 }
3421 /* Trees and subtrees share the same ID, so they are not not
3422 * unique like blobs. */
3423 flags = OPEN_RELOAD;
3424 request = REQ_VIEW_TREE;
3425 break;
3427 case LINE_TREE_FILE:
3428 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3429 request = REQ_VIEW_BLOB;
3430 break;
3432 default:
3433 return TRUE;
3434 }
3436 open_view(view, request, flags);
3437 if (request == REQ_VIEW_TREE) {
3438 view->lineno = tree_lineno;
3439 }
3441 return REQ_NONE;
3442 }
3444 static void
3445 tree_select(struct view *view, struct line *line)
3446 {
3447 char *text = (char *)line->data + STRING_SIZE("100644 blob ");
3449 if (line->type == LINE_TREE_FILE) {
3450 string_copy_rev(ref_blob, text);
3451 string_format(opt_file, "%s%s", opt_path, tree_path(line));
3453 } else if (line->type != LINE_TREE_DIR) {
3454 return;
3455 }
3457 string_copy_rev(view->ref, text);
3458 }
3460 static struct view_ops tree_ops = {
3461 "file",
3462 NULL,
3463 tree_read,
3464 pager_draw,
3465 tree_request,
3466 pager_grep,
3467 tree_select,
3468 };
3470 static bool
3471 blob_read(struct view *view, char *line)
3472 {
3473 if (!line)
3474 return TRUE;
3475 return add_line_text(view, line, LINE_DEFAULT) != NULL;
3476 }
3478 static struct view_ops blob_ops = {
3479 "line",
3480 NULL,
3481 blob_read,
3482 pager_draw,
3483 pager_request,
3484 pager_grep,
3485 pager_select,
3486 };
3488 /*
3489 * Blame backend
3490 *
3491 * Loading the blame view is a two phase job:
3492 *
3493 * 1. File content is read either using opt_file from the
3494 * filesystem or using git-cat-file.
3495 * 2. Then blame information is incrementally added by
3496 * reading output from git-blame.
3497 */
3499 struct blame_commit {
3500 char id[SIZEOF_REV]; /* SHA1 ID. */
3501 char title[128]; /* First line of the commit message. */
3502 char author[75]; /* Author of the commit. */
3503 struct tm time; /* Date from the author ident. */
3504 char filename[128]; /* Name of file. */
3505 };
3507 struct blame {
3508 struct blame_commit *commit;
3509 unsigned int header:1;
3510 char text[1];
3511 };
3513 #define BLAME_CAT_FILE_CMD "git cat-file blob %s:%s"
3514 #define BLAME_INCREMENTAL_CMD "git blame --incremental %s -- %s"
3516 static bool
3517 blame_open(struct view *view)
3518 {
3519 char path[SIZEOF_STR];
3520 char ref[SIZEOF_STR] = "";
3522 if (sq_quote(path, 0, opt_file) >= sizeof(path))
3523 return FALSE;
3525 if (*opt_ref && sq_quote(ref, 0, opt_ref) >= sizeof(ref))
3526 return FALSE;
3528 if (*opt_ref) {
3529 if (!string_format(view->cmd, BLAME_CAT_FILE_CMD, ref, path))
3530 return FALSE;
3531 } else {
3532 view->pipe = fopen(opt_file, "r");
3533 if (!view->pipe &&
3534 !string_format(view->cmd, BLAME_CAT_FILE_CMD, "HEAD", path))
3535 return FALSE;
3536 }
3538 if (!view->pipe)
3539 view->pipe = popen(view->cmd, "r");
3540 if (!view->pipe)
3541 return FALSE;
3543 if (!string_format(view->cmd, BLAME_INCREMENTAL_CMD, ref, path))
3544 return FALSE;
3546 reset_view(view);
3547 string_format(view->ref, "%s ...", opt_file);
3548 string_copy_rev(view->vid, opt_file);
3549 set_nonblocking_input(TRUE);
3550 view->start_time = time(NULL);
3552 return TRUE;
3553 }
3555 static struct blame_commit *
3556 get_blame_commit(struct view *view, const char *id)
3557 {
3558 size_t i;
3560 for (i = 0; i < view->lines; i++) {
3561 struct blame *blame = view->line[i].data;
3563 if (!blame->commit)
3564 continue;
3566 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
3567 return blame->commit;
3568 }
3570 {
3571 struct blame_commit *commit = calloc(1, sizeof(*commit));
3573 if (commit)
3574 string_ncopy(commit->id, id, SIZEOF_REV);
3575 return commit;
3576 }
3577 }
3579 static bool
3580 parse_number(const char **posref, size_t *number, size_t min, size_t max)
3581 {
3582 const char *pos = *posref;
3584 *posref = NULL;
3585 pos = strchr(pos + 1, ' ');
3586 if (!pos || !isdigit(pos[1]))
3587 return FALSE;
3588 *number = atoi(pos + 1);
3589 if (*number < min || *number > max)
3590 return FALSE;
3592 *posref = pos;
3593 return TRUE;
3594 }
3596 static struct blame_commit *
3597 parse_blame_commit(struct view *view, const char *text, int *blamed)
3598 {
3599 struct blame_commit *commit;
3600 struct blame *blame;
3601 const char *pos = text + SIZEOF_REV - 1;
3602 size_t lineno;
3603 size_t group;
3605 if (strlen(text) <= SIZEOF_REV || *pos != ' ')
3606 return NULL;
3608 if (!parse_number(&pos, &lineno, 1, view->lines) ||
3609 !parse_number(&pos, &group, 1, view->lines - lineno + 1))
3610 return NULL;
3612 commit = get_blame_commit(view, text);
3613 if (!commit)
3614 return NULL;
3616 *blamed += group;
3617 while (group--) {
3618 struct line *line = &view->line[lineno + group - 1];
3620 blame = line->data;
3621 blame->commit = commit;
3622 blame->header = !group;
3623 line->dirty = 1;
3624 }
3626 return commit;
3627 }
3629 static bool
3630 blame_read_file(struct view *view, const char *line)
3631 {
3632 if (!line) {
3633 FILE *pipe = NULL;
3635 if (view->lines > 0)
3636 pipe = popen(view->cmd, "r");
3637 else if (!view->parent)
3638 die("No blame exist for %s", view->vid);
3639 view->cmd[0] = 0;
3640 if (!pipe) {
3641 report("Failed to load blame data");
3642 return TRUE;
3643 }
3645 fclose(view->pipe);
3646 view->pipe = pipe;
3647 return FALSE;
3649 } else {
3650 size_t linelen = strlen(line);
3651 struct blame *blame = malloc(sizeof(*blame) + linelen);
3653 blame->commit = NULL;
3654 strncpy(blame->text, line, linelen);
3655 blame->text[linelen] = 0;
3656 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
3657 }
3658 }
3660 static bool
3661 match_blame_header(const char *name, char **line)
3662 {
3663 size_t namelen = strlen(name);
3664 bool matched = !strncmp(name, *line, namelen);
3666 if (matched)
3667 *line += namelen;
3669 return matched;
3670 }
3672 static bool
3673 blame_read(struct view *view, char *line)
3674 {
3675 static struct blame_commit *commit = NULL;
3676 static int blamed = 0;
3677 static time_t author_time;
3679 if (*view->cmd)
3680 return blame_read_file(view, line);
3682 if (!line) {
3683 /* Reset all! */
3684 commit = NULL;
3685 blamed = 0;
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_ENTER:
3768 if (!blame->commit) {
3769 report("No commit loaded yet");
3770 break;
3771 }
3773 if (!strcmp(blame->commit->id, NULL_ID)) {
3774 char path[SIZEOF_STR];
3776 if (sq_quote(path, 0, view->vid) >= sizeof(path))
3777 break;
3778 string_format(opt_cmd, "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s 2>/dev/null", path);
3779 }
3781 open_view(view, REQ_VIEW_DIFF, flags);
3782 break;
3784 default:
3785 return request;
3786 }
3788 return REQ_NONE;
3789 }
3791 static bool
3792 blame_grep(struct view *view, struct line *line)
3793 {
3794 struct blame *blame = line->data;
3795 struct blame_commit *commit = blame->commit;
3796 regmatch_t pmatch;
3798 #define MATCH(text, on) \
3799 (on && *text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3801 if (commit) {
3802 char buf[DATE_COLS + 1];
3804 if (MATCH(commit->title, 1) ||
3805 MATCH(commit->author, opt_author) ||
3806 MATCH(commit->id, opt_date))
3807 return TRUE;
3809 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
3810 MATCH(buf, 1))
3811 return TRUE;
3812 }
3814 return MATCH(blame->text, 1);
3816 #undef MATCH
3817 }
3819 static void
3820 blame_select(struct view *view, struct line *line)
3821 {
3822 struct blame *blame = line->data;
3823 struct blame_commit *commit = blame->commit;
3825 if (!commit)
3826 return;
3828 if (!strcmp(commit->id, NULL_ID))
3829 string_ncopy(ref_commit, "HEAD", 4);
3830 else
3831 string_copy_rev(ref_commit, commit->id);
3832 }
3834 static struct view_ops blame_ops = {
3835 "line",
3836 blame_open,
3837 blame_read,
3838 blame_draw,
3839 blame_request,
3840 blame_grep,
3841 blame_select,
3842 };
3844 /*
3845 * Status backend
3846 */
3848 struct status {
3849 char status;
3850 struct {
3851 mode_t mode;
3852 char rev[SIZEOF_REV];
3853 char name[SIZEOF_STR];
3854 } old;
3855 struct {
3856 mode_t mode;
3857 char rev[SIZEOF_REV];
3858 char name[SIZEOF_STR];
3859 } new;
3860 };
3862 static char status_onbranch[SIZEOF_STR];
3863 static struct status stage_status;
3864 static enum line_type stage_line_type;
3865 static size_t stage_chunks;
3866 static int *stage_chunk;
3868 /* This should work even for the "On branch" line. */
3869 static inline bool
3870 status_has_none(struct view *view, struct line *line)
3871 {
3872 return line < view->line + view->lines && !line[1].data;
3873 }
3875 /* Get fields from the diff line:
3876 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
3877 */
3878 static inline bool
3879 status_get_diff(struct status *file, const char *buf, size_t bufsize)
3880 {
3881 const char *old_mode = buf + 1;
3882 const char *new_mode = buf + 8;
3883 const char *old_rev = buf + 15;
3884 const char *new_rev = buf + 56;
3885 const char *status = buf + 97;
3887 if (bufsize < 99 ||
3888 old_mode[-1] != ':' ||
3889 new_mode[-1] != ' ' ||
3890 old_rev[-1] != ' ' ||
3891 new_rev[-1] != ' ' ||
3892 status[-1] != ' ')
3893 return FALSE;
3895 file->status = *status;
3897 string_copy_rev(file->old.rev, old_rev);
3898 string_copy_rev(file->new.rev, new_rev);
3900 file->old.mode = strtoul(old_mode, NULL, 8);
3901 file->new.mode = strtoul(new_mode, NULL, 8);
3903 file->old.name[0] = file->new.name[0] = 0;
3905 return TRUE;
3906 }
3908 static bool
3909 status_run(struct view *view, const char cmd[], char status, enum line_type type)
3910 {
3911 struct status *file = NULL;
3912 struct status *unmerged = NULL;
3913 char buf[SIZEOF_STR * 4];
3914 size_t bufsize = 0;
3915 FILE *pipe;
3917 pipe = popen(cmd, "r");
3918 if (!pipe)
3919 return FALSE;
3921 add_line_data(view, NULL, type);
3923 while (!feof(pipe) && !ferror(pipe)) {
3924 char *sep;
3925 size_t readsize;
3927 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
3928 if (!readsize)
3929 break;
3930 bufsize += readsize;
3932 /* Process while we have NUL chars. */
3933 while ((sep = memchr(buf, 0, bufsize))) {
3934 size_t sepsize = sep - buf + 1;
3936 if (!file) {
3937 if (!realloc_lines(view, view->line_size + 1))
3938 goto error_out;
3940 file = calloc(1, sizeof(*file));
3941 if (!file)
3942 goto error_out;
3944 add_line_data(view, file, type);
3945 }
3947 /* Parse diff info part. */
3948 if (status) {
3949 file->status = status;
3950 if (status == 'A')
3951 string_copy(file->old.rev, NULL_ID);
3953 } else if (!file->status) {
3954 if (!status_get_diff(file, buf, sepsize))
3955 goto error_out;
3957 bufsize -= sepsize;
3958 memmove(buf, sep + 1, bufsize);
3960 sep = memchr(buf, 0, bufsize);
3961 if (!sep)
3962 break;
3963 sepsize = sep - buf + 1;
3965 /* Collapse all 'M'odified entries that
3966 * follow a associated 'U'nmerged entry.
3967 */
3968 if (file->status == 'U') {
3969 unmerged = file;
3971 } else if (unmerged) {
3972 int collapse = !strcmp(buf, unmerged->new.name);
3974 unmerged = NULL;
3975 if (collapse) {
3976 free(file);
3977 view->lines--;
3978 continue;
3979 }
3980 }
3981 }
3983 /* Grab the old name for rename/copy. */
3984 if (!*file->old.name &&
3985 (file->status == 'R' || file->status == 'C')) {
3986 sepsize = sep - buf + 1;
3987 string_ncopy(file->old.name, buf, sepsize);
3988 bufsize -= sepsize;
3989 memmove(buf, sep + 1, bufsize);
3991 sep = memchr(buf, 0, bufsize);
3992 if (!sep)
3993 break;
3994 sepsize = sep - buf + 1;
3995 }
3997 /* git-ls-files just delivers a NUL separated
3998 * list of file names similar to the second half
3999 * of the git-diff-* output. */
4000 string_ncopy(file->new.name, buf, sepsize);
4001 if (!*file->old.name)
4002 string_copy(file->old.name, file->new.name);
4003 bufsize -= sepsize;
4004 memmove(buf, sep + 1, bufsize);
4005 file = NULL;
4006 }
4007 }
4009 if (ferror(pipe)) {
4010 error_out:
4011 pclose(pipe);
4012 return FALSE;
4013 }
4015 if (!view->line[view->lines - 1].data)
4016 add_line_data(view, NULL, LINE_STAT_NONE);
4018 pclose(pipe);
4019 return TRUE;
4020 }
4022 /* Don't show unmerged entries in the staged section. */
4023 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --diff-filter=ACDMRTXB --cached -M HEAD"
4024 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
4025 #define STATUS_LIST_OTHER_CMD \
4026 "git ls-files -z --others --exclude-standard"
4027 #define STATUS_LIST_NO_HEAD_CMD \
4028 "git ls-files -z --cached --exclude-standard"
4030 #define STATUS_DIFF_INDEX_SHOW_CMD \
4031 "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s %s 2>/dev/null"
4033 #define STATUS_DIFF_FILES_SHOW_CMD \
4034 "git diff-files --root --patch-with-stat -C -M -- %s %s 2>/dev/null"
4036 #define STATUS_DIFF_NO_HEAD_SHOW_CMD \
4037 "git diff --no-color --patch-with-stat /dev/null %s 2>/dev/null"
4039 /* First parse staged info using git-diff-index(1), then parse unstaged
4040 * info using git-diff-files(1), and finally untracked files using
4041 * git-ls-files(1). */
4042 static bool
4043 status_open(struct view *view)
4044 {
4045 unsigned long prev_lineno = view->lineno;
4047 reset_view(view);
4049 if (!realloc_lines(view, view->line_size + 7))
4050 return FALSE;
4052 add_line_data(view, NULL, LINE_STAT_HEAD);
4053 if (is_initial_commit())
4054 string_copy(status_onbranch, "Initial commit");
4055 else if (!*opt_head)
4056 string_copy(status_onbranch, "Not currently on any branch");
4057 else if (!string_format(status_onbranch, "On branch %s", opt_head))
4058 return FALSE;
4060 system("git update-index -q --refresh >/dev/null 2>/dev/null");
4062 if (is_initial_commit()) {
4063 if (!status_run(view, STATUS_LIST_NO_HEAD_CMD, 'A', LINE_STAT_STAGED))
4064 return FALSE;
4065 } else if (!status_run(view, STATUS_DIFF_INDEX_CMD, 0, LINE_STAT_STAGED)) {
4066 return FALSE;
4067 }
4069 if (!status_run(view, STATUS_DIFF_FILES_CMD, 0, LINE_STAT_UNSTAGED) ||
4070 !status_run(view, STATUS_LIST_OTHER_CMD, '?', LINE_STAT_UNTRACKED))
4071 return FALSE;
4073 /* If all went well restore the previous line number to stay in
4074 * the context or select a line with something that can be
4075 * updated. */
4076 if (prev_lineno >= view->lines)
4077 prev_lineno = view->lines - 1;
4078 while (prev_lineno < view->lines && !view->line[prev_lineno].data)
4079 prev_lineno++;
4080 while (prev_lineno > 0 && !view->line[prev_lineno].data)
4081 prev_lineno--;
4083 /* If the above fails, always skip the "On branch" line. */
4084 if (prev_lineno < view->lines)
4085 view->lineno = prev_lineno;
4086 else
4087 view->lineno = 1;
4089 if (view->lineno < view->offset)
4090 view->offset = view->lineno;
4091 else if (view->offset + view->height <= view->lineno)
4092 view->offset = view->lineno - view->height + 1;
4094 return TRUE;
4095 }
4097 static bool
4098 status_draw(struct view *view, struct line *line, unsigned int lineno)
4099 {
4100 struct status *status = line->data;
4101 enum line_type type;
4102 const char *text;
4104 if (!status) {
4105 switch (line->type) {
4106 case LINE_STAT_STAGED:
4107 type = LINE_STAT_SECTION;
4108 text = "Changes to be committed:";
4109 break;
4111 case LINE_STAT_UNSTAGED:
4112 type = LINE_STAT_SECTION;
4113 text = "Changed but not updated:";
4114 break;
4116 case LINE_STAT_UNTRACKED:
4117 type = LINE_STAT_SECTION;
4118 text = "Untracked files:";
4119 break;
4121 case LINE_STAT_NONE:
4122 type = LINE_DEFAULT;
4123 text = " (no files)";
4124 break;
4126 case LINE_STAT_HEAD:
4127 type = LINE_STAT_HEAD;
4128 text = status_onbranch;
4129 break;
4131 default:
4132 return FALSE;
4133 }
4134 } else {
4135 static char buf[] = { '?', ' ', ' ', ' ', 0 };
4137 buf[0] = status->status;
4138 if (draw_text(view, line->type, buf, TRUE))
4139 return TRUE;
4140 type = LINE_DEFAULT;
4141 text = status->new.name;
4142 }
4144 draw_text(view, type, text, TRUE);
4145 return TRUE;
4146 }
4148 static enum request
4149 status_enter(struct view *view, struct line *line)
4150 {
4151 struct status *status = line->data;
4152 char oldpath[SIZEOF_STR] = "";
4153 char newpath[SIZEOF_STR] = "";
4154 const char *info;
4155 size_t cmdsize = 0;
4156 enum open_flags split;
4158 if (line->type == LINE_STAT_NONE ||
4159 (!status && line[1].type == LINE_STAT_NONE)) {
4160 report("No file to diff");
4161 return REQ_NONE;
4162 }
4164 if (status) {
4165 if (sq_quote(oldpath, 0, status->old.name) >= sizeof(oldpath))
4166 return REQ_QUIT;
4167 /* Diffs for unmerged entries are empty when pasing the
4168 * new path, so leave it empty. */
4169 if (status->status != 'U' &&
4170 sq_quote(newpath, 0, status->new.name) >= sizeof(newpath))
4171 return REQ_QUIT;
4172 }
4174 if (opt_cdup[0] &&
4175 line->type != LINE_STAT_UNTRACKED &&
4176 !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
4177 return REQ_QUIT;
4179 switch (line->type) {
4180 case LINE_STAT_STAGED:
4181 if (is_initial_commit()) {
4182 if (!string_format_from(opt_cmd, &cmdsize,
4183 STATUS_DIFF_NO_HEAD_SHOW_CMD,
4184 newpath))
4185 return REQ_QUIT;
4186 } else {
4187 if (!string_format_from(opt_cmd, &cmdsize,
4188 STATUS_DIFF_INDEX_SHOW_CMD,
4189 oldpath, newpath))
4190 return REQ_QUIT;
4191 }
4193 if (status)
4194 info = "Staged changes to %s";
4195 else
4196 info = "Staged changes";
4197 break;
4199 case LINE_STAT_UNSTAGED:
4200 if (!string_format_from(opt_cmd, &cmdsize,
4201 STATUS_DIFF_FILES_SHOW_CMD, oldpath, newpath))
4202 return REQ_QUIT;
4203 if (status)
4204 info = "Unstaged changes to %s";
4205 else
4206 info = "Unstaged changes";
4207 break;
4209 case LINE_STAT_UNTRACKED:
4210 if (opt_pipe)
4211 return REQ_QUIT;
4213 if (!status) {
4214 report("No file to show");
4215 return REQ_NONE;
4216 }
4218 if (!suffixcmp(status->new.name, -1, "/")) {
4219 report("Cannot display a directory");
4220 return REQ_NONE;
4221 }
4223 opt_pipe = fopen(status->new.name, "r");
4224 info = "Untracked file %s";
4225 break;
4227 case LINE_STAT_HEAD:
4228 return REQ_NONE;
4230 default:
4231 die("line type %d not handled in switch", line->type);
4232 }
4234 split = view_is_displayed(view) ? OPEN_SPLIT : 0;
4235 open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | split);
4236 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
4237 if (status) {
4238 stage_status = *status;
4239 } else {
4240 memset(&stage_status, 0, sizeof(stage_status));
4241 }
4243 stage_line_type = line->type;
4244 stage_chunks = 0;
4245 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
4246 }
4248 return REQ_NONE;
4249 }
4251 static bool
4252 status_exists(struct status *status, enum line_type type)
4253 {
4254 struct view *view = VIEW(REQ_VIEW_STATUS);
4255 struct line *line;
4257 for (line = view->line; line < view->line + view->lines; line++) {
4258 struct status *pos = line->data;
4260 if (line->type == type && pos &&
4261 !strcmp(status->new.name, pos->new.name))
4262 return TRUE;
4263 }
4265 return FALSE;
4266 }
4269 static FILE *
4270 status_update_prepare(enum line_type type)
4271 {
4272 char cmd[SIZEOF_STR];
4273 size_t cmdsize = 0;
4275 if (opt_cdup[0] &&
4276 type != LINE_STAT_UNTRACKED &&
4277 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4278 return NULL;
4280 switch (type) {
4281 case LINE_STAT_STAGED:
4282 string_add(cmd, cmdsize, "git update-index -z --index-info");
4283 break;
4285 case LINE_STAT_UNSTAGED:
4286 case LINE_STAT_UNTRACKED:
4287 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
4288 break;
4290 default:
4291 die("line type %d not handled in switch", type);
4292 }
4294 return popen(cmd, "w");
4295 }
4297 static bool
4298 status_update_write(FILE *pipe, struct status *status, enum line_type type)
4299 {
4300 char buf[SIZEOF_STR];
4301 size_t bufsize = 0;
4302 size_t written = 0;
4304 switch (type) {
4305 case LINE_STAT_STAGED:
4306 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
4307 status->old.mode,
4308 status->old.rev,
4309 status->old.name, 0))
4310 return FALSE;
4311 break;
4313 case LINE_STAT_UNSTAGED:
4314 case LINE_STAT_UNTRACKED:
4315 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
4316 return FALSE;
4317 break;
4319 default:
4320 die("line type %d not handled in switch", type);
4321 }
4323 while (!ferror(pipe) && written < bufsize) {
4324 written += fwrite(buf + written, 1, bufsize - written, pipe);
4325 }
4327 return written == bufsize;
4328 }
4330 static bool
4331 status_update_file(struct status *status, enum line_type type)
4332 {
4333 FILE *pipe = status_update_prepare(type);
4334 bool result;
4336 if (!pipe)
4337 return FALSE;
4339 result = status_update_write(pipe, status, type);
4340 pclose(pipe);
4341 return result;
4342 }
4344 static bool
4345 status_update_files(struct view *view, struct line *line)
4346 {
4347 FILE *pipe = status_update_prepare(line->type);
4348 bool result = TRUE;
4349 struct line *pos = view->line + view->lines;
4350 int files = 0;
4351 int file, done;
4353 if (!pipe)
4354 return FALSE;
4356 for (pos = line; pos < view->line + view->lines && pos->data; pos++)
4357 files++;
4359 for (file = 0, done = 0; result && file < files; line++, file++) {
4360 int almost_done = file * 100 / files;
4362 if (almost_done > done) {
4363 done = almost_done;
4364 string_format(view->ref, "updating file %u of %u (%d%% done)",
4365 file, files, done);
4366 update_view_title(view);
4367 }
4368 result = status_update_write(pipe, line->data, line->type);
4369 }
4371 pclose(pipe);
4372 return result;
4373 }
4375 static bool
4376 status_update(struct view *view)
4377 {
4378 struct line *line = &view->line[view->lineno];
4380 assert(view->lines);
4382 if (!line->data) {
4383 /* This should work even for the "On branch" line. */
4384 if (line < view->line + view->lines && !line[1].data) {
4385 report("Nothing to update");
4386 return FALSE;
4387 }
4389 if (!status_update_files(view, line + 1)) {
4390 report("Failed to update file status");
4391 return FALSE;
4392 }
4394 } else if (!status_update_file(line->data, line->type)) {
4395 report("Failed to update file status");
4396 return FALSE;
4397 }
4399 return TRUE;
4400 }
4402 static bool
4403 status_revert(struct status *status, enum line_type type, bool has_none)
4404 {
4405 if (!status || type != LINE_STAT_UNSTAGED) {
4406 if (type == LINE_STAT_STAGED) {
4407 report("Cannot revert changes to staged files");
4408 } else if (type == LINE_STAT_UNTRACKED) {
4409 report("Cannot revert changes to untracked files");
4410 } else if (has_none) {
4411 report("Nothing to revert");
4412 } else {
4413 report("Cannot revert changes to multiple files");
4414 }
4415 return FALSE;
4417 } else {
4418 char cmd[SIZEOF_STR];
4419 char file_sq[SIZEOF_STR];
4421 if (sq_quote(file_sq, 0, status->old.name) >= sizeof(file_sq) ||
4422 !string_format(cmd, "git checkout -- %s%s", opt_cdup, file_sq))
4423 return FALSE;
4425 return run_confirm(cmd, "Are you sure you want to overwrite any changes?");
4426 }
4427 }
4429 static enum request
4430 status_request(struct view *view, enum request request, struct line *line)
4431 {
4432 struct status *status = line->data;
4434 switch (request) {
4435 case REQ_STATUS_UPDATE:
4436 if (!status_update(view))
4437 return REQ_NONE;
4438 break;
4440 case REQ_STATUS_REVERT:
4441 if (!status_revert(status, line->type, status_has_none(view, line)))
4442 return REQ_NONE;
4443 break;
4445 case REQ_STATUS_MERGE:
4446 if (!status || status->status != 'U') {
4447 report("Merging only possible for files with unmerged status ('U').");
4448 return REQ_NONE;
4449 }
4450 open_mergetool(status->new.name);
4451 break;
4453 case REQ_EDIT:
4454 if (!status)
4455 return request;
4456 if (status->status == 'D') {
4457 report("File has been deleted.");
4458 return REQ_NONE;
4459 }
4461 open_editor(status->status != '?', status->new.name);
4462 break;
4464 case REQ_VIEW_BLAME:
4465 if (status) {
4466 string_copy(opt_file, status->new.name);
4467 opt_ref[0] = 0;
4468 }
4469 return request;
4471 case REQ_ENTER:
4472 /* After returning the status view has been split to
4473 * show the stage view. No further reloading is
4474 * necessary. */
4475 status_enter(view, line);
4476 return REQ_NONE;
4478 case REQ_REFRESH:
4479 /* Simply reload the view. */
4480 break;
4482 default:
4483 return request;
4484 }
4486 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
4488 return REQ_NONE;
4489 }
4491 static void
4492 status_select(struct view *view, struct line *line)
4493 {
4494 struct status *status = line->data;
4495 char file[SIZEOF_STR] = "all files";
4496 const char *text;
4497 const char *key;
4499 if (status && !string_format(file, "'%s'", status->new.name))
4500 return;
4502 if (!status && line[1].type == LINE_STAT_NONE)
4503 line++;
4505 switch (line->type) {
4506 case LINE_STAT_STAGED:
4507 text = "Press %s to unstage %s for commit";
4508 break;
4510 case LINE_STAT_UNSTAGED:
4511 text = "Press %s to stage %s for commit";
4512 break;
4514 case LINE_STAT_UNTRACKED:
4515 text = "Press %s to stage %s for addition";
4516 break;
4518 case LINE_STAT_HEAD:
4519 case LINE_STAT_NONE:
4520 text = "Nothing to update";
4521 break;
4523 default:
4524 die("line type %d not handled in switch", line->type);
4525 }
4527 if (status && status->status == 'U') {
4528 text = "Press %s to resolve conflict in %s";
4529 key = get_key(REQ_STATUS_MERGE);
4531 } else {
4532 key = get_key(REQ_STATUS_UPDATE);
4533 }
4535 string_format(view->ref, text, key, file);
4536 }
4538 static bool
4539 status_grep(struct view *view, struct line *line)
4540 {
4541 struct status *status = line->data;
4542 enum { S_STATUS, S_NAME, S_END } state;
4543 char buf[2] = "?";
4544 regmatch_t pmatch;
4546 if (!status)
4547 return FALSE;
4549 for (state = S_STATUS; state < S_END; state++) {
4550 const char *text;
4552 switch (state) {
4553 case S_NAME: text = status->new.name; break;
4554 case S_STATUS:
4555 buf[0] = status->status;
4556 text = buf;
4557 break;
4559 default:
4560 return FALSE;
4561 }
4563 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4564 return TRUE;
4565 }
4567 return FALSE;
4568 }
4570 static struct view_ops status_ops = {
4571 "file",
4572 status_open,
4573 NULL,
4574 status_draw,
4575 status_request,
4576 status_grep,
4577 status_select,
4578 };
4581 static bool
4582 stage_diff_line(FILE *pipe, struct line *line)
4583 {
4584 const char *buf = line->data;
4585 size_t bufsize = strlen(buf);
4586 size_t written = 0;
4588 while (!ferror(pipe) && written < bufsize) {
4589 written += fwrite(buf + written, 1, bufsize - written, pipe);
4590 }
4592 fputc('\n', pipe);
4594 return written == bufsize;
4595 }
4597 static bool
4598 stage_diff_write(FILE *pipe, struct line *line, struct line *end)
4599 {
4600 while (line < end) {
4601 if (!stage_diff_line(pipe, line++))
4602 return FALSE;
4603 if (line->type == LINE_DIFF_CHUNK ||
4604 line->type == LINE_DIFF_HEADER)
4605 break;
4606 }
4608 return TRUE;
4609 }
4611 static struct line *
4612 stage_diff_find(struct view *view, struct line *line, enum line_type type)
4613 {
4614 for (; view->line < line; line--)
4615 if (line->type == type)
4616 return line;
4618 return NULL;
4619 }
4621 static bool
4622 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
4623 {
4624 char cmd[SIZEOF_STR];
4625 size_t cmdsize = 0;
4626 struct line *diff_hdr;
4627 FILE *pipe;
4629 diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
4630 if (!diff_hdr)
4631 return FALSE;
4633 if (opt_cdup[0] &&
4634 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4635 return FALSE;
4637 if (!string_format_from(cmd, &cmdsize,
4638 "git apply --whitespace=nowarn %s %s - && "
4639 "git update-index -q --unmerged --refresh 2>/dev/null",
4640 revert ? "" : "--cached",
4641 revert || stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
4642 return FALSE;
4644 pipe = popen(cmd, "w");
4645 if (!pipe)
4646 return FALSE;
4648 if (!stage_diff_write(pipe, diff_hdr, chunk) ||
4649 !stage_diff_write(pipe, chunk, view->line + view->lines))
4650 chunk = NULL;
4652 pclose(pipe);
4654 return chunk ? TRUE : FALSE;
4655 }
4657 static bool
4658 stage_update(struct view *view, struct line *line)
4659 {
4660 struct line *chunk = NULL;
4662 if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
4663 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4665 if (chunk) {
4666 if (!stage_apply_chunk(view, chunk, FALSE)) {
4667 report("Failed to apply chunk");
4668 return FALSE;
4669 }
4671 } else if (!stage_status.status) {
4672 view = VIEW(REQ_VIEW_STATUS);
4674 for (line = view->line; line < view->line + view->lines; line++)
4675 if (line->type == stage_line_type)
4676 break;
4678 if (!status_update_files(view, line + 1)) {
4679 report("Failed to update files");
4680 return FALSE;
4681 }
4683 } else if (!status_update_file(&stage_status, stage_line_type)) {
4684 report("Failed to update file");
4685 return FALSE;
4686 }
4688 return TRUE;
4689 }
4691 static bool
4692 stage_revert(struct view *view, struct line *line)
4693 {
4694 struct line *chunk = NULL;
4696 if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
4697 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
4699 if (chunk) {
4700 if (!prompt_yesno("Are you sure you want to revert changes?"))
4701 return FALSE;
4703 if (!stage_apply_chunk(view, chunk, TRUE)) {
4704 report("Failed to revert chunk");
4705 return FALSE;
4706 }
4707 return TRUE;
4709 } else {
4710 return status_revert(stage_status.status ? &stage_status : NULL,
4711 stage_line_type, FALSE);
4712 }
4713 }
4716 static void
4717 stage_next(struct view *view, struct line *line)
4718 {
4719 int i;
4721 if (!stage_chunks) {
4722 static size_t alloc = 0;
4723 int *tmp;
4725 for (line = view->line; line < view->line + view->lines; line++) {
4726 if (line->type != LINE_DIFF_CHUNK)
4727 continue;
4729 tmp = realloc_items(stage_chunk, &alloc,
4730 stage_chunks, sizeof(*tmp));
4731 if (!tmp) {
4732 report("Allocation failure");
4733 return;
4734 }
4736 stage_chunk = tmp;
4737 stage_chunk[stage_chunks++] = line - view->line;
4738 }
4739 }
4741 for (i = 0; i < stage_chunks; i++) {
4742 if (stage_chunk[i] > view->lineno) {
4743 do_scroll_view(view, stage_chunk[i] - view->lineno);
4744 report("Chunk %d of %d", i + 1, stage_chunks);
4745 return;
4746 }
4747 }
4749 report("No next chunk found");
4750 }
4752 static enum request
4753 stage_request(struct view *view, enum request request, struct line *line)
4754 {
4755 switch (request) {
4756 case REQ_STATUS_UPDATE:
4757 if (!stage_update(view, line))
4758 return REQ_NONE;
4759 break;
4761 case REQ_STATUS_REVERT:
4762 if (!stage_revert(view, line))
4763 return REQ_NONE;
4764 break;
4766 case REQ_STAGE_NEXT:
4767 if (stage_line_type == LINE_STAT_UNTRACKED) {
4768 report("File is untracked; press %s to add",
4769 get_key(REQ_STATUS_UPDATE));
4770 return REQ_NONE;
4771 }
4772 stage_next(view, line);
4773 return REQ_NONE;
4775 case REQ_EDIT:
4776 if (!stage_status.new.name[0])
4777 return request;
4778 if (stage_status.status == 'D') {
4779 report("File has been deleted.");
4780 return REQ_NONE;
4781 }
4783 open_editor(stage_status.status != '?', stage_status.new.name);
4784 break;
4786 case REQ_REFRESH:
4787 /* Reload everything ... */
4788 break;
4790 case REQ_VIEW_BLAME:
4791 if (stage_status.new.name[0]) {
4792 string_copy(opt_file, stage_status.new.name);
4793 opt_ref[0] = 0;
4794 }
4795 return request;
4797 case REQ_ENTER:
4798 return pager_request(view, request, line);
4800 default:
4801 return request;
4802 }
4804 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD | OPEN_NOMAXIMIZE);
4806 /* Check whether the staged entry still exists, and close the
4807 * stage view if it doesn't. */
4808 if (!status_exists(&stage_status, stage_line_type))
4809 return REQ_VIEW_CLOSE;
4811 if (stage_line_type == LINE_STAT_UNTRACKED) {
4812 if (!suffixcmp(stage_status.new.name, -1, "/")) {
4813 report("Cannot display a directory");
4814 return REQ_NONE;
4815 }
4817 opt_pipe = fopen(stage_status.new.name, "r");
4818 }
4819 open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
4821 return REQ_NONE;
4822 }
4824 static struct view_ops stage_ops = {
4825 "line",
4826 NULL,
4827 pager_read,
4828 pager_draw,
4829 stage_request,
4830 pager_grep,
4831 pager_select,
4832 };
4835 /*
4836 * Revision graph
4837 */
4839 struct commit {
4840 char id[SIZEOF_REV]; /* SHA1 ID. */
4841 char title[128]; /* First line of the commit message. */
4842 char author[75]; /* Author of the commit. */
4843 struct tm time; /* Date from the author ident. */
4844 struct ref **refs; /* Repository references. */
4845 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
4846 size_t graph_size; /* The width of the graph array. */
4847 bool has_parents; /* Rewritten --parents seen. */
4848 };
4850 /* Size of rev graph with no "padding" columns */
4851 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
4853 struct rev_graph {
4854 struct rev_graph *prev, *next, *parents;
4855 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
4856 size_t size;
4857 struct commit *commit;
4858 size_t pos;
4859 unsigned int boundary:1;
4860 };
4862 /* Parents of the commit being visualized. */
4863 static struct rev_graph graph_parents[4];
4865 /* The current stack of revisions on the graph. */
4866 static struct rev_graph graph_stacks[4] = {
4867 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
4868 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
4869 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
4870 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
4871 };
4873 static inline bool
4874 graph_parent_is_merge(struct rev_graph *graph)
4875 {
4876 return graph->parents->size > 1;
4877 }
4879 static inline void
4880 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
4881 {
4882 struct commit *commit = graph->commit;
4884 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
4885 commit->graph[commit->graph_size++] = symbol;
4886 }
4888 static void
4889 clear_rev_graph(struct rev_graph *graph)
4890 {
4891 graph->boundary = 0;
4892 graph->size = graph->pos = 0;
4893 graph->commit = NULL;
4894 memset(graph->parents, 0, sizeof(*graph->parents));
4895 }
4897 static void
4898 done_rev_graph(struct rev_graph *graph)
4899 {
4900 if (graph_parent_is_merge(graph) &&
4901 graph->pos < graph->size - 1 &&
4902 graph->next->size == graph->size + graph->parents->size - 1) {
4903 size_t i = graph->pos + graph->parents->size - 1;
4905 graph->commit->graph_size = i * 2;
4906 while (i < graph->next->size - 1) {
4907 append_to_rev_graph(graph, ' ');
4908 append_to_rev_graph(graph, '\\');
4909 i++;
4910 }
4911 }
4913 clear_rev_graph(graph);
4914 }
4916 static void
4917 push_rev_graph(struct rev_graph *graph, const char *parent)
4918 {
4919 int i;
4921 /* "Collapse" duplicate parents lines.
4922 *
4923 * FIXME: This needs to also update update the drawn graph but
4924 * for now it just serves as a method for pruning graph lines. */
4925 for (i = 0; i < graph->size; i++)
4926 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
4927 return;
4929 if (graph->size < SIZEOF_REVITEMS) {
4930 string_copy_rev(graph->rev[graph->size++], parent);
4931 }
4932 }
4934 static chtype
4935 get_rev_graph_symbol(struct rev_graph *graph)
4936 {
4937 chtype symbol;
4939 if (graph->boundary)
4940 symbol = REVGRAPH_BOUND;
4941 else if (graph->parents->size == 0)
4942 symbol = REVGRAPH_INIT;
4943 else if (graph_parent_is_merge(graph))
4944 symbol = REVGRAPH_MERGE;
4945 else if (graph->pos >= graph->size)
4946 symbol = REVGRAPH_BRANCH;
4947 else
4948 symbol = REVGRAPH_COMMIT;
4950 return symbol;
4951 }
4953 static void
4954 draw_rev_graph(struct rev_graph *graph)
4955 {
4956 struct rev_filler {
4957 chtype separator, line;
4958 };
4959 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
4960 static struct rev_filler fillers[] = {
4961 { ' ', '|' },
4962 { '`', '.' },
4963 { '\'', ' ' },
4964 { '/', ' ' },
4965 };
4966 chtype symbol = get_rev_graph_symbol(graph);
4967 struct rev_filler *filler;
4968 size_t i;
4970 if (opt_line_graphics)
4971 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
4973 filler = &fillers[DEFAULT];
4975 for (i = 0; i < graph->pos; i++) {
4976 append_to_rev_graph(graph, filler->line);
4977 if (graph_parent_is_merge(graph->prev) &&
4978 graph->prev->pos == i)
4979 filler = &fillers[RSHARP];
4981 append_to_rev_graph(graph, filler->separator);
4982 }
4984 /* Place the symbol for this revision. */
4985 append_to_rev_graph(graph, symbol);
4987 if (graph->prev->size > graph->size)
4988 filler = &fillers[RDIAG];
4989 else
4990 filler = &fillers[DEFAULT];
4992 i++;
4994 for (; i < graph->size; i++) {
4995 append_to_rev_graph(graph, filler->separator);
4996 append_to_rev_graph(graph, filler->line);
4997 if (graph_parent_is_merge(graph->prev) &&
4998 i < graph->prev->pos + graph->parents->size)
4999 filler = &fillers[RSHARP];
5000 if (graph->prev->size > graph->size)
5001 filler = &fillers[LDIAG];
5002 }
5004 if (graph->prev->size > graph->size) {
5005 append_to_rev_graph(graph, filler->separator);
5006 if (filler->line != ' ')
5007 append_to_rev_graph(graph, filler->line);
5008 }
5009 }
5011 /* Prepare the next rev graph */
5012 static void
5013 prepare_rev_graph(struct rev_graph *graph)
5014 {
5015 size_t i;
5017 /* First, traverse all lines of revisions up to the active one. */
5018 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5019 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5020 break;
5022 push_rev_graph(graph->next, graph->rev[graph->pos]);
5023 }
5025 /* Interleave the new revision parent(s). */
5026 for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5027 push_rev_graph(graph->next, graph->parents->rev[i]);
5029 /* Lastly, put any remaining revisions. */
5030 for (i = graph->pos + 1; i < graph->size; i++)
5031 push_rev_graph(graph->next, graph->rev[i]);
5032 }
5034 static void
5035 update_rev_graph(struct rev_graph *graph)
5036 {
5037 /* If this is the finalizing update ... */
5038 if (graph->commit)
5039 prepare_rev_graph(graph);
5041 /* Graph visualization needs a one rev look-ahead,
5042 * so the first update doesn't visualize anything. */
5043 if (!graph->prev->commit)
5044 return;
5046 draw_rev_graph(graph->prev);
5047 done_rev_graph(graph->prev->prev);
5048 }
5051 /*
5052 * Main view backend
5053 */
5055 static bool
5056 main_draw(struct view *view, struct line *line, unsigned int lineno)
5057 {
5058 struct commit *commit = line->data;
5060 if (!*commit->author)
5061 return FALSE;
5063 if (opt_date && draw_date(view, &commit->time))
5064 return TRUE;
5066 if (opt_author &&
5067 draw_field(view, LINE_MAIN_AUTHOR, commit->author, opt_author_cols, TRUE))
5068 return TRUE;
5070 if (opt_rev_graph && commit->graph_size &&
5071 draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5072 return TRUE;
5074 if (opt_show_refs && commit->refs) {
5075 size_t i = 0;
5077 do {
5078 enum line_type type;
5080 if (commit->refs[i]->head)
5081 type = LINE_MAIN_HEAD;
5082 else if (commit->refs[i]->ltag)
5083 type = LINE_MAIN_LOCAL_TAG;
5084 else if (commit->refs[i]->tag)
5085 type = LINE_MAIN_TAG;
5086 else if (commit->refs[i]->tracked)
5087 type = LINE_MAIN_TRACKED;
5088 else if (commit->refs[i]->remote)
5089 type = LINE_MAIN_REMOTE;
5090 else
5091 type = LINE_MAIN_REF;
5093 if (draw_text(view, type, "[", TRUE) ||
5094 draw_text(view, type, commit->refs[i]->name, TRUE) ||
5095 draw_text(view, type, "]", TRUE))
5096 return TRUE;
5098 if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5099 return TRUE;
5100 } while (commit->refs[i++]->next);
5101 }
5103 draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5104 return TRUE;
5105 }
5107 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5108 static bool
5109 main_read(struct view *view, char *line)
5110 {
5111 static struct rev_graph *graph = graph_stacks;
5112 enum line_type type;
5113 struct commit *commit;
5115 if (!line) {
5116 int i;
5118 if (!view->lines && !view->parent)
5119 die("No revisions match the given arguments.");
5120 if (view->lines > 0) {
5121 commit = view->line[view->lines - 1].data;
5122 if (!*commit->author) {
5123 view->lines--;
5124 free(commit);
5125 graph->commit = NULL;
5126 }
5127 }
5128 update_rev_graph(graph);
5130 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5131 clear_rev_graph(&graph_stacks[i]);
5132 return TRUE;
5133 }
5135 type = get_line_type(line);
5136 if (type == LINE_COMMIT) {
5137 commit = calloc(1, sizeof(struct commit));
5138 if (!commit)
5139 return FALSE;
5141 line += STRING_SIZE("commit ");
5142 if (*line == '-') {
5143 graph->boundary = 1;
5144 line++;
5145 }
5147 string_copy_rev(commit->id, line);
5148 commit->refs = get_refs(commit->id);
5149 graph->commit = commit;
5150 add_line_data(view, commit, LINE_MAIN_COMMIT);
5152 while ((line = strchr(line, ' '))) {
5153 line++;
5154 push_rev_graph(graph->parents, line);
5155 commit->has_parents = TRUE;
5156 }
5157 return TRUE;
5158 }
5160 if (!view->lines)
5161 return TRUE;
5162 commit = view->line[view->lines - 1].data;
5164 switch (type) {
5165 case LINE_PARENT:
5166 if (commit->has_parents)
5167 break;
5168 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5169 break;
5171 case LINE_AUTHOR:
5172 {
5173 /* Parse author lines where the name may be empty:
5174 * author <email@address.tld> 1138474660 +0100
5175 */
5176 char *ident = line + STRING_SIZE("author ");
5177 char *nameend = strchr(ident, '<');
5178 char *emailend = strchr(ident, '>');
5180 if (!nameend || !emailend)
5181 break;
5183 update_rev_graph(graph);
5184 graph = graph->next;
5186 *nameend = *emailend = 0;
5187 ident = chomp_string(ident);
5188 if (!*ident) {
5189 ident = chomp_string(nameend + 1);
5190 if (!*ident)
5191 ident = "Unknown";
5192 }
5194 string_ncopy(commit->author, ident, strlen(ident));
5196 /* Parse epoch and timezone */
5197 if (emailend[1] == ' ') {
5198 char *secs = emailend + 2;
5199 char *zone = strchr(secs, ' ');
5200 time_t time = (time_t) atol(secs);
5202 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
5203 long tz;
5205 zone++;
5206 tz = ('0' - zone[1]) * 60 * 60 * 10;
5207 tz += ('0' - zone[2]) * 60 * 60;
5208 tz += ('0' - zone[3]) * 60;
5209 tz += ('0' - zone[4]) * 60;
5211 if (zone[0] == '-')
5212 tz = -tz;
5214 time -= tz;
5215 }
5217 gmtime_r(&time, &commit->time);
5218 }
5219 break;
5220 }
5221 default:
5222 /* Fill in the commit title if it has not already been set. */
5223 if (commit->title[0])
5224 break;
5226 /* Require titles to start with a non-space character at the
5227 * offset used by git log. */
5228 if (strncmp(line, " ", 4))
5229 break;
5230 line += 4;
5231 /* Well, if the title starts with a whitespace character,
5232 * try to be forgiving. Otherwise we end up with no title. */
5233 while (isspace(*line))
5234 line++;
5235 if (*line == '\0')
5236 break;
5237 /* FIXME: More graceful handling of titles; append "..." to
5238 * shortened titles, etc. */
5240 string_ncopy(commit->title, line, strlen(line));
5241 }
5243 return TRUE;
5244 }
5246 static enum request
5247 main_request(struct view *view, enum request request, struct line *line)
5248 {
5249 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5251 switch (request) {
5252 case REQ_ENTER:
5253 open_view(view, REQ_VIEW_DIFF, flags);
5254 break;
5255 case REQ_REFRESH:
5256 load_refs();
5257 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
5258 break;
5259 default:
5260 return request;
5261 }
5263 return REQ_NONE;
5264 }
5266 static bool
5267 grep_refs(struct ref **refs, regex_t *regex)
5268 {
5269 regmatch_t pmatch;
5270 size_t i = 0;
5272 if (!refs)
5273 return FALSE;
5274 do {
5275 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
5276 return TRUE;
5277 } while (refs[i++]->next);
5279 return FALSE;
5280 }
5282 static bool
5283 main_grep(struct view *view, struct line *line)
5284 {
5285 struct commit *commit = line->data;
5286 enum { S_TITLE, S_AUTHOR, S_DATE, S_REFS, S_END } state;
5287 char buf[DATE_COLS + 1];
5288 regmatch_t pmatch;
5290 for (state = S_TITLE; state < S_END; state++) {
5291 char *text;
5293 switch (state) {
5294 case S_TITLE: text = commit->title; break;
5295 case S_AUTHOR:
5296 if (!opt_author)
5297 continue;
5298 text = commit->author;
5299 break;
5300 case S_DATE:
5301 if (!opt_date)
5302 continue;
5303 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
5304 continue;
5305 text = buf;
5306 break;
5307 case S_REFS:
5308 if (!opt_show_refs)
5309 continue;
5310 if (grep_refs(commit->refs, view->regex) == TRUE)
5311 return TRUE;
5312 continue;
5313 default:
5314 return FALSE;
5315 }
5317 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5318 return TRUE;
5319 }
5321 return FALSE;
5322 }
5324 static void
5325 main_select(struct view *view, struct line *line)
5326 {
5327 struct commit *commit = line->data;
5329 string_copy_rev(view->ref, commit->id);
5330 string_copy_rev(ref_commit, view->ref);
5331 }
5333 static struct view_ops main_ops = {
5334 "commit",
5335 NULL,
5336 main_read,
5337 main_draw,
5338 main_request,
5339 main_grep,
5340 main_select,
5341 };
5344 /*
5345 * Unicode / UTF-8 handling
5346 *
5347 * NOTE: Much of the following code for dealing with unicode is derived from
5348 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
5349 * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
5350 */
5352 /* I've (over)annotated a lot of code snippets because I am not entirely
5353 * confident that the approach taken by this small UTF-8 interface is correct.
5354 * --jonas */
5356 static inline int
5357 unicode_width(unsigned long c)
5358 {
5359 if (c >= 0x1100 &&
5360 (c <= 0x115f /* Hangul Jamo */
5361 || c == 0x2329
5362 || c == 0x232a
5363 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
5364 /* CJK ... Yi */
5365 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
5366 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
5367 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
5368 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
5369 || (c >= 0xffe0 && c <= 0xffe6)
5370 || (c >= 0x20000 && c <= 0x2fffd)
5371 || (c >= 0x30000 && c <= 0x3fffd)))
5372 return 2;
5374 if (c == '\t')
5375 return opt_tab_size;
5377 return 1;
5378 }
5380 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
5381 * Illegal bytes are set one. */
5382 static const unsigned char utf8_bytes[256] = {
5383 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,
5384 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,
5385 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,
5386 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,
5387 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,
5388 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,
5389 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,
5390 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,
5391 };
5393 /* Decode UTF-8 multi-byte representation into a unicode character. */
5394 static inline unsigned long
5395 utf8_to_unicode(const char *string, size_t length)
5396 {
5397 unsigned long unicode;
5399 switch (length) {
5400 case 1:
5401 unicode = string[0];
5402 break;
5403 case 2:
5404 unicode = (string[0] & 0x1f) << 6;
5405 unicode += (string[1] & 0x3f);
5406 break;
5407 case 3:
5408 unicode = (string[0] & 0x0f) << 12;
5409 unicode += ((string[1] & 0x3f) << 6);
5410 unicode += (string[2] & 0x3f);
5411 break;
5412 case 4:
5413 unicode = (string[0] & 0x0f) << 18;
5414 unicode += ((string[1] & 0x3f) << 12);
5415 unicode += ((string[2] & 0x3f) << 6);
5416 unicode += (string[3] & 0x3f);
5417 break;
5418 case 5:
5419 unicode = (string[0] & 0x0f) << 24;
5420 unicode += ((string[1] & 0x3f) << 18);
5421 unicode += ((string[2] & 0x3f) << 12);
5422 unicode += ((string[3] & 0x3f) << 6);
5423 unicode += (string[4] & 0x3f);
5424 break;
5425 case 6:
5426 unicode = (string[0] & 0x01) << 30;
5427 unicode += ((string[1] & 0x3f) << 24);
5428 unicode += ((string[2] & 0x3f) << 18);
5429 unicode += ((string[3] & 0x3f) << 12);
5430 unicode += ((string[4] & 0x3f) << 6);
5431 unicode += (string[5] & 0x3f);
5432 break;
5433 default:
5434 die("Invalid unicode length");
5435 }
5437 /* Invalid characters could return the special 0xfffd value but NUL
5438 * should be just as good. */
5439 return unicode > 0xffff ? 0 : unicode;
5440 }
5442 /* Calculates how much of string can be shown within the given maximum width
5443 * and sets trimmed parameter to non-zero value if all of string could not be
5444 * shown. If the reserve flag is TRUE, it will reserve at least one
5445 * trailing character, which can be useful when drawing a delimiter.
5446 *
5447 * Returns the number of bytes to output from string to satisfy max_width. */
5448 static size_t
5449 utf8_length(const char *string, int *width, size_t max_width, int *trimmed, bool reserve)
5450 {
5451 const char *start = string;
5452 const char *end = strchr(string, '\0');
5453 unsigned char last_bytes = 0;
5454 size_t last_ucwidth = 0;
5456 *width = 0;
5457 *trimmed = 0;
5459 while (string < end) {
5460 int c = *(unsigned char *) string;
5461 unsigned char bytes = utf8_bytes[c];
5462 size_t ucwidth;
5463 unsigned long unicode;
5465 if (string + bytes > end)
5466 break;
5468 /* Change representation to figure out whether
5469 * it is a single- or double-width character. */
5471 unicode = utf8_to_unicode(string, bytes);
5472 /* FIXME: Graceful handling of invalid unicode character. */
5473 if (!unicode)
5474 break;
5476 ucwidth = unicode_width(unicode);
5477 *width += ucwidth;
5478 if (*width > max_width) {
5479 *trimmed = 1;
5480 *width -= ucwidth;
5481 if (reserve && *width == max_width) {
5482 string -= last_bytes;
5483 *width -= last_ucwidth;
5484 }
5485 break;
5486 }
5488 string += bytes;
5489 last_bytes = bytes;
5490 last_ucwidth = ucwidth;
5491 }
5493 return string - start;
5494 }
5497 /*
5498 * Status management
5499 */
5501 /* Whether or not the curses interface has been initialized. */
5502 static bool cursed = FALSE;
5504 /* The status window is used for polling keystrokes. */
5505 static WINDOW *status_win;
5507 static bool status_empty = TRUE;
5509 /* Update status and title window. */
5510 static void
5511 report(const char *msg, ...)
5512 {
5513 struct view *view = display[current_view];
5515 if (input_mode)
5516 return;
5518 if (!view) {
5519 char buf[SIZEOF_STR];
5520 va_list args;
5522 va_start(args, msg);
5523 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
5524 buf[sizeof(buf) - 1] = 0;
5525 buf[sizeof(buf) - 2] = '.';
5526 buf[sizeof(buf) - 3] = '.';
5527 buf[sizeof(buf) - 4] = '.';
5528 }
5529 va_end(args);
5530 die("%s", buf);
5531 }
5533 if (!status_empty || *msg) {
5534 va_list args;
5536 va_start(args, msg);
5538 wmove(status_win, 0, 0);
5539 if (*msg) {
5540 vwprintw(status_win, msg, args);
5541 status_empty = FALSE;
5542 } else {
5543 status_empty = TRUE;
5544 }
5545 wclrtoeol(status_win);
5546 wrefresh(status_win);
5548 va_end(args);
5549 }
5551 update_view_title(view);
5552 update_display_cursor(view);
5553 }
5555 /* Controls when nodelay should be in effect when polling user input. */
5556 static void
5557 set_nonblocking_input(bool loading)
5558 {
5559 static unsigned int loading_views;
5561 if ((loading == FALSE && loading_views-- == 1) ||
5562 (loading == TRUE && loading_views++ == 0))
5563 nodelay(status_win, loading);
5564 }
5566 static void
5567 init_display(void)
5568 {
5569 int x, y;
5571 /* Initialize the curses library */
5572 if (isatty(STDIN_FILENO)) {
5573 cursed = !!initscr();
5574 opt_tty = stdin;
5575 } else {
5576 /* Leave stdin and stdout alone when acting as a pager. */
5577 opt_tty = fopen("/dev/tty", "r+");
5578 if (!opt_tty)
5579 die("Failed to open /dev/tty");
5580 cursed = !!newterm(NULL, opt_tty, opt_tty);
5581 }
5583 if (!cursed)
5584 die("Failed to initialize curses");
5586 nonl(); /* Tell curses not to do NL->CR/NL on output */
5587 cbreak(); /* Take input chars one at a time, no wait for \n */
5588 noecho(); /* Don't echo input */
5589 leaveok(stdscr, TRUE);
5591 if (has_colors())
5592 init_colors();
5594 getmaxyx(stdscr, y, x);
5595 status_win = newwin(1, 0, y - 1, 0);
5596 if (!status_win)
5597 die("Failed to create status window");
5599 /* Enable keyboard mapping */
5600 keypad(status_win, TRUE);
5601 wbkgdset(status_win, get_line_attr(LINE_STATUS));
5603 TABSIZE = opt_tab_size;
5604 if (opt_line_graphics) {
5605 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
5606 }
5607 }
5609 static bool
5610 prompt_yesno(const char *prompt)
5611 {
5612 enum { WAIT, STOP, CANCEL } status = WAIT;
5613 bool answer = FALSE;
5615 while (status == WAIT) {
5616 struct view *view;
5617 int i, key;
5619 input_mode = TRUE;
5621 foreach_view (view, i)
5622 update_view(view);
5624 input_mode = FALSE;
5626 mvwprintw(status_win, 0, 0, "%s [Yy]/[Nn]", prompt);
5627 wclrtoeol(status_win);
5629 /* Refresh, accept single keystroke of input */
5630 key = wgetch(status_win);
5631 switch (key) {
5632 case ERR:
5633 break;
5635 case 'y':
5636 case 'Y':
5637 answer = TRUE;
5638 status = STOP;
5639 break;
5641 case KEY_ESC:
5642 case KEY_RETURN:
5643 case KEY_ENTER:
5644 case KEY_BACKSPACE:
5645 case 'n':
5646 case 'N':
5647 case '\n':
5648 default:
5649 answer = FALSE;
5650 status = CANCEL;
5651 }
5652 }
5654 /* Clear the status window */
5655 status_empty = FALSE;
5656 report("");
5658 return answer;
5659 }
5661 static char *
5662 read_prompt(const char *prompt)
5663 {
5664 enum { READING, STOP, CANCEL } status = READING;
5665 static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
5666 int pos = 0;
5668 while (status == READING) {
5669 struct view *view;
5670 int i, key;
5672 input_mode = TRUE;
5674 foreach_view (view, i)
5675 update_view(view);
5677 input_mode = FALSE;
5679 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
5680 wclrtoeol(status_win);
5682 /* Refresh, accept single keystroke of input */
5683 key = wgetch(status_win);
5684 switch (key) {
5685 case KEY_RETURN:
5686 case KEY_ENTER:
5687 case '\n':
5688 status = pos ? STOP : CANCEL;
5689 break;
5691 case KEY_BACKSPACE:
5692 if (pos > 0)
5693 pos--;
5694 else
5695 status = CANCEL;
5696 break;
5698 case KEY_ESC:
5699 status = CANCEL;
5700 break;
5702 case ERR:
5703 break;
5705 default:
5706 if (pos >= sizeof(buf)) {
5707 report("Input string too long");
5708 return NULL;
5709 }
5711 if (isprint(key))
5712 buf[pos++] = (char) key;
5713 }
5714 }
5716 /* Clear the status window */
5717 status_empty = FALSE;
5718 report("");
5720 if (status == CANCEL)
5721 return NULL;
5723 buf[pos++] = 0;
5725 return buf;
5726 }
5728 /*
5729 * Repository references
5730 */
5732 static struct ref *refs = NULL;
5733 static size_t refs_alloc = 0;
5734 static size_t refs_size = 0;
5736 /* Id <-> ref store */
5737 static struct ref ***id_refs = NULL;
5738 static size_t id_refs_alloc = 0;
5739 static size_t id_refs_size = 0;
5741 static int
5742 compare_refs(const void *ref1_, const void *ref2_)
5743 {
5744 const struct ref *ref1 = *(const struct ref **)ref1_;
5745 const struct ref *ref2 = *(const struct ref **)ref2_;
5747 if (ref1->tag != ref2->tag)
5748 return ref2->tag - ref1->tag;
5749 if (ref1->ltag != ref2->ltag)
5750 return ref2->ltag - ref2->ltag;
5751 if (ref1->head != ref2->head)
5752 return ref2->head - ref1->head;
5753 if (ref1->tracked != ref2->tracked)
5754 return ref2->tracked - ref1->tracked;
5755 if (ref1->remote != ref2->remote)
5756 return ref2->remote - ref1->remote;
5757 return strcmp(ref1->name, ref2->name);
5758 }
5760 static struct ref **
5761 get_refs(const char *id)
5762 {
5763 struct ref ***tmp_id_refs;
5764 struct ref **ref_list = NULL;
5765 size_t ref_list_alloc = 0;
5766 size_t ref_list_size = 0;
5767 size_t i;
5769 for (i = 0; i < id_refs_size; i++)
5770 if (!strcmp(id, id_refs[i][0]->id))
5771 return id_refs[i];
5773 tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
5774 sizeof(*id_refs));
5775 if (!tmp_id_refs)
5776 return NULL;
5778 id_refs = tmp_id_refs;
5780 for (i = 0; i < refs_size; i++) {
5781 struct ref **tmp;
5783 if (strcmp(id, refs[i].id))
5784 continue;
5786 tmp = realloc_items(ref_list, &ref_list_alloc,
5787 ref_list_size + 1, sizeof(*ref_list));
5788 if (!tmp) {
5789 if (ref_list)
5790 free(ref_list);
5791 return NULL;
5792 }
5794 ref_list = tmp;
5795 ref_list[ref_list_size] = &refs[i];
5796 /* XXX: The properties of the commit chains ensures that we can
5797 * safely modify the shared ref. The repo references will
5798 * always be similar for the same id. */
5799 ref_list[ref_list_size]->next = 1;
5801 ref_list_size++;
5802 }
5804 if (ref_list) {
5805 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
5806 ref_list[ref_list_size - 1]->next = 0;
5807 id_refs[id_refs_size++] = ref_list;
5808 }
5810 return ref_list;
5811 }
5813 static int
5814 read_ref(char *id, size_t idlen, char *name, size_t namelen)
5815 {
5816 struct ref *ref;
5817 bool tag = FALSE;
5818 bool ltag = FALSE;
5819 bool remote = FALSE;
5820 bool tracked = FALSE;
5821 bool check_replace = FALSE;
5822 bool head = FALSE;
5824 if (!prefixcmp(name, "refs/tags/")) {
5825 if (!suffixcmp(name, namelen, "^{}")) {
5826 namelen -= 3;
5827 name[namelen] = 0;
5828 if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
5829 check_replace = TRUE;
5830 } else {
5831 ltag = TRUE;
5832 }
5834 tag = TRUE;
5835 namelen -= STRING_SIZE("refs/tags/");
5836 name += STRING_SIZE("refs/tags/");
5838 } else if (!prefixcmp(name, "refs/remotes/")) {
5839 remote = TRUE;
5840 namelen -= STRING_SIZE("refs/remotes/");
5841 name += STRING_SIZE("refs/remotes/");
5842 tracked = !strcmp(opt_remote, name);
5844 } else if (!prefixcmp(name, "refs/heads/")) {
5845 namelen -= STRING_SIZE("refs/heads/");
5846 name += STRING_SIZE("refs/heads/");
5847 head = !strncmp(opt_head, name, namelen);
5849 } else if (!strcmp(name, "HEAD")) {
5850 string_ncopy(opt_head_rev, id, idlen);
5851 return OK;
5852 }
5854 if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
5855 /* it's an annotated tag, replace the previous sha1 with the
5856 * resolved commit id; relies on the fact git-ls-remote lists
5857 * the commit id of an annotated tag right before the commit id
5858 * it points to. */
5859 refs[refs_size - 1].ltag = ltag;
5860 string_copy_rev(refs[refs_size - 1].id, id);
5862 return OK;
5863 }
5864 refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
5865 if (!refs)
5866 return ERR;
5868 ref = &refs[refs_size++];
5869 ref->name = malloc(namelen + 1);
5870 if (!ref->name)
5871 return ERR;
5873 strncpy(ref->name, name, namelen);
5874 ref->name[namelen] = 0;
5875 ref->head = head;
5876 ref->tag = tag;
5877 ref->ltag = ltag;
5878 ref->remote = remote;
5879 ref->tracked = tracked;
5880 string_copy_rev(ref->id, id);
5882 return OK;
5883 }
5885 static int
5886 load_refs(void)
5887 {
5888 const char *cmd_env = getenv("TIG_LS_REMOTE");
5889 const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
5891 if (!*opt_git_dir)
5892 return OK;
5894 while (refs_size > 0)
5895 free(refs[--refs_size].name);
5896 while (id_refs_size > 0)
5897 free(id_refs[--id_refs_size]);
5899 return read_properties(popen(cmd, "r"), "\t", read_ref);
5900 }
5902 static int
5903 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
5904 {
5905 if (!strcmp(name, "i18n.commitencoding"))
5906 string_ncopy(opt_encoding, value, valuelen);
5908 if (!strcmp(name, "core.editor"))
5909 string_ncopy(opt_editor, value, valuelen);
5911 /* branch.<head>.remote */
5912 if (*opt_head &&
5913 !strncmp(name, "branch.", 7) &&
5914 !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5915 !strcmp(name + 7 + strlen(opt_head), ".remote"))
5916 string_ncopy(opt_remote, value, valuelen);
5918 if (*opt_head && *opt_remote &&
5919 !strncmp(name, "branch.", 7) &&
5920 !strncmp(name + 7, opt_head, strlen(opt_head)) &&
5921 !strcmp(name + 7 + strlen(opt_head), ".merge")) {
5922 size_t from = strlen(opt_remote);
5924 if (!prefixcmp(value, "refs/heads/")) {
5925 value += STRING_SIZE("refs/heads/");
5926 valuelen -= STRING_SIZE("refs/heads/");
5927 }
5929 if (!string_format_from(opt_remote, &from, "/%s", value))
5930 opt_remote[0] = 0;
5931 }
5933 return OK;
5934 }
5936 static int
5937 load_git_config(void)
5938 {
5939 return read_properties(popen("git " GIT_CONFIG " --list", "r"),
5940 "=", read_repo_config_option);
5941 }
5943 static int
5944 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
5945 {
5946 if (!opt_git_dir[0]) {
5947 string_ncopy(opt_git_dir, name, namelen);
5949 } else if (opt_is_inside_work_tree == -1) {
5950 /* This can be 3 different values depending on the
5951 * version of git being used. If git-rev-parse does not
5952 * understand --is-inside-work-tree it will simply echo
5953 * the option else either "true" or "false" is printed.
5954 * Default to true for the unknown case. */
5955 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
5957 } else if (opt_cdup[0] == ' ') {
5958 string_ncopy(opt_cdup, name, namelen);
5959 } else {
5960 if (!prefixcmp(name, "refs/heads/")) {
5961 namelen -= STRING_SIZE("refs/heads/");
5962 name += STRING_SIZE("refs/heads/");
5963 string_ncopy(opt_head, name, namelen);
5964 }
5965 }
5967 return OK;
5968 }
5970 static int
5971 load_repo_info(void)
5972 {
5973 int result;
5974 FILE *pipe = popen("(git rev-parse --git-dir --is-inside-work-tree "
5975 " --show-cdup; git symbolic-ref HEAD) 2>/dev/null", "r");
5977 /* XXX: The line outputted by "--show-cdup" can be empty so
5978 * initialize it to something invalid to make it possible to
5979 * detect whether it has been set or not. */
5980 opt_cdup[0] = ' ';
5982 result = read_properties(pipe, "=", read_repo_info);
5983 if (opt_cdup[0] == ' ')
5984 opt_cdup[0] = 0;
5986 return result;
5987 }
5989 static int
5990 read_properties(FILE *pipe, const char *separators,
5991 int (*read_property)(char *, size_t, char *, size_t))
5992 {
5993 char buffer[BUFSIZ];
5994 char *name;
5995 int state = OK;
5997 if (!pipe)
5998 return ERR;
6000 while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
6001 char *value;
6002 size_t namelen;
6003 size_t valuelen;
6005 name = chomp_string(name);
6006 namelen = strcspn(name, separators);
6008 if (name[namelen]) {
6009 name[namelen] = 0;
6010 value = chomp_string(name + namelen + 1);
6011 valuelen = strlen(value);
6013 } else {
6014 value = "";
6015 valuelen = 0;
6016 }
6018 state = read_property(name, namelen, value, valuelen);
6019 }
6021 if (state != ERR && ferror(pipe))
6022 state = ERR;
6024 pclose(pipe);
6026 return state;
6027 }
6030 /*
6031 * Main
6032 */
6034 static void __NORETURN
6035 quit(int sig)
6036 {
6037 /* XXX: Restore tty modes and let the OS cleanup the rest! */
6038 if (cursed)
6039 endwin();
6040 exit(0);
6041 }
6043 static void __NORETURN
6044 die(const char *err, ...)
6045 {
6046 va_list args;
6048 endwin();
6050 va_start(args, err);
6051 fputs("tig: ", stderr);
6052 vfprintf(stderr, err, args);
6053 fputs("\n", stderr);
6054 va_end(args);
6056 exit(1);
6057 }
6059 static void
6060 warn(const char *msg, ...)
6061 {
6062 va_list args;
6064 va_start(args, msg);
6065 fputs("tig warning: ", stderr);
6066 vfprintf(stderr, msg, args);
6067 fputs("\n", stderr);
6068 va_end(args);
6069 }
6071 int
6072 main(int argc, const char *argv[])
6073 {
6074 struct view *view;
6075 enum request request;
6076 size_t i;
6078 signal(SIGINT, quit);
6080 if (setlocale(LC_ALL, "")) {
6081 char *codeset = nl_langinfo(CODESET);
6083 string_ncopy(opt_codeset, codeset, strlen(codeset));
6084 }
6086 if (load_repo_info() == ERR)
6087 die("Failed to load repo info.");
6089 if (load_options() == ERR)
6090 die("Failed to load user config.");
6092 if (load_git_config() == ERR)
6093 die("Failed to load repo config.");
6095 request = parse_options(argc, argv);
6096 if (request == REQ_NONE)
6097 return 0;
6099 /* Require a git repository unless when running in pager mode. */
6100 if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
6101 die("Not a git repository");
6103 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
6104 opt_utf8 = FALSE;
6106 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
6107 opt_iconv = iconv_open(opt_codeset, opt_encoding);
6108 if (opt_iconv == ICONV_NONE)
6109 die("Failed to initialize character set conversion");
6110 }
6112 if (load_refs() == ERR)
6113 die("Failed to load refs.");
6115 foreach_view (view, i)
6116 view->cmd_env = getenv(view->cmd_env);
6118 init_display();
6120 while (view_driver(display[current_view], request)) {
6121 int key;
6122 int i;
6124 foreach_view (view, i)
6125 update_view(view);
6126 view = display[current_view];
6128 /* Refresh, accept single keystroke of input */
6129 key = wgetch(status_win);
6131 /* wgetch() with nodelay() enabled returns ERR when there's no
6132 * input. */
6133 if (key == ERR) {
6134 request = REQ_NONE;
6135 continue;
6136 }
6138 request = get_keybinding(view->keymap, key);
6140 /* Some low-level request handling. This keeps access to
6141 * status_win restricted. */
6142 switch (request) {
6143 case REQ_PROMPT:
6144 {
6145 char *cmd = read_prompt(":");
6147 if (cmd && string_format(opt_cmd, "git %s", cmd)) {
6148 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
6149 request = REQ_VIEW_DIFF;
6150 } else {
6151 request = REQ_VIEW_PAGER;
6152 }
6154 /* Always reload^Wrerun commands from the prompt. */
6155 open_view(view, request, OPEN_RELOAD);
6156 }
6158 request = REQ_NONE;
6159 break;
6160 }
6161 case REQ_SEARCH:
6162 case REQ_SEARCH_BACK:
6163 {
6164 const char *prompt = request == REQ_SEARCH ? "/" : "?";
6165 char *search = read_prompt(prompt);
6167 if (search)
6168 string_ncopy(opt_search, search, strlen(search));
6169 else
6170 request = REQ_NONE;
6171 break;
6172 }
6173 case REQ_SCREEN_RESIZE:
6174 {
6175 int height, width;
6177 getmaxyx(stdscr, height, width);
6179 /* Resize the status view and let the view driver take
6180 * care of resizing the displayed views. */
6181 wresize(status_win, 1, width);
6182 mvwin(status_win, height - 1, 0);
6183 wrefresh(status_win);
6184 break;
6185 }
6186 default:
6187 break;
6188 }
6189 }
6191 quit(0);
6193 return 0;
6194 }