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