1 /*
2  * Copyright (c) 2018, 2019 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 <errno.h>
22 #define _XOPEN_SOURCE_EXTENDED
23 #include <curses.h>
24 #undef _XOPEN_SOURCE_EXTENDED
25 #include <panel.h>
26 #include <locale.h>
27 #include <stdlib.h>
28 #include <stdio.h>
29 #include <getopt.h>
30 #include <string.h>
31 #include <err.h>
32 #include <unistd.h>
33 #include <util.h>
34 #include <limits.h>
35 #include <wchar.h>
36 #include <time.h>
37 #include <pthread.h>
38 #include <libgen.h>
39 #include <regex.h>
40 
41 #include "got_version.h"
42 #include "got_error.h"
43 #include "got_object.h"
44 #include "got_reference.h"
45 #include "got_repository.h"
46 #include "got_diff.h"
47 #include "got_opentemp.h"
48 #include "got_utf8.h"
49 #include "got_cancel.h"
50 #include "got_commit_graph.h"
51 #include "got_blame.h"
52 #include "got_privsep.h"
53 #include "got_path.h"
54 #include "got_worktree.h"
55 
56 #ifndef MIN
57 #define	MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
58 #endif
59 
60 #ifndef MAX
61 #define	MAX(_a,_b) ((_a) > (_b) ? (_a) : (_b))
62 #endif
63 
64 #define CTRL(x)		((x) & 0x1f)
65 
66 #ifndef nitems
67 #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
68 #endif
69 
70 struct tog_cmd {
71 	const char *name;
72 	const struct got_error *(*cmd_main)(int, char *[]);
73 	void (*cmd_usage)(void);
74 };
75 
76 __dead static void	usage(int);
77 __dead static void	usage_log(void);
78 __dead static void	usage_diff(void);
79 __dead static void	usage_blame(void);
80 __dead static void	usage_tree(void);
81 
82 static const struct got_error*	cmd_log(int, char *[]);
83 static const struct got_error*	cmd_diff(int, char *[]);
84 static const struct got_error*	cmd_blame(int, char *[]);
85 static const struct got_error*	cmd_tree(int, char *[]);
86 
87 static struct tog_cmd tog_commands[] = {
88 	{ "log",	cmd_log,	usage_log },
89 	{ "diff",	cmd_diff,	usage_diff },
90 	{ "blame",	cmd_blame,	usage_blame },
91 	{ "tree",	cmd_tree,	usage_tree },
92 };
93 
94 enum tog_view_type {
95 	TOG_VIEW_DIFF,
96 	TOG_VIEW_LOG,
97 	TOG_VIEW_BLAME,
98 	TOG_VIEW_TREE
99 };
100 
101 #define TOG_EOF_STRING	"(END)"
102 
103 struct commit_queue_entry {
104 	TAILQ_ENTRY(commit_queue_entry) entry;
105 	struct got_object_id *id;
106 	struct got_commit_object *commit;
107 	int idx;
108 };
109 TAILQ_HEAD(commit_queue_head, commit_queue_entry);
110 struct commit_queue {
111 	int ncommits;
112 	struct commit_queue_head head;
113 };
114 
115 struct tog_color {
116 	SIMPLEQ_ENTRY(tog_color) entry;
117 	regex_t regex;
118 	short colorpair;
119 };
120 SIMPLEQ_HEAD(tog_colors, tog_color);
121 
122 static const struct got_error *
123 add_color(struct tog_colors *colors, const char *pattern,
124     int idx, short color)
125 {
126 	const struct got_error *err = NULL;
127 	struct tog_color *tc;
128 	int regerr = 0;
129 
130 	if (idx < 1 || idx > COLOR_PAIRS - 1)
131 		return NULL;
132 
133 	init_pair(idx, color, -1);
134 
135 	tc = calloc(1, sizeof(*tc));
136 	if (tc == NULL)
137 		return got_error_from_errno("calloc");
138 	regerr = regcomp(&tc->regex, pattern,
139 	    REG_EXTENDED | REG_NOSUB | REG_NEWLINE);
140 	if (regerr) {
141 		static char regerr_msg[512];
142 		static char err_msg[512];
143 		regerror(regerr, &tc->regex, regerr_msg,
144 		    sizeof(regerr_msg));
145 		snprintf(err_msg, sizeof(err_msg), "regcomp: %s",
146 		    regerr_msg);
147 		err = got_error_msg(GOT_ERR_REGEX, err_msg);
148 		free(tc);
149 		return err;
150 	}
151 	tc->colorpair = idx;
152 	SIMPLEQ_INSERT_HEAD(colors, tc, entry);
153 	return NULL;
154 }
155 
156 static void
157 free_colors(struct tog_colors *colors)
158 {
159 	struct tog_color *tc;
160 
161 	while (!SIMPLEQ_EMPTY(colors)) {
162 		tc = SIMPLEQ_FIRST(colors);
163 		SIMPLEQ_REMOVE_HEAD(colors, entry);
164 		regfree(&tc->regex);
165 		free(tc);
166 	}
167 }
168 
169 struct tog_color *
170 get_color(struct tog_colors *colors, int colorpair)
171 {
172 	struct tog_color *tc = NULL;
173 
174 	SIMPLEQ_FOREACH(tc, colors, entry) {
175 		if (tc->colorpair == colorpair)
176 			return tc;
177 	}
178 
179 	return NULL;
180 }
181 
182 static int
183 default_color_value(const char *envvar)
184 {
185 	if (strcmp(envvar, "TOG_COLOR_DIFF_MINUS") == 0)
186 		return COLOR_MAGENTA;
187 	if (strcmp(envvar, "TOG_COLOR_DIFF_PLUS") == 0)
188 		return COLOR_CYAN;
189 	if (strcmp(envvar, "TOG_COLOR_DIFF_CHUNK_HEADER") == 0)
190 		return COLOR_YELLOW;
191 	if (strcmp(envvar, "TOG_COLOR_DIFF_META") == 0)
192 		return COLOR_GREEN;
193 	if (strcmp(envvar, "TOG_COLOR_TREE_SUBMODULE") == 0)
194 		return COLOR_MAGENTA;
195 	if (strcmp(envvar, "TOG_COLOR_TREE_SYMLINK") == 0)
196 		return COLOR_CYAN;
197 	if (strcmp(envvar, "TOG_COLOR_TREE_DIRECTORY") == 0)
198 		return COLOR_BLUE;
199 	if (strcmp(envvar, "TOG_COLOR_TREE_EXECUTABLE") == 0)
200 		return COLOR_GREEN;
201 	if (strcmp(envvar, "TOG_COLOR_COMMIT") == 0)
202 		return COLOR_GREEN;
203 	if (strcmp(envvar, "TOG_COLOR_AUTHOR") == 0)
204 		return COLOR_CYAN;
205 	if (strcmp(envvar, "TOG_COLOR_DATE") == 0)
206 		return COLOR_YELLOW;
207 
208 	return -1;
209 }
210 
211 static int
212 get_color_value(const char *envvar)
213 {
214 	const char *val = getenv(envvar);
215 
216 	if (val == NULL)
217 		return default_color_value(envvar);
218 
219 	if (strcasecmp(val, "black") == 0)
220 		return COLOR_BLACK;
221 	if (strcasecmp(val, "red") == 0)
222 		return COLOR_RED;
223 	if (strcasecmp(val, "green") == 0)
224 		return COLOR_GREEN;
225 	if (strcasecmp(val, "yellow") == 0)
226 		return COLOR_YELLOW;
227 	if (strcasecmp(val, "blue") == 0)
228 		return COLOR_BLUE;
229 	if (strcasecmp(val, "magenta") == 0)
230 		return COLOR_MAGENTA;
231 	if (strcasecmp(val, "cyan") == 0)
232 		return COLOR_CYAN;
233 	if (strcasecmp(val, "white") == 0)
234 		return COLOR_WHITE;
235 	if (strcasecmp(val, "default") == 0)
236 		return -1;
237 
238 	return default_color_value(envvar);
239 }
240 
241 
242 struct tog_diff_view_state {
243 	struct got_object_id *id1, *id2;
244 	FILE *f;
245 	int first_displayed_line;
246 	int last_displayed_line;
247 	int eof;
248 	int diff_context;
249 	struct got_repository *repo;
250 	struct got_reflist_head *refs;
251 	struct tog_colors colors;
252 
253 	/* passed from log view; may be NULL */
254 	struct tog_view *log_view;
255 };
256 
257 pthread_mutex_t tog_mutex = PTHREAD_MUTEX_INITIALIZER;
258 
259 struct tog_log_thread_args {
260 	pthread_cond_t need_commits;
261 	int commits_needed;
262 	struct got_commit_graph *graph;
263 	struct commit_queue *commits;
264 	const char *in_repo_path;
265 	struct got_object_id *start_id;
266 	struct got_repository *repo;
267 	int log_complete;
268 	sig_atomic_t *quit;
269 	struct commit_queue_entry **first_displayed_entry;
270 	struct commit_queue_entry **selected_entry;
271 	int *searching;
272 	int *search_next_done;
273 	regex_t *regex;
274 };
275 
276 struct tog_log_view_state {
277 	struct commit_queue commits;
278 	struct commit_queue_entry *first_displayed_entry;
279 	struct commit_queue_entry *last_displayed_entry;
280 	struct commit_queue_entry *selected_entry;
281 	int selected;
282 	char *in_repo_path;
283 	const char *head_ref_name;
284 	struct got_repository *repo;
285 	struct got_reflist_head *refs;
286 	struct got_object_id *start_id;
287 	sig_atomic_t quit;
288 	pthread_t thread;
289 	struct tog_log_thread_args thread_args;
290 	struct commit_queue_entry *matched_entry;
291 	struct commit_queue_entry *search_entry;
292 	struct tog_colors colors;
293 };
294 
295 #define TOG_COLOR_DIFF_MINUS		1
296 #define TOG_COLOR_DIFF_PLUS		2
297 #define TOG_COLOR_DIFF_CHUNK_HEADER	3
298 #define TOG_COLOR_DIFF_META		4
299 #define TOG_COLOR_TREE_SUBMODULE	5
300 #define TOG_COLOR_TREE_SYMLINK		6
301 #define TOG_COLOR_TREE_DIRECTORY	7
302 #define TOG_COLOR_TREE_EXECUTABLE	8
303 #define TOG_COLOR_COMMIT		9
304 #define TOG_COLOR_AUTHOR		10
305 #define TOG_COLOR_DATE		11
306 
307 struct tog_blame_cb_args {
308 	struct tog_blame_line *lines; /* one per line */
309 	int nlines;
310 
311 	struct tog_view *view;
312 	struct got_object_id *commit_id;
313 	int *quit;
314 };
315 
316 struct tog_blame_thread_args {
317 	const char *path;
318 	struct got_repository *repo;
319 	struct tog_blame_cb_args *cb_args;
320 	int *complete;
321 	got_cancel_cb cancel_cb;
322 	void *cancel_arg;
323 };
324 
325 struct tog_blame {
326 	FILE *f;
327 	size_t filesize;
328 	struct tog_blame_line *lines;
329 	int nlines;
330 	off_t *line_offsets;
331 	pthread_t thread;
332 	struct tog_blame_thread_args thread_args;
333 	struct tog_blame_cb_args cb_args;
334 	const char *path;
335 };
336 
337 struct tog_blame_view_state {
338 	int first_displayed_line;
339 	int last_displayed_line;
340 	int selected_line;
341 	int blame_complete;
342 	int eof;
343 	int done;
344 	struct got_object_id_queue blamed_commits;
345 	struct got_object_qid *blamed_commit;
346 	char *path;
347 	struct got_repository *repo;
348 	struct got_reflist_head *refs;
349 	struct got_object_id *commit_id;
350 	struct tog_blame blame;
351 	int matched_line;
352 	struct tog_colors colors;
353 };
354 
355 struct tog_parent_tree {
356 	TAILQ_ENTRY(tog_parent_tree) entry;
357 	struct got_tree_object *tree;
358 	struct got_tree_entry *first_displayed_entry;
359 	struct got_tree_entry *selected_entry;
360 	int selected;
361 };
362 
363 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
364 
365 struct tog_tree_view_state {
366 	char *tree_label;
367 	struct got_tree_object *root;
368 	struct got_tree_object *tree;
369 	struct got_tree_entry *first_displayed_entry;
370 	struct got_tree_entry *last_displayed_entry;
371 	struct got_tree_entry *selected_entry;
372 	int ndisplayed, selected, show_ids;
373 	struct tog_parent_trees parents;
374 	struct got_object_id *commit_id;
375 	struct got_repository *repo;
376 	struct got_reflist_head *refs;
377 	struct got_tree_entry *matched_entry;
378 	struct tog_colors colors;
379 };
380 
381 /*
382  * We implement two types of views: parent views and child views.
383  *
384  * The 'Tab' key switches between a parent view and its child view.
385  * Child views are shown side-by-side to their parent view, provided
386  * there is enough screen estate.
387  *
388  * When a new view is opened from within a parent view, this new view
389  * becomes a child view of the parent view, replacing any existing child.
390  *
391  * When a new view is opened from within a child view, this new view
392  * becomes a parent view which will obscure the views below until the
393  * user quits the new parent view by typing 'q'.
394  *
395  * This list of views contains parent views only.
396  * Child views are only pointed to by their parent view.
397  */
398 TAILQ_HEAD(tog_view_list_head, tog_view);
399 
400 struct tog_view {
401 	TAILQ_ENTRY(tog_view) entry;
402 	WINDOW *window;
403 	PANEL *panel;
404 	int nlines, ncols, begin_y, begin_x;
405 	int lines, cols; /* copies of LINES and COLS */
406 	int focussed;
407 	struct tog_view *parent;
408 	struct tog_view *child;
409 	int child_focussed;
410 
411 	/* type-specific state */
412 	enum tog_view_type type;
413 	union {
414 		struct tog_diff_view_state diff;
415 		struct tog_log_view_state log;
416 		struct tog_blame_view_state blame;
417 		struct tog_tree_view_state tree;
418 	} state;
419 
420 	const struct got_error *(*show)(struct tog_view *);
421 	const struct got_error *(*input)(struct tog_view **,
422 	    struct tog_view **, struct tog_view**, struct tog_view *, int);
423 	const struct got_error *(*close)(struct tog_view *);
424 
425 	const struct got_error *(*search_start)(struct tog_view *);
426 	const struct got_error *(*search_next)(struct tog_view *);
427 	int searching;
428 #define TOG_SEARCH_FORWARD	1
429 #define TOG_SEARCH_BACKWARD	2
430 	int search_next_done;
431 	regex_t regex;
432 };
433 
434 static const struct got_error *open_diff_view(struct tog_view *,
435     struct got_object_id *, struct got_object_id *, struct tog_view *,
436     struct got_reflist_head *, struct got_repository *);
437 static const struct got_error *show_diff_view(struct tog_view *);
438 static const struct got_error *input_diff_view(struct tog_view **,
439     struct tog_view **, struct tog_view **, struct tog_view *, int);
440 static const struct got_error* close_diff_view(struct tog_view *);
441 
442 static const struct got_error *open_log_view(struct tog_view *,
443     struct got_object_id *, struct got_reflist_head *,
444     struct got_repository *, const char *, const char *, int);
445 static const struct got_error * show_log_view(struct tog_view *);
446 static const struct got_error *input_log_view(struct tog_view **,
447     struct tog_view **, struct tog_view **, struct tog_view *, int);
448 static const struct got_error *close_log_view(struct tog_view *);
449 static const struct got_error *search_start_log_view(struct tog_view *);
450 static const struct got_error *search_next_log_view(struct tog_view *);
451 
452 static const struct got_error *open_blame_view(struct tog_view *, char *,
453     struct got_object_id *, struct got_reflist_head *, struct got_repository *);
454 static const struct got_error *show_blame_view(struct tog_view *);
455 static const struct got_error *input_blame_view(struct tog_view **,
456     struct tog_view **, struct tog_view **, struct tog_view *, int);
457 static const struct got_error *close_blame_view(struct tog_view *);
458 static const struct got_error *search_start_blame_view(struct tog_view *);
459 static const struct got_error *search_next_blame_view(struct tog_view *);
460 
461 static const struct got_error *open_tree_view(struct tog_view *,
462     struct got_tree_object *, struct got_object_id *,
463     struct got_reflist_head *, struct got_repository *);
464 static const struct got_error *show_tree_view(struct tog_view *);
465 static const struct got_error *input_tree_view(struct tog_view **,
466     struct tog_view **, struct tog_view **, struct tog_view *, int);
467 static const struct got_error *close_tree_view(struct tog_view *);
468 static const struct got_error *search_start_tree_view(struct tog_view *);
469 static const struct got_error *search_next_tree_view(struct tog_view *);
470 
471 static volatile sig_atomic_t tog_sigwinch_received;
472 static volatile sig_atomic_t tog_sigpipe_received;
473 
474 static void
475 tog_sigwinch(int signo)
476 {
477 	tog_sigwinch_received = 1;
478 }
479 
480 static void
481 tog_sigpipe(int signo)
482 {
483 	tog_sigpipe_received = 1;
484 }
485 
486 static const struct got_error *
487 view_close(struct tog_view *view)
488 {
489 	const struct got_error *err = NULL;
490 
491 	if (view->child) {
492 		view_close(view->child);
493 		view->child = NULL;
494 	}
495 	if (view->close)
496 		err = view->close(view);
497 	if (view->panel)
498 		del_panel(view->panel);
499 	if (view->window)
500 		delwin(view->window);
501 	free(view);
502 	return err;
503 }
504 
505 static struct tog_view *
506 view_open(int nlines, int ncols, int begin_y, int begin_x,
507     enum tog_view_type type)
508 {
509 	struct tog_view *view = calloc(1, sizeof(*view));
510 
511 	if (view == NULL)
512 		return NULL;
513 
514 	view->type = type;
515 	view->lines = LINES;
516 	view->cols = COLS;
517 	view->nlines = nlines ? nlines : LINES - begin_y;
518 	view->ncols = ncols ? ncols : COLS - begin_x;
519 	view->begin_y = begin_y;
520 	view->begin_x = begin_x;
521 	view->window = newwin(nlines, ncols, begin_y, begin_x);
522 	if (view->window == NULL) {
523 		view_close(view);
524 		return NULL;
525 	}
526 	view->panel = new_panel(view->window);
527 	if (view->panel == NULL ||
528 	    set_panel_userptr(view->panel, view) != OK) {
529 		view_close(view);
530 		return NULL;
531 	}
532 
533 	keypad(view->window, TRUE);
534 	return view;
535 }
536 
537 static int
538 view_split_begin_x(int begin_x)
539 {
540 	if (begin_x > 0 || COLS < 120)
541 		return 0;
542 	return (COLS - MAX(COLS / 2, 80));
543 }
544 
545 static const struct got_error *view_resize(struct tog_view *);
546 
547 static const struct got_error *
548 view_splitscreen(struct tog_view *view)
549 {
550 	const struct got_error *err = NULL;
551 
552 	view->begin_y = 0;
553 	view->begin_x = view_split_begin_x(0);
554 	view->nlines = LINES;
555 	view->ncols = COLS - view->begin_x;
556 	view->lines = LINES;
557 	view->cols = COLS;
558 	err = view_resize(view);
559 	if (err)
560 		return err;
561 
562 	if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
563 		return got_error_from_errno("mvwin");
564 
565 	return NULL;
566 }
567 
568 static const struct got_error *
569 view_fullscreen(struct tog_view *view)
570 {
571 	const struct got_error *err = NULL;
572 
573 	view->begin_x = 0;
574 	view->begin_y = 0;
575 	view->nlines = LINES;
576 	view->ncols = COLS;
577 	view->lines = LINES;
578 	view->cols = COLS;
579 	err = view_resize(view);
580 	if (err)
581 		return err;
582 
583 	if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
584 		return got_error_from_errno("mvwin");
585 
586 	return NULL;
587 }
588 
589 static int
590 view_is_parent_view(struct tog_view *view)
591 {
592 	return view->parent == NULL;
593 }
594 
595 static const struct got_error *
596 view_resize(struct tog_view *view)
597 {
598 	int nlines, ncols;
599 
600 	if (view->lines > LINES)
601 		nlines = view->nlines - (view->lines - LINES);
602 	else
603 		nlines = view->nlines + (LINES - view->lines);
604 
605 	if (view->cols > COLS)
606 		ncols = view->ncols - (view->cols - COLS);
607 	else
608 		ncols = view->ncols + (COLS - view->cols);
609 
610 	if (wresize(view->window, nlines, ncols) == ERR)
611 		return got_error_from_errno("wresize");
612 	if (replace_panel(view->panel, view->window) == ERR)
613 		return got_error_from_errno("replace_panel");
614 	wclear(view->window);
615 
616 	view->nlines = nlines;
617 	view->ncols = ncols;
618 	view->lines = LINES;
619 	view->cols = COLS;
620 
621 	if (view->child) {
622 		view->child->begin_x = view_split_begin_x(view->begin_x);
623 		if (view->child->begin_x == 0) {
624 			view_fullscreen(view->child);
625 			if (view->child->focussed)
626 				show_panel(view->child->panel);
627 			else
628 				show_panel(view->panel);
629 		} else {
630 			view_splitscreen(view->child);
631 			show_panel(view->child->panel);
632 		}
633 	}
634 
635 	return NULL;
636 }
637 
638 static const struct got_error *
639 view_close_child(struct tog_view *view)
640 {
641 	const struct got_error *err = NULL;
642 
643 	if (view->child == NULL)
644 		return NULL;
645 
646 	err = view_close(view->child);
647 	view->child = NULL;
648 	return err;
649 }
650 
651 static const struct got_error *
652 view_set_child(struct tog_view *view, struct tog_view *child)
653 {
654 	const struct got_error *err = NULL;
655 
656 	view->child = child;
657 	child->parent = view;
658 	return err;
659 }
660 
661 static int
662 view_is_splitscreen(struct tog_view *view)
663 {
664 	return view->begin_x > 0;
665 }
666 
667 static void
668 tog_resizeterm(void)
669 {
670 	int cols, lines;
671 	struct winsize size;
672 
673 	if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &size) < 0) {
674 		cols = 80;     /* Default */
675 		lines = 24;
676 	} else {
677 		cols = size.ws_col;
678 		lines = size.ws_row;
679 	}
680 	resize_term(lines, cols);
681 }
682 
683 static const struct got_error *
684 view_search_start(struct tog_view *view)
685 {
686 	const struct got_error *err = NULL;
687 	char pattern[1024];
688 	int ret;
689 	int begin_x = 0;
690 
691 	if (view->nlines < 1)
692 		return NULL;
693 
694 	if (!view_is_parent_view(view))
695 		begin_x = view_split_begin_x(view->begin_x);
696 	mvwaddstr(view->window, view->begin_y + view->nlines - 1,
697 	    begin_x, "/");
698 	wclrtoeol(view->window);
699 
700 	nocbreak();
701 	echo();
702 	ret = wgetnstr(view->window, pattern, sizeof(pattern));
703 	cbreak();
704 	noecho();
705 	if (ret == ERR)
706 		return NULL;
707 
708 	if (view->searching) {
709 		regfree(&view->regex);
710 		view->searching = 0;
711 	}
712 
713 	if (regcomp(&view->regex, pattern,
714 	    REG_EXTENDED | REG_NOSUB | REG_NEWLINE) == 0) {
715 		err = view->search_start(view);
716 		if (err) {
717 			regfree(&view->regex);
718 			return err;
719 		}
720 		view->searching = TOG_SEARCH_FORWARD;
721 		view->search_next_done = 0;
722 		view->search_next(view);
723 	}
724 
725 	return NULL;
726 }
727 
728 static const struct got_error *
729 view_input(struct tog_view **new, struct tog_view **dead,
730     struct tog_view **focus, int *done, struct tog_view *view,
731     struct tog_view_list_head *views)
732 {
733 	const struct got_error *err = NULL;
734 	struct tog_view *v;
735 	int ch, errcode;
736 
737 	*new = NULL;
738 	*dead = NULL;
739 	*focus = NULL;
740 
741 	if (view->searching && !view->search_next_done) {
742 		errcode = pthread_mutex_unlock(&tog_mutex);
743 		if (errcode)
744 			return got_error_set_errno(errcode,
745 			    "pthread_mutex_unlock");
746 		pthread_yield();
747 		errcode = pthread_mutex_lock(&tog_mutex);
748 		if (errcode)
749 			return got_error_set_errno(errcode,
750 			    "pthread_mutex_lock");
751 		view->search_next(view);
752 		return NULL;
753 	}
754 
755 	nodelay(stdscr, FALSE);
756 	/* Allow threads to make progress while we are waiting for input. */
757 	errcode = pthread_mutex_unlock(&tog_mutex);
758 	if (errcode)
759 		return got_error_set_errno(errcode, "pthread_mutex_unlock");
760 	ch = wgetch(view->window);
761 	errcode = pthread_mutex_lock(&tog_mutex);
762 	if (errcode)
763 		return got_error_set_errno(errcode, "pthread_mutex_lock");
764 	nodelay(stdscr, TRUE);
765 
766 	if (tog_sigwinch_received) {
767 		tog_resizeterm();
768 		tog_sigwinch_received = 0;
769 		TAILQ_FOREACH(v, views, entry) {
770 			err = view_resize(v);
771 			if (err)
772 				return err;
773 			err = v->input(new, dead, focus, v, KEY_RESIZE);
774 			if (err)
775 				return err;
776 		}
777 	}
778 
779 	switch (ch) {
780 	case ERR:
781 		break;
782 	case '\t':
783 		if (view->child) {
784 			*focus = view->child;
785 			view->child_focussed = 1;
786 		} else if (view->parent) {
787 			*focus = view->parent;
788 			view->parent->child_focussed = 0;
789 		}
790 		break;
791 	case 'q':
792 		err = view->input(new, dead, focus, view, ch);
793 		*dead = view;
794 		break;
795 	case 'Q':
796 		*done = 1;
797 		break;
798 	case 'f':
799 		if (view_is_parent_view(view)) {
800 			if (view->child == NULL)
801 				break;
802 			if (view_is_splitscreen(view->child)) {
803 				*focus = view->child;
804 				view->child_focussed = 1;
805 				err = view_fullscreen(view->child);
806 			} else
807 				err = view_splitscreen(view->child);
808 			if (err)
809 				break;
810 			err = view->child->input(new, dead, focus,
811 			    view->child, KEY_RESIZE);
812 		} else {
813 			if (view_is_splitscreen(view)) {
814 				*focus = view;
815 				view->parent->child_focussed = 1;
816 				err = view_fullscreen(view);
817 			} else {
818 				err = view_splitscreen(view);
819 			}
820 			if (err)
821 				break;
822 			err = view->input(new, dead, focus, view,
823 			    KEY_RESIZE);
824 		}
825 		break;
826 	case KEY_RESIZE:
827 		break;
828 	case '/':
829 		if (view->search_start)
830 			view_search_start(view);
831 		else
832 			err = view->input(new, dead, focus, view, ch);
833 		break;
834 	case 'N':
835 	case 'n':
836 		if (view->search_next && view->searching) {
837 			view->searching = (ch == 'n' ?
838 			    TOG_SEARCH_FORWARD : TOG_SEARCH_BACKWARD);
839 			view->search_next_done = 0;
840 			view->search_next(view);
841 		} else
842 			err = view->input(new, dead, focus, view, ch);
843 		break;
844 	default:
845 		err = view->input(new, dead, focus, view, ch);
846 		break;
847 	}
848 
849 	return err;
850 }
851 
852 void
853 view_vborder(struct tog_view *view)
854 {
855 	PANEL *panel;
856 	struct tog_view *view_above;
857 
858 	if (view->parent)
859 		return view_vborder(view->parent);
860 
861 	panel = panel_above(view->panel);
862 	if (panel == NULL)
863 		return;
864 
865 	view_above = panel_userptr(panel);
866 	mvwvline(view->window, view->begin_y, view_above->begin_x - 1,
867 	    got_locale_is_utf8() ? ACS_VLINE : '|', view->nlines);
868 }
869 
870 int
871 view_needs_focus_indication(struct tog_view *view)
872 {
873 	if (view_is_parent_view(view)) {
874 		if (view->child == NULL || view->child_focussed)
875 			return 0;
876 		if (!view_is_splitscreen(view->child))
877 			return 0;
878 	} else if (!view_is_splitscreen(view))
879 		return 0;
880 
881 	return view->focussed;
882 }
883 
884 static const struct got_error *
885 view_loop(struct tog_view *view)
886 {
887 	const struct got_error *err = NULL;
888 	struct tog_view_list_head views;
889 	struct tog_view *new_view, *dead_view, *focus_view, *main_view;
890 	int fast_refresh = 10;
891 	int done = 0, errcode;
892 
893 	errcode = pthread_mutex_lock(&tog_mutex);
894 	if (errcode)
895 		return got_error_set_errno(errcode, "pthread_mutex_lock");
896 
897 	TAILQ_INIT(&views);
898 	TAILQ_INSERT_HEAD(&views, view, entry);
899 
900 	main_view = view;
901 	view->focussed = 1;
902 	err = view->show(view);
903 	if (err)
904 		return err;
905 	update_panels();
906 	doupdate();
907 	while (!TAILQ_EMPTY(&views) && !done && !tog_sigpipe_received) {
908 		/* Refresh fast during initialization, then become slower. */
909 		if (fast_refresh && fast_refresh-- == 0)
910 			halfdelay(10); /* switch to once per second */
911 
912 		err = view_input(&new_view, &dead_view, &focus_view, &done,
913 		    view, &views);
914 		if (err)
915 			break;
916 		if (dead_view) {
917 			struct tog_view *prev = NULL;
918 
919 			if (view_is_parent_view(dead_view))
920 				prev = TAILQ_PREV(dead_view,
921 				    tog_view_list_head, entry);
922 			else if (view->parent != dead_view)
923 				prev = view->parent;
924 
925 			if (dead_view->parent)
926 				dead_view->parent->child = NULL;
927 			else
928 				TAILQ_REMOVE(&views, dead_view, entry);
929 
930 			err = view_close(dead_view);
931 			if (err || (dead_view == main_view && new_view == NULL))
932 				goto done;
933 
934 			if (view == dead_view) {
935 				if (focus_view)
936 					view = focus_view;
937 				else if (prev)
938 					view = prev;
939 				else if (!TAILQ_EMPTY(&views))
940 					view = TAILQ_LAST(&views,
941 					    tog_view_list_head);
942 				else
943 					view = NULL;
944 				if (view) {
945 					if (view->child && view->child_focussed)
946 						focus_view = view->child;
947 					else
948 						focus_view = view;
949 				}
950 			}
951 		}
952 		if (new_view) {
953 			struct tog_view *v, *t;
954 			/* Only allow one parent view per type. */
955 			TAILQ_FOREACH_SAFE(v, &views, entry, t) {
956 				if (v->type != new_view->type)
957 					continue;
958 				TAILQ_REMOVE(&views, v, entry);
959 				err = view_close(v);
960 				if (err)
961 					goto done;
962 				break;
963 			}
964 			TAILQ_INSERT_TAIL(&views, new_view, entry);
965 			view = new_view;
966 			if (focus_view == NULL)
967 				focus_view = new_view;
968 		}
969 		if (focus_view) {
970 			show_panel(focus_view->panel);
971 			if (view)
972 				view->focussed = 0;
973 			focus_view->focussed = 1;
974 			view = focus_view;
975 			if (new_view)
976 				show_panel(new_view->panel);
977 			if (view->child && view_is_splitscreen(view->child))
978 				show_panel(view->child->panel);
979 		}
980 		if (view) {
981 			if (focus_view == NULL) {
982 				view->focussed = 1;
983 				show_panel(view->panel);
984 				if (view->child && view_is_splitscreen(view->child))
985 					show_panel(view->child->panel);
986 				focus_view = view;
987 			}
988 			if (view->parent) {
989 				err = view->parent->show(view->parent);
990 				if (err)
991 					goto done;
992 			}
993 			err = view->show(view);
994 			if (err)
995 				goto done;
996 			if (view->child) {
997 				err = view->child->show(view->child);
998 				if (err)
999 					goto done;
1000 			}
1001 			update_panels();
1002 			doupdate();
1003 		}
1004 	}
1005 done:
1006 	while (!TAILQ_EMPTY(&views)) {
1007 		view = TAILQ_FIRST(&views);
1008 		TAILQ_REMOVE(&views, view, entry);
1009 		view_close(view);
1010 	}
1011 
1012 	errcode = pthread_mutex_unlock(&tog_mutex);
1013 	if (errcode && err == NULL)
1014 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
1015 
1016 	return err;
1017 }
1018 
1019 __dead static void
1020 usage_log(void)
1021 {
1022 	endwin();
1023 	fprintf(stderr,
1024 	    "usage: %s log [-c commit] [-r repository-path] [path]\n",
1025 	    getprogname());
1026 	exit(1);
1027 }
1028 
1029 /* Create newly allocated wide-character string equivalent to a byte string. */
1030 static const struct got_error *
1031 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
1032 {
1033 	char *vis = NULL;
1034 	const struct got_error *err = NULL;
1035 
1036 	*ws = NULL;
1037 	*wlen = mbstowcs(NULL, s, 0);
1038 	if (*wlen == (size_t)-1) {
1039 		int vislen;
1040 		if (errno != EILSEQ)
1041 			return got_error_from_errno("mbstowcs");
1042 
1043 		/* byte string invalid in current encoding; try to "fix" it */
1044 		err = got_mbsavis(&vis, &vislen, s);
1045 		if (err)
1046 			return err;
1047 		*wlen = mbstowcs(NULL, vis, 0);
1048 		if (*wlen == (size_t)-1) {
1049 			err = got_error_from_errno("mbstowcs"); /* give up */
1050 			goto done;
1051 		}
1052 	}
1053 
1054 	*ws = calloc(*wlen + 1, sizeof(**ws));
1055 	if (*ws == NULL) {
1056 		err = got_error_from_errno("calloc");
1057 		goto done;
1058 	}
1059 
1060 	if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
1061 		err = got_error_from_errno("mbstowcs");
1062 done:
1063 	free(vis);
1064 	if (err) {
1065 		free(*ws);
1066 		*ws = NULL;
1067 		*wlen = 0;
1068 	}
1069 	return err;
1070 }
1071 
1072 /* Format a line for display, ensuring that it won't overflow a width limit. */
1073 static const struct got_error *
1074 format_line(wchar_t **wlinep, int *widthp, const char *line, int wlimit,
1075     int col_tab_align)
1076 {
1077 	const struct got_error *err = NULL;
1078 	int cols = 0;
1079 	wchar_t *wline = NULL;
1080 	size_t wlen;
1081 	int i;
1082 
1083 	*wlinep = NULL;
1084 	*widthp = 0;
1085 
1086 	err = mbs2ws(&wline, &wlen, line);
1087 	if (err)
1088 		return err;
1089 
1090 	i = 0;
1091 	while (i < wlen) {
1092 		int width = wcwidth(wline[i]);
1093 
1094 		if (width == 0) {
1095 			i++;
1096 			continue;
1097 		}
1098 
1099 		if (width == 1 || width == 2) {
1100 			if (cols + width > wlimit)
1101 				break;
1102 			cols += width;
1103 			i++;
1104 		} else if (width == -1) {
1105 			if (wline[i] == L'\t') {
1106 				width = TABSIZE -
1107 				    ((cols + col_tab_align) % TABSIZE);
1108 				if (cols + width > wlimit)
1109 					break;
1110 				cols += width;
1111 			}
1112 			i++;
1113 		} else {
1114 			err = got_error_from_errno("wcwidth");
1115 			goto done;
1116 		}
1117 	}
1118 	wline[i] = L'\0';
1119 	if (widthp)
1120 		*widthp = cols;
1121 done:
1122 	if (err)
1123 		free(wline);
1124 	else
1125 		*wlinep = wline;
1126 	return err;
1127 }
1128 
1129 static const struct got_error*
1130 build_refs_str(char **refs_str, struct got_reflist_head *refs,
1131     struct got_object_id *id, struct got_repository *repo)
1132 {
1133 	static const struct got_error *err = NULL;
1134 	struct got_reflist_entry *re;
1135 	char *s;
1136 	const char *name;
1137 
1138 	*refs_str = NULL;
1139 
1140 	SIMPLEQ_FOREACH(re, refs, entry) {
1141 		struct got_tag_object *tag = NULL;
1142 		int cmp;
1143 
1144 		name = got_ref_get_name(re->ref);
1145 		if (strcmp(name, GOT_REF_HEAD) == 0)
1146 			continue;
1147 		if (strncmp(name, "refs/", 5) == 0)
1148 			name += 5;
1149 		if (strncmp(name, "got/", 4) == 0)
1150 			continue;
1151 		if (strncmp(name, "heads/", 6) == 0)
1152 			name += 6;
1153 		if (strncmp(name, "remotes/", 8) == 0)
1154 			name += 8;
1155 		if (strncmp(name, "tags/", 5) == 0) {
1156 			err = got_object_open_as_tag(&tag, repo, re->id);
1157 			if (err) {
1158 				if (err->code != GOT_ERR_OBJ_TYPE)
1159 					break;
1160 				/* Ref points at something other than a tag. */
1161 				err = NULL;
1162 				tag = NULL;
1163 			}
1164 		}
1165 		cmp = got_object_id_cmp(tag ?
1166 		    got_object_tag_get_object_id(tag) : re->id, id);
1167 		if (tag)
1168 			got_object_tag_close(tag);
1169 		if (cmp != 0)
1170 			continue;
1171 		s = *refs_str;
1172 		if (asprintf(refs_str, "%s%s%s", s ? s : "",
1173 		    s ? ", " : "", name) == -1) {
1174 			err = got_error_from_errno("asprintf");
1175 			free(s);
1176 			*refs_str = NULL;
1177 			break;
1178 		}
1179 		free(s);
1180 	}
1181 
1182 	return err;
1183 }
1184 
1185 static const struct got_error *
1186 format_author(wchar_t **wauthor, int *author_width, char *author, int limit,
1187     int col_tab_align)
1188 {
1189 	char *smallerthan, *at;
1190 
1191 	smallerthan = strchr(author, '<');
1192 	if (smallerthan && smallerthan[1] != '\0')
1193 		author = smallerthan + 1;
1194 	at = strchr(author, '@');
1195 	if (at)
1196 		*at = '\0';
1197 	return format_line(wauthor, author_width, author, limit, col_tab_align);
1198 }
1199 
1200 static const struct got_error *
1201 draw_commit(struct tog_view *view, struct got_commit_object *commit,
1202     struct got_object_id *id, struct got_reflist_head *refs,
1203     const size_t date_display_cols, int author_display_cols,
1204     struct tog_colors *colors)
1205 {
1206 	const struct got_error *err = NULL;
1207 	char datebuf[12]; /* YYYY-MM-DD + SPACE + NUL */
1208 	char *logmsg0 = NULL, *logmsg = NULL;
1209 	char *author = NULL;
1210 	wchar_t *wlogmsg = NULL, *wauthor = NULL;
1211 	int author_width, logmsg_width;
1212 	char *newline, *line = NULL;
1213 	int col, limit;
1214 	const int avail = view->ncols;
1215 	struct tm tm;
1216 	time_t committer_time;
1217 	struct tog_color *tc;
1218 
1219 	committer_time = got_object_commit_get_committer_time(commit);
1220 	if (localtime_r(&committer_time, &tm) == NULL)
1221 		return got_error_from_errno("localtime_r");
1222 	if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d ", &tm)
1223 	    >= sizeof(datebuf))
1224 		return got_error(GOT_ERR_NO_SPACE);
1225 
1226 	if (avail <= date_display_cols)
1227 		limit = MIN(sizeof(datebuf) - 1, avail);
1228 	else
1229 		limit = MIN(date_display_cols, sizeof(datebuf) - 1);
1230 	tc = get_color(colors, TOG_COLOR_DATE);
1231 	if (tc)
1232 		wattr_on(view->window,
1233 		    COLOR_PAIR(tc->colorpair), NULL);
1234 	waddnstr(view->window, datebuf, limit);
1235 	if (tc)
1236 		wattr_off(view->window,
1237 		    COLOR_PAIR(tc->colorpair), NULL);
1238 	col = limit;
1239 	if (col > avail)
1240 		goto done;
1241 
1242 	if (avail >= 120) {
1243 		char *id_str;
1244 		err = got_object_id_str(&id_str, id);
1245 		if (err)
1246 			goto done;
1247 		tc = get_color(colors, TOG_COLOR_COMMIT);
1248 		if (tc)
1249 			wattr_on(view->window,
1250 			    COLOR_PAIR(tc->colorpair), NULL);
1251 		wprintw(view->window, "%.8s ", id_str);
1252 		if (tc)
1253 			wattr_off(view->window,
1254 			    COLOR_PAIR(tc->colorpair), NULL);
1255 		free(id_str);
1256 		col += 9;
1257 		if (col > avail)
1258 			goto done;
1259 	}
1260 
1261 	author = strdup(got_object_commit_get_author(commit));
1262 	if (author == NULL) {
1263 		err = got_error_from_errno("strdup");
1264 		goto done;
1265 	}
1266 	err = format_author(&wauthor, &author_width, author, avail - col, col);
1267 	if (err)
1268 		goto done;
1269 	tc = get_color(colors, TOG_COLOR_AUTHOR);
1270 	if (tc)
1271 		wattr_on(view->window,
1272 		    COLOR_PAIR(tc->colorpair), NULL);
1273 	waddwstr(view->window, wauthor);
1274 	if (tc)
1275 		wattr_off(view->window,
1276 		    COLOR_PAIR(tc->colorpair), NULL);
1277 	col += author_width;
1278 	while (col < avail && author_width < author_display_cols + 2) {
1279 		waddch(view->window, ' ');
1280 		col++;
1281 		author_width++;
1282 	}
1283 	if (col > avail)
1284 		goto done;
1285 
1286 	err = got_object_commit_get_logmsg(&logmsg0, commit);
1287 	if (err)
1288 		goto done;
1289 	logmsg = logmsg0;
1290 	while (*logmsg == '\n')
1291 		logmsg++;
1292 	newline = strchr(logmsg, '\n');
1293 	if (newline)
1294 		*newline = '\0';
1295 	limit = avail - col;
1296 	err = format_line(&wlogmsg, &logmsg_width, logmsg, limit, col);
1297 	if (err)
1298 		goto done;
1299 	waddwstr(view->window, wlogmsg);
1300 	col += logmsg_width;
1301 	while (col < avail) {
1302 		waddch(view->window, ' ');
1303 		col++;
1304 	}
1305 done:
1306 	free(logmsg0);
1307 	free(wlogmsg);
1308 	free(author);
1309 	free(wauthor);
1310 	free(line);
1311 	return err;
1312 }
1313 
1314 static struct commit_queue_entry *
1315 alloc_commit_queue_entry(struct got_commit_object *commit,
1316     struct got_object_id *id)
1317 {
1318 	struct commit_queue_entry *entry;
1319 
1320 	entry = calloc(1, sizeof(*entry));
1321 	if (entry == NULL)
1322 		return NULL;
1323 
1324 	entry->id = id;
1325 	entry->commit = commit;
1326 	return entry;
1327 }
1328 
1329 static void
1330 pop_commit(struct commit_queue *commits)
1331 {
1332 	struct commit_queue_entry *entry;
1333 
1334 	entry = TAILQ_FIRST(&commits->head);
1335 	TAILQ_REMOVE(&commits->head, entry, entry);
1336 	got_object_commit_close(entry->commit);
1337 	commits->ncommits--;
1338 	/* Don't free entry->id! It is owned by the commit graph. */
1339 	free(entry);
1340 }
1341 
1342 static void
1343 free_commits(struct commit_queue *commits)
1344 {
1345 	while (!TAILQ_EMPTY(&commits->head))
1346 		pop_commit(commits);
1347 }
1348 
1349 static const struct got_error *
1350 match_commit(int *have_match, struct got_object_id *id,
1351     struct got_commit_object *commit, regex_t *regex)
1352 {
1353 	const struct got_error *err = NULL;
1354 	regmatch_t regmatch;
1355 	char *id_str = NULL, *logmsg = NULL;
1356 
1357 	*have_match = 0;
1358 
1359 	err = got_object_id_str(&id_str, id);
1360 	if (err)
1361 		return err;
1362 
1363 	err = got_object_commit_get_logmsg(&logmsg, commit);
1364 	if (err)
1365 		goto done;
1366 
1367 	if (regexec(regex, got_object_commit_get_author(commit), 1,
1368 	    &regmatch, 0) == 0 ||
1369 	    regexec(regex, got_object_commit_get_committer(commit), 1,
1370 	    &regmatch, 0) == 0 ||
1371 	    regexec(regex, id_str, 1, &regmatch, 0) == 0 ||
1372 	    regexec(regex, logmsg, 1, &regmatch, 0) == 0)
1373 		*have_match = 1;
1374 done:
1375 	free(id_str);
1376 	free(logmsg);
1377 	return err;
1378 }
1379 
1380 static const struct got_error *
1381 queue_commits(struct got_commit_graph *graph, struct commit_queue *commits,
1382     int minqueue, struct got_repository *repo, const char *path,
1383     int *searching, int *search_next_done, regex_t *regex)
1384 {
1385 	const struct got_error *err = NULL;
1386 	int nqueued = 0, have_match = 0;
1387 
1388 	/*
1389 	 * We keep all commits open throughout the lifetime of the log
1390 	 * view in order to avoid having to re-fetch commits from disk
1391 	 * while updating the display.
1392 	 */
1393 	while (nqueued < minqueue ||
1394 	    (*searching == TOG_SEARCH_FORWARD && !*search_next_done)) {
1395 		struct got_object_id *id;
1396 		struct got_commit_object *commit;
1397 		struct commit_queue_entry *entry;
1398 		int errcode;
1399 
1400 		err = got_commit_graph_iter_next(&id, graph, repo, NULL, NULL);
1401 		if (err || id == NULL)
1402 			break;
1403 
1404 		err = got_object_open_as_commit(&commit, repo, id);
1405 		if (err)
1406 			break;
1407 		entry = alloc_commit_queue_entry(commit, id);
1408 		if (entry == NULL) {
1409 			err = got_error_from_errno("alloc_commit_queue_entry");
1410 			break;
1411 		}
1412 
1413 		errcode = pthread_mutex_lock(&tog_mutex);
1414 		if (errcode) {
1415 			err = got_error_set_errno(errcode,
1416 			    "pthread_mutex_lock");
1417 			break;
1418 		}
1419 
1420 		entry->idx = commits->ncommits;
1421 		TAILQ_INSERT_TAIL(&commits->head, entry, entry);
1422 		nqueued++;
1423 		commits->ncommits++;
1424 
1425 		if (*searching == TOG_SEARCH_FORWARD && !*search_next_done) {
1426 			err = match_commit(&have_match, id, commit, regex);
1427 			if (err) {
1428 				pthread_mutex_lock(&tog_mutex);
1429 				break;
1430 			}
1431 		}
1432 
1433 		errcode = pthread_mutex_unlock(&tog_mutex);
1434 		if (errcode && err == NULL)
1435 			err = got_error_set_errno(errcode,
1436 			    "pthread_mutex_unlock");
1437 
1438 		if (have_match)
1439 			break;
1440 	}
1441 
1442 	return err;
1443 }
1444 
1445 static const struct got_error *
1446 get_head_commit_id(struct got_object_id **head_id, const char *branch_name,
1447     struct got_repository *repo)
1448 {
1449 	const struct got_error *err = NULL;
1450 	struct got_reference *head_ref;
1451 
1452 	*head_id = NULL;
1453 
1454 	err = got_ref_open(&head_ref, repo, branch_name, 0);
1455 	if (err)
1456 		return err;
1457 
1458 	err = got_ref_resolve(head_id, repo, head_ref);
1459 	got_ref_close(head_ref);
1460 	if (err) {
1461 		*head_id = NULL;
1462 		return err;
1463 	}
1464 
1465 	return NULL;
1466 }
1467 
1468 static const struct got_error *
1469 draw_commits(struct tog_view *view, struct commit_queue_entry **last,
1470     struct commit_queue_entry **selected, struct commit_queue_entry *first,
1471     struct commit_queue *commits, int selected_idx, int limit,
1472     struct got_reflist_head *refs, const char *path, int commits_needed,
1473     struct tog_colors *colors)
1474 {
1475 	const struct got_error *err = NULL;
1476 	struct tog_log_view_state *s = &view->state.log;
1477 	struct commit_queue_entry *entry;
1478 	int width;
1479 	int ncommits, author_cols = 4;
1480 	char *id_str = NULL, *header = NULL, *ncommits_str = NULL;
1481 	char *refs_str = NULL;
1482 	wchar_t *wline;
1483 	struct tog_color *tc;
1484 	static const size_t date_display_cols = 12;
1485 
1486 	entry = first;
1487 	ncommits = 0;
1488 	while (entry) {
1489 		if (ncommits == selected_idx) {
1490 			*selected = entry;
1491 			break;
1492 		}
1493 		entry = TAILQ_NEXT(entry, entry);
1494 		ncommits++;
1495 	}
1496 
1497 	if (*selected && !(view->searching && view->search_next_done == 0)) {
1498 		err = got_object_id_str(&id_str, (*selected)->id);
1499 		if (err)
1500 			return err;
1501 		if (refs) {
1502 			err = build_refs_str(&refs_str, refs, (*selected)->id,
1503 			    s->repo);
1504 			if (err)
1505 				goto done;
1506 		}
1507 	}
1508 
1509 	if (commits_needed == 0)
1510 		halfdelay(10); /* disable fast refresh */
1511 
1512 	if (asprintf(&ncommits_str, " [%d/%d] %s",
1513 	    entry ? entry->idx + 1 : 0, commits->ncommits,
1514 	    commits_needed > 0 ?
1515 	    (view->searching && view->search_next_done == 0
1516 	    ? "searching..." : "loading... ") :
1517 	    (refs_str ? refs_str : "")) == -1) {
1518 		err = got_error_from_errno("asprintf");
1519 		goto done;
1520 	}
1521 
1522 	if (path && strcmp(path, "/") != 0) {
1523 		if (asprintf(&header, "commit %s %s%s",
1524 		    id_str ? id_str : "........................................",
1525 		    path, ncommits_str) == -1) {
1526 			err = got_error_from_errno("asprintf");
1527 			header = NULL;
1528 			goto done;
1529 		}
1530 	} else if (asprintf(&header, "commit %s%s",
1531 	    id_str ? id_str : "........................................",
1532 	    ncommits_str) == -1) {
1533 		err = got_error_from_errno("asprintf");
1534 		header = NULL;
1535 		goto done;
1536 	}
1537 	err = format_line(&wline, &width, header, view->ncols, 0);
1538 	if (err)
1539 		goto done;
1540 
1541 	werase(view->window);
1542 
1543 	if (view_needs_focus_indication(view))
1544 		wstandout(view->window);
1545 	tc = get_color(colors, TOG_COLOR_COMMIT);
1546 	if (tc)
1547 		wattr_on(view->window,
1548 		    COLOR_PAIR(tc->colorpair), NULL);
1549 	waddwstr(view->window, wline);
1550 	if (tc)
1551 		wattr_off(view->window,
1552 		    COLOR_PAIR(tc->colorpair), NULL);
1553 	while (width < view->ncols) {
1554 		waddch(view->window, ' ');
1555 		width++;
1556 	}
1557 	if (view_needs_focus_indication(view))
1558 		wstandend(view->window);
1559 	free(wline);
1560 	if (limit <= 1)
1561 		goto done;
1562 
1563 	/* Grow author column size if necessary. */
1564 	entry = first;
1565 	ncommits = 0;
1566 	while (entry) {
1567 		char *author;
1568 		wchar_t *wauthor;
1569 		int width;
1570 		if (ncommits >= limit - 1)
1571 			break;
1572 		author = strdup(got_object_commit_get_author(entry->commit));
1573 		if (author == NULL) {
1574 			err = got_error_from_errno("strdup");
1575 			goto done;
1576 		}
1577 		err = format_author(&wauthor, &width, author, COLS,
1578 		    date_display_cols);
1579 		if (author_cols < width)
1580 			author_cols = width;
1581 		free(wauthor);
1582 		free(author);
1583 		ncommits++;
1584 		entry = TAILQ_NEXT(entry, entry);
1585 	}
1586 
1587 	entry = first;
1588 	*last = first;
1589 	ncommits = 0;
1590 	while (entry) {
1591 		if (ncommits >= limit - 1)
1592 			break;
1593 		if (ncommits == selected_idx)
1594 			wstandout(view->window);
1595 		err = draw_commit(view, entry->commit, entry->id, refs,
1596 		    date_display_cols, author_cols, colors);
1597 		if (ncommits == selected_idx)
1598 			wstandend(view->window);
1599 		if (err)
1600 			goto done;
1601 		ncommits++;
1602 		*last = entry;
1603 		entry = TAILQ_NEXT(entry, entry);
1604 	}
1605 
1606 	view_vborder(view);
1607 done:
1608 	free(id_str);
1609 	free(refs_str);
1610 	free(ncommits_str);
1611 	free(header);
1612 	return err;
1613 }
1614 
1615 static void
1616 scroll_up(struct tog_view *view,
1617     struct commit_queue_entry **first_displayed_entry, int maxscroll,
1618     struct commit_queue *commits)
1619 {
1620 	struct commit_queue_entry *entry;
1621 	int nscrolled = 0;
1622 
1623 	entry = TAILQ_FIRST(&commits->head);
1624 	if (*first_displayed_entry == entry)
1625 		return;
1626 
1627 	entry = *first_displayed_entry;
1628 	while (entry && nscrolled < maxscroll) {
1629 		entry = TAILQ_PREV(entry, commit_queue_head, entry);
1630 		if (entry) {
1631 			*first_displayed_entry = entry;
1632 			nscrolled++;
1633 		}
1634 	}
1635 }
1636 
1637 static const struct got_error *
1638 trigger_log_thread(int load_all, int *commits_needed, int *log_complete,
1639     pthread_cond_t *need_commits)
1640 {
1641 	int errcode;
1642 	int max_wait = 20;
1643 
1644 	halfdelay(1); /* fast refresh while loading commits */
1645 
1646 	while (*commits_needed > 0) {
1647 		if (*log_complete)
1648 			break;
1649 
1650 		/* Wake the log thread. */
1651 		errcode = pthread_cond_signal(need_commits);
1652 		if (errcode)
1653 			return got_error_set_errno(errcode,
1654 			    "pthread_cond_signal");
1655 		errcode = pthread_mutex_unlock(&tog_mutex);
1656 		if (errcode)
1657 			return got_error_set_errno(errcode,
1658 			    "pthread_mutex_unlock");
1659 		pthread_yield();
1660 		errcode = pthread_mutex_lock(&tog_mutex);
1661 		if (errcode)
1662 			return got_error_set_errno(errcode,
1663 			    "pthread_mutex_lock");
1664 
1665 		if (*commits_needed > 0 && (!load_all || --max_wait <= 0)) {
1666 			/*
1667 			 * Thread is not done yet; lose a key press
1668 			 * and let the user retry... this way the GUI
1669 			 * remains interactive while logging deep paths
1670 			 * with few commits in history.
1671 			 */
1672 			return NULL;
1673 		}
1674 	}
1675 
1676 	return NULL;
1677 }
1678 
1679 static const struct got_error *
1680 scroll_down(struct tog_view *view,
1681     struct commit_queue_entry **first_displayed_entry, int maxscroll,
1682     struct commit_queue_entry **last_displayed_entry,
1683     struct commit_queue *commits, int *log_complete, int *commits_needed,
1684     pthread_cond_t *need_commits)
1685 {
1686 	const struct got_error *err = NULL;
1687 	struct commit_queue_entry *pentry;
1688 	int nscrolled = 0;
1689 
1690 	if (*last_displayed_entry == NULL)
1691 		return NULL;
1692 
1693 	pentry = TAILQ_NEXT(*last_displayed_entry, entry);
1694 	if (pentry == NULL && !*log_complete) {
1695 		/*
1696 		 * Ask the log thread for required amount of commits
1697 		 * plus some amount of pre-fetching.
1698 		 */
1699 		(*commits_needed) += maxscroll + 20;
1700 		err = trigger_log_thread(0, commits_needed, log_complete,
1701 		    need_commits);
1702 		if (err)
1703 			return err;
1704 	}
1705 
1706 	do {
1707 		pentry = TAILQ_NEXT(*last_displayed_entry, entry);
1708 		if (pentry == NULL)
1709 			break;
1710 
1711 		*last_displayed_entry = pentry;
1712 
1713 		pentry = TAILQ_NEXT(*first_displayed_entry, entry);
1714 		if (pentry == NULL)
1715 			break;
1716 		*first_displayed_entry = pentry;
1717 	} while (++nscrolled < maxscroll);
1718 
1719 	return err;
1720 }
1721 
1722 static const struct got_error *
1723 open_diff_view_for_commit(struct tog_view **new_view, int begin_x,
1724     struct got_commit_object *commit, struct got_object_id *commit_id,
1725     struct tog_view *log_view, struct got_reflist_head *refs,
1726     struct got_repository *repo)
1727 {
1728 	const struct got_error *err;
1729 	struct got_object_qid *parent_id;
1730 	struct tog_view *diff_view;
1731 
1732 	diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
1733 	if (diff_view == NULL)
1734 		return got_error_from_errno("view_open");
1735 
1736 	parent_id = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
1737 	err = open_diff_view(diff_view, parent_id ? parent_id->id : NULL,
1738 	    commit_id, log_view, refs, repo);
1739 	if (err == NULL)
1740 		*new_view = diff_view;
1741 	return err;
1742 }
1743 
1744 static const struct got_error *
1745 tree_view_visit_subtree(struct got_tree_object *subtree,
1746     struct tog_tree_view_state *s)
1747 {
1748 	struct tog_parent_tree *parent;
1749 
1750 	parent = calloc(1, sizeof(*parent));
1751 	if (parent == NULL)
1752 		return got_error_from_errno("calloc");
1753 
1754 	parent->tree = s->tree;
1755 	parent->first_displayed_entry = s->first_displayed_entry;
1756 	parent->selected_entry = s->selected_entry;
1757 	parent->selected = s->selected;
1758 	TAILQ_INSERT_HEAD(&s->parents, parent, entry);
1759 	s->tree = subtree;
1760 	s->selected = 0;
1761 	s->first_displayed_entry = NULL;
1762 	return NULL;
1763 }
1764 
1765 
1766 static const struct got_error *
1767 browse_commit_tree(struct tog_view **new_view, int begin_x,
1768     struct commit_queue_entry *entry, const char *path,
1769     struct got_reflist_head *refs, struct got_repository *repo)
1770 {
1771 	const struct got_error *err = NULL;
1772 	struct got_tree_object *tree;
1773 	struct tog_tree_view_state *s;
1774 	struct tog_view *tree_view;
1775 	char *slash, *subpath = NULL;
1776 	const char *p;
1777 
1778 	err = got_object_open_as_tree(&tree, repo,
1779 	    got_object_commit_get_tree_id(entry->commit));
1780 	if (err)
1781 		return err;
1782 
1783 	tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
1784 	if (tree_view == NULL)
1785 		return got_error_from_errno("view_open");
1786 
1787 	err = open_tree_view(tree_view, tree, entry->id, refs, repo);
1788 	if (err) {
1789 		got_object_tree_close(tree);
1790 		return err;
1791 	}
1792 	s = &tree_view->state.tree;
1793 
1794 	*new_view = tree_view;
1795 
1796 	/* Walk the path and open corresponding tree objects. */
1797 	p = path;
1798 	while (p[0] == '/')
1799 		p++;
1800 	while (*p) {
1801 		struct got_tree_entry *te;
1802 		struct got_object_id *tree_id;
1803 		char *te_name;
1804 
1805 		/* Ensure the correct subtree entry is selected. */
1806 		slash = strchr(p, '/');
1807 		if (slash == NULL)
1808 			slash = strchr(p, '\0');
1809 
1810 		te_name = strndup(p, slash -p);
1811 		if (te_name == NULL) {
1812 			err = got_error_from_errno("strndup");
1813 			break;
1814 		}
1815 		te = got_object_tree_find_entry(s->tree, te_name);
1816 		if (te == NULL) {
1817 			err = got_error_path(te_name, GOT_ERR_NO_TREE_ENTRY);
1818 			free(te_name);
1819 			break;
1820 		}
1821 		free(te_name);
1822 		s->selected_entry = te;
1823 		s->selected = got_tree_entry_get_index(te);
1824 		if (s->tree != s->root)
1825 			s->selected++; /* skip '..' */
1826 
1827 		if (!S_ISDIR(got_tree_entry_get_mode(s->selected_entry))) {
1828 			/* Jump to this file's entry. */
1829 			s->first_displayed_entry = s->selected_entry;
1830 			s->selected = 0;
1831 			break;
1832 		}
1833 
1834 		slash = strchr(p, '/');
1835 		if (slash)
1836 			subpath = strndup(path, slash - path);
1837 		else
1838 			subpath = strdup(path);
1839 		if (subpath == NULL) {
1840 			err = got_error_from_errno("strdup");
1841 			break;
1842 		}
1843 
1844 		err = got_object_id_by_path(&tree_id, repo, entry->id,
1845 		    subpath);
1846 		if (err)
1847 			break;
1848 
1849 		err = got_object_open_as_tree(&tree, repo, tree_id);
1850 		free(tree_id);
1851 		if (err)
1852 			break;
1853 
1854 		err = tree_view_visit_subtree(tree, s);
1855 		if (err) {
1856 			got_object_tree_close(tree);
1857 			break;
1858 		}
1859 		if (slash == NULL)
1860 			break;
1861 		free(subpath);
1862 		subpath = NULL;
1863 		p = slash;
1864 	}
1865 
1866 	free(subpath);
1867 	return err;
1868 }
1869 
1870 static void *
1871 log_thread(void *arg)
1872 {
1873 	const struct got_error *err = NULL;
1874 	int errcode = 0;
1875 	struct tog_log_thread_args *a = arg;
1876 	int done = 0;
1877 
1878 	err = got_commit_graph_iter_start(a->graph, a->start_id, a->repo,
1879 	    NULL, NULL);
1880 	if (err)
1881 		return (void *)err;
1882 
1883 	while (!done && !err && !tog_sigpipe_received) {
1884 		err = queue_commits(a->graph, a->commits, 1, a->repo,
1885 		    a->in_repo_path, a->searching, a->search_next_done,
1886 		    a->regex);
1887 		if (err) {
1888 			if (err->code != GOT_ERR_ITER_COMPLETED)
1889 				return (void *)err;
1890 			err = NULL;
1891 			done = 1;
1892 		} else if (a->commits_needed > 0)
1893 			a->commits_needed--;
1894 
1895 		errcode = pthread_mutex_lock(&tog_mutex);
1896 		if (errcode) {
1897 			err = got_error_set_errno(errcode,
1898 			    "pthread_mutex_lock");
1899 			break;
1900 		} else if (*a->quit)
1901 			done = 1;
1902 		else if (*a->first_displayed_entry == NULL) {
1903 			*a->first_displayed_entry =
1904 			    TAILQ_FIRST(&a->commits->head);
1905 			*a->selected_entry = *a->first_displayed_entry;
1906 		}
1907 
1908 		if (done)
1909 			a->commits_needed = 0;
1910 		else if (a->commits_needed == 0) {
1911 			errcode = pthread_cond_wait(&a->need_commits,
1912 			    &tog_mutex);
1913 			if (errcode)
1914 				err = got_error_set_errno(errcode,
1915 				    "pthread_cond_wait");
1916 		}
1917 
1918 		errcode = pthread_mutex_unlock(&tog_mutex);
1919 		if (errcode && err == NULL)
1920 			err = got_error_set_errno(errcode,
1921 			    "pthread_mutex_unlock");
1922 	}
1923 	a->log_complete = 1;
1924 	return (void *)err;
1925 }
1926 
1927 static const struct got_error *
1928 stop_log_thread(struct tog_log_view_state *s)
1929 {
1930 	const struct got_error *err = NULL;
1931 	int errcode;
1932 
1933 	if (s->thread) {
1934 		s->quit = 1;
1935 		errcode = pthread_cond_signal(&s->thread_args.need_commits);
1936 		if (errcode)
1937 			return got_error_set_errno(errcode,
1938 			    "pthread_cond_signal");
1939 		errcode = pthread_mutex_unlock(&tog_mutex);
1940 		if (errcode)
1941 			return got_error_set_errno(errcode,
1942 			    "pthread_mutex_unlock");
1943 		errcode = pthread_join(s->thread, (void **)&err);
1944 		if (errcode)
1945 			return got_error_set_errno(errcode, "pthread_join");
1946 		errcode = pthread_mutex_lock(&tog_mutex);
1947 		if (errcode)
1948 			return got_error_set_errno(errcode,
1949 			    "pthread_mutex_lock");
1950 		s->thread = NULL;
1951 	}
1952 
1953 	errcode = pthread_cond_destroy(&s->thread_args.need_commits);
1954 	if (errcode && err == NULL)
1955 		err = got_error_set_errno(errcode, "pthread_cond_destroy");
1956 
1957 	if (s->thread_args.repo) {
1958 		got_repo_close(s->thread_args.repo);
1959 		s->thread_args.repo = NULL;
1960 	}
1961 
1962 	if (s->thread_args.graph) {
1963 		got_commit_graph_close(s->thread_args.graph);
1964 		s->thread_args.graph = NULL;
1965 	}
1966 
1967 	return err;
1968 }
1969 
1970 static const struct got_error *
1971 close_log_view(struct tog_view *view)
1972 {
1973 	const struct got_error *err = NULL;
1974 	struct tog_log_view_state *s = &view->state.log;
1975 
1976 	err = stop_log_thread(s);
1977 	free_commits(&s->commits);
1978 	free(s->in_repo_path);
1979 	s->in_repo_path = NULL;
1980 	free(s->start_id);
1981 	s->start_id = NULL;
1982 	return err;
1983 }
1984 
1985 static const struct got_error *
1986 search_start_log_view(struct tog_view *view)
1987 {
1988 	struct tog_log_view_state *s = &view->state.log;
1989 
1990 	s->matched_entry = NULL;
1991 	s->search_entry = NULL;
1992 	return NULL;
1993 }
1994 
1995 static const struct got_error *
1996 search_next_log_view(struct tog_view *view)
1997 {
1998 	const struct got_error *err = NULL;
1999 	struct tog_log_view_state *s = &view->state.log;
2000 	struct commit_queue_entry *entry;
2001 
2002 	if (!view->searching) {
2003 		view->search_next_done = 1;
2004 		return NULL;
2005 	}
2006 
2007 	if (s->search_entry) {
2008 		int errcode, ch;
2009 		errcode = pthread_mutex_unlock(&tog_mutex);
2010 		if (errcode)
2011 			return got_error_set_errno(errcode,
2012 			    "pthread_mutex_unlock");
2013 		ch = wgetch(view->window);
2014 		errcode = pthread_mutex_lock(&tog_mutex);
2015 		if (errcode)
2016 			return got_error_set_errno(errcode,
2017 			    "pthread_mutex_lock");
2018 		if (ch == KEY_BACKSPACE) {
2019 			view->search_next_done = 1;
2020 			return NULL;
2021 		}
2022 		if (view->searching == TOG_SEARCH_FORWARD)
2023 			entry = TAILQ_NEXT(s->search_entry, entry);
2024 		else
2025 			entry = TAILQ_PREV(s->search_entry,
2026 			    commit_queue_head, entry);
2027 	} else if (s->matched_entry) {
2028 		if (view->searching == TOG_SEARCH_FORWARD)
2029 			entry = TAILQ_NEXT(s->selected_entry, entry);
2030 		else
2031 			entry = TAILQ_PREV(s->selected_entry,
2032 			    commit_queue_head, entry);
2033 	} else {
2034 		if (view->searching == TOG_SEARCH_FORWARD)
2035 			entry = TAILQ_FIRST(&s->commits.head);
2036 		else
2037 			entry = TAILQ_LAST(&s->commits.head, commit_queue_head);
2038 	}
2039 
2040 	while (1) {
2041 		int have_match = 0;
2042 
2043 		if (entry == NULL) {
2044 			if (s->thread_args.log_complete ||
2045 			    view->searching == TOG_SEARCH_BACKWARD) {
2046 				view->search_next_done = 1;
2047 				return NULL;
2048 			}
2049 			/*
2050 			 * Poke the log thread for more commits and return,
2051 			 * allowing the main loop to make progress. Search
2052 			 * will resume at s->search_entry once we come back.
2053 			 */
2054 			s->thread_args.commits_needed++;
2055 			return trigger_log_thread(1,
2056 			    &s->thread_args.commits_needed,
2057 			    &s->thread_args.log_complete,
2058 			    &s->thread_args.need_commits);
2059 		}
2060 
2061 		err = match_commit(&have_match, entry->id, entry->commit,
2062 		    &view->regex);
2063 		if (err)
2064 			break;
2065 		if (have_match) {
2066 			view->search_next_done = 1;
2067 			s->matched_entry = entry;
2068 			break;
2069 		}
2070 
2071 		s->search_entry = entry;
2072 		if (view->searching == TOG_SEARCH_FORWARD)
2073 			entry = TAILQ_NEXT(entry, entry);
2074 		else
2075 			entry = TAILQ_PREV(entry, commit_queue_head, entry);
2076 	}
2077 
2078 	if (s->matched_entry) {
2079 		int cur = s->selected_entry->idx;
2080 		while (cur < s->matched_entry->idx) {
2081 			err = input_log_view(NULL, NULL, NULL, view, KEY_DOWN);
2082 			if (err)
2083 				return err;
2084 			cur++;
2085 		}
2086 		while (cur > s->matched_entry->idx) {
2087 			err = input_log_view(NULL, NULL, NULL, view, KEY_UP);
2088 			if (err)
2089 				return err;
2090 			cur--;
2091 		}
2092 	}
2093 
2094 	s->search_entry = NULL;
2095 
2096 	return NULL;
2097 }
2098 
2099 static const struct got_error *
2100 open_log_view(struct tog_view *view, struct got_object_id *start_id,
2101     struct got_reflist_head *refs, struct got_repository *repo,
2102     const char *head_ref_name, const char *path, int check_disk)
2103 {
2104 	const struct got_error *err = NULL;
2105 	struct tog_log_view_state *s = &view->state.log;
2106 	struct got_repository *thread_repo = NULL;
2107 	struct got_commit_graph *thread_graph = NULL;
2108 	int errcode;
2109 
2110 	err = got_repo_map_path(&s->in_repo_path, repo, path, check_disk);
2111 	if (err != NULL)
2112 		goto done;
2113 
2114 	/* The commit queue only contains commits being displayed. */
2115 	TAILQ_INIT(&s->commits.head);
2116 	s->commits.ncommits = 0;
2117 
2118 	s->refs = refs;
2119 	s->repo = repo;
2120 	s->head_ref_name = head_ref_name;
2121 	s->start_id = got_object_id_dup(start_id);
2122 	if (s->start_id == NULL) {
2123 		err = got_error_from_errno("got_object_id_dup");
2124 		goto done;
2125 	}
2126 
2127 	SIMPLEQ_INIT(&s->colors);
2128 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
2129 		err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
2130 		    get_color_value("TOG_COLOR_COMMIT"));
2131 		if (err)
2132 			goto done;
2133 		err = add_color(&s->colors, "^$", TOG_COLOR_AUTHOR,
2134 		    get_color_value("TOG_COLOR_AUTHOR"));
2135 		if (err) {
2136 			free_colors(&s->colors);
2137 			goto done;
2138 		}
2139 		err = add_color(&s->colors, "^$", TOG_COLOR_DATE,
2140 		    get_color_value("TOG_COLOR_DATE"));
2141 		if (err) {
2142 			free_colors(&s->colors);
2143 			goto done;
2144 		}
2145 	}
2146 
2147 	view->show = show_log_view;
2148 	view->input = input_log_view;
2149 	view->close = close_log_view;
2150 	view->search_start = search_start_log_view;
2151 	view->search_next = search_next_log_view;
2152 
2153 	err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL);
2154 	if (err)
2155 		goto done;
2156 	err = got_commit_graph_open(&thread_graph, start_id, s->in_repo_path,
2157 	    0, thread_repo);
2158 	if (err)
2159 		goto done;
2160 
2161 	errcode = pthread_cond_init(&s->thread_args.need_commits, NULL);
2162 	if (errcode) {
2163 		err = got_error_set_errno(errcode, "pthread_cond_init");
2164 		goto done;
2165 	}
2166 
2167 	s->thread_args.commits_needed = view->nlines;
2168 	s->thread_args.graph = thread_graph;
2169 	s->thread_args.commits = &s->commits;
2170 	s->thread_args.in_repo_path = s->in_repo_path;
2171 	s->thread_args.start_id = s->start_id;
2172 	s->thread_args.repo = thread_repo;
2173 	s->thread_args.log_complete = 0;
2174 	s->thread_args.quit = &s->quit;
2175 	s->thread_args.first_displayed_entry = &s->first_displayed_entry;
2176 	s->thread_args.selected_entry = &s->selected_entry;
2177 	s->thread_args.searching = &view->searching;
2178 	s->thread_args.search_next_done = &view->search_next_done;
2179 	s->thread_args.regex = &view->regex;
2180 done:
2181 	if (err)
2182 		close_log_view(view);
2183 	return err;
2184 }
2185 
2186 static const struct got_error *
2187 show_log_view(struct tog_view *view)
2188 {
2189 	struct tog_log_view_state *s = &view->state.log;
2190 
2191 	if (s->thread == NULL) {
2192 		int errcode = pthread_create(&s->thread, NULL, log_thread,
2193 		    &s->thread_args);
2194 		if (errcode)
2195 			return got_error_set_errno(errcode, "pthread_create");
2196 	}
2197 
2198 	return draw_commits(view, &s->last_displayed_entry,
2199 	    &s->selected_entry, s->first_displayed_entry,
2200 	    &s->commits, s->selected, view->nlines, s->refs,
2201 	    s->in_repo_path, s->thread_args.commits_needed, &s->colors);
2202 }
2203 
2204 static const struct got_error *
2205 input_log_view(struct tog_view **new_view, struct tog_view **dead_view,
2206     struct tog_view **focus_view, struct tog_view *view, int ch)
2207 {
2208 	const struct got_error *err = NULL;
2209 	struct tog_log_view_state *s = &view->state.log;
2210 	char *parent_path, *in_repo_path = NULL;
2211 	struct tog_view *diff_view = NULL, *tree_view = NULL, *lv = NULL;
2212 	int begin_x = 0;
2213 	struct got_object_id *start_id;
2214 
2215 	switch (ch) {
2216 	case 'q':
2217 		s->quit = 1;
2218 		break;
2219 	case 'k':
2220 	case KEY_UP:
2221 	case '<':
2222 	case ',':
2223 		if (s->first_displayed_entry == NULL)
2224 			break;
2225 		if (s->selected > 0)
2226 			s->selected--;
2227 		else
2228 			scroll_up(view, &s->first_displayed_entry, 1,
2229 			    &s->commits);
2230 		break;
2231 	case KEY_PPAGE:
2232 	case CTRL('b'):
2233 		if (s->first_displayed_entry == NULL)
2234 			break;
2235 		if (TAILQ_FIRST(&s->commits.head) ==
2236 		    s->first_displayed_entry) {
2237 			s->selected = 0;
2238 			break;
2239 		}
2240 		scroll_up(view, &s->first_displayed_entry,
2241 		    view->nlines, &s->commits);
2242 		break;
2243 	case 'j':
2244 	case KEY_DOWN:
2245 	case '>':
2246 	case '.':
2247 		if (s->first_displayed_entry == NULL)
2248 			break;
2249 		if (s->selected < MIN(view->nlines - 2,
2250 		    s->commits.ncommits - 1)) {
2251 			s->selected++;
2252 			break;
2253 		}
2254 		err = scroll_down(view, &s->first_displayed_entry, 1,
2255 		    &s->last_displayed_entry, &s->commits,
2256 		    &s->thread_args.log_complete,
2257 		    &s->thread_args.commits_needed,
2258 		    &s->thread_args.need_commits);
2259 		break;
2260 	case KEY_NPAGE:
2261 	case CTRL('f'): {
2262 		struct commit_queue_entry *first;
2263 		first = s->first_displayed_entry;
2264 		if (first == NULL)
2265 			break;
2266 		err = scroll_down(view, &s->first_displayed_entry,
2267 		    view->nlines, &s->last_displayed_entry,
2268 		    &s->commits, &s->thread_args.log_complete,
2269 		    &s->thread_args.commits_needed,
2270 		    &s->thread_args.need_commits);
2271 		if (err)
2272 			break;
2273 		if (first == s->first_displayed_entry &&
2274 		    s->selected < MIN(view->nlines - 2,
2275 		    s->commits.ncommits - 1)) {
2276 			/* can't scroll further down */
2277 			s->selected = MIN(view->nlines - 2,
2278 			    s->commits.ncommits - 1);
2279 		}
2280 		err = NULL;
2281 		break;
2282 	}
2283 	case KEY_RESIZE:
2284 		if (s->selected > view->nlines - 2)
2285 			s->selected = view->nlines - 2;
2286 		if (s->selected > s->commits.ncommits - 1)
2287 			s->selected = s->commits.ncommits - 1;
2288 		break;
2289 	case KEY_ENTER:
2290 	case ' ':
2291 	case '\r':
2292 		if (s->selected_entry == NULL)
2293 			break;
2294 		if (view_is_parent_view(view))
2295 			begin_x = view_split_begin_x(view->begin_x);
2296 		err = open_diff_view_for_commit(&diff_view, begin_x,
2297 		    s->selected_entry->commit, s->selected_entry->id,
2298 		    view, s->refs, s->repo);
2299 		if (err)
2300 			break;
2301 		if (view_is_parent_view(view)) {
2302 			err = view_close_child(view);
2303 			if (err)
2304 				return err;
2305 			err = view_set_child(view, diff_view);
2306 			if (err) {
2307 				view_close(diff_view);
2308 				break;
2309 			}
2310 			*focus_view = diff_view;
2311 			view->child_focussed = 1;
2312 		} else
2313 			*new_view = diff_view;
2314 		break;
2315 	case 't':
2316 		if (s->selected_entry == NULL)
2317 			break;
2318 		if (view_is_parent_view(view))
2319 			begin_x = view_split_begin_x(view->begin_x);
2320 		err = browse_commit_tree(&tree_view, begin_x,
2321 		    s->selected_entry, s->in_repo_path, s->refs, s->repo);
2322 		if (err)
2323 			break;
2324 		if (view_is_parent_view(view)) {
2325 			err = view_close_child(view);
2326 			if (err)
2327 				return err;
2328 			err = view_set_child(view, tree_view);
2329 			if (err) {
2330 				view_close(tree_view);
2331 				break;
2332 			}
2333 			*focus_view = tree_view;
2334 			view->child_focussed = 1;
2335 		} else
2336 			*new_view = tree_view;
2337 		break;
2338 	case KEY_BACKSPACE:
2339 		if (strcmp(s->in_repo_path, "/") == 0)
2340 			break;
2341 		parent_path = dirname(s->in_repo_path);
2342 		if (parent_path && strcmp(parent_path, ".") != 0) {
2343 			err = stop_log_thread(s);
2344 			if (err)
2345 				return err;
2346 			lv = view_open(view->nlines, view->ncols,
2347 			    view->begin_y, view->begin_x, TOG_VIEW_LOG);
2348 			if (lv == NULL)
2349 				return got_error_from_errno(
2350 				    "view_open");
2351 			err = open_log_view(lv, s->start_id, s->refs,
2352 			    s->repo, s->head_ref_name, parent_path, 0);
2353 			if (err)
2354 				return err;;
2355 			if (view_is_parent_view(view))
2356 				*new_view = lv;
2357 			else {
2358 				view_set_child(view->parent, lv);
2359 				*focus_view = lv;
2360 			}
2361 			return NULL;
2362 		}
2363 		break;
2364 	case CTRL('l'):
2365 		err = stop_log_thread(s);
2366 		if (err)
2367 			return err;
2368 		lv = view_open(view->nlines, view->ncols,
2369 		    view->begin_y, view->begin_x, TOG_VIEW_LOG);
2370 		if (lv == NULL)
2371 			return got_error_from_errno("view_open");
2372 		err = get_head_commit_id(&start_id, s->head_ref_name ?
2373 		    s->head_ref_name : GOT_REF_HEAD, s->repo);
2374 		if (err) {
2375 			view_close(lv);
2376 			return err;
2377 		}
2378 		in_repo_path = strdup(s->in_repo_path);
2379 		if (in_repo_path == NULL) {
2380 			free(start_id);
2381 			view_close(lv);
2382 			return got_error_from_errno("strdup");
2383 		}
2384 		got_ref_list_free(s->refs);
2385 		err = got_ref_list(s->refs, s->repo, NULL,
2386 		    got_ref_cmp_by_name, NULL);
2387 		if (err) {
2388 			free(start_id);
2389 			view_close(lv);
2390 			return err;
2391 		}
2392 		err = open_log_view(lv, start_id, s->refs, s->repo,
2393 		    s->head_ref_name, in_repo_path, 0);
2394 		if (err) {
2395 			free(start_id);
2396 			view_close(lv);
2397 			return err;;
2398 		}
2399 		*dead_view = view;
2400 		*new_view = lv;
2401 		break;
2402 	default:
2403 		break;
2404 	}
2405 
2406 	return err;
2407 }
2408 
2409 static const struct got_error *
2410 apply_unveil(const char *repo_path, const char *worktree_path)
2411 {
2412 	const struct got_error *error;
2413 
2414 #ifdef PROFILE
2415 	if (unveil("gmon.out", "rwc") != 0)
2416 		return got_error_from_errno2("unveil", "gmon.out");
2417 #endif
2418 	if (repo_path && unveil(repo_path, "r") != 0)
2419 		return got_error_from_errno2("unveil", repo_path);
2420 
2421 	if (worktree_path && unveil(worktree_path, "rwc") != 0)
2422 		return got_error_from_errno2("unveil", worktree_path);
2423 
2424 	if (unveil("/tmp", "rwc") != 0)
2425 		return got_error_from_errno2("unveil", "/tmp");
2426 
2427 	error = got_privsep_unveil_exec_helpers();
2428 	if (error != NULL)
2429 		return error;
2430 
2431 	if (unveil(NULL, NULL) != 0)
2432 		return got_error_from_errno("unveil");
2433 
2434 	return NULL;
2435 }
2436 
2437 static void
2438 init_curses(void)
2439 {
2440 	initscr();
2441 	cbreak();
2442 	halfdelay(1); /* Do fast refresh while initial view is loading. */
2443 	noecho();
2444 	nonl();
2445 	intrflush(stdscr, FALSE);
2446 	keypad(stdscr, TRUE);
2447 	curs_set(0);
2448 	if (getenv("TOG_COLORS") != NULL) {
2449 		start_color();
2450 		use_default_colors();
2451 	}
2452 	signal(SIGWINCH, tog_sigwinch);
2453 	signal(SIGPIPE, tog_sigpipe);
2454 }
2455 
2456 static const struct got_error *
2457 cmd_log(int argc, char *argv[])
2458 {
2459 	const struct got_error *error;
2460 	struct got_repository *repo = NULL;
2461 	struct got_worktree *worktree = NULL;
2462 	struct got_reflist_head refs;
2463 	struct got_object_id *start_id = NULL;
2464 	char *path = NULL, *repo_path = NULL, *cwd = NULL;
2465 	char *start_commit = NULL, *head_ref_name = NULL;
2466 	int ch;
2467 	struct tog_view *view;
2468 
2469 	SIMPLEQ_INIT(&refs);
2470 
2471 #ifndef PROFILE
2472 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
2473 	    NULL) == -1)
2474 		err(1, "pledge");
2475 #endif
2476 
2477 	while ((ch = getopt(argc, argv, "c:r:")) != -1) {
2478 		switch (ch) {
2479 		case 'c':
2480 			start_commit = optarg;
2481 			break;
2482 		case 'r':
2483 			repo_path = realpath(optarg, NULL);
2484 			if (repo_path == NULL)
2485 				return got_error_from_errno2("realpath",
2486 				    optarg);
2487 			break;
2488 		default:
2489 			usage_log();
2490 			/* NOTREACHED */
2491 		}
2492 	}
2493 
2494 	argc -= optind;
2495 	argv += optind;
2496 
2497 	cwd = getcwd(NULL, 0);
2498 	if (cwd == NULL) {
2499 		error = got_error_from_errno("getcwd");
2500 		goto done;
2501 	}
2502 	error = got_worktree_open(&worktree, cwd);
2503 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
2504 		goto done;
2505 	error = NULL;
2506 
2507 	if (argc == 0) {
2508 		path = strdup("");
2509 		if (path == NULL) {
2510 			error = got_error_from_errno("strdup");
2511 			goto done;
2512 		}
2513 	} else if (argc == 1) {
2514 		if (worktree) {
2515 			error = got_worktree_resolve_path(&path, worktree,
2516 			    argv[0]);
2517 			if (error)
2518 				goto done;
2519 		} else {
2520 			path = strdup(argv[0]);
2521 			if (path == NULL) {
2522 				error = got_error_from_errno("strdup");
2523 				goto done;
2524 			}
2525 		}
2526 	} else
2527 		usage_log();
2528 
2529 	if (repo_path == NULL) {
2530 		if (worktree)
2531 			repo_path = strdup(
2532 			    got_worktree_get_repo_path(worktree));
2533 		else
2534 			repo_path = strdup(cwd);
2535 	}
2536 	if (repo_path == NULL) {
2537 		error = got_error_from_errno("strdup");
2538 		goto done;
2539 	}
2540 
2541 	init_curses();
2542 
2543 	error = got_repo_open(&repo, repo_path, NULL);
2544 	if (error != NULL)
2545 		goto done;
2546 
2547 	error = apply_unveil(got_repo_get_path(repo),
2548 	    worktree ? got_worktree_get_root_path(worktree) : NULL);
2549 	if (error)
2550 		goto done;
2551 
2552 	if (start_commit == NULL)
2553 		error = get_head_commit_id(&start_id, worktree ?
2554 		    got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
2555 		    repo);
2556 	else {
2557 		error = get_head_commit_id(&start_id, start_commit, repo);
2558 		if (error) {
2559 			if (error->code != GOT_ERR_NOT_REF)
2560 				goto done;
2561 			error = got_repo_match_object_id_prefix(&start_id,
2562 			    start_commit, GOT_OBJ_TYPE_COMMIT, repo);
2563 		}
2564 	}
2565 	if (error != NULL)
2566 		goto done;
2567 
2568 	error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
2569 	if (error)
2570 		goto done;
2571 
2572 	view = view_open(0, 0, 0, 0, TOG_VIEW_LOG);
2573 	if (view == NULL) {
2574 		error = got_error_from_errno("view_open");
2575 		goto done;
2576 	}
2577 	if (worktree) {
2578 		head_ref_name = strdup(
2579 		    got_worktree_get_head_ref_name(worktree));
2580 		if (head_ref_name == NULL) {
2581 			error = got_error_from_errno("strdup");
2582 			goto done;
2583 		}
2584 	}
2585 	error = open_log_view(view, start_id, &refs, repo, head_ref_name,
2586 	    path, 1);
2587 	if (error)
2588 		goto done;
2589 	if (worktree) {
2590 		/* Release work tree lock. */
2591 		got_worktree_close(worktree);
2592 		worktree = NULL;
2593 	}
2594 	error = view_loop(view);
2595 done:
2596 	free(repo_path);
2597 	free(cwd);
2598 	free(path);
2599 	free(start_id);
2600 	free(head_ref_name);
2601 	if (repo)
2602 		got_repo_close(repo);
2603 	if (worktree)
2604 		got_worktree_close(worktree);
2605 	got_ref_list_free(&refs);
2606 	return error;
2607 }
2608 
2609 __dead static void
2610 usage_diff(void)
2611 {
2612 	endwin();
2613 	fprintf(stderr, "usage: %s diff [repository-path] object1 object2\n",
2614 	    getprogname());
2615 	exit(1);
2616 }
2617 
2618 static char *
2619 parse_next_line(FILE *f, size_t *len)
2620 {
2621 	char *line;
2622 	size_t linelen;
2623 	size_t lineno;
2624 	const char delim[3] = { '\0', '\0', '\0'};
2625 
2626 	line = fparseln(f, &linelen, &lineno, delim, 0);
2627 	if (len)
2628 		*len = linelen;
2629 	return line;
2630 }
2631 
2632 static int
2633 match_line(const char *line, regex_t *regex)
2634 {
2635 	regmatch_t regmatch;
2636 
2637 	return regexec(regex, line, 1, &regmatch, 0) == 0;
2638 }
2639 
2640 struct tog_color *
2641 match_color(struct tog_colors *colors, const char *line)
2642 {
2643 	struct tog_color *tc = NULL;
2644 
2645 	SIMPLEQ_FOREACH(tc, colors, entry) {
2646 		if (match_line(line, &tc->regex))
2647 			return tc;
2648 	}
2649 
2650 	return NULL;
2651 }
2652 
2653 static const struct got_error *
2654 draw_file(struct tog_view *view, FILE *f, int *first_displayed_line,
2655     int *last_displayed_line, int *eof, int max_lines, char *header,
2656     struct tog_colors *colors)
2657 {
2658 	const struct got_error *err;
2659 	int nlines = 0, nprinted = 0;
2660 	char *line;
2661 	struct tog_color *tc;
2662 	size_t len;
2663 	wchar_t *wline;
2664 	int width;
2665 
2666 	rewind(f);
2667 	werase(view->window);
2668 
2669 	if (header) {
2670 		err = format_line(&wline, &width, header, view->ncols, 0);
2671 		if (err) {
2672 			return err;
2673 		}
2674 
2675 		if (view_needs_focus_indication(view))
2676 			wstandout(view->window);
2677 		waddwstr(view->window, wline);
2678 		if (view_needs_focus_indication(view))
2679 			wstandend(view->window);
2680 		if (width <= view->ncols - 1)
2681 			waddch(view->window, '\n');
2682 
2683 		if (max_lines <= 1)
2684 			return NULL;
2685 		max_lines--;
2686 	}
2687 
2688 	*eof = 0;
2689 	while (nprinted < max_lines) {
2690 		line = parse_next_line(f, &len);
2691 		if (line == NULL) {
2692 			*eof = 1;
2693 			break;
2694 		}
2695 		if (++nlines < *first_displayed_line) {
2696 			free(line);
2697 			continue;
2698 		}
2699 
2700 		err = format_line(&wline, &width, line, view->ncols, 0);
2701 		if (err) {
2702 			free(line);
2703 			return err;
2704 		}
2705 
2706 		tc = match_color(colors, line);
2707 		if (tc)
2708 			wattr_on(view->window,
2709 			    COLOR_PAIR(tc->colorpair), NULL);
2710 		waddwstr(view->window, wline);
2711 		if (tc)
2712 			wattr_off(view->window,
2713 			    COLOR_PAIR(tc->colorpair), NULL);
2714 		if (width <= view->ncols - 1)
2715 			waddch(view->window, '\n');
2716 		if (++nprinted == 1)
2717 			*first_displayed_line = nlines;
2718 		free(line);
2719 		free(wline);
2720 		wline = NULL;
2721 	}
2722 	*last_displayed_line = nlines;
2723 
2724 	view_vborder(view);
2725 
2726 	if (*eof) {
2727 		while (nprinted < view->nlines) {
2728 			waddch(view->window, '\n');
2729 			nprinted++;
2730 		}
2731 
2732 		err = format_line(&wline, &width, TOG_EOF_STRING, view->ncols, 0);
2733 		if (err) {
2734 			return err;
2735 		}
2736 
2737 		wstandout(view->window);
2738 		waddwstr(view->window, wline);
2739 		wstandend(view->window);
2740 	}
2741 
2742 	return NULL;
2743 }
2744 
2745 static char *
2746 get_datestr(time_t *time, char *datebuf)
2747 {
2748 	struct tm mytm, *tm;
2749 	char *p, *s;
2750 
2751 	tm = gmtime_r(time, &mytm);
2752 	if (tm == NULL)
2753 		return NULL;
2754 	s = asctime_r(tm, datebuf);
2755 	if (s == NULL)
2756 		return NULL;
2757 	p = strchr(s, '\n');
2758 	if (p)
2759 		*p = '\0';
2760 	return s;
2761 }
2762 
2763 static const struct got_error *
2764 write_commit_info(struct got_object_id *commit_id,
2765     struct got_reflist_head *refs, struct got_repository *repo, FILE *outfile)
2766 {
2767 	const struct got_error *err = NULL;
2768 	char datebuf[26], *datestr;
2769 	struct got_commit_object *commit;
2770 	char *id_str = NULL, *logmsg = NULL;
2771 	time_t committer_time;
2772 	const char *author, *committer;
2773 	char *refs_str = NULL;
2774 
2775 	if (refs) {
2776 		err = build_refs_str(&refs_str, refs, commit_id, repo);
2777 		if (err)
2778 			return err;
2779 	}
2780 
2781 	err = got_object_open_as_commit(&commit, repo, commit_id);
2782 	if (err)
2783 		return err;
2784 
2785 	err = got_object_id_str(&id_str, commit_id);
2786 	if (err) {
2787 		err = got_error_from_errno("got_object_id_str");
2788 		goto done;
2789 	}
2790 
2791 	if (fprintf(outfile, "commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
2792 	    refs_str ? refs_str : "", refs_str ? ")" : "") < 0) {
2793 		err = got_error_from_errno("fprintf");
2794 		goto done;
2795 	}
2796 	if (fprintf(outfile, "from: %s\n",
2797 	    got_object_commit_get_author(commit)) < 0) {
2798 		err = got_error_from_errno("fprintf");
2799 		goto done;
2800 	}
2801 	committer_time = got_object_commit_get_committer_time(commit);
2802 	datestr = get_datestr(&committer_time, datebuf);
2803 	if (datestr && fprintf(outfile, "date: %s UTC\n", datestr) < 0) {
2804 		err = got_error_from_errno("fprintf");
2805 		goto done;
2806 	}
2807 	author = got_object_commit_get_author(commit);
2808 	committer = got_object_commit_get_committer(commit);
2809 	if (strcmp(author, committer) != 0 &&
2810 	    fprintf(outfile, "via: %s\n", committer) < 0) {
2811 		err = got_error_from_errno("fprintf");
2812 		goto done;
2813 	}
2814 	err = got_object_commit_get_logmsg(&logmsg, commit);
2815 	if (err)
2816 		goto done;
2817 	if (fprintf(outfile, "%s\n", logmsg) < 0) {
2818 		err = got_error_from_errno("fprintf");
2819 		goto done;
2820 	}
2821 done:
2822 	free(id_str);
2823 	free(logmsg);
2824 	free(refs_str);
2825 	got_object_commit_close(commit);
2826 	return err;
2827 }
2828 
2829 static const struct got_error *
2830 create_diff(struct tog_diff_view_state *s)
2831 {
2832 	const struct got_error *err = NULL;
2833 	FILE *f = NULL;
2834 	int obj_type;
2835 
2836 	f = got_opentemp();
2837 	if (f == NULL) {
2838 		err = got_error_from_errno("got_opentemp");
2839 		goto done;
2840 	}
2841 	if (s->f && fclose(s->f) != 0) {
2842 		err = got_error_from_errno("fclose");
2843 		goto done;
2844 	}
2845 	s->f = f;
2846 
2847 	if (s->id1)
2848 		err = got_object_get_type(&obj_type, s->repo, s->id1);
2849 	else
2850 		err = got_object_get_type(&obj_type, s->repo, s->id2);
2851 	if (err)
2852 		goto done;
2853 
2854 	switch (obj_type) {
2855 	case GOT_OBJ_TYPE_BLOB:
2856 		err = got_diff_objects_as_blobs(s->id1, s->id2, NULL, NULL,
2857 		    s->diff_context, 0, s->repo, f);
2858 		break;
2859 	case GOT_OBJ_TYPE_TREE:
2860 		err = got_diff_objects_as_trees(s->id1, s->id2, "", "",
2861 		    s->diff_context, 0, s->repo, f);
2862 		break;
2863 	case GOT_OBJ_TYPE_COMMIT: {
2864 		const struct got_object_id_queue *parent_ids;
2865 		struct got_object_qid *pid;
2866 		struct got_commit_object *commit2;
2867 
2868 		err = got_object_open_as_commit(&commit2, s->repo, s->id2);
2869 		if (err)
2870 			break;
2871 		/* Show commit info if we're diffing to a parent/root commit. */
2872 		if (s->id1 == NULL)
2873 			write_commit_info(s->id2, s->refs, s->repo, f);
2874 		else {
2875 			parent_ids = got_object_commit_get_parent_ids(commit2);
2876 			SIMPLEQ_FOREACH(pid, parent_ids, entry) {
2877 				if (got_object_id_cmp(s->id1, pid->id) == 0) {
2878 					write_commit_info(s->id2, s->refs,
2879 					    s->repo, f);
2880 					break;
2881 				}
2882 			}
2883 		}
2884 		got_object_commit_close(commit2);
2885 
2886 		err = got_diff_objects_as_commits(s->id1, s->id2,
2887 		    s->diff_context, 0, s->repo, f);
2888 		break;
2889 	}
2890 	default:
2891 		err = got_error(GOT_ERR_OBJ_TYPE);
2892 		break;
2893 	}
2894 done:
2895 	if (f && fflush(f) != 0 && err == NULL)
2896 		err = got_error_from_errno("fflush");
2897 	return err;
2898 }
2899 
2900 static void
2901 diff_view_indicate_progress(struct tog_view *view)
2902 {
2903 	mvwaddstr(view->window, 0, 0, "diffing...");
2904 	update_panels();
2905 	doupdate();
2906 }
2907 
2908 static const struct got_error *
2909 open_diff_view(struct tog_view *view, struct got_object_id *id1,
2910     struct got_object_id *id2, struct tog_view *log_view,
2911     struct got_reflist_head *refs, struct got_repository *repo)
2912 {
2913 	const struct got_error *err;
2914 
2915 	if (id1 != NULL && id2 != NULL) {
2916 	    int type1, type2;
2917 	    err = got_object_get_type(&type1, repo, id1);
2918 	    if (err)
2919 		return err;
2920 	    err = got_object_get_type(&type2, repo, id2);
2921 	    if (err)
2922 		return err;
2923 
2924 	    if (type1 != type2)
2925 		return got_error(GOT_ERR_OBJ_TYPE);
2926 	}
2927 
2928 	if (id1) {
2929 		view->state.diff.id1 = got_object_id_dup(id1);
2930 		if (view->state.diff.id1 == NULL)
2931 			return got_error_from_errno("got_object_id_dup");
2932 	} else
2933 		view->state.diff.id1 = NULL;
2934 
2935 	view->state.diff.id2 = got_object_id_dup(id2);
2936 	if (view->state.diff.id2 == NULL) {
2937 		free(view->state.diff.id1);
2938 		view->state.diff.id1 = NULL;
2939 		return got_error_from_errno("got_object_id_dup");
2940 	}
2941 	view->state.diff.f = NULL;
2942 	view->state.diff.first_displayed_line = 1;
2943 	view->state.diff.last_displayed_line = view->nlines;
2944 	view->state.diff.diff_context = 3;
2945 	view->state.diff.log_view = log_view;
2946 	view->state.diff.repo = repo;
2947 	view->state.diff.refs = refs;
2948 	SIMPLEQ_INIT(&view->state.diff.colors);
2949 
2950 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
2951 		err = add_color(&view->state.diff.colors,
2952 		    "^-", TOG_COLOR_DIFF_MINUS,
2953 		    get_color_value("TOG_COLOR_DIFF_MINUS"));
2954 		if (err)
2955 			return err;
2956 		err = add_color(&view->state.diff.colors, "^\\+",
2957 		    TOG_COLOR_DIFF_PLUS,
2958 		    get_color_value("TOG_COLOR_DIFF_PLUS"));
2959 		if (err) {
2960 			free_colors(&view->state.diff.colors);
2961 			return err;
2962 		}
2963 		err = add_color(&view->state.diff.colors, 
2964 		    "^@@", TOG_COLOR_DIFF_CHUNK_HEADER,
2965 		    get_color_value("TOG_COLOR_DIFF_CHUNK_HEADER"));
2966 		if (err) {
2967 			free_colors(&view->state.diff.colors);
2968 			return err;
2969 		}
2970 
2971 		err = add_color(&view->state.diff.colors, 
2972 		    "^(commit|(blob|file) [-+] )", TOG_COLOR_DIFF_META,
2973 		    get_color_value("TOG_COLOR_DIFF_META"));
2974 		if (err) {
2975 			free_colors(&view->state.diff.colors);
2976 			return err;
2977 		}
2978 
2979 		err = add_color(&view->state.diff.colors, 
2980 		    "^(from|via): ", TOG_COLOR_AUTHOR,
2981 		    get_color_value("TOG_COLOR_AUTHOR"));
2982 		if (err) {
2983 			free_colors(&view->state.diff.colors);
2984 			return err;
2985 		}
2986 
2987 		err = add_color(&view->state.diff.colors, 
2988 		    "^date: ", TOG_COLOR_DATE,
2989 		    get_color_value("TOG_COLOR_DATE"));
2990 		if (err) {
2991 			free_colors(&view->state.diff.colors);
2992 			return err;
2993 		}
2994 	}
2995 
2996 	if (log_view && view_is_splitscreen(view))
2997 		show_log_view(log_view); /* draw vborder */
2998 	diff_view_indicate_progress(view);
2999 
3000 	err = create_diff(&view->state.diff);
3001 	if (err) {
3002 		free(view->state.diff.id1);
3003 		view->state.diff.id1 = NULL;
3004 		free(view->state.diff.id2);
3005 		view->state.diff.id2 = NULL;
3006 		return err;
3007 	}
3008 
3009 	view->show = show_diff_view;
3010 	view->input = input_diff_view;
3011 	view->close = close_diff_view;
3012 
3013 	return NULL;
3014 }
3015 
3016 static const struct got_error *
3017 close_diff_view(struct tog_view *view)
3018 {
3019 	const struct got_error *err = NULL;
3020 
3021 	free(view->state.diff.id1);
3022 	view->state.diff.id1 = NULL;
3023 	free(view->state.diff.id2);
3024 	view->state.diff.id2 = NULL;
3025 	if (view->state.diff.f && fclose(view->state.diff.f) == EOF)
3026 		err = got_error_from_errno("fclose");
3027 	free_colors(&view->state.diff.colors);
3028 	return err;
3029 }
3030 
3031 static const struct got_error *
3032 show_diff_view(struct tog_view *view)
3033 {
3034 	const struct got_error *err;
3035 	struct tog_diff_view_state *s = &view->state.diff;
3036 	char *id_str1 = NULL, *id_str2, *header;
3037 
3038 	if (s->id1) {
3039 		err = got_object_id_str(&id_str1, s->id1);
3040 		if (err)
3041 			return err;
3042 	}
3043 	err = got_object_id_str(&id_str2, s->id2);
3044 	if (err)
3045 		return err;
3046 
3047 	if (asprintf(&header, "diff %s %s",
3048 	    id_str1 ? id_str1 : "/dev/null", id_str2) == -1) {
3049 		err = got_error_from_errno("asprintf");
3050 		free(id_str1);
3051 		free(id_str2);
3052 		return err;
3053 	}
3054 	free(id_str1);
3055 	free(id_str2);
3056 
3057 	return draw_file(view, s->f, &s->first_displayed_line,
3058 	    &s->last_displayed_line, &s->eof, view->nlines,
3059 	    header, &s->colors);
3060 }
3061 
3062 static const struct got_error *
3063 set_selected_commit(struct tog_diff_view_state *s,
3064     struct commit_queue_entry *entry)
3065 {
3066 	const struct got_error *err;
3067 	const struct got_object_id_queue *parent_ids;
3068 	struct got_commit_object *selected_commit;
3069 	struct got_object_qid *pid;
3070 
3071 	free(s->id2);
3072 	s->id2 = got_object_id_dup(entry->id);
3073 	if (s->id2 == NULL)
3074 		return got_error_from_errno("got_object_id_dup");
3075 
3076 	err = got_object_open_as_commit(&selected_commit, s->repo, entry->id);
3077 	if (err)
3078 		return err;
3079 	parent_ids = got_object_commit_get_parent_ids(selected_commit);
3080 	free(s->id1);
3081 	pid = SIMPLEQ_FIRST(parent_ids);
3082 	s->id1 = pid ? got_object_id_dup(pid->id) : NULL;
3083 	got_object_commit_close(selected_commit);
3084 	return NULL;
3085 }
3086 
3087 static const struct got_error *
3088 input_diff_view(struct tog_view **new_view, struct tog_view **dead_view,
3089     struct tog_view **focus_view, struct tog_view *view, int ch)
3090 {
3091 	const struct got_error *err = NULL;
3092 	struct tog_diff_view_state *s = &view->state.diff;
3093 	struct tog_log_view_state *ls;
3094 	struct commit_queue_entry *entry;
3095 	int i;
3096 
3097 	switch (ch) {
3098 	case 'k':
3099 	case KEY_UP:
3100 		if (s->first_displayed_line > 1)
3101 			s->first_displayed_line--;
3102 		break;
3103 	case KEY_PPAGE:
3104 	case CTRL('b'):
3105 		if (s->first_displayed_line == 1)
3106 			break;
3107 		i = 0;
3108 		while (i++ < view->nlines - 1 &&
3109 		    s->first_displayed_line > 1)
3110 			s->first_displayed_line--;
3111 		break;
3112 	case 'j':
3113 	case KEY_DOWN:
3114 		if (!s->eof)
3115 			s->first_displayed_line++;
3116 		break;
3117 	case KEY_NPAGE:
3118 	case CTRL('f'):
3119 	case ' ':
3120 		if (s->eof)
3121 			break;
3122 		i = 0;
3123 		while (!s->eof && i++ < view->nlines - 1) {
3124 			char *line;
3125 			line = parse_next_line(s->f, NULL);
3126 			s->first_displayed_line++;
3127 			if (line == NULL)
3128 				break;
3129 		}
3130 		break;
3131 	case '[':
3132 		if (s->diff_context > 0) {
3133 			s->diff_context--;
3134 			diff_view_indicate_progress(view);
3135 			err = create_diff(s);
3136 		}
3137 		break;
3138 	case ']':
3139 		if (s->diff_context < GOT_DIFF_MAX_CONTEXT) {
3140 			s->diff_context++;
3141 			diff_view_indicate_progress(view);
3142 			err = create_diff(s);
3143 		}
3144 		break;
3145 	case '<':
3146 	case ',':
3147 		if (s->log_view == NULL)
3148 			break;
3149 		ls = &s->log_view->state.log;
3150 		entry = TAILQ_PREV(ls->selected_entry,
3151 		    commit_queue_head, entry);
3152 		if (entry == NULL)
3153 			break;
3154 
3155 		err = input_log_view(NULL, NULL, NULL, s->log_view,
3156 		    KEY_UP);
3157 		if (err)
3158 			break;
3159 
3160 		err = set_selected_commit(s, entry);
3161 		if (err)
3162 			break;
3163 
3164 		s->first_displayed_line = 1;
3165 		s->last_displayed_line = view->nlines;
3166 
3167 		diff_view_indicate_progress(view);
3168 		err = create_diff(s);
3169 		break;
3170 	case '>':
3171 	case '.':
3172 		if (s->log_view == NULL)
3173 			break;
3174 		ls = &s->log_view->state.log;
3175 
3176 		if (TAILQ_NEXT(ls->selected_entry, entry) == NULL) {
3177 			ls->thread_args.commits_needed++;
3178 
3179 			/* Display "loading..." in log view. */
3180 			show_log_view(s->log_view);
3181 			update_panels();
3182 			doupdate();
3183 
3184 			err = trigger_log_thread(1 /* load_all */,
3185 			    &ls->thread_args.commits_needed,
3186 			    &ls->thread_args.log_complete,
3187 			    &ls->thread_args.need_commits);
3188 			if (err)
3189 				break;
3190 		}
3191 		err = input_log_view(NULL, NULL, NULL, s->log_view,
3192 		    KEY_DOWN);
3193 		if (err)
3194 			break;
3195 
3196 		entry = TAILQ_NEXT(ls->selected_entry, entry);
3197 		if (entry == NULL)
3198 			break;
3199 
3200 		err = set_selected_commit(s, entry);
3201 		if (err)
3202 			break;
3203 
3204 		s->first_displayed_line = 1;
3205 		s->last_displayed_line = view->nlines;
3206 
3207 		diff_view_indicate_progress(view);
3208 		err = create_diff(s);
3209 		break;
3210 	default:
3211 		break;
3212 	}
3213 
3214 	return err;
3215 }
3216 
3217 static const struct got_error *
3218 cmd_diff(int argc, char *argv[])
3219 {
3220 	const struct got_error *error = NULL;
3221 	struct got_repository *repo = NULL;
3222 	struct got_reflist_head refs;
3223 	struct got_object_id *id1 = NULL, *id2 = NULL;
3224 	char *repo_path = NULL;
3225 	char *id_str1 = NULL, *id_str2 = NULL;
3226 	int ch;
3227 	struct tog_view *view;
3228 
3229 	SIMPLEQ_INIT(&refs);
3230 
3231 #ifndef PROFILE
3232 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
3233 	    NULL) == -1)
3234 		err(1, "pledge");
3235 #endif
3236 
3237 	while ((ch = getopt(argc, argv, "")) != -1) {
3238 		switch (ch) {
3239 		default:
3240 			usage_diff();
3241 			/* NOTREACHED */
3242 		}
3243 	}
3244 
3245 	argc -= optind;
3246 	argv += optind;
3247 
3248 	if (argc == 0) {
3249 		usage_diff(); /* TODO show local worktree changes */
3250 	} else if (argc == 2) {
3251 		repo_path = getcwd(NULL, 0);
3252 		if (repo_path == NULL)
3253 			return got_error_from_errno("getcwd");
3254 		id_str1 = argv[0];
3255 		id_str2 = argv[1];
3256 	} else if (argc == 3) {
3257 		repo_path = realpath(argv[0], NULL);
3258 		if (repo_path == NULL)
3259 			return got_error_from_errno2("realpath", argv[0]);
3260 		id_str1 = argv[1];
3261 		id_str2 = argv[2];
3262 	} else
3263 		usage_diff();
3264 
3265 	init_curses();
3266 
3267 	error = got_repo_open(&repo, repo_path, NULL);
3268 	if (error)
3269 		goto done;
3270 
3271 	error = apply_unveil(got_repo_get_path(repo), NULL);
3272 	if (error)
3273 		goto done;
3274 
3275 	error = got_repo_match_object_id_prefix(&id1, id_str1,
3276 	    GOT_OBJ_TYPE_ANY, repo);
3277 	if (error)
3278 		goto done;
3279 
3280 	error = got_repo_match_object_id_prefix(&id2, id_str2,
3281 	    GOT_OBJ_TYPE_ANY, repo);
3282 	if (error)
3283 		goto done;
3284 
3285 	error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
3286 	if (error)
3287 		goto done;
3288 
3289 	view = view_open(0, 0, 0, 0, TOG_VIEW_DIFF);
3290 	if (view == NULL) {
3291 		error = got_error_from_errno("view_open");
3292 		goto done;
3293 	}
3294 	error = open_diff_view(view, id1, id2, NULL, &refs, repo);
3295 	if (error)
3296 		goto done;
3297 	error = view_loop(view);
3298 done:
3299 	free(repo_path);
3300 	if (repo)
3301 		got_repo_close(repo);
3302 	got_ref_list_free(&refs);
3303 	return error;
3304 }
3305 
3306 __dead static void
3307 usage_blame(void)
3308 {
3309 	endwin();
3310 	fprintf(stderr, "usage: %s blame [-c commit] [-r repository-path] path\n",
3311 	    getprogname());
3312 	exit(1);
3313 }
3314 
3315 struct tog_blame_line {
3316 	int annotated;
3317 	struct got_object_id *id;
3318 };
3319 
3320 static const struct got_error *
3321 draw_blame(struct tog_view *view, struct got_object_id *id, FILE *f,
3322     const char *path, struct tog_blame_line *lines, int nlines,
3323     int blame_complete, int selected_line, int *first_displayed_line,
3324     int *last_displayed_line, int *eof, int max_lines,
3325     struct tog_colors *colors)
3326 {
3327 	const struct got_error *err;
3328 	int lineno = 0, nprinted = 0;
3329 	char *line;
3330 	size_t len;
3331 	wchar_t *wline;
3332 	int width;
3333 	struct tog_blame_line *blame_line;
3334 	struct got_object_id *prev_id = NULL;
3335 	char *id_str;
3336 	struct tog_color *tc;
3337 
3338 	err = got_object_id_str(&id_str, id);
3339 	if (err)
3340 		return err;
3341 
3342 	rewind(f);
3343 	werase(view->window);
3344 
3345 	if (asprintf(&line, "commit %s", id_str) == -1) {
3346 		err = got_error_from_errno("asprintf");
3347 		free(id_str);
3348 		return err;
3349 	}
3350 
3351 	err = format_line(&wline, &width, line, view->ncols, 0);
3352 	free(line);
3353 	line = NULL;
3354 	if (err)
3355 		return err;
3356 	if (view_needs_focus_indication(view))
3357 		wstandout(view->window);
3358 	tc = get_color(colors, TOG_COLOR_COMMIT);
3359 	if (tc)
3360 		wattr_on(view->window,
3361 		    COLOR_PAIR(tc->colorpair), NULL);
3362 	waddwstr(view->window, wline);
3363 	if (tc)
3364 		wattr_off(view->window,
3365 		    COLOR_PAIR(tc->colorpair), NULL);
3366 	if (view_needs_focus_indication(view))
3367 		wstandend(view->window);
3368 	free(wline);
3369 	wline = NULL;
3370 	if (width < view->ncols - 1)
3371 		waddch(view->window, '\n');
3372 
3373 	if (asprintf(&line, "[%d/%d] %s%s",
3374 	    *first_displayed_line - 1 + selected_line, nlines,
3375 	    blame_complete ? "" : "annotating... ", path) == -1) {
3376 		free(id_str);
3377 		return got_error_from_errno("asprintf");
3378 	}
3379 	free(id_str);
3380 	err = format_line(&wline, &width, line, view->ncols, 0);
3381 	free(line);
3382 	line = NULL;
3383 	if (err)
3384 		return err;
3385 	waddwstr(view->window, wline);
3386 	free(wline);
3387 	wline = NULL;
3388 	if (width < view->ncols - 1)
3389 		waddch(view->window, '\n');
3390 
3391 	*eof = 0;
3392 	while (nprinted < max_lines - 2) {
3393 		line = parse_next_line(f, &len);
3394 		if (line == NULL) {
3395 			*eof = 1;
3396 			break;
3397 		}
3398 		if (++lineno < *first_displayed_line) {
3399 			free(line);
3400 			continue;
3401 		}
3402 
3403 		if (view->ncols <= 9) {
3404 			width = 9;
3405 			wline = wcsdup(L"");
3406 			if (wline == NULL)
3407 				err = got_error_from_errno("wcsdup");
3408 		} else {
3409 			err = format_line(&wline, &width, line,
3410 			    view->ncols - 9, 9);
3411 			width += 9;
3412 		}
3413 		if (err) {
3414 			free(line);
3415 			return err;
3416 		}
3417 
3418 		if (view->focussed && nprinted == selected_line - 1)
3419 			wstandout(view->window);
3420 
3421 		if (nlines > 0) {
3422 			blame_line = &lines[lineno - 1];
3423 			if (blame_line->annotated && prev_id &&
3424 			    got_object_id_cmp(prev_id, blame_line->id) == 0 &&
3425 			    !(view->focussed &&
3426 			    nprinted == selected_line - 1)) {
3427 				waddstr(view->window, "        ");
3428 			} else if (blame_line->annotated) {
3429 				char *id_str;
3430 				err = got_object_id_str(&id_str, blame_line->id);
3431 				if (err) {
3432 					free(line);
3433 					free(wline);
3434 					return err;
3435 				}
3436 				tc = get_color(colors, TOG_COLOR_COMMIT);
3437 				if (tc)
3438 					wattr_on(view->window,
3439 					    COLOR_PAIR(tc->colorpair), NULL);
3440 				wprintw(view->window, "%.8s", id_str);
3441 				if (tc)
3442 					wattr_off(view->window,
3443 					    COLOR_PAIR(tc->colorpair), NULL);
3444 				free(id_str);
3445 				prev_id = blame_line->id;
3446 			} else {
3447 				waddstr(view->window, "........");
3448 				prev_id = NULL;
3449 			}
3450 		} else {
3451 			waddstr(view->window, "........");
3452 			prev_id = NULL;
3453 		}
3454 
3455 		if (view->focussed && nprinted == selected_line - 1)
3456 			wstandend(view->window);
3457 		waddstr(view->window, " ");
3458 
3459 		waddwstr(view->window, wline);
3460 		if (width <= view->ncols - 1)
3461 			waddch(view->window, '\n');
3462 		if (++nprinted == 1)
3463 			*first_displayed_line = lineno;
3464 		free(line);
3465 		free(wline);
3466 		wline = NULL;
3467 	}
3468 	*last_displayed_line = lineno;
3469 
3470 	view_vborder(view);
3471 
3472 	return NULL;
3473 }
3474 
3475 static const struct got_error *
3476 blame_cb(void *arg, int nlines, int lineno, struct got_object_id *id)
3477 {
3478 	const struct got_error *err = NULL;
3479 	struct tog_blame_cb_args *a = arg;
3480 	struct tog_blame_line *line;
3481 	int errcode;
3482 
3483 	if (nlines != a->nlines ||
3484 	    (lineno != -1 && lineno < 1) || lineno > a->nlines)
3485 		return got_error(GOT_ERR_RANGE);
3486 
3487 	errcode = pthread_mutex_lock(&tog_mutex);
3488 	if (errcode)
3489 		return got_error_set_errno(errcode, "pthread_mutex_lock");
3490 
3491 	if (*a->quit) {	/* user has quit the blame view */
3492 		err = got_error(GOT_ERR_ITER_COMPLETED);
3493 		goto done;
3494 	}
3495 
3496 	if (lineno == -1)
3497 		goto done; /* no change in this commit */
3498 
3499 	line = &a->lines[lineno - 1];
3500 	if (line->annotated)
3501 		goto done;
3502 
3503 	line->id = got_object_id_dup(id);
3504 	if (line->id == NULL) {
3505 		err = got_error_from_errno("got_object_id_dup");
3506 		goto done;
3507 	}
3508 	line->annotated = 1;
3509 done:
3510 	errcode = pthread_mutex_unlock(&tog_mutex);
3511 	if (errcode)
3512 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
3513 	return err;
3514 }
3515 
3516 static void *
3517 blame_thread(void *arg)
3518 {
3519 	const struct got_error *err;
3520 	struct tog_blame_thread_args *ta = arg;
3521 	struct tog_blame_cb_args *a = ta->cb_args;
3522 	int errcode;
3523 
3524 	err = got_blame(ta->path, a->commit_id, ta->repo,
3525 	    blame_cb, ta->cb_args, ta->cancel_cb, ta->cancel_arg);
3526 	if (err && err->code == GOT_ERR_CANCELLED)
3527 		err = NULL;
3528 
3529 	errcode = pthread_mutex_lock(&tog_mutex);
3530 	if (errcode)
3531 		return (void *)got_error_set_errno(errcode,
3532 		    "pthread_mutex_lock");
3533 
3534 	got_repo_close(ta->repo);
3535 	ta->repo = NULL;
3536 	*ta->complete = 1;
3537 
3538 	errcode = pthread_mutex_unlock(&tog_mutex);
3539 	if (errcode && err == NULL)
3540 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
3541 
3542 	return (void *)err;
3543 }
3544 
3545 static struct got_object_id *
3546 get_selected_commit_id(struct tog_blame_line *lines, int nlines,
3547     int first_displayed_line, int selected_line)
3548 {
3549 	struct tog_blame_line *line;
3550 
3551 	if (nlines <= 0)
3552 		return NULL;
3553 
3554 	line = &lines[first_displayed_line - 1 + selected_line - 1];
3555 	if (!line->annotated)
3556 		return NULL;
3557 
3558 	return line->id;
3559 }
3560 
3561 static const struct got_error *
3562 stop_blame(struct tog_blame *blame)
3563 {
3564 	const struct got_error *err = NULL;
3565 	int i;
3566 
3567 	if (blame->thread) {
3568 		int errcode;
3569 		errcode = pthread_mutex_unlock(&tog_mutex);
3570 		if (errcode)
3571 			return got_error_set_errno(errcode,
3572 			    "pthread_mutex_unlock");
3573 		errcode = pthread_join(blame->thread, (void **)&err);
3574 		if (errcode)
3575 			return got_error_set_errno(errcode, "pthread_join");
3576 		errcode = pthread_mutex_lock(&tog_mutex);
3577 		if (errcode)
3578 			return got_error_set_errno(errcode,
3579 			    "pthread_mutex_lock");
3580 		if (err && err->code == GOT_ERR_ITER_COMPLETED)
3581 			err = NULL;
3582 		blame->thread = NULL;
3583 	}
3584 	if (blame->thread_args.repo) {
3585 		got_repo_close(blame->thread_args.repo);
3586 		blame->thread_args.repo = NULL;
3587 	}
3588 	if (blame->f) {
3589 		if (fclose(blame->f) != 0 && err == NULL)
3590 			err = got_error_from_errno("fclose");
3591 		blame->f = NULL;
3592 	}
3593 	if (blame->lines) {
3594 		for (i = 0; i < blame->nlines; i++)
3595 			free(blame->lines[i].id);
3596 		free(blame->lines);
3597 		blame->lines = NULL;
3598 	}
3599 	free(blame->cb_args.commit_id);
3600 	blame->cb_args.commit_id = NULL;
3601 
3602 	return err;
3603 }
3604 
3605 static const struct got_error *
3606 cancel_blame_view(void *arg)
3607 {
3608 	const struct got_error *err = NULL;
3609 	int *done = arg;
3610 	int errcode;
3611 
3612 	errcode = pthread_mutex_lock(&tog_mutex);
3613 	if (errcode)
3614 		return got_error_set_errno(errcode,
3615 		    "pthread_mutex_unlock");
3616 
3617 	if (*done)
3618 		err = got_error(GOT_ERR_CANCELLED);
3619 
3620 	errcode = pthread_mutex_unlock(&tog_mutex);
3621 	if (errcode)
3622 		return got_error_set_errno(errcode,
3623 		    "pthread_mutex_lock");
3624 
3625 	return err;
3626 }
3627 
3628 static const struct got_error *
3629 run_blame(struct tog_blame *blame, struct tog_view *view, int *blame_complete,
3630     int *first_displayed_line, int *last_displayed_line, int *selected_line,
3631     int *done, int *eof, const char *path, struct got_object_id *commit_id,
3632     struct got_repository *repo)
3633 {
3634 	const struct got_error *err = NULL;
3635 	struct got_blob_object *blob = NULL;
3636 	struct got_repository *thread_repo = NULL;
3637 	struct got_object_id *obj_id = NULL;
3638 	int obj_type;
3639 
3640 	err = got_object_id_by_path(&obj_id, repo, commit_id, path);
3641 	if (err)
3642 		return err;
3643 	if (obj_id == NULL)
3644 		return got_error(GOT_ERR_NO_OBJ);
3645 
3646 	err = got_object_get_type(&obj_type, repo, obj_id);
3647 	if (err)
3648 		goto done;
3649 
3650 	if (obj_type != GOT_OBJ_TYPE_BLOB) {
3651 		err = got_error(GOT_ERR_OBJ_TYPE);
3652 		goto done;
3653 	}
3654 
3655 	err = got_object_open_as_blob(&blob, repo, obj_id, 8192);
3656 	if (err)
3657 		goto done;
3658 	blame->f = got_opentemp();
3659 	if (blame->f == NULL) {
3660 		err = got_error_from_errno("got_opentemp");
3661 		goto done;
3662 	}
3663 	err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
3664 	    &blame->line_offsets, blame->f, blob);
3665 	if (err || blame->nlines == 0)
3666 		goto done;
3667 
3668 	/* Don't include \n at EOF in the blame line count. */
3669 	if (blame->line_offsets[blame->nlines - 1] == blame->filesize)
3670 		blame->nlines--;
3671 
3672 	blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
3673 	if (blame->lines == NULL) {
3674 		err = got_error_from_errno("calloc");
3675 		goto done;
3676 	}
3677 
3678 	err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL);
3679 	if (err)
3680 		goto done;
3681 
3682 	blame->cb_args.view = view;
3683 	blame->cb_args.lines = blame->lines;
3684 	blame->cb_args.nlines = blame->nlines;
3685 	blame->cb_args.commit_id = got_object_id_dup(commit_id);
3686 	if (blame->cb_args.commit_id == NULL) {
3687 		err = got_error_from_errno("got_object_id_dup");
3688 		goto done;
3689 	}
3690 	blame->cb_args.quit = done;
3691 
3692 	blame->thread_args.path = path;
3693 	blame->thread_args.repo = thread_repo;
3694 	blame->thread_args.cb_args = &blame->cb_args;
3695 	blame->thread_args.complete = blame_complete;
3696 	blame->thread_args.cancel_cb = cancel_blame_view;
3697 	blame->thread_args.cancel_arg = done;
3698 	*blame_complete = 0;
3699 
3700 done:
3701 	if (blob)
3702 		got_object_blob_close(blob);
3703 	free(obj_id);
3704 	if (err)
3705 		stop_blame(blame);
3706 	return err;
3707 }
3708 
3709 static const struct got_error *
3710 open_blame_view(struct tog_view *view, char *path,
3711     struct got_object_id *commit_id, struct got_reflist_head *refs,
3712     struct got_repository *repo)
3713 {
3714 	const struct got_error *err = NULL;
3715 	struct tog_blame_view_state *s = &view->state.blame;
3716 
3717 	SIMPLEQ_INIT(&s->blamed_commits);
3718 
3719 	s->path = strdup(path);
3720 	if (s->path == NULL)
3721 		return got_error_from_errno("strdup");
3722 
3723 	err = got_object_qid_alloc(&s->blamed_commit, commit_id);
3724 	if (err) {
3725 		free(s->path);
3726 		return err;
3727 	}
3728 
3729 	SIMPLEQ_INSERT_HEAD(&s->blamed_commits, s->blamed_commit, entry);
3730 	s->first_displayed_line = 1;
3731 	s->last_displayed_line = view->nlines;
3732 	s->selected_line = 1;
3733 	s->blame_complete = 0;
3734 	s->repo = repo;
3735 	s->refs = refs;
3736 	s->commit_id = commit_id;
3737 	memset(&s->blame, 0, sizeof(s->blame));
3738 
3739 	SIMPLEQ_INIT(&s->colors);
3740 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
3741 		err = add_color(&s->colors, "^", TOG_COLOR_COMMIT,
3742 		    get_color_value("TOG_COLOR_COMMIT"));
3743 		if (err)
3744 			return err;
3745 	}
3746 
3747 	view->show = show_blame_view;
3748 	view->input = input_blame_view;
3749 	view->close = close_blame_view;
3750 	view->search_start = search_start_blame_view;
3751 	view->search_next = search_next_blame_view;
3752 
3753 	return run_blame(&s->blame, view, &s->blame_complete,
3754 	    &s->first_displayed_line, &s->last_displayed_line,
3755 	    &s->selected_line, &s->done, &s->eof, s->path,
3756 	    s->blamed_commit->id, s->repo);
3757 }
3758 
3759 static const struct got_error *
3760 close_blame_view(struct tog_view *view)
3761 {
3762 	const struct got_error *err = NULL;
3763 	struct tog_blame_view_state *s = &view->state.blame;
3764 
3765 	if (s->blame.thread)
3766 		err = stop_blame(&s->blame);
3767 
3768 	while (!SIMPLEQ_EMPTY(&s->blamed_commits)) {
3769 		struct got_object_qid *blamed_commit;
3770 		blamed_commit = SIMPLEQ_FIRST(&s->blamed_commits);
3771 		SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
3772 		got_object_qid_free(blamed_commit);
3773 	}
3774 
3775 	free(s->path);
3776 	free_colors(&s->colors);
3777 
3778 	return err;
3779 }
3780 
3781 static const struct got_error *
3782 search_start_blame_view(struct tog_view *view)
3783 {
3784 	struct tog_blame_view_state *s = &view->state.blame;
3785 
3786 	s->matched_line = 0;
3787 	return NULL;
3788 }
3789 
3790 static const struct got_error *
3791 search_next_blame_view(struct tog_view *view)
3792 {
3793 	struct tog_blame_view_state *s = &view->state.blame;
3794 	int lineno;
3795 
3796 	if (!view->searching) {
3797 		view->search_next_done = 1;
3798 		return NULL;
3799 	}
3800 
3801 	if (s->matched_line) {
3802 		if (view->searching == TOG_SEARCH_FORWARD)
3803 			lineno = s->matched_line + 1;
3804 		else
3805 			lineno = s->matched_line - 1;
3806 	} else {
3807 		if (view->searching == TOG_SEARCH_FORWARD)
3808 			lineno = 1;
3809 		else
3810 			lineno = s->blame.nlines;
3811 	}
3812 
3813 	while (1) {
3814 		char *line = NULL;
3815 		off_t offset;
3816 		size_t len;
3817 
3818 		if (lineno <= 0 || lineno > s->blame.nlines) {
3819 			if (s->matched_line == 0) {
3820 				view->search_next_done = 1;
3821 				free(line);
3822 				break;
3823 			}
3824 
3825 			if (view->searching == TOG_SEARCH_FORWARD)
3826 				lineno = 1;
3827 			else
3828 				lineno = s->blame.nlines;
3829 		}
3830 
3831 		offset = s->blame.line_offsets[lineno - 1];
3832 		if (fseeko(s->blame.f, offset, SEEK_SET) != 0) {
3833 			free(line);
3834 			return got_error_from_errno("fseeko");
3835 		}
3836 		free(line);
3837 		line = parse_next_line(s->blame.f, &len);
3838 		if (line && match_line(line, &view->regex)) {
3839 			view->search_next_done = 1;
3840 			s->matched_line = lineno;
3841 			free(line);
3842 			break;
3843 		}
3844 		free(line);
3845 		if (view->searching == TOG_SEARCH_FORWARD)
3846 			lineno++;
3847 		else
3848 			lineno--;
3849 	}
3850 
3851 	if (s->matched_line) {
3852 		s->first_displayed_line = s->matched_line;
3853 		s->selected_line = 1;
3854 	}
3855 
3856 	return NULL;
3857 }
3858 
3859 static const struct got_error *
3860 show_blame_view(struct tog_view *view)
3861 {
3862 	const struct got_error *err = NULL;
3863 	struct tog_blame_view_state *s = &view->state.blame;
3864 	int errcode;
3865 
3866 	if (s->blame.thread == NULL) {
3867 		errcode = pthread_create(&s->blame.thread, NULL, blame_thread,
3868 		    &s->blame.thread_args);
3869 		if (errcode)
3870 			return got_error_set_errno(errcode, "pthread_create");
3871 
3872 		halfdelay(1); /* fast refresh while annotating  */
3873 	}
3874 
3875 	if (s->blame_complete)
3876 		halfdelay(10); /* disable fast refresh */
3877 
3878 	err = draw_blame(view, s->blamed_commit->id, s->blame.f,
3879 	    s->path, s->blame.lines, s->blame.nlines, s->blame_complete,
3880 	    s->selected_line, &s->first_displayed_line,
3881 	    &s->last_displayed_line, &s->eof, view->nlines, &s->colors);
3882 
3883 	view_vborder(view);
3884 	return err;
3885 }
3886 
3887 static const struct got_error *
3888 input_blame_view(struct tog_view **new_view, struct tog_view **dead_view,
3889     struct tog_view **focus_view, struct tog_view *view, int ch)
3890 {
3891 	const struct got_error *err = NULL, *thread_err = NULL;
3892 	struct tog_view *diff_view;
3893 	struct tog_blame_view_state *s = &view->state.blame;
3894 	int begin_x = 0;
3895 
3896 	switch (ch) {
3897 	case 'q':
3898 		s->done = 1;
3899 		break;
3900 	case 'k':
3901 	case KEY_UP:
3902 		if (s->selected_line > 1)
3903 			s->selected_line--;
3904 		else if (s->selected_line == 1 &&
3905 		    s->first_displayed_line > 1)
3906 			s->first_displayed_line--;
3907 		break;
3908 	case KEY_PPAGE:
3909 		if (s->first_displayed_line == 1) {
3910 			s->selected_line = 1;
3911 			break;
3912 		}
3913 		if (s->first_displayed_line > view->nlines - 2)
3914 			s->first_displayed_line -=
3915 			    (view->nlines - 2);
3916 		else
3917 			s->first_displayed_line = 1;
3918 		break;
3919 	case 'j':
3920 	case KEY_DOWN:
3921 		if (s->selected_line < view->nlines - 2 &&
3922 		    s->first_displayed_line +
3923 		    s->selected_line <= s->blame.nlines)
3924 			s->selected_line++;
3925 		else if (s->last_displayed_line <
3926 		    s->blame.nlines)
3927 			s->first_displayed_line++;
3928 		break;
3929 	case 'b':
3930 	case 'p': {
3931 		struct got_object_id *id = NULL;
3932 		id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
3933 		    s->first_displayed_line, s->selected_line);
3934 		if (id == NULL)
3935 			break;
3936 		if (ch == 'p') {
3937 			struct got_commit_object *commit;
3938 			struct got_object_qid *pid;
3939 			struct got_object_id *blob_id = NULL;
3940 			int obj_type;
3941 			err = got_object_open_as_commit(&commit,
3942 			    s->repo, id);
3943 			if (err)
3944 				break;
3945 			pid = SIMPLEQ_FIRST(
3946 			    got_object_commit_get_parent_ids(commit));
3947 			if (pid == NULL) {
3948 				got_object_commit_close(commit);
3949 				break;
3950 			}
3951 			/* Check if path history ends here. */
3952 			err = got_object_id_by_path(&blob_id, s->repo,
3953 			    pid->id, s->path);
3954 			if (err) {
3955 				if (err->code == GOT_ERR_NO_TREE_ENTRY)
3956 					err = NULL;
3957 				got_object_commit_close(commit);
3958 				break;
3959 			}
3960 			err = got_object_get_type(&obj_type, s->repo,
3961 			    blob_id);
3962 			free(blob_id);
3963 			/* Can't blame non-blob type objects. */
3964 			if (obj_type != GOT_OBJ_TYPE_BLOB) {
3965 				got_object_commit_close(commit);
3966 				break;
3967 			}
3968 			err = got_object_qid_alloc(&s->blamed_commit,
3969 			    pid->id);
3970 			got_object_commit_close(commit);
3971 		} else {
3972 			if (got_object_id_cmp(id,
3973 			    s->blamed_commit->id) == 0)
3974 				break;
3975 			err = got_object_qid_alloc(&s->blamed_commit,
3976 			    id);
3977 		}
3978 		if (err)
3979 			break;
3980 		s->done = 1;
3981 		thread_err = stop_blame(&s->blame);
3982 		s->done = 0;
3983 		if (thread_err)
3984 			break;
3985 		SIMPLEQ_INSERT_HEAD(&s->blamed_commits,
3986 		    s->blamed_commit, entry);
3987 		err = run_blame(&s->blame, view, &s->blame_complete,
3988 		    &s->first_displayed_line, &s->last_displayed_line,
3989 		    &s->selected_line, &s->done, &s->eof,
3990 		    s->path, s->blamed_commit->id, s->repo);
3991 		if (err)
3992 			break;
3993 		break;
3994 	}
3995 	case 'B': {
3996 		struct got_object_qid *first;
3997 		first = SIMPLEQ_FIRST(&s->blamed_commits);
3998 		if (!got_object_id_cmp(first->id, s->commit_id))
3999 			break;
4000 		s->done = 1;
4001 		thread_err = stop_blame(&s->blame);
4002 		s->done = 0;
4003 		if (thread_err)
4004 			break;
4005 		SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
4006 		got_object_qid_free(s->blamed_commit);
4007 		s->blamed_commit =
4008 		    SIMPLEQ_FIRST(&s->blamed_commits);
4009 		err = run_blame(&s->blame, view, &s->blame_complete,
4010 		    &s->first_displayed_line, &s->last_displayed_line,
4011 		    &s->selected_line, &s->done, &s->eof, s->path,
4012 		    s->blamed_commit->id, s->repo);
4013 		if (err)
4014 			break;
4015 		break;
4016 	}
4017 	case KEY_ENTER:
4018 	case '\r': {
4019 		struct got_object_id *id = NULL;
4020 		struct got_object_qid *pid;
4021 		struct got_commit_object *commit = NULL;
4022 		id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
4023 		    s->first_displayed_line, s->selected_line);
4024 		if (id == NULL)
4025 			break;
4026 		err = got_object_open_as_commit(&commit, s->repo, id);
4027 		if (err)
4028 			break;
4029 		pid = SIMPLEQ_FIRST(
4030 		    got_object_commit_get_parent_ids(commit));
4031 		if (view_is_parent_view(view))
4032 		    begin_x = view_split_begin_x(view->begin_x);
4033 		diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
4034 		if (diff_view == NULL) {
4035 			got_object_commit_close(commit);
4036 			err = got_error_from_errno("view_open");
4037 			break;
4038 		}
4039 		err = open_diff_view(diff_view, pid ? pid->id : NULL,
4040 		    id, NULL, s->refs, s->repo);
4041 		got_object_commit_close(commit);
4042 		if (err) {
4043 			view_close(diff_view);
4044 			break;
4045 		}
4046 		if (view_is_parent_view(view)) {
4047 			err = view_close_child(view);
4048 			if (err)
4049 				break;
4050 			err = view_set_child(view, diff_view);
4051 			if (err) {
4052 				view_close(diff_view);
4053 				break;
4054 			}
4055 			*focus_view = diff_view;
4056 			view->child_focussed = 1;
4057 		} else
4058 			*new_view = diff_view;
4059 		if (err)
4060 			break;
4061 		break;
4062 	}
4063 	case KEY_NPAGE:
4064 	case ' ':
4065 		if (s->last_displayed_line >= s->blame.nlines &&
4066 		    s->selected_line >= MIN(s->blame.nlines,
4067 		    view->nlines - 2)) {
4068 			break;
4069 		}
4070 		if (s->last_displayed_line >= s->blame.nlines &&
4071 		    s->selected_line < view->nlines - 2) {
4072 			s->selected_line = MIN(s->blame.nlines,
4073 			    view->nlines - 2);
4074 			break;
4075 		}
4076 		if (s->last_displayed_line + view->nlines - 2
4077 		    <= s->blame.nlines)
4078 			s->first_displayed_line +=
4079 			    view->nlines - 2;
4080 		else
4081 			s->first_displayed_line =
4082 			    s->blame.nlines -
4083 			    (view->nlines - 3);
4084 		break;
4085 	case KEY_RESIZE:
4086 		if (s->selected_line > view->nlines - 2) {
4087 			s->selected_line = MIN(s->blame.nlines,
4088 			    view->nlines - 2);
4089 		}
4090 		break;
4091 	default:
4092 		break;
4093 	}
4094 	return thread_err ? thread_err : err;
4095 }
4096 
4097 static const struct got_error *
4098 cmd_blame(int argc, char *argv[])
4099 {
4100 	const struct got_error *error;
4101 	struct got_repository *repo = NULL;
4102 	struct got_reflist_head refs;
4103 	struct got_worktree *worktree = NULL;
4104 	char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
4105 	struct got_object_id *commit_id = NULL;
4106 	char *commit_id_str = NULL;
4107 	int ch;
4108 	struct tog_view *view;
4109 
4110 	SIMPLEQ_INIT(&refs);
4111 
4112 #ifndef PROFILE
4113 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
4114 	    NULL) == -1)
4115 		err(1, "pledge");
4116 #endif
4117 
4118 	while ((ch = getopt(argc, argv, "c:r:")) != -1) {
4119 		switch (ch) {
4120 		case 'c':
4121 			commit_id_str = optarg;
4122 			break;
4123 		case 'r':
4124 			repo_path = realpath(optarg, NULL);
4125 			if (repo_path == NULL)
4126 				return got_error_from_errno2("realpath",
4127 				    optarg);
4128 			break;
4129 		default:
4130 			usage_blame();
4131 			/* NOTREACHED */
4132 		}
4133 	}
4134 
4135 	argc -= optind;
4136 	argv += optind;
4137 
4138 	if (argc == 1)
4139 		path = argv[0];
4140 	else
4141 		usage_blame();
4142 
4143 	cwd = getcwd(NULL, 0);
4144 	if (cwd == NULL) {
4145 		error = got_error_from_errno("getcwd");
4146 		goto done;
4147 	}
4148 	if (repo_path == NULL) {
4149 		error = got_worktree_open(&worktree, cwd);
4150 		if (error && error->code != GOT_ERR_NOT_WORKTREE)
4151 			goto done;
4152 		else
4153 			error = NULL;
4154 		if (worktree) {
4155 			repo_path =
4156 			    strdup(got_worktree_get_repo_path(worktree));
4157 			if (repo_path == NULL)
4158 				error = got_error_from_errno("strdup");
4159 			if (error)
4160 				goto done;
4161 		} else {
4162 			repo_path = strdup(cwd);
4163 			if (repo_path == NULL) {
4164 				error = got_error_from_errno("strdup");
4165 				goto done;
4166 			}
4167 		}
4168 	}
4169 
4170 	init_curses();
4171 
4172 	error = got_repo_open(&repo, repo_path, NULL);
4173 	if (error != NULL)
4174 		goto done;
4175 
4176 	error = apply_unveil(got_repo_get_path(repo), NULL);
4177 	if (error)
4178 		goto done;
4179 
4180 	if (worktree) {
4181 		const char *prefix = got_worktree_get_path_prefix(worktree);
4182 		char *p, *worktree_subdir = cwd +
4183 		    strlen(got_worktree_get_root_path(worktree));
4184 		if (asprintf(&p, "%s%s%s%s%s",
4185 		    prefix, (strcmp(prefix, "/") != 0) ? "/" : "",
4186 		    worktree_subdir, worktree_subdir[0] ? "/" : "",
4187 		    path) == -1) {
4188 			error = got_error_from_errno("asprintf");
4189 			goto done;
4190 		}
4191 		error = got_repo_map_path(&in_repo_path, repo, p, 0);
4192 		free(p);
4193 	} else {
4194 		error = got_repo_map_path(&in_repo_path, repo, path, 1);
4195 	}
4196 	if (error)
4197 		goto done;
4198 
4199 	if (commit_id_str == NULL) {
4200 		struct got_reference *head_ref;
4201 		error = got_ref_open(&head_ref, repo, GOT_REF_HEAD, 0);
4202 		if (error != NULL)
4203 			goto done;
4204 		error = got_ref_resolve(&commit_id, repo, head_ref);
4205 		got_ref_close(head_ref);
4206 	} else {
4207 		error = get_head_commit_id(&commit_id, commit_id_str, repo);
4208 		if (error) {
4209 			if (error->code != GOT_ERR_NOT_REF)
4210 				goto done;
4211 			error = got_repo_match_object_id_prefix(&commit_id,
4212 			    commit_id_str, GOT_OBJ_TYPE_COMMIT, repo);
4213 		}
4214 	}
4215 	if (error != NULL)
4216 		goto done;
4217 
4218 	error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
4219 	if (error)
4220 		goto done;
4221 
4222 	view = view_open(0, 0, 0, 0, TOG_VIEW_BLAME);
4223 	if (view == NULL) {
4224 		error = got_error_from_errno("view_open");
4225 		goto done;
4226 	}
4227 	error = open_blame_view(view, in_repo_path, commit_id, &refs, repo);
4228 	if (error)
4229 		goto done;
4230 	if (worktree) {
4231 		/* Release work tree lock. */
4232 		got_worktree_close(worktree);
4233 		worktree = NULL;
4234 	}
4235 	error = view_loop(view);
4236 done:
4237 	free(repo_path);
4238 	free(cwd);
4239 	free(commit_id);
4240 	if (worktree)
4241 		got_worktree_close(worktree);
4242 	if (repo)
4243 		got_repo_close(repo);
4244 	got_ref_list_free(&refs);
4245 	return error;
4246 }
4247 
4248 static const struct got_error *
4249 draw_tree_entries(struct tog_view *view,
4250     struct got_tree_entry **first_displayed_entry,
4251     struct got_tree_entry **last_displayed_entry,
4252     struct got_tree_entry **selected_entry, int *ndisplayed,
4253     const char *label, int show_ids, const char *parent_path,
4254     struct got_tree_object *tree, int selected, int limit,
4255     int isroot, struct tog_colors *colors)
4256 {
4257 	const struct got_error *err = NULL;
4258 	struct got_tree_entry *te;
4259 	wchar_t *wline;
4260 	struct tog_color *tc;
4261 	int width, n, i, nentries;
4262 
4263 	*ndisplayed = 0;
4264 
4265 	werase(view->window);
4266 
4267 	if (limit == 0)
4268 		return NULL;
4269 
4270 	err = format_line(&wline, &width, label, view->ncols, 0);
4271 	if (err)
4272 		return err;
4273 	if (view_needs_focus_indication(view))
4274 		wstandout(view->window);
4275 	tc = get_color(colors, TOG_COLOR_COMMIT);
4276 	if (tc)
4277 		wattr_on(view->window,
4278 		    COLOR_PAIR(tc->colorpair), NULL);
4279 	waddwstr(view->window, wline);
4280 	if (tc)
4281 		wattr_off(view->window,
4282 		    COLOR_PAIR(tc->colorpair), NULL);
4283 	if (view_needs_focus_indication(view))
4284 		wstandend(view->window);
4285 	free(wline);
4286 	wline = NULL;
4287 	if (width < view->ncols - 1)
4288 		waddch(view->window, '\n');
4289 	if (--limit <= 0)
4290 		return NULL;
4291 	err = format_line(&wline, &width, parent_path, view->ncols, 0);
4292 	if (err)
4293 		return err;
4294 	waddwstr(view->window, wline);
4295 	free(wline);
4296 	wline = NULL;
4297 	if (width < view->ncols - 1)
4298 		waddch(view->window, '\n');
4299 	if (--limit <= 0)
4300 		return NULL;
4301 	waddch(view->window, '\n');
4302 	if (--limit <= 0)
4303 		return NULL;
4304 
4305 	if (*first_displayed_entry == NULL) {
4306 		te = got_object_tree_get_first_entry(tree);
4307 		if (selected == 0) {
4308 			if (view->focussed)
4309 				wstandout(view->window);
4310 			*selected_entry = NULL;
4311 		}
4312 		waddstr(view->window, "  ..\n");	/* parent directory */
4313 		if (selected == 0 && view->focussed)
4314 			wstandend(view->window);
4315 		(*ndisplayed)++;
4316 		if (--limit <= 0)
4317 			return NULL;
4318 		n = 1;
4319 	} else {
4320 		n = 0;
4321 		te = *first_displayed_entry;
4322 	}
4323 
4324 	nentries = got_object_tree_get_nentries(tree);
4325 	for (i = got_tree_entry_get_index(te); i < nentries; i++) {
4326 		char *line = NULL, *id_str = NULL;
4327 		const char *modestr = "";
4328 		mode_t mode;
4329 
4330 		te = got_object_tree_get_entry(tree, i);
4331 		mode = got_tree_entry_get_mode(te);
4332 
4333 		if (show_ids) {
4334 			err = got_object_id_str(&id_str,
4335 			    got_tree_entry_get_id(te));
4336 			if (err)
4337 				return got_error_from_errno(
4338 				    "got_object_id_str");
4339 		}
4340 		if (got_object_tree_entry_is_submodule(te))
4341 			modestr = "$";
4342 		else if (S_ISLNK(mode))
4343 			modestr = "@";
4344 		else if (S_ISDIR(mode))
4345 			modestr = "/";
4346 		else if (mode & S_IXUSR)
4347 			modestr = "*";
4348 		if (asprintf(&line, "%s  %s%s", id_str ? id_str : "",
4349 		    got_tree_entry_get_name(te), modestr) == -1) {
4350 			free(id_str);
4351 			return got_error_from_errno("asprintf");
4352 		}
4353 		free(id_str);
4354 		err = format_line(&wline, &width, line, view->ncols, 0);
4355 		if (err) {
4356 			free(line);
4357 			break;
4358 		}
4359 		if (n == selected) {
4360 			if (view->focussed)
4361 				wstandout(view->window);
4362 			*selected_entry = te;
4363 		}
4364 		tc = match_color(colors, line);
4365 		if (tc)
4366 			wattr_on(view->window,
4367 			    COLOR_PAIR(tc->colorpair), NULL);
4368 		waddwstr(view->window, wline);
4369 		if (tc)
4370 			wattr_off(view->window,
4371 			    COLOR_PAIR(tc->colorpair), NULL);
4372 		if (width < view->ncols - 1)
4373 			waddch(view->window, '\n');
4374 		if (n == selected && view->focussed)
4375 			wstandend(view->window);
4376 		free(line);
4377 		free(wline);
4378 		wline = NULL;
4379 		n++;
4380 		(*ndisplayed)++;
4381 		*last_displayed_entry = te;
4382 		if (--limit <= 0)
4383 			break;
4384 	}
4385 
4386 	return err;
4387 }
4388 
4389 static void
4390 tree_scroll_up(struct tog_view *view,
4391     struct got_tree_entry **first_displayed_entry, int maxscroll,
4392     struct got_tree_object *tree, int isroot)
4393 {
4394 	struct got_tree_entry *te;
4395 	int i;
4396 
4397 	if (*first_displayed_entry == NULL)
4398 		return;
4399 
4400 	te = got_object_tree_get_entry(tree, 0);
4401 	if (*first_displayed_entry == te) {
4402 		if (!isroot)
4403 			*first_displayed_entry = NULL;
4404 		return;
4405 	}
4406 
4407 	i = 0;
4408 	while (*first_displayed_entry && i < maxscroll) {
4409 		*first_displayed_entry = got_tree_entry_get_prev(tree,
4410 		    *first_displayed_entry);
4411 		i++;
4412 	}
4413 	if (!isroot && te == got_object_tree_get_first_entry(tree) && i < maxscroll)
4414 		*first_displayed_entry = NULL;
4415 }
4416 
4417 static int
4418 tree_scroll_down(struct got_tree_entry **first_displayed_entry, int maxscroll,
4419 	struct got_tree_entry *last_displayed_entry,
4420 	struct got_tree_object *tree)
4421 {
4422 	struct got_tree_entry *next, *last;
4423 	int n = 0;
4424 
4425 	if (*first_displayed_entry)
4426 		next = got_tree_entry_get_next(tree, *first_displayed_entry);
4427 	else
4428 		next = got_object_tree_get_first_entry(tree);
4429 
4430 	last = last_displayed_entry;
4431 	while (next && last && n++ < maxscroll) {
4432 		last = got_tree_entry_get_next(tree, last);
4433 		if (last) {
4434 			*first_displayed_entry = next;
4435 			next = got_tree_entry_get_next(tree, next);
4436 		}
4437 	}
4438 	return n;
4439 }
4440 
4441 static const struct got_error *
4442 tree_entry_path(char **path, struct tog_parent_trees *parents,
4443     struct got_tree_entry *te)
4444 {
4445 	const struct got_error *err = NULL;
4446 	struct tog_parent_tree *pt;
4447 	size_t len = 2; /* for leading slash and NUL */
4448 
4449 	TAILQ_FOREACH(pt, parents, entry)
4450 		len += strlen(got_tree_entry_get_name(pt->selected_entry))
4451 		    + 1 /* slash */;
4452 	if (te)
4453 		len += strlen(got_tree_entry_get_name(te));
4454 
4455 	*path = calloc(1, len);
4456 	if (path == NULL)
4457 		return got_error_from_errno("calloc");
4458 
4459 	(*path)[0] = '/';
4460 	pt = TAILQ_LAST(parents, tog_parent_trees);
4461 	while (pt) {
4462 		const char *name = got_tree_entry_get_name(pt->selected_entry);
4463 		if (strlcat(*path, name, len) >= len) {
4464 			err = got_error(GOT_ERR_NO_SPACE);
4465 			goto done;
4466 		}
4467 		if (strlcat(*path, "/", len) >= len) {
4468 			err = got_error(GOT_ERR_NO_SPACE);
4469 			goto done;
4470 		}
4471 		pt = TAILQ_PREV(pt, tog_parent_trees, entry);
4472 	}
4473 	if (te) {
4474 		if (strlcat(*path, got_tree_entry_get_name(te), len) >= len) {
4475 			err = got_error(GOT_ERR_NO_SPACE);
4476 			goto done;
4477 		}
4478 	}
4479 done:
4480 	if (err) {
4481 		free(*path);
4482 		*path = NULL;
4483 	}
4484 	return err;
4485 }
4486 
4487 static const struct got_error *
4488 blame_tree_entry(struct tog_view **new_view, int begin_x,
4489     struct got_tree_entry *te, struct tog_parent_trees *parents,
4490     struct got_object_id *commit_id, struct got_reflist_head *refs,
4491     struct got_repository *repo)
4492 {
4493 	const struct got_error *err = NULL;
4494 	char *path;
4495 	struct tog_view *blame_view;
4496 
4497 	*new_view = NULL;
4498 
4499 	err = tree_entry_path(&path, parents, te);
4500 	if (err)
4501 		return err;
4502 
4503 	blame_view = view_open(0, 0, 0, begin_x, TOG_VIEW_BLAME);
4504 	if (blame_view == NULL) {
4505 		err = got_error_from_errno("view_open");
4506 		goto done;
4507 	}
4508 
4509 	err = open_blame_view(blame_view, path, commit_id, refs, repo);
4510 	if (err) {
4511 		if (err->code == GOT_ERR_CANCELLED)
4512 			err = NULL;
4513 		view_close(blame_view);
4514 	} else
4515 		*new_view = blame_view;
4516 done:
4517 	free(path);
4518 	return err;
4519 }
4520 
4521 static const struct got_error *
4522 log_tree_entry(struct tog_view **new_view, int begin_x,
4523     struct got_tree_entry *te, struct tog_parent_trees *parents,
4524     struct got_object_id *commit_id, struct got_reflist_head *refs,
4525     struct got_repository *repo)
4526 {
4527 	struct tog_view *log_view;
4528 	const struct got_error *err = NULL;
4529 	char *path;
4530 
4531 	*new_view = NULL;
4532 
4533 	log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
4534 	if (log_view == NULL)
4535 		return got_error_from_errno("view_open");
4536 
4537 	err = tree_entry_path(&path, parents, te);
4538 	if (err)
4539 		return err;
4540 
4541 	err = open_log_view(log_view, commit_id, refs, repo, NULL, path, 0);
4542 	if (err)
4543 		view_close(log_view);
4544 	else
4545 		*new_view = log_view;
4546 	free(path);
4547 	return err;
4548 }
4549 
4550 static const struct got_error *
4551 open_tree_view(struct tog_view *view, struct got_tree_object *root,
4552     struct got_object_id *commit_id, struct got_reflist_head *refs,
4553     struct got_repository *repo)
4554 {
4555 	const struct got_error *err = NULL;
4556 	char *commit_id_str = NULL;
4557 	struct tog_tree_view_state *s = &view->state.tree;
4558 
4559 	TAILQ_INIT(&s->parents);
4560 
4561 	err = got_object_id_str(&commit_id_str, commit_id);
4562 	if (err != NULL)
4563 		goto done;
4564 
4565 	if (asprintf(&s->tree_label, "commit %s", commit_id_str) == -1) {
4566 		err = got_error_from_errno("asprintf");
4567 		goto done;
4568 	}
4569 
4570 	s->root = s->tree = root;
4571 	s->first_displayed_entry = got_object_tree_get_entry(s->tree, 0);
4572 	s->selected_entry = got_object_tree_get_entry(s->tree, 0);
4573 	s->commit_id = got_object_id_dup(commit_id);
4574 	if (s->commit_id == NULL) {
4575 		err = got_error_from_errno("got_object_id_dup");
4576 		goto done;
4577 	}
4578 	s->refs = refs;
4579 	s->repo = repo;
4580 
4581 	SIMPLEQ_INIT(&s->colors);
4582 
4583 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
4584 		err = add_color(&s->colors, "\\$$",
4585 		    TOG_COLOR_TREE_SUBMODULE,
4586 		    get_color_value("TOG_COLOR_TREE_SUBMODULE"));
4587 		if (err)
4588 			goto done;
4589 		err = add_color(&s->colors, "@$", TOG_COLOR_TREE_SYMLINK,
4590 		    get_color_value("TOG_COLOR_TREE_SYMLINK"));
4591 		if (err) {
4592 			free_colors(&s->colors);
4593 			goto done;
4594 		}
4595 		err = add_color(&s->colors, "/$",
4596 		    TOG_COLOR_TREE_DIRECTORY,
4597 		    get_color_value("TOG_COLOR_TREE_DIRECTORY"));
4598 		if (err) {
4599 			free_colors(&s->colors);
4600 			goto done;
4601 		}
4602 
4603 		err = add_color(&s->colors, "\\*$",
4604 		    TOG_COLOR_TREE_EXECUTABLE,
4605 		    get_color_value("TOG_COLOR_TREE_EXECUTABLE"));
4606 		if (err) {
4607 			free_colors(&s->colors);
4608 			goto done;
4609 		}
4610 
4611 		err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
4612 		    get_color_value("TOG_COLOR_COMMIT"));
4613 		if (err) {
4614 			free_colors(&s->colors);
4615 			goto done;
4616 		}
4617 	}
4618 
4619 	view->show = show_tree_view;
4620 	view->input = input_tree_view;
4621 	view->close = close_tree_view;
4622 	view->search_start = search_start_tree_view;
4623 	view->search_next = search_next_tree_view;
4624 done:
4625 	free(commit_id_str);
4626 	if (err) {
4627 		free(s->tree_label);
4628 		s->tree_label = NULL;
4629 	}
4630 	return err;
4631 }
4632 
4633 static const struct got_error *
4634 close_tree_view(struct tog_view *view)
4635 {
4636 	struct tog_tree_view_state *s = &view->state.tree;
4637 
4638 	free_colors(&s->colors);
4639 	free(s->tree_label);
4640 	s->tree_label = NULL;
4641 	free(s->commit_id);
4642 	s->commit_id = NULL;
4643 	while (!TAILQ_EMPTY(&s->parents)) {
4644 		struct tog_parent_tree *parent;
4645 		parent = TAILQ_FIRST(&s->parents);
4646 		TAILQ_REMOVE(&s->parents, parent, entry);
4647 		free(parent);
4648 
4649 	}
4650 	if (s->tree != s->root)
4651 		got_object_tree_close(s->tree);
4652 	got_object_tree_close(s->root);
4653 
4654 	return NULL;
4655 }
4656 
4657 static const struct got_error *
4658 search_start_tree_view(struct tog_view *view)
4659 {
4660 	struct tog_tree_view_state *s = &view->state.tree;
4661 
4662 	s->matched_entry = NULL;
4663 	return NULL;
4664 }
4665 
4666 static int
4667 match_tree_entry(struct got_tree_entry *te, regex_t *regex)
4668 {
4669 	regmatch_t regmatch;
4670 
4671 	return regexec(regex, got_tree_entry_get_name(te), 1, &regmatch,
4672 	    0) == 0;
4673 }
4674 
4675 static const struct got_error *
4676 search_next_tree_view(struct tog_view *view)
4677 {
4678 	struct tog_tree_view_state *s = &view->state.tree;
4679 	struct got_tree_entry *te = NULL;
4680 
4681 	if (!view->searching) {
4682 		view->search_next_done = 1;
4683 		return NULL;
4684 	}
4685 
4686 	if (s->matched_entry) {
4687 		if (view->searching == TOG_SEARCH_FORWARD) {
4688 			if (s->selected_entry)
4689 				te = got_tree_entry_get_next(s->tree,
4690 				    s->selected_entry);
4691 			else
4692 				te = got_object_tree_get_first_entry(s->tree);
4693 		} else {
4694 			if (s->selected_entry == NULL)
4695 				te = got_object_tree_get_last_entry(s->tree);
4696 			else
4697 				te = got_tree_entry_get_prev(s->tree,
4698 				    s->selected_entry);
4699 		}
4700 	} else {
4701 		if (view->searching == TOG_SEARCH_FORWARD)
4702 			te = got_object_tree_get_first_entry(s->tree);
4703 		else
4704 			te = got_object_tree_get_last_entry(s->tree);
4705 	}
4706 
4707 	while (1) {
4708 		if (te == NULL) {
4709 			if (s->matched_entry == NULL) {
4710 				view->search_next_done = 1;
4711 				return NULL;
4712 			}
4713 			if (view->searching == TOG_SEARCH_FORWARD)
4714 				te = got_object_tree_get_first_entry(s->tree);
4715 			else
4716 				te = got_object_tree_get_last_entry(s->tree);
4717 		}
4718 
4719 		if (match_tree_entry(te, &view->regex)) {
4720 			view->search_next_done = 1;
4721 			s->matched_entry = te;
4722 			break;
4723 		}
4724 
4725 		if (view->searching == TOG_SEARCH_FORWARD)
4726 			te = got_tree_entry_get_next(s->tree, te);
4727 		else
4728 			te = got_tree_entry_get_prev(s->tree, te);
4729 	}
4730 
4731 	if (s->matched_entry) {
4732 		s->first_displayed_entry = s->matched_entry;
4733 		s->selected = 0;
4734 	}
4735 
4736 	return NULL;
4737 }
4738 
4739 static const struct got_error *
4740 show_tree_view(struct tog_view *view)
4741 {
4742 	const struct got_error *err = NULL;
4743 	struct tog_tree_view_state *s = &view->state.tree;
4744 	char *parent_path;
4745 
4746 	err = tree_entry_path(&parent_path, &s->parents, NULL);
4747 	if (err)
4748 		return err;
4749 
4750 	err = draw_tree_entries(view, &s->first_displayed_entry,
4751 	    &s->last_displayed_entry, &s->selected_entry,
4752 	    &s->ndisplayed, s->tree_label, s->show_ids, parent_path,
4753 	    s->tree, s->selected, view->nlines, s->tree == s->root,
4754 	    &s->colors);
4755 	free(parent_path);
4756 
4757 	view_vborder(view);
4758 	return err;
4759 }
4760 
4761 static const struct got_error *
4762 input_tree_view(struct tog_view **new_view, struct tog_view **dead_view,
4763     struct tog_view **focus_view, struct tog_view *view, int ch)
4764 {
4765 	const struct got_error *err = NULL;
4766 	struct tog_tree_view_state *s = &view->state.tree;
4767 	struct tog_view *log_view;
4768 	int begin_x = 0, nscrolled;
4769 
4770 	switch (ch) {
4771 	case 'i':
4772 		s->show_ids = !s->show_ids;
4773 		break;
4774 	case 'l':
4775 		if (!s->selected_entry)
4776 			break;
4777 		if (view_is_parent_view(view))
4778 			begin_x = view_split_begin_x(view->begin_x);
4779 		err = log_tree_entry(&log_view, begin_x,
4780 		    s->selected_entry, &s->parents,
4781 		    s->commit_id, s->refs, s->repo);
4782 		if (view_is_parent_view(view)) {
4783 			err = view_close_child(view);
4784 			if (err)
4785 				return err;
4786 			err = view_set_child(view, log_view);
4787 			if (err) {
4788 				view_close(log_view);
4789 				break;
4790 			}
4791 			*focus_view = log_view;
4792 			view->child_focussed = 1;
4793 		} else
4794 			*new_view = log_view;
4795 		break;
4796 	case 'k':
4797 	case KEY_UP:
4798 		if (s->selected > 0) {
4799 			s->selected--;
4800 			if (s->selected == 0)
4801 				break;
4802 		}
4803 		if (s->selected > 0)
4804 			break;
4805 		tree_scroll_up(view, &s->first_displayed_entry, 1,
4806 		    s->tree, s->tree == s->root);
4807 		break;
4808 	case KEY_PPAGE:
4809 		tree_scroll_up(view, &s->first_displayed_entry,
4810 		    MAX(0, view->nlines - 4 - s->selected), s->tree,
4811 		    s->tree == s->root);
4812 		s->selected = 0;
4813 		if (got_object_tree_get_first_entry(s->tree) ==
4814 		    s->first_displayed_entry && s->tree != s->root)
4815 			s->first_displayed_entry = NULL;
4816 		break;
4817 	case 'j':
4818 	case KEY_DOWN:
4819 		if (s->selected < s->ndisplayed - 1) {
4820 			s->selected++;
4821 			break;
4822 		}
4823 		if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
4824 		    == NULL)
4825 			/* can't scroll any further */
4826 			break;
4827 		tree_scroll_down(&s->first_displayed_entry, 1,
4828 		    s->last_displayed_entry, s->tree);
4829 		break;
4830 	case KEY_NPAGE:
4831 		if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
4832 		    == NULL) {
4833 			/* can't scroll any further; move cursor down */
4834 			if (s->selected < s->ndisplayed - 1)
4835 				s->selected = s->ndisplayed - 1;
4836 			break;
4837 		}
4838 		nscrolled = tree_scroll_down(&s->first_displayed_entry,
4839 		    view->nlines, s->last_displayed_entry, s->tree);
4840 		if (nscrolled < view->nlines) {
4841 			int ndisplayed = 0;
4842 			struct got_tree_entry *te;
4843 			te = s->first_displayed_entry;
4844 			do {
4845 				ndisplayed++;
4846 				te = got_tree_entry_get_next(s->tree, te);
4847 			} while (te);
4848 			s->selected = ndisplayed - 1;
4849 		}
4850 		break;
4851 	case KEY_ENTER:
4852 	case '\r':
4853 	case KEY_BACKSPACE:
4854 		if (s->selected_entry == NULL || ch == KEY_BACKSPACE) {
4855 			struct tog_parent_tree *parent;
4856 			/* user selected '..' */
4857 			if (s->tree == s->root)
4858 				break;
4859 			parent = TAILQ_FIRST(&s->parents);
4860 			TAILQ_REMOVE(&s->parents, parent,
4861 			    entry);
4862 			got_object_tree_close(s->tree);
4863 			s->tree = parent->tree;
4864 			s->first_displayed_entry =
4865 			    parent->first_displayed_entry;
4866 			s->selected_entry =
4867 			    parent->selected_entry;
4868 			s->selected = parent->selected;
4869 			free(parent);
4870 		} else if (S_ISDIR(got_tree_entry_get_mode(
4871 		    s->selected_entry))) {
4872 			struct got_tree_object *subtree;
4873 			err = got_object_open_as_tree(&subtree, s->repo,
4874 			    got_tree_entry_get_id(s->selected_entry));
4875 			if (err)
4876 				break;
4877 			err = tree_view_visit_subtree(subtree, s);
4878 			if (err) {
4879 				got_object_tree_close(subtree);
4880 				break;
4881 			}
4882 		} else if (S_ISREG(got_tree_entry_get_mode(
4883 		    s->selected_entry))) {
4884 			struct tog_view *blame_view;
4885 			int begin_x = view_is_parent_view(view) ?
4886 			    view_split_begin_x(view->begin_x) : 0;
4887 
4888 			err = blame_tree_entry(&blame_view, begin_x,
4889 			    s->selected_entry, &s->parents,
4890 			    s->commit_id, s->refs, s->repo);
4891 			if (err)
4892 				break;
4893 			if (view_is_parent_view(view)) {
4894 				err = view_close_child(view);
4895 				if (err)
4896 					return err;
4897 				err = view_set_child(view, blame_view);
4898 				if (err) {
4899 					view_close(blame_view);
4900 					break;
4901 				}
4902 				*focus_view = blame_view;
4903 				view->child_focussed = 1;
4904 			} else
4905 				*new_view = blame_view;
4906 		}
4907 		break;
4908 	case KEY_RESIZE:
4909 		if (s->selected > view->nlines)
4910 			s->selected = s->ndisplayed - 1;
4911 		break;
4912 	default:
4913 		break;
4914 	}
4915 
4916 	return err;
4917 }
4918 
4919 __dead static void
4920 usage_tree(void)
4921 {
4922 	endwin();
4923 	fprintf(stderr, "usage: %s tree [-c commit] [repository-path]\n",
4924 	    getprogname());
4925 	exit(1);
4926 }
4927 
4928 static const struct got_error *
4929 cmd_tree(int argc, char *argv[])
4930 {
4931 	const struct got_error *error;
4932 	struct got_repository *repo = NULL;
4933 	struct got_reflist_head refs;
4934 	char *repo_path = NULL;
4935 	struct got_object_id *commit_id = NULL;
4936 	char *commit_id_arg = NULL;
4937 	struct got_commit_object *commit = NULL;
4938 	struct got_tree_object *tree = NULL;
4939 	int ch;
4940 	struct tog_view *view;
4941 
4942 	SIMPLEQ_INIT(&refs);
4943 
4944 #ifndef PROFILE
4945 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
4946 	    NULL) == -1)
4947 		err(1, "pledge");
4948 #endif
4949 
4950 	while ((ch = getopt(argc, argv, "c:")) != -1) {
4951 		switch (ch) {
4952 		case 'c':
4953 			commit_id_arg = optarg;
4954 			break;
4955 		default:
4956 			usage_tree();
4957 			/* NOTREACHED */
4958 		}
4959 	}
4960 
4961 	argc -= optind;
4962 	argv += optind;
4963 
4964 	if (argc == 0) {
4965 		struct got_worktree *worktree;
4966 		char *cwd = getcwd(NULL, 0);
4967 		if (cwd == NULL)
4968 			return got_error_from_errno("getcwd");
4969 		error = got_worktree_open(&worktree, cwd);
4970 		if (error && error->code != GOT_ERR_NOT_WORKTREE)
4971 			goto done;
4972 		if (worktree) {
4973 			free(cwd);
4974 			repo_path =
4975 			    strdup(got_worktree_get_repo_path(worktree));
4976 			got_worktree_close(worktree);
4977 		} else
4978 			repo_path = cwd;
4979 		if (repo_path == NULL) {
4980 			error = got_error_from_errno("strdup");
4981 			goto done;
4982 		}
4983 	} else if (argc == 1) {
4984 		repo_path = realpath(argv[0], NULL);
4985 		if (repo_path == NULL)
4986 			return got_error_from_errno2("realpath", argv[0]);
4987 	} else
4988 		usage_log();
4989 
4990 	init_curses();
4991 
4992 	error = got_repo_open(&repo, repo_path, NULL);
4993 	if (error != NULL)
4994 		goto done;
4995 
4996 	error = apply_unveil(got_repo_get_path(repo), NULL);
4997 	if (error)
4998 		goto done;
4999 
5000 	if (commit_id_arg == NULL)
5001 		error = get_head_commit_id(&commit_id, GOT_REF_HEAD, repo);
5002 	else {
5003 		error = get_head_commit_id(&commit_id, commit_id_arg, repo);
5004 		if (error) {
5005 			if (error->code != GOT_ERR_NOT_REF)
5006 				goto done;
5007 			error = got_repo_match_object_id_prefix(&commit_id,
5008 			    commit_id_arg, GOT_OBJ_TYPE_COMMIT, repo);
5009 		}
5010 	}
5011 	if (error != NULL)
5012 		goto done;
5013 
5014 	error = got_object_open_as_commit(&commit, repo, commit_id);
5015 	if (error != NULL)
5016 		goto done;
5017 
5018 	error = got_object_open_as_tree(&tree, repo,
5019 	    got_object_commit_get_tree_id(commit));
5020 	if (error != NULL)
5021 		goto done;
5022 
5023 	error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
5024 	if (error)
5025 		goto done;
5026 
5027 	view = view_open(0, 0, 0, 0, TOG_VIEW_TREE);
5028 	if (view == NULL) {
5029 		error = got_error_from_errno("view_open");
5030 		goto done;
5031 	}
5032 	error = open_tree_view(view, tree, commit_id, &refs, repo);
5033 	if (error)
5034 		goto done;
5035 	error = view_loop(view);
5036 done:
5037 	free(repo_path);
5038 	free(commit_id);
5039 	if (commit)
5040 		got_object_commit_close(commit);
5041 	if (tree)
5042 		got_object_tree_close(tree);
5043 	if (repo)
5044 		got_repo_close(repo);
5045 	got_ref_list_free(&refs);
5046 	return error;
5047 }
5048 
5049 static void
5050 list_commands(void)
5051 {
5052 	int i;
5053 
5054 	fprintf(stderr, "commands:");
5055 	for (i = 0; i < nitems(tog_commands); i++) {
5056 		struct tog_cmd *cmd = &tog_commands[i];
5057 		fprintf(stderr, " %s", cmd->name);
5058 	}
5059 	fputc('\n', stderr);
5060 }
5061 
5062 __dead static void
5063 usage(int hflag)
5064 {
5065 	fprintf(stderr, "usage: %s [-h] [-V] [command] [arg ...]\n",
5066 	    getprogname());
5067 	if (hflag)
5068 		list_commands();
5069 	exit(1);
5070 }
5071 
5072 static char **
5073 make_argv(const char *arg0, const char *arg1)
5074 {
5075 	char **argv;
5076 	int argc = (arg1 == NULL ? 1 : 2);
5077 
5078 	argv = calloc(argc, sizeof(char *));
5079 	if (argv == NULL)
5080 		err(1, "calloc");
5081 	argv[0] = strdup(arg0);
5082 	if (argv[0] == NULL)
5083 		err(1, "strdup");
5084 	if (arg1) {
5085 		argv[1] = strdup(arg1);
5086 		if (argv[1] == NULL)
5087 			err(1, "strdup");
5088 	}
5089 
5090 	return argv;
5091 }
5092 
5093 int
5094 main(int argc, char *argv[])
5095 {
5096 	const struct got_error *error = NULL;
5097 	struct tog_cmd *cmd = NULL;
5098 	int ch, hflag = 0, Vflag = 0;
5099 	char **cmd_argv = NULL;
5100 
5101 	setlocale(LC_CTYPE, "");
5102 
5103 	while ((ch = getopt(argc, argv, "hV")) != -1) {
5104 		switch (ch) {
5105 		case 'h':
5106 			hflag = 1;
5107 			break;
5108 		case 'V':
5109 			Vflag = 1;
5110 			break;
5111 		default:
5112 			usage(hflag);
5113 			/* NOTREACHED */
5114 		}
5115 	}
5116 
5117 	argc -= optind;
5118 	argv += optind;
5119 	optind = 0;
5120 	optreset = 1;
5121 
5122 	if (Vflag) {
5123 		got_version_print_str();
5124 		return 1;
5125 	}
5126 
5127 	if (argc == 0) {
5128 		if (hflag)
5129 			usage(hflag);
5130 		/* Build an argument vector which runs a default command. */
5131 		cmd = &tog_commands[0];
5132 		cmd_argv = make_argv(cmd->name, NULL);
5133 		argc = 1;
5134 	} else {
5135 		int i;
5136 
5137 		/* Did the user specific a command? */
5138 		for (i = 0; i < nitems(tog_commands); i++) {
5139 			if (strncmp(tog_commands[i].name, argv[0],
5140 			    strlen(argv[0])) == 0) {
5141 				cmd = &tog_commands[i];
5142 				break;
5143 			}
5144 		}
5145 
5146 		if (cmd == NULL) {
5147 			fprintf(stderr, "%s: unknown command '%s'\n",
5148 			    getprogname(), argv[0]);
5149 			list_commands();
5150 			return 1;
5151 		}
5152 	}
5153 
5154 	if (hflag)
5155 		cmd->cmd_usage();
5156 	else
5157 		error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
5158 
5159 	endwin();
5160 	free(cmd_argv);
5161 	if (error && error->code != GOT_ERR_CANCELLED)
5162 		fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
5163 	return 0;
5164 }