Blob


1 /*
2 * Copyright (c) 2018, 2019 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 <errno.h>
22 #define _XOPEN_SOURCE_EXTENDED
23 #include <curses.h>
24 #undef _XOPEN_SOURCE_EXTENDED
25 #include <panel.h>
26 #include <locale.h>
27 #include <stdlib.h>
28 #include <stdio.h>
29 #include <getopt.h>
30 #include <string.h>
31 #include <err.h>
32 #include <unistd.h>
33 #include <util.h>
34 #include <limits.h>
35 #include <wchar.h>
36 #include <time.h>
37 #include <pthread.h>
38 #include <libgen.h>
40 #include "got_error.h"
41 #include "got_object.h"
42 #include "got_reference.h"
43 #include "got_repository.h"
44 #include "got_diff.h"
45 #include "got_opentemp.h"
46 #include "got_commit_graph.h"
47 #include "got_utf8.h"
48 #include "got_blame.h"
49 #include "got_privsep.h"
50 #include "got_worktree.h"
52 #ifndef MIN
53 #define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
54 #endif
56 #ifndef MAX
57 #define MAX(_a,_b) ((_a) > (_b) ? (_a) : (_b))
58 #endif
61 #ifndef nitems
62 #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
63 #endif
65 struct tog_cmd {
66 const char *name;
67 const struct got_error *(*cmd_main)(int, char *[]);
68 void (*cmd_usage)(void);
69 const char *descr;
70 };
72 __dead static void usage(void);
73 __dead static void usage_log(void);
74 __dead static void usage_diff(void);
75 __dead static void usage_blame(void);
76 __dead static void usage_tree(void);
78 static const struct got_error* cmd_log(int, char *[]);
79 static const struct got_error* cmd_diff(int, char *[]);
80 static const struct got_error* cmd_blame(int, char *[]);
81 static const struct got_error* cmd_tree(int, char *[]);
83 static struct tog_cmd tog_commands[] = {
84 { "log", cmd_log, usage_log,
85 "show repository history" },
86 { "diff", cmd_diff, usage_diff,
87 "compare files and directories" },
88 { "blame", cmd_blame, usage_blame,
89 "show line-by-line file history" },
90 { "tree", cmd_tree, usage_tree,
91 "browse trees in repository" },
92 };
94 enum tog_view_type {
95 TOG_VIEW_DIFF,
96 TOG_VIEW_LOG,
97 TOG_VIEW_BLAME,
98 TOG_VIEW_TREE
99 };
101 struct tog_diff_view_state {
102 struct got_object_id *id1, *id2;
103 FILE *f;
104 int first_displayed_line;
105 int last_displayed_line;
106 int eof;
107 int diff_context;
108 struct got_repository *repo;
109 };
111 struct commit_queue_entry {
112 TAILQ_ENTRY(commit_queue_entry) entry;
113 struct got_object_id *id;
114 struct got_commit_object *commit;
115 int idx;
116 };
117 TAILQ_HEAD(commit_queue_head, commit_queue_entry);
118 struct commit_queue {
119 int ncommits;
120 struct commit_queue_head head;
121 };
123 pthread_mutex_t tog_mutex = PTHREAD_MUTEX_INITIALIZER;
125 struct tog_log_thread_args {
126 pthread_cond_t need_commits;
127 int commits_needed;
128 struct got_commit_graph *graph;
129 struct commit_queue *commits;
130 const char *in_repo_path;
131 struct got_object_id *start_id;
132 struct got_repository *repo;
133 int log_complete;
134 sig_atomic_t *quit;
135 struct tog_view *view;
136 struct commit_queue_entry **first_displayed_entry;
137 struct commit_queue_entry **selected_entry;
138 };
140 struct tog_log_view_state {
141 struct commit_queue commits;
142 struct commit_queue_entry *first_displayed_entry;
143 struct commit_queue_entry *last_displayed_entry;
144 struct commit_queue_entry *selected_entry;
145 int selected;
146 char *in_repo_path;
147 struct got_repository *repo;
148 struct got_object_id *start_id;
149 sig_atomic_t quit;
150 pthread_t thread;
151 struct tog_log_thread_args thread_args;
152 };
154 struct tog_blame_cb_args {
155 struct tog_blame_line *lines; /* one per line */
156 int nlines;
158 struct tog_view *view;
159 struct got_object_id *commit_id;
160 int *quit;
161 };
163 struct tog_blame_thread_args {
164 const char *path;
165 struct got_repository *repo;
166 struct tog_blame_cb_args *cb_args;
167 int *complete;
168 };
170 struct tog_blame {
171 FILE *f;
172 size_t filesize;
173 struct tog_blame_line *lines;
174 int nlines;
175 pthread_t thread;
176 struct tog_blame_thread_args thread_args;
177 struct tog_blame_cb_args cb_args;
178 const char *path;
179 };
181 struct tog_blame_view_state {
182 int first_displayed_line;
183 int last_displayed_line;
184 int selected_line;
185 int blame_complete;
186 int eof;
187 int done;
188 struct got_object_id_queue blamed_commits;
189 struct got_object_qid *blamed_commit;
190 char *path;
191 struct got_repository *repo;
192 struct got_object_id *commit_id;
193 struct tog_blame blame;
194 };
196 struct tog_parent_tree {
197 TAILQ_ENTRY(tog_parent_tree) entry;
198 struct got_tree_object *tree;
199 struct got_tree_entry *first_displayed_entry;
200 struct got_tree_entry *selected_entry;
201 int selected;
202 };
204 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
206 struct tog_tree_view_state {
207 char *tree_label;
208 struct got_tree_object *root;
209 struct got_tree_object *tree;
210 const struct got_tree_entries *entries;
211 struct got_tree_entry *first_displayed_entry;
212 struct got_tree_entry *last_displayed_entry;
213 struct got_tree_entry *selected_entry;
214 int ndisplayed, selected, show_ids;
215 struct tog_parent_trees parents;
216 struct got_object_id *commit_id;
217 struct got_repository *repo;
218 };
220 /*
221 * We implement two types of views: parent views and child views.
223 * The 'Tab' key switches between a parent view and its child view.
224 * Child views are shown side-by-side to their parent view, provided
225 * there is enough screen estate.
227 * When a new view is opened from within a parent view, this new view
228 * becomes a child view of the parent view, replacing any existing child.
230 * When a new view is opened from within a child view, this new view
231 * becomes a parent view which will obscure the views below until the
232 * user quits the new parent view by typing 'q'.
234 * This list of views contains parent views only.
235 * Child views are only pointed to by their parent view.
236 */
237 TAILQ_HEAD(tog_view_list_head, tog_view);
239 struct tog_view {
240 TAILQ_ENTRY(tog_view) entry;
241 WINDOW *window;
242 PANEL *panel;
243 int nlines, ncols, begin_y, begin_x;
244 int lines, cols; /* copies of LINES and COLS */
245 int focussed;
246 struct tog_view *parent;
247 struct tog_view *child;
248 int child_focussed;
250 /* type-specific state */
251 enum tog_view_type type;
252 union {
253 struct tog_diff_view_state diff;
254 struct tog_log_view_state log;
255 struct tog_blame_view_state blame;
256 struct tog_tree_view_state tree;
257 } state;
259 const struct got_error *(*show)(struct tog_view *);
260 const struct got_error *(*input)(struct tog_view **,
261 struct tog_view **, struct tog_view**, struct tog_view *, int);
262 const struct got_error *(*close)(struct tog_view *);
263 };
265 static const struct got_error *open_diff_view(struct tog_view *,
266 struct got_object_id *, struct got_object_id *, struct got_repository *);
267 static const struct got_error *show_diff_view(struct tog_view *);
268 static const struct got_error *input_diff_view(struct tog_view **,
269 struct tog_view **, struct tog_view **, struct tog_view *, int);
270 static const struct got_error* close_diff_view(struct tog_view *);
272 static const struct got_error *open_log_view(struct tog_view *,
273 struct got_object_id *, struct got_repository *, const char *, int);
274 static const struct got_error * show_log_view(struct tog_view *);
275 static const struct got_error *input_log_view(struct tog_view **,
276 struct tog_view **, struct tog_view **, struct tog_view *, int);
277 static const struct got_error *close_log_view(struct tog_view *);
279 static const struct got_error *open_blame_view(struct tog_view *, char *,
280 struct got_object_id *, struct got_repository *);
281 static const struct got_error *show_blame_view(struct tog_view *);
282 static const struct got_error *input_blame_view(struct tog_view **,
283 struct tog_view **, struct tog_view **, struct tog_view *, int);
284 static const struct got_error *close_blame_view(struct tog_view *);
286 static const struct got_error *open_tree_view(struct tog_view *,
287 struct got_tree_object *, struct got_object_id *, struct got_repository *);
288 static const struct got_error *show_tree_view(struct tog_view *);
289 static const struct got_error *input_tree_view(struct tog_view **,
290 struct tog_view **, struct tog_view **, struct tog_view *, int);
291 static const struct got_error *close_tree_view(struct tog_view *);
293 static volatile sig_atomic_t tog_sigwinch_received;
295 static void
296 tog_sigwinch(int signo)
298 tog_sigwinch_received = 1;
301 static const struct got_error *
302 view_close(struct tog_view *view)
304 const struct got_error *err = NULL;
306 if (view->child) {
307 view_close(view->child);
308 view->child = NULL;
310 if (view->close)
311 err = view->close(view);
312 if (view->panel)
313 del_panel(view->panel);
314 if (view->window)
315 delwin(view->window);
316 free(view);
317 return err;
320 static struct tog_view *
321 view_open(int nlines, int ncols, int begin_y, int begin_x,
322 enum tog_view_type type)
324 struct tog_view *view = calloc(1, sizeof(*view));
326 if (view == NULL)
327 return NULL;
329 view->type = type;
330 view->lines = LINES;
331 view->cols = COLS;
332 view->nlines = nlines ? nlines : LINES - begin_y;
333 view->ncols = ncols ? ncols : COLS - begin_x;
334 view->begin_y = begin_y;
335 view->begin_x = begin_x;
336 view->window = newwin(nlines, ncols, begin_y, begin_x);
337 if (view->window == NULL) {
338 view_close(view);
339 return NULL;
341 view->panel = new_panel(view->window);
342 if (view->panel == NULL ||
343 set_panel_userptr(view->panel, view) != OK) {
344 view_close(view);
345 return NULL;
348 keypad(view->window, TRUE);
349 return view;
352 static int
353 view_split_begin_x(int begin_x)
355 if (begin_x > 0 || COLS < 120)
356 return 0;
357 return (COLS - MAX(COLS / 2, 80));
360 static const struct got_error *view_resize(struct tog_view *);
362 static const struct got_error *
363 view_splitscreen(struct tog_view *view)
365 const struct got_error *err = NULL;
367 view->begin_y = 0;
368 view->begin_x = view_split_begin_x(0);
369 view->nlines = LINES;
370 view->ncols = COLS - view->begin_x;
371 view->lines = LINES;
372 view->cols = COLS;
373 err = view_resize(view);
374 if (err)
375 return err;
377 if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
378 return got_error_from_errno();
380 return NULL;
383 static const struct got_error *
384 view_fullscreen(struct tog_view *view)
386 const struct got_error *err = NULL;
388 view->begin_x = 0;
389 view->begin_y = 0;
390 view->nlines = LINES;
391 view->ncols = COLS;
392 view->lines = LINES;
393 view->cols = COLS;
394 err = view_resize(view);
395 if (err)
396 return err;
398 if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
399 return got_error_from_errno();
401 return NULL;
404 static int
405 view_is_parent_view(struct tog_view *view)
407 return view->parent == NULL;
410 static const struct got_error *
411 view_resize(struct tog_view *view)
413 int nlines, ncols;
415 if (view->lines > LINES)
416 nlines = view->nlines - (view->lines - LINES);
417 else
418 nlines = view->nlines + (LINES - view->lines);
420 if (view->cols > COLS)
421 ncols = view->ncols - (view->cols - COLS);
422 else
423 ncols = view->ncols + (COLS - view->cols);
425 if (wresize(view->window, nlines, ncols) == ERR)
426 return got_error_from_errno();
427 if (replace_panel(view->panel, view->window) == ERR)
428 return got_error_from_errno();
429 wclear(view->window);
431 view->nlines = nlines;
432 view->ncols = ncols;
433 view->lines = LINES;
434 view->cols = COLS;
436 if (view->child) {
437 view->child->begin_x = view_split_begin_x(view->begin_x);
438 if (view->child->begin_x == 0) {
439 view_fullscreen(view->child);
440 if (view->child->focussed)
441 show_panel(view->child->panel);
442 else
443 show_panel(view->panel);
444 } else {
445 view_splitscreen(view->child);
446 show_panel(view->child->panel);
450 return NULL;
453 static const struct got_error *
454 view_close_child(struct tog_view *view)
456 const struct got_error *err = NULL;
458 if (view->child == NULL)
459 return NULL;
461 err = view_close(view->child);
462 view->child = NULL;
463 return err;
466 static const struct got_error *
467 view_set_child(struct tog_view *view, struct tog_view *child)
469 const struct got_error *err = NULL;
471 view->child = child;
472 child->parent = view;
473 return err;
476 static int
477 view_is_splitscreen(struct tog_view *view)
479 return !view_is_parent_view(view) && view->begin_x > 0;
482 static void
483 tog_resizeterm(void)
485 int cols, lines;
486 struct winsize size;
488 if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &size) < 0) {
489 cols = 80; /* Default */
490 lines = 24;
491 } else {
492 cols = size.ws_col;
493 lines = size.ws_row;
495 resize_term(lines, cols);
498 static const struct got_error *
499 view_input(struct tog_view **new, struct tog_view **dead,
500 struct tog_view **focus, int *done, struct tog_view *view,
501 struct tog_view_list_head *views)
503 const struct got_error *err = NULL;
504 struct tog_view *v;
505 int ch, errcode;
507 *new = NULL;
508 *dead = NULL;
509 *focus = NULL;
511 nodelay(stdscr, FALSE);
512 /* Allow threads to make progress while we are waiting for input. */
513 errcode = pthread_mutex_unlock(&tog_mutex);
514 if (errcode)
515 return got_error_set_errno(errcode);
516 ch = wgetch(view->window);
517 errcode = pthread_mutex_lock(&tog_mutex);
518 if (errcode)
519 return got_error_set_errno(errcode);
520 nodelay(stdscr, TRUE);
522 if (tog_sigwinch_received) {
523 tog_resizeterm();
524 tog_sigwinch_received = 0;
525 TAILQ_FOREACH(v, views, entry) {
526 err = view_resize(v);
527 if (err)
528 return err;
529 err = v->input(new, dead, focus, v, KEY_RESIZE);
530 if (err)
531 return err;
535 switch (ch) {
536 case ERR:
537 break;
538 case '\t':
539 if (view->child) {
540 *focus = view->child;
541 view->child_focussed = 1;
542 } else if (view->parent) {
543 *focus = view->parent;
544 view->parent->child_focussed = 0;
546 break;
547 case 'q':
548 err = view->input(new, dead, focus, view, ch);
549 *dead = view;
550 break;
551 case 'Q':
552 *done = 1;
553 break;
554 case 'f':
555 if (view_is_parent_view(view)) {
556 if (view->child == NULL)
557 break;
558 if (view_is_splitscreen(view->child)) {
559 *focus = view->child;
560 view->child_focussed = 1;
561 err = view_fullscreen(view->child);
562 } else
563 err = view_splitscreen(view->child);
564 if (err)
565 break;
566 err = view->child->input(new, dead, focus,
567 view->child, KEY_RESIZE);
568 } else {
569 if (view_is_splitscreen(view)) {
570 *focus = view;
571 view->parent->child_focussed = 1;
572 err = view_fullscreen(view);
573 } else {
574 err = view_splitscreen(view);
576 if (err)
577 break;
578 err = view->input(new, dead, focus, view,
579 KEY_RESIZE);
581 break;
582 case KEY_RESIZE:
583 break;
584 default:
585 err = view->input(new, dead, focus, view, ch);
586 break;
589 return err;
592 void
593 view_vborder(struct tog_view *view)
595 PANEL *panel;
596 struct tog_view *view_above;
598 if (view->parent)
599 return view_vborder(view->parent);
601 panel = panel_above(view->panel);
602 if (panel == NULL)
603 return;
605 view_above = panel_userptr(panel);
606 mvwvline(view->window, view->begin_y, view_above->begin_x - 1,
607 got_locale_is_utf8() ? ACS_VLINE : '|', view->nlines);
610 int
611 view_needs_focus_indication(struct tog_view *view)
613 if (view_is_parent_view(view)) {
614 if (view->child == NULL || view->child_focussed)
615 return 0;
616 if (!view_is_splitscreen(view->child))
617 return 0;
618 } else if (!view_is_splitscreen(view))
619 return 0;
621 return view->focussed;
624 static const struct got_error *
625 view_loop(struct tog_view *view)
627 const struct got_error *err = NULL;
628 struct tog_view_list_head views;
629 struct tog_view *new_view, *dead_view, *focus_view, *main_view;
630 int fast_refresh = 10;
631 int done = 0, errcode;
633 errcode = pthread_mutex_lock(&tog_mutex);
634 if (errcode)
635 return got_error_set_errno(errcode);
637 TAILQ_INIT(&views);
638 TAILQ_INSERT_HEAD(&views, view, entry);
640 main_view = view;
641 view->focussed = 1;
642 err = view->show(view);
643 if (err)
644 return err;
645 update_panels();
646 doupdate();
647 while (!TAILQ_EMPTY(&views) && !done) {
648 /* Refresh fast during initialization, then become slower. */
649 if (fast_refresh && fast_refresh-- == 0)
650 halfdelay(10); /* switch to once per second */
652 err = view_input(&new_view, &dead_view, &focus_view, &done,
653 view, &views);
654 if (err)
655 break;
656 if (dead_view) {
657 struct tog_view *prev = NULL;
659 if (view_is_parent_view(dead_view))
660 prev = TAILQ_PREV(dead_view,
661 tog_view_list_head, entry);
662 else if (view->parent != dead_view)
663 prev = view->parent;
665 if (dead_view->parent)
666 dead_view->parent->child = NULL;
667 else
668 TAILQ_REMOVE(&views, dead_view, entry);
670 err = view_close(dead_view);
671 if (err || dead_view == main_view)
672 goto done;
674 if (view == dead_view) {
675 if (focus_view)
676 view = focus_view;
677 else if (prev)
678 view = prev;
679 else if (!TAILQ_EMPTY(&views))
680 view = TAILQ_LAST(&views,
681 tog_view_list_head);
682 else
683 view = NULL;
684 if (view) {
685 if (view->child && view->child_focussed)
686 focus_view = view->child;
687 else
688 focus_view = view;
692 if (new_view) {
693 struct tog_view *v, *t;
694 /* Only allow one parent view per type. */
695 TAILQ_FOREACH_SAFE(v, &views, entry, t) {
696 if (v->type != new_view->type)
697 continue;
698 TAILQ_REMOVE(&views, v, entry);
699 err = view_close(v);
700 if (err)
701 goto done;
702 break;
704 TAILQ_INSERT_TAIL(&views, new_view, entry);
705 view = new_view;
706 if (focus_view == NULL)
707 focus_view = new_view;
709 if (focus_view) {
710 show_panel(focus_view->panel);
711 if (view)
712 view->focussed = 0;
713 focus_view->focussed = 1;
714 view = focus_view;
715 if (new_view)
716 show_panel(new_view->panel);
717 if (view->child && view_is_splitscreen(view->child))
718 show_panel(view->child->panel);
720 if (view) {
721 if (focus_view == NULL) {
722 view->focussed = 1;
723 show_panel(view->panel);
724 if (view->child && view_is_splitscreen(view->child))
725 show_panel(view->child->panel);
726 focus_view = view;
728 if (view->parent) {
729 err = view->parent->show(view->parent);
730 if (err)
731 goto done;
733 err = view->show(view);
734 if (err)
735 goto done;
736 if (view->child) {
737 err = view->child->show(view->child);
738 if (err)
739 goto done;
741 update_panels();
742 doupdate();
745 done:
746 while (!TAILQ_EMPTY(&views)) {
747 view = TAILQ_FIRST(&views);
748 TAILQ_REMOVE(&views, view, entry);
749 view_close(view);
752 errcode = pthread_mutex_unlock(&tog_mutex);
753 if (errcode)
754 return got_error_set_errno(errcode);
756 return err;
759 __dead static void
760 usage_log(void)
762 endwin();
763 fprintf(stderr,
764 "usage: %s log [-c commit] [-r repository-path] [path]\n",
765 getprogname());
766 exit(1);
769 /* Create newly allocated wide-character string equivalent to a byte string. */
770 static const struct got_error *
771 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
773 char *vis = NULL;
774 const struct got_error *err = NULL;
776 *ws = NULL;
777 *wlen = mbstowcs(NULL, s, 0);
778 if (*wlen == (size_t)-1) {
779 int vislen;
780 if (errno != EILSEQ)
781 return got_error_from_errno();
783 /* byte string invalid in current encoding; try to "fix" it */
784 err = got_mbsavis(&vis, &vislen, s);
785 if (err)
786 return err;
787 *wlen = mbstowcs(NULL, vis, 0);
788 if (*wlen == (size_t)-1) {
789 err = got_error_from_errno(); /* give up */
790 goto done;
794 *ws = calloc(*wlen + 1, sizeof(*ws));
795 if (*ws == NULL) {
796 err = got_error_from_errno();
797 goto done;
800 if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
801 err = got_error_from_errno();
802 done:
803 free(vis);
804 if (err) {
805 free(*ws);
806 *ws = NULL;
807 *wlen = 0;
809 return err;
812 /* Format a line for display, ensuring that it won't overflow a width limit. */
813 static const struct got_error *
814 format_line(wchar_t **wlinep, int *widthp, const char *line, int wlimit)
816 const struct got_error *err = NULL;
817 int cols = 0;
818 wchar_t *wline = NULL;
819 size_t wlen;
820 int i;
822 *wlinep = NULL;
823 *widthp = 0;
825 err = mbs2ws(&wline, &wlen, line);
826 if (err)
827 return err;
829 i = 0;
830 while (i < wlen && cols < wlimit) {
831 int width = wcwidth(wline[i]);
832 switch (width) {
833 case 0:
834 i++;
835 break;
836 case 1:
837 case 2:
838 if (cols + width <= wlimit)
839 cols += width;
840 i++;
841 break;
842 case -1:
843 if (wline[i] == L'\t')
844 cols += TABSIZE - ((cols + 1) % TABSIZE);
845 i++;
846 break;
847 default:
848 err = got_error_from_errno();
849 goto done;
852 wline[i] = L'\0';
853 if (widthp)
854 *widthp = cols;
855 done:
856 if (err)
857 free(wline);
858 else
859 *wlinep = wline;
860 return err;
863 static const struct got_error *
864 draw_commit(struct tog_view *view, struct got_commit_object *commit,
865 struct got_object_id *id)
867 const struct got_error *err = NULL;
868 char datebuf[10]; /* YY-MM-DD + SPACE + NUL */
869 char *logmsg0 = NULL, *logmsg = NULL;
870 char *author0 = NULL, *author = NULL;
871 wchar_t *wlogmsg = NULL, *wauthor = NULL;
872 int author_width, logmsg_width;
873 char *newline, *smallerthan;
874 char *line = NULL;
875 int col, limit;
876 static const size_t date_display_cols = 9;
877 static const size_t author_display_cols = 16;
878 const int avail = view->ncols;
879 struct tm tm;
880 time_t committer_time;
882 committer_time = got_object_commit_get_committer_time(commit);
883 if (localtime_r(&committer_time, &tm) == NULL)
884 return got_error_from_errno();
885 if (strftime(datebuf, sizeof(datebuf), "%g/%m/%d ", &tm)
886 >= sizeof(datebuf))
887 return got_error(GOT_ERR_NO_SPACE);
889 if (avail < date_display_cols)
890 limit = MIN(sizeof(datebuf) - 1, avail);
891 else
892 limit = MIN(date_display_cols, sizeof(datebuf) - 1);
893 waddnstr(view->window, datebuf, limit);
894 col = limit + 1;
895 if (col > avail)
896 goto done;
898 author0 = strdup(got_object_commit_get_author(commit));
899 if (author0 == NULL) {
900 err = got_error_from_errno();
901 goto done;
903 author = author0;
904 smallerthan = strchr(author, '<');
905 if (smallerthan)
906 *smallerthan = '\0';
907 else {
908 char *at = strchr(author, '@');
909 if (at)
910 *at = '\0';
912 limit = avail - col;
913 err = format_line(&wauthor, &author_width, author, limit);
914 if (err)
915 goto done;
916 waddwstr(view->window, wauthor);
917 col += author_width;
918 while (col <= avail && author_width < author_display_cols + 1) {
919 waddch(view->window, ' ');
920 col++;
921 author_width++;
923 if (col > avail)
924 goto done;
926 logmsg0 = strdup(got_object_commit_get_logmsg(commit));
927 if (logmsg0 == NULL) {
928 err = got_error_from_errno();
929 goto done;
931 logmsg = logmsg0;
932 while (*logmsg == '\n')
933 logmsg++;
934 newline = strchr(logmsg, '\n');
935 if (newline)
936 *newline = '\0';
937 limit = avail - col;
938 err = format_line(&wlogmsg, &logmsg_width, logmsg, limit);
939 if (err)
940 goto done;
941 waddwstr(view->window, wlogmsg);
942 col += logmsg_width;
943 while (col <= avail) {
944 waddch(view->window, ' ');
945 col++;
947 done:
948 free(logmsg0);
949 free(wlogmsg);
950 free(author0);
951 free(wauthor);
952 free(line);
953 return err;
956 static struct commit_queue_entry *
957 alloc_commit_queue_entry(struct got_commit_object *commit,
958 struct got_object_id *id)
960 struct commit_queue_entry *entry;
962 entry = calloc(1, sizeof(*entry));
963 if (entry == NULL)
964 return NULL;
966 entry->id = id;
967 entry->commit = commit;
968 return entry;
971 static void
972 pop_commit(struct commit_queue *commits)
974 struct commit_queue_entry *entry;
976 entry = TAILQ_FIRST(&commits->head);
977 TAILQ_REMOVE(&commits->head, entry, entry);
978 got_object_commit_close(entry->commit);
979 commits->ncommits--;
980 /* Don't free entry->id! It is owned by the commit graph. */
981 free(entry);
984 static void
985 free_commits(struct commit_queue *commits)
987 while (!TAILQ_EMPTY(&commits->head))
988 pop_commit(commits);
991 static const struct got_error *
992 queue_commits(struct got_commit_graph *graph, struct commit_queue *commits,
993 int minqueue, struct got_repository *repo, const char *path)
995 const struct got_error *err = NULL;
996 int nqueued = 0;
998 /*
999 * We keep all commits open throughout the lifetime of the log
1000 * view in order to avoid having to re-fetch commits from disk
1001 * while updating the display.
1003 while (nqueued < minqueue) {
1004 struct got_object_id *id;
1005 struct got_commit_object *commit;
1006 struct commit_queue_entry *entry;
1007 int errcode;
1009 err = got_commit_graph_iter_next(&id, graph);
1010 if (err) {
1011 if (err->code != GOT_ERR_ITER_NEED_MORE)
1012 break;
1013 err = got_commit_graph_fetch_commits(graph,
1014 minqueue, repo);
1015 if (err)
1016 return err;
1017 continue;
1020 if (id == NULL)
1021 break;
1023 err = got_object_open_as_commit(&commit, repo, id);
1024 if (err)
1025 break;
1026 entry = alloc_commit_queue_entry(commit, id);
1027 if (entry == NULL) {
1028 err = got_error_from_errno();
1029 break;
1032 errcode = pthread_mutex_lock(&tog_mutex);
1033 if (errcode) {
1034 err = got_error_set_errno(errcode);
1035 break;
1038 entry->idx = commits->ncommits;
1039 TAILQ_INSERT_TAIL(&commits->head, entry, entry);
1040 nqueued++;
1041 commits->ncommits++;
1043 errcode = pthread_mutex_unlock(&tog_mutex);
1044 if (errcode && err == NULL)
1045 err = got_error_set_errno(errcode);
1048 return err;
1051 static const struct got_error *
1052 get_head_commit_id(struct got_object_id **head_id, struct got_repository *repo)
1054 const struct got_error *err = NULL;
1055 struct got_reference *head_ref;
1057 *head_id = NULL;
1059 err = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1060 if (err)
1061 return err;
1063 err = got_ref_resolve(head_id, repo, head_ref);
1064 got_ref_close(head_ref);
1065 if (err) {
1066 *head_id = NULL;
1067 return err;
1070 return NULL;
1073 static const struct got_error *
1074 draw_commits(struct tog_view *view, struct commit_queue_entry **last,
1075 struct commit_queue_entry **selected, struct commit_queue_entry *first,
1076 struct commit_queue *commits, int selected_idx, int limit,
1077 const char *path, int commits_needed)
1079 const struct got_error *err = NULL;
1080 struct commit_queue_entry *entry;
1081 int ncommits, width;
1082 char *id_str = NULL, *header = NULL, *ncommits_str = NULL;
1083 wchar_t *wline;
1085 entry = first;
1086 ncommits = 0;
1087 while (entry) {
1088 if (ncommits == selected_idx) {
1089 *selected = entry;
1090 break;
1092 entry = TAILQ_NEXT(entry, entry);
1093 ncommits++;
1096 if (*selected) {
1097 err = got_object_id_str(&id_str, (*selected)->id);
1098 if (err)
1099 return err;
1102 if (asprintf(&ncommits_str, " [%d/%d]%s ",
1103 entry ? entry->idx + 1 : 0, commits->ncommits,
1104 commits_needed == 0 ? "" : " loading...") == -1)
1105 return got_error_from_errno();
1107 if (path && strcmp(path, "/") != 0) {
1108 if (asprintf(&header, "commit %s %s%s",
1109 id_str ? id_str : "........................................",
1110 path, ncommits_str) == -1) {
1111 err = got_error_from_errno();
1112 header = NULL;
1113 goto done;
1115 } else if (asprintf(&header, "commit %s%s",
1116 id_str ? id_str : "........................................",
1117 ncommits_str) == -1) {
1118 err = got_error_from_errno();
1119 header = NULL;
1120 goto done;
1122 err = format_line(&wline, &width, header, view->ncols);
1123 if (err)
1124 goto done;
1126 werase(view->window);
1128 if (view_needs_focus_indication(view))
1129 wstandout(view->window);
1130 waddwstr(view->window, wline);
1131 while (width < view->ncols) {
1132 waddch(view->window, ' ');
1133 width++;
1135 if (view_needs_focus_indication(view))
1136 wstandend(view->window);
1137 free(wline);
1138 if (limit <= 1)
1139 goto done;
1141 entry = first;
1142 *last = first;
1143 ncommits = 0;
1144 while (entry) {
1145 if (ncommits >= limit - 1)
1146 break;
1147 if (view->focussed && ncommits == selected_idx)
1148 wstandout(view->window);
1149 err = draw_commit(view, entry->commit, entry->id);
1150 if (view->focussed && ncommits == selected_idx)
1151 wstandend(view->window);
1152 if (err)
1153 break;
1154 ncommits++;
1155 *last = entry;
1156 entry = TAILQ_NEXT(entry, entry);
1159 view_vborder(view);
1160 done:
1161 free(id_str);
1162 free(ncommits_str);
1163 free(header);
1164 return err;
1167 static void
1168 scroll_up(struct commit_queue_entry **first_displayed_entry, int maxscroll,
1169 struct commit_queue *commits)
1171 struct commit_queue_entry *entry;
1172 int nscrolled = 0;
1174 entry = TAILQ_FIRST(&commits->head);
1175 if (*first_displayed_entry == entry)
1176 return;
1178 entry = *first_displayed_entry;
1179 while (entry && nscrolled < maxscroll) {
1180 entry = TAILQ_PREV(entry, commit_queue_head, entry);
1181 if (entry) {
1182 *first_displayed_entry = entry;
1183 nscrolled++;
1188 static const struct got_error *
1189 scroll_down(struct commit_queue_entry **first_displayed_entry, int maxscroll,
1190 struct commit_queue_entry **last_displayed_entry,
1191 struct commit_queue *commits, int *log_complete, int *commits_needed,
1192 pthread_cond_t *need_commits)
1194 const struct got_error *err = NULL;
1195 struct commit_queue_entry *pentry;
1196 int nscrolled = 0;
1198 if (*last_displayed_entry == NULL)
1199 return NULL;
1201 do {
1202 pentry = TAILQ_NEXT(*last_displayed_entry, entry);
1203 if (pentry == NULL) {
1204 int errcode;
1205 if (*log_complete)
1206 return NULL;
1207 *commits_needed = maxscroll + 20;
1208 errcode = pthread_cond_signal(need_commits);
1209 if (errcode)
1210 return got_error_set_errno(errcode);
1211 return NULL;
1213 *last_displayed_entry = pentry;
1215 pentry = TAILQ_NEXT(*first_displayed_entry, entry);
1216 if (pentry == NULL)
1217 break;
1218 *first_displayed_entry = pentry;
1219 } while (++nscrolled < maxscroll);
1221 return err;
1224 static const struct got_error *
1225 open_diff_view_for_commit(struct tog_view **new_view, int begin_x,
1226 struct got_object_id *commit_id, struct got_commit_object *commit,
1227 struct got_repository *repo)
1229 const struct got_error *err;
1230 struct got_object_qid *parent_id;
1231 struct tog_view *diff_view;
1233 diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
1234 if (diff_view == NULL)
1235 return got_error_from_errno();
1237 parent_id = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
1238 err = open_diff_view(diff_view, parent_id ? parent_id->id : NULL,
1239 commit_id, repo);
1240 if (err == NULL)
1241 *new_view = diff_view;
1242 return err;
1245 static const struct got_error *
1246 browse_commit(struct tog_view **new_view, int begin_x,
1247 struct commit_queue_entry *entry, struct got_repository *repo)
1249 const struct got_error *err = NULL;
1250 struct got_tree_object *tree;
1251 struct tog_view *tree_view;
1253 err = got_object_open_as_tree(&tree, repo,
1254 got_object_commit_get_tree_id(entry->commit));
1255 if (err)
1256 return err;
1258 tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
1259 if (tree_view == NULL)
1260 return got_error_from_errno();
1262 err = open_tree_view(tree_view, tree, entry->id, repo);
1263 if (err)
1264 got_object_tree_close(tree);
1265 else
1266 *new_view = tree_view;
1267 return err;
1270 static void *
1271 log_thread(void *arg)
1273 const struct got_error *err = NULL;
1274 int errcode = 0;
1275 struct tog_log_thread_args *a = arg;
1276 int done = 0;
1278 err = got_commit_graph_iter_start(a->graph, a->start_id, a->repo);
1279 if (err)
1280 return (void *)err;
1282 while (!done && !err) {
1283 err = queue_commits(a->graph, a->commits, 1, a->repo,
1284 a->in_repo_path);
1285 if (err) {
1286 if (err->code != GOT_ERR_ITER_COMPLETED)
1287 return (void *)err;
1288 err = NULL;
1289 done = 1;
1290 } else if (a->commits_needed > 0)
1291 a->commits_needed--;
1293 errcode = pthread_mutex_lock(&tog_mutex);
1294 if (errcode)
1295 return (void *)got_error_set_errno(errcode);
1297 if (done)
1298 a->log_complete = 1;
1299 else if (*a->quit) {
1300 done = 1;
1301 a->log_complete = 1;
1302 } else if (*a->first_displayed_entry == NULL) {
1303 *a->first_displayed_entry =
1304 TAILQ_FIRST(&a->commits->head);
1305 *a->selected_entry = *a->first_displayed_entry;
1308 if (done)
1309 a->commits_needed = 0;
1310 else if (a->commits_needed == 0) {
1311 errcode = pthread_cond_wait(&a->need_commits,
1312 &tog_mutex);
1313 if (errcode)
1314 err = got_error_set_errno(errcode);
1317 errcode = pthread_mutex_unlock(&tog_mutex);
1318 if (errcode && err == NULL)
1319 err = got_error_set_errno(errcode);
1321 return (void *)err;
1324 static const struct got_error *
1325 stop_log_thread(struct tog_log_view_state *s)
1327 const struct got_error *err = NULL;
1328 int errcode;
1330 if (s->thread) {
1331 s->quit = 1;
1332 errcode = pthread_cond_signal(&s->thread_args.need_commits);
1333 if (errcode)
1334 return got_error_set_errno(errcode);
1335 errcode = pthread_mutex_unlock(&tog_mutex);
1336 if (errcode)
1337 return got_error_set_errno(errcode);
1338 errcode = pthread_join(s->thread, (void **)&err);
1339 if (errcode)
1340 return got_error_set_errno(errcode);
1341 errcode = pthread_mutex_lock(&tog_mutex);
1342 if (errcode)
1343 return got_error_set_errno(errcode);
1344 s->thread = NULL;
1347 errcode = pthread_cond_destroy(&s->thread_args.need_commits);
1348 if (errcode && err == NULL)
1349 err = got_error_set_errno(errcode);
1351 if (s->thread_args.repo) {
1352 got_repo_close(s->thread_args.repo);
1353 s->thread_args.repo = NULL;
1356 if (s->thread_args.graph) {
1357 got_commit_graph_close(s->thread_args.graph);
1358 s->thread_args.graph = NULL;
1361 return err;
1364 static const struct got_error *
1365 close_log_view(struct tog_view *view)
1367 const struct got_error *err = NULL;
1368 struct tog_log_view_state *s = &view->state.log;
1370 err = stop_log_thread(s);
1371 free_commits(&s->commits);
1372 free(s->in_repo_path);
1373 s->in_repo_path = NULL;
1374 free(s->start_id);
1375 s->start_id = NULL;
1376 return err;
1379 static const struct got_error *
1380 open_log_view(struct tog_view *view, struct got_object_id *start_id,
1381 struct got_repository *repo, const char *path, int check_disk)
1383 const struct got_error *err = NULL;
1384 struct tog_log_view_state *s = &view->state.log;
1385 struct got_repository *thread_repo = NULL;
1386 struct got_commit_graph *thread_graph = NULL;
1387 int errcode;
1389 err = got_repo_map_path(&s->in_repo_path, repo, path, check_disk);
1390 if (err != NULL)
1391 goto done;
1393 /* The commit queue only contains commits being displayed. */
1394 TAILQ_INIT(&s->commits.head);
1395 s->commits.ncommits = 0;
1397 s->repo = repo;
1398 s->start_id = got_object_id_dup(start_id);
1399 if (s->start_id == NULL) {
1400 err = got_error_from_errno();
1401 goto done;
1404 view->show = show_log_view;
1405 view->input = input_log_view;
1406 view->close = close_log_view;
1408 err = got_repo_open(&thread_repo, got_repo_get_path(repo));
1409 if (err)
1410 goto done;
1411 err = got_commit_graph_open(&thread_graph, start_id, s->in_repo_path,
1412 0, thread_repo);
1413 if (err)
1414 goto done;
1416 errcode = pthread_cond_init(&s->thread_args.need_commits, NULL);
1417 if (errcode) {
1418 err = got_error_set_errno(errcode);
1419 goto done;
1422 s->thread_args.commits_needed = view->nlines;
1423 s->thread_args.graph = thread_graph;
1424 s->thread_args.commits = &s->commits;
1425 s->thread_args.in_repo_path = s->in_repo_path;
1426 s->thread_args.start_id = s->start_id;
1427 s->thread_args.repo = thread_repo;
1428 s->thread_args.log_complete = 0;
1429 s->thread_args.quit = &s->quit;
1430 s->thread_args.view = view;
1431 s->thread_args.first_displayed_entry = &s->first_displayed_entry;
1432 s->thread_args.selected_entry = &s->selected_entry;
1433 done:
1434 if (err)
1435 close_log_view(view);
1436 return err;
1439 static const struct got_error *
1440 show_log_view(struct tog_view *view)
1442 struct tog_log_view_state *s = &view->state.log;
1444 if (s->thread == NULL) {
1445 int errcode = pthread_create(&s->thread, NULL, log_thread,
1446 &s->thread_args);
1447 if (errcode)
1448 return got_error_set_errno(errcode);
1451 return draw_commits(view, &s->last_displayed_entry,
1452 &s->selected_entry, s->first_displayed_entry,
1453 &s->commits, s->selected, view->nlines,
1454 s->in_repo_path, s->thread_args.commits_needed);
1457 static const struct got_error *
1458 input_log_view(struct tog_view **new_view, struct tog_view **dead_view,
1459 struct tog_view **focus_view, struct tog_view *view, int ch)
1461 const struct got_error *err = NULL;
1462 struct tog_log_view_state *s = &view->state.log;
1463 char *parent_path;
1464 struct tog_view *diff_view = NULL, *tree_view = NULL;
1465 int begin_x = 0;
1467 switch (ch) {
1468 case 'q':
1469 s->quit = 1;
1470 break;
1471 case 'k':
1472 case KEY_UP:
1473 if (s->first_displayed_entry == NULL)
1474 break;
1475 if (s->selected > 0)
1476 s->selected--;
1477 if (s->selected > 0)
1478 break;
1479 scroll_up(&s->first_displayed_entry, 1,
1480 &s->commits);
1481 break;
1482 case KEY_PPAGE:
1483 if (s->first_displayed_entry == NULL)
1484 break;
1485 if (TAILQ_FIRST(&s->commits.head) ==
1486 s->first_displayed_entry) {
1487 s->selected = 0;
1488 break;
1490 scroll_up(&s->first_displayed_entry,
1491 view->nlines, &s->commits);
1492 break;
1493 case 'j':
1494 case KEY_DOWN:
1495 if (s->first_displayed_entry == NULL)
1496 break;
1497 if (s->selected < MIN(view->nlines - 2,
1498 s->commits.ncommits - 1)) {
1499 s->selected++;
1500 break;
1502 err = scroll_down(&s->first_displayed_entry, 1,
1503 &s->last_displayed_entry, &s->commits,
1504 &s->thread_args.log_complete,
1505 &s->thread_args.commits_needed,
1506 &s->thread_args.need_commits);
1507 break;
1508 case KEY_NPAGE: {
1509 struct commit_queue_entry *first;
1510 first = s->first_displayed_entry;
1511 if (first == NULL)
1512 break;
1513 err = scroll_down(&s->first_displayed_entry,
1514 view->nlines, &s->last_displayed_entry,
1515 &s->commits, &s->thread_args.log_complete,
1516 &s->thread_args.commits_needed,
1517 &s->thread_args.need_commits);
1518 if (first == s->first_displayed_entry &&
1519 s->selected < MIN(view->nlines - 2,
1520 s->commits.ncommits - 1)) {
1521 /* can't scroll further down */
1522 s->selected = MIN(view->nlines - 2,
1523 s->commits.ncommits - 1);
1525 err = NULL;
1526 break;
1528 case KEY_RESIZE:
1529 if (s->selected > view->nlines - 2)
1530 s->selected = view->nlines - 2;
1531 if (s->selected > s->commits.ncommits - 1)
1532 s->selected = s->commits.ncommits - 1;
1533 break;
1534 case KEY_ENTER:
1535 case '\r':
1536 if (s->selected_entry == NULL)
1537 break;
1538 if (view_is_parent_view(view))
1539 begin_x = view_split_begin_x(view->begin_x);
1540 err = open_diff_view_for_commit(&diff_view, begin_x,
1541 s->selected_entry->id, s->selected_entry->commit,
1542 s->repo);
1543 if (err)
1544 break;
1545 if (view_is_parent_view(view)) {
1546 err = view_close_child(view);
1547 if (err)
1548 return err;
1549 err = view_set_child(view, diff_view);
1550 if (err) {
1551 view_close(diff_view);
1552 break;
1554 *focus_view = diff_view;
1555 view->child_focussed = 1;
1556 } else
1557 *new_view = diff_view;
1558 break;
1559 case 't':
1560 if (s->selected_entry == NULL)
1561 break;
1562 if (view_is_parent_view(view))
1563 begin_x = view_split_begin_x(view->begin_x);
1564 err = browse_commit(&tree_view, begin_x,
1565 s->selected_entry, s->repo);
1566 if (err)
1567 break;
1568 if (view_is_parent_view(view)) {
1569 err = view_close_child(view);
1570 if (err)
1571 return err;
1572 err = view_set_child(view, tree_view);
1573 if (err) {
1574 view_close(tree_view);
1575 break;
1577 *focus_view = tree_view;
1578 view->child_focussed = 1;
1579 } else
1580 *new_view = tree_view;
1581 break;
1582 case KEY_BACKSPACE:
1583 if (strcmp(s->in_repo_path, "/") == 0)
1584 break;
1585 parent_path = dirname(s->in_repo_path);
1586 if (parent_path && strcmp(parent_path, ".") != 0) {
1587 struct tog_view *lv;
1588 err = stop_log_thread(s);
1589 if (err)
1590 return err;
1591 lv = view_open(view->nlines, view->ncols,
1592 view->begin_y, view->begin_x, TOG_VIEW_LOG);
1593 if (lv == NULL)
1594 return got_error_from_errno();
1595 err = open_log_view(lv, s->start_id, s->repo,
1596 parent_path, 0);
1597 if (err)
1598 return err;;
1599 if (view_is_parent_view(view))
1600 *new_view = lv;
1601 else {
1602 view_set_child(view->parent, lv);
1603 *focus_view = lv;
1605 return NULL;
1607 break;
1608 default:
1609 break;
1612 return err;
1615 static const struct got_error *
1616 apply_unveil(const char *repo_path, const char *worktree_path)
1618 const struct got_error *error;
1620 if (repo_path && unveil(repo_path, "r") != 0)
1621 return got_error_from_errno();
1623 if (worktree_path && unveil(worktree_path, "rwc") != 0)
1624 return got_error_from_errno();
1626 if (unveil("/tmp", "rwc") != 0)
1627 return got_error_from_errno();
1629 error = got_privsep_unveil_exec_helpers();
1630 if (error != NULL)
1631 return error;
1633 if (unveil(NULL, NULL) != 0)
1634 return got_error_from_errno();
1636 return NULL;
1639 static const struct got_error *
1640 cmd_log(int argc, char *argv[])
1642 const struct got_error *error;
1643 struct got_repository *repo = NULL;
1644 struct got_object_id *start_id = NULL;
1645 char *path = NULL, *repo_path = NULL, *cwd = NULL;
1646 char *start_commit = NULL;
1647 int ch;
1648 struct tog_view *view;
1650 #ifndef PROFILE
1651 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
1652 NULL) == -1)
1653 err(1, "pledge");
1654 #endif
1656 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
1657 switch (ch) {
1658 case 'c':
1659 start_commit = optarg;
1660 break;
1661 case 'r':
1662 repo_path = realpath(optarg, NULL);
1663 if (repo_path == NULL)
1664 err(1, "-r option");
1665 break;
1666 default:
1667 usage();
1668 /* NOTREACHED */
1672 argc -= optind;
1673 argv += optind;
1675 if (argc == 0)
1676 path = strdup("");
1677 else if (argc == 1)
1678 path = strdup(argv[0]);
1679 else
1680 usage_log();
1681 if (path == NULL)
1682 return got_error_from_errno();
1684 cwd = getcwd(NULL, 0);
1685 if (cwd == NULL) {
1686 error = got_error_from_errno();
1687 goto done;
1689 if (repo_path == NULL) {
1690 struct got_worktree *worktree;
1691 error = got_worktree_open(&worktree, cwd);
1692 if (error && error->code != GOT_ERR_NOT_WORKTREE)
1693 goto done;
1694 if (worktree) {
1695 repo_path =
1696 strdup(got_worktree_get_repo_path(worktree));
1697 got_worktree_close(worktree);
1698 } else
1699 repo_path = strdup(cwd);
1700 if (repo_path == NULL) {
1701 error = got_error_from_errno();
1702 goto done;
1706 error = apply_unveil(repo_path, NULL);
1707 if (error)
1708 goto done;
1710 error = got_repo_open(&repo, repo_path);
1711 if (error != NULL)
1712 goto done;
1714 if (start_commit == NULL)
1715 error = get_head_commit_id(&start_id, repo);
1716 else
1717 error = got_object_resolve_id_str(&start_id, repo,
1718 start_commit);
1719 if (error != NULL)
1720 goto done;
1722 view = view_open(0, 0, 0, 0, TOG_VIEW_LOG);
1723 if (view == NULL) {
1724 error = got_error_from_errno();
1725 goto done;
1727 error = open_log_view(view, start_id, repo, path, 1);
1728 if (error)
1729 goto done;
1730 error = view_loop(view);
1731 done:
1732 free(repo_path);
1733 free(cwd);
1734 free(path);
1735 free(start_id);
1736 if (repo)
1737 got_repo_close(repo);
1738 return error;
1741 __dead static void
1742 usage_diff(void)
1744 endwin();
1745 fprintf(stderr, "usage: %s diff [repository-path] object1 object2\n",
1746 getprogname());
1747 exit(1);
1750 static char *
1751 parse_next_line(FILE *f, size_t *len)
1753 char *line;
1754 size_t linelen;
1755 size_t lineno;
1756 const char delim[3] = { '\0', '\0', '\0'};
1758 line = fparseln(f, &linelen, &lineno, delim, 0);
1759 if (len)
1760 *len = linelen;
1761 return line;
1764 static const struct got_error *
1765 draw_file(struct tog_view *view, FILE *f, int *first_displayed_line,
1766 int *last_displayed_line, int *eof, int max_lines,
1767 char * header)
1769 const struct got_error *err;
1770 int nlines = 0, nprinted = 0;
1771 char *line;
1772 size_t len;
1773 wchar_t *wline;
1774 int width;
1776 rewind(f);
1777 werase(view->window);
1779 if (header) {
1780 err = format_line(&wline, &width, header, view->ncols);
1781 if (err) {
1782 return err;
1785 if (view_needs_focus_indication(view))
1786 wstandout(view->window);
1787 waddwstr(view->window, wline);
1788 if (view_needs_focus_indication(view))
1789 wstandend(view->window);
1790 if (width < view->ncols)
1791 waddch(view->window, '\n');
1793 if (max_lines <= 1)
1794 return NULL;
1795 max_lines--;
1798 *eof = 0;
1799 while (nprinted < max_lines) {
1800 line = parse_next_line(f, &len);
1801 if (line == NULL) {
1802 *eof = 1;
1803 break;
1805 if (++nlines < *first_displayed_line) {
1806 free(line);
1807 continue;
1810 err = format_line(&wline, &width, line, view->ncols);
1811 if (err) {
1812 free(line);
1813 return err;
1815 waddwstr(view->window, wline);
1816 if (width < view->ncols)
1817 waddch(view->window, '\n');
1818 if (++nprinted == 1)
1819 *first_displayed_line = nlines;
1820 free(line);
1821 free(wline);
1822 wline = NULL;
1824 *last_displayed_line = nlines;
1826 view_vborder(view);
1828 return NULL;
1831 static char *
1832 get_datestr(time_t *time, char *datebuf)
1834 char *p, *s = ctime_r(time, datebuf);
1835 p = strchr(s, '\n');
1836 if (p)
1837 *p = '\0';
1838 return s;
1841 static const struct got_error *
1842 write_commit_info(struct got_object_id *commit_id, struct got_repository *repo,
1843 FILE *outfile)
1845 const struct got_error *err = NULL;
1846 char datebuf[26];
1847 struct got_commit_object *commit;
1848 char *id_str = NULL;
1849 time_t committer_time;
1850 const char *author, *committer;
1852 err = got_object_open_as_commit(&commit, repo, commit_id);
1853 if (err)
1854 return err;
1856 err = got_object_id_str(&id_str, commit_id);
1857 if (err) {
1858 err = got_error_from_errno();
1859 goto done;
1862 if (fprintf(outfile, "commit %s\n", id_str) < 0) {
1863 err = got_error_from_errno();
1864 goto done;
1866 if (fprintf(outfile, "from: %s\n",
1867 got_object_commit_get_author(commit)) < 0) {
1868 err = got_error_from_errno();
1869 goto done;
1871 committer_time = got_object_commit_get_committer_time(commit);
1872 if (fprintf(outfile, "date: %s UTC\n",
1873 get_datestr(&committer_time, datebuf)) < 0) {
1874 err = got_error_from_errno();
1875 goto done;
1877 author = got_object_commit_get_author(commit);
1878 committer = got_object_commit_get_committer(commit);
1879 if (strcmp(author, committer) != 0 &&
1880 fprintf(outfile, "via: %s\n", committer) < 0) {
1881 err = got_error_from_errno();
1882 goto done;
1884 if (fprintf(outfile, "%s\n",
1885 got_object_commit_get_logmsg(commit)) < 0) {
1886 err = got_error_from_errno();
1887 goto done;
1889 done:
1890 free(id_str);
1891 got_object_commit_close(commit);
1892 return err;
1895 static const struct got_error *
1896 create_diff(struct tog_diff_view_state *s)
1898 const struct got_error *err = NULL;
1899 FILE *f = NULL;
1900 int obj_type;
1902 f = got_opentemp();
1903 if (f == NULL) {
1904 err = got_error_from_errno();
1905 goto done;
1907 if (s->f)
1908 fclose(s->f);
1909 s->f = f;
1911 if (s->id1)
1912 err = got_object_get_type(&obj_type, s->repo, s->id1);
1913 else
1914 err = got_object_get_type(&obj_type, s->repo, s->id2);
1915 if (err)
1916 goto done;
1918 switch (obj_type) {
1919 case GOT_OBJ_TYPE_BLOB:
1920 err = got_diff_objects_as_blobs(s->id1, s->id2, NULL, NULL,
1921 s->diff_context, s->repo, f);
1922 break;
1923 case GOT_OBJ_TYPE_TREE:
1924 err = got_diff_objects_as_trees(s->id1, s->id2, "", "",
1925 s->diff_context, s->repo, f);
1926 break;
1927 case GOT_OBJ_TYPE_COMMIT: {
1928 const struct got_object_id_queue *parent_ids;
1929 struct got_object_qid *pid;
1930 struct got_commit_object *commit2;
1932 err = got_object_open_as_commit(&commit2, s->repo, s->id2);
1933 if (err)
1934 break;
1935 /* Show commit info if we're diffing to a parent commit. */
1936 parent_ids = got_object_commit_get_parent_ids(commit2);
1937 SIMPLEQ_FOREACH(pid, parent_ids, entry) {
1938 if (got_object_id_cmp(s->id1, pid->id) == 0) {
1939 write_commit_info(s->id2, s->repo, f);
1940 break;
1943 got_object_commit_close(commit2);
1945 err = got_diff_objects_as_commits(s->id1, s->id2,
1946 s->diff_context, s->repo, f);
1947 break;
1949 default:
1950 err = got_error(GOT_ERR_OBJ_TYPE);
1951 break;
1953 done:
1954 if (f)
1955 fflush(f);
1956 return err;
1959 static const struct got_error *
1960 open_diff_view(struct tog_view *view, struct got_object_id *id1,
1961 struct got_object_id *id2, struct got_repository *repo)
1963 const struct got_error *err;
1965 if (id1 != NULL && id2 != NULL) {
1966 int type1, type2;
1967 err = got_object_get_type(&type1, repo, id1);
1968 if (err)
1969 return err;
1970 err = got_object_get_type(&type2, repo, id2);
1971 if (err)
1972 return err;
1974 if (type1 != type2)
1975 return got_error(GOT_ERR_OBJ_TYPE);
1978 if (id1) {
1979 view->state.diff.id1 = got_object_id_dup(id1);
1980 if (view->state.diff.id1 == NULL)
1981 return got_error_from_errno();
1982 } else
1983 view->state.diff.id1 = NULL;
1985 view->state.diff.id2 = got_object_id_dup(id2);
1986 if (view->state.diff.id2 == NULL) {
1987 free(view->state.diff.id1);
1988 view->state.diff.id1 = NULL;
1989 return got_error_from_errno();
1991 view->state.diff.f = NULL;
1992 view->state.diff.first_displayed_line = 1;
1993 view->state.diff.last_displayed_line = view->nlines;
1994 view->state.diff.diff_context = 3;
1995 view->state.diff.repo = repo;
1997 err = create_diff(&view->state.diff);
1998 if (err) {
1999 free(view->state.diff.id1);
2000 view->state.diff.id1 = NULL;
2001 free(view->state.diff.id2);
2002 view->state.diff.id2 = NULL;
2003 return err;
2006 view->show = show_diff_view;
2007 view->input = input_diff_view;
2008 view->close = close_diff_view;
2010 return NULL;
2013 static const struct got_error *
2014 close_diff_view(struct tog_view *view)
2016 const struct got_error *err = NULL;
2018 free(view->state.diff.id1);
2019 view->state.diff.id1 = NULL;
2020 free(view->state.diff.id2);
2021 view->state.diff.id2 = NULL;
2022 if (view->state.diff.f && fclose(view->state.diff.f) == EOF)
2023 err = got_error_from_errno();
2024 return err;
2027 static const struct got_error *
2028 show_diff_view(struct tog_view *view)
2030 const struct got_error *err;
2031 struct tog_diff_view_state *s = &view->state.diff;
2032 char *id_str1 = NULL, *id_str2, *header;
2034 if (s->id1) {
2035 err = got_object_id_str(&id_str1, s->id1);
2036 if (err)
2037 return err;
2039 err = got_object_id_str(&id_str2, s->id2);
2040 if (err)
2041 return err;
2043 if (asprintf(&header, "diff %s %s",
2044 id_str1 ? id_str1 : "/dev/null", id_str2) == -1) {
2045 err = got_error_from_errno();
2046 free(id_str1);
2047 free(id_str2);
2048 return err;
2050 free(id_str1);
2051 free(id_str2);
2053 return draw_file(view, s->f, &s->first_displayed_line,
2054 &s->last_displayed_line, &s->eof, view->nlines,
2055 header);
2058 static const struct got_error *
2059 input_diff_view(struct tog_view **new_view, struct tog_view **dead_view,
2060 struct tog_view **focus_view, struct tog_view *view, int ch)
2062 const struct got_error *err = NULL;
2063 struct tog_diff_view_state *s = &view->state.diff;
2064 int i;
2066 switch (ch) {
2067 case 'k':
2068 case KEY_UP:
2069 if (s->first_displayed_line > 1)
2070 s->first_displayed_line--;
2071 break;
2072 case KEY_PPAGE:
2073 i = 0;
2074 while (i++ < view->nlines - 1 &&
2075 s->first_displayed_line > 1)
2076 s->first_displayed_line--;
2077 break;
2078 case 'j':
2079 case KEY_DOWN:
2080 if (!s->eof)
2081 s->first_displayed_line++;
2082 break;
2083 case KEY_NPAGE:
2084 case ' ':
2085 i = 0;
2086 while (!s->eof && i++ < view->nlines - 1) {
2087 char *line;
2088 line = parse_next_line(s->f, NULL);
2089 s->first_displayed_line++;
2090 if (line == NULL)
2091 break;
2093 break;
2094 case '[':
2095 if (s->diff_context > 0) {
2096 s->diff_context--;
2097 err = create_diff(s);
2099 break;
2100 case ']':
2101 if (s->diff_context < GOT_DIFF_MAX_CONTEXT) {
2102 s->diff_context++;
2103 err = create_diff(s);
2105 break;
2106 default:
2107 break;
2110 return err;
2113 static const struct got_error *
2114 cmd_diff(int argc, char *argv[])
2116 const struct got_error *error = NULL;
2117 struct got_repository *repo = NULL;
2118 struct got_object_id *id1 = NULL, *id2 = NULL;
2119 char *repo_path = NULL;
2120 char *id_str1 = NULL, *id_str2 = NULL;
2121 int ch;
2122 struct tog_view *view;
2124 #ifndef PROFILE
2125 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
2126 NULL) == -1)
2127 err(1, "pledge");
2128 #endif
2130 while ((ch = getopt(argc, argv, "")) != -1) {
2131 switch (ch) {
2132 default:
2133 usage();
2134 /* NOTREACHED */
2138 argc -= optind;
2139 argv += optind;
2141 if (argc == 0) {
2142 usage_diff(); /* TODO show local worktree changes */
2143 } else if (argc == 2) {
2144 repo_path = getcwd(NULL, 0);
2145 if (repo_path == NULL)
2146 return got_error_from_errno();
2147 id_str1 = argv[0];
2148 id_str2 = argv[1];
2149 } else if (argc == 3) {
2150 repo_path = realpath(argv[0], NULL);
2151 if (repo_path == NULL)
2152 return got_error_from_errno();
2153 id_str1 = argv[1];
2154 id_str2 = argv[2];
2155 } else
2156 usage_diff();
2158 error = apply_unveil(repo_path, NULL);
2159 if (error)
2160 goto done;
2162 error = got_repo_open(&repo, repo_path);
2163 free(repo_path);
2164 if (error)
2165 goto done;
2167 error = got_object_resolve_id_str(&id1, repo, id_str1);
2168 if (error)
2169 goto done;
2171 error = got_object_resolve_id_str(&id2, repo, id_str2);
2172 if (error)
2173 goto done;
2175 view = view_open(0, 0, 0, 0, TOG_VIEW_DIFF);
2176 if (view == NULL) {
2177 error = got_error_from_errno();
2178 goto done;
2180 error = open_diff_view(view, id1, id2, repo);
2181 if (error)
2182 goto done;
2183 error = view_loop(view);
2184 done:
2185 got_repo_close(repo);
2186 return error;
2189 __dead static void
2190 usage_blame(void)
2192 endwin();
2193 fprintf(stderr, "usage: %s blame [-c commit] [-r repository-path] path\n",
2194 getprogname());
2195 exit(1);
2198 struct tog_blame_line {
2199 int annotated;
2200 struct got_object_id *id;
2203 static const struct got_error *
2204 draw_blame(struct tog_view *view, struct got_object_id *id, FILE *f,
2205 const char *path, struct tog_blame_line *lines, int nlines,
2206 int blame_complete, int selected_line, int *first_displayed_line,
2207 int *last_displayed_line, int *eof, int max_lines)
2209 const struct got_error *err;
2210 int lineno = 0, nprinted = 0;
2211 char *line;
2212 size_t len;
2213 wchar_t *wline;
2214 int width, wlimit;
2215 struct tog_blame_line *blame_line;
2216 struct got_object_id *prev_id = NULL;
2217 char *id_str;
2219 err = got_object_id_str(&id_str, id);
2220 if (err)
2221 return err;
2223 rewind(f);
2224 werase(view->window);
2226 if (asprintf(&line, "commit %s", id_str) == -1) {
2227 err = got_error_from_errno();
2228 free(id_str);
2229 return err;
2232 err = format_line(&wline, &width, line, view->ncols);
2233 free(line);
2234 line = NULL;
2235 if (view_needs_focus_indication(view))
2236 wstandout(view->window);
2237 waddwstr(view->window, wline);
2238 if (view_needs_focus_indication(view))
2239 wstandend(view->window);
2240 free(wline);
2241 wline = NULL;
2242 if (width < view->ncols)
2243 waddch(view->window, '\n');
2245 if (asprintf(&line, "[%d/%d] %s%s",
2246 *first_displayed_line - 1 + selected_line, nlines,
2247 blame_complete ? "" : "annotating ", path) == -1) {
2248 free(id_str);
2249 return got_error_from_errno();
2251 free(id_str);
2252 err = format_line(&wline, &width, line, view->ncols);
2253 free(line);
2254 line = NULL;
2255 if (err)
2256 return err;
2257 waddwstr(view->window, wline);
2258 free(wline);
2259 wline = NULL;
2260 if (width < view->ncols)
2261 waddch(view->window, '\n');
2263 *eof = 0;
2264 while (nprinted < max_lines - 2) {
2265 line = parse_next_line(f, &len);
2266 if (line == NULL) {
2267 *eof = 1;
2268 break;
2270 if (++lineno < *first_displayed_line) {
2271 free(line);
2272 continue;
2275 wlimit = view->ncols < 9 ? 0 : view->ncols - 9;
2276 err = format_line(&wline, &width, line, wlimit);
2277 if (err) {
2278 free(line);
2279 return err;
2282 if (view->focussed && nprinted == selected_line - 1)
2283 wstandout(view->window);
2285 blame_line = &lines[lineno - 1];
2286 if (blame_line->annotated && prev_id &&
2287 got_object_id_cmp(prev_id, blame_line->id) == 0)
2288 waddstr(view->window, " ");
2289 else if (blame_line->annotated) {
2290 char *id_str;
2291 err = got_object_id_str(&id_str, blame_line->id);
2292 if (err) {
2293 free(line);
2294 free(wline);
2295 return err;
2297 wprintw(view->window, "%.8s ", id_str);
2298 free(id_str);
2299 prev_id = blame_line->id;
2300 } else {
2301 waddstr(view->window, "........ ");
2302 prev_id = NULL;
2305 waddwstr(view->window, wline);
2306 while (width < wlimit) {
2307 waddch(view->window, ' ');
2308 width++;
2310 if (view->focussed && nprinted == selected_line - 1)
2311 wstandend(view->window);
2312 if (++nprinted == 1)
2313 *first_displayed_line = lineno;
2314 free(line);
2315 free(wline);
2316 wline = NULL;
2318 *last_displayed_line = lineno;
2320 view_vborder(view);
2322 return NULL;
2325 static const struct got_error *
2326 blame_cb(void *arg, int nlines, int lineno, struct got_object_id *id)
2328 const struct got_error *err = NULL;
2329 struct tog_blame_cb_args *a = arg;
2330 struct tog_blame_line *line;
2331 int errcode;
2333 if (nlines != a->nlines ||
2334 (lineno != -1 && lineno < 1) || lineno > a->nlines)
2335 return got_error(GOT_ERR_RANGE);
2337 errcode = pthread_mutex_lock(&tog_mutex);
2338 if (errcode)
2339 return got_error_set_errno(errcode);
2341 if (*a->quit) { /* user has quit the blame view */
2342 err = got_error(GOT_ERR_ITER_COMPLETED);
2343 goto done;
2346 if (lineno == -1)
2347 goto done; /* no change in this commit */
2349 line = &a->lines[lineno - 1];
2350 if (line->annotated)
2351 goto done;
2353 line->id = got_object_id_dup(id);
2354 if (line->id == NULL) {
2355 err = got_error_from_errno();
2356 goto done;
2358 line->annotated = 1;
2359 done:
2360 errcode = pthread_mutex_unlock(&tog_mutex);
2361 if (errcode)
2362 err = got_error_set_errno(errcode);
2363 return err;
2366 static void *
2367 blame_thread(void *arg)
2369 const struct got_error *err;
2370 struct tog_blame_thread_args *ta = arg;
2371 struct tog_blame_cb_args *a = ta->cb_args;
2372 int errcode;
2374 err = got_blame_incremental(ta->path, a->commit_id, ta->repo,
2375 blame_cb, ta->cb_args);
2377 errcode = pthread_mutex_lock(&tog_mutex);
2378 if (errcode)
2379 return (void *)got_error_set_errno(errcode);
2381 got_repo_close(ta->repo);
2382 ta->repo = NULL;
2383 *ta->complete = 1;
2385 errcode = pthread_mutex_unlock(&tog_mutex);
2386 if (errcode && err == NULL)
2387 err = got_error_set_errno(errcode);
2389 return (void *)err;
2392 static struct got_object_id *
2393 get_selected_commit_id(struct tog_blame_line *lines, int first_displayed_line,
2394 int selected_line)
2396 struct tog_blame_line *line;
2398 line = &lines[first_displayed_line - 1 + selected_line - 1];
2399 if (!line->annotated)
2400 return NULL;
2402 return line->id;
2405 static const struct got_error *
2406 stop_blame(struct tog_blame *blame)
2408 const struct got_error *err = NULL;
2409 int i;
2411 if (blame->thread) {
2412 int errcode;
2413 errcode = pthread_mutex_unlock(&tog_mutex);
2414 if (errcode)
2415 return got_error_set_errno(errcode);
2416 errcode = pthread_join(blame->thread, (void **)&err);
2417 if (errcode)
2418 return got_error_set_errno(errcode);
2419 errcode = pthread_mutex_lock(&tog_mutex);
2420 if (errcode)
2421 return got_error_set_errno(errcode);
2422 if (err && err->code == GOT_ERR_ITER_COMPLETED)
2423 err = NULL;
2424 blame->thread = NULL;
2426 if (blame->thread_args.repo) {
2427 got_repo_close(blame->thread_args.repo);
2428 blame->thread_args.repo = NULL;
2430 if (blame->f) {
2431 fclose(blame->f);
2432 blame->f = NULL;
2434 if (blame->lines) {
2435 for (i = 0; i < blame->nlines; i++)
2436 free(blame->lines[i].id);
2437 free(blame->lines);
2438 blame->lines = NULL;
2440 free(blame->cb_args.commit_id);
2441 blame->cb_args.commit_id = NULL;
2443 return err;
2446 static const struct got_error *
2447 run_blame(struct tog_blame *blame, struct tog_view *view, int *blame_complete,
2448 int *first_displayed_line, int *last_displayed_line, int *selected_line,
2449 int *done, int *eof, const char *path, struct got_object_id *commit_id,
2450 struct got_repository *repo)
2452 const struct got_error *err = NULL;
2453 struct got_blob_object *blob = NULL;
2454 struct got_repository *thread_repo = NULL;
2455 struct got_object_id *obj_id = NULL;
2456 int obj_type;
2458 err = got_object_id_by_path(&obj_id, repo, commit_id, path);
2459 if (err)
2460 return err;
2461 if (obj_id == NULL)
2462 return got_error(GOT_ERR_NO_OBJ);
2464 err = got_object_get_type(&obj_type, repo, obj_id);
2465 if (err)
2466 goto done;
2468 if (obj_type != GOT_OBJ_TYPE_BLOB) {
2469 err = got_error(GOT_ERR_OBJ_TYPE);
2470 goto done;
2473 err = got_object_open_as_blob(&blob, repo, obj_id, 8192);
2474 if (err)
2475 goto done;
2476 blame->f = got_opentemp();
2477 if (blame->f == NULL) {
2478 err = got_error_from_errno();
2479 goto done;
2481 err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
2482 blame->f, blob);
2483 if (err)
2484 goto done;
2486 blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
2487 if (blame->lines == NULL) {
2488 err = got_error_from_errno();
2489 goto done;
2492 err = got_repo_open(&thread_repo, got_repo_get_path(repo));
2493 if (err)
2494 goto done;
2496 blame->cb_args.view = view;
2497 blame->cb_args.lines = blame->lines;
2498 blame->cb_args.nlines = blame->nlines;
2499 blame->cb_args.commit_id = got_object_id_dup(commit_id);
2500 if (blame->cb_args.commit_id == NULL) {
2501 err = got_error_from_errno();
2502 goto done;
2504 blame->cb_args.quit = done;
2506 blame->thread_args.path = path;
2507 blame->thread_args.repo = thread_repo;
2508 blame->thread_args.cb_args = &blame->cb_args;
2509 blame->thread_args.complete = blame_complete;
2510 *blame_complete = 0;
2512 done:
2513 if (blob)
2514 got_object_blob_close(blob);
2515 free(obj_id);
2516 if (err)
2517 stop_blame(blame);
2518 return err;
2521 static const struct got_error *
2522 open_blame_view(struct tog_view *view, char *path,
2523 struct got_object_id *commit_id, struct got_repository *repo)
2525 const struct got_error *err = NULL;
2526 struct tog_blame_view_state *s = &view->state.blame;
2528 SIMPLEQ_INIT(&s->blamed_commits);
2530 err = got_object_qid_alloc(&s->blamed_commit, commit_id);
2531 if (err)
2532 return err;
2534 SIMPLEQ_INSERT_HEAD(&s->blamed_commits, s->blamed_commit, entry);
2535 s->first_displayed_line = 1;
2536 s->last_displayed_line = view->nlines;
2537 s->selected_line = 1;
2538 s->blame_complete = 0;
2539 s->path = path;
2540 if (s->path == NULL)
2541 return got_error_from_errno();
2542 s->repo = repo;
2543 s->commit_id = commit_id;
2544 memset(&s->blame, 0, sizeof(s->blame));
2546 view->show = show_blame_view;
2547 view->input = input_blame_view;
2548 view->close = close_blame_view;
2550 return run_blame(&s->blame, view, &s->blame_complete,
2551 &s->first_displayed_line, &s->last_displayed_line,
2552 &s->selected_line, &s->done, &s->eof, s->path,
2553 s->blamed_commit->id, s->repo);
2556 static const struct got_error *
2557 close_blame_view(struct tog_view *view)
2559 const struct got_error *err = NULL;
2560 struct tog_blame_view_state *s = &view->state.blame;
2562 if (s->blame.thread)
2563 err = stop_blame(&s->blame);
2565 while (!SIMPLEQ_EMPTY(&s->blamed_commits)) {
2566 struct got_object_qid *blamed_commit;
2567 blamed_commit = SIMPLEQ_FIRST(&s->blamed_commits);
2568 SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
2569 got_object_qid_free(blamed_commit);
2572 free(s->path);
2574 return err;
2577 static const struct got_error *
2578 show_blame_view(struct tog_view *view)
2580 const struct got_error *err = NULL;
2581 struct tog_blame_view_state *s = &view->state.blame;
2582 int errcode;
2584 if (s->blame.thread == NULL) {
2585 errcode = pthread_create(&s->blame.thread, NULL, blame_thread,
2586 &s->blame.thread_args);
2587 if (errcode)
2588 return got_error_set_errno(errcode);
2591 err = draw_blame(view, s->blamed_commit->id, s->blame.f,
2592 s->path, s->blame.lines, s->blame.nlines, s->blame_complete,
2593 s->selected_line, &s->first_displayed_line,
2594 &s->last_displayed_line, &s->eof, view->nlines);
2596 view_vborder(view);
2597 return err;
2600 static const struct got_error *
2601 input_blame_view(struct tog_view **new_view, struct tog_view **dead_view,
2602 struct tog_view **focus_view, struct tog_view *view, int ch)
2604 const struct got_error *err = NULL, *thread_err = NULL;
2605 struct tog_view *diff_view;
2606 struct tog_blame_view_state *s = &view->state.blame;
2607 int begin_x = 0;
2609 switch (ch) {
2610 case 'q':
2611 s->done = 1;
2612 break;
2613 case 'k':
2614 case KEY_UP:
2615 if (s->selected_line > 1)
2616 s->selected_line--;
2617 else if (s->selected_line == 1 &&
2618 s->first_displayed_line > 1)
2619 s->first_displayed_line--;
2620 break;
2621 case KEY_PPAGE:
2622 if (s->first_displayed_line == 1) {
2623 s->selected_line = 1;
2624 break;
2626 if (s->first_displayed_line > view->nlines - 2)
2627 s->first_displayed_line -=
2628 (view->nlines - 2);
2629 else
2630 s->first_displayed_line = 1;
2631 break;
2632 case 'j':
2633 case KEY_DOWN:
2634 if (s->selected_line < view->nlines - 2 &&
2635 s->first_displayed_line +
2636 s->selected_line <= s->blame.nlines)
2637 s->selected_line++;
2638 else if (s->last_displayed_line <
2639 s->blame.nlines)
2640 s->first_displayed_line++;
2641 break;
2642 case 'b':
2643 case 'p': {
2644 struct got_object_id *id = NULL;
2645 id = get_selected_commit_id(s->blame.lines,
2646 s->first_displayed_line, s->selected_line);
2647 if (id == NULL)
2648 break;
2649 if (ch == 'p') {
2650 struct got_commit_object *commit;
2651 struct got_object_qid *pid;
2652 struct got_object_id *blob_id = NULL;
2653 int obj_type;
2654 err = got_object_open_as_commit(&commit,
2655 s->repo, id);
2656 if (err)
2657 break;
2658 pid = SIMPLEQ_FIRST(
2659 got_object_commit_get_parent_ids(commit));
2660 if (pid == NULL) {
2661 got_object_commit_close(commit);
2662 break;
2664 /* Check if path history ends here. */
2665 err = got_object_id_by_path(&blob_id, s->repo,
2666 pid->id, s->path);
2667 if (err) {
2668 if (err->code == GOT_ERR_NO_TREE_ENTRY)
2669 err = NULL;
2670 got_object_commit_close(commit);
2671 break;
2673 err = got_object_get_type(&obj_type, s->repo,
2674 blob_id);
2675 free(blob_id);
2676 /* Can't blame non-blob type objects. */
2677 if (obj_type != GOT_OBJ_TYPE_BLOB) {
2678 got_object_commit_close(commit);
2679 break;
2681 err = got_object_qid_alloc(&s->blamed_commit,
2682 pid->id);
2683 got_object_commit_close(commit);
2684 } else {
2685 if (got_object_id_cmp(id,
2686 s->blamed_commit->id) == 0)
2687 break;
2688 err = got_object_qid_alloc(&s->blamed_commit,
2689 id);
2691 if (err)
2692 break;
2693 s->done = 1;
2694 thread_err = stop_blame(&s->blame);
2695 s->done = 0;
2696 if (thread_err)
2697 break;
2698 SIMPLEQ_INSERT_HEAD(&s->blamed_commits,
2699 s->blamed_commit, entry);
2700 err = run_blame(&s->blame, view, &s->blame_complete,
2701 &s->first_displayed_line, &s->last_displayed_line,
2702 &s->selected_line, &s->done, &s->eof,
2703 s->path, s->blamed_commit->id, s->repo);
2704 if (err)
2705 break;
2706 break;
2708 case 'B': {
2709 struct got_object_qid *first;
2710 first = SIMPLEQ_FIRST(&s->blamed_commits);
2711 if (!got_object_id_cmp(first->id, s->commit_id))
2712 break;
2713 s->done = 1;
2714 thread_err = stop_blame(&s->blame);
2715 s->done = 0;
2716 if (thread_err)
2717 break;
2718 SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
2719 got_object_qid_free(s->blamed_commit);
2720 s->blamed_commit =
2721 SIMPLEQ_FIRST(&s->blamed_commits);
2722 err = run_blame(&s->blame, view, &s->blame_complete,
2723 &s->first_displayed_line, &s->last_displayed_line,
2724 &s->selected_line, &s->done, &s->eof, s->path,
2725 s->blamed_commit->id, s->repo);
2726 if (err)
2727 break;
2728 break;
2730 case KEY_ENTER:
2731 case '\r': {
2732 struct got_object_id *id = NULL;
2733 struct got_object_qid *pid;
2734 struct got_commit_object *commit = NULL;
2735 id = get_selected_commit_id(s->blame.lines,
2736 s->first_displayed_line, s->selected_line);
2737 if (id == NULL)
2738 break;
2739 err = got_object_open_as_commit(&commit, s->repo, id);
2740 if (err)
2741 break;
2742 pid = SIMPLEQ_FIRST(
2743 got_object_commit_get_parent_ids(commit));
2744 if (view_is_parent_view(view))
2745 begin_x = view_split_begin_x(view->begin_x);
2746 diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
2747 if (diff_view == NULL) {
2748 got_object_commit_close(commit);
2749 err = got_error_from_errno();
2750 break;
2752 err = open_diff_view(diff_view, pid ? pid->id : NULL,
2753 id, s->repo);
2754 got_object_commit_close(commit);
2755 if (err) {
2756 view_close(diff_view);
2757 break;
2759 if (view_is_parent_view(view)) {
2760 err = view_close_child(view);
2761 if (err)
2762 break;
2763 err = view_set_child(view, diff_view);
2764 if (err) {
2765 view_close(diff_view);
2766 break;
2768 *focus_view = diff_view;
2769 view->child_focussed = 1;
2770 } else
2771 *new_view = diff_view;
2772 if (err)
2773 break;
2774 break;
2776 case KEY_NPAGE:
2777 case ' ':
2778 if (s->last_displayed_line >= s->blame.nlines &&
2779 s->selected_line < view->nlines - 2) {
2780 s->selected_line = MIN(s->blame.nlines,
2781 view->nlines - 2);
2782 break;
2784 if (s->last_displayed_line + view->nlines - 2
2785 <= s->blame.nlines)
2786 s->first_displayed_line +=
2787 view->nlines - 2;
2788 else
2789 s->first_displayed_line =
2790 s->blame.nlines -
2791 (view->nlines - 3);
2792 break;
2793 case KEY_RESIZE:
2794 if (s->selected_line > view->nlines - 2) {
2795 s->selected_line = MIN(s->blame.nlines,
2796 view->nlines - 2);
2798 break;
2799 default:
2800 break;
2802 return thread_err ? thread_err : err;
2805 static const struct got_error *
2806 cmd_blame(int argc, char *argv[])
2808 const struct got_error *error;
2809 struct got_repository *repo = NULL;
2810 char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
2811 struct got_object_id *commit_id = NULL;
2812 char *commit_id_str = NULL;
2813 int ch;
2814 struct tog_view *view;
2816 #ifndef PROFILE
2817 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
2818 NULL) == -1)
2819 err(1, "pledge");
2820 #endif
2822 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
2823 switch (ch) {
2824 case 'c':
2825 commit_id_str = optarg;
2826 break;
2827 case 'r':
2828 repo_path = realpath(optarg, NULL);
2829 if (repo_path == NULL)
2830 err(1, "-r option");
2831 break;
2832 default:
2833 usage();
2834 /* NOTREACHED */
2838 argc -= optind;
2839 argv += optind;
2841 if (argc == 1)
2842 path = argv[0];
2843 else
2844 usage_blame();
2846 cwd = getcwd(NULL, 0);
2847 if (cwd == NULL) {
2848 error = got_error_from_errno();
2849 goto done;
2851 if (repo_path == NULL) {
2852 repo_path = strdup(cwd);
2853 if (repo_path == NULL) {
2854 error = got_error_from_errno();
2855 goto done;
2859 error = apply_unveil(repo_path, NULL);
2860 if (error)
2861 goto done;
2863 error = got_repo_open(&repo, repo_path);
2864 if (error != NULL)
2865 goto done;
2867 error = got_repo_map_path(&in_repo_path, repo, path, 1);
2868 if (error != NULL)
2869 goto done;
2871 if (commit_id_str == NULL) {
2872 struct got_reference *head_ref;
2873 error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
2874 if (error != NULL)
2875 goto done;
2876 error = got_ref_resolve(&commit_id, repo, head_ref);
2877 got_ref_close(head_ref);
2878 } else {
2879 error = got_object_resolve_id_str(&commit_id, repo,
2880 commit_id_str);
2882 if (error != NULL)
2883 goto done;
2885 view = view_open(0, 0, 0, 0, TOG_VIEW_BLAME);
2886 if (view == NULL) {
2887 error = got_error_from_errno();
2888 goto done;
2890 error = open_blame_view(view, in_repo_path, commit_id, repo);
2891 if (error)
2892 goto done;
2893 error = view_loop(view);
2894 done:
2895 free(repo_path);
2896 free(cwd);
2897 free(commit_id);
2898 if (repo)
2899 got_repo_close(repo);
2900 return error;
2903 static const struct got_error *
2904 draw_tree_entries(struct tog_view *view,
2905 struct got_tree_entry **first_displayed_entry,
2906 struct got_tree_entry **last_displayed_entry,
2907 struct got_tree_entry **selected_entry, int *ndisplayed,
2908 const char *label, int show_ids, const char *parent_path,
2909 const struct got_tree_entries *entries, int selected, int limit, int isroot)
2911 const struct got_error *err = NULL;
2912 struct got_tree_entry *te;
2913 wchar_t *wline;
2914 int width, n;
2916 *ndisplayed = 0;
2918 werase(view->window);
2920 if (limit == 0)
2921 return NULL;
2923 err = format_line(&wline, &width, label, view->ncols);
2924 if (err)
2925 return err;
2926 if (view_needs_focus_indication(view))
2927 wstandout(view->window);
2928 waddwstr(view->window, wline);
2929 if (view_needs_focus_indication(view))
2930 wstandend(view->window);
2931 free(wline);
2932 wline = NULL;
2933 if (width < view->ncols)
2934 waddch(view->window, '\n');
2935 if (--limit <= 0)
2936 return NULL;
2937 err = format_line(&wline, &width, parent_path, view->ncols);
2938 if (err)
2939 return err;
2940 waddwstr(view->window, wline);
2941 free(wline);
2942 wline = NULL;
2943 if (width < view->ncols)
2944 waddch(view->window, '\n');
2945 if (--limit <= 0)
2946 return NULL;
2947 waddch(view->window, '\n');
2948 if (--limit <= 0)
2949 return NULL;
2951 te = SIMPLEQ_FIRST(&entries->head);
2952 if (*first_displayed_entry == NULL) {
2953 if (selected == 0) {
2954 if (view->focussed)
2955 wstandout(view->window);
2956 *selected_entry = NULL;
2958 waddstr(view->window, " ..\n"); /* parent directory */
2959 if (selected == 0 && view->focussed)
2960 wstandend(view->window);
2961 (*ndisplayed)++;
2962 if (--limit <= 0)
2963 return NULL;
2964 n = 1;
2965 } else {
2966 n = 0;
2967 while (te != *first_displayed_entry)
2968 te = SIMPLEQ_NEXT(te, entry);
2971 while (te) {
2972 char *line = NULL, *id_str = NULL;
2974 if (show_ids) {
2975 err = got_object_id_str(&id_str, te->id);
2976 if (err)
2977 return got_error_from_errno();
2979 if (asprintf(&line, "%s %s%s", id_str ? id_str : "",
2980 te->name, S_ISDIR(te->mode) ? "/" : "") == -1) {
2981 free(id_str);
2982 return got_error_from_errno();
2984 free(id_str);
2985 err = format_line(&wline, &width, line, view->ncols);
2986 if (err) {
2987 free(line);
2988 break;
2990 if (n == selected) {
2991 if (view->focussed)
2992 wstandout(view->window);
2993 *selected_entry = te;
2995 waddwstr(view->window, wline);
2996 if (width < view->ncols)
2997 waddch(view->window, '\n');
2998 if (n == selected && view->focussed)
2999 wstandend(view->window);
3000 free(line);
3001 free(wline);
3002 wline = NULL;
3003 n++;
3004 (*ndisplayed)++;
3005 *last_displayed_entry = te;
3006 if (--limit <= 0)
3007 break;
3008 te = SIMPLEQ_NEXT(te, entry);
3011 return err;
3014 static void
3015 tree_scroll_up(struct got_tree_entry **first_displayed_entry, int maxscroll,
3016 const struct got_tree_entries *entries, int isroot)
3018 struct got_tree_entry *te, *prev;
3019 int i;
3021 if (*first_displayed_entry == NULL)
3022 return;
3024 te = SIMPLEQ_FIRST(&entries->head);
3025 if (*first_displayed_entry == te) {
3026 if (!isroot)
3027 *first_displayed_entry = NULL;
3028 return;
3031 /* XXX this is stupid... switch to TAILQ? */
3032 for (i = 0; i < maxscroll; i++) {
3033 while (te != *first_displayed_entry) {
3034 prev = te;
3035 te = SIMPLEQ_NEXT(te, entry);
3037 *first_displayed_entry = prev;
3038 te = SIMPLEQ_FIRST(&entries->head);
3040 if (!isroot && te == SIMPLEQ_FIRST(&entries->head) && i < maxscroll)
3041 *first_displayed_entry = NULL;
3044 static int
3045 tree_scroll_down(struct got_tree_entry **first_displayed_entry, int maxscroll,
3046 struct got_tree_entry *last_displayed_entry,
3047 const struct got_tree_entries *entries)
3049 struct got_tree_entry *next, *last;
3050 int n = 0;
3052 if (*first_displayed_entry)
3053 next = SIMPLEQ_NEXT(*first_displayed_entry, entry);
3054 else
3055 next = SIMPLEQ_FIRST(&entries->head);
3056 last = last_displayed_entry;
3057 while (next && last && n++ < maxscroll) {
3058 last = SIMPLEQ_NEXT(last, entry);
3059 if (last) {
3060 *first_displayed_entry = next;
3061 next = SIMPLEQ_NEXT(next, entry);
3064 return n;
3067 static const struct got_error *
3068 tree_entry_path(char **path, struct tog_parent_trees *parents,
3069 struct got_tree_entry *te)
3071 const struct got_error *err = NULL;
3072 struct tog_parent_tree *pt;
3073 size_t len = 2; /* for leading slash and NUL */
3075 TAILQ_FOREACH(pt, parents, entry)
3076 len += strlen(pt->selected_entry->name) + 1 /* slash */;
3077 if (te)
3078 len += strlen(te->name);
3080 *path = calloc(1, len);
3081 if (path == NULL)
3082 return got_error_from_errno();
3084 (*path)[0] = '/';
3085 pt = TAILQ_LAST(parents, tog_parent_trees);
3086 while (pt) {
3087 if (strlcat(*path, pt->selected_entry->name, len) >= len) {
3088 err = got_error(GOT_ERR_NO_SPACE);
3089 goto done;
3091 if (strlcat(*path, "/", len) >= len) {
3092 err = got_error(GOT_ERR_NO_SPACE);
3093 goto done;
3095 pt = TAILQ_PREV(pt, tog_parent_trees, entry);
3097 if (te) {
3098 if (strlcat(*path, te->name, len) >= len) {
3099 err = got_error(GOT_ERR_NO_SPACE);
3100 goto done;
3103 done:
3104 if (err) {
3105 free(*path);
3106 *path = NULL;
3108 return err;
3111 static const struct got_error *
3112 blame_tree_entry(struct tog_view **new_view, int begin_x,
3113 struct got_tree_entry *te, struct tog_parent_trees *parents,
3114 struct got_object_id *commit_id, struct got_repository *repo)
3116 const struct got_error *err = NULL;
3117 char *path;
3118 struct tog_view *blame_view;
3120 err = tree_entry_path(&path, parents, te);
3121 if (err)
3122 return err;
3124 blame_view = view_open(0, 0, 0, begin_x, TOG_VIEW_BLAME);
3125 if (blame_view == NULL)
3126 return got_error_from_errno();
3128 err = open_blame_view(blame_view, path, commit_id, repo);
3129 if (err) {
3130 view_close(blame_view);
3131 free(path);
3132 } else
3133 *new_view = blame_view;
3134 return err;
3137 static const struct got_error *
3138 log_tree_entry(struct tog_view **new_view, int begin_x,
3139 struct got_tree_entry *te, struct tog_parent_trees *parents,
3140 struct got_object_id *commit_id, struct got_repository *repo)
3142 struct tog_view *log_view;
3143 const struct got_error *err = NULL;
3144 char *path;
3146 log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
3147 if (log_view == NULL)
3148 return got_error_from_errno();
3150 err = tree_entry_path(&path, parents, te);
3151 if (err)
3152 return err;
3154 err = open_log_view(log_view, commit_id, repo, path, 0);
3155 if (err)
3156 view_close(log_view);
3157 else
3158 *new_view = log_view;
3159 free(path);
3160 return err;
3163 static const struct got_error *
3164 open_tree_view(struct tog_view *view, struct got_tree_object *root,
3165 struct got_object_id *commit_id, struct got_repository *repo)
3167 const struct got_error *err = NULL;
3168 char *commit_id_str = NULL;
3169 struct tog_tree_view_state *s = &view->state.tree;
3171 TAILQ_INIT(&s->parents);
3173 err = got_object_id_str(&commit_id_str, commit_id);
3174 if (err != NULL)
3175 goto done;
3177 if (asprintf(&s->tree_label, "commit %s", commit_id_str) == -1) {
3178 err = got_error_from_errno();
3179 goto done;
3182 s->root = s->tree = root;
3183 s->entries = got_object_tree_get_entries(root);
3184 s->first_displayed_entry = SIMPLEQ_FIRST(&s->entries->head);
3185 s->commit_id = got_object_id_dup(commit_id);
3186 if (s->commit_id == NULL) {
3187 err = got_error_from_errno();
3188 goto done;
3190 s->repo = repo;
3192 view->show = show_tree_view;
3193 view->input = input_tree_view;
3194 view->close = close_tree_view;
3195 done:
3196 free(commit_id_str);
3197 if (err) {
3198 free(s->tree_label);
3199 s->tree_label = NULL;
3201 return err;
3204 static const struct got_error *
3205 close_tree_view(struct tog_view *view)
3207 struct tog_tree_view_state *s = &view->state.tree;
3209 free(s->tree_label);
3210 s->tree_label = NULL;
3211 free(s->commit_id);
3212 s->commit_id = NULL;
3213 while (!TAILQ_EMPTY(&s->parents)) {
3214 struct tog_parent_tree *parent;
3215 parent = TAILQ_FIRST(&s->parents);
3216 TAILQ_REMOVE(&s->parents, parent, entry);
3217 free(parent);
3220 if (s->tree != s->root)
3221 got_object_tree_close(s->tree);
3222 got_object_tree_close(s->root);
3224 return NULL;
3227 static const struct got_error *
3228 show_tree_view(struct tog_view *view)
3230 const struct got_error *err = NULL;
3231 struct tog_tree_view_state *s = &view->state.tree;
3232 char *parent_path;
3234 err = tree_entry_path(&parent_path, &s->parents, NULL);
3235 if (err)
3236 return err;
3238 err = draw_tree_entries(view, &s->first_displayed_entry,
3239 &s->last_displayed_entry, &s->selected_entry,
3240 &s->ndisplayed, s->tree_label, s->show_ids, parent_path,
3241 s->entries, s->selected, view->nlines, s->tree == s->root);
3242 free(parent_path);
3244 view_vborder(view);
3245 return err;
3248 static const struct got_error *
3249 input_tree_view(struct tog_view **new_view, struct tog_view **dead_view,
3250 struct tog_view **focus_view, struct tog_view *view, int ch)
3252 const struct got_error *err = NULL;
3253 struct tog_tree_view_state *s = &view->state.tree;
3254 struct tog_view *log_view;
3255 int begin_x = 0, nscrolled;
3257 switch (ch) {
3258 case 'i':
3259 s->show_ids = !s->show_ids;
3260 break;
3261 case 'l':
3262 if (!s->selected_entry)
3263 break;
3264 if (view_is_parent_view(view))
3265 begin_x = view_split_begin_x(view->begin_x);
3266 err = log_tree_entry(&log_view, begin_x,
3267 s->selected_entry, &s->parents,
3268 s->commit_id, s->repo);
3269 if (view_is_parent_view(view)) {
3270 err = view_close_child(view);
3271 if (err)
3272 return err;
3273 err = view_set_child(view, log_view);
3274 if (err) {
3275 view_close(log_view);
3276 break;
3278 *focus_view = log_view;
3279 view->child_focussed = 1;
3280 } else
3281 *new_view = log_view;
3282 break;
3283 case 'k':
3284 case KEY_UP:
3285 if (s->selected > 0) {
3286 s->selected--;
3287 if (s->selected == 0)
3288 break;
3290 if (s->selected > 0)
3291 break;
3292 tree_scroll_up(&s->first_displayed_entry, 1,
3293 s->entries, s->tree == s->root);
3294 break;
3295 case KEY_PPAGE:
3296 tree_scroll_up(&s->first_displayed_entry,
3297 MAX(0, view->nlines - 4 - s->selected), s->entries,
3298 s->tree == s->root);
3299 s->selected = 0;
3300 if (SIMPLEQ_FIRST(&s->entries->head) ==
3301 s->first_displayed_entry && s->tree != s->root)
3302 s->first_displayed_entry = NULL;
3303 break;
3304 case 'j':
3305 case KEY_DOWN:
3306 if (s->selected < s->ndisplayed - 1) {
3307 s->selected++;
3308 break;
3310 if (SIMPLEQ_NEXT(s->last_displayed_entry, entry)
3311 == NULL) {
3312 /* can't scroll any further */
3313 break;
3315 tree_scroll_down(&s->first_displayed_entry, 1,
3316 s->last_displayed_entry, s->entries);
3317 break;
3318 case KEY_NPAGE:
3319 if (SIMPLEQ_NEXT(s->last_displayed_entry, entry)
3320 == NULL) {
3321 /* can't scroll any further; move cursor down */
3322 if (s->selected < s->ndisplayed - 1)
3323 s->selected = s->ndisplayed - 1;
3324 break;
3326 nscrolled = tree_scroll_down(&s->first_displayed_entry,
3327 view->nlines, s->last_displayed_entry, s->entries);
3328 if (nscrolled < view->nlines) {
3329 int ndisplayed = 0;
3330 struct got_tree_entry *te;
3331 te = s->first_displayed_entry;
3332 do {
3333 ndisplayed++;
3334 te = SIMPLEQ_NEXT(te, entry);
3335 } while (te);
3336 s->selected = ndisplayed - 1;
3338 break;
3339 case KEY_ENTER:
3340 case '\r':
3341 if (s->selected_entry == NULL) {
3342 struct tog_parent_tree *parent;
3343 case KEY_BACKSPACE:
3344 /* user selected '..' */
3345 if (s->tree == s->root)
3346 break;
3347 parent = TAILQ_FIRST(&s->parents);
3348 TAILQ_REMOVE(&s->parents, parent,
3349 entry);
3350 got_object_tree_close(s->tree);
3351 s->tree = parent->tree;
3352 s->entries =
3353 got_object_tree_get_entries(s->tree);
3354 s->first_displayed_entry =
3355 parent->first_displayed_entry;
3356 s->selected_entry =
3357 parent->selected_entry;
3358 s->selected = parent->selected;
3359 free(parent);
3360 } else if (S_ISDIR(s->selected_entry->mode)) {
3361 struct tog_parent_tree *parent;
3362 struct got_tree_object *child;
3363 err = got_object_open_as_tree(&child,
3364 s->repo, s->selected_entry->id);
3365 if (err)
3366 break;
3367 parent = calloc(1, sizeof(*parent));
3368 if (parent == NULL) {
3369 err = got_error_from_errno();
3370 break;
3372 parent->tree = s->tree;
3373 parent->first_displayed_entry =
3374 s->first_displayed_entry;
3375 parent->selected_entry = s->selected_entry;
3376 parent->selected = s->selected;
3377 TAILQ_INSERT_HEAD(&s->parents, parent, entry);
3378 s->tree = child;
3379 s->entries =
3380 got_object_tree_get_entries(s->tree);
3381 s->selected = 0;
3382 s->first_displayed_entry = NULL;
3383 } else if (S_ISREG(s->selected_entry->mode)) {
3384 struct tog_view *blame_view;
3385 int begin_x = view_is_parent_view(view) ?
3386 view_split_begin_x(view->begin_x) : 0;
3388 err = blame_tree_entry(&blame_view, begin_x,
3389 s->selected_entry, &s->parents, s->commit_id,
3390 s->repo);
3391 if (err)
3392 break;
3393 if (view_is_parent_view(view)) {
3394 err = view_close_child(view);
3395 if (err)
3396 return err;
3397 err = view_set_child(view, blame_view);
3398 if (err) {
3399 view_close(blame_view);
3400 break;
3402 *focus_view = blame_view;
3403 view->child_focussed = 1;
3404 } else
3405 *new_view = blame_view;
3407 break;
3408 case KEY_RESIZE:
3409 if (s->selected > view->nlines)
3410 s->selected = s->ndisplayed - 1;
3411 break;
3412 default:
3413 break;
3416 return err;
3419 __dead static void
3420 usage_tree(void)
3422 endwin();
3423 fprintf(stderr, "usage: %s tree [-c commit] [repository-path]\n",
3424 getprogname());
3425 exit(1);
3428 static const struct got_error *
3429 cmd_tree(int argc, char *argv[])
3431 const struct got_error *error;
3432 struct got_repository *repo = NULL;
3433 char *repo_path = NULL;
3434 struct got_object_id *commit_id = NULL;
3435 char *commit_id_arg = NULL;
3436 struct got_commit_object *commit = NULL;
3437 struct got_tree_object *tree = NULL;
3438 int ch;
3439 struct tog_view *view;
3441 #ifndef PROFILE
3442 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
3443 NULL) == -1)
3444 err(1, "pledge");
3445 #endif
3447 while ((ch = getopt(argc, argv, "c:")) != -1) {
3448 switch (ch) {
3449 case 'c':
3450 commit_id_arg = optarg;
3451 break;
3452 default:
3453 usage();
3454 /* NOTREACHED */
3458 argc -= optind;
3459 argv += optind;
3461 if (argc == 0) {
3462 struct got_worktree *worktree;
3463 char *cwd = getcwd(NULL, 0);
3464 if (cwd == NULL)
3465 return got_error_from_errno();
3466 error = got_worktree_open(&worktree, cwd);
3467 if (error && error->code != GOT_ERR_NOT_WORKTREE)
3468 goto done;
3469 if (worktree) {
3470 free(cwd);
3471 repo_path =
3472 strdup(got_worktree_get_repo_path(worktree));
3473 got_worktree_close(worktree);
3474 } else
3475 repo_path = cwd;
3476 if (repo_path == NULL) {
3477 error = got_error_from_errno();
3478 goto done;
3480 } else if (argc == 1) {
3481 repo_path = realpath(argv[0], NULL);
3482 if (repo_path == NULL)
3483 return got_error_from_errno();
3484 } else
3485 usage_log();
3487 error = apply_unveil(repo_path, NULL);
3488 if (error)
3489 goto done;
3491 error = got_repo_open(&repo, repo_path);
3492 if (error != NULL)
3493 goto done;
3495 if (commit_id_arg == NULL)
3496 error = get_head_commit_id(&commit_id, repo);
3497 else
3498 error = got_object_resolve_id_str(&commit_id, repo,
3499 commit_id_arg);
3500 if (error != NULL)
3501 goto done;
3503 error = got_object_open_as_commit(&commit, repo, commit_id);
3504 if (error != NULL)
3505 goto done;
3507 error = got_object_open_as_tree(&tree, repo,
3508 got_object_commit_get_tree_id(commit));
3509 if (error != NULL)
3510 goto done;
3512 view = view_open(0, 0, 0, 0, TOG_VIEW_TREE);
3513 if (view == NULL) {
3514 error = got_error_from_errno();
3515 goto done;
3517 error = open_tree_view(view, tree, commit_id, repo);
3518 if (error)
3519 goto done;
3520 error = view_loop(view);
3521 done:
3522 free(repo_path);
3523 free(commit_id);
3524 if (commit)
3525 got_object_commit_close(commit);
3526 if (tree)
3527 got_object_tree_close(tree);
3528 if (repo)
3529 got_repo_close(repo);
3530 return error;
3533 static void
3534 init_curses(void)
3536 initscr();
3537 cbreak();
3538 halfdelay(1); /* Do fast refresh while initial view is loading. */
3539 noecho();
3540 nonl();
3541 intrflush(stdscr, FALSE);
3542 keypad(stdscr, TRUE);
3543 curs_set(0);
3544 signal(SIGWINCH, tog_sigwinch);
3547 __dead static void
3548 usage(void)
3550 int i;
3552 fprintf(stderr, "usage: %s [-h] [command] [arg ...]\n\n"
3553 "Available commands:\n", getprogname());
3554 for (i = 0; i < nitems(tog_commands); i++) {
3555 struct tog_cmd *cmd = &tog_commands[i];
3556 fprintf(stderr, " %s: %s\n", cmd->name, cmd->descr);
3558 exit(1);
3561 static char **
3562 make_argv(const char *arg0, const char *arg1)
3564 char **argv;
3565 int argc = (arg1 == NULL ? 1 : 2);
3567 argv = calloc(argc, sizeof(char *));
3568 if (argv == NULL)
3569 err(1, "calloc");
3570 argv[0] = strdup(arg0);
3571 if (argv[0] == NULL)
3572 err(1, "calloc");
3573 if (arg1) {
3574 argv[1] = strdup(arg1);
3575 if (argv[1] == NULL)
3576 err(1, "calloc");
3579 return argv;
3582 int
3583 main(int argc, char *argv[])
3585 const struct got_error *error = NULL;
3586 struct tog_cmd *cmd = NULL;
3587 int ch, hflag = 0;
3588 char **cmd_argv = NULL;
3590 setlocale(LC_CTYPE, "");
3592 while ((ch = getopt(argc, argv, "h")) != -1) {
3593 switch (ch) {
3594 case 'h':
3595 hflag = 1;
3596 break;
3597 default:
3598 usage();
3599 /* NOTREACHED */
3603 argc -= optind;
3604 argv += optind;
3605 optind = 0;
3606 optreset = 1;
3608 if (argc == 0) {
3609 if (hflag)
3610 usage();
3611 /* Build an argument vector which runs a default command. */
3612 cmd = &tog_commands[0];
3613 cmd_argv = make_argv(cmd->name, NULL);
3614 argc = 1;
3615 } else {
3616 int i;
3618 /* Did the user specific a command? */
3619 for (i = 0; i < nitems(tog_commands); i++) {
3620 if (strncmp(tog_commands[i].name, argv[0],
3621 strlen(argv[0])) == 0) {
3622 cmd = &tog_commands[i];
3623 if (hflag)
3624 tog_commands[i].cmd_usage();
3625 break;
3628 if (cmd == NULL) {
3629 /* Did the user specify a repository? */
3630 char *repo_path = realpath(argv[0], NULL);
3631 if (repo_path) {
3632 struct got_repository *repo;
3633 error = got_repo_open(&repo, repo_path);
3634 if (error == NULL)
3635 got_repo_close(repo);
3636 } else
3637 error = got_error_from_errno();
3638 if (error) {
3639 if (hflag) {
3640 fprintf(stderr, "%s: '%s' is not a "
3641 "known command\n", getprogname(),
3642 argv[0]);
3643 usage();
3645 fprintf(stderr, "%s: '%s' is neither a known "
3646 "command nor a path to a repository\n",
3647 getprogname(), argv[0]);
3648 free(repo_path);
3649 return 1;
3651 cmd = &tog_commands[0];
3652 cmd_argv = make_argv(cmd->name, repo_path);
3653 argc = 2;
3654 free(repo_path);
3658 init_curses();
3660 error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
3661 if (error)
3662 goto done;
3663 done:
3664 endwin();
3665 free(cmd_argv);
3666 if (error)
3667 fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
3668 return 0;