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