1 /*
2  * Copyright (c) 2018, 2019, 2020 Stefan Sperling <stsp@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <sys/queue.h>
18 #include <sys/stat.h>
19 #include <sys/ioctl.h>
20 
21 #include <ctype.h>
22 #include <errno.h>
23 #define _XOPEN_SOURCE_EXTENDED /* for ncurses wide-character functions */
24 #include <curses.h>
25 #include <panel.h>
26 #include <locale.h>
27 #include <sha1.h>
28 #include <sha2.h>
29 #include <signal.h>
30 #include <stdlib.h>
31 #include <stdarg.h>
32 #include <stdio.h>
33 #include <getopt.h>
34 #include <string.h>
35 #include <err.h>
36 #include <unistd.h>
37 #include <limits.h>
38 #include <wchar.h>
39 #include <time.h>
40 #include <pthread.h>
41 #include <libgen.h>
42 #include <regex.h>
43 #include <sched.h>
44 
45 #include "got_version.h"
46 #include "got_error.h"
47 #include "got_object.h"
48 #include "got_reference.h"
49 #include "got_repository.h"
50 #include "got_diff.h"
51 #include "got_opentemp.h"
52 #include "got_utf8.h"
53 #include "got_cancel.h"
54 #include "got_commit_graph.h"
55 #include "got_blame.h"
56 #include "got_privsep.h"
57 #include "got_path.h"
58 #include "got_worktree.h"
59 #include "got_keyword.h"
60 
61 #ifndef MIN
62 #define	MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
63 #endif
64 
65 #ifndef MAX
66 #define	MAX(_a,_b) ((_a) > (_b) ? (_a) : (_b))
67 #endif
68 
69 #define CTRL(x)		((x) & 0x1f)
70 
71 #ifndef nitems
72 #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
73 #endif
74 
75 struct tog_cmd {
76 	const char *name;
77 	const struct got_error *(*cmd_main)(int, char *[]);
78 	void (*cmd_usage)(void);
79 };
80 
81 __dead static void	usage(int, int);
82 __dead static void	usage_log(void);
83 __dead static void	usage_diff(void);
84 __dead static void	usage_blame(void);
85 __dead static void	usage_tree(void);
86 __dead static void	usage_ref(void);
87 
88 static const struct got_error*	cmd_log(int, char *[]);
89 static const struct got_error*	cmd_diff(int, char *[]);
90 static const struct got_error*	cmd_blame(int, char *[]);
91 static const struct got_error*	cmd_tree(int, char *[]);
92 static const struct got_error*	cmd_ref(int, char *[]);
93 
94 static const struct tog_cmd tog_commands[] = {
95 	{ "log",	cmd_log,	usage_log },
96 	{ "diff",	cmd_diff,	usage_diff },
97 	{ "blame",	cmd_blame,	usage_blame },
98 	{ "tree",	cmd_tree,	usage_tree },
99 	{ "ref",	cmd_ref,	usage_ref },
100 };
101 
102 enum tog_view_type {
103 	TOG_VIEW_DIFF,
104 	TOG_VIEW_LOG,
105 	TOG_VIEW_BLAME,
106 	TOG_VIEW_TREE,
107 	TOG_VIEW_REF,
108 	TOG_VIEW_HELP
109 };
110 
111 /* Match _DIFF to _HELP with enum tog_view_type TOG_VIEW_* counterparts. */
112 enum tog_keymap_type {
113 	TOG_KEYMAP_KEYS = -2,
114 	TOG_KEYMAP_GLOBAL,
115 	TOG_KEYMAP_DIFF,
116 	TOG_KEYMAP_LOG,
117 	TOG_KEYMAP_BLAME,
118 	TOG_KEYMAP_TREE,
119 	TOG_KEYMAP_REF,
120 	TOG_KEYMAP_HELP
121 };
122 
123 enum tog_view_mode {
124 	TOG_VIEW_SPLIT_NONE,
125 	TOG_VIEW_SPLIT_VERT,
126 	TOG_VIEW_SPLIT_HRZN
127 };
128 
129 #define HSPLIT_SCALE	0.3f  /* default horizontal split scale */
130 
131 #define TOG_EOF_STRING	"(END)"
132 
133 struct commit_queue_entry {
134 	TAILQ_ENTRY(commit_queue_entry) entry;
135 	struct got_object_id *id;
136 	struct got_commit_object *commit;
137 	int idx;
138 };
139 TAILQ_HEAD(commit_queue_head, commit_queue_entry);
140 struct commit_queue {
141 	int ncommits;
142 	struct commit_queue_head head;
143 };
144 
145 struct tog_color {
146 	STAILQ_ENTRY(tog_color) entry;
147 	regex_t regex;
148 	short colorpair;
149 };
150 STAILQ_HEAD(tog_colors, tog_color);
151 
152 static struct got_reflist_head tog_refs = TAILQ_HEAD_INITIALIZER(tog_refs);
153 static struct got_reflist_object_id_map *tog_refs_idmap;
154 static struct {
155 	struct got_object_id	*id;
156 	int			 idx;
157 	char			 marker;
158 } tog_base_commit;
159 static enum got_diff_algorithm tog_diff_algo = GOT_DIFF_ALGORITHM_PATIENCE;
160 
161 static const struct got_error *
162 tog_ref_cmp_by_name(void *arg, int *cmp, struct got_reference *re1,
163     struct got_reference* re2)
164 {
165 	const char *name1 = got_ref_get_name(re1);
166 	const char *name2 = got_ref_get_name(re2);
167 	int isbackup1, isbackup2;
168 
169 	/* Sort backup refs towards the bottom of the list. */
170 	isbackup1 = strncmp(name1, "refs/got/backup/", 16) == 0;
171 	isbackup2 = strncmp(name2, "refs/got/backup/", 16) == 0;
172 	if (!isbackup1 && isbackup2) {
173 		*cmp = -1;
174 		return NULL;
175 	} else if (isbackup1 && !isbackup2) {
176 		*cmp = 1;
177 		return NULL;
178 	}
179 
180 	*cmp = got_path_cmp(name1, name2, strlen(name1), strlen(name2));
181 	return NULL;
182 }
183 
184 static const struct got_error *
185 tog_load_refs(struct got_repository *repo, int sort_by_date)
186 {
187 	const struct got_error *err;
188 
189 	err = got_ref_list(&tog_refs, repo, NULL, sort_by_date ?
190 	    got_ref_cmp_by_commit_timestamp_descending : tog_ref_cmp_by_name,
191 	    repo);
192 	if (err)
193 		return err;
194 
195 	return got_reflist_object_id_map_create(&tog_refs_idmap, &tog_refs,
196 	    repo);
197 }
198 
199 static void
200 tog_free_refs(void)
201 {
202 	if (tog_refs_idmap) {
203 		got_reflist_object_id_map_free(tog_refs_idmap);
204 		tog_refs_idmap = NULL;
205 	}
206 	got_ref_list_free(&tog_refs);
207 }
208 
209 static const struct got_error *
210 add_color(struct tog_colors *colors, const char *pattern,
211     int idx, short color)
212 {
213 	const struct got_error *err = NULL;
214 	struct tog_color *tc;
215 	int regerr = 0;
216 
217 	if (idx < 1 || idx > COLOR_PAIRS - 1)
218 		return NULL;
219 
220 	init_pair(idx, color, -1);
221 
222 	tc = calloc(1, sizeof(*tc));
223 	if (tc == NULL)
224 		return got_error_from_errno("calloc");
225 	regerr = regcomp(&tc->regex, pattern,
226 	    REG_EXTENDED | REG_NOSUB | REG_NEWLINE);
227 	if (regerr) {
228 		static char regerr_msg[512];
229 		static char err_msg[512];
230 		regerror(regerr, &tc->regex, regerr_msg,
231 		    sizeof(regerr_msg));
232 		snprintf(err_msg, sizeof(err_msg), "regcomp: %s",
233 		    regerr_msg);
234 		err = got_error_msg(GOT_ERR_REGEX, err_msg);
235 		free(tc);
236 		return err;
237 	}
238 	tc->colorpair = idx;
239 	STAILQ_INSERT_HEAD(colors, tc, entry);
240 	return NULL;
241 }
242 
243 static void
244 free_colors(struct tog_colors *colors)
245 {
246 	struct tog_color *tc;
247 
248 	while (!STAILQ_EMPTY(colors)) {
249 		tc = STAILQ_FIRST(colors);
250 		STAILQ_REMOVE_HEAD(colors, entry);
251 		regfree(&tc->regex);
252 		free(tc);
253 	}
254 }
255 
256 static struct tog_color *
257 get_color(struct tog_colors *colors, int colorpair)
258 {
259 	struct tog_color *tc = NULL;
260 
261 	STAILQ_FOREACH(tc, colors, entry) {
262 		if (tc->colorpair == colorpair)
263 			return tc;
264 	}
265 
266 	return NULL;
267 }
268 
269 static int
270 default_color_value(const char *envvar)
271 {
272 	if (strcmp(envvar, "TOG_COLOR_DIFF_MINUS") == 0)
273 		return COLOR_MAGENTA;
274 	if (strcmp(envvar, "TOG_COLOR_DIFF_PLUS") == 0)
275 		return COLOR_CYAN;
276 	if (strcmp(envvar, "TOG_COLOR_DIFF_CHUNK_HEADER") == 0)
277 		return COLOR_YELLOW;
278 	if (strcmp(envvar, "TOG_COLOR_DIFF_META") == 0)
279 		return COLOR_GREEN;
280 	if (strcmp(envvar, "TOG_COLOR_TREE_SUBMODULE") == 0)
281 		return COLOR_MAGENTA;
282 	if (strcmp(envvar, "TOG_COLOR_TREE_SYMLINK") == 0)
283 		return COLOR_MAGENTA;
284 	if (strcmp(envvar, "TOG_COLOR_TREE_DIRECTORY") == 0)
285 		return COLOR_CYAN;
286 	if (strcmp(envvar, "TOG_COLOR_TREE_EXECUTABLE") == 0)
287 		return COLOR_GREEN;
288 	if (strcmp(envvar, "TOG_COLOR_COMMIT") == 0)
289 		return COLOR_GREEN;
290 	if (strcmp(envvar, "TOG_COLOR_AUTHOR") == 0)
291 		return COLOR_CYAN;
292 	if (strcmp(envvar, "TOG_COLOR_DATE") == 0)
293 		return COLOR_YELLOW;
294 	if (strcmp(envvar, "TOG_COLOR_REFS_HEADS") == 0)
295 		return COLOR_GREEN;
296 	if (strcmp(envvar, "TOG_COLOR_REFS_TAGS") == 0)
297 		return COLOR_MAGENTA;
298 	if (strcmp(envvar, "TOG_COLOR_REFS_REMOTES") == 0)
299 		return COLOR_YELLOW;
300 	if (strcmp(envvar, "TOG_COLOR_REFS_BACKUP") == 0)
301 		return COLOR_CYAN;
302 
303 	return -1;
304 }
305 
306 static int
307 get_color_value(const char *envvar)
308 {
309 	const char *val = getenv(envvar);
310 
311 	if (val == NULL)
312 		return default_color_value(envvar);
313 
314 	if (strcasecmp(val, "black") == 0)
315 		return COLOR_BLACK;
316 	if (strcasecmp(val, "red") == 0)
317 		return COLOR_RED;
318 	if (strcasecmp(val, "green") == 0)
319 		return COLOR_GREEN;
320 	if (strcasecmp(val, "yellow") == 0)
321 		return COLOR_YELLOW;
322 	if (strcasecmp(val, "blue") == 0)
323 		return COLOR_BLUE;
324 	if (strcasecmp(val, "magenta") == 0)
325 		return COLOR_MAGENTA;
326 	if (strcasecmp(val, "cyan") == 0)
327 		return COLOR_CYAN;
328 	if (strcasecmp(val, "white") == 0)
329 		return COLOR_WHITE;
330 	if (strcasecmp(val, "default") == 0)
331 		return -1;
332 
333 	return default_color_value(envvar);
334 }
335 
336 struct tog_diff_view_state {
337 	struct got_object_id *id1, *id2;
338 	const char *label1, *label2;
339 	FILE *f, *f1, *f2;
340 	int fd1, fd2;
341 	int lineno;
342 	int first_displayed_line;
343 	int last_displayed_line;
344 	int eof;
345 	int diff_context;
346 	int ignore_whitespace;
347 	int force_text_diff;
348 	struct got_repository *repo;
349 	struct got_diff_line *lines;
350 	size_t nlines;
351 	int matched_line;
352 	int selected_line;
353 
354 	/* passed from log or blame view; may be NULL */
355 	struct tog_view *parent_view;
356 };
357 
358 pthread_mutex_t tog_mutex = PTHREAD_MUTEX_INITIALIZER;
359 static volatile sig_atomic_t tog_thread_error;
360 
361 struct tog_log_thread_args {
362 	pthread_cond_t need_commits;
363 	pthread_cond_t commit_loaded;
364 	int commits_needed;
365 	int load_all;
366 	struct got_commit_graph *graph;
367 	struct commit_queue *real_commits;
368 	const char *in_repo_path;
369 	struct got_object_id *start_id;
370 	struct got_repository *repo;
371 	int *pack_fds;
372 	int log_complete;
373 	pthread_cond_t log_loaded;
374 	sig_atomic_t *quit;
375 	struct commit_queue_entry **first_displayed_entry;
376 	struct commit_queue_entry **selected_entry;
377 	int *searching;
378 	int *search_next_done;
379 	regex_t *regex;
380 	int *limiting;
381 	int limit_match;
382 	regex_t *limit_regex;
383 	struct commit_queue *limit_commits;
384 	struct got_worktree *worktree;
385 	int need_commit_marker;
386 };
387 
388 struct tog_log_view_state {
389 	struct commit_queue *commits;
390 	struct commit_queue_entry *first_displayed_entry;
391 	struct commit_queue_entry *last_displayed_entry;
392 	struct commit_queue_entry *selected_entry;
393 	struct commit_queue real_commits;
394 	int selected;
395 	char *in_repo_path;
396 	char *head_ref_name;
397 	int log_branches;
398 	struct got_repository *repo;
399 	struct got_object_id *start_id;
400 	sig_atomic_t quit;
401 	pthread_t thread;
402 	struct tog_log_thread_args thread_args;
403 	struct commit_queue_entry *matched_entry;
404 	struct commit_queue_entry *search_entry;
405 	struct tog_colors colors;
406 	int use_committer;
407 	int limit_view;
408 	regex_t limit_regex;
409 	struct commit_queue limit_commits;
410 };
411 
412 #define TOG_COLOR_DIFF_MINUS		1
413 #define TOG_COLOR_DIFF_PLUS		2
414 #define TOG_COLOR_DIFF_CHUNK_HEADER	3
415 #define TOG_COLOR_DIFF_META		4
416 #define TOG_COLOR_TREE_SUBMODULE	5
417 #define TOG_COLOR_TREE_SYMLINK		6
418 #define TOG_COLOR_TREE_DIRECTORY	7
419 #define TOG_COLOR_TREE_EXECUTABLE	8
420 #define TOG_COLOR_COMMIT		9
421 #define TOG_COLOR_AUTHOR		10
422 #define TOG_COLOR_DATE			11
423 #define TOG_COLOR_REFS_HEADS		12
424 #define TOG_COLOR_REFS_TAGS		13
425 #define TOG_COLOR_REFS_REMOTES		14
426 #define TOG_COLOR_REFS_BACKUP		15
427 
428 struct tog_blame_cb_args {
429 	struct tog_blame_line *lines; /* one per line */
430 	int nlines;
431 
432 	struct tog_view *view;
433 	struct got_object_id *commit_id;
434 	int *quit;
435 };
436 
437 struct tog_blame_thread_args {
438 	const char *path;
439 	struct got_repository *repo;
440 	struct tog_blame_cb_args *cb_args;
441 	int *complete;
442 	got_cancel_cb cancel_cb;
443 	void *cancel_arg;
444 	pthread_cond_t blame_complete;
445 };
446 
447 struct tog_blame {
448 	FILE *f;
449 	off_t filesize;
450 	struct tog_blame_line *lines;
451 	int nlines;
452 	off_t *line_offsets;
453 	pthread_t thread;
454 	struct tog_blame_thread_args thread_args;
455 	struct tog_blame_cb_args cb_args;
456 	const char *path;
457 	int *pack_fds;
458 };
459 
460 struct tog_blame_view_state {
461 	int first_displayed_line;
462 	int last_displayed_line;
463 	int selected_line;
464 	int last_diffed_line;
465 	int blame_complete;
466 	int eof;
467 	int done;
468 	struct got_object_id_queue blamed_commits;
469 	struct got_object_qid *blamed_commit;
470 	char *path;
471 	struct got_repository *repo;
472 	struct got_object_id *commit_id;
473 	struct got_object_id *id_to_log;
474 	struct tog_blame blame;
475 	int matched_line;
476 	struct tog_colors colors;
477 };
478 
479 struct tog_parent_tree {
480 	TAILQ_ENTRY(tog_parent_tree) entry;
481 	struct got_tree_object *tree;
482 	struct got_tree_entry *first_displayed_entry;
483 	struct got_tree_entry *selected_entry;
484 	int selected;
485 };
486 
487 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
488 
489 struct tog_tree_view_state {
490 	char *tree_label;
491 	struct got_object_id *commit_id;/* commit which this tree belongs to */
492 	struct got_tree_object *root;	/* the commit's root tree entry */
493 	struct got_tree_object *tree;	/* currently displayed (sub-)tree */
494 	struct got_tree_entry *first_displayed_entry;
495 	struct got_tree_entry *last_displayed_entry;
496 	struct got_tree_entry *selected_entry;
497 	int ndisplayed, selected, show_ids;
498 	struct tog_parent_trees parents; /* parent trees of current sub-tree */
499 	char *head_ref_name;
500 	struct got_repository *repo;
501 	struct got_tree_entry *matched_entry;
502 	struct tog_colors colors;
503 };
504 
505 struct tog_reflist_entry {
506 	TAILQ_ENTRY(tog_reflist_entry) entry;
507 	struct got_reference *ref;
508 	int idx;
509 };
510 
511 TAILQ_HEAD(tog_reflist_head, tog_reflist_entry);
512 
513 struct tog_ref_view_state {
514 	struct tog_reflist_head refs;
515 	struct tog_reflist_entry *first_displayed_entry;
516 	struct tog_reflist_entry *last_displayed_entry;
517 	struct tog_reflist_entry *selected_entry;
518 	int nrefs, ndisplayed, selected, show_date, show_ids, sort_by_date;
519 	struct got_repository *repo;
520 	struct tog_reflist_entry *matched_entry;
521 	struct tog_colors colors;
522 };
523 
524 struct tog_help_view_state {
525 	FILE			*f;
526 	off_t			*line_offsets;
527 	size_t			 nlines;
528 	int			 lineno;
529 	int			 first_displayed_line;
530 	int			 last_displayed_line;
531 	int			 eof;
532 	int			 matched_line;
533 	int			 selected_line;
534 	int			 all;
535 	enum tog_keymap_type	 type;
536 };
537 
538 #define GENERATE_HELP \
539 	KEYMAP_("Global", TOG_KEYMAP_GLOBAL), \
540 	KEY_("H F1", "Open view-specific help (double tap for all help)"), \
541 	KEY_("k C-p Up", "Move cursor or page up one line"), \
542 	KEY_("j C-n Down", "Move cursor or page down one line"), \
543 	KEY_("C-b b PgUp", "Scroll the view up one page"), \
544 	KEY_("C-f f PgDn Space", "Scroll the view down one page"), \
545 	KEY_("C-u u", "Scroll the view up one half page"), \
546 	KEY_("C-d d", "Scroll the view down one half page"), \
547 	KEY_("g", "Go to line N (default: first line)"), \
548 	KEY_("Home =", "Go to the first line"), \
549 	KEY_("G", "Go to line N (default: last line)"), \
550 	KEY_("End *", "Go to the last line"), \
551 	KEY_("l Right", "Scroll the view right"), \
552 	KEY_("h Left", "Scroll the view left"), \
553 	KEY_("$", "Scroll view to the rightmost position"), \
554 	KEY_("0", "Scroll view to the leftmost position"), \
555 	KEY_("-", "Decrease size of the focussed split"), \
556 	KEY_("+", "Increase size of the focussed split"), \
557 	KEY_("Tab", "Switch focus between views"), \
558 	KEY_("F", "Toggle fullscreen mode"), \
559 	KEY_("S", "Switch split-screen layout"), \
560 	KEY_("/", "Open prompt to enter search term"), \
561 	KEY_("n", "Find next line/token matching the current search term"), \
562 	KEY_("N", "Find previous line/token matching the current search term"),\
563 	KEY_("q", "Quit the focussed view; Quit help screen"), \
564 	KEY_("Q", "Quit tog"), \
565 	\
566 	KEYMAP_("Log view", TOG_KEYMAP_LOG), \
567 	KEY_("< ,", "Move cursor up one commit"), \
568 	KEY_("> .", "Move cursor down one commit"), \
569 	KEY_("Enter", "Open diff view of the selected commit"), \
570 	KEY_("B", "Reload the log view and toggle display of merged commits"), \
571 	KEY_("R", "Open ref view of all repository references"), \
572 	KEY_("T", "Display tree view of the repository from the selected" \
573 	    " commit"), \
574 	KEY_("@", "Toggle between displaying author and committer name"), \
575 	KEY_("&", "Open prompt to enter term to limit commits displayed"), \
576 	KEY_("C-g Backspace", "Cancel current search or log operation"), \
577 	KEY_("C-l", "Reload the log view with new commits in the repository"), \
578 	\
579 	KEYMAP_("Diff view", TOG_KEYMAP_DIFF), \
580 	KEY_("K < ,", "Display diff of next line in the file/log entry"), \
581 	KEY_("J > .", "Display diff of previous line in the file/log entry"), \
582 	KEY_("A", "Toggle between Myers and Patience diff algorithm"), \
583 	KEY_("a", "Toggle treatment of file as ASCII irrespective of binary" \
584 	    " data"), \
585 	KEY_("(", "Go to the previous file in the diff"), \
586 	KEY_(")", "Go to the next file in the diff"), \
587 	KEY_("{", "Go to the previous hunk in the diff"), \
588 	KEY_("}", "Go to the next hunk in the diff"), \
589 	KEY_("[", "Decrease the number of context lines"), \
590 	KEY_("]", "Increase the number of context lines"), \
591 	KEY_("w", "Toggle ignore whitespace-only changes in the diff"), \
592 	\
593 	KEYMAP_("Blame view", TOG_KEYMAP_BLAME), \
594 	KEY_("Enter", "Display diff view of the selected line's commit"), \
595 	KEY_("A", "Toggle diff algorithm between Myers and Patience"), \
596 	KEY_("L", "Open log view for the currently selected annotated line"), \
597 	KEY_("C", "Reload view with the previously blamed commit"), \
598 	KEY_("c", "Reload view with the version of the file found in the" \
599 	    " selected line's commit"), \
600 	KEY_("p", "Reload view with the version of the file found in the" \
601 	    " selected line's parent commit"), \
602 	\
603 	KEYMAP_("Tree view", TOG_KEYMAP_TREE), \
604 	KEY_("Enter", "Enter selected directory or open blame view of the" \
605 	    " selected file"), \
606 	KEY_("L", "Open log view for the selected entry"), \
607 	KEY_("R", "Open ref view of all repository references"), \
608 	KEY_("i", "Show object IDs for all tree entries"), \
609 	KEY_("Backspace", "Return to the parent directory"), \
610 	\
611 	KEYMAP_("Ref view", TOG_KEYMAP_REF), \
612 	KEY_("Enter", "Display log view of the selected reference"), \
613 	KEY_("T", "Display tree view of the selected reference"), \
614 	KEY_("i", "Toggle display of IDs for all non-symbolic references"), \
615 	KEY_("m", "Toggle display of last modified date for each reference"), \
616 	KEY_("o", "Toggle reference sort order (name -> timestamp)"), \
617 	KEY_("C-l", "Reload view with all repository references")
618 
619 struct tog_key_map {
620 	const char		*keys;
621 	const char		*info;
622 	enum tog_keymap_type	 type;
623 };
624 
625 /* curses io for tog regress */
626 struct tog_io {
627 	FILE	*cin;
628 	FILE	*cout;
629 	FILE	*f;
630 	FILE	*sdump;
631 	char	*input_str;
632 	int	 wait_for_ui;
633 } tog_io;
634 static int using_mock_io;
635 
636 #define TOG_KEY_SCRDUMP	SHRT_MIN
637 
638 /*
639  * We implement two types of views: parent views and child views.
640  *
641  * The 'Tab' key switches focus between a parent view and its child view.
642  * Child views are shown side-by-side to their parent view, provided
643  * there is enough screen estate.
644  *
645  * When a new view is opened from within a parent view, this new view
646  * becomes a child view of the parent view, replacing any existing child.
647  *
648  * When a new view is opened from within a child view, this new view
649  * becomes a parent view which will obscure the views below until the
650  * user quits the new parent view by typing 'q'.
651  *
652  * This list of views contains parent views only.
653  * Child views are only pointed to by their parent view.
654  */
655 TAILQ_HEAD(tog_view_list_head, tog_view);
656 
657 struct tog_view {
658 	TAILQ_ENTRY(tog_view) entry;
659 	WINDOW *window;
660 	PANEL *panel;
661 	int nlines, ncols, begin_y, begin_x; /* based on split height/width */
662 	int resized_y, resized_x; /* begin_y/x based on user resizing */
663 	int maxx, x; /* max column and current start column */
664 	int lines, cols; /* copies of LINES and COLS */
665 	int nscrolled, offset; /* lines scrolled and hsplit line offset */
666 	int gline, hiline; /* navigate to and highlight this nG line */
667 	int ch, count; /* current keymap and count prefix */
668 	int resized; /* set when in a resize event */
669 	int focussed; /* Only set on one parent or child view at a time. */
670 	int dying;
671 	struct tog_view *parent;
672 	struct tog_view *child;
673 
674 	/*
675 	 * This flag is initially set on parent views when a new child view
676 	 * is created. It gets toggled when the 'Tab' key switches focus
677 	 * between parent and child.
678 	 * The flag indicates whether focus should be passed on to our child
679 	 * view if this parent view gets picked for focus after another parent
680 	 * view was closed. This prevents child views from losing focus in such
681 	 * situations.
682 	 */
683 	int focus_child;
684 
685 	enum tog_view_mode mode;
686 	/* type-specific state */
687 	enum tog_view_type type;
688 	union {
689 		struct tog_diff_view_state diff;
690 		struct tog_log_view_state log;
691 		struct tog_blame_view_state blame;
692 		struct tog_tree_view_state tree;
693 		struct tog_ref_view_state ref;
694 		struct tog_help_view_state help;
695 	} state;
696 
697 	const struct got_error *(*show)(struct tog_view *);
698 	const struct got_error *(*input)(struct tog_view **,
699 	    struct tog_view *, int);
700 	const struct got_error *(*reset)(struct tog_view *);
701 	const struct got_error *(*resize)(struct tog_view *, int);
702 	const struct got_error *(*close)(struct tog_view *);
703 
704 	const struct got_error *(*search_start)(struct tog_view *);
705 	const struct got_error *(*search_next)(struct tog_view *);
706 	void (*search_setup)(struct tog_view *, FILE **, off_t **, size_t *,
707 	    int **, int **, int **, int **);
708 	int search_started;
709 	int searching;
710 #define TOG_SEARCH_FORWARD	1
711 #define TOG_SEARCH_BACKWARD	2
712 	int search_next_done;
713 #define TOG_SEARCH_HAVE_MORE	1
714 #define TOG_SEARCH_NO_MORE	2
715 #define TOG_SEARCH_HAVE_NONE	3
716 	regex_t regex;
717 	regmatch_t regmatch;
718 	const char *action;
719 };
720 
721 static const struct got_error *open_diff_view(struct tog_view *,
722     struct got_object_id *, struct got_object_id *,
723     const char *, const char *, int, int, int, struct tog_view *,
724     struct got_repository *);
725 static const struct got_error *show_diff_view(struct tog_view *);
726 static const struct got_error *input_diff_view(struct tog_view **,
727     struct tog_view *, int);
728 static const struct got_error *reset_diff_view(struct tog_view *);
729 static const struct got_error* close_diff_view(struct tog_view *);
730 static const struct got_error *search_start_diff_view(struct tog_view *);
731 static void search_setup_diff_view(struct tog_view *, FILE **, off_t **,
732     size_t *, int **, int **, int **, int **);
733 static const struct got_error *search_next_view_match(struct tog_view *);
734 
735 static const struct got_error *open_log_view(struct tog_view *,
736     struct got_object_id *, struct got_repository *,
737     const char *, const char *, int, struct got_worktree *);
738 static const struct got_error * show_log_view(struct tog_view *);
739 static const struct got_error *input_log_view(struct tog_view **,
740     struct tog_view *, int);
741 static const struct got_error *resize_log_view(struct tog_view *, int);
742 static const struct got_error *close_log_view(struct tog_view *);
743 static const struct got_error *search_start_log_view(struct tog_view *);
744 static const struct got_error *search_next_log_view(struct tog_view *);
745 
746 static const struct got_error *open_blame_view(struct tog_view *, char *,
747     struct got_object_id *, struct got_repository *);
748 static const struct got_error *show_blame_view(struct tog_view *);
749 static const struct got_error *input_blame_view(struct tog_view **,
750     struct tog_view *, int);
751 static const struct got_error *reset_blame_view(struct tog_view *);
752 static const struct got_error *close_blame_view(struct tog_view *);
753 static const struct got_error *search_start_blame_view(struct tog_view *);
754 static void search_setup_blame_view(struct tog_view *, FILE **, off_t **,
755     size_t *, int **, int **, int **, int **);
756 
757 static const struct got_error *open_tree_view(struct tog_view *,
758     struct got_object_id *, const char *, struct got_repository *);
759 static const struct got_error *show_tree_view(struct tog_view *);
760 static const struct got_error *input_tree_view(struct tog_view **,
761     struct tog_view *, int);
762 static const struct got_error *close_tree_view(struct tog_view *);
763 static const struct got_error *search_start_tree_view(struct tog_view *);
764 static const struct got_error *search_next_tree_view(struct tog_view *);
765 
766 static const struct got_error *open_ref_view(struct tog_view *,
767     struct got_repository *);
768 static const struct got_error *show_ref_view(struct tog_view *);
769 static const struct got_error *input_ref_view(struct tog_view **,
770     struct tog_view *, int);
771 static const struct got_error *close_ref_view(struct tog_view *);
772 static const struct got_error *search_start_ref_view(struct tog_view *);
773 static const struct got_error *search_next_ref_view(struct tog_view *);
774 
775 static const struct got_error *open_help_view(struct tog_view *,
776     struct tog_view *);
777 static const struct got_error *show_help_view(struct tog_view *);
778 static const struct got_error *input_help_view(struct tog_view **,
779     struct tog_view *, int);
780 static const struct got_error *reset_help_view(struct tog_view *);
781 static const struct got_error* close_help_view(struct tog_view *);
782 static const struct got_error *search_start_help_view(struct tog_view *);
783 static void search_setup_help_view(struct tog_view *, FILE **, off_t **,
784     size_t *, int **, int **, int **, int **);
785 
786 static volatile sig_atomic_t tog_sigwinch_received;
787 static volatile sig_atomic_t tog_sigpipe_received;
788 static volatile sig_atomic_t tog_sigcont_received;
789 static volatile sig_atomic_t tog_sigint_received;
790 static volatile sig_atomic_t tog_sigterm_received;
791 
792 static void
793 tog_sigwinch(int signo)
794 {
795 	tog_sigwinch_received = 1;
796 }
797 
798 static void
799 tog_sigpipe(int signo)
800 {
801 	tog_sigpipe_received = 1;
802 }
803 
804 static void
805 tog_sigcont(int signo)
806 {
807 	tog_sigcont_received = 1;
808 }
809 
810 static void
811 tog_sigint(int signo)
812 {
813 	tog_sigint_received = 1;
814 }
815 
816 static void
817 tog_sigterm(int signo)
818 {
819 	tog_sigterm_received = 1;
820 }
821 
822 static int
823 tog_fatal_signal_received(void)
824 {
825 	return (tog_sigpipe_received ||
826 	    tog_sigint_received || tog_sigterm_received);
827 }
828 
829 static const struct got_error *
830 view_close(struct tog_view *view)
831 {
832 	const struct got_error *err = NULL, *child_err = NULL;
833 
834 	if (view->child) {
835 		child_err = view_close(view->child);
836 		view->child = NULL;
837 	}
838 	if (view->close)
839 		err = view->close(view);
840 	if (view->panel)
841 		del_panel(view->panel);
842 	if (view->window)
843 		delwin(view->window);
844 	free(view);
845 	return err ? err : child_err;
846 }
847 
848 static struct tog_view *
849 view_open(int nlines, int ncols, int begin_y, int begin_x,
850     enum tog_view_type type)
851 {
852 	struct tog_view *view = calloc(1, sizeof(*view));
853 
854 	if (view == NULL)
855 		return NULL;
856 
857 	view->type = type;
858 	view->lines = LINES;
859 	view->cols = COLS;
860 	view->nlines = nlines ? nlines : LINES - begin_y;
861 	view->ncols = ncols ? ncols : COLS - begin_x;
862 	view->begin_y = begin_y;
863 	view->begin_x = begin_x;
864 	view->window = newwin(nlines, ncols, begin_y, begin_x);
865 	if (view->window == NULL) {
866 		view_close(view);
867 		return NULL;
868 	}
869 	view->panel = new_panel(view->window);
870 	if (view->panel == NULL ||
871 	    set_panel_userptr(view->panel, view) != OK) {
872 		view_close(view);
873 		return NULL;
874 	}
875 
876 	keypad(view->window, TRUE);
877 	return view;
878 }
879 
880 static int
881 view_split_begin_x(int begin_x)
882 {
883 	if (begin_x > 0 || COLS < 120)
884 		return 0;
885 	return (COLS - MAX(COLS / 2, 80));
886 }
887 
888 /* XXX Stub till we decide what to do. */
889 static int
890 view_split_begin_y(int lines)
891 {
892 	return lines * HSPLIT_SCALE;
893 }
894 
895 static const struct got_error *view_resize(struct tog_view *);
896 
897 static const struct got_error *
898 view_splitscreen(struct tog_view *view)
899 {
900 	const struct got_error *err = NULL;
901 
902 	if (!view->resized && view->mode == TOG_VIEW_SPLIT_HRZN) {
903 		if (view->resized_y && view->resized_y < view->lines)
904 			view->begin_y = view->resized_y;
905 		else
906 			view->begin_y = view_split_begin_y(view->nlines);
907 		view->begin_x = 0;
908 	} else if (!view->resized) {
909 		if (view->resized_x && view->resized_x < view->cols - 1 &&
910 		    view->cols > 119)
911 			view->begin_x = view->resized_x;
912 		else
913 			view->begin_x = view_split_begin_x(0);
914 		view->begin_y = 0;
915 	}
916 	view->nlines = LINES - view->begin_y;
917 	view->ncols = COLS - view->begin_x;
918 	view->lines = LINES;
919 	view->cols = COLS;
920 	err = view_resize(view);
921 	if (err)
922 		return err;
923 
924 	if (view->parent && view->mode == TOG_VIEW_SPLIT_HRZN)
925 		view->parent->nlines = view->begin_y;
926 
927 	if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
928 		return got_error_from_errno("mvwin");
929 
930 	return NULL;
931 }
932 
933 static const struct got_error *
934 view_fullscreen(struct tog_view *view)
935 {
936 	const struct got_error *err = NULL;
937 
938 	view->begin_x = 0;
939 	view->begin_y = view->resized ? view->begin_y : 0;
940 	view->nlines = view->resized ? view->nlines : LINES;
941 	view->ncols = COLS;
942 	view->lines = LINES;
943 	view->cols = COLS;
944 	err = view_resize(view);
945 	if (err)
946 		return err;
947 
948 	if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
949 		return got_error_from_errno("mvwin");
950 
951 	return NULL;
952 }
953 
954 static int
955 view_is_parent_view(struct tog_view *view)
956 {
957 	return view->parent == NULL;
958 }
959 
960 static int
961 view_is_splitscreen(struct tog_view *view)
962 {
963 	return view->begin_x > 0 || view->begin_y > 0;
964 }
965 
966 static int
967 view_is_fullscreen(struct tog_view *view)
968 {
969 	return view->nlines == LINES && view->ncols == COLS;
970 }
971 
972 static int
973 view_is_hsplit_top(struct tog_view *view)
974 {
975 	return view->mode == TOG_VIEW_SPLIT_HRZN && view->child &&
976 	    view_is_splitscreen(view->child);
977 }
978 
979 static void
980 view_border(struct tog_view *view)
981 {
982 	PANEL *panel;
983 	const struct tog_view *view_above;
984 
985 	if (view->parent)
986 		return view_border(view->parent);
987 
988 	panel = panel_above(view->panel);
989 	if (panel == NULL)
990 		return;
991 
992 	view_above = panel_userptr(panel);
993 	if (view->mode == TOG_VIEW_SPLIT_HRZN)
994 		mvwhline(view->window, view_above->begin_y - 1,
995 		    view->begin_x, ACS_HLINE, view->ncols);
996 	else
997 		mvwvline(view->window, view->begin_y, view_above->begin_x - 1,
998 		    ACS_VLINE, view->nlines);
999 }
1000 
1001 static const struct got_error *view_init_hsplit(struct tog_view *, int);
1002 static const struct got_error *request_log_commits(struct tog_view *);
1003 static const struct got_error *offset_selection_down(struct tog_view *);
1004 static void offset_selection_up(struct tog_view *);
1005 static void view_get_split(struct tog_view *, int *, int *);
1006 
1007 static const struct got_error *
1008 view_resize(struct tog_view *view)
1009 {
1010 	const struct got_error	*err = NULL;
1011 	int			 dif, nlines, ncols;
1012 
1013 	dif = LINES - view->lines;  /* line difference */
1014 
1015 	if (view->lines > LINES)
1016 		nlines = view->nlines - (view->lines - LINES);
1017 	else
1018 		nlines = view->nlines + (LINES - view->lines);
1019 	if (view->cols > COLS)
1020 		ncols = view->ncols - (view->cols - COLS);
1021 	else
1022 		ncols = view->ncols + (COLS - view->cols);
1023 
1024 	if (view->child) {
1025 		int hs = view->child->begin_y;
1026 
1027 		if (!view_is_fullscreen(view))
1028 			view->child->begin_x = view_split_begin_x(view->begin_x);
1029 		if (view->mode == TOG_VIEW_SPLIT_HRZN ||
1030 		    view->child->begin_x == 0) {
1031 			ncols = COLS;
1032 
1033 			view_fullscreen(view->child);
1034 			if (view->child->focussed)
1035 				show_panel(view->child->panel);
1036 			else
1037 				show_panel(view->panel);
1038 		} else {
1039 			ncols = view->child->begin_x;
1040 
1041 			view_splitscreen(view->child);
1042 			show_panel(view->child->panel);
1043 		}
1044 		/*
1045 		 * XXX This is ugly and needs to be moved into the above
1046 		 * logic but "works" for now and my attempts at moving it
1047 		 * break either 'tab' or 'F' key maps in horizontal splits.
1048 		 */
1049 		if (hs) {
1050 			err = view_splitscreen(view->child);
1051 			if (err)
1052 				return err;
1053 			if (dif < 0) { /* top split decreased */
1054 				err = offset_selection_down(view);
1055 				if (err)
1056 					return err;
1057 			}
1058 			view_border(view);
1059 			update_panels();
1060 			doupdate();
1061 			show_panel(view->child->panel);
1062 			nlines = view->nlines;
1063 		}
1064 	} else if (view->parent == NULL)
1065 		ncols = COLS;
1066 
1067 	if (view->resize && dif > 0) {
1068 		err = view->resize(view, dif);
1069 		if (err)
1070 			return err;
1071 	}
1072 
1073 	if (wresize(view->window, nlines, ncols) == ERR)
1074 		return got_error_from_errno("wresize");
1075 	if (replace_panel(view->panel, view->window) == ERR)
1076 		return got_error_from_errno("replace_panel");
1077 	wclear(view->window);
1078 
1079 	view->nlines = nlines;
1080 	view->ncols = ncols;
1081 	view->lines = LINES;
1082 	view->cols = COLS;
1083 
1084 	return NULL;
1085 }
1086 
1087 static const struct got_error *
1088 resize_log_view(struct tog_view *view, int increase)
1089 {
1090 	struct tog_log_view_state	*s = &view->state.log;
1091 	const struct got_error		*err = NULL;
1092 	int				 n = 0;
1093 
1094 	if (s->selected_entry)
1095 		n = s->selected_entry->idx + view->lines - s->selected;
1096 
1097 	/*
1098 	 * Request commits to account for the increased
1099 	 * height so we have enough to populate the view.
1100 	 */
1101 	if (s->commits->ncommits < n) {
1102 		view->nscrolled = n - s->commits->ncommits + increase + 1;
1103 		err = request_log_commits(view);
1104 	}
1105 
1106 	return err;
1107 }
1108 
1109 static void
1110 view_adjust_offset(struct tog_view *view, int n)
1111 {
1112 	if (n == 0)
1113 		return;
1114 
1115 	if (view->parent && view->parent->offset) {
1116 		if (view->parent->offset + n >= 0)
1117 			view->parent->offset += n;
1118 		else
1119 			view->parent->offset = 0;
1120 	} else if (view->offset) {
1121 		if (view->offset - n >= 0)
1122 			view->offset -= n;
1123 		else
1124 			view->offset = 0;
1125 	}
1126 }
1127 
1128 static const struct got_error *
1129 view_resize_split(struct tog_view *view, int resize)
1130 {
1131 	const struct got_error	*err = NULL;
1132 	struct tog_view		*v = NULL;
1133 
1134 	if (view->parent)
1135 		v = view->parent;
1136 	else
1137 		v = view;
1138 
1139 	if (!v->child || !view_is_splitscreen(v->child))
1140 		return NULL;
1141 
1142 	v->resized = v->child->resized = resize;  /* lock for resize event */
1143 
1144 	if (view->mode == TOG_VIEW_SPLIT_HRZN) {
1145 		if (v->child->resized_y)
1146 			v->child->begin_y = v->child->resized_y;
1147 		if (view->parent)
1148 			v->child->begin_y -= resize;
1149 		else
1150 			v->child->begin_y += resize;
1151 		if (v->child->begin_y < 3) {
1152 			view->count = 0;
1153 			v->child->begin_y = 3;
1154 		} else if (v->child->begin_y > LINES - 1) {
1155 			view->count = 0;
1156 			v->child->begin_y = LINES - 1;
1157 		}
1158 		v->ncols = COLS;
1159 		v->child->ncols = COLS;
1160 		view_adjust_offset(view, resize);
1161 		err = view_init_hsplit(v, v->child->begin_y);
1162 		if (err)
1163 			return err;
1164 		v->child->resized_y = v->child->begin_y;
1165 	} else {
1166 		if (v->child->resized_x)
1167 			v->child->begin_x = v->child->resized_x;
1168 		if (view->parent)
1169 			v->child->begin_x -= resize;
1170 		else
1171 			v->child->begin_x += resize;
1172 		if (v->child->begin_x < 11) {
1173 			view->count = 0;
1174 			v->child->begin_x = 11;
1175 		} else if (v->child->begin_x > COLS - 1) {
1176 			view->count = 0;
1177 			v->child->begin_x = COLS - 1;
1178 		}
1179 		v->child->resized_x = v->child->begin_x;
1180 	}
1181 
1182 	v->child->mode = v->mode;
1183 	v->child->nlines = v->lines - v->child->begin_y;
1184 	v->child->ncols = v->cols - v->child->begin_x;
1185 	v->focus_child = 1;
1186 
1187 	err = view_fullscreen(v);
1188 	if (err)
1189 		return err;
1190 	err = view_splitscreen(v->child);
1191 	if (err)
1192 		return err;
1193 
1194 	if (v->mode == TOG_VIEW_SPLIT_HRZN) {
1195 		err = offset_selection_down(v->child);
1196 		if (err)
1197 			return err;
1198 	}
1199 
1200 	if (v->resize)
1201 		err = v->resize(v, 0);
1202 	else if (v->child->resize)
1203 		err = v->child->resize(v->child, 0);
1204 
1205 	v->resized = v->child->resized = 0;
1206 
1207 	return err;
1208 }
1209 
1210 static void
1211 view_transfer_size(struct tog_view *dst, struct tog_view *src)
1212 {
1213 	struct tog_view *v = src->child ? src->child : src;
1214 
1215 	dst->resized_x = v->resized_x;
1216 	dst->resized_y = v->resized_y;
1217 }
1218 
1219 static const struct got_error *
1220 view_close_child(struct tog_view *view)
1221 {
1222 	const struct got_error *err = NULL;
1223 
1224 	if (view->child == NULL)
1225 		return NULL;
1226 
1227 	err = view_close(view->child);
1228 	view->child = NULL;
1229 	return err;
1230 }
1231 
1232 static const struct got_error *
1233 view_set_child(struct tog_view *view, struct tog_view *child)
1234 {
1235 	const struct got_error *err = NULL;
1236 
1237 	view->child = child;
1238 	child->parent = view;
1239 
1240 	err = view_resize(view);
1241 	if (err)
1242 		return err;
1243 
1244 	if (view->child->resized_x || view->child->resized_y)
1245 		err = view_resize_split(view, 0);
1246 
1247 	return err;
1248 }
1249 
1250 static const struct got_error *view_dispatch_request(struct tog_view **,
1251     struct tog_view *, enum tog_view_type, int, int);
1252 
1253 static const struct got_error *
1254 view_request_new(struct tog_view **requested, struct tog_view *view,
1255     enum tog_view_type request)
1256 {
1257 	struct tog_view		*new_view = NULL;
1258 	const struct got_error	*err;
1259 	int			 y = 0, x = 0;
1260 
1261 	*requested = NULL;
1262 
1263 	if (view_is_parent_view(view) && request != TOG_VIEW_HELP)
1264 		view_get_split(view, &y, &x);
1265 
1266 	err = view_dispatch_request(&new_view, view, request, y, x);
1267 	if (err)
1268 		return err;
1269 
1270 	if (view_is_parent_view(view) && view->mode == TOG_VIEW_SPLIT_HRZN &&
1271 	    request != TOG_VIEW_HELP) {
1272 		err = view_init_hsplit(view, y);
1273 		if (err)
1274 			return err;
1275 	}
1276 
1277 	view->focussed = 0;
1278 	new_view->focussed = 1;
1279 	new_view->mode = view->mode;
1280 	new_view->nlines = request == TOG_VIEW_HELP ?
1281 	    view->lines : view->lines - y;
1282 
1283 	if (view_is_parent_view(view) && request != TOG_VIEW_HELP) {
1284 		view_transfer_size(new_view, view);
1285 		err = view_close_child(view);
1286 		if (err)
1287 			return err;
1288 		err = view_set_child(view, new_view);
1289 		if (err)
1290 			return err;
1291 		view->focus_child = 1;
1292 	} else
1293 		*requested = new_view;
1294 
1295 	return NULL;
1296 }
1297 
1298 static void
1299 tog_resizeterm(void)
1300 {
1301 	int cols, lines;
1302 	struct winsize size;
1303 
1304 	if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &size) < 0) {
1305 		cols = 80;     /* Default */
1306 		lines = 24;
1307 	} else {
1308 		cols = size.ws_col;
1309 		lines = size.ws_row;
1310 	}
1311 	resize_term(lines, cols);
1312 }
1313 
1314 static const struct got_error *
1315 view_search_start(struct tog_view *view, int fast_refresh)
1316 {
1317 	const struct got_error *err = NULL;
1318 	struct tog_view *v = view;
1319 	char pattern[1024];
1320 	int ret;
1321 
1322 	if (view->search_started) {
1323 		regfree(&view->regex);
1324 		view->searching = 0;
1325 		memset(&view->regmatch, 0, sizeof(view->regmatch));
1326 	}
1327 	view->search_started = 0;
1328 
1329 	if (view->nlines < 1)
1330 		return NULL;
1331 
1332 	if (view_is_hsplit_top(view))
1333 		v = view->child;
1334 	else if (view->mode == TOG_VIEW_SPLIT_VERT && view->parent)
1335 		v = view->parent;
1336 
1337 	if (tog_io.input_str != NULL) {
1338 		if (strlcpy(pattern, tog_io.input_str, sizeof(pattern)) >=
1339 		    sizeof(pattern))
1340 			return got_error(GOT_ERR_NO_SPACE);
1341 	} else {
1342 		mvwaddstr(v->window, v->nlines - 1, 0, "/");
1343 		wclrtoeol(v->window);
1344 		nodelay(v->window, FALSE);  /* block for search term input */
1345 		nocbreak();
1346 		echo();
1347 		ret = wgetnstr(v->window, pattern, sizeof(pattern));
1348 		wrefresh(v->window);
1349 		cbreak();
1350 		noecho();
1351 		nodelay(v->window, TRUE);
1352 		if (!fast_refresh && !using_mock_io)
1353 			halfdelay(10);
1354 		if (ret == ERR)
1355 			return NULL;
1356 	}
1357 
1358 	if (regcomp(&view->regex, pattern, REG_EXTENDED | REG_NEWLINE) == 0) {
1359 		err = view->search_start(view);
1360 		if (err) {
1361 			regfree(&view->regex);
1362 			return err;
1363 		}
1364 		view->search_started = 1;
1365 		view->searching = TOG_SEARCH_FORWARD;
1366 		view->search_next_done = 0;
1367 		view->search_next(view);
1368 	}
1369 
1370 	return NULL;
1371 }
1372 
1373 /* Switch split mode. If view is a parent or child, draw the new splitscreen. */
1374 static const struct got_error *
1375 switch_split(struct tog_view *view)
1376 {
1377 	const struct got_error	*err = NULL;
1378 	struct tog_view		*v = NULL;
1379 
1380 	if (view->parent)
1381 		v = view->parent;
1382 	else
1383 		v = view;
1384 
1385 	if (v->mode == TOG_VIEW_SPLIT_HRZN)
1386 		v->mode = TOG_VIEW_SPLIT_VERT;
1387 	else
1388 		v->mode = TOG_VIEW_SPLIT_HRZN;
1389 
1390 	if (!v->child)
1391 		return NULL;
1392 	else if (v->mode == TOG_VIEW_SPLIT_VERT && v->cols < 120)
1393 		v->mode = TOG_VIEW_SPLIT_NONE;
1394 
1395 	view_get_split(v, &v->child->begin_y, &v->child->begin_x);
1396 	if (v->mode == TOG_VIEW_SPLIT_HRZN && v->child->resized_y)
1397 		v->child->begin_y = v->child->resized_y;
1398 	else if (v->mode == TOG_VIEW_SPLIT_VERT && v->child->resized_x)
1399 		v->child->begin_x = v->child->resized_x;
1400 
1401 
1402 	if (v->mode == TOG_VIEW_SPLIT_HRZN) {
1403 		v->ncols = COLS;
1404 		v->child->ncols = COLS;
1405 		v->child->nscrolled = LINES - v->child->nlines;
1406 
1407 		err = view_init_hsplit(v, v->child->begin_y);
1408 		if (err)
1409 			return err;
1410 	}
1411 	v->child->mode = v->mode;
1412 	v->child->nlines = v->lines - v->child->begin_y;
1413 	v->focus_child = 1;
1414 
1415 	err = view_fullscreen(v);
1416 	if (err)
1417 		return err;
1418 	err = view_splitscreen(v->child);
1419 	if (err)
1420 		return err;
1421 
1422 	if (v->mode == TOG_VIEW_SPLIT_NONE)
1423 		v->mode = TOG_VIEW_SPLIT_VERT;
1424 	if (v->mode == TOG_VIEW_SPLIT_HRZN) {
1425 		err = offset_selection_down(v);
1426 		if (err)
1427 			return err;
1428 		err = offset_selection_down(v->child);
1429 		if (err)
1430 			return err;
1431 	} else {
1432 		offset_selection_up(v);
1433 		offset_selection_up(v->child);
1434 	}
1435 	if (v->resize)
1436 		err = v->resize(v, 0);
1437 	else if (v->child->resize)
1438 		err = v->child->resize(v->child, 0);
1439 
1440 	return err;
1441 }
1442 
1443 /*
1444  * Strip trailing whitespace from str starting at byte *n;
1445  * if *n < 0, use strlen(str). Return new str length in *n.
1446  */
1447 static void
1448 strip_trailing_ws(char *str, int *n)
1449 {
1450 	size_t x = *n;
1451 
1452 	if (str == NULL || *str == '\0')
1453 		return;
1454 
1455 	if (x < 0)
1456 		x = strlen(str);
1457 
1458 	while (x-- > 0 && isspace((unsigned char)str[x]))
1459 		str[x] = '\0';
1460 
1461 	*n = x + 1;
1462 }
1463 
1464 /*
1465  * Extract visible substring of line y from the curses screen
1466  * and strip trailing whitespace. If vline is set, overwrite
1467  * line[vline] with '|' because the ACS_VLINE character is
1468  * written out as 'x'. Write the line to file f.
1469  */
1470 static const struct got_error *
1471 view_write_line(FILE *f, int y, int vline)
1472 {
1473 	char	line[COLS * MB_LEN_MAX];  /* allow for multibyte chars */
1474 	int	r, w;
1475 
1476 	r = mvwinnstr(curscr, y, 0, line, sizeof(line));
1477 	if (r == ERR)
1478 		return got_error_fmt(GOT_ERR_RANGE,
1479 		    "failed to extract line %d", y);
1480 
1481 	/*
1482 	 * In some views, lines are padded with blanks to COLS width.
1483 	 * Strip them so we can diff without the -b flag when testing.
1484 	 */
1485 	strip_trailing_ws(line, &r);
1486 
1487 	if (vline > 0)
1488 		line[vline] = '|';
1489 
1490 	w = fprintf(f, "%s\n", line);
1491 	if (w != r + 1)		/* \n */
1492 		return got_ferror(f, GOT_ERR_IO);
1493 
1494 	return NULL;
1495 }
1496 
1497 /*
1498  * Capture the visible curses screen by writing each line to the
1499  * file at the path set via the TOG_SCR_DUMP environment variable.
1500  */
1501 static const struct got_error *
1502 screendump(struct tog_view *view)
1503 {
1504 	const struct got_error	*err;
1505 	int			 i;
1506 
1507 	err = got_opentemp_truncate(tog_io.sdump);
1508 	if (err)
1509 		return err;
1510 
1511 	if ((view->child && view->child->begin_x) ||
1512 	    (view->parent && view->begin_x)) {
1513 		int ncols = view->child ? view->ncols : view->parent->ncols;
1514 
1515 		/* vertical splitscreen */
1516 		for (i = 0; i < view->nlines; ++i) {
1517 			err = view_write_line(tog_io.sdump, i, ncols - 1);
1518 			if (err)
1519 				goto done;
1520 		}
1521 	} else {
1522 		int hline = 0;
1523 
1524 		/* fullscreen or horizontal splitscreen */
1525 		if ((view->child && view->child->begin_y) ||
1526 		    (view->parent && view->begin_y))	/* hsplit */
1527 			hline = view->child ?
1528 			    view->child->begin_y : view->begin_y;
1529 
1530 		for (i = 0; i < view->lines; i++) {
1531 			if (hline && i == hline - 1) {
1532 				int c;
1533 
1534 				/* ACS_HLINE writes out as 'q', overwrite it */
1535 				for (c = 0; c < view->cols; ++c)
1536 					fputc('-', tog_io.sdump);
1537 				fputc('\n', tog_io.sdump);
1538 				continue;
1539 			}
1540 
1541 			err = view_write_line(tog_io.sdump, i, 0);
1542 			if (err)
1543 				goto done;
1544 		}
1545 	}
1546 
1547 done:
1548 	return err;
1549 }
1550 
1551 /*
1552  * Compute view->count from numeric input. Assign total to view->count and
1553  * return first non-numeric key entered.
1554  */
1555 static int
1556 get_compound_key(struct tog_view *view, int c)
1557 {
1558 	struct tog_view	*v = view;
1559 	int		 x, n = 0;
1560 
1561 	if (view_is_hsplit_top(view))
1562 		v = view->child;
1563 	else if (view->mode == TOG_VIEW_SPLIT_VERT && view->parent)
1564 		v = view->parent;
1565 
1566 	view->count = 0;
1567 	cbreak();  /* block for input */
1568 	nodelay(view->window, FALSE);
1569 	wmove(v->window, v->nlines - 1, 0);
1570 	wclrtoeol(v->window);
1571 	waddch(v->window, ':');
1572 
1573 	do {
1574 		x = getcurx(v->window);
1575 		if (x != ERR && x < view->ncols) {
1576 			waddch(v->window, c);
1577 			wrefresh(v->window);
1578 		}
1579 
1580 		/*
1581 		 * Don't overflow. Max valid request should be the greatest
1582 		 * between the longest and total lines; cap at 10 million.
1583 		 */
1584 		if (n >= 9999999)
1585 			n = 9999999;
1586 		else
1587 			n = n * 10 + (c - '0');
1588 	} while (((c = wgetch(view->window))) >= '0' && c <= '9' && c != ERR);
1589 
1590 	if (c == 'G' || c == 'g') {	/* nG key map */
1591 		view->gline = view->hiline = n;
1592 		n = 0;
1593 		c = 0;
1594 	}
1595 
1596 	/* Massage excessive or inapplicable values at the input handler. */
1597 	view->count = n;
1598 
1599 	return c;
1600 }
1601 
1602 static void
1603 action_report(struct tog_view *view)
1604 {
1605 	struct tog_view *v = view;
1606 
1607 	if (view_is_hsplit_top(view))
1608 		v = view->child;
1609 	else if (view->mode == TOG_VIEW_SPLIT_VERT && view->parent)
1610 		v = view->parent;
1611 
1612 	wmove(v->window, v->nlines - 1, 0);
1613 	wclrtoeol(v->window);
1614 	wprintw(v->window, ":%s", view->action);
1615 	wrefresh(v->window);
1616 
1617 	/*
1618 	 * Clear action status report. Only clear in blame view
1619 	 * once annotating is complete, otherwise it's too fast.
1620 	 */
1621 	if (view->type == TOG_VIEW_BLAME) {
1622 		if (view->state.blame.blame_complete)
1623 			view->action = NULL;
1624 	} else
1625 		view->action = NULL;
1626 }
1627 
1628 /*
1629  * Read the next line from the test script and assign
1630  * key instruction to *ch. If at EOF, set the *done flag.
1631  */
1632 static const struct got_error *
1633 tog_read_script_key(FILE *script, struct tog_view *view, int *ch, int *done)
1634 {
1635 	const struct got_error	*err = NULL;
1636 	char			*line = NULL;
1637 	size_t			 linesz = 0;
1638 	ssize_t			 n;
1639 
1640 
1641 	if (view->count && --view->count) {
1642 		*ch = view->ch;
1643 		return NULL;
1644 	} else
1645 		*ch = -1;
1646 
1647 	if ((n = getline(&line, &linesz, script)) == -1) {
1648 		if (feof(script)) {
1649 			*done = 1;
1650 			goto done;
1651 		} else {
1652 			err = got_ferror(script, GOT_ERR_IO);
1653 			goto done;
1654 		}
1655 	}
1656 
1657 	if (strncasecmp(line, "WAIT_FOR_UI", 11) == 0)
1658 		tog_io.wait_for_ui = 1;
1659 	else if (strncasecmp(line, "KEY_ENTER", 9) == 0)
1660 		*ch = KEY_ENTER;
1661 	else if (strncasecmp(line, "KEY_RIGHT", 9) == 0)
1662 		*ch = KEY_RIGHT;
1663 	else if (strncasecmp(line, "KEY_LEFT", 8) == 0)
1664 		*ch = KEY_LEFT;
1665 	else if (strncasecmp(line, "KEY_DOWN", 8) == 0)
1666 		*ch = KEY_DOWN;
1667 	else if (strncasecmp(line, "KEY_UP", 6) == 0)
1668 		*ch = KEY_UP;
1669 	else if (strncasecmp(line, "TAB", 3) == 0)
1670 		*ch = '\t';
1671 	else if (strncasecmp(line, "SCREENDUMP", 10) == 0)
1672 		*ch = TOG_KEY_SCRDUMP;
1673 	else if (isdigit((unsigned char)*line)) {
1674 		char *t = line;
1675 
1676 		while (isdigit((unsigned char)*t))
1677 			++t;
1678 		view->ch = *ch = *t;
1679 		*t = '\0';
1680 		/* ignore error, view->count is 0 if instruction is invalid */
1681 		view->count = strtonum(line, 0, INT_MAX, NULL);
1682 	} else {
1683 		*ch = *line;
1684 		if (n > 2 && (*ch == '/' || *ch == '&')) {
1685 			/* skip leading keymap and trim trailing newline */
1686 			tog_io.input_str = strndup(line + 1, n - 2);
1687 			if (tog_io.input_str == NULL) {
1688 				err = got_error_from_errno("strndup");
1689 				goto done;
1690 			}
1691 		}
1692 	}
1693 
1694 done:
1695 	free(line);
1696 	return err;
1697 }
1698 
1699 static const struct got_error *
1700 view_input(struct tog_view **new, int *done, struct tog_view *view,
1701     struct tog_view_list_head *views, int fast_refresh)
1702 {
1703 	const struct got_error *err = NULL;
1704 	struct tog_view *v;
1705 	int ch, errcode;
1706 
1707 	*new = NULL;
1708 
1709 	if (view->action)
1710 		action_report(view);
1711 
1712 	/* Clear "no matches" indicator. */
1713 	if (view->search_next_done == TOG_SEARCH_NO_MORE ||
1714 	    view->search_next_done == TOG_SEARCH_HAVE_NONE) {
1715 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
1716 		view->count = 0;
1717 	}
1718 
1719 	if (view->searching && !view->search_next_done) {
1720 		errcode = pthread_mutex_unlock(&tog_mutex);
1721 		if (errcode)
1722 			return got_error_set_errno(errcode,
1723 			    "pthread_mutex_unlock");
1724 		sched_yield();
1725 		errcode = pthread_mutex_lock(&tog_mutex);
1726 		if (errcode)
1727 			return got_error_set_errno(errcode,
1728 			    "pthread_mutex_lock");
1729 		view->search_next(view);
1730 		return NULL;
1731 	}
1732 
1733 	/* Allow threads to make progress while we are waiting for input. */
1734 	errcode = pthread_mutex_unlock(&tog_mutex);
1735 	if (errcode)
1736 		return got_error_set_errno(errcode, "pthread_mutex_unlock");
1737 
1738 	if (using_mock_io) {
1739 		err = tog_read_script_key(tog_io.f, view, &ch, done);
1740 		if (err) {
1741 			errcode = pthread_mutex_lock(&tog_mutex);
1742 			return err;
1743 		}
1744 	} else if (view->count && --view->count) {
1745 		cbreak();
1746 		nodelay(view->window, TRUE);
1747 		ch = wgetch(view->window);
1748 		/* let C-g or backspace abort unfinished count */
1749 		if (ch == CTRL('g') || ch == KEY_BACKSPACE)
1750 			view->count = 0;
1751 		else
1752 			ch = view->ch;
1753 	} else {
1754 		ch = wgetch(view->window);
1755 		if (ch >= '1' && ch  <= '9')
1756 			view->ch = ch = get_compound_key(view, ch);
1757 	}
1758 	if (view->hiline && ch != ERR && ch != 0)
1759 		view->hiline = 0;  /* key pressed, clear line highlight */
1760 	nodelay(view->window, TRUE);
1761 	errcode = pthread_mutex_lock(&tog_mutex);
1762 	if (errcode)
1763 		return got_error_set_errno(errcode, "pthread_mutex_lock");
1764 
1765 	if (tog_sigwinch_received || tog_sigcont_received) {
1766 		tog_resizeterm();
1767 		tog_sigwinch_received = 0;
1768 		tog_sigcont_received = 0;
1769 		TAILQ_FOREACH(v, views, entry) {
1770 			err = view_resize(v);
1771 			if (err)
1772 				return err;
1773 			err = v->input(new, v, KEY_RESIZE);
1774 			if (err)
1775 				return err;
1776 			if (v->child) {
1777 				err = view_resize(v->child);
1778 				if (err)
1779 					return err;
1780 				err = v->child->input(new, v->child,
1781 				    KEY_RESIZE);
1782 				if (err)
1783 					return err;
1784 				if (v->child->resized_x || v->child->resized_y) {
1785 					err = view_resize_split(v, 0);
1786 					if (err)
1787 						return err;
1788 				}
1789 			}
1790 		}
1791 	}
1792 
1793 	switch (ch) {
1794 	case '?':
1795 	case 'H':
1796 	case KEY_F(1):
1797 		if (view->type == TOG_VIEW_HELP)
1798 			err = view->reset(view);
1799 		else
1800 			err = view_request_new(new, view, TOG_VIEW_HELP);
1801 		break;
1802 	case '\t':
1803 		view->count = 0;
1804 		if (view->child) {
1805 			view->focussed = 0;
1806 			view->child->focussed = 1;
1807 			view->focus_child = 1;
1808 		} else if (view->parent) {
1809 			view->focussed = 0;
1810 			view->parent->focussed = 1;
1811 			view->parent->focus_child = 0;
1812 			if (!view_is_splitscreen(view)) {
1813 				if (view->parent->resize) {
1814 					err = view->parent->resize(view->parent,
1815 					    0);
1816 					if (err)
1817 						return err;
1818 				}
1819 				offset_selection_up(view->parent);
1820 				err = view_fullscreen(view->parent);
1821 				if (err)
1822 					return err;
1823 			}
1824 		}
1825 		break;
1826 	case 'q':
1827 		if (view->parent && view->mode == TOG_VIEW_SPLIT_HRZN) {
1828 			if (view->parent->resize) {
1829 				/* might need more commits to fill fullscreen */
1830 				err = view->parent->resize(view->parent, 0);
1831 				if (err)
1832 					break;
1833 			}
1834 			offset_selection_up(view->parent);
1835 		}
1836 		err = view->input(new, view, ch);
1837 		view->dying = 1;
1838 		break;
1839 	case 'Q':
1840 		*done = 1;
1841 		break;
1842 	case 'F':
1843 		view->count = 0;
1844 		if (view_is_parent_view(view)) {
1845 			if (view->child == NULL)
1846 				break;
1847 			if (view_is_splitscreen(view->child)) {
1848 				view->focussed = 0;
1849 				view->child->focussed = 1;
1850 				err = view_fullscreen(view->child);
1851 			} else {
1852 				err = view_splitscreen(view->child);
1853 				if (!err)
1854 					err = view_resize_split(view, 0);
1855 			}
1856 			if (err)
1857 				break;
1858 			err = view->child->input(new, view->child,
1859 			    KEY_RESIZE);
1860 		} else {
1861 			if (view_is_splitscreen(view)) {
1862 				view->parent->focussed = 0;
1863 				view->focussed = 1;
1864 				err = view_fullscreen(view);
1865 			} else {
1866 				err = view_splitscreen(view);
1867 				if (!err && view->mode != TOG_VIEW_SPLIT_HRZN)
1868 					err = view_resize(view->parent);
1869 				if (!err)
1870 					err = view_resize_split(view, 0);
1871 			}
1872 			if (err)
1873 				break;
1874 			err = view->input(new, view, KEY_RESIZE);
1875 		}
1876 		if (err)
1877 			break;
1878 		if (view->resize) {
1879 			err = view->resize(view, 0);
1880 			if (err)
1881 				break;
1882 		}
1883 		if (view->parent) {
1884 			if (view->parent->resize) {
1885 				err = view->parent->resize(view->parent, 0);
1886 				if (err != NULL)
1887 					break;
1888 			}
1889 			err = offset_selection_down(view->parent);
1890 			if (err != NULL)
1891 				break;
1892 		}
1893 		err = offset_selection_down(view);
1894 		break;
1895 	case 'S':
1896 		view->count = 0;
1897 		err = switch_split(view);
1898 		break;
1899 	case '-':
1900 		err = view_resize_split(view, -1);
1901 		break;
1902 	case '+':
1903 		err = view_resize_split(view, 1);
1904 		break;
1905 	case KEY_RESIZE:
1906 		break;
1907 	case '/':
1908 		view->count = 0;
1909 		if (view->search_start)
1910 			view_search_start(view, fast_refresh);
1911 		else
1912 			err = view->input(new, view, ch);
1913 		break;
1914 	case 'N':
1915 	case 'n':
1916 		if (view->search_started && view->search_next) {
1917 			view->searching = (ch == 'n' ?
1918 			    TOG_SEARCH_FORWARD : TOG_SEARCH_BACKWARD);
1919 			view->search_next_done = 0;
1920 			view->search_next(view);
1921 		} else
1922 			err = view->input(new, view, ch);
1923 		break;
1924 	case 'A':
1925 		if (tog_diff_algo == GOT_DIFF_ALGORITHM_MYERS) {
1926 			tog_diff_algo = GOT_DIFF_ALGORITHM_PATIENCE;
1927 			view->action = "Patience diff algorithm";
1928 		} else {
1929 			tog_diff_algo = GOT_DIFF_ALGORITHM_MYERS;
1930 			view->action = "Myers diff algorithm";
1931 		}
1932 		TAILQ_FOREACH(v, views, entry) {
1933 			if (v->reset) {
1934 				err = v->reset(v);
1935 				if (err)
1936 					return err;
1937 			}
1938 			if (v->child && v->child->reset) {
1939 				err = v->child->reset(v->child);
1940 				if (err)
1941 					return err;
1942 			}
1943 		}
1944 		break;
1945 	case TOG_KEY_SCRDUMP:
1946 		err = screendump(view);
1947 		break;
1948 	default:
1949 		err = view->input(new, view, ch);
1950 		break;
1951 	}
1952 
1953 	return err;
1954 }
1955 
1956 static int
1957 view_needs_focus_indication(struct tog_view *view)
1958 {
1959 	if (view_is_parent_view(view)) {
1960 		if (view->child == NULL || view->child->focussed)
1961 			return 0;
1962 		if (!view_is_splitscreen(view->child))
1963 			return 0;
1964 	} else if (!view_is_splitscreen(view))
1965 		return 0;
1966 
1967 	return view->focussed;
1968 }
1969 
1970 static const struct got_error *
1971 tog_io_close(void)
1972 {
1973 	const struct got_error *err = NULL;
1974 
1975 	if (tog_io.cin && fclose(tog_io.cin) == EOF)
1976 		err = got_ferror(tog_io.cin, GOT_ERR_IO);
1977 	if (tog_io.cout && fclose(tog_io.cout) == EOF && err == NULL)
1978 		err = got_ferror(tog_io.cout, GOT_ERR_IO);
1979 	if (tog_io.f && fclose(tog_io.f) == EOF && err == NULL)
1980 		err = got_ferror(tog_io.f, GOT_ERR_IO);
1981 	if (tog_io.sdump && fclose(tog_io.sdump) == EOF && err == NULL)
1982 		err = got_ferror(tog_io.sdump, GOT_ERR_IO);
1983 	if (tog_io.input_str != NULL)
1984 		free(tog_io.input_str);
1985 
1986 	return err;
1987 }
1988 
1989 static const struct got_error *
1990 view_loop(struct tog_view *view)
1991 {
1992 	const struct got_error *err = NULL;
1993 	struct tog_view_list_head views;
1994 	struct tog_view *new_view;
1995 	char *mode;
1996 	int fast_refresh = 10;
1997 	int done = 0, errcode;
1998 
1999 	mode = getenv("TOG_VIEW_SPLIT_MODE");
2000 	if (!mode || !(*mode == 'h' || *mode == 'H'))
2001 		view->mode = TOG_VIEW_SPLIT_VERT;
2002 	else
2003 		view->mode = TOG_VIEW_SPLIT_HRZN;
2004 
2005 	errcode = pthread_mutex_lock(&tog_mutex);
2006 	if (errcode)
2007 		return got_error_set_errno(errcode, "pthread_mutex_lock");
2008 
2009 	TAILQ_INIT(&views);
2010 	TAILQ_INSERT_HEAD(&views, view, entry);
2011 
2012 	view->focussed = 1;
2013 	err = view->show(view);
2014 	if (err)
2015 		return err;
2016 	update_panels();
2017 	doupdate();
2018 	while (!TAILQ_EMPTY(&views) && !done && !tog_thread_error &&
2019 	    !tog_fatal_signal_received()) {
2020 		/* Refresh fast during initialization, then become slower. */
2021 		if (fast_refresh && --fast_refresh == 0 && !using_mock_io)
2022 			halfdelay(10); /* switch to once per second */
2023 
2024 		err = view_input(&new_view, &done, view, &views, fast_refresh);
2025 		if (err)
2026 			break;
2027 
2028 		if (view->dying && view == TAILQ_FIRST(&views) &&
2029 		    TAILQ_NEXT(view, entry) == NULL)
2030 			done = 1;
2031 		if (done) {
2032 			struct tog_view *v;
2033 
2034 			/*
2035 			 * When we quit, scroll the screen up a single line
2036 			 * so we don't lose any information.
2037 			 */
2038 			TAILQ_FOREACH(v, &views, entry) {
2039 				wmove(v->window, 0, 0);
2040 				wdeleteln(v->window);
2041 				wnoutrefresh(v->window);
2042 				if (v->child && !view_is_fullscreen(v)) {
2043 					wmove(v->child->window, 0, 0);
2044 					wdeleteln(v->child->window);
2045 					wnoutrefresh(v->child->window);
2046 				}
2047 			}
2048 			doupdate();
2049 		}
2050 
2051 		if (view->dying) {
2052 			struct tog_view *v, *prev = NULL;
2053 
2054 			if (view_is_parent_view(view))
2055 				prev = TAILQ_PREV(view, tog_view_list_head,
2056 				    entry);
2057 			else if (view->parent)
2058 				prev = view->parent;
2059 
2060 			if (view->parent) {
2061 				view->parent->child = NULL;
2062 				view->parent->focus_child = 0;
2063 				/* Restore fullscreen line height. */
2064 				view->parent->nlines = view->parent->lines;
2065 				err = view_resize(view->parent);
2066 				if (err)
2067 					break;
2068 				/* Make resized splits persist. */
2069 				view_transfer_size(view->parent, view);
2070 			} else
2071 				TAILQ_REMOVE(&views, view, entry);
2072 
2073 			err = view_close(view);
2074 			if (err)
2075 				goto done;
2076 
2077 			view = NULL;
2078 			TAILQ_FOREACH(v, &views, entry) {
2079 				if (v->focussed)
2080 					break;
2081 			}
2082 			if (view == NULL && new_view == NULL) {
2083 				/* No view has focus. Try to pick one. */
2084 				if (prev)
2085 					view = prev;
2086 				else if (!TAILQ_EMPTY(&views)) {
2087 					view = TAILQ_LAST(&views,
2088 					    tog_view_list_head);
2089 				}
2090 				if (view) {
2091 					if (view->focus_child) {
2092 						view->child->focussed = 1;
2093 						view = view->child;
2094 					} else
2095 						view->focussed = 1;
2096 				}
2097 			}
2098 		}
2099 		if (new_view) {
2100 			struct tog_view *v, *t;
2101 			/* Only allow one parent view per type. */
2102 			TAILQ_FOREACH_SAFE(v, &views, entry, t) {
2103 				if (v->type != new_view->type)
2104 					continue;
2105 				TAILQ_REMOVE(&views, v, entry);
2106 				err = view_close(v);
2107 				if (err)
2108 					goto done;
2109 				break;
2110 			}
2111 			TAILQ_INSERT_TAIL(&views, new_view, entry);
2112 			view = new_view;
2113 		}
2114 		if (view && !done) {
2115 			if (view_is_parent_view(view)) {
2116 				if (view->child && view->child->focussed)
2117 					view = view->child;
2118 			} else {
2119 				if (view->parent && view->parent->focussed)
2120 					view = view->parent;
2121 			}
2122 			show_panel(view->panel);
2123 			if (view->child && view_is_splitscreen(view->child))
2124 				show_panel(view->child->panel);
2125 			if (view->parent && view_is_splitscreen(view)) {
2126 				err = view->parent->show(view->parent);
2127 				if (err)
2128 					goto done;
2129 			}
2130 			err = view->show(view);
2131 			if (err)
2132 				goto done;
2133 			if (view->child) {
2134 				err = view->child->show(view->child);
2135 				if (err)
2136 					goto done;
2137 			}
2138 			update_panels();
2139 			doupdate();
2140 		}
2141 	}
2142 done:
2143 	while (!TAILQ_EMPTY(&views)) {
2144 		const struct got_error *close_err;
2145 		view = TAILQ_FIRST(&views);
2146 		TAILQ_REMOVE(&views, view, entry);
2147 		close_err = view_close(view);
2148 		if (close_err && err == NULL)
2149 			err = close_err;
2150 	}
2151 
2152 	errcode = pthread_mutex_unlock(&tog_mutex);
2153 	if (errcode && err == NULL)
2154 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
2155 
2156 	return err;
2157 }
2158 
2159 __dead static void
2160 usage_log(void)
2161 {
2162 	endwin();
2163 	fprintf(stderr,
2164 	    "usage: %s log [-b] [-c commit] [-r repository-path] [path]\n",
2165 	    getprogname());
2166 	exit(1);
2167 }
2168 
2169 /* Create newly allocated wide-character string equivalent to a byte string. */
2170 static const struct got_error *
2171 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
2172 {
2173 	char *vis = NULL;
2174 	const struct got_error *err = NULL;
2175 
2176 	*ws = NULL;
2177 	*wlen = mbstowcs(NULL, s, 0);
2178 	if (*wlen == (size_t)-1) {
2179 		int vislen;
2180 		if (errno != EILSEQ)
2181 			return got_error_from_errno("mbstowcs");
2182 
2183 		/* byte string invalid in current encoding; try to "fix" it */
2184 		err = got_mbsavis(&vis, &vislen, s);
2185 		if (err)
2186 			return err;
2187 		*wlen = mbstowcs(NULL, vis, 0);
2188 		if (*wlen == (size_t)-1) {
2189 			err = got_error_from_errno("mbstowcs"); /* give up */
2190 			goto done;
2191 		}
2192 	}
2193 
2194 	*ws = calloc(*wlen + 1, sizeof(**ws));
2195 	if (*ws == NULL) {
2196 		err = got_error_from_errno("calloc");
2197 		goto done;
2198 	}
2199 
2200 	if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
2201 		err = got_error_from_errno("mbstowcs");
2202 done:
2203 	free(vis);
2204 	if (err) {
2205 		free(*ws);
2206 		*ws = NULL;
2207 		*wlen = 0;
2208 	}
2209 	return err;
2210 }
2211 
2212 static const struct got_error *
2213 expand_tab(char **ptr, const char *src)
2214 {
2215 	char	*dst;
2216 	size_t	 len, n, idx = 0, sz = 0;
2217 
2218 	*ptr = NULL;
2219 	n = len = strlen(src);
2220 	dst = malloc(n + 1);
2221 	if (dst == NULL)
2222 		return got_error_from_errno("malloc");
2223 
2224 	while (idx < len && src[idx]) {
2225 		const char c = src[idx];
2226 
2227 		if (c == '\t') {
2228 			size_t nb = TABSIZE - sz % TABSIZE;
2229 			char *p;
2230 
2231 			p = realloc(dst, n + nb);
2232 			if (p == NULL) {
2233 				free(dst);
2234 				return got_error_from_errno("realloc");
2235 
2236 			}
2237 			dst = p;
2238 			n += nb;
2239 			memset(dst + sz, ' ', nb);
2240 			sz += nb;
2241 		} else
2242 			dst[sz++] = src[idx];
2243 		++idx;
2244 	}
2245 
2246 	dst[sz] = '\0';
2247 	*ptr = dst;
2248 	return NULL;
2249 }
2250 
2251 /*
2252  * Advance at most n columns from wline starting at offset off.
2253  * Return the index to the first character after the span operation.
2254  * Return the combined column width of all spanned wide characters in
2255  * *rcol.
2256  */
2257 static int
2258 span_wline(int *rcol, int off, wchar_t *wline, int n, int col_tab_align)
2259 {
2260 	int width, i, cols = 0;
2261 
2262 	if (n == 0) {
2263 		*rcol = cols;
2264 		return off;
2265 	}
2266 
2267 	for (i = off; wline[i] != L'\0'; ++i) {
2268 		if (wline[i] == L'\t')
2269 			width = TABSIZE - ((cols + col_tab_align) % TABSIZE);
2270 		else
2271 			width = wcwidth(wline[i]);
2272 
2273 		if (width == -1) {
2274 			width = 1;
2275 			wline[i] = L'.';
2276 		}
2277 
2278 		if (cols + width > n)
2279 			break;
2280 		cols += width;
2281 	}
2282 
2283 	*rcol = cols;
2284 	return i;
2285 }
2286 
2287 /*
2288  * Format a line for display, ensuring that it won't overflow a width limit.
2289  * With scrolling, the width returned refers to the scrolled version of the
2290  * line, which starts at (*wlinep)[*scrollxp]. The caller must free *wlinep.
2291  */
2292 static const struct got_error *
2293 format_line(wchar_t **wlinep, int *widthp, int *scrollxp,
2294     const char *line, int nscroll, int wlimit, int col_tab_align, int expand)
2295 {
2296 	const struct got_error *err = NULL;
2297 	int cols;
2298 	wchar_t *wline = NULL;
2299 	char *exstr = NULL;
2300 	size_t wlen;
2301 	int i, scrollx;
2302 
2303 	*wlinep = NULL;
2304 	*widthp = 0;
2305 
2306 	if (expand) {
2307 		err = expand_tab(&exstr, line);
2308 		if (err)
2309 			return err;
2310 	}
2311 
2312 	err = mbs2ws(&wline, &wlen, expand ? exstr : line);
2313 	free(exstr);
2314 	if (err)
2315 		return err;
2316 
2317 	scrollx = span_wline(&cols, 0, wline, nscroll, col_tab_align);
2318 
2319 	if (wlen > 0 && wline[wlen - 1] == L'\n') {
2320 		wline[wlen - 1] = L'\0';
2321 		wlen--;
2322 	}
2323 	if (wlen > 0 && wline[wlen - 1] == L'\r') {
2324 		wline[wlen - 1] = L'\0';
2325 		wlen--;
2326 	}
2327 
2328 	i = span_wline(&cols, scrollx, wline, wlimit, col_tab_align);
2329 	wline[i] = L'\0';
2330 
2331 	if (widthp)
2332 		*widthp = cols;
2333 	if (scrollxp)
2334 		*scrollxp = scrollx;
2335 	if (err)
2336 		free(wline);
2337 	else
2338 		*wlinep = wline;
2339 	return err;
2340 }
2341 
2342 static const struct got_error*
2343 build_refs_str(char **refs_str, struct got_reflist_head *refs,
2344     struct got_object_id *id, struct got_repository *repo)
2345 {
2346 	static const struct got_error *err = NULL;
2347 	struct got_reflist_entry *re;
2348 	char *s;
2349 	const char *name;
2350 
2351 	*refs_str = NULL;
2352 
2353 	if (refs == NULL)
2354 		return NULL;
2355 
2356 	TAILQ_FOREACH(re, refs, entry) {
2357 		struct got_tag_object *tag = NULL;
2358 		struct got_object_id *ref_id;
2359 		int cmp;
2360 
2361 		name = got_ref_get_name(re->ref);
2362 		if (strcmp(name, GOT_REF_HEAD) == 0)
2363 			continue;
2364 		if (strncmp(name, "refs/", 5) == 0)
2365 			name += 5;
2366 		if (strncmp(name, "got/", 4) == 0)
2367 			continue;
2368 		if (strncmp(name, "heads/", 6) == 0)
2369 			name += 6;
2370 		if (strncmp(name, "remotes/", 8) == 0) {
2371 			name += 8;
2372 			s = strstr(name, "/" GOT_REF_HEAD);
2373 			if (s != NULL && strcmp(s, "/" GOT_REF_HEAD) == 0)
2374 				continue;
2375 		}
2376 		err = got_ref_resolve(&ref_id, repo, re->ref);
2377 		if (err)
2378 			break;
2379 		if (strncmp(name, "tags/", 5) == 0) {
2380 			err = got_object_open_as_tag(&tag, repo, ref_id);
2381 			if (err) {
2382 				if (err->code != GOT_ERR_OBJ_TYPE) {
2383 					free(ref_id);
2384 					break;
2385 				}
2386 				/* Ref points at something other than a tag. */
2387 				err = NULL;
2388 				tag = NULL;
2389 			}
2390 		}
2391 		cmp = got_object_id_cmp(tag ?
2392 		    got_object_tag_get_object_id(tag) : ref_id, id);
2393 		free(ref_id);
2394 		if (tag)
2395 			got_object_tag_close(tag);
2396 		if (cmp != 0)
2397 			continue;
2398 		s = *refs_str;
2399 		if (asprintf(refs_str, "%s%s%s", s ? s : "",
2400 		    s ? ", " : "", name) == -1) {
2401 			err = got_error_from_errno("asprintf");
2402 			free(s);
2403 			*refs_str = NULL;
2404 			break;
2405 		}
2406 		free(s);
2407 	}
2408 
2409 	return err;
2410 }
2411 
2412 static const struct got_error *
2413 format_author(wchar_t **wauthor, int *author_width, char *author, int limit,
2414     int col_tab_align)
2415 {
2416 	char *smallerthan;
2417 
2418 	smallerthan = strchr(author, '<');
2419 	if (smallerthan && smallerthan[1] != '\0')
2420 		author = smallerthan + 1;
2421 	author[strcspn(author, "@>")] = '\0';
2422 	return format_line(wauthor, author_width, NULL, author, 0, limit,
2423 	    col_tab_align, 0);
2424 }
2425 
2426 static const struct got_error *
2427 draw_commit(struct tog_view *view, struct commit_queue_entry *entry,
2428     const size_t date_display_cols, int author_display_cols)
2429 {
2430 	struct tog_log_view_state *s = &view->state.log;
2431 	const struct got_error *err = NULL;
2432 	struct got_commit_object *commit = entry->commit;
2433 	struct got_object_id *id = entry->id;
2434 	char datebuf[12]; /* YYYY-MM-DD + SPACE + NUL */
2435 	char *refs_str = NULL;
2436 	char *logmsg0 = NULL, *logmsg = NULL;
2437 	char *author = NULL;
2438 	wchar_t *wrefstr = NULL, *wlogmsg = NULL, *wauthor = NULL;
2439 	int author_width, refstr_width, logmsg_width;
2440 	char *newline, *line = NULL;
2441 	int col, limit, scrollx, logmsg_x;
2442 	const int avail = view->ncols, marker_column = author_display_cols + 1;
2443 	struct tm tm;
2444 	time_t committer_time;
2445 	struct tog_color *tc;
2446 	struct got_reflist_head *refs;
2447 
2448 	if (tog_base_commit.id != NULL && tog_base_commit.idx == -1 &&
2449 	    got_object_id_cmp(id, tog_base_commit.id) == 0)
2450 		tog_base_commit.idx = entry->idx;
2451 	if (tog_io.wait_for_ui && s->thread_args.need_commit_marker) {
2452 		int rc;
2453 
2454 		rc = pthread_cond_wait(&s->thread_args.log_loaded, &tog_mutex);
2455 		if (rc)
2456 			return got_error_set_errno(rc, "pthread_cond_wait");
2457 	}
2458 
2459 	committer_time = got_object_commit_get_committer_time(commit);
2460 	if (gmtime_r(&committer_time, &tm) == NULL)
2461 		return got_error_from_errno("gmtime_r");
2462 	if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d ", &tm) == 0)
2463 		return got_error(GOT_ERR_NO_SPACE);
2464 
2465 	if (avail <= date_display_cols)
2466 		limit = MIN(sizeof(datebuf) - 1, avail);
2467 	else
2468 		limit = MIN(date_display_cols, sizeof(datebuf) - 1);
2469 	tc = get_color(&s->colors, TOG_COLOR_DATE);
2470 	if (tc)
2471 		wattr_on(view->window,
2472 		    COLOR_PAIR(tc->colorpair), NULL);
2473 	waddnstr(view->window, datebuf, limit);
2474 	if (tc)
2475 		wattr_off(view->window,
2476 		    COLOR_PAIR(tc->colorpair), NULL);
2477 	col = limit;
2478 	if (col > avail)
2479 		goto done;
2480 
2481 	if (avail >= 120) {
2482 		char *id_str;
2483 		err = got_object_id_str(&id_str, id);
2484 		if (err)
2485 			goto done;
2486 		tc = get_color(&s->colors, TOG_COLOR_COMMIT);
2487 		if (tc)
2488 			wattr_on(view->window,
2489 			    COLOR_PAIR(tc->colorpair), NULL);
2490 		wprintw(view->window, "%.8s ", id_str);
2491 		if (tc)
2492 			wattr_off(view->window,
2493 			    COLOR_PAIR(tc->colorpair), NULL);
2494 		free(id_str);
2495 		col += 9;
2496 		if (col > avail)
2497 			goto done;
2498 	}
2499 
2500 	if (s->use_committer)
2501 		author = strdup(got_object_commit_get_committer(commit));
2502 	else
2503 		author = strdup(got_object_commit_get_author(commit));
2504 	if (author == NULL) {
2505 		err = got_error_from_errno("strdup");
2506 		goto done;
2507 	}
2508 	err = format_author(&wauthor, &author_width, author, avail - col, col);
2509 	if (err)
2510 		goto done;
2511 	tc = get_color(&s->colors, TOG_COLOR_AUTHOR);
2512 	if (tc)
2513 		wattr_on(view->window,
2514 		    COLOR_PAIR(tc->colorpair), NULL);
2515 	waddwstr(view->window, wauthor);
2516 	col += author_width;
2517 	while (col < avail && author_width < author_display_cols + 2) {
2518 		if (tog_base_commit.marker != GOT_WORKTREE_STATE_UNKNOWN &&
2519 		    author_width == marker_column &&
2520 		    entry->idx == tog_base_commit.idx && !s->limit_view) {
2521 			tc = get_color(&s->colors, TOG_COLOR_COMMIT);
2522 			if (tc)
2523 				wattr_on(view->window,
2524 				    COLOR_PAIR(tc->colorpair), NULL);
2525 			waddch(view->window, tog_base_commit.marker);
2526 			if (tc)
2527 				wattr_off(view->window,
2528 				    COLOR_PAIR(tc->colorpair), NULL);
2529 		} else
2530 			waddch(view->window, ' ');
2531 		col++;
2532 		author_width++;
2533 	}
2534 	if (tc)
2535 		wattr_off(view->window,
2536 		    COLOR_PAIR(tc->colorpair), NULL);
2537 	if (col > avail)
2538 		goto done;
2539 
2540 	err = got_object_commit_get_logmsg(&logmsg0, commit);
2541 	if (err)
2542 		goto done;
2543 	logmsg = logmsg0;
2544 	while (*logmsg == '\n')
2545 		logmsg++;
2546 	newline = strchr(logmsg, '\n');
2547 	if (newline)
2548 		*newline = '\0';
2549 
2550 	limit = avail - col;
2551 	if (view->child && !view_is_hsplit_top(view) && limit > 0)
2552 		limit--;	/* for the border */
2553 
2554 	/* Prepend reference labels to log message if possible .*/
2555 	refs = got_reflist_object_id_map_lookup(tog_refs_idmap, id);
2556 	err = build_refs_str(&refs_str, refs, id, s->repo);
2557 	if (err)
2558 		goto done;
2559 	if (refs_str) {
2560 		char *rs;
2561 
2562 		if (asprintf(&rs, "[%s]", refs_str) == -1) {
2563 			err = got_error_from_errno("asprintf");
2564 			goto done;
2565 		}
2566 		err = format_line(&wrefstr, &refstr_width,
2567 		    &scrollx, rs, view->x, limit, col, 1);
2568 		free(rs);
2569 		if (err)
2570 			goto done;
2571 		tc = get_color(&s->colors, TOG_COLOR_COMMIT);
2572 		if (tc)
2573 			wattr_on(view->window,
2574 			    COLOR_PAIR(tc->colorpair), NULL);
2575 		waddwstr(view->window, &wrefstr[scrollx]);
2576 		if (tc)
2577 			wattr_off(view->window,
2578 			    COLOR_PAIR(tc->colorpair), NULL);
2579 		col += MAX(refstr_width, 0);
2580 		if (col > avail)
2581 			goto done;
2582 
2583 		if (col < avail) {
2584 			waddch(view->window, ' ');
2585 			col++;
2586 		}
2587 
2588 		if (refstr_width > 0)
2589 			logmsg_x = 0;
2590 		else {
2591 			int unscrolled_refstr_width;
2592 			size_t len = wcslen(wrefstr);
2593 
2594 			/*
2595 			 * No need to check for -1 return value here since
2596 			 * unprintables have been replaced by span_wline().
2597 			 */
2598 			unscrolled_refstr_width = wcswidth(wrefstr, len);
2599 			unscrolled_refstr_width += 1; /* trailing space */
2600 			logmsg_x = view->x - unscrolled_refstr_width;
2601 		}
2602 
2603 		limit = avail - col;
2604 		if (view->child && !view_is_hsplit_top(view) && limit > 0)
2605 			limit--;	/* for the border */
2606 	} else
2607 		logmsg_x = view->x;
2608 
2609 	err = format_line(&wlogmsg, &logmsg_width, &scrollx, logmsg, logmsg_x,
2610 	    limit, col, 1);
2611 	if (err)
2612 		goto done;
2613 	waddwstr(view->window, &wlogmsg[scrollx]);
2614 	col += MAX(logmsg_width, 0);
2615 	while (col < avail) {
2616 		waddch(view->window, ' ');
2617 		col++;
2618 	}
2619 done:
2620 	free(logmsg0);
2621 	free(wlogmsg);
2622 	free(wrefstr);
2623 	free(refs_str);
2624 	free(author);
2625 	free(wauthor);
2626 	free(line);
2627 	return err;
2628 }
2629 
2630 static struct commit_queue_entry *
2631 alloc_commit_queue_entry(struct got_commit_object *commit,
2632     struct got_object_id *id)
2633 {
2634 	struct commit_queue_entry *entry;
2635 	struct got_object_id *dup;
2636 
2637 	entry = calloc(1, sizeof(*entry));
2638 	if (entry == NULL)
2639 		return NULL;
2640 
2641 	dup = got_object_id_dup(id);
2642 	if (dup == NULL) {
2643 		free(entry);
2644 		return NULL;
2645 	}
2646 
2647 	entry->id = dup;
2648 	entry->commit = commit;
2649 	return entry;
2650 }
2651 
2652 static void
2653 pop_commit(struct commit_queue *commits)
2654 {
2655 	struct commit_queue_entry *entry;
2656 
2657 	entry = TAILQ_FIRST(&commits->head);
2658 	TAILQ_REMOVE(&commits->head, entry, entry);
2659 	got_object_commit_close(entry->commit);
2660 	commits->ncommits--;
2661 	free(entry->id);
2662 	free(entry);
2663 }
2664 
2665 static void
2666 free_commits(struct commit_queue *commits)
2667 {
2668 	while (!TAILQ_EMPTY(&commits->head))
2669 		pop_commit(commits);
2670 }
2671 
2672 static const struct got_error *
2673 match_commit(int *have_match, struct got_object_id *id,
2674     struct got_commit_object *commit, regex_t *regex)
2675 {
2676 	const struct got_error *err = NULL;
2677 	regmatch_t regmatch;
2678 	char *id_str = NULL, *logmsg = NULL;
2679 
2680 	*have_match = 0;
2681 
2682 	err = got_object_id_str(&id_str, id);
2683 	if (err)
2684 		return err;
2685 
2686 	err = got_object_commit_get_logmsg(&logmsg, commit);
2687 	if (err)
2688 		goto done;
2689 
2690 	if (regexec(regex, got_object_commit_get_author(commit), 1,
2691 	    &regmatch, 0) == 0 ||
2692 	    regexec(regex, got_object_commit_get_committer(commit), 1,
2693 	    &regmatch, 0) == 0 ||
2694 	    regexec(regex, id_str, 1, &regmatch, 0) == 0 ||
2695 	    regexec(regex, logmsg, 1, &regmatch, 0) == 0)
2696 		*have_match = 1;
2697 done:
2698 	free(id_str);
2699 	free(logmsg);
2700 	return err;
2701 }
2702 
2703 static const struct got_error *
2704 queue_commits(struct tog_log_thread_args *a)
2705 {
2706 	const struct got_error *err = NULL;
2707 
2708 	/*
2709 	 * We keep all commits open throughout the lifetime of the log
2710 	 * view in order to avoid having to re-fetch commits from disk
2711 	 * while updating the display.
2712 	 */
2713 	do {
2714 		struct got_object_id id;
2715 		struct got_commit_object *commit;
2716 		struct commit_queue_entry *entry;
2717 		int limit_match = 0;
2718 		int errcode;
2719 
2720 		err = got_commit_graph_iter_next(&id, a->graph, a->repo,
2721 		    NULL, NULL);
2722 		if (err)
2723 			break;
2724 
2725 		err = got_object_open_as_commit(&commit, a->repo, &id);
2726 		if (err)
2727 			break;
2728 		entry = alloc_commit_queue_entry(commit, &id);
2729 		if (entry == NULL) {
2730 			err = got_error_from_errno("alloc_commit_queue_entry");
2731 			break;
2732 		}
2733 
2734 		errcode = pthread_mutex_lock(&tog_mutex);
2735 		if (errcode) {
2736 			err = got_error_set_errno(errcode,
2737 			    "pthread_mutex_lock");
2738 			break;
2739 		}
2740 
2741 		entry->idx = a->real_commits->ncommits;
2742 		TAILQ_INSERT_TAIL(&a->real_commits->head, entry, entry);
2743 		a->real_commits->ncommits++;
2744 
2745 		if (*a->limiting) {
2746 			err = match_commit(&limit_match, &id, commit,
2747 			    a->limit_regex);
2748 			if (err)
2749 				break;
2750 
2751 			if (limit_match) {
2752 				struct commit_queue_entry *matched;
2753 
2754 				matched = alloc_commit_queue_entry(
2755 				    entry->commit, entry->id);
2756 				if (matched == NULL) {
2757 					err = got_error_from_errno(
2758 					    "alloc_commit_queue_entry");
2759 					break;
2760 				}
2761 				matched->commit = entry->commit;
2762 				got_object_commit_retain(entry->commit);
2763 
2764 				matched->idx = a->limit_commits->ncommits;
2765 				TAILQ_INSERT_TAIL(&a->limit_commits->head,
2766 				    matched, entry);
2767 				a->limit_commits->ncommits++;
2768 			}
2769 
2770 			/*
2771 			 * This is how we signal log_thread() that we
2772 			 * have found a match, and that it should be
2773 			 * counted as a new entry for the view.
2774 			 */
2775 			a->limit_match = limit_match;
2776 		}
2777 
2778 		if (*a->searching == TOG_SEARCH_FORWARD &&
2779 		    !*a->search_next_done) {
2780 			int have_match;
2781 			err = match_commit(&have_match, &id, commit, a->regex);
2782 			if (err)
2783 				break;
2784 
2785 			if (*a->limiting) {
2786 				if (limit_match && have_match)
2787 					*a->search_next_done =
2788 					    TOG_SEARCH_HAVE_MORE;
2789 			} else if (have_match)
2790 				*a->search_next_done = TOG_SEARCH_HAVE_MORE;
2791 		}
2792 
2793 		errcode = pthread_mutex_unlock(&tog_mutex);
2794 		if (errcode && err == NULL)
2795 			err = got_error_set_errno(errcode,
2796 			    "pthread_mutex_unlock");
2797 		if (err)
2798 			break;
2799 	} while (*a->searching == TOG_SEARCH_FORWARD && !*a->search_next_done);
2800 
2801 	return err;
2802 }
2803 
2804 static void
2805 select_commit(struct tog_log_view_state *s)
2806 {
2807 	struct commit_queue_entry *entry;
2808 	int ncommits = 0;
2809 
2810 	entry = s->first_displayed_entry;
2811 	while (entry) {
2812 		if (ncommits == s->selected) {
2813 			s->selected_entry = entry;
2814 			break;
2815 		}
2816 		entry = TAILQ_NEXT(entry, entry);
2817 		ncommits++;
2818 	}
2819 }
2820 
2821 static const struct got_error *
2822 draw_commits(struct tog_view *view)
2823 {
2824 	const struct got_error *err = NULL;
2825 	struct tog_log_view_state *s = &view->state.log;
2826 	struct commit_queue_entry *entry = s->selected_entry;
2827 	int limit = view->nlines;
2828 	int width;
2829 	int ncommits, author_cols = 4, refstr_cols;
2830 	char *id_str = NULL, *header = NULL, *ncommits_str = NULL;
2831 	char *refs_str = NULL;
2832 	wchar_t *wline;
2833 	struct tog_color *tc;
2834 	static const size_t date_display_cols = 12;
2835 	struct got_reflist_head *refs;
2836 
2837 	if (view_is_hsplit_top(view))
2838 		--limit;  /* account for border */
2839 
2840 	if (s->selected_entry &&
2841 	    !(view->searching && view->search_next_done == 0)) {
2842 		err = got_object_id_str(&id_str, s->selected_entry->id);
2843 		if (err)
2844 			return err;
2845 		refs = got_reflist_object_id_map_lookup(tog_refs_idmap,
2846 		    s->selected_entry->id);
2847 		err = build_refs_str(&refs_str, refs, s->selected_entry->id,
2848 		    s->repo);
2849 		if (err)
2850 			goto done;
2851 	}
2852 
2853 	if (s->thread_args.commits_needed == 0 && !using_mock_io)
2854 		halfdelay(10); /* disable fast refresh */
2855 
2856 	if (s->thread_args.commits_needed > 0 || s->thread_args.load_all) {
2857 		if (asprintf(&ncommits_str, " [%d/%d] %s",
2858 		    entry ? entry->idx + 1 : 0, s->commits->ncommits,
2859 		    (view->searching && !view->search_next_done) ?
2860 		    "searching..." : "loading...") == -1) {
2861 			err = got_error_from_errno("asprintf");
2862 			goto done;
2863 		}
2864 	} else {
2865 		const char *search_str = NULL;
2866 		const char *limit_str = NULL;
2867 
2868 		if (view->searching) {
2869 			if (view->search_next_done == TOG_SEARCH_NO_MORE)
2870 				search_str = "no more matches";
2871 			else if (view->search_next_done == TOG_SEARCH_HAVE_NONE)
2872 				search_str = "no matches found";
2873 			else if (!view->search_next_done)
2874 				search_str = "searching...";
2875 		}
2876 
2877 		if (s->limit_view && s->commits->ncommits == 0)
2878 			limit_str = "no matches found";
2879 
2880 		if (asprintf(&ncommits_str, " [%d/%d] %s %s",
2881 		    entry ? entry->idx + 1 : 0, s->commits->ncommits,
2882 		    search_str ? search_str : (refs_str ? refs_str : ""),
2883 		    limit_str ? limit_str : "") == -1) {
2884 			err = got_error_from_errno("asprintf");
2885 			goto done;
2886 		}
2887 	}
2888 
2889 	free(refs_str);
2890 	refs_str = NULL;
2891 
2892 	if (s->in_repo_path && strcmp(s->in_repo_path, "/") != 0) {
2893 		if (asprintf(&header, "commit %s %s%s", id_str ? id_str :
2894 		    "........................................",
2895 		    s->in_repo_path, ncommits_str) == -1) {
2896 			err = got_error_from_errno("asprintf");
2897 			header = NULL;
2898 			goto done;
2899 		}
2900 	} else if (asprintf(&header, "commit %s%s",
2901 	    id_str ? id_str : "........................................",
2902 	    ncommits_str) == -1) {
2903 		err = got_error_from_errno("asprintf");
2904 		header = NULL;
2905 		goto done;
2906 	}
2907 	err = format_line(&wline, &width, NULL, header, 0, view->ncols, 0, 0);
2908 	if (err)
2909 		goto done;
2910 
2911 	werase(view->window);
2912 
2913 	if (view_needs_focus_indication(view))
2914 		wstandout(view->window);
2915 	tc = get_color(&s->colors, TOG_COLOR_COMMIT);
2916 	if (tc)
2917 		wattr_on(view->window, COLOR_PAIR(tc->colorpair), NULL);
2918 	waddwstr(view->window, wline);
2919 	while (width < view->ncols) {
2920 		waddch(view->window, ' ');
2921 		width++;
2922 	}
2923 	if (tc)
2924 		wattr_off(view->window, COLOR_PAIR(tc->colorpair), NULL);
2925 	if (view_needs_focus_indication(view))
2926 		wstandend(view->window);
2927 	free(wline);
2928 	if (limit <= 1)
2929 		goto done;
2930 
2931 	/* Grow author column size if necessary, and set view->maxx. */
2932 	entry = s->first_displayed_entry;
2933 	ncommits = 0;
2934 	view->maxx = 0;
2935 	while (entry) {
2936 		struct got_commit_object *c = entry->commit;
2937 		char *author, *eol, *msg, *msg0;
2938 		wchar_t *wauthor, *wmsg;
2939 		int width;
2940 		if (ncommits >= limit - 1)
2941 			break;
2942 		if (s->use_committer)
2943 			author = strdup(got_object_commit_get_committer(c));
2944 		else
2945 			author = strdup(got_object_commit_get_author(c));
2946 		if (author == NULL) {
2947 			err = got_error_from_errno("strdup");
2948 			goto done;
2949 		}
2950 		err = format_author(&wauthor, &width, author, COLS,
2951 		    date_display_cols);
2952 		if (author_cols < width)
2953 			author_cols = width;
2954 		free(wauthor);
2955 		free(author);
2956 		if (err)
2957 			goto done;
2958 		refs = got_reflist_object_id_map_lookup(tog_refs_idmap,
2959 		    entry->id);
2960 		err = build_refs_str(&refs_str, refs, entry->id, s->repo);
2961 		if (err)
2962 			goto done;
2963 		if (refs_str) {
2964 			wchar_t *ws;
2965 			err = format_line(&ws, &width, NULL, refs_str,
2966 			    0, INT_MAX, date_display_cols + author_cols, 0);
2967 			free(ws);
2968 			free(refs_str);
2969 			refs_str = NULL;
2970 			if (err)
2971 				goto done;
2972 			refstr_cols = width + 3; /* account for [ ] + space */
2973 		} else
2974 			refstr_cols = 0;
2975 		err = got_object_commit_get_logmsg(&msg0, c);
2976 		if (err)
2977 			goto done;
2978 		msg = msg0;
2979 		while (*msg == '\n')
2980 			++msg;
2981 		if ((eol = strchr(msg, '\n')))
2982 			*eol = '\0';
2983 		err = format_line(&wmsg, &width, NULL, msg, 0, INT_MAX,
2984 		    date_display_cols + author_cols + refstr_cols, 0);
2985 		if (err)
2986 			goto done;
2987 		view->maxx = MAX(view->maxx, width + refstr_cols);
2988 		free(msg0);
2989 		free(wmsg);
2990 		ncommits++;
2991 		entry = TAILQ_NEXT(entry, entry);
2992 	}
2993 
2994 	entry = s->first_displayed_entry;
2995 	s->last_displayed_entry = s->first_displayed_entry;
2996 	ncommits = 0;
2997 	while (entry) {
2998 		if (ncommits >= limit - 1)
2999 			break;
3000 		if (ncommits == s->selected)
3001 			wstandout(view->window);
3002 		err = draw_commit(view, entry, date_display_cols, author_cols);
3003 		if (ncommits == s->selected)
3004 			wstandend(view->window);
3005 		if (err)
3006 			goto done;
3007 		ncommits++;
3008 		s->last_displayed_entry = entry;
3009 		entry = TAILQ_NEXT(entry, entry);
3010 	}
3011 
3012 	view_border(view);
3013 done:
3014 	free(id_str);
3015 	free(refs_str);
3016 	free(ncommits_str);
3017 	free(header);
3018 	return err;
3019 }
3020 
3021 static void
3022 log_scroll_up(struct tog_log_view_state *s, int maxscroll)
3023 {
3024 	struct commit_queue_entry *entry;
3025 	int nscrolled = 0;
3026 
3027 	entry = TAILQ_FIRST(&s->commits->head);
3028 	if (s->first_displayed_entry == entry)
3029 		return;
3030 
3031 	entry = s->first_displayed_entry;
3032 	while (entry && nscrolled < maxscroll) {
3033 		entry = TAILQ_PREV(entry, commit_queue_head, entry);
3034 		if (entry) {
3035 			s->first_displayed_entry = entry;
3036 			nscrolled++;
3037 		}
3038 	}
3039 }
3040 
3041 static const struct got_error *
3042 trigger_log_thread(struct tog_view *view, int wait)
3043 {
3044 	struct tog_log_thread_args *ta = &view->state.log.thread_args;
3045 	int errcode;
3046 
3047 	if (!using_mock_io)
3048 		halfdelay(1); /* fast refresh while loading commits */
3049 
3050 	while (!ta->log_complete && !tog_thread_error &&
3051 	    (ta->commits_needed > 0 || ta->load_all)) {
3052 		/* Wake the log thread. */
3053 		errcode = pthread_cond_signal(&ta->need_commits);
3054 		if (errcode)
3055 			return got_error_set_errno(errcode,
3056 			    "pthread_cond_signal");
3057 
3058 		/*
3059 		 * The mutex will be released while the view loop waits
3060 		 * in wgetch(), at which time the log thread will run.
3061 		 */
3062 		if (!wait)
3063 			break;
3064 
3065 		/* Display progress update in log view. */
3066 		show_log_view(view);
3067 		update_panels();
3068 		doupdate();
3069 
3070 		/* Wait right here while next commit is being loaded. */
3071 		errcode = pthread_cond_wait(&ta->commit_loaded, &tog_mutex);
3072 		if (errcode)
3073 			return got_error_set_errno(errcode,
3074 			    "pthread_cond_wait");
3075 
3076 		/* Display progress update in log view. */
3077 		show_log_view(view);
3078 		update_panels();
3079 		doupdate();
3080 	}
3081 
3082 	return NULL;
3083 }
3084 
3085 static const struct got_error *
3086 request_log_commits(struct tog_view *view)
3087 {
3088 	struct tog_log_view_state	*state = &view->state.log;
3089 	const struct got_error		*err = NULL;
3090 
3091 	if (state->thread_args.log_complete)
3092 		return NULL;
3093 
3094 	state->thread_args.commits_needed += view->nscrolled;
3095 	err = trigger_log_thread(view, 1);
3096 	view->nscrolled = 0;
3097 
3098 	return err;
3099 }
3100 
3101 static const struct got_error *
3102 log_scroll_down(struct tog_view *view, int maxscroll)
3103 {
3104 	struct tog_log_view_state *s = &view->state.log;
3105 	const struct got_error *err = NULL;
3106 	struct commit_queue_entry *pentry;
3107 	int nscrolled = 0, ncommits_needed;
3108 
3109 	if (s->last_displayed_entry == NULL)
3110 		return NULL;
3111 
3112 	ncommits_needed = s->last_displayed_entry->idx + 1 + maxscroll;
3113 	if (s->commits->ncommits < ncommits_needed &&
3114 	    !s->thread_args.log_complete) {
3115 		/*
3116 		 * Ask the log thread for required amount of commits.
3117 		 */
3118 		s->thread_args.commits_needed +=
3119 		    ncommits_needed - s->commits->ncommits;
3120 		err = trigger_log_thread(view, 1);
3121 		if (err)
3122 			return err;
3123 	}
3124 
3125 	do {
3126 		pentry = TAILQ_NEXT(s->last_displayed_entry, entry);
3127 		if (pentry == NULL && view->mode != TOG_VIEW_SPLIT_HRZN)
3128 			break;
3129 
3130 		s->last_displayed_entry = pentry ?
3131 		    pentry : s->last_displayed_entry;
3132 
3133 		pentry = TAILQ_NEXT(s->first_displayed_entry, entry);
3134 		if (pentry == NULL)
3135 			break;
3136 		s->first_displayed_entry = pentry;
3137 	} while (++nscrolled < maxscroll);
3138 
3139 	if (view->mode == TOG_VIEW_SPLIT_HRZN && !s->thread_args.log_complete)
3140 		view->nscrolled += nscrolled;
3141 	else
3142 		view->nscrolled = 0;
3143 
3144 	return err;
3145 }
3146 
3147 static const struct got_error *
3148 open_diff_view_for_commit(struct tog_view **new_view, int begin_y, int begin_x,
3149     struct got_commit_object *commit, struct got_object_id *commit_id,
3150     struct tog_view *log_view, struct got_repository *repo)
3151 {
3152 	const struct got_error *err;
3153 	struct got_object_qid *parent_id;
3154 	struct tog_view *diff_view;
3155 
3156 	diff_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_DIFF);
3157 	if (diff_view == NULL)
3158 		return got_error_from_errno("view_open");
3159 
3160 	parent_id = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
3161 	err = open_diff_view(diff_view, parent_id ? &parent_id->id : NULL,
3162 	    commit_id, NULL, NULL, 3, 0, 0, log_view, repo);
3163 	if (err == NULL)
3164 		*new_view = diff_view;
3165 	return err;
3166 }
3167 
3168 static const struct got_error *
3169 tree_view_visit_subtree(struct tog_tree_view_state *s,
3170     struct got_tree_object *subtree)
3171 {
3172 	struct tog_parent_tree *parent;
3173 
3174 	parent = calloc(1, sizeof(*parent));
3175 	if (parent == NULL)
3176 		return got_error_from_errno("calloc");
3177 
3178 	parent->tree = s->tree;
3179 	parent->first_displayed_entry = s->first_displayed_entry;
3180 	parent->selected_entry = s->selected_entry;
3181 	parent->selected = s->selected;
3182 	TAILQ_INSERT_HEAD(&s->parents, parent, entry);
3183 	s->tree = subtree;
3184 	s->selected = 0;
3185 	s->first_displayed_entry = NULL;
3186 	return NULL;
3187 }
3188 
3189 static const struct got_error *
3190 tree_view_walk_path(struct tog_tree_view_state *s,
3191     struct got_commit_object *commit, const char *path)
3192 {
3193 	const struct got_error *err = NULL;
3194 	struct got_tree_object *tree = NULL;
3195 	const char *p;
3196 	char *slash, *subpath = NULL;
3197 
3198 	/* Walk the path and open corresponding tree objects. */
3199 	p = path;
3200 	while (*p) {
3201 		struct got_tree_entry *te;
3202 		struct got_object_id *tree_id;
3203 		char *te_name;
3204 
3205 		while (p[0] == '/')
3206 			p++;
3207 
3208 		/* Ensure the correct subtree entry is selected. */
3209 		slash = strchr(p, '/');
3210 		if (slash == NULL)
3211 			te_name = strdup(p);
3212 		else
3213 			te_name = strndup(p, slash - p);
3214 		if (te_name == NULL) {
3215 			err = got_error_from_errno("strndup");
3216 			break;
3217 		}
3218 		te = got_object_tree_find_entry(s->tree, te_name);
3219 		if (te == NULL) {
3220 			err = got_error_path(te_name, GOT_ERR_NO_TREE_ENTRY);
3221 			free(te_name);
3222 			break;
3223 		}
3224 		free(te_name);
3225 		s->first_displayed_entry = s->selected_entry = te;
3226 
3227 		if (!S_ISDIR(got_tree_entry_get_mode(s->selected_entry)))
3228 			break; /* jump to this file's entry */
3229 
3230 		slash = strchr(p, '/');
3231 		if (slash)
3232 			subpath = strndup(path, slash - path);
3233 		else
3234 			subpath = strdup(path);
3235 		if (subpath == NULL) {
3236 			err = got_error_from_errno("strdup");
3237 			break;
3238 		}
3239 
3240 		err = got_object_id_by_path(&tree_id, s->repo, commit,
3241 		    subpath);
3242 		if (err)
3243 			break;
3244 
3245 		err = got_object_open_as_tree(&tree, s->repo, tree_id);
3246 		free(tree_id);
3247 		if (err)
3248 			break;
3249 
3250 		err = tree_view_visit_subtree(s, tree);
3251 		if (err) {
3252 			got_object_tree_close(tree);
3253 			break;
3254 		}
3255 		if (slash == NULL)
3256 			break;
3257 		free(subpath);
3258 		subpath = NULL;
3259 		p = slash;
3260 	}
3261 
3262 	free(subpath);
3263 	return err;
3264 }
3265 
3266 static const struct got_error *
3267 browse_commit_tree(struct tog_view **new_view, int begin_y, int begin_x,
3268     struct commit_queue_entry *entry, const char *path,
3269     const char *head_ref_name, struct got_repository *repo)
3270 {
3271 	const struct got_error *err = NULL;
3272 	struct tog_tree_view_state *s;
3273 	struct tog_view *tree_view;
3274 
3275 	tree_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_TREE);
3276 	if (tree_view == NULL)
3277 		return got_error_from_errno("view_open");
3278 
3279 	err = open_tree_view(tree_view, entry->id, head_ref_name, repo);
3280 	if (err)
3281 		return err;
3282 	s = &tree_view->state.tree;
3283 
3284 	*new_view = tree_view;
3285 
3286 	if (got_path_is_root_dir(path))
3287 		return NULL;
3288 
3289 	return tree_view_walk_path(s, entry->commit, path);
3290 }
3291 
3292 static const struct got_error *
3293 block_signals_used_by_main_thread(void)
3294 {
3295 	sigset_t sigset;
3296 	int errcode;
3297 
3298 	if (sigemptyset(&sigset) == -1)
3299 		return got_error_from_errno("sigemptyset");
3300 
3301 	/* tog handles SIGWINCH, SIGCONT, SIGINT, SIGTERM */
3302 	if (sigaddset(&sigset, SIGWINCH) == -1)
3303 		return got_error_from_errno("sigaddset");
3304 	if (sigaddset(&sigset, SIGCONT) == -1)
3305 		return got_error_from_errno("sigaddset");
3306 	if (sigaddset(&sigset, SIGINT) == -1)
3307 		return got_error_from_errno("sigaddset");
3308 	if (sigaddset(&sigset, SIGTERM) == -1)
3309 		return got_error_from_errno("sigaddset");
3310 
3311 	/* ncurses handles SIGTSTP */
3312 	if (sigaddset(&sigset, SIGTSTP) == -1)
3313 		return got_error_from_errno("sigaddset");
3314 
3315 	errcode = pthread_sigmask(SIG_BLOCK, &sigset, NULL);
3316 	if (errcode)
3317 		return got_error_set_errno(errcode, "pthread_sigmask");
3318 
3319 	return NULL;
3320 }
3321 
3322 static void *
3323 log_thread(void *arg)
3324 {
3325 	const struct got_error *err = NULL;
3326 	int errcode = 0;
3327 	struct tog_log_thread_args *a = arg;
3328 	int done = 0;
3329 
3330 	/*
3331 	 * Sync startup with main thread such that we begin our
3332 	 * work once view_input() has released the mutex.
3333 	 */
3334 	errcode = pthread_mutex_lock(&tog_mutex);
3335 	if (errcode) {
3336 		err = got_error_set_errno(errcode, "pthread_mutex_lock");
3337 		return (void *)err;
3338 	}
3339 
3340 	err = block_signals_used_by_main_thread();
3341 	if (err) {
3342 		pthread_mutex_unlock(&tog_mutex);
3343 		goto done;
3344 	}
3345 
3346 	while (!done && !err && !tog_fatal_signal_received()) {
3347 		errcode = pthread_mutex_unlock(&tog_mutex);
3348 		if (errcode) {
3349 			err = got_error_set_errno(errcode,
3350 			    "pthread_mutex_unlock");
3351 			goto done;
3352 		}
3353 		err = queue_commits(a);
3354 		if (err) {
3355 			if (err->code != GOT_ERR_ITER_COMPLETED)
3356 				goto done;
3357 			err = NULL;
3358 			done = 1;
3359 			a->commits_needed = 0;
3360 		} else if (a->commits_needed > 0 && !a->load_all) {
3361 			if (*a->limiting) {
3362 				if (a->limit_match)
3363 					a->commits_needed--;
3364 			} else
3365 				a->commits_needed--;
3366 		}
3367 
3368 		errcode = pthread_mutex_lock(&tog_mutex);
3369 		if (errcode) {
3370 			err = got_error_set_errno(errcode,
3371 			    "pthread_mutex_lock");
3372 			goto done;
3373 		} else if (*a->quit)
3374 			done = 1;
3375 		else if (*a->limiting && *a->first_displayed_entry == NULL) {
3376 			*a->first_displayed_entry =
3377 			    TAILQ_FIRST(&a->limit_commits->head);
3378 			*a->selected_entry = *a->first_displayed_entry;
3379 		} else if (*a->first_displayed_entry == NULL) {
3380 			*a->first_displayed_entry =
3381 			    TAILQ_FIRST(&a->real_commits->head);
3382 			*a->selected_entry = *a->first_displayed_entry;
3383 		}
3384 
3385 		errcode = pthread_cond_signal(&a->commit_loaded);
3386 		if (errcode) {
3387 			err = got_error_set_errno(errcode,
3388 			    "pthread_cond_signal");
3389 			pthread_mutex_unlock(&tog_mutex);
3390 			goto done;
3391 		}
3392 
3393 		if (a->commits_needed == 0 &&
3394 		    a->need_commit_marker && a->worktree) {
3395 			errcode = pthread_mutex_unlock(&tog_mutex);
3396 			if (errcode) {
3397 				err = got_error_set_errno(errcode,
3398 				    "pthread_mutex_unlock");
3399 				goto done;
3400 			}
3401 			err = got_worktree_get_state(&tog_base_commit.marker,
3402 			    a->repo, a->worktree, NULL, NULL);
3403 			if (err)
3404 				goto done;
3405 			errcode = pthread_mutex_lock(&tog_mutex);
3406 			if (errcode) {
3407 				err = got_error_set_errno(errcode,
3408 				    "pthread_mutex_lock");
3409 				goto done;
3410 			}
3411 			a->need_commit_marker = 0;
3412 			/*
3413 			 * The main thread did not close this
3414 			 * work tree yet. Close it now.
3415 			 */
3416 			got_worktree_close(a->worktree);
3417 			a->worktree = NULL;
3418 
3419 			if (*a->quit)
3420 				done = 1;
3421 		}
3422 
3423 		if (done)
3424 			a->commits_needed = 0;
3425 		else {
3426 			if (a->commits_needed == 0 && !a->load_all) {
3427 				if (tog_io.wait_for_ui) {
3428 					errcode = pthread_cond_signal(
3429 					    &a->log_loaded);
3430 					if (errcode && err == NULL)
3431 						err = got_error_set_errno(
3432 						    errcode,
3433 						    "pthread_cond_signal");
3434 				}
3435 
3436 				errcode = pthread_cond_wait(&a->need_commits,
3437 				    &tog_mutex);
3438 				if (errcode) {
3439 					err = got_error_set_errno(errcode,
3440 					    "pthread_cond_wait");
3441 					pthread_mutex_unlock(&tog_mutex);
3442 					goto done;
3443 				}
3444 				if (*a->quit)
3445 					done = 1;
3446 			}
3447 		}
3448 	}
3449 	a->log_complete = 1;
3450 	if (tog_io.wait_for_ui) {
3451 		errcode = pthread_cond_signal(&a->log_loaded);
3452 		if (errcode && err == NULL)
3453 			err = got_error_set_errno(errcode,
3454 			    "pthread_cond_signal");
3455 	}
3456 
3457 	errcode = pthread_mutex_unlock(&tog_mutex);
3458 	if (errcode)
3459 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
3460 done:
3461 	if (err) {
3462 		tog_thread_error = 1;
3463 		pthread_cond_signal(&a->commit_loaded);
3464 		if (a->worktree) {
3465 			got_worktree_close(a->worktree);
3466 			a->worktree = NULL;
3467 		}
3468 	}
3469 	return (void *)err;
3470 }
3471 
3472 static const struct got_error *
3473 stop_log_thread(struct tog_log_view_state *s)
3474 {
3475 	const struct got_error *err = NULL, *thread_err = NULL;
3476 	int errcode;
3477 
3478 	if (s->thread) {
3479 		s->quit = 1;
3480 		errcode = pthread_cond_signal(&s->thread_args.need_commits);
3481 		if (errcode)
3482 			return got_error_set_errno(errcode,
3483 			    "pthread_cond_signal");
3484 		errcode = pthread_mutex_unlock(&tog_mutex);
3485 		if (errcode)
3486 			return got_error_set_errno(errcode,
3487 			    "pthread_mutex_unlock");
3488 		errcode = pthread_join(s->thread, (void **)&thread_err);
3489 		if (errcode)
3490 			return got_error_set_errno(errcode, "pthread_join");
3491 		errcode = pthread_mutex_lock(&tog_mutex);
3492 		if (errcode)
3493 			return got_error_set_errno(errcode,
3494 			    "pthread_mutex_lock");
3495 		s->thread = NULL;
3496 	}
3497 
3498 	if (s->thread_args.repo) {
3499 		err = got_repo_close(s->thread_args.repo);
3500 		s->thread_args.repo = NULL;
3501 	}
3502 
3503 	if (s->thread_args.pack_fds) {
3504 		const struct got_error *pack_err =
3505 		    got_repo_pack_fds_close(s->thread_args.pack_fds);
3506 		if (err == NULL)
3507 			err = pack_err;
3508 		s->thread_args.pack_fds = NULL;
3509 	}
3510 
3511 	if (s->thread_args.graph) {
3512 		got_commit_graph_close(s->thread_args.graph);
3513 		s->thread_args.graph = NULL;
3514 	}
3515 
3516 	return err ? err : thread_err;
3517 }
3518 
3519 static const struct got_error *
3520 close_log_view(struct tog_view *view)
3521 {
3522 	const struct got_error *err = NULL;
3523 	struct tog_log_view_state *s = &view->state.log;
3524 	int errcode;
3525 
3526 	err = stop_log_thread(s);
3527 
3528 	errcode = pthread_cond_destroy(&s->thread_args.need_commits);
3529 	if (errcode && err == NULL)
3530 		err = got_error_set_errno(errcode, "pthread_cond_destroy");
3531 
3532 	errcode = pthread_cond_destroy(&s->thread_args.commit_loaded);
3533 	if (errcode && err == NULL)
3534 		err = got_error_set_errno(errcode, "pthread_cond_destroy");
3535 
3536 	free_commits(&s->limit_commits);
3537 	free_commits(&s->real_commits);
3538 	free_colors(&s->colors);
3539 	free(s->in_repo_path);
3540 	s->in_repo_path = NULL;
3541 	free(s->start_id);
3542 	s->start_id = NULL;
3543 	free(s->head_ref_name);
3544 	s->head_ref_name = NULL;
3545 	return err;
3546 }
3547 
3548 /*
3549  * We use two queues to implement the limit feature: first consists of
3550  * commits matching the current limit_regex; second is the real queue
3551  * of all known commits (real_commits). When the user starts limiting,
3552  * we swap queues such that all movement and displaying functionality
3553  * works with very slight change.
3554  */
3555 static const struct got_error *
3556 limit_log_view(struct tog_view *view)
3557 {
3558 	struct tog_log_view_state *s = &view->state.log;
3559 	struct commit_queue_entry *entry;
3560 	struct tog_view	*v = view;
3561 	const struct got_error *err = NULL;
3562 	char pattern[1024];
3563 	int ret;
3564 
3565 	if (view_is_hsplit_top(view))
3566 		v = view->child;
3567 	else if (view->mode == TOG_VIEW_SPLIT_VERT && view->parent)
3568 		v = view->parent;
3569 
3570 	if (tog_io.input_str != NULL) {
3571 		if (strlcpy(pattern, tog_io.input_str, sizeof(pattern)) >=
3572 		    sizeof(pattern))
3573 			return got_error(GOT_ERR_NO_SPACE);
3574 	} else {
3575 		wmove(v->window, v->nlines - 1, 0);
3576 		wclrtoeol(v->window);
3577 		mvwaddstr(v->window, v->nlines - 1, 0, "&/");
3578 		nodelay(v->window, FALSE);
3579 		nocbreak();
3580 		echo();
3581 		ret = wgetnstr(v->window, pattern, sizeof(pattern));
3582 		cbreak();
3583 		noecho();
3584 		nodelay(v->window, TRUE);
3585 		if (ret == ERR)
3586 			return NULL;
3587 	}
3588 
3589 	if (*pattern == '\0') {
3590 		/*
3591 		 * Safety measure for the situation where the user
3592 		 * resets limit without previously limiting anything.
3593 		 */
3594 		if (!s->limit_view)
3595 			return NULL;
3596 
3597 		/*
3598 		 * User could have pressed Ctrl+L, which refreshed the
3599 		 * commit queues, it means we can't save previously
3600 		 * (before limit took place) displayed entries,
3601 		 * because they would point to already free'ed memory,
3602 		 * so we are forced to always select first entry of
3603 		 * the queue.
3604 		 */
3605 		s->commits = &s->real_commits;
3606 		s->first_displayed_entry = TAILQ_FIRST(&s->real_commits.head);
3607 		s->selected_entry = s->first_displayed_entry;
3608 		s->selected = 0;
3609 		s->limit_view = 0;
3610 
3611 		return NULL;
3612 	}
3613 
3614 	if (regcomp(&s->limit_regex, pattern, REG_EXTENDED | REG_NEWLINE))
3615 		return NULL;
3616 
3617 	s->limit_view = 1;
3618 
3619 	/* Clear the screen while loading limit view */
3620 	s->first_displayed_entry = NULL;
3621 	s->last_displayed_entry = NULL;
3622 	s->selected_entry = NULL;
3623 	s->commits = &s->limit_commits;
3624 
3625 	/* Prepare limit queue for new search */
3626 	free_commits(&s->limit_commits);
3627 	s->limit_commits.ncommits = 0;
3628 
3629 	/* First process commits, which are in queue already */
3630 	TAILQ_FOREACH(entry, &s->real_commits.head, entry) {
3631 		int have_match = 0;
3632 
3633 		err = match_commit(&have_match, entry->id,
3634 		    entry->commit, &s->limit_regex);
3635 		if (err)
3636 			return err;
3637 
3638 		if (have_match) {
3639 			struct commit_queue_entry *matched;
3640 
3641 			matched = alloc_commit_queue_entry(entry->commit,
3642 			    entry->id);
3643 			if (matched == NULL) {
3644 				err = got_error_from_errno(
3645 				    "alloc_commit_queue_entry");
3646 				break;
3647 			}
3648 			matched->commit = entry->commit;
3649 			got_object_commit_retain(entry->commit);
3650 
3651 			matched->idx = s->limit_commits.ncommits;
3652 			TAILQ_INSERT_TAIL(&s->limit_commits.head,
3653 			    matched, entry);
3654 			s->limit_commits.ncommits++;
3655 		}
3656 	}
3657 
3658 	/* Second process all the commits, until we fill the screen */
3659 	if (s->limit_commits.ncommits < view->nlines - 1 &&
3660 	    !s->thread_args.log_complete) {
3661 		s->thread_args.commits_needed +=
3662 		    view->nlines - s->limit_commits.ncommits - 1;
3663 		err = trigger_log_thread(view, 1);
3664 		if (err)
3665 			return err;
3666 	}
3667 
3668 	s->first_displayed_entry = TAILQ_FIRST(&s->commits->head);
3669 	s->selected_entry = TAILQ_FIRST(&s->commits->head);
3670 	s->selected = 0;
3671 
3672 	return NULL;
3673 }
3674 
3675 static const struct got_error *
3676 search_start_log_view(struct tog_view *view)
3677 {
3678 	struct tog_log_view_state *s = &view->state.log;
3679 
3680 	s->matched_entry = NULL;
3681 	s->search_entry = NULL;
3682 	return NULL;
3683 }
3684 
3685 static const struct got_error *
3686 search_next_log_view(struct tog_view *view)
3687 {
3688 	const struct got_error *err = NULL;
3689 	struct tog_log_view_state *s = &view->state.log;
3690 	struct commit_queue_entry *entry;
3691 
3692 	/* Display progress update in log view. */
3693 	show_log_view(view);
3694 	update_panels();
3695 	doupdate();
3696 
3697 	if (s->search_entry) {
3698 		if (!using_mock_io) {
3699 			int errcode, ch;
3700 
3701 			errcode = pthread_mutex_unlock(&tog_mutex);
3702 			if (errcode)
3703 				return got_error_set_errno(errcode,
3704 				    "pthread_mutex_unlock");
3705 			ch = wgetch(view->window);
3706 			errcode = pthread_mutex_lock(&tog_mutex);
3707 			if (errcode)
3708 				return got_error_set_errno(errcode,
3709 				    "pthread_mutex_lock");
3710 			if (ch == CTRL('g') || ch == KEY_BACKSPACE) {
3711 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
3712 				return NULL;
3713 			}
3714 		}
3715 		if (view->searching == TOG_SEARCH_FORWARD)
3716 			entry = TAILQ_NEXT(s->search_entry, entry);
3717 		else
3718 			entry = TAILQ_PREV(s->search_entry,
3719 			    commit_queue_head, entry);
3720 	} else if (s->matched_entry) {
3721 		/*
3722 		 * If the user has moved the cursor after we hit a match,
3723 		 * the position from where we should continue searching
3724 		 * might have changed.
3725 		 */
3726 		if (view->searching == TOG_SEARCH_FORWARD)
3727 			entry = TAILQ_NEXT(s->selected_entry, entry);
3728 		else
3729 			entry = TAILQ_PREV(s->selected_entry, commit_queue_head,
3730 			    entry);
3731 	} else {
3732 		entry = s->selected_entry;
3733 	}
3734 
3735 	while (1) {
3736 		int have_match = 0;
3737 
3738 		if (entry == NULL) {
3739 			if (s->thread_args.log_complete ||
3740 			    view->searching == TOG_SEARCH_BACKWARD) {
3741 				view->search_next_done =
3742 				    (s->matched_entry == NULL ?
3743 				    TOG_SEARCH_HAVE_NONE : TOG_SEARCH_NO_MORE);
3744 				s->search_entry = NULL;
3745 				return NULL;
3746 			}
3747 			/*
3748 			 * Poke the log thread for more commits and return,
3749 			 * allowing the main loop to make progress. Search
3750 			 * will resume at s->search_entry once we come back.
3751 			 */
3752 			s->search_entry = s->selected_entry;
3753 			s->thread_args.commits_needed++;
3754 			return trigger_log_thread(view, 0);
3755 		}
3756 
3757 		err = match_commit(&have_match, entry->id, entry->commit,
3758 		    &view->regex);
3759 		if (err)
3760 			break;
3761 		if (have_match) {
3762 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
3763 			s->matched_entry = entry;
3764 			break;
3765 		}
3766 
3767 		s->search_entry = entry;
3768 		if (view->searching == TOG_SEARCH_FORWARD)
3769 			entry = TAILQ_NEXT(entry, entry);
3770 		else
3771 			entry = TAILQ_PREV(entry, commit_queue_head, entry);
3772 	}
3773 
3774 	if (s->matched_entry) {
3775 		int cur = s->selected_entry->idx;
3776 		while (cur < s->matched_entry->idx) {
3777 			err = input_log_view(NULL, view, KEY_DOWN);
3778 			if (err)
3779 				return err;
3780 			cur++;
3781 		}
3782 		while (cur > s->matched_entry->idx) {
3783 			err = input_log_view(NULL, view, KEY_UP);
3784 			if (err)
3785 				return err;
3786 			cur--;
3787 		}
3788 	}
3789 
3790 	s->search_entry = NULL;
3791 
3792 	return NULL;
3793 }
3794 
3795 static const struct got_error *
3796 open_log_view(struct tog_view *view, struct got_object_id *start_id,
3797     struct got_repository *repo, const char *head_ref_name,
3798     const char *in_repo_path, int log_branches,
3799     struct got_worktree *worktree)
3800 {
3801 	const struct got_error *err = NULL;
3802 	struct tog_log_view_state *s = &view->state.log;
3803 	struct got_repository *thread_repo = NULL;
3804 	struct got_commit_graph *thread_graph = NULL;
3805 	int errcode;
3806 
3807 	if (in_repo_path != s->in_repo_path) {
3808 		free(s->in_repo_path);
3809 		s->in_repo_path = strdup(in_repo_path);
3810 		if (s->in_repo_path == NULL) {
3811 			err = got_error_from_errno("strdup");
3812 			goto done;
3813 		}
3814 	}
3815 
3816 	/* The commit queue only contains commits being displayed. */
3817 	TAILQ_INIT(&s->real_commits.head);
3818 	s->real_commits.ncommits = 0;
3819 	s->commits = &s->real_commits;
3820 
3821 	TAILQ_INIT(&s->limit_commits.head);
3822 	s->limit_view = 0;
3823 	s->limit_commits.ncommits = 0;
3824 
3825 	s->repo = repo;
3826 	if (head_ref_name) {
3827 		s->head_ref_name = strdup(head_ref_name);
3828 		if (s->head_ref_name == NULL) {
3829 			err = got_error_from_errno("strdup");
3830 			goto done;
3831 		}
3832 	}
3833 	s->start_id = got_object_id_dup(start_id);
3834 	if (s->start_id == NULL) {
3835 		err = got_error_from_errno("got_object_id_dup");
3836 		goto done;
3837 	}
3838 	s->log_branches = log_branches;
3839 	s->use_committer = 1;
3840 
3841 	STAILQ_INIT(&s->colors);
3842 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
3843 		err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
3844 		    get_color_value("TOG_COLOR_COMMIT"));
3845 		if (err)
3846 			goto done;
3847 		err = add_color(&s->colors, "^$", TOG_COLOR_AUTHOR,
3848 		    get_color_value("TOG_COLOR_AUTHOR"));
3849 		if (err)
3850 			goto done;
3851 		err = add_color(&s->colors, "^$", TOG_COLOR_DATE,
3852 		    get_color_value("TOG_COLOR_DATE"));
3853 		if (err)
3854 			goto done;
3855 	}
3856 
3857 	view->show = show_log_view;
3858 	view->input = input_log_view;
3859 	view->resize = resize_log_view;
3860 	view->close = close_log_view;
3861 	view->search_start = search_start_log_view;
3862 	view->search_next = search_next_log_view;
3863 
3864 	if (s->thread_args.pack_fds == NULL) {
3865 		err = got_repo_pack_fds_open(&s->thread_args.pack_fds);
3866 		if (err)
3867 			goto done;
3868 	}
3869 	err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL,
3870 	    s->thread_args.pack_fds);
3871 	if (err)
3872 		goto done;
3873 	err = got_commit_graph_open(&thread_graph, s->in_repo_path,
3874 	    !s->log_branches);
3875 	if (err)
3876 		goto done;
3877 	err = got_commit_graph_iter_start(thread_graph, s->start_id,
3878 	    s->repo, NULL, NULL);
3879 	if (err)
3880 		goto done;
3881 
3882 	errcode = pthread_cond_init(&s->thread_args.need_commits, NULL);
3883 	if (errcode) {
3884 		err = got_error_set_errno(errcode, "pthread_cond_init");
3885 		goto done;
3886 	}
3887 	errcode = pthread_cond_init(&s->thread_args.commit_loaded, NULL);
3888 	if (errcode) {
3889 		err = got_error_set_errno(errcode, "pthread_cond_init");
3890 		goto done;
3891 	}
3892 
3893 	if (using_mock_io) {
3894 		int rc;
3895 
3896 		rc = pthread_cond_init(&s->thread_args.log_loaded, NULL);
3897 		if (rc)
3898 			return got_error_set_errno(rc, "pthread_cond_init");
3899 	}
3900 
3901 	s->thread_args.commits_needed = view->nlines;
3902 	s->thread_args.graph = thread_graph;
3903 	s->thread_args.real_commits = &s->real_commits;
3904 	s->thread_args.limit_commits = &s->limit_commits;
3905 	s->thread_args.in_repo_path = s->in_repo_path;
3906 	s->thread_args.start_id = s->start_id;
3907 	s->thread_args.repo = thread_repo;
3908 	s->thread_args.log_complete = 0;
3909 	s->thread_args.quit = &s->quit;
3910 	s->thread_args.first_displayed_entry = &s->first_displayed_entry;
3911 	s->thread_args.selected_entry = &s->selected_entry;
3912 	s->thread_args.searching = &view->searching;
3913 	s->thread_args.search_next_done = &view->search_next_done;
3914 	s->thread_args.regex = &view->regex;
3915 	s->thread_args.limiting = &s->limit_view;
3916 	s->thread_args.limit_regex = &s->limit_regex;
3917 	s->thread_args.limit_commits = &s->limit_commits;
3918 	s->thread_args.worktree = worktree;
3919 	if (worktree)
3920 		s->thread_args.need_commit_marker = 1;
3921 done:
3922 	if (err) {
3923 		if (view->close == NULL)
3924 			close_log_view(view);
3925 		view_close(view);
3926 	}
3927 	return err;
3928 }
3929 
3930 static const struct got_error *
3931 show_log_view(struct tog_view *view)
3932 {
3933 	const struct got_error *err;
3934 	struct tog_log_view_state *s = &view->state.log;
3935 
3936 	if (s->thread == NULL) {
3937 		int errcode = pthread_create(&s->thread, NULL, log_thread,
3938 		    &s->thread_args);
3939 		if (errcode)
3940 			return got_error_set_errno(errcode, "pthread_create");
3941 		if (s->thread_args.commits_needed > 0) {
3942 			err = trigger_log_thread(view, 1);
3943 			if (err)
3944 				return err;
3945 		}
3946 	}
3947 
3948 	return draw_commits(view);
3949 }
3950 
3951 static void
3952 log_move_cursor_up(struct tog_view *view, int page, int home)
3953 {
3954 	struct tog_log_view_state *s = &view->state.log;
3955 
3956 	if (s->first_displayed_entry == NULL)
3957 		return;
3958 	if (s->selected_entry->idx == 0)
3959 		view->count = 0;
3960 
3961 	if ((page && TAILQ_FIRST(&s->commits->head) == s->first_displayed_entry)
3962 	    || home)
3963 		s->selected = home ? 0 : MAX(0, s->selected - page - 1);
3964 
3965 	if (!page && !home && s->selected > 0)
3966 		--s->selected;
3967 	else
3968 		log_scroll_up(s, home ? s->commits->ncommits : MAX(page, 1));
3969 
3970 	select_commit(s);
3971 	return;
3972 }
3973 
3974 static const struct got_error *
3975 log_move_cursor_down(struct tog_view *view, int page)
3976 {
3977 	struct tog_log_view_state	*s = &view->state.log;
3978 	const struct got_error		*err = NULL;
3979 	int				 eos = view->nlines - 2;
3980 
3981 	if (s->first_displayed_entry == NULL)
3982 		return NULL;
3983 
3984 	if (s->thread_args.log_complete &&
3985 	    s->selected_entry->idx >= s->commits->ncommits - 1)
3986 		return NULL;
3987 
3988 	if (view_is_hsplit_top(view))
3989 		--eos;  /* border consumes the last line */
3990 
3991 	if (!page) {
3992 		if (s->selected < MIN(eos, s->commits->ncommits - 1))
3993 			++s->selected;
3994 		else
3995 			err = log_scroll_down(view, 1);
3996 	} else if (s->thread_args.load_all && s->thread_args.log_complete) {
3997 		struct commit_queue_entry *entry;
3998 		int n;
3999 
4000 		s->selected = 0;
4001 		entry = TAILQ_LAST(&s->commits->head, commit_queue_head);
4002 		s->last_displayed_entry = entry;
4003 		for (n = 0; n <= eos; n++) {
4004 			if (entry == NULL)
4005 				break;
4006 			s->first_displayed_entry = entry;
4007 			entry = TAILQ_PREV(entry, commit_queue_head, entry);
4008 		}
4009 		if (n > 0)
4010 			s->selected = n - 1;
4011 	} else {
4012 		if (s->last_displayed_entry->idx == s->commits->ncommits - 1 &&
4013 		    s->thread_args.log_complete)
4014 			s->selected += MIN(page,
4015 			    s->commits->ncommits - s->selected_entry->idx - 1);
4016 		else
4017 			err = log_scroll_down(view, page);
4018 	}
4019 	if (err)
4020 		return err;
4021 
4022 	/*
4023 	 * We might necessarily overshoot in horizontal
4024 	 * splits; if so, select the last displayed commit.
4025 	 */
4026 	if (s->first_displayed_entry && s->last_displayed_entry) {
4027 		s->selected = MIN(s->selected,
4028 		    s->last_displayed_entry->idx -
4029 		    s->first_displayed_entry->idx);
4030 	}
4031 
4032 	select_commit(s);
4033 
4034 	if (s->thread_args.log_complete &&
4035 	    s->selected_entry->idx == s->commits->ncommits - 1)
4036 		view->count = 0;
4037 
4038 	return NULL;
4039 }
4040 
4041 static void
4042 view_get_split(struct tog_view *view, int *y, int *x)
4043 {
4044 	*x = 0;
4045 	*y = 0;
4046 
4047 	if (view->mode == TOG_VIEW_SPLIT_HRZN) {
4048 		if (view->child && view->child->resized_y)
4049 			*y = view->child->resized_y;
4050 		else if (view->resized_y)
4051 			*y = view->resized_y;
4052 		else
4053 			*y = view_split_begin_y(view->lines);
4054 	} else if (view->mode == TOG_VIEW_SPLIT_VERT) {
4055 		if (view->child && view->child->resized_x)
4056 			*x = view->child->resized_x;
4057 		else if (view->resized_x)
4058 			*x = view->resized_x;
4059 		else
4060 			*x = view_split_begin_x(view->begin_x);
4061 	}
4062 }
4063 
4064 /* Split view horizontally at y and offset view->state->selected line. */
4065 static const struct got_error *
4066 view_init_hsplit(struct tog_view *view, int y)
4067 {
4068 	const struct got_error *err = NULL;
4069 
4070 	view->nlines = y;
4071 	view->ncols = COLS;
4072 	err = view_resize(view);
4073 	if (err)
4074 		return err;
4075 
4076 	err = offset_selection_down(view);
4077 
4078 	return err;
4079 }
4080 
4081 static const struct got_error *
4082 log_goto_line(struct tog_view *view, int nlines)
4083 {
4084 	const struct got_error		*err = NULL;
4085 	struct tog_log_view_state	*s = &view->state.log;
4086 	int				 g, idx = s->selected_entry->idx;
4087 
4088 	if (s->first_displayed_entry == NULL || s->last_displayed_entry == NULL)
4089 		return NULL;
4090 
4091 	g = view->gline;
4092 	view->gline = 0;
4093 
4094 	if (g >= s->first_displayed_entry->idx + 1 &&
4095 	    g <= s->last_displayed_entry->idx + 1 &&
4096 	    g - s->first_displayed_entry->idx - 1 < nlines) {
4097 		s->selected = g - s->first_displayed_entry->idx - 1;
4098 		select_commit(s);
4099 		return NULL;
4100 	}
4101 
4102 	if (idx + 1 < g) {
4103 		err = log_move_cursor_down(view, g - idx - 1);
4104 		if (!err && g > s->selected_entry->idx + 1)
4105 			err = log_move_cursor_down(view,
4106 			    g - s->first_displayed_entry->idx - 1);
4107 		if (err)
4108 			return err;
4109 	} else if (idx + 1 > g)
4110 		log_move_cursor_up(view, idx - g + 1, 0);
4111 
4112 	if (g < nlines && s->first_displayed_entry->idx == 0)
4113 		s->selected = g - 1;
4114 
4115 	select_commit(s);
4116 	return NULL;
4117 
4118 }
4119 
4120 static void
4121 horizontal_scroll_input(struct tog_view *view, int ch)
4122 {
4123 
4124 	switch (ch) {
4125 	case KEY_LEFT:
4126 	case 'h':
4127 		view->x -= MIN(view->x, 2);
4128 		if (view->x <= 0)
4129 			view->count = 0;
4130 		break;
4131 	case KEY_RIGHT:
4132 	case 'l':
4133 		if (view->x + view->ncols / 2 < view->maxx)
4134 			view->x += 2;
4135 		else
4136 			view->count = 0;
4137 		break;
4138 	case '0':
4139 		view->x = 0;
4140 		break;
4141 	case '$':
4142 		view->x = MAX(view->maxx - view->ncols / 2, 0);
4143 		view->count = 0;
4144 		break;
4145 	default:
4146 		break;
4147 	}
4148 }
4149 
4150 static const struct got_error *
4151 input_log_view(struct tog_view **new_view, struct tog_view *view, int ch)
4152 {
4153 	const struct got_error *err = NULL;
4154 	struct tog_log_view_state *s = &view->state.log;
4155 	int eos, nscroll;
4156 
4157 	if (s->thread_args.load_all) {
4158 		if (ch == CTRL('g') || ch == KEY_BACKSPACE)
4159 			s->thread_args.load_all = 0;
4160 		else if (s->thread_args.log_complete) {
4161 			err = log_move_cursor_down(view, s->commits->ncommits);
4162 			s->thread_args.load_all = 0;
4163 		}
4164 		if (err)
4165 			return err;
4166 	}
4167 
4168 	eos = nscroll = view->nlines - 1;
4169 	if (view_is_hsplit_top(view))
4170 		--eos;  /* border */
4171 
4172 	if (view->gline)
4173 		return log_goto_line(view, eos);
4174 
4175 	switch (ch) {
4176 	case '&':
4177 		err = limit_log_view(view);
4178 		break;
4179 	case 'q':
4180 		s->quit = 1;
4181 		break;
4182 	case '0':
4183 	case '$':
4184 	case KEY_RIGHT:
4185 	case 'l':
4186 	case KEY_LEFT:
4187 	case 'h':
4188 		horizontal_scroll_input(view, ch);
4189 		break;
4190 	case 'k':
4191 	case KEY_UP:
4192 	case '<':
4193 	case ',':
4194 	case CTRL('p'):
4195 		log_move_cursor_up(view, 0, 0);
4196 		break;
4197 	case 'g':
4198 	case '=':
4199 	case KEY_HOME:
4200 		log_move_cursor_up(view, 0, 1);
4201 		view->count = 0;
4202 		break;
4203 	case CTRL('u'):
4204 	case 'u':
4205 		nscroll /= 2;
4206 		/* FALL THROUGH */
4207 	case KEY_PPAGE:
4208 	case CTRL('b'):
4209 	case 'b':
4210 		log_move_cursor_up(view, nscroll, 0);
4211 		break;
4212 	case 'j':
4213 	case KEY_DOWN:
4214 	case '>':
4215 	case '.':
4216 	case CTRL('n'):
4217 		err = log_move_cursor_down(view, 0);
4218 		break;
4219 	case '@':
4220 		s->use_committer = !s->use_committer;
4221 		view->action = s->use_committer ?
4222 		    "show committer" : "show commit author";
4223 		break;
4224 	case 'G':
4225 	case '*':
4226 	case KEY_END: {
4227 		/* We don't know yet how many commits, so we're forced to
4228 		 * traverse them all. */
4229 		view->count = 0;
4230 		s->thread_args.load_all = 1;
4231 		if (!s->thread_args.log_complete)
4232 			return trigger_log_thread(view, 0);
4233 		err = log_move_cursor_down(view, s->commits->ncommits);
4234 		s->thread_args.load_all = 0;
4235 		break;
4236 	}
4237 	case CTRL('d'):
4238 	case 'd':
4239 		nscroll /= 2;
4240 		/* FALL THROUGH */
4241 	case KEY_NPAGE:
4242 	case CTRL('f'):
4243 	case 'f':
4244 	case ' ':
4245 		err = log_move_cursor_down(view, nscroll);
4246 		break;
4247 	case KEY_RESIZE:
4248 		if (s->selected > view->nlines - 2)
4249 			s->selected = view->nlines - 2;
4250 		if (s->selected > s->commits->ncommits - 1)
4251 			s->selected = s->commits->ncommits - 1;
4252 		select_commit(s);
4253 		if (s->commits->ncommits < view->nlines - 1 &&
4254 		    !s->thread_args.log_complete) {
4255 			s->thread_args.commits_needed += (view->nlines - 1) -
4256 			    s->commits->ncommits;
4257 			err = trigger_log_thread(view, 1);
4258 		}
4259 		break;
4260 	case KEY_ENTER:
4261 	case '\r':
4262 		view->count = 0;
4263 		if (s->selected_entry == NULL)
4264 			break;
4265 		err = view_request_new(new_view, view, TOG_VIEW_DIFF);
4266 		break;
4267 	case 'T':
4268 		view->count = 0;
4269 		if (s->selected_entry == NULL)
4270 			break;
4271 		err = view_request_new(new_view, view, TOG_VIEW_TREE);
4272 		break;
4273 	case KEY_BACKSPACE:
4274 	case CTRL('l'):
4275 	case 'B':
4276 		view->count = 0;
4277 		if (ch == KEY_BACKSPACE &&
4278 		    got_path_is_root_dir(s->in_repo_path))
4279 			break;
4280 		err = stop_log_thread(s);
4281 		if (err)
4282 			return err;
4283 		if (ch == KEY_BACKSPACE) {
4284 			char *parent_path;
4285 			err = got_path_dirname(&parent_path, s->in_repo_path);
4286 			if (err)
4287 				return err;
4288 			free(s->in_repo_path);
4289 			s->in_repo_path = parent_path;
4290 			s->thread_args.in_repo_path = s->in_repo_path;
4291 		} else if (ch == CTRL('l')) {
4292 			struct got_object_id *start_id;
4293 			err = got_repo_match_object_id(&start_id, NULL,
4294 			    s->head_ref_name ? s->head_ref_name : GOT_REF_HEAD,
4295 			    GOT_OBJ_TYPE_COMMIT, &tog_refs, s->repo);
4296 			if (err) {
4297 				if (s->head_ref_name == NULL ||
4298 				    err->code != GOT_ERR_NOT_REF)
4299 					return err;
4300 				/* Try to cope with deleted references. */
4301 				free(s->head_ref_name);
4302 				s->head_ref_name = NULL;
4303 				err = got_repo_match_object_id(&start_id,
4304 				    NULL, GOT_REF_HEAD, GOT_OBJ_TYPE_COMMIT,
4305 				    &tog_refs, s->repo);
4306 				if (err)
4307 					return err;
4308 			}
4309 			free(s->start_id);
4310 			s->start_id = start_id;
4311 			s->thread_args.start_id = s->start_id;
4312 		} else /* 'B' */
4313 			s->log_branches = !s->log_branches;
4314 
4315 		if (s->thread_args.pack_fds == NULL) {
4316 			err = got_repo_pack_fds_open(&s->thread_args.pack_fds);
4317 			if (err)
4318 				return err;
4319 		}
4320 		err = got_repo_open(&s->thread_args.repo,
4321 		    got_repo_get_path(s->repo), NULL,
4322 		    s->thread_args.pack_fds);
4323 		if (err)
4324 			return err;
4325 		tog_free_refs();
4326 		err = tog_load_refs(s->repo, 0);
4327 		if (err)
4328 			return err;
4329 		err = got_commit_graph_open(&s->thread_args.graph,
4330 		    s->in_repo_path, !s->log_branches);
4331 		if (err)
4332 			return err;
4333 		err = got_commit_graph_iter_start(s->thread_args.graph,
4334 		    s->start_id, s->repo, NULL, NULL);
4335 		if (err)
4336 			return err;
4337 		free_commits(&s->real_commits);
4338 		free_commits(&s->limit_commits);
4339 		s->first_displayed_entry = NULL;
4340 		s->last_displayed_entry = NULL;
4341 		s->selected_entry = NULL;
4342 		s->selected = 0;
4343 		s->thread_args.log_complete = 0;
4344 		s->quit = 0;
4345 		s->thread_args.commits_needed = view->lines;
4346 		s->matched_entry = NULL;
4347 		s->search_entry = NULL;
4348 		view->offset = 0;
4349 		break;
4350 	case 'R':
4351 		view->count = 0;
4352 		err = view_request_new(new_view, view, TOG_VIEW_REF);
4353 		break;
4354 	default:
4355 		view->count = 0;
4356 		break;
4357 	}
4358 
4359 	return err;
4360 }
4361 
4362 static const struct got_error *
4363 apply_unveil(const char *repo_path, const char *worktree_path)
4364 {
4365 	const struct got_error *error;
4366 
4367 #ifdef PROFILE
4368 	if (unveil("gmon.out", "rwc") != 0)
4369 		return got_error_from_errno2("unveil", "gmon.out");
4370 #endif
4371 	if (repo_path && unveil(repo_path, "r") != 0)
4372 		return got_error_from_errno2("unveil", repo_path);
4373 
4374 	if (worktree_path && unveil(worktree_path, "rwc") != 0)
4375 		return got_error_from_errno2("unveil", worktree_path);
4376 
4377 	if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
4378 		return got_error_from_errno2("unveil", GOT_TMPDIR_STR);
4379 
4380 	error = got_privsep_unveil_exec_helpers();
4381 	if (error != NULL)
4382 		return error;
4383 
4384 	if (unveil(NULL, NULL) != 0)
4385 		return got_error_from_errno("unveil");
4386 
4387 	return NULL;
4388 }
4389 
4390 static const struct got_error *
4391 init_mock_term(const char *test_script_path)
4392 {
4393 	const struct got_error	*err = NULL;
4394 	const char *screen_dump_path;
4395 	int in;
4396 
4397 	if (test_script_path == NULL || *test_script_path == '\0')
4398 		return got_error_msg(GOT_ERR_IO, "TOG_TEST_SCRIPT not defined");
4399 
4400 	tog_io.f = fopen(test_script_path, "re");
4401 	if (tog_io.f == NULL) {
4402 		err = got_error_from_errno_fmt("fopen: %s",
4403 		    test_script_path);
4404 		goto done;
4405 	}
4406 
4407 	/* test mode, we don't want any output */
4408 	tog_io.cout = fopen("/dev/null", "w+");
4409 	if (tog_io.cout == NULL) {
4410 		err = got_error_from_errno2("fopen", "/dev/null");
4411 		goto done;
4412 	}
4413 
4414 	in = dup(fileno(tog_io.cout));
4415 	if (in == -1) {
4416 		err = got_error_from_errno("dup");
4417 		goto done;
4418 	}
4419 	tog_io.cin = fdopen(in, "r");
4420 	if (tog_io.cin == NULL) {
4421 		err = got_error_from_errno("fdopen");
4422 		close(in);
4423 		goto done;
4424 	}
4425 
4426 	screen_dump_path = getenv("TOG_SCR_DUMP");
4427 	if (screen_dump_path == NULL || *screen_dump_path == '\0')
4428 		return got_error_msg(GOT_ERR_IO, "TOG_SCR_DUMP not defined");
4429 	tog_io.sdump = fopen(screen_dump_path, "we");
4430 	if (tog_io.sdump == NULL) {
4431 		err = got_error_from_errno2("fopen", screen_dump_path);
4432 		goto done;
4433 	}
4434 
4435 	if (fseeko(tog_io.f, 0L, SEEK_SET) == -1) {
4436 		err = got_error_from_errno("fseeko");
4437 		goto done;
4438 	}
4439 
4440 	if (newterm(NULL, tog_io.cout, tog_io.cin) == NULL)
4441 		err = got_error_msg(GOT_ERR_IO,
4442 		    "newterm: failed to initialise curses");
4443 
4444 	using_mock_io = 1;
4445 
4446 done:
4447 	if (err)
4448 		tog_io_close();
4449 	return err;
4450 }
4451 
4452 static void
4453 init_curses(void)
4454 {
4455 	if (using_mock_io) /* In test mode we use a fake terminal */
4456 		return;
4457 
4458 	initscr();
4459 
4460 	cbreak();
4461 	halfdelay(1); /* Fast refresh while initial view is loading. */
4462 	noecho();
4463 	nonl();
4464 	intrflush(stdscr, FALSE);
4465 	keypad(stdscr, TRUE);
4466 	curs_set(0);
4467 	if (getenv("TOG_COLORS") != NULL) {
4468 		start_color();
4469 		use_default_colors();
4470 	}
4471 
4472 	return;
4473 }
4474 
4475 static const struct got_error *
4476 set_tog_base_commit(struct got_repository *repo, struct got_worktree *worktree)
4477 {
4478 	tog_base_commit.id = got_object_id_dup(
4479 	    got_worktree_get_base_commit_id(worktree));
4480 	if (tog_base_commit.id == NULL)
4481 		return got_error_from_errno( "got_object_id_dup");
4482 
4483 	return NULL;
4484 }
4485 
4486 static const struct got_error *
4487 get_in_repo_path_from_argv0(char **in_repo_path, int argc, char *argv[],
4488     struct got_repository *repo, struct got_worktree *worktree)
4489 {
4490 	const struct got_error *err = NULL;
4491 
4492 	if (argc == 0) {
4493 		*in_repo_path = strdup("/");
4494 		if (*in_repo_path == NULL)
4495 			return got_error_from_errno("strdup");
4496 		return NULL;
4497 	}
4498 
4499 	if (worktree) {
4500 		const char *prefix = got_worktree_get_path_prefix(worktree);
4501 		char *p;
4502 
4503 		err = got_worktree_resolve_path(&p, worktree, argv[0]);
4504 		if (err)
4505 			return err;
4506 		if (asprintf(in_repo_path, "%s%s%s", prefix,
4507 		    (p[0] != '\0' && !got_path_is_root_dir(prefix)) ? "/" : "",
4508 		    p) == -1) {
4509 			err = got_error_from_errno("asprintf");
4510 			*in_repo_path = NULL;
4511 		}
4512 		free(p);
4513 	} else
4514 		err = got_repo_map_path(in_repo_path, repo, argv[0]);
4515 
4516 	return err;
4517 }
4518 
4519 static const struct got_error *
4520 cmd_log(int argc, char *argv[])
4521 {
4522 	const struct got_error *error;
4523 	struct got_repository *repo = NULL;
4524 	struct got_worktree *worktree = NULL;
4525 	struct got_object_id *start_id = NULL;
4526 	char *in_repo_path = NULL, *repo_path = NULL, *cwd = NULL;
4527 	char *keyword_idstr = NULL, *start_commit = NULL, *label = NULL;
4528 	struct got_reference *ref = NULL;
4529 	const char *head_ref_name = NULL;
4530 	int ch, log_branches = 0;
4531 	struct tog_view *view;
4532 	int *pack_fds = NULL;
4533 
4534 	while ((ch = getopt(argc, argv, "bc:r:")) != -1) {
4535 		switch (ch) {
4536 		case 'b':
4537 			log_branches = 1;
4538 			break;
4539 		case 'c':
4540 			start_commit = optarg;
4541 			break;
4542 		case 'r':
4543 			repo_path = realpath(optarg, NULL);
4544 			if (repo_path == NULL)
4545 				return got_error_from_errno2("realpath",
4546 				    optarg);
4547 			break;
4548 		default:
4549 			usage_log();
4550 			/* NOTREACHED */
4551 		}
4552 	}
4553 
4554 	argc -= optind;
4555 	argv += optind;
4556 
4557 	if (argc > 1)
4558 		usage_log();
4559 
4560 	error = got_repo_pack_fds_open(&pack_fds);
4561 	if (error != NULL)
4562 		goto done;
4563 
4564 	if (repo_path == NULL) {
4565 		cwd = getcwd(NULL, 0);
4566 		if (cwd == NULL) {
4567 			error = got_error_from_errno("getcwd");
4568 			goto done;
4569 		}
4570 		error = got_worktree_open(&worktree, cwd, NULL);
4571 		if (error && error->code != GOT_ERR_NOT_WORKTREE)
4572 			goto done;
4573 		if (worktree)
4574 			repo_path =
4575 			    strdup(got_worktree_get_repo_path(worktree));
4576 		else
4577 			repo_path = strdup(cwd);
4578 		if (repo_path == NULL) {
4579 			error = got_error_from_errno("strdup");
4580 			goto done;
4581 		}
4582 	}
4583 
4584 	error = got_repo_open(&repo, repo_path, NULL, pack_fds);
4585 	if (error != NULL)
4586 		goto done;
4587 
4588 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
4589 	    repo, worktree);
4590 	if (error)
4591 		goto done;
4592 
4593 	init_curses();
4594 
4595 	error = apply_unveil(got_repo_get_path(repo),
4596 	    worktree ? got_worktree_get_root_path(worktree) : NULL);
4597 	if (error)
4598 		goto done;
4599 
4600 	/* already loaded by tog_log_with_path()? */
4601 	if (TAILQ_EMPTY(&tog_refs)) {
4602 		error = tog_load_refs(repo, 0);
4603 		if (error)
4604 			goto done;
4605 	}
4606 
4607 	if (start_commit == NULL) {
4608 		error = got_repo_match_object_id(&start_id, &label,
4609 		    worktree ? got_worktree_get_head_ref_name(worktree) :
4610 		    GOT_REF_HEAD, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
4611 		if (error)
4612 			goto done;
4613 		head_ref_name = label;
4614 	} else {
4615 		error = got_keyword_to_idstr(&keyword_idstr, start_commit,
4616 		    repo, worktree);
4617 		if (error != NULL)
4618 			goto done;
4619 		if (keyword_idstr != NULL)
4620 			start_commit = keyword_idstr;
4621 
4622 		error = got_ref_open(&ref, repo, start_commit, 0);
4623 		if (error == NULL)
4624 			head_ref_name = got_ref_get_name(ref);
4625 		else if (error->code != GOT_ERR_NOT_REF)
4626 			goto done;
4627 		error = got_repo_match_object_id(&start_id, NULL,
4628 		    start_commit, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
4629 		if (error)
4630 			goto done;
4631 	}
4632 
4633 	view = view_open(0, 0, 0, 0, TOG_VIEW_LOG);
4634 	if (view == NULL) {
4635 		error = got_error_from_errno("view_open");
4636 		goto done;
4637 	}
4638 
4639 	if (worktree) {
4640 		error = set_tog_base_commit(repo, worktree);
4641 		if (error != NULL)
4642 			goto done;
4643 	}
4644 
4645 	error = open_log_view(view, start_id, repo, head_ref_name,
4646 	    in_repo_path, log_branches, worktree);
4647 	if (error)
4648 		goto done;
4649 
4650 	if (worktree) {
4651 		/* The work tree will be closed by the log thread. */
4652 		worktree = NULL;
4653 	}
4654 
4655 	error = view_loop(view);
4656 
4657 done:
4658 	free(tog_base_commit.id);
4659 	free(keyword_idstr);
4660 	free(in_repo_path);
4661 	free(repo_path);
4662 	free(cwd);
4663 	free(start_id);
4664 	free(label);
4665 	if (ref)
4666 		got_ref_close(ref);
4667 	if (repo) {
4668 		const struct got_error *close_err = got_repo_close(repo);
4669 		if (error == NULL)
4670 			error = close_err;
4671 	}
4672 	if (worktree)
4673 		got_worktree_close(worktree);
4674 	if (pack_fds) {
4675 		const struct got_error *pack_err =
4676 		    got_repo_pack_fds_close(pack_fds);
4677 		if (error == NULL)
4678 			error = pack_err;
4679 	}
4680 	tog_free_refs();
4681 	return error;
4682 }
4683 
4684 __dead static void
4685 usage_diff(void)
4686 {
4687 	endwin();
4688 	fprintf(stderr, "usage: %s diff [-aw] [-C number] [-r repository-path] "
4689 	    "object1 object2\n", getprogname());
4690 	exit(1);
4691 }
4692 
4693 static int
4694 match_line(const char *line, regex_t *regex, size_t nmatch,
4695     regmatch_t *regmatch)
4696 {
4697 	return regexec(regex, line, nmatch, regmatch, 0) == 0;
4698 }
4699 
4700 static struct tog_color *
4701 match_color(struct tog_colors *colors, const char *line)
4702 {
4703 	struct tog_color *tc = NULL;
4704 
4705 	STAILQ_FOREACH(tc, colors, entry) {
4706 		if (match_line(line, &tc->regex, 0, NULL))
4707 			return tc;
4708 	}
4709 
4710 	return NULL;
4711 }
4712 
4713 static const struct got_error *
4714 add_matched_line(int *wtotal, const char *line, int wlimit, int col_tab_align,
4715     WINDOW *window, int skipcol, regmatch_t *regmatch)
4716 {
4717 	const struct got_error *err = NULL;
4718 	char *exstr = NULL;
4719 	wchar_t *wline = NULL;
4720 	int rme, rms, n, width, scrollx;
4721 	int width0 = 0, width1 = 0, width2 = 0;
4722 	char *seg0 = NULL, *seg1 = NULL, *seg2 = NULL;
4723 
4724 	*wtotal = 0;
4725 
4726 	rms = regmatch->rm_so;
4727 	rme = regmatch->rm_eo;
4728 
4729 	err = expand_tab(&exstr, line);
4730 	if (err)
4731 		return err;
4732 
4733 	/* Split the line into 3 segments, according to match offsets. */
4734 	seg0 = strndup(exstr, rms);
4735 	if (seg0 == NULL) {
4736 		err = got_error_from_errno("strndup");
4737 		goto done;
4738 	}
4739 	seg1 = strndup(exstr + rms, rme - rms);
4740 	if (seg1 == NULL) {
4741 		err = got_error_from_errno("strndup");
4742 		goto done;
4743 	}
4744 	seg2 = strdup(exstr + rme);
4745 	if (seg2 == NULL) {
4746 		err = got_error_from_errno("strndup");
4747 		goto done;
4748 	}
4749 
4750 	/* draw up to matched token if we haven't scrolled past it */
4751 	err = format_line(&wline, &width0, NULL, seg0, 0, wlimit,
4752 	    col_tab_align, 1);
4753 	if (err)
4754 		goto done;
4755 	n = MAX(width0 - skipcol, 0);
4756 	if (n) {
4757 		free(wline);
4758 		err = format_line(&wline, &width, &scrollx, seg0, skipcol,
4759 		    wlimit, col_tab_align, 1);
4760 		if (err)
4761 			goto done;
4762 		waddwstr(window, &wline[scrollx]);
4763 		wlimit -= width;
4764 		*wtotal += width;
4765 	}
4766 
4767 	if (wlimit > 0) {
4768 		int i = 0, w = 0;
4769 		size_t wlen;
4770 
4771 		free(wline);
4772 		err = format_line(&wline, &width1, NULL, seg1, 0, wlimit,
4773 		    col_tab_align, 1);
4774 		if (err)
4775 			goto done;
4776 		wlen = wcslen(wline);
4777 		while (i < wlen) {
4778 			width = wcwidth(wline[i]);
4779 			if (width == -1) {
4780 				/* should not happen, tabs are expanded */
4781 				err = got_error(GOT_ERR_RANGE);
4782 				goto done;
4783 			}
4784 			if (width0 + w + width > skipcol)
4785 				break;
4786 			w += width;
4787 			i++;
4788 		}
4789 		/* draw (visible part of) matched token (if scrolled into it) */
4790 		if (width1 - w > 0) {
4791 			wattron(window, A_STANDOUT);
4792 			waddwstr(window, &wline[i]);
4793 			wattroff(window, A_STANDOUT);
4794 			wlimit -= (width1 - w);
4795 			*wtotal += (width1 - w);
4796 		}
4797 	}
4798 
4799 	if (wlimit > 0) {  /* draw rest of line */
4800 		free(wline);
4801 		if (skipcol > width0 + width1) {
4802 			err = format_line(&wline, &width2, &scrollx, seg2,
4803 			    skipcol - (width0 + width1), wlimit,
4804 			    col_tab_align, 1);
4805 			if (err)
4806 				goto done;
4807 			waddwstr(window, &wline[scrollx]);
4808 		} else {
4809 			err = format_line(&wline, &width2, NULL, seg2, 0,
4810 			    wlimit, col_tab_align, 1);
4811 			if (err)
4812 				goto done;
4813 			waddwstr(window, wline);
4814 		}
4815 		*wtotal += width2;
4816 	}
4817 done:
4818 	free(wline);
4819 	free(exstr);
4820 	free(seg0);
4821 	free(seg1);
4822 	free(seg2);
4823 	return err;
4824 }
4825 
4826 static int
4827 gotoline(struct tog_view *view, int *lineno, int *nprinted)
4828 {
4829 	FILE	*f = NULL;
4830 	int	*eof, *first, *selected;
4831 
4832 	if (view->type == TOG_VIEW_DIFF) {
4833 		struct tog_diff_view_state *s = &view->state.diff;
4834 
4835 		first = &s->first_displayed_line;
4836 		selected = first;
4837 		eof = &s->eof;
4838 		f = s->f;
4839 	} else if (view->type == TOG_VIEW_HELP) {
4840 		struct tog_help_view_state *s = &view->state.help;
4841 
4842 		first = &s->first_displayed_line;
4843 		selected = first;
4844 		eof = &s->eof;
4845 		f = s->f;
4846 	} else if (view->type == TOG_VIEW_BLAME) {
4847 		struct tog_blame_view_state *s = &view->state.blame;
4848 
4849 		first = &s->first_displayed_line;
4850 		selected = &s->selected_line;
4851 		eof = &s->eof;
4852 		f = s->blame.f;
4853 	} else
4854 		return 0;
4855 
4856 	/* Center gline in the middle of the page like vi(1). */
4857 	if (*lineno < view->gline - (view->nlines - 3) / 2)
4858 		return 0;
4859 	if (*first != 1 && (*lineno > view->gline - (view->nlines - 3) / 2)) {
4860 		rewind(f);
4861 		*eof = 0;
4862 		*first = 1;
4863 		*lineno = 0;
4864 		*nprinted = 0;
4865 		return 0;
4866 	}
4867 
4868 	*selected = view->gline <= (view->nlines - 3) / 2 ?
4869 		view->gline : (view->nlines - 3) / 2 + 1;
4870 	view->gline = 0;
4871 
4872 	return 1;
4873 }
4874 
4875 static const struct got_error *
4876 draw_file(struct tog_view *view, const char *header)
4877 {
4878 	struct tog_diff_view_state *s = &view->state.diff;
4879 	regmatch_t *regmatch = &view->regmatch;
4880 	const struct got_error *err;
4881 	int nprinted = 0;
4882 	char *line;
4883 	size_t linesize = 0;
4884 	ssize_t linelen;
4885 	wchar_t *wline;
4886 	int width;
4887 	int max_lines = view->nlines;
4888 	int nlines = s->nlines;
4889 	off_t line_offset;
4890 
4891 	s->lineno = s->first_displayed_line - 1;
4892 	line_offset = s->lines[s->first_displayed_line - 1].offset;
4893 	if (fseeko(s->f, line_offset, SEEK_SET) == -1)
4894 		return got_error_from_errno("fseek");
4895 
4896 	werase(view->window);
4897 
4898 	if (view->gline > s->nlines - 1)
4899 		view->gline = s->nlines - 1;
4900 
4901 	if (header) {
4902 		int ln = view->gline ? view->gline <= (view->nlines - 3) / 2 ?
4903 		    1 : view->gline - (view->nlines - 3) / 2 :
4904 		    s->lineno + s->selected_line;
4905 
4906 		if (asprintf(&line, "[%d/%d] %s", ln, nlines, header) == -1)
4907 			return got_error_from_errno("asprintf");
4908 		err = format_line(&wline, &width, NULL, line, 0, view->ncols,
4909 		    0, 0);
4910 		free(line);
4911 		if (err)
4912 			return err;
4913 
4914 		if (view_needs_focus_indication(view))
4915 			wstandout(view->window);
4916 		waddwstr(view->window, wline);
4917 		free(wline);
4918 		wline = NULL;
4919 		while (width++ < view->ncols)
4920 			waddch(view->window, ' ');
4921 		if (view_needs_focus_indication(view))
4922 			wstandend(view->window);
4923 
4924 		if (max_lines <= 1)
4925 			return NULL;
4926 		max_lines--;
4927 	}
4928 
4929 	s->eof = 0;
4930 	view->maxx = 0;
4931 	line = NULL;
4932 	while (max_lines > 0 && nprinted < max_lines) {
4933 		enum got_diff_line_type linetype;
4934 		attr_t attr = 0;
4935 
4936 		linelen = getline(&line, &linesize, s->f);
4937 		if (linelen == -1) {
4938 			if (feof(s->f)) {
4939 				s->eof = 1;
4940 				break;
4941 			}
4942 			free(line);
4943 			return got_ferror(s->f, GOT_ERR_IO);
4944 		}
4945 
4946 		if (++s->lineno < s->first_displayed_line)
4947 			continue;
4948 		if (view->gline && !gotoline(view, &s->lineno, &nprinted))
4949 			continue;
4950 		if (s->lineno == view->hiline)
4951 			attr = A_STANDOUT;
4952 
4953 		/* Set view->maxx based on full line length. */
4954 		err = format_line(&wline, &width, NULL, line, 0, INT_MAX, 0,
4955 		    view->x ? 1 : 0);
4956 		if (err) {
4957 			free(line);
4958 			return err;
4959 		}
4960 		view->maxx = MAX(view->maxx, width);
4961 		free(wline);
4962 		wline = NULL;
4963 
4964 		linetype = s->lines[s->lineno].type;
4965 		if (linetype > GOT_DIFF_LINE_LOGMSG &&
4966 		    linetype < GOT_DIFF_LINE_CONTEXT)
4967 			attr |= COLOR_PAIR(linetype);
4968 		if (attr)
4969 			wattron(view->window, attr);
4970 		if (s->first_displayed_line + nprinted == s->matched_line &&
4971 		    regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
4972 			err = add_matched_line(&width, line, view->ncols, 0,
4973 			    view->window, view->x, regmatch);
4974 			if (err) {
4975 				free(line);
4976 				return err;
4977 			}
4978 		} else {
4979 			int skip;
4980 			err = format_line(&wline, &width, &skip, line,
4981 			    view->x, view->ncols, 0, view->x ? 1 : 0);
4982 			if (err) {
4983 				free(line);
4984 				return err;
4985 			}
4986 			waddwstr(view->window, &wline[skip]);
4987 			free(wline);
4988 			wline = NULL;
4989 		}
4990 		if (s->lineno == view->hiline) {
4991 			/* highlight full gline length */
4992 			while (width++ < view->ncols)
4993 				waddch(view->window, ' ');
4994 		} else {
4995 			if (width <= view->ncols - 1)
4996 				waddch(view->window, '\n');
4997 		}
4998 		if (attr)
4999 			wattroff(view->window, attr);
5000 		if (++nprinted == 1)
5001 			s->first_displayed_line = s->lineno;
5002 	}
5003 	free(line);
5004 	if (nprinted >= 1)
5005 		s->last_displayed_line = s->first_displayed_line +
5006 		    (nprinted - 1);
5007 	else
5008 		s->last_displayed_line = s->first_displayed_line;
5009 
5010 	view_border(view);
5011 
5012 	if (s->eof) {
5013 		while (nprinted < view->nlines) {
5014 			waddch(view->window, '\n');
5015 			nprinted++;
5016 		}
5017 
5018 		err = format_line(&wline, &width, NULL, TOG_EOF_STRING, 0,
5019 		    view->ncols, 0, 0);
5020 		if (err) {
5021 			return err;
5022 		}
5023 
5024 		wstandout(view->window);
5025 		waddwstr(view->window, wline);
5026 		free(wline);
5027 		wline = NULL;
5028 		wstandend(view->window);
5029 	}
5030 
5031 	return NULL;
5032 }
5033 
5034 static char *
5035 get_datestr(time_t *time, char *datebuf)
5036 {
5037 	struct tm mytm, *tm;
5038 	char *p, *s;
5039 
5040 	tm = gmtime_r(time, &mytm);
5041 	if (tm == NULL)
5042 		return NULL;
5043 	s = asctime_r(tm, datebuf);
5044 	if (s == NULL)
5045 		return NULL;
5046 	p = strchr(s, '\n');
5047 	if (p)
5048 		*p = '\0';
5049 	return s;
5050 }
5051 
5052 static const struct got_error *
5053 add_line_metadata(struct got_diff_line **lines, size_t *nlines,
5054     off_t off, uint8_t type)
5055 {
5056 	struct got_diff_line *p;
5057 
5058 	p = reallocarray(*lines, *nlines + 1, sizeof(**lines));
5059 	if (p == NULL)
5060 		return got_error_from_errno("reallocarray");
5061 	*lines = p;
5062 	(*lines)[*nlines].offset = off;
5063 	(*lines)[*nlines].type = type;
5064 	(*nlines)++;
5065 
5066 	return NULL;
5067 }
5068 
5069 static const struct got_error *
5070 cat_diff(FILE *dst, FILE *src, struct got_diff_line **d_lines, size_t *d_nlines,
5071     struct got_diff_line *s_lines, size_t s_nlines)
5072 {
5073 	struct got_diff_line	*p;
5074 	char			 buf[BUFSIZ];
5075 	size_t			 i, r;
5076 
5077 	if (fseeko(src, 0L, SEEK_SET) == -1)
5078 		return got_error_from_errno("fseeko");
5079 
5080 	for (;;) {
5081 		r = fread(buf, 1, sizeof(buf), src);
5082 		if (r == 0) {
5083 			if (ferror(src))
5084 				return got_error_from_errno("fread");
5085 			if (feof(src))
5086 				break;
5087 		}
5088 		if (fwrite(buf, 1, r, dst) != r)
5089 			return got_ferror(dst, GOT_ERR_IO);
5090 	}
5091 
5092 	if (s_nlines == 0 && *d_nlines == 0)
5093 		return NULL;
5094 
5095 	/*
5096 	 * If commit info was in dst, increment line offsets
5097 	 * of the appended diff content, but skip s_lines[0]
5098 	 * because offset zero is already in *d_lines.
5099 	 */
5100 	if (*d_nlines > 0) {
5101 		for (i = 1; i < s_nlines; ++i)
5102 			s_lines[i].offset += (*d_lines)[*d_nlines - 1].offset;
5103 
5104 		if (s_nlines > 0) {
5105 			--s_nlines;
5106 			++s_lines;
5107 		}
5108 	}
5109 
5110 	p = reallocarray(*d_lines, *d_nlines + s_nlines, sizeof(*p));
5111 	if (p == NULL) {
5112 		/* d_lines is freed in close_diff_view() */
5113 		return got_error_from_errno("reallocarray");
5114 	}
5115 
5116 	*d_lines = p;
5117 
5118 	memcpy(*d_lines + *d_nlines, s_lines, s_nlines * sizeof(*s_lines));
5119 	*d_nlines += s_nlines;
5120 
5121 	return NULL;
5122 }
5123 
5124 static const struct got_error *
5125 write_commit_info(struct got_diff_line **lines, size_t *nlines,
5126     struct got_object_id *commit_id, struct got_reflist_head *refs,
5127     struct got_repository *repo, int ignore_ws, int force_text_diff,
5128     struct got_diffstat_cb_arg *dsa, FILE *outfile)
5129 {
5130 	const struct got_error *err = NULL;
5131 	char datebuf[26], *datestr;
5132 	struct got_commit_object *commit;
5133 	char *id_str = NULL, *logmsg = NULL, *s = NULL, *line;
5134 	time_t committer_time;
5135 	const char *author, *committer;
5136 	char *refs_str = NULL;
5137 	struct got_pathlist_entry *pe;
5138 	off_t outoff = 0;
5139 	int n;
5140 
5141 	err = build_refs_str(&refs_str, refs, commit_id, repo);
5142 	if (err)
5143 		return err;
5144 
5145 	err = got_object_open_as_commit(&commit, repo, commit_id);
5146 	if (err)
5147 		return err;
5148 
5149 	err = got_object_id_str(&id_str, commit_id);
5150 	if (err) {
5151 		err = got_error_from_errno("got_object_id_str");
5152 		goto done;
5153 	}
5154 
5155 	err = add_line_metadata(lines, nlines, 0, GOT_DIFF_LINE_NONE);
5156 	if (err)
5157 		goto done;
5158 
5159 	n = fprintf(outfile, "commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
5160 	    refs_str ? refs_str : "", refs_str ? ")" : "");
5161 	if (n < 0) {
5162 		err = got_error_from_errno("fprintf");
5163 		goto done;
5164 	}
5165 	outoff += n;
5166 	err = add_line_metadata(lines, nlines, outoff, GOT_DIFF_LINE_META);
5167 	if (err)
5168 		goto done;
5169 
5170 	n = fprintf(outfile, "from: %s\n",
5171 	    got_object_commit_get_author(commit));
5172 	if (n < 0) {
5173 		err = got_error_from_errno("fprintf");
5174 		goto done;
5175 	}
5176 	outoff += n;
5177 	err = add_line_metadata(lines, nlines, outoff, GOT_DIFF_LINE_AUTHOR);
5178 	if (err)
5179 		goto done;
5180 
5181 	author = got_object_commit_get_author(commit);
5182 	committer = got_object_commit_get_committer(commit);
5183 	if (strcmp(author, committer) != 0) {
5184 		n = fprintf(outfile, "via: %s\n", committer);
5185 		if (n < 0) {
5186 			err = got_error_from_errno("fprintf");
5187 			goto done;
5188 		}
5189 		outoff += n;
5190 		err = add_line_metadata(lines, nlines, outoff,
5191 		    GOT_DIFF_LINE_AUTHOR);
5192 		if (err)
5193 			goto done;
5194 	}
5195 	committer_time = got_object_commit_get_committer_time(commit);
5196 	datestr = get_datestr(&committer_time, datebuf);
5197 	if (datestr) {
5198 		n = fprintf(outfile, "date: %s UTC\n", datestr);
5199 		if (n < 0) {
5200 			err = got_error_from_errno("fprintf");
5201 			goto done;
5202 		}
5203 		outoff += n;
5204 		err = add_line_metadata(lines, nlines, outoff,
5205 		    GOT_DIFF_LINE_DATE);
5206 		if (err)
5207 			goto done;
5208 	}
5209 	if (got_object_commit_get_nparents(commit) > 1) {
5210 		const struct got_object_id_queue *parent_ids;
5211 		struct got_object_qid *qid;
5212 		int pn = 1;
5213 		parent_ids = got_object_commit_get_parent_ids(commit);
5214 		STAILQ_FOREACH(qid, parent_ids, entry) {
5215 			err = got_object_id_str(&id_str, &qid->id);
5216 			if (err)
5217 				goto done;
5218 			n = fprintf(outfile, "parent %d: %s\n", pn++, id_str);
5219 			if (n < 0) {
5220 				err = got_error_from_errno("fprintf");
5221 				goto done;
5222 			}
5223 			outoff += n;
5224 			err = add_line_metadata(lines, nlines, outoff,
5225 			    GOT_DIFF_LINE_META);
5226 			if (err)
5227 				goto done;
5228 			free(id_str);
5229 			id_str = NULL;
5230 		}
5231 	}
5232 
5233 	err = got_object_commit_get_logmsg(&logmsg, commit);
5234 	if (err)
5235 		goto done;
5236 	s = logmsg;
5237 	while ((line = strsep(&s, "\n")) != NULL) {
5238 		n = fprintf(outfile, "%s\n", line);
5239 		if (n < 0) {
5240 			err = got_error_from_errno("fprintf");
5241 			goto done;
5242 		}
5243 		outoff += n;
5244 		err = add_line_metadata(lines, nlines, outoff,
5245 		    GOT_DIFF_LINE_LOGMSG);
5246 		if (err)
5247 			goto done;
5248 	}
5249 
5250 	TAILQ_FOREACH(pe, dsa->paths, entry) {
5251 		struct got_diff_changed_path *cp = pe->data;
5252 		int pad = dsa->max_path_len - pe->path_len + 1;
5253 
5254 		n = fprintf(outfile, "%c  %s%*c | %*d+ %*d-\n", cp->status,
5255 		    pe->path, pad, ' ', dsa->add_cols + 1, cp->add,
5256 		    dsa->rm_cols + 1, cp->rm);
5257 		if (n < 0) {
5258 			err = got_error_from_errno("fprintf");
5259 			goto done;
5260 		}
5261 		outoff += n;
5262 		err = add_line_metadata(lines, nlines, outoff,
5263 		    GOT_DIFF_LINE_CHANGES);
5264 		if (err)
5265 			goto done;
5266 	}
5267 
5268 	fputc('\n', outfile);
5269 	outoff++;
5270 	err = add_line_metadata(lines, nlines, outoff, GOT_DIFF_LINE_NONE);
5271 	if (err)
5272 		goto done;
5273 
5274 	n = fprintf(outfile,
5275 	    "%d file%s changed, %d insertion%s(+), %d deletion%s(-)\n",
5276 	    dsa->nfiles, dsa->nfiles > 1 ? "s" : "", dsa->ins,
5277 	    dsa->ins != 1 ? "s" : "", dsa->del, dsa->del != 1 ? "s" : "");
5278 	if (n < 0) {
5279 		err = got_error_from_errno("fprintf");
5280 		goto done;
5281 	}
5282 	outoff += n;
5283 	err = add_line_metadata(lines, nlines, outoff, GOT_DIFF_LINE_NONE);
5284 	if (err)
5285 		goto done;
5286 
5287 	fputc('\n', outfile);
5288 	outoff++;
5289 	err = add_line_metadata(lines, nlines, outoff, GOT_DIFF_LINE_NONE);
5290 done:
5291 	free(id_str);
5292 	free(logmsg);
5293 	free(refs_str);
5294 	got_object_commit_close(commit);
5295 	if (err) {
5296 		free(*lines);
5297 		*lines = NULL;
5298 		*nlines = 0;
5299 	}
5300 	return err;
5301 }
5302 
5303 static const struct got_error *
5304 create_diff(struct tog_diff_view_state *s)
5305 {
5306 	const struct got_error *err = NULL;
5307 	FILE *f = NULL, *tmp_diff_file = NULL;
5308 	int obj_type;
5309 	struct got_diff_line *lines = NULL;
5310 	struct got_pathlist_head changed_paths;
5311 	struct got_commit_object *commit2 = NULL;
5312 
5313 	TAILQ_INIT(&changed_paths);
5314 
5315 	free(s->lines);
5316 	s->lines = malloc(sizeof(*s->lines));
5317 	if (s->lines == NULL)
5318 		return got_error_from_errno("malloc");
5319 	s->nlines = 0;
5320 
5321 	f = got_opentemp();
5322 	if (f == NULL) {
5323 		err = got_error_from_errno("got_opentemp");
5324 		goto done;
5325 	}
5326 	tmp_diff_file = got_opentemp();
5327 	if (tmp_diff_file == NULL) {
5328 		err = got_error_from_errno("got_opentemp");
5329 		goto done;
5330 	}
5331 	if (s->f && fclose(s->f) == EOF) {
5332 		err = got_error_from_errno("fclose");
5333 		goto done;
5334 	}
5335 	s->f = f;
5336 
5337 	if (s->id1)
5338 		err = got_object_get_type(&obj_type, s->repo, s->id1);
5339 	else
5340 		err = got_object_get_type(&obj_type, s->repo, s->id2);
5341 	if (err)
5342 		goto done;
5343 
5344 	switch (obj_type) {
5345 	case GOT_OBJ_TYPE_BLOB:
5346 		err = got_diff_objects_as_blobs(&s->lines, &s->nlines,
5347 		    s->f1, s->f2, s->fd1, s->fd2, s->id1, s->id2,
5348 		    s->label1, s->label2, tog_diff_algo, s->diff_context,
5349 		    s->ignore_whitespace, s->force_text_diff, NULL, s->repo,
5350 		    s->f);
5351 		break;
5352 	case GOT_OBJ_TYPE_TREE:
5353 		err = got_diff_objects_as_trees(&s->lines, &s->nlines,
5354 		    s->f1, s->f2, s->fd1, s->fd2, s->id1, s->id2, NULL, "", "",
5355 		    tog_diff_algo, s->diff_context, s->ignore_whitespace,
5356 		    s->force_text_diff, NULL, s->repo, s->f);
5357 		break;
5358 	case GOT_OBJ_TYPE_COMMIT: {
5359 		const struct got_object_id_queue *parent_ids;
5360 		struct got_object_qid *pid;
5361 		struct got_reflist_head *refs;
5362 		size_t nlines = 0;
5363 		struct got_diffstat_cb_arg dsa = {
5364 			0, 0, 0, 0, 0, 0,
5365 			&changed_paths,
5366 			s->ignore_whitespace,
5367 			s->force_text_diff,
5368 			tog_diff_algo
5369 		};
5370 
5371 		lines = malloc(sizeof(*lines));
5372 		if (lines == NULL) {
5373 			err = got_error_from_errno("malloc");
5374 			goto done;
5375 		}
5376 
5377 		/* build diff first in tmp file then append to commit info */
5378 		err = got_diff_objects_as_commits(&lines, &nlines,
5379 		    s->f1, s->f2, s->fd1, s->fd2, s->id1, s->id2, NULL,
5380 		    tog_diff_algo, s->diff_context, s->ignore_whitespace,
5381 		    s->force_text_diff, &dsa, s->repo, tmp_diff_file);
5382 		if (err)
5383 			break;
5384 
5385 		refs = got_reflist_object_id_map_lookup(tog_refs_idmap, s->id2);
5386 		/* Show commit info if we're diffing to a parent/root commit. */
5387 		if (s->id1 == NULL) {
5388 			err = write_commit_info(&s->lines, &s->nlines, s->id2,
5389 			    refs, s->repo, s->ignore_whitespace,
5390 			    s->force_text_diff, &dsa, s->f);
5391 			if (err)
5392 				goto done;
5393 		} else {
5394 			err = got_object_open_as_commit(&commit2, s->repo,
5395 			    s->id2);
5396 			if (err)
5397 				goto done;
5398 
5399 			parent_ids = got_object_commit_get_parent_ids(commit2);
5400 			STAILQ_FOREACH(pid, parent_ids, entry) {
5401 				if (got_object_id_cmp(s->id1, &pid->id) == 0) {
5402 					err = write_commit_info(&s->lines,
5403 					    &s->nlines, s->id2, refs, s->repo,
5404 					    s->ignore_whitespace,
5405 					    s->force_text_diff, &dsa, s->f);
5406 					if (err)
5407 						goto done;
5408 					break;
5409 				}
5410 			}
5411 		}
5412 
5413 		err = cat_diff(s->f, tmp_diff_file, &s->lines, &s->nlines,
5414 		    lines, nlines);
5415 		break;
5416 	}
5417 	default:
5418 		err = got_error(GOT_ERR_OBJ_TYPE);
5419 		break;
5420 	}
5421 done:
5422 	free(lines);
5423 	if (commit2 != NULL)
5424 		got_object_commit_close(commit2);
5425 	got_pathlist_free(&changed_paths, GOT_PATHLIST_FREE_ALL);
5426 	if (s->f && fflush(s->f) != 0 && err == NULL)
5427 		err = got_error_from_errno("fflush");
5428 	if (tmp_diff_file && fclose(tmp_diff_file) == EOF && err == NULL)
5429 		err = got_error_from_errno("fclose");
5430 	return err;
5431 }
5432 
5433 static void
5434 diff_view_indicate_progress(struct tog_view *view)
5435 {
5436 	mvwaddstr(view->window, 0, 0, "diffing...");
5437 	update_panels();
5438 	doupdate();
5439 }
5440 
5441 static const struct got_error *
5442 search_start_diff_view(struct tog_view *view)
5443 {
5444 	struct tog_diff_view_state *s = &view->state.diff;
5445 
5446 	s->matched_line = 0;
5447 	return NULL;
5448 }
5449 
5450 static void
5451 search_setup_diff_view(struct tog_view *view, FILE **f, off_t **line_offsets,
5452     size_t *nlines, int **first, int **last, int **match, int **selected)
5453 {
5454 	struct tog_diff_view_state *s = &view->state.diff;
5455 
5456 	*f = s->f;
5457 	*nlines = s->nlines;
5458 	*line_offsets = NULL;
5459 	*match = &s->matched_line;
5460 	*first = &s->first_displayed_line;
5461 	*last = &s->last_displayed_line;
5462 	*selected = &s->selected_line;
5463 }
5464 
5465 static const struct got_error *
5466 search_next_view_match(struct tog_view *view)
5467 {
5468 	const struct got_error *err = NULL;
5469 	FILE *f;
5470 	int lineno;
5471 	char *line = NULL;
5472 	size_t linesize = 0;
5473 	ssize_t linelen;
5474 	off_t *line_offsets;
5475 	size_t nlines = 0;
5476 	int *first, *last, *match, *selected;
5477 
5478 	if (!view->search_setup)
5479 		return got_error_msg(GOT_ERR_NOT_IMPL,
5480 		    "view search not supported");
5481 	view->search_setup(view, &f, &line_offsets, &nlines, &first, &last,
5482 	    &match, &selected);
5483 
5484 	if (!view->searching) {
5485 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
5486 		return NULL;
5487 	}
5488 
5489 	if (*match) {
5490 		if (view->searching == TOG_SEARCH_FORWARD)
5491 			lineno = *first + 1;
5492 		else
5493 			lineno = *first - 1;
5494 	} else
5495 		lineno = *first - 1 + *selected;
5496 
5497 	while (1) {
5498 		off_t offset;
5499 
5500 		if (lineno <= 0 || lineno > nlines) {
5501 			if (*match == 0) {
5502 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
5503 				break;
5504 			}
5505 
5506 			if (view->searching == TOG_SEARCH_FORWARD)
5507 				lineno = 1;
5508 			else
5509 				lineno = nlines;
5510 		}
5511 
5512 		offset = view->type == TOG_VIEW_DIFF ?
5513 		    view->state.diff.lines[lineno - 1].offset :
5514 		    line_offsets[lineno - 1];
5515 		if (fseeko(f, offset, SEEK_SET) != 0) {
5516 			free(line);
5517 			return got_error_from_errno("fseeko");
5518 		}
5519 		linelen = getline(&line, &linesize, f);
5520 		if (linelen != -1) {
5521 			char *exstr;
5522 			err = expand_tab(&exstr, line);
5523 			if (err)
5524 				break;
5525 			if (match_line(exstr, &view->regex, 1,
5526 			    &view->regmatch)) {
5527 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
5528 				*match = lineno;
5529 				free(exstr);
5530 				break;
5531 			}
5532 			free(exstr);
5533 		}
5534 		if (view->searching == TOG_SEARCH_FORWARD)
5535 			lineno++;
5536 		else
5537 			lineno--;
5538 	}
5539 	free(line);
5540 
5541 	if (*match) {
5542 		*first = *match;
5543 		*selected = 1;
5544 	}
5545 
5546 	return err;
5547 }
5548 
5549 static const struct got_error *
5550 close_diff_view(struct tog_view *view)
5551 {
5552 	const struct got_error *err = NULL;
5553 	struct tog_diff_view_state *s = &view->state.diff;
5554 
5555 	free(s->id1);
5556 	s->id1 = NULL;
5557 	free(s->id2);
5558 	s->id2 = NULL;
5559 	if (s->f && fclose(s->f) == EOF)
5560 		err = got_error_from_errno("fclose");
5561 	s->f = NULL;
5562 	if (s->f1 && fclose(s->f1) == EOF && err == NULL)
5563 		err = got_error_from_errno("fclose");
5564 	s->f1 = NULL;
5565 	if (s->f2 && fclose(s->f2) == EOF && err == NULL)
5566 		err = got_error_from_errno("fclose");
5567 	s->f2 = NULL;
5568 	if (s->fd1 != -1 && close(s->fd1) == -1 && err == NULL)
5569 		err = got_error_from_errno("close");
5570 	s->fd1 = -1;
5571 	if (s->fd2 != -1 && close(s->fd2) == -1 && err == NULL)
5572 		err = got_error_from_errno("close");
5573 	s->fd2 = -1;
5574 	free(s->lines);
5575 	s->lines = NULL;
5576 	s->nlines = 0;
5577 	return err;
5578 }
5579 
5580 static const struct got_error *
5581 open_diff_view(struct tog_view *view, struct got_object_id *id1,
5582     struct got_object_id *id2, const char *label1, const char *label2,
5583     int diff_context, int ignore_whitespace, int force_text_diff,
5584     struct tog_view *parent_view, struct got_repository *repo)
5585 {
5586 	const struct got_error *err;
5587 	struct tog_diff_view_state *s = &view->state.diff;
5588 
5589 	memset(s, 0, sizeof(*s));
5590 	s->fd1 = -1;
5591 	s->fd2 = -1;
5592 
5593 	if (id1 != NULL && id2 != NULL) {
5594 		int type1, type2;
5595 
5596 		err = got_object_get_type(&type1, repo, id1);
5597 		if (err)
5598 			goto done;
5599 		err = got_object_get_type(&type2, repo, id2);
5600 		if (err)
5601 			goto done;
5602 
5603 		if (type1 != type2) {
5604 			err = got_error(GOT_ERR_OBJ_TYPE);
5605 			goto done;
5606 		}
5607 	}
5608 	s->first_displayed_line = 1;
5609 	s->last_displayed_line = view->nlines;
5610 	s->selected_line = 1;
5611 	s->repo = repo;
5612 	s->label1 = label1;
5613 	s->label2 = label2;
5614 
5615 	if (id1) {
5616 		s->id1 = got_object_id_dup(id1);
5617 		if (s->id1 == NULL) {
5618 			err = got_error_from_errno("got_object_id_dup");
5619 			goto done;
5620 		}
5621 	} else
5622 		s->id1 = NULL;
5623 
5624 	s->id2 = got_object_id_dup(id2);
5625 	if (s->id2 == NULL) {
5626 		err = got_error_from_errno("got_object_id_dup");
5627 		goto done;
5628 	}
5629 
5630 	s->f1 = got_opentemp();
5631 	if (s->f1 == NULL) {
5632 		err = got_error_from_errno("got_opentemp");
5633 		goto done;
5634 	}
5635 
5636 	s->f2 = got_opentemp();
5637 	if (s->f2 == NULL) {
5638 		err = got_error_from_errno("got_opentemp");
5639 		goto done;
5640 	}
5641 
5642 	s->fd1 = got_opentempfd();
5643 	if (s->fd1 == -1) {
5644 		err = got_error_from_errno("got_opentempfd");
5645 		goto done;
5646 	}
5647 
5648 	s->fd2 = got_opentempfd();
5649 	if (s->fd2 == -1) {
5650 		err = got_error_from_errno("got_opentempfd");
5651 		goto done;
5652 	}
5653 
5654 	s->diff_context = diff_context;
5655 	s->ignore_whitespace = ignore_whitespace;
5656 	s->force_text_diff = force_text_diff;
5657 	s->parent_view = parent_view;
5658 	s->repo = repo;
5659 
5660 	if (has_colors() && getenv("TOG_COLORS") != NULL && !using_mock_io) {
5661 		int rc;
5662 
5663 		rc = init_pair(GOT_DIFF_LINE_MINUS,
5664 		    get_color_value("TOG_COLOR_DIFF_MINUS"), -1);
5665 		if (rc != ERR)
5666 			rc = init_pair(GOT_DIFF_LINE_PLUS,
5667 			    get_color_value("TOG_COLOR_DIFF_PLUS"), -1);
5668 		if (rc != ERR)
5669 			rc = init_pair(GOT_DIFF_LINE_HUNK,
5670 			    get_color_value("TOG_COLOR_DIFF_CHUNK_HEADER"), -1);
5671 		if (rc != ERR)
5672 			rc = init_pair(GOT_DIFF_LINE_META,
5673 			    get_color_value("TOG_COLOR_DIFF_META"), -1);
5674 		if (rc != ERR)
5675 			rc = init_pair(GOT_DIFF_LINE_CHANGES,
5676 			    get_color_value("TOG_COLOR_DIFF_META"), -1);
5677 		if (rc != ERR)
5678 			rc = init_pair(GOT_DIFF_LINE_BLOB_MIN,
5679 			    get_color_value("TOG_COLOR_DIFF_META"), -1);
5680 		if (rc != ERR)
5681 			rc = init_pair(GOT_DIFF_LINE_BLOB_PLUS,
5682 			    get_color_value("TOG_COLOR_DIFF_META"), -1);
5683 		if (rc != ERR)
5684 			rc = init_pair(GOT_DIFF_LINE_AUTHOR,
5685 			    get_color_value("TOG_COLOR_AUTHOR"), -1);
5686 		if (rc != ERR)
5687 			rc = init_pair(GOT_DIFF_LINE_DATE,
5688 			    get_color_value("TOG_COLOR_DATE"), -1);
5689 		if (rc == ERR) {
5690 			err = got_error(GOT_ERR_RANGE);
5691 			goto done;
5692 		}
5693 	}
5694 
5695 	if (parent_view && parent_view->type == TOG_VIEW_LOG &&
5696 	    view_is_splitscreen(view))
5697 		show_log_view(parent_view); /* draw border */
5698 	diff_view_indicate_progress(view);
5699 
5700 	err = create_diff(s);
5701 
5702 	view->show = show_diff_view;
5703 	view->input = input_diff_view;
5704 	view->reset = reset_diff_view;
5705 	view->close = close_diff_view;
5706 	view->search_start = search_start_diff_view;
5707 	view->search_setup = search_setup_diff_view;
5708 	view->search_next = search_next_view_match;
5709 done:
5710 	if (err) {
5711 		if (view->close == NULL)
5712 			close_diff_view(view);
5713 		view_close(view);
5714 	}
5715 	return err;
5716 }
5717 
5718 static const struct got_error *
5719 show_diff_view(struct tog_view *view)
5720 {
5721 	const struct got_error *err;
5722 	struct tog_diff_view_state *s = &view->state.diff;
5723 	char *id_str1 = NULL, *id_str2, *header;
5724 	const char *label1, *label2;
5725 
5726 	if (s->id1) {
5727 		err = got_object_id_str(&id_str1, s->id1);
5728 		if (err)
5729 			return err;
5730 		label1 = s->label1 ? s->label1 : id_str1;
5731 	} else
5732 		label1 = "/dev/null";
5733 
5734 	err = got_object_id_str(&id_str2, s->id2);
5735 	if (err)
5736 		return err;
5737 	label2 = s->label2 ? s->label2 : id_str2;
5738 
5739 	if (asprintf(&header, "diff %s %s", label1, label2) == -1) {
5740 		err = got_error_from_errno("asprintf");
5741 		free(id_str1);
5742 		free(id_str2);
5743 		return err;
5744 	}
5745 	free(id_str1);
5746 	free(id_str2);
5747 
5748 	err = draw_file(view, header);
5749 	free(header);
5750 	return err;
5751 }
5752 
5753 static const struct got_error *
5754 set_selected_commit(struct tog_diff_view_state *s,
5755     struct commit_queue_entry *entry)
5756 {
5757 	const struct got_error *err;
5758 	const struct got_object_id_queue *parent_ids;
5759 	struct got_commit_object *selected_commit;
5760 	struct got_object_qid *pid;
5761 
5762 	free(s->id2);
5763 	s->id2 = got_object_id_dup(entry->id);
5764 	if (s->id2 == NULL)
5765 		return got_error_from_errno("got_object_id_dup");
5766 
5767 	err = got_object_open_as_commit(&selected_commit, s->repo, entry->id);
5768 	if (err)
5769 		return err;
5770 	parent_ids = got_object_commit_get_parent_ids(selected_commit);
5771 	free(s->id1);
5772 	pid = STAILQ_FIRST(parent_ids);
5773 	s->id1 = pid ? got_object_id_dup(&pid->id) : NULL;
5774 	got_object_commit_close(selected_commit);
5775 	return NULL;
5776 }
5777 
5778 static const struct got_error *
5779 reset_diff_view(struct tog_view *view)
5780 {
5781 	struct tog_diff_view_state *s = &view->state.diff;
5782 
5783 	view->count = 0;
5784 	wclear(view->window);
5785 	s->first_displayed_line = 1;
5786 	s->last_displayed_line = view->nlines;
5787 	s->matched_line = 0;
5788 	diff_view_indicate_progress(view);
5789 	return create_diff(s);
5790 }
5791 
5792 static void
5793 diff_prev_index(struct tog_diff_view_state *s, enum got_diff_line_type type)
5794 {
5795 	int start, i;
5796 
5797 	i = start = s->first_displayed_line - 1;
5798 
5799 	while (s->lines[i].type != type) {
5800 		if (i == 0)
5801 			i = s->nlines - 1;
5802 		if (--i == start)
5803 			return; /* do nothing, requested type not in file */
5804 	}
5805 
5806 	s->selected_line = 1;
5807 	s->first_displayed_line = i;
5808 }
5809 
5810 static void
5811 diff_next_index(struct tog_diff_view_state *s, enum got_diff_line_type type)
5812 {
5813 	int start, i;
5814 
5815 	i = start = s->first_displayed_line + 1;
5816 
5817 	while (s->lines[i].type != type) {
5818 		if (i == s->nlines - 1)
5819 			i = 0;
5820 		if (++i == start)
5821 			return; /* do nothing, requested type not in file */
5822 	}
5823 
5824 	s->selected_line = 1;
5825 	s->first_displayed_line = i;
5826 }
5827 
5828 static struct got_object_id *get_selected_commit_id(struct tog_blame_line *,
5829     int, int, int);
5830 static struct got_object_id *get_annotation_for_line(struct tog_blame_line *,
5831     int, int);
5832 
5833 static const struct got_error *
5834 input_diff_view(struct tog_view **new_view, struct tog_view *view, int ch)
5835 {
5836 	const struct got_error *err = NULL;
5837 	struct tog_diff_view_state *s = &view->state.diff;
5838 	struct tog_log_view_state *ls;
5839 	struct commit_queue_entry *old_selected_entry;
5840 	char *line = NULL;
5841 	size_t linesize = 0;
5842 	ssize_t linelen;
5843 	int i, nscroll = view->nlines - 1, up = 0;
5844 
5845 	s->lineno = s->first_displayed_line - 1 + s->selected_line;
5846 
5847 	switch (ch) {
5848 	case '0':
5849 	case '$':
5850 	case KEY_RIGHT:
5851 	case 'l':
5852 	case KEY_LEFT:
5853 	case 'h':
5854 		horizontal_scroll_input(view, ch);
5855 		break;
5856 	case 'a':
5857 	case 'w':
5858 		if (ch == 'a') {
5859 			s->force_text_diff = !s->force_text_diff;
5860 			view->action = s->force_text_diff ?
5861 			    "force ASCII text enabled" :
5862 			    "force ASCII text disabled";
5863 		}
5864 		else if (ch == 'w') {
5865 			s->ignore_whitespace = !s->ignore_whitespace;
5866 			view->action = s->ignore_whitespace ?
5867 			    "ignore whitespace enabled" :
5868 			    "ignore whitespace disabled";
5869 		}
5870 		err = reset_diff_view(view);
5871 		break;
5872 	case 'g':
5873 	case KEY_HOME:
5874 		s->first_displayed_line = 1;
5875 		view->count = 0;
5876 		break;
5877 	case 'G':
5878 	case KEY_END:
5879 		view->count = 0;
5880 		if (s->eof)
5881 			break;
5882 
5883 		s->first_displayed_line = (s->nlines - view->nlines) + 2;
5884 		s->eof = 1;
5885 		break;
5886 	case 'k':
5887 	case KEY_UP:
5888 	case CTRL('p'):
5889 		if (s->first_displayed_line > 1)
5890 			s->first_displayed_line--;
5891 		else
5892 			view->count = 0;
5893 		break;
5894 	case CTRL('u'):
5895 	case 'u':
5896 		nscroll /= 2;
5897 		/* FALL THROUGH */
5898 	case KEY_PPAGE:
5899 	case CTRL('b'):
5900 	case 'b':
5901 		if (s->first_displayed_line == 1) {
5902 			view->count = 0;
5903 			break;
5904 		}
5905 		i = 0;
5906 		while (i++ < nscroll && s->first_displayed_line > 1)
5907 			s->first_displayed_line--;
5908 		break;
5909 	case 'j':
5910 	case KEY_DOWN:
5911 	case CTRL('n'):
5912 		if (!s->eof)
5913 			s->first_displayed_line++;
5914 		else
5915 			view->count = 0;
5916 		break;
5917 	case CTRL('d'):
5918 	case 'd':
5919 		nscroll /= 2;
5920 		/* FALL THROUGH */
5921 	case KEY_NPAGE:
5922 	case CTRL('f'):
5923 	case 'f':
5924 	case ' ':
5925 		if (s->eof) {
5926 			view->count = 0;
5927 			break;
5928 		}
5929 		i = 0;
5930 		while (!s->eof && i++ < nscroll) {
5931 			linelen = getline(&line, &linesize, s->f);
5932 			s->first_displayed_line++;
5933 			if (linelen == -1) {
5934 				if (feof(s->f)) {
5935 					s->eof = 1;
5936 				} else
5937 					err = got_ferror(s->f, GOT_ERR_IO);
5938 				break;
5939 			}
5940 		}
5941 		free(line);
5942 		break;
5943 	case '(':
5944 		diff_prev_index(s, GOT_DIFF_LINE_BLOB_MIN);
5945 		break;
5946 	case ')':
5947 		diff_next_index(s, GOT_DIFF_LINE_BLOB_MIN);
5948 		break;
5949 	case '{':
5950 		diff_prev_index(s, GOT_DIFF_LINE_HUNK);
5951 		break;
5952 	case '}':
5953 		diff_next_index(s, GOT_DIFF_LINE_HUNK);
5954 		break;
5955 	case '[':
5956 		if (s->diff_context > 0) {
5957 			s->diff_context--;
5958 			s->matched_line = 0;
5959 			diff_view_indicate_progress(view);
5960 			err = create_diff(s);
5961 			if (s->first_displayed_line + view->nlines - 1 >
5962 			    s->nlines) {
5963 				s->first_displayed_line = 1;
5964 				s->last_displayed_line = view->nlines;
5965 			}
5966 		} else
5967 			view->count = 0;
5968 		break;
5969 	case ']':
5970 		if (s->diff_context < GOT_DIFF_MAX_CONTEXT) {
5971 			s->diff_context++;
5972 			s->matched_line = 0;
5973 			diff_view_indicate_progress(view);
5974 			err = create_diff(s);
5975 		} else
5976 			view->count = 0;
5977 		break;
5978 	case '<':
5979 	case ',':
5980 	case 'K':
5981 		up = 1;
5982 		/* FALL THROUGH */
5983 	case '>':
5984 	case '.':
5985 	case 'J':
5986 		if (s->parent_view == NULL) {
5987 			view->count = 0;
5988 			break;
5989 		}
5990 		s->parent_view->count = view->count;
5991 
5992 		if (s->parent_view->type == TOG_VIEW_LOG) {
5993 			ls = &s->parent_view->state.log;
5994 			old_selected_entry = ls->selected_entry;
5995 
5996 			err = input_log_view(NULL, s->parent_view,
5997 			    up ? KEY_UP : KEY_DOWN);
5998 			if (err)
5999 				break;
6000 			view->count = s->parent_view->count;
6001 
6002 			if (old_selected_entry == ls->selected_entry)
6003 				break;
6004 
6005 			err = set_selected_commit(s, ls->selected_entry);
6006 			if (err)
6007 				break;
6008 		} else if (s->parent_view->type == TOG_VIEW_BLAME) {
6009 			struct tog_blame_view_state *bs;
6010 			struct got_object_id *id, *prev_id;
6011 
6012 			bs = &s->parent_view->state.blame;
6013 			prev_id = get_annotation_for_line(bs->blame.lines,
6014 			    bs->blame.nlines, bs->last_diffed_line);
6015 
6016 			err = input_blame_view(&view, s->parent_view,
6017 			    up ? KEY_UP : KEY_DOWN);
6018 			if (err)
6019 				break;
6020 			view->count = s->parent_view->count;
6021 
6022 			if (prev_id == NULL)
6023 				break;
6024 			id = get_selected_commit_id(bs->blame.lines,
6025 			    bs->blame.nlines, bs->first_displayed_line,
6026 			    bs->selected_line);
6027 			if (id == NULL)
6028 				break;
6029 
6030 			if (!got_object_id_cmp(prev_id, id))
6031 				break;
6032 
6033 			err = input_blame_view(&view, s->parent_view, KEY_ENTER);
6034 			if (err)
6035 				break;
6036 		}
6037 		s->first_displayed_line = 1;
6038 		s->last_displayed_line = view->nlines;
6039 		s->matched_line = 0;
6040 		view->x = 0;
6041 
6042 		diff_view_indicate_progress(view);
6043 		err = create_diff(s);
6044 		break;
6045 	default:
6046 		view->count = 0;
6047 		break;
6048 	}
6049 
6050 	return err;
6051 }
6052 
6053 static const struct got_error *
6054 cmd_diff(int argc, char *argv[])
6055 {
6056 	const struct got_error *error;
6057 	struct got_repository *repo = NULL;
6058 	struct got_worktree *worktree = NULL;
6059 	struct got_object_id *id1 = NULL, *id2 = NULL;
6060 	char *repo_path = NULL, *cwd = NULL;
6061 	char *id_str1 = NULL, *id_str2 = NULL;
6062 	char *keyword_idstr1 = NULL, *keyword_idstr2 = NULL;
6063 	char *label1 = NULL, *label2 = NULL;
6064 	int diff_context = 3, ignore_whitespace = 0;
6065 	int ch, force_text_diff = 0;
6066 	const char *errstr;
6067 	struct tog_view *view;
6068 	int *pack_fds = NULL;
6069 
6070 	while ((ch = getopt(argc, argv, "aC:r:w")) != -1) {
6071 		switch (ch) {
6072 		case 'a':
6073 			force_text_diff = 1;
6074 			break;
6075 		case 'C':
6076 			diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
6077 			    &errstr);
6078 			if (errstr != NULL)
6079 				errx(1, "number of context lines is %s: %s",
6080 				    errstr, errstr);
6081 			break;
6082 		case 'r':
6083 			repo_path = realpath(optarg, NULL);
6084 			if (repo_path == NULL)
6085 				return got_error_from_errno2("realpath",
6086 				    optarg);
6087 			got_path_strip_trailing_slashes(repo_path);
6088 			break;
6089 		case 'w':
6090 			ignore_whitespace = 1;
6091 			break;
6092 		default:
6093 			usage_diff();
6094 			/* NOTREACHED */
6095 		}
6096 	}
6097 
6098 	argc -= optind;
6099 	argv += optind;
6100 
6101 	if (argc == 0) {
6102 		usage_diff(); /* TODO show local worktree changes */
6103 	} else if (argc == 2) {
6104 		id_str1 = argv[0];
6105 		id_str2 = argv[1];
6106 	} else
6107 		usage_diff();
6108 
6109 	error = got_repo_pack_fds_open(&pack_fds);
6110 	if (error)
6111 		goto done;
6112 
6113 	if (repo_path == NULL) {
6114 		cwd = getcwd(NULL, 0);
6115 		if (cwd == NULL)
6116 			return got_error_from_errno("getcwd");
6117 		error = got_worktree_open(&worktree, cwd, NULL);
6118 		if (error && error->code != GOT_ERR_NOT_WORKTREE)
6119 			goto done;
6120 		if (worktree)
6121 			repo_path =
6122 			    strdup(got_worktree_get_repo_path(worktree));
6123 		else
6124 			repo_path = strdup(cwd);
6125 		if (repo_path == NULL) {
6126 			error = got_error_from_errno("strdup");
6127 			goto done;
6128 		}
6129 	}
6130 
6131 	error = got_repo_open(&repo, repo_path, NULL, pack_fds);
6132 	if (error)
6133 		goto done;
6134 
6135 	init_curses();
6136 
6137 	error = apply_unveil(got_repo_get_path(repo), NULL);
6138 	if (error)
6139 		goto done;
6140 
6141 	error = tog_load_refs(repo, 0);
6142 	if (error)
6143 		goto done;
6144 
6145 	if (id_str1 != NULL) {
6146 		error = got_keyword_to_idstr(&keyword_idstr1, id_str1,
6147 		    repo, worktree);
6148 		if (error != NULL)
6149 			goto done;
6150 		if (keyword_idstr1 != NULL)
6151 			id_str1 = keyword_idstr1;
6152 	}
6153 	if (id_str2 != NULL) {
6154 		error = got_keyword_to_idstr(&keyword_idstr2, id_str2,
6155 		    repo, worktree);
6156 		if (error != NULL)
6157 			goto done;
6158 		if (keyword_idstr2 != NULL)
6159 			id_str2 = keyword_idstr2;
6160 	}
6161 
6162 	error = got_repo_match_object_id(&id1, &label1, id_str1,
6163 	    GOT_OBJ_TYPE_ANY, &tog_refs, repo);
6164 	if (error)
6165 		goto done;
6166 
6167 	error = got_repo_match_object_id(&id2, &label2, id_str2,
6168 	    GOT_OBJ_TYPE_ANY, &tog_refs, repo);
6169 	if (error)
6170 		goto done;
6171 
6172 	view = view_open(0, 0, 0, 0, TOG_VIEW_DIFF);
6173 	if (view == NULL) {
6174 		error = got_error_from_errno("view_open");
6175 		goto done;
6176 	}
6177 	error = open_diff_view(view, id1, id2, label1, label2, diff_context,
6178 	    ignore_whitespace, force_text_diff, NULL,  repo);
6179 	if (error)
6180 		goto done;
6181 
6182 	if (worktree) {
6183 		error = set_tog_base_commit(repo, worktree);
6184 		if (error != NULL)
6185 			goto done;
6186 
6187 		/* Release work tree lock. */
6188 		got_worktree_close(worktree);
6189 		worktree = NULL;
6190 	}
6191 
6192 	error = view_loop(view);
6193 
6194 done:
6195 	free(tog_base_commit.id);
6196 	free(keyword_idstr1);
6197 	free(keyword_idstr2);
6198 	free(label1);
6199 	free(label2);
6200 	free(id1);
6201 	free(id2);
6202 	free(repo_path);
6203 	free(cwd);
6204 	if (repo) {
6205 		const struct got_error *close_err = got_repo_close(repo);
6206 		if (error == NULL)
6207 			error = close_err;
6208 	}
6209 	if (worktree)
6210 		got_worktree_close(worktree);
6211 	if (pack_fds) {
6212 		const struct got_error *pack_err =
6213 		    got_repo_pack_fds_close(pack_fds);
6214 		if (error == NULL)
6215 			error = pack_err;
6216 	}
6217 	tog_free_refs();
6218 	return error;
6219 }
6220 
6221 __dead static void
6222 usage_blame(void)
6223 {
6224 	endwin();
6225 	fprintf(stderr,
6226 	    "usage: %s blame [-c commit] [-r repository-path] path\n",
6227 	    getprogname());
6228 	exit(1);
6229 }
6230 
6231 struct tog_blame_line {
6232 	int annotated;
6233 	struct got_object_id *id;
6234 };
6235 
6236 static const struct got_error *
6237 draw_blame(struct tog_view *view)
6238 {
6239 	struct tog_blame_view_state *s = &view->state.blame;
6240 	struct tog_blame *blame = &s->blame;
6241 	regmatch_t *regmatch = &view->regmatch;
6242 	const struct got_error *err;
6243 	int lineno = 0, nprinted = 0;
6244 	char *line = NULL;
6245 	size_t linesize = 0;
6246 	ssize_t linelen;
6247 	wchar_t *wline;
6248 	int width;
6249 	struct tog_blame_line *blame_line;
6250 	struct got_object_id *prev_id = NULL;
6251 	char *id_str;
6252 	struct tog_color *tc;
6253 
6254 	err = got_object_id_str(&id_str, &s->blamed_commit->id);
6255 	if (err)
6256 		return err;
6257 
6258 	rewind(blame->f);
6259 	werase(view->window);
6260 
6261 	if (asprintf(&line, "commit %s", id_str) == -1) {
6262 		err = got_error_from_errno("asprintf");
6263 		free(id_str);
6264 		return err;
6265 	}
6266 
6267 	err = format_line(&wline, &width, NULL, line, 0, view->ncols, 0, 0);
6268 	free(line);
6269 	line = NULL;
6270 	if (err)
6271 		return err;
6272 	if (view_needs_focus_indication(view))
6273 		wstandout(view->window);
6274 	tc = get_color(&s->colors, TOG_COLOR_COMMIT);
6275 	if (tc)
6276 		wattr_on(view->window, COLOR_PAIR(tc->colorpair), NULL);
6277 	waddwstr(view->window, wline);
6278 	while (width++ < view->ncols)
6279 		waddch(view->window, ' ');
6280 	if (tc)
6281 		wattr_off(view->window, COLOR_PAIR(tc->colorpair), NULL);
6282 	if (view_needs_focus_indication(view))
6283 		wstandend(view->window);
6284 	free(wline);
6285 	wline = NULL;
6286 
6287 	if (view->gline > blame->nlines)
6288 		view->gline = blame->nlines;
6289 
6290 	if (tog_io.wait_for_ui) {
6291 		struct tog_blame_thread_args *bta = &s->blame.thread_args;
6292 		int rc;
6293 
6294 		rc = pthread_cond_wait(&bta->blame_complete, &tog_mutex);
6295 		if (rc)
6296 			return got_error_set_errno(rc, "pthread_cond_wait");
6297 		tog_io.wait_for_ui = 0;
6298 	}
6299 
6300 	if (asprintf(&line, "[%d/%d] %s%s", view->gline ? view->gline :
6301 	    s->first_displayed_line - 1 + s->selected_line, blame->nlines,
6302 	    s->blame_complete ? "" : "annotating... ", s->path) == -1) {
6303 		free(id_str);
6304 		return got_error_from_errno("asprintf");
6305 	}
6306 	free(id_str);
6307 	err = format_line(&wline, &width, NULL, line, 0, view->ncols, 0, 0);
6308 	free(line);
6309 	line = NULL;
6310 	if (err)
6311 		return err;
6312 	waddwstr(view->window, wline);
6313 	free(wline);
6314 	wline = NULL;
6315 	if (width < view->ncols - 1)
6316 		waddch(view->window, '\n');
6317 
6318 	s->eof = 0;
6319 	view->maxx = 0;
6320 	while (nprinted < view->nlines - 2) {
6321 		linelen = getline(&line, &linesize, blame->f);
6322 		if (linelen == -1) {
6323 			if (feof(blame->f)) {
6324 				s->eof = 1;
6325 				break;
6326 			}
6327 			free(line);
6328 			return got_ferror(blame->f, GOT_ERR_IO);
6329 		}
6330 		if (++lineno < s->first_displayed_line)
6331 			continue;
6332 		if (view->gline && !gotoline(view, &lineno, &nprinted))
6333 			continue;
6334 
6335 		/* Set view->maxx based on full line length. */
6336 		err = format_line(&wline, &width, NULL, line, 0, INT_MAX, 9, 1);
6337 		if (err) {
6338 			free(line);
6339 			return err;
6340 		}
6341 		free(wline);
6342 		wline = NULL;
6343 		view->maxx = MAX(view->maxx, width);
6344 
6345 		if (nprinted == s->selected_line - 1)
6346 			wstandout(view->window);
6347 
6348 		if (blame->nlines > 0) {
6349 			blame_line = &blame->lines[lineno - 1];
6350 			if (blame_line->annotated && prev_id &&
6351 			    got_object_id_cmp(prev_id, blame_line->id) == 0 &&
6352 			    !(nprinted == s->selected_line - 1)) {
6353 				waddstr(view->window, "        ");
6354 			} else if (blame_line->annotated) {
6355 				char *id_str;
6356 				err = got_object_id_str(&id_str,
6357 				    blame_line->id);
6358 				if (err) {
6359 					free(line);
6360 					return err;
6361 				}
6362 				tc = get_color(&s->colors, TOG_COLOR_COMMIT);
6363 				if (tc)
6364 					wattr_on(view->window,
6365 					    COLOR_PAIR(tc->colorpair), NULL);
6366 				wprintw(view->window, "%.8s", id_str);
6367 				if (tc)
6368 					wattr_off(view->window,
6369 					    COLOR_PAIR(tc->colorpair), NULL);
6370 				free(id_str);
6371 				prev_id = blame_line->id;
6372 			} else {
6373 				waddstr(view->window, "........");
6374 				prev_id = NULL;
6375 			}
6376 		} else {
6377 			waddstr(view->window, "........");
6378 			prev_id = NULL;
6379 		}
6380 
6381 		if (nprinted == s->selected_line - 1)
6382 			wstandend(view->window);
6383 		waddstr(view->window, " ");
6384 
6385 		if (view->ncols <= 9) {
6386 			width = 9;
6387 		} else if (s->first_displayed_line + nprinted ==
6388 		    s->matched_line &&
6389 		    regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
6390 			err = add_matched_line(&width, line, view->ncols - 9, 9,
6391 			    view->window, view->x, regmatch);
6392 			if (err) {
6393 				free(line);
6394 				return err;
6395 			}
6396 			width += 9;
6397 		} else {
6398 			int skip;
6399 			err = format_line(&wline, &width, &skip, line,
6400 			    view->x, view->ncols - 9, 9, 1);
6401 			if (err) {
6402 				free(line);
6403 				return err;
6404 			}
6405 			waddwstr(view->window, &wline[skip]);
6406 			width += 9;
6407 			free(wline);
6408 			wline = NULL;
6409 		}
6410 
6411 		if (width <= view->ncols - 1)
6412 			waddch(view->window, '\n');
6413 		if (++nprinted == 1)
6414 			s->first_displayed_line = lineno;
6415 	}
6416 	free(line);
6417 	s->last_displayed_line = lineno;
6418 
6419 	view_border(view);
6420 
6421 	return NULL;
6422 }
6423 
6424 static const struct got_error *
6425 blame_cb(void *arg, int nlines, int lineno,
6426     struct got_commit_object *commit, struct got_object_id *id)
6427 {
6428 	const struct got_error *err = NULL;
6429 	struct tog_blame_cb_args *a = arg;
6430 	struct tog_blame_line *line;
6431 	int errcode;
6432 
6433 	if (nlines != a->nlines ||
6434 	    (lineno != -1 && lineno < 1) || lineno > a->nlines)
6435 		return got_error(GOT_ERR_RANGE);
6436 
6437 	errcode = pthread_mutex_lock(&tog_mutex);
6438 	if (errcode)
6439 		return got_error_set_errno(errcode, "pthread_mutex_lock");
6440 
6441 	if (*a->quit) {	/* user has quit the blame view */
6442 		err = got_error(GOT_ERR_ITER_COMPLETED);
6443 		goto done;
6444 	}
6445 
6446 	if (lineno == -1)
6447 		goto done; /* no change in this commit */
6448 
6449 	line = &a->lines[lineno - 1];
6450 	if (line->annotated)
6451 		goto done;
6452 
6453 	line->id = got_object_id_dup(id);
6454 	if (line->id == NULL) {
6455 		err = got_error_from_errno("got_object_id_dup");
6456 		goto done;
6457 	}
6458 	line->annotated = 1;
6459 done:
6460 	errcode = pthread_mutex_unlock(&tog_mutex);
6461 	if (errcode)
6462 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
6463 	return err;
6464 }
6465 
6466 static void *
6467 blame_thread(void *arg)
6468 {
6469 	const struct got_error *err, *close_err;
6470 	struct tog_blame_thread_args *ta = arg;
6471 	struct tog_blame_cb_args *a = ta->cb_args;
6472 	int errcode, fd1 = -1, fd2 = -1;
6473 	FILE *f1 = NULL, *f2 = NULL;
6474 
6475 	fd1 = got_opentempfd();
6476 	if (fd1 == -1)
6477 		return (void *)got_error_from_errno("got_opentempfd");
6478 
6479 	fd2 = got_opentempfd();
6480 	if (fd2 == -1) {
6481 		err = got_error_from_errno("got_opentempfd");
6482 		goto done;
6483 	}
6484 
6485 	f1 = got_opentemp();
6486 	if (f1 == NULL) {
6487 		err = (void *)got_error_from_errno("got_opentemp");
6488 		goto done;
6489 	}
6490 	f2 = got_opentemp();
6491 	if (f2 == NULL) {
6492 		err = (void *)got_error_from_errno("got_opentemp");
6493 		goto done;
6494 	}
6495 
6496 	err = block_signals_used_by_main_thread();
6497 	if (err)
6498 		goto done;
6499 
6500 	err = got_blame(ta->path, a->commit_id, ta->repo,
6501 	    tog_diff_algo, blame_cb, ta->cb_args,
6502 	    ta->cancel_cb, ta->cancel_arg, fd1, fd2, f1, f2);
6503 	if (err && err->code == GOT_ERR_CANCELLED)
6504 		err = NULL;
6505 
6506 	errcode = pthread_mutex_lock(&tog_mutex);
6507 	if (errcode) {
6508 		err = got_error_set_errno(errcode, "pthread_mutex_lock");
6509 		goto done;
6510 	}
6511 
6512 	close_err = got_repo_close(ta->repo);
6513 	if (err == NULL)
6514 		err = close_err;
6515 	ta->repo = NULL;
6516 	*ta->complete = 1;
6517 
6518 	if (tog_io.wait_for_ui) {
6519 		errcode = pthread_cond_signal(&ta->blame_complete);
6520 		if (errcode && err == NULL)
6521 			err = got_error_set_errno(errcode,
6522 			    "pthread_cond_signal");
6523 	}
6524 
6525 	errcode = pthread_mutex_unlock(&tog_mutex);
6526 	if (errcode && err == NULL)
6527 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
6528 
6529 done:
6530 	if (fd1 != -1 && close(fd1) == -1 && err == NULL)
6531 		err = got_error_from_errno("close");
6532 	if (fd2 != -1 && close(fd2) == -1 && err == NULL)
6533 		err = got_error_from_errno("close");
6534 	if (f1 && fclose(f1) == EOF && err == NULL)
6535 		err = got_error_from_errno("fclose");
6536 	if (f2 && fclose(f2) == EOF && err == NULL)
6537 		err = got_error_from_errno("fclose");
6538 
6539 	return (void *)err;
6540 }
6541 
6542 static struct got_object_id *
6543 get_selected_commit_id(struct tog_blame_line *lines, int nlines,
6544     int first_displayed_line, int selected_line)
6545 {
6546 	struct tog_blame_line *line;
6547 
6548 	if (nlines <= 0)
6549 		return NULL;
6550 
6551 	line = &lines[first_displayed_line - 1 + selected_line - 1];
6552 	if (!line->annotated)
6553 		return NULL;
6554 
6555 	return line->id;
6556 }
6557 
6558 static struct got_object_id *
6559 get_annotation_for_line(struct tog_blame_line *lines, int nlines,
6560     int lineno)
6561 {
6562 	struct tog_blame_line *line;
6563 
6564 	if (nlines <= 0 || lineno >= nlines)
6565 		return NULL;
6566 
6567 	line = &lines[lineno - 1];
6568 	if (!line->annotated)
6569 		return NULL;
6570 
6571 	return line->id;
6572 }
6573 
6574 static const struct got_error *
6575 stop_blame(struct tog_blame *blame)
6576 {
6577 	const struct got_error *err = NULL;
6578 	int i;
6579 
6580 	if (blame->thread) {
6581 		int errcode;
6582 		errcode = pthread_mutex_unlock(&tog_mutex);
6583 		if (errcode)
6584 			return got_error_set_errno(errcode,
6585 			    "pthread_mutex_unlock");
6586 		errcode = pthread_join(blame->thread, (void **)&err);
6587 		if (errcode)
6588 			return got_error_set_errno(errcode, "pthread_join");
6589 		errcode = pthread_mutex_lock(&tog_mutex);
6590 		if (errcode)
6591 			return got_error_set_errno(errcode,
6592 			    "pthread_mutex_lock");
6593 		if (err && err->code == GOT_ERR_ITER_COMPLETED)
6594 			err = NULL;
6595 		blame->thread = NULL;
6596 	}
6597 	if (blame->thread_args.repo) {
6598 		const struct got_error *close_err;
6599 		close_err = got_repo_close(blame->thread_args.repo);
6600 		if (err == NULL)
6601 			err = close_err;
6602 		blame->thread_args.repo = NULL;
6603 	}
6604 	if (blame->f) {
6605 		if (fclose(blame->f) == EOF && err == NULL)
6606 			err = got_error_from_errno("fclose");
6607 		blame->f = NULL;
6608 	}
6609 	if (blame->lines) {
6610 		for (i = 0; i < blame->nlines; i++)
6611 			free(blame->lines[i].id);
6612 		free(blame->lines);
6613 		blame->lines = NULL;
6614 	}
6615 	free(blame->cb_args.commit_id);
6616 	blame->cb_args.commit_id = NULL;
6617 	if (blame->pack_fds) {
6618 		const struct got_error *pack_err =
6619 		    got_repo_pack_fds_close(blame->pack_fds);
6620 		if (err == NULL)
6621 			err = pack_err;
6622 		blame->pack_fds = NULL;
6623 	}
6624 	free(blame->line_offsets);
6625 	blame->line_offsets = NULL;
6626 	return err;
6627 }
6628 
6629 static const struct got_error *
6630 cancel_blame_view(void *arg)
6631 {
6632 	const struct got_error *err = NULL;
6633 	int *done = arg;
6634 	int errcode;
6635 
6636 	errcode = pthread_mutex_lock(&tog_mutex);
6637 	if (errcode)
6638 		return got_error_set_errno(errcode,
6639 		    "pthread_mutex_unlock");
6640 
6641 	if (*done)
6642 		err = got_error(GOT_ERR_CANCELLED);
6643 
6644 	errcode = pthread_mutex_unlock(&tog_mutex);
6645 	if (errcode)
6646 		return got_error_set_errno(errcode,
6647 		    "pthread_mutex_lock");
6648 
6649 	return err;
6650 }
6651 
6652 static const struct got_error *
6653 run_blame(struct tog_view *view)
6654 {
6655 	struct tog_blame_view_state *s = &view->state.blame;
6656 	struct tog_blame *blame = &s->blame;
6657 	const struct got_error *err = NULL;
6658 	struct got_commit_object *commit = NULL;
6659 	struct got_blob_object *blob = NULL;
6660 	struct got_repository *thread_repo = NULL;
6661 	struct got_object_id *obj_id = NULL;
6662 	int obj_type, fd = -1;
6663 	int *pack_fds = NULL;
6664 
6665 	err = got_object_open_as_commit(&commit, s->repo,
6666 	    &s->blamed_commit->id);
6667 	if (err)
6668 		return err;
6669 
6670 	fd = got_opentempfd();
6671 	if (fd == -1) {
6672 		err = got_error_from_errno("got_opentempfd");
6673 		goto done;
6674 	}
6675 
6676 	err = got_object_id_by_path(&obj_id, s->repo, commit, s->path);
6677 	if (err)
6678 		goto done;
6679 
6680 	err = got_object_get_type(&obj_type, s->repo, obj_id);
6681 	if (err)
6682 		goto done;
6683 
6684 	if (obj_type != GOT_OBJ_TYPE_BLOB) {
6685 		err = got_error(GOT_ERR_OBJ_TYPE);
6686 		goto done;
6687 	}
6688 
6689 	err = got_object_open_as_blob(&blob, s->repo, obj_id, 8192, fd);
6690 	if (err)
6691 		goto done;
6692 	blame->f = got_opentemp();
6693 	if (blame->f == NULL) {
6694 		err = got_error_from_errno("got_opentemp");
6695 		goto done;
6696 	}
6697 	err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
6698 	    &blame->line_offsets, blame->f, blob);
6699 	if (err)
6700 		goto done;
6701 	if (blame->nlines == 0) {
6702 		s->blame_complete = 1;
6703 		goto done;
6704 	}
6705 
6706 	/* Don't include \n at EOF in the blame line count. */
6707 	if (blame->line_offsets[blame->nlines - 1] == blame->filesize)
6708 		blame->nlines--;
6709 
6710 	blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
6711 	if (blame->lines == NULL) {
6712 		err = got_error_from_errno("calloc");
6713 		goto done;
6714 	}
6715 
6716 	err = got_repo_pack_fds_open(&pack_fds);
6717 	if (err)
6718 		goto done;
6719 	err = got_repo_open(&thread_repo, got_repo_get_path(s->repo), NULL,
6720 	    pack_fds);
6721 	if (err)
6722 		goto done;
6723 
6724 	blame->pack_fds = pack_fds;
6725 	blame->cb_args.view = view;
6726 	blame->cb_args.lines = blame->lines;
6727 	blame->cb_args.nlines = blame->nlines;
6728 	blame->cb_args.commit_id = got_object_id_dup(&s->blamed_commit->id);
6729 	if (blame->cb_args.commit_id == NULL) {
6730 		err = got_error_from_errno("got_object_id_dup");
6731 		goto done;
6732 	}
6733 	blame->cb_args.quit = &s->done;
6734 
6735 	blame->thread_args.path = s->path;
6736 	blame->thread_args.repo = thread_repo;
6737 	blame->thread_args.cb_args = &blame->cb_args;
6738 	blame->thread_args.complete = &s->blame_complete;
6739 	blame->thread_args.cancel_cb = cancel_blame_view;
6740 	blame->thread_args.cancel_arg = &s->done;
6741 	s->blame_complete = 0;
6742 
6743 	if (s->first_displayed_line + view->nlines - 1 > blame->nlines) {
6744 		s->first_displayed_line = 1;
6745 		s->last_displayed_line = view->nlines;
6746 		s->selected_line = 1;
6747 	}
6748 	s->matched_line = 0;
6749 
6750 done:
6751 	if (commit)
6752 		got_object_commit_close(commit);
6753 	if (fd != -1 && close(fd) == -1 && err == NULL)
6754 		err = got_error_from_errno("close");
6755 	if (blob)
6756 		got_object_blob_close(blob);
6757 	free(obj_id);
6758 	if (err)
6759 		stop_blame(blame);
6760 	return err;
6761 }
6762 
6763 static const struct got_error *
6764 open_blame_view(struct tog_view *view, char *path,
6765     struct got_object_id *commit_id, struct got_repository *repo)
6766 {
6767 	const struct got_error *err = NULL;
6768 	struct tog_blame_view_state *s = &view->state.blame;
6769 
6770 	STAILQ_INIT(&s->blamed_commits);
6771 
6772 	s->path = strdup(path);
6773 	if (s->path == NULL)
6774 		return got_error_from_errno("strdup");
6775 
6776 	err = got_object_qid_alloc(&s->blamed_commit, commit_id);
6777 	if (err) {
6778 		free(s->path);
6779 		return err;
6780 	}
6781 
6782 	STAILQ_INSERT_HEAD(&s->blamed_commits, s->blamed_commit, entry);
6783 	s->first_displayed_line = 1;
6784 	s->last_displayed_line = view->nlines;
6785 	s->selected_line = 1;
6786 	s->blame_complete = 0;
6787 	s->repo = repo;
6788 	s->commit_id = commit_id;
6789 	memset(&s->blame, 0, sizeof(s->blame));
6790 
6791 	STAILQ_INIT(&s->colors);
6792 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
6793 		err = add_color(&s->colors, "^", TOG_COLOR_COMMIT,
6794 		    get_color_value("TOG_COLOR_COMMIT"));
6795 		if (err)
6796 			return err;
6797 	}
6798 
6799 	view->show = show_blame_view;
6800 	view->input = input_blame_view;
6801 	view->reset = reset_blame_view;
6802 	view->close = close_blame_view;
6803 	view->search_start = search_start_blame_view;
6804 	view->search_setup = search_setup_blame_view;
6805 	view->search_next = search_next_view_match;
6806 
6807 	if (using_mock_io) {
6808 		struct tog_blame_thread_args *bta = &s->blame.thread_args;
6809 		int rc;
6810 
6811 		rc = pthread_cond_init(&bta->blame_complete, NULL);
6812 		if (rc)
6813 			return got_error_set_errno(rc, "pthread_cond_init");
6814 	}
6815 
6816 	return run_blame(view);
6817 }
6818 
6819 static const struct got_error *
6820 close_blame_view(struct tog_view *view)
6821 {
6822 	const struct got_error *err = NULL;
6823 	struct tog_blame_view_state *s = &view->state.blame;
6824 
6825 	if (s->blame.thread)
6826 		err = stop_blame(&s->blame);
6827 
6828 	while (!STAILQ_EMPTY(&s->blamed_commits)) {
6829 		struct got_object_qid *blamed_commit;
6830 		blamed_commit = STAILQ_FIRST(&s->blamed_commits);
6831 		STAILQ_REMOVE_HEAD(&s->blamed_commits, entry);
6832 		got_object_qid_free(blamed_commit);
6833 	}
6834 
6835 	if (using_mock_io) {
6836 		struct tog_blame_thread_args *bta = &s->blame.thread_args;
6837 		int rc;
6838 
6839 		rc = pthread_cond_destroy(&bta->blame_complete);
6840 		if (rc && err == NULL)
6841 			err = got_error_set_errno(rc, "pthread_cond_destroy");
6842 	}
6843 
6844 	free(s->path);
6845 	free_colors(&s->colors);
6846 	return err;
6847 }
6848 
6849 static const struct got_error *
6850 search_start_blame_view(struct tog_view *view)
6851 {
6852 	struct tog_blame_view_state *s = &view->state.blame;
6853 
6854 	s->matched_line = 0;
6855 	return NULL;
6856 }
6857 
6858 static void
6859 search_setup_blame_view(struct tog_view *view, FILE **f, off_t **line_offsets,
6860     size_t *nlines, int **first, int **last, int **match, int **selected)
6861 {
6862 	struct tog_blame_view_state *s = &view->state.blame;
6863 
6864 	*f = s->blame.f;
6865 	*nlines = s->blame.nlines;
6866 	*line_offsets = s->blame.line_offsets;
6867 	*match = &s->matched_line;
6868 	*first = &s->first_displayed_line;
6869 	*last = &s->last_displayed_line;
6870 	*selected = &s->selected_line;
6871 }
6872 
6873 static const struct got_error *
6874 show_blame_view(struct tog_view *view)
6875 {
6876 	const struct got_error *err = NULL;
6877 	struct tog_blame_view_state *s = &view->state.blame;
6878 	int errcode;
6879 
6880 	if (s->blame.thread == NULL && !s->blame_complete) {
6881 		errcode = pthread_create(&s->blame.thread, NULL, blame_thread,
6882 		    &s->blame.thread_args);
6883 		if (errcode)
6884 			return got_error_set_errno(errcode, "pthread_create");
6885 
6886 		if (!using_mock_io)
6887 			halfdelay(1); /* fast refresh while annotating  */
6888 	}
6889 
6890 	if (s->blame_complete && !using_mock_io)
6891 		halfdelay(10); /* disable fast refresh */
6892 
6893 	err = draw_blame(view);
6894 
6895 	view_border(view);
6896 	return err;
6897 }
6898 
6899 static const struct got_error *
6900 log_annotated_line(struct tog_view **new_view, int begin_y, int begin_x,
6901     struct got_repository *repo, struct got_object_id *id)
6902 {
6903 	struct tog_view		*log_view;
6904 	const struct got_error	*err = NULL;
6905 
6906 	*new_view = NULL;
6907 
6908 	log_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_LOG);
6909 	if (log_view == NULL)
6910 		return got_error_from_errno("view_open");
6911 
6912 	err = open_log_view(log_view, id, repo, GOT_REF_HEAD, "", 0, NULL);
6913 	if (err)
6914 		view_close(log_view);
6915 	else
6916 		*new_view = log_view;
6917 
6918 	return err;
6919 }
6920 
6921 static const struct got_error *
6922 input_blame_view(struct tog_view **new_view, struct tog_view *view, int ch)
6923 {
6924 	const struct got_error *err = NULL, *thread_err = NULL;
6925 	struct tog_view *diff_view;
6926 	struct tog_blame_view_state *s = &view->state.blame;
6927 	int eos, nscroll, begin_y = 0, begin_x = 0;
6928 
6929 	eos = nscroll = view->nlines - 2;
6930 	if (view_is_hsplit_top(view))
6931 		--eos;  /* border */
6932 
6933 	switch (ch) {
6934 	case '0':
6935 	case '$':
6936 	case KEY_RIGHT:
6937 	case 'l':
6938 	case KEY_LEFT:
6939 	case 'h':
6940 		horizontal_scroll_input(view, ch);
6941 		break;
6942 	case 'q':
6943 		s->done = 1;
6944 		break;
6945 	case 'g':
6946 	case KEY_HOME:
6947 		s->selected_line = 1;
6948 		s->first_displayed_line = 1;
6949 		view->count = 0;
6950 		break;
6951 	case 'G':
6952 	case KEY_END:
6953 		if (s->blame.nlines < eos) {
6954 			s->selected_line = s->blame.nlines;
6955 			s->first_displayed_line = 1;
6956 		} else {
6957 			s->selected_line = eos;
6958 			s->first_displayed_line = s->blame.nlines - (eos - 1);
6959 		}
6960 		view->count = 0;
6961 		break;
6962 	case 'k':
6963 	case KEY_UP:
6964 	case CTRL('p'):
6965 		if (s->selected_line > 1)
6966 			s->selected_line--;
6967 		else if (s->selected_line == 1 &&
6968 		    s->first_displayed_line > 1)
6969 			s->first_displayed_line--;
6970 		else
6971 			view->count = 0;
6972 		break;
6973 	case CTRL('u'):
6974 	case 'u':
6975 		nscroll /= 2;
6976 		/* FALL THROUGH */
6977 	case KEY_PPAGE:
6978 	case CTRL('b'):
6979 	case 'b':
6980 		if (s->first_displayed_line == 1) {
6981 			if (view->count > 1)
6982 				nscroll += nscroll;
6983 			s->selected_line = MAX(1, s->selected_line - nscroll);
6984 			view->count = 0;
6985 			break;
6986 		}
6987 		if (s->first_displayed_line > nscroll)
6988 			s->first_displayed_line -= nscroll;
6989 		else
6990 			s->first_displayed_line = 1;
6991 		break;
6992 	case 'j':
6993 	case KEY_DOWN:
6994 	case CTRL('n'):
6995 		if (s->selected_line < eos && s->first_displayed_line +
6996 		    s->selected_line <= s->blame.nlines)
6997 			s->selected_line++;
6998 		else if (s->first_displayed_line < s->blame.nlines - (eos - 1))
6999 			s->first_displayed_line++;
7000 		else
7001 			view->count = 0;
7002 		break;
7003 	case 'c':
7004 	case 'p': {
7005 		struct got_object_id *id = NULL;
7006 
7007 		view->count = 0;
7008 		id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
7009 		    s->first_displayed_line, s->selected_line);
7010 		if (id == NULL)
7011 			break;
7012 		if (ch == 'p') {
7013 			struct got_commit_object *commit, *pcommit;
7014 			struct got_object_qid *pid;
7015 			struct got_object_id *blob_id = NULL;
7016 			int obj_type;
7017 			err = got_object_open_as_commit(&commit,
7018 			    s->repo, id);
7019 			if (err)
7020 				break;
7021 			pid = STAILQ_FIRST(
7022 			    got_object_commit_get_parent_ids(commit));
7023 			if (pid == NULL) {
7024 				got_object_commit_close(commit);
7025 				break;
7026 			}
7027 			/* Check if path history ends here. */
7028 			err = got_object_open_as_commit(&pcommit,
7029 			    s->repo, &pid->id);
7030 			if (err)
7031 				break;
7032 			err = got_object_id_by_path(&blob_id, s->repo,
7033 			    pcommit, s->path);
7034 			got_object_commit_close(pcommit);
7035 			if (err) {
7036 				if (err->code == GOT_ERR_NO_TREE_ENTRY)
7037 					err = NULL;
7038 				got_object_commit_close(commit);
7039 				break;
7040 			}
7041 			err = got_object_get_type(&obj_type, s->repo,
7042 			    blob_id);
7043 			free(blob_id);
7044 			/* Can't blame non-blob type objects. */
7045 			if (obj_type != GOT_OBJ_TYPE_BLOB) {
7046 				got_object_commit_close(commit);
7047 				break;
7048 			}
7049 			err = got_object_qid_alloc(&s->blamed_commit,
7050 			    &pid->id);
7051 			got_object_commit_close(commit);
7052 		} else {
7053 			if (got_object_id_cmp(id,
7054 			    &s->blamed_commit->id) == 0)
7055 				break;
7056 			err = got_object_qid_alloc(&s->blamed_commit,
7057 			    id);
7058 		}
7059 		if (err)
7060 			break;
7061 		s->done = 1;
7062 		thread_err = stop_blame(&s->blame);
7063 		s->done = 0;
7064 		if (thread_err)
7065 			break;
7066 		STAILQ_INSERT_HEAD(&s->blamed_commits,
7067 		    s->blamed_commit, entry);
7068 		err = run_blame(view);
7069 		if (err)
7070 			break;
7071 		break;
7072 	}
7073 	case 'C': {
7074 		struct got_object_qid *first;
7075 
7076 		view->count = 0;
7077 		first = STAILQ_FIRST(&s->blamed_commits);
7078 		if (!got_object_id_cmp(&first->id, s->commit_id))
7079 			break;
7080 		s->done = 1;
7081 		thread_err = stop_blame(&s->blame);
7082 		s->done = 0;
7083 		if (thread_err)
7084 			break;
7085 		STAILQ_REMOVE_HEAD(&s->blamed_commits, entry);
7086 		got_object_qid_free(s->blamed_commit);
7087 		s->blamed_commit =
7088 		    STAILQ_FIRST(&s->blamed_commits);
7089 		err = run_blame(view);
7090 		if (err)
7091 			break;
7092 		break;
7093 	}
7094 	case 'L':
7095 		view->count = 0;
7096 		s->id_to_log = get_selected_commit_id(s->blame.lines,
7097 		    s->blame.nlines, s->first_displayed_line, s->selected_line);
7098 		if (s->id_to_log)
7099 			err = view_request_new(new_view, view, TOG_VIEW_LOG);
7100 		break;
7101 	case KEY_ENTER:
7102 	case '\r': {
7103 		struct got_object_id *id = NULL;
7104 		struct got_object_qid *pid;
7105 		struct got_commit_object *commit = NULL;
7106 
7107 		view->count = 0;
7108 		id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
7109 		    s->first_displayed_line, s->selected_line);
7110 		if (id == NULL)
7111 			break;
7112 		err = got_object_open_as_commit(&commit, s->repo, id);
7113 		if (err)
7114 			break;
7115 		pid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
7116 		if (*new_view) {
7117 			/* traversed from diff view, release diff resources  */
7118 			err = close_diff_view(*new_view);
7119 			if (err)
7120 				break;
7121 			diff_view = *new_view;
7122 		} else {
7123 			if (view_is_parent_view(view))
7124 				view_get_split(view, &begin_y, &begin_x);
7125 
7126 			diff_view = view_open(0, 0, begin_y, begin_x,
7127 			    TOG_VIEW_DIFF);
7128 			if (diff_view == NULL) {
7129 				got_object_commit_close(commit);
7130 				err = got_error_from_errno("view_open");
7131 				break;
7132 			}
7133 		}
7134 		err = open_diff_view(diff_view, pid ? &pid->id : NULL,
7135 		    id, NULL, NULL, 3, 0, 0, view, s->repo);
7136 		got_object_commit_close(commit);
7137 		if (err)
7138 			break;
7139 		s->last_diffed_line = s->first_displayed_line - 1 +
7140 		    s->selected_line;
7141 		if (*new_view)
7142 			break;	/* still open from active diff view */
7143 		if (view_is_parent_view(view) &&
7144 		    view->mode == TOG_VIEW_SPLIT_HRZN) {
7145 			err = view_init_hsplit(view, begin_y);
7146 			if (err)
7147 				break;
7148 		}
7149 
7150 		view->focussed = 0;
7151 		diff_view->focussed = 1;
7152 		diff_view->mode = view->mode;
7153 		diff_view->nlines = view->lines - begin_y;
7154 		if (view_is_parent_view(view)) {
7155 			view_transfer_size(diff_view, view);
7156 			err = view_close_child(view);
7157 			if (err)
7158 				break;
7159 			err = view_set_child(view, diff_view);
7160 			if (err)
7161 				break;
7162 			view->focus_child = 1;
7163 		} else
7164 			*new_view = diff_view;
7165 		if (err)
7166 			break;
7167 		break;
7168 	}
7169 	case CTRL('d'):
7170 	case 'd':
7171 		nscroll /= 2;
7172 		/* FALL THROUGH */
7173 	case KEY_NPAGE:
7174 	case CTRL('f'):
7175 	case 'f':
7176 	case ' ':
7177 		if (s->last_displayed_line >= s->blame.nlines &&
7178 		    s->selected_line >= MIN(s->blame.nlines,
7179 		    view->nlines - 2)) {
7180 			view->count = 0;
7181 			break;
7182 		}
7183 		if (s->last_displayed_line >= s->blame.nlines &&
7184 		    s->selected_line < view->nlines - 2) {
7185 			s->selected_line +=
7186 			    MIN(nscroll, s->last_displayed_line -
7187 			    s->first_displayed_line - s->selected_line + 1);
7188 		}
7189 		if (s->last_displayed_line + nscroll <= s->blame.nlines)
7190 			s->first_displayed_line += nscroll;
7191 		else
7192 			s->first_displayed_line =
7193 			    s->blame.nlines - (view->nlines - 3);
7194 		break;
7195 	case KEY_RESIZE:
7196 		if (s->selected_line > view->nlines - 2) {
7197 			s->selected_line = MIN(s->blame.nlines,
7198 			    view->nlines - 2);
7199 		}
7200 		break;
7201 	default:
7202 		view->count = 0;
7203 		break;
7204 	}
7205 	return thread_err ? thread_err : err;
7206 }
7207 
7208 static const struct got_error *
7209 reset_blame_view(struct tog_view *view)
7210 {
7211 	const struct got_error *err;
7212 	struct tog_blame_view_state *s = &view->state.blame;
7213 
7214 	view->count = 0;
7215 	s->done = 1;
7216 	err = stop_blame(&s->blame);
7217 	s->done = 0;
7218 	if (err)
7219 		return err;
7220 	return run_blame(view);
7221 }
7222 
7223 static const struct got_error *
7224 cmd_blame(int argc, char *argv[])
7225 {
7226 	const struct got_error *error;
7227 	struct got_repository *repo = NULL;
7228 	struct got_worktree *worktree = NULL;
7229 	char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
7230 	char *link_target = NULL;
7231 	struct got_object_id *commit_id = NULL;
7232 	struct got_commit_object *commit = NULL;
7233 	char *keyword_idstr = NULL, *commit_id_str = NULL;
7234 	int ch;
7235 	struct tog_view *view = NULL;
7236 	int *pack_fds = NULL;
7237 
7238 	while ((ch = getopt(argc, argv, "c:r:")) != -1) {
7239 		switch (ch) {
7240 		case 'c':
7241 			commit_id_str = optarg;
7242 			break;
7243 		case 'r':
7244 			repo_path = realpath(optarg, NULL);
7245 			if (repo_path == NULL)
7246 				return got_error_from_errno2("realpath",
7247 				    optarg);
7248 			break;
7249 		default:
7250 			usage_blame();
7251 			/* NOTREACHED */
7252 		}
7253 	}
7254 
7255 	argc -= optind;
7256 	argv += optind;
7257 
7258 	if (argc != 1)
7259 		usage_blame();
7260 
7261 	error = got_repo_pack_fds_open(&pack_fds);
7262 	if (error != NULL)
7263 		goto done;
7264 
7265 	if (repo_path == NULL) {
7266 		cwd = getcwd(NULL, 0);
7267 		if (cwd == NULL)
7268 			return got_error_from_errno("getcwd");
7269 		error = got_worktree_open(&worktree, cwd, NULL);
7270 		if (error && error->code != GOT_ERR_NOT_WORKTREE)
7271 			goto done;
7272 		if (worktree)
7273 			repo_path =
7274 			    strdup(got_worktree_get_repo_path(worktree));
7275 		else
7276 			repo_path = strdup(cwd);
7277 		if (repo_path == NULL) {
7278 			error = got_error_from_errno("strdup");
7279 			goto done;
7280 		}
7281 	}
7282 
7283 	error = got_repo_open(&repo, repo_path, NULL, pack_fds);
7284 	if (error != NULL)
7285 		goto done;
7286 
7287 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv, repo,
7288 	    worktree);
7289 	if (error)
7290 		goto done;
7291 
7292 	init_curses();
7293 
7294 	error = apply_unveil(got_repo_get_path(repo), NULL);
7295 	if (error)
7296 		goto done;
7297 
7298 	error = tog_load_refs(repo, 0);
7299 	if (error)
7300 		goto done;
7301 
7302 	if (commit_id_str == NULL) {
7303 		struct got_reference *head_ref;
7304 		error = got_ref_open(&head_ref, repo, worktree ?
7305 		    got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD, 0);
7306 		if (error != NULL)
7307 			goto done;
7308 		error = got_ref_resolve(&commit_id, repo, head_ref);
7309 		got_ref_close(head_ref);
7310 	} else {
7311 		error = got_keyword_to_idstr(&keyword_idstr, commit_id_str,
7312 		    repo, worktree);
7313 		if (error != NULL)
7314 			goto done;
7315 		if (keyword_idstr != NULL)
7316 			commit_id_str = keyword_idstr;
7317 
7318 		error = got_repo_match_object_id(&commit_id, NULL,
7319 		    commit_id_str, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
7320 	}
7321 	if (error != NULL)
7322 		goto done;
7323 
7324 	error = got_object_open_as_commit(&commit, repo, commit_id);
7325 	if (error)
7326 		goto done;
7327 
7328 	error = got_object_resolve_symlinks(&link_target, in_repo_path,
7329 	    commit, repo);
7330 	if (error)
7331 		goto done;
7332 
7333 	view = view_open(0, 0, 0, 0, TOG_VIEW_BLAME);
7334 	if (view == NULL) {
7335 		error = got_error_from_errno("view_open");
7336 		goto done;
7337 	}
7338 	error = open_blame_view(view, link_target ? link_target : in_repo_path,
7339 	    commit_id, repo);
7340 	if (error != NULL) {
7341 		if (view->close == NULL)
7342 			close_blame_view(view);
7343 		view_close(view);
7344 		goto done;
7345 	}
7346 
7347 	if (worktree) {
7348 		error = set_tog_base_commit(repo, worktree);
7349 		if (error != NULL)
7350 			goto done;
7351 
7352 		/* Release work tree lock. */
7353 		got_worktree_close(worktree);
7354 		worktree = NULL;
7355 	}
7356 
7357 	error = view_loop(view);
7358 
7359 done:
7360 	free(tog_base_commit.id);
7361 	free(repo_path);
7362 	free(in_repo_path);
7363 	free(link_target);
7364 	free(cwd);
7365 	free(commit_id);
7366 	free(keyword_idstr);
7367 	if (commit)
7368 		got_object_commit_close(commit);
7369 	if (worktree)
7370 		got_worktree_close(worktree);
7371 	if (repo) {
7372 		const struct got_error *close_err = got_repo_close(repo);
7373 		if (error == NULL)
7374 			error = close_err;
7375 	}
7376 	if (pack_fds) {
7377 		const struct got_error *pack_err =
7378 		    got_repo_pack_fds_close(pack_fds);
7379 		if (error == NULL)
7380 			error = pack_err;
7381 	}
7382 	tog_free_refs();
7383 	return error;
7384 }
7385 
7386 static const struct got_error *
7387 draw_tree_entries(struct tog_view *view, const char *parent_path)
7388 {
7389 	struct tog_tree_view_state *s = &view->state.tree;
7390 	const struct got_error *err = NULL;
7391 	struct got_tree_entry *te;
7392 	wchar_t *wline;
7393 	char *index = NULL;
7394 	struct tog_color *tc;
7395 	int width, n, nentries, scrollx, i = 1;
7396 	int limit = view->nlines;
7397 
7398 	s->ndisplayed = 0;
7399 	if (view_is_hsplit_top(view))
7400 		--limit;  /* border */
7401 
7402 	werase(view->window);
7403 
7404 	if (limit == 0)
7405 		return NULL;
7406 
7407 	err = format_line(&wline, &width, NULL, s->tree_label, 0, view->ncols,
7408 	    0, 0);
7409 	if (err)
7410 		return err;
7411 	if (view_needs_focus_indication(view))
7412 		wstandout(view->window);
7413 	tc = get_color(&s->colors, TOG_COLOR_COMMIT);
7414 	if (tc)
7415 		wattr_on(view->window, COLOR_PAIR(tc->colorpair), NULL);
7416 	waddwstr(view->window, wline);
7417 	free(wline);
7418 	wline = NULL;
7419 	while (width++ < view->ncols)
7420 		waddch(view->window, ' ');
7421 	if (tc)
7422 		wattr_off(view->window, COLOR_PAIR(tc->colorpair), NULL);
7423 	if (view_needs_focus_indication(view))
7424 		wstandend(view->window);
7425 	if (--limit <= 0)
7426 		return NULL;
7427 
7428 	i += s->selected;
7429 	if (s->first_displayed_entry) {
7430 		i += got_tree_entry_get_index(s->first_displayed_entry);
7431 		if (s->tree != s->root)
7432 			++i;  /* account for ".." entry */
7433 	}
7434 	nentries = got_object_tree_get_nentries(s->tree);
7435 	if (asprintf(&index, "[%d/%d] %s",
7436 	    i, nentries + (s->tree == s->root ? 0 : 1), parent_path) == -1)
7437 		return got_error_from_errno("asprintf");
7438 	err = format_line(&wline, &width, NULL, index, 0, view->ncols, 0, 0);
7439 	free(index);
7440 	if (err)
7441 		return err;
7442 	waddwstr(view->window, wline);
7443 	free(wline);
7444 	wline = NULL;
7445 	if (width < view->ncols - 1)
7446 		waddch(view->window, '\n');
7447 	if (--limit <= 0)
7448 		return NULL;
7449 	waddch(view->window, '\n');
7450 	if (--limit <= 0)
7451 		return NULL;
7452 
7453 	if (s->first_displayed_entry == NULL) {
7454 		te = got_object_tree_get_first_entry(s->tree);
7455 		if (s->selected == 0) {
7456 			if (view->focussed)
7457 				wstandout(view->window);
7458 			s->selected_entry = NULL;
7459 		}
7460 		waddstr(view->window, "  ..\n");	/* parent directory */
7461 		if (s->selected == 0 && view->focussed)
7462 			wstandend(view->window);
7463 		s->ndisplayed++;
7464 		if (--limit <= 0)
7465 			return NULL;
7466 		n = 1;
7467 	} else {
7468 		n = 0;
7469 		te = s->first_displayed_entry;
7470 	}
7471 
7472 	view->maxx = 0;
7473 	for (i = got_tree_entry_get_index(te); i < nentries; i++) {
7474 		char *line = NULL, *id_str = NULL, *link_target = NULL;
7475 		const char *modestr = "";
7476 		mode_t mode;
7477 
7478 		te = got_object_tree_get_entry(s->tree, i);
7479 		mode = got_tree_entry_get_mode(te);
7480 
7481 		if (s->show_ids) {
7482 			err = got_object_id_str(&id_str,
7483 			    got_tree_entry_get_id(te));
7484 			if (err)
7485 				return got_error_from_errno(
7486 				    "got_object_id_str");
7487 		}
7488 		if (got_object_tree_entry_is_submodule(te))
7489 			modestr = "$";
7490 		else if (S_ISLNK(mode)) {
7491 			int i;
7492 
7493 			err = got_tree_entry_get_symlink_target(&link_target,
7494 			    te, s->repo);
7495 			if (err) {
7496 				free(id_str);
7497 				return err;
7498 			}
7499 			for (i = 0; link_target[i] != '\0'; i++) {
7500 				if (!isprint((unsigned char)link_target[i]))
7501 					link_target[i] = '?';
7502 			}
7503 			modestr = "@";
7504 		}
7505 		else if (S_ISDIR(mode))
7506 			modestr = "/";
7507 		else if (mode & S_IXUSR)
7508 			modestr = "*";
7509 		if (asprintf(&line, "%s  %s%s%s%s", id_str ? id_str : "",
7510 		    got_tree_entry_get_name(te), modestr,
7511 		    link_target ? " -> ": "",
7512 		    link_target ? link_target : "") == -1) {
7513 			free(id_str);
7514 			free(link_target);
7515 			return got_error_from_errno("asprintf");
7516 		}
7517 		free(id_str);
7518 		free(link_target);
7519 
7520 		/* use full line width to determine view->maxx */
7521 		err = format_line(&wline, &width, NULL, line, 0, INT_MAX, 0, 0);
7522 		if (err) {
7523 			free(line);
7524 			break;
7525 		}
7526 		view->maxx = MAX(view->maxx, width);
7527 		free(wline);
7528 		wline = NULL;
7529 
7530 		err = format_line(&wline, &width, &scrollx, line, view->x,
7531 		    view->ncols, 0, 0);
7532 		if (err) {
7533 			free(line);
7534 			break;
7535 		}
7536 		if (n == s->selected) {
7537 			if (view->focussed)
7538 				wstandout(view->window);
7539 			s->selected_entry = te;
7540 		}
7541 		tc = match_color(&s->colors, line);
7542 		if (tc)
7543 			wattr_on(view->window,
7544 			    COLOR_PAIR(tc->colorpair), NULL);
7545 		waddwstr(view->window, &wline[scrollx]);
7546 		if (tc)
7547 			wattr_off(view->window,
7548 			    COLOR_PAIR(tc->colorpair), NULL);
7549 		if (width < view->ncols)
7550 			waddch(view->window, '\n');
7551 		if (n == s->selected && view->focussed)
7552 			wstandend(view->window);
7553 		free(line);
7554 		free(wline);
7555 		wline = NULL;
7556 		n++;
7557 		s->ndisplayed++;
7558 		s->last_displayed_entry = te;
7559 		if (--limit <= 0)
7560 			break;
7561 	}
7562 
7563 	return err;
7564 }
7565 
7566 static void
7567 tree_scroll_up(struct tog_tree_view_state *s, int maxscroll)
7568 {
7569 	struct got_tree_entry *te;
7570 	int isroot = s->tree == s->root;
7571 	int i = 0;
7572 
7573 	if (s->first_displayed_entry == NULL)
7574 		return;
7575 
7576 	te = got_tree_entry_get_prev(s->tree, s->first_displayed_entry);
7577 	while (i++ < maxscroll) {
7578 		if (te == NULL) {
7579 			if (!isroot)
7580 				s->first_displayed_entry = NULL;
7581 			break;
7582 		}
7583 		s->first_displayed_entry = te;
7584 		te = got_tree_entry_get_prev(s->tree, te);
7585 	}
7586 }
7587 
7588 static const struct got_error *
7589 tree_scroll_down(struct tog_view *view, int maxscroll)
7590 {
7591 	struct tog_tree_view_state *s = &view->state.tree;
7592 	struct got_tree_entry *next, *last;
7593 	int n = 0;
7594 
7595 	if (s->first_displayed_entry)
7596 		next = got_tree_entry_get_next(s->tree,
7597 		    s->first_displayed_entry);
7598 	else
7599 		next = got_object_tree_get_first_entry(s->tree);
7600 
7601 	last = s->last_displayed_entry;
7602 	while (next && n++ < maxscroll) {
7603 		if (last) {
7604 			s->last_displayed_entry = last;
7605 			last = got_tree_entry_get_next(s->tree, last);
7606 		}
7607 		if (last || (view->mode == TOG_VIEW_SPLIT_HRZN && next)) {
7608 			s->first_displayed_entry = next;
7609 			next = got_tree_entry_get_next(s->tree, next);
7610 		}
7611 	}
7612 
7613 	return NULL;
7614 }
7615 
7616 static const struct got_error *
7617 tree_entry_path(char **path, struct tog_parent_trees *parents,
7618     struct got_tree_entry *te)
7619 {
7620 	const struct got_error *err = NULL;
7621 	struct tog_parent_tree *pt;
7622 	size_t len = 2; /* for leading slash and NUL */
7623 
7624 	TAILQ_FOREACH(pt, parents, entry)
7625 		len += strlen(got_tree_entry_get_name(pt->selected_entry))
7626 		    + 1 /* slash */;
7627 	if (te)
7628 		len += strlen(got_tree_entry_get_name(te));
7629 
7630 	*path = calloc(1, len);
7631 	if (path == NULL)
7632 		return got_error_from_errno("calloc");
7633 
7634 	(*path)[0] = '/';
7635 	pt = TAILQ_LAST(parents, tog_parent_trees);
7636 	while (pt) {
7637 		const char *name = got_tree_entry_get_name(pt->selected_entry);
7638 		if (strlcat(*path, name, len) >= len) {
7639 			err = got_error(GOT_ERR_NO_SPACE);
7640 			goto done;
7641 		}
7642 		if (strlcat(*path, "/", len) >= len) {
7643 			err = got_error(GOT_ERR_NO_SPACE);
7644 			goto done;
7645 		}
7646 		pt = TAILQ_PREV(pt, tog_parent_trees, entry);
7647 	}
7648 	if (te) {
7649 		if (strlcat(*path, got_tree_entry_get_name(te), len) >= len) {
7650 			err = got_error(GOT_ERR_NO_SPACE);
7651 			goto done;
7652 		}
7653 	}
7654 done:
7655 	if (err) {
7656 		free(*path);
7657 		*path = NULL;
7658 	}
7659 	return err;
7660 }
7661 
7662 static const struct got_error *
7663 blame_tree_entry(struct tog_view **new_view, int begin_y, int begin_x,
7664     struct got_tree_entry *te, struct tog_parent_trees *parents,
7665     struct got_object_id *commit_id, struct got_repository *repo)
7666 {
7667 	const struct got_error *err = NULL;
7668 	char *path;
7669 	struct tog_view *blame_view;
7670 
7671 	*new_view = NULL;
7672 
7673 	err = tree_entry_path(&path, parents, te);
7674 	if (err)
7675 		return err;
7676 
7677 	blame_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_BLAME);
7678 	if (blame_view == NULL) {
7679 		err = got_error_from_errno("view_open");
7680 		goto done;
7681 	}
7682 
7683 	err = open_blame_view(blame_view, path, commit_id, repo);
7684 	if (err) {
7685 		if (err->code == GOT_ERR_CANCELLED)
7686 			err = NULL;
7687 		view_close(blame_view);
7688 	} else
7689 		*new_view = blame_view;
7690 done:
7691 	free(path);
7692 	return err;
7693 }
7694 
7695 static const struct got_error *
7696 log_selected_tree_entry(struct tog_view **new_view, int begin_y, int begin_x,
7697     struct tog_tree_view_state *s)
7698 {
7699 	struct tog_view *log_view;
7700 	const struct got_error *err = NULL;
7701 	char *path;
7702 
7703 	*new_view = NULL;
7704 
7705 	log_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_LOG);
7706 	if (log_view == NULL)
7707 		return got_error_from_errno("view_open");
7708 
7709 	err = tree_entry_path(&path, &s->parents, s->selected_entry);
7710 	if (err)
7711 		return err;
7712 
7713 	err = open_log_view(log_view, s->commit_id, s->repo, s->head_ref_name,
7714 	    path, 0, NULL);
7715 	if (err)
7716 		view_close(log_view);
7717 	else
7718 		*new_view = log_view;
7719 	free(path);
7720 	return err;
7721 }
7722 
7723 static const struct got_error *
7724 open_tree_view(struct tog_view *view, struct got_object_id *commit_id,
7725     const char *head_ref_name, struct got_repository *repo)
7726 {
7727 	const struct got_error *err = NULL;
7728 	char *commit_id_str = NULL;
7729 	struct tog_tree_view_state *s = &view->state.tree;
7730 	struct got_commit_object *commit = NULL;
7731 
7732 	TAILQ_INIT(&s->parents);
7733 	STAILQ_INIT(&s->colors);
7734 
7735 	s->commit_id = got_object_id_dup(commit_id);
7736 	if (s->commit_id == NULL) {
7737 		err = got_error_from_errno("got_object_id_dup");
7738 		goto done;
7739 	}
7740 
7741 	err = got_object_open_as_commit(&commit, repo, commit_id);
7742 	if (err)
7743 		goto done;
7744 
7745 	/*
7746 	 * The root is opened here and will be closed when the view is closed.
7747 	 * Any visited subtrees and their path-wise parents are opened and
7748 	 * closed on demand.
7749 	 */
7750 	err = got_object_open_as_tree(&s->root, repo,
7751 	    got_object_commit_get_tree_id(commit));
7752 	if (err)
7753 		goto done;
7754 	s->tree = s->root;
7755 
7756 	err = got_object_id_str(&commit_id_str, commit_id);
7757 	if (err != NULL)
7758 		goto done;
7759 
7760 	if (asprintf(&s->tree_label, "commit %s", commit_id_str) == -1) {
7761 		err = got_error_from_errno("asprintf");
7762 		goto done;
7763 	}
7764 
7765 	s->first_displayed_entry = got_object_tree_get_entry(s->tree, 0);
7766 	s->selected_entry = got_object_tree_get_entry(s->tree, 0);
7767 	if (head_ref_name) {
7768 		s->head_ref_name = strdup(head_ref_name);
7769 		if (s->head_ref_name == NULL) {
7770 			err = got_error_from_errno("strdup");
7771 			goto done;
7772 		}
7773 	}
7774 	s->repo = repo;
7775 
7776 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
7777 		err = add_color(&s->colors, "\\$$",
7778 		    TOG_COLOR_TREE_SUBMODULE,
7779 		    get_color_value("TOG_COLOR_TREE_SUBMODULE"));
7780 		if (err)
7781 			goto done;
7782 		err = add_color(&s->colors, "@$", TOG_COLOR_TREE_SYMLINK,
7783 		    get_color_value("TOG_COLOR_TREE_SYMLINK"));
7784 		if (err)
7785 			goto done;
7786 		err = add_color(&s->colors, "/$",
7787 		    TOG_COLOR_TREE_DIRECTORY,
7788 		    get_color_value("TOG_COLOR_TREE_DIRECTORY"));
7789 		if (err)
7790 			goto done;
7791 
7792 		err = add_color(&s->colors, "\\*$",
7793 		    TOG_COLOR_TREE_EXECUTABLE,
7794 		    get_color_value("TOG_COLOR_TREE_EXECUTABLE"));
7795 		if (err)
7796 			goto done;
7797 
7798 		err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
7799 		    get_color_value("TOG_COLOR_COMMIT"));
7800 		if (err)
7801 			goto done;
7802 	}
7803 
7804 	view->show = show_tree_view;
7805 	view->input = input_tree_view;
7806 	view->close = close_tree_view;
7807 	view->search_start = search_start_tree_view;
7808 	view->search_next = search_next_tree_view;
7809 done:
7810 	free(commit_id_str);
7811 	if (commit)
7812 		got_object_commit_close(commit);
7813 	if (err) {
7814 		if (view->close == NULL)
7815 			close_tree_view(view);
7816 		view_close(view);
7817 	}
7818 	return err;
7819 }
7820 
7821 static const struct got_error *
7822 close_tree_view(struct tog_view *view)
7823 {
7824 	struct tog_tree_view_state *s = &view->state.tree;
7825 
7826 	free_colors(&s->colors);
7827 	free(s->tree_label);
7828 	s->tree_label = NULL;
7829 	free(s->commit_id);
7830 	s->commit_id = NULL;
7831 	free(s->head_ref_name);
7832 	s->head_ref_name = NULL;
7833 	while (!TAILQ_EMPTY(&s->parents)) {
7834 		struct tog_parent_tree *parent;
7835 		parent = TAILQ_FIRST(&s->parents);
7836 		TAILQ_REMOVE(&s->parents, parent, entry);
7837 		if (parent->tree != s->root)
7838 			got_object_tree_close(parent->tree);
7839 		free(parent);
7840 
7841 	}
7842 	if (s->tree != NULL && s->tree != s->root)
7843 		got_object_tree_close(s->tree);
7844 	if (s->root)
7845 		got_object_tree_close(s->root);
7846 	return NULL;
7847 }
7848 
7849 static const struct got_error *
7850 search_start_tree_view(struct tog_view *view)
7851 {
7852 	struct tog_tree_view_state *s = &view->state.tree;
7853 
7854 	s->matched_entry = NULL;
7855 	return NULL;
7856 }
7857 
7858 static int
7859 match_tree_entry(struct got_tree_entry *te, regex_t *regex)
7860 {
7861 	regmatch_t regmatch;
7862 
7863 	return regexec(regex, got_tree_entry_get_name(te), 1, &regmatch,
7864 	    0) == 0;
7865 }
7866 
7867 static const struct got_error *
7868 search_next_tree_view(struct tog_view *view)
7869 {
7870 	struct tog_tree_view_state *s = &view->state.tree;
7871 	struct got_tree_entry *te = NULL;
7872 
7873 	if (!view->searching) {
7874 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
7875 		return NULL;
7876 	}
7877 
7878 	if (s->matched_entry) {
7879 		if (view->searching == TOG_SEARCH_FORWARD) {
7880 			if (s->selected_entry)
7881 				te = got_tree_entry_get_next(s->tree,
7882 				    s->selected_entry);
7883 			else
7884 				te = got_object_tree_get_first_entry(s->tree);
7885 		} else {
7886 			if (s->selected_entry == NULL)
7887 				te = got_object_tree_get_last_entry(s->tree);
7888 			else
7889 				te = got_tree_entry_get_prev(s->tree,
7890 				    s->selected_entry);
7891 		}
7892 	} else {
7893 		if (s->selected_entry)
7894 			te = s->selected_entry;
7895 		else if (view->searching == TOG_SEARCH_FORWARD)
7896 			te = got_object_tree_get_first_entry(s->tree);
7897 		else
7898 			te = got_object_tree_get_last_entry(s->tree);
7899 	}
7900 
7901 	while (1) {
7902 		if (te == NULL) {
7903 			if (s->matched_entry == NULL) {
7904 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
7905 				return NULL;
7906 			}
7907 			if (view->searching == TOG_SEARCH_FORWARD)
7908 				te = got_object_tree_get_first_entry(s->tree);
7909 			else
7910 				te = got_object_tree_get_last_entry(s->tree);
7911 		}
7912 
7913 		if (match_tree_entry(te, &view->regex)) {
7914 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
7915 			s->matched_entry = te;
7916 			break;
7917 		}
7918 
7919 		if (view->searching == TOG_SEARCH_FORWARD)
7920 			te = got_tree_entry_get_next(s->tree, te);
7921 		else
7922 			te = got_tree_entry_get_prev(s->tree, te);
7923 	}
7924 
7925 	if (s->matched_entry) {
7926 		s->first_displayed_entry = s->matched_entry;
7927 		s->selected = 0;
7928 	}
7929 
7930 	return NULL;
7931 }
7932 
7933 static const struct got_error *
7934 show_tree_view(struct tog_view *view)
7935 {
7936 	const struct got_error *err = NULL;
7937 	struct tog_tree_view_state *s = &view->state.tree;
7938 	char *parent_path;
7939 
7940 	err = tree_entry_path(&parent_path, &s->parents, NULL);
7941 	if (err)
7942 		return err;
7943 
7944 	err = draw_tree_entries(view, parent_path);
7945 	free(parent_path);
7946 
7947 	view_border(view);
7948 	return err;
7949 }
7950 
7951 static const struct got_error *
7952 tree_goto_line(struct tog_view *view, int nlines)
7953 {
7954 	const struct got_error		 *err = NULL;
7955 	struct tog_tree_view_state	 *s = &view->state.tree;
7956 	struct got_tree_entry		**fte, **lte, **ste;
7957 	int				  g, last, first = 1, i = 1;
7958 	int				  root = s->tree == s->root;
7959 	int				  off = root ? 1 : 2;
7960 
7961 	g = view->gline;
7962 	view->gline = 0;
7963 
7964 	if (g == 0)
7965 		g = 1;
7966 	else if (g > got_object_tree_get_nentries(s->tree))
7967 		g = got_object_tree_get_nentries(s->tree) + (root ? 0 : 1);
7968 
7969 	fte = &s->first_displayed_entry;
7970 	lte = &s->last_displayed_entry;
7971 	ste = &s->selected_entry;
7972 
7973 	if (*fte != NULL) {
7974 		first = got_tree_entry_get_index(*fte);
7975 		first += off;  /* account for ".." */
7976 	}
7977 	last = got_tree_entry_get_index(*lte);
7978 	last += off;
7979 
7980 	if (g >= first && g <= last && g - first < nlines) {
7981 		s->selected = g - first;
7982 		return NULL;	/* gline is on the current page */
7983 	}
7984 
7985 	if (*ste != NULL) {
7986 		i = got_tree_entry_get_index(*ste);
7987 		i += off;
7988 	}
7989 
7990 	if (i < g) {
7991 		err = tree_scroll_down(view, g - i);
7992 		if (err)
7993 			return err;
7994 		if (got_tree_entry_get_index(*lte) >=
7995 		    got_object_tree_get_nentries(s->tree) - 1 &&
7996 		    first + s->selected < g &&
7997 		    s->selected < s->ndisplayed - 1) {
7998 			first = got_tree_entry_get_index(*fte);
7999 			first += off;
8000 			s->selected = g - first;
8001 		}
8002 	} else if (i > g)
8003 		tree_scroll_up(s, i - g);
8004 
8005 	if (g < nlines &&
8006 	    (*fte == NULL || (root && !got_tree_entry_get_index(*fte))))
8007 		s->selected = g - 1;
8008 
8009 	return NULL;
8010 }
8011 
8012 static const struct got_error *
8013 input_tree_view(struct tog_view **new_view, struct tog_view *view, int ch)
8014 {
8015 	const struct got_error *err = NULL;
8016 	struct tog_tree_view_state *s = &view->state.tree;
8017 	struct got_tree_entry *te;
8018 	int n, nscroll = view->nlines - 3;
8019 
8020 	if (view->gline)
8021 		return tree_goto_line(view, nscroll);
8022 
8023 	switch (ch) {
8024 	case '0':
8025 	case '$':
8026 	case KEY_RIGHT:
8027 	case 'l':
8028 	case KEY_LEFT:
8029 	case 'h':
8030 		horizontal_scroll_input(view, ch);
8031 		break;
8032 	case 'i':
8033 		s->show_ids = !s->show_ids;
8034 		view->count = 0;
8035 		break;
8036 	case 'L':
8037 		view->count = 0;
8038 		if (!s->selected_entry)
8039 			break;
8040 		err = view_request_new(new_view, view, TOG_VIEW_LOG);
8041 		break;
8042 	case 'R':
8043 		view->count = 0;
8044 		err = view_request_new(new_view, view, TOG_VIEW_REF);
8045 		break;
8046 	case 'g':
8047 	case '=':
8048 	case KEY_HOME:
8049 		s->selected = 0;
8050 		view->count = 0;
8051 		if (s->tree == s->root)
8052 			s->first_displayed_entry =
8053 			    got_object_tree_get_first_entry(s->tree);
8054 		else
8055 			s->first_displayed_entry = NULL;
8056 		break;
8057 	case 'G':
8058 	case '*':
8059 	case KEY_END: {
8060 		int eos = view->nlines - 3;
8061 
8062 		if (view->mode == TOG_VIEW_SPLIT_HRZN)
8063 			--eos;  /* border */
8064 		s->selected = 0;
8065 		view->count = 0;
8066 		te = got_object_tree_get_last_entry(s->tree);
8067 		for (n = 0; n < eos; n++) {
8068 			if (te == NULL) {
8069 				if (s->tree != s->root) {
8070 					s->first_displayed_entry = NULL;
8071 					n++;
8072 				}
8073 				break;
8074 			}
8075 			s->first_displayed_entry = te;
8076 			te = got_tree_entry_get_prev(s->tree, te);
8077 		}
8078 		if (n > 0)
8079 			s->selected = n - 1;
8080 		break;
8081 	}
8082 	case 'k':
8083 	case KEY_UP:
8084 	case CTRL('p'):
8085 		if (s->selected > 0) {
8086 			s->selected--;
8087 			break;
8088 		}
8089 		tree_scroll_up(s, 1);
8090 		if (s->selected_entry == NULL ||
8091 		    (s->tree == s->root && s->selected_entry ==
8092 		     got_object_tree_get_first_entry(s->tree)))
8093 			view->count = 0;
8094 		break;
8095 	case CTRL('u'):
8096 	case 'u':
8097 		nscroll /= 2;
8098 		/* FALL THROUGH */
8099 	case KEY_PPAGE:
8100 	case CTRL('b'):
8101 	case 'b':
8102 		if (s->tree == s->root) {
8103 			if (got_object_tree_get_first_entry(s->tree) ==
8104 			    s->first_displayed_entry)
8105 				s->selected -= MIN(s->selected, nscroll);
8106 		} else {
8107 			if (s->first_displayed_entry == NULL)
8108 				s->selected -= MIN(s->selected, nscroll);
8109 		}
8110 		tree_scroll_up(s, MAX(0, nscroll));
8111 		if (s->selected_entry == NULL ||
8112 		    (s->tree == s->root && s->selected_entry ==
8113 		     got_object_tree_get_first_entry(s->tree)))
8114 			view->count = 0;
8115 		break;
8116 	case 'j':
8117 	case KEY_DOWN:
8118 	case CTRL('n'):
8119 		if (s->selected < s->ndisplayed - 1) {
8120 			s->selected++;
8121 			break;
8122 		}
8123 		if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
8124 		    == NULL) {
8125 			/* can't scroll any further */
8126 			view->count = 0;
8127 			break;
8128 		}
8129 		tree_scroll_down(view, 1);
8130 		break;
8131 	case CTRL('d'):
8132 	case 'd':
8133 		nscroll /= 2;
8134 		/* FALL THROUGH */
8135 	case KEY_NPAGE:
8136 	case CTRL('f'):
8137 	case 'f':
8138 	case ' ':
8139 		if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
8140 		    == NULL) {
8141 			/* can't scroll any further; move cursor down */
8142 			if (s->selected < s->ndisplayed - 1)
8143 				s->selected += MIN(nscroll,
8144 				    s->ndisplayed - s->selected - 1);
8145 			else
8146 				view->count = 0;
8147 			break;
8148 		}
8149 		tree_scroll_down(view, nscroll);
8150 		break;
8151 	case KEY_ENTER:
8152 	case '\r':
8153 	case KEY_BACKSPACE:
8154 		if (s->selected_entry == NULL || ch == KEY_BACKSPACE) {
8155 			struct tog_parent_tree *parent;
8156 			/* user selected '..' */
8157 			if (s->tree == s->root) {
8158 				view->count = 0;
8159 				break;
8160 			}
8161 			parent = TAILQ_FIRST(&s->parents);
8162 			TAILQ_REMOVE(&s->parents, parent,
8163 			    entry);
8164 			got_object_tree_close(s->tree);
8165 			s->tree = parent->tree;
8166 			s->first_displayed_entry =
8167 			    parent->first_displayed_entry;
8168 			s->selected_entry =
8169 			    parent->selected_entry;
8170 			s->selected = parent->selected;
8171 			if (s->selected > view->nlines - 3) {
8172 				err = offset_selection_down(view);
8173 				if (err)
8174 					break;
8175 			}
8176 			free(parent);
8177 		} else if (S_ISDIR(got_tree_entry_get_mode(
8178 		    s->selected_entry))) {
8179 			struct got_tree_object *subtree;
8180 			view->count = 0;
8181 			err = got_object_open_as_tree(&subtree, s->repo,
8182 			    got_tree_entry_get_id(s->selected_entry));
8183 			if (err)
8184 				break;
8185 			err = tree_view_visit_subtree(s, subtree);
8186 			if (err) {
8187 				got_object_tree_close(subtree);
8188 				break;
8189 			}
8190 		} else if (S_ISREG(got_tree_entry_get_mode(s->selected_entry)))
8191 			err = view_request_new(new_view, view, TOG_VIEW_BLAME);
8192 		break;
8193 	case KEY_RESIZE:
8194 		if (view->nlines >= 4 && s->selected >= view->nlines - 3)
8195 			s->selected = view->nlines - 4;
8196 		view->count = 0;
8197 		break;
8198 	default:
8199 		view->count = 0;
8200 		break;
8201 	}
8202 
8203 	return err;
8204 }
8205 
8206 __dead static void
8207 usage_tree(void)
8208 {
8209 	endwin();
8210 	fprintf(stderr,
8211 	    "usage: %s tree [-c commit] [-r repository-path] [path]\n",
8212 	    getprogname());
8213 	exit(1);
8214 }
8215 
8216 static const struct got_error *
8217 cmd_tree(int argc, char *argv[])
8218 {
8219 	const struct got_error *error;
8220 	struct got_repository *repo = NULL;
8221 	struct got_worktree *worktree = NULL;
8222 	char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
8223 	struct got_object_id *commit_id = NULL;
8224 	struct got_commit_object *commit = NULL;
8225 	const char *commit_id_arg = NULL;
8226 	char *keyword_idstr = NULL, *label = NULL;
8227 	struct got_reference *ref = NULL;
8228 	const char *head_ref_name = NULL;
8229 	int ch;
8230 	struct tog_view *view;
8231 	int *pack_fds = NULL;
8232 
8233 	while ((ch = getopt(argc, argv, "c:r:")) != -1) {
8234 		switch (ch) {
8235 		case 'c':
8236 			commit_id_arg = optarg;
8237 			break;
8238 		case 'r':
8239 			repo_path = realpath(optarg, NULL);
8240 			if (repo_path == NULL)
8241 				return got_error_from_errno2("realpath",
8242 				    optarg);
8243 			break;
8244 		default:
8245 			usage_tree();
8246 			/* NOTREACHED */
8247 		}
8248 	}
8249 
8250 	argc -= optind;
8251 	argv += optind;
8252 
8253 	if (argc > 1)
8254 		usage_tree();
8255 
8256 	error = got_repo_pack_fds_open(&pack_fds);
8257 	if (error != NULL)
8258 		goto done;
8259 
8260 	if (repo_path == NULL) {
8261 		cwd = getcwd(NULL, 0);
8262 		if (cwd == NULL)
8263 			return got_error_from_errno("getcwd");
8264 		error = got_worktree_open(&worktree, cwd, NULL);
8265 		if (error && error->code != GOT_ERR_NOT_WORKTREE)
8266 			goto done;
8267 		if (worktree)
8268 			repo_path =
8269 			    strdup(got_worktree_get_repo_path(worktree));
8270 		else
8271 			repo_path = strdup(cwd);
8272 		if (repo_path == NULL) {
8273 			error = got_error_from_errno("strdup");
8274 			goto done;
8275 		}
8276 	}
8277 
8278 	error = got_repo_open(&repo, repo_path, NULL, pack_fds);
8279 	if (error != NULL)
8280 		goto done;
8281 
8282 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
8283 	    repo, worktree);
8284 	if (error)
8285 		goto done;
8286 
8287 	init_curses();
8288 
8289 	error = apply_unveil(got_repo_get_path(repo), NULL);
8290 	if (error)
8291 		goto done;
8292 
8293 	error = tog_load_refs(repo, 0);
8294 	if (error)
8295 		goto done;
8296 
8297 	if (commit_id_arg == NULL) {
8298 		error = got_repo_match_object_id(&commit_id, &label,
8299 		    worktree ? got_worktree_get_head_ref_name(worktree) :
8300 		    GOT_REF_HEAD, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
8301 		if (error)
8302 			goto done;
8303 		head_ref_name = label;
8304 	} else {
8305 		error = got_keyword_to_idstr(&keyword_idstr, commit_id_arg,
8306 		    repo, worktree);
8307 		if (error != NULL)
8308 			goto done;
8309 		if (keyword_idstr != NULL)
8310 			commit_id_arg = keyword_idstr;
8311 
8312 		error = got_ref_open(&ref, repo, commit_id_arg, 0);
8313 		if (error == NULL)
8314 			head_ref_name = got_ref_get_name(ref);
8315 		else if (error->code != GOT_ERR_NOT_REF)
8316 			goto done;
8317 		error = got_repo_match_object_id(&commit_id, NULL,
8318 		    commit_id_arg, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
8319 		if (error)
8320 			goto done;
8321 	}
8322 
8323 	error = got_object_open_as_commit(&commit, repo, commit_id);
8324 	if (error)
8325 		goto done;
8326 
8327 	view = view_open(0, 0, 0, 0, TOG_VIEW_TREE);
8328 	if (view == NULL) {
8329 		error = got_error_from_errno("view_open");
8330 		goto done;
8331 	}
8332 	error = open_tree_view(view, commit_id, head_ref_name, repo);
8333 	if (error)
8334 		goto done;
8335 	if (!got_path_is_root_dir(in_repo_path)) {
8336 		error = tree_view_walk_path(&view->state.tree, commit,
8337 		    in_repo_path);
8338 		if (error)
8339 			goto done;
8340 	}
8341 
8342 	if (worktree) {
8343 		error = set_tog_base_commit(repo, worktree);
8344 		if (error != NULL)
8345 			goto done;
8346 
8347 		/* Release work tree lock. */
8348 		got_worktree_close(worktree);
8349 		worktree = NULL;
8350 	}
8351 
8352 	error = view_loop(view);
8353 
8354 done:
8355 	free(tog_base_commit.id);
8356 	free(keyword_idstr);
8357 	free(repo_path);
8358 	free(cwd);
8359 	free(commit_id);
8360 	free(label);
8361 	if (commit != NULL)
8362 		got_object_commit_close(commit);
8363 	if (ref)
8364 		got_ref_close(ref);
8365 	if (worktree != NULL)
8366 		got_worktree_close(worktree);
8367 	if (repo) {
8368 		const struct got_error *close_err = got_repo_close(repo);
8369 		if (error == NULL)
8370 			error = close_err;
8371 	}
8372 	if (pack_fds) {
8373 		const struct got_error *pack_err =
8374 		    got_repo_pack_fds_close(pack_fds);
8375 		if (error == NULL)
8376 			error = pack_err;
8377 	}
8378 	tog_free_refs();
8379 	return error;
8380 }
8381 
8382 static const struct got_error *
8383 ref_view_load_refs(struct tog_ref_view_state *s)
8384 {
8385 	struct got_reflist_entry *sre;
8386 	struct tog_reflist_entry *re;
8387 
8388 	s->nrefs = 0;
8389 	TAILQ_FOREACH(sre, &tog_refs, entry) {
8390 		if (strncmp(got_ref_get_name(sre->ref),
8391 		    "refs/got/", 9) == 0 &&
8392 		    strncmp(got_ref_get_name(sre->ref),
8393 		    "refs/got/backup/", 16) != 0)
8394 			continue;
8395 
8396 		re = malloc(sizeof(*re));
8397 		if (re == NULL)
8398 			return got_error_from_errno("malloc");
8399 
8400 		re->ref = got_ref_dup(sre->ref);
8401 		if (re->ref == NULL)
8402 			return got_error_from_errno("got_ref_dup");
8403 		re->idx = s->nrefs++;
8404 		TAILQ_INSERT_TAIL(&s->refs, re, entry);
8405 	}
8406 
8407 	s->first_displayed_entry = TAILQ_FIRST(&s->refs);
8408 	return NULL;
8409 }
8410 
8411 static void
8412 ref_view_free_refs(struct tog_ref_view_state *s)
8413 {
8414 	struct tog_reflist_entry *re;
8415 
8416 	while (!TAILQ_EMPTY(&s->refs)) {
8417 		re = TAILQ_FIRST(&s->refs);
8418 		TAILQ_REMOVE(&s->refs, re, entry);
8419 		got_ref_close(re->ref);
8420 		free(re);
8421 	}
8422 }
8423 
8424 static const struct got_error *
8425 open_ref_view(struct tog_view *view, struct got_repository *repo)
8426 {
8427 	const struct got_error *err = NULL;
8428 	struct tog_ref_view_state *s = &view->state.ref;
8429 
8430 	s->selected_entry = 0;
8431 	s->repo = repo;
8432 
8433 	TAILQ_INIT(&s->refs);
8434 	STAILQ_INIT(&s->colors);
8435 
8436 	err = ref_view_load_refs(s);
8437 	if (err)
8438 		goto done;
8439 
8440 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
8441 		err = add_color(&s->colors, "^refs/heads/",
8442 		    TOG_COLOR_REFS_HEADS,
8443 		    get_color_value("TOG_COLOR_REFS_HEADS"));
8444 		if (err)
8445 			goto done;
8446 
8447 		err = add_color(&s->colors, "^refs/tags/",
8448 		    TOG_COLOR_REFS_TAGS,
8449 		    get_color_value("TOG_COLOR_REFS_TAGS"));
8450 		if (err)
8451 			goto done;
8452 
8453 		err = add_color(&s->colors, "^refs/remotes/",
8454 		    TOG_COLOR_REFS_REMOTES,
8455 		    get_color_value("TOG_COLOR_REFS_REMOTES"));
8456 		if (err)
8457 			goto done;
8458 
8459 		err = add_color(&s->colors, "^refs/got/backup/",
8460 		    TOG_COLOR_REFS_BACKUP,
8461 		    get_color_value("TOG_COLOR_REFS_BACKUP"));
8462 		if (err)
8463 			goto done;
8464 	}
8465 
8466 	view->show = show_ref_view;
8467 	view->input = input_ref_view;
8468 	view->close = close_ref_view;
8469 	view->search_start = search_start_ref_view;
8470 	view->search_next = search_next_ref_view;
8471 done:
8472 	if (err) {
8473 		if (view->close == NULL)
8474 			close_ref_view(view);
8475 		view_close(view);
8476 	}
8477 	return err;
8478 }
8479 
8480 static const struct got_error *
8481 close_ref_view(struct tog_view *view)
8482 {
8483 	struct tog_ref_view_state *s = &view->state.ref;
8484 
8485 	ref_view_free_refs(s);
8486 	free_colors(&s->colors);
8487 
8488 	return NULL;
8489 }
8490 
8491 static const struct got_error *
8492 resolve_reflist_entry(struct got_object_id **commit_id,
8493     struct tog_reflist_entry *re, struct got_repository *repo)
8494 {
8495 	const struct got_error *err = NULL;
8496 	struct got_object_id *obj_id;
8497 	struct got_tag_object *tag = NULL;
8498 	int obj_type;
8499 
8500 	*commit_id = NULL;
8501 
8502 	err = got_ref_resolve(&obj_id, repo, re->ref);
8503 	if (err)
8504 		return err;
8505 
8506 	err = got_object_get_type(&obj_type, repo, obj_id);
8507 	if (err)
8508 		goto done;
8509 
8510 	switch (obj_type) {
8511 	case GOT_OBJ_TYPE_COMMIT:
8512 		*commit_id = obj_id;
8513 		break;
8514 	case GOT_OBJ_TYPE_TAG:
8515 		err = got_object_open_as_tag(&tag, repo, obj_id);
8516 		if (err)
8517 			goto done;
8518 		free(obj_id);
8519 		err = got_object_get_type(&obj_type, repo,
8520 		    got_object_tag_get_object_id(tag));
8521 		if (err)
8522 			goto done;
8523 		if (obj_type != GOT_OBJ_TYPE_COMMIT) {
8524 			err = got_error(GOT_ERR_OBJ_TYPE);
8525 			goto done;
8526 		}
8527 		*commit_id = got_object_id_dup(
8528 		    got_object_tag_get_object_id(tag));
8529 		if (*commit_id == NULL) {
8530 			err = got_error_from_errno("got_object_id_dup");
8531 			goto done;
8532 		}
8533 		break;
8534 	default:
8535 		err = got_error(GOT_ERR_OBJ_TYPE);
8536 		break;
8537 	}
8538 
8539 done:
8540 	if (tag)
8541 		got_object_tag_close(tag);
8542 	if (err) {
8543 		free(*commit_id);
8544 		*commit_id = NULL;
8545 	}
8546 	return err;
8547 }
8548 
8549 static const struct got_error *
8550 log_ref_entry(struct tog_view **new_view, int begin_y, int begin_x,
8551     struct tog_reflist_entry *re, struct got_repository *repo)
8552 {
8553 	struct tog_view *log_view;
8554 	const struct got_error *err = NULL;
8555 	struct got_object_id *commit_id = NULL;
8556 
8557 	*new_view = NULL;
8558 
8559 	err = resolve_reflist_entry(&commit_id, re, repo);
8560 	if (err) {
8561 		if (err->code != GOT_ERR_OBJ_TYPE)
8562 			return err;
8563 		else
8564 			return NULL;
8565 	}
8566 
8567 	log_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_LOG);
8568 	if (log_view == NULL) {
8569 		err = got_error_from_errno("view_open");
8570 		goto done;
8571 	}
8572 
8573 	err = open_log_view(log_view, commit_id, repo,
8574 	    got_ref_get_name(re->ref), "", 0, NULL);
8575 done:
8576 	if (err)
8577 		view_close(log_view);
8578 	else
8579 		*new_view = log_view;
8580 	free(commit_id);
8581 	return err;
8582 }
8583 
8584 static void
8585 ref_scroll_up(struct tog_ref_view_state *s, int maxscroll)
8586 {
8587 	struct tog_reflist_entry *re;
8588 	int i = 0;
8589 
8590 	if (s->first_displayed_entry == TAILQ_FIRST(&s->refs))
8591 		return;
8592 
8593 	re = TAILQ_PREV(s->first_displayed_entry, tog_reflist_head, entry);
8594 	while (i++ < maxscroll) {
8595 		if (re == NULL)
8596 			break;
8597 		s->first_displayed_entry = re;
8598 		re = TAILQ_PREV(re, tog_reflist_head, entry);
8599 	}
8600 }
8601 
8602 static const struct got_error *
8603 ref_scroll_down(struct tog_view *view, int maxscroll)
8604 {
8605 	struct tog_ref_view_state *s = &view->state.ref;
8606 	struct tog_reflist_entry *next, *last;
8607 	int n = 0;
8608 
8609 	if (s->first_displayed_entry)
8610 		next = TAILQ_NEXT(s->first_displayed_entry, entry);
8611 	else
8612 		next = TAILQ_FIRST(&s->refs);
8613 
8614 	last = s->last_displayed_entry;
8615 	while (next && n++ < maxscroll) {
8616 		if (last) {
8617 			s->last_displayed_entry = last;
8618 			last = TAILQ_NEXT(last, entry);
8619 		}
8620 		if (last || (view->mode == TOG_VIEW_SPLIT_HRZN)) {
8621 			s->first_displayed_entry = next;
8622 			next = TAILQ_NEXT(next, entry);
8623 		}
8624 	}
8625 
8626 	return NULL;
8627 }
8628 
8629 static const struct got_error *
8630 search_start_ref_view(struct tog_view *view)
8631 {
8632 	struct tog_ref_view_state *s = &view->state.ref;
8633 
8634 	s->matched_entry = NULL;
8635 	return NULL;
8636 }
8637 
8638 static int
8639 match_reflist_entry(struct tog_reflist_entry *re, regex_t *regex)
8640 {
8641 	regmatch_t regmatch;
8642 
8643 	return regexec(regex, got_ref_get_name(re->ref), 1, &regmatch,
8644 	    0) == 0;
8645 }
8646 
8647 static const struct got_error *
8648 search_next_ref_view(struct tog_view *view)
8649 {
8650 	struct tog_ref_view_state *s = &view->state.ref;
8651 	struct tog_reflist_entry *re = NULL;
8652 
8653 	if (!view->searching) {
8654 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
8655 		return NULL;
8656 	}
8657 
8658 	if (s->matched_entry) {
8659 		if (view->searching == TOG_SEARCH_FORWARD) {
8660 			if (s->selected_entry)
8661 				re = TAILQ_NEXT(s->selected_entry, entry);
8662 			else
8663 				re = TAILQ_PREV(s->selected_entry,
8664 				    tog_reflist_head, entry);
8665 		} else {
8666 			if (s->selected_entry == NULL)
8667 				re = TAILQ_LAST(&s->refs, tog_reflist_head);
8668 			else
8669 				re = TAILQ_PREV(s->selected_entry,
8670 				    tog_reflist_head, entry);
8671 		}
8672 	} else {
8673 		if (s->selected_entry)
8674 			re = s->selected_entry;
8675 		else if (view->searching == TOG_SEARCH_FORWARD)
8676 			re = TAILQ_FIRST(&s->refs);
8677 		else
8678 			re = TAILQ_LAST(&s->refs, tog_reflist_head);
8679 	}
8680 
8681 	while (1) {
8682 		if (re == NULL) {
8683 			if (s->matched_entry == NULL) {
8684 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
8685 				return NULL;
8686 			}
8687 			if (view->searching == TOG_SEARCH_FORWARD)
8688 				re = TAILQ_FIRST(&s->refs);
8689 			else
8690 				re = TAILQ_LAST(&s->refs, tog_reflist_head);
8691 		}
8692 
8693 		if (match_reflist_entry(re, &view->regex)) {
8694 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
8695 			s->matched_entry = re;
8696 			break;
8697 		}
8698 
8699 		if (view->searching == TOG_SEARCH_FORWARD)
8700 			re = TAILQ_NEXT(re, entry);
8701 		else
8702 			re = TAILQ_PREV(re, tog_reflist_head, entry);
8703 	}
8704 
8705 	if (s->matched_entry) {
8706 		s->first_displayed_entry = s->matched_entry;
8707 		s->selected = 0;
8708 	}
8709 
8710 	return NULL;
8711 }
8712 
8713 static const struct got_error *
8714 show_ref_view(struct tog_view *view)
8715 {
8716 	const struct got_error *err = NULL;
8717 	struct tog_ref_view_state *s = &view->state.ref;
8718 	struct tog_reflist_entry *re;
8719 	char *line = NULL;
8720 	wchar_t *wline;
8721 	struct tog_color *tc;
8722 	int width, n, scrollx;
8723 	int limit = view->nlines;
8724 
8725 	werase(view->window);
8726 
8727 	s->ndisplayed = 0;
8728 	if (view_is_hsplit_top(view))
8729 		--limit;  /* border */
8730 
8731 	if (limit == 0)
8732 		return NULL;
8733 
8734 	re = s->first_displayed_entry;
8735 
8736 	if (asprintf(&line, "references [%d/%d]", re->idx + s->selected + 1,
8737 	    s->nrefs) == -1)
8738 		return got_error_from_errno("asprintf");
8739 
8740 	err = format_line(&wline, &width, NULL, line, 0, view->ncols, 0, 0);
8741 	if (err) {
8742 		free(line);
8743 		return err;
8744 	}
8745 	if (view_needs_focus_indication(view))
8746 		wstandout(view->window);
8747 	waddwstr(view->window, wline);
8748 	while (width++ < view->ncols)
8749 		waddch(view->window, ' ');
8750 	if (view_needs_focus_indication(view))
8751 		wstandend(view->window);
8752 	free(wline);
8753 	wline = NULL;
8754 	free(line);
8755 	line = NULL;
8756 	if (--limit <= 0)
8757 		return NULL;
8758 
8759 	n = 0;
8760 	view->maxx = 0;
8761 	while (re && limit > 0) {
8762 		char *line = NULL;
8763 		char ymd[13];  /* YYYY-MM-DD + "  " + NUL */
8764 
8765 		if (s->show_date) {
8766 			struct got_commit_object *ci;
8767 			struct got_tag_object *tag;
8768 			struct got_object_id *id;
8769 			struct tm tm;
8770 			time_t t;
8771 
8772 			err = got_ref_resolve(&id, s->repo, re->ref);
8773 			if (err)
8774 				return err;
8775 			err = got_object_open_as_tag(&tag, s->repo, id);
8776 			if (err) {
8777 				if (err->code != GOT_ERR_OBJ_TYPE) {
8778 					free(id);
8779 					return err;
8780 				}
8781 				err = got_object_open_as_commit(&ci, s->repo,
8782 				    id);
8783 				if (err) {
8784 					free(id);
8785 					return err;
8786 				}
8787 				t = got_object_commit_get_committer_time(ci);
8788 				got_object_commit_close(ci);
8789 			} else {
8790 				t = got_object_tag_get_tagger_time(tag);
8791 				got_object_tag_close(tag);
8792 			}
8793 			free(id);
8794 			if (gmtime_r(&t, &tm) == NULL)
8795 				return got_error_from_errno("gmtime_r");
8796 			if (strftime(ymd, sizeof(ymd), "%G-%m-%d  ", &tm) == 0)
8797 				return got_error(GOT_ERR_NO_SPACE);
8798 		}
8799 		if (got_ref_is_symbolic(re->ref)) {
8800 			if (asprintf(&line, "%s%s -> %s", s->show_date ?
8801 			    ymd : "", got_ref_get_name(re->ref),
8802 			    got_ref_get_symref_target(re->ref)) == -1)
8803 				return got_error_from_errno("asprintf");
8804 		} else if (s->show_ids) {
8805 			struct got_object_id *id;
8806 			char *id_str;
8807 			err = got_ref_resolve(&id, s->repo, re->ref);
8808 			if (err)
8809 				return err;
8810 			err = got_object_id_str(&id_str, id);
8811 			if (err) {
8812 				free(id);
8813 				return err;
8814 			}
8815 			if (asprintf(&line, "%s%s: %s", s->show_date ? ymd : "",
8816 			    got_ref_get_name(re->ref), id_str) == -1) {
8817 				err = got_error_from_errno("asprintf");
8818 				free(id);
8819 				free(id_str);
8820 				return err;
8821 			}
8822 			free(id);
8823 			free(id_str);
8824 		} else if (asprintf(&line, "%s%s", s->show_date ? ymd : "",
8825 		    got_ref_get_name(re->ref)) == -1)
8826 			return got_error_from_errno("asprintf");
8827 
8828 		/* use full line width to determine view->maxx */
8829 		err = format_line(&wline, &width, NULL, line, 0, INT_MAX, 0, 0);
8830 		if (err) {
8831 			free(line);
8832 			return err;
8833 		}
8834 		view->maxx = MAX(view->maxx, width);
8835 		free(wline);
8836 		wline = NULL;
8837 
8838 		err = format_line(&wline, &width, &scrollx, line, view->x,
8839 		    view->ncols, 0, 0);
8840 		if (err) {
8841 			free(line);
8842 			return err;
8843 		}
8844 		if (n == s->selected) {
8845 			if (view->focussed)
8846 				wstandout(view->window);
8847 			s->selected_entry = re;
8848 		}
8849 		tc = match_color(&s->colors, got_ref_get_name(re->ref));
8850 		if (tc)
8851 			wattr_on(view->window,
8852 			    COLOR_PAIR(tc->colorpair), NULL);
8853 		waddwstr(view->window, &wline[scrollx]);
8854 		if (tc)
8855 			wattr_off(view->window,
8856 			    COLOR_PAIR(tc->colorpair), NULL);
8857 		if (width < view->ncols)
8858 			waddch(view->window, '\n');
8859 		if (n == s->selected && view->focussed)
8860 			wstandend(view->window);
8861 		free(line);
8862 		free(wline);
8863 		wline = NULL;
8864 		n++;
8865 		s->ndisplayed++;
8866 		s->last_displayed_entry = re;
8867 
8868 		limit--;
8869 		re = TAILQ_NEXT(re, entry);
8870 	}
8871 
8872 	view_border(view);
8873 	return err;
8874 }
8875 
8876 static const struct got_error *
8877 browse_ref_tree(struct tog_view **new_view, int begin_y, int begin_x,
8878     struct tog_reflist_entry *re, struct got_repository *repo)
8879 {
8880 	const struct got_error *err = NULL;
8881 	struct got_object_id *commit_id = NULL;
8882 	struct tog_view *tree_view;
8883 
8884 	*new_view = NULL;
8885 
8886 	err = resolve_reflist_entry(&commit_id, re, repo);
8887 	if (err) {
8888 		if (err->code != GOT_ERR_OBJ_TYPE)
8889 			return err;
8890 		else
8891 			return NULL;
8892 	}
8893 
8894 
8895 	tree_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_TREE);
8896 	if (tree_view == NULL) {
8897 		err = got_error_from_errno("view_open");
8898 		goto done;
8899 	}
8900 
8901 	err = open_tree_view(tree_view, commit_id,
8902 	    got_ref_get_name(re->ref), repo);
8903 	if (err)
8904 		goto done;
8905 
8906 	*new_view = tree_view;
8907 done:
8908 	free(commit_id);
8909 	return err;
8910 }
8911 
8912 static const struct got_error *
8913 ref_goto_line(struct tog_view *view, int nlines)
8914 {
8915 	const struct got_error		*err = NULL;
8916 	struct tog_ref_view_state	*s = &view->state.ref;
8917 	int				 g, idx = s->selected_entry->idx;
8918 
8919 	g = view->gline;
8920 	view->gline = 0;
8921 
8922 	if (g == 0)
8923 		g = 1;
8924 	else if (g > s->nrefs)
8925 		g = s->nrefs;
8926 
8927 	if (g >= s->first_displayed_entry->idx + 1 &&
8928 	    g <= s->last_displayed_entry->idx + 1 &&
8929 	    g - s->first_displayed_entry->idx - 1 < nlines) {
8930 		s->selected = g - s->first_displayed_entry->idx - 1;
8931 		return NULL;
8932 	}
8933 
8934 	if (idx + 1 < g) {
8935 		err = ref_scroll_down(view, g - idx - 1);
8936 		if (err)
8937 			return err;
8938 		if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL &&
8939 		    s->first_displayed_entry->idx + s->selected < g &&
8940 		    s->selected < s->ndisplayed - 1)
8941 			s->selected = g - s->first_displayed_entry->idx - 1;
8942 	} else if (idx + 1 > g)
8943 		ref_scroll_up(s, idx - g + 1);
8944 
8945 	if (g < nlines && s->first_displayed_entry->idx == 0)
8946 		s->selected = g - 1;
8947 
8948 	return NULL;
8949 
8950 }
8951 
8952 static const struct got_error *
8953 input_ref_view(struct tog_view **new_view, struct tog_view *view, int ch)
8954 {
8955 	const struct got_error *err = NULL;
8956 	struct tog_ref_view_state *s = &view->state.ref;
8957 	struct tog_reflist_entry *re;
8958 	int n, nscroll = view->nlines - 1;
8959 
8960 	if (view->gline)
8961 		return ref_goto_line(view, nscroll);
8962 
8963 	switch (ch) {
8964 	case '0':
8965 	case '$':
8966 	case KEY_RIGHT:
8967 	case 'l':
8968 	case KEY_LEFT:
8969 	case 'h':
8970 		horizontal_scroll_input(view, ch);
8971 		break;
8972 	case 'i':
8973 		s->show_ids = !s->show_ids;
8974 		view->count = 0;
8975 		break;
8976 	case 'm':
8977 		s->show_date = !s->show_date;
8978 		view->count = 0;
8979 		break;
8980 	case 'o':
8981 		s->sort_by_date = !s->sort_by_date;
8982 		view->action = s->sort_by_date ? "sort by date" : "sort by name";
8983 		view->count = 0;
8984 		err = got_reflist_sort(&tog_refs, s->sort_by_date ?
8985 		    got_ref_cmp_by_commit_timestamp_descending :
8986 		    tog_ref_cmp_by_name, s->repo);
8987 		if (err)
8988 			break;
8989 		got_reflist_object_id_map_free(tog_refs_idmap);
8990 		err = got_reflist_object_id_map_create(&tog_refs_idmap,
8991 		    &tog_refs, s->repo);
8992 		if (err)
8993 			break;
8994 		ref_view_free_refs(s);
8995 		err = ref_view_load_refs(s);
8996 		break;
8997 	case KEY_ENTER:
8998 	case '\r':
8999 		view->count = 0;
9000 		if (!s->selected_entry)
9001 			break;
9002 		err = view_request_new(new_view, view, TOG_VIEW_LOG);
9003 		break;
9004 	case 'T':
9005 		view->count = 0;
9006 		if (!s->selected_entry)
9007 			break;
9008 		err = view_request_new(new_view, view, TOG_VIEW_TREE);
9009 		break;
9010 	case 'g':
9011 	case '=':
9012 	case KEY_HOME:
9013 		s->selected = 0;
9014 		view->count = 0;
9015 		s->first_displayed_entry = TAILQ_FIRST(&s->refs);
9016 		break;
9017 	case 'G':
9018 	case '*':
9019 	case KEY_END: {
9020 		int eos = view->nlines - 1;
9021 
9022 		if (view->mode == TOG_VIEW_SPLIT_HRZN)
9023 			--eos;  /* border */
9024 		s->selected = 0;
9025 		view->count = 0;
9026 		re = TAILQ_LAST(&s->refs, tog_reflist_head);
9027 		for (n = 0; n < eos; n++) {
9028 			if (re == NULL)
9029 				break;
9030 			s->first_displayed_entry = re;
9031 			re = TAILQ_PREV(re, tog_reflist_head, entry);
9032 		}
9033 		if (n > 0)
9034 			s->selected = n - 1;
9035 		break;
9036 	}
9037 	case 'k':
9038 	case KEY_UP:
9039 	case CTRL('p'):
9040 		if (s->selected > 0) {
9041 			s->selected--;
9042 			break;
9043 		}
9044 		ref_scroll_up(s, 1);
9045 		if (s->selected_entry == TAILQ_FIRST(&s->refs))
9046 			view->count = 0;
9047 		break;
9048 	case CTRL('u'):
9049 	case 'u':
9050 		nscroll /= 2;
9051 		/* FALL THROUGH */
9052 	case KEY_PPAGE:
9053 	case CTRL('b'):
9054 	case 'b':
9055 		if (s->first_displayed_entry == TAILQ_FIRST(&s->refs))
9056 			s->selected -= MIN(nscroll, s->selected);
9057 		ref_scroll_up(s, MAX(0, nscroll));
9058 		if (s->selected_entry == TAILQ_FIRST(&s->refs))
9059 			view->count = 0;
9060 		break;
9061 	case 'j':
9062 	case KEY_DOWN:
9063 	case CTRL('n'):
9064 		if (s->selected < s->ndisplayed - 1) {
9065 			s->selected++;
9066 			break;
9067 		}
9068 		if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL) {
9069 			/* can't scroll any further */
9070 			view->count = 0;
9071 			break;
9072 		}
9073 		ref_scroll_down(view, 1);
9074 		break;
9075 	case CTRL('d'):
9076 	case 'd':
9077 		nscroll /= 2;
9078 		/* FALL THROUGH */
9079 	case KEY_NPAGE:
9080 	case CTRL('f'):
9081 	case 'f':
9082 	case ' ':
9083 		if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL) {
9084 			/* can't scroll any further; move cursor down */
9085 			if (s->selected < s->ndisplayed - 1)
9086 				s->selected += MIN(nscroll,
9087 				    s->ndisplayed - s->selected - 1);
9088 			if (view->count > 1 && s->selected < s->ndisplayed - 1)
9089 				s->selected += s->ndisplayed - s->selected - 1;
9090 			view->count = 0;
9091 			break;
9092 		}
9093 		ref_scroll_down(view, nscroll);
9094 		break;
9095 	case CTRL('l'):
9096 		view->count = 0;
9097 		tog_free_refs();
9098 		err = tog_load_refs(s->repo, s->sort_by_date);
9099 		if (err)
9100 			break;
9101 		ref_view_free_refs(s);
9102 		err = ref_view_load_refs(s);
9103 		break;
9104 	case KEY_RESIZE:
9105 		if (view->nlines >= 2 && s->selected >= view->nlines - 1)
9106 			s->selected = view->nlines - 2;
9107 		break;
9108 	default:
9109 		view->count = 0;
9110 		break;
9111 	}
9112 
9113 	return err;
9114 }
9115 
9116 __dead static void
9117 usage_ref(void)
9118 {
9119 	endwin();
9120 	fprintf(stderr, "usage: %s ref [-r repository-path]\n",
9121 	    getprogname());
9122 	exit(1);
9123 }
9124 
9125 static const struct got_error *
9126 cmd_ref(int argc, char *argv[])
9127 {
9128 	const struct got_error *error;
9129 	struct got_repository *repo = NULL;
9130 	struct got_worktree *worktree = NULL;
9131 	char *cwd = NULL, *repo_path = NULL;
9132 	int ch;
9133 	struct tog_view *view;
9134 	int *pack_fds = NULL;
9135 
9136 	while ((ch = getopt(argc, argv, "r:")) != -1) {
9137 		switch (ch) {
9138 		case 'r':
9139 			repo_path = realpath(optarg, NULL);
9140 			if (repo_path == NULL)
9141 				return got_error_from_errno2("realpath",
9142 				    optarg);
9143 			break;
9144 		default:
9145 			usage_ref();
9146 			/* NOTREACHED */
9147 		}
9148 	}
9149 
9150 	argc -= optind;
9151 	argv += optind;
9152 
9153 	if (argc > 1)
9154 		usage_ref();
9155 
9156 	error = got_repo_pack_fds_open(&pack_fds);
9157 	if (error != NULL)
9158 		goto done;
9159 
9160 	if (repo_path == NULL) {
9161 		cwd = getcwd(NULL, 0);
9162 		if (cwd == NULL)
9163 			return got_error_from_errno("getcwd");
9164 		error = got_worktree_open(&worktree, cwd, NULL);
9165 		if (error && error->code != GOT_ERR_NOT_WORKTREE)
9166 			goto done;
9167 		if (worktree)
9168 			repo_path =
9169 			    strdup(got_worktree_get_repo_path(worktree));
9170 		else
9171 			repo_path = strdup(cwd);
9172 		if (repo_path == NULL) {
9173 			error = got_error_from_errno("strdup");
9174 			goto done;
9175 		}
9176 	}
9177 
9178 	error = got_repo_open(&repo, repo_path, NULL, pack_fds);
9179 	if (error != NULL)
9180 		goto done;
9181 
9182 	init_curses();
9183 
9184 	error = apply_unveil(got_repo_get_path(repo), NULL);
9185 	if (error)
9186 		goto done;
9187 
9188 	error = tog_load_refs(repo, 0);
9189 	if (error)
9190 		goto done;
9191 
9192 	view = view_open(0, 0, 0, 0, TOG_VIEW_REF);
9193 	if (view == NULL) {
9194 		error = got_error_from_errno("view_open");
9195 		goto done;
9196 	}
9197 
9198 	error = open_ref_view(view, repo);
9199 	if (error)
9200 		goto done;
9201 
9202 	if (worktree) {
9203 		error = set_tog_base_commit(repo, worktree);
9204 		if (error != NULL)
9205 			goto done;
9206 
9207 		/* Release work tree lock. */
9208 		got_worktree_close(worktree);
9209 		worktree = NULL;
9210 	}
9211 
9212 	error = view_loop(view);
9213 
9214 done:
9215 	free(tog_base_commit.id);
9216 	free(repo_path);
9217 	free(cwd);
9218 	if (worktree != NULL)
9219 		got_worktree_close(worktree);
9220 	if (repo) {
9221 		const struct got_error *close_err = got_repo_close(repo);
9222 		if (close_err)
9223 			error = close_err;
9224 	}
9225 	if (pack_fds) {
9226 		const struct got_error *pack_err =
9227 		    got_repo_pack_fds_close(pack_fds);
9228 		if (error == NULL)
9229 			error = pack_err;
9230 	}
9231 	tog_free_refs();
9232 	return error;
9233 }
9234 
9235 static const struct got_error*
9236 win_draw_center(WINDOW *win, size_t y, size_t x, size_t maxx, int focus,
9237     const char *str)
9238 {
9239 	size_t len;
9240 
9241 	if (win == NULL)
9242 		win = stdscr;
9243 
9244 	len = strlen(str);
9245 	x = x ? x : maxx > len ? (maxx - len) / 2 : 0;
9246 
9247 	if (focus)
9248 		wstandout(win);
9249 	if (mvwprintw(win, y, x, "%s", str) == ERR)
9250 		return got_error_msg(GOT_ERR_RANGE, "mvwprintw");
9251 	if (focus)
9252 		wstandend(win);
9253 
9254 	return NULL;
9255 }
9256 
9257 static const struct got_error *
9258 add_line_offset(off_t **line_offsets, size_t *nlines, off_t off)
9259 {
9260 	off_t *p;
9261 
9262 	p = reallocarray(*line_offsets, *nlines + 1, sizeof(off_t));
9263 	if (p == NULL) {
9264 		free(*line_offsets);
9265 		*line_offsets = NULL;
9266 		return got_error_from_errno("reallocarray");
9267 	}
9268 
9269 	*line_offsets = p;
9270 	(*line_offsets)[*nlines] = off;
9271 	++(*nlines);
9272 	return NULL;
9273 }
9274 
9275 static const struct got_error *
9276 max_key_str(int *ret, const struct tog_key_map *km, size_t n)
9277 {
9278 	*ret = 0;
9279 
9280 	for (;n > 0; --n, ++km) {
9281 		char	*t0, *t, *k;
9282 		size_t	 len = 1;
9283 
9284 		if (km->keys == NULL)
9285 			continue;
9286 
9287 		t = t0 = strdup(km->keys);
9288 		if (t0 == NULL)
9289 			return got_error_from_errno("strdup");
9290 
9291 		len += strlen(t);
9292 		while ((k = strsep(&t, " ")) != NULL)
9293 			len += strlen(k) > 1 ? 2 : 0;
9294 		free(t0);
9295 		*ret = MAX(*ret, len);
9296 	}
9297 
9298 	return NULL;
9299 }
9300 
9301 /*
9302  * Write keymap section headers, keys, and key info in km to f.
9303  * Save line offset to *off. If terminal has UTF8 encoding enabled,
9304  * wrap control and symbolic keys in guillemets, else use <>.
9305  */
9306 static const struct got_error *
9307 format_help_line(off_t *off, FILE *f, const struct tog_key_map *km, int width)
9308 {
9309 	int n, len = width;
9310 
9311 	if (km->keys) {
9312 		static const char *u8_glyph[] = {
9313 			"\xe2\x80\xb9", /* U+2039  (utf8 <) */
9314 			"\xe2\x80\xba"  /* U+203A  (utf8 >) */
9315 		};
9316 		char	*t0, *t, *k;
9317 		int	 cs, s, first = 1;
9318 
9319 		cs = got_locale_is_utf8();
9320 
9321 		t = t0 = strdup(km->keys);
9322 		if (t0 == NULL)
9323 			return got_error_from_errno("strdup");
9324 
9325 		len = strlen(km->keys);
9326 		while ((k = strsep(&t, " ")) != NULL) {
9327 			s = strlen(k) > 1;  /* control or symbolic key */
9328 			n = fprintf(f, "%s%s%s%s%s", first ? "  " : "",
9329 			    cs && s ? u8_glyph[0] : s ? "<" : "", k,
9330 			    cs && s ? u8_glyph[1] : s ? ">" : "", t ? " " : "");
9331 			if (n < 0) {
9332 				free(t0);
9333 				return got_error_from_errno("fprintf");
9334 			}
9335 			first = 0;
9336 			len += s ? 2 : 0;
9337 			*off += n;
9338 		}
9339 		free(t0);
9340 	}
9341 	n = fprintf(f, "%*s%s\n", width - len, width - len ? " " : "", km->info);
9342 	if (n < 0)
9343 		return got_error_from_errno("fprintf");
9344 	*off += n;
9345 
9346 	return NULL;
9347 }
9348 
9349 static const struct got_error *
9350 format_help(struct tog_help_view_state *s)
9351 {
9352 	const struct got_error		*err = NULL;
9353 	off_t				 off = 0;
9354 	int				 i, max, n, show = s->all;
9355 	static const struct tog_key_map	 km[] = {
9356 #define KEYMAP_(info, type)	{ NULL, (info), type }
9357 #define KEY_(keys, info)	{ (keys), (info), TOG_KEYMAP_KEYS }
9358 		GENERATE_HELP
9359 #undef KEYMAP_
9360 #undef KEY_
9361 	};
9362 
9363 	err = add_line_offset(&s->line_offsets, &s->nlines, 0);
9364 	if (err)
9365 		return err;
9366 
9367 	n = nitems(km);
9368 	err = max_key_str(&max, km, n);
9369 	if (err)
9370 		return err;
9371 
9372 	for (i = 0; i < n; ++i) {
9373 		if (km[i].keys == NULL) {
9374 			show = s->all;
9375 			if (km[i].type == TOG_KEYMAP_GLOBAL ||
9376 			    km[i].type == s->type || s->all)
9377 				show = 1;
9378 		}
9379 		if (show) {
9380 			err = format_help_line(&off, s->f, &km[i], max);
9381 			if (err)
9382 				return err;
9383 			err = add_line_offset(&s->line_offsets, &s->nlines, off);
9384 			if (err)
9385 				return err;
9386 		}
9387 	}
9388 	fputc('\n', s->f);
9389 	++off;
9390 	err = add_line_offset(&s->line_offsets, &s->nlines, off);
9391 	return err;
9392 }
9393 
9394 static const struct got_error *
9395 create_help(struct tog_help_view_state *s)
9396 {
9397 	FILE			*f;
9398 	const struct got_error	*err;
9399 
9400 	free(s->line_offsets);
9401 	s->line_offsets = NULL;
9402 	s->nlines = 0;
9403 
9404 	f = got_opentemp();
9405 	if (f == NULL)
9406 		return got_error_from_errno("got_opentemp");
9407 	s->f = f;
9408 
9409 	err = format_help(s);
9410 	if (err)
9411 		return err;
9412 
9413 	if (s->f && fflush(s->f) != 0)
9414 		return got_error_from_errno("fflush");
9415 
9416 	return NULL;
9417 }
9418 
9419 static const struct got_error *
9420 search_start_help_view(struct tog_view *view)
9421 {
9422 	view->state.help.matched_line = 0;
9423 	return NULL;
9424 }
9425 
9426 static void
9427 search_setup_help_view(struct tog_view *view, FILE **f, off_t **line_offsets,
9428     size_t *nlines, int **first, int **last, int **match, int **selected)
9429 {
9430 	struct tog_help_view_state *s = &view->state.help;
9431 
9432 	*f = s->f;
9433 	*nlines = s->nlines;
9434 	*line_offsets = s->line_offsets;
9435 	*match = &s->matched_line;
9436 	*first = &s->first_displayed_line;
9437 	*last = &s->last_displayed_line;
9438 	*selected = &s->selected_line;
9439 }
9440 
9441 static const struct got_error *
9442 show_help_view(struct tog_view *view)
9443 {
9444 	struct tog_help_view_state	*s = &view->state.help;
9445 	const struct got_error		*err;
9446 	regmatch_t			*regmatch = &view->regmatch;
9447 	wchar_t				*wline;
9448 	char				*line;
9449 	ssize_t				 linelen;
9450 	size_t				 linesz = 0;
9451 	int				 width, nprinted = 0, rc = 0;
9452 	int				 eos = view->nlines;
9453 
9454 	if (view_is_hsplit_top(view))
9455 		--eos;  /* account for border */
9456 
9457 	s->lineno = 0;
9458 	rewind(s->f);
9459 	werase(view->window);
9460 
9461 	if (view->gline > s->nlines - 1)
9462 		view->gline = s->nlines - 1;
9463 
9464 	err = win_draw_center(view->window, 0, 0, view->ncols,
9465 	    view_needs_focus_indication(view),
9466 	    "tog help (press q to return to tog)");
9467 	if (err)
9468 		return err;
9469 	if (eos <= 1)
9470 		return NULL;
9471 	waddstr(view->window, "\n\n");
9472 	eos -= 2;
9473 
9474 	s->eof = 0;
9475 	view->maxx = 0;
9476 	line = NULL;
9477 	while (eos > 0 && nprinted < eos) {
9478 		attr_t attr = 0;
9479 
9480 		linelen = getline(&line, &linesz, s->f);
9481 		if (linelen == -1) {
9482 			if (!feof(s->f)) {
9483 				free(line);
9484 				return got_ferror(s->f, GOT_ERR_IO);
9485 			}
9486 			s->eof = 1;
9487 			break;
9488 		}
9489 		if (++s->lineno < s->first_displayed_line)
9490 			continue;
9491 		if (view->gline && !gotoline(view, &s->lineno, &nprinted))
9492 			continue;
9493 		if (s->lineno == view->hiline)
9494 			attr = A_STANDOUT;
9495 
9496 		err = format_line(&wline, &width, NULL, line, 0, INT_MAX, 0,
9497 		    view->x ? 1 : 0);
9498 		if (err) {
9499 			free(line);
9500 			return err;
9501 		}
9502 		view->maxx = MAX(view->maxx, width);
9503 		free(wline);
9504 		wline = NULL;
9505 
9506 		if (attr)
9507 			wattron(view->window, attr);
9508 		if (s->first_displayed_line + nprinted == s->matched_line &&
9509 		    regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
9510 			err = add_matched_line(&width, line, view->ncols - 1, 0,
9511 			    view->window, view->x, regmatch);
9512 			if (err) {
9513 				free(line);
9514 				return err;
9515 			}
9516 		} else {
9517 			int skip;
9518 
9519 			err = format_line(&wline, &width, &skip, line,
9520 			    view->x, view->ncols, 0, view->x ? 1 : 0);
9521 			if (err) {
9522 				free(line);
9523 				return err;
9524 			}
9525 			waddwstr(view->window, &wline[skip]);
9526 			free(wline);
9527 			wline = NULL;
9528 		}
9529 		if (s->lineno == view->hiline) {
9530 			while (width++ < view->ncols)
9531 				waddch(view->window, ' ');
9532 		} else {
9533 			if (width < view->ncols)
9534 				waddch(view->window, '\n');
9535 		}
9536 		if (attr)
9537 			wattroff(view->window, attr);
9538 		if (++nprinted == 1)
9539 			s->first_displayed_line = s->lineno;
9540 	}
9541 	free(line);
9542 	if (nprinted > 0)
9543 		s->last_displayed_line = s->first_displayed_line + nprinted - 1;
9544 	else
9545 		s->last_displayed_line = s->first_displayed_line;
9546 
9547 	view_border(view);
9548 
9549 	if (s->eof) {
9550 		rc = waddnstr(view->window,
9551 		    "See the tog(1) manual page for full documentation",
9552 		    view->ncols - 1);
9553 		if (rc == ERR)
9554 			return got_error_msg(GOT_ERR_RANGE, "waddnstr");
9555 	} else {
9556 		wmove(view->window, view->nlines - 1, 0);
9557 		wclrtoeol(view->window);
9558 		wstandout(view->window);
9559 		rc = waddnstr(view->window, "scroll down for more...",
9560 		    view->ncols - 1);
9561 		if (rc == ERR)
9562 			return got_error_msg(GOT_ERR_RANGE, "waddnstr");
9563 		if (getcurx(view->window) < view->ncols - 6) {
9564 			rc = wprintw(view->window, "[%.0f%%]",
9565 			    100.00 * s->last_displayed_line / s->nlines);
9566 			if (rc == ERR)
9567 				return got_error_msg(GOT_ERR_IO, "wprintw");
9568 		}
9569 		wstandend(view->window);
9570 	}
9571 
9572 	return NULL;
9573 }
9574 
9575 static const struct got_error *
9576 input_help_view(struct tog_view **new_view, struct tog_view *view, int ch)
9577 {
9578 	struct tog_help_view_state	*s = &view->state.help;
9579 	const struct got_error		*err = NULL;
9580 	char				*line = NULL;
9581 	ssize_t				 linelen;
9582 	size_t				 linesz = 0;
9583 	int				 eos, nscroll;
9584 
9585 	eos = nscroll = view->nlines;
9586 	if (view_is_hsplit_top(view))
9587 		--eos;  /* border */
9588 
9589 	s->lineno = s->first_displayed_line - 1 + s->selected_line;
9590 
9591 	switch (ch) {
9592 	case '0':
9593 	case '$':
9594 	case KEY_RIGHT:
9595 	case 'l':
9596 	case KEY_LEFT:
9597 	case 'h':
9598 		horizontal_scroll_input(view, ch);
9599 		break;
9600 	case 'g':
9601 	case KEY_HOME:
9602 		s->first_displayed_line = 1;
9603 		view->count = 0;
9604 		break;
9605 	case 'G':
9606 	case KEY_END:
9607 		view->count = 0;
9608 		if (s->eof)
9609 			break;
9610 		s->first_displayed_line = (s->nlines - eos) + 3;
9611 		s->eof = 1;
9612 		break;
9613 	case 'k':
9614 	case KEY_UP:
9615 		if (s->first_displayed_line > 1)
9616 			--s->first_displayed_line;
9617 		else
9618 			view->count = 0;
9619 		break;
9620 	case CTRL('u'):
9621 	case 'u':
9622 		nscroll /= 2;
9623 		/* FALL THROUGH */
9624 	case KEY_PPAGE:
9625 	case CTRL('b'):
9626 	case 'b':
9627 		if (s->first_displayed_line == 1) {
9628 			view->count = 0;
9629 			break;
9630 		}
9631 		while (--nscroll > 0 && s->first_displayed_line > 1)
9632 			s->first_displayed_line--;
9633 		break;
9634 	case 'j':
9635 	case KEY_DOWN:
9636 	case CTRL('n'):
9637 		if (!s->eof)
9638 			++s->first_displayed_line;
9639 		else
9640 			view->count = 0;
9641 		break;
9642 	case CTRL('d'):
9643 	case 'd':
9644 		nscroll /= 2;
9645 		/* FALL THROUGH */
9646 	case KEY_NPAGE:
9647 	case CTRL('f'):
9648 	case 'f':
9649 	case ' ':
9650 		if (s->eof) {
9651 			view->count = 0;
9652 			break;
9653 		}
9654 		while (!s->eof && --nscroll > 0) {
9655 			linelen = getline(&line, &linesz, s->f);
9656 			s->first_displayed_line++;
9657 			if (linelen == -1) {
9658 				if (feof(s->f))
9659 					s->eof = 1;
9660 				else
9661 					err = got_ferror(s->f, GOT_ERR_IO);
9662 				break;
9663 			}
9664 		}
9665 		free(line);
9666 		break;
9667 	default:
9668 		view->count = 0;
9669 		break;
9670 	}
9671 
9672 	return err;
9673 }
9674 
9675 static const struct got_error *
9676 close_help_view(struct tog_view *view)
9677 {
9678 	struct tog_help_view_state *s = &view->state.help;
9679 
9680 	free(s->line_offsets);
9681 	s->line_offsets = NULL;
9682 	if (fclose(s->f) == EOF)
9683 		return got_error_from_errno("fclose");
9684 
9685 	return NULL;
9686 }
9687 
9688 static const struct got_error *
9689 reset_help_view(struct tog_view *view)
9690 {
9691 	struct tog_help_view_state *s = &view->state.help;
9692 
9693 
9694 	if (s->f && fclose(s->f) == EOF)
9695 		return got_error_from_errno("fclose");
9696 
9697 	wclear(view->window);
9698 	view->count = 0;
9699 	view->x = 0;
9700 	s->all = !s->all;
9701 	s->first_displayed_line = 1;
9702 	s->last_displayed_line = view->nlines;
9703 	s->matched_line = 0;
9704 
9705 	return create_help(s);
9706 }
9707 
9708 static const struct got_error *
9709 open_help_view(struct tog_view *view, struct tog_view *parent)
9710 {
9711 	const struct got_error		*err = NULL;
9712 	struct tog_help_view_state	*s = &view->state.help;
9713 
9714 	s->type = (enum tog_keymap_type)parent->type;
9715 	s->first_displayed_line = 1;
9716 	s->last_displayed_line = view->nlines;
9717 	s->selected_line = 1;
9718 
9719 	view->show = show_help_view;
9720 	view->input = input_help_view;
9721 	view->reset = reset_help_view;
9722 	view->close = close_help_view;
9723 	view->search_start = search_start_help_view;
9724 	view->search_setup = search_setup_help_view;
9725 	view->search_next = search_next_view_match;
9726 
9727 	err = create_help(s);
9728 	return err;
9729 }
9730 
9731 static const struct got_error *
9732 view_dispatch_request(struct tog_view **new_view, struct tog_view *view,
9733     enum tog_view_type request, int y, int x)
9734 {
9735 	const struct got_error *err = NULL;
9736 
9737 	*new_view = NULL;
9738 
9739 	switch (request) {
9740 	case TOG_VIEW_DIFF:
9741 		if (view->type == TOG_VIEW_LOG) {
9742 			struct tog_log_view_state *s = &view->state.log;
9743 
9744 			err = open_diff_view_for_commit(new_view, y, x,
9745 			    s->selected_entry->commit, s->selected_entry->id,
9746 			    view, s->repo);
9747 		} else
9748 			return got_error_msg(GOT_ERR_NOT_IMPL,
9749 			    "parent/child view pair not supported");
9750 		break;
9751 	case TOG_VIEW_BLAME:
9752 		if (view->type == TOG_VIEW_TREE) {
9753 			struct tog_tree_view_state *s = &view->state.tree;
9754 
9755 			err = blame_tree_entry(new_view, y, x,
9756 			    s->selected_entry, &s->parents, s->commit_id,
9757 			    s->repo);
9758 		} else
9759 			return got_error_msg(GOT_ERR_NOT_IMPL,
9760 			    "parent/child view pair not supported");
9761 		break;
9762 	case TOG_VIEW_LOG:
9763 		if (view->type == TOG_VIEW_BLAME)
9764 			err = log_annotated_line(new_view, y, x,
9765 			    view->state.blame.repo, view->state.blame.id_to_log);
9766 		else if (view->type == TOG_VIEW_TREE)
9767 			err = log_selected_tree_entry(new_view, y, x,
9768 			    &view->state.tree);
9769 		else if (view->type == TOG_VIEW_REF)
9770 			err = log_ref_entry(new_view, y, x,
9771 			    view->state.ref.selected_entry,
9772 			    view->state.ref.repo);
9773 		else
9774 			return got_error_msg(GOT_ERR_NOT_IMPL,
9775 			    "parent/child view pair not supported");
9776 		break;
9777 	case TOG_VIEW_TREE:
9778 		if (view->type == TOG_VIEW_LOG)
9779 			err = browse_commit_tree(new_view, y, x,
9780 			    view->state.log.selected_entry,
9781 			    view->state.log.in_repo_path,
9782 			    view->state.log.head_ref_name,
9783 			    view->state.log.repo);
9784 		else if (view->type == TOG_VIEW_REF)
9785 			err = browse_ref_tree(new_view, y, x,
9786 			    view->state.ref.selected_entry,
9787 			    view->state.ref.repo);
9788 		else
9789 			return got_error_msg(GOT_ERR_NOT_IMPL,
9790 			    "parent/child view pair not supported");
9791 		break;
9792 	case TOG_VIEW_REF:
9793 		*new_view = view_open(0, 0, y, x, TOG_VIEW_REF);
9794 		if (*new_view == NULL)
9795 			return got_error_from_errno("view_open");
9796 		if (view->type == TOG_VIEW_LOG)
9797 			err = open_ref_view(*new_view, view->state.log.repo);
9798 		else if (view->type == TOG_VIEW_TREE)
9799 			err = open_ref_view(*new_view, view->state.tree.repo);
9800 		else
9801 			err = got_error_msg(GOT_ERR_NOT_IMPL,
9802 			    "parent/child view pair not supported");
9803 		if (err)
9804 			view_close(*new_view);
9805 		break;
9806 	case TOG_VIEW_HELP:
9807 		*new_view = view_open(0, 0, 0, 0, TOG_VIEW_HELP);
9808 		if (*new_view == NULL)
9809 			return got_error_from_errno("view_open");
9810 		err = open_help_view(*new_view, view);
9811 		if (err)
9812 			view_close(*new_view);
9813 		break;
9814 	default:
9815 		return got_error_msg(GOT_ERR_NOT_IMPL, "invalid view");
9816 	}
9817 
9818 	return err;
9819 }
9820 
9821 /*
9822  * If view was scrolled down to move the selected line into view when opening a
9823  * horizontal split, scroll back up when closing the split/toggling fullscreen.
9824  */
9825 static void
9826 offset_selection_up(struct tog_view *view)
9827 {
9828 	switch (view->type) {
9829 	case TOG_VIEW_BLAME: {
9830 		struct tog_blame_view_state *s = &view->state.blame;
9831 		if (s->first_displayed_line == 1) {
9832 			s->selected_line = MAX(s->selected_line - view->offset,
9833 			    1);
9834 			break;
9835 		}
9836 		if (s->first_displayed_line > view->offset)
9837 			s->first_displayed_line -= view->offset;
9838 		else
9839 			s->first_displayed_line = 1;
9840 		s->selected_line += view->offset;
9841 		break;
9842 	}
9843 	case TOG_VIEW_LOG:
9844 		log_scroll_up(&view->state.log, view->offset);
9845 		view->state.log.selected += view->offset;
9846 		break;
9847 	case TOG_VIEW_REF:
9848 		ref_scroll_up(&view->state.ref, view->offset);
9849 		view->state.ref.selected += view->offset;
9850 		break;
9851 	case TOG_VIEW_TREE:
9852 		tree_scroll_up(&view->state.tree, view->offset);
9853 		view->state.tree.selected += view->offset;
9854 		break;
9855 	default:
9856 		break;
9857 	}
9858 
9859 	view->offset = 0;
9860 }
9861 
9862 /*
9863  * If the selected line is in the section of screen covered by the bottom split,
9864  * scroll down offset lines to move it into view and index its new position.
9865  */
9866 static const struct got_error *
9867 offset_selection_down(struct tog_view *view)
9868 {
9869 	const struct got_error	*err = NULL;
9870 	const struct got_error	*(*scrolld)(struct tog_view *, int);
9871 	int			*selected = NULL;
9872 	int			 header, offset;
9873 
9874 	switch (view->type) {
9875 	case TOG_VIEW_BLAME: {
9876 		struct tog_blame_view_state *s = &view->state.blame;
9877 		header = 3;
9878 		scrolld = NULL;
9879 		if (s->selected_line > view->nlines - header) {
9880 			offset = abs(view->nlines - s->selected_line - header);
9881 			s->first_displayed_line += offset;
9882 			s->selected_line -= offset;
9883 			view->offset = offset;
9884 		}
9885 		break;
9886 	}
9887 	case TOG_VIEW_LOG: {
9888 		struct tog_log_view_state *s = &view->state.log;
9889 		scrolld = &log_scroll_down;
9890 		header = view_is_parent_view(view) ? 3 : 2;
9891 		selected = &s->selected;
9892 		break;
9893 	}
9894 	case TOG_VIEW_REF: {
9895 		struct tog_ref_view_state *s = &view->state.ref;
9896 		scrolld = &ref_scroll_down;
9897 		header = 3;
9898 		selected = &s->selected;
9899 		break;
9900 	}
9901 	case TOG_VIEW_TREE: {
9902 		struct tog_tree_view_state *s = &view->state.tree;
9903 		scrolld = &tree_scroll_down;
9904 		header = 5;
9905 		selected = &s->selected;
9906 		break;
9907 	}
9908 	default:
9909 		selected = NULL;
9910 		scrolld = NULL;
9911 		header = 0;
9912 		break;
9913 	}
9914 
9915 	if (selected && *selected > view->nlines - header) {
9916 		offset = abs(view->nlines - *selected - header);
9917 		view->offset = offset;
9918 		if (scrolld && offset) {
9919 			err = scrolld(view, offset);
9920 			*selected -= offset;
9921 		}
9922 	}
9923 
9924 	return err;
9925 }
9926 
9927 static void
9928 list_commands(FILE *fp)
9929 {
9930 	size_t i;
9931 
9932 	fprintf(fp, "commands:");
9933 	for (i = 0; i < nitems(tog_commands); i++) {
9934 		const struct tog_cmd *cmd = &tog_commands[i];
9935 		fprintf(fp, " %s", cmd->name);
9936 	}
9937 	fputc('\n', fp);
9938 }
9939 
9940 __dead static void
9941 usage(int hflag, int status)
9942 {
9943 	FILE *fp = (status == 0) ? stdout : stderr;
9944 
9945 	fprintf(fp, "usage: %s [-hV] command [arg ...]\n",
9946 	    getprogname());
9947 	if (hflag) {
9948 		fprintf(fp, "lazy usage: %s path\n", getprogname());
9949 		list_commands(fp);
9950 	}
9951 	exit(status);
9952 }
9953 
9954 static char **
9955 make_argv(int argc, ...)
9956 {
9957 	va_list ap;
9958 	char **argv;
9959 	int i;
9960 
9961 	va_start(ap, argc);
9962 
9963 	argv = calloc(argc, sizeof(char *));
9964 	if (argv == NULL)
9965 		err(1, "calloc");
9966 	for (i = 0; i < argc; i++) {
9967 		argv[i] = strdup(va_arg(ap, char *));
9968 		if (argv[i] == NULL)
9969 			err(1, "strdup");
9970 	}
9971 
9972 	va_end(ap);
9973 	return argv;
9974 }
9975 
9976 /*
9977  * Try to convert 'tog path' into a 'tog log path' command.
9978  * The user could simply have mistyped the command rather than knowingly
9979  * provided a path. So check whether argv[0] can in fact be resolved
9980  * to a path in the HEAD commit and print a special error if not.
9981  * This hack is for mpi@ <3
9982  */
9983 static const struct got_error *
9984 tog_log_with_path(int argc, char *argv[])
9985 {
9986 	const struct got_error *error = NULL, *close_err;
9987 	const struct tog_cmd *cmd = NULL;
9988 	struct got_repository *repo = NULL;
9989 	struct got_worktree *worktree = NULL;
9990 	struct got_object_id *commit_id = NULL, *id = NULL;
9991 	struct got_commit_object *commit = NULL;
9992 	char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
9993 	char *commit_id_str = NULL, **cmd_argv = NULL;
9994 	int *pack_fds = NULL;
9995 
9996 	cwd = getcwd(NULL, 0);
9997 	if (cwd == NULL)
9998 		return got_error_from_errno("getcwd");
9999 
10000 	error = got_repo_pack_fds_open(&pack_fds);
10001 	if (error != NULL)
10002 		goto done;
10003 
10004 	error = got_worktree_open(&worktree, cwd, NULL);
10005 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
10006 		goto done;
10007 
10008 	if (worktree)
10009 		repo_path = strdup(got_worktree_get_repo_path(worktree));
10010 	else
10011 		repo_path = strdup(cwd);
10012 	if (repo_path == NULL) {
10013 		error = got_error_from_errno("strdup");
10014 		goto done;
10015 	}
10016 
10017 	error = got_repo_open(&repo, repo_path, NULL, pack_fds);
10018 	if (error != NULL)
10019 		goto done;
10020 
10021 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
10022 	    repo, worktree);
10023 	if (error)
10024 		goto done;
10025 
10026 	error = tog_load_refs(repo, 0);
10027 	if (error)
10028 		goto done;
10029 	error = got_repo_match_object_id(&commit_id, NULL, worktree ?
10030 	    got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
10031 	    GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
10032 	if (error)
10033 		goto done;
10034 
10035 	if (worktree) {
10036 		got_worktree_close(worktree);
10037 		worktree = NULL;
10038 	}
10039 
10040 	error = got_object_open_as_commit(&commit, repo, commit_id);
10041 	if (error)
10042 		goto done;
10043 
10044 	error = got_object_id_by_path(&id, repo, commit, in_repo_path);
10045 	if (error) {
10046 		if (error->code != GOT_ERR_NO_TREE_ENTRY)
10047 			goto done;
10048 		fprintf(stderr, "%s: '%s' is no known command or path\n",
10049 		    getprogname(), argv[0]);
10050 		usage(1, 1);
10051 		/* not reached */
10052 	}
10053 
10054 	error = got_object_id_str(&commit_id_str, commit_id);
10055 	if (error)
10056 		goto done;
10057 
10058 	cmd = &tog_commands[0]; /* log */
10059 	argc = 4;
10060 	cmd_argv = make_argv(argc, cmd->name, "-c", commit_id_str, argv[0]);
10061 	error = cmd->cmd_main(argc, cmd_argv);
10062 done:
10063 	if (repo) {
10064 		close_err = got_repo_close(repo);
10065 		if (error == NULL)
10066 			error = close_err;
10067 	}
10068 	if (commit)
10069 		got_object_commit_close(commit);
10070 	if (worktree)
10071 		got_worktree_close(worktree);
10072 	if (pack_fds) {
10073 		const struct got_error *pack_err =
10074 		    got_repo_pack_fds_close(pack_fds);
10075 		if (error == NULL)
10076 			error = pack_err;
10077 	}
10078 	free(id);
10079 	free(commit_id_str);
10080 	free(commit_id);
10081 	free(cwd);
10082 	free(repo_path);
10083 	free(in_repo_path);
10084 	if (cmd_argv) {
10085 		int i;
10086 		for (i = 0; i < argc; i++)
10087 			free(cmd_argv[i]);
10088 		free(cmd_argv);
10089 	}
10090 	tog_free_refs();
10091 	return error;
10092 }
10093 
10094 int
10095 main(int argc, char *argv[])
10096 {
10097 	const struct got_error *io_err, *error = NULL;
10098 	const struct tog_cmd *cmd = NULL;
10099 	int ch, hflag = 0, Vflag = 0;
10100 	char **cmd_argv = NULL;
10101 	static const struct option longopts[] = {
10102 	    { "version", no_argument, NULL, 'V' },
10103 	    { NULL, 0, NULL, 0}
10104 	};
10105 	char *diff_algo_str = NULL;
10106 	const char *test_script_path;
10107 
10108 	setlocale(LC_CTYPE, "");
10109 
10110 	/*
10111 	 * Override default signal handlers before starting ncurses.
10112 	 * This should prevent ncurses from installing its own
10113 	 * broken cleanup() signal handler.
10114 	 */
10115 	signal(SIGWINCH, tog_sigwinch);
10116 	signal(SIGPIPE, tog_sigpipe);
10117 	signal(SIGCONT, tog_sigcont);
10118 	signal(SIGINT, tog_sigint);
10119 	signal(SIGTERM, tog_sigterm);
10120 
10121 	/*
10122 	 * Test mode init must happen before pledge() because "tty" will
10123 	 * not allow TTY-related ioctls to occur via regular files.
10124 	 */
10125 	test_script_path = getenv("TOG_TEST_SCRIPT");
10126 	if (test_script_path != NULL) {
10127 		error = init_mock_term(test_script_path);
10128 		if (error) {
10129 			fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
10130 			return 1;
10131 		}
10132 	} else if (!isatty(STDIN_FILENO))
10133 		errx(1, "standard input is not a tty");
10134 
10135 #if !defined(PROFILE)
10136 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
10137 	    NULL) == -1)
10138 		err(1, "pledge");
10139 #endif
10140 
10141 	while ((ch = getopt_long(argc, argv, "+hV", longopts, NULL)) != -1) {
10142 		switch (ch) {
10143 		case 'h':
10144 			hflag = 1;
10145 			break;
10146 		case 'V':
10147 			Vflag = 1;
10148 			break;
10149 		default:
10150 			usage(hflag, 1);
10151 			/* NOTREACHED */
10152 		}
10153 	}
10154 
10155 	argc -= optind;
10156 	argv += optind;
10157 	optind = 1;
10158 	optreset = 1;
10159 
10160 	if (Vflag) {
10161 		got_version_print_str();
10162 		return 0;
10163 	}
10164 
10165 	if (argc == 0) {
10166 		if (hflag)
10167 			usage(hflag, 0);
10168 		/* Build an argument vector which runs a default command. */
10169 		cmd = &tog_commands[0];
10170 		argc = 1;
10171 		cmd_argv = make_argv(argc, cmd->name);
10172 	} else {
10173 		size_t i;
10174 
10175 		/* Did the user specify a command? */
10176 		for (i = 0; i < nitems(tog_commands); i++) {
10177 			if (strncmp(tog_commands[i].name, argv[0],
10178 			    strlen(argv[0])) == 0) {
10179 				cmd = &tog_commands[i];
10180 				break;
10181 			}
10182 		}
10183 	}
10184 
10185 	diff_algo_str = getenv("TOG_DIFF_ALGORITHM");
10186 	if (diff_algo_str) {
10187 		if (strcasecmp(diff_algo_str, "patience") == 0)
10188 			tog_diff_algo = GOT_DIFF_ALGORITHM_PATIENCE;
10189 		if (strcasecmp(diff_algo_str, "myers") == 0)
10190 			tog_diff_algo = GOT_DIFF_ALGORITHM_MYERS;
10191 	}
10192 
10193 	tog_base_commit.idx = -1;
10194 	tog_base_commit.marker = GOT_WORKTREE_STATE_UNKNOWN;
10195 
10196 	if (cmd == NULL) {
10197 		if (argc != 1)
10198 			usage(0, 1);
10199 		/* No command specified; try log with a path */
10200 		error = tog_log_with_path(argc, argv);
10201 	} else {
10202 		if (hflag)
10203 			cmd->cmd_usage();
10204 		else
10205 			error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
10206 	}
10207 
10208 	if (using_mock_io) {
10209 		io_err = tog_io_close();
10210 		if (error == NULL)
10211 			error = io_err;
10212 	}
10213 	endwin();
10214 	if (cmd_argv) {
10215 		int i;
10216 		for (i = 0; i < argc; i++)
10217 			free(cmd_argv[i]);
10218 		free(cmd_argv);
10219 	}
10220 
10221 	if (error && error->code != GOT_ERR_CANCELLED &&
10222 	    error->code != GOT_ERR_EOF &&
10223 	    error->code != GOT_ERR_PRIVSEP_EXIT &&
10224 	    error->code != GOT_ERR_PRIVSEP_PIPE &&
10225 	    !(error->code == GOT_ERR_ERRNO && errno == EINTR)) {
10226 		fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
10227 		return 1;
10228 	}
10229 	return 0;
10230 }