?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Remove unused OPEN_BACKGROUNDED flag
[tig.git] / tig.c
blob4990de5f5961311d3b20b4680c82ebadedf10b44
1 /* Copyright (c) 2006-2009 Jonas Fonseca <fonseca@diku.dk>
2 *
3 * This program is free software; you can redistribute it and/or
4 * modify it under the terms of the GNU General Public License as
5 * published by the Free Software Foundation; either version 2 of
6 * the License, or (at your option) any later version.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 */
13
14 #ifdef HAVE_CONFIG_H
15 #include "config.h"
16 #endif
17
18 #ifndef TIG_VERSION
19 #define TIG_VERSION "unknown-version"
20 #endif
21
22 #ifndef DEBUG
23 #define NDEBUG
24 #endif
25
26 #include <assert.h>
27 #include <errno.h>
28 #include <ctype.h>
29 #include <signal.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <stdlib.h>
33 #include <string.h>
34 #include <sys/types.h>
35 #include <sys/wait.h>
36 #include <sys/stat.h>
37 #include <sys/select.h>
38 #include <unistd.h>
39 #include <time.h>
40 #include <fcntl.h>
41
42 #include <regex.h>
43
44 #include <locale.h>
45 #include <langinfo.h>
46 #include <iconv.h>
47
48 /* ncurses(3): Must be defined to have extended wide-character functions. */
49 #define _XOPEN_SOURCE_EXTENDED
50
51 #ifdef HAVE_NCURSESW_NCURSES_H
52 #include <ncursesw/ncurses.h>
53 #else
54 #ifdef HAVE_NCURSES_NCURSES_H
55 #include <ncurses/ncurses.h>
56 #else
57 #include <ncurses.h>
58 #endif
59 #endif
60
61 #if __GNUC__ >= 3
62 #define __NORETURN __attribute__((__noreturn__))
63 #else
64 #define __NORETURN
65 #endif
66
67 static void __NORETURN die(const char *err, ...);
68 static void warn(const char *msg, ...);
69 static void report(const char *msg, ...);
70 static void set_nonblocking_input(bool loading);
71 static int load_refs(void);
72 static size_t utf8_length(const char **string, size_t col, int *width, size_t max_width, int *trimmed, bool reserve);
73
74 #define ABS(x) ((x) >= 0 ? (x) : -(x))
75 #define MIN(x, y) ((x) < (y) ? (x) : (y))
76
77 #define ARRAY_SIZE(x) (sizeof(x) / sizeof(x[0]))
78 #define STRING_SIZE(x) (sizeof(x) - 1)
79
80 #define SIZEOF_STR 1024 /* Default string size. */
81 #define SIZEOF_REF 256 /* Size of symbolic or SHA1 ID. */
82 #define SIZEOF_REV 41 /* Holds a SHA-1 and an ending NUL. */
83 #define SIZEOF_ARG 32 /* Default argument array size. */
84
85 /* Revision graph */
86
87 #define REVGRAPH_INIT 'I'
88 #define REVGRAPH_MERGE 'M'
89 #define REVGRAPH_BRANCH '+'
90 #define REVGRAPH_COMMIT '*'
91 #define REVGRAPH_BOUND '^'
92
93 #define SIZEOF_REVGRAPH 19 /* Size of revision ancestry graphics. */
94
95 /* This color name can be used to refer to the default term colors. */
96 #define COLOR_DEFAULT (-1)
97
98 #define ICONV_NONE ((iconv_t) -1)
99 #ifndef ICONV_CONST
100 #define ICONV_CONST /* nothing */
101 #endif
102
103 /* The format and size of the date column in the main view. */
104 #define DATE_FORMAT "%Y-%m-%d %H:%M"
105 #define DATE_COLS STRING_SIZE("2006-04-29 14:21 ")
106
107 #define AUTHOR_COLS 20
108 #define ID_COLS 8
109
110 /* The default interval between line numbers. */
111 #define NUMBER_INTERVAL 5
112 #define SCROLL_INTERVAL 1
113
114 #define TAB_SIZE 8
115
116 #define SCALE_SPLIT_VIEW(height) ((height) * 2 / 3)
117
118 #define NULL_ID "0000000000000000000000000000000000000000"
119
120 #ifndef GIT_CONFIG
121 #define GIT_CONFIG "config"
122 #endif
123
124 /* Some ASCII-shorthands fitted into the ncurses namespace. */
125 #define KEY_TAB '\t'
126 #define KEY_RETURN '\r'
127 #define KEY_ESC 27
128
129
130 struct ref {
131 char *name; /* Ref name; tag or head names are shortened. */
132 char id[SIZEOF_REV]; /* Commit SHA1 ID */
133 unsigned int head:1; /* Is it the current HEAD? */
134 unsigned int tag:1; /* Is it a tag? */
135 unsigned int ltag:1; /* If so, is the tag local? */
136 unsigned int remote:1; /* Is it a remote ref? */
137 unsigned int tracked:1; /* Is it the remote for the current HEAD? */
138 unsigned int next:1; /* For ref lists: are there more refs? */
139 };
140
141 static struct ref **get_refs(const char *id);
142
143 enum format_flags {
144 FORMAT_ALL, /* Perform replacement in all arguments. */
145 FORMAT_DASH, /* Perform replacement up until "--". */
146 FORMAT_NONE /* No replacement should be performed. */
147 };
148
149 static bool format_argv(const char *dst[], const char *src[], enum format_flags flags);
150
151 enum input_status {
152 INPUT_OK,
153 INPUT_SKIP,
154 INPUT_STOP,
155 INPUT_CANCEL
156 };
157
158 typedef enum input_status (*input_handler)(void *data, char *buf, int c);
159
160 static char *prompt_input(const char *prompt, input_handler handler, void *data);
161 static bool prompt_yesno(const char *prompt);
162
163 /*
164 * String helpers
165 */
166
167 static inline void
168 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
169 {
170 if (srclen > dstlen - 1)
171 srclen = dstlen - 1;
172
173 strncpy(dst, src, srclen);
174 dst[srclen] = 0;
175 }
176
177 /* Shorthands for safely copying into a fixed buffer. */
178
179 #define string_copy(dst, src) \
180 string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
181
182 #define string_ncopy(dst, src, srclen) \
183 string_ncopy_do(dst, sizeof(dst), src, srclen)
184
185 #define string_copy_rev(dst, src) \
186 string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
187
188 #define string_add(dst, from, src) \
189 string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
190
191 static size_t
192 string_expand_length(const char *line, int tabsize)
193 {
194 size_t size, pos;
195
196 for (pos = 0; line[pos]; pos++) {
197 if (line[pos] == '\t' && tabsize > 0)
198 size += tabsize - (size % tabsize);
199 else
200 size++;
201 }
202 return size;
203 }
204
205 static void
206 string_expand(char *dst, size_t dstlen, const char *src, int tabsize)
207 {
208 size_t size, pos;
209
210 for (size = pos = 0; size < dstlen - 1 && src[pos]; pos++) {
211 if (src[pos] == '\t') {
212 size_t expanded = tabsize - (size % tabsize);
213
214 if (expanded + size >= dstlen - 1)
215 expanded = dstlen - size - 1;
216 memcpy(dst + size, " ", expanded);
217 size += expanded;
218 } else {
219 dst[size++] = src[pos];
220 }
221 }
222
223 dst[size] = 0;
224 }
225
226 static char *
227 chomp_string(char *name)
228 {
229 int namelen;
230
231 while (isspace(*name))
232 name++;
233
234 namelen = strlen(name) - 1;
235 while (namelen > 0 && isspace(name[namelen]))
236 name[namelen--] = 0;
237
238 return name;
239 }
240
241 static bool
242 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
243 {
244 va_list args;
245 size_t pos = bufpos ? *bufpos : 0;
246
247 va_start(args, fmt);
248 pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
249 va_end(args);
250
251 if (bufpos)
252 *bufpos = pos;
253
254 return pos >= bufsize ? FALSE : TRUE;
255 }
256
257 #define string_format(buf, fmt, args...) \
258 string_nformat(buf, sizeof(buf), NULL, fmt, args)
259
260 #define string_format_from(buf, from, fmt, args...) \
261 string_nformat(buf, sizeof(buf), from, fmt, args)
262
263 static int
264 string_enum_compare(const char *str1, const char *str2, int len)
265 {
266 size_t i;
267
268 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
269
270 /* Diff-Header == DIFF_HEADER */
271 for (i = 0; i < len; i++) {
272 if (toupper(str1[i]) == toupper(str2[i]))
273 continue;
274
275 if (string_enum_sep(str1[i]) &&
276 string_enum_sep(str2[i]))
277 continue;
278
279 return str1[i] - str2[i];
280 }
281
282 return 0;
283 }
284
285 struct enum_map {
286 const char *name;
287 int namelen;
288 int value;
289 };
290
291 #define ENUM_MAP(name, value) { name, STRING_SIZE(name), value }
292
293 static bool
294 map_enum_do(struct enum_map *map, size_t map_size, int *value, const char *name)
295 {
296 size_t namelen = strlen(name);
297 int i;
298
299 for (i = 0; i < map_size; i++)
300 if (namelen == map[i].namelen &&
301 !string_enum_compare(name, map[i].name, namelen)) {
302 *value = map[i].value;
303 return TRUE;
304 }
305
306 return FALSE;
307 }
308
309 #define map_enum(attr, map, name) \
310 map_enum_do(map, ARRAY_SIZE(map), attr, name)
311
312 #define prefixcmp(str1, str2) \
313 strncmp(str1, str2, STRING_SIZE(str2))
314
315 static inline int
316 suffixcmp(const char *str, int slen, const char *suffix)
317 {
318 size_t len = slen >= 0 ? slen : strlen(str);
319 size_t suffixlen = strlen(suffix);
320
321 return suffixlen < len ? strcmp(str + len - suffixlen, suffix) : -1;
322 }
323
324
325 static bool
326 argv_from_string(const char *argv[SIZEOF_ARG], int *argc, char *cmd)
327 {
328 int valuelen;
329
330 while (*cmd && *argc < SIZEOF_ARG && (valuelen = strcspn(cmd, " \t"))) {
331 bool advance = cmd[valuelen] != 0;
332
333 cmd[valuelen] = 0;
334 argv[(*argc)++] = chomp_string(cmd);
335 cmd = chomp_string(cmd + valuelen + advance);
336 }
337
338 if (*argc < SIZEOF_ARG)
339 argv[*argc] = NULL;
340 return *argc < SIZEOF_ARG;
341 }
342
343 static void
344 argv_from_env(const char **argv, const char *name)
345 {
346 char *env = argv ? getenv(name) : NULL;
347 int argc = 0;
348
349 if (env && *env)
350 env = strdup(env);
351 if (env && !argv_from_string(argv, &argc, env))
352 die("Too many arguments in the `%s` environment variable", name);
353 }
354
355
356 /*
357 * Executing external commands.
358 */
359
360 enum io_type {
361 IO_FD, /* File descriptor based IO. */
362 IO_BG, /* Execute command in the background. */
363 IO_FG, /* Execute command with same std{in,out,err}. */
364 IO_RD, /* Read only fork+exec IO. */
365 IO_WR, /* Write only fork+exec IO. */
366 IO_AP, /* Append fork+exec output to file. */
367 };
368
369 struct io {
370 enum io_type type; /* The requested type of pipe. */
371 const char *dir; /* Directory from which to execute. */
372 pid_t pid; /* Pipe for reading or writing. */
373 int pipe; /* Pipe end for reading or writing. */
374 int error; /* Error status. */
375 const char *argv[SIZEOF_ARG]; /* Shell command arguments. */
376 char *buf; /* Read buffer. */
377 size_t bufalloc; /* Allocated buffer size. */
378 size_t bufsize; /* Buffer content size. */
379 char *bufpos; /* Current buffer position. */
380 unsigned int eof:1; /* Has end of file been reached. */
381 };
382
383 static void
384 reset_io(struct io *io)
385 {
386 io->pipe = -1;
387 io->pid = 0;
388 io->buf = io->bufpos = NULL;
389 io->bufalloc = io->bufsize = 0;
390 io->error = 0;
391 io->eof = 0;
392 }
393
394 static void
395 init_io(struct io *io, const char *dir, enum io_type type)
396 {
397 reset_io(io);
398 io->type = type;
399 io->dir = dir;
400 }
401
402 static bool
403 init_io_rd(struct io *io, const char *argv[], const char *dir,
404 enum format_flags flags)
405 {
406 init_io(io, dir, IO_RD);
407 return format_argv(io->argv, argv, flags);
408 }
409
410 static bool
411 io_open(struct io *io, const char *name)
412 {
413 init_io(io, NULL, IO_FD);
414 io->pipe = *name ? open(name, O_RDONLY) : STDIN_FILENO;
415 return io->pipe != -1;
416 }
417
418 static bool
419 kill_io(struct io *io)
420 {
421 return io->pid == 0 || kill(io->pid, SIGKILL) != -1;
422 }
423
424 static bool
425 done_io(struct io *io)
426 {
427 pid_t pid = io->pid;
428
429 if (io->pipe != -1)
430 close(io->pipe);
431 free(io->buf);
432 reset_io(io);
433
434 while (pid > 0) {
435 int status;
436 pid_t waiting = waitpid(pid, &status, 0);
437
438 if (waiting < 0) {
439 if (errno == EINTR)
440 continue;
441 report("waitpid failed (%s)", strerror(errno));
442 return FALSE;
443 }
444
445 return waiting == pid &&
446 !WIFSIGNALED(status) &&
447 WIFEXITED(status) &&
448 !WEXITSTATUS(status);
449 }
450
451 return TRUE;
452 }
453
454 static bool
455 start_io(struct io *io)
456 {
457 int pipefds[2] = { -1, -1 };
458
459 if (io->type == IO_FD)
460 return TRUE;
461
462 if ((io->type == IO_RD || io->type == IO_WR) &&
463 pipe(pipefds) < 0)
464 return FALSE;
465 else if (io->type == IO_AP)
466 pipefds[1] = io->pipe;
467
468 if ((io->pid = fork())) {
469 if (pipefds[!(io->type == IO_WR)] != -1)
470 close(pipefds[!(io->type == IO_WR)]);
471 if (io->pid != -1) {
472 io->pipe = pipefds[!!(io->type == IO_WR)];
473 return TRUE;
474 }
475
476 } else {
477 if (io->type != IO_FG) {
478 int devnull = open("/dev/null", O_RDWR);
479 int readfd = io->type == IO_WR ? pipefds[0] : devnull;
480 int writefd = (io->type == IO_RD || io->type == IO_AP)
481 ? pipefds[1] : devnull;
482
483 dup2(readfd, STDIN_FILENO);
484 dup2(writefd, STDOUT_FILENO);
485 dup2(devnull, STDERR_FILENO);
486
487 close(devnull);
488 if (pipefds[0] != -1)
489 close(pipefds[0]);
490 if (pipefds[1] != -1)
491 close(pipefds[1]);
492 }
493
494 if (io->dir && *io->dir && chdir(io->dir) == -1)
495 die("Failed to change directory: %s", strerror(errno));
496
497 execvp(io->argv[0], (char *const*) io->argv);
498 die("Failed to execute program: %s", strerror(errno));
499 }
500
501 if (pipefds[!!(io->type == IO_WR)] != -1)
502 close(pipefds[!!(io->type == IO_WR)]);
503 return FALSE;
504 }
505
506 static bool
507 run_io(struct io *io, const char **argv, const char *dir, enum io_type type)
508 {
509 init_io(io, dir, type);
510 if (!format_argv(io->argv, argv, FORMAT_NONE))
511 return FALSE;
512 return start_io(io);
513 }
514
515 static int
516 run_io_do(struct io *io)
517 {
518 return start_io(io) && done_io(io);
519 }
520
521 static int
522 run_io_bg(const char **argv)
523 {
524 struct io io = {};
525
526 init_io(&io, NULL, IO_BG);
527 if (!format_argv(io.argv, argv, FORMAT_NONE))
528 return FALSE;
529 return run_io_do(&io);
530 }
531
532 static bool
533 run_io_fg(const char **argv, const char *dir)
534 {
535 struct io io = {};
536
537 init_io(&io, dir, IO_FG);
538 if (!format_argv(io.argv, argv, FORMAT_NONE))
539 return FALSE;
540 return run_io_do(&io);
541 }
542
543 static bool
544 run_io_append(const char **argv, enum format_flags flags, int fd)
545 {
546 struct io io = {};
547
548 init_io(&io, NULL, IO_AP);
549 io.pipe = fd;
550 if (format_argv(io.argv, argv, flags))
551 return run_io_do(&io);
552 close(fd);
553 return FALSE;
554 }
555
556 static bool
557 run_io_rd(struct io *io, const char **argv, enum format_flags flags)
558 {
559 return init_io_rd(io, argv, NULL, flags) && start_io(io);
560 }
561
562 static bool
563 io_eof(struct io *io)
564 {
565 return io->eof;
566 }
567
568 static int
569 io_error(struct io *io)
570 {
571 return io->error;
572 }
573
574 static bool
575 io_strerror(struct io *io)
576 {
577 return strerror(io->error);
578 }
579
580 static bool
581 io_can_read(struct io *io)
582 {
583 struct timeval tv = { 0, 500 };
584 fd_set fds;
585
586 FD_ZERO(&fds);
587 FD_SET(io->pipe, &fds);
588
589 return select(io->pipe + 1, &fds, NULL, NULL, &tv) > 0;
590 }
591
592 static ssize_t
593 io_read(struct io *io, void *buf, size_t bufsize)
594 {
595 do {
596 ssize_t readsize = read(io->pipe, buf, bufsize);
597
598 if (readsize < 0 && (errno == EAGAIN || errno == EINTR))
599 continue;
600 else if (readsize == -1)
601 io->error = errno;
602 else if (readsize == 0)
603 io->eof = 1;
604 return readsize;
605 } while (1);
606 }
607
608 static char *
609 io_get(struct io *io, int c, bool can_read)
610 {
611 char *eol;
612 ssize_t readsize;
613
614 if (!io->buf) {
615 io->buf = io->bufpos = malloc(BUFSIZ);
616 if (!io->buf)
617 return NULL;
618 io->bufalloc = BUFSIZ;
619 io->bufsize = 0;
620 }
621
622 while (TRUE) {
623 if (io->bufsize > 0) {
624 eol = memchr(io->bufpos, c, io->bufsize);
625 if (eol) {
626 char *line = io->bufpos;
627
628 *eol = 0;
629 io->bufpos = eol + 1;
630 io->bufsize -= io->bufpos - line;
631 return line;
632 }
633 }
634
635 if (io_eof(io)) {
636 if (io->bufsize) {
637 io->bufpos[io->bufsize] = 0;
638 io->bufsize = 0;
639 return io->bufpos;
640 }
641 return NULL;
642 }
643
644 if (!can_read)
645 return NULL;
646
647 if (io->bufsize > 0 && io->bufpos > io->buf)
648 memmove(io->buf, io->bufpos, io->bufsize);
649
650 io->bufpos = io->buf;
651 readsize = io_read(io, io->buf + io->bufsize, io->bufalloc - io->bufsize);
652 if (io_error(io))
653 return NULL;
654 io->bufsize += readsize;
655 }
656 }
657
658 static bool
659 io_write(struct io *io, const void *buf, size_t bufsize)
660 {
661 size_t written = 0;
662
663 while (!io_error(io) && written < bufsize) {
664 ssize_t size;
665
666 size = write(io->pipe, buf + written, bufsize - written);
667 if (size < 0 && (errno == EAGAIN || errno == EINTR))
668 continue;
669 else if (size == -1)
670 io->error = errno;
671 else
672 written += size;
673 }
674
675 return written == bufsize;
676 }
677
678 static bool
679 run_io_buf(const char **argv, char buf[], size_t bufsize)
680 {
681 struct io io = {};
682 bool error;
683
684 if (!run_io_rd(&io, argv, FORMAT_NONE))
685 return FALSE;
686
687 io.buf = io.bufpos = buf;
688 io.bufalloc = bufsize;
689 error = !io_get(&io, '\n', TRUE) && io_error(&io);
690 io.buf = NULL;
691
692 return done_io(&io) || error;
693 }
694
695 static int
696 io_load(struct io *io, const char *separators,
697 int (*read_property)(char *, size_t, char *, size_t))
698 {
699 char *name;
700 int state = OK;
701
702 if (!start_io(io))
703 return ERR;
704
705 while (state == OK && (name = io_get(io, '\n', TRUE))) {
706 char *value;
707 size_t namelen;
708 size_t valuelen;
709
710 name = chomp_string(name);
711 namelen = strcspn(name, separators);
712
713 if (name[namelen]) {
714 name[namelen] = 0;
715 value = chomp_string(name + namelen + 1);
716 valuelen = strlen(value);
717
718 } else {
719 value = "";
720 valuelen = 0;
721 }
722
723 state = read_property(name, namelen, value, valuelen);
724 }
725
726 if (state != ERR && io_error(io))
727 state = ERR;
728 done_io(io);
729
730 return state;
731 }
732
733 static int
734 run_io_load(const char **argv, const char *separators,
735 int (*read_property)(char *, size_t, char *, size_t))
736 {
737 struct io io = {};
738
739 return init_io_rd(&io, argv, NULL, FORMAT_NONE)
740 ? io_load(&io, separators, read_property) : ERR;
741 }
742
743
744 /*
745 * User requests
746 */
747
748 #define REQ_INFO \
749 /* XXX: Keep the view request first and in sync with views[]. */ \
750 REQ_GROUP("View switching") \
751 REQ_(VIEW_MAIN, "Show main view"), \
752 REQ_(VIEW_DIFF, "Show diff view"), \
753 REQ_(VIEW_LOG, "Show log view"), \
754 REQ_(VIEW_TREE, "Show tree view"), \
755 REQ_(VIEW_BLOB, "Show blob view"), \
756 REQ_(VIEW_BLAME, "Show blame view"), \
757 REQ_(VIEW_HELP, "Show help page"), \
758 REQ_(VIEW_PAGER, "Show pager view"), \
759 REQ_(VIEW_STATUS, "Show status view"), \
760 REQ_(VIEW_STAGE, "Show stage view"), \
761 \
762 REQ_GROUP("View manipulation") \
763 REQ_(ENTER, "Enter current line and scroll"), \
764 REQ_(NEXT, "Move to next"), \
765 REQ_(PREVIOUS, "Move to previous"), \
766 REQ_(PARENT, "Move to parent"), \
767 REQ_(VIEW_NEXT, "Move focus to next view"), \
768 REQ_(REFRESH, "Reload and refresh"), \
769 REQ_(MAXIMIZE, "Maximize the current view"), \
770 REQ_(VIEW_CLOSE, "Close the current view"), \
771 REQ_(QUIT, "Close all views and quit"), \
772 \
773 REQ_GROUP("View specific requests") \
774 REQ_(STATUS_UPDATE, "Update file status"), \
775 REQ_(STATUS_REVERT, "Revert file changes"), \
776 REQ_(STATUS_MERGE, "Merge file using external tool"), \
777 REQ_(STAGE_NEXT, "Find next chunk to stage"), \
778 \
779 REQ_GROUP("Cursor navigation") \
780 REQ_(MOVE_UP, "Move cursor one line up"), \
781 REQ_(MOVE_DOWN, "Move cursor one line down"), \
782 REQ_(MOVE_PAGE_DOWN, "Move cursor one page down"), \
783 REQ_(MOVE_PAGE_UP, "Move cursor one page up"), \
784 REQ_(MOVE_FIRST_LINE, "Move cursor to first line"), \
785 REQ_(MOVE_LAST_LINE, "Move cursor to last line"), \
786 \
787 REQ_GROUP("Scrolling") \
788 REQ_(SCROLL_LEFT, "Scroll two columns left"), \
789 REQ_(SCROLL_RIGHT, "Scroll two columns right"), \
790 REQ_(SCROLL_LINE_UP, "Scroll one line up"), \
791 REQ_(SCROLL_LINE_DOWN, "Scroll one line down"), \
792 REQ_(SCROLL_PAGE_UP, "Scroll one page up"), \
793 REQ_(SCROLL_PAGE_DOWN, "Scroll one page down"), \
794 \
795 REQ_GROUP("Searching") \
796 REQ_(SEARCH, "Search the view"), \
797 REQ_(SEARCH_BACK, "Search backwards in the view"), \
798 REQ_(FIND_NEXT, "Find next search match"), \
799 REQ_(FIND_PREV, "Find previous search match"), \
800 \
801 REQ_GROUP("Option manipulation") \
802 REQ_(TOGGLE_LINENO, "Toggle line numbers"), \
803 REQ_(TOGGLE_DATE, "Toggle date display"), \
804 REQ_(TOGGLE_AUTHOR, "Toggle author display"), \
805 REQ_(TOGGLE_REV_GRAPH, "Toggle revision graph visualization"), \
806 REQ_(TOGGLE_REFS, "Toggle reference display (tags/branches)"), \
807 \
808 REQ_GROUP("Misc") \
809 REQ_(PROMPT, "Bring up the prompt"), \
810 REQ_(SCREEN_REDRAW, "Redraw the screen"), \
811 REQ_(SHOW_VERSION, "Show version information"), \
812 REQ_(STOP_LOADING, "Stop all loading views"), \
813 REQ_(EDIT, "Open in editor"), \
814 REQ_(NONE, "Do nothing")
815
816
817 /* User action requests. */
818 enum request {
819 #define REQ_GROUP(help)
820 #define REQ_(req, help) REQ_##req
821
822 /* Offset all requests to avoid conflicts with ncurses getch values. */
823 REQ_OFFSET = KEY_MAX + 1,
824 REQ_INFO
825
826 #undef REQ_GROUP
827 #undef REQ_
828 };
829
830 struct request_info {
831 enum request request;
832 const char *name;
833 int namelen;
834 const char *help;
835 };
836
837 static struct request_info req_info[] = {
838 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
839 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
840 REQ_INFO
841 #undef REQ_GROUP
842 #undef REQ_
843 };
844
845 static enum request
846 get_request(const char *name)
847 {
848 int namelen = strlen(name);
849 int i;
850
851 for (i = 0; i < ARRAY_SIZE(req_info); i++)
852 if (req_info[i].namelen == namelen &&
853 !string_enum_compare(req_info[i].name, name, namelen))
854 return req_info[i].request;
855
856 return REQ_NONE;
857 }
858
859
860 /*
861 * Options
862 */
863
864 /* Option and state variables. */
865 static bool opt_date = TRUE;
866 static bool opt_author = TRUE;
867 static bool opt_line_number = FALSE;
868 static bool opt_line_graphics = TRUE;
869 static bool opt_rev_graph = FALSE;
870 static bool opt_show_refs = TRUE;
871 static int opt_num_interval = NUMBER_INTERVAL;
872 static int opt_tab_size = TAB_SIZE;
873 static int opt_author_cols = AUTHOR_COLS-1;
874 static char opt_path[SIZEOF_STR] = "";
875 static char opt_file[SIZEOF_STR] = "";
876 static char opt_ref[SIZEOF_REF] = "";
877 static char opt_head[SIZEOF_REF] = "";
878 static char opt_head_rev[SIZEOF_REV] = "";
879 static char opt_remote[SIZEOF_REF] = "";
880 static char opt_encoding[20] = "UTF-8";
881 static bool opt_utf8 = TRUE;
882 static char opt_codeset[20] = "UTF-8";
883 static iconv_t opt_iconv = ICONV_NONE;
884 static char opt_search[SIZEOF_STR] = "";
885 static char opt_cdup[SIZEOF_STR] = "";
886 static char opt_prefix[SIZEOF_STR] = "";
887 static char opt_git_dir[SIZEOF_STR] = "";
888 static signed char opt_is_inside_work_tree = -1; /* set to TRUE or FALSE */
889 static char opt_editor[SIZEOF_STR] = "";
890 static FILE *opt_tty = NULL;
891
892 #define is_initial_commit() (!*opt_head_rev)
893 #define is_head_commit(rev) (!strcmp((rev), "HEAD") || !strcmp(opt_head_rev, (rev)))
894
895
896 /*
897 * Line-oriented content detection.
898 */
899
900 #define LINE_INFO \
901 LINE(DIFF_HEADER, "diff --git ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
902 LINE(DIFF_CHUNK, "@@", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
903 LINE(DIFF_ADD, "+", COLOR_GREEN, COLOR_DEFAULT, 0), \
904 LINE(DIFF_DEL, "-", COLOR_RED, COLOR_DEFAULT, 0), \
905 LINE(DIFF_INDEX, "index ", COLOR_BLUE, COLOR_DEFAULT, 0), \
906 LINE(DIFF_OLDMODE, "old file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
907 LINE(DIFF_NEWMODE, "new file mode ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
908 LINE(DIFF_COPY_FROM, "copy from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
909 LINE(DIFF_COPY_TO, "copy to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
910 LINE(DIFF_RENAME_FROM, "rename from", COLOR_YELLOW, COLOR_DEFAULT, 0), \
911 LINE(DIFF_RENAME_TO, "rename to", COLOR_YELLOW, COLOR_DEFAULT, 0), \
912 LINE(DIFF_SIMILARITY, "similarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
913 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
914 LINE(DIFF_TREE, "diff-tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
915 LINE(PP_AUTHOR, "Author: ", COLOR_CYAN, COLOR_DEFAULT, 0), \
916 LINE(PP_COMMIT, "Commit: ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
917 LINE(PP_MERGE, "Merge: ", COLOR_BLUE, COLOR_DEFAULT, 0), \
918 LINE(PP_DATE, "Date: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
919 LINE(PP_ADATE, "AuthorDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
920 LINE(PP_CDATE, "CommitDate: ", COLOR_YELLOW, COLOR_DEFAULT, 0), \
921 LINE(PP_REFS, "Refs: ", COLOR_RED, COLOR_DEFAULT, 0), \
922 LINE(COMMIT, "commit ", COLOR_GREEN, COLOR_DEFAULT, 0), \
923 LINE(PARENT, "parent ", COLOR_BLUE, COLOR_DEFAULT, 0), \
924 LINE(TREE, "tree ", COLOR_BLUE, COLOR_DEFAULT, 0), \
925 LINE(AUTHOR, "author ", COLOR_GREEN, COLOR_DEFAULT, 0), \
926 LINE(COMMITTER, "committer ", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
927 LINE(SIGNOFF, " Signed-off-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
928 LINE(ACKED, " Acked-by", COLOR_YELLOW, COLOR_DEFAULT, 0), \
929 LINE(DEFAULT, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
930 LINE(CURSOR, "", COLOR_WHITE, COLOR_GREEN, A_BOLD), \
931 LINE(STATUS, "", COLOR_GREEN, COLOR_DEFAULT, 0), \
932 LINE(DELIMITER, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
933 LINE(DATE, "", COLOR_BLUE, COLOR_DEFAULT, 0), \
934 LINE(MODE, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
935 LINE(LINE_NUMBER, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
936 LINE(TITLE_BLUR, "", COLOR_WHITE, COLOR_BLUE, 0), \
937 LINE(TITLE_FOCUS, "", COLOR_WHITE, COLOR_BLUE, A_BOLD), \
938 LINE(MAIN_COMMIT, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
939 LINE(MAIN_TAG, "", COLOR_MAGENTA, COLOR_DEFAULT, A_BOLD), \
940 LINE(MAIN_LOCAL_TAG,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
941 LINE(MAIN_REMOTE, "", COLOR_YELLOW, COLOR_DEFAULT, 0), \
942 LINE(MAIN_TRACKED, "", COLOR_YELLOW, COLOR_DEFAULT, A_BOLD), \
943 LINE(MAIN_REF, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
944 LINE(MAIN_HEAD, "", COLOR_CYAN, COLOR_DEFAULT, A_BOLD), \
945 LINE(MAIN_REVGRAPH,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
946 LINE(TREE_HEAD, "", COLOR_DEFAULT, COLOR_DEFAULT, A_BOLD), \
947 LINE(TREE_DIR, "", COLOR_YELLOW, COLOR_DEFAULT, A_NORMAL), \
948 LINE(TREE_FILE, "", COLOR_DEFAULT, COLOR_DEFAULT, A_NORMAL), \
949 LINE(STAT_HEAD, "", COLOR_YELLOW, COLOR_DEFAULT, 0), \
950 LINE(STAT_SECTION, "", COLOR_CYAN, COLOR_DEFAULT, 0), \
951 LINE(STAT_NONE, "", COLOR_DEFAULT, COLOR_DEFAULT, 0), \
952 LINE(STAT_STAGED, "", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
953 LINE(STAT_UNSTAGED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
954 LINE(STAT_UNTRACKED,"", COLOR_MAGENTA, COLOR_DEFAULT, 0), \
955 LINE(BLAME_ID, "", COLOR_MAGENTA, COLOR_DEFAULT, 0)
956
957 enum line_type {
958 #define LINE(type, line, fg, bg, attr) \
959 LINE_##type
960 LINE_INFO,
961 LINE_NONE
962 #undef LINE
963 };
964
965 struct line_info {
966 const char *name; /* Option name. */
967 int namelen; /* Size of option name. */
968 const char *line; /* The start of line to match. */
969 int linelen; /* Size of string to match. */
970 int fg, bg, attr; /* Color and text attributes for the lines. */
971 };
972
973 static struct line_info line_info[] = {
974 #define LINE(type, line, fg, bg, attr) \
975 { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
976 LINE_INFO
977 #undef LINE
978 };
979
980 static enum line_type
981 get_line_type(const char *line)
982 {
983 int linelen = strlen(line);
984 enum line_type type;
985
986 for (type = 0; type < ARRAY_SIZE(line_info); type++)
987 /* Case insensitive search matches Signed-off-by lines better. */
988 if (linelen >= line_info[type].linelen &&
989 !strncasecmp(line_info[type].line, line, line_info[type].linelen))
990 return type;
991
992 return LINE_DEFAULT;
993 }
994
995 static inline int
996 get_line_attr(enum line_type type)
997 {
998 assert(type < ARRAY_SIZE(line_info));
999 return COLOR_PAIR(type) | line_info[type].attr;
1000 }
1001
1002 static struct line_info *
1003 get_line_info(const char *name)
1004 {
1005 size_t namelen = strlen(name);
1006 enum line_type type;
1007
1008 for (type = 0; type < ARRAY_SIZE(line_info); type++)
1009 if (namelen == line_info[type].namelen &&
1010 !string_enum_compare(line_info[type].name, name, namelen))
1011 return &line_info[type];
1012
1013 return NULL;
1014 }
1015
1016 static void
1017 init_colors(void)
1018 {
1019 int default_bg = line_info[LINE_DEFAULT].bg;
1020 int default_fg = line_info[LINE_DEFAULT].fg;
1021 enum line_type type;
1022
1023 start_color();
1024
1025 if (assume_default_colors(default_fg, default_bg) == ERR) {
1026 default_bg = COLOR_BLACK;
1027 default_fg = COLOR_WHITE;
1028 }
1029
1030 for (type = 0; type < ARRAY_SIZE(line_info); type++) {
1031 struct line_info *info = &line_info[type];
1032 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
1033 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
1034
1035 init_pair(type, fg, bg);
1036 }
1037 }
1038
1039 struct line {
1040 enum line_type type;
1041
1042 /* State flags */
1043 unsigned int selected:1;
1044 unsigned int dirty:1;
1045 unsigned int cleareol:1;
1046
1047 void *data; /* User data */
1048 };
1049
1050
1051 /*
1052 * Keys
1053 */
1054
1055 struct keybinding {
1056 int alias;
1057 enum request request;
1058 };
1059
1060 static struct keybinding default_keybindings[] = {
1061 /* View switching */
1062 { 'm', REQ_VIEW_MAIN },
1063 { 'd', REQ_VIEW_DIFF },
1064 { 'l', REQ_VIEW_LOG },
1065 { 't', REQ_VIEW_TREE },
1066 { 'f', REQ_VIEW_BLOB },
1067 { 'B', REQ_VIEW_BLAME },
1068 { 'p', REQ_VIEW_PAGER },
1069 { 'h', REQ_VIEW_HELP },
1070 { 'S', REQ_VIEW_STATUS },
1071 { 'c', REQ_VIEW_STAGE },
1072
1073 /* View manipulation */
1074 { 'q', REQ_VIEW_CLOSE },
1075 { KEY_TAB, REQ_VIEW_NEXT },
1076 { KEY_RETURN, REQ_ENTER },
1077 { KEY_UP, REQ_PREVIOUS },
1078 { KEY_DOWN, REQ_NEXT },
1079 { 'R', REQ_REFRESH },
1080 { KEY_F(5), REQ_REFRESH },
1081 { 'O', REQ_MAXIMIZE },
1082
1083 /* Cursor navigation */
1084 { 'k', REQ_MOVE_UP },
1085 { 'j', REQ_MOVE_DOWN },
1086 { KEY_HOME, REQ_MOVE_FIRST_LINE },
1087 { KEY_END, REQ_MOVE_LAST_LINE },
1088 { KEY_NPAGE, REQ_MOVE_PAGE_DOWN },
1089 { ' ', REQ_MOVE_PAGE_DOWN },
1090 { KEY_PPAGE, REQ_MOVE_PAGE_UP },
1091 { 'b', REQ_MOVE_PAGE_UP },
1092 { '-', REQ_MOVE_PAGE_UP },
1093
1094 /* Scrolling */
1095 { KEY_LEFT, REQ_SCROLL_LEFT },
1096 { KEY_RIGHT, REQ_SCROLL_RIGHT },
1097 { KEY_IC, REQ_SCROLL_LINE_UP },
1098 { KEY_DC, REQ_SCROLL_LINE_DOWN },
1099 { 'w', REQ_SCROLL_PAGE_UP },
1100 { 's', REQ_SCROLL_PAGE_DOWN },
1101
1102 /* Searching */
1103 { '/', REQ_SEARCH },
1104 { '?', REQ_SEARCH_BACK },
1105 { 'n', REQ_FIND_NEXT },
1106 { 'N', REQ_FIND_PREV },
1107
1108 /* Misc */
1109 { 'Q', REQ_QUIT },
1110 { 'z', REQ_STOP_LOADING },
1111 { 'v', REQ_SHOW_VERSION },
1112 { 'r', REQ_SCREEN_REDRAW },
1113 { '.', REQ_TOGGLE_LINENO },
1114 { 'D', REQ_TOGGLE_DATE },
1115 { 'A', REQ_TOGGLE_AUTHOR },
1116 { 'g', REQ_TOGGLE_REV_GRAPH },
1117 { 'F', REQ_TOGGLE_REFS },
1118 { ':', REQ_PROMPT },
1119 { 'u', REQ_STATUS_UPDATE },
1120 { '!', REQ_STATUS_REVERT },
1121 { 'M', REQ_STATUS_MERGE },
1122 { '@', REQ_STAGE_NEXT },
1123 { ',', REQ_PARENT },
1124 { 'e', REQ_EDIT },
1125 };
1126
1127 #define KEYMAP_INFO \
1128 KEYMAP_(GENERIC), \
1129 KEYMAP_(MAIN), \
1130 KEYMAP_(DIFF), \
1131 KEYMAP_(LOG), \
1132 KEYMAP_(TREE), \
1133 KEYMAP_(BLOB), \
1134 KEYMAP_(BLAME), \
1135 KEYMAP_(PAGER), \
1136 KEYMAP_(HELP), \
1137 KEYMAP_(STATUS), \
1138 KEYMAP_(STAGE)
1139
1140 enum keymap {
1141 #define KEYMAP_(name) KEYMAP_##name
1142 KEYMAP_INFO
1143 #undef KEYMAP_
1144 };
1145
1146 static struct enum_map keymap_table[] = {
1147 #define KEYMAP_(name) ENUM_MAP(#name, KEYMAP_##name)
1148 KEYMAP_INFO
1149 #undef KEYMAP_
1150 };
1151
1152 #define set_keymap(map, name) map_enum(map, keymap_table, name)
1153
1154 struct keybinding_table {
1155 struct keybinding *data;
1156 size_t size;
1157 };
1158
1159 static struct keybinding_table keybindings[ARRAY_SIZE(keymap_table)];
1160
1161 static void
1162 add_keybinding(enum keymap keymap, enum request request, int key)
1163 {
1164 struct keybinding_table *table = &keybindings[keymap];
1165
1166 table->data = realloc(table->data, (table->size + 1) * sizeof(*table->data));
1167 if (!table->data)
1168 die("Failed to allocate keybinding");
1169 table->data[table->size].alias = key;
1170 table->data[table->size++].request = request;
1171 }
1172
1173 /* Looks for a key binding first in the given map, then in the generic map, and
1174 * lastly in the default keybindings. */
1175 static enum request
1176 get_keybinding(enum keymap keymap, int key)
1177 {
1178 size_t i;
1179
1180 for (i = 0; i < keybindings[keymap].size; i++)
1181 if (keybindings[keymap].data[i].alias == key)
1182 return keybindings[keymap].data[i].request;
1183
1184 for (i = 0; i < keybindings[KEYMAP_GENERIC].size; i++)
1185 if (keybindings[KEYMAP_GENERIC].data[i].alias == key)
1186 return keybindings[KEYMAP_GENERIC].data[i].request;
1187
1188 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
1189 if (default_keybindings[i].alias == key)
1190 return default_keybindings[i].request;
1191
1192 return (enum request) key;
1193 }
1194
1195
1196 struct key {
1197 const char *name;
1198 int value;
1199 };
1200
1201 static struct key key_table[] = {
1202 { "Enter", KEY_RETURN },
1203 { "Space", ' ' },
1204 { "Backspace", KEY_BACKSPACE },
1205 { "Tab", KEY_TAB },
1206 { "Escape", KEY_ESC },
1207 { "Left", KEY_LEFT },
1208 { "Right", KEY_RIGHT },
1209 { "Up", KEY_UP },
1210 { "Down", KEY_DOWN },
1211 { "Insert", KEY_IC },
1212 { "Delete", KEY_DC },
1213 { "Hash", '#' },
1214 { "Home", KEY_HOME },
1215 { "End", KEY_END },
1216 { "PageUp", KEY_PPAGE },
1217 { "PageDown", KEY_NPAGE },
1218 { "F1", KEY_F(1) },
1219 { "F2", KEY_F(2) },
1220 { "F3", KEY_F(3) },
1221 { "F4", KEY_F(4) },
1222 { "F5", KEY_F(5) },
1223 { "F6", KEY_F(6) },
1224 { "F7", KEY_F(7) },
1225 { "F8", KEY_F(8) },
1226 { "F9", KEY_F(9) },
1227 { "F10", KEY_F(10) },
1228 { "F11", KEY_F(11) },
1229 { "F12", KEY_F(12) },
1230 };
1231
1232 static int
1233 get_key_value(const char *name)
1234 {
1235 int i;
1236
1237 for (i = 0; i < ARRAY_SIZE(key_table); i++)
1238 if (!strcasecmp(key_table[i].name, name))
1239 return key_table[i].value;
1240
1241 if (strlen(name) == 1 && isprint(*name))
1242 return (int) *name;
1243
1244 return ERR;
1245 }
1246
1247 static const char *
1248 get_key_name(int key_value)
1249 {
1250 static char key_char[] = "'X'";
1251 const char *seq = NULL;
1252 int key;
1253
1254 for (key = 0; key < ARRAY_SIZE(key_table); key++)
1255 if (key_table[key].value == key_value)
1256 seq = key_table[key].name;
1257
1258 if (seq == NULL &&
1259 key_value < 127 &&
1260 isprint(key_value)) {
1261 key_char[1] = (char) key_value;
1262 seq = key_char;
1263 }
1264
1265 return seq ? seq : "(no key)";
1266 }
1267
1268 static const char *
1269 get_key(enum request request)
1270 {
1271 static char buf[BUFSIZ];
1272 size_t pos = 0;
1273 char *sep = "";
1274 int i;
1275
1276 buf[pos] = 0;
1277
1278 for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
1279 struct keybinding *keybinding = &default_keybindings[i];
1280
1281 if (keybinding->request != request)
1282 continue;
1283
1284 if (!string_format_from(buf, &pos, "%s%s", sep,
1285 get_key_name(keybinding->alias)))
1286 return "Too many keybindings!";
1287 sep = ", ";
1288 }
1289
1290 return buf;
1291 }
1292
1293 struct run_request {
1294 enum keymap keymap;
1295 int key;
1296 const char *argv[SIZEOF_ARG];
1297 };
1298
1299 static struct run_request *run_request;
1300 static size_t run_requests;
1301
1302 static enum request
1303 add_run_request(enum keymap keymap, int key, int argc, const char **argv)
1304 {
1305 struct run_request *req;
1306
1307 if (argc >= ARRAY_SIZE(req->argv) - 1)
1308 return REQ_NONE;
1309
1310 req = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
1311 if (!req)
1312 return REQ_NONE;
1313
1314 run_request = req;
1315 req = &run_request[run_requests];
1316 req->keymap = keymap;
1317 req->key = key;
1318 req->argv[0] = NULL;
1319
1320 if (!format_argv(req->argv, argv, FORMAT_NONE))
1321 return REQ_NONE;
1322
1323 return REQ_NONE + ++run_requests;
1324 }
1325
1326 static struct run_request *
1327 get_run_request(enum request request)
1328 {
1329 if (request <= REQ_NONE)
1330 return NULL;
1331 return &run_request[request - REQ_NONE - 1];
1332 }
1333
1334 static void
1335 add_builtin_run_requests(void)
1336 {
1337 const char *cherry_pick[] = { "git", "cherry-pick", "%(commit)", NULL };
1338 const char *gc[] = { "git", "gc", NULL };
1339 struct {
1340 enum keymap keymap;
1341 int key;
1342 int argc;
1343 const char **argv;
1344 } reqs[] = {
1345 { KEYMAP_MAIN, 'C', ARRAY_SIZE(cherry_pick) - 1, cherry_pick },
1346 { KEYMAP_GENERIC, 'G', ARRAY_SIZE(gc) - 1, gc },
1347 };
1348 int i;
1349
1350 for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1351 enum request req;
1352
1353 req = add_run_request(reqs[i].keymap, reqs[i].key, reqs[i].argc, reqs[i].argv);
1354 if (req != REQ_NONE)
1355 add_keybinding(reqs[i].keymap, req, reqs[i].key);
1356 }
1357 }
1358
1359 /*
1360 * User config file handling.
1361 */
1362
1363 static struct enum_map color_map[] = {
1364 #define COLOR_MAP(name) ENUM_MAP(#name, COLOR_##name)
1365 COLOR_MAP(DEFAULT),
1366 COLOR_MAP(BLACK),
1367 COLOR_MAP(BLUE),
1368 COLOR_MAP(CYAN),
1369 COLOR_MAP(GREEN),
1370 COLOR_MAP(MAGENTA),
1371 COLOR_MAP(RED),
1372 COLOR_MAP(WHITE),
1373 COLOR_MAP(YELLOW),
1374 };
1375
1376 static struct enum_map attr_map[] = {
1377 #define ATTR_MAP(name) ENUM_MAP(#name, A_##name)
1378 ATTR_MAP(NORMAL),
1379 ATTR_MAP(BLINK),
1380 ATTR_MAP(BOLD),
1381 ATTR_MAP(DIM),
1382 ATTR_MAP(REVERSE),
1383 ATTR_MAP(STANDOUT),
1384 ATTR_MAP(UNDERLINE),
1385 };
1386
1387 #define set_color(color, name) map_enum(color, color_map, name)
1388 #define set_attribute(attr, name) map_enum(attr, attr_map, name)
1389
1390 static int config_lineno;
1391 static bool config_errors;
1392 static const char *config_msg;
1393
1394 /* Wants: object fgcolor bgcolor [attribute] */
1395 static int
1396 option_color_command(int argc, const char *argv[])
1397 {
1398 struct line_info *info;
1399
1400 if (argc != 3 && argc != 4) {
1401 config_msg = "Wrong number of arguments given to color command";
1402 return ERR;
1403 }
1404
1405 info = get_line_info(argv[0]);
1406 if (!info) {
1407 static struct enum_map obsolete[] = {
1408 ENUM_MAP("main-delim", LINE_DELIMITER),
1409 ENUM_MAP("main-date", LINE_DATE),
1410 ENUM_MAP("main-author", LINE_AUTHOR),
1411 };
1412 int index;
1413
1414 if (!map_enum(&index, obsolete, argv[0])) {
1415 config_msg = "Unknown color name";
1416 return ERR;
1417 }
1418 info = &line_info[index];
1419 }
1420
1421 if (!set_color(&info->fg, argv[1]) ||
1422 !set_color(&info->bg, argv[2])) {
1423 config_msg = "Unknown color";
1424 return ERR;
1425 }
1426
1427 if (argc == 4 && !set_attribute(&info->attr, argv[3])) {
1428 config_msg = "Unknown attribute";
1429 return ERR;
1430 }
1431
1432 return OK;
1433 }
1434
1435 static int parse_bool(bool *opt, const char *arg)
1436 {
1437 *opt = (!strcmp(arg, "1") || !strcmp(arg, "true") || !strcmp(arg, "yes"))
1438 ? TRUE : FALSE;
1439 return OK;
1440 }
1441
1442 static int
1443 parse_int(int *opt, const char *arg, int min, int max)
1444 {
1445 int value = atoi(arg);
1446
1447 if (min <= value && value <= max)
1448 *opt = value;
1449 return OK;
1450 }
1451
1452 static int
1453 parse_string(char *opt, const char *arg, size_t optsize)
1454 {
1455 int arglen = strlen(arg);
1456
1457 switch (arg[0]) {
1458 case '\"':
1459 case '\'':
1460 if (arglen == 1 || arg[arglen - 1] != arg[0]) {
1461 config_msg = "Unmatched quotation";
1462 return ERR;
1463 }
1464 arg += 1; arglen -= 2;
1465 default:
1466 string_ncopy_do(opt, optsize, arg, strlen(arg));
1467 return OK;
1468 }
1469 }
1470
1471 /* Wants: name = value */
1472 static int
1473 option_set_command(int argc, const char *argv[])
1474 {
1475 if (argc != 3) {
1476 config_msg = "Wrong number of arguments given to set command";
1477 return ERR;
1478 }
1479
1480 if (strcmp(argv[1], "=")) {
1481 config_msg = "No value assigned";
1482 return ERR;
1483 }
1484
1485 if (!strcmp(argv[0], "show-author"))
1486 return parse_bool(&opt_author, argv[2]);
1487
1488 if (!strcmp(argv[0], "show-date"))
1489 return parse_bool(&opt_date, argv[2]);
1490
1491 if (!strcmp(argv[0], "show-rev-graph"))
1492 return parse_bool(&opt_rev_graph, argv[2]);
1493
1494 if (!strcmp(argv[0], "show-refs"))
1495 return parse_bool(&opt_show_refs, argv[2]);
1496
1497 if (!strcmp(argv[0], "show-line-numbers"))
1498 return parse_bool(&opt_line_number, argv[2]);
1499
1500 if (!strcmp(argv[0], "line-graphics"))
1501 return parse_bool(&opt_line_graphics, argv[2]);
1502
1503 if (!strcmp(argv[0], "line-number-interval"))
1504 return parse_int(&opt_num_interval, argv[2], 1, 1024);
1505
1506 if (!strcmp(argv[0], "author-width"))
1507 return parse_int(&opt_author_cols, argv[2], 0, 1024);
1508
1509 if (!strcmp(argv[0], "tab-size"))
1510 return parse_int(&opt_tab_size, argv[2], 1, 1024);
1511
1512 if (!strcmp(argv[0], "commit-encoding"))
1513 return parse_string(opt_encoding, argv[2], sizeof(opt_encoding));
1514
1515 config_msg = "Unknown variable name";
1516 return ERR;
1517 }
1518
1519 /* Wants: mode request key */
1520 static int
1521 option_bind_command(int argc, const char *argv[])
1522 {
1523 enum request request;
1524 int keymap;
1525 int key;
1526
1527 if (argc < 3) {
1528 config_msg = "Wrong number of arguments given to bind command";
1529 return ERR;
1530 }
1531
1532 if (set_keymap(&keymap, argv[0]) == ERR) {
1533 config_msg = "Unknown key map";
1534 return ERR;
1535 }
1536
1537 key = get_key_value(argv[1]);
1538 if (key == ERR) {
1539 config_msg = "Unknown key";
1540 return ERR;
1541 }
1542
1543 request = get_request(argv[2]);
1544 if (request == REQ_NONE) {
1545 static struct enum_map obsolete[] = {
1546 ENUM_MAP("cherry-pick", REQ_NONE),
1547 ENUM_MAP("screen-resize", REQ_NONE),
1548 ENUM_MAP("tree-parent", REQ_PARENT),
1549 };
1550 int alias;
1551
1552 if (map_enum(&alias, obsolete, argv[2])) {
1553 if (alias != REQ_NONE)
1554 add_keybinding(keymap, alias, key);
1555 config_msg = "Obsolete request name";
1556 return ERR;
1557 }
1558 }
1559 if (request == REQ_NONE && *argv[2]++ == '!')
1560 request = add_run_request(keymap, key, argc - 2, argv + 2);
1561 if (request == REQ_NONE) {
1562 config_msg = "Unknown request name";
1563 return ERR;
1564 }
1565
1566 add_keybinding(keymap, request, key);
1567
1568 return OK;
1569 }
1570
1571 static int
1572 set_option(const char *opt, char *value)
1573 {
1574 const char *argv[SIZEOF_ARG];
1575 int argc = 0;
1576
1577 if (!argv_from_string(argv, &argc, value)) {
1578 config_msg = "Too many option arguments";
1579 return ERR;
1580 }
1581
1582 if (!strcmp(opt, "color"))
1583 return option_color_command(argc, argv);
1584
1585 if (!strcmp(opt, "set"))
1586 return option_set_command(argc, argv);
1587
1588 if (!strcmp(opt, "bind"))
1589 return option_bind_command(argc, argv);
1590
1591 config_msg = "Unknown option command";
1592 return ERR;
1593 }
1594
1595 static int
1596 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1597 {
1598 int status = OK;
1599
1600 config_lineno++;
1601 config_msg = "Internal error";
1602
1603 /* Check for comment markers, since read_properties() will
1604 * only ensure opt and value are split at first " \t". */
1605 optlen = strcspn(opt, "#");
1606 if (optlen == 0)
1607 return OK;
1608
1609 if (opt[optlen] != 0) {
1610 config_msg = "No option value";
1611 status = ERR;
1612
1613 } else {
1614 /* Look for comment endings in the value. */
1615 size_t len = strcspn(value, "#");
1616
1617 if (len < valuelen) {
1618 valuelen = len;
1619 value[valuelen] = 0;
1620 }
1621
1622 status = set_option(opt, value);
1623 }
1624
1625 if (status == ERR) {
1626 warn("Error on line %d, near '%.*s': %s",
1627 config_lineno, (int) optlen, opt, config_msg);
1628 config_errors = TRUE;
1629 }
1630
1631 /* Always keep going if errors are encountered. */
1632 return OK;
1633 }
1634
1635 static void
1636 load_option_file(const char *path)
1637 {
1638 struct io io = {};
1639
1640 /* It's OK that the file doesn't exist. */
1641 if (!io_open(&io, path))
1642 return;
1643
1644 config_lineno = 0;
1645 config_errors = FALSE;
1646
1647 if (io_load(&io, " \t", read_option) == ERR ||
1648 config_errors == TRUE)
1649 warn("Errors while loading %s.", path);
1650 }
1651
1652 static int
1653 load_options(void)
1654 {
1655 const char *home = getenv("HOME");
1656 const char *tigrc_user = getenv("TIGRC_USER");
1657 const char *tigrc_system = getenv("TIGRC_SYSTEM");
1658 char buf[SIZEOF_STR];
1659
1660 add_builtin_run_requests();
1661
1662 if (!tigrc_system) {
1663 if (!string_format(buf, "%s/tigrc", SYSCONFDIR))
1664 return ERR;
1665 tigrc_system = buf;
1666 }
1667 load_option_file(tigrc_system);
1668
1669 if (!tigrc_user) {
1670 if (!home || !string_format(buf, "%s/.tigrc", home))
1671 return ERR;
1672 tigrc_user = buf;
1673 }
1674 load_option_file(tigrc_user);
1675
1676 return OK;
1677 }
1678
1679
1680 /*
1681 * The viewer
1682 */
1683
1684 struct view;
1685 struct view_ops;
1686
1687 /* The display array of active views and the index of the current view. */
1688 static struct view *display[2];
1689 static unsigned int current_view;
1690
1691 #define foreach_displayed_view(view, i) \
1692 for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1693
1694 #define displayed_views() (display[1] != NULL ? 2 : 1)
1695
1696 /* Current head and commit ID */
1697 static char ref_blob[SIZEOF_REF] = "";
1698 static char ref_commit[SIZEOF_REF] = "HEAD";
1699 static char ref_head[SIZEOF_REF] = "HEAD";
1700
1701 struct view {
1702 const char *name; /* View name */
1703 const char *cmd_env; /* Command line set via environment */
1704 const char *id; /* Points to either of ref_{head,commit,blob} */
1705
1706 struct view_ops *ops; /* View operations */
1707
1708 enum keymap keymap; /* What keymap does this view have */
1709 bool git_dir; /* Whether the view requires a git directory. */
1710
1711 char ref[SIZEOF_REF]; /* Hovered commit reference */
1712 char vid[SIZEOF_REF]; /* View ID. Set to id member when updating. */
1713
1714 int height, width; /* The width and height of the main window */
1715 WINDOW *win; /* The main window */
1716 WINDOW *title; /* The title window living below the main window */
1717
1718 /* Navigation */
1719 unsigned long offset; /* Offset of the window top */
1720 unsigned long yoffset; /* Offset from the window side. */
1721 unsigned long lineno; /* Current line number */
1722 unsigned long p_offset; /* Previous offset of the window top */
1723 unsigned long p_yoffset;/* Previous offset from the window side */
1724 unsigned long p_lineno; /* Previous current line number */
1725 bool p_restore; /* Should the previous position be restored. */
1726
1727 /* Searching */
1728 char grep[SIZEOF_STR]; /* Search string */
1729 regex_t *regex; /* Pre-compiled regexp */
1730
1731 /* If non-NULL, points to the view that opened this view. If this view
1732 * is closed tig will switch back to the parent view. */
1733 struct view *parent;
1734
1735 /* Buffering */
1736 size_t lines; /* Total number of lines */
1737 struct line *line; /* Line index */
1738 size_t line_alloc; /* Total number of allocated lines */
1739 unsigned int digits; /* Number of digits in the lines member. */
1740
1741 /* Drawing */
1742 struct line *curline; /* Line currently being drawn. */
1743 enum line_type curtype; /* Attribute currently used for drawing. */
1744 unsigned long col; /* Column when drawing. */
1745 bool has_scrolled; /* View was scrolled. */
1746 bool can_hscroll; /* View can be scrolled horizontally. */
1747
1748 /* Loading */
1749 struct io io;
1750 struct io *pipe;
1751 time_t start_time;
1752 time_t update_secs;
1753 };
1754
1755 struct view_ops {
1756 /* What type of content being displayed. Used in the title bar. */
1757 const char *type;
1758 /* Default command arguments. */
1759 const char **argv;
1760 /* Open and reads in all view content. */
1761 bool (*open)(struct view *view);
1762 /* Read one line; updates view->line. */
1763 bool (*read)(struct view *view, char *data);
1764 /* Draw one line; @lineno must be < view->height. */
1765 bool (*draw)(struct view *view, struct line *line, unsigned int lineno);
1766 /* Depending on view handle a special requests. */
1767 enum request (*request)(struct view *view, enum request request, struct line *line);
1768 /* Search for regexp in a line. */
1769 bool (*grep)(struct view *view, struct line *line);
1770 /* Select line */
1771 void (*select)(struct view *view, struct line *line);
1772 };
1773
1774 static struct view_ops blame_ops;
1775 static struct view_ops blob_ops;
1776 static struct view_ops diff_ops;
1777 static struct view_ops help_ops;
1778 static struct view_ops log_ops;
1779 static struct view_ops main_ops;
1780 static struct view_ops pager_ops;
1781 static struct view_ops stage_ops;
1782 static struct view_ops status_ops;
1783 static struct view_ops tree_ops;
1784
1785 #define VIEW_STR(name, env, ref, ops, map, git) \
1786 { name, #env, ref, ops, map, git }
1787
1788 #define VIEW_(id, name, ops, git, ref) \
1789 VIEW_STR(name, TIG_##id##_CMD, ref, ops, KEYMAP_##id, git)
1790
1791
1792 static struct view views[] = {
1793 VIEW_(MAIN, "main", &main_ops, TRUE, ref_head),
1794 VIEW_(DIFF, "diff", &diff_ops, TRUE, ref_commit),
1795 VIEW_(LOG, "log", &log_ops, TRUE, ref_head),
1796 VIEW_(TREE, "tree", &tree_ops, TRUE, ref_commit),
1797 VIEW_(BLOB, "blob", &blob_ops, TRUE, ref_blob),
1798 VIEW_(BLAME, "blame", &blame_ops, TRUE, ref_commit),
1799 VIEW_(HELP, "help", &help_ops, FALSE, ""),
1800 VIEW_(PAGER, "pager", &pager_ops, FALSE, "stdin"),
1801 VIEW_(STATUS, "status", &status_ops, TRUE, ""),
1802 VIEW_(STAGE, "stage", &stage_ops, TRUE, ""),
1803 };
1804
1805 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1806 #define VIEW_REQ(view) ((view) - views + REQ_OFFSET + 1)
1807
1808 #define foreach_view(view, i) \
1809 for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1810
1811 #define view_is_displayed(view) \
1812 (view == display[0] || view == display[1])
1813
1814
1815 enum line_graphic {
1816 LINE_GRAPHIC_VLINE
1817 };
1818
1819 static int line_graphics[] = {
1820 /* LINE_GRAPHIC_VLINE: */ '|'
1821 };
1822
1823 static inline void
1824 set_view_attr(struct view *view, enum line_type type)
1825 {
1826 if (!view->curline->selected && view->curtype != type) {
1827 wattrset(view->win, get_line_attr(type));
1828 wchgat(view->win, -1, 0, type, NULL);
1829 view->curtype = type;
1830 }
1831 }
1832
1833 static int
1834 draw_chars(struct view *view, enum line_type type, const char *string,
1835 int max_len, bool use_tilde)
1836 {
1837 int len = 0;
1838 int col = 0;
1839 int trimmed = FALSE;
1840 size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
1841
1842 if (max_len <= 0)
1843 return 0;
1844
1845 if (opt_utf8) {
1846 len = utf8_length(&string, skip, &col, max_len, &trimmed, use_tilde);
1847 } else {
1848 col = len = strlen(string);
1849 if (len > max_len) {
1850 if (use_tilde) {
1851 max_len -= 1;
1852 }
1853 col = len = max_len;
1854 trimmed = TRUE;
1855 }
1856 }
1857
1858 set_view_attr(view, type);
1859 if (len > 0)
1860 waddnstr(view->win, string, len);
1861 if (trimmed && use_tilde) {
1862 set_view_attr(view, LINE_DELIMITER);
1863 waddch(view->win, '~');
1864 col++;
1865 }
1866
1867 if (view->col + col >= view->width + view->yoffset)
1868 view->can_hscroll = TRUE;
1869
1870 return col;
1871 }
1872
1873 static int
1874 draw_space(struct view *view, enum line_type type, int max, int spaces)
1875 {
1876 static char space[] = " ";
1877 int col = 0;
1878
1879 spaces = MIN(max, spaces);
1880
1881 while (spaces > 0) {
1882 int len = MIN(spaces, sizeof(space) - 1);
1883
1884 col += draw_chars(view, type, space, spaces, FALSE);
1885 spaces -= len;
1886 }
1887
1888 return col;
1889 }
1890
1891 static bool
1892 draw_lineno(struct view *view, unsigned int lineno)
1893 {
1894 size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
1895 char number[10];
1896 int digits3 = view->digits < 3 ? 3 : view->digits;
1897 int max_number = MIN(digits3, STRING_SIZE(number));
1898 int max = view->width - view->col;
1899 int col;
1900
1901 if (max < max_number)
1902 max_number = max;
1903
1904 lineno += view->offset + 1;
1905 if (lineno == 1 || (lineno % opt_num_interval) == 0) {
1906 static char fmt[] = "%1ld";
1907
1908 if (view->digits <= 9)
1909 fmt[1] = '0' + digits3;
1910
1911 if (!string_format(number, fmt, lineno))
1912 number[0] = 0;
1913 col = draw_chars(view, LINE_LINE_NUMBER, number, max_number, TRUE);
1914 } else {
1915 col = draw_space(view, LINE_LINE_NUMBER, max_number, max_number);
1916 }
1917
1918 if (col < max && skip <= col) {
1919 set_view_attr(view, LINE_DEFAULT);
1920 waddch(view->win, line_graphics[LINE_GRAPHIC_VLINE]);
1921 }
1922 col++;
1923
1924 view->col += col;
1925 if (col < max && skip <= col)
1926 col = draw_space(view, LINE_DEFAULT, max - col, 1);
1927 view->col++;
1928
1929 return view->width + view->yoffset <= view->col;
1930 }
1931
1932 static bool
1933 draw_text(struct view *view, enum line_type type, const char *string, bool trim)
1934 {
1935 view->col += draw_chars(view, type, string, view->width + view->yoffset - view->col, trim);
1936 return view->width - view->col <= 0;
1937 }
1938
1939 static bool
1940 draw_graphic(struct view *view, enum line_type type, chtype graphic[], size_t size)
1941 {
1942 size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
1943 int max = view->width - view->col;
1944 int i;
1945
1946 if (max < size)
1947 size = max;
1948
1949 set_view_attr(view, type);
1950 /* Using waddch() instead of waddnstr() ensures that
1951 * they'll be rendered correctly for the cursor line. */
1952 for (i = skip; i < size; i++)
1953 waddch(view->win, graphic[i]);
1954
1955 view->col += size;
1956 if (size < max && skip <= size)
1957 waddch(view->win, ' ');
1958 view->col++;
1959
1960 return view->width - view->col <= 0;
1961 }
1962
1963 static bool
1964 draw_field(struct view *view, enum line_type type, const char *text, int len, bool trim)
1965 {
1966 int max = MIN(view->width - view->col, len);
1967 int col;
1968
1969 if (text)
1970 col = draw_chars(view, type, text, max - 1, trim);
1971 else
1972 col = draw_space(view, type, max - 1, max - 1);
1973
1974 view->col += col;
1975 view->col += draw_space(view, LINE_DEFAULT, max - col, max - col);
1976 return view->width + view->yoffset <= view->col;
1977 }
1978
1979 static bool
1980 draw_date(struct view *view, struct tm *time)
1981 {
1982 char buf[DATE_COLS];
1983 char *date;
1984 int timelen = 0;
1985
1986 if (time)
1987 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, time);
1988 date = timelen ? buf : NULL;
1989
1990 return draw_field(view, LINE_DATE, date, DATE_COLS, FALSE);
1991 }
1992
1993 static bool
1994 draw_author(struct view *view, const char *author)
1995 {
1996 bool trim = opt_author_cols == 0 || opt_author_cols > 5 || !author;
1997
1998 if (!trim) {
1999 static char initials[10];
2000 size_t pos;
2001
2002 #define is_initial_sep(c) (isspace(c) || ispunct(c) || (c) == '@')
2003
2004 memset(initials, 0, sizeof(initials));
2005 for (pos = 0; *author && pos < opt_author_cols - 1; author++, pos++) {
2006 while (is_initial_sep(*author))
2007 author++;
2008 strncpy(&initials[pos], author, sizeof(initials) - 1 - pos);
2009 while (*author && !is_initial_sep(author[1]))
2010 author++;
2011 }
2012
2013 author = initials;
2014 }
2015
2016 return draw_field(view, LINE_AUTHOR, author, opt_author_cols, trim);
2017 }
2018
2019 static bool
2020 draw_mode(struct view *view, mode_t mode)
2021 {
2022 static const char dir_mode[] = "drwxr-xr-x";
2023 static const char link_mode[] = "lrwxrwxrwx";
2024 static const char exe_mode[] = "-rwxr-xr-x";
2025 static const char file_mode[] = "-rw-r--r--";
2026 const char *str;
2027
2028 if (S_ISDIR(mode))
2029 str = dir_mode;
2030 else if (S_ISLNK(mode))
2031 str = link_mode;
2032 else if (mode & S_IXUSR)
2033 str = exe_mode;
2034 else
2035 str = file_mode;
2036
2037 return draw_field(view, LINE_MODE, str, sizeof(file_mode), FALSE);
2038 }
2039
2040 static bool
2041 draw_view_line(struct view *view, unsigned int lineno)
2042 {
2043 struct line *line;
2044 bool selected = (view->offset + lineno == view->lineno);
2045
2046 assert(view_is_displayed(view));
2047
2048 if (view->offset + lineno >= view->lines)
2049 return FALSE;
2050
2051 line = &view->line[view->offset + lineno];
2052
2053 wmove(view->win, lineno, 0);
2054 if (line->cleareol)
2055 wclrtoeol(view->win);
2056 view->col = 0;
2057 view->curline = line;
2058 view->curtype = LINE_NONE;
2059 line->selected = FALSE;
2060 line->dirty = line->cleareol = 0;
2061
2062 if (selected) {
2063 set_view_attr(view, LINE_CURSOR);
2064 line->selected = TRUE;
2065 view->ops->select(view, line);
2066 }
2067
2068 return view->ops->draw(view, line, lineno);
2069 }
2070
2071 static void
2072 redraw_view_dirty(struct view *view)
2073 {
2074 bool dirty = FALSE;
2075 int lineno;
2076
2077 for (lineno = 0; lineno < view->height; lineno++) {
2078 if (view->offset + lineno >= view->lines)
2079 break;
2080 if (!view->line[view->offset + lineno].dirty)
2081 continue;
2082 dirty = TRUE;
2083 if (!draw_view_line(view, lineno))
2084 break;
2085 }
2086
2087 if (!dirty)
2088 return;
2089 wnoutrefresh(view->win);
2090 }
2091
2092 static void
2093 redraw_view_from(struct view *view, int lineno)
2094 {
2095 assert(0 <= lineno && lineno < view->height);
2096
2097 if (lineno == 0)
2098 view->can_hscroll = FALSE;
2099
2100 for (; lineno < view->height; lineno++) {
2101 if (!draw_view_line(view, lineno))
2102 break;
2103 }
2104
2105 wnoutrefresh(view->win);
2106 }
2107
2108 static void
2109 redraw_view(struct view *view)
2110 {
2111 werase(view->win);
2112 redraw_view_from(view, 0);
2113 }
2114
2115
2116 static void
2117 update_view_title(struct view *view)
2118 {
2119 char buf[SIZEOF_STR];
2120 char state[SIZEOF_STR];
2121 size_t bufpos = 0, statelen = 0;
2122
2123 assert(view_is_displayed(view));
2124
2125 if (view != VIEW(REQ_VIEW_STATUS) && view->lines) {
2126 unsigned int view_lines = view->offset + view->height;
2127 unsigned int lines = view->lines
2128 ? MIN(view_lines, view->lines) * 100 / view->lines
2129 : 0;
2130
2131 string_format_from(state, &statelen, " - %s %d of %d (%d%%)",
2132 view->ops->type,
2133 view->lineno + 1,
2134 view->lines,
2135 lines);
2136
2137 }
2138
2139 if (view->pipe) {
2140 time_t secs = time(NULL) - view->start_time;
2141
2142 /* Three git seconds are a long time ... */
2143 if (secs > 2)
2144 string_format_from(state, &statelen, " loading %lds", secs);
2145 }
2146
2147 string_format_from(buf, &bufpos, "[%s]", view->name);
2148 if (*view->ref && bufpos < view->width) {
2149 size_t refsize = strlen(view->ref);
2150 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
2151
2152 if (minsize < view->width)
2153 refsize = view->width - minsize + 7;
2154 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
2155 }
2156
2157 if (statelen && bufpos < view->width) {
2158 string_format_from(buf, &bufpos, "%s", state);
2159 }
2160
2161 if (view == display[current_view])
2162 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
2163 else
2164 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
2165
2166 mvwaddnstr(view->title, 0, 0, buf, bufpos);
2167 wclrtoeol(view->title);
2168 wnoutrefresh(view->title);
2169 }
2170
2171 static void
2172 resize_display(void)
2173 {
2174 int offset, i;
2175 struct view *base = display[0];
2176 struct view *view = display[1] ? display[1] : display[0];
2177
2178 /* Setup window dimensions */
2179
2180 getmaxyx(stdscr, base->height, base->width);
2181
2182 /* Make room for the status window. */
2183 base->height -= 1;
2184
2185 if (view != base) {
2186 /* Horizontal split. */
2187 view->width = base->width;
2188 view->height = SCALE_SPLIT_VIEW(base->height);
2189 base->height -= view->height;
2190
2191 /* Make room for the title bar. */
2192 view->height -= 1;
2193 }
2194
2195 /* Make room for the title bar. */
2196 base->height -= 1;
2197
2198 offset = 0;
2199
2200 foreach_displayed_view (view, i) {
2201 if (!view->win) {
2202 view->win = newwin(view->height, 0, offset, 0);
2203 if (!view->win)
2204 die("Failed to create %s view", view->name);
2205
2206 scrollok(view->win, FALSE);
2207
2208 view->title = newwin(1, 0, offset + view->height, 0);
2209 if (!view->title)
2210 die("Failed to create title window");
2211
2212 } else {
2213 wresize(view->win, view->height, view->width);
2214 mvwin(view->win, offset, 0);
2215 mvwin(view->title, offset + view->height, 0);
2216 }
2217
2218 offset += view->height + 1;
2219 }
2220 }
2221
2222 static void
2223 redraw_display(bool clear)
2224 {
2225 struct view *view;
2226 int i;
2227
2228 foreach_displayed_view (view, i) {
2229 if (clear)
2230 wclear(view->win);
2231 redraw_view(view);
2232 update_view_title(view);
2233 }
2234 }
2235
2236 static void
2237 toggle_view_option(bool *option, const char *help)
2238 {
2239 *option = !*option;
2240 redraw_display(FALSE);
2241 report("%sabling %s", *option ? "En" : "Dis", help);
2242 }
2243
2244 /*
2245 * Navigation
2246 */
2247
2248 /* Scrolling backend */
2249 static void
2250 do_scroll_view(struct view *view, int lines)
2251 {
2252 bool redraw_current_line = FALSE;
2253
2254 /* The rendering expects the new offset. */
2255 view->offset += lines;
2256
2257 assert(0 <= view->offset && view->offset < view->lines);
2258 assert(lines);
2259
2260 /* Move current line into the view. */
2261 if (view->lineno < view->offset) {
2262 view->lineno = view->offset;
2263 redraw_current_line = TRUE;
2264 } else if (view->lineno >= view->offset + view->height) {
2265 view->lineno = view->offset + view->height - 1;
2266 redraw_current_line = TRUE;
2267 }
2268
2269 assert(view->offset <= view->lineno && view->lineno < view->lines);
2270
2271 /* Redraw the whole screen if scrolling is pointless. */
2272 if (view->height < ABS(lines)) {
2273 redraw_view(view);
2274
2275 } else {
2276 int line = lines > 0 ? view->height - lines : 0;
2277 int end = line + ABS(lines);
2278
2279 scrollok(view->win, TRUE);
2280 wscrl(view->win, lines);
2281 scrollok(view->win, FALSE);
2282
2283 while (line < end && draw_view_line(view, line))
2284 line++;
2285
2286 if (redraw_current_line)
2287 draw_view_line(view, view->lineno - view->offset);
2288 wnoutrefresh(view->win);
2289 }
2290
2291 view->has_scrolled = TRUE;
2292 report("");
2293 }
2294
2295 /* Scroll frontend */
2296 static void
2297 scroll_view(struct view *view, enum request request)
2298 {
2299 int lines = 1;
2300
2301 assert(view_is_displayed(view));
2302
2303 switch (request) {
2304 case REQ_SCROLL_LEFT:
2305 if (view->yoffset == 0) {
2306 report("Cannot scroll beyond the first column");
2307 return;
2308 }
2309 if (view->yoffset <= SCROLL_INTERVAL)
2310 view->yoffset = 0;
2311 else
2312 view->yoffset -= SCROLL_INTERVAL;
2313 redraw_view_from(view, 0);
2314 report("");
2315 return;
2316 case REQ_SCROLL_RIGHT:
2317 if (!view->can_hscroll) {
2318 report("Cannot scroll beyond the last column");
2319 return;
2320 }
2321 view->yoffset += SCROLL_INTERVAL;
2322 redraw_view(view);
2323 report("");
2324 return;
2325 case REQ_SCROLL_PAGE_DOWN:
2326 lines = view->height;
2327 case REQ_SCROLL_LINE_DOWN:
2328 if (view->offset + lines > view->lines)
2329 lines = view->lines - view->offset;
2330
2331 if (lines == 0 || view->offset + view->height >= view->lines) {
2332 report("Cannot scroll beyond the last line");
2333 return;
2334 }
2335 break;
2336
2337 case REQ_SCROLL_PAGE_UP:
2338 lines = view->height;
2339 case REQ_SCROLL_LINE_UP:
2340 if (lines > view->offset)
2341 lines = view->offset;
2342
2343 if (lines == 0) {
2344 report("Cannot scroll beyond the first line");
2345 return;
2346 }
2347
2348 lines = -lines;
2349 break;
2350
2351 default:
2352 die("request %d not handled in switch", request);
2353 }
2354
2355 do_scroll_view(view, lines);
2356 }
2357
2358 /* Cursor moving */
2359 static void
2360 move_view(struct view *view, enum request request)
2361 {
2362 int scroll_steps = 0;
2363 int steps;
2364
2365 switch (request) {
2366 case REQ_MOVE_FIRST_LINE:
2367 steps = -view->lineno;
2368 break;
2369
2370 case REQ_MOVE_LAST_LINE:
2371 steps = view->lines - view->lineno - 1;
2372 break;
2373
2374 case REQ_MOVE_PAGE_UP:
2375 steps = view->height > view->lineno
2376 ? -view->lineno : -view->height;
2377 break;
2378
2379 case REQ_MOVE_PAGE_DOWN:
2380 steps = view->lineno + view->height >= view->lines
2381 ? view->lines - view->lineno - 1 : view->height;
2382 break;
2383
2384 case REQ_MOVE_UP:
2385 steps = -1;
2386 break;
2387
2388 case REQ_MOVE_DOWN:
2389 steps = 1;
2390 break;
2391
2392 default:
2393 die("request %d not handled in switch", request);
2394 }
2395
2396 if (steps <= 0 && view->lineno == 0) {
2397 report("Cannot move beyond the first line");
2398 return;
2399
2400 } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
2401 report("Cannot move beyond the last line");
2402 return;
2403 }
2404
2405 /* Move the current line */
2406 view->lineno += steps;
2407 assert(0 <= view->lineno && view->lineno < view->lines);
2408
2409 /* Check whether the view needs to be scrolled */
2410 if (view->lineno < view->offset ||
2411 view->lineno >= view->offset + view->height) {
2412 scroll_steps = steps;
2413 if (steps < 0 && -steps > view->offset) {
2414 scroll_steps = -view->offset;
2415
2416 } else if (steps > 0) {
2417 if (view->lineno == view->lines - 1 &&
2418 view->lines > view->height) {
2419 scroll_steps = view->lines - view->offset - 1;
2420 if (scroll_steps >= view->height)
2421 scroll_steps -= view->height - 1;
2422 }
2423 }
2424 }
2425
2426 if (!view_is_displayed(view)) {
2427 view->offset += scroll_steps;
2428 assert(0 <= view->offset && view->offset < view->lines);
2429 view->ops->select(view, &view->line[view->lineno]);
2430 return;
2431 }
2432
2433 /* Repaint the old "current" line if we be scrolling */
2434 if (ABS(steps) < view->height)
2435 draw_view_line(view, view->lineno - steps - view->offset);
2436
2437 if (scroll_steps) {
2438 do_scroll_view(view, scroll_steps);
2439 return;
2440 }
2441
2442 /* Draw the current line */
2443 draw_view_line(view, view->lineno - view->offset);
2444
2445 wnoutrefresh(view->win);
2446 report("");
2447 }
2448
2449
2450 /*
2451 * Searching
2452 */
2453
2454 static void search_view(struct view *view, enum request request);
2455
2456 static void
2457 select_view_line(struct view *view, unsigned long lineno)
2458 {
2459 if (lineno - view->offset >= view->height) {
2460 view->offset = lineno;
2461 view->lineno = lineno;
2462 if (view_is_displayed(view))
2463 redraw_view(view);
2464
2465 } else {
2466 unsigned long old_lineno = view->lineno - view->offset;
2467
2468 view->lineno = lineno;
2469 if (view_is_displayed(view)) {
2470 draw_view_line(view, old_lineno);
2471 draw_view_line(view, view->lineno - view->offset);
2472 wnoutrefresh(view->win);
2473 } else {
2474 view->ops->select(view, &view->line[view->lineno]);
2475 }
2476 }
2477 }
2478
2479 static void
2480 find_next(struct view *view, enum request request)
2481 {
2482 unsigned long lineno = view->lineno;
2483 int direction;
2484
2485 if (!*view->grep) {
2486 if (!*opt_search)
2487 report("No previous search");
2488 else
2489 search_view(view, request);
2490 return;
2491 }
2492
2493 switch (request) {
2494 case REQ_SEARCH:
2495 case REQ_FIND_NEXT:
2496 direction = 1;
2497 break;
2498
2499 case REQ_SEARCH_BACK:
2500 case REQ_FIND_PREV:
2501 direction = -1;
2502 break;
2503
2504 default:
2505 return;
2506 }
2507
2508 if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
2509 lineno += direction;
2510
2511 /* Note, lineno is unsigned long so will wrap around in which case it
2512 * will become bigger than view->lines. */
2513 for (; lineno < view->lines; lineno += direction) {
2514 if (view->ops->grep(view, &view->line[lineno])) {
2515 select_view_line(view, lineno);
2516 report("Line %ld matches '%s'", lineno + 1, view->grep);
2517 return;
2518 }
2519 }
2520
2521 report("No match found for '%s'", view->grep);
2522 }
2523
2524 static void
2525 search_view(struct view *view, enum request request)
2526 {
2527 int regex_err;
2528
2529 if (view->regex) {
2530 regfree(view->regex);
2531 *view->grep = 0;
2532 } else {
2533 view->regex = calloc(1, sizeof(*view->regex));
2534 if (!view->regex)
2535 return;
2536 }
2537
2538 regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
2539 if (regex_err != 0) {
2540 char buf[SIZEOF_STR] = "unknown error";
2541
2542 regerror(regex_err, view->regex, buf, sizeof(buf));
2543 report("Search failed: %s", buf);
2544 return;
2545 }
2546
2547 string_copy(view->grep, opt_search);
2548
2549 find_next(view, request);
2550 }
2551
2552 /*
2553 * Incremental updating
2554 */
2555
2556 static void
2557 reset_view(struct view *view)
2558 {
2559 int i;
2560
2561 for (i = 0; i < view->lines; i++)
2562 free(view->line[i].data);
2563 free(view->line);
2564
2565 view->p_offset = view->offset;
2566 view->p_yoffset = view->yoffset;
2567 view->p_lineno = view->lineno;
2568
2569 view->line = NULL;
2570 view->offset = 0;
2571 view->yoffset = 0;
2572 view->lines = 0;
2573 view->lineno = 0;
2574 view->line_alloc = 0;
2575 view->vid[0] = 0;
2576 view->update_secs = 0;
2577 }
2578
2579 static void
2580 free_argv(const char *argv[])
2581 {
2582 int argc;
2583
2584 for (argc = 0; argv[argc]; argc++)
2585 free((void *) argv[argc]);
2586 }
2587
2588 static bool
2589 format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags flags)
2590 {
2591 char buf[SIZEOF_STR];
2592 int argc;
2593 bool noreplace = flags == FORMAT_NONE;
2594
2595 free_argv(dst_argv);
2596
2597 for (argc = 0; src_argv[argc]; argc++) {
2598 const char *arg = src_argv[argc];
2599 size_t bufpos = 0;
2600
2601 while (arg) {
2602 char *next = strstr(arg, "%(");
2603 int len = next - arg;
2604 const char *value;
2605
2606 if (!next || noreplace) {
2607 if (flags == FORMAT_DASH && !strcmp(arg, "--"))
2608 noreplace = TRUE;
2609 len = strlen(arg);
2610 value = "";
2611
2612 } else if (!prefixcmp(next, "%(directory)")) {
2613 value = opt_path;
2614
2615 } else if (!prefixcmp(next, "%(file)")) {
2616 value = opt_file;
2617
2618 } else if (!prefixcmp(next, "%(ref)")) {
2619 value = *opt_ref ? opt_ref : "HEAD";
2620
2621 } else if (!prefixcmp(next, "%(head)")) {
2622 value = ref_head;
2623
2624 } else if (!prefixcmp(next, "%(commit)")) {
2625 value = ref_commit;
2626
2627 } else if (!prefixcmp(next, "%(blob)")) {
2628 value = ref_blob;
2629
2630 } else {
2631 report("Unknown replacement: `%s`", next);
2632 return FALSE;
2633 }
2634
2635 if (!string_format_from(buf, &bufpos, "%.*s%s", len, arg, value))
2636 return FALSE;
2637
2638 arg = next && !noreplace ? strchr(next, ')') + 1 : NULL;
2639 }
2640
2641 dst_argv[argc] = strdup(buf);
2642 if (!dst_argv[argc])
2643 break;
2644 }
2645
2646 dst_argv[argc] = NULL;
2647
2648 return src_argv[argc] == NULL;
2649 }
2650
2651 static bool
2652 restore_view_position(struct view *view)
2653 {
2654 if (!view->p_restore || (view->pipe && view->lines <= view->p_lineno))
2655 return FALSE;
2656
2657 /* Changing the view position cancels the restoring. */
2658 /* FIXME: Changing back to the first line is not detected. */
2659 if (view->offset != 0 || view->lineno != 0) {
2660 view->p_restore = FALSE;
2661 return FALSE;
2662 }
2663
2664 if (view->p_lineno >= view->lines) {
2665 view->p_lineno = view->lines > 0 ? view->lines - 1 : 0;
2666 if (view->p_offset >= view->p_lineno) {
2667 unsigned long half = view->height / 2;
2668
2669 if (view->p_lineno > half)
2670 view->p_offset = view->p_lineno - half;
2671 else
2672 view->p_offset = 0;
2673 }
2674 }
2675
2676 if (view_is_displayed(view) &&
2677 view->offset != view->p_offset &&
2678 view->lineno != view->p_lineno)
2679 werase(view->win);
2680
2681 view->offset = view->p_offset;
2682 view->yoffset = view->p_yoffset;
2683 view->lineno = view->p_lineno;
2684 view->p_restore = FALSE;
2685
2686 return TRUE;
2687 }
2688
2689 static void
2690 end_update(struct view *view, bool force)
2691 {
2692 if (!view->pipe)
2693 return;
2694 while (!view->ops->read(view, NULL))
2695 if (!force)
2696 return;
2697 set_nonblocking_input(FALSE);
2698 if (force)
2699 kill_io(view->pipe);
2700 done_io(view->pipe);
2701 view->pipe = NULL;
2702 }
2703
2704 static void
2705 setup_update(struct view *view, const char *vid)
2706 {
2707 set_nonblocking_input(TRUE);
2708 reset_view(view);
2709 string_copy_rev(view->vid, vid);
2710 view->pipe = &view->io;
2711 view->start_time = time(NULL);
2712 }
2713
2714 static bool
2715 prepare_update(struct view *view, const char *argv[], const char *dir,
2716 enum format_flags flags)
2717 {
2718 if (view->pipe)
2719 end_update(view, TRUE);
2720 return init_io_rd(&view->io, argv, dir, flags);
2721 }
2722
2723 static bool
2724 prepare_update_file(struct view *view, const char *name)
2725 {
2726 if (view->pipe)
2727 end_update(view, TRUE);
2728 return io_open(&view->io, name);
2729 }
2730
2731 static bool
2732 begin_update(struct view *view, bool refresh)
2733 {
2734 if (view->pipe)
2735 end_update(view, TRUE);
2736
2737 if (refresh) {
2738 if (!start_io(&view->io))
2739 return FALSE;
2740
2741 } else {
2742 if (view == VIEW(REQ_VIEW_TREE) && strcmp(view->vid, view->id))
2743 opt_path[0] = 0;
2744
2745 if (!run_io_rd(&view->io, view->ops->argv, FORMAT_ALL))
2746 return FALSE;
2747
2748 /* Put the current ref_* value to the view title ref
2749 * member. This is needed by the blob view. Most other
2750 * views sets it automatically after loading because the
2751 * first line is a commit line. */
2752 string_copy_rev(view->ref, view->id);
2753 }
2754
2755 setup_update(view, view->id);
2756
2757 return TRUE;
2758 }
2759
2760 #define ITEM_CHUNK_SIZE 256
2761 static void *
2762 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2763 {
2764 size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2765 size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2766
2767 if (mem == NULL || num_chunks != num_chunks_new) {
2768 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2769 mem = realloc(mem, *size * item_size);
2770 }
2771
2772 return mem;
2773 }
2774
2775 static struct line *
2776 realloc_lines(struct view *view, size_t line_size)
2777 {
2778 size_t alloc = view->line_alloc;
2779 struct line *tmp = realloc_items(view->line, &alloc, line_size,
2780 sizeof(*view->line));
2781
2782 if (!tmp)
2783 return NULL;
2784
2785 view->line = tmp;
2786 view->line_alloc = alloc;
2787 return view->line;
2788 }
2789
2790 static bool
2791 update_view(struct view *view)
2792 {
2793 char out_buffer[BUFSIZ * 2];
2794 char *line;
2795 /* Clear the view and redraw everything since the tree sorting
2796 * might have rearranged things. */
2797 bool redraw = view->lines == 0;
2798 bool can_read = TRUE;
2799
2800 if (!view->pipe)
2801 return TRUE;
2802
2803 if (!io_can_read(view->pipe)) {
2804 if (view->lines == 0) {
2805 time_t secs = time(NULL) - view->start_time;
2806
2807 if (secs > 1 && secs > view->update_secs) {
2808 if (view->update_secs == 0)
2809 redraw_view(view);
2810 update_view_title(view);
2811 view->update_secs = secs;
2812 }
2813 }
2814 return TRUE;
2815 }
2816
2817 for (; (line = io_get(view->pipe, '\n', can_read)); can_read = FALSE) {
2818 if (opt_iconv != ICONV_NONE) {
2819 ICONV_CONST char *inbuf = line;
2820 size_t inlen = strlen(line) + 1;
2821
2822 char *outbuf = out_buffer;
2823 size_t outlen = sizeof(out_buffer);
2824
2825 size_t ret;
2826
2827 ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2828 if (ret != (size_t) -1)
2829 line = out_buffer;
2830 }
2831
2832 if (!view->ops->read(view, line)) {
2833 report("Allocation failure");
2834 end_update(view, TRUE);
2835 return FALSE;
2836 }
2837 }
2838
2839 {
2840 unsigned long lines = view->lines;
2841 int digits;
2842
2843 for (digits = 0; lines; digits++)
2844 lines /= 10;
2845
2846 /* Keep the displayed view in sync with line number scaling. */
2847 if (digits != view->digits) {
2848 view->digits = digits;
2849 if (opt_line_number || view == VIEW(REQ_VIEW_BLAME))
2850 redraw = TRUE;
2851 }
2852 }
2853
2854 if (io_error(view->pipe)) {
2855 report("Failed to read: %s", io_strerror(view->pipe));
2856 end_update(view, TRUE);
2857
2858 } else if (io_eof(view->pipe)) {
2859 report("");
2860 end_update(view, FALSE);
2861 }
2862
2863 if (restore_view_position(view))
2864 redraw = TRUE;
2865
2866 if (!view_is_displayed(view))
2867 return TRUE;
2868
2869 if (redraw)
2870 redraw_view_from(view, 0);
2871 else
2872 redraw_view_dirty(view);
2873
2874 /* Update the title _after_ the redraw so that if the redraw picks up a
2875 * commit reference in view->ref it'll be available here. */
2876 update_view_title(view);
2877 return TRUE;
2878 }
2879
2880 static struct line *
2881 add_line_data(struct view *view, void *data, enum line_type type)
2882 {
2883 struct line *line;
2884
2885 if (!realloc_lines(view, view->lines + 1))
2886 return NULL;
2887
2888 line = &view->line[view->lines++];
2889 memset(line, 0, sizeof(*line));
2890 line->type = type;
2891 line->data = data;
2892 line->dirty = 1;
2893
2894 return line;
2895 }
2896
2897 static struct line *
2898 add_line_text(struct view *view, const char *text, enum line_type type)
2899 {
2900 char *data = text ? strdup(text) : NULL;
2901
2902 return data ? add_line_data(view, data, type) : NULL;
2903 }
2904
2905 static struct line *
2906 add_line_format(struct view *view, enum line_type type, const char *fmt, ...)
2907 {
2908 char buf[SIZEOF_STR];
2909 va_list args;
2910
2911 va_start(args, fmt);
2912 if (vsnprintf(buf, sizeof(buf), fmt, args) >= sizeof(buf))
2913 buf[0] = 0;
2914 va_end(args);
2915
2916 return buf[0] ? add_line_text(view, buf, type) : NULL;
2917 }
2918
2919 /*
2920 * View opening
2921 */
2922
2923 enum open_flags {
2924 OPEN_DEFAULT = 0, /* Use default view switching. */
2925 OPEN_SPLIT = 1, /* Split current view. */
2926 OPEN_RELOAD = 4, /* Reload view even if it is the current. */
2927 OPEN_NOMAXIMIZE = 8, /* Do not maximize the current view. */
2928 OPEN_REFRESH = 16, /* Refresh view using previous command. */
2929 OPEN_PREPARED = 32, /* Open already prepared command. */
2930 };
2931
2932 static void
2933 open_view(struct view *prev, enum request request, enum open_flags flags)
2934 {
2935 bool split = !!(flags & OPEN_SPLIT);
2936 bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH | OPEN_PREPARED));
2937 bool nomaximize = !!(flags & (OPEN_NOMAXIMIZE | OPEN_REFRESH));
2938 struct view *view = VIEW(request);
2939 int nviews = displayed_views();
2940 struct view *base_view = display[0];
2941
2942 if (view == prev && nviews == 1 && !reload) {
2943 report("Already in %s view", view->name);
2944 return;
2945 }
2946
2947 if (view->git_dir && !opt_git_dir[0]) {
2948 report("The %s view is disabled in pager view", view->name);
2949 return;
2950 }
2951
2952 if (split) {
2953 display[1] = view;
2954 current_view = 1;
2955 } else if (!nomaximize) {
2956 /* Maximize the current view. */
2957 memset(display, 0, sizeof(display));
2958 current_view = 0;
2959 display[current_view] = view;
2960 }
2961
2962 /* Resize the view when switching between split- and full-screen,
2963 * or when switching between two different full-screen views. */
2964 if (nviews != displayed_views() ||
2965 (nviews == 1 && base_view != display[0]))
2966 resize_display();
2967
2968 if (view->ops->open) {
2969 if (view->pipe)
2970 end_update(view, TRUE);
2971 if (!view->ops->open(view)) {
2972 report("Failed to load %s view", view->name);
2973 return;
2974 }
2975 restore_view_position(view);
2976
2977 } else if ((reload || strcmp(view->vid, view->id)) &&
2978 !begin_update(view, flags & (OPEN_REFRESH | OPEN_PREPARED))) {
2979 report("Failed to load %s view", view->name);
2980 return;
2981 }
2982
2983 if (split && prev->lineno - prev->offset >= prev->height) {
2984 /* Take the title line into account. */
2985 int lines = prev->lineno - prev->offset - prev->height + 1;
2986
2987 /* Scroll the view that was split if the current line is
2988 * outside the new limited view. */
2989 do_scroll_view(prev, lines);
2990 }
2991
2992 if (prev && view != prev) {
2993 if (split) {
2994 /* "Blur" the previous view. */
2995 update_view_title(prev);
2996 }
2997
2998 view->parent = prev;
2999 }
3000
3001 if (view->pipe && view->lines == 0) {
3002 /* Clear the old view and let the incremental updating refill
3003 * the screen. */
3004 werase(view->win);
3005 view->p_restore = flags & (OPEN_RELOAD | OPEN_REFRESH);
3006 report("");
3007 } else if (view_is_displayed(view)) {
3008 redraw_view(view);
3009 report("");
3010 }
3011 }
3012
3013 static void
3014 open_external_viewer(const char *argv[], const char *dir)
3015 {
3016 def_prog_mode(); /* save current tty modes */
3017 endwin(); /* restore original tty modes */
3018 run_io_fg(argv, dir);
3019 fprintf(stderr, "Press Enter to continue");
3020 getc(opt_tty);
3021 reset_prog_mode();
3022 redraw_display(TRUE);
3023 }
3024
3025 static void
3026 open_mergetool(const char *file)
3027 {
3028 const char *mergetool_argv[] = { "git", "mergetool", file, NULL };
3029
3030 open_external_viewer(mergetool_argv, opt_cdup);
3031 }
3032
3033 static void
3034 open_editor(bool from_root, const char *file)
3035 {
3036 const char *editor_argv[] = { "vi", file, NULL };
3037 const char *editor;
3038
3039 editor = getenv("GIT_EDITOR");
3040 if (!editor && *opt_editor)
3041 editor = opt_editor;
3042 if (!editor)
3043 editor = getenv("VISUAL");
3044 if (!editor)
3045 editor = getenv("EDITOR");
3046 if (!editor)
3047 editor = "vi";
3048
3049 editor_argv[0] = editor;
3050 open_external_viewer(editor_argv, from_root ? opt_cdup : NULL);
3051 }
3052
3053 static void
3054 open_run_request(enum request request)
3055 {
3056 struct run_request *req = get_run_request(request);
3057 const char *argv[ARRAY_SIZE(req->argv)] = { NULL };
3058
3059 if (!req) {
3060 report("Unknown run request");
3061 return;
3062 }
3063
3064 if (format_argv(argv, req->argv, FORMAT_ALL))
3065 open_external_viewer(argv, NULL);
3066 free_argv(argv);
3067 }
3068
3069 /*
3070 * User request switch noodle
3071 */
3072
3073 static int
3074 view_driver(struct view *view, enum request request)
3075 {
3076 int i;
3077
3078 if (request == REQ_NONE) {
3079 doupdate();
3080 return TRUE;
3081 }
3082
3083 if (request > REQ_NONE) {
3084 open_run_request(request);
3085 /* FIXME: When all views can refresh always do this. */
3086 if (view == VIEW(REQ_VIEW_STATUS) ||
3087 view == VIEW(REQ_VIEW_MAIN) ||
3088 view == VIEW(REQ_VIEW_LOG) ||
3089 view == VIEW(REQ_VIEW_STAGE))
3090 request = REQ_REFRESH;
3091 else
3092 return TRUE;
3093 }
3094
3095 if (view && view->lines) {
3096 request = view->ops->request(view, request, &view->line[view->lineno]);
3097 if (request == REQ_NONE)
3098 return TRUE;
3099 }
3100
3101 switch (request) {
3102 case REQ_MOVE_UP:
3103 case REQ_MOVE_DOWN:
3104 case REQ_MOVE_PAGE_UP:
3105 case REQ_MOVE_PAGE_DOWN:
3106 case REQ_MOVE_FIRST_LINE:
3107 case REQ_MOVE_LAST_LINE:
3108 move_view(view, request);
3109 break;
3110
3111 case REQ_SCROLL_LEFT:
3112 case REQ_SCROLL_RIGHT:
3113 case REQ_SCROLL_LINE_DOWN:
3114 case REQ_SCROLL_LINE_UP:
3115 case REQ_SCROLL_PAGE_DOWN:
3116 case REQ_SCROLL_PAGE_UP:
3117 scroll_view(view, request);
3118 break;
3119
3120 case REQ_VIEW_BLAME:
3121 if (!opt_file[0]) {
3122 report("No file chosen, press %s to open tree view",
3123 get_key(REQ_VIEW_TREE));
3124 break;
3125 }
3126 open_view(view, request, OPEN_DEFAULT);
3127 break;
3128
3129 case REQ_VIEW_BLOB:
3130 if (!ref_blob[0]) {
3131 report("No file chosen, press %s to open tree view",
3132 get_key(REQ_VIEW_TREE));
3133 break;
3134 }
3135 open_view(view, request, OPEN_DEFAULT);
3136 break;
3137
3138 case REQ_VIEW_PAGER:
3139 if (!VIEW(REQ_VIEW_PAGER)->pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
3140 report("No pager content, press %s to run command from prompt",
3141 get_key(REQ_PROMPT));
3142 break;
3143 }
3144 open_view(view, request, OPEN_DEFAULT);
3145 break;
3146
3147 case REQ_VIEW_STAGE:
3148 if (!VIEW(REQ_VIEW_STAGE)->lines) {
3149 report("No stage content, press %s to open the status view and choose file",
3150 get_key(REQ_VIEW_STATUS));
3151 break;
3152 }
3153 open_view(view, request, OPEN_DEFAULT);
3154 break;
3155
3156 case REQ_VIEW_STATUS:
3157 if (opt_is_inside_work_tree == FALSE) {
3158 report("The status view requires a working tree");
3159 break;
3160 }
3161 open_view(view, request, OPEN_DEFAULT);
3162 break;
3163
3164 case REQ_VIEW_MAIN:
3165 case REQ_VIEW_DIFF:
3166 case REQ_VIEW_LOG:
3167 case REQ_VIEW_TREE:
3168 case REQ_VIEW_HELP:
3169 open_view(view, request, OPEN_DEFAULT);
3170 break;
3171
3172 case REQ_NEXT:
3173 case REQ_PREVIOUS:
3174 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
3175
3176 if ((view == VIEW(REQ_VIEW_DIFF) &&
3177 view->parent == VIEW(REQ_VIEW_MAIN)) ||
3178 (view == VIEW(REQ_VIEW_DIFF) &&
3179 view->parent == VIEW(REQ_VIEW_BLAME)) ||
3180 (view == VIEW(REQ_VIEW_STAGE) &&
3181 view->parent == VIEW(REQ_VIEW_STATUS)) ||
3182 (view == VIEW(REQ_VIEW_BLOB) &&
3183 view->parent == VIEW(REQ_VIEW_TREE))) {
3184 int line;
3185
3186 view = view->parent;
3187 line = view->lineno;
3188 move_view(view, request);
3189 if (view_is_displayed(view))
3190 update_view_title(view);
3191 if (line != view->lineno)
3192 view->ops->request(view, REQ_ENTER,
3193 &view->line[view->lineno]);
3194
3195 } else {
3196 move_view(view, request);
3197 }
3198 break;
3199
3200 case REQ_VIEW_NEXT:
3201 {
3202 int nviews = displayed_views();
3203 int next_view = (current_view + 1) % nviews;
3204
3205 if (next_view == current_view) {
3206 report("Only one view is displayed");
3207 break;
3208 }
3209
3210 current_view = next_view;
3211 /* Blur out the title of the previous view. */
3212 update_view_title(view);
3213 report("");
3214 break;
3215 }
3216 case REQ_REFRESH:
3217 report("Refreshing is not yet supported for the %s view", view->name);
3218 break;
3219
3220 case REQ_MAXIMIZE:
3221 if (displayed_views() == 2)
3222 open_view(view, VIEW_REQ(view), OPEN_DEFAULT);
3223 break;
3224
3225 case REQ_TOGGLE_LINENO:
3226 toggle_view_option(&opt_line_number, "line numbers");
3227 break;
3228
3229 case REQ_TOGGLE_DATE:
3230 toggle_view_option(&opt_date, "date display");
3231 break;
3232
3233 case REQ_TOGGLE_AUTHOR:
3234 toggle_view_option(&opt_author, "author display");
3235 break;
3236
3237 case REQ_TOGGLE_REV_GRAPH:
3238 toggle_view_option(&opt_rev_graph, "revision graph display");
3239 break;
3240
3241 case REQ_TOGGLE_REFS:
3242 toggle_view_option(&opt_show_refs, "reference display");
3243 break;
3244
3245 case REQ_SEARCH:
3246 case REQ_SEARCH_BACK:
3247 search_view(view, request);
3248 break;
3249
3250 case REQ_FIND_NEXT:
3251 case REQ_FIND_PREV:
3252 find_next(view, request);
3253 break;
3254
3255 case REQ_STOP_LOADING:
3256 for (i = 0; i < ARRAY_SIZE(views); i++) {
3257 view = &views[i];
3258 if (view->pipe)
3259 report("Stopped loading the %s view", view->name),
3260 end_update(view, TRUE);
3261 }
3262 break;
3263
3264 case REQ_SHOW_VERSION:
3265 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
3266 return TRUE;
3267
3268 case REQ_SCREEN_REDRAW:
3269 redraw_display(TRUE);
3270 break;
3271
3272 case REQ_EDIT:
3273 report("Nothing to edit");
3274 break;
3275
3276 case REQ_ENTER:
3277 report("Nothing to enter");
3278 break;
3279
3280 case REQ_VIEW_CLOSE:
3281 /* XXX: Mark closed views by letting view->parent point to the
3282 * view itself. Parents to closed view should never be
3283 * followed. */
3284 if (view->parent &&
3285 view->parent->parent != view->parent) {
3286 memset(display, 0, sizeof(display));
3287 current_view = 0;
3288 display[current_view] = view->parent;
3289 view->parent = view;
3290 resize_display();
3291 redraw_display(FALSE);
3292 report("");
3293 break;
3294 }
3295 /* Fall-through */
3296 case REQ_QUIT:
3297 return FALSE;
3298
3299 default:
3300 report("Unknown key, press 'h' for help");
3301 return TRUE;
3302 }
3303
3304 return TRUE;
3305 }
3306
3307
3308 /*
3309 * View backend utilities
3310 */
3311
3312 /* Parse author lines where the name may be empty:
3313 * author <email@address.tld> 1138474660 +0100
3314 */
3315 static void
3316 parse_author_line(char *ident, char *author, size_t authorsize, struct tm *tm)
3317 {
3318 char *nameend = strchr(ident, '<');
3319 char *emailend = strchr(ident, '>');
3320
3321 if (nameend && emailend)
3322 *nameend = *emailend = 0;
3323 ident = chomp_string(ident);
3324 if (!*ident) {
3325 if (nameend)
3326 ident = chomp_string(nameend + 1);
3327 if (!*ident)
3328 ident = "Unknown";
3329 }
3330
3331 string_ncopy_do(author, authorsize, ident, strlen(ident));
3332
3333 /* Parse epoch and timezone */
3334 if (emailend && emailend[1] == ' ') {
3335 char *secs = emailend + 2;
3336 char *zone = strchr(secs, ' ');
3337 time_t time = (time_t) atol(secs);
3338
3339 if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
3340 long tz;
3341
3342 zone++;
3343 tz = ('0' - zone[1]) * 60 * 60 * 10;
3344 tz += ('0' - zone[2]) * 60 * 60;
3345 tz += ('0' - zone[3]) * 60;
3346 tz += ('0' - zone[4]) * 60;
3347
3348 if (zone[0] == '-')
3349 tz = -tz;
3350
3351 time -= tz;
3352 }
3353
3354 gmtime_r(&time, tm);
3355 }
3356 }
3357
3358 static enum input_status
3359 select_commit_parent_handler(void *data, char *buf, int c)
3360 {
3361 size_t parents = *(size_t *) data;
3362 int parent = 0;
3363
3364 if (!isdigit(c))
3365 return INPUT_SKIP;
3366
3367 if (*buf)
3368 parent = atoi(buf) * 10;
3369 parent += c - '0';
3370
3371 if (parent > parents)
3372 return INPUT_SKIP;
3373 return INPUT_OK;
3374 }
3375
3376 static bool
3377 select_commit_parent(const char *id, char rev[SIZEOF_REV])
3378 {
3379 char buf[SIZEOF_STR * 4];
3380 const char *revlist_argv[] = {
3381 "git", "rev-list", "-1", "--parents", id, NULL
3382 };
3383 int parents;
3384
3385 if (!run_io_buf(revlist_argv, buf, sizeof(buf)) ||
3386 !*chomp_string(buf) ||
3387 (parents = (strlen(buf) / 40) - 1) < 0) {
3388 report("Failed to get parent information");
3389 return FALSE;
3390
3391 } else if (parents == 0) {
3392 report("The selected commit has no parents");
3393 return FALSE;
3394 }
3395
3396 if (parents > 1) {
3397 char prompt[SIZEOF_STR];
3398 char *result;
3399
3400 if (!string_format(prompt, "Which parent? [1..%d] ", parents))
3401 return FALSE;
3402 result = prompt_input(prompt, select_commit_parent_handler, &parents);
3403 if (!result)
3404 return FALSE;
3405 parents = atoi(result);
3406 }
3407
3408 string_copy_rev(rev, &buf[41 * parents]);
3409 return TRUE;
3410 }
3411
3412 /*
3413 * Pager backend
3414 */
3415
3416 static bool
3417 pager_draw(struct view *view, struct line *line, unsigned int lineno)
3418 {
3419 char text[SIZEOF_STR];
3420
3421 if (opt_line_number && draw_lineno(view, lineno))
3422 return TRUE;
3423
3424 string_expand(text, sizeof(text), line->data, opt_tab_size);
3425 draw_text(view, line->type, text, TRUE);
3426 return TRUE;
3427 }
3428
3429 static bool
3430 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
3431 {
3432 const char *describe_argv[] = { "git", "describe", commit_id, NULL };
3433 char refbuf[SIZEOF_STR];
3434 char *ref = NULL;
3435
3436 if (run_io_buf(describe_argv, refbuf, sizeof(refbuf)))
3437 ref = chomp_string(refbuf);
3438
3439 if (!ref || !*ref)
3440 return TRUE;
3441
3442 /* This is the only fatal call, since it can "corrupt" the buffer. */
3443 if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
3444 return FALSE;
3445
3446 return TRUE;
3447 }
3448
3449 static void
3450 add_pager_refs(struct view *view, struct line *line)
3451 {
3452 char buf[SIZEOF_STR];
3453 char *commit_id = (char *)line->data + STRING_SIZE("commit ");
3454 struct ref **refs;
3455 size_t bufpos = 0, refpos = 0;
3456 const char *sep = "Refs: ";
3457 bool is_tag = FALSE;
3458
3459 assert(line->type == LINE_COMMIT);
3460
3461 refs = get_refs(commit_id);
3462 if (!refs) {
3463 if (view == VIEW(REQ_VIEW_DIFF))
3464 goto try_add_describe_ref;
3465 return;
3466 }
3467
3468 do {
3469 struct ref *ref = refs[refpos];
3470 const char *fmt = ref->tag ? "%s[%s]" :
3471 ref->remote ? "%s<%s>" : "%s%s";
3472
3473 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
3474 return;
3475 sep = ", ";
3476 if (ref->tag)
3477 is_tag = TRUE;
3478 } while (refs[refpos++]->next);
3479
3480 if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
3481 try_add_describe_ref:
3482 /* Add <tag>-g<commit_id> "fake" reference. */
3483 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
3484 return;
3485 }
3486
3487 if (bufpos == 0)
3488 return;
3489
3490 add_line_text(view, buf, LINE_PP_REFS);
3491 }
3492
3493 static bool
3494 pager_read(struct view *view, char *data)
3495 {
3496 struct line *line;
3497
3498 if (!data)
3499 return TRUE;
3500
3501 line = add_line_text(view, data, get_line_type(data));
3502 if (!line)
3503 return FALSE;
3504
3505 if (line->type == LINE_COMMIT &&
3506 (view == VIEW(REQ_VIEW_DIFF) ||
3507 view == VIEW(REQ_VIEW_LOG)))
3508 add_pager_refs(view, line);
3509
3510 return TRUE;
3511 }
3512
3513 static enum request
3514 pager_request(struct view *view, enum request request, struct line *line)
3515 {
3516 int split = 0;
3517
3518 if (request != REQ_ENTER)
3519 return request;
3520
3521 if (line->type == LINE_COMMIT &&
3522 (view == VIEW(REQ_VIEW_LOG) ||
3523 view == VIEW(REQ_VIEW_PAGER))) {
3524 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
3525 split = 1;
3526 }
3527
3528 /* Always scroll the view even if it was split. That way
3529 * you can use Enter to scroll through the log view and
3530 * split open each commit diff. */
3531 scroll_view(view, REQ_SCROLL_LINE_DOWN);
3532
3533 /* FIXME: A minor workaround. Scrolling the view will call report("")
3534 * but if we are scrolling a non-current view this won't properly
3535 * update the view title. */
3536 if (split)
3537 update_view_title(view);
3538
3539 return REQ_NONE;
3540 }
3541
3542 static bool
3543 pager_grep(struct view *view, struct line *line)
3544 {
3545 regmatch_t pmatch;
3546 char *text = line->data;
3547
3548 if (!*text)
3549 return FALSE;
3550
3551 if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
3552 return FALSE;
3553
3554 return TRUE;
3555 }
3556
3557 static void
3558 pager_select(struct view *view, struct line *line)
3559 {
3560 if (line->type == LINE_COMMIT) {
3561 char *text = (char *)line->data + STRING_SIZE("commit ");
3562
3563 if (view != VIEW(REQ_VIEW_PAGER))
3564 string_copy_rev(view->ref, text);
3565 string_copy_rev(ref_commit, text);
3566 }
3567 }
3568
3569 static struct view_ops pager_ops = {
3570 "line",
3571 NULL,
3572 NULL,
3573 pager_read,
3574 pager_draw,
3575 pager_request,
3576 pager_grep,
3577 pager_select,
3578 };
3579
3580 static const char *log_argv[SIZEOF_ARG] = {
3581 "git", "log", "--no-color", "--cc", "--stat", "-n100", "%(head)", NULL
3582 };
3583
3584 static enum request
3585 log_request(struct view *view, enum request request, struct line *line)
3586 {
3587 switch (request) {
3588 case REQ_REFRESH:
3589 load_refs();
3590 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
3591 return REQ_NONE;
3592 default:
3593 return pager_request(view, request, line);
3594 }
3595 }
3596
3597 static struct view_ops log_ops = {
3598 "line",
3599 log_argv,
3600 NULL,
3601 pager_read,
3602 pager_draw,
3603 log_request,
3604 pager_grep,
3605 pager_select,
3606 };
3607
3608 static const char *diff_argv[SIZEOF_ARG] = {
3609 "git", "show", "--pretty=fuller", "--no-color", "--root",
3610 "--patch-with-stat", "--find-copies-harder", "-C", "%(commit)", NULL
3611 };
3612
3613 static struct view_ops diff_ops = {
3614 "line",
3615 diff_argv,
3616 NULL,
3617 pager_read,
3618 pager_draw,
3619 pager_request,
3620 pager_grep,
3621 pager_select,
3622 };
3623
3624 /*
3625 * Help backend
3626 */
3627
3628 static bool
3629 help_open(struct view *view)
3630 {
3631 char buf[SIZEOF_STR];
3632 size_t bufpos;
3633 int i;
3634
3635 if (view->lines > 0)
3636 return TRUE;
3637
3638 add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3639
3640 for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3641 const char *key;
3642
3643 if (req_info[i].request == REQ_NONE)
3644 continue;
3645
3646 if (!req_info[i].request) {
3647 add_line_text(view, "", LINE_DEFAULT);
3648 add_line_text(view, req_info[i].help, LINE_DEFAULT);
3649 continue;
3650 }
3651
3652 key = get_key(req_info[i].request);
3653 if (!*key)
3654 key = "(no key defined)";
3655
3656 for (bufpos = 0; bufpos <= req_info[i].namelen; bufpos++) {
3657 buf[bufpos] = tolower(req_info[i].name[bufpos]);
3658 if (buf[bufpos] == '_')
3659 buf[bufpos] = '-';
3660 }
3661
3662 add_line_format(view, LINE_DEFAULT, " %-25s %-20s %s",
3663 key, buf, req_info[i].help);
3664 }
3665
3666 if (run_requests) {
3667 add_line_text(view, "", LINE_DEFAULT);
3668 add_line_text(view, "External commands:", LINE_DEFAULT);
3669 }
3670
3671 for (i = 0; i < run_requests; i++) {
3672 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3673 const char *key;
3674 int argc;
3675
3676 if (!req)
3677 continue;
3678
3679 key = get_key_name(req->key);
3680 if (!*key)
3681 key = "(no key defined)";
3682
3683 for (bufpos = 0, argc = 0; req->argv[argc]; argc++)
3684 if (!string_format_from(buf, &bufpos, "%s%s",
3685 argc ? " " : "", req->argv[argc]))
3686 return REQ_NONE;
3687
3688 add_line_format(view, LINE_DEFAULT, " %-10s %-14s `%s`",
3689 keymap_table[req->keymap].name, key, buf);
3690 }
3691
3692 return TRUE;
3693 }
3694
3695 static struct view_ops help_ops = {
3696 "line",
3697 NULL,
3698 help_open,
3699 NULL,
3700 pager_draw,
3701 pager_request,
3702 pager_grep,
3703 pager_select,
3704 };
3705
3706
3707 /*
3708 * Tree backend
3709 */
3710
3711 struct tree_stack_entry {
3712 struct tree_stack_entry *prev; /* Entry below this in the stack */
3713 unsigned long lineno; /* Line number to restore */
3714 char *name; /* Position of name in opt_path */
3715 };
3716
3717 /* The top of the path stack. */
3718 static struct tree_stack_entry *tree_stack = NULL;
3719 unsigned long tree_lineno = 0;
3720
3721 static void
3722 pop_tree_stack_entry(void)
3723 {
3724 struct tree_stack_entry *entry = tree_stack;
3725
3726 tree_lineno = entry->lineno;
3727 entry->name[0] = 0;
3728 tree_stack = entry->prev;
3729 free(entry);
3730 }
3731
3732 static void
3733 push_tree_stack_entry(const char *name, unsigned long lineno)
3734 {
3735 struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3736 size_t pathlen = strlen(opt_path);
3737
3738 if (!entry)
3739 return;
3740
3741 entry->prev = tree_stack;
3742 entry->name = opt_path + pathlen;
3743 tree_stack = entry;
3744
3745 if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3746 pop_tree_stack_entry();
3747 return;
3748 }
3749
3750 /* Move the current line to the first tree entry. */
3751 tree_lineno = 1;
3752 entry->lineno = lineno;
3753 }
3754
3755 /* Parse output from git-ls-tree(1):
3756 *
3757 * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3758 */
3759
3760 #define SIZEOF_TREE_ATTR \
3761 STRING_SIZE("100644 blob f931e1d229c3e185caad4449bf5b66ed72462657\t")
3762
3763 #define SIZEOF_TREE_MODE \
3764 STRING_SIZE("100644 ")
3765
3766 #define TREE_ID_OFFSET \
3767 STRING_SIZE("100644 blob ")
3768
3769 struct tree_entry {
3770 char id[SIZEOF_REV];
3771 mode_t mode;
3772 struct tm time; /* Date from the author ident. */
3773 char author[75]; /* Author of the commit. */
3774 char name[1];
3775 };
3776
3777 static const char *
3778 tree_path(struct line *line)
3779 {
3780 return ((struct tree_entry *) line->data)->name;
3781 }
3782
3783
3784 static int
3785 tree_compare_entry(struct line *line1, struct line *line2)
3786 {
3787 if (line1->type != line2->type)
3788 return line1->type == LINE_TREE_DIR ? -1 : 1;
3789 return strcmp(tree_path(line1), tree_path(line2));
3790 }
3791
3792 static struct line *
3793 tree_entry(struct view *view, enum line_type type, const char *path,
3794 const char *mode, const char *id)
3795 {
3796 struct tree_entry *entry = calloc(1, sizeof(*entry) + strlen(path));
3797 struct line *line = entry ? add_line_data(view, entry, type) : NULL;
3798
3799 if (!entry || !line) {
3800 free(entry);
3801 return NULL;
3802 }
3803
3804 strncpy(entry->name, path, strlen(path));
3805 if (mode)
3806 entry->mode = strtoul(mode, NULL, 8);
3807 if (id)
3808 string_copy_rev(entry->id, id);
3809
3810 return line;
3811 }
3812
3813 static bool
3814 tree_read_date(struct view *view, char *text, bool *read_date)
3815 {
3816 static char author_name[SIZEOF_STR];
3817 static struct tm author_time;
3818
3819 if (!text && *read_date) {
3820 *read_date = FALSE;
3821 return TRUE;
3822
3823 } else if (!text) {
3824 char *path = *opt_path ? opt_path : ".";
3825 /* Find next entry to process */
3826 const char *log_file[] = {
3827 "git", "log", "--no-color", "--pretty=raw",
3828 "--cc", "--raw", view->id, "--", path, NULL
3829 };
3830 struct io io = {};
3831
3832 if (!view->lines) {
3833 tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL);
3834 report("Tree is empty");
3835 return TRUE;
3836 }
3837
3838 if (!run_io_rd(&io, log_file, FORMAT_NONE)) {
3839 report("Failed to load tree data");
3840 return TRUE;
3841 }
3842
3843 done_io(view->pipe);
3844 view->io = io;
3845 *read_date = TRUE;
3846 return FALSE;
3847
3848 } else if (*text == 'a' && get_line_type(text) == LINE_AUTHOR) {
3849 parse_author_line(text + STRING_SIZE("author "),
3850 author_name, sizeof(author_name), &author_time);
3851
3852 } else if (*text == ':') {
3853 char *pos;
3854 size_t annotated = 1;
3855 size_t i;
3856
3857 pos = strchr(text, '\t');
3858 if (!pos)
3859 return TRUE;
3860 text = pos + 1;
3861 if (*opt_prefix && !strncmp(text, opt_prefix, strlen(opt_prefix)))
3862 text += strlen(opt_prefix);
3863 if (*opt_path && !strncmp(text, opt_path, strlen(opt_path)))
3864 text += strlen(opt_path);
3865 pos = strchr(text, '/');
3866 if (pos)
3867 *pos = 0;
3868
3869 for (i = 1; i < view->lines; i++) {
3870 struct line *line = &view->line[i];
3871 struct tree_entry *entry = line->data;
3872
3873 annotated += !!*entry->author;
3874 if (*entry->author || strcmp(entry->name, text))
3875 continue;
3876
3877 string_copy(entry->author, author_name);
3878 memcpy(&entry->time, &author_time, sizeof(entry->time));
3879 line->dirty = 1;
3880 break;
3881 }
3882
3883 if (annotated == view->lines)
3884 kill_io(view->pipe);
3885 }
3886 return TRUE;
3887 }
3888
3889 static bool
3890 tree_read(struct view *view, char *text)
3891 {
3892 static bool read_date = FALSE;
3893 struct tree_entry *data;
3894 struct line *entry, *line;
3895 enum line_type type;
3896 size_t textlen = text ? strlen(text) : 0;
3897 char *path = text + SIZEOF_TREE_ATTR;
3898
3899 if (read_date || !text)
3900 return tree_read_date(view, text, &read_date);
3901
3902 if (textlen <= SIZEOF_TREE_ATTR)
3903 return FALSE;
3904 if (view->lines == 0 &&
3905 !tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL))
3906 return FALSE;
3907
3908 /* Strip the path part ... */
3909 if (*opt_path) {
3910 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3911 size_t striplen = strlen(opt_path);
3912
3913 if (pathlen > striplen)
3914 memmove(path, path + striplen,
3915 pathlen - striplen + 1);
3916
3917 /* Insert "link" to parent directory. */
3918 if (view->lines == 1 &&
3919 !tree_entry(view, LINE_TREE_DIR, "..", "040000", view->ref))
3920 return FALSE;
3921 }
3922
3923 type = text[SIZEOF_TREE_MODE] == 't' ? LINE_TREE_DIR : LINE_TREE_FILE;
3924 entry = tree_entry(view, type, path, text, text + TREE_ID_OFFSET);
3925 if (!entry)
3926 return FALSE;
3927 data = entry->data;
3928
3929 /* Skip "Directory ..." and ".." line. */
3930 for (line = &view->line[1 + !!*opt_path]; line < entry; line++) {
3931 if (tree_compare_entry(line, entry) <= 0)
3932 continue;
3933
3934 memmove(line + 1, line, (entry - line) * sizeof(*entry));
3935
3936 line->data = data;
3937 line->type = type;
3938 for (; line <= entry; line++)
3939 line->dirty = line->cleareol = 1;
3940 return TRUE;
3941 }
3942
3943 if (tree_lineno > view->lineno) {
3944 view->lineno = tree_lineno;
3945 tree_lineno = 0;
3946 }
3947
3948 return TRUE;
3949 }
3950
3951 static bool
3952 tree_draw(struct view *view, struct line *line, unsigned int lineno)
3953 {
3954 struct tree_entry *entry = line->data;
3955
3956 if (line->type == LINE_TREE_HEAD) {
3957 if (draw_text(view, line->type, "Directory path /", TRUE))
3958 return TRUE;
3959 } else {
3960 if (draw_mode(view, entry->mode))
3961 return TRUE;
3962
3963 if (opt_author && draw_author(view, entry->author))
3964 return TRUE;
3965
3966 if (opt_date && draw_date(view, *entry->author ? &entry->time : NULL))
3967 return TRUE;
3968 }
3969 if (draw_text(view, line->type, entry->name, TRUE))
3970 return TRUE;
3971 return TRUE;
3972 }
3973
3974 static void
3975 open_blob_editor()
3976 {
3977 char file[SIZEOF_STR] = "/tmp/tigblob.XXXXXX";
3978 int fd = mkstemp(file);
3979
3980 if (fd == -1)
3981 report("Failed to create temporary file");
3982 else if (!run_io_append(blob_ops.argv, FORMAT_ALL, fd))
3983 report("Failed to save blob data to file");
3984 else
3985 open_editor(FALSE, file);
3986 if (fd != -1)
3987 unlink(file);
3988 }
3989
3990 static enum request
3991 tree_request(struct view *view, enum request request, struct line *line)
3992 {
3993 enum open_flags flags;
3994
3995 switch (request) {
3996 case REQ_VIEW_BLAME:
3997 if (line->type != LINE_TREE_FILE) {
3998 report("Blame only supported for files");
3999 return REQ_NONE;
4000 }
4001
4002 string_copy(opt_ref, view->vid);
4003 return request;
4004
4005 case REQ_EDIT:
4006 if (line->type != LINE_TREE_FILE) {
4007 report("Edit only supported for files");
4008 } else if (!is_head_commit(view->vid)) {
4009 open_blob_editor();
4010 } else {
4011 open_editor(TRUE, opt_file);
4012 }
4013 return REQ_NONE;
4014
4015 case REQ_PARENT:
4016 if (!*opt_path) {
4017 /* quit view if at top of tree */
4018 return REQ_VIEW_CLOSE;
4019 }
4020 /* fake 'cd ..' */
4021 line = &view->line[1];
4022 break;
4023
4024 case REQ_ENTER:
4025 break;
4026
4027 default:
4028 return request;
4029 }
4030
4031 /* Cleanup the stack if the tree view is at a different tree. */
4032 while (!*opt_path && tree_stack)
4033 pop_tree_stack_entry();
4034
4035 switch (line->type) {
4036 case LINE_TREE_DIR:
4037 /* Depending on whether it is a subdirectory or parent link
4038 * mangle the path buffer. */
4039 if (line == &view->line[1] && *opt_path) {
4040 pop_tree_stack_entry();
4041
4042 } else {
4043 const char *basename = tree_path(line);
4044
4045 push_tree_stack_entry(basename, view->lineno);
4046 }
4047
4048 /* Trees and subtrees share the same ID, so they are not not
4049 * unique like blobs. */
4050 flags = OPEN_RELOAD;
4051 request = REQ_VIEW_TREE;
4052 break;
4053
4054 case LINE_TREE_FILE:
4055 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4056 request = REQ_VIEW_BLOB;
4057 break;
4058
4059 default:
4060 return REQ_NONE;
4061 }
4062
4063 open_view(view, request, flags);
4064 if (request == REQ_VIEW_TREE)
4065 view->lineno = tree_lineno;
4066
4067 return REQ_NONE;
4068 }
4069
4070 static void
4071 tree_select(struct view *view, struct line *line)
4072 {
4073 struct tree_entry *entry = line->data;
4074
4075 if (line->type == LINE_TREE_FILE) {
4076 string_copy_rev(ref_blob, entry->id);
4077 string_format(opt_file, "%s%s", opt_path, tree_path(line));
4078
4079 } else if (line->type != LINE_TREE_DIR) {
4080 return;
4081 }
4082
4083 string_copy_rev(view->ref, entry->id);
4084 }
4085
4086 static const char *tree_argv[SIZEOF_ARG] = {
4087 "git", "ls-tree", "%(commit)", "%(directory)", NULL
4088 };
4089
4090 static struct view_ops tree_ops = {
4091 "file",
4092 tree_argv,
4093 NULL,
4094 tree_read,
4095 tree_draw,
4096 tree_request,
4097 pager_grep,
4098 tree_select,
4099 };
4100
4101 static bool
4102 blob_read(struct view *view, char *line)
4103 {
4104 if (!line)
4105 return TRUE;
4106 return add_line_text(view, line, LINE_DEFAULT) != NULL;
4107 }
4108
4109 static enum request
4110 blob_request(struct view *view, enum request request, struct line *line)
4111 {
4112 switch (request) {
4113 case REQ_EDIT:
4114 open_blob_editor();
4115 return REQ_NONE;
4116 default:
4117 return pager_request(view, request, line);
4118 }
4119 }
4120
4121 static const char *blob_argv[SIZEOF_ARG] = {
4122 "git", "cat-file", "blob", "%(blob)", NULL
4123 };
4124
4125 static struct view_ops blob_ops = {
4126 "line",
4127 blob_argv,
4128 NULL,
4129 blob_read,
4130 pager_draw,
4131 blob_request,
4132 pager_grep,
4133 pager_select,
4134 };
4135
4136 /*
4137 * Blame backend
4138 *
4139 * Loading the blame view is a two phase job:
4140 *
4141 * 1. File content is read either using opt_file from the
4142 * filesystem or using git-cat-file.
4143 * 2. Then blame information is incrementally added by
4144 * reading output from git-blame.
4145 */
4146
4147 static const char *blame_head_argv[] = {
4148 "git", "blame", "--incremental", "--", "%(file)", NULL
4149 };
4150
4151 static const char *blame_ref_argv[] = {
4152 "git", "blame", "--incremental", "%(ref)", "--", "%(file)", NULL
4153 };
4154
4155 static const char *blame_cat_file_argv[] = {
4156 "git", "cat-file", "blob", "%(ref):%(file)", NULL
4157 };
4158
4159 struct blame_commit {
4160 char id[SIZEOF_REV]; /* SHA1 ID. */
4161 char title[128]; /* First line of the commit message. */
4162 char author[75]; /* Author of the commit. */
4163 struct tm time; /* Date from the author ident. */
4164 char filename[128]; /* Name of file. */
4165 bool has_previous; /* Was a "previous" line detected. */
4166 };
4167
4168 struct blame {
4169 struct blame_commit *commit;
4170 char text[1];
4171 };
4172
4173 static bool
4174 blame_open(struct view *view)
4175 {
4176 if (*opt_ref || !io_open(&view->io, opt_file)) {
4177 if (!run_io_rd(&view->io, blame_cat_file_argv, FORMAT_ALL))
4178 return FALSE;
4179 }
4180
4181 setup_update(view, opt_file);
4182 string_format(view->ref, "%s ...", opt_file);
4183
4184 return TRUE;
4185 }
4186
4187 static struct blame_commit *
4188 get_blame_commit(struct view *view, const char *id)
4189 {
4190 size_t i;
4191
4192 for (i = 0; i < view->lines; i++) {
4193 struct blame *blame = view->line[i].data;
4194
4195 if (!blame->commit)
4196 continue;
4197
4198 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
4199 return blame->commit;
4200 }
4201
4202 {
4203 struct blame_commit *commit = calloc(1, sizeof(*commit));
4204
4205 if (commit)
4206 string_ncopy(commit->id, id, SIZEOF_REV);
4207 return commit;
4208 }
4209 }
4210
4211 static bool
4212 parse_number(const char **posref, size_t *number, size_t min, size_t max)
4213 {
4214 const char *pos = *posref;
4215
4216 *posref = NULL;
4217 pos = strchr(pos + 1, ' ');
4218 if (!pos || !isdigit(pos[1]))
4219 return FALSE;
4220 *number = atoi(pos + 1);
4221 if (*number < min || *number > max)
4222 return FALSE;
4223
4224 *posref = pos;
4225 return TRUE;
4226 }
4227
4228 static struct blame_commit *
4229 parse_blame_commit(struct view *view, const char *text, int *blamed)
4230 {
4231 struct blame_commit *commit;
4232 struct blame *blame;
4233 const char *pos = text + SIZEOF_REV - 1;
4234 size_t lineno;
4235 size_t group;
4236
4237 if (strlen(text) <= SIZEOF_REV || *pos != ' ')
4238 return NULL;
4239
4240 if (!parse_number(&pos, &lineno, 1, view->lines) ||
4241 !parse_number(&pos, &group, 1, view->lines - lineno + 1))
4242 return NULL;
4243
4244 commit = get_blame_commit(view, text);
4245 if (!commit)
4246 return NULL;
4247
4248 *blamed += group;
4249 while (group--) {
4250 struct line *line = &view->line[lineno + group - 1];
4251
4252 blame = line->data;
4253 blame->commit = commit;
4254 line->dirty = 1;
4255 }
4256
4257 return commit;
4258 }
4259
4260 static bool
4261 blame_read_file(struct view *view, const char *line, bool *read_file)
4262 {
4263 if (!line) {
4264 const char **argv = *opt_ref ? blame_ref_argv : blame_head_argv;
4265 struct io io = {};
4266
4267 if (view->lines == 0 && !view->parent)
4268 die("No blame exist for %s", view->vid);
4269
4270 if (view->lines == 0 || !run_io_rd(&io, argv, FORMAT_ALL)) {
4271 report("Failed to load blame data");
4272 return TRUE;
4273 }
4274
4275 done_io(view->pipe);
4276 view->io = io;
4277 *read_file = FALSE;
4278 return FALSE;
4279
4280 } else {
4281 size_t linelen = string_expand_length(line, opt_tab_size);
4282 struct blame *blame = malloc(sizeof(*blame) + linelen);
4283
4284 if (!blame)
4285 return FALSE;
4286
4287 blame->commit = NULL;
4288 string_expand(blame->text, linelen + 1, line, opt_tab_size);
4289 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
4290 }
4291 }
4292
4293 static bool
4294 match_blame_header(const char *name, char **line)
4295 {
4296 size_t namelen = strlen(name);
4297 bool matched = !strncmp(name, *line, namelen);
4298
4299 if (matched)
4300 *line += namelen;
4301
4302 return matched;
4303 }
4304
4305 static bool
4306 blame_read(struct view *view, char *line)
4307 {
4308 static struct blame_commit *commit = NULL;
4309 static int blamed = 0;
4310 static time_t author_time;
4311 static bool read_file = TRUE;
4312
4313 if (read_file)
4314 return blame_read_file(view, line, &read_file);
4315
4316 if (!line) {
4317 /* Reset all! */
4318 commit = NULL;
4319 blamed = 0;
4320 read_file = TRUE;
4321 string_format(view->ref, "%s", view->vid);
4322 if (view_is_displayed(view)) {
4323 update_view_title(view);
4324 redraw_view_from(view, 0);
4325 }
4326 return TRUE;
4327 }
4328
4329 if (!commit) {
4330 commit = parse_blame_commit(view, line, &blamed);
4331 string_format(view->ref, "%s %2d%%", view->vid,
4332 view->lines ? blamed * 100 / view->lines : 0);
4333
4334 } else if (match_blame_header("author ", &line)) {
4335 string_ncopy(commit->author, line, strlen(line));
4336
4337 } else if (match_blame_header("author-time ", &line)) {
4338 author_time = (time_t) atol(line);
4339
4340 } else if (match_blame_header("author-tz ", &line)) {
4341 long tz;
4342
4343 tz = ('0' - line[1]) * 60 * 60 * 10;
4344 tz += ('0' - line[2]) * 60 * 60;
4345 tz += ('0' - line[3]) * 60;
4346 tz += ('0' - line[4]) * 60;
4347
4348 if (line[0] == '-')
4349 tz = -tz;
4350
4351 author_time -= tz;
4352 gmtime_r(&author_time, &commit->time);
4353
4354 } else if (match_blame_header("summary ", &line)) {
4355 string_ncopy(commit->title, line, strlen(line));
4356
4357 } else if (match_blame_header("previous ", &line)) {
4358 commit->has_previous = TRUE;
4359
4360 } else if (match_blame_header("filename ", &line)) {
4361 string_ncopy(commit->filename, line, strlen(line));
4362 commit = NULL;
4363 }
4364
4365 return TRUE;
4366 }
4367
4368 static bool
4369 blame_draw(struct view *view, struct line *line, unsigned int lineno)
4370 {
4371 struct blame *blame = line->data;
4372 struct tm *time = NULL;
4373 const char *id = NULL, *author = NULL;
4374
4375 if (blame->commit && *blame->commit->filename) {
4376 id = blame->commit->id;
4377 author = blame->commit->author;
4378 time = &blame->commit->time;
4379 }
4380
4381 if (opt_date && draw_date(view, time))
4382 return TRUE;
4383
4384 if (opt_author && draw_author(view, author))
4385 return TRUE;
4386
4387 if (draw_field(view, LINE_BLAME_ID, id, ID_COLS, FALSE))
4388 return TRUE;
4389
4390 if (draw_lineno(view, lineno))
4391 return TRUE;
4392
4393 draw_text(view, LINE_DEFAULT, blame->text, TRUE);
4394 return TRUE;
4395 }
4396
4397 static bool
4398 check_blame_commit(struct blame *blame)
4399 {
4400 if (!blame->commit)
4401 report("Commit data not loaded yet");
4402 else if (!strcmp(blame->commit->id, NULL_ID))
4403 report("No commit exist for the selected line");
4404 else
4405 return TRUE;
4406 return FALSE;
4407 }
4408
4409 static enum request
4410 blame_request(struct view *view, enum request request, struct line *line)
4411 {
4412 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4413 struct blame *blame = line->data;
4414
4415 switch (request) {
4416 case REQ_VIEW_BLAME:
4417 if (check_blame_commit(blame)) {
4418 string_copy(opt_ref, blame->commit->id);
4419 open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4420 }
4421 break;
4422
4423 case REQ_PARENT:
4424 if (check_blame_commit(blame) &&
4425 select_commit_parent(blame->commit->id, opt_ref))
4426 open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4427 break;
4428
4429 case REQ_ENTER:
4430 if (!blame->commit) {
4431 report("No commit loaded yet");
4432 break;
4433 }
4434
4435 if (view_is_displayed(VIEW(REQ_VIEW_DIFF)) &&
4436 !strcmp(blame->commit->id, VIEW(REQ_VIEW_DIFF)->ref))
4437 break;
4438
4439 if (!strcmp(blame->commit->id, NULL_ID)) {
4440 struct view *diff = VIEW(REQ_VIEW_DIFF);
4441 const char *diff_index_argv[] = {
4442 "git", "diff-index", "--root", "--patch-with-stat",
4443 "-C", "-M", "HEAD", "--", view->vid, NULL
4444 };
4445
4446 if (!blame->commit->has_previous) {
4447 diff_index_argv[1] = "diff";
4448 diff_index_argv[2] = "--no-color";
4449 diff_index_argv[6] = "--";
4450 diff_index_argv[7] = "/dev/null";
4451 }
4452
4453 if (!prepare_update(diff, diff_index_argv, NULL, FORMAT_DASH)) {
4454 report("Failed to allocate diff command");
4455 break;
4456 }
4457 flags |= OPEN_PREPARED;
4458 }
4459
4460 open_view(view, REQ_VIEW_DIFF, flags);
4461 if (VIEW(REQ_VIEW_DIFF)->pipe && !strcmp(blame->commit->id, NULL_ID))
4462 string_copy_rev(VIEW(REQ_VIEW_DIFF)->ref, NULL_ID);
4463 break;
4464
4465 default:
4466 return request;
4467 }
4468
4469 return REQ_NONE;
4470 }
4471
4472 static bool
4473 blame_grep(struct view *view, struct line *line)
4474 {
4475 struct blame *blame = line->data;
4476 struct blame_commit *commit = blame->commit;
4477 regmatch_t pmatch;
4478
4479 #define MATCH(text, on) \
4480 (on && *text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4481
4482 if (commit) {
4483 char buf[DATE_COLS + 1];
4484
4485 if (MATCH(commit->title, 1) ||
4486 MATCH(commit->author, opt_author) ||
4487 MATCH(commit->id, opt_date))
4488 return TRUE;
4489
4490 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
4491 MATCH(buf, 1))
4492 return TRUE;
4493 }
4494
4495 return MATCH(blame->text, 1);
4496
4497 #undef MATCH
4498 }
4499
4500 static void
4501 blame_select(struct view *view, struct line *line)
4502 {
4503 struct blame *blame = line->data;
4504 struct blame_commit *commit = blame->commit;
4505
4506 if (!commit)
4507 return;
4508
4509 if (!strcmp(commit->id, NULL_ID))
4510 string_ncopy(ref_commit, "HEAD", 4);
4511 else
4512 string_copy_rev(ref_commit, commit->id);
4513 }
4514
4515 static struct view_ops blame_ops = {
4516 "line",
4517 NULL,
4518 blame_open,
4519 blame_read,
4520 blame_draw,
4521 blame_request,
4522 blame_grep,
4523 blame_select,
4524 };
4525
4526 /*
4527 * Status backend
4528 */
4529
4530 struct status {
4531 char status;
4532 struct {
4533 mode_t mode;
4534 char rev[SIZEOF_REV];
4535 char name[SIZEOF_STR];
4536 } old;
4537 struct {
4538 mode_t mode;
4539 char rev[SIZEOF_REV];
4540 char name[SIZEOF_STR];
4541 } new;
4542 };
4543
4544 static char status_onbranch[SIZEOF_STR];
4545 static struct status stage_status;
4546 static enum line_type stage_line_type;
4547 static size_t stage_chunks;
4548 static int *stage_chunk;
4549
4550 /* This should work even for the "On branch" line. */
4551 static inline bool
4552 status_has_none(struct view *view, struct line *line)
4553 {
4554 return line < view->line + view->lines && !line[1].data;
4555 }
4556
4557 /* Get fields from the diff line:
4558 * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
4559 */
4560 static inline bool
4561 status_get_diff(struct status *file, const char *buf, size_t bufsize)
4562 {
4563 const char *old_mode = buf + 1;
4564 const char *new_mode = buf + 8;
4565 const char *old_rev = buf + 15;
4566 const char *new_rev = buf + 56;
4567 const char *status = buf + 97;
4568
4569 if (bufsize < 98 ||
4570 old_mode[-1] != ':' ||
4571 new_mode[-1] != ' ' ||
4572 old_rev[-1] != ' ' ||
4573 new_rev[-1] != ' ' ||
4574 status[-1] != ' ')
4575 return FALSE;
4576
4577 file->status = *status;
4578
4579 string_copy_rev(file->old.rev, old_rev);
4580 string_copy_rev(file->new.rev, new_rev);
4581
4582 file->old.mode = strtoul(old_mode, NULL, 8);
4583 file->new.mode = strtoul(new_mode, NULL, 8);
4584
4585 file->old.name[0] = file->new.name[0] = 0;
4586
4587 return TRUE;
4588 }
4589
4590 static bool
4591 status_run(struct view *view, const char *argv[], char status, enum line_type type)
4592 {
4593 struct status *unmerged = NULL;
4594 char *buf;
4595 struct io io = {};
4596
4597 if (!run_io(&io, argv, NULL, IO_RD))
4598 return FALSE;
4599
4600 add_line_data(view, NULL, type);
4601
4602 while ((buf = io_get(&io, 0, TRUE))) {
4603 struct status *file = unmerged;
4604
4605 if (!file) {
4606 file = calloc(1, sizeof(*file));
4607 if (!file || !add_line_data(view, file, type))
4608 goto error_out;
4609 }
4610
4611 /* Parse diff info part. */
4612 if (status) {
4613 file->status = status;
4614 if (status == 'A')
4615 string_copy(file->old.rev, NULL_ID);
4616
4617 } else if (!file->status || file == unmerged) {
4618 if (!status_get_diff(file, buf, strlen(buf)))
4619 goto error_out;
4620
4621 buf = io_get(&io, 0, TRUE);
4622 if (!buf)
4623 break;
4624
4625 /* Collapse all modified entries that follow an
4626 * associated unmerged entry. */
4627 if (unmerged == file) {
4628 unmerged->status = 'U';
4629 unmerged = NULL;
4630 } else if (file->status == 'U') {
4631 unmerged = file;
4632 }
4633 }
4634
4635 /* Grab the old name for rename/copy. */
4636 if (!*file->old.name &&
4637 (file->status == 'R' || file->status == 'C')) {
4638 string_ncopy(file->old.name, buf, strlen(buf));
4639
4640 buf = io_get(&io, 0, TRUE);
4641 if (!buf)
4642 break;
4643 }
4644
4645 /* git-ls-files just delivers a NUL separated list of
4646 * file names similar to the second half of the
4647 * git-diff-* output. */
4648 string_ncopy(file->new.name, buf, strlen(buf));
4649 if (!*file->old.name)
4650 string_copy(file->old.name, file->new.name);
4651 file = NULL;
4652 }
4653
4654 if (io_error(&io)) {
4655 error_out:
4656 done_io(&io);
4657 return FALSE;
4658 }
4659
4660 if (!view->line[view->lines - 1].data)
4661 add_line_data(view, NULL, LINE_STAT_NONE);
4662
4663 done_io(&io);
4664 return TRUE;
4665 }
4666
4667 /* Don't show unmerged entries in the staged section. */
4668 static const char *status_diff_index_argv[] = {
4669 "git", "diff-index", "-z", "--diff-filter=ACDMRTXB",
4670 "--cached", "-M", "HEAD", NULL
4671 };
4672
4673 static const char *status_diff_files_argv[] = {
4674 "git", "diff-files", "-z", NULL
4675 };
4676
4677 static const char *status_list_other_argv[] = {
4678 "git", "ls-files", "-z", "--others", "--exclude-standard", NULL
4679 };
4680
4681 static const char *status_list_no_head_argv[] = {
4682 "git", "ls-files", "-z", "--cached", "--exclude-standard", NULL
4683 };
4684
4685 static const char *update_index_argv[] = {
4686 "git", "update-index", "-q", "--unmerged", "--refresh", NULL
4687 };
4688
4689 /* Restore the previous line number to stay in the context or select a
4690 * line with something that can be updated. */
4691 static void
4692 status_restore(struct view *view)
4693 {
4694 if (view->p_lineno >= view->lines)
4695 view->p_lineno = view->lines - 1;
4696 while (view->p_lineno < view->lines && !view->line[view->p_lineno].data)
4697 view->p_lineno++;
4698 while (view->p_lineno > 0 && !view->line[view->p_lineno].data)
4699 view->p_lineno--;
4700
4701 /* If the above fails, always skip the "On branch" line. */
4702 if (view->p_lineno < view->lines)
4703 view->lineno = view->p_lineno;
4704 else
4705 view->lineno = 1;
4706
4707 if (view->lineno < view->offset)
4708 view->offset = view->lineno;
4709 else if (view->offset + view->height <= view->lineno)
4710 view->offset = view->lineno - view->height + 1;
4711
4712 view->p_restore = FALSE;
4713 }
4714
4715 /* First parse staged info using git-diff-index(1), then parse unstaged
4716 * info using git-diff-files(1), and finally untracked files using
4717 * git-ls-files(1). */
4718 static bool
4719 status_open(struct view *view)
4720 {
4721 reset_view(view);
4722
4723 add_line_data(view, NULL, LINE_STAT_HEAD);
4724 if (is_initial_commit())
4725 string_copy(status_onbranch, "Initial commit");
4726 else if (!*opt_head)
4727 string_copy(status_onbranch, "Not currently on any branch");
4728 else if (!string_format(status_onbranch, "On branch %s", opt_head))
4729 return FALSE;
4730
4731 run_io_bg(update_index_argv);
4732
4733 if (is_initial_commit()) {
4734 if (!status_run(view, status_list_no_head_argv, 'A', LINE_STAT_STAGED))
4735 return FALSE;
4736 } else if (!status_run(view, status_diff_index_argv, 0, LINE_STAT_STAGED)) {
4737 return FALSE;
4738 }
4739
4740 if (!status_run(view, status_diff_files_argv, 0, LINE_STAT_UNSTAGED) ||
4741 !status_run(view, status_list_other_argv, '?', LINE_STAT_UNTRACKED))
4742 return FALSE;
4743
4744 /* Restore the exact position or use the specialized restore
4745 * mode? */
4746 if (!view->p_restore)
4747 status_restore(view);
4748 return TRUE;
4749 }
4750
4751 static bool
4752 status_draw(struct view *view, struct line *line, unsigned int lineno)
4753 {
4754 struct status *status = line->data;
4755 enum line_type type;
4756 const char *text;
4757
4758 if (!status) {
4759 switch (line->type) {
4760 case LINE_STAT_STAGED:
4761 type = LINE_STAT_SECTION;
4762 text = "Changes to be committed:";
4763 break;
4764
4765 case LINE_STAT_UNSTAGED:
4766 type = LINE_STAT_SECTION;
4767 text = "Changed but not updated:";
4768 break;
4769
4770 case LINE_STAT_UNTRACKED:
4771 type = LINE_STAT_SECTION;
4772 text = "Untracked files:";
4773 break;
4774
4775 case LINE_STAT_NONE:
4776 type = LINE_DEFAULT;
4777 text = " (no files)";
4778 break;
4779
4780 case LINE_STAT_HEAD:
4781 type = LINE_STAT_HEAD;
4782 text = status_onbranch;
4783 break;
4784
4785 default:
4786 return FALSE;
4787 }
4788 } else {
4789 static char buf[] = { '?', ' ', ' ', ' ', 0 };
4790
4791 buf[0] = status->status;
4792 if (draw_text(view, line->type, buf, TRUE))
4793 return TRUE;
4794 type = LINE_DEFAULT;
4795 text = status->new.name;
4796 }
4797
4798 draw_text(view, type, text, TRUE);
4799 return TRUE;
4800 }
4801
4802 static enum request
4803 status_enter(struct view *view, struct line *line)
4804 {
4805 struct status *status = line->data;
4806 const char *oldpath = status ? status->old.name : NULL;
4807 /* Diffs for unmerged entries are empty when passing the new
4808 * path, so leave it empty. */
4809 const char *newpath = status && status->status != 'U' ? status->new.name : NULL;
4810 const char *info;
4811 enum open_flags split;
4812 struct view *stage = VIEW(REQ_VIEW_STAGE);
4813
4814 if (line->type == LINE_STAT_NONE ||
4815 (!status && line[1].type == LINE_STAT_NONE)) {
4816 report("No file to diff");
4817 return REQ_NONE;
4818 }
4819
4820 switch (line->type) {
4821 case LINE_STAT_STAGED:
4822 if (is_initial_commit()) {
4823 const char *no_head_diff_argv[] = {
4824 "git", "diff", "--no-color", "--patch-with-stat",
4825 "--", "/dev/null", newpath, NULL
4826 };
4827
4828 if (!prepare_update(stage, no_head_diff_argv, opt_cdup, FORMAT_DASH))
4829 return REQ_QUIT;
4830 } else {
4831 const char *index_show_argv[] = {
4832 "git", "diff-index", "--root", "--patch-with-stat",
4833 "-C", "-M", "--cached", "HEAD", "--",
4834 oldpath, newpath, NULL
4835 };
4836
4837 if (!prepare_update(stage, index_show_argv, opt_cdup, FORMAT_DASH))
4838 return REQ_QUIT;
4839 }
4840
4841 if (status)
4842 info = "Staged changes to %s";
4843 else
4844 info = "Staged changes";
4845 break;
4846
4847 case LINE_STAT_UNSTAGED:
4848 {
4849 const char *files_show_argv[] = {
4850 "git", "diff-files", "--root", "--patch-with-stat",
4851 "-C", "-M", "--", oldpath, newpath, NULL
4852 };
4853
4854 if (!prepare_update(stage, files_show_argv, opt_cdup, FORMAT_DASH))
4855 return REQ_QUIT;
4856 if (status)
4857 info = "Unstaged changes to %s";
4858 else
4859 info = "Unstaged changes";
4860 break;
4861 }
4862 case LINE_STAT_UNTRACKED:
4863 if (!newpath) {
4864 report("No file to show");
4865 return REQ_NONE;
4866 }
4867
4868 if (!suffixcmp(status->new.name, -1, "/")) {
4869 report("Cannot display a directory");
4870 return REQ_NONE;
4871 }
4872
4873 if (!prepare_update_file(stage, newpath))
4874 return REQ_QUIT;
4875 info = "Untracked file %s";
4876 break;
4877
4878 case LINE_STAT_HEAD:
4879 return REQ_NONE;
4880
4881 default:
4882 die("line type %d not handled in switch", line->type);
4883 }
4884
4885 split = view_is_displayed(view) ? OPEN_SPLIT : 0;
4886 open_view(view, REQ_VIEW_STAGE, OPEN_PREPARED | split);
4887 if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
4888 if (status) {
4889 stage_status = *status;
4890 } else {
4891 memset(&stage_status, 0, sizeof(stage_status));
4892 }
4893
4894 stage_line_type = line->type;
4895 stage_chunks = 0;
4896 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
4897 }
4898
4899 return REQ_NONE;
4900 }
4901
4902 static bool
4903 status_exists(struct status *status, enum line_type type)
4904 {
4905 struct view *view = VIEW(REQ_VIEW_STATUS);
4906 unsigned long lineno;
4907
4908 for (lineno = 0; lineno < view->lines; lineno++) {
4909 struct line *line = &view->line[lineno];
4910 struct status *pos = line->data;
4911
4912 if (line->type != type)
4913 continue;
4914 if (!pos && (!status || !status->status) && line[1].data) {
4915 select_view_line(view, lineno);
4916 return TRUE;
4917 }
4918 if (pos && !strcmp(status->new.name, pos->new.name)) {
4919 select_view_line(view, lineno);
4920 return TRUE;
4921 }
4922 }
4923
4924 return FALSE;
4925 }
4926
4927
4928 static bool
4929 status_update_prepare(struct io *io, enum line_type type)
4930 {
4931 const char *staged_argv[] = {
4932 "git", "update-index", "-z", "--index-info", NULL
4933 };
4934 const char *others_argv[] = {
4935 "git", "update-index", "-z", "--add", "--remove", "--stdin", NULL
4936 };
4937
4938 switch (type) {
4939 case LINE_STAT_STAGED:
4940 return run_io(io, staged_argv, opt_cdup, IO_WR);
4941
4942 case LINE_STAT_UNSTAGED:
4943 return run_io(io, others_argv, opt_cdup, IO_WR);
4944
4945 case LINE_STAT_UNTRACKED:
4946 return run_io(io, others_argv, NULL, IO_WR);
4947
4948 default:
4949 die("line type %d not handled in switch", type);
4950 return FALSE;
4951 }
4952 }
4953
4954 static bool
4955 status_update_write(struct io *io, struct status *status, enum line_type type)
4956 {
4957 char buf[SIZEOF_STR];
4958 size_t bufsize = 0;
4959
4960 switch (type) {
4961 case LINE_STAT_STAGED:
4962 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
4963 status->old.mode,
4964 status->old.rev,
4965 status->old.name, 0))
4966 return FALSE;
4967 break;
4968
4969 case LINE_STAT_UNSTAGED:
4970 case LINE_STAT_UNTRACKED:
4971 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
4972 return FALSE;
4973 break;
4974
4975 default:
4976 die("line type %d not handled in switch", type);
4977 }
4978
4979 return io_write(io, buf, bufsize);
4980 }
4981
4982 static bool
4983 status_update_file(struct status *status, enum line_type type)
4984 {
4985 struct io io = {};
4986 bool result;
4987
4988 if (!status_update_prepare(&io, type))
4989 return FALSE;
4990
4991 result = status_update_write(&io, status, type);
4992 done_io(&io);
4993 return result;
4994 }
4995
4996 static bool
4997 status_update_files(struct view *view, struct line *line)
4998 {
4999 struct io io = {};
5000 bool result = TRUE;
5001 struct line *pos = view->line + view->lines;
5002 int files = 0;
5003 int file, done;
5004
5005 if (!status_update_prepare(&io, line->type))
5006 return FALSE;
5007
5008 for (pos = line; pos < view->line + view->lines && pos->data; pos++)
5009 files++;
5010
5011 for (file = 0, done = 0; result && file < files; line++, file++) {
5012 int almost_done = file * 100 / files;
5013
5014 if (almost_done > done) {
5015 done = almost_done;
5016 string_format(view->ref, "updating file %u of %u (%d%% done)",
5017 file, files, done);
5018 update_view_title(view);
5019 }
5020 result = status_update_write(&io, line->data, line->type);
5021 }
5022
5023 done_io(&io);
5024 return result;
5025 }
5026
5027 static bool
5028 status_update(struct view *view)
5029 {
5030 struct line *line = &view->line[view->lineno];
5031
5032 assert(view->lines);
5033
5034 if (!line->data) {
5035 /* This should work even for the "On branch" line. */
5036 if (line < view->line + view->lines && !line[1].data) {
5037 report("Nothing to update");
5038 return FALSE;
5039 }
5040
5041 if (!status_update_files(view, line + 1)) {
5042 report("Failed to update file status");
5043 return FALSE;
5044 }
5045
5046 } else if (!status_update_file(line->data, line->type)) {
5047 report("Failed to update file status");
5048 return FALSE;
5049 }
5050
5051 return TRUE;
5052 }
5053
5054 static bool
5055 status_revert(struct status *status, enum line_type type, bool has_none)
5056 {
5057 if (!status || type != LINE_STAT_UNSTAGED) {
5058 if (type == LINE_STAT_STAGED) {
5059 report("Cannot revert changes to staged files");
5060 } else if (type == LINE_STAT_UNTRACKED) {
5061 report("Cannot revert changes to untracked files");
5062 } else if (has_none) {
5063 report("Nothing to revert");
5064 } else {
5065 report("Cannot revert changes to multiple files");
5066 }
5067 return FALSE;
5068
5069 } else {
5070 char mode[10] = "100644";
5071 const char *reset_argv[] = {
5072 "git", "update-index", "--cacheinfo", mode,
5073 status->old.rev, status->old.name, NULL
5074 };
5075 const char *checkout_argv[] = {
5076 "git", "checkout", "--", status->old.name, NULL
5077 };
5078
5079 if (!prompt_yesno("Are you sure you want to overwrite any changes?"))
5080 return FALSE;
5081 string_format(mode, "%o", status->old.mode);
5082 return (status->status != 'U' || run_io_fg(reset_argv, opt_cdup)) &&
5083 run_io_fg(checkout_argv, opt_cdup);
5084 }
5085 }
5086
5087 static enum request
5088 status_request(struct view *view, enum request request, struct line *line)
5089 {
5090 struct status *status = line->data;
5091
5092 switch (request) {
5093 case REQ_STATUS_UPDATE:
5094 if (!status_update(view))
5095 return REQ_NONE;
5096 break;
5097
5098 case REQ_STATUS_REVERT:
5099 if (!status_revert(status, line->type, status_has_none(view, line)))
5100 return REQ_NONE;
5101 break;
5102
5103 case REQ_STATUS_MERGE:
5104 if (!status || status->status != 'U') {
5105 report("Merging only possible for files with unmerged status ('U').");
5106 return REQ_NONE;
5107 }
5108 open_mergetool(status->new.name);
5109 break;
5110
5111 case REQ_EDIT:
5112 if (!status)
5113 return request;
5114 if (status->status == 'D') {
5115 report("File has been deleted.");
5116 return REQ_NONE;
5117 }
5118
5119 open_editor(status->status != '?', status->new.name);
5120 break;
5121
5122 case REQ_VIEW_BLAME:
5123 if (status) {
5124 string_copy(opt_file, status->new.name);
5125 opt_ref[0] = 0;
5126 }
5127 return request;
5128
5129 case REQ_ENTER:
5130 /* After returning the status view has been split to
5131 * show the stage view. No further reloading is
5132 * necessary. */
5133 status_enter(view, line);
5134 return REQ_NONE;
5135
5136 case REQ_REFRESH:
5137 /* Simply reload the view. */
5138 break;
5139
5140 default:
5141 return request;
5142 }
5143
5144 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
5145
5146 return REQ_NONE;
5147 }
5148
5149 static void
5150 status_select(struct view *view, struct line *line)
5151 {
5152 struct status *status = line->data;
5153 char file[SIZEOF_STR] = "all files";
5154 const char *text;
5155 const char *key;
5156
5157 if (status && !string_format(file, "'%s'", status->new.name))
5158 return;
5159
5160 if (!status && line[1].type == LINE_STAT_NONE)
5161 line++;
5162
5163 switch (line->type) {
5164 case LINE_STAT_STAGED:
5165 text = "Press %s to unstage %s for commit";
5166 break;
5167
5168 case LINE_STAT_UNSTAGED:
5169 text = "Press %s to stage %s for commit";
5170 break;
5171
5172 case LINE_STAT_UNTRACKED:
5173 text = "Press %s to stage %s for addition";
5174 break;
5175
5176 case LINE_STAT_HEAD:
5177 case LINE_STAT_NONE:
5178 text = "Nothing to update";
5179 break;
5180
5181 default:
5182 die("line type %d not handled in switch", line->type);
5183 }
5184
5185 if (status && status->status == 'U') {
5186 text = "Press %s to resolve conflict in %s";
5187 key = get_key(REQ_STATUS_MERGE);
5188
5189 } else {
5190 key = get_key(REQ_STATUS_UPDATE);
5191 }
5192
5193 string_format(view->ref, text, key, file);
5194 }
5195
5196 static bool
5197 status_grep(struct view *view, struct line *line)
5198 {
5199 struct status *status = line->data;
5200 enum { S_STATUS, S_NAME, S_END } state;
5201 char buf[2] = "?";
5202 regmatch_t pmatch;
5203
5204 if (!status)
5205 return FALSE;
5206
5207 for (state = S_STATUS; state < S_END; state++) {
5208 const char *text;
5209
5210 switch (state) {
5211 case S_NAME: text = status->new.name; break;
5212 case S_STATUS:
5213 buf[0] = status->status;
5214 text = buf;
5215 break;
5216
5217 default:
5218 return FALSE;
5219 }
5220
5221 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5222 return TRUE;
5223 }
5224
5225 return FALSE;
5226 }
5227
5228 static struct view_ops status_ops = {
5229 "file",
5230 NULL,
5231 status_open,
5232 NULL,
5233 status_draw,
5234 status_request,
5235 status_grep,
5236 status_select,
5237 };
5238
5239
5240 static bool
5241 stage_diff_write(struct io *io, struct line *line, struct line *end)
5242 {
5243 while (line < end) {
5244 if (!io_write(io, line->data, strlen(line->data)) ||
5245 !io_write(io, "\n", 1))
5246 return FALSE;
5247 line++;
5248 if (line->type == LINE_DIFF_CHUNK ||
5249 line->type == LINE_DIFF_HEADER)
5250 break;
5251 }
5252
5253 return TRUE;
5254 }
5255
5256 static struct line *
5257 stage_diff_find(struct view *view, struct line *line, enum line_type type)
5258 {
5259 for (; view->line < line; line--)
5260 if (line->type == type)
5261 return line;
5262
5263 return NULL;
5264 }
5265
5266 static bool
5267 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
5268 {
5269 const char *apply_argv[SIZEOF_ARG] = {
5270 "git", "apply", "--whitespace=nowarn", NULL
5271 };
5272 struct line *diff_hdr;
5273 struct io io = {};
5274 int argc = 3;
5275
5276 diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
5277 if (!diff_hdr)
5278 return FALSE;
5279
5280 if (!revert)
5281 apply_argv[argc++] = "--cached";
5282 if (revert || stage_line_type == LINE_STAT_STAGED)
5283 apply_argv[argc++] = "-R";
5284 apply_argv[argc++] = "-";
5285 apply_argv[argc++] = NULL;
5286 if (!run_io(&io, apply_argv, opt_cdup, IO_WR))
5287 return FALSE;
5288
5289 if (!stage_diff_write(&io, diff_hdr, chunk) ||
5290 !stage_diff_write(&io, chunk, view->line + view->lines))
5291 chunk = NULL;
5292
5293 done_io(&io);
5294 run_io_bg(update_index_argv);
5295
5296 return chunk ? TRUE : FALSE;
5297 }
5298
5299 static bool
5300 stage_update(struct view *view, struct line *line)
5301 {
5302 struct line *chunk = NULL;
5303
5304 if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
5305 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5306
5307 if (chunk) {
5308 if (!stage_apply_chunk(view, chunk, FALSE)) {
5309 report("Failed to apply chunk");
5310 return FALSE;
5311 }
5312
5313 } else if (!stage_status.status) {
5314 view = VIEW(REQ_VIEW_STATUS);
5315
5316 for (line = view->line; line < view->line + view->lines; line++)
5317 if (line->type == stage_line_type)
5318 break;
5319
5320 if (!status_update_files(view, line + 1)) {
5321 report("Failed to update files");
5322 return FALSE;
5323 }
5324
5325 } else if (!status_update_file(&stage_status, stage_line_type)) {
5326 report("Failed to update file");
5327 return FALSE;
5328 }
5329
5330 return TRUE;
5331 }
5332
5333 static bool
5334 stage_revert(struct view *view, struct line *line)
5335 {
5336 struct line *chunk = NULL;
5337
5338 if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
5339 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5340
5341 if (chunk) {
5342 if (!prompt_yesno("Are you sure you want to revert changes?"))
5343 return FALSE;
5344
5345 if (!stage_apply_chunk(view, chunk, TRUE)) {
5346 report("Failed to revert chunk");
5347 return FALSE;
5348 }
5349 return TRUE;
5350
5351 } else {
5352 return status_revert(stage_status.status ? &stage_status : NULL,
5353 stage_line_type, FALSE);
5354 }
5355 }
5356
5357
5358 static void
5359 stage_next(struct view *view, struct line *line)
5360 {
5361 int i;
5362
5363 if (!stage_chunks) {
5364 static size_t alloc = 0;
5365 int *tmp;
5366
5367 for (line = view->line; line < view->line + view->lines; line++) {
5368 if (line->type != LINE_DIFF_CHUNK)
5369 continue;
5370
5371 tmp = realloc_items(stage_chunk, &alloc,
5372 stage_chunks, sizeof(*tmp));
5373 if (!tmp) {
5374 report("Allocation failure");
5375 return;
5376 }
5377
5378 stage_chunk = tmp;
5379 stage_chunk[stage_chunks++] = line - view->line;
5380 }
5381 }
5382
5383 for (i = 0; i < stage_chunks; i++) {
5384 if (stage_chunk[i] > view->lineno) {
5385 do_scroll_view(view, stage_chunk[i] - view->lineno);
5386 report("Chunk %d of %d", i + 1, stage_chunks);
5387 return;
5388 }
5389 }
5390
5391 report("No next chunk found");
5392 }
5393
5394 static enum request
5395 stage_request(struct view *view, enum request request, struct line *line)
5396 {
5397 switch (request) {
5398 case REQ_STATUS_UPDATE:
5399 if (!stage_update(view, line))
5400 return REQ_NONE;
5401 break;
5402
5403 case REQ_STATUS_REVERT:
5404 if (!stage_revert(view, line))
5405 return REQ_NONE;
5406 break;
5407
5408 case REQ_STAGE_NEXT:
5409 if (stage_line_type == LINE_STAT_UNTRACKED) {
5410 report("File is untracked; press %s to add",
5411 get_key(REQ_STATUS_UPDATE));
5412 return REQ_NONE;
5413 }
5414 stage_next(view, line);
5415 return REQ_NONE;
5416
5417 case REQ_EDIT:
5418 if (!stage_status.new.name[0])
5419 return request;
5420 if (stage_status.status == 'D') {
5421 report("File has been deleted.");
5422 return REQ_NONE;
5423 }
5424
5425 open_editor(stage_status.status != '?', stage_status.new.name);
5426 break;
5427
5428 case REQ_REFRESH:
5429 /* Reload everything ... */
5430 break;
5431
5432 case REQ_VIEW_BLAME:
5433 if (stage_status.new.name[0]) {
5434 string_copy(opt_file, stage_status.new.name);
5435 opt_ref[0] = 0;
5436 }
5437 return request;
5438
5439 case REQ_ENTER:
5440 return pager_request(view, request, line);
5441
5442 default:
5443 return request;
5444 }
5445
5446 VIEW(REQ_VIEW_STATUS)->p_restore = TRUE;
5447 open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD | OPEN_NOMAXIMIZE);
5448
5449 /* Check whether the staged entry still exists, and close the
5450 * stage view if it doesn't. */
5451 if (!status_exists(&stage_status, stage_line_type)) {
5452 status_restore(VIEW(REQ_VIEW_STATUS));
5453 return REQ_VIEW_CLOSE;
5454 }
5455
5456 if (stage_line_type == LINE_STAT_UNTRACKED) {
5457 if (!suffixcmp(stage_status.new.name, -1, "/")) {
5458 report("Cannot display a directory");
5459 return REQ_NONE;
5460 }
5461
5462 if (!prepare_update_file(view, stage_status.new.name)) {
5463 report("Failed to open file: %s", strerror(errno));
5464 return REQ_NONE;
5465 }
5466 }
5467 open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
5468
5469 return REQ_NONE;
5470 }
5471
5472 static struct view_ops stage_ops = {
5473 "line",
5474 NULL,
5475 NULL,
5476 pager_read,
5477 pager_draw,
5478 stage_request,
5479 pager_grep,
5480 pager_select,
5481 };
5482
5483
5484 /*
5485 * Revision graph
5486 */
5487
5488 struct commit {
5489 char id[SIZEOF_REV]; /* SHA1 ID. */
5490 char title[128]; /* First line of the commit message. */
5491 char author[75]; /* Author of the commit. */
5492 struct tm time; /* Date from the author ident. */
5493 struct ref **refs; /* Repository references. */
5494 chtype graph[SIZEOF_REVGRAPH]; /* Ancestry chain graphics. */
5495 size_t graph_size; /* The width of the graph array. */
5496 bool has_parents; /* Rewritten --parents seen. */
5497 };
5498
5499 /* Size of rev graph with no "padding" columns */
5500 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
5501
5502 struct rev_graph {
5503 struct rev_graph *prev, *next, *parents;
5504 char rev[SIZEOF_REVITEMS][SIZEOF_REV];
5505 size_t size;
5506 struct commit *commit;
5507 size_t pos;
5508 unsigned int boundary:1;
5509 };
5510
5511 /* Parents of the commit being visualized. */
5512 static struct rev_graph graph_parents[4];
5513
5514 /* The current stack of revisions on the graph. */
5515 static struct rev_graph graph_stacks[4] = {
5516 { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
5517 { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
5518 { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
5519 { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
5520 };
5521
5522 static inline bool
5523 graph_parent_is_merge(struct rev_graph *graph)
5524 {
5525 return graph->parents->size > 1;
5526 }
5527
5528 static inline void
5529 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
5530 {
5531 struct commit *commit = graph->commit;
5532
5533 if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
5534 commit->graph[commit->graph_size++] = symbol;
5535 }
5536
5537 static void
5538 clear_rev_graph(struct rev_graph *graph)
5539 {
5540 graph->boundary = 0;
5541 graph->size = graph->pos = 0;
5542 graph->commit = NULL;
5543 memset(graph->parents, 0, sizeof(*graph->parents));
5544 }
5545
5546 static void
5547 done_rev_graph(struct rev_graph *graph)
5548 {
5549 if (graph_parent_is_merge(graph) &&
5550 graph->pos < graph->size - 1 &&
5551 graph->next->size == graph->size + graph->parents->size - 1) {
5552 size_t i = graph->pos + graph->parents->size - 1;
5553
5554 graph->commit->graph_size = i * 2;
5555 while (i < graph->next->size - 1) {
5556 append_to_rev_graph(graph, ' ');
5557 append_to_rev_graph(graph, '\\');
5558 i++;
5559 }
5560 }
5561
5562 clear_rev_graph(graph);
5563 }
5564
5565 static void
5566 push_rev_graph(struct rev_graph *graph, const char *parent)
5567 {
5568 int i;
5569
5570 /* "Collapse" duplicate parents lines.
5571 *
5572 * FIXME: This needs to also update update the drawn graph but
5573 * for now it just serves as a method for pruning graph lines. */
5574 for (i = 0; i < graph->size; i++)
5575 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
5576 return;
5577
5578 if (graph->size < SIZEOF_REVITEMS) {
5579 string_copy_rev(graph->rev[graph->size++], parent);
5580 }
5581 }
5582
5583 static chtype
5584 get_rev_graph_symbol(struct rev_graph *graph)
5585 {
5586 chtype symbol;
5587
5588 if (graph->boundary)
5589 symbol = REVGRAPH_BOUND;
5590 else if (graph->parents->size == 0)
5591 symbol = REVGRAPH_INIT;
5592 else if (graph_parent_is_merge(graph))
5593 symbol = REVGRAPH_MERGE;
5594 else if (graph->pos >= graph->size)
5595 symbol = REVGRAPH_BRANCH;
5596 else
5597 symbol = REVGRAPH_COMMIT;
5598
5599 return symbol;
5600 }
5601
5602 static void
5603 draw_rev_graph(struct rev_graph *graph)
5604 {
5605 struct rev_filler {
5606 chtype separator, line;
5607 };
5608 enum { DEFAULT, RSHARP, RDIAG, LDIAG };
5609 static struct rev_filler fillers[] = {
5610 { ' ', '|' },
5611 { '`', '.' },
5612 { '\'', ' ' },
5613 { '/', ' ' },
5614 };
5615 chtype symbol = get_rev_graph_symbol(graph);
5616 struct rev_filler *filler;
5617 size_t i;
5618
5619 if (opt_line_graphics)
5620 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
5621
5622 filler = &fillers[DEFAULT];
5623
5624 for (i = 0; i < graph->pos; i++) {
5625 append_to_rev_graph(graph, filler->line);
5626 if (graph_parent_is_merge(graph->prev) &&
5627 graph->prev->pos == i)
5628 filler = &fillers[RSHARP];
5629
5630 append_to_rev_graph(graph, filler->separator);
5631 }
5632
5633 /* Place the symbol for this revision. */
5634 append_to_rev_graph(graph, symbol);
5635
5636 if (graph->prev->size > graph->size)
5637 filler = &fillers[RDIAG];
5638 else
5639 filler = &fillers[DEFAULT];
5640
5641 i++;
5642
5643 for (; i < graph->size; i++) {
5644 append_to_rev_graph(graph, filler->separator);
5645 append_to_rev_graph(graph, filler->line);
5646 if (graph_parent_is_merge(graph->prev) &&
5647 i < graph->prev->pos + graph->parents->size)
5648 filler = &fillers[RSHARP];
5649 if (graph->prev->size > graph->size)
5650 filler = &fillers[LDIAG];
5651 }
5652
5653 if (graph->prev->size > graph->size) {
5654 append_to_rev_graph(graph, filler->separator);
5655 if (filler->line != ' ')
5656 append_to_rev_graph(graph, filler->line);
5657 }
5658 }
5659
5660 /* Prepare the next rev graph */
5661 static void
5662 prepare_rev_graph(struct rev_graph *graph)
5663 {
5664 size_t i;
5665
5666 /* First, traverse all lines of revisions up to the active one. */
5667 for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5668 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5669 break;
5670
5671 push_rev_graph(graph->next, graph->rev[graph->pos]);
5672 }
5673
5674 /* Interleave the new revision parent(s). */
5675 for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5676 push_rev_graph(graph->next, graph->parents->rev[i]);
5677
5678 /* Lastly, put any remaining revisions. */
5679 for (i = graph->pos + 1; i < graph->size; i++)
5680 push_rev_graph(graph->next, graph->rev[i]);
5681 }
5682
5683 static void
5684 update_rev_graph(struct view *view, struct rev_graph *graph)
5685 {
5686 /* If this is the finalizing update ... */
5687 if (graph->commit)
5688 prepare_rev_graph(graph);
5689
5690 /* Graph visualization needs a one rev look-ahead,
5691 * so the first update doesn't visualize anything. */
5692 if (!graph->prev->commit)
5693 return;
5694
5695 if (view->lines > 2)
5696 view->line[view->lines - 3].dirty = 1;
5697 if (view->lines > 1)
5698 view->line[view->lines - 2].dirty = 1;
5699 draw_rev_graph(graph->prev);
5700 done_rev_graph(graph->prev->prev);
5701 }
5702
5703
5704 /*
5705 * Main view backend
5706 */
5707
5708 static const char *main_argv[SIZEOF_ARG] = {
5709 "git", "log", "--no-color", "--pretty=raw", "--parents",
5710 "--topo-order", "%(head)", NULL
5711 };
5712
5713 static bool
5714 main_draw(struct view *view, struct line *line, unsigned int lineno)
5715 {
5716 struct commit *commit = line->data;
5717
5718 if (!*commit->author)
5719 return FALSE;
5720
5721 if (opt_date && draw_date(view, &commit->time))
5722 return TRUE;
5723
5724 if (opt_author && draw_author(view, commit->author))
5725 return TRUE;
5726
5727 if (opt_rev_graph && commit->graph_size &&
5728 draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5729 return TRUE;
5730
5731 if (opt_show_refs && commit->refs) {
5732 size_t i = 0;
5733
5734 do {
5735 enum line_type type;
5736
5737 if (commit->refs[i]->head)
5738 type = LINE_MAIN_HEAD;
5739 else if (commit->refs[i]->ltag)
5740 type = LINE_MAIN_LOCAL_TAG;
5741 else if (commit->refs[i]->tag)
5742 type = LINE_MAIN_TAG;
5743 else if (commit->refs[i]->tracked)
5744 type = LINE_MAIN_TRACKED;
5745 else if (commit->refs[i]->remote)
5746 type = LINE_MAIN_REMOTE;
5747 else
5748 type = LINE_MAIN_REF;
5749
5750 if (draw_text(view, type, "[", TRUE) ||
5751 draw_text(view, type, commit->refs[i]->name, TRUE) ||
5752 draw_text(view, type, "]", TRUE))
5753 return TRUE;
5754
5755 if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5756 return TRUE;
5757 } while (commit->refs[i++]->next);
5758 }
5759
5760 draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5761 return TRUE;
5762 }
5763
5764 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5765 static bool
5766 main_read(struct view *view, char *line)
5767 {
5768 static struct rev_graph *graph = graph_stacks;
5769 enum line_type type;
5770 struct commit *commit;
5771
5772 if (!line) {
5773 int i;
5774
5775 if (!view->lines && !view->parent)
5776 die("No revisions match the given arguments.");
5777 if (view->lines > 0) {
5778 commit = view->line[view->lines - 1].data;
5779 view->line[view->lines - 1].dirty = 1;
5780 if (!*commit->author) {
5781 view->lines--;
5782 free(commit);
5783 graph->commit = NULL;
5784 }
5785 }
5786 update_rev_graph(view, graph);
5787
5788 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5789 clear_rev_graph(&graph_stacks[i]);
5790 return TRUE;
5791 }
5792
5793 type = get_line_type(line);
5794 if (type == LINE_COMMIT) {
5795 commit = calloc(1, sizeof(struct commit));
5796 if (!commit)
5797 return FALSE;
5798
5799 line += STRING_SIZE("commit ");
5800 if (*line == '-') {
5801 graph->boundary = 1;
5802 line++;
5803 }
5804
5805 string_copy_rev(commit->id, line);
5806 commit->refs = get_refs(commit->id);
5807 graph->commit = commit;
5808 add_line_data(view, commit, LINE_MAIN_COMMIT);
5809
5810 while ((line = strchr(line, ' '))) {
5811 line++;
5812 push_rev_graph(graph->parents, line);
5813 commit->has_parents = TRUE;
5814 }
5815 return TRUE;
5816 }
5817
5818 if (!view->lines)
5819 return TRUE;
5820 commit = view->line[view->lines - 1].data;
5821
5822 switch (type) {
5823 case LINE_PARENT:
5824 if (commit->has_parents)
5825 break;
5826 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5827 break;
5828
5829 case LINE_AUTHOR:
5830 parse_author_line(line + STRING_SIZE("author "),
5831 commit->author, sizeof(commit->author),
5832 &commit->time);
5833 update_rev_graph(view, graph);
5834 graph = graph->next;
5835 break;
5836
5837 default:
5838 /* Fill in the commit title if it has not already been set. */
5839 if (commit->title[0])
5840 break;
5841
5842 /* Require titles to start with a non-space character at the
5843 * offset used by git log. */
5844 if (strncmp(line, " ", 4))
5845 break;
5846 line += 4;
5847 /* Well, if the title starts with a whitespace character,
5848 * try to be forgiving. Otherwise we end up with no title. */
5849 while (isspace(*line))
5850 line++;
5851 if (*line == '\0')
5852 break;
5853 /* FIXME: More graceful handling of titles; append "..." to
5854 * shortened titles, etc. */
5855
5856 string_expand(commit->title, sizeof(commit->title), line, 1);
5857 view->line[view->lines - 1].dirty = 1;
5858 }
5859
5860 return TRUE;
5861 }
5862
5863 static enum request
5864 main_request(struct view *view, enum request request, struct line *line)
5865 {
5866 enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5867
5868 switch (request) {
5869 case REQ_ENTER:
5870 open_view(view, REQ_VIEW_DIFF, flags);
5871 break;
5872 case REQ_REFRESH:
5873 load_refs();
5874 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
5875 break;
5876 default:
5877 return request;
5878 }
5879
5880 return REQ_NONE;
5881 }
5882
5883 static bool
5884 grep_refs(struct ref **refs, regex_t *regex)
5885 {
5886 regmatch_t pmatch;
5887 size_t i = 0;
5888
5889 if (!refs)
5890 return FALSE;
5891 do {
5892 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
5893 return TRUE;
5894 } while (refs[i++]->next);
5895
5896 return FALSE;
5897 }
5898
5899 static bool
5900 main_grep(struct view *view, struct line *line)
5901 {
5902 struct commit *commit = line->data;
5903 enum { S_TITLE, S_AUTHOR, S_DATE, S_REFS, S_END } state;
5904 char buf[DATE_COLS + 1];
5905 regmatch_t pmatch;
5906
5907 for (state = S_TITLE; state < S_END; state++) {
5908 char *text;
5909
5910 switch (state) {
5911 case S_TITLE: text = commit->title; break;
5912 case S_AUTHOR:
5913 if (!opt_author)
5914 continue;
5915 text = commit->author;
5916 break;
5917 case S_DATE:
5918 if (!opt_date)
5919 continue;
5920 if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
5921 continue;
5922 text = buf;
5923 break;
5924 case S_REFS:
5925 if (!opt_show_refs)
5926 continue;
5927 if (grep_refs(commit->refs, view->regex) == TRUE)
5928 return TRUE;
5929 continue;
5930 default:
5931 return FALSE;
5932 }
5933
5934 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5935 return TRUE;
5936 }
5937
5938 return FALSE;
5939 }
5940
5941 static void
5942 main_select(struct view *view, struct line *line)
5943 {
5944 struct commit *commit = line->data;
5945
5946 string_copy_rev(view->ref, commit->id);
5947 string_copy_rev(ref_commit, view->ref);
5948 }
5949
5950 static struct view_ops main_ops = {
5951 "commit",
5952 main_argv,
5953 NULL,
5954 main_read,
5955 main_draw,
5956 main_request,
5957 main_grep,
5958 main_select,
5959 };
5960
5961
5962 /*
5963 * Unicode / UTF-8 handling
5964 *
5965 * NOTE: Much of the following code for dealing with Unicode is derived from
5966 * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
5967 * src/intl/charset.c from the UTF-8 branch commit elinks-0.11.0-g31f2c28.
5968 */
5969
5970 static inline int
5971 unicode_width(unsigned long c)
5972 {
5973 if (c >= 0x1100 &&
5974 (c <= 0x115f /* Hangul Jamo */
5975 || c == 0x2329
5976 || c == 0x232a
5977 || (c >= 0x2e80 && c <= 0xa4cf && c != 0x303f)
5978 /* CJK ... Yi */
5979 || (c >= 0xac00 && c <= 0xd7a3) /* Hangul Syllables */
5980 || (c >= 0xf900 && c <= 0xfaff) /* CJK Compatibility Ideographs */
5981 || (c >= 0xfe30 && c <= 0xfe6f) /* CJK Compatibility Forms */
5982 || (c >= 0xff00 && c <= 0xff60) /* Fullwidth Forms */
5983 || (c >= 0xffe0 && c <= 0xffe6)
5984 || (c >= 0x20000 && c <= 0x2fffd)
5985 || (c >= 0x30000 && c <= 0x3fffd)))
5986 return 2;
5987
5988 if (c == '\t')
5989 return opt_tab_size;
5990
5991 return 1;
5992 }
5993
5994 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
5995 * Illegal bytes are set one. */
5996 static const unsigned char utf8_bytes[256] = {
5997 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,
5998 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,
5999 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,
6000 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,
6001 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,
6002 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,
6003 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,
6004 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,
6005 };
6006
6007 /* Decode UTF-8 multi-byte representation into a Unicode character. */
6008 static inline unsigned long
6009 utf8_to_unicode(const char *string, size_t length)
6010 {
6011 unsigned long unicode;
6012
6013 switch (length) {
6014 case 1:
6015 unicode = string[0];
6016 break;
6017 case 2:
6018 unicode = (string[0] & 0x1f) << 6;
6019 unicode += (string[1] & 0x3f);
6020 break;
6021 case 3:
6022 unicode = (string[0] & 0x0f) << 12;
6023 unicode += ((string[1] & 0x3f) << 6);
6024 unicode += (string[2] & 0x3f);
6025 break;
6026 case 4:
6027 unicode = (string[0] & 0x0f) << 18;
6028 unicode += ((string[1] & 0x3f) << 12);
6029 unicode += ((string[2] & 0x3f) << 6);
6030 unicode += (string[3] & 0x3f);
6031 break;
6032 case 5:
6033 unicode = (string[0] & 0x0f) << 24;
6034 unicode += ((string[1] & 0x3f) << 18);
6035 unicode += ((string[2] & 0x3f) << 12);
6036 unicode += ((string[3] & 0x3f) << 6);
6037 unicode += (string[4] & 0x3f);
6038 break;
6039 case 6:
6040 unicode = (string[0] & 0x01) << 30;
6041 unicode += ((string[1] & 0x3f) << 24);
6042 unicode += ((string[2] & 0x3f) << 18);
6043 unicode += ((string[3] & 0x3f) << 12);
6044 unicode += ((string[4] & 0x3f) << 6);
6045 unicode += (string[5] & 0x3f);
6046 break;
6047 default:
6048 die("Invalid Unicode length");
6049 }
6050
6051 /* Invalid characters could return the special 0xfffd value but NUL
6052 * should be just as good. */
6053 return unicode > 0xffff ? 0 : unicode;
6054 }
6055
6056 /* Calculates how much of string can be shown within the given maximum width
6057 * and sets trimmed parameter to non-zero value if all of string could not be
6058 * shown. If the reserve flag is TRUE, it will reserve at least one
6059 * trailing character, which can be useful when drawing a delimiter.
6060 *
6061 * Returns the number of bytes to output from string to satisfy max_width. */
6062 static size_t
6063 utf8_length(const char **start, size_t skip, int *width, size_t max_width, int *trimmed, bool reserve)
6064 {
6065 const char *string = *start;
6066 const char *end = strchr(string, '\0');
6067 unsigned char last_bytes = 0;
6068 size_t last_ucwidth = 0;
6069
6070 *width = 0;
6071 *trimmed = 0;
6072
6073 while (string < end) {
6074 int c = *(unsigned char *) string;
6075 unsigned char bytes = utf8_bytes[c];
6076 size_t ucwidth;
6077 unsigned long unicode;
6078
6079 if (string + bytes > end)
6080 break;
6081
6082 /* Change representation to figure out whether
6083 * it is a single- or double-width character. */
6084
6085 unicode = utf8_to_unicode(string, bytes);
6086 /* FIXME: Graceful handling of invalid Unicode character. */
6087 if (!unicode)
6088 break;
6089
6090 ucwidth = unicode_width(unicode);
6091 if (skip > 0) {
6092 skip -= ucwidth <= skip ? ucwidth : skip;
6093 *start += bytes;
6094 }
6095 *width += ucwidth;
6096 if (*width > max_width) {
6097 *trimmed = 1;
6098 *width -= ucwidth;
6099 if (reserve && *width == max_width) {
6100 string -= last_bytes;
6101 *width -= last_ucwidth;
6102 }
6103 break;
6104 }
6105
6106 string += bytes;
6107 last_bytes = ucwidth ? bytes : 0;
6108 last_ucwidth = ucwidth;
6109 }
6110
6111 return string - *start;
6112 }
6113
6114
6115 /*
6116 * Status management
6117 */
6118
6119 /* Whether or not the curses interface has been initialized. */
6120 static bool cursed = FALSE;
6121
6122 /* Terminal hacks and workarounds. */
6123 static bool use_scroll_redrawwin;
6124 static bool use_scroll_status_wclear;
6125
6126 /* The status window is used for polling keystrokes. */
6127 static WINDOW *status_win;
6128
6129 /* Reading from the prompt? */
6130 static bool input_mode = FALSE;
6131
6132 static bool status_empty = FALSE;
6133
6134 /* Update status and title window. */
6135 static void
6136 report(const char *msg, ...)
6137 {
6138 struct view *view = display[current_view];
6139
6140 if (input_mode)
6141 return;
6142
6143 if (!view) {
6144 char buf[SIZEOF_STR];
6145 va_list args;
6146
6147 va_start(args, msg);
6148 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
6149 buf[sizeof(buf) - 1] = 0;
6150 buf[sizeof(buf) - 2] = '.';
6151 buf[sizeof(buf) - 3] = '.';
6152 buf[sizeof(buf) - 4] = '.';
6153 }
6154 va_end(args);
6155 die("%s", buf);
6156 }
6157
6158 if (!status_empty || *msg) {
6159 va_list args;
6160
6161 va_start(args, msg);
6162
6163 wmove(status_win, 0, 0);
6164 if (view->has_scrolled && use_scroll_status_wclear)
6165 wclear(status_win);
6166 if (*msg) {
6167 vwprintw(status_win, msg, args);
6168 status_empty = FALSE;
6169 } else {
6170 status_empty = TRUE;
6171 }
6172 wclrtoeol(status_win);
6173 wnoutrefresh(status_win);
6174
6175 va_end(args);
6176 }
6177
6178 update_view_title(view);
6179 }
6180
6181 /* Controls when nodelay should be in effect when polling user input. */
6182 static void
6183 set_nonblocking_input(bool loading)
6184 {
6185 static unsigned int loading_views;
6186
6187 if ((loading == FALSE && loading_views-- == 1) ||
6188 (loading == TRUE && loading_views++ == 0))
6189 nodelay(status_win, loading);
6190 }
6191
6192 static void
6193 init_display(void)
6194 {
6195 const char *term;
6196 int x, y;
6197
6198 /* Initialize the curses library */
6199 if (isatty(STDIN_FILENO)) {
6200 cursed = !!initscr();
6201 opt_tty = stdin;
6202 } else {
6203 /* Leave stdin and stdout alone when acting as a pager. */
6204 opt_tty = fopen("/dev/tty", "r+");
6205 if (!opt_tty)
6206 die("Failed to open /dev/tty");
6207 cursed = !!newterm(NULL, opt_tty, opt_tty);
6208 }
6209
6210 if (!cursed)
6211 die("Failed to initialize curses");
6212
6213 nonl(); /* Disable conversion and detect newlines from input. */
6214 cbreak(); /* Take input chars one at a time, no wait for \n */
6215 noecho(); /* Don't echo input */
6216 leaveok(stdscr, FALSE);
6217
6218 if (has_colors())
6219 init_colors();
6220
6221 getmaxyx(stdscr, y, x);
6222 status_win = newwin(1, 0, y - 1, 0);
6223 if (!status_win)
6224 die("Failed to create status window");
6225
6226 /* Enable keyboard mapping */
6227 keypad(status_win, TRUE);
6228 wbkgdset(status_win, get_line_attr(LINE_STATUS));
6229
6230 TABSIZE = opt_tab_size;
6231 if (opt_line_graphics) {
6232 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
6233 }
6234
6235 term = getenv("XTERM_VERSION") ? NULL : getenv("COLORTERM");
6236 if (term && !strcmp(term, "gnome-terminal")) {
6237 /* In the gnome-terminal-emulator, the message from
6238 * scrolling up one line when impossible followed by
6239 * scrolling down one line causes corruption of the
6240 * status line. This is fixed by calling wclear. */
6241 use_scroll_status_wclear = TRUE;
6242 use_scroll_redrawwin = FALSE;
6243
6244 } else if (term && !strcmp(term, "xrvt-xpm")) {
6245 /* No problems with full optimizations in xrvt-(unicode)
6246 * and aterm. */
6247 use_scroll_status_wclear = use_scroll_redrawwin = FALSE;
6248
6249 } else {
6250 /* When scrolling in (u)xterm the last line in the
6251 * scrolling direction will update slowly. */
6252 use_scroll_redrawwin = TRUE;
6253 use_scroll_status_wclear = FALSE;
6254 }
6255 }
6256
6257 static int
6258 get_input(int prompt_position)
6259 {
6260 struct view *view;
6261 int i, key, cursor_y, cursor_x;
6262
6263 if (prompt_position)
6264 input_mode = TRUE;
6265
6266 while (TRUE) {
6267 foreach_view (view, i) {
6268 update_view(view);
6269 if (view_is_displayed(view) && view->has_scrolled &&
6270 use_scroll_redrawwin)
6271 redrawwin(view->win);
6272 view->has_scrolled = FALSE;
6273 }
6274
6275 /* Update the cursor position. */
6276 if (prompt_position) {
6277 getbegyx(status_win, cursor_y, cursor_x);
6278 cursor_x = prompt_position;
6279 } else {
6280 view = display[current_view];
6281 getbegyx(view->win, cursor_y, cursor_x);
6282 cursor_x = view->width - 1;
6283 cursor_y += view->lineno - view->offset;
6284 }
6285 setsyx(cursor_y, cursor_x);
6286
6287 /* Refresh, accept single keystroke of input */
6288 doupdate();
6289 key = wgetch(status_win);
6290
6291 /* wgetch() with nodelay() enabled returns ERR when
6292 * there's no input. */
6293 if (key == ERR) {
6294
6295 } else if (key == KEY_RESIZE) {
6296 int height, width;
6297
6298 getmaxyx(stdscr, height, width);
6299
6300 wresize(status_win, 1, width);
6301 mvwin(status_win, height - 1, 0);
6302 wnoutrefresh(status_win);
6303 resize_display();
6304 redraw_display(TRUE);
6305
6306 } else {
6307 input_mode = FALSE;
6308 return key;
6309 }
6310 }
6311 }
6312
6313 static char *
6314 prompt_input(const char *prompt, input_handler handler, void *data)
6315 {
6316 enum input_status status = INPUT_OK;
6317 static char buf[SIZEOF_STR];
6318 size_t pos = 0;
6319
6320 buf[pos] = 0;
6321
6322 while (status == INPUT_OK || status == INPUT_SKIP) {
6323 int key;
6324
6325 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
6326 wclrtoeol(status_win);
6327
6328 key = get_input(pos + 1);
6329 switch (key) {
6330 case KEY_RETURN:
6331 case KEY_ENTER:
6332 case '\n':
6333 status = pos ? INPUT_STOP : INPUT_CANCEL;
6334 break;
6335
6336 case KEY_BACKSPACE:
6337 if (pos > 0)
6338 buf[--pos] = 0;
6339 else
6340 status = INPUT_CANCEL;
6341 break;
6342
6343 case KEY_ESC:
6344 status = INPUT_CANCEL;
6345 break;
6346
6347 default:
6348 if (pos >= sizeof(buf)) {
6349 report("Input string too long");
6350 return NULL;
6351 }
6352
6353 status = handler(data, buf, key);
6354 if (status == INPUT_OK)
6355 buf[pos++] = (char) key;
6356 }
6357 }
6358
6359 /* Clear the status window */
6360 status_empty = FALSE;
6361 report("");
6362
6363 if (status == INPUT_CANCEL)
6364 return NULL;
6365
6366 buf[pos++] = 0;
6367
6368 return buf;
6369 }
6370
6371 static enum input_status
6372 prompt_yesno_handler(void *data, char *buf, int c)
6373 {
6374 if (c == 'y' || c == 'Y')
6375 return INPUT_STOP;
6376 if (c == 'n' || c == 'N')
6377 return INPUT_CANCEL;
6378 return INPUT_SKIP;
6379 }
6380
6381 static bool
6382 prompt_yesno(const char *prompt)
6383 {
6384 char prompt2[SIZEOF_STR];
6385
6386 if (!string_format(prompt2, "%s [Yy/Nn]", prompt))
6387 return FALSE;
6388
6389 return !!prompt_input(prompt2, prompt_yesno_handler, NULL);
6390 }
6391
6392 static enum input_status
6393 read_prompt_handler(void *data, char *buf, int c)
6394 {
6395 return isprint(c) ? INPUT_OK : INPUT_SKIP;
6396 }
6397
6398 static char *
6399 read_prompt(const char *prompt)
6400 {
6401 return prompt_input(prompt, read_prompt_handler, NULL);
6402 }
6403
6404 /*
6405 * Repository properties
6406 */
6407
6408 static struct ref *refs = NULL;
6409 static size_t refs_alloc = 0;
6410 static size_t refs_size = 0;
6411
6412 /* Id <-> ref store */
6413 static struct ref ***id_refs = NULL;
6414 static size_t id_refs_alloc = 0;
6415 static size_t id_refs_size = 0;
6416
6417 static int
6418 compare_refs(const void *ref1_, const void *ref2_)
6419 {
6420 const struct ref *ref1 = *(const struct ref **)ref1_;
6421 const struct ref *ref2 = *(const struct ref **)ref2_;
6422
6423 if (ref1->tag != ref2->tag)
6424 return ref2->tag - ref1->tag;
6425 if (ref1->ltag != ref2->ltag)
6426 return ref2->ltag - ref2->ltag;
6427 if (ref1->head != ref2->head)
6428 return ref2->head - ref1->head;
6429 if (ref1->tracked != ref2->tracked)
6430 return ref2->tracked - ref1->tracked;
6431 if (ref1->remote != ref2->remote)
6432 return ref2->remote - ref1->remote;
6433 return strcmp(ref1->name, ref2->name);
6434 }
6435
6436 static struct ref **
6437 get_refs(const char *id)
6438 {
6439 struct ref ***tmp_id_refs;
6440 struct ref **ref_list = NULL;
6441 size_t ref_list_alloc = 0;
6442 size_t ref_list_size = 0;
6443 size_t i;
6444
6445 for (i = 0; i < id_refs_size; i++)
6446 if (!strcmp(id, id_refs[i][0]->id))
6447 return id_refs[i];
6448
6449 tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
6450 sizeof(*id_refs));
6451 if (!tmp_id_refs)
6452 return NULL;
6453
6454 id_refs = tmp_id_refs;
6455
6456 for (i = 0; i < refs_size; i++) {
6457 struct ref **tmp;
6458
6459 if (strcmp(id, refs[i].id))
6460 continue;
6461
6462 tmp = realloc_items(ref_list, &ref_list_alloc,
6463 ref_list_size + 1, sizeof(*ref_list));
6464 if (!tmp) {
6465 if (ref_list)
6466 free(ref_list);
6467 return NULL;
6468 }
6469
6470 ref_list = tmp;
6471 ref_list[ref_list_size] = &refs[i];
6472 /* XXX: The properties of the commit chains ensures that we can
6473 * safely modify the shared ref. The repo references will
6474 * always be similar for the same id. */
6475 ref_list[ref_list_size]->next = 1;
6476
6477 ref_list_size++;
6478 }
6479
6480 if (ref_list) {
6481 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
6482 ref_list[ref_list_size - 1]->next = 0;
6483 id_refs[id_refs_size++] = ref_list;
6484 }
6485
6486 return ref_list;
6487 }
6488
6489 static int
6490 read_ref(char *id, size_t idlen, char *name, size_t namelen)
6491 {
6492 struct ref *ref;
6493 bool tag = FALSE;
6494 bool ltag = FALSE;
6495 bool remote = FALSE;
6496 bool tracked = FALSE;
6497 bool check_replace = FALSE;
6498 bool head = FALSE;
6499
6500 if (!prefixcmp(name, "refs/tags/")) {
6501 if (!suffixcmp(name, namelen, "^{}")) {
6502 namelen -= 3;
6503 name[namelen] = 0;
6504 if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
6505 check_replace = TRUE;
6506 } else {
6507 ltag = TRUE;
6508 }
6509
6510 tag = TRUE;
6511 namelen -= STRING_SIZE("refs/tags/");
6512 name += STRING_SIZE("refs/tags/");
6513
6514 } else if (!prefixcmp(name, "refs/remotes/")) {
6515 remote = TRUE;
6516 namelen -= STRING_SIZE("refs/remotes/");
6517 name += STRING_SIZE("refs/remotes/");
6518 tracked = !strcmp(opt_remote, name);
6519
6520 } else if (!prefixcmp(name, "refs/heads/")) {
6521 namelen -= STRING_SIZE("refs/heads/");
6522 name += STRING_SIZE("refs/heads/");
6523 head = !strncmp(opt_head, name, namelen);
6524
6525 } else if (!strcmp(name, "HEAD")) {
6526 string_ncopy(opt_head_rev, id, idlen);
6527 return OK;
6528 }
6529
6530 if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
6531 /* it's an annotated tag, replace the previous SHA1 with the
6532 * resolved commit id; relies on the fact git-ls-remote lists
6533 * the commit id of an annotated tag right before the commit id
6534 * it points to. */
6535 refs[refs_size - 1].ltag = ltag;
6536 string_copy_rev(refs[refs_size - 1].id, id);
6537
6538 return OK;
6539 }
6540 refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
6541 if (!refs)
6542 return ERR;
6543
6544 ref = &refs[refs_size++];
6545 ref->name = malloc(namelen + 1);
6546 if (!ref->name)
6547 return ERR;
6548
6549 strncpy(ref->name, name, namelen);
6550 ref->name[namelen] = 0;
6551 ref->head = head;
6552 ref->tag = tag;
6553 ref->ltag = ltag;
6554 ref->remote = remote;
6555 ref->tracked = tracked;
6556 string_copy_rev(ref->id, id);
6557
6558 return OK;
6559 }
6560
6561 static int
6562 load_refs(void)
6563 {
6564 static const char *ls_remote_argv[SIZEOF_ARG] = {
6565 "git", "ls-remote", ".", NULL
6566 };
6567 static bool init = FALSE;
6568
6569 if (!init) {
6570 argv_from_env(ls_remote_argv, "TIG_LS_REMOTE");
6571 init = TRUE;
6572 }
6573
6574 if (!*opt_git_dir)
6575 return OK;
6576
6577 while (refs_size > 0)
6578 free(refs[--refs_size].name);
6579 while (id_refs_size > 0)
6580 free(id_refs[--id_refs_size]);
6581
6582 return run_io_load(ls_remote_argv, "\t", read_ref);
6583 }
6584
6585 static void
6586 set_repo_config_option(char *name, char *value, int (*cmd)(int, const char **))
6587 {
6588 const char *argv[SIZEOF_ARG] = { name, "=" };
6589 int argc = 1 + (cmd == option_set_command);
6590 int error = ERR;
6591
6592 if (!argv_from_string(argv, &argc, value))
6593 config_msg = "Too many option arguments";
6594 else
6595 error = cmd(argc, argv);
6596
6597 if (error == ERR)
6598 warn("Option 'tig.%s': %s", name, config_msg);
6599 }
6600
6601 static int
6602 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
6603 {
6604 if (!strcmp(name, "i18n.commitencoding"))
6605 string_ncopy(opt_encoding, value, valuelen);
6606
6607 if (!strcmp(name, "core.editor"))
6608 string_ncopy(opt_editor, value, valuelen);
6609
6610 if (!prefixcmp(name, "tig.color."))
6611 set_repo_config_option(name + 10, value, option_color_command);
6612
6613 else if (!prefixcmp(name, "tig.bind."))
6614 set_repo_config_option(name + 9, value, option_bind_command);
6615
6616 else if (!prefixcmp(name, "tig."))
6617 set_repo_config_option(name + 4, value, option_set_command);
6618
6619 /* branch.<head>.remote */
6620 if (*opt_head &&
6621 !strncmp(name, "branch.", 7) &&
6622 !strncmp(name + 7, opt_head, strlen(opt_head)) &&
6623 !strcmp(name + 7 + strlen(opt_head), ".remote"))
6624 string_ncopy(opt_remote, value, valuelen);
6625
6626 if (*opt_head && *opt_remote &&
6627 !strncmp(name, "branch.", 7) &&
6628 !strncmp(name + 7, opt_head, strlen(opt_head)) &&
6629 !strcmp(name + 7 + strlen(opt_head), ".merge")) {
6630 size_t from = strlen(opt_remote);
6631
6632 if (!prefixcmp(value, "refs/heads/")) {
6633 value += STRING_SIZE("refs/heads/");
6634 valuelen -= STRING_SIZE("refs/heads/");
6635 }
6636
6637 if (!string_format_from(opt_remote, &from, "/%s", value))
6638 opt_remote[0] = 0;
6639 }
6640
6641 return OK;
6642 }
6643
6644 static int
6645 load_git_config(void)
6646 {
6647 const char *config_list_argv[] = { "git", GIT_CONFIG, "--list", NULL };
6648
6649 return run_io_load(config_list_argv, "=", read_repo_config_option);
6650 }
6651
6652 static int
6653 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
6654 {
6655 if (!opt_git_dir[0]) {
6656 string_ncopy(opt_git_dir, name, namelen);
6657
6658 } else if (opt_is_inside_work_tree == -1) {
6659 /* This can be 3 different values depending on the
6660 * version of git being used. If git-rev-parse does not
6661 * understand --is-inside-work-tree it will simply echo
6662 * the option else either "true" or "false" is printed.
6663 * Default to true for the unknown case. */
6664 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
6665
6666 } else if (*name == '.') {
6667 string_ncopy(opt_cdup, name, namelen);
6668
6669 } else {
6670 string_ncopy(opt_prefix, name, namelen);
6671 }
6672
6673 return OK;
6674 }
6675
6676 static int
6677 load_repo_info(void)
6678 {
6679 const char *head_argv[] = {
6680 "git", "symbolic-ref", "HEAD", NULL
6681 };
6682 const char *rev_parse_argv[] = {
6683 "git", "rev-parse", "--git-dir", "--is-inside-work-tree",
6684 "--show-cdup", "--show-prefix", NULL
6685 };
6686
6687 if (run_io_buf(head_argv, opt_head, sizeof(opt_head))) {
6688 chomp_string(opt_head);
6689 if (!prefixcmp(opt_head, "refs/heads/")) {
6690 char *offset = opt_head + STRING_SIZE("refs/heads/");
6691
6692 memmove(opt_head, offset, strlen(offset) + 1);
6693 }
6694 }
6695
6696 return run_io_load(rev_parse_argv, "=", read_repo_info);
6697 }
6698
6699
6700 /*
6701 * Main
6702 */
6703
6704 static const char usage[] =
6705 "tig " TIG_VERSION " (" __DATE__ ")\n"
6706 "\n"
6707 "Usage: tig [options] [revs] [--] [paths]\n"
6708 " or: tig show [options] [revs] [--] [paths]\n"
6709 " or: tig blame [rev] path\n"
6710 " or: tig status\n"
6711 " or: tig < [git command output]\n"
6712 "\n"
6713 "Options:\n"
6714 " -v, --version Show version and exit\n"
6715 " -h, --help Show help message and exit";
6716
6717 static void __NORETURN
6718 quit(int sig)
6719 {
6720 /* XXX: Restore tty modes and let the OS cleanup the rest! */
6721 if (cursed)
6722 endwin();
6723 exit(0);
6724 }
6725
6726 static void __NORETURN
6727 die(const char *err, ...)
6728 {
6729 va_list args;
6730
6731 endwin();
6732
6733 va_start(args, err);
6734 fputs("tig: ", stderr);
6735 vfprintf(stderr, err, args);
6736 fputs("\n", stderr);
6737 va_end(args);
6738
6739 exit(1);
6740 }
6741
6742 static void
6743 warn(const char *msg, ...)
6744 {
6745 va_list args;
6746
6747 va_start(args, msg);
6748 fputs("tig warning: ", stderr);
6749 vfprintf(stderr, msg, args);
6750 fputs("\n", stderr);
6751 va_end(args);
6752 }
6753
6754 static enum request
6755 parse_options(int argc, const char *argv[])
6756 {
6757 enum request request = REQ_VIEW_MAIN;
6758 const char *subcommand;
6759 bool seen_dashdash = FALSE;
6760 /* XXX: This is vulnerable to the user overriding options
6761 * required for the main view parser. */
6762 const char *custom_argv[SIZEOF_ARG] = {
6763 "git", "log", "--no-color", "--pretty=raw", "--parents",
6764 "--topo-order", NULL
6765 };
6766 int i, j = 6;
6767
6768 if (!isatty(STDIN_FILENO)) {
6769 io_open(&VIEW(REQ_VIEW_PAGER)->io, "");
6770 return REQ_VIEW_PAGER;
6771 }
6772
6773 if (argc <= 1)
6774 return REQ_NONE;
6775
6776 subcommand = argv[1];
6777 if (!strcmp(subcommand, "status")) {
6778 if (argc > 2)
6779 warn("ignoring arguments after `%s'", subcommand);
6780 return REQ_VIEW_STATUS;
6781
6782 } else if (!strcmp(subcommand, "blame")) {
6783 if (argc <= 2 || argc > 4)
6784 die("invalid number of options to blame\n\n%s", usage);
6785
6786 i = 2;
6787 if (argc == 4) {
6788 string_ncopy(opt_ref, argv[i], strlen(argv[i]));
6789 i++;
6790 }
6791
6792 string_ncopy(opt_file, argv[i], strlen(argv[i]));
6793 return REQ_VIEW_BLAME;
6794
6795 } else if (!strcmp(subcommand, "show")) {
6796 request = REQ_VIEW_DIFF;
6797
6798 } else {
6799 subcommand = NULL;
6800 }
6801
6802 if (subcommand) {
6803 custom_argv[1] = subcommand;
6804 j = 2;
6805 }
6806
6807 for (i = 1 + !!subcommand; i < argc; i++) {
6808 const char *opt = argv[i];
6809
6810 if (seen_dashdash || !strcmp(opt, "--")) {
6811 seen_dashdash = TRUE;
6812
6813 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
6814 printf("tig version %s\n", TIG_VERSION);
6815 quit(0);
6816
6817 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
6818 printf("%s\n", usage);
6819 quit(0);
6820 }
6821
6822 custom_argv[j++] = opt;
6823 if (j >= ARRAY_SIZE(custom_argv))
6824 die("command too long");
6825 }
6826
6827 if (!prepare_update(VIEW(request), custom_argv, NULL, FORMAT_NONE))
6828 die("Failed to format arguments");
6829
6830 return request;
6831 }
6832
6833 int
6834 main(int argc, const char *argv[])
6835 {
6836 enum request request = parse_options(argc, argv);
6837 struct view *view;
6838 size_t i;
6839
6840 signal(SIGINT, quit);
6841
6842 if (setlocale(LC_ALL, "")) {
6843 char *codeset = nl_langinfo(CODESET);
6844
6845 string_ncopy(opt_codeset, codeset, strlen(codeset));
6846 }
6847
6848 if (load_repo_info() == ERR)
6849 die("Failed to load repo info.");
6850
6851 if (load_options() == ERR)
6852 die("Failed to load user config.");
6853
6854 if (load_git_config() == ERR)
6855 die("Failed to load repo config.");
6856
6857 /* Require a git repository unless when running in pager mode. */
6858 if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
6859 die("Not a git repository");
6860
6861 if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
6862 opt_utf8 = FALSE;
6863
6864 if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
6865 opt_iconv = iconv_open(opt_codeset, opt_encoding);
6866 if (opt_iconv == ICONV_NONE)
6867 die("Failed to initialize character set conversion");
6868 }
6869
6870 if (load_refs() == ERR)
6871 die("Failed to load refs.");
6872
6873 foreach_view (view, i)
6874 argv_from_env(view->ops->argv, view->cmd_env);
6875
6876 init_display();
6877
6878 if (request != REQ_NONE)
6879 open_view(NULL, request, OPEN_PREPARED);
6880 request = request == REQ_NONE ? REQ_VIEW_MAIN : REQ_NONE;
6881
6882 while (view_driver(display[current_view], request)) {
6883 int key = get_input(0);
6884
6885 view = display[current_view];
6886 request = get_keybinding(view->keymap, key);
6887
6888 /* Some low-level request handling. This keeps access to
6889 * status_win restricted. */
6890 switch (request) {
6891 case REQ_PROMPT:
6892 {
6893 char *cmd = read_prompt(":");
6894
6895 if (cmd) {
6896 struct view *next = VIEW(REQ_VIEW_PAGER);
6897 const char *argv[SIZEOF_ARG] = { "git" };
6898 int argc = 1;
6899
6900 /* When running random commands, initially show the
6901 * command in the title. However, it maybe later be
6902 * overwritten if a commit line is selected. */
6903 string_ncopy(next->ref, cmd, strlen(cmd));
6904
6905 if (!argv_from_string(argv, &argc, cmd)) {
6906 report("Too many arguments");
6907 } else if (!prepare_update(next, argv, NULL, FORMAT_DASH)) {
6908 report("Failed to format command");
6909 } else {
6910 open_view(view, REQ_VIEW_PAGER, OPEN_PREPARED);
6911 }
6912 }
6913
6914 request = REQ_NONE;
6915 break;
6916 }
6917 case REQ_SEARCH:
6918 case REQ_SEARCH_BACK:
6919 {
6920 const char *prompt = request == REQ_SEARCH ? "/" : "?";
6921 char *search = read_prompt(prompt);
6922
6923 if (search)
6924 string_ncopy(opt_search, search, strlen(search));
6925 else if (*opt_search)
6926 request = request == REQ_SEARCH ?
6927 REQ_FIND_NEXT :
6928 REQ_FIND_PREV;
6929 else
6930 request = REQ_NONE;
6931 break;
6932 }
6933 default:
6934 break;
6935 }
6936 }
6937
6938 quit(0);
6939
6940 return 0;
6941 }
Text-mode interface for git