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