git.oblomov.eu Git - tig/blob - tig.c

   1 /* Copyright (c) 2006-2007 Jonas Fonseca <fonseca@diku.dk>
   2  *
   3  * This program is free software; you can redistribute it and/or
   4  * modify it under the terms of the GNU General Public License as
   5  * published by the Free Software Foundation; either version 2 of
   6  * the License, or (at your option) any later version.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  */
  13
  14 #ifdef HAVE_CONFIG_H
  15 #include "config.h"
  16 #endif
  17
  18 #ifndef TIG_VERSION
  19 #define TIG_VERSION "unknown-version"
  20 #endif
  21
  22 #ifndef DEBUG
  23 #define NDEBUG
  24 #endif
  25
  26 #include <assert.h>
  27 #include <errno.h>
  28 #include <ctype.h>
  29 #include <signal.h>
  30 #include <stdarg.h>
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34 #include <sys/types.h>
  35 #include <sys/stat.h>
  36 #include <unistd.h>
  37 #include <time.h>
  38
  39 #include <regex.h>
  40
  41 #include <locale.h>
  42 #include <langinfo.h>
  43 #include <iconv.h>
  44
  45 #include <curses.h>
  46
  47 #if __GNUC__ >= 3
  48 #define __NORETURN __attribute__((__noreturn__))
  49 #else
  50 #define __NORETURN
  51 #endif
  52
  53 static void __NORETURN die(const char *err, ...);
  54 static void report(const char *msg, ...);
  55 static int read_properties(FILE *pipe, const char *separators, int (*read)(char *, size_t, char *, size_t));
  56 static void set_nonblocking_input(bool loading);
  57 static size_t utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed);
  58
  59 #define ABS(x)          ((x) >= 0  ? (x) : -(x))
  60 #define MIN(x, y)       ((x) < (y) ? (x) :  (y))
  61
  62 #define ARRAY_SIZE(x)   (sizeof(x) / sizeof(x[0]))
  63 #define STRING_SIZE(x)  (sizeof(x) - 1)
  64
  65 #define SIZEOF_STR      1024    /* Default string size. */
  66 #define SIZEOF_REF      256     /* Size of symbolic or SHA1 ID. */
  67 #define SIZEOF_REV      41      /* Holds a SHA-1 and an ending NUL */
  68
  69 /* Revision graph */
  70
  71 #define REVGRAPH_INIT   'I'
  72 #define REVGRAPH_MERGE  'M'
  73 #define REVGRAPH_BRANCH '+'
  74 #define REVGRAPH_COMMIT '*'
  75 #define REVGRAPH_LINE   '|'
  76
  77 #define SIZEOF_REVGRAPH 19      /* Size of revision ancestry graphics. */
  78
  79 /* This color name can be used to refer to the default term colors. */
  80 #define COLOR_DEFAULT   (-1)
  81
  82 #define ICONV_NONE      ((iconv_t) -1)
  83 #ifndef ICONV_CONST
  84 #define ICONV_CONST     /* nothing */
  85 #endif
  86
  87 /* The format and size of the date column in the main view. */
  88 #define DATE_FORMAT     "%Y-%m-%d %H:%M"
  89 #define DATE_COLS       STRING_SIZE("2006-04-29 14:21 ")
  90
  91 #define AUTHOR_COLS     20
  92
  93 /* The default interval between line numbers. */
  94 #define NUMBER_INTERVAL 1
  95
  96 #define TABSIZE         8
  97
  98 #define SCALE_SPLIT_VIEW(height)        ((height) * 2 / 3)
  99
 100 #ifndef GIT_CONFIG
 101 #define GIT_CONFIG "git config"
 102 #endif
 103
 104 #define TIG_LS_REMOTE \
 105         "git ls-remote $(git rev-parse --git-dir) 2>/dev/null"
 106
 107 #define TIG_DIFF_CMD \
 108         "git show --no-color --root --patch-with-stat --find-copies-harder -B -C %s 2>/dev/null"
 109
 110 #define TIG_LOG_CMD     \
 111         "git log --no-color --cc --stat -n100 %s 2>/dev/null"
 112
 113 #define TIG_MAIN_CMD \
 114         "git log --no-color --topo-order --pretty=raw %s 2>/dev/null"
 115
 116 #define TIG_TREE_CMD    \
 117         "git ls-tree %s %s"
 118
 119 #define TIG_BLOB_CMD    \
 120         "git cat-file blob %s"
 121
 122 /* XXX: Needs to be defined to the empty string. */
 123 #define TIG_HELP_CMD    ""
 124 #define TIG_PAGER_CMD   ""
 125 #define TIG_STATUS_CMD  ""
 126 #define TIG_STAGE_CMD   ""
 127
 128 /* Some ascii-shorthands fitted into the ncurses namespace. */
 129 #define KEY_TAB         '\t'
 130 #define KEY_RETURN      '\r'
 131 #define KEY_ESC         27
 132
 133
 134 struct ref {
 135         char *name;             /* Ref name; tag or head names are shortened. */
 136         char id[SIZEOF_REV];    /* Commit SHA1 ID */
 137         unsigned int tag:1;     /* Is it a tag? */
 138         unsigned int remote:1;  /* Is it a remote ref? */
 139         unsigned int next:1;    /* For ref lists: are there more refs? */
 140 };
 141
 142 static struct ref **get_refs(char *id);
 143
 144 struct int_map {
 145         const char *name;
 146         int namelen;
 147         int value;
 148 };
 149
 150 static int
 151 set_from_int_map(struct int_map *map, size_t map_size,
 152                  int *value, const char *name, int namelen)
 153 {
 154
 155         int i;
 156
 157         for (i = 0; i < map_size; i++)
 158                 if (namelen == map[i].namelen &&
 159                     !strncasecmp(name, map[i].name, namelen)) {
 160                         *value = map[i].value;
 161                         return OK;
 162                 }
 163
 164         return ERR;
 165 }
 166
 167
 168 /*
 169  * String helpers
 170  */
 171
 172 static inline void
 173 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
 174 {
 175         if (srclen > dstlen - 1)
 176                 srclen = dstlen - 1;
 177
 178         strncpy(dst, src, srclen);
 179         dst[srclen] = 0;
 180 }
 181
 182 /* Shorthands for safely copying into a fixed buffer. */
 183
 184 #define string_copy(dst, src) \
 185         string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
 186
 187 #define string_ncopy(dst, src, srclen) \
 188         string_ncopy_do(dst, sizeof(dst), src, srclen)
 189
 190 #define string_copy_rev(dst, src) \
 191         string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
 192
 193 #define string_add(dst, from, src) \
 194         string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
 195
 196 static char *
 197 chomp_string(char *name)
 198 {
 199         int namelen;
 200
 201         while (isspace(*name))
 202                 name++;
 203
 204         namelen = strlen(name) - 1;
 205         while (namelen > 0 && isspace(name[namelen]))
 206                 name[namelen--] = 0;
 207
 208         return name;
 209 }
 210
 211 static bool
 212 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
 213 {
 214         va_list args;
 215         size_t pos = bufpos ? *bufpos : 0;
 216
 217         va_start(args, fmt);
 218         pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
 219         va_end(args);
 220
 221         if (bufpos)
 222                 *bufpos = pos;
 223
 224         return pos >= bufsize ? FALSE : TRUE;
 225 }
 226
 227 #define string_format(buf, fmt, args...) \
 228         string_nformat(buf, sizeof(buf), NULL, fmt, args)
 229
 230 #define string_format_from(buf, from, fmt, args...) \
 231         string_nformat(buf, sizeof(buf), from, fmt, args)
 232
 233 static int
 234 string_enum_compare(const char *str1, const char *str2, int len)
 235 {
 236         size_t i;
 237
 238 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
 239
 240         /* Diff-Header == DIFF_HEADER */
 241         for (i = 0; i < len; i++) {
 242                 if (toupper(str1[i]) == toupper(str2[i]))
 243                         continue;
 244
 245                 if (string_enum_sep(str1[i]) &&
 246                     string_enum_sep(str2[i]))
 247                         continue;
 248
 249                 return str1[i] - str2[i];
 250         }
 251
 252         return 0;
 253 }
 254
 255 /* Shell quoting
 256  *
 257  * NOTE: The following is a slightly modified copy of the git project's shell
 258  * quoting routines found in the quote.c file.
 259  *
 260  * Help to copy the thing properly quoted for the shell safety.  any single
 261  * quote is replaced with '\'', any exclamation point is replaced with '\!',
 262  * and the whole thing is enclosed in a
 263  *
 264  * E.g.
 265  *  original     sq_quote     result
 266  *  name     ==> name      ==> 'name'
 267  *  a b      ==> a b       ==> 'a b'
 268  *  a'b      ==> a'\''b    ==> 'a'\''b'
 269  *  a!b      ==> a'\!'b    ==> 'a'\!'b'
 270  */
 271
 272 static size_t
 273 sq_quote(char buf[SIZEOF_STR], size_t bufsize, const char *src)
 274 {
 275         char c;
 276
 277 #define BUFPUT(x) do { if (bufsize < SIZEOF_STR) buf[bufsize++] = (x); } while (0)
 278
 279         BUFPUT('\'');
 280         while ((c = *src++)) {
 281                 if (c == '\'' || c == '!') {
 282                         BUFPUT('\'');
 283                         BUFPUT('\\');
 284                         BUFPUT(c);
 285                         BUFPUT('\'');
 286                 } else {
 287                         BUFPUT(c);
 288                 }
 289         }
 290         BUFPUT('\'');
 291
 292         if (bufsize < SIZEOF_STR)
 293                 buf[bufsize] = 0;
 294
 295         return bufsize;
 296 }
 297
 298
 299 /*
 300  * User requests
 301  */
 302
 303 #define REQ_INFO \
 304         /* XXX: Keep the view request first and in sync with views[]. */ \
 305         REQ_GROUP("View switching") \
 306         REQ_(VIEW_MAIN,         "Show main view"), \
 307         REQ_(VIEW_DIFF,         "Show diff view"), \
 308         REQ_(VIEW_LOG,          "Show log view"), \
 309         REQ_(VIEW_TREE,         "Show tree view"), \
 310         REQ_(VIEW_BLOB,         "Show blob view"), \
 311         REQ_(VIEW_HELP,         "Show help page"), \
 312         REQ_(VIEW_PAGER,        "Show pager view"), \
 313         REQ_(VIEW_STATUS,       "Show status view"), \
 314         REQ_(VIEW_STAGE,        "Show stage view"), \
 315         \
 316         REQ_GROUP("View manipulation") \
 317         REQ_(ENTER,             "Enter current line and scroll"), \
 318         REQ_(NEXT,              "Move to next"), \
 319         REQ_(PREVIOUS,          "Move to previous"), \
 320         REQ_(VIEW_NEXT,         "Move focus to next view"), \
 321         REQ_(REFRESH,           "Reload and refresh"), \
 322         REQ_(VIEW_CLOSE,        "Close the current view"), \
 323         REQ_(QUIT,              "Close all views and quit"), \
 324         \
 325         REQ_GROUP("Cursor navigation") \
 326         REQ_(MOVE_UP,           "Move cursor one line up"), \
 327         REQ_(MOVE_DOWN,         "Move cursor one line down"), \
 328         REQ_(MOVE_PAGE_DOWN,    "Move cursor one page down"), \
 329         REQ_(MOVE_PAGE_UP,      "Move cursor one page up"), \
 330         REQ_(MOVE_FIRST_LINE,   "Move cursor to first line"), \
 331         REQ_(MOVE_LAST_LINE,    "Move cursor to last line"), \
 332         \
 333         REQ_GROUP("Scrolling") \
 334         REQ_(SCROLL_LINE_UP,    "Scroll one line up"), \
 335         REQ_(SCROLL_LINE_DOWN,  "Scroll one line down"), \
 336         REQ_(SCROLL_PAGE_UP,    "Scroll one page up"), \
 337         REQ_(SCROLL_PAGE_DOWN,  "Scroll one page down"), \
 338         \
 339         REQ_GROUP("Searching") \
 340         REQ_(SEARCH,            "Search the view"), \
 341         REQ_(SEARCH_BACK,       "Search backwards in the view"), \
 342         REQ_(FIND_NEXT,         "Find next search match"), \
 343         REQ_(FIND_PREV,         "Find previous search match"), \
 344         \
 345         REQ_GROUP("Misc") \
 346         REQ_(PROMPT,            "Bring up the prompt"), \
 347         REQ_(SCREEN_REDRAW,     "Redraw the screen"), \
 348         REQ_(SCREEN_RESIZE,     "Resize the screen"), \
 349         REQ_(SHOW_VERSION,      "Show version information"), \
 350         REQ_(STOP_LOADING,      "Stop all loading views"), \
 351         REQ_(TOGGLE_LINENO,     "Toggle line numbers"), \
 352         REQ_(TOGGLE_REV_GRAPH,  "Toggle revision graph visualization"), \
 353         REQ_(STATUS_UPDATE,     "Update file status"), \
 354         REQ_(STATUS_MERGE,      "Merge file using external tool"), \
 355         REQ_(EDIT,              "Open in editor"), \
 356         REQ_(NONE,              "Do nothing")
 357
 358
 359 /* User action requests. */
 360 enum request {
 361 #define REQ_GROUP(help)
 362 #define REQ_(req, help) REQ_##req
 363
 364         /* Offset all requests to avoid conflicts with ncurses getch values. */
 365         REQ_OFFSET = KEY_MAX + 1,
 366         REQ_INFO
 367
 368 #undef  REQ_GROUP
 369 #undef  REQ_
 370 };
 371
 372 struct request_info {
 373         enum request request;
 374         char *name;
 375         int namelen;
 376         char *help;
 377 };
 378
 379 static struct request_info req_info[] = {
 380 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
 381 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
 382         REQ_INFO
 383 #undef  REQ_GROUP
 384 #undef  REQ_
 385 };
 386
 387 static enum request
 388 get_request(const char *name)
 389 {
 390         int namelen = strlen(name);
 391         int i;
 392
 393         for (i = 0; i < ARRAY_SIZE(req_info); i++)
 394                 if (req_info[i].namelen == namelen &&
 395                     !string_enum_compare(req_info[i].name, name, namelen))
 396                         return req_info[i].request;
 397
 398         return REQ_NONE;
 399 }
 400
 401
 402 /*
 403  * Options
 404  */
 405
 406 static const char usage[] =
 407 "tig " TIG_VERSION " (" __DATE__ ")\n"
 408 "\n"
 409 "Usage: tig [options]\n"
 410 "   or: tig [options] [--] [git log options]\n"
 411 "   or: tig [options] log  [git log options]\n"
 412 "   or: tig [options] diff [git diff options]\n"
 413 "   or: tig [options] show [git show options]\n"
 414 "   or: tig [options] <    [git command output]\n"
 415 "\n"
 416 "Options:\n"
 417 "  -l                          Start up in log view\n"
 418 "  -d                          Start up in diff view\n"
 419 "  -S                          Start up in status view\n"
 420 "  -n[I], --line-number[=I]    Show line numbers with given interval\n"
 421 "  -b[N], --tab-size[=N]       Set number of spaces for tab expansion\n"
 422 "  --                          Mark end of tig options\n"
 423 "  -v, --version               Show version and exit\n"
 424 "  -h, --help                  Show help message and exit\n";
 425
 426 /* Option and state variables. */
 427 static bool opt_line_number             = FALSE;
 428 static bool opt_rev_graph               = FALSE;
 429 static int opt_num_interval             = NUMBER_INTERVAL;
 430 static int opt_tab_size                 = TABSIZE;
 431 static enum request opt_request         = REQ_VIEW_MAIN;
 432 static char opt_cmd[SIZEOF_STR]         = "";
 433 static char opt_path[SIZEOF_STR]        = "";
 434 static FILE *opt_pipe                   = NULL;
 435 static char opt_encoding[20]            = "UTF-8";
 436 static bool opt_utf8                    = TRUE;
 437 static char opt_codeset[20]             = "UTF-8";
 438 static iconv_t opt_iconv                = ICONV_NONE;
 439 static char opt_search[SIZEOF_STR]      = "";
 440 static char opt_cdup[SIZEOF_STR]        = "";
 441 static char opt_git_dir[SIZEOF_STR]     = "";
 442 static char opt_is_inside_work_tree     = -1; /* set to TRUE or FALSE */
 443 static char opt_editor[SIZEOF_STR]      = "";
 444
 445 enum option_type {
 446         OPT_NONE,
 447         OPT_INT,
 448 };
 449
 450 static bool
 451 check_option(char *opt, char short_name, char *name, enum option_type type, ...)
 452 {
 453         va_list args;
 454         char *value = "";
 455         int *number;
 456
 457         if (opt[0] != '-')
 458                 return FALSE;
 459
 460         if (opt[1] == '-') {
 461                 int namelen = strlen(name);
 462
 463                 opt += 2;
 464
 465                 if (strncmp(opt, name, namelen))
 466                         return FALSE;
 467
 468                 if (opt[namelen] == '=')
 469                         value = opt + namelen + 1;
 470
 471         } else {
 472                 if (!short_name || opt[1] != short_name)
 473                         return FALSE;
 474                 value = opt + 2;
 475         }
 476
 477         va_start(args, type);
 478         if (type == OPT_INT) {
 479                 number = va_arg(args, int *);
 480                 if (isdigit(*value))
 481                         *number = atoi(value);
 482         }
 483         va_end(args);
 484
 485         return TRUE;
 486 }
 487
 488 /* Returns the index of log or diff command or -1 to exit. */
 489 static bool
 490 parse_options(int argc, char *argv[])
 491 {
 492         int i;
 493
 494         for (i = 1; i < argc; i++) {
 495                 char *opt = argv[i];
 496
 497                 if (!strcmp(opt, "log") ||
 498                     !strcmp(opt, "diff") ||
 499                     !strcmp(opt, "show")) {
 500                         opt_request = opt[0] == 'l'
 501                                     ? REQ_VIEW_LOG : REQ_VIEW_DIFF;
 502                         break;
 503                 }
 504
 505                 if (opt[0] && opt[0] != '-')
 506                         break;
 507
 508                 if (!strcmp(opt, "--")) {
 509                         i++;
 510                         break;
 511                 }
 512
 513                 if (check_option(opt, 'v', "version", OPT_NONE)) {
 514                         printf("tig version %s\n", TIG_VERSION);
 515                         return FALSE;
 516                 }
 517
 518                 if (check_option(opt, 'h', "help", OPT_NONE)) {
 519                         printf(usage);
 520                         return FALSE;
 521                 }
 522
 523                 if (!strcmp(opt, "-S")) {
 524                         opt_request = REQ_VIEW_STATUS;
 525                         continue;
 526                 }
 527
 528                 if (!strcmp(opt, "-l")) {
 529                         opt_request = REQ_VIEW_LOG;
 530                         continue;
 531                 }
 532
 533                 if (!strcmp(opt, "-d")) {
 534                         opt_request = REQ_VIEW_DIFF;
 535                         continue;
 536                 }
 537
 538                 if (check_option(opt, 'n', "line-number", OPT_INT, &opt_num_interval)) {
 539                         opt_line_number = TRUE;
 540                         continue;
 541                 }
 542
 543                 if (check_option(opt, 'b', "tab-size", OPT_INT, &opt_tab_size)) {
 544                         opt_tab_size = MIN(opt_tab_size, TABSIZE);
 545                         continue;
 546                 }
 547
 548                 die("unknown option '%s'\n\n%s", opt, usage);
 549         }
 550
 551         if (!isatty(STDIN_FILENO)) {
 552                 opt_request = REQ_VIEW_PAGER;
 553                 opt_pipe = stdin;
 554
 555         } else if (i < argc) {
 556                 size_t buf_size;
 557
 558                 if (opt_request == REQ_VIEW_MAIN)
 559                         /* XXX: This is vulnerable to the user overriding
 560                          * options required for the main view parser. */
 561                         string_copy(opt_cmd, "git log --no-color --pretty=raw");
 562                 else
 563                         string_copy(opt_cmd, "git");
 564                 buf_size = strlen(opt_cmd);
 565
 566                 while (buf_size < sizeof(opt_cmd) && i < argc) {
 567                         opt_cmd[buf_size++] = ' ';
 568                         buf_size = sq_quote(opt_cmd, buf_size, argv[i++]);
 569                 }
 570
 571                 if (buf_size >= sizeof(opt_cmd))
 572                         die("command too long");
 573
 574                 opt_cmd[buf_size] = 0;
 575         }
 576
 577         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
 578                 opt_utf8 = FALSE;
 579
 580         return TRUE;
 581 }
 582
 583
 584 /*
 585  * Line-oriented content detection.
 586  */
 587
 588 #define LINE_INFO \
 589 LINE(DIFF_HEADER,  "diff --git ",       COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 590 LINE(DIFF_CHUNK,   "@@",                COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 591 LINE(DIFF_ADD,     "+",                 COLOR_GREEN,    COLOR_DEFAULT,  0), \
 592 LINE(DIFF_DEL,     "-",                 COLOR_RED,      COLOR_DEFAULT,  0), \
 593 LINE(DIFF_INDEX,        "index ",         COLOR_BLUE,   COLOR_DEFAULT,  0), \
 594 LINE(DIFF_OLDMODE,      "old file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
 595 LINE(DIFF_NEWMODE,      "new file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
 596 LINE(DIFF_COPY_FROM,    "copy from",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
 597 LINE(DIFF_COPY_TO,      "copy to",        COLOR_YELLOW, COLOR_DEFAULT,  0), \
 598 LINE(DIFF_RENAME_FROM,  "rename from",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
 599 LINE(DIFF_RENAME_TO,    "rename to",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
 600 LINE(DIFF_SIMILARITY,   "similarity ",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
 601 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
 602 LINE(DIFF_TREE,         "diff-tree ",     COLOR_BLUE,   COLOR_DEFAULT,  0), \
 603 LINE(PP_AUTHOR,    "Author: ",          COLOR_CYAN,     COLOR_DEFAULT,  0), \
 604 LINE(PP_COMMIT,    "Commit: ",          COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 605 LINE(PP_MERGE,     "Merge: ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
 606 LINE(PP_DATE,      "Date:   ",          COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 607 LINE(PP_ADATE,     "AuthorDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 608 LINE(PP_CDATE,     "CommitDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 609 LINE(PP_REFS,      "Refs: ",            COLOR_RED,      COLOR_DEFAULT,  0), \
 610 LINE(COMMIT,       "commit ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
 611 LINE(PARENT,       "parent ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
 612 LINE(TREE,         "tree ",             COLOR_BLUE,     COLOR_DEFAULT,  0), \
 613 LINE(AUTHOR,       "author ",           COLOR_CYAN,     COLOR_DEFAULT,  0), \
 614 LINE(COMMITTER,    "committer ",        COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 615 LINE(SIGNOFF,      "    Signed-off-by", COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 616 LINE(ACKED,        "    Acked-by",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 617 LINE(DEFAULT,      "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
 618 LINE(CURSOR,       "",                  COLOR_WHITE,    COLOR_GREEN,    A_BOLD), \
 619 LINE(STATUS,       "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
 620 LINE(TITLE_BLUR,   "",                  COLOR_WHITE,    COLOR_BLUE,     0), \
 621 LINE(TITLE_FOCUS,  "",                  COLOR_WHITE,    COLOR_BLUE,     A_BOLD), \
 622 LINE(MAIN_DATE,    "",                  COLOR_BLUE,     COLOR_DEFAULT,  0), \
 623 LINE(MAIN_AUTHOR,  "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
 624 LINE(MAIN_COMMIT,  "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
 625 LINE(MAIN_DELIM,   "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 626 LINE(MAIN_TAG,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  A_BOLD), \
 627 LINE(MAIN_REMOTE,  "",                  COLOR_YELLOW,   COLOR_DEFAULT,  A_BOLD), \
 628 LINE(MAIN_REF,     "",                  COLOR_CYAN,     COLOR_DEFAULT,  A_BOLD), \
 629 LINE(TREE_DIR,     "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
 630 LINE(TREE_FILE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
 631 LINE(STAT_SECTION, "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
 632 LINE(STAT_NONE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
 633 LINE(STAT_STAGED,  "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 634 LINE(STAT_UNSTAGED,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 635 LINE(STAT_UNTRACKED,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0)
 636
 637 enum line_type {
 638 #define LINE(type, line, fg, bg, attr) \
 639         LINE_##type
 640         LINE_INFO
 641 #undef  LINE
 642 };
 643
 644 struct line_info {
 645         const char *name;       /* Option name. */
 646         int namelen;            /* Size of option name. */
 647         const char *line;       /* The start of line to match. */
 648         int linelen;            /* Size of string to match. */
 649         int fg, bg, attr;       /* Color and text attributes for the lines. */
 650 };
 651
 652 static struct line_info line_info[] = {
 653 #define LINE(type, line, fg, bg, attr) \
 654         { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
 655         LINE_INFO
 656 #undef  LINE
 657 };
 658
 659 static enum line_type
 660 get_line_type(char *line)
 661 {
 662         int linelen = strlen(line);
 663         enum line_type type;
 664
 665         for (type = 0; type < ARRAY_SIZE(line_info); type++)
 666                 /* Case insensitive search matches Signed-off-by lines better. */
 667                 if (linelen >= line_info[type].linelen &&
 668                     !strncasecmp(line_info[type].line, line, line_info[type].linelen))
 669                         return type;
 670
 671         return LINE_DEFAULT;
 672 }
 673
 674 static inline int
 675 get_line_attr(enum line_type type)
 676 {
 677         assert(type < ARRAY_SIZE(line_info));
 678         return COLOR_PAIR(type) | line_info[type].attr;
 679 }
 680
 681 static struct line_info *
 682 get_line_info(char *name, int namelen)
 683 {
 684         enum line_type type;
 685
 686         for (type = 0; type < ARRAY_SIZE(line_info); type++)
 687                 if (namelen == line_info[type].namelen &&
 688                     !string_enum_compare(line_info[type].name, name, namelen))
 689                         return &line_info[type];
 690
 691         return NULL;
 692 }
 693
 694 static void
 695 init_colors(void)
 696 {
 697         int default_bg = COLOR_BLACK;
 698         int default_fg = COLOR_WHITE;
 699         enum line_type type;
 700
 701         start_color();
 702
 703         if (use_default_colors() != ERR) {
 704                 default_bg = -1;
 705                 default_fg = -1;
 706         }
 707
 708         for (type = 0; type < ARRAY_SIZE(line_info); type++) {
 709                 struct line_info *info = &line_info[type];
 710                 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
 711                 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
 712
 713                 init_pair(type, fg, bg);
 714         }
 715 }
 716
 717 struct line {
 718         enum line_type type;
 719
 720         /* State flags */
 721         unsigned int selected:1;
 722
 723         void *data;             /* User data */
 724 };
 725
 726
 727 /*
 728  * Keys
 729  */
 730
 731 struct keybinding {
 732         int alias;
 733         enum request request;
 734         struct keybinding *next;
 735 };
 736
 737 static struct keybinding default_keybindings[] = {
 738         /* View switching */
 739         { 'm',          REQ_VIEW_MAIN },
 740         { 'd',          REQ_VIEW_DIFF },
 741         { 'l',          REQ_VIEW_LOG },
 742         { 't',          REQ_VIEW_TREE },
 743         { 'f',          REQ_VIEW_BLOB },
 744         { 'p',          REQ_VIEW_PAGER },
 745         { 'h',          REQ_VIEW_HELP },
 746         { 'S',          REQ_VIEW_STATUS },
 747         { 'c',          REQ_VIEW_STAGE },
 748
 749         /* View manipulation */
 750         { 'q',          REQ_VIEW_CLOSE },
 751         { KEY_TAB,      REQ_VIEW_NEXT },
 752         { KEY_RETURN,   REQ_ENTER },
 753         { KEY_UP,       REQ_PREVIOUS },
 754         { KEY_DOWN,     REQ_NEXT },
 755         { 'R',          REQ_REFRESH },
 756
 757         /* Cursor navigation */
 758         { 'k',          REQ_MOVE_UP },
 759         { 'j',          REQ_MOVE_DOWN },
 760         { KEY_HOME,     REQ_MOVE_FIRST_LINE },
 761         { KEY_END,      REQ_MOVE_LAST_LINE },
 762         { KEY_NPAGE,    REQ_MOVE_PAGE_DOWN },
 763         { ' ',          REQ_MOVE_PAGE_DOWN },
 764         { KEY_PPAGE,    REQ_MOVE_PAGE_UP },
 765         { 'b',          REQ_MOVE_PAGE_UP },
 766         { '-',          REQ_MOVE_PAGE_UP },
 767
 768         /* Scrolling */
 769         { KEY_IC,       REQ_SCROLL_LINE_UP },
 770         { KEY_DC,       REQ_SCROLL_LINE_DOWN },
 771         { 'w',          REQ_SCROLL_PAGE_UP },
 772         { 's',          REQ_SCROLL_PAGE_DOWN },
 773
 774         /* Searching */
 775         { '/',          REQ_SEARCH },
 776         { '?',          REQ_SEARCH_BACK },
 777         { 'n',          REQ_FIND_NEXT },
 778         { 'N',          REQ_FIND_PREV },
 779
 780         /* Misc */
 781         { 'Q',          REQ_QUIT },
 782         { 'z',          REQ_STOP_LOADING },
 783         { 'v',          REQ_SHOW_VERSION },
 784         { 'r',          REQ_SCREEN_REDRAW },
 785         { '.',          REQ_TOGGLE_LINENO },
 786         { 'g',          REQ_TOGGLE_REV_GRAPH },
 787         { ':',          REQ_PROMPT },
 788         { 'u',          REQ_STATUS_UPDATE },
 789         { 'M',          REQ_STATUS_MERGE },
 790         { 'e',          REQ_EDIT },
 791
 792         /* Using the ncurses SIGWINCH handler. */
 793         { KEY_RESIZE,   REQ_SCREEN_RESIZE },
 794 };
 795
 796 #define KEYMAP_INFO \
 797         KEYMAP_(GENERIC), \
 798         KEYMAP_(MAIN), \
 799         KEYMAP_(DIFF), \
 800         KEYMAP_(LOG), \
 801         KEYMAP_(TREE), \
 802         KEYMAP_(BLOB), \
 803         KEYMAP_(PAGER), \
 804         KEYMAP_(HELP), \
 805         KEYMAP_(STATUS), \
 806         KEYMAP_(STAGE)
 807
 808 enum keymap {
 809 #define KEYMAP_(name) KEYMAP_##name
 810         KEYMAP_INFO
 811 #undef  KEYMAP_
 812 };
 813
 814 static struct int_map keymap_table[] = {
 815 #define KEYMAP_(name) { #name, STRING_SIZE(#name), KEYMAP_##name }
 816         KEYMAP_INFO
 817 #undef  KEYMAP_
 818 };
 819
 820 #define set_keymap(map, name) \
 821         set_from_int_map(keymap_table, ARRAY_SIZE(keymap_table), map, name, strlen(name))
 822
 823 static struct keybinding *keybindings[ARRAY_SIZE(keymap_table)];
 824
 825 static void
 826 add_keybinding(enum keymap keymap, enum request request, int key)
 827 {
 828         struct keybinding *keybinding;
 829
 830         keybinding = calloc(1, sizeof(*keybinding));
 831         if (!keybinding)
 832                 die("Failed to allocate keybinding");
 833
 834         keybinding->alias = key;
 835         keybinding->request = request;
 836         keybinding->next = keybindings[keymap];
 837         keybindings[keymap] = keybinding;
 838 }
 839
 840 /* Looks for a key binding first in the given map, then in the generic map, and
 841  * lastly in the default keybindings. */
 842 static enum request
 843 get_keybinding(enum keymap keymap, int key)
 844 {
 845         struct keybinding *kbd;
 846         int i;
 847
 848         for (kbd = keybindings[keymap]; kbd; kbd = kbd->next)
 849                 if (kbd->alias == key)
 850                         return kbd->request;
 851
 852         for (kbd = keybindings[KEYMAP_GENERIC]; kbd; kbd = kbd->next)
 853                 if (kbd->alias == key)
 854                         return kbd->request;
 855
 856         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
 857                 if (default_keybindings[i].alias == key)
 858                         return default_keybindings[i].request;
 859
 860         return (enum request) key;
 861 }
 862
 863
 864 struct key {
 865         char *name;
 866         int value;
 867 };
 868
 869 static struct key key_table[] = {
 870         { "Enter",      KEY_RETURN },
 871         { "Space",      ' ' },
 872         { "Backspace",  KEY_BACKSPACE },
 873         { "Tab",        KEY_TAB },
 874         { "Escape",     KEY_ESC },
 875         { "Left",       KEY_LEFT },
 876         { "Right",      KEY_RIGHT },
 877         { "Up",         KEY_UP },
 878         { "Down",       KEY_DOWN },
 879         { "Insert",     KEY_IC },
 880         { "Delete",     KEY_DC },
 881         { "Hash",       '#' },
 882         { "Home",       KEY_HOME },
 883         { "End",        KEY_END },
 884         { "PageUp",     KEY_PPAGE },
 885         { "PageDown",   KEY_NPAGE },
 886         { "F1",         KEY_F(1) },
 887         { "F2",         KEY_F(2) },
 888         { "F3",         KEY_F(3) },
 889         { "F4",         KEY_F(4) },
 890         { "F5",         KEY_F(5) },
 891         { "F6",         KEY_F(6) },
 892         { "F7",         KEY_F(7) },
 893         { "F8",         KEY_F(8) },
 894         { "F9",         KEY_F(9) },
 895         { "F10",        KEY_F(10) },
 896         { "F11",        KEY_F(11) },
 897         { "F12",        KEY_F(12) },
 898 };
 899
 900 static int
 901 get_key_value(const char *name)
 902 {
 903         int i;
 904
 905         for (i = 0; i < ARRAY_SIZE(key_table); i++)
 906                 if (!strcasecmp(key_table[i].name, name))
 907                         return key_table[i].value;
 908
 909         if (strlen(name) == 1 && isprint(*name))
 910                 return (int) *name;
 911
 912         return ERR;
 913 }
 914
 915 static char *
 916 get_key_name(int key_value)
 917 {
 918         static char key_char[] = "'X'";
 919         char *seq = NULL;
 920         int key;
 921
 922         for (key = 0; key < ARRAY_SIZE(key_table); key++)
 923                 if (key_table[key].value == key_value)
 924                         seq = key_table[key].name;
 925
 926         if (seq == NULL &&
 927             key_value < 127 &&
 928             isprint(key_value)) {
 929                 key_char[1] = (char) key_value;
 930                 seq = key_char;
 931         }
 932
 933         return seq ? seq : "'?'";
 934 }
 935
 936 static char *
 937 get_key(enum request request)
 938 {
 939         static char buf[BUFSIZ];
 940         size_t pos = 0;
 941         char *sep = "";
 942         int i;
 943
 944         buf[pos] = 0;
 945
 946         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
 947                 struct keybinding *keybinding = &default_keybindings[i];
 948
 949                 if (keybinding->request != request)
 950                         continue;
 951
 952                 if (!string_format_from(buf, &pos, "%s%s", sep,
 953                                         get_key_name(keybinding->alias)))
 954                         return "Too many keybindings!";
 955                 sep = ", ";
 956         }
 957
 958         return buf;
 959 }
 960
 961 struct run_request {
 962         enum keymap keymap;
 963         int key;
 964         char cmd[SIZEOF_STR];
 965 };
 966
 967 static struct run_request *run_request;
 968 static size_t run_requests;
 969
 970 static enum request
 971 add_run_request(enum keymap keymap, int key, int argc, char **argv)
 972 {
 973         struct run_request *tmp;
 974         struct run_request req = { keymap, key };
 975         size_t bufpos;
 976
 977         for (bufpos = 0; argc > 0; argc--, argv++)
 978                 if (!string_format_from(req.cmd, &bufpos, "%s ", *argv))
 979                         return REQ_NONE;
 980
 981         req.cmd[bufpos - 1] = 0;
 982
 983         tmp = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
 984         if (!tmp)
 985                 return REQ_NONE;
 986
 987         run_request = tmp;
 988         run_request[run_requests++] = req;
 989
 990         return REQ_NONE + run_requests;
 991 }
 992
 993 static struct run_request *
 994 get_run_request(enum request request)
 995 {
 996         if (request <= REQ_NONE)
 997                 return NULL;
 998         return &run_request[request - REQ_NONE - 1];
 999 }
1000
1001 static void
1002 add_builtin_run_requests(void)
1003 {
1004         struct {
1005                 enum keymap keymap;
1006                 int key;
1007                 char *argv[1];
1008         } reqs[] = {
1009                 { KEYMAP_MAIN,    'C', { "git cherry-pick %(commit)" } },
1010                 { KEYMAP_GENERIC, 'G', { "git gc" } },
1011         };
1012         int i;
1013
1014         for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1015                 enum request req;
1016
1017                 req = add_run_request(reqs[i].keymap, reqs[i].key, 1, reqs[i].argv);
1018                 if (req != REQ_NONE)
1019                         add_keybinding(reqs[i].keymap, req, reqs[i].key);
1020         }
1021 }
1022
1023 /*
1024  * User config file handling.
1025  */
1026
1027 static struct int_map color_map[] = {
1028 #define COLOR_MAP(name) { #name, STRING_SIZE(#name), COLOR_##name }
1029         COLOR_MAP(DEFAULT),
1030         COLOR_MAP(BLACK),
1031         COLOR_MAP(BLUE),
1032         COLOR_MAP(CYAN),
1033         COLOR_MAP(GREEN),
1034         COLOR_MAP(MAGENTA),
1035         COLOR_MAP(RED),
1036         COLOR_MAP(WHITE),
1037         COLOR_MAP(YELLOW),
1038 };
1039
1040 #define set_color(color, name) \
1041         set_from_int_map(color_map, ARRAY_SIZE(color_map), color, name, strlen(name))
1042
1043 static struct int_map attr_map[] = {
1044 #define ATTR_MAP(name) { #name, STRING_SIZE(#name), A_##name }
1045         ATTR_MAP(NORMAL),
1046         ATTR_MAP(BLINK),
1047         ATTR_MAP(BOLD),
1048         ATTR_MAP(DIM),
1049         ATTR_MAP(REVERSE),
1050         ATTR_MAP(STANDOUT),
1051         ATTR_MAP(UNDERLINE),
1052 };
1053
1054 #define set_attribute(attr, name) \
1055         set_from_int_map(attr_map, ARRAY_SIZE(attr_map), attr, name, strlen(name))
1056
1057 static int   config_lineno;
1058 static bool  config_errors;
1059 static char *config_msg;
1060
1061 /* Wants: object fgcolor bgcolor [attr] */
1062 static int
1063 option_color_command(int argc, char *argv[])
1064 {
1065         struct line_info *info;
1066
1067         if (argc != 3 && argc != 4) {
1068                 config_msg = "Wrong number of arguments given to color command";
1069                 return ERR;
1070         }
1071
1072         info = get_line_info(argv[0], strlen(argv[0]));
1073         if (!info) {
1074                 config_msg = "Unknown color name";
1075                 return ERR;
1076         }
1077
1078         if (set_color(&info->fg, argv[1]) == ERR ||
1079             set_color(&info->bg, argv[2]) == ERR) {
1080                 config_msg = "Unknown color";
1081                 return ERR;
1082         }
1083
1084         if (argc == 4 && set_attribute(&info->attr, argv[3]) == ERR) {
1085                 config_msg = "Unknown attribute";
1086                 return ERR;
1087         }
1088
1089         return OK;
1090 }
1091
1092 /* Wants: name = value */
1093 static int
1094 option_set_command(int argc, char *argv[])
1095 {
1096         if (argc != 3) {
1097                 config_msg = "Wrong number of arguments given to set command";
1098                 return ERR;
1099         }
1100
1101         if (strcmp(argv[1], "=")) {
1102                 config_msg = "No value assigned";
1103                 return ERR;
1104         }
1105
1106         if (!strcmp(argv[0], "show-rev-graph")) {
1107                 opt_rev_graph = (!strcmp(argv[2], "1") ||
1108                                  !strcmp(argv[2], "true") ||
1109                                  !strcmp(argv[2], "yes"));
1110                 return OK;
1111         }
1112
1113         if (!strcmp(argv[0], "line-number-interval")) {
1114                 opt_num_interval = atoi(argv[2]);
1115                 return OK;
1116         }
1117
1118         if (!strcmp(argv[0], "tab-size")) {
1119                 opt_tab_size = atoi(argv[2]);
1120                 return OK;
1121         }
1122
1123         if (!strcmp(argv[0], "commit-encoding")) {
1124                 char *arg = argv[2];
1125                 int delimiter = *arg;
1126                 int i;
1127
1128                 switch (delimiter) {
1129                 case '"':
1130                 case '\'':
1131                         for (arg++, i = 0; arg[i]; i++)
1132                                 if (arg[i] == delimiter) {
1133                                         arg[i] = 0;
1134                                         break;
1135                                 }
1136                 default:
1137                         string_ncopy(opt_encoding, arg, strlen(arg));
1138                         return OK;
1139                 }
1140         }
1141
1142         config_msg = "Unknown variable name";
1143         return ERR;
1144 }
1145
1146 /* Wants: mode request key */
1147 static int
1148 option_bind_command(int argc, char *argv[])
1149 {
1150         enum request request;
1151         int keymap;
1152         int key;
1153
1154         if (argc < 3) {
1155                 config_msg = "Wrong number of arguments given to bind command";
1156                 return ERR;
1157         }
1158
1159         if (set_keymap(&keymap, argv[0]) == ERR) {
1160                 config_msg = "Unknown key map";
1161                 return ERR;
1162         }
1163
1164         key = get_key_value(argv[1]);
1165         if (key == ERR) {
1166                 config_msg = "Unknown key";
1167                 return ERR;
1168         }
1169
1170         request = get_request(argv[2]);
1171         if (request == REQ_NONE) {
1172                 const char *obsolete[] = { "cherry-pick" };
1173                 size_t namelen = strlen(argv[2]);
1174                 int i;
1175
1176                 for (i = 0; i < ARRAY_SIZE(obsolete); i++) {
1177                         if (namelen == strlen(obsolete[i]) &&
1178                             !string_enum_compare(obsolete[i], argv[2], namelen)) {
1179                                 config_msg = "Obsolete request name";
1180                                 return ERR;
1181                         }
1182                 }
1183         }
1184         if (request == REQ_NONE && *argv[2]++ == '!')
1185                 request = add_run_request(keymap, key, argc - 2, argv + 2);
1186         if (request == REQ_NONE) {
1187                 config_msg = "Unknown request name";
1188                 return ERR;
1189         }
1190
1191         add_keybinding(keymap, request, key);
1192
1193         return OK;
1194 }
1195
1196 static int
1197 set_option(char *opt, char *value)
1198 {
1199         char *argv[16];
1200         int valuelen;
1201         int argc = 0;
1202
1203         /* Tokenize */
1204         while (argc < ARRAY_SIZE(argv) && (valuelen = strcspn(value, " \t"))) {
1205                 argv[argc++] = value;
1206                 value += valuelen;
1207
1208                 /* Nothing more to tokenize or last available token. */
1209                 if (!*value || argc >= ARRAY_SIZE(argv))
1210                         break;
1211
1212                 *value++ = 0;
1213                 while (isspace(*value))
1214                         value++;
1215         }
1216
1217         if (!strcmp(opt, "color"))
1218                 return option_color_command(argc, argv);
1219
1220         if (!strcmp(opt, "set"))
1221                 return option_set_command(argc, argv);
1222
1223         if (!strcmp(opt, "bind"))
1224                 return option_bind_command(argc, argv);
1225
1226         config_msg = "Unknown option command";
1227         return ERR;
1228 }
1229
1230 static int
1231 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1232 {
1233         int status = OK;
1234
1235         config_lineno++;
1236         config_msg = "Internal error";
1237
1238         /* Check for comment markers, since read_properties() will
1239          * only ensure opt and value are split at first " \t". */
1240         optlen = strcspn(opt, "#");
1241         if (optlen == 0)
1242                 return OK;
1243
1244         if (opt[optlen] != 0) {
1245                 config_msg = "No option value";
1246                 status = ERR;
1247
1248         }  else {
1249                 /* Look for comment endings in the value. */
1250                 size_t len = strcspn(value, "#");
1251
1252                 if (len < valuelen) {
1253                         valuelen = len;
1254                         value[valuelen] = 0;
1255                 }
1256
1257                 status = set_option(opt, value);
1258         }
1259
1260         if (status == ERR) {
1261                 fprintf(stderr, "Error on line %d, near '%.*s': %s\n",
1262                         config_lineno, (int) optlen, opt, config_msg);
1263                 config_errors = TRUE;
1264         }
1265
1266         /* Always keep going if errors are encountered. */
1267         return OK;
1268 }
1269
1270 static int
1271 load_options(void)
1272 {
1273         char *home = getenv("HOME");
1274         char buf[SIZEOF_STR];
1275         FILE *file;
1276
1277         config_lineno = 0;
1278         config_errors = FALSE;
1279
1280         add_builtin_run_requests();
1281
1282         if (!home || !string_format(buf, "%s/.tigrc", home))
1283                 return ERR;
1284
1285         /* It's ok that the file doesn't exist. */
1286         file = fopen(buf, "r");
1287         if (!file)
1288                 return OK;
1289
1290         if (read_properties(file, " \t", read_option) == ERR ||
1291             config_errors == TRUE)
1292                 fprintf(stderr, "Errors while loading %s.\n", buf);
1293
1294         return OK;
1295 }
1296
1297
1298 /*
1299  * The viewer
1300  */
1301
1302 struct view;
1303 struct view_ops;
1304
1305 /* The display array of active views and the index of the current view. */
1306 static struct view *display[2];
1307 static unsigned int current_view;
1308
1309 /* Reading from the prompt? */
1310 static bool input_mode = FALSE;
1311
1312 #define foreach_displayed_view(view, i) \
1313         for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1314
1315 #define displayed_views()       (display[1] != NULL ? 2 : 1)
1316
1317 /* Current head and commit ID */
1318 static char ref_blob[SIZEOF_REF]        = "";
1319 static char ref_commit[SIZEOF_REF]      = "HEAD";
1320 static char ref_head[SIZEOF_REF]        = "HEAD";
1321
1322 struct view {
1323         const char *name;       /* View name */
1324         const char *cmd_fmt;    /* Default command line format */
1325         const char *cmd_env;    /* Command line set via environment */
1326         const char *id;         /* Points to either of ref_{head,commit,blob} */
1327
1328         struct view_ops *ops;   /* View operations */
1329
1330         enum keymap keymap;     /* What keymap does this view have */
1331
1332         char cmd[SIZEOF_STR];   /* Command buffer */
1333         char ref[SIZEOF_REF];   /* Hovered commit reference */
1334         char vid[SIZEOF_REF];   /* View ID. Set to id member when updating. */
1335
1336         int height, width;      /* The width and height of the main window */
1337         WINDOW *win;            /* The main window */
1338         WINDOW *title;          /* The title window living below the main window */
1339
1340         /* Navigation */
1341         unsigned long offset;   /* Offset of the window top */
1342         unsigned long lineno;   /* Current line number */
1343
1344         /* Searching */
1345         char grep[SIZEOF_STR];  /* Search string */
1346         regex_t *regex;         /* Pre-compiled regex */
1347
1348         /* If non-NULL, points to the view that opened this view. If this view
1349          * is closed tig will switch back to the parent view. */
1350         struct view *parent;
1351
1352         /* Buffering */
1353         unsigned long lines;    /* Total number of lines */
1354         struct line *line;      /* Line index */
1355         unsigned long line_size;/* Total number of allocated lines */
1356         unsigned int digits;    /* Number of digits in the lines member. */
1357
1358         /* Loading */
1359         FILE *pipe;
1360         time_t start_time;
1361 };
1362
1363 struct view_ops {
1364         /* What type of content being displayed. Used in the title bar. */
1365         const char *type;
1366         /* Open and reads in all view content. */
1367         bool (*open)(struct view *view);
1368         /* Read one line; updates view->line. */
1369         bool (*read)(struct view *view, char *data);
1370         /* Draw one line; @lineno must be < view->height. */
1371         bool (*draw)(struct view *view, struct line *line, unsigned int lineno, bool selected);
1372         /* Depending on view handle a special requests. */
1373         enum request (*request)(struct view *view, enum request request, struct line *line);
1374         /* Search for regex in a line. */
1375         bool (*grep)(struct view *view, struct line *line);
1376         /* Select line */
1377         void (*select)(struct view *view, struct line *line);
1378 };
1379
1380 static struct view_ops pager_ops;
1381 static struct view_ops main_ops;
1382 static struct view_ops tree_ops;
1383 static struct view_ops blob_ops;
1384 static struct view_ops help_ops;
1385 static struct view_ops status_ops;
1386 static struct view_ops stage_ops;
1387
1388 #define VIEW_STR(name, cmd, env, ref, ops, map) \
1389         { name, cmd, #env, ref, ops, map}
1390
1391 #define VIEW_(id, name, ops, ref) \
1392         VIEW_STR(name, TIG_##id##_CMD,  TIG_##id##_CMD, ref, ops, KEYMAP_##id)
1393
1394
1395 static struct view views[] = {
1396         VIEW_(MAIN,   "main",   &main_ops,   ref_head),
1397         VIEW_(DIFF,   "diff",   &pager_ops,  ref_commit),
1398         VIEW_(LOG,    "log",    &pager_ops,  ref_head),
1399         VIEW_(TREE,   "tree",   &tree_ops,   ref_commit),
1400         VIEW_(BLOB,   "blob",   &blob_ops,   ref_blob),
1401         VIEW_(HELP,   "help",   &help_ops,   ""),
1402         VIEW_(PAGER,  "pager",  &pager_ops,  "stdin"),
1403         VIEW_(STATUS, "status", &status_ops, ""),
1404         VIEW_(STAGE,  "stage",  &stage_ops,  ""),
1405 };
1406
1407 #define VIEW(req) (&views[(req) - REQ_OFFSET - 1])
1408
1409 #define foreach_view(view, i) \
1410         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1411
1412 #define view_is_displayed(view) \
1413         (view == display[0] || view == display[1])
1414
1415 static bool
1416 draw_view_line(struct view *view, unsigned int lineno)
1417 {
1418         struct line *line;
1419         bool selected = (view->offset + lineno == view->lineno);
1420         bool draw_ok;
1421
1422         assert(view_is_displayed(view));
1423
1424         if (view->offset + lineno >= view->lines)
1425                 return FALSE;
1426
1427         line = &view->line[view->offset + lineno];
1428
1429         if (selected) {
1430                 line->selected = TRUE;
1431                 view->ops->select(view, line);
1432         } else if (line->selected) {
1433                 line->selected = FALSE;
1434                 wmove(view->win, lineno, 0);
1435                 wclrtoeol(view->win);
1436         }
1437
1438         scrollok(view->win, FALSE);
1439         draw_ok = view->ops->draw(view, line, lineno, selected);
1440         scrollok(view->win, TRUE);
1441
1442         return draw_ok;
1443 }
1444
1445 static void
1446 redraw_view_from(struct view *view, int lineno)
1447 {
1448         assert(0 <= lineno && lineno < view->height);
1449
1450         for (; lineno < view->height; lineno++) {
1451                 if (!draw_view_line(view, lineno))
1452                         break;
1453         }
1454
1455         redrawwin(view->win);
1456         if (input_mode)
1457                 wnoutrefresh(view->win);
1458         else
1459                 wrefresh(view->win);
1460 }
1461
1462 static void
1463 redraw_view(struct view *view)
1464 {
1465         wclear(view->win);
1466         redraw_view_from(view, 0);
1467 }
1468
1469
1470 static void
1471 update_view_title(struct view *view)
1472 {
1473         char buf[SIZEOF_STR];
1474         char state[SIZEOF_STR];
1475         size_t bufpos = 0, statelen = 0;
1476
1477         assert(view_is_displayed(view));
1478
1479         if (view != VIEW(REQ_VIEW_STATUS) && (view->lines || view->pipe)) {
1480                 unsigned int view_lines = view->offset + view->height;
1481                 unsigned int lines = view->lines
1482                                    ? MIN(view_lines, view->lines) * 100 / view->lines
1483                                    : 0;
1484
1485                 string_format_from(state, &statelen, "- %s %d of %d (%d%%)",
1486                                    view->ops->type,
1487                                    view->lineno + 1,
1488                                    view->lines,
1489                                    lines);
1490
1491                 if (view->pipe) {
1492                         time_t secs = time(NULL) - view->start_time;
1493
1494                         /* Three git seconds are a long time ... */
1495                         if (secs > 2)
1496                                 string_format_from(state, &statelen, " %lds", secs);
1497                 }
1498         }
1499
1500         string_format_from(buf, &bufpos, "[%s]", view->name);
1501         if (*view->ref && bufpos < view->width) {
1502                 size_t refsize = strlen(view->ref);
1503                 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
1504
1505                 if (minsize < view->width)
1506                         refsize = view->width - minsize + 7;
1507                 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
1508         }
1509
1510         if (statelen && bufpos < view->width) {
1511                 string_format_from(buf, &bufpos, " %s", state);
1512         }
1513
1514         if (view == display[current_view])
1515                 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
1516         else
1517                 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
1518
1519         mvwaddnstr(view->title, 0, 0, buf, bufpos);
1520         wclrtoeol(view->title);
1521         wmove(view->title, 0, view->width - 1);
1522
1523         if (input_mode)
1524                 wnoutrefresh(view->title);
1525         else
1526                 wrefresh(view->title);
1527 }
1528
1529 static void
1530 resize_display(void)
1531 {
1532         int offset, i;
1533         struct view *base = display[0];
1534         struct view *view = display[1] ? display[1] : display[0];
1535
1536         /* Setup window dimensions */
1537
1538         getmaxyx(stdscr, base->height, base->width);
1539
1540         /* Make room for the status window. */
1541         base->height -= 1;
1542
1543         if (view != base) {
1544                 /* Horizontal split. */
1545                 view->width   = base->width;
1546                 view->height  = SCALE_SPLIT_VIEW(base->height);
1547                 base->height -= view->height;
1548
1549                 /* Make room for the title bar. */
1550                 view->height -= 1;
1551         }
1552
1553         /* Make room for the title bar. */
1554         base->height -= 1;
1555
1556         offset = 0;
1557
1558         foreach_displayed_view (view, i) {
1559                 if (!view->win) {
1560                         view->win = newwin(view->height, 0, offset, 0);
1561                         if (!view->win)
1562                                 die("Failed to create %s view", view->name);
1563
1564                         scrollok(view->win, TRUE);
1565
1566                         view->title = newwin(1, 0, offset + view->height, 0);
1567                         if (!view->title)
1568                                 die("Failed to create title window");
1569
1570                 } else {
1571                         wresize(view->win, view->height, view->width);
1572                         mvwin(view->win,   offset, 0);
1573                         mvwin(view->title, offset + view->height, 0);
1574                 }
1575
1576                 offset += view->height + 1;
1577         }
1578 }
1579
1580 static void
1581 redraw_display(void)
1582 {
1583         struct view *view;
1584         int i;
1585
1586         foreach_displayed_view (view, i) {
1587                 redraw_view(view);
1588                 update_view_title(view);
1589         }
1590 }
1591
1592 static void
1593 update_display_cursor(struct view *view)
1594 {
1595         /* Move the cursor to the right-most column of the cursor line.
1596          *
1597          * XXX: This could turn out to be a bit expensive, but it ensures that
1598          * the cursor does not jump around. */
1599         if (view->lines) {
1600                 wmove(view->win, view->lineno - view->offset, view->width - 1);
1601                 wrefresh(view->win);
1602         }
1603 }
1604
1605 /*
1606  * Navigation
1607  */
1608
1609 /* Scrolling backend */
1610 static void
1611 do_scroll_view(struct view *view, int lines)
1612 {
1613         bool redraw_current_line = FALSE;
1614
1615         /* The rendering expects the new offset. */
1616         view->offset += lines;
1617
1618         assert(0 <= view->offset && view->offset < view->lines);
1619         assert(lines);
1620
1621         /* Move current line into the view. */
1622         if (view->lineno < view->offset) {
1623                 view->lineno = view->offset;
1624                 redraw_current_line = TRUE;
1625         } else if (view->lineno >= view->offset + view->height) {
1626                 view->lineno = view->offset + view->height - 1;
1627                 redraw_current_line = TRUE;
1628         }
1629
1630         assert(view->offset <= view->lineno && view->lineno < view->lines);
1631
1632         /* Redraw the whole screen if scrolling is pointless. */
1633         if (view->height < ABS(lines)) {
1634                 redraw_view(view);
1635
1636         } else {
1637                 int line = lines > 0 ? view->height - lines : 0;
1638                 int end = line + ABS(lines);
1639
1640                 wscrl(view->win, lines);
1641
1642                 for (; line < end; line++) {
1643                         if (!draw_view_line(view, line))
1644                                 break;
1645                 }
1646
1647                 if (redraw_current_line)
1648                         draw_view_line(view, view->lineno - view->offset);
1649         }
1650
1651         redrawwin(view->win);
1652         wrefresh(view->win);
1653         report("");
1654 }
1655
1656 /* Scroll frontend */
1657 static void
1658 scroll_view(struct view *view, enum request request)
1659 {
1660         int lines = 1;
1661
1662         assert(view_is_displayed(view));
1663
1664         switch (request) {
1665         case REQ_SCROLL_PAGE_DOWN:
1666                 lines = view->height;
1667         case REQ_SCROLL_LINE_DOWN:
1668                 if (view->offset + lines > view->lines)
1669                         lines = view->lines - view->offset;
1670
1671                 if (lines == 0 || view->offset + view->height >= view->lines) {
1672                         report("Cannot scroll beyond the last line");
1673                         return;
1674                 }
1675                 break;
1676
1677         case REQ_SCROLL_PAGE_UP:
1678                 lines = view->height;
1679         case REQ_SCROLL_LINE_UP:
1680                 if (lines > view->offset)
1681                         lines = view->offset;
1682
1683                 if (lines == 0) {
1684                         report("Cannot scroll beyond the first line");
1685                         return;
1686                 }
1687
1688                 lines = -lines;
1689                 break;
1690
1691         default:
1692                 die("request %d not handled in switch", request);
1693         }
1694
1695         do_scroll_view(view, lines);
1696 }
1697
1698 /* Cursor moving */
1699 static void
1700 move_view(struct view *view, enum request request)
1701 {
1702         int scroll_steps = 0;
1703         int steps;
1704
1705         switch (request) {
1706         case REQ_MOVE_FIRST_LINE:
1707                 steps = -view->lineno;
1708                 break;
1709
1710         case REQ_MOVE_LAST_LINE:
1711                 steps = view->lines - view->lineno - 1;
1712                 break;
1713
1714         case REQ_MOVE_PAGE_UP:
1715                 steps = view->height > view->lineno
1716                       ? -view->lineno : -view->height;
1717                 break;
1718
1719         case REQ_MOVE_PAGE_DOWN:
1720                 steps = view->lineno + view->height >= view->lines
1721                       ? view->lines - view->lineno - 1 : view->height;
1722                 break;
1723
1724         case REQ_MOVE_UP:
1725                 steps = -1;
1726                 break;
1727
1728         case REQ_MOVE_DOWN:
1729                 steps = 1;
1730                 break;
1731
1732         default:
1733                 die("request %d not handled in switch", request);
1734         }
1735
1736         if (steps <= 0 && view->lineno == 0) {
1737                 report("Cannot move beyond the first line");
1738                 return;
1739
1740         } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
1741                 report("Cannot move beyond the last line");
1742                 return;
1743         }
1744
1745         /* Move the current line */
1746         view->lineno += steps;
1747         assert(0 <= view->lineno && view->lineno < view->lines);
1748
1749         /* Check whether the view needs to be scrolled */
1750         if (view->lineno < view->offset ||
1751             view->lineno >= view->offset + view->height) {
1752                 scroll_steps = steps;
1753                 if (steps < 0 && -steps > view->offset) {
1754                         scroll_steps = -view->offset;
1755
1756                 } else if (steps > 0) {
1757                         if (view->lineno == view->lines - 1 &&
1758                             view->lines > view->height) {
1759                                 scroll_steps = view->lines - view->offset - 1;
1760                                 if (scroll_steps >= view->height)
1761                                         scroll_steps -= view->height - 1;
1762                         }
1763                 }
1764         }
1765
1766         if (!view_is_displayed(view)) {
1767                 view->offset += scroll_steps;
1768                 assert(0 <= view->offset && view->offset < view->lines);
1769                 view->ops->select(view, &view->line[view->lineno]);
1770                 return;
1771         }
1772
1773         /* Repaint the old "current" line if we be scrolling */
1774         if (ABS(steps) < view->height)
1775                 draw_view_line(view, view->lineno - steps - view->offset);
1776
1777         if (scroll_steps) {
1778                 do_scroll_view(view, scroll_steps);
1779                 return;
1780         }
1781
1782         /* Draw the current line */
1783         draw_view_line(view, view->lineno - view->offset);
1784
1785         redrawwin(view->win);
1786         wrefresh(view->win);
1787         report("");
1788 }
1789
1790
1791 /*
1792  * Searching
1793  */
1794
1795 static void search_view(struct view *view, enum request request);
1796
1797 static bool
1798 find_next_line(struct view *view, unsigned long lineno, struct line *line)
1799 {
1800         assert(view_is_displayed(view));
1801
1802         if (!view->ops->grep(view, line))
1803                 return FALSE;
1804
1805         if (lineno - view->offset >= view->height) {
1806                 view->offset = lineno;
1807                 view->lineno = lineno;
1808                 redraw_view(view);
1809
1810         } else {
1811                 unsigned long old_lineno = view->lineno - view->offset;
1812
1813                 view->lineno = lineno;
1814                 draw_view_line(view, old_lineno);
1815
1816                 draw_view_line(view, view->lineno - view->offset);
1817                 redrawwin(view->win);
1818                 wrefresh(view->win);
1819         }
1820
1821         report("Line %ld matches '%s'", lineno + 1, view->grep);
1822         return TRUE;
1823 }
1824
1825 static void
1826 find_next(struct view *view, enum request request)
1827 {
1828         unsigned long lineno = view->lineno;
1829         int direction;
1830
1831         if (!*view->grep) {
1832                 if (!*opt_search)
1833                         report("No previous search");
1834                 else
1835                         search_view(view, request);
1836                 return;
1837         }
1838
1839         switch (request) {
1840         case REQ_SEARCH:
1841         case REQ_FIND_NEXT:
1842                 direction = 1;
1843                 break;
1844
1845         case REQ_SEARCH_BACK:
1846         case REQ_FIND_PREV:
1847                 direction = -1;
1848                 break;
1849
1850         default:
1851                 return;
1852         }
1853
1854         if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
1855                 lineno += direction;
1856
1857         /* Note, lineno is unsigned long so will wrap around in which case it
1858          * will become bigger than view->lines. */
1859         for (; lineno < view->lines; lineno += direction) {
1860                 struct line *line = &view->line[lineno];
1861
1862                 if (find_next_line(view, lineno, line))
1863                         return;
1864         }
1865
1866         report("No match found for '%s'", view->grep);
1867 }
1868
1869 static void
1870 search_view(struct view *view, enum request request)
1871 {
1872         int regex_err;
1873
1874         if (view->regex) {
1875                 regfree(view->regex);
1876                 *view->grep = 0;
1877         } else {
1878                 view->regex = calloc(1, sizeof(*view->regex));
1879                 if (!view->regex)
1880                         return;
1881         }
1882
1883         regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
1884         if (regex_err != 0) {
1885                 char buf[SIZEOF_STR] = "unknown error";
1886
1887                 regerror(regex_err, view->regex, buf, sizeof(buf));
1888                 report("Search failed: %s", buf);
1889                 return;
1890         }
1891
1892         string_copy(view->grep, opt_search);
1893
1894         find_next(view, request);
1895 }
1896
1897 /*
1898  * Incremental updating
1899  */
1900
1901 static void
1902 end_update(struct view *view)
1903 {
1904         if (!view->pipe)
1905                 return;
1906         set_nonblocking_input(FALSE);
1907         if (view->pipe == stdin)
1908                 fclose(view->pipe);
1909         else
1910                 pclose(view->pipe);
1911         view->pipe = NULL;
1912 }
1913
1914 static bool
1915 begin_update(struct view *view)
1916 {
1917         if (view->pipe)
1918                 end_update(view);
1919
1920         if (opt_cmd[0]) {
1921                 string_copy(view->cmd, opt_cmd);
1922                 opt_cmd[0] = 0;
1923                 /* When running random commands, initially show the
1924                  * command in the title. However, it maybe later be
1925                  * overwritten if a commit line is selected. */
1926                 if (view == VIEW(REQ_VIEW_PAGER))
1927                         string_copy(view->ref, view->cmd);
1928                 else
1929                         view->ref[0] = 0;
1930
1931         } else if (view == VIEW(REQ_VIEW_TREE)) {
1932                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1933                 char path[SIZEOF_STR];
1934
1935                 if (strcmp(view->vid, view->id))
1936                         opt_path[0] = path[0] = 0;
1937                 else if (sq_quote(path, 0, opt_path) >= sizeof(path))
1938                         return FALSE;
1939
1940                 if (!string_format(view->cmd, format, view->id, path))
1941                         return FALSE;
1942
1943         } else {
1944                 const char *format = view->cmd_env ? view->cmd_env : view->cmd_fmt;
1945                 const char *id = view->id;
1946
1947                 if (!string_format(view->cmd, format, id, id, id, id, id))
1948                         return FALSE;
1949
1950                 /* Put the current ref_* value to the view title ref
1951                  * member. This is needed by the blob view. Most other
1952                  * views sets it automatically after loading because the
1953                  * first line is a commit line. */
1954                 string_copy_rev(view->ref, view->id);
1955         }
1956
1957         /* Special case for the pager view. */
1958         if (opt_pipe) {
1959                 view->pipe = opt_pipe;
1960                 opt_pipe = NULL;
1961         } else {
1962                 view->pipe = popen(view->cmd, "r");
1963         }
1964
1965         if (!view->pipe)
1966                 return FALSE;
1967
1968         set_nonblocking_input(TRUE);
1969
1970         view->offset = 0;
1971         view->lines  = 0;
1972         view->lineno = 0;
1973         string_copy_rev(view->vid, view->id);
1974
1975         if (view->line) {
1976                 int i;
1977
1978                 for (i = 0; i < view->lines; i++)
1979                         if (view->line[i].data)
1980                                 free(view->line[i].data);
1981
1982                 free(view->line);
1983                 view->line = NULL;
1984         }
1985
1986         view->start_time = time(NULL);
1987
1988         return TRUE;
1989 }
1990
1991 static struct line *
1992 realloc_lines(struct view *view, size_t line_size)
1993 {
1994         struct line *tmp = realloc(view->line, sizeof(*view->line) * line_size);
1995
1996         if (!tmp)
1997                 return NULL;
1998
1999         view->line = tmp;
2000         view->line_size = line_size;
2001         return view->line;
2002 }
2003
2004 static bool
2005 update_view(struct view *view)
2006 {
2007         char in_buffer[BUFSIZ];
2008         char out_buffer[BUFSIZ * 2];
2009         char *line;
2010         /* The number of lines to read. If too low it will cause too much
2011          * redrawing (and possible flickering), if too high responsiveness
2012          * will suffer. */
2013         unsigned long lines = view->height;
2014         int redraw_from = -1;
2015
2016         if (!view->pipe)
2017                 return TRUE;
2018
2019         /* Only redraw if lines are visible. */
2020         if (view->offset + view->height >= view->lines)
2021                 redraw_from = view->lines - view->offset;
2022
2023         /* FIXME: This is probably not perfect for backgrounded views. */
2024         if (!realloc_lines(view, view->lines + lines))
2025                 goto alloc_error;
2026
2027         while ((line = fgets(in_buffer, sizeof(in_buffer), view->pipe))) {
2028                 size_t linelen = strlen(line);
2029
2030                 if (linelen)
2031                         line[linelen - 1] = 0;
2032
2033                 if (opt_iconv != ICONV_NONE) {
2034                         ICONV_CONST char *inbuf = line;
2035                         size_t inlen = linelen;
2036
2037                         char *outbuf = out_buffer;
2038                         size_t outlen = sizeof(out_buffer);
2039
2040                         size_t ret;
2041
2042                         ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2043                         if (ret != (size_t) -1) {
2044                                 line = out_buffer;
2045                                 linelen = strlen(out_buffer);
2046                         }
2047                 }
2048
2049                 if (!view->ops->read(view, line))
2050                         goto alloc_error;
2051
2052                 if (lines-- == 1)
2053                         break;
2054         }
2055
2056         {
2057                 int digits;
2058
2059                 lines = view->lines;
2060                 for (digits = 0; lines; digits++)
2061                         lines /= 10;
2062
2063                 /* Keep the displayed view in sync with line number scaling. */
2064                 if (digits != view->digits) {
2065                         view->digits = digits;
2066                         redraw_from = 0;
2067                 }
2068         }
2069
2070         if (!view_is_displayed(view))
2071                 goto check_pipe;
2072
2073         if (view == VIEW(REQ_VIEW_TREE)) {
2074                 /* Clear the view and redraw everything since the tree sorting
2075                  * might have rearranged things. */
2076                 redraw_view(view);
2077
2078         } else if (redraw_from >= 0) {
2079                 /* If this is an incremental update, redraw the previous line
2080                  * since for commits some members could have changed when
2081                  * loading the main view. */
2082                 if (redraw_from > 0)
2083                         redraw_from--;
2084
2085                 /* Since revision graph visualization requires knowledge
2086                  * about the parent commit, it causes a further one-off
2087                  * needed to be redrawn for incremental updates. */
2088                 if (redraw_from > 0 && opt_rev_graph)
2089                         redraw_from--;
2090
2091                 /* Incrementally draw avoids flickering. */
2092                 redraw_view_from(view, redraw_from);
2093         }
2094
2095         /* Update the title _after_ the redraw so that if the redraw picks up a
2096          * commit reference in view->ref it'll be available here. */
2097         update_view_title(view);
2098
2099 check_pipe:
2100         if (ferror(view->pipe)) {
2101                 report("Failed to read: %s", strerror(errno));
2102                 goto end;
2103
2104         } else if (feof(view->pipe)) {
2105                 report("");
2106                 goto end;
2107         }
2108
2109         return TRUE;
2110
2111 alloc_error:
2112         report("Allocation failure");
2113
2114 end:
2115         view->ops->read(view, NULL);
2116         end_update(view);
2117         return FALSE;
2118 }
2119
2120 static struct line *
2121 add_line_data(struct view *view, void *data, enum line_type type)
2122 {
2123         struct line *line = &view->line[view->lines++];
2124
2125         memset(line, 0, sizeof(*line));
2126         line->type = type;
2127         line->data = data;
2128
2129         return line;
2130 }
2131
2132 static struct line *
2133 add_line_text(struct view *view, char *data, enum line_type type)
2134 {
2135         if (data)
2136                 data = strdup(data);
2137
2138         return data ? add_line_data(view, data, type) : NULL;
2139 }
2140
2141
2142 /*
2143  * View opening
2144  */
2145
2146 enum open_flags {
2147         OPEN_DEFAULT = 0,       /* Use default view switching. */
2148         OPEN_SPLIT = 1,         /* Split current view. */
2149         OPEN_BACKGROUNDED = 2,  /* Backgrounded. */
2150         OPEN_RELOAD = 4,        /* Reload view even if it is the current. */
2151 };
2152
2153 static void
2154 open_view(struct view *prev, enum request request, enum open_flags flags)
2155 {
2156         bool backgrounded = !!(flags & OPEN_BACKGROUNDED);
2157         bool split = !!(flags & OPEN_SPLIT);
2158         bool reload = !!(flags & OPEN_RELOAD);
2159         struct view *view = VIEW(request);
2160         int nviews = displayed_views();
2161         struct view *base_view = display[0];
2162
2163         if (view == prev && nviews == 1 && !reload) {
2164                 report("Already in %s view", view->name);
2165                 return;
2166         }
2167
2168         if (view->ops->open) {
2169                 if (!view->ops->open(view)) {
2170                         report("Failed to load %s view", view->name);
2171                         return;
2172                 }
2173
2174         } else if ((reload || strcmp(view->vid, view->id)) &&
2175                    !begin_update(view)) {
2176                 report("Failed to load %s view", view->name);
2177                 return;
2178         }
2179
2180         if (split) {
2181                 display[1] = view;
2182                 if (!backgrounded)
2183                         current_view = 1;
2184         } else {
2185                 /* Maximize the current view. */
2186                 memset(display, 0, sizeof(display));
2187                 current_view = 0;
2188                 display[current_view] = view;
2189         }
2190
2191         /* Resize the view when switching between split- and full-screen,
2192          * or when switching between two different full-screen views. */
2193         if (nviews != displayed_views() ||
2194             (nviews == 1 && base_view != display[0]))
2195                 resize_display();
2196
2197         if (split && prev->lineno - prev->offset >= prev->height) {
2198                 /* Take the title line into account. */
2199                 int lines = prev->lineno - prev->offset - prev->height + 1;
2200
2201                 /* Scroll the view that was split if the current line is
2202                  * outside the new limited view. */
2203                 do_scroll_view(prev, lines);
2204         }
2205
2206         if (prev && view != prev) {
2207                 if (split && !backgrounded) {
2208                         /* "Blur" the previous view. */
2209                         update_view_title(prev);
2210                 }
2211
2212                 view->parent = prev;
2213         }
2214
2215         if (view->pipe && view->lines == 0) {
2216                 /* Clear the old view and let the incremental updating refill
2217                  * the screen. */
2218                 wclear(view->win);
2219                 report("");
2220         } else {
2221                 redraw_view(view);
2222                 report("");
2223         }
2224
2225         /* If the view is backgrounded the above calls to report()
2226          * won't redraw the view title. */
2227         if (backgrounded)
2228                 update_view_title(view);
2229 }
2230
2231 static void
2232 open_external_viewer(const char *cmd)
2233 {
2234         def_prog_mode();           /* save current tty modes */
2235         endwin();                  /* restore original tty modes */
2236         system(cmd);
2237         fprintf(stderr, "Press Enter to continue");
2238         getc(stdin);
2239         reset_prog_mode();
2240         redraw_display();
2241 }
2242
2243 static void
2244 open_mergetool(const char *file)
2245 {
2246         char cmd[SIZEOF_STR];
2247         char file_sq[SIZEOF_STR];
2248
2249         if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2250             string_format(cmd, "git mergetool %s", file_sq)) {
2251                 open_external_viewer(cmd);
2252         }
2253 }
2254
2255 static void
2256 open_editor(bool from_root, const char *file)
2257 {
2258         char cmd[SIZEOF_STR];
2259         char file_sq[SIZEOF_STR];
2260         char *editor;
2261         char *prefix = from_root ? opt_cdup : "";
2262
2263         editor = getenv("GIT_EDITOR");
2264         if (!editor && *opt_editor)
2265                 editor = opt_editor;
2266         if (!editor)
2267                 editor = getenv("VISUAL");
2268         if (!editor)
2269                 editor = getenv("EDITOR");
2270         if (!editor)
2271                 editor = "vi";
2272
2273         if (sq_quote(file_sq, 0, file) < sizeof(file_sq) &&
2274             string_format(cmd, "%s %s%s", editor, prefix, file_sq)) {
2275                 open_external_viewer(cmd);
2276         }
2277 }
2278
2279 static void
2280 open_run_request(enum request request)
2281 {
2282         struct run_request *req = get_run_request(request);
2283         char buf[SIZEOF_STR * 2];
2284         size_t bufpos;
2285         char *cmd;
2286
2287         if (!req) {
2288                 report("Unknown run request");
2289                 return;
2290         }
2291
2292         bufpos = 0;
2293         cmd = req->cmd;
2294
2295         while (cmd) {
2296                 char *next = strstr(cmd, "%(");
2297                 int len = next - cmd;
2298                 char *value;
2299
2300                 if (!next) {
2301                         len = strlen(cmd);
2302                         value = "";
2303
2304                 } else if (!strncmp(next, "%(head)", 7)) {
2305                         value = ref_head;
2306
2307                 } else if (!strncmp(next, "%(commit)", 9)) {
2308                         value = ref_commit;
2309
2310                 } else if (!strncmp(next, "%(blob)", 7)) {
2311                         value = ref_blob;
2312
2313                 } else {
2314                         report("Unknown replacement in run request: `%s`", req->cmd);
2315                         return;
2316                 }
2317
2318                 if (!string_format_from(buf, &bufpos, "%.*s%s", len, cmd, value))
2319                         return;
2320
2321                 if (next)
2322                         next = strchr(next, ')') + 1;
2323                 cmd = next;
2324         }
2325
2326         open_external_viewer(buf);
2327 }
2328
2329 /*
2330  * User request switch noodle
2331  */
2332
2333 static int
2334 view_driver(struct view *view, enum request request)
2335 {
2336         int i;
2337
2338         if (request == REQ_NONE) {
2339                 doupdate();
2340                 return TRUE;
2341         }
2342
2343         if (request > REQ_NONE) {
2344                 open_run_request(request);
2345                 return TRUE;
2346         }
2347
2348         if (view && view->lines) {
2349                 request = view->ops->request(view, request, &view->line[view->lineno]);
2350                 if (request == REQ_NONE)
2351                         return TRUE;
2352         }
2353
2354         switch (request) {
2355         case REQ_MOVE_UP:
2356         case REQ_MOVE_DOWN:
2357         case REQ_MOVE_PAGE_UP:
2358         case REQ_MOVE_PAGE_DOWN:
2359         case REQ_MOVE_FIRST_LINE:
2360         case REQ_MOVE_LAST_LINE:
2361                 move_view(view, request);
2362                 break;
2363
2364         case REQ_SCROLL_LINE_DOWN:
2365         case REQ_SCROLL_LINE_UP:
2366         case REQ_SCROLL_PAGE_DOWN:
2367         case REQ_SCROLL_PAGE_UP:
2368                 scroll_view(view, request);
2369                 break;
2370
2371         case REQ_VIEW_BLOB:
2372                 if (!ref_blob[0]) {
2373                         report("No file chosen, press %s to open tree view",
2374                                get_key(REQ_VIEW_TREE));
2375                         break;
2376                 }
2377                 open_view(view, request, OPEN_DEFAULT);
2378                 break;
2379
2380         case REQ_VIEW_PAGER:
2381                 if (!opt_pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
2382                         report("No pager content, press %s to run command from prompt",
2383                                get_key(REQ_PROMPT));
2384                         break;
2385                 }
2386                 open_view(view, request, OPEN_DEFAULT);
2387                 break;
2388
2389         case REQ_VIEW_STAGE:
2390                 if (!VIEW(REQ_VIEW_STAGE)->lines) {
2391                         report("No stage content, press %s to open the status view and choose file",
2392                                get_key(REQ_VIEW_STATUS));
2393                         break;
2394                 }
2395                 open_view(view, request, OPEN_DEFAULT);
2396                 break;
2397
2398         case REQ_VIEW_STATUS:
2399                 if (opt_is_inside_work_tree == FALSE) {
2400                         report("The status view requires a working tree");
2401                         break;
2402                 }
2403                 open_view(view, request, OPEN_DEFAULT);
2404                 break;
2405
2406         case REQ_VIEW_MAIN:
2407         case REQ_VIEW_DIFF:
2408         case REQ_VIEW_LOG:
2409         case REQ_VIEW_TREE:
2410         case REQ_VIEW_HELP:
2411                 open_view(view, request, OPEN_DEFAULT);
2412                 break;
2413
2414         case REQ_NEXT:
2415         case REQ_PREVIOUS:
2416                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
2417
2418                 if ((view == VIEW(REQ_VIEW_DIFF) &&
2419                      view->parent == VIEW(REQ_VIEW_MAIN)) ||
2420                    (view == VIEW(REQ_VIEW_STAGE) &&
2421                      view->parent == VIEW(REQ_VIEW_STATUS)) ||
2422                    (view == VIEW(REQ_VIEW_BLOB) &&
2423                      view->parent == VIEW(REQ_VIEW_TREE))) {
2424                         int line;
2425
2426                         view = view->parent;
2427                         line = view->lineno;
2428                         move_view(view, request);
2429                         if (view_is_displayed(view))
2430                                 update_view_title(view);
2431                         if (line != view->lineno)
2432                                 view->ops->request(view, REQ_ENTER,
2433                                                    &view->line[view->lineno]);
2434
2435                 } else {
2436                         move_view(view, request);
2437                 }
2438                 break;
2439
2440         case REQ_VIEW_NEXT:
2441         {
2442                 int nviews = displayed_views();
2443                 int next_view = (current_view + 1) % nviews;
2444
2445                 if (next_view == current_view) {
2446                         report("Only one view is displayed");
2447                         break;
2448                 }
2449
2450                 current_view = next_view;
2451                 /* Blur out the title of the previous view. */
2452                 update_view_title(view);
2453                 report("");
2454                 break;
2455         }
2456         case REQ_REFRESH:
2457                 report("Refreshing is not yet supported for the %s view", view->name);
2458                 break;
2459
2460         case REQ_TOGGLE_LINENO:
2461                 opt_line_number = !opt_line_number;
2462                 redraw_display();
2463                 break;
2464
2465         case REQ_TOGGLE_REV_GRAPH:
2466                 opt_rev_graph = !opt_rev_graph;
2467                 redraw_display();
2468                 break;
2469
2470         case REQ_PROMPT:
2471                 /* Always reload^Wrerun commands from the prompt. */
2472                 open_view(view, opt_request, OPEN_RELOAD);
2473                 break;
2474
2475         case REQ_SEARCH:
2476         case REQ_SEARCH_BACK:
2477                 search_view(view, request);
2478                 break;
2479
2480         case REQ_FIND_NEXT:
2481         case REQ_FIND_PREV:
2482                 find_next(view, request);
2483                 break;
2484
2485         case REQ_STOP_LOADING:
2486                 for (i = 0; i < ARRAY_SIZE(views); i++) {
2487                         view = &views[i];
2488                         if (view->pipe)
2489                                 report("Stopped loading the %s view", view->name),
2490                         end_update(view);
2491                 }
2492                 break;
2493
2494         case REQ_SHOW_VERSION:
2495                 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
2496                 return TRUE;
2497
2498         case REQ_SCREEN_RESIZE:
2499                 resize_display();
2500                 /* Fall-through */
2501         case REQ_SCREEN_REDRAW:
2502                 redraw_display();
2503                 break;
2504
2505         case REQ_EDIT:
2506                 report("Nothing to edit");
2507                 break;
2508
2509
2510         case REQ_ENTER:
2511                 report("Nothing to enter");
2512                 break;
2513
2514
2515         case REQ_VIEW_CLOSE:
2516                 /* XXX: Mark closed views by letting view->parent point to the
2517                  * view itself. Parents to closed view should never be
2518                  * followed. */
2519                 if (view->parent &&
2520                     view->parent->parent != view->parent) {
2521                         memset(display, 0, sizeof(display));
2522                         current_view = 0;
2523                         display[current_view] = view->parent;
2524                         view->parent = view;
2525                         resize_display();
2526                         redraw_display();
2527                         break;
2528                 }
2529                 /* Fall-through */
2530         case REQ_QUIT:
2531                 return FALSE;
2532
2533         default:
2534                 /* An unknown key will show most commonly used commands. */
2535                 report("Unknown key, press 'h' for help");
2536                 return TRUE;
2537         }
2538
2539         return TRUE;
2540 }
2541
2542
2543 /*
2544  * Pager backend
2545  */
2546
2547 static bool
2548 pager_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
2549 {
2550         char *text = line->data;
2551         enum line_type type = line->type;
2552         int textlen = strlen(text);
2553         int attr;
2554
2555         wmove(view->win, lineno, 0);
2556
2557         if (selected) {
2558                 type = LINE_CURSOR;
2559                 wchgat(view->win, -1, 0, type, NULL);
2560         }
2561
2562         attr = get_line_attr(type);
2563         wattrset(view->win, attr);
2564
2565         if (opt_line_number || opt_tab_size < TABSIZE) {
2566                 static char spaces[] = "                    ";
2567                 int col_offset = 0, col = 0;
2568
2569                 if (opt_line_number) {
2570                         unsigned long real_lineno = view->offset + lineno + 1;
2571
2572                         if (real_lineno == 1 ||
2573                             (real_lineno % opt_num_interval) == 0) {
2574                                 wprintw(view->win, "%.*d", view->digits, real_lineno);
2575
2576                         } else {
2577                                 waddnstr(view->win, spaces,
2578                                          MIN(view->digits, STRING_SIZE(spaces)));
2579                         }
2580                         waddstr(view->win, ": ");
2581                         col_offset = view->digits + 2;
2582                 }
2583
2584                 while (text && col_offset + col < view->width) {
2585                         int cols_max = view->width - col_offset - col;
2586                         char *pos = text;
2587                         int cols;
2588
2589                         if (*text == '\t') {
2590                                 text++;
2591                                 assert(sizeof(spaces) > TABSIZE);
2592                                 pos = spaces;
2593                                 cols = opt_tab_size - (col % opt_tab_size);
2594
2595                         } else {
2596                                 text = strchr(text, '\t');
2597                                 cols = line ? text - pos : strlen(pos);
2598                         }
2599
2600                         waddnstr(view->win, pos, MIN(cols, cols_max));
2601                         col += cols;
2602                 }
2603
2604         } else {
2605                 int col = 0, pos = 0;
2606
2607                 for (; pos < textlen && col < view->width; pos++, col++)
2608                         if (text[pos] == '\t')
2609                                 col += TABSIZE - (col % TABSIZE) - 1;
2610
2611                 waddnstr(view->win, text, pos);
2612         }
2613
2614         return TRUE;
2615 }
2616
2617 static bool
2618 add_describe_ref(char *buf, size_t *bufpos, char *commit_id, const char *sep)
2619 {
2620         char refbuf[SIZEOF_STR];
2621         char *ref = NULL;
2622         FILE *pipe;
2623
2624         if (!string_format(refbuf, "git describe %s 2>/dev/null", commit_id))
2625                 return TRUE;
2626
2627         pipe = popen(refbuf, "r");
2628         if (!pipe)
2629                 return TRUE;
2630
2631         if ((ref = fgets(refbuf, sizeof(refbuf), pipe)))
2632                 ref = chomp_string(ref);
2633         pclose(pipe);
2634
2635         if (!ref || !*ref)
2636                 return TRUE;
2637
2638         /* This is the only fatal call, since it can "corrupt" the buffer. */
2639         if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
2640                 return FALSE;
2641
2642         return TRUE;
2643 }
2644
2645 static void
2646 add_pager_refs(struct view *view, struct line *line)
2647 {
2648         char buf[SIZEOF_STR];
2649         char *commit_id = line->data + STRING_SIZE("commit ");
2650         struct ref **refs;
2651         size_t bufpos = 0, refpos = 0;
2652         const char *sep = "Refs: ";
2653         bool is_tag = FALSE;
2654
2655         assert(line->type == LINE_COMMIT);
2656
2657         refs = get_refs(commit_id);
2658         if (!refs) {
2659                 if (view == VIEW(REQ_VIEW_DIFF))
2660                         goto try_add_describe_ref;
2661                 return;
2662         }
2663
2664         do {
2665                 struct ref *ref = refs[refpos];
2666                 char *fmt = ref->tag    ? "%s[%s]" :
2667                             ref->remote ? "%s<%s>" : "%s%s";
2668
2669                 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
2670                         return;
2671                 sep = ", ";
2672                 if (ref->tag)
2673                         is_tag = TRUE;
2674         } while (refs[refpos++]->next);
2675
2676         if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
2677 try_add_describe_ref:
2678                 /* Add <tag>-g<commit_id> "fake" reference. */
2679                 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
2680                         return;
2681         }
2682
2683         if (bufpos == 0)
2684                 return;
2685
2686         if (!realloc_lines(view, view->line_size + 1))
2687                 return;
2688
2689         add_line_text(view, buf, LINE_PP_REFS);
2690 }
2691
2692 static bool
2693 pager_read(struct view *view, char *data)
2694 {
2695         struct line *line;
2696
2697         if (!data)
2698                 return TRUE;
2699
2700         line = add_line_text(view, data, get_line_type(data));
2701         if (!line)
2702                 return FALSE;
2703
2704         if (line->type == LINE_COMMIT &&
2705             (view == VIEW(REQ_VIEW_DIFF) ||
2706              view == VIEW(REQ_VIEW_LOG)))
2707                 add_pager_refs(view, line);
2708
2709         return TRUE;
2710 }
2711
2712 static enum request
2713 pager_request(struct view *view, enum request request, struct line *line)
2714 {
2715         int split = 0;
2716
2717         if (request != REQ_ENTER)
2718                 return request;
2719
2720         if (line->type == LINE_COMMIT &&
2721            (view == VIEW(REQ_VIEW_LOG) ||
2722             view == VIEW(REQ_VIEW_PAGER))) {
2723                 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
2724                 split = 1;
2725         }
2726
2727         /* Always scroll the view even if it was split. That way
2728          * you can use Enter to scroll through the log view and
2729          * split open each commit diff. */
2730         scroll_view(view, REQ_SCROLL_LINE_DOWN);
2731
2732         /* FIXME: A minor workaround. Scrolling the view will call report("")
2733          * but if we are scrolling a non-current view this won't properly
2734          * update the view title. */
2735         if (split)
2736                 update_view_title(view);
2737
2738         return REQ_NONE;
2739 }
2740
2741 static bool
2742 pager_grep(struct view *view, struct line *line)
2743 {
2744         regmatch_t pmatch;
2745         char *text = line->data;
2746
2747         if (!*text)
2748                 return FALSE;
2749
2750         if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
2751                 return FALSE;
2752
2753         return TRUE;
2754 }
2755
2756 static void
2757 pager_select(struct view *view, struct line *line)
2758 {
2759         if (line->type == LINE_COMMIT) {
2760                 char *text = line->data + STRING_SIZE("commit ");
2761
2762                 if (view != VIEW(REQ_VIEW_PAGER))
2763                         string_copy_rev(view->ref, text);
2764                 string_copy_rev(ref_commit, text);
2765         }
2766 }
2767
2768 static struct view_ops pager_ops = {
2769         "line",
2770         NULL,
2771         pager_read,
2772         pager_draw,
2773         pager_request,
2774         pager_grep,
2775         pager_select,
2776 };
2777
2778
2779 /*
2780  * Help backend
2781  */
2782
2783 static bool
2784 help_open(struct view *view)
2785 {
2786         char buf[BUFSIZ];
2787         int lines = ARRAY_SIZE(req_info) + 2;
2788         int i;
2789
2790         if (view->lines > 0)
2791                 return TRUE;
2792
2793         for (i = 0; i < ARRAY_SIZE(req_info); i++)
2794                 if (!req_info[i].request)
2795                         lines++;
2796
2797         lines += run_requests + 1;
2798
2799         view->line = calloc(lines, sizeof(*view->line));
2800         if (!view->line)
2801                 return FALSE;
2802
2803         add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
2804
2805         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
2806                 char *key;
2807
2808                 if (req_info[i].request == REQ_NONE)
2809                         continue;
2810
2811                 if (!req_info[i].request) {
2812                         add_line_text(view, "", LINE_DEFAULT);
2813                         add_line_text(view, req_info[i].help, LINE_DEFAULT);
2814                         continue;
2815                 }
2816
2817                 key = get_key(req_info[i].request);
2818                 if (!*key)
2819                         key = "(no key defined)";
2820
2821                 if (!string_format(buf, "    %-25s %s", key, req_info[i].help))
2822                         continue;
2823
2824                 add_line_text(view, buf, LINE_DEFAULT);
2825         }
2826
2827         if (run_requests) {
2828                 add_line_text(view, "", LINE_DEFAULT);
2829                 add_line_text(view, "External commands:", LINE_DEFAULT);
2830         }
2831
2832         for (i = 0; i < run_requests; i++) {
2833                 struct run_request *req = get_run_request(REQ_NONE + i + 1);
2834                 char *key;
2835
2836                 if (!req)
2837                         continue;
2838
2839                 key = get_key_name(req->key);
2840                 if (!*key)
2841                         key = "(no key defined)";
2842
2843                 if (!string_format(buf, "    %-10s %-14s `%s`",
2844                                    keymap_table[req->keymap].name,
2845                                    key, req->cmd))
2846                         continue;
2847
2848                 add_line_text(view, buf, LINE_DEFAULT);
2849         }
2850
2851         return TRUE;
2852 }
2853
2854 static struct view_ops help_ops = {
2855         "line",
2856         help_open,
2857         NULL,
2858         pager_draw,
2859         pager_request,
2860         pager_grep,
2861         pager_select,
2862 };
2863
2864
2865 /*
2866  * Tree backend
2867  */
2868
2869 struct tree_stack_entry {
2870         struct tree_stack_entry *prev;  /* Entry below this in the stack */
2871         unsigned long lineno;           /* Line number to restore */
2872         char *name;                     /* Position of name in opt_path */
2873 };
2874
2875 /* The top of the path stack. */
2876 static struct tree_stack_entry *tree_stack = NULL;
2877 unsigned long tree_lineno = 0;
2878
2879 static void
2880 pop_tree_stack_entry(void)
2881 {
2882         struct tree_stack_entry *entry = tree_stack;
2883
2884         tree_lineno = entry->lineno;
2885         entry->name[0] = 0;
2886         tree_stack = entry->prev;
2887         free(entry);
2888 }
2889
2890 static void
2891 push_tree_stack_entry(char *name, unsigned long lineno)
2892 {
2893         struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
2894         size_t pathlen = strlen(opt_path);
2895
2896         if (!entry)
2897                 return;
2898
2899         entry->prev = tree_stack;
2900         entry->name = opt_path + pathlen;
2901         tree_stack = entry;
2902
2903         if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
2904                 pop_tree_stack_entry();
2905                 return;
2906         }
2907
2908         /* Move the current line to the first tree entry. */
2909         tree_lineno = 1;
2910         entry->lineno = lineno;
2911 }
2912
2913 /* Parse output from git-ls-tree(1):
2914  *
2915  * 100644 blob fb0e31ea6cc679b7379631188190e975f5789c26 Makefile
2916  * 100644 blob 5304ca4260aaddaee6498f9630e7d471b8591ea6 README
2917  * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
2918  * 100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38 web.conf
2919  */
2920
2921 #define SIZEOF_TREE_ATTR \
2922         STRING_SIZE("100644 blob ed09fe897f3c7c9af90bcf80cae92558ea88ae38\t")
2923
2924 #define TREE_UP_FORMAT "040000 tree %s\t.."
2925
2926 static int
2927 tree_compare_entry(enum line_type type1, char *name1,
2928                    enum line_type type2, char *name2)
2929 {
2930         if (type1 != type2) {
2931                 if (type1 == LINE_TREE_DIR)
2932                         return -1;
2933                 return 1;
2934         }
2935
2936         return strcmp(name1, name2);
2937 }
2938
2939 static bool
2940 tree_read(struct view *view, char *text)
2941 {
2942         size_t textlen = text ? strlen(text) : 0;
2943         char buf[SIZEOF_STR];
2944         unsigned long pos;
2945         enum line_type type;
2946         bool first_read = view->lines == 0;
2947
2948         if (textlen <= SIZEOF_TREE_ATTR)
2949                 return FALSE;
2950
2951         type = text[STRING_SIZE("100644 ")] == 't'
2952              ? LINE_TREE_DIR : LINE_TREE_FILE;
2953
2954         if (first_read) {
2955                 /* Add path info line */
2956                 if (!string_format(buf, "Directory path /%s", opt_path) ||
2957                     !realloc_lines(view, view->line_size + 1) ||
2958                     !add_line_text(view, buf, LINE_DEFAULT))
2959                         return FALSE;
2960
2961                 /* Insert "link" to parent directory. */
2962                 if (*opt_path) {
2963                         if (!string_format(buf, TREE_UP_FORMAT, view->ref) ||
2964                             !realloc_lines(view, view->line_size + 1) ||
2965                             !add_line_text(view, buf, LINE_TREE_DIR))
2966                                 return FALSE;
2967                 }
2968         }
2969
2970         /* Strip the path part ... */
2971         if (*opt_path) {
2972                 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
2973                 size_t striplen = strlen(opt_path);
2974                 char *path = text + SIZEOF_TREE_ATTR;
2975
2976                 if (pathlen > striplen)
2977                         memmove(path, path + striplen,
2978                                 pathlen - striplen + 1);
2979         }
2980
2981         /* Skip "Directory ..." and ".." line. */
2982         for (pos = 1 + !!*opt_path; pos < view->lines; pos++) {
2983                 struct line *line = &view->line[pos];
2984                 char *path1 = ((char *) line->data) + SIZEOF_TREE_ATTR;
2985                 char *path2 = text + SIZEOF_TREE_ATTR;
2986                 int cmp = tree_compare_entry(line->type, path1, type, path2);
2987
2988                 if (cmp <= 0)
2989                         continue;
2990
2991                 text = strdup(text);
2992                 if (!text)
2993                         return FALSE;
2994
2995                 if (view->lines > pos)
2996                         memmove(&view->line[pos + 1], &view->line[pos],
2997                                 (view->lines - pos) * sizeof(*line));
2998
2999                 line = &view->line[pos];
3000                 line->data = text;
3001                 line->type = type;
3002                 view->lines++;
3003                 return TRUE;
3004         }
3005
3006         if (!add_line_text(view, text, type))
3007                 return FALSE;
3008
3009         if (tree_lineno > view->lineno) {
3010                 view->lineno = tree_lineno;
3011                 tree_lineno = 0;
3012         }
3013
3014         return TRUE;
3015 }
3016
3017 static enum request
3018 tree_request(struct view *view, enum request request, struct line *line)
3019 {
3020         enum open_flags flags;
3021
3022         if (request != REQ_ENTER)
3023                 return request;
3024
3025         /* Cleanup the stack if the tree view is at a different tree. */
3026         while (!*opt_path && tree_stack)
3027                 pop_tree_stack_entry();
3028
3029         switch (line->type) {
3030         case LINE_TREE_DIR:
3031                 /* Depending on whether it is a subdir or parent (updir?) link
3032                  * mangle the path buffer. */
3033                 if (line == &view->line[1] && *opt_path) {
3034                         pop_tree_stack_entry();
3035
3036                 } else {
3037                         char *data = line->data;
3038                         char *basename = data + SIZEOF_TREE_ATTR;
3039
3040                         push_tree_stack_entry(basename, view->lineno);
3041                 }
3042
3043                 /* Trees and subtrees share the same ID, so they are not not
3044                  * unique like blobs. */
3045                 flags = OPEN_RELOAD;
3046                 request = REQ_VIEW_TREE;
3047                 break;
3048
3049         case LINE_TREE_FILE:
3050                 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
3051                 request = REQ_VIEW_BLOB;
3052                 break;
3053
3054         default:
3055                 return TRUE;
3056         }
3057
3058         open_view(view, request, flags);
3059         if (request == REQ_VIEW_TREE) {
3060                 view->lineno = tree_lineno;
3061         }
3062
3063         return REQ_NONE;
3064 }
3065
3066 static void
3067 tree_select(struct view *view, struct line *line)
3068 {
3069         char *text = line->data + STRING_SIZE("100644 blob ");
3070
3071         if (line->type == LINE_TREE_FILE) {
3072                 string_copy_rev(ref_blob, text);
3073
3074         } else if (line->type != LINE_TREE_DIR) {
3075                 return;
3076         }
3077
3078         string_copy_rev(view->ref, text);
3079 }
3080
3081 static struct view_ops tree_ops = {
3082         "file",
3083         NULL,
3084         tree_read,
3085         pager_draw,
3086         tree_request,
3087         pager_grep,
3088         tree_select,
3089 };
3090
3091 static bool
3092 blob_read(struct view *view, char *line)
3093 {
3094         return add_line_text(view, line, LINE_DEFAULT) != NULL;
3095 }
3096
3097 static struct view_ops blob_ops = {
3098         "line",
3099         NULL,
3100         blob_read,
3101         pager_draw,
3102         pager_request,
3103         pager_grep,
3104         pager_select,
3105 };
3106
3107
3108 /*
3109  * Status backend
3110  */
3111
3112 struct status {
3113         char status;
3114         struct {
3115                 mode_t mode;
3116                 char rev[SIZEOF_REV];
3117         } old;
3118         struct {
3119                 mode_t mode;
3120                 char rev[SIZEOF_REV];
3121         } new;
3122         char name[SIZEOF_STR];
3123 };
3124
3125 static struct status stage_status;
3126 static enum line_type stage_line_type;
3127
3128 /* Get fields from the diff line:
3129  * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
3130  */
3131 static inline bool
3132 status_get_diff(struct status *file, char *buf, size_t bufsize)
3133 {
3134         char *old_mode = buf +  1;
3135         char *new_mode = buf +  8;
3136         char *old_rev  = buf + 15;
3137         char *new_rev  = buf + 56;
3138         char *status   = buf + 97;
3139
3140         if (bufsize != 99 ||
3141             old_mode[-1] != ':' ||
3142             new_mode[-1] != ' ' ||
3143             old_rev[-1]  != ' ' ||
3144             new_rev[-1]  != ' ' ||
3145             status[-1]   != ' ')
3146                 return FALSE;
3147
3148         file->status = *status;
3149
3150         string_copy_rev(file->old.rev, old_rev);
3151         string_copy_rev(file->new.rev, new_rev);
3152
3153         file->old.mode = strtoul(old_mode, NULL, 8);
3154         file->new.mode = strtoul(new_mode, NULL, 8);
3155
3156         file->name[0] = 0;
3157
3158         return TRUE;
3159 }
3160
3161 static bool
3162 status_run(struct view *view, const char cmd[], bool diff, enum line_type type)
3163 {
3164         struct status *file = NULL;
3165         struct status *unmerged = NULL;
3166         char buf[SIZEOF_STR * 4];
3167         size_t bufsize = 0;
3168         FILE *pipe;
3169
3170         pipe = popen(cmd, "r");
3171         if (!pipe)
3172                 return FALSE;
3173
3174         add_line_data(view, NULL, type);
3175
3176         while (!feof(pipe) && !ferror(pipe)) {
3177                 char *sep;
3178                 size_t readsize;
3179
3180                 readsize = fread(buf + bufsize, 1, sizeof(buf) - bufsize, pipe);
3181                 if (!readsize)
3182                         break;
3183                 bufsize += readsize;
3184
3185                 /* Process while we have NUL chars. */
3186                 while ((sep = memchr(buf, 0, bufsize))) {
3187                         size_t sepsize = sep - buf + 1;
3188
3189                         if (!file) {
3190                                 if (!realloc_lines(view, view->line_size + 1))
3191                                         goto error_out;
3192
3193                                 file = calloc(1, sizeof(*file));
3194                                 if (!file)
3195                                         goto error_out;
3196
3197                                 add_line_data(view, file, type);
3198                         }
3199
3200                         /* Parse diff info part. */
3201                         if (!diff) {
3202                                 file->status = '?';
3203
3204                         } else if (!file->status) {
3205                                 if (!status_get_diff(file, buf, sepsize))
3206                                         goto error_out;
3207
3208                                 bufsize -= sepsize;
3209                                 memmove(buf, sep + 1, bufsize);
3210
3211                                 sep = memchr(buf, 0, bufsize);
3212                                 if (!sep)
3213                                         break;
3214                                 sepsize = sep - buf + 1;
3215
3216                                 /* Collapse all 'M'odified entries that
3217                                  * follow a associated 'U'nmerged entry.
3218                                  */
3219                                 if (file->status == 'U') {
3220                                         unmerged = file;
3221
3222                                 } else if (unmerged) {
3223                                         int collapse = !strcmp(buf, unmerged->name);
3224
3225                                         unmerged = NULL;
3226                                         if (collapse) {
3227                                                 free(file);
3228                                                 view->lines--;
3229                                                 continue;
3230                                         }
3231                                 }
3232                         }
3233
3234                         /* git-ls-files just delivers a NUL separated
3235                          * list of file names similar to the second half
3236                          * of the git-diff-* output. */
3237                         string_ncopy(file->name, buf, sepsize);
3238                         bufsize -= sepsize;
3239                         memmove(buf, sep + 1, bufsize);
3240                         file = NULL;
3241                 }
3242         }
3243
3244         if (ferror(pipe)) {
3245 error_out:
3246                 pclose(pipe);
3247                 return FALSE;
3248         }
3249
3250         if (!view->line[view->lines - 1].data)
3251                 add_line_data(view, NULL, LINE_STAT_NONE);
3252
3253         pclose(pipe);
3254         return TRUE;
3255 }
3256
3257 /* Don't show unmerged entries in the staged section. */
3258 #define STATUS_DIFF_INDEX_CMD "git diff-index -z --diff-filter=ACDMRTXB --cached HEAD"
3259 #define STATUS_DIFF_FILES_CMD "git diff-files -z"
3260 #define STATUS_LIST_OTHER_CMD \
3261         "git ls-files -z --others --exclude-per-directory=.gitignore"
3262
3263 #define STATUS_DIFF_INDEX_SHOW_CMD \
3264         "git diff-index --root --patch-with-stat --find-copies-harder -B -C --cached HEAD -- %s 2>/dev/null"
3265
3266 #define STATUS_DIFF_FILES_SHOW_CMD \
3267         "git diff-files --root --patch-with-stat --find-copies-harder -B -C -- %s 2>/dev/null"
3268
3269 /* First parse staged info using git-diff-index(1), then parse unstaged
3270  * info using git-diff-files(1), and finally untracked files using
3271  * git-ls-files(1). */
3272 static bool
3273 status_open(struct view *view)
3274 {
3275         struct stat statbuf;
3276         char exclude[SIZEOF_STR];
3277         char cmd[SIZEOF_STR];
3278         unsigned long prev_lineno = view->lineno;
3279         size_t i;
3280
3281         for (i = 0; i < view->lines; i++)
3282                 free(view->line[i].data);
3283         free(view->line);
3284         view->lines = view->line_size = view->lineno = 0;
3285         view->line = NULL;
3286
3287         if (!realloc_lines(view, view->line_size + 6))
3288                 return FALSE;
3289
3290         if (!string_format(exclude, "%s/info/exclude", opt_git_dir))
3291                 return FALSE;
3292
3293         string_copy(cmd, STATUS_LIST_OTHER_CMD);
3294
3295         if (stat(exclude, &statbuf) >= 0) {
3296                 size_t cmdsize = strlen(cmd);
3297
3298                 if (!string_format_from(cmd, &cmdsize, " %s", "--exclude-from=") ||
3299                     sq_quote(cmd, cmdsize, exclude) >= sizeof(cmd))
3300                         return FALSE;
3301         }
3302
3303         if (!status_run(view, STATUS_DIFF_INDEX_CMD, TRUE, LINE_STAT_STAGED) ||
3304             !status_run(view, STATUS_DIFF_FILES_CMD, TRUE, LINE_STAT_UNSTAGED) ||
3305             !status_run(view, cmd, FALSE, LINE_STAT_UNTRACKED))
3306                 return FALSE;
3307
3308         /* If all went well restore the previous line number to stay in
3309          * the context. */
3310         if (prev_lineno < view->lines)
3311                 view->lineno = prev_lineno;
3312         else
3313                 view->lineno = view->lines - 1;
3314
3315         return TRUE;
3316 }
3317
3318 static bool
3319 status_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
3320 {
3321         struct status *status = line->data;
3322
3323         wmove(view->win, lineno, 0);
3324
3325         if (selected) {
3326                 wattrset(view->win, get_line_attr(LINE_CURSOR));
3327                 wchgat(view->win, -1, 0, LINE_CURSOR, NULL);
3328
3329         } else if (!status && line->type != LINE_STAT_NONE) {
3330                 wattrset(view->win, get_line_attr(LINE_STAT_SECTION));
3331                 wchgat(view->win, -1, 0, LINE_STAT_SECTION, NULL);
3332
3333         } else {
3334                 wattrset(view->win, get_line_attr(line->type));
3335         }
3336
3337         if (!status) {
3338                 char *text;
3339
3340                 switch (line->type) {
3341                 case LINE_STAT_STAGED:
3342                         text = "Changes to be committed:";
3343                         break;
3344
3345                 case LINE_STAT_UNSTAGED:
3346                         text = "Changed but not updated:";
3347                         break;
3348
3349                 case LINE_STAT_UNTRACKED:
3350                         text = "Untracked files:";
3351                         break;
3352
3353                 case LINE_STAT_NONE:
3354                         text = "    (no files)";
3355                         break;
3356
3357                 default:
3358                         return FALSE;
3359                 }
3360
3361                 waddstr(view->win, text);
3362                 return TRUE;
3363         }
3364
3365         waddch(view->win, status->status);
3366         if (!selected)
3367                 wattrset(view->win, A_NORMAL);
3368         wmove(view->win, lineno, 4);
3369         waddstr(view->win, status->name);
3370
3371         return TRUE;
3372 }
3373
3374 static enum request
3375 status_enter(struct view *view, struct line *line)
3376 {
3377         struct status *status = line->data;
3378         char path[SIZEOF_STR] = "";
3379         char *info;
3380         size_t cmdsize = 0;
3381
3382         if (line->type == LINE_STAT_NONE ||
3383             (!status && line[1].type == LINE_STAT_NONE)) {
3384                 report("No file to diff");
3385                 return REQ_NONE;
3386         }
3387
3388         if (status && sq_quote(path, 0, status->name) >= sizeof(path))
3389                 return REQ_QUIT;
3390
3391         if (opt_cdup[0] &&
3392             line->type != LINE_STAT_UNTRACKED &&
3393             !string_format_from(opt_cmd, &cmdsize, "cd %s;", opt_cdup))
3394                 return REQ_QUIT;
3395
3396         switch (line->type) {
3397         case LINE_STAT_STAGED:
3398                 if (!string_format_from(opt_cmd, &cmdsize,
3399                                         STATUS_DIFF_INDEX_SHOW_CMD, path))
3400                         return REQ_QUIT;
3401                 if (status)
3402                         info = "Staged changes to %s";
3403                 else
3404                         info = "Staged changes";
3405                 break;
3406
3407         case LINE_STAT_UNSTAGED:
3408                 if (!string_format_from(opt_cmd, &cmdsize,
3409                                         STATUS_DIFF_FILES_SHOW_CMD, path))
3410                         return REQ_QUIT;
3411                 if (status)
3412                         info = "Unstaged changes to %s";
3413                 else
3414                         info = "Unstaged changes";
3415                 break;
3416
3417         case LINE_STAT_UNTRACKED:
3418                 if (opt_pipe)
3419                         return REQ_QUIT;
3420
3421
3422                 if (!status) {
3423                         report("No file to show");
3424                         return REQ_NONE;
3425                 }
3426
3427                 opt_pipe = fopen(status->name, "r");
3428                 info = "Untracked file %s";
3429                 break;
3430
3431         default:
3432                 die("line type %d not handled in switch", line->type);
3433         }
3434
3435         open_view(view, REQ_VIEW_STAGE, OPEN_RELOAD | OPEN_SPLIT);
3436         if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
3437                 if (status) {
3438                         stage_status = *status;
3439                 } else {
3440                         memset(&stage_status, 0, sizeof(stage_status));
3441                 }
3442
3443                 stage_line_type = line->type;
3444                 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.name);
3445         }
3446
3447         return REQ_NONE;
3448 }
3449
3450
3451 static bool
3452 status_update_file(struct view *view, struct status *status, enum line_type type)
3453 {
3454         char cmd[SIZEOF_STR];
3455         char buf[SIZEOF_STR];
3456         size_t cmdsize = 0;
3457         size_t bufsize = 0;
3458         size_t written = 0;
3459         FILE *pipe;
3460
3461         if (opt_cdup[0] &&
3462             type != LINE_STAT_UNTRACKED &&
3463             !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
3464                 return FALSE;
3465
3466         switch (type) {
3467         case LINE_STAT_STAGED:
3468                 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
3469                                         status->old.mode,
3470                                         status->old.rev,
3471                                         status->name, 0))
3472                         return FALSE;
3473
3474                 string_add(cmd, cmdsize, "git update-index -z --index-info");
3475                 break;
3476
3477         case LINE_STAT_UNSTAGED:
3478         case LINE_STAT_UNTRACKED:
3479                 if (!string_format_from(buf, &bufsize, "%s%c", status->name, 0))
3480                         return FALSE;
3481
3482                 string_add(cmd, cmdsize, "git update-index -z --add --remove --stdin");
3483                 break;
3484
3485         default:
3486                 die("line type %d not handled in switch", type);
3487         }
3488
3489         pipe = popen(cmd, "w");
3490         if (!pipe)
3491                 return FALSE;
3492
3493         while (!ferror(pipe) && written < bufsize) {
3494                 written += fwrite(buf + written, 1, bufsize - written, pipe);
3495         }
3496
3497         pclose(pipe);
3498
3499         if (written != bufsize)
3500                 return FALSE;
3501
3502         return TRUE;
3503 }
3504
3505 static void
3506 status_update(struct view *view)
3507 {
3508         struct line *line = &view->line[view->lineno];
3509
3510         assert(view->lines);
3511
3512         if (!line->data) {
3513                 while (++line < view->line + view->lines && line->data) {
3514                         if (!status_update_file(view, line->data, line->type))
3515                                 report("Failed to update file status");
3516                 }
3517
3518                 if (!line[-1].data) {
3519                         report("Nothing to update");
3520                         return;
3521                 }
3522
3523         } else if (!status_update_file(view, line->data, line->type)) {
3524                 report("Failed to update file status");
3525         }
3526 }
3527
3528 static enum request
3529 status_request(struct view *view, enum request request, struct line *line)
3530 {
3531         struct status *status = line->data;
3532
3533         switch (request) {
3534         case REQ_STATUS_UPDATE:
3535                 status_update(view);
3536                 break;
3537
3538         case REQ_STATUS_MERGE:
3539                 if (!status || status->status != 'U') {
3540                         report("Merging only possible for files with unmerged status ('U').");
3541                         return REQ_NONE;
3542                 }
3543                 open_mergetool(status->name);
3544                 break;
3545
3546         case REQ_EDIT:
3547                 if (!status)
3548                         return request;
3549
3550                 open_editor(status->status != '?', status->name);
3551                 break;
3552
3553         case REQ_ENTER:
3554                 /* After returning the status view has been split to
3555                  * show the stage view. No further reloading is
3556                  * necessary. */
3557                 status_enter(view, line);
3558                 return REQ_NONE;
3559
3560         case REQ_REFRESH:
3561                 /* Simply reload the view. */
3562                 break;
3563
3564         default:
3565                 return request;
3566         }
3567
3568         open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
3569
3570         return REQ_NONE;
3571 }
3572
3573 static void
3574 status_select(struct view *view, struct line *line)
3575 {
3576         struct status *status = line->data;
3577         char file[SIZEOF_STR] = "all files";
3578         char *text;
3579         char *key;
3580
3581         if (status && !string_format(file, "'%s'", status->name))
3582                 return;
3583
3584         if (!status && line[1].type == LINE_STAT_NONE)
3585                 line++;
3586
3587         switch (line->type) {
3588         case LINE_STAT_STAGED:
3589                 text = "Press %s to unstage %s for commit";
3590                 break;
3591
3592         case LINE_STAT_UNSTAGED:
3593                 text = "Press %s to stage %s for commit";
3594                 break;
3595
3596         case LINE_STAT_UNTRACKED:
3597                 text = "Press %s to stage %s for addition";
3598                 break;
3599
3600         case LINE_STAT_NONE:
3601                 text = "Nothing to update";
3602                 break;
3603
3604         default:
3605                 die("line type %d not handled in switch", line->type);
3606         }
3607
3608         if (status && status->status == 'U') {
3609                 text = "Press %s to resolve conflict in %s";
3610                 key = get_key(REQ_STATUS_MERGE);
3611
3612         } else {
3613                 key = get_key(REQ_STATUS_UPDATE);
3614         }
3615
3616         string_format(view->ref, text, key, file);
3617 }
3618
3619 static bool
3620 status_grep(struct view *view, struct line *line)
3621 {
3622         struct status *status = line->data;
3623         enum { S_STATUS, S_NAME, S_END } state;
3624         char buf[2] = "?";
3625         regmatch_t pmatch;
3626
3627         if (!status)
3628                 return FALSE;
3629
3630         for (state = S_STATUS; state < S_END; state++) {
3631                 char *text;
3632
3633                 switch (state) {
3634                 case S_NAME:    text = status->name;    break;
3635                 case S_STATUS:
3636                         buf[0] = status->status;
3637                         text = buf;
3638                         break;
3639
3640                 default:
3641                         return FALSE;
3642                 }
3643
3644                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
3645                         return TRUE;
3646         }
3647
3648         return FALSE;
3649 }
3650
3651 static struct view_ops status_ops = {
3652         "file",
3653         status_open,
3654         NULL,
3655         status_draw,
3656         status_request,
3657         status_grep,
3658         status_select,
3659 };
3660
3661
3662 static bool
3663 stage_diff_line(FILE *pipe, struct line *line)
3664 {
3665         char *buf = line->data;
3666         size_t bufsize = strlen(buf);
3667         size_t written = 0;
3668
3669         while (!ferror(pipe) && written < bufsize) {
3670                 written += fwrite(buf + written, 1, bufsize - written, pipe);
3671         }
3672
3673         fputc('\n', pipe);
3674
3675         return written == bufsize;
3676 }
3677
3678 static struct line *
3679 stage_diff_hdr(struct view *view, struct line *line)
3680 {
3681         int diff_hdr_dir = line->type == LINE_DIFF_CHUNK ? -1 : 1;
3682         struct line *diff_hdr;
3683
3684         if (line->type == LINE_DIFF_CHUNK)
3685                 diff_hdr = line - 1;
3686         else
3687                 diff_hdr = view->line + 1;
3688
3689         while (diff_hdr > view->line && diff_hdr < view->line + view->lines) {
3690                 if (diff_hdr->type == LINE_DIFF_HEADER)
3691                         return diff_hdr;
3692
3693                 diff_hdr += diff_hdr_dir;
3694         }
3695
3696         return NULL;
3697 }
3698
3699 static bool
3700 stage_update_chunk(struct view *view, struct line *line)
3701 {
3702         char cmd[SIZEOF_STR];
3703         size_t cmdsize = 0;
3704         struct line *diff_hdr, *diff_chunk, *diff_end;
3705         FILE *pipe;
3706
3707         diff_hdr = stage_diff_hdr(view, line);
3708         if (!diff_hdr)
3709                 return FALSE;
3710
3711         if (opt_cdup[0] &&
3712             !string_format_from(cmd, &cmdsize, "cd %s;", opt_cdup))
3713                 return FALSE;
3714
3715         if (!string_format_from(cmd, &cmdsize,
3716                                 "git apply --cached %s - && "
3717                                 "git update-index -q --unmerged --refresh 2>/dev/null",
3718                                 stage_line_type == LINE_STAT_STAGED ? "-R" : ""))
3719                 return FALSE;
3720
3721         pipe = popen(cmd, "w");
3722         if (!pipe)
3723                 return FALSE;
3724
3725         diff_end = view->line + view->lines;
3726         if (line->type != LINE_DIFF_CHUNK) {
3727                 diff_chunk = diff_hdr;
3728
3729         } else {
3730                 for (diff_chunk = line + 1; diff_chunk < diff_end; diff_chunk++)
3731                         if (diff_chunk->type == LINE_DIFF_CHUNK ||
3732                             diff_chunk->type == LINE_DIFF_HEADER)
3733                                 diff_end = diff_chunk;
3734
3735                 diff_chunk = line;
3736
3737                 while (diff_hdr->type != LINE_DIFF_CHUNK) {
3738                         switch (diff_hdr->type) {
3739                         case LINE_DIFF_HEADER:
3740                         case LINE_DIFF_INDEX:
3741                         case LINE_DIFF_ADD:
3742                         case LINE_DIFF_DEL:
3743                                 break;
3744
3745                         default:
3746                                 diff_hdr++;
3747                                 continue;
3748                         }
3749
3750                         if (!stage_diff_line(pipe, diff_hdr++)) {
3751                                 pclose(pipe);
3752                                 return FALSE;
3753                         }
3754                 }
3755         }
3756
3757         while (diff_chunk < diff_end && stage_diff_line(pipe, diff_chunk))
3758                 diff_chunk++;
3759
3760         pclose(pipe);
3761
3762         if (diff_chunk != diff_end)
3763                 return FALSE;
3764
3765         return TRUE;
3766 }
3767
3768 static void
3769 stage_update(struct view *view, struct line *line)
3770 {
3771         if (stage_line_type != LINE_STAT_UNTRACKED &&
3772             (line->type == LINE_DIFF_CHUNK || !stage_status.status)) {
3773                 if (!stage_update_chunk(view, line)) {
3774                         report("Failed to apply chunk");
3775                         return;
3776                 }
3777
3778         } else if (!status_update_file(view, &stage_status, stage_line_type)) {
3779                 report("Failed to update file");
3780                 return;
3781         }
3782
3783         open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
3784
3785         view = VIEW(REQ_VIEW_STATUS);
3786         if (view_is_displayed(view))
3787                 status_enter(view, &view->line[view->lineno]);
3788 }
3789
3790 static enum request
3791 stage_request(struct view *view, enum request request, struct line *line)
3792 {
3793         switch (request) {
3794         case REQ_STATUS_UPDATE:
3795                 stage_update(view, line);
3796                 break;
3797
3798         case REQ_EDIT:
3799                 if (!stage_status.name[0])
3800                         return request;
3801
3802                 open_editor(stage_status.status != '?', stage_status.name);
3803                 break;
3804
3805         case REQ_ENTER:
3806                 pager_request(view, request, line);
3807                 break;
3808
3809         default:
3810                 return request;
3811         }
3812
3813         return REQ_NONE;
3814 }
3815
3816 static struct view_ops stage_ops = {
3817         "line",
3818         NULL,
3819         pager_read,
3820         pager_draw,
3821         stage_request,
3822         pager_grep,
3823         pager_select,
3824 };
3825
3826
3827 /*
3828  * Revision graph
3829  */
3830
3831 struct commit {
3832         char id[SIZEOF_REV];            /* SHA1 ID. */
3833         char title[128];                /* First line of the commit message. */
3834         char author[75];                /* Author of the commit. */
3835         struct tm time;                 /* Date from the author ident. */
3836         struct ref **refs;              /* Repository references. */
3837         chtype graph[SIZEOF_REVGRAPH];  /* Ancestry chain graphics. */
3838         size_t graph_size;              /* The width of the graph array. */
3839 };
3840
3841 /* Size of rev graph with no  "padding" columns */
3842 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
3843
3844 struct rev_graph {
3845         struct rev_graph *prev, *next, *parents;
3846         char rev[SIZEOF_REVITEMS][SIZEOF_REV];
3847         size_t size;
3848         struct commit *commit;
3849         size_t pos;
3850 };
3851
3852 /* Parents of the commit being visualized. */
3853 static struct rev_graph graph_parents[4];
3854
3855 /* The current stack of revisions on the graph. */
3856 static struct rev_graph graph_stacks[4] = {
3857         { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
3858         { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
3859         { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
3860         { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
3861 };
3862
3863 static inline bool
3864 graph_parent_is_merge(struct rev_graph *graph)
3865 {
3866         return graph->parents->size > 1;
3867 }
3868
3869 static inline void
3870 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
3871 {
3872         struct commit *commit = graph->commit;
3873
3874         if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
3875                 commit->graph[commit->graph_size++] = symbol;
3876 }
3877
3878 static void
3879 done_rev_graph(struct rev_graph *graph)
3880 {
3881         if (graph_parent_is_merge(graph) &&
3882             graph->pos < graph->size - 1 &&
3883             graph->next->size == graph->size + graph->parents->size - 1) {
3884                 size_t i = graph->pos + graph->parents->size - 1;
3885
3886                 graph->commit->graph_size = i * 2;
3887                 while (i < graph->next->size - 1) {
3888                         append_to_rev_graph(graph, ' ');
3889                         append_to_rev_graph(graph, '\\');
3890                         i++;
3891                 }
3892         }
3893
3894         graph->size = graph->pos = 0;
3895         graph->commit = NULL;
3896         memset(graph->parents, 0, sizeof(*graph->parents));
3897 }
3898
3899 static void
3900 push_rev_graph(struct rev_graph *graph, char *parent)
3901 {
3902         int i;
3903
3904         /* "Collapse" duplicate parents lines.
3905          *
3906          * FIXME: This needs to also update update the drawn graph but
3907          * for now it just serves as a method for pruning graph lines. */
3908         for (i = 0; i < graph->size; i++)
3909                 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
3910                         return;
3911
3912         if (graph->size < SIZEOF_REVITEMS) {
3913                 string_copy_rev(graph->rev[graph->size++], parent);
3914         }
3915 }
3916
3917 static chtype
3918 get_rev_graph_symbol(struct rev_graph *graph)
3919 {
3920         chtype symbol;
3921
3922         if (graph->parents->size == 0)
3923                 symbol = REVGRAPH_INIT;
3924         else if (graph_parent_is_merge(graph))
3925                 symbol = REVGRAPH_MERGE;
3926         else if (graph->pos >= graph->size)
3927                 symbol = REVGRAPH_BRANCH;
3928         else
3929                 symbol = REVGRAPH_COMMIT;
3930
3931         return symbol;
3932 }
3933
3934 static void
3935 draw_rev_graph(struct rev_graph *graph)
3936 {
3937         struct rev_filler {
3938                 chtype separator, line;
3939         };
3940         enum { DEFAULT, RSHARP, RDIAG, LDIAG };
3941         static struct rev_filler fillers[] = {
3942                 { ' ',  REVGRAPH_LINE },
3943                 { '`',  '.' },
3944                 { '\'', ' ' },
3945                 { '/',  ' ' },
3946         };
3947         chtype symbol = get_rev_graph_symbol(graph);
3948         struct rev_filler *filler;
3949         size_t i;
3950
3951         filler = &fillers[DEFAULT];
3952
3953         for (i = 0; i < graph->pos; i++) {
3954                 append_to_rev_graph(graph, filler->line);
3955                 if (graph_parent_is_merge(graph->prev) &&
3956                     graph->prev->pos == i)
3957                         filler = &fillers[RSHARP];
3958
3959                 append_to_rev_graph(graph, filler->separator);
3960         }
3961
3962         /* Place the symbol for this revision. */
3963         append_to_rev_graph(graph, symbol);
3964
3965         if (graph->prev->size > graph->size)
3966                 filler = &fillers[RDIAG];
3967         else
3968                 filler = &fillers[DEFAULT];
3969
3970         i++;
3971
3972         for (; i < graph->size; i++) {
3973                 append_to_rev_graph(graph, filler->separator);
3974                 append_to_rev_graph(graph, filler->line);
3975                 if (graph_parent_is_merge(graph->prev) &&
3976                     i < graph->prev->pos + graph->parents->size)
3977                         filler = &fillers[RSHARP];
3978                 if (graph->prev->size > graph->size)
3979                         filler = &fillers[LDIAG];
3980         }
3981
3982         if (graph->prev->size > graph->size) {
3983                 append_to_rev_graph(graph, filler->separator);
3984                 if (filler->line != ' ')
3985                         append_to_rev_graph(graph, filler->line);
3986         }
3987 }
3988
3989 /* Prepare the next rev graph */
3990 static void
3991 prepare_rev_graph(struct rev_graph *graph)
3992 {
3993         size_t i;
3994
3995         /* First, traverse all lines of revisions up to the active one. */
3996         for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
3997                 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
3998                         break;
3999
4000                 push_rev_graph(graph->next, graph->rev[graph->pos]);
4001         }
4002
4003         /* Interleave the new revision parent(s). */
4004         for (i = 0; i < graph->parents->size; i++)
4005                 push_rev_graph(graph->next, graph->parents->rev[i]);
4006
4007         /* Lastly, put any remaining revisions. */
4008         for (i = graph->pos + 1; i < graph->size; i++)
4009                 push_rev_graph(graph->next, graph->rev[i]);
4010 }
4011
4012 static void
4013 update_rev_graph(struct rev_graph *graph)
4014 {
4015         /* If this is the finalizing update ... */
4016         if (graph->commit)
4017                 prepare_rev_graph(graph);
4018
4019         /* Graph visualization needs a one rev look-ahead,
4020          * so the first update doesn't visualize anything. */
4021         if (!graph->prev->commit)
4022                 return;
4023
4024         draw_rev_graph(graph->prev);
4025         done_rev_graph(graph->prev->prev);
4026 }
4027
4028
4029 /*
4030  * Main view backend
4031  */
4032
4033 static bool
4034 main_draw(struct view *view, struct line *line, unsigned int lineno, bool selected)
4035 {
4036         char buf[DATE_COLS + 1];
4037         struct commit *commit = line->data;
4038         enum line_type type;
4039         int col = 0;
4040         size_t timelen;
4041         size_t authorlen;
4042         int trimmed = 1;
4043
4044         if (!*commit->author)
4045                 return FALSE;
4046
4047         wmove(view->win, lineno, col);
4048
4049         if (selected) {
4050                 type = LINE_CURSOR;
4051                 wattrset(view->win, get_line_attr(type));
4052                 wchgat(view->win, -1, 0, type, NULL);
4053
4054         } else {
4055                 type = LINE_MAIN_COMMIT;
4056                 wattrset(view->win, get_line_attr(LINE_MAIN_DATE));
4057         }
4058
4059         timelen = strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time);
4060         waddnstr(view->win, buf, timelen);
4061         waddstr(view->win, " ");
4062
4063         col += DATE_COLS;
4064         wmove(view->win, lineno, col);
4065         if (type != LINE_CURSOR)
4066                 wattrset(view->win, get_line_attr(LINE_MAIN_AUTHOR));
4067
4068         if (opt_utf8) {
4069                 authorlen = utf8_length(commit->author, AUTHOR_COLS - 2, &col, &trimmed);
4070         } else {
4071                 authorlen = strlen(commit->author);
4072                 if (authorlen > AUTHOR_COLS - 2) {
4073                         authorlen = AUTHOR_COLS - 2;
4074                         trimmed = 1;
4075                 }
4076         }
4077
4078         if (trimmed) {
4079                 waddnstr(view->win, commit->author, authorlen);
4080                 if (type != LINE_CURSOR)
4081                         wattrset(view->win, get_line_attr(LINE_MAIN_DELIM));
4082                 waddch(view->win, '~');
4083         } else {
4084                 waddstr(view->win, commit->author);
4085         }
4086
4087         col += AUTHOR_COLS;
4088         if (type != LINE_CURSOR)
4089                 wattrset(view->win, A_NORMAL);
4090
4091         if (opt_rev_graph && commit->graph_size) {
4092                 size_t i;
4093
4094                 wmove(view->win, lineno, col);
4095                 /* Using waddch() instead of waddnstr() ensures that
4096                  * they'll be rendered correctly for the cursor line. */
4097                 for (i = 0; i < commit->graph_size; i++)
4098                         waddch(view->win, commit->graph[i]);
4099
4100                 waddch(view->win, ' ');
4101                 col += commit->graph_size + 1;
4102         }
4103
4104         wmove(view->win, lineno, col);
4105
4106         if (commit->refs) {
4107                 size_t i = 0;
4108
4109                 do {
4110                         if (type == LINE_CURSOR)
4111                                 ;
4112                         else if (commit->refs[i]->tag)
4113                                 wattrset(view->win, get_line_attr(LINE_MAIN_TAG));
4114                         else if (commit->refs[i]->remote)
4115                                 wattrset(view->win, get_line_attr(LINE_MAIN_REMOTE));
4116                         else
4117                                 wattrset(view->win, get_line_attr(LINE_MAIN_REF));
4118                         waddstr(view->win, "[");
4119                         waddstr(view->win, commit->refs[i]->name);
4120                         waddstr(view->win, "]");
4121                         if (type != LINE_CURSOR)
4122                                 wattrset(view->win, A_NORMAL);
4123                         waddstr(view->win, " ");
4124                         col += strlen(commit->refs[i]->name) + STRING_SIZE("[] ");
4125                 } while (commit->refs[i++]->next);
4126         }
4127
4128         if (type != LINE_CURSOR)
4129                 wattrset(view->win, get_line_attr(type));
4130
4131         {
4132                 int titlelen = strlen(commit->title);
4133
4134                 if (col + titlelen > view->width)
4135                         titlelen = view->width - col;
4136
4137                 waddnstr(view->win, commit->title, titlelen);
4138         }
4139
4140         return TRUE;
4141 }
4142
4143 /* Reads git log --pretty=raw output and parses it into the commit struct. */
4144 static bool
4145 main_read(struct view *view, char *line)
4146 {
4147         static struct rev_graph *graph = graph_stacks;
4148         enum line_type type;
4149         struct commit *commit;
4150
4151         if (!line) {
4152                 update_rev_graph(graph);
4153                 return TRUE;
4154         }
4155
4156         type = get_line_type(line);
4157         if (type == LINE_COMMIT) {
4158                 commit = calloc(1, sizeof(struct commit));
4159                 if (!commit)
4160                         return FALSE;
4161
4162                 string_copy_rev(commit->id, line + STRING_SIZE("commit "));
4163                 commit->refs = get_refs(commit->id);
4164                 graph->commit = commit;
4165                 add_line_data(view, commit, LINE_MAIN_COMMIT);
4166                 return TRUE;
4167         }
4168
4169         if (!view->lines)
4170                 return TRUE;
4171         commit = view->line[view->lines - 1].data;
4172
4173         switch (type) {
4174         case LINE_PARENT:
4175                 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
4176                 break;
4177
4178         case LINE_AUTHOR:
4179         {
4180                 /* Parse author lines where the name may be empty:
4181                  *      author  <email@address.tld> 1138474660 +0100
4182                  */
4183                 char *ident = line + STRING_SIZE("author ");
4184                 char *nameend = strchr(ident, '<');
4185                 char *emailend = strchr(ident, '>');
4186
4187                 if (!nameend || !emailend)
4188                         break;
4189
4190                 update_rev_graph(graph);
4191                 graph = graph->next;
4192
4193                 *nameend = *emailend = 0;
4194                 ident = chomp_string(ident);
4195                 if (!*ident) {
4196                         ident = chomp_string(nameend + 1);
4197                         if (!*ident)
4198                                 ident = "Unknown";
4199                 }
4200
4201                 string_ncopy(commit->author, ident, strlen(ident));
4202
4203                 /* Parse epoch and timezone */
4204                 if (emailend[1] == ' ') {
4205                         char *secs = emailend + 2;
4206                         char *zone = strchr(secs, ' ');
4207                         time_t time = (time_t) atol(secs);
4208
4209                         if (zone && strlen(zone) == STRING_SIZE(" +0700")) {
4210                                 long tz;
4211
4212                                 zone++;
4213                                 tz  = ('0' - zone[1]) * 60 * 60 * 10;
4214                                 tz += ('0' - zone[2]) * 60 * 60;
4215                                 tz += ('0' - zone[3]) * 60;
4216                                 tz += ('0' - zone[4]) * 60;
4217
4218                                 if (zone[0] == '-')
4219                                         tz = -tz;
4220
4221                                 time -= tz;
4222                         }
4223
4224                         gmtime_r(&time, &commit->time);
4225                 }
4226                 break;
4227         }
4228         default:
4229                 /* Fill in the commit title if it has not already been set. */
4230                 if (commit->title[0])
4231                         break;
4232
4233                 /* Require titles to start with a non-space character at the
4234                  * offset used by git log. */
4235                 if (strncmp(line, "    ", 4))
4236                         break;
4237                 line += 4;
4238                 /* Well, if the title starts with a whitespace character,
4239                  * try to be forgiving.  Otherwise we end up with no title. */
4240                 while (isspace(*line))
4241                         line++;
4242                 if (*line == '\0')
4243                         break;
4244                 /* FIXME: More graceful handling of titles; append "..." to
4245                  * shortened titles, etc. */
4246
4247                 string_ncopy(commit->title, line, strlen(line));
4248         }
4249
4250         return TRUE;
4251 }
4252
4253 static enum request
4254 main_request(struct view *view, enum request request, struct line *line)
4255 {
4256         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4257
4258         if (request == REQ_ENTER)
4259                 open_view(view, REQ_VIEW_DIFF, flags);
4260         else
4261                 return request;
4262
4263         return REQ_NONE;
4264 }
4265
4266 static bool
4267 main_grep(struct view *view, struct line *line)
4268 {
4269         struct commit *commit = line->data;
4270         enum { S_TITLE, S_AUTHOR, S_DATE, S_END } state;
4271         char buf[DATE_COLS + 1];
4272         regmatch_t pmatch;
4273
4274         for (state = S_TITLE; state < S_END; state++) {
4275                 char *text;
4276
4277                 switch (state) {
4278                 case S_TITLE:   text = commit->title;   break;
4279                 case S_AUTHOR:  text = commit->author;  break;
4280                 case S_DATE:
4281                         if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
4282                                 continue;
4283                         text = buf;
4284                         break;
4285
4286                 default:
4287                         return FALSE;
4288                 }
4289
4290                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4291                         return TRUE;
4292         }
4293
4294         return FALSE;
4295 }
4296
4297 static void
4298 main_select(struct view *view, struct line *line)
4299 {
4300         struct commit *commit = line->data;
4301
4302         string_copy_rev(view->ref, commit->id);
4303         string_copy_rev(ref_commit, view->ref);
4304 }
4305
4306 static struct view_ops main_ops = {
4307         "commit",
4308         NULL,
4309         main_read,
4310         main_draw,
4311         main_request,
4312         main_grep,
4313         main_select,
4314 };
4315
4316
4317 /*
4318  * Unicode / UTF-8 handling
4319  *
4320  * NOTE: Much of the following code for dealing with unicode is derived from
4321  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
4322  * src/intl/charset.c from the utf8 branch commit elinks-0.11.0-g31f2c28.
4323  */
4324
4325 /* I've (over)annotated a lot of code snippets because I am not entirely
4326  * confident that the approach taken by this small UTF-8 interface is correct.
4327  * --jonas */
4328
4329 static inline int
4330 unicode_width(unsigned long c)
4331 {
4332         if (c >= 0x1100 &&
4333            (c <= 0x115f                         /* Hangul Jamo */
4334             || c == 0x2329
4335             || c == 0x232a
4336             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
4337                                                 /* CJK ... Yi */
4338             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
4339             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
4340             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
4341             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
4342             || (c >= 0xffe0  && c <= 0xffe6)
4343             || (c >= 0x20000 && c <= 0x2fffd)
4344             || (c >= 0x30000 && c <= 0x3fffd)))
4345                 return 2;
4346
4347         return 1;
4348 }
4349
4350 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
4351  * Illegal bytes are set one. */
4352 static const unsigned char utf8_bytes[256] = {
4353         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,
4354         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,
4355         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,
4356         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,
4357         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,
4358         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,
4359         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,
4360         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,
4361 };
4362
4363 /* Decode UTF-8 multi-byte representation into a unicode character. */
4364 static inline unsigned long
4365 utf8_to_unicode(const char *string, size_t length)
4366 {
4367         unsigned long unicode;
4368
4369         switch (length) {
4370         case 1:
4371                 unicode  =   string[0];
4372                 break;
4373         case 2:
4374                 unicode  =  (string[0] & 0x1f) << 6;
4375                 unicode +=  (string[1] & 0x3f);
4376                 break;
4377         case 3:
4378                 unicode  =  (string[0] & 0x0f) << 12;
4379                 unicode += ((string[1] & 0x3f) << 6);
4380                 unicode +=  (string[2] & 0x3f);
4381                 break;
4382         case 4:
4383                 unicode  =  (string[0] & 0x0f) << 18;
4384                 unicode += ((string[1] & 0x3f) << 12);
4385                 unicode += ((string[2] & 0x3f) << 6);
4386                 unicode +=  (string[3] & 0x3f);
4387                 break;
4388         case 5:
4389                 unicode  =  (string[0] & 0x0f) << 24;
4390                 unicode += ((string[1] & 0x3f) << 18);
4391                 unicode += ((string[2] & 0x3f) << 12);
4392                 unicode += ((string[3] & 0x3f) << 6);
4393                 unicode +=  (string[4] & 0x3f);
4394                 break;
4395         case 6:
4396                 unicode  =  (string[0] & 0x01) << 30;
4397                 unicode += ((string[1] & 0x3f) << 24);
4398                 unicode += ((string[2] & 0x3f) << 18);
4399                 unicode += ((string[3] & 0x3f) << 12);
4400                 unicode += ((string[4] & 0x3f) << 6);
4401                 unicode +=  (string[5] & 0x3f);
4402                 break;
4403         default:
4404                 die("Invalid unicode length");
4405         }
4406
4407         /* Invalid characters could return the special 0xfffd value but NUL
4408          * should be just as good. */
4409         return unicode > 0xffff ? 0 : unicode;
4410 }
4411
4412 /* Calculates how much of string can be shown within the given maximum width
4413  * and sets trimmed parameter to non-zero value if all of string could not be
4414  * shown.
4415  *
4416  * Additionally, adds to coloffset how many many columns to move to align with
4417  * the expected position. Takes into account how multi-byte and double-width
4418  * characters will effect the cursor position.
4419  *
4420  * Returns the number of bytes to output from string to satisfy max_width. */
4421 static size_t
4422 utf8_length(const char *string, size_t max_width, int *coloffset, int *trimmed)
4423 {
4424         const char *start = string;
4425         const char *end = strchr(string, '\0');
4426         size_t mbwidth = 0;
4427         size_t width = 0;
4428
4429         *trimmed = 0;
4430
4431         while (string < end) {
4432                 int c = *(unsigned char *) string;
4433                 unsigned char bytes = utf8_bytes[c];
4434                 size_t ucwidth;
4435                 unsigned long unicode;
4436
4437                 if (string + bytes > end)
4438                         break;
4439
4440                 /* Change representation to figure out whether
4441                  * it is a single- or double-width character. */
4442
4443                 unicode = utf8_to_unicode(string, bytes);
4444                 /* FIXME: Graceful handling of invalid unicode character. */
4445                 if (!unicode)
4446                         break;
4447
4448                 ucwidth = unicode_width(unicode);
4449                 width  += ucwidth;
4450                 if (width > max_width) {
4451                         *trimmed = 1;
4452                         break;
4453                 }
4454
4455                 /* The column offset collects the differences between the
4456                  * number of bytes encoding a character and the number of
4457                  * columns will be used for rendering said character.
4458                  *
4459                  * So if some character A is encoded in 2 bytes, but will be
4460                  * represented on the screen using only 1 byte this will and up
4461                  * adding 1 to the multi-byte column offset.
4462                  *
4463                  * Assumes that no double-width character can be encoding in
4464                  * less than two bytes. */
4465                 if (bytes > ucwidth)
4466                         mbwidth += bytes - ucwidth;
4467
4468                 string  += bytes;
4469         }
4470
4471         *coloffset += mbwidth;
4472
4473         return string - start;
4474 }
4475
4476
4477 /*
4478  * Status management
4479  */
4480
4481 /* Whether or not the curses interface has been initialized. */
4482 static bool cursed = FALSE;
4483
4484 /* The status window is used for polling keystrokes. */
4485 static WINDOW *status_win;
4486
4487 static bool status_empty = TRUE;
4488
4489 /* Update status and title window. */
4490 static void
4491 report(const char *msg, ...)
4492 {
4493         struct view *view = display[current_view];
4494
4495         if (input_mode)
4496                 return;
4497
4498         if (!view) {
4499                 char buf[SIZEOF_STR];
4500                 va_list args;
4501
4502                 va_start(args, msg);
4503                 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
4504                         buf[sizeof(buf) - 1] = 0;
4505                         buf[sizeof(buf) - 2] = '.';
4506                         buf[sizeof(buf) - 3] = '.';
4507                         buf[sizeof(buf) - 4] = '.';
4508                 }
4509                 va_end(args);
4510                 die("%s", buf);
4511         }
4512
4513         if (!status_empty || *msg) {
4514                 va_list args;
4515
4516                 va_start(args, msg);
4517
4518                 wmove(status_win, 0, 0);
4519                 if (*msg) {
4520                         vwprintw(status_win, msg, args);
4521                         status_empty = FALSE;
4522                 } else {
4523                         status_empty = TRUE;
4524                 }
4525                 wclrtoeol(status_win);
4526                 wrefresh(status_win);
4527
4528                 va_end(args);
4529         }
4530
4531         update_view_title(view);
4532         update_display_cursor(view);
4533 }
4534
4535 /* Controls when nodelay should be in effect when polling user input. */
4536 static void
4537 set_nonblocking_input(bool loading)
4538 {
4539         static unsigned int loading_views;
4540
4541         if ((loading == FALSE && loading_views-- == 1) ||
4542             (loading == TRUE  && loading_views++ == 0))
4543                 nodelay(status_win, loading);
4544 }
4545
4546 static void
4547 init_display(void)
4548 {
4549         int x, y;
4550
4551         /* Initialize the curses library */
4552         if (isatty(STDIN_FILENO)) {
4553                 cursed = !!initscr();
4554         } else {
4555                 /* Leave stdin and stdout alone when acting as a pager. */
4556                 FILE *io = fopen("/dev/tty", "r+");
4557
4558                 if (!io)
4559                         die("Failed to open /dev/tty");
4560                 cursed = !!newterm(NULL, io, io);
4561         }
4562
4563         if (!cursed)
4564                 die("Failed to initialize curses");
4565
4566         nonl();         /* Tell curses not to do NL->CR/NL on output */
4567         cbreak();       /* Take input chars one at a time, no wait for \n */
4568         noecho();       /* Don't echo input */
4569         leaveok(stdscr, TRUE);
4570
4571         if (has_colors())
4572                 init_colors();
4573
4574         getmaxyx(stdscr, y, x);
4575         status_win = newwin(1, 0, y - 1, 0);
4576         if (!status_win)
4577                 die("Failed to create status window");
4578
4579         /* Enable keyboard mapping */
4580         keypad(status_win, TRUE);
4581         wbkgdset(status_win, get_line_attr(LINE_STATUS));
4582 }
4583
4584 static char *
4585 read_prompt(const char *prompt)
4586 {
4587         enum { READING, STOP, CANCEL } status = READING;
4588         static char buf[sizeof(opt_cmd) - STRING_SIZE("git \0")];
4589         int pos = 0;
4590
4591         while (status == READING) {
4592                 struct view *view;
4593                 int i, key;
4594
4595                 input_mode = TRUE;
4596
4597                 foreach_view (view, i)
4598                         update_view(view);
4599
4600                 input_mode = FALSE;
4601
4602                 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
4603                 wclrtoeol(status_win);
4604
4605                 /* Refresh, accept single keystroke of input */
4606                 key = wgetch(status_win);
4607                 switch (key) {
4608                 case KEY_RETURN:
4609                 case KEY_ENTER:
4610                 case '\n':
4611                         status = pos ? STOP : CANCEL;
4612                         break;
4613
4614                 case KEY_BACKSPACE:
4615                         if (pos > 0)
4616                                 pos--;
4617                         else
4618                                 status = CANCEL;
4619                         break;
4620
4621                 case KEY_ESC:
4622                         status = CANCEL;
4623                         break;
4624
4625                 case ERR:
4626                         break;
4627
4628                 default:
4629                         if (pos >= sizeof(buf)) {
4630                                 report("Input string too long");
4631                                 return NULL;
4632                         }
4633
4634                         if (isprint(key))
4635                                 buf[pos++] = (char) key;
4636                 }
4637         }
4638
4639         /* Clear the status window */
4640         status_empty = FALSE;
4641         report("");
4642
4643         if (status == CANCEL)
4644                 return NULL;
4645
4646         buf[pos++] = 0;
4647
4648         return buf;
4649 }
4650
4651 /*
4652  * Repository references
4653  */
4654
4655 static struct ref *refs;
4656 static size_t refs_size;
4657
4658 /* Id <-> ref store */
4659 static struct ref ***id_refs;
4660 static size_t id_refs_size;
4661
4662 static struct ref **
4663 get_refs(char *id)
4664 {
4665         struct ref ***tmp_id_refs;
4666         struct ref **ref_list = NULL;
4667         size_t ref_list_size = 0;
4668         size_t i;
4669
4670         for (i = 0; i < id_refs_size; i++)
4671                 if (!strcmp(id, id_refs[i][0]->id))
4672                         return id_refs[i];
4673
4674         tmp_id_refs = realloc(id_refs, (id_refs_size + 1) * sizeof(*id_refs));
4675         if (!tmp_id_refs)
4676                 return NULL;
4677
4678         id_refs = tmp_id_refs;
4679
4680         for (i = 0; i < refs_size; i++) {
4681                 struct ref **tmp;
4682
4683                 if (strcmp(id, refs[i].id))
4684                         continue;
4685
4686                 tmp = realloc(ref_list, (ref_list_size + 1) * sizeof(*ref_list));
4687                 if (!tmp) {
4688                         if (ref_list)
4689                                 free(ref_list);
4690                         return NULL;
4691                 }
4692
4693                 ref_list = tmp;
4694                 if (ref_list_size > 0)
4695                         ref_list[ref_list_size - 1]->next = 1;
4696                 ref_list[ref_list_size] = &refs[i];
4697
4698                 /* XXX: The properties of the commit chains ensures that we can
4699                  * safely modify the shared ref. The repo references will
4700                  * always be similar for the same id. */
4701                 ref_list[ref_list_size]->next = 0;
4702                 ref_list_size++;
4703         }
4704
4705         if (ref_list)
4706                 id_refs[id_refs_size++] = ref_list;
4707
4708         return ref_list;
4709 }
4710
4711 static int
4712 read_ref(char *id, size_t idlen, char *name, size_t namelen)
4713 {
4714         struct ref *ref;
4715         bool tag = FALSE;
4716         bool remote = FALSE;
4717
4718         if (!strncmp(name, "refs/tags/", STRING_SIZE("refs/tags/"))) {
4719                 /* Commits referenced by tags has "^{}" appended. */
4720                 if (name[namelen - 1] != '}')
4721                         return OK;
4722
4723                 while (namelen > 0 && name[namelen] != '^')
4724                         namelen--;
4725
4726                 tag = TRUE;
4727                 namelen -= STRING_SIZE("refs/tags/");
4728                 name    += STRING_SIZE("refs/tags/");
4729
4730         } else if (!strncmp(name, "refs/remotes/", STRING_SIZE("refs/remotes/"))) {
4731                 remote = TRUE;
4732                 namelen -= STRING_SIZE("refs/remotes/");
4733                 name    += STRING_SIZE("refs/remotes/");
4734
4735         } else if (!strncmp(name, "refs/heads/", STRING_SIZE("refs/heads/"))) {
4736                 namelen -= STRING_SIZE("refs/heads/");
4737                 name    += STRING_SIZE("refs/heads/");
4738
4739         } else if (!strcmp(name, "HEAD")) {
4740                 return OK;
4741         }
4742
4743         refs = realloc(refs, sizeof(*refs) * (refs_size + 1));
4744         if (!refs)
4745                 return ERR;
4746
4747         ref = &refs[refs_size++];
4748         ref->name = malloc(namelen + 1);
4749         if (!ref->name)
4750                 return ERR;
4751
4752         strncpy(ref->name, name, namelen);
4753         ref->name[namelen] = 0;
4754         ref->tag = tag;
4755         ref->remote = remote;
4756         string_copy_rev(ref->id, id);
4757
4758         return OK;
4759 }
4760
4761 static int
4762 load_refs(void)
4763 {
4764         const char *cmd_env = getenv("TIG_LS_REMOTE");
4765         const char *cmd = cmd_env && *cmd_env ? cmd_env : TIG_LS_REMOTE;
4766
4767         return read_properties(popen(cmd, "r"), "\t", read_ref);
4768 }
4769
4770 static int
4771 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
4772 {
4773         if (!strcmp(name, "i18n.commitencoding"))
4774                 string_ncopy(opt_encoding, value, valuelen);
4775
4776         if (!strcmp(name, "core.editor"))
4777                 string_ncopy(opt_editor, value, valuelen);
4778
4779         return OK;
4780 }
4781
4782 static int
4783 load_repo_config(void)
4784 {
4785         return read_properties(popen(GIT_CONFIG " --list", "r"),
4786                                "=", read_repo_config_option);
4787 }
4788
4789 static int
4790 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
4791 {
4792         if (!opt_git_dir[0]) {
4793                 string_ncopy(opt_git_dir, name, namelen);
4794
4795         } else if (opt_is_inside_work_tree == -1) {
4796                 /* This can be 3 different values depending on the
4797                  * version of git being used. If git-rev-parse does not
4798                  * understand --is-inside-work-tree it will simply echo
4799                  * the option else either "true" or "false" is printed.
4800                  * Default to true for the unknown case. */
4801                 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
4802
4803         } else {
4804                 string_ncopy(opt_cdup, name, namelen);
4805         }
4806
4807         return OK;
4808 }
4809
4810 /* XXX: The line outputted by "--show-cdup" can be empty so the option
4811  * must be the last one! */
4812 static int
4813 load_repo_info(void)
4814 {
4815         return read_properties(popen("git rev-parse --git-dir --is-inside-work-tree --show-cdup 2>/dev/null", "r"),
4816                                "=", read_repo_info);
4817 }
4818
4819 static int
4820 read_properties(FILE *pipe, const char *separators,
4821                 int (*read_property)(char *, size_t, char *, size_t))
4822 {
4823         char buffer[BUFSIZ];
4824         char *name;
4825         int state = OK;
4826
4827         if (!pipe)
4828                 return ERR;
4829
4830         while (state == OK && (name = fgets(buffer, sizeof(buffer), pipe))) {
4831                 char *value;
4832                 size_t namelen;
4833                 size_t valuelen;
4834
4835                 name = chomp_string(name);
4836                 namelen = strcspn(name, separators);
4837
4838                 if (name[namelen]) {
4839                         name[namelen] = 0;
4840                         value = chomp_string(name + namelen + 1);
4841                         valuelen = strlen(value);
4842
4843                 } else {
4844                         value = "";
4845                         valuelen = 0;
4846                 }
4847
4848                 state = read_property(name, namelen, value, valuelen);
4849         }
4850
4851         if (state != ERR && ferror(pipe))
4852                 state = ERR;
4853
4854         pclose(pipe);
4855
4856         return state;
4857 }
4858
4859
4860 /*
4861  * Main
4862  */
4863
4864 static void __NORETURN
4865 quit(int sig)
4866 {
4867         /* XXX: Restore tty modes and let the OS cleanup the rest! */
4868         if (cursed)
4869                 endwin();
4870         exit(0);
4871 }
4872
4873 static void __NORETURN
4874 die(const char *err, ...)
4875 {
4876         va_list args;
4877
4878         endwin();
4879
4880         va_start(args, err);
4881         fputs("tig: ", stderr);
4882         vfprintf(stderr, err, args);
4883         fputs("\n", stderr);
4884         va_end(args);
4885
4886         exit(1);
4887 }
4888
4889 int
4890 main(int argc, char *argv[])
4891 {
4892         struct view *view;
4893         enum request request;
4894         size_t i;
4895
4896         signal(SIGINT, quit);
4897
4898         if (setlocale(LC_ALL, "")) {
4899                 char *codeset = nl_langinfo(CODESET);
4900
4901                 string_ncopy(opt_codeset, codeset, strlen(codeset));
4902         }
4903
4904         if (load_repo_info() == ERR)
4905                 die("Failed to load repo info.");
4906
4907         if (load_options() == ERR)
4908                 die("Failed to load user config.");
4909
4910         /* Load the repo config file so options can be overwritten from
4911          * the command line. */
4912         if (load_repo_config() == ERR)
4913                 die("Failed to load repo config.");
4914
4915         if (!parse_options(argc, argv))
4916                 return 0;
4917
4918         /* Require a git repository unless when running in pager mode. */
4919         if (!opt_git_dir[0])
4920                 die("Not a git repository");
4921
4922         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
4923                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
4924                 if (opt_iconv == ICONV_NONE)
4925                         die("Failed to initialize character set conversion");
4926         }
4927
4928         if (load_refs() == ERR)
4929                 die("Failed to load refs.");
4930
4931         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
4932                 view->cmd_env = getenv(view->cmd_env);
4933
4934         request = opt_request;
4935
4936         init_display();
4937
4938         while (view_driver(display[current_view], request)) {
4939                 int key;
4940                 int i;
4941
4942                 foreach_view (view, i)
4943                         update_view(view);
4944
4945                 /* Refresh, accept single keystroke of input */
4946                 key = wgetch(status_win);
4947
4948                 /* wgetch() with nodelay() enabled returns ERR when there's no
4949                  * input. */
4950                 if (key == ERR) {
4951                         request = REQ_NONE;
4952                         continue;
4953                 }
4954
4955                 request = get_keybinding(display[current_view]->keymap, key);
4956
4957                 /* Some low-level request handling. This keeps access to
4958                  * status_win restricted. */
4959                 switch (request) {
4960                 case REQ_PROMPT:
4961                 {
4962                         char *cmd = read_prompt(":");
4963
4964                         if (cmd && string_format(opt_cmd, "git %s", cmd)) {
4965                                 if (strncmp(cmd, "show", 4) && isspace(cmd[4])) {
4966                                         opt_request = REQ_VIEW_DIFF;
4967                                 } else {
4968                                         opt_request = REQ_VIEW_PAGER;
4969                                 }
4970                                 break;
4971                         }
4972
4973                         request = REQ_NONE;
4974                         break;
4975                 }
4976                 case REQ_SEARCH:
4977                 case REQ_SEARCH_BACK:
4978                 {
4979                         const char *prompt = request == REQ_SEARCH
4980                                            ? "/" : "?";
4981                         char *search = read_prompt(prompt);
4982
4983                         if (search)
4984                                 string_ncopy(opt_search, search, strlen(search));
4985                         else
4986                                 request = REQ_NONE;
4987                         break;
4988                 }
4989                 case REQ_SCREEN_RESIZE:
4990                 {
4991                         int height, width;
4992
4993                         getmaxyx(stdscr, height, width);
4994
4995                         /* Resize the status view and let the view driver take
4996                          * care of resizing the displayed views. */
4997                         wresize(status_win, 1, width);
4998                         mvwin(status_win, height - 1, 0);
4999                         wrefresh(status_win);
5000                         break;
5001                 }
5002                 default:
5003                         break;
5004                 }
5005         }
5006
5007         quit(0);
5008
5009         return 0;
5010 }