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