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