1 /*
2  * Copyright (c) 2018 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 
20 #include <errno.h>
21 #define _XOPEN_SOURCE_EXTENDED
22 #include <curses.h>
23 #undef _XOPEN_SOURCE_EXTENDED
24 #include <panel.h>
25 #include <locale.h>
26 #include <stdlib.h>
27 #include <stdio.h>
28 #include <getopt.h>
29 #include <string.h>
30 #include <err.h>
31 #include <unistd.h>
32 #include <util.h>
33 #include <limits.h>
34 #include <wchar.h>
35 #include <time.h>
36 #include <pthread.h>
37 
38 #include "got_error.h"
39 #include "got_object.h"
40 #include "got_reference.h"
41 #include "got_repository.h"
42 #include "got_diff.h"
43 #include "got_opentemp.h"
44 #include "got_commit_graph.h"
45 #include "got_utf8.h"
46 #include "got_blame.h"
47 
48 #ifndef MIN
49 #define	MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
50 #endif
51 
52 #ifndef nitems
53 #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
54 #endif
55 
56 struct tog_cmd {
57 	const char *name;
58 	const struct got_error *(*cmd_main)(int, char *[]);
59 	void (*cmd_usage)(void);
60 	const char *descr;
61 };
62 
63 __dead static void	usage(void);
64 __dead static void	usage_log(void);
65 __dead static void	usage_diff(void);
66 __dead static void	usage_blame(void);
67 __dead static void	usage_tree(void);
68 
69 static const struct got_error*	cmd_log(int, char *[]);
70 static const struct got_error*	cmd_diff(int, char *[]);
71 static const struct got_error*	cmd_blame(int, char *[]);
72 static const struct got_error*	cmd_tree(int, char *[]);
73 
74 static struct tog_cmd tog_commands[] = {
75 	{ "log",	cmd_log,	usage_log,
76 	    "show repository history" },
77 	{ "diff",	cmd_diff,	usage_diff,
78 	    "compare files and directories" },
79 	{ "blame",	cmd_blame,	usage_blame,
80 	    "show line-by-line file history" },
81 	{ "tree",	cmd_tree,	usage_tree,
82 	    "browse trees in repository" },
83 };
84 
85 static struct tog_view {
86 	WINDOW *window;
87 	PANEL *panel;
88 } tog_log_view, tog_diff_view, tog_blame_view, tog_tree_view;
89 
90 static const struct got_error *
91 show_diff_view(struct got_object *, struct got_object *,
92     struct got_repository *);
93 static const struct got_error *
94 show_log_view(struct got_object_id *, struct got_repository *);
95 static const struct got_error *
96 show_blame_view(const char *, struct got_object_id *, struct got_repository *);
97 static const struct got_error *
98 show_tree_view(struct got_tree_object *, struct got_object_id *,
99     struct got_repository *);
100 
101 __dead static void
102 usage_log(void)
103 {
104 	endwin();
105 	fprintf(stderr, "usage: %s log [-c commit] [repository-path]\n",
106 	    getprogname());
107 	exit(1);
108 }
109 
110 /* Create newly allocated wide-character string equivalent to a byte string. */
111 static const struct got_error *
112 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
113 {
114 	char *vis = NULL;
115 	const struct got_error *err = NULL;
116 
117 	*ws = NULL;
118 	*wlen = mbstowcs(NULL, s, 0);
119 	if (*wlen == (size_t)-1) {
120 		int vislen;
121 		if (errno != EILSEQ)
122 			return got_error_from_errno();
123 
124 		/* byte string invalid in current encoding; try to "fix" it */
125 		err = got_mbsavis(&vis, &vislen, s);
126 		if (err)
127 			return err;
128 		*wlen = mbstowcs(NULL, vis, 0);
129 		if (*wlen == (size_t)-1) {
130 			err = got_error_from_errno(); /* give up */
131 			goto done;
132 		}
133 	}
134 
135 	*ws = calloc(*wlen + 1, sizeof(*ws));
136 	if (*ws == NULL) {
137 		err = got_error_from_errno();
138 		goto done;
139 	}
140 
141 	if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
142 		err = got_error_from_errno();
143 done:
144 	free(vis);
145 	if (err) {
146 		free(*ws);
147 		*ws = NULL;
148 		*wlen = 0;
149 	}
150 	return err;
151 }
152 
153 /* Format a line for display, ensuring that it won't overflow a width limit. */
154 static const struct got_error *
155 format_line(wchar_t **wlinep, int *widthp, const char *line, int wlimit)
156 {
157 	const struct got_error *err = NULL;
158 	int cols = 0;
159 	wchar_t *wline = NULL;
160 	size_t wlen;
161 	int i;
162 
163 	*wlinep = NULL;
164 	*widthp = 0;
165 
166 	err = mbs2ws(&wline, &wlen, line);
167 	if (err)
168 		return err;
169 
170 	i = 0;
171 	while (i < wlen && cols < wlimit) {
172 		int width = wcwidth(wline[i]);
173 		switch (width) {
174 		case 0:
175 			i++;
176 			break;
177 		case 1:
178 		case 2:
179 			if (cols + width <= wlimit) {
180 				cols += width;
181 				i++;
182 			}
183 			break;
184 		case -1:
185 			if (wline[i] == L'\t')
186 				cols += TABSIZE - ((cols + 1) % TABSIZE);
187 			i++;
188 			break;
189 		default:
190 			err = got_error_from_errno();
191 			goto done;
192 		}
193 	}
194 	wline[i] = L'\0';
195 	if (widthp)
196 		*widthp = cols;
197 done:
198 	if (err)
199 		free(wline);
200 	else
201 		*wlinep = wline;
202 	return err;
203 }
204 
205 static const struct got_error *
206 draw_commit(struct got_commit_object *commit, struct got_object_id *id)
207 {
208 	const struct got_error *err = NULL;
209 	char datebuf[10]; /* YY-MM-DD + SPACE + NUL */
210 	char *logmsg0 = NULL, *logmsg = NULL;
211 	char *author0 = NULL, *author = NULL;
212 	wchar_t *wlogmsg = NULL, *wauthor = NULL;
213 	int author_width, logmsg_width;
214 	char *newline, *smallerthan;
215 	char *line = NULL;
216 	int col, limit;
217 	static const size_t date_display_cols = 9;
218 	static const size_t author_display_cols = 16;
219 	const int avail = COLS;
220 
221 	if (strftime(datebuf, sizeof(datebuf), "%g/%m/%d ", &commit->tm_committer)
222 	    >= sizeof(datebuf))
223 		return got_error(GOT_ERR_NO_SPACE);
224 
225 	if (avail < date_display_cols)
226 		limit = MIN(sizeof(datebuf) - 1, avail);
227 	else
228 		limit = MIN(date_display_cols, sizeof(datebuf) - 1);
229 	waddnstr(tog_log_view.window, datebuf, limit);
230 	col = limit + 1;
231 	if (col > avail)
232 		goto done;
233 
234 	author0 = strdup(commit->author);
235 	if (author0 == NULL) {
236 		err = got_error_from_errno();
237 		goto done;
238 	}
239 	author = author0;
240 	smallerthan = strchr(author, '<');
241 	if (smallerthan)
242 		*smallerthan = '\0';
243 	else {
244 		char *at = strchr(author, '@');
245 		if (at)
246 			*at = '\0';
247 	}
248 	limit = avail - col;
249 	err = format_line(&wauthor, &author_width, author, limit);
250 	if (err)
251 		goto done;
252 	waddwstr(tog_log_view.window, wauthor);
253 	col += author_width;
254 	while (col <= avail && author_width < author_display_cols + 1) {
255 		waddch(tog_log_view.window, ' ');
256 		col++;
257 		author_width++;
258 	}
259 	if (col > avail)
260 		goto done;
261 
262 	logmsg0 = strdup(commit->logmsg);
263 	if (logmsg0 == NULL) {
264 		err = got_error_from_errno();
265 		goto done;
266 	}
267 	logmsg = logmsg0;
268 	while (*logmsg == '\n')
269 		logmsg++;
270 	newline = strchr(logmsg, '\n');
271 	if (newline)
272 		*newline = '\0';
273 	limit = avail - col;
274 	err = format_line(&wlogmsg, &logmsg_width, logmsg, limit);
275 	if (err)
276 		goto done;
277 	waddwstr(tog_log_view.window, wlogmsg);
278 	col += logmsg_width;
279 	while (col <= avail) {
280 		waddch(tog_log_view.window, ' ');
281 		col++;
282 	}
283 done:
284 	free(logmsg0);
285 	free(wlogmsg);
286 	free(author0);
287 	free(wauthor);
288 	free(line);
289 	return err;
290 }
291 
292 struct commit_queue_entry {
293 	TAILQ_ENTRY(commit_queue_entry) entry;
294 	struct got_object_id *id;
295 	struct got_commit_object *commit;
296 };
297 TAILQ_HEAD(commit_queue, commit_queue_entry);
298 
299 static struct commit_queue_entry *
300 alloc_commit_queue_entry(struct got_commit_object *commit,
301     struct got_object_id *id)
302 {
303 	struct commit_queue_entry *entry;
304 
305 	entry = calloc(1, sizeof(*entry));
306 	if (entry == NULL)
307 		return NULL;
308 
309 	entry->id = id;
310 	entry->commit = commit;
311 	return entry;
312 }
313 
314 static void
315 pop_commit(struct commit_queue *commits)
316 {
317 	struct commit_queue_entry *entry;
318 
319 	entry = TAILQ_FIRST(commits);
320 	TAILQ_REMOVE(commits, entry, entry);
321 	got_object_commit_close(entry->commit);
322 	/* Don't free entry->id! It is owned by the commit graph. */
323 	free(entry);
324 }
325 
326 static void
327 free_commits(struct commit_queue *commits)
328 {
329 	while (!TAILQ_EMPTY(commits))
330 		pop_commit(commits);
331 }
332 
333 static const struct got_error *
334 queue_commits(struct got_commit_graph *graph, struct commit_queue *commits,
335     struct got_object_id *start_id, struct got_repository *repo)
336 {
337 	const struct got_error *err = NULL;
338 	struct got_object_id *id;
339 	struct commit_queue_entry *entry;
340 
341 	err = got_commit_graph_iter_start(graph, start_id);
342 	if (err)
343 		return err;
344 
345 	entry = TAILQ_LAST(commits, commit_queue);
346 	if (entry && got_object_id_cmp(entry->id, start_id) == 0) {
347 		int nfetched;
348 
349 		/* Start ID's commit is already on the queue; skip over it. */
350 		err = got_commit_graph_iter_next(&id, graph);
351 		if (err && err->code != GOT_ERR_ITER_NEED_MORE)
352 			return err;
353 
354 		err = got_commit_graph_fetch_commits(&nfetched, graph, 1, repo);
355 		if (err)
356 			return err;
357 	}
358 
359 	while (1) {
360 		struct got_commit_object *commit;
361 
362 		err = got_commit_graph_iter_next(&id, graph);
363 		if (err) {
364 			if (err->code == GOT_ERR_ITER_NEED_MORE)
365 				err = NULL;
366 			break;
367 		}
368 
369 		err = got_object_open_as_commit(&commit, repo, id);
370 		if (err)
371 			break;
372 
373 		entry = alloc_commit_queue_entry(commit, id);
374 		if (entry == NULL) {
375 			err = got_error_from_errno();
376 			break;
377 		}
378 
379 		TAILQ_INSERT_TAIL(commits, entry, entry);
380 	}
381 
382 	return err;
383 }
384 
385 static const struct got_error *
386 fetch_next_commit(struct commit_queue_entry **pentry,
387     struct commit_queue_entry *entry, struct commit_queue *commits,
388     struct got_commit_graph *graph, struct got_repository *repo)
389 {
390 	const struct got_error *err = NULL;
391 	struct got_object_qid *qid;
392 
393 	*pentry = NULL;
394 
395 	/* Populate commit graph with entry's parent commits. */
396 	SIMPLEQ_FOREACH(qid, &entry->commit->parent_ids, entry) {
397 		int nfetched;
398 		err = got_commit_graph_fetch_commits_up_to(&nfetched,
399 			graph, qid->id, repo);
400 		if (err)
401 			return err;
402 	}
403 
404 	/* Append outstanding commits to queue in graph sort order. */
405 	err = queue_commits(graph, commits, entry->id, repo);
406 	if (err) {
407 		if (err->code == GOT_ERR_ITER_COMPLETED)
408 			err = NULL;
409 		return err;
410 	}
411 
412 	/* Next entry to display should now be available. */
413 	*pentry = TAILQ_NEXT(entry, entry);
414 	if (*pentry == NULL)
415 		return got_error(GOT_ERR_NO_OBJ);
416 
417 	return NULL;
418 }
419 
420 static const struct got_error *
421 get_head_commit_id(struct got_object_id **head_id, struct got_repository *repo)
422 {
423 	const struct got_error *err = NULL;
424 	struct got_reference *head_ref;
425 
426 	*head_id = NULL;
427 
428 	err = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
429 	if (err)
430 		return err;
431 
432 	err = got_ref_resolve(head_id, repo, head_ref);
433 	got_ref_close(head_ref);
434 	if (err) {
435 		*head_id = NULL;
436 		return err;
437 	}
438 
439 	return NULL;
440 }
441 
442 static const struct got_error *
443 draw_commits(struct commit_queue_entry **last, struct commit_queue_entry **selected,
444     struct commit_queue_entry *first, int selected_idx, int limit)
445 {
446 	const struct got_error *err = NULL;
447 	struct commit_queue_entry *entry;
448 	int ncommits;
449 	char *id_str, *header;
450 	size_t header_len;
451 
452 	entry = first;
453 	*selected = NULL;
454 	ncommits = 0;
455 	while (entry) {
456 		if (++ncommits - 1 == selected_idx) {
457 			*selected = entry;
458 			break;
459 		}
460 		entry = TAILQ_NEXT(entry, entry);
461 	}
462 	if (*selected == NULL)
463 		return got_error(GOT_ERR_RANGE);
464 
465 	err = got_object_id_str(&id_str, (*selected)->id);
466 	if (err)
467 		return err;
468 
469 	if (asprintf(&header, "commit: %s", id_str) == -1) {
470 		err = got_error_from_errno();
471 		free(id_str);
472 		return err;
473 	}
474 
475 	werase(tog_log_view.window);
476 
477 	header_len = strlen(header);
478 	if (header_len > COLS) {
479 		id_str[COLS + 1] = '\0';
480 		header_len = COLS;
481 	}
482 	wprintw(tog_log_view.window, header);
483 	while (header_len < COLS) {
484 		waddch(tog_log_view.window, ' ');
485 		header_len++;
486 	}
487 	free(id_str);
488 	free(header);
489 
490 	entry = first;
491 	*last = first;
492 	ncommits = 0;
493 	while (entry) {
494 		if (ncommits == limit - 1)
495 			break;
496 		if (ncommits == selected_idx) {
497 			wstandout(tog_log_view.window);
498 			*selected = entry;
499 		}
500 		err = draw_commit(entry->commit, entry->id);
501 		if (ncommits == selected_idx)
502 			wstandend(tog_log_view.window);
503 		if (err)
504 			break;
505 		ncommits++;
506 		*last = entry;
507 		entry = TAILQ_NEXT(entry, entry);
508 	}
509 
510 	update_panels();
511 	doupdate();
512 
513 	return err;
514 }
515 
516 static void
517 scroll_up(struct commit_queue_entry **first_displayed_entry, int maxscroll,
518     struct commit_queue *commits)
519 {
520 	struct commit_queue_entry *entry;
521 	int nscrolled = 0;
522 
523 	entry = TAILQ_FIRST(commits);
524 	if (*first_displayed_entry == entry)
525 		return;
526 
527 	entry = *first_displayed_entry;
528 	while (entry && nscrolled < maxscroll) {
529 		entry = TAILQ_PREV(entry, commit_queue, entry);
530 		if (entry) {
531 			*first_displayed_entry = entry;
532 			nscrolled++;
533 		}
534 	}
535 }
536 
537 static const struct got_error *
538 scroll_down(struct commit_queue_entry **first_displayed_entry, int maxscroll,
539     struct commit_queue_entry *last_displayed_entry,
540     struct commit_queue *commits, struct got_commit_graph *graph,
541     struct got_repository *repo)
542 {
543 	const struct got_error *err = NULL;
544 	struct commit_queue_entry *pentry;
545 	int nscrolled = 0;
546 
547 	do {
548 		pentry = TAILQ_NEXT(last_displayed_entry, entry);
549 		if (pentry == NULL) {
550 			err = fetch_next_commit(&pentry, last_displayed_entry,
551 			    commits, graph, repo);
552 			if (err || pentry == NULL)
553 				break;
554 		}
555 		last_displayed_entry = pentry;
556 
557 		pentry = TAILQ_NEXT(*first_displayed_entry, entry);
558 		if (pentry == NULL)
559 			break;
560 		*first_displayed_entry = pentry;
561 	} while (++nscrolled < maxscroll);
562 
563 	return err;
564 }
565 
566 static int
567 num_parents(struct commit_queue_entry *entry)
568 {
569 	int nparents = 0;
570 
571 	while (entry) {
572 		entry = TAILQ_NEXT(entry, entry);
573 		nparents++;
574 	}
575 
576 	return nparents;
577 }
578 
579 static const struct got_error *
580 show_commit(struct commit_queue_entry *entry, struct got_repository *repo)
581 {
582 	const struct got_error *err;
583 	struct got_object *obj1 = NULL, *obj2 = NULL;
584 	struct got_object_qid *parent_id;
585 
586 	err = got_object_open(&obj2, repo, entry->id);
587 	if (err)
588 		return err;
589 
590 	parent_id = SIMPLEQ_FIRST(&entry->commit->parent_ids);
591 	if (parent_id) {
592 		err = got_object_open(&obj1, repo, parent_id->id);
593 		if (err)
594 			goto done;
595 	}
596 
597 	err = show_diff_view(obj1, obj2, repo);
598 done:
599 	if (obj1)
600 		got_object_close(obj1);
601 	if (obj2)
602 		got_object_close(obj2);
603 	return err;
604 }
605 
606 static const struct got_error *
607 browse_commit(struct commit_queue_entry *entry, struct got_repository *repo)
608 {
609 	const struct got_error *err = NULL;
610 	struct got_tree_object *tree;
611 
612 	err = got_object_open_as_tree(&tree, repo, entry->commit->tree_id);
613 	if (err)
614 		return err;
615 
616 	err = show_tree_view(tree, entry->id, repo);
617 	got_object_tree_close(tree);
618 	return err;
619 }
620 
621 static const struct got_error *
622 show_log_view(struct got_object_id *start_id, struct got_repository *repo)
623 {
624 	const struct got_error *err = NULL;
625 	struct got_object_id *head_id = NULL;
626 	int ch, done = 0, selected = 0, nparents, nfetched;
627 	struct got_commit_graph *graph;
628 	struct commit_queue commits;
629 	struct commit_queue_entry *entry = NULL;
630 	struct commit_queue_entry *first_displayed_entry = NULL;
631 	struct commit_queue_entry *last_displayed_entry = NULL;
632 	struct commit_queue_entry *selected_entry = NULL;
633 
634 	if (tog_log_view.window == NULL) {
635 		tog_log_view.window = newwin(0, 0, 0, 0);
636 		if (tog_log_view.window == NULL)
637 			return got_error_from_errno();
638 		keypad(tog_log_view.window, TRUE);
639 	}
640 	if (tog_log_view.panel == NULL) {
641 		tog_log_view.panel = new_panel(tog_log_view.window);
642 		if (tog_log_view.panel == NULL)
643 			return got_error_from_errno();
644 	} else
645 		show_panel(tog_log_view.panel);
646 
647 	err = get_head_commit_id(&head_id, repo);
648 	if (err)
649 		return err;
650 
651 	TAILQ_INIT(&commits);
652 
653 	err = got_commit_graph_open(&graph, head_id, 0, repo);
654 	if (err)
655 		goto done;
656 
657 	/* Populate commit graph with a sufficient number of commits. */
658 	err = got_commit_graph_fetch_commits_up_to(&nfetched, graph, start_id,
659 	    repo);
660 	if (err)
661 		goto done;
662 	err = got_commit_graph_fetch_commits(&nfetched, graph, LINES, repo);
663 	if (err)
664 		goto done;
665 
666 	/*
667 	 * Open the initial batch of commits, sorted in commit graph order.
668 	 * We keep all commits open throughout the lifetime of the log view
669 	 * in order to avoid having to re-fetch commits from disk while
670 	 * updating the display.
671 	 */
672 	err = queue_commits(graph, &commits, head_id, repo);
673 	if (err && err->code != GOT_ERR_ITER_COMPLETED)
674 		goto done;
675 
676 	/* Find entry corresponding to the first commit to display. */
677 	TAILQ_FOREACH(entry, &commits, entry) {
678 		if (got_object_id_cmp(entry->id, start_id) == 0) {
679 			first_displayed_entry = entry;
680 			break;
681 		}
682 	}
683 	if (first_displayed_entry == NULL) {
684 		err = got_error(GOT_ERR_NO_OBJ);
685 		goto done;
686 	}
687 
688 	selected_entry = first_displayed_entry;
689 	while (!done) {
690 		err = draw_commits(&last_displayed_entry, &selected_entry,
691 		    first_displayed_entry, selected, LINES);
692 		if (err)
693 			goto done;
694 
695 		nodelay(stdscr, FALSE);
696 		ch = wgetch(tog_log_view.window);
697 		nodelay(stdscr, TRUE);
698 		switch (ch) {
699 			case ERR:
700 				if (errno) {
701 					err = got_error_from_errno();
702 					goto done;
703 				}
704 				break;
705 			case 'q':
706 				done = 1;
707 				break;
708 			case 'k':
709 			case KEY_UP:
710 				if (selected > 0)
711 					selected--;
712 				if (selected > 0)
713 					break;
714 				scroll_up(&first_displayed_entry, 1, &commits);
715 				break;
716 			case KEY_PPAGE:
717 				if (TAILQ_FIRST(&commits) ==
718 				    first_displayed_entry) {
719 					selected = 0;
720 					break;
721 				}
722 				scroll_up(&first_displayed_entry, LINES,
723 				    &commits);
724 				break;
725 			case 'j':
726 			case KEY_DOWN:
727 				nparents = num_parents(first_displayed_entry);
728 				if (selected < LINES - 2 &&
729 				    selected < nparents - 1) {
730 					selected++;
731 					break;
732 				}
733 				err = scroll_down(&first_displayed_entry, 1,
734 				    last_displayed_entry, &commits, graph,
735 				    repo);
736 				if (err)
737 					goto done;
738 				break;
739 			case KEY_NPAGE:
740 				err = scroll_down(&first_displayed_entry, LINES,
741 				    last_displayed_entry, &commits, graph,
742 				    repo);
743 				if (err)
744 					goto done;
745 				if (last_displayed_entry->commit->nparents > 0)
746 					break;
747 				/* can't scroll any further; move cursor down */
748 				nparents = num_parents(first_displayed_entry);
749 				if (selected < LINES - 2 ||
750 				    selected < nparents - 1)
751 					selected = MIN(LINES - 2, nparents - 1);
752 				break;
753 			case KEY_RESIZE:
754 				if (selected > LINES - 1)
755 					selected = LINES - 2;
756 				break;
757 			case KEY_ENTER:
758 			case '\r':
759 				err = show_commit(selected_entry, repo);
760 				if (err)
761 					goto done;
762 				show_panel(tog_log_view.panel);
763 				break;
764 			case 't':
765 				err = browse_commit(selected_entry, repo);
766 				if (err)
767 					goto done;
768 				show_panel(tog_log_view.panel);
769 				break;
770 			default:
771 				break;
772 		}
773 	}
774 done:
775 	free(head_id);
776 	if (graph)
777 		got_commit_graph_close(graph);
778 	free_commits(&commits);
779 	return err;
780 }
781 
782 static const struct got_error *
783 cmd_log(int argc, char *argv[])
784 {
785 	const struct got_error *error;
786 	struct got_repository *repo;
787 	struct got_object_id *start_id = NULL;
788 	char *repo_path = NULL;
789 	char *start_commit = NULL;
790 	int ch;
791 
792 #ifndef PROFILE
793 	if (pledge("stdio rpath wpath cpath flock proc tty", NULL) == -1)
794 		err(1, "pledge");
795 #endif
796 
797 	while ((ch = getopt(argc, argv, "c:")) != -1) {
798 		switch (ch) {
799 		case 'c':
800 			start_commit = optarg;
801 			break;
802 		default:
803 			usage();
804 			/* NOTREACHED */
805 		}
806 	}
807 
808 	argc -= optind;
809 	argv += optind;
810 
811 	if (argc == 0) {
812 		repo_path = getcwd(NULL, 0);
813 		if (repo_path == NULL)
814 			return got_error_from_errno();
815 	} else if (argc == 1) {
816 		repo_path = realpath(argv[0], NULL);
817 		if (repo_path == NULL)
818 			return got_error_from_errno();
819 	} else
820 		usage_log();
821 
822 	error = got_repo_open(&repo, repo_path);
823 	free(repo_path);
824 	if (error != NULL)
825 		return error;
826 
827 	if (start_commit == NULL) {
828 		error = get_head_commit_id(&start_id, repo);
829 		if (error != NULL)
830 			return error;
831 	} else {
832 		struct got_object *obj;
833 		error = got_object_open_by_id_str(&obj, repo, start_commit);
834 		if (error == NULL) {
835 			start_id = got_object_get_id(obj);
836 			if (start_id == NULL)
837 				error = got_error_from_errno();
838 		}
839 	}
840 	if (error != NULL)
841 		return error;
842 	error = show_log_view(start_id, repo);
843 	free(start_id);
844 	got_repo_close(repo);
845 	return error;
846 }
847 
848 __dead static void
849 usage_diff(void)
850 {
851 	endwin();
852 	fprintf(stderr, "usage: %s diff [repository-path] object1 object2\n",
853 	    getprogname());
854 	exit(1);
855 }
856 
857 static char *
858 parse_next_line(FILE *f, size_t *len)
859 {
860 	char *line;
861 	size_t linelen;
862 	size_t lineno;
863 	const char delim[3] = { '\0', '\0', '\0'};
864 
865 	line = fparseln(f, &linelen, &lineno, delim, 0);
866 	if (len)
867 		*len = linelen;
868 	return line;
869 }
870 
871 static const struct got_error *
872 draw_file(WINDOW *window, FILE *f, int *first_displayed_line,
873     int *last_displayed_line, int *eof, int max_lines)
874 {
875 	const struct got_error *err;
876 	int nlines = 0, nprinted = 0;
877 	char *line;
878 	size_t len;
879 	wchar_t *wline;
880 	int width;
881 
882 	rewind(f);
883 	werase(window);
884 
885 	*eof = 0;
886 	while (nprinted < max_lines) {
887 		line = parse_next_line(f, &len);
888 		if (line == NULL) {
889 			*eof = 1;
890 			break;
891 		}
892 		if (++nlines < *first_displayed_line) {
893 			free(line);
894 			continue;
895 		}
896 
897 		err = format_line(&wline, &width, line, COLS);
898 		if (err) {
899 			free(line);
900 			return err;
901 		}
902 		waddwstr(window, wline);
903 		if (width < COLS)
904 			waddch(window, '\n');
905 		if (++nprinted == 1)
906 			*first_displayed_line = nlines;
907 		free(line);
908 	}
909 	*last_displayed_line = nlines;
910 
911 	update_panels();
912 	doupdate();
913 
914 	return NULL;
915 }
916 
917 static const struct got_error *
918 show_diff_view(struct got_object *obj1, struct got_object *obj2,
919     struct got_repository *repo)
920 {
921 	const struct got_error *err;
922 	FILE *f;
923 	int ch, done = 0, first_displayed_line = 1, last_displayed_line = LINES;
924 	int eof, i;
925 
926 	if (obj1 != NULL && obj2 != NULL &&
927 	    got_object_get_type(obj1) != got_object_get_type(obj2))
928 		return got_error(GOT_ERR_OBJ_TYPE);
929 
930 	f = got_opentemp();
931 	if (f == NULL)
932 		return got_error_from_errno();
933 
934 	switch (got_object_get_type(obj1 ? obj1 : obj2)) {
935 	case GOT_OBJ_TYPE_BLOB:
936 		err = got_diff_objects_as_blobs(obj1, obj2, repo, f);
937 		break;
938 	case GOT_OBJ_TYPE_TREE:
939 		err = got_diff_objects_as_trees(obj1, obj2, repo, f);
940 		break;
941 	case GOT_OBJ_TYPE_COMMIT:
942 		err = got_diff_objects_as_commits(obj1, obj2, repo, f);
943 		break;
944 	default:
945 		return got_error(GOT_ERR_OBJ_TYPE);
946 	}
947 
948 	fflush(f);
949 
950 	if (tog_diff_view.window == NULL) {
951 		tog_diff_view.window = newwin(0, 0, 0, 0);
952 		if (tog_diff_view.window == NULL)
953 			return got_error_from_errno();
954 		keypad(tog_diff_view.window, TRUE);
955 	}
956 	if (tog_diff_view.panel == NULL) {
957 		tog_diff_view.panel = new_panel(tog_diff_view.window);
958 		if (tog_diff_view.panel == NULL)
959 			return got_error_from_errno();
960 	} else
961 		show_panel(tog_diff_view.panel);
962 
963 	while (!done) {
964 		err = draw_file(tog_diff_view.window, f, &first_displayed_line,
965 		    &last_displayed_line, &eof, LINES);
966 		if (err)
967 			break;
968 		nodelay(stdscr, FALSE);
969 		ch = wgetch(tog_diff_view.window);
970 		nodelay(stdscr, TRUE);
971 		switch (ch) {
972 			case 'q':
973 				done = 1;
974 				break;
975 			case 'k':
976 			case KEY_UP:
977 				if (first_displayed_line > 1)
978 					first_displayed_line--;
979 				break;
980 			case KEY_PPAGE:
981 			case KEY_BACKSPACE:
982 				i = 0;
983 				while (i++ < LINES - 1 &&
984 				    first_displayed_line > 1)
985 					first_displayed_line--;
986 				break;
987 			case 'j':
988 			case KEY_DOWN:
989 				if (!eof)
990 					first_displayed_line++;
991 				break;
992 			case KEY_NPAGE:
993 			case ' ':
994 				i = 0;
995 				while (!eof && i++ < LINES - 1) {
996 					char *line = parse_next_line(f, NULL);
997 					first_displayed_line++;
998 					if (line == NULL)
999 						break;
1000 				}
1001 				break;
1002 			default:
1003 				break;
1004 		}
1005 	}
1006 	fclose(f);
1007 	return err;
1008 }
1009 
1010 static const struct got_error *
1011 cmd_diff(int argc, char *argv[])
1012 {
1013 	const struct got_error *error = NULL;
1014 	struct got_repository *repo = NULL;
1015 	struct got_object *obj1 = NULL, *obj2 = NULL;
1016 	char *repo_path = NULL;
1017 	char *obj_id_str1 = NULL, *obj_id_str2 = NULL;
1018 	int ch;
1019 
1020 #ifndef PROFILE
1021 	if (pledge("stdio rpath wpath cpath flock proc tty", NULL) == -1)
1022 		err(1, "pledge");
1023 #endif
1024 
1025 	while ((ch = getopt(argc, argv, "")) != -1) {
1026 		switch (ch) {
1027 		default:
1028 			usage();
1029 			/* NOTREACHED */
1030 		}
1031 	}
1032 
1033 	argc -= optind;
1034 	argv += optind;
1035 
1036 	if (argc == 0) {
1037 		usage_diff(); /* TODO show local worktree changes */
1038 	} else if (argc == 2) {
1039 		repo_path = getcwd(NULL, 0);
1040 		if (repo_path == NULL)
1041 			return got_error_from_errno();
1042 		obj_id_str1 = argv[0];
1043 		obj_id_str2 = argv[1];
1044 	} else if (argc == 3) {
1045 		repo_path = realpath(argv[0], NULL);
1046 		if (repo_path == NULL)
1047 			return got_error_from_errno();
1048 		obj_id_str1 = argv[1];
1049 		obj_id_str2 = argv[2];
1050 	} else
1051 		usage_diff();
1052 
1053 	error = got_repo_open(&repo, repo_path);
1054 	free(repo_path);
1055 	if (error)
1056 		goto done;
1057 
1058 	error = got_object_open_by_id_str(&obj1, repo, obj_id_str1);
1059 	if (error)
1060 		goto done;
1061 
1062 	error = got_object_open_by_id_str(&obj2, repo, obj_id_str2);
1063 	if (error)
1064 		goto done;
1065 
1066 	error = show_diff_view(obj1, obj2, repo);
1067 done:
1068 	got_repo_close(repo);
1069 	if (obj1)
1070 		got_object_close(obj1);
1071 	if (obj2)
1072 		got_object_close(obj2);
1073 	return error;
1074 }
1075 
1076 __dead static void
1077 usage_blame(void)
1078 {
1079 	endwin();
1080 	fprintf(stderr, "usage: %s blame [-c commit] [repository-path] path\n",
1081 	    getprogname());
1082 	exit(1);
1083 }
1084 
1085 struct tog_blame_line {
1086 	int annotated;
1087 	struct got_object_id *id;
1088 };
1089 
1090 static const struct got_error *
1091 draw_blame(WINDOW *window, struct got_object_id *id, FILE *f, const char *path,
1092     struct tog_blame_line *lines, int nlines, int blame_complete,
1093     int selected_line, int *first_displayed_line, int *last_displayed_line,
1094     int *eof, int max_lines)
1095 {
1096 	const struct got_error *err;
1097 	int lineno = 0, nprinted = 0;
1098 	char *line;
1099 	size_t len;
1100 	wchar_t *wline;
1101 	int width, wlimit;
1102 	struct tog_blame_line *blame_line;
1103 	struct got_object_id *prev_id = NULL;
1104 	char *id_str;
1105 
1106 	err = got_object_id_str(&id_str, id);
1107 	if (err)
1108 		return err;
1109 
1110 	rewind(f);
1111 	werase(window);
1112 
1113 	if (asprintf(&line, "commit: %s", id_str) == -1) {
1114 		err = got_error_from_errno();
1115 		free(id_str);
1116 		return err;
1117 	}
1118 
1119 	err = format_line(&wline, &width, line, COLS);
1120 	free(line);
1121 	waddwstr(window, wline);
1122 	if (width < COLS)
1123 		waddch(window, '\n');
1124 
1125 	if (asprintf(&line, "[%d/%d] %s%s",
1126 	    *first_displayed_line - 1 + selected_line, nlines,
1127 	    blame_complete ? "" : "annotating ", path) == -1) {
1128 		free(id_str);
1129 		return got_error_from_errno();
1130 	}
1131 	free(id_str);
1132 	err = format_line(&wline, &width, line, COLS);
1133 	free(line);
1134 	if (err)
1135 		return err;
1136 	waddwstr(window, wline);
1137 	if (width < COLS)
1138 		waddch(window, '\n');
1139 
1140 	*eof = 0;
1141 	while (nprinted < max_lines - 2) {
1142 		line = parse_next_line(f, &len);
1143 		if (line == NULL) {
1144 			*eof = 1;
1145 			break;
1146 		}
1147 		if (++lineno < *first_displayed_line) {
1148 			free(line);
1149 			continue;
1150 		}
1151 
1152 		wlimit = COLS < 9 ? 0 : COLS - 9;
1153 		err = format_line(&wline, &width, line, wlimit);
1154 		if (err) {
1155 			free(line);
1156 			return err;
1157 		}
1158 
1159 		if (nprinted == selected_line - 1)
1160 			wstandout(window);
1161 
1162 		blame_line = &lines[lineno - 1];
1163 		if (blame_line->annotated && prev_id &&
1164 		    got_object_id_cmp(prev_id, blame_line->id) == 0)
1165 			waddstr(window, "         ");
1166 		else if (blame_line->annotated) {
1167 			char *id_str;
1168 			err = got_object_id_str(&id_str, blame_line->id);
1169 			if (err) {
1170 				free(line);
1171 				return err;
1172 			}
1173 			wprintw(window, "%.8s ", id_str);
1174 			free(id_str);
1175 			prev_id = blame_line->id;
1176 		} else {
1177 			waddstr(window, "........ ");
1178 			prev_id = NULL;
1179 		}
1180 
1181 		waddwstr(window, wline);
1182 		while (width < wlimit) {
1183 			waddch(window, ' '); /* width == wlimit - 1 ? '\n' : ' '); */
1184 			width++;
1185 		}
1186 		if (nprinted == selected_line - 1)
1187 			wstandend(window);
1188 		if (++nprinted == 1)
1189 			*first_displayed_line = lineno;
1190 		free(line);
1191 	}
1192 	*last_displayed_line = lineno;
1193 
1194 	update_panels();
1195 	doupdate();
1196 
1197 	return NULL;
1198 }
1199 
1200 struct tog_blame_cb_args {
1201 	pthread_mutex_t *mutex;
1202 	struct tog_blame_line *lines; /* one per line */
1203 	int nlines;
1204 
1205 	struct got_object_id *commit_id;
1206 	FILE *f;
1207 	const char *path;
1208 	int *first_displayed_line;
1209 	int *last_displayed_line;
1210 	int *selected_line;
1211 	int *quit;
1212 };
1213 
1214 static const struct got_error *
1215 blame_cb(void *arg, int nlines, int lineno, struct got_object_id *id)
1216 {
1217 	const struct got_error *err = NULL;
1218 	struct tog_blame_cb_args *a = arg;
1219 	struct tog_blame_line *line;
1220 	int eof;
1221 
1222 	if (nlines != a->nlines ||
1223 	    (lineno != -1 && lineno < 1) || lineno > a->nlines)
1224 		return got_error(GOT_ERR_RANGE);
1225 
1226 	if (pthread_mutex_lock(a->mutex) != 0)
1227 		return got_error_from_errno();
1228 
1229 	if (*a->quit) {	/* user has quit the blame view */
1230 		err = got_error(GOT_ERR_ITER_COMPLETED);
1231 		goto done;
1232 	}
1233 
1234 	if (lineno == -1)
1235 		goto done; /* no change in this commit */
1236 
1237 	line = &a->lines[lineno - 1];
1238 	if (line->annotated)
1239 		goto done;
1240 
1241 	line->id = got_object_id_dup(id);
1242 	if (line->id == NULL) {
1243 		err = got_error_from_errno();
1244 		goto done;
1245 	}
1246 	line->annotated = 1;
1247 
1248 	err = draw_blame(tog_blame_view.window, a->commit_id, a->f, a->path,
1249 	    a->lines, a->nlines, 0, *a->selected_line, a->first_displayed_line,
1250 	    a->last_displayed_line, &eof, LINES);
1251 done:
1252 	if (pthread_mutex_unlock(a->mutex) != 0)
1253 		return got_error_from_errno();
1254 	return err;
1255 }
1256 
1257 struct tog_blame_thread_args {
1258 	const char *path;
1259 	struct got_repository *repo;
1260 	struct tog_blame_cb_args *cb_args;
1261 	int *complete;
1262 };
1263 
1264 static void *
1265 blame_thread(void *arg)
1266 {
1267 	const struct got_error *err;
1268 	struct tog_blame_thread_args *ta = arg;
1269 	struct tog_blame_cb_args *a = ta->cb_args;
1270 	int eof;
1271 
1272 	err = got_blame_incremental(ta->path, a->commit_id, ta->repo,
1273 	    blame_cb, ta->cb_args);
1274 	got_repo_close(ta->repo);
1275 	ta->repo = NULL;
1276 	*ta->complete = 1;
1277 	if (err)
1278 		return (void *)err;
1279 
1280 	if (pthread_mutex_lock(a->mutex) != 0)
1281 		return (void *)got_error_from_errno();
1282 
1283 	err = draw_blame(tog_blame_view.window, a->commit_id, a->f, a->path,
1284 	    a->lines, a->nlines, 1, *a->selected_line, a->first_displayed_line,
1285 	    a->last_displayed_line, &eof, LINES);
1286 
1287 	if (pthread_mutex_unlock(a->mutex) != 0 && err == NULL)
1288 		err = got_error_from_errno();
1289 
1290 	return (void *)err;
1291 }
1292 
1293 static struct got_object_id *
1294 get_selected_commit_id(struct tog_blame_line *lines,
1295     int first_displayed_line, int selected_line)
1296 {
1297 	struct tog_blame_line *line;
1298 
1299 	line = &lines[first_displayed_line - 1 + selected_line - 1];
1300 	if (!line->annotated)
1301 		return NULL;
1302 
1303 	return line->id;
1304 }
1305 
1306 static const struct got_error *
1307 open_selected_commit(struct got_object **pobj, struct got_object **obj,
1308     struct tog_blame_line *lines, int first_displayed_line,
1309     int selected_line, struct got_repository *repo)
1310 {
1311 	const struct got_error *err = NULL;
1312 	struct got_commit_object *commit = NULL;
1313 	struct got_object_id *selected_id;
1314 	struct got_object_qid *pid;
1315 
1316 	*pobj = NULL;
1317 	*obj = NULL;
1318 
1319 	selected_id = get_selected_commit_id(lines,
1320 	    first_displayed_line, selected_line);
1321 	if (selected_id == NULL)
1322 		return NULL;
1323 
1324 	err = got_object_open(obj, repo, selected_id);
1325 	if (err)
1326 		goto done;
1327 
1328 	err = got_object_commit_open(&commit, repo, *obj);
1329 	if (err)
1330 		goto done;
1331 
1332 	pid = SIMPLEQ_FIRST(&commit->parent_ids);
1333 	if (pid) {
1334 		err = got_object_open(pobj, repo, pid->id);
1335 		if (err)
1336 			goto done;
1337 	}
1338 done:
1339 	if (commit)
1340 		got_object_commit_close(commit);
1341 	return err;
1342 }
1343 
1344 struct tog_blame {
1345 	FILE *f;
1346 	size_t filesize;
1347 	struct tog_blame_line *lines;
1348 	size_t nlines;
1349 	pthread_t thread;
1350 	struct tog_blame_thread_args thread_args;
1351 	struct tog_blame_cb_args cb_args;
1352 	const char *path;
1353 	struct got_object_id *commit_id;
1354 };
1355 
1356 static const struct got_error *
1357 stop_blame(struct tog_blame *blame)
1358 {
1359 	const struct got_error *err = NULL;
1360 	int i;
1361 
1362 	if (blame->thread) {
1363 		if (pthread_join(blame->thread, (void **)&err) != 0)
1364 			err = got_error_from_errno();
1365 		if (err && err->code == GOT_ERR_ITER_COMPLETED)
1366 			err = NULL;
1367 		blame->thread = NULL;
1368 	}
1369 	if (blame->thread_args.repo) {
1370 		got_repo_close(blame->thread_args.repo);
1371 		blame->thread_args.repo = NULL;
1372 	}
1373 	if (blame->f) {
1374 		fclose(blame->f);
1375 		blame->f = NULL;
1376 	}
1377 	for (i = 0; i < blame->nlines; i++)
1378 		free(blame->lines[i].id);
1379 	free(blame->lines);
1380 	blame->lines = NULL;
1381 	free(blame->commit_id);
1382 	blame->commit_id = NULL;
1383 
1384 	return err;
1385 }
1386 
1387 static const struct got_error *
1388 run_blame(struct tog_blame *blame, pthread_mutex_t *mutex, int *blame_complete,
1389     int *first_displayed_line, int *last_displayed_line,
1390     int *selected_line, int *done, const char *path,
1391     struct got_object_id *commit_id,
1392     struct got_repository *repo)
1393 {
1394 	const struct got_error *err = NULL;
1395 	struct got_blob_object *blob = NULL;
1396 	struct got_repository *thread_repo = NULL;
1397 	struct got_object *obj;
1398 
1399 	err = got_object_open_by_path(&obj, repo, commit_id, path);
1400 	if (err)
1401 		goto done;
1402 	if (got_object_get_type(obj) != GOT_OBJ_TYPE_BLOB) {
1403 		err = got_error(GOT_ERR_OBJ_TYPE);
1404 		goto done;
1405 	}
1406 
1407 	err = got_object_blob_open(&blob, repo, obj, 8192);
1408 	if (err)
1409 		goto done;
1410 	blame->f = got_opentemp();
1411 	if (blame->f == NULL) {
1412 		err = got_error_from_errno();
1413 		goto done;
1414 	}
1415 	err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
1416 	    blame->f, blob);
1417 	if (err)
1418 		goto done;
1419 
1420 	blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
1421 	if (blame->lines == NULL) {
1422 		err = got_error_from_errno();
1423 		goto done;
1424 	}
1425 
1426 	err = got_repo_open(&thread_repo, got_repo_get_path(repo));
1427 	if (err)
1428 		goto done;
1429 
1430 	blame->cb_args.lines = blame->lines;
1431 	blame->cb_args.nlines = blame->nlines;
1432 	blame->cb_args.mutex = mutex;
1433 	blame->cb_args.commit_id = got_object_id_dup(commit_id);
1434 	if (blame->cb_args.commit_id == NULL) {
1435 		err = got_error_from_errno();
1436 		goto done;
1437 	}
1438 	blame->cb_args.f = blame->f;
1439 	blame->cb_args.path = path;
1440 	blame->cb_args.first_displayed_line = first_displayed_line;
1441 	blame->cb_args.selected_line = selected_line;
1442 	blame->cb_args.last_displayed_line = last_displayed_line;
1443 	blame->cb_args.quit = done;
1444 
1445 	blame->thread_args.path = path;
1446 	blame->thread_args.repo = thread_repo;
1447 	blame->thread_args.cb_args = &blame->cb_args;
1448 	blame->thread_args.complete = blame_complete;
1449 	*blame_complete = 0;
1450 
1451 	if (pthread_create(&blame->thread, NULL, blame_thread,
1452 	    &blame->thread_args) != 0) {
1453 		err = got_error_from_errno();
1454 		goto done;
1455 	}
1456 
1457 done:
1458 	if (blob)
1459 		got_object_blob_close(blob);
1460 	if (obj)
1461 		got_object_close(obj);
1462 	if (err)
1463 		stop_blame(blame);
1464 	return err;
1465 }
1466 
1467 static const struct got_error *
1468 show_blame_view(const char *path, struct got_object_id *commit_id,
1469     struct got_repository *repo)
1470 {
1471 	const struct got_error *err = NULL, *thread_err = NULL;
1472 	int ch, done = 0, first_displayed_line = 1, last_displayed_line = LINES;
1473 	int selected_line = first_displayed_line;
1474 	int eof, blame_complete = 0;
1475 	struct got_object *obj = NULL, *pobj = NULL;
1476 	pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
1477 	struct tog_blame blame;
1478 	int blame_running = 0;
1479 	struct got_object_id_queue blamed_commits;
1480 	struct got_object_qid *blamed_commit = NULL;
1481 
1482 	SIMPLEQ_INIT(&blamed_commits);
1483 
1484 	if (pthread_mutex_init(&mutex, NULL) != 0) {
1485 		err = got_error_from_errno();
1486 		goto done;
1487 	}
1488 
1489 	err = got_object_qid_alloc(&blamed_commit, commit_id);
1490 	if (err)
1491 		goto done;
1492 	SIMPLEQ_INSERT_HEAD(&blamed_commits, blamed_commit, entry);
1493 
1494 	if (tog_blame_view.window == NULL) {
1495 		tog_blame_view.window = newwin(0, 0, 0, 0);
1496 		if (tog_blame_view.window == NULL)
1497 			return got_error_from_errno();
1498 		keypad(tog_blame_view.window, TRUE);
1499 	}
1500 	if (tog_blame_view.panel == NULL) {
1501 		tog_blame_view.panel = new_panel(tog_blame_view.window);
1502 		if (tog_blame_view.panel == NULL)
1503 			return got_error_from_errno();
1504 	} else
1505 		show_panel(tog_blame_view.panel);
1506 
1507 	memset(&blame, 0, sizeof(blame));
1508 	err = run_blame(&blame, &mutex, &blame_complete,
1509 	    &first_displayed_line, &last_displayed_line,
1510 	    &selected_line, &done, path, blamed_commit->id, repo);
1511 	if (err)
1512 		return err;
1513 
1514 	while (!done) {
1515 		if (pthread_mutex_lock(&mutex) != 0) {
1516 			err = got_error_from_errno();
1517 			goto done;
1518 		}
1519 		err = draw_blame(tog_blame_view.window, blamed_commit->id,
1520 		    blame.f, path, blame.lines, blame.nlines, blame_complete,
1521 		    selected_line, &first_displayed_line, &last_displayed_line,
1522 		    &eof, LINES);
1523 		if (pthread_mutex_unlock(&mutex) != 0) {
1524 			err = got_error_from_errno();
1525 			goto done;
1526 		}
1527 		if (err)
1528 			break;
1529 		nodelay(stdscr, FALSE);
1530 		ch = wgetch(tog_blame_view.window);
1531 		nodelay(stdscr, TRUE);
1532 		if (pthread_mutex_lock(&mutex) != 0) {
1533 			err = got_error_from_errno();
1534 			goto done;
1535 		}
1536 		switch (ch) {
1537 			case 'q':
1538 				done = 1;
1539 				break;
1540 			case 'k':
1541 			case KEY_UP:
1542 				if (selected_line > 1)
1543 					selected_line--;
1544 				else if (selected_line == 1 &&
1545 				    first_displayed_line > 1)
1546 					first_displayed_line--;
1547 				break;
1548 			case KEY_PPAGE:
1549 			case KEY_BACKSPACE:
1550 				if (first_displayed_line == 1) {
1551 					selected_line = 1;
1552 					break;
1553 				}
1554 				if (first_displayed_line > LINES - 2)
1555 					first_displayed_line -= (LINES - 2);
1556 				else
1557 					first_displayed_line = 1;
1558 				break;
1559 			case 'j':
1560 			case KEY_DOWN:
1561 				if (selected_line < LINES - 2 &&
1562 				    first_displayed_line + selected_line <=
1563 				    blame.nlines)
1564 					selected_line++;
1565 				else if (last_displayed_line < blame.nlines)
1566 					first_displayed_line++;
1567 				break;
1568 			case 'b':
1569 			case 'p': {
1570 				struct got_object_id *id;
1571 				id = get_selected_commit_id(blame.lines,
1572 				    first_displayed_line, selected_line);
1573 				if (id == NULL || got_object_id_cmp(id,
1574 				    blamed_commit->id) == 0)
1575 					break;
1576 				err = open_selected_commit(&pobj, &obj,
1577 				    blame.lines, first_displayed_line,
1578 				    selected_line, repo);
1579 				if (err)
1580 					break;
1581 				if (pobj == NULL && obj == NULL)
1582 					break;
1583 				if (ch == 'p' && pobj == NULL)
1584 					break;
1585 				done = 1;
1586 				if (pthread_mutex_unlock(&mutex) != 0) {
1587 					err = got_error_from_errno();
1588 					goto done;
1589 				}
1590 				thread_err = stop_blame(&blame);
1591 				blame_running = 0;
1592 				done = 0;
1593 				if (pthread_mutex_lock(&mutex) != 0) {
1594 					err = got_error_from_errno();
1595 					goto done;
1596 				}
1597 				if (thread_err)
1598 					break;
1599 				id = got_object_get_id(ch == 'b' ? obj : pobj);
1600 				got_object_close(obj);
1601 				obj = NULL;
1602 				if (pobj) {
1603 					got_object_close(pobj);
1604 					pobj = NULL;
1605 				}
1606 				if (id == NULL) {
1607 					err = got_error_from_errno();
1608 					break;
1609 				}
1610 				err = got_object_qid_alloc(&blamed_commit, id);
1611 				free(id);
1612 				if (err)
1613 					goto done;
1614 				SIMPLEQ_INSERT_HEAD(&blamed_commits,
1615 				    blamed_commit, entry);
1616 				err = run_blame(&blame, &mutex,
1617 				    &blame_complete, &first_displayed_line,
1618 				    &last_displayed_line, &selected_line,
1619 				    &done, path, blamed_commit->id, repo);
1620 				if (err)
1621 					break;
1622 				blame_running = 1;
1623 				break;
1624 			}
1625 			case 'B': {
1626 				struct got_object_qid *first;
1627 				first = SIMPLEQ_FIRST(&blamed_commits);
1628 				if (!got_object_id_cmp(first->id, commit_id))
1629 					break;
1630 				done = 1;
1631 				if (pthread_mutex_unlock(&mutex) != 0) {
1632 					err = got_error_from_errno();
1633 					goto done;
1634 				}
1635 				thread_err = stop_blame(&blame);
1636 				blame_running = 0;
1637 				done = 0;
1638 				if (pthread_mutex_lock(&mutex) != 0) {
1639 					err = got_error_from_errno();
1640 					goto done;
1641 				}
1642 				if (thread_err)
1643 					break;
1644 				SIMPLEQ_REMOVE_HEAD(&blamed_commits, entry);
1645 				got_object_qid_free(blamed_commit);
1646 				blamed_commit = SIMPLEQ_FIRST(&blamed_commits);
1647 				err = run_blame(&blame, &mutex,
1648 				    &blame_complete, &first_displayed_line,
1649 				    &last_displayed_line, &selected_line,
1650 				    &done, path, blamed_commit->id, repo);
1651 				if (err)
1652 					break;
1653 				blame_running = 1;
1654 				break;
1655 			}
1656 			case KEY_ENTER:
1657 			case '\r':
1658 				err = open_selected_commit(&pobj, &obj,
1659 				    blame.lines, first_displayed_line,
1660 				    selected_line, repo);
1661 				if (err)
1662 					break;
1663 				if (pobj == NULL && obj == NULL)
1664 					break;
1665 				err = show_diff_view(pobj, obj, repo);
1666 				if (pobj) {
1667 					got_object_close(pobj);
1668 					pobj = NULL;
1669 				}
1670 				got_object_close(obj);
1671 				obj = NULL;
1672 				show_panel(tog_blame_view.panel);
1673 				if (err)
1674 					break;
1675 				break;
1676 			case KEY_NPAGE:
1677 			case ' ':
1678 				if (last_displayed_line >= blame.nlines &&
1679 				    selected_line < LINES - 2) {
1680 					selected_line = MIN(blame.nlines,
1681 					    LINES - 2);
1682 					break;
1683 				}
1684 				if (last_displayed_line + LINES - 2 <=
1685 				    blame.nlines)
1686 					first_displayed_line += LINES - 2;
1687 				else
1688 					first_displayed_line =
1689 					    blame.nlines - (LINES - 3);
1690 				break;
1691 			default:
1692 				break;
1693 		}
1694 		if (pthread_mutex_unlock(&mutex) != 0)
1695 			err = got_error_from_errno();
1696 		if (err || thread_err)
1697 			break;
1698 	}
1699 done:
1700 	if (pobj)
1701 		got_object_close(pobj);
1702 	if (blame_running)
1703 		thread_err = stop_blame(&blame);
1704 	while (!SIMPLEQ_EMPTY(&blamed_commits)) {
1705 		blamed_commit = SIMPLEQ_FIRST(&blamed_commits);
1706 		SIMPLEQ_REMOVE_HEAD(&blamed_commits, entry);
1707 		got_object_qid_free(blamed_commit);
1708 	}
1709 	return thread_err ? thread_err : err;
1710 }
1711 
1712 static const struct got_error *
1713 cmd_blame(int argc, char *argv[])
1714 {
1715 	const struct got_error *error;
1716 	struct got_repository *repo = NULL;
1717 	char *repo_path = NULL;
1718 	char *path = NULL;
1719 	struct got_object_id *commit_id = NULL;
1720 	char *commit_id_str = NULL;
1721 	int ch;
1722 
1723 #ifndef PROFILE
1724 	if (pledge("stdio rpath wpath cpath flock proc tty", NULL) == -1)
1725 		err(1, "pledge");
1726 #endif
1727 
1728 	while ((ch = getopt(argc, argv, "c:")) != -1) {
1729 		switch (ch) {
1730 		case 'c':
1731 			commit_id_str = optarg;
1732 			break;
1733 		default:
1734 			usage();
1735 			/* NOTREACHED */
1736 		}
1737 	}
1738 
1739 	argc -= optind;
1740 	argv += optind;
1741 
1742 	if (argc == 0) {
1743 		usage_blame();
1744 	} else if (argc == 1) {
1745 		repo_path = getcwd(NULL, 0);
1746 		if (repo_path == NULL)
1747 			return got_error_from_errno();
1748 		path = argv[0];
1749 	} else if (argc == 2) {
1750 		repo_path = realpath(argv[0], NULL);
1751 		if (repo_path == NULL)
1752 			return got_error_from_errno();
1753 		path = argv[1];
1754 	} else
1755 		usage_blame();
1756 
1757 	error = got_repo_open(&repo, repo_path);
1758 	free(repo_path);
1759 	if (error != NULL)
1760 		return error;
1761 
1762 	if (commit_id_str == NULL) {
1763 		struct got_reference *head_ref;
1764 		error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1765 		if (error != NULL)
1766 			goto done;
1767 		error = got_ref_resolve(&commit_id, repo, head_ref);
1768 		got_ref_close(head_ref);
1769 	} else {
1770 		struct got_object *obj;
1771 		error = got_object_open_by_id_str(&obj, repo, commit_id_str);
1772 		if (error != NULL)
1773 			goto done;
1774 		commit_id = got_object_get_id(obj);
1775 		if (commit_id == NULL)
1776 			error = got_error_from_errno();
1777 		got_object_close(obj);
1778 	}
1779 	if (error != NULL)
1780 		goto done;
1781 
1782 	error = show_blame_view(path, commit_id, repo);
1783 done:
1784 	free(commit_id);
1785 	if (repo)
1786 		got_repo_close(repo);
1787 	return error;
1788 }
1789 
1790 static const struct got_error *
1791 draw_tree_entries(struct got_tree_entry **first_displayed_entry,
1792     struct got_tree_entry **last_displayed_entry,
1793     struct got_tree_entry **selected_entry, int *ndisplayed,
1794     WINDOW *window, const char *label, int show_ids, const char *parent_path,
1795     const struct got_tree_entries *entries, int selected, int limit, int isroot)
1796 {
1797 	const struct got_error *err = NULL;
1798 	struct got_tree_entry *te;
1799 	wchar_t *wline;
1800 	int width, n;
1801 
1802 	*ndisplayed = 0;
1803 
1804 	werase(window);
1805 
1806 	if (limit == 0)
1807 		return NULL;
1808 
1809 	err = format_line(&wline, &width, label, COLS);
1810 	if (err)
1811 		return err;
1812 	waddwstr(window, wline);
1813 	if (width < COLS)
1814 		waddch(window, '\n');
1815 	if (--limit <= 0)
1816 		return NULL;
1817 	err = format_line(&wline, &width, parent_path, COLS);
1818 	if (err)
1819 		return err;
1820 	waddwstr(window, wline);
1821 	if (width < COLS)
1822 		waddch(window, '\n');
1823 	if (--limit <= 0)
1824 		return NULL;
1825 	waddch(window, '\n');
1826 	if (--limit <= 0)
1827 		return NULL;
1828 
1829 	te = SIMPLEQ_FIRST(&entries->head);
1830 	if (*first_displayed_entry == NULL) {
1831 		if (selected == 0) {
1832 			wstandout(window);
1833 			*selected_entry = NULL;
1834 		}
1835 		waddstr(window, "  ..\n");	/* parent directory */
1836 		if (selected == 0)
1837 			wstandend(window);
1838 		(*ndisplayed)++;
1839 		if (--limit <= 0)
1840 			return NULL;
1841 		n = 1;
1842 	} else {
1843 		n = 0;
1844 		while (te != *first_displayed_entry)
1845 			te = SIMPLEQ_NEXT(te, entry);
1846 	}
1847 
1848 	while (te) {
1849 		char *line = NULL, *id_str = NULL;
1850 
1851 		if (show_ids) {
1852 			err = got_object_id_str(&id_str, te->id);
1853 			if (err)
1854 				return got_error_from_errno();
1855 		}
1856 		if (asprintf(&line, "%s  %s%s", id_str ? id_str : "",
1857 		    te->name, S_ISDIR(te->mode) ? "/" : "") == -1) {
1858 			free(id_str);
1859 			return got_error_from_errno();
1860 		}
1861 		free(id_str);
1862 		err = format_line(&wline, &width, line, COLS);
1863 		if (err) {
1864 			free(line);
1865 			break;
1866 		}
1867 		if (n == selected) {
1868 			wstandout(window);
1869 			*selected_entry = te;
1870 		}
1871 		waddwstr(window, wline);
1872 		if (width < COLS)
1873 			waddch(window, '\n');
1874 		if (n == selected)
1875 			wstandend(window);
1876 		free(line);
1877 		n++;
1878 		(*ndisplayed)++;
1879 		*last_displayed_entry = te;
1880 		if (--limit <= 0)
1881 			break;
1882 		te = SIMPLEQ_NEXT(te, entry);
1883 	}
1884 
1885 	return err;
1886 }
1887 
1888 static void
1889 tree_scroll_up(struct got_tree_entry **first_displayed_entry, int maxscroll,
1890     const struct got_tree_entries *entries, int isroot)
1891 {
1892 	struct got_tree_entry *te, *prev;
1893 	int i;
1894 
1895 	if (*first_displayed_entry == NULL)
1896 		return;
1897 
1898 	te = SIMPLEQ_FIRST(&entries->head);
1899 	if (*first_displayed_entry == te) {
1900 		if (!isroot)
1901 			*first_displayed_entry = NULL;
1902 		return;
1903 	}
1904 
1905 	/* XXX this is stupid... switch to TAILQ? */
1906 	for (i = 0; i < maxscroll; i++) {
1907 		while (te != *first_displayed_entry) {
1908 			prev = te;
1909 			te = SIMPLEQ_NEXT(te, entry);
1910 		}
1911 		*first_displayed_entry = prev;
1912 		te = SIMPLEQ_FIRST(&entries->head);
1913 	}
1914 	if (!isroot && te == SIMPLEQ_FIRST(&entries->head) && i < maxscroll)
1915 		*first_displayed_entry = NULL;
1916 }
1917 
1918 static void
1919 tree_scroll_down(struct got_tree_entry **first_displayed_entry, int maxscroll,
1920 	struct got_tree_entry *last_displayed_entry,
1921 	const struct got_tree_entries *entries)
1922 {
1923 	struct got_tree_entry *next;
1924 	int n = 0;
1925 
1926 	if (SIMPLEQ_NEXT(last_displayed_entry, entry) == NULL)
1927 		return;
1928 
1929 	if (*first_displayed_entry)
1930 		next = SIMPLEQ_NEXT(*first_displayed_entry, entry);
1931 	else
1932 		next = SIMPLEQ_FIRST(&entries->head);
1933 	while (next) {
1934 		*first_displayed_entry = next;
1935 		if (++n >= maxscroll)
1936 			break;
1937 		next = SIMPLEQ_NEXT(next, entry);
1938 	}
1939 }
1940 
1941 struct tog_parent_tree {
1942 	TAILQ_ENTRY(tog_parent_tree) entry;
1943 	struct got_tree_object *tree;
1944 	struct got_tree_entry *first_displayed_entry;
1945 	struct got_tree_entry *selected_entry;
1946 	int selected;
1947 };
1948 
1949 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
1950 
1951 static const struct got_error *
1952 tree_entry_path(char **path, struct tog_parent_trees *parents,
1953     struct got_tree_entry *te)
1954 {
1955 	const struct got_error *err = NULL;
1956 	struct tog_parent_tree *pt;
1957 	size_t len = 2; /* for leading slash and NUL */
1958 
1959 	TAILQ_FOREACH(pt, parents, entry)
1960 		len += strlen(pt->selected_entry->name) + 1 /* slash */;
1961 	if (te)
1962 		len += strlen(te->name);
1963 
1964 	*path = calloc(1, len);
1965 	if (path == NULL)
1966 		return got_error_from_errno();
1967 
1968 	(*path)[0] = '/';
1969 	pt = TAILQ_LAST(parents, tog_parent_trees);
1970 	while (pt) {
1971 		if (strlcat(*path, pt->selected_entry->name, len) >= len) {
1972 			err = got_error(GOT_ERR_NO_SPACE);
1973 			goto done;
1974 		}
1975 		if (strlcat(*path, "/", len) >= len) {
1976 			err = got_error(GOT_ERR_NO_SPACE);
1977 			goto done;
1978 		}
1979 		pt = TAILQ_PREV(pt, tog_parent_trees, entry);
1980 	}
1981 	if (te) {
1982 		if (strlcat(*path, te->name, len) >= len) {
1983 			err = got_error(GOT_ERR_NO_SPACE);
1984 			goto done;
1985 		}
1986 	}
1987 done:
1988 	if (err) {
1989 		free(*path);
1990 		*path = NULL;
1991 	}
1992 	return err;
1993 }
1994 
1995 static const struct got_error *
1996 blame_tree_entry(struct got_tree_entry *te, struct tog_parent_trees *parents,
1997     struct got_object_id *commit_id, struct got_repository *repo)
1998 {
1999 	const struct got_error *err = NULL;
2000 	char *path;
2001 	
2002 	err = tree_entry_path(&path, parents, te);
2003 	if (err)
2004 		return err;
2005 
2006 	err = show_blame_view(path, commit_id, repo);
2007 	free(path);
2008 	return err;
2009 }
2010 
2011 static const struct got_error *
2012 show_tree_view(struct got_tree_object *root, struct got_object_id *commit_id,
2013     struct got_repository *repo)
2014 {
2015 	const struct got_error *err = NULL;
2016 	int ch, done = 0, selected = 0, show_ids = 0;
2017 	struct got_tree_object *tree = root;
2018 	const struct got_tree_entries *entries;
2019 	struct got_tree_entry *first_displayed_entry = NULL;
2020 	struct got_tree_entry *last_displayed_entry = NULL;
2021 	struct got_tree_entry *selected_entry = NULL;
2022 	char *commit_id_str = NULL, *tree_label = NULL;
2023 	int nentries, ndisplayed;
2024 	struct tog_parent_trees parents;
2025 
2026 	TAILQ_INIT(&parents);
2027 
2028 	err = got_object_id_str(&commit_id_str, commit_id);
2029 	if (err != NULL)
2030 		goto done;
2031 
2032 	if (asprintf(&tree_label, "commit: %s", commit_id_str) == -1) {
2033 		err = got_error_from_errno();
2034 		goto done;
2035 	}
2036 
2037 	if (tog_tree_view.window == NULL) {
2038 		tog_tree_view.window = newwin(0, 0, 0, 0);
2039 		if (tog_tree_view.window == NULL)
2040 			return got_error_from_errno();
2041 		keypad(tog_tree_view.window, TRUE);
2042 	}
2043 	if (tog_tree_view.panel == NULL) {
2044 		tog_tree_view.panel = new_panel(tog_tree_view.window);
2045 		if (tog_tree_view.panel == NULL)
2046 			return got_error_from_errno();
2047 	} else
2048 		show_panel(tog_tree_view.panel);
2049 
2050 	entries = got_object_tree_get_entries(root);
2051 	first_displayed_entry = SIMPLEQ_FIRST(&entries->head);
2052 	while (!done) {
2053 		char *parent_path;
2054 		entries = got_object_tree_get_entries(tree);
2055 		nentries = entries->nentries;
2056 		if (tree != root)
2057 			nentries++; /* '..' directory */
2058 
2059 		err = tree_entry_path(&parent_path, &parents, NULL);
2060 		if (err)
2061 			goto done;
2062 
2063 		err = draw_tree_entries(&first_displayed_entry,
2064 		    &last_displayed_entry, &selected_entry, &ndisplayed,
2065 		    tog_tree_view.window, tree_label, show_ids,
2066 		     parent_path, entries, selected, LINES, tree == root);
2067 		free(parent_path);
2068 		if (err)
2069 			break;
2070 
2071 		nodelay(stdscr, FALSE);
2072 		ch = wgetch(tog_tree_view.window);
2073 		nodelay(stdscr, TRUE);
2074 		switch (ch) {
2075 			case 'q':
2076 				done = 1;
2077 				break;
2078 			case 'i':
2079 				show_ids = !show_ids;
2080 				break;
2081 			case 'k':
2082 			case KEY_UP:
2083 				if (selected > 0)
2084 					selected--;
2085 				if (selected > 0)
2086 					break;
2087 				tree_scroll_up(&first_displayed_entry, 1,
2088 				    entries, tree == root);
2089 				break;
2090 			case KEY_PPAGE:
2091 				if (SIMPLEQ_FIRST(&entries->head) ==
2092 				    first_displayed_entry) {
2093 					if (tree != root)
2094 						first_displayed_entry = NULL;
2095 					selected = 0;
2096 					break;
2097 				}
2098 				tree_scroll_up(&first_displayed_entry, LINES,
2099 				    entries, tree == root);
2100 				break;
2101 			case 'j':
2102 			case KEY_DOWN:
2103 				if (selected < ndisplayed - 1) {
2104 					selected++;
2105 					break;
2106 				}
2107 				tree_scroll_down(&first_displayed_entry, 1,
2108 				    last_displayed_entry, entries);
2109 				break;
2110 			case KEY_NPAGE:
2111 				tree_scroll_down(&first_displayed_entry, LINES,
2112 				    last_displayed_entry, entries);
2113 				if (SIMPLEQ_NEXT(last_displayed_entry, entry))
2114 					break;
2115 				/* can't scroll any further; move cursor down */
2116 				if (selected < ndisplayed - 1)
2117 					selected = ndisplayed - 1;
2118 				break;
2119 			case KEY_ENTER:
2120 			case '\r':
2121 				if (selected_entry == NULL) {
2122 					struct tog_parent_tree *parent;
2123 			case KEY_BACKSPACE:
2124 					/* user selected '..' */
2125 					if (tree == root)
2126 						break;
2127 					parent = TAILQ_FIRST(&parents);
2128 					TAILQ_REMOVE(&parents, parent, entry);
2129 					got_object_tree_close(tree);
2130 					tree = parent->tree;
2131 					first_displayed_entry =
2132 					    parent->first_displayed_entry;
2133 					selected_entry = parent->selected_entry;
2134 					selected = parent->selected;
2135 					free(parent);
2136 				} else if (S_ISDIR(selected_entry->mode)) {
2137 					struct tog_parent_tree *parent;
2138 					struct got_tree_object *child;
2139 					err = got_object_open_as_tree(
2140 					    &child, repo, selected_entry->id);
2141 					if (err)
2142 						goto done;
2143 					parent = calloc(1, sizeof(*parent));
2144 					if (parent == NULL) {
2145 						err = got_error_from_errno();
2146 						goto done;
2147 					}
2148 					parent->tree = tree;
2149 					parent->first_displayed_entry =
2150 					   first_displayed_entry;
2151 					parent->selected_entry = selected_entry;
2152 					parent->selected = selected;
2153 					TAILQ_INSERT_HEAD(&parents, parent,
2154 					    entry);
2155 					tree = child;
2156 					selected = 0;
2157 					first_displayed_entry = NULL;
2158 				} else if (S_ISREG(selected_entry->mode)) {
2159 					err = blame_tree_entry(selected_entry,
2160 					    &parents, commit_id, repo);
2161 					if (err)
2162 						goto done;
2163 				}
2164 				break;
2165 			case KEY_RESIZE:
2166 				if (selected > LINES)
2167 					selected = ndisplayed - 1;
2168 				break;
2169 			default:
2170 				break;
2171 		}
2172 	}
2173 done:
2174 	free(tree_label);
2175 	free(commit_id_str);
2176 	while (!TAILQ_EMPTY(&parents)) {
2177 		struct tog_parent_tree *parent;
2178 		parent = TAILQ_FIRST(&parents);
2179 		TAILQ_REMOVE(&parents, parent, entry);
2180 		free(parent);
2181 
2182 	}
2183 	if (tree != root)
2184 		got_object_tree_close(tree);
2185 	return err;
2186 }
2187 
2188 __dead static void
2189 usage_tree(void)
2190 {
2191 	endwin();
2192 	fprintf(stderr, "usage: %s tree [-c commit] [repository-path]\n",
2193 	    getprogname());
2194 	exit(1);
2195 }
2196 
2197 static const struct got_error *
2198 cmd_tree(int argc, char *argv[])
2199 {
2200 	const struct got_error *error;
2201 	struct got_repository *repo = NULL;
2202 	char *repo_path = NULL;
2203 	struct got_object_id *commit_id = NULL;
2204 	char *commit_id_arg = NULL;
2205 	struct got_commit_object *commit = NULL;
2206 	struct got_tree_object *tree = NULL;
2207 	int ch;
2208 
2209 #ifndef PROFILE
2210 	if (pledge("stdio rpath wpath cpath flock proc tty", NULL) == -1)
2211 		err(1, "pledge");
2212 #endif
2213 
2214 	while ((ch = getopt(argc, argv, "c:")) != -1) {
2215 		switch (ch) {
2216 		case 'c':
2217 			commit_id_arg = optarg;
2218 			break;
2219 		default:
2220 			usage();
2221 			/* NOTREACHED */
2222 		}
2223 	}
2224 
2225 	argc -= optind;
2226 	argv += optind;
2227 
2228 	if (argc == 0) {
2229 		repo_path = getcwd(NULL, 0);
2230 		if (repo_path == NULL)
2231 			return got_error_from_errno();
2232 	} else if (argc == 1) {
2233 		repo_path = realpath(argv[0], NULL);
2234 		if (repo_path == NULL)
2235 			return got_error_from_errno();
2236 	} else
2237 		usage_log();
2238 
2239 	error = got_repo_open(&repo, repo_path);
2240 	free(repo_path);
2241 	if (error != NULL)
2242 		return error;
2243 
2244 	if (commit_id_arg == NULL) {
2245 		error = get_head_commit_id(&commit_id, repo);
2246 		if (error != NULL)
2247 			goto done;
2248 	} else {
2249 		struct got_object *obj;
2250 		error = got_object_open_by_id_str(&obj, repo, commit_id_arg);
2251 		if (error == NULL) {
2252 			commit_id = got_object_get_id(obj);
2253 			if (commit_id == NULL)
2254 				error = got_error_from_errno();
2255 		}
2256 	}
2257 	if (error != NULL)
2258 		goto done;
2259 
2260 	error = got_object_open_as_commit(&commit, repo, commit_id);
2261 	if (error != NULL)
2262 		goto done;
2263 
2264 	error = got_object_open_as_tree(&tree, repo, commit->tree_id);
2265 	if (error != NULL)
2266 		goto done;
2267 
2268 	error = show_tree_view(tree, commit_id, repo);
2269 done:
2270 	free(commit_id);
2271 	if (commit)
2272 		got_object_commit_close(commit);
2273 	if (tree)
2274 		got_object_tree_close(tree);
2275 	if (repo)
2276 		got_repo_close(repo);
2277 	return error;
2278 }
2279 static void
2280 init_curses(void)
2281 {
2282 	initscr();
2283 	cbreak();
2284 	noecho();
2285 	nonl();
2286 	intrflush(stdscr, FALSE);
2287 	keypad(stdscr, TRUE);
2288 	curs_set(0);
2289 }
2290 
2291 __dead static void
2292 usage(void)
2293 {
2294 	int i;
2295 
2296 	fprintf(stderr, "usage: %s [-h] [command] [arg ...]\n\n"
2297 	    "Available commands:\n", getprogname());
2298 	for (i = 0; i < nitems(tog_commands); i++) {
2299 		struct tog_cmd *cmd = &tog_commands[i];
2300 		fprintf(stderr, "    %s: %s\n", cmd->name, cmd->descr);
2301 	}
2302 	exit(1);
2303 }
2304 
2305 static char **
2306 make_argv(const char *arg0, const char *arg1)
2307 {
2308 	char **argv;
2309 	int argc = (arg1 == NULL ? 1 : 2);
2310 
2311 	argv = calloc(argc, sizeof(char *));
2312 	if (argv == NULL)
2313 		err(1, "calloc");
2314 	argv[0] = strdup(arg0);
2315 	if (argv[0] == NULL)
2316 		err(1, "calloc");
2317 	if (arg1) {
2318 		argv[1] = strdup(arg1);
2319 		if (argv[1] == NULL)
2320 			err(1, "calloc");
2321 	}
2322 
2323 	return argv;
2324 }
2325 
2326 int
2327 main(int argc, char *argv[])
2328 {
2329 	const struct got_error *error = NULL;
2330 	struct tog_cmd *cmd = NULL;
2331 	int ch, hflag = 0;
2332 	char **cmd_argv = NULL;
2333 
2334 	setlocale(LC_ALL, "");
2335 
2336 	while ((ch = getopt(argc, argv, "h")) != -1) {
2337 		switch (ch) {
2338 		case 'h':
2339 			hflag = 1;
2340 			break;
2341 		default:
2342 			usage();
2343 			/* NOTREACHED */
2344 		}
2345 	}
2346 
2347 	argc -= optind;
2348 	argv += optind;
2349 	optind = 0;
2350 	optreset = 1;
2351 
2352 	if (argc == 0) {
2353 		if (hflag)
2354 			usage();
2355 		/* Build an argument vector which runs a default command. */
2356 		cmd = &tog_commands[0];
2357 		cmd_argv = make_argv(cmd->name, NULL);
2358 		argc = 1;
2359 	} else {
2360 		int i;
2361 
2362 		/* Did the user specific a command? */
2363 		for (i = 0; i < nitems(tog_commands); i++) {
2364 			if (strncmp(tog_commands[i].name, argv[0],
2365 			    strlen(argv[0])) == 0) {
2366 				cmd = &tog_commands[i];
2367 				if (hflag)
2368 					tog_commands[i].cmd_usage();
2369 				break;
2370 			}
2371 		}
2372 		if (cmd == NULL) {
2373 			/* Did the user specify a repository? */
2374 			char *repo_path = realpath(argv[0], NULL);
2375 			if (repo_path) {
2376 				struct got_repository *repo;
2377 				error = got_repo_open(&repo, repo_path);
2378 				if (error == NULL)
2379 					got_repo_close(repo);
2380 			} else
2381 				error = got_error_from_errno();
2382 			if (error) {
2383 				if (hflag) {
2384 					fprintf(stderr, "%s: '%s' is not a "
2385 					    "known command\n", getprogname(),
2386 					    argv[0]);
2387 					usage();
2388 				}
2389 				fprintf(stderr, "%s: '%s' is neither a known "
2390 				    "command nor a path to a repository\n",
2391 				    getprogname(), argv[0]);
2392 				free(repo_path);
2393 				return 1;
2394 			}
2395 			cmd = &tog_commands[0];
2396 			cmd_argv = make_argv(cmd->name, repo_path);
2397 			argc = 2;
2398 			free(repo_path);
2399 		}
2400 	}
2401 
2402 	init_curses();
2403 
2404 	error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
2405 	if (error)
2406 		goto done;
2407 done:
2408 	endwin();
2409 	free(cmd_argv);
2410 	if (error)
2411 		fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
2412 	return 0;
2413 }