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