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