~mdw / tig / blob
? search:
summary | shortlog | log | commit | commitdiff | tree
blame | history | raw | HEAD
Save current branch from rev-parse output and highlight it
[tig] / tig.c
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 */
13
14 #ifdef HAVE_CONFIG_H
15 #include "config.h"
16 #endif
17
18 #ifndef TIG_VERSION
19 #define TIG_VERSION "unknown-version"
20 #endif
21
22 #ifndef DEBUG
23 #define NDEBUG
24 #endif
25
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>
38
39 #include <regex.h>
40
41 #include <locale.h>
42 #include <langinfo.h>
43 #include <iconv.h>
44
45 /* ncurses(3): Must be defined to have extended wide-character functions. */
46 #define _XOPEN_SOURCE_EXTENDED
47
48 #include <curses.h>
49
50 #if __GNUC__ >= 3
51 #define __NORETURN __attribute__((__noreturn__))
52 #else
53 #define __NORETURN
54 #endif
55
56 static void __NORETURN die(const char *err, ...);
57 static void warn(const char *msg, ...);
58 static void report(const char *msg, ...);
59 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, size_t, char *, size_t));
60 static void set_nonblocking_input(bool loading);
61 static size_t utf8_length(const char *string, size_t max_width, int *trimmed, bool reserve);
62
63 #define ABS(x) ((x) >= 0 ? (x) : -(x))
64 #define MIN(x, y) ((x) < (y) ? (x) : (y))
65
66 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
67 #define STRING_SIZE(x) (sizeof(x) - 1)
68
69 #define SIZEOF_STR 1024 /* Default string size. */
70 #define SIZEOF_REF 256 /* Size of symbolic or SHA1 ID. */
71 #define SIZEOF_REV 41 /* Holds a SHA-1 and an ending NUL */
72
73 /* Revision graph */
74
75 #define REVGRAPH_INIT 'I'
76 #define REVGRAPH_MERGE 'M'
77 #define REVGRAPH_BRANCH '+'
78 #define REVGRAPH_COMMIT '*'
79 #define REVGRAPH_BOUND '^'
80 #define REVGRAPH_LINE '|'
81
82 #define SIZEOF_REVGRAPH 19 /* Size of revision ancestry graphics. */
83
84 /* This color name can be used to refer to the default term colors. */
85 #define COLOR_DEFAULT (-1)
86
87 #define ICONV_NONE ((iconv_t) -1)
88 #ifndef ICONV_CONST
89 #define ICONV_CONST /* nothing */
90 #endif
91
92 /* The format and size of the date column in the main view. */
93 #define DATE_FORMAT "%Y-%m-%d %H:%M"
94 #define DATE_COLS STRING_SIZE("2006-04-29 14:21 ")
95
96 #define AUTHOR_COLS 20
97 #define ID_COLS 8
98
99 /* The default interval between line numbers. */
100 #define NUMBER_INTERVAL 5
101
102 #define TABSIZE 8
103
104 #define SCALE_SPLIT_VIEW(height) ((height) * 2 / 3)
105
106 #ifndef GIT_CONFIG
107 #define GIT_CONFIG "git config"
108 #endif
109
110 #define TIG_LS_REMOTE \
111 "git ls-remote $(git rev-parse --git-dir) 2>/dev/null"
112
113 #define TIG_DIFF_CMD \
114 "git show --pretty=fuller --no-color --root --patch-with-stat --find-copies-harder -C %s 2>/dev/null"
115
116 #define TIG_LOG_CMD \
117 "git log --no-color --cc --stat -n100 %s 2>/dev/null"
118
119 #define TIG_MAIN_CMD \
120 "git log --no-color --topo-order --parents --boundary --pretty=raw %s 2>/dev/null"
121
122 #define TIG_TREE_CMD \
123 "git ls-tree %s %s"
124
125 #define TIG_BLOB_CMD \
126 "git cat-file blob %s"
127
128 /* XXX: Needs to be defined to the empty string. */
129 #define TIG_HELP_CMD ""
130 #define TIG_PAGER_CMD ""
131 #define TIG_STATUS_CMD ""
132 #define TIG_STAGE_CMD ""
133 #define TIG_BLAME_CMD ""
134
135 /* Some ascii-shorthands fitted into the ncurses namespace. */
136 #define KEY_TAB '\t'
137 #define KEY_RETURN '\r'
138 #define KEY_ESC 27
139
140
141 struct ref {
142 char *name; /* Ref name; tag or head names are shortened. */
143 char id[SIZEOF_REV]; /* Commit SHA1 ID */
144 unsigned int tag:1; /* Is it a tag? */
145 unsigned int ltag:1; /* If so, is the tag local? */
146 unsigned int remote:1; /* Is it a remote ref? */
147 unsigned int next:1; /* For ref lists: are there more refs? */
148 unsigned int head:1; /* Is it the current HEAD? */
149 };
150
151 static struct ref **get_refs(char *id);
152
153 struct int_map {
154 const char *name;
155 int namelen;
156 int value;
157 };
158
159 static int
160 set_from_int_map(struct int_map *map, size_t map_size,
161 int *value, const char *name, int namelen)
162 {
163
164 int i;
165
166 for (i = 0; i < map_size; i++)
167 if (namelen == map[i].namelen &&
168 !strncasecmp(name, map[i].name, namelen)) {
169 *value = map[i].value;
170 return OK;
171 }
172
173 return ERR;
174 }
175
176
177 /*
178 * String helpers
179 */
180
181 static inline void
182 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
183 {
184 if (srclen > dstlen - 1)
185 srclen = dstlen - 1;
186
187 strncpy(dst, src, srclen);
188 dst[srclen] = 0;
189 }
190
191 /* Shorthands for safely copying into a fixed buffer. */
192
193 #define string_copy(dst, src) \
194 string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
195
196 #define string_ncopy(dst, src, srclen) \
197 string_ncopy_do(dst, sizeof(dst), src, srclen)
198
199 #define string_copy_rev(dst, src) \
200 string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
201
202 #define string_add(dst, from, src) \
203 string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
204
205 static char *
206 chomp_string(char *name)
207 {
208 int namelen;
209
210 while (isspace(*name))
211 name++;
212
213 namelen = strlen(name) - 1;
214 while (namelen > 0 && isspace(name[namelen]))
215 name[namelen--] = 0;
216
217 return name;
218 }
219
220 static bool
221 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
222 {
223 va_list args;
224 size_t pos = bufpos ? *bufpos : 0;
225
226 va_start(args, fmt);
227 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
228 va_end(args);
229
230 if (bufpos)
231 *bufpos = pos;
232
233 return pos >= bufsize ? FALSE : TRUE;
234 }
235
236 #define string_format(buf, fmt, args...) \
237 string_nformat(buf, sizeof(buf), NULL, fmt, args)
238
239 #define string_format_from(buf, from, fmt, args...) \
240 string_nformat(buf, sizeof(buf), from, fmt, args)
241
242 static int
243 string_enum_compare(const char *str1, const char *str2, int len)
244 {
245 size_t i;
246
247 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
248
249 /* Diff-Header == DIFF_HEADER */
250 for (i = 0; i < len; i++) {
251 if (toupper(str1[i]) == toupper(str2[i]))
252 continue;
253
254 if (string_enum_sep(str1[i]) &&
255 string_enum_sep(str2[i]))
256 continue;
257
258 return str1[i] - str2[i];
259 }
260
261 return 0;
262 }
263
264 /* Shell quoting
265 *
266 * NOTE: The following is a slightly modified copy of the git project's shell
267 * quoting routines found in the quote.c file.
268 *
269 * Help to copy the thing properly quoted for the shell safety. any single
270 * quote is replaced with '\'', any exclamation point is replaced with '\!',
271 * and the whole thing is enclosed in a
272 *
273 * E.g.
274 * original sq_quote result
275 * name ==> name ==> 'name'
276 * a b ==> a b ==> 'a b'
277 * a'b ==> a'\''b ==> 'a'\''b'
278 * a!b ==> a'\!'b ==> 'a'\!'b'
279 */
280
281 static size_t
282 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
283 {
284 char c;
285
286 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
287
288 BUFPUT('\'');
289 while ((c = *src++)) {
290 if (c == '\'' || c == '!') {
291 BUFPUT('\'');
292 BUFPUT('\\');
293 BUFPUT(c);
294 BUFPUT('\'');
295 } else {
296 BUFPUT(c);
297 }
298 }
299 BUFPUT('\'');
300
301 if (bufsize < SIZEOF_STR)
302 buf[bufsize] = 0;
303
304 return bufsize;
305 }
306
307
308 /*
309 * User requests
310 */
311
312 #define REQ_INFO \
313 /* XXX: Keep the view request first and in sync with views[]. */ \
314 REQ_GROUP("View switching") \
315 REQ_(VIEW_MAIN, "Show main view"), \
316 REQ_(VIEW_DIFF, "Show diff view"), \
317 REQ_(VIEW_LOG, "Show log view"), \
318 REQ_(VIEW_TREE, "Show tree view"), \
319 REQ_(VIEW_BLOB, "Show blob view"), \
320 REQ_(VIEW_BLAME, "Show blame view"), \
321 REQ_(VIEW_HELP, "Show help page"), \
322 REQ_(VIEW_PAGER, "Show pager view"), \
323 REQ_(VIEW_STATUS, "Show status view"), \
324 REQ_(VIEW_STAGE, "Show stage view"), \
325 \
326 REQ_GROUP("View manipulation") \
327 REQ_(ENTER, "Enter current line and scroll"), \
328 REQ_(NEXT, "Move to next"), \
329 REQ_(PREVIOUS, "Move to previous"), \
330 REQ_(VIEW_NEXT, "Move focus to next view"), \
331 REQ_(REFRESH, "Reload and refresh"), \
332 REQ_(VIEW_CLOSE, "Close the current view"), \
333 REQ_(QUIT, "Close all views and quit"), \
334 \
335 REQ_GROUP("Cursor navigation") \
336 REQ_(MOVE_UP, "Move cursor one line up"), \
337 REQ_(MOVE_DOWN, "Move cursor one line down"), \
338 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
339 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
340 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
341 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
342 \
343 REQ_GROUP("Scrolling") \
344 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
345 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
346 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
347 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
348 \
349 REQ_GROUP("Searching") \
350 REQ_(SEARCH, "Search the view"), \
351 REQ_(SEARCH_BACK, "Search backwards in the view"), \
352 REQ_(FIND_NEXT, "Find next search match"), \
353 REQ_(FIND_PREV, "Find previous search match"), \
354 \
355 REQ_GROUP("Misc") \
356 REQ_(PROMPT, "Bring up the prompt"), \
357 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
358 REQ_(SCREEN_RESIZE, "Resize the screen"), \
359 REQ_(SHOW_VERSION, "Show version information"), \
360 REQ_(STOP_LOADING, "Stop all loading views"), \
361 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
362 REQ_(TOGGLE_DATE, "Toggle date display"), \
363 REQ_(TOGGLE_AUTHOR, "Toggle author display"), \
364 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization"), \
365 REQ_(TOGGLE_REFS, "Toggle reference display (tags/branches)"), \
366 REQ_(STATUS_UPDATE, "Update file status"), \
367 REQ_(STATUS_MERGE, "Merge file using external tool"), \
368 REQ_(TREE_PARENT, "Switch to parent directory in tree view"), \
369 REQ_(EDIT, "Open in editor"), \
370 REQ_(NONE, "Do nothing")
371
372
373 /* User action requests. */
374 enum request {
375 #define REQ_GROUP(help)
376 #define REQ_(req, help) REQ_##req
377
378 /* Offset all requests to avoid conflicts with ncurses getch values. */
379 REQ_OFFSET = KEY_MAX + 1,
380 REQ_INFO
381
382 #undef REQ_GROUP
383 #undef REQ_
384 };
385
386 struct request_info {
387 enum request request;
388 char *name;
389 int namelen;
390 char *help;
391 };
392
393 static struct request_info req_info[] = {
394 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
395 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
396 REQ_INFO
397 #undef REQ_GROUP
398 #undef REQ_
399 };
400
401 static enum request
402 get_request(const char *name)
403 {
404 int namelen = strlen(name);
405 int i;
406
407 for (i = 0; i < ARRAY_SIZE(req_info); i++)
408 if (req_info[i].namelen == namelen &&
409 !string_enum_compare(req_info[i].name, name, namelen))
410 return req_info[i].request;
411
412 return REQ_NONE;
413 }
414
415
416 /*
417 * Options
418 */
419
420 static const char usage[] =
421 "tig " TIG_VERSION " (" __DATE__ ")\n"
422 "\n"
423 "Usage: tig [options] [revs] [--] [paths]\n"
424 " or: tig show [options] [revs] [--] [paths]\n"
425 " or: tig blame [rev] path\n"
426 " or: tig status\n"
427 " or: tig < [git command output]\n"
428 "\n"
429 "Options:\n"
430 " -v, --version Show version and exit\n"
431 " -h, --help Show help message and exit";
432
433 /* Option and state variables. */
434 static bool opt_date = TRUE;
435 static bool opt_author = TRUE;
436 static bool opt_line_number = FALSE;
437 static bool opt_rev_graph = FALSE;
438 static bool opt_show_refs = TRUE;
439 static int opt_num_interval = NUMBER_INTERVAL;
440 static int opt_tab_size = TABSIZE;
441 static enum request opt_request = REQ_VIEW_MAIN;
442 static char opt_cmd[SIZEOF_STR] = "";
443 static char opt_path[SIZEOF_STR] = "";
444 static char opt_file[SIZEOF_STR] = "";
445 static char opt_ref[SIZEOF_REF] = "";
446 static char opt_head[SIZEOF_REF] = "";
447 static FILE *opt_pipe = NULL;
448 static char opt_encoding[20] = "UTF-8";
449 static bool opt_utf8 = TRUE;
450 static char opt_codeset[20] = "UTF-8";
451 static iconv_t opt_iconv = ICONV_NONE;
452 static char opt_search[SIZEOF_STR] = "";
453 static char opt_cdup[SIZEOF_STR] = "";
454 static char opt_git_dir[SIZEOF_STR] = "";
455 static signed char opt_is_inside_work_tree = -1; /* set to TRUE or FALSE */
456 static char opt_editor[SIZEOF_STR] = "";
457
458 static bool
459 parse_options(int argc, char *argv[])
460 {
461 size_t buf_size;
462 char *subcommand;
463 bool seen_dashdash = FALSE;
464 int i;
465
466 if (argc <= 1)
467 return TRUE;
468
469 subcommand = argv[1];
470 if (!strcmp(subcommand, "status") || !strcmp(subcommand, "-S")) {
471 opt_request = REQ_VIEW_STATUS;
472 if (!strcmp(subcommand, "-S"))
473 warn("`-S' has been deprecated; use `tig status' instead");
474 if (argc > 2)
475 warn("ignoring arguments after `%s'", subcommand);
476 return TRUE;
477
478 } else if (!strcmp(subcommand, "blame")) {
479 opt_request = REQ_VIEW_BLAME;
480 if (argc <= 2 || argc > 4)
481 die("invalid number of options to blame\n\n%s", usage);
482
483 i = 2;
484 if (argc == 4) {
485 string_ncopy(opt_ref, argv[i], strlen(argv[i]));
486 i++;
487 }
488
489 string_ncopy(opt_file, argv[i], strlen(argv[i]));
490 return TRUE;
491
492 } else if (!strcmp(subcommand, "show")) {
493 opt_request = REQ_VIEW_DIFF;
494
495 } else if (!strcmp(subcommand, "log") || !strcmp(subcommand, "diff")) {
496 opt_request = subcommand[0] == 'l'
497 ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
498 warn("`tig %s' has been deprecated", subcommand);
499
500 } else {
501 subcommand = NULL;
502 }
503
504 if (!subcommand)
505 /* XXX: This is vulnerable to the user overriding
506 * options required for the main view parser. */
507 string_copy(opt_cmd, "git log --no-color --pretty=raw --boundary --parents");
508 else
509 string_format(opt_cmd, "git %s", subcommand);
510
511 buf_size = strlen(opt_cmd);
512
513 for (i = 1 + !!subcommand; i < argc; i++) {
514 char *opt = argv[i];
515
516 if (seen_dashdash || !strcmp(opt, "--")) {
517 seen_dashdash = TRUE;
518
519 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
520 printf("tig version %s\n", TIG_VERSION);
521 return FALSE;
522
523 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
524 printf(usage);
525 return FALSE;
526 }
527
528 opt_cmd[buf_size++] = ' ';
529 buf_size = sq_quote(opt_cmd, buf_size, opt);
530 if (buf_size >= sizeof(opt_cmd))
531 die("command too long");
532 }
533
534 if (!isatty(STDIN_FILENO)) {
535 opt_request = REQ_VIEW_PAGER;
536 opt_pipe = stdin;
537 buf_size = 0;
538 }
539
540 opt_cmd[buf_size] = 0;
541
542 return TRUE;
543 }
544
545
546 /*
547 * Line-oriented content detection.
548 */
549
550 #define LINE_INFO \
551 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
552 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
553 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
554 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
555 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
556 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
557 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
558 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
559 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
560 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
561 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
562 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
563 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
564 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
565 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
566 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
567 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
568 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
569 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
570 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
571 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
572 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
573 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
574 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
575 LINE(AUTHOR, "author ", COLOR_CYAN, COLOR_DEFAULT, 0), \
576 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
577 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
578 LINE(ACKED, " Acked-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
579 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
580 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
581 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
582 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
583 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
584 LINE(MAIN_DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
585 LINE(MAIN_AUTHOR, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
586 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
587 LINE(MAIN_DELIM, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
588 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
589 LINE(MAIN_LOCAL_TAG,"", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
590 LINE(MAIN_REMOTE, "", COLOR_YELLOW, COLOR_DEFAULT, A_BOLD), \
591 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
592 LINE(MAIN_HEAD, "", COLOR_RED, COLOR_DEFAULT, A_BOLD), \
593 LINE(MAIN_REVGRAPH,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
594 LINE(TREE_DIR, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
595 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
596 LINE(STAT_SECTION, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
597 LINE(STAT_NONE, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
598 LINE(STAT_STAGED, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
599 LINE(STAT_UNSTAGED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
600 LINE(STAT_UNTRACKED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
601 LINE(BLAME_DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
602 LINE(BLAME_AUTHOR, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
603 LINE(BLAME_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
604 LINE(BLAME_ID, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
605 LINE(BLAME_LINENO, "", COLOR_CYAN, COLOR_DEFAULT, 0)
606
607 enum line_type {
608 #define LINE(type, line, fg, bg, attr) \
609 LINE_##type
610 LINE_INFO
611 #undef LINE
612 };
613
614 struct line_info {
615 const char *name; /* Option name. */
616 int namelen; /* Size of option name. */
617 const char *line; /* The start of line to match. */
618 int linelen; /* Size of string to match. */
619 int fg, bg, attr; /* Color and text attributes for the lines. */
620 };
621
622 static struct line_info line_info[] = {
623 #define LINE(type, line, fg, bg, attr) \
624 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
625 LINE_INFO
626 #undef LINE
627 };
628
629 static enum line_type
630 get_line_type(char *line)
631 {
632 int linelen = strlen(line);
633 enum line_type type;
634
635 for (type = 0; type < ARRAY_SIZE(line_info); type++)
636 /* Case insensitive search matches Signed-off-by lines better. */
637 if (linelen >= line_info[type].linelen &&
638 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
639 return type;
640
641 return LINE_DEFAULT;
642 }
643
644 static inline int
645 get_line_attr(enum line_type type)
646 {
647 assert(type < ARRAY_SIZE(line_info));
648 return COLOR_PAIR(type) | line_info[type].attr;
649 }
650
651 static struct line_info *
652 get_line_info(char *name, int namelen)
653 {
654 enum line_type type;
655
656 for (type = 0; type < ARRAY_SIZE(line_info); type++)
657 if (namelen == line_info[type].namelen &&
658 !string_enum_compare(line_info[type].name, name, namelen))
659 return &line_info[type];
660
661 return NULL;
662 }
663
664 static void
665 init_colors(void)
666 {
667 int default_bg = line_info[LINE_DEFAULT].bg;
668 int default_fg = line_info[LINE_DEFAULT].fg;
669 enum line_type type;
670
671 start_color();
672
673 if (assume_default_colors(default_fg, default_bg) == ERR) {
674 default_bg = COLOR_BLACK;
675 default_fg = COLOR_WHITE;
676 }
677
678 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
679 struct line_info *info = &line_info[type];
680 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
681 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
682
683 init_pair(type, fg, bg);
684 }
685 }
686
687 struct line {
688 enum line_type type;
689
690 /* State flags */
691 unsigned int selected:1;
692 unsigned int dirty:1;
693
694 void *data; /* User data */
695 };
696
697
698 /*
699 * Keys
700 */
701
702 struct keybinding {
703 int alias;
704 enum request request;
705 struct keybinding *next;
706 };
707
708 static struct keybinding default_keybindings[] = {
709 /* View switching */
710 { 'm', REQ_VIEW_MAIN },
711 { 'd', REQ_VIEW_DIFF },
712 { 'l', REQ_VIEW_LOG },
713 { 't', REQ_VIEW_TREE },
714 { 'f', REQ_VIEW_BLOB },
715 { 'B', REQ_VIEW_BLAME },
716 { 'p', REQ_VIEW_PAGER },
717 { 'h', REQ_VIEW_HELP },
718 { 'S', REQ_VIEW_STATUS },
719 { 'c', REQ_VIEW_STAGE },
720
721 /* View manipulation */
722 { 'q', REQ_VIEW_CLOSE },
723 { KEY_TAB, REQ_VIEW_NEXT },
724 { KEY_RETURN, REQ_ENTER },
725 { KEY_UP, REQ_PREVIOUS },
726 { KEY_DOWN, REQ_NEXT },
727 { 'R', REQ_REFRESH },
728
729 /* Cursor navigation */
730 { 'k', REQ_MOVE_UP },
731 { 'j', REQ_MOVE_DOWN },
732 { KEY_HOME, REQ_MOVE_FIRST_LINE },
733 { KEY_END, REQ_MOVE_LAST_LINE },
734 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
735 { ' ', REQ_MOVE_PAGE_DOWN },
736 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
737 { 'b', REQ_MOVE_PAGE_UP },
738 { '-', REQ_MOVE_PAGE_UP },
739
740 /* Scrolling */
741 { KEY_IC, REQ_SCROLL_LINE_UP },
742 { KEY_DC, REQ_SCROLL_LINE_DOWN },
743 { 'w', REQ_SCROLL_PAGE_UP },
744 { 's', REQ_SCROLL_PAGE_DOWN },
745
746 /* Searching */
747 { '/', REQ_SEARCH },
748 { '?', REQ_SEARCH_BACK },
749 { 'n', REQ_FIND_NEXT },
750 { 'N', REQ_FIND_PREV },
751
752 /* Misc */
753 { 'Q', REQ_QUIT },
754 { 'z', REQ_STOP_LOADING },
755 { 'v', REQ_SHOW_VERSION },
756 { 'r', REQ_SCREEN_REDRAW },
757 { '.', REQ_TOGGLE_LINENO },
758 { 'D', REQ_TOGGLE_DATE },
759 { 'A', REQ_TOGGLE_AUTHOR },
760 { 'g', REQ_TOGGLE_REV_GRAPH },
761 { 'F', REQ_TOGGLE_REFS },
762 { ':', REQ_PROMPT },
763 { 'u', REQ_STATUS_UPDATE },
764 { 'M', REQ_STATUS_MERGE },
765 { ',', REQ_TREE_PARENT },
766 { 'e', REQ_EDIT },
767
768 /* Using the ncurses SIGWINCH handler. */
769 { KEY_RESIZE, REQ_SCREEN_RESIZE },
770 };
771
772 #define KEYMAP_INFO \
773 KEYMAP_(GENERIC), \
774 KEYMAP_(MAIN), \
775 KEYMAP_(DIFF), \
776 KEYMAP_(LOG), \
777 KEYMAP_(TREE), \
778 KEYMAP_(BLOB), \
779 KEYMAP_(BLAME), \
780 KEYMAP_(PAGER), \
781 KEYMAP_(HELP), \
782 KEYMAP_(STATUS), \
783 KEYMAP_(STAGE)
784
785 enum keymap {
786 #define KEYMAP_(name) KEYMAP_##name
787 KEYMAP_INFO
788 #undef KEYMAP_
789 };
790
791 static struct int_map keymap_table[] = {
792 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
793 KEYMAP_INFO
794 #undef KEYMAP_
795 };
796
797 #define set_keymap(map, name) \
798 set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
799
800 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
801
802 static void
803 add_keybinding(enum keymap keymap, enum request request, int key)
804 {
805 struct keybinding *keybinding;
806
807 keybinding = calloc(1, sizeof(*keybinding));
808 if (!keybinding)
809 die("Failed to allocate keybinding");
810
811 keybinding->alias = key;
812 keybinding->request = request;
813 keybinding->next = keybindings[keymap];
814 keybindings[keymap] = keybinding;
815 }
816
817 /* Looks for a key binding first in the given map, then in the generic map, and
818 * lastly in the default keybindings. */
819 static enum request
820 get_keybinding(enum keymap keymap, int key)
821 {
822 struct keybinding *kbd;
823 int i;
824
825 for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
826 if (kbd->alias == key)
827 return kbd->request;
828
829 for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
830 if (kbd->alias == key)
831 return kbd->request;
832
833 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
834 if (default_keybindings[i].alias == key)
835 return default_keybindings[i].request;
836
837 return (enum request) key;
838 }
839
840
841 struct key {
842 char *name;
843 int value;
844 };
845
846 static struct key key_table[] = {
847 { "Enter", KEY_RETURN },
848 { "Space", ' ' },
849 { "Backspace", KEY_BACKSPACE },
850 { "Tab", KEY_TAB },
851 { "Escape", KEY_ESC },
852 { "Left", KEY_LEFT },
853 { "Right", KEY_RIGHT },
854 { "Up", KEY_UP },
855 { "Down", KEY_DOWN },
856 { "Insert", KEY_IC },
857 { "Delete", KEY_DC },
858 { "Hash", '#' },
859 { "Home", KEY_HOME },
860 { "End", KEY_END },
861 { "PageUp", KEY_PPAGE },
862 { "PageDown", KEY_NPAGE },
863 { "F1", KEY_F(1) },
864 { "F2", KEY_F(2) },
865 { "F3", KEY_F(3) },
866 { "F4", KEY_F(4) },
867 { "F5", KEY_F(5) },
868 { "F6", KEY_F(6) },
869 { "F7", KEY_F(7) },
870 { "F8", KEY_F(8) },
871 { "F9", KEY_F(9) },
872 { "F10", KEY_F(10) },
873 { "F11", KEY_F(11) },
874 { "F12", KEY_F(12) },
875 };
876
877 static int
878 get_key_value(const char *name)
879 {
880 int i;
881
882 for (i = 0; i < ARRAY_SIZE(key_table); i++)
883 if (!strcasecmp(key_table[i].name, name))
884 return key_table[i].value;
885
886 if (strlen(name) == 1 && isprint(*name))
887 return (int) *name;
888
889 return ERR;
890 }
891
892 static char *
893 get_key_name(int key_value)
894 {
895 static char key_char[] = "'X'";
896 char *seq = NULL;
897 int key;
898
899 for (key = 0; key < ARRAY_SIZE(key_table); key++)
900 if (key_table[key].value == key_value)
901 seq = key_table[key].name;
902
903 if (seq == NULL &&
904 key_value < 127 &&
905 isprint(key_value)) {
906 key_char[1] = (char) key_value;
907 seq = key_char;
908 }
909
910 return seq ? seq : "'?'";
911 }
912
913 static char *
914 get_key(enum request request)
915 {
916 static char buf[BUFSIZ];
917 size_t pos = 0;
918 char *sep = "";
919 int i;
920
921 buf[pos] = 0;
922
923 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
924 struct keybinding *keybinding = &default_keybindings[i];
925
926 if (keybinding->request != request)
927 continue;
928
929 if (!string_format_from(buf, &pos, "%s%s", sep,
930 get_key_name(keybinding->alias)))
931 return "Too many keybindings!";
932 sep = ", ";
933 }
934
935 return buf;
936 }
937
938 struct run_request {
939 enum keymap keymap;
940 int key;
941 char cmd[SIZEOF_STR];
942 };
943
944 static struct run_request *run_request;
945 static size_t run_requests;
946
947 static enum request
948 add_run_request(enum keymap keymap, int key, int argc, char **argv)
949 {
950 struct run_request *tmp;
951 struct run_request req = { keymap, key };
952 size_t bufpos;
953
954 for (bufpos = 0; argc > 0; argc--, argv++)
955 if (!string_format_from(req.cmd, &bufpos, "%s ", *argv))
956 return REQ_NONE;
957
958 req.cmd[bufpos - 1] = 0;
959
960 tmp = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
961 if (!tmp)
962 return REQ_NONE;
963
964 run_request = tmp;
965 run_request[run_requests++] = req;
966
967 return REQ_NONE + run_requests;
968 }
969
970 static struct run_request *
971 get_run_request(enum request request)
972 {
973 if (request <= REQ_NONE)
974 return NULL;
975 return &run_request[request - REQ_NONE - 1];
976 }
977
978 static void
979 add_builtin_run_requests(void)
980 {
981 struct {
982 enum keymap keymap;
983 int key;
984 char *argv[1];
985 } reqs[] = {
986 { KEYMAP_MAIN, 'C', { "git cherry-pick %(commit)" } },
987 { KEYMAP_GENERIC, 'G', { "git gc" } },
988 };
989 int i;
990
991 for (i = 0; i < ARRAY_SIZE(reqs); i++) {
992 enum request req;
993
994 req = add_run_request(reqs[i].keymap, reqs[i].key, 1, reqs[i].argv);
995 if (req != REQ_NONE)
996 add_keybinding(reqs[i].keymap, req, reqs[i].key);
997 }
998 }
999
1000 /*
1001 * User config file handling.
1002 */
1003
1004 static struct int_map color_map[] = {
1005 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
1006 COLOR_MAP(DEFAULT),
1007 COLOR_MAP(BLACK),
1008 COLOR_MAP(BLUE),
1009 COLOR_MAP(CYAN),
1010 COLOR_MAP(GREEN),
1011 COLOR_MAP(MAGENTA),
1012 COLOR_MAP(RED),
1013 COLOR_MAP(WHITE),
1014 COLOR_MAP(YELLOW),
1015 };
1016
1017 #define set_color(color, name) \
1018 set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
1019
1020 static struct int_map attr_map[] = {
1021 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
1022 ATTR_MAP(NORMAL),
1023 ATTR_MAP(BLINK),
1024 ATTR_MAP(BOLD),
1025 ATTR_MAP(DIM),
1026 ATTR_MAP(REVERSE),
1027 ATTR_MAP(STANDOUT),
1028 ATTR_MAP(UNDERLINE),
1029 };
1030
1031 #define set_attribute(attr, name) \
1032 set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
1033
1034 static int config_lineno;
1035 static bool config_errors;
1036 static char *config_msg;
1037
1038 /* Wants: object fgcolor bgcolor [attr] */
1039 static int
1040 option_color_command(int argc, char *argv[])
1041 {
1042 struct line_info *info;
1043
1044 if (argc != 3 && argc != 4) {
1045 config_msg = "Wrong number of arguments given to color command";
1046 return ERR;
1047 }
1048
1049 info = get_line_info(argv[0], strlen(argv[0]));
1050 if (!info) {
1051 config_msg = "Unknown color name";
1052 return ERR;
1053 }
1054
1055 if (set_color(&info->fg, argv[1]) == ERR ||
1056 set_color(&info->bg, argv[2]) == ERR) {
1057 config_msg = "Unknown color";
1058 return ERR;
1059 }
1060
1061 if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
1062 config_msg = "Unknown attribute";
1063 return ERR;
1064 }
1065
1066 return OK;
1067 }
1068
1069 static bool parse_bool(const char *s)
1070 {
1071 return (!strcmp(s, "1") || !strcmp(s, "true") ||
1072 !strcmp(s, "yes")) ? TRUE : FALSE;
1073 }
1074
1075 /* Wants: name = value */
1076 static int
1077 option_set_command(int argc, char *argv[])
1078 {
1079 if (argc != 3) {
1080 config_msg = "Wrong number of arguments given to set command";
1081 return ERR;
1082 }
1083
1084 if (strcmp(argv[1], "=")) {
1085 config_msg = "No value assigned";
1086 return ERR;
1087 }
1088
1089 if (!strcmp(argv[0], "show-author")) {
1090 opt_author = parse_bool(argv[2]);
1091 return OK;
1092 }
1093
1094 if (!strcmp(argv[0], "show-date")) {
1095 opt_date = parse_bool(argv[2]);
1096 return OK;
1097 }
1098
1099 if (!strcmp(argv[0], "show-rev-graph")) {
1100 opt_rev_graph = parse_bool(argv[2]);
1101 return OK;
1102 }
1103
1104 if (!strcmp(argv[0], "show-refs")) {
1105 opt_show_refs = parse_bool(argv[2]);
1106 return OK;
1107 }
1108
1109 if (!strcmp(argv[0], "show-line-numbers")) {
1110 opt_line_number = parse_bool(argv[2]);
1111 return OK;
1112 }
1113
1114 if (!strcmp(argv[0], "line-number-interval")) {
1115 opt_num_interval = atoi(argv[2]);
1116 return OK;
1117 }
1118
1119 if (!strcmp(argv[0], "tab-size")) {
1120 opt_tab_size = atoi(argv[2]);
1121 return OK;
1122 }
1123
1124 if (!strcmp(argv[0], "commit-encoding")) {
1125 char *arg = argv[2];
1126 int delimiter = *arg;
1127 int i;
1128
1129 switch (delimiter) {
1130 case '"':
1131 case '\'':
1132 for (arg++, i = 0; arg[i]; i++)
1133 if (arg[i] == delimiter) {
1134 arg[i] = 0;
1135 break;
1136 }
1137 default:
1138 string_ncopy(opt_encoding, arg, strlen(arg));
1139 return OK;
1140 }
1141 }
1142
1143 config_msg = "Unknown variable name";
1144 return ERR;
1145 }
1146
1147 /* Wants: mode request key */
1148 static int
1149 option_bind_command(int argc, char *argv[])
1150 {
1151 enum request request;
1152 int keymap;
1153 int key;
1154
1155 if (argc < 3) {
1156 config_msg = "Wrong number of arguments given to bind command";
1157 return ERR;
1158 }
1159
1160 if (set_keymap(&keymap, argv[0]) == ERR) {
1161 config_msg = "Unknown key map";
1162 return ERR;
1163 }
1164
1165 key = get_key_value(argv[1]);
1166 if (key == ERR) {
1167 config_msg = "Unknown key";
1168 return ERR;
1169 }
1170
1171 request = get_request(argv[2]);
1172 if (request == REQ_NONE) {
1173 const char *obsolete[] = { "cherry-pick" };
1174 size_t namelen = strlen(argv[2]);
1175 int i;
1176
1177 for (i = 0; i < ARRAY_SIZE(obsolete); i++) {
1178 if (namelen == strlen(obsolete[i]) &&
1179 !string_enum_compare(obsolete[i], argv[2], namelen)) {
1180 config_msg = "Obsolete request name";
1181 return ERR;
1182 }
1183 }
1184 }
1185 if (request == REQ_NONE && *argv[2]++ == '!')
1186 request = add_run_request(keymap, key, argc - 2, argv + 2);
1187 if (request == REQ_NONE) {
1188 config_msg = "Unknown request name";
1189 return ERR;
1190 }
1191
1192 add_keybinding(keymap, request, key);
1193
1194 return OK;
1195 }
1196
1197 static int
1198 set_option(char *opt, char *value)
1199 {
1200 char *argv[16];
1201 int valuelen;
1202 int argc = 0;
1203
1204 /* Tokenize */
1205 while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1206 argv[argc++] = value;
1207 value += valuelen;
1208
1209 /* Nothing more to tokenize or last available token. */
1210 if (!*value || argc >= ARRAY_SIZE(argv))
1211 break;
1212
1213 *value++ = 0;
1214 while (isspace(*value))
1215 value++;
1216 }
1217
1218 if (!strcmp(opt, "color"))
1219 return option_color_command(argc, argv);
1220
1221 if (!strcmp(opt, "set"))
1222 return option_set_command(argc, argv);
1223
1224 if (!strcmp(opt, "bind"))
1225 return option_bind_command(argc, argv);
1226
1227 config_msg = "Unknown option command";
1228 return ERR;
1229 }
1230
1231 static int
1232 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1233 {
1234 int status = OK;
1235
1236 config_lineno++;
1237 config_msg = "Internal error";
1238
1239 /* Check for comment markers, since read_properties() will
1240 * only ensure opt and value are split at first " \t". */
1241 optlen = strcspn(opt, "#");
1242 if (optlen == 0)
1243 return OK;
1244
1245 if (opt[optlen] != 0) {
1246 config_msg = "No option value";
1247 status = ERR;
1248
1249 } else {
1250 /* Look for comment endings in the value. */
1251 size_t len = strcspn(value, "#");
1252
1253 if (len < valuelen) {
1254 valuelen = len;
1255 value[valuelen] = 0;
1256 }
1257
1258 status = set_option(opt, value);
1259 }
1260
1261 if (status == ERR) {
1262 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1263 config_lineno, (int) optlen, opt, config_msg);
1264 config_errors = TRUE;
1265 }
1266
1267 /* Always keep going if errors are encountered. */
1268 return OK;
1269 }
1270
1271 static void
1272 load_option_file(const char *path)
1273 {
1274 FILE *file;
1275
1276 /* It's ok that the file doesn't exist. */
1277 file = fopen(path, "r");
1278 if (!file)
1279 return;
1280
1281 config_lineno = 0;
1282 config_errors = FALSE;
1283
1284 if (read_properties(file, " \t", read_option) == ERR ||
1285 config_errors == TRUE)
1286 fprintf(stderr, "Errors while loading %s.\n", path);
1287 }
1288
1289 static int
1290 load_options(void)
1291 {
1292 char *home = getenv("HOME");
1293 char *tigrc_user = getenv("TIGRC_USER");
1294 char *tigrc_system = getenv("TIGRC_SYSTEM");
1295 char buf[SIZEOF_STR];
1296
1297 add_builtin_run_requests();
1298
1299 if (!tigrc_system) {
1300 if (!string_format(buf, "%s/tigrc", SYSCONFDIR))
1301 return ERR;
1302 tigrc_system = buf;
1303 }
1304 load_option_file(tigrc_system);
1305
1306 if (!tigrc_user) {
1307 if (!home || !string_format(buf, "%s/.tigrc", home))
1308 return ERR;
1309 tigrc_user = buf;
1310 }
1311 load_option_file(tigrc_user);
1312
1313 return OK;
1314 }
1315
1316
1317 /*
1318 * The viewer
1319 */
1320
1321 struct view;
1322 struct view_ops;
1323
1324 /* The display array of active views and the index of the current view. */
1325 static struct view *display[2];
1326 static unsigned int current_view;
1327
1328 /* Reading from the prompt? */
1329 static bool input_mode = FALSE;
1330
1331 #define foreach_displayed_view(view, i) \
1332 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1333
1334 #define displayed_views() (display[1] != NULL ? 2 : 1)
1335
1336 /* Current head and commit ID */
1337 static char ref_blob[SIZEOF_REF] = "";
1338 static char ref_commit[SIZEOF_REF] = "HEAD";
1339 static char ref_head[SIZEOF_REF] = "HEAD";
1340
1341 struct view {
1342 const char *name; /* View name */
1343 const char *cmd_fmt; /* Default command line format */
1344 const char *cmd_env; /* Command line set via environment */
1345 const char *id; /* Points to either of ref_{head,commit,blob} */
1346
1347 struct view_ops *ops; /* View operations */
1348
1349 enum keymap keymap; /* What keymap does this view have */
1350
1351 char cmd[SIZEOF_STR]; /* Command buffer */
1352 char ref[SIZEOF_REF]; /* Hovered commit reference */
1353 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
1354
1355 int height, width; /* The width and height of the main window */
1356 WINDOW *win; /* The main window */
1357 WINDOW *title; /* The title window living below the main window */
1358
1359 /* Navigation */
1360 unsigned long offset; /* Offset of the window top */
1361 unsigned long lineno; /* Current line number */
1362
1363 /* Searching */
1364 char grep[SIZEOF_STR]; /* Search string */
1365 regex_t *regex; /* Pre-compiled regex */
1366
1367 /* If non-NULL, points to the view that opened this view. If this view
1368 * is closed tig will switch back to the parent view. */
1369 struct view *parent;
1370
1371 /* Buffering */
1372 size_t lines; /* Total number of lines */
1373 struct line *line; /* Line index */
1374 size_t line_alloc; /* Total number of allocated lines */
1375 size_t line_size; /* Total number of used lines */
1376 unsigned int digits; /* Number of digits in the lines member. */
1377
1378 /* Loading */
1379 FILE *pipe;
1380 time_t start_time;
1381 };
1382
1383 struct view_ops {
1384 /* What type of content being displayed. Used in the title bar. */
1385 const char *type;
1386 /* Open and reads in all view content. */
1387 bool (*open)(struct view *view);
1388 /* Read one line; updates view->line. */
1389 bool (*read)(struct view *view, char *data);
1390 /* Draw one line; @lineno must be < view->height. */
1391 bool (*draw)(struct view *view, struct line *line, unsigned int lineno, bool selected);
1392 /* Depending on view handle a special requests. */
1393 enum request (*request)(struct view *view, enum request request, struct line *line);
1394 /* Search for regex in a line. */
1395 bool (*grep)(struct view *view, struct line *line);
1396 /* Select line */
1397 void (*select)(struct view *view, struct line *line);
1398 };
1399
1400 static struct view_ops pager_ops;
1401 static struct view_ops main_ops;
1402 static struct view_ops tree_ops;
1403 static struct view_ops blob_ops;
1404 static struct view_ops blame_ops;
1405 static struct view_ops help_ops;
1406 static struct view_ops status_ops;
1407 static struct view_ops stage_ops;
1408
1409 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1410 { name, cmd, #env, ref, ops, map}
1411
1412 #define VIEW_(id, name, ops, ref) \
1413 VIEW_STR(name, TIG_##id##_CMD, TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1414
1415
1416 static struct view views[] = {
1417 VIEW_(MAIN, "main", &main_ops, ref_head),
1418 VIEW_(DIFF, "diff", &pager_ops, ref_commit),
1419 VIEW_(LOG, "log", &pager_ops, ref_head),
1420 VIEW_(TREE, "tree", &tree_ops, ref_commit),
1421 VIEW_(BLOB, "blob", &blob_ops, ref_blob),
1422 VIEW_(BLAME, "blame", &blame_ops, ref_commit),
1423 VIEW_(HELP, "help", &help_ops, ""),
1424 VIEW_(PAGER, "pager", &pager_ops, "stdin"),
1425 VIEW_(STATUS, "status", &status_ops, ""),
1426 VIEW_(STAGE, "stage", &stage_ops, ""),
1427 };
1428
1429 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1430
1431 #define foreach_view(view, i) \
1432 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1433
1434 #define view_is_displayed(view) \
1435 (view == display[0] || view == display[1])
1436
1437 static int
1438 draw_text(struct view *view, const char *string, int max_len,
1439 bool use_tilde, int tilde_attr)
1440 {
1441 int len = 0;
1442 int trimmed = FALSE;
1443
1444 if (max_len <= 0)
1445 return 0;
1446
1447 if (opt_utf8) {
1448 len = utf8_length(string, max_len, &trimmed, use_tilde);
1449 } else {
1450 len = strlen(string);
1451 if (len > max_len) {
1452 if (use_tilde) {
1453 max_len -= 1;
1454 }
1455 len = max_len;
1456 trimmed = TRUE;
1457 }
1458 }
1459
1460 waddnstr(view->win, string, len);
1461 if (trimmed && use_tilde) {
1462 if (tilde_attr != -1)
1463 wattrset(view->win, tilde_attr);
1464 waddch(view->win, '~');
1465 len++;
1466 }
1467
1468 return len;
1469 }
1470
1471 static bool
1472 draw_view_line(struct view *view, unsigned int lineno)
1473 {
1474 struct line *line;
1475 bool selected = (view->offset + lineno == view->lineno);
1476 bool draw_ok;
1477
1478 assert(view_is_displayed(view));
1479
1480 if (view->offset + lineno >= view->lines)
1481 return FALSE;
1482
1483 line = &view->line[view->offset + lineno];
1484
1485 if (selected) {
1486 line->selected = TRUE;
1487 view->ops->select(view, line);
1488 } else if (line->selected) {
1489 line->selected = FALSE;
1490 wmove(view->win, lineno, 0);
1491 wclrtoeol(view->win);
1492 }
1493
1494 scrollok(view->win, FALSE);
1495 draw_ok = view->ops->draw(view, line, lineno, selected);
1496 scrollok(view->win, TRUE);
1497
1498 return draw_ok;
1499 }
1500
1501 static void
1502 redraw_view_dirty(struct view *view)
1503 {
1504 bool dirty = FALSE;
1505 int lineno;
1506
1507 for (lineno = 0; lineno < view->height; lineno++) {
1508 struct line *line = &view->line[view->offset + lineno];
1509
1510 if (!line->dirty)
1511 continue;
1512 line->dirty = 0;
1513 dirty = TRUE;
1514 if (!draw_view_line(view, lineno))
1515 break;
1516 }
1517
1518 if (!dirty)
1519 return;
1520 redrawwin(view->win);
1521 if (input_mode)
1522 wnoutrefresh(view->win);
1523 else
1524 wrefresh(view->win);
1525 }
1526
1527 static void
1528 redraw_view_from(struct view *view, int lineno)
1529 {
1530 assert(0 <= lineno && lineno < view->height);
1531
1532 for (; lineno < view->height; lineno++) {
1533 if (!draw_view_line(view, lineno))
1534 break;
1535 }
1536
1537 redrawwin(view->win);
1538 if (input_mode)
1539 wnoutrefresh(view->win);
1540 else
1541 wrefresh(view->win);
1542 }
1543
1544 static void
1545 redraw_view(struct view *view)
1546 {
1547 wclear(view->win);
1548 redraw_view_from(view, 0);
1549 }
1550
1551
1552 static void
1553 update_view_title(struct view *view)
1554 {
1555 char buf[SIZEOF_STR];
1556 char state[SIZEOF_STR];
1557 size_t bufpos = 0, statelen = 0;
1558
1559 assert(view_is_displayed(view));
1560
1561 if (view != VIEW(REQ_VIEW_STATUS) && (view->lines || view->pipe)) {
1562 unsigned int view_lines = view->offset + view->height;
1563 unsigned int lines = view->lines
1564 ? MIN(view_lines, view->lines) * 100 / view->lines
1565 : 0;
1566
1567 string_format_from(state, &statelen, "- %s %d of %d (%d%%)",
1568 view->ops->type,
1569 view->lineno + 1,
1570 view->lines,
1571 lines);
1572
1573 if (view->pipe) {
1574 time_t secs = time(NULL) - view->start_time;
1575
1576 /* Three git seconds are a long time ... */
1577 if (secs > 2)
1578 string_format_from(state, &statelen, " %lds", secs);
1579 }
1580 }
1581
1582 string_format_from(buf, &bufpos, "[%s]", view->name);
1583 if (*view->ref && bufpos < view->width) {
1584 size_t refsize = strlen(view->ref);
1585 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
1586
1587 if (minsize < view->width)
1588 refsize = view->width - minsize + 7;
1589 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
1590 }
1591
1592 if (statelen && bufpos < view->width) {
1593 string_format_from(buf, &bufpos, " %s", state);
1594 }
1595
1596 if (view == display[current_view])
1597 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1598 else
1599 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1600
1601 mvwaddnstr(view->title, 0, 0, buf, bufpos);
1602 wclrtoeol(view->title);
1603 wmove(view->title, 0, view->width - 1);
1604
1605 if (input_mode)
1606 wnoutrefresh(view->title);
1607 else
1608 wrefresh(view->title);
1609 }
1610
1611 static void
1612 resize_display(void)
1613 {
1614 int offset, i;
1615 struct view *base = display[0];
1616 struct view *view = display[1] ? display[1] : display[0];
1617
1618 /* Setup window dimensions */
1619
1620 getmaxyx(stdscr, base->height, base->width);
1621
1622 /* Make room for the status window. */
1623 base->height -= 1;
1624
1625 if (view != base) {
1626 /* Horizontal split. */
1627 view->width = base->width;
1628 view->height = SCALE_SPLIT_VIEW(base->height);
1629 base->height -= view->height;
1630
1631 /* Make room for the title bar. */
1632 view->height -= 1;
1633 }
1634
1635 /* Make room for the title bar. */
1636 base->height -= 1;
1637
1638 offset = 0;
1639
1640 foreach_displayed_view (view, i) {
1641 if (!view->win) {
1642 view->win = newwin(view->height, 0, offset, 0);
1643 if (!view->win)
1644 die("Failed to create %s view", view->name);
1645
1646 scrollok(view->win, TRUE);
1647
1648 view->title = newwin(1, 0, offset + view->height, 0);
1649 if (!view->title)
1650 die("Failed to create title window");
1651
1652 } else {
1653 wresize(view->win, view->height, view->width);
1654 mvwin(view->win, offset, 0);
1655 mvwin(view->title, offset + view->height, 0);
1656 }
1657
1658 offset += view->height + 1;
1659 }
1660 }
1661
1662 static void
1663 redraw_display(void)
1664 {
1665 struct view *view;
1666 int i;
1667
1668 foreach_displayed_view (view, i) {
1669 redraw_view(view);
1670 update_view_title(view);
1671 }
1672 }
1673
1674 static void
1675 update_display_cursor(struct view *view)
1676 {
1677 /* Move the cursor to the right-most column of the cursor line.
1678 *
1679 * XXX: This could turn out to be a bit expensive, but it ensures that
1680 * the cursor does not jump around. */
1681 if (view->lines) {
1682 wmove(view->win, view->lineno - view->offset, view->width - 1);
1683 wrefresh(view->win);
1684 }
1685 }
1686
1687 /*
1688 * Navigation
1689 */
1690
1691 /* Scrolling backend */
1692 static void
1693 do_scroll_view(struct view *view, int lines)
1694 {
1695 bool redraw_current_line = FALSE;
1696
1697 /* The rendering expects the new offset. */
1698 view->offset += lines;
1699
1700 assert(0 <= view->offset && view->offset < view->lines);
1701 assert(lines);
1702
1703 /* Move current line into the view. */
1704 if (view->lineno < view->offset) {
1705 view->lineno = view->offset;
1706 redraw_current_line = TRUE;
1707 } else if (view->lineno >= view->offset + view->height) {
1708 view->lineno = view->offset + view->height - 1;
1709 redraw_current_line = TRUE;
1710 }
1711
1712 assert(view->offset <= view->lineno && view->lineno < view->lines);
1713
1714 /* Redraw the whole screen if scrolling is pointless. */
1715 if (view->height < ABS(lines)) {
1716 redraw_view(view);
1717
1718 } else {
1719 int line = lines > 0 ? view->height - lines : 0;
1720 int end = line + ABS(lines);
1721
1722 wscrl(view->win, lines);
1723
1724 for (; line < end; line++) {
1725 if (!draw_view_line(view, line))
1726 break;
1727 }
1728
1729 if (redraw_current_line)
1730 draw_view_line(view, view->lineno - view->offset);
1731 }
1732
1733 redrawwin(view->win);
1734 wrefresh(view->win);
1735 report("");
1736 }
1737
1738 /* Scroll frontend */
1739 static void
1740 scroll_view(struct view *view, enum request request)
1741 {
1742 int lines = 1;
1743
1744 assert(view_is_displayed(view));
1745
1746 switch (request) {
1747 case REQ_SCROLL_PAGE_DOWN:
1748 lines = view->height;
1749 case REQ_SCROLL_LINE_DOWN:
1750 if (view->offset + lines > view->lines)
1751 lines = view->lines - view->offset;
1752
1753 if (lines == 0 || view->offset + view->height >= view->lines) {
1754 report("Cannot scroll beyond the last line");
1755 return;
1756 }
1757 break;
1758
1759 case REQ_SCROLL_PAGE_UP:
1760 lines = view->height;
1761 case REQ_SCROLL_LINE_UP:
1762 if (lines > view->offset)
1763 lines = view->offset;
1764
1765 if (lines == 0) {
1766 report("Cannot scroll beyond the first line");
1767 return;
1768 }
1769
1770 lines = -lines;
1771 break;
1772
1773 default:
1774 die("request %d not handled in switch", request);
1775 }
1776
1777 do_scroll_view(view, lines);
1778 }
1779
1780 /* Cursor moving */
1781 static void
1782 move_view(struct view *view, enum request request)
1783 {
1784 int scroll_steps = 0;
1785 int steps;
1786
1787 switch (request) {
1788 case REQ_MOVE_FIRST_LINE:
1789 steps = -view->lineno;
1790 break;
1791
1792 case REQ_MOVE_LAST_LINE:
1793 steps = view->lines - view->lineno - 1;
1794 break;
1795
1796 case REQ_MOVE_PAGE_UP:
1797 steps = view->height > view->lineno
1798 ? -view->lineno : -view->height;
1799 break;
1800
1801 case REQ_MOVE_PAGE_DOWN:
1802 steps = view->lineno + view->height >= view->lines
1803 ? view->lines - view->lineno - 1 : view->height;
1804 break;
1805
1806 case REQ_MOVE_UP:
1807 steps = -1;
1808 break;
1809
1810 case REQ_MOVE_DOWN:
1811 steps = 1;
1812 break;
1813
1814 default:
1815 die("request %d not handled in switch", request);
1816 }
1817
1818 if (steps <= 0 && view->lineno == 0) {
1819 report("Cannot move beyond the first line");
1820 return;
1821
1822 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1823 report("Cannot move beyond the last line");
1824 return;
1825 }
1826
1827 /* Move the current line */
1828 view->lineno += steps;
1829 assert(0 <= view->lineno && view->lineno < view->lines);
1830
1831 /* Check whether the view needs to be scrolled */
1832 if (view->lineno < view->offset ||
1833 view->lineno >= view->offset + view->height) {
1834 scroll_steps = steps;
1835 if (steps < 0 && -steps > view->offset) {
1836 scroll_steps = -view->offset;
1837
1838 } else if (steps > 0) {
1839 if (view->lineno == view->lines - 1 &&
1840 view->lines > view->height) {
1841 scroll_steps = view->lines - view->offset - 1;
1842 if (scroll_steps >= view->height)
1843 scroll_steps -= view->height - 1;
1844 }
1845 }
1846 }
1847
1848 if (!view_is_displayed(view)) {
1849 view->offset += scroll_steps;
1850 assert(0 <= view->offset && view->offset < view->lines);
1851 view->ops->select(view, &view->line[view->lineno]);
1852 return;
1853 }
1854
1855 /* Repaint the old "current" line if we be scrolling */
1856 if (ABS(steps) < view->height)
1857 draw_view_line(view, view->lineno - steps - view->offset);
1858
1859 if (scroll_steps) {
1860 do_scroll_view(view, scroll_steps);
1861 return;
1862 }
1863
1864 /* Draw the current line */
1865 draw_view_line(view, view->lineno - view->offset);
1866
1867 redrawwin(view->win);
1868 wrefresh(view->win);
1869 report("");
1870 }
1871
1872
1873 /*
1874 * Searching
1875 */
1876
1877 static void search_view(struct view *view, enum request request);
1878
1879 static bool
1880 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1881 {
1882 assert(view_is_displayed(view));
1883
1884 if (!view->ops->grep(view, line))
1885 return FALSE;
1886
1887 if (lineno - view->offset >= view->height) {
1888 view->offset = lineno;
1889 view->lineno = lineno;
1890 redraw_view(view);
1891
1892 } else {
1893 unsigned long old_lineno = view->lineno - view->offset;
1894
1895 view->lineno = lineno;
1896 draw_view_line(view, old_lineno);
1897
1898 draw_view_line(view, view->lineno - view->offset);
1899 redrawwin(view->win);
1900 wrefresh(view->win);
1901 }
1902
1903 report("Line %ld matches '%s'", lineno + 1, view->grep);
1904 return TRUE;
1905 }
1906
1907 static void
1908 find_next(struct view *view, enum request request)
1909 {
1910 unsigned long lineno = view->lineno;
1911 int direction;
1912
1913 if (!*view->grep) {
1914 if (!*opt_search)
1915 report("No previous search");
1916 else
1917 search_view(view, request);
1918 return;
1919 }
1920
1921 switch (request) {
1922 case REQ_SEARCH:
1923 case REQ_FIND_NEXT:
1924 direction = 1;
1925 break;
1926
1927 case REQ_SEARCH_BACK:
1928 case REQ_FIND_PREV:
1929 direction = -1;
1930 break;
1931
1932 default:
1933 return;
1934 }
1935
1936 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1937 lineno += direction;
1938
1939 /* Note, lineno is unsigned long so will wrap around in which case it
1940 * will become bigger than view->lines. */
1941 for (; lineno < view->lines; lineno += direction) {
1942 struct line *line = &view->line[lineno];
1943
1944 if (find_next_line(view, lineno, line))
1945 return;
1946 }
1947
1948 report("No match found for '%s'", view->grep);
1949 }
1950
1951 static void
1952 search_view(struct view *view, enum request request)
1953 {
1954 int regex_err;
1955
1956 if (view->regex) {
1957 regfree(view->regex);
1958 *view->grep = 0;
1959 } else {
1960 view->regex = calloc(1, sizeof(*view->regex));
1961 if (!view->regex)
1962 return;
1963 }
1964
1965 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
1966 if (regex_err != 0) {
1967 char buf[SIZEOF_STR] = "unknown error";
1968
1969 regerror(regex_err, view->regex, buf, sizeof(buf));
1970 report("Search failed: %s", buf);
1971 return;
1972 }
1973
1974 string_copy(view->grep, opt_search);
1975
1976 find_next(view, request);
1977 }
1978
1979 /*
1980 * Incremental updating
1981 */
1982
1983 static void
1984 end_update(struct view *view)
1985 {
1986 if (!view->pipe)
1987 return;
1988 set_nonblocking_input(FALSE);
1989 if (view->pipe == stdin)
1990 fclose(view->pipe);
1991 else
1992 pclose(view->pipe);
1993 view->pipe = NULL;
1994 }
1995
1996 static bool
1997 begin_update(struct view *view)
1998 {
1999 if (view->pipe)
2000 end_update(view);
2001
2002 if (opt_cmd[0]) {
2003 string_copy(view->cmd, opt_cmd);
2004 opt_cmd[0] = 0;
2005 /* When running random commands, initially show the
2006 * command in the title. However, it maybe later be
2007 * overwritten if a commit line is selected. */
2008 if (view == VIEW(REQ_VIEW_PAGER))
2009 string_copy(view->ref, view->cmd);
2010 else
2011 view->ref[0] = 0;
2012
2013 } else if (view == VIEW(REQ_VIEW_TREE)) {
2014 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2015 char path[SIZEOF_STR];
2016
2017 if (strcmp(view->vid, view->id))
2018 opt_path[0] = path[0] = 0;
2019 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
2020 return FALSE;
2021
2022 if (!string_format(view->cmd, format, view->id, path))
2023 return FALSE;
2024
2025 } else {
2026 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
2027 const char *id = view->id;
2028
2029 if (!string_format(view->cmd, format, id, id, id, id, id))
2030 return FALSE;
2031
2032 /* Put the current ref_* value to the view title ref
2033 * member. This is needed by the blob view. Most other
2034 * views sets it automatically after loading because the
2035 * first line is a commit line. */
2036 string_copy_rev(view->ref, view->id);
2037 }
2038
2039 /* Special case for the pager view. */
2040 if (opt_pipe) {
2041 view->pipe = opt_pipe;
2042 opt_pipe = NULL;
2043 } else {
2044 view->pipe = popen(view->cmd, "r");
2045 }
2046
2047 if (!view->pipe)
2048 return FALSE;
2049
2050 set_nonblocking_input(TRUE);
2051
2052 view->offset = 0;
2053 view->lines = 0;
2054 view->lineno = 0;
2055 string_copy_rev(view->vid, view->id);
2056
2057 if (view->line) {
2058 int i;
2059
2060 for (i = 0; i < view->lines; i++)
2061 if (view->line[i].data)
2062 free(view->line[i].data);
2063
2064 free(view->line);
2065 view->line = NULL;
2066 }
2067
2068 view->start_time = time(NULL);
2069
2070 return TRUE;
2071 }
2072
2073 #define ITEM_CHUNK_SIZE 256
2074 static void *
2075 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2076 {
2077 size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2078 size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2079
2080 if (mem == NULL || num_chunks != num_chunks_new) {
2081 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2082 mem = realloc(mem, *size * item_size);
2083 }
2084
2085 return mem;
2086 }
2087
2088 static struct line *
2089 realloc_lines(struct view *view, size_t line_size)
2090 {
2091 size_t alloc = view->line_alloc;
2092 struct line *tmp = realloc_items(view->line, &alloc, line_size,
2093 sizeof(*view->line));
2094
2095 if (!tmp)
2096 return NULL;
2097
2098 view->line = tmp;
2099 view->line_alloc = alloc;
2100 view->line_size = line_size;
2101 return view->line;
2102 }
2103
2104 static bool
2105 update_view(struct view *view)
2106 {
2107 char in_buffer[BUFSIZ];
2108 char out_buffer[BUFSIZ * 2];
2109 char *line;
2110 /* The number of lines to read. If too low it will cause too much
2111 * redrawing (and possible flickering), if too high responsiveness
2112 * will suffer. */
2113 unsigned long lines = view->height;
2114 int redraw_from = -1;
2115
2116 if (!view->pipe)
2117 return TRUE;
2118
2119 /* Only redraw if lines are visible. */
2120 if (view->offset + view->height >= view->lines)
2121 redraw_from = view->lines - view->offset;
2122
2123 /* FIXME: This is probably not perfect for backgrounded views. */
2124 if (!realloc_lines(view, view->lines + lines))
2125 goto alloc_error;
2126
2127 while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
2128 size_t linelen = strlen(line);
2129
2130 if (linelen)
2131 line[linelen - 1] = 0;
2132
2133 if (opt_iconv != ICONV_NONE) {
2134 ICONV_CONST char *inbuf = line;
2135 size_t inlen = linelen;
2136
2137 char *outbuf = out_buffer;
2138 size_t outlen = sizeof(out_buffer);
2139
2140 size_t ret;
2141
2142 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2143 if (ret != (size_t) -1) {
2144 line = out_buffer;
2145 linelen = strlen(out_buffer);
2146 }
2147 }
2148
2149 if (!view->ops->read(view, line))
2150 goto alloc_error;
2151
2152 if (lines-- == 1)
2153 break;
2154 }
2155
2156 {
2157 int digits;
2158
2159 lines = view->lines;
2160 for (digits = 0; lines; digits++)
2161 lines /= 10;
2162
2163 /* Keep the displayed view in sync with line number scaling. */
2164 if (digits != view->digits) {
2165 view->digits = digits;
2166 redraw_from = 0;
2167 }
2168 }
2169
2170 if (!view_is_displayed(view))
2171 goto check_pipe;
2172
2173 if (view == VIEW(REQ_VIEW_TREE)) {
2174 /* Clear the view and redraw everything since the tree sorting
2175 * might have rearranged things. */
2176 redraw_view(view);
2177
2178
2179 } else if (redraw_from >= 0) {
2180 /* If this is an incremental update, redraw the previous line
2181 * since for commits some members could have changed when
2182 * loading the main view. */
2183 if (redraw_from > 0)
2184 redraw_from--;
2185
2186 /* Since revision graph visualization requires knowledge
2187 * about the parent commit, it causes a further one-off
2188 * needed to be redrawn for incremental updates. */
2189 if (redraw_from > 0 && opt_rev_graph)
2190 redraw_from--;
2191
2192 /* Incrementally draw avoids flickering. */
2193 redraw_view_from(view, redraw_from);
2194 }
2195
2196 if (view == VIEW(REQ_VIEW_BLAME))
2197 redraw_view_dirty(view);
2198
2199 /* Update the title _after_ the redraw so that if the redraw picks up a
2200 * commit reference in view->ref it'll be available here. */
2201 update_view_title(view);
2202
2203 check_pipe:
2204 if (ferror(view->pipe)) {
2205 report("Failed to read: %s", strerror(errno));
2206 goto end;
2207
2208 } else if (feof(view->pipe)) {
2209 report("");
2210 goto end;
2211 }
2212
2213 return TRUE;
2214
2215 alloc_error:
2216 report("Allocation failure");
2217
2218 end:
2219 if (view->ops->read(view, NULL))
2220 end_update(view);
2221 return FALSE;
2222 }
2223
2224 static struct line *
2225 add_line_data(struct view *view, void *data, enum line_type type)
2226 {
2227 struct line *line = &view->line[view->lines++];
2228
2229 memset(line, 0, sizeof(*line));
2230 line->type = type;
2231 line->data = data;
2232
2233 return line;
2234 }
2235
2236 static struct line *
2237 add_line_text(struct view *view, char *data, enum line_type type)
2238 {
2239 if (data)
2240 data = strdup(data);
2241
2242 return data ? add_line_data(view, data, type) : NULL;
2243 }
2244
2245
2246 /*
2247 * View opening
2248 */
2249
2250 enum open_flags {
2251 OPEN_DEFAULT = 0, /* Use default view switching. */
2252 OPEN_SPLIT = 1, /* Split current view. */
2253 OPEN_BACKGROUNDED = 2, /* Backgrounded. */
2254 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
2255 };
2256
2257 static void
2258 open_view(struct view *prev, enum request request, enum open_flags flags)
2259 {
2260 bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2261 bool split = !!(flags & OPEN_SPLIT);
2262 bool reload = !!(flags & OPEN_RELOAD);
2263 struct view *view = VIEW(request);
2264 int nviews = displayed_views();
2265 struct view *base_view = display[0];
2266
2267 if (view == prev && nviews == 1 && !reload) {
2268 report("Already in %s view", view->name);
2269 return;
2270 }
2271
2272 if (view->ops->open) {
2273 if (!view->ops->open(view)) {
2274 report("Failed to load %s view", view->name);
2275 return;
2276 }
2277
2278 } else if ((reload || strcmp(view->vid, view->id)) &&
2279 !begin_update(view)) {
2280 report("Failed to load %s view", view->name);
2281 return;
2282 }
2283
2284 if (split) {
2285 display[1] = view;
2286 if (!backgrounded)
2287 current_view = 1;
2288 } else {
2289 /* Maximize the current view. */
2290 memset(display, 0, sizeof(display));
2291 current_view = 0;
2292 display[current_view] = view;
2293 }
2294
2295 /* Resize the view when switching between split- and full-screen,
2296 * or when switching between two different full-screen views. */
2297 if (nviews != displayed_views() ||
2298 (nviews == 1 && base_view != display[0]))
2299 resize_display();
2300
2301 if (split && prev->lineno - prev->offset >= prev->height) {
2302 /* Take the title line into account. */
2303 int lines = prev->lineno - prev->offset - prev->height + 1;
2304
2305 /* Scroll the view that was split if the current line is
2306 * outside the new limited view. */
2307 do_scroll_view(prev, lines);
2308 }
2309
2310 if (prev && view != prev) {
2311 if (split && !backgrounded) {
2312 /* "Blur" the previous view. */
2313 update_view_title(prev);
2314 }
2315
2316 view->parent = prev;
2317 }
2318
2319 if (view->pipe && view->lines == 0) {
2320 /* Clear the old view and let the incremental updating refill
2321 * the screen. */
2322 wclear(view->win);
2323 report("");
2324 } else {
2325 redraw_view(view);
2326 report("");
2327 }
2328
2329 /* If the view is backgrounded the above calls to report()
2330 * won't redraw the view title. */
2331 if (backgrounded)
2332 update_view_title(view);
2333 }
2334
2335 static void
2336 open_external_viewer(const char *cmd)
2337 {
2338 def_prog_mode(); /* save current tty modes */
2339 endwin(); /* restore original tty modes */
2340 system(cmd);
2341 fprintf(stderr, "Press Enter to continue");
2342 getc(stdin);
2343 reset_prog_mode();
2344 redraw_display();
2345 }
2346
2347 static void
2348 open_mergetool(const char *file)
2349 {
2350 char cmd[SIZEOF_STR];
2351 char file_sq[SIZEOF_STR];
2352
2353 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2354 string_format(cmd, "git mergetool %s", file_sq)) {
2355 open_external_viewer(cmd);
2356 }
2357 }
2358
2359 static void
2360 open_editor(bool from_root, const char *file)
2361 {
2362 char cmd[SIZEOF_STR];
2363 char file_sq[SIZEOF_STR];
2364 char *editor;
2365 char *prefix = from_root ? opt_cdup : "";
2366
2367 editor = getenv("GIT_EDITOR");
2368 if (!editor && *opt_editor)
2369 editor = opt_editor;
2370 if (!editor)
2371 editor = getenv("VISUAL");
2372 if (!editor)
2373 editor = getenv("EDITOR");
2374 if (!editor)
2375 editor = "vi";
2376
2377 if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2378 string_format(cmd, "%s %s%s", editor, prefix, file_sq)) {
2379 open_external_viewer(cmd);
2380 }
2381 }
2382
2383 static void
2384 open_run_request(enum request request)
2385 {
2386 struct run_request *req = get_run_request(request);
2387 char buf[SIZEOF_STR * 2];
2388 size_t bufpos;
2389 char *cmd;
2390
2391 if (!req) {
2392 report("Unknown run request");
2393 return;
2394 }
2395
2396 bufpos = 0;
2397 cmd = req->cmd;
2398
2399 while (cmd) {
2400 char *next = strstr(cmd, "%(");
2401 int len = next - cmd;
2402 char *value;
2403
2404 if (!next) {
2405 len = strlen(cmd);
2406 value = "";
2407
2408 } else if (!strncmp(next, "%(head)", 7)) {
2409 value = ref_head;
2410
2411 } else if (!strncmp(next, "%(commit)", 9)) {
2412 value = ref_commit;
2413
2414 } else if (!strncmp(next, "%(blob)", 7)) {
2415 value = ref_blob;
2416
2417 } else {
2418 report("Unknown replacement in run request: `%s`", req->cmd);
2419 return;
2420 }
2421
2422 if (!string_format_from(buf, &bufpos, "%.*s%s", len, cmd, value))
2423 return;
2424
2425 if (next)
2426 next = strchr(next, ')') + 1;
2427 cmd = next;
2428 }
2429
2430 open_external_viewer(buf);
2431 }
2432
2433 /*
2434 * User request switch noodle
2435 */
2436
2437 static int
2438 view_driver(struct view *view, enum request request)
2439 {
2440 int i;
2441
2442 if (request == REQ_NONE) {
2443 doupdate();
2444 return TRUE;
2445 }
2446
2447 if (request > REQ_NONE) {
2448 open_run_request(request);
2449 return TRUE;
2450 }
2451
2452 if (view && view->lines) {
2453 request = view->ops->request(view, request, &view->line[view->lineno]);
2454 if (request == REQ_NONE)
2455 return TRUE;
2456 }
2457
2458 switch (request) {
2459 case REQ_MOVE_UP:
2460 case REQ_MOVE_DOWN:
2461 case REQ_MOVE_PAGE_UP:
2462 case REQ_MOVE_PAGE_DOWN:
2463 case REQ_MOVE_FIRST_LINE:
2464 case REQ_MOVE_LAST_LINE:
2465 move_view(view, request);
2466 break;
2467
2468 case REQ_SCROLL_LINE_DOWN:
2469 case REQ_SCROLL_LINE_UP:
2470 case REQ_SCROLL_PAGE_DOWN:
2471 case REQ_SCROLL_PAGE_UP:
2472 scroll_view(view, request);
2473 break;
2474
2475 case REQ_VIEW_BLAME:
2476 if (!opt_file[0]) {
2477 report("No file chosen, press %s to open tree view",
2478 get_key(REQ_VIEW_TREE));
2479 break;
2480 }
2481 open_view(view, request, OPEN_DEFAULT);
2482 break;
2483
2484 case REQ_VIEW_BLOB:
2485 if (!ref_blob[0]) {
2486 report("No file chosen, press %s to open tree view",
2487 get_key(REQ_VIEW_TREE));
2488 break;
2489 }
2490 open_view(view, request, OPEN_DEFAULT);
2491 break;
2492
2493 case REQ_VIEW_PAGER:
2494 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2495 report("No pager content, press %s to run command from prompt",
2496 get_key(REQ_PROMPT));
2497 break;
2498 }
2499 open_view(view, request, OPEN_DEFAULT);
2500 break;
2501
2502 case REQ_VIEW_STAGE:
2503 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2504 report("No stage content, press %s to open the status view and choose file",
2505 get_key(REQ_VIEW_STATUS));
2506 break;
2507 }
2508 open_view(view, request, OPEN_DEFAULT);
2509 break;
2510
2511 case REQ_VIEW_STATUS:
2512 if (opt_is_inside_work_tree == FALSE) {
2513 report("The status view requires a working tree");
2514 break;
2515 }
2516 open_view(view, request, OPEN_DEFAULT);
2517 break;
2518
2519 case REQ_VIEW_MAIN:
2520 case REQ_VIEW_DIFF:
2521 case REQ_VIEW_LOG:
2522 case REQ_VIEW_TREE:
2523 case REQ_VIEW_HELP:
2524 open_view(view, request, OPEN_DEFAULT);
2525 break;
2526
2527 case REQ_NEXT:
2528 case REQ_PREVIOUS:
2529 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2530
2531 if ((view == VIEW(REQ_VIEW_DIFF) &&
2532 view->parent == VIEW(REQ_VIEW_MAIN)) ||
2533 (view == VIEW(REQ_VIEW_DIFF) &&
2534 view->parent == VIEW(REQ_VIEW_BLAME)) ||
2535 (view == VIEW(REQ_VIEW_STAGE) &&
2536 view->parent == VIEW(REQ_VIEW_STATUS)) ||
2537 (view == VIEW(REQ_VIEW_BLOB) &&
2538 view->parent == VIEW(REQ_VIEW_TREE))) {
2539 int line;
2540
2541 view = view->parent;
2542 line = view->lineno;
2543 move_view(view, request);
2544 if (view_is_displayed(view))
2545 update_view_title(view);
2546 if (line != view->lineno)
2547 view->ops->request(view, REQ_ENTER,
2548 &view->line[view->lineno]);
2549
2550 } else {
2551 move_view(view, request);
2552 }
2553 break;
2554
2555 case REQ_VIEW_NEXT:
2556 {
2557 int nviews = displayed_views();
2558 int next_view = (current_view + 1) % nviews;
2559
2560 if (next_view == current_view) {
2561 report("Only one view is displayed");
2562 break;
2563 }
2564
2565 current_view = next_view;
2566 /* Blur out the title of the previous view. */
2567 update_view_title(view);
2568 report("");
2569 break;
2570 }
2571 case REQ_REFRESH:
2572 report("Refreshing is not yet supported for the %s view", view->name);
2573 break;
2574
2575 case REQ_TOGGLE_LINENO:
2576 opt_line_number = !opt_line_number;
2577 redraw_display();
2578 break;
2579
2580 case REQ_TOGGLE_DATE:
2581 opt_date = !opt_date;
2582 redraw_display();
2583 break;
2584
2585 case REQ_TOGGLE_AUTHOR:
2586 opt_author = !opt_author;
2587 redraw_display();
2588 break;
2589
2590 case REQ_TOGGLE_REV_GRAPH:
2591 opt_rev_graph = !opt_rev_graph;
2592 redraw_display();
2593 break;
2594
2595 case REQ_TOGGLE_REFS:
2596 opt_show_refs = !opt_show_refs;
2597 redraw_display();
2598 break;
2599
2600 case REQ_PROMPT:
2601 /* Always reload^Wrerun commands from the prompt. */
2602 open_view(view, opt_request, OPEN_RELOAD);
2603 break;
2604
2605 case REQ_SEARCH:
2606 case REQ_SEARCH_BACK:
2607 search_view(view, request);
2608 break;
2609
2610 case REQ_FIND_NEXT:
2611 case REQ_FIND_PREV:
2612 find_next(view, request);
2613 break;
2614
2615 case REQ_STOP_LOADING:
2616 for (i = 0; i < ARRAY_SIZE(views); i++) {
2617 view = &views[i];
2618 if (view->pipe)
2619 report("Stopped loading the %s view", view->name),
2620 end_update(view);
2621 }
2622 break;
2623
2624 case REQ_SHOW_VERSION:
2625 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2626 return TRUE;
2627
2628 case REQ_SCREEN_RESIZE:
2629 resize_display();
2630 /* Fall-through */
2631 case REQ_SCREEN_REDRAW:
2632 redraw_display();
2633 break;
2634
2635 case REQ_EDIT:
2636 report("Nothing to edit");
2637 break;
2638
2639
2640 case REQ_ENTER:
2641 report("Nothing to enter");
2642 break;
2643
2644
2645 case REQ_VIEW_CLOSE:
2646 /* XXX: Mark closed views by letting view->parent point to the
2647 * view itself. Parents to closed view should never be
2648 * followed. */
2649 if (view->parent &&
2650 view->parent->parent != view->parent) {
2651 memset(display, 0, sizeof(display));
2652 current_view = 0;
2653 display[current_view] = view->parent;
2654 view->parent = view;
2655 resize_display();
2656 redraw_display();
2657 break;
2658 }
2659 /* Fall-through */
2660 case REQ_QUIT:
2661 return FALSE;
2662
2663 default:
2664 /* An unknown key will show most commonly used commands. */
2665 report("Unknown key, press 'h' for help");
2666 return TRUE;
2667 }
2668
2669 return TRUE;
2670 }
2671
2672
2673 /*
2674 * Pager backend
2675 */
2676
2677 static bool
2678 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2679 {
2680 char *text = line->data;
2681 enum line_type type = line->type;
2682 int attr;
2683
2684 wmove(view->win, lineno, 0);
2685
2686 if (selected) {
2687 type = LINE_CURSOR;
2688 wchgat(view->win, -1, 0, type, NULL);
2689 }
2690
2691 attr = get_line_attr(type);
2692 wattrset(view->win, attr);
2693
2694 if (opt_line_number || opt_tab_size < TABSIZE) {
2695 static char spaces[] = " ";
2696 int col_offset = 0, col = 0;
2697
2698 if (opt_line_number) {
2699 unsigned long real_lineno = view->offset + lineno + 1;
2700
2701 if (real_lineno == 1 ||
2702 (real_lineno % opt_num_interval) == 0) {
2703 wprintw(view->win, "%.*d", view->digits, real_lineno);
2704
2705 } else {
2706 waddnstr(view->win, spaces,
2707 MIN(view->digits, STRING_SIZE(spaces)));
2708 }
2709 waddstr(view->win, ": ");
2710 col_offset = view->digits + 2;
2711 }
2712
2713 while (text && col_offset + col < view->width) {
2714 int cols_max = view->width - col_offset - col;
2715 char *pos = text;
2716 int cols;
2717
2718 if (*text == '\t') {
2719 text++;
2720 assert(sizeof(spaces) > TABSIZE);
2721 pos = spaces;
2722 cols = opt_tab_size - (col % opt_tab_size);
2723
2724 } else {
2725 text = strchr(text, '\t');
2726 cols = line ? text - pos : strlen(pos);
2727 }
2728
2729 waddnstr(view->win, pos, MIN(cols, cols_max));
2730 col += cols;
2731 }
2732
2733 } else {
2734 int tilde_attr = get_line_attr(LINE_MAIN_DELIM);
2735
2736 draw_text(view, text, view->width, TRUE, tilde_attr);
2737 }
2738
2739 return TRUE;
2740 }
2741
2742 static bool
2743 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2744 {
2745 char refbuf[SIZEOF_STR];
2746 char *ref = NULL;
2747 FILE *pipe;
2748
2749 if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2750 return TRUE;
2751
2752 pipe = popen(refbuf, "r");
2753 if (!pipe)
2754 return TRUE;
2755
2756 if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2757 ref = chomp_string(ref);
2758 pclose(pipe);
2759
2760 if (!ref || !*ref)
2761 return TRUE;
2762
2763 /* This is the only fatal call, since it can "corrupt" the buffer. */
2764 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2765 return FALSE;
2766
2767 return TRUE;
2768 }
2769
2770 static void
2771 add_pager_refs(struct view *view, struct line *line)
2772 {
2773 char buf[SIZEOF_STR];
2774 char *commit_id = (char *)line->data + STRING_SIZE("commit ");
2775 struct ref **refs;
2776 size_t bufpos = 0, refpos = 0;
2777 const char *sep = "Refs: ";
2778 bool is_tag = FALSE;
2779
2780 assert(line->type == LINE_COMMIT);
2781
2782 refs = get_refs(commit_id);
2783 if (!refs) {
2784 if (view == VIEW(REQ_VIEW_DIFF))
2785 goto try_add_describe_ref;
2786 return;
2787 }
2788
2789 do {
2790 struct ref *ref = refs[refpos];
2791 char *fmt = ref->tag ? "%s[%s]" :
2792 ref->remote ? "%s<%s>" : "%s%s";
2793
2794 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2795 return;
2796 sep = ", ";
2797 if (ref->tag)
2798 is_tag = TRUE;
2799 } while (refs[refpos++]->next);
2800
2801 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2802 try_add_describe_ref:
2803 /* Add <tag>-g<commit_id> "fake" reference. */
2804 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2805 return;
2806 }
2807
2808 if (bufpos == 0)
2809 return;
2810
2811 if (!realloc_lines(view, view->line_size + 1))
2812 return;
2813
2814 add_line_text(view, buf, LINE_PP_REFS);
2815 }
2816
2817 static bool
2818 pager_read(struct view *view, char *data)
2819 {
2820 struct line *line;
2821
2822 if (!data)
2823 return TRUE;
2824
2825 line = add_line_text(view, data, get_line_type(data));
2826 if (!line)
2827 return FALSE;
2828
2829 if (line->type == LINE_COMMIT &&
2830 (view == VIEW(REQ_VIEW_DIFF) ||
2831 view == VIEW(REQ_VIEW_LOG)))
2832 add_pager_refs(view, line);
2833
2834 return TRUE;
2835 }
2836
2837 static enum request
2838 pager_request(struct view *view, enum request request, struct line *line)
2839 {
2840 int split = 0;
2841
2842 if (request != REQ_ENTER)
2843 return request;
2844
2845 if (line->type == LINE_COMMIT &&
2846 (view == VIEW(REQ_VIEW_LOG) ||
2847 view == VIEW(REQ_VIEW_PAGER))) {
2848 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2849 split = 1;
2850 }
2851
2852 /* Always scroll the view even if it was split. That way
2853 * you can use Enter to scroll through the log view and
2854 * split open each commit diff. */
2855 scroll_view(view, REQ_SCROLL_LINE_DOWN);
2856
2857 /* FIXME: A minor workaround. Scrolling the view will call report("")
2858 * but if we are scrolling a non-current view this won't properly
2859 * update the view title. */
2860 if (split)
2861 update_view_title(view);
2862
2863 return REQ_NONE;
2864 }
2865
2866 static bool
2867 pager_grep(struct view *view, struct line *line)
2868 {
2869 regmatch_t pmatch;
2870 char *text = line->data;
2871
2872 if (!*text)
2873 return FALSE;
2874
2875 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2876 return FALSE;
2877
2878 return TRUE;
2879 }
2880
2881 static void
2882 pager_select(struct view *view, struct line *line)
2883 {
2884 if (line->type == LINE_COMMIT) {
2885 char *text = (char *)line->data + STRING_SIZE("commit ");
2886
2887 if (view != VIEW(REQ_VIEW_PAGER))
2888 string_copy_rev(view->ref, text);
2889 string_copy_rev(ref_commit, text);
2890 }
2891 }
2892
2893 static struct view_ops pager_ops = {
2894 "line",
2895 NULL,
2896 pager_read,
2897 pager_draw,
2898 pager_request,
2899 pager_grep,
2900 pager_select,
2901 };
2902
2903
2904 /*
2905 * Help backend
2906 */
2907
2908 static bool
2909 help_open(struct view *view)
2910 {
2911 char buf[BUFSIZ];
2912 int lines = ARRAY_SIZE(req_info) + 2;
2913 int i;
2914
2915 if (view->lines > 0)
2916 return TRUE;
2917
2918 for (i = 0; i < ARRAY_SIZE(req_info); i++)
2919 if (!req_info[i].request)
2920 lines++;
2921
2922 lines += run_requests + 1;
2923
2924 view->line = calloc(lines, sizeof(*view->line));
2925 if (!view->line)
2926 return FALSE;
2927
2928 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
2929
2930 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
2931 char *key;
2932
2933 if (req_info[i].request == REQ_NONE)
2934 continue;
2935
2936 if (!req_info[i].request) {
2937 add_line_text(view, "", LINE_DEFAULT);
2938 add_line_text(view, req_info[i].help, LINE_DEFAULT);
2939 continue;
2940 }
2941
2942 key = get_key(req_info[i].request);
2943 if (!*key)
2944 key = "(no key defined)";
2945
2946 if (!string_format(buf, " %-25s %s", key, req_info[i].help))
2947 continue;
2948
2949 add_line_text(view, buf, LINE_DEFAULT);
2950 }
2951
2952 if (run_requests) {
2953 add_line_text(view, "", LINE_DEFAULT);
2954 add_line_text(view, "External commands:", LINE_DEFAULT);
2955 }
2956
2957 for (i = 0; i < run_requests; i++) {
2958 struct run_request *req = get_run_request(REQ_NONE + i + 1);
2959 char *key;
2960
2961 if (!req)
2962 continue;
2963
2964 key = get_key_name(req->key);
2965 if (!*key)
2966 key = "(no key defined)";
2967
2968 if (!string_format(buf, " %-10s %-14s `%s`",
2969 keymap_table[req->keymap].name,
2970 key, req->cmd))
2971 continue;
2972
2973 add_line_text(view, buf, LINE_DEFAULT);
2974 }
2975
2976 return TRUE;
2977 }
2978
2979 static struct view_ops help_ops = {
2980 "line",
2981 help_open,
2982 NULL,
2983 pager_draw,
2984 pager_request,
2985 pager_grep,
2986 pager_select,
2987 };
2988
2989
2990 /*
2991 * Tree backend
2992 */
2993
2994 struct tree_stack_entry {
2995 struct tree_stack_entry *prev; /* Entry below this in the stack */
2996 unsigned long lineno; /* Line number to restore */
2997 char *name; /* Position of name in opt_path */
2998 };
2999
3000 /* The top of the path stack. */
3001 static struct tree_stack_entry *tree_stack = NULL;
3002 unsigned long tree_lineno = 0;
3003
3004 static void
3005 pop_tree_stack_entry(void)
3006 {
3007 struct tree_stack_entry *entry = tree_stack;
3008
3009 tree_lineno = entry->lineno;
3010 entry->name[0] = 0;
3011 tree_stack = entry->prev;
3012 free(entry);
3013 }
3014
3015 static void
3016 push_tree_stack_entry(char *name, unsigned long lineno)
3017 {
3018 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3019 size_t pathlen = strlen(opt_path);
3020
3021 if (!entry)
3022 return;
3023
3024 entry->prev = tree_stack;
3025 entry->name = opt_path + pathlen;
3026 tree_stack = entry;
3027
3028 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3029 pop_tree_stack_entry();
3030 return;
3031 }
3032
3033 /* Move the current line to the first tree entry. */
3034 tree_lineno = 1;
3035 entry->lineno = lineno;
3036 }
3037
3038 /* Parse output from git-ls-tree(1):
3039 *
3040 * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
3041 * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
3042 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3043 * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
3044 */
3045
3046 #define SIZEOF_TREE_ATTR \
3047 STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
3048
3049 #define TREE_UP_FORMAT "040000 tree %s\t.."
3050
3051 static int
3052 tree_compare_entry(enum line_type type1, char *name1,
3053 enum line_type type2, char *name2)
3054 {
3055 if (type1 != type2) {
3056 if (type1 == LINE_TREE_DIR)
3057 return -1;
3058 return 1;
3059 }
3060
3061 return strcmp(name1, name2);
3062 }
3063
3064 static char *
3065 tree_path(struct line *line)
3066 {
3067 char *path = line->data;
3068
3069 return path + SIZEOF_TREE_ATTR;
3070 }
3071
3072 static bool
3073 tree_read(struct view *view, char *text)
3074 {
3075 size_t textlen = text ? strlen(text) : 0;
3076 char buf[SIZEOF_STR];
3077 unsigned long pos;
3078 enum line_type type;
3079 bool first_read = view->lines == 0;
3080
3081 if (!text)
3082 return TRUE;
3083 if (textlen <= SIZEOF_TREE_ATTR)
3084 return FALSE;
3085
3086 type = text[STRING_SIZE("100644 ")] == 't'
3087 ? LINE_TREE_DIR : LINE_TREE_FILE;
3088
3089 if (first_read) {
3090 /* Add path info line */
3091 if (!string_format(buf, "Directory path /%s", opt_path) ||
3092 !realloc_lines(view, view->line_size + 1) ||
3093 !add_line_text(view, buf, LINE_DEFAULT))
3094 return FALSE;
3095
3096 /* Insert "link" to parent directory. */
3097 if (*opt_path) {
3098 if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
3099 !realloc_lines(view, view->line_size + 1) ||
3100 !add_line_text(view, buf, LINE_TREE_DIR))
3101 return FALSE;
3102 }
3103 }
3104
3105 /* Strip the path part ... */
3106 if (*opt_path) {
3107 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3108 size_t striplen = strlen(opt_path);
3109 char *path = text + SIZEOF_TREE_ATTR;
3110
3111 if (pathlen > striplen)
3112 memmove(path, path + striplen,
3113 pathlen - striplen + 1);
3114 }
3115
3116 /* Skip "Directory ..." and ".." line. */
3117 for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
3118 struct line *line = &view->line[pos];
3119 char *path1 = tree_path(line);
3120 char *path2 = text + SIZEOF_TREE_ATTR;
3121 int cmp = tree_compare_entry(line->type, path1, type, path2);
3122
3123 if (cmp <= 0)
3124 continue;
3125
3126 text = strdup(text);
3127 if (!text)
3128 return FALSE;
3129
3130 if (view->lines > pos)
3131 memmove(&view->line[pos + 1], &view->line[pos],
3132 (view->lines - pos) * sizeof(*line));
3133
3134 line = &view->line[pos];
3135 line->data = text;
3136 line->type = type;
3137 view->lines++;
3138 return TRUE;
3139 }
3140
3141 if (!add_line_text(view, text, type))
3142 return FALSE;
3143
3144 if (tree_lineno > view->lineno) {
3145 view->lineno = tree_lineno;
3146 tree_lineno = 0;
3147 }
3148
3149 return TRUE;
3150 }
3151
3152 static enum request
3153 tree_request(struct view *view, enum request request, struct line *line)
3154 {
3155 enum open_flags flags;
3156
3157 if (request == REQ_VIEW_BLAME) {
3158 char *filename = tree_path(line);
3159
3160 if (line->type == LINE_TREE_DIR) {
3161 report("Cannot show blame for directory %s", opt_path);
3162 return REQ_NONE;
3163 }
3164
3165 string_copy(opt_ref, view->vid);
3166 string_format(opt_file, "%s%s", opt_path, filename);
3167 return request;
3168 }
3169 if (request == REQ_TREE_PARENT) {
3170 if (*opt_path) {
3171 /* fake 'cd ..' */
3172 request = REQ_ENTER;
3173 line = &view->line[1];
3174 } else {
3175 /* quit view if at top of tree */
3176 return REQ_VIEW_CLOSE;
3177 }
3178 }
3179 if (request != REQ_ENTER)
3180 return request;
3181
3182 /* Cleanup the stack if the tree view is at a different tree. */
3183 while (!*opt_path && tree_stack)
3184 pop_tree_stack_entry();
3185
3186 switch (line->type) {
3187 case LINE_TREE_DIR:
3188 /* Depending on whether it is a subdir or parent (updir?) link
3189 * mangle the path buffer. */
3190 if (line == &view->line[1] && *opt_path) {
3191 pop_tree_stack_entry();
3192
3193 } else {
3194 char *basename = tree_path(line);
3195
3196 push_tree_stack_entry(basename, view->lineno);
3197 }
3198
3199 /* Trees and subtrees share the same ID, so they are not not
3200 * unique like blobs. */
3201 flags = OPEN_RELOAD;
3202 request = REQ_VIEW_TREE;
3203 break;
3204
3205 case LINE_TREE_FILE:
3206 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3207 request = REQ_VIEW_BLOB;
3208 break;
3209
3210 default:
3211 return TRUE;
3212 }
3213
3214 open_view(view, request, flags);
3215 if (request == REQ_VIEW_TREE) {
3216 view->lineno = tree_lineno;
3217 }
3218
3219 return REQ_NONE;
3220 }
3221
3222 static void
3223 tree_select(struct view *view, struct line *line)
3224 {
3225 char *text = (char *)line->data + STRING_SIZE("100644 blob ");
3226
3227 if (line->type == LINE_TREE_FILE) {
3228 string_copy_rev(ref_blob, text);
3229
3230 } else if (line->type != LINE_TREE_DIR) {
3231 return;
3232 }
3233
3234 string_copy_rev(view->ref, text);
3235 }
3236
3237 static struct view_ops tree_ops = {
3238 "file",
3239 NULL,
3240 tree_read,
3241 pager_draw,
3242 tree_request,
3243 pager_grep,
3244 tree_select,
3245 };
3246
3247 static bool
3248 blob_read(struct view *view, char *line)
3249 {
3250 if (!line)
3251 return TRUE;
3252 return add_line_text(view, line, LINE_DEFAULT) != NULL;
3253 }
3254
3255 static struct view_ops blob_ops = {
3256 "line",
3257 NULL,
3258 blob_read,
3259 pager_draw,
3260 pager_request,
3261 pager_grep,
3262 pager_select,
3263 };
3264
3265 /*
3266 * Blame backend
3267 *
3268 * Loading the blame view is a two phase job:
3269 *
3270 * 1. File content is read either using opt_file from the
3271 * filesystem or using git-cat-file.
3272 * 2. Then blame information is incrementally added by
3273 * reading output from git-blame.
3274 */
3275
3276 struct blame_commit {
3277 char id[SIZEOF_REV]; /* SHA1 ID. */
3278 char title[128]; /* First line of the commit message. */
3279 char author[75]; /* Author of the commit. */
3280 struct tm time; /* Date from the author ident. */
3281 char filename[128]; /* Name of file. */
3282 };
3283
3284 struct blame {
3285 struct blame_commit *commit;
3286 unsigned int header:1;
3287 char text[1];
3288 };
3289
3290 #define BLAME_CAT_FILE_CMD "git cat-file blob %s:%s"
3291 #define BLAME_INCREMENTAL_CMD "git blame --incremental %s %s"
3292
3293 static bool
3294 blame_open(struct view *view)
3295 {
3296 char path[SIZEOF_STR];
3297 char ref[SIZEOF_STR] = "";
3298
3299 if (sq_quote(path, 0, opt_file) >= sizeof(path))
3300 return FALSE;
3301
3302 if (*opt_ref && sq_quote(ref, 0, opt_ref) >= sizeof(ref))
3303 return FALSE;
3304
3305 if (*opt_ref) {
3306 if (!string_format(view->cmd, BLAME_CAT_FILE_CMD, ref, path))
3307 return FALSE;
3308 } else {
3309 view->pipe = fopen(opt_file, "r");
3310 if (!view->pipe &&
3311 !string_format(view->cmd, BLAME_CAT_FILE_CMD, "HEAD", path))
3312 return FALSE;
3313 }
3314
3315 if (!view->pipe)
3316 view->pipe = popen(view->cmd, "r");
3317 if (!view->pipe)
3318 return FALSE;
3319
3320 if (!string_format(view->cmd, BLAME_INCREMENTAL_CMD, ref, path))
3321 return FALSE;
3322
3323 string_format(view->ref, "%s ...", opt_file);
3324 string_copy_rev(view->vid, opt_file);
3325 set_nonblocking_input(TRUE);
3326
3327 if (view->line) {
3328 int i;
3329
3330 for (i = 0; i < view->lines; i++)
3331 free(view->line[i].data);
3332 free(view->line);
3333 }
3334
3335 view->lines = view->line_alloc = view->line_size = view->lineno = 0;
3336 view->offset = view->lines = view->lineno = 0;
3337 view->line = NULL;
3338 view->start_time = time(NULL);
3339
3340 return TRUE;
3341 }
3342
3343 static struct blame_commit *
3344 get_blame_commit(struct view *view, const char *id)
3345 {
3346 size_t i;
3347
3348 for (i = 0; i < view->lines; i++) {
3349 struct blame *blame = view->line[i].data;
3350
3351 if (!blame->commit)
3352 continue;
3353
3354 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
3355 return blame->commit;
3356 }
3357
3358 {
3359 struct blame_commit *commit = calloc(1, sizeof(*commit));
3360
3361 if (commit)
3362 string_ncopy(commit->id, id, SIZEOF_REV);
3363 return commit;
3364 }
3365 }
3366
3367 static bool
3368 parse_number(char **posref, size_t *number, size_t min, size_t max)
3369 {
3370 char *pos = *posref;
3371
3372 *posref = NULL;
3373 pos = strchr(pos + 1, ' ');
3374 if (!pos || !isdigit(pos[1]))
3375 return FALSE;
3376 *number = atoi(pos + 1);
3377 if (*number < min || *number > max)
3378 return FALSE;
3379
3380 *posref = pos;
3381 return TRUE;
3382 }
3383
3384 static struct blame_commit *
3385 parse_blame_commit(struct view *view, char *text, int *blamed)
3386 {
3387 struct blame_commit *commit;
3388 struct blame *blame;
3389 char *pos = text + SIZEOF_REV - 1;
3390 size_t lineno;
3391 size_t group;
3392
3393 if (strlen(text) <= SIZEOF_REV || *pos != ' ')
3394 return NULL;
3395
3396 if (!parse_number(&pos, &lineno, 1, view->lines) ||
3397 !parse_number(&pos, &group, 1, view->lines - lineno + 1))
3398 return NULL;
3399
3400 commit = get_blame_commit(view, text);
3401 if (!commit)
3402 return NULL;
3403
3404 *blamed += group;
3405 while (group--) {
3406 struct line *line = &view->line[lineno + group - 1];
3407
3408 blame = line->data;
3409 blame->commit = commit;
3410 line->dirty = 1;
3411 }
3412 blame->header = 1;
3413
3414 return commit;
3415 }
3416
3417 static bool
3418 blame_read_file(struct view *view, char *line)
3419 {
3420 if (!line) {
3421 FILE *pipe = NULL;
3422
3423 if (view->lines > 0)
3424 pipe = popen(view->cmd, "r");
3425 view->cmd[0] = 0;
3426 if (!pipe) {
3427 report("Failed to load blame data");
3428 return TRUE;
3429 }
3430
3431 fclose(view->pipe);
3432 view->pipe = pipe;
3433 return FALSE;
3434
3435 } else {
3436 size_t linelen = strlen(line);
3437 struct blame *blame = malloc(sizeof(*blame) + linelen);
3438
3439 if (!line)
3440 return FALSE;
3441
3442 blame->commit = NULL;
3443 strncpy(blame->text, line, linelen);
3444 blame->text[linelen] = 0;
3445 return add_line_data(view, blame, LINE_BLAME_COMMIT) != NULL;
3446 }
3447 }
3448
3449 static bool
3450 match_blame_header(const char *name, char **line)
3451 {
3452 size_t namelen = strlen(name);
3453 bool matched = !strncmp(name, *line, namelen);
3454
3455 if (matched)
3456 *line += namelen;
3457
3458 return matched;
3459 }
3460
3461 static bool
3462 blame_read(struct view *view, char *line)
3463 {
3464 static struct blame_commit *commit = NULL;
3465 static int blamed = 0;
3466 static time_t author_time;
3467
3468 if (*view->cmd)
3469 return blame_read_file(view, line);
3470
3471 if (!line) {
3472 /* Reset all! */
3473 commit = NULL;
3474 blamed = 0;
3475 string_format(view->ref, "%s", view->vid);
3476 if (view_is_displayed(view)) {
3477 update_view_title(view);
3478 redraw_view_from(view, 0);
3479 }
3480 return TRUE;
3481 }
3482
3483 if (!commit) {
3484 commit = parse_blame_commit(view, line, &blamed);
3485 string_format(view->ref, "%s %2d%%", view->vid,
3486 blamed * 100 / view->lines);
3487
3488 } else if (match_blame_header("author ", &line)) {
3489 string_ncopy(commit->author, line, strlen(line));
3490
3491 } else if (match_blame_header("author-time ", &line)) {
3492 author_time = (time_t) atol(line);
3493
3494 } else if (match_blame_header("author-tz ", &line)) {
3495 long tz;
3496
3497 tz = ('0' - line[1]) * 60 * 60 * 10;
3498 tz += ('0' - line[2]) * 60 * 60;
3499 tz += ('0' - line[3]) * 60;
3500 tz += ('0' - line[4]) * 60;
3501
3502 if (line[0] == '-')
3503 tz = -tz;
3504
3505 author_time -= tz;
3506 gmtime_r(&author_time, &commit->time);
3507
3508 } else if (match_blame_header("summary ", &line)) {
3509 string_ncopy(commit->title, line, strlen(line));
3510
3511 } else if (match_blame_header("filename ", &line)) {
3512 string_ncopy(commit->filename, line, strlen(line));
3513 commit = NULL;
3514 }
3515
3516 return TRUE;
3517 }
3518
3519 static bool
3520 blame_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3521 {
3522 int tilde_attr = -1;
3523 struct blame *blame = line->data;
3524 int col = 0;
3525
3526 wmove(view->win, lineno, 0);
3527
3528 if (selected) {
3529 wattrset(view->win, get_line_attr(LINE_CURSOR));
3530 wchgat(view->win, -1, 0, LINE_CURSOR, NULL);
3531 } else {
3532 wattrset(view->win, A_NORMAL);
3533 tilde_attr = get_line_attr(LINE_MAIN_DELIM);
3534 }
3535
3536 if (opt_date) {
3537 int n;
3538
3539 if (!selected)
3540 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
3541 if (blame->commit) {
3542 char buf[DATE_COLS + 1];
3543 int timelen;
3544
3545 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &blame->commit->time);
3546 n = draw_text(view, buf, view->width - col, FALSE, tilde_attr);
3547 draw_text(view, " ", view->width - col - n, FALSE, tilde_attr);
3548 }
3549
3550 col += DATE_COLS;
3551 wmove(view->win, lineno, col);
3552 if (col >= view->width)
3553 return TRUE;
3554 }
3555
3556 if (opt_author) {
3557 int max = MIN(AUTHOR_COLS - 1, view->width - col);
3558
3559 if (!selected)
3560 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
3561 if (blame->commit)
3562 draw_text(view, blame->commit->author, max, TRUE, tilde_attr);
3563 col += AUTHOR_COLS;
3564 if (col >= view->width)
3565 return TRUE;
3566 wmove(view->win, lineno, col);
3567 }
3568
3569 {
3570 int max = MIN(ID_COLS - 1, view->width - col);
3571
3572 if (!selected)
3573 wattrset(view->win, get_line_attr(LINE_BLAME_ID));
3574 if (blame->commit)
3575 draw_text(view, blame->commit->id, max, FALSE, -1);
3576 col += ID_COLS;
3577 if (col >= view->width)
3578 return TRUE;
3579 wmove(view->win, lineno, col);
3580 }
3581
3582 {
3583 unsigned long real_lineno = view->offset + lineno + 1;
3584 char number[10] = " ";
3585 int max = MIN(view->digits, STRING_SIZE(number));
3586 bool showtrimmed = FALSE;
3587
3588 if (real_lineno == 1 ||
3589 (real_lineno % opt_num_interval) == 0) {
3590 char fmt[] = "%1ld";
3591
3592 if (view->digits <= 9)
3593 fmt[1] = '0' + view->digits;
3594
3595 if (!string_format(number, fmt, real_lineno))
3596 number[0] = 0;
3597 showtrimmed = TRUE;
3598 }
3599
3600 if (max > view->width - col)
3601 max = view->width - col;
3602 if (!selected)
3603 wattrset(view->win, get_line_attr(LINE_BLAME_LINENO));
3604 col += draw_text(view, number, max, showtrimmed, tilde_attr);
3605 if (col >= view->width)
3606 return TRUE;
3607 }
3608
3609 if (!selected)
3610 wattrset(view->win, A_NORMAL);
3611
3612 if (col >= view->width)
3613 return TRUE;
3614 waddch(view->win, ACS_VLINE);
3615 col++;
3616 if (col >= view->width)
3617 return TRUE;
3618 waddch(view->win, ' ');
3619 col++;
3620 col += draw_text(view, blame->text, view->width - col, TRUE, tilde_attr);
3621
3622 return TRUE;
3623 }
3624
3625 static enum request
3626 blame_request(struct view *view, enum request request, struct line *line)
3627 {
3628 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3629 struct blame *blame = line->data;
3630
3631 switch (request) {
3632 case REQ_ENTER:
3633 if (!blame->commit) {
3634 report("No commit loaded yet");
3635 break;
3636 }
3637
3638 if (!strcmp(blame->commit->id, "0000000000000000000000000000000000000000")) {
3639 char path[SIZEOF_STR];
3640
3641 if (sq_quote(path, 0, view->vid) >= sizeof(path))
3642 break;
3643 string_format(opt_cmd, "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s 2>/dev/null", path);
3644 }
3645
3646 open_view(view, REQ_VIEW_DIFF, flags);
3647 break;
3648
3649 default:
3650 return request;
3651 }
3652
3653 return REQ_NONE;
3654 }
3655
3656 static bool
3657 blame_grep(struct view *view, struct line *line)
3658 {
3659 struct blame *blame = line->data;
3660 struct blame_commit *commit = blame->commit;
3661 regmatch_t pmatch;
3662
3663 #define MATCH(text) \
3664 (*text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3665
3666 if (commit) {
3667 char buf[DATE_COLS + 1];
3668
3669 if (MATCH(commit->title) ||
3670 MATCH(commit->author) ||
3671 MATCH(commit->id))
3672 return TRUE;
3673
3674 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
3675 MATCH(buf))
3676 return TRUE;
3677 }
3678
3679 return MATCH(blame->text);
3680
3681 #undef MATCH
3682 }
3683
3684 static void
3685 blame_select(struct view *view, struct line *line)
3686 {
3687 struct blame *blame = line->data;
3688 struct blame_commit *commit = blame->commit;
3689
3690 if (!commit)
3691 return;
3692
3693 if (!strcmp(commit->id, "0000000000000000000000000000000000000000"))
3694 string_ncopy(ref_commit, "HEAD", 4);
3695 else
3696 string_copy_rev(ref_commit, commit->id);
3697 }
3698
3699 static struct view_ops blame_ops = {
3700 "line",
3701 blame_open,
3702 blame_read,
3703 blame_draw,
3704 blame_request,
3705 blame_grep,
3706 blame_select,
3707 };
3708
3709 /*
3710 * Status backend
3711 */
3712
3713 struct status {
3714 char status;
3715 struct {
3716 mode_t mode;
3717 char rev[SIZEOF_REV];
3718 char name[SIZEOF_STR];
3719 } old;
3720 struct {
3721 mode_t mode;
3722 char rev[SIZEOF_REV];
3723 char name[SIZEOF_STR];
3724 } new;
3725 };
3726
3727 static struct status stage_status;
3728 static enum line_type stage_line_type;
3729
3730 /* Get fields from the diff line:
3731 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
3732 */
3733 static inline bool
3734 status_get_diff(struct status *file, char *buf, size_t bufsize)
3735 {
3736 char *old_mode = buf + 1;
3737 char *new_mode = buf + 8;
3738 char *old_rev = buf + 15;
3739 char *new_rev = buf + 56;
3740 char *status = buf + 97;
3741
3742 if (bufsize < 99 ||
3743 old_mode[-1] != ':' ||
3744 new_mode[-1] != ' ' ||
3745 old_rev[-1] != ' ' ||
3746 new_rev[-1] != ' ' ||
3747 status[-1] != ' ')
3748 return FALSE;
3749
3750 file->status = *status;
3751
3752 string_copy_rev(file->old.rev, old_rev);
3753 string_copy_rev(file->new.rev, new_rev);
3754
3755 file->old.mode = strtoul(old_mode, NULL, 8);
3756 file->new.mode = strtoul(new_mode, NULL, 8);
3757
3758 file->old.name[0] = file->new.name[0] = 0;
3759
3760 return TRUE;
3761 }
3762
3763 static bool
3764 status_run(struct view *view, const char cmd[], bool diff, enum line_type type)
3765 {
3766 struct status *file = NULL;
3767 struct status *unmerged = NULL;
3768 char buf[SIZEOF_STR * 4];
3769 size_t bufsize = 0;
3770 FILE *pipe;
3771
3772 pipe = popen(cmd, "r");
3773 if (!pipe)
3774 return FALSE;
3775
3776 add_line_data(view, NULL, type);
3777
3778 while (!feof(pipe) && !ferror(pipe)) {
3779 char *sep;
3780 size_t readsize;
3781
3782 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
3783 if (!readsize)
3784 break;
3785 bufsize += readsize;
3786
3787 /* Process while we have NUL chars. */
3788 while ((sep = memchr(buf, 0, bufsize))) {
3789 size_t sepsize = sep - buf + 1;
3790
3791 if (!file) {
3792 if (!realloc_lines(view, view->line_size + 1))
3793 goto error_out;
3794
3795 file = calloc(1, sizeof(*file));
3796 if (!file)
3797 goto error_out;
3798
3799 add_line_data(view, file, type);
3800 }
3801
3802 /* Parse diff info part. */
3803 if (!diff) {
3804 file->status = '?';
3805
3806 } else if (!file->status) {
3807 if (!status_get_diff(file, buf, sepsize))
3808 goto error_out;
3809
3810 bufsize -= sepsize;
3811 memmove(buf, sep + 1, bufsize);
3812
3813 sep = memchr(buf, 0, bufsize);
3814 if (!sep)
3815 break;
3816 sepsize = sep - buf + 1;
3817
3818 /* Collapse all 'M'odified entries that
3819 * follow a associated 'U'nmerged entry.
3820 */
3821 if (file->status == 'U') {
3822 unmerged = file;
3823
3824 } else if (unmerged) {
3825 int collapse = !strcmp(buf, unmerged->new.name);
3826
3827 unmerged = NULL;
3828 if (collapse) {
3829 free(file);
3830 view->lines--;
3831 continue;
3832 }
3833 }
3834 }
3835
3836 /* Grab the old name for rename/copy. */
3837 if (!*file->old.name &&
3838 (file->status == 'R' || file->status == 'C')) {
3839 sepsize = sep - buf + 1;
3840 string_ncopy(file->old.name, buf, sepsize);
3841 bufsize -= sepsize;
3842 memmove(buf, sep + 1, bufsize);
3843
3844 sep = memchr(buf, 0, bufsize);
3845 if (!sep)
3846 break;
3847 sepsize = sep - buf + 1;
3848 }
3849
3850 /* git-ls-files just delivers a NUL separated
3851 * list of file names similar to the second half
3852 * of the git-diff-* output. */
3853 string_ncopy(file->new.name, buf, sepsize);
3854 if (!*file->old.name)
3855 string_copy(file->old.name, file->new.name);
3856 bufsize -= sepsize;
3857 memmove(buf, sep + 1, bufsize);
3858 file = NULL;
3859 }
3860 }
3861
3862 if (ferror(pipe)) {
3863 error_out:
3864 pclose(pipe);
3865 return FALSE;
3866 }
3867
3868 if (!view->line[view->lines - 1].data)
3869 add_line_data(view, NULL, LINE_STAT_NONE);
3870
3871 pclose(pipe);
3872 return TRUE;
3873 }
3874
3875 /* Don't show unmerged entries in the staged section. */
3876 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --diff-filter=ACDMRTXB --cached -M HEAD"
3877 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
3878 #define STATUS_LIST_OTHER_CMD \
3879 "git ls-files -z --others --exclude-per-directory=.gitignore"
3880
3881 #define STATUS_DIFF_INDEX_SHOW_CMD \
3882 "git diff-index --root --patch-with-stat -C -M --cached HEAD -- %s %s 2>/dev/null"
3883
3884 #define STATUS_DIFF_FILES_SHOW_CMD \
3885 "git diff-files --root --patch-with-stat -C -M -- %s %s 2>/dev/null"
3886
3887 /* First parse staged info using git-diff-index(1), then parse unstaged
3888 * info using git-diff-files(1), and finally untracked files using
3889 * git-ls-files(1). */
3890 static bool
3891 status_open(struct view *view)
3892 {
3893 struct stat statbuf;
3894 char exclude[SIZEOF_STR];
3895 char cmd[SIZEOF_STR];
3896 unsigned long prev_lineno = view->lineno;
3897 size_t i;
3898
3899 for (i = 0; i < view->lines; i++)
3900 free(view->line[i].data);
3901 free(view->line);
3902 view->lines = view->line_alloc = view->line_size = view->lineno = 0;
3903 view->line = NULL;
3904
3905 if (!realloc_lines(view, view->line_size + 6))
3906 return FALSE;
3907
3908 if (!string_format(exclude, "%s/info/exclude", opt_git_dir))
3909 return FALSE;
3910
3911 string_copy(cmd, STATUS_LIST_OTHER_CMD);
3912
3913 if (stat(exclude, &statbuf) >= 0) {
3914 size_t cmdsize = strlen(cmd);
3915
3916 if (!string_format_from(cmd, &cmdsize, " %s", "--exclude-from=") ||
3917 sq_quote(cmd, cmdsize, exclude) >= sizeof(cmd))
3918 return FALSE;
3919 }
3920
3921 system("git update-index -q --refresh");
3922
3923 if (!status_run(view, STATUS_DIFF_INDEX_CMD, TRUE, LINE_STAT_STAGED) ||
3924 !status_run(view, STATUS_DIFF_FILES_CMD, TRUE, LINE_STAT_UNSTAGED) ||
3925 !status_run(view, cmd, FALSE, LINE_STAT_UNTRACKED))
3926 return FALSE;
3927
3928 /* If all went well restore the previous line number to stay in
3929 * the context. */
3930 if (prev_lineno < view->lines)
3931 view->lineno = prev_lineno;
3932 else
3933 view->lineno = view->lines - 1;
3934
3935 return TRUE;
3936 }
3937
3938 static bool
3939 status_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3940 {
3941 struct status *status = line->data;
3942 int tilde_attr = get_line_attr(LINE_MAIN_DELIM);
3943
3944 wmove(view->win, lineno, 0);
3945
3946 if (selected) {
3947 wattrset(view->win, get_line_attr(LINE_CURSOR));
3948 wchgat(view->win, -1, 0, LINE_CURSOR, NULL);
3949 tilde_attr = -1;
3950
3951 } else if (!status && line->type != LINE_STAT_NONE) {
3952 wattrset(view->win, get_line_attr(LINE_STAT_SECTION));
3953 wchgat(view->win, -1, 0, LINE_STAT_SECTION, NULL);
3954
3955 } else {
3956 wattrset(view->win, get_line_attr(line->type));
3957 }
3958
3959 if (!status) {
3960 char *text;
3961
3962 switch (line->type) {
3963 case LINE_STAT_STAGED:
3964 text = "Changes to be committed:";
3965 break;
3966
3967 case LINE_STAT_UNSTAGED:
3968 text = "Changed but not updated:";
3969 break;
3970
3971 case LINE_STAT_UNTRACKED:
3972 text = "Untracked files:";
3973 break;
3974
3975 case LINE_STAT_NONE:
3976 text = " (no files)";
3977 break;
3978
3979 default:
3980 return FALSE;
3981 }
3982
3983 draw_text(view, text, view->width, TRUE, tilde_attr);
3984 return TRUE;
3985 }
3986
3987 waddch(view->win, status->status);
3988 if (!selected)
3989 wattrset(view->win, A_NORMAL);
3990 wmove(view->win, lineno, 4);
3991 if (view->width < 5)
3992 return TRUE;
3993
3994 draw_text(view, status->new.name, view->width - 5, TRUE, tilde_attr);
3995 return TRUE;
3996 }
3997
3998 static enum request
3999 status_enter(struct view *view, struct line *line)
4000 {
4001 struct status *status = line->data;
4002 char oldpath[SIZEOF_STR] = "";
4003 char newpath[SIZEOF_STR] = "";
4004 char *info;
4005 size_t cmdsize = 0;
4006
4007 if (line->type == LINE_STAT_NONE ||
4008 (!status && line[1].type == LINE_STAT_NONE)) {
4009 report("No file to diff");
4010 return REQ_NONE;
4011 }
4012
4013 if (status) {
4014 if (sq_quote(oldpath, 0, status->old.name) >= sizeof(oldpath))
4015 return REQ_QUIT;
4016 /* Diffs for unmerged entries are empty when pasing the
4017 * new path, so leave it empty. */
4018 if (status->status != 'U' &&
4019 sq_quote(newpath, 0, status->new.name) >= sizeof(newpath))
4020 return REQ_QUIT;
4021 }
4022
4023 if (opt_cdup[0] &&
4024 line->type != LINE_STAT_UNTRACKED &&
4025 !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
4026 return REQ_QUIT;
4027
4028 switch (line->type) {
4029 case LINE_STAT_STAGED:
4030 if (!string_format_from(opt_cmd, &cmdsize,
4031 STATUS_DIFF_INDEX_SHOW_CMD, oldpath, newpath))
4032 return REQ_QUIT;
4033 if (status)
4034 info = "Staged changes to %s";
4035 else
4036 info = "Staged changes";
4037 break;
4038
4039 case LINE_STAT_UNSTAGED:
4040 if (!string_format_from(opt_cmd, &cmdsize,
4041 STATUS_DIFF_FILES_SHOW_CMD, oldpath, newpath))
4042 return REQ_QUIT;
4043 if (status)
4044 info = "Unstaged changes to %s";
4045 else
4046 info = "Unstaged changes";
4047 break;
4048
4049 case LINE_STAT_UNTRACKED:
4050 if (opt_pipe)
4051 return REQ_QUIT;
4052
4053
4054 if (!status) {
4055 report("No file to show");
4056 return REQ_NONE;
4057 }
4058
4059 opt_pipe = fopen(status->new.name, "r");
4060 info = "Untracked file %s";
4061 break;
4062
4063 default:
4064 die("line type %d not handled in switch", line->type);
4065 }
4066
4067 open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | OPEN_SPLIT);
4068 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
4069 if (status) {
4070 stage_status = *status;
4071 } else {
4072 memset(&stage_status, 0, sizeof(stage_status));
4073 }
4074
4075 stage_line_type = line->type;
4076 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
4077 }
4078
4079 return REQ_NONE;
4080 }
4081
4082
4083 static bool
4084 status_update_file(struct view *view, struct status *status, enum line_type type)
4085 {
4086 char cmd[SIZEOF_STR];
4087 char buf[SIZEOF_STR];
4088 size_t cmdsize = 0;
4089 size_t bufsize = 0;
4090 size_t written = 0;
4091 FILE *pipe;
4092
4093 if (opt_cdup[0] &&
4094 type != LINE_STAT_UNTRACKED &&
4095 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4096 return FALSE;
4097
4098 switch (type) {
4099 case LINE_STAT_STAGED:
4100 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
4101 status->old.mode,
4102 status->old.rev,
4103 status->old.name, 0))
4104 return FALSE;
4105
4106 string_add(cmd, cmdsize, "git update-index -z --index-info");
4107 break;
4108
4109 case LINE_STAT_UNSTAGED:
4110 case LINE_STAT_UNTRACKED:
4111 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
4112 return FALSE;
4113
4114 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
4115 break;
4116
4117 default:
4118 die("line type %d not handled in switch", type);
4119 }
4120
4121 pipe = popen(cmd, "w");
4122 if (!pipe)
4123 return FALSE;
4124
4125 while (!ferror(pipe) && written < bufsize) {
4126 written += fwrite(buf + written, 1, bufsize - written, pipe);
4127 }
4128
4129 pclose(pipe);
4130
4131 if (written != bufsize)
4132 return FALSE;
4133
4134 return TRUE;
4135 }
4136
4137 static void
4138 status_update(struct view *view)
4139 {
4140 struct line *line = &view->line[view->lineno];
4141
4142 assert(view->lines);
4143
4144 if (!line->data) {
4145 while (++line < view->line + view->lines && line->data) {
4146 if (!status_update_file(view, line->data, line->type))
4147 report("Failed to update file status");
4148 }
4149
4150 if (!line[-1].data) {
4151 report("Nothing to update");
4152 return;
4153 }
4154
4155 } else if (!status_update_file(view, line->data, line->type)) {
4156 report("Failed to update file status");
4157 }
4158 }
4159
4160 static enum request
4161 status_request(struct view *view, enum request request, struct line *line)
4162 {
4163 struct status *status = line->data;
4164
4165 switch (request) {
4166 case REQ_STATUS_UPDATE:
4167 status_update(view);
4168 break;
4169
4170 case REQ_STATUS_MERGE:
4171 if (!status || status->status != 'U') {
4172 report("Merging only possible for files with unmerged status ('U').");
4173 return REQ_NONE;
4174 }
4175 open_mergetool(status->new.name);
4176 break;
4177
4178 case REQ_EDIT:
4179 if (!status)
4180 return request;
4181
4182 open_editor(status->status != '?', status->new.name);
4183 break;
4184
4185 case REQ_VIEW_BLAME:
4186 if (status) {
4187 string_copy(opt_file, status->new.name);
4188 opt_ref[0] = 0;
4189 }
4190 return request;
4191
4192 case REQ_ENTER:
4193 /* After returning the status view has been split to
4194 * show the stage view. No further reloading is
4195 * necessary. */
4196 status_enter(view, line);
4197 return REQ_NONE;
4198
4199 case REQ_REFRESH:
4200 /* Simply reload the view. */
4201 break;
4202
4203 default:
4204 return request;
4205 }
4206
4207 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
4208
4209 return REQ_NONE;
4210 }
4211
4212 static void
4213 status_select(struct view *view, struct line *line)
4214 {
4215 struct status *status = line->data;
4216 char file[SIZEOF_STR] = "all files";
4217 char *text;
4218 char *key;
4219
4220 if (status && !string_format(file, "'%s'", status->new.name))
4221 return;
4222
4223 if (!status && line[1].type == LINE_STAT_NONE)
4224 line++;
4225
4226 switch (line->type) {
4227 case LINE_STAT_STAGED:
4228 text = "Press %s to unstage %s for commit";
4229 break;
4230
4231 case LINE_STAT_UNSTAGED:
4232 text = "Press %s to stage %s for commit";
4233 break;
4234
4235 case LINE_STAT_UNTRACKED:
4236 text = "Press %s to stage %s for addition";
4237 break;
4238
4239 case LINE_STAT_NONE:
4240 text = "Nothing to update";
4241 break;
4242
4243 default:
4244 die("line type %d not handled in switch", line->type);
4245 }
4246
4247 if (status && status->status == 'U') {
4248 text = "Press %s to resolve conflict in %s";
4249 key = get_key(REQ_STATUS_MERGE);
4250
4251 } else {
4252 key = get_key(REQ_STATUS_UPDATE);
4253 }
4254
4255 string_format(view->ref, text, key, file);
4256 }
4257
4258 static bool
4259 status_grep(struct view *view, struct line *line)
4260 {
4261 struct status *status = line->data;
4262 enum { S_STATUS, S_NAME, S_END } state;
4263 char buf[2] = "?";
4264 regmatch_t pmatch;
4265
4266 if (!status)
4267 return FALSE;
4268
4269 for (state = S_STATUS; state < S_END; state++) {
4270 char *text;
4271
4272 switch (state) {
4273 case S_NAME: text = status->new.name; break;
4274 case S_STATUS:
4275 buf[0] = status->status;
4276 text = buf;
4277 break;
4278
4279 default:
4280 return FALSE;
4281 }
4282
4283 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4284 return TRUE;
4285 }
4286
4287 return FALSE;
4288 }
4289
4290 static struct view_ops status_ops = {
4291 "file",
4292 status_open,
4293 NULL,
4294 status_draw,
4295 status_request,
4296 status_grep,
4297 status_select,
4298 };
4299
4300
4301 static bool
4302 stage_diff_line(FILE *pipe, struct line *line)
4303 {
4304 char *buf = line->data;
4305 size_t bufsize = strlen(buf);
4306 size_t written = 0;
4307
4308 while (!ferror(pipe) && written < bufsize) {
4309 written += fwrite(buf + written, 1, bufsize - written, pipe);
4310 }
4311
4312 fputc('\n', pipe);
4313
4314 return written == bufsize;
4315 }
4316
4317 static struct line *
4318 stage_diff_hdr(struct view *view, struct line *line)
4319 {
4320 int diff_hdr_dir = line->type == LINE_DIFF_CHUNK ? -1 : 1;
4321 struct line *diff_hdr;
4322
4323 if (line->type == LINE_DIFF_CHUNK)
4324 diff_hdr = line - 1;
4325 else
4326 diff_hdr = view->line + 1;
4327
4328 while (diff_hdr > view->line && diff_hdr < view->line + view->lines) {
4329 if (diff_hdr->type == LINE_DIFF_HEADER)
4330 return diff_hdr;
4331
4332 diff_hdr += diff_hdr_dir;
4333 }
4334
4335 return NULL;
4336 }
4337
4338 static bool
4339 stage_update_chunk(struct view *view, struct line *line)
4340 {
4341 char cmd[SIZEOF_STR];
4342 size_t cmdsize = 0;
4343 struct line *diff_hdr, *diff_chunk, *diff_end;
4344 FILE *pipe;
4345
4346 diff_hdr = stage_diff_hdr(view, line);
4347 if (!diff_hdr)
4348 return FALSE;
4349
4350 if (opt_cdup[0] &&
4351 !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
4352 return FALSE;
4353
4354 if (!string_format_from(cmd, &cmdsize,
4355 "git apply --cached %s - && "
4356 "git update-index -q --unmerged --refresh 2>/dev/null",
4357 stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
4358 return FALSE;
4359
4360 pipe = popen(cmd, "w");
4361 if (!pipe)
4362 return FALSE;
4363
4364 diff_end = view->line + view->lines;
4365 if (line->type != LINE_DIFF_CHUNK) {
4366 diff_chunk = diff_hdr;
4367
4368 } else {
4369 for (diff_chunk = line + 1; diff_chunk < diff_end; diff_chunk++)
4370 if (diff_chunk->type == LINE_DIFF_CHUNK ||
4371 diff_chunk->type == LINE_DIFF_HEADER)
4372 diff_end = diff_chunk;
4373
4374 diff_chunk = line;
4375
4376 while (diff_hdr->type != LINE_DIFF_CHUNK) {
4377 switch (diff_hdr->type) {
4378 case LINE_DIFF_HEADER:
4379 case LINE_DIFF_INDEX:
4380 case LINE_DIFF_ADD:
4381 case LINE_DIFF_DEL:
4382 break;
4383
4384 default:
4385 diff_hdr++;
4386 continue;
4387 }
4388
4389 if (!stage_diff_line(pipe, diff_hdr++)) {
4390 pclose(pipe);
4391 return FALSE;
4392 }
4393 }
4394 }
4395
4396 while (diff_chunk < diff_end && stage_diff_line(pipe, diff_chunk))
4397 diff_chunk++;
4398
4399 pclose(pipe);
4400
4401 if (diff_chunk != diff_end)
4402 return FALSE;
4403
4404 return TRUE;
4405 }
4406
4407 static void
4408 stage_update(struct view *view, struct line *line)
4409 {
4410 if (stage_line_type != LINE_STAT_UNTRACKED &&
4411 (line->type == LINE_DIFF_CHUNK || !stage_status.status)) {
4412 if (!stage_update_chunk(view, line)) {
4413 report("Failed to apply chunk");
4414 return;
4415 }
4416
4417 } else if (!status_update_file(view, &stage_status, stage_line_type)) {
4418 report("Failed to update file");
4419 return;
4420 }
4421
4422 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
4423
4424 view = VIEW(REQ_VIEW_STATUS);
4425 if (view_is_displayed(view))
4426 status_enter(view, &view->line[view->lineno]);
4427 }
4428
4429 static enum request
4430 stage_request(struct view *view, enum request request, struct line *line)
4431 {
4432 switch (request) {
4433 case REQ_STATUS_UPDATE:
4434 stage_update(view, line);
4435 break;
4436
4437 case REQ_EDIT:
4438 if (!stage_status.new.name[0])
4439 return request;
4440
4441 open_editor(stage_status.status != '?', stage_status.new.name);
4442 break;
4443
4444 case REQ_VIEW_BLAME:
4445 if (stage_status.new.name[0]) {
4446 string_copy(opt_file, stage_status.new.name);
4447 opt_ref[0] = 0;
4448 }
4449 return request;
4450
4451 case REQ_ENTER:
4452 pager_request(view, request, line);
4453 break;
4454
4455 default:
4456 return request;
4457 }
4458
4459 return REQ_NONE;
4460 }
4461
4462 static struct view_ops stage_ops = {
4463 "line",
4464 NULL,
4465 pager_read,
4466 pager_draw,
4467 stage_request,
4468 pager_grep,
4469 pager_select,
4470 };
4471
4472
4473 /*
4474 * Revision graph
4475 */
4476
4477 struct commit {
4478 char id[SIZEOF_REV]; /* SHA1 ID. */
4479 char title[128]; /* First line of the commit message. */
4480 char author[75]; /* Author of the commit. */
4481 struct tm time; /* Date from the author ident. */
4482 struct ref **refs; /* Repository references. */
4483 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
4484 size_t graph_size; /* The width of the graph array. */
4485 bool has_parents; /* Rewritten --parents seen. */
4486 };
4487
4488 /* Size of rev graph with no "padding" columns */
4489 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
4490
4491 struct rev_graph {
4492 struct rev_graph *prev, *next, *parents;
4493 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
4494 size_t size;
4495 struct commit *commit;
4496 size_t pos;
4497 unsigned int boundary:1;
4498 };
4499
4500 /* Parents of the commit being visualized. */
4501 static struct rev_graph graph_parents[4];
4502
4503 /* The current stack of revisions on the graph. */
4504 static struct rev_graph graph_stacks[4] = {
4505 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
4506 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
4507 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
4508 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
4509 };
4510
4511 static inline bool
4512 graph_parent_is_merge(struct rev_graph *graph)
4513 {
4514 return graph->parents->size > 1;
4515 }
4516
4517 static inline void
4518 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
4519 {
4520 struct commit *commit = graph->commit;
4521
4522 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
4523 commit->graph[commit->graph_size++] = symbol;
4524 }
4525
4526 static void
4527 done_rev_graph(struct rev_graph *graph)
4528 {
4529 if (graph_parent_is_merge(graph) &&
4530 graph->pos < graph->size - 1 &&
4531 graph->next->size == graph->size + graph->parents->size - 1) {
4532 size_t i = graph->pos + graph->parents->size - 1;
4533
4534 graph->commit->graph_size = i * 2;
4535 while (i < graph->next->size - 1) {
4536 append_to_rev_graph(graph, ' ');
4537 append_to_rev_graph(graph, '\\');
4538 i++;
4539 }
4540 }
4541
4542 graph->size = graph->pos = 0;
4543 graph->commit = NULL;
4544 memset(graph->parents, 0, sizeof(*graph->parents));
4545 }
4546
4547 static void
4548 push_rev_graph(struct rev_graph *graph, char *parent)
4549 {
4550 int i;
4551
4552 /* "Collapse" duplicate parents lines.
4553 *
4554 * FIXME: This needs to also update update the drawn graph but
4555 * for now it just serves as a method for pruning graph lines. */
4556 for (i = 0; i < graph->size; i++)
4557 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
4558 return;
4559
4560 if (graph->size < SIZEOF_REVITEMS) {
4561 string_copy_rev(graph->rev[graph->size++], parent);
4562 }
4563 }
4564
4565 static chtype
4566 get_rev_graph_symbol(struct rev_graph *graph)
4567 {
4568 chtype symbol;
4569
4570 if (graph->boundary)
4571 symbol = REVGRAPH_BOUND;
4572 else if (graph->parents->size == 0)
4573 symbol = REVGRAPH_INIT;
4574 else if (graph_parent_is_merge(graph))
4575 symbol = REVGRAPH_MERGE;
4576 else if (graph->pos >= graph->size)
4577 symbol = REVGRAPH_BRANCH;
4578 else
4579 symbol = REVGRAPH_COMMIT;
4580
4581 return symbol;
4582 }
4583
4584 static void
4585 draw_rev_graph(struct rev_graph *graph)
4586 {
4587 struct rev_filler {
4588 chtype separator, line;
4589 };
4590 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
4591 static struct rev_filler fillers[] = {
4592 { ' ', REVGRAPH_LINE },
4593 { '`', '.' },
4594 { '\'', ' ' },
4595 { '/', ' ' },
4596 };
4597 chtype symbol = get_rev_graph_symbol(graph);
4598 struct rev_filler *filler;
4599 size_t i;
4600
4601 filler = &fillers[DEFAULT];
4602
4603 for (i = 0; i < graph->pos; i++) {
4604 append_to_rev_graph(graph, filler->line);
4605 if (graph_parent_is_merge(graph->prev) &&
4606 graph->prev->pos == i)
4607 filler = &fillers[RSHARP];
4608
4609 append_to_rev_graph(graph, filler->separator);
4610 }
4611
4612 /* Place the symbol for this revision. */
4613 append_to_rev_graph(graph, symbol);
4614
4615 if (graph->prev->size > graph->size)
4616 filler = &fillers[RDIAG];
4617 else
4618 filler = &fillers[DEFAULT];
4619
4620 i++;
4621
4622 for (; i < graph->size; i++) {
4623 append_to_rev_graph(graph, filler->separator);
4624 append_to_rev_graph(graph, filler->line);
4625 if (graph_parent_is_merge(graph->prev) &&
4626 i < graph->prev->pos + graph->parents->size)
4627 filler = &fillers[RSHARP];
4628 if (graph->prev->size > graph->size)
4629 filler = &fillers[LDIAG];
4630 }
4631
4632 if (graph->prev->size > graph->size) {
4633 append_to_rev_graph(graph, filler->separator);
4634 if (filler->line != ' ')
4635 append_to_rev_graph(graph, filler->line);
4636 }
4637 }
4638
4639 /* Prepare the next rev graph */
4640 static void
4641 prepare_rev_graph(struct rev_graph *graph)
4642 {
4643 size_t i;
4644
4645 /* First, traverse all lines of revisions up to the active one. */
4646 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
4647 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
4648 break;
4649
4650 push_rev_graph(graph->next, graph->rev[graph->pos]);
4651 }
4652
4653 /* Interleave the new revision parent(s). */
4654 for (i = 0; !graph->boundary && i < graph->parents->size; i++)
4655 push_rev_graph(graph->next, graph->parents->rev[i]);
4656
4657 /* Lastly, put any remaining revisions. */
4658 for (i = graph->pos + 1; i < graph->size; i++)
4659 push_rev_graph(graph->next, graph->rev[i]);
4660 }
4661
4662 static void
4663 update_rev_graph(struct rev_graph *graph)
4664 {
4665 /* If this is the finalizing update ... */
4666 if (graph->commit)
4667 prepare_rev_graph(graph);
4668
4669 /* Graph visualization needs a one rev look-ahead,
4670 * so the first update doesn't visualize anything. */
4671 if (!graph->prev->commit)
4672 return;
4673
4674 draw_rev_graph(graph->prev);
4675 done_rev_graph(graph->prev->prev);
4676 }
4677
4678
4679 /*
4680 * Main view backend
4681 */
4682
4683 static bool
4684 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
4685 {
4686 char buf[DATE_COLS + 1];
4687 struct commit *commit = line->data;
4688 enum line_type type;
4689 int col = 0;
4690 size_t timelen;
4691 int tilde_attr;
4692 int space;
4693
4694 if (!*commit->author)
4695 return FALSE;
4696
4697 space = view->width;
4698 wmove(view->win, lineno, col);
4699
4700 if (selected) {
4701 type = LINE_CURSOR;
4702 wattrset(view->win, get_line_attr(type));
4703 wchgat(view->win, -1, 0, type, NULL);
4704 tilde_attr = -1;
4705 } else {
4706 type = LINE_MAIN_COMMIT;
4707 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
4708 tilde_attr = get_line_attr(LINE_MAIN_DELIM);
4709 }
4710
4711 if (opt_date) {
4712 int n;
4713
4714 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
4715 n = draw_text(view, buf, view->width - col, FALSE, tilde_attr);
4716 draw_text(view, " ", view->width - col - n, FALSE, tilde_attr);
4717
4718 col += DATE_COLS;
4719 wmove(view->win, lineno, col);
4720 if (col >= view->width)
4721 return TRUE;
4722 }
4723 if (type != LINE_CURSOR)
4724 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
4725
4726 if (opt_author) {
4727 int max_len;
4728
4729 max_len = view->width - col;
4730 if (max_len > AUTHOR_COLS - 1)
4731 max_len = AUTHOR_COLS - 1;
4732 draw_text(view, commit->author, max_len, TRUE, tilde_attr);
4733 col += AUTHOR_COLS;
4734 if (col >= view->width)
4735 return TRUE;
4736 }
4737
4738 if (opt_rev_graph && commit->graph_size) {
4739 size_t graph_size = view->width - col;
4740 size_t i;
4741
4742 if (type != LINE_CURSOR)
4743 wattrset(view->win, get_line_attr(LINE_MAIN_REVGRAPH));
4744 wmove(view->win, lineno, col);
4745 if (graph_size > commit->graph_size)
4746 graph_size = commit->graph_size;
4747 /* Using waddch() instead of waddnstr() ensures that
4748 * they'll be rendered correctly for the cursor line. */
4749 for (i = 0; i < graph_size; i++)
4750 waddch(view->win, commit->graph[i]);
4751
4752 col += commit->graph_size + 1;
4753 if (col >= view->width)
4754 return TRUE;
4755 waddch(view->win, ' ');
4756 }
4757 if (type != LINE_CURSOR)
4758 wattrset(view->win, A_NORMAL);
4759
4760 wmove(view->win, lineno, col);
4761
4762 if (opt_show_refs && commit->refs) {
4763 size_t i = 0;
4764
4765 do {
4766 if (type == LINE_CURSOR)
4767 ;
4768 else if (commit->refs[i]->head)
4769 wattrset(view->win, get_line_attr(LINE_MAIN_HEAD));
4770 else if (commit->refs[i]->ltag)
4771 wattrset(view->win, get_line_attr(LINE_MAIN_LOCAL_TAG));
4772 else if (commit->refs[i]->tag)
4773 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
4774 else if (commit->refs[i]->remote)
4775 wattrset(view->win, get_line_attr(LINE_MAIN_REMOTE));
4776 else
4777 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
4778
4779 col += draw_text(view, "[", view->width - col, TRUE, tilde_attr);
4780 col += draw_text(view, commit->refs[i]->name, view->width - col,
4781 TRUE, tilde_attr);
4782 col += draw_text(view, "]", view->width - col, TRUE, tilde_attr);
4783 if (type != LINE_CURSOR)
4784 wattrset(view->win, A_NORMAL);
4785 col += draw_text(view, " ", view->width - col, TRUE, tilde_attr);
4786 if (col >= view->width)
4787 return TRUE;
4788 } while (commit->refs[i++]->next);
4789 }
4790
4791 if (type != LINE_CURSOR)
4792 wattrset(view->win, get_line_attr(type));
4793
4794 draw_text(view, commit->title, view->width - col, TRUE, tilde_attr);
4795 return TRUE;
4796 }
4797
4798 /* Reads git log --pretty=raw output and parses it into the commit struct. */
4799 static bool
4800 main_read(struct view *view, char *line)
4801 {
4802 static struct rev_graph *graph = graph_stacks;
4803 enum line_type type;
4804 struct commit *commit;
4805
4806 if (!line) {
4807 update_rev_graph(graph);
4808 return TRUE;
4809 }
4810
4811 type = get_line_type(line);
4812 if (type == LINE_COMMIT) {
4813 commit = calloc(1, sizeof(struct commit));
4814 if (!commit)
4815 return FALSE;
4816
4817 line += STRING_SIZE("commit ");
4818 if (*line == '-') {
4819 graph->boundary = 1;
4820 line++;
4821 }
4822
4823 string_copy_rev(commit->id, line);
4824 commit->refs = get_refs(commit->id);
4825 graph->commit = commit;
4826 add_line_data(view, commit, LINE_MAIN_COMMIT);
4827
4828 while ((line = strchr(line, ' '))) {
4829 line++;
4830 push_rev_graph(graph->parents, line);
4831 commit->has_parents = TRUE;
4832 }
4833 return TRUE;
4834 }
4835
4836 if (!view->lines)
4837 return TRUE;
4838 commit = view->line[view->lines - 1].data;
4839
4840 switch (type) {
4841 case LINE_PARENT:
4842 if (commit->has_parents)
4843 break;
4844 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
4845 break;
4846
4847 case LINE_AUTHOR:
4848 {
4849 /* Parse author lines where the name may be empty:
4850 * author <email@address.tld> 1138474660 +0100
4851 */
4852 char *ident = line + STRING_SIZE("author ");
4853 char *nameend = strchr(ident, '<');
4854 char *emailend = strchr(ident, '>');
4855
4856 if (!nameend || !emailend)
4857 break;
4858
4859 update_rev_graph(graph);
4860 graph = graph->next;
4861
4862 *nameend = *emailend = 0;
4863 ident = chomp_string(ident);
4864 if (!*ident) {
4865 ident = chomp_string(nameend + 1);
4866 if (!*ident)
4867 ident = "Unknown";
4868 }
4869
4870 string_ncopy(commit->author, ident, strlen(ident));
4871
4872 /* Parse epoch and timezone */
4873 if (emailend[1] == ' ') {
4874 char *secs = emailend + 2;
4875 char *zone = strchr(secs, ' ');
4876 time_t time = (time_t) atol(secs);
4877
4878 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
4879 long tz;
4880
4881 zone++;
4882 tz = ('0' - zone[1]) * 60 * 60 * 10;
4883 tz += ('0' - zone[2]) * 60 * 60;
4884 tz += ('0' - zone[3]) * 60;
4885 tz += ('0' - zone[4]) * 60;
4886
4887 if (zone[0] == '-')
4888 tz = -tz;
4889
4890 time -= tz;
4891 }
4892
4893 gmtime_r(&time, &commit->time);
4894 }
4895 break;
4896 }
4897 default:
4898 /* Fill in the commit title if it has not already been set. */
4899 if (commit->title[0])
4900 break;
4901
4902 /* Require titles to start with a non-space character at the
4903 * offset used by git log. */
4904 if (strncmp(line, " ", 4))
4905 break;
4906 line += 4;
4907 /* Well, if the title starts with a whitespace character,
4908 * try to be forgiving. Otherwise we end up with no title. */
4909 while (isspace(*line))
4910 line++;
4911 if (*line == '\0')
4912 break;
4913 /* FIXME: More graceful handling of titles; append "..." to
4914 * shortened titles, etc. */
4915
4916 string_ncopy(commit->title, line, strlen(line));
4917 }
4918
4919 return TRUE;
4920 }
4921
4922 static enum request
4923 main_request(struct view *view, enum request request, struct line *line)
4924 {
4925 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4926
4927 if (request == REQ_ENTER)
4928 open_view(view, REQ_VIEW_DIFF, flags);
4929 else
4930 return request;
4931
4932 return REQ_NONE;
4933 }
4934
4935 static bool
4936 main_grep(struct view *view, struct line *line)
4937 {
4938 struct commit *commit = line->data;
4939 enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
4940 char buf[DATE_COLS + 1];
4941 regmatch_t pmatch;
4942
4943 for (state = S_TITLE; state < S_END; state++) {
4944 char *text;
4945
4946 switch (state) {
4947 case S_TITLE: text = commit->title; break;
4948 case S_AUTHOR: text = commit->author; break;
4949 case S_DATE:
4950 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
4951 continue;
4952 text = buf;
4953 break;
4954
4955 default:
4956 return FALSE;
4957 }
4958
4959 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4960 return TRUE;
4961 }
4962
4963 return FALSE;
4964 }
4965
4966 static void
4967 main_select(struct view *view, struct line *line)
4968 {
4969 struct commit *commit = line->data;
4970
4971 string_copy_rev(view->ref, commit->id);
4972 string_copy_rev(ref_commit, view->ref);
4973 }
4974
4975 static struct view_ops main_ops = {
4976 "commit",
4977 NULL,
4978 main_read,
4979 main_draw,
4980 main_request,
4981 main_grep,
4982 main_select,
4983 };
4984
4985
4986 /*
4987 * Unicode / UTF-8 handling
4988 *
4989 * NOTE: Much of the following code for dealing with unicode is derived from
4990 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
4991 * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
4992 */
4993
4994 /* I've (over)annotated a lot of code snippets because I am not entirely
4995 * confident that the approach taken by this small UTF-8 interface is correct.
4996 * --jonas */
4997
4998 static inline int
4999 unicode_width(unsigned long c)
5000 {
5001 if (c >= 0x1100 &&
5002 (c <= 0x115f /* Hangul Jamo */
5003 || c == 0x2329
5004 || c == 0x232a
5005 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
5006 /* CJK ... Yi */
5007 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
5008 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
5009 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
5010 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
5011 || (c >= 0xffe0 && c <= 0xffe6)
5012 || (c >= 0x20000 && c <= 0x2fffd)
5013 || (c >= 0x30000 && c <= 0x3fffd)))
5014 return 2;
5015
5016 if (c == '\t')
5017 return opt_tab_size;
5018
5019 return 1;
5020 }
5021
5022 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
5023 * Illegal bytes are set one. */
5024 static const unsigned char utf8_bytes[256] = {
5025 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,
5026 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,
5027 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,
5028 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,
5029 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,
5030 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,
5031 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,
5032 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,
5033 };
5034
5035 /* Decode UTF-8 multi-byte representation into a unicode character. */
5036 static inline unsigned long
5037 utf8_to_unicode(const char *string, size_t length)
5038 {
5039 unsigned long unicode;
5040
5041 switch (length) {
5042 case 1:
5043 unicode = string[0];
5044 break;
5045 case 2:
5046 unicode = (string[0] & 0x1f) << 6;
5047 unicode += (string[1] & 0x3f);
5048 break;
5049 case 3:
5050 unicode = (string[0] & 0x0f) << 12;
5051 unicode += ((string[1] & 0x3f) << 6);
5052 unicode += (string[2] & 0x3f);
5053 break;
5054 case 4:
5055 unicode = (string[0] & 0x0f) << 18;
5056 unicode += ((string[1] & 0x3f) << 12);
5057 unicode += ((string[2] & 0x3f) << 6);
5058 unicode += (string[3] & 0x3f);
5059 break;
5060 case 5:
5061 unicode = (string[0] & 0x0f) << 24;
5062 unicode += ((string[1] & 0x3f) << 18);
5063 unicode += ((string[2] & 0x3f) << 12);
5064 unicode += ((string[3] & 0x3f) << 6);
5065 unicode += (string[4] & 0x3f);
5066 break;
5067 case 6:
5068 unicode = (string[0] & 0x01) << 30;
5069 unicode += ((string[1] & 0x3f) << 24);
5070 unicode += ((string[2] & 0x3f) << 18);
5071 unicode += ((string[3] & 0x3f) << 12);
5072 unicode += ((string[4] & 0x3f) << 6);
5073 unicode += (string[5] & 0x3f);
5074 break;
5075 default:
5076 die("Invalid unicode length");
5077 }
5078
5079 /* Invalid characters could return the special 0xfffd value but NUL
5080 * should be just as good. */
5081 return unicode > 0xffff ? 0 : unicode;
5082 }
5083
5084 /* Calculates how much of string can be shown within the given maximum width
5085 * and sets trimmed parameter to non-zero value if all of string could not be
5086 * shown. If the reserve flag is TRUE, it will reserve at least one
5087 * trailing character, which can be useful when drawing a delimiter.
5088 *
5089 * Returns the number of bytes to output from string to satisfy max_width. */
5090 static size_t
5091 utf8_length(const char *string, size_t max_width, int *trimmed, bool reserve)
5092 {
5093 const char *start = string;
5094 const char *end = strchr(string, '\0');
5095 unsigned char last_bytes = 0;
5096 size_t width = 0;
5097
5098 *trimmed = 0;
5099
5100 while (string < end) {
5101 int c = *(unsigned char *) string;
5102 unsigned char bytes = utf8_bytes[c];
5103 size_t ucwidth;
5104 unsigned long unicode;
5105
5106 if (string + bytes > end)
5107 break;
5108
5109 /* Change representation to figure out whether
5110 * it is a single- or double-width character. */
5111
5112 unicode = utf8_to_unicode(string, bytes);
5113 /* FIXME: Graceful handling of invalid unicode character. */
5114 if (!unicode)
5115 break;
5116
5117 ucwidth = unicode_width(unicode);
5118 width += ucwidth;
5119 if (width > max_width) {
5120 *trimmed = 1;
5121 if (reserve && width - ucwidth == max_width) {
5122 string -= last_bytes;
5123 }
5124 break;
5125 }
5126
5127 string += bytes;
5128 last_bytes = bytes;
5129 }
5130
5131 return string - start;
5132 }
5133
5134
5135 /*
5136 * Status management
5137 */
5138
5139 /* Whether or not the curses interface has been initialized. */
5140 static bool cursed = FALSE;
5141
5142 /* The status window is used for polling keystrokes. */
5143 static WINDOW *status_win;
5144
5145 static bool status_empty = TRUE;
5146
5147 /* Update status and title window. */
5148 static void
5149 report(const char *msg, ...)
5150 {
5151 struct view *view = display[current_view];
5152
5153 if (input_mode)
5154 return;
5155
5156 if (!view) {
5157 char buf[SIZEOF_STR];
5158 va_list args;
5159
5160 va_start(args, msg);
5161 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
5162 buf[sizeof(buf) - 1] = 0;
5163 buf[sizeof(buf) - 2] = '.';
5164 buf[sizeof(buf) - 3] = '.';
5165 buf[sizeof(buf) - 4] = '.';
5166 }
5167 va_end(args);
5168 die("%s", buf);
5169 }
5170
5171 if (!status_empty || *msg) {
5172 va_list args;
5173
5174 va_start(args, msg);
5175
5176 wmove(status_win, 0, 0);
5177 if (*msg) {
5178 vwprintw(status_win, msg, args);
5179 status_empty = FALSE;
5180 } else {
5181 status_empty = TRUE;
5182 }
5183 wclrtoeol(status_win);
5184 wrefresh(status_win);
5185
5186 va_end(args);
5187 }
5188
5189 update_view_title(view);
5190 update_display_cursor(view);
5191 }
5192
5193 /* Controls when nodelay should be in effect when polling user input. */
5194 static void
5195 set_nonblocking_input(bool loading)
5196 {
5197 static unsigned int loading_views;
5198
5199 if ((loading == FALSE && loading_views-- == 1) ||
5200 (loading == TRUE && loading_views++ == 0))
5201 nodelay(status_win, loading);
5202 }
5203
5204 static void
5205 init_display(void)
5206 {
5207 int x, y;
5208
5209 /* Initialize the curses library */
5210 if (isatty(STDIN_FILENO)) {
5211 cursed = !!initscr();
5212 } else {
5213 /* Leave stdin and stdout alone when acting as a pager. */
5214 FILE *io = fopen("/dev/tty", "r+");
5215
5216 if (!io)
5217 die("Failed to open /dev/tty");
5218 cursed = !!newterm(NULL, io, io);
5219 }
5220
5221 if (!cursed)
5222 die("Failed to initialize curses");
5223
5224 nonl(); /* Tell curses not to do NL->CR/NL on output */
5225 cbreak(); /* Take input chars one at a time, no wait for \n */
5226 noecho(); /* Don't echo input */
5227 leaveok(stdscr, TRUE);
5228
5229 if (has_colors())
5230 init_colors();
5231
5232 getmaxyx(stdscr, y, x);
5233 status_win = newwin(1, 0, y - 1, 0);
5234 if (!status_win)
5235 die("Failed to create status window");
5236
5237 /* Enable keyboard mapping */
5238 keypad(status_win, TRUE);
5239 wbkgdset(status_win, get_line_attr(LINE_STATUS));
5240 }
5241
5242 static char *
5243 read_prompt(const char *prompt)
5244 {
5245 enum { READING, STOP, CANCEL } status = READING;
5246 static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
5247 int pos = 0;
5248
5249 while (status == READING) {
5250 struct view *view;
5251 int i, key;
5252
5253 input_mode = TRUE;
5254
5255 foreach_view (view, i)
5256 update_view(view);
5257
5258 input_mode = FALSE;
5259
5260 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
5261 wclrtoeol(status_win);
5262
5263 /* Refresh, accept single keystroke of input */
5264 key = wgetch(status_win);
5265 switch (key) {
5266 case KEY_RETURN:
5267 case KEY_ENTER:
5268 case '\n':
5269 status = pos ? STOP : CANCEL;
5270 break;
5271
5272 case KEY_BACKSPACE:
5273 if (pos > 0)
5274 pos--;
5275 else
5276 status = CANCEL;
5277 break;
5278
5279 case KEY_ESC:
5280 status = CANCEL;
5281 break;
5282
5283 case ERR:
5284 break;
5285
5286 default:
5287 if (pos >= sizeof(buf)) {
5288 report("Input string too long");
5289 return NULL;
5290 }
5291
5292 if (isprint(key))
5293 buf[pos++] = (char) key;
5294 }
5295 }
5296
5297 /* Clear the status window */
5298 status_empty = FALSE;
5299 report("");
5300
5301 if (status == CANCEL)
5302 return NULL;
5303
5304 buf[pos++] = 0;
5305
5306 return buf;
5307 }
5308
5309 /*
5310 * Repository references
5311 */
5312
5313 static struct ref *refs = NULL;
5314 static size_t refs_alloc = 0;
5315 static size_t refs_size = 0;
5316
5317 /* Id <-> ref store */
5318 static struct ref ***id_refs = NULL;
5319 static size_t id_refs_alloc = 0;
5320 static size_t id_refs_size = 0;
5321
5322 static struct ref **
5323 get_refs(char *id)
5324 {
5325 struct ref ***tmp_id_refs;
5326 struct ref **ref_list = NULL;
5327 size_t ref_list_alloc = 0;
5328 size_t ref_list_size = 0;
5329 size_t i;
5330
5331 for (i = 0; i < id_refs_size; i++)
5332 if (!strcmp(id, id_refs[i][0]->id))
5333 return id_refs[i];
5334
5335 tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
5336 sizeof(*id_refs));
5337 if (!tmp_id_refs)
5338 return NULL;
5339
5340 id_refs = tmp_id_refs;
5341
5342 for (i = 0; i < refs_size; i++) {
5343 struct ref **tmp;
5344
5345 if (strcmp(id, refs[i].id))
5346 continue;
5347
5348 tmp = realloc_items(ref_list, &ref_list_alloc,
5349 ref_list_size + 1, sizeof(*ref_list));
5350 if (!tmp) {
5351 if (ref_list)
5352 free(ref_list);
5353 return NULL;
5354 }
5355
5356 ref_list = tmp;
5357 if (ref_list_size > 0)
5358 ref_list[ref_list_size - 1]->next = 1;
5359 ref_list[ref_list_size] = &refs[i];
5360
5361 /* XXX: The properties of the commit chains ensures that we can
5362 * safely modify the shared ref. The repo references will
5363 * always be similar for the same id. */
5364 ref_list[ref_list_size]->next = 0;
5365 ref_list_size++;
5366 }
5367
5368 if (ref_list)
5369 id_refs[id_refs_size++] = ref_list;
5370
5371 return ref_list;
5372 }
5373
5374 static int
5375 read_ref(char *id, size_t idlen, char *name, size_t namelen)
5376 {
5377 struct ref *ref;
5378 bool tag = FALSE;
5379 bool ltag = FALSE;
5380 bool remote = FALSE;
5381 bool check_replace = FALSE;
5382 bool head = FALSE;
5383
5384 if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
5385 if (!strcmp(name + namelen - 3, "^{}")) {
5386 namelen -= 3;
5387 name[namelen] = 0;
5388 if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
5389 check_replace = TRUE;
5390 } else {
5391 ltag = TRUE;
5392 }
5393
5394 tag = TRUE;
5395 namelen -= STRING_SIZE("refs/tags/");
5396 name += STRING_SIZE("refs/tags/");
5397
5398 } else if (!strncmp(name, "refs/remotes/", STRING_SIZE("refs/remotes/"))) {
5399 remote = TRUE;
5400 namelen -= STRING_SIZE("refs/remotes/");
5401 name += STRING_SIZE("refs/remotes/");
5402
5403 } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
5404 namelen -= STRING_SIZE("refs/heads/");
5405 name += STRING_SIZE("refs/heads/");
5406 head = !strncmp(opt_head, name, namelen);
5407
5408 } else if (!strcmp(name, "HEAD")) {
5409 return OK;
5410 }
5411
5412 if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
5413 /* it's an annotated tag, replace the previous sha1 with the
5414 * resolved commit id; relies on the fact git-ls-remote lists
5415 * the commit id of an annotated tag right beofre the commit id
5416 * it points to. */
5417 refs[refs_size - 1].ltag = ltag;
5418 string_copy_rev(refs[refs_size - 1].id, id);
5419
5420 return OK;
5421 }
5422 refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
5423 if (!refs)
5424 return ERR;
5425
5426 ref = &refs[refs_size++];
5427 ref->name = malloc(namelen + 1);
5428 if (!ref->name)
5429 return ERR;
5430
5431 strncpy(ref->name, name, namelen);
5432 ref->name[namelen] = 0;
5433 ref->tag = tag;
5434 ref->ltag = ltag;
5435 ref->remote = remote;
5436 ref->head = head;
5437 string_copy_rev(ref->id, id);
5438
5439 return OK;
5440 }
5441
5442 static int
5443 load_refs(void)
5444 {
5445 const char *cmd_env = getenv("TIG_LS_REMOTE");
5446 const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
5447
5448 return read_properties(popen(cmd, "r"), "\t", read_ref);
5449 }
5450
5451 static int
5452 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
5453 {
5454 if (!strcmp(name, "i18n.commitencoding"))
5455 string_ncopy(opt_encoding, value, valuelen);
5456
5457 if (!strcmp(name, "core.editor"))
5458 string_ncopy(opt_editor, value, valuelen);
5459
5460 return OK;
5461 }
5462
5463 static int
5464 load_repo_config(void)
5465 {
5466 return read_properties(popen(GIT_CONFIG " --list", "r"),
5467 "=", read_repo_config_option);
5468 }
5469
5470 static int
5471 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
5472 {
5473 if (!opt_git_dir[0]) {
5474 string_ncopy(opt_git_dir, name, namelen);
5475
5476 } else if (opt_is_inside_work_tree == -1) {
5477 /* This can be 3 different values depending on the
5478 * version of git being used. If git-rev-parse does not
5479 * understand --is-inside-work-tree it will simply echo
5480 * the option else either "true" or "false" is printed.
5481 * Default to true for the unknown case. */
5482 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
5483
5484 } else if (opt_cdup[0] == ' ') {
5485 string_ncopy(opt_cdup, name, namelen);
5486 } else {
5487 if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
5488 namelen -= STRING_SIZE("refs/heads/");
5489 name += STRING_SIZE("refs/heads/");
5490 string_ncopy(opt_head, name, namelen);
5491 }
5492 }
5493
5494 return OK;
5495 }
5496
5497 static int
5498 load_repo_info(void)
5499 {
5500 int result;
5501 FILE *pipe = popen("git rev-parse --git-dir --is-inside-work-tree "
5502 " --show-cdup --symbolic-full-name HEAD 2>/dev/null", "r");
5503
5504 /* XXX: The line outputted by "--show-cdup" can be empty so
5505 * initialize it to something invalid to make it possible to
5506 * detect whether it has been set or not. */
5507 opt_cdup[0] = ' ';
5508
5509 result = read_properties(pipe, "=", read_repo_info);
5510 if (opt_cdup[0] == ' ')
5511 opt_cdup[0] = 0;
5512
5513 return result;
5514 }
5515
5516 static int
5517 read_properties(FILE *pipe, const char *separators,
5518 int (*read_property)(char *, size_t, char *, size_t))
5519 {
5520 char buffer[BUFSIZ];
5521 char *name;
5522 int state = OK;
5523
5524 if (!pipe)
5525 return ERR;
5526
5527 while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
5528 char *value;
5529 size_t namelen;
5530 size_t valuelen;
5531
5532 name = chomp_string(name);
5533 namelen = strcspn(name, separators);
5534
5535 if (name[namelen]) {
5536 name[namelen] = 0;
5537 value = chomp_string(name + namelen + 1);
5538 valuelen = strlen(value);
5539
5540 } else {
5541 value = "";
5542 valuelen = 0;
5543 }
5544
5545 state = read_property(name, namelen, value, valuelen);
5546 }
5547
5548 if (state != ERR && ferror(pipe))
5549 state = ERR;
5550
5551 pclose(pipe);
5552
5553 return state;
5554 }
5555
5556
5557 /*
5558 * Main
5559 */
5560
5561 static void __NORETURN
5562 quit(int sig)
5563 {
5564 /* XXX: Restore tty modes and let the OS cleanup the rest! */
5565 if (cursed)
5566 endwin();
5567 exit(0);
5568 }
5569
5570 static void __NORETURN
5571 die(const char *err, ...)
5572 {
5573 va_list args;
5574
5575 endwin();
5576
5577 va_start(args, err);
5578 fputs("tig: ", stderr);
5579 vfprintf(stderr, err, args);
5580 fputs("\n", stderr);
5581 va_end(args);
5582
5583 exit(1);
5584 }
5585
5586 static void
5587 warn(const char *msg, ...)
5588 {
5589 va_list args;
5590
5591 va_start(args, msg);
5592 fputs("tig warning: ", stderr);
5593 vfprintf(stderr, msg, args);
5594 fputs("\n", stderr);
5595 va_end(args);
5596 }
5597
5598 int
5599 main(int argc, char *argv[])
5600 {
5601 struct view *view;
5602 enum request request;
5603 size_t i;
5604
5605 signal(SIGINT, quit);
5606
5607 if (setlocale(LC_ALL, "")) {
5608 char *codeset = nl_langinfo(CODESET);
5609
5610 string_ncopy(opt_codeset, codeset, strlen(codeset));
5611 }
5612
5613 if (load_repo_info() == ERR)
5614 die("Failed to load repo info.");
5615
5616 if (load_options() == ERR)
5617 die("Failed to load user config.");
5618
5619 /* Load the repo config file so options can be overwritten from
5620 * the command line. */
5621 if (load_repo_config() == ERR)
5622 die("Failed to load repo config.");
5623
5624 if (!parse_options(argc, argv))
5625 return 0;
5626
5627 /* Require a git repository unless when running in pager mode. */
5628 if (!opt_git_dir[0])
5629 die("Not a git repository");
5630
5631 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
5632 opt_utf8 = FALSE;
5633
5634 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
5635 opt_iconv = iconv_open(opt_codeset, opt_encoding);
5636 if (opt_iconv == ICONV_NONE)
5637 die("Failed to initialize character set conversion");
5638 }
5639
5640 if (load_refs() == ERR)
5641 die("Failed to load refs.");
5642
5643 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
5644 view->cmd_env = getenv(view->cmd_env);
5645
5646 request = opt_request;
5647
5648 init_display();
5649
5650 while (view_driver(display[current_view], request)) {
5651 int key;
5652 int i;
5653
5654 foreach_view (view, i)
5655 update_view(view);
5656
5657 /* Refresh, accept single keystroke of input */
5658 key = wgetch(status_win);
5659
5660 /* wgetch() with nodelay() enabled returns ERR when there's no
5661 * input. */
5662 if (key == ERR) {
5663 request = REQ_NONE;
5664 continue;
5665 }
5666
5667 request = get_keybinding(display[current_view]->keymap, key);
5668
5669 /* Some low-level request handling. This keeps access to
5670 * status_win restricted. */
5671 switch (request) {
5672 case REQ_PROMPT:
5673 {
5674 char *cmd = read_prompt(":");
5675
5676 if (cmd && string_format(opt_cmd, "git %s", cmd)) {
5677 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
5678 opt_request = REQ_VIEW_DIFF;
5679 } else {
5680 opt_request = REQ_VIEW_PAGER;
5681 }
5682 break;
5683 }
5684
5685 request = REQ_NONE;
5686 break;
5687 }
5688 case REQ_SEARCH:
5689 case REQ_SEARCH_BACK:
5690 {
5691 const char *prompt = request == REQ_SEARCH
5692 ? "/" : "?";
5693 char *search = read_prompt(prompt);
5694
5695 if (search)
5696 string_ncopy(opt_search, search, strlen(search));
5697 else
5698 request = REQ_NONE;
5699 break;
5700 }
5701 case REQ_SCREEN_RESIZE:
5702 {
5703 int height, width;
5704
5705 getmaxyx(stdscr, height, width);
5706
5707 /* Resize the status view and let the view driver take
5708 * care of resizing the displayed views. */
5709 wresize(status_win, 1, width);
5710 mvwin(status_win, height - 1, 0);
5711 wrefresh(status_win);
5712 break;
5713 }
5714 default:
5715 break;
5716 }
5717 }
5718
5719 quit(0);
5720
5721 return 0;
5722 }
Full-screen terminal Git browser.