1 /*
2  * Copyright (c) 2018, 2019, 2020 Stefan Sperling <stsp@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <sys/queue.h>
18 #include <sys/stat.h>
19 #include <sys/ioctl.h>
20 
21 #include <ctype.h>
22 #include <errno.h>
23 #define _XOPEN_SOURCE_EXTENDED
24 #include <curses.h>
25 #undef _XOPEN_SOURCE_EXTENDED
26 #include <panel.h>
27 #include <locale.h>
28 #include <signal.h>
29 #include <stdlib.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <getopt.h>
33 #include <string.h>
34 #include <err.h>
35 #include <unistd.h>
36 #include <util.h>
37 #include <limits.h>
38 #include <wchar.h>
39 #include <time.h>
40 #include <pthread.h>
41 #include <libgen.h>
42 #include <regex.h>
43 
44 #include "got_version.h"
45 #include "got_error.h"
46 #include "got_object.h"
47 #include "got_reference.h"
48 #include "got_repository.h"
49 #include "got_diff.h"
50 #include "got_opentemp.h"
51 #include "got_utf8.h"
52 #include "got_cancel.h"
53 #include "got_commit_graph.h"
54 #include "got_blame.h"
55 #include "got_privsep.h"
56 #include "got_path.h"
57 #include "got_worktree.h"
58 
59 #ifndef MIN
60 #define	MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
61 #endif
62 
63 #ifndef MAX
64 #define	MAX(_a,_b) ((_a) > (_b) ? (_a) : (_b))
65 #endif
66 
67 #define CTRL(x)		((x) & 0x1f)
68 
69 #ifndef nitems
70 #define nitems(_a)	(sizeof((_a)) / sizeof((_a)[0]))
71 #endif
72 
73 struct tog_cmd {
74 	const char *name;
75 	const struct got_error *(*cmd_main)(int, char *[]);
76 	void (*cmd_usage)(void);
77 };
78 
79 __dead static void	usage(int, int);
80 __dead static void	usage_log(void);
81 __dead static void	usage_diff(void);
82 __dead static void	usage_blame(void);
83 __dead static void	usage_tree(void);
84 __dead static void	usage_ref(void);
85 
86 static const struct got_error*	cmd_log(int, char *[]);
87 static const struct got_error*	cmd_diff(int, char *[]);
88 static const struct got_error*	cmd_blame(int, char *[]);
89 static const struct got_error*	cmd_tree(int, char *[]);
90 static const struct got_error*	cmd_ref(int, char *[]);
91 
92 static struct tog_cmd tog_commands[] = {
93 	{ "log",	cmd_log,	usage_log },
94 	{ "diff",	cmd_diff,	usage_diff },
95 	{ "blame",	cmd_blame,	usage_blame },
96 	{ "tree",	cmd_tree,	usage_tree },
97 	{ "ref",	cmd_ref,	usage_ref },
98 };
99 
100 enum tog_view_type {
101 	TOG_VIEW_DIFF,
102 	TOG_VIEW_LOG,
103 	TOG_VIEW_BLAME,
104 	TOG_VIEW_TREE,
105 	TOG_VIEW_REF,
106 };
107 
108 #define TOG_EOF_STRING	"(END)"
109 
110 struct commit_queue_entry {
111 	TAILQ_ENTRY(commit_queue_entry) entry;
112 	struct got_object_id *id;
113 	struct got_commit_object *commit;
114 	int idx;
115 };
116 TAILQ_HEAD(commit_queue_head, commit_queue_entry);
117 struct commit_queue {
118 	int ncommits;
119 	struct commit_queue_head head;
120 };
121 
122 struct tog_color {
123 	SIMPLEQ_ENTRY(tog_color) entry;
124 	regex_t regex;
125 	short colorpair;
126 };
127 SIMPLEQ_HEAD(tog_colors, tog_color);
128 
129 static const struct got_error *
130 add_color(struct tog_colors *colors, const char *pattern,
131     int idx, short color)
132 {
133 	const struct got_error *err = NULL;
134 	struct tog_color *tc;
135 	int regerr = 0;
136 
137 	if (idx < 1 || idx > COLOR_PAIRS - 1)
138 		return NULL;
139 
140 	init_pair(idx, color, -1);
141 
142 	tc = calloc(1, sizeof(*tc));
143 	if (tc == NULL)
144 		return got_error_from_errno("calloc");
145 	regerr = regcomp(&tc->regex, pattern,
146 	    REG_EXTENDED | REG_NOSUB | REG_NEWLINE);
147 	if (regerr) {
148 		static char regerr_msg[512];
149 		static char err_msg[512];
150 		regerror(regerr, &tc->regex, regerr_msg,
151 		    sizeof(regerr_msg));
152 		snprintf(err_msg, sizeof(err_msg), "regcomp: %s",
153 		    regerr_msg);
154 		err = got_error_msg(GOT_ERR_REGEX, err_msg);
155 		free(tc);
156 		return err;
157 	}
158 	tc->colorpair = idx;
159 	SIMPLEQ_INSERT_HEAD(colors, tc, entry);
160 	return NULL;
161 }
162 
163 static void
164 free_colors(struct tog_colors *colors)
165 {
166 	struct tog_color *tc;
167 
168 	while (!SIMPLEQ_EMPTY(colors)) {
169 		tc = SIMPLEQ_FIRST(colors);
170 		SIMPLEQ_REMOVE_HEAD(colors, entry);
171 		regfree(&tc->regex);
172 		free(tc);
173 	}
174 }
175 
176 struct tog_color *
177 get_color(struct tog_colors *colors, int colorpair)
178 {
179 	struct tog_color *tc = NULL;
180 
181 	SIMPLEQ_FOREACH(tc, colors, entry) {
182 		if (tc->colorpair == colorpair)
183 			return tc;
184 	}
185 
186 	return NULL;
187 }
188 
189 static int
190 default_color_value(const char *envvar)
191 {
192 	if (strcmp(envvar, "TOG_COLOR_DIFF_MINUS") == 0)
193 		return COLOR_MAGENTA;
194 	if (strcmp(envvar, "TOG_COLOR_DIFF_PLUS") == 0)
195 		return COLOR_CYAN;
196 	if (strcmp(envvar, "TOG_COLOR_DIFF_CHUNK_HEADER") == 0)
197 		return COLOR_YELLOW;
198 	if (strcmp(envvar, "TOG_COLOR_DIFF_META") == 0)
199 		return COLOR_GREEN;
200 	if (strcmp(envvar, "TOG_COLOR_TREE_SUBMODULE") == 0)
201 		return COLOR_MAGENTA;
202 	if (strcmp(envvar, "TOG_COLOR_TREE_SYMLINK") == 0)
203 		return COLOR_MAGENTA;
204 	if (strcmp(envvar, "TOG_COLOR_TREE_DIRECTORY") == 0)
205 		return COLOR_CYAN;
206 	if (strcmp(envvar, "TOG_COLOR_TREE_EXECUTABLE") == 0)
207 		return COLOR_GREEN;
208 	if (strcmp(envvar, "TOG_COLOR_COMMIT") == 0)
209 		return COLOR_GREEN;
210 	if (strcmp(envvar, "TOG_COLOR_AUTHOR") == 0)
211 		return COLOR_CYAN;
212 	if (strcmp(envvar, "TOG_COLOR_DATE") == 0)
213 		return COLOR_YELLOW;
214 	if (strcmp(envvar, "TOG_COLOR_REFS_HEADS") == 0)
215 		return COLOR_GREEN;
216 	if (strcmp(envvar, "TOG_COLOR_REFS_TAGS") == 0)
217 		return COLOR_MAGENTA;
218 	if (strcmp(envvar, "TOG_COLOR_REFS_REMOTES") == 0)
219 		return COLOR_YELLOW;
220 
221 	return -1;
222 }
223 
224 static int
225 get_color_value(const char *envvar)
226 {
227 	const char *val = getenv(envvar);
228 
229 	if (val == NULL)
230 		return default_color_value(envvar);
231 
232 	if (strcasecmp(val, "black") == 0)
233 		return COLOR_BLACK;
234 	if (strcasecmp(val, "red") == 0)
235 		return COLOR_RED;
236 	if (strcasecmp(val, "green") == 0)
237 		return COLOR_GREEN;
238 	if (strcasecmp(val, "yellow") == 0)
239 		return COLOR_YELLOW;
240 	if (strcasecmp(val, "blue") == 0)
241 		return COLOR_BLUE;
242 	if (strcasecmp(val, "magenta") == 0)
243 		return COLOR_MAGENTA;
244 	if (strcasecmp(val, "cyan") == 0)
245 		return COLOR_CYAN;
246 	if (strcasecmp(val, "white") == 0)
247 		return COLOR_WHITE;
248 	if (strcasecmp(val, "default") == 0)
249 		return -1;
250 
251 	return default_color_value(envvar);
252 }
253 
254 
255 struct tog_diff_view_state {
256 	struct got_object_id *id1, *id2;
257 	const char *label1, *label2;
258 	FILE *f;
259 	int first_displayed_line;
260 	int last_displayed_line;
261 	int eof;
262 	int diff_context;
263 	int ignore_whitespace;
264 	int force_text_diff;
265 	struct got_repository *repo;
266 	struct got_reflist_head refs;
267 	struct tog_colors colors;
268 	size_t nlines;
269 	off_t *line_offsets;
270 	int matched_line;
271 	int selected_line;
272 
273 	/* passed from log view; may be NULL */
274 	struct tog_view *log_view;
275 };
276 
277 pthread_mutex_t tog_mutex = PTHREAD_MUTEX_INITIALIZER;
278 
279 struct tog_log_thread_args {
280 	pthread_cond_t need_commits;
281 	pthread_cond_t commit_loaded;
282 	int commits_needed;
283 	struct got_commit_graph *graph;
284 	struct commit_queue *commits;
285 	const char *in_repo_path;
286 	struct got_object_id *start_id;
287 	struct got_repository *repo;
288 	int log_complete;
289 	sig_atomic_t *quit;
290 	struct commit_queue_entry **first_displayed_entry;
291 	struct commit_queue_entry **selected_entry;
292 	int *searching;
293 	int *search_next_done;
294 	regex_t *regex;
295 };
296 
297 struct tog_log_view_state {
298 	struct commit_queue commits;
299 	struct commit_queue_entry *first_displayed_entry;
300 	struct commit_queue_entry *last_displayed_entry;
301 	struct commit_queue_entry *selected_entry;
302 	int selected;
303 	char *in_repo_path;
304 	const char *head_ref_name;
305 	int log_branches;
306 	struct got_repository *repo;
307 	struct got_reflist_head refs;
308 	struct got_object_id *start_id;
309 	sig_atomic_t quit;
310 	pthread_t thread;
311 	struct tog_log_thread_args thread_args;
312 	struct commit_queue_entry *matched_entry;
313 	struct commit_queue_entry *search_entry;
314 	struct tog_colors colors;
315 };
316 
317 #define TOG_COLOR_DIFF_MINUS		1
318 #define TOG_COLOR_DIFF_PLUS		2
319 #define TOG_COLOR_DIFF_CHUNK_HEADER	3
320 #define TOG_COLOR_DIFF_META		4
321 #define TOG_COLOR_TREE_SUBMODULE	5
322 #define TOG_COLOR_TREE_SYMLINK		6
323 #define TOG_COLOR_TREE_DIRECTORY	7
324 #define TOG_COLOR_TREE_EXECUTABLE	8
325 #define TOG_COLOR_COMMIT		9
326 #define TOG_COLOR_AUTHOR		10
327 #define TOG_COLOR_DATE		11
328 #define TOG_COLOR_REFS_HEADS		12
329 #define TOG_COLOR_REFS_TAGS		13
330 #define TOG_COLOR_REFS_REMOTES		14
331 
332 struct tog_blame_cb_args {
333 	struct tog_blame_line *lines; /* one per line */
334 	int nlines;
335 
336 	struct tog_view *view;
337 	struct got_object_id *commit_id;
338 	int *quit;
339 };
340 
341 struct tog_blame_thread_args {
342 	const char *path;
343 	struct got_repository *repo;
344 	struct tog_blame_cb_args *cb_args;
345 	int *complete;
346 	got_cancel_cb cancel_cb;
347 	void *cancel_arg;
348 };
349 
350 struct tog_blame {
351 	FILE *f;
352 	off_t filesize;
353 	struct tog_blame_line *lines;
354 	int nlines;
355 	off_t *line_offsets;
356 	pthread_t thread;
357 	struct tog_blame_thread_args thread_args;
358 	struct tog_blame_cb_args cb_args;
359 	const char *path;
360 };
361 
362 struct tog_blame_view_state {
363 	int first_displayed_line;
364 	int last_displayed_line;
365 	int selected_line;
366 	int blame_complete;
367 	int eof;
368 	int done;
369 	struct got_object_id_queue blamed_commits;
370 	struct got_object_qid *blamed_commit;
371 	char *path;
372 	struct got_repository *repo;
373 	struct got_object_id *commit_id;
374 	struct tog_blame blame;
375 	int matched_line;
376 	struct tog_colors colors;
377 };
378 
379 struct tog_parent_tree {
380 	TAILQ_ENTRY(tog_parent_tree) entry;
381 	struct got_tree_object *tree;
382 	struct got_tree_entry *first_displayed_entry;
383 	struct got_tree_entry *selected_entry;
384 	int selected;
385 };
386 
387 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
388 
389 struct tog_tree_view_state {
390 	char *tree_label;
391 	struct got_tree_object *root;
392 	struct got_tree_object *tree;
393 	struct got_tree_entry *first_displayed_entry;
394 	struct got_tree_entry *last_displayed_entry;
395 	struct got_tree_entry *selected_entry;
396 	int ndisplayed, selected, show_ids;
397 	struct tog_parent_trees parents;
398 	struct got_object_id *commit_id;
399 	struct got_repository *repo;
400 	struct got_tree_entry *matched_entry;
401 	struct tog_colors colors;
402 };
403 
404 struct tog_reflist_entry {
405 	TAILQ_ENTRY(tog_reflist_entry) entry;
406 	struct got_reference *ref;
407 	int idx;
408 };
409 
410 TAILQ_HEAD(tog_reflist_head, tog_reflist_entry);
411 
412 struct tog_ref_view_state {
413 	struct got_reflist_head simplerefs; /* SIMPLEQ */
414 	struct tog_reflist_head refs;	/* TAILQ */
415 	struct tog_reflist_entry *first_displayed_entry;
416 	struct tog_reflist_entry *last_displayed_entry;
417 	struct tog_reflist_entry *selected_entry;
418 	int nrefs, ndisplayed, selected, show_ids;
419 	struct got_repository *repo;
420 	struct tog_reflist_entry *matched_entry;
421 	struct tog_colors colors;
422 };
423 
424 /*
425  * We implement two types of views: parent views and child views.
426  *
427  * The 'Tab' key switches between a parent view and its child view.
428  * Child views are shown side-by-side to their parent view, provided
429  * there is enough screen estate.
430  *
431  * When a new view is opened from within a parent view, this new view
432  * becomes a child view of the parent view, replacing any existing child.
433  *
434  * When a new view is opened from within a child view, this new view
435  * becomes a parent view which will obscure the views below until the
436  * user quits the new parent view by typing 'q'.
437  *
438  * This list of views contains parent views only.
439  * Child views are only pointed to by their parent view.
440  */
441 TAILQ_HEAD(tog_view_list_head, tog_view);
442 
443 struct tog_view {
444 	TAILQ_ENTRY(tog_view) entry;
445 	WINDOW *window;
446 	PANEL *panel;
447 	int nlines, ncols, begin_y, begin_x;
448 	int lines, cols; /* copies of LINES and COLS */
449 	int focussed;
450 	struct tog_view *parent;
451 	struct tog_view *child;
452 	int child_focussed;
453 
454 	/* type-specific state */
455 	enum tog_view_type type;
456 	union {
457 		struct tog_diff_view_state diff;
458 		struct tog_log_view_state log;
459 		struct tog_blame_view_state blame;
460 		struct tog_tree_view_state tree;
461 		struct tog_ref_view_state ref;
462 	} state;
463 
464 	const struct got_error *(*show)(struct tog_view *);
465 	const struct got_error *(*input)(struct tog_view **,
466 	    struct tog_view **, struct tog_view**, struct tog_view *, int);
467 	const struct got_error *(*close)(struct tog_view *);
468 
469 	const struct got_error *(*search_start)(struct tog_view *);
470 	const struct got_error *(*search_next)(struct tog_view *);
471 	int searching;
472 #define TOG_SEARCH_FORWARD	1
473 #define TOG_SEARCH_BACKWARD	2
474 	int search_next_done;
475 #define TOG_SEARCH_HAVE_MORE	1
476 #define TOG_SEARCH_NO_MORE	2
477 #define TOG_SEARCH_HAVE_NONE	3
478 	regex_t regex;
479 	regmatch_t regmatch;
480 };
481 
482 static const struct got_error *open_diff_view(struct tog_view *,
483     struct got_object_id *, struct got_object_id *,
484     const char *, const char *, int, int, int, struct tog_view *,
485     struct got_repository *);
486 static const struct got_error *show_diff_view(struct tog_view *);
487 static const struct got_error *input_diff_view(struct tog_view **,
488     struct tog_view **, struct tog_view **, struct tog_view *, int);
489 static const struct got_error* close_diff_view(struct tog_view *);
490 static const struct got_error *search_start_diff_view(struct tog_view *);
491 static const struct got_error *search_next_diff_view(struct tog_view *);
492 
493 static const struct got_error *open_log_view(struct tog_view *,
494     struct got_object_id *, struct got_repository *,
495     const char *, const char *, int);
496 static const struct got_error * show_log_view(struct tog_view *);
497 static const struct got_error *input_log_view(struct tog_view **,
498     struct tog_view **, struct tog_view **, struct tog_view *, int);
499 static const struct got_error *close_log_view(struct tog_view *);
500 static const struct got_error *search_start_log_view(struct tog_view *);
501 static const struct got_error *search_next_log_view(struct tog_view *);
502 
503 static const struct got_error *open_blame_view(struct tog_view *, char *,
504     struct got_object_id *, struct got_repository *);
505 static const struct got_error *show_blame_view(struct tog_view *);
506 static const struct got_error *input_blame_view(struct tog_view **,
507     struct tog_view **, struct tog_view **, struct tog_view *, int);
508 static const struct got_error *close_blame_view(struct tog_view *);
509 static const struct got_error *search_start_blame_view(struct tog_view *);
510 static const struct got_error *search_next_blame_view(struct tog_view *);
511 
512 static const struct got_error *open_tree_view(struct tog_view *,
513     struct got_tree_object *, struct got_object_id *, struct got_repository *);
514 static const struct got_error *show_tree_view(struct tog_view *);
515 static const struct got_error *input_tree_view(struct tog_view **,
516     struct tog_view **, struct tog_view **, struct tog_view *, int);
517 static const struct got_error *close_tree_view(struct tog_view *);
518 static const struct got_error *search_start_tree_view(struct tog_view *);
519 static const struct got_error *search_next_tree_view(struct tog_view *);
520 
521 static const struct got_error *open_ref_view(struct tog_view *,
522     struct got_repository *);
523 static const struct got_error *show_ref_view(struct tog_view *);
524 static const struct got_error *input_ref_view(struct tog_view **,
525     struct tog_view **, struct tog_view **, struct tog_view *, int);
526 static const struct got_error *close_ref_view(struct tog_view *);
527 static const struct got_error *search_start_ref_view(struct tog_view *);
528 static const struct got_error *search_next_ref_view(struct tog_view *);
529 
530 static volatile sig_atomic_t tog_sigwinch_received;
531 static volatile sig_atomic_t tog_sigpipe_received;
532 static volatile sig_atomic_t tog_sigcont_received;
533 
534 static void
535 tog_sigwinch(int signo)
536 {
537 	tog_sigwinch_received = 1;
538 }
539 
540 static void
541 tog_sigpipe(int signo)
542 {
543 	tog_sigpipe_received = 1;
544 }
545 
546 static void
547 tog_sigcont(int signo)
548 {
549 	tog_sigcont_received = 1;
550 }
551 
552 static const struct got_error *
553 view_close(struct tog_view *view)
554 {
555 	const struct got_error *err = NULL;
556 
557 	if (view->child) {
558 		view_close(view->child);
559 		view->child = NULL;
560 	}
561 	if (view->close)
562 		err = view->close(view);
563 	if (view->panel)
564 		del_panel(view->panel);
565 	if (view->window)
566 		delwin(view->window);
567 	free(view);
568 	return err;
569 }
570 
571 static struct tog_view *
572 view_open(int nlines, int ncols, int begin_y, int begin_x,
573     enum tog_view_type type)
574 {
575 	struct tog_view *view = calloc(1, sizeof(*view));
576 
577 	if (view == NULL)
578 		return NULL;
579 
580 	view->type = type;
581 	view->lines = LINES;
582 	view->cols = COLS;
583 	view->nlines = nlines ? nlines : LINES - begin_y;
584 	view->ncols = ncols ? ncols : COLS - begin_x;
585 	view->begin_y = begin_y;
586 	view->begin_x = begin_x;
587 	view->window = newwin(nlines, ncols, begin_y, begin_x);
588 	if (view->window == NULL) {
589 		view_close(view);
590 		return NULL;
591 	}
592 	view->panel = new_panel(view->window);
593 	if (view->panel == NULL ||
594 	    set_panel_userptr(view->panel, view) != OK) {
595 		view_close(view);
596 		return NULL;
597 	}
598 
599 	keypad(view->window, TRUE);
600 	return view;
601 }
602 
603 static int
604 view_split_begin_x(int begin_x)
605 {
606 	if (begin_x > 0 || COLS < 120)
607 		return 0;
608 	return (COLS - MAX(COLS / 2, 80));
609 }
610 
611 static const struct got_error *view_resize(struct tog_view *);
612 
613 static const struct got_error *
614 view_splitscreen(struct tog_view *view)
615 {
616 	const struct got_error *err = NULL;
617 
618 	view->begin_y = 0;
619 	view->begin_x = view_split_begin_x(0);
620 	view->nlines = LINES;
621 	view->ncols = COLS - view->begin_x;
622 	view->lines = LINES;
623 	view->cols = COLS;
624 	err = view_resize(view);
625 	if (err)
626 		return err;
627 
628 	if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
629 		return got_error_from_errno("mvwin");
630 
631 	return NULL;
632 }
633 
634 static const struct got_error *
635 view_fullscreen(struct tog_view *view)
636 {
637 	const struct got_error *err = NULL;
638 
639 	view->begin_x = 0;
640 	view->begin_y = 0;
641 	view->nlines = LINES;
642 	view->ncols = COLS;
643 	view->lines = LINES;
644 	view->cols = COLS;
645 	err = view_resize(view);
646 	if (err)
647 		return err;
648 
649 	if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
650 		return got_error_from_errno("mvwin");
651 
652 	return NULL;
653 }
654 
655 static int
656 view_is_parent_view(struct tog_view *view)
657 {
658 	return view->parent == NULL;
659 }
660 
661 static const struct got_error *
662 view_resize(struct tog_view *view)
663 {
664 	int nlines, ncols;
665 
666 	if (view->lines > LINES)
667 		nlines = view->nlines - (view->lines - LINES);
668 	else
669 		nlines = view->nlines + (LINES - view->lines);
670 
671 	if (view->cols > COLS)
672 		ncols = view->ncols - (view->cols - COLS);
673 	else
674 		ncols = view->ncols + (COLS - view->cols);
675 
676 	if (wresize(view->window, nlines, ncols) == ERR)
677 		return got_error_from_errno("wresize");
678 	if (replace_panel(view->panel, view->window) == ERR)
679 		return got_error_from_errno("replace_panel");
680 	wclear(view->window);
681 
682 	view->nlines = nlines;
683 	view->ncols = ncols;
684 	view->lines = LINES;
685 	view->cols = COLS;
686 
687 	if (view->child) {
688 		view->child->begin_x = view_split_begin_x(view->begin_x);
689 		if (view->child->begin_x == 0) {
690 			view_fullscreen(view->child);
691 			if (view->child->focussed)
692 				show_panel(view->child->panel);
693 			else
694 				show_panel(view->panel);
695 		} else {
696 			view_splitscreen(view->child);
697 			show_panel(view->child->panel);
698 		}
699 	}
700 
701 	return NULL;
702 }
703 
704 static const struct got_error *
705 view_close_child(struct tog_view *view)
706 {
707 	const struct got_error *err = NULL;
708 
709 	if (view->child == NULL)
710 		return NULL;
711 
712 	err = view_close(view->child);
713 	view->child = NULL;
714 	return err;
715 }
716 
717 static const struct got_error *
718 view_set_child(struct tog_view *view, struct tog_view *child)
719 {
720 	const struct got_error *err = NULL;
721 
722 	view->child = child;
723 	child->parent = view;
724 	return err;
725 }
726 
727 static int
728 view_is_splitscreen(struct tog_view *view)
729 {
730 	return view->begin_x > 0;
731 }
732 
733 static void
734 tog_resizeterm(void)
735 {
736 	int cols, lines;
737 	struct winsize size;
738 
739 	if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &size) < 0) {
740 		cols = 80;     /* Default */
741 		lines = 24;
742 	} else {
743 		cols = size.ws_col;
744 		lines = size.ws_row;
745 	}
746 	resize_term(lines, cols);
747 }
748 
749 static const struct got_error *
750 view_search_start(struct tog_view *view)
751 {
752 	const struct got_error *err = NULL;
753 	char pattern[1024];
754 	int ret;
755 
756 	if (view->nlines < 1)
757 		return NULL;
758 
759 	mvwaddstr(view->window, view->begin_y + view->nlines - 1, 0, "/");
760 	wclrtoeol(view->window);
761 
762 	nocbreak();
763 	echo();
764 	ret = wgetnstr(view->window, pattern, sizeof(pattern));
765 	cbreak();
766 	noecho();
767 	if (ret == ERR)
768 		return NULL;
769 
770 	if (view->searching) {
771 		regfree(&view->regex);
772 		view->searching = 0;
773 	}
774 
775 	if (regcomp(&view->regex, pattern, REG_EXTENDED | REG_NEWLINE) == 0) {
776 		err = view->search_start(view);
777 		if (err) {
778 			regfree(&view->regex);
779 			return err;
780 		}
781 		view->searching = TOG_SEARCH_FORWARD;
782 		view->search_next_done = 0;
783 		view->search_next(view);
784 	}
785 
786 	return NULL;
787 }
788 
789 static const struct got_error *
790 view_input(struct tog_view **new, struct tog_view **dead,
791     struct tog_view **focus, int *done, struct tog_view *view,
792     struct tog_view_list_head *views)
793 {
794 	const struct got_error *err = NULL;
795 	struct tog_view *v;
796 	int ch, errcode;
797 
798 	*new = NULL;
799 	*dead = NULL;
800 	*focus = NULL;
801 
802 	/* Clear "no matches" indicator. */
803 	if (view->search_next_done == TOG_SEARCH_NO_MORE ||
804 	    view->search_next_done == TOG_SEARCH_HAVE_NONE)
805 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
806 
807 	if (view->searching && !view->search_next_done) {
808 		errcode = pthread_mutex_unlock(&tog_mutex);
809 		if (errcode)
810 			return got_error_set_errno(errcode,
811 			    "pthread_mutex_unlock");
812 		pthread_yield();
813 		errcode = pthread_mutex_lock(&tog_mutex);
814 		if (errcode)
815 			return got_error_set_errno(errcode,
816 			    "pthread_mutex_lock");
817 		view->search_next(view);
818 		return NULL;
819 	}
820 
821 	nodelay(stdscr, FALSE);
822 	/* Allow threads to make progress while we are waiting for input. */
823 	errcode = pthread_mutex_unlock(&tog_mutex);
824 	if (errcode)
825 		return got_error_set_errno(errcode, "pthread_mutex_unlock");
826 	ch = wgetch(view->window);
827 	errcode = pthread_mutex_lock(&tog_mutex);
828 	if (errcode)
829 		return got_error_set_errno(errcode, "pthread_mutex_lock");
830 	nodelay(stdscr, TRUE);
831 
832 	if (tog_sigwinch_received || tog_sigcont_received) {
833 		tog_resizeterm();
834 		tog_sigwinch_received = 0;
835 		tog_sigcont_received = 0;
836 		TAILQ_FOREACH(v, views, entry) {
837 			err = view_resize(v);
838 			if (err)
839 				return err;
840 			err = v->input(new, dead, focus, v, KEY_RESIZE);
841 			if (err)
842 				return err;
843 		}
844 	}
845 
846 	switch (ch) {
847 	case ERR:
848 		break;
849 	case '\t':
850 		if (view->child) {
851 			*focus = view->child;
852 			view->child_focussed = 1;
853 		} else if (view->parent) {
854 			*focus = view->parent;
855 			view->parent->child_focussed = 0;
856 		}
857 		break;
858 	case 'q':
859 		err = view->input(new, dead, focus, view, ch);
860 		*dead = view;
861 		break;
862 	case 'Q':
863 		*done = 1;
864 		break;
865 	case 'f':
866 		if (view_is_parent_view(view)) {
867 			if (view->child == NULL)
868 				break;
869 			if (view_is_splitscreen(view->child)) {
870 				*focus = view->child;
871 				view->child_focussed = 1;
872 				err = view_fullscreen(view->child);
873 			} else
874 				err = view_splitscreen(view->child);
875 			if (err)
876 				break;
877 			err = view->child->input(new, dead, focus,
878 			    view->child, KEY_RESIZE);
879 		} else {
880 			if (view_is_splitscreen(view)) {
881 				*focus = view;
882 				view->parent->child_focussed = 1;
883 				err = view_fullscreen(view);
884 			} else {
885 				err = view_splitscreen(view);
886 			}
887 			if (err)
888 				break;
889 			err = view->input(new, dead, focus, view,
890 			    KEY_RESIZE);
891 		}
892 		break;
893 	case KEY_RESIZE:
894 		break;
895 	case '/':
896 		if (view->search_start)
897 			view_search_start(view);
898 		else
899 			err = view->input(new, dead, focus, view, ch);
900 		break;
901 	case 'N':
902 	case 'n':
903 		if (view->search_next) {
904 			view->searching = (ch == 'n' ?
905 			    TOG_SEARCH_FORWARD : TOG_SEARCH_BACKWARD);
906 			view->search_next_done = 0;
907 			view->search_next(view);
908 		} else
909 			err = view->input(new, dead, focus, view, ch);
910 		break;
911 	default:
912 		err = view->input(new, dead, focus, view, ch);
913 		break;
914 	}
915 
916 	return err;
917 }
918 
919 void
920 view_vborder(struct tog_view *view)
921 {
922 	PANEL *panel;
923 	struct tog_view *view_above;
924 
925 	if (view->parent)
926 		return view_vborder(view->parent);
927 
928 	panel = panel_above(view->panel);
929 	if (panel == NULL)
930 		return;
931 
932 	view_above = panel_userptr(panel);
933 	mvwvline(view->window, view->begin_y, view_above->begin_x - 1,
934 	    got_locale_is_utf8() ? ACS_VLINE : '|', view->nlines);
935 }
936 
937 int
938 view_needs_focus_indication(struct tog_view *view)
939 {
940 	if (view_is_parent_view(view)) {
941 		if (view->child == NULL || view->child_focussed)
942 			return 0;
943 		if (!view_is_splitscreen(view->child))
944 			return 0;
945 	} else if (!view_is_splitscreen(view))
946 		return 0;
947 
948 	return view->focussed;
949 }
950 
951 static const struct got_error *
952 view_loop(struct tog_view *view)
953 {
954 	const struct got_error *err = NULL;
955 	struct tog_view_list_head views;
956 	struct tog_view *new_view, *dead_view, *focus_view, *main_view;
957 	int fast_refresh = 10;
958 	int done = 0, errcode;
959 
960 	errcode = pthread_mutex_lock(&tog_mutex);
961 	if (errcode)
962 		return got_error_set_errno(errcode, "pthread_mutex_lock");
963 
964 	TAILQ_INIT(&views);
965 	TAILQ_INSERT_HEAD(&views, view, entry);
966 
967 	main_view = view;
968 	view->focussed = 1;
969 	err = view->show(view);
970 	if (err)
971 		return err;
972 	update_panels();
973 	doupdate();
974 	while (!TAILQ_EMPTY(&views) && !done && !tog_sigpipe_received) {
975 		/* Refresh fast during initialization, then become slower. */
976 		if (fast_refresh && fast_refresh-- == 0)
977 			halfdelay(10); /* switch to once per second */
978 
979 		err = view_input(&new_view, &dead_view, &focus_view, &done,
980 		    view, &views);
981 		if (err)
982 			break;
983 		if (dead_view) {
984 			struct tog_view *prev = NULL;
985 
986 			if (view_is_parent_view(dead_view))
987 				prev = TAILQ_PREV(dead_view,
988 				    tog_view_list_head, entry);
989 			else if (view->parent != dead_view)
990 				prev = view->parent;
991 
992 			if (dead_view->parent)
993 				dead_view->parent->child = NULL;
994 			else
995 				TAILQ_REMOVE(&views, dead_view, entry);
996 
997 			err = view_close(dead_view);
998 			if (err || (dead_view == main_view && new_view == NULL))
999 				goto done;
1000 
1001 			if (view == dead_view) {
1002 				if (focus_view)
1003 					view = focus_view;
1004 				else if (prev)
1005 					view = prev;
1006 				else if (!TAILQ_EMPTY(&views))
1007 					view = TAILQ_LAST(&views,
1008 					    tog_view_list_head);
1009 				else
1010 					view = NULL;
1011 				if (view) {
1012 					if (view->child && view->child_focussed)
1013 						focus_view = view->child;
1014 					else
1015 						focus_view = view;
1016 				}
1017 			}
1018 		}
1019 		if (new_view) {
1020 			struct tog_view *v, *t;
1021 			/* Only allow one parent view per type. */
1022 			TAILQ_FOREACH_SAFE(v, &views, entry, t) {
1023 				if (v->type != new_view->type)
1024 					continue;
1025 				TAILQ_REMOVE(&views, v, entry);
1026 				err = view_close(v);
1027 				if (err)
1028 					goto done;
1029 				break;
1030 			}
1031 			TAILQ_INSERT_TAIL(&views, new_view, entry);
1032 			view = new_view;
1033 			if (focus_view == NULL)
1034 				focus_view = new_view;
1035 		}
1036 		if (focus_view) {
1037 			show_panel(focus_view->panel);
1038 			if (view)
1039 				view->focussed = 0;
1040 			focus_view->focussed = 1;
1041 			view = focus_view;
1042 			if (new_view)
1043 				show_panel(new_view->panel);
1044 			if (view->child && view_is_splitscreen(view->child))
1045 				show_panel(view->child->panel);
1046 		}
1047 		if (view) {
1048 			if (focus_view == NULL) {
1049 				view->focussed = 1;
1050 				show_panel(view->panel);
1051 				if (view->child && view_is_splitscreen(view->child))
1052 					show_panel(view->child->panel);
1053 				focus_view = view;
1054 			}
1055 			if (view->parent) {
1056 				err = view->parent->show(view->parent);
1057 				if (err)
1058 					goto done;
1059 			}
1060 			err = view->show(view);
1061 			if (err)
1062 				goto done;
1063 			if (view->child) {
1064 				err = view->child->show(view->child);
1065 				if (err)
1066 					goto done;
1067 			}
1068 			update_panels();
1069 			doupdate();
1070 		}
1071 	}
1072 done:
1073 	while (!TAILQ_EMPTY(&views)) {
1074 		view = TAILQ_FIRST(&views);
1075 		TAILQ_REMOVE(&views, view, entry);
1076 		view_close(view);
1077 	}
1078 
1079 	errcode = pthread_mutex_unlock(&tog_mutex);
1080 	if (errcode && err == NULL)
1081 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
1082 
1083 	return err;
1084 }
1085 
1086 __dead static void
1087 usage_log(void)
1088 {
1089 	endwin();
1090 	fprintf(stderr,
1091 	    "usage: %s log [-b] [-c commit] [-r repository-path] [path]\n",
1092 	    getprogname());
1093 	exit(1);
1094 }
1095 
1096 /* Create newly allocated wide-character string equivalent to a byte string. */
1097 static const struct got_error *
1098 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
1099 {
1100 	char *vis = NULL;
1101 	const struct got_error *err = NULL;
1102 
1103 	*ws = NULL;
1104 	*wlen = mbstowcs(NULL, s, 0);
1105 	if (*wlen == (size_t)-1) {
1106 		int vislen;
1107 		if (errno != EILSEQ)
1108 			return got_error_from_errno("mbstowcs");
1109 
1110 		/* byte string invalid in current encoding; try to "fix" it */
1111 		err = got_mbsavis(&vis, &vislen, s);
1112 		if (err)
1113 			return err;
1114 		*wlen = mbstowcs(NULL, vis, 0);
1115 		if (*wlen == (size_t)-1) {
1116 			err = got_error_from_errno("mbstowcs"); /* give up */
1117 			goto done;
1118 		}
1119 	}
1120 
1121 	*ws = calloc(*wlen + 1, sizeof(**ws));
1122 	if (*ws == NULL) {
1123 		err = got_error_from_errno("calloc");
1124 		goto done;
1125 	}
1126 
1127 	if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
1128 		err = got_error_from_errno("mbstowcs");
1129 done:
1130 	free(vis);
1131 	if (err) {
1132 		free(*ws);
1133 		*ws = NULL;
1134 		*wlen = 0;
1135 	}
1136 	return err;
1137 }
1138 
1139 /* Format a line for display, ensuring that it won't overflow a width limit. */
1140 static const struct got_error *
1141 format_line(wchar_t **wlinep, int *widthp, const char *line, int wlimit,
1142     int col_tab_align)
1143 {
1144 	const struct got_error *err = NULL;
1145 	int cols = 0;
1146 	wchar_t *wline = NULL;
1147 	size_t wlen;
1148 	int i;
1149 
1150 	*wlinep = NULL;
1151 	*widthp = 0;
1152 
1153 	err = mbs2ws(&wline, &wlen, line);
1154 	if (err)
1155 		return err;
1156 
1157 	i = 0;
1158 	while (i < wlen) {
1159 		int width = wcwidth(wline[i]);
1160 
1161 		if (width == 0) {
1162 			i++;
1163 			continue;
1164 		}
1165 
1166 		if (width == 1 || width == 2) {
1167 			if (cols + width > wlimit)
1168 				break;
1169 			cols += width;
1170 			i++;
1171 		} else if (width == -1) {
1172 			if (wline[i] == L'\t') {
1173 				width = TABSIZE -
1174 				    ((cols + col_tab_align) % TABSIZE);
1175 				if (cols + width > wlimit)
1176 					break;
1177 				cols += width;
1178 			}
1179 			i++;
1180 		} else {
1181 			err = got_error_from_errno("wcwidth");
1182 			goto done;
1183 		}
1184 	}
1185 	wline[i] = L'\0';
1186 	if (widthp)
1187 		*widthp = cols;
1188 done:
1189 	if (err)
1190 		free(wline);
1191 	else
1192 		*wlinep = wline;
1193 	return err;
1194 }
1195 
1196 static const struct got_error*
1197 build_refs_str(char **refs_str, struct got_reflist_head *refs,
1198     struct got_object_id *id, struct got_repository *repo)
1199 {
1200 	static const struct got_error *err = NULL;
1201 	struct got_reflist_entry *re;
1202 	char *s;
1203 	const char *name;
1204 
1205 	*refs_str = NULL;
1206 
1207 	SIMPLEQ_FOREACH(re, refs, entry) {
1208 		struct got_tag_object *tag = NULL;
1209 		struct got_object_id *ref_id;
1210 		int cmp;
1211 
1212 		name = got_ref_get_name(re->ref);
1213 		if (strcmp(name, GOT_REF_HEAD) == 0)
1214 			continue;
1215 		if (strncmp(name, "refs/", 5) == 0)
1216 			name += 5;
1217 		if (strncmp(name, "got/", 4) == 0)
1218 			continue;
1219 		if (strncmp(name, "heads/", 6) == 0)
1220 			name += 6;
1221 		if (strncmp(name, "remotes/", 8) == 0) {
1222 			name += 8;
1223 			s = strstr(name, "/" GOT_REF_HEAD);
1224 			if (s != NULL && s[strlen(s)] == '\0')
1225 				continue;
1226 		}
1227 		err = got_ref_resolve(&ref_id, repo, re->ref);
1228 		if (err)
1229 			break;
1230 		if (strncmp(name, "tags/", 5) == 0) {
1231 			err = got_object_open_as_tag(&tag, repo, ref_id);
1232 			if (err) {
1233 				if (err->code != GOT_ERR_OBJ_TYPE) {
1234 					free(ref_id);
1235 					break;
1236 				}
1237 				/* Ref points at something other than a tag. */
1238 				err = NULL;
1239 				tag = NULL;
1240 			}
1241 		}
1242 		cmp = got_object_id_cmp(tag ?
1243 		    got_object_tag_get_object_id(tag) : ref_id, id);
1244 		free(ref_id);
1245 		if (tag)
1246 			got_object_tag_close(tag);
1247 		if (cmp != 0)
1248 			continue;
1249 		s = *refs_str;
1250 		if (asprintf(refs_str, "%s%s%s", s ? s : "",
1251 		    s ? ", " : "", name) == -1) {
1252 			err = got_error_from_errno("asprintf");
1253 			free(s);
1254 			*refs_str = NULL;
1255 			break;
1256 		}
1257 		free(s);
1258 	}
1259 
1260 	return err;
1261 }
1262 
1263 static const struct got_error *
1264 format_author(wchar_t **wauthor, int *author_width, char *author, int limit,
1265     int col_tab_align)
1266 {
1267 	char *smallerthan, *at;
1268 
1269 	smallerthan = strchr(author, '<');
1270 	if (smallerthan && smallerthan[1] != '\0')
1271 		author = smallerthan + 1;
1272 	at = strchr(author, '@');
1273 	if (at)
1274 		*at = '\0';
1275 	return format_line(wauthor, author_width, author, limit, col_tab_align);
1276 }
1277 
1278 static const struct got_error *
1279 draw_commit(struct tog_view *view, struct got_commit_object *commit,
1280     struct got_object_id *id, struct got_reflist_head *refs,
1281     const size_t date_display_cols, int author_display_cols,
1282     struct tog_colors *colors)
1283 {
1284 	const struct got_error *err = NULL;
1285 	char datebuf[12]; /* YYYY-MM-DD + SPACE + NUL */
1286 	char *logmsg0 = NULL, *logmsg = NULL;
1287 	char *author = NULL;
1288 	wchar_t *wlogmsg = NULL, *wauthor = NULL;
1289 	int author_width, logmsg_width;
1290 	char *newline, *line = NULL;
1291 	int col, limit;
1292 	const int avail = view->ncols;
1293 	struct tm tm;
1294 	time_t committer_time;
1295 	struct tog_color *tc;
1296 
1297 	committer_time = got_object_commit_get_committer_time(commit);
1298 	if (localtime_r(&committer_time, &tm) == NULL)
1299 		return got_error_from_errno("localtime_r");
1300 	if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d ", &tm)
1301 	    >= sizeof(datebuf))
1302 		return got_error(GOT_ERR_NO_SPACE);
1303 
1304 	if (avail <= date_display_cols)
1305 		limit = MIN(sizeof(datebuf) - 1, avail);
1306 	else
1307 		limit = MIN(date_display_cols, sizeof(datebuf) - 1);
1308 	tc = get_color(colors, TOG_COLOR_DATE);
1309 	if (tc)
1310 		wattr_on(view->window,
1311 		    COLOR_PAIR(tc->colorpair), NULL);
1312 	waddnstr(view->window, datebuf, limit);
1313 	if (tc)
1314 		wattr_off(view->window,
1315 		    COLOR_PAIR(tc->colorpair), NULL);
1316 	col = limit;
1317 	if (col > avail)
1318 		goto done;
1319 
1320 	if (avail >= 120) {
1321 		char *id_str;
1322 		err = got_object_id_str(&id_str, id);
1323 		if (err)
1324 			goto done;
1325 		tc = get_color(colors, TOG_COLOR_COMMIT);
1326 		if (tc)
1327 			wattr_on(view->window,
1328 			    COLOR_PAIR(tc->colorpair), NULL);
1329 		wprintw(view->window, "%.8s ", id_str);
1330 		if (tc)
1331 			wattr_off(view->window,
1332 			    COLOR_PAIR(tc->colorpair), NULL);
1333 		free(id_str);
1334 		col += 9;
1335 		if (col > avail)
1336 			goto done;
1337 	}
1338 
1339 	author = strdup(got_object_commit_get_author(commit));
1340 	if (author == NULL) {
1341 		err = got_error_from_errno("strdup");
1342 		goto done;
1343 	}
1344 	err = format_author(&wauthor, &author_width, author, avail - col, col);
1345 	if (err)
1346 		goto done;
1347 	tc = get_color(colors, TOG_COLOR_AUTHOR);
1348 	if (tc)
1349 		wattr_on(view->window,
1350 		    COLOR_PAIR(tc->colorpair), NULL);
1351 	waddwstr(view->window, wauthor);
1352 	if (tc)
1353 		wattr_off(view->window,
1354 		    COLOR_PAIR(tc->colorpair), NULL);
1355 	col += author_width;
1356 	while (col < avail && author_width < author_display_cols + 2) {
1357 		waddch(view->window, ' ');
1358 		col++;
1359 		author_width++;
1360 	}
1361 	if (col > avail)
1362 		goto done;
1363 
1364 	err = got_object_commit_get_logmsg(&logmsg0, commit);
1365 	if (err)
1366 		goto done;
1367 	logmsg = logmsg0;
1368 	while (*logmsg == '\n')
1369 		logmsg++;
1370 	newline = strchr(logmsg, '\n');
1371 	if (newline)
1372 		*newline = '\0';
1373 	limit = avail - col;
1374 	err = format_line(&wlogmsg, &logmsg_width, logmsg, limit, col);
1375 	if (err)
1376 		goto done;
1377 	waddwstr(view->window, wlogmsg);
1378 	col += logmsg_width;
1379 	while (col < avail) {
1380 		waddch(view->window, ' ');
1381 		col++;
1382 	}
1383 done:
1384 	free(logmsg0);
1385 	free(wlogmsg);
1386 	free(author);
1387 	free(wauthor);
1388 	free(line);
1389 	return err;
1390 }
1391 
1392 static struct commit_queue_entry *
1393 alloc_commit_queue_entry(struct got_commit_object *commit,
1394     struct got_object_id *id)
1395 {
1396 	struct commit_queue_entry *entry;
1397 
1398 	entry = calloc(1, sizeof(*entry));
1399 	if (entry == NULL)
1400 		return NULL;
1401 
1402 	entry->id = id;
1403 	entry->commit = commit;
1404 	return entry;
1405 }
1406 
1407 static void
1408 pop_commit(struct commit_queue *commits)
1409 {
1410 	struct commit_queue_entry *entry;
1411 
1412 	entry = TAILQ_FIRST(&commits->head);
1413 	TAILQ_REMOVE(&commits->head, entry, entry);
1414 	got_object_commit_close(entry->commit);
1415 	commits->ncommits--;
1416 	/* Don't free entry->id! It is owned by the commit graph. */
1417 	free(entry);
1418 }
1419 
1420 static void
1421 free_commits(struct commit_queue *commits)
1422 {
1423 	while (!TAILQ_EMPTY(&commits->head))
1424 		pop_commit(commits);
1425 }
1426 
1427 static const struct got_error *
1428 match_commit(int *have_match, struct got_object_id *id,
1429     struct got_commit_object *commit, regex_t *regex)
1430 {
1431 	const struct got_error *err = NULL;
1432 	regmatch_t regmatch;
1433 	char *id_str = NULL, *logmsg = NULL;
1434 
1435 	*have_match = 0;
1436 
1437 	err = got_object_id_str(&id_str, id);
1438 	if (err)
1439 		return err;
1440 
1441 	err = got_object_commit_get_logmsg(&logmsg, commit);
1442 	if (err)
1443 		goto done;
1444 
1445 	if (regexec(regex, got_object_commit_get_author(commit), 1,
1446 	    &regmatch, 0) == 0 ||
1447 	    regexec(regex, got_object_commit_get_committer(commit), 1,
1448 	    &regmatch, 0) == 0 ||
1449 	    regexec(regex, id_str, 1, &regmatch, 0) == 0 ||
1450 	    regexec(regex, logmsg, 1, &regmatch, 0) == 0)
1451 		*have_match = 1;
1452 done:
1453 	free(id_str);
1454 	free(logmsg);
1455 	return err;
1456 }
1457 
1458 static const struct got_error *
1459 queue_commits(struct got_commit_graph *graph, struct commit_queue *commits,
1460     int minqueue, struct got_repository *repo, const char *path,
1461     int *searching, int *search_next_done, regex_t *regex)
1462 {
1463 	const struct got_error *err = NULL;
1464 	int nqueued = 0;
1465 
1466 	/*
1467 	 * We keep all commits open throughout the lifetime of the log
1468 	 * view in order to avoid having to re-fetch commits from disk
1469 	 * while updating the display.
1470 	 */
1471 	while (nqueued < minqueue ||
1472 	    (*searching == TOG_SEARCH_FORWARD && !*search_next_done)) {
1473 		struct got_object_id *id;
1474 		struct got_commit_object *commit;
1475 		struct commit_queue_entry *entry;
1476 		int errcode;
1477 
1478 		err = got_commit_graph_iter_next(&id, graph, repo, NULL, NULL);
1479 		if (err || id == NULL)
1480 			break;
1481 
1482 		err = got_object_open_as_commit(&commit, repo, id);
1483 		if (err)
1484 			break;
1485 		entry = alloc_commit_queue_entry(commit, id);
1486 		if (entry == NULL) {
1487 			err = got_error_from_errno("alloc_commit_queue_entry");
1488 			break;
1489 		}
1490 
1491 		errcode = pthread_mutex_lock(&tog_mutex);
1492 		if (errcode) {
1493 			err = got_error_set_errno(errcode,
1494 			    "pthread_mutex_lock");
1495 			break;
1496 		}
1497 
1498 		entry->idx = commits->ncommits;
1499 		TAILQ_INSERT_TAIL(&commits->head, entry, entry);
1500 		nqueued++;
1501 		commits->ncommits++;
1502 
1503 		if (*searching == TOG_SEARCH_FORWARD && !*search_next_done) {
1504 			int have_match;
1505 			err = match_commit(&have_match, id, commit, regex);
1506 			if (err)
1507 				break;
1508 			if (have_match)
1509 				*search_next_done = TOG_SEARCH_HAVE_MORE;
1510 		}
1511 
1512 		errcode = pthread_mutex_unlock(&tog_mutex);
1513 		if (errcode && err == NULL)
1514 			err = got_error_set_errno(errcode,
1515 			    "pthread_mutex_unlock");
1516 		if (err)
1517 			break;
1518 	}
1519 
1520 	return err;
1521 }
1522 
1523 static const struct got_error *
1524 draw_commits(struct tog_view *view, struct commit_queue_entry **last,
1525     struct commit_queue_entry **selected, struct commit_queue_entry *first,
1526     struct commit_queue *commits, int selected_idx, int limit,
1527     struct got_reflist_head *refs, const char *path, int commits_needed,
1528     struct tog_colors *colors)
1529 {
1530 	const struct got_error *err = NULL;
1531 	struct tog_log_view_state *s = &view->state.log;
1532 	struct commit_queue_entry *entry;
1533 	int width;
1534 	int ncommits, author_cols = 4;
1535 	char *id_str = NULL, *header = NULL, *ncommits_str = NULL;
1536 	char *refs_str = NULL;
1537 	wchar_t *wline;
1538 	struct tog_color *tc;
1539 	static const size_t date_display_cols = 12;
1540 
1541 	entry = first;
1542 	ncommits = 0;
1543 	while (entry) {
1544 		if (ncommits == selected_idx) {
1545 			*selected = entry;
1546 			break;
1547 		}
1548 		entry = TAILQ_NEXT(entry, entry);
1549 		ncommits++;
1550 	}
1551 
1552 	if (*selected && !(view->searching && view->search_next_done == 0)) {
1553 		err = got_object_id_str(&id_str, (*selected)->id);
1554 		if (err)
1555 			return err;
1556 		if (refs) {
1557 			err = build_refs_str(&refs_str, refs, (*selected)->id,
1558 			    s->repo);
1559 			if (err)
1560 				goto done;
1561 		}
1562 	}
1563 
1564 	if (commits_needed == 0)
1565 		halfdelay(10); /* disable fast refresh */
1566 
1567 	if (commits_needed > 0) {
1568 		if (asprintf(&ncommits_str, " [%d/%d] %s",
1569 		    entry ? entry->idx + 1 : 0, commits->ncommits,
1570 		    (view->searching && !view->search_next_done) ?
1571 		    "searching..." : "loading...") == -1) {
1572 			err = got_error_from_errno("asprintf");
1573 			goto done;
1574 		}
1575 	} else {
1576 		const char *search_str = NULL;
1577 
1578 		if (view->searching) {
1579 			if (view->search_next_done == TOG_SEARCH_NO_MORE)
1580 				search_str = "no more matches";
1581 			else if (view->search_next_done == TOG_SEARCH_HAVE_NONE)
1582 				search_str = "no matches found";
1583 			else if (!view->search_next_done)
1584 				search_str = "searching...";
1585 		}
1586 
1587 		if (asprintf(&ncommits_str, " [%d/%d] %s",
1588 		    entry ? entry->idx + 1 : 0, commits->ncommits,
1589 		    search_str ? search_str :
1590 		    (refs_str ? refs_str : "")) == -1) {
1591 			err = got_error_from_errno("asprintf");
1592 			goto done;
1593 		}
1594 	}
1595 
1596 	if (path && strcmp(path, "/") != 0) {
1597 		if (asprintf(&header, "commit %s %s%s",
1598 		    id_str ? id_str : "........................................",
1599 		    path, ncommits_str) == -1) {
1600 			err = got_error_from_errno("asprintf");
1601 			header = NULL;
1602 			goto done;
1603 		}
1604 	} else if (asprintf(&header, "commit %s%s",
1605 	    id_str ? id_str : "........................................",
1606 	    ncommits_str) == -1) {
1607 		err = got_error_from_errno("asprintf");
1608 		header = NULL;
1609 		goto done;
1610 	}
1611 	err = format_line(&wline, &width, header, view->ncols, 0);
1612 	if (err)
1613 		goto done;
1614 
1615 	werase(view->window);
1616 
1617 	if (view_needs_focus_indication(view))
1618 		wstandout(view->window);
1619 	tc = get_color(colors, TOG_COLOR_COMMIT);
1620 	if (tc)
1621 		wattr_on(view->window,
1622 		    COLOR_PAIR(tc->colorpair), NULL);
1623 	waddwstr(view->window, wline);
1624 	if (tc)
1625 		wattr_off(view->window,
1626 		    COLOR_PAIR(tc->colorpair), NULL);
1627 	while (width < view->ncols) {
1628 		waddch(view->window, ' ');
1629 		width++;
1630 	}
1631 	if (view_needs_focus_indication(view))
1632 		wstandend(view->window);
1633 	free(wline);
1634 	if (limit <= 1)
1635 		goto done;
1636 
1637 	/* Grow author column size if necessary. */
1638 	entry = first;
1639 	ncommits = 0;
1640 	while (entry) {
1641 		char *author;
1642 		wchar_t *wauthor;
1643 		int width;
1644 		if (ncommits >= limit - 1)
1645 			break;
1646 		author = strdup(got_object_commit_get_author(entry->commit));
1647 		if (author == NULL) {
1648 			err = got_error_from_errno("strdup");
1649 			goto done;
1650 		}
1651 		err = format_author(&wauthor, &width, author, COLS,
1652 		    date_display_cols);
1653 		if (author_cols < width)
1654 			author_cols = width;
1655 		free(wauthor);
1656 		free(author);
1657 		ncommits++;
1658 		entry = TAILQ_NEXT(entry, entry);
1659 	}
1660 
1661 	entry = first;
1662 	*last = first;
1663 	ncommits = 0;
1664 	while (entry) {
1665 		if (ncommits >= limit - 1)
1666 			break;
1667 		if (ncommits == selected_idx)
1668 			wstandout(view->window);
1669 		err = draw_commit(view, entry->commit, entry->id, refs,
1670 		    date_display_cols, author_cols, colors);
1671 		if (ncommits == selected_idx)
1672 			wstandend(view->window);
1673 		if (err)
1674 			goto done;
1675 		ncommits++;
1676 		*last = entry;
1677 		entry = TAILQ_NEXT(entry, entry);
1678 	}
1679 
1680 	view_vborder(view);
1681 done:
1682 	free(id_str);
1683 	free(refs_str);
1684 	free(ncommits_str);
1685 	free(header);
1686 	return err;
1687 }
1688 
1689 static void
1690 scroll_up(struct tog_view *view,
1691     struct commit_queue_entry **first_displayed_entry, int maxscroll,
1692     struct commit_queue *commits)
1693 {
1694 	struct commit_queue_entry *entry;
1695 	int nscrolled = 0;
1696 
1697 	entry = TAILQ_FIRST(&commits->head);
1698 	if (*first_displayed_entry == entry)
1699 		return;
1700 
1701 	entry = *first_displayed_entry;
1702 	while (entry && nscrolled < maxscroll) {
1703 		entry = TAILQ_PREV(entry, commit_queue_head, entry);
1704 		if (entry) {
1705 			*first_displayed_entry = entry;
1706 			nscrolled++;
1707 		}
1708 	}
1709 }
1710 
1711 static const struct got_error *
1712 trigger_log_thread(struct tog_view *log_view, int wait,
1713     int *commits_needed, int *log_complete,
1714     pthread_cond_t *need_commits, pthread_cond_t *commit_loaded)
1715 {
1716 	int errcode;
1717 
1718 	halfdelay(1); /* fast refresh while loading commits */
1719 
1720 	while (*commits_needed > 0) {
1721 		if (*log_complete)
1722 			break;
1723 
1724 		/* Wake the log thread. */
1725 		errcode = pthread_cond_signal(need_commits);
1726 		if (errcode)
1727 			return got_error_set_errno(errcode,
1728 			    "pthread_cond_signal");
1729 
1730 		/*
1731 		 * The mutex will be released while the view loop waits
1732 		 * in wgetch(), at which time the log thread will run.
1733 		 */
1734 		if (!wait)
1735 			break;
1736 
1737 		/* Display progress update in log view. */
1738 		show_log_view(log_view);
1739 		update_panels();
1740 		doupdate();
1741 
1742 		/* Wait right here while next commit is being loaded. */
1743 		errcode = pthread_cond_wait(commit_loaded, &tog_mutex);
1744 		if (errcode)
1745 			return got_error_set_errno(errcode,
1746 			    "pthread_cond_wait");
1747 
1748 		/* Display progress update in log view. */
1749 		show_log_view(log_view);
1750 		update_panels();
1751 		doupdate();
1752 	}
1753 
1754 	return NULL;
1755 }
1756 
1757 static const struct got_error *
1758 scroll_down(struct tog_view *log_view,
1759     struct commit_queue_entry **first_displayed_entry, int maxscroll,
1760     struct commit_queue_entry **last_displayed_entry,
1761     struct commit_queue *commits, int *log_complete, int *commits_needed,
1762     pthread_cond_t *need_commits, pthread_cond_t *commit_loaded)
1763 {
1764 	const struct got_error *err = NULL;
1765 	struct commit_queue_entry *pentry;
1766 	int nscrolled = 0, ncommits_needed;
1767 
1768 	if (*last_displayed_entry == NULL)
1769 		return NULL;
1770 
1771 	ncommits_needed = (*last_displayed_entry)->idx + 1 + maxscroll;
1772 	if (commits->ncommits < ncommits_needed && !*log_complete) {
1773 		/*
1774 		 * Ask the log thread for required amount of commits.
1775 		 */
1776 		(*commits_needed) += maxscroll;
1777 		err = trigger_log_thread(log_view, 1, commits_needed,
1778 		    log_complete, need_commits, commit_loaded);
1779 		if (err)
1780 			return err;
1781 	}
1782 
1783 	do {
1784 		pentry = TAILQ_NEXT(*last_displayed_entry, entry);
1785 		if (pentry == NULL)
1786 			break;
1787 
1788 		*last_displayed_entry = pentry;
1789 
1790 		pentry = TAILQ_NEXT(*first_displayed_entry, entry);
1791 		if (pentry == NULL)
1792 			break;
1793 		*first_displayed_entry = pentry;
1794 	} while (++nscrolled < maxscroll);
1795 
1796 	return err;
1797 }
1798 
1799 static const struct got_error *
1800 open_diff_view_for_commit(struct tog_view **new_view, int begin_x,
1801     struct got_commit_object *commit, struct got_object_id *commit_id,
1802     struct tog_view *log_view, struct got_repository *repo)
1803 {
1804 	const struct got_error *err;
1805 	struct got_object_qid *parent_id;
1806 	struct tog_view *diff_view;
1807 
1808 	diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
1809 	if (diff_view == NULL)
1810 		return got_error_from_errno("view_open");
1811 
1812 	parent_id = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
1813 	err = open_diff_view(diff_view, parent_id ? parent_id->id : NULL,
1814 	    commit_id, NULL, NULL, 3, 0, 0, log_view, repo);
1815 	if (err == NULL)
1816 		*new_view = diff_view;
1817 	return err;
1818 }
1819 
1820 static const struct got_error *
1821 tree_view_visit_subtree(struct got_tree_object *subtree,
1822     struct tog_tree_view_state *s)
1823 {
1824 	struct tog_parent_tree *parent;
1825 
1826 	parent = calloc(1, sizeof(*parent));
1827 	if (parent == NULL)
1828 		return got_error_from_errno("calloc");
1829 
1830 	parent->tree = s->tree;
1831 	parent->first_displayed_entry = s->first_displayed_entry;
1832 	parent->selected_entry = s->selected_entry;
1833 	parent->selected = s->selected;
1834 	TAILQ_INSERT_HEAD(&s->parents, parent, entry);
1835 	s->tree = subtree;
1836 	s->selected = 0;
1837 	s->first_displayed_entry = NULL;
1838 	return NULL;
1839 }
1840 
1841 static const struct got_error *
1842 tree_view_walk_path(struct tog_tree_view_state *s,
1843     struct got_object_id *commit_id,
1844     const char *path, struct got_repository *repo)
1845 {
1846 	const struct got_error *err = NULL;
1847 	struct got_tree_object *tree = NULL;
1848 	const char *p;
1849 	char *slash, *subpath = NULL;
1850 
1851 	/* Walk the path and open corresponding tree objects. */
1852 	p = path;
1853 	while (*p) {
1854 		struct got_tree_entry *te;
1855 		struct got_object_id *tree_id;
1856 		char *te_name;
1857 
1858 		while (p[0] == '/')
1859 			p++;
1860 
1861 		/* Ensure the correct subtree entry is selected. */
1862 		slash = strchr(p, '/');
1863 		if (slash == NULL)
1864 			te_name = strdup(p);
1865 		else
1866 			te_name = strndup(p, slash - p);
1867 		if (te_name == NULL) {
1868 			err = got_error_from_errno("strndup");
1869 			break;
1870 		}
1871 		te = got_object_tree_find_entry(s->tree, te_name);
1872 		if (te == NULL) {
1873 			err = got_error_path(te_name, GOT_ERR_NO_TREE_ENTRY);
1874 			free(te_name);
1875 			break;
1876 		}
1877 		free(te_name);
1878 		s->selected_entry = te;
1879 		s->selected = got_tree_entry_get_index(te);
1880 		if (s->tree != s->root)
1881 			s->selected++; /* skip '..' */
1882 
1883 		if (!S_ISDIR(got_tree_entry_get_mode(s->selected_entry))) {
1884 			/* Jump to this file's entry. */
1885 			s->first_displayed_entry = s->selected_entry;
1886 			s->selected = 0;
1887 			break;
1888 		}
1889 
1890 		slash = strchr(p, '/');
1891 		if (slash)
1892 			subpath = strndup(path, slash - path);
1893 		else
1894 			subpath = strdup(path);
1895 		if (subpath == NULL) {
1896 			err = got_error_from_errno("strdup");
1897 			break;
1898 		}
1899 
1900 		err = got_object_id_by_path(&tree_id, repo, commit_id,
1901 		    subpath);
1902 		if (err)
1903 			break;
1904 
1905 		err = got_object_open_as_tree(&tree, repo, tree_id);
1906 		free(tree_id);
1907 		if (err)
1908 			break;
1909 
1910 		err = tree_view_visit_subtree(tree, s);
1911 		if (err) {
1912 			got_object_tree_close(tree);
1913 			break;
1914 		}
1915 		if (slash == NULL)
1916 			break;
1917 		free(subpath);
1918 		subpath = NULL;
1919 		p = slash;
1920 	}
1921 
1922 	free(subpath);
1923 	return err;
1924 }
1925 
1926 static const struct got_error *
1927 browse_commit_tree(struct tog_view **new_view, int begin_x,
1928     struct commit_queue_entry *entry, const char *path,
1929     struct got_repository *repo)
1930 {
1931 	const struct got_error *err = NULL;
1932 	struct got_tree_object *tree;
1933 	struct tog_tree_view_state *s;
1934 	struct tog_view *tree_view;
1935 
1936 	err = got_object_open_as_tree(&tree, repo,
1937 	    got_object_commit_get_tree_id(entry->commit));
1938 	if (err)
1939 		return err;
1940 
1941 	tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
1942 	if (tree_view == NULL)
1943 		return got_error_from_errno("view_open");
1944 
1945 	err = open_tree_view(tree_view, tree, entry->id, repo);
1946 	if (err) {
1947 		got_object_tree_close(tree);
1948 		return err;
1949 	}
1950 	s = &tree_view->state.tree;
1951 
1952 	*new_view = tree_view;
1953 
1954 	if (got_path_is_root_dir(path))
1955 		return NULL;
1956 
1957 	return tree_view_walk_path(s, entry->id, path, repo);
1958 }
1959 
1960 static const struct got_error *
1961 block_signals_used_by_main_thread(void)
1962 {
1963 	sigset_t sigset;
1964 	int errcode;
1965 
1966 	if (sigemptyset(&sigset) == -1)
1967 		return got_error_from_errno("sigemptyset");
1968 
1969 	/* tog handles SIGWINCH and SIGCONT */
1970 	if (sigaddset(&sigset, SIGWINCH) == -1)
1971 		return got_error_from_errno("sigaddset");
1972 	if (sigaddset(&sigset, SIGCONT) == -1)
1973 		return got_error_from_errno("sigaddset");
1974 
1975 	/* ncurses handles SIGTSTP */
1976 	if (sigaddset(&sigset, SIGTSTP) == -1)
1977 		return got_error_from_errno("sigaddset");
1978 
1979 	errcode = pthread_sigmask(SIG_BLOCK, &sigset, NULL);
1980 	if (errcode)
1981 		return got_error_set_errno(errcode, "pthread_sigmask");
1982 
1983 	return NULL;
1984 }
1985 
1986 static void *
1987 log_thread(void *arg)
1988 {
1989 	const struct got_error *err = NULL;
1990 	int errcode = 0;
1991 	struct tog_log_thread_args *a = arg;
1992 	int done = 0;
1993 
1994 	err = block_signals_used_by_main_thread();
1995 	if (err)
1996 		return (void *)err;
1997 
1998 	while (!done && !err && !tog_sigpipe_received) {
1999 		err = queue_commits(a->graph, a->commits, 1, a->repo,
2000 		    a->in_repo_path, a->searching, a->search_next_done,
2001 		    a->regex);
2002 		if (err) {
2003 			if (err->code != GOT_ERR_ITER_COMPLETED)
2004 				return (void *)err;
2005 			err = NULL;
2006 			done = 1;
2007 		} else if (a->commits_needed > 0)
2008 			a->commits_needed--;
2009 
2010 		errcode = pthread_mutex_lock(&tog_mutex);
2011 		if (errcode) {
2012 			err = got_error_set_errno(errcode,
2013 			    "pthread_mutex_lock");
2014 			break;
2015 		} else if (*a->quit)
2016 			done = 1;
2017 		else if (*a->first_displayed_entry == NULL) {
2018 			*a->first_displayed_entry =
2019 			    TAILQ_FIRST(&a->commits->head);
2020 			*a->selected_entry = *a->first_displayed_entry;
2021 		}
2022 
2023 		errcode = pthread_cond_signal(&a->commit_loaded);
2024 		if (errcode) {
2025 			err = got_error_set_errno(errcode,
2026 			    "pthread_cond_signal");
2027 			pthread_mutex_unlock(&tog_mutex);
2028 			break;
2029 		}
2030 
2031 		if (done)
2032 			a->commits_needed = 0;
2033 		else {
2034 			if (a->commits_needed == 0) {
2035 				errcode = pthread_cond_wait(&a->need_commits,
2036 				    &tog_mutex);
2037 				if (errcode)
2038 					err = got_error_set_errno(errcode,
2039 					    "pthread_cond_wait");
2040 			}
2041 		}
2042 
2043 		errcode = pthread_mutex_unlock(&tog_mutex);
2044 		if (errcode && err == NULL)
2045 			err = got_error_set_errno(errcode,
2046 			    "pthread_mutex_unlock");
2047 	}
2048 	a->log_complete = 1;
2049 	return (void *)err;
2050 }
2051 
2052 static const struct got_error *
2053 stop_log_thread(struct tog_log_view_state *s)
2054 {
2055 	const struct got_error *err = NULL;
2056 	int errcode;
2057 
2058 	if (s->thread) {
2059 		s->quit = 1;
2060 		errcode = pthread_cond_signal(&s->thread_args.need_commits);
2061 		if (errcode)
2062 			return got_error_set_errno(errcode,
2063 			    "pthread_cond_signal");
2064 		errcode = pthread_mutex_unlock(&tog_mutex);
2065 		if (errcode)
2066 			return got_error_set_errno(errcode,
2067 			    "pthread_mutex_unlock");
2068 		errcode = pthread_join(s->thread, (void **)&err);
2069 		if (errcode)
2070 			return got_error_set_errno(errcode, "pthread_join");
2071 		errcode = pthread_mutex_lock(&tog_mutex);
2072 		if (errcode)
2073 			return got_error_set_errno(errcode,
2074 			    "pthread_mutex_lock");
2075 		s->thread = NULL;
2076 	}
2077 
2078 	if (s->thread_args.repo) {
2079 		got_repo_close(s->thread_args.repo);
2080 		s->thread_args.repo = NULL;
2081 	}
2082 
2083 	if (s->thread_args.graph) {
2084 		got_commit_graph_close(s->thread_args.graph);
2085 		s->thread_args.graph = NULL;
2086 	}
2087 
2088 	return err;
2089 }
2090 
2091 static const struct got_error *
2092 close_log_view(struct tog_view *view)
2093 {
2094 	const struct got_error *err = NULL;
2095 	struct tog_log_view_state *s = &view->state.log;
2096 	int errcode;
2097 
2098 	err = stop_log_thread(s);
2099 
2100 	errcode = pthread_cond_destroy(&s->thread_args.need_commits);
2101 	if (errcode && err == NULL)
2102 		err = got_error_set_errno(errcode, "pthread_cond_destroy");
2103 
2104 	errcode = pthread_cond_destroy(&s->thread_args.commit_loaded);
2105 	if (errcode && err == NULL)
2106 		err = got_error_set_errno(errcode, "pthread_cond_destroy");
2107 
2108 	free_commits(&s->commits);
2109 	free(s->in_repo_path);
2110 	s->in_repo_path = NULL;
2111 	free(s->start_id);
2112 	s->start_id = NULL;
2113 	got_ref_list_free(&s->refs);
2114 	return err;
2115 }
2116 
2117 static const struct got_error *
2118 search_start_log_view(struct tog_view *view)
2119 {
2120 	struct tog_log_view_state *s = &view->state.log;
2121 
2122 	s->matched_entry = NULL;
2123 	s->search_entry = NULL;
2124 	return NULL;
2125 }
2126 
2127 static const struct got_error *
2128 search_next_log_view(struct tog_view *view)
2129 {
2130 	const struct got_error *err = NULL;
2131 	struct tog_log_view_state *s = &view->state.log;
2132 	struct commit_queue_entry *entry;
2133 
2134 	/* Display progress update in log view. */
2135 	show_log_view(view);
2136 	update_panels();
2137 	doupdate();
2138 
2139 	if (s->search_entry) {
2140 		int errcode, ch;
2141 		errcode = pthread_mutex_unlock(&tog_mutex);
2142 		if (errcode)
2143 			return got_error_set_errno(errcode,
2144 			    "pthread_mutex_unlock");
2145 		ch = wgetch(view->window);
2146 		errcode = pthread_mutex_lock(&tog_mutex);
2147 		if (errcode)
2148 			return got_error_set_errno(errcode,
2149 			    "pthread_mutex_lock");
2150 		if (ch == KEY_BACKSPACE) {
2151 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
2152 			return NULL;
2153 		}
2154 		if (view->searching == TOG_SEARCH_FORWARD)
2155 			entry = TAILQ_NEXT(s->search_entry, entry);
2156 		else
2157 			entry = TAILQ_PREV(s->search_entry,
2158 			    commit_queue_head, entry);
2159 	} else if (s->matched_entry) {
2160 		if (view->searching == TOG_SEARCH_FORWARD)
2161 			entry = TAILQ_NEXT(s->matched_entry, entry);
2162 		else
2163 			entry = TAILQ_PREV(s->matched_entry,
2164 			    commit_queue_head, entry);
2165 	} else {
2166 		if (view->searching == TOG_SEARCH_FORWARD)
2167 			entry = TAILQ_FIRST(&s->commits.head);
2168 		else
2169 			entry = TAILQ_LAST(&s->commits.head, commit_queue_head);
2170 	}
2171 
2172 	while (1) {
2173 		int have_match = 0;
2174 
2175 		if (entry == NULL) {
2176 			if (s->thread_args.log_complete ||
2177 			    view->searching == TOG_SEARCH_BACKWARD) {
2178 				view->search_next_done =
2179 				    (s->matched_entry == NULL ?
2180 				    TOG_SEARCH_HAVE_NONE : TOG_SEARCH_NO_MORE);
2181 				s->search_entry = NULL;
2182 				return NULL;
2183 			}
2184 			/*
2185 			 * Poke the log thread for more commits and return,
2186 			 * allowing the main loop to make progress. Search
2187 			 * will resume at s->search_entry once we come back.
2188 			 */
2189 			s->thread_args.commits_needed++;
2190 			return trigger_log_thread(view, 0,
2191 			    &s->thread_args.commits_needed,
2192 			    &s->thread_args.log_complete,
2193 			    &s->thread_args.need_commits,
2194 			    &s->thread_args.commit_loaded);
2195 		}
2196 
2197 		err = match_commit(&have_match, entry->id, entry->commit,
2198 		    &view->regex);
2199 		if (err)
2200 			break;
2201 		if (have_match) {
2202 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
2203 			s->matched_entry = entry;
2204 			break;
2205 		}
2206 
2207 		s->search_entry = entry;
2208 		if (view->searching == TOG_SEARCH_FORWARD)
2209 			entry = TAILQ_NEXT(entry, entry);
2210 		else
2211 			entry = TAILQ_PREV(entry, commit_queue_head, entry);
2212 	}
2213 
2214 	if (s->matched_entry) {
2215 		int cur = s->selected_entry->idx;
2216 		while (cur < s->matched_entry->idx) {
2217 			err = input_log_view(NULL, NULL, NULL, view, KEY_DOWN);
2218 			if (err)
2219 				return err;
2220 			cur++;
2221 		}
2222 		while (cur > s->matched_entry->idx) {
2223 			err = input_log_view(NULL, NULL, NULL, view, KEY_UP);
2224 			if (err)
2225 				return err;
2226 			cur--;
2227 		}
2228 	}
2229 
2230 	s->search_entry = NULL;
2231 
2232 	return NULL;
2233 }
2234 
2235 static const struct got_error *
2236 open_log_view(struct tog_view *view, struct got_object_id *start_id,
2237     struct got_repository *repo, const char *head_ref_name,
2238     const char *in_repo_path, int log_branches)
2239 {
2240 	const struct got_error *err = NULL;
2241 	struct tog_log_view_state *s = &view->state.log;
2242 	struct got_repository *thread_repo = NULL;
2243 	struct got_commit_graph *thread_graph = NULL;
2244 	int errcode;
2245 
2246 	SIMPLEQ_INIT(&s->refs);
2247 
2248 	if (in_repo_path != s->in_repo_path) {
2249 		free(s->in_repo_path);
2250 		s->in_repo_path = strdup(in_repo_path);
2251 		if (s->in_repo_path == NULL)
2252 			return got_error_from_errno("strdup");
2253 	}
2254 
2255 	/* The commit queue only contains commits being displayed. */
2256 	TAILQ_INIT(&s->commits.head);
2257 	s->commits.ncommits = 0;
2258 
2259 	err = got_ref_list(&s->refs, repo, NULL, got_ref_cmp_by_name, NULL);
2260 	if (err)
2261 		goto done;
2262 
2263 	s->repo = repo;
2264 	s->head_ref_name = head_ref_name;
2265 	s->start_id = got_object_id_dup(start_id);
2266 	if (s->start_id == NULL) {
2267 		err = got_error_from_errno("got_object_id_dup");
2268 		goto done;
2269 	}
2270 	s->log_branches = log_branches;
2271 
2272 	SIMPLEQ_INIT(&s->colors);
2273 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
2274 		err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
2275 		    get_color_value("TOG_COLOR_COMMIT"));
2276 		if (err)
2277 			goto done;
2278 		err = add_color(&s->colors, "^$", TOG_COLOR_AUTHOR,
2279 		    get_color_value("TOG_COLOR_AUTHOR"));
2280 		if (err) {
2281 			free_colors(&s->colors);
2282 			goto done;
2283 		}
2284 		err = add_color(&s->colors, "^$", TOG_COLOR_DATE,
2285 		    get_color_value("TOG_COLOR_DATE"));
2286 		if (err) {
2287 			free_colors(&s->colors);
2288 			goto done;
2289 		}
2290 	}
2291 
2292 	view->show = show_log_view;
2293 	view->input = input_log_view;
2294 	view->close = close_log_view;
2295 	view->search_start = search_start_log_view;
2296 	view->search_next = search_next_log_view;
2297 
2298 	err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL);
2299 	if (err)
2300 		goto done;
2301 	err = got_commit_graph_open(&thread_graph, s->in_repo_path,
2302 	    !s->log_branches);
2303 	if (err)
2304 		goto done;
2305 	err = got_commit_graph_iter_start(thread_graph, s->start_id,
2306 	    s->repo, NULL, NULL);
2307 	if (err)
2308 		goto done;
2309 
2310 	errcode = pthread_cond_init(&s->thread_args.need_commits, NULL);
2311 	if (errcode) {
2312 		err = got_error_set_errno(errcode, "pthread_cond_init");
2313 		goto done;
2314 	}
2315 	errcode = pthread_cond_init(&s->thread_args.commit_loaded, NULL);
2316 	if (errcode) {
2317 		err = got_error_set_errno(errcode, "pthread_cond_init");
2318 		goto done;
2319 	}
2320 
2321 	s->thread_args.commits_needed = view->nlines;
2322 	s->thread_args.graph = thread_graph;
2323 	s->thread_args.commits = &s->commits;
2324 	s->thread_args.in_repo_path = s->in_repo_path;
2325 	s->thread_args.start_id = s->start_id;
2326 	s->thread_args.repo = thread_repo;
2327 	s->thread_args.log_complete = 0;
2328 	s->thread_args.quit = &s->quit;
2329 	s->thread_args.first_displayed_entry = &s->first_displayed_entry;
2330 	s->thread_args.selected_entry = &s->selected_entry;
2331 	s->thread_args.searching = &view->searching;
2332 	s->thread_args.search_next_done = &view->search_next_done;
2333 	s->thread_args.regex = &view->regex;
2334 done:
2335 	if (err)
2336 		close_log_view(view);
2337 	return err;
2338 }
2339 
2340 static const struct got_error *
2341 show_log_view(struct tog_view *view)
2342 {
2343 	const struct got_error *err;
2344 	struct tog_log_view_state *s = &view->state.log;
2345 
2346 	if (s->thread == NULL) {
2347 		int errcode = pthread_create(&s->thread, NULL, log_thread,
2348 		    &s->thread_args);
2349 		if (errcode)
2350 			return got_error_set_errno(errcode, "pthread_create");
2351 		if (s->thread_args.commits_needed > 0) {
2352 			err = trigger_log_thread(view, 1,
2353 			    &s->thread_args.commits_needed,
2354 			    &s->thread_args.log_complete,
2355 			    &s->thread_args.need_commits,
2356 			    &s->thread_args.commit_loaded);
2357 			if (err)
2358 				return err;
2359 		}
2360 	}
2361 
2362 	return draw_commits(view, &s->last_displayed_entry,
2363 	    &s->selected_entry, s->first_displayed_entry,
2364 	    &s->commits, s->selected, view->nlines, &s->refs,
2365 	    s->in_repo_path, s->thread_args.commits_needed, &s->colors);
2366 }
2367 
2368 static const struct got_error *
2369 input_log_view(struct tog_view **new_view, struct tog_view **dead_view,
2370     struct tog_view **focus_view, struct tog_view *view, int ch)
2371 {
2372 	const struct got_error *err = NULL;
2373 	struct tog_log_view_state *s = &view->state.log;
2374 	char *parent_path, *in_repo_path = NULL;
2375 	struct tog_view *diff_view = NULL, *tree_view = NULL, *lv = NULL;
2376 	struct tog_view *ref_view = NULL;
2377 	int begin_x = 0;
2378 	struct got_object_id *start_id;
2379 
2380 	switch (ch) {
2381 	case 'q':
2382 		s->quit = 1;
2383 		break;
2384 	case 'k':
2385 	case KEY_UP:
2386 	case '<':
2387 	case ',':
2388 		if (s->first_displayed_entry == NULL)
2389 			break;
2390 		if (s->selected > 0)
2391 			s->selected--;
2392 		else
2393 			scroll_up(view, &s->first_displayed_entry, 1,
2394 			    &s->commits);
2395 		break;
2396 	case KEY_PPAGE:
2397 	case CTRL('b'):
2398 		if (s->first_displayed_entry == NULL)
2399 			break;
2400 		if (TAILQ_FIRST(&s->commits.head) ==
2401 		    s->first_displayed_entry) {
2402 			s->selected = 0;
2403 			break;
2404 		}
2405 		scroll_up(view, &s->first_displayed_entry,
2406 		    view->nlines - 1, &s->commits);
2407 		break;
2408 	case 'j':
2409 	case KEY_DOWN:
2410 	case '>':
2411 	case '.':
2412 		if (s->first_displayed_entry == NULL)
2413 			break;
2414 		if (s->selected < MIN(view->nlines - 2,
2415 		    s->commits.ncommits - 1)) {
2416 			s->selected++;
2417 			break;
2418 		}
2419 		err = scroll_down(view, &s->first_displayed_entry, 1,
2420 		    &s->last_displayed_entry, &s->commits,
2421 		    &s->thread_args.log_complete,
2422 		    &s->thread_args.commits_needed,
2423 		    &s->thread_args.need_commits,
2424 		    &s->thread_args.commit_loaded);
2425 		break;
2426 	case KEY_NPAGE:
2427 	case CTRL('f'): {
2428 		struct commit_queue_entry *first;
2429 		first = s->first_displayed_entry;
2430 		if (first == NULL)
2431 			break;
2432 		err = scroll_down(view, &s->first_displayed_entry,
2433 		    view->nlines - 1, &s->last_displayed_entry,
2434 		    &s->commits, &s->thread_args.log_complete,
2435 		    &s->thread_args.commits_needed,
2436 		    &s->thread_args.need_commits,
2437 		    &s->thread_args.commit_loaded);
2438 		if (err)
2439 			break;
2440 		if (first == s->first_displayed_entry &&
2441 		    s->selected < MIN(view->nlines - 2,
2442 		    s->commits.ncommits - 1)) {
2443 			/* can't scroll further down */
2444 			s->selected = MIN(view->nlines - 2,
2445 			    s->commits.ncommits - 1);
2446 		}
2447 		err = NULL;
2448 		break;
2449 	}
2450 	case KEY_RESIZE:
2451 		if (s->selected > view->nlines - 2)
2452 			s->selected = view->nlines - 2;
2453 		if (s->selected > s->commits.ncommits - 1)
2454 			s->selected = s->commits.ncommits - 1;
2455 		break;
2456 	case KEY_ENTER:
2457 	case ' ':
2458 	case '\r':
2459 		if (s->selected_entry == NULL)
2460 			break;
2461 		if (view_is_parent_view(view))
2462 			begin_x = view_split_begin_x(view->begin_x);
2463 		err = open_diff_view_for_commit(&diff_view, begin_x,
2464 		    s->selected_entry->commit, s->selected_entry->id,
2465 		    view, s->repo);
2466 		if (err)
2467 			break;
2468 		if (view_is_parent_view(view)) {
2469 			err = view_close_child(view);
2470 			if (err)
2471 				return err;
2472 			err = view_set_child(view, diff_view);
2473 			if (err) {
2474 				view_close(diff_view);
2475 				break;
2476 			}
2477 			*focus_view = diff_view;
2478 			view->child_focussed = 1;
2479 		} else
2480 			*new_view = diff_view;
2481 		break;
2482 	case 't':
2483 		if (s->selected_entry == NULL)
2484 			break;
2485 		if (view_is_parent_view(view))
2486 			begin_x = view_split_begin_x(view->begin_x);
2487 		err = browse_commit_tree(&tree_view, begin_x,
2488 		    s->selected_entry, s->in_repo_path, s->repo);
2489 		if (err)
2490 			break;
2491 		if (view_is_parent_view(view)) {
2492 			err = view_close_child(view);
2493 			if (err)
2494 				return err;
2495 			err = view_set_child(view, tree_view);
2496 			if (err) {
2497 				view_close(tree_view);
2498 				break;
2499 			}
2500 			*focus_view = tree_view;
2501 			view->child_focussed = 1;
2502 		} else
2503 			*new_view = tree_view;
2504 		break;
2505 	case KEY_BACKSPACE:
2506 		if (got_path_cmp(s->in_repo_path, "/",
2507 		    strlen(s->in_repo_path), 1) == 0)
2508 			break;
2509 		err = got_path_dirname(&parent_path, s->in_repo_path);
2510 		if (err)
2511 			return err;
2512 		err = stop_log_thread(s);
2513 		if (err) {
2514 			free(parent_path);
2515 			return err;
2516 		}
2517 		lv = view_open(view->nlines, view->ncols,
2518 		    view->begin_y, view->begin_x, TOG_VIEW_LOG);
2519 		if (lv == NULL) {
2520 			free(parent_path);
2521 			return got_error_from_errno("view_open");
2522 		}
2523 		err = open_log_view(lv, s->start_id, s->repo, s->head_ref_name,
2524 		    parent_path, s->log_branches);
2525 		free(parent_path);
2526 		if (err)
2527 			return err;;
2528 		if (view_is_parent_view(view))
2529 			*new_view = lv;
2530 		else {
2531 			view_set_child(view->parent, lv);
2532 			*focus_view = lv;
2533 		}
2534 		break;
2535 	case CTRL('l'):
2536 		err = stop_log_thread(s);
2537 		if (err)
2538 			return err;
2539 		lv = view_open(view->nlines, view->ncols,
2540 		    view->begin_y, view->begin_x, TOG_VIEW_LOG);
2541 		if (lv == NULL)
2542 			return got_error_from_errno("view_open");
2543 		err = got_repo_match_object_id(&start_id, NULL,
2544 		    s->head_ref_name ? s->head_ref_name : GOT_REF_HEAD,
2545 		    GOT_OBJ_TYPE_COMMIT, 1, s->repo);
2546 		if (err) {
2547 			view_close(lv);
2548 			return err;
2549 		}
2550 		in_repo_path = strdup(s->in_repo_path);
2551 		if (in_repo_path == NULL) {
2552 			free(start_id);
2553 			view_close(lv);
2554 			return got_error_from_errno("strdup");
2555 		}
2556 		err = open_log_view(lv, start_id, s->repo, s->head_ref_name,
2557 		    in_repo_path, s->log_branches);
2558 		if (err) {
2559 			free(start_id);
2560 			view_close(lv);
2561 			return err;;
2562 		}
2563 		*dead_view = view;
2564 		*new_view = lv;
2565 		break;
2566 	case 'B':
2567 		s->log_branches = !s->log_branches;
2568 		err = stop_log_thread(s);
2569 		if (err)
2570 			return err;
2571 		lv = view_open(view->nlines, view->ncols,
2572 		    view->begin_y, view->begin_x, TOG_VIEW_LOG);
2573 		if (lv == NULL)
2574 			return got_error_from_errno("view_open");
2575 		err = open_log_view(lv, s->start_id, s->repo,
2576 		    s->head_ref_name, s->in_repo_path, s->log_branches);
2577 		if (err) {
2578 			view_close(lv);
2579 			return err;;
2580 		}
2581 		*dead_view = view;
2582 		*new_view = lv;
2583 		break;
2584 	case 'r':
2585 		if (view_is_parent_view(view))
2586 			begin_x = view_split_begin_x(view->begin_x);
2587 		ref_view = view_open(view->nlines, view->ncols,
2588 		    view->begin_y, begin_x, TOG_VIEW_REF);
2589 		if (ref_view == NULL)
2590 			return got_error_from_errno("view_open");
2591 		err = open_ref_view(ref_view, s->repo);
2592 		if (err) {
2593 			view_close(ref_view);
2594 			return err;
2595 		}
2596 		if (view_is_parent_view(view)) {
2597 			err = view_close_child(view);
2598 			if (err)
2599 				return err;
2600 			err = view_set_child(view, ref_view);
2601 			if (err) {
2602 				view_close(ref_view);
2603 				break;
2604 			}
2605 			*focus_view = ref_view;
2606 			view->child_focussed = 1;
2607 		} else
2608 			*new_view = ref_view;
2609 		break;
2610 	default:
2611 		break;
2612 	}
2613 
2614 	return err;
2615 }
2616 
2617 static const struct got_error *
2618 apply_unveil(const char *repo_path, const char *worktree_path)
2619 {
2620 	const struct got_error *error;
2621 
2622 #ifdef PROFILE
2623 	if (unveil("gmon.out", "rwc") != 0)
2624 		return got_error_from_errno2("unveil", "gmon.out");
2625 #endif
2626 	if (repo_path && unveil(repo_path, "r") != 0)
2627 		return got_error_from_errno2("unveil", repo_path);
2628 
2629 	if (worktree_path && unveil(worktree_path, "rwc") != 0)
2630 		return got_error_from_errno2("unveil", worktree_path);
2631 
2632 	if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
2633 		return got_error_from_errno2("unveil", GOT_TMPDIR_STR);
2634 
2635 	error = got_privsep_unveil_exec_helpers();
2636 	if (error != NULL)
2637 		return error;
2638 
2639 	if (unveil(NULL, NULL) != 0)
2640 		return got_error_from_errno("unveil");
2641 
2642 	return NULL;
2643 }
2644 
2645 static void
2646 init_curses(void)
2647 {
2648 	initscr();
2649 	cbreak();
2650 	halfdelay(1); /* Do fast refresh while initial view is loading. */
2651 	noecho();
2652 	nonl();
2653 	intrflush(stdscr, FALSE);
2654 	keypad(stdscr, TRUE);
2655 	curs_set(0);
2656 	if (getenv("TOG_COLORS") != NULL) {
2657 		start_color();
2658 		use_default_colors();
2659 	}
2660 	signal(SIGWINCH, tog_sigwinch);
2661 	signal(SIGPIPE, tog_sigpipe);
2662 	signal(SIGCONT, tog_sigcont);
2663 }
2664 
2665 static const struct got_error *
2666 get_in_repo_path_from_argv0(char **in_repo_path, int argc, char *argv[],
2667     struct got_repository *repo, struct got_worktree *worktree)
2668 {
2669 	const struct got_error *err = NULL;
2670 
2671 	if (argc == 0) {
2672 		*in_repo_path = strdup("/");
2673 		if (*in_repo_path == NULL)
2674 			return got_error_from_errno("strdup");
2675 		return NULL;
2676 	}
2677 
2678 	if (worktree) {
2679 		const char *prefix = got_worktree_get_path_prefix(worktree);
2680 		char *p;
2681 
2682 		err = got_worktree_resolve_path(&p, worktree, argv[0]);
2683 		if (err)
2684 			return err;
2685 		if (asprintf(in_repo_path, "%s%s%s", prefix,
2686 		    (p[0] != '\0' && !got_path_is_root_dir(prefix)) ? "/" : "",
2687 		    p) == -1) {
2688 			err = got_error_from_errno("asprintf");
2689 			*in_repo_path = NULL;
2690 		}
2691 		free(p);
2692 	} else
2693 		err = got_repo_map_path(in_repo_path, repo, argv[0]);
2694 
2695 	return err;
2696 }
2697 
2698 static const struct got_error *
2699 cmd_log(int argc, char *argv[])
2700 {
2701 	const struct got_error *error;
2702 	struct got_repository *repo = NULL;
2703 	struct got_worktree *worktree = NULL;
2704 	struct got_object_id *start_id = NULL;
2705 	char *in_repo_path = NULL, *repo_path = NULL, *cwd = NULL;
2706 	char *start_commit = NULL, *head_ref_name = NULL;
2707 	int ch, log_branches = 0;
2708 	struct tog_view *view;
2709 
2710 #ifndef PROFILE
2711 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
2712 	    NULL) == -1)
2713 		err(1, "pledge");
2714 #endif
2715 
2716 	while ((ch = getopt(argc, argv, "bc:r:")) != -1) {
2717 		switch (ch) {
2718 		case 'b':
2719 			log_branches = 1;
2720 			break;
2721 		case 'c':
2722 			start_commit = optarg;
2723 			break;
2724 		case 'r':
2725 			repo_path = realpath(optarg, NULL);
2726 			if (repo_path == NULL)
2727 				return got_error_from_errno2("realpath",
2728 				    optarg);
2729 			break;
2730 		default:
2731 			usage_log();
2732 			/* NOTREACHED */
2733 		}
2734 	}
2735 
2736 	argc -= optind;
2737 	argv += optind;
2738 
2739 	if (argc > 1)
2740 		usage_log();
2741 
2742 	cwd = getcwd(NULL, 0);
2743 	if (cwd == NULL)
2744 		return got_error_from_errno("getcwd");
2745 
2746 	error = got_worktree_open(&worktree, cwd);
2747 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
2748 		goto done;
2749 
2750 	if (repo_path == NULL) {
2751 		if (worktree)
2752 			repo_path =
2753 			    strdup(got_worktree_get_repo_path(worktree));
2754 		else
2755 			repo_path = strdup(cwd);
2756 	}
2757 	if (repo_path == NULL) {
2758 		error = got_error_from_errno("strdup");
2759 		goto done;
2760 	}
2761 
2762 	error = got_repo_open(&repo, repo_path, NULL);
2763 	if (error != NULL)
2764 		goto done;
2765 
2766 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
2767 	    repo, worktree);
2768 	if (error)
2769 		goto done;
2770 
2771 	init_curses();
2772 
2773 	error = apply_unveil(got_repo_get_path(repo),
2774 	    worktree ? got_worktree_get_root_path(worktree) : NULL);
2775 	if (error)
2776 		goto done;
2777 
2778 	if (start_commit == NULL)
2779 		error = got_repo_match_object_id(&start_id, NULL, worktree ?
2780 		    got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
2781 		    GOT_OBJ_TYPE_COMMIT, 1, repo);
2782 	else
2783 		error = got_repo_match_object_id(&start_id, NULL, start_commit,
2784 		    GOT_OBJ_TYPE_COMMIT, 1, repo);
2785 	if (error != NULL)
2786 		goto done;
2787 
2788 	view = view_open(0, 0, 0, 0, TOG_VIEW_LOG);
2789 	if (view == NULL) {
2790 		error = got_error_from_errno("view_open");
2791 		goto done;
2792 	}
2793 	if (worktree) {
2794 		head_ref_name = strdup(
2795 		    got_worktree_get_head_ref_name(worktree));
2796 		if (head_ref_name == NULL) {
2797 			error = got_error_from_errno("strdup");
2798 			goto done;
2799 		}
2800 	}
2801 	error = open_log_view(view, start_id, repo, head_ref_name,
2802 	    in_repo_path, log_branches);
2803 	if (error)
2804 		goto done;
2805 	if (worktree) {
2806 		/* Release work tree lock. */
2807 		got_worktree_close(worktree);
2808 		worktree = NULL;
2809 	}
2810 	error = view_loop(view);
2811 done:
2812 	free(in_repo_path);
2813 	free(repo_path);
2814 	free(cwd);
2815 	free(start_id);
2816 	free(head_ref_name);
2817 	if (repo)
2818 		got_repo_close(repo);
2819 	if (worktree)
2820 		got_worktree_close(worktree);
2821 	return error;
2822 }
2823 
2824 __dead static void
2825 usage_diff(void)
2826 {
2827 	endwin();
2828 	fprintf(stderr, "usage: %s diff [-a] [-C number] [-r repository-path] "
2829 	    "[-w] object1 object2\n", getprogname());
2830 	exit(1);
2831 }
2832 
2833 static char *
2834 parse_next_line(FILE *f, size_t *len)
2835 {
2836 	char *line;
2837 	size_t linelen;
2838 	size_t lineno;
2839 	const char delim[3] = { '\0', '\0', '\0'};
2840 
2841 	line = fparseln(f, &linelen, &lineno, delim, 0);
2842 	if (len)
2843 		*len = linelen;
2844 	return line;
2845 }
2846 
2847 static int
2848 match_line(const char *line, regex_t *regex, size_t nmatch,
2849     regmatch_t *regmatch)
2850 {
2851 	return regexec(regex, line, nmatch, regmatch, 0) == 0;
2852 }
2853 
2854 struct tog_color *
2855 match_color(struct tog_colors *colors, const char *line)
2856 {
2857 	struct tog_color *tc = NULL;
2858 
2859 	SIMPLEQ_FOREACH(tc, colors, entry) {
2860 		if (match_line(line, &tc->regex, 0, NULL))
2861 			return tc;
2862 	}
2863 
2864 	return NULL;
2865 }
2866 
2867 static const struct got_error *
2868 add_matched_line(int *wtotal, const char *line, int wlimit, int col_tab_align,
2869     WINDOW *window, regmatch_t *regmatch)
2870 {
2871 	const struct got_error *err = NULL;
2872 	wchar_t *wline;
2873 	int width;
2874 	char *s;
2875 
2876 	*wtotal = 0;
2877 
2878 	s = strndup(line, regmatch->rm_so);
2879 	if (s == NULL)
2880 		return got_error_from_errno("strndup");
2881 
2882 	err = format_line(&wline, &width, s, wlimit, col_tab_align);
2883 	if (err) {
2884 		free(s);
2885 		return err;
2886 	}
2887 	waddwstr(window, wline);
2888 	free(wline);
2889 	free(s);
2890 	wlimit -= width;
2891 	*wtotal += width;
2892 
2893 	if (wlimit > 0) {
2894 		s = strndup(line + regmatch->rm_so,
2895 		    regmatch->rm_eo - regmatch->rm_so);
2896 		if (s == NULL) {
2897 			err = got_error_from_errno("strndup");
2898 			free(s);
2899 			return err;
2900 		}
2901 		err = format_line(&wline, &width, s, wlimit, col_tab_align);
2902 		if (err) {
2903 			free(s);
2904 			return err;
2905 		}
2906 		wattr_on(window, A_STANDOUT, NULL);
2907 		waddwstr(window, wline);
2908 		wattr_off(window, A_STANDOUT, NULL);
2909 		free(wline);
2910 		free(s);
2911 		wlimit -= width;
2912 		*wtotal += width;
2913 	}
2914 
2915 	if (wlimit > 0 && strlen(line) > regmatch->rm_eo) {
2916 		err = format_line(&wline, &width,
2917 		     line + regmatch->rm_eo, wlimit, col_tab_align);
2918 		if (err)
2919 			return err;
2920 		waddwstr(window, wline);
2921 		free(wline);
2922 		*wtotal += width;
2923 	}
2924 
2925 	return NULL;
2926 }
2927 
2928 static const struct got_error *
2929 draw_file(struct tog_view *view, FILE *f, int first_displayed_line, int nlines,
2930     off_t *line_offsets, int selected_line, int max_lines,
2931     int *last_displayed_line, int *eof, const char *header,
2932     struct tog_colors *colors, int matched_line, regmatch_t *regmatch)
2933 {
2934 	const struct got_error *err;
2935 	int nprinted = 0;
2936 	char *line;
2937 	struct tog_color *tc;
2938 	size_t len;
2939 	wchar_t *wline;
2940 	int width;
2941 	off_t line_offset;
2942 
2943 	line_offset = line_offsets[first_displayed_line - 1];
2944 	if (fseeko(f, line_offset, SEEK_SET) == -1)
2945 		return got_error_from_errno("fseek");
2946 
2947 	werase(view->window);
2948 
2949 	if (header) {
2950 		if (asprintf(&line, "[%d/%d] %s",
2951 		    first_displayed_line - 1 + selected_line, nlines,
2952 		    header) == -1)
2953 			return got_error_from_errno("asprintf");
2954 		err = format_line(&wline, &width, line, view->ncols, 0);
2955 		free(line);
2956 		if (err)
2957 			return err;
2958 
2959 		if (view_needs_focus_indication(view))
2960 			wstandout(view->window);
2961 		waddwstr(view->window, wline);
2962 		free(wline);
2963 		wline = NULL;
2964 		if (view_needs_focus_indication(view))
2965 			wstandend(view->window);
2966 		if (width <= view->ncols - 1)
2967 			waddch(view->window, '\n');
2968 
2969 		if (max_lines <= 1)
2970 			return NULL;
2971 		max_lines--;
2972 	}
2973 
2974 	*eof = 0;
2975 	while (max_lines > 0 && nprinted < max_lines) {
2976 		line = parse_next_line(f, &len);
2977 		if (line == NULL) {
2978 			*eof = 1;
2979 			break;
2980 		}
2981 
2982 		tc = match_color(colors, line);
2983 		if (tc)
2984 			wattr_on(view->window,
2985 			    COLOR_PAIR(tc->colorpair), NULL);
2986 		if (first_displayed_line + nprinted == matched_line &&
2987 		    regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
2988 			err = add_matched_line(&width, line, view->ncols, 0,
2989 			    view->window, regmatch);
2990 			if (err) {
2991 				free(line);
2992 				return err;
2993 			}
2994 		} else {
2995 			err = format_line(&wline, &width, line, view->ncols, 0);
2996 			if (err) {
2997 				free(line);
2998 				return err;
2999 			}
3000 			waddwstr(view->window, wline);
3001 			free(wline);
3002 			wline = NULL;
3003 		}
3004 		if (tc)
3005 			wattr_off(view->window,
3006 			    COLOR_PAIR(tc->colorpair), NULL);
3007 		if (width <= view->ncols - 1)
3008 			waddch(view->window, '\n');
3009 		nprinted++;
3010 		free(line);
3011 	}
3012 	if (nprinted >= 1)
3013 		*last_displayed_line = first_displayed_line + (nprinted - 1);
3014 	else
3015 		*last_displayed_line = first_displayed_line;
3016 
3017 	view_vborder(view);
3018 
3019 	if (*eof) {
3020 		while (nprinted < view->nlines) {
3021 			waddch(view->window, '\n');
3022 			nprinted++;
3023 		}
3024 
3025 		err = format_line(&wline, &width, TOG_EOF_STRING, view->ncols, 0);
3026 		if (err) {
3027 			return err;
3028 		}
3029 
3030 		wstandout(view->window);
3031 		waddwstr(view->window, wline);
3032 		free(wline);
3033 		wline = NULL;
3034 		wstandend(view->window);
3035 	}
3036 
3037 	return NULL;
3038 }
3039 
3040 static char *
3041 get_datestr(time_t *time, char *datebuf)
3042 {
3043 	struct tm mytm, *tm;
3044 	char *p, *s;
3045 
3046 	tm = gmtime_r(time, &mytm);
3047 	if (tm == NULL)
3048 		return NULL;
3049 	s = asctime_r(tm, datebuf);
3050 	if (s == NULL)
3051 		return NULL;
3052 	p = strchr(s, '\n');
3053 	if (p)
3054 		*p = '\0';
3055 	return s;
3056 }
3057 
3058 static const struct got_error *
3059 get_changed_paths(struct got_pathlist_head *paths,
3060     struct got_commit_object *commit, struct got_repository *repo)
3061 {
3062 	const struct got_error *err = NULL;
3063 	struct got_object_id *tree_id1 = NULL, *tree_id2 = NULL;
3064 	struct got_tree_object *tree1 = NULL, *tree2 = NULL;
3065 	struct got_object_qid *qid;
3066 
3067 	qid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
3068 	if (qid != NULL) {
3069 		struct got_commit_object *pcommit;
3070 		err = got_object_open_as_commit(&pcommit, repo,
3071 		    qid->id);
3072 		if (err)
3073 			return err;
3074 
3075 		tree_id1 = got_object_commit_get_tree_id(pcommit);
3076 		got_object_commit_close(pcommit);
3077 
3078 	}
3079 
3080 	if (tree_id1) {
3081 		err = got_object_open_as_tree(&tree1, repo, tree_id1);
3082 		if (err)
3083 			goto done;
3084 	}
3085 
3086 	tree_id2 = got_object_commit_get_tree_id(commit);
3087 	err = got_object_open_as_tree(&tree2, repo, tree_id2);
3088 	if (err)
3089 		goto done;
3090 
3091 	err = got_diff_tree(tree1, tree2, "", "", repo,
3092 	    got_diff_tree_collect_changed_paths, paths, 0);
3093 done:
3094 	if (tree1)
3095 		got_object_tree_close(tree1);
3096 	if (tree2)
3097 		got_object_tree_close(tree2);
3098 	return err;
3099 }
3100 
3101 static const struct got_error *
3102 add_line_offset(off_t **line_offsets, size_t *nlines, off_t off)
3103 {
3104 	off_t *p;
3105 
3106 	p = reallocarray(*line_offsets, *nlines + 1, sizeof(off_t));
3107 	if (p == NULL)
3108 		return got_error_from_errno("reallocarray");
3109 	*line_offsets = p;
3110 	(*line_offsets)[*nlines] = off;
3111 	(*nlines)++;
3112 	return NULL;
3113 }
3114 
3115 static const struct got_error *
3116 write_commit_info(off_t **line_offsets, size_t *nlines,
3117     struct got_object_id *commit_id, struct got_reflist_head *refs,
3118     struct got_repository *repo, FILE *outfile)
3119 {
3120 	const struct got_error *err = NULL;
3121 	char datebuf[26], *datestr;
3122 	struct got_commit_object *commit;
3123 	char *id_str = NULL, *logmsg = NULL, *s = NULL, *line;
3124 	time_t committer_time;
3125 	const char *author, *committer;
3126 	char *refs_str = NULL;
3127 	struct got_pathlist_head changed_paths;
3128 	struct got_pathlist_entry *pe;
3129 	off_t outoff = 0;
3130 	int n;
3131 
3132 	TAILQ_INIT(&changed_paths);
3133 
3134 	if (refs) {
3135 		err = build_refs_str(&refs_str, refs, commit_id, repo);
3136 		if (err)
3137 			return err;
3138 	}
3139 
3140 	err = got_object_open_as_commit(&commit, repo, commit_id);
3141 	if (err)
3142 		return err;
3143 
3144 	err = got_object_id_str(&id_str, commit_id);
3145 	if (err) {
3146 		err = got_error_from_errno("got_object_id_str");
3147 		goto done;
3148 	}
3149 
3150 	err = add_line_offset(line_offsets, nlines, 0);
3151 	if (err)
3152 		goto done;
3153 
3154 	n = fprintf(outfile, "commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
3155 	    refs_str ? refs_str : "", refs_str ? ")" : "");
3156 	if (n < 0) {
3157 		err = got_error_from_errno("fprintf");
3158 		goto done;
3159 	}
3160 	outoff += n;
3161 	err = add_line_offset(line_offsets, nlines, outoff);
3162 	if (err)
3163 		goto done;
3164 
3165 	n = fprintf(outfile, "from: %s\n",
3166 	    got_object_commit_get_author(commit));
3167 	if (n < 0) {
3168 		err = got_error_from_errno("fprintf");
3169 		goto done;
3170 	}
3171 	outoff += n;
3172 	err = add_line_offset(line_offsets, nlines, outoff);
3173 	if (err)
3174 		goto done;
3175 
3176 	committer_time = got_object_commit_get_committer_time(commit);
3177 	datestr = get_datestr(&committer_time, datebuf);
3178 	if (datestr) {
3179 		n = fprintf(outfile, "date: %s UTC\n", datestr);
3180 		if (n < 0) {
3181 			err = got_error_from_errno("fprintf");
3182 			goto done;
3183 		}
3184 		outoff += n;
3185 		err = add_line_offset(line_offsets, nlines, outoff);
3186 		if (err)
3187 			goto done;
3188 	}
3189 	author = got_object_commit_get_author(commit);
3190 	committer = got_object_commit_get_committer(commit);
3191 	if (strcmp(author, committer) != 0) {
3192 		n = fprintf(outfile, "via: %s\n", committer);
3193 		if (n < 0) {
3194 			err = got_error_from_errno("fprintf");
3195 			goto done;
3196 		}
3197 		outoff += n;
3198 		err = add_line_offset(line_offsets, nlines, outoff);
3199 		if (err)
3200 			goto done;
3201 	}
3202 	err = got_object_commit_get_logmsg(&logmsg, commit);
3203 	if (err)
3204 		goto done;
3205 	s = logmsg;
3206 	while ((line = strsep(&s, "\n")) != NULL) {
3207 		n = fprintf(outfile, "%s\n", line);
3208 		if (n < 0) {
3209 			err = got_error_from_errno("fprintf");
3210 			goto done;
3211 		}
3212 		outoff += n;
3213 		err = add_line_offset(line_offsets, nlines, outoff);
3214 		if (err)
3215 			goto done;
3216 	}
3217 
3218 	err = get_changed_paths(&changed_paths, commit, repo);
3219 	if (err)
3220 		goto done;
3221 	TAILQ_FOREACH(pe, &changed_paths, entry) {
3222 		struct got_diff_changed_path *cp = pe->data;
3223 		n = fprintf(outfile, "%c  %s\n", cp->status, pe->path);
3224 		if (n < 0) {
3225 			err = got_error_from_errno("fprintf");
3226 			goto done;
3227 		}
3228 		outoff += n;
3229 		err = add_line_offset(line_offsets, nlines, outoff);
3230 		if (err)
3231 			goto done;
3232 		free((char *)pe->path);
3233 		free(pe->data);
3234 	}
3235 
3236 	fputc('\n', outfile);
3237 	outoff++;
3238 	err = add_line_offset(line_offsets, nlines, outoff);
3239 done:
3240 	got_pathlist_free(&changed_paths);
3241 	free(id_str);
3242 	free(logmsg);
3243 	free(refs_str);
3244 	got_object_commit_close(commit);
3245 	if (err) {
3246 		free(*line_offsets);
3247 		*line_offsets = NULL;
3248 		*nlines = 0;
3249 	}
3250 	return err;
3251 }
3252 
3253 static const struct got_error *
3254 create_diff(struct tog_diff_view_state *s)
3255 {
3256 	const struct got_error *err = NULL;
3257 	FILE *f = NULL;
3258 	int obj_type;
3259 
3260 	free(s->line_offsets);
3261 	s->line_offsets = malloc(sizeof(off_t));
3262 	if (s->line_offsets == NULL)
3263 		return got_error_from_errno("malloc");
3264 	s->nlines = 0;
3265 
3266 	f = got_opentemp();
3267 	if (f == NULL) {
3268 		err = got_error_from_errno("got_opentemp");
3269 		goto done;
3270 	}
3271 	if (s->f && fclose(s->f) != 0) {
3272 		err = got_error_from_errno("fclose");
3273 		goto done;
3274 	}
3275 	s->f = f;
3276 
3277 	if (s->id1)
3278 		err = got_object_get_type(&obj_type, s->repo, s->id1);
3279 	else
3280 		err = got_object_get_type(&obj_type, s->repo, s->id2);
3281 	if (err)
3282 		goto done;
3283 
3284 	switch (obj_type) {
3285 	case GOT_OBJ_TYPE_BLOB:
3286 		err = got_diff_objects_as_blobs(&s->line_offsets, &s->nlines,
3287 		    s->id1, s->id2, s->label1, s->label2, s->diff_context,
3288 		    s->ignore_whitespace, s->force_text_diff, s->repo, s->f);
3289 		break;
3290 	case GOT_OBJ_TYPE_TREE:
3291 		err = got_diff_objects_as_trees(&s->line_offsets, &s->nlines,
3292 		    s->id1, s->id2, "", "", s->diff_context,
3293 		    s->ignore_whitespace, s->force_text_diff, s->repo, s->f);
3294 		break;
3295 	case GOT_OBJ_TYPE_COMMIT: {
3296 		const struct got_object_id_queue *parent_ids;
3297 		struct got_object_qid *pid;
3298 		struct got_commit_object *commit2;
3299 
3300 		err = got_object_open_as_commit(&commit2, s->repo, s->id2);
3301 		if (err)
3302 			goto done;
3303 		/* Show commit info if we're diffing to a parent/root commit. */
3304 		if (s->id1 == NULL) {
3305 			err = write_commit_info(&s->line_offsets, &s->nlines,
3306 			    s->id2, &s->refs, s->repo, s->f);
3307 			if (err)
3308 				goto done;
3309 		} else {
3310 			parent_ids = got_object_commit_get_parent_ids(commit2);
3311 			SIMPLEQ_FOREACH(pid, parent_ids, entry) {
3312 				if (got_object_id_cmp(s->id1, pid->id) == 0) {
3313 					err = write_commit_info(
3314 					    &s->line_offsets, &s->nlines,
3315 					    s->id2, &s->refs, s->repo, s->f);
3316 					if (err)
3317 						goto done;
3318 					break;
3319 				}
3320 			}
3321 		}
3322 		got_object_commit_close(commit2);
3323 
3324 		err = got_diff_objects_as_commits(&s->line_offsets, &s->nlines,
3325 		    s->id1, s->id2, s->diff_context, s->ignore_whitespace,
3326 		    s->force_text_diff, s->repo, s->f);
3327 		break;
3328 	}
3329 	default:
3330 		err = got_error(GOT_ERR_OBJ_TYPE);
3331 		break;
3332 	}
3333 	if (err)
3334 		goto done;
3335 done:
3336 	if (s->f && fflush(s->f) != 0 && err == NULL)
3337 		err = got_error_from_errno("fflush");
3338 	return err;
3339 }
3340 
3341 static void
3342 diff_view_indicate_progress(struct tog_view *view)
3343 {
3344 	mvwaddstr(view->window, 0, 0, "diffing...");
3345 	update_panels();
3346 	doupdate();
3347 }
3348 
3349 static const struct got_error *
3350 search_start_diff_view(struct tog_view *view)
3351 {
3352 	struct tog_diff_view_state *s = &view->state.diff;
3353 
3354 	s->matched_line = 0;
3355 	return NULL;
3356 }
3357 
3358 static const struct got_error *
3359 search_next_diff_view(struct tog_view *view)
3360 {
3361 	struct tog_diff_view_state *s = &view->state.diff;
3362 	int lineno;
3363 
3364 	if (!view->searching) {
3365 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
3366 		return NULL;
3367 	}
3368 
3369 	if (s->matched_line) {
3370 		if (view->searching == TOG_SEARCH_FORWARD)
3371 			lineno = s->matched_line + 1;
3372 		else
3373 			lineno = s->matched_line - 1;
3374 	} else {
3375 		if (view->searching == TOG_SEARCH_FORWARD)
3376 			lineno = 1;
3377 		else
3378 			lineno = s->nlines;
3379 	}
3380 
3381 	while (1) {
3382 		char *line = NULL;
3383 		off_t offset;
3384 		size_t len;
3385 
3386 		if (lineno <= 0 || lineno > s->nlines) {
3387 			if (s->matched_line == 0) {
3388 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
3389 				free(line);
3390 				break;
3391 			}
3392 
3393 			if (view->searching == TOG_SEARCH_FORWARD)
3394 				lineno = 1;
3395 			else
3396 				lineno = s->nlines;
3397 		}
3398 
3399 		offset = s->line_offsets[lineno - 1];
3400 		if (fseeko(s->f, offset, SEEK_SET) != 0) {
3401 			free(line);
3402 			return got_error_from_errno("fseeko");
3403 		}
3404 		free(line);
3405 		line = parse_next_line(s->f, &len);
3406 		if (line &&
3407 		    match_line(line, &view->regex, 1, &view->regmatch)) {
3408 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
3409 			s->matched_line = lineno;
3410 			free(line);
3411 			break;
3412 		}
3413 		free(line);
3414 		if (view->searching == TOG_SEARCH_FORWARD)
3415 			lineno++;
3416 		else
3417 			lineno--;
3418 	}
3419 
3420 	if (s->matched_line) {
3421 		s->first_displayed_line = s->matched_line;
3422 		s->selected_line = 1;
3423 	}
3424 
3425 	return NULL;
3426 }
3427 
3428 static const struct got_error *
3429 open_diff_view(struct tog_view *view, struct got_object_id *id1,
3430     struct got_object_id *id2, const char *label1, const char *label2,
3431     int diff_context, int ignore_whitespace, int force_text_diff,
3432     struct tog_view *log_view, struct got_repository *repo)
3433 {
3434 	const struct got_error *err;
3435 	struct tog_diff_view_state *s = &view->state.diff;
3436 
3437 	SIMPLEQ_INIT(&s->refs);
3438 
3439 	if (id1 != NULL && id2 != NULL) {
3440 	    int type1, type2;
3441 	    err = got_object_get_type(&type1, repo, id1);
3442 	    if (err)
3443 		return err;
3444 	    err = got_object_get_type(&type2, repo, id2);
3445 	    if (err)
3446 		return err;
3447 
3448 	    if (type1 != type2)
3449 		return got_error(GOT_ERR_OBJ_TYPE);
3450 	}
3451 	s->first_displayed_line = 1;
3452 	s->last_displayed_line = view->nlines;
3453 	s->selected_line = 1;
3454 	s->repo = repo;
3455 	s->id1 = id1;
3456 	s->id2 = id2;
3457 	s->label1 = label1;
3458 	s->label2 = label2;
3459 
3460 	if (id1) {
3461 		s->id1 = got_object_id_dup(id1);
3462 		if (s->id1 == NULL)
3463 			return got_error_from_errno("got_object_id_dup");
3464 	} else
3465 		s->id1 = NULL;
3466 
3467 	s->id2 = got_object_id_dup(id2);
3468 	if (s->id2 == NULL) {
3469 		free(s->id1);
3470 		s->id1 = NULL;
3471 		return got_error_from_errno("got_object_id_dup");
3472 	}
3473 	s->f = NULL;
3474 	s->first_displayed_line = 1;
3475 	s->last_displayed_line = view->nlines;
3476 	s->diff_context = diff_context;
3477 	s->ignore_whitespace = ignore_whitespace;
3478 	s->force_text_diff = force_text_diff;
3479 	s->log_view = log_view;
3480 	s->repo = repo;
3481 
3482 	SIMPLEQ_INIT(&s->colors);
3483 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
3484 		err = add_color(&s->colors,
3485 		    "^-", TOG_COLOR_DIFF_MINUS,
3486 		    get_color_value("TOG_COLOR_DIFF_MINUS"));
3487 		if (err)
3488 			return err;
3489 		err = add_color(&s->colors, "^\\+",
3490 		    TOG_COLOR_DIFF_PLUS,
3491 		    get_color_value("TOG_COLOR_DIFF_PLUS"));
3492 		if (err) {
3493 			free_colors(&s->colors);
3494 			return err;
3495 		}
3496 		err = add_color(&s->colors,
3497 		    "^@@", TOG_COLOR_DIFF_CHUNK_HEADER,
3498 		    get_color_value("TOG_COLOR_DIFF_CHUNK_HEADER"));
3499 		if (err) {
3500 			free_colors(&s->colors);
3501 			return err;
3502 		}
3503 
3504 		err = add_color(&s->colors,
3505 		    "^(commit [0-9a-f]|(blob|file) [-+] |[MDmA]  [^ ])",
3506 		    TOG_COLOR_DIFF_META,
3507 		    get_color_value("TOG_COLOR_DIFF_META"));
3508 		if (err) {
3509 			free_colors(&s->colors);
3510 			return err;
3511 		}
3512 
3513 		err = add_color(&s->colors,
3514 		    "^(from|via): ", TOG_COLOR_AUTHOR,
3515 		    get_color_value("TOG_COLOR_AUTHOR"));
3516 		if (err) {
3517 			free_colors(&s->colors);
3518 			return err;
3519 		}
3520 
3521 		err = add_color(&s->colors,
3522 		    "^date: ", TOG_COLOR_DATE,
3523 		    get_color_value("TOG_COLOR_DATE"));
3524 		if (err) {
3525 			free_colors(&s->colors);
3526 			return err;
3527 		}
3528 	}
3529 
3530 	err = got_ref_list(&s->refs, repo, NULL, got_ref_cmp_by_name, NULL);
3531 	if (err) {
3532 		free(s->id1);
3533 		s->id1 = NULL;
3534 		free(s->id2);
3535 		s->id2 = NULL;
3536 		free_colors(&s->colors);
3537 		return err;
3538 	}
3539 
3540 	if (log_view && view_is_splitscreen(view))
3541 		show_log_view(log_view); /* draw vborder */
3542 	diff_view_indicate_progress(view);
3543 
3544 	s->line_offsets = NULL;
3545 	s->nlines = 0;
3546 	err = create_diff(s);
3547 	if (err) {
3548 		free(s->id1);
3549 		s->id1 = NULL;
3550 		free(s->id2);
3551 		s->id2 = NULL;
3552 		free_colors(&s->colors);
3553 		got_ref_list_free(&s->refs);
3554 		return err;
3555 	}
3556 
3557 	view->show = show_diff_view;
3558 	view->input = input_diff_view;
3559 	view->close = close_diff_view;
3560 	view->search_start = search_start_diff_view;
3561 	view->search_next = search_next_diff_view;
3562 
3563 	return NULL;
3564 }
3565 
3566 static const struct got_error *
3567 close_diff_view(struct tog_view *view)
3568 {
3569 	const struct got_error *err = NULL;
3570 	struct tog_diff_view_state *s = &view->state.diff;
3571 
3572 	free(s->id1);
3573 	s->id1 = NULL;
3574 	free(s->id2);
3575 	s->id2 = NULL;
3576 	if (s->f && fclose(s->f) == EOF)
3577 		err = got_error_from_errno("fclose");
3578 	free_colors(&s->colors);
3579 	free(s->line_offsets);
3580 	s->line_offsets = NULL;
3581 	s->nlines = 0;
3582 	got_ref_list_free(&s->refs);
3583 	return err;
3584 }
3585 
3586 static const struct got_error *
3587 show_diff_view(struct tog_view *view)
3588 {
3589 	const struct got_error *err;
3590 	struct tog_diff_view_state *s = &view->state.diff;
3591 	char *id_str1 = NULL, *id_str2, *header;
3592 	const char *label1, *label2;
3593 
3594 	if (s->id1) {
3595 		err = got_object_id_str(&id_str1, s->id1);
3596 		if (err)
3597 			return err;
3598 		label1 = s->label1 ? : id_str1;
3599 	} else
3600 		label1 = "/dev/null";
3601 
3602 	err = got_object_id_str(&id_str2, s->id2);
3603 	if (err)
3604 		return err;
3605 	label2 = s->label2 ? : id_str2;
3606 
3607 	if (asprintf(&header, "diff %s %s", label1, label2) == -1) {
3608 		err = got_error_from_errno("asprintf");
3609 		free(id_str1);
3610 		free(id_str2);
3611 		return err;
3612 	}
3613 	free(id_str1);
3614 	free(id_str2);
3615 
3616 	return draw_file(view, s->f, s->first_displayed_line, s->nlines,
3617 	    s->line_offsets, s->selected_line, view->nlines,
3618 	    &s->last_displayed_line, &s->eof, header, &s->colors,
3619 	    s->matched_line, &view->regmatch);
3620 }
3621 
3622 static const struct got_error *
3623 set_selected_commit(struct tog_diff_view_state *s,
3624     struct commit_queue_entry *entry)
3625 {
3626 	const struct got_error *err;
3627 	const struct got_object_id_queue *parent_ids;
3628 	struct got_commit_object *selected_commit;
3629 	struct got_object_qid *pid;
3630 
3631 	free(s->id2);
3632 	s->id2 = got_object_id_dup(entry->id);
3633 	if (s->id2 == NULL)
3634 		return got_error_from_errno("got_object_id_dup");
3635 
3636 	err = got_object_open_as_commit(&selected_commit, s->repo, entry->id);
3637 	if (err)
3638 		return err;
3639 	parent_ids = got_object_commit_get_parent_ids(selected_commit);
3640 	free(s->id1);
3641 	pid = SIMPLEQ_FIRST(parent_ids);
3642 	s->id1 = pid ? got_object_id_dup(pid->id) : NULL;
3643 	got_object_commit_close(selected_commit);
3644 	return NULL;
3645 }
3646 
3647 static const struct got_error *
3648 input_diff_view(struct tog_view **new_view, struct tog_view **dead_view,
3649     struct tog_view **focus_view, struct tog_view *view, int ch)
3650 {
3651 	const struct got_error *err = NULL;
3652 	struct tog_diff_view_state *s = &view->state.diff;
3653 	struct tog_log_view_state *ls;
3654 	struct commit_queue_entry *entry;
3655 	int i;
3656 
3657 	switch (ch) {
3658 	case 'a':
3659 	case 'w':
3660 		if (ch == 'a')
3661 			s->force_text_diff = !s->force_text_diff;
3662 		if (ch == 'w')
3663 			s->ignore_whitespace = !s->ignore_whitespace;
3664 		wclear(view->window);
3665 		s->first_displayed_line = 1;
3666 		s->last_displayed_line = view->nlines;
3667 		diff_view_indicate_progress(view);
3668 		err = create_diff(s);
3669 		break;
3670 	case 'k':
3671 	case KEY_UP:
3672 		if (s->first_displayed_line > 1)
3673 			s->first_displayed_line--;
3674 		break;
3675 	case KEY_PPAGE:
3676 	case CTRL('b'):
3677 		if (s->first_displayed_line == 1)
3678 			break;
3679 		i = 0;
3680 		while (i++ < view->nlines - 1 &&
3681 		    s->first_displayed_line > 1)
3682 			s->first_displayed_line--;
3683 		break;
3684 	case 'j':
3685 	case KEY_DOWN:
3686 		if (!s->eof)
3687 			s->first_displayed_line++;
3688 		break;
3689 	case KEY_NPAGE:
3690 	case CTRL('f'):
3691 	case ' ':
3692 		if (s->eof)
3693 			break;
3694 		i = 0;
3695 		while (!s->eof && i++ < view->nlines - 1) {
3696 			char *line;
3697 			line = parse_next_line(s->f, NULL);
3698 			s->first_displayed_line++;
3699 			if (line == NULL)
3700 				break;
3701 		}
3702 		break;
3703 	case '[':
3704 		if (s->diff_context > 0) {
3705 			s->diff_context--;
3706 			diff_view_indicate_progress(view);
3707 			err = create_diff(s);
3708 			if (s->first_displayed_line + view->nlines - 1 >
3709 			    s->nlines) {
3710 				s->first_displayed_line = 1;
3711 				s->last_displayed_line = view->nlines;
3712 			}
3713 		}
3714 		break;
3715 	case ']':
3716 		if (s->diff_context < GOT_DIFF_MAX_CONTEXT) {
3717 			s->diff_context++;
3718 			diff_view_indicate_progress(view);
3719 			err = create_diff(s);
3720 		}
3721 		break;
3722 	case '<':
3723 	case ',':
3724 		if (s->log_view == NULL)
3725 			break;
3726 		ls = &s->log_view->state.log;
3727 		entry = TAILQ_PREV(ls->selected_entry,
3728 		    commit_queue_head, entry);
3729 		if (entry == NULL)
3730 			break;
3731 
3732 		err = input_log_view(NULL, NULL, NULL, s->log_view,
3733 		    KEY_UP);
3734 		if (err)
3735 			break;
3736 
3737 		err = set_selected_commit(s, entry);
3738 		if (err)
3739 			break;
3740 
3741 		s->first_displayed_line = 1;
3742 		s->last_displayed_line = view->nlines;
3743 
3744 		diff_view_indicate_progress(view);
3745 		err = create_diff(s);
3746 		break;
3747 	case '>':
3748 	case '.':
3749 		if (s->log_view == NULL)
3750 			break;
3751 		ls = &s->log_view->state.log;
3752 
3753 		if (TAILQ_NEXT(ls->selected_entry, entry) == NULL) {
3754 			ls->thread_args.commits_needed++;
3755 			err = trigger_log_thread(s->log_view, 1,
3756 			    &ls->thread_args.commits_needed,
3757 			    &ls->thread_args.log_complete,
3758 			    &ls->thread_args.need_commits,
3759 			    &ls->thread_args.commit_loaded);
3760 			if (err)
3761 				break;
3762 		}
3763 		err = input_log_view(NULL, NULL, NULL, s->log_view,
3764 		    KEY_DOWN);
3765 		if (err)
3766 			break;
3767 
3768 		entry = TAILQ_NEXT(ls->selected_entry, entry);
3769 		if (entry == NULL)
3770 			break;
3771 
3772 		err = set_selected_commit(s, entry);
3773 		if (err)
3774 			break;
3775 
3776 		s->first_displayed_line = 1;
3777 		s->last_displayed_line = view->nlines;
3778 
3779 		diff_view_indicate_progress(view);
3780 		err = create_diff(s);
3781 		break;
3782 	default:
3783 		break;
3784 	}
3785 
3786 	return err;
3787 }
3788 
3789 static const struct got_error *
3790 cmd_diff(int argc, char *argv[])
3791 {
3792 	const struct got_error *error = NULL;
3793 	struct got_repository *repo = NULL;
3794 	struct got_worktree *worktree = NULL;
3795 	struct got_object_id *id1 = NULL, *id2 = NULL;
3796 	char *repo_path = NULL, *cwd = NULL;
3797 	char *id_str1 = NULL, *id_str2 = NULL;
3798 	char *label1 = NULL, *label2 = NULL;
3799 	int diff_context = 3, ignore_whitespace = 0;
3800 	int ch, force_text_diff = 0;
3801 	const char *errstr;
3802 	struct tog_view *view;
3803 
3804 #ifndef PROFILE
3805 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
3806 	    NULL) == -1)
3807 		err(1, "pledge");
3808 #endif
3809 	while ((ch = getopt(argc, argv, "aC:r:w")) != -1) {
3810 		switch (ch) {
3811 		case 'a':
3812 			force_text_diff = 1;
3813 			break;
3814 		case 'C':
3815 			diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
3816 			    &errstr);
3817 			if (errstr != NULL)
3818 				err(1, "-C option %s", errstr);
3819 			break;
3820 		case 'r':
3821 			repo_path = realpath(optarg, NULL);
3822 			if (repo_path == NULL)
3823 				return got_error_from_errno2("realpath",
3824 				    optarg);
3825 			got_path_strip_trailing_slashes(repo_path);
3826 			break;
3827 		case 'w':
3828 			ignore_whitespace = 1;
3829 			break;
3830 		default:
3831 			usage_diff();
3832 			/* NOTREACHED */
3833 		}
3834 	}
3835 
3836 	argc -= optind;
3837 	argv += optind;
3838 
3839 	if (argc == 0) {
3840 		usage_diff(); /* TODO show local worktree changes */
3841 	} else if (argc == 2) {
3842 		id_str1 = argv[0];
3843 		id_str2 = argv[1];
3844 	} else
3845 		usage_diff();
3846 
3847 	cwd = getcwd(NULL, 0);
3848 	if (cwd == NULL)
3849 		return got_error_from_errno("getcwd");
3850 
3851 	error = got_worktree_open(&worktree, cwd);
3852 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
3853 		goto done;
3854 
3855 	if (repo_path == NULL) {
3856 		if (worktree)
3857 			repo_path =
3858 			    strdup(got_worktree_get_repo_path(worktree));
3859 		else
3860 			repo_path = strdup(cwd);
3861 	}
3862 	if (repo_path == NULL) {
3863 		error = got_error_from_errno("strdup");
3864 		goto done;
3865 	}
3866 
3867 	error = got_repo_open(&repo, repo_path, NULL);
3868 	if (error)
3869 		goto done;
3870 
3871 	init_curses();
3872 
3873 	error = apply_unveil(got_repo_get_path(repo), NULL);
3874 	if (error)
3875 		goto done;
3876 
3877 	error = got_repo_match_object_id(&id1, &label1, id_str1,
3878 	    GOT_OBJ_TYPE_ANY, 1, repo);
3879 	if (error)
3880 		goto done;
3881 
3882 	error = got_repo_match_object_id(&id2, &label2, id_str2,
3883 	    GOT_OBJ_TYPE_ANY, 1, repo);
3884 	if (error)
3885 		goto done;
3886 
3887 	view = view_open(0, 0, 0, 0, TOG_VIEW_DIFF);
3888 	if (view == NULL) {
3889 		error = got_error_from_errno("view_open");
3890 		goto done;
3891 	}
3892 	error = open_diff_view(view, id1, id2, label1, label2, diff_context,
3893 	    ignore_whitespace, force_text_diff, NULL,  repo);
3894 	if (error)
3895 		goto done;
3896 	error = view_loop(view);
3897 done:
3898 	free(label1);
3899 	free(label2);
3900 	free(repo_path);
3901 	free(cwd);
3902 	if (repo)
3903 		got_repo_close(repo);
3904 	if (worktree)
3905 		got_worktree_close(worktree);
3906 	return error;
3907 }
3908 
3909 __dead static void
3910 usage_blame(void)
3911 {
3912 	endwin();
3913 	fprintf(stderr, "usage: %s blame [-c commit] [-r repository-path] path\n",
3914 	    getprogname());
3915 	exit(1);
3916 }
3917 
3918 struct tog_blame_line {
3919 	int annotated;
3920 	struct got_object_id *id;
3921 };
3922 
3923 static const struct got_error *
3924 draw_blame(struct tog_view *view, struct got_object_id *id, FILE *f,
3925     const char *path, struct tog_blame_line *lines, int nlines,
3926     int blame_complete, int selected_line, int *first_displayed_line,
3927     int *last_displayed_line, int *eof, int max_lines,
3928     struct tog_colors *colors, int matched_line, regmatch_t *regmatch)
3929 {
3930 	const struct got_error *err;
3931 	int lineno = 0, nprinted = 0;
3932 	char *line;
3933 	size_t len;
3934 	wchar_t *wline;
3935 	int width;
3936 	struct tog_blame_line *blame_line;
3937 	struct got_object_id *prev_id = NULL;
3938 	char *id_str;
3939 	struct tog_color *tc;
3940 
3941 	err = got_object_id_str(&id_str, id);
3942 	if (err)
3943 		return err;
3944 
3945 	rewind(f);
3946 	werase(view->window);
3947 
3948 	if (asprintf(&line, "commit %s", id_str) == -1) {
3949 		err = got_error_from_errno("asprintf");
3950 		free(id_str);
3951 		return err;
3952 	}
3953 
3954 	err = format_line(&wline, &width, line, view->ncols, 0);
3955 	free(line);
3956 	line = NULL;
3957 	if (err)
3958 		return err;
3959 	if (view_needs_focus_indication(view))
3960 		wstandout(view->window);
3961 	tc = get_color(colors, TOG_COLOR_COMMIT);
3962 	if (tc)
3963 		wattr_on(view->window,
3964 		    COLOR_PAIR(tc->colorpair), NULL);
3965 	waddwstr(view->window, wline);
3966 	if (tc)
3967 		wattr_off(view->window,
3968 		    COLOR_PAIR(tc->colorpair), NULL);
3969 	if (view_needs_focus_indication(view))
3970 		wstandend(view->window);
3971 	free(wline);
3972 	wline = NULL;
3973 	if (width < view->ncols - 1)
3974 		waddch(view->window, '\n');
3975 
3976 	if (asprintf(&line, "[%d/%d] %s%s",
3977 	    *first_displayed_line - 1 + selected_line, nlines,
3978 	    blame_complete ? "" : "annotating... ", path) == -1) {
3979 		free(id_str);
3980 		return got_error_from_errno("asprintf");
3981 	}
3982 	free(id_str);
3983 	err = format_line(&wline, &width, line, view->ncols, 0);
3984 	free(line);
3985 	line = NULL;
3986 	if (err)
3987 		return err;
3988 	waddwstr(view->window, wline);
3989 	free(wline);
3990 	wline = NULL;
3991 	if (width < view->ncols - 1)
3992 		waddch(view->window, '\n');
3993 
3994 	*eof = 0;
3995 	while (nprinted < max_lines - 2) {
3996 		line = parse_next_line(f, &len);
3997 		if (line == NULL) {
3998 			*eof = 1;
3999 			break;
4000 		}
4001 		if (++lineno < *first_displayed_line) {
4002 			free(line);
4003 			continue;
4004 		}
4005 
4006 		if (view->focussed && nprinted == selected_line - 1)
4007 			wstandout(view->window);
4008 
4009 		if (nlines > 0) {
4010 			blame_line = &lines[lineno - 1];
4011 			if (blame_line->annotated && prev_id &&
4012 			    got_object_id_cmp(prev_id, blame_line->id) == 0 &&
4013 			    !(view->focussed &&
4014 			    nprinted == selected_line - 1)) {
4015 				waddstr(view->window, "        ");
4016 			} else if (blame_line->annotated) {
4017 				char *id_str;
4018 				err = got_object_id_str(&id_str, blame_line->id);
4019 				if (err) {
4020 					free(line);
4021 					return err;
4022 				}
4023 				tc = get_color(colors, TOG_COLOR_COMMIT);
4024 				if (tc)
4025 					wattr_on(view->window,
4026 					    COLOR_PAIR(tc->colorpair), NULL);
4027 				wprintw(view->window, "%.8s", id_str);
4028 				if (tc)
4029 					wattr_off(view->window,
4030 					    COLOR_PAIR(tc->colorpair), NULL);
4031 				free(id_str);
4032 				prev_id = blame_line->id;
4033 			} else {
4034 				waddstr(view->window, "........");
4035 				prev_id = NULL;
4036 			}
4037 		} else {
4038 			waddstr(view->window, "........");
4039 			prev_id = NULL;
4040 		}
4041 
4042 		if (view->focussed && nprinted == selected_line - 1)
4043 			wstandend(view->window);
4044 		waddstr(view->window, " ");
4045 
4046 		if (view->ncols <= 9) {
4047 			width = 9;
4048 			wline = wcsdup(L"");
4049 			if (wline == NULL) {
4050 				err = got_error_from_errno("wcsdup");
4051 				free(line);
4052 				return err;
4053 			}
4054 		} else if (*first_displayed_line + nprinted == matched_line &&
4055 		    regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
4056 			err = add_matched_line(&width, line, view->ncols - 9, 9,
4057 			    view->window, regmatch);
4058 			if (err) {
4059 				free(line);
4060 				return err;
4061 			}
4062 			width += 9;
4063 		} else {
4064 			err = format_line(&wline, &width, line,
4065 			    view->ncols - 9, 9);
4066 			waddwstr(view->window, wline);
4067 			free(wline);
4068 			wline = NULL;
4069 			width += 9;
4070 		}
4071 
4072 		if (width <= view->ncols - 1)
4073 			waddch(view->window, '\n');
4074 		if (++nprinted == 1)
4075 			*first_displayed_line = lineno;
4076 		free(line);
4077 	}
4078 	*last_displayed_line = lineno;
4079 
4080 	view_vborder(view);
4081 
4082 	return NULL;
4083 }
4084 
4085 static const struct got_error *
4086 blame_cb(void *arg, int nlines, int lineno, struct got_object_id *id)
4087 {
4088 	const struct got_error *err = NULL;
4089 	struct tog_blame_cb_args *a = arg;
4090 	struct tog_blame_line *line;
4091 	int errcode;
4092 
4093 	if (nlines != a->nlines ||
4094 	    (lineno != -1 && lineno < 1) || lineno > a->nlines)
4095 		return got_error(GOT_ERR_RANGE);
4096 
4097 	errcode = pthread_mutex_lock(&tog_mutex);
4098 	if (errcode)
4099 		return got_error_set_errno(errcode, "pthread_mutex_lock");
4100 
4101 	if (*a->quit) {	/* user has quit the blame view */
4102 		err = got_error(GOT_ERR_ITER_COMPLETED);
4103 		goto done;
4104 	}
4105 
4106 	if (lineno == -1)
4107 		goto done; /* no change in this commit */
4108 
4109 	line = &a->lines[lineno - 1];
4110 	if (line->annotated)
4111 		goto done;
4112 
4113 	line->id = got_object_id_dup(id);
4114 	if (line->id == NULL) {
4115 		err = got_error_from_errno("got_object_id_dup");
4116 		goto done;
4117 	}
4118 	line->annotated = 1;
4119 done:
4120 	errcode = pthread_mutex_unlock(&tog_mutex);
4121 	if (errcode)
4122 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
4123 	return err;
4124 }
4125 
4126 static void *
4127 blame_thread(void *arg)
4128 {
4129 	const struct got_error *err;
4130 	struct tog_blame_thread_args *ta = arg;
4131 	struct tog_blame_cb_args *a = ta->cb_args;
4132 	int errcode;
4133 
4134 	err = block_signals_used_by_main_thread();
4135 	if (err)
4136 		return (void *)err;
4137 
4138 	err = got_blame(ta->path, a->commit_id, ta->repo,
4139 	    blame_cb, ta->cb_args, ta->cancel_cb, ta->cancel_arg);
4140 	if (err && err->code == GOT_ERR_CANCELLED)
4141 		err = NULL;
4142 
4143 	errcode = pthread_mutex_lock(&tog_mutex);
4144 	if (errcode)
4145 		return (void *)got_error_set_errno(errcode,
4146 		    "pthread_mutex_lock");
4147 
4148 	got_repo_close(ta->repo);
4149 	ta->repo = NULL;
4150 	*ta->complete = 1;
4151 
4152 	errcode = pthread_mutex_unlock(&tog_mutex);
4153 	if (errcode && err == NULL)
4154 		err = got_error_set_errno(errcode, "pthread_mutex_unlock");
4155 
4156 	return (void *)err;
4157 }
4158 
4159 static struct got_object_id *
4160 get_selected_commit_id(struct tog_blame_line *lines, int nlines,
4161     int first_displayed_line, int selected_line)
4162 {
4163 	struct tog_blame_line *line;
4164 
4165 	if (nlines <= 0)
4166 		return NULL;
4167 
4168 	line = &lines[first_displayed_line - 1 + selected_line - 1];
4169 	if (!line->annotated)
4170 		return NULL;
4171 
4172 	return line->id;
4173 }
4174 
4175 static const struct got_error *
4176 stop_blame(struct tog_blame *blame)
4177 {
4178 	const struct got_error *err = NULL;
4179 	int i;
4180 
4181 	if (blame->thread) {
4182 		int errcode;
4183 		errcode = pthread_mutex_unlock(&tog_mutex);
4184 		if (errcode)
4185 			return got_error_set_errno(errcode,
4186 			    "pthread_mutex_unlock");
4187 		errcode = pthread_join(blame->thread, (void **)&err);
4188 		if (errcode)
4189 			return got_error_set_errno(errcode, "pthread_join");
4190 		errcode = pthread_mutex_lock(&tog_mutex);
4191 		if (errcode)
4192 			return got_error_set_errno(errcode,
4193 			    "pthread_mutex_lock");
4194 		if (err && err->code == GOT_ERR_ITER_COMPLETED)
4195 			err = NULL;
4196 		blame->thread = NULL;
4197 	}
4198 	if (blame->thread_args.repo) {
4199 		got_repo_close(blame->thread_args.repo);
4200 		blame->thread_args.repo = NULL;
4201 	}
4202 	if (blame->f) {
4203 		if (fclose(blame->f) != 0 && err == NULL)
4204 			err = got_error_from_errno("fclose");
4205 		blame->f = NULL;
4206 	}
4207 	if (blame->lines) {
4208 		for (i = 0; i < blame->nlines; i++)
4209 			free(blame->lines[i].id);
4210 		free(blame->lines);
4211 		blame->lines = NULL;
4212 	}
4213 	free(blame->cb_args.commit_id);
4214 	blame->cb_args.commit_id = NULL;
4215 
4216 	return err;
4217 }
4218 
4219 static const struct got_error *
4220 cancel_blame_view(void *arg)
4221 {
4222 	const struct got_error *err = NULL;
4223 	int *done = arg;
4224 	int errcode;
4225 
4226 	errcode = pthread_mutex_lock(&tog_mutex);
4227 	if (errcode)
4228 		return got_error_set_errno(errcode,
4229 		    "pthread_mutex_unlock");
4230 
4231 	if (*done)
4232 		err = got_error(GOT_ERR_CANCELLED);
4233 
4234 	errcode = pthread_mutex_unlock(&tog_mutex);
4235 	if (errcode)
4236 		return got_error_set_errno(errcode,
4237 		    "pthread_mutex_lock");
4238 
4239 	return err;
4240 }
4241 
4242 static const struct got_error *
4243 run_blame(struct tog_blame *blame, struct tog_view *view, int *blame_complete,
4244     int *first_displayed_line, int *last_displayed_line, int *selected_line,
4245     int *done, int *eof, const char *path, struct got_object_id *commit_id,
4246     struct got_repository *repo)
4247 {
4248 	const struct got_error *err = NULL;
4249 	struct got_blob_object *blob = NULL;
4250 	struct got_repository *thread_repo = NULL;
4251 	struct got_object_id *obj_id = NULL;
4252 	int obj_type;
4253 
4254 	err = got_object_id_by_path(&obj_id, repo, commit_id, path);
4255 	if (err)
4256 		return err;
4257 
4258 	err = got_object_get_type(&obj_type, repo, obj_id);
4259 	if (err)
4260 		goto done;
4261 
4262 	if (obj_type != GOT_OBJ_TYPE_BLOB) {
4263 		err = got_error(GOT_ERR_OBJ_TYPE);
4264 		goto done;
4265 	}
4266 
4267 	err = got_object_open_as_blob(&blob, repo, obj_id, 8192);
4268 	if (err)
4269 		goto done;
4270 	blame->f = got_opentemp();
4271 	if (blame->f == NULL) {
4272 		err = got_error_from_errno("got_opentemp");
4273 		goto done;
4274 	}
4275 	err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
4276 	    &blame->line_offsets, blame->f, blob);
4277 	if (err || blame->nlines == 0)
4278 		goto done;
4279 
4280 	/* Don't include \n at EOF in the blame line count. */
4281 	if (blame->line_offsets[blame->nlines - 1] == blame->filesize)
4282 		blame->nlines--;
4283 
4284 	blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
4285 	if (blame->lines == NULL) {
4286 		err = got_error_from_errno("calloc");
4287 		goto done;
4288 	}
4289 
4290 	err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL);
4291 	if (err)
4292 		goto done;
4293 
4294 	blame->cb_args.view = view;
4295 	blame->cb_args.lines = blame->lines;
4296 	blame->cb_args.nlines = blame->nlines;
4297 	blame->cb_args.commit_id = got_object_id_dup(commit_id);
4298 	if (blame->cb_args.commit_id == NULL) {
4299 		err = got_error_from_errno("got_object_id_dup");
4300 		goto done;
4301 	}
4302 	blame->cb_args.quit = done;
4303 
4304 	blame->thread_args.path = path;
4305 	blame->thread_args.repo = thread_repo;
4306 	blame->thread_args.cb_args = &blame->cb_args;
4307 	blame->thread_args.complete = blame_complete;
4308 	blame->thread_args.cancel_cb = cancel_blame_view;
4309 	blame->thread_args.cancel_arg = done;
4310 	*blame_complete = 0;
4311 
4312 done:
4313 	if (blob)
4314 		got_object_blob_close(blob);
4315 	free(obj_id);
4316 	if (err)
4317 		stop_blame(blame);
4318 	return err;
4319 }
4320 
4321 static const struct got_error *
4322 open_blame_view(struct tog_view *view, char *path,
4323     struct got_object_id *commit_id, struct got_repository *repo)
4324 {
4325 	const struct got_error *err = NULL;
4326 	struct tog_blame_view_state *s = &view->state.blame;
4327 
4328 	SIMPLEQ_INIT(&s->blamed_commits);
4329 
4330 	s->path = strdup(path);
4331 	if (s->path == NULL)
4332 		return got_error_from_errno("strdup");
4333 
4334 	err = got_object_qid_alloc(&s->blamed_commit, commit_id);
4335 	if (err) {
4336 		free(s->path);
4337 		return err;
4338 	}
4339 
4340 	SIMPLEQ_INSERT_HEAD(&s->blamed_commits, s->blamed_commit, entry);
4341 	s->first_displayed_line = 1;
4342 	s->last_displayed_line = view->nlines;
4343 	s->selected_line = 1;
4344 	s->blame_complete = 0;
4345 	s->repo = repo;
4346 	s->commit_id = commit_id;
4347 	memset(&s->blame, 0, sizeof(s->blame));
4348 
4349 	SIMPLEQ_INIT(&s->colors);
4350 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
4351 		err = add_color(&s->colors, "^", TOG_COLOR_COMMIT,
4352 		    get_color_value("TOG_COLOR_COMMIT"));
4353 		if (err)
4354 			return err;
4355 	}
4356 
4357 	view->show = show_blame_view;
4358 	view->input = input_blame_view;
4359 	view->close = close_blame_view;
4360 	view->search_start = search_start_blame_view;
4361 	view->search_next = search_next_blame_view;
4362 
4363 	return run_blame(&s->blame, view, &s->blame_complete,
4364 	    &s->first_displayed_line, &s->last_displayed_line,
4365 	    &s->selected_line, &s->done, &s->eof, s->path,
4366 	    s->blamed_commit->id, s->repo);
4367 }
4368 
4369 static const struct got_error *
4370 close_blame_view(struct tog_view *view)
4371 {
4372 	const struct got_error *err = NULL;
4373 	struct tog_blame_view_state *s = &view->state.blame;
4374 
4375 	if (s->blame.thread)
4376 		err = stop_blame(&s->blame);
4377 
4378 	while (!SIMPLEQ_EMPTY(&s->blamed_commits)) {
4379 		struct got_object_qid *blamed_commit;
4380 		blamed_commit = SIMPLEQ_FIRST(&s->blamed_commits);
4381 		SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
4382 		got_object_qid_free(blamed_commit);
4383 	}
4384 
4385 	free(s->path);
4386 	free_colors(&s->colors);
4387 
4388 	return err;
4389 }
4390 
4391 static const struct got_error *
4392 search_start_blame_view(struct tog_view *view)
4393 {
4394 	struct tog_blame_view_state *s = &view->state.blame;
4395 
4396 	s->matched_line = 0;
4397 	return NULL;
4398 }
4399 
4400 static const struct got_error *
4401 search_next_blame_view(struct tog_view *view)
4402 {
4403 	struct tog_blame_view_state *s = &view->state.blame;
4404 	int lineno;
4405 
4406 	if (!view->searching) {
4407 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
4408 		return NULL;
4409 	}
4410 
4411 	if (s->matched_line) {
4412 		if (view->searching == TOG_SEARCH_FORWARD)
4413 			lineno = s->matched_line + 1;
4414 		else
4415 			lineno = s->matched_line - 1;
4416 	} else {
4417 		if (view->searching == TOG_SEARCH_FORWARD)
4418 			lineno = 1;
4419 		else
4420 			lineno = s->blame.nlines;
4421 	}
4422 
4423 	while (1) {
4424 		char *line = NULL;
4425 		off_t offset;
4426 		size_t len;
4427 
4428 		if (lineno <= 0 || lineno > s->blame.nlines) {
4429 			if (s->matched_line == 0) {
4430 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
4431 				free(line);
4432 				break;
4433 			}
4434 
4435 			if (view->searching == TOG_SEARCH_FORWARD)
4436 				lineno = 1;
4437 			else
4438 				lineno = s->blame.nlines;
4439 		}
4440 
4441 		offset = s->blame.line_offsets[lineno - 1];
4442 		if (fseeko(s->blame.f, offset, SEEK_SET) != 0) {
4443 			free(line);
4444 			return got_error_from_errno("fseeko");
4445 		}
4446 		free(line);
4447 		line = parse_next_line(s->blame.f, &len);
4448 		if (line &&
4449 		    match_line(line, &view->regex, 1, &view->regmatch)) {
4450 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
4451 			s->matched_line = lineno;
4452 			free(line);
4453 			break;
4454 		}
4455 		free(line);
4456 		if (view->searching == TOG_SEARCH_FORWARD)
4457 			lineno++;
4458 		else
4459 			lineno--;
4460 	}
4461 
4462 	if (s->matched_line) {
4463 		s->first_displayed_line = s->matched_line;
4464 		s->selected_line = 1;
4465 	}
4466 
4467 	return NULL;
4468 }
4469 
4470 static const struct got_error *
4471 show_blame_view(struct tog_view *view)
4472 {
4473 	const struct got_error *err = NULL;
4474 	struct tog_blame_view_state *s = &view->state.blame;
4475 	int errcode;
4476 
4477 	if (s->blame.thread == NULL) {
4478 		errcode = pthread_create(&s->blame.thread, NULL, blame_thread,
4479 		    &s->blame.thread_args);
4480 		if (errcode)
4481 			return got_error_set_errno(errcode, "pthread_create");
4482 
4483 		halfdelay(1); /* fast refresh while annotating  */
4484 	}
4485 
4486 	if (s->blame_complete)
4487 		halfdelay(10); /* disable fast refresh */
4488 
4489 	err = draw_blame(view, s->blamed_commit->id, s->blame.f,
4490 	    s->path, s->blame.lines, s->blame.nlines, s->blame_complete,
4491 	    s->selected_line, &s->first_displayed_line,
4492 	    &s->last_displayed_line, &s->eof, view->nlines, &s->colors,
4493 	    s->matched_line, &view->regmatch);
4494 
4495 	view_vborder(view);
4496 	return err;
4497 }
4498 
4499 static const struct got_error *
4500 input_blame_view(struct tog_view **new_view, struct tog_view **dead_view,
4501     struct tog_view **focus_view, struct tog_view *view, int ch)
4502 {
4503 	const struct got_error *err = NULL, *thread_err = NULL;
4504 	struct tog_view *diff_view;
4505 	struct tog_blame_view_state *s = &view->state.blame;
4506 	int begin_x = 0;
4507 
4508 	switch (ch) {
4509 	case 'q':
4510 		s->done = 1;
4511 		break;
4512 	case 'k':
4513 	case KEY_UP:
4514 		if (s->selected_line > 1)
4515 			s->selected_line--;
4516 		else if (s->selected_line == 1 &&
4517 		    s->first_displayed_line > 1)
4518 			s->first_displayed_line--;
4519 		break;
4520 	case KEY_PPAGE:
4521 	case CTRL('b'):
4522 		if (s->first_displayed_line == 1) {
4523 			s->selected_line = 1;
4524 			break;
4525 		}
4526 		if (s->first_displayed_line > view->nlines - 2)
4527 			s->first_displayed_line -=
4528 			    (view->nlines - 2);
4529 		else
4530 			s->first_displayed_line = 1;
4531 		break;
4532 	case 'j':
4533 	case KEY_DOWN:
4534 		if (s->selected_line < view->nlines - 2 &&
4535 		    s->first_displayed_line +
4536 		    s->selected_line <= s->blame.nlines)
4537 			s->selected_line++;
4538 		else if (s->last_displayed_line <
4539 		    s->blame.nlines)
4540 			s->first_displayed_line++;
4541 		break;
4542 	case 'b':
4543 	case 'p': {
4544 		struct got_object_id *id = NULL;
4545 		id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
4546 		    s->first_displayed_line, s->selected_line);
4547 		if (id == NULL)
4548 			break;
4549 		if (ch == 'p') {
4550 			struct got_commit_object *commit;
4551 			struct got_object_qid *pid;
4552 			struct got_object_id *blob_id = NULL;
4553 			int obj_type;
4554 			err = got_object_open_as_commit(&commit,
4555 			    s->repo, id);
4556 			if (err)
4557 				break;
4558 			pid = SIMPLEQ_FIRST(
4559 			    got_object_commit_get_parent_ids(commit));
4560 			if (pid == NULL) {
4561 				got_object_commit_close(commit);
4562 				break;
4563 			}
4564 			/* Check if path history ends here. */
4565 			err = got_object_id_by_path(&blob_id, s->repo,
4566 			    pid->id, s->path);
4567 			if (err) {
4568 				if (err->code == GOT_ERR_NO_TREE_ENTRY)
4569 					err = NULL;
4570 				got_object_commit_close(commit);
4571 				break;
4572 			}
4573 			err = got_object_get_type(&obj_type, s->repo,
4574 			    blob_id);
4575 			free(blob_id);
4576 			/* Can't blame non-blob type objects. */
4577 			if (obj_type != GOT_OBJ_TYPE_BLOB) {
4578 				got_object_commit_close(commit);
4579 				break;
4580 			}
4581 			err = got_object_qid_alloc(&s->blamed_commit,
4582 			    pid->id);
4583 			got_object_commit_close(commit);
4584 		} else {
4585 			if (got_object_id_cmp(id,
4586 			    s->blamed_commit->id) == 0)
4587 				break;
4588 			err = got_object_qid_alloc(&s->blamed_commit,
4589 			    id);
4590 		}
4591 		if (err)
4592 			break;
4593 		s->done = 1;
4594 		thread_err = stop_blame(&s->blame);
4595 		s->done = 0;
4596 		if (thread_err)
4597 			break;
4598 		SIMPLEQ_INSERT_HEAD(&s->blamed_commits,
4599 		    s->blamed_commit, entry);
4600 		err = run_blame(&s->blame, view, &s->blame_complete,
4601 		    &s->first_displayed_line, &s->last_displayed_line,
4602 		    &s->selected_line, &s->done, &s->eof,
4603 		    s->path, s->blamed_commit->id, s->repo);
4604 		if (err)
4605 			break;
4606 		break;
4607 	}
4608 	case 'B': {
4609 		struct got_object_qid *first;
4610 		first = SIMPLEQ_FIRST(&s->blamed_commits);
4611 		if (!got_object_id_cmp(first->id, s->commit_id))
4612 			break;
4613 		s->done = 1;
4614 		thread_err = stop_blame(&s->blame);
4615 		s->done = 0;
4616 		if (thread_err)
4617 			break;
4618 		SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
4619 		got_object_qid_free(s->blamed_commit);
4620 		s->blamed_commit =
4621 		    SIMPLEQ_FIRST(&s->blamed_commits);
4622 		err = run_blame(&s->blame, view, &s->blame_complete,
4623 		    &s->first_displayed_line, &s->last_displayed_line,
4624 		    &s->selected_line, &s->done, &s->eof, s->path,
4625 		    s->blamed_commit->id, s->repo);
4626 		if (err)
4627 			break;
4628 		break;
4629 	}
4630 	case KEY_ENTER:
4631 	case '\r': {
4632 		struct got_object_id *id = NULL;
4633 		struct got_object_qid *pid;
4634 		struct got_commit_object *commit = NULL;
4635 		id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
4636 		    s->first_displayed_line, s->selected_line);
4637 		if (id == NULL)
4638 			break;
4639 		err = got_object_open_as_commit(&commit, s->repo, id);
4640 		if (err)
4641 			break;
4642 		pid = SIMPLEQ_FIRST(
4643 		    got_object_commit_get_parent_ids(commit));
4644 		if (view_is_parent_view(view))
4645 		    begin_x = view_split_begin_x(view->begin_x);
4646 		diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
4647 		if (diff_view == NULL) {
4648 			got_object_commit_close(commit);
4649 			err = got_error_from_errno("view_open");
4650 			break;
4651 		}
4652 		err = open_diff_view(diff_view, pid ? pid->id : NULL,
4653 		    id, NULL, NULL, 3, 0, 0, NULL, s->repo);
4654 		got_object_commit_close(commit);
4655 		if (err) {
4656 			view_close(diff_view);
4657 			break;
4658 		}
4659 		if (view_is_parent_view(view)) {
4660 			err = view_close_child(view);
4661 			if (err)
4662 				break;
4663 			err = view_set_child(view, diff_view);
4664 			if (err) {
4665 				view_close(diff_view);
4666 				break;
4667 			}
4668 			*focus_view = diff_view;
4669 			view->child_focussed = 1;
4670 		} else
4671 			*new_view = diff_view;
4672 		if (err)
4673 			break;
4674 		break;
4675 	}
4676 	case KEY_NPAGE:
4677 	case CTRL('f'):
4678 	case ' ':
4679 		if (s->last_displayed_line >= s->blame.nlines &&
4680 		    s->selected_line >= MIN(s->blame.nlines,
4681 		    view->nlines - 2)) {
4682 			break;
4683 		}
4684 		if (s->last_displayed_line >= s->blame.nlines &&
4685 		    s->selected_line < view->nlines - 2) {
4686 			s->selected_line = MIN(s->blame.nlines,
4687 			    view->nlines - 2);
4688 			break;
4689 		}
4690 		if (s->last_displayed_line + view->nlines - 2
4691 		    <= s->blame.nlines)
4692 			s->first_displayed_line +=
4693 			    view->nlines - 2;
4694 		else
4695 			s->first_displayed_line =
4696 			    s->blame.nlines -
4697 			    (view->nlines - 3);
4698 		break;
4699 	case KEY_RESIZE:
4700 		if (s->selected_line > view->nlines - 2) {
4701 			s->selected_line = MIN(s->blame.nlines,
4702 			    view->nlines - 2);
4703 		}
4704 		break;
4705 	default:
4706 		break;
4707 	}
4708 	return thread_err ? thread_err : err;
4709 }
4710 
4711 static const struct got_error *
4712 cmd_blame(int argc, char *argv[])
4713 {
4714 	const struct got_error *error;
4715 	struct got_repository *repo = NULL;
4716 	struct got_worktree *worktree = NULL;
4717 	char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
4718 	char *link_target = NULL;
4719 	struct got_object_id *commit_id = NULL;
4720 	char *commit_id_str = NULL;
4721 	int ch;
4722 	struct tog_view *view;
4723 
4724 #ifndef PROFILE
4725 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
4726 	    NULL) == -1)
4727 		err(1, "pledge");
4728 #endif
4729 
4730 	while ((ch = getopt(argc, argv, "c:r:")) != -1) {
4731 		switch (ch) {
4732 		case 'c':
4733 			commit_id_str = optarg;
4734 			break;
4735 		case 'r':
4736 			repo_path = realpath(optarg, NULL);
4737 			if (repo_path == NULL)
4738 				return got_error_from_errno2("realpath",
4739 				    optarg);
4740 			break;
4741 		default:
4742 			usage_blame();
4743 			/* NOTREACHED */
4744 		}
4745 	}
4746 
4747 	argc -= optind;
4748 	argv += optind;
4749 
4750 	if (argc != 1)
4751 		usage_blame();
4752 
4753 	cwd = getcwd(NULL, 0);
4754 	if (cwd == NULL)
4755 		return got_error_from_errno("getcwd");
4756 
4757 	error = got_worktree_open(&worktree, cwd);
4758 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
4759 		goto done;
4760 
4761 	if (repo_path == NULL) {
4762 		if (worktree)
4763 			repo_path =
4764 			    strdup(got_worktree_get_repo_path(worktree));
4765 		else
4766 			repo_path = strdup(cwd);
4767 	}
4768 	if (repo_path == NULL) {
4769 		error = got_error_from_errno("strdup");
4770 		goto done;
4771 	}
4772 
4773 	error = got_repo_open(&repo, repo_path, NULL);
4774 	if (error != NULL)
4775 		goto done;
4776 
4777 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv, repo,
4778 	    worktree);
4779 	if (error)
4780 		goto done;
4781 
4782 	init_curses();
4783 
4784 	error = apply_unveil(got_repo_get_path(repo), NULL);
4785 	if (error)
4786 		goto done;
4787 
4788 	if (commit_id_str == NULL) {
4789 		struct got_reference *head_ref;
4790 		error = got_ref_open(&head_ref, repo, worktree ?
4791 		    got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD, 0);
4792 		if (error != NULL)
4793 			goto done;
4794 		error = got_ref_resolve(&commit_id, repo, head_ref);
4795 		got_ref_close(head_ref);
4796 	} else {
4797 		error = got_repo_match_object_id(&commit_id, NULL,
4798 		    commit_id_str, GOT_OBJ_TYPE_COMMIT, 1, repo);
4799 	}
4800 	if (error != NULL)
4801 		goto done;
4802 
4803 	view = view_open(0, 0, 0, 0, TOG_VIEW_BLAME);
4804 	if (view == NULL) {
4805 		error = got_error_from_errno("view_open");
4806 		goto done;
4807 	}
4808 
4809 	error = got_object_resolve_symlinks(&link_target, in_repo_path,
4810 	    commit_id, repo);
4811 	if (error)
4812 		goto done;
4813 
4814 	error = open_blame_view(view, link_target ? link_target : in_repo_path,
4815 	    commit_id, repo);
4816 	if (error)
4817 		goto done;
4818 	if (worktree) {
4819 		/* Release work tree lock. */
4820 		got_worktree_close(worktree);
4821 		worktree = NULL;
4822 	}
4823 	error = view_loop(view);
4824 done:
4825 	free(repo_path);
4826 	free(in_repo_path);
4827 	free(link_target);
4828 	free(cwd);
4829 	free(commit_id);
4830 	if (worktree)
4831 		got_worktree_close(worktree);
4832 	if (repo)
4833 		got_repo_close(repo);
4834 	return error;
4835 }
4836 
4837 static const struct got_error *
4838 draw_tree_entries(struct tog_view *view,
4839     struct got_tree_entry **first_displayed_entry,
4840     struct got_tree_entry **last_displayed_entry,
4841     struct got_tree_entry **selected_entry, int *ndisplayed,
4842     const char *label, int show_ids, const char *parent_path,
4843     struct got_tree_object *tree, int selected, int limit,
4844     int isroot, struct tog_colors *colors, struct got_repository *repo)
4845 {
4846 	const struct got_error *err = NULL;
4847 	struct got_tree_entry *te;
4848 	wchar_t *wline;
4849 	struct tog_color *tc;
4850 	int width, n, i, nentries;
4851 
4852 	*ndisplayed = 0;
4853 
4854 	werase(view->window);
4855 
4856 	if (limit == 0)
4857 		return NULL;
4858 
4859 	err = format_line(&wline, &width, label, view->ncols, 0);
4860 	if (err)
4861 		return err;
4862 	if (view_needs_focus_indication(view))
4863 		wstandout(view->window);
4864 	tc = get_color(colors, TOG_COLOR_COMMIT);
4865 	if (tc)
4866 		wattr_on(view->window,
4867 		    COLOR_PAIR(tc->colorpair), NULL);
4868 	waddwstr(view->window, wline);
4869 	if (tc)
4870 		wattr_off(view->window,
4871 		    COLOR_PAIR(tc->colorpair), NULL);
4872 	if (view_needs_focus_indication(view))
4873 		wstandend(view->window);
4874 	free(wline);
4875 	wline = NULL;
4876 	if (width < view->ncols - 1)
4877 		waddch(view->window, '\n');
4878 	if (--limit <= 0)
4879 		return NULL;
4880 	err = format_line(&wline, &width, parent_path, view->ncols, 0);
4881 	if (err)
4882 		return err;
4883 	waddwstr(view->window, wline);
4884 	free(wline);
4885 	wline = NULL;
4886 	if (width < view->ncols - 1)
4887 		waddch(view->window, '\n');
4888 	if (--limit <= 0)
4889 		return NULL;
4890 	waddch(view->window, '\n');
4891 	if (--limit <= 0)
4892 		return NULL;
4893 
4894 	if (*first_displayed_entry == NULL) {
4895 		te = got_object_tree_get_first_entry(tree);
4896 		if (selected == 0) {
4897 			if (view->focussed)
4898 				wstandout(view->window);
4899 			*selected_entry = NULL;
4900 		}
4901 		waddstr(view->window, "  ..\n");	/* parent directory */
4902 		if (selected == 0 && view->focussed)
4903 			wstandend(view->window);
4904 		(*ndisplayed)++;
4905 		if (--limit <= 0)
4906 			return NULL;
4907 		n = 1;
4908 	} else {
4909 		n = 0;
4910 		te = *first_displayed_entry;
4911 	}
4912 
4913 	nentries = got_object_tree_get_nentries(tree);
4914 	for (i = got_tree_entry_get_index(te); i < nentries; i++) {
4915 		char *line = NULL, *id_str = NULL, *link_target = NULL;
4916 		const char *modestr = "";
4917 		mode_t mode;
4918 
4919 		te = got_object_tree_get_entry(tree, i);
4920 		mode = got_tree_entry_get_mode(te);
4921 
4922 		if (show_ids) {
4923 			err = got_object_id_str(&id_str,
4924 			    got_tree_entry_get_id(te));
4925 			if (err)
4926 				return got_error_from_errno(
4927 				    "got_object_id_str");
4928 		}
4929 		if (got_object_tree_entry_is_submodule(te))
4930 			modestr = "$";
4931 		else if (S_ISLNK(mode)) {
4932 			int i;
4933 
4934 			err = got_tree_entry_get_symlink_target(&link_target,
4935 			    te, repo);
4936 			if (err) {
4937 				free(id_str);
4938 				return err;
4939 			}
4940 			for (i = 0; i < strlen(link_target); i++) {
4941 				if (!isprint((unsigned char)link_target[i]))
4942 					link_target[i] = '?';
4943 			}
4944 			modestr = "@";
4945 		}
4946 		else if (S_ISDIR(mode))
4947 			modestr = "/";
4948 		else if (mode & S_IXUSR)
4949 			modestr = "*";
4950 		if (asprintf(&line, "%s  %s%s%s%s", id_str ? id_str : "",
4951 		    got_tree_entry_get_name(te), modestr,
4952 		    link_target ? " -> ": "",
4953 		    link_target ? link_target : "") == -1) {
4954 			free(id_str);
4955 			free(link_target);
4956 			return got_error_from_errno("asprintf");
4957 		}
4958 		free(id_str);
4959 		free(link_target);
4960 		err = format_line(&wline, &width, line, view->ncols, 0);
4961 		if (err) {
4962 			free(line);
4963 			break;
4964 		}
4965 		if (n == selected) {
4966 			if (view->focussed)
4967 				wstandout(view->window);
4968 			*selected_entry = te;
4969 		}
4970 		tc = match_color(colors, line);
4971 		if (tc)
4972 			wattr_on(view->window,
4973 			    COLOR_PAIR(tc->colorpair), NULL);
4974 		waddwstr(view->window, wline);
4975 		if (tc)
4976 			wattr_off(view->window,
4977 			    COLOR_PAIR(tc->colorpair), NULL);
4978 		if (width < view->ncols - 1)
4979 			waddch(view->window, '\n');
4980 		if (n == selected && view->focussed)
4981 			wstandend(view->window);
4982 		free(line);
4983 		free(wline);
4984 		wline = NULL;
4985 		n++;
4986 		(*ndisplayed)++;
4987 		*last_displayed_entry = te;
4988 		if (--limit <= 0)
4989 			break;
4990 	}
4991 
4992 	return err;
4993 }
4994 
4995 static void
4996 tree_scroll_up(struct tog_view *view,
4997     struct got_tree_entry **first_displayed_entry, int maxscroll,
4998     struct got_tree_object *tree, int isroot)
4999 {
5000 	struct got_tree_entry *te;
5001 	int i;
5002 
5003 	if (*first_displayed_entry == NULL)
5004 		return;
5005 
5006 	te = got_object_tree_get_entry(tree, 0);
5007 	if (*first_displayed_entry == te) {
5008 		if (!isroot)
5009 			*first_displayed_entry = NULL;
5010 		return;
5011 	}
5012 
5013 	i = 0;
5014 	while (*first_displayed_entry && i < maxscroll) {
5015 		*first_displayed_entry = got_tree_entry_get_prev(tree,
5016 		    *first_displayed_entry);
5017 		i++;
5018 	}
5019 	if (!isroot && te == got_object_tree_get_first_entry(tree) && i < maxscroll)
5020 		*first_displayed_entry = NULL;
5021 }
5022 
5023 static int
5024 tree_scroll_down(struct got_tree_entry **first_displayed_entry, int maxscroll,
5025 	struct got_tree_entry *last_displayed_entry,
5026 	struct got_tree_object *tree)
5027 {
5028 	struct got_tree_entry *next, *last;
5029 	int n = 0;
5030 
5031 	if (*first_displayed_entry)
5032 		next = got_tree_entry_get_next(tree, *first_displayed_entry);
5033 	else
5034 		next = got_object_tree_get_first_entry(tree);
5035 
5036 	last = last_displayed_entry;
5037 	while (next && last && n++ < maxscroll) {
5038 		last = got_tree_entry_get_next(tree, last);
5039 		if (last) {
5040 			*first_displayed_entry = next;
5041 			next = got_tree_entry_get_next(tree, next);
5042 		}
5043 	}
5044 	return n;
5045 }
5046 
5047 static const struct got_error *
5048 tree_entry_path(char **path, struct tog_parent_trees *parents,
5049     struct got_tree_entry *te)
5050 {
5051 	const struct got_error *err = NULL;
5052 	struct tog_parent_tree *pt;
5053 	size_t len = 2; /* for leading slash and NUL */
5054 
5055 	TAILQ_FOREACH(pt, parents, entry)
5056 		len += strlen(got_tree_entry_get_name(pt->selected_entry))
5057 		    + 1 /* slash */;
5058 	if (te)
5059 		len += strlen(got_tree_entry_get_name(te));
5060 
5061 	*path = calloc(1, len);
5062 	if (path == NULL)
5063 		return got_error_from_errno("calloc");
5064 
5065 	(*path)[0] = '/';
5066 	pt = TAILQ_LAST(parents, tog_parent_trees);
5067 	while (pt) {
5068 		const char *name = got_tree_entry_get_name(pt->selected_entry);
5069 		if (strlcat(*path, name, len) >= len) {
5070 			err = got_error(GOT_ERR_NO_SPACE);
5071 			goto done;
5072 		}
5073 		if (strlcat(*path, "/", len) >= len) {
5074 			err = got_error(GOT_ERR_NO_SPACE);
5075 			goto done;
5076 		}
5077 		pt = TAILQ_PREV(pt, tog_parent_trees, entry);
5078 	}
5079 	if (te) {
5080 		if (strlcat(*path, got_tree_entry_get_name(te), len) >= len) {
5081 			err = got_error(GOT_ERR_NO_SPACE);
5082 			goto done;
5083 		}
5084 	}
5085 done:
5086 	if (err) {
5087 		free(*path);
5088 		*path = NULL;
5089 	}
5090 	return err;
5091 }
5092 
5093 static const struct got_error *
5094 blame_tree_entry(struct tog_view **new_view, int begin_x,
5095     struct got_tree_entry *te, struct tog_parent_trees *parents,
5096     struct got_object_id *commit_id, struct got_repository *repo)
5097 {
5098 	const struct got_error *err = NULL;
5099 	char *path;
5100 	struct tog_view *blame_view;
5101 
5102 	*new_view = NULL;
5103 
5104 	err = tree_entry_path(&path, parents, te);
5105 	if (err)
5106 		return err;
5107 
5108 	blame_view = view_open(0, 0, 0, begin_x, TOG_VIEW_BLAME);
5109 	if (blame_view == NULL) {
5110 		err = got_error_from_errno("view_open");
5111 		goto done;
5112 	}
5113 
5114 	err = open_blame_view(blame_view, path, commit_id, repo);
5115 	if (err) {
5116 		if (err->code == GOT_ERR_CANCELLED)
5117 			err = NULL;
5118 		view_close(blame_view);
5119 	} else
5120 		*new_view = blame_view;
5121 done:
5122 	free(path);
5123 	return err;
5124 }
5125 
5126 static const struct got_error *
5127 log_tree_entry(struct tog_view **new_view, int begin_x,
5128     struct got_tree_entry *te, struct tog_parent_trees *parents,
5129     struct got_object_id *commit_id, struct got_repository *repo)
5130 {
5131 	struct tog_view *log_view;
5132 	const struct got_error *err = NULL;
5133 	char *path;
5134 
5135 	*new_view = NULL;
5136 
5137 	log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
5138 	if (log_view == NULL)
5139 		return got_error_from_errno("view_open");
5140 
5141 	err = tree_entry_path(&path, parents, te);
5142 	if (err)
5143 		return err;
5144 
5145 	err = open_log_view(log_view, commit_id, repo, NULL, path, 0);
5146 	if (err)
5147 		view_close(log_view);
5148 	else
5149 		*new_view = log_view;
5150 	free(path);
5151 	return err;
5152 }
5153 
5154 static const struct got_error *
5155 open_tree_view(struct tog_view *view, struct got_tree_object *root,
5156     struct got_object_id *commit_id, struct got_repository *repo)
5157 {
5158 	const struct got_error *err = NULL;
5159 	char *commit_id_str = NULL;
5160 	struct tog_tree_view_state *s = &view->state.tree;
5161 
5162 	TAILQ_INIT(&s->parents);
5163 
5164 	err = got_object_id_str(&commit_id_str, commit_id);
5165 	if (err != NULL)
5166 		goto done;
5167 
5168 	if (asprintf(&s->tree_label, "commit %s", commit_id_str) == -1) {
5169 		err = got_error_from_errno("asprintf");
5170 		goto done;
5171 	}
5172 
5173 	s->root = s->tree = root;
5174 	s->first_displayed_entry = got_object_tree_get_entry(s->tree, 0);
5175 	s->selected_entry = got_object_tree_get_entry(s->tree, 0);
5176 	s->commit_id = got_object_id_dup(commit_id);
5177 	if (s->commit_id == NULL) {
5178 		err = got_error_from_errno("got_object_id_dup");
5179 		goto done;
5180 	}
5181 	s->repo = repo;
5182 
5183 	SIMPLEQ_INIT(&s->colors);
5184 
5185 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
5186 		err = add_color(&s->colors, "\\$$",
5187 		    TOG_COLOR_TREE_SUBMODULE,
5188 		    get_color_value("TOG_COLOR_TREE_SUBMODULE"));
5189 		if (err)
5190 			goto done;
5191 		err = add_color(&s->colors, "@$", TOG_COLOR_TREE_SYMLINK,
5192 		    get_color_value("TOG_COLOR_TREE_SYMLINK"));
5193 		if (err) {
5194 			free_colors(&s->colors);
5195 			goto done;
5196 		}
5197 		err = add_color(&s->colors, "/$",
5198 		    TOG_COLOR_TREE_DIRECTORY,
5199 		    get_color_value("TOG_COLOR_TREE_DIRECTORY"));
5200 		if (err) {
5201 			free_colors(&s->colors);
5202 			goto done;
5203 		}
5204 
5205 		err = add_color(&s->colors, "\\*$",
5206 		    TOG_COLOR_TREE_EXECUTABLE,
5207 		    get_color_value("TOG_COLOR_TREE_EXECUTABLE"));
5208 		if (err) {
5209 			free_colors(&s->colors);
5210 			goto done;
5211 		}
5212 
5213 		err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
5214 		    get_color_value("TOG_COLOR_COMMIT"));
5215 		if (err) {
5216 			free_colors(&s->colors);
5217 			goto done;
5218 		}
5219 	}
5220 
5221 	view->show = show_tree_view;
5222 	view->input = input_tree_view;
5223 	view->close = close_tree_view;
5224 	view->search_start = search_start_tree_view;
5225 	view->search_next = search_next_tree_view;
5226 done:
5227 	free(commit_id_str);
5228 	if (err) {
5229 		free(s->tree_label);
5230 		s->tree_label = NULL;
5231 	}
5232 	return err;
5233 }
5234 
5235 static const struct got_error *
5236 close_tree_view(struct tog_view *view)
5237 {
5238 	struct tog_tree_view_state *s = &view->state.tree;
5239 
5240 	free_colors(&s->colors);
5241 	free(s->tree_label);
5242 	s->tree_label = NULL;
5243 	free(s->commit_id);
5244 	s->commit_id = NULL;
5245 	while (!TAILQ_EMPTY(&s->parents)) {
5246 		struct tog_parent_tree *parent;
5247 		parent = TAILQ_FIRST(&s->parents);
5248 		TAILQ_REMOVE(&s->parents, parent, entry);
5249 		free(parent);
5250 
5251 	}
5252 	if (s->tree != s->root)
5253 		got_object_tree_close(s->tree);
5254 	got_object_tree_close(s->root);
5255 	return NULL;
5256 }
5257 
5258 static const struct got_error *
5259 search_start_tree_view(struct tog_view *view)
5260 {
5261 	struct tog_tree_view_state *s = &view->state.tree;
5262 
5263 	s->matched_entry = NULL;
5264 	return NULL;
5265 }
5266 
5267 static int
5268 match_tree_entry(struct got_tree_entry *te, regex_t *regex)
5269 {
5270 	regmatch_t regmatch;
5271 
5272 	return regexec(regex, got_tree_entry_get_name(te), 1, &regmatch,
5273 	    0) == 0;
5274 }
5275 
5276 static const struct got_error *
5277 search_next_tree_view(struct tog_view *view)
5278 {
5279 	struct tog_tree_view_state *s = &view->state.tree;
5280 	struct got_tree_entry *te = NULL;
5281 
5282 	if (!view->searching) {
5283 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
5284 		return NULL;
5285 	}
5286 
5287 	if (s->matched_entry) {
5288 		if (view->searching == TOG_SEARCH_FORWARD) {
5289 			if (s->selected_entry)
5290 				te = got_tree_entry_get_next(s->tree,
5291 				    s->selected_entry);
5292 			else
5293 				te = got_object_tree_get_first_entry(s->tree);
5294 		} else {
5295 			if (s->selected_entry == NULL)
5296 				te = got_object_tree_get_last_entry(s->tree);
5297 			else
5298 				te = got_tree_entry_get_prev(s->tree,
5299 				    s->selected_entry);
5300 		}
5301 	} else {
5302 		if (view->searching == TOG_SEARCH_FORWARD)
5303 			te = got_object_tree_get_first_entry(s->tree);
5304 		else
5305 			te = got_object_tree_get_last_entry(s->tree);
5306 	}
5307 
5308 	while (1) {
5309 		if (te == NULL) {
5310 			if (s->matched_entry == NULL) {
5311 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
5312 				return NULL;
5313 			}
5314 			if (view->searching == TOG_SEARCH_FORWARD)
5315 				te = got_object_tree_get_first_entry(s->tree);
5316 			else
5317 				te = got_object_tree_get_last_entry(s->tree);
5318 		}
5319 
5320 		if (match_tree_entry(te, &view->regex)) {
5321 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
5322 			s->matched_entry = te;
5323 			break;
5324 		}
5325 
5326 		if (view->searching == TOG_SEARCH_FORWARD)
5327 			te = got_tree_entry_get_next(s->tree, te);
5328 		else
5329 			te = got_tree_entry_get_prev(s->tree, te);
5330 	}
5331 
5332 	if (s->matched_entry) {
5333 		s->first_displayed_entry = s->matched_entry;
5334 		s->selected = 0;
5335 	}
5336 
5337 	return NULL;
5338 }
5339 
5340 static const struct got_error *
5341 show_tree_view(struct tog_view *view)
5342 {
5343 	const struct got_error *err = NULL;
5344 	struct tog_tree_view_state *s = &view->state.tree;
5345 	char *parent_path;
5346 
5347 	err = tree_entry_path(&parent_path, &s->parents, NULL);
5348 	if (err)
5349 		return err;
5350 
5351 	err = draw_tree_entries(view, &s->first_displayed_entry,
5352 	    &s->last_displayed_entry, &s->selected_entry,
5353 	    &s->ndisplayed, s->tree_label, s->show_ids, parent_path,
5354 	    s->tree, s->selected, view->nlines, s->tree == s->root,
5355 	    &s->colors, s->repo);
5356 	free(parent_path);
5357 
5358 	view_vborder(view);
5359 	return err;
5360 }
5361 
5362 static const struct got_error *
5363 input_tree_view(struct tog_view **new_view, struct tog_view **dead_view,
5364     struct tog_view **focus_view, struct tog_view *view, int ch)
5365 {
5366 	const struct got_error *err = NULL;
5367 	struct tog_tree_view_state *s = &view->state.tree;
5368 	struct tog_view *log_view;
5369 	int begin_x = 0, nscrolled;
5370 
5371 	switch (ch) {
5372 	case 'i':
5373 		s->show_ids = !s->show_ids;
5374 		break;
5375 	case 'l':
5376 		if (!s->selected_entry)
5377 			break;
5378 		if (view_is_parent_view(view))
5379 			begin_x = view_split_begin_x(view->begin_x);
5380 		err = log_tree_entry(&log_view, begin_x, s->selected_entry,
5381 		    &s->parents, s->commit_id, s->repo);
5382 		if (view_is_parent_view(view)) {
5383 			err = view_close_child(view);
5384 			if (err)
5385 				return err;
5386 			err = view_set_child(view, log_view);
5387 			if (err) {
5388 				view_close(log_view);
5389 				break;
5390 			}
5391 			*focus_view = log_view;
5392 			view->child_focussed = 1;
5393 		} else
5394 			*new_view = log_view;
5395 		break;
5396 	case 'k':
5397 	case KEY_UP:
5398 		if (s->selected > 0) {
5399 			s->selected--;
5400 			if (s->selected == 0)
5401 				break;
5402 		}
5403 		if (s->selected > 0)
5404 			break;
5405 		tree_scroll_up(view, &s->first_displayed_entry, 1,
5406 		    s->tree, s->tree == s->root);
5407 		break;
5408 	case KEY_PPAGE:
5409 	case CTRL('b'):
5410 		tree_scroll_up(view, &s->first_displayed_entry,
5411 		    MAX(0, view->nlines - 4 - s->selected), s->tree,
5412 		    s->tree == s->root);
5413 		s->selected = 0;
5414 		if (got_object_tree_get_first_entry(s->tree) ==
5415 		    s->first_displayed_entry && s->tree != s->root)
5416 			s->first_displayed_entry = NULL;
5417 		break;
5418 	case 'j':
5419 	case KEY_DOWN:
5420 		if (s->selected < s->ndisplayed - 1) {
5421 			s->selected++;
5422 			break;
5423 		}
5424 		if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
5425 		    == NULL)
5426 			/* can't scroll any further */
5427 			break;
5428 		tree_scroll_down(&s->first_displayed_entry, 1,
5429 		    s->last_displayed_entry, s->tree);
5430 		break;
5431 	case KEY_NPAGE:
5432 	case CTRL('f'):
5433 		if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
5434 		    == NULL) {
5435 			/* can't scroll any further; move cursor down */
5436 			if (s->selected < s->ndisplayed - 1)
5437 				s->selected = s->ndisplayed - 1;
5438 			break;
5439 		}
5440 		nscrolled = tree_scroll_down(&s->first_displayed_entry,
5441 		    view->nlines, s->last_displayed_entry, s->tree);
5442 		if (nscrolled < view->nlines) {
5443 			int ndisplayed = 0;
5444 			struct got_tree_entry *te;
5445 			te = s->first_displayed_entry;
5446 			do {
5447 				ndisplayed++;
5448 				te = got_tree_entry_get_next(s->tree, te);
5449 			} while (te);
5450 			s->selected = ndisplayed - 1;
5451 		}
5452 		break;
5453 	case KEY_ENTER:
5454 	case '\r':
5455 	case KEY_BACKSPACE:
5456 		if (s->selected_entry == NULL || ch == KEY_BACKSPACE) {
5457 			struct tog_parent_tree *parent;
5458 			/* user selected '..' */
5459 			if (s->tree == s->root)
5460 				break;
5461 			parent = TAILQ_FIRST(&s->parents);
5462 			TAILQ_REMOVE(&s->parents, parent,
5463 			    entry);
5464 			got_object_tree_close(s->tree);
5465 			s->tree = parent->tree;
5466 			s->first_displayed_entry =
5467 			    parent->first_displayed_entry;
5468 			s->selected_entry =
5469 			    parent->selected_entry;
5470 			s->selected = parent->selected;
5471 			free(parent);
5472 		} else if (S_ISDIR(got_tree_entry_get_mode(
5473 		    s->selected_entry))) {
5474 			struct got_tree_object *subtree;
5475 			err = got_object_open_as_tree(&subtree, s->repo,
5476 			    got_tree_entry_get_id(s->selected_entry));
5477 			if (err)
5478 				break;
5479 			err = tree_view_visit_subtree(subtree, s);
5480 			if (err) {
5481 				got_object_tree_close(subtree);
5482 				break;
5483 			}
5484 		} else if (S_ISREG(got_tree_entry_get_mode(
5485 		    s->selected_entry))) {
5486 			struct tog_view *blame_view;
5487 			int begin_x = view_is_parent_view(view) ?
5488 			    view_split_begin_x(view->begin_x) : 0;
5489 
5490 			err = blame_tree_entry(&blame_view, begin_x,
5491 			    s->selected_entry, &s->parents,
5492 			    s->commit_id, s->repo);
5493 			if (err)
5494 				break;
5495 			if (view_is_parent_view(view)) {
5496 				err = view_close_child(view);
5497 				if (err)
5498 					return err;
5499 				err = view_set_child(view, blame_view);
5500 				if (err) {
5501 					view_close(blame_view);
5502 					break;
5503 				}
5504 				*focus_view = blame_view;
5505 				view->child_focussed = 1;
5506 			} else
5507 				*new_view = blame_view;
5508 		}
5509 		break;
5510 	case KEY_RESIZE:
5511 		if (s->selected > view->nlines)
5512 			s->selected = s->ndisplayed - 1;
5513 		break;
5514 	default:
5515 		break;
5516 	}
5517 
5518 	return err;
5519 }
5520 
5521 __dead static void
5522 usage_tree(void)
5523 {
5524 	endwin();
5525 	fprintf(stderr, "usage: %s tree [-c commit] [-r repository-path] [path]\n",
5526 	    getprogname());
5527 	exit(1);
5528 }
5529 
5530 static const struct got_error *
5531 cmd_tree(int argc, char *argv[])
5532 {
5533 	const struct got_error *error;
5534 	struct got_repository *repo = NULL;
5535 	struct got_worktree *worktree = NULL;
5536 	char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5537 	struct got_object_id *commit_id = NULL;
5538 	char *commit_id_arg = NULL;
5539 	struct got_commit_object *commit = NULL;
5540 	struct got_tree_object *tree = NULL;
5541 	int ch;
5542 	struct tog_view *view;
5543 
5544 #ifndef PROFILE
5545 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
5546 	    NULL) == -1)
5547 		err(1, "pledge");
5548 #endif
5549 
5550 	while ((ch = getopt(argc, argv, "c:r:")) != -1) {
5551 		switch (ch) {
5552 		case 'c':
5553 			commit_id_arg = optarg;
5554 			break;
5555 		case 'r':
5556 			repo_path = realpath(optarg, NULL);
5557 			if (repo_path == NULL)
5558 				return got_error_from_errno2("realpath",
5559 				    optarg);
5560 			break;
5561 		default:
5562 			usage_tree();
5563 			/* NOTREACHED */
5564 		}
5565 	}
5566 
5567 	argc -= optind;
5568 	argv += optind;
5569 
5570 	if (argc > 1)
5571 		usage_tree();
5572 
5573 	cwd = getcwd(NULL, 0);
5574 	if (cwd == NULL)
5575 		return got_error_from_errno("getcwd");
5576 
5577 	error = got_worktree_open(&worktree, cwd);
5578 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
5579 		goto done;
5580 
5581 	if (repo_path == NULL) {
5582 		if (worktree)
5583 			repo_path =
5584 			    strdup(got_worktree_get_repo_path(worktree));
5585 		else
5586 			repo_path = strdup(cwd);
5587 	}
5588 	if (repo_path == NULL) {
5589 		error = got_error_from_errno("strdup");
5590 		goto done;
5591 	}
5592 
5593 	error = got_repo_open(&repo, repo_path, NULL);
5594 	if (error != NULL)
5595 		goto done;
5596 
5597 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
5598 	    repo, worktree);
5599 	if (error)
5600 		goto done;
5601 
5602 	init_curses();
5603 
5604 	error = apply_unveil(got_repo_get_path(repo), NULL);
5605 	if (error)
5606 		goto done;
5607 
5608 	error = got_repo_match_object_id(&commit_id, NULL,
5609 	    commit_id_arg ? commit_id_arg : GOT_REF_HEAD,
5610 	    GOT_OBJ_TYPE_COMMIT, 1, repo);
5611 	if (error)
5612 		goto done;
5613 
5614 	error = got_object_open_as_commit(&commit, repo, commit_id);
5615 	if (error)
5616 		goto done;
5617 
5618 	error = got_object_open_as_tree(&tree, repo,
5619 	    got_object_commit_get_tree_id(commit));
5620 	if (error)
5621 		goto done;
5622 
5623 	view = view_open(0, 0, 0, 0, TOG_VIEW_TREE);
5624 	if (view == NULL) {
5625 		error = got_error_from_errno("view_open");
5626 		goto done;
5627 	}
5628 	error = open_tree_view(view, tree, commit_id, repo);
5629 	if (error)
5630 		goto done;
5631 	if (!got_path_is_root_dir(in_repo_path)) {
5632 		error = tree_view_walk_path(&view->state.tree, commit_id,
5633 		    in_repo_path, repo);
5634 		if (error)
5635 			goto done;
5636 	}
5637 
5638 	if (worktree) {
5639 		/* Release work tree lock. */
5640 		got_worktree_close(worktree);
5641 		worktree = NULL;
5642 	}
5643 	error = view_loop(view);
5644 done:
5645 	free(repo_path);
5646 	free(cwd);
5647 	free(commit_id);
5648 	if (commit)
5649 		got_object_commit_close(commit);
5650 	if (tree)
5651 		got_object_tree_close(tree);
5652 	if (repo)
5653 		got_repo_close(repo);
5654 	return error;
5655 }
5656 
5657 static const struct got_error *
5658 ref_view_load_refs(struct tog_ref_view_state *s)
5659 {
5660 	const struct got_error *err;
5661 	struct got_reflist_entry *sre;
5662 	struct tog_reflist_entry *re;
5663 
5664 	err = got_ref_list(&s->simplerefs, s->repo, NULL,
5665 	    got_ref_cmp_by_name, NULL);
5666 	if (err)
5667 		return err;
5668 
5669 	s->nrefs = 0;
5670 	SIMPLEQ_FOREACH(sre, &s->simplerefs, entry) {
5671 		if (strncmp(got_ref_get_name(sre->ref), "refs/got/", 9) == 0)
5672 			continue;
5673 
5674 		re = malloc(sizeof(*re));
5675 		if (re == NULL)
5676 			return got_error_from_errno("malloc");
5677 
5678 		re->ref = sre->ref;
5679 		re->idx = s->nrefs++;
5680 		TAILQ_INSERT_TAIL(&s->refs, re, entry);
5681 	}
5682 
5683 	return NULL;
5684 }
5685 
5686 void
5687 ref_view_free_refs(struct tog_ref_view_state *s)
5688 {
5689 	struct tog_reflist_entry *re;
5690 
5691 	while (!TAILQ_EMPTY(&s->refs)) {
5692 		re = TAILQ_FIRST(&s->refs);
5693 		TAILQ_REMOVE(&s->refs, re, entry);
5694 		free(re);
5695 	}
5696 	got_ref_list_free(&s->simplerefs);
5697 }
5698 
5699 static const struct got_error *
5700 open_ref_view(struct tog_view *view, struct got_repository *repo)
5701 {
5702 	const struct got_error *err = NULL;
5703 	struct tog_ref_view_state *s = &view->state.ref;
5704 
5705 	s->first_displayed_entry = 0;
5706 	s->selected_entry = 0;
5707 	s->repo = repo;
5708 
5709 	SIMPLEQ_INIT(&s->simplerefs);
5710 	TAILQ_INIT(&s->refs);
5711 	SIMPLEQ_INIT(&s->colors);
5712 
5713 	err = ref_view_load_refs(s);
5714 	if (err)
5715 		return err;
5716 
5717 	if (has_colors() && getenv("TOG_COLORS") != NULL) {
5718 		err = add_color(&s->colors, "^refs/heads/",
5719 		    TOG_COLOR_REFS_HEADS,
5720 		    get_color_value("TOG_COLOR_REFS_HEADS"));
5721 		if (err)
5722 			goto done;
5723 
5724 		err = add_color(&s->colors, "^refs/tags/",
5725 		    TOG_COLOR_REFS_TAGS,
5726 		    get_color_value("TOG_COLOR_REFS_TAGS"));
5727 		if (err)
5728 			goto done;
5729 
5730 		err = add_color(&s->colors, "^refs/remotes/",
5731 		    TOG_COLOR_REFS_REMOTES,
5732 		    get_color_value("TOG_COLOR_REFS_REMOTES"));
5733 		if (err)
5734 			goto done;
5735 	}
5736 
5737 	view->show = show_ref_view;
5738 	view->input = input_ref_view;
5739 	view->close = close_ref_view;
5740 	view->search_start = search_start_ref_view;
5741 	view->search_next = search_next_ref_view;
5742 done:
5743 	if (err)
5744 		free_colors(&s->colors);
5745 	return err;
5746 }
5747 
5748 static const struct got_error *
5749 close_ref_view(struct tog_view *view)
5750 {
5751 	struct tog_ref_view_state *s = &view->state.ref;
5752 
5753 	ref_view_free_refs(s);
5754 	free_colors(&s->colors);
5755 
5756 	return NULL;
5757 }
5758 
5759 static const struct got_error *
5760 resolve_reflist_entry(struct got_object_id **commit_id,
5761     struct tog_reflist_entry *re, struct got_repository *repo)
5762 {
5763 	const struct got_error *err = NULL;
5764 	struct got_object_id *obj_id;
5765 	struct got_tag_object *tag = NULL;
5766 	int obj_type;
5767 
5768 	*commit_id = NULL;
5769 
5770 	err = got_ref_resolve(&obj_id, repo, re->ref);
5771 	if (err)
5772 		return err;
5773 
5774 	err = got_object_get_type(&obj_type, repo, obj_id);
5775 	if (err)
5776 		goto done;
5777 
5778 	switch (obj_type) {
5779 	case GOT_OBJ_TYPE_COMMIT:
5780 		*commit_id = obj_id;
5781 		break;
5782 	case GOT_OBJ_TYPE_TAG:
5783 		err = got_object_open_as_tag(&tag, repo, obj_id);
5784 		if (err)
5785 			goto done;
5786 		free(obj_id);
5787 		err = got_object_get_type(&obj_type, repo,
5788 		    got_object_tag_get_object_id(tag));
5789 		if (err)
5790 			goto done;
5791 		if (obj_type != GOT_OBJ_TYPE_COMMIT) {
5792 			err = got_error(GOT_ERR_OBJ_TYPE);
5793 			goto done;
5794 		}
5795 		*commit_id = got_object_id_dup(
5796 		    got_object_tag_get_object_id(tag));
5797 		if (*commit_id == NULL) {
5798 			err = got_error_from_errno("got_object_id_dup");
5799 			goto done;
5800 		}
5801 		break;
5802 	default:
5803 		err = got_error(GOT_ERR_OBJ_TYPE);
5804 		break;
5805 	}
5806 
5807 done:
5808 	if (tag)
5809 		got_object_tag_close(tag);
5810 	if (err) {
5811 		free(*commit_id);
5812 		*commit_id = NULL;
5813 	}
5814 	return err;
5815 }
5816 
5817 static const struct got_error *
5818 log_ref_entry(struct tog_view **new_view, int begin_x,
5819     struct tog_reflist_entry *re, struct got_repository *repo)
5820 {
5821 	struct tog_view *log_view;
5822 	const struct got_error *err = NULL;
5823 	struct got_object_id *commit_id = NULL;
5824 
5825 	*new_view = NULL;
5826 
5827 	err = resolve_reflist_entry(&commit_id, re, repo);
5828 	if (err) {
5829 		if (err->code != GOT_ERR_OBJ_TYPE)
5830 			return err;
5831 		else
5832 			return NULL;
5833 	}
5834 
5835 	log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
5836 	if (log_view == NULL) {
5837 		err = got_error_from_errno("view_open");
5838 		goto done;
5839 	}
5840 
5841 	err = open_log_view(log_view, commit_id, repo, NULL, "", 0);
5842 done:
5843 	if (err)
5844 		view_close(log_view);
5845 	else
5846 		*new_view = log_view;
5847 	free(commit_id);
5848 	return err;
5849 }
5850 
5851 static void
5852 ref_scroll_up(struct tog_view *view,
5853     struct tog_reflist_entry **first_displayed_entry, int maxscroll,
5854     struct tog_reflist_head *refs)
5855 {
5856 	int i;
5857 
5858 	if (*first_displayed_entry == TAILQ_FIRST(refs))
5859 		return;
5860 
5861 	i = 0;
5862 	while (*first_displayed_entry && i < maxscroll) {
5863 		*first_displayed_entry = TAILQ_PREV(*first_displayed_entry,
5864 		    tog_reflist_head, entry);
5865 		i++;
5866 	}
5867 }
5868 
5869 static int
5870 ref_scroll_down(struct tog_reflist_entry **first_displayed_entry, int maxscroll,
5871 	struct tog_reflist_entry *last_displayed_entry,
5872 	struct tog_reflist_head *refs)
5873 {
5874 	struct tog_reflist_entry *next, *last;
5875 	int n = 0;
5876 
5877 	if (*first_displayed_entry)
5878 		next = TAILQ_NEXT(*first_displayed_entry, entry);
5879 	else
5880 		next = TAILQ_FIRST(refs);
5881 
5882 	last = last_displayed_entry;
5883 	while (next && last && n++ < maxscroll) {
5884 		last = TAILQ_NEXT(last, entry);
5885 		if (last) {
5886 			*first_displayed_entry = next;
5887 			next = TAILQ_NEXT(next, entry);
5888 		}
5889 	}
5890 	return n;
5891 }
5892 
5893 static const struct got_error *
5894 search_start_ref_view(struct tog_view *view)
5895 {
5896 	struct tog_ref_view_state *s = &view->state.ref;
5897 
5898 	s->matched_entry = NULL;
5899 	return NULL;
5900 }
5901 
5902 static int
5903 match_reflist_entry(struct tog_reflist_entry *re, regex_t *regex)
5904 {
5905 	regmatch_t regmatch;
5906 
5907 	return regexec(regex, got_ref_get_name(re->ref), 1, &regmatch,
5908 	    0) == 0;
5909 }
5910 
5911 static const struct got_error *
5912 search_next_ref_view(struct tog_view *view)
5913 {
5914 	struct tog_ref_view_state *s = &view->state.ref;
5915 	struct tog_reflist_entry *re = NULL;
5916 
5917 	if (!view->searching) {
5918 		view->search_next_done = TOG_SEARCH_HAVE_MORE;
5919 		return NULL;
5920 	}
5921 
5922 	if (s->matched_entry) {
5923 		if (view->searching == TOG_SEARCH_FORWARD) {
5924 			if (s->selected_entry)
5925 				re = TAILQ_NEXT(s->selected_entry, entry);
5926 			else
5927 				re = TAILQ_PREV(s->selected_entry,
5928 				    tog_reflist_head, entry);
5929 		} else {
5930 			if (s->selected_entry == NULL)
5931 				re = TAILQ_LAST(&s->refs, tog_reflist_head);
5932 			else
5933 				re = TAILQ_PREV(s->selected_entry,
5934 				    tog_reflist_head, entry);
5935 		}
5936 	} else {
5937 		if (view->searching == TOG_SEARCH_FORWARD)
5938 			re = TAILQ_FIRST(&s->refs);
5939 		else
5940 			re = TAILQ_LAST(&s->refs, tog_reflist_head);
5941 	}
5942 
5943 	while (1) {
5944 		if (re == NULL) {
5945 			if (s->matched_entry == NULL) {
5946 				view->search_next_done = TOG_SEARCH_HAVE_MORE;
5947 				return NULL;
5948 			}
5949 			if (view->searching == TOG_SEARCH_FORWARD)
5950 				re = TAILQ_FIRST(&s->refs);
5951 			else
5952 				re = TAILQ_LAST(&s->refs, tog_reflist_head);
5953 		}
5954 
5955 		if (match_reflist_entry(re, &view->regex)) {
5956 			view->search_next_done = TOG_SEARCH_HAVE_MORE;
5957 			s->matched_entry = re;
5958 			break;
5959 		}
5960 
5961 		if (view->searching == TOG_SEARCH_FORWARD)
5962 			re = TAILQ_NEXT(re, entry);
5963 		else
5964 			re = TAILQ_PREV(re, tog_reflist_head, entry);
5965 	}
5966 
5967 	if (s->matched_entry) {
5968 		s->first_displayed_entry = s->matched_entry;
5969 		s->selected = 0;
5970 	}
5971 
5972 	return NULL;
5973 }
5974 
5975 static const struct got_error *
5976 show_ref_view(struct tog_view *view)
5977 {
5978 	const struct got_error *err = NULL;
5979 	struct tog_ref_view_state *s = &view->state.ref;
5980 	struct tog_reflist_entry *re;
5981 	char *line = NULL;
5982 	wchar_t *wline;
5983 	struct tog_color *tc;
5984 	int width, n;
5985 	int limit = view->nlines;
5986 
5987 	werase(view->window);
5988 
5989 	s->ndisplayed = 0;
5990 
5991 	if (limit == 0)
5992 		return NULL;
5993 
5994 	if (s->first_displayed_entry)
5995 		re = s->first_displayed_entry;
5996 	else
5997 		re = TAILQ_FIRST(&s->refs);
5998 
5999 	if (asprintf(&line, "references [%d/%d]", re->idx + s->selected + 1,
6000 	    s->nrefs) == -1)
6001 		return got_error_from_errno("asprintf");
6002 
6003 	err = format_line(&wline, &width, line, view->ncols, 0);
6004 	if (err) {
6005 		free(line);
6006 		return err;
6007 	}
6008 	if (view_needs_focus_indication(view))
6009 		wstandout(view->window);
6010 	waddwstr(view->window, wline);
6011 	if (view_needs_focus_indication(view))
6012 		wstandend(view->window);
6013 	free(wline);
6014 	wline = NULL;
6015 	free(line);
6016 	line = NULL;
6017 	if (width < view->ncols - 1)
6018 		waddch(view->window, '\n');
6019 	if (--limit <= 0)
6020 		return NULL;
6021 
6022 	n = 0;
6023 	while (re && limit > 0) {
6024 		char *line = NULL;
6025 
6026 		if (got_ref_is_symbolic(re->ref)) {
6027 			if (asprintf(&line, "%s -> %s",
6028 			    got_ref_get_name(re->ref),
6029 			    got_ref_get_symref_target(re->ref)) == -1)
6030 				return got_error_from_errno("asprintf");
6031 		} else if (s->show_ids) {
6032 			struct got_object_id *id;
6033 			char *id_str;
6034 			err = got_ref_resolve(&id, s->repo, re->ref);
6035 			if (err)
6036 				return err;
6037 			err = got_object_id_str(&id_str, id);
6038 			if (err) {
6039 				free(id);
6040 				return err;
6041 			}
6042 			if (asprintf(&line, "%s: %s",
6043 			    got_ref_get_name(re->ref), id_str) == -1) {
6044 				err = got_error_from_errno("asprintf");
6045 				free(id);
6046 				free(id_str);
6047 				return err;
6048 			}
6049 			free(id);
6050 			free(id_str);
6051 		} else {
6052 			line = strdup(got_ref_get_name(re->ref));
6053 			if (line == NULL)
6054 				return got_error_from_errno("strdup");
6055 		}
6056 
6057 		err = format_line(&wline, &width, line, view->ncols, 0);
6058 		if (err) {
6059 			free(line);
6060 			return err;
6061 		}
6062 		if (n == s->selected) {
6063 			if (view->focussed)
6064 				wstandout(view->window);
6065 			s->selected_entry = re;
6066 		}
6067 		tc = match_color(&s->colors, got_ref_get_name(re->ref));
6068 		if (tc)
6069 			wattr_on(view->window,
6070 			    COLOR_PAIR(tc->colorpair), NULL);
6071 		waddwstr(view->window, wline);
6072 		if (tc)
6073 			wattr_off(view->window,
6074 			    COLOR_PAIR(tc->colorpair), NULL);
6075 		if (width < view->ncols - 1)
6076 			waddch(view->window, '\n');
6077 		if (n == s->selected && view->focussed)
6078 			wstandend(view->window);
6079 		free(line);
6080 		free(wline);
6081 		wline = NULL;
6082 		n++;
6083 		s->ndisplayed++;
6084 		s->last_displayed_entry = re;
6085 
6086 		limit--;
6087 		re = TAILQ_NEXT(re, entry);
6088 	}
6089 
6090 	view_vborder(view);
6091 	return err;
6092 }
6093 
6094 static const struct got_error *
6095 browse_ref_tree(struct tog_view **new_view, int begin_x,
6096     struct tog_reflist_entry *re, struct got_repository *repo)
6097 {
6098 	const struct got_error *err = NULL;
6099 	struct got_object_id *commit_id = NULL, *tree_id = NULL;
6100 	struct got_tree_object *tree = NULL;
6101 	struct tog_view *tree_view;
6102 
6103 	*new_view = NULL;
6104 
6105 	err = resolve_reflist_entry(&commit_id, re, repo);
6106 	if (err) {
6107 		if (err->code != GOT_ERR_OBJ_TYPE)
6108 			return err;
6109 		else
6110 			return NULL;
6111 	}
6112 
6113 	err = got_object_id_by_path(&tree_id, repo, commit_id, "/");
6114 	if (err)
6115 		goto done;
6116 
6117 	err = got_object_open_as_tree(&tree, repo, tree_id);
6118 	if (err)
6119 		goto done;
6120 
6121 	tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
6122 	if (tree_view == NULL) {
6123 		err = got_error_from_errno("view_open");
6124 		goto done;
6125 	}
6126 
6127 	err = open_tree_view(tree_view, tree, commit_id, repo);
6128 	if (err)
6129 		goto done;
6130 
6131 	*new_view = tree_view;
6132 done:
6133 	free(commit_id);
6134 	free(tree_id);
6135 	if (err) {
6136 		if (tree)
6137 			got_object_tree_close(tree);
6138 	}
6139 	return err;
6140 }
6141 static const struct got_error *
6142 input_ref_view(struct tog_view **new_view, struct tog_view **dead_view,
6143     struct tog_view **focus_view, struct tog_view *view, int ch)
6144 {
6145 	const struct got_error *err = NULL;
6146 	struct tog_ref_view_state *s = &view->state.ref;
6147 	struct tog_view *log_view, *tree_view;
6148 	int begin_x = 0, nscrolled;
6149 
6150 	switch (ch) {
6151 	case 'i':
6152 		s->show_ids = !s->show_ids;
6153 		break;
6154 	case KEY_ENTER:
6155 	case '\r':
6156 		if (!s->selected_entry)
6157 			break;
6158 		if (view_is_parent_view(view))
6159 			begin_x = view_split_begin_x(view->begin_x);
6160 		err = log_ref_entry(&log_view, begin_x, s->selected_entry,
6161 		    s->repo);
6162 		if (view_is_parent_view(view)) {
6163 			err = view_close_child(view);
6164 			if (err)
6165 				return err;
6166 			err = view_set_child(view, log_view);
6167 			if (err) {
6168 				view_close(log_view);
6169 				break;
6170 			}
6171 			*focus_view = log_view;
6172 			view->child_focussed = 1;
6173 		} else
6174 			*new_view = log_view;
6175 		break;
6176 	case 't':
6177 		if (!s->selected_entry)
6178 			break;
6179 		if (view_is_parent_view(view))
6180 			begin_x = view_split_begin_x(view->begin_x);
6181 		err = browse_ref_tree(&tree_view, begin_x, s->selected_entry,
6182 		    s->repo);
6183 		if (err || tree_view == NULL)
6184 			break;
6185 		if (view_is_parent_view(view)) {
6186 			err = view_close_child(view);
6187 			if (err)
6188 				return err;
6189 			err = view_set_child(view, tree_view);
6190 			if (err) {
6191 				view_close(tree_view);
6192 				break;
6193 			}
6194 			*focus_view = tree_view;
6195 			view->child_focussed = 1;
6196 		} else
6197 			*new_view = tree_view;
6198 		break;
6199 	case 'k':
6200 	case KEY_UP:
6201 		if (s->selected > 0) {
6202 			s->selected--;
6203 			if (s->selected == 0)
6204 				break;
6205 		}
6206 		if (s->selected > 0)
6207 			break;
6208 		ref_scroll_up(view, &s->first_displayed_entry, 1, &s->refs);
6209 		break;
6210 	case KEY_PPAGE:
6211 	case CTRL('b'):
6212 		ref_scroll_up(view, &s->first_displayed_entry,
6213 		    MAX(0, view->nlines - 4 - s->selected), &s->refs);
6214 		s->selected = 0;
6215 		break;
6216 	case 'j':
6217 	case KEY_DOWN:
6218 		if (s->selected < s->ndisplayed - 1) {
6219 			s->selected++;
6220 			break;
6221 		}
6222 		if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL)
6223 			/* can't scroll any further */
6224 			break;
6225 		ref_scroll_down(&s->first_displayed_entry, 1,
6226 		    s->last_displayed_entry, &s->refs);
6227 		break;
6228 	case KEY_NPAGE:
6229 	case CTRL('f'):
6230 		if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL) {
6231 			/* can't scroll any further; move cursor down */
6232 			if (s->selected < s->ndisplayed - 1)
6233 				s->selected = s->ndisplayed - 1;
6234 			break;
6235 		}
6236 		nscrolled = ref_scroll_down(&s->first_displayed_entry,
6237 		    view->nlines, s->last_displayed_entry, &s->refs);
6238 		if (nscrolled < view->nlines) {
6239 			int ndisplayed = 0;
6240 			struct tog_reflist_entry *re;
6241 			re = s->first_displayed_entry;
6242 			do {
6243 				ndisplayed++;
6244 				re = TAILQ_NEXT(re, entry);
6245 			} while (re);
6246 			s->selected = ndisplayed - 1;
6247 		}
6248 		break;
6249 	case CTRL('l'):
6250 		ref_view_free_refs(s);
6251 		err = ref_view_load_refs(s);
6252 		break;
6253 	case KEY_RESIZE:
6254 		if (s->selected > view->nlines)
6255 			s->selected = s->ndisplayed - 1;
6256 		break;
6257 	default:
6258 		break;
6259 	}
6260 
6261 	return err;
6262 }
6263 
6264 __dead static void
6265 usage_ref(void)
6266 {
6267 	endwin();
6268 	fprintf(stderr, "usage: %s ref [-r repository-path]\n",
6269 	    getprogname());
6270 	exit(1);
6271 }
6272 
6273 static const struct got_error *
6274 cmd_ref(int argc, char *argv[])
6275 {
6276 	const struct got_error *error;
6277 	struct got_repository *repo = NULL;
6278 	struct got_worktree *worktree = NULL;
6279 	char *cwd = NULL, *repo_path = NULL;
6280 	int ch;
6281 	struct tog_view *view;
6282 
6283 #ifndef PROFILE
6284 	if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
6285 	    NULL) == -1)
6286 		err(1, "pledge");
6287 #endif
6288 
6289 	while ((ch = getopt(argc, argv, "r:")) != -1) {
6290 		switch (ch) {
6291 		case 'r':
6292 			repo_path = realpath(optarg, NULL);
6293 			if (repo_path == NULL)
6294 				return got_error_from_errno2("realpath",
6295 				    optarg);
6296 			break;
6297 		default:
6298 			usage_ref();
6299 			/* NOTREACHED */
6300 		}
6301 	}
6302 
6303 	argc -= optind;
6304 	argv += optind;
6305 
6306 	if (argc > 1)
6307 		usage_ref();
6308 
6309 	cwd = getcwd(NULL, 0);
6310 	if (cwd == NULL)
6311 		return got_error_from_errno("getcwd");
6312 
6313 	error = got_worktree_open(&worktree, cwd);
6314 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
6315 		goto done;
6316 
6317 	if (repo_path == NULL) {
6318 		if (worktree)
6319 			repo_path =
6320 			    strdup(got_worktree_get_repo_path(worktree));
6321 		else
6322 			repo_path = strdup(cwd);
6323 	}
6324 	if (repo_path == NULL) {
6325 		error = got_error_from_errno("strdup");
6326 		goto done;
6327 	}
6328 
6329 	error = got_repo_open(&repo, repo_path, NULL);
6330 	if (error != NULL)
6331 		goto done;
6332 
6333 	init_curses();
6334 
6335 	error = apply_unveil(got_repo_get_path(repo), NULL);
6336 	if (error)
6337 		goto done;
6338 
6339 	view = view_open(0, 0, 0, 0, TOG_VIEW_REF);
6340 	if (view == NULL) {
6341 		error = got_error_from_errno("view_open");
6342 		goto done;
6343 	}
6344 
6345 	error = open_ref_view(view, repo);
6346 	if (error)
6347 		goto done;
6348 
6349 	if (worktree) {
6350 		/* Release work tree lock. */
6351 		got_worktree_close(worktree);
6352 		worktree = NULL;
6353 	}
6354 	error = view_loop(view);
6355 done:
6356 	free(repo_path);
6357 	free(cwd);
6358 	if (repo)
6359 		got_repo_close(repo);
6360 	return error;
6361 }
6362 
6363 static void
6364 list_commands(FILE *fp)
6365 {
6366 	int i;
6367 
6368 	fprintf(fp, "commands:");
6369 	for (i = 0; i < nitems(tog_commands); i++) {
6370 		struct tog_cmd *cmd = &tog_commands[i];
6371 		fprintf(fp, " %s", cmd->name);
6372 	}
6373 	fputc('\n', fp);
6374 }
6375 
6376 __dead static void
6377 usage(int hflag, int status)
6378 {
6379 	FILE *fp = (status == 0) ? stdout : stderr;
6380 
6381 	fprintf(fp, "usage: %s [-h] [-V | --version] [command] [arg ...]\n",
6382 	    getprogname());
6383 	if (hflag) {
6384 		fprintf(fp, "lazy usage: %s path\n", getprogname());
6385 		list_commands(fp);
6386 	}
6387 	exit(status);
6388 }
6389 
6390 static char **
6391 make_argv(int argc, ...)
6392 {
6393 	va_list ap;
6394 	char **argv;
6395 	int i;
6396 
6397 	va_start(ap, argc);
6398 
6399 	argv = calloc(argc, sizeof(char *));
6400 	if (argv == NULL)
6401 		err(1, "calloc");
6402 	for (i = 0; i < argc; i++) {
6403 		argv[i] = strdup(va_arg(ap, char *));
6404 		if (argv[i] == NULL)
6405 			err(1, "strdup");
6406 	}
6407 
6408 	va_end(ap);
6409 	return argv;
6410 }
6411 
6412 /*
6413  * Try to convert 'tog path' into a 'tog log path' command.
6414  * The user could simply have mistyped the command rather than knowingly
6415  * provided a path. So check whether argv[0] can in fact be resolved
6416  * to a path in the HEAD commit and print a special error if not.
6417  * This hack is for mpi@ <3
6418  */
6419 static const struct got_error *
6420 tog_log_with_path(int argc, char *argv[])
6421 {
6422 	const struct got_error *error = NULL;
6423 	struct tog_cmd *cmd = NULL;
6424 	struct got_repository *repo = NULL;
6425 	struct got_worktree *worktree = NULL;
6426 	struct got_object_id *commit_id = NULL, *id = NULL;
6427 	char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
6428 	char *commit_id_str = NULL, **cmd_argv = NULL;
6429 
6430 	cwd = getcwd(NULL, 0);
6431 	if (cwd == NULL)
6432 		return got_error_from_errno("getcwd");
6433 
6434 	error = got_worktree_open(&worktree, cwd);
6435 	if (error && error->code != GOT_ERR_NOT_WORKTREE)
6436 		goto done;
6437 
6438 	if (worktree)
6439 		repo_path = strdup(got_worktree_get_repo_path(worktree));
6440 	else
6441 		repo_path = strdup(cwd);
6442 	if (repo_path == NULL) {
6443 		error = got_error_from_errno("strdup");
6444 		goto done;
6445 	}
6446 
6447 	error = got_repo_open(&repo, repo_path, NULL);
6448 	if (error != NULL)
6449 		goto done;
6450 
6451 	error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
6452 	    repo, worktree);
6453 	if (error)
6454 		goto done;
6455 
6456 	error = got_repo_match_object_id(&commit_id, NULL, worktree ?
6457 	    got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
6458 	    GOT_OBJ_TYPE_COMMIT, 1, repo);
6459 	if (error)
6460 		goto done;
6461 
6462 	if (worktree) {
6463 		got_worktree_close(worktree);
6464 		worktree = NULL;
6465 	}
6466 
6467 	error = got_object_id_by_path(&id, repo, commit_id, in_repo_path);
6468 	if (error) {
6469 		if (error->code != GOT_ERR_NO_TREE_ENTRY)
6470 			goto done;
6471 		fprintf(stderr, "%s: '%s' is no known command or path\n",
6472 		    getprogname(), argv[0]);
6473 		usage(1, 1);
6474 		/* not reached */
6475 	}
6476 
6477 	got_repo_close(repo);
6478 	repo = NULL;
6479 
6480 	error = got_object_id_str(&commit_id_str, commit_id);
6481 	if (error)
6482 		goto done;
6483 
6484 	cmd = &tog_commands[0]; /* log */
6485 	argc = 4;
6486 	cmd_argv = make_argv(argc, cmd->name, "-c", commit_id_str, argv[0]);
6487 	error = cmd->cmd_main(argc, cmd_argv);
6488 done:
6489 	if (repo)
6490 		got_repo_close(repo);
6491 	if (worktree)
6492 		got_worktree_close(worktree);
6493 	free(id);
6494 	free(commit_id_str);
6495 	free(commit_id);
6496 	free(cwd);
6497 	free(repo_path);
6498 	free(in_repo_path);
6499 	if (cmd_argv) {
6500 		int i;
6501 		for (i = 0; i < argc; i++)
6502 			free(cmd_argv[i]);
6503 		free(cmd_argv);
6504 	}
6505 	return error;
6506 }
6507 
6508 int
6509 main(int argc, char *argv[])
6510 {
6511 	const struct got_error *error = NULL;
6512 	struct tog_cmd *cmd = NULL;
6513 	int ch, hflag = 0, Vflag = 0;
6514 	char **cmd_argv = NULL;
6515 	static struct option longopts[] = {
6516 	    { "version", no_argument, NULL, 'V' },
6517 	    { NULL, 0, NULL, 0}
6518 	};
6519 
6520 	setlocale(LC_CTYPE, "");
6521 
6522 	while ((ch = getopt_long(argc, argv, "+hV", longopts, NULL)) != -1) {
6523 		switch (ch) {
6524 		case 'h':
6525 			hflag = 1;
6526 			break;
6527 		case 'V':
6528 			Vflag = 1;
6529 			break;
6530 		default:
6531 			usage(hflag, 1);
6532 			/* NOTREACHED */
6533 		}
6534 	}
6535 
6536 	argc -= optind;
6537 	argv += optind;
6538 	optind = 1;
6539 	optreset = 1;
6540 
6541 	if (Vflag) {
6542 		got_version_print_str();
6543 		return 0;
6544 	}
6545 
6546 	if (argc == 0) {
6547 		if (hflag)
6548 			usage(hflag, 0);
6549 		/* Build an argument vector which runs a default command. */
6550 		cmd = &tog_commands[0];
6551 		argc = 1;
6552 		cmd_argv = make_argv(argc, cmd->name);
6553 	} else {
6554 		int i;
6555 
6556 		/* Did the user specify a command? */
6557 		for (i = 0; i < nitems(tog_commands); i++) {
6558 			if (strncmp(tog_commands[i].name, argv[0],
6559 			    strlen(argv[0])) == 0) {
6560 				cmd = &tog_commands[i];
6561 				break;
6562 			}
6563 		}
6564 	}
6565 
6566 	if (cmd == NULL) {
6567 		if (argc != 1)
6568 			usage(0, 1);
6569 		/* No command specified; try log with a path */
6570 		error = tog_log_with_path(argc, argv);
6571 	} else {
6572 		if (hflag)
6573 			cmd->cmd_usage();
6574 		else
6575 			error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
6576 	}
6577 
6578 	endwin();
6579 	putchar('\n');
6580 	if (cmd_argv) {
6581 		int i;
6582 		for (i = 0; i < argc; i++)
6583 			free(cmd_argv[i]);
6584 		free(cmd_argv);
6585 	}
6586 
6587 	if (error && error->code != GOT_ERR_CANCELLED)
6588 		fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
6589 	return 0;
6590 }