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