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