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