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/stat.h>
18 #include <sys/ioctl.h>
20 #include <ctype.h>
21 #include <errno.h>
22 #if defined(__FreeBSD__) || defined(__APPLE__)
23 #define _XOPEN_SOURCE_EXTENDED /* for ncurses wide-character functions */
24 #endif
25 #include <curses.h>
26 #include <panel.h>
27 #include <locale.h>
28 #include <signal.h>
29 #include <stdlib.h>
30 #include <stdarg.h>
31 #include <stdio.h>
32 #include <getopt.h>
33 #include <string.h>
34 #include <err.h>
35 #include <unistd.h>
36 #include <limits.h>
37 #include <wchar.h>
38 #include <time.h>
39 #include <pthread.h>
40 #include <libgen.h>
41 #include <regex.h>
42 #include <sched.h>
44 #include "got_compat.h"
46 #include "got_version.h"
47 #include "got_error.h"
48 #include "got_object.h"
49 #include "got_reference.h"
50 #include "got_repository.h"
51 #include "got_diff.h"
52 #include "got_opentemp.h"
53 #include "got_utf8.h"
54 #include "got_cancel.h"
55 #include "got_commit_graph.h"
56 #include "got_blame.h"
57 #include "got_privsep.h"
58 #include "got_path.h"
59 #include "got_worktree.h"
61 //#define update_panels() (0)
62 //#define doupdate() (0)
64 #ifndef MIN
65 #define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
66 #endif
68 #ifndef MAX
69 #define MAX(_a,_b) ((_a) > (_b) ? (_a) : (_b))
70 #endif
72 #ifndef CTRL
73 #define CTRL(x) ((x) & 0x1f)
74 #endif
76 #ifndef nitems
77 #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
78 #endif
80 struct tog_cmd {
81 const char *name;
82 const struct got_error *(*cmd_main)(int, char *[]);
83 void (*cmd_usage)(void);
84 };
86 __dead static void usage(int, int);
87 __dead static void usage_log(void);
88 __dead static void usage_diff(void);
89 __dead static void usage_blame(void);
90 __dead static void usage_tree(void);
91 __dead static void usage_ref(void);
93 static const struct got_error* cmd_log(int, char *[]);
94 static const struct got_error* cmd_diff(int, char *[]);
95 static const struct got_error* cmd_blame(int, char *[]);
96 static const struct got_error* cmd_tree(int, char *[]);
97 static const struct got_error* cmd_ref(int, char *[]);
99 static const struct tog_cmd tog_commands[] = {
100 { "log", cmd_log, usage_log },
101 { "diff", cmd_diff, usage_diff },
102 { "blame", cmd_blame, usage_blame },
103 { "tree", cmd_tree, usage_tree },
104 { "ref", cmd_ref, usage_ref },
105 };
107 enum tog_view_type {
108 TOG_VIEW_DIFF,
109 TOG_VIEW_LOG,
110 TOG_VIEW_BLAME,
111 TOG_VIEW_TREE,
112 TOG_VIEW_REF,
113 };
115 #define TOG_EOF_STRING "(END)"
117 struct commit_queue_entry {
118 TAILQ_ENTRY(commit_queue_entry) entry;
119 struct got_object_id *id;
120 struct got_commit_object *commit;
121 int idx;
122 };
123 TAILQ_HEAD(commit_queue_head, commit_queue_entry);
124 struct commit_queue {
125 int ncommits;
126 struct commit_queue_head head;
127 };
129 struct tog_color {
130 STAILQ_ENTRY(tog_color) entry;
131 regex_t regex;
132 short colorpair;
133 };
134 STAILQ_HEAD(tog_colors, tog_color);
136 static struct got_reflist_head tog_refs = TAILQ_HEAD_INITIALIZER(tog_refs);
137 static struct got_reflist_object_id_map *tog_refs_idmap;
139 static const struct got_error *
140 tog_ref_cmp_by_name(void *arg, int *cmp, struct got_reference *re1,
141 struct got_reference* re2)
143 const char *name1 = got_ref_get_name(re1);
144 const char *name2 = got_ref_get_name(re2);
145 int isbackup1, isbackup2;
147 /* Sort backup refs towards the bottom of the list. */
148 isbackup1 = strncmp(name1, "refs/got/backup/", 16) == 0;
149 isbackup2 = strncmp(name2, "refs/got/backup/", 16) == 0;
150 if (!isbackup1 && isbackup2) {
151 *cmp = -1;
152 return NULL;
153 } else if (isbackup1 && !isbackup2) {
154 *cmp = 1;
155 return NULL;
158 *cmp = got_path_cmp(name1, name2, strlen(name1), strlen(name2));
159 return NULL;
162 static const struct got_error *
163 tog_load_refs(struct got_repository *repo, int sort_by_date)
165 const struct got_error *err;
167 err = got_ref_list(&tog_refs, repo, NULL, sort_by_date ?
168 got_ref_cmp_by_commit_timestamp_descending : tog_ref_cmp_by_name,
169 repo);
170 if (err)
171 return err;
173 return got_reflist_object_id_map_create(&tog_refs_idmap, &tog_refs,
174 repo);
177 static void
178 tog_free_refs(void)
180 if (tog_refs_idmap) {
181 got_reflist_object_id_map_free(tog_refs_idmap);
182 tog_refs_idmap = NULL;
184 got_ref_list_free(&tog_refs);
187 static const struct got_error *
188 add_color(struct tog_colors *colors, const char *pattern,
189 int idx, short color)
191 const struct got_error *err = NULL;
192 struct tog_color *tc;
193 int regerr = 0;
195 if (idx < 1 || idx > COLOR_PAIRS - 1)
196 return NULL;
198 init_pair(idx, color, -1);
200 tc = calloc(1, sizeof(*tc));
201 if (tc == NULL)
202 return got_error_from_errno("calloc");
203 regerr = regcomp(&tc->regex, pattern,
204 REG_EXTENDED | REG_NOSUB | REG_NEWLINE);
205 if (regerr) {
206 static char regerr_msg[512];
207 static char err_msg[512];
208 regerror(regerr, &tc->regex, regerr_msg,
209 sizeof(regerr_msg));
210 snprintf(err_msg, sizeof(err_msg), "regcomp: %s",
211 regerr_msg);
212 err = got_error_msg(GOT_ERR_REGEX, err_msg);
213 free(tc);
214 return err;
216 tc->colorpair = idx;
217 STAILQ_INSERT_HEAD(colors, tc, entry);
218 return NULL;
221 static void
222 free_colors(struct tog_colors *colors)
224 struct tog_color *tc;
226 while (!STAILQ_EMPTY(colors)) {
227 tc = STAILQ_FIRST(colors);
228 STAILQ_REMOVE_HEAD(colors, entry);
229 regfree(&tc->regex);
230 free(tc);
234 struct tog_color *
235 get_color(struct tog_colors *colors, int colorpair)
237 struct tog_color *tc = NULL;
239 STAILQ_FOREACH(tc, colors, entry) {
240 if (tc->colorpair == colorpair)
241 return tc;
244 return NULL;
247 static int
248 default_color_value(const char *envvar)
250 if (strcmp(envvar, "TOG_COLOR_DIFF_MINUS") == 0)
251 return COLOR_MAGENTA;
252 if (strcmp(envvar, "TOG_COLOR_DIFF_PLUS") == 0)
253 return COLOR_CYAN;
254 if (strcmp(envvar, "TOG_COLOR_DIFF_CHUNK_HEADER") == 0)
255 return COLOR_YELLOW;
256 if (strcmp(envvar, "TOG_COLOR_DIFF_META") == 0)
257 return COLOR_GREEN;
258 if (strcmp(envvar, "TOG_COLOR_TREE_SUBMODULE") == 0)
259 return COLOR_MAGENTA;
260 if (strcmp(envvar, "TOG_COLOR_TREE_SYMLINK") == 0)
261 return COLOR_MAGENTA;
262 if (strcmp(envvar, "TOG_COLOR_TREE_DIRECTORY") == 0)
263 return COLOR_CYAN;
264 if (strcmp(envvar, "TOG_COLOR_TREE_EXECUTABLE") == 0)
265 return COLOR_GREEN;
266 if (strcmp(envvar, "TOG_COLOR_COMMIT") == 0)
267 return COLOR_GREEN;
268 if (strcmp(envvar, "TOG_COLOR_AUTHOR") == 0)
269 return COLOR_CYAN;
270 if (strcmp(envvar, "TOG_COLOR_DATE") == 0)
271 return COLOR_YELLOW;
272 if (strcmp(envvar, "TOG_COLOR_REFS_HEADS") == 0)
273 return COLOR_GREEN;
274 if (strcmp(envvar, "TOG_COLOR_REFS_TAGS") == 0)
275 return COLOR_MAGENTA;
276 if (strcmp(envvar, "TOG_COLOR_REFS_REMOTES") == 0)
277 return COLOR_YELLOW;
278 if (strcmp(envvar, "TOG_COLOR_REFS_BACKUP") == 0)
279 return COLOR_CYAN;
281 return -1;
284 static int
285 get_color_value(const char *envvar)
287 const char *val = getenv(envvar);
289 if (val == NULL)
290 return default_color_value(envvar);
292 if (strcasecmp(val, "black") == 0)
293 return COLOR_BLACK;
294 if (strcasecmp(val, "red") == 0)
295 return COLOR_RED;
296 if (strcasecmp(val, "green") == 0)
297 return COLOR_GREEN;
298 if (strcasecmp(val, "yellow") == 0)
299 return COLOR_YELLOW;
300 if (strcasecmp(val, "blue") == 0)
301 return COLOR_BLUE;
302 if (strcasecmp(val, "magenta") == 0)
303 return COLOR_MAGENTA;
304 if (strcasecmp(val, "cyan") == 0)
305 return COLOR_CYAN;
306 if (strcasecmp(val, "white") == 0)
307 return COLOR_WHITE;
308 if (strcasecmp(val, "default") == 0)
309 return -1;
311 return default_color_value(envvar);
315 struct tog_diff_view_state {
316 struct got_object_id *id1, *id2;
317 const char *label1, *label2;
318 FILE *f, *f1, *f2;
319 int first_displayed_line;
320 int last_displayed_line;
321 int eof;
322 int diff_context;
323 int ignore_whitespace;
324 int force_text_diff;
325 struct got_repository *repo;
326 struct tog_colors colors;
327 size_t nlines;
328 off_t *line_offsets;
329 int matched_line;
330 int selected_line;
332 /* passed from log view; may be NULL */
333 struct tog_view *log_view;
334 };
336 pthread_mutex_t tog_mutex = PTHREAD_MUTEX_INITIALIZER;
338 struct tog_log_thread_args {
339 pthread_cond_t need_commits;
340 pthread_cond_t commit_loaded;
341 int commits_needed;
342 int load_all;
343 struct got_commit_graph *graph;
344 struct commit_queue *commits;
345 const char *in_repo_path;
346 struct got_object_id *start_id;
347 struct got_repository *repo;
348 int *pack_fds;
349 int log_complete;
350 sig_atomic_t *quit;
351 struct commit_queue_entry **first_displayed_entry;
352 struct commit_queue_entry **selected_entry;
353 int *searching;
354 int *search_next_done;
355 regex_t *regex;
356 };
358 struct tog_log_view_state {
359 struct commit_queue commits;
360 struct commit_queue_entry *first_displayed_entry;
361 struct commit_queue_entry *last_displayed_entry;
362 struct commit_queue_entry *selected_entry;
363 int selected;
364 char *in_repo_path;
365 char *head_ref_name;
366 int log_branches;
367 struct got_repository *repo;
368 struct got_object_id *start_id;
369 sig_atomic_t quit;
370 pthread_t thread;
371 struct tog_log_thread_args thread_args;
372 struct commit_queue_entry *matched_entry;
373 struct commit_queue_entry *search_entry;
374 struct tog_colors colors;
375 };
377 #define TOG_COLOR_DIFF_MINUS 1
378 #define TOG_COLOR_DIFF_PLUS 2
379 #define TOG_COLOR_DIFF_CHUNK_HEADER 3
380 #define TOG_COLOR_DIFF_META 4
381 #define TOG_COLOR_TREE_SUBMODULE 5
382 #define TOG_COLOR_TREE_SYMLINK 6
383 #define TOG_COLOR_TREE_DIRECTORY 7
384 #define TOG_COLOR_TREE_EXECUTABLE 8
385 #define TOG_COLOR_COMMIT 9
386 #define TOG_COLOR_AUTHOR 10
387 #define TOG_COLOR_DATE 11
388 #define TOG_COLOR_REFS_HEADS 12
389 #define TOG_COLOR_REFS_TAGS 13
390 #define TOG_COLOR_REFS_REMOTES 14
391 #define TOG_COLOR_REFS_BACKUP 15
393 struct tog_blame_cb_args {
394 struct tog_blame_line *lines; /* one per line */
395 int nlines;
397 struct tog_view *view;
398 struct got_object_id *commit_id;
399 int *quit;
400 };
402 struct tog_blame_thread_args {
403 const char *path;
404 struct got_repository *repo;
405 struct tog_blame_cb_args *cb_args;
406 int *complete;
407 got_cancel_cb cancel_cb;
408 void *cancel_arg;
409 };
411 struct tog_blame {
412 FILE *f;
413 off_t filesize;
414 struct tog_blame_line *lines;
415 int nlines;
416 off_t *line_offsets;
417 pthread_t thread;
418 struct tog_blame_thread_args thread_args;
419 struct tog_blame_cb_args cb_args;
420 const char *path;
421 int *pack_fds;
422 };
424 struct tog_blame_view_state {
425 int first_displayed_line;
426 int last_displayed_line;
427 int selected_line;
428 int blame_complete;
429 int eof;
430 int done;
431 struct got_object_id_queue blamed_commits;
432 struct got_object_qid *blamed_commit;
433 char *path;
434 struct got_repository *repo;
435 struct got_object_id *commit_id;
436 struct tog_blame blame;
437 int matched_line;
438 struct tog_colors colors;
439 };
441 struct tog_parent_tree {
442 TAILQ_ENTRY(tog_parent_tree) entry;
443 struct got_tree_object *tree;
444 struct got_tree_entry *first_displayed_entry;
445 struct got_tree_entry *selected_entry;
446 int selected;
447 };
449 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
451 struct tog_tree_view_state {
452 char *tree_label;
453 struct got_object_id *commit_id;/* commit which this tree belongs to */
454 struct got_tree_object *root; /* the commit's root tree entry */
455 struct got_tree_object *tree; /* currently displayed (sub-)tree */
456 struct got_tree_entry *first_displayed_entry;
457 struct got_tree_entry *last_displayed_entry;
458 struct got_tree_entry *selected_entry;
459 int ndisplayed, selected, show_ids;
460 struct tog_parent_trees parents; /* parent trees of current sub-tree */
461 char *head_ref_name;
462 struct got_repository *repo;
463 struct got_tree_entry *matched_entry;
464 struct tog_colors colors;
465 };
467 struct tog_reflist_entry {
468 TAILQ_ENTRY(tog_reflist_entry) entry;
469 struct got_reference *ref;
470 int idx;
471 };
473 TAILQ_HEAD(tog_reflist_head, tog_reflist_entry);
475 struct tog_ref_view_state {
476 struct tog_reflist_head refs;
477 struct tog_reflist_entry *first_displayed_entry;
478 struct tog_reflist_entry *last_displayed_entry;
479 struct tog_reflist_entry *selected_entry;
480 int nrefs, ndisplayed, selected, show_date, show_ids, sort_by_date;
481 struct got_repository *repo;
482 struct tog_reflist_entry *matched_entry;
483 struct tog_colors colors;
484 };
486 /*
487 * We implement two types of views: parent views and child views.
489 * The 'Tab' key switches focus between a parent view and its child view.
490 * Child views are shown side-by-side to their parent view, provided
491 * there is enough screen estate.
493 * When a new view is opened from within a parent view, this new view
494 * becomes a child view of the parent view, replacing any existing child.
496 * When a new view is opened from within a child view, this new view
497 * becomes a parent view which will obscure the views below until the
498 * user quits the new parent view by typing 'q'.
500 * This list of views contains parent views only.
501 * Child views are only pointed to by their parent view.
502 */
503 TAILQ_HEAD(tog_view_list_head, tog_view);
505 struct tog_view {
506 TAILQ_ENTRY(tog_view) entry;
507 WINDOW *window;
508 PANEL *panel;
509 int nlines, ncols, begin_y, begin_x;
510 int maxx, x; /* max column and current start column */
511 int lines, cols; /* copies of LINES and COLS */
512 int focussed; /* Only set on one parent or child view at a time. */
513 int dying;
514 struct tog_view *parent;
515 struct tog_view *child;
517 /*
518 * This flag is initially set on parent views when a new child view
519 * is created. It gets toggled when the 'Tab' key switches focus
520 * between parent and child.
521 * The flag indicates whether focus should be passed on to our child
522 * view if this parent view gets picked for focus after another parent
523 * view was closed. This prevents child views from losing focus in such
524 * situations.
525 */
526 int focus_child;
528 /* type-specific state */
529 enum tog_view_type type;
530 union {
531 struct tog_diff_view_state diff;
532 struct tog_log_view_state log;
533 struct tog_blame_view_state blame;
534 struct tog_tree_view_state tree;
535 struct tog_ref_view_state ref;
536 } state;
538 const struct got_error *(*show)(struct tog_view *);
539 const struct got_error *(*input)(struct tog_view **,
540 struct tog_view *, int);
541 const struct got_error *(*close)(struct tog_view *);
543 const struct got_error *(*search_start)(struct tog_view *);
544 const struct got_error *(*search_next)(struct tog_view *);
545 int search_started;
546 int searching;
547 #define TOG_SEARCH_FORWARD 1
548 #define TOG_SEARCH_BACKWARD 2
549 int search_next_done;
550 #define TOG_SEARCH_HAVE_MORE 1
551 #define TOG_SEARCH_NO_MORE 2
552 #define TOG_SEARCH_HAVE_NONE 3
553 regex_t regex;
554 regmatch_t regmatch;
555 };
557 static const struct got_error *open_diff_view(struct tog_view *,
558 struct got_object_id *, struct got_object_id *,
559 const char *, const char *, int, int, int, struct tog_view *,
560 struct got_repository *);
561 static const struct got_error *show_diff_view(struct tog_view *);
562 static const struct got_error *input_diff_view(struct tog_view **,
563 struct tog_view *, int);
564 static const struct got_error* close_diff_view(struct tog_view *);
565 static const struct got_error *search_start_diff_view(struct tog_view *);
566 static const struct got_error *search_next_diff_view(struct tog_view *);
568 static const struct got_error *open_log_view(struct tog_view *,
569 struct got_object_id *, struct got_repository *,
570 const char *, const char *, int);
571 static const struct got_error * show_log_view(struct tog_view *);
572 static const struct got_error *input_log_view(struct tog_view **,
573 struct tog_view *, int);
574 static const struct got_error *close_log_view(struct tog_view *);
575 static const struct got_error *search_start_log_view(struct tog_view *);
576 static const struct got_error *search_next_log_view(struct tog_view *);
578 static const struct got_error *open_blame_view(struct tog_view *, char *,
579 struct got_object_id *, struct got_repository *);
580 static const struct got_error *show_blame_view(struct tog_view *);
581 static const struct got_error *input_blame_view(struct tog_view **,
582 struct tog_view *, int);
583 static const struct got_error *close_blame_view(struct tog_view *);
584 static const struct got_error *search_start_blame_view(struct tog_view *);
585 static const struct got_error *search_next_blame_view(struct tog_view *);
587 static const struct got_error *open_tree_view(struct tog_view *,
588 struct got_object_id *, const char *, struct got_repository *);
589 static const struct got_error *show_tree_view(struct tog_view *);
590 static const struct got_error *input_tree_view(struct tog_view **,
591 struct tog_view *, int);
592 static const struct got_error *close_tree_view(struct tog_view *);
593 static const struct got_error *search_start_tree_view(struct tog_view *);
594 static const struct got_error *search_next_tree_view(struct tog_view *);
596 static const struct got_error *open_ref_view(struct tog_view *,
597 struct got_repository *);
598 static const struct got_error *show_ref_view(struct tog_view *);
599 static const struct got_error *input_ref_view(struct tog_view **,
600 struct tog_view *, int);
601 static const struct got_error *close_ref_view(struct tog_view *);
602 static const struct got_error *search_start_ref_view(struct tog_view *);
603 static const struct got_error *search_next_ref_view(struct tog_view *);
605 static volatile sig_atomic_t tog_sigwinch_received;
606 static volatile sig_atomic_t tog_sigpipe_received;
607 static volatile sig_atomic_t tog_sigcont_received;
608 static volatile sig_atomic_t tog_sigint_received;
609 static volatile sig_atomic_t tog_sigterm_received;
611 static void
612 tog_sigwinch(int signo)
614 tog_sigwinch_received = 1;
617 static void
618 tog_sigpipe(int signo)
620 tog_sigpipe_received = 1;
623 static void
624 tog_sigcont(int signo)
626 tog_sigcont_received = 1;
629 static void
630 tog_sigint(int signo)
632 tog_sigint_received = 1;
635 static void
636 tog_sigterm(int signo)
638 tog_sigterm_received = 1;
641 static int
642 tog_fatal_signal_received(void)
644 return (tog_sigpipe_received ||
645 tog_sigint_received || tog_sigint_received);
649 static const struct got_error *
650 view_close(struct tog_view *view)
652 const struct got_error *err = NULL;
654 if (view->child) {
655 view_close(view->child);
656 view->child = NULL;
658 if (view->close)
659 err = view->close(view);
660 if (view->panel)
661 del_panel(view->panel);
662 if (view->window)
663 delwin(view->window);
664 free(view);
665 return err;
668 static struct tog_view *
669 view_open(int nlines, int ncols, int begin_y, int begin_x,
670 enum tog_view_type type)
672 struct tog_view *view = calloc(1, sizeof(*view));
674 if (view == NULL)
675 return NULL;
677 view->type = type;
678 view->lines = LINES;
679 view->cols = COLS;
680 view->nlines = nlines ? nlines : LINES - begin_y;
681 view->ncols = ncols ? ncols : COLS - begin_x;
682 view->begin_y = begin_y;
683 view->begin_x = begin_x;
684 view->window = newwin(nlines, ncols, begin_y, begin_x);
685 if (view->window == NULL) {
686 view_close(view);
687 return NULL;
689 view->panel = new_panel(view->window);
690 if (view->panel == NULL ||
691 set_panel_userptr(view->panel, view) != OK) {
692 view_close(view);
693 return NULL;
696 keypad(view->window, TRUE);
697 return view;
700 static int
701 view_split_begin_x(int begin_x)
703 if (begin_x > 0 || COLS < 120)
704 return 0;
705 return (COLS - MAX(COLS / 2, 80));
708 static const struct got_error *view_resize(struct tog_view *);
710 static const struct got_error *
711 view_splitscreen(struct tog_view *view)
713 const struct got_error *err = NULL;
715 view->begin_y = 0;
716 view->begin_x = view_split_begin_x(0);
717 view->nlines = LINES;
718 view->ncols = COLS - view->begin_x;
719 view->lines = LINES;
720 view->cols = COLS;
721 err = view_resize(view);
722 if (err)
723 return err;
725 if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
726 return got_error_from_errno("mvwin");
728 return NULL;
731 static const struct got_error *
732 view_fullscreen(struct tog_view *view)
734 const struct got_error *err = NULL;
736 view->begin_x = 0;
737 view->begin_y = 0;
738 view->nlines = LINES;
739 view->ncols = COLS;
740 view->lines = LINES;
741 view->cols = COLS;
742 err = view_resize(view);
743 if (err)
744 return err;
746 if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
747 return got_error_from_errno("mvwin");
749 return NULL;
752 static int
753 view_is_parent_view(struct tog_view *view)
755 return view->parent == NULL;
758 static const struct got_error *
759 view_resize(struct tog_view *view)
761 int nlines, ncols;
763 if (view->lines > LINES)
764 nlines = view->nlines - (view->lines - LINES);
765 else
766 nlines = view->nlines + (LINES - view->lines);
768 if (view->cols > COLS)
769 ncols = view->ncols - (view->cols - COLS);
770 else
771 ncols = view->ncols + (COLS - view->cols);
773 if (view->child) {
774 view->child->begin_x = view_split_begin_x(view->begin_x);
775 if (view->child->begin_x == 0) {
776 ncols = COLS;
778 view_fullscreen(view->child);
779 if (view->child->focussed)
780 show_panel(view->child->panel);
781 else
782 show_panel(view->panel);
783 } else {
784 ncols = view->child->begin_x;
786 view_splitscreen(view->child);
787 show_panel(view->child->panel);
789 } else if (view->parent == NULL)
790 ncols = COLS;
792 if (wresize(view->window, nlines, ncols) == ERR)
793 return got_error_from_errno("wresize");
794 if (replace_panel(view->panel, view->window) == ERR)
795 return got_error_from_errno("replace_panel");
796 wclear(view->window);
798 view->nlines = nlines;
799 view->ncols = ncols;
800 view->lines = LINES;
801 view->cols = COLS;
803 return NULL;
806 static const struct got_error *
807 view_close_child(struct tog_view *view)
809 const struct got_error *err = NULL;
811 if (view->child == NULL)
812 return NULL;
814 err = view_close(view->child);
815 view->child = NULL;
816 return err;
819 static const struct got_error *
820 view_set_child(struct tog_view *view, struct tog_view *child)
822 view->child = child;
823 child->parent = view;
825 return view_resize(view);
828 static int
829 view_is_splitscreen(struct tog_view *view)
831 return view->begin_x > 0;
834 static void
835 tog_resizeterm(void)
837 int cols, lines;
838 struct winsize size;
840 if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &size) < 0) {
841 cols = 80; /* Default */
842 lines = 24;
843 } else {
844 cols = size.ws_col;
845 lines = size.ws_row;
847 resize_term(lines, cols);
850 static const struct got_error *
851 view_search_start(struct tog_view *view)
853 const struct got_error *err = NULL;
854 char pattern[1024];
855 int ret;
857 if (view->search_started) {
858 regfree(&view->regex);
859 view->searching = 0;
860 memset(&view->regmatch, 0, sizeof(view->regmatch));
862 view->search_started = 0;
864 if (view->nlines < 1)
865 return NULL;
867 mvwaddstr(view->window, view->begin_y + view->nlines - 1, 0, "/");
868 wclrtoeol(view->window);
870 nocbreak();
871 echo();
872 ret = wgetnstr(view->window, pattern, sizeof(pattern));
873 cbreak();
874 noecho();
875 if (ret == ERR)
876 return NULL;
878 if (regcomp(&view->regex, pattern, REG_EXTENDED | REG_NEWLINE) == 0) {
879 err = view->search_start(view);
880 if (err) {
881 regfree(&view->regex);
882 return err;
884 view->search_started = 1;
885 view->searching = TOG_SEARCH_FORWARD;
886 view->search_next_done = 0;
887 view->search_next(view);
890 return NULL;
893 static const struct got_error *
894 view_input(struct tog_view **new, int *done, struct tog_view *view,
895 struct tog_view_list_head *views)
897 const struct got_error *err = NULL;
898 struct tog_view *v;
899 int ch, errcode;
901 *new = NULL;
903 /* Clear "no matches" indicator. */
904 if (view->search_next_done == TOG_SEARCH_NO_MORE ||
905 view->search_next_done == TOG_SEARCH_HAVE_NONE)
906 view->search_next_done = TOG_SEARCH_HAVE_MORE;
908 if (view->searching && !view->search_next_done) {
909 errcode = pthread_mutex_unlock(&tog_mutex);
910 if (errcode)
911 return got_error_set_errno(errcode,
912 "pthread_mutex_unlock");
913 sched_yield();
914 errcode = pthread_mutex_lock(&tog_mutex);
915 if (errcode)
916 return got_error_set_errno(errcode,
917 "pthread_mutex_lock");
918 view->search_next(view);
919 return NULL;
922 nodelay(stdscr, FALSE);
923 /* Allow threads to make progress while we are waiting for input. */
924 errcode = pthread_mutex_unlock(&tog_mutex);
925 if (errcode)
926 return got_error_set_errno(errcode, "pthread_mutex_unlock");
927 ch = wgetch(view->window);
928 errcode = pthread_mutex_lock(&tog_mutex);
929 if (errcode)
930 return got_error_set_errno(errcode, "pthread_mutex_lock");
931 nodelay(stdscr, TRUE);
933 if (tog_sigwinch_received || tog_sigcont_received) {
934 tog_resizeterm();
935 tog_sigwinch_received = 0;
936 tog_sigcont_received = 0;
937 TAILQ_FOREACH(v, views, entry) {
938 err = view_resize(v);
939 if (err)
940 return err;
941 err = v->input(new, v, KEY_RESIZE);
942 if (err)
943 return err;
944 if (v->child) {
945 err = view_resize(v->child);
946 if (err)
947 return err;
948 err = v->child->input(new, v->child,
949 KEY_RESIZE);
950 if (err)
951 return err;
956 switch (ch) {
957 case '\t':
958 if (view->child) {
959 view->focussed = 0;
960 view->child->focussed = 1;
961 view->focus_child = 1;
962 } else if (view->parent) {
963 view->focussed = 0;
964 view->parent->focussed = 1;
965 view->parent->focus_child = 0;
967 break;
968 case 'q':
969 err = view->input(new, view, ch);
970 view->dying = 1;
971 break;
972 case 'Q':
973 *done = 1;
974 break;
975 case 'f':
976 if (view_is_parent_view(view)) {
977 if (view->child == NULL)
978 break;
979 if (view_is_splitscreen(view->child)) {
980 view->focussed = 0;
981 view->child->focussed = 1;
982 err = view_fullscreen(view->child);
983 } else
984 err = view_splitscreen(view->child);
985 if (err)
986 break;
987 err = view->child->input(new, view->child,
988 KEY_RESIZE);
989 } else {
990 if (view_is_splitscreen(view)) {
991 view->parent->focussed = 0;
992 view->focussed = 1;
993 err = view_fullscreen(view);
994 } else {
995 err = view_splitscreen(view);
997 if (err)
998 break;
999 err = view->input(new, view, KEY_RESIZE);
1001 break;
1002 case KEY_RESIZE:
1003 break;
1004 case '/':
1005 if (view->search_start)
1006 view_search_start(view);
1007 else
1008 err = view->input(new, view, ch);
1009 break;
1010 case 'N':
1011 case 'n':
1012 if (view->search_started && view->search_next) {
1013 view->searching = (ch == 'n' ?
1014 TOG_SEARCH_FORWARD : TOG_SEARCH_BACKWARD);
1015 view->search_next_done = 0;
1016 view->search_next(view);
1017 } else
1018 err = view->input(new, view, ch);
1019 break;
1020 default:
1021 err = view->input(new, view, ch);
1022 break;
1025 return err;
1028 void
1029 view_vborder(struct tog_view *view)
1031 PANEL *panel;
1032 const struct tog_view *view_above;
1034 if (view->parent)
1035 return view_vborder(view->parent);
1037 panel = panel_above(view->panel);
1038 if (panel == NULL)
1039 return;
1041 view_above = panel_userptr(panel);
1042 mvwvline(view->window, view->begin_y, view_above->begin_x - 1,
1043 got_locale_is_utf8() ? ACS_VLINE : '|', view->nlines);
1046 int
1047 view_needs_focus_indication(struct tog_view *view)
1049 if (view_is_parent_view(view)) {
1050 if (view->child == NULL || view->child->focussed)
1051 return 0;
1052 if (!view_is_splitscreen(view->child))
1053 return 0;
1054 } else if (!view_is_splitscreen(view))
1055 return 0;
1057 return view->focussed;
1060 static const struct got_error *
1061 view_loop(struct tog_view *view)
1063 const struct got_error *err = NULL;
1064 struct tog_view_list_head views;
1065 struct tog_view *new_view;
1066 int fast_refresh = 10;
1067 int done = 0, errcode;
1069 errcode = pthread_mutex_lock(&tog_mutex);
1070 if (errcode)
1071 return got_error_set_errno(errcode, "pthread_mutex_lock");
1073 TAILQ_INIT(&views);
1074 TAILQ_INSERT_HEAD(&views, view, entry);
1076 view->focussed = 1;
1077 err = view->show(view);
1078 if (err)
1079 return err;
1080 update_panels();
1081 doupdate();
1082 while (!TAILQ_EMPTY(&views) && !done && !tog_fatal_signal_received()) {
1083 /* Refresh fast during initialization, then become slower. */
1084 if (fast_refresh && fast_refresh-- == 0)
1085 halfdelay(10); /* switch to once per second */
1087 err = view_input(&new_view, &done, view, &views);
1088 if (err)
1089 break;
1090 if (view->dying) {
1091 struct tog_view *v, *prev = NULL;
1093 if (view_is_parent_view(view))
1094 prev = TAILQ_PREV(view, tog_view_list_head,
1095 entry);
1096 else if (view->parent)
1097 prev = view->parent;
1099 if (view->parent) {
1100 view->parent->child = NULL;
1101 view->parent->focus_child = 0;
1103 err = view_resize(view->parent);
1104 if (err)
1105 break;
1106 } else
1107 TAILQ_REMOVE(&views, view, entry);
1109 err = view_close(view);
1110 if (err)
1111 goto done;
1113 view = NULL;
1114 TAILQ_FOREACH(v, &views, entry) {
1115 if (v->focussed)
1116 break;
1118 if (view == NULL && new_view == NULL) {
1119 /* No view has focus. Try to pick one. */
1120 if (prev)
1121 view = prev;
1122 else if (!TAILQ_EMPTY(&views)) {
1123 view = TAILQ_LAST(&views,
1124 tog_view_list_head);
1126 if (view) {
1127 if (view->focus_child) {
1128 view->child->focussed = 1;
1129 view = view->child;
1130 } else
1131 view->focussed = 1;
1135 if (new_view) {
1136 struct tog_view *v, *t;
1137 /* Only allow one parent view per type. */
1138 TAILQ_FOREACH_SAFE(v, &views, entry, t) {
1139 if (v->type != new_view->type)
1140 continue;
1141 TAILQ_REMOVE(&views, v, entry);
1142 err = view_close(v);
1143 if (err)
1144 goto done;
1145 break;
1147 TAILQ_INSERT_TAIL(&views, new_view, entry);
1148 view = new_view;
1150 if (view) {
1151 if (view_is_parent_view(view)) {
1152 if (view->child && view->child->focussed)
1153 view = view->child;
1154 } else {
1155 if (view->parent && view->parent->focussed)
1156 view = view->parent;
1158 show_panel(view->panel);
1159 if (view->child && view_is_splitscreen(view->child))
1160 show_panel(view->child->panel);
1161 if (view->parent && view_is_splitscreen(view)) {
1162 err = view->parent->show(view->parent);
1163 if (err)
1164 goto done;
1166 err = view->show(view);
1167 if (err)
1168 goto done;
1169 if (view->child) {
1170 err = view->child->show(view->child);
1171 if (err)
1172 goto done;
1174 update_panels();
1175 doupdate();
1178 done:
1179 while (!TAILQ_EMPTY(&views)) {
1180 view = TAILQ_FIRST(&views);
1181 TAILQ_REMOVE(&views, view, entry);
1182 view_close(view);
1185 errcode = pthread_mutex_unlock(&tog_mutex);
1186 if (errcode && err == NULL)
1187 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
1189 return err;
1192 __dead static void
1193 usage_log(void)
1195 endwin();
1196 fprintf(stderr,
1197 "usage: %s log [-b] [-c commit] [-r repository-path] [path]\n",
1198 getprogname());
1199 exit(1);
1202 /* Create newly allocated wide-character string equivalent to a byte string. */
1203 static const struct got_error *
1204 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
1206 char *vis = NULL;
1207 const struct got_error *err = NULL;
1209 *ws = NULL;
1210 *wlen = mbstowcs(NULL, s, 0);
1211 if (*wlen == (size_t)-1) {
1212 int vislen;
1213 if (errno != EILSEQ)
1214 return got_error_from_errno("mbstowcs");
1216 /* byte string invalid in current encoding; try to "fix" it */
1217 err = got_mbsavis(&vis, &vislen, s);
1218 if (err)
1219 return err;
1220 *wlen = mbstowcs(NULL, vis, 0);
1221 if (*wlen == (size_t)-1) {
1222 err = got_error_from_errno("mbstowcs"); /* give up */
1223 goto done;
1227 *ws = calloc(*wlen + 1, sizeof(**ws));
1228 if (*ws == NULL) {
1229 err = got_error_from_errno("calloc");
1230 goto done;
1233 if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
1234 err = got_error_from_errno("mbstowcs");
1235 done:
1236 free(vis);
1237 if (err) {
1238 free(*ws);
1239 *ws = NULL;
1240 *wlen = 0;
1242 return err;
1245 static const struct got_error *
1246 expand_tab(char **ptr, const char *src)
1248 char *dst;
1249 size_t len, n, idx = 0, sz = 0;
1251 *ptr = NULL;
1252 n = len = strlen(src);
1253 dst = malloc(n + 1);
1254 if (dst == NULL)
1255 return got_error_from_errno("malloc");
1257 while (idx < len && src[idx]) {
1258 const char c = src[idx];
1260 if (c == '\t') {
1261 size_t nb = TABSIZE - sz % TABSIZE;
1262 char *p;
1264 p = realloc(dst, n + nb);
1265 if (p == NULL) {
1266 free(dst);
1267 return got_error_from_errno("realloc");
1270 dst = p;
1271 n += nb;
1272 memset(dst + sz, ' ', nb);
1273 sz += nb;
1274 } else
1275 dst[sz++] = src[idx];
1276 ++idx;
1279 dst[sz] = '\0';
1280 *ptr = dst;
1281 return NULL;
1285 * Advance at most n columns from wline starting at offset off.
1286 * Return the index to the first character after the span operation.
1287 * Return the combined column width of all spanned wide character in
1288 * *rcol.
1290 static int
1291 span_wline(int *rcol, int off, wchar_t *wline, int n, int col_tab_align)
1293 int width, i, cols = 0;
1295 if (n == 0) {
1296 *rcol = cols;
1297 return off;
1300 for (i = off; wline[i] != L'\0'; ++i) {
1301 if (wline[i] == L'\t')
1302 width = TABSIZE - ((cols + col_tab_align) % TABSIZE);
1303 else
1304 width = wcwidth(wline[i]);
1306 if (width == -1) {
1307 width = 1;
1308 wline[i] = L'.';
1311 if (cols + width > n)
1312 break;
1313 cols += width;
1316 *rcol = cols;
1317 return i;
1321 * Format a line for display, ensuring that it won't overflow a width limit.
1322 * With scrolling, the width returned refers to the scrolled version of the
1323 * line, which starts at (*wlinep)[*scrollxp]. The caller must free *wlinep.
1325 static const struct got_error *
1326 format_line(wchar_t **wlinep, int *widthp, int *scrollxp,
1327 const char *line, int nscroll, int wlimit, int col_tab_align, int expand)
1329 const struct got_error *err = NULL;
1330 int cols;
1331 wchar_t *wline = NULL;
1332 char *exstr = NULL;
1333 size_t wlen;
1334 int i, scrollx;
1336 *wlinep = NULL;
1337 *widthp = 0;
1339 if (expand) {
1340 err = expand_tab(&exstr, line);
1341 if (err)
1342 return err;
1345 err = mbs2ws(&wline, &wlen, expand ? exstr : line);
1346 free(exstr);
1347 if (err)
1348 return err;
1350 scrollx = span_wline(&cols, 0, wline, nscroll, col_tab_align);
1352 if (wlen > 0 && wline[wlen - 1] == L'\n') {
1353 wline[wlen - 1] = L'\0';
1354 wlen--;
1356 if (wlen > 0 && wline[wlen - 1] == L'\r') {
1357 wline[wlen - 1] = L'\0';
1358 wlen--;
1361 i = span_wline(&cols, scrollx, wline, wlimit, col_tab_align);
1362 wline[i] = L'\0';
1364 if (widthp)
1365 *widthp = cols;
1366 if (scrollxp)
1367 *scrollxp = scrollx;
1368 if (err)
1369 free(wline);
1370 else
1371 *wlinep = wline;
1372 return err;
1375 static const struct got_error*
1376 build_refs_str(char **refs_str, struct got_reflist_head *refs,
1377 struct got_object_id *id, struct got_repository *repo)
1379 static const struct got_error *err = NULL;
1380 struct got_reflist_entry *re;
1381 char *s;
1382 const char *name;
1384 *refs_str = NULL;
1386 TAILQ_FOREACH(re, refs, entry) {
1387 struct got_tag_object *tag = NULL;
1388 struct got_object_id *ref_id;
1389 int cmp;
1391 name = got_ref_get_name(re->ref);
1392 if (strcmp(name, GOT_REF_HEAD) == 0)
1393 continue;
1394 if (strncmp(name, "refs/", 5) == 0)
1395 name += 5;
1396 if (strncmp(name, "got/", 4) == 0 &&
1397 strncmp(name, "got/backup/", 11) != 0)
1398 continue;
1399 if (strncmp(name, "heads/", 6) == 0)
1400 name += 6;
1401 if (strncmp(name, "remotes/", 8) == 0) {
1402 name += 8;
1403 s = strstr(name, "/" GOT_REF_HEAD);
1404 if (s != NULL && s[strlen(s)] == '\0')
1405 continue;
1407 err = got_ref_resolve(&ref_id, repo, re->ref);
1408 if (err)
1409 break;
1410 if (strncmp(name, "tags/", 5) == 0) {
1411 err = got_object_open_as_tag(&tag, repo, ref_id);
1412 if (err) {
1413 if (err->code != GOT_ERR_OBJ_TYPE) {
1414 free(ref_id);
1415 break;
1417 /* Ref points at something other than a tag. */
1418 err = NULL;
1419 tag = NULL;
1422 cmp = got_object_id_cmp(tag ?
1423 got_object_tag_get_object_id(tag) : ref_id, id);
1424 free(ref_id);
1425 if (tag)
1426 got_object_tag_close(tag);
1427 if (cmp != 0)
1428 continue;
1429 s = *refs_str;
1430 if (asprintf(refs_str, "%s%s%s", s ? s : "",
1431 s ? ", " : "", name) == -1) {
1432 err = got_error_from_errno("asprintf");
1433 free(s);
1434 *refs_str = NULL;
1435 break;
1437 free(s);
1440 return err;
1443 static const struct got_error *
1444 format_author(wchar_t **wauthor, int *author_width, char *author, int limit,
1445 int col_tab_align)
1447 char *smallerthan;
1449 smallerthan = strchr(author, '<');
1450 if (smallerthan && smallerthan[1] != '\0')
1451 author = smallerthan + 1;
1452 author[strcspn(author, "@>")] = '\0';
1453 return format_line(wauthor, author_width, NULL, author, 0, limit,
1454 col_tab_align, 0);
1457 static const struct got_error *
1458 draw_commit(struct tog_view *view, struct got_commit_object *commit,
1459 struct got_object_id *id, const size_t date_display_cols,
1460 int author_display_cols)
1462 struct tog_log_view_state *s = &view->state.log;
1463 const struct got_error *err = NULL;
1464 char datebuf[12]; /* YYYY-MM-DD + SPACE + NUL */
1465 char *logmsg0 = NULL, *logmsg = NULL;
1466 char *author = NULL;
1467 wchar_t *wlogmsg = NULL, *wauthor = NULL;
1468 int author_width, logmsg_width;
1469 char *newline, *line = NULL;
1470 int col, limit, scrollx;
1471 const int avail = view->ncols;
1472 struct tm tm;
1473 time_t committer_time;
1474 struct tog_color *tc;
1476 committer_time = got_object_commit_get_committer_time(commit);
1477 if (gmtime_r(&committer_time, &tm) == NULL)
1478 return got_error_from_errno("gmtime_r");
1479 if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d ", &tm) == 0)
1480 return got_error(GOT_ERR_NO_SPACE);
1482 if (avail <= date_display_cols)
1483 limit = MIN(sizeof(datebuf) - 1, avail);
1484 else
1485 limit = MIN(date_display_cols, sizeof(datebuf) - 1);
1486 tc = get_color(&s->colors, TOG_COLOR_DATE);
1487 if (tc)
1488 wattr_on(view->window,
1489 COLOR_PAIR(tc->colorpair), NULL);
1490 waddnstr(view->window, datebuf, limit);
1491 if (tc)
1492 wattr_off(view->window,
1493 COLOR_PAIR(tc->colorpair), NULL);
1494 col = limit;
1495 if (col > avail)
1496 goto done;
1498 if (avail >= 120) {
1499 char *id_str;
1500 err = got_object_id_str(&id_str, id);
1501 if (err)
1502 goto done;
1503 tc = get_color(&s->colors, TOG_COLOR_COMMIT);
1504 if (tc)
1505 wattr_on(view->window,
1506 COLOR_PAIR(tc->colorpair), NULL);
1507 wprintw(view->window, "%.8s ", id_str);
1508 if (tc)
1509 wattr_off(view->window,
1510 COLOR_PAIR(tc->colorpair), NULL);
1511 free(id_str);
1512 col += 9;
1513 if (col > avail)
1514 goto done;
1517 author = strdup(got_object_commit_get_author(commit));
1518 if (author == NULL) {
1519 err = got_error_from_errno("strdup");
1520 goto done;
1522 err = format_author(&wauthor, &author_width, author, avail - col, col);
1523 if (err)
1524 goto done;
1525 tc = get_color(&s->colors, TOG_COLOR_AUTHOR);
1526 if (tc)
1527 wattr_on(view->window,
1528 COLOR_PAIR(tc->colorpair), NULL);
1529 waddwstr(view->window, wauthor);
1530 if (tc)
1531 wattr_off(view->window,
1532 COLOR_PAIR(tc->colorpair), NULL);
1533 col += author_width;
1534 while (col < avail && author_width < author_display_cols + 2) {
1535 waddch(view->window, ' ');
1536 col++;
1537 author_width++;
1539 if (col > avail)
1540 goto done;
1542 err = got_object_commit_get_logmsg(&logmsg0, commit);
1543 if (err)
1544 goto done;
1545 logmsg = logmsg0;
1546 while (*logmsg == '\n')
1547 logmsg++;
1548 newline = strchr(logmsg, '\n');
1549 if (newline)
1550 *newline = '\0';
1551 limit = avail - col;
1552 if (view->child && limit > 0)
1553 limit--; /* for the border */
1554 err = format_line(&wlogmsg, &logmsg_width, &scrollx, logmsg, view->x,
1555 limit, col, 1);
1556 if (err)
1557 goto done;
1558 waddwstr(view->window, &wlogmsg[scrollx]);
1559 col += MAX(logmsg_width, 0);
1560 while (col < avail) {
1561 waddch(view->window, ' ');
1562 col++;
1564 done:
1565 free(logmsg0);
1566 free(wlogmsg);
1567 free(author);
1568 free(wauthor);
1569 free(line);
1570 return err;
1573 static struct commit_queue_entry *
1574 alloc_commit_queue_entry(struct got_commit_object *commit,
1575 struct got_object_id *id)
1577 struct commit_queue_entry *entry;
1579 entry = calloc(1, sizeof(*entry));
1580 if (entry == NULL)
1581 return NULL;
1583 entry->id = id;
1584 entry->commit = commit;
1585 return entry;
1588 static void
1589 pop_commit(struct commit_queue *commits)
1591 struct commit_queue_entry *entry;
1593 entry = TAILQ_FIRST(&commits->head);
1594 TAILQ_REMOVE(&commits->head, entry, entry);
1595 got_object_commit_close(entry->commit);
1596 commits->ncommits--;
1597 /* Don't free entry->id! It is owned by the commit graph. */
1598 free(entry);
1601 static void
1602 free_commits(struct commit_queue *commits)
1604 while (!TAILQ_EMPTY(&commits->head))
1605 pop_commit(commits);
1608 static const struct got_error *
1609 match_commit(int *have_match, struct got_object_id *id,
1610 struct got_commit_object *commit, regex_t *regex)
1612 const struct got_error *err = NULL;
1613 regmatch_t regmatch;
1614 char *id_str = NULL, *logmsg = NULL;
1616 *have_match = 0;
1618 err = got_object_id_str(&id_str, id);
1619 if (err)
1620 return err;
1622 err = got_object_commit_get_logmsg(&logmsg, commit);
1623 if (err)
1624 goto done;
1626 if (regexec(regex, got_object_commit_get_author(commit), 1,
1627 &regmatch, 0) == 0 ||
1628 regexec(regex, got_object_commit_get_committer(commit), 1,
1629 &regmatch, 0) == 0 ||
1630 regexec(regex, id_str, 1, &regmatch, 0) == 0 ||
1631 regexec(regex, logmsg, 1, &regmatch, 0) == 0)
1632 *have_match = 1;
1633 done:
1634 free(id_str);
1635 free(logmsg);
1636 return err;
1639 static const struct got_error *
1640 queue_commits(struct tog_log_thread_args *a)
1642 const struct got_error *err = NULL;
1645 * We keep all commits open throughout the lifetime of the log
1646 * view in order to avoid having to re-fetch commits from disk
1647 * while updating the display.
1649 do {
1650 struct got_object_id *id;
1651 struct got_commit_object *commit;
1652 struct commit_queue_entry *entry;
1653 int errcode;
1655 err = got_commit_graph_iter_next(&id, a->graph, a->repo,
1656 NULL, NULL);
1657 if (err || id == NULL)
1658 break;
1660 err = got_object_open_as_commit(&commit, a->repo, id);
1661 if (err)
1662 break;
1663 entry = alloc_commit_queue_entry(commit, id);
1664 if (entry == NULL) {
1665 err = got_error_from_errno("alloc_commit_queue_entry");
1666 break;
1669 errcode = pthread_mutex_lock(&tog_mutex);
1670 if (errcode) {
1671 err = got_error_set_errno(errcode,
1672 "pthread_mutex_lock");
1673 break;
1676 entry->idx = a->commits->ncommits;
1677 TAILQ_INSERT_TAIL(&a->commits->head, entry, entry);
1678 a->commits->ncommits++;
1680 if (*a->searching == TOG_SEARCH_FORWARD &&
1681 !*a->search_next_done) {
1682 int have_match;
1683 err = match_commit(&have_match, id, commit, a->regex);
1684 if (err)
1685 break;
1686 if (have_match)
1687 *a->search_next_done = TOG_SEARCH_HAVE_MORE;
1690 errcode = pthread_mutex_unlock(&tog_mutex);
1691 if (errcode && err == NULL)
1692 err = got_error_set_errno(errcode,
1693 "pthread_mutex_unlock");
1694 if (err)
1695 break;
1696 } while (*a->searching == TOG_SEARCH_FORWARD && !*a->search_next_done);
1698 return err;
1701 static void
1702 select_commit(struct tog_log_view_state *s)
1704 struct commit_queue_entry *entry;
1705 int ncommits = 0;
1707 entry = s->first_displayed_entry;
1708 while (entry) {
1709 if (ncommits == s->selected) {
1710 s->selected_entry = entry;
1711 break;
1713 entry = TAILQ_NEXT(entry, entry);
1714 ncommits++;
1718 static const struct got_error *
1719 draw_commits(struct tog_view *view)
1721 const struct got_error *err = NULL;
1722 struct tog_log_view_state *s = &view->state.log;
1723 struct commit_queue_entry *entry = s->selected_entry;
1724 const int limit = view->nlines;
1725 int width;
1726 int ncommits, author_cols = 4;
1727 char *id_str = NULL, *header = NULL, *ncommits_str = NULL;
1728 char *refs_str = NULL;
1729 wchar_t *wline;
1730 struct tog_color *tc;
1731 static const size_t date_display_cols = 12;
1733 if (s->selected_entry &&
1734 !(view->searching && view->search_next_done == 0)) {
1735 struct got_reflist_head *refs;
1736 err = got_object_id_str(&id_str, s->selected_entry->id);
1737 if (err)
1738 return err;
1739 refs = got_reflist_object_id_map_lookup(tog_refs_idmap,
1740 s->selected_entry->id);
1741 if (refs) {
1742 err = build_refs_str(&refs_str, refs,
1743 s->selected_entry->id, s->repo);
1744 if (err)
1745 goto done;
1749 if (s->thread_args.commits_needed == 0)
1750 halfdelay(10); /* disable fast refresh */
1752 if (s->thread_args.commits_needed > 0 || s->thread_args.load_all) {
1753 if (asprintf(&ncommits_str, " [%d/%d] %s",
1754 entry ? entry->idx + 1 : 0, s->commits.ncommits,
1755 (view->searching && !view->search_next_done) ?
1756 "searching..." : "loading...") == -1) {
1757 err = got_error_from_errno("asprintf");
1758 goto done;
1760 } else {
1761 const char *search_str = NULL;
1763 if (view->searching) {
1764 if (view->search_next_done == TOG_SEARCH_NO_MORE)
1765 search_str = "no more matches";
1766 else if (view->search_next_done == TOG_SEARCH_HAVE_NONE)
1767 search_str = "no matches found";
1768 else if (!view->search_next_done)
1769 search_str = "searching...";
1772 if (asprintf(&ncommits_str, " [%d/%d] %s",
1773 entry ? entry->idx + 1 : 0, s->commits.ncommits,
1774 search_str ? search_str :
1775 (refs_str ? refs_str : "")) == -1) {
1776 err = got_error_from_errno("asprintf");
1777 goto done;
1781 if (s->in_repo_path && strcmp(s->in_repo_path, "/") != 0) {
1782 if (asprintf(&header, "commit %s %s%s", id_str ? id_str :
1783 "........................................",
1784 s->in_repo_path, ncommits_str) == -1) {
1785 err = got_error_from_errno("asprintf");
1786 header = NULL;
1787 goto done;
1789 } else if (asprintf(&header, "commit %s%s",
1790 id_str ? id_str : "........................................",
1791 ncommits_str) == -1) {
1792 err = got_error_from_errno("asprintf");
1793 header = NULL;
1794 goto done;
1796 err = format_line(&wline, &width, NULL, header, 0, view->ncols, 0, 0);
1797 if (err)
1798 goto done;
1800 werase(view->window);
1802 if (view_needs_focus_indication(view))
1803 wstandout(view->window);
1804 tc = get_color(&s->colors, TOG_COLOR_COMMIT);
1805 if (tc)
1806 wattr_on(view->window,
1807 COLOR_PAIR(tc->colorpair), NULL);
1808 waddwstr(view->window, wline);
1809 if (tc)
1810 wattr_off(view->window,
1811 COLOR_PAIR(tc->colorpair), NULL);
1812 while (width < view->ncols) {
1813 waddch(view->window, ' ');
1814 width++;
1816 if (view_needs_focus_indication(view))
1817 wstandend(view->window);
1818 free(wline);
1819 if (limit <= 1)
1820 goto done;
1822 /* Grow author column size if necessary, and set view->maxx. */
1823 entry = s->first_displayed_entry;
1824 ncommits = 0;
1825 view->maxx = 0;
1826 while (entry) {
1827 char *author, *eol, *msg, *msg0;
1828 wchar_t *wauthor, *wmsg;
1829 int width;
1830 if (ncommits >= limit - 1)
1831 break;
1832 author = strdup(got_object_commit_get_author(entry->commit));
1833 if (author == NULL) {
1834 err = got_error_from_errno("strdup");
1835 goto done;
1837 err = format_author(&wauthor, &width, author, COLS,
1838 date_display_cols);
1839 if (author_cols < width)
1840 author_cols = width;
1841 free(wauthor);
1842 free(author);
1843 err = got_object_commit_get_logmsg(&msg0, entry->commit);
1844 if (err)
1845 goto done;
1846 msg = msg0;
1847 while (*msg == '\n')
1848 ++msg;
1849 if ((eol = strchr(msg, '\n')))
1850 *eol = '\0';
1851 err = format_line(&wmsg, &width, NULL, msg, 0, INT_MAX,
1852 date_display_cols + author_cols, 0);
1853 if (err)
1854 goto done;
1855 view->maxx = MAX(view->maxx, width);
1856 free(msg0);
1857 free(wmsg);
1858 ncommits++;
1859 entry = TAILQ_NEXT(entry, entry);
1862 entry = s->first_displayed_entry;
1863 s->last_displayed_entry = s->first_displayed_entry;
1864 ncommits = 0;
1865 while (entry) {
1866 if (ncommits >= limit - 1)
1867 break;
1868 if (ncommits == s->selected)
1869 wstandout(view->window);
1870 err = draw_commit(view, entry->commit, entry->id,
1871 date_display_cols, author_cols);
1872 if (ncommits == s->selected)
1873 wstandend(view->window);
1874 if (err)
1875 goto done;
1876 ncommits++;
1877 s->last_displayed_entry = entry;
1878 entry = TAILQ_NEXT(entry, entry);
1881 view_vborder(view);
1882 update_panels();
1883 doupdate();
1884 done:
1885 free(id_str);
1886 free(refs_str);
1887 free(ncommits_str);
1888 free(header);
1889 return err;
1892 static void
1893 log_scroll_up(struct tog_log_view_state *s, int maxscroll)
1895 struct commit_queue_entry *entry;
1896 int nscrolled = 0;
1898 entry = TAILQ_FIRST(&s->commits.head);
1899 if (s->first_displayed_entry == entry)
1900 return;
1902 entry = s->first_displayed_entry;
1903 while (entry && nscrolled < maxscroll) {
1904 entry = TAILQ_PREV(entry, commit_queue_head, entry);
1905 if (entry) {
1906 s->first_displayed_entry = entry;
1907 nscrolled++;
1912 static const struct got_error *
1913 trigger_log_thread(struct tog_view *view, int wait)
1915 struct tog_log_thread_args *ta = &view->state.log.thread_args;
1916 int errcode;
1918 halfdelay(1); /* fast refresh while loading commits */
1920 while (ta->commits_needed > 0 || ta->load_all) {
1921 if (ta->log_complete)
1922 break;
1924 /* Wake the log thread. */
1925 errcode = pthread_cond_signal(&ta->need_commits);
1926 if (errcode)
1927 return got_error_set_errno(errcode,
1928 "pthread_cond_signal");
1931 * The mutex will be released while the view loop waits
1932 * in wgetch(), at which time the log thread will run.
1934 if (!wait)
1935 break;
1937 /* Display progress update in log view. */
1938 show_log_view(view);
1939 update_panels();
1940 doupdate();
1942 /* Wait right here while next commit is being loaded. */
1943 errcode = pthread_cond_wait(&ta->commit_loaded, &tog_mutex);
1944 if (errcode)
1945 return got_error_set_errno(errcode,
1946 "pthread_cond_wait");
1948 /* Display progress update in log view. */
1949 show_log_view(view);
1950 update_panels();
1951 doupdate();
1954 return NULL;
1957 static const struct got_error *
1958 log_scroll_down(struct tog_view *view, int maxscroll)
1960 struct tog_log_view_state *s = &view->state.log;
1961 const struct got_error *err = NULL;
1962 struct commit_queue_entry *pentry;
1963 int nscrolled = 0, ncommits_needed;
1965 if (s->last_displayed_entry == NULL)
1966 return NULL;
1968 ncommits_needed = s->last_displayed_entry->idx + 1 + maxscroll;
1969 if (s->commits.ncommits < ncommits_needed &&
1970 !s->thread_args.log_complete) {
1972 * Ask the log thread for required amount of commits.
1974 s->thread_args.commits_needed += maxscroll;
1975 err = trigger_log_thread(view, 1);
1976 if (err)
1977 return err;
1980 do {
1981 pentry = TAILQ_NEXT(s->last_displayed_entry, entry);
1982 if (pentry == NULL)
1983 break;
1985 s->last_displayed_entry = pentry;
1987 pentry = TAILQ_NEXT(s->first_displayed_entry, entry);
1988 if (pentry == NULL)
1989 break;
1990 s->first_displayed_entry = pentry;
1991 } while (++nscrolled < maxscroll);
1993 return err;
1996 static const struct got_error *
1997 open_diff_view_for_commit(struct tog_view **new_view, int begin_x,
1998 struct got_commit_object *commit, struct got_object_id *commit_id,
1999 struct tog_view *log_view, struct got_repository *repo)
2001 const struct got_error *err;
2002 struct got_object_qid *parent_id;
2003 struct tog_view *diff_view;
2005 diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
2006 if (diff_view == NULL)
2007 return got_error_from_errno("view_open");
2009 parent_id = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
2010 err = open_diff_view(diff_view, parent_id ? &parent_id->id : NULL,
2011 commit_id, NULL, NULL, 3, 0, 0, log_view, repo);
2012 if (err == NULL)
2013 *new_view = diff_view;
2014 return err;
2017 static const struct got_error *
2018 tree_view_visit_subtree(struct tog_tree_view_state *s,
2019 struct got_tree_object *subtree)
2021 struct tog_parent_tree *parent;
2023 parent = calloc(1, sizeof(*parent));
2024 if (parent == NULL)
2025 return got_error_from_errno("calloc");
2027 parent->tree = s->tree;
2028 parent->first_displayed_entry = s->first_displayed_entry;
2029 parent->selected_entry = s->selected_entry;
2030 parent->selected = s->selected;
2031 TAILQ_INSERT_HEAD(&s->parents, parent, entry);
2032 s->tree = subtree;
2033 s->selected = 0;
2034 s->first_displayed_entry = NULL;
2035 return NULL;
2038 static const struct got_error *
2039 tree_view_walk_path(struct tog_tree_view_state *s,
2040 struct got_commit_object *commit, const char *path)
2042 const struct got_error *err = NULL;
2043 struct got_tree_object *tree = NULL;
2044 const char *p;
2045 char *slash, *subpath = NULL;
2047 /* Walk the path and open corresponding tree objects. */
2048 p = path;
2049 while (*p) {
2050 struct got_tree_entry *te;
2051 struct got_object_id *tree_id;
2052 char *te_name;
2054 while (p[0] == '/')
2055 p++;
2057 /* Ensure the correct subtree entry is selected. */
2058 slash = strchr(p, '/');
2059 if (slash == NULL)
2060 te_name = strdup(p);
2061 else
2062 te_name = strndup(p, slash - p);
2063 if (te_name == NULL) {
2064 err = got_error_from_errno("strndup");
2065 break;
2067 te = got_object_tree_find_entry(s->tree, te_name);
2068 if (te == NULL) {
2069 err = got_error_path(te_name, GOT_ERR_NO_TREE_ENTRY);
2070 free(te_name);
2071 break;
2073 free(te_name);
2074 s->first_displayed_entry = s->selected_entry = te;
2076 if (!S_ISDIR(got_tree_entry_get_mode(s->selected_entry)))
2077 break; /* jump to this file's entry */
2079 slash = strchr(p, '/');
2080 if (slash)
2081 subpath = strndup(path, slash - path);
2082 else
2083 subpath = strdup(path);
2084 if (subpath == NULL) {
2085 err = got_error_from_errno("strdup");
2086 break;
2089 err = got_object_id_by_path(&tree_id, s->repo, commit,
2090 subpath);
2091 if (err)
2092 break;
2094 err = got_object_open_as_tree(&tree, s->repo, tree_id);
2095 free(tree_id);
2096 if (err)
2097 break;
2099 err = tree_view_visit_subtree(s, tree);
2100 if (err) {
2101 got_object_tree_close(tree);
2102 break;
2104 if (slash == NULL)
2105 break;
2106 free(subpath);
2107 subpath = NULL;
2108 p = slash;
2111 free(subpath);
2112 return err;
2115 static const struct got_error *
2116 browse_commit_tree(struct tog_view **new_view, int begin_x,
2117 struct commit_queue_entry *entry, const char *path,
2118 const char *head_ref_name, struct got_repository *repo)
2120 const struct got_error *err = NULL;
2121 struct tog_tree_view_state *s;
2122 struct tog_view *tree_view;
2124 tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
2125 if (tree_view == NULL)
2126 return got_error_from_errno("view_open");
2128 err = open_tree_view(tree_view, entry->id, head_ref_name, repo);
2129 if (err)
2130 return err;
2131 s = &tree_view->state.tree;
2133 *new_view = tree_view;
2135 if (got_path_is_root_dir(path))
2136 return NULL;
2138 return tree_view_walk_path(s, entry->commit, path);
2141 static const struct got_error *
2142 block_signals_used_by_main_thread(void)
2144 sigset_t sigset;
2145 int errcode;
2147 if (sigemptyset(&sigset) == -1)
2148 return got_error_from_errno("sigemptyset");
2150 /* tog handles SIGWINCH, SIGCONT, SIGINT, SIGTERM */
2151 if (sigaddset(&sigset, SIGWINCH) == -1)
2152 return got_error_from_errno("sigaddset");
2153 if (sigaddset(&sigset, SIGCONT) == -1)
2154 return got_error_from_errno("sigaddset");
2155 if (sigaddset(&sigset, SIGINT) == -1)
2156 return got_error_from_errno("sigaddset");
2157 if (sigaddset(&sigset, SIGTERM) == -1)
2158 return got_error_from_errno("sigaddset");
2160 /* ncurses handles SIGTSTP */
2161 if (sigaddset(&sigset, SIGTSTP) == -1)
2162 return got_error_from_errno("sigaddset");
2164 errcode = pthread_sigmask(SIG_BLOCK, &sigset, NULL);
2165 if (errcode)
2166 return got_error_set_errno(errcode, "pthread_sigmask");
2168 return NULL;
2171 static void *
2172 log_thread(void *arg)
2174 const struct got_error *err = NULL;
2175 int errcode = 0;
2176 struct tog_log_thread_args *a = arg;
2177 int done = 0;
2179 err = block_signals_used_by_main_thread();
2180 if (err)
2181 return (void *)err;
2183 while (!done && !err && !tog_fatal_signal_received()) {
2184 err = queue_commits(a);
2185 if (err) {
2186 if (err->code != GOT_ERR_ITER_COMPLETED)
2187 return (void *)err;
2188 err = NULL;
2189 done = 1;
2190 } else if (a->commits_needed > 0 && !a->load_all)
2191 a->commits_needed--;
2193 errcode = pthread_mutex_lock(&tog_mutex);
2194 if (errcode) {
2195 err = got_error_set_errno(errcode,
2196 "pthread_mutex_lock");
2197 break;
2198 } else if (*a->quit)
2199 done = 1;
2200 else if (*a->first_displayed_entry == NULL) {
2201 *a->first_displayed_entry =
2202 TAILQ_FIRST(&a->commits->head);
2203 *a->selected_entry = *a->first_displayed_entry;
2206 errcode = pthread_cond_signal(&a->commit_loaded);
2207 if (errcode) {
2208 err = got_error_set_errno(errcode,
2209 "pthread_cond_signal");
2210 pthread_mutex_unlock(&tog_mutex);
2211 break;
2214 if (done)
2215 a->commits_needed = 0;
2216 else {
2217 if (a->commits_needed == 0 && !a->load_all) {
2218 errcode = pthread_cond_wait(&a->need_commits,
2219 &tog_mutex);
2220 if (errcode)
2221 err = got_error_set_errno(errcode,
2222 "pthread_cond_wait");
2223 if (*a->quit)
2224 done = 1;
2228 errcode = pthread_mutex_unlock(&tog_mutex);
2229 if (errcode && err == NULL)
2230 err = got_error_set_errno(errcode,
2231 "pthread_mutex_unlock");
2233 a->log_complete = 1;
2234 return (void *)err;
2237 static const struct got_error *
2238 stop_log_thread(struct tog_log_view_state *s)
2240 const struct got_error *err = NULL;
2241 int errcode;
2243 if (s->thread) {
2244 s->quit = 1;
2245 errcode = pthread_cond_signal(&s->thread_args.need_commits);
2246 if (errcode)
2247 return got_error_set_errno(errcode,
2248 "pthread_cond_signal");
2249 errcode = pthread_mutex_unlock(&tog_mutex);
2250 if (errcode)
2251 return got_error_set_errno(errcode,
2252 "pthread_mutex_unlock");
2253 errcode = pthread_join(s->thread, (void **)&err);
2254 if (errcode)
2255 return got_error_set_errno(errcode, "pthread_join");
2256 errcode = pthread_mutex_lock(&tog_mutex);
2257 if (errcode)
2258 return got_error_set_errno(errcode,
2259 "pthread_mutex_lock");
2260 s->thread = 0; //NULL;
2263 if (s->thread_args.repo) {
2264 err = got_repo_close(s->thread_args.repo);
2265 s->thread_args.repo = NULL;
2268 if (s->thread_args.pack_fds) {
2269 const struct got_error *pack_err =
2270 got_repo_pack_fds_close(s->thread_args.pack_fds);
2271 if (err == NULL)
2272 err = pack_err;
2273 s->thread_args.pack_fds = NULL;
2276 if (s->thread_args.graph) {
2277 got_commit_graph_close(s->thread_args.graph);
2278 s->thread_args.graph = NULL;
2281 return err;
2284 static const struct got_error *
2285 close_log_view(struct tog_view *view)
2287 const struct got_error *err = NULL;
2288 struct tog_log_view_state *s = &view->state.log;
2289 int errcode;
2291 err = stop_log_thread(s);
2293 errcode = pthread_cond_destroy(&s->thread_args.need_commits);
2294 if (errcode && err == NULL)
2295 err = got_error_set_errno(errcode, "pthread_cond_destroy");
2297 errcode = pthread_cond_destroy(&s->thread_args.commit_loaded);
2298 if (errcode && err == NULL)
2299 err = got_error_set_errno(errcode, "pthread_cond_destroy");
2301 free_commits(&s->commits);
2302 free(s->in_repo_path);
2303 s->in_repo_path = NULL;
2304 free(s->start_id);
2305 s->start_id = NULL;
2306 free(s->head_ref_name);
2307 s->head_ref_name = NULL;
2308 return err;
2311 static const struct got_error *
2312 search_start_log_view(struct tog_view *view)
2314 struct tog_log_view_state *s = &view->state.log;
2316 s->matched_entry = NULL;
2317 s->search_entry = NULL;
2318 return NULL;
2321 static const struct got_error *
2322 search_next_log_view(struct tog_view *view)
2324 const struct got_error *err = NULL;
2325 struct tog_log_view_state *s = &view->state.log;
2326 struct commit_queue_entry *entry;
2328 /* Display progress update in log view. */
2329 show_log_view(view);
2330 update_panels();
2331 doupdate();
2333 if (s->search_entry) {
2334 int errcode, ch;
2335 errcode = pthread_mutex_unlock(&tog_mutex);
2336 if (errcode)
2337 return got_error_set_errno(errcode,
2338 "pthread_mutex_unlock");
2339 ch = wgetch(view->window);
2340 errcode = pthread_mutex_lock(&tog_mutex);
2341 if (errcode)
2342 return got_error_set_errno(errcode,
2343 "pthread_mutex_lock");
2344 if (ch == KEY_BACKSPACE) {
2345 view->search_next_done = TOG_SEARCH_HAVE_MORE;
2346 return NULL;
2348 if (view->searching == TOG_SEARCH_FORWARD)
2349 entry = TAILQ_NEXT(s->search_entry, entry);
2350 else
2351 entry = TAILQ_PREV(s->search_entry,
2352 commit_queue_head, entry);
2353 } else if (s->matched_entry) {
2354 if (view->searching == TOG_SEARCH_FORWARD)
2355 entry = TAILQ_NEXT(s->matched_entry, entry);
2356 else
2357 entry = TAILQ_PREV(s->matched_entry,
2358 commit_queue_head, entry);
2359 } else {
2360 entry = s->selected_entry;
2363 while (1) {
2364 int have_match = 0;
2366 if (entry == NULL) {
2367 if (s->thread_args.log_complete ||
2368 view->searching == TOG_SEARCH_BACKWARD) {
2369 view->search_next_done =
2370 (s->matched_entry == NULL ?
2371 TOG_SEARCH_HAVE_NONE : TOG_SEARCH_NO_MORE);
2372 s->search_entry = NULL;
2373 return NULL;
2376 * Poke the log thread for more commits and return,
2377 * allowing the main loop to make progress. Search
2378 * will resume at s->search_entry once we come back.
2380 s->thread_args.commits_needed++;
2381 return trigger_log_thread(view, 0);
2384 err = match_commit(&have_match, entry->id, entry->commit,
2385 &view->regex);
2386 if (err)
2387 break;
2388 if (have_match) {
2389 view->search_next_done = TOG_SEARCH_HAVE_MORE;
2390 s->matched_entry = entry;
2391 break;
2394 s->search_entry = entry;
2395 if (view->searching == TOG_SEARCH_FORWARD)
2396 entry = TAILQ_NEXT(entry, entry);
2397 else
2398 entry = TAILQ_PREV(entry, commit_queue_head, entry);
2401 if (s->matched_entry) {
2402 int cur = s->selected_entry->idx;
2403 while (cur < s->matched_entry->idx) {
2404 err = input_log_view(NULL, view, KEY_DOWN);
2405 if (err)
2406 return err;
2407 cur++;
2409 while (cur > s->matched_entry->idx) {
2410 err = input_log_view(NULL, view, KEY_UP);
2411 if (err)
2412 return err;
2413 cur--;
2417 s->search_entry = NULL;
2419 return NULL;
2422 static const struct got_error *
2423 open_log_view(struct tog_view *view, struct got_object_id *start_id,
2424 struct got_repository *repo, const char *head_ref_name,
2425 const char *in_repo_path, int log_branches)
2427 const struct got_error *err = NULL;
2428 struct tog_log_view_state *s = &view->state.log;
2429 struct got_repository *thread_repo = NULL;
2430 struct got_commit_graph *thread_graph = NULL;
2431 int errcode;
2433 if (in_repo_path != s->in_repo_path) {
2434 free(s->in_repo_path);
2435 s->in_repo_path = strdup(in_repo_path);
2436 if (s->in_repo_path == NULL)
2437 return got_error_from_errno("strdup");
2440 /* The commit queue only contains commits being displayed. */
2441 TAILQ_INIT(&s->commits.head);
2442 s->commits.ncommits = 0;
2444 s->repo = repo;
2445 if (head_ref_name) {
2446 s->head_ref_name = strdup(head_ref_name);
2447 if (s->head_ref_name == NULL) {
2448 err = got_error_from_errno("strdup");
2449 goto done;
2452 s->start_id = got_object_id_dup(start_id);
2453 if (s->start_id == NULL) {
2454 err = got_error_from_errno("got_object_id_dup");
2455 goto done;
2457 s->log_branches = log_branches;
2459 STAILQ_INIT(&s->colors);
2460 if (has_colors() && getenv("TOG_COLORS") != NULL) {
2461 err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
2462 get_color_value("TOG_COLOR_COMMIT"));
2463 if (err)
2464 goto done;
2465 err = add_color(&s->colors, "^$", TOG_COLOR_AUTHOR,
2466 get_color_value("TOG_COLOR_AUTHOR"));
2467 if (err) {
2468 free_colors(&s->colors);
2469 goto done;
2471 err = add_color(&s->colors, "^$", TOG_COLOR_DATE,
2472 get_color_value("TOG_COLOR_DATE"));
2473 if (err) {
2474 free_colors(&s->colors);
2475 goto done;
2479 view->show = show_log_view;
2480 view->input = input_log_view;
2481 view->close = close_log_view;
2482 view->search_start = search_start_log_view;
2483 view->search_next = search_next_log_view;
2485 if (s->thread_args.pack_fds == NULL) {
2486 err = got_repo_pack_fds_open(&s->thread_args.pack_fds);
2487 if (err)
2488 goto done;
2490 err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL,
2491 s->thread_args.pack_fds);
2492 if (err)
2493 goto done;
2494 err = got_commit_graph_open(&thread_graph, s->in_repo_path,
2495 !s->log_branches);
2496 if (err)
2497 goto done;
2498 err = got_commit_graph_iter_start(thread_graph, s->start_id,
2499 s->repo, NULL, NULL);
2500 if (err)
2501 goto done;
2503 errcode = pthread_cond_init(&s->thread_args.need_commits, NULL);
2504 if (errcode) {
2505 err = got_error_set_errno(errcode, "pthread_cond_init");
2506 goto done;
2508 errcode = pthread_cond_init(&s->thread_args.commit_loaded, NULL);
2509 if (errcode) {
2510 err = got_error_set_errno(errcode, "pthread_cond_init");
2511 goto done;
2514 s->thread_args.commits_needed = view->nlines;
2515 s->thread_args.graph = thread_graph;
2516 s->thread_args.commits = &s->commits;
2517 s->thread_args.in_repo_path = s->in_repo_path;
2518 s->thread_args.start_id = s->start_id;
2519 s->thread_args.repo = thread_repo;
2520 s->thread_args.log_complete = 0;
2521 s->thread_args.quit = &s->quit;
2522 s->thread_args.first_displayed_entry = &s->first_displayed_entry;
2523 s->thread_args.selected_entry = &s->selected_entry;
2524 s->thread_args.searching = &view->searching;
2525 s->thread_args.search_next_done = &view->search_next_done;
2526 s->thread_args.regex = &view->regex;
2527 done:
2528 if (err)
2529 close_log_view(view);
2530 return err;
2533 static const struct got_error *
2534 show_log_view(struct tog_view *view)
2536 const struct got_error *err;
2537 struct tog_log_view_state *s = &view->state.log;
2539 if (s->thread == 0) { //NULL) {
2540 int errcode = pthread_create(&s->thread, NULL, log_thread,
2541 &s->thread_args);
2542 if (errcode)
2543 return got_error_set_errno(errcode, "pthread_create");
2544 if (s->thread_args.commits_needed > 0) {
2545 err = trigger_log_thread(view, 1);
2546 if (err)
2547 return err;
2551 return draw_commits(view);
2554 static const struct got_error *
2555 input_log_view(struct tog_view **new_view, struct tog_view *view, int ch)
2557 const struct got_error *err = NULL;
2558 struct tog_log_view_state *s = &view->state.log;
2559 struct tog_view *diff_view = NULL, *tree_view = NULL;
2560 struct tog_view *ref_view = NULL;
2561 struct commit_queue_entry *entry;
2562 int begin_x = 0, n, nscroll = view->nlines - 1;
2564 if (s->thread_args.load_all) {
2565 if (ch == KEY_BACKSPACE)
2566 s->thread_args.load_all = 0;
2567 else if (s->thread_args.log_complete) {
2568 s->thread_args.load_all = 0;
2569 log_scroll_down(view, s->commits.ncommits);
2570 s->selected = MIN(view->nlines - 2,
2571 s->commits.ncommits - 1);
2572 select_commit(s);
2574 return NULL;
2577 switch (ch) {
2578 case 'q':
2579 s->quit = 1;
2580 break;
2581 case '0':
2582 view->x = 0;
2583 break;
2584 case '$':
2585 view->x = MAX(view->maxx - view->ncols / 2, 0);
2586 break;
2587 case KEY_RIGHT:
2588 case 'l':
2589 if (view->x + view->ncols / 2 < view->maxx)
2590 view->x += 2; /* move two columns right */
2591 break;
2592 case KEY_LEFT:
2593 case 'h':
2594 view->x -= MIN(view->x, 2); /* move two columns back */
2595 break;
2596 case 'k':
2597 case KEY_UP:
2598 case '<':
2599 case ',':
2600 case CTRL('p'):
2601 if (s->first_displayed_entry == NULL)
2602 break;
2603 if (s->selected > 0)
2604 s->selected--;
2605 else
2606 log_scroll_up(s, 1);
2607 select_commit(s);
2608 break;
2609 case 'g':
2610 case KEY_HOME:
2611 s->selected = 0;
2612 s->first_displayed_entry = TAILQ_FIRST(&s->commits.head);
2613 select_commit(s);
2614 break;
2615 case CTRL('u'):
2616 case 'u':
2617 nscroll /= 2;
2618 /* FALL THROUGH */
2619 case KEY_PPAGE:
2620 case CTRL('b'):
2621 if (s->first_displayed_entry == NULL)
2622 break;
2623 if (TAILQ_FIRST(&s->commits.head) == s->first_displayed_entry)
2624 s->selected = MAX(0, s->selected - nscroll - 1);
2625 else
2626 log_scroll_up(s, nscroll);
2627 select_commit(s);
2628 break;
2629 case 'j':
2630 case KEY_DOWN:
2631 case '>':
2632 case '.':
2633 case CTRL('n'):
2634 if (s->first_displayed_entry == NULL)
2635 break;
2636 if (s->selected < MIN(view->nlines - 2,
2637 s->commits.ncommits - 1))
2638 s->selected++;
2639 else {
2640 err = log_scroll_down(view, 1);
2641 if (err)
2642 break;
2644 select_commit(s);
2645 break;
2646 case 'G':
2647 case KEY_END: {
2648 /* We don't know yet how many commits, so we're forced to
2649 * traverse them all. */
2650 if (!s->thread_args.log_complete) {
2651 s->thread_args.load_all = 1;
2652 return trigger_log_thread(view, 0);
2655 s->selected = 0;
2656 entry = TAILQ_LAST(&s->commits.head, commit_queue_head);
2657 for (n = 0; n < view->nlines - 1; n++) {
2658 if (entry == NULL)
2659 break;
2660 s->first_displayed_entry = entry;
2661 entry = TAILQ_PREV(entry, commit_queue_head, entry);
2663 if (n > 0)
2664 s->selected = n - 1;
2665 select_commit(s);
2666 break;
2668 case CTRL('d'):
2669 case 'd':
2670 nscroll /= 2;
2671 /* FALL THROUGH */
2672 case KEY_NPAGE:
2673 case CTRL('f'): {
2674 struct commit_queue_entry *first;
2675 first = s->first_displayed_entry;
2676 if (first == NULL)
2677 break;
2678 err = log_scroll_down(view, nscroll);
2679 if (err)
2680 break;
2681 if (first == s->first_displayed_entry &&
2682 s->selected < MIN(view->nlines - 2,
2683 s->commits.ncommits - 1)) {
2684 /* can't scroll further down */
2685 s->selected += MIN(s->last_displayed_entry->idx -
2686 s->selected_entry->idx, nscroll + 1);
2688 select_commit(s);
2689 break;
2691 case KEY_RESIZE:
2692 if (s->selected > view->nlines - 2)
2693 s->selected = view->nlines - 2;
2694 if (s->selected > s->commits.ncommits - 1)
2695 s->selected = s->commits.ncommits - 1;
2696 select_commit(s);
2697 if (s->commits.ncommits < view->nlines - 1 &&
2698 !s->thread_args.log_complete) {
2699 s->thread_args.commits_needed += (view->nlines - 1) -
2700 s->commits.ncommits;
2701 err = trigger_log_thread(view, 1);
2703 break;
2704 case KEY_ENTER:
2705 case ' ':
2706 case '\r':
2707 if (s->selected_entry == NULL)
2708 break;
2709 if (view_is_parent_view(view))
2710 begin_x = view_split_begin_x(view->begin_x);
2711 err = open_diff_view_for_commit(&diff_view, begin_x,
2712 s->selected_entry->commit, s->selected_entry->id,
2713 view, s->repo);
2714 if (err)
2715 break;
2716 view->focussed = 0;
2717 diff_view->focussed = 1;
2718 if (view_is_parent_view(view)) {
2719 err = view_close_child(view);
2720 if (err)
2721 return err;
2722 err = view_set_child(view, diff_view);
2723 if (err)
2724 return err;
2725 view->focus_child = 1;
2726 } else
2727 *new_view = diff_view;
2728 break;
2729 case 't':
2730 if (s->selected_entry == NULL)
2731 break;
2732 if (view_is_parent_view(view))
2733 begin_x = view_split_begin_x(view->begin_x);
2734 err = browse_commit_tree(&tree_view, begin_x,
2735 s->selected_entry, s->in_repo_path, s->head_ref_name,
2736 s->repo);
2737 if (err)
2738 break;
2739 view->focussed = 0;
2740 tree_view->focussed = 1;
2741 if (view_is_parent_view(view)) {
2742 err = view_close_child(view);
2743 if (err)
2744 return err;
2745 err = view_set_child(view, tree_view);
2746 if (err)
2747 return err;
2748 view->focus_child = 1;
2749 } else
2750 *new_view = tree_view;
2751 break;
2752 case KEY_BACKSPACE:
2753 case CTRL('l'):
2754 case 'B':
2755 if (ch == KEY_BACKSPACE &&
2756 got_path_is_root_dir(s->in_repo_path))
2757 break;
2758 err = stop_log_thread(s);
2759 if (err)
2760 return err;
2761 if (ch == KEY_BACKSPACE) {
2762 char *parent_path;
2763 err = got_path_dirname(&parent_path, s->in_repo_path);
2764 if (err)
2765 return err;
2766 free(s->in_repo_path);
2767 s->in_repo_path = parent_path;
2768 s->thread_args.in_repo_path = s->in_repo_path;
2769 } else if (ch == CTRL('l')) {
2770 struct got_object_id *start_id;
2771 err = got_repo_match_object_id(&start_id, NULL,
2772 s->head_ref_name ? s->head_ref_name : GOT_REF_HEAD,
2773 GOT_OBJ_TYPE_COMMIT, &tog_refs, s->repo);
2774 if (err)
2775 return err;
2776 free(s->start_id);
2777 s->start_id = start_id;
2778 s->thread_args.start_id = s->start_id;
2779 } else /* 'B' */
2780 s->log_branches = !s->log_branches;
2782 if (s->thread_args.pack_fds == NULL) {
2783 err = got_repo_pack_fds_open(&s->thread_args.pack_fds);
2784 if (err)
2785 return err;
2787 err = got_repo_open(&s->thread_args.repo,
2788 got_repo_get_path(s->repo), NULL,
2789 s->thread_args.pack_fds);
2790 if (err)
2791 return err;
2792 tog_free_refs();
2793 err = tog_load_refs(s->repo, 0);
2794 if (err)
2795 return err;
2796 err = got_commit_graph_open(&s->thread_args.graph,
2797 s->in_repo_path, !s->log_branches);
2798 if (err)
2799 return err;
2800 err = got_commit_graph_iter_start(s->thread_args.graph,
2801 s->start_id, s->repo, NULL, NULL);
2802 if (err)
2803 return err;
2804 free_commits(&s->commits);
2805 s->first_displayed_entry = NULL;
2806 s->last_displayed_entry = NULL;
2807 s->selected_entry = NULL;
2808 s->selected = 0;
2809 s->thread_args.log_complete = 0;
2810 s->quit = 0;
2811 s->thread_args.commits_needed = view->nlines;
2812 break;
2813 case 'r':
2814 if (view_is_parent_view(view))
2815 begin_x = view_split_begin_x(view->begin_x);
2816 ref_view = view_open(view->nlines, view->ncols,
2817 view->begin_y, begin_x, TOG_VIEW_REF);
2818 if (ref_view == NULL)
2819 return got_error_from_errno("view_open");
2820 err = open_ref_view(ref_view, s->repo);
2821 if (err) {
2822 view_close(ref_view);
2823 return err;
2825 view->focussed = 0;
2826 ref_view->focussed = 1;
2827 if (view_is_parent_view(view)) {
2828 err = view_close_child(view);
2829 if (err)
2830 return err;
2831 err = view_set_child(view, ref_view);
2832 if (err)
2833 return err;
2834 view->focus_child = 1;
2835 } else
2836 *new_view = ref_view;
2837 break;
2838 default:
2839 break;
2842 return err;
2845 static const struct got_error *
2846 apply_unveil(const char *repo_path, const char *worktree_path)
2848 const struct got_error *error;
2850 #ifdef PROFILE
2851 if (unveil("gmon.out", "rwc") != 0)
2852 return got_error_from_errno2("unveil", "gmon.out");
2853 #endif
2854 if (repo_path && unveil(repo_path, "r") != 0)
2855 return got_error_from_errno2("unveil", repo_path);
2857 if (worktree_path && unveil(worktree_path, "rwc") != 0)
2858 return got_error_from_errno2("unveil", worktree_path);
2860 if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
2861 return got_error_from_errno2("unveil", GOT_TMPDIR_STR);
2863 error = got_privsep_unveil_exec_helpers();
2864 if (error != NULL)
2865 return error;
2867 if (unveil(NULL, NULL) != 0)
2868 return got_error_from_errno("unveil");
2870 return NULL;
2873 static void
2874 init_curses(void)
2877 * Override default signal handlers before starting ncurses.
2878 * This should prevent ncurses from installing its own
2879 * broken cleanup() signal handler.
2881 signal(SIGWINCH, tog_sigwinch);
2882 signal(SIGPIPE, tog_sigpipe);
2883 signal(SIGCONT, tog_sigcont);
2884 signal(SIGINT, tog_sigint);
2885 signal(SIGTERM, tog_sigterm);
2887 initscr();
2888 cbreak();
2889 halfdelay(1); /* Do fast refresh while initial view is loading. */
2890 noecho();
2891 nonl();
2892 intrflush(stdscr, FALSE);
2893 keypad(stdscr, TRUE);
2894 curs_set(0);
2895 if (getenv("TOG_COLORS") != NULL) {
2896 start_color();
2897 use_default_colors();
2901 static const struct got_error *
2902 get_in_repo_path_from_argv0(char **in_repo_path, int argc, char *argv[],
2903 struct got_repository *repo, struct got_worktree *worktree)
2905 const struct got_error *err = NULL;
2907 if (argc == 0) {
2908 *in_repo_path = strdup("/");
2909 if (*in_repo_path == NULL)
2910 return got_error_from_errno("strdup");
2911 return NULL;
2914 if (worktree) {
2915 const char *prefix = got_worktree_get_path_prefix(worktree);
2916 char *p;
2918 err = got_worktree_resolve_path(&p, worktree, argv[0]);
2919 if (err)
2920 return err;
2921 if (asprintf(in_repo_path, "%s%s%s", prefix,
2922 (p[0] != '\0' && !got_path_is_root_dir(prefix)) ? "/" : "",
2923 p) == -1) {
2924 err = got_error_from_errno("asprintf");
2925 *in_repo_path = NULL;
2927 free(p);
2928 } else
2929 err = got_repo_map_path(in_repo_path, repo, argv[0]);
2931 return err;
2934 static const struct got_error *
2935 cmd_log(int argc, char *argv[])
2937 const struct got_error *error;
2938 struct got_repository *repo = NULL;
2939 struct got_worktree *worktree = NULL;
2940 struct got_object_id *start_id = NULL;
2941 char *in_repo_path = NULL, *repo_path = NULL, *cwd = NULL;
2942 char *start_commit = NULL, *label = NULL;
2943 struct got_reference *ref = NULL;
2944 const char *head_ref_name = NULL;
2945 int ch, log_branches = 0;
2946 struct tog_view *view;
2947 int *pack_fds = NULL;
2949 while ((ch = getopt(argc, argv, "bc:r:")) != -1) {
2950 switch (ch) {
2951 case 'b':
2952 log_branches = 1;
2953 break;
2954 case 'c':
2955 start_commit = optarg;
2956 break;
2957 case 'r':
2958 repo_path = realpath(optarg, NULL);
2959 if (repo_path == NULL)
2960 return got_error_from_errno2("realpath",
2961 optarg);
2962 break;
2963 default:
2964 usage_log();
2965 /* NOTREACHED */
2969 argc -= optind;
2970 argv += optind;
2972 if (argc > 1)
2973 usage_log();
2975 error = got_repo_pack_fds_open(&pack_fds);
2976 if (error != NULL)
2977 goto done;
2979 if (repo_path == NULL) {
2980 cwd = getcwd(NULL, 0);
2981 if (cwd == NULL)
2982 return got_error_from_errno("getcwd");
2983 error = got_worktree_open(&worktree, cwd);
2984 if (error && error->code != GOT_ERR_NOT_WORKTREE)
2985 goto done;
2986 if (worktree)
2987 repo_path =
2988 strdup(got_worktree_get_repo_path(worktree));
2989 else
2990 repo_path = strdup(cwd);
2991 if (repo_path == NULL) {
2992 error = got_error_from_errno("strdup");
2993 goto done;
2997 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
2998 if (error != NULL)
2999 goto done;
3001 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
3002 repo, worktree);
3003 if (error)
3004 goto done;
3006 init_curses();
3008 error = apply_unveil(got_repo_get_path(repo),
3009 worktree ? got_worktree_get_root_path(worktree) : NULL);
3010 if (error)
3011 goto done;
3013 /* already loaded by tog_log_with_path()? */
3014 if (TAILQ_EMPTY(&tog_refs)) {
3015 error = tog_load_refs(repo, 0);
3016 if (error)
3017 goto done;
3020 if (start_commit == NULL) {
3021 error = got_repo_match_object_id(&start_id, &label,
3022 worktree ? got_worktree_get_head_ref_name(worktree) :
3023 GOT_REF_HEAD, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
3024 if (error)
3025 goto done;
3026 head_ref_name = label;
3027 } else {
3028 error = got_ref_open(&ref, repo, start_commit, 0);
3029 if (error == NULL)
3030 head_ref_name = got_ref_get_name(ref);
3031 else if (error->code != GOT_ERR_NOT_REF)
3032 goto done;
3033 error = got_repo_match_object_id(&start_id, NULL,
3034 start_commit, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
3035 if (error)
3036 goto done;
3039 view = view_open(0, 0, 0, 0, TOG_VIEW_LOG);
3040 if (view == NULL) {
3041 error = got_error_from_errno("view_open");
3042 goto done;
3044 error = open_log_view(view, start_id, repo, head_ref_name,
3045 in_repo_path, log_branches);
3046 if (error)
3047 goto done;
3048 if (worktree) {
3049 /* Release work tree lock. */
3050 got_worktree_close(worktree);
3051 worktree = NULL;
3053 error = view_loop(view);
3054 done:
3055 free(in_repo_path);
3056 free(repo_path);
3057 free(cwd);
3058 free(start_id);
3059 free(label);
3060 if (ref)
3061 got_ref_close(ref);
3062 if (repo) {
3063 const struct got_error *close_err = got_repo_close(repo);
3064 if (error == NULL)
3065 error = close_err;
3067 if (worktree)
3068 got_worktree_close(worktree);
3069 if (pack_fds) {
3070 const struct got_error *pack_err =
3071 got_repo_pack_fds_close(pack_fds);
3072 if (error == NULL)
3073 error = pack_err;
3075 tog_free_refs();
3076 return error;
3079 __dead static void
3080 usage_diff(void)
3082 endwin();
3083 fprintf(stderr, "usage: %s diff [-a] [-C number] [-r repository-path] "
3084 "[-w] object1 object2\n", getprogname());
3085 exit(1);
3088 static int
3089 match_line(const char *line, regex_t *regex, size_t nmatch,
3090 regmatch_t *regmatch)
3092 return regexec(regex, line, nmatch, regmatch, 0) == 0;
3095 struct tog_color *
3096 match_color(struct tog_colors *colors, const char *line)
3098 struct tog_color *tc = NULL;
3100 STAILQ_FOREACH(tc, colors, entry) {
3101 if (match_line(line, &tc->regex, 0, NULL))
3102 return tc;
3105 return NULL;
3108 static const struct got_error *
3109 add_matched_line(int *wtotal, const char *line, int wlimit, int col_tab_align,
3110 WINDOW *window, int skipcol, regmatch_t *regmatch)
3112 const struct got_error *err = NULL;
3113 char *exstr = NULL;
3114 wchar_t *wline = NULL;
3115 int rme, rms, n, width, scrollx;
3116 int width0 = 0, width1 = 0, width2 = 0;
3117 char *seg0 = NULL, *seg1 = NULL, *seg2 = NULL;
3119 *wtotal = 0;
3121 rms = regmatch->rm_so;
3122 rme = regmatch->rm_eo;
3124 err = expand_tab(&exstr, line);
3125 if (err)
3126 return err;
3128 /* Split the line into 3 segments, according to match offsets. */
3129 seg0 = strndup(exstr, rms);
3130 if (seg0 == NULL) {
3131 err = got_error_from_errno("strndup");
3132 goto done;
3134 seg1 = strndup(exstr + rms, rme - rms);
3135 if (seg1 == NULL) {
3136 err = got_error_from_errno("strndup");
3137 goto done;
3139 seg2 = strdup(exstr + rme);
3140 if (seg2 == NULL) {
3141 err = got_error_from_errno("strndup");
3142 goto done;
3145 /* draw up to matched token if we haven't scrolled past it */
3146 err = format_line(&wline, &width0, NULL, seg0, 0, wlimit,
3147 col_tab_align, 1);
3148 if (err)
3149 goto done;
3150 n = MAX(width0 - skipcol, 0);
3151 if (n) {
3152 free(wline);
3153 err = format_line(&wline, &width, &scrollx, seg0, skipcol,
3154 wlimit, col_tab_align, 1);
3155 if (err)
3156 goto done;
3157 waddwstr(window, &wline[scrollx]);
3158 wlimit -= width;
3159 *wtotal += width;
3162 if (wlimit > 0) {
3163 int i = 0, w = 0;
3164 size_t wlen;
3166 free(wline);
3167 err = format_line(&wline, &width1, NULL, seg1, 0, wlimit,
3168 col_tab_align, 1);
3169 if (err)
3170 goto done;
3171 wlen = wcslen(wline);
3172 while (i < wlen) {
3173 width = wcwidth(wline[i]);
3174 if (width == -1) {
3175 /* should not happen, tabs are expanded */
3176 err = got_error(GOT_ERR_RANGE);
3177 goto done;
3179 if (width0 + w + width > skipcol)
3180 break;
3181 w += width;
3182 i++;
3184 /* draw (visible part of) matched token (if scrolled into it) */
3185 if (width1 - w > 0) {
3186 wattron(window, A_STANDOUT);
3187 waddwstr(window, &wline[i]);
3188 wattroff(window, A_STANDOUT);
3189 wlimit -= (width1 - w);
3190 *wtotal += (width1 - w);
3194 if (wlimit > 0) { /* draw rest of line */
3195 free(wline);
3196 if (skipcol > width0 + width1) {
3197 err = format_line(&wline, &width2, &scrollx, seg2,
3198 skipcol - (width0 + width1), wlimit,
3199 col_tab_align, 1);
3200 if (err)
3201 goto done;
3202 waddwstr(window, &wline[scrollx]);
3203 } else {
3204 err = format_line(&wline, &width2, NULL, seg2, 0,
3205 wlimit, col_tab_align, 1);
3206 if (err)
3207 goto done;
3208 waddwstr(window, wline);
3210 *wtotal += width2;
3212 done:
3213 free(wline);
3214 free(exstr);
3215 free(seg0);
3216 free(seg1);
3217 free(seg2);
3218 return err;
3221 static const struct got_error *
3222 draw_file(struct tog_view *view, const char *header)
3224 struct tog_diff_view_state *s = &view->state.diff;
3225 regmatch_t *regmatch = &view->regmatch;
3226 const struct got_error *err;
3227 int nprinted = 0;
3228 char *line;
3229 size_t linesize = 0;
3230 ssize_t linelen;
3231 struct tog_color *tc;
3232 wchar_t *wline;
3233 int width;
3234 int max_lines = view->nlines;
3235 int nlines = s->nlines;
3236 off_t line_offset;
3238 line_offset = s->line_offsets[s->first_displayed_line - 1];
3239 if (fseeko(s->f, line_offset, SEEK_SET) == -1)
3240 return got_error_from_errno("fseek");
3242 werase(view->window);
3244 if (header) {
3245 if (asprintf(&line, "[%d/%d] %s",
3246 s->first_displayed_line - 1 + s->selected_line, nlines,
3247 header) == -1)
3248 return got_error_from_errno("asprintf");
3249 err = format_line(&wline, &width, NULL, line, 0, view->ncols,
3250 0, 0);
3251 free(line);
3252 if (err)
3253 return err;
3255 if (view_needs_focus_indication(view))
3256 wstandout(view->window);
3257 waddwstr(view->window, wline);
3258 free(wline);
3259 wline = NULL;
3260 if (view_needs_focus_indication(view))
3261 wstandend(view->window);
3262 if (width <= view->ncols - 1)
3263 waddch(view->window, '\n');
3265 if (max_lines <= 1)
3266 return NULL;
3267 max_lines--;
3270 s->eof = 0;
3271 view->maxx = 0;
3272 line = NULL;
3273 while (max_lines > 0 && nprinted < max_lines) {
3274 linelen = getline(&line, &linesize, s->f);
3275 if (linelen == -1) {
3276 if (feof(s->f)) {
3277 s->eof = 1;
3278 break;
3280 free(line);
3281 return got_ferror(s->f, GOT_ERR_IO);
3284 /* Set view->maxx based on full line length. */
3285 err = format_line(&wline, &width, NULL, line, 0, INT_MAX, 0,
3286 view->x ? 1 : 0);
3287 if (err) {
3288 free(line);
3289 return err;
3291 view->maxx = MAX(view->maxx, width);
3292 free(wline);
3293 wline = NULL;
3295 tc = match_color(&s->colors, line);
3296 if (tc)
3297 wattr_on(view->window,
3298 COLOR_PAIR(tc->colorpair), NULL);
3299 if (s->first_displayed_line + nprinted == s->matched_line &&
3300 regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
3301 err = add_matched_line(&width, line, view->ncols, 0,
3302 view->window, view->x, regmatch);
3303 if (err) {
3304 free(line);
3305 return err;
3307 } else {
3308 int skip;
3309 err = format_line(&wline, &width, &skip, line,
3310 view->x, view->ncols, 0, view->x ? 1 : 0);
3311 if (err) {
3312 free(line);
3313 return err;
3315 waddwstr(view->window, &wline[skip]);
3316 free(wline);
3317 wline = NULL;
3319 if (tc)
3320 wattr_off(view->window,
3321 COLOR_PAIR(tc->colorpair), NULL);
3322 if (width <= view->ncols - 1)
3323 waddch(view->window, '\n');
3324 nprinted++;
3326 free(line);
3327 if (nprinted >= 1)
3328 s->last_displayed_line = s->first_displayed_line +
3329 (nprinted - 1);
3330 else
3331 s->last_displayed_line = s->first_displayed_line;
3333 view_vborder(view);
3335 if (s->eof) {
3336 while (nprinted < view->nlines) {
3337 waddch(view->window, '\n');
3338 nprinted++;
3341 err = format_line(&wline, &width, NULL, TOG_EOF_STRING, 0,
3342 view->ncols, 0, 0);
3343 if (err) {
3344 return err;
3347 wstandout(view->window);
3348 waddwstr(view->window, wline);
3349 free(wline);
3350 wline = NULL;
3351 wstandend(view->window);
3354 return NULL;
3357 static char *
3358 get_datestr(time_t *time, char *datebuf)
3360 struct tm mytm, *tm;
3361 char *p, *s;
3363 tm = gmtime_r(time, &mytm);
3364 if (tm == NULL)
3365 return NULL;
3366 s = asctime_r(tm, datebuf);
3367 if (s == NULL)
3368 return NULL;
3369 p = strchr(s, '\n');
3370 if (p)
3371 *p = '\0';
3372 return s;
3375 static const struct got_error *
3376 get_changed_paths(struct got_pathlist_head *paths,
3377 struct got_commit_object *commit, struct got_repository *repo)
3379 const struct got_error *err = NULL;
3380 struct got_object_id *tree_id1 = NULL, *tree_id2 = NULL;
3381 struct got_tree_object *tree1 = NULL, *tree2 = NULL;
3382 struct got_object_qid *qid;
3384 qid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
3385 if (qid != NULL) {
3386 struct got_commit_object *pcommit;
3387 err = got_object_open_as_commit(&pcommit, repo,
3388 &qid->id);
3389 if (err)
3390 return err;
3392 tree_id1 = got_object_id_dup(
3393 got_object_commit_get_tree_id(pcommit));
3394 if (tree_id1 == NULL) {
3395 got_object_commit_close(pcommit);
3396 return got_error_from_errno("got_object_id_dup");
3398 got_object_commit_close(pcommit);
3402 if (tree_id1) {
3403 err = got_object_open_as_tree(&tree1, repo, tree_id1);
3404 if (err)
3405 goto done;
3408 tree_id2 = got_object_commit_get_tree_id(commit);
3409 err = got_object_open_as_tree(&tree2, repo, tree_id2);
3410 if (err)
3411 goto done;
3413 err = got_diff_tree(tree1, tree2, NULL, NULL, "", "", repo,
3414 got_diff_tree_collect_changed_paths, paths, 0);
3415 done:
3416 if (tree1)
3417 got_object_tree_close(tree1);
3418 if (tree2)
3419 got_object_tree_close(tree2);
3420 free(tree_id1);
3421 return err;
3424 static const struct got_error *
3425 add_line_offset(off_t **line_offsets, size_t *nlines, off_t off)
3427 off_t *p;
3429 p = reallocarray(*line_offsets, *nlines + 1, sizeof(off_t));
3430 if (p == NULL)
3431 return got_error_from_errno("reallocarray");
3432 *line_offsets = p;
3433 (*line_offsets)[*nlines] = off;
3434 (*nlines)++;
3435 return NULL;
3438 static const struct got_error *
3439 write_commit_info(off_t **line_offsets, size_t *nlines,
3440 struct got_object_id *commit_id, struct got_reflist_head *refs,
3441 struct got_repository *repo, FILE *outfile)
3443 const struct got_error *err = NULL;
3444 char datebuf[26], *datestr;
3445 struct got_commit_object *commit;
3446 char *id_str = NULL, *logmsg = NULL, *s = NULL, *line;
3447 time_t committer_time;
3448 const char *author, *committer;
3449 char *refs_str = NULL;
3450 struct got_pathlist_head changed_paths;
3451 struct got_pathlist_entry *pe;
3452 off_t outoff = 0;
3453 int n;
3455 TAILQ_INIT(&changed_paths);
3457 if (refs) {
3458 err = build_refs_str(&refs_str, refs, commit_id, repo);
3459 if (err)
3460 return err;
3463 err = got_object_open_as_commit(&commit, repo, commit_id);
3464 if (err)
3465 return err;
3467 err = got_object_id_str(&id_str, commit_id);
3468 if (err) {
3469 err = got_error_from_errno("got_object_id_str");
3470 goto done;
3473 err = add_line_offset(line_offsets, nlines, 0);
3474 if (err)
3475 goto done;
3477 n = fprintf(outfile, "commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
3478 refs_str ? refs_str : "", refs_str ? ")" : "");
3479 if (n < 0) {
3480 err = got_error_from_errno("fprintf");
3481 goto done;
3483 outoff += n;
3484 err = add_line_offset(line_offsets, nlines, outoff);
3485 if (err)
3486 goto done;
3488 n = fprintf(outfile, "from: %s\n",
3489 got_object_commit_get_author(commit));
3490 if (n < 0) {
3491 err = got_error_from_errno("fprintf");
3492 goto done;
3494 outoff += n;
3495 err = add_line_offset(line_offsets, nlines, outoff);
3496 if (err)
3497 goto done;
3499 committer_time = got_object_commit_get_committer_time(commit);
3500 datestr = get_datestr(&committer_time, datebuf);
3501 if (datestr) {
3502 n = fprintf(outfile, "date: %s UTC\n", datestr);
3503 if (n < 0) {
3504 err = got_error_from_errno("fprintf");
3505 goto done;
3507 outoff += n;
3508 err = add_line_offset(line_offsets, nlines, outoff);
3509 if (err)
3510 goto done;
3512 author = got_object_commit_get_author(commit);
3513 committer = got_object_commit_get_committer(commit);
3514 if (strcmp(author, committer) != 0) {
3515 n = fprintf(outfile, "via: %s\n", committer);
3516 if (n < 0) {
3517 err = got_error_from_errno("fprintf");
3518 goto done;
3520 outoff += n;
3521 err = add_line_offset(line_offsets, nlines, outoff);
3522 if (err)
3523 goto done;
3525 if (got_object_commit_get_nparents(commit) > 1) {
3526 const struct got_object_id_queue *parent_ids;
3527 struct got_object_qid *qid;
3528 int pn = 1;
3529 parent_ids = got_object_commit_get_parent_ids(commit);
3530 STAILQ_FOREACH(qid, parent_ids, entry) {
3531 err = got_object_id_str(&id_str, &qid->id);
3532 if (err)
3533 goto done;
3534 n = fprintf(outfile, "parent %d: %s\n", pn++, id_str);
3535 if (n < 0) {
3536 err = got_error_from_errno("fprintf");
3537 goto done;
3539 outoff += n;
3540 err = add_line_offset(line_offsets, nlines, outoff);
3541 if (err)
3542 goto done;
3543 free(id_str);
3544 id_str = NULL;
3548 err = got_object_commit_get_logmsg(&logmsg, commit);
3549 if (err)
3550 goto done;
3551 s = logmsg;
3552 while ((line = strsep(&s, "\n")) != NULL) {
3553 n = fprintf(outfile, "%s\n", line);
3554 if (n < 0) {
3555 err = got_error_from_errno("fprintf");
3556 goto done;
3558 outoff += n;
3559 err = add_line_offset(line_offsets, nlines, outoff);
3560 if (err)
3561 goto done;
3564 err = get_changed_paths(&changed_paths, commit, repo);
3565 if (err)
3566 goto done;
3567 TAILQ_FOREACH(pe, &changed_paths, entry) {
3568 struct got_diff_changed_path *cp = pe->data;
3569 n = fprintf(outfile, "%c %s\n", cp->status, pe->path);
3570 if (n < 0) {
3571 err = got_error_from_errno("fprintf");
3572 goto done;
3574 outoff += n;
3575 err = add_line_offset(line_offsets, nlines, outoff);
3576 if (err)
3577 goto done;
3578 free((char *)pe->path);
3579 free(pe->data);
3582 fputc('\n', outfile);
3583 outoff++;
3584 err = add_line_offset(line_offsets, nlines, outoff);
3585 done:
3586 got_pathlist_free(&changed_paths);
3587 free(id_str);
3588 free(logmsg);
3589 free(refs_str);
3590 got_object_commit_close(commit);
3591 if (err) {
3592 free(*line_offsets);
3593 *line_offsets = NULL;
3594 *nlines = 0;
3596 return err;
3599 static const struct got_error *
3600 create_diff(struct tog_diff_view_state *s)
3602 const struct got_error *err = NULL;
3603 FILE *f = NULL;
3604 int obj_type;
3606 free(s->line_offsets);
3607 s->line_offsets = malloc(sizeof(off_t));
3608 if (s->line_offsets == NULL)
3609 return got_error_from_errno("malloc");
3610 s->nlines = 0;
3612 f = got_opentemp();
3613 if (f == NULL) {
3614 err = got_error_from_errno("got_opentemp");
3615 goto done;
3617 if (s->f && fclose(s->f) == EOF) {
3618 err = got_error_from_errno("fclose");
3619 goto done;
3621 s->f = f;
3623 if (s->id1)
3624 err = got_object_get_type(&obj_type, s->repo, s->id1);
3625 else
3626 err = got_object_get_type(&obj_type, s->repo, s->id2);
3627 if (err)
3628 goto done;
3630 switch (obj_type) {
3631 case GOT_OBJ_TYPE_BLOB:
3632 err = got_diff_objects_as_blobs(&s->line_offsets, &s->nlines,
3633 s->f1, s->f2, s->id1, s->id2, s->label1, s->label2,
3634 s->diff_context, s->ignore_whitespace, s->force_text_diff,
3635 s->repo, s->f);
3636 break;
3637 case GOT_OBJ_TYPE_TREE:
3638 err = got_diff_objects_as_trees(&s->line_offsets, &s->nlines,
3639 s->f1, s->f2, s->id1, s->id2, NULL, "", "", s->diff_context,
3640 s->ignore_whitespace, s->force_text_diff, s->repo, s->f);
3641 break;
3642 case GOT_OBJ_TYPE_COMMIT: {
3643 const struct got_object_id_queue *parent_ids;
3644 struct got_object_qid *pid;
3645 struct got_commit_object *commit2;
3646 struct got_reflist_head *refs;
3648 err = got_object_open_as_commit(&commit2, s->repo, s->id2);
3649 if (err)
3650 goto done;
3651 refs = got_reflist_object_id_map_lookup(tog_refs_idmap, s->id2);
3652 /* Show commit info if we're diffing to a parent/root commit. */
3653 if (s->id1 == NULL) {
3654 err = write_commit_info(&s->line_offsets, &s->nlines,
3655 s->id2, refs, s->repo, s->f);
3656 if (err)
3657 goto done;
3658 } else {
3659 parent_ids = got_object_commit_get_parent_ids(commit2);
3660 STAILQ_FOREACH(pid, parent_ids, entry) {
3661 if (got_object_id_cmp(s->id1, &pid->id) == 0) {
3662 err = write_commit_info(
3663 &s->line_offsets, &s->nlines,
3664 s->id2, refs, s->repo, s->f);
3665 if (err)
3666 goto done;
3667 break;
3671 got_object_commit_close(commit2);
3673 err = got_diff_objects_as_commits(&s->line_offsets, &s->nlines,
3674 s->f1, s->f2, s->id1, s->id2, NULL, s->diff_context,
3675 s->ignore_whitespace, s->force_text_diff, s->repo, s->f);
3676 break;
3678 default:
3679 err = got_error(GOT_ERR_OBJ_TYPE);
3680 break;
3682 if (err)
3683 goto done;
3684 done:
3685 if (s->f && fflush(s->f) != 0 && err == NULL)
3686 err = got_error_from_errno("fflush");
3687 return err;
3690 static void
3691 diff_view_indicate_progress(struct tog_view *view)
3693 mvwaddstr(view->window, 0, 0, "diffing...");
3694 update_panels();
3695 doupdate();
3698 static const struct got_error *
3699 search_start_diff_view(struct tog_view *view)
3701 struct tog_diff_view_state *s = &view->state.diff;
3703 s->matched_line = 0;
3704 return NULL;
3707 static const struct got_error *
3708 search_next_diff_view(struct tog_view *view)
3710 struct tog_diff_view_state *s = &view->state.diff;
3711 const struct got_error *err = NULL;
3712 int lineno;
3713 char *line = NULL;
3714 size_t linesize = 0;
3715 ssize_t linelen;
3717 if (!view->searching) {
3718 view->search_next_done = TOG_SEARCH_HAVE_MORE;
3719 return NULL;
3722 if (s->matched_line) {
3723 if (view->searching == TOG_SEARCH_FORWARD)
3724 lineno = s->matched_line + 1;
3725 else
3726 lineno = s->matched_line - 1;
3727 } else
3728 lineno = s->first_displayed_line;
3730 while (1) {
3731 off_t offset;
3733 if (lineno <= 0 || lineno > s->nlines) {
3734 if (s->matched_line == 0) {
3735 view->search_next_done = TOG_SEARCH_HAVE_MORE;
3736 break;
3739 if (view->searching == TOG_SEARCH_FORWARD)
3740 lineno = 1;
3741 else
3742 lineno = s->nlines;
3745 offset = s->line_offsets[lineno - 1];
3746 if (fseeko(s->f, offset, SEEK_SET) != 0) {
3747 free(line);
3748 return got_error_from_errno("fseeko");
3750 linelen = getline(&line, &linesize, s->f);
3751 if (linelen != -1) {
3752 char *exstr;
3753 err = expand_tab(&exstr, line);
3754 if (err)
3755 break;
3756 if (match_line(exstr, &view->regex, 1,
3757 &view->regmatch)) {
3758 view->search_next_done = TOG_SEARCH_HAVE_MORE;
3759 s->matched_line = lineno;
3760 free(exstr);
3761 break;
3763 free(exstr);
3765 if (view->searching == TOG_SEARCH_FORWARD)
3766 lineno++;
3767 else
3768 lineno--;
3770 free(line);
3772 if (s->matched_line) {
3773 s->first_displayed_line = s->matched_line;
3774 s->selected_line = 1;
3777 return err;
3780 static const struct got_error *
3781 close_diff_view(struct tog_view *view)
3783 const struct got_error *err = NULL;
3784 struct tog_diff_view_state *s = &view->state.diff;
3786 free(s->id1);
3787 s->id1 = NULL;
3788 free(s->id2);
3789 s->id2 = NULL;
3790 if (s->f && fclose(s->f) == EOF)
3791 err = got_error_from_errno("fclose");
3792 s->f = NULL;
3793 if (s->f1 && fclose(s->f1) == EOF)
3794 err = got_error_from_errno("fclose");
3795 s->f1 = NULL;
3796 if (s->f2 && fclose(s->f2) == EOF)
3797 err = got_error_from_errno("fclose");
3798 s->f2 = NULL;
3799 free_colors(&s->colors);
3800 free(s->line_offsets);
3801 s->line_offsets = NULL;
3802 s->nlines = 0;
3803 return err;
3806 static const struct got_error *
3807 open_diff_view(struct tog_view *view, struct got_object_id *id1,
3808 struct got_object_id *id2, const char *label1, const char *label2,
3809 int diff_context, int ignore_whitespace, int force_text_diff,
3810 struct tog_view *log_view, struct got_repository *repo)
3812 const struct got_error *err;
3813 struct tog_diff_view_state *s = &view->state.diff;
3815 memset(s, 0, sizeof(*s));
3817 if (id1 != NULL && id2 != NULL) {
3818 int type1, type2;
3819 err = got_object_get_type(&type1, repo, id1);
3820 if (err)
3821 return err;
3822 err = got_object_get_type(&type2, repo, id2);
3823 if (err)
3824 return err;
3826 if (type1 != type2)
3827 return got_error(GOT_ERR_OBJ_TYPE);
3829 s->first_displayed_line = 1;
3830 s->last_displayed_line = view->nlines;
3831 s->selected_line = 1;
3832 s->repo = repo;
3833 s->id1 = id1;
3834 s->id2 = id2;
3835 s->label1 = label1;
3836 s->label2 = label2;
3838 if (id1) {
3839 s->id1 = got_object_id_dup(id1);
3840 if (s->id1 == NULL)
3841 return got_error_from_errno("got_object_id_dup");
3842 } else
3843 s->id1 = NULL;
3845 s->id2 = got_object_id_dup(id2);
3846 if (s->id2 == NULL) {
3847 err = got_error_from_errno("got_object_id_dup");
3848 goto done;
3851 s->f1 = got_opentemp();
3852 if (s->f1 == NULL) {
3853 err = got_error_from_errno("got_opentemp");
3854 goto done;
3857 s->f2 = got_opentemp();
3858 if (s->f2 == NULL) {
3859 err = got_error_from_errno("got_opentemp");
3860 goto done;
3863 s->first_displayed_line = 1;
3864 s->last_displayed_line = view->nlines;
3865 s->diff_context = diff_context;
3866 s->ignore_whitespace = ignore_whitespace;
3867 s->force_text_diff = force_text_diff;
3868 s->log_view = log_view;
3869 s->repo = repo;
3871 STAILQ_INIT(&s->colors);
3872 if (has_colors() && getenv("TOG_COLORS") != NULL) {
3873 err = add_color(&s->colors,
3874 "^-", TOG_COLOR_DIFF_MINUS,
3875 get_color_value("TOG_COLOR_DIFF_MINUS"));
3876 if (err)
3877 goto done;
3878 err = add_color(&s->colors, "^\\+",
3879 TOG_COLOR_DIFF_PLUS,
3880 get_color_value("TOG_COLOR_DIFF_PLUS"));
3881 if (err)
3882 goto done;
3883 err = add_color(&s->colors,
3884 "^@@", TOG_COLOR_DIFF_CHUNK_HEADER,
3885 get_color_value("TOG_COLOR_DIFF_CHUNK_HEADER"));
3886 if (err)
3887 goto done;
3889 err = add_color(&s->colors,
3890 "^(commit [0-9a-f]|parent [0-9]|(blob|file) [-+] |"
3891 "[MDmA] [^ ])", TOG_COLOR_DIFF_META,
3892 get_color_value("TOG_COLOR_DIFF_META"));
3893 if (err)
3894 goto done;
3896 err = add_color(&s->colors,
3897 "^(from|via): ", TOG_COLOR_AUTHOR,
3898 get_color_value("TOG_COLOR_AUTHOR"));
3899 if (err)
3900 goto done;
3902 err = add_color(&s->colors,
3903 "^date: ", TOG_COLOR_DATE,
3904 get_color_value("TOG_COLOR_DATE"));
3905 if (err)
3906 goto done;
3909 if (log_view && view_is_splitscreen(view))
3910 show_log_view(log_view); /* draw vborder */
3911 diff_view_indicate_progress(view);
3913 err = create_diff(s);
3915 view->show = show_diff_view;
3916 view->input = input_diff_view;
3917 view->close = close_diff_view;
3918 view->search_start = search_start_diff_view;
3919 view->search_next = search_next_diff_view;
3920 done:
3921 if (err)
3922 close_diff_view(view);
3923 return err;
3926 static const struct got_error *
3927 show_diff_view(struct tog_view *view)
3929 const struct got_error *err;
3930 struct tog_diff_view_state *s = &view->state.diff;
3931 char *id_str1 = NULL, *id_str2, *header;
3932 const char *label1, *label2;
3934 if (s->id1) {
3935 err = got_object_id_str(&id_str1, s->id1);
3936 if (err)
3937 return err;
3938 label1 = s->label1 ? : id_str1;
3939 } else
3940 label1 = "/dev/null";
3942 err = got_object_id_str(&id_str2, s->id2);
3943 if (err)
3944 return err;
3945 label2 = s->label2 ? : id_str2;
3947 if (asprintf(&header, "diff %s %s", label1, label2) == -1) {
3948 err = got_error_from_errno("asprintf");
3949 free(id_str1);
3950 free(id_str2);
3951 return err;
3953 free(id_str1);
3954 free(id_str2);
3956 err = draw_file(view, header);
3957 free(header);
3958 return err;
3961 static const struct got_error *
3962 set_selected_commit(struct tog_diff_view_state *s,
3963 struct commit_queue_entry *entry)
3965 const struct got_error *err;
3966 const struct got_object_id_queue *parent_ids;
3967 struct got_commit_object *selected_commit;
3968 struct got_object_qid *pid;
3970 free(s->id2);
3971 s->id2 = got_object_id_dup(entry->id);
3972 if (s->id2 == NULL)
3973 return got_error_from_errno("got_object_id_dup");
3975 err = got_object_open_as_commit(&selected_commit, s->repo, entry->id);
3976 if (err)
3977 return err;
3978 parent_ids = got_object_commit_get_parent_ids(selected_commit);
3979 free(s->id1);
3980 pid = STAILQ_FIRST(parent_ids);
3981 s->id1 = pid ? got_object_id_dup(&pid->id) : NULL;
3982 got_object_commit_close(selected_commit);
3983 return NULL;
3986 static const struct got_error *
3987 input_diff_view(struct tog_view **new_view, struct tog_view *view, int ch)
3989 const struct got_error *err = NULL;
3990 struct tog_diff_view_state *s = &view->state.diff;
3991 struct tog_log_view_state *ls;
3992 struct commit_queue_entry *old_selected_entry;
3993 char *line = NULL;
3994 size_t linesize = 0;
3995 ssize_t linelen;
3996 int i, nscroll = view->nlines - 1;
3998 switch (ch) {
3999 case '0':
4000 view->x = 0;
4001 break;
4002 case '$':
4003 view->x = MAX(view->maxx - view->ncols / 3, 0);
4004 break;
4005 case KEY_RIGHT:
4006 case 'l':
4007 if (view->x + view->ncols / 3 < view->maxx)
4008 view->x += 2; /* move two columns right */
4009 break;
4010 case KEY_LEFT:
4011 case 'h':
4012 view->x -= MIN(view->x, 2); /* move two columns back */
4013 break;
4014 case 'a':
4015 case 'w':
4016 if (ch == 'a')
4017 s->force_text_diff = !s->force_text_diff;
4018 if (ch == 'w')
4019 s->ignore_whitespace = !s->ignore_whitespace;
4020 wclear(view->window);
4021 s->first_displayed_line = 1;
4022 s->last_displayed_line = view->nlines;
4023 s->matched_line = 0;
4024 diff_view_indicate_progress(view);
4025 err = create_diff(s);
4026 break;
4027 case 'g':
4028 case KEY_HOME:
4029 s->first_displayed_line = 1;
4030 break;
4031 case 'G':
4032 case KEY_END:
4033 if (s->eof)
4034 break;
4036 s->first_displayed_line = (s->nlines - view->nlines) + 2;
4037 s->eof = 1;
4038 break;
4039 case 'k':
4040 case KEY_UP:
4041 case CTRL('p'):
4042 if (s->first_displayed_line > 1)
4043 s->first_displayed_line--;
4044 break;
4045 case CTRL('u'):
4046 case 'u':
4047 nscroll /= 2;
4048 /* FALL THROUGH */
4049 case KEY_PPAGE:
4050 case CTRL('b'):
4051 if (s->first_displayed_line == 1)
4052 break;
4053 i = 0;
4054 while (i++ < nscroll && s->first_displayed_line > 1)
4055 s->first_displayed_line--;
4056 break;
4057 case 'j':
4058 case KEY_DOWN:
4059 case CTRL('n'):
4060 if (!s->eof)
4061 s->first_displayed_line++;
4062 break;
4063 case CTRL('d'):
4064 case 'd':
4065 nscroll /= 2;
4066 /* FALL THROUGH */
4067 case KEY_NPAGE:
4068 case CTRL('f'):
4069 case ' ':
4070 if (s->eof)
4071 break;
4072 i = 0;
4073 while (!s->eof && i++ < nscroll) {
4074 linelen = getline(&line, &linesize, s->f);
4075 s->first_displayed_line++;
4076 if (linelen == -1) {
4077 if (feof(s->f)) {
4078 s->eof = 1;
4079 } else
4080 err = got_ferror(s->f, GOT_ERR_IO);
4081 break;
4084 free(line);
4085 break;
4086 case '[':
4087 if (s->diff_context > 0) {
4088 s->diff_context--;
4089 s->matched_line = 0;
4090 diff_view_indicate_progress(view);
4091 err = create_diff(s);
4092 if (s->first_displayed_line + view->nlines - 1 >
4093 s->nlines) {
4094 s->first_displayed_line = 1;
4095 s->last_displayed_line = view->nlines;
4098 break;
4099 case ']':
4100 if (s->diff_context < GOT_DIFF_MAX_CONTEXT) {
4101 s->diff_context++;
4102 s->matched_line = 0;
4103 diff_view_indicate_progress(view);
4104 err = create_diff(s);
4106 break;
4107 case '<':
4108 case ',':
4109 if (s->log_view == NULL)
4110 break;
4111 ls = &s->log_view->state.log;
4112 old_selected_entry = ls->selected_entry;
4114 err = input_log_view(NULL, s->log_view, KEY_UP);
4115 if (err)
4116 break;
4118 if (old_selected_entry == ls->selected_entry)
4119 break;
4121 err = set_selected_commit(s, ls->selected_entry);
4122 if (err)
4123 break;
4125 s->first_displayed_line = 1;
4126 s->last_displayed_line = view->nlines;
4127 s->matched_line = 0;
4128 view->x = 0;
4130 diff_view_indicate_progress(view);
4131 err = create_diff(s);
4132 break;
4133 case '>':
4134 case '.':
4135 if (s->log_view == NULL)
4136 break;
4137 ls = &s->log_view->state.log;
4138 old_selected_entry = ls->selected_entry;
4140 err = input_log_view(NULL, s->log_view, KEY_DOWN);
4141 if (err)
4142 break;
4144 if (old_selected_entry == ls->selected_entry)
4145 break;
4147 err = set_selected_commit(s, ls->selected_entry);
4148 if (err)
4149 break;
4151 s->first_displayed_line = 1;
4152 s->last_displayed_line = view->nlines;
4153 s->matched_line = 0;
4154 view->x = 0;
4156 diff_view_indicate_progress(view);
4157 err = create_diff(s);
4158 break;
4159 default:
4160 break;
4163 return err;
4166 static const struct got_error *
4167 cmd_diff(int argc, char *argv[])
4169 const struct got_error *error = NULL;
4170 struct got_repository *repo = NULL;
4171 struct got_worktree *worktree = NULL;
4172 struct got_object_id *id1 = NULL, *id2 = NULL;
4173 char *repo_path = NULL, *cwd = NULL;
4174 char *id_str1 = NULL, *id_str2 = NULL;
4175 char *label1 = NULL, *label2 = NULL;
4176 int diff_context = 3, ignore_whitespace = 0;
4177 int ch, force_text_diff = 0;
4178 const char *errstr;
4179 struct tog_view *view;
4180 int *pack_fds = NULL;
4182 while ((ch = getopt(argc, argv, "aC:r:w")) != -1) {
4183 switch (ch) {
4184 case 'a':
4185 force_text_diff = 1;
4186 break;
4187 case 'C':
4188 diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
4189 &errstr);
4190 if (errstr != NULL)
4191 errx(1, "number of context lines is %s: %s",
4192 errstr, errstr);
4193 break;
4194 case 'r':
4195 repo_path = realpath(optarg, NULL);
4196 if (repo_path == NULL)
4197 return got_error_from_errno2("realpath",
4198 optarg);
4199 got_path_strip_trailing_slashes(repo_path);
4200 break;
4201 case 'w':
4202 ignore_whitespace = 1;
4203 break;
4204 default:
4205 usage_diff();
4206 /* NOTREACHED */
4210 argc -= optind;
4211 argv += optind;
4213 if (argc == 0) {
4214 usage_diff(); /* TODO show local worktree changes */
4215 } else if (argc == 2) {
4216 id_str1 = argv[0];
4217 id_str2 = argv[1];
4218 } else
4219 usage_diff();
4221 error = got_repo_pack_fds_open(&pack_fds);
4222 if (error)
4223 goto done;
4225 if (repo_path == NULL) {
4226 cwd = getcwd(NULL, 0);
4227 if (cwd == NULL)
4228 return got_error_from_errno("getcwd");
4229 error = got_worktree_open(&worktree, cwd);
4230 if (error && error->code != GOT_ERR_NOT_WORKTREE)
4231 goto done;
4232 if (worktree)
4233 repo_path =
4234 strdup(got_worktree_get_repo_path(worktree));
4235 else
4236 repo_path = strdup(cwd);
4237 if (repo_path == NULL) {
4238 error = got_error_from_errno("strdup");
4239 goto done;
4243 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
4244 if (error)
4245 goto done;
4247 init_curses();
4249 error = apply_unveil(got_repo_get_path(repo), NULL);
4250 if (error)
4251 goto done;
4253 error = tog_load_refs(repo, 0);
4254 if (error)
4255 goto done;
4257 error = got_repo_match_object_id(&id1, &label1, id_str1,
4258 GOT_OBJ_TYPE_ANY, &tog_refs, repo);
4259 if (error)
4260 goto done;
4262 error = got_repo_match_object_id(&id2, &label2, id_str2,
4263 GOT_OBJ_TYPE_ANY, &tog_refs, repo);
4264 if (error)
4265 goto done;
4267 view = view_open(0, 0, 0, 0, TOG_VIEW_DIFF);
4268 if (view == NULL) {
4269 error = got_error_from_errno("view_open");
4270 goto done;
4272 error = open_diff_view(view, id1, id2, label1, label2, diff_context,
4273 ignore_whitespace, force_text_diff, NULL, repo);
4274 if (error)
4275 goto done;
4276 error = view_loop(view);
4277 done:
4278 free(label1);
4279 free(label2);
4280 free(repo_path);
4281 free(cwd);
4282 if (repo) {
4283 const struct got_error *close_err = got_repo_close(repo);
4284 if (error == NULL)
4285 error = close_err;
4287 if (worktree)
4288 got_worktree_close(worktree);
4289 if (pack_fds) {
4290 const struct got_error *pack_err =
4291 got_repo_pack_fds_close(pack_fds);
4292 if (error == NULL)
4293 error = pack_err;
4295 tog_free_refs();
4296 return error;
4299 __dead static void
4300 usage_blame(void)
4302 endwin();
4303 fprintf(stderr,
4304 "usage: %s blame [-c commit] [-r repository-path] path\n",
4305 getprogname());
4306 exit(1);
4309 struct tog_blame_line {
4310 int annotated;
4311 struct got_object_id *id;
4314 static const struct got_error *
4315 draw_blame(struct tog_view *view)
4317 struct tog_blame_view_state *s = &view->state.blame;
4318 struct tog_blame *blame = &s->blame;
4319 regmatch_t *regmatch = &view->regmatch;
4320 const struct got_error *err;
4321 int lineno = 0, nprinted = 0;
4322 char *line = NULL;
4323 size_t linesize = 0;
4324 ssize_t linelen;
4325 wchar_t *wline;
4326 int width;
4327 struct tog_blame_line *blame_line;
4328 struct got_object_id *prev_id = NULL;
4329 char *id_str;
4330 struct tog_color *tc;
4332 err = got_object_id_str(&id_str, &s->blamed_commit->id);
4333 if (err)
4334 return err;
4336 rewind(blame->f);
4337 werase(view->window);
4339 if (asprintf(&line, "commit %s", id_str) == -1) {
4340 err = got_error_from_errno("asprintf");
4341 free(id_str);
4342 return err;
4345 err = format_line(&wline, &width, NULL, line, 0, view->ncols, 0, 0);
4346 free(line);
4347 line = NULL;
4348 if (err)
4349 return err;
4350 if (view_needs_focus_indication(view))
4351 wstandout(view->window);
4352 tc = get_color(&s->colors, TOG_COLOR_COMMIT);
4353 if (tc)
4354 wattr_on(view->window,
4355 COLOR_PAIR(tc->colorpair), NULL);
4356 waddwstr(view->window, wline);
4357 if (tc)
4358 wattr_off(view->window,
4359 COLOR_PAIR(tc->colorpair), NULL);
4360 if (view_needs_focus_indication(view))
4361 wstandend(view->window);
4362 free(wline);
4363 wline = NULL;
4364 if (width < view->ncols - 1)
4365 waddch(view->window, '\n');
4367 if (asprintf(&line, "[%d/%d] %s%s",
4368 s->first_displayed_line - 1 + s->selected_line, blame->nlines,
4369 s->blame_complete ? "" : "annotating... ", s->path) == -1) {
4370 free(id_str);
4371 return got_error_from_errno("asprintf");
4373 free(id_str);
4374 err = format_line(&wline, &width, NULL, line, 0, view->ncols, 0, 0);
4375 free(line);
4376 line = NULL;
4377 if (err)
4378 return err;
4379 waddwstr(view->window, wline);
4380 free(wline);
4381 wline = NULL;
4382 if (width < view->ncols - 1)
4383 waddch(view->window, '\n');
4385 s->eof = 0;
4386 view->maxx = 0;
4387 while (nprinted < view->nlines - 2) {
4388 linelen = getline(&line, &linesize, blame->f);
4389 if (linelen == -1) {
4390 if (feof(blame->f)) {
4391 s->eof = 1;
4392 break;
4394 free(line);
4395 return got_ferror(blame->f, GOT_ERR_IO);
4397 if (++lineno < s->first_displayed_line)
4398 continue;
4400 /* Set view->maxx based on full line length. */
4401 err = format_line(&wline, &width, NULL, line, 0, INT_MAX, 9, 1);
4402 if (err) {
4403 free(line);
4404 return err;
4406 free(wline);
4407 wline = NULL;
4408 view->maxx = MAX(view->maxx, width);
4410 if (view->focussed && nprinted == s->selected_line - 1)
4411 wstandout(view->window);
4413 if (blame->nlines > 0) {
4414 blame_line = &blame->lines[lineno - 1];
4415 if (blame_line->annotated && prev_id &&
4416 got_object_id_cmp(prev_id, blame_line->id) == 0 &&
4417 !(view->focussed &&
4418 nprinted == s->selected_line - 1)) {
4419 waddstr(view->window, " ");
4420 } else if (blame_line->annotated) {
4421 char *id_str;
4422 err = got_object_id_str(&id_str,
4423 blame_line->id);
4424 if (err) {
4425 free(line);
4426 return err;
4428 tc = get_color(&s->colors, TOG_COLOR_COMMIT);
4429 if (tc)
4430 wattr_on(view->window,
4431 COLOR_PAIR(tc->colorpair), NULL);
4432 wprintw(view->window, "%.8s", id_str);
4433 if (tc)
4434 wattr_off(view->window,
4435 COLOR_PAIR(tc->colorpair), NULL);
4436 free(id_str);
4437 prev_id = blame_line->id;
4438 } else {
4439 waddstr(view->window, "........");
4440 prev_id = NULL;
4442 } else {
4443 waddstr(view->window, "........");
4444 prev_id = NULL;
4447 if (view->focussed && nprinted == s->selected_line - 1)
4448 wstandend(view->window);
4449 waddstr(view->window, " ");
4451 if (view->ncols <= 9) {
4452 width = 9;
4453 } else if (s->first_displayed_line + nprinted ==
4454 s->matched_line &&
4455 regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
4456 err = add_matched_line(&width, line, view->ncols - 9, 9,
4457 view->window, view->x, regmatch);
4458 if (err) {
4459 free(line);
4460 return err;
4462 width += 9;
4463 } else {
4464 int skip;
4465 err = format_line(&wline, &width, &skip, line,
4466 view->x, view->ncols - 9, 9, 1);
4467 if (err) {
4468 free(line);
4469 return err;
4471 waddwstr(view->window, &wline[skip]);
4472 width += 9;
4473 free(wline);
4474 wline = NULL;
4477 if (width <= view->ncols - 1)
4478 waddch(view->window, '\n');
4479 if (++nprinted == 1)
4480 s->first_displayed_line = lineno;
4482 free(line);
4483 s->last_displayed_line = lineno;
4485 view_vborder(view);
4487 return NULL;
4490 static const struct got_error *
4491 blame_cb(void *arg, int nlines, int lineno,
4492 struct got_commit_object *commit, struct got_object_id *id)
4494 const struct got_error *err = NULL;
4495 struct tog_blame_cb_args *a = arg;
4496 struct tog_blame_line *line;
4497 int errcode;
4499 if (nlines != a->nlines ||
4500 (lineno != -1 && lineno < 1) || lineno > a->nlines)
4501 return got_error(GOT_ERR_RANGE);
4503 errcode = pthread_mutex_lock(&tog_mutex);
4504 if (errcode)
4505 return got_error_set_errno(errcode, "pthread_mutex_lock");
4507 if (*a->quit) { /* user has quit the blame view */
4508 err = got_error(GOT_ERR_ITER_COMPLETED);
4509 goto done;
4512 if (lineno == -1)
4513 goto done; /* no change in this commit */
4515 line = &a->lines[lineno - 1];
4516 if (line->annotated)
4517 goto done;
4519 line->id = got_object_id_dup(id);
4520 if (line->id == NULL) {
4521 err = got_error_from_errno("got_object_id_dup");
4522 goto done;
4524 line->annotated = 1;
4525 done:
4526 errcode = pthread_mutex_unlock(&tog_mutex);
4527 if (errcode)
4528 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
4529 return err;
4532 static void *
4533 blame_thread(void *arg)
4535 const struct got_error *err, *close_err;
4536 struct tog_blame_thread_args *ta = arg;
4537 struct tog_blame_cb_args *a = ta->cb_args;
4538 int errcode;
4540 err = block_signals_used_by_main_thread();
4541 if (err)
4542 return (void *)err;
4544 err = got_blame(ta->path, a->commit_id, ta->repo,
4545 blame_cb, ta->cb_args, ta->cancel_cb, ta->cancel_arg);
4546 if (err && err->code == GOT_ERR_CANCELLED)
4547 err = NULL;
4549 errcode = pthread_mutex_lock(&tog_mutex);
4550 if (errcode)
4551 return (void *)got_error_set_errno(errcode,
4552 "pthread_mutex_lock");
4554 close_err = got_repo_close(ta->repo);
4555 if (err == NULL)
4556 err = close_err;
4557 ta->repo = NULL;
4558 *ta->complete = 1;
4560 errcode = pthread_mutex_unlock(&tog_mutex);
4561 if (errcode && err == NULL)
4562 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
4564 return (void *)err;
4567 static struct got_object_id *
4568 get_selected_commit_id(struct tog_blame_line *lines, int nlines,
4569 int first_displayed_line, int selected_line)
4571 struct tog_blame_line *line;
4573 if (nlines <= 0)
4574 return NULL;
4576 line = &lines[first_displayed_line - 1 + selected_line - 1];
4577 if (!line->annotated)
4578 return NULL;
4580 return line->id;
4583 static const struct got_error *
4584 stop_blame(struct tog_blame *blame)
4586 const struct got_error *err = NULL;
4587 int i;
4589 if (blame->thread) {
4590 int errcode;
4591 errcode = pthread_mutex_unlock(&tog_mutex);
4592 if (errcode)
4593 return got_error_set_errno(errcode,
4594 "pthread_mutex_unlock");
4595 errcode = pthread_join(blame->thread, (void **)&err);
4596 if (errcode)
4597 return got_error_set_errno(errcode, "pthread_join");
4598 errcode = pthread_mutex_lock(&tog_mutex);
4599 if (errcode)
4600 return got_error_set_errno(errcode,
4601 "pthread_mutex_lock");
4602 if (err && err->code == GOT_ERR_ITER_COMPLETED)
4603 err = NULL;
4604 blame->thread = 0; //NULL;
4606 if (blame->thread_args.repo) {
4607 const struct got_error *close_err;
4608 close_err = got_repo_close(blame->thread_args.repo);
4609 if (err == NULL)
4610 err = close_err;
4611 blame->thread_args.repo = NULL;
4613 if (blame->f) {
4614 if (fclose(blame->f) == EOF && err == NULL)
4615 err = got_error_from_errno("fclose");
4616 blame->f = NULL;
4618 if (blame->lines) {
4619 for (i = 0; i < blame->nlines; i++)
4620 free(blame->lines[i].id);
4621 free(blame->lines);
4622 blame->lines = NULL;
4624 free(blame->cb_args.commit_id);
4625 blame->cb_args.commit_id = NULL;
4626 if (blame->pack_fds) {
4627 const struct got_error *pack_err =
4628 got_repo_pack_fds_close(blame->pack_fds);
4629 if (err == NULL)
4630 err = pack_err;
4631 blame->pack_fds = NULL;
4633 return err;
4636 static const struct got_error *
4637 cancel_blame_view(void *arg)
4639 const struct got_error *err = NULL;
4640 int *done = arg;
4641 int errcode;
4643 errcode = pthread_mutex_lock(&tog_mutex);
4644 if (errcode)
4645 return got_error_set_errno(errcode,
4646 "pthread_mutex_unlock");
4648 if (*done)
4649 err = got_error(GOT_ERR_CANCELLED);
4651 errcode = pthread_mutex_unlock(&tog_mutex);
4652 if (errcode)
4653 return got_error_set_errno(errcode,
4654 "pthread_mutex_lock");
4656 return err;
4659 static const struct got_error *
4660 run_blame(struct tog_view *view)
4662 struct tog_blame_view_state *s = &view->state.blame;
4663 struct tog_blame *blame = &s->blame;
4664 const struct got_error *err = NULL;
4665 struct got_commit_object *commit = NULL;
4666 struct got_blob_object *blob = NULL;
4667 struct got_repository *thread_repo = NULL;
4668 struct got_object_id *obj_id = NULL;
4669 int obj_type;
4670 int *pack_fds = NULL;
4672 err = got_object_open_as_commit(&commit, s->repo,
4673 &s->blamed_commit->id);
4674 if (err)
4675 return err;
4677 err = got_object_id_by_path(&obj_id, s->repo, commit, s->path);
4678 if (err)
4679 goto done;
4681 err = got_object_get_type(&obj_type, s->repo, obj_id);
4682 if (err)
4683 goto done;
4685 if (obj_type != GOT_OBJ_TYPE_BLOB) {
4686 err = got_error(GOT_ERR_OBJ_TYPE);
4687 goto done;
4690 err = got_object_open_as_blob(&blob, s->repo, obj_id, 8192);
4691 if (err)
4692 goto done;
4693 blame->f = got_opentemp();
4694 if (blame->f == NULL) {
4695 err = got_error_from_errno("got_opentemp");
4696 goto done;
4698 err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
4699 &blame->line_offsets, blame->f, blob);
4700 if (err)
4701 goto done;
4702 if (blame->nlines == 0) {
4703 s->blame_complete = 1;
4704 goto done;
4707 /* Don't include \n at EOF in the blame line count. */
4708 if (blame->line_offsets[blame->nlines - 1] == blame->filesize)
4709 blame->nlines--;
4711 blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
4712 if (blame->lines == NULL) {
4713 err = got_error_from_errno("calloc");
4714 goto done;
4717 err = got_repo_pack_fds_open(&pack_fds);
4718 if (err)
4719 goto done;
4720 err = got_repo_open(&thread_repo, got_repo_get_path(s->repo), NULL,
4721 pack_fds);
4722 if (err)
4723 goto done;
4725 blame->pack_fds = pack_fds;
4726 blame->cb_args.view = view;
4727 blame->cb_args.lines = blame->lines;
4728 blame->cb_args.nlines = blame->nlines;
4729 blame->cb_args.commit_id = got_object_id_dup(&s->blamed_commit->id);
4730 if (blame->cb_args.commit_id == NULL) {
4731 err = got_error_from_errno("got_object_id_dup");
4732 goto done;
4734 blame->cb_args.quit = &s->done;
4736 blame->thread_args.path = s->path;
4737 blame->thread_args.repo = thread_repo;
4738 blame->thread_args.cb_args = &blame->cb_args;
4739 blame->thread_args.complete = &s->blame_complete;
4740 blame->thread_args.cancel_cb = cancel_blame_view;
4741 blame->thread_args.cancel_arg = &s->done;
4742 s->blame_complete = 0;
4744 if (s->first_displayed_line + view->nlines - 1 > blame->nlines) {
4745 s->first_displayed_line = 1;
4746 s->last_displayed_line = view->nlines;
4747 s->selected_line = 1;
4749 s->matched_line = 0;
4751 done:
4752 if (commit)
4753 got_object_commit_close(commit);
4754 if (blob)
4755 got_object_blob_close(blob);
4756 free(obj_id);
4757 if (err)
4758 stop_blame(blame);
4759 return err;
4762 static const struct got_error *
4763 open_blame_view(struct tog_view *view, char *path,
4764 struct got_object_id *commit_id, struct got_repository *repo)
4766 const struct got_error *err = NULL;
4767 struct tog_blame_view_state *s = &view->state.blame;
4769 STAILQ_INIT(&s->blamed_commits);
4771 s->path = strdup(path);
4772 if (s->path == NULL)
4773 return got_error_from_errno("strdup");
4775 err = got_object_qid_alloc(&s->blamed_commit, commit_id);
4776 if (err) {
4777 free(s->path);
4778 return err;
4781 STAILQ_INSERT_HEAD(&s->blamed_commits, s->blamed_commit, entry);
4782 s->first_displayed_line = 1;
4783 s->last_displayed_line = view->nlines;
4784 s->selected_line = 1;
4785 s->blame_complete = 0;
4786 s->repo = repo;
4787 s->commit_id = commit_id;
4788 memset(&s->blame, 0, sizeof(s->blame));
4790 STAILQ_INIT(&s->colors);
4791 if (has_colors() && getenv("TOG_COLORS") != NULL) {
4792 err = add_color(&s->colors, "^", TOG_COLOR_COMMIT,
4793 get_color_value("TOG_COLOR_COMMIT"));
4794 if (err)
4795 return err;
4798 view->show = show_blame_view;
4799 view->input = input_blame_view;
4800 view->close = close_blame_view;
4801 view->search_start = search_start_blame_view;
4802 view->search_next = search_next_blame_view;
4804 return run_blame(view);
4807 static const struct got_error *
4808 close_blame_view(struct tog_view *view)
4810 const struct got_error *err = NULL;
4811 struct tog_blame_view_state *s = &view->state.blame;
4813 if (s->blame.thread)
4814 err = stop_blame(&s->blame);
4816 while (!STAILQ_EMPTY(&s->blamed_commits)) {
4817 struct got_object_qid *blamed_commit;
4818 blamed_commit = STAILQ_FIRST(&s->blamed_commits);
4819 STAILQ_REMOVE_HEAD(&s->blamed_commits, entry);
4820 got_object_qid_free(blamed_commit);
4823 free(s->path);
4824 free_colors(&s->colors);
4825 return err;
4828 static const struct got_error *
4829 search_start_blame_view(struct tog_view *view)
4831 struct tog_blame_view_state *s = &view->state.blame;
4833 s->matched_line = 0;
4834 return NULL;
4837 static const struct got_error *
4838 search_next_blame_view(struct tog_view *view)
4840 struct tog_blame_view_state *s = &view->state.blame;
4841 const struct got_error *err = NULL;
4842 int lineno;
4843 char *line = NULL;
4844 size_t linesize = 0;
4845 ssize_t linelen;
4847 if (!view->searching) {
4848 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4849 return NULL;
4852 if (s->matched_line) {
4853 if (view->searching == TOG_SEARCH_FORWARD)
4854 lineno = s->matched_line + 1;
4855 else
4856 lineno = s->matched_line - 1;
4857 } else
4858 lineno = s->first_displayed_line - 1 + s->selected_line;
4860 while (1) {
4861 off_t offset;
4863 if (lineno <= 0 || lineno > s->blame.nlines) {
4864 if (s->matched_line == 0) {
4865 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4866 break;
4869 if (view->searching == TOG_SEARCH_FORWARD)
4870 lineno = 1;
4871 else
4872 lineno = s->blame.nlines;
4875 offset = s->blame.line_offsets[lineno - 1];
4876 if (fseeko(s->blame.f, offset, SEEK_SET) != 0) {
4877 free(line);
4878 return got_error_from_errno("fseeko");
4880 linelen = getline(&line, &linesize, s->blame.f);
4881 if (linelen != -1) {
4882 char *exstr;
4883 err = expand_tab(&exstr, line);
4884 if (err)
4885 break;
4886 if (match_line(exstr, &view->regex, 1,
4887 &view->regmatch)) {
4888 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4889 s->matched_line = lineno;
4890 free(exstr);
4891 break;
4893 free(exstr);
4895 if (view->searching == TOG_SEARCH_FORWARD)
4896 lineno++;
4897 else
4898 lineno--;
4900 free(line);
4902 if (s->matched_line) {
4903 s->first_displayed_line = s->matched_line;
4904 s->selected_line = 1;
4907 return err;
4910 static const struct got_error *
4911 show_blame_view(struct tog_view *view)
4913 const struct got_error *err = NULL;
4914 struct tog_blame_view_state *s = &view->state.blame;
4915 int errcode;
4917 if (s->blame.thread == 0 && !s->blame_complete) {
4918 errcode = pthread_create(&s->blame.thread, NULL, blame_thread,
4919 &s->blame.thread_args);
4920 if (errcode)
4921 return got_error_set_errno(errcode, "pthread_create");
4923 halfdelay(1); /* fast refresh while annotating */
4926 if (s->blame_complete)
4927 halfdelay(10); /* disable fast refresh */
4929 err = draw_blame(view);
4931 view_vborder(view);
4932 return err;
4935 static const struct got_error *
4936 input_blame_view(struct tog_view **new_view, struct tog_view *view, int ch)
4938 const struct got_error *err = NULL, *thread_err = NULL;
4939 struct tog_view *diff_view;
4940 struct tog_blame_view_state *s = &view->state.blame;
4941 int begin_x = 0, nscroll = view->nlines - 2;
4943 switch (ch) {
4944 case '0':
4945 view->x = 0;
4946 break;
4947 case '$':
4948 view->x = MAX(view->maxx - view->ncols / 3, 0);
4949 break;
4950 case KEY_RIGHT:
4951 case 'l':
4952 if (view->x + view->ncols / 3 < view->maxx)
4953 view->x += 2; /* move two columns right */
4954 break;
4955 case KEY_LEFT:
4956 case 'h':
4957 view->x -= MIN(view->x, 2); /* move two columns back */
4958 break;
4959 case 'q':
4960 s->done = 1;
4961 break;
4962 case 'g':
4963 case KEY_HOME:
4964 s->selected_line = 1;
4965 s->first_displayed_line = 1;
4966 break;
4967 case 'G':
4968 case KEY_END:
4969 if (s->blame.nlines < view->nlines - 2) {
4970 s->selected_line = s->blame.nlines;
4971 s->first_displayed_line = 1;
4972 } else {
4973 s->selected_line = view->nlines - 2;
4974 s->first_displayed_line = s->blame.nlines -
4975 (view->nlines - 3);
4977 break;
4978 case 'k':
4979 case KEY_UP:
4980 case CTRL('p'):
4981 if (s->selected_line > 1)
4982 s->selected_line--;
4983 else if (s->selected_line == 1 &&
4984 s->first_displayed_line > 1)
4985 s->first_displayed_line--;
4986 break;
4987 case CTRL('u'):
4988 case 'u':
4989 nscroll /= 2;
4990 /* FALL THROUGH */
4991 case KEY_PPAGE:
4992 case CTRL('b'):
4993 if (s->first_displayed_line == 1) {
4994 s->selected_line = MAX(1, s->selected_line - nscroll);
4995 break;
4997 if (s->first_displayed_line > nscroll)
4998 s->first_displayed_line -= nscroll;
4999 else
5000 s->first_displayed_line = 1;
5001 break;
5002 case 'j':
5003 case KEY_DOWN:
5004 case CTRL('n'):
5005 if (s->selected_line < view->nlines - 2 &&
5006 s->first_displayed_line +
5007 s->selected_line <= s->blame.nlines)
5008 s->selected_line++;
5009 else if (s->last_displayed_line <
5010 s->blame.nlines)
5011 s->first_displayed_line++;
5012 break;
5013 case 'b':
5014 case 'p': {
5015 struct got_object_id *id = NULL;
5016 id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
5017 s->first_displayed_line, s->selected_line);
5018 if (id == NULL)
5019 break;
5020 if (ch == 'p') {
5021 struct got_commit_object *commit, *pcommit;
5022 struct got_object_qid *pid;
5023 struct got_object_id *blob_id = NULL;
5024 int obj_type;
5025 err = got_object_open_as_commit(&commit,
5026 s->repo, id);
5027 if (err)
5028 break;
5029 pid = STAILQ_FIRST(
5030 got_object_commit_get_parent_ids(commit));
5031 if (pid == NULL) {
5032 got_object_commit_close(commit);
5033 break;
5035 /* Check if path history ends here. */
5036 err = got_object_open_as_commit(&pcommit,
5037 s->repo, &pid->id);
5038 if (err)
5039 break;
5040 err = got_object_id_by_path(&blob_id, s->repo,
5041 pcommit, s->path);
5042 got_object_commit_close(pcommit);
5043 if (err) {
5044 if (err->code == GOT_ERR_NO_TREE_ENTRY)
5045 err = NULL;
5046 got_object_commit_close(commit);
5047 break;
5049 err = got_object_get_type(&obj_type, s->repo,
5050 blob_id);
5051 free(blob_id);
5052 /* Can't blame non-blob type objects. */
5053 if (obj_type != GOT_OBJ_TYPE_BLOB) {
5054 got_object_commit_close(commit);
5055 break;
5057 err = got_object_qid_alloc(&s->blamed_commit,
5058 &pid->id);
5059 got_object_commit_close(commit);
5060 } else {
5061 if (got_object_id_cmp(id,
5062 &s->blamed_commit->id) == 0)
5063 break;
5064 err = got_object_qid_alloc(&s->blamed_commit,
5065 id);
5067 if (err)
5068 break;
5069 s->done = 1;
5070 thread_err = stop_blame(&s->blame);
5071 s->done = 0;
5072 if (thread_err)
5073 break;
5074 STAILQ_INSERT_HEAD(&s->blamed_commits,
5075 s->blamed_commit, entry);
5076 err = run_blame(view);
5077 if (err)
5078 break;
5079 break;
5081 case 'B': {
5082 struct got_object_qid *first;
5083 first = STAILQ_FIRST(&s->blamed_commits);
5084 if (!got_object_id_cmp(&first->id, s->commit_id))
5085 break;
5086 s->done = 1;
5087 thread_err = stop_blame(&s->blame);
5088 s->done = 0;
5089 if (thread_err)
5090 break;
5091 STAILQ_REMOVE_HEAD(&s->blamed_commits, entry);
5092 got_object_qid_free(s->blamed_commit);
5093 s->blamed_commit =
5094 STAILQ_FIRST(&s->blamed_commits);
5095 err = run_blame(view);
5096 if (err)
5097 break;
5098 break;
5100 case KEY_ENTER:
5101 case '\r': {
5102 struct got_object_id *id = NULL;
5103 struct got_object_qid *pid;
5104 struct got_commit_object *commit = NULL;
5105 id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
5106 s->first_displayed_line, s->selected_line);
5107 if (id == NULL)
5108 break;
5109 err = got_object_open_as_commit(&commit, s->repo, id);
5110 if (err)
5111 break;
5112 pid = STAILQ_FIRST(
5113 got_object_commit_get_parent_ids(commit));
5114 if (view_is_parent_view(view))
5115 begin_x = view_split_begin_x(view->begin_x);
5116 diff_view = view_open(0, 0, 0, begin_x, TOG_VIEW_DIFF);
5117 if (diff_view == NULL) {
5118 got_object_commit_close(commit);
5119 err = got_error_from_errno("view_open");
5120 break;
5122 err = open_diff_view(diff_view, pid ? &pid->id : NULL,
5123 id, NULL, NULL, 3, 0, 0, NULL, s->repo);
5124 got_object_commit_close(commit);
5125 if (err) {
5126 view_close(diff_view);
5127 break;
5129 view->focussed = 0;
5130 diff_view->focussed = 1;
5131 if (view_is_parent_view(view)) {
5132 err = view_close_child(view);
5133 if (err)
5134 break;
5135 err = view_set_child(view, diff_view);
5136 if (err)
5137 break;
5138 view->focus_child = 1;
5139 } else
5140 *new_view = diff_view;
5141 if (err)
5142 break;
5143 break;
5145 case CTRL('d'):
5146 case 'd':
5147 nscroll /= 2;
5148 /* FALL THROUGH */
5149 case KEY_NPAGE:
5150 case CTRL('f'):
5151 case ' ':
5152 if (s->last_displayed_line >= s->blame.nlines &&
5153 s->selected_line >= MIN(s->blame.nlines,
5154 view->nlines - 2)) {
5155 break;
5157 if (s->last_displayed_line >= s->blame.nlines &&
5158 s->selected_line < view->nlines - 2) {
5159 s->selected_line +=
5160 MIN(nscroll, s->last_displayed_line -
5161 s->first_displayed_line - s->selected_line + 1);
5163 if (s->last_displayed_line + nscroll <= s->blame.nlines)
5164 s->first_displayed_line += nscroll;
5165 else
5166 s->first_displayed_line =
5167 s->blame.nlines - (view->nlines - 3);
5168 break;
5169 case KEY_RESIZE:
5170 if (s->selected_line > view->nlines - 2) {
5171 s->selected_line = MIN(s->blame.nlines,
5172 view->nlines - 2);
5174 break;
5175 default:
5176 break;
5178 return thread_err ? thread_err : err;
5181 static const struct got_error *
5182 cmd_blame(int argc, char *argv[])
5184 const struct got_error *error;
5185 struct got_repository *repo = NULL;
5186 struct got_worktree *worktree = NULL;
5187 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
5188 char *link_target = NULL;
5189 struct got_object_id *commit_id = NULL;
5190 struct got_commit_object *commit = NULL;
5191 char *commit_id_str = NULL;
5192 int ch;
5193 struct tog_view *view;
5194 int *pack_fds = NULL;
5196 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
5197 switch (ch) {
5198 case 'c':
5199 commit_id_str = optarg;
5200 break;
5201 case 'r':
5202 repo_path = realpath(optarg, NULL);
5203 if (repo_path == NULL)
5204 return got_error_from_errno2("realpath",
5205 optarg);
5206 break;
5207 default:
5208 usage_blame();
5209 /* NOTREACHED */
5213 argc -= optind;
5214 argv += optind;
5216 if (argc != 1)
5217 usage_blame();
5219 error = got_repo_pack_fds_open(&pack_fds);
5220 if (error != NULL)
5221 goto done;
5223 if (repo_path == NULL) {
5224 cwd = getcwd(NULL, 0);
5225 if (cwd == NULL)
5226 return got_error_from_errno("getcwd");
5227 error = got_worktree_open(&worktree, cwd);
5228 if (error && error->code != GOT_ERR_NOT_WORKTREE)
5229 goto done;
5230 if (worktree)
5231 repo_path =
5232 strdup(got_worktree_get_repo_path(worktree));
5233 else
5234 repo_path = strdup(cwd);
5235 if (repo_path == NULL) {
5236 error = got_error_from_errno("strdup");
5237 goto done;
5241 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
5242 if (error != NULL)
5243 goto done;
5245 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv, repo,
5246 worktree);
5247 if (error)
5248 goto done;
5250 init_curses();
5252 error = apply_unveil(got_repo_get_path(repo), NULL);
5253 if (error)
5254 goto done;
5256 error = tog_load_refs(repo, 0);
5257 if (error)
5258 goto done;
5260 if (commit_id_str == NULL) {
5261 struct got_reference *head_ref;
5262 error = got_ref_open(&head_ref, repo, worktree ?
5263 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD, 0);
5264 if (error != NULL)
5265 goto done;
5266 error = got_ref_resolve(&commit_id, repo, head_ref);
5267 got_ref_close(head_ref);
5268 } else {
5269 error = got_repo_match_object_id(&commit_id, NULL,
5270 commit_id_str, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
5272 if (error != NULL)
5273 goto done;
5275 view = view_open(0, 0, 0, 0, TOG_VIEW_BLAME);
5276 if (view == NULL) {
5277 error = got_error_from_errno("view_open");
5278 goto done;
5281 error = got_object_open_as_commit(&commit, repo, commit_id);
5282 if (error)
5283 goto done;
5285 error = got_object_resolve_symlinks(&link_target, in_repo_path,
5286 commit, repo);
5287 if (error)
5288 goto done;
5290 error = open_blame_view(view, link_target ? link_target : in_repo_path,
5291 commit_id, repo);
5292 if (error)
5293 goto done;
5294 if (worktree) {
5295 /* Release work tree lock. */
5296 got_worktree_close(worktree);
5297 worktree = NULL;
5299 error = view_loop(view);
5300 done:
5301 free(repo_path);
5302 free(in_repo_path);
5303 free(link_target);
5304 free(cwd);
5305 free(commit_id);
5306 if (commit)
5307 got_object_commit_close(commit);
5308 if (worktree)
5309 got_worktree_close(worktree);
5310 if (repo) {
5311 const struct got_error *close_err = got_repo_close(repo);
5312 if (error == NULL)
5313 error = close_err;
5315 if (pack_fds) {
5316 const struct got_error *pack_err =
5317 got_repo_pack_fds_close(pack_fds);
5318 if (error == NULL)
5319 error = pack_err;
5321 tog_free_refs();
5322 return error;
5325 static const struct got_error *
5326 draw_tree_entries(struct tog_view *view, const char *parent_path)
5328 struct tog_tree_view_state *s = &view->state.tree;
5329 const struct got_error *err = NULL;
5330 struct got_tree_entry *te;
5331 wchar_t *wline;
5332 struct tog_color *tc;
5333 int width, n, i, nentries;
5334 int limit = view->nlines;
5336 s->ndisplayed = 0;
5338 werase(view->window);
5340 if (limit == 0)
5341 return NULL;
5343 err = format_line(&wline, &width, NULL, s->tree_label, 0, view->ncols,
5344 0, 0);
5345 if (err)
5346 return err;
5347 if (view_needs_focus_indication(view))
5348 wstandout(view->window);
5349 tc = get_color(&s->colors, TOG_COLOR_COMMIT);
5350 if (tc)
5351 wattr_on(view->window,
5352 COLOR_PAIR(tc->colorpair), NULL);
5353 waddwstr(view->window, wline);
5354 if (tc)
5355 wattr_off(view->window,
5356 COLOR_PAIR(tc->colorpair), NULL);
5357 if (view_needs_focus_indication(view))
5358 wstandend(view->window);
5359 free(wline);
5360 wline = NULL;
5361 if (width < view->ncols - 1)
5362 waddch(view->window, '\n');
5363 if (--limit <= 0)
5364 return NULL;
5365 err = format_line(&wline, &width, NULL, parent_path, 0, view->ncols,
5366 0, 0);
5367 if (err)
5368 return err;
5369 waddwstr(view->window, wline);
5370 free(wline);
5371 wline = NULL;
5372 if (width < view->ncols - 1)
5373 waddch(view->window, '\n');
5374 if (--limit <= 0)
5375 return NULL;
5376 waddch(view->window, '\n');
5377 if (--limit <= 0)
5378 return NULL;
5380 if (s->first_displayed_entry == NULL) {
5381 te = got_object_tree_get_first_entry(s->tree);
5382 if (s->selected == 0) {
5383 if (view->focussed)
5384 wstandout(view->window);
5385 s->selected_entry = NULL;
5387 waddstr(view->window, " ..\n"); /* parent directory */
5388 if (s->selected == 0 && view->focussed)
5389 wstandend(view->window);
5390 s->ndisplayed++;
5391 if (--limit <= 0)
5392 return NULL;
5393 n = 1;
5394 } else {
5395 n = 0;
5396 te = s->first_displayed_entry;
5399 nentries = got_object_tree_get_nentries(s->tree);
5400 for (i = got_tree_entry_get_index(te); i < nentries; i++) {
5401 char *line = NULL, *id_str = NULL, *link_target = NULL;
5402 const char *modestr = "";
5403 mode_t mode;
5405 te = got_object_tree_get_entry(s->tree, i);
5406 mode = got_tree_entry_get_mode(te);
5408 if (s->show_ids) {
5409 err = got_object_id_str(&id_str,
5410 got_tree_entry_get_id(te));
5411 if (err)
5412 return got_error_from_errno(
5413 "got_object_id_str");
5415 if (got_object_tree_entry_is_submodule(te))
5416 modestr = "$";
5417 else if (S_ISLNK(mode)) {
5418 int i;
5420 err = got_tree_entry_get_symlink_target(&link_target,
5421 te, s->repo);
5422 if (err) {
5423 free(id_str);
5424 return err;
5426 for (i = 0; i < strlen(link_target); i++) {
5427 if (!isprint((unsigned char)link_target[i]))
5428 link_target[i] = '?';
5430 modestr = "@";
5432 else if (S_ISDIR(mode))
5433 modestr = "/";
5434 else if (mode & S_IXUSR)
5435 modestr = "*";
5436 if (asprintf(&line, "%s %s%s%s%s", id_str ? id_str : "",
5437 got_tree_entry_get_name(te), modestr,
5438 link_target ? " -> ": "",
5439 link_target ? link_target : "") == -1) {
5440 free(id_str);
5441 free(link_target);
5442 return got_error_from_errno("asprintf");
5444 free(id_str);
5445 free(link_target);
5446 err = format_line(&wline, &width, NULL, line, 0, view->ncols,
5447 0, 0);
5448 if (err) {
5449 free(line);
5450 break;
5452 if (n == s->selected) {
5453 if (view->focussed)
5454 wstandout(view->window);
5455 s->selected_entry = te;
5457 tc = match_color(&s->colors, line);
5458 if (tc)
5459 wattr_on(view->window,
5460 COLOR_PAIR(tc->colorpair), NULL);
5461 waddwstr(view->window, wline);
5462 if (tc)
5463 wattr_off(view->window,
5464 COLOR_PAIR(tc->colorpair), NULL);
5465 if (width < view->ncols - 1)
5466 waddch(view->window, '\n');
5467 if (n == s->selected && view->focussed)
5468 wstandend(view->window);
5469 free(line);
5470 free(wline);
5471 wline = NULL;
5472 n++;
5473 s->ndisplayed++;
5474 s->last_displayed_entry = te;
5475 if (--limit <= 0)
5476 break;
5479 return err;
5482 static void
5483 tree_scroll_up(struct tog_tree_view_state *s, int maxscroll)
5485 struct got_tree_entry *te;
5486 int isroot = s->tree == s->root;
5487 int i = 0;
5489 if (s->first_displayed_entry == NULL)
5490 return;
5492 te = got_tree_entry_get_prev(s->tree, s->first_displayed_entry);
5493 while (i++ < maxscroll) {
5494 if (te == NULL) {
5495 if (!isroot)
5496 s->first_displayed_entry = NULL;
5497 break;
5499 s->first_displayed_entry = te;
5500 te = got_tree_entry_get_prev(s->tree, te);
5504 static void
5505 tree_scroll_down(struct tog_tree_view_state *s, int maxscroll)
5507 struct got_tree_entry *next, *last;
5508 int n = 0;
5510 if (s->first_displayed_entry)
5511 next = got_tree_entry_get_next(s->tree,
5512 s->first_displayed_entry);
5513 else
5514 next = got_object_tree_get_first_entry(s->tree);
5516 last = s->last_displayed_entry;
5517 while (next && last && n++ < maxscroll) {
5518 last = got_tree_entry_get_next(s->tree, last);
5519 if (last) {
5520 s->first_displayed_entry = next;
5521 next = got_tree_entry_get_next(s->tree, next);
5526 static const struct got_error *
5527 tree_entry_path(char **path, struct tog_parent_trees *parents,
5528 struct got_tree_entry *te)
5530 const struct got_error *err = NULL;
5531 struct tog_parent_tree *pt;
5532 size_t len = 2; /* for leading slash and NUL */
5534 TAILQ_FOREACH(pt, parents, entry)
5535 len += strlen(got_tree_entry_get_name(pt->selected_entry))
5536 + 1 /* slash */;
5537 if (te)
5538 len += strlen(got_tree_entry_get_name(te));
5540 *path = calloc(1, len);
5541 if (path == NULL)
5542 return got_error_from_errno("calloc");
5544 (*path)[0] = '/';
5545 pt = TAILQ_LAST(parents, tog_parent_trees);
5546 while (pt) {
5547 const char *name = got_tree_entry_get_name(pt->selected_entry);
5548 if (strlcat(*path, name, len) >= len) {
5549 err = got_error(GOT_ERR_NO_SPACE);
5550 goto done;
5552 if (strlcat(*path, "/", len) >= len) {
5553 err = got_error(GOT_ERR_NO_SPACE);
5554 goto done;
5556 pt = TAILQ_PREV(pt, tog_parent_trees, entry);
5558 if (te) {
5559 if (strlcat(*path, got_tree_entry_get_name(te), len) >= len) {
5560 err = got_error(GOT_ERR_NO_SPACE);
5561 goto done;
5564 done:
5565 if (err) {
5566 free(*path);
5567 *path = NULL;
5569 return err;
5572 static const struct got_error *
5573 blame_tree_entry(struct tog_view **new_view, int begin_x,
5574 struct got_tree_entry *te, struct tog_parent_trees *parents,
5575 struct got_object_id *commit_id, struct got_repository *repo)
5577 const struct got_error *err = NULL;
5578 char *path;
5579 struct tog_view *blame_view;
5581 *new_view = NULL;
5583 err = tree_entry_path(&path, parents, te);
5584 if (err)
5585 return err;
5587 blame_view = view_open(0, 0, 0, begin_x, TOG_VIEW_BLAME);
5588 if (blame_view == NULL) {
5589 err = got_error_from_errno("view_open");
5590 goto done;
5593 err = open_blame_view(blame_view, path, commit_id, repo);
5594 if (err) {
5595 if (err->code == GOT_ERR_CANCELLED)
5596 err = NULL;
5597 view_close(blame_view);
5598 } else
5599 *new_view = blame_view;
5600 done:
5601 free(path);
5602 return err;
5605 static const struct got_error *
5606 log_selected_tree_entry(struct tog_view **new_view, int begin_x,
5607 struct tog_tree_view_state *s)
5609 struct tog_view *log_view;
5610 const struct got_error *err = NULL;
5611 char *path;
5613 *new_view = NULL;
5615 log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
5616 if (log_view == NULL)
5617 return got_error_from_errno("view_open");
5619 err = tree_entry_path(&path, &s->parents, s->selected_entry);
5620 if (err)
5621 return err;
5623 err = open_log_view(log_view, s->commit_id, s->repo, s->head_ref_name,
5624 path, 0);
5625 if (err)
5626 view_close(log_view);
5627 else
5628 *new_view = log_view;
5629 free(path);
5630 return err;
5633 static const struct got_error *
5634 open_tree_view(struct tog_view *view, struct got_object_id *commit_id,
5635 const char *head_ref_name, struct got_repository *repo)
5637 const struct got_error *err = NULL;
5638 char *commit_id_str = NULL;
5639 struct tog_tree_view_state *s = &view->state.tree;
5640 struct got_commit_object *commit = NULL;
5642 TAILQ_INIT(&s->parents);
5643 STAILQ_INIT(&s->colors);
5645 s->commit_id = got_object_id_dup(commit_id);
5646 if (s->commit_id == NULL)
5647 return got_error_from_errno("got_object_id_dup");
5649 err = got_object_open_as_commit(&commit, repo, commit_id);
5650 if (err)
5651 goto done;
5654 * The root is opened here and will be closed when the view is closed.
5655 * Any visited subtrees and their path-wise parents are opened and
5656 * closed on demand.
5658 err = got_object_open_as_tree(&s->root, repo,
5659 got_object_commit_get_tree_id(commit));
5660 if (err)
5661 goto done;
5662 s->tree = s->root;
5664 err = got_object_id_str(&commit_id_str, commit_id);
5665 if (err != NULL)
5666 goto done;
5668 if (asprintf(&s->tree_label, "commit %s", commit_id_str) == -1) {
5669 err = got_error_from_errno("asprintf");
5670 goto done;
5673 s->first_displayed_entry = got_object_tree_get_entry(s->tree, 0);
5674 s->selected_entry = got_object_tree_get_entry(s->tree, 0);
5675 if (head_ref_name) {
5676 s->head_ref_name = strdup(head_ref_name);
5677 if (s->head_ref_name == NULL) {
5678 err = got_error_from_errno("strdup");
5679 goto done;
5682 s->repo = repo;
5684 if (has_colors() && getenv("TOG_COLORS") != NULL) {
5685 err = add_color(&s->colors, "\\$$",
5686 TOG_COLOR_TREE_SUBMODULE,
5687 get_color_value("TOG_COLOR_TREE_SUBMODULE"));
5688 if (err)
5689 goto done;
5690 err = add_color(&s->colors, "@$", TOG_COLOR_TREE_SYMLINK,
5691 get_color_value("TOG_COLOR_TREE_SYMLINK"));
5692 if (err)
5693 goto done;
5694 err = add_color(&s->colors, "/$",
5695 TOG_COLOR_TREE_DIRECTORY,
5696 get_color_value("TOG_COLOR_TREE_DIRECTORY"));
5697 if (err)
5698 goto done;
5700 err = add_color(&s->colors, "\\*$",
5701 TOG_COLOR_TREE_EXECUTABLE,
5702 get_color_value("TOG_COLOR_TREE_EXECUTABLE"));
5703 if (err)
5704 goto done;
5706 err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
5707 get_color_value("TOG_COLOR_COMMIT"));
5708 if (err)
5709 goto done;
5712 view->show = show_tree_view;
5713 view->input = input_tree_view;
5714 view->close = close_tree_view;
5715 view->search_start = search_start_tree_view;
5716 view->search_next = search_next_tree_view;
5717 done:
5718 free(commit_id_str);
5719 if (commit)
5720 got_object_commit_close(commit);
5721 if (err)
5722 close_tree_view(view);
5723 return err;
5726 static const struct got_error *
5727 close_tree_view(struct tog_view *view)
5729 struct tog_tree_view_state *s = &view->state.tree;
5731 free_colors(&s->colors);
5732 free(s->tree_label);
5733 s->tree_label = NULL;
5734 free(s->commit_id);
5735 s->commit_id = NULL;
5736 free(s->head_ref_name);
5737 s->head_ref_name = NULL;
5738 while (!TAILQ_EMPTY(&s->parents)) {
5739 struct tog_parent_tree *parent;
5740 parent = TAILQ_FIRST(&s->parents);
5741 TAILQ_REMOVE(&s->parents, parent, entry);
5742 if (parent->tree != s->root)
5743 got_object_tree_close(parent->tree);
5744 free(parent);
5747 if (s->tree != NULL && s->tree != s->root)
5748 got_object_tree_close(s->tree);
5749 if (s->root)
5750 got_object_tree_close(s->root);
5751 return NULL;
5754 static const struct got_error *
5755 search_start_tree_view(struct tog_view *view)
5757 struct tog_tree_view_state *s = &view->state.tree;
5759 s->matched_entry = NULL;
5760 return NULL;
5763 static int
5764 match_tree_entry(struct got_tree_entry *te, regex_t *regex)
5766 regmatch_t regmatch;
5768 return regexec(regex, got_tree_entry_get_name(te), 1, &regmatch,
5769 0) == 0;
5772 static const struct got_error *
5773 search_next_tree_view(struct tog_view *view)
5775 struct tog_tree_view_state *s = &view->state.tree;
5776 struct got_tree_entry *te = NULL;
5778 if (!view->searching) {
5779 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5780 return NULL;
5783 if (s->matched_entry) {
5784 if (view->searching == TOG_SEARCH_FORWARD) {
5785 if (s->selected_entry)
5786 te = got_tree_entry_get_next(s->tree,
5787 s->selected_entry);
5788 else
5789 te = got_object_tree_get_first_entry(s->tree);
5790 } else {
5791 if (s->selected_entry == NULL)
5792 te = got_object_tree_get_last_entry(s->tree);
5793 else
5794 te = got_tree_entry_get_prev(s->tree,
5795 s->selected_entry);
5797 } else {
5798 if (s->selected_entry)
5799 te = s->selected_entry;
5800 else if (view->searching == TOG_SEARCH_FORWARD)
5801 te = got_object_tree_get_first_entry(s->tree);
5802 else
5803 te = got_object_tree_get_last_entry(s->tree);
5806 while (1) {
5807 if (te == NULL) {
5808 if (s->matched_entry == NULL) {
5809 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5810 return NULL;
5812 if (view->searching == TOG_SEARCH_FORWARD)
5813 te = got_object_tree_get_first_entry(s->tree);
5814 else
5815 te = got_object_tree_get_last_entry(s->tree);
5818 if (match_tree_entry(te, &view->regex)) {
5819 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5820 s->matched_entry = te;
5821 break;
5824 if (view->searching == TOG_SEARCH_FORWARD)
5825 te = got_tree_entry_get_next(s->tree, te);
5826 else
5827 te = got_tree_entry_get_prev(s->tree, te);
5830 if (s->matched_entry) {
5831 s->first_displayed_entry = s->matched_entry;
5832 s->selected = 0;
5835 return NULL;
5838 static const struct got_error *
5839 show_tree_view(struct tog_view *view)
5841 const struct got_error *err = NULL;
5842 struct tog_tree_view_state *s = &view->state.tree;
5843 char *parent_path;
5845 err = tree_entry_path(&parent_path, &s->parents, NULL);
5846 if (err)
5847 return err;
5849 err = draw_tree_entries(view, parent_path);
5850 free(parent_path);
5852 view_vborder(view);
5853 return err;
5856 static const struct got_error *
5857 input_tree_view(struct tog_view **new_view, struct tog_view *view, int ch)
5859 const struct got_error *err = NULL;
5860 struct tog_tree_view_state *s = &view->state.tree;
5861 struct tog_view *log_view, *ref_view;
5862 struct got_tree_entry *te;
5863 int begin_x = 0, n, nscroll = view->nlines - 3;
5865 switch (ch) {
5866 case 'i':
5867 s->show_ids = !s->show_ids;
5868 break;
5869 case 'l':
5870 if (!s->selected_entry)
5871 break;
5872 if (view_is_parent_view(view))
5873 begin_x = view_split_begin_x(view->begin_x);
5874 err = log_selected_tree_entry(&log_view, begin_x, s);
5875 view->focussed = 0;
5876 log_view->focussed = 1;
5877 if (view_is_parent_view(view)) {
5878 err = view_close_child(view);
5879 if (err)
5880 return err;
5881 err = view_set_child(view, log_view);
5882 if (err)
5883 return err;
5884 view->focus_child = 1;
5885 } else
5886 *new_view = log_view;
5887 break;
5888 case 'r':
5889 if (view_is_parent_view(view))
5890 begin_x = view_split_begin_x(view->begin_x);
5891 ref_view = view_open(view->nlines, view->ncols,
5892 view->begin_y, begin_x, TOG_VIEW_REF);
5893 if (ref_view == NULL)
5894 return got_error_from_errno("view_open");
5895 err = open_ref_view(ref_view, s->repo);
5896 if (err) {
5897 view_close(ref_view);
5898 return err;
5900 view->focussed = 0;
5901 ref_view->focussed = 1;
5902 if (view_is_parent_view(view)) {
5903 err = view_close_child(view);
5904 if (err)
5905 return err;
5906 err = view_set_child(view, ref_view);
5907 if (err)
5908 return err;
5909 view->focus_child = 1;
5910 } else
5911 *new_view = ref_view;
5912 break;
5913 case 'g':
5914 case KEY_HOME:
5915 s->selected = 0;
5916 if (s->tree == s->root)
5917 s->first_displayed_entry =
5918 got_object_tree_get_first_entry(s->tree);
5919 else
5920 s->first_displayed_entry = NULL;
5921 break;
5922 case 'G':
5923 case KEY_END:
5924 s->selected = 0;
5925 te = got_object_tree_get_last_entry(s->tree);
5926 for (n = 0; n < view->nlines - 3; n++) {
5927 if (te == NULL) {
5928 if(s->tree != s->root) {
5929 s->first_displayed_entry = NULL;
5930 n++;
5932 break;
5934 s->first_displayed_entry = te;
5935 te = got_tree_entry_get_prev(s->tree, te);
5937 if (n > 0)
5938 s->selected = n - 1;
5939 break;
5940 case 'k':
5941 case KEY_UP:
5942 case CTRL('p'):
5943 if (s->selected > 0) {
5944 s->selected--;
5945 break;
5947 tree_scroll_up(s, 1);
5948 break;
5949 case CTRL('u'):
5950 case 'u':
5951 nscroll /= 2;
5952 /* FALL THROUGH */
5953 case KEY_PPAGE:
5954 case CTRL('b'):
5955 if (s->tree == s->root) {
5956 if (got_object_tree_get_first_entry(s->tree) ==
5957 s->first_displayed_entry)
5958 s->selected -= MIN(s->selected, nscroll);
5959 } else {
5960 if (s->first_displayed_entry == NULL)
5961 s->selected -= MIN(s->selected, nscroll);
5963 tree_scroll_up(s, MAX(0, nscroll));
5964 break;
5965 case 'j':
5966 case KEY_DOWN:
5967 case CTRL('n'):
5968 if (s->selected < s->ndisplayed - 1) {
5969 s->selected++;
5970 break;
5972 if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
5973 == NULL)
5974 /* can't scroll any further */
5975 break;
5976 tree_scroll_down(s, 1);
5977 break;
5978 case CTRL('d'):
5979 case 'd':
5980 nscroll /= 2;
5981 /* FALL THROUGH */
5982 case KEY_NPAGE:
5983 case CTRL('f'):
5984 if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
5985 == NULL) {
5986 /* can't scroll any further; move cursor down */
5987 if (s->selected < s->ndisplayed - 1)
5988 s->selected += MIN(nscroll,
5989 s->ndisplayed - s->selected - 1);
5990 break;
5992 tree_scroll_down(s, nscroll);
5993 break;
5994 case KEY_ENTER:
5995 case '\r':
5996 case KEY_BACKSPACE:
5997 if (s->selected_entry == NULL || ch == KEY_BACKSPACE) {
5998 struct tog_parent_tree *parent;
5999 /* user selected '..' */
6000 if (s->tree == s->root)
6001 break;
6002 parent = TAILQ_FIRST(&s->parents);
6003 TAILQ_REMOVE(&s->parents, parent,
6004 entry);
6005 got_object_tree_close(s->tree);
6006 s->tree = parent->tree;
6007 s->first_displayed_entry =
6008 parent->first_displayed_entry;
6009 s->selected_entry =
6010 parent->selected_entry;
6011 s->selected = parent->selected;
6012 free(parent);
6013 } else if (S_ISDIR(got_tree_entry_get_mode(
6014 s->selected_entry))) {
6015 struct got_tree_object *subtree;
6016 err = got_object_open_as_tree(&subtree, s->repo,
6017 got_tree_entry_get_id(s->selected_entry));
6018 if (err)
6019 break;
6020 err = tree_view_visit_subtree(s, subtree);
6021 if (err) {
6022 got_object_tree_close(subtree);
6023 break;
6025 } else if (S_ISREG(got_tree_entry_get_mode(
6026 s->selected_entry))) {
6027 struct tog_view *blame_view;
6028 int begin_x = view_is_parent_view(view) ?
6029 view_split_begin_x(view->begin_x) : 0;
6031 err = blame_tree_entry(&blame_view, begin_x,
6032 s->selected_entry, &s->parents,
6033 s->commit_id, s->repo);
6034 if (err)
6035 break;
6036 view->focussed = 0;
6037 blame_view->focussed = 1;
6038 if (view_is_parent_view(view)) {
6039 err = view_close_child(view);
6040 if (err)
6041 return err;
6042 err = view_set_child(view, blame_view);
6043 if (err)
6044 return err;
6045 view->focus_child = 1;
6046 } else
6047 *new_view = blame_view;
6049 break;
6050 case KEY_RESIZE:
6051 if (view->nlines >= 4 && s->selected >= view->nlines - 3)
6052 s->selected = view->nlines - 4;
6053 break;
6054 default:
6055 break;
6058 return err;
6061 __dead static void
6062 usage_tree(void)
6064 endwin();
6065 fprintf(stderr,
6066 "usage: %s tree [-c commit] [-r repository-path] [path]\n",
6067 getprogname());
6068 exit(1);
6071 static const struct got_error *
6072 cmd_tree(int argc, char *argv[])
6074 const struct got_error *error;
6075 struct got_repository *repo = NULL;
6076 struct got_worktree *worktree = NULL;
6077 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
6078 struct got_object_id *commit_id = NULL;
6079 struct got_commit_object *commit = NULL;
6080 const char *commit_id_arg = NULL;
6081 char *label = NULL;
6082 struct got_reference *ref = NULL;
6083 const char *head_ref_name = NULL;
6084 int ch;
6085 struct tog_view *view;
6086 int *pack_fds = NULL;
6088 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
6089 switch (ch) {
6090 case 'c':
6091 commit_id_arg = optarg;
6092 break;
6093 case 'r':
6094 repo_path = realpath(optarg, NULL);
6095 if (repo_path == NULL)
6096 return got_error_from_errno2("realpath",
6097 optarg);
6098 break;
6099 default:
6100 usage_tree();
6101 /* NOTREACHED */
6105 argc -= optind;
6106 argv += optind;
6108 if (argc > 1)
6109 usage_tree();
6111 error = got_repo_pack_fds_open(&pack_fds);
6112 if (error != NULL)
6113 goto done;
6115 if (repo_path == NULL) {
6116 cwd = getcwd(NULL, 0);
6117 if (cwd == NULL)
6118 return got_error_from_errno("getcwd");
6119 error = got_worktree_open(&worktree, cwd);
6120 if (error && error->code != GOT_ERR_NOT_WORKTREE)
6121 goto done;
6122 if (worktree)
6123 repo_path =
6124 strdup(got_worktree_get_repo_path(worktree));
6125 else
6126 repo_path = strdup(cwd);
6127 if (repo_path == NULL) {
6128 error = got_error_from_errno("strdup");
6129 goto done;
6133 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
6134 if (error != NULL)
6135 goto done;
6137 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
6138 repo, worktree);
6139 if (error)
6140 goto done;
6142 init_curses();
6144 error = apply_unveil(got_repo_get_path(repo), NULL);
6145 if (error)
6146 goto done;
6148 error = tog_load_refs(repo, 0);
6149 if (error)
6150 goto done;
6152 if (commit_id_arg == NULL) {
6153 error = got_repo_match_object_id(&commit_id, &label,
6154 worktree ? got_worktree_get_head_ref_name(worktree) :
6155 GOT_REF_HEAD, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
6156 if (error)
6157 goto done;
6158 head_ref_name = label;
6159 } else {
6160 error = got_ref_open(&ref, repo, commit_id_arg, 0);
6161 if (error == NULL)
6162 head_ref_name = got_ref_get_name(ref);
6163 else if (error->code != GOT_ERR_NOT_REF)
6164 goto done;
6165 error = got_repo_match_object_id(&commit_id, NULL,
6166 commit_id_arg, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
6167 if (error)
6168 goto done;
6171 error = got_object_open_as_commit(&commit, repo, commit_id);
6172 if (error)
6173 goto done;
6175 view = view_open(0, 0, 0, 0, TOG_VIEW_TREE);
6176 if (view == NULL) {
6177 error = got_error_from_errno("view_open");
6178 goto done;
6180 error = open_tree_view(view, commit_id, head_ref_name, repo);
6181 if (error)
6182 goto done;
6183 if (!got_path_is_root_dir(in_repo_path)) {
6184 error = tree_view_walk_path(&view->state.tree, commit,
6185 in_repo_path);
6186 if (error)
6187 goto done;
6190 if (worktree) {
6191 /* Release work tree lock. */
6192 got_worktree_close(worktree);
6193 worktree = NULL;
6195 error = view_loop(view);
6196 done:
6197 free(repo_path);
6198 free(cwd);
6199 free(commit_id);
6200 free(label);
6201 if (ref)
6202 got_ref_close(ref);
6203 if (repo) {
6204 const struct got_error *close_err = got_repo_close(repo);
6205 if (error == NULL)
6206 error = close_err;
6208 if (pack_fds) {
6209 const struct got_error *pack_err =
6210 got_repo_pack_fds_close(pack_fds);
6211 if (error == NULL)
6212 error = pack_err;
6214 tog_free_refs();
6215 return error;
6218 static const struct got_error *
6219 ref_view_load_refs(struct tog_ref_view_state *s)
6221 struct got_reflist_entry *sre;
6222 struct tog_reflist_entry *re;
6224 s->nrefs = 0;
6225 TAILQ_FOREACH(sre, &tog_refs, entry) {
6226 if (strncmp(got_ref_get_name(sre->ref),
6227 "refs/got/", 9) == 0 &&
6228 strncmp(got_ref_get_name(sre->ref),
6229 "refs/got/backup/", 16) != 0)
6230 continue;
6232 re = malloc(sizeof(*re));
6233 if (re == NULL)
6234 return got_error_from_errno("malloc");
6236 re->ref = got_ref_dup(sre->ref);
6237 if (re->ref == NULL)
6238 return got_error_from_errno("got_ref_dup");
6239 re->idx = s->nrefs++;
6240 TAILQ_INSERT_TAIL(&s->refs, re, entry);
6243 s->first_displayed_entry = TAILQ_FIRST(&s->refs);
6244 return NULL;
6247 void
6248 ref_view_free_refs(struct tog_ref_view_state *s)
6250 struct tog_reflist_entry *re;
6252 while (!TAILQ_EMPTY(&s->refs)) {
6253 re = TAILQ_FIRST(&s->refs);
6254 TAILQ_REMOVE(&s->refs, re, entry);
6255 got_ref_close(re->ref);
6256 free(re);
6260 static const struct got_error *
6261 open_ref_view(struct tog_view *view, struct got_repository *repo)
6263 const struct got_error *err = NULL;
6264 struct tog_ref_view_state *s = &view->state.ref;
6266 s->selected_entry = 0;
6267 s->repo = repo;
6269 TAILQ_INIT(&s->refs);
6270 STAILQ_INIT(&s->colors);
6272 err = ref_view_load_refs(s);
6273 if (err)
6274 return err;
6276 if (has_colors() && getenv("TOG_COLORS") != NULL) {
6277 err = add_color(&s->colors, "^refs/heads/",
6278 TOG_COLOR_REFS_HEADS,
6279 get_color_value("TOG_COLOR_REFS_HEADS"));
6280 if (err)
6281 goto done;
6283 err = add_color(&s->colors, "^refs/tags/",
6284 TOG_COLOR_REFS_TAGS,
6285 get_color_value("TOG_COLOR_REFS_TAGS"));
6286 if (err)
6287 goto done;
6289 err = add_color(&s->colors, "^refs/remotes/",
6290 TOG_COLOR_REFS_REMOTES,
6291 get_color_value("TOG_COLOR_REFS_REMOTES"));
6292 if (err)
6293 goto done;
6295 err = add_color(&s->colors, "^refs/got/backup/",
6296 TOG_COLOR_REFS_BACKUP,
6297 get_color_value("TOG_COLOR_REFS_BACKUP"));
6298 if (err)
6299 goto done;
6302 view->show = show_ref_view;
6303 view->input = input_ref_view;
6304 view->close = close_ref_view;
6305 view->search_start = search_start_ref_view;
6306 view->search_next = search_next_ref_view;
6307 done:
6308 if (err)
6309 free_colors(&s->colors);
6310 return err;
6313 static const struct got_error *
6314 close_ref_view(struct tog_view *view)
6316 struct tog_ref_view_state *s = &view->state.ref;
6318 ref_view_free_refs(s);
6319 free_colors(&s->colors);
6321 return NULL;
6324 static const struct got_error *
6325 resolve_reflist_entry(struct got_object_id **commit_id,
6326 struct tog_reflist_entry *re, struct got_repository *repo)
6328 const struct got_error *err = NULL;
6329 struct got_object_id *obj_id;
6330 struct got_tag_object *tag = NULL;
6331 int obj_type;
6333 *commit_id = NULL;
6335 err = got_ref_resolve(&obj_id, repo, re->ref);
6336 if (err)
6337 return err;
6339 err = got_object_get_type(&obj_type, repo, obj_id);
6340 if (err)
6341 goto done;
6343 switch (obj_type) {
6344 case GOT_OBJ_TYPE_COMMIT:
6345 *commit_id = obj_id;
6346 break;
6347 case GOT_OBJ_TYPE_TAG:
6348 err = got_object_open_as_tag(&tag, repo, obj_id);
6349 if (err)
6350 goto done;
6351 free(obj_id);
6352 err = got_object_get_type(&obj_type, repo,
6353 got_object_tag_get_object_id(tag));
6354 if (err)
6355 goto done;
6356 if (obj_type != GOT_OBJ_TYPE_COMMIT) {
6357 err = got_error(GOT_ERR_OBJ_TYPE);
6358 goto done;
6360 *commit_id = got_object_id_dup(
6361 got_object_tag_get_object_id(tag));
6362 if (*commit_id == NULL) {
6363 err = got_error_from_errno("got_object_id_dup");
6364 goto done;
6366 break;
6367 default:
6368 err = got_error(GOT_ERR_OBJ_TYPE);
6369 break;
6372 done:
6373 if (tag)
6374 got_object_tag_close(tag);
6375 if (err) {
6376 free(*commit_id);
6377 *commit_id = NULL;
6379 return err;
6382 static const struct got_error *
6383 log_ref_entry(struct tog_view **new_view, int begin_x,
6384 struct tog_reflist_entry *re, struct got_repository *repo)
6386 struct tog_view *log_view;
6387 const struct got_error *err = NULL;
6388 struct got_object_id *commit_id = NULL;
6390 *new_view = NULL;
6392 err = resolve_reflist_entry(&commit_id, re, repo);
6393 if (err) {
6394 if (err->code != GOT_ERR_OBJ_TYPE)
6395 return err;
6396 else
6397 return NULL;
6400 log_view = view_open(0, 0, 0, begin_x, TOG_VIEW_LOG);
6401 if (log_view == NULL) {
6402 err = got_error_from_errno("view_open");
6403 goto done;
6406 err = open_log_view(log_view, commit_id, repo,
6407 got_ref_get_name(re->ref), "", 0);
6408 done:
6409 if (err)
6410 view_close(log_view);
6411 else
6412 *new_view = log_view;
6413 free(commit_id);
6414 return err;
6417 static void
6418 ref_scroll_up(struct tog_ref_view_state *s, int maxscroll)
6420 struct tog_reflist_entry *re;
6421 int i = 0;
6423 if (s->first_displayed_entry == TAILQ_FIRST(&s->refs))
6424 return;
6426 re = TAILQ_PREV(s->first_displayed_entry, tog_reflist_head, entry);
6427 while (i++ < maxscroll) {
6428 if (re == NULL)
6429 break;
6430 s->first_displayed_entry = re;
6431 re = TAILQ_PREV(re, tog_reflist_head, entry);
6435 static void
6436 ref_scroll_down(struct tog_ref_view_state *s, int maxscroll)
6438 struct tog_reflist_entry *next, *last;
6439 int n = 0;
6441 if (s->first_displayed_entry)
6442 next = TAILQ_NEXT(s->first_displayed_entry, entry);
6443 else
6444 next = TAILQ_FIRST(&s->refs);
6446 last = s->last_displayed_entry;
6447 while (next && last && n++ < maxscroll) {
6448 last = TAILQ_NEXT(last, entry);
6449 if (last) {
6450 s->first_displayed_entry = next;
6451 next = TAILQ_NEXT(next, entry);
6456 static const struct got_error *
6457 search_start_ref_view(struct tog_view *view)
6459 struct tog_ref_view_state *s = &view->state.ref;
6461 s->matched_entry = NULL;
6462 return NULL;
6465 static int
6466 match_reflist_entry(struct tog_reflist_entry *re, regex_t *regex)
6468 regmatch_t regmatch;
6470 return regexec(regex, got_ref_get_name(re->ref), 1, &regmatch,
6471 0) == 0;
6474 static const struct got_error *
6475 search_next_ref_view(struct tog_view *view)
6477 struct tog_ref_view_state *s = &view->state.ref;
6478 struct tog_reflist_entry *re = NULL;
6480 if (!view->searching) {
6481 view->search_next_done = TOG_SEARCH_HAVE_MORE;
6482 return NULL;
6485 if (s->matched_entry) {
6486 if (view->searching == TOG_SEARCH_FORWARD) {
6487 if (s->selected_entry)
6488 re = TAILQ_NEXT(s->selected_entry, entry);
6489 else
6490 re = TAILQ_PREV(s->selected_entry,
6491 tog_reflist_head, entry);
6492 } else {
6493 if (s->selected_entry == NULL)
6494 re = TAILQ_LAST(&s->refs, tog_reflist_head);
6495 else
6496 re = TAILQ_PREV(s->selected_entry,
6497 tog_reflist_head, entry);
6499 } else {
6500 if (s->selected_entry)
6501 re = s->selected_entry;
6502 else if (view->searching == TOG_SEARCH_FORWARD)
6503 re = TAILQ_FIRST(&s->refs);
6504 else
6505 re = TAILQ_LAST(&s->refs, tog_reflist_head);
6508 while (1) {
6509 if (re == NULL) {
6510 if (s->matched_entry == NULL) {
6511 view->search_next_done = TOG_SEARCH_HAVE_MORE;
6512 return NULL;
6514 if (view->searching == TOG_SEARCH_FORWARD)
6515 re = TAILQ_FIRST(&s->refs);
6516 else
6517 re = TAILQ_LAST(&s->refs, tog_reflist_head);
6520 if (match_reflist_entry(re, &view->regex)) {
6521 view->search_next_done = TOG_SEARCH_HAVE_MORE;
6522 s->matched_entry = re;
6523 break;
6526 if (view->searching == TOG_SEARCH_FORWARD)
6527 re = TAILQ_NEXT(re, entry);
6528 else
6529 re = TAILQ_PREV(re, tog_reflist_head, entry);
6532 if (s->matched_entry) {
6533 s->first_displayed_entry = s->matched_entry;
6534 s->selected = 0;
6537 return NULL;
6540 static const struct got_error *
6541 show_ref_view(struct tog_view *view)
6543 const struct got_error *err = NULL;
6544 struct tog_ref_view_state *s = &view->state.ref;
6545 struct tog_reflist_entry *re;
6546 char *line = NULL;
6547 wchar_t *wline;
6548 struct tog_color *tc;
6549 int width, n;
6550 int limit = view->nlines;
6552 werase(view->window);
6554 s->ndisplayed = 0;
6556 if (limit == 0)
6557 return NULL;
6559 re = s->first_displayed_entry;
6561 if (asprintf(&line, "references [%d/%d]", re->idx + s->selected + 1,
6562 s->nrefs) == -1)
6563 return got_error_from_errno("asprintf");
6565 err = format_line(&wline, &width, NULL, line, 0, view->ncols, 0, 0);
6566 if (err) {
6567 free(line);
6568 return err;
6570 if (view_needs_focus_indication(view))
6571 wstandout(view->window);
6572 waddwstr(view->window, wline);
6573 if (view_needs_focus_indication(view))
6574 wstandend(view->window);
6575 free(wline);
6576 wline = NULL;
6577 free(line);
6578 line = NULL;
6579 if (width < view->ncols - 1)
6580 waddch(view->window, '\n');
6581 if (--limit <= 0)
6582 return NULL;
6584 n = 0;
6585 while (re && limit > 0) {
6586 char *line = NULL;
6587 char ymd[13]; /* YYYY-MM-DD + " " + NUL */
6589 if (s->show_date) {
6590 struct got_commit_object *ci;
6591 struct got_tag_object *tag;
6592 struct got_object_id *id;
6593 struct tm tm;
6594 time_t t;
6596 err = got_ref_resolve(&id, s->repo, re->ref);
6597 if (err)
6598 return err;
6599 err = got_object_open_as_tag(&tag, s->repo, id);
6600 if (err) {
6601 if (err->code != GOT_ERR_OBJ_TYPE) {
6602 free(id);
6603 return err;
6605 err = got_object_open_as_commit(&ci, s->repo,
6606 id);
6607 if (err) {
6608 free(id);
6609 return err;
6611 t = got_object_commit_get_committer_time(ci);
6612 got_object_commit_close(ci);
6613 } else {
6614 t = got_object_tag_get_tagger_time(tag);
6615 got_object_tag_close(tag);
6617 free(id);
6618 if (gmtime_r(&t, &tm) == NULL)
6619 return got_error_from_errno("gmtime_r");
6620 if (strftime(ymd, sizeof(ymd), "%G-%m-%d ", &tm) == 0)
6621 return got_error(GOT_ERR_NO_SPACE);
6623 if (got_ref_is_symbolic(re->ref)) {
6624 if (asprintf(&line, "%s%s -> %s", s->show_date ?
6625 ymd : "", got_ref_get_name(re->ref),
6626 got_ref_get_symref_target(re->ref)) == -1)
6627 return got_error_from_errno("asprintf");
6628 } else if (s->show_ids) {
6629 struct got_object_id *id;
6630 char *id_str;
6631 err = got_ref_resolve(&id, s->repo, re->ref);
6632 if (err)
6633 return err;
6634 err = got_object_id_str(&id_str, id);
6635 if (err) {
6636 free(id);
6637 return err;
6639 if (asprintf(&line, "%s%s: %s", s->show_date ? ymd : "",
6640 got_ref_get_name(re->ref), id_str) == -1) {
6641 err = got_error_from_errno("asprintf");
6642 free(id);
6643 free(id_str);
6644 return err;
6646 free(id);
6647 free(id_str);
6648 } else if (asprintf(&line, "%s%s", s->show_date ? ymd : "",
6649 got_ref_get_name(re->ref)) == -1)
6650 return got_error_from_errno("asprintf");
6652 err = format_line(&wline, &width, NULL, line, 0, view->ncols,
6653 0, 0);
6654 if (err) {
6655 free(line);
6656 return err;
6658 if (n == s->selected) {
6659 if (view->focussed)
6660 wstandout(view->window);
6661 s->selected_entry = re;
6663 tc = match_color(&s->colors, got_ref_get_name(re->ref));
6664 if (tc)
6665 wattr_on(view->window,
6666 COLOR_PAIR(tc->colorpair), NULL);
6667 waddwstr(view->window, wline);
6668 if (tc)
6669 wattr_off(view->window,
6670 COLOR_PAIR(tc->colorpair), NULL);
6671 if (width < view->ncols - 1)
6672 waddch(view->window, '\n');
6673 if (n == s->selected && view->focussed)
6674 wstandend(view->window);
6675 free(line);
6676 free(wline);
6677 wline = NULL;
6678 n++;
6679 s->ndisplayed++;
6680 s->last_displayed_entry = re;
6682 limit--;
6683 re = TAILQ_NEXT(re, entry);
6686 view_vborder(view);
6687 return err;
6690 static const struct got_error *
6691 browse_ref_tree(struct tog_view **new_view, int begin_x,
6692 struct tog_reflist_entry *re, struct got_repository *repo)
6694 const struct got_error *err = NULL;
6695 struct got_object_id *commit_id = NULL;
6696 struct tog_view *tree_view;
6698 *new_view = NULL;
6700 err = resolve_reflist_entry(&commit_id, re, repo);
6701 if (err) {
6702 if (err->code != GOT_ERR_OBJ_TYPE)
6703 return err;
6704 else
6705 return NULL;
6709 tree_view = view_open(0, 0, 0, begin_x, TOG_VIEW_TREE);
6710 if (tree_view == NULL) {
6711 err = got_error_from_errno("view_open");
6712 goto done;
6715 err = open_tree_view(tree_view, commit_id,
6716 got_ref_get_name(re->ref), repo);
6717 if (err)
6718 goto done;
6720 *new_view = tree_view;
6721 done:
6722 free(commit_id);
6723 return err;
6725 static const struct got_error *
6726 input_ref_view(struct tog_view **new_view, struct tog_view *view, int ch)
6728 const struct got_error *err = NULL;
6729 struct tog_ref_view_state *s = &view->state.ref;
6730 struct tog_view *log_view, *tree_view;
6731 struct tog_reflist_entry *re;
6732 int begin_x = 0, n, nscroll = view->nlines - 1;
6734 switch (ch) {
6735 case 'i':
6736 s->show_ids = !s->show_ids;
6737 break;
6738 case 'm':
6739 s->show_date = !s->show_date;
6740 break;
6741 case 'o':
6742 s->sort_by_date = !s->sort_by_date;
6743 err = got_reflist_sort(&tog_refs, s->sort_by_date ?
6744 got_ref_cmp_by_commit_timestamp_descending :
6745 tog_ref_cmp_by_name, s->repo);
6746 if (err)
6747 break;
6748 got_reflist_object_id_map_free(tog_refs_idmap);
6749 err = got_reflist_object_id_map_create(&tog_refs_idmap,
6750 &tog_refs, s->repo);
6751 if (err)
6752 break;
6753 ref_view_free_refs(s);
6754 err = ref_view_load_refs(s);
6755 break;
6756 case KEY_ENTER:
6757 case '\r':
6758 if (!s->selected_entry)
6759 break;
6760 if (view_is_parent_view(view))
6761 begin_x = view_split_begin_x(view->begin_x);
6762 err = log_ref_entry(&log_view, begin_x, s->selected_entry,
6763 s->repo);
6764 view->focussed = 0;
6765 log_view->focussed = 1;
6766 if (view_is_parent_view(view)) {
6767 err = view_close_child(view);
6768 if (err)
6769 return err;
6770 err = view_set_child(view, log_view);
6771 if (err)
6772 return err;
6773 view->focus_child = 1;
6774 } else
6775 *new_view = log_view;
6776 break;
6777 case 't':
6778 if (!s->selected_entry)
6779 break;
6780 if (view_is_parent_view(view))
6781 begin_x = view_split_begin_x(view->begin_x);
6782 err = browse_ref_tree(&tree_view, begin_x, s->selected_entry,
6783 s->repo);
6784 if (err || tree_view == NULL)
6785 break;
6786 view->focussed = 0;
6787 tree_view->focussed = 1;
6788 if (view_is_parent_view(view)) {
6789 err = view_close_child(view);
6790 if (err)
6791 return err;
6792 err = view_set_child(view, tree_view);
6793 if (err)
6794 return err;
6795 view->focus_child = 1;
6796 } else
6797 *new_view = tree_view;
6798 break;
6799 case 'g':
6800 case KEY_HOME:
6801 s->selected = 0;
6802 s->first_displayed_entry = TAILQ_FIRST(&s->refs);
6803 break;
6804 case 'G':
6805 case KEY_END:
6806 s->selected = 0;
6807 re = TAILQ_LAST(&s->refs, tog_reflist_head);
6808 for (n = 0; n < view->nlines - 1; n++) {
6809 if (re == NULL)
6810 break;
6811 s->first_displayed_entry = re;
6812 re = TAILQ_PREV(re, tog_reflist_head, entry);
6814 if (n > 0)
6815 s->selected = n - 1;
6816 break;
6817 case 'k':
6818 case KEY_UP:
6819 case CTRL('p'):
6820 if (s->selected > 0) {
6821 s->selected--;
6822 break;
6824 ref_scroll_up(s, 1);
6825 break;
6826 case CTRL('u'):
6827 case 'u':
6828 nscroll /= 2;
6829 /* FALL THROUGH */
6830 case KEY_PPAGE:
6831 case CTRL('b'):
6832 if (s->first_displayed_entry == TAILQ_FIRST(&s->refs))
6833 s->selected -= MIN(nscroll, s->selected);
6834 ref_scroll_up(s, MAX(0, nscroll));
6835 break;
6836 case 'j':
6837 case KEY_DOWN:
6838 case CTRL('n'):
6839 if (s->selected < s->ndisplayed - 1) {
6840 s->selected++;
6841 break;
6843 if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL)
6844 /* can't scroll any further */
6845 break;
6846 ref_scroll_down(s, 1);
6847 break;
6848 case CTRL('d'):
6849 case 'd':
6850 nscroll /= 2;
6851 /* FALL THROUGH */
6852 case KEY_NPAGE:
6853 case CTRL('f'):
6854 if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL) {
6855 /* can't scroll any further; move cursor down */
6856 if (s->selected < s->ndisplayed - 1)
6857 s->selected += MIN(nscroll,
6858 s->ndisplayed - s->selected - 1);
6859 break;
6861 ref_scroll_down(s, nscroll);
6862 break;
6863 case CTRL('l'):
6864 tog_free_refs();
6865 err = tog_load_refs(s->repo, s->sort_by_date);
6866 if (err)
6867 break;
6868 ref_view_free_refs(s);
6869 err = ref_view_load_refs(s);
6870 break;
6871 case KEY_RESIZE:
6872 if (view->nlines >= 2 && s->selected >= view->nlines - 1)
6873 s->selected = view->nlines - 2;
6874 break;
6875 default:
6876 break;
6879 return err;
6882 __dead static void
6883 usage_ref(void)
6885 endwin();
6886 fprintf(stderr, "usage: %s ref [-r repository-path]\n",
6887 getprogname());
6888 exit(1);
6891 static const struct got_error *
6892 cmd_ref(int argc, char *argv[])
6894 const struct got_error *error;
6895 struct got_repository *repo = NULL;
6896 struct got_worktree *worktree = NULL;
6897 char *cwd = NULL, *repo_path = NULL;
6898 int ch;
6899 struct tog_view *view;
6900 int *pack_fds = NULL;
6902 while ((ch = getopt(argc, argv, "r:")) != -1) {
6903 switch (ch) {
6904 case 'r':
6905 repo_path = realpath(optarg, NULL);
6906 if (repo_path == NULL)
6907 return got_error_from_errno2("realpath",
6908 optarg);
6909 break;
6910 default:
6911 usage_ref();
6912 /* NOTREACHED */
6916 argc -= optind;
6917 argv += optind;
6919 if (argc > 1)
6920 usage_ref();
6922 error = got_repo_pack_fds_open(&pack_fds);
6923 if (error != NULL)
6924 goto done;
6926 if (repo_path == NULL) {
6927 cwd = getcwd(NULL, 0);
6928 if (cwd == NULL)
6929 return got_error_from_errno("getcwd");
6930 error = got_worktree_open(&worktree, cwd);
6931 if (error && error->code != GOT_ERR_NOT_WORKTREE)
6932 goto done;
6933 if (worktree)
6934 repo_path =
6935 strdup(got_worktree_get_repo_path(worktree));
6936 else
6937 repo_path = strdup(cwd);
6938 if (repo_path == NULL) {
6939 error = got_error_from_errno("strdup");
6940 goto done;
6944 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
6945 if (error != NULL)
6946 goto done;
6948 init_curses();
6950 error = apply_unveil(got_repo_get_path(repo), NULL);
6951 if (error)
6952 goto done;
6954 error = tog_load_refs(repo, 0);
6955 if (error)
6956 goto done;
6958 view = view_open(0, 0, 0, 0, TOG_VIEW_REF);
6959 if (view == NULL) {
6960 error = got_error_from_errno("view_open");
6961 goto done;
6964 error = open_ref_view(view, repo);
6965 if (error)
6966 goto done;
6968 if (worktree) {
6969 /* Release work tree lock. */
6970 got_worktree_close(worktree);
6971 worktree = NULL;
6973 error = view_loop(view);
6974 done:
6975 free(repo_path);
6976 free(cwd);
6977 if (repo) {
6978 const struct got_error *close_err = got_repo_close(repo);
6979 if (close_err)
6980 error = close_err;
6982 if (pack_fds) {
6983 const struct got_error *pack_err =
6984 got_repo_pack_fds_close(pack_fds);
6985 if (error == NULL)
6986 error = pack_err;
6988 tog_free_refs();
6989 return error;
6992 static void
6993 list_commands(FILE *fp)
6995 size_t i;
6997 fprintf(fp, "commands:");
6998 for (i = 0; i < nitems(tog_commands); i++) {
6999 const struct tog_cmd *cmd = &tog_commands[i];
7000 fprintf(fp, " %s", cmd->name);
7002 fputc('\n', fp);
7005 __dead static void
7006 usage(int hflag, int status)
7008 FILE *fp = (status == 0) ? stdout : stderr;
7010 fprintf(fp, "usage: %s [-h] [-V | --version] [command] [arg ...]\n",
7011 getprogname());
7012 if (hflag) {
7013 fprintf(fp, "lazy usage: %s path\n", getprogname());
7014 list_commands(fp);
7016 exit(status);
7019 static char **
7020 make_argv(int argc, ...)
7022 va_list ap;
7023 char **argv;
7024 int i;
7026 va_start(ap, argc);
7028 argv = calloc(argc, sizeof(char *));
7029 if (argv == NULL)
7030 err(1, "calloc");
7031 for (i = 0; i < argc; i++) {
7032 argv[i] = strdup(va_arg(ap, char *));
7033 if (argv[i] == NULL)
7034 err(1, "strdup");
7037 va_end(ap);
7038 return argv;
7042 * Try to convert 'tog path' into a 'tog log path' command.
7043 * The user could simply have mistyped the command rather than knowingly
7044 * provided a path. So check whether argv[0] can in fact be resolved
7045 * to a path in the HEAD commit and print a special error if not.
7046 * This hack is for mpi@ <3
7048 static const struct got_error *
7049 tog_log_with_path(int argc, char *argv[])
7051 const struct got_error *error = NULL, *close_err;
7052 const struct tog_cmd *cmd = NULL;
7053 struct got_repository *repo = NULL;
7054 struct got_worktree *worktree = NULL;
7055 struct got_object_id *commit_id = NULL, *id = NULL;
7056 struct got_commit_object *commit = NULL;
7057 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
7058 char *commit_id_str = NULL, **cmd_argv = NULL;
7059 int *pack_fds = NULL;
7061 cwd = getcwd(NULL, 0);
7062 if (cwd == NULL)
7063 return got_error_from_errno("getcwd");
7065 error = got_repo_pack_fds_open(&pack_fds);
7066 if (error != NULL)
7067 goto done;
7069 error = got_worktree_open(&worktree, cwd);
7070 if (error && error->code != GOT_ERR_NOT_WORKTREE)
7071 goto done;
7073 if (worktree)
7074 repo_path = strdup(got_worktree_get_repo_path(worktree));
7075 else
7076 repo_path = strdup(cwd);
7077 if (repo_path == NULL) {
7078 error = got_error_from_errno("strdup");
7079 goto done;
7082 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
7083 if (error != NULL)
7084 goto done;
7086 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
7087 repo, worktree);
7088 if (error)
7089 goto done;
7091 error = tog_load_refs(repo, 0);
7092 if (error)
7093 goto done;
7094 error = got_repo_match_object_id(&commit_id, NULL, worktree ?
7095 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
7096 GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
7097 if (error)
7098 goto done;
7100 if (worktree) {
7101 got_worktree_close(worktree);
7102 worktree = NULL;
7105 error = got_object_open_as_commit(&commit, repo, commit_id);
7106 if (error)
7107 goto done;
7109 error = got_object_id_by_path(&id, repo, commit, in_repo_path);
7110 if (error) {
7111 if (error->code != GOT_ERR_NO_TREE_ENTRY)
7112 goto done;
7113 fprintf(stderr, "%s: '%s' is no known command or path\n",
7114 getprogname(), argv[0]);
7115 usage(1, 1);
7116 /* not reached */
7119 close_err = got_repo_close(repo);
7120 if (error == NULL)
7121 error = close_err;
7122 repo = NULL;
7124 error = got_object_id_str(&commit_id_str, commit_id);
7125 if (error)
7126 goto done;
7128 cmd = &tog_commands[0]; /* log */
7129 argc = 4;
7130 cmd_argv = make_argv(argc, cmd->name, "-c", commit_id_str, argv[0]);
7131 error = cmd->cmd_main(argc, cmd_argv);
7132 done:
7133 if (repo) {
7134 close_err = got_repo_close(repo);
7135 if (error == NULL)
7136 error = close_err;
7138 if (commit)
7139 got_object_commit_close(commit);
7140 if (worktree)
7141 got_worktree_close(worktree);
7142 if (pack_fds) {
7143 const struct got_error *pack_err =
7144 got_repo_pack_fds_close(pack_fds);
7145 if (error == NULL)
7146 error = pack_err;
7148 free(id);
7149 free(commit_id_str);
7150 free(commit_id);
7151 free(cwd);
7152 free(repo_path);
7153 free(in_repo_path);
7154 if (cmd_argv) {
7155 int i;
7156 for (i = 0; i < argc; i++)
7157 free(cmd_argv[i]);
7158 free(cmd_argv);
7160 tog_free_refs();
7161 return error;
7164 int
7165 main(int argc, char *argv[])
7167 const struct got_error *error = NULL;
7168 const struct tog_cmd *cmd = NULL;
7169 int ch, hflag = 0, Vflag = 0;
7170 char **cmd_argv = NULL;
7171 static const struct option longopts[] = {
7172 { "version", no_argument, NULL, 'V' },
7173 { NULL, 0, NULL, 0}
7176 setlocale(LC_CTYPE, "");
7178 while ((ch = getopt_long(argc, argv, "+hV", longopts, NULL)) != -1) {
7179 switch (ch) {
7180 case 'h':
7181 hflag = 1;
7182 break;
7183 case 'V':
7184 Vflag = 1;
7185 break;
7186 default:
7187 usage(hflag, 1);
7188 /* NOTREACHED */
7192 argc -= optind;
7193 argv += optind;
7194 optind = 1;
7195 optreset = 1;
7197 if (Vflag) {
7198 got_version_print_str();
7199 return 0;
7202 #ifndef PROFILE
7203 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
7204 NULL) == -1)
7205 err(1, "pledge");
7206 #endif
7208 if (argc == 0) {
7209 if (hflag)
7210 usage(hflag, 0);
7211 /* Build an argument vector which runs a default command. */
7212 cmd = &tog_commands[0];
7213 argc = 1;
7214 cmd_argv = make_argv(argc, cmd->name);
7215 } else {
7216 size_t i;
7218 /* Did the user specify a command? */
7219 for (i = 0; i < nitems(tog_commands); i++) {
7220 if (strncmp(tog_commands[i].name, argv[0],
7221 strlen(argv[0])) == 0) {
7222 cmd = &tog_commands[i];
7223 break;
7228 if (cmd == NULL) {
7229 if (argc != 1)
7230 usage(0, 1);
7231 /* No command specified; try log with a path */
7232 error = tog_log_with_path(argc, argv);
7233 } else {
7234 if (hflag)
7235 cmd->cmd_usage();
7236 else
7237 error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
7240 endwin();
7241 putchar('\n');
7242 if (cmd_argv) {
7243 int i;
7244 for (i = 0; i < argc; i++)
7245 free(cmd_argv[i]);
7246 free(cmd_argv);
7249 if (error && error->code != GOT_ERR_CANCELLED)
7250 fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
7251 return 0;