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