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 commit_queue_entry {
102 TAILQ_ENTRY(commit_queue_entry) entry;
103 struct got_object_id *id;
104 struct got_commit_object *commit;
105 int idx;
106 };
107 TAILQ_HEAD(commit_queue_head, commit_queue_entry);
108 struct commit_queue {
109 int ncommits;
110 struct commit_queue_head head;
111 };
113 struct tog_diff_view_state {
114 struct got_object_id *id1, *id2;
115 FILE *f;
116 int first_displayed_line;
117 int last_displayed_line;
118 int eof;
119 int diff_context;
120 struct got_repository *repo;
121 struct got_reflist_head *refs;
123 /* passed from log view; may be NULL */
124 struct tog_view *log_view;
125 };
127 pthread_mutex_t tog_mutex = PTHREAD_MUTEX_INITIALIZER;
129 struct tog_log_thread_args {
130 pthread_cond_t need_commits;
131 int commits_needed;
132 struct got_commit_graph *graph;
133 struct commit_queue *commits;
134 const char *in_repo_path;
135 struct got_object_id *start_id;
136 struct got_repository *repo;
137 int log_complete;
138 sig_atomic_t *quit;
139 struct tog_view *view;
140 struct commit_queue_entry **first_displayed_entry;
141 struct commit_queue_entry **selected_entry;
142 };
144 struct tog_log_view_state {
145 struct commit_queue commits;
146 struct commit_queue_entry *first_displayed_entry;
147 struct commit_queue_entry *last_displayed_entry;
148 struct commit_queue_entry *selected_entry;
149 int selected;
150 char *in_repo_path;
151 struct got_repository *repo;
152 struct got_reflist_head *refs;
153 struct got_object_id *start_id;
154 sig_atomic_t quit;
155 pthread_t thread;
156 struct tog_log_thread_args thread_args;
157 };
159 struct tog_blame_cb_args {
160 struct tog_blame_line *lines; /* one per line */
161 int nlines;
163 struct tog_view *view;
164 struct got_object_id *commit_id;
165 int *quit;
166 };
168 struct tog_blame_thread_args {
169 const char *path;
170 struct got_repository *repo;
171 struct tog_blame_cb_args *cb_args;
172 int *complete;
173 };
175 struct tog_blame {
176 FILE *f;
177 size_t filesize;
178 struct tog_blame_line *lines;
179 int nlines;
180 pthread_t thread;
181 struct tog_blame_thread_args thread_args;
182 struct tog_blame_cb_args cb_args;
183 const char *path;
184 };
186 struct tog_blame_view_state {
187 int first_displayed_line;
188 int last_displayed_line;
189 int selected_line;
190 int blame_complete;
191 int eof;
192 int done;
193 struct got_object_id_queue blamed_commits;
194 struct got_object_qid *blamed_commit;
195 char *path;
196 struct got_repository *repo;
197 struct got_reflist_head *refs;
198 struct got_object_id *commit_id;
199 struct tog_blame blame;
200 };
202 struct tog_parent_tree {
203 TAILQ_ENTRY(tog_parent_tree) entry;
204 struct got_tree_object *tree;
205 struct got_tree_entry *first_displayed_entry;
206 struct got_tree_entry *selected_entry;
207 int selected;
208 };
210 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
212 struct tog_tree_view_state {
213 char *tree_label;
214 struct got_tree_object *root;
215 struct got_tree_object *tree;
216 const struct got_tree_entries *entries;
217 struct got_tree_entry *first_displayed_entry;
218 struct got_tree_entry *last_displayed_entry;
219 struct got_tree_entry *selected_entry;
220 int ndisplayed, selected, show_ids;
221 struct tog_parent_trees parents;
222 struct got_object_id *commit_id;
223 struct got_repository *repo;
224 struct got_reflist_head *refs;
225 };
227 /*
228 * We implement two types of views: parent views and child views.
230 * The 'Tab' key switches between a parent view and its child view.
231 * Child views are shown side-by-side to their parent view, provided
232 * there is enough screen estate.
234 * When a new view is opened from within a parent view, this new view
235 * becomes a child view of the parent view, replacing any existing child.
237 * When a new view is opened from within a child view, this new view
238 * becomes a parent view which will obscure the views below until the
239 * user quits the new parent view by typing 'q'.
241 * This list of views contains parent views only.
242 * Child views are only pointed to by their parent view.
243 */
244 TAILQ_HEAD(tog_view_list_head, tog_view);
246 struct tog_view {
247 TAILQ_ENTRY(tog_view) entry;
248 WINDOW *window;
249 PANEL *panel;
250 int nlines, ncols, begin_y, begin_x;
251 int lines, cols; /* copies of LINES and COLS */
252 int focussed;
253 struct tog_view *parent;
254 struct tog_view *child;
255 int child_focussed;
257 /* type-specific state */
258 enum tog_view_type type;
259 union {
260 struct tog_diff_view_state diff;
261 struct tog_log_view_state log;
262 struct tog_blame_view_state blame;
263 struct tog_tree_view_state tree;
264 } state;
266 const struct got_error *(*show)(struct tog_view *);
267 const struct got_error *(*input)(struct tog_view **,
268 struct tog_view **, struct tog_view**, struct tog_view *, int);
269 const struct got_error *(*close)(struct tog_view *);
270 };
272 static const struct got_error *open_diff_view(struct tog_view *,
273 struct got_object_id *, struct got_object_id *, struct tog_view *,
274 struct got_reflist_head *, struct got_repository *);
275 static const struct got_error *show_diff_view(struct tog_view *);
276 static const struct got_error *input_diff_view(struct tog_view **,
277 struct tog_view **, struct tog_view **, struct tog_view *, int);
278 static const struct got_error* close_diff_view(struct tog_view *);
280 static const struct got_error *open_log_view(struct tog_view *,
281 struct got_object_id *, struct got_reflist_head *,
282 struct got_repository *, const char *, int);
283 static const struct got_error * show_log_view(struct tog_view *);
284 static const struct got_error *input_log_view(struct tog_view **,
285 struct tog_view **, struct tog_view **, struct tog_view *, int);
286 static const struct got_error *close_log_view(struct tog_view *);
288 static const struct got_error *open_blame_view(struct tog_view *, char *,
289 struct got_object_id *, struct got_reflist_head *, struct got_repository *);
290 static const struct got_error *show_blame_view(struct tog_view *);
291 static const struct got_error *input_blame_view(struct tog_view **,
292 struct tog_view **, struct tog_view **, struct tog_view *, int);
293 static const struct got_error *close_blame_view(struct tog_view *);
295 static const struct got_error *open_tree_view(struct tog_view *,
296 struct got_tree_object *, struct got_object_id *,
297 struct got_reflist_head *, struct got_repository *);
298 static const struct got_error *show_tree_view(struct tog_view *);
299 static const struct got_error *input_tree_view(struct tog_view **,
300 struct tog_view **, struct tog_view **, struct tog_view *, int);
301 static const struct got_error *close_tree_view(struct tog_view *);
303 static volatile sig_atomic_t tog_sigwinch_received;
305 static void
306 tog_sigwinch(int signo)
308 tog_sigwinch_received = 1;
311 static const struct got_error *
312 view_close(struct tog_view *view)
314 const struct got_error *err = NULL;
316 if (view->child) {
317 view_close(view->child);
318 view->child = NULL;
320 if (view->close)
321 err = view->close(view);
322 if (view->panel)
323 del_panel(view->panel);
324 if (view->window)
325 delwin(view->window);
326 free(view);
327 return err;
330 static struct tog_view *
331 view_open(int nlines, int ncols, int begin_y, int begin_x,
332 enum tog_view_type type)
334 struct tog_view *view = calloc(1, sizeof(*view));
336 if (view == NULL)
337 return NULL;
339 view->type = type;
340 view->lines = LINES;
341 view->cols = COLS;
342 view->nlines = nlines ? nlines : LINES - begin_y;
343 view->ncols = ncols ? ncols : COLS - begin_x;
344 view->begin_y = begin_y;
345 view->begin_x = begin_x;
346 view->window = newwin(nlines, ncols, begin_y, begin_x);
347 if (view->window == NULL) {
348 view_close(view);
349 return NULL;
351 view->panel = new_panel(view->window);
352 if (view->panel == NULL ||
353 set_panel_userptr(view->panel, view) != OK) {
354 view_close(view);
355 return NULL;
358 keypad(view->window, TRUE);
359 return view;
362 static int
363 view_split_begin_x(int begin_x)
365 if (begin_x > 0 || COLS < 120)
366 return 0;
367 return (COLS - MAX(COLS / 2, 80));
370 static const struct got_error *view_resize(struct tog_view *);
372 static const struct got_error *
373 view_splitscreen(struct tog_view *view)
375 const struct got_error *err = NULL;
377 view->begin_y = 0;
378 view->begin_x = view_split_begin_x(0);
379 view->nlines = LINES;
380 view->ncols = COLS - view->begin_x;
381 view->lines = LINES;
382 view->cols = COLS;
383 err = view_resize(view);
384 if (err)
385 return err;
387 if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
388 return got_error_from_errno();
390 return NULL;
393 static const struct got_error *
394 view_fullscreen(struct tog_view *view)
396 const struct got_error *err = NULL;
398 view->begin_x = 0;
399 view->begin_y = 0;
400 view->nlines = LINES;
401 view->ncols = COLS;
402 view->lines = LINES;
403 view->cols = COLS;
404 err = view_resize(view);
405 if (err)
406 return err;
408 if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
409 return got_error_from_errno();
411 return NULL;
414 static int
415 view_is_parent_view(struct tog_view *view)
417 return view->parent == NULL;
420 static const struct got_error *
421 view_resize(struct tog_view *view)
423 int nlines, ncols;
425 if (view->lines > LINES)
426 nlines = view->nlines - (view->lines - LINES);
427 else
428 nlines = view->nlines + (LINES - view->lines);
430 if (view->cols > COLS)
431 ncols = view->ncols - (view->cols - COLS);
432 else
433 ncols = view->ncols + (COLS - view->cols);
435 if (wresize(view->window, nlines, ncols) == ERR)
436 return got_error_from_errno();
437 if (replace_panel(view->panel, view->window) == ERR)
438 return got_error_from_errno();
439 wclear(view->window);
441 view->nlines = nlines;
442 view->ncols = ncols;
443 view->lines = LINES;
444 view->cols = COLS;
446 if (view->child) {
447 view->child->begin_x = view_split_begin_x(view->begin_x);
448 if (view->child->begin_x == 0) {
449 view_fullscreen(view->child);
450 if (view->child->focussed)
451 show_panel(view->child->panel);
452 else
453 show_panel(view->panel);
454 } else {
455 view_splitscreen(view->child);
456 show_panel(view->child->panel);
460 return NULL;
463 static const struct got_error *
464 view_close_child(struct tog_view *view)
466 const struct got_error *err = NULL;
468 if (view->child == NULL)
469 return NULL;
471 err = view_close(view->child);
472 view->child = NULL;
473 return err;
476 static const struct got_error *
477 view_set_child(struct tog_view *view, struct tog_view *child)
479 const struct got_error *err = NULL;
481 view->child = child;
482 child->parent = view;
483 return err;
486 static int
487 view_is_splitscreen(struct tog_view *view)
489 return view->begin_x > 0;
492 /*
493 * Erase all content of the view. Can be used to "flash" the view because
494 * the view loop will redraw it quickly, providing a more subtle visual
495 * effect than curs_flash(3) would provide.
496 */
497 static void
498 view_flash(struct tog_view *view)
500 werase(view->window);
501 update_panels();
502 doupdate();
505 static void
506 tog_resizeterm(void)
508 int cols, lines;
509 struct winsize size;
511 if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &size) < 0) {
512 cols = 80; /* Default */
513 lines = 24;
514 } else {
515 cols = size.ws_col;
516 lines = size.ws_row;
518 resize_term(lines, cols);
521 static const struct got_error *
522 view_input(struct tog_view **new, struct tog_view **dead,
523 struct tog_view **focus, int *done, struct tog_view *view,
524 struct tog_view_list_head *views)
526 const struct got_error *err = NULL;
527 struct tog_view *v;
528 int ch, errcode;
530 *new = NULL;
531 *dead = NULL;
532 *focus = NULL;
534 nodelay(stdscr, FALSE);
535 /* Allow threads to make progress while we are waiting for input. */
536 errcode = pthread_mutex_unlock(&tog_mutex);
537 if (errcode)
538 return got_error_set_errno(errcode);
539 ch = wgetch(view->window);
540 errcode = pthread_mutex_lock(&tog_mutex);
541 if (errcode)
542 return got_error_set_errno(errcode);
543 nodelay(stdscr, TRUE);
545 if (tog_sigwinch_received) {
546 tog_resizeterm();
547 tog_sigwinch_received = 0;
548 TAILQ_FOREACH(v, views, entry) {
549 err = view_resize(v);
550 if (err)
551 return err;
552 err = v->input(new, dead, focus, v, KEY_RESIZE);
553 if (err)
554 return err;
558 switch (ch) {
559 case ERR:
560 break;
561 case '\t':
562 if (view->child) {
563 *focus = view->child;
564 view->child_focussed = 1;
565 } else if (view->parent) {
566 *focus = view->parent;
567 view->parent->child_focussed = 0;
569 break;
570 case 'q':
571 err = view->input(new, dead, focus, view, ch);
572 *dead = view;
573 break;
574 case 'Q':
575 *done = 1;
576 break;
577 case 'f':
578 if (view_is_parent_view(view)) {
579 if (view->child == NULL)
580 break;
581 if (view_is_splitscreen(view->child)) {
582 *focus = view->child;
583 view->child_focussed = 1;
584 err = view_fullscreen(view->child);
585 } else
586 err = view_splitscreen(view->child);
587 if (err)
588 break;
589 err = view->child->input(new, dead, focus,
590 view->child, KEY_RESIZE);
591 } else {
592 if (view_is_splitscreen(view)) {
593 *focus = view;
594 view->parent->child_focussed = 1;
595 err = view_fullscreen(view);
596 } else {
597 err = view_splitscreen(view);
599 if (err)
600 break;
601 err = view->input(new, dead, focus, view,
602 KEY_RESIZE);
604 break;
605 case KEY_RESIZE:
606 break;
607 default:
608 err = view->input(new, dead, focus, view, ch);
609 break;
612 return err;
615 void
616 view_vborder(struct tog_view *view)
618 PANEL *panel;
619 struct tog_view *view_above;
621 if (view->parent)
622 return view_vborder(view->parent);
624 panel = panel_above(view->panel);
625 if (panel == NULL)
626 return;
628 view_above = panel_userptr(panel);
629 mvwvline(view->window, view->begin_y, view_above->begin_x - 1,
630 got_locale_is_utf8() ? ACS_VLINE : '|', view->nlines);
633 int
634 view_needs_focus_indication(struct tog_view *view)
636 if (view_is_parent_view(view)) {
637 if (view->child == NULL || view->child_focussed)
638 return 0;
639 if (!view_is_splitscreen(view->child))
640 return 0;
641 } else if (!view_is_splitscreen(view))
642 return 0;
644 return view->focussed;
647 static const struct got_error *
648 view_loop(struct tog_view *view)
650 const struct got_error *err = NULL;
651 struct tog_view_list_head views;
652 struct tog_view *new_view, *dead_view, *focus_view, *main_view;
653 int fast_refresh = 10;
654 int done = 0, errcode;
656 errcode = pthread_mutex_lock(&tog_mutex);
657 if (errcode)
658 return got_error_set_errno(errcode);
660 TAILQ_INIT(&views);
661 TAILQ_INSERT_HEAD(&views, view, entry);
663 main_view = view;
664 view->focussed = 1;
665 err = view->show(view);
666 if (err)
667 return err;
668 update_panels();
669 doupdate();
670 while (!TAILQ_EMPTY(&views) && !done) {
671 /* Refresh fast during initialization, then become slower. */
672 if (fast_refresh && fast_refresh-- == 0)
673 halfdelay(10); /* switch to once per second */
675 err = view_input(&new_view, &dead_view, &focus_view, &done,
676 view, &views);
677 if (err)
678 break;
679 if (dead_view) {
680 struct tog_view *prev = NULL;
682 if (view_is_parent_view(dead_view))
683 prev = TAILQ_PREV(dead_view,
684 tog_view_list_head, entry);
685 else if (view->parent != dead_view)
686 prev = view->parent;
688 if (dead_view->parent)
689 dead_view->parent->child = NULL;
690 else
691 TAILQ_REMOVE(&views, dead_view, entry);
693 err = view_close(dead_view);
694 if (err || dead_view == main_view)
695 goto done;
697 if (view == dead_view) {
698 if (focus_view)
699 view = focus_view;
700 else if (prev)
701 view = prev;
702 else if (!TAILQ_EMPTY(&views))
703 view = TAILQ_LAST(&views,
704 tog_view_list_head);
705 else
706 view = NULL;
707 if (view) {
708 if (view->child && view->child_focussed)
709 focus_view = view->child;
710 else
711 focus_view = view;
715 if (new_view) {
716 struct tog_view *v, *t;
717 /* Only allow one parent view per type. */
718 TAILQ_FOREACH_SAFE(v, &views, entry, t) {
719 if (v->type != new_view->type)
720 continue;
721 TAILQ_REMOVE(&views, v, entry);
722 err = view_close(v);
723 if (err)
724 goto done;
725 break;
727 TAILQ_INSERT_TAIL(&views, new_view, entry);
728 view = new_view;
729 if (focus_view == NULL)
730 focus_view = new_view;
732 if (focus_view) {
733 show_panel(focus_view->panel);
734 if (view)
735 view->focussed = 0;
736 focus_view->focussed = 1;
737 view = focus_view;
738 if (new_view)
739 show_panel(new_view->panel);
740 if (view->child && view_is_splitscreen(view->child))
741 show_panel(view->child->panel);
743 if (view) {
744 if (focus_view == NULL) {
745 view->focussed = 1;
746 show_panel(view->panel);
747 if (view->child && view_is_splitscreen(view->child))
748 show_panel(view->child->panel);
749 focus_view = view;
751 if (view->parent) {
752 err = view->parent->show(view->parent);
753 if (err)
754 goto done;
756 err = view->show(view);
757 if (err)
758 goto done;
759 if (view->child) {
760 err = view->child->show(view->child);
761 if (err)
762 goto done;
764 update_panels();
765 doupdate();
768 done:
769 while (!TAILQ_EMPTY(&views)) {
770 view = TAILQ_FIRST(&views);
771 TAILQ_REMOVE(&views, view, entry);
772 view_close(view);
775 errcode = pthread_mutex_unlock(&tog_mutex);
776 if (errcode)
777 return got_error_set_errno(errcode);
779 return err;
782 __dead static void
783 usage_log(void)
785 endwin();
786 fprintf(stderr,
787 "usage: %s log [-c commit] [-r repository-path] [path]\n",
788 getprogname());
789 exit(1);
792 /* Create newly allocated wide-character string equivalent to a byte string. */
793 static const struct got_error *
794 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
796 char *vis = NULL;
797 const struct got_error *err = NULL;
799 *ws = NULL;
800 *wlen = mbstowcs(NULL, s, 0);
801 if (*wlen == (size_t)-1) {
802 int vislen;
803 if (errno != EILSEQ)
804 return got_error_from_errno();
806 /* byte string invalid in current encoding; try to "fix" it */
807 err = got_mbsavis(&vis, &vislen, s);
808 if (err)
809 return err;
810 *wlen = mbstowcs(NULL, vis, 0);
811 if (*wlen == (size_t)-1) {
812 err = got_error_from_errno(); /* give up */
813 goto done;
817 *ws = calloc(*wlen + 1, sizeof(*ws));
818 if (*ws == NULL) {
819 err = got_error_from_errno();
820 goto done;
823 if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
824 err = got_error_from_errno();
825 done:
826 free(vis);
827 if (err) {
828 free(*ws);
829 *ws = NULL;
830 *wlen = 0;
832 return err;
835 /* Format a line for display, ensuring that it won't overflow a width limit. */
836 static const struct got_error *
837 format_line(wchar_t **wlinep, int *widthp, const char *line, int wlimit)
839 const struct got_error *err = NULL;
840 int cols = 0;
841 wchar_t *wline = NULL;
842 size_t wlen;
843 int i;
845 *wlinep = NULL;
846 *widthp = 0;
848 err = mbs2ws(&wline, &wlen, line);
849 if (err)
850 return err;
852 i = 0;
853 while (i < wlen && cols < wlimit) {
854 int width = wcwidth(wline[i]);
855 switch (width) {
856 case 0:
857 i++;
858 break;
859 case 1:
860 case 2:
861 if (cols + width <= wlimit)
862 cols += width;
863 i++;
864 break;
865 case -1:
866 if (wline[i] == L'\t')
867 cols += TABSIZE - ((cols + 1) % TABSIZE);
868 i++;
869 break;
870 default:
871 err = got_error_from_errno();
872 goto done;
875 wline[i] = L'\0';
876 if (widthp)
877 *widthp = cols;
878 done:
879 if (err)
880 free(wline);
881 else
882 *wlinep = wline;
883 return err;
886 static const struct got_error*
887 build_refs_str(char **refs_str, struct got_reflist_head *refs,
888 struct got_object_id *id)
890 static const struct got_error *err = NULL;
891 struct got_reflist_entry *re;
892 char *s;
893 const char *name;
895 *refs_str = NULL;
897 SIMPLEQ_FOREACH(re, refs, entry) {
898 if (got_object_id_cmp(re->id, id) != 0)
899 continue;
900 name = got_ref_get_name(re->ref);
901 if (strcmp(name, GOT_REF_HEAD) == 0)
902 continue;
903 if (strncmp(name, "refs/", 5) == 0)
904 name += 5;
905 if (strncmp(name, "heads/", 6) == 0)
906 name += 6;
907 if (strncmp(name, "remotes/", 8) == 0)
908 name += 8;
909 s = *refs_str;
910 if (asprintf(refs_str, "%s%s%s", s ? s : "",
911 s ? ", " : "", name) == -1) {
912 err = got_error_from_errno();
913 free(s);
914 *refs_str = NULL;
915 break;
917 free(s);
920 return err;
923 static const struct got_error *
924 format_author(wchar_t **wauthor, int *author_width, char *author, int limit)
926 char *smallerthan, *at;
928 smallerthan = strchr(author, '<');
929 if (smallerthan && smallerthan[1] != '\0')
930 author = smallerthan + 1;
931 at = strchr(author, '@');
932 if (at)
933 *at = '\0';
934 return format_line(wauthor, author_width, author, limit);
937 static const struct got_error *
938 draw_commit(struct tog_view *view, struct got_commit_object *commit,
939 struct got_object_id *id, struct got_reflist_head *refs,
940 int author_display_cols)
942 const struct got_error *err = NULL;
943 char datebuf[10]; /* YY-MM-DD + SPACE + NUL */
944 char *logmsg0 = NULL, *logmsg = NULL;
945 char *author = NULL;
946 wchar_t *wlogmsg = NULL, *wauthor = NULL;
947 int author_width, logmsg_width;
948 char *newline, *line = NULL;
949 int col, limit;
950 static const size_t date_display_cols = 9;
951 const int avail = view->ncols;
952 struct tm tm;
953 time_t committer_time;
955 committer_time = got_object_commit_get_committer_time(commit);
956 if (localtime_r(&committer_time, &tm) == NULL)
957 return got_error_from_errno();
958 if (strftime(datebuf, sizeof(datebuf), "%g/%m/%d ", &tm)
959 >= sizeof(datebuf))
960 return got_error(GOT_ERR_NO_SPACE);
962 if (avail < date_display_cols)
963 limit = MIN(sizeof(datebuf) - 1, avail);
964 else
965 limit = MIN(date_display_cols, sizeof(datebuf) - 1);
966 waddnstr(view->window, datebuf, limit);
967 col = limit + 1;
968 if (col > avail)
969 goto done;
971 author = strdup(got_object_commit_get_author(commit));
972 if (author == NULL) {
973 err = got_error_from_errno();
974 goto done;
976 err = format_author(&wauthor, &author_width, author, avail - col);
977 if (err)
978 goto done;
979 waddwstr(view->window, wauthor);
980 col += author_width;
981 while (col <= avail && author_width < author_display_cols + 2) {
982 waddch(view->window, ' ');
983 col++;
984 author_width++;
986 if (col > avail)
987 goto done;
989 logmsg0 = strdup(got_object_commit_get_logmsg(commit));
990 if (logmsg0 == NULL) {
991 err = got_error_from_errno();
992 goto done;
994 logmsg = logmsg0;
995 while (*logmsg == '\n')
996 logmsg++;
997 newline = strchr(logmsg, '\n');
998 if (newline)
999 *newline = '\0';
1000 limit = avail - col;
1001 err = format_line(&wlogmsg, &logmsg_width, logmsg, limit);
1002 if (err)
1003 goto done;
1004 waddwstr(view->window, wlogmsg);
1005 col += logmsg_width;
1006 while (col <= avail) {
1007 waddch(view->window, ' ');
1008 col++;
1010 done:
1011 free(logmsg0);
1012 free(wlogmsg);
1013 free(author);
1014 free(wauthor);
1015 free(line);
1016 return err;
1019 static struct commit_queue_entry *
1020 alloc_commit_queue_entry(struct got_commit_object *commit,
1021 struct got_object_id *id)
1023 struct commit_queue_entry *entry;
1025 entry = calloc(1, sizeof(*entry));
1026 if (entry == NULL)
1027 return NULL;
1029 entry->id = id;
1030 entry->commit = commit;
1031 return entry;
1034 static void
1035 pop_commit(struct commit_queue *commits)
1037 struct commit_queue_entry *entry;
1039 entry = TAILQ_FIRST(&commits->head);
1040 TAILQ_REMOVE(&commits->head, entry, entry);
1041 got_object_commit_close(entry->commit);
1042 commits->ncommits--;
1043 /* Don't free entry->id! It is owned by the commit graph. */
1044 free(entry);
1047 static void
1048 free_commits(struct commit_queue *commits)
1050 while (!TAILQ_EMPTY(&commits->head))
1051 pop_commit(commits);
1054 static const struct got_error *
1055 queue_commits(struct got_commit_graph *graph, struct commit_queue *commits,
1056 int minqueue, struct got_repository *repo, const char *path)
1058 const struct got_error *err = NULL;
1059 int nqueued = 0;
1062 * We keep all commits open throughout the lifetime of the log
1063 * view in order to avoid having to re-fetch commits from disk
1064 * while updating the display.
1066 while (nqueued < minqueue) {
1067 struct got_object_id *id;
1068 struct got_commit_object *commit;
1069 struct commit_queue_entry *entry;
1070 int errcode;
1072 err = got_commit_graph_iter_next(&id, graph);
1073 if (err) {
1074 if (err->code != GOT_ERR_ITER_NEED_MORE)
1075 break;
1076 err = got_commit_graph_fetch_commits(graph,
1077 minqueue, repo);
1078 if (err)
1079 return err;
1080 continue;
1083 if (id == NULL)
1084 break;
1086 err = got_object_open_as_commit(&commit, repo, id);
1087 if (err)
1088 break;
1089 entry = alloc_commit_queue_entry(commit, id);
1090 if (entry == NULL) {
1091 err = got_error_from_errno();
1092 break;
1095 errcode = pthread_mutex_lock(&tog_mutex);
1096 if (errcode) {
1097 err = got_error_set_errno(errcode);
1098 break;
1101 entry->idx = commits->ncommits;
1102 TAILQ_INSERT_TAIL(&commits->head, entry, entry);
1103 nqueued++;
1104 commits->ncommits++;
1106 errcode = pthread_mutex_unlock(&tog_mutex);
1107 if (errcode && err == NULL)
1108 err = got_error_set_errno(errcode);
1111 return err;
1114 static const struct got_error *
1115 get_head_commit_id(struct got_object_id **head_id, struct got_repository *repo)
1117 const struct got_error *err = NULL;
1118 struct got_reference *head_ref;
1120 *head_id = NULL;
1122 err = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1123 if (err)
1124 return err;
1126 err = got_ref_resolve(head_id, repo, head_ref);
1127 got_ref_close(head_ref);
1128 if (err) {
1129 *head_id = NULL;
1130 return err;
1133 return NULL;
1136 static const struct got_error *
1137 draw_commits(struct tog_view *view, struct commit_queue_entry **last,
1138 struct commit_queue_entry **selected, struct commit_queue_entry *first,
1139 struct commit_queue *commits, int selected_idx, int limit,
1140 struct got_reflist_head *refs, const char *path, int commits_needed)
1142 const struct got_error *err = NULL;
1143 struct commit_queue_entry *entry;
1144 int ncommits, width;
1145 int author_cols = 10;
1146 char *id_str = NULL, *header = NULL, *ncommits_str = NULL;
1147 char *refs_str = NULL;
1148 wchar_t *wline;
1150 entry = first;
1151 ncommits = 0;
1152 while (entry) {
1153 if (ncommits == selected_idx) {
1154 *selected = entry;
1155 break;
1157 entry = TAILQ_NEXT(entry, entry);
1158 ncommits++;
1161 if (*selected) {
1162 err = got_object_id_str(&id_str, (*selected)->id);
1163 if (err)
1164 return err;
1165 if (refs) {
1166 err = build_refs_str(&refs_str, refs, (*selected)->id);
1167 if (err)
1168 goto done;
1172 if (commits_needed == 0)
1173 halfdelay(10); /* disable fast refresh */
1175 if (asprintf(&ncommits_str, " [%d/%d] %s",
1176 entry ? entry->idx + 1 : 0, commits->ncommits,
1177 commits_needed > 0 ? "loading... " :
1178 (refs_str ? refs_str : "")) == -1) {
1179 err = got_error_from_errno();
1180 goto done;
1183 if (path && strcmp(path, "/") != 0) {
1184 if (asprintf(&header, "commit %s %s%s",
1185 id_str ? id_str : "........................................",
1186 path, ncommits_str) == -1) {
1187 err = got_error_from_errno();
1188 header = NULL;
1189 goto done;
1191 } else if (asprintf(&header, "commit %s%s",
1192 id_str ? id_str : "........................................",
1193 ncommits_str) == -1) {
1194 err = got_error_from_errno();
1195 header = NULL;
1196 goto done;
1198 err = format_line(&wline, &width, header, view->ncols);
1199 if (err)
1200 goto done;
1202 werase(view->window);
1204 if (view_needs_focus_indication(view))
1205 wstandout(view->window);
1206 waddwstr(view->window, wline);
1207 while (width < view->ncols) {
1208 waddch(view->window, ' ');
1209 width++;
1211 if (view_needs_focus_indication(view))
1212 wstandend(view->window);
1213 free(wline);
1214 if (limit <= 1)
1215 goto done;
1217 /* Grow author column size if necessary. */
1218 entry = first;
1219 ncommits = 0;
1220 while (entry) {
1221 char *author;
1222 wchar_t *wauthor;
1223 int width;
1224 if (ncommits >= limit - 1)
1225 break;
1226 author = strdup(got_object_commit_get_author(entry->commit));
1227 if (author == NULL) {
1228 err = got_error_from_errno();
1229 goto done;
1231 err = format_author(&wauthor, &width, author, COLS);
1232 if (author_cols < width)
1233 author_cols = width;
1234 free(wauthor);
1235 free(author);
1236 entry = TAILQ_NEXT(entry, entry);
1239 entry = first;
1240 *last = first;
1241 ncommits = 0;
1242 while (entry) {
1243 if (ncommits >= limit - 1)
1244 break;
1245 if (ncommits == selected_idx)
1246 wstandout(view->window);
1247 err = draw_commit(view, entry->commit, entry->id, refs,
1248 author_cols);
1249 if (ncommits == selected_idx)
1250 wstandend(view->window);
1251 if (err)
1252 break;
1253 ncommits++;
1254 *last = entry;
1255 entry = TAILQ_NEXT(entry, entry);
1258 view_vborder(view);
1259 done:
1260 free(id_str);
1261 free(refs_str);
1262 free(ncommits_str);
1263 free(header);
1264 return err;
1267 static void
1268 scroll_up(struct tog_view *view,
1269 struct commit_queue_entry **first_displayed_entry, int maxscroll,
1270 struct commit_queue *commits)
1272 struct commit_queue_entry *entry;
1273 int nscrolled = 0;
1275 entry = TAILQ_FIRST(&commits->head);
1276 if (*first_displayed_entry == entry) {
1277 view_flash(view);
1278 return;
1281 entry = *first_displayed_entry;
1282 while (entry && nscrolled < maxscroll) {
1283 entry = TAILQ_PREV(entry, commit_queue_head, entry);
1284 if (entry) {
1285 *first_displayed_entry = entry;
1286 nscrolled++;
1291 static const struct got_error *
1292 trigger_log_thread(int load_all, int *commits_needed, int *log_complete,
1293 pthread_cond_t *need_commits)
1295 int errcode;
1296 int max_wait = 20;
1298 halfdelay(1); /* fast refresh while loading commits */
1300 while (*commits_needed > 0) {
1301 if (*log_complete)
1302 break;
1304 /* Wake the log thread. */
1305 errcode = pthread_cond_signal(need_commits);
1306 if (errcode)
1307 return got_error_set_errno(errcode);
1308 errcode = pthread_mutex_unlock(&tog_mutex);
1309 if (errcode)
1310 return got_error_set_errno(errcode);
1311 pthread_yield();
1312 errcode = pthread_mutex_lock(&tog_mutex);
1313 if (errcode)
1314 return got_error_set_errno(errcode);
1316 if (*commits_needed > 0 && (!load_all || --max_wait <= 0)) {
1318 * Thread is not done yet; lose a key press
1319 * and let the user retry... this way the GUI
1320 * remains interactive while logging deep paths
1321 * with few commits in history.
1323 return NULL;
1327 return NULL;
1330 static const struct got_error *
1331 scroll_down(struct tog_view *view,
1332 struct commit_queue_entry **first_displayed_entry, int maxscroll,
1333 struct commit_queue_entry **last_displayed_entry,
1334 struct commit_queue *commits, int *log_complete, int *commits_needed,
1335 pthread_cond_t *need_commits)
1337 const struct got_error *err = NULL;
1338 struct commit_queue_entry *pentry;
1339 int nscrolled = 0;
1341 if (*last_displayed_entry == NULL)
1342 return NULL;
1344 pentry = TAILQ_NEXT(*last_displayed_entry, entry);
1345 if (pentry == NULL && !*log_complete) {
1347 * Ask the log thread for required amount of commits
1348 * plus some amount of pre-fetching.
1350 (*commits_needed) += maxscroll + 20;
1351 err = trigger_log_thread(0, commits_needed, log_complete,
1352 need_commits);
1353 if (err)
1354 return err;
1357 do {
1358 pentry = TAILQ_NEXT(*last_displayed_entry, entry);
1359 if (pentry == NULL) {
1360 if (*log_complete)
1361 view_flash(view);
1362 break;
1365 *last_displayed_entry = pentry;
1367 pentry = TAILQ_NEXT(*first_displayed_entry, entry);
1368 if (pentry == NULL)
1369 break;
1370 *first_displayed_entry = pentry;
1371 } while (++nscrolled < maxscroll);
1373 return err;
1376 static const struct got_error *
1377 open_diff_view_for_commit(struct tog_view **new_view, int begin_x,
1378 struct got_commit_object *commit, struct got_object_id *commit_id,
1379 struct tog_view *log_view, struct got_reflist_head *refs,
1380 struct got_repository *repo)
1382 const struct got_error *err;
1383 struct got_object_qid *parent_id;
1384 struct tog_view *diff_view;
1386 diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
1387 if (diff_view == NULL)
1388 return got_error_from_errno();
1390 parent_id = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
1391 err = open_diff_view(diff_view, parent_id ? parent_id->id : NULL,
1392 commit_id, log_view, refs, repo);
1393 if (err == NULL)
1394 *new_view = diff_view;
1395 return err;
1398 static const struct got_error *
1399 browse_commit(struct tog_view **new_view, int begin_x,
1400 struct commit_queue_entry *entry, struct got_reflist_head *refs,
1401 struct got_repository *repo)
1403 const struct got_error *err = NULL;
1404 struct got_tree_object *tree;
1405 struct tog_view *tree_view;
1407 err = got_object_open_as_tree(&tree, repo,
1408 got_object_commit_get_tree_id(entry->commit));
1409 if (err)
1410 return err;
1412 tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
1413 if (tree_view == NULL)
1414 return got_error_from_errno();
1416 err = open_tree_view(tree_view, tree, entry->id, refs, repo);
1417 if (err)
1418 got_object_tree_close(tree);
1419 else
1420 *new_view = tree_view;
1421 return err;
1424 static void *
1425 log_thread(void *arg)
1427 const struct got_error *err = NULL;
1428 int errcode = 0;
1429 struct tog_log_thread_args *a = arg;
1430 int done = 0;
1432 err = got_commit_graph_iter_start(a->graph, a->start_id, a->repo);
1433 if (err)
1434 return (void *)err;
1436 while (!done && !err) {
1437 err = queue_commits(a->graph, a->commits, 1, a->repo,
1438 a->in_repo_path);
1439 if (err) {
1440 if (err->code != GOT_ERR_ITER_COMPLETED)
1441 return (void *)err;
1442 err = NULL;
1443 done = 1;
1444 } else if (a->commits_needed > 0)
1445 a->commits_needed--;
1447 errcode = pthread_mutex_lock(&tog_mutex);
1448 if (errcode)
1449 return (void *)got_error_set_errno(errcode);
1451 if (done)
1452 a->log_complete = 1;
1453 else if (*a->quit) {
1454 done = 1;
1455 a->log_complete = 1;
1456 } else if (*a->first_displayed_entry == NULL) {
1457 *a->first_displayed_entry =
1458 TAILQ_FIRST(&a->commits->head);
1459 *a->selected_entry = *a->first_displayed_entry;
1462 if (done)
1463 a->commits_needed = 0;
1464 else if (a->commits_needed == 0) {
1465 errcode = pthread_cond_wait(&a->need_commits,
1466 &tog_mutex);
1467 if (errcode)
1468 err = got_error_set_errno(errcode);
1471 errcode = pthread_mutex_unlock(&tog_mutex);
1472 if (errcode && err == NULL)
1473 err = got_error_set_errno(errcode);
1475 return (void *)err;
1478 static const struct got_error *
1479 stop_log_thread(struct tog_log_view_state *s)
1481 const struct got_error *err = NULL;
1482 int errcode;
1484 if (s->thread) {
1485 s->quit = 1;
1486 errcode = pthread_cond_signal(&s->thread_args.need_commits);
1487 if (errcode)
1488 return got_error_set_errno(errcode);
1489 errcode = pthread_mutex_unlock(&tog_mutex);
1490 if (errcode)
1491 return got_error_set_errno(errcode);
1492 errcode = pthread_join(s->thread, (void **)&err);
1493 if (errcode)
1494 return got_error_set_errno(errcode);
1495 errcode = pthread_mutex_lock(&tog_mutex);
1496 if (errcode)
1497 return got_error_set_errno(errcode);
1498 s->thread = NULL;
1501 errcode = pthread_cond_destroy(&s->thread_args.need_commits);
1502 if (errcode && err == NULL)
1503 err = got_error_set_errno(errcode);
1505 if (s->thread_args.repo) {
1506 got_repo_close(s->thread_args.repo);
1507 s->thread_args.repo = NULL;
1510 if (s->thread_args.graph) {
1511 got_commit_graph_close(s->thread_args.graph);
1512 s->thread_args.graph = NULL;
1515 return err;
1518 static const struct got_error *
1519 close_log_view(struct tog_view *view)
1521 const struct got_error *err = NULL;
1522 struct tog_log_view_state *s = &view->state.log;
1524 err = stop_log_thread(s);
1525 free_commits(&s->commits);
1526 free(s->in_repo_path);
1527 s->in_repo_path = NULL;
1528 free(s->start_id);
1529 s->start_id = NULL;
1530 return err;
1533 static const struct got_error *
1534 open_log_view(struct tog_view *view, struct got_object_id *start_id,
1535 struct got_reflist_head *refs, struct got_repository *repo,
1536 const char *path, int check_disk)
1538 const struct got_error *err = NULL;
1539 struct tog_log_view_state *s = &view->state.log;
1540 struct got_repository *thread_repo = NULL;
1541 struct got_commit_graph *thread_graph = NULL;
1542 int errcode;
1544 err = got_repo_map_path(&s->in_repo_path, repo, path, check_disk);
1545 if (err != NULL)
1546 goto done;
1548 /* The commit queue only contains commits being displayed. */
1549 TAILQ_INIT(&s->commits.head);
1550 s->commits.ncommits = 0;
1552 s->refs = refs;
1553 s->repo = repo;
1554 s->start_id = got_object_id_dup(start_id);
1555 if (s->start_id == NULL) {
1556 err = got_error_from_errno();
1557 goto done;
1560 view->show = show_log_view;
1561 view->input = input_log_view;
1562 view->close = close_log_view;
1564 err = got_repo_open(&thread_repo, got_repo_get_path(repo));
1565 if (err)
1566 goto done;
1567 err = got_commit_graph_open(&thread_graph, start_id, s->in_repo_path,
1568 0, thread_repo);
1569 if (err)
1570 goto done;
1572 errcode = pthread_cond_init(&s->thread_args.need_commits, NULL);
1573 if (errcode) {
1574 err = got_error_set_errno(errcode);
1575 goto done;
1578 s->thread_args.commits_needed = view->nlines;
1579 s->thread_args.graph = thread_graph;
1580 s->thread_args.commits = &s->commits;
1581 s->thread_args.in_repo_path = s->in_repo_path;
1582 s->thread_args.start_id = s->start_id;
1583 s->thread_args.repo = thread_repo;
1584 s->thread_args.log_complete = 0;
1585 s->thread_args.quit = &s->quit;
1586 s->thread_args.view = view;
1587 s->thread_args.first_displayed_entry = &s->first_displayed_entry;
1588 s->thread_args.selected_entry = &s->selected_entry;
1589 done:
1590 if (err)
1591 close_log_view(view);
1592 return err;
1595 static const struct got_error *
1596 show_log_view(struct tog_view *view)
1598 struct tog_log_view_state *s = &view->state.log;
1600 if (s->thread == NULL) {
1601 int errcode = pthread_create(&s->thread, NULL, log_thread,
1602 &s->thread_args);
1603 if (errcode)
1604 return got_error_set_errno(errcode);
1607 return draw_commits(view, &s->last_displayed_entry,
1608 &s->selected_entry, s->first_displayed_entry,
1609 &s->commits, s->selected, view->nlines, s->refs,
1610 s->in_repo_path, s->thread_args.commits_needed);
1613 static const struct got_error *
1614 input_log_view(struct tog_view **new_view, struct tog_view **dead_view,
1615 struct tog_view **focus_view, struct tog_view *view, int ch)
1617 const struct got_error *err = NULL;
1618 struct tog_log_view_state *s = &view->state.log;
1619 char *parent_path;
1620 struct tog_view *diff_view = NULL, *tree_view = NULL;
1621 int begin_x = 0;
1623 switch (ch) {
1624 case 'q':
1625 s->quit = 1;
1626 break;
1627 case 'k':
1628 case KEY_UP:
1629 case '<':
1630 case ',':
1631 if (s->first_displayed_entry == NULL)
1632 break;
1633 if (s->selected > 0)
1634 s->selected--;
1635 if (s->selected > 0)
1636 break;
1637 scroll_up(view, &s->first_displayed_entry, 1,
1638 &s->commits);
1639 break;
1640 case KEY_PPAGE:
1641 if (s->first_displayed_entry == NULL)
1642 break;
1643 if (TAILQ_FIRST(&s->commits.head) ==
1644 s->first_displayed_entry) {
1645 if (s->selected == 0) {
1646 view_flash(view);
1647 break;
1649 s->selected = 0;
1650 break;
1652 scroll_up(view, &s->first_displayed_entry,
1653 view->nlines, &s->commits);
1654 break;
1655 case 'j':
1656 case KEY_DOWN:
1657 case '>':
1658 case '.':
1659 if (s->first_displayed_entry == NULL)
1660 break;
1661 if (s->selected < MIN(view->nlines - 2,
1662 s->commits.ncommits - 1)) {
1663 s->selected++;
1664 break;
1666 err = scroll_down(view, &s->first_displayed_entry, 1,
1667 &s->last_displayed_entry, &s->commits,
1668 &s->thread_args.log_complete,
1669 &s->thread_args.commits_needed,
1670 &s->thread_args.need_commits);
1671 break;
1672 case KEY_NPAGE: {
1673 struct commit_queue_entry *first;
1674 first = s->first_displayed_entry;
1675 if (first == NULL)
1676 break;
1677 err = scroll_down(view, &s->first_displayed_entry,
1678 view->nlines, &s->last_displayed_entry,
1679 &s->commits, &s->thread_args.log_complete,
1680 &s->thread_args.commits_needed,
1681 &s->thread_args.need_commits);
1682 if (first == s->first_displayed_entry &&
1683 s->selected < MIN(view->nlines - 2,
1684 s->commits.ncommits - 1)) {
1685 /* can't scroll further down */
1686 s->selected = MIN(view->nlines - 2,
1687 s->commits.ncommits - 1);
1689 err = NULL;
1690 break;
1692 case KEY_RESIZE:
1693 if (s->selected > view->nlines - 2)
1694 s->selected = view->nlines - 2;
1695 if (s->selected > s->commits.ncommits - 1)
1696 s->selected = s->commits.ncommits - 1;
1697 break;
1698 case KEY_ENTER:
1699 case '\r':
1700 if (s->selected_entry == NULL)
1701 break;
1702 if (view_is_parent_view(view))
1703 begin_x = view_split_begin_x(view->begin_x);
1704 err = open_diff_view_for_commit(&diff_view, begin_x,
1705 s->selected_entry->commit, s->selected_entry->id,
1706 view, s->refs, s->repo);
1707 if (err)
1708 break;
1709 if (view_is_parent_view(view)) {
1710 err = view_close_child(view);
1711 if (err)
1712 return err;
1713 err = view_set_child(view, diff_view);
1714 if (err) {
1715 view_close(diff_view);
1716 break;
1718 *focus_view = diff_view;
1719 view->child_focussed = 1;
1720 } else
1721 *new_view = diff_view;
1722 break;
1723 case 't':
1724 if (s->selected_entry == NULL)
1725 break;
1726 if (view_is_parent_view(view))
1727 begin_x = view_split_begin_x(view->begin_x);
1728 err = browse_commit(&tree_view, begin_x,
1729 s->selected_entry, s->refs, s->repo);
1730 if (err)
1731 break;
1732 if (view_is_parent_view(view)) {
1733 err = view_close_child(view);
1734 if (err)
1735 return err;
1736 err = view_set_child(view, tree_view);
1737 if (err) {
1738 view_close(tree_view);
1739 break;
1741 *focus_view = tree_view;
1742 view->child_focussed = 1;
1743 } else
1744 *new_view = tree_view;
1745 break;
1746 case KEY_BACKSPACE:
1747 if (strcmp(s->in_repo_path, "/") == 0)
1748 break;
1749 parent_path = dirname(s->in_repo_path);
1750 if (parent_path && strcmp(parent_path, ".") != 0) {
1751 struct tog_view *lv;
1752 err = stop_log_thread(s);
1753 if (err)
1754 return err;
1755 lv = view_open(view->nlines, view->ncols,
1756 view->begin_y, view->begin_x, TOG_VIEW_LOG);
1757 if (lv == NULL)
1758 return got_error_from_errno();
1759 err = open_log_view(lv, s->start_id, s->refs,
1760 s->repo, parent_path, 0);
1761 if (err)
1762 return err;;
1763 if (view_is_parent_view(view))
1764 *new_view = lv;
1765 else {
1766 view_set_child(view->parent, lv);
1767 *focus_view = lv;
1769 return NULL;
1771 break;
1772 default:
1773 break;
1776 return err;
1779 static const struct got_error *
1780 apply_unveil(const char *repo_path, const char *worktree_path)
1782 const struct got_error *error;
1784 if (repo_path && unveil(repo_path, "r") != 0)
1785 return got_error_from_errno();
1787 if (worktree_path && unveil(worktree_path, "rwc") != 0)
1788 return got_error_from_errno();
1790 if (unveil("/tmp", "rwc") != 0)
1791 return got_error_from_errno();
1793 error = got_privsep_unveil_exec_helpers();
1794 if (error != NULL)
1795 return error;
1797 if (unveil(NULL, NULL) != 0)
1798 return got_error_from_errno();
1800 return NULL;
1803 static void
1804 init_curses(void)
1806 initscr();
1807 cbreak();
1808 halfdelay(1); /* Do fast refresh while initial view is loading. */
1809 noecho();
1810 nonl();
1811 intrflush(stdscr, FALSE);
1812 keypad(stdscr, TRUE);
1813 curs_set(0);
1814 signal(SIGWINCH, tog_sigwinch);
1817 static const struct got_error *
1818 cmd_log(int argc, char *argv[])
1820 const struct got_error *error;
1821 struct got_repository *repo = NULL;
1822 struct got_worktree *worktree = NULL;
1823 struct got_reflist_head refs;
1824 struct got_object_id *start_id = NULL;
1825 char *path = NULL, *repo_path = NULL, *cwd = NULL;
1826 char *start_commit = NULL;
1827 int ch;
1828 struct tog_view *view;
1830 #ifndef PROFILE
1831 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
1832 NULL) == -1)
1833 err(1, "pledge");
1834 #endif
1836 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
1837 switch (ch) {
1838 case 'c':
1839 start_commit = optarg;
1840 break;
1841 case 'r':
1842 repo_path = realpath(optarg, NULL);
1843 if (repo_path == NULL)
1844 err(1, "-r option");
1845 break;
1846 default:
1847 usage_log();
1848 /* NOTREACHED */
1852 argc -= optind;
1853 argv += optind;
1855 if (argc == 0)
1856 path = strdup("");
1857 else if (argc == 1)
1858 path = strdup(argv[0]);
1859 else
1860 usage_log();
1861 if (path == NULL)
1862 return got_error_from_errno();
1864 cwd = getcwd(NULL, 0);
1865 if (cwd == NULL) {
1866 error = got_error_from_errno();
1867 goto done;
1869 if (repo_path == NULL) {
1870 error = got_worktree_open(&worktree, cwd);
1871 if (error && error->code != GOT_ERR_NOT_WORKTREE)
1872 goto done;
1873 if (worktree) {
1874 repo_path =
1875 strdup(got_worktree_get_repo_path(worktree));
1876 } else
1877 repo_path = strdup(cwd);
1878 if (repo_path == NULL) {
1879 error = got_error_from_errno();
1880 goto done;
1884 init_curses();
1886 error = apply_unveil(repo_path,
1887 worktree ? got_worktree_get_root_path(worktree) : NULL);
1888 if (error)
1889 goto done;
1891 error = got_repo_open(&repo, repo_path);
1892 if (error != NULL)
1893 goto done;
1895 if (start_commit == NULL)
1896 error = get_head_commit_id(&start_id, repo);
1897 else
1898 error = got_object_resolve_id_str(&start_id, repo,
1899 start_commit);
1900 if (error != NULL)
1901 goto done;
1903 SIMPLEQ_INIT(&refs);
1904 error = got_ref_list(&refs, repo);
1905 if (error)
1906 goto done;
1908 view = view_open(0, 0, 0, 0, TOG_VIEW_LOG);
1909 if (view == NULL) {
1910 error = got_error_from_errno();
1911 goto done;
1913 error = open_log_view(view, start_id, &refs, repo, path, 1);
1914 if (error)
1915 goto done;
1916 error = view_loop(view);
1917 done:
1918 free(repo_path);
1919 free(cwd);
1920 free(path);
1921 free(start_id);
1922 if (repo)
1923 got_repo_close(repo);
1924 if (worktree)
1925 got_worktree_close(worktree);
1926 return error;
1929 __dead static void
1930 usage_diff(void)
1932 endwin();
1933 fprintf(stderr, "usage: %s diff [repository-path] object1 object2\n",
1934 getprogname());
1935 exit(1);
1938 static char *
1939 parse_next_line(FILE *f, size_t *len)
1941 char *line;
1942 size_t linelen;
1943 size_t lineno;
1944 const char delim[3] = { '\0', '\0', '\0'};
1946 line = fparseln(f, &linelen, &lineno, delim, 0);
1947 if (len)
1948 *len = linelen;
1949 return line;
1952 static const struct got_error *
1953 draw_file(struct tog_view *view, FILE *f, int *first_displayed_line,
1954 int *last_displayed_line, int *eof, int max_lines,
1955 char * header)
1957 const struct got_error *err;
1958 int nlines = 0, nprinted = 0;
1959 char *line;
1960 size_t len;
1961 wchar_t *wline;
1962 int width;
1964 rewind(f);
1965 werase(view->window);
1967 if (header) {
1968 err = format_line(&wline, &width, header, view->ncols);
1969 if (err) {
1970 return err;
1973 if (view_needs_focus_indication(view))
1974 wstandout(view->window);
1975 waddwstr(view->window, wline);
1976 if (view_needs_focus_indication(view))
1977 wstandend(view->window);
1978 if (width < view->ncols)
1979 waddch(view->window, '\n');
1981 if (max_lines <= 1)
1982 return NULL;
1983 max_lines--;
1986 *eof = 0;
1987 while (nprinted < max_lines) {
1988 line = parse_next_line(f, &len);
1989 if (line == NULL) {
1990 *eof = 1;
1991 break;
1993 if (++nlines < *first_displayed_line) {
1994 free(line);
1995 continue;
1998 err = format_line(&wline, &width, line, view->ncols);
1999 if (err) {
2000 free(line);
2001 return err;
2003 waddwstr(view->window, wline);
2004 if (width < view->ncols)
2005 waddch(view->window, '\n');
2006 if (++nprinted == 1)
2007 *first_displayed_line = nlines;
2008 free(line);
2009 free(wline);
2010 wline = NULL;
2012 *last_displayed_line = nlines;
2014 view_vborder(view);
2016 return NULL;
2019 static char *
2020 get_datestr(time_t *time, char *datebuf)
2022 char *p, *s = ctime_r(time, datebuf);
2023 p = strchr(s, '\n');
2024 if (p)
2025 *p = '\0';
2026 return s;
2029 static const struct got_error *
2030 write_commit_info(struct got_object_id *commit_id,
2031 struct got_reflist_head *refs, struct got_repository *repo, FILE *outfile)
2033 const struct got_error *err = NULL;
2034 char datebuf[26];
2035 struct got_commit_object *commit;
2036 char *id_str = NULL;
2037 time_t committer_time;
2038 const char *author, *committer;
2039 char *refs_str = NULL;
2041 if (refs) {
2042 err = build_refs_str(&refs_str, refs, commit_id);
2043 if (err)
2044 return err;
2047 err = got_object_open_as_commit(&commit, repo, commit_id);
2048 if (err)
2049 return err;
2051 err = got_object_id_str(&id_str, commit_id);
2052 if (err) {
2053 err = got_error_from_errno();
2054 goto done;
2057 if (fprintf(outfile, "commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
2058 refs_str ? refs_str : "", refs_str ? ")" : "") < 0) {
2059 err = got_error_from_errno();
2060 goto done;
2062 if (fprintf(outfile, "from: %s\n",
2063 got_object_commit_get_author(commit)) < 0) {
2064 err = got_error_from_errno();
2065 goto done;
2067 committer_time = got_object_commit_get_committer_time(commit);
2068 if (fprintf(outfile, "date: %s UTC\n",
2069 get_datestr(&committer_time, datebuf)) < 0) {
2070 err = got_error_from_errno();
2071 goto done;
2073 author = got_object_commit_get_author(commit);
2074 committer = got_object_commit_get_committer(commit);
2075 if (strcmp(author, committer) != 0 &&
2076 fprintf(outfile, "via: %s\n", committer) < 0) {
2077 err = got_error_from_errno();
2078 goto done;
2080 if (fprintf(outfile, "%s\n",
2081 got_object_commit_get_logmsg(commit)) < 0) {
2082 err = got_error_from_errno();
2083 goto done;
2085 done:
2086 free(id_str);
2087 free(refs_str);
2088 got_object_commit_close(commit);
2089 return err;
2092 static const struct got_error *
2093 create_diff(struct tog_diff_view_state *s)
2095 const struct got_error *err = NULL;
2096 FILE *f = NULL;
2097 int obj_type;
2099 f = got_opentemp();
2100 if (f == NULL) {
2101 err = got_error_from_errno();
2102 goto done;
2104 if (s->f && fclose(s->f) != 0) {
2105 err = got_error_from_errno();
2106 goto done;
2108 s->f = f;
2110 if (s->id1)
2111 err = got_object_get_type(&obj_type, s->repo, s->id1);
2112 else
2113 err = got_object_get_type(&obj_type, s->repo, s->id2);
2114 if (err)
2115 goto done;
2117 switch (obj_type) {
2118 case GOT_OBJ_TYPE_BLOB:
2119 err = got_diff_objects_as_blobs(s->id1, s->id2, NULL, NULL,
2120 s->diff_context, s->repo, f);
2121 break;
2122 case GOT_OBJ_TYPE_TREE:
2123 err = got_diff_objects_as_trees(s->id1, s->id2, "", "",
2124 s->diff_context, s->repo, f);
2125 break;
2126 case GOT_OBJ_TYPE_COMMIT: {
2127 const struct got_object_id_queue *parent_ids;
2128 struct got_object_qid *pid;
2129 struct got_commit_object *commit2;
2131 err = got_object_open_as_commit(&commit2, s->repo, s->id2);
2132 if (err)
2133 break;
2134 /* Show commit info if we're diffing to a parent/root commit. */
2135 if (s->id1 == NULL)
2136 write_commit_info(s->id2, s->refs, s->repo, f);
2137 else {
2138 parent_ids = got_object_commit_get_parent_ids(commit2);
2139 SIMPLEQ_FOREACH(pid, parent_ids, entry) {
2140 if (got_object_id_cmp(s->id1, pid->id) == 0) {
2141 write_commit_info(s->id2, s->refs,
2142 s->repo, f);
2143 break;
2147 got_object_commit_close(commit2);
2149 err = got_diff_objects_as_commits(s->id1, s->id2,
2150 s->diff_context, s->repo, f);
2151 break;
2153 default:
2154 err = got_error(GOT_ERR_OBJ_TYPE);
2155 break;
2157 done:
2158 if (f && fflush(f) != 0 && err == NULL)
2159 err = got_error_from_errno();
2160 return err;
2163 static void
2164 diff_view_indicate_progress(struct tog_view *view)
2166 werase(view->window);
2167 waddstr(view->window, "diffing...");
2168 update_panels();
2169 doupdate();
2172 static const struct got_error *
2173 open_diff_view(struct tog_view *view, struct got_object_id *id1,
2174 struct got_object_id *id2, struct tog_view *log_view,
2175 struct got_reflist_head *refs, struct got_repository *repo)
2177 const struct got_error *err;
2179 if (id1 != NULL && id2 != NULL) {
2180 int type1, type2;
2181 err = got_object_get_type(&type1, repo, id1);
2182 if (err)
2183 return err;
2184 err = got_object_get_type(&type2, repo, id2);
2185 if (err)
2186 return err;
2188 if (type1 != type2)
2189 return got_error(GOT_ERR_OBJ_TYPE);
2192 if (id1) {
2193 view->state.diff.id1 = got_object_id_dup(id1);
2194 if (view->state.diff.id1 == NULL)
2195 return got_error_from_errno();
2196 } else
2197 view->state.diff.id1 = NULL;
2199 view->state.diff.id2 = got_object_id_dup(id2);
2200 if (view->state.diff.id2 == NULL) {
2201 free(view->state.diff.id1);
2202 view->state.diff.id1 = NULL;
2203 return got_error_from_errno();
2205 view->state.diff.f = NULL;
2206 view->state.diff.first_displayed_line = 1;
2207 view->state.diff.last_displayed_line = view->nlines;
2208 view->state.diff.diff_context = 3;
2209 view->state.diff.log_view = log_view;
2210 view->state.diff.repo = repo;
2211 view->state.diff.refs = refs;
2213 if (log_view && view_is_splitscreen(view))
2214 show_log_view(log_view); /* draw vborder */
2215 diff_view_indicate_progress(view);
2217 err = create_diff(&view->state.diff);
2218 if (err) {
2219 free(view->state.diff.id1);
2220 view->state.diff.id1 = NULL;
2221 free(view->state.diff.id2);
2222 view->state.diff.id2 = NULL;
2223 return err;
2226 view->show = show_diff_view;
2227 view->input = input_diff_view;
2228 view->close = close_diff_view;
2230 return NULL;
2233 static const struct got_error *
2234 close_diff_view(struct tog_view *view)
2236 const struct got_error *err = NULL;
2238 free(view->state.diff.id1);
2239 view->state.diff.id1 = NULL;
2240 free(view->state.diff.id2);
2241 view->state.diff.id2 = NULL;
2242 if (view->state.diff.f && fclose(view->state.diff.f) == EOF)
2243 err = got_error_from_errno();
2244 return err;
2247 static const struct got_error *
2248 show_diff_view(struct tog_view *view)
2250 const struct got_error *err;
2251 struct tog_diff_view_state *s = &view->state.diff;
2252 char *id_str1 = NULL, *id_str2, *header;
2254 if (s->id1) {
2255 err = got_object_id_str(&id_str1, s->id1);
2256 if (err)
2257 return err;
2259 err = got_object_id_str(&id_str2, s->id2);
2260 if (err)
2261 return err;
2263 if (asprintf(&header, "diff %s %s",
2264 id_str1 ? id_str1 : "/dev/null", id_str2) == -1) {
2265 err = got_error_from_errno();
2266 free(id_str1);
2267 free(id_str2);
2268 return err;
2270 free(id_str1);
2271 free(id_str2);
2273 return draw_file(view, s->f, &s->first_displayed_line,
2274 &s->last_displayed_line, &s->eof, view->nlines,
2275 header);
2278 static const struct got_error *
2279 set_selected_commit(struct tog_diff_view_state *s,
2280 struct commit_queue_entry *entry)
2282 const struct got_error *err;
2283 const struct got_object_id_queue *parent_ids;
2284 struct got_commit_object *selected_commit;
2285 struct got_object_qid *pid;
2287 free(s->id2);
2288 s->id2 = got_object_id_dup(entry->id);
2289 if (s->id2 == NULL)
2290 return got_error_from_errno();
2292 err = got_object_open_as_commit(&selected_commit, s->repo, entry->id);
2293 if (err)
2294 return err;
2295 parent_ids = got_object_commit_get_parent_ids(selected_commit);
2296 free(s->id1);
2297 pid = SIMPLEQ_FIRST(parent_ids);
2298 s->id1 = pid ? got_object_id_dup(pid->id) : NULL;
2299 got_object_commit_close(selected_commit);
2300 return NULL;
2303 static const struct got_error *
2304 input_diff_view(struct tog_view **new_view, struct tog_view **dead_view,
2305 struct tog_view **focus_view, struct tog_view *view, int ch)
2307 const struct got_error *err = NULL;
2308 struct tog_diff_view_state *s = &view->state.diff;
2309 struct tog_log_view_state *ls;
2310 struct commit_queue_entry *entry;
2311 int i;
2313 switch (ch) {
2314 case 'k':
2315 case KEY_UP:
2316 if (s->first_displayed_line > 1)
2317 s->first_displayed_line--;
2318 else
2319 view_flash(view);
2320 break;
2321 case KEY_PPAGE:
2322 if (s->first_displayed_line == 1) {
2323 view_flash(view);
2324 break;
2326 i = 0;
2327 while (i++ < view->nlines - 1 &&
2328 s->first_displayed_line > 1)
2329 s->first_displayed_line--;
2330 break;
2331 case 'j':
2332 case KEY_DOWN:
2333 if (!s->eof)
2334 s->first_displayed_line++;
2335 else
2336 view_flash(view);
2337 break;
2338 case KEY_NPAGE:
2339 case ' ':
2340 if (s->eof) {
2341 view_flash(view);
2342 break;
2344 i = 0;
2345 while (!s->eof && i++ < view->nlines - 1) {
2346 char *line;
2347 line = parse_next_line(s->f, NULL);
2348 s->first_displayed_line++;
2349 if (line == NULL)
2350 break;
2352 break;
2353 case '[':
2354 if (s->diff_context > 0) {
2355 s->diff_context--;
2356 diff_view_indicate_progress(view);
2357 err = create_diff(s);
2359 break;
2360 case ']':
2361 if (s->diff_context < GOT_DIFF_MAX_CONTEXT) {
2362 s->diff_context++;
2363 diff_view_indicate_progress(view);
2364 err = create_diff(s);
2366 break;
2367 case '<':
2368 case ',':
2369 if (s->log_view == NULL)
2370 break;
2371 ls = &s->log_view->state.log;
2372 entry = TAILQ_PREV(ls->selected_entry,
2373 commit_queue_head, entry);
2374 if (entry == NULL)
2375 break;
2377 err = input_log_view(NULL, NULL, NULL, s->log_view,
2378 KEY_UP);
2379 if (err)
2380 break;
2382 err = set_selected_commit(s, entry);
2383 if (err)
2384 break;
2386 s->first_displayed_line = 1;
2387 s->last_displayed_line = view->nlines;
2389 diff_view_indicate_progress(view);
2390 err = create_diff(s);
2391 break;
2392 case '>':
2393 case '.':
2394 if (s->log_view == NULL)
2395 break;
2396 ls = &s->log_view->state.log;
2398 if (TAILQ_NEXT(ls->selected_entry, entry) == NULL) {
2399 ls->thread_args.commits_needed++;
2401 /* Display "loading..." in log view. */
2402 show_log_view(s->log_view);
2403 update_panels();
2404 doupdate();
2406 err = trigger_log_thread(1 /* load_all */,
2407 &ls->thread_args.commits_needed,
2408 &ls->thread_args.log_complete,
2409 &ls->thread_args.need_commits);
2410 if (err)
2411 break;
2413 err = input_log_view(NULL, NULL, NULL, s->log_view,
2414 KEY_DOWN);
2415 if (err)
2416 break;
2418 entry = TAILQ_NEXT(ls->selected_entry, entry);
2419 if (entry == NULL)
2420 break;
2422 err = set_selected_commit(s, entry);
2423 if (err)
2424 break;
2426 s->first_displayed_line = 1;
2427 s->last_displayed_line = view->nlines;
2429 diff_view_indicate_progress(view);
2430 err = create_diff(s);
2431 break;
2432 default:
2433 break;
2436 return err;
2439 static const struct got_error *
2440 cmd_diff(int argc, char *argv[])
2442 const struct got_error *error = NULL;
2443 struct got_repository *repo = NULL;
2444 struct got_reflist_head refs;
2445 struct got_object_id *id1 = NULL, *id2 = NULL;
2446 char *repo_path = NULL;
2447 char *id_str1 = NULL, *id_str2 = NULL;
2448 int ch;
2449 struct tog_view *view;
2451 #ifndef PROFILE
2452 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
2453 NULL) == -1)
2454 err(1, "pledge");
2455 #endif
2457 while ((ch = getopt(argc, argv, "")) != -1) {
2458 switch (ch) {
2459 default:
2460 usage_diff();
2461 /* NOTREACHED */
2465 argc -= optind;
2466 argv += optind;
2468 if (argc == 0) {
2469 usage_diff(); /* TODO show local worktree changes */
2470 } else if (argc == 2) {
2471 repo_path = getcwd(NULL, 0);
2472 if (repo_path == NULL)
2473 return got_error_from_errno();
2474 id_str1 = argv[0];
2475 id_str2 = argv[1];
2476 } else if (argc == 3) {
2477 repo_path = realpath(argv[0], NULL);
2478 if (repo_path == NULL)
2479 return got_error_from_errno();
2480 id_str1 = argv[1];
2481 id_str2 = argv[2];
2482 } else
2483 usage_diff();
2485 init_curses();
2487 error = apply_unveil(repo_path, NULL);
2488 if (error)
2489 goto done;
2491 error = got_repo_open(&repo, repo_path);
2492 free(repo_path);
2493 if (error)
2494 goto done;
2496 error = got_object_resolve_id_str(&id1, repo, id_str1);
2497 if (error)
2498 goto done;
2500 error = got_object_resolve_id_str(&id2, repo, id_str2);
2501 if (error)
2502 goto done;
2504 SIMPLEQ_INIT(&refs);
2505 error = got_ref_list(&refs, repo);
2506 if (error)
2507 goto done;
2509 view = view_open(0, 0, 0, 0, TOG_VIEW_DIFF);
2510 if (view == NULL) {
2511 error = got_error_from_errno();
2512 goto done;
2514 error = open_diff_view(view, id1, id2, NULL, &refs, repo);
2515 if (error)
2516 goto done;
2517 error = view_loop(view);
2518 done:
2519 got_repo_close(repo);
2520 return error;
2523 __dead static void
2524 usage_blame(void)
2526 endwin();
2527 fprintf(stderr, "usage: %s blame [-c commit] [-r repository-path] path\n",
2528 getprogname());
2529 exit(1);
2532 struct tog_blame_line {
2533 int annotated;
2534 struct got_object_id *id;
2537 static const struct got_error *
2538 draw_blame(struct tog_view *view, struct got_object_id *id, FILE *f,
2539 const char *path, struct tog_blame_line *lines, int nlines,
2540 int blame_complete, int selected_line, int *first_displayed_line,
2541 int *last_displayed_line, int *eof, int max_lines)
2543 const struct got_error *err;
2544 int lineno = 0, nprinted = 0;
2545 char *line;
2546 size_t len;
2547 wchar_t *wline;
2548 int width, wlimit;
2549 struct tog_blame_line *blame_line;
2550 struct got_object_id *prev_id = NULL;
2551 char *id_str;
2553 err = got_object_id_str(&id_str, id);
2554 if (err)
2555 return err;
2557 rewind(f);
2558 werase(view->window);
2560 if (asprintf(&line, "commit %s", id_str) == -1) {
2561 err = got_error_from_errno();
2562 free(id_str);
2563 return err;
2566 err = format_line(&wline, &width, line, view->ncols);
2567 free(line);
2568 line = NULL;
2569 if (view_needs_focus_indication(view))
2570 wstandout(view->window);
2571 waddwstr(view->window, wline);
2572 if (view_needs_focus_indication(view))
2573 wstandend(view->window);
2574 free(wline);
2575 wline = NULL;
2576 if (width < view->ncols)
2577 waddch(view->window, '\n');
2579 if (asprintf(&line, "[%d/%d] %s%s",
2580 *first_displayed_line - 1 + selected_line, nlines,
2581 blame_complete ? "" : "annotating... ", path) == -1) {
2582 free(id_str);
2583 return got_error_from_errno();
2585 free(id_str);
2586 err = format_line(&wline, &width, line, view->ncols);
2587 free(line);
2588 line = NULL;
2589 if (err)
2590 return err;
2591 waddwstr(view->window, wline);
2592 free(wline);
2593 wline = NULL;
2594 if (width < view->ncols)
2595 waddch(view->window, '\n');
2597 *eof = 0;
2598 while (nprinted < max_lines - 2) {
2599 line = parse_next_line(f, &len);
2600 if (line == NULL) {
2601 *eof = 1;
2602 break;
2604 if (++lineno < *first_displayed_line) {
2605 free(line);
2606 continue;
2609 wlimit = view->ncols < 9 ? 0 : view->ncols - 9;
2610 err = format_line(&wline, &width, line, wlimit);
2611 if (err) {
2612 free(line);
2613 return err;
2616 if (view->focussed && nprinted == selected_line - 1)
2617 wstandout(view->window);
2619 blame_line = &lines[lineno - 1];
2620 if (blame_line->annotated && prev_id &&
2621 got_object_id_cmp(prev_id, blame_line->id) == 0)
2622 waddstr(view->window, " ");
2623 else if (blame_line->annotated) {
2624 char *id_str;
2625 err = got_object_id_str(&id_str, blame_line->id);
2626 if (err) {
2627 free(line);
2628 free(wline);
2629 return err;
2631 wprintw(view->window, "%.8s ", id_str);
2632 free(id_str);
2633 prev_id = blame_line->id;
2634 } else {
2635 waddstr(view->window, "........ ");
2636 prev_id = NULL;
2639 waddwstr(view->window, wline);
2640 while (width < wlimit) {
2641 waddch(view->window, ' ');
2642 width++;
2644 if (view->focussed && nprinted == selected_line - 1)
2645 wstandend(view->window);
2646 if (++nprinted == 1)
2647 *first_displayed_line = lineno;
2648 free(line);
2649 free(wline);
2650 wline = NULL;
2652 *last_displayed_line = lineno;
2654 view_vborder(view);
2656 return NULL;
2659 static const struct got_error *
2660 blame_cb(void *arg, int nlines, int lineno, struct got_object_id *id)
2662 const struct got_error *err = NULL;
2663 struct tog_blame_cb_args *a = arg;
2664 struct tog_blame_line *line;
2665 int errcode;
2667 if (nlines != a->nlines ||
2668 (lineno != -1 && lineno < 1) || lineno > a->nlines)
2669 return got_error(GOT_ERR_RANGE);
2671 errcode = pthread_mutex_lock(&tog_mutex);
2672 if (errcode)
2673 return got_error_set_errno(errcode);
2675 if (*a->quit) { /* user has quit the blame view */
2676 err = got_error(GOT_ERR_ITER_COMPLETED);
2677 goto done;
2680 if (lineno == -1)
2681 goto done; /* no change in this commit */
2683 line = &a->lines[lineno - 1];
2684 if (line->annotated)
2685 goto done;
2687 line->id = got_object_id_dup(id);
2688 if (line->id == NULL) {
2689 err = got_error_from_errno();
2690 goto done;
2692 line->annotated = 1;
2693 done:
2694 errcode = pthread_mutex_unlock(&tog_mutex);
2695 if (errcode)
2696 err = got_error_set_errno(errcode);
2697 return err;
2700 static void *
2701 blame_thread(void *arg)
2703 const struct got_error *err;
2704 struct tog_blame_thread_args *ta = arg;
2705 struct tog_blame_cb_args *a = ta->cb_args;
2706 int errcode;
2708 err = got_blame_incremental(ta->path, a->commit_id, ta->repo,
2709 blame_cb, ta->cb_args);
2711 errcode = pthread_mutex_lock(&tog_mutex);
2712 if (errcode)
2713 return (void *)got_error_set_errno(errcode);
2715 got_repo_close(ta->repo);
2716 ta->repo = NULL;
2717 *ta->complete = 1;
2719 errcode = pthread_mutex_unlock(&tog_mutex);
2720 if (errcode && err == NULL)
2721 err = got_error_set_errno(errcode);
2723 return (void *)err;
2726 static struct got_object_id *
2727 get_selected_commit_id(struct tog_blame_line *lines, int first_displayed_line,
2728 int selected_line)
2730 struct tog_blame_line *line;
2732 line = &lines[first_displayed_line - 1 + selected_line - 1];
2733 if (!line->annotated)
2734 return NULL;
2736 return line->id;
2739 static const struct got_error *
2740 stop_blame(struct tog_blame *blame)
2742 const struct got_error *err = NULL;
2743 int i;
2745 if (blame->thread) {
2746 int errcode;
2747 errcode = pthread_mutex_unlock(&tog_mutex);
2748 if (errcode)
2749 return got_error_set_errno(errcode);
2750 errcode = pthread_join(blame->thread, (void **)&err);
2751 if (errcode)
2752 return got_error_set_errno(errcode);
2753 errcode = pthread_mutex_lock(&tog_mutex);
2754 if (errcode)
2755 return got_error_set_errno(errcode);
2756 if (err && err->code == GOT_ERR_ITER_COMPLETED)
2757 err = NULL;
2758 blame->thread = NULL;
2760 if (blame->thread_args.repo) {
2761 got_repo_close(blame->thread_args.repo);
2762 blame->thread_args.repo = NULL;
2764 if (blame->f) {
2765 if (fclose(blame->f) != 0 && err == NULL)
2766 err = got_error_from_errno();
2767 blame->f = NULL;
2769 if (blame->lines) {
2770 for (i = 0; i < blame->nlines; i++)
2771 free(blame->lines[i].id);
2772 free(blame->lines);
2773 blame->lines = NULL;
2775 free(blame->cb_args.commit_id);
2776 blame->cb_args.commit_id = NULL;
2778 return err;
2781 static const struct got_error *
2782 run_blame(struct tog_blame *blame, struct tog_view *view, int *blame_complete,
2783 int *first_displayed_line, int *last_displayed_line, int *selected_line,
2784 int *done, int *eof, const char *path, struct got_object_id *commit_id,
2785 struct got_repository *repo)
2787 const struct got_error *err = NULL;
2788 struct got_blob_object *blob = NULL;
2789 struct got_repository *thread_repo = NULL;
2790 struct got_object_id *obj_id = NULL;
2791 int obj_type;
2793 err = got_object_id_by_path(&obj_id, repo, commit_id, path);
2794 if (err)
2795 return err;
2796 if (obj_id == NULL)
2797 return got_error(GOT_ERR_NO_OBJ);
2799 err = got_object_get_type(&obj_type, repo, obj_id);
2800 if (err)
2801 goto done;
2803 if (obj_type != GOT_OBJ_TYPE_BLOB) {
2804 err = got_error(GOT_ERR_OBJ_TYPE);
2805 goto done;
2808 err = got_object_open_as_blob(&blob, repo, obj_id, 8192);
2809 if (err)
2810 goto done;
2811 blame->f = got_opentemp();
2812 if (blame->f == NULL) {
2813 err = got_error_from_errno();
2814 goto done;
2816 err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
2817 blame->f, blob);
2818 if (err)
2819 goto done;
2821 blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
2822 if (blame->lines == NULL) {
2823 err = got_error_from_errno();
2824 goto done;
2827 err = got_repo_open(&thread_repo, got_repo_get_path(repo));
2828 if (err)
2829 goto done;
2831 blame->cb_args.view = view;
2832 blame->cb_args.lines = blame->lines;
2833 blame->cb_args.nlines = blame->nlines;
2834 blame->cb_args.commit_id = got_object_id_dup(commit_id);
2835 if (blame->cb_args.commit_id == NULL) {
2836 err = got_error_from_errno();
2837 goto done;
2839 blame->cb_args.quit = done;
2841 blame->thread_args.path = path;
2842 blame->thread_args.repo = thread_repo;
2843 blame->thread_args.cb_args = &blame->cb_args;
2844 blame->thread_args.complete = blame_complete;
2845 *blame_complete = 0;
2847 done:
2848 if (blob)
2849 got_object_blob_close(blob);
2850 free(obj_id);
2851 if (err)
2852 stop_blame(blame);
2853 return err;
2856 static const struct got_error *
2857 open_blame_view(struct tog_view *view, char *path,
2858 struct got_object_id *commit_id, struct got_reflist_head *refs,
2859 struct got_repository *repo)
2861 const struct got_error *err = NULL;
2862 struct tog_blame_view_state *s = &view->state.blame;
2864 SIMPLEQ_INIT(&s->blamed_commits);
2866 err = got_object_qid_alloc(&s->blamed_commit, commit_id);
2867 if (err)
2868 return err;
2870 SIMPLEQ_INSERT_HEAD(&s->blamed_commits, s->blamed_commit, entry);
2871 s->first_displayed_line = 1;
2872 s->last_displayed_line = view->nlines;
2873 s->selected_line = 1;
2874 s->blame_complete = 0;
2875 s->path = path;
2876 if (s->path == NULL)
2877 return got_error_from_errno();
2878 s->repo = repo;
2879 s->refs = refs;
2880 s->commit_id = commit_id;
2881 memset(&s->blame, 0, sizeof(s->blame));
2883 view->show = show_blame_view;
2884 view->input = input_blame_view;
2885 view->close = close_blame_view;
2887 return run_blame(&s->blame, view, &s->blame_complete,
2888 &s->first_displayed_line, &s->last_displayed_line,
2889 &s->selected_line, &s->done, &s->eof, s->path,
2890 s->blamed_commit->id, s->repo);
2893 static const struct got_error *
2894 close_blame_view(struct tog_view *view)
2896 const struct got_error *err = NULL;
2897 struct tog_blame_view_state *s = &view->state.blame;
2899 if (s->blame.thread)
2900 err = stop_blame(&s->blame);
2902 while (!SIMPLEQ_EMPTY(&s->blamed_commits)) {
2903 struct got_object_qid *blamed_commit;
2904 blamed_commit = SIMPLEQ_FIRST(&s->blamed_commits);
2905 SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
2906 got_object_qid_free(blamed_commit);
2909 free(s->path);
2911 return err;
2914 static const struct got_error *
2915 show_blame_view(struct tog_view *view)
2917 const struct got_error *err = NULL;
2918 struct tog_blame_view_state *s = &view->state.blame;
2919 int errcode;
2921 if (s->blame.thread == NULL) {
2922 errcode = pthread_create(&s->blame.thread, NULL, blame_thread,
2923 &s->blame.thread_args);
2924 if (errcode)
2925 return got_error_set_errno(errcode);
2928 err = draw_blame(view, s->blamed_commit->id, s->blame.f,
2929 s->path, s->blame.lines, s->blame.nlines, s->blame_complete,
2930 s->selected_line, &s->first_displayed_line,
2931 &s->last_displayed_line, &s->eof, view->nlines);
2933 view_vborder(view);
2934 return err;
2937 static const struct got_error *
2938 input_blame_view(struct tog_view **new_view, struct tog_view **dead_view,
2939 struct tog_view **focus_view, struct tog_view *view, int ch)
2941 const struct got_error *err = NULL, *thread_err = NULL;
2942 struct tog_view *diff_view;
2943 struct tog_blame_view_state *s = &view->state.blame;
2944 int begin_x = 0;
2946 switch (ch) {
2947 case 'q':
2948 s->done = 1;
2949 break;
2950 case 'k':
2951 case KEY_UP:
2952 if (s->selected_line > 1)
2953 s->selected_line--;
2954 else if (s->selected_line == 1 &&
2955 s->first_displayed_line > 1)
2956 s->first_displayed_line--;
2957 else
2958 view_flash(view);
2959 break;
2960 case KEY_PPAGE:
2961 if (s->first_displayed_line == 1) {
2962 if (s->selected_line == 1) {
2963 view_flash(view);
2964 break;
2966 s->selected_line = 1;
2967 break;
2969 if (s->first_displayed_line > view->nlines - 2)
2970 s->first_displayed_line -=
2971 (view->nlines - 2);
2972 else
2973 s->first_displayed_line = 1;
2974 break;
2975 case 'j':
2976 case KEY_DOWN:
2977 if (s->selected_line < view->nlines - 2 &&
2978 s->first_displayed_line +
2979 s->selected_line <= s->blame.nlines)
2980 s->selected_line++;
2981 else if (s->last_displayed_line <
2982 s->blame.nlines)
2983 s->first_displayed_line++;
2984 else
2985 view_flash(view);
2986 break;
2987 case 'b':
2988 case 'p': {
2989 struct got_object_id *id = NULL;
2990 id = get_selected_commit_id(s->blame.lines,
2991 s->first_displayed_line, s->selected_line);
2992 if (id == NULL)
2993 break;
2994 if (ch == 'p') {
2995 struct got_commit_object *commit;
2996 struct got_object_qid *pid;
2997 struct got_object_id *blob_id = NULL;
2998 int obj_type;
2999 err = got_object_open_as_commit(&commit,
3000 s->repo, id);
3001 if (err)
3002 break;
3003 pid = SIMPLEQ_FIRST(
3004 got_object_commit_get_parent_ids(commit));
3005 if (pid == NULL) {
3006 got_object_commit_close(commit);
3007 break;
3009 /* Check if path history ends here. */
3010 err = got_object_id_by_path(&blob_id, s->repo,
3011 pid->id, s->path);
3012 if (err) {
3013 if (err->code == GOT_ERR_NO_TREE_ENTRY)
3014 err = NULL;
3015 got_object_commit_close(commit);
3016 break;
3018 err = got_object_get_type(&obj_type, s->repo,
3019 blob_id);
3020 free(blob_id);
3021 /* Can't blame non-blob type objects. */
3022 if (obj_type != GOT_OBJ_TYPE_BLOB) {
3023 got_object_commit_close(commit);
3024 break;
3026 err = got_object_qid_alloc(&s->blamed_commit,
3027 pid->id);
3028 got_object_commit_close(commit);
3029 } else {
3030 if (got_object_id_cmp(id,
3031 s->blamed_commit->id) == 0)
3032 break;
3033 err = got_object_qid_alloc(&s->blamed_commit,
3034 id);
3036 if (err)
3037 break;
3038 s->done = 1;
3039 thread_err = stop_blame(&s->blame);
3040 s->done = 0;
3041 if (thread_err)
3042 break;
3043 SIMPLEQ_INSERT_HEAD(&s->blamed_commits,
3044 s->blamed_commit, entry);
3045 err = run_blame(&s->blame, view, &s->blame_complete,
3046 &s->first_displayed_line, &s->last_displayed_line,
3047 &s->selected_line, &s->done, &s->eof,
3048 s->path, s->blamed_commit->id, s->repo);
3049 if (err)
3050 break;
3051 break;
3053 case 'B': {
3054 struct got_object_qid *first;
3055 first = SIMPLEQ_FIRST(&s->blamed_commits);
3056 if (!got_object_id_cmp(first->id, s->commit_id))
3057 break;
3058 s->done = 1;
3059 thread_err = stop_blame(&s->blame);
3060 s->done = 0;
3061 if (thread_err)
3062 break;
3063 SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
3064 got_object_qid_free(s->blamed_commit);
3065 s->blamed_commit =
3066 SIMPLEQ_FIRST(&s->blamed_commits);
3067 err = run_blame(&s->blame, view, &s->blame_complete,
3068 &s->first_displayed_line, &s->last_displayed_line,
3069 &s->selected_line, &s->done, &s->eof, s->path,
3070 s->blamed_commit->id, s->repo);
3071 if (err)
3072 break;
3073 break;
3075 case KEY_ENTER:
3076 case '\r': {
3077 struct got_object_id *id = NULL;
3078 struct got_object_qid *pid;
3079 struct got_commit_object *commit = NULL;
3080 id = get_selected_commit_id(s->blame.lines,
3081 s->first_displayed_line, s->selected_line);
3082 if (id == NULL)
3083 break;
3084 err = got_object_open_as_commit(&commit, s->repo, id);
3085 if (err)
3086 break;
3087 pid = SIMPLEQ_FIRST(
3088 got_object_commit_get_parent_ids(commit));
3089 if (view_is_parent_view(view))
3090 begin_x = view_split_begin_x(view->begin_x);
3091 diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
3092 if (diff_view == NULL) {
3093 got_object_commit_close(commit);
3094 err = got_error_from_errno();
3095 break;
3097 err = open_diff_view(diff_view, pid ? pid->id : NULL,
3098 id, NULL, s->refs, s->repo);
3099 got_object_commit_close(commit);
3100 if (err) {
3101 view_close(diff_view);
3102 break;
3104 if (view_is_parent_view(view)) {
3105 err = view_close_child(view);
3106 if (err)
3107 break;
3108 err = view_set_child(view, diff_view);
3109 if (err) {
3110 view_close(diff_view);
3111 break;
3113 *focus_view = diff_view;
3114 view->child_focussed = 1;
3115 } else
3116 *new_view = diff_view;
3117 if (err)
3118 break;
3119 break;
3121 case KEY_NPAGE:
3122 case ' ':
3123 if (s->last_displayed_line >= s->blame.nlines &&
3124 s->selected_line >= MIN(s->blame.nlines,
3125 view->nlines - 2)) {
3126 view_flash(view);
3127 break;
3129 if (s->last_displayed_line >= s->blame.nlines &&
3130 s->selected_line < view->nlines - 2) {
3131 s->selected_line = MIN(s->blame.nlines,
3132 view->nlines - 2);
3133 break;
3135 if (s->last_displayed_line + view->nlines - 2
3136 <= s->blame.nlines)
3137 s->first_displayed_line +=
3138 view->nlines - 2;
3139 else
3140 s->first_displayed_line =
3141 s->blame.nlines -
3142 (view->nlines - 3);
3143 break;
3144 case KEY_RESIZE:
3145 if (s->selected_line > view->nlines - 2) {
3146 s->selected_line = MIN(s->blame.nlines,
3147 view->nlines - 2);
3149 break;
3150 default:
3151 break;
3153 return thread_err ? thread_err : err;
3156 static const struct got_error *
3157 cmd_blame(int argc, char *argv[])
3159 const struct got_error *error;
3160 struct got_repository *repo = NULL;
3161 struct got_reflist_head refs;
3162 struct got_worktree *worktree = NULL;
3163 char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
3164 struct got_object_id *commit_id = NULL;
3165 char *commit_id_str = NULL;
3166 int ch;
3167 struct tog_view *view;
3169 #ifndef PROFILE
3170 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
3171 NULL) == -1)
3172 err(1, "pledge");
3173 #endif
3175 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
3176 switch (ch) {
3177 case 'c':
3178 commit_id_str = optarg;
3179 break;
3180 case 'r':
3181 repo_path = realpath(optarg, NULL);
3182 if (repo_path == NULL)
3183 err(1, "-r option");
3184 break;
3185 default:
3186 usage_blame();
3187 /* NOTREACHED */
3191 argc -= optind;
3192 argv += optind;
3194 if (argc == 1)
3195 path = argv[0];
3196 else
3197 usage_blame();
3199 cwd = getcwd(NULL, 0);
3200 if (cwd == NULL) {
3201 error = got_error_from_errno();
3202 goto done;
3204 if (repo_path == NULL) {
3205 error = got_worktree_open(&worktree, cwd);
3206 if (error && error->code != GOT_ERR_NOT_WORKTREE)
3207 goto done;
3208 else
3209 error = NULL;
3210 if (worktree) {
3211 repo_path =
3212 strdup(got_worktree_get_repo_path(worktree));
3213 if (repo_path == NULL)
3214 error = got_error_from_errno();
3215 if (error)
3216 goto done;
3217 } else {
3218 repo_path = strdup(cwd);
3219 if (repo_path == NULL) {
3220 error = got_error_from_errno();
3221 goto done;
3226 init_curses();
3228 error = apply_unveil(repo_path, NULL);
3229 if (error)
3230 goto done;
3232 error = got_repo_open(&repo, repo_path);
3233 if (error != NULL)
3234 goto done;
3236 if (worktree) {
3237 const char *prefix = got_worktree_get_path_prefix(worktree);
3238 char *p, *worktree_subdir = cwd +
3239 strlen(got_worktree_get_root_path(worktree));
3240 if (asprintf(&p, "%s%s%s%s%s",
3241 prefix, (strcmp(prefix, "/") != 0) ? "/" : "",
3242 worktree_subdir, worktree_subdir[0] ? "/" : "",
3243 path) == -1) {
3244 error = got_error_from_errno();
3245 goto done;
3247 error = got_repo_map_path(&in_repo_path, repo, p, 0);
3248 free(p);
3249 } else {
3250 error = got_repo_map_path(&in_repo_path, repo, path, 1);
3252 if (error)
3253 goto done;
3255 if (commit_id_str == NULL) {
3256 struct got_reference *head_ref;
3257 error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
3258 if (error != NULL)
3259 goto done;
3260 error = got_ref_resolve(&commit_id, repo, head_ref);
3261 got_ref_close(head_ref);
3262 } else {
3263 error = got_object_resolve_id_str(&commit_id, repo,
3264 commit_id_str);
3266 if (error != NULL)
3267 goto done;
3269 SIMPLEQ_INIT(&refs);
3270 error = got_ref_list(&refs, repo);
3271 if (error)
3272 goto done;
3274 view = view_open(0, 0, 0, 0, TOG_VIEW_BLAME);
3275 if (view == NULL) {
3276 error = got_error_from_errno();
3277 goto done;
3279 error = open_blame_view(view, in_repo_path, commit_id, &refs, repo);
3280 if (error)
3281 goto done;
3282 error = view_loop(view);
3283 done:
3284 free(repo_path);
3285 free(cwd);
3286 free(commit_id);
3287 if (worktree)
3288 got_worktree_close(worktree);
3289 if (repo)
3290 got_repo_close(repo);
3291 return error;
3294 static const struct got_error *
3295 draw_tree_entries(struct tog_view *view,
3296 struct got_tree_entry **first_displayed_entry,
3297 struct got_tree_entry **last_displayed_entry,
3298 struct got_tree_entry **selected_entry, int *ndisplayed,
3299 const char *label, int show_ids, const char *parent_path,
3300 const struct got_tree_entries *entries, int selected, int limit, int isroot)
3302 const struct got_error *err = NULL;
3303 struct got_tree_entry *te;
3304 wchar_t *wline;
3305 int width, n;
3307 *ndisplayed = 0;
3309 werase(view->window);
3311 if (limit == 0)
3312 return NULL;
3314 err = format_line(&wline, &width, label, view->ncols);
3315 if (err)
3316 return err;
3317 if (view_needs_focus_indication(view))
3318 wstandout(view->window);
3319 waddwstr(view->window, wline);
3320 if (view_needs_focus_indication(view))
3321 wstandend(view->window);
3322 free(wline);
3323 wline = NULL;
3324 if (width < view->ncols)
3325 waddch(view->window, '\n');
3326 if (--limit <= 0)
3327 return NULL;
3328 err = format_line(&wline, &width, parent_path, view->ncols);
3329 if (err)
3330 return err;
3331 waddwstr(view->window, wline);
3332 free(wline);
3333 wline = NULL;
3334 if (width < view->ncols)
3335 waddch(view->window, '\n');
3336 if (--limit <= 0)
3337 return NULL;
3338 waddch(view->window, '\n');
3339 if (--limit <= 0)
3340 return NULL;
3342 te = SIMPLEQ_FIRST(&entries->head);
3343 if (*first_displayed_entry == NULL) {
3344 if (selected == 0) {
3345 if (view->focussed)
3346 wstandout(view->window);
3347 *selected_entry = NULL;
3349 waddstr(view->window, " ..\n"); /* parent directory */
3350 if (selected == 0 && view->focussed)
3351 wstandend(view->window);
3352 (*ndisplayed)++;
3353 if (--limit <= 0)
3354 return NULL;
3355 n = 1;
3356 } else {
3357 n = 0;
3358 while (te != *first_displayed_entry)
3359 te = SIMPLEQ_NEXT(te, entry);
3362 while (te) {
3363 char *line = NULL, *id_str = NULL;
3365 if (show_ids) {
3366 err = got_object_id_str(&id_str, te->id);
3367 if (err)
3368 return got_error_from_errno();
3370 if (asprintf(&line, "%s %s%s", id_str ? id_str : "",
3371 te->name, S_ISDIR(te->mode) ? "/" :
3372 ((te->mode & S_IXUSR) ? "*" : "")) == -1) {
3373 free(id_str);
3374 return got_error_from_errno();
3376 free(id_str);
3377 err = format_line(&wline, &width, line, view->ncols);
3378 if (err) {
3379 free(line);
3380 break;
3382 if (n == selected) {
3383 if (view->focussed)
3384 wstandout(view->window);
3385 *selected_entry = te;
3387 waddwstr(view->window, wline);
3388 if (width < view->ncols)
3389 waddch(view->window, '\n');
3390 if (n == selected && view->focussed)
3391 wstandend(view->window);
3392 free(line);
3393 free(wline);
3394 wline = NULL;
3395 n++;
3396 (*ndisplayed)++;
3397 *last_displayed_entry = te;
3398 if (--limit <= 0)
3399 break;
3400 te = SIMPLEQ_NEXT(te, entry);
3403 return err;
3406 static void
3407 tree_scroll_up(struct tog_view *view,
3408 struct got_tree_entry **first_displayed_entry, int maxscroll,
3409 const struct got_tree_entries *entries, int isroot)
3411 struct got_tree_entry *te, *prev;
3412 int i;
3414 if (*first_displayed_entry == NULL) {
3415 view_flash(view);
3416 return;
3419 te = SIMPLEQ_FIRST(&entries->head);
3420 if (*first_displayed_entry == te) {
3421 view_flash(view);
3422 if (!isroot)
3423 *first_displayed_entry = NULL;
3424 return;
3427 /* XXX this is stupid... switch to TAILQ? */
3428 for (i = 0; i < maxscroll; i++) {
3429 while (te != *first_displayed_entry) {
3430 prev = te;
3431 te = SIMPLEQ_NEXT(te, entry);
3433 *first_displayed_entry = prev;
3434 te = SIMPLEQ_FIRST(&entries->head);
3436 if (!isroot && te == SIMPLEQ_FIRST(&entries->head) && i < maxscroll)
3437 *first_displayed_entry = NULL;
3440 static int
3441 tree_scroll_down(struct got_tree_entry **first_displayed_entry, int maxscroll,
3442 struct got_tree_entry *last_displayed_entry,
3443 const struct got_tree_entries *entries)
3445 struct got_tree_entry *next, *last;
3446 int n = 0;
3448 if (*first_displayed_entry)
3449 next = SIMPLEQ_NEXT(*first_displayed_entry, entry);
3450 else
3451 next = SIMPLEQ_FIRST(&entries->head);
3452 last = last_displayed_entry;
3453 while (next && last && n++ < maxscroll) {
3454 last = SIMPLEQ_NEXT(last, entry);
3455 if (last) {
3456 *first_displayed_entry = next;
3457 next = SIMPLEQ_NEXT(next, entry);
3460 return n;
3463 static const struct got_error *
3464 tree_entry_path(char **path, struct tog_parent_trees *parents,
3465 struct got_tree_entry *te)
3467 const struct got_error *err = NULL;
3468 struct tog_parent_tree *pt;
3469 size_t len = 2; /* for leading slash and NUL */
3471 TAILQ_FOREACH(pt, parents, entry)
3472 len += strlen(pt->selected_entry->name) + 1 /* slash */;
3473 if (te)
3474 len += strlen(te->name);
3476 *path = calloc(1, len);
3477 if (path == NULL)
3478 return got_error_from_errno();
3480 (*path)[0] = '/';
3481 pt = TAILQ_LAST(parents, tog_parent_trees);
3482 while (pt) {
3483 if (strlcat(*path, pt->selected_entry->name, len) >= len) {
3484 err = got_error(GOT_ERR_NO_SPACE);
3485 goto done;
3487 if (strlcat(*path, "/", len) >= len) {
3488 err = got_error(GOT_ERR_NO_SPACE);
3489 goto done;
3491 pt = TAILQ_PREV(pt, tog_parent_trees, entry);
3493 if (te) {
3494 if (strlcat(*path, te->name, len) >= len) {
3495 err = got_error(GOT_ERR_NO_SPACE);
3496 goto done;
3499 done:
3500 if (err) {
3501 free(*path);
3502 *path = NULL;
3504 return err;
3507 static const struct got_error *
3508 blame_tree_entry(struct tog_view **new_view, int begin_x,
3509 struct got_tree_entry *te, struct tog_parent_trees *parents,
3510 struct got_object_id *commit_id, struct got_reflist_head *refs,
3511 struct got_repository *repo)
3513 const struct got_error *err = NULL;
3514 char *path;
3515 struct tog_view *blame_view;
3517 err = tree_entry_path(&path, parents, te);
3518 if (err)
3519 return err;
3521 blame_view = view_open(0, 0, 0, begin_x, TOG_VIEW_BLAME);
3522 if (blame_view == NULL)
3523 return got_error_from_errno();
3525 err = open_blame_view(blame_view, path, commit_id, refs, repo);
3526 if (err) {
3527 view_close(blame_view);
3528 free(path);
3529 } else
3530 *new_view = blame_view;
3531 return err;
3534 static const struct got_error *
3535 log_tree_entry(struct tog_view **new_view, int begin_x,
3536 struct got_tree_entry *te, struct tog_parent_trees *parents,
3537 struct got_object_id *commit_id, struct got_reflist_head *refs,
3538 struct got_repository *repo)
3540 struct tog_view *log_view;
3541 const struct got_error *err = NULL;
3542 char *path;
3544 log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
3545 if (log_view == NULL)
3546 return got_error_from_errno();
3548 err = tree_entry_path(&path, parents, te);
3549 if (err)
3550 return err;
3552 err = open_log_view(log_view, commit_id, refs, repo, path, 0);
3553 if (err)
3554 view_close(log_view);
3555 else
3556 *new_view = log_view;
3557 free(path);
3558 return err;
3561 static const struct got_error *
3562 open_tree_view(struct tog_view *view, struct got_tree_object *root,
3563 struct got_object_id *commit_id, struct got_reflist_head *refs,
3564 struct got_repository *repo)
3566 const struct got_error *err = NULL;
3567 char *commit_id_str = NULL;
3568 struct tog_tree_view_state *s = &view->state.tree;
3570 TAILQ_INIT(&s->parents);
3572 err = got_object_id_str(&commit_id_str, commit_id);
3573 if (err != NULL)
3574 goto done;
3576 if (asprintf(&s->tree_label, "commit %s", commit_id_str) == -1) {
3577 err = got_error_from_errno();
3578 goto done;
3581 s->root = s->tree = root;
3582 s->entries = got_object_tree_get_entries(root);
3583 s->first_displayed_entry = SIMPLEQ_FIRST(&s->entries->head);
3584 s->commit_id = got_object_id_dup(commit_id);
3585 if (s->commit_id == NULL) {
3586 err = got_error_from_errno();
3587 goto done;
3589 s->refs = refs;
3590 s->repo = repo;
3592 view->show = show_tree_view;
3593 view->input = input_tree_view;
3594 view->close = close_tree_view;
3595 done:
3596 free(commit_id_str);
3597 if (err) {
3598 free(s->tree_label);
3599 s->tree_label = NULL;
3601 return err;
3604 static const struct got_error *
3605 close_tree_view(struct tog_view *view)
3607 struct tog_tree_view_state *s = &view->state.tree;
3609 free(s->tree_label);
3610 s->tree_label = NULL;
3611 free(s->commit_id);
3612 s->commit_id = NULL;
3613 while (!TAILQ_EMPTY(&s->parents)) {
3614 struct tog_parent_tree *parent;
3615 parent = TAILQ_FIRST(&s->parents);
3616 TAILQ_REMOVE(&s->parents, parent, entry);
3617 free(parent);
3620 if (s->tree != s->root)
3621 got_object_tree_close(s->tree);
3622 got_object_tree_close(s->root);
3624 return NULL;
3627 static const struct got_error *
3628 show_tree_view(struct tog_view *view)
3630 const struct got_error *err = NULL;
3631 struct tog_tree_view_state *s = &view->state.tree;
3632 char *parent_path;
3634 err = tree_entry_path(&parent_path, &s->parents, NULL);
3635 if (err)
3636 return err;
3638 err = draw_tree_entries(view, &s->first_displayed_entry,
3639 &s->last_displayed_entry, &s->selected_entry,
3640 &s->ndisplayed, s->tree_label, s->show_ids, parent_path,
3641 s->entries, s->selected, view->nlines, s->tree == s->root);
3642 free(parent_path);
3644 view_vborder(view);
3645 return err;
3648 static const struct got_error *
3649 input_tree_view(struct tog_view **new_view, struct tog_view **dead_view,
3650 struct tog_view **focus_view, struct tog_view *view, int ch)
3652 const struct got_error *err = NULL;
3653 struct tog_tree_view_state *s = &view->state.tree;
3654 struct tog_view *log_view;
3655 int begin_x = 0, nscrolled;
3657 switch (ch) {
3658 case 'i':
3659 s->show_ids = !s->show_ids;
3660 break;
3661 case 'l':
3662 if (!s->selected_entry)
3663 break;
3664 if (view_is_parent_view(view))
3665 begin_x = view_split_begin_x(view->begin_x);
3666 err = log_tree_entry(&log_view, begin_x,
3667 s->selected_entry, &s->parents,
3668 s->commit_id, s->refs, s->repo);
3669 if (view_is_parent_view(view)) {
3670 err = view_close_child(view);
3671 if (err)
3672 return err;
3673 err = view_set_child(view, log_view);
3674 if (err) {
3675 view_close(log_view);
3676 break;
3678 *focus_view = log_view;
3679 view->child_focussed = 1;
3680 } else
3681 *new_view = log_view;
3682 break;
3683 case 'k':
3684 case KEY_UP:
3685 if (s->selected > 0) {
3686 s->selected--;
3687 if (s->selected == 0)
3688 break;
3690 if (s->selected > 0)
3691 break;
3692 tree_scroll_up(view, &s->first_displayed_entry, 1,
3693 s->entries, s->tree == s->root);
3694 break;
3695 case KEY_PPAGE:
3696 tree_scroll_up(view, &s->first_displayed_entry,
3697 MAX(0, view->nlines - 4 - s->selected), s->entries,
3698 s->tree == s->root);
3699 s->selected = 0;
3700 if (SIMPLEQ_FIRST(&s->entries->head) ==
3701 s->first_displayed_entry && s->tree != s->root)
3702 s->first_displayed_entry = NULL;
3703 break;
3704 case 'j':
3705 case KEY_DOWN:
3706 if (s->selected < s->ndisplayed - 1) {
3707 s->selected++;
3708 break;
3710 if (SIMPLEQ_NEXT(s->last_displayed_entry, entry)
3711 == NULL) {
3712 /* can't scroll any further */
3713 view_flash(view);
3714 break;
3716 tree_scroll_down(&s->first_displayed_entry, 1,
3717 s->last_displayed_entry, s->entries);
3718 break;
3719 case KEY_NPAGE:
3720 if (SIMPLEQ_NEXT(s->last_displayed_entry, entry)
3721 == NULL) {
3722 /* can't scroll any further; move cursor down */
3723 if (s->selected < s->ndisplayed - 1)
3724 s->selected = s->ndisplayed - 1;
3725 else
3726 view_flash(view);
3727 break;
3729 nscrolled = tree_scroll_down(&s->first_displayed_entry,
3730 view->nlines, s->last_displayed_entry, s->entries);
3731 if (nscrolled < view->nlines) {
3732 int ndisplayed = 0;
3733 struct got_tree_entry *te;
3734 te = s->first_displayed_entry;
3735 do {
3736 ndisplayed++;
3737 te = SIMPLEQ_NEXT(te, entry);
3738 } while (te);
3739 s->selected = ndisplayed - 1;
3741 break;
3742 case KEY_ENTER:
3743 case '\r':
3744 if (s->selected_entry == NULL) {
3745 struct tog_parent_tree *parent;
3746 case KEY_BACKSPACE:
3747 /* user selected '..' */
3748 if (s->tree == s->root)
3749 break;
3750 parent = TAILQ_FIRST(&s->parents);
3751 TAILQ_REMOVE(&s->parents, parent,
3752 entry);
3753 got_object_tree_close(s->tree);
3754 s->tree = parent->tree;
3755 s->entries =
3756 got_object_tree_get_entries(s->tree);
3757 s->first_displayed_entry =
3758 parent->first_displayed_entry;
3759 s->selected_entry =
3760 parent->selected_entry;
3761 s->selected = parent->selected;
3762 free(parent);
3763 } else if (S_ISDIR(s->selected_entry->mode)) {
3764 struct tog_parent_tree *parent;
3765 struct got_tree_object *child;
3766 err = got_object_open_as_tree(&child,
3767 s->repo, s->selected_entry->id);
3768 if (err)
3769 break;
3770 parent = calloc(1, sizeof(*parent));
3771 if (parent == NULL) {
3772 err = got_error_from_errno();
3773 break;
3775 parent->tree = s->tree;
3776 parent->first_displayed_entry =
3777 s->first_displayed_entry;
3778 parent->selected_entry = s->selected_entry;
3779 parent->selected = s->selected;
3780 TAILQ_INSERT_HEAD(&s->parents, parent, entry);
3781 s->tree = child;
3782 s->entries =
3783 got_object_tree_get_entries(s->tree);
3784 s->selected = 0;
3785 s->first_displayed_entry = NULL;
3786 } else if (S_ISREG(s->selected_entry->mode)) {
3787 struct tog_view *blame_view;
3788 int begin_x = view_is_parent_view(view) ?
3789 view_split_begin_x(view->begin_x) : 0;
3791 err = blame_tree_entry(&blame_view, begin_x,
3792 s->selected_entry, &s->parents,
3793 s->commit_id, s->refs, s->repo);
3794 if (err)
3795 break;
3796 if (view_is_parent_view(view)) {
3797 err = view_close_child(view);
3798 if (err)
3799 return err;
3800 err = view_set_child(view, blame_view);
3801 if (err) {
3802 view_close(blame_view);
3803 break;
3805 *focus_view = blame_view;
3806 view->child_focussed = 1;
3807 } else
3808 *new_view = blame_view;
3810 break;
3811 case KEY_RESIZE:
3812 if (s->selected > view->nlines)
3813 s->selected = s->ndisplayed - 1;
3814 break;
3815 default:
3816 break;
3819 return err;
3822 __dead static void
3823 usage_tree(void)
3825 endwin();
3826 fprintf(stderr, "usage: %s tree [-c commit] [repository-path]\n",
3827 getprogname());
3828 exit(1);
3831 static const struct got_error *
3832 cmd_tree(int argc, char *argv[])
3834 const struct got_error *error;
3835 struct got_repository *repo = NULL;
3836 struct got_reflist_head refs;
3837 char *repo_path = NULL;
3838 struct got_object_id *commit_id = NULL;
3839 char *commit_id_arg = NULL;
3840 struct got_commit_object *commit = NULL;
3841 struct got_tree_object *tree = NULL;
3842 int ch;
3843 struct tog_view *view;
3845 #ifndef PROFILE
3846 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
3847 NULL) == -1)
3848 err(1, "pledge");
3849 #endif
3851 while ((ch = getopt(argc, argv, "c:")) != -1) {
3852 switch (ch) {
3853 case 'c':
3854 commit_id_arg = optarg;
3855 break;
3856 default:
3857 usage_tree();
3858 /* NOTREACHED */
3862 argc -= optind;
3863 argv += optind;
3865 if (argc == 0) {
3866 struct got_worktree *worktree;
3867 char *cwd = getcwd(NULL, 0);
3868 if (cwd == NULL)
3869 return got_error_from_errno();
3870 error = got_worktree_open(&worktree, cwd);
3871 if (error && error->code != GOT_ERR_NOT_WORKTREE)
3872 goto done;
3873 if (worktree) {
3874 free(cwd);
3875 repo_path =
3876 strdup(got_worktree_get_repo_path(worktree));
3877 got_worktree_close(worktree);
3878 } else
3879 repo_path = cwd;
3880 if (repo_path == NULL) {
3881 error = got_error_from_errno();
3882 goto done;
3884 } else if (argc == 1) {
3885 repo_path = realpath(argv[0], NULL);
3886 if (repo_path == NULL)
3887 return got_error_from_errno();
3888 } else
3889 usage_log();
3891 init_curses();
3893 error = apply_unveil(repo_path, NULL);
3894 if (error)
3895 goto done;
3897 error = got_repo_open(&repo, repo_path);
3898 if (error != NULL)
3899 goto done;
3901 if (commit_id_arg == NULL)
3902 error = get_head_commit_id(&commit_id, repo);
3903 else
3904 error = got_object_resolve_id_str(&commit_id, repo,
3905 commit_id_arg);
3906 if (error != NULL)
3907 goto done;
3909 error = got_object_open_as_commit(&commit, repo, commit_id);
3910 if (error != NULL)
3911 goto done;
3913 error = got_object_open_as_tree(&tree, repo,
3914 got_object_commit_get_tree_id(commit));
3915 if (error != NULL)
3916 goto done;
3918 SIMPLEQ_INIT(&refs);
3919 error = got_ref_list(&refs, repo);
3920 if (error)
3921 goto done;
3923 view = view_open(0, 0, 0, 0, TOG_VIEW_TREE);
3924 if (view == NULL) {
3925 error = got_error_from_errno();
3926 goto done;
3928 error = open_tree_view(view, tree, commit_id, &refs, repo);
3929 if (error)
3930 goto done;
3931 error = view_loop(view);
3932 done:
3933 free(repo_path);
3934 free(commit_id);
3935 if (commit)
3936 got_object_commit_close(commit);
3937 if (tree)
3938 got_object_tree_close(tree);
3939 if (repo)
3940 got_repo_close(repo);
3941 return error;
3944 __dead static void
3945 usage(void)
3947 int i;
3949 fprintf(stderr, "usage: %s [-h] [command] [arg ...]\n\n"
3950 "Available commands:\n", getprogname());
3951 for (i = 0; i < nitems(tog_commands); i++) {
3952 struct tog_cmd *cmd = &tog_commands[i];
3953 fprintf(stderr, " %s: %s\n", cmd->name, cmd->descr);
3955 exit(1);
3958 static char **
3959 make_argv(const char *arg0, const char *arg1)
3961 char **argv;
3962 int argc = (arg1 == NULL ? 1 : 2);
3964 argv = calloc(argc, sizeof(char *));
3965 if (argv == NULL)
3966 err(1, "calloc");
3967 argv[0] = strdup(arg0);
3968 if (argv[0] == NULL)
3969 err(1, "calloc");
3970 if (arg1) {
3971 argv[1] = strdup(arg1);
3972 if (argv[1] == NULL)
3973 err(1, "calloc");
3976 return argv;
3979 int
3980 main(int argc, char *argv[])
3982 const struct got_error *error = NULL;
3983 struct tog_cmd *cmd = NULL;
3984 int ch, hflag = 0;
3985 char **cmd_argv = NULL;
3987 setlocale(LC_CTYPE, "");
3989 while ((ch = getopt(argc, argv, "h")) != -1) {
3990 switch (ch) {
3991 case 'h':
3992 hflag = 1;
3993 break;
3994 default:
3995 usage();
3996 /* NOTREACHED */
4000 argc -= optind;
4001 argv += optind;
4002 optind = 0;
4003 optreset = 1;
4005 if (argc == 0) {
4006 if (hflag)
4007 usage();
4008 /* Build an argument vector which runs a default command. */
4009 cmd = &tog_commands[0];
4010 cmd_argv = make_argv(cmd->name, NULL);
4011 argc = 1;
4012 } else {
4013 int i;
4015 /* Did the user specific a command? */
4016 for (i = 0; i < nitems(tog_commands); i++) {
4017 if (strncmp(tog_commands[i].name, argv[0],
4018 strlen(argv[0])) == 0) {
4019 cmd = &tog_commands[i];
4020 if (hflag)
4021 tog_commands[i].cmd_usage();
4022 break;
4025 if (cmd == NULL) {
4026 /* Did the user specify a repository? */
4027 char *repo_path = realpath(argv[0], NULL);
4028 if (repo_path) {
4029 struct got_repository *repo;
4030 error = got_repo_open(&repo, repo_path);
4031 if (error == NULL)
4032 got_repo_close(repo);
4033 } else
4034 error = got_error_from_errno();
4035 if (error) {
4036 if (hflag) {
4037 fprintf(stderr, "%s: '%s' is not a "
4038 "known command\n", getprogname(),
4039 argv[0]);
4040 usage();
4042 fprintf(stderr, "%s: '%s' is neither a known "
4043 "command nor a path to a repository\n",
4044 getprogname(), argv[0]);
4045 free(repo_path);
4046 return 1;
4048 cmd = &tog_commands[0];
4049 cmd_argv = make_argv(cmd->name, repo_path);
4050 argc = 2;
4051 free(repo_path);
4055 error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
4056 if (error)
4057 goto done;
4058 done:
4059 endwin();
4060 free(cmd_argv);
4061 if (error)
4062 fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
4063 return 0;