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