Fix redrawing of old current line
[tig] / tig.c
1 /* Copyright (c) 2006 Jonas Fonseca <fonseca@diku.dk>
2  *
3  * This program is free software; you can redistribute it and/or
4  * modify it under the terms of the GNU General Public License as
5  * published by the Free Software Foundation; either version 2 of
6  * the License, or (at your option) any later version.
7  *
8  * This program is distributed in the hope that it will be useful,
9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11  * GNU General Public License for more details.
12  */
13
14 #ifndef VERSION
15 #define VERSION "tig-0.4.git"
16 #endif
17
18 #ifndef DEBUG
19 #define NDEBUG
20 #endif
21
22 #include <assert.h>
23 #include <errno.h>
24 #include <ctype.h>
25 #include <signal.h>
26 #include <stdarg.h>
27 #include <stdio.h>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <unistd.h>
31 #include <time.h>
32
33 #include <sys/types.h>
34 #include <regex.h>
35
36 #include <locale.h>
37 #include <langinfo.h>
38 #include <iconv.h>
39
40 #include <curses.h>
41
42 #if __GNUC__ >= 3
43 #define __NORETURN __attribute__((__noreturn__))
44 #else
45 #define __NORETURN
46 #endif
47
48 static void __NORETURN die(const char *err, ...);
49 static void report(const char *msg, ...);
50 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, int, char *, int));
51 static void set_nonblocking_input(bool loading);
52 static size_t utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed);
53
54 #define ABS(x)          ((x) >= 0  ? (x) : -(x))
55 #define MIN(x, y)       ((x) < (y) ? (x) :  (y))
56
57 #define ARRAY_SIZE(x)   (sizeof(x) / sizeof(x[0]))
58 #define STRING_SIZE(x)  (sizeof(x) - 1)
59
60 #define SIZEOF_STR      1024    /* Default string size. */
61 #define SIZEOF_REF      256     /* Size of symbolic or SHA1 ID. */
62 #define SIZEOF_REVGRAPH 19      /* Size of revision ancestry graphics. */
63
64 /* This color name can be used to refer to the default term colors. */
65 #define COLOR_DEFAULT   (-1)
66
67 #define ICONV_NONE      ((iconv_t) -1)
68
69 /* The format and size of the date column in the main view. */
70 #define DATE_FORMAT     "%Y-%m-%d %H:%M"
71 #define DATE_COLS       STRING_SIZE("2006-04-29 14:21 ")
72
73 #define AUTHOR_COLS     20
74
75 /* The default interval between line numbers. */
76 #define NUMBER_INTERVAL 1
77
78 #define TABSIZE         8
79
80 #define SCALE_SPLIT_VIEW(height)        ((height) * 2 / 3)
81
82 #define TIG_LS_REMOTE \
83         "git ls-remote . 2>/dev/null"
84
85 #define TIG_DIFF_CMD \
86         "git show --root --patch-with-stat --find-copies-harder -B -C %s 2>/dev/null"
87
88 #define TIG_LOG_CMD     \
89         "git log --cc --stat -n100 %s 2>/dev/null"
90
91 #define TIG_MAIN_CMD \
92         "git log --topo-order --pretty=raw %s 2>/dev/null"
93
94 #define TIG_TREE_CMD    \
95         "git ls-tree %s %s"
96
97 #define TIG_BLOB_CMD    \
98         "git cat-file blob %s"
99
100 /* XXX: Needs to be defined to the empty string. */
101 #define TIG_HELP_CMD    ""
102 #define TIG_PAGER_CMD   ""
103
104 /* Some ascii-shorthands fitted into the ncurses namespace. */
105 #define KEY_TAB         '\t'
106 #define KEY_RETURN      '\r'
107 #define KEY_ESC         27
108
109
110 struct ref {
111         char *name;             /* Ref name; tag or head names are shortened. */
112         char id[41];            /* Commit SHA1 ID */
113         unsigned int tag:1;     /* Is it a tag? */
114         unsigned int next:1;    /* For ref lists: are there more refs? */
115 };
116
117 static struct ref **get_refs(char *id);
118
119 struct int_map {
120         const char *name;
121         int namelen;
122         int value;
123 };
124
125 static int
126 set_from_int_map(struct int_map *map, size_t map_size,
127                  int *value, const char *name, int namelen)
128 {
129
130         int i;
131
132         for (i = 0; i < map_size; i++)
133                 if (namelen == map[i].namelen &&
134                     !strncasecmp(name, map[i].name, namelen)) {
135                         *value = map[i].value;
136                         return OK;
137                 }
138
139         return ERR;
140 }
141
142
143 /*
144  * String helpers
145  */
146
147 static inline void
148 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
149 {
150         if (srclen > dstlen - 1)
151                 srclen = dstlen - 1;
152
153         strncpy(dst, src, srclen);
154         dst[srclen] = 0;
155 }
156
157 /* Shorthands for safely copying into a fixed buffer. */
158
159 #define string_copy(dst, src) \
160         string_ncopy_do(dst, sizeof(dst), src, sizeof(dst))
161
162 #define string_ncopy(dst, src, srclen) \
163         string_ncopy_do(dst, sizeof(dst), src, srclen)
164
165 static char *
166 chomp_string(char *name)
167 {
168         int namelen;
169
170         while (isspace(*name))
171                 name++;
172
173         namelen = strlen(name) - 1;
174         while (namelen > 0 && isspace(name[namelen]))
175                 name[namelen--] = 0;
176
177         return name;
178 }
179
180 static bool
181 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
182 {
183         va_list args;
184         size_t pos = bufpos ? *bufpos : 0;
185
186         va_start(args, fmt);
187         pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
188         va_end(args);
189
190         if (bufpos)
191                 *bufpos = pos;
192
193         return pos >= bufsize ? FALSE : TRUE;
194 }
195
196 #define string_format(buf, fmt, args...) \
197         string_nformat(buf, sizeof(buf), NULL, fmt, args)
198
199 #define string_format_from(buf, from, fmt, args...) \
200         string_nformat(buf, sizeof(buf), from, fmt, args)
201
202 static int
203 string_enum_compare(const char *str1, const char *str2, int len)
204 {
205         size_t i;
206
207 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
208
209         /* Diff-Header == DIFF_HEADER */
210         for (i = 0; i < len; i++) {
211                 if (toupper(str1[i]) == toupper(str2[i]))
212                         continue;
213
214                 if (string_enum_sep(str1[i]) &&
215                     string_enum_sep(str2[i]))
216                         continue;
217
218                 return str1[i] - str2[i];
219         }
220
221         return 0;
222 }
223
224 /* Shell quoting
225  *
226  * NOTE: The following is a slightly modified copy of the git project's shell
227  * quoting routines found in the quote.c file.
228  *
229  * Help to copy the thing properly quoted for the shell safety.  any single
230  * quote is replaced with '\'', any exclamation point is replaced with '\!',
231  * and the whole thing is enclosed in a
232  *
233  * E.g.
234  *  original     sq_quote     result
235  *  name     ==> name      ==> 'name'
236  *  a b      ==> a b       ==> 'a b'
237  *  a'b      ==> a'\''b    ==> 'a'\''b'
238  *  a!b      ==> a'\!'b    ==> 'a'\!'b'
239  */
240
241 static size_t
242 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
243 {
244         char c;
245
246 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
247
248         BUFPUT('\'');
249         while ((c = *src++)) {
250                 if (c == '\'' || c == '!') {
251                         BUFPUT('\'');
252                         BUFPUT('\\');
253                         BUFPUT(c);
254                         BUFPUT('\'');
255                 } else {
256                         BUFPUT(c);
257                 }
258         }
259         BUFPUT('\'');
260
261         return bufsize;
262 }
263
264
265 /*
266  * User requests
267  */
268
269 #define REQ_INFO \
270         /* XXX: Keep the view request first and in sync with views[]. */ \
271         REQ_GROUP("View switching") \
272         REQ_(VIEW_MAIN,         "Show main view"), \
273         REQ_(VIEW_DIFF,         "Show diff view"), \
274         REQ_(VIEW_LOG,          "Show log view"), \
275         REQ_(VIEW_TREE,         "Show tree view"), \
276         REQ_(VIEW_BLOB,         "Show blob view"), \
277         REQ_(VIEW_HELP,         "Show help page"), \
278         REQ_(VIEW_PAGER,        "Show pager view"), \
279         \
280         REQ_GROUP("View manipulation") \
281         REQ_(ENTER,             "Enter current line and scroll"), \
282         REQ_(NEXT,              "Move to next"), \
283         REQ_(PREVIOUS,          "Move to previous"), \
284         REQ_(VIEW_NEXT,         "Move focus to next view"), \
285         REQ_(VIEW_CLOSE,        "Close the current view"), \
286         REQ_(QUIT,              "Close all views and quit"), \
287         \
288         REQ_GROUP("Cursor navigation") \
289         REQ_(MOVE_UP,           "Move cursor one line up"), \
290         REQ_(MOVE_DOWN,         "Move cursor one line down"), \
291         REQ_(MOVE_PAGE_DOWN,    "Move cursor one page down"), \
292         REQ_(MOVE_PAGE_UP,      "Move cursor one page up"), \
293         REQ_(MOVE_FIRST_LINE,   "Move cursor to first line"), \
294         REQ_(MOVE_LAST_LINE,    "Move cursor to last line"), \
295         \
296         REQ_GROUP("Scrolling") \
297         REQ_(SCROLL_LINE_UP,    "Scroll one line up"), \
298         REQ_(SCROLL_LINE_DOWN,  "Scroll one line down"), \
299         REQ_(SCROLL_PAGE_UP,    "Scroll one page up"), \
300         REQ_(SCROLL_PAGE_DOWN,  "Scroll one page down"), \
301         \
302         REQ_GROUP("Searching") \
303         REQ_(SEARCH,            "Search the view"), \
304         REQ_(SEARCH_BACK,       "Search backwards in the view"), \
305         REQ_(FIND_NEXT,         "Find next search match"), \
306         REQ_(FIND_PREV,         "Find previous search match"), \
307         \
308         REQ_GROUP("Misc") \
309         REQ_(NONE,              "Do nothing"), \
310         REQ_(PROMPT,            "Bring up the prompt"), \
311         REQ_(SCREEN_REDRAW,     "Redraw the screen"), \
312         REQ_(SCREEN_RESIZE,     "Resize the screen"), \
313         REQ_(SHOW_VERSION,      "Show version information"), \
314         REQ_(STOP_LOADING,      "Stop all loading views"), \
315         REQ_(TOGGLE_LINENO,     "Toggle line numbers"), \
316         REQ_(TOGGLE_REV_GRAPH,  "Toggle revision graph visualization")
317
318
319 /* User action requests. */
320 enum request {
321 #define REQ_GROUP(help)
322 #define REQ_(req, help) REQ_##req
323
324         /* Offset all requests to avoid conflicts with ncurses getch values. */
325         REQ_OFFSET = KEY_MAX + 1,
326         REQ_INFO,
327         REQ_UNKNOWN,
328
329 #undef  REQ_GROUP
330 #undef  REQ_
331 };
332
333 struct request_info {
334         enum request request;
335         char *name;
336         int namelen;
337         char *help;
338 };
339
340 static struct request_info req_info[] = {
341 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
342 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
343         REQ_INFO
344 #undef  REQ_GROUP
345 #undef  REQ_
346 };
347
348 static enum request
349 get_request(const char *name)
350 {
351         int namelen = strlen(name);
352         int i;
353
354         for (i = 0; i < ARRAY_SIZE(req_info); i++)
355                 if (req_info[i].namelen == namelen &&
356                     !string_enum_compare(req_info[i].name, name, namelen))
357                         return req_info[i].request;
358
359         return REQ_UNKNOWN;
360 }
361
362
363 /*
364  * Options
365  */
366
367 static const char usage[] =
368 VERSION " (" __DATE__ ")\n"
369 "\n"
370 "Usage: tig [options]\n"
371 "   or: tig [options] [--] [git log options]\n"
372 "   or: tig [options] log  [git log options]\n"
373 "   or: tig [options] diff [git diff options]\n"
374 "   or: tig [options] show [git show options]\n"
375 "   or: tig [options] <    [git command output]\n"
376 "\n"
377 "Options:\n"
378 "  -l                          Start up in log view\n"
379 "  -d                          Start up in diff view\n"
380 "  -n[I], --line-number[=I]    Show line numbers with given interval\n"
381 "  -b[N], --tab-size[=N]       Set number of spaces for tab expansion\n"
382 "  --                          Mark end of tig options\n"
383 "  -v, --version               Show version and exit\n"
384 "  -h, --help                  Show help message and exit\n";
385
386 /* Option and state variables. */
387 static bool opt_line_number             = FALSE;
388 static bool opt_rev_graph               = TRUE;
389 static int opt_num_interval             = NUMBER_INTERVAL;
390 static int opt_tab_size                 = TABSIZE;
391 static enum request opt_request         = REQ_VIEW_MAIN;
392 static char opt_cmd[SIZEOF_STR]         = "";
393 static char opt_path[SIZEOF_STR]        = "";
394 static FILE *opt_pipe                   = NULL;
395 static char opt_encoding[20]            = "UTF-8";
396 static bool opt_utf8                    = TRUE;
397 static char opt_codeset[20]             = "UTF-8";
398 static iconv_t opt_iconv                = ICONV_NONE;
399 static char opt_search[SIZEOF_STR]      = "";
400
401 enum option_type {
402         OPT_NONE,
403         OPT_INT,
404 };
405
406 static bool
407 check_option(char *opt, char short_name, char *name, enum option_type type, ...)
408 {
409         va_list args;
410         char *value = "";
411         int *number;
412
413         if (opt[0] != '-')
414                 return FALSE;
415
416         if (opt[1] == '-') {
417                 int namelen = strlen(name);
418
419                 opt += 2;
420
421                 if (strncmp(opt, name, namelen))
422                         return FALSE;
423
424                 if (opt[namelen] == '=')
425                         value = opt + namelen + 1;
426
427         } else {
428                 if (!short_name || opt[1] != short_name)
429                         return FALSE;
430                 value = opt + 2;
431         }
432
433         va_start(args, type);
434         if (type == OPT_INT) {
435                 number = va_arg(args, int *);
436                 if (isdigit(*value))
437                         *number = atoi(value);
438         }
439         va_end(args);
440
441         return TRUE;
442 }
443
444 /* Returns the index of log or diff command or -1 to exit. */
445 static bool
446 parse_options(int argc, char *argv[])
447 {
448         int i;
449
450         for (i = 1; i < argc; i++) {
451                 char *opt = argv[i];
452
453                 if (!strcmp(opt, "-l")) {
454                         opt_request = REQ_VIEW_LOG;
455                         continue;
456                 }
457
458                 if (!strcmp(opt, "-d")) {
459                         opt_request = REQ_VIEW_DIFF;
460                         continue;
461                 }
462
463                 if (check_option(opt, 'n', "line-number", OPT_INT, &opt_num_interval)) {
464                         opt_line_number = TRUE;
465                         continue;
466                 }
467
468                 if (check_option(opt, 'b', "tab-size", OPT_INT, &opt_tab_size)) {
469                         opt_tab_size = MIN(opt_tab_size, TABSIZE);
470                         continue;
471                 }
472
473                 if (check_option(opt, 'v', "version", OPT_NONE)) {
474                         printf("tig version %s\n", VERSION);
475                         return FALSE;
476                 }
477
478                 if (check_option(opt, 'h', "help", OPT_NONE)) {
479                         printf(usage);
480                         return FALSE;
481                 }
482
483                 if (!strcmp(opt, "--")) {
484                         i++;
485                         break;
486                 }
487
488                 if (!strcmp(opt, "log") ||
489                     !strcmp(opt, "diff") ||
490                     !strcmp(opt, "show")) {
491                         opt_request = opt[0] == 'l'
492                                     ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
493                         break;
494                 }
495
496                 if (opt[0] && opt[0] != '-')
497                         break;
498
499                 die("unknown option '%s'\n\n%s", opt, usage);
500         }
501
502         if (!isatty(STDIN_FILENO)) {
503                 opt_request = REQ_VIEW_PAGER;
504                 opt_pipe = stdin;
505
506         } else if (i < argc) {
507                 size_t buf_size;
508
509                 if (opt_request == REQ_VIEW_MAIN)
510                         /* XXX: This is vulnerable to the user overriding
511                          * options required for the main view parser. */
512                         string_copy(opt_cmd, "git log --stat --pretty=raw");
513                 else
514                         string_copy(opt_cmd, "git");
515                 buf_size = strlen(opt_cmd);
516
517                 while (buf_size < sizeof(opt_cmd) && i < argc) {
518                         opt_cmd[buf_size++] = ' ';
519                         buf_size = sq_quote(opt_cmd, buf_size, argv[i++]);
520                 }
521
522                 if (buf_size >= sizeof(opt_cmd))
523                         die("command too long");
524
525                 opt_cmd[buf_size] = 0;
526
527         }
528
529         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
530                 opt_utf8 = FALSE;
531
532         return TRUE;
533 }
534
535
536 /*
537  * Line-oriented content detection.
538  */
539
540 #define LINE_INFO \
541 LINE(DIFF_HEADER,  "diff --git ",       COLOR_YELLOW,   COLOR_DEFAULT,  0), \
542 LINE(DIFF_CHUNK,   "@@",                COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
543 LINE(DIFF_ADD,     "+",                 COLOR_GREEN,    COLOR_DEFAULT,  0), \
544 LINE(DIFF_DEL,     "-",                 COLOR_RED,      COLOR_DEFAULT,  0), \
545 LINE(DIFF_INDEX,        "index ",         COLOR_BLUE,   COLOR_DEFAULT,  0), \
546 LINE(DIFF_OLDMODE,      "old file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
547 LINE(DIFF_NEWMODE,      "new file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
548 LINE(DIFF_COPY_FROM,    "copy from",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
549 LINE(DIFF_COPY_TO,      "copy to",        COLOR_YELLOW, COLOR_DEFAULT,  0), \
550 LINE(DIFF_RENAME_FROM,  "rename from",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
551 LINE(DIFF_RENAME_TO,    "rename to",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
552 LINE(DIFF_SIMILARITY,   "similarity ",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
553 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
554 LINE(DIFF_TREE,         "diff-tree ",     COLOR_BLUE,   COLOR_DEFAULT,  0), \
555 LINE(PP_AUTHOR,    "Author: ",          COLOR_CYAN,     COLOR_DEFAULT,  0), \
556 LINE(PP_COMMIT,    "Commit: ",          COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
557 LINE(PP_MERGE,     "Merge: ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
558 LINE(PP_DATE,      "Date:   ",          COLOR_YELLOW,   COLOR_DEFAULT,  0), \
559 LINE(PP_ADATE,     "AuthorDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
560 LINE(PP_CDATE,     "CommitDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
561 LINE(PP_REFS,      "Refs: ",            COLOR_RED,      COLOR_DEFAULT,  0), \
562 LINE(COMMIT,       "commit ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
563 LINE(PARENT,       "parent ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
564 LINE(TREE,         "tree ",             COLOR_BLUE,     COLOR_DEFAULT,  0), \
565 LINE(AUTHOR,       "author ",           COLOR_CYAN,     COLOR_DEFAULT,  0), \
566 LINE(COMMITTER,    "committer ",        COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
567 LINE(SIGNOFF,      "    Signed-off-by", COLOR_YELLOW,   COLOR_DEFAULT,  0), \
568 LINE(DEFAULT,      "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
569 LINE(CURSOR,       "",                  COLOR_WHITE,    COLOR_GREEN,    A_BOLD), \
570 LINE(STATUS,       "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
571 LINE(TITLE_BLUR,   "",                  COLOR_WHITE,    COLOR_BLUE,     0), \
572 LINE(TITLE_FOCUS,  "",                  COLOR_WHITE,    COLOR_BLUE,     A_BOLD), \
573 LINE(MAIN_DATE,    "",                  COLOR_BLUE,     COLOR_DEFAULT,  0), \
574 LINE(MAIN_AUTHOR,  "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
575 LINE(MAIN_COMMIT,  "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
576 LINE(MAIN_DELIM,   "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
577 LINE(MAIN_TAG,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  A_BOLD), \
578 LINE(MAIN_REF,     "",                  COLOR_CYAN,     COLOR_DEFAULT,  A_BOLD), \
579 LINE(TREE_DIR,     "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
580 LINE(TREE_FILE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL)
581
582 enum line_type {
583 #define LINE(type, line, fg, bg, attr) \
584         LINE_##type
585         LINE_INFO
586 #undef  LINE
587 };
588
589 struct line_info {
590         const char *name;       /* Option name. */
591         int namelen;            /* Size of option name. */
592         const char *line;       /* The start of line to match. */
593         int linelen;            /* Size of string to match. */
594         int fg, bg, attr;       /* Color and text attributes for the lines. */
595 };
596
597 static struct line_info line_info[] = {
598 #define LINE(type, line, fg, bg, attr) \
599         { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
600         LINE_INFO
601 #undef  LINE
602 };
603
604 static enum line_type
605 get_line_type(char *line)
606 {
607         int linelen = strlen(line);
608         enum line_type type;
609
610         for (type = 0; type < ARRAY_SIZE(line_info); type++)
611                 /* Case insensitive search matches Signed-off-by lines better. */
612                 if (linelen >= line_info[type].linelen &&
613                     !strncasecmp(line_info[type].line, line, line_info[type].linelen))
614                         return type;
615
616         return LINE_DEFAULT;
617 }
618
619 static inline int
620 get_line_attr(enum line_type type)
621 {
622         assert(type < ARRAY_SIZE(line_info));
623         return COLOR_PAIR(type) | line_info[type].attr;
624 }
625
626 static struct line_info *
627 get_line_info(char *name, int namelen)
628 {
629         enum line_type type;
630
631         for (type = 0; type < ARRAY_SIZE(line_info); type++)
632                 if (namelen == line_info[type].namelen &&
633                     !string_enum_compare(line_info[type].name, name, namelen))
634                         return &line_info[type];
635
636         return NULL;
637 }
638
639 static void
640 init_colors(void)
641 {
642         int default_bg = COLOR_BLACK;
643         int default_fg = COLOR_WHITE;
644         enum line_type type;
645
646         start_color();
647
648         if (use_default_colors() != ERR) {
649                 default_bg = -1;
650                 default_fg = -1;
651         }
652
653         for (type = 0; type < ARRAY_SIZE(line_info); type++) {
654                 struct line_info *info = &line_info[type];
655                 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
656                 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
657
658                 init_pair(type, fg, bg);
659         }
660 }
661
662 struct line {
663         enum line_type type;
664
665         /* State flags */
666         unsigned int selected:1;
667
668         void *data;             /* User data */
669 };
670
671
672 /*
673  * Keys
674  */
675
676 struct keybinding {
677         int alias;
678         enum request request;
679         struct keybinding *next;
680 };
681
682 static struct keybinding default_keybindings[] = {
683         /* View switching */
684         { 'm',          REQ_VIEW_MAIN },
685         { 'd',          REQ_VIEW_DIFF },
686         { 'l',          REQ_VIEW_LOG },
687         { 't',          REQ_VIEW_TREE },
688         { 'f',          REQ_VIEW_BLOB },
689         { 'p',          REQ_VIEW_PAGER },
690         { 'h',          REQ_VIEW_HELP },
691
692         /* View manipulation */
693         { 'q',          REQ_VIEW_CLOSE },
694         { KEY_TAB,      REQ_VIEW_NEXT },
695         { KEY_RETURN,   REQ_ENTER },
696         { KEY_UP,       REQ_PREVIOUS },
697         { KEY_DOWN,     REQ_NEXT },
698
699         /* Cursor navigation */
700         { 'k',          REQ_MOVE_UP },
701         { 'j',          REQ_MOVE_DOWN },
702         { KEY_HOME,     REQ_MOVE_FIRST_LINE },
703         { KEY_END,      REQ_MOVE_LAST_LINE },
704         { KEY_NPAGE,    REQ_MOVE_PAGE_DOWN },
705         { ' ',          REQ_MOVE_PAGE_DOWN },
706         { KEY_PPAGE,    REQ_MOVE_PAGE_UP },
707         { 'b',          REQ_MOVE_PAGE_UP },
708         { '-',          REQ_MOVE_PAGE_UP },
709
710         /* Scrolling */
711         { KEY_IC,       REQ_SCROLL_LINE_UP },
712         { KEY_DC,       REQ_SCROLL_LINE_DOWN },
713         { 'w',          REQ_SCROLL_PAGE_UP },
714         { 's',          REQ_SCROLL_PAGE_DOWN },
715
716         /* Searching */
717         { '/',          REQ_SEARCH },
718         { '?',          REQ_SEARCH_BACK },
719         { 'n',          REQ_FIND_NEXT },
720         { 'N',          REQ_FIND_PREV },
721
722         /* Misc */
723         { 'Q',          REQ_QUIT },
724         { 'z',          REQ_STOP_LOADING },
725         { 'v',          REQ_SHOW_VERSION },
726         { 'r',          REQ_SCREEN_REDRAW },
727         { '.',          REQ_TOGGLE_LINENO },
728         { 'g',          REQ_TOGGLE_REV_GRAPH },
729         { ':',          REQ_PROMPT },
730
731         /* wgetch() with nodelay() enabled returns ERR when there's no input. */
732         { ERR,          REQ_NONE },
733
734         /* Using the ncurses SIGWINCH handler. */
735         { KEY_RESIZE,   REQ_SCREEN_RESIZE },
736 };
737
738 #define KEYMAP_INFO \
739         KEYMAP_(GENERIC), \
740         KEYMAP_(MAIN), \
741         KEYMAP_(DIFF), \
742         KEYMAP_(LOG), \
743         KEYMAP_(TREE), \
744         KEYMAP_(BLOB), \
745         KEYMAP_(PAGER), \
746         KEYMAP_(HELP) \
747
748 enum keymap {
749 #define KEYMAP_(name) KEYMAP_##name
750         KEYMAP_INFO
751 #undef  KEYMAP_
752 };
753
754 static struct int_map keymap_table[] = {
755 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
756         KEYMAP_INFO
757 #undef  KEYMAP_
758 };
759
760 #define set_keymap(map, name) \
761         set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
762
763 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
764
765 static void
766 add_keybinding(enum keymap keymap, enum request request, int key)
767 {
768         struct keybinding *keybinding;
769
770         keybinding = calloc(1, sizeof(*keybinding));
771         if (!keybinding)
772                 die("Failed to allocate keybinding");
773
774         keybinding->alias = key;
775         keybinding->request = request;
776         keybinding->next = keybindings[keymap];
777         keybindings[keymap] = keybinding;
778 }
779
780 /* Looks for a key binding first in the given map, then in the generic map, and
781  * lastly in the default keybindings. */
782 static enum request
783 get_keybinding(enum keymap keymap, int key)
784 {
785         struct keybinding *kbd;
786         int i;
787
788         for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
789                 if (kbd->alias == key)
790                         return kbd->request;
791
792         for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
793                 if (kbd->alias == key)
794                         return kbd->request;
795
796         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
797                 if (default_keybindings[i].alias == key)
798                         return default_keybindings[i].request;
799
800         return (enum request) key;
801 }
802
803
804 struct key {
805         char *name;
806         int value;
807 };
808
809 static struct key key_table[] = {
810         { "Enter",      KEY_RETURN },
811         { "Space",      ' ' },
812         { "Backspace",  KEY_BACKSPACE },
813         { "Tab",        KEY_TAB },
814         { "Escape",     KEY_ESC },
815         { "Left",       KEY_LEFT },
816         { "Right",      KEY_RIGHT },
817         { "Up",         KEY_UP },
818         { "Down",       KEY_DOWN },
819         { "Insert",     KEY_IC },
820         { "Delete",     KEY_DC },
821         { "Hash",       '#' },
822         { "Home",       KEY_HOME },
823         { "End",        KEY_END },
824         { "PageUp",     KEY_PPAGE },
825         { "PageDown",   KEY_NPAGE },
826         { "F1",         KEY_F(1) },
827         { "F2",         KEY_F(2) },
828         { "F3",         KEY_F(3) },
829         { "F4",         KEY_F(4) },
830         { "F5",         KEY_F(5) },
831         { "F6",         KEY_F(6) },
832         { "F7",         KEY_F(7) },
833         { "F8",         KEY_F(8) },
834         { "F9",         KEY_F(9) },
835         { "F10",        KEY_F(10) },
836         { "F11",        KEY_F(11) },
837         { "F12",        KEY_F(12) },
838 };
839
840 static int
841 get_key_value(const char *name)
842 {
843         int i;
844
845         for (i = 0; i < ARRAY_SIZE(key_table); i++)
846                 if (!strcasecmp(key_table[i].name, name))
847                         return key_table[i].value;
848
849         if (strlen(name) == 1 && isprint(*name))
850                 return (int) *name;
851
852         return ERR;
853 }
854
855 static char *
856 get_key(enum request request)
857 {
858         static char buf[BUFSIZ];
859         static char key_char[] = "'X'";
860         size_t pos = 0;
861         char *sep = "    ";
862         int i;
863
864         buf[pos] = 0;
865
866         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
867                 struct keybinding *keybinding = &default_keybindings[i];
868                 char *seq = NULL;
869                 int key;
870
871                 if (keybinding->request != request)
872                         continue;
873
874                 for (key = 0; key < ARRAY_SIZE(key_table); key++)
875                         if (key_table[key].value == keybinding->alias)
876                                 seq = key_table[key].name;
877
878                 if (seq == NULL &&
879                     keybinding->alias < 127 &&
880                     isprint(keybinding->alias)) {
881                         key_char[1] = (char) keybinding->alias;
882                         seq = key_char;
883                 }
884
885                 if (!seq)
886                         seq = "'?'";
887
888                 if (!string_format_from(buf, &pos, "%s%s", sep, seq))
889                         return "Too many keybindings!";
890                 sep = ", ";
891         }
892
893         return buf;
894 }
895
896
897 /*
898  * User config file handling.
899  */
900
901 static struct int_map color_map[] = {
902 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
903         COLOR_MAP(DEFAULT),
904         COLOR_MAP(BLACK),
905         COLOR_MAP(BLUE),
906         COLOR_MAP(CYAN),
907         COLOR_MAP(GREEN),
908         COLOR_MAP(MAGENTA),
909         COLOR_MAP(RED),
910         COLOR_MAP(WHITE),
911         COLOR_MAP(YELLOW),
912 };
913
914 #define set_color(color, name) \
915         set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
916
917 static struct int_map attr_map[] = {
918 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
919         ATTR_MAP(NORMAL),
920         ATTR_MAP(BLINK),
921         ATTR_MAP(BOLD),
922         ATTR_MAP(DIM),
923         ATTR_MAP(REVERSE),
924         ATTR_MAP(STANDOUT),
925         ATTR_MAP(UNDERLINE),
926 };
927
928 #define set_attribute(attr, name) \
929         set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
930
931 static int   config_lineno;
932 static bool  config_errors;
933 static char *config_msg;
934
935 /* Wants: object fgcolor bgcolor [attr] */
936 static int
937 option_color_command(int argc, char *argv[])
938 {
939         struct line_info *info;
940
941         if (argc != 3 && argc != 4) {
942                 config_msg = "Wrong number of arguments given to color command";
943                 return ERR;
944         }
945
946         info = get_line_info(argv[0], strlen(argv[0]));
947         if (!info) {
948                 config_msg = "Unknown color name";
949                 return ERR;
950         }
951
952         if (set_color(&info->fg, argv[1]) == ERR ||
953             set_color(&info->bg, argv[2]) == ERR) {
954                 config_msg = "Unknown color";
955                 return ERR;
956         }
957
958         if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
959                 config_msg = "Unknown attribute";
960                 return ERR;
961         }
962
963         return OK;
964 }
965
966 /* Wants: name = value */
967 static int
968 option_set_command(int argc, char *argv[])
969 {
970         if (argc != 3) {
971                 config_msg = "Wrong number of arguments given to set command";
972                 return ERR;
973         }
974
975         if (strcmp(argv[1], "=")) {
976                 config_msg = "No value assigned";
977                 return ERR;
978         }
979
980         if (!strcmp(argv[0], "show-rev-graph")) {
981                 opt_rev_graph = (!strcmp(argv[2], "1") ||
982                                  !strcmp(argv[2], "true") ||
983                                  !strcmp(argv[2], "yes"));
984                 return OK;
985         }
986
987         if (!strcmp(argv[0], "line-number-interval")) {
988                 opt_num_interval = atoi(argv[2]);
989                 return OK;
990         }
991
992         if (!strcmp(argv[0], "tab-size")) {
993                 opt_tab_size = atoi(argv[2]);
994                 return OK;
995         }
996
997         if (!strcmp(argv[0], "commit-encoding")) {
998                 char *arg = argv[2];
999                 int delimiter = *arg;
1000                 int i;
1001
1002                 switch (delimiter) {
1003                 case '"':
1004                 case '\'':
1005                         for (arg++, i = 0; arg[i]; i++)
1006                                 if (arg[i] == delimiter) {
1007                                         arg[i] = 0;
1008                                         break;
1009                                 }
1010                 default:
1011                         string_copy(opt_encoding, arg);
1012                         return OK;
1013                 }
1014         }
1015
1016         config_msg = "Unknown variable name";
1017         return ERR;
1018 }
1019
1020 /* Wants: mode request key */
1021 static int
1022 option_bind_command(int argc, char *argv[])
1023 {
1024         enum request request;
1025         int keymap;
1026         int key;
1027
1028         if (argc != 3) {
1029                 config_msg = "Wrong number of arguments given to bind command";
1030                 return ERR;
1031         }
1032
1033         if (set_keymap(&keymap, argv[0]) == ERR) {
1034                 config_msg = "Unknown key map";
1035                 return ERR;
1036         }
1037
1038         key = get_key_value(argv[1]);
1039         if (key == ERR) {
1040                 config_msg = "Unknown key";
1041                 return ERR;
1042         }
1043
1044         request = get_request(argv[2]);
1045         if (request == REQ_UNKNOWN) {
1046                 config_msg = "Unknown request name";
1047                 return ERR;
1048         }
1049
1050         add_keybinding(keymap, request, key);
1051
1052         return OK;
1053 }
1054
1055 static int
1056 set_option(char *opt, char *value)
1057 {
1058         char *argv[16];
1059         int valuelen;
1060         int argc = 0;
1061
1062         /* Tokenize */
1063         while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1064                 argv[argc++] = value;
1065
1066                 value += valuelen;
1067                 if (!*value)
1068                         break;
1069
1070                 *value++ = 0;
1071                 while (isspace(*value))
1072                         value++;
1073         }
1074
1075         if (!strcmp(opt, "color"))
1076                 return option_color_command(argc, argv);
1077
1078         if (!strcmp(opt, "set"))
1079                 return option_set_command(argc, argv);
1080
1081         if (!strcmp(opt, "bind"))
1082                 return option_bind_command(argc, argv);
1083
1084         config_msg = "Unknown option command";
1085         return ERR;
1086 }
1087
1088 static int
1089 read_option(char *opt, int optlen, char *value, int valuelen)
1090 {
1091         int status = OK;
1092
1093         config_lineno++;
1094         config_msg = "Internal error";
1095
1096         /* Check for comment markers, since read_properties() will
1097          * only ensure opt and value are split at first " \t". */
1098         optlen = strcspn(opt, "#");
1099         if (optlen == 0)
1100                 return OK;
1101
1102         if (opt[optlen] != 0) {
1103                 config_msg = "No option value";
1104                 status = ERR;
1105
1106         }  else {
1107                 /* Look for comment endings in the value. */
1108                 int len = strcspn(value, "#");
1109
1110                 if (len < valuelen) {
1111                         valuelen = len;
1112                         value[valuelen] = 0;
1113                 }
1114
1115                 status = set_option(opt, value);
1116         }
1117
1118         if (status == ERR) {
1119                 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1120                         config_lineno, optlen, opt, config_msg);
1121                 config_errors = TRUE;
1122         }
1123
1124         /* Always keep going if errors are encountered. */
1125         return OK;
1126 }
1127
1128 static int
1129 load_options(void)
1130 {
1131         char *home = getenv("HOME");
1132         char buf[SIZEOF_STR];
1133         FILE *file;
1134
1135         config_lineno = 0;
1136         config_errors = FALSE;
1137
1138         if (!home || !string_format(buf, "%s/.tigrc", home))
1139                 return ERR;
1140
1141         /* It's ok that the file doesn't exist. */
1142         file = fopen(buf, "r");
1143         if (!file)
1144                 return OK;
1145
1146         if (read_properties(file, " \t", read_option) == ERR ||
1147             config_errors == TRUE)
1148                 fprintf(stderr, "Errors while loading %s.\n", buf);
1149
1150         return OK;
1151 }
1152
1153
1154 /*
1155  * The viewer
1156  */
1157
1158 struct view;
1159 struct view_ops;
1160
1161 /* The display array of active views and the index of the current view. */
1162 static struct view *display[2];
1163 static unsigned int current_view;
1164
1165 #define foreach_displayed_view(view, i) \
1166         for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1167
1168 #define displayed_views()       (display[1] != NULL ? 2 : 1)
1169
1170 /* Current head and commit ID */
1171 static char ref_blob[SIZEOF_REF]        = "";
1172 static char ref_commit[SIZEOF_REF]      = "HEAD";
1173 static char ref_head[SIZEOF_REF]        = "HEAD";
1174
1175 struct view {
1176         const char *name;       /* View name */
1177         const char *cmd_fmt;    /* Default command line format */
1178         const char *cmd_env;    /* Command line set via environment */
1179         const char *id;         /* Points to either of ref_{head,commit,blob} */
1180
1181         struct view_ops *ops;   /* View operations */
1182
1183         enum keymap keymap;     /* What keymap does this view have */
1184
1185         char cmd[SIZEOF_STR];   /* Command buffer */
1186         char ref[SIZEOF_REF];   /* Hovered commit reference */
1187         char vid[SIZEOF_REF];   /* View ID. Set to id member when updating. */
1188
1189         int height, width;      /* The width and height of the main window */
1190         WINDOW *win;            /* The main window */
1191         WINDOW *title;          /* The title window living below the main window */
1192
1193         /* Navigation */
1194         unsigned long offset;   /* Offset of the window top */
1195         unsigned long lineno;   /* Current line number */
1196
1197         /* Searching */
1198         char grep[SIZEOF_STR];  /* Search string */
1199         regex_t *regex;         /* Pre-compiled regex */
1200
1201         /* If non-NULL, points to the view that opened this view. If this view
1202          * is closed tig will switch back to the parent view. */
1203         struct view *parent;
1204
1205         /* Buffering */
1206         unsigned long lines;    /* Total number of lines */
1207         struct line *line;      /* Line index */
1208         unsigned long line_size;/* Total number of allocated lines */
1209         unsigned int digits;    /* Number of digits in the lines member. */
1210
1211         /* Loading */
1212         FILE *pipe;
1213         time_t start_time;
1214 };
1215
1216 struct view_ops {
1217         /* What type of content being displayed. Used in the title bar. */
1218         const char *type;
1219         /* Draw one line; @lineno must be < view->height. */
1220         bool (*draw)(struct view *view, struct line *line, unsigned int lineno, bool selected);
1221         /* Read one line; updates view->line. */
1222         bool (*read)(struct view *view, char *data);
1223         /* Depending on view, change display based on current line. */
1224         bool (*enter)(struct view *view, struct line *line);
1225         /* Search for regex in a line. */
1226         bool (*grep)(struct view *view, struct line *line);
1227         /* Select line */
1228         void (*select)(struct view *view, struct line *line);
1229 };
1230
1231 static struct view_ops pager_ops;
1232 static struct view_ops main_ops;
1233 static struct view_ops tree_ops;
1234 static struct view_ops blob_ops;
1235
1236 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1237         { name, cmd, #env, ref, ops, map}
1238
1239 #define VIEW_(id, name, ops, ref) \
1240         VIEW_STR(name, TIG_##id##_CMD,  TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1241
1242
1243 static struct view views[] = {
1244         VIEW_(MAIN,  "main",  &main_ops,  ref_head),
1245         VIEW_(DIFF,  "diff",  &pager_ops, ref_commit),
1246         VIEW_(LOG,   "log",   &pager_ops, ref_head),
1247         VIEW_(TREE,  "tree",  &tree_ops,  ref_commit),
1248         VIEW_(BLOB,  "blob",  &blob_ops,  ref_blob),
1249         VIEW_(HELP,  "help",  &pager_ops, "static"),
1250         VIEW_(PAGER, "pager", &pager_ops, "static"),
1251 };
1252
1253 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1254
1255 #define foreach_view(view, i) \
1256         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1257
1258 #define view_is_displayed(view) \
1259         (view == display[0] || view == display[1])
1260
1261 static bool
1262 draw_view_line(struct view *view, unsigned int lineno)
1263 {
1264         struct line *line;
1265         bool selected = (view->offset + lineno == view->lineno);
1266
1267         assert(view_is_displayed(view));
1268
1269         if (view->offset + lineno >= view->lines)
1270                 return FALSE;
1271
1272         line = &view->line[view->offset + lineno];
1273
1274         if (selected) {
1275                 line->selected = TRUE;
1276                 view->ops->select(view, line);
1277         } else if (line->selected) {
1278                 line->selected = FALSE;
1279                 wmove(view->win, lineno, 0);
1280                 wclrtoeol(view->win);
1281         }
1282
1283         return view->ops->draw(view, line, lineno, selected);
1284 }
1285
1286 static void
1287 redraw_view_from(struct view *view, int lineno)
1288 {
1289         assert(0 <= lineno && lineno < view->height);
1290
1291         for (; lineno < view->height; lineno++) {
1292                 if (!draw_view_line(view, lineno))
1293                         break;
1294         }
1295
1296         redrawwin(view->win);
1297         wrefresh(view->win);
1298 }
1299
1300 static void
1301 redraw_view(struct view *view)
1302 {
1303         wclear(view->win);
1304         redraw_view_from(view, 0);
1305 }
1306
1307
1308 static void
1309 update_view_title(struct view *view)
1310 {
1311         assert(view_is_displayed(view));
1312
1313         if (view == display[current_view])
1314                 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1315         else
1316                 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1317
1318         werase(view->title);
1319         wmove(view->title, 0, 0);
1320
1321         if (*view->ref)
1322                 wprintw(view->title, "[%s] %s", view->name, view->ref);
1323         else
1324                 wprintw(view->title, "[%s]", view->name);
1325
1326         if (view->lines || view->pipe) {
1327                 unsigned int view_lines = view->offset + view->height;
1328                 unsigned int lines = view->lines
1329                                    ? MIN(view_lines, view->lines) * 100 / view->lines
1330                                    : 0;
1331
1332                 wprintw(view->title, " - %s %d of %d (%d%%)",
1333                         view->ops->type,
1334                         view->lineno + 1,
1335                         view->lines,
1336                         lines);
1337         }
1338
1339         if (view->pipe) {
1340                 time_t secs = time(NULL) - view->start_time;
1341
1342                 /* Three git seconds are a long time ... */
1343                 if (secs > 2)
1344                         wprintw(view->title, " %lds", secs);
1345         }
1346
1347         wmove(view->title, 0, view->width - 1);
1348         wrefresh(view->title);
1349 }
1350
1351 static void
1352 resize_display(void)
1353 {
1354         int offset, i;
1355         struct view *base = display[0];
1356         struct view *view = display[1] ? display[1] : display[0];
1357
1358         /* Setup window dimensions */
1359
1360         getmaxyx(stdscr, base->height, base->width);
1361
1362         /* Make room for the status window. */
1363         base->height -= 1;
1364
1365         if (view != base) {
1366                 /* Horizontal split. */
1367                 view->width   = base->width;
1368                 view->height  = SCALE_SPLIT_VIEW(base->height);
1369                 base->height -= view->height;
1370
1371                 /* Make room for the title bar. */
1372                 view->height -= 1;
1373         }
1374
1375         /* Make room for the title bar. */
1376         base->height -= 1;
1377
1378         offset = 0;
1379
1380         foreach_displayed_view (view, i) {
1381                 if (!view->win) {
1382                         view->win = newwin(view->height, 0, offset, 0);
1383                         if (!view->win)
1384                                 die("Failed to create %s view", view->name);
1385
1386                         scrollok(view->win, TRUE);
1387
1388                         view->title = newwin(1, 0, offset + view->height, 0);
1389                         if (!view->title)
1390                                 die("Failed to create title window");
1391
1392                 } else {
1393                         wresize(view->win, view->height, view->width);
1394                         mvwin(view->win,   offset, 0);
1395                         mvwin(view->title, offset + view->height, 0);
1396                 }
1397
1398                 offset += view->height + 1;
1399         }
1400 }
1401
1402 static void
1403 redraw_display(void)
1404 {
1405         struct view *view;
1406         int i;
1407
1408         foreach_displayed_view (view, i) {
1409                 redraw_view(view);
1410                 update_view_title(view);
1411         }
1412 }
1413
1414 static void
1415 update_display_cursor(void)
1416 {
1417         struct view *view = display[current_view];
1418
1419         /* Move the cursor to the right-most column of the cursor line.
1420          *
1421          * XXX: This could turn out to be a bit expensive, but it ensures that
1422          * the cursor does not jump around. */
1423         if (view->lines) {
1424                 wmove(view->win, view->lineno - view->offset, view->width - 1);
1425                 wrefresh(view->win);
1426         }
1427 }
1428
1429 /*
1430  * Navigation
1431  */
1432
1433 /* Scrolling backend */
1434 static void
1435 do_scroll_view(struct view *view, int lines)
1436 {
1437         bool redraw_current_line = FALSE;
1438
1439         /* The rendering expects the new offset. */
1440         view->offset += lines;
1441
1442         assert(0 <= view->offset && view->offset < view->lines);
1443         assert(lines);
1444
1445         /* Move current line into the view. */
1446         if (view->lineno < view->offset) {
1447                 view->lineno = view->offset;
1448                 redraw_current_line = TRUE;
1449         } else if (view->lineno >= view->offset + view->height) {
1450                 view->lineno = view->offset + view->height - 1;
1451                 redraw_current_line = TRUE;
1452         }
1453
1454         assert(view->offset <= view->lineno && view->lineno < view->lines);
1455
1456         /* Redraw the whole screen if scrolling is pointless. */
1457         if (view->height < ABS(lines)) {
1458                 redraw_view(view);
1459
1460         } else {
1461                 int line = lines > 0 ? view->height - lines : 0;
1462                 int end = line + ABS(lines);
1463
1464                 wscrl(view->win, lines);
1465
1466                 for (; line < end; line++) {
1467                         if (!draw_view_line(view, line))
1468                                 break;
1469                 }
1470
1471                 if (redraw_current_line)
1472                         draw_view_line(view, view->lineno - view->offset);
1473         }
1474
1475         redrawwin(view->win);
1476         wrefresh(view->win);
1477         report("");
1478 }
1479
1480 /* Scroll frontend */
1481 static void
1482 scroll_view(struct view *view, enum request request)
1483 {
1484         int lines = 1;
1485
1486         assert(view_is_displayed(view));
1487
1488         switch (request) {
1489         case REQ_SCROLL_PAGE_DOWN:
1490                 lines = view->height;
1491         case REQ_SCROLL_LINE_DOWN:
1492                 if (view->offset + lines > view->lines)
1493                         lines = view->lines - view->offset;
1494
1495                 if (lines == 0 || view->offset + view->height >= view->lines) {
1496                         report("Cannot scroll beyond the last line");
1497                         return;
1498                 }
1499                 break;
1500
1501         case REQ_SCROLL_PAGE_UP:
1502                 lines = view->height;
1503         case REQ_SCROLL_LINE_UP:
1504                 if (lines > view->offset)
1505                         lines = view->offset;
1506
1507                 if (lines == 0) {
1508                         report("Cannot scroll beyond the first line");
1509                         return;
1510                 }
1511
1512                 lines = -lines;
1513                 break;
1514
1515         default:
1516                 die("request %d not handled in switch", request);
1517         }
1518
1519         do_scroll_view(view, lines);
1520 }
1521
1522 /* Cursor moving */
1523 static void
1524 move_view(struct view *view, enum request request)
1525 {
1526         int scroll_steps = 0;
1527         int steps;
1528
1529         switch (request) {
1530         case REQ_MOVE_FIRST_LINE:
1531                 steps = -view->lineno;
1532                 break;
1533
1534         case REQ_MOVE_LAST_LINE:
1535                 steps = view->lines - view->lineno - 1;
1536                 break;
1537
1538         case REQ_MOVE_PAGE_UP:
1539                 steps = view->height > view->lineno
1540                       ? -view->lineno : -view->height;
1541                 break;
1542
1543         case REQ_MOVE_PAGE_DOWN:
1544                 steps = view->lineno + view->height >= view->lines
1545                       ? view->lines - view->lineno - 1 : view->height;
1546                 break;
1547
1548         case REQ_MOVE_UP:
1549                 steps = -1;
1550                 break;
1551
1552         case REQ_MOVE_DOWN:
1553                 steps = 1;
1554                 break;
1555
1556         default:
1557                 die("request %d not handled in switch", request);
1558         }
1559
1560         if (steps <= 0 && view->lineno == 0) {
1561                 report("Cannot move beyond the first line");
1562                 return;
1563
1564         } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1565                 report("Cannot move beyond the last line");
1566                 return;
1567         }
1568
1569         /* Move the current line */
1570         view->lineno += steps;
1571         assert(0 <= view->lineno && view->lineno < view->lines);
1572
1573         /* Check whether the view needs to be scrolled */
1574         if (view->lineno < view->offset ||
1575             view->lineno >= view->offset + view->height) {
1576                 scroll_steps = steps;
1577                 if (steps < 0 && -steps > view->offset) {
1578                         scroll_steps = -view->offset;
1579
1580                 } else if (steps > 0) {
1581                         if (view->lineno == view->lines - 1 &&
1582                             view->lines > view->height) {
1583                                 scroll_steps = view->lines - view->offset - 1;
1584                                 if (scroll_steps >= view->height)
1585                                         scroll_steps -= view->height - 1;
1586                         }
1587                 }
1588         }
1589
1590         if (!view_is_displayed(view)) {
1591                 view->offset += steps;
1592                 view->ops->select(view, &view->line[view->lineno]);
1593                 return;
1594         }
1595
1596         /* Repaint the old "current" line if we be scrolling */
1597         if (ABS(steps) < view->height)
1598                 draw_view_line(view, view->lineno - steps - view->offset);
1599
1600         if (scroll_steps) {
1601                 do_scroll_view(view, scroll_steps);
1602                 return;
1603         }
1604
1605         /* Draw the current line */
1606         draw_view_line(view, view->lineno - view->offset);
1607
1608         redrawwin(view->win);
1609         wrefresh(view->win);
1610         report("");
1611 }
1612
1613
1614 /*
1615  * Searching
1616  */
1617
1618 static void search_view(struct view *view, enum request request);
1619
1620 static bool
1621 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1622 {
1623         assert(view_is_displayed(view));
1624
1625         if (!view->ops->grep(view, line))
1626                 return FALSE;
1627
1628         if (lineno - view->offset >= view->height) {
1629                 view->offset = lineno;
1630                 view->lineno = lineno;
1631                 redraw_view(view);
1632
1633         } else {
1634                 unsigned long old_lineno = view->lineno - view->offset;
1635
1636                 view->lineno = lineno;
1637                 draw_view_line(view, old_lineno);
1638
1639                 draw_view_line(view, view->lineno - view->offset);
1640                 redrawwin(view->win);
1641                 wrefresh(view->win);
1642         }
1643
1644         report("Line %ld matches '%s'", lineno + 1, view->grep);
1645         return TRUE;
1646 }
1647
1648 static void
1649 find_next(struct view *view, enum request request)
1650 {
1651         unsigned long lineno = view->lineno;
1652         int direction;
1653
1654         if (!*view->grep) {
1655                 if (!*opt_search)
1656                         report("No previous search");
1657                 else
1658                         search_view(view, request);
1659                 return;
1660         }
1661
1662         switch (request) {
1663         case REQ_SEARCH:
1664         case REQ_FIND_NEXT:
1665                 direction = 1;
1666                 break;
1667
1668         case REQ_SEARCH_BACK:
1669         case REQ_FIND_PREV:
1670                 direction = -1;
1671                 break;
1672
1673         default:
1674                 return;
1675         }
1676
1677         if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1678                 lineno += direction;
1679
1680         /* Note, lineno is unsigned long so will wrap around in which case it
1681          * will become bigger than view->lines. */
1682         for (; lineno < view->lines; lineno += direction) {
1683                 struct line *line = &view->line[lineno];
1684
1685                 if (find_next_line(view, lineno, line))
1686                         return;
1687         }
1688
1689         report("No match found for '%s'", view->grep);
1690 }
1691
1692 static void
1693 search_view(struct view *view, enum request request)
1694 {
1695         int regex_err;
1696
1697         if (view->regex) {
1698                 regfree(view->regex);
1699                 *view->grep = 0;
1700         } else {
1701                 view->regex = calloc(1, sizeof(*view->regex));
1702                 if (!view->regex)
1703                         return;
1704         }
1705
1706         regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
1707         if (regex_err != 0) {
1708                 char buf[SIZEOF_STR] = "unknown error";
1709
1710                 regerror(regex_err, view->regex, buf, sizeof(buf));
1711                 report("Search failed: %s", buf);
1712                 return;
1713         }
1714
1715         string_copy(view->grep, opt_search);
1716
1717         find_next(view, request);
1718 }
1719
1720 /*
1721  * Incremental updating
1722  */
1723
1724 static void
1725 end_update(struct view *view)
1726 {
1727         if (!view->pipe)
1728                 return;
1729         set_nonblocking_input(FALSE);
1730         if (view->pipe == stdin)
1731                 fclose(view->pipe);
1732         else
1733                 pclose(view->pipe);
1734         view->pipe = NULL;
1735 }
1736
1737 static bool
1738 begin_update(struct view *view)
1739 {
1740         const char *id = view->id;
1741
1742         if (view->pipe)
1743                 end_update(view);
1744
1745         if (opt_cmd[0]) {
1746                 string_copy(view->cmd, opt_cmd);
1747                 opt_cmd[0] = 0;
1748                 /* When running random commands, the view ref could have become
1749                  * invalid so clear it. */
1750                 view->ref[0] = 0;
1751
1752         } else if (view == VIEW(REQ_VIEW_TREE)) {
1753                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1754
1755                 if (strcmp(view->vid, view->id))
1756                         opt_path[0] = 0;
1757
1758                 if (!string_format(view->cmd, format, id, opt_path))
1759                         return FALSE;
1760
1761         } else {
1762                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1763
1764                 if (!string_format(view->cmd, format, id, id, id, id, id))
1765                         return FALSE;
1766         }
1767
1768         /* Special case for the pager view. */
1769         if (opt_pipe) {
1770                 view->pipe = opt_pipe;
1771                 opt_pipe = NULL;
1772         } else {
1773                 view->pipe = popen(view->cmd, "r");
1774         }
1775
1776         if (!view->pipe)
1777                 return FALSE;
1778
1779         set_nonblocking_input(TRUE);
1780
1781         view->offset = 0;
1782         view->lines  = 0;
1783         view->lineno = 0;
1784         string_copy(view->vid, id);
1785
1786         if (view->line) {
1787                 int i;
1788
1789                 for (i = 0; i < view->lines; i++)
1790                         if (view->line[i].data)
1791                                 free(view->line[i].data);
1792
1793                 free(view->line);
1794                 view->line = NULL;
1795         }
1796
1797         view->start_time = time(NULL);
1798
1799         return TRUE;
1800 }
1801
1802 static struct line *
1803 realloc_lines(struct view *view, size_t line_size)
1804 {
1805         struct line *tmp = realloc(view->line, sizeof(*view->line) * line_size);
1806
1807         if (!tmp)
1808                 return NULL;
1809
1810         view->line = tmp;
1811         view->line_size = line_size;
1812         return view->line;
1813 }
1814
1815 static bool
1816 update_view(struct view *view)
1817 {
1818         char in_buffer[BUFSIZ];
1819         char out_buffer[BUFSIZ * 2];
1820         char *line;
1821         /* The number of lines to read. If too low it will cause too much
1822          * redrawing (and possible flickering), if too high responsiveness
1823          * will suffer. */
1824         unsigned long lines = view->height;
1825         int redraw_from = -1;
1826
1827         if (!view->pipe)
1828                 return TRUE;
1829
1830         /* Only redraw if lines are visible. */
1831         if (view->offset + view->height >= view->lines)
1832                 redraw_from = view->lines - view->offset;
1833
1834         /* FIXME: This is probably not perfect for backgrounded views. */
1835         if (!realloc_lines(view, view->lines + lines))
1836                 goto alloc_error;
1837
1838         while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
1839                 size_t linelen = strlen(line);
1840
1841                 if (linelen)
1842                         line[linelen - 1] = 0;
1843
1844                 if (opt_iconv != ICONV_NONE) {
1845                         char *inbuf = line;
1846                         size_t inlen = linelen;
1847
1848                         char *outbuf = out_buffer;
1849                         size_t outlen = sizeof(out_buffer);
1850
1851                         size_t ret;
1852
1853                         ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
1854                         if (ret != (size_t) -1) {
1855                                 line = out_buffer;
1856                                 linelen = strlen(out_buffer);
1857                         }
1858                 }
1859
1860                 if (!view->ops->read(view, line))
1861                         goto alloc_error;
1862
1863                 if (lines-- == 1)
1864                         break;
1865         }
1866
1867         {
1868                 int digits;
1869
1870                 lines = view->lines;
1871                 for (digits = 0; lines; digits++)
1872                         lines /= 10;
1873
1874                 /* Keep the displayed view in sync with line number scaling. */
1875                 if (digits != view->digits) {
1876                         view->digits = digits;
1877                         redraw_from = 0;
1878                 }
1879         }
1880
1881         if (!view_is_displayed(view))
1882                 goto check_pipe;
1883
1884         if (view == VIEW(REQ_VIEW_TREE)) {
1885                 /* Clear the view and redraw everything since the tree sorting
1886                  * might have rearranged things. */
1887                 redraw_view(view);
1888
1889         } else if (redraw_from >= 0) {
1890                 /* If this is an incremental update, redraw the previous line
1891                  * since for commits some members could have changed when
1892                  * loading the main view. */
1893                 if (redraw_from > 0)
1894                         redraw_from--;
1895
1896                 /* Incrementally draw avoids flickering. */
1897                 redraw_view_from(view, redraw_from);
1898         }
1899
1900         /* Update the title _after_ the redraw so that if the redraw picks up a
1901          * commit reference in view->ref it'll be available here. */
1902         update_view_title(view);
1903
1904 check_pipe:
1905         if (ferror(view->pipe)) {
1906                 report("Failed to read: %s", strerror(errno));
1907                 goto end;
1908
1909         } else if (feof(view->pipe)) {
1910                 report("");
1911                 goto end;
1912         }
1913
1914         return TRUE;
1915
1916 alloc_error:
1917         report("Allocation failure");
1918
1919 end:
1920         end_update(view);
1921         return FALSE;
1922 }
1923
1924
1925 /*
1926  * View opening
1927  */
1928
1929 static void open_help_view(struct view *view)
1930 {
1931         char buf[BUFSIZ];
1932         int lines = ARRAY_SIZE(req_info) + 2;
1933         int i;
1934
1935         if (view->lines > 0)
1936                 return;
1937
1938         for (i = 0; i < ARRAY_SIZE(req_info); i++)
1939                 if (!req_info[i].request)
1940                         lines++;
1941
1942         view->line = calloc(lines, sizeof(*view->line));
1943         if (!view->line) {
1944                 report("Allocation failure");
1945                 return;
1946         }
1947
1948         view->ops->read(view, "Quick reference for tig keybindings:");
1949
1950         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
1951                 char *key;
1952
1953                 if (!req_info[i].request) {
1954                         view->ops->read(view, "");
1955                         view->ops->read(view, req_info[i].help);
1956                         continue;
1957                 }
1958
1959                 key = get_key(req_info[i].request);
1960                 if (!string_format(buf, "%-25s %s", key, req_info[i].help))
1961                         continue;
1962
1963                 view->ops->read(view, buf);
1964         }
1965 }
1966
1967 enum open_flags {
1968         OPEN_DEFAULT = 0,       /* Use default view switching. */
1969         OPEN_SPLIT = 1,         /* Split current view. */
1970         OPEN_BACKGROUNDED = 2,  /* Backgrounded. */
1971         OPEN_RELOAD = 4,        /* Reload view even if it is the current. */
1972 };
1973
1974 static void
1975 open_view(struct view *prev, enum request request, enum open_flags flags)
1976 {
1977         bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
1978         bool split = !!(flags & OPEN_SPLIT);
1979         bool reload = !!(flags & OPEN_RELOAD);
1980         struct view *view = VIEW(request);
1981         int nviews = displayed_views();
1982         struct view *base_view = display[0];
1983
1984         if (view == prev && nviews == 1 && !reload) {
1985                 report("Already in %s view", view->name);
1986                 return;
1987         }
1988
1989         if (view == VIEW(REQ_VIEW_HELP)) {
1990                 open_help_view(view);
1991
1992         } else if ((reload || strcmp(view->vid, view->id)) &&
1993                    !begin_update(view)) {
1994                 report("Failed to load %s view", view->name);
1995                 return;
1996         }
1997
1998         if (split) {
1999                 display[1] = view;
2000                 if (!backgrounded)
2001                         current_view = 1;
2002         } else {
2003                 /* Maximize the current view. */
2004                 memset(display, 0, sizeof(display));
2005                 current_view = 0;
2006                 display[current_view] = view;
2007         }
2008
2009         /* Resize the view when switching between split- and full-screen,
2010          * or when switching between two different full-screen views. */
2011         if (nviews != displayed_views() ||
2012             (nviews == 1 && base_view != display[0]))
2013                 resize_display();
2014
2015         if (split && prev->lineno - prev->offset >= prev->height) {
2016                 /* Take the title line into account. */
2017                 int lines = prev->lineno - prev->offset - prev->height + 1;
2018
2019                 /* Scroll the view that was split if the current line is
2020                  * outside the new limited view. */
2021                 do_scroll_view(prev, lines);
2022         }
2023
2024         if (prev && view != prev) {
2025                 if (split && !backgrounded) {
2026                         /* "Blur" the previous view. */
2027                         update_view_title(prev);
2028                 }
2029
2030                 view->parent = prev;
2031         }
2032
2033         if (view->pipe && view->lines == 0) {
2034                 /* Clear the old view and let the incremental updating refill
2035                  * the screen. */
2036                 wclear(view->win);
2037                 report("");
2038         } else {
2039                 redraw_view(view);
2040                 report("");
2041         }
2042
2043         /* If the view is backgrounded the above calls to report()
2044          * won't redraw the view title. */
2045         if (backgrounded)
2046                 update_view_title(view);
2047 }
2048
2049
2050 /*
2051  * User request switch noodle
2052  */
2053
2054 static int
2055 view_driver(struct view *view, enum request request)
2056 {
2057         int i;
2058
2059         switch (request) {
2060         case REQ_MOVE_UP:
2061         case REQ_MOVE_DOWN:
2062         case REQ_MOVE_PAGE_UP:
2063         case REQ_MOVE_PAGE_DOWN:
2064         case REQ_MOVE_FIRST_LINE:
2065         case REQ_MOVE_LAST_LINE:
2066                 move_view(view, request);
2067                 break;
2068
2069         case REQ_SCROLL_LINE_DOWN:
2070         case REQ_SCROLL_LINE_UP:
2071         case REQ_SCROLL_PAGE_DOWN:
2072         case REQ_SCROLL_PAGE_UP:
2073                 scroll_view(view, request);
2074                 break;
2075
2076         case REQ_VIEW_BLOB:
2077                 if (!ref_blob[0]) {
2078                         report("No file chosen, press 't' to open tree view");
2079                         break;
2080                 }
2081                 /* Fall-through */
2082         case REQ_VIEW_MAIN:
2083         case REQ_VIEW_DIFF:
2084         case REQ_VIEW_LOG:
2085         case REQ_VIEW_TREE:
2086         case REQ_VIEW_HELP:
2087         case REQ_VIEW_PAGER:
2088                 open_view(view, request, OPEN_DEFAULT);
2089                 break;
2090
2091         case REQ_NEXT:
2092         case REQ_PREVIOUS:
2093                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2094
2095                 if ((view == VIEW(REQ_VIEW_DIFF) &&
2096                      view->parent == VIEW(REQ_VIEW_MAIN)) ||
2097                    (view == VIEW(REQ_VIEW_BLOB) &&
2098                      view->parent == VIEW(REQ_VIEW_TREE))) {
2099                         view = view->parent;
2100                         move_view(view, request);
2101                         if (view_is_displayed(view))
2102                                 update_view_title(view);
2103                 } else {
2104                         move_view(view, request);
2105                         break;
2106                 }
2107                 /* Fall-through */
2108
2109         case REQ_ENTER:
2110                 if (!view->lines) {
2111                         report("Nothing to enter");
2112                         break;
2113                 }
2114                 return view->ops->enter(view, &view->line[view->lineno]);
2115
2116         case REQ_VIEW_NEXT:
2117         {
2118                 int nviews = displayed_views();
2119                 int next_view = (current_view + 1) % nviews;
2120
2121                 if (next_view == current_view) {
2122                         report("Only one view is displayed");
2123                         break;
2124                 }
2125
2126                 current_view = next_view;
2127                 /* Blur out the title of the previous view. */
2128                 update_view_title(view);
2129                 report("");
2130                 break;
2131         }
2132         case REQ_TOGGLE_LINENO:
2133                 opt_line_number = !opt_line_number;
2134                 redraw_display();
2135                 break;
2136
2137         case REQ_TOGGLE_REV_GRAPH:
2138                 opt_rev_graph = !opt_rev_graph;
2139                 redraw_display();
2140                 break;
2141
2142         case REQ_PROMPT:
2143                 /* Always reload^Wrerun commands from the prompt. */
2144                 open_view(view, opt_request, OPEN_RELOAD);
2145                 break;
2146
2147         case REQ_SEARCH:
2148         case REQ_SEARCH_BACK:
2149                 search_view(view, request);
2150                 break;
2151
2152         case REQ_FIND_NEXT:
2153         case REQ_FIND_PREV:
2154                 find_next(view, request);
2155                 break;
2156
2157         case REQ_STOP_LOADING:
2158                 for (i = 0; i < ARRAY_SIZE(views); i++) {
2159                         view = &views[i];
2160                         if (view->pipe)
2161                                 report("Stopped loading the %s view", view->name),
2162                         end_update(view);
2163                 }
2164                 break;
2165
2166         case REQ_SHOW_VERSION:
2167                 report("%s (built %s)", VERSION, __DATE__);
2168                 return TRUE;
2169
2170         case REQ_SCREEN_RESIZE:
2171                 resize_display();
2172                 /* Fall-through */
2173         case REQ_SCREEN_REDRAW:
2174                 redraw_display();
2175                 break;
2176
2177         case REQ_NONE:
2178                 doupdate();
2179                 return TRUE;
2180
2181         case REQ_VIEW_CLOSE:
2182                 /* XXX: Mark closed views by letting view->parent point to the
2183                  * view itself. Parents to closed view should never be
2184                  * followed. */
2185                 if (view->parent &&
2186                     view->parent->parent != view->parent) {
2187                         memset(display, 0, sizeof(display));
2188                         current_view = 0;
2189                         display[current_view] = view->parent;
2190                         view->parent = view;
2191                         resize_display();
2192                         redraw_display();
2193                         break;
2194                 }
2195                 /* Fall-through */
2196         case REQ_QUIT:
2197                 return FALSE;
2198
2199         default:
2200                 /* An unknown key will show most commonly used commands. */
2201                 report("Unknown key, press 'h' for help");
2202                 return TRUE;
2203         }
2204
2205         return TRUE;
2206 }
2207
2208
2209 /*
2210  * Pager backend
2211  */
2212
2213 static bool
2214 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2215 {
2216         char *text = line->data;
2217         enum line_type type = line->type;
2218         int textlen = strlen(text);
2219         int attr;
2220
2221         wmove(view->win, lineno, 0);
2222
2223         if (selected) {
2224                 type = LINE_CURSOR;
2225                 wchgat(view->win, -1, 0, type, NULL);
2226         }
2227
2228         attr = get_line_attr(type);
2229         wattrset(view->win, attr);
2230
2231         if (opt_line_number || opt_tab_size < TABSIZE) {
2232                 static char spaces[] = "                    ";
2233                 int col_offset = 0, col = 0;
2234
2235                 if (opt_line_number) {
2236                         unsigned long real_lineno = view->offset + lineno + 1;
2237
2238                         if (real_lineno == 1 ||
2239                             (real_lineno % opt_num_interval) == 0) {
2240                                 wprintw(view->win, "%.*d", view->digits, real_lineno);
2241
2242                         } else {
2243                                 waddnstr(view->win, spaces,
2244                                          MIN(view->digits, STRING_SIZE(spaces)));
2245                         }
2246                         waddstr(view->win, ": ");
2247                         col_offset = view->digits + 2;
2248                 }
2249
2250                 while (text && col_offset + col < view->width) {
2251                         int cols_max = view->width - col_offset - col;
2252                         char *pos = text;
2253                         int cols;
2254
2255                         if (*text == '\t') {
2256                                 text++;
2257                                 assert(sizeof(spaces) > TABSIZE);
2258                                 pos = spaces;
2259                                 cols = opt_tab_size - (col % opt_tab_size);
2260
2261                         } else {
2262                                 text = strchr(text, '\t');
2263                                 cols = line ? text - pos : strlen(pos);
2264                         }
2265
2266                         waddnstr(view->win, pos, MIN(cols, cols_max));
2267                         col += cols;
2268                 }
2269
2270         } else {
2271                 int col = 0, pos = 0;
2272
2273                 for (; pos < textlen && col < view->width; pos++, col++)
2274                         if (text[pos] == '\t')
2275                                 col += TABSIZE - (col % TABSIZE) - 1;
2276
2277                 waddnstr(view->win, text, pos);
2278         }
2279
2280         return TRUE;
2281 }
2282
2283 static bool
2284 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2285 {
2286         char refbuf[SIZEOF_STR];
2287         char *ref = NULL;
2288         FILE *pipe;
2289
2290         if (!string_format(refbuf, "git describe %s", commit_id))
2291                 return TRUE;
2292
2293         pipe = popen(refbuf, "r");
2294         if (!pipe)
2295                 return TRUE;
2296
2297         if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2298                 ref = chomp_string(ref);
2299         pclose(pipe);
2300
2301         if (!ref || !*ref)
2302                 return TRUE;
2303
2304         /* This is the only fatal call, since it can "corrupt" the buffer. */
2305         if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2306                 return FALSE;
2307
2308         return TRUE;
2309 }
2310
2311 static void
2312 add_pager_refs(struct view *view, struct line *line)
2313 {
2314         char buf[SIZEOF_STR];
2315         char *commit_id = line->data + STRING_SIZE("commit ");
2316         struct ref **refs;
2317         size_t bufpos = 0, refpos = 0;
2318         const char *sep = "Refs: ";
2319         bool is_tag = FALSE;
2320
2321         assert(line->type == LINE_COMMIT);
2322
2323         refs = get_refs(commit_id);
2324         if (!refs) {
2325                 if (view == VIEW(REQ_VIEW_DIFF))
2326                         goto try_add_describe_ref;
2327                 return;
2328         }
2329
2330         do {
2331                 struct ref *ref = refs[refpos];
2332                 char *fmt = ref->tag ? "%s[%s]" : "%s%s";
2333
2334                 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2335                         return;
2336                 sep = ", ";
2337                 if (ref->tag)
2338                         is_tag = TRUE;
2339         } while (refs[refpos++]->next);
2340
2341         if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2342 try_add_describe_ref:
2343                 /* Add <tag>-g<commit_id> "fake" reference. */
2344                 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2345                         return;
2346         }
2347
2348         if (bufpos == 0)
2349                 return;
2350
2351         if (!realloc_lines(view, view->line_size + 1))
2352                 return;
2353
2354         line = &view->line[view->lines];
2355         line->data = strdup(buf);
2356         if (!line->data)
2357                 return;
2358
2359         line->type = LINE_PP_REFS;
2360         view->lines++;
2361 }
2362
2363 static bool
2364 pager_read(struct view *view, char *data)
2365 {
2366         struct line *line = &view->line[view->lines];
2367
2368         line->data = strdup(data);
2369         if (!line->data)
2370                 return FALSE;
2371
2372         line->type = get_line_type(line->data);
2373         view->lines++;
2374
2375         if (line->type == LINE_COMMIT &&
2376             (view == VIEW(REQ_VIEW_DIFF) ||
2377              view == VIEW(REQ_VIEW_LOG)))
2378                 add_pager_refs(view, line);
2379
2380         return TRUE;
2381 }
2382
2383 static bool
2384 pager_enter(struct view *view, struct line *line)
2385 {
2386         int split = 0;
2387
2388         if (line->type == LINE_COMMIT &&
2389            (view == VIEW(REQ_VIEW_LOG) ||
2390             view == VIEW(REQ_VIEW_PAGER))) {
2391                 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2392                 split = 1;
2393         }
2394
2395         /* Always scroll the view even if it was split. That way
2396          * you can use Enter to scroll through the log view and
2397          * split open each commit diff. */
2398         scroll_view(view, REQ_SCROLL_LINE_DOWN);
2399
2400         /* FIXME: A minor workaround. Scrolling the view will call report("")
2401          * but if we are scrolling a non-current view this won't properly
2402          * update the view title. */
2403         if (split)
2404                 update_view_title(view);
2405
2406         return TRUE;
2407 }
2408
2409 static bool
2410 pager_grep(struct view *view, struct line *line)
2411 {
2412         regmatch_t pmatch;
2413         char *text = line->data;
2414
2415         if (!*text)
2416                 return FALSE;
2417
2418         if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2419                 return FALSE;
2420
2421         return TRUE;
2422 }
2423
2424 static void
2425 pager_select(struct view *view, struct line *line)
2426 {
2427         if (line->type == LINE_COMMIT) {
2428                 char *text = line->data;
2429
2430                 string_copy(view->ref, text + STRING_SIZE("commit "));
2431                 string_copy(ref_commit, view->ref);
2432         }
2433 }
2434
2435 static struct view_ops pager_ops = {
2436         "line",
2437         pager_draw,
2438         pager_read,
2439         pager_enter,
2440         pager_grep,
2441         pager_select,
2442 };
2443
2444
2445 /*
2446  * Tree backend
2447  */
2448
2449 /* Parse output from git ls-tree:
2450  *
2451  * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
2452  * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
2453  * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
2454  * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
2455  */
2456
2457 #define SIZEOF_TREE_ATTR \
2458         STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
2459
2460 #define TREE_UP_FORMAT "040000 tree %s\t.."
2461
2462 static int
2463 tree_compare_entry(enum line_type type1, char *name1,
2464                    enum line_type type2, char *name2)
2465 {
2466         if (type1 != type2) {
2467                 if (type1 == LINE_TREE_DIR)
2468                         return -1;
2469                 return 1;
2470         }
2471
2472         return strcmp(name1, name2);
2473 }
2474
2475 static bool
2476 tree_read(struct view *view, char *text)
2477 {
2478         size_t textlen = strlen(text);
2479         char buf[SIZEOF_STR];
2480         unsigned long pos;
2481         enum line_type type;
2482         bool first_read = view->lines == 0;
2483
2484         if (textlen <= SIZEOF_TREE_ATTR)
2485                 return FALSE;
2486
2487         type = text[STRING_SIZE("100644 ")] == 't'
2488              ? LINE_TREE_DIR : LINE_TREE_FILE;
2489
2490         if (first_read) {
2491                 /* Add path info line */
2492                 if (string_format(buf, "Directory path /%s", opt_path) &&
2493                     realloc_lines(view, view->line_size + 1) &&
2494                     pager_read(view, buf))
2495                         view->line[view->lines - 1].type = LINE_DEFAULT;
2496                 else
2497                         return FALSE;
2498
2499                 /* Insert "link" to parent directory. */
2500                 if (*opt_path &&
2501                     string_format(buf, TREE_UP_FORMAT, view->ref) &&
2502                     realloc_lines(view, view->line_size + 1) &&
2503                     pager_read(view, buf))
2504                         view->line[view->lines - 1].type = LINE_TREE_DIR;
2505                 else if (*opt_path)
2506                         return FALSE;
2507         }
2508
2509         /* Strip the path part ... */
2510         if (*opt_path) {
2511                 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
2512                 size_t striplen = strlen(opt_path);
2513                 char *path = text + SIZEOF_TREE_ATTR;
2514
2515                 if (pathlen > striplen)
2516                         memmove(path, path + striplen,
2517                                 pathlen - striplen + 1);
2518         }
2519
2520         /* Skip "Directory ..." and ".." line. */
2521         for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
2522                 struct line *line = &view->line[pos];
2523                 char *path1 = ((char *) line->data) + SIZEOF_TREE_ATTR;
2524                 char *path2 = text + SIZEOF_TREE_ATTR;
2525                 int cmp = tree_compare_entry(line->type, path1, type, path2);
2526
2527                 if (cmp <= 0)
2528                         continue;
2529
2530                 text = strdup(text);
2531                 if (!text)
2532                         return FALSE;
2533
2534                 if (view->lines > pos)
2535                         memmove(&view->line[pos + 1], &view->line[pos],
2536                                 (view->lines - pos) * sizeof(*line));
2537
2538                 line = &view->line[pos];
2539                 line->data = text;
2540                 line->type = type;
2541                 view->lines++;
2542                 return TRUE;
2543         }
2544
2545         if (!pager_read(view, text))
2546                 return FALSE;
2547
2548         /* Move the current line to the first tree entry. */
2549         if (first_read)
2550                 view->lineno++;
2551
2552         view->line[view->lines - 1].type = type;
2553         return TRUE;
2554 }
2555
2556 static bool
2557 tree_enter(struct view *view, struct line *line)
2558 {
2559         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2560         char *data = line->data;
2561         enum request request;
2562
2563         switch (line->type) {
2564         case LINE_TREE_DIR:
2565                 /* Depending on whether it is a subdir or parent (updir?) link
2566                  * mangle the path buffer. */
2567                 if (line == &view->line[1] && *opt_path) {
2568                         size_t path_len = strlen(opt_path);
2569                         char *dirsep = opt_path + path_len - 1;
2570
2571                         while (dirsep > opt_path && dirsep[-1] != '/')
2572                                 dirsep--;
2573
2574                         dirsep[0] = 0;
2575
2576                 } else {
2577                         size_t pathlen = strlen(opt_path);
2578                         size_t origlen = pathlen;
2579                         char *basename = data + SIZEOF_TREE_ATTR;
2580
2581                         if (!string_format_from(opt_path, &pathlen, "%s/", basename)) {
2582                                 opt_path[origlen] = 0;
2583                                 return TRUE;
2584                         }
2585                 }
2586
2587                 /* Trees and subtrees share the same ID, so they are not not
2588                  * unique like blobs. */
2589                 flags |= OPEN_RELOAD;
2590                 request = REQ_VIEW_TREE;
2591                 break;
2592
2593         case LINE_TREE_FILE:
2594                 request = REQ_VIEW_BLOB;
2595                 break;
2596
2597         default:
2598                 return TRUE;
2599         }
2600
2601         open_view(view, request, flags);
2602
2603         if (!VIEW(request)->pipe)
2604                 return TRUE;
2605
2606         /* For tree views insert the path to the parent as the first line. */
2607         if (request == REQ_VIEW_BLOB) {
2608                 /* Mirror what is showed in the title bar. */
2609                 string_ncopy(ref_blob, data + STRING_SIZE("100644 blob "), 40);
2610                 string_copy(VIEW(REQ_VIEW_BLOB)->ref, ref_blob);
2611                 return TRUE;
2612         }
2613
2614         return TRUE;
2615 }
2616
2617 static void
2618 tree_select(struct view *view, struct line *line)
2619 {
2620         if (line->type == LINE_TREE_DIR || line->type == LINE_TREE_FILE) {
2621                 char *text = line->data;
2622
2623                 string_ncopy(view->ref, text + STRING_SIZE("100644 blob "), 40);
2624                 string_copy(ref_blob, view->ref);
2625         }
2626 }
2627
2628 static struct view_ops tree_ops = {
2629         "file",
2630         pager_draw,
2631         tree_read,
2632         tree_enter,
2633         pager_grep,
2634         tree_select,
2635 };
2636
2637 static bool
2638 blob_read(struct view *view, char *line)
2639 {
2640         bool state = pager_read(view, line);
2641
2642         if (state == TRUE)
2643                 view->line[view->lines - 1].type = LINE_DEFAULT;
2644
2645         return state;
2646 }
2647
2648 static struct view_ops blob_ops = {
2649         "line",
2650         pager_draw,
2651         blob_read,
2652         pager_enter,
2653         pager_grep,
2654         pager_select,
2655 };
2656
2657
2658 /*
2659  * Main view backend
2660  */
2661
2662 struct commit {
2663         char id[41];                    /* SHA1 ID. */
2664         char title[75];                 /* First line of the commit message. */
2665         char author[75];                /* Author of the commit. */
2666         struct tm time;                 /* Date from the author ident. */
2667         struct ref **refs;              /* Repository references. */
2668         chtype graph[SIZEOF_REVGRAPH];  /* Ancestry chain graphics. */
2669         size_t graph_size;              /* The width of the graph array. */
2670 };
2671
2672 static bool
2673 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2674 {
2675         char buf[DATE_COLS + 1];
2676         struct commit *commit = line->data;
2677         enum line_type type;
2678         int col = 0;
2679         size_t timelen;
2680         size_t authorlen;
2681         int trimmed = 1;
2682
2683         if (!*commit->author)
2684                 return FALSE;
2685
2686         wmove(view->win, lineno, col);
2687
2688         if (selected) {
2689                 type = LINE_CURSOR;
2690                 wattrset(view->win, get_line_attr(type));
2691                 wchgat(view->win, -1, 0, type, NULL);
2692
2693         } else {
2694                 type = LINE_MAIN_COMMIT;
2695                 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
2696         }
2697
2698         timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
2699         waddnstr(view->win, buf, timelen);
2700         waddstr(view->win, " ");
2701
2702         col += DATE_COLS;
2703         wmove(view->win, lineno, col);
2704         if (type != LINE_CURSOR)
2705                 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
2706
2707         if (opt_utf8) {
2708                 authorlen = utf8_length(commit->author, AUTHOR_COLS - 2, &col, &trimmed);
2709         } else {
2710                 authorlen = strlen(commit->author);
2711                 if (authorlen > AUTHOR_COLS - 2) {
2712                         authorlen = AUTHOR_COLS - 2;
2713                         trimmed = 1;
2714                 }
2715         }
2716
2717         if (trimmed) {
2718                 waddnstr(view->win, commit->author, authorlen);
2719                 if (type != LINE_CURSOR)
2720                         wattrset(view->win, get_line_attr(LINE_MAIN_DELIM));
2721                 waddch(view->win, '~');
2722         } else {
2723                 waddstr(view->win, commit->author);
2724         }
2725
2726         col += AUTHOR_COLS;
2727         if (type != LINE_CURSOR)
2728                 wattrset(view->win, A_NORMAL);
2729
2730         if (opt_rev_graph && commit->graph_size) {
2731                 size_t i;
2732
2733                 wmove(view->win, lineno, col);
2734                 /* Using waddch() instead of waddnstr() ensures that
2735                  * they'll be rendered correctly for the cursor line. */
2736                 for (i = 0; i < commit->graph_size; i++)
2737                         waddch(view->win, commit->graph[i]);
2738
2739                 col += commit->graph_size + 1;
2740         }
2741
2742         wmove(view->win, lineno, col);
2743
2744         if (commit->refs) {
2745                 size_t i = 0;
2746
2747                 do {
2748                         if (type == LINE_CURSOR)
2749                                 ;
2750                         else if (commit->refs[i]->tag)
2751                                 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
2752                         else
2753                                 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
2754                         waddstr(view->win, "[");
2755                         waddstr(view->win, commit->refs[i]->name);
2756                         waddstr(view->win, "]");
2757                         if (type != LINE_CURSOR)
2758                                 wattrset(view->win, A_NORMAL);
2759                         waddstr(view->win, " ");
2760                         col += strlen(commit->refs[i]->name) + STRING_SIZE("[] ");
2761                 } while (commit->refs[i++]->next);
2762         }
2763
2764         if (type != LINE_CURSOR)
2765                 wattrset(view->win, get_line_attr(type));
2766
2767         {
2768                 int titlelen = strlen(commit->title);
2769
2770                 if (col + titlelen > view->width)
2771                         titlelen = view->width - col;
2772
2773                 waddnstr(view->win, commit->title, titlelen);
2774         }
2775
2776         return TRUE;
2777 }
2778
2779 /* Reads git log --pretty=raw output and parses it into the commit struct. */
2780 static bool
2781 main_read(struct view *view, char *line)
2782 {
2783         enum line_type type = get_line_type(line);
2784         struct commit *commit = view->lines
2785                               ? view->line[view->lines - 1].data : NULL;
2786
2787         switch (type) {
2788         case LINE_COMMIT:
2789                 commit = calloc(1, sizeof(struct commit));
2790                 if (!commit)
2791                         return FALSE;
2792
2793                 line += STRING_SIZE("commit ");
2794
2795                 view->line[view->lines++].data = commit;
2796                 string_copy(commit->id, line);
2797                 commit->refs = get_refs(commit->id);
2798                 commit->graph[commit->graph_size++] = ACS_LTEE;
2799                 break;
2800
2801         case LINE_AUTHOR:
2802         {
2803                 char *ident = line + STRING_SIZE("author ");
2804                 char *end = strchr(ident, '<');
2805
2806                 if (!commit)
2807                         break;
2808
2809                 if (end) {
2810                         char *email = end + 1;
2811
2812                         for (; end > ident && isspace(end[-1]); end--) ;
2813
2814                         if (end == ident && *email) {
2815                                 ident = email;
2816                                 end = strchr(ident, '>');
2817                                 for (; end > ident && isspace(end[-1]); end--) ;
2818                         }
2819                         *end = 0;
2820                 }
2821
2822                 /* End is NULL or ident meaning there's no author. */
2823                 if (end <= ident)
2824                         ident = "Unknown";
2825
2826                 string_copy(commit->author, ident);
2827
2828                 /* Parse epoch and timezone */
2829                 if (end) {
2830                         char *secs = strchr(end + 1, '>');
2831                         char *zone;
2832                         time_t time;
2833
2834                         if (!secs || secs[1] != ' ')
2835                                 break;
2836
2837                         secs += 2;
2838                         time = (time_t) atol(secs);
2839                         zone = strchr(secs, ' ');
2840                         if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
2841                                 long tz;
2842
2843                                 zone++;
2844                                 tz  = ('0' - zone[1]) * 60 * 60 * 10;
2845                                 tz += ('0' - zone[2]) * 60 * 60;
2846                                 tz += ('0' - zone[3]) * 60;
2847                                 tz += ('0' - zone[4]) * 60;
2848
2849                                 if (zone[0] == '-')
2850                                         tz = -tz;
2851
2852                                 time -= tz;
2853                         }
2854                         gmtime_r(&time, &commit->time);
2855                 }
2856                 break;
2857         }
2858         default:
2859                 if (!commit)
2860                         break;
2861
2862                 /* Fill in the commit title if it has not already been set. */
2863                 if (commit->title[0])
2864                         break;
2865
2866                 /* Require titles to start with a non-space character at the
2867                  * offset used by git log. */
2868                 /* FIXME: More gracefull handling of titles; append "..." to
2869                  * shortened titles, etc. */
2870                 if (strncmp(line, "    ", 4) ||
2871                     isspace(line[4]))
2872                         break;
2873
2874                 string_copy(commit->title, line + 4);
2875         }
2876
2877         return TRUE;
2878 }
2879
2880 static bool
2881 main_enter(struct view *view, struct line *line)
2882 {
2883         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
2884
2885         open_view(view, REQ_VIEW_DIFF, flags);
2886         return TRUE;
2887 }
2888
2889 static bool
2890 main_grep(struct view *view, struct line *line)
2891 {
2892         struct commit *commit = line->data;
2893         enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
2894         char buf[DATE_COLS + 1];
2895         regmatch_t pmatch;
2896
2897         for (state = S_TITLE; state < S_END; state++) {
2898                 char *text;
2899
2900                 switch (state) {
2901                 case S_TITLE:   text = commit->title;   break;
2902                 case S_AUTHOR:  text = commit->author;  break;
2903                 case S_DATE:
2904                         if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
2905                                 continue;
2906                         text = buf;
2907                         break;
2908
2909                 default:
2910                         return FALSE;
2911                 }
2912
2913                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
2914                         return TRUE;
2915         }
2916
2917         return FALSE;
2918 }
2919
2920 static void
2921 main_select(struct view *view, struct line *line)
2922 {
2923         struct commit *commit = line->data;
2924
2925         string_copy(view->ref, commit->id);
2926         string_copy(ref_commit, view->ref);
2927 }
2928
2929 static struct view_ops main_ops = {
2930         "commit",
2931         main_draw,
2932         main_read,
2933         main_enter,
2934         main_grep,
2935         main_select,
2936 };
2937
2938
2939 /*
2940  * Unicode / UTF-8 handling
2941  *
2942  * NOTE: Much of the following code for dealing with unicode is derived from
2943  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
2944  * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
2945  */
2946
2947 /* I've (over)annotated a lot of code snippets because I am not entirely
2948  * confident that the approach taken by this small UTF-8 interface is correct.
2949  * --jonas */
2950
2951 static inline int
2952 unicode_width(unsigned long c)
2953 {
2954         if (c >= 0x1100 &&
2955            (c <= 0x115f                         /* Hangul Jamo */
2956             || c == 0x2329
2957             || c == 0x232a
2958             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
2959                                                 /* CJK ... Yi */
2960             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
2961             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
2962             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
2963             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
2964             || (c >= 0xffe0  && c <= 0xffe6)
2965             || (c >= 0x20000 && c <= 0x2fffd)
2966             || (c >= 0x30000 && c <= 0x3fffd)))
2967                 return 2;
2968
2969         return 1;
2970 }
2971
2972 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
2973  * Illegal bytes are set one. */
2974 static const unsigned char utf8_bytes[256] = {
2975         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,
2976         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,
2977         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,
2978         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,
2979         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,
2980         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,
2981         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,
2982         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,
2983 };
2984
2985 /* Decode UTF-8 multi-byte representation into a unicode character. */
2986 static inline unsigned long
2987 utf8_to_unicode(const char *string, size_t length)
2988 {
2989         unsigned long unicode;
2990
2991         switch (length) {
2992         case 1:
2993                 unicode  =   string[0];
2994                 break;
2995         case 2:
2996                 unicode  =  (string[0] & 0x1f) << 6;
2997                 unicode +=  (string[1] & 0x3f);
2998                 break;
2999         case 3:
3000                 unicode  =  (string[0] & 0x0f) << 12;
3001                 unicode += ((string[1] & 0x3f) << 6);
3002                 unicode +=  (string[2] & 0x3f);
3003                 break;
3004         case 4:
3005                 unicode  =  (string[0] & 0x0f) << 18;
3006                 unicode += ((string[1] & 0x3f) << 12);
3007                 unicode += ((string[2] & 0x3f) << 6);
3008                 unicode +=  (string[3] & 0x3f);
3009                 break;
3010         case 5:
3011                 unicode  =  (string[0] & 0x0f) << 24;
3012                 unicode += ((string[1] & 0x3f) << 18);
3013                 unicode += ((string[2] & 0x3f) << 12);
3014                 unicode += ((string[3] & 0x3f) << 6);
3015                 unicode +=  (string[4] & 0x3f);
3016                 break;
3017         case 6:
3018                 unicode  =  (string[0] & 0x01) << 30;
3019                 unicode += ((string[1] & 0x3f) << 24);
3020                 unicode += ((string[2] & 0x3f) << 18);
3021                 unicode += ((string[3] & 0x3f) << 12);
3022                 unicode += ((string[4] & 0x3f) << 6);
3023                 unicode +=  (string[5] & 0x3f);
3024                 break;
3025         default:
3026                 die("Invalid unicode length");
3027         }
3028
3029         /* Invalid characters could return the special 0xfffd value but NUL
3030          * should be just as good. */
3031         return unicode > 0xffff ? 0 : unicode;
3032 }
3033
3034 /* Calculates how much of string can be shown within the given maximum width
3035  * and sets trimmed parameter to non-zero value if all of string could not be
3036  * shown.
3037  *
3038  * Additionally, adds to coloffset how many many columns to move to align with
3039  * the expected position. Takes into account how multi-byte and double-width
3040  * characters will effect the cursor position.
3041  *
3042  * Returns the number of bytes to output from string to satisfy max_width. */
3043 static size_t
3044 utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed)
3045 {
3046         const char *start = string;
3047         const char *end = strchr(string, '\0');
3048         size_t mbwidth = 0;
3049         size_t width = 0;
3050
3051         *trimmed = 0;
3052
3053         while (string < end) {
3054                 int c = *(unsigned char *) string;
3055                 unsigned char bytes = utf8_bytes[c];
3056                 size_t ucwidth;
3057                 unsigned long unicode;
3058
3059                 if (string + bytes > end)
3060                         break;
3061
3062                 /* Change representation to figure out whether
3063                  * it is a single- or double-width character. */
3064
3065                 unicode = utf8_to_unicode(string, bytes);
3066                 /* FIXME: Graceful handling of invalid unicode character. */
3067                 if (!unicode)
3068                         break;
3069
3070                 ucwidth = unicode_width(unicode);
3071                 width  += ucwidth;
3072                 if (width > max_width) {
3073                         *trimmed = 1;
3074                         break;
3075                 }
3076
3077                 /* The column offset collects the differences between the
3078                  * number of bytes encoding a character and the number of
3079                  * columns will be used for rendering said character.
3080                  *
3081                  * So if some character A is encoded in 2 bytes, but will be
3082                  * represented on the screen using only 1 byte this will and up
3083                  * adding 1 to the multi-byte column offset.
3084                  *
3085                  * Assumes that no double-width character can be encoding in
3086                  * less than two bytes. */
3087                 if (bytes > ucwidth)
3088                         mbwidth += bytes - ucwidth;
3089
3090                 string  += bytes;
3091         }
3092
3093         *coloffset += mbwidth;
3094
3095         return string - start;
3096 }
3097
3098
3099 /*
3100  * Status management
3101  */
3102
3103 /* Whether or not the curses interface has been initialized. */
3104 static bool cursed = FALSE;
3105
3106 /* The status window is used for polling keystrokes. */
3107 static WINDOW *status_win;
3108
3109 /* Update status and title window. */
3110 static void
3111 report(const char *msg, ...)
3112 {
3113         static bool empty = TRUE;
3114         struct view *view = display[current_view];
3115
3116         if (!empty || *msg) {
3117                 va_list args;
3118
3119                 va_start(args, msg);
3120
3121                 werase(status_win);
3122                 wmove(status_win, 0, 0);
3123                 if (*msg) {
3124                         vwprintw(status_win, msg, args);
3125                         empty = FALSE;
3126                 } else {
3127                         empty = TRUE;
3128                 }
3129                 wrefresh(status_win);
3130
3131                 va_end(args);
3132         }
3133
3134         update_view_title(view);
3135         update_display_cursor();
3136 }
3137
3138 /* Controls when nodelay should be in effect when polling user input. */
3139 static void
3140 set_nonblocking_input(bool loading)
3141 {
3142         static unsigned int loading_views;
3143
3144         if ((loading == FALSE && loading_views-- == 1) ||
3145             (loading == TRUE  && loading_views++ == 0))
3146                 nodelay(status_win, loading);
3147 }
3148
3149 static void
3150 init_display(void)
3151 {
3152         int x, y;
3153
3154         /* Initialize the curses library */
3155         if (isatty(STDIN_FILENO)) {
3156                 cursed = !!initscr();
3157         } else {
3158                 /* Leave stdin and stdout alone when acting as a pager. */
3159                 FILE *io = fopen("/dev/tty", "r+");
3160
3161                 if (!io)
3162                         die("Failed to open /dev/tty");
3163                 cursed = !!newterm(NULL, io, io);
3164         }
3165
3166         if (!cursed)
3167                 die("Failed to initialize curses");
3168
3169         nonl();         /* Tell curses not to do NL->CR/NL on output */
3170         cbreak();       /* Take input chars one at a time, no wait for \n */
3171         noecho();       /* Don't echo input */
3172         leaveok(stdscr, TRUE);
3173
3174         if (has_colors())
3175                 init_colors();
3176
3177         getmaxyx(stdscr, y, x);
3178         status_win = newwin(1, 0, y - 1, 0);
3179         if (!status_win)
3180                 die("Failed to create status window");
3181
3182         /* Enable keyboard mapping */
3183         keypad(status_win, TRUE);
3184         wbkgdset(status_win, get_line_attr(LINE_STATUS));
3185 }
3186
3187 static char *
3188 read_prompt(const char *prompt)
3189 {
3190         enum { READING, STOP, CANCEL } status = READING;
3191         static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
3192         int pos = 0;
3193
3194         while (status == READING) {
3195                 struct view *view;
3196                 int i, key;
3197
3198                 foreach_view (view, i)
3199                         update_view(view);
3200
3201                 report("%s%.*s", prompt, pos, buf);
3202                 /* Refresh, accept single keystroke of input */
3203                 key = wgetch(status_win);
3204                 switch (key) {
3205                 case KEY_RETURN:
3206                 case KEY_ENTER:
3207                 case '\n':
3208                         status = pos ? STOP : CANCEL;
3209                         break;
3210
3211                 case KEY_BACKSPACE:
3212                         if (pos > 0)
3213                                 pos--;
3214                         else
3215                                 status = CANCEL;
3216                         break;
3217
3218                 case KEY_ESC:
3219                         status = CANCEL;
3220                         break;
3221
3222                 case ERR:
3223                         break;
3224
3225                 default:
3226                         if (pos >= sizeof(buf)) {
3227                                 report("Input string too long");
3228                                 return NULL;
3229                         }
3230
3231                         if (isprint(key))
3232                                 buf[pos++] = (char) key;
3233                 }
3234         }
3235
3236         if (status == CANCEL) {
3237                 /* Clear the status window */
3238                 report("");
3239                 return NULL;
3240         }
3241
3242         buf[pos++] = 0;
3243
3244         return buf;
3245 }
3246
3247 /*
3248  * Repository references
3249  */
3250
3251 static struct ref *refs;
3252 static size_t refs_size;
3253
3254 /* Id <-> ref store */
3255 static struct ref ***id_refs;
3256 static size_t id_refs_size;
3257
3258 static struct ref **
3259 get_refs(char *id)
3260 {
3261         struct ref ***tmp_id_refs;
3262         struct ref **ref_list = NULL;
3263         size_t ref_list_size = 0;
3264         size_t i;
3265
3266         for (i = 0; i < id_refs_size; i++)
3267                 if (!strcmp(id, id_refs[i][0]->id))
3268                         return id_refs[i];
3269
3270         tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
3271         if (!tmp_id_refs)
3272                 return NULL;
3273
3274         id_refs = tmp_id_refs;
3275
3276         for (i = 0; i < refs_size; i++) {
3277                 struct ref **tmp;
3278
3279                 if (strcmp(id, refs[i].id))
3280                         continue;
3281
3282                 tmp = realloc(ref_list, (ref_list_size + 1) * sizeof(*ref_list));
3283                 if (!tmp) {
3284                         if (ref_list)
3285                                 free(ref_list);
3286                         return NULL;
3287                 }
3288
3289                 ref_list = tmp;
3290                 if (ref_list_size > 0)
3291                         ref_list[ref_list_size - 1]->next = 1;
3292                 ref_list[ref_list_size] = &refs[i];
3293
3294                 /* XXX: The properties of the commit chains ensures that we can
3295                  * safely modify the shared ref. The repo references will
3296                  * always be similar for the same id. */
3297                 ref_list[ref_list_size]->next = 0;
3298                 ref_list_size++;
3299         }
3300
3301         if (ref_list)
3302                 id_refs[id_refs_size++] = ref_list;
3303
3304         return ref_list;
3305 }
3306
3307 static int
3308 read_ref(char *id, int idlen, char *name, int namelen)
3309 {
3310         struct ref *ref;
3311         bool tag = FALSE;
3312
3313         if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
3314                 /* Commits referenced by tags has "^{}" appended. */
3315                 if (name[namelen - 1] != '}')
3316                         return OK;
3317
3318                 while (namelen > 0 && name[namelen] != '^')
3319                         namelen--;
3320
3321                 tag = TRUE;
3322                 namelen -= STRING_SIZE("refs/tags/");
3323                 name    += STRING_SIZE("refs/tags/");
3324
3325         } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
3326                 namelen -= STRING_SIZE("refs/heads/");
3327                 name    += STRING_SIZE("refs/heads/");
3328
3329         } else if (!strcmp(name, "HEAD")) {
3330                 return OK;
3331         }
3332
3333         refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
3334         if (!refs)
3335                 return ERR;
3336
3337         ref = &refs[refs_size++];
3338         ref->name = malloc(namelen + 1);
3339         if (!ref->name)
3340                 return ERR;
3341
3342         strncpy(ref->name, name, namelen);
3343         ref->name[namelen] = 0;
3344         ref->tag = tag;
3345         string_copy(ref->id, id);
3346
3347         return OK;
3348 }
3349
3350 static int
3351 load_refs(void)
3352 {
3353         const char *cmd_env = getenv("TIG_LS_REMOTE");
3354         const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
3355
3356         return read_properties(popen(cmd, "r"), "\t", read_ref);
3357 }
3358
3359 static int
3360 read_repo_config_option(char *name, int namelen, char *value, int valuelen)
3361 {
3362         if (!strcmp(name, "i18n.commitencoding"))
3363                 string_copy(opt_encoding, value);
3364
3365         return OK;
3366 }
3367
3368 static int
3369 load_repo_config(void)
3370 {
3371         return read_properties(popen("git repo-config --list", "r"),
3372                                "=", read_repo_config_option);
3373 }
3374
3375 static int
3376 read_properties(FILE *pipe, const char *separators,
3377                 int (*read_property)(char *, int, char *, int))
3378 {
3379         char buffer[BUFSIZ];
3380         char *name;
3381         int state = OK;
3382
3383         if (!pipe)
3384                 return ERR;
3385
3386         while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
3387                 char *value;
3388                 size_t namelen;
3389                 size_t valuelen;
3390
3391                 name = chomp_string(name);
3392                 namelen = strcspn(name, separators);
3393
3394                 if (name[namelen]) {
3395                         name[namelen] = 0;
3396                         value = chomp_string(name + namelen + 1);
3397                         valuelen = strlen(value);
3398
3399                 } else {
3400                         value = "";
3401                         valuelen = 0;
3402                 }
3403
3404                 state = read_property(name, namelen, value, valuelen);
3405         }
3406
3407         if (state != ERR && ferror(pipe))
3408                 state = ERR;
3409
3410         pclose(pipe);
3411
3412         return state;
3413 }
3414
3415
3416 /*
3417  * Main
3418  */
3419
3420 static void __NORETURN
3421 quit(int sig)
3422 {
3423         /* XXX: Restore tty modes and let the OS cleanup the rest! */
3424         if (cursed)
3425                 endwin();
3426         exit(0);
3427 }
3428
3429 static void __NORETURN
3430 die(const char *err, ...)
3431 {
3432         va_list args;
3433
3434         endwin();
3435
3436         va_start(args, err);
3437         fputs("tig: ", stderr);
3438         vfprintf(stderr, err, args);
3439         fputs("\n", stderr);
3440         va_end(args);
3441
3442         exit(1);
3443 }
3444
3445 int
3446 main(int argc, char *argv[])
3447 {
3448         struct view *view;
3449         enum request request;
3450         size_t i;
3451
3452         signal(SIGINT, quit);
3453
3454         if (setlocale(LC_ALL, "")) {
3455                 string_copy(opt_codeset, nl_langinfo(CODESET));
3456         }
3457
3458         if (load_options() == ERR)
3459                 die("Failed to load user config.");
3460
3461         /* Load the repo config file so options can be overwritten from
3462          * the command line.  */
3463         if (load_repo_config() == ERR)
3464                 die("Failed to load repo config.");
3465
3466         if (!parse_options(argc, argv))
3467                 return 0;
3468
3469         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
3470                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
3471                 if (opt_iconv == ICONV_NONE)
3472                         die("Failed to initialize character set conversion");
3473         }
3474
3475         if (load_refs() == ERR)
3476                 die("Failed to load refs.");
3477
3478         /* Require a git repository unless when running in pager mode. */
3479         if (refs_size == 0 && opt_request != REQ_VIEW_PAGER)
3480                 die("Not a git repository");
3481
3482         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
3483                 view->cmd_env = getenv(view->cmd_env);
3484
3485         request = opt_request;
3486
3487         init_display();
3488
3489         while (view_driver(display[current_view], request)) {
3490                 int key;
3491                 int i;
3492
3493                 foreach_view (view, i)
3494                         update_view(view);
3495
3496                 /* Refresh, accept single keystroke of input */
3497                 key = wgetch(status_win);
3498
3499                 request = get_keybinding(display[current_view]->keymap, key);
3500
3501                 /* Some low-level request handling. This keeps access to
3502                  * status_win restricted. */
3503                 switch (request) {
3504                 case REQ_PROMPT:
3505                 {
3506                         char *cmd = read_prompt(":");
3507
3508                         if (cmd && string_format(opt_cmd, "git %s", cmd)) {
3509                                 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
3510                                         opt_request = REQ_VIEW_DIFF;
3511                                 } else {
3512                                         opt_request = REQ_VIEW_PAGER;
3513                                 }
3514                                 break;
3515                         }
3516
3517                         request = REQ_NONE;
3518                         break;
3519                 }
3520                 case REQ_SEARCH:
3521                 case REQ_SEARCH_BACK:
3522                 {
3523                         const char *prompt = request == REQ_SEARCH
3524                                            ? "/" : "?";
3525                         char *search = read_prompt(prompt);
3526
3527                         if (search)
3528                                 string_copy(opt_search, search);
3529                         else
3530                                 request = REQ_NONE;
3531                         break;
3532                 }
3533                 case REQ_SCREEN_RESIZE:
3534                 {
3535                         int height, width;
3536
3537                         getmaxyx(stdscr, height, width);
3538
3539                         /* Resize the status view and let the view driver take
3540                          * care of resizing the displayed views. */
3541                         wresize(status_win, 1, width);
3542                         mvwin(status_win, height - 1, 0);
3543                         wrefresh(status_win);
3544                         break;
3545                 }
3546                 default:
3547                         break;
3548                 }
3549         }
3550
3551         quit(0);
3552
3553         return 0;
3554 }