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