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