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

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