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
24 #include <curses.h>
25 #undef _XOPEN_SOURCE_EXTENDED
26 #include <panel.h>
27 #include <locale.h>
28 #include <signal.h>
29 #include <stdlib.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <getopt.h>
33 #include <string.h>
34 #include <err.h>
35 #include <unistd.h>
36 #include <util.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 
44 #include "got_version.h"
45 #include "got_error.h"
46 #include "got_object.h"
47 #include "got_reference.h"
48 #include "got_repository.h"
49 #include "got_diff.h"
50 #include "got_opentemp.h"
51 #include "got_utf8.h"
52 #include "got_cancel.h"
53 #include "got_commit_graph.h"
54 #include "got_blame.h"
55 #include "got_privsep.h"
56 #include "got_path.h"
57 #include "got_worktree.h"
58 
59 #ifndef MIN
60 #define	MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
61 #endif
62 
63 #ifndef MAX
64 #define	MAX(_a,_b) ((_a) > (_b) ? (_a) : (_b))
65 #endif
66 
67 #define CTRL(x)		((x) & 0x1f)
68 
69 #ifndef nitems
70 #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
71 #endif
72 
73 struct tog_cmd {
74 	const char *name;
75 	const struct got_error *(*cmd_main)(int, char *[]);
76 	void (*cmd_usage)(void);
77 };
78 
79 __dead static void	usage(int, int);
80 __dead static void	usage_log(void);
81 __dead static void	usage_diff(void);
82 __dead static void	usage_blame(void);
83 __dead static void	usage_tree(void);
84 __dead static void	usage_ref(void);
85 
86 static const struct got_error*	cmd_log(int, char *[]);
87 static const struct got_error*	cmd_diff(int, char *[]);
88 static const struct got_error*	cmd_blame(int, char *[]);
89 static const struct got_error*	cmd_tree(int, char *[]);
90 static const struct got_error*	cmd_ref(int, char *[]);
91 
92 static struct tog_cmd tog_commands[] = {
93 	{ "log",	cmd_log,	usage_log },
94 	{ "diff",	cmd_diff,	usage_diff },
95 	{ "blame",	cmd_blame,	usage_blame },
96 	{ "tree",	cmd_tree,	usage_tree },
97 	{ "ref",	cmd_ref,	usage_ref },
98 };
99 
100 enum tog_view_type {
101 	TOG_VIEW_DIFF,
102 	TOG_VIEW_LOG,
103 	TOG_VIEW_BLAME,
104 	TOG_VIEW_TREE,
105 	TOG_VIEW_REF,
106 };
107 
108 #define TOG_EOF_STRING	"(END)"
109 
110 struct commit_queue_entry {
111 	TAILQ_ENTRY(commit_queue_entry) entry;
112 	struct got_object_id *id;
113 	struct got_commit_object *commit;
114 	int idx;
115 };
116 TAILQ_HEAD(commit_queue_head, commit_queue_entry);
117 struct commit_queue {
118 	int ncommits;
119 	struct commit_queue_head head;
120 };
121 
122 struct tog_color {
123 	SIMPLEQ_ENTRY(tog_color) entry;
124 	regex_t regex;
125 	short colorpair;
126 };
127 SIMPLEQ_HEAD(tog_colors, tog_color);
128 
129 static const struct got_error *
130 add_color(struct tog_colors *colors, const char *pattern,
131     int idx, short color)
132 {
133 	const struct got_error *err = NULL;
134 	struct tog_color *tc;
135 	int regerr = 0;
136 
137 	if (idx < 1 || idx > COLOR_PAIRS - 1)
138 		return NULL;
139 
140 	init_pair(idx, color, -1);
141 
142 	tc = calloc(1, sizeof(*tc));
143 	if (tc == NULL)
144 		return got_error_from_errno("calloc");
145 	regerr = regcomp(&tc->regex, pattern,
146 	    REG_EXTENDED | REG_NOSUB | REG_NEWLINE);
147 	if (regerr) {
148 		static char regerr_msg[512];
149 		static char err_msg[512];
150 		regerror(regerr, &tc->regex, regerr_msg,
151 		    sizeof(regerr_msg));
152 		snprintf(err_msg, sizeof(err_msg), "regcomp: %s",
153 		    regerr_msg);
154 		err = got_error_msg(GOT_ERR_REGEX, err_msg);
155 		free(tc);
156 		return err;
157 	}
158 	tc->colorpair = idx;
159 	SIMPLEQ_INSERT_HEAD(colors, tc, entry);
160 	return NULL;
161 }
162 
163 static void
164 free_colors(struct tog_colors *colors)
165 {
166 	struct tog_color *tc;
167 
168 	while (!SIMPLEQ_EMPTY(colors)) {
169 		tc = SIMPLEQ_FIRST(colors);
170 		SIMPLEQ_REMOVE_HEAD(colors, entry);
171 		regfree(&tc->regex);
172 		free(tc);
173 	}
174 }
175 
176 struct tog_color *
177 get_color(struct tog_colors *colors, int colorpair)
178 {
179 	struct tog_color *tc = NULL;
180 
181 	SIMPLEQ_FOREACH(tc, colors, entry) {
182 		if (tc->colorpair == colorpair)
183 			return tc;
184 	}
185 
186 	return NULL;
187 }
188 
189 static int
190 default_color_value(const char *envvar)
191 {
192 	if (strcmp(envvar, "TOG_COLOR_DIFF_MINUS") == 0)
193 		return COLOR_MAGENTA;
194 	if (strcmp(envvar, "TOG_COLOR_DIFF_PLUS") == 0)
195 		return COLOR_CYAN;
196 	if (strcmp(envvar, "TOG_COLOR_DIFF_CHUNK_HEADER") == 0)
197 		return COLOR_YELLOW;
198 	if (strcmp(envvar, "TOG_COLOR_DIFF_META") == 0)
199 		return COLOR_GREEN;
200 	if (strcmp(envvar, "TOG_COLOR_TREE_SUBMODULE") == 0)
201 		return COLOR_MAGENTA;
202 	if (strcmp(envvar, "TOG_COLOR_TREE_SYMLINK") == 0)
203 		return COLOR_MAGENTA;
204 	if (strcmp(envvar, "TOG_COLOR_TREE_DIRECTORY") == 0)
205 		return COLOR_CYAN;
206 	if (strcmp(envvar, "TOG_COLOR_TREE_EXECUTABLE") == 0)
207 		return COLOR_GREEN;
208 	if (strcmp(envvar, "TOG_COLOR_COMMIT") == 0)
209 		return COLOR_GREEN;
210 	if (strcmp(envvar, "TOG_COLOR_AUTHOR") == 0)
211 		return COLOR_CYAN;
212 	if (strcmp(envvar, "TOG_COLOR_DATE") == 0)
213 		return COLOR_YELLOW;
214 	if (strcmp(envvar, "TOG_COLOR_REFS_HEADS") == 0)
215 		return COLOR_GREEN;
216 	if (strcmp(envvar, "TOG_COLOR_REFS_TAGS") == 0)
217 		return COLOR_MAGENTA;
218 	if (strcmp(envvar, "TOG_COLOR_REFS_REMOTES") == 0)
219 		return COLOR_YELLOW;
220 
221 	return -1;
222 }
223 
224 static int
225 get_color_value(const char *envvar)
226 {
227 	const char *val = getenv(envvar);
228 
229 	if (val == NULL)
230 		return default_color_value(envvar);
231 
232 	if (strcasecmp(val, "black") == 0)
233 		return COLOR_BLACK;
234 	if (strcasecmp(val, "red") == 0)
235 		return COLOR_RED;
236 	if (strcasecmp(val, "green") == 0)
237 		return COLOR_GREEN;
238 	if (strcasecmp(val, "yellow") == 0)
239 		return COLOR_YELLOW;
240 	if (strcasecmp(val, "blue") == 0)
241 		return COLOR_BLUE;
242 	if (strcasecmp(val, "magenta") == 0)
243 		return COLOR_MAGENTA;
244 	if (strcasecmp(val, "cyan") == 0)
245 		return COLOR_CYAN;
246 	if (strcasecmp(val, "white") == 0)
247 		return COLOR_WHITE;
248 	if (strcasecmp(val, "default") == 0)
249 		return -1;
250 
251 	return default_color_value(envvar);
252 }
253 
254 
255 struct tog_diff_view_state {
256 	struct got_object_id *id1, *id2;
257 	const char *label1, *label2;
258 	FILE *f;
259 	int first_displayed_line;
260 	int last_displayed_line;
261 	int eof;
262 	int diff_context;
263 	int ignore_whitespace;
264 	int force_text_diff;
265 	struct got_repository *repo;
266 	struct got_reflist_head refs;
267 	struct tog_colors colors;
268 	size_t nlines;
269 	off_t *line_offsets;
270 	int matched_line;
271 	int selected_line;
272 
273 	/* passed from log view; may be NULL */
274 	struct tog_view *log_view;
275 };
276 
277 pthread_mutex_t tog_mutex = PTHREAD_MUTEX_INITIALIZER;
278 
279 struct tog_log_thread_args {
280 	pthread_cond_t need_commits;
281 	pthread_cond_t commit_loaded;
282 	int commits_needed;
283 	struct got_commit_graph *graph;
284 	struct commit_queue *commits;
285 	const char *in_repo_path;
286 	struct got_object_id *start_id;
287 	struct got_repository *repo;
288 	int log_complete;
289 	sig_atomic_t *quit;
290 	struct commit_queue_entry **first_displayed_entry;
291 	struct commit_queue_entry **selected_entry;
292 	int *searching;
293 	int *search_next_done;
294 	regex_t *regex;
295 };
296 
297 struct tog_log_view_state {
298 	struct commit_queue commits;
299 	struct commit_queue_entry *first_displayed_entry;
300 	struct commit_queue_entry *last_displayed_entry;
301 	struct commit_queue_entry *selected_entry;
302 	int selected;
303 	char *in_repo_path;
304 	const char *head_ref_name;
305 	int log_branches;
306 	struct got_repository *repo;
307 	struct got_reflist_head refs;
308 	struct got_object_id *start_id;
309 	sig_atomic_t quit;
310 	pthread_t thread;
311 	struct tog_log_thread_args thread_args;
312 	struct commit_queue_entry *matched_entry;
313 	struct commit_queue_entry *search_entry;
314 	struct tog_colors colors;
315 };
316 
317 #define TOG_COLOR_DIFF_MINUS		1
318 #define TOG_COLOR_DIFF_PLUS		2
319 #define TOG_COLOR_DIFF_CHUNK_HEADER	3
320 #define TOG_COLOR_DIFF_META		4
321 #define TOG_COLOR_TREE_SUBMODULE	5
322 #define TOG_COLOR_TREE_SYMLINK		6
323 #define TOG_COLOR_TREE_DIRECTORY	7
324 #define TOG_COLOR_TREE_EXECUTABLE	8
325 #define TOG_COLOR_COMMIT		9
326 #define TOG_COLOR_AUTHOR		10
327 #define TOG_COLOR_DATE		11
328 #define TOG_COLOR_REFS_HEADS		12
329 #define TOG_COLOR_REFS_TAGS		13
330 #define TOG_COLOR_REFS_REMOTES		14
331 
332 struct tog_blame_cb_args {
333 	struct tog_blame_line *lines; /* one per line */
334 	int nlines;
335 
336 	struct tog_view *view;
337 	struct got_object_id *commit_id;
338 	int *quit;
339 };
340 
341 struct tog_blame_thread_args {
342 	const char *path;
343 	struct got_repository *repo;
344 	struct tog_blame_cb_args *cb_args;
345 	int *complete;
346 	got_cancel_cb cancel_cb;
347 	void *cancel_arg;
348 };
349 
350 struct tog_blame {
351 	FILE *f;
352 	off_t filesize;
353 	struct tog_blame_line *lines;
354 	int nlines;
355 	off_t *line_offsets;
356 	pthread_t thread;
357 	struct tog_blame_thread_args thread_args;
358 	struct tog_blame_cb_args cb_args;
359 	const char *path;
360 };
361 
362 struct tog_blame_view_state {
363 	int first_displayed_line;
364 	int last_displayed_line;
365 	int selected_line;
366 	int blame_complete;
367 	int eof;
368 	int done;
369 	struct got_object_id_queue blamed_commits;
370 	struct got_object_qid *blamed_commit;
371 	char *path;
372 	struct got_repository *repo;
373 	struct got_object_id *commit_id;
374 	struct tog_blame blame;
375 	int matched_line;
376 	struct tog_colors colors;
377 };
378 
379 struct tog_parent_tree {
380 	TAILQ_ENTRY(tog_parent_tree) entry;
381 	struct got_tree_object *tree;
382 	struct got_tree_entry *first_displayed_entry;
383 	struct got_tree_entry *selected_entry;
384 	int selected;
385 };
386 
387 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
388 
389 struct tog_tree_view_state {
390 	char *tree_label;
391 	struct got_tree_object *root;
392 	struct got_tree_object *tree;
393 	struct got_tree_entry *first_displayed_entry;
394 	struct got_tree_entry *last_displayed_entry;
395 	struct got_tree_entry *selected_entry;
396 	int ndisplayed, selected, show_ids;
397 	struct tog_parent_trees parents;
398 	struct got_object_id *commit_id;
399 	struct got_repository *repo;
400 	struct got_tree_entry *matched_entry;
401 	struct tog_colors colors;
402 };
403 
404 struct tog_reflist_entry {
405 	TAILQ_ENTRY(tog_reflist_entry) entry;
406 	struct got_reference *ref;
407 	int idx;
408 };
409 
410 TAILQ_HEAD(tog_reflist_head, tog_reflist_entry);
411 
412 struct tog_ref_view_state {
413 	struct got_reflist_head simplerefs; /* SIMPLEQ */
414 	struct tog_reflist_head refs;	/* TAILQ */
415 	struct tog_reflist_entry *first_displayed_entry;
416 	struct tog_reflist_entry *last_displayed_entry;
417 	struct tog_reflist_entry *selected_entry;
418 	int nrefs, ndisplayed, selected, show_ids;
419 	struct got_repository *repo;
420 	struct tog_reflist_entry *matched_entry;
421 	struct tog_colors colors;
422 };
423 
424 /*
425  * We implement two types of views: parent views and child views.
426  *
427  * The 'Tab' key switches between a parent view and its child view.
428  * Child views are shown side-by-side to their parent view, provided
429  * there is enough screen estate.
430  *
431  * When a new view is opened from within a parent view, this new view
432  * becomes a child view of the parent view, replacing any existing child.
433  *
434  * When a new view is opened from within a child view, this new view
435  * becomes a parent view which will obscure the views below until the
436  * user quits the new parent view by typing 'q'.
437  *
438  * This list of views contains parent views only.
439  * Child views are only pointed to by their parent view.
440  */
441 TAILQ_HEAD(tog_view_list_head, tog_view);
442 
443 struct tog_view {
444 	TAILQ_ENTRY(tog_view) entry;
445 	WINDOW *window;
446 	PANEL *panel;
447 	int nlines, ncols, begin_y, begin_x;
448 	int lines, cols; /* copies of LINES and COLS */
449 	int focussed;
450 	struct tog_view *parent;
451 	struct tog_view *child;
452 	int child_focussed;
453 
454 	/* type-specific state */
455 	enum tog_view_type type;
456 	union {
457 		struct tog_diff_view_state diff;
458 		struct tog_log_view_state log;
459 		struct tog_blame_view_state blame;
460 		struct tog_tree_view_state tree;
461 		struct tog_ref_view_state ref;
462 	} state;
463 
464 	const struct got_error *(*show)(struct tog_view *);
465 	const struct got_error *(*input)(struct tog_view **,
466 	    struct tog_view **, struct tog_view**, struct tog_view *, int);
467 	const struct got_error *(*close)(struct tog_view *);
468 
469 	const struct got_error *(*search_start)(struct tog_view *);
470 	const struct got_error *(*search_next)(struct tog_view *);
471 	int searching;
472 #define TOG_SEARCH_FORWARD	1
473 #define TOG_SEARCH_BACKWARD	2
474 	int search_next_done;
475 #define TOG_SEARCH_HAVE_MORE	1
476 #define TOG_SEARCH_NO_MORE	2
477 #define TOG_SEARCH_HAVE_NONE	3
478 	regex_t regex;
479 	regmatch_t regmatch;
480 };
481 
482 static const struct got_error *open_diff_view(struct tog_view *,
483     struct got_object_id *, struct got_object_id *,
484     const char *, const char *, int, int, int, struct tog_view *,
485     struct got_repository *);
486 static const struct got_error *show_diff_view(struct tog_view *);
487 static const struct got_error *input_diff_view(struct tog_view **,
488     struct tog_view **, struct tog_view **, struct tog_view *, int);
489 static const struct got_error* close_diff_view(struct tog_view *);
490 static const struct got_error *search_start_diff_view(struct tog_view *);
491 static const struct got_error *search_next_diff_view(struct tog_view *);
492 
493 static const struct got_error *open_log_view(struct tog_view *,
494     struct got_object_id *, struct got_repository *,
495     const char *, const char *, int);
496 static const struct got_error * show_log_view(struct tog_view *);
497 static const struct got_error *input_log_view(struct tog_view **,
498     struct tog_view **, struct tog_view **, struct tog_view *, int);
499 static const struct got_error *close_log_view(struct tog_view *);
500 static const struct got_error *search_start_log_view(struct tog_view *);
501 static const struct got_error *search_next_log_view(struct tog_view *);
502 
503 static const struct got_error *open_blame_view(struct tog_view *, char *,
504     struct got_object_id *, struct got_repository *);
505 static const struct got_error *show_blame_view(struct tog_view *);
506 static const struct got_error *input_blame_view(struct tog_view **,
507     struct tog_view **, struct tog_view **, struct tog_view *, int);
508 static const struct got_error *close_blame_view(struct tog_view *);
509 static const struct got_error *search_start_blame_view(struct tog_view *);
510 static const struct got_error *search_next_blame_view(struct tog_view *);
511 
512 static const struct got_error *open_tree_view(struct tog_view *,
513     struct got_tree_object *, struct got_object_id *, struct got_repository *);
514 static const struct got_error *show_tree_view(struct tog_view *);
515 static const struct got_error *input_tree_view(struct tog_view **,
516     struct tog_view **, struct tog_view **, struct tog_view *, int);
517 static const struct got_error *close_tree_view(struct tog_view *);
518 static const struct got_error *search_start_tree_view(struct tog_view *);
519 static const struct got_error *search_next_tree_view(struct tog_view *);
520 
521 static const struct got_error *open_ref_view(struct tog_view *,
522     struct got_repository *);
523 static const struct got_error *show_ref_view(struct tog_view *);
524 static const struct got_error *input_ref_view(struct tog_view **,
525     struct tog_view **, struct tog_view **, struct tog_view *, int);
526 static const struct got_error *close_ref_view(struct tog_view *);
527 static const struct got_error *search_start_ref_view(struct tog_view *);
528 static const struct got_error *search_next_ref_view(struct tog_view *);
529 
530 static volatile sig_atomic_t tog_sigwinch_received;
531 static volatile sig_atomic_t tog_sigpipe_received;
532 static volatile sig_atomic_t tog_sigcont_received;
533 
534 static void
535 tog_sigwinch(int signo)
536 {
537 	tog_sigwinch_received = 1;
538 }
539 
540 static void
541 tog_sigpipe(int signo)
542 {
543 	tog_sigpipe_received = 1;
544 }
545 
546 static void
547 tog_sigcont(int signo)
548 {
549 	tog_sigcont_received = 1;
550 }
551 
552 static const struct got_error *
553 view_close(struct tog_view *view)
554 {
555 	const struct got_error *err = NULL;
556 
557 	if (view->child) {
558 		view_close(view->child);
559 		view->child = NULL;
560 	}
561 	if (view->close)
562 		err = view->close(view);
563 	if (view->panel)
564 		del_panel(view->panel);
565 	if (view->window)
566 		delwin(view->window);
567 	free(view);
568 	return err;
569 }
570 
571 static struct tog_view *
572 view_open(int nlines, int ncols, int begin_y, int begin_x,
573     enum tog_view_type type)
574 {
575 	struct tog_view *view = calloc(1, sizeof(*view));
576 
577 	if (view == NULL)
578 		return NULL;
579 
580 	view->type = type;
581 	view->lines = LINES;
582 	view->cols = COLS;
583 	view->nlines = nlines ? nlines : LINES - begin_y;
584 	view->ncols = ncols ? ncols : COLS - begin_x;
585 	view->begin_y = begin_y;
586 	view->begin_x = begin_x;
587 	view->window = newwin(nlines, ncols, begin_y, begin_x);
588 	if (view->window == NULL) {
589 		view_close(view);
590 		return NULL;
591 	}
592 	view->panel = new_panel(view->window);
593 	if (view->panel == NULL ||
594 	    set_panel_userptr(view->panel, view) != OK) {
595 		view_close(view);
596 		return NULL;
597 	}
598 
599 	keypad(view->window, TRUE);
600 	return view;
601 }
602 
603 static int
604 view_split_begin_x(int begin_x)
605 {
606 	if (begin_x > 0 || COLS < 120)
607 		return 0;
608 	return (COLS - MAX(COLS / 2, 80));
609 }
610 
611 static const struct got_error *view_resize(struct tog_view *);
612 
613 static const struct got_error *
614 view_splitscreen(struct tog_view *view)
615 {
616 	const struct got_error *err = NULL;
617 
618 	view->begin_y = 0;
619 	view->begin_x = view_split_begin_x(0);
620 	view->nlines = LINES;
621 	view->ncols = COLS - view->begin_x;
622 	view->lines = LINES;
623 	view->cols = COLS;
624 	err = view_resize(view);
625 	if (err)
626 		return err;
627 
628 	if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
629 		return got_error_from_errno("mvwin");
630 
631 	return NULL;
632 }
633 
634 static const struct got_error *
635 view_fullscreen(struct tog_view *view)
636 {
637 	const struct got_error *err = NULL;
638 
639 	view->begin_x = 0;
640 	view->begin_y = 0;
641 	view->nlines = LINES;
642 	view->ncols = COLS;
643 	view->lines = LINES;
644 	view->cols = COLS;
645 	err = view_resize(view);
646 	if (err)
647 		return err;
648 
649 	if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
650 		return got_error_from_errno("mvwin");
651 
652 	return NULL;
653 }
654 
655 static int
656 view_is_parent_view(struct tog_view *view)
657 {
658 	return view->parent == NULL;
659 }
660 
661 static const struct got_error *
662 view_resize(struct tog_view *view)
663 {
664 	int nlines, ncols;
665 
666 	if (view->lines > LINES)
667 		nlines = view->nlines - (view->lines - LINES);
668 	else
669 		nlines = view->nlines + (LINES - view->lines);
670 
671 	if (view->cols > COLS)
672 		ncols = view->ncols - (view->cols - COLS);
673 	else
674 		ncols = view->ncols + (COLS - view->cols);
675 
676 	if (wresize(view->window, nlines, ncols) == ERR)
677 		return got_error_from_errno("wresize");
678 	if (replace_panel(view->panel, view->window) == ERR)
679 		return got_error_from_errno("replace_panel");
680 	wclear(view->window);
681 
682 	view->nlines = nlines;
683 	view->ncols = ncols;
684 	view->lines = LINES;
685 	view->cols = COLS;
686 
687 	if (view->child) {
688 		view->child->begin_x = view_split_begin_x(view->begin_x);
689 		if (view->child->begin_x == 0) {
690 			view_fullscreen(view->child);
691 			if (view->child->focussed)
692 				show_panel(view->child->panel);
693 			else
694 				show_panel(view->panel);
695 		} else {
696 			view_splitscreen(view->child);
697 			show_panel(view->child->panel);
698 		}
699 	}
700 
701 	return NULL;
702 }
703 
704 static const struct got_error *
705 view_close_child(struct tog_view *view)
706 {
707 	const struct got_error *err = NULL;
708 
709 	if (view->child == NULL)
710 		return NULL;
711 
712 	err = view_close(view->child);
713 	view->child = NULL;
714 	return err;
715 }
716 
717 static const struct got_error *
718 view_set_child(struct tog_view *view, struct tog_view *child)
719 {
720 	const struct got_error *err = NULL;
721 
722 	view->child = child;
723 	child->parent = view;
724 	return err;
725 }
726 
727 static int
728 view_is_splitscreen(struct tog_view *view)
729 {
730 	return view->begin_x > 0;
731 }
732 
733 static void
734 tog_resizeterm(void)
735 {
736 	int cols, lines;
737 	struct winsize size;
738 
739 	if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &size) < 0) {
740 		cols = 80;     /* Default */
741 		lines = 24;
742 	} else {
743 		cols = size.ws_col;
744 		lines = size.ws_row;
745 	}
746 	resize_term(lines, cols);
747 }
748 
749 static const struct got_error *
750 view_search_start(struct tog_view *view)
751 {
752 	const struct got_error *err = NULL;
753 	char pattern[1024];
754 	int ret;
755 
756 	if (view->nlines < 1)
757 		return NULL;
758 
759 	mvwaddstr(view->window, view->begin_y + view->nlines - 1, 0, "/");
760 	wclrtoeol(view->window);
761 
762 	nocbreak();
763 	echo();
764 	ret = wgetnstr(view->window, pattern, sizeof(pattern));
765 	cbreak();
766 	noecho();
767 	if (ret == ERR)
768 		return NULL;
769 
770 	if (view->searching) {
771 		regfree(&view->regex);
772 		view->searching = 0;
773 	}
774 
775 	if (regcomp(&view->regex, pattern, REG_EXTENDED | REG_NEWLINE) == 0) {
776 		err = view->search_start(view);
777 		if (err) {
778 			regfree(&view->regex);
779 			return err;
780 		}
781 		view->searching = TOG_SEARCH_FORWARD;
782 		view->search_next_done = 0;
783 		view->search_next(view);
784 	}
785 
786 	return NULL;
787 }
788 
789 static const struct got_error *
790 view_input(struct tog_view **new, struct tog_view **dead,
791     struct tog_view **focus, int *done, struct tog_view *view,
792     struct tog_view_list_head *views)
793 {
794 	const struct got_error *err = NULL;
795 	struct tog_view *v;
796 	int ch, errcode;
797 
798 	*new = NULL;
799 	*dead = NULL;
800 	*focus = NULL;
801 
802 	/* Clear "no matches" indicator. */
803 	if (view->search_next_done == TOG_SEARCH_NO_MORE ||
804 	    view->search_next_done == TOG_SEARCH_HAVE_NONE)
805 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
806 
807 	if (view->searching && !view->search_next_done) {
808 		errcode = pthread_mutex_unlock(&tog_mutex);
809 		if (errcode)
810 			return got_error_set_errno(errcode,
811 			    "pthread_mutex_unlock");
812 		pthread_yield();
813 		errcode = pthread_mutex_lock(&tog_mutex);
814 		if (errcode)
815 			return got_error_set_errno(errcode,
816 			    "pthread_mutex_lock");
817 		view->search_next(view);
818 		return NULL;
819 	}
820 
821 	nodelay(stdscr, FALSE);
822 	/* Allow threads to make progress while we are waiting for input. */
823 	errcode = pthread_mutex_unlock(&tog_mutex);
824 	if (errcode)
825 		return got_error_set_errno(errcode, "pthread_mutex_unlock");
826 	ch = wgetch(view->window);
827 	errcode = pthread_mutex_lock(&tog_mutex);
828 	if (errcode)
829 		return got_error_set_errno(errcode, "pthread_mutex_lock");
830 	nodelay(stdscr, TRUE);
831 
832 	if (tog_sigwinch_received || tog_sigcont_received) {
833 		tog_resizeterm();
834 		tog_sigwinch_received = 0;
835 		tog_sigcont_received = 0;
836 		TAILQ_FOREACH(v, views, entry) {
837 			err = view_resize(v);
838 			if (err)
839 				return err;
840 			err = v->input(new, dead, focus, v, KEY_RESIZE);
841 			if (err)
842 				return err;
843 		}
844 	}
845 
846 	switch (ch) {
847 	case ERR:
848 		break;
849 	case '\t':
850 		if (view->child) {
851 			*focus = view->child;
852 			view->child_focussed = 1;
853 		} else if (view->parent) {
854 			*focus = view->parent;
855 			view->parent->child_focussed = 0;
856 		}
857 		break;
858 	case 'q':
859 		err = view->input(new, dead, focus, view, ch);
860 		*dead = view;
861 		break;
862 	case 'Q':
863 		*done = 1;
864 		break;
865 	case 'f':
866 		if (view_is_parent_view(view)) {
867 			if (view->child == NULL)
868 				break;
869 			if (view_is_splitscreen(view->child)) {
870 				*focus = view->child;
871 				view->child_focussed = 1;
872 				err = view_fullscreen(view->child);
873 			} else
874 				err = view_splitscreen(view->child);
875 			if (err)
876 				break;
877 			err = view->child->input(new, dead, focus,
878 			    view->child, KEY_RESIZE);
879 		} else {
880 			if (view_is_splitscreen(view)) {
881 				*focus = view;
882 				view->parent->child_focussed = 1;
883 				err = view_fullscreen(view);
884 			} else {
885 				err = view_splitscreen(view);
886 			}
887 			if (err)
888 				break;
889 			err = view->input(new, dead, focus, view,
890 			    KEY_RESIZE);
891 		}
892 		break;
893 	case KEY_RESIZE:
894 		break;
895 	case '/':
896 		if (view->search_start)
897 			view_search_start(view);
898 		else
899 			err = view->input(new, dead, focus, view, ch);
900 		break;
901 	case 'N':
902 	case 'n':
903 		if (view->search_next) {
904 			view->searching = (ch == 'n' ?
905 			    TOG_SEARCH_FORWARD : TOG_SEARCH_BACKWARD);
906 			view->search_next_done = 0;
907 			view->search_next(view);
908 		} else
909 			err = view->input(new, dead, focus, view, ch);
910 		break;
911 	default:
912 		err = view->input(new, dead, focus, view, ch);
913 		break;
914 	}
915 
916 	return err;
917 }
918 
919 void
920 view_vborder(struct tog_view *view)
921 {
922 	PANEL *panel;
923 	struct tog_view *view_above;
924 
925 	if (view->parent)
926 		return view_vborder(view->parent);
927 
928 	panel = panel_above(view->panel);
929 	if (panel == NULL)
930 		return;
931 
932 	view_above = panel_userptr(panel);
933 	mvwvline(view->window, view->begin_y, view_above->begin_x - 1,
934 	    got_locale_is_utf8() ? ACS_VLINE : '|', view->nlines);
935 }
936 
937 int
938 view_needs_focus_indication(struct tog_view *view)
939 {
940 	if (view_is_parent_view(view)) {
941 		if (view->child == NULL || view->child_focussed)
942 			return 0;
943 		if (!view_is_splitscreen(view->child))
944 			return 0;
945 	} else if (!view_is_splitscreen(view))
946 		return 0;
947 
948 	return view->focussed;
949 }
950 
951 static const struct got_error *
952 view_loop(struct tog_view *view)
953 {
954 	const struct got_error *err = NULL;
955 	struct tog_view_list_head views;
956 	struct tog_view *new_view, *dead_view, *focus_view, *main_view;
957 	int fast_refresh = 10;
958 	int done = 0, errcode;
959 
960 	errcode = pthread_mutex_lock(&tog_mutex);
961 	if (errcode)
962 		return got_error_set_errno(errcode, "pthread_mutex_lock");
963 
964 	TAILQ_INIT(&views);
965 	TAILQ_INSERT_HEAD(&views, view, entry);
966 
967 	main_view = view;
968 	view->focussed = 1;
969 	err = view->show(view);
970 	if (err)
971 		return err;
972 	update_panels();
973 	doupdate();
974 	while (!TAILQ_EMPTY(&views) && !done && !tog_sigpipe_received) {
975 		/* Refresh fast during initialization, then become slower. */
976 		if (fast_refresh && fast_refresh-- == 0)
977 			halfdelay(10); /* switch to once per second */
978 
979 		err = view_input(&new_view, &dead_view, &focus_view, &done,
980 		    view, &views);
981 		if (err)
982 			break;
983 		if (dead_view) {
984 			struct tog_view *prev = NULL;
985 
986 			if (view_is_parent_view(dead_view))
987 				prev = TAILQ_PREV(dead_view,
988 				    tog_view_list_head, entry);
989 			else if (view->parent != dead_view)
990 				prev = view->parent;
991 
992 			if (dead_view->parent)
993 				dead_view->parent->child = NULL;
994 			else
995 				TAILQ_REMOVE(&views, dead_view, entry);
996 
997 			err = view_close(dead_view);
998 			if (err || (dead_view == main_view && new_view == NULL))
999 				goto done;
1000 
1001 			if (view == dead_view) {
1002 				if (focus_view)
1003 					view = focus_view;
1004 				else if (prev)
1005 					view = prev;
1006 				else if (!TAILQ_EMPTY(&views))
1007 					view = TAILQ_LAST(&views,
1008 					    tog_view_list_head);
1009 				else
1010 					view = NULL;
1011 				if (view) {
1012 					if (view->child && view->child_focussed)
1013 						focus_view = view->child;
1014 					else
1015 						focus_view = view;
1016 				}
1017 			}
1018 		}
1019 		if (new_view) {
1020 			struct tog_view *v, *t;
1021 			/* Only allow one parent view per type. */
1022 			TAILQ_FOREACH_SAFE(v, &views, entry, t) {
1023 				if (v->type != new_view->type)
1024 					continue;
1025 				TAILQ_REMOVE(&views, v, entry);
1026 				err = view_close(v);
1027 				if (err)
1028 					goto done;
1029 				break;
1030 			}
1031 			TAILQ_INSERT_TAIL(&views, new_view, entry);
1032 			view = new_view;
1033 			if (focus_view == NULL)
1034 				focus_view = new_view;
1035 		}
1036 		if (focus_view) {
1037 			show_panel(focus_view->panel);
1038 			if (view)
1039 				view->focussed = 0;
1040 			focus_view->focussed = 1;
1041 			view = focus_view;
1042 			if (new_view)
1043 				show_panel(new_view->panel);
1044 			if (view->child && view_is_splitscreen(view->child))
1045 				show_panel(view->child->panel);
1046 		}
1047 		if (view) {
1048 			if (focus_view == NULL) {
1049 				view->focussed = 1;
1050 				show_panel(view->panel);
1051 				if (view->child && view_is_splitscreen(view->child))
1052 					show_panel(view->child->panel);
1053 				focus_view = view;
1054 			}
1055 			if (view->parent) {
1056 				err = view->parent->show(view->parent);
1057 				if (err)
1058 					goto done;
1059 			}
1060 			err = view->show(view);
1061 			if (err)
1062 				goto done;
1063 			if (view->child) {
1064 				err = view->child->show(view->child);
1065 				if (err)
1066 					goto done;
1067 			}
1068 			update_panels();
1069 			doupdate();
1070 		}
1071 	}
1072 done:
1073 	while (!TAILQ_EMPTY(&views)) {
1074 		view = TAILQ_FIRST(&views);
1075 		TAILQ_REMOVE(&views, view, entry);
1076 		view_close(view);
1077 	}
1078 
1079 	errcode = pthread_mutex_unlock(&tog_mutex);
1080 	if (errcode && err == NULL)
1081 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
1082 
1083 	return err;
1084 }
1085 
1086 __dead static void
1087 usage_log(void)
1088 {
1089 	endwin();
1090 	fprintf(stderr,
1091 	    "usage: %s log [-b] [-c commit] [-r repository-path] [path]\n",
1092 	    getprogname());
1093 	exit(1);
1094 }
1095 
1096 /* Create newly allocated wide-character string equivalent to a byte string. */
1097 static const struct got_error *
1098 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
1099 {
1100 	char *vis = NULL;
1101 	const struct got_error *err = NULL;
1102 
1103 	*ws = NULL;
1104 	*wlen = mbstowcs(NULL, s, 0);
1105 	if (*wlen == (size_t)-1) {
1106 		int vislen;
1107 		if (errno != EILSEQ)
1108 			return got_error_from_errno("mbstowcs");
1109 
1110 		/* byte string invalid in current encoding; try to "fix" it */
1111 		err = got_mbsavis(&vis, &vislen, s);
1112 		if (err)
1113 			return err;
1114 		*wlen = mbstowcs(NULL, vis, 0);
1115 		if (*wlen == (size_t)-1) {
1116 			err = got_error_from_errno("mbstowcs"); /* give up */
1117 			goto done;
1118 		}
1119 	}
1120 
1121 	*ws = calloc(*wlen + 1, sizeof(**ws));
1122 	if (*ws == NULL) {
1123 		err = got_error_from_errno("calloc");
1124 		goto done;
1125 	}
1126 
1127 	if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
1128 		err = got_error_from_errno("mbstowcs");
1129 done:
1130 	free(vis);
1131 	if (err) {
1132 		free(*ws);
1133 		*ws = NULL;
1134 		*wlen = 0;
1135 	}
1136 	return err;
1137 }
1138 
1139 /* Format a line for display, ensuring that it won't overflow a width limit. */
1140 static const struct got_error *
1141 format_line(wchar_t **wlinep, int *widthp, const char *line, int wlimit,
1142     int col_tab_align)
1143 {
1144 	const struct got_error *err = NULL;
1145 	int cols = 0;
1146 	wchar_t *wline = NULL;
1147 	size_t wlen;
1148 	int i;
1149 
1150 	*wlinep = NULL;
1151 	*widthp = 0;
1152 
1153 	err = mbs2ws(&wline, &wlen, line);
1154 	if (err)
1155 		return err;
1156 
1157 	i = 0;
1158 	while (i < wlen) {
1159 		int width = wcwidth(wline[i]);
1160 
1161 		if (width == 0) {
1162 			i++;
1163 			continue;
1164 		}
1165 
1166 		if (width == 1 || width == 2) {
1167 			if (cols + width > wlimit)
1168 				break;
1169 			cols += width;
1170 			i++;
1171 		} else if (width == -1) {
1172 			if (wline[i] == L'\t') {
1173 				width = TABSIZE -
1174 				    ((cols + col_tab_align) % TABSIZE);
1175 				if (cols + width > wlimit)
1176 					break;
1177 				cols += width;
1178 			}
1179 			i++;
1180 		} else {
1181 			err = got_error_from_errno("wcwidth");
1182 			goto done;
1183 		}
1184 	}
1185 	wline[i] = L'\0';
1186 	if (widthp)
1187 		*widthp = cols;
1188 done:
1189 	if (err)
1190 		free(wline);
1191 	else
1192 		*wlinep = wline;
1193 	return err;
1194 }
1195 
1196 static const struct got_error*
1197 build_refs_str(char **refs_str, struct got_reflist_head *refs,
1198     struct got_object_id *id, struct got_repository *repo)
1199 {
1200 	static const struct got_error *err = NULL;
1201 	struct got_reflist_entry *re;
1202 	char *s;
1203 	const char *name;
1204 
1205 	*refs_str = NULL;
1206 
1207 	SIMPLEQ_FOREACH(re, refs, entry) {
1208 		struct got_tag_object *tag = NULL;
1209 		struct got_object_id *ref_id;
1210 		int cmp;
1211 
1212 		name = got_ref_get_name(re->ref);
1213 		if (strcmp(name, GOT_REF_HEAD) == 0)
1214 			continue;
1215 		if (strncmp(name, "refs/", 5) == 0)
1216 			name += 5;
1217 		if (strncmp(name, "got/", 4) == 0)
1218 			continue;
1219 		if (strncmp(name, "heads/", 6) == 0)
1220 			name += 6;
1221 		if (strncmp(name, "remotes/", 8) == 0) {
1222 			name += 8;
1223 			s = strstr(name, "/" GOT_REF_HEAD);
1224 			if (s != NULL && s[strlen(s)] == '\0')
1225 				continue;
1226 		}
1227 		err = got_ref_resolve(&ref_id, repo, re->ref);
1228 		if (err)
1229 			break;
1230 		if (strncmp(name, "tags/", 5) == 0) {
1231 			err = got_object_open_as_tag(&tag, repo, ref_id);
1232 			if (err) {
1233 				if (err->code != GOT_ERR_OBJ_TYPE) {
1234 					free(ref_id);
1235 					break;
1236 				}
1237 				/* Ref points at something other than a tag. */
1238 				err = NULL;
1239 				tag = NULL;
1240 			}
1241 		}
1242 		cmp = got_object_id_cmp(tag ?
1243 		    got_object_tag_get_object_id(tag) : ref_id, id);
1244 		free(ref_id);
1245 		if (tag)
1246 			got_object_tag_close(tag);
1247 		if (cmp != 0)
1248 			continue;
1249 		s = *refs_str;
1250 		if (asprintf(refs_str, "%s%s%s", s ? s : "",
1251 		    s ? ", " : "", name) == -1) {
1252 			err = got_error_from_errno("asprintf");
1253 			free(s);
1254 			*refs_str = NULL;
1255 			break;
1256 		}
1257 		free(s);
1258 	}
1259 
1260 	return err;
1261 }
1262 
1263 static const struct got_error *
1264 format_author(wchar_t **wauthor, int *author_width, char *author, int limit,
1265     int col_tab_align)
1266 {
1267 	char *smallerthan, *at;
1268 
1269 	smallerthan = strchr(author, '<');
1270 	if (smallerthan && smallerthan[1] != '\0')
1271 		author = smallerthan + 1;
1272 	at = strchr(author, '@');
1273 	if (at)
1274 		*at = '\0';
1275 	return format_line(wauthor, author_width, author, limit, col_tab_align);
1276 }
1277 
1278 static const struct got_error *
1279 draw_commit(struct tog_view *view, struct got_commit_object *commit,
1280     struct got_object_id *id, struct got_reflist_head *refs,
1281     const size_t date_display_cols, int author_display_cols,
1282     struct tog_colors *colors)
1283 {
1284 	const struct got_error *err = NULL;
1285 	char datebuf[12]; /* YYYY-MM-DD + SPACE + NUL */
1286 	char *logmsg0 = NULL, *logmsg = NULL;
1287 	char *author = NULL;
1288 	wchar_t *wlogmsg = NULL, *wauthor = NULL;
1289 	int author_width, logmsg_width;
1290 	char *newline, *line = NULL;
1291 	int col, limit;
1292 	const int avail = view->ncols;
1293 	struct tm tm;
1294 	time_t committer_time;
1295 	struct tog_color *tc;
1296 
1297 	committer_time = got_object_commit_get_committer_time(commit);
1298 	if (localtime_r(&committer_time, &tm) == NULL)
1299 		return got_error_from_errno("localtime_r");
1300 	if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d ", &tm)
1301 	    >= sizeof(datebuf))
1302 		return got_error(GOT_ERR_NO_SPACE);
1303 
1304 	if (avail <= date_display_cols)
1305 		limit = MIN(sizeof(datebuf) - 1, avail);
1306 	else
1307 		limit = MIN(date_display_cols, sizeof(datebuf) - 1);
1308 	tc = get_color(colors, TOG_COLOR_DATE);
1309 	if (tc)
1310 		wattr_on(view->window,
1311 		    COLOR_PAIR(tc->colorpair), NULL);
1312 	waddnstr(view->window, datebuf, limit);
1313 	if (tc)
1314 		wattr_off(view->window,
1315 		    COLOR_PAIR(tc->colorpair), NULL);
1316 	col = limit;
1317 	if (col > avail)
1318 		goto done;
1319 
1320 	if (avail >= 120) {
1321 		char *id_str;
1322 		err = got_object_id_str(&id_str, id);
1323 		if (err)
1324 			goto done;
1325 		tc = get_color(colors, TOG_COLOR_COMMIT);
1326 		if (tc)
1327 			wattr_on(view->window,
1328 			    COLOR_PAIR(tc->colorpair), NULL);
1329 		wprintw(view->window, "%.8s ", id_str);
1330 		if (tc)
1331 			wattr_off(view->window,
1332 			    COLOR_PAIR(tc->colorpair), NULL);
1333 		free(id_str);
1334 		col += 9;
1335 		if (col > avail)
1336 			goto done;
1337 	}
1338 
1339 	author = strdup(got_object_commit_get_author(commit));
1340 	if (author == NULL) {
1341 		err = got_error_from_errno("strdup");
1342 		goto done;
1343 	}
1344 	err = format_author(&wauthor, &author_width, author, avail - col, col);
1345 	if (err)
1346 		goto done;
1347 	tc = get_color(colors, TOG_COLOR_AUTHOR);
1348 	if (tc)
1349 		wattr_on(view->window,
1350 		    COLOR_PAIR(tc->colorpair), NULL);
1351 	waddwstr(view->window, wauthor);
1352 	if (tc)
1353 		wattr_off(view->window,
1354 		    COLOR_PAIR(tc->colorpair), NULL);
1355 	col += author_width;
1356 	while (col < avail && author_width < author_display_cols + 2) {
1357 		waddch(view->window, ' ');
1358 		col++;
1359 		author_width++;
1360 	}
1361 	if (col > avail)
1362 		goto done;
1363 
1364 	err = got_object_commit_get_logmsg(&logmsg0, commit);
1365 	if (err)
1366 		goto done;
1367 	logmsg = logmsg0;
1368 	while (*logmsg == '\n')
1369 		logmsg++;
1370 	newline = strchr(logmsg, '\n');
1371 	if (newline)
1372 		*newline = '\0';
1373 	limit = avail - col;
1374 	err = format_line(&wlogmsg, &logmsg_width, logmsg, limit, col);
1375 	if (err)
1376 		goto done;
1377 	waddwstr(view->window, wlogmsg);
1378 	col += logmsg_width;
1379 	while (col < avail) {
1380 		waddch(view->window, ' ');
1381 		col++;
1382 	}
1383 done:
1384 	free(logmsg0);
1385 	free(wlogmsg);
1386 	free(author);
1387 	free(wauthor);
1388 	free(line);
1389 	return err;
1390 }
1391 
1392 static struct commit_queue_entry *
1393 alloc_commit_queue_entry(struct got_commit_object *commit,
1394     struct got_object_id *id)
1395 {
1396 	struct commit_queue_entry *entry;
1397 
1398 	entry = calloc(1, sizeof(*entry));
1399 	if (entry == NULL)
1400 		return NULL;
1401 
1402 	entry->id = id;
1403 	entry->commit = commit;
1404 	return entry;
1405 }
1406 
1407 static void
1408 pop_commit(struct commit_queue *commits)
1409 {
1410 	struct commit_queue_entry *entry;
1411 
1412 	entry = TAILQ_FIRST(&commits->head);
1413 	TAILQ_REMOVE(&commits->head, entry, entry);
1414 	got_object_commit_close(entry->commit);
1415 	commits->ncommits--;
1416 	/* Don't free entry->id! It is owned by the commit graph. */
1417 	free(entry);
1418 }
1419 
1420 static void
1421 free_commits(struct commit_queue *commits)
1422 {
1423 	while (!TAILQ_EMPTY(&commits->head))
1424 		pop_commit(commits);
1425 }
1426 
1427 static const struct got_error *
1428 match_commit(int *have_match, struct got_object_id *id,
1429     struct got_commit_object *commit, regex_t *regex)
1430 {
1431 	const struct got_error *err = NULL;
1432 	regmatch_t regmatch;
1433 	char *id_str = NULL, *logmsg = NULL;
1434 
1435 	*have_match = 0;
1436 
1437 	err = got_object_id_str(&id_str, id);
1438 	if (err)
1439 		return err;
1440 
1441 	err = got_object_commit_get_logmsg(&logmsg, commit);
1442 	if (err)
1443 		goto done;
1444 
1445 	if (regexec(regex, got_object_commit_get_author(commit), 1,
1446 	    &regmatch, 0) == 0 ||
1447 	    regexec(regex, got_object_commit_get_committer(commit), 1,
1448 	    &regmatch, 0) == 0 ||
1449 	    regexec(regex, id_str, 1, &regmatch, 0) == 0 ||
1450 	    regexec(regex, logmsg, 1, &regmatch, 0) == 0)
1451 		*have_match = 1;
1452 done:
1453 	free(id_str);
1454 	free(logmsg);
1455 	return err;
1456 }
1457 
1458 static const struct got_error *
1459 queue_commits(struct got_commit_graph *graph, struct commit_queue *commits,
1460     int minqueue, struct got_repository *repo, const char *path,
1461     int *searching, int *search_next_done, regex_t *regex)
1462 {
1463 	const struct got_error *err = NULL;
1464 	int nqueued = 0;
1465 
1466 	/*
1467 	 * We keep all commits open throughout the lifetime of the log
1468 	 * view in order to avoid having to re-fetch commits from disk
1469 	 * while updating the display.
1470 	 */
1471 	while (nqueued < minqueue ||
1472 	    (*searching == TOG_SEARCH_FORWARD && !*search_next_done)) {
1473 		struct got_object_id *id;
1474 		struct got_commit_object *commit;
1475 		struct commit_queue_entry *entry;
1476 		int errcode;
1477 
1478 		err = got_commit_graph_iter_next(&id, graph, repo, NULL, NULL);
1479 		if (err || id == NULL)
1480 			break;
1481 
1482 		err = got_object_open_as_commit(&commit, repo, id);
1483 		if (err)
1484 			break;
1485 		entry = alloc_commit_queue_entry(commit, id);
1486 		if (entry == NULL) {
1487 			err = got_error_from_errno("alloc_commit_queue_entry");
1488 			break;
1489 		}
1490 
1491 		errcode = pthread_mutex_lock(&tog_mutex);
1492 		if (errcode) {
1493 			err = got_error_set_errno(errcode,
1494 			    "pthread_mutex_lock");
1495 			break;
1496 		}
1497 
1498 		entry->idx = commits->ncommits;
1499 		TAILQ_INSERT_TAIL(&commits->head, entry, entry);
1500 		nqueued++;
1501 		commits->ncommits++;
1502 
1503 		if (*searching == TOG_SEARCH_FORWARD && !*search_next_done) {
1504 			int have_match;
1505 			err = match_commit(&have_match, id, commit, regex);
1506 			if (err)
1507 				break;
1508 			if (have_match)
1509 				*search_next_done = TOG_SEARCH_HAVE_MORE;
1510 		}
1511 
1512 		errcode = pthread_mutex_unlock(&tog_mutex);
1513 		if (errcode && err == NULL)
1514 			err = got_error_set_errno(errcode,
1515 			    "pthread_mutex_unlock");
1516 		if (err)
1517 			break;
1518 	}
1519 
1520 	return err;
1521 }
1522 
1523 static const struct got_error *
1524 draw_commits(struct tog_view *view, struct commit_queue_entry **last,
1525     struct commit_queue_entry **selected, struct commit_queue_entry *first,
1526     struct commit_queue *commits, int selected_idx, int limit,
1527     struct got_reflist_head *refs, const char *path, int commits_needed,
1528     struct tog_colors *colors)
1529 {
1530 	const struct got_error *err = NULL;
1531 	struct tog_log_view_state *s = &view->state.log;
1532 	struct commit_queue_entry *entry;
1533 	int width;
1534 	int ncommits, author_cols = 4;
1535 	char *id_str = NULL, *header = NULL, *ncommits_str = NULL;
1536 	char *refs_str = NULL;
1537 	wchar_t *wline;
1538 	struct tog_color *tc;
1539 	static const size_t date_display_cols = 12;
1540 
1541 	entry = first;
1542 	ncommits = 0;
1543 	while (entry) {
1544 		if (ncommits == selected_idx) {
1545 			*selected = entry;
1546 			break;
1547 		}
1548 		entry = TAILQ_NEXT(entry, entry);
1549 		ncommits++;
1550 	}
1551 
1552 	if (*selected && !(view->searching && view->search_next_done == 0)) {
1553 		err = got_object_id_str(&id_str, (*selected)->id);
1554 		if (err)
1555 			return err;
1556 		if (refs) {
1557 			err = build_refs_str(&refs_str, refs, (*selected)->id,
1558 			    s->repo);
1559 			if (err)
1560 				goto done;
1561 		}
1562 	}
1563 
1564 	if (commits_needed == 0)
1565 		halfdelay(10); /* disable fast refresh */
1566 
1567 	if (commits_needed > 0) {
1568 		if (asprintf(&ncommits_str, " [%d/%d] %s",
1569 		    entry ? entry->idx + 1 : 0, commits->ncommits,
1570 		    (view->searching && !view->search_next_done) ?
1571 		    "searching..." : "loading...") == -1) {
1572 			err = got_error_from_errno("asprintf");
1573 			goto done;
1574 		}
1575 	} else {
1576 		const char *search_str = NULL;
1577 
1578 		if (view->searching) {
1579 			if (view->search_next_done == TOG_SEARCH_NO_MORE)
1580 				search_str = "no more matches";
1581 			else if (view->search_next_done == TOG_SEARCH_HAVE_NONE)
1582 				search_str = "no matches found";
1583 			else if (!view->search_next_done)
1584 				search_str = "searching...";
1585 		}
1586 
1587 		if (asprintf(&ncommits_str, " [%d/%d] %s",
1588 		    entry ? entry->idx + 1 : 0, commits->ncommits,
1589 		    search_str ? search_str :
1590 		    (refs_str ? refs_str : "")) == -1) {
1591 			err = got_error_from_errno("asprintf");
1592 			goto done;
1593 		}
1594 	}
1595 
1596 	if (path && strcmp(path, "/") != 0) {
1597 		if (asprintf(&header, "commit %s %s%s",
1598 		    id_str ? id_str : "........................................",
1599 		    path, ncommits_str) == -1) {
1600 			err = got_error_from_errno("asprintf");
1601 			header = NULL;
1602 			goto done;
1603 		}
1604 	} else if (asprintf(&header, "commit %s%s",
1605 	    id_str ? id_str : "........................................",
1606 	    ncommits_str) == -1) {
1607 		err = got_error_from_errno("asprintf");
1608 		header = NULL;
1609 		goto done;
1610 	}
1611 	err = format_line(&wline, &width, header, view->ncols, 0);
1612 	if (err)
1613 		goto done;
1614 
1615 	werase(view->window);
1616 
1617 	if (view_needs_focus_indication(view))
1618 		wstandout(view->window);
1619 	tc = get_color(colors, TOG_COLOR_COMMIT);
1620 	if (tc)
1621 		wattr_on(view->window,
1622 		    COLOR_PAIR(tc->colorpair), NULL);
1623 	waddwstr(view->window, wline);
1624 	if (tc)
1625 		wattr_off(view->window,
1626 		    COLOR_PAIR(tc->colorpair), NULL);
1627 	while (width < view->ncols) {
1628 		waddch(view->window, ' ');
1629 		width++;
1630 	}
1631 	if (view_needs_focus_indication(view))
1632 		wstandend(view->window);
1633 	free(wline);
1634 	if (limit <= 1)
1635 		goto done;
1636 
1637 	/* Grow author column size if necessary. */
1638 	entry = first;
1639 	ncommits = 0;
1640 	while (entry) {
1641 		char *author;
1642 		wchar_t *wauthor;
1643 		int width;
1644 		if (ncommits >= limit - 1)
1645 			break;
1646 		author = strdup(got_object_commit_get_author(entry->commit));
1647 		if (author == NULL) {
1648 			err = got_error_from_errno("strdup");
1649 			goto done;
1650 		}
1651 		err = format_author(&wauthor, &width, author, COLS,
1652 		    date_display_cols);
1653 		if (author_cols < width)
1654 			author_cols = width;
1655 		free(wauthor);
1656 		free(author);
1657 		ncommits++;
1658 		entry = TAILQ_NEXT(entry, entry);
1659 	}
1660 
1661 	entry = first;
1662 	*last = first;
1663 	ncommits = 0;
1664 	while (entry) {
1665 		if (ncommits >= limit - 1)
1666 			break;
1667 		if (ncommits == selected_idx)
1668 			wstandout(view->window);
1669 		err = draw_commit(view, entry->commit, entry->id, refs,
1670 		    date_display_cols, author_cols, colors);
1671 		if (ncommits == selected_idx)
1672 			wstandend(view->window);
1673 		if (err)
1674 			goto done;
1675 		ncommits++;
1676 		*last = entry;
1677 		entry = TAILQ_NEXT(entry, entry);
1678 	}
1679 
1680 	view_vborder(view);
1681 done:
1682 	free(id_str);
1683 	free(refs_str);
1684 	free(ncommits_str);
1685 	free(header);
1686 	return err;
1687 }
1688 
1689 static void
1690 scroll_up(struct tog_view *view,
1691     struct commit_queue_entry **first_displayed_entry, int maxscroll,
1692     struct commit_queue *commits)
1693 {
1694 	struct commit_queue_entry *entry;
1695 	int nscrolled = 0;
1696 
1697 	entry = TAILQ_FIRST(&commits->head);
1698 	if (*first_displayed_entry == entry)
1699 		return;
1700 
1701 	entry = *first_displayed_entry;
1702 	while (entry && nscrolled < maxscroll) {
1703 		entry = TAILQ_PREV(entry, commit_queue_head, entry);
1704 		if (entry) {
1705 			*first_displayed_entry = entry;
1706 			nscrolled++;
1707 		}
1708 	}
1709 }
1710 
1711 static const struct got_error *
1712 trigger_log_thread(struct tog_view *log_view, int wait,
1713     int *commits_needed, int *log_complete,
1714     pthread_cond_t *need_commits, pthread_cond_t *commit_loaded)
1715 {
1716 	int errcode;
1717 
1718 	halfdelay(1); /* fast refresh while loading commits */
1719 
1720 	while (*commits_needed > 0) {
1721 		if (*log_complete)
1722 			break;
1723 
1724 		/* Wake the log thread. */
1725 		errcode = pthread_cond_signal(need_commits);
1726 		if (errcode)
1727 			return got_error_set_errno(errcode,
1728 			    "pthread_cond_signal");
1729 
1730 		/*
1731 		 * The mutex will be released while the view loop waits
1732 		 * in wgetch(), at which time the log thread will run.
1733 		 */
1734 		if (!wait)
1735 			break;
1736 
1737 		/* Display progress update in log view. */
1738 		show_log_view(log_view);
1739 		update_panels();
1740 		doupdate();
1741 
1742 		/* Wait right here while next commit is being loaded. */
1743 		errcode = pthread_cond_wait(commit_loaded, &tog_mutex);
1744 		if (errcode)
1745 			return got_error_set_errno(errcode,
1746 			    "pthread_cond_wait");
1747 
1748 		/* Display progress update in log view. */
1749 		show_log_view(log_view);
1750 		update_panels();
1751 		doupdate();
1752 	}
1753 
1754 	return NULL;
1755 }
1756 
1757 static const struct got_error *
1758 scroll_down(struct tog_view *log_view,
1759     struct commit_queue_entry **first_displayed_entry, int maxscroll,
1760     struct commit_queue_entry **last_displayed_entry,
1761     struct commit_queue *commits, int *log_complete, int *commits_needed,
1762     pthread_cond_t *need_commits, pthread_cond_t *commit_loaded)
1763 {
1764 	const struct got_error *err = NULL;
1765 	struct commit_queue_entry *pentry;
1766 	int nscrolled = 0, ncommits_needed;
1767 
1768 	if (*last_displayed_entry == NULL)
1769 		return NULL;
1770 
1771 	ncommits_needed = (*last_displayed_entry)->idx + 1 + maxscroll;
1772 	if (commits->ncommits < ncommits_needed && !*log_complete) {
1773 		/*
1774 		 * Ask the log thread for required amount of commits.
1775 		 */
1776 		(*commits_needed) += maxscroll;
1777 		err = trigger_log_thread(log_view, 1, commits_needed,
1778 		    log_complete, need_commits, commit_loaded);
1779 		if (err)
1780 			return err;
1781 	}
1782 
1783 	do {
1784 		pentry = TAILQ_NEXT(*last_displayed_entry, entry);
1785 		if (pentry == NULL)
1786 			break;
1787 
1788 		*last_displayed_entry = pentry;
1789 
1790 		pentry = TAILQ_NEXT(*first_displayed_entry, entry);
1791 		if (pentry == NULL)
1792 			break;
1793 		*first_displayed_entry = pentry;
1794 	} while (++nscrolled < maxscroll);
1795 
1796 	return err;
1797 }
1798 
1799 static const struct got_error *
1800 open_diff_view_for_commit(struct tog_view **new_view, int begin_x,
1801     struct got_commit_object *commit, struct got_object_id *commit_id,
1802     struct tog_view *log_view, struct got_repository *repo)
1803 {
1804 	const struct got_error *err;
1805 	struct got_object_qid *parent_id;
1806 	struct tog_view *diff_view;
1807 
1808 	diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
1809 	if (diff_view == NULL)
1810 		return got_error_from_errno("view_open");
1811 
1812 	parent_id = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
1813 	err = open_diff_view(diff_view, parent_id ? parent_id->id : NULL,
1814 	    commit_id, NULL, NULL, 3, 0, 0, log_view, repo);
1815 	if (err == NULL)
1816 		*new_view = diff_view;
1817 	return err;
1818 }
1819 
1820 static const struct got_error *
1821 tree_view_visit_subtree(struct got_tree_object *subtree,
1822     struct tog_tree_view_state *s)
1823 {
1824 	struct tog_parent_tree *parent;
1825 
1826 	parent = calloc(1, sizeof(*parent));
1827 	if (parent == NULL)
1828 		return got_error_from_errno("calloc");
1829 
1830 	parent->tree = s->tree;
1831 	parent->first_displayed_entry = s->first_displayed_entry;
1832 	parent->selected_entry = s->selected_entry;
1833 	parent->selected = s->selected;
1834 	TAILQ_INSERT_HEAD(&s->parents, parent, entry);
1835 	s->tree = subtree;
1836 	s->selected = 0;
1837 	s->first_displayed_entry = NULL;
1838 	return NULL;
1839 }
1840 
1841 static const struct got_error *
1842 tree_view_walk_path(struct tog_tree_view_state *s,
1843     struct got_object_id *commit_id,
1844     const char *path, struct got_repository *repo)
1845 {
1846 	const struct got_error *err = NULL;
1847 	struct got_tree_object *tree = NULL;
1848 	const char *p;
1849 	char *slash, *subpath = NULL;
1850 
1851 	/* Walk the path and open corresponding tree objects. */
1852 	p = path;
1853 	while (*p) {
1854 		struct got_tree_entry *te;
1855 		struct got_object_id *tree_id;
1856 		char *te_name;
1857 
1858 		while (p[0] == '/')
1859 			p++;
1860 
1861 		/* Ensure the correct subtree entry is selected. */
1862 		slash = strchr(p, '/');
1863 		if (slash == NULL)
1864 			te_name = strdup(p);
1865 		else
1866 			te_name = strndup(p, slash - p);
1867 		if (te_name == NULL) {
1868 			err = got_error_from_errno("strndup");
1869 			break;
1870 		}
1871 		te = got_object_tree_find_entry(s->tree, te_name);
1872 		if (te == NULL) {
1873 			err = got_error_path(te_name, GOT_ERR_NO_TREE_ENTRY);
1874 			free(te_name);
1875 			break;
1876 		}
1877 		free(te_name);
1878 		s->first_displayed_entry = s->selected_entry = te;
1879 
1880 		if (!S_ISDIR(got_tree_entry_get_mode(s->selected_entry)))
1881 			break; /* jump to this file's entry */
1882 
1883 		slash = strchr(p, '/');
1884 		if (slash)
1885 			subpath = strndup(path, slash - path);
1886 		else
1887 			subpath = strdup(path);
1888 		if (subpath == NULL) {
1889 			err = got_error_from_errno("strdup");
1890 			break;
1891 		}
1892 
1893 		err = got_object_id_by_path(&tree_id, repo, commit_id,
1894 		    subpath);
1895 		if (err)
1896 			break;
1897 
1898 		err = got_object_open_as_tree(&tree, repo, tree_id);
1899 		free(tree_id);
1900 		if (err)
1901 			break;
1902 
1903 		err = tree_view_visit_subtree(tree, s);
1904 		if (err) {
1905 			got_object_tree_close(tree);
1906 			break;
1907 		}
1908 		if (slash == NULL)
1909 			break;
1910 		free(subpath);
1911 		subpath = NULL;
1912 		p = slash;
1913 	}
1914 
1915 	free(subpath);
1916 	return err;
1917 }
1918 
1919 static const struct got_error *
1920 browse_commit_tree(struct tog_view **new_view, int begin_x,
1921     struct commit_queue_entry *entry, const char *path,
1922     struct got_repository *repo)
1923 {
1924 	const struct got_error *err = NULL;
1925 	struct got_tree_object *tree;
1926 	struct tog_tree_view_state *s;
1927 	struct tog_view *tree_view;
1928 
1929 	err = got_object_open_as_tree(&tree, repo,
1930 	    got_object_commit_get_tree_id(entry->commit));
1931 	if (err)
1932 		return err;
1933 
1934 	tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
1935 	if (tree_view == NULL)
1936 		return got_error_from_errno("view_open");
1937 
1938 	err = open_tree_view(tree_view, tree, entry->id, repo);
1939 	if (err) {
1940 		got_object_tree_close(tree);
1941 		return err;
1942 	}
1943 	s = &tree_view->state.tree;
1944 
1945 	*new_view = tree_view;
1946 
1947 	if (got_path_is_root_dir(path))
1948 		return NULL;
1949 
1950 	return tree_view_walk_path(s, entry->id, path, repo);
1951 }
1952 
1953 static const struct got_error *
1954 block_signals_used_by_main_thread(void)
1955 {
1956 	sigset_t sigset;
1957 	int errcode;
1958 
1959 	if (sigemptyset(&sigset) == -1)
1960 		return got_error_from_errno("sigemptyset");
1961 
1962 	/* tog handles SIGWINCH and SIGCONT */
1963 	if (sigaddset(&sigset, SIGWINCH) == -1)
1964 		return got_error_from_errno("sigaddset");
1965 	if (sigaddset(&sigset, SIGCONT) == -1)
1966 		return got_error_from_errno("sigaddset");
1967 
1968 	/* ncurses handles SIGTSTP */
1969 	if (sigaddset(&sigset, SIGTSTP) == -1)
1970 		return got_error_from_errno("sigaddset");
1971 
1972 	errcode = pthread_sigmask(SIG_BLOCK, &sigset, NULL);
1973 	if (errcode)
1974 		return got_error_set_errno(errcode, "pthread_sigmask");
1975 
1976 	return NULL;
1977 }
1978 
1979 static void *
1980 log_thread(void *arg)
1981 {
1982 	const struct got_error *err = NULL;
1983 	int errcode = 0;
1984 	struct tog_log_thread_args *a = arg;
1985 	int done = 0;
1986 
1987 	err = block_signals_used_by_main_thread();
1988 	if (err)
1989 		return (void *)err;
1990 
1991 	while (!done && !err && !tog_sigpipe_received) {
1992 		err = queue_commits(a->graph, a->commits, 1, a->repo,
1993 		    a->in_repo_path, a->searching, a->search_next_done,
1994 		    a->regex);
1995 		if (err) {
1996 			if (err->code != GOT_ERR_ITER_COMPLETED)
1997 				return (void *)err;
1998 			err = NULL;
1999 			done = 1;
2000 		} else if (a->commits_needed > 0)
2001 			a->commits_needed--;
2002 
2003 		errcode = pthread_mutex_lock(&tog_mutex);
2004 		if (errcode) {
2005 			err = got_error_set_errno(errcode,
2006 			    "pthread_mutex_lock");
2007 			break;
2008 		} else if (*a->quit)
2009 			done = 1;
2010 		else if (*a->first_displayed_entry == NULL) {
2011 			*a->first_displayed_entry =
2012 			    TAILQ_FIRST(&a->commits->head);
2013 			*a->selected_entry = *a->first_displayed_entry;
2014 		}
2015 
2016 		errcode = pthread_cond_signal(&a->commit_loaded);
2017 		if (errcode) {
2018 			err = got_error_set_errno(errcode,
2019 			    "pthread_cond_signal");
2020 			pthread_mutex_unlock(&tog_mutex);
2021 			break;
2022 		}
2023 
2024 		if (done)
2025 			a->commits_needed = 0;
2026 		else {
2027 			if (a->commits_needed == 0) {
2028 				errcode = pthread_cond_wait(&a->need_commits,
2029 				    &tog_mutex);
2030 				if (errcode)
2031 					err = got_error_set_errno(errcode,
2032 					    "pthread_cond_wait");
2033 			}
2034 		}
2035 
2036 		errcode = pthread_mutex_unlock(&tog_mutex);
2037 		if (errcode && err == NULL)
2038 			err = got_error_set_errno(errcode,
2039 			    "pthread_mutex_unlock");
2040 	}
2041 	a->log_complete = 1;
2042 	return (void *)err;
2043 }
2044 
2045 static const struct got_error *
2046 stop_log_thread(struct tog_log_view_state *s)
2047 {
2048 	const struct got_error *err = NULL;
2049 	int errcode;
2050 
2051 	if (s->thread) {
2052 		s->quit = 1;
2053 		errcode = pthread_cond_signal(&s->thread_args.need_commits);
2054 		if (errcode)
2055 			return got_error_set_errno(errcode,
2056 			    "pthread_cond_signal");
2057 		errcode = pthread_mutex_unlock(&tog_mutex);
2058 		if (errcode)
2059 			return got_error_set_errno(errcode,
2060 			    "pthread_mutex_unlock");
2061 		errcode = pthread_join(s->thread, (void **)&err);
2062 		if (errcode)
2063 			return got_error_set_errno(errcode, "pthread_join");
2064 		errcode = pthread_mutex_lock(&tog_mutex);
2065 		if (errcode)
2066 			return got_error_set_errno(errcode,
2067 			    "pthread_mutex_lock");
2068 		s->thread = NULL;
2069 	}
2070 
2071 	if (s->thread_args.repo) {
2072 		got_repo_close(s->thread_args.repo);
2073 		s->thread_args.repo = NULL;
2074 	}
2075 
2076 	if (s->thread_args.graph) {
2077 		got_commit_graph_close(s->thread_args.graph);
2078 		s->thread_args.graph = NULL;
2079 	}
2080 
2081 	return err;
2082 }
2083 
2084 static const struct got_error *
2085 close_log_view(struct tog_view *view)
2086 {
2087 	const struct got_error *err = NULL;
2088 	struct tog_log_view_state *s = &view->state.log;
2089 	int errcode;
2090 
2091 	err = stop_log_thread(s);
2092 
2093 	errcode = pthread_cond_destroy(&s->thread_args.need_commits);
2094 	if (errcode && err == NULL)
2095 		err = got_error_set_errno(errcode, "pthread_cond_destroy");
2096 
2097 	errcode = pthread_cond_destroy(&s->thread_args.commit_loaded);
2098 	if (errcode && err == NULL)
2099 		err = got_error_set_errno(errcode, "pthread_cond_destroy");
2100 
2101 	free_commits(&s->commits);
2102 	free(s->in_repo_path);
2103 	s->in_repo_path = NULL;
2104 	free(s->start_id);
2105 	s->start_id = NULL;
2106 	got_ref_list_free(&s->refs);
2107 	return err;
2108 }
2109 
2110 static const struct got_error *
2111 search_start_log_view(struct tog_view *view)
2112 {
2113 	struct tog_log_view_state *s = &view->state.log;
2114 
2115 	s->matched_entry = NULL;
2116 	s->search_entry = NULL;
2117 	return NULL;
2118 }
2119 
2120 static const struct got_error *
2121 search_next_log_view(struct tog_view *view)
2122 {
2123 	const struct got_error *err = NULL;
2124 	struct tog_log_view_state *s = &view->state.log;
2125 	struct commit_queue_entry *entry;
2126 
2127 	/* Display progress update in log view. */
2128 	show_log_view(view);
2129 	update_panels();
2130 	doupdate();
2131 
2132 	if (s->search_entry) {
2133 		int errcode, ch;
2134 		errcode = pthread_mutex_unlock(&tog_mutex);
2135 		if (errcode)
2136 			return got_error_set_errno(errcode,
2137 			    "pthread_mutex_unlock");
2138 		ch = wgetch(view->window);
2139 		errcode = pthread_mutex_lock(&tog_mutex);
2140 		if (errcode)
2141 			return got_error_set_errno(errcode,
2142 			    "pthread_mutex_lock");
2143 		if (ch == KEY_BACKSPACE) {
2144 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
2145 			return NULL;
2146 		}
2147 		if (view->searching == TOG_SEARCH_FORWARD)
2148 			entry = TAILQ_NEXT(s->search_entry, entry);
2149 		else
2150 			entry = TAILQ_PREV(s->search_entry,
2151 			    commit_queue_head, entry);
2152 	} else if (s->matched_entry) {
2153 		if (view->searching == TOG_SEARCH_FORWARD)
2154 			entry = TAILQ_NEXT(s->matched_entry, entry);
2155 		else
2156 			entry = TAILQ_PREV(s->matched_entry,
2157 			    commit_queue_head, entry);
2158 	} else {
2159 		if (view->searching == TOG_SEARCH_FORWARD)
2160 			entry = TAILQ_FIRST(&s->commits.head);
2161 		else
2162 			entry = TAILQ_LAST(&s->commits.head, commit_queue_head);
2163 	}
2164 
2165 	while (1) {
2166 		int have_match = 0;
2167 
2168 		if (entry == NULL) {
2169 			if (s->thread_args.log_complete ||
2170 			    view->searching == TOG_SEARCH_BACKWARD) {
2171 				view->search_next_done =
2172 				    (s->matched_entry == NULL ?
2173 				    TOG_SEARCH_HAVE_NONE : TOG_SEARCH_NO_MORE);
2174 				s->search_entry = NULL;
2175 				return NULL;
2176 			}
2177 			/*
2178 			 * Poke the log thread for more commits and return,
2179 			 * allowing the main loop to make progress. Search
2180 			 * will resume at s->search_entry once we come back.
2181 			 */
2182 			s->thread_args.commits_needed++;
2183 			return trigger_log_thread(view, 0,
2184 			    &s->thread_args.commits_needed,
2185 			    &s->thread_args.log_complete,
2186 			    &s->thread_args.need_commits,
2187 			    &s->thread_args.commit_loaded);
2188 		}
2189 
2190 		err = match_commit(&have_match, entry->id, entry->commit,
2191 		    &view->regex);
2192 		if (err)
2193 			break;
2194 		if (have_match) {
2195 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
2196 			s->matched_entry = entry;
2197 			break;
2198 		}
2199 
2200 		s->search_entry = entry;
2201 		if (view->searching == TOG_SEARCH_FORWARD)
2202 			entry = TAILQ_NEXT(entry, entry);
2203 		else
2204 			entry = TAILQ_PREV(entry, commit_queue_head, entry);
2205 	}
2206 
2207 	if (s->matched_entry) {
2208 		int cur = s->selected_entry->idx;
2209 		while (cur < s->matched_entry->idx) {
2210 			err = input_log_view(NULL, NULL, NULL, view, KEY_DOWN);
2211 			if (err)
2212 				return err;
2213 			cur++;
2214 		}
2215 		while (cur > s->matched_entry->idx) {
2216 			err = input_log_view(NULL, NULL, NULL, view, KEY_UP);
2217 			if (err)
2218 				return err;
2219 			cur--;
2220 		}
2221 	}
2222 
2223 	s->search_entry = NULL;
2224 
2225 	return NULL;
2226 }
2227 
2228 static const struct got_error *
2229 open_log_view(struct tog_view *view, struct got_object_id *start_id,
2230     struct got_repository *repo, const char *head_ref_name,
2231     const char *in_repo_path, int log_branches)
2232 {
2233 	const struct got_error *err = NULL;
2234 	struct tog_log_view_state *s = &view->state.log;
2235 	struct got_repository *thread_repo = NULL;
2236 	struct got_commit_graph *thread_graph = NULL;
2237 	int errcode;
2238 
2239 	SIMPLEQ_INIT(&s->refs);
2240 
2241 	if (in_repo_path != s->in_repo_path) {
2242 		free(s->in_repo_path);
2243 		s->in_repo_path = strdup(in_repo_path);
2244 		if (s->in_repo_path == NULL)
2245 			return got_error_from_errno("strdup");
2246 	}
2247 
2248 	/* The commit queue only contains commits being displayed. */
2249 	TAILQ_INIT(&s->commits.head);
2250 	s->commits.ncommits = 0;
2251 
2252 	err = got_ref_list(&s->refs, repo, NULL, got_ref_cmp_by_name, NULL);
2253 	if (err)
2254 		goto done;
2255 
2256 	s->repo = repo;
2257 	s->head_ref_name = head_ref_name;
2258 	s->start_id = got_object_id_dup(start_id);
2259 	if (s->start_id == NULL) {
2260 		err = got_error_from_errno("got_object_id_dup");
2261 		goto done;
2262 	}
2263 	s->log_branches = log_branches;
2264 
2265 	SIMPLEQ_INIT(&s->colors);
2266 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
2267 		err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
2268 		    get_color_value("TOG_COLOR_COMMIT"));
2269 		if (err)
2270 			goto done;
2271 		err = add_color(&s->colors, "^$", TOG_COLOR_AUTHOR,
2272 		    get_color_value("TOG_COLOR_AUTHOR"));
2273 		if (err) {
2274 			free_colors(&s->colors);
2275 			goto done;
2276 		}
2277 		err = add_color(&s->colors, "^$", TOG_COLOR_DATE,
2278 		    get_color_value("TOG_COLOR_DATE"));
2279 		if (err) {
2280 			free_colors(&s->colors);
2281 			goto done;
2282 		}
2283 	}
2284 
2285 	view->show = show_log_view;
2286 	view->input = input_log_view;
2287 	view->close = close_log_view;
2288 	view->search_start = search_start_log_view;
2289 	view->search_next = search_next_log_view;
2290 
2291 	err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL);
2292 	if (err)
2293 		goto done;
2294 	err = got_commit_graph_open(&thread_graph, s->in_repo_path,
2295 	    !s->log_branches);
2296 	if (err)
2297 		goto done;
2298 	err = got_commit_graph_iter_start(thread_graph, s->start_id,
2299 	    s->repo, NULL, NULL);
2300 	if (err)
2301 		goto done;
2302 
2303 	errcode = pthread_cond_init(&s->thread_args.need_commits, NULL);
2304 	if (errcode) {
2305 		err = got_error_set_errno(errcode, "pthread_cond_init");
2306 		goto done;
2307 	}
2308 	errcode = pthread_cond_init(&s->thread_args.commit_loaded, NULL);
2309 	if (errcode) {
2310 		err = got_error_set_errno(errcode, "pthread_cond_init");
2311 		goto done;
2312 	}
2313 
2314 	s->thread_args.commits_needed = view->nlines;
2315 	s->thread_args.graph = thread_graph;
2316 	s->thread_args.commits = &s->commits;
2317 	s->thread_args.in_repo_path = s->in_repo_path;
2318 	s->thread_args.start_id = s->start_id;
2319 	s->thread_args.repo = thread_repo;
2320 	s->thread_args.log_complete = 0;
2321 	s->thread_args.quit = &s->quit;
2322 	s->thread_args.first_displayed_entry = &s->first_displayed_entry;
2323 	s->thread_args.selected_entry = &s->selected_entry;
2324 	s->thread_args.searching = &view->searching;
2325 	s->thread_args.search_next_done = &view->search_next_done;
2326 	s->thread_args.regex = &view->regex;
2327 done:
2328 	if (err)
2329 		close_log_view(view);
2330 	return err;
2331 }
2332 
2333 static const struct got_error *
2334 show_log_view(struct tog_view *view)
2335 {
2336 	const struct got_error *err;
2337 	struct tog_log_view_state *s = &view->state.log;
2338 
2339 	if (s->thread == NULL) {
2340 		int errcode = pthread_create(&s->thread, NULL, log_thread,
2341 		    &s->thread_args);
2342 		if (errcode)
2343 			return got_error_set_errno(errcode, "pthread_create");
2344 		if (s->thread_args.commits_needed > 0) {
2345 			err = trigger_log_thread(view, 1,
2346 			    &s->thread_args.commits_needed,
2347 			    &s->thread_args.log_complete,
2348 			    &s->thread_args.need_commits,
2349 			    &s->thread_args.commit_loaded);
2350 			if (err)
2351 				return err;
2352 		}
2353 	}
2354 
2355 	return draw_commits(view, &s->last_displayed_entry,
2356 	    &s->selected_entry, s->first_displayed_entry,
2357 	    &s->commits, s->selected, view->nlines, &s->refs,
2358 	    s->in_repo_path, s->thread_args.commits_needed, &s->colors);
2359 }
2360 
2361 static const struct got_error *
2362 input_log_view(struct tog_view **new_view, struct tog_view **dead_view,
2363     struct tog_view **focus_view, struct tog_view *view, int ch)
2364 {
2365 	const struct got_error *err = NULL;
2366 	struct tog_log_view_state *s = &view->state.log;
2367 	char *parent_path, *in_repo_path = NULL;
2368 	struct tog_view *diff_view = NULL, *tree_view = NULL, *lv = NULL;
2369 	struct tog_view *ref_view = NULL;
2370 	int begin_x = 0;
2371 	struct got_object_id *start_id;
2372 
2373 	switch (ch) {
2374 	case 'q':
2375 		s->quit = 1;
2376 		break;
2377 	case 'k':
2378 	case KEY_UP:
2379 	case '<':
2380 	case ',':
2381 		if (s->first_displayed_entry == NULL)
2382 			break;
2383 		if (s->selected > 0)
2384 			s->selected--;
2385 		else
2386 			scroll_up(view, &s->first_displayed_entry, 1,
2387 			    &s->commits);
2388 		break;
2389 	case KEY_PPAGE:
2390 	case CTRL('b'):
2391 		if (s->first_displayed_entry == NULL)
2392 			break;
2393 		if (TAILQ_FIRST(&s->commits.head) ==
2394 		    s->first_displayed_entry) {
2395 			s->selected = 0;
2396 			break;
2397 		}
2398 		scroll_up(view, &s->first_displayed_entry,
2399 		    view->nlines - 1, &s->commits);
2400 		break;
2401 	case 'j':
2402 	case KEY_DOWN:
2403 	case '>':
2404 	case '.':
2405 		if (s->first_displayed_entry == NULL)
2406 			break;
2407 		if (s->selected < MIN(view->nlines - 2,
2408 		    s->commits.ncommits - 1)) {
2409 			s->selected++;
2410 			break;
2411 		}
2412 		err = scroll_down(view, &s->first_displayed_entry, 1,
2413 		    &s->last_displayed_entry, &s->commits,
2414 		    &s->thread_args.log_complete,
2415 		    &s->thread_args.commits_needed,
2416 		    &s->thread_args.need_commits,
2417 		    &s->thread_args.commit_loaded);
2418 		break;
2419 	case KEY_NPAGE:
2420 	case CTRL('f'): {
2421 		struct commit_queue_entry *first;
2422 		first = s->first_displayed_entry;
2423 		if (first == NULL)
2424 			break;
2425 		err = scroll_down(view, &s->first_displayed_entry,
2426 		    view->nlines - 1, &s->last_displayed_entry,
2427 		    &s->commits, &s->thread_args.log_complete,
2428 		    &s->thread_args.commits_needed,
2429 		    &s->thread_args.need_commits,
2430 		    &s->thread_args.commit_loaded);
2431 		if (err)
2432 			break;
2433 		if (first == s->first_displayed_entry &&
2434 		    s->selected < MIN(view->nlines - 2,
2435 		    s->commits.ncommits - 1)) {
2436 			/* can't scroll further down */
2437 			s->selected = MIN(view->nlines - 2,
2438 			    s->commits.ncommits - 1);
2439 		}
2440 		err = NULL;
2441 		break;
2442 	}
2443 	case KEY_RESIZE:
2444 		if (s->selected > view->nlines - 2)
2445 			s->selected = view->nlines - 2;
2446 		if (s->selected > s->commits.ncommits - 1)
2447 			s->selected = s->commits.ncommits - 1;
2448 		break;
2449 	case KEY_ENTER:
2450 	case ' ':
2451 	case '\r':
2452 		if (s->selected_entry == NULL)
2453 			break;
2454 		if (view_is_parent_view(view))
2455 			begin_x = view_split_begin_x(view->begin_x);
2456 		err = open_diff_view_for_commit(&diff_view, begin_x,
2457 		    s->selected_entry->commit, s->selected_entry->id,
2458 		    view, s->repo);
2459 		if (err)
2460 			break;
2461 		if (view_is_parent_view(view)) {
2462 			err = view_close_child(view);
2463 			if (err)
2464 				return err;
2465 			err = view_set_child(view, diff_view);
2466 			if (err) {
2467 				view_close(diff_view);
2468 				break;
2469 			}
2470 			*focus_view = diff_view;
2471 			view->child_focussed = 1;
2472 		} else
2473 			*new_view = diff_view;
2474 		break;
2475 	case 't':
2476 		if (s->selected_entry == NULL)
2477 			break;
2478 		if (view_is_parent_view(view))
2479 			begin_x = view_split_begin_x(view->begin_x);
2480 		err = browse_commit_tree(&tree_view, begin_x,
2481 		    s->selected_entry, s->in_repo_path, s->repo);
2482 		if (err)
2483 			break;
2484 		if (view_is_parent_view(view)) {
2485 			err = view_close_child(view);
2486 			if (err)
2487 				return err;
2488 			err = view_set_child(view, tree_view);
2489 			if (err) {
2490 				view_close(tree_view);
2491 				break;
2492 			}
2493 			*focus_view = tree_view;
2494 			view->child_focussed = 1;
2495 		} else
2496 			*new_view = tree_view;
2497 		break;
2498 	case KEY_BACKSPACE:
2499 		if (got_path_cmp(s->in_repo_path, "/",
2500 		    strlen(s->in_repo_path), 1) == 0)
2501 			break;
2502 		err = got_path_dirname(&parent_path, s->in_repo_path);
2503 		if (err)
2504 			return err;
2505 		err = stop_log_thread(s);
2506 		if (err) {
2507 			free(parent_path);
2508 			return err;
2509 		}
2510 		lv = view_open(view->nlines, view->ncols,
2511 		    view->begin_y, view->begin_x, TOG_VIEW_LOG);
2512 		if (lv == NULL) {
2513 			free(parent_path);
2514 			return got_error_from_errno("view_open");
2515 		}
2516 		err = open_log_view(lv, s->start_id, s->repo, s->head_ref_name,
2517 		    parent_path, s->log_branches);
2518 		free(parent_path);
2519 		if (err)
2520 			return err;;
2521 		if (view_is_parent_view(view))
2522 			*new_view = lv;
2523 		else {
2524 			view_set_child(view->parent, lv);
2525 			*focus_view = lv;
2526 		}
2527 		break;
2528 	case CTRL('l'):
2529 		err = stop_log_thread(s);
2530 		if (err)
2531 			return err;
2532 		lv = view_open(view->nlines, view->ncols,
2533 		    view->begin_y, view->begin_x, TOG_VIEW_LOG);
2534 		if (lv == NULL)
2535 			return got_error_from_errno("view_open");
2536 		err = got_repo_match_object_id(&start_id, NULL,
2537 		    s->head_ref_name ? s->head_ref_name : GOT_REF_HEAD,
2538 		    GOT_OBJ_TYPE_COMMIT, 1, s->repo);
2539 		if (err) {
2540 			view_close(lv);
2541 			return err;
2542 		}
2543 		in_repo_path = strdup(s->in_repo_path);
2544 		if (in_repo_path == NULL) {
2545 			free(start_id);
2546 			view_close(lv);
2547 			return got_error_from_errno("strdup");
2548 		}
2549 		err = open_log_view(lv, start_id, s->repo, s->head_ref_name,
2550 		    in_repo_path, s->log_branches);
2551 		if (err) {
2552 			free(start_id);
2553 			view_close(lv);
2554 			return err;;
2555 		}
2556 		*dead_view = view;
2557 		*new_view = lv;
2558 		break;
2559 	case 'B':
2560 		s->log_branches = !s->log_branches;
2561 		err = stop_log_thread(s);
2562 		if (err)
2563 			return err;
2564 		lv = view_open(view->nlines, view->ncols,
2565 		    view->begin_y, view->begin_x, TOG_VIEW_LOG);
2566 		if (lv == NULL)
2567 			return got_error_from_errno("view_open");
2568 		err = open_log_view(lv, s->start_id, s->repo,
2569 		    s->head_ref_name, s->in_repo_path, s->log_branches);
2570 		if (err) {
2571 			view_close(lv);
2572 			return err;;
2573 		}
2574 		*dead_view = view;
2575 		*new_view = lv;
2576 		break;
2577 	case 'r':
2578 		if (view_is_parent_view(view))
2579 			begin_x = view_split_begin_x(view->begin_x);
2580 		ref_view = view_open(view->nlines, view->ncols,
2581 		    view->begin_y, begin_x, TOG_VIEW_REF);
2582 		if (ref_view == NULL)
2583 			return got_error_from_errno("view_open");
2584 		err = open_ref_view(ref_view, s->repo);
2585 		if (err) {
2586 			view_close(ref_view);
2587 			return err;
2588 		}
2589 		if (view_is_parent_view(view)) {
2590 			err = view_close_child(view);
2591 			if (err)
2592 				return err;
2593 			err = view_set_child(view, ref_view);
2594 			if (err) {
2595 				view_close(ref_view);
2596 				break;
2597 			}
2598 			*focus_view = ref_view;
2599 			view->child_focussed = 1;
2600 		} else
2601 			*new_view = ref_view;
2602 		break;
2603 	default:
2604 		break;
2605 	}
2606 
2607 	return err;
2608 }
2609 
2610 static const struct got_error *
2611 apply_unveil(const char *repo_path, const char *worktree_path)
2612 {
2613 	const struct got_error *error;
2614 
2615 #ifdef PROFILE
2616 	if (unveil("gmon.out", "rwc") != 0)
2617 		return got_error_from_errno2("unveil", "gmon.out");
2618 #endif
2619 	if (repo_path && unveil(repo_path, "r") != 0)
2620 		return got_error_from_errno2("unveil", repo_path);
2621 
2622 	if (worktree_path && unveil(worktree_path, "rwc") != 0)
2623 		return got_error_from_errno2("unveil", worktree_path);
2624 
2625 	if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
2626 		return got_error_from_errno2("unveil", GOT_TMPDIR_STR);
2627 
2628 	error = got_privsep_unveil_exec_helpers();
2629 	if (error != NULL)
2630 		return error;
2631 
2632 	if (unveil(NULL, NULL) != 0)
2633 		return got_error_from_errno("unveil");
2634 
2635 	return NULL;
2636 }
2637 
2638 static void
2639 init_curses(void)
2640 {
2641 	initscr();
2642 	cbreak();
2643 	halfdelay(1); /* Do fast refresh while initial view is loading. */
2644 	noecho();
2645 	nonl();
2646 	intrflush(stdscr, FALSE);
2647 	keypad(stdscr, TRUE);
2648 	curs_set(0);
2649 	if (getenv("TOG_COLORS") != NULL) {
2650 		start_color();
2651 		use_default_colors();
2652 	}
2653 	signal(SIGWINCH, tog_sigwinch);
2654 	signal(SIGPIPE, tog_sigpipe);
2655 	signal(SIGCONT, tog_sigcont);
2656 }
2657 
2658 static const struct got_error *
2659 get_in_repo_path_from_argv0(char **in_repo_path, int argc, char *argv[],
2660     struct got_repository *repo, struct got_worktree *worktree)
2661 {
2662 	const struct got_error *err = NULL;
2663 
2664 	if (argc == 0) {
2665 		*in_repo_path = strdup("/");
2666 		if (*in_repo_path == NULL)
2667 			return got_error_from_errno("strdup");
2668 		return NULL;
2669 	}
2670 
2671 	if (worktree) {
2672 		const char *prefix = got_worktree_get_path_prefix(worktree);
2673 		char *p;
2674 
2675 		err = got_worktree_resolve_path(&p, worktree, argv[0]);
2676 		if (err)
2677 			return err;
2678 		if (asprintf(in_repo_path, "%s%s%s", prefix,
2679 		    (p[0] != '\0' && !got_path_is_root_dir(prefix)) ? "/" : "",
2680 		    p) == -1) {
2681 			err = got_error_from_errno("asprintf");
2682 			*in_repo_path = NULL;
2683 		}
2684 		free(p);
2685 	} else
2686 		err = got_repo_map_path(in_repo_path, repo, argv[0]);
2687 
2688 	return err;
2689 }
2690 
2691 static const struct got_error *
2692 cmd_log(int argc, char *argv[])
2693 {
2694 	const struct got_error *error;
2695 	struct got_repository *repo = NULL;
2696 	struct got_worktree *worktree = NULL;
2697 	struct got_object_id *start_id = NULL;
2698 	char *in_repo_path = NULL, *repo_path = NULL, *cwd = NULL;
2699 	char *start_commit = NULL, *head_ref_name = NULL;
2700 	int ch, log_branches = 0;
2701 	struct tog_view *view;
2702 
2703 #ifndef PROFILE
2704 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
2705 	    NULL) == -1)
2706 		err(1, "pledge");
2707 #endif
2708 
2709 	while ((ch = getopt(argc, argv, "bc:r:")) != -1) {
2710 		switch (ch) {
2711 		case 'b':
2712 			log_branches = 1;
2713 			break;
2714 		case 'c':
2715 			start_commit = optarg;
2716 			break;
2717 		case 'r':
2718 			repo_path = realpath(optarg, NULL);
2719 			if (repo_path == NULL)
2720 				return got_error_from_errno2("realpath",
2721 				    optarg);
2722 			break;
2723 		default:
2724 			usage_log();
2725 			/* NOTREACHED */
2726 		}
2727 	}
2728 
2729 	argc -= optind;
2730 	argv += optind;
2731 
2732 	if (argc > 1)
2733 		usage_log();
2734 
2735 	cwd = getcwd(NULL, 0);
2736 	if (cwd == NULL)
2737 		return got_error_from_errno("getcwd");
2738 
2739 	error = got_worktree_open(&worktree, cwd);
2740 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
2741 		goto done;
2742 
2743 	if (repo_path == NULL) {
2744 		if (worktree)
2745 			repo_path =
2746 			    strdup(got_worktree_get_repo_path(worktree));
2747 		else
2748 			repo_path = strdup(cwd);
2749 	}
2750 	if (repo_path == NULL) {
2751 		error = got_error_from_errno("strdup");
2752 		goto done;
2753 	}
2754 
2755 	error = got_repo_open(&repo, repo_path, NULL);
2756 	if (error != NULL)
2757 		goto done;
2758 
2759 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
2760 	    repo, worktree);
2761 	if (error)
2762 		goto done;
2763 
2764 	init_curses();
2765 
2766 	error = apply_unveil(got_repo_get_path(repo),
2767 	    worktree ? got_worktree_get_root_path(worktree) : NULL);
2768 	if (error)
2769 		goto done;
2770 
2771 	if (start_commit == NULL)
2772 		error = got_repo_match_object_id(&start_id, NULL, worktree ?
2773 		    got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
2774 		    GOT_OBJ_TYPE_COMMIT, 1, repo);
2775 	else
2776 		error = got_repo_match_object_id(&start_id, NULL, start_commit,
2777 		    GOT_OBJ_TYPE_COMMIT, 1, repo);
2778 	if (error != NULL)
2779 		goto done;
2780 
2781 	view = view_open(0, 0, 0, 0, TOG_VIEW_LOG);
2782 	if (view == NULL) {
2783 		error = got_error_from_errno("view_open");
2784 		goto done;
2785 	}
2786 	if (worktree) {
2787 		head_ref_name = strdup(
2788 		    got_worktree_get_head_ref_name(worktree));
2789 		if (head_ref_name == NULL) {
2790 			error = got_error_from_errno("strdup");
2791 			goto done;
2792 		}
2793 	}
2794 	error = open_log_view(view, start_id, repo, head_ref_name,
2795 	    in_repo_path, log_branches);
2796 	if (error)
2797 		goto done;
2798 	if (worktree) {
2799 		/* Release work tree lock. */
2800 		got_worktree_close(worktree);
2801 		worktree = NULL;
2802 	}
2803 	error = view_loop(view);
2804 done:
2805 	free(in_repo_path);
2806 	free(repo_path);
2807 	free(cwd);
2808 	free(start_id);
2809 	free(head_ref_name);
2810 	if (repo)
2811 		got_repo_close(repo);
2812 	if (worktree)
2813 		got_worktree_close(worktree);
2814 	return error;
2815 }
2816 
2817 __dead static void
2818 usage_diff(void)
2819 {
2820 	endwin();
2821 	fprintf(stderr, "usage: %s diff [-a] [-C number] [-r repository-path] "
2822 	    "[-w] object1 object2\n", getprogname());
2823 	exit(1);
2824 }
2825 
2826 static char *
2827 parse_next_line(FILE *f, size_t *len)
2828 {
2829 	char *line;
2830 	size_t linelen;
2831 	size_t lineno;
2832 	const char delim[3] = { '\0', '\0', '\0'};
2833 
2834 	line = fparseln(f, &linelen, &lineno, delim, 0);
2835 	if (len)
2836 		*len = linelen;
2837 	return line;
2838 }
2839 
2840 static int
2841 match_line(const char *line, regex_t *regex, size_t nmatch,
2842     regmatch_t *regmatch)
2843 {
2844 	return regexec(regex, line, nmatch, regmatch, 0) == 0;
2845 }
2846 
2847 struct tog_color *
2848 match_color(struct tog_colors *colors, const char *line)
2849 {
2850 	struct tog_color *tc = NULL;
2851 
2852 	SIMPLEQ_FOREACH(tc, colors, entry) {
2853 		if (match_line(line, &tc->regex, 0, NULL))
2854 			return tc;
2855 	}
2856 
2857 	return NULL;
2858 }
2859 
2860 static const struct got_error *
2861 add_matched_line(int *wtotal, const char *line, int wlimit, int col_tab_align,
2862     WINDOW *window, regmatch_t *regmatch)
2863 {
2864 	const struct got_error *err = NULL;
2865 	wchar_t *wline;
2866 	int width;
2867 	char *s;
2868 
2869 	*wtotal = 0;
2870 
2871 	s = strndup(line, regmatch->rm_so);
2872 	if (s == NULL)
2873 		return got_error_from_errno("strndup");
2874 
2875 	err = format_line(&wline, &width, s, wlimit, col_tab_align);
2876 	if (err) {
2877 		free(s);
2878 		return err;
2879 	}
2880 	waddwstr(window, wline);
2881 	free(wline);
2882 	free(s);
2883 	wlimit -= width;
2884 	*wtotal += width;
2885 
2886 	if (wlimit > 0) {
2887 		s = strndup(line + regmatch->rm_so,
2888 		    regmatch->rm_eo - regmatch->rm_so);
2889 		if (s == NULL) {
2890 			err = got_error_from_errno("strndup");
2891 			free(s);
2892 			return err;
2893 		}
2894 		err = format_line(&wline, &width, s, wlimit, col_tab_align);
2895 		if (err) {
2896 			free(s);
2897 			return err;
2898 		}
2899 		wattr_on(window, A_STANDOUT, NULL);
2900 		waddwstr(window, wline);
2901 		wattr_off(window, A_STANDOUT, NULL);
2902 		free(wline);
2903 		free(s);
2904 		wlimit -= width;
2905 		*wtotal += width;
2906 	}
2907 
2908 	if (wlimit > 0 && strlen(line) > regmatch->rm_eo) {
2909 		err = format_line(&wline, &width,
2910 		     line + regmatch->rm_eo, wlimit, col_tab_align);
2911 		if (err)
2912 			return err;
2913 		waddwstr(window, wline);
2914 		free(wline);
2915 		*wtotal += width;
2916 	}
2917 
2918 	return NULL;
2919 }
2920 
2921 static const struct got_error *
2922 draw_file(struct tog_view *view, FILE *f, int first_displayed_line, int nlines,
2923     off_t *line_offsets, int selected_line, int max_lines,
2924     int *last_displayed_line, int *eof, const char *header,
2925     struct tog_colors *colors, int matched_line, regmatch_t *regmatch)
2926 {
2927 	const struct got_error *err;
2928 	int nprinted = 0;
2929 	char *line;
2930 	struct tog_color *tc;
2931 	size_t len;
2932 	wchar_t *wline;
2933 	int width;
2934 	off_t line_offset;
2935 
2936 	line_offset = line_offsets[first_displayed_line - 1];
2937 	if (fseeko(f, line_offset, SEEK_SET) == -1)
2938 		return got_error_from_errno("fseek");
2939 
2940 	werase(view->window);
2941 
2942 	if (header) {
2943 		if (asprintf(&line, "[%d/%d] %s",
2944 		    first_displayed_line - 1 + selected_line, nlines,
2945 		    header) == -1)
2946 			return got_error_from_errno("asprintf");
2947 		err = format_line(&wline, &width, line, view->ncols, 0);
2948 		free(line);
2949 		if (err)
2950 			return err;
2951 
2952 		if (view_needs_focus_indication(view))
2953 			wstandout(view->window);
2954 		waddwstr(view->window, wline);
2955 		free(wline);
2956 		wline = NULL;
2957 		if (view_needs_focus_indication(view))
2958 			wstandend(view->window);
2959 		if (width <= view->ncols - 1)
2960 			waddch(view->window, '\n');
2961 
2962 		if (max_lines <= 1)
2963 			return NULL;
2964 		max_lines--;
2965 	}
2966 
2967 	*eof = 0;
2968 	while (max_lines > 0 && nprinted < max_lines) {
2969 		line = parse_next_line(f, &len);
2970 		if (line == NULL) {
2971 			*eof = 1;
2972 			break;
2973 		}
2974 
2975 		tc = match_color(colors, line);
2976 		if (tc)
2977 			wattr_on(view->window,
2978 			    COLOR_PAIR(tc->colorpair), NULL);
2979 		if (first_displayed_line + nprinted == matched_line &&
2980 		    regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
2981 			err = add_matched_line(&width, line, view->ncols, 0,
2982 			    view->window, regmatch);
2983 			if (err) {
2984 				free(line);
2985 				return err;
2986 			}
2987 		} else {
2988 			err = format_line(&wline, &width, line, view->ncols, 0);
2989 			if (err) {
2990 				free(line);
2991 				return err;
2992 			}
2993 			waddwstr(view->window, wline);
2994 			free(wline);
2995 			wline = NULL;
2996 		}
2997 		if (tc)
2998 			wattr_off(view->window,
2999 			    COLOR_PAIR(tc->colorpair), NULL);
3000 		if (width <= view->ncols - 1)
3001 			waddch(view->window, '\n');
3002 		nprinted++;
3003 		free(line);
3004 	}
3005 	if (nprinted >= 1)
3006 		*last_displayed_line = first_displayed_line + (nprinted - 1);
3007 	else
3008 		*last_displayed_line = first_displayed_line;
3009 
3010 	view_vborder(view);
3011 
3012 	if (*eof) {
3013 		while (nprinted < view->nlines) {
3014 			waddch(view->window, '\n');
3015 			nprinted++;
3016 		}
3017 
3018 		err = format_line(&wline, &width, TOG_EOF_STRING, view->ncols, 0);
3019 		if (err) {
3020 			return err;
3021 		}
3022 
3023 		wstandout(view->window);
3024 		waddwstr(view->window, wline);
3025 		free(wline);
3026 		wline = NULL;
3027 		wstandend(view->window);
3028 	}
3029 
3030 	return NULL;
3031 }
3032 
3033 static char *
3034 get_datestr(time_t *time, char *datebuf)
3035 {
3036 	struct tm mytm, *tm;
3037 	char *p, *s;
3038 
3039 	tm = gmtime_r(time, &mytm);
3040 	if (tm == NULL)
3041 		return NULL;
3042 	s = asctime_r(tm, datebuf);
3043 	if (s == NULL)
3044 		return NULL;
3045 	p = strchr(s, '\n');
3046 	if (p)
3047 		*p = '\0';
3048 	return s;
3049 }
3050 
3051 static const struct got_error *
3052 get_changed_paths(struct got_pathlist_head *paths,
3053     struct got_commit_object *commit, struct got_repository *repo)
3054 {
3055 	const struct got_error *err = NULL;
3056 	struct got_object_id *tree_id1 = NULL, *tree_id2 = NULL;
3057 	struct got_tree_object *tree1 = NULL, *tree2 = NULL;
3058 	struct got_object_qid *qid;
3059 
3060 	qid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
3061 	if (qid != NULL) {
3062 		struct got_commit_object *pcommit;
3063 		err = got_object_open_as_commit(&pcommit, repo,
3064 		    qid->id);
3065 		if (err)
3066 			return err;
3067 
3068 		tree_id1 = got_object_commit_get_tree_id(pcommit);
3069 		got_object_commit_close(pcommit);
3070 
3071 	}
3072 
3073 	if (tree_id1) {
3074 		err = got_object_open_as_tree(&tree1, repo, tree_id1);
3075 		if (err)
3076 			goto done;
3077 	}
3078 
3079 	tree_id2 = got_object_commit_get_tree_id(commit);
3080 	err = got_object_open_as_tree(&tree2, repo, tree_id2);
3081 	if (err)
3082 		goto done;
3083 
3084 	err = got_diff_tree(tree1, tree2, "", "", repo,
3085 	    got_diff_tree_collect_changed_paths, paths, 0);
3086 done:
3087 	if (tree1)
3088 		got_object_tree_close(tree1);
3089 	if (tree2)
3090 		got_object_tree_close(tree2);
3091 	return err;
3092 }
3093 
3094 static const struct got_error *
3095 add_line_offset(off_t **line_offsets, size_t *nlines, off_t off)
3096 {
3097 	off_t *p;
3098 
3099 	p = reallocarray(*line_offsets, *nlines + 1, sizeof(off_t));
3100 	if (p == NULL)
3101 		return got_error_from_errno("reallocarray");
3102 	*line_offsets = p;
3103 	(*line_offsets)[*nlines] = off;
3104 	(*nlines)++;
3105 	return NULL;
3106 }
3107 
3108 static const struct got_error *
3109 write_commit_info(off_t **line_offsets, size_t *nlines,
3110     struct got_object_id *commit_id, struct got_reflist_head *refs,
3111     struct got_repository *repo, FILE *outfile)
3112 {
3113 	const struct got_error *err = NULL;
3114 	char datebuf[26], *datestr;
3115 	struct got_commit_object *commit;
3116 	char *id_str = NULL, *logmsg = NULL, *s = NULL, *line;
3117 	time_t committer_time;
3118 	const char *author, *committer;
3119 	char *refs_str = NULL;
3120 	struct got_pathlist_head changed_paths;
3121 	struct got_pathlist_entry *pe;
3122 	off_t outoff = 0;
3123 	int n;
3124 
3125 	TAILQ_INIT(&changed_paths);
3126 
3127 	if (refs) {
3128 		err = build_refs_str(&refs_str, refs, commit_id, repo);
3129 		if (err)
3130 			return err;
3131 	}
3132 
3133 	err = got_object_open_as_commit(&commit, repo, commit_id);
3134 	if (err)
3135 		return err;
3136 
3137 	err = got_object_id_str(&id_str, commit_id);
3138 	if (err) {
3139 		err = got_error_from_errno("got_object_id_str");
3140 		goto done;
3141 	}
3142 
3143 	err = add_line_offset(line_offsets, nlines, 0);
3144 	if (err)
3145 		goto done;
3146 
3147 	n = fprintf(outfile, "commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
3148 	    refs_str ? refs_str : "", refs_str ? ")" : "");
3149 	if (n < 0) {
3150 		err = got_error_from_errno("fprintf");
3151 		goto done;
3152 	}
3153 	outoff += n;
3154 	err = add_line_offset(line_offsets, nlines, outoff);
3155 	if (err)
3156 		goto done;
3157 
3158 	n = fprintf(outfile, "from: %s\n",
3159 	    got_object_commit_get_author(commit));
3160 	if (n < 0) {
3161 		err = got_error_from_errno("fprintf");
3162 		goto done;
3163 	}
3164 	outoff += n;
3165 	err = add_line_offset(line_offsets, nlines, outoff);
3166 	if (err)
3167 		goto done;
3168 
3169 	committer_time = got_object_commit_get_committer_time(commit);
3170 	datestr = get_datestr(&committer_time, datebuf);
3171 	if (datestr) {
3172 		n = fprintf(outfile, "date: %s UTC\n", datestr);
3173 		if (n < 0) {
3174 			err = got_error_from_errno("fprintf");
3175 			goto done;
3176 		}
3177 		outoff += n;
3178 		err = add_line_offset(line_offsets, nlines, outoff);
3179 		if (err)
3180 			goto done;
3181 	}
3182 	author = got_object_commit_get_author(commit);
3183 	committer = got_object_commit_get_committer(commit);
3184 	if (strcmp(author, committer) != 0) {
3185 		n = fprintf(outfile, "via: %s\n", committer);
3186 		if (n < 0) {
3187 			err = got_error_from_errno("fprintf");
3188 			goto done;
3189 		}
3190 		outoff += n;
3191 		err = add_line_offset(line_offsets, nlines, outoff);
3192 		if (err)
3193 			goto done;
3194 	}
3195 	err = got_object_commit_get_logmsg(&logmsg, commit);
3196 	if (err)
3197 		goto done;
3198 	s = logmsg;
3199 	while ((line = strsep(&s, "\n")) != NULL) {
3200 		n = fprintf(outfile, "%s\n", line);
3201 		if (n < 0) {
3202 			err = got_error_from_errno("fprintf");
3203 			goto done;
3204 		}
3205 		outoff += n;
3206 		err = add_line_offset(line_offsets, nlines, outoff);
3207 		if (err)
3208 			goto done;
3209 	}
3210 
3211 	err = get_changed_paths(&changed_paths, commit, repo);
3212 	if (err)
3213 		goto done;
3214 	TAILQ_FOREACH(pe, &changed_paths, entry) {
3215 		struct got_diff_changed_path *cp = pe->data;
3216 		n = fprintf(outfile, "%c  %s\n", cp->status, pe->path);
3217 		if (n < 0) {
3218 			err = got_error_from_errno("fprintf");
3219 			goto done;
3220 		}
3221 		outoff += n;
3222 		err = add_line_offset(line_offsets, nlines, outoff);
3223 		if (err)
3224 			goto done;
3225 		free((char *)pe->path);
3226 		free(pe->data);
3227 	}
3228 
3229 	fputc('\n', outfile);
3230 	outoff++;
3231 	err = add_line_offset(line_offsets, nlines, outoff);
3232 done:
3233 	got_pathlist_free(&changed_paths);
3234 	free(id_str);
3235 	free(logmsg);
3236 	free(refs_str);
3237 	got_object_commit_close(commit);
3238 	if (err) {
3239 		free(*line_offsets);
3240 		*line_offsets = NULL;
3241 		*nlines = 0;
3242 	}
3243 	return err;
3244 }
3245 
3246 static const struct got_error *
3247 create_diff(struct tog_diff_view_state *s)
3248 {
3249 	const struct got_error *err = NULL;
3250 	FILE *f = NULL;
3251 	int obj_type;
3252 
3253 	free(s->line_offsets);
3254 	s->line_offsets = malloc(sizeof(off_t));
3255 	if (s->line_offsets == NULL)
3256 		return got_error_from_errno("malloc");
3257 	s->nlines = 0;
3258 
3259 	f = got_opentemp();
3260 	if (f == NULL) {
3261 		err = got_error_from_errno("got_opentemp");
3262 		goto done;
3263 	}
3264 	if (s->f && fclose(s->f) != 0) {
3265 		err = got_error_from_errno("fclose");
3266 		goto done;
3267 	}
3268 	s->f = f;
3269 
3270 	if (s->id1)
3271 		err = got_object_get_type(&obj_type, s->repo, s->id1);
3272 	else
3273 		err = got_object_get_type(&obj_type, s->repo, s->id2);
3274 	if (err)
3275 		goto done;
3276 
3277 	switch (obj_type) {
3278 	case GOT_OBJ_TYPE_BLOB:
3279 		err = got_diff_objects_as_blobs(&s->line_offsets, &s->nlines,
3280 		    s->id1, s->id2, s->label1, s->label2, s->diff_context,
3281 		    s->ignore_whitespace, s->force_text_diff, s->repo, s->f);
3282 		break;
3283 	case GOT_OBJ_TYPE_TREE:
3284 		err = got_diff_objects_as_trees(&s->line_offsets, &s->nlines,
3285 		    s->id1, s->id2, "", "", s->diff_context,
3286 		    s->ignore_whitespace, s->force_text_diff, s->repo, s->f);
3287 		break;
3288 	case GOT_OBJ_TYPE_COMMIT: {
3289 		const struct got_object_id_queue *parent_ids;
3290 		struct got_object_qid *pid;
3291 		struct got_commit_object *commit2;
3292 
3293 		err = got_object_open_as_commit(&commit2, s->repo, s->id2);
3294 		if (err)
3295 			goto done;
3296 		/* Show commit info if we're diffing to a parent/root commit. */
3297 		if (s->id1 == NULL) {
3298 			err = write_commit_info(&s->line_offsets, &s->nlines,
3299 			    s->id2, &s->refs, s->repo, s->f);
3300 			if (err)
3301 				goto done;
3302 		} else {
3303 			parent_ids = got_object_commit_get_parent_ids(commit2);
3304 			SIMPLEQ_FOREACH(pid, parent_ids, entry) {
3305 				if (got_object_id_cmp(s->id1, pid->id) == 0) {
3306 					err = write_commit_info(
3307 					    &s->line_offsets, &s->nlines,
3308 					    s->id2, &s->refs, s->repo, s->f);
3309 					if (err)
3310 						goto done;
3311 					break;
3312 				}
3313 			}
3314 		}
3315 		got_object_commit_close(commit2);
3316 
3317 		err = got_diff_objects_as_commits(&s->line_offsets, &s->nlines,
3318 		    s->id1, s->id2, s->diff_context, s->ignore_whitespace,
3319 		    s->force_text_diff, s->repo, s->f);
3320 		break;
3321 	}
3322 	default:
3323 		err = got_error(GOT_ERR_OBJ_TYPE);
3324 		break;
3325 	}
3326 	if (err)
3327 		goto done;
3328 done:
3329 	if (s->f && fflush(s->f) != 0 && err == NULL)
3330 		err = got_error_from_errno("fflush");
3331 	return err;
3332 }
3333 
3334 static void
3335 diff_view_indicate_progress(struct tog_view *view)
3336 {
3337 	mvwaddstr(view->window, 0, 0, "diffing...");
3338 	update_panels();
3339 	doupdate();
3340 }
3341 
3342 static const struct got_error *
3343 search_start_diff_view(struct tog_view *view)
3344 {
3345 	struct tog_diff_view_state *s = &view->state.diff;
3346 
3347 	s->matched_line = 0;
3348 	return NULL;
3349 }
3350 
3351 static const struct got_error *
3352 search_next_diff_view(struct tog_view *view)
3353 {
3354 	struct tog_diff_view_state *s = &view->state.diff;
3355 	int lineno;
3356 
3357 	if (!view->searching) {
3358 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
3359 		return NULL;
3360 	}
3361 
3362 	if (s->matched_line) {
3363 		if (view->searching == TOG_SEARCH_FORWARD)
3364 			lineno = s->matched_line + 1;
3365 		else
3366 			lineno = s->matched_line - 1;
3367 	} else {
3368 		if (view->searching == TOG_SEARCH_FORWARD)
3369 			lineno = 1;
3370 		else
3371 			lineno = s->nlines;
3372 	}
3373 
3374 	while (1) {
3375 		char *line = NULL;
3376 		off_t offset;
3377 		size_t len;
3378 
3379 		if (lineno <= 0 || lineno > s->nlines) {
3380 			if (s->matched_line == 0) {
3381 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
3382 				free(line);
3383 				break;
3384 			}
3385 
3386 			if (view->searching == TOG_SEARCH_FORWARD)
3387 				lineno = 1;
3388 			else
3389 				lineno = s->nlines;
3390 		}
3391 
3392 		offset = s->line_offsets[lineno - 1];
3393 		if (fseeko(s->f, offset, SEEK_SET) != 0) {
3394 			free(line);
3395 			return got_error_from_errno("fseeko");
3396 		}
3397 		free(line);
3398 		line = parse_next_line(s->f, &len);
3399 		if (line &&
3400 		    match_line(line, &view->regex, 1, &view->regmatch)) {
3401 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
3402 			s->matched_line = lineno;
3403 			free(line);
3404 			break;
3405 		}
3406 		free(line);
3407 		if (view->searching == TOG_SEARCH_FORWARD)
3408 			lineno++;
3409 		else
3410 			lineno--;
3411 	}
3412 
3413 	if (s->matched_line) {
3414 		s->first_displayed_line = s->matched_line;
3415 		s->selected_line = 1;
3416 	}
3417 
3418 	return NULL;
3419 }
3420 
3421 static const struct got_error *
3422 open_diff_view(struct tog_view *view, struct got_object_id *id1,
3423     struct got_object_id *id2, const char *label1, const char *label2,
3424     int diff_context, int ignore_whitespace, int force_text_diff,
3425     struct tog_view *log_view, struct got_repository *repo)
3426 {
3427 	const struct got_error *err;
3428 	struct tog_diff_view_state *s = &view->state.diff;
3429 
3430 	SIMPLEQ_INIT(&s->refs);
3431 
3432 	if (id1 != NULL && id2 != NULL) {
3433 	    int type1, type2;
3434 	    err = got_object_get_type(&type1, repo, id1);
3435 	    if (err)
3436 		return err;
3437 	    err = got_object_get_type(&type2, repo, id2);
3438 	    if (err)
3439 		return err;
3440 
3441 	    if (type1 != type2)
3442 		return got_error(GOT_ERR_OBJ_TYPE);
3443 	}
3444 	s->first_displayed_line = 1;
3445 	s->last_displayed_line = view->nlines;
3446 	s->selected_line = 1;
3447 	s->repo = repo;
3448 	s->id1 = id1;
3449 	s->id2 = id2;
3450 	s->label1 = label1;
3451 	s->label2 = label2;
3452 
3453 	if (id1) {
3454 		s->id1 = got_object_id_dup(id1);
3455 		if (s->id1 == NULL)
3456 			return got_error_from_errno("got_object_id_dup");
3457 	} else
3458 		s->id1 = NULL;
3459 
3460 	s->id2 = got_object_id_dup(id2);
3461 	if (s->id2 == NULL) {
3462 		free(s->id1);
3463 		s->id1 = NULL;
3464 		return got_error_from_errno("got_object_id_dup");
3465 	}
3466 	s->f = NULL;
3467 	s->first_displayed_line = 1;
3468 	s->last_displayed_line = view->nlines;
3469 	s->diff_context = diff_context;
3470 	s->ignore_whitespace = ignore_whitespace;
3471 	s->force_text_diff = force_text_diff;
3472 	s->log_view = log_view;
3473 	s->repo = repo;
3474 
3475 	SIMPLEQ_INIT(&s->colors);
3476 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
3477 		err = add_color(&s->colors,
3478 		    "^-", TOG_COLOR_DIFF_MINUS,
3479 		    get_color_value("TOG_COLOR_DIFF_MINUS"));
3480 		if (err)
3481 			return err;
3482 		err = add_color(&s->colors, "^\\+",
3483 		    TOG_COLOR_DIFF_PLUS,
3484 		    get_color_value("TOG_COLOR_DIFF_PLUS"));
3485 		if (err) {
3486 			free_colors(&s->colors);
3487 			return err;
3488 		}
3489 		err = add_color(&s->colors,
3490 		    "^@@", TOG_COLOR_DIFF_CHUNK_HEADER,
3491 		    get_color_value("TOG_COLOR_DIFF_CHUNK_HEADER"));
3492 		if (err) {
3493 			free_colors(&s->colors);
3494 			return err;
3495 		}
3496 
3497 		err = add_color(&s->colors,
3498 		    "^(commit [0-9a-f]|(blob|file) [-+] |[MDmA]  [^ ])",
3499 		    TOG_COLOR_DIFF_META,
3500 		    get_color_value("TOG_COLOR_DIFF_META"));
3501 		if (err) {
3502 			free_colors(&s->colors);
3503 			return err;
3504 		}
3505 
3506 		err = add_color(&s->colors,
3507 		    "^(from|via): ", TOG_COLOR_AUTHOR,
3508 		    get_color_value("TOG_COLOR_AUTHOR"));
3509 		if (err) {
3510 			free_colors(&s->colors);
3511 			return err;
3512 		}
3513 
3514 		err = add_color(&s->colors,
3515 		    "^date: ", TOG_COLOR_DATE,
3516 		    get_color_value("TOG_COLOR_DATE"));
3517 		if (err) {
3518 			free_colors(&s->colors);
3519 			return err;
3520 		}
3521 	}
3522 
3523 	err = got_ref_list(&s->refs, repo, NULL, got_ref_cmp_by_name, NULL);
3524 	if (err) {
3525 		free(s->id1);
3526 		s->id1 = NULL;
3527 		free(s->id2);
3528 		s->id2 = NULL;
3529 		free_colors(&s->colors);
3530 		return err;
3531 	}
3532 
3533 	if (log_view && view_is_splitscreen(view))
3534 		show_log_view(log_view); /* draw vborder */
3535 	diff_view_indicate_progress(view);
3536 
3537 	s->line_offsets = NULL;
3538 	s->nlines = 0;
3539 	err = create_diff(s);
3540 	if (err) {
3541 		free(s->id1);
3542 		s->id1 = NULL;
3543 		free(s->id2);
3544 		s->id2 = NULL;
3545 		free_colors(&s->colors);
3546 		got_ref_list_free(&s->refs);
3547 		return err;
3548 	}
3549 
3550 	view->show = show_diff_view;
3551 	view->input = input_diff_view;
3552 	view->close = close_diff_view;
3553 	view->search_start = search_start_diff_view;
3554 	view->search_next = search_next_diff_view;
3555 
3556 	return NULL;
3557 }
3558 
3559 static const struct got_error *
3560 close_diff_view(struct tog_view *view)
3561 {
3562 	const struct got_error *err = NULL;
3563 	struct tog_diff_view_state *s = &view->state.diff;
3564 
3565 	free(s->id1);
3566 	s->id1 = NULL;
3567 	free(s->id2);
3568 	s->id2 = NULL;
3569 	if (s->f && fclose(s->f) == EOF)
3570 		err = got_error_from_errno("fclose");
3571 	free_colors(&s->colors);
3572 	free(s->line_offsets);
3573 	s->line_offsets = NULL;
3574 	s->nlines = 0;
3575 	got_ref_list_free(&s->refs);
3576 	return err;
3577 }
3578 
3579 static const struct got_error *
3580 show_diff_view(struct tog_view *view)
3581 {
3582 	const struct got_error *err;
3583 	struct tog_diff_view_state *s = &view->state.diff;
3584 	char *id_str1 = NULL, *id_str2, *header;
3585 	const char *label1, *label2;
3586 
3587 	if (s->id1) {
3588 		err = got_object_id_str(&id_str1, s->id1);
3589 		if (err)
3590 			return err;
3591 		label1 = s->label1 ? : id_str1;
3592 	} else
3593 		label1 = "/dev/null";
3594 
3595 	err = got_object_id_str(&id_str2, s->id2);
3596 	if (err)
3597 		return err;
3598 	label2 = s->label2 ? : id_str2;
3599 
3600 	if (asprintf(&header, "diff %s %s", label1, label2) == -1) {
3601 		err = got_error_from_errno("asprintf");
3602 		free(id_str1);
3603 		free(id_str2);
3604 		return err;
3605 	}
3606 	free(id_str1);
3607 	free(id_str2);
3608 
3609 	return draw_file(view, s->f, s->first_displayed_line, s->nlines,
3610 	    s->line_offsets, s->selected_line, view->nlines,
3611 	    &s->last_displayed_line, &s->eof, header, &s->colors,
3612 	    s->matched_line, &view->regmatch);
3613 }
3614 
3615 static const struct got_error *
3616 set_selected_commit(struct tog_diff_view_state *s,
3617     struct commit_queue_entry *entry)
3618 {
3619 	const struct got_error *err;
3620 	const struct got_object_id_queue *parent_ids;
3621 	struct got_commit_object *selected_commit;
3622 	struct got_object_qid *pid;
3623 
3624 	free(s->id2);
3625 	s->id2 = got_object_id_dup(entry->id);
3626 	if (s->id2 == NULL)
3627 		return got_error_from_errno("got_object_id_dup");
3628 
3629 	err = got_object_open_as_commit(&selected_commit, s->repo, entry->id);
3630 	if (err)
3631 		return err;
3632 	parent_ids = got_object_commit_get_parent_ids(selected_commit);
3633 	free(s->id1);
3634 	pid = SIMPLEQ_FIRST(parent_ids);
3635 	s->id1 = pid ? got_object_id_dup(pid->id) : NULL;
3636 	got_object_commit_close(selected_commit);
3637 	return NULL;
3638 }
3639 
3640 static const struct got_error *
3641 input_diff_view(struct tog_view **new_view, struct tog_view **dead_view,
3642     struct tog_view **focus_view, struct tog_view *view, int ch)
3643 {
3644 	const struct got_error *err = NULL;
3645 	struct tog_diff_view_state *s = &view->state.diff;
3646 	struct tog_log_view_state *ls;
3647 	struct commit_queue_entry *entry;
3648 	int i;
3649 
3650 	switch (ch) {
3651 	case 'a':
3652 	case 'w':
3653 		if (ch == 'a')
3654 			s->force_text_diff = !s->force_text_diff;
3655 		if (ch == 'w')
3656 			s->ignore_whitespace = !s->ignore_whitespace;
3657 		wclear(view->window);
3658 		s->first_displayed_line = 1;
3659 		s->last_displayed_line = view->nlines;
3660 		diff_view_indicate_progress(view);
3661 		err = create_diff(s);
3662 		break;
3663 	case 'k':
3664 	case KEY_UP:
3665 		if (s->first_displayed_line > 1)
3666 			s->first_displayed_line--;
3667 		break;
3668 	case KEY_PPAGE:
3669 	case CTRL('b'):
3670 		if (s->first_displayed_line == 1)
3671 			break;
3672 		i = 0;
3673 		while (i++ < view->nlines - 1 &&
3674 		    s->first_displayed_line > 1)
3675 			s->first_displayed_line--;
3676 		break;
3677 	case 'j':
3678 	case KEY_DOWN:
3679 		if (!s->eof)
3680 			s->first_displayed_line++;
3681 		break;
3682 	case KEY_NPAGE:
3683 	case CTRL('f'):
3684 	case ' ':
3685 		if (s->eof)
3686 			break;
3687 		i = 0;
3688 		while (!s->eof && i++ < view->nlines - 1) {
3689 			char *line;
3690 			line = parse_next_line(s->f, NULL);
3691 			s->first_displayed_line++;
3692 			if (line == NULL)
3693 				break;
3694 		}
3695 		break;
3696 	case '[':
3697 		if (s->diff_context > 0) {
3698 			s->diff_context--;
3699 			diff_view_indicate_progress(view);
3700 			err = create_diff(s);
3701 			if (s->first_displayed_line + view->nlines - 1 >
3702 			    s->nlines) {
3703 				s->first_displayed_line = 1;
3704 				s->last_displayed_line = view->nlines;
3705 			}
3706 		}
3707 		break;
3708 	case ']':
3709 		if (s->diff_context < GOT_DIFF_MAX_CONTEXT) {
3710 			s->diff_context++;
3711 			diff_view_indicate_progress(view);
3712 			err = create_diff(s);
3713 		}
3714 		break;
3715 	case '<':
3716 	case ',':
3717 		if (s->log_view == NULL)
3718 			break;
3719 		ls = &s->log_view->state.log;
3720 		entry = TAILQ_PREV(ls->selected_entry,
3721 		    commit_queue_head, entry);
3722 		if (entry == NULL)
3723 			break;
3724 
3725 		err = input_log_view(NULL, NULL, NULL, s->log_view,
3726 		    KEY_UP);
3727 		if (err)
3728 			break;
3729 
3730 		err = set_selected_commit(s, entry);
3731 		if (err)
3732 			break;
3733 
3734 		s->first_displayed_line = 1;
3735 		s->last_displayed_line = view->nlines;
3736 
3737 		diff_view_indicate_progress(view);
3738 		err = create_diff(s);
3739 		break;
3740 	case '>':
3741 	case '.':
3742 		if (s->log_view == NULL)
3743 			break;
3744 		ls = &s->log_view->state.log;
3745 
3746 		if (TAILQ_NEXT(ls->selected_entry, entry) == NULL) {
3747 			ls->thread_args.commits_needed++;
3748 			err = trigger_log_thread(s->log_view, 1,
3749 			    &ls->thread_args.commits_needed,
3750 			    &ls->thread_args.log_complete,
3751 			    &ls->thread_args.need_commits,
3752 			    &ls->thread_args.commit_loaded);
3753 			if (err)
3754 				break;
3755 		}
3756 		err = input_log_view(NULL, NULL, NULL, s->log_view,
3757 		    KEY_DOWN);
3758 		if (err)
3759 			break;
3760 
3761 		entry = TAILQ_NEXT(ls->selected_entry, entry);
3762 		if (entry == NULL)
3763 			break;
3764 
3765 		err = set_selected_commit(s, entry);
3766 		if (err)
3767 			break;
3768 
3769 		s->first_displayed_line = 1;
3770 		s->last_displayed_line = view->nlines;
3771 
3772 		diff_view_indicate_progress(view);
3773 		err = create_diff(s);
3774 		break;
3775 	default:
3776 		break;
3777 	}
3778 
3779 	return err;
3780 }
3781 
3782 static const struct got_error *
3783 cmd_diff(int argc, char *argv[])
3784 {
3785 	const struct got_error *error = NULL;
3786 	struct got_repository *repo = NULL;
3787 	struct got_worktree *worktree = NULL;
3788 	struct got_object_id *id1 = NULL, *id2 = NULL;
3789 	char *repo_path = NULL, *cwd = NULL;
3790 	char *id_str1 = NULL, *id_str2 = NULL;
3791 	char *label1 = NULL, *label2 = NULL;
3792 	int diff_context = 3, ignore_whitespace = 0;
3793 	int ch, force_text_diff = 0;
3794 	const char *errstr;
3795 	struct tog_view *view;
3796 
3797 #ifndef PROFILE
3798 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
3799 	    NULL) == -1)
3800 		err(1, "pledge");
3801 #endif
3802 	while ((ch = getopt(argc, argv, "aC:r:w")) != -1) {
3803 		switch (ch) {
3804 		case 'a':
3805 			force_text_diff = 1;
3806 			break;
3807 		case 'C':
3808 			diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
3809 			    &errstr);
3810 			if (errstr != NULL)
3811 				err(1, "-C option %s", errstr);
3812 			break;
3813 		case 'r':
3814 			repo_path = realpath(optarg, NULL);
3815 			if (repo_path == NULL)
3816 				return got_error_from_errno2("realpath",
3817 				    optarg);
3818 			got_path_strip_trailing_slashes(repo_path);
3819 			break;
3820 		case 'w':
3821 			ignore_whitespace = 1;
3822 			break;
3823 		default:
3824 			usage_diff();
3825 			/* NOTREACHED */
3826 		}
3827 	}
3828 
3829 	argc -= optind;
3830 	argv += optind;
3831 
3832 	if (argc == 0) {
3833 		usage_diff(); /* TODO show local worktree changes */
3834 	} else if (argc == 2) {
3835 		id_str1 = argv[0];
3836 		id_str2 = argv[1];
3837 	} else
3838 		usage_diff();
3839 
3840 	cwd = getcwd(NULL, 0);
3841 	if (cwd == NULL)
3842 		return got_error_from_errno("getcwd");
3843 
3844 	error = got_worktree_open(&worktree, cwd);
3845 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
3846 		goto done;
3847 
3848 	if (repo_path == NULL) {
3849 		if (worktree)
3850 			repo_path =
3851 			    strdup(got_worktree_get_repo_path(worktree));
3852 		else
3853 			repo_path = strdup(cwd);
3854 	}
3855 	if (repo_path == NULL) {
3856 		error = got_error_from_errno("strdup");
3857 		goto done;
3858 	}
3859 
3860 	error = got_repo_open(&repo, repo_path, NULL);
3861 	if (error)
3862 		goto done;
3863 
3864 	init_curses();
3865 
3866 	error = apply_unveil(got_repo_get_path(repo), NULL);
3867 	if (error)
3868 		goto done;
3869 
3870 	error = got_repo_match_object_id(&id1, &label1, id_str1,
3871 	    GOT_OBJ_TYPE_ANY, 1, repo);
3872 	if (error)
3873 		goto done;
3874 
3875 	error = got_repo_match_object_id(&id2, &label2, id_str2,
3876 	    GOT_OBJ_TYPE_ANY, 1, repo);
3877 	if (error)
3878 		goto done;
3879 
3880 	view = view_open(0, 0, 0, 0, TOG_VIEW_DIFF);
3881 	if (view == NULL) {
3882 		error = got_error_from_errno("view_open");
3883 		goto done;
3884 	}
3885 	error = open_diff_view(view, id1, id2, label1, label2, diff_context,
3886 	    ignore_whitespace, force_text_diff, NULL,  repo);
3887 	if (error)
3888 		goto done;
3889 	error = view_loop(view);
3890 done:
3891 	free(label1);
3892 	free(label2);
3893 	free(repo_path);
3894 	free(cwd);
3895 	if (repo)
3896 		got_repo_close(repo);
3897 	if (worktree)
3898 		got_worktree_close(worktree);
3899 	return error;
3900 }
3901 
3902 __dead static void
3903 usage_blame(void)
3904 {
3905 	endwin();
3906 	fprintf(stderr, "usage: %s blame [-c commit] [-r repository-path] path\n",
3907 	    getprogname());
3908 	exit(1);
3909 }
3910 
3911 struct tog_blame_line {
3912 	int annotated;
3913 	struct got_object_id *id;
3914 };
3915 
3916 static const struct got_error *
3917 draw_blame(struct tog_view *view, struct got_object_id *id, FILE *f,
3918     const char *path, struct tog_blame_line *lines, int nlines,
3919     int blame_complete, int selected_line, int *first_displayed_line,
3920     int *last_displayed_line, int *eof, int max_lines,
3921     struct tog_colors *colors, int matched_line, regmatch_t *regmatch)
3922 {
3923 	const struct got_error *err;
3924 	int lineno = 0, nprinted = 0;
3925 	char *line;
3926 	size_t len;
3927 	wchar_t *wline;
3928 	int width;
3929 	struct tog_blame_line *blame_line;
3930 	struct got_object_id *prev_id = NULL;
3931 	char *id_str;
3932 	struct tog_color *tc;
3933 
3934 	err = got_object_id_str(&id_str, id);
3935 	if (err)
3936 		return err;
3937 
3938 	rewind(f);
3939 	werase(view->window);
3940 
3941 	if (asprintf(&line, "commit %s", id_str) == -1) {
3942 		err = got_error_from_errno("asprintf");
3943 		free(id_str);
3944 		return err;
3945 	}
3946 
3947 	err = format_line(&wline, &width, line, view->ncols, 0);
3948 	free(line);
3949 	line = NULL;
3950 	if (err)
3951 		return err;
3952 	if (view_needs_focus_indication(view))
3953 		wstandout(view->window);
3954 	tc = get_color(colors, TOG_COLOR_COMMIT);
3955 	if (tc)
3956 		wattr_on(view->window,
3957 		    COLOR_PAIR(tc->colorpair), NULL);
3958 	waddwstr(view->window, wline);
3959 	if (tc)
3960 		wattr_off(view->window,
3961 		    COLOR_PAIR(tc->colorpair), NULL);
3962 	if (view_needs_focus_indication(view))
3963 		wstandend(view->window);
3964 	free(wline);
3965 	wline = NULL;
3966 	if (width < view->ncols - 1)
3967 		waddch(view->window, '\n');
3968 
3969 	if (asprintf(&line, "[%d/%d] %s%s",
3970 	    *first_displayed_line - 1 + selected_line, nlines,
3971 	    blame_complete ? "" : "annotating... ", path) == -1) {
3972 		free(id_str);
3973 		return got_error_from_errno("asprintf");
3974 	}
3975 	free(id_str);
3976 	err = format_line(&wline, &width, line, view->ncols, 0);
3977 	free(line);
3978 	line = NULL;
3979 	if (err)
3980 		return err;
3981 	waddwstr(view->window, wline);
3982 	free(wline);
3983 	wline = NULL;
3984 	if (width < view->ncols - 1)
3985 		waddch(view->window, '\n');
3986 
3987 	*eof = 0;
3988 	while (nprinted < max_lines - 2) {
3989 		line = parse_next_line(f, &len);
3990 		if (line == NULL) {
3991 			*eof = 1;
3992 			break;
3993 		}
3994 		if (++lineno < *first_displayed_line) {
3995 			free(line);
3996 			continue;
3997 		}
3998 
3999 		if (view->focussed && nprinted == selected_line - 1)
4000 			wstandout(view->window);
4001 
4002 		if (nlines > 0) {
4003 			blame_line = &lines[lineno - 1];
4004 			if (blame_line->annotated && prev_id &&
4005 			    got_object_id_cmp(prev_id, blame_line->id) == 0 &&
4006 			    !(view->focussed &&
4007 			    nprinted == selected_line - 1)) {
4008 				waddstr(view->window, "        ");
4009 			} else if (blame_line->annotated) {
4010 				char *id_str;
4011 				err = got_object_id_str(&id_str, blame_line->id);
4012 				if (err) {
4013 					free(line);
4014 					return err;
4015 				}
4016 				tc = get_color(colors, TOG_COLOR_COMMIT);
4017 				if (tc)
4018 					wattr_on(view->window,
4019 					    COLOR_PAIR(tc->colorpair), NULL);
4020 				wprintw(view->window, "%.8s", id_str);
4021 				if (tc)
4022 					wattr_off(view->window,
4023 					    COLOR_PAIR(tc->colorpair), NULL);
4024 				free(id_str);
4025 				prev_id = blame_line->id;
4026 			} else {
4027 				waddstr(view->window, "........");
4028 				prev_id = NULL;
4029 			}
4030 		} else {
4031 			waddstr(view->window, "........");
4032 			prev_id = NULL;
4033 		}
4034 
4035 		if (view->focussed && nprinted == selected_line - 1)
4036 			wstandend(view->window);
4037 		waddstr(view->window, " ");
4038 
4039 		if (view->ncols <= 9) {
4040 			width = 9;
4041 			wline = wcsdup(L"");
4042 			if (wline == NULL) {
4043 				err = got_error_from_errno("wcsdup");
4044 				free(line);
4045 				return err;
4046 			}
4047 		} else if (*first_displayed_line + nprinted == matched_line &&
4048 		    regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
4049 			err = add_matched_line(&width, line, view->ncols - 9, 9,
4050 			    view->window, regmatch);
4051 			if (err) {
4052 				free(line);
4053 				return err;
4054 			}
4055 			width += 9;
4056 		} else {
4057 			err = format_line(&wline, &width, line,
4058 			    view->ncols - 9, 9);
4059 			waddwstr(view->window, wline);
4060 			free(wline);
4061 			wline = NULL;
4062 			width += 9;
4063 		}
4064 
4065 		if (width <= view->ncols - 1)
4066 			waddch(view->window, '\n');
4067 		if (++nprinted == 1)
4068 			*first_displayed_line = lineno;
4069 		free(line);
4070 	}
4071 	*last_displayed_line = lineno;
4072 
4073 	view_vborder(view);
4074 
4075 	return NULL;
4076 }
4077 
4078 static const struct got_error *
4079 blame_cb(void *arg, int nlines, int lineno, struct got_object_id *id)
4080 {
4081 	const struct got_error *err = NULL;
4082 	struct tog_blame_cb_args *a = arg;
4083 	struct tog_blame_line *line;
4084 	int errcode;
4085 
4086 	if (nlines != a->nlines ||
4087 	    (lineno != -1 && lineno < 1) || lineno > a->nlines)
4088 		return got_error(GOT_ERR_RANGE);
4089 
4090 	errcode = pthread_mutex_lock(&tog_mutex);
4091 	if (errcode)
4092 		return got_error_set_errno(errcode, "pthread_mutex_lock");
4093 
4094 	if (*a->quit) {	/* user has quit the blame view */
4095 		err = got_error(GOT_ERR_ITER_COMPLETED);
4096 		goto done;
4097 	}
4098 
4099 	if (lineno == -1)
4100 		goto done; /* no change in this commit */
4101 
4102 	line = &a->lines[lineno - 1];
4103 	if (line->annotated)
4104 		goto done;
4105 
4106 	line->id = got_object_id_dup(id);
4107 	if (line->id == NULL) {
4108 		err = got_error_from_errno("got_object_id_dup");
4109 		goto done;
4110 	}
4111 	line->annotated = 1;
4112 done:
4113 	errcode = pthread_mutex_unlock(&tog_mutex);
4114 	if (errcode)
4115 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
4116 	return err;
4117 }
4118 
4119 static void *
4120 blame_thread(void *arg)
4121 {
4122 	const struct got_error *err;
4123 	struct tog_blame_thread_args *ta = arg;
4124 	struct tog_blame_cb_args *a = ta->cb_args;
4125 	int errcode;
4126 
4127 	err = block_signals_used_by_main_thread();
4128 	if (err)
4129 		return (void *)err;
4130 
4131 	err = got_blame(ta->path, a->commit_id, ta->repo,
4132 	    blame_cb, ta->cb_args, ta->cancel_cb, ta->cancel_arg);
4133 	if (err && err->code == GOT_ERR_CANCELLED)
4134 		err = NULL;
4135 
4136 	errcode = pthread_mutex_lock(&tog_mutex);
4137 	if (errcode)
4138 		return (void *)got_error_set_errno(errcode,
4139 		    "pthread_mutex_lock");
4140 
4141 	got_repo_close(ta->repo);
4142 	ta->repo = NULL;
4143 	*ta->complete = 1;
4144 
4145 	errcode = pthread_mutex_unlock(&tog_mutex);
4146 	if (errcode && err == NULL)
4147 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
4148 
4149 	return (void *)err;
4150 }
4151 
4152 static struct got_object_id *
4153 get_selected_commit_id(struct tog_blame_line *lines, int nlines,
4154     int first_displayed_line, int selected_line)
4155 {
4156 	struct tog_blame_line *line;
4157 
4158 	if (nlines <= 0)
4159 		return NULL;
4160 
4161 	line = &lines[first_displayed_line - 1 + selected_line - 1];
4162 	if (!line->annotated)
4163 		return NULL;
4164 
4165 	return line->id;
4166 }
4167 
4168 static const struct got_error *
4169 stop_blame(struct tog_blame *blame)
4170 {
4171 	const struct got_error *err = NULL;
4172 	int i;
4173 
4174 	if (blame->thread) {
4175 		int errcode;
4176 		errcode = pthread_mutex_unlock(&tog_mutex);
4177 		if (errcode)
4178 			return got_error_set_errno(errcode,
4179 			    "pthread_mutex_unlock");
4180 		errcode = pthread_join(blame->thread, (void **)&err);
4181 		if (errcode)
4182 			return got_error_set_errno(errcode, "pthread_join");
4183 		errcode = pthread_mutex_lock(&tog_mutex);
4184 		if (errcode)
4185 			return got_error_set_errno(errcode,
4186 			    "pthread_mutex_lock");
4187 		if (err && err->code == GOT_ERR_ITER_COMPLETED)
4188 			err = NULL;
4189 		blame->thread = NULL;
4190 	}
4191 	if (blame->thread_args.repo) {
4192 		got_repo_close(blame->thread_args.repo);
4193 		blame->thread_args.repo = NULL;
4194 	}
4195 	if (blame->f) {
4196 		if (fclose(blame->f) != 0 && err == NULL)
4197 			err = got_error_from_errno("fclose");
4198 		blame->f = NULL;
4199 	}
4200 	if (blame->lines) {
4201 		for (i = 0; i < blame->nlines; i++)
4202 			free(blame->lines[i].id);
4203 		free(blame->lines);
4204 		blame->lines = NULL;
4205 	}
4206 	free(blame->cb_args.commit_id);
4207 	blame->cb_args.commit_id = NULL;
4208 
4209 	return err;
4210 }
4211 
4212 static const struct got_error *
4213 cancel_blame_view(void *arg)
4214 {
4215 	const struct got_error *err = NULL;
4216 	int *done = arg;
4217 	int errcode;
4218 
4219 	errcode = pthread_mutex_lock(&tog_mutex);
4220 	if (errcode)
4221 		return got_error_set_errno(errcode,
4222 		    "pthread_mutex_unlock");
4223 
4224 	if (*done)
4225 		err = got_error(GOT_ERR_CANCELLED);
4226 
4227 	errcode = pthread_mutex_unlock(&tog_mutex);
4228 	if (errcode)
4229 		return got_error_set_errno(errcode,
4230 		    "pthread_mutex_lock");
4231 
4232 	return err;
4233 }
4234 
4235 static const struct got_error *
4236 run_blame(struct tog_blame *blame, struct tog_view *view, int *blame_complete,
4237     int *first_displayed_line, int *last_displayed_line, int *selected_line,
4238     int *done, int *eof, const char *path, struct got_object_id *commit_id,
4239     struct got_repository *repo)
4240 {
4241 	const struct got_error *err = NULL;
4242 	struct got_blob_object *blob = NULL;
4243 	struct got_repository *thread_repo = NULL;
4244 	struct got_object_id *obj_id = NULL;
4245 	int obj_type;
4246 
4247 	err = got_object_id_by_path(&obj_id, repo, commit_id, path);
4248 	if (err)
4249 		return err;
4250 
4251 	err = got_object_get_type(&obj_type, repo, obj_id);
4252 	if (err)
4253 		goto done;
4254 
4255 	if (obj_type != GOT_OBJ_TYPE_BLOB) {
4256 		err = got_error(GOT_ERR_OBJ_TYPE);
4257 		goto done;
4258 	}
4259 
4260 	err = got_object_open_as_blob(&blob, repo, obj_id, 8192);
4261 	if (err)
4262 		goto done;
4263 	blame->f = got_opentemp();
4264 	if (blame->f == NULL) {
4265 		err = got_error_from_errno("got_opentemp");
4266 		goto done;
4267 	}
4268 	err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
4269 	    &blame->line_offsets, blame->f, blob);
4270 	if (err || blame->nlines == 0)
4271 		goto done;
4272 
4273 	/* Don't include \n at EOF in the blame line count. */
4274 	if (blame->line_offsets[blame->nlines - 1] == blame->filesize)
4275 		blame->nlines--;
4276 
4277 	blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
4278 	if (blame->lines == NULL) {
4279 		err = got_error_from_errno("calloc");
4280 		goto done;
4281 	}
4282 
4283 	err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL);
4284 	if (err)
4285 		goto done;
4286 
4287 	blame->cb_args.view = view;
4288 	blame->cb_args.lines = blame->lines;
4289 	blame->cb_args.nlines = blame->nlines;
4290 	blame->cb_args.commit_id = got_object_id_dup(commit_id);
4291 	if (blame->cb_args.commit_id == NULL) {
4292 		err = got_error_from_errno("got_object_id_dup");
4293 		goto done;
4294 	}
4295 	blame->cb_args.quit = done;
4296 
4297 	blame->thread_args.path = path;
4298 	blame->thread_args.repo = thread_repo;
4299 	blame->thread_args.cb_args = &blame->cb_args;
4300 	blame->thread_args.complete = blame_complete;
4301 	blame->thread_args.cancel_cb = cancel_blame_view;
4302 	blame->thread_args.cancel_arg = done;
4303 	*blame_complete = 0;
4304 
4305 done:
4306 	if (blob)
4307 		got_object_blob_close(blob);
4308 	free(obj_id);
4309 	if (err)
4310 		stop_blame(blame);
4311 	return err;
4312 }
4313 
4314 static const struct got_error *
4315 open_blame_view(struct tog_view *view, char *path,
4316     struct got_object_id *commit_id, struct got_repository *repo)
4317 {
4318 	const struct got_error *err = NULL;
4319 	struct tog_blame_view_state *s = &view->state.blame;
4320 
4321 	SIMPLEQ_INIT(&s->blamed_commits);
4322 
4323 	s->path = strdup(path);
4324 	if (s->path == NULL)
4325 		return got_error_from_errno("strdup");
4326 
4327 	err = got_object_qid_alloc(&s->blamed_commit, commit_id);
4328 	if (err) {
4329 		free(s->path);
4330 		return err;
4331 	}
4332 
4333 	SIMPLEQ_INSERT_HEAD(&s->blamed_commits, s->blamed_commit, entry);
4334 	s->first_displayed_line = 1;
4335 	s->last_displayed_line = view->nlines;
4336 	s->selected_line = 1;
4337 	s->blame_complete = 0;
4338 	s->repo = repo;
4339 	s->commit_id = commit_id;
4340 	memset(&s->blame, 0, sizeof(s->blame));
4341 
4342 	SIMPLEQ_INIT(&s->colors);
4343 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
4344 		err = add_color(&s->colors, "^", TOG_COLOR_COMMIT,
4345 		    get_color_value("TOG_COLOR_COMMIT"));
4346 		if (err)
4347 			return err;
4348 	}
4349 
4350 	view->show = show_blame_view;
4351 	view->input = input_blame_view;
4352 	view->close = close_blame_view;
4353 	view->search_start = search_start_blame_view;
4354 	view->search_next = search_next_blame_view;
4355 
4356 	return run_blame(&s->blame, view, &s->blame_complete,
4357 	    &s->first_displayed_line, &s->last_displayed_line,
4358 	    &s->selected_line, &s->done, &s->eof, s->path,
4359 	    s->blamed_commit->id, s->repo);
4360 }
4361 
4362 static const struct got_error *
4363 close_blame_view(struct tog_view *view)
4364 {
4365 	const struct got_error *err = NULL;
4366 	struct tog_blame_view_state *s = &view->state.blame;
4367 
4368 	if (s->blame.thread)
4369 		err = stop_blame(&s->blame);
4370 
4371 	while (!SIMPLEQ_EMPTY(&s->blamed_commits)) {
4372 		struct got_object_qid *blamed_commit;
4373 		blamed_commit = SIMPLEQ_FIRST(&s->blamed_commits);
4374 		SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
4375 		got_object_qid_free(blamed_commit);
4376 	}
4377 
4378 	free(s->path);
4379 	free_colors(&s->colors);
4380 
4381 	return err;
4382 }
4383 
4384 static const struct got_error *
4385 search_start_blame_view(struct tog_view *view)
4386 {
4387 	struct tog_blame_view_state *s = &view->state.blame;
4388 
4389 	s->matched_line = 0;
4390 	return NULL;
4391 }
4392 
4393 static const struct got_error *
4394 search_next_blame_view(struct tog_view *view)
4395 {
4396 	struct tog_blame_view_state *s = &view->state.blame;
4397 	int lineno;
4398 
4399 	if (!view->searching) {
4400 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
4401 		return NULL;
4402 	}
4403 
4404 	if (s->matched_line) {
4405 		if (view->searching == TOG_SEARCH_FORWARD)
4406 			lineno = s->matched_line + 1;
4407 		else
4408 			lineno = s->matched_line - 1;
4409 	} else {
4410 		if (view->searching == TOG_SEARCH_FORWARD)
4411 			lineno = 1;
4412 		else
4413 			lineno = s->blame.nlines;
4414 	}
4415 
4416 	while (1) {
4417 		char *line = NULL;
4418 		off_t offset;
4419 		size_t len;
4420 
4421 		if (lineno <= 0 || lineno > s->blame.nlines) {
4422 			if (s->matched_line == 0) {
4423 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
4424 				free(line);
4425 				break;
4426 			}
4427 
4428 			if (view->searching == TOG_SEARCH_FORWARD)
4429 				lineno = 1;
4430 			else
4431 				lineno = s->blame.nlines;
4432 		}
4433 
4434 		offset = s->blame.line_offsets[lineno - 1];
4435 		if (fseeko(s->blame.f, offset, SEEK_SET) != 0) {
4436 			free(line);
4437 			return got_error_from_errno("fseeko");
4438 		}
4439 		free(line);
4440 		line = parse_next_line(s->blame.f, &len);
4441 		if (line &&
4442 		    match_line(line, &view->regex, 1, &view->regmatch)) {
4443 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
4444 			s->matched_line = lineno;
4445 			free(line);
4446 			break;
4447 		}
4448 		free(line);
4449 		if (view->searching == TOG_SEARCH_FORWARD)
4450 			lineno++;
4451 		else
4452 			lineno--;
4453 	}
4454 
4455 	if (s->matched_line) {
4456 		s->first_displayed_line = s->matched_line;
4457 		s->selected_line = 1;
4458 	}
4459 
4460 	return NULL;
4461 }
4462 
4463 static const struct got_error *
4464 show_blame_view(struct tog_view *view)
4465 {
4466 	const struct got_error *err = NULL;
4467 	struct tog_blame_view_state *s = &view->state.blame;
4468 	int errcode;
4469 
4470 	if (s->blame.thread == NULL) {
4471 		errcode = pthread_create(&s->blame.thread, NULL, blame_thread,
4472 		    &s->blame.thread_args);
4473 		if (errcode)
4474 			return got_error_set_errno(errcode, "pthread_create");
4475 
4476 		halfdelay(1); /* fast refresh while annotating  */
4477 	}
4478 
4479 	if (s->blame_complete)
4480 		halfdelay(10); /* disable fast refresh */
4481 
4482 	err = draw_blame(view, s->blamed_commit->id, s->blame.f,
4483 	    s->path, s->blame.lines, s->blame.nlines, s->blame_complete,
4484 	    s->selected_line, &s->first_displayed_line,
4485 	    &s->last_displayed_line, &s->eof, view->nlines, &s->colors,
4486 	    s->matched_line, &view->regmatch);
4487 
4488 	view_vborder(view);
4489 	return err;
4490 }
4491 
4492 static const struct got_error *
4493 input_blame_view(struct tog_view **new_view, struct tog_view **dead_view,
4494     struct tog_view **focus_view, struct tog_view *view, int ch)
4495 {
4496 	const struct got_error *err = NULL, *thread_err = NULL;
4497 	struct tog_view *diff_view;
4498 	struct tog_blame_view_state *s = &view->state.blame;
4499 	int begin_x = 0;
4500 
4501 	switch (ch) {
4502 	case 'q':
4503 		s->done = 1;
4504 		break;
4505 	case 'k':
4506 	case KEY_UP:
4507 		if (s->selected_line > 1)
4508 			s->selected_line--;
4509 		else if (s->selected_line == 1 &&
4510 		    s->first_displayed_line > 1)
4511 			s->first_displayed_line--;
4512 		break;
4513 	case KEY_PPAGE:
4514 	case CTRL('b'):
4515 		if (s->first_displayed_line == 1) {
4516 			s->selected_line = 1;
4517 			break;
4518 		}
4519 		if (s->first_displayed_line > view->nlines - 2)
4520 			s->first_displayed_line -=
4521 			    (view->nlines - 2);
4522 		else
4523 			s->first_displayed_line = 1;
4524 		break;
4525 	case 'j':
4526 	case KEY_DOWN:
4527 		if (s->selected_line < view->nlines - 2 &&
4528 		    s->first_displayed_line +
4529 		    s->selected_line <= s->blame.nlines)
4530 			s->selected_line++;
4531 		else if (s->last_displayed_line <
4532 		    s->blame.nlines)
4533 			s->first_displayed_line++;
4534 		break;
4535 	case 'b':
4536 	case 'p': {
4537 		struct got_object_id *id = NULL;
4538 		id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
4539 		    s->first_displayed_line, s->selected_line);
4540 		if (id == NULL)
4541 			break;
4542 		if (ch == 'p') {
4543 			struct got_commit_object *commit;
4544 			struct got_object_qid *pid;
4545 			struct got_object_id *blob_id = NULL;
4546 			int obj_type;
4547 			err = got_object_open_as_commit(&commit,
4548 			    s->repo, id);
4549 			if (err)
4550 				break;
4551 			pid = SIMPLEQ_FIRST(
4552 			    got_object_commit_get_parent_ids(commit));
4553 			if (pid == NULL) {
4554 				got_object_commit_close(commit);
4555 				break;
4556 			}
4557 			/* Check if path history ends here. */
4558 			err = got_object_id_by_path(&blob_id, s->repo,
4559 			    pid->id, s->path);
4560 			if (err) {
4561 				if (err->code == GOT_ERR_NO_TREE_ENTRY)
4562 					err = NULL;
4563 				got_object_commit_close(commit);
4564 				break;
4565 			}
4566 			err = got_object_get_type(&obj_type, s->repo,
4567 			    blob_id);
4568 			free(blob_id);
4569 			/* Can't blame non-blob type objects. */
4570 			if (obj_type != GOT_OBJ_TYPE_BLOB) {
4571 				got_object_commit_close(commit);
4572 				break;
4573 			}
4574 			err = got_object_qid_alloc(&s->blamed_commit,
4575 			    pid->id);
4576 			got_object_commit_close(commit);
4577 		} else {
4578 			if (got_object_id_cmp(id,
4579 			    s->blamed_commit->id) == 0)
4580 				break;
4581 			err = got_object_qid_alloc(&s->blamed_commit,
4582 			    id);
4583 		}
4584 		if (err)
4585 			break;
4586 		s->done = 1;
4587 		thread_err = stop_blame(&s->blame);
4588 		s->done = 0;
4589 		if (thread_err)
4590 			break;
4591 		SIMPLEQ_INSERT_HEAD(&s->blamed_commits,
4592 		    s->blamed_commit, entry);
4593 		err = run_blame(&s->blame, view, &s->blame_complete,
4594 		    &s->first_displayed_line, &s->last_displayed_line,
4595 		    &s->selected_line, &s->done, &s->eof,
4596 		    s->path, s->blamed_commit->id, s->repo);
4597 		if (err)
4598 			break;
4599 		break;
4600 	}
4601 	case 'B': {
4602 		struct got_object_qid *first;
4603 		first = SIMPLEQ_FIRST(&s->blamed_commits);
4604 		if (!got_object_id_cmp(first->id, s->commit_id))
4605 			break;
4606 		s->done = 1;
4607 		thread_err = stop_blame(&s->blame);
4608 		s->done = 0;
4609 		if (thread_err)
4610 			break;
4611 		SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
4612 		got_object_qid_free(s->blamed_commit);
4613 		s->blamed_commit =
4614 		    SIMPLEQ_FIRST(&s->blamed_commits);
4615 		err = run_blame(&s->blame, view, &s->blame_complete,
4616 		    &s->first_displayed_line, &s->last_displayed_line,
4617 		    &s->selected_line, &s->done, &s->eof, s->path,
4618 		    s->blamed_commit->id, s->repo);
4619 		if (err)
4620 			break;
4621 		break;
4622 	}
4623 	case KEY_ENTER:
4624 	case '\r': {
4625 		struct got_object_id *id = NULL;
4626 		struct got_object_qid *pid;
4627 		struct got_commit_object *commit = NULL;
4628 		id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
4629 		    s->first_displayed_line, s->selected_line);
4630 		if (id == NULL)
4631 			break;
4632 		err = got_object_open_as_commit(&commit, s->repo, id);
4633 		if (err)
4634 			break;
4635 		pid = SIMPLEQ_FIRST(
4636 		    got_object_commit_get_parent_ids(commit));
4637 		if (view_is_parent_view(view))
4638 		    begin_x = view_split_begin_x(view->begin_x);
4639 		diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
4640 		if (diff_view == NULL) {
4641 			got_object_commit_close(commit);
4642 			err = got_error_from_errno("view_open");
4643 			break;
4644 		}
4645 		err = open_diff_view(diff_view, pid ? pid->id : NULL,
4646 		    id, NULL, NULL, 3, 0, 0, NULL, s->repo);
4647 		got_object_commit_close(commit);
4648 		if (err) {
4649 			view_close(diff_view);
4650 			break;
4651 		}
4652 		if (view_is_parent_view(view)) {
4653 			err = view_close_child(view);
4654 			if (err)
4655 				break;
4656 			err = view_set_child(view, diff_view);
4657 			if (err) {
4658 				view_close(diff_view);
4659 				break;
4660 			}
4661 			*focus_view = diff_view;
4662 			view->child_focussed = 1;
4663 		} else
4664 			*new_view = diff_view;
4665 		if (err)
4666 			break;
4667 		break;
4668 	}
4669 	case KEY_NPAGE:
4670 	case CTRL('f'):
4671 	case ' ':
4672 		if (s->last_displayed_line >= s->blame.nlines &&
4673 		    s->selected_line >= MIN(s->blame.nlines,
4674 		    view->nlines - 2)) {
4675 			break;
4676 		}
4677 		if (s->last_displayed_line >= s->blame.nlines &&
4678 		    s->selected_line < view->nlines - 2) {
4679 			s->selected_line = MIN(s->blame.nlines,
4680 			    view->nlines - 2);
4681 			break;
4682 		}
4683 		if (s->last_displayed_line + view->nlines - 2
4684 		    <= s->blame.nlines)
4685 			s->first_displayed_line +=
4686 			    view->nlines - 2;
4687 		else
4688 			s->first_displayed_line =
4689 			    s->blame.nlines -
4690 			    (view->nlines - 3);
4691 		break;
4692 	case KEY_RESIZE:
4693 		if (s->selected_line > view->nlines - 2) {
4694 			s->selected_line = MIN(s->blame.nlines,
4695 			    view->nlines - 2);
4696 		}
4697 		break;
4698 	default:
4699 		break;
4700 	}
4701 	return thread_err ? thread_err : err;
4702 }
4703 
4704 static const struct got_error *
4705 cmd_blame(int argc, char *argv[])
4706 {
4707 	const struct got_error *error;
4708 	struct got_repository *repo = NULL;
4709 	struct got_worktree *worktree = NULL;
4710 	char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
4711 	char *link_target = NULL;
4712 	struct got_object_id *commit_id = NULL;
4713 	char *commit_id_str = NULL;
4714 	int ch;
4715 	struct tog_view *view;
4716 
4717 #ifndef PROFILE
4718 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
4719 	    NULL) == -1)
4720 		err(1, "pledge");
4721 #endif
4722 
4723 	while ((ch = getopt(argc, argv, "c:r:")) != -1) {
4724 		switch (ch) {
4725 		case 'c':
4726 			commit_id_str = optarg;
4727 			break;
4728 		case 'r':
4729 			repo_path = realpath(optarg, NULL);
4730 			if (repo_path == NULL)
4731 				return got_error_from_errno2("realpath",
4732 				    optarg);
4733 			break;
4734 		default:
4735 			usage_blame();
4736 			/* NOTREACHED */
4737 		}
4738 	}
4739 
4740 	argc -= optind;
4741 	argv += optind;
4742 
4743 	if (argc != 1)
4744 		usage_blame();
4745 
4746 	cwd = getcwd(NULL, 0);
4747 	if (cwd == NULL)
4748 		return got_error_from_errno("getcwd");
4749 
4750 	error = got_worktree_open(&worktree, cwd);
4751 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
4752 		goto done;
4753 
4754 	if (repo_path == NULL) {
4755 		if (worktree)
4756 			repo_path =
4757 			    strdup(got_worktree_get_repo_path(worktree));
4758 		else
4759 			repo_path = strdup(cwd);
4760 	}
4761 	if (repo_path == NULL) {
4762 		error = got_error_from_errno("strdup");
4763 		goto done;
4764 	}
4765 
4766 	error = got_repo_open(&repo, repo_path, NULL);
4767 	if (error != NULL)
4768 		goto done;
4769 
4770 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv, repo,
4771 	    worktree);
4772 	if (error)
4773 		goto done;
4774 
4775 	init_curses();
4776 
4777 	error = apply_unveil(got_repo_get_path(repo), NULL);
4778 	if (error)
4779 		goto done;
4780 
4781 	if (commit_id_str == NULL) {
4782 		struct got_reference *head_ref;
4783 		error = got_ref_open(&head_ref, repo, worktree ?
4784 		    got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD, 0);
4785 		if (error != NULL)
4786 			goto done;
4787 		error = got_ref_resolve(&commit_id, repo, head_ref);
4788 		got_ref_close(head_ref);
4789 	} else {
4790 		error = got_repo_match_object_id(&commit_id, NULL,
4791 		    commit_id_str, GOT_OBJ_TYPE_COMMIT, 1, repo);
4792 	}
4793 	if (error != NULL)
4794 		goto done;
4795 
4796 	view = view_open(0, 0, 0, 0, TOG_VIEW_BLAME);
4797 	if (view == NULL) {
4798 		error = got_error_from_errno("view_open");
4799 		goto done;
4800 	}
4801 
4802 	error = got_object_resolve_symlinks(&link_target, in_repo_path,
4803 	    commit_id, repo);
4804 	if (error)
4805 		goto done;
4806 
4807 	error = open_blame_view(view, link_target ? link_target : in_repo_path,
4808 	    commit_id, repo);
4809 	if (error)
4810 		goto done;
4811 	if (worktree) {
4812 		/* Release work tree lock. */
4813 		got_worktree_close(worktree);
4814 		worktree = NULL;
4815 	}
4816 	error = view_loop(view);
4817 done:
4818 	free(repo_path);
4819 	free(in_repo_path);
4820 	free(link_target);
4821 	free(cwd);
4822 	free(commit_id);
4823 	if (worktree)
4824 		got_worktree_close(worktree);
4825 	if (repo)
4826 		got_repo_close(repo);
4827 	return error;
4828 }
4829 
4830 static const struct got_error *
4831 draw_tree_entries(struct tog_view *view,
4832     struct got_tree_entry **first_displayed_entry,
4833     struct got_tree_entry **last_displayed_entry,
4834     struct got_tree_entry **selected_entry, int *ndisplayed,
4835     const char *label, int show_ids, const char *parent_path,
4836     struct got_tree_object *tree, int selected, int limit,
4837     int isroot, struct tog_colors *colors, struct got_repository *repo)
4838 {
4839 	const struct got_error *err = NULL;
4840 	struct got_tree_entry *te;
4841 	wchar_t *wline;
4842 	struct tog_color *tc;
4843 	int width, n, i, nentries;
4844 
4845 	*ndisplayed = 0;
4846 
4847 	werase(view->window);
4848 
4849 	if (limit == 0)
4850 		return NULL;
4851 
4852 	err = format_line(&wline, &width, label, view->ncols, 0);
4853 	if (err)
4854 		return err;
4855 	if (view_needs_focus_indication(view))
4856 		wstandout(view->window);
4857 	tc = get_color(colors, TOG_COLOR_COMMIT);
4858 	if (tc)
4859 		wattr_on(view->window,
4860 		    COLOR_PAIR(tc->colorpair), NULL);
4861 	waddwstr(view->window, wline);
4862 	if (tc)
4863 		wattr_off(view->window,
4864 		    COLOR_PAIR(tc->colorpair), NULL);
4865 	if (view_needs_focus_indication(view))
4866 		wstandend(view->window);
4867 	free(wline);
4868 	wline = NULL;
4869 	if (width < view->ncols - 1)
4870 		waddch(view->window, '\n');
4871 	if (--limit <= 0)
4872 		return NULL;
4873 	err = format_line(&wline, &width, parent_path, view->ncols, 0);
4874 	if (err)
4875 		return err;
4876 	waddwstr(view->window, wline);
4877 	free(wline);
4878 	wline = NULL;
4879 	if (width < view->ncols - 1)
4880 		waddch(view->window, '\n');
4881 	if (--limit <= 0)
4882 		return NULL;
4883 	waddch(view->window, '\n');
4884 	if (--limit <= 0)
4885 		return NULL;
4886 
4887 	if (*first_displayed_entry == NULL) {
4888 		te = got_object_tree_get_first_entry(tree);
4889 		if (selected == 0) {
4890 			if (view->focussed)
4891 				wstandout(view->window);
4892 			*selected_entry = NULL;
4893 		}
4894 		waddstr(view->window, "  ..\n");	/* parent directory */
4895 		if (selected == 0 && view->focussed)
4896 			wstandend(view->window);
4897 		(*ndisplayed)++;
4898 		if (--limit <= 0)
4899 			return NULL;
4900 		n = 1;
4901 	} else {
4902 		n = 0;
4903 		te = *first_displayed_entry;
4904 	}
4905 
4906 	nentries = got_object_tree_get_nentries(tree);
4907 	for (i = got_tree_entry_get_index(te); i < nentries; i++) {
4908 		char *line = NULL, *id_str = NULL, *link_target = NULL;
4909 		const char *modestr = "";
4910 		mode_t mode;
4911 
4912 		te = got_object_tree_get_entry(tree, i);
4913 		mode = got_tree_entry_get_mode(te);
4914 
4915 		if (show_ids) {
4916 			err = got_object_id_str(&id_str,
4917 			    got_tree_entry_get_id(te));
4918 			if (err)
4919 				return got_error_from_errno(
4920 				    "got_object_id_str");
4921 		}
4922 		if (got_object_tree_entry_is_submodule(te))
4923 			modestr = "$";
4924 		else if (S_ISLNK(mode)) {
4925 			int i;
4926 
4927 			err = got_tree_entry_get_symlink_target(&link_target,
4928 			    te, repo);
4929 			if (err) {
4930 				free(id_str);
4931 				return err;
4932 			}
4933 			for (i = 0; i < strlen(link_target); i++) {
4934 				if (!isprint((unsigned char)link_target[i]))
4935 					link_target[i] = '?';
4936 			}
4937 			modestr = "@";
4938 		}
4939 		else if (S_ISDIR(mode))
4940 			modestr = "/";
4941 		else if (mode & S_IXUSR)
4942 			modestr = "*";
4943 		if (asprintf(&line, "%s  %s%s%s%s", id_str ? id_str : "",
4944 		    got_tree_entry_get_name(te), modestr,
4945 		    link_target ? " -> ": "",
4946 		    link_target ? link_target : "") == -1) {
4947 			free(id_str);
4948 			free(link_target);
4949 			return got_error_from_errno("asprintf");
4950 		}
4951 		free(id_str);
4952 		free(link_target);
4953 		err = format_line(&wline, &width, line, view->ncols, 0);
4954 		if (err) {
4955 			free(line);
4956 			break;
4957 		}
4958 		if (n == selected) {
4959 			if (view->focussed)
4960 				wstandout(view->window);
4961 			*selected_entry = te;
4962 		}
4963 		tc = match_color(colors, line);
4964 		if (tc)
4965 			wattr_on(view->window,
4966 			    COLOR_PAIR(tc->colorpair), NULL);
4967 		waddwstr(view->window, wline);
4968 		if (tc)
4969 			wattr_off(view->window,
4970 			    COLOR_PAIR(tc->colorpair), NULL);
4971 		if (width < view->ncols - 1)
4972 			waddch(view->window, '\n');
4973 		if (n == selected && view->focussed)
4974 			wstandend(view->window);
4975 		free(line);
4976 		free(wline);
4977 		wline = NULL;
4978 		n++;
4979 		(*ndisplayed)++;
4980 		*last_displayed_entry = te;
4981 		if (--limit <= 0)
4982 			break;
4983 	}
4984 
4985 	return err;
4986 }
4987 
4988 static void
4989 tree_scroll_up(struct tog_view *view,
4990     struct got_tree_entry **first_displayed_entry, int maxscroll,
4991     struct got_tree_object *tree, int isroot)
4992 {
4993 	struct got_tree_entry *te;
4994 	int i = 0;
4995 
4996 	if (*first_displayed_entry == NULL)
4997 		return;
4998 
4999 	te = got_tree_entry_get_prev(tree, *first_displayed_entry);
5000 	while (i++ < maxscroll) {
5001 		if (te == NULL) {
5002 			if (!isroot)
5003 				*first_displayed_entry = NULL;
5004 			break;
5005 		}
5006 		*first_displayed_entry = te;
5007 		te = got_tree_entry_get_prev(tree, te);
5008 	}
5009 }
5010 
5011 static void
5012 tree_scroll_down(struct got_tree_entry **first_displayed_entry, int maxscroll,
5013 	struct got_tree_entry *last_displayed_entry,
5014 	struct got_tree_object *tree)
5015 {
5016 	struct got_tree_entry *next, *last;
5017 	int n = 0;
5018 
5019 	if (*first_displayed_entry)
5020 		next = got_tree_entry_get_next(tree, *first_displayed_entry);
5021 	else
5022 		next = got_object_tree_get_first_entry(tree);
5023 
5024 	last = last_displayed_entry;
5025 	while (next && last && n++ < maxscroll) {
5026 		last = got_tree_entry_get_next(tree, last);
5027 		if (last) {
5028 			*first_displayed_entry = next;
5029 			next = got_tree_entry_get_next(tree, next);
5030 		}
5031 	}
5032 }
5033 
5034 static const struct got_error *
5035 tree_entry_path(char **path, struct tog_parent_trees *parents,
5036     struct got_tree_entry *te)
5037 {
5038 	const struct got_error *err = NULL;
5039 	struct tog_parent_tree *pt;
5040 	size_t len = 2; /* for leading slash and NUL */
5041 
5042 	TAILQ_FOREACH(pt, parents, entry)
5043 		len += strlen(got_tree_entry_get_name(pt->selected_entry))
5044 		    + 1 /* slash */;
5045 	if (te)
5046 		len += strlen(got_tree_entry_get_name(te));
5047 
5048 	*path = calloc(1, len);
5049 	if (path == NULL)
5050 		return got_error_from_errno("calloc");
5051 
5052 	(*path)[0] = '/';
5053 	pt = TAILQ_LAST(parents, tog_parent_trees);
5054 	while (pt) {
5055 		const char *name = got_tree_entry_get_name(pt->selected_entry);
5056 		if (strlcat(*path, name, len) >= len) {
5057 			err = got_error(GOT_ERR_NO_SPACE);
5058 			goto done;
5059 		}
5060 		if (strlcat(*path, "/", len) >= len) {
5061 			err = got_error(GOT_ERR_NO_SPACE);
5062 			goto done;
5063 		}
5064 		pt = TAILQ_PREV(pt, tog_parent_trees, entry);
5065 	}
5066 	if (te) {
5067 		if (strlcat(*path, got_tree_entry_get_name(te), len) >= len) {
5068 			err = got_error(GOT_ERR_NO_SPACE);
5069 			goto done;
5070 		}
5071 	}
5072 done:
5073 	if (err) {
5074 		free(*path);
5075 		*path = NULL;
5076 	}
5077 	return err;
5078 }
5079 
5080 static const struct got_error *
5081 blame_tree_entry(struct tog_view **new_view, int begin_x,
5082     struct got_tree_entry *te, struct tog_parent_trees *parents,
5083     struct got_object_id *commit_id, struct got_repository *repo)
5084 {
5085 	const struct got_error *err = NULL;
5086 	char *path;
5087 	struct tog_view *blame_view;
5088 
5089 	*new_view = NULL;
5090 
5091 	err = tree_entry_path(&path, parents, te);
5092 	if (err)
5093 		return err;
5094 
5095 	blame_view = view_open(0, 0, 0, begin_x, TOG_VIEW_BLAME);
5096 	if (blame_view == NULL) {
5097 		err = got_error_from_errno("view_open");
5098 		goto done;
5099 	}
5100 
5101 	err = open_blame_view(blame_view, path, commit_id, repo);
5102 	if (err) {
5103 		if (err->code == GOT_ERR_CANCELLED)
5104 			err = NULL;
5105 		view_close(blame_view);
5106 	} else
5107 		*new_view = blame_view;
5108 done:
5109 	free(path);
5110 	return err;
5111 }
5112 
5113 static const struct got_error *
5114 log_tree_entry(struct tog_view **new_view, int begin_x,
5115     struct got_tree_entry *te, struct tog_parent_trees *parents,
5116     struct got_object_id *commit_id, struct got_repository *repo)
5117 {
5118 	struct tog_view *log_view;
5119 	const struct got_error *err = NULL;
5120 	char *path;
5121 
5122 	*new_view = NULL;
5123 
5124 	log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
5125 	if (log_view == NULL)
5126 		return got_error_from_errno("view_open");
5127 
5128 	err = tree_entry_path(&path, parents, te);
5129 	if (err)
5130 		return err;
5131 
5132 	err = open_log_view(log_view, commit_id, repo, NULL, path, 0);
5133 	if (err)
5134 		view_close(log_view);
5135 	else
5136 		*new_view = log_view;
5137 	free(path);
5138 	return err;
5139 }
5140 
5141 static const struct got_error *
5142 open_tree_view(struct tog_view *view, struct got_tree_object *root,
5143     struct got_object_id *commit_id, struct got_repository *repo)
5144 {
5145 	const struct got_error *err = NULL;
5146 	char *commit_id_str = NULL;
5147 	struct tog_tree_view_state *s = &view->state.tree;
5148 
5149 	TAILQ_INIT(&s->parents);
5150 
5151 	err = got_object_id_str(&commit_id_str, commit_id);
5152 	if (err != NULL)
5153 		goto done;
5154 
5155 	if (asprintf(&s->tree_label, "commit %s", commit_id_str) == -1) {
5156 		err = got_error_from_errno("asprintf");
5157 		goto done;
5158 	}
5159 
5160 	s->root = s->tree = root;
5161 	s->first_displayed_entry = got_object_tree_get_entry(s->tree, 0);
5162 	s->selected_entry = got_object_tree_get_entry(s->tree, 0);
5163 	s->commit_id = got_object_id_dup(commit_id);
5164 	if (s->commit_id == NULL) {
5165 		err = got_error_from_errno("got_object_id_dup");
5166 		goto done;
5167 	}
5168 	s->repo = repo;
5169 
5170 	SIMPLEQ_INIT(&s->colors);
5171 
5172 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
5173 		err = add_color(&s->colors, "\\$$",
5174 		    TOG_COLOR_TREE_SUBMODULE,
5175 		    get_color_value("TOG_COLOR_TREE_SUBMODULE"));
5176 		if (err)
5177 			goto done;
5178 		err = add_color(&s->colors, "@$", TOG_COLOR_TREE_SYMLINK,
5179 		    get_color_value("TOG_COLOR_TREE_SYMLINK"));
5180 		if (err) {
5181 			free_colors(&s->colors);
5182 			goto done;
5183 		}
5184 		err = add_color(&s->colors, "/$",
5185 		    TOG_COLOR_TREE_DIRECTORY,
5186 		    get_color_value("TOG_COLOR_TREE_DIRECTORY"));
5187 		if (err) {
5188 			free_colors(&s->colors);
5189 			goto done;
5190 		}
5191 
5192 		err = add_color(&s->colors, "\\*$",
5193 		    TOG_COLOR_TREE_EXECUTABLE,
5194 		    get_color_value("TOG_COLOR_TREE_EXECUTABLE"));
5195 		if (err) {
5196 			free_colors(&s->colors);
5197 			goto done;
5198 		}
5199 
5200 		err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
5201 		    get_color_value("TOG_COLOR_COMMIT"));
5202 		if (err) {
5203 			free_colors(&s->colors);
5204 			goto done;
5205 		}
5206 	}
5207 
5208 	view->show = show_tree_view;
5209 	view->input = input_tree_view;
5210 	view->close = close_tree_view;
5211 	view->search_start = search_start_tree_view;
5212 	view->search_next = search_next_tree_view;
5213 done:
5214 	free(commit_id_str);
5215 	if (err) {
5216 		free(s->tree_label);
5217 		s->tree_label = NULL;
5218 	}
5219 	return err;
5220 }
5221 
5222 static const struct got_error *
5223 close_tree_view(struct tog_view *view)
5224 {
5225 	struct tog_tree_view_state *s = &view->state.tree;
5226 
5227 	free_colors(&s->colors);
5228 	free(s->tree_label);
5229 	s->tree_label = NULL;
5230 	free(s->commit_id);
5231 	s->commit_id = NULL;
5232 	while (!TAILQ_EMPTY(&s->parents)) {
5233 		struct tog_parent_tree *parent;
5234 		parent = TAILQ_FIRST(&s->parents);
5235 		TAILQ_REMOVE(&s->parents, parent, entry);
5236 		free(parent);
5237 
5238 	}
5239 	if (s->tree != s->root)
5240 		got_object_tree_close(s->tree);
5241 	got_object_tree_close(s->root);
5242 	return NULL;
5243 }
5244 
5245 static const struct got_error *
5246 search_start_tree_view(struct tog_view *view)
5247 {
5248 	struct tog_tree_view_state *s = &view->state.tree;
5249 
5250 	s->matched_entry = NULL;
5251 	return NULL;
5252 }
5253 
5254 static int
5255 match_tree_entry(struct got_tree_entry *te, regex_t *regex)
5256 {
5257 	regmatch_t regmatch;
5258 
5259 	return regexec(regex, got_tree_entry_get_name(te), 1, &regmatch,
5260 	    0) == 0;
5261 }
5262 
5263 static const struct got_error *
5264 search_next_tree_view(struct tog_view *view)
5265 {
5266 	struct tog_tree_view_state *s = &view->state.tree;
5267 	struct got_tree_entry *te = NULL;
5268 
5269 	if (!view->searching) {
5270 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
5271 		return NULL;
5272 	}
5273 
5274 	if (s->matched_entry) {
5275 		if (view->searching == TOG_SEARCH_FORWARD) {
5276 			if (s->selected_entry)
5277 				te = got_tree_entry_get_next(s->tree,
5278 				    s->selected_entry);
5279 			else
5280 				te = got_object_tree_get_first_entry(s->tree);
5281 		} else {
5282 			if (s->selected_entry == NULL)
5283 				te = got_object_tree_get_last_entry(s->tree);
5284 			else
5285 				te = got_tree_entry_get_prev(s->tree,
5286 				    s->selected_entry);
5287 		}
5288 	} else {
5289 		if (view->searching == TOG_SEARCH_FORWARD)
5290 			te = got_object_tree_get_first_entry(s->tree);
5291 		else
5292 			te = got_object_tree_get_last_entry(s->tree);
5293 	}
5294 
5295 	while (1) {
5296 		if (te == NULL) {
5297 			if (s->matched_entry == NULL) {
5298 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
5299 				return NULL;
5300 			}
5301 			if (view->searching == TOG_SEARCH_FORWARD)
5302 				te = got_object_tree_get_first_entry(s->tree);
5303 			else
5304 				te = got_object_tree_get_last_entry(s->tree);
5305 		}
5306 
5307 		if (match_tree_entry(te, &view->regex)) {
5308 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
5309 			s->matched_entry = te;
5310 			break;
5311 		}
5312 
5313 		if (view->searching == TOG_SEARCH_FORWARD)
5314 			te = got_tree_entry_get_next(s->tree, te);
5315 		else
5316 			te = got_tree_entry_get_prev(s->tree, te);
5317 	}
5318 
5319 	if (s->matched_entry) {
5320 		s->first_displayed_entry = s->matched_entry;
5321 		s->selected = 0;
5322 	}
5323 
5324 	return NULL;
5325 }
5326 
5327 static const struct got_error *
5328 show_tree_view(struct tog_view *view)
5329 {
5330 	const struct got_error *err = NULL;
5331 	struct tog_tree_view_state *s = &view->state.tree;
5332 	char *parent_path;
5333 
5334 	err = tree_entry_path(&parent_path, &s->parents, NULL);
5335 	if (err)
5336 		return err;
5337 
5338 	err = draw_tree_entries(view, &s->first_displayed_entry,
5339 	    &s->last_displayed_entry, &s->selected_entry,
5340 	    &s->ndisplayed, s->tree_label, s->show_ids, parent_path,
5341 	    s->tree, s->selected, view->nlines, s->tree == s->root,
5342 	    &s->colors, s->repo);
5343 	free(parent_path);
5344 
5345 	view_vborder(view);
5346 	return err;
5347 }
5348 
5349 static const struct got_error *
5350 input_tree_view(struct tog_view **new_view, struct tog_view **dead_view,
5351     struct tog_view **focus_view, struct tog_view *view, int ch)
5352 {
5353 	const struct got_error *err = NULL;
5354 	struct tog_tree_view_state *s = &view->state.tree;
5355 	struct tog_view *log_view, *ref_view;
5356 	int begin_x = 0;
5357 
5358 	switch (ch) {
5359 	case 'i':
5360 		s->show_ids = !s->show_ids;
5361 		break;
5362 	case 'l':
5363 		if (!s->selected_entry)
5364 			break;
5365 		if (view_is_parent_view(view))
5366 			begin_x = view_split_begin_x(view->begin_x);
5367 		err = log_tree_entry(&log_view, begin_x, s->selected_entry,
5368 		    &s->parents, s->commit_id, s->repo);
5369 		if (view_is_parent_view(view)) {
5370 			err = view_close_child(view);
5371 			if (err)
5372 				return err;
5373 			err = view_set_child(view, log_view);
5374 			if (err) {
5375 				view_close(log_view);
5376 				break;
5377 			}
5378 			*focus_view = log_view;
5379 			view->child_focussed = 1;
5380 		} else
5381 			*new_view = log_view;
5382 		break;
5383 	case 'r':
5384 		if (view_is_parent_view(view))
5385 			begin_x = view_split_begin_x(view->begin_x);
5386 		ref_view = view_open(view->nlines, view->ncols,
5387 		    view->begin_y, begin_x, TOG_VIEW_REF);
5388 		if (ref_view == NULL)
5389 			return got_error_from_errno("view_open");
5390 		err = open_ref_view(ref_view, s->repo);
5391 		if (err) {
5392 			view_close(ref_view);
5393 			return err;
5394 		}
5395 		if (view_is_parent_view(view)) {
5396 			err = view_close_child(view);
5397 			if (err)
5398 				return err;
5399 			err = view_set_child(view, ref_view);
5400 			if (err) {
5401 				view_close(ref_view);
5402 				break;
5403 			}
5404 			*focus_view = ref_view;
5405 			view->child_focussed = 1;
5406 		} else
5407 			*new_view = ref_view;
5408 		break;
5409 	case 'k':
5410 	case KEY_UP:
5411 		if (s->selected > 0) {
5412 			s->selected--;
5413 			break;
5414 		}
5415 		tree_scroll_up(view, &s->first_displayed_entry, 1,
5416 		    s->tree, s->tree == s->root);
5417 		break;
5418 	case KEY_PPAGE:
5419 	case CTRL('b'):
5420 		if (s->tree == s->root) {
5421 			if (got_object_tree_get_first_entry(s->tree) ==
5422 			    s->first_displayed_entry)
5423 				s->selected = 0;
5424 		} else {
5425 			if (s->first_displayed_entry == NULL)
5426 				s->selected = 0;
5427 		}
5428 		tree_scroll_up(view, &s->first_displayed_entry,
5429 		    MAX(0, view->nlines - 3), s->tree, s->tree == s->root);
5430 		break;
5431 	case 'j':
5432 	case KEY_DOWN:
5433 		if (s->selected < s->ndisplayed - 1) {
5434 			s->selected++;
5435 			break;
5436 		}
5437 		if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
5438 		    == NULL)
5439 			/* can't scroll any further */
5440 			break;
5441 		tree_scroll_down(&s->first_displayed_entry, 1,
5442 		    s->last_displayed_entry, s->tree);
5443 		break;
5444 	case KEY_NPAGE:
5445 	case CTRL('f'):
5446 		if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
5447 		    == NULL) {
5448 			/* can't scroll any further; move cursor down */
5449 			if (s->selected < s->ndisplayed - 1)
5450 				s->selected = s->ndisplayed - 1;
5451 			break;
5452 		}
5453 		tree_scroll_down(&s->first_displayed_entry, view->nlines - 3,
5454 		    s->last_displayed_entry, s->tree);
5455 		break;
5456 	case KEY_ENTER:
5457 	case '\r':
5458 	case KEY_BACKSPACE:
5459 		if (s->selected_entry == NULL || ch == KEY_BACKSPACE) {
5460 			struct tog_parent_tree *parent;
5461 			/* user selected '..' */
5462 			if (s->tree == s->root)
5463 				break;
5464 			parent = TAILQ_FIRST(&s->parents);
5465 			TAILQ_REMOVE(&s->parents, parent,
5466 			    entry);
5467 			got_object_tree_close(s->tree);
5468 			s->tree = parent->tree;
5469 			s->first_displayed_entry =
5470 			    parent->first_displayed_entry;
5471 			s->selected_entry =
5472 			    parent->selected_entry;
5473 			s->selected = parent->selected;
5474 			free(parent);
5475 		} else if (S_ISDIR(got_tree_entry_get_mode(
5476 		    s->selected_entry))) {
5477 			struct got_tree_object *subtree;
5478 			err = got_object_open_as_tree(&subtree, s->repo,
5479 			    got_tree_entry_get_id(s->selected_entry));
5480 			if (err)
5481 				break;
5482 			err = tree_view_visit_subtree(subtree, s);
5483 			if (err) {
5484 				got_object_tree_close(subtree);
5485 				break;
5486 			}
5487 		} else if (S_ISREG(got_tree_entry_get_mode(
5488 		    s->selected_entry))) {
5489 			struct tog_view *blame_view;
5490 			int begin_x = view_is_parent_view(view) ?
5491 			    view_split_begin_x(view->begin_x) : 0;
5492 
5493 			err = blame_tree_entry(&blame_view, begin_x,
5494 			    s->selected_entry, &s->parents,
5495 			    s->commit_id, s->repo);
5496 			if (err)
5497 				break;
5498 			if (view_is_parent_view(view)) {
5499 				err = view_close_child(view);
5500 				if (err)
5501 					return err;
5502 				err = view_set_child(view, blame_view);
5503 				if (err) {
5504 					view_close(blame_view);
5505 					break;
5506 				}
5507 				*focus_view = blame_view;
5508 				view->child_focussed = 1;
5509 			} else
5510 				*new_view = blame_view;
5511 		}
5512 		break;
5513 	case KEY_RESIZE:
5514 		if (s->selected > view->nlines)
5515 			s->selected = s->ndisplayed - 1;
5516 		break;
5517 	default:
5518 		break;
5519 	}
5520 
5521 	return err;
5522 }
5523 
5524 __dead static void
5525 usage_tree(void)
5526 {
5527 	endwin();
5528 	fprintf(stderr, "usage: %s tree [-c commit] [-r repository-path] [path]\n",
5529 	    getprogname());
5530 	exit(1);
5531 }
5532 
5533 static const struct got_error *
5534 cmd_tree(int argc, char *argv[])
5535 {
5536 	const struct got_error *error;
5537 	struct got_repository *repo = NULL;
5538 	struct got_worktree *worktree = NULL;
5539 	char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5540 	struct got_object_id *commit_id = NULL;
5541 	char *commit_id_arg = NULL;
5542 	struct got_commit_object *commit = NULL;
5543 	struct got_tree_object *tree = NULL;
5544 	int ch;
5545 	struct tog_view *view;
5546 
5547 #ifndef PROFILE
5548 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
5549 	    NULL) == -1)
5550 		err(1, "pledge");
5551 #endif
5552 
5553 	while ((ch = getopt(argc, argv, "c:r:")) != -1) {
5554 		switch (ch) {
5555 		case 'c':
5556 			commit_id_arg = optarg;
5557 			break;
5558 		case 'r':
5559 			repo_path = realpath(optarg, NULL);
5560 			if (repo_path == NULL)
5561 				return got_error_from_errno2("realpath",
5562 				    optarg);
5563 			break;
5564 		default:
5565 			usage_tree();
5566 			/* NOTREACHED */
5567 		}
5568 	}
5569 
5570 	argc -= optind;
5571 	argv += optind;
5572 
5573 	if (argc > 1)
5574 		usage_tree();
5575 
5576 	cwd = getcwd(NULL, 0);
5577 	if (cwd == NULL)
5578 		return got_error_from_errno("getcwd");
5579 
5580 	error = got_worktree_open(&worktree, cwd);
5581 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
5582 		goto done;
5583 
5584 	if (repo_path == NULL) {
5585 		if (worktree)
5586 			repo_path =
5587 			    strdup(got_worktree_get_repo_path(worktree));
5588 		else
5589 			repo_path = strdup(cwd);
5590 	}
5591 	if (repo_path == NULL) {
5592 		error = got_error_from_errno("strdup");
5593 		goto done;
5594 	}
5595 
5596 	error = got_repo_open(&repo, repo_path, NULL);
5597 	if (error != NULL)
5598 		goto done;
5599 
5600 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
5601 	    repo, worktree);
5602 	if (error)
5603 		goto done;
5604 
5605 	init_curses();
5606 
5607 	error = apply_unveil(got_repo_get_path(repo), NULL);
5608 	if (error)
5609 		goto done;
5610 
5611 	error = got_repo_match_object_id(&commit_id, NULL,
5612 	    commit_id_arg ? commit_id_arg : GOT_REF_HEAD,
5613 	    GOT_OBJ_TYPE_COMMIT, 1, repo);
5614 	if (error)
5615 		goto done;
5616 
5617 	error = got_object_open_as_commit(&commit, repo, commit_id);
5618 	if (error)
5619 		goto done;
5620 
5621 	error = got_object_open_as_tree(&tree, repo,
5622 	    got_object_commit_get_tree_id(commit));
5623 	if (error)
5624 		goto done;
5625 
5626 	view = view_open(0, 0, 0, 0, TOG_VIEW_TREE);
5627 	if (view == NULL) {
5628 		error = got_error_from_errno("view_open");
5629 		goto done;
5630 	}
5631 	error = open_tree_view(view, tree, commit_id, repo);
5632 	if (error)
5633 		goto done;
5634 	if (!got_path_is_root_dir(in_repo_path)) {
5635 		error = tree_view_walk_path(&view->state.tree, commit_id,
5636 		    in_repo_path, repo);
5637 		if (error)
5638 			goto done;
5639 	}
5640 
5641 	if (worktree) {
5642 		/* Release work tree lock. */
5643 		got_worktree_close(worktree);
5644 		worktree = NULL;
5645 	}
5646 	error = view_loop(view);
5647 done:
5648 	free(repo_path);
5649 	free(cwd);
5650 	free(commit_id);
5651 	if (commit)
5652 		got_object_commit_close(commit);
5653 	if (tree)
5654 		got_object_tree_close(tree);
5655 	if (repo)
5656 		got_repo_close(repo);
5657 	return error;
5658 }
5659 
5660 static const struct got_error *
5661 ref_view_load_refs(struct tog_ref_view_state *s)
5662 {
5663 	const struct got_error *err;
5664 	struct got_reflist_entry *sre;
5665 	struct tog_reflist_entry *re;
5666 
5667 	err = got_ref_list(&s->simplerefs, s->repo, NULL,
5668 	    got_ref_cmp_by_name, NULL);
5669 	if (err)
5670 		return err;
5671 
5672 	s->nrefs = 0;
5673 	SIMPLEQ_FOREACH(sre, &s->simplerefs, entry) {
5674 		if (strncmp(got_ref_get_name(sre->ref), "refs/got/", 9) == 0)
5675 			continue;
5676 
5677 		re = malloc(sizeof(*re));
5678 		if (re == NULL)
5679 			return got_error_from_errno("malloc");
5680 
5681 		re->ref = sre->ref;
5682 		re->idx = s->nrefs++;
5683 		TAILQ_INSERT_TAIL(&s->refs, re, entry);
5684 	}
5685 
5686 	return NULL;
5687 }
5688 
5689 void
5690 ref_view_free_refs(struct tog_ref_view_state *s)
5691 {
5692 	struct tog_reflist_entry *re;
5693 
5694 	while (!TAILQ_EMPTY(&s->refs)) {
5695 		re = TAILQ_FIRST(&s->refs);
5696 		TAILQ_REMOVE(&s->refs, re, entry);
5697 		free(re);
5698 	}
5699 	got_ref_list_free(&s->simplerefs);
5700 }
5701 
5702 static const struct got_error *
5703 open_ref_view(struct tog_view *view, struct got_repository *repo)
5704 {
5705 	const struct got_error *err = NULL;
5706 	struct tog_ref_view_state *s = &view->state.ref;
5707 
5708 	s->selected_entry = 0;
5709 	s->repo = repo;
5710 
5711 	SIMPLEQ_INIT(&s->simplerefs);
5712 	TAILQ_INIT(&s->refs);
5713 	SIMPLEQ_INIT(&s->colors);
5714 
5715 	err = ref_view_load_refs(s);
5716 	if (err)
5717 		return err;
5718 
5719 	s->first_displayed_entry = TAILQ_FIRST(&s->refs);
5720 
5721 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
5722 		err = add_color(&s->colors, "^refs/heads/",
5723 		    TOG_COLOR_REFS_HEADS,
5724 		    get_color_value("TOG_COLOR_REFS_HEADS"));
5725 		if (err)
5726 			goto done;
5727 
5728 		err = add_color(&s->colors, "^refs/tags/",
5729 		    TOG_COLOR_REFS_TAGS,
5730 		    get_color_value("TOG_COLOR_REFS_TAGS"));
5731 		if (err)
5732 			goto done;
5733 
5734 		err = add_color(&s->colors, "^refs/remotes/",
5735 		    TOG_COLOR_REFS_REMOTES,
5736 		    get_color_value("TOG_COLOR_REFS_REMOTES"));
5737 		if (err)
5738 			goto done;
5739 	}
5740 
5741 	view->show = show_ref_view;
5742 	view->input = input_ref_view;
5743 	view->close = close_ref_view;
5744 	view->search_start = search_start_ref_view;
5745 	view->search_next = search_next_ref_view;
5746 done:
5747 	if (err)
5748 		free_colors(&s->colors);
5749 	return err;
5750 }
5751 
5752 static const struct got_error *
5753 close_ref_view(struct tog_view *view)
5754 {
5755 	struct tog_ref_view_state *s = &view->state.ref;
5756 
5757 	ref_view_free_refs(s);
5758 	free_colors(&s->colors);
5759 
5760 	return NULL;
5761 }
5762 
5763 static const struct got_error *
5764 resolve_reflist_entry(struct got_object_id **commit_id,
5765     struct tog_reflist_entry *re, struct got_repository *repo)
5766 {
5767 	const struct got_error *err = NULL;
5768 	struct got_object_id *obj_id;
5769 	struct got_tag_object *tag = NULL;
5770 	int obj_type;
5771 
5772 	*commit_id = NULL;
5773 
5774 	err = got_ref_resolve(&obj_id, repo, re->ref);
5775 	if (err)
5776 		return err;
5777 
5778 	err = got_object_get_type(&obj_type, repo, obj_id);
5779 	if (err)
5780 		goto done;
5781 
5782 	switch (obj_type) {
5783 	case GOT_OBJ_TYPE_COMMIT:
5784 		*commit_id = obj_id;
5785 		break;
5786 	case GOT_OBJ_TYPE_TAG:
5787 		err = got_object_open_as_tag(&tag, repo, obj_id);
5788 		if (err)
5789 			goto done;
5790 		free(obj_id);
5791 		err = got_object_get_type(&obj_type, repo,
5792 		    got_object_tag_get_object_id(tag));
5793 		if (err)
5794 			goto done;
5795 		if (obj_type != GOT_OBJ_TYPE_COMMIT) {
5796 			err = got_error(GOT_ERR_OBJ_TYPE);
5797 			goto done;
5798 		}
5799 		*commit_id = got_object_id_dup(
5800 		    got_object_tag_get_object_id(tag));
5801 		if (*commit_id == NULL) {
5802 			err = got_error_from_errno("got_object_id_dup");
5803 			goto done;
5804 		}
5805 		break;
5806 	default:
5807 		err = got_error(GOT_ERR_OBJ_TYPE);
5808 		break;
5809 	}
5810 
5811 done:
5812 	if (tag)
5813 		got_object_tag_close(tag);
5814 	if (err) {
5815 		free(*commit_id);
5816 		*commit_id = NULL;
5817 	}
5818 	return err;
5819 }
5820 
5821 static const struct got_error *
5822 log_ref_entry(struct tog_view **new_view, int begin_x,
5823     struct tog_reflist_entry *re, struct got_repository *repo)
5824 {
5825 	struct tog_view *log_view;
5826 	const struct got_error *err = NULL;
5827 	struct got_object_id *commit_id = NULL;
5828 
5829 	*new_view = NULL;
5830 
5831 	err = resolve_reflist_entry(&commit_id, re, repo);
5832 	if (err) {
5833 		if (err->code != GOT_ERR_OBJ_TYPE)
5834 			return err;
5835 		else
5836 			return NULL;
5837 	}
5838 
5839 	log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
5840 	if (log_view == NULL) {
5841 		err = got_error_from_errno("view_open");
5842 		goto done;
5843 	}
5844 
5845 	err = open_log_view(log_view, commit_id, repo, NULL, "", 0);
5846 done:
5847 	if (err)
5848 		view_close(log_view);
5849 	else
5850 		*new_view = log_view;
5851 	free(commit_id);
5852 	return err;
5853 }
5854 
5855 static void
5856 ref_scroll_up(struct tog_view *view,
5857     struct tog_reflist_entry **first_displayed_entry, int maxscroll,
5858     struct tog_reflist_head *refs)
5859 {
5860 	struct tog_reflist_entry *re;
5861 	int i = 0;
5862 
5863 	if (*first_displayed_entry == TAILQ_FIRST(refs))
5864 		return;
5865 
5866 	re = TAILQ_PREV(*first_displayed_entry, tog_reflist_head, entry);
5867 	while (i++ < maxscroll) {
5868 		if (re == NULL)
5869 			break;
5870 		*first_displayed_entry = re;
5871 		re = TAILQ_PREV(re, tog_reflist_head, entry);
5872 	}
5873 }
5874 
5875 static void
5876 ref_scroll_down(struct tog_reflist_entry **first_displayed_entry, int maxscroll,
5877 	struct tog_reflist_entry *last_displayed_entry,
5878 	struct tog_reflist_head *refs)
5879 {
5880 	struct tog_reflist_entry *next, *last;
5881 	int n = 0;
5882 
5883 	if (*first_displayed_entry)
5884 		next = TAILQ_NEXT(*first_displayed_entry, entry);
5885 	else
5886 		next = TAILQ_FIRST(refs);
5887 
5888 	last = last_displayed_entry;
5889 	while (next && last && n++ < maxscroll) {
5890 		last = TAILQ_NEXT(last, entry);
5891 		if (last) {
5892 			*first_displayed_entry = next;
5893 			next = TAILQ_NEXT(next, entry);
5894 		}
5895 	}
5896 }
5897 
5898 static const struct got_error *
5899 search_start_ref_view(struct tog_view *view)
5900 {
5901 	struct tog_ref_view_state *s = &view->state.ref;
5902 
5903 	s->matched_entry = NULL;
5904 	return NULL;
5905 }
5906 
5907 static int
5908 match_reflist_entry(struct tog_reflist_entry *re, regex_t *regex)
5909 {
5910 	regmatch_t regmatch;
5911 
5912 	return regexec(regex, got_ref_get_name(re->ref), 1, &regmatch,
5913 	    0) == 0;
5914 }
5915 
5916 static const struct got_error *
5917 search_next_ref_view(struct tog_view *view)
5918 {
5919 	struct tog_ref_view_state *s = &view->state.ref;
5920 	struct tog_reflist_entry *re = NULL;
5921 
5922 	if (!view->searching) {
5923 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
5924 		return NULL;
5925 	}
5926 
5927 	if (s->matched_entry) {
5928 		if (view->searching == TOG_SEARCH_FORWARD) {
5929 			if (s->selected_entry)
5930 				re = TAILQ_NEXT(s->selected_entry, entry);
5931 			else
5932 				re = TAILQ_PREV(s->selected_entry,
5933 				    tog_reflist_head, entry);
5934 		} else {
5935 			if (s->selected_entry == NULL)
5936 				re = TAILQ_LAST(&s->refs, tog_reflist_head);
5937 			else
5938 				re = TAILQ_PREV(s->selected_entry,
5939 				    tog_reflist_head, entry);
5940 		}
5941 	} else {
5942 		if (view->searching == TOG_SEARCH_FORWARD)
5943 			re = TAILQ_FIRST(&s->refs);
5944 		else
5945 			re = TAILQ_LAST(&s->refs, tog_reflist_head);
5946 	}
5947 
5948 	while (1) {
5949 		if (re == NULL) {
5950 			if (s->matched_entry == NULL) {
5951 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
5952 				return NULL;
5953 			}
5954 			if (view->searching == TOG_SEARCH_FORWARD)
5955 				re = TAILQ_FIRST(&s->refs);
5956 			else
5957 				re = TAILQ_LAST(&s->refs, tog_reflist_head);
5958 		}
5959 
5960 		if (match_reflist_entry(re, &view->regex)) {
5961 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
5962 			s->matched_entry = re;
5963 			break;
5964 		}
5965 
5966 		if (view->searching == TOG_SEARCH_FORWARD)
5967 			re = TAILQ_NEXT(re, entry);
5968 		else
5969 			re = TAILQ_PREV(re, tog_reflist_head, entry);
5970 	}
5971 
5972 	if (s->matched_entry) {
5973 		s->first_displayed_entry = s->matched_entry;
5974 		s->selected = 0;
5975 	}
5976 
5977 	return NULL;
5978 }
5979 
5980 static const struct got_error *
5981 show_ref_view(struct tog_view *view)
5982 {
5983 	const struct got_error *err = NULL;
5984 	struct tog_ref_view_state *s = &view->state.ref;
5985 	struct tog_reflist_entry *re;
5986 	char *line = NULL;
5987 	wchar_t *wline;
5988 	struct tog_color *tc;
5989 	int width, n;
5990 	int limit = view->nlines;
5991 
5992 	werase(view->window);
5993 
5994 	s->ndisplayed = 0;
5995 
5996 	if (limit == 0)
5997 		return NULL;
5998 
5999 	re = s->first_displayed_entry;
6000 
6001 	if (asprintf(&line, "references [%d/%d]", re->idx + s->selected + 1,
6002 	    s->nrefs) == -1)
6003 		return got_error_from_errno("asprintf");
6004 
6005 	err = format_line(&wline, &width, line, view->ncols, 0);
6006 	if (err) {
6007 		free(line);
6008 		return err;
6009 	}
6010 	if (view_needs_focus_indication(view))
6011 		wstandout(view->window);
6012 	waddwstr(view->window, wline);
6013 	if (view_needs_focus_indication(view))
6014 		wstandend(view->window);
6015 	free(wline);
6016 	wline = NULL;
6017 	free(line);
6018 	line = NULL;
6019 	if (width < view->ncols - 1)
6020 		waddch(view->window, '\n');
6021 	if (--limit <= 0)
6022 		return NULL;
6023 
6024 	n = 0;
6025 	while (re && limit > 0) {
6026 		char *line = NULL;
6027 
6028 		if (got_ref_is_symbolic(re->ref)) {
6029 			if (asprintf(&line, "%s -> %s",
6030 			    got_ref_get_name(re->ref),
6031 			    got_ref_get_symref_target(re->ref)) == -1)
6032 				return got_error_from_errno("asprintf");
6033 		} else if (s->show_ids) {
6034 			struct got_object_id *id;
6035 			char *id_str;
6036 			err = got_ref_resolve(&id, s->repo, re->ref);
6037 			if (err)
6038 				return err;
6039 			err = got_object_id_str(&id_str, id);
6040 			if (err) {
6041 				free(id);
6042 				return err;
6043 			}
6044 			if (asprintf(&line, "%s: %s",
6045 			    got_ref_get_name(re->ref), id_str) == -1) {
6046 				err = got_error_from_errno("asprintf");
6047 				free(id);
6048 				free(id_str);
6049 				return err;
6050 			}
6051 			free(id);
6052 			free(id_str);
6053 		} else {
6054 			line = strdup(got_ref_get_name(re->ref));
6055 			if (line == NULL)
6056 				return got_error_from_errno("strdup");
6057 		}
6058 
6059 		err = format_line(&wline, &width, line, view->ncols, 0);
6060 		if (err) {
6061 			free(line);
6062 			return err;
6063 		}
6064 		if (n == s->selected) {
6065 			if (view->focussed)
6066 				wstandout(view->window);
6067 			s->selected_entry = re;
6068 		}
6069 		tc = match_color(&s->colors, got_ref_get_name(re->ref));
6070 		if (tc)
6071 			wattr_on(view->window,
6072 			    COLOR_PAIR(tc->colorpair), NULL);
6073 		waddwstr(view->window, wline);
6074 		if (tc)
6075 			wattr_off(view->window,
6076 			    COLOR_PAIR(tc->colorpair), NULL);
6077 		if (width < view->ncols - 1)
6078 			waddch(view->window, '\n');
6079 		if (n == s->selected && view->focussed)
6080 			wstandend(view->window);
6081 		free(line);
6082 		free(wline);
6083 		wline = NULL;
6084 		n++;
6085 		s->ndisplayed++;
6086 		s->last_displayed_entry = re;
6087 
6088 		limit--;
6089 		re = TAILQ_NEXT(re, entry);
6090 	}
6091 
6092 	view_vborder(view);
6093 	return err;
6094 }
6095 
6096 static const struct got_error *
6097 browse_ref_tree(struct tog_view **new_view, int begin_x,
6098     struct tog_reflist_entry *re, struct got_repository *repo)
6099 {
6100 	const struct got_error *err = NULL;
6101 	struct got_object_id *commit_id = NULL, *tree_id = NULL;
6102 	struct got_tree_object *tree = NULL;
6103 	struct tog_view *tree_view;
6104 
6105 	*new_view = NULL;
6106 
6107 	err = resolve_reflist_entry(&commit_id, re, repo);
6108 	if (err) {
6109 		if (err->code != GOT_ERR_OBJ_TYPE)
6110 			return err;
6111 		else
6112 			return NULL;
6113 	}
6114 
6115 	err = got_object_id_by_path(&tree_id, repo, commit_id, "/");
6116 	if (err)
6117 		goto done;
6118 
6119 	err = got_object_open_as_tree(&tree, repo, tree_id);
6120 	if (err)
6121 		goto done;
6122 
6123 	tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
6124 	if (tree_view == NULL) {
6125 		err = got_error_from_errno("view_open");
6126 		goto done;
6127 	}
6128 
6129 	err = open_tree_view(tree_view, tree, commit_id, repo);
6130 	if (err)
6131 		goto done;
6132 
6133 	*new_view = tree_view;
6134 done:
6135 	free(commit_id);
6136 	free(tree_id);
6137 	if (err) {
6138 		if (tree)
6139 			got_object_tree_close(tree);
6140 	}
6141 	return err;
6142 }
6143 static const struct got_error *
6144 input_ref_view(struct tog_view **new_view, struct tog_view **dead_view,
6145     struct tog_view **focus_view, struct tog_view *view, int ch)
6146 {
6147 	const struct got_error *err = NULL;
6148 	struct tog_ref_view_state *s = &view->state.ref;
6149 	struct tog_view *log_view, *tree_view;
6150 	int begin_x = 0;
6151 
6152 	switch (ch) {
6153 	case 'i':
6154 		s->show_ids = !s->show_ids;
6155 		break;
6156 	case KEY_ENTER:
6157 	case '\r':
6158 		if (!s->selected_entry)
6159 			break;
6160 		if (view_is_parent_view(view))
6161 			begin_x = view_split_begin_x(view->begin_x);
6162 		err = log_ref_entry(&log_view, begin_x, s->selected_entry,
6163 		    s->repo);
6164 		if (view_is_parent_view(view)) {
6165 			err = view_close_child(view);
6166 			if (err)
6167 				return err;
6168 			err = view_set_child(view, log_view);
6169 			if (err) {
6170 				view_close(log_view);
6171 				break;
6172 			}
6173 			*focus_view = log_view;
6174 			view->child_focussed = 1;
6175 		} else
6176 			*new_view = log_view;
6177 		break;
6178 	case 't':
6179 		if (!s->selected_entry)
6180 			break;
6181 		if (view_is_parent_view(view))
6182 			begin_x = view_split_begin_x(view->begin_x);
6183 		err = browse_ref_tree(&tree_view, begin_x, s->selected_entry,
6184 		    s->repo);
6185 		if (err || tree_view == NULL)
6186 			break;
6187 		if (view_is_parent_view(view)) {
6188 			err = view_close_child(view);
6189 			if (err)
6190 				return err;
6191 			err = view_set_child(view, tree_view);
6192 			if (err) {
6193 				view_close(tree_view);
6194 				break;
6195 			}
6196 			*focus_view = tree_view;
6197 			view->child_focussed = 1;
6198 		} else
6199 			*new_view = tree_view;
6200 		break;
6201 	case 'k':
6202 	case KEY_UP:
6203 		if (s->selected > 0) {
6204 			s->selected--;
6205 			break;
6206 		}
6207 		ref_scroll_up(view, &s->first_displayed_entry, 1, &s->refs);
6208 		break;
6209 	case KEY_PPAGE:
6210 	case CTRL('b'):
6211 		if (s->first_displayed_entry == TAILQ_FIRST(&s->refs))
6212 			s->selected = 0;
6213 		ref_scroll_up(view, &s->first_displayed_entry,
6214 		    MAX(0, view->nlines - 1), &s->refs);
6215 		break;
6216 	case 'j':
6217 	case KEY_DOWN:
6218 		if (s->selected < s->ndisplayed - 1) {
6219 			s->selected++;
6220 			break;
6221 		}
6222 		if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL)
6223 			/* can't scroll any further */
6224 			break;
6225 		ref_scroll_down(&s->first_displayed_entry, 1,
6226 		    s->last_displayed_entry, &s->refs);
6227 		break;
6228 	case KEY_NPAGE:
6229 	case CTRL('f'):
6230 		if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL) {
6231 			/* can't scroll any further; move cursor down */
6232 			if (s->selected < s->ndisplayed - 1)
6233 				s->selected = s->ndisplayed - 1;
6234 			break;
6235 		}
6236 		ref_scroll_down(&s->first_displayed_entry,
6237 		    view->nlines - 1, s->last_displayed_entry, &s->refs);
6238 		break;
6239 	case CTRL('l'):
6240 		ref_view_free_refs(s);
6241 		err = ref_view_load_refs(s);
6242 		break;
6243 	case KEY_RESIZE:
6244 		if (s->selected > view->nlines)
6245 			s->selected = s->ndisplayed - 1;
6246 		break;
6247 	default:
6248 		break;
6249 	}
6250 
6251 	return err;
6252 }
6253 
6254 __dead static void
6255 usage_ref(void)
6256 {
6257 	endwin();
6258 	fprintf(stderr, "usage: %s ref [-r repository-path]\n",
6259 	    getprogname());
6260 	exit(1);
6261 }
6262 
6263 static const struct got_error *
6264 cmd_ref(int argc, char *argv[])
6265 {
6266 	const struct got_error *error;
6267 	struct got_repository *repo = NULL;
6268 	struct got_worktree *worktree = NULL;
6269 	char *cwd = NULL, *repo_path = NULL;
6270 	int ch;
6271 	struct tog_view *view;
6272 
6273 #ifndef PROFILE
6274 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
6275 	    NULL) == -1)
6276 		err(1, "pledge");
6277 #endif
6278 
6279 	while ((ch = getopt(argc, argv, "r:")) != -1) {
6280 		switch (ch) {
6281 		case 'r':
6282 			repo_path = realpath(optarg, NULL);
6283 			if (repo_path == NULL)
6284 				return got_error_from_errno2("realpath",
6285 				    optarg);
6286 			break;
6287 		default:
6288 			usage_ref();
6289 			/* NOTREACHED */
6290 		}
6291 	}
6292 
6293 	argc -= optind;
6294 	argv += optind;
6295 
6296 	if (argc > 1)
6297 		usage_ref();
6298 
6299 	cwd = getcwd(NULL, 0);
6300 	if (cwd == NULL)
6301 		return got_error_from_errno("getcwd");
6302 
6303 	error = got_worktree_open(&worktree, cwd);
6304 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
6305 		goto done;
6306 
6307 	if (repo_path == NULL) {
6308 		if (worktree)
6309 			repo_path =
6310 			    strdup(got_worktree_get_repo_path(worktree));
6311 		else
6312 			repo_path = strdup(cwd);
6313 	}
6314 	if (repo_path == NULL) {
6315 		error = got_error_from_errno("strdup");
6316 		goto done;
6317 	}
6318 
6319 	error = got_repo_open(&repo, repo_path, NULL);
6320 	if (error != NULL)
6321 		goto done;
6322 
6323 	init_curses();
6324 
6325 	error = apply_unveil(got_repo_get_path(repo), NULL);
6326 	if (error)
6327 		goto done;
6328 
6329 	view = view_open(0, 0, 0, 0, TOG_VIEW_REF);
6330 	if (view == NULL) {
6331 		error = got_error_from_errno("view_open");
6332 		goto done;
6333 	}
6334 
6335 	error = open_ref_view(view, repo);
6336 	if (error)
6337 		goto done;
6338 
6339 	if (worktree) {
6340 		/* Release work tree lock. */
6341 		got_worktree_close(worktree);
6342 		worktree = NULL;
6343 	}
6344 	error = view_loop(view);
6345 done:
6346 	free(repo_path);
6347 	free(cwd);
6348 	if (repo)
6349 		got_repo_close(repo);
6350 	return error;
6351 }
6352 
6353 static void
6354 list_commands(FILE *fp)
6355 {
6356 	int i;
6357 
6358 	fprintf(fp, "commands:");
6359 	for (i = 0; i < nitems(tog_commands); i++) {
6360 		struct tog_cmd *cmd = &tog_commands[i];
6361 		fprintf(fp, " %s", cmd->name);
6362 	}
6363 	fputc('\n', fp);
6364 }
6365 
6366 __dead static void
6367 usage(int hflag, int status)
6368 {
6369 	FILE *fp = (status == 0) ? stdout : stderr;
6370 
6371 	fprintf(fp, "usage: %s [-h] [-V | --version] [command] [arg ...]\n",
6372 	    getprogname());
6373 	if (hflag) {
6374 		fprintf(fp, "lazy usage: %s path\n", getprogname());
6375 		list_commands(fp);
6376 	}
6377 	exit(status);
6378 }
6379 
6380 static char **
6381 make_argv(int argc, ...)
6382 {
6383 	va_list ap;
6384 	char **argv;
6385 	int i;
6386 
6387 	va_start(ap, argc);
6388 
6389 	argv = calloc(argc, sizeof(char *));
6390 	if (argv == NULL)
6391 		err(1, "calloc");
6392 	for (i = 0; i < argc; i++) {
6393 		argv[i] = strdup(va_arg(ap, char *));
6394 		if (argv[i] == NULL)
6395 			err(1, "strdup");
6396 	}
6397 
6398 	va_end(ap);
6399 	return argv;
6400 }
6401 
6402 /*
6403  * Try to convert 'tog path' into a 'tog log path' command.
6404  * The user could simply have mistyped the command rather than knowingly
6405  * provided a path. So check whether argv[0] can in fact be resolved
6406  * to a path in the HEAD commit and print a special error if not.
6407  * This hack is for mpi@ <3
6408  */
6409 static const struct got_error *
6410 tog_log_with_path(int argc, char *argv[])
6411 {
6412 	const struct got_error *error = NULL;
6413 	struct tog_cmd *cmd = NULL;
6414 	struct got_repository *repo = NULL;
6415 	struct got_worktree *worktree = NULL;
6416 	struct got_object_id *commit_id = NULL, *id = NULL;
6417 	char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
6418 	char *commit_id_str = NULL, **cmd_argv = NULL;
6419 
6420 	cwd = getcwd(NULL, 0);
6421 	if (cwd == NULL)
6422 		return got_error_from_errno("getcwd");
6423 
6424 	error = got_worktree_open(&worktree, cwd);
6425 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
6426 		goto done;
6427 
6428 	if (worktree)
6429 		repo_path = strdup(got_worktree_get_repo_path(worktree));
6430 	else
6431 		repo_path = strdup(cwd);
6432 	if (repo_path == NULL) {
6433 		error = got_error_from_errno("strdup");
6434 		goto done;
6435 	}
6436 
6437 	error = got_repo_open(&repo, repo_path, NULL);
6438 	if (error != NULL)
6439 		goto done;
6440 
6441 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
6442 	    repo, worktree);
6443 	if (error)
6444 		goto done;
6445 
6446 	error = got_repo_match_object_id(&commit_id, NULL, worktree ?
6447 	    got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
6448 	    GOT_OBJ_TYPE_COMMIT, 1, repo);
6449 	if (error)
6450 		goto done;
6451 
6452 	if (worktree) {
6453 		got_worktree_close(worktree);
6454 		worktree = NULL;
6455 	}
6456 
6457 	error = got_object_id_by_path(&id, repo, commit_id, in_repo_path);
6458 	if (error) {
6459 		if (error->code != GOT_ERR_NO_TREE_ENTRY)
6460 			goto done;
6461 		fprintf(stderr, "%s: '%s' is no known command or path\n",
6462 		    getprogname(), argv[0]);
6463 		usage(1, 1);
6464 		/* not reached */
6465 	}
6466 
6467 	got_repo_close(repo);
6468 	repo = NULL;
6469 
6470 	error = got_object_id_str(&commit_id_str, commit_id);
6471 	if (error)
6472 		goto done;
6473 
6474 	cmd = &tog_commands[0]; /* log */
6475 	argc = 4;
6476 	cmd_argv = make_argv(argc, cmd->name, "-c", commit_id_str, argv[0]);
6477 	error = cmd->cmd_main(argc, cmd_argv);
6478 done:
6479 	if (repo)
6480 		got_repo_close(repo);
6481 	if (worktree)
6482 		got_worktree_close(worktree);
6483 	free(id);
6484 	free(commit_id_str);
6485 	free(commit_id);
6486 	free(cwd);
6487 	free(repo_path);
6488 	free(in_repo_path);
6489 	if (cmd_argv) {
6490 		int i;
6491 		for (i = 0; i < argc; i++)
6492 			free(cmd_argv[i]);
6493 		free(cmd_argv);
6494 	}
6495 	return error;
6496 }
6497 
6498 int
6499 main(int argc, char *argv[])
6500 {
6501 	const struct got_error *error = NULL;
6502 	struct tog_cmd *cmd = NULL;
6503 	int ch, hflag = 0, Vflag = 0;
6504 	char **cmd_argv = NULL;
6505 	static struct option longopts[] = {
6506 	    { "version", no_argument, NULL, 'V' },
6507 	    { NULL, 0, NULL, 0}
6508 	};
6509 
6510 	setlocale(LC_CTYPE, "");
6511 
6512 	while ((ch = getopt_long(argc, argv, "+hV", longopts, NULL)) != -1) {
6513 		switch (ch) {
6514 		case 'h':
6515 			hflag = 1;
6516 			break;
6517 		case 'V':
6518 			Vflag = 1;
6519 			break;
6520 		default:
6521 			usage(hflag, 1);
6522 			/* NOTREACHED */
6523 		}
6524 	}
6525 
6526 	argc -= optind;
6527 	argv += optind;
6528 	optind = 1;
6529 	optreset = 1;
6530 
6531 	if (Vflag) {
6532 		got_version_print_str();
6533 		return 0;
6534 	}
6535 
6536 	if (argc == 0) {
6537 		if (hflag)
6538 			usage(hflag, 0);
6539 		/* Build an argument vector which runs a default command. */
6540 		cmd = &tog_commands[0];
6541 		argc = 1;
6542 		cmd_argv = make_argv(argc, cmd->name);
6543 	} else {
6544 		int i;
6545 
6546 		/* Did the user specify a command? */
6547 		for (i = 0; i < nitems(tog_commands); i++) {
6548 			if (strncmp(tog_commands[i].name, argv[0],
6549 			    strlen(argv[0])) == 0) {
6550 				cmd = &tog_commands[i];
6551 				break;
6552 			}
6553 		}
6554 	}
6555 
6556 	if (cmd == NULL) {
6557 		if (argc != 1)
6558 			usage(0, 1);
6559 		/* No command specified; try log with a path */
6560 		error = tog_log_with_path(argc, argv);
6561 	} else {
6562 		if (hflag)
6563 			cmd->cmd_usage();
6564 		else
6565 			error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
6566 	}
6567 
6568 	endwin();
6569 	putchar('\n');
6570 	if (cmd_argv) {
6571 		int i;
6572 		for (i = 0; i < argc; i++)
6573 			free(cmd_argv[i]);
6574 		free(cmd_argv);
6575 	}
6576 
6577 	if (error && error->code != GOT_ERR_CANCELLED)
6578 		fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
6579 	return 0;
6580 }