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