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 <errno.h>
22 #define _XOPEN_SOURCE_EXTENDED
23 #include <curses.h>
24 #undef _XOPEN_SOURCE_EXTENDED
25 #include <panel.h>
26 #include <locale.h>
27 #include <signal.h>
28 #include <stdlib.h>
29 #include <stdarg.h>
30 #include <stdio.h>
31 #include <getopt.h>
32 #include <string.h>
33 #include <err.h>
34 #include <unistd.h>
35 #include <util.h>
36 #include <limits.h>
37 #include <wchar.h>
38 #include <time.h>
39 #include <pthread.h>
40 #include <libgen.h>
41 #include <regex.h>
43 #include "got_version.h"
44 #include "got_error.h"
45 #include "got_object.h"
46 #include "got_reference.h"
47 #include "got_repository.h"
48 #include "got_diff.h"
49 #include "got_opentemp.h"
50 #include "got_utf8.h"
51 #include "got_cancel.h"
52 #include "got_commit_graph.h"
53 #include "got_blame.h"
54 #include "got_privsep.h"
55 #include "got_path.h"
56 #include "got_worktree.h"
58 #ifndef MIN
59 #define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
60 #endif
62 #ifndef MAX
63 #define MAX(_a,_b) ((_a) > (_b) ? (_a) : (_b))
64 #endif
66 #define CTRL(x) ((x) & 0x1f)
68 #ifndef nitems
69 #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
70 #endif
72 struct tog_cmd {
73 const char *name;
74 const struct got_error *(*cmd_main)(int, char *[]);
75 void (*cmd_usage)(void);
76 };
78 __dead static void usage(int);
79 __dead static void usage_log(void);
80 __dead static void usage_diff(void);
81 __dead static void usage_blame(void);
82 __dead static void usage_tree(void);
84 static const struct got_error* cmd_log(int, char *[]);
85 static const struct got_error* cmd_diff(int, char *[]);
86 static const struct got_error* cmd_blame(int, char *[]);
87 static const struct got_error* cmd_tree(int, char *[]);
89 static struct tog_cmd tog_commands[] = {
90 { "log", cmd_log, usage_log },
91 { "diff", cmd_diff, usage_diff },
92 { "blame", cmd_blame, usage_blame },
93 { "tree", cmd_tree, usage_tree },
94 };
96 enum tog_view_type {
97 TOG_VIEW_DIFF,
98 TOG_VIEW_LOG,
99 TOG_VIEW_BLAME,
100 TOG_VIEW_TREE
101 };
103 #define TOG_EOF_STRING "(END)"
105 struct commit_queue_entry {
106 TAILQ_ENTRY(commit_queue_entry) entry;
107 struct got_object_id *id;
108 struct got_commit_object *commit;
109 int idx;
110 };
111 TAILQ_HEAD(commit_queue_head, commit_queue_entry);
112 struct commit_queue {
113 int ncommits;
114 struct commit_queue_head head;
115 };
117 struct tog_color {
118 SIMPLEQ_ENTRY(tog_color) entry;
119 regex_t regex;
120 short colorpair;
121 };
122 SIMPLEQ_HEAD(tog_colors, tog_color);
124 static const struct got_error *
125 add_color(struct tog_colors *colors, const char *pattern,
126 int idx, short color)
128 const struct got_error *err = NULL;
129 struct tog_color *tc;
130 int regerr = 0;
132 if (idx < 1 || idx > COLOR_PAIRS - 1)
133 return NULL;
135 init_pair(idx, color, -1);
137 tc = calloc(1, sizeof(*tc));
138 if (tc == NULL)
139 return got_error_from_errno("calloc");
140 regerr = regcomp(&tc->regex, pattern,
141 REG_EXTENDED | REG_NOSUB | REG_NEWLINE);
142 if (regerr) {
143 static char regerr_msg[512];
144 static char err_msg[512];
145 regerror(regerr, &tc->regex, regerr_msg,
146 sizeof(regerr_msg));
147 snprintf(err_msg, sizeof(err_msg), "regcomp: %s",
148 regerr_msg);
149 err = got_error_msg(GOT_ERR_REGEX, err_msg);
150 free(tc);
151 return err;
153 tc->colorpair = idx;
154 SIMPLEQ_INSERT_HEAD(colors, tc, entry);
155 return NULL;
158 static void
159 free_colors(struct tog_colors *colors)
161 struct tog_color *tc;
163 while (!SIMPLEQ_EMPTY(colors)) {
164 tc = SIMPLEQ_FIRST(colors);
165 SIMPLEQ_REMOVE_HEAD(colors, entry);
166 regfree(&tc->regex);
167 free(tc);
171 struct tog_color *
172 get_color(struct tog_colors *colors, int colorpair)
174 struct tog_color *tc = NULL;
176 SIMPLEQ_FOREACH(tc, colors, entry) {
177 if (tc->colorpair == colorpair)
178 return tc;
181 return NULL;
184 static int
185 default_color_value(const char *envvar)
187 if (strcmp(envvar, "TOG_COLOR_DIFF_MINUS") == 0)
188 return COLOR_MAGENTA;
189 if (strcmp(envvar, "TOG_COLOR_DIFF_PLUS") == 0)
190 return COLOR_CYAN;
191 if (strcmp(envvar, "TOG_COLOR_DIFF_CHUNK_HEADER") == 0)
192 return COLOR_YELLOW;
193 if (strcmp(envvar, "TOG_COLOR_DIFF_META") == 0)
194 return COLOR_GREEN;
195 if (strcmp(envvar, "TOG_COLOR_TREE_SUBMODULE") == 0)
196 return COLOR_MAGENTA;
197 if (strcmp(envvar, "TOG_COLOR_TREE_SYMLINK") == 0)
198 return COLOR_MAGENTA;
199 if (strcmp(envvar, "TOG_COLOR_TREE_DIRECTORY") == 0)
200 return COLOR_CYAN;
201 if (strcmp(envvar, "TOG_COLOR_TREE_EXECUTABLE") == 0)
202 return COLOR_GREEN;
203 if (strcmp(envvar, "TOG_COLOR_COMMIT") == 0)
204 return COLOR_GREEN;
205 if (strcmp(envvar, "TOG_COLOR_AUTHOR") == 0)
206 return COLOR_CYAN;
207 if (strcmp(envvar, "TOG_COLOR_DATE") == 0)
208 return COLOR_YELLOW;
210 return -1;
213 static int
214 get_color_value(const char *envvar)
216 const char *val = getenv(envvar);
218 if (val == NULL)
219 return default_color_value(envvar);
221 if (strcasecmp(val, "black") == 0)
222 return COLOR_BLACK;
223 if (strcasecmp(val, "red") == 0)
224 return COLOR_RED;
225 if (strcasecmp(val, "green") == 0)
226 return COLOR_GREEN;
227 if (strcasecmp(val, "yellow") == 0)
228 return COLOR_YELLOW;
229 if (strcasecmp(val, "blue") == 0)
230 return COLOR_BLUE;
231 if (strcasecmp(val, "magenta") == 0)
232 return COLOR_MAGENTA;
233 if (strcasecmp(val, "cyan") == 0)
234 return COLOR_CYAN;
235 if (strcasecmp(val, "white") == 0)
236 return COLOR_WHITE;
237 if (strcasecmp(val, "default") == 0)
238 return -1;
240 return default_color_value(envvar);
244 struct tog_diff_view_state {
245 struct got_object_id *id1, *id2;
246 FILE *f;
247 int first_displayed_line;
248 int last_displayed_line;
249 int eof;
250 int diff_context;
251 struct got_repository *repo;
252 struct got_reflist_head *refs;
253 struct tog_colors colors;
254 int nlines;
255 off_t *line_offsets;
256 int matched_line;
257 int selected_line;
258 size_t filesize;
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 int cmp;
1167 name = got_ref_get_name(re->ref);
1168 if (strcmp(name, GOT_REF_HEAD) == 0)
1169 continue;
1170 if (strncmp(name, "refs/", 5) == 0)
1171 name += 5;
1172 if (strncmp(name, "got/", 4) == 0)
1173 continue;
1174 if (strncmp(name, "heads/", 6) == 0)
1175 name += 6;
1176 if (strncmp(name, "remotes/", 8) == 0) {
1177 name += 8;
1178 s = strstr(name, "/" GOT_REF_HEAD);
1179 if (s != NULL && s[strlen(s)] == '\0')
1180 continue;
1182 if (strncmp(name, "tags/", 5) == 0) {
1183 err = got_object_open_as_tag(&tag, repo, re->id);
1184 if (err) {
1185 if (err->code != GOT_ERR_OBJ_TYPE)
1186 break;
1187 /* Ref points at something other than a tag. */
1188 err = NULL;
1189 tag = NULL;
1192 cmp = got_object_id_cmp(tag ?
1193 got_object_tag_get_object_id(tag) : re->id, id);
1194 if (tag)
1195 got_object_tag_close(tag);
1196 if (cmp != 0)
1197 continue;
1198 s = *refs_str;
1199 if (asprintf(refs_str, "%s%s%s", s ? s : "",
1200 s ? ", " : "", name) == -1) {
1201 err = got_error_from_errno("asprintf");
1202 free(s);
1203 *refs_str = NULL;
1204 break;
1206 free(s);
1209 return err;
1212 static const struct got_error *
1213 format_author(wchar_t **wauthor, int *author_width, char *author, int limit,
1214 int col_tab_align)
1216 char *smallerthan, *at;
1218 smallerthan = strchr(author, '<');
1219 if (smallerthan && smallerthan[1] != '\0')
1220 author = smallerthan + 1;
1221 at = strchr(author, '@');
1222 if (at)
1223 *at = '\0';
1224 return format_line(wauthor, author_width, author, limit, col_tab_align);
1227 static const struct got_error *
1228 draw_commit(struct tog_view *view, struct got_commit_object *commit,
1229 struct got_object_id *id, struct got_reflist_head *refs,
1230 const size_t date_display_cols, int author_display_cols,
1231 struct tog_colors *colors)
1233 const struct got_error *err = NULL;
1234 char datebuf[12]; /* YYYY-MM-DD + SPACE + NUL */
1235 char *logmsg0 = NULL, *logmsg = NULL;
1236 char *author = NULL;
1237 wchar_t *wlogmsg = NULL, *wauthor = NULL;
1238 int author_width, logmsg_width;
1239 char *newline, *line = NULL;
1240 int col, limit;
1241 const int avail = view->ncols;
1242 struct tm tm;
1243 time_t committer_time;
1244 struct tog_color *tc;
1246 committer_time = got_object_commit_get_committer_time(commit);
1247 if (localtime_r(&committer_time, &tm) == NULL)
1248 return got_error_from_errno("localtime_r");
1249 if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d ", &tm)
1250 >= sizeof(datebuf))
1251 return got_error(GOT_ERR_NO_SPACE);
1253 if (avail <= date_display_cols)
1254 limit = MIN(sizeof(datebuf) - 1, avail);
1255 else
1256 limit = MIN(date_display_cols, sizeof(datebuf) - 1);
1257 tc = get_color(colors, TOG_COLOR_DATE);
1258 if (tc)
1259 wattr_on(view->window,
1260 COLOR_PAIR(tc->colorpair), NULL);
1261 waddnstr(view->window, datebuf, limit);
1262 if (tc)
1263 wattr_off(view->window,
1264 COLOR_PAIR(tc->colorpair), NULL);
1265 col = limit;
1266 if (col > avail)
1267 goto done;
1269 if (avail >= 120) {
1270 char *id_str;
1271 err = got_object_id_str(&id_str, id);
1272 if (err)
1273 goto done;
1274 tc = get_color(colors, TOG_COLOR_COMMIT);
1275 if (tc)
1276 wattr_on(view->window,
1277 COLOR_PAIR(tc->colorpair), NULL);
1278 wprintw(view->window, "%.8s ", id_str);
1279 if (tc)
1280 wattr_off(view->window,
1281 COLOR_PAIR(tc->colorpair), NULL);
1282 free(id_str);
1283 col += 9;
1284 if (col > avail)
1285 goto done;
1288 author = strdup(got_object_commit_get_author(commit));
1289 if (author == NULL) {
1290 err = got_error_from_errno("strdup");
1291 goto done;
1293 err = format_author(&wauthor, &author_width, author, avail - col, col);
1294 if (err)
1295 goto done;
1296 tc = get_color(colors, TOG_COLOR_AUTHOR);
1297 if (tc)
1298 wattr_on(view->window,
1299 COLOR_PAIR(tc->colorpair), NULL);
1300 waddwstr(view->window, wauthor);
1301 if (tc)
1302 wattr_off(view->window,
1303 COLOR_PAIR(tc->colorpair), NULL);
1304 col += author_width;
1305 while (col < avail && author_width < author_display_cols + 2) {
1306 waddch(view->window, ' ');
1307 col++;
1308 author_width++;
1310 if (col > avail)
1311 goto done;
1313 err = got_object_commit_get_logmsg(&logmsg0, commit);
1314 if (err)
1315 goto done;
1316 logmsg = logmsg0;
1317 while (*logmsg == '\n')
1318 logmsg++;
1319 newline = strchr(logmsg, '\n');
1320 if (newline)
1321 *newline = '\0';
1322 limit = avail - col;
1323 err = format_line(&wlogmsg, &logmsg_width, logmsg, limit, col);
1324 if (err)
1325 goto done;
1326 waddwstr(view->window, wlogmsg);
1327 col += logmsg_width;
1328 while (col < avail) {
1329 waddch(view->window, ' ');
1330 col++;
1332 done:
1333 free(logmsg0);
1334 free(wlogmsg);
1335 free(author);
1336 free(wauthor);
1337 free(line);
1338 return err;
1341 static struct commit_queue_entry *
1342 alloc_commit_queue_entry(struct got_commit_object *commit,
1343 struct got_object_id *id)
1345 struct commit_queue_entry *entry;
1347 entry = calloc(1, sizeof(*entry));
1348 if (entry == NULL)
1349 return NULL;
1351 entry->id = id;
1352 entry->commit = commit;
1353 return entry;
1356 static void
1357 pop_commit(struct commit_queue *commits)
1359 struct commit_queue_entry *entry;
1361 entry = TAILQ_FIRST(&commits->head);
1362 TAILQ_REMOVE(&commits->head, entry, entry);
1363 got_object_commit_close(entry->commit);
1364 commits->ncommits--;
1365 /* Don't free entry->id! It is owned by the commit graph. */
1366 free(entry);
1369 static void
1370 free_commits(struct commit_queue *commits)
1372 while (!TAILQ_EMPTY(&commits->head))
1373 pop_commit(commits);
1376 static const struct got_error *
1377 match_commit(int *have_match, struct got_object_id *id,
1378 struct got_commit_object *commit, regex_t *regex)
1380 const struct got_error *err = NULL;
1381 regmatch_t regmatch;
1382 char *id_str = NULL, *logmsg = NULL;
1384 *have_match = 0;
1386 err = got_object_id_str(&id_str, id);
1387 if (err)
1388 return err;
1390 err = got_object_commit_get_logmsg(&logmsg, commit);
1391 if (err)
1392 goto done;
1394 if (regexec(regex, got_object_commit_get_author(commit), 1,
1395 &regmatch, 0) == 0 ||
1396 regexec(regex, got_object_commit_get_committer(commit), 1,
1397 &regmatch, 0) == 0 ||
1398 regexec(regex, id_str, 1, &regmatch, 0) == 0 ||
1399 regexec(regex, logmsg, 1, &regmatch, 0) == 0)
1400 *have_match = 1;
1401 done:
1402 free(id_str);
1403 free(logmsg);
1404 return err;
1407 static const struct got_error *
1408 queue_commits(struct got_commit_graph *graph, struct commit_queue *commits,
1409 int minqueue, struct got_repository *repo, const char *path,
1410 int *searching, int *search_next_done, regex_t *regex)
1412 const struct got_error *err = NULL;
1413 int nqueued = 0;
1416 * We keep all commits open throughout the lifetime of the log
1417 * view in order to avoid having to re-fetch commits from disk
1418 * while updating the display.
1420 while (nqueued < minqueue ||
1421 (*searching == TOG_SEARCH_FORWARD && !*search_next_done)) {
1422 struct got_object_id *id;
1423 struct got_commit_object *commit;
1424 struct commit_queue_entry *entry;
1425 int errcode;
1427 err = got_commit_graph_iter_next(&id, graph, repo, NULL, NULL);
1428 if (err || id == NULL)
1429 break;
1431 err = got_object_open_as_commit(&commit, repo, id);
1432 if (err)
1433 break;
1434 entry = alloc_commit_queue_entry(commit, id);
1435 if (entry == NULL) {
1436 err = got_error_from_errno("alloc_commit_queue_entry");
1437 break;
1440 errcode = pthread_mutex_lock(&tog_mutex);
1441 if (errcode) {
1442 err = got_error_set_errno(errcode,
1443 "pthread_mutex_lock");
1444 break;
1447 entry->idx = commits->ncommits;
1448 TAILQ_INSERT_TAIL(&commits->head, entry, entry);
1449 nqueued++;
1450 commits->ncommits++;
1452 if (*searching == TOG_SEARCH_FORWARD && !*search_next_done) {
1453 int have_match;
1454 err = match_commit(&have_match, id, commit, regex);
1455 if (err)
1456 break;
1457 if (have_match)
1458 *search_next_done = TOG_SEARCH_HAVE_MORE;
1461 errcode = pthread_mutex_unlock(&tog_mutex);
1462 if (errcode && err == NULL)
1463 err = got_error_set_errno(errcode,
1464 "pthread_mutex_unlock");
1465 if (err)
1466 break;
1469 return err;
1472 static const struct got_error *
1473 draw_commits(struct tog_view *view, struct commit_queue_entry **last,
1474 struct commit_queue_entry **selected, struct commit_queue_entry *first,
1475 struct commit_queue *commits, int selected_idx, int limit,
1476 struct got_reflist_head *refs, const char *path, int commits_needed,
1477 struct tog_colors *colors)
1479 const struct got_error *err = NULL;
1480 struct tog_log_view_state *s = &view->state.log;
1481 struct commit_queue_entry *entry;
1482 int width;
1483 int ncommits, author_cols = 4;
1484 char *id_str = NULL, *header = NULL, *ncommits_str = NULL;
1485 char *refs_str = NULL;
1486 wchar_t *wline;
1487 struct tog_color *tc;
1488 static const size_t date_display_cols = 12;
1490 entry = first;
1491 ncommits = 0;
1492 while (entry) {
1493 if (ncommits == selected_idx) {
1494 *selected = entry;
1495 break;
1497 entry = TAILQ_NEXT(entry, entry);
1498 ncommits++;
1501 if (*selected && !(view->searching && view->search_next_done == 0)) {
1502 err = got_object_id_str(&id_str, (*selected)->id);
1503 if (err)
1504 return err;
1505 if (refs) {
1506 err = build_refs_str(&refs_str, refs, (*selected)->id,
1507 s->repo);
1508 if (err)
1509 goto done;
1513 if (commits_needed == 0)
1514 halfdelay(10); /* disable fast refresh */
1516 if (commits_needed > 0) {
1517 if (asprintf(&ncommits_str, " [%d/%d] %s",
1518 entry ? entry->idx + 1 : 0, commits->ncommits,
1519 (view->searching && !view->search_next_done) ?
1520 "searching..." : "loading...") == -1) {
1521 err = got_error_from_errno("asprintf");
1522 goto done;
1524 } else {
1525 const char *search_str = NULL;
1527 if (view->searching) {
1528 if (view->search_next_done == TOG_SEARCH_NO_MORE)
1529 search_str = "no more matches";
1530 else if (view->search_next_done == TOG_SEARCH_HAVE_NONE)
1531 search_str = "no matches found";
1532 else if (!view->search_next_done)
1533 search_str = "searching...";
1536 if (asprintf(&ncommits_str, " [%d/%d] %s",
1537 entry ? entry->idx + 1 : 0, commits->ncommits,
1538 search_str ? search_str :
1539 (refs_str ? refs_str : "")) == -1) {
1540 err = got_error_from_errno("asprintf");
1541 goto done;
1545 if (path && strcmp(path, "/") != 0) {
1546 if (asprintf(&header, "commit %s %s%s",
1547 id_str ? id_str : "........................................",
1548 path, ncommits_str) == -1) {
1549 err = got_error_from_errno("asprintf");
1550 header = NULL;
1551 goto done;
1553 } else if (asprintf(&header, "commit %s%s",
1554 id_str ? id_str : "........................................",
1555 ncommits_str) == -1) {
1556 err = got_error_from_errno("asprintf");
1557 header = NULL;
1558 goto done;
1560 err = format_line(&wline, &width, header, view->ncols, 0);
1561 if (err)
1562 goto done;
1564 werase(view->window);
1566 if (view_needs_focus_indication(view))
1567 wstandout(view->window);
1568 tc = get_color(colors, TOG_COLOR_COMMIT);
1569 if (tc)
1570 wattr_on(view->window,
1571 COLOR_PAIR(tc->colorpair), NULL);
1572 waddwstr(view->window, wline);
1573 if (tc)
1574 wattr_off(view->window,
1575 COLOR_PAIR(tc->colorpair), NULL);
1576 while (width < view->ncols) {
1577 waddch(view->window, ' ');
1578 width++;
1580 if (view_needs_focus_indication(view))
1581 wstandend(view->window);
1582 free(wline);
1583 if (limit <= 1)
1584 goto done;
1586 /* Grow author column size if necessary. */
1587 entry = first;
1588 ncommits = 0;
1589 while (entry) {
1590 char *author;
1591 wchar_t *wauthor;
1592 int width;
1593 if (ncommits >= limit - 1)
1594 break;
1595 author = strdup(got_object_commit_get_author(entry->commit));
1596 if (author == NULL) {
1597 err = got_error_from_errno("strdup");
1598 goto done;
1600 err = format_author(&wauthor, &width, author, COLS,
1601 date_display_cols);
1602 if (author_cols < width)
1603 author_cols = width;
1604 free(wauthor);
1605 free(author);
1606 ncommits++;
1607 entry = TAILQ_NEXT(entry, entry);
1610 entry = first;
1611 *last = first;
1612 ncommits = 0;
1613 while (entry) {
1614 if (ncommits >= limit - 1)
1615 break;
1616 if (ncommits == selected_idx)
1617 wstandout(view->window);
1618 err = draw_commit(view, entry->commit, entry->id, refs,
1619 date_display_cols, author_cols, colors);
1620 if (ncommits == selected_idx)
1621 wstandend(view->window);
1622 if (err)
1623 goto done;
1624 ncommits++;
1625 *last = entry;
1626 entry = TAILQ_NEXT(entry, entry);
1629 view_vborder(view);
1630 done:
1631 free(id_str);
1632 free(refs_str);
1633 free(ncommits_str);
1634 free(header);
1635 return err;
1638 static void
1639 scroll_up(struct tog_view *view,
1640 struct commit_queue_entry **first_displayed_entry, int maxscroll,
1641 struct commit_queue *commits)
1643 struct commit_queue_entry *entry;
1644 int nscrolled = 0;
1646 entry = TAILQ_FIRST(&commits->head);
1647 if (*first_displayed_entry == entry)
1648 return;
1650 entry = *first_displayed_entry;
1651 while (entry && nscrolled < maxscroll) {
1652 entry = TAILQ_PREV(entry, commit_queue_head, entry);
1653 if (entry) {
1654 *first_displayed_entry = entry;
1655 nscrolled++;
1660 static const struct got_error *
1661 trigger_log_thread(struct tog_view *log_view, int wait,
1662 int *commits_needed, int *log_complete,
1663 pthread_cond_t *need_commits, pthread_cond_t *commit_loaded)
1665 int errcode;
1667 halfdelay(1); /* fast refresh while loading commits */
1669 while (*commits_needed > 0) {
1670 if (*log_complete)
1671 break;
1673 /* Wake the log thread. */
1674 errcode = pthread_cond_signal(need_commits);
1675 if (errcode)
1676 return got_error_set_errno(errcode,
1677 "pthread_cond_signal");
1680 * The mutex will be released while the view loop waits
1681 * in wgetch(), at which time the log thread will run.
1683 if (!wait)
1684 break;
1686 /* Display progress update in log view. */
1687 show_log_view(log_view);
1688 update_panels();
1689 doupdate();
1691 /* Wait right here while next commit is being loaded. */
1692 errcode = pthread_cond_wait(commit_loaded, &tog_mutex);
1693 if (errcode)
1694 return got_error_set_errno(errcode,
1695 "pthread_cond_wait");
1697 /* Display progress update in log view. */
1698 show_log_view(log_view);
1699 update_panels();
1700 doupdate();
1703 return NULL;
1706 static const struct got_error *
1707 scroll_down(struct tog_view *log_view,
1708 struct commit_queue_entry **first_displayed_entry, int maxscroll,
1709 struct commit_queue_entry **last_displayed_entry,
1710 struct commit_queue *commits, int *log_complete, int *commits_needed,
1711 pthread_cond_t *need_commits, pthread_cond_t *commit_loaded)
1713 const struct got_error *err = NULL;
1714 struct commit_queue_entry *pentry;
1715 int nscrolled = 0, ncommits_needed;
1717 if (*last_displayed_entry == NULL)
1718 return NULL;
1720 ncommits_needed = (*last_displayed_entry)->idx + 1 + maxscroll;
1721 if (commits->ncommits < ncommits_needed && !*log_complete) {
1723 * Ask the log thread for required amount of commits.
1725 (*commits_needed) += maxscroll;
1726 err = trigger_log_thread(log_view, 1, commits_needed,
1727 log_complete, need_commits, commit_loaded);
1728 if (err)
1729 return err;
1732 do {
1733 pentry = TAILQ_NEXT(*last_displayed_entry, entry);
1734 if (pentry == NULL)
1735 break;
1737 *last_displayed_entry = pentry;
1739 pentry = TAILQ_NEXT(*first_displayed_entry, entry);
1740 if (pentry == NULL)
1741 break;
1742 *first_displayed_entry = pentry;
1743 } while (++nscrolled < maxscroll);
1745 return err;
1748 static const struct got_error *
1749 open_diff_view_for_commit(struct tog_view **new_view, int begin_x,
1750 struct got_commit_object *commit, struct got_object_id *commit_id,
1751 struct tog_view *log_view, struct got_reflist_head *refs,
1752 struct got_repository *repo)
1754 const struct got_error *err;
1755 struct got_object_qid *parent_id;
1756 struct tog_view *diff_view;
1758 diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
1759 if (diff_view == NULL)
1760 return got_error_from_errno("view_open");
1762 parent_id = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
1763 err = open_diff_view(diff_view, parent_id ? parent_id->id : NULL,
1764 commit_id, log_view, refs, repo);
1765 if (err == NULL)
1766 *new_view = diff_view;
1767 return err;
1770 static const struct got_error *
1771 tree_view_visit_subtree(struct got_tree_object *subtree,
1772 struct tog_tree_view_state *s)
1774 struct tog_parent_tree *parent;
1776 parent = calloc(1, sizeof(*parent));
1777 if (parent == NULL)
1778 return got_error_from_errno("calloc");
1780 parent->tree = s->tree;
1781 parent->first_displayed_entry = s->first_displayed_entry;
1782 parent->selected_entry = s->selected_entry;
1783 parent->selected = s->selected;
1784 TAILQ_INSERT_HEAD(&s->parents, parent, entry);
1785 s->tree = subtree;
1786 s->selected = 0;
1787 s->first_displayed_entry = NULL;
1788 return NULL;
1791 static const struct got_error *
1792 tree_view_walk_path(struct tog_tree_view_state *s,
1793 struct got_object_id *commit_id,
1794 const char *path, struct got_repository *repo)
1796 const struct got_error *err = NULL;
1797 struct got_tree_object *tree = NULL;
1798 const char *p;
1799 char *slash, *subpath = NULL;
1801 /* Walk the path and open corresponding tree objects. */
1802 p = path;
1803 while (*p) {
1804 struct got_tree_entry *te;
1805 struct got_object_id *tree_id;
1806 char *te_name;
1808 while (p[0] == '/')
1809 p++;
1811 /* Ensure the correct subtree entry is selected. */
1812 slash = strchr(p, '/');
1813 if (slash == NULL)
1814 te_name = strdup(p);
1815 else
1816 te_name = strndup(p, slash - p);
1817 if (te_name == NULL) {
1818 err = got_error_from_errno("strndup");
1819 break;
1821 te = got_object_tree_find_entry(s->tree, te_name);
1822 if (te == NULL) {
1823 err = got_error_path(te_name, GOT_ERR_NO_TREE_ENTRY);
1824 free(te_name);
1825 break;
1827 free(te_name);
1828 s->selected_entry = te;
1829 s->selected = got_tree_entry_get_index(te);
1830 if (s->tree != s->root)
1831 s->selected++; /* skip '..' */
1833 if (!S_ISDIR(got_tree_entry_get_mode(s->selected_entry))) {
1834 /* Jump to this file's entry. */
1835 s->first_displayed_entry = s->selected_entry;
1836 s->selected = 0;
1837 break;
1840 slash = strchr(p, '/');
1841 if (slash)
1842 subpath = strndup(path, slash - path);
1843 else
1844 subpath = strdup(path);
1845 if (subpath == NULL) {
1846 err = got_error_from_errno("strdup");
1847 break;
1850 err = got_object_id_by_path(&tree_id, repo, commit_id,
1851 subpath);
1852 if (err)
1853 break;
1855 err = got_object_open_as_tree(&tree, repo, tree_id);
1856 free(tree_id);
1857 if (err)
1858 break;
1860 err = tree_view_visit_subtree(tree, s);
1861 if (err) {
1862 got_object_tree_close(tree);
1863 break;
1865 if (slash == NULL)
1866 break;
1867 free(subpath);
1868 subpath = NULL;
1869 p = slash;
1872 free(subpath);
1873 return err;
1876 static const struct got_error *
1877 browse_commit_tree(struct tog_view **new_view, int begin_x,
1878 struct commit_queue_entry *entry, const char *path,
1879 struct got_reflist_head *refs, struct got_repository *repo)
1881 const struct got_error *err = NULL;
1882 struct got_tree_object *tree;
1883 struct tog_tree_view_state *s;
1884 struct tog_view *tree_view;
1886 err = got_object_open_as_tree(&tree, repo,
1887 got_object_commit_get_tree_id(entry->commit));
1888 if (err)
1889 return err;
1891 tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
1892 if (tree_view == NULL)
1893 return got_error_from_errno("view_open");
1895 err = open_tree_view(tree_view, tree, entry->id, refs, repo);
1896 if (err) {
1897 got_object_tree_close(tree);
1898 return err;
1900 s = &tree_view->state.tree;
1902 *new_view = tree_view;
1904 if (got_path_is_root_dir(path))
1905 return NULL;
1907 return tree_view_walk_path(s, entry->id, path, repo);
1910 static const struct got_error *
1911 block_signals_used_by_main_thread(void)
1913 sigset_t sigset;
1914 int errcode;
1916 if (sigemptyset(&sigset) == -1)
1917 return got_error_from_errno("sigemptyset");
1919 /* tog handles SIGWINCH and SIGCONT */
1920 if (sigaddset(&sigset, SIGWINCH) == -1)
1921 return got_error_from_errno("sigaddset");
1922 if (sigaddset(&sigset, SIGCONT) == -1)
1923 return got_error_from_errno("sigaddset");
1925 /* ncurses handles SIGTSTP */
1926 if (sigaddset(&sigset, SIGTSTP) == -1)
1927 return got_error_from_errno("sigaddset");
1929 errcode = pthread_sigmask(SIG_BLOCK, &sigset, NULL);
1930 if (errcode)
1931 return got_error_set_errno(errcode, "pthread_sigmask");
1933 return NULL;
1936 static void *
1937 log_thread(void *arg)
1939 const struct got_error *err = NULL;
1940 int errcode = 0;
1941 struct tog_log_thread_args *a = arg;
1942 int done = 0;
1944 err = block_signals_used_by_main_thread();
1945 if (err)
1946 return (void *)err;
1948 while (!done && !err && !tog_sigpipe_received) {
1949 err = queue_commits(a->graph, a->commits, 1, a->repo,
1950 a->in_repo_path, a->searching, a->search_next_done,
1951 a->regex);
1952 if (err) {
1953 if (err->code != GOT_ERR_ITER_COMPLETED)
1954 return (void *)err;
1955 err = NULL;
1956 done = 1;
1957 } else if (a->commits_needed > 0)
1958 a->commits_needed--;
1960 errcode = pthread_mutex_lock(&tog_mutex);
1961 if (errcode) {
1962 err = got_error_set_errno(errcode,
1963 "pthread_mutex_lock");
1964 break;
1965 } else if (*a->quit)
1966 done = 1;
1967 else if (*a->first_displayed_entry == NULL) {
1968 *a->first_displayed_entry =
1969 TAILQ_FIRST(&a->commits->head);
1970 *a->selected_entry = *a->first_displayed_entry;
1973 errcode = pthread_cond_signal(&a->commit_loaded);
1974 if (errcode) {
1975 err = got_error_set_errno(errcode,
1976 "pthread_cond_signal");
1977 pthread_mutex_unlock(&tog_mutex);
1978 break;
1981 if (done)
1982 a->commits_needed = 0;
1983 else {
1984 if (a->commits_needed == 0) {
1985 errcode = pthread_cond_wait(&a->need_commits,
1986 &tog_mutex);
1987 if (errcode)
1988 err = got_error_set_errno(errcode,
1989 "pthread_cond_wait");
1993 errcode = pthread_mutex_unlock(&tog_mutex);
1994 if (errcode && err == NULL)
1995 err = got_error_set_errno(errcode,
1996 "pthread_mutex_unlock");
1998 a->log_complete = 1;
1999 return (void *)err;
2002 static const struct got_error *
2003 stop_log_thread(struct tog_log_view_state *s)
2005 const struct got_error *err = NULL;
2006 int errcode;
2008 if (s->thread) {
2009 s->quit = 1;
2010 errcode = pthread_cond_signal(&s->thread_args.need_commits);
2011 if (errcode)
2012 return got_error_set_errno(errcode,
2013 "pthread_cond_signal");
2014 errcode = pthread_mutex_unlock(&tog_mutex);
2015 if (errcode)
2016 return got_error_set_errno(errcode,
2017 "pthread_mutex_unlock");
2018 errcode = pthread_join(s->thread, (void **)&err);
2019 if (errcode)
2020 return got_error_set_errno(errcode, "pthread_join");
2021 errcode = pthread_mutex_lock(&tog_mutex);
2022 if (errcode)
2023 return got_error_set_errno(errcode,
2024 "pthread_mutex_lock");
2025 s->thread = NULL;
2028 errcode = pthread_cond_destroy(&s->thread_args.need_commits);
2029 if (errcode && err == NULL)
2030 err = got_error_set_errno(errcode, "pthread_cond_destroy");
2032 if (s->thread_args.repo) {
2033 got_repo_close(s->thread_args.repo);
2034 s->thread_args.repo = NULL;
2037 if (s->thread_args.graph) {
2038 got_commit_graph_close(s->thread_args.graph);
2039 s->thread_args.graph = NULL;
2042 return err;
2045 static const struct got_error *
2046 close_log_view(struct tog_view *view)
2048 const struct got_error *err = NULL;
2049 struct tog_log_view_state *s = &view->state.log;
2051 err = stop_log_thread(s);
2052 free_commits(&s->commits);
2053 free(s->in_repo_path);
2054 s->in_repo_path = NULL;
2055 free(s->start_id);
2056 s->start_id = NULL;
2057 return err;
2060 static const struct got_error *
2061 search_start_log_view(struct tog_view *view)
2063 struct tog_log_view_state *s = &view->state.log;
2065 s->matched_entry = NULL;
2066 s->search_entry = NULL;
2067 return NULL;
2070 static const struct got_error *
2071 search_next_log_view(struct tog_view *view)
2073 const struct got_error *err = NULL;
2074 struct tog_log_view_state *s = &view->state.log;
2075 struct commit_queue_entry *entry;
2077 /* Display progress update in log view. */
2078 show_log_view(view);
2079 update_panels();
2080 doupdate();
2082 if (s->search_entry) {
2083 int errcode, ch;
2084 errcode = pthread_mutex_unlock(&tog_mutex);
2085 if (errcode)
2086 return got_error_set_errno(errcode,
2087 "pthread_mutex_unlock");
2088 ch = wgetch(view->window);
2089 errcode = pthread_mutex_lock(&tog_mutex);
2090 if (errcode)
2091 return got_error_set_errno(errcode,
2092 "pthread_mutex_lock");
2093 if (ch == KEY_BACKSPACE) {
2094 view->search_next_done = TOG_SEARCH_HAVE_MORE;
2095 return NULL;
2097 if (view->searching == TOG_SEARCH_FORWARD)
2098 entry = TAILQ_NEXT(s->search_entry, entry);
2099 else
2100 entry = TAILQ_PREV(s->search_entry,
2101 commit_queue_head, entry);
2102 } else if (s->matched_entry) {
2103 if (view->searching == TOG_SEARCH_FORWARD)
2104 entry = TAILQ_NEXT(s->matched_entry, entry);
2105 else
2106 entry = TAILQ_PREV(s->matched_entry,
2107 commit_queue_head, entry);
2108 } else {
2109 if (view->searching == TOG_SEARCH_FORWARD)
2110 entry = TAILQ_FIRST(&s->commits.head);
2111 else
2112 entry = TAILQ_LAST(&s->commits.head, commit_queue_head);
2115 while (1) {
2116 int have_match = 0;
2118 if (entry == NULL) {
2119 if (s->thread_args.log_complete ||
2120 view->searching == TOG_SEARCH_BACKWARD) {
2121 view->search_next_done =
2122 (s->matched_entry == NULL ?
2123 TOG_SEARCH_HAVE_NONE : TOG_SEARCH_NO_MORE);
2124 s->search_entry = NULL;
2125 return NULL;
2128 * Poke the log thread for more commits and return,
2129 * allowing the main loop to make progress. Search
2130 * will resume at s->search_entry once we come back.
2132 s->thread_args.commits_needed++;
2133 return trigger_log_thread(view, 0,
2134 &s->thread_args.commits_needed,
2135 &s->thread_args.log_complete,
2136 &s->thread_args.need_commits,
2137 &s->thread_args.commit_loaded);
2140 err = match_commit(&have_match, entry->id, entry->commit,
2141 &view->regex);
2142 if (err)
2143 break;
2144 if (have_match) {
2145 view->search_next_done = TOG_SEARCH_HAVE_MORE;
2146 s->matched_entry = entry;
2147 break;
2150 s->search_entry = entry;
2151 if (view->searching == TOG_SEARCH_FORWARD)
2152 entry = TAILQ_NEXT(entry, entry);
2153 else
2154 entry = TAILQ_PREV(entry, commit_queue_head, entry);
2157 if (s->matched_entry) {
2158 int cur = s->selected_entry->idx;
2159 while (cur < s->matched_entry->idx) {
2160 err = input_log_view(NULL, NULL, NULL, view, KEY_DOWN);
2161 if (err)
2162 return err;
2163 cur++;
2165 while (cur > s->matched_entry->idx) {
2166 err = input_log_view(NULL, NULL, NULL, view, KEY_UP);
2167 if (err)
2168 return err;
2169 cur--;
2173 s->search_entry = NULL;
2175 return NULL;
2178 static const struct got_error *
2179 open_log_view(struct tog_view *view, struct got_object_id *start_id,
2180 struct got_reflist_head *refs, struct got_repository *repo,
2181 const char *head_ref_name, const char *in_repo_path,
2182 int log_branches)
2184 const struct got_error *err = NULL;
2185 struct tog_log_view_state *s = &view->state.log;
2186 struct got_repository *thread_repo = NULL;
2187 struct got_commit_graph *thread_graph = NULL;
2188 int errcode;
2190 if (in_repo_path != s->in_repo_path) {
2191 free(s->in_repo_path);
2192 s->in_repo_path = strdup(in_repo_path);
2193 if (s->in_repo_path == NULL)
2194 return got_error_from_errno("strdup");
2197 /* The commit queue only contains commits being displayed. */
2198 TAILQ_INIT(&s->commits.head);
2199 s->commits.ncommits = 0;
2201 s->refs = refs;
2202 s->repo = repo;
2203 s->head_ref_name = head_ref_name;
2204 s->start_id = got_object_id_dup(start_id);
2205 if (s->start_id == NULL) {
2206 err = got_error_from_errno("got_object_id_dup");
2207 goto done;
2209 s->log_branches = log_branches;
2211 SIMPLEQ_INIT(&s->colors);
2212 if (has_colors() && getenv("TOG_COLORS") != NULL) {
2213 err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
2214 get_color_value("TOG_COLOR_COMMIT"));
2215 if (err)
2216 goto done;
2217 err = add_color(&s->colors, "^$", TOG_COLOR_AUTHOR,
2218 get_color_value("TOG_COLOR_AUTHOR"));
2219 if (err) {
2220 free_colors(&s->colors);
2221 goto done;
2223 err = add_color(&s->colors, "^$", TOG_COLOR_DATE,
2224 get_color_value("TOG_COLOR_DATE"));
2225 if (err) {
2226 free_colors(&s->colors);
2227 goto done;
2231 view->show = show_log_view;
2232 view->input = input_log_view;
2233 view->close = close_log_view;
2234 view->search_start = search_start_log_view;
2235 view->search_next = search_next_log_view;
2237 err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL);
2238 if (err)
2239 goto done;
2240 err = got_commit_graph_open(&thread_graph, s->in_repo_path,
2241 !s->log_branches);
2242 if (err)
2243 goto done;
2244 err = got_commit_graph_iter_start(thread_graph,
2245 s->start_id, s->repo, NULL, NULL);
2246 if (err)
2247 goto done;
2249 errcode = pthread_cond_init(&s->thread_args.need_commits, NULL);
2250 if (errcode) {
2251 err = got_error_set_errno(errcode, "pthread_cond_init");
2252 goto done;
2254 errcode = pthread_cond_init(&s->thread_args.commit_loaded, NULL);
2255 if (errcode) {
2256 err = got_error_set_errno(errcode, "pthread_cond_init");
2257 goto done;
2260 s->thread_args.commits_needed = view->nlines;
2261 s->thread_args.graph = thread_graph;
2262 s->thread_args.commits = &s->commits;
2263 s->thread_args.in_repo_path = s->in_repo_path;
2264 s->thread_args.start_id = s->start_id;
2265 s->thread_args.repo = thread_repo;
2266 s->thread_args.log_complete = 0;
2267 s->thread_args.quit = &s->quit;
2268 s->thread_args.first_displayed_entry = &s->first_displayed_entry;
2269 s->thread_args.selected_entry = &s->selected_entry;
2270 s->thread_args.searching = &view->searching;
2271 s->thread_args.search_next_done = &view->search_next_done;
2272 s->thread_args.regex = &view->regex;
2273 done:
2274 if (err)
2275 close_log_view(view);
2276 return err;
2279 static const struct got_error *
2280 show_log_view(struct tog_view *view)
2282 struct tog_log_view_state *s = &view->state.log;
2284 if (s->thread == NULL) {
2285 int errcode = pthread_create(&s->thread, NULL, log_thread,
2286 &s->thread_args);
2287 if (errcode)
2288 return got_error_set_errno(errcode, "pthread_create");
2291 return draw_commits(view, &s->last_displayed_entry,
2292 &s->selected_entry, s->first_displayed_entry,
2293 &s->commits, s->selected, view->nlines, s->refs,
2294 s->in_repo_path, s->thread_args.commits_needed, &s->colors);
2297 static const struct got_error *
2298 input_log_view(struct tog_view **new_view, struct tog_view **dead_view,
2299 struct tog_view **focus_view, struct tog_view *view, int ch)
2301 const struct got_error *err = NULL;
2302 struct tog_log_view_state *s = &view->state.log;
2303 char *parent_path, *in_repo_path = NULL;
2304 struct tog_view *diff_view = NULL, *tree_view = NULL, *lv = NULL;
2305 int begin_x = 0;
2306 struct got_object_id *start_id;
2308 switch (ch) {
2309 case 'q':
2310 s->quit = 1;
2311 break;
2312 case 'k':
2313 case KEY_UP:
2314 case '<':
2315 case ',':
2316 if (s->first_displayed_entry == NULL)
2317 break;
2318 if (s->selected > 0)
2319 s->selected--;
2320 else
2321 scroll_up(view, &s->first_displayed_entry, 1,
2322 &s->commits);
2323 break;
2324 case KEY_PPAGE:
2325 case CTRL('b'):
2326 if (s->first_displayed_entry == NULL)
2327 break;
2328 if (TAILQ_FIRST(&s->commits.head) ==
2329 s->first_displayed_entry) {
2330 s->selected = 0;
2331 break;
2333 scroll_up(view, &s->first_displayed_entry,
2334 view->nlines - 1, &s->commits);
2335 break;
2336 case 'j':
2337 case KEY_DOWN:
2338 case '>':
2339 case '.':
2340 if (s->first_displayed_entry == NULL)
2341 break;
2342 if (s->selected < MIN(view->nlines - 2,
2343 s->commits.ncommits - 1)) {
2344 s->selected++;
2345 break;
2347 err = scroll_down(view, &s->first_displayed_entry, 1,
2348 &s->last_displayed_entry, &s->commits,
2349 &s->thread_args.log_complete,
2350 &s->thread_args.commits_needed,
2351 &s->thread_args.need_commits,
2352 &s->thread_args.commit_loaded);
2353 break;
2354 case KEY_NPAGE:
2355 case CTRL('f'): {
2356 struct commit_queue_entry *first;
2357 first = s->first_displayed_entry;
2358 if (first == NULL)
2359 break;
2360 err = scroll_down(view, &s->first_displayed_entry,
2361 view->nlines - 1, &s->last_displayed_entry,
2362 &s->commits, &s->thread_args.log_complete,
2363 &s->thread_args.commits_needed,
2364 &s->thread_args.need_commits,
2365 &s->thread_args.commit_loaded);
2366 if (err)
2367 break;
2368 if (first == s->first_displayed_entry &&
2369 s->selected < MIN(view->nlines - 2,
2370 s->commits.ncommits - 1)) {
2371 /* can't scroll further down */
2372 s->selected = MIN(view->nlines - 2,
2373 s->commits.ncommits - 1);
2375 err = NULL;
2376 break;
2378 case KEY_RESIZE:
2379 if (s->selected > view->nlines - 2)
2380 s->selected = view->nlines - 2;
2381 if (s->selected > s->commits.ncommits - 1)
2382 s->selected = s->commits.ncommits - 1;
2383 break;
2384 case KEY_ENTER:
2385 case ' ':
2386 case '\r':
2387 if (s->selected_entry == NULL)
2388 break;
2389 if (view_is_parent_view(view))
2390 begin_x = view_split_begin_x(view->begin_x);
2391 err = open_diff_view_for_commit(&diff_view, begin_x,
2392 s->selected_entry->commit, s->selected_entry->id,
2393 view, s->refs, s->repo);
2394 if (err)
2395 break;
2396 if (view_is_parent_view(view)) {
2397 err = view_close_child(view);
2398 if (err)
2399 return err;
2400 err = view_set_child(view, diff_view);
2401 if (err) {
2402 view_close(diff_view);
2403 break;
2405 *focus_view = diff_view;
2406 view->child_focussed = 1;
2407 } else
2408 *new_view = diff_view;
2409 break;
2410 case 't':
2411 if (s->selected_entry == NULL)
2412 break;
2413 if (view_is_parent_view(view))
2414 begin_x = view_split_begin_x(view->begin_x);
2415 err = browse_commit_tree(&tree_view, begin_x,
2416 s->selected_entry, s->in_repo_path, s->refs, s->repo);
2417 if (err)
2418 break;
2419 if (view_is_parent_view(view)) {
2420 err = view_close_child(view);
2421 if (err)
2422 return err;
2423 err = view_set_child(view, tree_view);
2424 if (err) {
2425 view_close(tree_view);
2426 break;
2428 *focus_view = tree_view;
2429 view->child_focussed = 1;
2430 } else
2431 *new_view = tree_view;
2432 break;
2433 case KEY_BACKSPACE:
2434 if (strcmp(s->in_repo_path, "/") == 0)
2435 break;
2436 parent_path = dirname(s->in_repo_path);
2437 if (parent_path && strcmp(parent_path, ".") != 0) {
2438 err = stop_log_thread(s);
2439 if (err)
2440 return err;
2441 lv = view_open(view->nlines, view->ncols,
2442 view->begin_y, view->begin_x, TOG_VIEW_LOG);
2443 if (lv == NULL)
2444 return got_error_from_errno(
2445 "view_open");
2446 err = open_log_view(lv, s->start_id, s->refs,
2447 s->repo, s->head_ref_name, parent_path,
2448 s->log_branches);
2449 if (err)
2450 return err;;
2451 if (view_is_parent_view(view))
2452 *new_view = lv;
2453 else {
2454 view_set_child(view->parent, lv);
2455 *focus_view = lv;
2457 return NULL;
2459 break;
2460 case CTRL('l'):
2461 err = stop_log_thread(s);
2462 if (err)
2463 return err;
2464 lv = view_open(view->nlines, view->ncols,
2465 view->begin_y, view->begin_x, TOG_VIEW_LOG);
2466 if (lv == NULL)
2467 return got_error_from_errno("view_open");
2468 err = got_repo_match_object_id(&start_id, NULL,
2469 s->head_ref_name ? s->head_ref_name : GOT_REF_HEAD,
2470 GOT_OBJ_TYPE_COMMIT, 1, s->repo);
2471 if (err) {
2472 view_close(lv);
2473 return err;
2475 in_repo_path = strdup(s->in_repo_path);
2476 if (in_repo_path == NULL) {
2477 free(start_id);
2478 view_close(lv);
2479 return got_error_from_errno("strdup");
2481 got_ref_list_free(s->refs);
2482 err = got_ref_list(s->refs, s->repo, NULL,
2483 got_ref_cmp_by_name, NULL);
2484 if (err) {
2485 free(start_id);
2486 view_close(lv);
2487 return err;
2489 err = open_log_view(lv, start_id, s->refs, s->repo,
2490 s->head_ref_name, in_repo_path, s->log_branches);
2491 if (err) {
2492 free(start_id);
2493 view_close(lv);
2494 return err;;
2496 *dead_view = view;
2497 *new_view = lv;
2498 break;
2499 case 'B':
2500 s->log_branches = !s->log_branches;
2501 err = stop_log_thread(s);
2502 if (err)
2503 return err;
2504 lv = view_open(view->nlines, view->ncols,
2505 view->begin_y, view->begin_x, TOG_VIEW_LOG);
2506 if (lv == NULL)
2507 return got_error_from_errno("view_open");
2508 err = open_log_view(lv, s->start_id, s->refs, s->repo,
2509 s->head_ref_name, s->in_repo_path, s->log_branches);
2510 if (err) {
2511 view_close(lv);
2512 return err;;
2514 *dead_view = view;
2515 *new_view = lv;
2516 break;
2517 default:
2518 break;
2521 return err;
2524 static const struct got_error *
2525 apply_unveil(const char *repo_path, const char *worktree_path)
2527 const struct got_error *error;
2529 #ifdef PROFILE
2530 if (unveil("gmon.out", "rwc") != 0)
2531 return got_error_from_errno2("unveil", "gmon.out");
2532 #endif
2533 if (repo_path && unveil(repo_path, "r") != 0)
2534 return got_error_from_errno2("unveil", repo_path);
2536 if (worktree_path && unveil(worktree_path, "rwc") != 0)
2537 return got_error_from_errno2("unveil", worktree_path);
2539 if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
2540 return got_error_from_errno2("unveil", GOT_TMPDIR_STR);
2542 error = got_privsep_unveil_exec_helpers();
2543 if (error != NULL)
2544 return error;
2546 if (unveil(NULL, NULL) != 0)
2547 return got_error_from_errno("unveil");
2549 return NULL;
2552 static void
2553 init_curses(void)
2555 initscr();
2556 cbreak();
2557 halfdelay(1); /* Do fast refresh while initial view is loading. */
2558 noecho();
2559 nonl();
2560 intrflush(stdscr, FALSE);
2561 keypad(stdscr, TRUE);
2562 curs_set(0);
2563 if (getenv("TOG_COLORS") != NULL) {
2564 start_color();
2565 use_default_colors();
2567 signal(SIGWINCH, tog_sigwinch);
2568 signal(SIGPIPE, tog_sigpipe);
2569 signal(SIGCONT, tog_sigcont);
2572 static const struct got_error *
2573 get_in_repo_path_from_argv0(char **in_repo_path, int argc, char *argv[],
2574 struct got_repository *repo, struct got_worktree *worktree)
2576 const struct got_error *err = NULL;
2578 if (argc == 0) {
2579 *in_repo_path = strdup("/");
2580 if (*in_repo_path == NULL)
2581 return got_error_from_errno("strdup");
2582 return NULL;
2585 if (worktree) {
2586 const char *prefix = got_worktree_get_path_prefix(worktree);
2587 char *wt_path, *p;
2589 err = got_worktree_resolve_path(&wt_path, worktree, argv[0]);
2590 if (err)
2591 return err;
2593 if (asprintf(&p, "%s%s%s", prefix,
2594 (strcmp(prefix, "/") != 0) ? "/" : "", wt_path) == -1) {
2595 err = got_error_from_errno("asprintf");
2596 free(wt_path);
2597 return err;
2599 err = got_repo_map_path(in_repo_path, repo, p, 0);
2600 free(p);
2601 free(wt_path);
2602 } else
2603 err = got_repo_map_path(in_repo_path, repo, argv[0], 1);
2605 return err;
2608 static const struct got_error *
2609 cmd_log(int argc, char *argv[])
2611 const struct got_error *error;
2612 struct got_repository *repo = NULL;
2613 struct got_worktree *worktree = NULL;
2614 struct got_reflist_head refs;
2615 struct got_object_id *start_id = NULL;
2616 char *in_repo_path = NULL, *repo_path = NULL, *cwd = NULL;
2617 char *start_commit = NULL, *head_ref_name = NULL;
2618 int ch, log_branches = 0;
2619 struct tog_view *view;
2621 SIMPLEQ_INIT(&refs);
2623 #ifndef PROFILE
2624 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
2625 NULL) == -1)
2626 err(1, "pledge");
2627 #endif
2629 while ((ch = getopt(argc, argv, "bc:r:")) != -1) {
2630 switch (ch) {
2631 case 'b':
2632 log_branches = 1;
2633 break;
2634 case 'c':
2635 start_commit = optarg;
2636 break;
2637 case 'r':
2638 repo_path = realpath(optarg, NULL);
2639 if (repo_path == NULL)
2640 return got_error_from_errno2("realpath",
2641 optarg);
2642 break;
2643 default:
2644 usage_log();
2645 /* NOTREACHED */
2649 argc -= optind;
2650 argv += optind;
2652 if (argc > 1)
2653 usage_log();
2655 cwd = getcwd(NULL, 0);
2656 if (cwd == NULL)
2657 return got_error_from_errno("getcwd");
2659 error = got_worktree_open(&worktree, cwd);
2660 if (error && error->code != GOT_ERR_NOT_WORKTREE)
2661 goto done;
2663 if (repo_path == NULL) {
2664 if (worktree)
2665 repo_path =
2666 strdup(got_worktree_get_repo_path(worktree));
2667 else
2668 repo_path = strdup(cwd);
2670 if (repo_path == NULL) {
2671 error = got_error_from_errno("strdup");
2672 goto done;
2675 error = got_repo_open(&repo, repo_path, NULL);
2676 if (error != NULL)
2677 goto done;
2679 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
2680 repo, worktree);
2681 if (error)
2682 goto done;
2684 init_curses();
2686 error = apply_unveil(got_repo_get_path(repo),
2687 worktree ? got_worktree_get_root_path(worktree) : NULL);
2688 if (error)
2689 goto done;
2691 if (start_commit == NULL)
2692 error = got_repo_match_object_id(&start_id, NULL, worktree ?
2693 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
2694 GOT_OBJ_TYPE_COMMIT, 1, repo);
2695 else
2696 error = got_repo_match_object_id(&start_id, NULL, start_commit,
2697 GOT_OBJ_TYPE_COMMIT, 1, repo);
2698 if (error != NULL)
2699 goto done;
2701 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
2702 if (error)
2703 goto done;
2705 view = view_open(0, 0, 0, 0, TOG_VIEW_LOG);
2706 if (view == NULL) {
2707 error = got_error_from_errno("view_open");
2708 goto done;
2710 if (worktree) {
2711 head_ref_name = strdup(
2712 got_worktree_get_head_ref_name(worktree));
2713 if (head_ref_name == NULL) {
2714 error = got_error_from_errno("strdup");
2715 goto done;
2718 error = open_log_view(view, start_id, &refs, repo, head_ref_name,
2719 in_repo_path, log_branches);
2720 if (error)
2721 goto done;
2722 if (worktree) {
2723 /* Release work tree lock. */
2724 got_worktree_close(worktree);
2725 worktree = NULL;
2727 error = view_loop(view);
2728 done:
2729 free(in_repo_path);
2730 free(repo_path);
2731 free(cwd);
2732 free(start_id);
2733 free(head_ref_name);
2734 if (repo)
2735 got_repo_close(repo);
2736 if (worktree)
2737 got_worktree_close(worktree);
2738 got_ref_list_free(&refs);
2739 return error;
2742 __dead static void
2743 usage_diff(void)
2745 endwin();
2746 fprintf(stderr, "usage: %s diff [-r repository-path] object1 object2\n",
2747 getprogname());
2748 exit(1);
2751 static char *
2752 parse_next_line(FILE *f, size_t *len)
2754 char *line;
2755 size_t linelen;
2756 size_t lineno;
2757 const char delim[3] = { '\0', '\0', '\0'};
2759 line = fparseln(f, &linelen, &lineno, delim, 0);
2760 if (len)
2761 *len = linelen;
2762 return line;
2765 static int
2766 match_line(const char *line, regex_t *regex)
2768 regmatch_t regmatch;
2770 return regexec(regex, line, 1, &regmatch, 0) == 0;
2773 struct tog_color *
2774 match_color(struct tog_colors *colors, const char *line)
2776 struct tog_color *tc = NULL;
2778 SIMPLEQ_FOREACH(tc, colors, entry) {
2779 if (match_line(line, &tc->regex))
2780 return tc;
2783 return NULL;
2786 static const struct got_error *
2787 draw_file(struct tog_view *view, FILE *f, int *first_displayed_line, int nlines,
2788 int selected_line, int max_lines, int *last_displayed_line, int *eof,
2789 char *header, struct tog_colors *colors)
2791 const struct got_error *err;
2792 int lineno = 0, nprinted = 0;
2793 char *line;
2794 struct tog_color *tc;
2795 size_t len;
2796 wchar_t *wline;
2797 int width;
2799 rewind(f);
2800 werase(view->window);
2802 if (header) {
2803 err = format_line(&wline, &width, header, view->ncols, 0);
2804 if (err) {
2805 return err;
2808 if (view_needs_focus_indication(view))
2809 wstandout(view->window);
2810 waddwstr(view->window, wline);
2811 if (view_needs_focus_indication(view))
2812 wstandend(view->window);
2813 if (width <= view->ncols - 1)
2814 waddch(view->window, '\n');
2816 if (max_lines <= 1)
2817 return NULL;
2818 max_lines--;
2821 *eof = 0;
2822 while (nprinted < max_lines) {
2823 line = parse_next_line(f, &len);
2824 if (line == NULL) {
2825 *eof = 1;
2826 break;
2828 if (++lineno < *first_displayed_line) {
2829 free(line);
2830 continue;
2833 err = format_line(&wline, &width, line, view->ncols, 0);
2834 if (err) {
2835 free(line);
2836 return err;
2839 tc = match_color(colors, line);
2840 if (tc)
2841 wattr_on(view->window,
2842 COLOR_PAIR(tc->colorpair), NULL);
2843 waddwstr(view->window, wline);
2844 if (tc)
2845 wattr_off(view->window,
2846 COLOR_PAIR(tc->colorpair), NULL);
2847 if (width <= view->ncols - 1)
2848 waddch(view->window, '\n');
2849 if (++nprinted == 1)
2850 *first_displayed_line = lineno;
2851 free(line);
2852 free(wline);
2853 wline = NULL;
2855 *last_displayed_line = lineno;
2857 view_vborder(view);
2859 if (*eof) {
2860 while (nprinted < view->nlines) {
2861 waddch(view->window, '\n');
2862 nprinted++;
2865 err = format_line(&wline, &width, TOG_EOF_STRING, view->ncols, 0);
2866 if (err) {
2867 return err;
2870 wstandout(view->window);
2871 waddwstr(view->window, wline);
2872 wstandend(view->window);
2875 return NULL;
2878 static char *
2879 get_datestr(time_t *time, char *datebuf)
2881 struct tm mytm, *tm;
2882 char *p, *s;
2884 tm = gmtime_r(time, &mytm);
2885 if (tm == NULL)
2886 return NULL;
2887 s = asctime_r(tm, datebuf);
2888 if (s == NULL)
2889 return NULL;
2890 p = strchr(s, '\n');
2891 if (p)
2892 *p = '\0';
2893 return s;
2896 static const struct got_error *
2897 get_changed_paths(struct got_pathlist_head *paths,
2898 struct got_commit_object *commit, struct got_repository *repo)
2900 const struct got_error *err = NULL;
2901 struct got_object_id *tree_id1 = NULL, *tree_id2 = NULL;
2902 struct got_tree_object *tree1 = NULL, *tree2 = NULL;
2903 struct got_object_qid *qid;
2905 qid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
2906 if (qid != NULL) {
2907 struct got_commit_object *pcommit;
2908 err = got_object_open_as_commit(&pcommit, repo,
2909 qid->id);
2910 if (err)
2911 return err;
2913 tree_id1 = got_object_commit_get_tree_id(pcommit);
2914 got_object_commit_close(pcommit);
2918 if (tree_id1) {
2919 err = got_object_open_as_tree(&tree1, repo, tree_id1);
2920 if (err)
2921 goto done;
2924 tree_id2 = got_object_commit_get_tree_id(commit);
2925 err = got_object_open_as_tree(&tree2, repo, tree_id2);
2926 if (err)
2927 goto done;
2929 err = got_diff_tree(tree1, tree2, "", "", repo,
2930 got_diff_tree_collect_changed_paths, paths, 0);
2931 done:
2932 if (tree1)
2933 got_object_tree_close(tree1);
2934 if (tree2)
2935 got_object_tree_close(tree2);
2936 return err;
2939 static const struct got_error *
2940 write_commit_info(struct got_object_id *commit_id,
2941 struct got_reflist_head *refs, struct got_repository *repo, FILE *outfile)
2943 const struct got_error *err = NULL;
2944 char datebuf[26], *datestr;
2945 struct got_commit_object *commit;
2946 char *id_str = NULL, *logmsg = NULL;
2947 time_t committer_time;
2948 const char *author, *committer;
2949 char *refs_str = NULL;
2950 struct got_pathlist_head changed_paths;
2951 struct got_pathlist_entry *pe;
2953 TAILQ_INIT(&changed_paths);
2955 if (refs) {
2956 err = build_refs_str(&refs_str, refs, commit_id, repo);
2957 if (err)
2958 return err;
2961 err = got_object_open_as_commit(&commit, repo, commit_id);
2962 if (err)
2963 return err;
2965 err = got_object_id_str(&id_str, commit_id);
2966 if (err) {
2967 err = got_error_from_errno("got_object_id_str");
2968 goto done;
2971 if (fprintf(outfile, "commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
2972 refs_str ? refs_str : "", refs_str ? ")" : "") < 0) {
2973 err = got_error_from_errno("fprintf");
2974 goto done;
2976 if (fprintf(outfile, "from: %s\n",
2977 got_object_commit_get_author(commit)) < 0) {
2978 err = got_error_from_errno("fprintf");
2979 goto done;
2981 committer_time = got_object_commit_get_committer_time(commit);
2982 datestr = get_datestr(&committer_time, datebuf);
2983 if (datestr && fprintf(outfile, "date: %s UTC\n", datestr) < 0) {
2984 err = got_error_from_errno("fprintf");
2985 goto done;
2987 author = got_object_commit_get_author(commit);
2988 committer = got_object_commit_get_committer(commit);
2989 if (strcmp(author, committer) != 0 &&
2990 fprintf(outfile, "via: %s\n", committer) < 0) {
2991 err = got_error_from_errno("fprintf");
2992 goto done;
2994 err = got_object_commit_get_logmsg(&logmsg, commit);
2995 if (err)
2996 goto done;
2997 if (fprintf(outfile, "%s\n", logmsg) < 0) {
2998 err = got_error_from_errno("fprintf");
2999 goto done;
3001 err = get_changed_paths(&changed_paths, commit, repo);
3002 if (err)
3003 goto done;
3004 TAILQ_FOREACH(pe, &changed_paths, entry) {
3005 struct got_diff_changed_path *cp = pe->data;
3006 fprintf(outfile, "%c %s\n", cp->status, pe->path);
3007 free((char *)pe->path);
3008 free(pe->data);
3010 fputc('\n', outfile);
3011 done:
3012 got_pathlist_free(&changed_paths);
3013 free(id_str);
3014 free(logmsg);
3015 free(refs_str);
3016 got_object_commit_close(commit);
3017 return err;
3020 const struct got_error *
3021 get_filestream_info(size_t *filesize, int *nlines, off_t **line_offsets,
3022 FILE *infile)
3024 size_t len;
3025 char *buf = NULL;
3026 int i;
3027 size_t noffsets = 0;
3028 off_t off = 0;
3030 if (line_offsets)
3031 *line_offsets = NULL;
3032 if (filesize)
3033 *filesize = 0;
3034 if (nlines)
3035 *nlines = 0;
3037 if (fseek(infile, 0, SEEK_END) == -1)
3038 return got_error_from_errno("fseek");
3039 len = ftell(infile) + 1;
3040 if (ferror(infile))
3041 return got_error_from_errno("ftell");
3042 if (fseek(infile, 0, SEEK_SET) == -1)
3043 return got_error_from_errno("fseek");
3045 if (len == 0)
3046 return NULL;
3047 if ((buf = calloc(len, sizeof(char *))) == NULL)
3048 return got_error_from_errno("calloc");
3050 fread(buf, 1, len, infile);
3051 if (ferror(infile))
3052 return got_error_from_errno("fread");
3054 i = 0;
3055 if (line_offsets && nlines) {
3056 if (*line_offsets == NULL) {
3057 /* Have some data but perhaps no '\n'. */
3058 noffsets = 1;
3059 *nlines = 1;
3060 *line_offsets = calloc(1, sizeof(**line_offsets));
3061 if (*line_offsets == NULL)
3062 return got_error_from_errno("calloc");
3063 /* Skip forward over end of first line. */
3064 while (i < len) {
3065 if (buf[i] == '\n')
3066 break;
3067 i++;
3070 /* Scan '\n' offsets in remaining chunk of data. */
3071 while (i < len) {
3072 if (buf[i] != '\n') {
3073 i++;
3074 continue;
3076 (*nlines)++;
3077 if (noffsets < *nlines) {
3078 off_t *o = recallocarray(*line_offsets,
3079 noffsets, *nlines,
3080 sizeof(**line_offsets));
3081 if (o == NULL) {
3082 free(*line_offsets);
3083 *line_offsets = NULL;
3084 return got_error_from_errno(
3085 "recallocarray");
3087 *line_offsets = o;
3088 noffsets = *nlines;
3090 off = i + 1;
3091 (*line_offsets)[*nlines - 1] = off;
3092 i++;
3096 if (fflush(infile) != 0)
3097 return got_error_from_errno("fflush");
3098 rewind(infile);
3100 if (filesize)
3101 *filesize = len;
3103 return NULL;
3106 static const struct got_error *
3107 create_diff(struct tog_diff_view_state *s)
3109 const struct got_error *err = NULL;
3110 FILE *f = NULL;
3111 int obj_type;
3113 f = got_opentemp();
3114 if (f == NULL) {
3115 err = got_error_from_errno("got_opentemp");
3116 goto done;
3118 if (s->f && fclose(s->f) != 0) {
3119 err = got_error_from_errno("fclose");
3120 goto done;
3122 s->f = f;
3124 if (s->id1)
3125 err = got_object_get_type(&obj_type, s->repo, s->id1);
3126 else
3127 err = got_object_get_type(&obj_type, s->repo, s->id2);
3128 if (err)
3129 goto done;
3131 switch (obj_type) {
3132 case GOT_OBJ_TYPE_BLOB:
3133 err = got_diff_objects_as_blobs(s->id1, s->id2, NULL, NULL,
3134 s->diff_context, 0, s->repo, s->f);
3135 break;
3136 case GOT_OBJ_TYPE_TREE:
3137 err = got_diff_objects_as_trees(s->id1, s->id2, "", "",
3138 s->diff_context, 0, s->repo, s->f);
3139 break;
3140 case GOT_OBJ_TYPE_COMMIT: {
3141 const struct got_object_id_queue *parent_ids;
3142 struct got_object_qid *pid;
3143 struct got_commit_object *commit2;
3145 err = got_object_open_as_commit(&commit2, s->repo, s->id2);
3146 if (err)
3147 goto done;
3148 /* Show commit info if we're diffing to a parent/root commit. */
3149 if (s->id1 == NULL) {
3150 err = write_commit_info(s->id2, s->refs, s->repo, s->f);
3151 if (err)
3152 goto done;
3153 } else {
3154 parent_ids = got_object_commit_get_parent_ids(commit2);
3155 SIMPLEQ_FOREACH(pid, parent_ids, entry) {
3156 if (got_object_id_cmp(s->id1, pid->id) == 0) {
3157 err = write_commit_info(s->id2, s->refs,
3158 s->repo, s->f);
3159 if (err)
3160 goto done;
3161 break;
3165 got_object_commit_close(commit2);
3167 err = got_diff_objects_as_commits(s->id1, s->id2,
3168 s->diff_context, 0, s->repo, s->f);
3169 break;
3171 default:
3172 err = got_error(GOT_ERR_OBJ_TYPE);
3173 break;
3175 if (err)
3176 goto done;
3177 err = get_filestream_info(&s->filesize, &s->nlines, &s->line_offsets,
3178 s->f);
3179 done:
3180 if (s->f && fflush(s->f) != 0 && err == NULL)
3181 err = got_error_from_errno("fflush");
3182 return err;
3185 static void
3186 diff_view_indicate_progress(struct tog_view *view)
3188 mvwaddstr(view->window, 0, 0, "diffing...");
3189 update_panels();
3190 doupdate();
3193 static const struct got_error *
3194 search_start_diff_view(struct tog_view *view)
3196 struct tog_diff_view_state *s = &view->state.diff;
3198 s->matched_line = 0;
3199 return NULL;
3202 static const struct got_error *
3203 search_next_diff_view(struct tog_view *view)
3205 struct tog_diff_view_state *s = &view->state.diff;
3206 int lineno;
3208 if (!view->searching) {
3209 view->search_next_done = TOG_SEARCH_HAVE_MORE;
3210 return NULL;
3213 if (s->matched_line) {
3214 if (view->searching == TOG_SEARCH_FORWARD)
3215 lineno = s->matched_line + 1;
3216 else
3217 lineno = s->matched_line - 1;
3218 } else {
3219 if (view->searching == TOG_SEARCH_FORWARD)
3220 lineno = 1;
3221 else
3222 lineno = s->nlines;
3225 while (1) {
3226 char *line = NULL;
3227 off_t offset;
3228 size_t len;
3230 if (lineno <= 0 || lineno > s->nlines) {
3231 if (s->matched_line == 0) {
3232 view->search_next_done = TOG_SEARCH_HAVE_MORE;
3233 free(line);
3234 break;
3237 if (view->searching == TOG_SEARCH_FORWARD)
3238 lineno = 1;
3239 else
3240 lineno = s->nlines;
3243 offset = s->line_offsets[lineno - 1];
3244 if (fseeko(s->f, offset, SEEK_SET) != 0) {
3245 free(line);
3246 return got_error_from_errno("fseeko");
3248 free(line);
3249 line = parse_next_line(s->f, &len);
3250 if (line && match_line(line, &view->regex)) {
3251 view->search_next_done = TOG_SEARCH_HAVE_MORE;
3252 s->matched_line = lineno;
3253 free(line);
3254 break;
3256 free(line);
3257 if (view->searching == TOG_SEARCH_FORWARD)
3258 lineno++;
3259 else
3260 lineno--;
3263 if (s->matched_line) {
3264 s->first_displayed_line = s->matched_line;
3265 s->selected_line = 1;
3268 return NULL;
3271 static const struct got_error *
3272 open_diff_view(struct tog_view *view, struct got_object_id *id1,
3273 struct got_object_id *id2, struct tog_view *log_view,
3274 struct got_reflist_head *refs, struct got_repository *repo)
3276 const struct got_error *err;
3277 struct tog_diff_view_state *s = &view->state.diff;
3279 if (id1 != NULL && id2 != NULL) {
3280 int type1, type2;
3281 err = got_object_get_type(&type1, repo, id1);
3282 if (err)
3283 return err;
3284 err = got_object_get_type(&type2, repo, id2);
3285 if (err)
3286 return err;
3288 if (type1 != type2)
3289 return got_error(GOT_ERR_OBJ_TYPE);
3291 s->first_displayed_line = 1;
3292 s->last_displayed_line = view->nlines;
3293 s->selected_line = 1;
3294 s->repo = repo;
3295 s->refs = refs;
3296 s->id1 = id1;
3297 s->id2 = id2;
3299 if (id1) {
3300 s->id1 = got_object_id_dup(id1);
3301 if (s->id1 == NULL)
3302 return got_error_from_errno("got_object_id_dup");
3303 } else
3304 s->id1 = NULL;
3306 s->id2 = got_object_id_dup(id2);
3307 if (s->id2 == NULL) {
3308 free(s->id1);
3309 s->id1 = NULL;
3310 return got_error_from_errno("got_object_id_dup");
3312 s->f = NULL;
3313 s->first_displayed_line = 1;
3314 s->last_displayed_line = view->nlines;
3315 s->diff_context = 3;
3316 s->log_view = log_view;
3317 s->repo = repo;
3318 s->refs = refs;
3320 SIMPLEQ_INIT(&s->colors);
3321 if (has_colors() && getenv("TOG_COLORS") != NULL) {
3322 err = add_color(&s->colors,
3323 "^-", TOG_COLOR_DIFF_MINUS,
3324 get_color_value("TOG_COLOR_DIFF_MINUS"));
3325 if (err)
3326 return err;
3327 err = add_color(&s->colors, "^\\+",
3328 TOG_COLOR_DIFF_PLUS,
3329 get_color_value("TOG_COLOR_DIFF_PLUS"));
3330 if (err) {
3331 free_colors(&s->colors);
3332 return err;
3334 err = add_color(&s->colors,
3335 "^@@", TOG_COLOR_DIFF_CHUNK_HEADER,
3336 get_color_value("TOG_COLOR_DIFF_CHUNK_HEADER"));
3337 if (err) {
3338 free_colors(&s->colors);
3339 return err;
3342 err = add_color(&s->colors,
3343 "^(commit|(blob|file) [-+] |[MDmA] [^ ])",
3344 TOG_COLOR_DIFF_META,
3345 get_color_value("TOG_COLOR_DIFF_META"));
3346 if (err) {
3347 free_colors(&s->colors);
3348 return err;
3351 err = add_color(&s->colors,
3352 "^(from|via): ", TOG_COLOR_AUTHOR,
3353 get_color_value("TOG_COLOR_AUTHOR"));
3354 if (err) {
3355 free_colors(&s->colors);
3356 return err;
3359 err = add_color(&s->colors,
3360 "^date: ", TOG_COLOR_DATE,
3361 get_color_value("TOG_COLOR_DATE"));
3362 if (err) {
3363 free_colors(&s->colors);
3364 return err;
3368 if (log_view && view_is_splitscreen(view))
3369 show_log_view(log_view); /* draw vborder */
3370 diff_view_indicate_progress(view);
3372 err = create_diff(s);
3373 if (err) {
3374 free(s->id1);
3375 s->id1 = NULL;
3376 free(s->id2);
3377 s->id2 = NULL;
3378 return err;
3381 view->show = show_diff_view;
3382 view->input = input_diff_view;
3383 view->close = close_diff_view;
3384 view->search_start = search_start_diff_view;
3385 view->search_next = search_next_diff_view;
3387 return NULL;
3390 static const struct got_error *
3391 close_diff_view(struct tog_view *view)
3393 const struct got_error *err = NULL;
3394 struct tog_diff_view_state *s = &view->state.diff;
3396 free(s->id1);
3397 s->id1 = NULL;
3398 free(s->id2);
3399 s->id2 = NULL;
3400 if (s->f && fclose(s->f) == EOF)
3401 err = got_error_from_errno("fclose");
3402 free_colors(&s->colors);
3403 free(s->line_offsets);
3404 return err;
3407 static const struct got_error *
3408 show_diff_view(struct tog_view *view)
3410 const struct got_error *err;
3411 struct tog_diff_view_state *s = &view->state.diff;
3412 char *id_str1 = NULL, *id_str2, *header;
3414 if (s->id1) {
3415 err = got_object_id_str(&id_str1, s->id1);
3416 if (err)
3417 return err;
3419 err = got_object_id_str(&id_str2, s->id2);
3420 if (err)
3421 return err;
3423 if (asprintf(&header, "diff %s %s",
3424 id_str1 ? id_str1 : "/dev/null", id_str2) == -1) {
3425 err = got_error_from_errno("asprintf");
3426 free(id_str1);
3427 free(id_str2);
3428 return err;
3430 free(id_str1);
3431 free(id_str2);
3433 return draw_file(view, s->f, &s->first_displayed_line, s->nlines,
3434 s->selected_line, view->nlines, &s->last_displayed_line, &s->eof,
3435 header, &s->colors);
3438 static const struct got_error *
3439 set_selected_commit(struct tog_diff_view_state *s,
3440 struct commit_queue_entry *entry)
3442 const struct got_error *err;
3443 const struct got_object_id_queue *parent_ids;
3444 struct got_commit_object *selected_commit;
3445 struct got_object_qid *pid;
3447 free(s->id2);
3448 s->id2 = got_object_id_dup(entry->id);
3449 if (s->id2 == NULL)
3450 return got_error_from_errno("got_object_id_dup");
3452 err = got_object_open_as_commit(&selected_commit, s->repo, entry->id);
3453 if (err)
3454 return err;
3455 parent_ids = got_object_commit_get_parent_ids(selected_commit);
3456 free(s->id1);
3457 pid = SIMPLEQ_FIRST(parent_ids);
3458 s->id1 = pid ? got_object_id_dup(pid->id) : NULL;
3459 got_object_commit_close(selected_commit);
3460 return NULL;
3463 static const struct got_error *
3464 input_diff_view(struct tog_view **new_view, struct tog_view **dead_view,
3465 struct tog_view **focus_view, struct tog_view *view, int ch)
3467 const struct got_error *err = NULL;
3468 struct tog_diff_view_state *s = &view->state.diff;
3469 struct tog_log_view_state *ls;
3470 struct commit_queue_entry *entry;
3471 int i;
3473 switch (ch) {
3474 case 'k':
3475 case KEY_UP:
3476 if (s->first_displayed_line > 1)
3477 s->first_displayed_line--;
3478 break;
3479 case KEY_PPAGE:
3480 case CTRL('b'):
3481 if (s->first_displayed_line == 1)
3482 break;
3483 i = 0;
3484 while (i++ < view->nlines - 1 &&
3485 s->first_displayed_line > 1)
3486 s->first_displayed_line--;
3487 break;
3488 case 'j':
3489 case KEY_DOWN:
3490 if (!s->eof)
3491 s->first_displayed_line++;
3492 break;
3493 case KEY_NPAGE:
3494 case CTRL('f'):
3495 case ' ':
3496 if (s->eof)
3497 break;
3498 i = 0;
3499 while (!s->eof && i++ < view->nlines - 1) {
3500 char *line;
3501 line = parse_next_line(s->f, NULL);
3502 s->first_displayed_line++;
3503 if (line == NULL)
3504 break;
3506 break;
3507 case '[':
3508 if (s->diff_context > 0) {
3509 s->diff_context--;
3510 diff_view_indicate_progress(view);
3511 err = create_diff(s);
3513 break;
3514 case ']':
3515 if (s->diff_context < GOT_DIFF_MAX_CONTEXT) {
3516 s->diff_context++;
3517 diff_view_indicate_progress(view);
3518 err = create_diff(s);
3520 break;
3521 case '<':
3522 case ',':
3523 if (s->log_view == NULL)
3524 break;
3525 ls = &s->log_view->state.log;
3526 entry = TAILQ_PREV(ls->selected_entry,
3527 commit_queue_head, entry);
3528 if (entry == NULL)
3529 break;
3531 err = input_log_view(NULL, NULL, NULL, s->log_view,
3532 KEY_UP);
3533 if (err)
3534 break;
3536 err = set_selected_commit(s, entry);
3537 if (err)
3538 break;
3540 s->first_displayed_line = 1;
3541 s->last_displayed_line = view->nlines;
3543 diff_view_indicate_progress(view);
3544 err = create_diff(s);
3545 break;
3546 case '>':
3547 case '.':
3548 if (s->log_view == NULL)
3549 break;
3550 ls = &s->log_view->state.log;
3552 if (TAILQ_NEXT(ls->selected_entry, entry) == NULL) {
3553 ls->thread_args.commits_needed++;
3554 err = trigger_log_thread(s->log_view, 1,
3555 &ls->thread_args.commits_needed,
3556 &ls->thread_args.log_complete,
3557 &ls->thread_args.need_commits,
3558 &ls->thread_args.commit_loaded);
3559 if (err)
3560 break;
3562 err = input_log_view(NULL, NULL, NULL, s->log_view,
3563 KEY_DOWN);
3564 if (err)
3565 break;
3567 entry = TAILQ_NEXT(ls->selected_entry, entry);
3568 if (entry == NULL)
3569 break;
3571 err = set_selected_commit(s, entry);
3572 if (err)
3573 break;
3575 s->first_displayed_line = 1;
3576 s->last_displayed_line = view->nlines;
3578 diff_view_indicate_progress(view);
3579 err = create_diff(s);
3580 break;
3581 default:
3582 break;
3585 return err;
3588 static const struct got_error *
3589 cmd_diff(int argc, char *argv[])
3591 const struct got_error *error = NULL;
3592 struct got_repository *repo = NULL;
3593 struct got_worktree *worktree = NULL;
3594 struct got_reflist_head refs;
3595 struct got_object_id *id1 = NULL, *id2 = NULL;
3596 char *repo_path = NULL, *cwd = NULL;
3597 char *id_str1 = NULL, *id_str2 = NULL;
3598 int ch;
3599 struct tog_view *view;
3601 SIMPLEQ_INIT(&refs);
3603 #ifndef PROFILE
3604 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
3605 NULL) == -1)
3606 err(1, "pledge");
3607 #endif
3609 while ((ch = getopt(argc, argv, "r:")) != -1) {
3610 switch (ch) {
3611 case 'r':
3612 repo_path = realpath(optarg, NULL);
3613 if (repo_path == NULL)
3614 return got_error_from_errno2("realpath",
3615 optarg);
3616 break;
3617 default:
3618 usage_diff();
3619 /* NOTREACHED */
3623 argc -= optind;
3624 argv += optind;
3626 if (argc == 0) {
3627 usage_diff(); /* TODO show local worktree changes */
3628 } else if (argc == 2) {
3629 id_str1 = argv[0];
3630 id_str2 = argv[1];
3631 } else
3632 usage_diff();
3634 cwd = getcwd(NULL, 0);
3635 if (cwd == NULL)
3636 return got_error_from_errno("getcwd");
3638 error = got_worktree_open(&worktree, cwd);
3639 if (error && error->code != GOT_ERR_NOT_WORKTREE)
3640 goto done;
3642 if (repo_path == NULL) {
3643 if (worktree)
3644 repo_path =
3645 strdup(got_worktree_get_repo_path(worktree));
3646 else
3647 repo_path = strdup(cwd);
3649 if (repo_path == NULL) {
3650 error = got_error_from_errno("strdup");
3651 goto done;
3654 error = got_repo_open(&repo, repo_path, NULL);
3655 if (error)
3656 goto done;
3658 init_curses();
3660 error = apply_unveil(got_repo_get_path(repo), NULL);
3661 if (error)
3662 goto done;
3664 error = got_repo_match_object_id_prefix(&id1, id_str1,
3665 GOT_OBJ_TYPE_ANY, repo);
3666 if (error)
3667 goto done;
3669 error = got_repo_match_object_id_prefix(&id2, id_str2,
3670 GOT_OBJ_TYPE_ANY, repo);
3671 if (error)
3672 goto done;
3674 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
3675 if (error)
3676 goto done;
3678 view = view_open(0, 0, 0, 0, TOG_VIEW_DIFF);
3679 if (view == NULL) {
3680 error = got_error_from_errno("view_open");
3681 goto done;
3683 error = open_diff_view(view, id1, id2, NULL, &refs, repo);
3684 if (error)
3685 goto done;
3686 error = view_loop(view);
3687 done:
3688 free(repo_path);
3689 free(cwd);
3690 if (repo)
3691 got_repo_close(repo);
3692 if (worktree)
3693 got_worktree_close(worktree);
3694 got_ref_list_free(&refs);
3695 return error;
3698 __dead static void
3699 usage_blame(void)
3701 endwin();
3702 fprintf(stderr, "usage: %s blame [-c commit] [-r repository-path] path\n",
3703 getprogname());
3704 exit(1);
3707 struct tog_blame_line {
3708 int annotated;
3709 struct got_object_id *id;
3712 static const struct got_error *
3713 draw_blame(struct tog_view *view, struct got_object_id *id, FILE *f,
3714 const char *path, struct tog_blame_line *lines, int nlines,
3715 int blame_complete, int selected_line, int *first_displayed_line,
3716 int *last_displayed_line, int *eof, int max_lines,
3717 struct tog_colors *colors)
3719 const struct got_error *err;
3720 int lineno = 0, nprinted = 0;
3721 char *line;
3722 size_t len;
3723 wchar_t *wline;
3724 int width;
3725 struct tog_blame_line *blame_line;
3726 struct got_object_id *prev_id = NULL;
3727 char *id_str;
3728 struct tog_color *tc;
3730 err = got_object_id_str(&id_str, id);
3731 if (err)
3732 return err;
3734 rewind(f);
3735 werase(view->window);
3737 if (asprintf(&line, "commit %s", id_str) == -1) {
3738 err = got_error_from_errno("asprintf");
3739 free(id_str);
3740 return err;
3743 err = format_line(&wline, &width, line, view->ncols, 0);
3744 free(line);
3745 line = NULL;
3746 if (err)
3747 return err;
3748 if (view_needs_focus_indication(view))
3749 wstandout(view->window);
3750 tc = get_color(colors, TOG_COLOR_COMMIT);
3751 if (tc)
3752 wattr_on(view->window,
3753 COLOR_PAIR(tc->colorpair), NULL);
3754 waddwstr(view->window, wline);
3755 if (tc)
3756 wattr_off(view->window,
3757 COLOR_PAIR(tc->colorpair), NULL);
3758 if (view_needs_focus_indication(view))
3759 wstandend(view->window);
3760 free(wline);
3761 wline = NULL;
3762 if (width < view->ncols - 1)
3763 waddch(view->window, '\n');
3765 if (asprintf(&line, "[%d/%d] %s%s",
3766 *first_displayed_line - 1 + selected_line, nlines,
3767 blame_complete ? "" : "annotating... ", path) == -1) {
3768 free(id_str);
3769 return got_error_from_errno("asprintf");
3771 free(id_str);
3772 err = format_line(&wline, &width, line, view->ncols, 0);
3773 free(line);
3774 line = NULL;
3775 if (err)
3776 return err;
3777 waddwstr(view->window, wline);
3778 free(wline);
3779 wline = NULL;
3780 if (width < view->ncols - 1)
3781 waddch(view->window, '\n');
3783 *eof = 0;
3784 while (nprinted < max_lines - 2) {
3785 line = parse_next_line(f, &len);
3786 if (line == NULL) {
3787 *eof = 1;
3788 break;
3790 if (++lineno < *first_displayed_line) {
3791 free(line);
3792 continue;
3795 if (view->ncols <= 9) {
3796 width = 9;
3797 wline = wcsdup(L"");
3798 if (wline == NULL)
3799 err = got_error_from_errno("wcsdup");
3800 } else {
3801 err = format_line(&wline, &width, line,
3802 view->ncols - 9, 9);
3803 width += 9;
3805 if (err) {
3806 free(line);
3807 return err;
3810 if (view->focussed && nprinted == selected_line - 1)
3811 wstandout(view->window);
3813 if (nlines > 0) {
3814 blame_line = &lines[lineno - 1];
3815 if (blame_line->annotated && prev_id &&
3816 got_object_id_cmp(prev_id, blame_line->id) == 0 &&
3817 !(view->focussed &&
3818 nprinted == selected_line - 1)) {
3819 waddstr(view->window, " ");
3820 } else if (blame_line->annotated) {
3821 char *id_str;
3822 err = got_object_id_str(&id_str, blame_line->id);
3823 if (err) {
3824 free(line);
3825 free(wline);
3826 return err;
3828 tc = get_color(colors, TOG_COLOR_COMMIT);
3829 if (tc)
3830 wattr_on(view->window,
3831 COLOR_PAIR(tc->colorpair), NULL);
3832 wprintw(view->window, "%.8s", id_str);
3833 if (tc)
3834 wattr_off(view->window,
3835 COLOR_PAIR(tc->colorpair), NULL);
3836 free(id_str);
3837 prev_id = blame_line->id;
3838 } else {
3839 waddstr(view->window, "........");
3840 prev_id = NULL;
3842 } else {
3843 waddstr(view->window, "........");
3844 prev_id = NULL;
3847 if (view->focussed && nprinted == selected_line - 1)
3848 wstandend(view->window);
3849 waddstr(view->window, " ");
3851 waddwstr(view->window, wline);
3852 if (width <= view->ncols - 1)
3853 waddch(view->window, '\n');
3854 if (++nprinted == 1)
3855 *first_displayed_line = lineno;
3856 free(line);
3857 free(wline);
3858 wline = NULL;
3860 *last_displayed_line = lineno;
3862 view_vborder(view);
3864 return NULL;
3867 static const struct got_error *
3868 blame_cb(void *arg, int nlines, int lineno, struct got_object_id *id)
3870 const struct got_error *err = NULL;
3871 struct tog_blame_cb_args *a = arg;
3872 struct tog_blame_line *line;
3873 int errcode;
3875 if (nlines != a->nlines ||
3876 (lineno != -1 && lineno < 1) || lineno > a->nlines)
3877 return got_error(GOT_ERR_RANGE);
3879 errcode = pthread_mutex_lock(&tog_mutex);
3880 if (errcode)
3881 return got_error_set_errno(errcode, "pthread_mutex_lock");
3883 if (*a->quit) { /* user has quit the blame view */
3884 err = got_error(GOT_ERR_ITER_COMPLETED);
3885 goto done;
3888 if (lineno == -1)
3889 goto done; /* no change in this commit */
3891 line = &a->lines[lineno - 1];
3892 if (line->annotated)
3893 goto done;
3895 line->id = got_object_id_dup(id);
3896 if (line->id == NULL) {
3897 err = got_error_from_errno("got_object_id_dup");
3898 goto done;
3900 line->annotated = 1;
3901 done:
3902 errcode = pthread_mutex_unlock(&tog_mutex);
3903 if (errcode)
3904 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
3905 return err;
3908 static void *
3909 blame_thread(void *arg)
3911 const struct got_error *err;
3912 struct tog_blame_thread_args *ta = arg;
3913 struct tog_blame_cb_args *a = ta->cb_args;
3914 int errcode;
3916 err = block_signals_used_by_main_thread();
3917 if (err)
3918 return (void *)err;
3920 err = got_blame(ta->path, a->commit_id, ta->repo,
3921 blame_cb, ta->cb_args, ta->cancel_cb, ta->cancel_arg);
3922 if (err && err->code == GOT_ERR_CANCELLED)
3923 err = NULL;
3925 errcode = pthread_mutex_lock(&tog_mutex);
3926 if (errcode)
3927 return (void *)got_error_set_errno(errcode,
3928 "pthread_mutex_lock");
3930 got_repo_close(ta->repo);
3931 ta->repo = NULL;
3932 *ta->complete = 1;
3934 errcode = pthread_mutex_unlock(&tog_mutex);
3935 if (errcode && err == NULL)
3936 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
3938 return (void *)err;
3941 static struct got_object_id *
3942 get_selected_commit_id(struct tog_blame_line *lines, int nlines,
3943 int first_displayed_line, int selected_line)
3945 struct tog_blame_line *line;
3947 if (nlines <= 0)
3948 return NULL;
3950 line = &lines[first_displayed_line - 1 + selected_line - 1];
3951 if (!line->annotated)
3952 return NULL;
3954 return line->id;
3957 static const struct got_error *
3958 stop_blame(struct tog_blame *blame)
3960 const struct got_error *err = NULL;
3961 int i;
3963 if (blame->thread) {
3964 int errcode;
3965 errcode = pthread_mutex_unlock(&tog_mutex);
3966 if (errcode)
3967 return got_error_set_errno(errcode,
3968 "pthread_mutex_unlock");
3969 errcode = pthread_join(blame->thread, (void **)&err);
3970 if (errcode)
3971 return got_error_set_errno(errcode, "pthread_join");
3972 errcode = pthread_mutex_lock(&tog_mutex);
3973 if (errcode)
3974 return got_error_set_errno(errcode,
3975 "pthread_mutex_lock");
3976 if (err && err->code == GOT_ERR_ITER_COMPLETED)
3977 err = NULL;
3978 blame->thread = NULL;
3980 if (blame->thread_args.repo) {
3981 got_repo_close(blame->thread_args.repo);
3982 blame->thread_args.repo = NULL;
3984 if (blame->f) {
3985 if (fclose(blame->f) != 0 && err == NULL)
3986 err = got_error_from_errno("fclose");
3987 blame->f = NULL;
3989 if (blame->lines) {
3990 for (i = 0; i < blame->nlines; i++)
3991 free(blame->lines[i].id);
3992 free(blame->lines);
3993 blame->lines = NULL;
3995 free(blame->cb_args.commit_id);
3996 blame->cb_args.commit_id = NULL;
3998 return err;
4001 static const struct got_error *
4002 cancel_blame_view(void *arg)
4004 const struct got_error *err = NULL;
4005 int *done = arg;
4006 int errcode;
4008 errcode = pthread_mutex_lock(&tog_mutex);
4009 if (errcode)
4010 return got_error_set_errno(errcode,
4011 "pthread_mutex_unlock");
4013 if (*done)
4014 err = got_error(GOT_ERR_CANCELLED);
4016 errcode = pthread_mutex_unlock(&tog_mutex);
4017 if (errcode)
4018 return got_error_set_errno(errcode,
4019 "pthread_mutex_lock");
4021 return err;
4024 static const struct got_error *
4025 run_blame(struct tog_blame *blame, struct tog_view *view, int *blame_complete,
4026 int *first_displayed_line, int *last_displayed_line, int *selected_line,
4027 int *done, int *eof, const char *path, struct got_object_id *commit_id,
4028 struct got_repository *repo)
4030 const struct got_error *err = NULL;
4031 struct got_blob_object *blob = NULL;
4032 struct got_repository *thread_repo = NULL;
4033 struct got_object_id *obj_id = NULL;
4034 int obj_type;
4036 err = got_object_id_by_path(&obj_id, repo, commit_id, path);
4037 if (err)
4038 return err;
4040 err = got_object_get_type(&obj_type, repo, obj_id);
4041 if (err)
4042 goto done;
4044 if (obj_type != GOT_OBJ_TYPE_BLOB) {
4045 err = got_error(GOT_ERR_OBJ_TYPE);
4046 goto done;
4049 err = got_object_open_as_blob(&blob, repo, obj_id, 8192);
4050 if (err)
4051 goto done;
4052 blame->f = got_opentemp();
4053 if (blame->f == NULL) {
4054 err = got_error_from_errno("got_opentemp");
4055 goto done;
4057 err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
4058 &blame->line_offsets, blame->f, blob);
4059 if (err || blame->nlines == 0)
4060 goto done;
4062 /* Don't include \n at EOF in the blame line count. */
4063 if (blame->line_offsets[blame->nlines - 1] == blame->filesize)
4064 blame->nlines--;
4066 blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
4067 if (blame->lines == NULL) {
4068 err = got_error_from_errno("calloc");
4069 goto done;
4072 err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL);
4073 if (err)
4074 goto done;
4076 blame->cb_args.view = view;
4077 blame->cb_args.lines = blame->lines;
4078 blame->cb_args.nlines = blame->nlines;
4079 blame->cb_args.commit_id = got_object_id_dup(commit_id);
4080 if (blame->cb_args.commit_id == NULL) {
4081 err = got_error_from_errno("got_object_id_dup");
4082 goto done;
4084 blame->cb_args.quit = done;
4086 blame->thread_args.path = path;
4087 blame->thread_args.repo = thread_repo;
4088 blame->thread_args.cb_args = &blame->cb_args;
4089 blame->thread_args.complete = blame_complete;
4090 blame->thread_args.cancel_cb = cancel_blame_view;
4091 blame->thread_args.cancel_arg = done;
4092 *blame_complete = 0;
4094 done:
4095 if (blob)
4096 got_object_blob_close(blob);
4097 free(obj_id);
4098 if (err)
4099 stop_blame(blame);
4100 return err;
4103 static const struct got_error *
4104 open_blame_view(struct tog_view *view, char *path,
4105 struct got_object_id *commit_id, struct got_reflist_head *refs,
4106 struct got_repository *repo)
4108 const struct got_error *err = NULL;
4109 struct tog_blame_view_state *s = &view->state.blame;
4111 SIMPLEQ_INIT(&s->blamed_commits);
4113 s->path = strdup(path);
4114 if (s->path == NULL)
4115 return got_error_from_errno("strdup");
4117 err = got_object_qid_alloc(&s->blamed_commit, commit_id);
4118 if (err) {
4119 free(s->path);
4120 return err;
4123 SIMPLEQ_INSERT_HEAD(&s->blamed_commits, s->blamed_commit, entry);
4124 s->first_displayed_line = 1;
4125 s->last_displayed_line = view->nlines;
4126 s->selected_line = 1;
4127 s->blame_complete = 0;
4128 s->repo = repo;
4129 s->refs = refs;
4130 s->commit_id = commit_id;
4131 memset(&s->blame, 0, sizeof(s->blame));
4133 SIMPLEQ_INIT(&s->colors);
4134 if (has_colors() && getenv("TOG_COLORS") != NULL) {
4135 err = add_color(&s->colors, "^", TOG_COLOR_COMMIT,
4136 get_color_value("TOG_COLOR_COMMIT"));
4137 if (err)
4138 return err;
4141 view->show = show_blame_view;
4142 view->input = input_blame_view;
4143 view->close = close_blame_view;
4144 view->search_start = search_start_blame_view;
4145 view->search_next = search_next_blame_view;
4147 return run_blame(&s->blame, view, &s->blame_complete,
4148 &s->first_displayed_line, &s->last_displayed_line,
4149 &s->selected_line, &s->done, &s->eof, s->path,
4150 s->blamed_commit->id, s->repo);
4153 static const struct got_error *
4154 close_blame_view(struct tog_view *view)
4156 const struct got_error *err = NULL;
4157 struct tog_blame_view_state *s = &view->state.blame;
4159 if (s->blame.thread)
4160 err = stop_blame(&s->blame);
4162 while (!SIMPLEQ_EMPTY(&s->blamed_commits)) {
4163 struct got_object_qid *blamed_commit;
4164 blamed_commit = SIMPLEQ_FIRST(&s->blamed_commits);
4165 SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
4166 got_object_qid_free(blamed_commit);
4169 free(s->path);
4170 free_colors(&s->colors);
4172 return err;
4175 static const struct got_error *
4176 search_start_blame_view(struct tog_view *view)
4178 struct tog_blame_view_state *s = &view->state.blame;
4180 s->matched_line = 0;
4181 return NULL;
4184 static const struct got_error *
4185 search_next_blame_view(struct tog_view *view)
4187 struct tog_blame_view_state *s = &view->state.blame;
4188 int lineno;
4190 if (!view->searching) {
4191 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4192 return NULL;
4195 if (s->matched_line) {
4196 if (view->searching == TOG_SEARCH_FORWARD)
4197 lineno = s->matched_line + 1;
4198 else
4199 lineno = s->matched_line - 1;
4200 } else {
4201 if (view->searching == TOG_SEARCH_FORWARD)
4202 lineno = 1;
4203 else
4204 lineno = s->blame.nlines;
4207 while (1) {
4208 char *line = NULL;
4209 off_t offset;
4210 size_t len;
4212 if (lineno <= 0 || lineno > s->blame.nlines) {
4213 if (s->matched_line == 0) {
4214 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4215 free(line);
4216 break;
4219 if (view->searching == TOG_SEARCH_FORWARD)
4220 lineno = 1;
4221 else
4222 lineno = s->blame.nlines;
4225 offset = s->blame.line_offsets[lineno - 1];
4226 if (fseeko(s->blame.f, offset, SEEK_SET) != 0) {
4227 free(line);
4228 return got_error_from_errno("fseeko");
4230 free(line);
4231 line = parse_next_line(s->blame.f, &len);
4232 if (line && match_line(line, &view->regex)) {
4233 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4234 s->matched_line = lineno;
4235 free(line);
4236 break;
4238 free(line);
4239 if (view->searching == TOG_SEARCH_FORWARD)
4240 lineno++;
4241 else
4242 lineno--;
4245 if (s->matched_line) {
4246 s->first_displayed_line = s->matched_line;
4247 s->selected_line = 1;
4250 return NULL;
4253 static const struct got_error *
4254 show_blame_view(struct tog_view *view)
4256 const struct got_error *err = NULL;
4257 struct tog_blame_view_state *s = &view->state.blame;
4258 int errcode;
4260 if (s->blame.thread == NULL) {
4261 errcode = pthread_create(&s->blame.thread, NULL, blame_thread,
4262 &s->blame.thread_args);
4263 if (errcode)
4264 return got_error_set_errno(errcode, "pthread_create");
4266 halfdelay(1); /* fast refresh while annotating */
4269 if (s->blame_complete)
4270 halfdelay(10); /* disable fast refresh */
4272 err = draw_blame(view, s->blamed_commit->id, s->blame.f,
4273 s->path, s->blame.lines, s->blame.nlines, s->blame_complete,
4274 s->selected_line, &s->first_displayed_line,
4275 &s->last_displayed_line, &s->eof, view->nlines, &s->colors);
4277 view_vborder(view);
4278 return err;
4281 static const struct got_error *
4282 input_blame_view(struct tog_view **new_view, struct tog_view **dead_view,
4283 struct tog_view **focus_view, struct tog_view *view, int ch)
4285 const struct got_error *err = NULL, *thread_err = NULL;
4286 struct tog_view *diff_view;
4287 struct tog_blame_view_state *s = &view->state.blame;
4288 int begin_x = 0;
4290 switch (ch) {
4291 case 'q':
4292 s->done = 1;
4293 break;
4294 case 'k':
4295 case KEY_UP:
4296 if (s->selected_line > 1)
4297 s->selected_line--;
4298 else if (s->selected_line == 1 &&
4299 s->first_displayed_line > 1)
4300 s->first_displayed_line--;
4301 break;
4302 case KEY_PPAGE:
4303 case CTRL('b'):
4304 if (s->first_displayed_line == 1) {
4305 s->selected_line = 1;
4306 break;
4308 if (s->first_displayed_line > view->nlines - 2)
4309 s->first_displayed_line -=
4310 (view->nlines - 2);
4311 else
4312 s->first_displayed_line = 1;
4313 break;
4314 case 'j':
4315 case KEY_DOWN:
4316 if (s->selected_line < view->nlines - 2 &&
4317 s->first_displayed_line +
4318 s->selected_line <= s->blame.nlines)
4319 s->selected_line++;
4320 else if (s->last_displayed_line <
4321 s->blame.nlines)
4322 s->first_displayed_line++;
4323 break;
4324 case 'b':
4325 case 'p': {
4326 struct got_object_id *id = NULL;
4327 id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
4328 s->first_displayed_line, s->selected_line);
4329 if (id == NULL)
4330 break;
4331 if (ch == 'p') {
4332 struct got_commit_object *commit;
4333 struct got_object_qid *pid;
4334 struct got_object_id *blob_id = NULL;
4335 int obj_type;
4336 err = got_object_open_as_commit(&commit,
4337 s->repo, id);
4338 if (err)
4339 break;
4340 pid = SIMPLEQ_FIRST(
4341 got_object_commit_get_parent_ids(commit));
4342 if (pid == NULL) {
4343 got_object_commit_close(commit);
4344 break;
4346 /* Check if path history ends here. */
4347 err = got_object_id_by_path(&blob_id, s->repo,
4348 pid->id, s->path);
4349 if (err) {
4350 if (err->code == GOT_ERR_NO_TREE_ENTRY)
4351 err = NULL;
4352 got_object_commit_close(commit);
4353 break;
4355 err = got_object_get_type(&obj_type, s->repo,
4356 blob_id);
4357 free(blob_id);
4358 /* Can't blame non-blob type objects. */
4359 if (obj_type != GOT_OBJ_TYPE_BLOB) {
4360 got_object_commit_close(commit);
4361 break;
4363 err = got_object_qid_alloc(&s->blamed_commit,
4364 pid->id);
4365 got_object_commit_close(commit);
4366 } else {
4367 if (got_object_id_cmp(id,
4368 s->blamed_commit->id) == 0)
4369 break;
4370 err = got_object_qid_alloc(&s->blamed_commit,
4371 id);
4373 if (err)
4374 break;
4375 s->done = 1;
4376 thread_err = stop_blame(&s->blame);
4377 s->done = 0;
4378 if (thread_err)
4379 break;
4380 SIMPLEQ_INSERT_HEAD(&s->blamed_commits,
4381 s->blamed_commit, entry);
4382 err = run_blame(&s->blame, view, &s->blame_complete,
4383 &s->first_displayed_line, &s->last_displayed_line,
4384 &s->selected_line, &s->done, &s->eof,
4385 s->path, s->blamed_commit->id, s->repo);
4386 if (err)
4387 break;
4388 break;
4390 case 'B': {
4391 struct got_object_qid *first;
4392 first = SIMPLEQ_FIRST(&s->blamed_commits);
4393 if (!got_object_id_cmp(first->id, s->commit_id))
4394 break;
4395 s->done = 1;
4396 thread_err = stop_blame(&s->blame);
4397 s->done = 0;
4398 if (thread_err)
4399 break;
4400 SIMPLEQ_REMOVE_HEAD(&s->blamed_commits, entry);
4401 got_object_qid_free(s->blamed_commit);
4402 s->blamed_commit =
4403 SIMPLEQ_FIRST(&s->blamed_commits);
4404 err = run_blame(&s->blame, view, &s->blame_complete,
4405 &s->first_displayed_line, &s->last_displayed_line,
4406 &s->selected_line, &s->done, &s->eof, s->path,
4407 s->blamed_commit->id, s->repo);
4408 if (err)
4409 break;
4410 break;
4412 case KEY_ENTER:
4413 case '\r': {
4414 struct got_object_id *id = NULL;
4415 struct got_object_qid *pid;
4416 struct got_commit_object *commit = NULL;
4417 id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
4418 s->first_displayed_line, s->selected_line);
4419 if (id == NULL)
4420 break;
4421 err = got_object_open_as_commit(&commit, s->repo, id);
4422 if (err)
4423 break;
4424 pid = SIMPLEQ_FIRST(
4425 got_object_commit_get_parent_ids(commit));
4426 if (view_is_parent_view(view))
4427 begin_x = view_split_begin_x(view->begin_x);
4428 diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
4429 if (diff_view == NULL) {
4430 got_object_commit_close(commit);
4431 err = got_error_from_errno("view_open");
4432 break;
4434 err = open_diff_view(diff_view, pid ? pid->id : NULL,
4435 id, NULL, s->refs, s->repo);
4436 got_object_commit_close(commit);
4437 if (err) {
4438 view_close(diff_view);
4439 break;
4441 if (view_is_parent_view(view)) {
4442 err = view_close_child(view);
4443 if (err)
4444 break;
4445 err = view_set_child(view, diff_view);
4446 if (err) {
4447 view_close(diff_view);
4448 break;
4450 *focus_view = diff_view;
4451 view->child_focussed = 1;
4452 } else
4453 *new_view = diff_view;
4454 if (err)
4455 break;
4456 break;
4458 case KEY_NPAGE:
4459 case CTRL('f'):
4460 case ' ':
4461 if (s->last_displayed_line >= s->blame.nlines &&
4462 s->selected_line >= MIN(s->blame.nlines,
4463 view->nlines - 2)) {
4464 break;
4466 if (s->last_displayed_line >= s->blame.nlines &&
4467 s->selected_line < view->nlines - 2) {
4468 s->selected_line = MIN(s->blame.nlines,
4469 view->nlines - 2);
4470 break;
4472 if (s->last_displayed_line + view->nlines - 2
4473 <= s->blame.nlines)
4474 s->first_displayed_line +=
4475 view->nlines - 2;
4476 else
4477 s->first_displayed_line =
4478 s->blame.nlines -
4479 (view->nlines - 3);
4480 break;
4481 case KEY_RESIZE:
4482 if (s->selected_line > view->nlines - 2) {
4483 s->selected_line = MIN(s->blame.nlines,
4484 view->nlines - 2);
4486 break;
4487 default:
4488 break;
4490 return thread_err ? thread_err : err;
4493 static const struct got_error *
4494 cmd_blame(int argc, char *argv[])
4496 const struct got_error *error;
4497 struct got_repository *repo = NULL;
4498 struct got_reflist_head refs;
4499 struct got_worktree *worktree = NULL;
4500 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
4501 struct got_object_id *commit_id = NULL;
4502 char *commit_id_str = NULL;
4503 int ch;
4504 struct tog_view *view;
4506 SIMPLEQ_INIT(&refs);
4508 #ifndef PROFILE
4509 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
4510 NULL) == -1)
4511 err(1, "pledge");
4512 #endif
4514 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
4515 switch (ch) {
4516 case 'c':
4517 commit_id_str = optarg;
4518 break;
4519 case 'r':
4520 repo_path = realpath(optarg, NULL);
4521 if (repo_path == NULL)
4522 return got_error_from_errno2("realpath",
4523 optarg);
4524 break;
4525 default:
4526 usage_blame();
4527 /* NOTREACHED */
4531 argc -= optind;
4532 argv += optind;
4534 if (argc != 1)
4535 usage_blame();
4537 cwd = getcwd(NULL, 0);
4538 if (cwd == NULL)
4539 return got_error_from_errno("getcwd");
4541 error = got_worktree_open(&worktree, cwd);
4542 if (error && error->code != GOT_ERR_NOT_WORKTREE)
4543 goto done;
4545 if (repo_path == NULL) {
4546 if (worktree)
4547 repo_path =
4548 strdup(got_worktree_get_repo_path(worktree));
4549 else
4550 repo_path = strdup(cwd);
4552 if (repo_path == NULL) {
4553 error = got_error_from_errno("strdup");
4554 goto done;
4557 error = got_repo_open(&repo, repo_path, NULL);
4558 if (error != NULL)
4559 goto done;
4561 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv, repo,
4562 worktree);
4563 if (error)
4564 goto done;
4566 init_curses();
4568 error = apply_unveil(got_repo_get_path(repo), NULL);
4569 if (error)
4570 goto done;
4572 if (commit_id_str == NULL) {
4573 struct got_reference *head_ref;
4574 error = got_ref_open(&head_ref, repo, worktree ?
4575 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD, 0);
4576 if (error != NULL)
4577 goto done;
4578 error = got_ref_resolve(&commit_id, repo, head_ref);
4579 got_ref_close(head_ref);
4580 } else {
4581 error = got_repo_match_object_id(&commit_id, NULL,
4582 commit_id_str, GOT_OBJ_TYPE_COMMIT, 1, repo);
4584 if (error != NULL)
4585 goto done;
4587 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
4588 if (error)
4589 goto done;
4591 view = view_open(0, 0, 0, 0, TOG_VIEW_BLAME);
4592 if (view == NULL) {
4593 error = got_error_from_errno("view_open");
4594 goto done;
4596 error = open_blame_view(view, in_repo_path, commit_id, &refs, repo);
4597 if (error)
4598 goto done;
4599 if (worktree) {
4600 /* Release work tree lock. */
4601 got_worktree_close(worktree);
4602 worktree = NULL;
4604 error = view_loop(view);
4605 done:
4606 free(repo_path);
4607 free(in_repo_path);
4608 free(cwd);
4609 free(commit_id);
4610 if (worktree)
4611 got_worktree_close(worktree);
4612 if (repo)
4613 got_repo_close(repo);
4614 got_ref_list_free(&refs);
4615 return error;
4618 static const struct got_error *
4619 draw_tree_entries(struct tog_view *view,
4620 struct got_tree_entry **first_displayed_entry,
4621 struct got_tree_entry **last_displayed_entry,
4622 struct got_tree_entry **selected_entry, int *ndisplayed,
4623 const char *label, int show_ids, const char *parent_path,
4624 struct got_tree_object *tree, int selected, int limit,
4625 int isroot, struct tog_colors *colors)
4627 const struct got_error *err = NULL;
4628 struct got_tree_entry *te;
4629 wchar_t *wline;
4630 struct tog_color *tc;
4631 int width, n, i, nentries;
4633 *ndisplayed = 0;
4635 werase(view->window);
4637 if (limit == 0)
4638 return NULL;
4640 err = format_line(&wline, &width, label, view->ncols, 0);
4641 if (err)
4642 return err;
4643 if (view_needs_focus_indication(view))
4644 wstandout(view->window);
4645 tc = get_color(colors, TOG_COLOR_COMMIT);
4646 if (tc)
4647 wattr_on(view->window,
4648 COLOR_PAIR(tc->colorpair), NULL);
4649 waddwstr(view->window, wline);
4650 if (tc)
4651 wattr_off(view->window,
4652 COLOR_PAIR(tc->colorpair), NULL);
4653 if (view_needs_focus_indication(view))
4654 wstandend(view->window);
4655 free(wline);
4656 wline = NULL;
4657 if (width < view->ncols - 1)
4658 waddch(view->window, '\n');
4659 if (--limit <= 0)
4660 return NULL;
4661 err = format_line(&wline, &width, parent_path, view->ncols, 0);
4662 if (err)
4663 return err;
4664 waddwstr(view->window, wline);
4665 free(wline);
4666 wline = NULL;
4667 if (width < view->ncols - 1)
4668 waddch(view->window, '\n');
4669 if (--limit <= 0)
4670 return NULL;
4671 waddch(view->window, '\n');
4672 if (--limit <= 0)
4673 return NULL;
4675 if (*first_displayed_entry == NULL) {
4676 te = got_object_tree_get_first_entry(tree);
4677 if (selected == 0) {
4678 if (view->focussed)
4679 wstandout(view->window);
4680 *selected_entry = NULL;
4682 waddstr(view->window, " ..\n"); /* parent directory */
4683 if (selected == 0 && view->focussed)
4684 wstandend(view->window);
4685 (*ndisplayed)++;
4686 if (--limit <= 0)
4687 return NULL;
4688 n = 1;
4689 } else {
4690 n = 0;
4691 te = *first_displayed_entry;
4694 nentries = got_object_tree_get_nentries(tree);
4695 for (i = got_tree_entry_get_index(te); i < nentries; i++) {
4696 char *line = NULL, *id_str = NULL;
4697 const char *modestr = "";
4698 mode_t mode;
4700 te = got_object_tree_get_entry(tree, i);
4701 mode = got_tree_entry_get_mode(te);
4703 if (show_ids) {
4704 err = got_object_id_str(&id_str,
4705 got_tree_entry_get_id(te));
4706 if (err)
4707 return got_error_from_errno(
4708 "got_object_id_str");
4710 if (got_object_tree_entry_is_submodule(te))
4711 modestr = "$";
4712 else if (S_ISLNK(mode))
4713 modestr = "@";
4714 else if (S_ISDIR(mode))
4715 modestr = "/";
4716 else if (mode & S_IXUSR)
4717 modestr = "*";
4718 if (asprintf(&line, "%s %s%s", id_str ? id_str : "",
4719 got_tree_entry_get_name(te), modestr) == -1) {
4720 free(id_str);
4721 return got_error_from_errno("asprintf");
4723 free(id_str);
4724 err = format_line(&wline, &width, line, view->ncols, 0);
4725 if (err) {
4726 free(line);
4727 break;
4729 if (n == selected) {
4730 if (view->focussed)
4731 wstandout(view->window);
4732 *selected_entry = te;
4734 tc = match_color(colors, line);
4735 if (tc)
4736 wattr_on(view->window,
4737 COLOR_PAIR(tc->colorpair), NULL);
4738 waddwstr(view->window, wline);
4739 if (tc)
4740 wattr_off(view->window,
4741 COLOR_PAIR(tc->colorpair), NULL);
4742 if (width < view->ncols - 1)
4743 waddch(view->window, '\n');
4744 if (n == selected && view->focussed)
4745 wstandend(view->window);
4746 free(line);
4747 free(wline);
4748 wline = NULL;
4749 n++;
4750 (*ndisplayed)++;
4751 *last_displayed_entry = te;
4752 if (--limit <= 0)
4753 break;
4756 return err;
4759 static void
4760 tree_scroll_up(struct tog_view *view,
4761 struct got_tree_entry **first_displayed_entry, int maxscroll,
4762 struct got_tree_object *tree, int isroot)
4764 struct got_tree_entry *te;
4765 int i;
4767 if (*first_displayed_entry == NULL)
4768 return;
4770 te = got_object_tree_get_entry(tree, 0);
4771 if (*first_displayed_entry == te) {
4772 if (!isroot)
4773 *first_displayed_entry = NULL;
4774 return;
4777 i = 0;
4778 while (*first_displayed_entry && i < maxscroll) {
4779 *first_displayed_entry = got_tree_entry_get_prev(tree,
4780 *first_displayed_entry);
4781 i++;
4783 if (!isroot && te == got_object_tree_get_first_entry(tree) && i < maxscroll)
4784 *first_displayed_entry = NULL;
4787 static int
4788 tree_scroll_down(struct got_tree_entry **first_displayed_entry, int maxscroll,
4789 struct got_tree_entry *last_displayed_entry,
4790 struct got_tree_object *tree)
4792 struct got_tree_entry *next, *last;
4793 int n = 0;
4795 if (*first_displayed_entry)
4796 next = got_tree_entry_get_next(tree, *first_displayed_entry);
4797 else
4798 next = got_object_tree_get_first_entry(tree);
4800 last = last_displayed_entry;
4801 while (next && last && n++ < maxscroll) {
4802 last = got_tree_entry_get_next(tree, last);
4803 if (last) {
4804 *first_displayed_entry = next;
4805 next = got_tree_entry_get_next(tree, next);
4808 return n;
4811 static const struct got_error *
4812 tree_entry_path(char **path, struct tog_parent_trees *parents,
4813 struct got_tree_entry *te)
4815 const struct got_error *err = NULL;
4816 struct tog_parent_tree *pt;
4817 size_t len = 2; /* for leading slash and NUL */
4819 TAILQ_FOREACH(pt, parents, entry)
4820 len += strlen(got_tree_entry_get_name(pt->selected_entry))
4821 + 1 /* slash */;
4822 if (te)
4823 len += strlen(got_tree_entry_get_name(te));
4825 *path = calloc(1, len);
4826 if (path == NULL)
4827 return got_error_from_errno("calloc");
4829 (*path)[0] = '/';
4830 pt = TAILQ_LAST(parents, tog_parent_trees);
4831 while (pt) {
4832 const char *name = got_tree_entry_get_name(pt->selected_entry);
4833 if (strlcat(*path, name, len) >= len) {
4834 err = got_error(GOT_ERR_NO_SPACE);
4835 goto done;
4837 if (strlcat(*path, "/", len) >= len) {
4838 err = got_error(GOT_ERR_NO_SPACE);
4839 goto done;
4841 pt = TAILQ_PREV(pt, tog_parent_trees, entry);
4843 if (te) {
4844 if (strlcat(*path, got_tree_entry_get_name(te), len) >= len) {
4845 err = got_error(GOT_ERR_NO_SPACE);
4846 goto done;
4849 done:
4850 if (err) {
4851 free(*path);
4852 *path = NULL;
4854 return err;
4857 static const struct got_error *
4858 blame_tree_entry(struct tog_view **new_view, int begin_x,
4859 struct got_tree_entry *te, struct tog_parent_trees *parents,
4860 struct got_object_id *commit_id, struct got_reflist_head *refs,
4861 struct got_repository *repo)
4863 const struct got_error *err = NULL;
4864 char *path;
4865 struct tog_view *blame_view;
4867 *new_view = NULL;
4869 err = tree_entry_path(&path, parents, te);
4870 if (err)
4871 return err;
4873 blame_view = view_open(0, 0, 0, begin_x, TOG_VIEW_BLAME);
4874 if (blame_view == NULL) {
4875 err = got_error_from_errno("view_open");
4876 goto done;
4879 err = open_blame_view(blame_view, path, commit_id, refs, repo);
4880 if (err) {
4881 if (err->code == GOT_ERR_CANCELLED)
4882 err = NULL;
4883 view_close(blame_view);
4884 } else
4885 *new_view = blame_view;
4886 done:
4887 free(path);
4888 return err;
4891 static const struct got_error *
4892 log_tree_entry(struct tog_view **new_view, int begin_x,
4893 struct got_tree_entry *te, struct tog_parent_trees *parents,
4894 struct got_object_id *commit_id, struct got_reflist_head *refs,
4895 struct got_repository *repo)
4897 struct tog_view *log_view;
4898 const struct got_error *err = NULL;
4899 char *path;
4901 *new_view = NULL;
4903 log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
4904 if (log_view == NULL)
4905 return got_error_from_errno("view_open");
4907 err = tree_entry_path(&path, parents, te);
4908 if (err)
4909 return err;
4911 err = open_log_view(log_view, commit_id, refs, repo, NULL, path, 0);
4912 if (err)
4913 view_close(log_view);
4914 else
4915 *new_view = log_view;
4916 free(path);
4917 return err;
4920 static const struct got_error *
4921 open_tree_view(struct tog_view *view, struct got_tree_object *root,
4922 struct got_object_id *commit_id, struct got_reflist_head *refs,
4923 struct got_repository *repo)
4925 const struct got_error *err = NULL;
4926 char *commit_id_str = NULL;
4927 struct tog_tree_view_state *s = &view->state.tree;
4929 TAILQ_INIT(&s->parents);
4931 err = got_object_id_str(&commit_id_str, commit_id);
4932 if (err != NULL)
4933 goto done;
4935 if (asprintf(&s->tree_label, "commit %s", commit_id_str) == -1) {
4936 err = got_error_from_errno("asprintf");
4937 goto done;
4940 s->root = s->tree = root;
4941 s->first_displayed_entry = got_object_tree_get_entry(s->tree, 0);
4942 s->selected_entry = got_object_tree_get_entry(s->tree, 0);
4943 s->commit_id = got_object_id_dup(commit_id);
4944 if (s->commit_id == NULL) {
4945 err = got_error_from_errno("got_object_id_dup");
4946 goto done;
4948 s->refs = refs;
4949 s->repo = repo;
4951 SIMPLEQ_INIT(&s->colors);
4953 if (has_colors() && getenv("TOG_COLORS") != NULL) {
4954 err = add_color(&s->colors, "\\$$",
4955 TOG_COLOR_TREE_SUBMODULE,
4956 get_color_value("TOG_COLOR_TREE_SUBMODULE"));
4957 if (err)
4958 goto done;
4959 err = add_color(&s->colors, "@$", TOG_COLOR_TREE_SYMLINK,
4960 get_color_value("TOG_COLOR_TREE_SYMLINK"));
4961 if (err) {
4962 free_colors(&s->colors);
4963 goto done;
4965 err = add_color(&s->colors, "/$",
4966 TOG_COLOR_TREE_DIRECTORY,
4967 get_color_value("TOG_COLOR_TREE_DIRECTORY"));
4968 if (err) {
4969 free_colors(&s->colors);
4970 goto done;
4973 err = add_color(&s->colors, "\\*$",
4974 TOG_COLOR_TREE_EXECUTABLE,
4975 get_color_value("TOG_COLOR_TREE_EXECUTABLE"));
4976 if (err) {
4977 free_colors(&s->colors);
4978 goto done;
4981 err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
4982 get_color_value("TOG_COLOR_COMMIT"));
4983 if (err) {
4984 free_colors(&s->colors);
4985 goto done;
4989 view->show = show_tree_view;
4990 view->input = input_tree_view;
4991 view->close = close_tree_view;
4992 view->search_start = search_start_tree_view;
4993 view->search_next = search_next_tree_view;
4994 done:
4995 free(commit_id_str);
4996 if (err) {
4997 free(s->tree_label);
4998 s->tree_label = NULL;
5000 return err;
5003 static const struct got_error *
5004 close_tree_view(struct tog_view *view)
5006 struct tog_tree_view_state *s = &view->state.tree;
5008 free_colors(&s->colors);
5009 free(s->tree_label);
5010 s->tree_label = NULL;
5011 free(s->commit_id);
5012 s->commit_id = NULL;
5013 while (!TAILQ_EMPTY(&s->parents)) {
5014 struct tog_parent_tree *parent;
5015 parent = TAILQ_FIRST(&s->parents);
5016 TAILQ_REMOVE(&s->parents, parent, entry);
5017 free(parent);
5020 if (s->tree != s->root)
5021 got_object_tree_close(s->tree);
5022 got_object_tree_close(s->root);
5024 return NULL;
5027 static const struct got_error *
5028 search_start_tree_view(struct tog_view *view)
5030 struct tog_tree_view_state *s = &view->state.tree;
5032 s->matched_entry = NULL;
5033 return NULL;
5036 static int
5037 match_tree_entry(struct got_tree_entry *te, regex_t *regex)
5039 regmatch_t regmatch;
5041 return regexec(regex, got_tree_entry_get_name(te), 1, &regmatch,
5042 0) == 0;
5045 static const struct got_error *
5046 search_next_tree_view(struct tog_view *view)
5048 struct tog_tree_view_state *s = &view->state.tree;
5049 struct got_tree_entry *te = NULL;
5051 if (!view->searching) {
5052 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5053 return NULL;
5056 if (s->matched_entry) {
5057 if (view->searching == TOG_SEARCH_FORWARD) {
5058 if (s->selected_entry)
5059 te = got_tree_entry_get_next(s->tree,
5060 s->selected_entry);
5061 else
5062 te = got_object_tree_get_first_entry(s->tree);
5063 } else {
5064 if (s->selected_entry == NULL)
5065 te = got_object_tree_get_last_entry(s->tree);
5066 else
5067 te = got_tree_entry_get_prev(s->tree,
5068 s->selected_entry);
5070 } else {
5071 if (view->searching == TOG_SEARCH_FORWARD)
5072 te = got_object_tree_get_first_entry(s->tree);
5073 else
5074 te = got_object_tree_get_last_entry(s->tree);
5077 while (1) {
5078 if (te == NULL) {
5079 if (s->matched_entry == NULL) {
5080 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5081 return NULL;
5083 if (view->searching == TOG_SEARCH_FORWARD)
5084 te = got_object_tree_get_first_entry(s->tree);
5085 else
5086 te = got_object_tree_get_last_entry(s->tree);
5089 if (match_tree_entry(te, &view->regex)) {
5090 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5091 s->matched_entry = te;
5092 break;
5095 if (view->searching == TOG_SEARCH_FORWARD)
5096 te = got_tree_entry_get_next(s->tree, te);
5097 else
5098 te = got_tree_entry_get_prev(s->tree, te);
5101 if (s->matched_entry) {
5102 s->first_displayed_entry = s->matched_entry;
5103 s->selected = 0;
5106 return NULL;
5109 static const struct got_error *
5110 show_tree_view(struct tog_view *view)
5112 const struct got_error *err = NULL;
5113 struct tog_tree_view_state *s = &view->state.tree;
5114 char *parent_path;
5116 err = tree_entry_path(&parent_path, &s->parents, NULL);
5117 if (err)
5118 return err;
5120 err = draw_tree_entries(view, &s->first_displayed_entry,
5121 &s->last_displayed_entry, &s->selected_entry,
5122 &s->ndisplayed, s->tree_label, s->show_ids, parent_path,
5123 s->tree, s->selected, view->nlines, s->tree == s->root,
5124 &s->colors);
5125 free(parent_path);
5127 view_vborder(view);
5128 return err;
5131 static const struct got_error *
5132 input_tree_view(struct tog_view **new_view, struct tog_view **dead_view,
5133 struct tog_view **focus_view, struct tog_view *view, int ch)
5135 const struct got_error *err = NULL;
5136 struct tog_tree_view_state *s = &view->state.tree;
5137 struct tog_view *log_view;
5138 int begin_x = 0, nscrolled;
5140 switch (ch) {
5141 case 'i':
5142 s->show_ids = !s->show_ids;
5143 break;
5144 case 'l':
5145 if (!s->selected_entry)
5146 break;
5147 if (view_is_parent_view(view))
5148 begin_x = view_split_begin_x(view->begin_x);
5149 err = log_tree_entry(&log_view, begin_x,
5150 s->selected_entry, &s->parents,
5151 s->commit_id, s->refs, s->repo);
5152 if (view_is_parent_view(view)) {
5153 err = view_close_child(view);
5154 if (err)
5155 return err;
5156 err = view_set_child(view, log_view);
5157 if (err) {
5158 view_close(log_view);
5159 break;
5161 *focus_view = log_view;
5162 view->child_focussed = 1;
5163 } else
5164 *new_view = log_view;
5165 break;
5166 case 'k':
5167 case KEY_UP:
5168 if (s->selected > 0) {
5169 s->selected--;
5170 if (s->selected == 0)
5171 break;
5173 if (s->selected > 0)
5174 break;
5175 tree_scroll_up(view, &s->first_displayed_entry, 1,
5176 s->tree, s->tree == s->root);
5177 break;
5178 case KEY_PPAGE:
5179 case CTRL('b'):
5180 tree_scroll_up(view, &s->first_displayed_entry,
5181 MAX(0, view->nlines - 4 - s->selected), s->tree,
5182 s->tree == s->root);
5183 s->selected = 0;
5184 if (got_object_tree_get_first_entry(s->tree) ==
5185 s->first_displayed_entry && s->tree != s->root)
5186 s->first_displayed_entry = NULL;
5187 break;
5188 case 'j':
5189 case KEY_DOWN:
5190 if (s->selected < s->ndisplayed - 1) {
5191 s->selected++;
5192 break;
5194 if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
5195 == NULL)
5196 /* can't scroll any further */
5197 break;
5198 tree_scroll_down(&s->first_displayed_entry, 1,
5199 s->last_displayed_entry, s->tree);
5200 break;
5201 case KEY_NPAGE:
5202 case CTRL('f'):
5203 if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
5204 == NULL) {
5205 /* can't scroll any further; move cursor down */
5206 if (s->selected < s->ndisplayed - 1)
5207 s->selected = s->ndisplayed - 1;
5208 break;
5210 nscrolled = tree_scroll_down(&s->first_displayed_entry,
5211 view->nlines, s->last_displayed_entry, s->tree);
5212 if (nscrolled < view->nlines) {
5213 int ndisplayed = 0;
5214 struct got_tree_entry *te;
5215 te = s->first_displayed_entry;
5216 do {
5217 ndisplayed++;
5218 te = got_tree_entry_get_next(s->tree, te);
5219 } while (te);
5220 s->selected = ndisplayed - 1;
5222 break;
5223 case KEY_ENTER:
5224 case '\r':
5225 case KEY_BACKSPACE:
5226 if (s->selected_entry == NULL || ch == KEY_BACKSPACE) {
5227 struct tog_parent_tree *parent;
5228 /* user selected '..' */
5229 if (s->tree == s->root)
5230 break;
5231 parent = TAILQ_FIRST(&s->parents);
5232 TAILQ_REMOVE(&s->parents, parent,
5233 entry);
5234 got_object_tree_close(s->tree);
5235 s->tree = parent->tree;
5236 s->first_displayed_entry =
5237 parent->first_displayed_entry;
5238 s->selected_entry =
5239 parent->selected_entry;
5240 s->selected = parent->selected;
5241 free(parent);
5242 } else if (S_ISDIR(got_tree_entry_get_mode(
5243 s->selected_entry))) {
5244 struct got_tree_object *subtree;
5245 err = got_object_open_as_tree(&subtree, s->repo,
5246 got_tree_entry_get_id(s->selected_entry));
5247 if (err)
5248 break;
5249 err = tree_view_visit_subtree(subtree, s);
5250 if (err) {
5251 got_object_tree_close(subtree);
5252 break;
5254 } else if (S_ISREG(got_tree_entry_get_mode(
5255 s->selected_entry))) {
5256 struct tog_view *blame_view;
5257 int begin_x = view_is_parent_view(view) ?
5258 view_split_begin_x(view->begin_x) : 0;
5260 err = blame_tree_entry(&blame_view, begin_x,
5261 s->selected_entry, &s->parents,
5262 s->commit_id, s->refs, s->repo);
5263 if (err)
5264 break;
5265 if (view_is_parent_view(view)) {
5266 err = view_close_child(view);
5267 if (err)
5268 return err;
5269 err = view_set_child(view, blame_view);
5270 if (err) {
5271 view_close(blame_view);
5272 break;
5274 *focus_view = blame_view;
5275 view->child_focussed = 1;
5276 } else
5277 *new_view = blame_view;
5279 break;
5280 case KEY_RESIZE:
5281 if (s->selected > view->nlines)
5282 s->selected = s->ndisplayed - 1;
5283 break;
5284 default:
5285 break;
5288 return err;
5291 __dead static void
5292 usage_tree(void)
5294 endwin();
5295 fprintf(stderr, "usage: %s tree [-c commit] [-r repository-path] [path]\n",
5296 getprogname());
5297 exit(1);
5300 static const struct got_error *
5301 cmd_tree(int argc, char *argv[])
5303 const struct got_error *error;
5304 struct got_repository *repo = NULL;
5305 struct got_worktree *worktree = NULL;
5306 struct got_reflist_head refs;
5307 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5308 struct got_object_id *commit_id = NULL;
5309 char *commit_id_arg = NULL;
5310 struct got_commit_object *commit = NULL;
5311 struct got_tree_object *tree = NULL;
5312 int ch;
5313 struct tog_view *view;
5315 SIMPLEQ_INIT(&refs);
5317 #ifndef PROFILE
5318 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
5319 NULL) == -1)
5320 err(1, "pledge");
5321 #endif
5323 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
5324 switch (ch) {
5325 case 'c':
5326 commit_id_arg = optarg;
5327 break;
5328 case 'r':
5329 repo_path = realpath(optarg, NULL);
5330 if (repo_path == NULL)
5331 return got_error_from_errno2("realpath",
5332 optarg);
5333 break;
5334 default:
5335 usage_tree();
5336 /* NOTREACHED */
5340 argc -= optind;
5341 argv += optind;
5343 if (argc > 1)
5344 usage_tree();
5346 cwd = getcwd(NULL, 0);
5347 if (cwd == NULL)
5348 return got_error_from_errno("getcwd");
5350 error = got_worktree_open(&worktree, cwd);
5351 if (error && error->code != GOT_ERR_NOT_WORKTREE)
5352 goto done;
5354 if (repo_path == NULL) {
5355 if (worktree)
5356 repo_path =
5357 strdup(got_worktree_get_repo_path(worktree));
5358 else
5359 repo_path = strdup(cwd);
5361 if (repo_path == NULL) {
5362 error = got_error_from_errno("strdup");
5363 goto done;
5366 error = got_repo_open(&repo, repo_path, NULL);
5367 if (error != NULL)
5368 goto done;
5370 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
5371 repo, worktree);
5372 if (error)
5373 goto done;
5375 init_curses();
5377 error = apply_unveil(got_repo_get_path(repo), NULL);
5378 if (error)
5379 goto done;
5381 error = got_repo_match_object_id(&commit_id, NULL,
5382 commit_id_arg ? commit_id_arg : GOT_REF_HEAD,
5383 GOT_OBJ_TYPE_COMMIT, 1, repo);
5384 if (error)
5385 goto done;
5387 error = got_object_open_as_commit(&commit, repo, commit_id);
5388 if (error)
5389 goto done;
5391 error = got_object_open_as_tree(&tree, repo,
5392 got_object_commit_get_tree_id(commit));
5393 if (error)
5394 goto done;
5396 error = got_ref_list(&refs, repo, NULL, got_ref_cmp_by_name, NULL);
5397 if (error)
5398 goto done;
5400 view = view_open(0, 0, 0, 0, TOG_VIEW_TREE);
5401 if (view == NULL) {
5402 error = got_error_from_errno("view_open");
5403 goto done;
5405 error = open_tree_view(view, tree, commit_id, &refs, repo);
5406 if (error)
5407 goto done;
5408 if (!got_path_is_root_dir(in_repo_path)) {
5409 error = tree_view_walk_path(&view->state.tree, commit_id,
5410 in_repo_path, repo);
5411 if (error)
5412 goto done;
5415 if (worktree) {
5416 /* Release work tree lock. */
5417 got_worktree_close(worktree);
5418 worktree = NULL;
5420 error = view_loop(view);
5421 done:
5422 free(repo_path);
5423 free(cwd);
5424 free(commit_id);
5425 if (commit)
5426 got_object_commit_close(commit);
5427 if (tree)
5428 got_object_tree_close(tree);
5429 if (repo)
5430 got_repo_close(repo);
5431 got_ref_list_free(&refs);
5432 return error;
5435 static void
5436 list_commands(void)
5438 int i;
5440 fprintf(stderr, "commands:");
5441 for (i = 0; i < nitems(tog_commands); i++) {
5442 struct tog_cmd *cmd = &tog_commands[i];
5443 fprintf(stderr, " %s", cmd->name);
5445 fputc('\n', stderr);
5448 __dead static void
5449 usage(int hflag)
5451 fprintf(stderr, "usage: %s [-h] [-V | --version] [command] "
5452 "[arg ...]\n", getprogname());
5453 if (hflag) {
5454 fprintf(stderr, "lazy usage: %s path\n", getprogname());
5455 list_commands();
5457 exit(1);
5460 static char **
5461 make_argv(int argc, ...)
5463 va_list ap;
5464 char **argv;
5465 int i;
5467 va_start(ap, argc);
5469 argv = calloc(argc, sizeof(char *));
5470 if (argv == NULL)
5471 err(1, "calloc");
5472 for (i = 0; i < argc; i++) {
5473 argv[i] = strdup(va_arg(ap, char *));
5474 if (argv[i] == NULL)
5475 err(1, "strdup");
5478 va_end(ap);
5479 return argv;
5483 * Try to convert 'tog path' into a 'tog log path' command.
5484 * The user could simply have mistyped the command rather than knowingly
5485 * provided a path. So check whether argv[0] can in fact be resolved
5486 * to a path in the HEAD commit and print a special error if not.
5487 * This hack is for mpi@ <3
5489 static const struct got_error *
5490 tog_log_with_path(int argc, char *argv[])
5492 const struct got_error *error = NULL;
5493 struct tog_cmd *cmd = NULL;
5494 struct got_repository *repo = NULL;
5495 struct got_worktree *worktree = NULL;
5496 struct got_object_id *commit_id = NULL, *id = NULL;
5497 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5498 char *commit_id_str = NULL, **cmd_argv = NULL;
5500 cwd = getcwd(NULL, 0);
5501 if (cwd == NULL)
5502 return got_error_from_errno("getcwd");
5504 error = got_worktree_open(&worktree, cwd);
5505 if (error && error->code != GOT_ERR_NOT_WORKTREE)
5506 goto done;
5508 if (worktree)
5509 repo_path = strdup(got_worktree_get_repo_path(worktree));
5510 else
5511 repo_path = strdup(cwd);
5512 if (repo_path == NULL) {
5513 error = got_error_from_errno("strdup");
5514 goto done;
5517 error = got_repo_open(&repo, repo_path, NULL);
5518 if (error != NULL)
5519 goto done;
5521 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
5522 repo, worktree);
5523 if (error)
5524 goto done;
5526 error = got_repo_match_object_id(&commit_id, NULL, worktree ?
5527 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
5528 GOT_OBJ_TYPE_COMMIT, 1, repo);
5529 if (error)
5530 goto done;
5532 if (worktree) {
5533 got_worktree_close(worktree);
5534 worktree = NULL;
5537 error = got_object_id_by_path(&id, repo, commit_id, in_repo_path);
5538 if (error) {
5539 if (error->code != GOT_ERR_NO_TREE_ENTRY)
5540 goto done;
5541 fprintf(stderr, "%s: '%s' is no known command or path\n",
5542 getprogname(), argv[0]);
5543 usage(1);
5544 /* not reached */
5547 got_repo_close(repo);
5548 repo = NULL;
5550 error = got_object_id_str(&commit_id_str, commit_id);
5551 if (error)
5552 goto done;
5554 cmd = &tog_commands[0]; /* log */
5555 argc = 4;
5556 cmd_argv = make_argv(argc, cmd->name, "-c", commit_id_str, argv[0]);
5557 error = cmd->cmd_main(argc, cmd_argv);
5558 done:
5559 if (repo)
5560 got_repo_close(repo);
5561 if (worktree)
5562 got_worktree_close(worktree);
5563 free(id);
5564 free(commit_id_str);
5565 free(commit_id);
5566 free(cwd);
5567 free(repo_path);
5568 free(in_repo_path);
5569 if (cmd_argv) {
5570 int i;
5571 for (i = 0; i < argc; i++)
5572 free(cmd_argv[i]);
5573 free(cmd_argv);
5575 return error;
5578 int
5579 main(int argc, char *argv[])
5581 const struct got_error *error = NULL;
5582 struct tog_cmd *cmd = NULL;
5583 int ch, hflag = 0, Vflag = 0;
5584 char **cmd_argv = NULL;
5585 static struct option longopts[] = {
5586 { "version", no_argument, NULL, 'V' },
5587 { NULL, 0, NULL, 0}
5590 setlocale(LC_CTYPE, "");
5592 while ((ch = getopt_long(argc, argv, "+hV", longopts, NULL)) != -1) {
5593 switch (ch) {
5594 case 'h':
5595 hflag = 1;
5596 break;
5597 case 'V':
5598 Vflag = 1;
5599 break;
5600 default:
5601 usage(hflag);
5602 /* NOTREACHED */
5606 argc -= optind;
5607 argv += optind;
5608 optind = 0;
5609 optreset = 1;
5611 if (Vflag) {
5612 got_version_print_str();
5613 return 1;
5616 if (argc == 0) {
5617 if (hflag)
5618 usage(hflag);
5619 /* Build an argument vector which runs a default command. */
5620 cmd = &tog_commands[0];
5621 argc = 1;
5622 cmd_argv = make_argv(argc, cmd->name);
5623 } else {
5624 int i;
5626 /* Did the user specify a command? */
5627 for (i = 0; i < nitems(tog_commands); i++) {
5628 if (strncmp(tog_commands[i].name, argv[0],
5629 strlen(argv[0])) == 0) {
5630 cmd = &tog_commands[i];
5631 break;
5636 if (cmd == NULL) {
5637 if (argc != 1)
5638 usage(0);
5639 /* No command specified; try log with a path */
5640 error = tog_log_with_path(argc, argv);
5641 } else {
5642 if (hflag)
5643 cmd->cmd_usage();
5644 else
5645 error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
5648 endwin();
5649 if (cmd_argv) {
5650 int i;
5651 for (i = 0; i < argc; i++)
5652 free(cmd_argv[i]);
5653 free(cmd_argv);
5656 if (error && error->code != GOT_ERR_CANCELLED)
5657 fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
5658 return 0;