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

   1 /* Copyright (c) 2006-2009 Jonas Fonseca <fonseca@diku.dk>
   2  *
   3  * This program is free software; you can redistribute it and/or
   4  * modify it under the terms of the GNU General Public License as
   5  * published by the Free Software Foundation; either version 2 of
   6  * the License, or (at your option) any later version.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  */
  13
  14 #ifdef HAVE_CONFIG_H
  15 #include "config.h"
  16 #endif
  17
  18 #ifndef TIG_VERSION
  19 #define TIG_VERSION "unknown-version"
  20 #endif
  21
  22 #ifndef DEBUG
  23 #define NDEBUG
  24 #endif
  25
  26 #include <assert.h>
  27 #include <errno.h>
  28 #include <ctype.h>
  29 #include <signal.h>
  30 #include <stdarg.h>
  31 #include <stdio.h>
  32 #include <stdlib.h>
  33 #include <string.h>
  34 #include <sys/types.h>
  35 #include <sys/wait.h>
  36 #include <sys/stat.h>
  37 #include <sys/select.h>
  38 #include <unistd.h>
  39 #include <time.h>
  40 #include <fcntl.h>
  41
  42 #include <regex.h>
  43
  44 #include <locale.h>
  45 #include <langinfo.h>
  46 #include <iconv.h>
  47
  48 /* ncurses(3): Must be defined to have extended wide-character functions. */
  49 #define _XOPEN_SOURCE_EXTENDED
  50
  51 #ifdef HAVE_NCURSESW_NCURSES_H
  52 #include <ncursesw/ncurses.h>
  53 #else
  54 #ifdef HAVE_NCURSES_NCURSES_H
  55 #include <ncurses/ncurses.h>
  56 #else
  57 #include <ncurses.h>
  58 #endif
  59 #endif
  60
  61 #if __GNUC__ >= 3
  62 #define __NORETURN __attribute__((__noreturn__))
  63 #else
  64 #define __NORETURN
  65 #endif
  66
  67 static void __NORETURN die(const char *err, ...);
  68 static void warn(const char *msg, ...);
  69 static void report(const char *msg, ...);
  70 static void set_nonblocking_input(bool loading);
  71 static int load_refs(void);
  72 static size_t utf8_length(const char **string, size_t col, int *width, size_t max_width, int *trimmed, bool reserve);
  73
  74 #define ABS(x)          ((x) >= 0  ? (x) : -(x))
  75 #define MIN(x, y)       ((x) < (y) ? (x) :  (y))
  76
  77 #define ARRAY_SIZE(x)   (sizeof(x) / sizeof(x[0]))
  78 #define STRING_SIZE(x)  (sizeof(x) - 1)
  79
  80 #define SIZEOF_STR      1024    /* Default string size. */
  81 #define SIZEOF_REF      256     /* Size of symbolic or SHA1 ID. */
  82 #define SIZEOF_REV      41      /* Holds a SHA-1 and an ending NUL. */
  83 #define SIZEOF_ARG      32      /* Default argument array size. */
  84
  85 /* Revision graph */
  86
  87 #define REVGRAPH_INIT   'I'
  88 #define REVGRAPH_MERGE  'M'
  89 #define REVGRAPH_BRANCH '+'
  90 #define REVGRAPH_COMMIT '*'
  91 #define REVGRAPH_BOUND  '^'
  92
  93 #define SIZEOF_REVGRAPH 19      /* Size of revision ancestry graphics. */
  94
  95 /* This color name can be used to refer to the default term colors. */
  96 #define COLOR_DEFAULT   (-1)
  97
  98 #define ICONV_NONE      ((iconv_t) -1)
  99 #ifndef ICONV_CONST
 100 #define ICONV_CONST     /* nothing */
 101 #endif
 102
 103 /* The format and size of the date column in the main view. */
 104 #define DATE_FORMAT     "%Y-%m-%d %H:%M"
 105 #define DATE_COLS       STRING_SIZE("2006-04-29 14:21 ")
 106
 107 #define AUTHOR_COLS     20
 108 #define ID_COLS         8
 109
 110 /* The default interval between line numbers. */
 111 #define NUMBER_INTERVAL 5
 112
 113 #define TAB_SIZE        8
 114
 115 #define SCALE_SPLIT_VIEW(height)        ((height) * 2 / 3)
 116
 117 #define NULL_ID         "0000000000000000000000000000000000000000"
 118
 119 #define S_ISGITLINK(mode) (((mode) & S_IFMT) == 0160000)
 120
 121 #ifndef GIT_CONFIG
 122 #define GIT_CONFIG "config"
 123 #endif
 124
 125 /* Some ASCII-shorthands fitted into the ncurses namespace. */
 126 #define KEY_TAB         '\t'
 127 #define KEY_RETURN      '\r'
 128 #define KEY_ESC         27
 129
 130
 131 struct ref {
 132         char *name;             /* Ref name; tag or head names are shortened. */
 133         char id[SIZEOF_REV];    /* Commit SHA1 ID */
 134         unsigned int head:1;    /* Is it the current HEAD? */
 135         unsigned int tag:1;     /* Is it a tag? */
 136         unsigned int ltag:1;    /* If so, is the tag local? */
 137         unsigned int remote:1;  /* Is it a remote ref? */
 138         unsigned int tracked:1; /* Is it the remote for the current HEAD? */
 139         unsigned int next:1;    /* For ref lists: are there more refs? */
 140 };
 141
 142 static struct ref **get_refs(const char *id);
 143
 144 enum format_flags {
 145         FORMAT_ALL,             /* Perform replacement in all arguments. */
 146         FORMAT_DASH,            /* Perform replacement up until "--". */
 147         FORMAT_NONE             /* No replacement should be performed. */
 148 };
 149
 150 static bool format_argv(const char *dst[], const char *src[], enum format_flags flags);
 151
 152 enum input_status {
 153         INPUT_OK,
 154         INPUT_SKIP,
 155         INPUT_STOP,
 156         INPUT_CANCEL
 157 };
 158
 159 typedef enum input_status (*input_handler)(void *data, char *buf, int c);
 160
 161 static char *prompt_input(const char *prompt, input_handler handler, void *data);
 162 static bool prompt_yesno(const char *prompt);
 163
 164 /*
 165  * String helpers
 166  */
 167
 168 static inline void
 169 string_ncopy_do(char *dst, size_t dstlen, const char *src, size_t srclen)
 170 {
 171         if (srclen > dstlen - 1)
 172                 srclen = dstlen - 1;
 173
 174         strncpy(dst, src, srclen);
 175         dst[srclen] = 0;
 176 }
 177
 178 /* Shorthands for safely copying into a fixed buffer. */
 179
 180 #define string_copy(dst, src) \
 181         string_ncopy_do(dst, sizeof(dst), src, sizeof(src))
 182
 183 #define string_ncopy(dst, src, srclen) \
 184         string_ncopy_do(dst, sizeof(dst), src, srclen)
 185
 186 #define string_copy_rev(dst, src) \
 187         string_ncopy_do(dst, SIZEOF_REV, src, SIZEOF_REV - 1)
 188
 189 #define string_add(dst, from, src) \
 190         string_ncopy_do(dst + (from), sizeof(dst) - (from), src, sizeof(src))
 191
 192 static void
 193 string_expand(char *dst, size_t dstlen, const char *src, int tabsize)
 194 {
 195         size_t size, pos;
 196
 197         for (size = pos = 0; size < dstlen - 1 && src[pos]; pos++) {
 198                 if (src[pos] == '\t') {
 199                         size_t expanded = tabsize - (size % tabsize);
 200
 201                         if (expanded + size >= dstlen - 1)
 202                                 expanded = dstlen - size - 1;
 203                         memcpy(dst + size, "        ", expanded);
 204                         size += expanded;
 205                 } else {
 206                         dst[size++] = src[pos];
 207                 }
 208         }
 209
 210         dst[size] = 0;
 211 }
 212
 213 static char *
 214 chomp_string(char *name)
 215 {
 216         int namelen;
 217
 218         while (isspace(*name))
 219                 name++;
 220
 221         namelen = strlen(name) - 1;
 222         while (namelen > 0 && isspace(name[namelen]))
 223                 name[namelen--] = 0;
 224
 225         return name;
 226 }
 227
 228 static bool
 229 string_nformat(char *buf, size_t bufsize, size_t *bufpos, const char *fmt, ...)
 230 {
 231         va_list args;
 232         size_t pos = bufpos ? *bufpos : 0;
 233
 234         va_start(args, fmt);
 235         pos += vsnprintf(buf + pos, bufsize - pos, fmt, args);
 236         va_end(args);
 237
 238         if (bufpos)
 239                 *bufpos = pos;
 240
 241         return pos >= bufsize ? FALSE : TRUE;
 242 }
 243
 244 #define string_format(buf, fmt, args...) \
 245         string_nformat(buf, sizeof(buf), NULL, fmt, args)
 246
 247 #define string_format_from(buf, from, fmt, args...) \
 248         string_nformat(buf, sizeof(buf), from, fmt, args)
 249
 250 static int
 251 string_enum_compare(const char *str1, const char *str2, int len)
 252 {
 253         size_t i;
 254
 255 #define string_enum_sep(x) ((x) == '-' || (x) == '_' || (x) == '.')
 256
 257         /* Diff-Header == DIFF_HEADER */
 258         for (i = 0; i < len; i++) {
 259                 if (toupper(str1[i]) == toupper(str2[i]))
 260                         continue;
 261
 262                 if (string_enum_sep(str1[i]) &&
 263                     string_enum_sep(str2[i]))
 264                         continue;
 265
 266                 return str1[i] - str2[i];
 267         }
 268
 269         return 0;
 270 }
 271
 272 struct enum_map {
 273         const char *name;
 274         int namelen;
 275         int value;
 276 };
 277
 278 #define ENUM_MAP(name, value) { name, STRING_SIZE(name), value }
 279
 280 static bool
 281 map_enum_do(const struct enum_map *map, size_t map_size, int *value, const char *name)
 282 {
 283         size_t namelen = strlen(name);
 284         int i;
 285
 286         for (i = 0; i < map_size; i++)
 287                 if (namelen == map[i].namelen &&
 288                     !string_enum_compare(name, map[i].name, namelen)) {
 289                         *value = map[i].value;
 290                         return TRUE;
 291                 }
 292
 293         return FALSE;
 294 }
 295
 296 #define map_enum(attr, map, name) \
 297         map_enum_do(map, ARRAY_SIZE(map), attr, name)
 298
 299 #define prefixcmp(str1, str2) \
 300         strncmp(str1, str2, STRING_SIZE(str2))
 301
 302 static inline int
 303 suffixcmp(const char *str, int slen, const char *suffix)
 304 {
 305         size_t len = slen >= 0 ? slen : strlen(str);
 306         size_t suffixlen = strlen(suffix);
 307
 308         return suffixlen < len ? strcmp(str + len - suffixlen, suffix) : -1;
 309 }
 310
 311
 312 static bool
 313 argv_from_string(const char *argv[SIZEOF_ARG], int *argc, char *cmd)
 314 {
 315         int valuelen;
 316
 317         while (*cmd && *argc < SIZEOF_ARG && (valuelen = strcspn(cmd, " \t"))) {
 318                 bool advance = cmd[valuelen] != 0;
 319
 320                 cmd[valuelen] = 0;
 321                 argv[(*argc)++] = chomp_string(cmd);
 322                 cmd = chomp_string(cmd + valuelen + advance);
 323         }
 324
 325         if (*argc < SIZEOF_ARG)
 326                 argv[*argc] = NULL;
 327         return *argc < SIZEOF_ARG;
 328 }
 329
 330 static void
 331 argv_from_env(const char **argv, const char *name)
 332 {
 333         char *env = argv ? getenv(name) : NULL;
 334         int argc = 0;
 335
 336         if (env && *env)
 337                 env = strdup(env);
 338         if (env && !argv_from_string(argv, &argc, env))
 339                 die("Too many arguments in the `%s` environment variable", name);
 340 }
 341
 342
 343 /*
 344  * Executing external commands.
 345  */
 346
 347 enum io_type {
 348         IO_FD,                  /* File descriptor based IO. */
 349         IO_BG,                  /* Execute command in the background. */
 350         IO_FG,                  /* Execute command with same std{in,out,err}. */
 351         IO_RD,                  /* Read only fork+exec IO. */
 352         IO_WR,                  /* Write only fork+exec IO. */
 353         IO_AP,                  /* Append fork+exec output to file. */
 354 };
 355
 356 struct io {
 357         enum io_type type;      /* The requested type of pipe. */
 358         const char *dir;        /* Directory from which to execute. */
 359         pid_t pid;              /* Pipe for reading or writing. */
 360         int pipe;               /* Pipe end for reading or writing. */
 361         int error;              /* Error status. */
 362         const char *argv[SIZEOF_ARG];   /* Shell command arguments. */
 363         char *buf;              /* Read buffer. */
 364         size_t bufalloc;        /* Allocated buffer size. */
 365         size_t bufsize;         /* Buffer content size. */
 366         char *bufpos;           /* Current buffer position. */
 367         unsigned int eof:1;     /* Has end of file been reached. */
 368 };
 369
 370 static void
 371 reset_io(struct io *io)
 372 {
 373         io->pipe = -1;
 374         io->pid = 0;
 375         io->buf = io->bufpos = NULL;
 376         io->bufalloc = io->bufsize = 0;
 377         io->error = 0;
 378         io->eof = 0;
 379 }
 380
 381 static void
 382 init_io(struct io *io, const char *dir, enum io_type type)
 383 {
 384         reset_io(io);
 385         io->type = type;
 386         io->dir = dir;
 387 }
 388
 389 static bool
 390 init_io_rd(struct io *io, const char *argv[], const char *dir,
 391                 enum format_flags flags)
 392 {
 393         init_io(io, dir, IO_RD);
 394         return format_argv(io->argv, argv, flags);
 395 }
 396
 397 static bool
 398 io_open(struct io *io, const char *name)
 399 {
 400         init_io(io, NULL, IO_FD);
 401         io->pipe = *name ? open(name, O_RDONLY) : STDIN_FILENO;
 402         if (io->pipe == -1)
 403                 io->error = errno;
 404         return io->pipe != -1;
 405 }
 406
 407 static bool
 408 kill_io(struct io *io)
 409 {
 410         return io->pid == 0 || kill(io->pid, SIGKILL) != -1;
 411 }
 412
 413 static bool
 414 done_io(struct io *io)
 415 {
 416         pid_t pid = io->pid;
 417
 418         if (io->pipe != -1)
 419                 close(io->pipe);
 420         free(io->buf);
 421         reset_io(io);
 422
 423         while (pid > 0) {
 424                 int status;
 425                 pid_t waiting = waitpid(pid, &status, 0);
 426
 427                 if (waiting < 0) {
 428                         if (errno == EINTR)
 429                                 continue;
 430                         report("waitpid failed (%s)", strerror(errno));
 431                         return FALSE;
 432                 }
 433
 434                 return waiting == pid &&
 435                        !WIFSIGNALED(status) &&
 436                        WIFEXITED(status) &&
 437                        !WEXITSTATUS(status);
 438         }
 439
 440         return TRUE;
 441 }
 442
 443 static bool
 444 start_io(struct io *io)
 445 {
 446         int pipefds[2] = { -1, -1 };
 447
 448         if (io->type == IO_FD)
 449                 return TRUE;
 450
 451         if ((io->type == IO_RD || io->type == IO_WR) &&
 452             pipe(pipefds) < 0)
 453                 return FALSE;
 454         else if (io->type == IO_AP)
 455                 pipefds[1] = io->pipe;
 456
 457         if ((io->pid = fork())) {
 458                 if (pipefds[!(io->type == IO_WR)] != -1)
 459                         close(pipefds[!(io->type == IO_WR)]);
 460                 if (io->pid != -1) {
 461                         io->pipe = pipefds[!!(io->type == IO_WR)];
 462                         return TRUE;
 463                 }
 464
 465         } else {
 466                 if (io->type != IO_FG) {
 467                         int devnull = open("/dev/null", O_RDWR);
 468                         int readfd  = io->type == IO_WR ? pipefds[0] : devnull;
 469                         int writefd = (io->type == IO_RD || io->type == IO_AP)
 470                                                         ? pipefds[1] : devnull;
 471
 472                         dup2(readfd,  STDIN_FILENO);
 473                         dup2(writefd, STDOUT_FILENO);
 474                         dup2(devnull, STDERR_FILENO);
 475
 476                         close(devnull);
 477                         if (pipefds[0] != -1)
 478                                 close(pipefds[0]);
 479                         if (pipefds[1] != -1)
 480                                 close(pipefds[1]);
 481                 }
 482
 483                 if (io->dir && *io->dir && chdir(io->dir) == -1)
 484                         die("Failed to change directory: %s", strerror(errno));
 485
 486                 execvp(io->argv[0], (char *const*) io->argv);
 487                 die("Failed to execute program: %s", strerror(errno));
 488         }
 489
 490         if (pipefds[!!(io->type == IO_WR)] != -1)
 491                 close(pipefds[!!(io->type == IO_WR)]);
 492         return FALSE;
 493 }
 494
 495 static bool
 496 run_io(struct io *io, const char **argv, const char *dir, enum io_type type)
 497 {
 498         init_io(io, dir, type);
 499         if (!format_argv(io->argv, argv, FORMAT_NONE))
 500                 return FALSE;
 501         return start_io(io);
 502 }
 503
 504 static int
 505 run_io_do(struct io *io)
 506 {
 507         return start_io(io) && done_io(io);
 508 }
 509
 510 static int
 511 run_io_bg(const char **argv)
 512 {
 513         struct io io = {};
 514
 515         init_io(&io, NULL, IO_BG);
 516         if (!format_argv(io.argv, argv, FORMAT_NONE))
 517                 return FALSE;
 518         return run_io_do(&io);
 519 }
 520
 521 static bool
 522 run_io_fg(const char **argv, const char *dir)
 523 {
 524         struct io io = {};
 525
 526         init_io(&io, dir, IO_FG);
 527         if (!format_argv(io.argv, argv, FORMAT_NONE))
 528                 return FALSE;
 529         return run_io_do(&io);
 530 }
 531
 532 static bool
 533 run_io_append(const char **argv, enum format_flags flags, int fd)
 534 {
 535         struct io io = {};
 536
 537         init_io(&io, NULL, IO_AP);
 538         io.pipe = fd;
 539         if (format_argv(io.argv, argv, flags))
 540                 return run_io_do(&io);
 541         close(fd);
 542         return FALSE;
 543 }
 544
 545 static bool
 546 run_io_rd(struct io *io, const char **argv, enum format_flags flags)
 547 {
 548         return init_io_rd(io, argv, NULL, flags) && start_io(io);
 549 }
 550
 551 static bool
 552 io_eof(struct io *io)
 553 {
 554         return io->eof;
 555 }
 556
 557 static int
 558 io_error(struct io *io)
 559 {
 560         return io->error;
 561 }
 562
 563 static char *
 564 io_strerror(struct io *io)
 565 {
 566         return strerror(io->error);
 567 }
 568
 569 static bool
 570 io_can_read(struct io *io)
 571 {
 572         struct timeval tv = { 0, 500 };
 573         fd_set fds;
 574
 575         FD_ZERO(&fds);
 576         FD_SET(io->pipe, &fds);
 577
 578         return select(io->pipe + 1, &fds, NULL, NULL, &tv) > 0;
 579 }
 580
 581 static ssize_t
 582 io_read(struct io *io, void *buf, size_t bufsize)
 583 {
 584         do {
 585                 ssize_t readsize = read(io->pipe, buf, bufsize);
 586
 587                 if (readsize < 0 && (errno == EAGAIN || errno == EINTR))
 588                         continue;
 589                 else if (readsize == -1)
 590                         io->error = errno;
 591                 else if (readsize == 0)
 592                         io->eof = 1;
 593                 return readsize;
 594         } while (1);
 595 }
 596
 597 static char *
 598 io_get(struct io *io, int c, bool can_read)
 599 {
 600         char *eol;
 601         ssize_t readsize;
 602
 603         if (!io->buf) {
 604                 io->buf = io->bufpos = malloc(BUFSIZ);
 605                 if (!io->buf)
 606                         return NULL;
 607                 io->bufalloc = BUFSIZ;
 608                 io->bufsize = 0;
 609         }
 610
 611         while (TRUE) {
 612                 if (io->bufsize > 0) {
 613                         eol = memchr(io->bufpos, c, io->bufsize);
 614                         if (eol) {
 615                                 char *line = io->bufpos;
 616
 617                                 *eol = 0;
 618                                 io->bufpos = eol + 1;
 619                                 io->bufsize -= io->bufpos - line;
 620                                 return line;
 621                         }
 622                 }
 623
 624                 if (io_eof(io)) {
 625                         if (io->bufsize) {
 626                                 io->bufpos[io->bufsize] = 0;
 627                                 io->bufsize = 0;
 628                                 return io->bufpos;
 629                         }
 630                         return NULL;
 631                 }
 632
 633                 if (!can_read)
 634                         return NULL;
 635
 636                 if (io->bufsize > 0 && io->bufpos > io->buf)
 637                         memmove(io->buf, io->bufpos, io->bufsize);
 638
 639                 io->bufpos = io->buf;
 640                 readsize = io_read(io, io->buf + io->bufsize, io->bufalloc - io->bufsize);
 641                 if (io_error(io))
 642                         return NULL;
 643                 io->bufsize += readsize;
 644         }
 645 }
 646
 647 static bool
 648 io_write(struct io *io, const void *buf, size_t bufsize)
 649 {
 650         size_t written = 0;
 651
 652         while (!io_error(io) && written < bufsize) {
 653                 ssize_t size;
 654
 655                 size = write(io->pipe, buf + written, bufsize - written);
 656                 if (size < 0 && (errno == EAGAIN || errno == EINTR))
 657                         continue;
 658                 else if (size == -1)
 659                         io->error = errno;
 660                 else
 661                         written += size;
 662         }
 663
 664         return written == bufsize;
 665 }
 666
 667 static bool
 668 io_read_buf(struct io *io, char buf[], size_t bufsize)
 669 {
 670         bool error;
 671
 672         io->buf = io->bufpos = buf;
 673         io->bufalloc = bufsize;
 674         error = !io_get(io, '\n', TRUE) && io_error(io);
 675         io->buf = NULL;
 676
 677         return done_io(io) || error;
 678 }
 679
 680 static bool
 681 run_io_buf(const char **argv, char buf[], size_t bufsize)
 682 {
 683         struct io io = {};
 684
 685         return run_io_rd(&io, argv, FORMAT_NONE) && io_read_buf(&io, buf, bufsize);
 686 }
 687
 688 static int
 689 io_load(struct io *io, const char *separators,
 690         int (*read_property)(char *, size_t, char *, size_t))
 691 {
 692         char *name;
 693         int state = OK;
 694
 695         if (!start_io(io))
 696                 return ERR;
 697
 698         while (state == OK && (name = io_get(io, '\n', TRUE))) {
 699                 char *value;
 700                 size_t namelen;
 701                 size_t valuelen;
 702
 703                 name = chomp_string(name);
 704                 namelen = strcspn(name, separators);
 705
 706                 if (name[namelen]) {
 707                         name[namelen] = 0;
 708                         value = chomp_string(name + namelen + 1);
 709                         valuelen = strlen(value);
 710
 711                 } else {
 712                         value = "";
 713                         valuelen = 0;
 714                 }
 715
 716                 state = read_property(name, namelen, value, valuelen);
 717         }
 718
 719         if (state != ERR && io_error(io))
 720                 state = ERR;
 721         done_io(io);
 722
 723         return state;
 724 }
 725
 726 static int
 727 run_io_load(const char **argv, const char *separators,
 728             int (*read_property)(char *, size_t, char *, size_t))
 729 {
 730         struct io io = {};
 731
 732         return init_io_rd(&io, argv, NULL, FORMAT_NONE)
 733                 ? io_load(&io, separators, read_property) : ERR;
 734 }
 735
 736
 737 /*
 738  * User requests
 739  */
 740
 741 #define REQ_INFO \
 742         /* XXX: Keep the view request first and in sync with views[]. */ \
 743         REQ_GROUP("View switching") \
 744         REQ_(VIEW_MAIN,         "Show main view"), \
 745         REQ_(VIEW_DIFF,         "Show diff view"), \
 746         REQ_(VIEW_LOG,          "Show log view"), \
 747         REQ_(VIEW_TREE,         "Show tree view"), \
 748         REQ_(VIEW_BLOB,         "Show blob view"), \
 749         REQ_(VIEW_BLAME,        "Show blame view"), \
 750         REQ_(VIEW_HELP,         "Show help page"), \
 751         REQ_(VIEW_PAGER,        "Show pager view"), \
 752         REQ_(VIEW_STATUS,       "Show status view"), \
 753         REQ_(VIEW_STAGE,        "Show stage view"), \
 754         \
 755         REQ_GROUP("View manipulation") \
 756         REQ_(ENTER,             "Enter current line and scroll"), \
 757         REQ_(NEXT,              "Move to next"), \
 758         REQ_(PREVIOUS,          "Move to previous"), \
 759         REQ_(PARENT,            "Move to parent"), \
 760         REQ_(VIEW_NEXT,         "Move focus to next view"), \
 761         REQ_(REFRESH,           "Reload and refresh"), \
 762         REQ_(MAXIMIZE,          "Maximize the current view"), \
 763         REQ_(VIEW_CLOSE,        "Close the current view"), \
 764         REQ_(QUIT,              "Close all views and quit"), \
 765         \
 766         REQ_GROUP("View specific requests") \
 767         REQ_(STATUS_UPDATE,     "Update file status"), \
 768         REQ_(STATUS_REVERT,     "Revert file changes"), \
 769         REQ_(STATUS_MERGE,      "Merge file using external tool"), \
 770         REQ_(STAGE_NEXT,        "Find next chunk to stage"), \
 771         \
 772         REQ_GROUP("Cursor navigation") \
 773         REQ_(MOVE_UP,           "Move cursor one line up"), \
 774         REQ_(MOVE_DOWN,         "Move cursor one line down"), \
 775         REQ_(MOVE_PAGE_DOWN,    "Move cursor one page down"), \
 776         REQ_(MOVE_PAGE_UP,      "Move cursor one page up"), \
 777         REQ_(MOVE_FIRST_LINE,   "Move cursor to first line"), \
 778         REQ_(MOVE_LAST_LINE,    "Move cursor to last line"), \
 779         \
 780         REQ_GROUP("Scrolling") \
 781         REQ_(SCROLL_LEFT,       "Scroll two columns left"), \
 782         REQ_(SCROLL_RIGHT,      "Scroll two columns right"), \
 783         REQ_(SCROLL_LINE_UP,    "Scroll one line up"), \
 784         REQ_(SCROLL_LINE_DOWN,  "Scroll one line down"), \
 785         REQ_(SCROLL_PAGE_UP,    "Scroll one page up"), \
 786         REQ_(SCROLL_PAGE_DOWN,  "Scroll one page down"), \
 787         \
 788         REQ_GROUP("Searching") \
 789         REQ_(SEARCH,            "Search the view"), \
 790         REQ_(SEARCH_BACK,       "Search backwards in the view"), \
 791         REQ_(FIND_NEXT,         "Find next search match"), \
 792         REQ_(FIND_PREV,         "Find previous search match"), \
 793         \
 794         REQ_GROUP("Option manipulation") \
 795         REQ_(TOGGLE_LINENO,     "Toggle line numbers"), \
 796         REQ_(TOGGLE_DATE,       "Toggle date display"), \
 797         REQ_(TOGGLE_AUTHOR,     "Toggle author display"), \
 798         REQ_(TOGGLE_REV_GRAPH,  "Toggle revision graph visualization"), \
 799         REQ_(TOGGLE_REFS,       "Toggle reference display (tags/branches)"), \
 800         \
 801         REQ_GROUP("Misc") \
 802         REQ_(PROMPT,            "Bring up the prompt"), \
 803         REQ_(SCREEN_REDRAW,     "Redraw the screen"), \
 804         REQ_(SHOW_VERSION,      "Show version information"), \
 805         REQ_(STOP_LOADING,      "Stop all loading views"), \
 806         REQ_(EDIT,              "Open in editor"), \
 807         REQ_(NONE,              "Do nothing")
 808
 809
 810 /* User action requests. */
 811 enum request {
 812 #define REQ_GROUP(help)
 813 #define REQ_(req, help) REQ_##req
 814
 815         /* Offset all requests to avoid conflicts with ncurses getch values. */
 816         REQ_OFFSET = KEY_MAX + 1,
 817         REQ_INFO
 818
 819 #undef  REQ_GROUP
 820 #undef  REQ_
 821 };
 822
 823 struct request_info {
 824         enum request request;
 825         const char *name;
 826         int namelen;
 827         const char *help;
 828 };
 829
 830 static const struct request_info req_info[] = {
 831 #define REQ_GROUP(help) { 0, NULL, 0, (help) },
 832 #define REQ_(req, help) { REQ_##req, (#req), STRING_SIZE(#req), (help) }
 833         REQ_INFO
 834 #undef  REQ_GROUP
 835 #undef  REQ_
 836 };
 837
 838 static enum request
 839 get_request(const char *name)
 840 {
 841         int namelen = strlen(name);
 842         int i;
 843
 844         for (i = 0; i < ARRAY_SIZE(req_info); i++)
 845                 if (req_info[i].namelen == namelen &&
 846                     !string_enum_compare(req_info[i].name, name, namelen))
 847                         return req_info[i].request;
 848
 849         return REQ_NONE;
 850 }
 851
 852
 853 /*
 854  * Options
 855  */
 856
 857 /* Option and state variables. */
 858 static bool opt_date                    = TRUE;
 859 static bool opt_author                  = TRUE;
 860 static bool opt_line_number             = FALSE;
 861 static bool opt_line_graphics           = TRUE;
 862 static bool opt_rev_graph               = FALSE;
 863 static bool opt_show_refs               = TRUE;
 864 static int opt_num_interval             = NUMBER_INTERVAL;
 865 static double opt_hscroll               = 0.50;
 866 static int opt_tab_size                 = TAB_SIZE;
 867 static int opt_author_cols              = AUTHOR_COLS-1;
 868 static char opt_path[SIZEOF_STR]        = "";
 869 static char opt_file[SIZEOF_STR]        = "";
 870 static char opt_ref[SIZEOF_REF]         = "";
 871 static char opt_head[SIZEOF_REF]        = "";
 872 static char opt_head_rev[SIZEOF_REV]    = "";
 873 static char opt_remote[SIZEOF_REF]      = "";
 874 static char opt_encoding[20]            = "UTF-8";
 875 static bool opt_utf8                    = TRUE;
 876 static char opt_codeset[20]             = "UTF-8";
 877 static iconv_t opt_iconv                = ICONV_NONE;
 878 static char opt_search[SIZEOF_STR]      = "";
 879 static char opt_cdup[SIZEOF_STR]        = "";
 880 static char opt_prefix[SIZEOF_STR]      = "";
 881 static char opt_git_dir[SIZEOF_STR]     = "";
 882 static signed char opt_is_inside_work_tree      = -1; /* set to TRUE or FALSE */
 883 static char opt_editor[SIZEOF_STR]      = "";
 884 static FILE *opt_tty                    = NULL;
 885
 886 #define is_initial_commit()     (!*opt_head_rev)
 887 #define is_head_commit(rev)     (!strcmp((rev), "HEAD") || !strcmp(opt_head_rev, (rev)))
 888
 889
 890 /*
 891  * Line-oriented content detection.
 892  */
 893
 894 #define LINE_INFO \
 895 LINE(DIFF_HEADER,  "diff --git ",       COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 896 LINE(DIFF_CHUNK,   "@@",                COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 897 LINE(DIFF_ADD,     "+",                 COLOR_GREEN,    COLOR_DEFAULT,  0), \
 898 LINE(DIFF_DEL,     "-",                 COLOR_RED,      COLOR_DEFAULT,  0), \
 899 LINE(DIFF_INDEX,        "index ",         COLOR_BLUE,   COLOR_DEFAULT,  0), \
 900 LINE(DIFF_OLDMODE,      "old file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
 901 LINE(DIFF_NEWMODE,      "new file mode ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
 902 LINE(DIFF_COPY_FROM,    "copy from",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
 903 LINE(DIFF_COPY_TO,      "copy to",        COLOR_YELLOW, COLOR_DEFAULT,  0), \
 904 LINE(DIFF_RENAME_FROM,  "rename from",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
 905 LINE(DIFF_RENAME_TO,    "rename to",      COLOR_YELLOW, COLOR_DEFAULT,  0), \
 906 LINE(DIFF_SIMILARITY,   "similarity ",    COLOR_YELLOW, COLOR_DEFAULT,  0), \
 907 LINE(DIFF_DISSIMILARITY,"dissimilarity ", COLOR_YELLOW, COLOR_DEFAULT,  0), \
 908 LINE(DIFF_TREE,         "diff-tree ",     COLOR_BLUE,   COLOR_DEFAULT,  0), \
 909 LINE(PP_AUTHOR,    "Author: ",          COLOR_CYAN,     COLOR_DEFAULT,  0), \
 910 LINE(PP_COMMIT,    "Commit: ",          COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 911 LINE(PP_MERGE,     "Merge: ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
 912 LINE(PP_DATE,      "Date:   ",          COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 913 LINE(PP_ADATE,     "AuthorDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 914 LINE(PP_CDATE,     "CommitDate: ",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 915 LINE(PP_REFS,      "Refs: ",            COLOR_RED,      COLOR_DEFAULT,  0), \
 916 LINE(COMMIT,       "commit ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
 917 LINE(PARENT,       "parent ",           COLOR_BLUE,     COLOR_DEFAULT,  0), \
 918 LINE(TREE,         "tree ",             COLOR_BLUE,     COLOR_DEFAULT,  0), \
 919 LINE(AUTHOR,       "author ",           COLOR_GREEN,    COLOR_DEFAULT,  0), \
 920 LINE(COMMITTER,    "committer ",        COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 921 LINE(SIGNOFF,      "    Signed-off-by", COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 922 LINE(ACKED,        "    Acked-by",      COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 923 LINE(DEFAULT,      "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
 924 LINE(CURSOR,       "",                  COLOR_WHITE,    COLOR_GREEN,    A_BOLD), \
 925 LINE(STATUS,       "",                  COLOR_GREEN,    COLOR_DEFAULT,  0), \
 926 LINE(DELIMITER,    "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 927 LINE(DATE,         "",                  COLOR_BLUE,     COLOR_DEFAULT,  0), \
 928 LINE(MODE,         "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
 929 LINE(LINE_NUMBER,  "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
 930 LINE(TITLE_BLUR,   "",                  COLOR_WHITE,    COLOR_BLUE,     0), \
 931 LINE(TITLE_FOCUS,  "",                  COLOR_WHITE,    COLOR_BLUE,     A_BOLD), \
 932 LINE(MAIN_COMMIT,  "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
 933 LINE(MAIN_TAG,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  A_BOLD), \
 934 LINE(MAIN_LOCAL_TAG,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 935 LINE(MAIN_REMOTE,  "",                  COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 936 LINE(MAIN_TRACKED, "",                  COLOR_YELLOW,   COLOR_DEFAULT,  A_BOLD), \
 937 LINE(MAIN_REF,     "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
 938 LINE(MAIN_HEAD,    "",                  COLOR_CYAN,     COLOR_DEFAULT,  A_BOLD), \
 939 LINE(MAIN_REVGRAPH,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 940 LINE(TREE_HEAD,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_BOLD), \
 941 LINE(TREE_DIR,     "",                  COLOR_YELLOW,   COLOR_DEFAULT,  A_NORMAL), \
 942 LINE(TREE_FILE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  A_NORMAL), \
 943 LINE(STAT_HEAD,    "",                  COLOR_YELLOW,   COLOR_DEFAULT,  0), \
 944 LINE(STAT_SECTION, "",                  COLOR_CYAN,     COLOR_DEFAULT,  0), \
 945 LINE(STAT_NONE,    "",                  COLOR_DEFAULT,  COLOR_DEFAULT,  0), \
 946 LINE(STAT_STAGED,  "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 947 LINE(STAT_UNSTAGED,"",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 948 LINE(STAT_UNTRACKED,"",                 COLOR_MAGENTA,  COLOR_DEFAULT,  0), \
 949 LINE(BLAME_ID,     "",                  COLOR_MAGENTA,  COLOR_DEFAULT,  0)
 950
 951 enum line_type {
 952 #define LINE(type, line, fg, bg, attr) \
 953         LINE_##type
 954         LINE_INFO,
 955         LINE_NONE
 956 #undef  LINE
 957 };
 958
 959 struct line_info {
 960         const char *name;       /* Option name. */
 961         int namelen;            /* Size of option name. */
 962         const char *line;       /* The start of line to match. */
 963         int linelen;            /* Size of string to match. */
 964         int fg, bg, attr;       /* Color and text attributes for the lines. */
 965 };
 966
 967 static struct line_info line_info[] = {
 968 #define LINE(type, line, fg, bg, attr) \
 969         { #type, STRING_SIZE(#type), (line), STRING_SIZE(line), (fg), (bg), (attr) }
 970         LINE_INFO
 971 #undef  LINE
 972 };
 973
 974 static enum line_type
 975 get_line_type(const char *line)
 976 {
 977         int linelen = strlen(line);
 978         enum line_type type;
 979
 980         for (type = 0; type < ARRAY_SIZE(line_info); type++)
 981                 /* Case insensitive search matches Signed-off-by lines better. */
 982                 if (linelen >= line_info[type].linelen &&
 983                     !strncasecmp(line_info[type].line, line, line_info[type].linelen))
 984                         return type;
 985
 986         return LINE_DEFAULT;
 987 }
 988
 989 static inline int
 990 get_line_attr(enum line_type type)
 991 {
 992         assert(type < ARRAY_SIZE(line_info));
 993         return COLOR_PAIR(type) | line_info[type].attr;
 994 }
 995
 996 static struct line_info *
 997 get_line_info(const char *name)
 998 {
 999         size_t namelen = strlen(name);
1000         enum line_type type;
1001
1002         for (type = 0; type < ARRAY_SIZE(line_info); type++)
1003                 if (namelen == line_info[type].namelen &&
1004                     !string_enum_compare(line_info[type].name, name, namelen))
1005                         return &line_info[type];
1006
1007         return NULL;
1008 }
1009
1010 static void
1011 init_colors(void)
1012 {
1013         int default_bg = line_info[LINE_DEFAULT].bg;
1014         int default_fg = line_info[LINE_DEFAULT].fg;
1015         enum line_type type;
1016
1017         start_color();
1018
1019         if (assume_default_colors(default_fg, default_bg) == ERR) {
1020                 default_bg = COLOR_BLACK;
1021                 default_fg = COLOR_WHITE;
1022         }
1023
1024         for (type = 0; type < ARRAY_SIZE(line_info); type++) {
1025                 struct line_info *info = &line_info[type];
1026                 int bg = info->bg == COLOR_DEFAULT ? default_bg : info->bg;
1027                 int fg = info->fg == COLOR_DEFAULT ? default_fg : info->fg;
1028
1029                 init_pair(type, fg, bg);
1030         }
1031 }
1032
1033 struct line {
1034         enum line_type type;
1035
1036         /* State flags */
1037         unsigned int selected:1;
1038         unsigned int dirty:1;
1039         unsigned int cleareol:1;
1040
1041         void *data;             /* User data */
1042 };
1043
1044
1045 /*
1046  * Keys
1047  */
1048
1049 struct keybinding {
1050         int alias;
1051         enum request request;
1052 };
1053
1054 static const struct keybinding default_keybindings[] = {
1055         /* View switching */
1056         { 'm',          REQ_VIEW_MAIN },
1057         { 'd',          REQ_VIEW_DIFF },
1058         { 'l',          REQ_VIEW_LOG },
1059         { 't',          REQ_VIEW_TREE },
1060         { 'f',          REQ_VIEW_BLOB },
1061         { 'B',          REQ_VIEW_BLAME },
1062         { 'p',          REQ_VIEW_PAGER },
1063         { 'h',          REQ_VIEW_HELP },
1064         { 'S',          REQ_VIEW_STATUS },
1065         { 'c',          REQ_VIEW_STAGE },
1066
1067         /* View manipulation */
1068         { 'q',          REQ_VIEW_CLOSE },
1069         { KEY_TAB,      REQ_VIEW_NEXT },
1070         { KEY_RETURN,   REQ_ENTER },
1071         { KEY_UP,       REQ_PREVIOUS },
1072         { KEY_DOWN,     REQ_NEXT },
1073         { 'R',          REQ_REFRESH },
1074         { KEY_F(5),     REQ_REFRESH },
1075         { 'O',          REQ_MAXIMIZE },
1076
1077         /* Cursor navigation */
1078         { 'k',          REQ_MOVE_UP },
1079         { 'j',          REQ_MOVE_DOWN },
1080         { KEY_HOME,     REQ_MOVE_FIRST_LINE },
1081         { KEY_END,      REQ_MOVE_LAST_LINE },
1082         { KEY_NPAGE,    REQ_MOVE_PAGE_DOWN },
1083         { ' ',          REQ_MOVE_PAGE_DOWN },
1084         { KEY_PPAGE,    REQ_MOVE_PAGE_UP },
1085         { 'b',          REQ_MOVE_PAGE_UP },
1086         { '-',          REQ_MOVE_PAGE_UP },
1087
1088         /* Scrolling */
1089         { KEY_LEFT,     REQ_SCROLL_LEFT },
1090         { KEY_RIGHT,    REQ_SCROLL_RIGHT },
1091         { KEY_IC,       REQ_SCROLL_LINE_UP },
1092         { KEY_DC,       REQ_SCROLL_LINE_DOWN },
1093         { 'w',          REQ_SCROLL_PAGE_UP },
1094         { 's',          REQ_SCROLL_PAGE_DOWN },
1095
1096         /* Searching */
1097         { '/',          REQ_SEARCH },
1098         { '?',          REQ_SEARCH_BACK },
1099         { 'n',          REQ_FIND_NEXT },
1100         { 'N',          REQ_FIND_PREV },
1101
1102         /* Misc */
1103         { 'Q',          REQ_QUIT },
1104         { 'z',          REQ_STOP_LOADING },
1105         { 'v',          REQ_SHOW_VERSION },
1106         { 'r',          REQ_SCREEN_REDRAW },
1107         { '.',          REQ_TOGGLE_LINENO },
1108         { 'D',          REQ_TOGGLE_DATE },
1109         { 'A',          REQ_TOGGLE_AUTHOR },
1110         { 'g',          REQ_TOGGLE_REV_GRAPH },
1111         { 'F',          REQ_TOGGLE_REFS },
1112         { ':',          REQ_PROMPT },
1113         { 'u',          REQ_STATUS_UPDATE },
1114         { '!',          REQ_STATUS_REVERT },
1115         { 'M',          REQ_STATUS_MERGE },
1116         { '@',          REQ_STAGE_NEXT },
1117         { ',',          REQ_PARENT },
1118         { 'e',          REQ_EDIT },
1119 };
1120
1121 #define KEYMAP_INFO \
1122         KEYMAP_(GENERIC), \
1123         KEYMAP_(MAIN), \
1124         KEYMAP_(DIFF), \
1125         KEYMAP_(LOG), \
1126         KEYMAP_(TREE), \
1127         KEYMAP_(BLOB), \
1128         KEYMAP_(BLAME), \
1129         KEYMAP_(PAGER), \
1130         KEYMAP_(HELP), \
1131         KEYMAP_(STATUS), \
1132         KEYMAP_(STAGE)
1133
1134 enum keymap {
1135 #define KEYMAP_(name) KEYMAP_##name
1136         KEYMAP_INFO
1137 #undef  KEYMAP_
1138 };
1139
1140 static const struct enum_map keymap_table[] = {
1141 #define KEYMAP_(name) ENUM_MAP(#name, KEYMAP_##name)
1142         KEYMAP_INFO
1143 #undef  KEYMAP_
1144 };
1145
1146 #define set_keymap(map, name) map_enum(map, keymap_table, name)
1147
1148 struct keybinding_table {
1149         struct keybinding *data;
1150         size_t size;
1151 };
1152
1153 static struct keybinding_table keybindings[ARRAY_SIZE(keymap_table)];
1154
1155 static void
1156 add_keybinding(enum keymap keymap, enum request request, int key)
1157 {
1158         struct keybinding_table *table = &keybindings[keymap];
1159
1160         table->data = realloc(table->data, (table->size + 1) * sizeof(*table->data));
1161         if (!table->data)
1162                 die("Failed to allocate keybinding");
1163         table->data[table->size].alias = key;
1164         table->data[table->size++].request = request;
1165 }
1166
1167 /* Looks for a key binding first in the given map, then in the generic map, and
1168  * lastly in the default keybindings. */
1169 static enum request
1170 get_keybinding(enum keymap keymap, int key)
1171 {
1172         size_t i;
1173
1174         for (i = 0; i < keybindings[keymap].size; i++)
1175                 if (keybindings[keymap].data[i].alias == key)
1176                         return keybindings[keymap].data[i].request;
1177
1178         for (i = 0; i < keybindings[KEYMAP_GENERIC].size; i++)
1179                 if (keybindings[KEYMAP_GENERIC].data[i].alias == key)
1180                         return keybindings[KEYMAP_GENERIC].data[i].request;
1181
1182         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++)
1183                 if (default_keybindings[i].alias == key)
1184                         return default_keybindings[i].request;
1185
1186         return (enum request) key;
1187 }
1188
1189
1190 struct key {
1191         const char *name;
1192         int value;
1193 };
1194
1195 static const struct key key_table[] = {
1196         { "Enter",      KEY_RETURN },
1197         { "Space",      ' ' },
1198         { "Backspace",  KEY_BACKSPACE },
1199         { "Tab",        KEY_TAB },
1200         { "Escape",     KEY_ESC },
1201         { "Left",       KEY_LEFT },
1202         { "Right",      KEY_RIGHT },
1203         { "Up",         KEY_UP },
1204         { "Down",       KEY_DOWN },
1205         { "Insert",     KEY_IC },
1206         { "Delete",     KEY_DC },
1207         { "Hash",       '#' },
1208         { "Home",       KEY_HOME },
1209         { "End",        KEY_END },
1210         { "PageUp",     KEY_PPAGE },
1211         { "PageDown",   KEY_NPAGE },
1212         { "F1",         KEY_F(1) },
1213         { "F2",         KEY_F(2) },
1214         { "F3",         KEY_F(3) },
1215         { "F4",         KEY_F(4) },
1216         { "F5",         KEY_F(5) },
1217         { "F6",         KEY_F(6) },
1218         { "F7",         KEY_F(7) },
1219         { "F8",         KEY_F(8) },
1220         { "F9",         KEY_F(9) },
1221         { "F10",        KEY_F(10) },
1222         { "F11",        KEY_F(11) },
1223         { "F12",        KEY_F(12) },
1224 };
1225
1226 static int
1227 get_key_value(const char *name)
1228 {
1229         int i;
1230
1231         for (i = 0; i < ARRAY_SIZE(key_table); i++)
1232                 if (!strcasecmp(key_table[i].name, name))
1233                         return key_table[i].value;
1234
1235         if (strlen(name) == 1 && isprint(*name))
1236                 return (int) *name;
1237
1238         return ERR;
1239 }
1240
1241 static const char *
1242 get_key_name(int key_value)
1243 {
1244         static char key_char[] = "'X'";
1245         const char *seq = NULL;
1246         int key;
1247
1248         for (key = 0; key < ARRAY_SIZE(key_table); key++)
1249                 if (key_table[key].value == key_value)
1250                         seq = key_table[key].name;
1251
1252         if (seq == NULL &&
1253             key_value < 127 &&
1254             isprint(key_value)) {
1255                 key_char[1] = (char) key_value;
1256                 seq = key_char;
1257         }
1258
1259         return seq ? seq : "(no key)";
1260 }
1261
1262 static const char *
1263 get_key(enum request request)
1264 {
1265         static char buf[BUFSIZ];
1266         size_t pos = 0;
1267         char *sep = "";
1268         int i;
1269
1270         buf[pos] = 0;
1271
1272         for (i = 0; i < ARRAY_SIZE(default_keybindings); i++) {
1273                 const struct keybinding *keybinding = &default_keybindings[i];
1274
1275                 if (keybinding->request != request)
1276                         continue;
1277
1278                 if (!string_format_from(buf, &pos, "%s%s", sep,
1279                                         get_key_name(keybinding->alias)))
1280                         return "Too many keybindings!";
1281                 sep = ", ";
1282         }
1283
1284         return buf;
1285 }
1286
1287 struct run_request {
1288         enum keymap keymap;
1289         int key;
1290         const char *argv[SIZEOF_ARG];
1291 };
1292
1293 static struct run_request *run_request;
1294 static size_t run_requests;
1295
1296 static enum request
1297 add_run_request(enum keymap keymap, int key, int argc, const char **argv)
1298 {
1299         struct run_request *req;
1300
1301         if (argc >= ARRAY_SIZE(req->argv) - 1)
1302                 return REQ_NONE;
1303
1304         req = realloc(run_request, (run_requests + 1) * sizeof(*run_request));
1305         if (!req)
1306                 return REQ_NONE;
1307
1308         run_request = req;
1309         req = &run_request[run_requests];
1310         req->keymap = keymap;
1311         req->key = key;
1312         req->argv[0] = NULL;
1313
1314         if (!format_argv(req->argv, argv, FORMAT_NONE))
1315                 return REQ_NONE;
1316
1317         return REQ_NONE + ++run_requests;
1318 }
1319
1320 static struct run_request *
1321 get_run_request(enum request request)
1322 {
1323         if (request <= REQ_NONE)
1324                 return NULL;
1325         return &run_request[request - REQ_NONE - 1];
1326 }
1327
1328 static void
1329 add_builtin_run_requests(void)
1330 {
1331         const char *cherry_pick[] = { "git", "cherry-pick", "%(commit)", NULL };
1332         const char *gc[] = { "git", "gc", NULL };
1333         struct {
1334                 enum keymap keymap;
1335                 int key;
1336                 int argc;
1337                 const char **argv;
1338         } reqs[] = {
1339                 { KEYMAP_MAIN,    'C', ARRAY_SIZE(cherry_pick) - 1, cherry_pick },
1340                 { KEYMAP_GENERIC, 'G', ARRAY_SIZE(gc) - 1, gc },
1341         };
1342         int i;
1343
1344         for (i = 0; i < ARRAY_SIZE(reqs); i++) {
1345                 enum request req;
1346
1347                 req = add_run_request(reqs[i].keymap, reqs[i].key, reqs[i].argc, reqs[i].argv);
1348                 if (req != REQ_NONE)
1349                         add_keybinding(reqs[i].keymap, req, reqs[i].key);
1350         }
1351 }
1352
1353 /*
1354  * User config file handling.
1355  */
1356
1357 static int   config_lineno;
1358 static bool  config_errors;
1359 static const char *config_msg;
1360
1361 static const struct enum_map color_map[] = {
1362 #define COLOR_MAP(name) ENUM_MAP(#name, COLOR_##name)
1363         COLOR_MAP(DEFAULT),
1364         COLOR_MAP(BLACK),
1365         COLOR_MAP(BLUE),
1366         COLOR_MAP(CYAN),
1367         COLOR_MAP(GREEN),
1368         COLOR_MAP(MAGENTA),
1369         COLOR_MAP(RED),
1370         COLOR_MAP(WHITE),
1371         COLOR_MAP(YELLOW),
1372 };
1373
1374 static const struct enum_map attr_map[] = {
1375 #define ATTR_MAP(name) ENUM_MAP(#name, A_##name)
1376         ATTR_MAP(NORMAL),
1377         ATTR_MAP(BLINK),
1378         ATTR_MAP(BOLD),
1379         ATTR_MAP(DIM),
1380         ATTR_MAP(REVERSE),
1381         ATTR_MAP(STANDOUT),
1382         ATTR_MAP(UNDERLINE),
1383 };
1384
1385 #define set_attribute(attr, name)       map_enum(attr, attr_map, name)
1386
1387 static int parse_step(double *opt, const char *arg)
1388 {
1389         *opt = atoi(arg);
1390         if (!strchr(arg, '%'))
1391                 return OK;
1392
1393         /* "Shift down" so 100% and 1 does not conflict. */
1394         *opt = (*opt - 1) / 100;
1395         if (*opt >= 1.0) {
1396                 *opt = 0.99;
1397                 config_msg = "Step value larger than 100%";
1398                 return ERR;
1399         }
1400         if (*opt < 0.0) {
1401                 *opt = 1;
1402                 config_msg = "Invalid step value";
1403                 return ERR;
1404         }
1405         return OK;
1406 }
1407
1408 static int
1409 parse_int(int *opt, const char *arg, int min, int max)
1410 {
1411         int value = atoi(arg);
1412
1413         if (min <= value && value <= max) {
1414                 *opt = value;
1415                 return OK;
1416         }
1417
1418         config_msg = "Integer value out of bound";
1419         return ERR;
1420 }
1421
1422 static bool
1423 set_color(int *color, const char *name)
1424 {
1425         if (map_enum(color, color_map, name))
1426                 return TRUE;
1427         if (!prefixcmp(name, "color"))
1428                 return parse_int(color, name + 5, 0, 255) == OK;
1429         return FALSE;
1430 }
1431
1432 /* Wants: object fgcolor bgcolor [attribute] */
1433 static int
1434 option_color_command(int argc, const char *argv[])
1435 {
1436         struct line_info *info;
1437
1438         if (argc != 3 && argc != 4) {
1439                 config_msg = "Wrong number of arguments given to color command";
1440                 return ERR;
1441         }
1442
1443         info = get_line_info(argv[0]);
1444         if (!info) {
1445                 static const struct enum_map obsolete[] = {
1446                         ENUM_MAP("main-delim",  LINE_DELIMITER),
1447                         ENUM_MAP("main-date",   LINE_DATE),
1448                         ENUM_MAP("main-author", LINE_AUTHOR),
1449                 };
1450                 int index;
1451
1452                 if (!map_enum(&index, obsolete, argv[0])) {
1453                         config_msg = "Unknown color name";
1454                         return ERR;
1455                 }
1456                 info = &line_info[index];
1457         }
1458
1459         if (!set_color(&info->fg, argv[1]) ||
1460             !set_color(&info->bg, argv[2])) {
1461                 config_msg = "Unknown color";
1462                 return ERR;
1463         }
1464
1465         if (argc == 4 && !set_attribute(&info->attr, argv[3])) {
1466                 config_msg = "Unknown attribute";
1467                 return ERR;
1468         }
1469
1470         return OK;
1471 }
1472
1473 static int parse_bool(bool *opt, const char *arg)
1474 {
1475         *opt = (!strcmp(arg, "1") || !strcmp(arg, "true") || !strcmp(arg, "yes"))
1476                 ? TRUE : FALSE;
1477         return OK;
1478 }
1479
1480 static int
1481 parse_string(char *opt, const char *arg, size_t optsize)
1482 {
1483         int arglen = strlen(arg);
1484
1485         switch (arg[0]) {
1486         case '\"':
1487         case '\'':
1488                 if (arglen == 1 || arg[arglen - 1] != arg[0]) {
1489                         config_msg = "Unmatched quotation";
1490                         return ERR;
1491                 }
1492                 arg += 1; arglen -= 2;
1493         default:
1494                 string_ncopy_do(opt, optsize, arg, arglen);
1495                 return OK;
1496         }
1497 }
1498
1499 /* Wants: name = value */
1500 static int
1501 option_set_command(int argc, const char *argv[])
1502 {
1503         if (argc != 3) {
1504                 config_msg = "Wrong number of arguments given to set command";
1505                 return ERR;
1506         }
1507
1508         if (strcmp(argv[1], "=")) {
1509                 config_msg = "No value assigned";
1510                 return ERR;
1511         }
1512
1513         if (!strcmp(argv[0], "show-author"))
1514                 return parse_bool(&opt_author, argv[2]);
1515
1516         if (!strcmp(argv[0], "show-date"))
1517                 return parse_bool(&opt_date, argv[2]);
1518
1519         if (!strcmp(argv[0], "show-rev-graph"))
1520                 return parse_bool(&opt_rev_graph, argv[2]);
1521
1522         if (!strcmp(argv[0], "show-refs"))
1523                 return parse_bool(&opt_show_refs, argv[2]);
1524
1525         if (!strcmp(argv[0], "show-line-numbers"))
1526                 return parse_bool(&opt_line_number, argv[2]);
1527
1528         if (!strcmp(argv[0], "line-graphics"))
1529                 return parse_bool(&opt_line_graphics, argv[2]);
1530
1531         if (!strcmp(argv[0], "line-number-interval"))
1532                 return parse_int(&opt_num_interval, argv[2], 1, 1024);
1533
1534         if (!strcmp(argv[0], "author-width"))
1535                 return parse_int(&opt_author_cols, argv[2], 0, 1024);
1536
1537         if (!strcmp(argv[0], "horizontal-scroll"))
1538                 return parse_step(&opt_hscroll, argv[2]);
1539
1540         if (!strcmp(argv[0], "tab-size"))
1541                 return parse_int(&opt_tab_size, argv[2], 1, 1024);
1542
1543         if (!strcmp(argv[0], "commit-encoding"))
1544                 return parse_string(opt_encoding, argv[2], sizeof(opt_encoding));
1545
1546         config_msg = "Unknown variable name";
1547         return ERR;
1548 }
1549
1550 /* Wants: mode request key */
1551 static int
1552 option_bind_command(int argc, const char *argv[])
1553 {
1554         enum request request;
1555         int keymap;
1556         int key;
1557
1558         if (argc < 3) {
1559                 config_msg = "Wrong number of arguments given to bind command";
1560                 return ERR;
1561         }
1562
1563         if (set_keymap(&keymap, argv[0]) == ERR) {
1564                 config_msg = "Unknown key map";
1565                 return ERR;
1566         }
1567
1568         key = get_key_value(argv[1]);
1569         if (key == ERR) {
1570                 config_msg = "Unknown key";
1571                 return ERR;
1572         }
1573
1574         request = get_request(argv[2]);
1575         if (request == REQ_NONE) {
1576                 static const struct enum_map obsolete[] = {
1577                         ENUM_MAP("cherry-pick",         REQ_NONE),
1578                         ENUM_MAP("screen-resize",       REQ_NONE),
1579                         ENUM_MAP("tree-parent",         REQ_PARENT),
1580                 };
1581                 int alias;
1582
1583                 if (map_enum(&alias, obsolete, argv[2])) {
1584                         if (alias != REQ_NONE)
1585                                 add_keybinding(keymap, alias, key);
1586                         config_msg = "Obsolete request name";
1587                         return ERR;
1588                 }
1589         }
1590         if (request == REQ_NONE && *argv[2]++ == '!')
1591                 request = add_run_request(keymap, key, argc - 2, argv + 2);
1592         if (request == REQ_NONE) {
1593                 config_msg = "Unknown request name";
1594                 return ERR;
1595         }
1596
1597         add_keybinding(keymap, request, key);
1598
1599         return OK;
1600 }
1601
1602 static int
1603 set_option(const char *opt, char *value)
1604 {
1605         const char *argv[SIZEOF_ARG];
1606         int argc = 0;
1607
1608         if (!argv_from_string(argv, &argc, value)) {
1609                 config_msg = "Too many option arguments";
1610                 return ERR;
1611         }
1612
1613         if (!strcmp(opt, "color"))
1614                 return option_color_command(argc, argv);
1615
1616         if (!strcmp(opt, "set"))
1617                 return option_set_command(argc, argv);
1618
1619         if (!strcmp(opt, "bind"))
1620                 return option_bind_command(argc, argv);
1621
1622         config_msg = "Unknown option command";
1623         return ERR;
1624 }
1625
1626 static int
1627 read_option(char *opt, size_t optlen, char *value, size_t valuelen)
1628 {
1629         int status = OK;
1630
1631         config_lineno++;
1632         config_msg = "Internal error";
1633
1634         /* Check for comment markers, since read_properties() will
1635          * only ensure opt and value are split at first " \t". */
1636         optlen = strcspn(opt, "#");
1637         if (optlen == 0)
1638                 return OK;
1639
1640         if (opt[optlen] != 0) {
1641                 config_msg = "No option value";
1642                 status = ERR;
1643
1644         }  else {
1645                 /* Look for comment endings in the value. */
1646                 size_t len = strcspn(value, "#");
1647
1648                 if (len < valuelen) {
1649                         valuelen = len;
1650                         value[valuelen] = 0;
1651                 }
1652
1653                 status = set_option(opt, value);
1654         }
1655
1656         if (status == ERR) {
1657                 warn("Error on line %d, near '%.*s': %s",
1658                      config_lineno, (int) optlen, opt, config_msg);
1659                 config_errors = TRUE;
1660         }
1661
1662         /* Always keep going if errors are encountered. */
1663         return OK;
1664 }
1665
1666 static void
1667 load_option_file(const char *path)
1668 {
1669         struct io io = {};
1670
1671         /* It's OK that the file doesn't exist. */
1672         if (!io_open(&io, path))
1673                 return;
1674
1675         config_lineno = 0;
1676         config_errors = FALSE;
1677
1678         if (io_load(&io, " \t", read_option) == ERR ||
1679             config_errors == TRUE)
1680                 warn("Errors while loading %s.", path);
1681 }
1682
1683 static int
1684 load_options(void)
1685 {
1686         const char *home = getenv("HOME");
1687         const char *tigrc_user = getenv("TIGRC_USER");
1688         const char *tigrc_system = getenv("TIGRC_SYSTEM");
1689         char buf[SIZEOF_STR];
1690
1691         add_builtin_run_requests();
1692
1693         if (!tigrc_system)
1694                 tigrc_system = SYSCONFDIR "/tigrc";
1695         load_option_file(tigrc_system);
1696
1697         if (!tigrc_user) {
1698                 if (!home || !string_format(buf, "%s/.tigrc", home))
1699                         return ERR;
1700                 tigrc_user = buf;
1701         }
1702         load_option_file(tigrc_user);
1703
1704         return OK;
1705 }
1706
1707
1708 /*
1709  * The viewer
1710  */
1711
1712 struct view;
1713 struct view_ops;
1714
1715 /* The display array of active views and the index of the current view. */
1716 static struct view *display[2];
1717 static unsigned int current_view;
1718
1719 #define foreach_displayed_view(view, i) \
1720         for (i = 0; i < ARRAY_SIZE(display) && (view = display[i]); i++)
1721
1722 #define displayed_views()       (display[1] != NULL ? 2 : 1)
1723
1724 /* Current head and commit ID */
1725 static char ref_blob[SIZEOF_REF]        = "";
1726 static char ref_commit[SIZEOF_REF]      = "HEAD";
1727 static char ref_head[SIZEOF_REF]        = "HEAD";
1728
1729 struct view {
1730         const char *name;       /* View name */
1731         const char *cmd_env;    /* Command line set via environment */
1732         const char *id;         /* Points to either of ref_{head,commit,blob} */
1733
1734         struct view_ops *ops;   /* View operations */
1735
1736         enum keymap keymap;     /* What keymap does this view have */
1737         bool git_dir;           /* Whether the view requires a git directory. */
1738
1739         char ref[SIZEOF_REF];   /* Hovered commit reference */
1740         char vid[SIZEOF_REF];   /* View ID. Set to id member when updating. */
1741
1742         int height, width;      /* The width and height of the main window */
1743         WINDOW *win;            /* The main window */
1744         WINDOW *title;          /* The title window living below the main window */
1745
1746         /* Navigation */
1747         unsigned long offset;   /* Offset of the window top */
1748         unsigned long yoffset;  /* Offset from the window side. */
1749         unsigned long lineno;   /* Current line number */
1750         unsigned long p_offset; /* Previous offset of the window top */
1751         unsigned long p_yoffset;/* Previous offset from the window side */
1752         unsigned long p_lineno; /* Previous current line number */
1753         bool p_restore;         /* Should the previous position be restored. */
1754
1755         /* Searching */
1756         char grep[SIZEOF_STR];  /* Search string */
1757         regex_t *regex;         /* Pre-compiled regexp */
1758
1759         /* If non-NULL, points to the view that opened this view. If this view
1760          * is closed tig will switch back to the parent view. */
1761         struct view *parent;
1762
1763         /* Buffering */
1764         size_t lines;           /* Total number of lines */
1765         struct line *line;      /* Line index */
1766         size_t line_alloc;      /* Total number of allocated lines */
1767         unsigned int digits;    /* Number of digits in the lines member. */
1768
1769         /* Drawing */
1770         struct line *curline;   /* Line currently being drawn. */
1771         enum line_type curtype; /* Attribute currently used for drawing. */
1772         unsigned long col;      /* Column when drawing. */
1773         bool has_scrolled;      /* View was scrolled. */
1774
1775         /* Loading */
1776         struct io io;
1777         struct io *pipe;
1778         time_t start_time;
1779         time_t update_secs;
1780 };
1781
1782 struct view_ops {
1783         /* What type of content being displayed. Used in the title bar. */
1784         const char *type;
1785         /* Default command arguments. */
1786         const char **argv;
1787         /* Open and reads in all view content. */
1788         bool (*open)(struct view *view);
1789         /* Read one line; updates view->line. */
1790         bool (*read)(struct view *view, char *data);
1791         /* Draw one line; @lineno must be < view->height. */
1792         bool (*draw)(struct view *view, struct line *line, unsigned int lineno);
1793         /* Depending on view handle a special requests. */
1794         enum request (*request)(struct view *view, enum request request, struct line *line);
1795         /* Search for regexp in a line. */
1796         bool (*grep)(struct view *view, struct line *line);
1797         /* Select line */
1798         void (*select)(struct view *view, struct line *line);
1799 };
1800
1801 static struct view_ops blame_ops;
1802 static struct view_ops blob_ops;
1803 static struct view_ops diff_ops;
1804 static struct view_ops help_ops;
1805 static struct view_ops log_ops;
1806 static struct view_ops main_ops;
1807 static struct view_ops pager_ops;
1808 static struct view_ops stage_ops;
1809 static struct view_ops status_ops;
1810 static struct view_ops tree_ops;
1811
1812 #define VIEW_STR(name, env, ref, ops, map, git) \
1813         { name, #env, ref, ops, map, git }
1814
1815 #define VIEW_(id, name, ops, git, ref) \
1816         VIEW_STR(name, TIG_##id##_CMD, ref, ops, KEYMAP_##id, git)
1817
1818
1819 static struct view views[] = {
1820         VIEW_(MAIN,   "main",   &main_ops,   TRUE,  ref_head),
1821         VIEW_(DIFF,   "diff",   &diff_ops,   TRUE,  ref_commit),
1822         VIEW_(LOG,    "log",    &log_ops,    TRUE,  ref_head),
1823         VIEW_(TREE,   "tree",   &tree_ops,   TRUE,  ref_commit),
1824         VIEW_(BLOB,   "blob",   &blob_ops,   TRUE,  ref_blob),
1825         VIEW_(BLAME,  "blame",  &blame_ops,  TRUE,  ref_commit),
1826         VIEW_(HELP,   "help",   &help_ops,   FALSE, ""),
1827         VIEW_(PAGER,  "pager",  &pager_ops,  FALSE, "stdin"),
1828         VIEW_(STATUS, "status", &status_ops, TRUE,  ""),
1829         VIEW_(STAGE,  "stage",  &stage_ops,  TRUE,  ""),
1830 };
1831
1832 #define VIEW(req)       (&views[(req) - REQ_OFFSET - 1])
1833 #define VIEW_REQ(view)  ((view) - views + REQ_OFFSET + 1)
1834
1835 #define foreach_view(view, i) \
1836         for (i = 0; i < ARRAY_SIZE(views) && (view = &views[i]); i++)
1837
1838 #define view_is_displayed(view) \
1839         (view == display[0] || view == display[1])
1840
1841
1842 enum line_graphic {
1843         LINE_GRAPHIC_VLINE
1844 };
1845
1846 static chtype line_graphics[] = {
1847         /* LINE_GRAPHIC_VLINE: */ '|'
1848 };
1849
1850 static inline void
1851 set_view_attr(struct view *view, enum line_type type)
1852 {
1853         if (!view->curline->selected && view->curtype != type) {
1854                 wattrset(view->win, get_line_attr(type));
1855                 wchgat(view->win, -1, 0, type, NULL);
1856                 view->curtype = type;
1857         }
1858 }
1859
1860 static int
1861 draw_chars(struct view *view, enum line_type type, const char *string,
1862            int max_len, bool use_tilde)
1863 {
1864         int len = 0;
1865         int col = 0;
1866         int trimmed = FALSE;
1867         size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
1868
1869         if (max_len <= 0)
1870                 return 0;
1871
1872         if (opt_utf8) {
1873                 len = utf8_length(&string, skip, &col, max_len, &trimmed, use_tilde);
1874         } else {
1875                 col = len = strlen(string);
1876                 if (len > max_len) {
1877                         if (use_tilde) {
1878                                 max_len -= 1;
1879                         }
1880                         col = len = max_len;
1881                         trimmed = TRUE;
1882                 }
1883         }
1884
1885         set_view_attr(view, type);
1886         if (len > 0)
1887                 waddnstr(view->win, string, len);
1888         if (trimmed && use_tilde) {
1889                 set_view_attr(view, LINE_DELIMITER);
1890                 waddch(view->win, '~');
1891                 col++;
1892         }
1893
1894         return col;
1895 }
1896
1897 static int
1898 draw_space(struct view *view, enum line_type type, int max, int spaces)
1899 {
1900         static char space[] = "                    ";
1901         int col = 0;
1902
1903         spaces = MIN(max, spaces);
1904
1905         while (spaces > 0) {
1906                 int len = MIN(spaces, sizeof(space) - 1);
1907
1908                 col += draw_chars(view, type, space, len, FALSE);
1909                 spaces -= len;
1910         }
1911
1912         return col;
1913 }
1914
1915 static bool
1916 draw_text(struct view *view, enum line_type type, const char *string, bool trim)
1917 {
1918         view->col += draw_chars(view, type, string, view->width + view->yoffset - view->col, trim);
1919         return view->width + view->yoffset <= view->col;
1920 }
1921
1922 static bool
1923 draw_graphic(struct view *view, enum line_type type, chtype graphic[], size_t size)
1924 {
1925         size_t skip = view->yoffset > view->col ? view->yoffset - view->col : 0;
1926         int max = view->width + view->yoffset - view->col;
1927         int i;
1928
1929         if (max < size)
1930                 size = max;
1931
1932         set_view_attr(view, type);
1933         /* Using waddch() instead of waddnstr() ensures that
1934          * they'll be rendered correctly for the cursor line. */
1935         for (i = skip; i < size; i++)
1936                 waddch(view->win, graphic[i]);
1937
1938         view->col += size;
1939         if (size < max && skip <= size)
1940                 waddch(view->win, ' ');
1941         view->col++;
1942
1943         return view->width + view->yoffset <= view->col;
1944 }
1945
1946 static bool
1947 draw_field(struct view *view, enum line_type type, const char *text, int len, bool trim)
1948 {
1949         int max = MIN(view->width + view->yoffset - view->col, len);
1950         int col;
1951
1952         if (text)
1953                 col = draw_chars(view, type, text, max - 1, trim);
1954         else
1955                 col = draw_space(view, type, max - 1, max - 1);
1956
1957         view->col += col;
1958         view->col += draw_space(view, LINE_DEFAULT, max - col, max - col);
1959         return view->width + view->yoffset <= view->col;
1960 }
1961
1962 static bool
1963 draw_date(struct view *view, struct tm *time)
1964 {
1965         char buf[DATE_COLS];
1966         char *date;
1967         int timelen = 0;
1968
1969         if (time)
1970                 timelen = strftime(buf, sizeof(buf), DATE_FORMAT, time);
1971         date = timelen ? buf : NULL;
1972
1973         return draw_field(view, LINE_DATE, date, DATE_COLS, FALSE);
1974 }
1975
1976 static bool
1977 draw_author(struct view *view, const char *author)
1978 {
1979         bool trim = opt_author_cols == 0 || opt_author_cols > 5 || !author;
1980
1981         if (!trim) {
1982                 static char initials[10];
1983                 size_t pos;
1984
1985 #define is_initial_sep(c) (isspace(c) || ispunct(c) || (c) == '@')
1986
1987                 memset(initials, 0, sizeof(initials));
1988                 for (pos = 0; *author && pos < opt_author_cols - 1; author++, pos++) {
1989                         while (is_initial_sep(*author))
1990                                 author++;
1991                         strncpy(&initials[pos], author, sizeof(initials) - 1 - pos);
1992                         while (*author && !is_initial_sep(author[1]))
1993                                 author++;
1994                 }
1995
1996                 author = initials;
1997         }
1998
1999         return draw_field(view, LINE_AUTHOR, author, opt_author_cols, trim);
2000 }
2001
2002 static bool
2003 draw_mode(struct view *view, mode_t mode)
2004 {
2005         const char *str;
2006
2007         if (S_ISDIR(mode))
2008                 str = "drwxr-xr-x";
2009         else if (S_ISLNK(mode))
2010                 str = "lrwxrwxrwx";
2011         else if (S_ISGITLINK(mode))
2012                 str = "m---------";
2013         else if (S_ISREG(mode) && mode & S_IXUSR)
2014                 str = "-rwxr-xr-x";
2015         else if (S_ISREG(mode))
2016                 str = "-rw-r--r--";
2017         else
2018                 str = "----------";
2019
2020         return draw_field(view, LINE_MODE, str, STRING_SIZE("-rw-r--r-- "), FALSE);
2021 }
2022
2023 static bool
2024 draw_lineno(struct view *view, unsigned int lineno)
2025 {
2026         char number[10];
2027         int digits3 = view->digits < 3 ? 3 : view->digits;
2028         int max = MIN(view->width + view->yoffset - view->col, digits3);
2029         char *text = NULL;
2030
2031         lineno += view->offset + 1;
2032         if (lineno == 1 || (lineno % opt_num_interval) == 0) {
2033                 static char fmt[] = "%1ld";
2034
2035                 fmt[1] = '0' + (view->digits <= 9 ? digits3 : 1);
2036                 if (string_format(number, fmt, lineno))
2037                         text = number;
2038         }
2039         if (text)
2040                 view->col += draw_chars(view, LINE_LINE_NUMBER, text, max, TRUE);
2041         else
2042                 view->col += draw_space(view, LINE_LINE_NUMBER, max, digits3);
2043         return draw_graphic(view, LINE_DEFAULT, &line_graphics[LINE_GRAPHIC_VLINE], 1);
2044 }
2045
2046 static bool
2047 draw_view_line(struct view *view, unsigned int lineno)
2048 {
2049         struct line *line;
2050         bool selected = (view->offset + lineno == view->lineno);
2051
2052         assert(view_is_displayed(view));
2053
2054         if (view->offset + lineno >= view->lines)
2055                 return FALSE;
2056
2057         line = &view->line[view->offset + lineno];
2058
2059         wmove(view->win, lineno, 0);
2060         if (line->cleareol)
2061                 wclrtoeol(view->win);
2062         view->col = 0;
2063         view->curline = line;
2064         view->curtype = LINE_NONE;
2065         line->selected = FALSE;
2066         line->dirty = line->cleareol = 0;
2067
2068         if (selected) {
2069                 set_view_attr(view, LINE_CURSOR);
2070                 line->selected = TRUE;
2071                 view->ops->select(view, line);
2072         }
2073
2074         return view->ops->draw(view, line, lineno);
2075 }
2076
2077 static void
2078 redraw_view_dirty(struct view *view)
2079 {
2080         bool dirty = FALSE;
2081         int lineno;
2082
2083         for (lineno = 0; lineno < view->height; lineno++) {
2084                 if (view->offset + lineno >= view->lines)
2085                         break;
2086                 if (!view->line[view->offset + lineno].dirty)
2087                         continue;
2088                 dirty = TRUE;
2089                 if (!draw_view_line(view, lineno))
2090                         break;
2091         }
2092
2093         if (!dirty)
2094                 return;
2095         wnoutrefresh(view->win);
2096 }
2097
2098 static void
2099 redraw_view_from(struct view *view, int lineno)
2100 {
2101         assert(0 <= lineno && lineno < view->height);
2102
2103         for (; lineno < view->height; lineno++) {
2104                 if (!draw_view_line(view, lineno))
2105                         break;
2106         }
2107
2108         wnoutrefresh(view->win);
2109 }
2110
2111 static void
2112 redraw_view(struct view *view)
2113 {
2114         werase(view->win);
2115         redraw_view_from(view, 0);
2116 }
2117
2118
2119 static void
2120 update_view_title(struct view *view)
2121 {
2122         char buf[SIZEOF_STR];
2123         char state[SIZEOF_STR];
2124         size_t bufpos = 0, statelen = 0;
2125
2126         assert(view_is_displayed(view));
2127
2128         if (view != VIEW(REQ_VIEW_STATUS) && view->lines) {
2129                 unsigned int view_lines = view->offset + view->height;
2130                 unsigned int lines = view->lines
2131                                    ? MIN(view_lines, view->lines) * 100 / view->lines
2132                                    : 0;
2133
2134                 string_format_from(state, &statelen, " - %s %d of %d (%d%%)",
2135                                    view->ops->type,
2136                                    view->lineno + 1,
2137                                    view->lines,
2138                                    lines);
2139
2140         }
2141
2142         if (view->pipe) {
2143                 time_t secs = time(NULL) - view->start_time;
2144
2145                 /* Three git seconds are a long time ... */
2146                 if (secs > 2)
2147                         string_format_from(state, &statelen, " loading %lds", secs);
2148         }
2149
2150         string_format_from(buf, &bufpos, "[%s]", view->name);
2151         if (*view->ref && bufpos < view->width) {
2152                 size_t refsize = strlen(view->ref);
2153                 size_t minsize = bufpos + 1 + /* abbrev= */ 7 + 1 + statelen;
2154
2155                 if (minsize < view->width)
2156                         refsize = view->width - minsize + 7;
2157                 string_format_from(buf, &bufpos, " %.*s", (int) refsize, view->ref);
2158         }
2159
2160         if (statelen && bufpos < view->width) {
2161                 string_format_from(buf, &bufpos, "%s", state);
2162         }
2163
2164         if (view == display[current_view])
2165                 wbkgdset(view->title, get_line_attr(LINE_TITLE_FOCUS));
2166         else
2167                 wbkgdset(view->title, get_line_attr(LINE_TITLE_BLUR));
2168
2169         mvwaddnstr(view->title, 0, 0, buf, bufpos);
2170         wclrtoeol(view->title);
2171         wnoutrefresh(view->title);
2172 }
2173
2174 static void
2175 resize_display(void)
2176 {
2177         int offset, i;
2178         struct view *base = display[0];
2179         struct view *view = display[1] ? display[1] : display[0];
2180
2181         /* Setup window dimensions */
2182
2183         getmaxyx(stdscr, base->height, base->width);
2184
2185         /* Make room for the status window. */
2186         base->height -= 1;
2187
2188         if (view != base) {
2189                 /* Horizontal split. */
2190                 view->width   = base->width;
2191                 view->height  = SCALE_SPLIT_VIEW(base->height);
2192                 base->height -= view->height;
2193
2194                 /* Make room for the title bar. */
2195                 view->height -= 1;
2196         }
2197
2198         /* Make room for the title bar. */
2199         base->height -= 1;
2200
2201         offset = 0;
2202
2203         foreach_displayed_view (view, i) {
2204                 if (!view->win) {
2205                         view->win = newwin(view->height, 0, offset, 0);
2206                         if (!view->win)
2207                                 die("Failed to create %s view", view->name);
2208
2209                         scrollok(view->win, FALSE);
2210
2211                         view->title = newwin(1, 0, offset + view->height, 0);
2212                         if (!view->title)
2213                                 die("Failed to create title window");
2214
2215                 } else {
2216                         wresize(view->win, view->height, view->width);
2217                         mvwin(view->win,   offset, 0);
2218                         mvwin(view->title, offset + view->height, 0);
2219                 }
2220
2221                 offset += view->height + 1;
2222         }
2223 }
2224
2225 static void
2226 redraw_display(bool clear)
2227 {
2228         struct view *view;
2229         int i;
2230
2231         foreach_displayed_view (view, i) {
2232                 if (clear)
2233                         wclear(view->win);
2234                 redraw_view(view);
2235                 update_view_title(view);
2236         }
2237 }
2238
2239 static void
2240 toggle_view_option(bool *option, const char *help)
2241 {
2242         *option = !*option;
2243         redraw_display(FALSE);
2244         report("%sabling %s", *option ? "En" : "Dis", help);
2245 }
2246
2247 static void
2248 maximize_view(struct view *view)
2249 {
2250         memset(display, 0, sizeof(display));
2251         current_view = 0;
2252         display[current_view] = view;
2253         resize_display();
2254         redraw_display(FALSE);
2255         report("");
2256 }
2257
2258
2259 /*
2260  * Navigation
2261  */
2262
2263 static bool
2264 goto_view_line(struct view *view, unsigned long offset, unsigned long lineno)
2265 {
2266         if (lineno >= view->lines)
2267                 lineno = view->lines > 0 ? view->lines - 1 : 0;
2268
2269         if (offset > lineno || offset + view->height <= lineno) {
2270                 unsigned long half = view->height / 2;
2271
2272                 if (lineno > half)
2273                         offset = lineno - half;
2274                 else
2275                         offset = 0;
2276         }
2277
2278         if (offset != view->offset || lineno != view->lineno) {
2279                 view->offset = offset;
2280                 view->lineno = lineno;
2281                 return TRUE;
2282         }
2283
2284         return FALSE;
2285 }
2286
2287 static int
2288 apply_step(double step, int value)
2289 {
2290         if (step >= 1)
2291                 return (int) step;
2292         value *= step + 0.01;
2293         return value ? value : 1;
2294 }
2295
2296 /* Scrolling backend */
2297 static void
2298 do_scroll_view(struct view *view, int lines)
2299 {
2300         bool redraw_current_line = FALSE;
2301
2302         /* The rendering expects the new offset. */
2303         view->offset += lines;
2304
2305         assert(0 <= view->offset && view->offset < view->lines);
2306         assert(lines);
2307
2308         /* Move current line into the view. */
2309         if (view->lineno < view->offset) {
2310                 view->lineno = view->offset;
2311                 redraw_current_line = TRUE;
2312         } else if (view->lineno >= view->offset + view->height) {
2313                 view->lineno = view->offset + view->height - 1;
2314                 redraw_current_line = TRUE;
2315         }
2316
2317         assert(view->offset <= view->lineno && view->lineno < view->lines);
2318
2319         /* Redraw the whole screen if scrolling is pointless. */
2320         if (view->height < ABS(lines)) {
2321                 redraw_view(view);
2322
2323         } else {
2324                 int line = lines > 0 ? view->height - lines : 0;
2325                 int end = line + ABS(lines);
2326
2327                 scrollok(view->win, TRUE);
2328                 wscrl(view->win, lines);
2329                 scrollok(view->win, FALSE);
2330
2331                 while (line < end && draw_view_line(view, line))
2332                         line++;
2333
2334                 if (redraw_current_line)
2335                         draw_view_line(view, view->lineno - view->offset);
2336                 wnoutrefresh(view->win);
2337         }
2338
2339         view->has_scrolled = TRUE;
2340         report("");
2341 }
2342
2343 /* Scroll frontend */
2344 static void
2345 scroll_view(struct view *view, enum request request)
2346 {
2347         int lines = 1;
2348
2349         assert(view_is_displayed(view));
2350
2351         switch (request) {
2352         case REQ_SCROLL_LEFT:
2353                 if (view->yoffset == 0) {
2354                         report("Cannot scroll beyond the first column");
2355                         return;
2356                 }
2357                 if (view->yoffset <= apply_step(opt_hscroll, view->width))
2358                         view->yoffset = 0;
2359                 else
2360                         view->yoffset -= apply_step(opt_hscroll, view->width);
2361                 redraw_view_from(view, 0);
2362                 report("");
2363                 return;
2364         case REQ_SCROLL_RIGHT:
2365                 view->yoffset += apply_step(opt_hscroll, view->width);
2366                 redraw_view(view);
2367                 report("");
2368                 return;
2369         case REQ_SCROLL_PAGE_DOWN:
2370                 lines = view->height;
2371         case REQ_SCROLL_LINE_DOWN:
2372                 if (view->offset + lines > view->lines)
2373                         lines = view->lines - view->offset;
2374
2375                 if (lines == 0 || view->offset + view->height >= view->lines) {
2376                         report("Cannot scroll beyond the last line");
2377                         return;
2378                 }
2379                 break;
2380
2381         case REQ_SCROLL_PAGE_UP:
2382                 lines = view->height;
2383         case REQ_SCROLL_LINE_UP:
2384                 if (lines > view->offset)
2385                         lines = view->offset;
2386
2387                 if (lines == 0) {
2388                         report("Cannot scroll beyond the first line");
2389                         return;
2390                 }
2391
2392                 lines = -lines;
2393                 break;
2394
2395         default:
2396                 die("request %d not handled in switch", request);
2397         }
2398
2399         do_scroll_view(view, lines);
2400 }
2401
2402 /* Cursor moving */
2403 static void
2404 move_view(struct view *view, enum request request)
2405 {
2406         int scroll_steps = 0;
2407         int steps;
2408
2409         switch (request) {
2410         case REQ_MOVE_FIRST_LINE:
2411                 steps = -view->lineno;
2412                 break;
2413
2414         case REQ_MOVE_LAST_LINE:
2415                 steps = view->lines - view->lineno - 1;
2416                 break;
2417
2418         case REQ_MOVE_PAGE_UP:
2419                 steps = view->height > view->lineno
2420                       ? -view->lineno : -view->height;
2421                 break;
2422
2423         case REQ_MOVE_PAGE_DOWN:
2424                 steps = view->lineno + view->height >= view->lines
2425                       ? view->lines - view->lineno - 1 : view->height;
2426                 break;
2427
2428         case REQ_MOVE_UP:
2429                 steps = -1;
2430                 break;
2431
2432         case REQ_MOVE_DOWN:
2433                 steps = 1;
2434                 break;
2435
2436         default:
2437                 die("request %d not handled in switch", request);
2438         }
2439
2440         if (steps <= 0 && view->lineno == 0) {
2441                 report("Cannot move beyond the first line");
2442                 return;
2443
2444         } else if (steps >= 0 && view->lineno + 1 >= view->lines) {
2445                 report("Cannot move beyond the last line");
2446                 return;
2447         }
2448
2449         /* Move the current line */
2450         view->lineno += steps;
2451         assert(0 <= view->lineno && view->lineno < view->lines);
2452
2453         /* Check whether the view needs to be scrolled */
2454         if (view->lineno < view->offset ||
2455             view->lineno >= view->offset + view->height) {
2456                 scroll_steps = steps;
2457                 if (steps < 0 && -steps > view->offset) {
2458                         scroll_steps = -view->offset;
2459
2460                 } else if (steps > 0) {
2461                         if (view->lineno == view->lines - 1 &&
2462                             view->lines > view->height) {
2463                                 scroll_steps = view->lines - view->offset - 1;
2464                                 if (scroll_steps >= view->height)
2465                                         scroll_steps -= view->height - 1;
2466                         }
2467                 }
2468         }
2469
2470         if (!view_is_displayed(view)) {
2471                 view->offset += scroll_steps;
2472                 assert(0 <= view->offset && view->offset < view->lines);
2473                 view->ops->select(view, &view->line[view->lineno]);
2474                 return;
2475         }
2476
2477         /* Repaint the old "current" line if we be scrolling */
2478         if (ABS(steps) < view->height)
2479                 draw_view_line(view, view->lineno - steps - view->offset);
2480
2481         if (scroll_steps) {
2482                 do_scroll_view(view, scroll_steps);
2483                 return;
2484         }
2485
2486         /* Draw the current line */
2487         draw_view_line(view, view->lineno - view->offset);
2488
2489         wnoutrefresh(view->win);
2490         report("");
2491 }
2492
2493
2494 /*
2495  * Searching
2496  */
2497
2498 static void search_view(struct view *view, enum request request);
2499
2500 static void
2501 select_view_line(struct view *view, unsigned long lineno)
2502 {
2503         unsigned long old_lineno = view->lineno;
2504         unsigned long old_offset = view->offset;
2505
2506         if (goto_view_line(view, view->offset, lineno)) {
2507                 if (view_is_displayed(view)) {
2508                         if (old_offset != view->offset) {
2509                                 redraw_view(view);
2510                         } else {
2511                                 draw_view_line(view, old_lineno - view->offset);
2512                                 draw_view_line(view, view->lineno - view->offset);
2513                                 wnoutrefresh(view->win);
2514                         }
2515                 } else {
2516                         view->ops->select(view, &view->line[view->lineno]);
2517                 }
2518         }
2519 }
2520
2521 static void
2522 find_next(struct view *view, enum request request)
2523 {
2524         unsigned long lineno = view->lineno;
2525         int direction;
2526
2527         if (!*view->grep) {
2528                 if (!*opt_search)
2529                         report("No previous search");
2530                 else
2531                         search_view(view, request);
2532                 return;
2533         }
2534
2535         switch (request) {
2536         case REQ_SEARCH:
2537         case REQ_FIND_NEXT:
2538                 direction = 1;
2539                 break;
2540
2541         case REQ_SEARCH_BACK:
2542         case REQ_FIND_PREV:
2543                 direction = -1;
2544                 break;
2545
2546         default:
2547                 return;
2548         }
2549
2550         if (request == REQ_FIND_NEXT || request == REQ_FIND_PREV)
2551                 lineno += direction;
2552
2553         /* Note, lineno is unsigned long so will wrap around in which case it
2554          * will become bigger than view->lines. */
2555         for (; lineno < view->lines; lineno += direction) {
2556                 if (view->ops->grep(view, &view->line[lineno])) {
2557                         select_view_line(view, lineno);
2558                         report("Line %ld matches '%s'", lineno + 1, view->grep);
2559                         return;
2560                 }
2561         }
2562
2563         report("No match found for '%s'", view->grep);
2564 }
2565
2566 static void
2567 search_view(struct view *view, enum request request)
2568 {
2569         int regex_err;
2570
2571         if (view->regex) {
2572                 regfree(view->regex);
2573                 *view->grep = 0;
2574         } else {
2575                 view->regex = calloc(1, sizeof(*view->regex));
2576                 if (!view->regex)
2577                         return;
2578         }
2579
2580         regex_err = regcomp(view->regex, opt_search, REG_EXTENDED);
2581         if (regex_err != 0) {
2582                 char buf[SIZEOF_STR] = "unknown error";
2583
2584                 regerror(regex_err, view->regex, buf, sizeof(buf));
2585                 report("Search failed: %s", buf);
2586                 return;
2587         }
2588
2589         string_copy(view->grep, opt_search);
2590
2591         find_next(view, request);
2592 }
2593
2594 /*
2595  * Incremental updating
2596  */
2597
2598 static void
2599 reset_view(struct view *view)
2600 {
2601         int i;
2602
2603         for (i = 0; i < view->lines; i++)
2604                 free(view->line[i].data);
2605         free(view->line);
2606
2607         view->p_offset = view->offset;
2608         view->p_yoffset = view->yoffset;
2609         view->p_lineno = view->lineno;
2610
2611         view->line = NULL;
2612         view->offset = 0;
2613         view->yoffset = 0;
2614         view->lines  = 0;
2615         view->lineno = 0;
2616         view->line_alloc = 0;
2617         view->vid[0] = 0;
2618         view->update_secs = 0;
2619 }
2620
2621 static void
2622 free_argv(const char *argv[])
2623 {
2624         int argc;
2625
2626         for (argc = 0; argv[argc]; argc++)
2627                 free((void *) argv[argc]);
2628 }
2629
2630 static bool
2631 format_argv(const char *dst_argv[], const char *src_argv[], enum format_flags flags)
2632 {
2633         char buf[SIZEOF_STR];
2634         int argc;
2635         bool noreplace = flags == FORMAT_NONE;
2636
2637         free_argv(dst_argv);
2638
2639         for (argc = 0; src_argv[argc]; argc++) {
2640                 const char *arg = src_argv[argc];
2641                 size_t bufpos = 0;
2642
2643                 while (arg) {
2644                         char *next = strstr(arg, "%(");
2645                         int len = next - arg;
2646                         const char *value;
2647
2648                         if (!next || noreplace) {
2649                                 if (flags == FORMAT_DASH && !strcmp(arg, "--"))
2650                                         noreplace = TRUE;
2651                                 len = strlen(arg);
2652                                 value = "";
2653
2654                         } else if (!prefixcmp(next, "%(directory)")) {
2655                                 value = opt_path;
2656
2657                         } else if (!prefixcmp(next, "%(file)")) {
2658                                 value = opt_file;
2659
2660                         } else if (!prefixcmp(next, "%(ref)")) {
2661                                 value = *opt_ref ? opt_ref : "HEAD";
2662
2663                         } else if (!prefixcmp(next, "%(head)")) {
2664                                 value = ref_head;
2665
2666                         } else if (!prefixcmp(next, "%(commit)")) {
2667                                 value = ref_commit;
2668
2669                         } else if (!prefixcmp(next, "%(blob)")) {
2670                                 value = ref_blob;
2671
2672                         } else {
2673                                 report("Unknown replacement: `%s`", next);
2674                                 return FALSE;
2675                         }
2676
2677                         if (!string_format_from(buf, &bufpos, "%.*s%s", len, arg, value))
2678                                 return FALSE;
2679
2680                         arg = next && !noreplace ? strchr(next, ')') + 1 : NULL;
2681                 }
2682
2683                 dst_argv[argc] = strdup(buf);
2684                 if (!dst_argv[argc])
2685                         break;
2686         }
2687
2688         dst_argv[argc] = NULL;
2689
2690         return src_argv[argc] == NULL;
2691 }
2692
2693 static bool
2694 restore_view_position(struct view *view)
2695 {
2696         if (!view->p_restore || (view->pipe && view->lines <= view->p_lineno))
2697                 return FALSE;
2698
2699         /* Changing the view position cancels the restoring. */
2700         /* FIXME: Changing back to the first line is not detected. */
2701         if (view->offset != 0 || view->lineno != 0) {
2702                 view->p_restore = FALSE;
2703                 return FALSE;
2704         }
2705
2706         if (goto_view_line(view, view->p_offset, view->p_lineno) &&
2707             view_is_displayed(view))
2708                 werase(view->win);
2709
2710         view->yoffset = view->p_yoffset;
2711         view->p_restore = FALSE;
2712
2713         return TRUE;
2714 }
2715
2716 static void
2717 end_update(struct view *view, bool force)
2718 {
2719         if (!view->pipe)
2720                 return;
2721         while (!view->ops->read(view, NULL))
2722                 if (!force)
2723                         return;
2724         set_nonblocking_input(FALSE);
2725         if (force)
2726                 kill_io(view->pipe);
2727         done_io(view->pipe);
2728         view->pipe = NULL;
2729 }
2730
2731 static void
2732 setup_update(struct view *view, const char *vid)
2733 {
2734         set_nonblocking_input(TRUE);
2735         reset_view(view);
2736         string_copy_rev(view->vid, vid);
2737         view->pipe = &view->io;
2738         view->start_time = time(NULL);
2739 }
2740
2741 static bool
2742 prepare_update(struct view *view, const char *argv[], const char *dir,
2743                enum format_flags flags)
2744 {
2745         if (view->pipe)
2746                 end_update(view, TRUE);
2747         return init_io_rd(&view->io, argv, dir, flags);
2748 }
2749
2750 static bool
2751 prepare_update_file(struct view *view, const char *name)
2752 {
2753         if (view->pipe)
2754                 end_update(view, TRUE);
2755         return io_open(&view->io, name);
2756 }
2757
2758 static bool
2759 begin_update(struct view *view, bool refresh)
2760 {
2761         if (view->pipe)
2762                 end_update(view, TRUE);
2763
2764         if (refresh) {
2765                 if (!start_io(&view->io))
2766                         return FALSE;
2767
2768         } else {
2769                 if (view == VIEW(REQ_VIEW_TREE) && strcmp(view->vid, view->id))
2770                         opt_path[0] = 0;
2771
2772                 if (!run_io_rd(&view->io, view->ops->argv, FORMAT_ALL))
2773                         return FALSE;
2774
2775                 /* Put the current ref_* value to the view title ref
2776                  * member. This is needed by the blob view. Most other
2777                  * views sets it automatically after loading because the
2778                  * first line is a commit line. */
2779                 string_copy_rev(view->ref, view->id);
2780         }
2781
2782         setup_update(view, view->id);
2783
2784         return TRUE;
2785 }
2786
2787 #define ITEM_CHUNK_SIZE 256
2788 static void *
2789 realloc_items(void *mem, size_t *size, size_t new_size, size_t item_size)
2790 {
2791         size_t num_chunks = *size / ITEM_CHUNK_SIZE;
2792         size_t num_chunks_new = (new_size + ITEM_CHUNK_SIZE - 1) / ITEM_CHUNK_SIZE;
2793
2794         if (mem == NULL || num_chunks != num_chunks_new) {
2795                 *size = num_chunks_new * ITEM_CHUNK_SIZE;
2796                 mem = realloc(mem, *size * item_size);
2797         }
2798
2799         return mem;
2800 }
2801
2802 static struct line *
2803 realloc_lines(struct view *view, size_t line_size)
2804 {
2805         size_t alloc = view->line_alloc;
2806         struct line *tmp = realloc_items(view->line, &alloc, line_size,
2807                                          sizeof(*view->line));
2808
2809         if (!tmp)
2810                 return NULL;
2811
2812         view->line = tmp;
2813         view->line_alloc = alloc;
2814         return view->line;
2815 }
2816
2817 static bool
2818 update_view(struct view *view)
2819 {
2820         char out_buffer[BUFSIZ * 2];
2821         char *line;
2822         /* Clear the view and redraw everything since the tree sorting
2823          * might have rearranged things. */
2824         bool redraw = view->lines == 0;
2825         bool can_read = TRUE;
2826
2827         if (!view->pipe)
2828                 return TRUE;
2829
2830         if (!io_can_read(view->pipe)) {
2831                 if (view->lines == 0 && view_is_displayed(view)) {
2832                         time_t secs = time(NULL) - view->start_time;
2833
2834                         if (secs > 1 && secs > view->update_secs) {
2835                                 if (view->update_secs == 0)
2836                                         redraw_view(view);
2837                                 update_view_title(view);
2838                                 view->update_secs = secs;
2839                         }
2840                 }
2841                 return TRUE;
2842         }
2843
2844         for (; (line = io_get(view->pipe, '\n', can_read)); can_read = FALSE) {
2845                 if (opt_iconv != ICONV_NONE) {
2846                         ICONV_CONST char *inbuf = line;
2847                         size_t inlen = strlen(line) + 1;
2848
2849                         char *outbuf = out_buffer;
2850                         size_t outlen = sizeof(out_buffer);
2851
2852                         size_t ret;
2853
2854                         ret = iconv(opt_iconv, &inbuf, &inlen, &outbuf, &outlen);
2855                         if (ret != (size_t) -1)
2856                                 line = out_buffer;
2857                 }
2858
2859                 if (!view->ops->read(view, line)) {
2860                         report("Allocation failure");
2861                         end_update(view, TRUE);
2862                         return FALSE;
2863                 }
2864         }
2865
2866         {
2867                 unsigned long lines = view->lines;
2868                 int digits;
2869
2870                 for (digits = 0; lines; digits++)
2871                         lines /= 10;
2872
2873                 /* Keep the displayed view in sync with line number scaling. */
2874                 if (digits != view->digits) {
2875                         view->digits = digits;
2876                         if (opt_line_number || view == VIEW(REQ_VIEW_BLAME))
2877                                 redraw = TRUE;
2878                 }
2879         }
2880
2881         if (io_error(view->pipe)) {
2882                 report("Failed to read: %s", io_strerror(view->pipe));
2883                 end_update(view, TRUE);
2884
2885         } else if (io_eof(view->pipe)) {
2886                 report("");
2887                 end_update(view, FALSE);
2888         }
2889
2890         if (restore_view_position(view))
2891                 redraw = TRUE;
2892
2893         if (!view_is_displayed(view))
2894                 return TRUE;
2895
2896         if (redraw)
2897                 redraw_view_from(view, 0);
2898         else
2899                 redraw_view_dirty(view);
2900
2901         /* Update the title _after_ the redraw so that if the redraw picks up a
2902          * commit reference in view->ref it'll be available here. */
2903         update_view_title(view);
2904         return TRUE;
2905 }
2906
2907 static struct line *
2908 add_line_data(struct view *view, void *data, enum line_type type)
2909 {
2910         struct line *line;
2911
2912         if (!realloc_lines(view, view->lines + 1))
2913                 return NULL;
2914
2915         line = &view->line[view->lines++];
2916         memset(line, 0, sizeof(*line));
2917         line->type = type;
2918         line->data = data;
2919         line->dirty = 1;
2920
2921         return line;
2922 }
2923
2924 static struct line *
2925 add_line_text(struct view *view, const char *text, enum line_type type)
2926 {
2927         char *data = text ? strdup(text) : NULL;
2928
2929         return data ? add_line_data(view, data, type) : NULL;
2930 }
2931
2932 static struct line *
2933 add_line_format(struct view *view, enum line_type type, const char *fmt, ...)
2934 {
2935         char buf[SIZEOF_STR];
2936         va_list args;
2937
2938         va_start(args, fmt);
2939         if (vsnprintf(buf, sizeof(buf), fmt, args) >= sizeof(buf))
2940                 buf[0] = 0;
2941         va_end(args);
2942
2943         return buf[0] ? add_line_text(view, buf, type) : NULL;
2944 }
2945
2946 /*
2947  * View opening
2948  */
2949
2950 enum open_flags {
2951         OPEN_DEFAULT = 0,       /* Use default view switching. */
2952         OPEN_SPLIT = 1,         /* Split current view. */
2953         OPEN_RELOAD = 4,        /* Reload view even if it is the current. */
2954         OPEN_REFRESH = 16,      /* Refresh view using previous command. */
2955         OPEN_PREPARED = 32,     /* Open already prepared command. */
2956 };
2957
2958 static void
2959 open_view(struct view *prev, enum request request, enum open_flags flags)
2960 {
2961         bool split = !!(flags & OPEN_SPLIT);
2962         bool reload = !!(flags & (OPEN_RELOAD | OPEN_REFRESH | OPEN_PREPARED));
2963         bool nomaximize = !!(flags & OPEN_REFRESH);
2964         struct view *view = VIEW(request);
2965         int nviews = displayed_views();
2966         struct view *base_view = display[0];
2967
2968         if (view == prev && nviews == 1 && !reload) {
2969                 report("Already in %s view", view->name);
2970                 return;
2971         }
2972
2973         if (view->git_dir && !opt_git_dir[0]) {
2974                 report("The %s view is disabled in pager view", view->name);
2975                 return;
2976         }
2977
2978         if (split) {
2979                 display[1] = view;
2980                 current_view = 1;
2981         } else if (!nomaximize) {
2982                 /* Maximize the current view. */
2983                 memset(display, 0, sizeof(display));
2984                 current_view = 0;
2985                 display[current_view] = view;
2986         }
2987
2988         /* Resize the view when switching between split- and full-screen,
2989          * or when switching between two different full-screen views. */
2990         if (nviews != displayed_views() ||
2991             (nviews == 1 && base_view != display[0]))
2992                 resize_display();
2993
2994         if (view->ops->open) {
2995                 if (view->pipe)
2996                         end_update(view, TRUE);
2997                 if (!view->ops->open(view)) {
2998                         report("Failed to load %s view", view->name);
2999                         return;
3000                 }
3001                 restore_view_position(view);
3002
3003         } else if ((reload || strcmp(view->vid, view->id)) &&
3004                    !begin_update(view, flags & (OPEN_REFRESH | OPEN_PREPARED))) {
3005                 report("Failed to load %s view", view->name);
3006                 return;
3007         }
3008
3009         if (split && prev->lineno - prev->offset >= prev->height) {
3010                 /* Take the title line into account. */
3011                 int lines = prev->lineno - prev->offset - prev->height + 1;
3012
3013                 /* Scroll the view that was split if the current line is
3014                  * outside the new limited view. */
3015                 do_scroll_view(prev, lines);
3016         }
3017
3018         if (prev && view != prev) {
3019                 if (split) {
3020                         /* "Blur" the previous view. */
3021                         update_view_title(prev);
3022                 }
3023
3024                 view->parent = prev;
3025         }
3026
3027         if (view->pipe && view->lines == 0) {
3028                 /* Clear the old view and let the incremental updating refill
3029                  * the screen. */
3030                 werase(view->win);
3031                 view->p_restore = flags & (OPEN_RELOAD | OPEN_REFRESH);
3032                 report("");
3033         } else if (view_is_displayed(view)) {
3034                 redraw_view(view);
3035                 report("");
3036         }
3037 }
3038
3039 static void
3040 open_external_viewer(const char *argv[], const char *dir)
3041 {
3042         def_prog_mode();           /* save current tty modes */
3043         endwin();                  /* restore original tty modes */
3044         run_io_fg(argv, dir);
3045         fprintf(stderr, "Press Enter to continue");
3046         getc(opt_tty);
3047         reset_prog_mode();
3048         redraw_display(TRUE);
3049 }
3050
3051 static void
3052 open_mergetool(const char *file)
3053 {
3054         const char *mergetool_argv[] = { "git", "mergetool", file, NULL };
3055
3056         open_external_viewer(mergetool_argv, opt_cdup);
3057 }
3058
3059 static void
3060 open_editor(bool from_root, const char *file)
3061 {
3062         const char *editor_argv[] = { "vi", file, NULL };
3063         const char *editor;
3064
3065         editor = getenv("GIT_EDITOR");
3066         if (!editor && *opt_editor)
3067                 editor = opt_editor;
3068         if (!editor)
3069                 editor = getenv("VISUAL");
3070         if (!editor)
3071                 editor = getenv("EDITOR");
3072         if (!editor)
3073                 editor = "vi";
3074
3075         editor_argv[0] = editor;
3076         open_external_viewer(editor_argv, from_root ? opt_cdup : NULL);
3077 }
3078
3079 static void
3080 open_run_request(enum request request)
3081 {
3082         struct run_request *req = get_run_request(request);
3083         const char *argv[ARRAY_SIZE(req->argv)] = { NULL };
3084
3085         if (!req) {
3086                 report("Unknown run request");
3087                 return;
3088         }
3089
3090         if (format_argv(argv, req->argv, FORMAT_ALL))
3091                 open_external_viewer(argv, NULL);
3092         free_argv(argv);
3093 }
3094
3095 /*
3096  * User request switch noodle
3097  */
3098
3099 static int
3100 view_driver(struct view *view, enum request request)
3101 {
3102         int i;
3103
3104         if (request == REQ_NONE) {
3105                 doupdate();
3106                 return TRUE;
3107         }
3108
3109         if (request > REQ_NONE) {
3110                 open_run_request(request);
3111                 /* FIXME: When all views can refresh always do this. */
3112                 if (view == VIEW(REQ_VIEW_STATUS) ||
3113                     view == VIEW(REQ_VIEW_MAIN) ||
3114                     view == VIEW(REQ_VIEW_LOG) ||
3115                     view == VIEW(REQ_VIEW_STAGE))
3116                         request = REQ_REFRESH;
3117                 else
3118                         return TRUE;
3119         }
3120
3121         if (view && view->lines) {
3122                 request = view->ops->request(view, request, &view->line[view->lineno]);
3123                 if (request == REQ_NONE)
3124                         return TRUE;
3125         }
3126
3127         switch (request) {
3128         case REQ_MOVE_UP:
3129         case REQ_MOVE_DOWN:
3130         case REQ_MOVE_PAGE_UP:
3131         case REQ_MOVE_PAGE_DOWN:
3132         case REQ_MOVE_FIRST_LINE:
3133         case REQ_MOVE_LAST_LINE:
3134                 move_view(view, request);
3135                 break;
3136
3137         case REQ_SCROLL_LEFT:
3138         case REQ_SCROLL_RIGHT:
3139         case REQ_SCROLL_LINE_DOWN:
3140         case REQ_SCROLL_LINE_UP:
3141         case REQ_SCROLL_PAGE_DOWN:
3142         case REQ_SCROLL_PAGE_UP:
3143                 scroll_view(view, request);
3144                 break;
3145
3146         case REQ_VIEW_BLAME:
3147                 if (!opt_file[0]) {
3148                         report("No file chosen, press %s to open tree view",
3149                                get_key(REQ_VIEW_TREE));
3150                         break;
3151                 }
3152                 open_view(view, request, OPEN_DEFAULT);
3153                 break;
3154
3155         case REQ_VIEW_BLOB:
3156                 if (!ref_blob[0]) {
3157                         report("No file chosen, press %s to open tree view",
3158                                get_key(REQ_VIEW_TREE));
3159                         break;
3160                 }
3161                 open_view(view, request, OPEN_DEFAULT);
3162                 break;
3163
3164         case REQ_VIEW_PAGER:
3165                 if (!VIEW(REQ_VIEW_PAGER)->pipe && !VIEW(REQ_VIEW_PAGER)->lines) {
3166                         report("No pager content, press %s to run command from prompt",
3167                                get_key(REQ_PROMPT));
3168                         break;
3169                 }
3170                 open_view(view, request, OPEN_DEFAULT);
3171                 break;
3172
3173         case REQ_VIEW_STAGE:
3174                 if (!VIEW(REQ_VIEW_STAGE)->lines) {
3175                         report("No stage content, press %s to open the status view and choose file",
3176                                get_key(REQ_VIEW_STATUS));
3177                         break;
3178                 }
3179                 open_view(view, request, OPEN_DEFAULT);
3180                 break;
3181
3182         case REQ_VIEW_STATUS:
3183                 if (opt_is_inside_work_tree == FALSE) {
3184                         report("The status view requires a working tree");
3185                         break;
3186                 }
3187                 open_view(view, request, OPEN_DEFAULT);
3188                 break;
3189
3190         case REQ_VIEW_MAIN:
3191         case REQ_VIEW_DIFF:
3192         case REQ_VIEW_LOG:
3193         case REQ_VIEW_TREE:
3194         case REQ_VIEW_HELP:
3195                 open_view(view, request, OPEN_DEFAULT);
3196                 break;
3197
3198         case REQ_NEXT:
3199         case REQ_PREVIOUS:
3200                 request = request == REQ_NEXT ? REQ_MOVE_DOWN : REQ_MOVE_UP;
3201
3202                 if ((view == VIEW(REQ_VIEW_DIFF) &&
3203                      view->parent == VIEW(REQ_VIEW_MAIN)) ||
3204                    (view == VIEW(REQ_VIEW_DIFF) &&
3205                      view->parent == VIEW(REQ_VIEW_BLAME)) ||
3206                    (view == VIEW(REQ_VIEW_STAGE) &&
3207                      view->parent == VIEW(REQ_VIEW_STATUS)) ||
3208                    (view == VIEW(REQ_VIEW_BLOB) &&
3209                      view->parent == VIEW(REQ_VIEW_TREE))) {
3210                         int line;
3211
3212                         view = view->parent;
3213                         line = view->lineno;
3214                         move_view(view, request);
3215                         if (view_is_displayed(view))
3216                                 update_view_title(view);
3217                         if (line != view->lineno)
3218                                 view->ops->request(view, REQ_ENTER,
3219                                                    &view->line[view->lineno]);
3220
3221                 } else {
3222                         move_view(view, request);
3223                 }
3224                 break;
3225
3226         case REQ_VIEW_NEXT:
3227         {
3228                 int nviews = displayed_views();
3229                 int next_view = (current_view + 1) % nviews;
3230
3231                 if (next_view == current_view) {
3232                         report("Only one view is displayed");
3233                         break;
3234                 }
3235
3236                 current_view = next_view;
3237                 /* Blur out the title of the previous view. */
3238                 update_view_title(view);
3239                 report("");
3240                 break;
3241         }
3242         case REQ_REFRESH:
3243                 report("Refreshing is not yet supported for the %s view", view->name);
3244                 break;
3245
3246         case REQ_MAXIMIZE:
3247                 if (displayed_views() == 2)
3248                         maximize_view(view);
3249                 break;
3250
3251         case REQ_TOGGLE_LINENO:
3252                 toggle_view_option(&opt_line_number, "line numbers");
3253                 break;
3254
3255         case REQ_TOGGLE_DATE:
3256                 toggle_view_option(&opt_date, "date display");
3257                 break;
3258
3259         case REQ_TOGGLE_AUTHOR:
3260                 toggle_view_option(&opt_author, "author display");
3261                 break;
3262
3263         case REQ_TOGGLE_REV_GRAPH:
3264                 toggle_view_option(&opt_rev_graph, "revision graph display");
3265                 break;
3266
3267         case REQ_TOGGLE_REFS:
3268                 toggle_view_option(&opt_show_refs, "reference display");
3269                 break;
3270
3271         case REQ_SEARCH:
3272         case REQ_SEARCH_BACK:
3273                 search_view(view, request);
3274                 break;
3275
3276         case REQ_FIND_NEXT:
3277         case REQ_FIND_PREV:
3278                 find_next(view, request);
3279                 break;
3280
3281         case REQ_STOP_LOADING:
3282                 for (i = 0; i < ARRAY_SIZE(views); i++) {
3283                         view = &views[i];
3284                         if (view->pipe)
3285                                 report("Stopped loading the %s view", view->name),
3286                         end_update(view, TRUE);
3287                 }
3288                 break;
3289
3290         case REQ_SHOW_VERSION:
3291                 report("tig-%s (built %s)", TIG_VERSION, __DATE__);
3292                 return TRUE;
3293
3294         case REQ_SCREEN_REDRAW:
3295                 redraw_display(TRUE);
3296                 break;
3297
3298         case REQ_EDIT:
3299                 report("Nothing to edit");
3300                 break;
3301
3302         case REQ_ENTER:
3303                 report("Nothing to enter");
3304                 break;
3305
3306         case REQ_VIEW_CLOSE:
3307                 /* XXX: Mark closed views by letting view->parent point to the
3308                  * view itself. Parents to closed view should never be
3309                  * followed. */
3310                 if (view->parent &&
3311                     view->parent->parent != view->parent) {
3312                         maximize_view(view->parent);
3313                         view->parent = view;
3314                         break;
3315                 }
3316                 /* Fall-through */
3317         case REQ_QUIT:
3318                 return FALSE;
3319
3320         default:
3321                 report("Unknown key, press 'h' for help");
3322                 return TRUE;
3323         }
3324
3325         return TRUE;
3326 }
3327
3328
3329 /*
3330  * View backend utilities
3331  */
3332
3333 static void
3334 parse_timezone(time_t *time, const char *zone)
3335 {
3336         long tz;
3337
3338         tz  = ('0' - zone[1]) * 60 * 60 * 10;
3339         tz += ('0' - zone[2]) * 60 * 60;
3340         tz += ('0' - zone[3]) * 60;
3341         tz += ('0' - zone[4]);
3342
3343         if (zone[0] == '-')
3344                 tz = -tz;
3345
3346         *time -= tz;
3347 }
3348
3349 /* Parse author lines where the name may be empty:
3350  *      author  <email@address.tld> 1138474660 +0100
3351  */
3352 static void
3353 parse_author_line(char *ident, char *author, size_t authorsize, struct tm *tm)
3354 {
3355         char *nameend = strchr(ident, '<');
3356         char *emailend = strchr(ident, '>');
3357
3358         if (nameend && emailend)
3359                 *nameend = *emailend = 0;
3360         ident = chomp_string(ident);
3361         if (!*ident) {
3362                 if (nameend)
3363                         ident = chomp_string(nameend + 1);
3364                 if (!*ident)
3365                         ident = "Unknown";
3366         }
3367
3368         string_ncopy_do(author, authorsize, ident, strlen(ident));
3369
3370         /* Parse epoch and timezone */
3371         if (emailend && emailend[1] == ' ') {
3372                 char *secs = emailend + 2;
3373                 char *zone = strchr(secs, ' ');
3374                 time_t time = (time_t) atol(secs);
3375
3376                 if (zone && strlen(zone) == STRING_SIZE(" +0700"))
3377                         parse_timezone(&time, zone + 1);
3378
3379                 gmtime_r(&time, tm);
3380         }
3381 }
3382
3383 static enum input_status
3384 select_commit_parent_handler(void *data, char *buf, int c)
3385 {
3386         size_t parents = *(size_t *) data;
3387         int parent = 0;
3388
3389         if (!isdigit(c))
3390                 return INPUT_SKIP;
3391
3392         if (*buf)
3393                 parent = atoi(buf) * 10;
3394         parent += c - '0';
3395
3396         if (parent > parents)
3397                 return INPUT_SKIP;
3398         return INPUT_OK;
3399 }
3400
3401 static bool
3402 select_commit_parent(const char *id, char rev[SIZEOF_REV], const char *path)
3403 {
3404         char buf[SIZEOF_STR * 4];
3405         const char *revlist_argv[] = {
3406                 "git", "rev-list", "-1", "--parents", id, "--", path, NULL
3407         };
3408         int parents;
3409
3410         if (!run_io_buf(revlist_argv, buf, sizeof(buf)) ||
3411             !*chomp_string(buf) ||
3412             (parents = (strlen(buf) / 40) - 1) < 0) {
3413                 report("Failed to get parent information");
3414                 return FALSE;
3415
3416         } else if (parents == 0) {
3417                 if (path)
3418                         report("Path '%s' does not exist in the parent", path);
3419                 else
3420                         report("The selected commit has no parents");
3421                 return FALSE;
3422         }
3423
3424         if (parents > 1) {
3425                 char prompt[SIZEOF_STR];
3426                 char *result;
3427
3428                 if (!string_format(prompt, "Which parent? [1..%d] ", parents))
3429                         return FALSE;
3430                 result = prompt_input(prompt, select_commit_parent_handler, &parents);
3431                 if (!result)
3432                         return FALSE;
3433                 parents = atoi(result);
3434         }
3435
3436         string_copy_rev(rev, &buf[41 * parents]);
3437         return TRUE;
3438 }
3439
3440 /*
3441  * Pager backend
3442  */
3443
3444 static bool
3445 pager_draw(struct view *view, struct line *line, unsigned int lineno)
3446 {
3447         char text[SIZEOF_STR];
3448
3449         if (opt_line_number && draw_lineno(view, lineno))
3450                 return TRUE;
3451
3452         string_expand(text, sizeof(text), line->data, opt_tab_size);
3453         draw_text(view, line->type, text, TRUE);
3454         return TRUE;
3455 }
3456
3457 static bool
3458 add_describe_ref(char *buf, size_t *bufpos, const char *commit_id, const char *sep)
3459 {
3460         const char *describe_argv[] = { "git", "describe", commit_id, NULL };
3461         char refbuf[SIZEOF_STR];
3462         char *ref = NULL;
3463
3464         if (run_io_buf(describe_argv, refbuf, sizeof(refbuf)))
3465                 ref = chomp_string(refbuf);
3466
3467         if (!ref || !*ref)
3468                 return TRUE;
3469
3470         /* This is the only fatal call, since it can "corrupt" the buffer. */
3471         if (!string_nformat(buf, SIZEOF_STR, bufpos, "%s%s", sep, ref))
3472                 return FALSE;
3473
3474         return TRUE;
3475 }
3476
3477 static void
3478 add_pager_refs(struct view *view, struct line *line)
3479 {
3480         char buf[SIZEOF_STR];
3481         char *commit_id = (char *)line->data + STRING_SIZE("commit ");
3482         struct ref **refs;
3483         size_t bufpos = 0, refpos = 0;
3484         const char *sep = "Refs: ";
3485         bool is_tag = FALSE;
3486
3487         assert(line->type == LINE_COMMIT);
3488
3489         refs = get_refs(commit_id);
3490         if (!refs) {
3491                 if (view == VIEW(REQ_VIEW_DIFF))
3492                         goto try_add_describe_ref;
3493                 return;
3494         }
3495
3496         do {
3497                 struct ref *ref = refs[refpos];
3498                 const char *fmt = ref->tag    ? "%s[%s]" :
3499                                   ref->remote ? "%s<%s>" : "%s%s";
3500
3501                 if (!string_format_from(buf, &bufpos, fmt, sep, ref->name))
3502                         return;
3503                 sep = ", ";
3504                 if (ref->tag)
3505                         is_tag = TRUE;
3506         } while (refs[refpos++]->next);
3507
3508         if (!is_tag && view == VIEW(REQ_VIEW_DIFF)) {
3509 try_add_describe_ref:
3510                 /* Add <tag>-g<commit_id> "fake" reference. */
3511                 if (!add_describe_ref(buf, &bufpos, commit_id, sep))
3512                         return;
3513         }
3514
3515         if (bufpos == 0)
3516                 return;
3517
3518         add_line_text(view, buf, LINE_PP_REFS);
3519 }
3520
3521 static bool
3522 pager_read(struct view *view, char *data)
3523 {
3524         struct line *line;
3525
3526         if (!data)
3527                 return TRUE;
3528
3529         line = add_line_text(view, data, get_line_type(data));
3530         if (!line)
3531                 return FALSE;
3532
3533         if (line->type == LINE_COMMIT &&
3534             (view == VIEW(REQ_VIEW_DIFF) ||
3535              view == VIEW(REQ_VIEW_LOG)))
3536                 add_pager_refs(view, line);
3537
3538         return TRUE;
3539 }
3540
3541 static enum request
3542 pager_request(struct view *view, enum request request, struct line *line)
3543 {
3544         int split = 0;
3545
3546         if (request != REQ_ENTER)
3547                 return request;
3548
3549         if (line->type == LINE_COMMIT &&
3550            (view == VIEW(REQ_VIEW_LOG) ||
3551             view == VIEW(REQ_VIEW_PAGER))) {
3552                 open_view(view, REQ_VIEW_DIFF, OPEN_SPLIT);
3553                 split = 1;
3554         }
3555
3556         /* Always scroll the view even if it was split. That way
3557          * you can use Enter to scroll through the log view and
3558          * split open each commit diff. */
3559         scroll_view(view, REQ_SCROLL_LINE_DOWN);
3560
3561         /* FIXME: A minor workaround. Scrolling the view will call report("")
3562          * but if we are scrolling a non-current view this won't properly
3563          * update the view title. */
3564         if (split)
3565                 update_view_title(view);
3566
3567         return REQ_NONE;
3568 }
3569
3570 static bool
3571 pager_grep(struct view *view, struct line *line)
3572 {
3573         regmatch_t pmatch;
3574         char *text = line->data;
3575
3576         if (!*text)
3577                 return FALSE;
3578
3579         if (regexec(view->regex, text, 1, &pmatch, 0) == REG_NOMATCH)
3580                 return FALSE;
3581
3582         return TRUE;
3583 }
3584
3585 static void
3586 pager_select(struct view *view, struct line *line)
3587 {
3588         if (line->type == LINE_COMMIT) {
3589                 char *text = (char *)line->data + STRING_SIZE("commit ");
3590
3591                 if (view != VIEW(REQ_VIEW_PAGER))
3592                         string_copy_rev(view->ref, text);
3593                 string_copy_rev(ref_commit, text);
3594         }
3595 }
3596
3597 static struct view_ops pager_ops = {
3598         "line",
3599         NULL,
3600         NULL,
3601         pager_read,
3602         pager_draw,
3603         pager_request,
3604         pager_grep,
3605         pager_select,
3606 };
3607
3608 static const char *log_argv[SIZEOF_ARG] = {
3609         "git", "log", "--no-color", "--cc", "--stat", "-n100", "%(head)", NULL
3610 };
3611
3612 static enum request
3613 log_request(struct view *view, enum request request, struct line *line)
3614 {
3615         switch (request) {
3616         case REQ_REFRESH:
3617                 load_refs();
3618                 open_view(view, REQ_VIEW_LOG, OPEN_REFRESH);
3619                 return REQ_NONE;
3620         default:
3621                 return pager_request(view, request, line);
3622         }
3623 }
3624
3625 static struct view_ops log_ops = {
3626         "line",
3627         log_argv,
3628         NULL,
3629         pager_read,
3630         pager_draw,
3631         log_request,
3632         pager_grep,
3633         pager_select,
3634 };
3635
3636 static const char *diff_argv[SIZEOF_ARG] = {
3637         "git", "show", "--pretty=fuller", "--no-color", "--root",
3638                 "--patch-with-stat", "--find-copies-harder", "-C", "%(commit)", NULL
3639 };
3640
3641 static struct view_ops diff_ops = {
3642         "line",
3643         diff_argv,
3644         NULL,
3645         pager_read,
3646         pager_draw,
3647         pager_request,
3648         pager_grep,
3649         pager_select,
3650 };
3651
3652 /*
3653  * Help backend
3654  */
3655
3656 static bool
3657 help_open(struct view *view)
3658 {
3659         char buf[SIZEOF_STR];
3660         size_t bufpos;
3661         int i;
3662
3663         if (view->lines > 0)
3664                 return TRUE;
3665
3666         add_line_text(view, "Quick reference for tig keybindings:", LINE_DEFAULT);
3667
3668         for (i = 0; i < ARRAY_SIZE(req_info); i++) {
3669                 const char *key;
3670
3671                 if (req_info[i].request == REQ_NONE)
3672                         continue;
3673
3674                 if (!req_info[i].request) {
3675                         add_line_text(view, "", LINE_DEFAULT);
3676                         add_line_text(view, req_info[i].help, LINE_DEFAULT);
3677                         continue;
3678                 }
3679
3680                 key = get_key(req_info[i].request);
3681                 if (!*key)
3682                         key = "(no key defined)";
3683
3684                 for (bufpos = 0; bufpos <= req_info[i].namelen; bufpos++) {
3685                         buf[bufpos] = tolower(req_info[i].name[bufpos]);
3686                         if (buf[bufpos] == '_')
3687                                 buf[bufpos] = '-';
3688                 }
3689
3690                 add_line_format(view, LINE_DEFAULT, "    %-25s %-20s %s",
3691                                 key, buf, req_info[i].help);
3692         }
3693
3694         if (run_requests) {
3695                 add_line_text(view, "", LINE_DEFAULT);
3696                 add_line_text(view, "External commands:", LINE_DEFAULT);
3697         }
3698
3699         for (i = 0; i < run_requests; i++) {
3700                 struct run_request *req = get_run_request(REQ_NONE + i + 1);
3701                 const char *key;
3702                 int argc;
3703
3704                 if (!req)
3705                         continue;
3706
3707                 key = get_key_name(req->key);
3708                 if (!*key)
3709                         key = "(no key defined)";
3710
3711                 for (bufpos = 0, argc = 0; req->argv[argc]; argc++)
3712                         if (!string_format_from(buf, &bufpos, "%s%s",
3713                                                 argc ? " " : "", req->argv[argc]))
3714                                 return REQ_NONE;
3715
3716                 add_line_format(view, LINE_DEFAULT, "    %-10s %-14s `%s`",
3717                                 keymap_table[req->keymap].name, key, buf);
3718         }
3719
3720         return TRUE;
3721 }
3722
3723 static struct view_ops help_ops = {
3724         "line",
3725         NULL,
3726         help_open,
3727         NULL,
3728         pager_draw,
3729         pager_request,
3730         pager_grep,
3731         pager_select,
3732 };
3733
3734
3735 /*
3736  * Tree backend
3737  */
3738
3739 struct tree_stack_entry {
3740         struct tree_stack_entry *prev;  /* Entry below this in the stack */
3741         unsigned long lineno;           /* Line number to restore */
3742         char *name;                     /* Position of name in opt_path */
3743 };
3744
3745 /* The top of the path stack. */
3746 static struct tree_stack_entry *tree_stack = NULL;
3747 unsigned long tree_lineno = 0;
3748
3749 static void
3750 pop_tree_stack_entry(void)
3751 {
3752         struct tree_stack_entry *entry = tree_stack;
3753
3754         tree_lineno = entry->lineno;
3755         entry->name[0] = 0;
3756         tree_stack = entry->prev;
3757         free(entry);
3758 }
3759
3760 static void
3761 push_tree_stack_entry(const char *name, unsigned long lineno)
3762 {
3763         struct tree_stack_entry *entry = calloc(1, sizeof(*entry));
3764         size_t pathlen = strlen(opt_path);
3765
3766         if (!entry)
3767                 return;
3768
3769         entry->prev = tree_stack;
3770         entry->name = opt_path + pathlen;
3771         tree_stack = entry;
3772
3773         if (!string_format_from(opt_path, &pathlen, "%s/", name)) {
3774                 pop_tree_stack_entry();
3775                 return;
3776         }
3777
3778         /* Move the current line to the first tree entry. */
3779         tree_lineno = 1;
3780         entry->lineno = lineno;
3781 }
3782
3783 /* Parse output from git-ls-tree(1):
3784  *
3785  * 100644 blob f931e1d229c3e185caad4449bf5b66ed72462657 tig.c
3786  */
3787
3788 #define SIZEOF_TREE_ATTR \
3789         STRING_SIZE("100644 blob f931e1d229c3e185caad4449bf5b66ed72462657\t")
3790
3791 #define SIZEOF_TREE_MODE \
3792         STRING_SIZE("100644 ")
3793
3794 #define TREE_ID_OFFSET \
3795         STRING_SIZE("100644 blob ")
3796
3797 struct tree_entry {
3798         char id[SIZEOF_REV];
3799         mode_t mode;
3800         struct tm time;                 /* Date from the author ident. */
3801         char author[75];                /* Author of the commit. */
3802         char name[1];
3803 };
3804
3805 static const char *
3806 tree_path(struct line *line)
3807 {
3808         return ((struct tree_entry *) line->data)->name;
3809 }
3810
3811
3812 static int
3813 tree_compare_entry(struct line *line1, struct line *line2)
3814 {
3815         if (line1->type != line2->type)
3816                 return line1->type == LINE_TREE_DIR ? -1 : 1;
3817         return strcmp(tree_path(line1), tree_path(line2));
3818 }
3819
3820 static struct line *
3821 tree_entry(struct view *view, enum line_type type, const char *path,
3822            const char *mode, const char *id)
3823 {
3824         struct tree_entry *entry = calloc(1, sizeof(*entry) + strlen(path));
3825         struct line *line = entry ? add_line_data(view, entry, type) : NULL;
3826
3827         if (!entry || !line) {
3828                 free(entry);
3829                 return NULL;
3830         }
3831
3832         strncpy(entry->name, path, strlen(path));
3833         if (mode)
3834                 entry->mode = strtoul(mode, NULL, 8);
3835         if (id)
3836                 string_copy_rev(entry->id, id);
3837
3838         return line;
3839 }
3840
3841 static bool
3842 tree_read_date(struct view *view, char *text, bool *read_date)
3843 {
3844         static char author_name[SIZEOF_STR];
3845         static struct tm author_time;
3846
3847         if (!text && *read_date) {
3848                 *read_date = FALSE;
3849                 return TRUE;
3850
3851         } else if (!text) {
3852                 char *path = *opt_path ? opt_path : ".";
3853                 /* Find next entry to process */
3854                 const char *log_file[] = {
3855                         "git", "log", "--no-color", "--pretty=raw",
3856                                 "--cc", "--raw", view->id, "--", path, NULL
3857                 };
3858                 struct io io = {};
3859
3860                 if (!view->lines) {
3861                         tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL);
3862                         report("Tree is empty");
3863                         return TRUE;
3864                 }
3865
3866                 if (!run_io_rd(&io, log_file, FORMAT_NONE)) {
3867                         report("Failed to load tree data");
3868                         return TRUE;
3869                 }
3870
3871                 done_io(view->pipe);
3872                 view->io = io;
3873                 *read_date = TRUE;
3874                 return FALSE;
3875
3876         } else if (*text == 'a' && get_line_type(text) == LINE_AUTHOR) {
3877                 parse_author_line(text + STRING_SIZE("author "),
3878                                   author_name, sizeof(author_name), &author_time);
3879
3880         } else if (*text == ':') {
3881                 char *pos;
3882                 size_t annotated = 1;
3883                 size_t i;
3884
3885                 pos = strchr(text, '\t');
3886                 if (!pos)
3887                         return TRUE;
3888                 text = pos + 1;
3889                 if (*opt_prefix && !strncmp(text, opt_prefix, strlen(opt_prefix)))
3890                         text += strlen(opt_prefix);
3891                 if (*opt_path && !strncmp(text, opt_path, strlen(opt_path)))
3892                         text += strlen(opt_path);
3893                 pos = strchr(text, '/');
3894                 if (pos)
3895                         *pos = 0;
3896
3897                 for (i = 1; i < view->lines; i++) {
3898                         struct line *line = &view->line[i];
3899                         struct tree_entry *entry = line->data;
3900
3901                         annotated += !!*entry->author;
3902                         if (*entry->author || strcmp(entry->name, text))
3903                                 continue;
3904
3905                         string_copy(entry->author, author_name);
3906                         memcpy(&entry->time, &author_time, sizeof(entry->time));
3907                         line->dirty = 1;
3908                         break;
3909                 }
3910
3911                 if (annotated == view->lines)
3912                         kill_io(view->pipe);
3913         }
3914         return TRUE;
3915 }
3916
3917 static bool
3918 tree_read(struct view *view, char *text)
3919 {
3920         static bool read_date = FALSE;
3921         struct tree_entry *data;
3922         struct line *entry, *line;
3923         enum line_type type;
3924         size_t textlen = text ? strlen(text) : 0;
3925         char *path = text + SIZEOF_TREE_ATTR;
3926
3927         if (read_date || !text)
3928                 return tree_read_date(view, text, &read_date);
3929
3930         if (textlen <= SIZEOF_TREE_ATTR)
3931                 return FALSE;
3932         if (view->lines == 0 &&
3933             !tree_entry(view, LINE_TREE_HEAD, opt_path, NULL, NULL))
3934                 return FALSE;
3935
3936         /* Strip the path part ... */
3937         if (*opt_path) {
3938                 size_t pathlen = textlen - SIZEOF_TREE_ATTR;
3939                 size_t striplen = strlen(opt_path);
3940
3941                 if (pathlen > striplen)
3942                         memmove(path, path + striplen,
3943                                 pathlen - striplen + 1);
3944
3945                 /* Insert "link" to parent directory. */
3946                 if (view->lines == 1 &&
3947                     !tree_entry(view, LINE_TREE_DIR, "..", "040000", view->ref))
3948                         return FALSE;
3949         }
3950
3951         type = text[SIZEOF_TREE_MODE] == 't' ? LINE_TREE_DIR : LINE_TREE_FILE;
3952         entry = tree_entry(view, type, path, text, text + TREE_ID_OFFSET);
3953         if (!entry)
3954                 return FALSE;
3955         data = entry->data;
3956
3957         /* Skip "Directory ..." and ".." line. */
3958         for (line = &view->line[1 + !!*opt_path]; line < entry; line++) {
3959                 if (tree_compare_entry(line, entry) <= 0)
3960                         continue;
3961
3962                 memmove(line + 1, line, (entry - line) * sizeof(*entry));
3963
3964                 line->data = data;
3965                 line->type = type;
3966                 for (; line <= entry; line++)
3967                         line->dirty = line->cleareol = 1;
3968                 return TRUE;
3969         }
3970
3971         if (tree_lineno > view->lineno) {
3972                 view->lineno = tree_lineno;
3973                 tree_lineno = 0;
3974         }
3975
3976         return TRUE;
3977 }
3978
3979 static bool
3980 tree_draw(struct view *view, struct line *line, unsigned int lineno)
3981 {
3982         struct tree_entry *entry = line->data;
3983
3984         if (line->type == LINE_TREE_HEAD) {
3985                 if (draw_text(view, line->type, "Directory path /", TRUE))
3986                         return TRUE;
3987         } else {
3988                 if (draw_mode(view, entry->mode))
3989                         return TRUE;
3990
3991                 if (opt_author && draw_author(view, entry->author))
3992                         return TRUE;
3993
3994                 if (opt_date && draw_date(view, *entry->author ? &entry->time : NULL))
3995                         return TRUE;
3996         }
3997         if (draw_text(view, line->type, entry->name, TRUE))
3998                 return TRUE;
3999         return TRUE;
4000 }
4001
4002 static void
4003 open_blob_editor()
4004 {
4005         char file[SIZEOF_STR] = "/tmp/tigblob.XXXXXX";
4006         int fd = mkstemp(file);
4007
4008         if (fd == -1)
4009                 report("Failed to create temporary file");
4010         else if (!run_io_append(blob_ops.argv, FORMAT_ALL, fd))
4011                 report("Failed to save blob data to file");
4012         else
4013                 open_editor(FALSE, file);
4014         if (fd != -1)
4015                 unlink(file);
4016 }
4017
4018 static enum request
4019 tree_request(struct view *view, enum request request, struct line *line)
4020 {
4021         enum open_flags flags;
4022
4023         switch (request) {
4024         case REQ_VIEW_BLAME:
4025                 if (line->type != LINE_TREE_FILE) {
4026                         report("Blame only supported for files");
4027                         return REQ_NONE;
4028                 }
4029
4030                 string_copy(opt_ref, view->vid);
4031                 return request;
4032
4033         case REQ_EDIT:
4034                 if (line->type != LINE_TREE_FILE) {
4035                         report("Edit only supported for files");
4036                 } else if (!is_head_commit(view->vid)) {
4037                         open_blob_editor();
4038                 } else {
4039                         open_editor(TRUE, opt_file);
4040                 }
4041                 return REQ_NONE;
4042
4043         case REQ_PARENT:
4044                 if (!*opt_path) {
4045                         /* quit view if at top of tree */
4046                         return REQ_VIEW_CLOSE;
4047                 }
4048                 /* fake 'cd  ..' */
4049                 line = &view->line[1];
4050                 break;
4051
4052         case REQ_ENTER:
4053                 break;
4054
4055         default:
4056                 return request;
4057         }
4058
4059         /* Cleanup the stack if the tree view is at a different tree. */
4060         while (!*opt_path && tree_stack)
4061                 pop_tree_stack_entry();
4062
4063         switch (line->type) {
4064         case LINE_TREE_DIR:
4065                 /* Depending on whether it is a subdirectory or parent link
4066                  * mangle the path buffer. */
4067                 if (line == &view->line[1] && *opt_path) {
4068                         pop_tree_stack_entry();
4069
4070                 } else {
4071                         const char *basename = tree_path(line);
4072
4073                         push_tree_stack_entry(basename, view->lineno);
4074                 }
4075
4076                 /* Trees and subtrees share the same ID, so they are not not
4077                  * unique like blobs. */
4078                 flags = OPEN_RELOAD;
4079                 request = REQ_VIEW_TREE;
4080                 break;
4081
4082         case LINE_TREE_FILE:
4083                 flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4084                 request = REQ_VIEW_BLOB;
4085                 break;
4086
4087         default:
4088                 return REQ_NONE;
4089         }
4090
4091         open_view(view, request, flags);
4092         if (request == REQ_VIEW_TREE)
4093                 view->lineno = tree_lineno;
4094
4095         return REQ_NONE;
4096 }
4097
4098 static void
4099 tree_select(struct view *view, struct line *line)
4100 {
4101         struct tree_entry *entry = line->data;
4102
4103         if (line->type == LINE_TREE_FILE) {
4104                 string_copy_rev(ref_blob, entry->id);
4105                 string_format(opt_file, "%s%s", opt_path, tree_path(line));
4106
4107         } else if (line->type != LINE_TREE_DIR) {
4108                 return;
4109         }
4110
4111         string_copy_rev(view->ref, entry->id);
4112 }
4113
4114 static const char *tree_argv[SIZEOF_ARG] = {
4115         "git", "ls-tree", "%(commit)", "%(directory)", NULL
4116 };
4117
4118 static struct view_ops tree_ops = {
4119         "file",
4120         tree_argv,
4121         NULL,
4122         tree_read,
4123         tree_draw,
4124         tree_request,
4125         pager_grep,
4126         tree_select,
4127 };
4128
4129 static bool
4130 blob_read(struct view *view, char *line)
4131 {
4132         if (!line)
4133                 return TRUE;
4134         return add_line_text(view, line, LINE_DEFAULT) != NULL;
4135 }
4136
4137 static enum request
4138 blob_request(struct view *view, enum request request, struct line *line)
4139 {
4140         switch (request) {
4141         case REQ_EDIT:
4142                 open_blob_editor();
4143                 return REQ_NONE;
4144         default:
4145                 return pager_request(view, request, line);
4146         }
4147 }
4148
4149 static const char *blob_argv[SIZEOF_ARG] = {
4150         "git", "cat-file", "blob", "%(blob)", NULL
4151 };
4152
4153 static struct view_ops blob_ops = {
4154         "line",
4155         blob_argv,
4156         NULL,
4157         blob_read,
4158         pager_draw,
4159         blob_request,
4160         pager_grep,
4161         pager_select,
4162 };
4163
4164 /*
4165  * Blame backend
4166  *
4167  * Loading the blame view is a two phase job:
4168  *
4169  *  1. File content is read either using opt_file from the
4170  *     filesystem or using git-cat-file.
4171  *  2. Then blame information is incrementally added by
4172  *     reading output from git-blame.
4173  */
4174
4175 static const char *blame_head_argv[] = {
4176         "git", "blame", "--incremental", "--", "%(file)", NULL
4177 };
4178
4179 static const char *blame_ref_argv[] = {
4180         "git", "blame", "--incremental", "%(ref)", "--", "%(file)", NULL
4181 };
4182
4183 static const char *blame_cat_file_argv[] = {
4184         "git", "cat-file", "blob", "%(ref):%(file)", NULL
4185 };
4186
4187 struct blame_commit {
4188         char id[SIZEOF_REV];            /* SHA1 ID. */
4189         char title[128];                /* First line of the commit message. */
4190         char author[75];                /* Author of the commit. */
4191         struct tm time;                 /* Date from the author ident. */
4192         char filename[128];             /* Name of file. */
4193         bool has_previous;              /* Was a "previous" line detected. */
4194 };
4195
4196 struct blame {
4197         struct blame_commit *commit;
4198         unsigned long lineno;
4199         char text[1];
4200 };
4201
4202 static bool
4203 blame_open(struct view *view)
4204 {
4205         if (*opt_ref || !io_open(&view->io, opt_file)) {
4206                 if (!run_io_rd(&view->io, blame_cat_file_argv, FORMAT_ALL))
4207                         return FALSE;
4208         }
4209
4210         setup_update(view, opt_file);
4211         string_format(view->ref, "%s ...", opt_file);
4212
4213         return TRUE;
4214 }
4215
4216 static struct blame_commit *
4217 get_blame_commit(struct view *view, const char *id)
4218 {
4219         size_t i;
4220
4221         for (i = 0; i < view->lines; i++) {
4222                 struct blame *blame = view->line[i].data;
4223
4224                 if (!blame->commit)
4225                         continue;
4226
4227                 if (!strncmp(blame->commit->id, id, SIZEOF_REV - 1))
4228                         return blame->commit;
4229         }
4230
4231         {
4232                 struct blame_commit *commit = calloc(1, sizeof(*commit));
4233
4234                 if (commit)
4235                         string_ncopy(commit->id, id, SIZEOF_REV);
4236                 return commit;
4237         }
4238 }
4239
4240 static bool
4241 parse_number(const char **posref, size_t *number, size_t min, size_t max)
4242 {
4243         const char *pos = *posref;
4244
4245         *posref = NULL;
4246         pos = strchr(pos + 1, ' ');
4247         if (!pos || !isdigit(pos[1]))
4248                 return FALSE;
4249         *number = atoi(pos + 1);
4250         if (*number < min || *number > max)
4251                 return FALSE;
4252
4253         *posref = pos;
4254         return TRUE;
4255 }
4256
4257 static struct blame_commit *
4258 parse_blame_commit(struct view *view, const char *text, int *blamed)
4259 {
4260         struct blame_commit *commit;
4261         struct blame *blame;
4262         const char *pos = text + SIZEOF_REV - 2;
4263         size_t orig_lineno = 0;
4264         size_t lineno;
4265         size_t group;
4266
4267         if (strlen(text) <= SIZEOF_REV || pos[1] != ' ')
4268                 return NULL;
4269
4270         if (!parse_number(&pos, &orig_lineno, 1, 9999999) ||
4271             !parse_number(&pos, &lineno, 1, view->lines) ||
4272             !parse_number(&pos, &group, 1, view->lines - lineno + 1))
4273                 return NULL;
4274
4275         commit = get_blame_commit(view, text);
4276         if (!commit)
4277                 return NULL;
4278
4279         *blamed += group;
4280         while (group--) {
4281                 struct line *line = &view->line[lineno + group - 1];
4282
4283                 blame = line->data;
4284                 blame->commit = commit;
4285                 blame->lineno = orig_lineno + group - 1;
4286                 line->dirty = 1;
4287         }
4288
4289         return commit;
4290 }
4291
4292 static bool
4293 blame_read_file(struct view *view, const char *line, bool *read_file)
4294 {
4295         if (!line) {
4296                 const char **argv = *opt_ref ? blame_ref_argv : blame_head_argv;
4297                 struct io io = {};
4298
4299                 if (view->lines == 0 && !view->parent)
4300                         die("No blame exist for %s", view->vid);
4301
4302                 if (view->lines == 0 || !run_io_rd(&io, argv, FORMAT_ALL)) {
4303                         report("Failed to load blame data");
4304                         return TRUE;
4305                 }
4306
4307                 done_io(view->pipe);
4308                 view->io = io;
4309                 *read_file = FALSE;
4310                 return FALSE;
4311
4312         } else {
4313                 size_t linelen = strlen(line);
4314                 struct blame *blame = malloc(sizeof(*blame) + linelen);
4315
4316                 if (!blame)
4317                         return FALSE;
4318
4319                 blame->commit = NULL;
4320                 strncpy(blame->text, line, linelen);
4321                 blame->text[linelen] = 0;
4322                 return add_line_data(view, blame, LINE_BLAME_ID) != NULL;
4323         }
4324 }
4325
4326 static bool
4327 match_blame_header(const char *name, char **line)
4328 {
4329         size_t namelen = strlen(name);
4330         bool matched = !strncmp(name, *line, namelen);
4331
4332         if (matched)
4333                 *line += namelen;
4334
4335         return matched;
4336 }
4337
4338 static bool
4339 blame_read(struct view *view, char *line)
4340 {
4341         static struct blame_commit *commit = NULL;
4342         static int blamed = 0;
4343         static time_t author_time;
4344         static bool read_file = TRUE;
4345
4346         if (read_file)
4347                 return blame_read_file(view, line, &read_file);
4348
4349         if (!line) {
4350                 /* Reset all! */
4351                 commit = NULL;
4352                 blamed = 0;
4353                 read_file = TRUE;
4354                 string_format(view->ref, "%s", view->vid);
4355                 if (view_is_displayed(view)) {
4356                         update_view_title(view);
4357                         redraw_view_from(view, 0);
4358                 }
4359                 return TRUE;
4360         }
4361
4362         if (!commit) {
4363                 commit = parse_blame_commit(view, line, &blamed);
4364                 string_format(view->ref, "%s %2d%%", view->vid,
4365                               view->lines ? blamed * 100 / view->lines : 0);
4366
4367         } else if (match_blame_header("author ", &line)) {
4368                 string_ncopy(commit->author, line, strlen(line));
4369
4370         } else if (match_blame_header("author-time ", &line)) {
4371                 author_time = (time_t) atol(line);
4372
4373         } else if (match_blame_header("author-tz ", &line)) {
4374                 parse_timezone(&author_time, line);
4375                 gmtime_r(&author_time, &commit->time);
4376
4377         } else if (match_blame_header("summary ", &line)) {
4378                 string_ncopy(commit->title, line, strlen(line));
4379
4380         } else if (match_blame_header("previous ", &line)) {
4381                 commit->has_previous = TRUE;
4382
4383         } else if (match_blame_header("filename ", &line)) {
4384                 string_ncopy(commit->filename, line, strlen(line));
4385                 commit = NULL;
4386         }
4387
4388         return TRUE;
4389 }
4390
4391 static bool
4392 blame_draw(struct view *view, struct line *line, unsigned int lineno)
4393 {
4394         struct blame *blame = line->data;
4395         struct tm *time = NULL;
4396         const char *id = NULL, *author = NULL;
4397         char text[SIZEOF_STR];
4398
4399         if (blame->commit && *blame->commit->filename) {
4400                 id = blame->commit->id;
4401                 author = blame->commit->author;
4402                 time = &blame->commit->time;
4403         }
4404
4405         if (opt_date && draw_date(view, time))
4406                 return TRUE;
4407
4408         if (opt_author && draw_author(view, author))
4409                 return TRUE;
4410
4411         if (draw_field(view, LINE_BLAME_ID, id, ID_COLS, FALSE))
4412                 return TRUE;
4413
4414         if (draw_lineno(view, lineno))
4415                 return TRUE;
4416
4417         string_expand(text, sizeof(text), blame->text, opt_tab_size);
4418         draw_text(view, LINE_DEFAULT, text, TRUE);
4419         return TRUE;
4420 }
4421
4422 static bool
4423 check_blame_commit(struct blame *blame, bool check_null_id)
4424 {
4425         if (!blame->commit)
4426                 report("Commit data not loaded yet");
4427         else if (check_null_id && !strcmp(blame->commit->id, NULL_ID))
4428                 report("No commit exist for the selected line");
4429         else
4430                 return TRUE;
4431         return FALSE;
4432 }
4433
4434 static void
4435 setup_blame_parent_line(struct view *view, struct blame *blame)
4436 {
4437         const char *diff_tree_argv[] = {
4438                 "git", "diff-tree", "-U0", blame->commit->id,
4439                         "--", blame->commit->filename, NULL
4440         };
4441         struct io io = {};
4442         int parent_lineno = -1;
4443         int blamed_lineno = -1;
4444         char *line;
4445
4446         if (!run_io(&io, diff_tree_argv, NULL, IO_RD))
4447                 return;
4448
4449         while ((line = io_get(&io, '\n', TRUE))) {
4450                 if (*line == '@') {
4451                         char *pos = strchr(line, '+');
4452
4453                         parent_lineno = atoi(line + 4);
4454                         if (pos)
4455                                 blamed_lineno = atoi(pos + 1);
4456
4457                 } else if (*line == '+' && parent_lineno != -1) {
4458                         if (blame->lineno == blamed_lineno - 1 &&
4459                             !strcmp(blame->text, line + 1)) {
4460                                 view->lineno = parent_lineno ? parent_lineno - 1 : 0;
4461                                 break;
4462                         }
4463                         blamed_lineno++;
4464                 }
4465         }
4466
4467         done_io(&io);
4468 }
4469
4470 static enum request
4471 blame_request(struct view *view, enum request request, struct line *line)
4472 {
4473         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
4474         struct blame *blame = line->data;
4475
4476         switch (request) {
4477         case REQ_VIEW_BLAME:
4478                 if (check_blame_commit(blame, TRUE)) {
4479                         string_copy(opt_ref, blame->commit->id);
4480                         string_copy(opt_file, blame->commit->filename);
4481                         if (blame->lineno)
4482                                 view->lineno = blame->lineno;
4483                         open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4484                 }
4485                 break;
4486
4487         case REQ_PARENT:
4488                 if (check_blame_commit(blame, TRUE) &&
4489                     select_commit_parent(blame->commit->id, opt_ref,
4490                                          blame->commit->filename)) {
4491                         string_copy(opt_file, blame->commit->filename);
4492                         setup_blame_parent_line(view, blame);
4493                         open_view(view, REQ_VIEW_BLAME, OPEN_REFRESH);
4494                 }
4495                 break;
4496
4497         case REQ_ENTER:
4498                 if (!check_blame_commit(blame, FALSE))
4499                         break;
4500
4501                 if (view_is_displayed(VIEW(REQ_VIEW_DIFF)) &&
4502                     !strcmp(blame->commit->id, VIEW(REQ_VIEW_DIFF)->ref))
4503                         break;
4504
4505                 if (!strcmp(blame->commit->id, NULL_ID)) {
4506                         struct view *diff = VIEW(REQ_VIEW_DIFF);
4507                         const char *diff_index_argv[] = {
4508                                 "git", "diff-index", "--root", "--patch-with-stat",
4509                                         "-C", "-M", "HEAD", "--", view->vid, NULL
4510                         };
4511
4512                         if (!blame->commit->has_previous) {
4513                                 diff_index_argv[1] = "diff";
4514                                 diff_index_argv[2] = "--no-color";
4515                                 diff_index_argv[6] = "--";
4516                                 diff_index_argv[7] = "/dev/null";
4517                         }
4518
4519                         if (!prepare_update(diff, diff_index_argv, NULL, FORMAT_DASH)) {
4520                                 report("Failed to allocate diff command");
4521                                 break;
4522                         }
4523                         flags |= OPEN_PREPARED;
4524                 }
4525
4526                 open_view(view, REQ_VIEW_DIFF, flags);
4527                 if (VIEW(REQ_VIEW_DIFF)->pipe && !strcmp(blame->commit->id, NULL_ID))
4528                         string_copy_rev(VIEW(REQ_VIEW_DIFF)->ref, NULL_ID);
4529                 break;
4530
4531         default:
4532                 return request;
4533         }
4534
4535         return REQ_NONE;
4536 }
4537
4538 static bool
4539 blame_grep(struct view *view, struct line *line)
4540 {
4541         struct blame *blame = line->data;
4542         struct blame_commit *commit = blame->commit;
4543         regmatch_t pmatch;
4544
4545 #define MATCH(text, on)                                                 \
4546         (on && *text && regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
4547
4548         if (commit) {
4549                 char buf[DATE_COLS + 1];
4550
4551                 if (MATCH(commit->title, 1) ||
4552                     MATCH(commit->author, opt_author) ||
4553                     MATCH(commit->id, opt_date))
4554                         return TRUE;
4555
4556                 if (strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time) &&
4557                     MATCH(buf, 1))
4558                         return TRUE;
4559         }
4560
4561         return MATCH(blame->text, 1);
4562
4563 #undef MATCH
4564 }
4565
4566 static void
4567 blame_select(struct view *view, struct line *line)
4568 {
4569         struct blame *blame = line->data;
4570         struct blame_commit *commit = blame->commit;
4571
4572         if (!commit)
4573                 return;
4574
4575         if (!strcmp(commit->id, NULL_ID))
4576                 string_ncopy(ref_commit, "HEAD", 4);
4577         else
4578                 string_copy_rev(ref_commit, commit->id);
4579 }
4580
4581 static struct view_ops blame_ops = {
4582         "line",
4583         NULL,
4584         blame_open,
4585         blame_read,
4586         blame_draw,
4587         blame_request,
4588         blame_grep,
4589         blame_select,
4590 };
4591
4592 /*
4593  * Status backend
4594  */
4595
4596 struct status {
4597         char status;
4598         struct {
4599                 mode_t mode;
4600                 char rev[SIZEOF_REV];
4601                 char name[SIZEOF_STR];
4602         } old;
4603         struct {
4604                 mode_t mode;
4605                 char rev[SIZEOF_REV];
4606                 char name[SIZEOF_STR];
4607         } new;
4608 };
4609
4610 static char status_onbranch[SIZEOF_STR];
4611 static struct status stage_status;
4612 static enum line_type stage_line_type;
4613 static size_t stage_chunks;
4614 static int *stage_chunk;
4615
4616 /* This should work even for the "On branch" line. */
4617 static inline bool
4618 status_has_none(struct view *view, struct line *line)
4619 {
4620         return line < view->line + view->lines && !line[1].data;
4621 }
4622
4623 /* Get fields from the diff line:
4624  * :100644 100644 06a5d6ae9eca55be2e0e585a152e6b1336f2b20e 0000000000000000000000000000000000000000 M
4625  */
4626 static inline bool
4627 status_get_diff(struct status *file, const char *buf, size_t bufsize)
4628 {
4629         const char *old_mode = buf +  1;
4630         const char *new_mode = buf +  8;
4631         const char *old_rev  = buf + 15;
4632         const char *new_rev  = buf + 56;
4633         const char *status   = buf + 97;
4634
4635         if (bufsize < 98 ||
4636             old_mode[-1] != ':' ||
4637             new_mode[-1] != ' ' ||
4638             old_rev[-1]  != ' ' ||
4639             new_rev[-1]  != ' ' ||
4640             status[-1]   != ' ')
4641                 return FALSE;
4642
4643         file->status = *status;
4644
4645         string_copy_rev(file->old.rev, old_rev);
4646         string_copy_rev(file->new.rev, new_rev);
4647
4648         file->old.mode = strtoul(old_mode, NULL, 8);
4649         file->new.mode = strtoul(new_mode, NULL, 8);
4650
4651         file->old.name[0] = file->new.name[0] = 0;
4652
4653         return TRUE;
4654 }
4655
4656 static bool
4657 status_run(struct view *view, const char *argv[], char status, enum line_type type)
4658 {
4659         struct status *unmerged = NULL;
4660         char *buf;
4661         struct io io = {};
4662
4663         if (!run_io(&io, argv, NULL, IO_RD))
4664                 return FALSE;
4665
4666         add_line_data(view, NULL, type);
4667
4668         while ((buf = io_get(&io, 0, TRUE))) {
4669                 struct status *file = unmerged;
4670
4671                 if (!file) {
4672                         file = calloc(1, sizeof(*file));
4673                         if (!file || !add_line_data(view, file, type))
4674                                 goto error_out;
4675                 }
4676
4677                 /* Parse diff info part. */
4678                 if (status) {
4679                         file->status = status;
4680                         if (status == 'A')
4681                                 string_copy(file->old.rev, NULL_ID);
4682
4683                 } else if (!file->status || file == unmerged) {
4684                         if (!status_get_diff(file, buf, strlen(buf)))
4685                                 goto error_out;
4686
4687                         buf = io_get(&io, 0, TRUE);
4688                         if (!buf)
4689                                 break;
4690
4691                         /* Collapse all modified entries that follow an
4692                          * associated unmerged entry. */
4693                         if (unmerged == file) {
4694                                 unmerged->status = 'U';
4695                                 unmerged = NULL;
4696                         } else if (file->status == 'U') {
4697                                 unmerged = file;
4698                         }
4699                 }
4700
4701                 /* Grab the old name for rename/copy. */
4702                 if (!*file->old.name &&
4703                     (file->status == 'R' || file->status == 'C')) {
4704                         string_ncopy(file->old.name, buf, strlen(buf));
4705
4706                         buf = io_get(&io, 0, TRUE);
4707                         if (!buf)
4708                                 break;
4709                 }
4710
4711                 /* git-ls-files just delivers a NUL separated list of
4712                  * file names similar to the second half of the
4713                  * git-diff-* output. */
4714                 string_ncopy(file->new.name, buf, strlen(buf));
4715                 if (!*file->old.name)
4716                         string_copy(file->old.name, file->new.name);
4717                 file = NULL;
4718         }
4719
4720         if (io_error(&io)) {
4721 error_out:
4722                 done_io(&io);
4723                 return FALSE;
4724         }
4725
4726         if (!view->line[view->lines - 1].data)
4727                 add_line_data(view, NULL, LINE_STAT_NONE);
4728
4729         done_io(&io);
4730         return TRUE;
4731 }
4732
4733 /* Don't show unmerged entries in the staged section. */
4734 static const char *status_diff_index_argv[] = {
4735         "git", "diff-index", "-z", "--diff-filter=ACDMRTXB",
4736                              "--cached", "-M", "HEAD", NULL
4737 };
4738
4739 static const char *status_diff_files_argv[] = {
4740         "git", "diff-files", "-z", NULL
4741 };
4742
4743 static const char *status_list_other_argv[] = {
4744         "git", "ls-files", "-z", "--others", "--exclude-standard", NULL
4745 };
4746
4747 static const char *status_list_no_head_argv[] = {
4748         "git", "ls-files", "-z", "--cached", "--exclude-standard", NULL
4749 };
4750
4751 static const char *update_index_argv[] = {
4752         "git", "update-index", "-q", "--unmerged", "--refresh", NULL
4753 };
4754
4755 /* Restore the previous line number to stay in the context or select a
4756  * line with something that can be updated. */
4757 static void
4758 status_restore(struct view *view)
4759 {
4760         if (view->p_lineno >= view->lines)
4761                 view->p_lineno = view->lines - 1;
4762         while (view->p_lineno < view->lines && !view->line[view->p_lineno].data)
4763                 view->p_lineno++;
4764         while (view->p_lineno > 0 && !view->line[view->p_lineno].data)
4765                 view->p_lineno--;
4766
4767         /* If the above fails, always skip the "On branch" line. */
4768         if (view->p_lineno < view->lines)
4769                 view->lineno = view->p_lineno;
4770         else
4771                 view->lineno = 1;
4772
4773         if (view->lineno < view->offset)
4774                 view->offset = view->lineno;
4775         else if (view->offset + view->height <= view->lineno)
4776                 view->offset = view->lineno - view->height + 1;
4777
4778         view->p_restore = FALSE;
4779 }
4780
4781 static void
4782 status_update_onbranch(void)
4783 {
4784         static const char *paths[][2] = {
4785                 { "rebase-apply/rebasing",      "Rebasing" },
4786                 { "rebase-apply/applying",      "Applying mailbox" },
4787                 { "rebase-apply/",              "Rebasing mailbox" },
4788                 { "rebase-merge/interactive",   "Interactive rebase" },
4789                 { "rebase-merge/",              "Rebase merge" },
4790                 { "MERGE_HEAD",                 "Merging" },
4791                 { "BISECT_LOG",                 "Bisecting" },
4792                 { "HEAD",                       "On branch" },
4793         };
4794         char buf[SIZEOF_STR];
4795         struct stat stat;
4796         int i;
4797
4798         if (is_initial_commit()) {
4799                 string_copy(status_onbranch, "Initial commit");
4800                 return;
4801         }
4802
4803         for (i = 0; i < ARRAY_SIZE(paths); i++) {
4804                 char *head = opt_head;
4805
4806                 if (!string_format(buf, "%s/%s", opt_git_dir, paths[i][0]) ||
4807                     lstat(buf, &stat) < 0)
4808                         continue;
4809
4810                 if (!*opt_head) {
4811                         struct io io = {};
4812
4813                         if (string_format(buf, "%s/rebase-merge/head-name", opt_git_dir) &&
4814                             io_open(&io, buf) &&
4815                             io_read_buf(&io, buf, sizeof(buf))) {
4816                                 head = chomp_string(buf);
4817                                 if (!prefixcmp(head, "refs/heads/"))
4818                                         head += STRING_SIZE("refs/heads/");
4819                         }
4820                 }
4821
4822                 if (!string_format(status_onbranch, "%s %s", paths[i][1], head))
4823                         string_copy(status_onbranch, opt_head);
4824                 return;
4825         }
4826
4827         string_copy(status_onbranch, "Not currently on any branch");
4828 }
4829
4830 /* First parse staged info using git-diff-index(1), then parse unstaged
4831  * info using git-diff-files(1), and finally untracked files using
4832  * git-ls-files(1). */
4833 static bool
4834 status_open(struct view *view)
4835 {
4836         reset_view(view);
4837
4838         add_line_data(view, NULL, LINE_STAT_HEAD);
4839         status_update_onbranch();
4840
4841         run_io_bg(update_index_argv);
4842
4843         if (is_initial_commit()) {
4844                 if (!status_run(view, status_list_no_head_argv, 'A', LINE_STAT_STAGED))
4845                         return FALSE;
4846         } else if (!status_run(view, status_diff_index_argv, 0, LINE_STAT_STAGED)) {
4847                 return FALSE;
4848         }
4849
4850         if (!status_run(view, status_diff_files_argv, 0, LINE_STAT_UNSTAGED) ||
4851             !status_run(view, status_list_other_argv, '?', LINE_STAT_UNTRACKED))
4852                 return FALSE;
4853
4854         /* Restore the exact position or use the specialized restore
4855          * mode? */
4856         if (!view->p_restore)
4857                 status_restore(view);
4858         return TRUE;
4859 }
4860
4861 static bool
4862 status_draw(struct view *view, struct line *line, unsigned int lineno)
4863 {
4864         struct status *status = line->data;
4865         enum line_type type;
4866         const char *text;
4867
4868         if (!status) {
4869                 switch (line->type) {
4870                 case LINE_STAT_STAGED:
4871                         type = LINE_STAT_SECTION;
4872                         text = "Changes to be committed:";
4873                         break;
4874
4875                 case LINE_STAT_UNSTAGED:
4876                         type = LINE_STAT_SECTION;
4877                         text = "Changed but not updated:";
4878                         break;
4879
4880                 case LINE_STAT_UNTRACKED:
4881                         type = LINE_STAT_SECTION;
4882                         text = "Untracked files:";
4883                         break;
4884
4885                 case LINE_STAT_NONE:
4886                         type = LINE_DEFAULT;
4887                         text = "  (no files)";
4888                         break;
4889
4890                 case LINE_STAT_HEAD:
4891                         type = LINE_STAT_HEAD;
4892                         text = status_onbranch;
4893                         break;
4894
4895                 default:
4896                         return FALSE;
4897                 }
4898         } else {
4899                 static char buf[] = { '?', ' ', ' ', ' ', 0 };
4900
4901                 buf[0] = status->status;
4902                 if (draw_text(view, line->type, buf, TRUE))
4903                         return TRUE;
4904                 type = LINE_DEFAULT;
4905                 text = status->new.name;
4906         }
4907
4908         draw_text(view, type, text, TRUE);
4909         return TRUE;
4910 }
4911
4912 static enum request
4913 status_load_error(struct view *view, struct view *stage, const char *path)
4914 {
4915         if (displayed_views() == 2 || display[current_view] != view)
4916                 maximize_view(view);
4917         report("Failed to load '%s': %s", path, io_strerror(&stage->io));
4918         return REQ_NONE;
4919 }
4920
4921 static enum request
4922 status_enter(struct view *view, struct line *line)
4923 {
4924         struct status *status = line->data;
4925         const char *oldpath = status ? status->old.name : NULL;
4926         /* Diffs for unmerged entries are empty when passing the new
4927          * path, so leave it empty. */
4928         const char *newpath = status && status->status != 'U' ? status->new.name : NULL;
4929         const char *info;
4930         enum open_flags split;
4931         struct view *stage = VIEW(REQ_VIEW_STAGE);
4932
4933         if (line->type == LINE_STAT_NONE ||
4934             (!status && line[1].type == LINE_STAT_NONE)) {
4935                 report("No file to diff");
4936                 return REQ_NONE;
4937         }
4938
4939         switch (line->type) {
4940         case LINE_STAT_STAGED:
4941                 if (is_initial_commit()) {
4942                         const char *no_head_diff_argv[] = {
4943                                 "git", "diff", "--no-color", "--patch-with-stat",
4944                                         "--", "/dev/null", newpath, NULL
4945                         };
4946
4947                         if (!prepare_update(stage, no_head_diff_argv, opt_cdup, FORMAT_DASH))
4948                                 return status_load_error(view, stage, newpath);
4949                 } else {
4950                         const char *index_show_argv[] = {
4951                                 "git", "diff-index", "--root", "--patch-with-stat",
4952                                         "-C", "-M", "--cached", "HEAD", "--",
4953                                         oldpath, newpath, NULL
4954                         };
4955
4956                         if (!prepare_update(stage, index_show_argv, opt_cdup, FORMAT_DASH))
4957                                 return status_load_error(view, stage, newpath);
4958                 }
4959
4960                 if (status)
4961                         info = "Staged changes to %s";
4962                 else
4963                         info = "Staged changes";
4964                 break;
4965
4966         case LINE_STAT_UNSTAGED:
4967         {
4968                 const char *files_show_argv[] = {
4969                         "git", "diff-files", "--root", "--patch-with-stat",
4970                                 "-C", "-M", "--", oldpath, newpath, NULL
4971                 };
4972
4973                 if (!prepare_update(stage, files_show_argv, opt_cdup, FORMAT_DASH))
4974                         return status_load_error(view, stage, newpath);
4975                 if (status)
4976                         info = "Unstaged changes to %s";
4977                 else
4978                         info = "Unstaged changes";
4979                 break;
4980         }
4981         case LINE_STAT_UNTRACKED:
4982                 if (!newpath) {
4983                         report("No file to show");
4984                         return REQ_NONE;
4985                 }
4986
4987                 if (!suffixcmp(status->new.name, -1, "/")) {
4988                         report("Cannot display a directory");
4989                         return REQ_NONE;
4990                 }
4991
4992                 if (!prepare_update_file(stage, newpath))
4993                         return status_load_error(view, stage, newpath);
4994                 info = "Untracked file %s";
4995                 break;
4996
4997         case LINE_STAT_HEAD:
4998                 return REQ_NONE;
4999
5000         default:
5001                 die("line type %d not handled in switch", line->type);
5002         }
5003
5004         split = view_is_displayed(view) ? OPEN_SPLIT : 0;
5005         open_view(view, REQ_VIEW_STAGE, OPEN_PREPARED | split);
5006         if (view_is_displayed(VIEW(REQ_VIEW_STAGE))) {
5007                 if (status) {
5008                         stage_status = *status;
5009                 } else {
5010                         memset(&stage_status, 0, sizeof(stage_status));
5011                 }
5012
5013                 stage_line_type = line->type;
5014                 stage_chunks = 0;
5015                 string_format(VIEW(REQ_VIEW_STAGE)->ref, info, stage_status.new.name);
5016         }
5017
5018         return REQ_NONE;
5019 }
5020
5021 static bool
5022 status_exists(struct status *status, enum line_type type)
5023 {
5024         struct view *view = VIEW(REQ_VIEW_STATUS);
5025         unsigned long lineno;
5026
5027         for (lineno = 0; lineno < view->lines; lineno++) {
5028                 struct line *line = &view->line[lineno];
5029                 struct status *pos = line->data;
5030
5031                 if (line->type != type)
5032                         continue;
5033                 if (!pos && (!status || !status->status) && line[1].data) {
5034                         select_view_line(view, lineno);
5035                         return TRUE;
5036                 }
5037                 if (pos && !strcmp(status->new.name, pos->new.name)) {
5038                         select_view_line(view, lineno);
5039                         return TRUE;
5040                 }
5041         }
5042
5043         return FALSE;
5044 }
5045
5046
5047 static bool
5048 status_update_prepare(struct io *io, enum line_type type)
5049 {
5050         const char *staged_argv[] = {
5051                 "git", "update-index", "-z", "--index-info", NULL
5052         };
5053         const char *others_argv[] = {
5054                 "git", "update-index", "-z", "--add", "--remove", "--stdin", NULL
5055         };
5056
5057         switch (type) {
5058         case LINE_STAT_STAGED:
5059                 return run_io(io, staged_argv, opt_cdup, IO_WR);
5060
5061         case LINE_STAT_UNSTAGED:
5062                 return run_io(io, others_argv, opt_cdup, IO_WR);
5063
5064         case LINE_STAT_UNTRACKED:
5065                 return run_io(io, others_argv, NULL, IO_WR);
5066
5067         default:
5068                 die("line type %d not handled in switch", type);
5069                 return FALSE;
5070         }
5071 }
5072
5073 static bool
5074 status_update_write(struct io *io, struct status *status, enum line_type type)
5075 {
5076         char buf[SIZEOF_STR];
5077         size_t bufsize = 0;
5078
5079         switch (type) {
5080         case LINE_STAT_STAGED:
5081                 if (!string_format_from(buf, &bufsize, "%06o %s\t%s%c",
5082                                         status->old.mode,
5083                                         status->old.rev,
5084                                         status->old.name, 0))
5085                         return FALSE;
5086                 break;
5087
5088         case LINE_STAT_UNSTAGED:
5089         case LINE_STAT_UNTRACKED:
5090                 if (!string_format_from(buf, &bufsize, "%s%c", status->new.name, 0))
5091                         return FALSE;
5092                 break;
5093
5094         default:
5095                 die("line type %d not handled in switch", type);
5096         }
5097
5098         return io_write(io, buf, bufsize);
5099 }
5100
5101 static bool
5102 status_update_file(struct status *status, enum line_type type)
5103 {
5104         struct io io = {};
5105         bool result;
5106
5107         if (!status_update_prepare(&io, type))
5108                 return FALSE;
5109
5110         result = status_update_write(&io, status, type);
5111         return done_io(&io) && result;
5112 }
5113
5114 static bool
5115 status_update_files(struct view *view, struct line *line)
5116 {
5117         char buf[sizeof(view->ref)];
5118         struct io io = {};
5119         bool result = TRUE;
5120         struct line *pos = view->line + view->lines;
5121         int files = 0;
5122         int file, done;
5123         int cursor_y, cursor_x;
5124
5125         if (!status_update_prepare(&io, line->type))
5126                 return FALSE;
5127
5128         for (pos = line; pos < view->line + view->lines && pos->data; pos++)
5129                 files++;
5130
5131         string_copy(buf, view->ref);
5132         getsyx(cursor_y, cursor_x);
5133         for (file = 0, done = 5; result && file < files; line++, file++) {
5134                 int almost_done = file * 100 / files;
5135
5136                 if (almost_done > done) {
5137                         done = almost_done;
5138                         string_format(view->ref, "updating file %u of %u (%d%% done)",
5139                                       file, files, done);
5140                         update_view_title(view);
5141                         setsyx(cursor_y, cursor_x);
5142                         doupdate();
5143                 }
5144                 result = status_update_write(&io, line->data, line->type);
5145         }
5146         string_copy(view->ref, buf);
5147
5148         return done_io(&io) && result;
5149 }
5150
5151 static bool
5152 status_update(struct view *view)
5153 {
5154         struct line *line = &view->line[view->lineno];
5155
5156         assert(view->lines);
5157
5158         if (!line->data) {
5159                 /* This should work even for the "On branch" line. */
5160                 if (line < view->line + view->lines && !line[1].data) {
5161                         report("Nothing to update");
5162                         return FALSE;
5163                 }
5164
5165                 if (!status_update_files(view, line + 1)) {
5166                         report("Failed to update file status");
5167                         return FALSE;
5168                 }
5169
5170         } else if (!status_update_file(line->data, line->type)) {
5171                 report("Failed to update file status");
5172                 return FALSE;
5173         }
5174
5175         return TRUE;
5176 }
5177
5178 static bool
5179 status_revert(struct status *status, enum line_type type, bool has_none)
5180 {
5181         if (!status || type != LINE_STAT_UNSTAGED) {
5182                 if (type == LINE_STAT_STAGED) {
5183                         report("Cannot revert changes to staged files");
5184                 } else if (type == LINE_STAT_UNTRACKED) {
5185                         report("Cannot revert changes to untracked files");
5186                 } else if (has_none) {
5187                         report("Nothing to revert");
5188                 } else {
5189                         report("Cannot revert changes to multiple files");
5190                 }
5191                 return FALSE;
5192
5193         } else {
5194                 char mode[10] = "100644";
5195                 const char *reset_argv[] = {
5196                         "git", "update-index", "--cacheinfo", mode,
5197                                 status->old.rev, status->old.name, NULL
5198                 };
5199                 const char *checkout_argv[] = {
5200                         "git", "checkout", "--", status->old.name, NULL
5201                 };
5202
5203                 if (!prompt_yesno("Are you sure you want to overwrite any changes?"))
5204                         return FALSE;
5205                 string_format(mode, "%o", status->old.mode);
5206                 return (status->status != 'U' || run_io_fg(reset_argv, opt_cdup)) &&
5207                         run_io_fg(checkout_argv, opt_cdup);
5208         }
5209 }
5210
5211 static enum request
5212 status_request(struct view *view, enum request request, struct line *line)
5213 {
5214         struct status *status = line->data;
5215
5216         switch (request) {
5217         case REQ_STATUS_UPDATE:
5218                 if (!status_update(view))
5219                         return REQ_NONE;
5220                 break;
5221
5222         case REQ_STATUS_REVERT:
5223                 if (!status_revert(status, line->type, status_has_none(view, line)))
5224                         return REQ_NONE;
5225                 break;
5226
5227         case REQ_STATUS_MERGE:
5228                 if (!status || status->status != 'U') {
5229                         report("Merging only possible for files with unmerged status ('U').");
5230                         return REQ_NONE;
5231                 }
5232                 open_mergetool(status->new.name);
5233                 break;
5234
5235         case REQ_EDIT:
5236                 if (!status)
5237                         return request;
5238                 if (status->status == 'D') {
5239                         report("File has been deleted.");
5240                         return REQ_NONE;
5241                 }
5242
5243                 open_editor(status->status != '?', status->new.name);
5244                 break;
5245
5246         case REQ_VIEW_BLAME:
5247                 if (status) {
5248                         string_copy(opt_file, status->new.name);
5249                         opt_ref[0] = 0;
5250                 }
5251                 return request;
5252
5253         case REQ_ENTER:
5254                 /* After returning the status view has been split to
5255                  * show the stage view. No further reloading is
5256                  * necessary. */
5257                 return status_enter(view, line);
5258
5259         case REQ_REFRESH:
5260                 /* Simply reload the view. */
5261                 break;
5262
5263         default:
5264                 return request;
5265         }
5266
5267         open_view(view, REQ_VIEW_STATUS, OPEN_RELOAD);
5268
5269         return REQ_NONE;
5270 }
5271
5272 static void
5273 status_select(struct view *view, struct line *line)
5274 {
5275         struct status *status = line->data;
5276         char file[SIZEOF_STR] = "all files";
5277         const char *text;
5278         const char *key;
5279
5280         if (status && !string_format(file, "'%s'", status->new.name))
5281                 return;
5282
5283         if (!status && line[1].type == LINE_STAT_NONE)
5284                 line++;
5285
5286         switch (line->type) {
5287         case LINE_STAT_STAGED:
5288                 text = "Press %s to unstage %s for commit";
5289                 break;
5290
5291         case LINE_STAT_UNSTAGED:
5292                 text = "Press %s to stage %s for commit";
5293                 break;
5294
5295         case LINE_STAT_UNTRACKED:
5296                 text = "Press %s to stage %s for addition";
5297                 break;
5298
5299         case LINE_STAT_HEAD:
5300         case LINE_STAT_NONE:
5301                 text = "Nothing to update";
5302                 break;
5303
5304         default:
5305                 die("line type %d not handled in switch", line->type);
5306         }
5307
5308         if (status && status->status == 'U') {
5309                 text = "Press %s to resolve conflict in %s";
5310                 key = get_key(REQ_STATUS_MERGE);
5311
5312         } else {
5313                 key = get_key(REQ_STATUS_UPDATE);
5314         }
5315
5316         string_format(view->ref, text, key, file);
5317 }
5318
5319 static bool
5320 status_grep(struct view *view, struct line *line)
5321 {
5322         struct status *status = line->data;
5323         enum { S_STATUS, S_NAME, S_END } state;
5324         char buf[2] = "?";
5325         regmatch_t pmatch;
5326
5327         if (!status)
5328                 return FALSE;
5329
5330         for (state = S_STATUS; state < S_END; state++) {
5331                 const char *text;
5332
5333                 switch (state) {
5334                 case S_NAME:    text = status->new.name;        break;
5335                 case S_STATUS:
5336                         buf[0] = status->status;
5337                         text = buf;
5338                         break;
5339
5340                 default:
5341                         return FALSE;
5342                 }
5343
5344                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
5345                         return TRUE;
5346         }
5347
5348         return FALSE;
5349 }
5350
5351 static struct view_ops status_ops = {
5352         "file",
5353         NULL,
5354         status_open,
5355         NULL,
5356         status_draw,
5357         status_request,
5358         status_grep,
5359         status_select,
5360 };
5361
5362
5363 static bool
5364 stage_diff_write(struct io *io, struct line *line, struct line *end)
5365 {
5366         while (line < end) {
5367                 if (!io_write(io, line->data, strlen(line->data)) ||
5368                     !io_write(io, "\n", 1))
5369                         return FALSE;
5370                 line++;
5371                 if (line->type == LINE_DIFF_CHUNK ||
5372                     line->type == LINE_DIFF_HEADER)
5373                         break;
5374         }
5375
5376         return TRUE;
5377 }
5378
5379 static struct line *
5380 stage_diff_find(struct view *view, struct line *line, enum line_type type)
5381 {
5382         for (; view->line < line; line--)
5383                 if (line->type == type)
5384                         return line;
5385
5386         return NULL;
5387 }
5388
5389 static bool
5390 stage_apply_chunk(struct view *view, struct line *chunk, bool revert)
5391 {
5392         const char *apply_argv[SIZEOF_ARG] = {
5393                 "git", "apply", "--whitespace=nowarn", NULL
5394         };
5395         struct line *diff_hdr;
5396         struct io io = {};
5397         int argc = 3;
5398
5399         diff_hdr = stage_diff_find(view, chunk, LINE_DIFF_HEADER);
5400         if (!diff_hdr)
5401                 return FALSE;
5402
5403         if (!revert)
5404                 apply_argv[argc++] = "--cached";
5405         if (revert || stage_line_type == LINE_STAT_STAGED)
5406                 apply_argv[argc++] = "-R";
5407         apply_argv[argc++] = "-";
5408         apply_argv[argc++] = NULL;
5409         if (!run_io(&io, apply_argv, opt_cdup, IO_WR))
5410                 return FALSE;
5411
5412         if (!stage_diff_write(&io, diff_hdr, chunk) ||
5413             !stage_diff_write(&io, chunk, view->line + view->lines))
5414                 chunk = NULL;
5415
5416         done_io(&io);
5417         run_io_bg(update_index_argv);
5418
5419         return chunk ? TRUE : FALSE;
5420 }
5421
5422 static bool
5423 stage_update(struct view *view, struct line *line)
5424 {
5425         struct line *chunk = NULL;
5426
5427         if (!is_initial_commit() && stage_line_type != LINE_STAT_UNTRACKED)
5428                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5429
5430         if (chunk) {
5431                 if (!stage_apply_chunk(view, chunk, FALSE)) {
5432                         report("Failed to apply chunk");
5433                         return FALSE;
5434                 }
5435
5436         } else if (!stage_status.status) {
5437                 view = VIEW(REQ_VIEW_STATUS);
5438
5439                 for (line = view->line; line < view->line + view->lines; line++)
5440                         if (line->type == stage_line_type)
5441                                 break;
5442
5443                 if (!status_update_files(view, line + 1)) {
5444                         report("Failed to update files");
5445                         return FALSE;
5446                 }
5447
5448         } else if (!status_update_file(&stage_status, stage_line_type)) {
5449                 report("Failed to update file");
5450                 return FALSE;
5451         }
5452
5453         return TRUE;
5454 }
5455
5456 static bool
5457 stage_revert(struct view *view, struct line *line)
5458 {
5459         struct line *chunk = NULL;
5460
5461         if (!is_initial_commit() && stage_line_type == LINE_STAT_UNSTAGED)
5462                 chunk = stage_diff_find(view, line, LINE_DIFF_CHUNK);
5463
5464         if (chunk) {
5465                 if (!prompt_yesno("Are you sure you want to revert changes?"))
5466                         return FALSE;
5467
5468                 if (!stage_apply_chunk(view, chunk, TRUE)) {
5469                         report("Failed to revert chunk");
5470                         return FALSE;
5471                 }
5472                 return TRUE;
5473
5474         } else {
5475                 return status_revert(stage_status.status ? &stage_status : NULL,
5476                                      stage_line_type, FALSE);
5477         }
5478 }
5479
5480
5481 static void
5482 stage_next(struct view *view, struct line *line)
5483 {
5484         int i;
5485
5486         if (!stage_chunks) {
5487                 static size_t alloc = 0;
5488                 int *tmp;
5489
5490                 for (line = view->line; line < view->line + view->lines; line++) {
5491                         if (line->type != LINE_DIFF_CHUNK)
5492                                 continue;
5493
5494                         tmp = realloc_items(stage_chunk, &alloc,
5495                                             stage_chunks, sizeof(*tmp));
5496                         if (!tmp) {
5497                                 report("Allocation failure");
5498                                 return;
5499                         }
5500
5501                         stage_chunk = tmp;
5502                         stage_chunk[stage_chunks++] = line - view->line;
5503                 }
5504         }
5505
5506         for (i = 0; i < stage_chunks; i++) {
5507                 if (stage_chunk[i] > view->lineno) {
5508                         do_scroll_view(view, stage_chunk[i] - view->lineno);
5509                         report("Chunk %d of %d", i + 1, stage_chunks);
5510                         return;
5511                 }
5512         }
5513
5514         report("No next chunk found");
5515 }
5516
5517 static enum request
5518 stage_request(struct view *view, enum request request, struct line *line)
5519 {
5520         switch (request) {
5521         case REQ_STATUS_UPDATE:
5522                 if (!stage_update(view, line))
5523                         return REQ_NONE;
5524                 break;
5525
5526         case REQ_STATUS_REVERT:
5527                 if (!stage_revert(view, line))
5528                         return REQ_NONE;
5529                 break;
5530
5531         case REQ_STAGE_NEXT:
5532                 if (stage_line_type == LINE_STAT_UNTRACKED) {
5533                         report("File is untracked; press %s to add",
5534                                get_key(REQ_STATUS_UPDATE));
5535                         return REQ_NONE;
5536                 }
5537                 stage_next(view, line);
5538                 return REQ_NONE;
5539
5540         case REQ_EDIT:
5541                 if (!stage_status.new.name[0])
5542                         return request;
5543                 if (stage_status.status == 'D') {
5544                         report("File has been deleted.");
5545                         return REQ_NONE;
5546                 }
5547
5548                 open_editor(stage_status.status != '?', stage_status.new.name);
5549                 break;
5550
5551         case REQ_REFRESH:
5552                 /* Reload everything ... */
5553                 break;
5554
5555         case REQ_VIEW_BLAME:
5556                 if (stage_status.new.name[0]) {
5557                         string_copy(opt_file, stage_status.new.name);
5558                         opt_ref[0] = 0;
5559                 }
5560                 return request;
5561
5562         case REQ_ENTER:
5563                 return pager_request(view, request, line);
5564
5565         default:
5566                 return request;
5567         }
5568
5569         VIEW(REQ_VIEW_STATUS)->p_restore = TRUE;
5570         open_view(view, REQ_VIEW_STATUS, OPEN_REFRESH);
5571
5572         /* Check whether the staged entry still exists, and close the
5573          * stage view if it doesn't. */
5574         if (!status_exists(&stage_status, stage_line_type)) {
5575                 status_restore(VIEW(REQ_VIEW_STATUS));
5576                 return REQ_VIEW_CLOSE;
5577         }
5578
5579         if (stage_line_type == LINE_STAT_UNTRACKED) {
5580                 if (!suffixcmp(stage_status.new.name, -1, "/")) {
5581                         report("Cannot display a directory");
5582                         return REQ_NONE;
5583                 }
5584
5585                 if (!prepare_update_file(view, stage_status.new.name)) {
5586                         report("Failed to open file: %s", strerror(errno));
5587                         return REQ_NONE;
5588                 }
5589         }
5590         open_view(view, REQ_VIEW_STAGE, OPEN_REFRESH);
5591
5592         return REQ_NONE;
5593 }
5594
5595 static struct view_ops stage_ops = {
5596         "line",
5597         NULL,
5598         NULL,
5599         pager_read,
5600         pager_draw,
5601         stage_request,
5602         pager_grep,
5603         pager_select,
5604 };
5605
5606
5607 /*
5608  * Revision graph
5609  */
5610
5611 struct commit {
5612         char id[SIZEOF_REV];            /* SHA1 ID. */
5613         char title[128];                /* First line of the commit message. */
5614         char author[75];                /* Author of the commit. */
5615         struct tm time;                 /* Date from the author ident. */
5616         struct ref **refs;              /* Repository references. */
5617         chtype graph[SIZEOF_REVGRAPH];  /* Ancestry chain graphics. */
5618         size_t graph_size;              /* The width of the graph array. */
5619         bool has_parents;               /* Rewritten --parents seen. */
5620 };
5621
5622 /* Size of rev graph with no  "padding" columns */
5623 #define SIZEOF_REVITEMS (SIZEOF_REVGRAPH - (SIZEOF_REVGRAPH / 2))
5624
5625 struct rev_graph {
5626         struct rev_graph *prev, *next, *parents;
5627         char rev[SIZEOF_REVITEMS][SIZEOF_REV];
5628         size_t size;
5629         struct commit *commit;
5630         size_t pos;
5631         unsigned int boundary:1;
5632 };
5633
5634 /* Parents of the commit being visualized. */
5635 static struct rev_graph graph_parents[4];
5636
5637 /* The current stack of revisions on the graph. */
5638 static struct rev_graph graph_stacks[4] = {
5639         { &graph_stacks[3], &graph_stacks[1], &graph_parents[0] },
5640         { &graph_stacks[0], &graph_stacks[2], &graph_parents[1] },
5641         { &graph_stacks[1], &graph_stacks[3], &graph_parents[2] },
5642         { &graph_stacks[2], &graph_stacks[0], &graph_parents[3] },
5643 };
5644
5645 static inline bool
5646 graph_parent_is_merge(struct rev_graph *graph)
5647 {
5648         return graph->parents->size > 1;
5649 }
5650
5651 static inline void
5652 append_to_rev_graph(struct rev_graph *graph, chtype symbol)
5653 {
5654         struct commit *commit = graph->commit;
5655
5656         if (commit->graph_size < ARRAY_SIZE(commit->graph) - 1)
5657                 commit->graph[commit->graph_size++] = symbol;
5658 }
5659
5660 static void
5661 clear_rev_graph(struct rev_graph *graph)
5662 {
5663         graph->boundary = 0;
5664         graph->size = graph->pos = 0;
5665         graph->commit = NULL;
5666         memset(graph->parents, 0, sizeof(*graph->parents));
5667 }
5668
5669 static void
5670 done_rev_graph(struct rev_graph *graph)
5671 {
5672         if (graph_parent_is_merge(graph) &&
5673             graph->pos < graph->size - 1 &&
5674             graph->next->size == graph->size + graph->parents->size - 1) {
5675                 size_t i = graph->pos + graph->parents->size - 1;
5676
5677                 graph->commit->graph_size = i * 2;
5678                 while (i < graph->next->size - 1) {
5679                         append_to_rev_graph(graph, ' ');
5680                         append_to_rev_graph(graph, '\\');
5681                         i++;
5682                 }
5683         }
5684
5685         clear_rev_graph(graph);
5686 }
5687
5688 static void
5689 push_rev_graph(struct rev_graph *graph, const char *parent)
5690 {
5691         int i;
5692
5693         /* "Collapse" duplicate parents lines.
5694          *
5695          * FIXME: This needs to also update update the drawn graph but
5696          * for now it just serves as a method for pruning graph lines. */
5697         for (i = 0; i < graph->size; i++)
5698                 if (!strncmp(graph->rev[i], parent, SIZEOF_REV))
5699                         return;
5700
5701         if (graph->size < SIZEOF_REVITEMS) {
5702                 string_copy_rev(graph->rev[graph->size++], parent);
5703         }
5704 }
5705
5706 static chtype
5707 get_rev_graph_symbol(struct rev_graph *graph)
5708 {
5709         chtype symbol;
5710
5711         if (graph->boundary)
5712                 symbol = REVGRAPH_BOUND;
5713         else if (graph->parents->size == 0)
5714                 symbol = REVGRAPH_INIT;
5715         else if (graph_parent_is_merge(graph))
5716                 symbol = REVGRAPH_MERGE;
5717         else if (graph->pos >= graph->size)
5718                 symbol = REVGRAPH_BRANCH;
5719         else
5720                 symbol = REVGRAPH_COMMIT;
5721
5722         return symbol;
5723 }
5724
5725 static void
5726 draw_rev_graph(struct rev_graph *graph)
5727 {
5728         struct rev_filler {
5729                 chtype separator, line;
5730         };
5731         enum { DEFAULT, RSHARP, RDIAG, LDIAG };
5732         static struct rev_filler fillers[] = {
5733                 { ' ',  '|' },
5734                 { '`',  '.' },
5735                 { '\'', ' ' },
5736                 { '/',  ' ' },
5737         };
5738         chtype symbol = get_rev_graph_symbol(graph);
5739         struct rev_filler *filler;
5740         size_t i;
5741
5742         if (opt_line_graphics)
5743                 fillers[DEFAULT].line = line_graphics[LINE_GRAPHIC_VLINE];
5744
5745         filler = &fillers[DEFAULT];
5746
5747         for (i = 0; i < graph->pos; i++) {
5748                 append_to_rev_graph(graph, filler->line);
5749                 if (graph_parent_is_merge(graph->prev) &&
5750                     graph->prev->pos == i)
5751                         filler = &fillers[RSHARP];
5752
5753                 append_to_rev_graph(graph, filler->separator);
5754         }
5755
5756         /* Place the symbol for this revision. */
5757         append_to_rev_graph(graph, symbol);
5758
5759         if (graph->prev->size > graph->size)
5760                 filler = &fillers[RDIAG];
5761         else
5762                 filler = &fillers[DEFAULT];
5763
5764         i++;
5765
5766         for (; i < graph->size; i++) {
5767                 append_to_rev_graph(graph, filler->separator);
5768                 append_to_rev_graph(graph, filler->line);
5769                 if (graph_parent_is_merge(graph->prev) &&
5770                     i < graph->prev->pos + graph->parents->size)
5771                         filler = &fillers[RSHARP];
5772                 if (graph->prev->size > graph->size)
5773                         filler = &fillers[LDIAG];
5774         }
5775
5776         if (graph->prev->size > graph->size) {
5777                 append_to_rev_graph(graph, filler->separator);
5778                 if (filler->line != ' ')
5779                         append_to_rev_graph(graph, filler->line);
5780         }
5781 }
5782
5783 /* Prepare the next rev graph */
5784 static void
5785 prepare_rev_graph(struct rev_graph *graph)
5786 {
5787         size_t i;
5788
5789         /* First, traverse all lines of revisions up to the active one. */
5790         for (graph->pos = 0; graph->pos < graph->size; graph->pos++) {
5791                 if (!strcmp(graph->rev[graph->pos], graph->commit->id))
5792                         break;
5793
5794                 push_rev_graph(graph->next, graph->rev[graph->pos]);
5795         }
5796
5797         /* Interleave the new revision parent(s). */
5798         for (i = 0; !graph->boundary && i < graph->parents->size; i++)
5799                 push_rev_graph(graph->next, graph->parents->rev[i]);
5800
5801         /* Lastly, put any remaining revisions. */
5802         for (i = graph->pos + 1; i < graph->size; i++)
5803                 push_rev_graph(graph->next, graph->rev[i]);
5804 }
5805
5806 static void
5807 update_rev_graph(struct view *view, struct rev_graph *graph)
5808 {
5809         /* If this is the finalizing update ... */
5810         if (graph->commit)
5811                 prepare_rev_graph(graph);
5812
5813         /* Graph visualization needs a one rev look-ahead,
5814          * so the first update doesn't visualize anything. */
5815         if (!graph->prev->commit)
5816                 return;
5817
5818         if (view->lines > 2)
5819                 view->line[view->lines - 3].dirty = 1;
5820         if (view->lines > 1)
5821                 view->line[view->lines - 2].dirty = 1;
5822         draw_rev_graph(graph->prev);
5823         done_rev_graph(graph->prev->prev);
5824 }
5825
5826
5827 /*
5828  * Main view backend
5829  */
5830
5831 static const char *main_argv[SIZEOF_ARG] = {
5832         "git", "log", "--no-color", "--pretty=raw", "--parents",
5833                       "--topo-order", "%(head)", NULL
5834 };
5835
5836 static bool
5837 main_draw(struct view *view, struct line *line, unsigned int lineno)
5838 {
5839         struct commit *commit = line->data;
5840
5841         if (!*commit->author)
5842                 return FALSE;
5843
5844         if (opt_date && draw_date(view, &commit->time))
5845                 return TRUE;
5846
5847         if (opt_author && draw_author(view, commit->author))
5848                 return TRUE;
5849
5850         if (opt_rev_graph && commit->graph_size &&
5851             draw_graphic(view, LINE_MAIN_REVGRAPH, commit->graph, commit->graph_size))
5852                 return TRUE;
5853
5854         if (opt_show_refs && commit->refs) {
5855                 size_t i = 0;
5856
5857                 do {
5858                         enum line_type type;
5859
5860                         if (commit->refs[i]->head)
5861                                 type = LINE_MAIN_HEAD;
5862                         else if (commit->refs[i]->ltag)
5863                                 type = LINE_MAIN_LOCAL_TAG;
5864                         else if (commit->refs[i]->tag)
5865                                 type = LINE_MAIN_TAG;
5866                         else if (commit->refs[i]->tracked)
5867                                 type = LINE_MAIN_TRACKED;
5868                         else if (commit->refs[i]->remote)
5869                                 type = LINE_MAIN_REMOTE;
5870                         else
5871                                 type = LINE_MAIN_REF;
5872
5873                         if (draw_text(view, type, "[", TRUE) ||
5874                             draw_text(view, type, commit->refs[i]->name, TRUE) ||
5875                             draw_text(view, type, "]", TRUE))
5876                                 return TRUE;
5877
5878                         if (draw_text(view, LINE_DEFAULT, " ", TRUE))
5879                                 return TRUE;
5880                 } while (commit->refs[i++]->next);
5881         }
5882
5883         draw_text(view, LINE_DEFAULT, commit->title, TRUE);
5884         return TRUE;
5885 }
5886
5887 /* Reads git log --pretty=raw output and parses it into the commit struct. */
5888 static bool
5889 main_read(struct view *view, char *line)
5890 {
5891         static struct rev_graph *graph = graph_stacks;
5892         enum line_type type;
5893         struct commit *commit;
5894
5895         if (!line) {
5896                 int i;
5897
5898                 if (!view->lines && !view->parent)
5899                         die("No revisions match the given arguments.");
5900                 if (view->lines > 0) {
5901                         commit = view->line[view->lines - 1].data;
5902                         view->line[view->lines - 1].dirty = 1;
5903                         if (!*commit->author) {
5904                                 view->lines--;
5905                                 free(commit);
5906                                 graph->commit = NULL;
5907                         }
5908                 }
5909                 update_rev_graph(view, graph);
5910
5911                 for (i = 0; i < ARRAY_SIZE(graph_stacks); i++)
5912                         clear_rev_graph(&graph_stacks[i]);
5913                 return TRUE;
5914         }
5915
5916         type = get_line_type(line);
5917         if (type == LINE_COMMIT) {
5918                 commit = calloc(1, sizeof(struct commit));
5919                 if (!commit)
5920                         return FALSE;
5921
5922                 line += STRING_SIZE("commit ");
5923                 if (*line == '-') {
5924                         graph->boundary = 1;
5925                         line++;
5926                 }
5927
5928                 string_copy_rev(commit->id, line);
5929                 commit->refs = get_refs(commit->id);
5930                 graph->commit = commit;
5931                 add_line_data(view, commit, LINE_MAIN_COMMIT);
5932
5933                 while ((line = strchr(line, ' '))) {
5934                         line++;
5935                         push_rev_graph(graph->parents, line);
5936                         commit->has_parents = TRUE;
5937                 }
5938                 return TRUE;
5939         }
5940
5941         if (!view->lines)
5942                 return TRUE;
5943         commit = view->line[view->lines - 1].data;
5944
5945         switch (type) {
5946         case LINE_PARENT:
5947                 if (commit->has_parents)
5948                         break;
5949                 push_rev_graph(graph->parents, line + STRING_SIZE("parent "));
5950                 break;
5951
5952         case LINE_AUTHOR:
5953                 parse_author_line(line + STRING_SIZE("author "),
5954                                   commit->author, sizeof(commit->author),
5955                                   &commit->time);
5956                 update_rev_graph(view, graph);
5957                 graph = graph->next;
5958                 break;
5959
5960         default:
5961                 /* Fill in the commit title if it has not already been set. */
5962                 if (commit->title[0])
5963                         break;
5964
5965                 /* Require titles to start with a non-space character at the
5966                  * offset used by git log. */
5967                 if (strncmp(line, "    ", 4))
5968                         break;
5969                 line += 4;
5970                 /* Well, if the title starts with a whitespace character,
5971                  * try to be forgiving.  Otherwise we end up with no title. */
5972                 while (isspace(*line))
5973                         line++;
5974                 if (*line == '\0')
5975                         break;
5976                 /* FIXME: More graceful handling of titles; append "..." to
5977                  * shortened titles, etc. */
5978
5979                 string_expand(commit->title, sizeof(commit->title), line, 1);
5980                 view->line[view->lines - 1].dirty = 1;
5981         }
5982
5983         return TRUE;
5984 }
5985
5986 static enum request
5987 main_request(struct view *view, enum request request, struct line *line)
5988 {
5989         enum open_flags flags = display[0] == view ? OPEN_SPLIT : OPEN_DEFAULT;
5990
5991         switch (request) {
5992         case REQ_ENTER:
5993                 open_view(view, REQ_VIEW_DIFF, flags);
5994                 break;
5995         case REQ_REFRESH:
5996                 load_refs();
5997                 open_view(view, REQ_VIEW_MAIN, OPEN_REFRESH);
5998                 break;
5999         default:
6000                 return request;
6001         }
6002
6003         return REQ_NONE;
6004 }
6005
6006 static bool
6007 grep_refs(struct ref **refs, regex_t *regex)
6008 {
6009         regmatch_t pmatch;
6010         size_t i = 0;
6011
6012         if (!refs)
6013                 return FALSE;
6014         do {
6015                 if (regexec(regex, refs[i]->name, 1, &pmatch, 0) != REG_NOMATCH)
6016                         return TRUE;
6017         } while (refs[i++]->next);
6018
6019         return FALSE;
6020 }
6021
6022 static bool
6023 main_grep(struct view *view, struct line *line)
6024 {
6025         struct commit *commit = line->data;
6026         enum { S_TITLE, S_AUTHOR, S_DATE, S_REFS, S_END } state;
6027         char buf[DATE_COLS + 1];
6028         regmatch_t pmatch;
6029
6030         for (state = S_TITLE; state < S_END; state++) {
6031                 char *text;
6032
6033                 switch (state) {
6034                 case S_TITLE:   text = commit->title;   break;
6035                 case S_AUTHOR:
6036                         if (!opt_author)
6037                                 continue;
6038                         text = commit->author;
6039                         break;
6040                 case S_DATE:
6041                         if (!opt_date)
6042                                 continue;
6043                         if (!strftime(buf, sizeof(buf), DATE_FORMAT, &commit->time))
6044                                 continue;
6045                         text = buf;
6046                         break;
6047                 case S_REFS:
6048                         if (!opt_show_refs)
6049                                 continue;
6050                         if (grep_refs(commit->refs, view->regex) == TRUE)
6051                                 return TRUE;
6052                         continue;
6053                 default:
6054                         return FALSE;
6055                 }
6056
6057                 if (regexec(view->regex, text, 1, &pmatch, 0) != REG_NOMATCH)
6058                         return TRUE;
6059         }
6060
6061         return FALSE;
6062 }
6063
6064 static void
6065 main_select(struct view *view, struct line *line)
6066 {
6067         struct commit *commit = line->data;
6068
6069         string_copy_rev(view->ref, commit->id);
6070         string_copy_rev(ref_commit, view->ref);
6071 }
6072
6073 static struct view_ops main_ops = {
6074         "commit",
6075         main_argv,
6076         NULL,
6077         main_read,
6078         main_draw,
6079         main_request,
6080         main_grep,
6081         main_select,
6082 };
6083
6084
6085 /*
6086  * Unicode / UTF-8 handling
6087  *
6088  * NOTE: Much of the following code for dealing with Unicode is derived from
6089  * ELinks' UTF-8 code developed by Scrool <scroolik@gmail.com>. Origin file is
6090  * src/intl/charset.c from the UTF-8 branch commit elinks-0.11.0-g31f2c28.
6091  */
6092
6093 static inline int
6094 unicode_width(unsigned long c)
6095 {
6096         if (c >= 0x1100 &&
6097            (c <= 0x115f                         /* Hangul Jamo */
6098             || c == 0x2329
6099             || c == 0x232a
6100             || (c >= 0x2e80  && c <= 0xa4cf && c != 0x303f)
6101                                                 /* CJK ... Yi */
6102             || (c >= 0xac00  && c <= 0xd7a3)    /* Hangul Syllables */
6103             || (c >= 0xf900  && c <= 0xfaff)    /* CJK Compatibility Ideographs */
6104             || (c >= 0xfe30  && c <= 0xfe6f)    /* CJK Compatibility Forms */
6105             || (c >= 0xff00  && c <= 0xff60)    /* Fullwidth Forms */
6106             || (c >= 0xffe0  && c <= 0xffe6)
6107             || (c >= 0x20000 && c <= 0x2fffd)
6108             || (c >= 0x30000 && c <= 0x3fffd)))
6109                 return 2;
6110
6111         if (c == '\t')
6112                 return opt_tab_size;
6113
6114         return 1;
6115 }
6116
6117 /* Number of bytes used for encoding a UTF-8 character indexed by first byte.
6118  * Illegal bytes are set one. */
6119 static const unsigned char utf8_bytes[256] = {
6120         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,
6121         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,
6122         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,
6123         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,
6124         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,
6125         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,
6126         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,
6127         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,
6128 };
6129
6130 /* Decode UTF-8 multi-byte representation into a Unicode character. */
6131 static inline unsigned long
6132 utf8_to_unicode(const char *string, size_t length)
6133 {
6134         unsigned long unicode;
6135
6136         switch (length) {
6137         case 1:
6138                 unicode  =   string[0];
6139                 break;
6140         case 2:
6141                 unicode  =  (string[0] & 0x1f) << 6;
6142                 unicode +=  (string[1] & 0x3f);
6143                 break;
6144         case 3:
6145                 unicode  =  (string[0] & 0x0f) << 12;
6146                 unicode += ((string[1] & 0x3f) << 6);
6147                 unicode +=  (string[2] & 0x3f);
6148                 break;
6149         case 4:
6150                 unicode  =  (string[0] & 0x0f) << 18;
6151                 unicode += ((string[1] & 0x3f) << 12);
6152                 unicode += ((string[2] & 0x3f) << 6);
6153                 unicode +=  (string[3] & 0x3f);
6154                 break;
6155         case 5:
6156                 unicode  =  (string[0] & 0x0f) << 24;
6157                 unicode += ((string[1] & 0x3f) << 18);
6158                 unicode += ((string[2] & 0x3f) << 12);
6159                 unicode += ((string[3] & 0x3f) << 6);
6160                 unicode +=  (string[4] & 0x3f);
6161                 break;
6162         case 6:
6163                 unicode  =  (string[0] & 0x01) << 30;
6164                 unicode += ((string[1] & 0x3f) << 24);
6165                 unicode += ((string[2] & 0x3f) << 18);
6166                 unicode += ((string[3] & 0x3f) << 12);
6167                 unicode += ((string[4] & 0x3f) << 6);
6168                 unicode +=  (string[5] & 0x3f);
6169                 break;
6170         default:
6171                 die("Invalid Unicode length");
6172         }
6173
6174         /* Invalid characters could return the special 0xfffd value but NUL
6175          * should be just as good. */
6176         return unicode > 0xffff ? 0 : unicode;
6177 }
6178
6179 /* Calculates how much of string can be shown within the given maximum width
6180  * and sets trimmed parameter to non-zero value if all of string could not be
6181  * shown. If the reserve flag is TRUE, it will reserve at least one
6182  * trailing character, which can be useful when drawing a delimiter.
6183  *
6184  * Returns the number of bytes to output from string to satisfy max_width. */
6185 static size_t
6186 utf8_length(const char **start, size_t skip, int *width, size_t max_width, int *trimmed, bool reserve)
6187 {
6188         const char *string = *start;
6189         const char *end = strchr(string, '\0');
6190         unsigned char last_bytes = 0;
6191         size_t last_ucwidth = 0;
6192
6193         *width = 0;
6194         *trimmed = 0;
6195
6196         while (string < end) {
6197                 int c = *(unsigned char *) string;
6198                 unsigned char bytes = utf8_bytes[c];
6199                 size_t ucwidth;
6200                 unsigned long unicode;
6201
6202                 if (string + bytes > end)
6203                         break;
6204
6205                 /* Change representation to figure out whether
6206                  * it is a single- or double-width character. */
6207
6208                 unicode = utf8_to_unicode(string, bytes);
6209                 /* FIXME: Graceful handling of invalid Unicode character. */
6210                 if (!unicode)
6211                         break;
6212
6213                 ucwidth = unicode_width(unicode);
6214                 if (skip > 0) {
6215                         skip -= ucwidth <= skip ? ucwidth : skip;
6216                         *start += bytes;
6217                 }
6218                 *width  += ucwidth;
6219                 if (*width > max_width) {
6220                         *trimmed = 1;
6221                         *width -= ucwidth;
6222                         if (reserve && *width == max_width) {
6223                                 string -= last_bytes;
6224                                 *width -= last_ucwidth;
6225                         }
6226                         break;
6227                 }
6228
6229                 string  += bytes;
6230                 last_bytes = ucwidth ? bytes : 0;
6231                 last_ucwidth = ucwidth;
6232         }
6233
6234         return string - *start;
6235 }
6236
6237
6238 /*
6239  * Status management
6240  */
6241
6242 /* Whether or not the curses interface has been initialized. */
6243 static bool cursed = FALSE;
6244
6245 /* Terminal hacks and workarounds. */
6246 static bool use_scroll_redrawwin;
6247 static bool use_scroll_status_wclear;
6248
6249 /* The status window is used for polling keystrokes. */
6250 static WINDOW *status_win;
6251
6252 /* Reading from the prompt? */
6253 static bool input_mode = FALSE;
6254
6255 static bool status_empty = FALSE;
6256
6257 /* Update status and title window. */
6258 static void
6259 report(const char *msg, ...)
6260 {
6261         struct view *view = display[current_view];
6262
6263         if (input_mode)
6264                 return;
6265
6266         if (!view) {
6267                 char buf[SIZEOF_STR];
6268                 va_list args;
6269
6270                 va_start(args, msg);
6271                 if (vsnprintf(buf, sizeof(buf), msg, args) >= sizeof(buf)) {
6272                         buf[sizeof(buf) - 1] = 0;
6273                         buf[sizeof(buf) - 2] = '.';
6274                         buf[sizeof(buf) - 3] = '.';
6275                         buf[sizeof(buf) - 4] = '.';
6276                 }
6277                 va_end(args);
6278                 die("%s", buf);
6279         }
6280
6281         if (!status_empty || *msg) {
6282                 va_list args;
6283
6284                 va_start(args, msg);
6285
6286                 wmove(status_win, 0, 0);
6287                 if (view->has_scrolled && use_scroll_status_wclear)
6288                         wclear(status_win);
6289                 if (*msg) {
6290                         vwprintw(status_win, msg, args);
6291                         status_empty = FALSE;
6292                 } else {
6293                         status_empty = TRUE;
6294                 }
6295                 wclrtoeol(status_win);
6296                 wnoutrefresh(status_win);
6297
6298                 va_end(args);
6299         }
6300
6301         update_view_title(view);
6302 }
6303
6304 /* Controls when nodelay should be in effect when polling user input. */
6305 static void
6306 set_nonblocking_input(bool loading)
6307 {
6308         static unsigned int loading_views;
6309
6310         if ((loading == FALSE && loading_views-- == 1) ||
6311             (loading == TRUE  && loading_views++ == 0))
6312                 nodelay(status_win, loading);
6313 }
6314
6315 static void
6316 init_display(void)
6317 {
6318         const char *term;
6319         int x, y;
6320
6321         /* Initialize the curses library */
6322         if (isatty(STDIN_FILENO)) {
6323                 cursed = !!initscr();
6324                 opt_tty = stdin;
6325         } else {
6326                 /* Leave stdin and stdout alone when acting as a pager. */
6327                 opt_tty = fopen("/dev/tty", "r+");
6328                 if (!opt_tty)
6329                         die("Failed to open /dev/tty");
6330                 cursed = !!newterm(NULL, opt_tty, opt_tty);
6331         }
6332
6333         if (!cursed)
6334                 die("Failed to initialize curses");
6335
6336         nonl();         /* Disable conversion and detect newlines from input. */
6337         cbreak();       /* Take input chars one at a time, no wait for \n */
6338         noecho();       /* Don't echo input */
6339         leaveok(stdscr, FALSE);
6340
6341         if (has_colors())
6342                 init_colors();
6343
6344         getmaxyx(stdscr, y, x);
6345         status_win = newwin(1, 0, y - 1, 0);
6346         if (!status_win)
6347                 die("Failed to create status window");
6348
6349         /* Enable keyboard mapping */
6350         keypad(status_win, TRUE);
6351         wbkgdset(status_win, get_line_attr(LINE_STATUS));
6352
6353         TABSIZE = opt_tab_size;
6354         if (opt_line_graphics) {
6355                 line_graphics[LINE_GRAPHIC_VLINE] = ACS_VLINE;
6356         }
6357
6358         term = getenv("XTERM_VERSION") ? NULL : getenv("COLORTERM");
6359         if (term && !strcmp(term, "gnome-terminal")) {
6360                 /* In the gnome-terminal-emulator, the message from
6361                  * scrolling up one line when impossible followed by
6362                  * scrolling down one line causes corruption of the
6363                  * status line. This is fixed by calling wclear. */
6364                 use_scroll_status_wclear = TRUE;
6365                 use_scroll_redrawwin = FALSE;
6366
6367         } else if (term && !strcmp(term, "xrvt-xpm")) {
6368                 /* No problems with full optimizations in xrvt-(unicode)
6369                  * and aterm. */
6370                 use_scroll_status_wclear = use_scroll_redrawwin = FALSE;
6371
6372         } else {
6373                 /* When scrolling in (u)xterm the last line in the
6374                  * scrolling direction will update slowly. */
6375                 use_scroll_redrawwin = TRUE;
6376                 use_scroll_status_wclear = FALSE;
6377         }
6378 }
6379
6380 static int
6381 get_input(int prompt_position)
6382 {
6383         struct view *view;
6384         int i, key, cursor_y, cursor_x;
6385
6386         if (prompt_position)
6387                 input_mode = TRUE;
6388
6389         while (TRUE) {
6390                 foreach_view (view, i) {
6391                         update_view(view);
6392                         if (view_is_displayed(view) && view->has_scrolled &&
6393                             use_scroll_redrawwin)
6394                                 redrawwin(view->win);
6395                         view->has_scrolled = FALSE;
6396                 }
6397
6398                 /* Update the cursor position. */
6399                 if (prompt_position) {
6400                         getbegyx(status_win, cursor_y, cursor_x);
6401                         cursor_x = prompt_position;
6402                 } else {
6403                         view = display[current_view];
6404                         getbegyx(view->win, cursor_y, cursor_x);
6405                         cursor_x = view->width - 1;
6406                         cursor_y += view->lineno - view->offset;
6407                 }
6408                 setsyx(cursor_y, cursor_x);
6409
6410                 /* Refresh, accept single keystroke of input */
6411                 doupdate();
6412                 key = wgetch(status_win);
6413
6414                 /* wgetch() with nodelay() enabled returns ERR when
6415                  * there's no input. */
6416                 if (key == ERR) {
6417
6418                 } else if (key == KEY_RESIZE) {
6419                         int height, width;
6420
6421                         getmaxyx(stdscr, height, width);
6422
6423                         wresize(status_win, 1, width);
6424                         mvwin(status_win, height - 1, 0);
6425                         wnoutrefresh(status_win);
6426                         resize_display();
6427                         redraw_display(TRUE);
6428
6429                 } else {
6430                         input_mode = FALSE;
6431                         return key;
6432                 }
6433         }
6434 }
6435
6436 static char *
6437 prompt_input(const char *prompt, input_handler handler, void *data)
6438 {
6439         enum input_status status = INPUT_OK;
6440         static char buf[SIZEOF_STR];
6441         size_t pos = 0;
6442
6443         buf[pos] = 0;
6444
6445         while (status == INPUT_OK || status == INPUT_SKIP) {
6446                 int key;
6447
6448                 mvwprintw(status_win, 0, 0, "%s%.*s", prompt, pos, buf);
6449                 wclrtoeol(status_win);
6450
6451                 key = get_input(pos + 1);
6452                 switch (key) {
6453                 case KEY_RETURN:
6454                 case KEY_ENTER:
6455                 case '\n':
6456                         status = pos ? INPUT_STOP : INPUT_CANCEL;
6457                         break;
6458
6459                 case KEY_BACKSPACE:
6460                         if (pos > 0)
6461                                 buf[--pos] = 0;
6462                         else
6463                                 status = INPUT_CANCEL;
6464                         break;
6465
6466                 case KEY_ESC:
6467                         status = INPUT_CANCEL;
6468                         break;
6469
6470                 default:
6471                         if (pos >= sizeof(buf)) {
6472                                 report("Input string too long");
6473                                 return NULL;
6474                         }
6475
6476                         status = handler(data, buf, key);
6477                         if (status == INPUT_OK)
6478                                 buf[pos++] = (char) key;
6479                 }
6480         }
6481
6482         /* Clear the status window */
6483         status_empty = FALSE;
6484         report("");
6485
6486         if (status == INPUT_CANCEL)
6487                 return NULL;
6488
6489         buf[pos++] = 0;
6490
6491         return buf;
6492 }
6493
6494 static enum input_status
6495 prompt_yesno_handler(void *data, char *buf, int c)
6496 {
6497         if (c == 'y' || c == 'Y')
6498                 return INPUT_STOP;
6499         if (c == 'n' || c == 'N')
6500                 return INPUT_CANCEL;
6501         return INPUT_SKIP;
6502 }
6503
6504 static bool
6505 prompt_yesno(const char *prompt)
6506 {
6507         char prompt2[SIZEOF_STR];
6508
6509         if (!string_format(prompt2, "%s [Yy/Nn]", prompt))
6510                 return FALSE;
6511
6512         return !!prompt_input(prompt2, prompt_yesno_handler, NULL);
6513 }
6514
6515 static enum input_status
6516 read_prompt_handler(void *data, char *buf, int c)
6517 {
6518         return isprint(c) ? INPUT_OK : INPUT_SKIP;
6519 }
6520
6521 static char *
6522 read_prompt(const char *prompt)
6523 {
6524         return prompt_input(prompt, read_prompt_handler, NULL);
6525 }
6526
6527 /*
6528  * Repository properties
6529  */
6530
6531 static struct ref *refs = NULL;
6532 static size_t refs_alloc = 0;
6533 static size_t refs_size = 0;
6534
6535 /* Id <-> ref store */
6536 static struct ref ***id_refs = NULL;
6537 static size_t id_refs_alloc = 0;
6538 static size_t id_refs_size = 0;
6539
6540 static int
6541 compare_refs(const void *ref1_, const void *ref2_)
6542 {
6543         const struct ref *ref1 = *(const struct ref **)ref1_;
6544         const struct ref *ref2 = *(const struct ref **)ref2_;
6545
6546         if (ref1->tag != ref2->tag)
6547                 return ref2->tag - ref1->tag;
6548         if (ref1->ltag != ref2->ltag)
6549                 return ref2->ltag - ref2->ltag;
6550         if (ref1->head != ref2->head)
6551                 return ref2->head - ref1->head;
6552         if (ref1->tracked != ref2->tracked)
6553                 return ref2->tracked - ref1->tracked;
6554         if (ref1->remote != ref2->remote)
6555                 return ref2->remote - ref1->remote;
6556         return strcmp(ref1->name, ref2->name);
6557 }
6558
6559 static struct ref **
6560 get_refs(const char *id)
6561 {
6562         struct ref ***tmp_id_refs;
6563         struct ref **ref_list = NULL;
6564         size_t ref_list_alloc = 0;
6565         size_t ref_list_size = 0;
6566         size_t i;
6567
6568         for (i = 0; i < id_refs_size; i++)
6569                 if (!strcmp(id, id_refs[i][0]->id))
6570                         return id_refs[i];
6571
6572         tmp_id_refs = realloc_items(id_refs, &id_refs_alloc, id_refs_size + 1,
6573                                     sizeof(*id_refs));
6574         if (!tmp_id_refs)
6575                 return NULL;
6576
6577         id_refs = tmp_id_refs;
6578
6579         for (i = 0; i < refs_size; i++) {
6580                 struct ref **tmp;
6581
6582                 if (strcmp(id, refs[i].id))
6583                         continue;
6584
6585                 tmp = realloc_items(ref_list, &ref_list_alloc,
6586                                     ref_list_size + 1, sizeof(*ref_list));
6587                 if (!tmp) {
6588                         if (ref_list)
6589                                 free(ref_list);
6590                         return NULL;
6591                 }
6592
6593                 ref_list = tmp;
6594                 ref_list[ref_list_size] = &refs[i];
6595                 /* XXX: The properties of the commit chains ensures that we can
6596                  * safely modify the shared ref. The repo references will
6597                  * always be similar for the same id. */
6598                 ref_list[ref_list_size]->next = 1;
6599
6600                 ref_list_size++;
6601         }
6602
6603         if (ref_list) {
6604                 qsort(ref_list, ref_list_size, sizeof(*ref_list), compare_refs);
6605                 ref_list[ref_list_size - 1]->next = 0;
6606                 id_refs[id_refs_size++] = ref_list;
6607         }
6608
6609         return ref_list;
6610 }
6611
6612 static int
6613 read_ref(char *id, size_t idlen, char *name, size_t namelen)
6614 {
6615         struct ref *ref;
6616         bool tag = FALSE;
6617         bool ltag = FALSE;
6618         bool remote = FALSE;
6619         bool tracked = FALSE;
6620         bool check_replace = FALSE;
6621         bool head = FALSE;
6622
6623         if (!prefixcmp(name, "refs/tags/")) {
6624                 if (!suffixcmp(name, namelen, "^{}")) {
6625                         namelen -= 3;
6626                         name[namelen] = 0;
6627                         if (refs_size > 0 && refs[refs_size - 1].ltag == TRUE)
6628                                 check_replace = TRUE;
6629                 } else {
6630                         ltag = TRUE;
6631                 }
6632
6633                 tag = TRUE;
6634                 namelen -= STRING_SIZE("refs/tags/");
6635                 name    += STRING_SIZE("refs/tags/");
6636
6637         } else if (!prefixcmp(name, "refs/remotes/")) {
6638                 remote = TRUE;
6639                 namelen -= STRING_SIZE("refs/remotes/");
6640                 name    += STRING_SIZE("refs/remotes/");
6641                 tracked  = !strcmp(opt_remote, name);
6642
6643         } else if (!prefixcmp(name, "refs/heads/")) {
6644                 namelen -= STRING_SIZE("refs/heads/");
6645                 name    += STRING_SIZE("refs/heads/");
6646                 head     = !strncmp(opt_head, name, namelen);
6647
6648         } else if (!strcmp(name, "HEAD")) {
6649                 string_ncopy(opt_head_rev, id, idlen);
6650                 return OK;
6651         }
6652
6653         if (check_replace && !strcmp(name, refs[refs_size - 1].name)) {
6654                 /* it's an annotated tag, replace the previous SHA1 with the
6655                  * resolved commit id; relies on the fact git-ls-remote lists
6656                  * the commit id of an annotated tag right before the commit id
6657                  * it points to. */
6658                 refs[refs_size - 1].ltag = ltag;
6659                 string_copy_rev(refs[refs_size - 1].id, id);
6660
6661                 return OK;
6662         }
6663         refs = realloc_items(refs, &refs_alloc, refs_size + 1, sizeof(*refs));
6664         if (!refs)
6665                 return ERR;
6666
6667         ref = &refs[refs_size++];
6668         ref->name = malloc(namelen + 1);
6669         if (!ref->name)
6670                 return ERR;
6671
6672         strncpy(ref->name, name, namelen);
6673         ref->name[namelen] = 0;
6674         ref->head = head;
6675         ref->tag = tag;
6676         ref->ltag = ltag;
6677         ref->remote = remote;
6678         ref->tracked = tracked;
6679         string_copy_rev(ref->id, id);
6680
6681         return OK;
6682 }
6683
6684 static int
6685 load_refs(void)
6686 {
6687         static const char *ls_remote_argv[SIZEOF_ARG] = {
6688                 "git", "ls-remote", opt_git_dir, NULL
6689         };
6690         static bool init = FALSE;
6691
6692         if (!init) {
6693                 argv_from_env(ls_remote_argv, "TIG_LS_REMOTE");
6694                 init = TRUE;
6695         }
6696
6697         if (!*opt_git_dir)
6698                 return OK;
6699
6700         while (refs_size > 0)
6701                 free(refs[--refs_size].name);
6702         while (id_refs_size > 0)
6703                 free(id_refs[--id_refs_size]);
6704
6705         return run_io_load(ls_remote_argv, "\t", read_ref);
6706 }
6707
6708 static void
6709 set_remote_branch(const char *name, const char *value, size_t valuelen)
6710 {
6711         if (!strcmp(name, ".remote")) {
6712                 string_ncopy(opt_remote, value, valuelen);
6713
6714         } else if (*opt_remote && !strcmp(name, ".merge")) {
6715                 size_t from = strlen(opt_remote);
6716
6717                 if (!prefixcmp(value, "refs/heads/"))
6718                         value += STRING_SIZE("refs/heads/");
6719
6720                 if (!string_format_from(opt_remote, &from, "/%s", value))
6721                         opt_remote[0] = 0;
6722         }
6723 }
6724
6725 static void
6726 set_repo_config_option(char *name, char *value, int (*cmd)(int, const char **))
6727 {
6728         const char *argv[SIZEOF_ARG] = { name, "=" };
6729         int argc = 1 + (cmd == option_set_command);
6730         int error = ERR;
6731
6732         if (!argv_from_string(argv, &argc, value))
6733                 config_msg = "Too many option arguments";
6734         else
6735                 error = cmd(argc, argv);
6736
6737         if (error == ERR)
6738                 warn("Option 'tig.%s': %s", name, config_msg);
6739 }
6740
6741 static bool
6742 set_environment_variable(const char *name, const char *value)
6743 {
6744         size_t len = strlen(name) + 1 + strlen(value) + 1;
6745         char *env = malloc(len);
6746
6747         if (env &&
6748             string_nformat(env, len, NULL, "%s=%s", name, value) &&
6749             putenv(env) == 0)
6750                 return TRUE;
6751         free(env);
6752         return FALSE;
6753 }
6754
6755 static void
6756 set_work_tree(const char *value)
6757 {
6758         char cwd[SIZEOF_STR];
6759
6760         if (!getcwd(cwd, sizeof(cwd)))
6761                 die("Failed to get cwd path: %s", strerror(errno));
6762         if (chdir(opt_git_dir) < 0)
6763                 die("Failed to chdir(%s): %s", strerror(errno));
6764         if (!getcwd(opt_git_dir, sizeof(opt_git_dir)))
6765                 die("Failed to get git path: %s", strerror(errno));
6766         if (chdir(cwd) < 0)
6767                 die("Failed to chdir(%s): %s", cwd, strerror(errno));
6768         if (chdir(value) < 0)
6769                 die("Failed to chdir(%s): %s", value, strerror(errno));
6770         if (!getcwd(cwd, sizeof(cwd)))
6771                 die("Failed to get cwd path: %s", strerror(errno));
6772         if (!set_environment_variable("GIT_WORK_TREE", cwd))
6773                 die("Failed to set GIT_WORK_TREE to '%s'", cwd);
6774         if (!set_environment_variable("GIT_DIR", opt_git_dir))
6775                 die("Failed to set GIT_DIR to '%s'", opt_git_dir);
6776         opt_is_inside_work_tree = TRUE;
6777 }
6778
6779 static int
6780 read_repo_config_option(char *name, size_t namelen, char *value, size_t valuelen)
6781 {
6782         if (!strcmp(name, "i18n.commitencoding"))
6783                 string_ncopy(opt_encoding, value, valuelen);
6784
6785         else if (!strcmp(name, "core.editor"))
6786                 string_ncopy(opt_editor, value, valuelen);
6787
6788         else if (!strcmp(name, "core.worktree"))
6789                 set_work_tree(value);
6790
6791         else if (!prefixcmp(name, "tig.color."))
6792                 set_repo_config_option(name + 10, value, option_color_command);
6793
6794         else if (!prefixcmp(name, "tig.bind."))
6795                 set_repo_config_option(name + 9, value, option_bind_command);
6796
6797         else if (!prefixcmp(name, "tig."))
6798                 set_repo_config_option(name + 4, value, option_set_command);
6799
6800         else if (*opt_head && !prefixcmp(name, "branch.") &&
6801                  !strncmp(name + 7, opt_head, strlen(opt_head)))
6802                 set_remote_branch(name + 7 + strlen(opt_head), value, valuelen);
6803
6804         return OK;
6805 }
6806
6807 static int
6808 load_git_config(void)
6809 {
6810         const char *config_list_argv[] = { "git", GIT_CONFIG, "--list", NULL };
6811
6812         return run_io_load(config_list_argv, "=", read_repo_config_option);
6813 }
6814
6815 static int
6816 read_repo_info(char *name, size_t namelen, char *value, size_t valuelen)
6817 {
6818         if (!opt_git_dir[0]) {
6819                 string_ncopy(opt_git_dir, name, namelen);
6820
6821         } else if (opt_is_inside_work_tree == -1) {
6822                 /* This can be 3 different values depending on the
6823                  * version of git being used. If git-rev-parse does not
6824                  * understand --is-inside-work-tree it will simply echo
6825                  * the option else either "true" or "false" is printed.
6826                  * Default to true for the unknown case. */
6827                 opt_is_inside_work_tree = strcmp(name, "false") ? TRUE : FALSE;
6828
6829         } else if (*name == '.') {
6830                 string_ncopy(opt_cdup, name, namelen);
6831
6832         } else {
6833                 string_ncopy(opt_prefix, name, namelen);
6834         }
6835
6836         return OK;
6837 }
6838
6839 static int
6840 load_repo_info(void)
6841 {
6842         const char *head_argv[] = {
6843                 "git", "symbolic-ref", "HEAD", NULL
6844         };
6845         const char *rev_parse_argv[] = {
6846                 "git", "rev-parse", "--git-dir", "--is-inside-work-tree",
6847                         "--show-cdup", "--show-prefix", NULL
6848         };
6849
6850         if (run_io_buf(head_argv, opt_head, sizeof(opt_head))) {
6851                 chomp_string(opt_head);
6852                 if (!prefixcmp(opt_head, "refs/heads/")) {
6853                         char *offset = opt_head + STRING_SIZE("refs/heads/");
6854
6855                         memmove(opt_head, offset, strlen(offset) + 1);
6856                 }
6857         }
6858
6859         return run_io_load(rev_parse_argv, "=", read_repo_info);
6860 }
6861
6862
6863 /*
6864  * Main
6865  */
6866
6867 static const char usage[] =
6868 "tig " TIG_VERSION " (" __DATE__ ")\n"
6869 "\n"
6870 "Usage: tig        [options] [revs] [--] [paths]\n"
6871 "   or: tig show   [options] [revs] [--] [paths]\n"
6872 "   or: tig blame  [rev] path\n"
6873 "   or: tig status\n"
6874 "   or: tig <      [git command output]\n"
6875 "\n"
6876 "Options:\n"
6877 "  -v, --version   Show version and exit\n"
6878 "  -h, --help      Show help message and exit";
6879
6880 static void __NORETURN
6881 quit(int sig)
6882 {
6883         /* XXX: Restore tty modes and let the OS cleanup the rest! */
6884         if (cursed)
6885                 endwin();
6886         exit(0);
6887 }
6888
6889 static void __NORETURN
6890 die(const char *err, ...)
6891 {
6892         va_list args;
6893
6894         endwin();
6895
6896         va_start(args, err);
6897         fputs("tig: ", stderr);
6898         vfprintf(stderr, err, args);
6899         fputs("\n", stderr);
6900         va_end(args);
6901
6902         exit(1);
6903 }
6904
6905 static void
6906 warn(const char *msg, ...)
6907 {
6908         va_list args;
6909
6910         va_start(args, msg);
6911         fputs("tig warning: ", stderr);
6912         vfprintf(stderr, msg, args);
6913         fputs("\n", stderr);
6914         va_end(args);
6915 }
6916
6917 static enum request
6918 parse_options(int argc, const char *argv[])
6919 {
6920         enum request request = REQ_VIEW_MAIN;
6921         const char *subcommand;
6922         bool seen_dashdash = FALSE;
6923         /* XXX: This is vulnerable to the user overriding options
6924          * required for the main view parser. */
6925         const char *custom_argv[SIZEOF_ARG] = {
6926                 "git", "log", "--no-color", "--pretty=raw", "--parents",
6927                         "--topo-order", NULL
6928         };
6929         int i, j = 6;
6930
6931         if (!isatty(STDIN_FILENO)) {
6932                 io_open(&VIEW(REQ_VIEW_PAGER)->io, "");
6933                 return REQ_VIEW_PAGER;
6934         }
6935
6936         if (argc <= 1)
6937                 return REQ_NONE;
6938
6939         subcommand = argv[1];
6940         if (!strcmp(subcommand, "status")) {
6941                 if (argc > 2)
6942                         warn("ignoring arguments after `%s'", subcommand);
6943                 return REQ_VIEW_STATUS;
6944
6945         } else if (!strcmp(subcommand, "blame")) {
6946                 if (argc <= 2 || argc > 4)
6947                         die("invalid number of options to blame\n\n%s", usage);
6948
6949                 i = 2;
6950                 if (argc == 4) {
6951                         string_ncopy(opt_ref, argv[i], strlen(argv[i]));
6952                         i++;
6953                 }
6954
6955                 string_ncopy(opt_file, argv[i], strlen(argv[i]));
6956                 return REQ_VIEW_BLAME;
6957
6958         } else if (!strcmp(subcommand, "show")) {
6959                 request = REQ_VIEW_DIFF;
6960
6961         } else {
6962                 subcommand = NULL;
6963         }
6964
6965         if (subcommand) {
6966                 custom_argv[1] = subcommand;
6967                 j = 2;
6968         }
6969
6970         for (i = 1 + !!subcommand; i < argc; i++) {
6971                 const char *opt = argv[i];
6972
6973                 if (seen_dashdash || !strcmp(opt, "--")) {
6974                         seen_dashdash = TRUE;
6975
6976                 } else if (!strcmp(opt, "-v") || !strcmp(opt, "--version")) {
6977                         printf("tig version %s\n", TIG_VERSION);
6978                         quit(0);
6979
6980                 } else if (!strcmp(opt, "-h") || !strcmp(opt, "--help")) {
6981                         printf("%s\n", usage);
6982                         quit(0);
6983                 }
6984
6985                 custom_argv[j++] = opt;
6986                 if (j >= ARRAY_SIZE(custom_argv))
6987                         die("command too long");
6988         }
6989
6990         if (!prepare_update(VIEW(request), custom_argv, NULL, FORMAT_NONE))
6991                 die("Failed to format arguments");
6992
6993         return request;
6994 }
6995
6996 int
6997 main(int argc, const char *argv[])
6998 {
6999         enum request request = parse_options(argc, argv);
7000         struct view *view;
7001         size_t i;
7002
7003         signal(SIGINT, quit);
7004         signal(SIGPIPE, SIG_IGN);
7005
7006         if (setlocale(LC_ALL, "")) {
7007                 char *codeset = nl_langinfo(CODESET);
7008
7009                 string_ncopy(opt_codeset, codeset, strlen(codeset));
7010         }
7011
7012         if (load_repo_info() == ERR)
7013                 die("Failed to load repo info.");
7014
7015         if (load_options() == ERR)
7016                 die("Failed to load user config.");
7017
7018         if (load_git_config() == ERR)
7019                 die("Failed to load repo config.");
7020
7021         /* Require a git repository unless when running in pager mode. */
7022         if (!opt_git_dir[0] && request != REQ_VIEW_PAGER)
7023                 die("Not a git repository");
7024
7025         if (*opt_encoding && strcasecmp(opt_encoding, "UTF-8"))
7026                 opt_utf8 = FALSE;
7027
7028         if (*opt_codeset && strcmp(opt_codeset, opt_encoding)) {
7029                 opt_iconv = iconv_open(opt_codeset, opt_encoding);
7030                 if (opt_iconv == ICONV_NONE)
7031                         die("Failed to initialize character set conversion");
7032         }
7033
7034         if (load_refs() == ERR)
7035                 die("Failed to load refs.");
7036
7037         foreach_view (view, i)
7038                 argv_from_env(view->ops->argv, view->cmd_env);
7039
7040         init_display();
7041
7042         if (request != REQ_NONE)
7043                 open_view(NULL, request, OPEN_PREPARED);
7044         request = request == REQ_NONE ? REQ_VIEW_MAIN : REQ_NONE;
7045
7046         while (view_driver(display[current_view], request)) {
7047                 int key = get_input(0);
7048
7049                 view = display[current_view];
7050                 request = get_keybinding(view->keymap, key);
7051
7052                 /* Some low-level request handling. This keeps access to
7053                  * status_win restricted. */
7054                 switch (request) {
7055                 case REQ_PROMPT:
7056                 {
7057                         char *cmd = read_prompt(":");
7058
7059                         if (cmd && isdigit(*cmd)) {
7060                                 int lineno = view->lineno + 1;
7061
7062                                 if (parse_int(&lineno, cmd, 1, view->lines + 1) == OK) {
7063                                         select_view_line(view, lineno - 1);
7064                                         report("");
7065                                 } else {
7066                                         report("Unable to parse '%s' as a line number", cmd);
7067                                 }
7068
7069                         } else if (cmd) {
7070                                 struct view *next = VIEW(REQ_VIEW_PAGER);
7071                                 const char *argv[SIZEOF_ARG] = { "git" };
7072                                 int argc = 1;
7073
7074                                 /* When running random commands, initially show the
7075                                  * command in the title. However, it maybe later be
7076                                  * overwritten if a commit line is selected. */
7077                                 string_ncopy(next->ref, cmd, strlen(cmd));
7078
7079                                 if (!argv_from_string(argv, &argc, cmd)) {
7080                                         report("Too many arguments");
7081                                 } else if (!prepare_update(next, argv, NULL, FORMAT_DASH)) {
7082                                         report("Failed to format command");
7083                                 } else {
7084                                         open_view(view, REQ_VIEW_PAGER, OPEN_PREPARED);
7085                                 }
7086                         }
7087
7088                         request = REQ_NONE;
7089                         break;
7090                 }
7091                 case REQ_SEARCH:
7092                 case REQ_SEARCH_BACK:
7093                 {
7094                         const char *prompt = request == REQ_SEARCH ? "/" : "?";
7095                         char *search = read_prompt(prompt);
7096
7097                         if (search)
7098                                 string_ncopy(opt_search, search, strlen(search));
7099                         else if (*opt_search)
7100                                 request = request == REQ_SEARCH ?
7101                                         REQ_FIND_NEXT :
7102                                         REQ_FIND_PREV;
7103                         else
7104                                 request = REQ_NONE;
7105                         break;
7106                 }
7107                 default:
7108                         break;
7109                 }
7110         }
7111
7112         quit(0);
7113
7114         return 0;
7115 }