Blob


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 */
17 #include <sys/queue.h>
18 #include <sys/stat.h>
19 #include <sys/ioctl.h>
21 #include <ctype.h>
22 #include <errno.h>
23 #define _XOPEN_SOURCE_EXTENDED
24 #include <curses.h>
25 #undef _XOPEN_SOURCE_EXTENDED
26 #include <panel.h>
27 #include <locale.h>
28 #include <signal.h>
29 #include <stdlib.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <getopt.h>
33 #include <string.h>
34 #include <err.h>
35 #include <unistd.h>
36 #include <util.h>
37 #include <limits.h>
38 #include <wchar.h>
39 #include <time.h>
40 #include <pthread.h>
41 #include <libgen.h>
42 #include <regex.h>
44 #include "got_version.h"
45 #include "got_error.h"
46 #include "got_object.h"
47 #include "got_reference.h"
48 #include "got_repository.h"
49 #include "got_diff.h"
50 #include "got_opentemp.h"
51 #include "got_utf8.h"
52 #include "got_cancel.h"
53 #include "got_commit_graph.h"
54 #include "got_blame.h"
55 #include "got_privsep.h"
56 #include "got_path.h"
57 #include "got_worktree.h"
59 #ifndef MIN
60 #define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
61 #endif
63 #ifndef MAX
64 #define MAX(_a,_b) ((_a) > (_b) ? (_a) : (_b))
65 #endif
67 #define CTRL(x) ((x) & 0x1f)
69 #ifndef nitems
70 #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
71 #endif
73 struct tog_cmd {
74 const char *name;
75 const struct got_error *(*cmd_main)(int, char *[]);
76 void (*cmd_usage)(void);
77 };
79 __dead static void usage(int, int);
80 __dead static void usage_log(void);
81 __dead static void usage_diff(void);
82 __dead static void usage_blame(void);
83 __dead static void usage_tree(void);
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 *[]);
90 static struct tog_cmd tog_commands[] = {
91 { "log", cmd_log, usage_log },
92 { "diff", cmd_diff, usage_diff },
93 { "blame", cmd_blame, usage_blame },
94 { "tree", cmd_tree, usage_tree },
95 };
97 enum tog_view_type {
98 TOG_VIEW_DIFF,
99 TOG_VIEW_LOG,
100 TOG_VIEW_BLAME,
101 TOG_VIEW_TREE
102 };
104 #define TOG_EOF_STRING "(END)"
106 struct commit_queue_entry {
107 TAILQ_ENTRY(commit_queue_entry) entry;
108 struct got_object_id *id;
109 struct got_commit_object *commit;
110 int idx;
111 };
112 TAILQ_HEAD(commit_queue_head, commit_queue_entry);
113 struct commit_queue {
114 int ncommits;
115 struct commit_queue_head head;
116 };
118 struct tog_color {
119 SIMPLEQ_ENTRY(tog_color) entry;
120 regex_t regex;
121 short colorpair;
122 };
123 SIMPLEQ_HEAD(tog_colors, tog_color);
125 static const struct got_error *
126 add_color(struct tog_colors *colors, const char *pattern,
127 int idx, short color)
129 const struct got_error *err = NULL;
130 struct tog_color *tc;
131 int regerr = 0;
133 if (idx < 1 || idx > COLOR_PAIRS - 1)
134 return NULL;
136 init_pair(idx, color, -1);
138 tc = calloc(1, sizeof(*tc));
139 if (tc == NULL)
140 return got_error_from_errno("calloc");
141 regerr = regcomp(&tc->regex, pattern,
142 REG_EXTENDED | REG_NOSUB | REG_NEWLINE);
143 if (regerr) {
144 static char regerr_msg[512];
145 static char err_msg[512];
146 regerror(regerr, &tc->regex, regerr_msg,
147 sizeof(regerr_msg));
148 snprintf(err_msg, sizeof(err_msg), "regcomp: %s",
149 regerr_msg);
150 err = got_error_msg(GOT_ERR_REGEX, err_msg);
151 free(tc);
152 return err;
154 tc->colorpair = idx;
155 SIMPLEQ_INSERT_HEAD(colors, tc, entry);
156 return NULL;
159 static void
160 free_colors(struct tog_colors *colors)
162 struct tog_color *tc;
164 while (!SIMPLEQ_EMPTY(colors)) {
165 tc = SIMPLEQ_FIRST(colors);
166 SIMPLEQ_REMOVE_HEAD(colors, entry);
167 regfree(&tc->regex);
168 free(tc);
172 struct tog_color *
173 get_color(struct tog_colors *colors, int colorpair)
175 struct tog_color *tc = NULL;
177 SIMPLEQ_FOREACH(tc, colors, entry) {
178 if (tc->colorpair == colorpair)
179 return tc;
182 return NULL;
185 static int
186 default_color_value(const char *envvar)
188 if (strcmp(envvar, "TOG_COLOR_DIFF_MINUS") == 0)
189 return COLOR_MAGENTA;
190 if (strcmp(envvar, "TOG_COLOR_DIFF_PLUS") == 0)
191 return COLOR_CYAN;
192 if (strcmp(envvar, "TOG_COLOR_DIFF_CHUNK_HEADER") == 0)
193 return COLOR_YELLOW;
194 if (strcmp(envvar, "TOG_COLOR_DIFF_META") == 0)
195 return COLOR_GREEN;
196 if (strcmp(envvar, "TOG_COLOR_TREE_SUBMODULE") == 0)
197 return COLOR_MAGENTA;
198 if (strcmp(envvar, "TOG_COLOR_TREE_SYMLINK") == 0)
199 return COLOR_MAGENTA;
200 if (strcmp(envvar, "TOG_COLOR_TREE_DIRECTORY") == 0)
201 return COLOR_CYAN;
202 if (strcmp(envvar, "TOG_COLOR_TREE_EXECUTABLE") == 0)
203 return COLOR_GREEN;
204 if (strcmp(envvar, "TOG_COLOR_COMMIT") == 0)
205 return COLOR_GREEN;
206 if (strcmp(envvar, "TOG_COLOR_AUTHOR") == 0)
207 return COLOR_CYAN;
208 if (strcmp(envvar, "TOG_COLOR_DATE") == 0)
209 return COLOR_YELLOW;
211 return -1;
214 static int
215 get_color_value(const char *envvar)
217 const char *val = getenv(envvar);
219 if (val == NULL)
220 return default_color_value(envvar);
222 if (strcasecmp(val, "black") == 0)
223 return COLOR_BLACK;
224 if (strcasecmp(val, "red") == 0)
225 return COLOR_RED;
226 if (strcasecmp(val, "green") == 0)
227 return COLOR_GREEN;
228 if (strcasecmp(val, "yellow") == 0)
229 return COLOR_YELLOW;
230 if (strcasecmp(val, "blue") == 0)
231 return COLOR_BLUE;
232 if (strcasecmp(val, "magenta") == 0)
233 return COLOR_MAGENTA;
234 if (strcasecmp(val, "cyan") == 0)
235 return COLOR_CYAN;
236 if (strcasecmp(val, "white") == 0)
237 return COLOR_WHITE;
238 if (strcasecmp(val, "default") == 0)
239 return -1;
241 return default_color_value(envvar);
245 struct tog_diff_view_state {
246 struct got_object_id *id1, *id2;
247 FILE *f;
248 int first_displayed_line;
249 int last_displayed_line;
250 int eof;
251 int diff_context;
252 struct got_repository *repo;
253 struct got_reflist_head *refs;
254 struct tog_colors colors;
255 size_t nlines;
256 off_t *line_offsets;
257 int matched_line;
258 int selected_line;
260 /* passed from log view; may be NULL */
261 struct tog_view *log_view;
262 };
264 pthread_mutex_t tog_mutex = PTHREAD_MUTEX_INITIALIZER;
266 struct tog_log_thread_args {
267 pthread_cond_t need_commits;
268 pthread_cond_t commit_loaded;
269 int commits_needed;
270 struct got_commit_graph *graph;
271 struct commit_queue *commits;
272 const char *in_repo_path;
273 struct got_object_id *start_id;
274 struct got_repository *repo;
275 int log_complete;
276 sig_atomic_t *quit;
277 struct commit_queue_entry **first_displayed_entry;
278 struct commit_queue_entry **selected_entry;
279 int *searching;
280 int *search_next_done;
281 regex_t *regex;
282 };
284 struct tog_log_view_state {
285 struct commit_queue commits;
286 struct commit_queue_entry *first_displayed_entry;
287 struct commit_queue_entry *last_displayed_entry;
288 struct commit_queue_entry *selected_entry;
289 int selected;
290 char *in_repo_path;
291 const char *head_ref_name;
292 int log_branches;
293 struct got_repository *repo;
294 struct got_reflist_head *refs;
295 struct got_object_id *start_id;
296 sig_atomic_t quit;
297 pthread_t thread;
298 struct tog_log_thread_args thread_args;
299 struct commit_queue_entry *matched_entry;
300 struct commit_queue_entry *search_entry;
301 struct tog_colors colors;
302 };
304 #define TOG_COLOR_DIFF_MINUS 1
305 #define TOG_COLOR_DIFF_PLUS 2
306 #define TOG_COLOR_DIFF_CHUNK_HEADER 3
307 #define TOG_COLOR_DIFF_META 4
308 #define TOG_COLOR_TREE_SUBMODULE 5
309 #define TOG_COLOR_TREE_SYMLINK 6
310 #define TOG_COLOR_TREE_DIRECTORY 7
311 #define TOG_COLOR_TREE_EXECUTABLE 8
312 #define TOG_COLOR_COMMIT 9
313 #define TOG_COLOR_AUTHOR 10
314 #define TOG_COLOR_DATE 11
316 struct tog_blame_cb_args {
317 struct tog_blame_line *lines; /* one per line */
318 int nlines;
320 struct tog_view *view;
321 struct got_object_id *commit_id;
322 int *quit;
323 };
325 struct tog_blame_thread_args {
326 const char *path;
327 struct got_repository *repo;
328 struct tog_blame_cb_args *cb_args;
329 int *complete;
330 got_cancel_cb cancel_cb;
331 void *cancel_arg;
332 };
334 struct tog_blame {
335 FILE *f;
336 size_t filesize;
337 struct tog_blame_line *lines;
338 int nlines;
339 off_t *line_offsets;
340 pthread_t thread;
341 struct tog_blame_thread_args thread_args;
342 struct tog_blame_cb_args cb_args;
343 const char *path;
344 };
346 struct tog_blame_view_state {
347 int first_displayed_line;
348 int last_displayed_line;
349 int selected_line;
350 int blame_complete;
351 int eof;
352 int done;
353 struct got_object_id_queue blamed_commits;
354 struct got_object_qid *blamed_commit;
355 char *path;
356 struct got_repository *repo;
357 struct got_reflist_head *refs;
358 struct got_object_id *commit_id;
359 struct tog_blame blame;
360 int matched_line;
361 struct tog_colors colors;
362 };
364 struct tog_parent_tree {
365 TAILQ_ENTRY(tog_parent_tree) entry;
366 struct got_tree_object *tree;
367 struct got_tree_entry *first_displayed_entry;
368 struct got_tree_entry *selected_entry;
369 int selected;
370 };
372 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
374 struct tog_tree_view_state {
375 char *tree_label;
376 struct got_tree_object *root;
377 struct got_tree_object *tree;
378 struct got_tree_entry *first_displayed_entry;
379 struct got_tree_entry *last_displayed_entry;
380 struct got_tree_entry *selected_entry;
381 int ndisplayed, selected, show_ids;
382 struct tog_parent_trees parents;
383 struct got_object_id *commit_id;
384 struct got_repository *repo;
385 struct got_reflist_head *refs;
386 struct got_tree_entry *matched_entry;
387 struct tog_colors colors;
388 };
390 /*
391 * We implement two types of views: parent views and child views.
393 * The 'Tab' key switches between a parent view and its child view.
394 * Child views are shown side-by-side to their parent view, provided
395 * there is enough screen estate.
397 * When a new view is opened from within a parent view, this new view
398 * becomes a child view of the parent view, replacing any existing child.
400 * When a new view is opened from within a child view, this new view
401 * becomes a parent view which will obscure the views below until the
402 * user quits the new parent view by typing 'q'.
404 * This list of views contains parent views only.
405 * Child views are only pointed to by their parent view.
406 */
407 TAILQ_HEAD(tog_view_list_head, tog_view);
409 struct tog_view {
410 TAILQ_ENTRY(tog_view) entry;
411 WINDOW *window;
412 PANEL *panel;
413 int nlines, ncols, begin_y, begin_x;
414 int lines, cols; /* copies of LINES and COLS */
415 int focussed;
416 struct tog_view *parent;
417 struct tog_view *child;
418 int child_focussed;
420 /* type-specific state */
421 enum tog_view_type type;
422 union {
423 struct tog_diff_view_state diff;
424 struct tog_log_view_state log;
425 struct tog_blame_view_state blame;
426 struct tog_tree_view_state tree;
427 } state;
429 const struct got_error *(*show)(struct tog_view *);
430 const struct got_error *(*input)(struct tog_view **,
431 struct tog_view **, struct tog_view**, struct tog_view *, int);
432 const struct got_error *(*close)(struct tog_view *);
434 const struct got_error *(*search_start)(struct tog_view *);
435 const struct got_error *(*search_next)(struct tog_view *);
436 int searching;
437 #define TOG_SEARCH_FORWARD 1
438 #define TOG_SEARCH_BACKWARD 2
439 int search_next_done;
440 #define TOG_SEARCH_HAVE_MORE 1
441 #define TOG_SEARCH_NO_MORE 2
442 #define TOG_SEARCH_HAVE_NONE 3
443 regex_t regex;
444 };
446 static const struct got_error *open_diff_view(struct tog_view *,
447 struct got_object_id *, struct got_object_id *, struct tog_view *,
448 struct got_reflist_head *, struct got_repository *);
449 static const struct got_error *show_diff_view(struct tog_view *);
450 static const struct got_error *input_diff_view(struct tog_view **,
451 struct tog_view **, struct tog_view **, struct tog_view *, int);
452 static const struct got_error* close_diff_view(struct tog_view *);
453 static const struct got_error *search_start_diff_view(struct tog_view *);
454 static const struct got_error *search_next_diff_view(struct tog_view *);
456 static const struct got_error *open_log_view(struct tog_view *,
457 struct got_object_id *, struct got_reflist_head *,
458 struct got_repository *, const char *, const char *, int);
459 static const struct got_error * show_log_view(struct tog_view *);
460 static const struct got_error *input_log_view(struct tog_view **,
461 struct tog_view **, struct tog_view **, struct tog_view *, int);
462 static const struct got_error *close_log_view(struct tog_view *);
463 static const struct got_error *search_start_log_view(struct tog_view *);
464 static const struct got_error *search_next_log_view(struct tog_view *);
466 static const struct got_error *open_blame_view(struct tog_view *, char *,
467 struct got_object_id *, struct got_reflist_head *, struct got_repository *);
468 static const struct got_error *show_blame_view(struct tog_view *);
469 static const struct got_error *input_blame_view(struct tog_view **,
470 struct tog_view **, struct tog_view **, struct tog_view *, int);
471 static const struct got_error *close_blame_view(struct tog_view *);
472 static const struct got_error *search_start_blame_view(struct tog_view *);
473 static const struct got_error *search_next_blame_view(struct tog_view *);
475 static const struct got_error *open_tree_view(struct tog_view *,
476 struct got_tree_object *, struct got_object_id *, struct got_reflist_head *,
477 struct got_repository *);
478 static const struct got_error *show_tree_view(struct tog_view *);
479 static const struct got_error *input_tree_view(struct tog_view **,
480 struct tog_view **, struct tog_view **, struct tog_view *, int);
481 static const struct got_error *close_tree_view(struct tog_view *);
482 static const struct got_error *search_start_tree_view(struct tog_view *);
483 static const struct got_error *search_next_tree_view(struct tog_view *);
485 static volatile sig_atomic_t tog_sigwinch_received;
486 static volatile sig_atomic_t tog_sigpipe_received;
487 static volatile sig_atomic_t tog_sigcont_received;
489 static void
490 tog_sigwinch(int signo)
492 tog_sigwinch_received = 1;
495 static void
496 tog_sigpipe(int signo)
498 tog_sigpipe_received = 1;
501 static void
502 tog_sigcont(int signo)
504 tog_sigcont_received = 1;
507 static const struct got_error *
508 view_close(struct tog_view *view)
510 const struct got_error *err = NULL;
512 if (view->child) {
513 view_close(view->child);
514 view->child = NULL;
516 if (view->close)
517 err = view->close(view);
518 if (view->panel)
519 del_panel(view->panel);
520 if (view->window)
521 delwin(view->window);
522 free(view);
523 return err;
526 static struct tog_view *
527 view_open(int nlines, int ncols, int begin_y, int begin_x,
528 enum tog_view_type type)
530 struct tog_view *view = calloc(1, sizeof(*view));
532 if (view == NULL)
533 return NULL;
535 view->type = type;
536 view->lines = LINES;
537 view->cols = COLS;
538 view->nlines = nlines ? nlines : LINES - begin_y;
539 view->ncols = ncols ? ncols : COLS - begin_x;
540 view->begin_y = begin_y;
541 view->begin_x = begin_x;
542 view->window = newwin(nlines, ncols, begin_y, begin_x);
543 if (view->window == NULL) {
544 view_close(view);
545 return NULL;
547 view->panel = new_panel(view->window);
548 if (view->panel == NULL ||
549 set_panel_userptr(view->panel, view) != OK) {
550 view_close(view);
551 return NULL;
554 keypad(view->window, TRUE);
555 return view;
558 static int
559 view_split_begin_x(int begin_x)
561 if (begin_x > 0 || COLS < 120)
562 return 0;
563 return (COLS - MAX(COLS / 2, 80));
566 static const struct got_error *view_resize(struct tog_view *);
568 static const struct got_error *
569 view_splitscreen(struct tog_view *view)
571 const struct got_error *err = NULL;
573 view->begin_y = 0;
574 view->begin_x = view_split_begin_x(0);
575 view->nlines = LINES;
576 view->ncols = COLS - view->begin_x;
577 view->lines = LINES;
578 view->cols = COLS;
579 err = view_resize(view);
580 if (err)
581 return err;
583 if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
584 return got_error_from_errno("mvwin");
586 return NULL;
589 static const struct got_error *
590 view_fullscreen(struct tog_view *view)
592 const struct got_error *err = NULL;
594 view->begin_x = 0;
595 view->begin_y = 0;
596 view->nlines = LINES;
597 view->ncols = COLS;
598 view->lines = LINES;
599 view->cols = COLS;
600 err = view_resize(view);
601 if (err)
602 return err;
604 if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
605 return got_error_from_errno("mvwin");
607 return NULL;
610 static int
611 view_is_parent_view(struct tog_view *view)
613 return view->parent == NULL;
616 static const struct got_error *
617 view_resize(struct tog_view *view)
619 int nlines, ncols;
621 if (view->lines > LINES)
622 nlines = view->nlines - (view->lines - LINES);
623 else
624 nlines = view->nlines + (LINES - view->lines);
626 if (view->cols > COLS)
627 ncols = view->ncols - (view->cols - COLS);
628 else
629 ncols = view->ncols + (COLS - view->cols);
631 if (wresize(view->window, nlines, ncols) == ERR)
632 return got_error_from_errno("wresize");
633 if (replace_panel(view->panel, view->window) == ERR)
634 return got_error_from_errno("replace_panel");
635 wclear(view->window);
637 view->nlines = nlines;
638 view->ncols = ncols;
639 view->lines = LINES;
640 view->cols = COLS;
642 if (view->child) {
643 view->child->begin_x = view_split_begin_x(view->begin_x);
644 if (view->child->begin_x == 0) {
645 view_fullscreen(view->child);
646 if (view->child->focussed)
647 show_panel(view->child->panel);
648 else
649 show_panel(view->panel);
650 } else {
651 view_splitscreen(view->child);
652 show_panel(view->child->panel);
656 return NULL;
659 static const struct got_error *
660 view_close_child(struct tog_view *view)
662 const struct got_error *err = NULL;
664 if (view->child == NULL)
665 return NULL;
667 err = view_close(view->child);
668 view->child = NULL;
669 return err;
672 static const struct got_error *
673 view_set_child(struct tog_view *view, struct tog_view *child)
675 const struct got_error *err = NULL;
677 view->child = child;
678 child->parent = view;
679 return err;
682 static int
683 view_is_splitscreen(struct tog_view *view)
685 return view->begin_x > 0;
688 static void
689 tog_resizeterm(void)
691 int cols, lines;
692 struct winsize size;
694 if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &size) < 0) {
695 cols = 80; /* Default */
696 lines = 24;
697 } else {
698 cols = size.ws_col;
699 lines = size.ws_row;
701 resize_term(lines, cols);
704 static const struct got_error *
705 view_search_start(struct tog_view *view)
707 const struct got_error *err = NULL;
708 char pattern[1024];
709 int ret;
711 if (view->nlines < 1)
712 return NULL;
714 mvwaddstr(view->window, view->begin_y + view->nlines - 1, 0, "/");
715 wclrtoeol(view->window);
717 nocbreak();
718 echo();
719 ret = wgetnstr(view->window, pattern, sizeof(pattern));
720 cbreak();
721 noecho();
722 if (ret == ERR)
723 return NULL;
725 if (view->searching) {
726 regfree(&view->regex);
727 view->searching = 0;
730 if (regcomp(&view->regex, pattern,
731 REG_EXTENDED | REG_NOSUB | REG_NEWLINE) == 0) {
732 err = view->search_start(view);
733 if (err) {
734 regfree(&view->regex);
735 return err;
737 view->searching = TOG_SEARCH_FORWARD;
738 view->search_next_done = 0;
739 view->search_next(view);
742 return NULL;
745 static const struct got_error *
746 view_input(struct tog_view **new, struct tog_view **dead,
747 struct tog_view **focus, int *done, struct tog_view *view,
748 struct tog_view_list_head *views)
750 const struct got_error *err = NULL;
751 struct tog_view *v;
752 int ch, errcode;
754 *new = NULL;
755 *dead = NULL;
756 *focus = NULL;
758 /* Clear "no matches" indicator. */
759 if (view->search_next_done == TOG_SEARCH_NO_MORE ||
760 view->search_next_done == TOG_SEARCH_HAVE_NONE)
761 view->search_next_done = TOG_SEARCH_HAVE_MORE;
763 if (view->searching && !view->search_next_done) {
764 errcode = pthread_mutex_unlock(&tog_mutex);
765 if (errcode)
766 return got_error_set_errno(errcode,
767 "pthread_mutex_unlock");
768 pthread_yield();
769 errcode = pthread_mutex_lock(&tog_mutex);
770 if (errcode)
771 return got_error_set_errno(errcode,
772 "pthread_mutex_lock");
773 view->search_next(view);
774 return NULL;
777 nodelay(stdscr, FALSE);
778 /* Allow threads to make progress while we are waiting for input. */
779 errcode = pthread_mutex_unlock(&tog_mutex);
780 if (errcode)
781 return got_error_set_errno(errcode, "pthread_mutex_unlock");
782 ch = wgetch(view->window);
783 errcode = pthread_mutex_lock(&tog_mutex);
784 if (errcode)
785 return got_error_set_errno(errcode, "pthread_mutex_lock");
786 nodelay(stdscr, TRUE);
788 if (tog_sigwinch_received || tog_sigcont_received) {
789 tog_resizeterm();
790 tog_sigwinch_received = 0;
791 tog_sigcont_received = 0;
792 TAILQ_FOREACH(v, views, entry) {
793 err = view_resize(v);
794 if (err)
795 return err;
796 err = v->input(new, dead, focus, v, KEY_RESIZE);
797 if (err)
798 return err;
802 switch (ch) {
803 case ERR:
804 break;
805 case '\t':
806 if (view->child) {
807 *focus = view->child;
808 view->child_focussed = 1;
809 } else if (view->parent) {
810 *focus = view->parent;
811 view->parent->child_focussed = 0;
813 break;
814 case 'q':
815 err = view->input(new, dead, focus, view, ch);
816 *dead = view;
817 break;
818 case 'Q':
819 *done = 1;
820 break;
821 case 'f':
822 if (view_is_parent_view(view)) {
823 if (view->child == NULL)
824 break;
825 if (view_is_splitscreen(view->child)) {
826 *focus = view->child;
827 view->child_focussed = 1;
828 err = view_fullscreen(view->child);
829 } else
830 err = view_splitscreen(view->child);
831 if (err)
832 break;
833 err = view->child->input(new, dead, focus,
834 view->child, KEY_RESIZE);
835 } else {
836 if (view_is_splitscreen(view)) {
837 *focus = view;
838 view->parent->child_focussed = 1;
839 err = view_fullscreen(view);
840 } else {
841 err = view_splitscreen(view);
843 if (err)
844 break;
845 err = view->input(new, dead, focus, view,
846 KEY_RESIZE);
848 break;
849 case KEY_RESIZE:
850 break;
851 case '/':
852 if (view->search_start)
853 view_search_start(view);
854 else
855 err = view->input(new, dead, focus, view, ch);
856 break;
857 case 'N':
858 case 'n':
859 if (view->search_next) {
860 view->searching = (ch == 'n' ?
861 TOG_SEARCH_FORWARD : TOG_SEARCH_BACKWARD);
862 view->search_next_done = 0;
863 view->search_next(view);
864 } else
865 err = view->input(new, dead, focus, view, ch);
866 break;
867 default:
868 err = view->input(new, dead, focus, view, ch);
869 break;
872 return err;
875 void
876 view_vborder(struct tog_view *view)
878 PANEL *panel;
879 struct tog_view *view_above;
881 if (view->parent)
882 return view_vborder(view->parent);
884 panel = panel_above(view->panel);
885 if (panel == NULL)
886 return;
888 view_above = panel_userptr(panel);
889 mvwvline(view->window, view->begin_y, view_above->begin_x - 1,
890 got_locale_is_utf8() ? ACS_VLINE : '|', view->nlines);
893 int
894 view_needs_focus_indication(struct tog_view *view)
896 if (view_is_parent_view(view)) {
897 if (view->child == NULL || view->child_focussed)
898 return 0;
899 if (!view_is_splitscreen(view->child))
900 return 0;
901 } else if (!view_is_splitscreen(view))
902 return 0;
904 return view->focussed;
907 static const struct got_error *
908 view_loop(struct tog_view *view)
910 const struct got_error *err = NULL;
911 struct tog_view_list_head views;
912 struct tog_view *new_view, *dead_view, *focus_view, *main_view;
913 int fast_refresh = 10;
914 int done = 0, errcode;
916 errcode = pthread_mutex_lock(&tog_mutex);
917 if (errcode)
918 return got_error_set_errno(errcode, "pthread_mutex_lock");
920 TAILQ_INIT(&views);
921 TAILQ_INSERT_HEAD(&views, view, entry);
923 main_view = view;
924 view->focussed = 1;
925 err = view->show(view);
926 if (err)
927 return err;
928 update_panels();
929 doupdate();
930 while (!TAILQ_EMPTY(&views) && !done && !tog_sigpipe_received) {
931 /* Refresh fast during initialization, then become slower. */
932 if (fast_refresh && fast_refresh-- == 0)
933 halfdelay(10); /* switch to once per second */
935 err = view_input(&new_view, &dead_view, &focus_view, &done,
936 view, &views);
937 if (err)
938 break;
939 if (dead_view) {
940 struct tog_view *prev = NULL;
942 if (view_is_parent_view(dead_view))
943 prev = TAILQ_PREV(dead_view,
944 tog_view_list_head, entry);
945 else if (view->parent != dead_view)
946 prev = view->parent;
948 if (dead_view->parent)
949 dead_view->parent->child = NULL;
950 else
951 TAILQ_REMOVE(&views, dead_view, entry);
953 err = view_close(dead_view);
954 if (err || (dead_view == main_view && new_view == NULL))
955 goto done;
957 if (view == dead_view) {
958 if (focus_view)
959 view = focus_view;
960 else if (prev)
961 view = prev;
962 else if (!TAILQ_EMPTY(&views))
963 view = TAILQ_LAST(&views,
964 tog_view_list_head);
965 else
966 view = NULL;
967 if (view) {
968 if (view->child && view->child_focussed)
969 focus_view = view->child;
970 else
971 focus_view = view;
975 if (new_view) {
976 struct tog_view *v, *t;
977 /* Only allow one parent view per type. */
978 TAILQ_FOREACH_SAFE(v, &views, entry, t) {
979 if (v->type != new_view->type)
980 continue;
981 TAILQ_REMOVE(&views, v, entry);
982 err = view_close(v);
983 if (err)
984 goto done;
985 break;
987 TAILQ_INSERT_TAIL(&views, new_view, entry);
988 view = new_view;
989 if (focus_view == NULL)
990 focus_view = new_view;
992 if (focus_view) {
993 show_panel(focus_view->panel);
994 if (view)
995 view->focussed = 0;
996 focus_view->focussed = 1;
997 view = focus_view;
998 if (new_view)
999 show_panel(new_view->panel);
1000 if (view->child && view_is_splitscreen(view->child))
1001 show_panel(view->child->panel);
1003 if (view) {
1004 if (focus_view == NULL) {
1005 view->focussed = 1;
1006 show_panel(view->panel);
1007 if (view->child && view_is_splitscreen(view->child))
1008 show_panel(view->child->panel);
1009 focus_view = view;
1011 if (view->parent) {
1012 err = view->parent->show(view->parent);
1013 if (err)
1014 goto done;
1016 err = view->show(view);
1017 if (err)
1018 goto done;
1019 if (view->child) {
1020 err = view->child->show(view->child);
1021 if (err)
1022 goto done;
1024 update_panels();
1025 doupdate();
1028 done:
1029 while (!TAILQ_EMPTY(&views)) {
1030 view = TAILQ_FIRST(&views);
1031 TAILQ_REMOVE(&views, view, entry);
1032 view_close(view);
1035 errcode = pthread_mutex_unlock(&tog_mutex);
1036 if (errcode && err == NULL)
1037 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
1039 return err;
1042 __dead static void
1043 usage_log(void)
1045 endwin();
1046 fprintf(stderr,
1047 "usage: %s log [-b] [-c commit] [-r repository-path] [path]\n",
1048 getprogname());
1049 exit(1);
1052 /* Create newly allocated wide-character string equivalent to a byte string. */
1053 static const struct got_error *
1054 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
1056 char *vis = NULL;
1057 const struct got_error *err = NULL;
1059 *ws = NULL;
1060 *wlen = mbstowcs(NULL, s, 0);
1061 if (*wlen == (size_t)-1) {
1062 int vislen;
1063 if (errno != EILSEQ)
1064 return got_error_from_errno("mbstowcs");
1066 /* byte string invalid in current encoding; try to "fix" it */
1067 err = got_mbsavis(&vis, &vislen, s);
1068 if (err)
1069 return err;
1070 *wlen = mbstowcs(NULL, vis, 0);
1071 if (*wlen == (size_t)-1) {
1072 err = got_error_from_errno("mbstowcs"); /* give up */
1073 goto done;
1077 *ws = calloc(*wlen + 1, sizeof(**ws));
1078 if (*ws == NULL) {
1079 err = got_error_from_errno("calloc");
1080 goto done;
1083 if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
1084 err = got_error_from_errno("mbstowcs");
1085 done:
1086 free(vis);
1087 if (err) {
1088 free(*ws);
1089 *ws = NULL;
1090 *wlen = 0;
1092 return err;
1095 /* Format a line for display, ensuring that it won't overflow a width limit. */
1096 static const struct got_error *
1097 format_line(wchar_t **wlinep, int *widthp, const char *line, int wlimit,
1098 int col_tab_align)
1100 const struct got_error *err = NULL;
1101 int cols = 0;
1102 wchar_t *wline = NULL;
1103 size_t wlen;
1104 int i;
1106 *wlinep = NULL;
1107 *widthp = 0;
1109 err = mbs2ws(&wline, &wlen, line);
1110 if (err)
1111 return err;
1113 i = 0;
1114 while (i < wlen) {
1115 int width = wcwidth(wline[i]);
1117 if (width == 0) {
1118 i++;
1119 continue;
1122 if (width == 1 || width == 2) {
1123 if (cols + width > wlimit)
1124 break;
1125 cols += width;
1126 i++;
1127 } else if (width == -1) {
1128 if (wline[i] == L'\t') {
1129 width = TABSIZE -
1130 ((cols + col_tab_align) % TABSIZE);
1131 if (cols + width > wlimit)
1132 break;
1133 cols += width;
1135 i++;
1136 } else {
1137 err = got_error_from_errno("wcwidth");
1138 goto done;
1141 wline[i] = L'\0';
1142 if (widthp)
1143 *widthp = cols;
1144 done:
1145 if (err)
1146 free(wline);
1147 else
1148 *wlinep = wline;
1149 return err;
1152 static const struct got_error*
1153 build_refs_str(char **refs_str, struct got_reflist_head *refs,
1154 struct got_object_id *id, struct got_repository *repo)
1156 static const struct got_error *err = NULL;
1157 struct got_reflist_entry *re;
1158 char *s;
1159 const char *name;
1161 *refs_str = NULL;
1163 SIMPLEQ_FOREACH(re, refs, entry) {
1164 struct got_tag_object *tag = NULL;
1165 struct got_object_id *ref_id;
1166 int cmp;
1168 name = got_ref_get_name(re->ref);
1169 if (strcmp(name, GOT_REF_HEAD) == 0)
1170 continue;
1171 if (strncmp(name, "refs/", 5) == 0)
1172 name += 5;
1173 if (strncmp(name, "got/", 4) == 0)
1174 continue;
1175 if (strncmp(name, "heads/", 6) == 0)
1176 name += 6;
1177 if (strncmp(name, "remotes/", 8) == 0) {
1178 name += 8;
1179 s = strstr(name, "/" GOT_REF_HEAD);
1180 if (s != NULL && s[strlen(s)] == '\0')
1181 continue;
1183 err = got_ref_resolve(&ref_id, repo, re->ref);
1184 if (err)
1185 break;
1186 if (strncmp(name, "tags/", 5) == 0) {
1187 err = got_object_open_as_tag(&tag, repo, ref_id);
1188 if (err) {
1189 if (err->code != GOT_ERR_OBJ_TYPE) {
1190 free(ref_id);
1191 break;
1193 /* Ref points at something other than a tag. */
1194 err = NULL;
1195 tag = NULL;
1198 cmp = got_object_id_cmp(tag ?
1199 got_object_tag_get_object_id(tag) : ref_id, id);
1200 free(ref_id);
1201 if (tag)
1202 got_object_tag_close(tag);
1203 if (cmp != 0)
1204 continue;
1205 s = *refs_str;
1206 if (asprintf(refs_str, "%s%s%s", s ? s : "",
1207 s ? ", " : "", name) == -1) {
1208 err = got_error_from_errno("asprintf");
1209 free(s);
1210 *refs_str = NULL;
1211 break;
1213 free(s);
1216 return err;
1219 static const struct got_error *
1220 format_author(wchar_t **wauthor, int *author_width, char *author, int limit,
1221 int col_tab_align)
1223 char *smallerthan, *at;
1225 smallerthan = strchr(author, '<');
1226 if (smallerthan && smallerthan[1] != '\0')
1227 author = smallerthan + 1;
1228 at = strchr(author, '@');
1229 if (at)
1230 *at = '\0';
1231 return format_line(wauthor, author_width, author, limit, col_tab_align);
1234 static const struct got_error *
1235 draw_commit(struct tog_view *view, struct got_commit_object *commit,
1236 struct got_object_id *id, struct got_reflist_head *refs,
1237 const size_t date_display_cols, int author_display_cols,
1238 struct tog_colors *colors)
1240 const struct got_error *err = NULL;
1241 char datebuf[12]; /* YYYY-MM-DD + SPACE + NUL */
1242 char *logmsg0 = NULL, *logmsg = NULL;
1243 char *author = NULL;
1244 wchar_t *wlogmsg = NULL, *wauthor = NULL;
1245 int author_width, logmsg_width;
1246 char *newline, *line = NULL;
1247 int col, limit;
1248 const int avail = view->ncols;
1249 struct tm tm;
1250 time_t committer_time;
1251 struct tog_color *tc;
1253 committer_time = got_object_commit_get_committer_time(commit);
1254 if (localtime_r(&committer_time, &tm) == NULL)
1255 return got_error_from_errno("localtime_r");
1256 if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d ", &tm)
1257 >= sizeof(datebuf))
1258 return got_error(GOT_ERR_NO_SPACE);
1260 if (avail <= date_display_cols)
1261 limit = MIN(sizeof(datebuf) - 1, avail);
1262 else
1263 limit = MIN(date_display_cols, sizeof(datebuf) - 1);
1264 tc = get_color(colors, TOG_COLOR_DATE);
1265 if (tc)
1266 wattr_on(view->window,
1267 COLOR_PAIR(tc->colorpair), NULL);
1268 waddnstr(view->window, datebuf, limit);
1269 if (tc)
1270 wattr_off(view->window,
1271 COLOR_PAIR(tc->colorpair), NULL);
1272 col = limit;
1273 if (col > avail)
1274 goto done;
1276 if (avail >= 120) {
1277 char *id_str;
1278 err = got_object_id_str(&id_str, id);
1279 if (err)
1280 goto done;
1281 tc = get_color(colors, TOG_COLOR_COMMIT);
1282 if (tc)
1283 wattr_on(view->window,
1284 COLOR_PAIR(tc->colorpair), NULL);
1285 wprintw(view->window, "%.8s ", id_str);
1286 if (tc)
1287 wattr_off(view->window,
1288 COLOR_PAIR(tc->colorpair), NULL);
1289 free(id_str);
1290 col += 9;
1291 if (col > avail)
1292 goto done;
1295 author = strdup(got_object_commit_get_author(commit));
1296 if (author == NULL) {
1297 err = got_error_from_errno("strdup");
1298 goto done;
1300 err = format_author(&wauthor, &author_width, author, avail - col, col);
1301 if (err)
1302 goto done;
1303 tc = get_color(colors, TOG_COLOR_AUTHOR);
1304 if (tc)
1305 wattr_on(view->window,
1306 COLOR_PAIR(tc->colorpair), NULL);
1307 waddwstr(view->window, wauthor);
1308 if (tc)
1309 wattr_off(view->window,
1310 COLOR_PAIR(tc->colorpair), NULL);
1311 col += author_width;
1312 while (col < avail && author_width < author_display_cols + 2) {
1313 waddch(view->window, ' ');
1314 col++;
1315 author_width++;
1317 if (col > avail)
1318 goto done;
1320 err = got_object_commit_get_logmsg(&logmsg0, commit);
1321 if (err)
1322 goto done;
1323 logmsg = logmsg0;
1324 while (*logmsg == '\n')
1325 logmsg++;
1326 newline = strchr(logmsg, '\n');
1327 if (newline)
1328 *newline = '\0';
1329 limit = avail - col;
1330 err = format_line(&wlogmsg, &logmsg_width, logmsg, limit, col);
1331 if (err)
1332 goto done;
1333 waddwstr(view->window, wlogmsg);
1334 col += logmsg_width;
1335 while (col < avail) {
1336 waddch(view->window, ' ');
1337 col++;
1339 done:
1340 free(logmsg0);
1341 free(wlogmsg);
1342 free(author);
1343 free(wauthor);
1344 free(line);
1345 return err;
1348 static struct commit_queue_entry *
1349 alloc_commit_queue_entry(struct got_commit_object *commit,
1350 struct got_object_id *id)
1352 struct commit_queue_entry *entry;
1354 entry = calloc(1, sizeof(*entry));
1355 if (entry == NULL)
1356 return NULL;
1358 entry->id = id;
1359 entry->commit = commit;
1360 return entry;
1363 static void
1364 pop_commit(struct commit_queue *commits)
1366 struct commit_queue_entry *entry;
1368 entry = TAILQ_FIRST(&commits->head);
1369 TAILQ_REMOVE(&commits->head, entry, entry);
1370 got_object_commit_close(entry->commit);
1371 commits->ncommits--;
1372 /* Don't free entry->id! It is owned by the commit graph. */
1373 free(entry);
1376 static void
1377 free_commits(struct commit_queue *commits)
1379 while (!TAILQ_EMPTY(&commits->head))
1380 pop_commit(commits);
1383 static const struct got_error *
1384 match_commit(int *have_match, struct got_object_id *id,
1385 struct got_commit_object *commit, regex_t *regex)
1387 const struct got_error *err = NULL;
1388 regmatch_t regmatch;
1389 char *id_str = NULL, *logmsg = NULL;
1391 *have_match = 0;
1393 err = got_object_id_str(&id_str, id);
1394 if (err)
1395 return err;
1397 err = got_object_commit_get_logmsg(&logmsg, commit);
1398 if (err)
1399 goto done;
1401 if (regexec(regex, got_object_commit_get_author(commit), 1,
1402 &regmatch, 0) == 0 ||
1403 regexec(regex, got_object_commit_get_committer(commit), 1,
1404 &regmatch, 0) == 0 ||
1405 regexec(regex, id_str, 1, &regmatch, 0) == 0 ||
1406 regexec(regex, logmsg, 1, &regmatch, 0) == 0)
1407 *have_match = 1;
1408 done:
1409 free(id_str);
1410 free(logmsg);
1411 return err;
1414 static const struct got_error *
1415 queue_commits(struct got_commit_graph *graph, struct commit_queue *commits,
1416 int minqueue, struct got_repository *repo, const char *path,
1417 int *searching, int *search_next_done, regex_t *regex)
1419 const struct got_error *err = NULL;
1420 int nqueued = 0;
1423 * We keep all commits open throughout the lifetime of the log
1424 * view in order to avoid having to re-fetch commits from disk
1425 * while updating the display.
1427 while (nqueued < minqueue ||
1428 (*searching == TOG_SEARCH_FORWARD && !*search_next_done)) {
1429 struct got_object_id *id;
1430 struct got_commit_object *commit;
1431 struct commit_queue_entry *entry;
1432 int errcode;
1434 err = got_commit_graph_iter_next(&id, graph, repo, NULL, NULL);
1435 if (err || id == NULL)
1436 break;
1438 err = got_object_open_as_commit(&commit, repo, id);
1439 if (err)
1440 break;
1441 entry = alloc_commit_queue_entry(commit, id);
1442 if (entry == NULL) {
1443 err = got_error_from_errno("alloc_commit_queue_entry");
1444 break;
1447 errcode = pthread_mutex_lock(&tog_mutex);
1448 if (errcode) {
1449 err = got_error_set_errno(errcode,
1450 "pthread_mutex_lock");
1451 break;
1454 entry->idx = commits->ncommits;
1455 TAILQ_INSERT_TAIL(&commits->head, entry, entry);
1456 nqueued++;
1457 commits->ncommits++;
1459 if (*searching == TOG_SEARCH_FORWARD && !*search_next_done) {
1460 int have_match;
1461 err = match_commit(&have_match, id, commit, regex);
1462 if (err)
1463 break;
1464 if (have_match)
1465 *search_next_done = TOG_SEARCH_HAVE_MORE;
1468 errcode = pthread_mutex_unlock(&tog_mutex);
1469 if (errcode && err == NULL)
1470 err = got_error_set_errno(errcode,
1471 "pthread_mutex_unlock");
1472 if (err)
1473 break;
1476 return err;
1479 static const struct got_error *
1480 draw_commits(struct tog_view *view, struct commit_queue_entry **last,
1481 struct commit_queue_entry **selected, struct commit_queue_entry *first,
1482 struct commit_queue *commits, int selected_idx, int limit,
1483 struct got_reflist_head *refs, const char *path, int commits_needed,
1484 struct tog_colors *colors)
1486 const struct got_error *err = NULL;
1487 struct tog_log_view_state *s = &view->state.log;
1488 struct commit_queue_entry *entry;
1489 int width;
1490 int ncommits, author_cols = 4;
1491 char *id_str = NULL, *header = NULL, *ncommits_str = NULL;
1492 char *refs_str = NULL;
1493 wchar_t *wline;
1494 struct tog_color *tc;
1495 static const size_t date_display_cols = 12;
1497 entry = first;
1498 ncommits = 0;
1499 while (entry) {
1500 if (ncommits == selected_idx) {
1501 *selected = entry;
1502 break;
1504 entry = TAILQ_NEXT(entry, entry);
1505 ncommits++;
1508 if (*selected && !(view->searching && view->search_next_done == 0)) {
1509 err = got_object_id_str(&id_str, (*selected)->id);
1510 if (err)
1511 return err;
1512 if (refs) {
1513 err = build_refs_str(&refs_str, refs, (*selected)->id,
1514 s->repo);
1515 if (err)
1516 goto done;
1520 if (commits_needed == 0)
1521 halfdelay(10); /* disable fast refresh */
1523 if (commits_needed > 0) {
1524 if (asprintf(&ncommits_str, " [%d/%d] %s",
1525 entry ? entry->idx + 1 : 0, commits->ncommits,
1526 (view->searching && !view->search_next_done) ?
1527 "searching..." : "loading...") == -1) {
1528 err = got_error_from_errno("asprintf");
1529 goto done;
1531 } else {
1532 const char *search_str = NULL;
1534 if (view->searching) {
1535 if (view->search_next_done == TOG_SEARCH_NO_MORE)
1536 search_str = "no more matches";
1537 else if (view->search_next_done == TOG_SEARCH_HAVE_NONE)
1538 search_str = "no matches found";
1539 else if (!view->search_next_done)
1540 search_str = "searching...";
1543 if (asprintf(&ncommits_str, " [%d/%d] %s",
1544 entry ? entry->idx + 1 : 0, commits->ncommits,
1545 search_str ? search_str :
1546 (refs_str ? refs_str : "")) == -1) {
1547 err = got_error_from_errno("asprintf");
1548 goto done;
1552 if (path && strcmp(path, "/") != 0) {
1553 if (asprintf(&header, "commit %s %s%s",
1554 id_str ? id_str : "........................................",
1555 path, ncommits_str) == -1) {
1556 err = got_error_from_errno("asprintf");
1557 header = NULL;
1558 goto done;
1560 } else if (asprintf(&header, "commit %s%s",
1561 id_str ? id_str : "........................................",
1562 ncommits_str) == -1) {
1563 err = got_error_from_errno("asprintf");
1564 header = NULL;
1565 goto done;
1567 err = format_line(&wline, &width, header, view->ncols, 0);
1568 if (err)
1569 goto done;
1571 werase(view->window);
1573 if (view_needs_focus_indication(view))
1574 wstandout(view->window);
1575 tc = get_color(colors, TOG_COLOR_COMMIT);
1576 if (tc)
1577 wattr_on(view->window,
1578 COLOR_PAIR(tc->colorpair), NULL);
1579 waddwstr(view->window, wline);
1580 if (tc)
1581 wattr_off(view->window,
1582 COLOR_PAIR(tc->colorpair), NULL);
1583 while (width < view->ncols) {
1584 waddch(view->window, ' ');
1585 width++;
1587 if (view_needs_focus_indication(view))
1588 wstandend(view->window);
1589 free(wline);
1590 if (limit <= 1)
1591 goto done;
1593 /* Grow author column size if necessary. */
1594 entry = first;
1595 ncommits = 0;
1596 while (entry) {
1597 char *author;
1598 wchar_t *wauthor;
1599 int width;
1600 if (ncommits >= limit - 1)
1601 break;
1602 author = strdup(got_object_commit_get_author(entry->commit));
1603 if (author == NULL) {
1604 err = got_error_from_errno("strdup");
1605 goto done;
1607 err = format_author(&wauthor, &width, author, COLS,
1608 date_display_cols);
1609 if (author_cols < width)
1610 author_cols = width;
1611 free(wauthor);
1612 free(author);
1613 ncommits++;
1614 entry = TAILQ_NEXT(entry, entry);
1617 entry = first;
1618 *last = first;
1619 ncommits = 0;
1620 while (entry) {
1621 if (ncommits >= limit - 1)
1622 break;
1623 if (ncommits == selected_idx)
1624 wstandout(view->window);
1625 err = draw_commit(view, entry->commit, entry->id, refs,
1626 date_display_cols, author_cols, colors);
1627 if (ncommits == selected_idx)
1628 wstandend(view->window);
1629 if (err)
1630 goto done;
1631 ncommits++;
1632 *last = entry;
1633 entry = TAILQ_NEXT(entry, entry);
1636 view_vborder(view);
1637 done:
1638 free(id_str);
1639 free(refs_str);
1640 free(ncommits_str);
1641 free(header);
1642 return err;
1645 static void
1646 scroll_up(struct tog_view *view,
1647 struct commit_queue_entry **first_displayed_entry, int maxscroll,
1648 struct commit_queue *commits)
1650 struct commit_queue_entry *entry;
1651 int nscrolled = 0;
1653 entry = TAILQ_FIRST(&commits->head);
1654 if (*first_displayed_entry == entry)
1655 return;
1657 entry = *first_displayed_entry;
1658 while (entry && nscrolled < maxscroll) {
1659 entry = TAILQ_PREV(entry, commit_queue_head, entry);
1660 if (entry) {
1661 *first_displayed_entry = entry;
1662 nscrolled++;
1667 static const struct got_error *
1668 trigger_log_thread(struct tog_view *log_view, int wait,
1669 int *commits_needed, int *log_complete,
1670 pthread_cond_t *need_commits, pthread_cond_t *commit_loaded)
1672 int errcode;
1674 halfdelay(1); /* fast refresh while loading commits */
1676 while (*commits_needed > 0) {
1677 if (*log_complete)
1678 break;
1680 /* Wake the log thread. */
1681 errcode = pthread_cond_signal(need_commits);
1682 if (errcode)
1683 return got_error_set_errno(errcode,
1684 "pthread_cond_signal");
1687 * The mutex will be released while the view loop waits
1688 * in wgetch(), at which time the log thread will run.
1690 if (!wait)
1691 break;
1693 /* Display progress update in log view. */
1694 show_log_view(log_view);
1695 update_panels();
1696 doupdate();
1698 /* Wait right here while next commit is being loaded. */
1699 errcode = pthread_cond_wait(commit_loaded, &tog_mutex);
1700 if (errcode)
1701 return got_error_set_errno(errcode,
1702 "pthread_cond_wait");
1704 /* Display progress update in log view. */
1705 show_log_view(log_view);
1706 update_panels();
1707 doupdate();
1710 return NULL;
1713 static const struct got_error *
1714 scroll_down(struct tog_view *log_view,
1715 struct commit_queue_entry **first_displayed_entry, int maxscroll,
1716 struct commit_queue_entry **last_displayed_entry,
1717 struct commit_queue *commits, int *log_complete, int *commits_needed,
1718 pthread_cond_t *need_commits, pthread_cond_t *commit_loaded)
1720 const struct got_error *err = NULL;
1721 struct commit_queue_entry *pentry;
1722 int nscrolled = 0, ncommits_needed;
1724 if (*last_displayed_entry == NULL)
1725 return NULL;
1727 ncommits_needed = (*last_displayed_entry)->idx + 1 + maxscroll;
1728 if (commits->ncommits < ncommits_needed && !*log_complete) {
1730 * Ask the log thread for required amount of commits.
1732 (*commits_needed) += maxscroll;
1733 err = trigger_log_thread(log_view, 1, commits_needed,
1734 log_complete, need_commits, commit_loaded);
1735 if (err)
1736 return err;
1739 do {
1740 pentry = TAILQ_NEXT(*last_displayed_entry, entry);
1741 if (pentry == NULL)
1742 break;
1744 *last_displayed_entry = pentry;
1746 pentry = TAILQ_NEXT(*first_displayed_entry, entry);
1747 if (pentry == NULL)
1748 break;
1749 *first_displayed_entry = pentry;
1750 } while (++nscrolled < maxscroll);
1752 return err;
1755 static const struct got_error *
1756 open_diff_view_for_commit(struct tog_view **new_view, int begin_x,
1757 struct got_commit_object *commit, struct got_object_id *commit_id,
1758 struct tog_view *log_view, struct got_reflist_head *refs,
1759 struct got_repository *repo)
1761 const struct got_error *err;
1762 struct got_object_qid *parent_id;
1763 struct tog_view *diff_view;
1765 diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
1766 if (diff_view == NULL)
1767 return got_error_from_errno("view_open");
1769 parent_id = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
1770 err = open_diff_view(diff_view, parent_id ? parent_id->id : NULL,
1771 commit_id, log_view, refs, repo);
1772 if (err == NULL)
1773 *new_view = diff_view;
1774 return err;
1777 static const struct got_error *
1778 tree_view_visit_subtree(struct got_tree_object *subtree,
1779 struct tog_tree_view_state *s)
1781 struct tog_parent_tree *parent;
1783 parent = calloc(1, sizeof(*parent));
1784 if (parent == NULL)
1785 return got_error_from_errno("calloc");
1787 parent->tree = s->tree;
1788 parent->first_displayed_entry = s->first_displayed_entry;
1789 parent->selected_entry = s->selected_entry;
1790 parent->selected = s->selected;
1791 TAILQ_INSERT_HEAD(&s->parents, parent, entry);
1792 s->tree = subtree;
1793 s->selected = 0;
1794 s->first_displayed_entry = NULL;
1795 return NULL;
1798 static const struct got_error *
1799 tree_view_walk_path(struct tog_tree_view_state *s,
1800 struct got_object_id *commit_id,
1801 const char *path, struct got_repository *repo)
1803 const struct got_error *err = NULL;
1804 struct got_tree_object *tree = NULL;
1805 const char *p;
1806 char *slash, *subpath = NULL;
1808 /* Walk the path and open corresponding tree objects. */
1809 p = path;
1810 while (*p) {
1811 struct got_tree_entry *te;
1812 struct got_object_id *tree_id;
1813 char *te_name;
1815 while (p[0] == '/')
1816 p++;
1818 /* Ensure the correct subtree entry is selected. */
1819 slash = strchr(p, '/');
1820 if (slash == NULL)
1821 te_name = strdup(p);
1822 else
1823 te_name = strndup(p, slash - p);
1824 if (te_name == NULL) {
1825 err = got_error_from_errno("strndup");
1826 break;
1828 te = got_object_tree_find_entry(s->tree, te_name);
1829 if (te == NULL) {
1830 err = got_error_path(te_name, GOT_ERR_NO_TREE_ENTRY);
1831 free(te_name);
1832 break;
1834 free(te_name);
1835 s->selected_entry = te;
1836 s->selected = got_tree_entry_get_index(te);
1837 if (s->tree != s->root)
1838 s->selected++; /* skip '..' */
1840 if (!S_ISDIR(got_tree_entry_get_mode(s->selected_entry))) {
1841 /* Jump to this file's entry. */
1842 s->first_displayed_entry = s->selected_entry;
1843 s->selected = 0;
1844 break;
1847 slash = strchr(p, '/');
1848 if (slash)
1849 subpath = strndup(path, slash - path);
1850 else
1851 subpath = strdup(path);
1852 if (subpath == NULL) {
1853 err = got_error_from_errno("strdup");
1854 break;
1857 err = got_object_id_by_path(&tree_id, repo, commit_id,
1858 subpath);
1859 if (err)
1860 break;
1862 err = got_object_open_as_tree(&tree, repo, tree_id);
1863 free(tree_id);
1864 if (err)
1865 break;
1867 err = tree_view_visit_subtree(tree, s);
1868 if (err) {
1869 got_object_tree_close(tree);
1870 break;
1872 if (slash == NULL)
1873 break;
1874 free(subpath);
1875 subpath = NULL;
1876 p = slash;
1879 free(subpath);
1880 return err;
1883 static const struct got_error *
1884 browse_commit_tree(struct tog_view **new_view, int begin_x,
1885 struct commit_queue_entry *entry, const char *path,
1886 struct got_reflist_head *refs, struct got_repository *repo)
1888 const struct got_error *err = NULL;
1889 struct got_tree_object *tree;
1890 struct tog_tree_view_state *s;
1891 struct tog_view *tree_view;
1893 err = got_object_open_as_tree(&tree, repo,
1894 got_object_commit_get_tree_id(entry->commit));
1895 if (err)
1896 return err;
1898 tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
1899 if (tree_view == NULL)
1900 return got_error_from_errno("view_open");
1902 err = open_tree_view(tree_view, tree, entry->id, refs, repo);
1903 if (err) {
1904 got_object_tree_close(tree);
1905 return err;
1907 s = &tree_view->state.tree;
1909 *new_view = tree_view;
1911 if (got_path_is_root_dir(path))
1912 return NULL;
1914 return tree_view_walk_path(s, entry->id, path, repo);
1917 static const struct got_error *
1918 block_signals_used_by_main_thread(void)
1920 sigset_t sigset;
1921 int errcode;
1923 if (sigemptyset(&sigset) == -1)
1924 return got_error_from_errno("sigemptyset");
1926 /* tog handles SIGWINCH and SIGCONT */
1927 if (sigaddset(&sigset, SIGWINCH) == -1)
1928 return got_error_from_errno("sigaddset");
1929 if (sigaddset(&sigset, SIGCONT) == -1)
1930 return got_error_from_errno("sigaddset");
1932 /* ncurses handles SIGTSTP */
1933 if (sigaddset(&sigset, SIGTSTP) == -1)
1934 return got_error_from_errno("sigaddset");
1936 errcode = pthread_sigmask(SIG_BLOCK, &sigset, NULL);
1937 if (errcode)
1938 return got_error_set_errno(errcode, "pthread_sigmask");
1940 return NULL;
1943 static void *
1944 log_thread(void *arg)
1946 const struct got_error *err = NULL;
1947 int errcode = 0;
1948 struct tog_log_thread_args *a = arg;
1949 int done = 0;
1951 err = block_signals_used_by_main_thread();
1952 if (err)
1953 return (void *)err;
1955 while (!done && !err && !tog_sigpipe_received) {
1956 err = queue_commits(a->graph, a->commits, 1, a->repo,
1957 a->in_repo_path, a->searching, a->search_next_done,
1958 a->regex);
1959 if (err) {
1960 if (err->code != GOT_ERR_ITER_COMPLETED)
1961 return (void *)err;
1962 err = NULL;
1963 done = 1;
1964 } else if (a->commits_needed > 0)
1965 a->commits_needed--;
1967 errcode = pthread_mutex_lock(&tog_mutex);
1968 if (errcode) {
1969 err = got_error_set_errno(errcode,
1970 "pthread_mutex_lock");
1971 break;
1972 } else if (*a->quit)
1973 done = 1;
1974 else if (*a->first_displayed_entry == NULL) {
1975 *a->first_displayed_entry =
1976 TAILQ_FIRST(&a->commits->head);
1977 *a->selected_entry = *a->first_displayed_entry;
1980 errcode = pthread_cond_signal(&a->commit_loaded);
1981 if (errcode) {
1982 err = got_error_set_errno(errcode,
1983 "pthread_cond_signal");
1984 pthread_mutex_unlock(&tog_mutex);
1985 break;
1988 if (done)
1989 a->commits_needed = 0;
1990 else {
1991 if (a->commits_needed == 0) {
1992 errcode = pthread_cond_wait(&a->need_commits,
1993 &tog_mutex);
1994 if (errcode)
1995 err = got_error_set_errno(errcode,
1996 "pthread_cond_wait");
2000 errcode = pthread_mutex_unlock(&tog_mutex);
2001 if (errcode && err == NULL)
2002 err = got_error_set_errno(errcode,
2003 "pthread_mutex_unlock");
2005 a->log_complete = 1;
2006 return (void *)err;
2009 static const struct got_error *
2010 stop_log_thread(struct tog_log_view_state *s)
2012 const struct got_error *err = NULL;
2013 int errcode;
2015 if (s->thread) {
2016 s->quit = 1;
2017 errcode = pthread_cond_signal(&s->thread_args.need_commits);
2018 if (errcode)
2019 return got_error_set_errno(errcode,
2020 "pthread_cond_signal");
2021 errcode = pthread_mutex_unlock(&tog_mutex);
2022 if (errcode)
2023 return got_error_set_errno(errcode,
2024 "pthread_mutex_unlock");
2025 errcode = pthread_join(s->thread, (void **)&err);
2026 if (errcode)
2027 return got_error_set_errno(errcode, "pthread_join");
2028 errcode = pthread_mutex_lock(&tog_mutex);
2029 if (errcode)
2030 return got_error_set_errno(errcode,
2031 "pthread_mutex_lock");
2032 s->thread = NULL;
2035 errcode = pthread_cond_destroy(&s->thread_args.need_commits);
2036 if (errcode && err == NULL)
2037 err = got_error_set_errno(errcode, "pthread_cond_destroy");
2039 if (s->thread_args.repo) {
2040 got_repo_close(s->thread_args.repo);
2041 s->thread_args.repo = NULL;
2044 if (s->thread_args.graph) {
2045 got_commit_graph_close(s->thread_args.graph);
2046 s->thread_args.graph = NULL;
2049 return err;
2052 static const struct got_error *
2053 close_log_view(struct tog_view *view)
2055 const struct got_error *err = NULL;
2056 struct tog_log_view_state *s = &view->state.log;
2058 err = stop_log_thread(s);
2059 free_commits(&s->commits);
2060 free(s->in_repo_path);
2061 s->in_repo_path = NULL;
2062 free(s->start_id);
2063 s->start_id = NULL;
2064 return err;
2067 static const struct got_error *
2068 search_start_log_view(struct tog_view *view)
2070 struct tog_log_view_state *s = &view->state.log;
2072 s->matched_entry = NULL;
2073 s->search_entry = NULL;
2074 return NULL;
2077 static const struct got_error *
2078 search_next_log_view(struct tog_view *view)
2080 const struct got_error *err = NULL;
2081 struct tog_log_view_state *s = &view->state.log;
2082 struct commit_queue_entry *entry;
2084 /* Display progress update in log view. */
2085 show_log_view(view);
2086 update_panels();
2087 doupdate();
2089 if (s->search_entry) {
2090 int errcode, ch;
2091 errcode = pthread_mutex_unlock(&tog_mutex);
2092 if (errcode)
2093 return got_error_set_errno(errcode,
2094 "pthread_mutex_unlock");
2095 ch = wgetch(view->window);
2096 errcode = pthread_mutex_lock(&tog_mutex);
2097 if (errcode)
2098 return got_error_set_errno(errcode,
2099 "pthread_mutex_lock");
2100 if (ch == KEY_BACKSPACE) {
2101 view->search_next_done = TOG_SEARCH_HAVE_MORE;
2102 return NULL;
2104 if (view->searching == TOG_SEARCH_FORWARD)
2105 entry = TAILQ_NEXT(s->search_entry, entry);
2106 else
2107 entry = TAILQ_PREV(s->search_entry,
2108 commit_queue_head, entry);
2109 } else if (s->matched_entry) {
2110 if (view->searching == TOG_SEARCH_FORWARD)
2111 entry = TAILQ_NEXT(s->matched_entry, entry);
2112 else
2113 entry = TAILQ_PREV(s->matched_entry,
2114 commit_queue_head, entry);
2115 } else {
2116 if (view->searching == TOG_SEARCH_FORWARD)
2117 entry = TAILQ_FIRST(&s->commits.head);
2118 else
2119 entry = TAILQ_LAST(&s->commits.head, commit_queue_head);
2122 while (1) {
2123 int have_match = 0;
2125 if (entry == NULL) {
2126 if (s->thread_args.log_complete ||
2127 view->searching == TOG_SEARCH_BACKWARD) {
2128 view->search_next_done =
2129 (s->matched_entry == NULL ?
2130 TOG_SEARCH_HAVE_NONE : TOG_SEARCH_NO_MORE);
2131 s->search_entry = NULL;
2132 return NULL;
2135 * Poke the log thread for more commits and return,
2136 * allowing the main loop to make progress. Search
2137 * will resume at s->search_entry once we come back.
2139 s->thread_args.commits_needed++;
2140 return trigger_log_thread(view, 0,
2141 &s->thread_args.commits_needed,
2142 &s->thread_args.log_complete,
2143 &s->thread_args.need_commits,
2144 &s->thread_args.commit_loaded);
2147 err = match_commit(&have_match, entry->id, entry->commit,
2148 &view->regex);
2149 if (err)
2150 break;
2151 if (have_match) {
2152 view->search_next_done = TOG_SEARCH_HAVE_MORE;
2153 s->matched_entry = entry;
2154 break;
2157 s->search_entry = entry;
2158 if (view->searching == TOG_SEARCH_FORWARD)
2159 entry = TAILQ_NEXT(entry, entry);
2160 else
2161 entry = TAILQ_PREV(entry, commit_queue_head, entry);
2164 if (s->matched_entry) {
2165 int cur = s->selected_entry->idx;
2166 while (cur < s->matched_entry->idx) {
2167 err = input_log_view(NULL, NULL, NULL, view, KEY_DOWN);
2168 if (err)
2169 return err;
2170 cur++;
2172 while (cur > s->matched_entry->idx) {
2173 err = input_log_view(NULL, NULL, NULL, view, KEY_UP);
2174 if (err)
2175 return err;
2176 cur--;
2180 s->search_entry = NULL;
2182 return NULL;
2185 static const struct got_error *
2186 open_log_view(struct tog_view *view, struct got_object_id *start_id,
2187 struct got_reflist_head *refs, struct got_repository *repo,
2188 const char *head_ref_name, const char *in_repo_path,
2189 int log_branches)
2191 const struct got_error *err = NULL;
2192 struct tog_log_view_state *s = &view->state.log;
2193 struct got_repository *thread_repo = NULL;
2194 struct got_commit_graph *thread_graph = NULL;
2195 int errcode;
2197 if (in_repo_path != s->in_repo_path) {
2198 free(s->in_repo_path);
2199 s->in_repo_path = strdup(in_repo_path);
2200 if (s->in_repo_path == NULL)
2201 return got_error_from_errno("strdup");
2204 /* The commit queue only contains commits being displayed. */
2205 TAILQ_INIT(&s->commits.head);
2206 s->commits.ncommits = 0;
2208 s->refs = refs;
2209 s->repo = repo;
2210 s->head_ref_name = head_ref_name;
2211 s->start_id = got_object_id_dup(start_id);
2212 if (s->start_id == NULL) {
2213 err = got_error_from_errno("got_object_id_dup");
2214 goto done;
2216 s->log_branches = log_branches;
2218 SIMPLEQ_INIT(&s->colors);
2219 if (has_colors() && getenv("TOG_COLORS") != NULL) {
2220 err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
2221 get_color_value("TOG_COLOR_COMMIT"));
2222 if (err)
2223 goto done;
2224 err = add_color(&s->colors, "^$", TOG_COLOR_AUTHOR,
2225 get_color_value("TOG_COLOR_AUTHOR"));
2226 if (err) {
2227 free_colors(&s->colors);
2228 goto done;
2230 err = add_color(&s->colors, "^$", TOG_COLOR_DATE,
2231 get_color_value("TOG_COLOR_DATE"));
2232 if (err) {
2233 free_colors(&s->colors);
2234 goto done;
2238 view->show = show_log_view;
2239 view->input = input_log_view;
2240 view->close = close_log_view;
2241 view->search_start = search_start_log_view;
2242 view->search_next = search_next_log_view;
2244 err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL);
2245 if (err)
2246 goto done;
2247 err = got_commit_graph_open(&thread_graph, s->in_repo_path,
2248 !s->log_branches);
2249 if (err)
2250 goto done;
2251 err = got_commit_graph_iter_start(thread_graph, s->start_id,
2252 s->repo, NULL, NULL);
2253 if (err)
2254 goto done;
2256 errcode = pthread_cond_init(&s->thread_args.need_commits, NULL);
2257 if (errcode) {
2258 err = got_error_set_errno(errcode, "pthread_cond_init");
2259 goto done;
2261 errcode = pthread_cond_init(&s->thread_args.commit_loaded, NULL);
2262 if (errcode) {
2263 err = got_error_set_errno(errcode, "pthread_cond_init");
2264 goto done;
2267 s->thread_args.commits_needed = view->nlines;
2268 s->thread_args.graph = thread_graph;
2269 s->thread_args.commits = &s->commits;
2270 s->thread_args.in_repo_path = s->in_repo_path;
2271 s->thread_args.start_id = s->start_id;
2272 s->thread_args.repo = thread_repo;
2273 s->thread_args.log_complete = 0;
2274 s->thread_args.quit = &s->quit;
2275 s->thread_args.first_displayed_entry = &s->first_displayed_entry;
2276 s->thread_args.selected_entry = &s->selected_entry;
2277 s->thread_args.searching = &view->searching;
2278 s->thread_args.search_next_done = &view->search_next_done;
2279 s->thread_args.regex = &view->regex;
2280 done:
2281 if (err)
2282 close_log_view(view);
2283 return err;
2286 static const struct got_error *
2287 show_log_view(struct tog_view *view)
2289 struct tog_log_view_state *s = &view->state.log;
2291 if (s->thread == NULL) {
2292 int errcode = pthread_create(&s->thread, NULL, log_thread,
2293 &s->thread_args);
2294 if (errcode)
2295 return got_error_set_errno(errcode, "pthread_create");
2298 return draw_commits(view, &s->last_displayed_entry,
2299 &s->selected_entry, s->first_displayed_entry,
2300 &s->commits, s->selected, view->nlines, s->refs,
2301 s->in_repo_path, s->thread_args.commits_needed, &s->colors);
2304 static const struct got_error *
2305 input_log_view(struct tog_view **new_view, struct tog_view **dead_view,
2306 struct tog_view **focus_view, struct tog_view *view, int ch)
2308 const struct got_error *err = NULL;
2309 struct tog_log_view_state *s = &view->state.log;
2310 char *parent_path, *in_repo_path = NULL;
2311 struct tog_view *diff_view = NULL, *tree_view = NULL, *lv = NULL;
2312 int begin_x = 0;
2313 struct got_object_id *start_id;
2315 switch (ch) {
2316 case 'q':
2317 s->quit = 1;
2318 break;
2319 case 'k':
2320 case KEY_UP:
2321 case '<':
2322 case ',':
2323 if (s->first_displayed_entry == NULL)
2324 break;
2325 if (s->selected > 0)
2326 s->selected--;
2327 else
2328 scroll_up(view, &s->first_displayed_entry, 1,
2329 &s->commits);
2330 break;
2331 case KEY_PPAGE:
2332 case CTRL('b'):
2333 if (s->first_displayed_entry == NULL)
2334 break;
2335 if (TAILQ_FIRST(&s->commits.head) ==
2336 s->first_displayed_entry) {
2337 s->selected = 0;
2338 break;
2340 scroll_up(view, &s->first_displayed_entry,
2341 view->nlines - 1, &s->commits);
2342 break;
2343 case 'j':
2344 case KEY_DOWN:
2345 case '>':
2346 case '.':
2347 if (s->first_displayed_entry == NULL)
2348 break;
2349 if (s->selected < MIN(view->nlines - 2,
2350 s->commits.ncommits - 1)) {
2351 s->selected++;
2352 break;
2354 err = scroll_down(view, &s->first_displayed_entry, 1,
2355 &s->last_displayed_entry, &s->commits,
2356 &s->thread_args.log_complete,
2357 &s->thread_args.commits_needed,
2358 &s->thread_args.need_commits,
2359 &s->thread_args.commit_loaded);
2360 break;
2361 case KEY_NPAGE:
2362 case CTRL('f'): {
2363 struct commit_queue_entry *first;
2364 first = s->first_displayed_entry;
2365 if (first == NULL)
2366 break;
2367 err = scroll_down(view, &s->first_displayed_entry,
2368 view->nlines - 1, &s->last_displayed_entry,
2369 &s->commits, &s->thread_args.log_complete,
2370 &s->thread_args.commits_needed,
2371 &s->thread_args.need_commits,
2372 &s->thread_args.commit_loaded);
2373 if (err)
2374 break;
2375 if (first == s->first_displayed_entry &&
2376 s->selected < MIN(view->nlines - 2,
2377 s->commits.ncommits - 1)) {
2378 /* can't scroll further down */
2379 s->selected = MIN(view->nlines - 2,
2380 s->commits.ncommits - 1);
2382 err = NULL;
2383 break;
2385 case KEY_RESIZE:
2386 if (s->selected > view->nlines - 2)
2387 s->selected = view->nlines - 2;
2388 if (s->selected > s->commits.ncommits - 1)
2389 s->selected = s->commits.ncommits - 1;
2390 break;
2391 case KEY_ENTER:
2392 case ' ':
2393 case '\r':
2394 if (s->selected_entry == NULL)
2395 break;
2396 if (view_is_parent_view(view))
2397 begin_x = view_split_begin_x(view->begin_x);
2398 err = open_diff_view_for_commit(&diff_view, begin_x,
2399 s->selected_entry->commit, s->selected_entry->id,
2400 view, s->refs, s->repo);
2401 if (err)
2402 break;
2403 if (view_is_parent_view(view)) {
2404 err = view_close_child(view);
2405 if (err)
2406 return err;
2407 err = view_set_child(view, diff_view);
2408 if (err) {
2409 view_close(diff_view);
2410 break;
2412 *focus_view = diff_view;
2413 view->child_focussed = 1;
2414 } else
2415 *new_view = diff_view;
2416 break;
2417 case 't':
2418 if (s->selected_entry == NULL)
2419 break;
2420 if (view_is_parent_view(view))
2421 begin_x = view_split_begin_x(view->begin_x);
2422 err = browse_commit_tree(&tree_view, begin_x,
2423 s->selected_entry, s->in_repo_path, s->refs, s->repo);
2424 if (err)
2425 break;
2426 if (view_is_parent_view(view)) {
2427 err = view_close_child(view);
2428 if (err)
2429 return err;
2430 err = view_set_child(view, tree_view);
2431 if (err) {
2432 view_close(tree_view);
2433 break;
2435 *focus_view = tree_view;
2436 view->child_focussed = 1;
2437 } else
2438 *new_view = tree_view;
2439 break;
2440 case KEY_BACKSPACE:
2441 if (got_path_cmp(s->in_repo_path, "/",
2442 strlen(s->in_repo_path), 1) == 0)
2443 break;
2444 err = got_path_dirname(&parent_path, s->in_repo_path);
2445 if (err)
2446 return err;
2447 err = stop_log_thread(s);
2448 if (err) {
2449 free(parent_path);
2450 return err;
2452 lv = view_open(view->nlines, view->ncols,
2453 view->begin_y, view->begin_x, TOG_VIEW_LOG);
2454 if (lv == NULL) {
2455 free(parent_path);
2456 return got_error_from_errno("view_open");
2458 err = open_log_view(lv, s->start_id, s->refs,
2459 s->repo, s->head_ref_name, parent_path,
2460 s->log_branches);
2461 free(parent_path);
2462 if (err)
2463 return err;;
2464 if (view_is_parent_view(view))
2465 *new_view = lv;
2466 else {
2467 view_set_child(view->parent, lv);
2468 *focus_view = lv;
2470 break;
2471 case CTRL('l'):
2472 err = stop_log_thread(s);
2473 if (err)
2474 return err;
2475 lv = view_open(view->nlines, view->ncols,
2476 view->begin_y, view->begin_x, TOG_VIEW_LOG);
2477 if (lv == NULL)
2478 return got_error_from_errno("view_open");
2479 err = got_repo_match_object_id(&start_id, NULL,
2480 s->head_ref_name ? s->head_ref_name : GOT_REF_HEAD,
2481 GOT_OBJ_TYPE_COMMIT, 1, s->repo);
2482 if (err) {
2483 view_close(lv);
2484 return err;
2486 in_repo_path = strdup(s->in_repo_path);
2487 if (in_repo_path == NULL) {
2488 free(start_id);
2489 view_close(lv);
2490 return got_error_from_errno("strdup");
2492 got_ref_list_free(s->refs);
2493 err = got_ref_list(s->refs, s->repo, NULL,
2494 got_ref_cmp_by_name, NULL);
2495 if (err) {
2496 free(start_id);
2497 view_close(lv);
2498 return err;
2500 err = open_log_view(lv, start_id, s->refs, s->repo,
2501 s->head_ref_name, in_repo_path, s->log_branches);
2502 if (err) {
2503 free(start_id);
2504 view_close(lv);
2505 return err;;
2507 *dead_view = view;
2508 *new_view = lv;
2509 break;
2510 case 'B':
2511 s->log_branches = !s->log_branches;
2512 err = stop_log_thread(s);
2513 if (err)
2514 return err;
2515 lv = view_open(view->nlines, view->ncols,
2516 view->begin_y, view->begin_x, TOG_VIEW_LOG);
2517 if (lv == NULL)
2518 return got_error_from_errno("view_open");
2519 err = open_log_view(lv, s->start_id, s->refs, s->repo,
2520 s->head_ref_name, s->in_repo_path, s->log_branches);
2521 if (err) {
2522 view_close(lv);
2523 return err;;
2525 *dead_view = view;
2526 *new_view = lv;
2527 break;
2528 default:
2529 break;
2532 return err;
2535 static const struct got_error *
2536 apply_unveil(const char *repo_path, const char *worktree_path)
2538 const struct got_error *error;
2540 #ifdef PROFILE
2541 if (unveil("gmon.out", "rwc") != 0)
2542 return got_error_from_errno2("unveil", "gmon.out");
2543 #endif
2544 if (repo_path && unveil(repo_path, "r") != 0)
2545 return got_error_from_errno2("unveil", repo_path);
2547 if (worktree_path && unveil(worktree_path, "rwc") != 0)
2548 return got_error_from_errno2("unveil", worktree_path);
2550 if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
2551 return got_error_from_errno2("unveil", GOT_TMPDIR_STR);
2553 error = got_privsep_unveil_exec_helpers();
2554 if (error != NULL)
2555 return error;
2557 if (unveil(NULL, NULL) != 0)
2558 return got_error_from_errno("unveil");
2560 return NULL;
2563 static void
2564 init_curses(void)
2566 initscr();
2567 cbreak();
2568 halfdelay(1); /* Do fast refresh while initial view is loading. */
2569 noecho();
2570 nonl();
2571 intrflush(stdscr, FALSE);
2572 keypad(stdscr, TRUE);
2573 curs_set(0);
2574 if (getenv("TOG_COLORS") != NULL) {
2575 start_color();
2576 use_default_colors();
2578 signal(SIGWINCH, tog_sigwinch);
2579 signal(SIGPIPE, tog_sigpipe);
2580 signal(SIGCONT, tog_sigcont);
2583 static const struct got_error *
2584 get_in_repo_path_from_argv0(char **in_repo_path, int argc, char *argv[],
2585 struct got_repository *repo, struct got_worktree *worktree)
2587 const struct got_error *err = NULL;
2589 if (argc == 0) {
2590 *in_repo_path = strdup("/");
2591 if (*in_repo_path == NULL)
2592 return got_error_from_errno("strdup");
2593 return NULL;
2596 if (worktree) {
2597 const char *prefix = got_worktree_get_path_prefix(worktree);
2598 char *wt_path, *p;
2600 err = got_worktree_resolve_path(&wt_path, worktree, argv[0]);
2601 if (err)
2602 return err;
2604 if (asprintf(&p, "%s%s%s", prefix,
2605 (strcmp(prefix, "/") != 0) ? "/" : "", wt_path) == -1) {
2606 err = got_error_from_errno("asprintf");
2607 free(wt_path);
2608 return err;
2610 err = got_repo_map_path(in_repo_path, repo, p, 0);
2611 free(p);
2612 free(wt_path);
2613 } else
2614 err = got_repo_map_path(in_repo_path, repo, argv[0], 1);
2616 return err;
2619 static const struct got_error *
2620 cmd_log(int argc, char *argv[])
2622 const struct got_error *error;
2623 struct got_repository *repo = NULL;
2624 struct got_worktree *worktree = NULL;
2625 struct got_reflist_head refs;
2626 struct got_object_id *start_id = NULL;
2627 char *in_repo_path = NULL, *repo_path = NULL, *cwd = NULL;
2628 char *start_commit = NULL, *head_ref_name = NULL;
2629 int ch, log_branches = 0;
2630 struct tog_view *view;
2632 SIMPLEQ_INIT(&refs);
2634 #ifndef PROFILE
2635 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
2636 NULL) == -1)
2637 err(1, "pledge");
2638 #endif
2640 while ((ch = getopt(argc, argv, "bc:r:")) != -1) {
2641 switch (ch) {
2642 case 'b':
2643 log_branches = 1;
2644 break;
2645 case 'c':
2646 start_commit = optarg;
2647 break;
2648 case 'r':
2649 repo_path = realpath(optarg, NULL);
2650 if (repo_path == NULL)
2651 return got_error_from_errno2("realpath",
2652 optarg);
2653 break;
2654 default:
2655 usage_log();
2656 /* NOTREACHED */
2660 argc -= optind;
2661 argv += optind;
2663 if (argc > 1)
2664 usage_log();
2666 cwd = getcwd(NULL, 0);
2667 if (cwd == NULL)
2668 return got_error_from_errno("getcwd");
2670 error = got_worktree_open(&worktree, cwd);
2671 if (error && error->code != GOT_ERR_NOT_WORKTREE)
2672 goto done;
2674 if (repo_path == NULL) {
2675 if (worktree)
2676 repo_path =
2677 strdup(got_worktree_get_repo_path(worktree));
2678 else
2679 repo_path = strdup(cwd);
2681 if (repo_path == NULL) {
2682 error = got_error_from_errno("strdup");
2683 goto done;
2686 error = got_repo_open(&repo, repo_path, NULL);
2687 if (error != NULL)
2688 goto done;
2690 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
2691 repo, worktree);
2692 if (error)
2693 goto done;
2695 init_curses();
2697 error = apply_unveil(got_repo_get_path(repo),
2698 worktree ? got_worktree_get_root_path(worktree) : NULL);
2699 if (error)
2700 goto done;
2702 if (start_commit == NULL)
2703 error = got_repo_match_object_id(&start_id, NULL, worktree ?
2704 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
2705 GOT_OBJ_TYPE_COMMIT, 1, repo);
2706 else
2707 error = got_repo_match_object_id(&start_id, NULL, start_commit,
2708 GOT_OBJ_TYPE_COMMIT, 1, repo);
2709 if (error != NULL)
2710 goto done;
2712 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
2713 if (error)
2714 goto done;
2716 view = view_open(0, 0, 0, 0, TOG_VIEW_LOG);
2717 if (view == NULL) {
2718 error = got_error_from_errno("view_open");
2719 goto done;
2721 if (worktree) {
2722 head_ref_name = strdup(
2723 got_worktree_get_head_ref_name(worktree));
2724 if (head_ref_name == NULL) {
2725 error = got_error_from_errno("strdup");
2726 goto done;
2729 error = open_log_view(view, start_id, &refs, repo, head_ref_name,
2730 in_repo_path, log_branches);
2731 if (error)
2732 goto done;
2733 if (worktree) {
2734 /* Release work tree lock. */
2735 got_worktree_close(worktree);
2736 worktree = NULL;
2738 error = view_loop(view);
2739 done:
2740 free(in_repo_path);
2741 free(repo_path);
2742 free(cwd);
2743 free(start_id);
2744 free(head_ref_name);
2745 if (repo)
2746 got_repo_close(repo);
2747 if (worktree)
2748 got_worktree_close(worktree);
2749 got_ref_list_free(&refs);
2750 return error;
2753 __dead static void
2754 usage_diff(void)
2756 endwin();
2757 fprintf(stderr, "usage: %s diff [-r repository-path] object1 object2\n",
2758 getprogname());
2759 exit(1);
2762 static char *
2763 parse_next_line(FILE *f, size_t *len)
2765 char *line;
2766 size_t linelen;
2767 size_t lineno;
2768 const char delim[3] = { '\0', '\0', '\0'};
2770 line = fparseln(f, &linelen, &lineno, delim, 0);
2771 if (len)
2772 *len = linelen;
2773 return line;
2776 static int
2777 match_line(const char *line, regex_t *regex)
2779 regmatch_t regmatch;
2781 return regexec(regex, line, 1, &regmatch, 0) == 0;
2784 struct tog_color *
2785 match_color(struct tog_colors *colors, const char *line)
2787 struct tog_color *tc = NULL;
2789 SIMPLEQ_FOREACH(tc, colors, entry) {
2790 if (match_line(line, &tc->regex))
2791 return tc;
2794 return NULL;
2797 static const struct got_error *
2798 draw_file(struct tog_view *view, FILE *f, int first_displayed_line, int nlines,
2799 off_t *line_offsets, int selected_line, int max_lines,
2800 int *last_displayed_line, int *eof, char *header, struct tog_colors *colors)
2802 const struct got_error *err;
2803 int nprinted = 0;
2804 char *line;
2805 struct tog_color *tc;
2806 size_t len;
2807 wchar_t *wline;
2808 int width;
2809 off_t line_offset;
2811 line_offset = line_offsets[first_displayed_line - 1];
2812 if (fseeko(f, line_offset, SEEK_SET) == -1)
2813 return got_error_from_errno("fseek");
2815 werase(view->window);
2817 if (header) {
2818 err = format_line(&wline, &width, header, view->ncols, 0);
2819 if (err) {
2820 return err;
2823 if (view_needs_focus_indication(view))
2824 wstandout(view->window);
2825 waddwstr(view->window, wline);
2826 free(wline);
2827 wline = NULL;
2828 if (view_needs_focus_indication(view))
2829 wstandend(view->window);
2830 if (width <= view->ncols - 1)
2831 waddch(view->window, '\n');
2833 if (max_lines <= 1)
2834 return NULL;
2835 max_lines--;
2838 *eof = 0;
2839 while (max_lines > 0 && nprinted < max_lines) {
2840 line = parse_next_line(f, &len);
2841 if (line == NULL) {
2842 *eof = 1;
2843 break;
2845 err = format_line(&wline, &width, line, view->ncols, 0);
2846 if (err) {
2847 free(line);
2848 return err;
2851 tc = match_color(colors, line);
2852 if (tc)
2853 wattr_on(view->window,
2854 COLOR_PAIR(tc->colorpair), NULL);
2855 waddwstr(view->window, wline);
2856 if (tc)
2857 wattr_off(view->window,
2858 COLOR_PAIR(tc->colorpair), NULL);
2859 if (width <= view->ncols - 1)
2860 waddch(view->window, '\n');
2861 nprinted++;
2862 free(line);
2863 free(wline);
2864 wline = NULL;
2866 if (nprinted >= 1)
2867 *last_displayed_line = first_displayed_line + (nprinted - 1);
2868 else
2869 *last_displayed_line = first_displayed_line;
2871 view_vborder(view);
2873 if (*eof) {
2874 while (nprinted < view->nlines) {
2875 waddch(view->window, '\n');
2876 nprinted++;
2879 err = format_line(&wline, &width, TOG_EOF_STRING, view->ncols, 0);
2880 if (err) {
2881 return err;
2884 wstandout(view->window);
2885 waddwstr(view->window, wline);
2886 free(wline);
2887 wline = NULL;
2888 wstandend(view->window);
2891 return NULL;
2894 static char *
2895 get_datestr(time_t *time, char *datebuf)
2897 struct tm mytm, *tm;
2898 char *p, *s;
2900 tm = gmtime_r(time, &mytm);
2901 if (tm == NULL)
2902 return NULL;
2903 s = asctime_r(tm, datebuf);
2904 if (s == NULL)
2905 return NULL;
2906 p = strchr(s, '\n');
2907 if (p)
2908 *p = '\0';
2909 return s;
2912 static const struct got_error *
2913 get_changed_paths(struct got_pathlist_head *paths,
2914 struct got_commit_object *commit, struct got_repository *repo)
2916 const struct got_error *err = NULL;
2917 struct got_object_id *tree_id1 = NULL, *tree_id2 = NULL;
2918 struct got_tree_object *tree1 = NULL, *tree2 = NULL;
2919 struct got_object_qid *qid;
2921 qid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
2922 if (qid != NULL) {
2923 struct got_commit_object *pcommit;
2924 err = got_object_open_as_commit(&pcommit, repo,
2925 qid->id);
2926 if (err)
2927 return err;
2929 tree_id1 = got_object_commit_get_tree_id(pcommit);
2930 got_object_commit_close(pcommit);
2934 if (tree_id1) {
2935 err = got_object_open_as_tree(&tree1, repo, tree_id1);
2936 if (err)
2937 goto done;
2940 tree_id2 = got_object_commit_get_tree_id(commit);
2941 err = got_object_open_as_tree(&tree2, repo, tree_id2);
2942 if (err)
2943 goto done;
2945 err = got_diff_tree(tree1, tree2, "", "", repo,
2946 got_diff_tree_collect_changed_paths, paths, 0);
2947 done:
2948 if (tree1)
2949 got_object_tree_close(tree1);
2950 if (tree2)
2951 got_object_tree_close(tree2);
2952 return err;
2955 static const struct got_error *
2956 add_line_offset(off_t **line_offsets, size_t *nlines, off_t off)
2958 off_t *p;
2960 p = reallocarray(*line_offsets, *nlines + 1, sizeof(off_t));
2961 if (p == NULL)
2962 return got_error_from_errno("reallocarray");
2963 *line_offsets = p;
2964 (*line_offsets)[*nlines] = off;
2965 (*nlines)++;
2966 return NULL;
2969 static const struct got_error *
2970 write_commit_info(off_t **line_offsets, size_t *nlines,
2971 struct got_object_id *commit_id, struct got_reflist_head *refs,
2972 struct got_repository *repo, FILE *outfile)
2974 const struct got_error *err = NULL;
2975 char datebuf[26], *datestr;
2976 struct got_commit_object *commit;
2977 char *id_str = NULL, *logmsg = NULL, *s = NULL, *line;
2978 time_t committer_time;
2979 const char *author, *committer;
2980 char *refs_str = NULL;
2981 struct got_pathlist_head changed_paths;
2982 struct got_pathlist_entry *pe;
2983 off_t outoff = 0;
2984 int n;
2986 TAILQ_INIT(&changed_paths);
2988 if (refs) {
2989 err = build_refs_str(&refs_str, refs, commit_id, repo);
2990 if (err)
2991 return err;
2994 err = got_object_open_as_commit(&commit, repo, commit_id);
2995 if (err)
2996 return err;
2998 err = got_object_id_str(&id_str, commit_id);
2999 if (err) {
3000 err = got_error_from_errno("got_object_id_str");
3001 goto done;
3004 err = add_line_offset(line_offsets, nlines, 0);
3005 if (err)
3006 goto done;
3008 n = fprintf(outfile, "commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
3009 refs_str ? refs_str : "", refs_str ? ")" : "");
3010 if (n < 0) {
3011 err = got_error_from_errno("fprintf");
3012 goto done;
3014 outoff += n;
3015 err = add_line_offset(line_offsets, nlines, outoff);
3016 if (err)
3017 goto done;
3019 n = fprintf(outfile, "from: %s\n",
3020 got_object_commit_get_author(commit));
3021 if (n < 0) {
3022 err = got_error_from_errno("fprintf");
3023 goto done;
3025 outoff += n;
3026 err = add_line_offset(line_offsets, nlines, outoff);
3027 if (err)
3028 goto done;
3030 committer_time = got_object_commit_get_committer_time(commit);
3031 datestr = get_datestr(&committer_time, datebuf);
3032 if (datestr) {
3033 n = fprintf(outfile, "date: %s UTC\n", datestr);
3034 if (n < 0) {
3035 err = got_error_from_errno("fprintf");
3036 goto done;
3038 outoff += n;
3039 err = add_line_offset(line_offsets, nlines, outoff);
3040 if (err)
3041 goto done;
3043 author = got_object_commit_get_author(commit);
3044 committer = got_object_commit_get_committer(commit);
3045 if (strcmp(author, committer) != 0) {
3046 n = fprintf(outfile, "via: %s\n", committer);
3047 if (n < 0) {
3048 err = got_error_from_errno("fprintf");
3049 goto done;
3051 outoff += n;
3052 err = add_line_offset(line_offsets, nlines, outoff);
3053 if (err)
3054 goto done;
3056 err = got_object_commit_get_logmsg(&logmsg, commit);
3057 if (err)
3058 goto done;
3059 s = logmsg;
3060 while ((line = strsep(&s, "\n")) != NULL) {
3061 n = fprintf(outfile, "%s\n", line);
3062 if (n < 0) {
3063 err = got_error_from_errno("fprintf");
3064 goto done;
3066 outoff += n;
3067 err = add_line_offset(line_offsets, nlines, outoff);
3068 if (err)
3069 goto done;
3072 err = get_changed_paths(&changed_paths, commit, repo);
3073 if (err)
3074 goto done;
3075 TAILQ_FOREACH(pe, &changed_paths, entry) {
3076 struct got_diff_changed_path *cp = pe->data;
3077 n = fprintf(outfile, "%c %s\n", cp->status, pe->path);
3078 if (n < 0) {
3079 err = got_error_from_errno("fprintf");
3080 goto done;
3082 outoff += n;
3083 err = add_line_offset(line_offsets, nlines, outoff);
3084 if (err)
3085 goto done;
3086 free((char *)pe->path);
3087 free(pe->data);
3090 fputc('\n', outfile);
3091 outoff++;
3092 err = add_line_offset(line_offsets, nlines, outoff);
3093 done:
3094 got_pathlist_free(&changed_paths);
3095 free(id_str);
3096 free(logmsg);
3097 free(refs_str);
3098 got_object_commit_close(commit);
3099 if (err) {
3100 free(*line_offsets);
3101 *line_offsets = NULL;
3102 *nlines = 0;
3104 return err;
3107 static const struct got_error *
3108 create_diff(struct tog_diff_view_state *s)
3110 const struct got_error *err = NULL;
3111 FILE *f = NULL;
3112 int obj_type;
3114 free(s->line_offsets);
3115 s->line_offsets = malloc(sizeof(off_t));
3116 if (s->line_offsets == NULL)
3117 return got_error_from_errno("malloc");
3118 s->nlines = 0;
3120 f = got_opentemp();
3121 if (f == NULL) {
3122 err = got_error_from_errno("got_opentemp");
3123 goto done;
3125 if (s->f && fclose(s->f) != 0) {
3126 err = got_error_from_errno("fclose");
3127 goto done;
3129 s->f = f;
3131 if (s->id1)
3132 err = got_object_get_type(&obj_type, s->repo, s->id1);
3133 else
3134 err = got_object_get_type(&obj_type, s->repo, s->id2);
3135 if (err)
3136 goto done;
3138 switch (obj_type) {
3139 case GOT_OBJ_TYPE_BLOB:
3140 err = got_diff_objects_as_blobs(&s->line_offsets, &s->nlines,
3141 s->id1, s->id2, NULL, NULL, s->diff_context, 0,
3142 s->repo, s->f);
3143 break;
3144 case GOT_OBJ_TYPE_TREE:
3145 err = got_diff_objects_as_trees(&s->line_offsets, &s->nlines,
3146 s->id1, s->id2, "", "", s->diff_context, 0, s->repo, s->f);
3147 break;
3148 case GOT_OBJ_TYPE_COMMIT: {
3149 const struct got_object_id_queue *parent_ids;
3150 struct got_object_qid *pid;
3151 struct got_commit_object *commit2;
3153 err = got_object_open_as_commit(&commit2, s->repo, s->id2);
3154 if (err)
3155 goto done;
3156 /* Show commit info if we're diffing to a parent/root commit. */
3157 if (s->id1 == NULL) {
3158 err = write_commit_info(&s->line_offsets, &s->nlines,
3159 s->id2, s->refs, s->repo, s->f);
3160 if (err)
3161 goto done;
3162 } else {
3163 parent_ids = got_object_commit_get_parent_ids(commit2);
3164 SIMPLEQ_FOREACH(pid, parent_ids, entry) {
3165 if (got_object_id_cmp(s->id1, pid->id) == 0) {
3166 err = write_commit_info(
3167 &s->line_offsets, &s->nlines,
3168 s->id2, s->refs, s->repo, s->f);
3169 if (err)
3170 goto done;
3171 break;
3175 got_object_commit_close(commit2);
3177 err = got_diff_objects_as_commits(&s->line_offsets, &s->nlines,
3178 s->id1, s->id2, s->diff_context, 0, s->repo, s->f);
3179 break;
3181 default:
3182 err = got_error(GOT_ERR_OBJ_TYPE);
3183 break;
3185 if (err)
3186 goto done;
3187 done:
3188 if (s->f && fflush(s->f) != 0 && err == NULL)
3189 err = got_error_from_errno("fflush");
3190 return err;
3193 static void
3194 diff_view_indicate_progress(struct tog_view *view)
3196 mvwaddstr(view->window, 0, 0, "diffing...");
3197 update_panels();
3198 doupdate();
3201 static const struct got_error *
3202 search_start_diff_view(struct tog_view *view)
3204 struct tog_diff_view_state *s = &view->state.diff;
3206 s->matched_line = 0;
3207 return NULL;
3210 static const struct got_error *
3211 search_next_diff_view(struct tog_view *view)
3213 struct tog_diff_view_state *s = &view->state.diff;
3214 int lineno;
3216 if (!view->searching) {
3217 view->search_next_done = TOG_SEARCH_HAVE_MORE;
3218 return NULL;
3221 if (s->matched_line) {
3222 if (view->searching == TOG_SEARCH_FORWARD)
3223 lineno = s->matched_line + 1;
3224 else
3225 lineno = s->matched_line - 1;
3226 } else {
3227 if (view->searching == TOG_SEARCH_FORWARD)
3228 lineno = 1;
3229 else
3230 lineno = s->nlines;
3233 while (1) {
3234 char *line = NULL;
3235 off_t offset;
3236 size_t len;
3238 if (lineno <= 0 || lineno > s->nlines) {
3239 if (s->matched_line == 0) {
3240 view->search_next_done = TOG_SEARCH_HAVE_MORE;
3241 free(line);
3242 break;
3245 if (view->searching == TOG_SEARCH_FORWARD)
3246 lineno = 1;
3247 else
3248 lineno = s->nlines;
3251 offset = s->line_offsets[lineno - 1];
3252 if (fseeko(s->f, offset, SEEK_SET) != 0) {
3253 free(line);
3254 return got_error_from_errno("fseeko");
3256 free(line);
3257 line = parse_next_line(s->f, &len);
3258 if (line && match_line(line, &view->regex)) {
3259 view->search_next_done = TOG_SEARCH_HAVE_MORE;
3260 s->matched_line = lineno;
3261 free(line);
3262 break;
3264 free(line);
3265 if (view->searching == TOG_SEARCH_FORWARD)
3266 lineno++;
3267 else
3268 lineno--;
3271 if (s->matched_line) {
3272 s->first_displayed_line = s->matched_line;
3273 s->selected_line = 1;
3276 return NULL;
3279 static const struct got_error *
3280 open_diff_view(struct tog_view *view, struct got_object_id *id1,
3281 struct got_object_id *id2, struct tog_view *log_view,
3282 struct got_reflist_head *refs, struct got_repository *repo)
3284 const struct got_error *err;
3285 struct tog_diff_view_state *s = &view->state.diff;
3287 if (id1 != NULL && id2 != NULL) {
3288 int type1, type2;
3289 err = got_object_get_type(&type1, repo, id1);
3290 if (err)
3291 return err;
3292 err = got_object_get_type(&type2, repo, id2);
3293 if (err)
3294 return err;
3296 if (type1 != type2)
3297 return got_error(GOT_ERR_OBJ_TYPE);
3299 s->first_displayed_line = 1;
3300 s->last_displayed_line = view->nlines;
3301 s->selected_line = 1;
3302 s->repo = repo;
3303 s->refs = refs;
3304 s->id1 = id1;
3305 s->id2 = id2;
3307 if (id1) {
3308 s->id1 = got_object_id_dup(id1);
3309 if (s->id1 == NULL)
3310 return got_error_from_errno("got_object_id_dup");
3311 } else
3312 s->id1 = NULL;
3314 s->id2 = got_object_id_dup(id2);
3315 if (s->id2 == NULL) {
3316 free(s->id1);
3317 s->id1 = NULL;
3318 return got_error_from_errno("got_object_id_dup");
3320 s->f = NULL;
3321 s->first_displayed_line = 1;
3322 s->last_displayed_line = view->nlines;
3323 s->diff_context = 3;
3324 s->log_view = log_view;
3325 s->repo = repo;
3326 s->refs = refs;
3328 SIMPLEQ_INIT(&s->colors);
3329 if (has_colors() && getenv("TOG_COLORS") != NULL) {
3330 err = add_color(&s->colors,
3331 "^-", TOG_COLOR_DIFF_MINUS,
3332 get_color_value("TOG_COLOR_DIFF_MINUS"));
3333 if (err)
3334 return err;
3335 err = add_color(&s->colors, "^\\+",
3336 TOG_COLOR_DIFF_PLUS,
3337 get_color_value("TOG_COLOR_DIFF_PLUS"));
3338 if (err) {
3339 free_colors(&s->colors);
3340 return err;
3342 err = add_color(&s->colors,
3343 "^@@", TOG_COLOR_DIFF_CHUNK_HEADER,
3344 get_color_value("TOG_COLOR_DIFF_CHUNK_HEADER"));
3345 if (err) {
3346 free_colors(&s->colors);
3347 return err;
3350 err = add_color(&s->colors,
3351 "^(commit [0-9a-f]|(blob|file) [-+] |[MDmA] [^ ])",
3352 TOG_COLOR_DIFF_META,
3353 get_color_value("TOG_COLOR_DIFF_META"));
3354 if (err) {
3355 free_colors(&s->colors);
3356 return err;
3359 err = add_color(&s->colors,
3360 "^(from|via): ", TOG_COLOR_AUTHOR,
3361 get_color_value("TOG_COLOR_AUTHOR"));
3362 if (err) {
3363 free_colors(&s->colors);
3364 return err;
3367 err = add_color(&s->colors,
3368 "^date: ", TOG_COLOR_DATE,
3369 get_color_value("TOG_COLOR_DATE"));
3370 if (err) {
3371 free_colors(&s->colors);
3372 return err;
3376 if (log_view && view_is_splitscreen(view))
3377 show_log_view(log_view); /* draw vborder */
3378 diff_view_indicate_progress(view);
3380 s->line_offsets = NULL;
3381 s->nlines = 0;
3382 err = create_diff(s);
3383 if (err) {
3384 free(s->id1);
3385 s->id1 = NULL;
3386 free(s->id2);
3387 s->id2 = NULL;
3388 return err;
3391 view->show = show_diff_view;
3392 view->input = input_diff_view;
3393 view->close = close_diff_view;
3394 view->search_start = search_start_diff_view;
3395 view->search_next = search_next_diff_view;
3397 return NULL;
3400 static const struct got_error *
3401 close_diff_view(struct tog_view *view)
3403 const struct got_error *err = NULL;
3404 struct tog_diff_view_state *s = &view->state.diff;
3406 free(s->id1);
3407 s->id1 = NULL;
3408 free(s->id2);
3409 s->id2 = NULL;
3410 if (s->f && fclose(s->f) == EOF)
3411 err = got_error_from_errno("fclose");
3412 free_colors(&s->colors);
3413 free(s->line_offsets);
3414 s->line_offsets = NULL;
3415 s->nlines = 0;
3416 return err;
3419 static const struct got_error *
3420 show_diff_view(struct tog_view *view)
3422 const struct got_error *err;
3423 struct tog_diff_view_state *s = &view->state.diff;
3424 char *id_str1 = NULL, *id_str2, *header;
3426 if (s->id1) {
3427 err = got_object_id_str(&id_str1, s->id1);
3428 if (err)
3429 return err;
3431 err = got_object_id_str(&id_str2, s->id2);
3432 if (err)
3433 return err;
3435 if (asprintf(&header, "diff %s %s",
3436 id_str1 ? id_str1 : "/dev/null", id_str2) == -1) {
3437 err = got_error_from_errno("asprintf");
3438 free(id_str1);
3439 free(id_str2);
3440 return err;
3442 free(id_str1);
3443 free(id_str2);
3445 return draw_file(view, s->f, s->first_displayed_line, s->nlines,
3446 s->line_offsets, s->selected_line, view->nlines,
3447 &s->last_displayed_line, &s->eof, header, &s->colors);
3450 static const struct got_error *
3451 set_selected_commit(struct tog_diff_view_state *s,
3452 struct commit_queue_entry *entry)
3454 const struct got_error *err;
3455 const struct got_object_id_queue *parent_ids;
3456 struct got_commit_object *selected_commit;
3457 struct got_object_qid *pid;
3459 free(s->id2);
3460 s->id2 = got_object_id_dup(entry->id);
3461 if (s->id2 == NULL)
3462 return got_error_from_errno("got_object_id_dup");
3464 err = got_object_open_as_commit(&selected_commit, s->repo, entry->id);
3465 if (err)
3466 return err;
3467 parent_ids = got_object_commit_get_parent_ids(selected_commit);
3468 free(s->id1);
3469 pid = SIMPLEQ_FIRST(parent_ids);
3470 s->id1 = pid ? got_object_id_dup(pid->id) : NULL;
3471 got_object_commit_close(selected_commit);
3472 return NULL;
3475 static const struct got_error *
3476 input_diff_view(struct tog_view **new_view, struct tog_view **dead_view,
3477 struct tog_view **focus_view, struct tog_view *view, int ch)
3479 const struct got_error *err = NULL;
3480 struct tog_diff_view_state *s = &view->state.diff;
3481 struct tog_log_view_state *ls;
3482 struct commit_queue_entry *entry;
3483 int i;
3485 switch (ch) {
3486 case 'k':
3487 case KEY_UP:
3488 if (s->first_displayed_line > 1)
3489 s->first_displayed_line--;
3490 break;
3491 case KEY_PPAGE:
3492 case CTRL('b'):
3493 if (s->first_displayed_line == 1)
3494 break;
3495 i = 0;
3496 while (i++ < view->nlines - 1 &&
3497 s->first_displayed_line > 1)
3498 s->first_displayed_line--;
3499 break;
3500 case 'j':
3501 case KEY_DOWN:
3502 if (!s->eof)
3503 s->first_displayed_line++;
3504 break;
3505 case KEY_NPAGE:
3506 case CTRL('f'):
3507 case ' ':
3508 if (s->eof)
3509 break;
3510 i = 0;
3511 while (!s->eof && i++ < view->nlines - 1) {
3512 char *line;
3513 line = parse_next_line(s->f, NULL);
3514 s->first_displayed_line++;
3515 if (line == NULL)
3516 break;
3518 break;
3519 case '[':
3520 if (s->diff_context > 0) {
3521 s->diff_context--;
3522 diff_view_indicate_progress(view);
3523 err = create_diff(s);
3525 break;
3526 case ']':
3527 if (s->diff_context < GOT_DIFF_MAX_CONTEXT) {
3528 s->diff_context++;
3529 diff_view_indicate_progress(view);
3530 err = create_diff(s);
3532 break;
3533 case '<':
3534 case ',':
3535 if (s->log_view == NULL)
3536 break;
3537 ls = &s->log_view->state.log;
3538 entry = TAILQ_PREV(ls->selected_entry,
3539 commit_queue_head, entry);
3540 if (entry == NULL)
3541 break;
3543 err = input_log_view(NULL, NULL, NULL, s->log_view,
3544 KEY_UP);
3545 if (err)
3546 break;
3548 err = set_selected_commit(s, entry);
3549 if (err)
3550 break;
3552 s->first_displayed_line = 1;
3553 s->last_displayed_line = view->nlines;
3555 diff_view_indicate_progress(view);
3556 err = create_diff(s);
3557 break;
3558 case '>':
3559 case '.':
3560 if (s->log_view == NULL)
3561 break;
3562 ls = &s->log_view->state.log;
3564 if (TAILQ_NEXT(ls->selected_entry, entry) == NULL) {
3565 ls->thread_args.commits_needed++;
3566 err = trigger_log_thread(s->log_view, 1,
3567 &ls->thread_args.commits_needed,
3568 &ls->thread_args.log_complete,
3569 &ls->thread_args.need_commits,
3570 &ls->thread_args.commit_loaded);
3571 if (err)
3572 break;
3574 err = input_log_view(NULL, NULL, NULL, s->log_view,
3575 KEY_DOWN);
3576 if (err)
3577 break;
3579 entry = TAILQ_NEXT(ls->selected_entry, entry);
3580 if (entry == NULL)
3581 break;
3583 err = set_selected_commit(s, entry);
3584 if (err)
3585 break;
3587 s->first_displayed_line = 1;
3588 s->last_displayed_line = view->nlines;
3590 diff_view_indicate_progress(view);
3591 err = create_diff(s);
3592 break;
3593 default:
3594 break;
3597 return err;
3600 static const struct got_error *
3601 cmd_diff(int argc, char *argv[])
3603 const struct got_error *error = NULL;
3604 struct got_repository *repo = NULL;
3605 struct got_worktree *worktree = NULL;
3606 struct got_reflist_head refs;
3607 struct got_object_id *id1 = NULL, *id2 = NULL;
3608 char *repo_path = NULL, *cwd = NULL;
3609 char *id_str1 = NULL, *id_str2 = NULL;
3610 int ch;
3611 struct tog_view *view;
3613 SIMPLEQ_INIT(&refs);
3615 #ifndef PROFILE
3616 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
3617 NULL) == -1)
3618 err(1, "pledge");
3619 #endif
3621 while ((ch = getopt(argc, argv, "r:")) != -1) {
3622 switch (ch) {
3623 case 'r':
3624 repo_path = realpath(optarg, NULL);
3625 if (repo_path == NULL)
3626 return got_error_from_errno2("realpath",
3627 optarg);
3628 break;
3629 default:
3630 usage_diff();
3631 /* NOTREACHED */
3635 argc -= optind;
3636 argv += optind;
3638 if (argc == 0) {
3639 usage_diff(); /* TODO show local worktree changes */
3640 } else if (argc == 2) {
3641 id_str1 = argv[0];
3642 id_str2 = argv[1];
3643 } else
3644 usage_diff();
3646 cwd = getcwd(NULL, 0);
3647 if (cwd == NULL)
3648 return got_error_from_errno("getcwd");
3650 error = got_worktree_open(&worktree, cwd);
3651 if (error && error->code != GOT_ERR_NOT_WORKTREE)
3652 goto done;
3654 if (repo_path == NULL) {
3655 if (worktree)
3656 repo_path =
3657 strdup(got_worktree_get_repo_path(worktree));
3658 else
3659 repo_path = strdup(cwd);
3661 if (repo_path == NULL) {
3662 error = got_error_from_errno("strdup");
3663 goto done;
3666 error = got_repo_open(&repo, repo_path, NULL);
3667 if (error)
3668 goto done;
3670 init_curses();
3672 error = apply_unveil(got_repo_get_path(repo), NULL);
3673 if (error)
3674 goto done;
3676 error = got_repo_match_object_id_prefix(&id1, id_str1,
3677 GOT_OBJ_TYPE_ANY, repo);
3678 if (error)
3679 goto done;
3681 error = got_repo_match_object_id_prefix(&id2, id_str2,
3682 GOT_OBJ_TYPE_ANY, repo);
3683 if (error)
3684 goto done;
3686 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
3687 if (error)
3688 goto done;
3690 view = view_open(0, 0, 0, 0, TOG_VIEW_DIFF);
3691 if (view == NULL) {
3692 error = got_error_from_errno("view_open");
3693 goto done;
3695 error = open_diff_view(view, id1, id2, NULL, &refs, repo);
3696 if (error)
3697 goto done;
3698 error = view_loop(view);
3699 done:
3700 free(repo_path);
3701 free(cwd);
3702 if (repo)
3703 got_repo_close(repo);
3704 if (worktree)
3705 got_worktree_close(worktree);
3706 got_ref_list_free(&refs);
3707 return error;
3710 __dead static void
3711 usage_blame(void)
3713 endwin();
3714 fprintf(stderr, "usage: %s blame [-c commit] [-r repository-path] path\n",
3715 getprogname());
3716 exit(1);
3719 struct tog_blame_line {
3720 int annotated;
3721 struct got_object_id *id;
3724 static const struct got_error *
3725 draw_blame(struct tog_view *view, struct got_object_id *id, FILE *f,
3726 const char *path, struct tog_blame_line *lines, int nlines,
3727 int blame_complete, int selected_line, int *first_displayed_line,
3728 int *last_displayed_line, int *eof, int max_lines,
3729 struct tog_colors *colors)
3731 const struct got_error *err;
3732 int lineno = 0, nprinted = 0;
3733 char *line;
3734 size_t len;
3735 wchar_t *wline;
3736 int width;
3737 struct tog_blame_line *blame_line;
3738 struct got_object_id *prev_id = NULL;
3739 char *id_str;
3740 struct tog_color *tc;
3742 err = got_object_id_str(&id_str, id);
3743 if (err)
3744 return err;
3746 rewind(f);
3747 werase(view->window);
3749 if (asprintf(&line, "commit %s", id_str) == -1) {
3750 err = got_error_from_errno("asprintf");
3751 free(id_str);
3752 return err;
3755 err = format_line(&wline, &width, line, view->ncols, 0);
3756 free(line);
3757 line = NULL;
3758 if (err)
3759 return err;
3760 if (view_needs_focus_indication(view))
3761 wstandout(view->window);
3762 tc = get_color(colors, TOG_COLOR_COMMIT);
3763 if (tc)
3764 wattr_on(view->window,
3765 COLOR_PAIR(tc->colorpair), NULL);
3766 waddwstr(view->window, wline);
3767 if (tc)
3768 wattr_off(view->window,
3769 COLOR_PAIR(tc->colorpair), NULL);
3770 if (view_needs_focus_indication(view))
3771 wstandend(view->window);
3772 free(wline);
3773 wline = NULL;
3774 if (width < view->ncols - 1)
3775 waddch(view->window, '\n');
3777 if (asprintf(&line, "[%d/%d] %s%s",
3778 *first_displayed_line - 1 + selected_line, nlines,
3779 blame_complete ? "" : "annotating... ", path) == -1) {
3780 free(id_str);
3781 return got_error_from_errno("asprintf");
3783 free(id_str);
3784 err = format_line(&wline, &width, line, view->ncols, 0);
3785 free(line);
3786 line = NULL;
3787 if (err)
3788 return err;
3789 waddwstr(view->window, wline);
3790 free(wline);
3791 wline = NULL;
3792 if (width < view->ncols - 1)
3793 waddch(view->window, '\n');
3795 *eof = 0;
3796 while (nprinted < max_lines - 2) {
3797 line = parse_next_line(f, &len);
3798 if (line == NULL) {
3799 *eof = 1;
3800 break;
3802 if (++lineno < *first_displayed_line) {
3803 free(line);
3804 continue;
3807 if (view->ncols <= 9) {
3808 width = 9;
3809 wline = wcsdup(L"");
3810 if (wline == NULL)
3811 err = got_error_from_errno("wcsdup");
3812 } else {
3813 err = format_line(&wline, &width, line,
3814 view->ncols - 9, 9);
3815 width += 9;
3817 if (err) {
3818 free(line);
3819 return err;
3822 if (view->focussed && nprinted == selected_line - 1)
3823 wstandout(view->window);
3825 if (nlines > 0) {
3826 blame_line = &lines[lineno - 1];
3827 if (blame_line->annotated && prev_id &&
3828 got_object_id_cmp(prev_id, blame_line->id) == 0 &&
3829 !(view->focussed &&
3830 nprinted == selected_line - 1)) {
3831 waddstr(view->window, " ");
3832 } else if (blame_line->annotated) {
3833 char *id_str;
3834 err = got_object_id_str(&id_str, blame_line->id);
3835 if (err) {
3836 free(line);
3837 free(wline);
3838 return err;
3840 tc = get_color(colors, TOG_COLOR_COMMIT);
3841 if (tc)
3842 wattr_on(view->window,
3843 COLOR_PAIR(tc->colorpair), NULL);
3844 wprintw(view->window, "%.8s", id_str);
3845 if (tc)
3846 wattr_off(view->window,
3847 COLOR_PAIR(tc->colorpair), NULL);
3848 free(id_str);
3849 prev_id = blame_line->id;
3850 } else {
3851 waddstr(view->window, "........");
3852 prev_id = NULL;
3854 } else {
3855 waddstr(view->window, "........");
3856 prev_id = NULL;
3859 if (view->focussed && nprinted == selected_line - 1)
3860 wstandend(view->window);
3861 waddstr(view->window, " ");
3863 waddwstr(view->window, wline);
3864 if (width <= view->ncols - 1)
3865 waddch(view->window, '\n');
3866 if (++nprinted == 1)
3867 *first_displayed_line = lineno;
3868 free(line);
3869 free(wline);
3870 wline = NULL;
3872 *last_displayed_line = lineno;
3874 view_vborder(view);
3876 return NULL;
3879 static const struct got_error *
3880 blame_cb(void *arg, int nlines, int lineno, struct got_object_id *id)
3882 const struct got_error *err = NULL;
3883 struct tog_blame_cb_args *a = arg;
3884 struct tog_blame_line *line;
3885 int errcode;
3887 if (nlines != a->nlines ||
3888 (lineno != -1 && lineno < 1) || lineno > a->nlines)
3889 return got_error(GOT_ERR_RANGE);
3891 errcode = pthread_mutex_lock(&tog_mutex);
3892 if (errcode)
3893 return got_error_set_errno(errcode, "pthread_mutex_lock");
3895 if (*a->quit) { /* user has quit the blame view */
3896 err = got_error(GOT_ERR_ITER_COMPLETED);
3897 goto done;
3900 if (lineno == -1)
3901 goto done; /* no change in this commit */
3903 line = &a->lines[lineno - 1];
3904 if (line->annotated)
3905 goto done;
3907 line->id = got_object_id_dup(id);
3908 if (line->id == NULL) {
3909 err = got_error_from_errno("got_object_id_dup");
3910 goto done;
3912 line->annotated = 1;
3913 done:
3914 errcode = pthread_mutex_unlock(&tog_mutex);
3915 if (errcode)
3916 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
3917 return err;
3920 static void *
3921 blame_thread(void *arg)
3923 const struct got_error *err;
3924 struct tog_blame_thread_args *ta = arg;
3925 struct tog_blame_cb_args *a = ta->cb_args;
3926 int errcode;
3928 err = block_signals_used_by_main_thread();
3929 if (err)
3930 return (void *)err;
3932 err = got_blame(ta->path, a->commit_id, ta->repo,
3933 blame_cb, ta->cb_args, ta->cancel_cb, ta->cancel_arg);
3934 if (err && err->code == GOT_ERR_CANCELLED)
3935 err = NULL;
3937 errcode = pthread_mutex_lock(&tog_mutex);
3938 if (errcode)
3939 return (void *)got_error_set_errno(errcode,
3940 "pthread_mutex_lock");
3942 got_repo_close(ta->repo);
3943 ta->repo = NULL;
3944 *ta->complete = 1;
3946 errcode = pthread_mutex_unlock(&tog_mutex);
3947 if (errcode && err == NULL)
3948 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
3950 return (void *)err;
3953 static struct got_object_id *
3954 get_selected_commit_id(struct tog_blame_line *lines, int nlines,
3955 int first_displayed_line, int selected_line)
3957 struct tog_blame_line *line;
3959 if (nlines <= 0)
3960 return NULL;
3962 line = &lines[first_displayed_line - 1 + selected_line - 1];
3963 if (!line->annotated)
3964 return NULL;
3966 return line->id;
3969 static const struct got_error *
3970 stop_blame(struct tog_blame *blame)
3972 const struct got_error *err = NULL;
3973 int i;
3975 if (blame->thread) {
3976 int errcode;
3977 errcode = pthread_mutex_unlock(&tog_mutex);
3978 if (errcode)
3979 return got_error_set_errno(errcode,
3980 "pthread_mutex_unlock");
3981 errcode = pthread_join(blame->thread, (void **)&err);
3982 if (errcode)
3983 return got_error_set_errno(errcode, "pthread_join");
3984 errcode = pthread_mutex_lock(&tog_mutex);
3985 if (errcode)
3986 return got_error_set_errno(errcode,
3987 "pthread_mutex_lock");
3988 if (err && err->code == GOT_ERR_ITER_COMPLETED)
3989 err = NULL;
3990 blame->thread = NULL;
3992 if (blame->thread_args.repo) {
3993 got_repo_close(blame->thread_args.repo);
3994 blame->thread_args.repo = NULL;
3996 if (blame->f) {
3997 if (fclose(blame->f) != 0 && err == NULL)
3998 err = got_error_from_errno("fclose");
3999 blame->f = NULL;
4001 if (blame->lines) {
4002 for (i = 0; i < blame->nlines; i++)
4003 free(blame->lines[i].id);
4004 free(blame->lines);
4005 blame->lines = NULL;
4007 free(blame->cb_args.commit_id);
4008 blame->cb_args.commit_id = NULL;
4010 return err;
4013 static const struct got_error *
4014 cancel_blame_view(void *arg)
4016 const struct got_error *err = NULL;
4017 int *done = arg;
4018 int errcode;
4020 errcode = pthread_mutex_lock(&tog_mutex);
4021 if (errcode)
4022 return got_error_set_errno(errcode,
4023 "pthread_mutex_unlock");
4025 if (*done)
4026 err = got_error(GOT_ERR_CANCELLED);
4028 errcode = pthread_mutex_unlock(&tog_mutex);
4029 if (errcode)
4030 return got_error_set_errno(errcode,
4031 "pthread_mutex_lock");
4033 return err;
4036 static const struct got_error *
4037 run_blame(struct tog_blame *blame, struct tog_view *view, int *blame_complete,
4038 int *first_displayed_line, int *last_displayed_line, int *selected_line,
4039 int *done, int *eof, const char *path, struct got_object_id *commit_id,
4040 struct got_repository *repo)
4042 const struct got_error *err = NULL;
4043 struct got_blob_object *blob = NULL;
4044 struct got_repository *thread_repo = NULL;
4045 struct got_object_id *obj_id = NULL;
4046 int obj_type;
4048 err = got_object_id_by_path(&obj_id, repo, commit_id, path);
4049 if (err)
4050 return err;
4052 err = got_object_get_type(&obj_type, repo, obj_id);
4053 if (err)
4054 goto done;
4056 if (obj_type != GOT_OBJ_TYPE_BLOB) {
4057 err = got_error(GOT_ERR_OBJ_TYPE);
4058 goto done;
4061 err = got_object_open_as_blob(&blob, repo, obj_id, 8192);
4062 if (err)
4063 goto done;
4064 blame->f = got_opentemp();
4065 if (blame->f == NULL) {
4066 err = got_error_from_errno("got_opentemp");
4067 goto done;
4069 err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
4070 &blame->line_offsets, blame->f, blob);
4071 if (err || blame->nlines == 0)
4072 goto done;
4074 /* Don't include \n at EOF in the blame line count. */
4075 if (blame->line_offsets[blame->nlines - 1] == blame->filesize)
4076 blame->nlines--;
4078 blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
4079 if (blame->lines == NULL) {
4080 err = got_error_from_errno("calloc");
4081 goto done;
4084 err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL);
4085 if (err)
4086 goto done;
4088 blame->cb_args.view = view;
4089 blame->cb_args.lines = blame->lines;
4090 blame->cb_args.nlines = blame->nlines;
4091 blame->cb_args.commit_id = got_object_id_dup(commit_id);
4092 if (blame->cb_args.commit_id == NULL) {
4093 err = got_error_from_errno("got_object_id_dup");
4094 goto done;
4096 blame->cb_args.quit = done;
4098 blame->thread_args.path = path;
4099 blame->thread_args.repo = thread_repo;
4100 blame->thread_args.cb_args = &blame->cb_args;
4101 blame->thread_args.complete = blame_complete;
4102 blame->thread_args.cancel_cb = cancel_blame_view;
4103 blame->thread_args.cancel_arg = done;
4104 *blame_complete = 0;
4106 done:
4107 if (blob)
4108 got_object_blob_close(blob);
4109 free(obj_id);
4110 if (err)
4111 stop_blame(blame);
4112 return err;
4115 static const struct got_error *
4116 open_blame_view(struct tog_view *view, char *path,
4117 struct got_object_id *commit_id, struct got_reflist_head *refs,
4118 struct got_repository *repo)
4120 const struct got_error *err = NULL;
4121 struct tog_blame_view_state *s = &view->state.blame;
4123 SIMPLEQ_INIT(&s->blamed_commits);
4125 s->path = strdup(path);
4126 if (s->path == NULL)
4127 return got_error_from_errno("strdup");
4129 err = got_object_qid_alloc(&s->blamed_commit, commit_id);
4130 if (err) {
4131 free(s->path);
4132 return err;
4135 SIMPLEQ_INSERT_HEAD(&s->blamed_commits, s->blamed_commit, entry);
4136 s->first_displayed_line = 1;
4137 s->last_displayed_line = view->nlines;
4138 s->selected_line = 1;
4139 s->blame_complete = 0;
4140 s->repo = repo;
4141 s->refs = refs;
4142 s->commit_id = commit_id;
4143 memset(&s->blame, 0, sizeof(s->blame));
4145 SIMPLEQ_INIT(&s->colors);
4146 if (has_colors() && getenv("TOG_COLORS") != NULL) {
4147 err = add_color(&s->colors, "^", TOG_COLOR_COMMIT,
4148 get_color_value("TOG_COLOR_COMMIT"));
4149 if (err)
4150 return err;
4153 view->show = show_blame_view;
4154 view->input = input_blame_view;
4155 view->close = close_blame_view;
4156 view->search_start = search_start_blame_view;
4157 view->search_next = search_next_blame_view;
4159 return run_blame(&s->blame, view, &s->blame_complete,
4160 &s->first_displayed_line, &s->last_displayed_line,
4161 &s->selected_line, &s->done, &s->eof, s->path,
4162 s->blamed_commit->id, s->repo);
4165 static const struct got_error *
4166 close_blame_view(struct tog_view *view)
4168 const struct got_error *err = NULL;
4169 struct tog_blame_view_state *s = &view->state.blame;
4171 if (s->blame.thread)
4172 err = stop_blame(&s->blame);
4174 while (!SIMPLEQ_EMPTY(&s->blamed_commits)) {
4175 struct got_object_qid *blamed_commit;
4176 blamed_commit = SIMPLEQ_FIRST(&s->blamed_commits);
4177 SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
4178 got_object_qid_free(blamed_commit);
4181 free(s->path);
4182 free_colors(&s->colors);
4184 return err;
4187 static const struct got_error *
4188 search_start_blame_view(struct tog_view *view)
4190 struct tog_blame_view_state *s = &view->state.blame;
4192 s->matched_line = 0;
4193 return NULL;
4196 static const struct got_error *
4197 search_next_blame_view(struct tog_view *view)
4199 struct tog_blame_view_state *s = &view->state.blame;
4200 int lineno;
4202 if (!view->searching) {
4203 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4204 return NULL;
4207 if (s->matched_line) {
4208 if (view->searching == TOG_SEARCH_FORWARD)
4209 lineno = s->matched_line + 1;
4210 else
4211 lineno = s->matched_line - 1;
4212 } else {
4213 if (view->searching == TOG_SEARCH_FORWARD)
4214 lineno = 1;
4215 else
4216 lineno = s->blame.nlines;
4219 while (1) {
4220 char *line = NULL;
4221 off_t offset;
4222 size_t len;
4224 if (lineno <= 0 || lineno > s->blame.nlines) {
4225 if (s->matched_line == 0) {
4226 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4227 free(line);
4228 break;
4231 if (view->searching == TOG_SEARCH_FORWARD)
4232 lineno = 1;
4233 else
4234 lineno = s->blame.nlines;
4237 offset = s->blame.line_offsets[lineno - 1];
4238 if (fseeko(s->blame.f, offset, SEEK_SET) != 0) {
4239 free(line);
4240 return got_error_from_errno("fseeko");
4242 free(line);
4243 line = parse_next_line(s->blame.f, &len);
4244 if (line && match_line(line, &view->regex)) {
4245 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4246 s->matched_line = lineno;
4247 free(line);
4248 break;
4250 free(line);
4251 if (view->searching == TOG_SEARCH_FORWARD)
4252 lineno++;
4253 else
4254 lineno--;
4257 if (s->matched_line) {
4258 s->first_displayed_line = s->matched_line;
4259 s->selected_line = 1;
4262 return NULL;
4265 static const struct got_error *
4266 show_blame_view(struct tog_view *view)
4268 const struct got_error *err = NULL;
4269 struct tog_blame_view_state *s = &view->state.blame;
4270 int errcode;
4272 if (s->blame.thread == NULL) {
4273 errcode = pthread_create(&s->blame.thread, NULL, blame_thread,
4274 &s->blame.thread_args);
4275 if (errcode)
4276 return got_error_set_errno(errcode, "pthread_create");
4278 halfdelay(1); /* fast refresh while annotating */
4281 if (s->blame_complete)
4282 halfdelay(10); /* disable fast refresh */
4284 err = draw_blame(view, s->blamed_commit->id, s->blame.f,
4285 s->path, s->blame.lines, s->blame.nlines, s->blame_complete,
4286 s->selected_line, &s->first_displayed_line,
4287 &s->last_displayed_line, &s->eof, view->nlines, &s->colors);
4289 view_vborder(view);
4290 return err;
4293 static const struct got_error *
4294 input_blame_view(struct tog_view **new_view, struct tog_view **dead_view,
4295 struct tog_view **focus_view, struct tog_view *view, int ch)
4297 const struct got_error *err = NULL, *thread_err = NULL;
4298 struct tog_view *diff_view;
4299 struct tog_blame_view_state *s = &view->state.blame;
4300 int begin_x = 0;
4302 switch (ch) {
4303 case 'q':
4304 s->done = 1;
4305 break;
4306 case 'k':
4307 case KEY_UP:
4308 if (s->selected_line > 1)
4309 s->selected_line--;
4310 else if (s->selected_line == 1 &&
4311 s->first_displayed_line > 1)
4312 s->first_displayed_line--;
4313 break;
4314 case KEY_PPAGE:
4315 case CTRL('b'):
4316 if (s->first_displayed_line == 1) {
4317 s->selected_line = 1;
4318 break;
4320 if (s->first_displayed_line > view->nlines - 2)
4321 s->first_displayed_line -=
4322 (view->nlines - 2);
4323 else
4324 s->first_displayed_line = 1;
4325 break;
4326 case 'j':
4327 case KEY_DOWN:
4328 if (s->selected_line < view->nlines - 2 &&
4329 s->first_displayed_line +
4330 s->selected_line <= s->blame.nlines)
4331 s->selected_line++;
4332 else if (s->last_displayed_line <
4333 s->blame.nlines)
4334 s->first_displayed_line++;
4335 break;
4336 case 'b':
4337 case 'p': {
4338 struct got_object_id *id = NULL;
4339 id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
4340 s->first_displayed_line, s->selected_line);
4341 if (id == NULL)
4342 break;
4343 if (ch == 'p') {
4344 struct got_commit_object *commit;
4345 struct got_object_qid *pid;
4346 struct got_object_id *blob_id = NULL;
4347 int obj_type;
4348 err = got_object_open_as_commit(&commit,
4349 s->repo, id);
4350 if (err)
4351 break;
4352 pid = SIMPLEQ_FIRST(
4353 got_object_commit_get_parent_ids(commit));
4354 if (pid == NULL) {
4355 got_object_commit_close(commit);
4356 break;
4358 /* Check if path history ends here. */
4359 err = got_object_id_by_path(&blob_id, s->repo,
4360 pid->id, s->path);
4361 if (err) {
4362 if (err->code == GOT_ERR_NO_TREE_ENTRY)
4363 err = NULL;
4364 got_object_commit_close(commit);
4365 break;
4367 err = got_object_get_type(&obj_type, s->repo,
4368 blob_id);
4369 free(blob_id);
4370 /* Can't blame non-blob type objects. */
4371 if (obj_type != GOT_OBJ_TYPE_BLOB) {
4372 got_object_commit_close(commit);
4373 break;
4375 err = got_object_qid_alloc(&s->blamed_commit,
4376 pid->id);
4377 got_object_commit_close(commit);
4378 } else {
4379 if (got_object_id_cmp(id,
4380 s->blamed_commit->id) == 0)
4381 break;
4382 err = got_object_qid_alloc(&s->blamed_commit,
4383 id);
4385 if (err)
4386 break;
4387 s->done = 1;
4388 thread_err = stop_blame(&s->blame);
4389 s->done = 0;
4390 if (thread_err)
4391 break;
4392 SIMPLEQ_INSERT_HEAD(&s->blamed_commits,
4393 s->blamed_commit, entry);
4394 err = run_blame(&s->blame, view, &s->blame_complete,
4395 &s->first_displayed_line, &s->last_displayed_line,
4396 &s->selected_line, &s->done, &s->eof,
4397 s->path, s->blamed_commit->id, s->repo);
4398 if (err)
4399 break;
4400 break;
4402 case 'B': {
4403 struct got_object_qid *first;
4404 first = SIMPLEQ_FIRST(&s->blamed_commits);
4405 if (!got_object_id_cmp(first->id, s->commit_id))
4406 break;
4407 s->done = 1;
4408 thread_err = stop_blame(&s->blame);
4409 s->done = 0;
4410 if (thread_err)
4411 break;
4412 SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
4413 got_object_qid_free(s->blamed_commit);
4414 s->blamed_commit =
4415 SIMPLEQ_FIRST(&s->blamed_commits);
4416 err = run_blame(&s->blame, view, &s->blame_complete,
4417 &s->first_displayed_line, &s->last_displayed_line,
4418 &s->selected_line, &s->done, &s->eof, s->path,
4419 s->blamed_commit->id, s->repo);
4420 if (err)
4421 break;
4422 break;
4424 case KEY_ENTER:
4425 case '\r': {
4426 struct got_object_id *id = NULL;
4427 struct got_object_qid *pid;
4428 struct got_commit_object *commit = NULL;
4429 id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
4430 s->first_displayed_line, s->selected_line);
4431 if (id == NULL)
4432 break;
4433 err = got_object_open_as_commit(&commit, s->repo, id);
4434 if (err)
4435 break;
4436 pid = SIMPLEQ_FIRST(
4437 got_object_commit_get_parent_ids(commit));
4438 if (view_is_parent_view(view))
4439 begin_x = view_split_begin_x(view->begin_x);
4440 diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
4441 if (diff_view == NULL) {
4442 got_object_commit_close(commit);
4443 err = got_error_from_errno("view_open");
4444 break;
4446 err = open_diff_view(diff_view, pid ? pid->id : NULL,
4447 id, NULL, s->refs, s->repo);
4448 got_object_commit_close(commit);
4449 if (err) {
4450 view_close(diff_view);
4451 break;
4453 if (view_is_parent_view(view)) {
4454 err = view_close_child(view);
4455 if (err)
4456 break;
4457 err = view_set_child(view, diff_view);
4458 if (err) {
4459 view_close(diff_view);
4460 break;
4462 *focus_view = diff_view;
4463 view->child_focussed = 1;
4464 } else
4465 *new_view = diff_view;
4466 if (err)
4467 break;
4468 break;
4470 case KEY_NPAGE:
4471 case CTRL('f'):
4472 case ' ':
4473 if (s->last_displayed_line >= s->blame.nlines &&
4474 s->selected_line >= MIN(s->blame.nlines,
4475 view->nlines - 2)) {
4476 break;
4478 if (s->last_displayed_line >= s->blame.nlines &&
4479 s->selected_line < view->nlines - 2) {
4480 s->selected_line = MIN(s->blame.nlines,
4481 view->nlines - 2);
4482 break;
4484 if (s->last_displayed_line + view->nlines - 2
4485 <= s->blame.nlines)
4486 s->first_displayed_line +=
4487 view->nlines - 2;
4488 else
4489 s->first_displayed_line =
4490 s->blame.nlines -
4491 (view->nlines - 3);
4492 break;
4493 case KEY_RESIZE:
4494 if (s->selected_line > view->nlines - 2) {
4495 s->selected_line = MIN(s->blame.nlines,
4496 view->nlines - 2);
4498 break;
4499 default:
4500 break;
4502 return thread_err ? thread_err : err;
4505 static const struct got_error *
4506 cmd_blame(int argc, char *argv[])
4508 const struct got_error *error;
4509 struct got_repository *repo = NULL;
4510 struct got_reflist_head refs;
4511 struct got_worktree *worktree = NULL;
4512 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
4513 char *link_target = NULL;
4514 struct got_object_id *commit_id = NULL;
4515 char *commit_id_str = NULL;
4516 int ch;
4517 struct tog_view *view;
4519 SIMPLEQ_INIT(&refs);
4521 #ifndef PROFILE
4522 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
4523 NULL) == -1)
4524 err(1, "pledge");
4525 #endif
4527 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
4528 switch (ch) {
4529 case 'c':
4530 commit_id_str = optarg;
4531 break;
4532 case 'r':
4533 repo_path = realpath(optarg, NULL);
4534 if (repo_path == NULL)
4535 return got_error_from_errno2("realpath",
4536 optarg);
4537 break;
4538 default:
4539 usage_blame();
4540 /* NOTREACHED */
4544 argc -= optind;
4545 argv += optind;
4547 if (argc != 1)
4548 usage_blame();
4550 cwd = getcwd(NULL, 0);
4551 if (cwd == NULL)
4552 return got_error_from_errno("getcwd");
4554 error = got_worktree_open(&worktree, cwd);
4555 if (error && error->code != GOT_ERR_NOT_WORKTREE)
4556 goto done;
4558 if (repo_path == NULL) {
4559 if (worktree)
4560 repo_path =
4561 strdup(got_worktree_get_repo_path(worktree));
4562 else
4563 repo_path = strdup(cwd);
4565 if (repo_path == NULL) {
4566 error = got_error_from_errno("strdup");
4567 goto done;
4570 error = got_repo_open(&repo, repo_path, NULL);
4571 if (error != NULL)
4572 goto done;
4574 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv, repo,
4575 worktree);
4576 if (error)
4577 goto done;
4579 init_curses();
4581 error = apply_unveil(got_repo_get_path(repo), NULL);
4582 if (error)
4583 goto done;
4585 if (commit_id_str == NULL) {
4586 struct got_reference *head_ref;
4587 error = got_ref_open(&head_ref, repo, worktree ?
4588 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD, 0);
4589 if (error != NULL)
4590 goto done;
4591 error = got_ref_resolve(&commit_id, repo, head_ref);
4592 got_ref_close(head_ref);
4593 } else {
4594 error = got_repo_match_object_id(&commit_id, NULL,
4595 commit_id_str, GOT_OBJ_TYPE_COMMIT, 1, repo);
4597 if (error != NULL)
4598 goto done;
4600 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
4601 if (error)
4602 goto done;
4604 view = view_open(0, 0, 0, 0, TOG_VIEW_BLAME);
4605 if (view == NULL) {
4606 error = got_error_from_errno("view_open");
4607 goto done;
4610 error = got_object_resolve_symlinks(&link_target, in_repo_path,
4611 commit_id, repo);
4612 if (error)
4613 goto done;
4615 error = open_blame_view(view, link_target ? link_target : in_repo_path,
4616 commit_id, &refs, repo);
4617 if (error)
4618 goto done;
4619 if (worktree) {
4620 /* Release work tree lock. */
4621 got_worktree_close(worktree);
4622 worktree = NULL;
4624 error = view_loop(view);
4625 done:
4626 free(repo_path);
4627 free(in_repo_path);
4628 free(link_target);
4629 free(cwd);
4630 free(commit_id);
4631 if (worktree)
4632 got_worktree_close(worktree);
4633 if (repo)
4634 got_repo_close(repo);
4635 got_ref_list_free(&refs);
4636 return error;
4639 static const struct got_error *
4640 draw_tree_entries(struct tog_view *view,
4641 struct got_tree_entry **first_displayed_entry,
4642 struct got_tree_entry **last_displayed_entry,
4643 struct got_tree_entry **selected_entry, int *ndisplayed,
4644 const char *label, int show_ids, const char *parent_path,
4645 struct got_tree_object *tree, int selected, int limit,
4646 int isroot, struct tog_colors *colors, struct got_repository *repo)
4648 const struct got_error *err = NULL;
4649 struct got_tree_entry *te;
4650 wchar_t *wline;
4651 struct tog_color *tc;
4652 int width, n, i, nentries;
4654 *ndisplayed = 0;
4656 werase(view->window);
4658 if (limit == 0)
4659 return NULL;
4661 err = format_line(&wline, &width, label, view->ncols, 0);
4662 if (err)
4663 return err;
4664 if (view_needs_focus_indication(view))
4665 wstandout(view->window);
4666 tc = get_color(colors, TOG_COLOR_COMMIT);
4667 if (tc)
4668 wattr_on(view->window,
4669 COLOR_PAIR(tc->colorpair), NULL);
4670 waddwstr(view->window, wline);
4671 if (tc)
4672 wattr_off(view->window,
4673 COLOR_PAIR(tc->colorpair), NULL);
4674 if (view_needs_focus_indication(view))
4675 wstandend(view->window);
4676 free(wline);
4677 wline = NULL;
4678 if (width < view->ncols - 1)
4679 waddch(view->window, '\n');
4680 if (--limit <= 0)
4681 return NULL;
4682 err = format_line(&wline, &width, parent_path, view->ncols, 0);
4683 if (err)
4684 return err;
4685 waddwstr(view->window, wline);
4686 free(wline);
4687 wline = NULL;
4688 if (width < view->ncols - 1)
4689 waddch(view->window, '\n');
4690 if (--limit <= 0)
4691 return NULL;
4692 waddch(view->window, '\n');
4693 if (--limit <= 0)
4694 return NULL;
4696 if (*first_displayed_entry == NULL) {
4697 te = got_object_tree_get_first_entry(tree);
4698 if (selected == 0) {
4699 if (view->focussed)
4700 wstandout(view->window);
4701 *selected_entry = NULL;
4703 waddstr(view->window, " ..\n"); /* parent directory */
4704 if (selected == 0 && view->focussed)
4705 wstandend(view->window);
4706 (*ndisplayed)++;
4707 if (--limit <= 0)
4708 return NULL;
4709 n = 1;
4710 } else {
4711 n = 0;
4712 te = *first_displayed_entry;
4715 nentries = got_object_tree_get_nentries(tree);
4716 for (i = got_tree_entry_get_index(te); i < nentries; i++) {
4717 char *line = NULL, *id_str = NULL, *link_target = NULL;
4718 const char *modestr = "";
4719 mode_t mode;
4721 te = got_object_tree_get_entry(tree, i);
4722 mode = got_tree_entry_get_mode(te);
4724 if (show_ids) {
4725 err = got_object_id_str(&id_str,
4726 got_tree_entry_get_id(te));
4727 if (err)
4728 return got_error_from_errno(
4729 "got_object_id_str");
4731 if (got_object_tree_entry_is_submodule(te))
4732 modestr = "$";
4733 else if (S_ISLNK(mode)) {
4734 int i;
4736 err = got_tree_entry_get_symlink_target(&link_target,
4737 te, repo);
4738 if (err) {
4739 free(id_str);
4740 return err;
4742 for (i = 0; i < strlen(link_target); i++) {
4743 if (!isprint((unsigned char)link_target[i]))
4744 link_target[i] = '?';
4746 modestr = "@";
4748 else if (S_ISDIR(mode))
4749 modestr = "/";
4750 else if (mode & S_IXUSR)
4751 modestr = "*";
4752 if (asprintf(&line, "%s %s%s%s%s", id_str ? id_str : "",
4753 got_tree_entry_get_name(te), modestr,
4754 link_target ? " -> ": "",
4755 link_target ? link_target : "") == -1) {
4756 free(id_str);
4757 free(link_target);
4758 return got_error_from_errno("asprintf");
4760 free(id_str);
4761 free(link_target);
4762 err = format_line(&wline, &width, line, view->ncols, 0);
4763 if (err) {
4764 free(line);
4765 break;
4767 if (n == selected) {
4768 if (view->focussed)
4769 wstandout(view->window);
4770 *selected_entry = te;
4772 tc = match_color(colors, line);
4773 if (tc)
4774 wattr_on(view->window,
4775 COLOR_PAIR(tc->colorpair), NULL);
4776 waddwstr(view->window, wline);
4777 if (tc)
4778 wattr_off(view->window,
4779 COLOR_PAIR(tc->colorpair), NULL);
4780 if (width < view->ncols - 1)
4781 waddch(view->window, '\n');
4782 if (n == selected && view->focussed)
4783 wstandend(view->window);
4784 free(line);
4785 free(wline);
4786 wline = NULL;
4787 n++;
4788 (*ndisplayed)++;
4789 *last_displayed_entry = te;
4790 if (--limit <= 0)
4791 break;
4794 return err;
4797 static void
4798 tree_scroll_up(struct tog_view *view,
4799 struct got_tree_entry **first_displayed_entry, int maxscroll,
4800 struct got_tree_object *tree, int isroot)
4802 struct got_tree_entry *te;
4803 int i;
4805 if (*first_displayed_entry == NULL)
4806 return;
4808 te = got_object_tree_get_entry(tree, 0);
4809 if (*first_displayed_entry == te) {
4810 if (!isroot)
4811 *first_displayed_entry = NULL;
4812 return;
4815 i = 0;
4816 while (*first_displayed_entry && i < maxscroll) {
4817 *first_displayed_entry = got_tree_entry_get_prev(tree,
4818 *first_displayed_entry);
4819 i++;
4821 if (!isroot && te == got_object_tree_get_first_entry(tree) && i < maxscroll)
4822 *first_displayed_entry = NULL;
4825 static int
4826 tree_scroll_down(struct got_tree_entry **first_displayed_entry, int maxscroll,
4827 struct got_tree_entry *last_displayed_entry,
4828 struct got_tree_object *tree)
4830 struct got_tree_entry *next, *last;
4831 int n = 0;
4833 if (*first_displayed_entry)
4834 next = got_tree_entry_get_next(tree, *first_displayed_entry);
4835 else
4836 next = got_object_tree_get_first_entry(tree);
4838 last = last_displayed_entry;
4839 while (next && last && n++ < maxscroll) {
4840 last = got_tree_entry_get_next(tree, last);
4841 if (last) {
4842 *first_displayed_entry = next;
4843 next = got_tree_entry_get_next(tree, next);
4846 return n;
4849 static const struct got_error *
4850 tree_entry_path(char **path, struct tog_parent_trees *parents,
4851 struct got_tree_entry *te)
4853 const struct got_error *err = NULL;
4854 struct tog_parent_tree *pt;
4855 size_t len = 2; /* for leading slash and NUL */
4857 TAILQ_FOREACH(pt, parents, entry)
4858 len += strlen(got_tree_entry_get_name(pt->selected_entry))
4859 + 1 /* slash */;
4860 if (te)
4861 len += strlen(got_tree_entry_get_name(te));
4863 *path = calloc(1, len);
4864 if (path == NULL)
4865 return got_error_from_errno("calloc");
4867 (*path)[0] = '/';
4868 pt = TAILQ_LAST(parents, tog_parent_trees);
4869 while (pt) {
4870 const char *name = got_tree_entry_get_name(pt->selected_entry);
4871 if (strlcat(*path, name, len) >= len) {
4872 err = got_error(GOT_ERR_NO_SPACE);
4873 goto done;
4875 if (strlcat(*path, "/", len) >= len) {
4876 err = got_error(GOT_ERR_NO_SPACE);
4877 goto done;
4879 pt = TAILQ_PREV(pt, tog_parent_trees, entry);
4881 if (te) {
4882 if (strlcat(*path, got_tree_entry_get_name(te), len) >= len) {
4883 err = got_error(GOT_ERR_NO_SPACE);
4884 goto done;
4887 done:
4888 if (err) {
4889 free(*path);
4890 *path = NULL;
4892 return err;
4895 static const struct got_error *
4896 blame_tree_entry(struct tog_view **new_view, int begin_x,
4897 struct got_tree_entry *te, struct tog_parent_trees *parents,
4898 struct got_object_id *commit_id, struct got_reflist_head *refs,
4899 struct got_repository *repo)
4901 const struct got_error *err = NULL;
4902 char *path;
4903 struct tog_view *blame_view;
4905 *new_view = NULL;
4907 err = tree_entry_path(&path, parents, te);
4908 if (err)
4909 return err;
4911 blame_view = view_open(0, 0, 0, begin_x, TOG_VIEW_BLAME);
4912 if (blame_view == NULL) {
4913 err = got_error_from_errno("view_open");
4914 goto done;
4917 err = open_blame_view(blame_view, path, commit_id, refs, repo);
4918 if (err) {
4919 if (err->code == GOT_ERR_CANCELLED)
4920 err = NULL;
4921 view_close(blame_view);
4922 } else
4923 *new_view = blame_view;
4924 done:
4925 free(path);
4926 return err;
4929 static const struct got_error *
4930 log_tree_entry(struct tog_view **new_view, int begin_x,
4931 struct got_tree_entry *te, struct tog_parent_trees *parents,
4932 struct got_object_id *commit_id, struct got_reflist_head *refs,
4933 struct got_repository *repo)
4935 struct tog_view *log_view;
4936 const struct got_error *err = NULL;
4937 char *path;
4939 *new_view = NULL;
4941 log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
4942 if (log_view == NULL)
4943 return got_error_from_errno("view_open");
4945 err = tree_entry_path(&path, parents, te);
4946 if (err)
4947 return err;
4949 err = open_log_view(log_view, commit_id, refs, repo, NULL, path, 0);
4950 if (err)
4951 view_close(log_view);
4952 else
4953 *new_view = log_view;
4954 free(path);
4955 return err;
4958 static const struct got_error *
4959 open_tree_view(struct tog_view *view, struct got_tree_object *root,
4960 struct got_object_id *commit_id, struct got_reflist_head *refs,
4961 struct got_repository *repo)
4963 const struct got_error *err = NULL;
4964 char *commit_id_str = NULL;
4965 struct tog_tree_view_state *s = &view->state.tree;
4967 TAILQ_INIT(&s->parents);
4969 err = got_object_id_str(&commit_id_str, commit_id);
4970 if (err != NULL)
4971 goto done;
4973 if (asprintf(&s->tree_label, "commit %s", commit_id_str) == -1) {
4974 err = got_error_from_errno("asprintf");
4975 goto done;
4978 s->root = s->tree = root;
4979 s->first_displayed_entry = got_object_tree_get_entry(s->tree, 0);
4980 s->selected_entry = got_object_tree_get_entry(s->tree, 0);
4981 s->commit_id = got_object_id_dup(commit_id);
4982 if (s->commit_id == NULL) {
4983 err = got_error_from_errno("got_object_id_dup");
4984 goto done;
4986 s->refs = refs;
4987 s->repo = repo;
4989 SIMPLEQ_INIT(&s->colors);
4991 if (has_colors() && getenv("TOG_COLORS") != NULL) {
4992 err = add_color(&s->colors, "\\$$",
4993 TOG_COLOR_TREE_SUBMODULE,
4994 get_color_value("TOG_COLOR_TREE_SUBMODULE"));
4995 if (err)
4996 goto done;
4997 err = add_color(&s->colors, "@$", TOG_COLOR_TREE_SYMLINK,
4998 get_color_value("TOG_COLOR_TREE_SYMLINK"));
4999 if (err) {
5000 free_colors(&s->colors);
5001 goto done;
5003 err = add_color(&s->colors, "/$",
5004 TOG_COLOR_TREE_DIRECTORY,
5005 get_color_value("TOG_COLOR_TREE_DIRECTORY"));
5006 if (err) {
5007 free_colors(&s->colors);
5008 goto done;
5011 err = add_color(&s->colors, "\\*$",
5012 TOG_COLOR_TREE_EXECUTABLE,
5013 get_color_value("TOG_COLOR_TREE_EXECUTABLE"));
5014 if (err) {
5015 free_colors(&s->colors);
5016 goto done;
5019 err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
5020 get_color_value("TOG_COLOR_COMMIT"));
5021 if (err) {
5022 free_colors(&s->colors);
5023 goto done;
5027 view->show = show_tree_view;
5028 view->input = input_tree_view;
5029 view->close = close_tree_view;
5030 view->search_start = search_start_tree_view;
5031 view->search_next = search_next_tree_view;
5032 done:
5033 free(commit_id_str);
5034 if (err) {
5035 free(s->tree_label);
5036 s->tree_label = NULL;
5038 return err;
5041 static const struct got_error *
5042 close_tree_view(struct tog_view *view)
5044 struct tog_tree_view_state *s = &view->state.tree;
5046 free_colors(&s->colors);
5047 free(s->tree_label);
5048 s->tree_label = NULL;
5049 free(s->commit_id);
5050 s->commit_id = NULL;
5051 while (!TAILQ_EMPTY(&s->parents)) {
5052 struct tog_parent_tree *parent;
5053 parent = TAILQ_FIRST(&s->parents);
5054 TAILQ_REMOVE(&s->parents, parent, entry);
5055 free(parent);
5058 if (s->tree != s->root)
5059 got_object_tree_close(s->tree);
5060 got_object_tree_close(s->root);
5062 return NULL;
5065 static const struct got_error *
5066 search_start_tree_view(struct tog_view *view)
5068 struct tog_tree_view_state *s = &view->state.tree;
5070 s->matched_entry = NULL;
5071 return NULL;
5074 static int
5075 match_tree_entry(struct got_tree_entry *te, regex_t *regex)
5077 regmatch_t regmatch;
5079 return regexec(regex, got_tree_entry_get_name(te), 1, &regmatch,
5080 0) == 0;
5083 static const struct got_error *
5084 search_next_tree_view(struct tog_view *view)
5086 struct tog_tree_view_state *s = &view->state.tree;
5087 struct got_tree_entry *te = NULL;
5089 if (!view->searching) {
5090 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5091 return NULL;
5094 if (s->matched_entry) {
5095 if (view->searching == TOG_SEARCH_FORWARD) {
5096 if (s->selected_entry)
5097 te = got_tree_entry_get_next(s->tree,
5098 s->selected_entry);
5099 else
5100 te = got_object_tree_get_first_entry(s->tree);
5101 } else {
5102 if (s->selected_entry == NULL)
5103 te = got_object_tree_get_last_entry(s->tree);
5104 else
5105 te = got_tree_entry_get_prev(s->tree,
5106 s->selected_entry);
5108 } else {
5109 if (view->searching == TOG_SEARCH_FORWARD)
5110 te = got_object_tree_get_first_entry(s->tree);
5111 else
5112 te = got_object_tree_get_last_entry(s->tree);
5115 while (1) {
5116 if (te == NULL) {
5117 if (s->matched_entry == NULL) {
5118 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5119 return NULL;
5121 if (view->searching == TOG_SEARCH_FORWARD)
5122 te = got_object_tree_get_first_entry(s->tree);
5123 else
5124 te = got_object_tree_get_last_entry(s->tree);
5127 if (match_tree_entry(te, &view->regex)) {
5128 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5129 s->matched_entry = te;
5130 break;
5133 if (view->searching == TOG_SEARCH_FORWARD)
5134 te = got_tree_entry_get_next(s->tree, te);
5135 else
5136 te = got_tree_entry_get_prev(s->tree, te);
5139 if (s->matched_entry) {
5140 s->first_displayed_entry = s->matched_entry;
5141 s->selected = 0;
5144 return NULL;
5147 static const struct got_error *
5148 show_tree_view(struct tog_view *view)
5150 const struct got_error *err = NULL;
5151 struct tog_tree_view_state *s = &view->state.tree;
5152 char *parent_path;
5154 err = tree_entry_path(&parent_path, &s->parents, NULL);
5155 if (err)
5156 return err;
5158 err = draw_tree_entries(view, &s->first_displayed_entry,
5159 &s->last_displayed_entry, &s->selected_entry,
5160 &s->ndisplayed, s->tree_label, s->show_ids, parent_path,
5161 s->tree, s->selected, view->nlines, s->tree == s->root,
5162 &s->colors, s->repo);
5163 free(parent_path);
5165 view_vborder(view);
5166 return err;
5169 static const struct got_error *
5170 input_tree_view(struct tog_view **new_view, struct tog_view **dead_view,
5171 struct tog_view **focus_view, struct tog_view *view, int ch)
5173 const struct got_error *err = NULL;
5174 struct tog_tree_view_state *s = &view->state.tree;
5175 struct tog_view *log_view;
5176 int begin_x = 0, nscrolled;
5178 switch (ch) {
5179 case 'i':
5180 s->show_ids = !s->show_ids;
5181 break;
5182 case 'l':
5183 if (!s->selected_entry)
5184 break;
5185 if (view_is_parent_view(view))
5186 begin_x = view_split_begin_x(view->begin_x);
5187 err = log_tree_entry(&log_view, begin_x,
5188 s->selected_entry, &s->parents,
5189 s->commit_id, s->refs, s->repo);
5190 if (view_is_parent_view(view)) {
5191 err = view_close_child(view);
5192 if (err)
5193 return err;
5194 err = view_set_child(view, log_view);
5195 if (err) {
5196 view_close(log_view);
5197 break;
5199 *focus_view = log_view;
5200 view->child_focussed = 1;
5201 } else
5202 *new_view = log_view;
5203 break;
5204 case 'k':
5205 case KEY_UP:
5206 if (s->selected > 0) {
5207 s->selected--;
5208 if (s->selected == 0)
5209 break;
5211 if (s->selected > 0)
5212 break;
5213 tree_scroll_up(view, &s->first_displayed_entry, 1,
5214 s->tree, s->tree == s->root);
5215 break;
5216 case KEY_PPAGE:
5217 case CTRL('b'):
5218 tree_scroll_up(view, &s->first_displayed_entry,
5219 MAX(0, view->nlines - 4 - s->selected), s->tree,
5220 s->tree == s->root);
5221 s->selected = 0;
5222 if (got_object_tree_get_first_entry(s->tree) ==
5223 s->first_displayed_entry && s->tree != s->root)
5224 s->first_displayed_entry = NULL;
5225 break;
5226 case 'j':
5227 case KEY_DOWN:
5228 if (s->selected < s->ndisplayed - 1) {
5229 s->selected++;
5230 break;
5232 if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
5233 == NULL)
5234 /* can't scroll any further */
5235 break;
5236 tree_scroll_down(&s->first_displayed_entry, 1,
5237 s->last_displayed_entry, s->tree);
5238 break;
5239 case KEY_NPAGE:
5240 case CTRL('f'):
5241 if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
5242 == NULL) {
5243 /* can't scroll any further; move cursor down */
5244 if (s->selected < s->ndisplayed - 1)
5245 s->selected = s->ndisplayed - 1;
5246 break;
5248 nscrolled = tree_scroll_down(&s->first_displayed_entry,
5249 view->nlines, s->last_displayed_entry, s->tree);
5250 if (nscrolled < view->nlines) {
5251 int ndisplayed = 0;
5252 struct got_tree_entry *te;
5253 te = s->first_displayed_entry;
5254 do {
5255 ndisplayed++;
5256 te = got_tree_entry_get_next(s->tree, te);
5257 } while (te);
5258 s->selected = ndisplayed - 1;
5260 break;
5261 case KEY_ENTER:
5262 case '\r':
5263 case KEY_BACKSPACE:
5264 if (s->selected_entry == NULL || ch == KEY_BACKSPACE) {
5265 struct tog_parent_tree *parent;
5266 /* user selected '..' */
5267 if (s->tree == s->root)
5268 break;
5269 parent = TAILQ_FIRST(&s->parents);
5270 TAILQ_REMOVE(&s->parents, parent,
5271 entry);
5272 got_object_tree_close(s->tree);
5273 s->tree = parent->tree;
5274 s->first_displayed_entry =
5275 parent->first_displayed_entry;
5276 s->selected_entry =
5277 parent->selected_entry;
5278 s->selected = parent->selected;
5279 free(parent);
5280 } else if (S_ISDIR(got_tree_entry_get_mode(
5281 s->selected_entry))) {
5282 struct got_tree_object *subtree;
5283 err = got_object_open_as_tree(&subtree, s->repo,
5284 got_tree_entry_get_id(s->selected_entry));
5285 if (err)
5286 break;
5287 err = tree_view_visit_subtree(subtree, s);
5288 if (err) {
5289 got_object_tree_close(subtree);
5290 break;
5292 } else if (S_ISREG(got_tree_entry_get_mode(
5293 s->selected_entry))) {
5294 struct tog_view *blame_view;
5295 int begin_x = view_is_parent_view(view) ?
5296 view_split_begin_x(view->begin_x) : 0;
5298 err = blame_tree_entry(&blame_view, begin_x,
5299 s->selected_entry, &s->parents,
5300 s->commit_id, s->refs, s->repo);
5301 if (err)
5302 break;
5303 if (view_is_parent_view(view)) {
5304 err = view_close_child(view);
5305 if (err)
5306 return err;
5307 err = view_set_child(view, blame_view);
5308 if (err) {
5309 view_close(blame_view);
5310 break;
5312 *focus_view = blame_view;
5313 view->child_focussed = 1;
5314 } else
5315 *new_view = blame_view;
5317 break;
5318 case KEY_RESIZE:
5319 if (s->selected > view->nlines)
5320 s->selected = s->ndisplayed - 1;
5321 break;
5322 default:
5323 break;
5326 return err;
5329 __dead static void
5330 usage_tree(void)
5332 endwin();
5333 fprintf(stderr, "usage: %s tree [-c commit] [-r repository-path] [path]\n",
5334 getprogname());
5335 exit(1);
5338 static const struct got_error *
5339 cmd_tree(int argc, char *argv[])
5341 const struct got_error *error;
5342 struct got_repository *repo = NULL;
5343 struct got_worktree *worktree = NULL;
5344 struct got_reflist_head refs;
5345 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5346 struct got_object_id *commit_id = NULL;
5347 char *commit_id_arg = NULL;
5348 struct got_commit_object *commit = NULL;
5349 struct got_tree_object *tree = NULL;
5350 int ch;
5351 struct tog_view *view;
5353 SIMPLEQ_INIT(&refs);
5355 #ifndef PROFILE
5356 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
5357 NULL) == -1)
5358 err(1, "pledge");
5359 #endif
5361 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
5362 switch (ch) {
5363 case 'c':
5364 commit_id_arg = optarg;
5365 break;
5366 case 'r':
5367 repo_path = realpath(optarg, NULL);
5368 if (repo_path == NULL)
5369 return got_error_from_errno2("realpath",
5370 optarg);
5371 break;
5372 default:
5373 usage_tree();
5374 /* NOTREACHED */
5378 argc -= optind;
5379 argv += optind;
5381 if (argc > 1)
5382 usage_tree();
5384 cwd = getcwd(NULL, 0);
5385 if (cwd == NULL)
5386 return got_error_from_errno("getcwd");
5388 error = got_worktree_open(&worktree, cwd);
5389 if (error && error->code != GOT_ERR_NOT_WORKTREE)
5390 goto done;
5392 if (repo_path == NULL) {
5393 if (worktree)
5394 repo_path =
5395 strdup(got_worktree_get_repo_path(worktree));
5396 else
5397 repo_path = strdup(cwd);
5399 if (repo_path == NULL) {
5400 error = got_error_from_errno("strdup");
5401 goto done;
5404 error = got_repo_open(&repo, repo_path, NULL);
5405 if (error != NULL)
5406 goto done;
5408 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
5409 repo, worktree);
5410 if (error)
5411 goto done;
5413 init_curses();
5415 error = apply_unveil(got_repo_get_path(repo), NULL);
5416 if (error)
5417 goto done;
5419 error = got_repo_match_object_id(&commit_id, NULL,
5420 commit_id_arg ? commit_id_arg : GOT_REF_HEAD,
5421 GOT_OBJ_TYPE_COMMIT, 1, repo);
5422 if (error)
5423 goto done;
5425 error = got_object_open_as_commit(&commit, repo, commit_id);
5426 if (error)
5427 goto done;
5429 error = got_object_open_as_tree(&tree, repo,
5430 got_object_commit_get_tree_id(commit));
5431 if (error)
5432 goto done;
5434 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
5435 if (error)
5436 goto done;
5438 view = view_open(0, 0, 0, 0, TOG_VIEW_TREE);
5439 if (view == NULL) {
5440 error = got_error_from_errno("view_open");
5441 goto done;
5443 error = open_tree_view(view, tree, commit_id, &refs, repo);
5444 if (error)
5445 goto done;
5446 if (!got_path_is_root_dir(in_repo_path)) {
5447 error = tree_view_walk_path(&view->state.tree, commit_id,
5448 in_repo_path, repo);
5449 if (error)
5450 goto done;
5453 if (worktree) {
5454 /* Release work tree lock. */
5455 got_worktree_close(worktree);
5456 worktree = NULL;
5458 error = view_loop(view);
5459 done:
5460 free(repo_path);
5461 free(cwd);
5462 free(commit_id);
5463 if (commit)
5464 got_object_commit_close(commit);
5465 if (tree)
5466 got_object_tree_close(tree);
5467 if (repo)
5468 got_repo_close(repo);
5469 got_ref_list_free(&refs);
5470 return error;
5473 static void
5474 list_commands(FILE *fp)
5476 int i;
5478 fprintf(fp, "commands:");
5479 for (i = 0; i < nitems(tog_commands); i++) {
5480 struct tog_cmd *cmd = &tog_commands[i];
5481 fprintf(fp, " %s", cmd->name);
5483 fputc('\n', fp);
5486 __dead static void
5487 usage(int hflag, int status)
5489 FILE *fp = (status == 0) ? stdout : stderr;
5491 fprintf(fp, "usage: %s [-h] [-V | --version] [command] [arg ...]\n",
5492 getprogname());
5493 if (hflag) {
5494 fprintf(fp, "lazy usage: %s path\n", getprogname());
5495 list_commands(fp);
5497 exit(status);
5500 static char **
5501 make_argv(int argc, ...)
5503 va_list ap;
5504 char **argv;
5505 int i;
5507 va_start(ap, argc);
5509 argv = calloc(argc, sizeof(char *));
5510 if (argv == NULL)
5511 err(1, "calloc");
5512 for (i = 0; i < argc; i++) {
5513 argv[i] = strdup(va_arg(ap, char *));
5514 if (argv[i] == NULL)
5515 err(1, "strdup");
5518 va_end(ap);
5519 return argv;
5523 * Try to convert 'tog path' into a 'tog log path' command.
5524 * The user could simply have mistyped the command rather than knowingly
5525 * provided a path. So check whether argv[0] can in fact be resolved
5526 * to a path in the HEAD commit and print a special error if not.
5527 * This hack is for mpi@ <3
5529 static const struct got_error *
5530 tog_log_with_path(int argc, char *argv[])
5532 const struct got_error *error = NULL;
5533 struct tog_cmd *cmd = NULL;
5534 struct got_repository *repo = NULL;
5535 struct got_worktree *worktree = NULL;
5536 struct got_object_id *commit_id = NULL, *id = NULL;
5537 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5538 char *commit_id_str = NULL, **cmd_argv = NULL;
5540 cwd = getcwd(NULL, 0);
5541 if (cwd == NULL)
5542 return got_error_from_errno("getcwd");
5544 error = got_worktree_open(&worktree, cwd);
5545 if (error && error->code != GOT_ERR_NOT_WORKTREE)
5546 goto done;
5548 if (worktree)
5549 repo_path = strdup(got_worktree_get_repo_path(worktree));
5550 else
5551 repo_path = strdup(cwd);
5552 if (repo_path == NULL) {
5553 error = got_error_from_errno("strdup");
5554 goto done;
5557 error = got_repo_open(&repo, repo_path, NULL);
5558 if (error != NULL)
5559 goto done;
5561 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
5562 repo, worktree);
5563 if (error)
5564 goto done;
5566 error = got_repo_match_object_id(&commit_id, NULL, worktree ?
5567 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
5568 GOT_OBJ_TYPE_COMMIT, 1, repo);
5569 if (error)
5570 goto done;
5572 if (worktree) {
5573 got_worktree_close(worktree);
5574 worktree = NULL;
5577 error = got_object_id_by_path(&id, repo, commit_id, in_repo_path);
5578 if (error) {
5579 if (error->code != GOT_ERR_NO_TREE_ENTRY)
5580 goto done;
5581 fprintf(stderr, "%s: '%s' is no known command or path\n",
5582 getprogname(), argv[0]);
5583 usage(1, 1);
5584 /* not reached */
5587 got_repo_close(repo);
5588 repo = NULL;
5590 error = got_object_id_str(&commit_id_str, commit_id);
5591 if (error)
5592 goto done;
5594 cmd = &tog_commands[0]; /* log */
5595 argc = 4;
5596 cmd_argv = make_argv(argc, cmd->name, "-c", commit_id_str, argv[0]);
5597 error = cmd->cmd_main(argc, cmd_argv);
5598 done:
5599 if (repo)
5600 got_repo_close(repo);
5601 if (worktree)
5602 got_worktree_close(worktree);
5603 free(id);
5604 free(commit_id_str);
5605 free(commit_id);
5606 free(cwd);
5607 free(repo_path);
5608 free(in_repo_path);
5609 if (cmd_argv) {
5610 int i;
5611 for (i = 0; i < argc; i++)
5612 free(cmd_argv[i]);
5613 free(cmd_argv);
5615 return error;
5618 int
5619 main(int argc, char *argv[])
5621 const struct got_error *error = NULL;
5622 struct tog_cmd *cmd = NULL;
5623 int ch, hflag = 0, Vflag = 0;
5624 char **cmd_argv = NULL;
5625 static struct option longopts[] = {
5626 { "version", no_argument, NULL, 'V' },
5627 { NULL, 0, NULL, 0}
5630 setlocale(LC_CTYPE, "");
5632 while ((ch = getopt_long(argc, argv, "+hV", longopts, NULL)) != -1) {
5633 switch (ch) {
5634 case 'h':
5635 hflag = 1;
5636 break;
5637 case 'V':
5638 Vflag = 1;
5639 break;
5640 default:
5641 usage(hflag, 1);
5642 /* NOTREACHED */
5646 argc -= optind;
5647 argv += optind;
5648 optind = 1;
5649 optreset = 1;
5651 if (Vflag) {
5652 got_version_print_str();
5653 return 0;
5656 if (argc == 0) {
5657 if (hflag)
5658 usage(hflag, 0);
5659 /* Build an argument vector which runs a default command. */
5660 cmd = &tog_commands[0];
5661 argc = 1;
5662 cmd_argv = make_argv(argc, cmd->name);
5663 } else {
5664 int i;
5666 /* Did the user specify a command? */
5667 for (i = 0; i < nitems(tog_commands); i++) {
5668 if (strncmp(tog_commands[i].name, argv[0],
5669 strlen(argv[0])) == 0) {
5670 cmd = &tog_commands[i];
5671 break;
5676 if (cmd == NULL) {
5677 if (argc != 1)
5678 usage(0, 1);
5679 /* No command specified; try log with a path */
5680 error = tog_log_with_path(argc, argv);
5681 } else {
5682 if (hflag)
5683 cmd->cmd_usage();
5684 else
5685 error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
5688 endwin();
5689 putchar('\n');
5690 if (cmd_argv) {
5691 int i;
5692 for (i = 0; i < argc; i++)
5693 free(cmd_argv[i]);
5694 free(cmd_argv);
5697 if (error && error->code != GOT_ERR_CANCELLED)
5698 fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
5699 return 0;