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