Blob


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