Blob


1 /*
2 * Copyright (c) 2018, 2019, 2020 Stefan Sperling <stsp@openbsd.org>
3 *
4 * Permission to use, copy, modify, and distribute this software for any
5 * purpose with or without fee is hereby granted, provided that the above
6 * copyright notice and this permission notice appear in all copies.
7 *
8 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 */
17 #include <sys/queue.h>
18 #include <sys/stat.h>
19 #include <sys/ioctl.h>
21 #include <ctype.h>
22 #include <errno.h>
23 #if defined(__FreeBSD__) || defined(__APPLE__)
24 #define _XOPEN_SOURCE_EXTENDED /* for ncurses wide-character functions */
25 #endif
26 #include <curses.h>
27 #include <panel.h>
28 #include <locale.h>
29 #include <signal.h>
30 #include <stdlib.h>
31 #include <stdarg.h>
32 #include <stdio.h>
33 #include <getopt.h>
34 #include <string.h>
35 #include <err.h>
36 #include <unistd.h>
37 #include <limits.h>
38 #include <wchar.h>
39 #include <time.h>
40 #include <pthread.h>
41 #include <libgen.h>
42 #include <regex.h>
43 #include <sched.h>
45 #include "got_compat.h"
47 #include "got_version.h"
48 #include "got_error.h"
49 #include "got_object.h"
50 #include "got_reference.h"
51 #include "got_repository.h"
52 #include "got_diff.h"
53 #include "got_opentemp.h"
54 #include "got_utf8.h"
55 #include "got_cancel.h"
56 #include "got_commit_graph.h"
57 #include "got_blame.h"
58 #include "got_privsep.h"
59 #include "got_path.h"
60 #include "got_worktree.h"
62 #ifndef MIN
63 #define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
64 #endif
66 #ifndef MAX
67 #define MAX(_a,_b) ((_a) > (_b) ? (_a) : (_b))
68 #endif
70 #ifndef CTRL
71 #define CTRL(x) ((x) & 0x1f)
72 #endif
74 #ifndef nitems
75 #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
76 #endif
78 struct tog_cmd {
79 const char *name;
80 const struct got_error *(*cmd_main)(int, char *[]);
81 void (*cmd_usage)(void);
82 };
84 __dead static void usage(int, int);
85 __dead static void usage_log(void);
86 __dead static void usage_diff(void);
87 __dead static void usage_blame(void);
88 __dead static void usage_tree(void);
89 __dead static void usage_ref(void);
91 static const struct got_error* cmd_log(int, char *[]);
92 static const struct got_error* cmd_diff(int, char *[]);
93 static const struct got_error* cmd_blame(int, char *[]);
94 static const struct got_error* cmd_tree(int, char *[]);
95 static const struct got_error* cmd_ref(int, char *[]);
97 static const struct tog_cmd tog_commands[] = {
98 { "log", cmd_log, usage_log },
99 { "diff", cmd_diff, usage_diff },
100 { "blame", cmd_blame, usage_blame },
101 { "tree", cmd_tree, usage_tree },
102 { "ref", cmd_ref, usage_ref },
103 };
105 enum tog_view_type {
106 TOG_VIEW_DIFF,
107 TOG_VIEW_LOG,
108 TOG_VIEW_BLAME,
109 TOG_VIEW_TREE,
110 TOG_VIEW_REF,
111 };
113 enum tog_view_mode {
114 TOG_VIEW_SPLIT_NONE,
115 TOG_VIEW_SPLIT_VERT,
116 TOG_VIEW_SPLIT_HRZN
117 };
119 #define HSPLIT_SCALE 0.3 /* default horizontal split scale */
121 #define TOG_EOF_STRING "(END)"
123 struct commit_queue_entry {
124 TAILQ_ENTRY(commit_queue_entry) entry;
125 struct got_object_id *id;
126 struct got_commit_object *commit;
127 int idx;
128 };
129 TAILQ_HEAD(commit_queue_head, commit_queue_entry);
130 struct commit_queue {
131 int ncommits;
132 struct commit_queue_head head;
133 };
135 struct tog_color {
136 STAILQ_ENTRY(tog_color) entry;
137 regex_t regex;
138 short colorpair;
139 };
140 STAILQ_HEAD(tog_colors, tog_color);
142 static struct got_reflist_head tog_refs = TAILQ_HEAD_INITIALIZER(tog_refs);
143 static struct got_reflist_object_id_map *tog_refs_idmap;
144 static enum got_diff_algorithm tog_diff_algo = GOT_DIFF_ALGORITHM_MYERS;
146 static const struct got_error *
147 tog_ref_cmp_by_name(void *arg, int *cmp, struct got_reference *re1,
148 struct got_reference* re2)
150 const char *name1 = got_ref_get_name(re1);
151 const char *name2 = got_ref_get_name(re2);
152 int isbackup1, isbackup2;
154 /* Sort backup refs towards the bottom of the list. */
155 isbackup1 = strncmp(name1, "refs/got/backup/", 16) == 0;
156 isbackup2 = strncmp(name2, "refs/got/backup/", 16) == 0;
157 if (!isbackup1 && isbackup2) {
158 *cmp = -1;
159 return NULL;
160 } else if (isbackup1 && !isbackup2) {
161 *cmp = 1;
162 return NULL;
165 *cmp = got_path_cmp(name1, name2, strlen(name1), strlen(name2));
166 return NULL;
169 static const struct got_error *
170 tog_load_refs(struct got_repository *repo, int sort_by_date)
172 const struct got_error *err;
174 err = got_ref_list(&tog_refs, repo, NULL, sort_by_date ?
175 got_ref_cmp_by_commit_timestamp_descending : tog_ref_cmp_by_name,
176 repo);
177 if (err)
178 return err;
180 return got_reflist_object_id_map_create(&tog_refs_idmap, &tog_refs,
181 repo);
184 static void
185 tog_free_refs(void)
187 if (tog_refs_idmap) {
188 got_reflist_object_id_map_free(tog_refs_idmap);
189 tog_refs_idmap = NULL;
191 got_ref_list_free(&tog_refs);
194 static const struct got_error *
195 add_color(struct tog_colors *colors, const char *pattern,
196 int idx, short color)
198 const struct got_error *err = NULL;
199 struct tog_color *tc;
200 int regerr = 0;
202 if (idx < 1 || idx > COLOR_PAIRS - 1)
203 return NULL;
205 init_pair(idx, color, -1);
207 tc = calloc(1, sizeof(*tc));
208 if (tc == NULL)
209 return got_error_from_errno("calloc");
210 regerr = regcomp(&tc->regex, pattern,
211 REG_EXTENDED | REG_NOSUB | REG_NEWLINE);
212 if (regerr) {
213 static char regerr_msg[512];
214 static char err_msg[512];
215 regerror(regerr, &tc->regex, regerr_msg,
216 sizeof(regerr_msg));
217 snprintf(err_msg, sizeof(err_msg), "regcomp: %s",
218 regerr_msg);
219 err = got_error_msg(GOT_ERR_REGEX, err_msg);
220 free(tc);
221 return err;
223 tc->colorpair = idx;
224 STAILQ_INSERT_HEAD(colors, tc, entry);
225 return NULL;
228 static void
229 free_colors(struct tog_colors *colors)
231 struct tog_color *tc;
233 while (!STAILQ_EMPTY(colors)) {
234 tc = STAILQ_FIRST(colors);
235 STAILQ_REMOVE_HEAD(colors, entry);
236 regfree(&tc->regex);
237 free(tc);
241 static struct tog_color *
242 get_color(struct tog_colors *colors, int colorpair)
244 struct tog_color *tc = NULL;
246 STAILQ_FOREACH(tc, colors, entry) {
247 if (tc->colorpair == colorpair)
248 return tc;
251 return NULL;
254 static int
255 default_color_value(const char *envvar)
257 if (strcmp(envvar, "TOG_COLOR_DIFF_MINUS") == 0)
258 return COLOR_MAGENTA;
259 if (strcmp(envvar, "TOG_COLOR_DIFF_PLUS") == 0)
260 return COLOR_CYAN;
261 if (strcmp(envvar, "TOG_COLOR_DIFF_CHUNK_HEADER") == 0)
262 return COLOR_YELLOW;
263 if (strcmp(envvar, "TOG_COLOR_DIFF_META") == 0)
264 return COLOR_GREEN;
265 if (strcmp(envvar, "TOG_COLOR_TREE_SUBMODULE") == 0)
266 return COLOR_MAGENTA;
267 if (strcmp(envvar, "TOG_COLOR_TREE_SYMLINK") == 0)
268 return COLOR_MAGENTA;
269 if (strcmp(envvar, "TOG_COLOR_TREE_DIRECTORY") == 0)
270 return COLOR_CYAN;
271 if (strcmp(envvar, "TOG_COLOR_TREE_EXECUTABLE") == 0)
272 return COLOR_GREEN;
273 if (strcmp(envvar, "TOG_COLOR_COMMIT") == 0)
274 return COLOR_GREEN;
275 if (strcmp(envvar, "TOG_COLOR_AUTHOR") == 0)
276 return COLOR_CYAN;
277 if (strcmp(envvar, "TOG_COLOR_DATE") == 0)
278 return COLOR_YELLOW;
279 if (strcmp(envvar, "TOG_COLOR_REFS_HEADS") == 0)
280 return COLOR_GREEN;
281 if (strcmp(envvar, "TOG_COLOR_REFS_TAGS") == 0)
282 return COLOR_MAGENTA;
283 if (strcmp(envvar, "TOG_COLOR_REFS_REMOTES") == 0)
284 return COLOR_YELLOW;
285 if (strcmp(envvar, "TOG_COLOR_REFS_BACKUP") == 0)
286 return COLOR_CYAN;
288 return -1;
291 static int
292 get_color_value(const char *envvar)
294 const char *val = getenv(envvar);
296 if (val == NULL)
297 return default_color_value(envvar);
299 if (strcasecmp(val, "black") == 0)
300 return COLOR_BLACK;
301 if (strcasecmp(val, "red") == 0)
302 return COLOR_RED;
303 if (strcasecmp(val, "green") == 0)
304 return COLOR_GREEN;
305 if (strcasecmp(val, "yellow") == 0)
306 return COLOR_YELLOW;
307 if (strcasecmp(val, "blue") == 0)
308 return COLOR_BLUE;
309 if (strcasecmp(val, "magenta") == 0)
310 return COLOR_MAGENTA;
311 if (strcasecmp(val, "cyan") == 0)
312 return COLOR_CYAN;
313 if (strcasecmp(val, "white") == 0)
314 return COLOR_WHITE;
315 if (strcasecmp(val, "default") == 0)
316 return -1;
318 return default_color_value(envvar);
322 struct tog_diff_view_state {
323 struct got_object_id *id1, *id2;
324 const char *label1, *label2;
325 FILE *f, *f1, *f2;
326 int fd1, fd2;
327 int first_displayed_line;
328 int last_displayed_line;
329 int eof;
330 int diff_context;
331 int ignore_whitespace;
332 int force_text_diff;
333 struct got_repository *repo;
334 struct tog_colors colors;
335 size_t nlines;
336 off_t *line_offsets;
337 int matched_line;
338 int selected_line;
340 /* passed from log or blame view; may be NULL */
341 struct tog_view *parent_view;
342 };
344 pthread_mutex_t tog_mutex = PTHREAD_MUTEX_INITIALIZER;
345 static volatile sig_atomic_t tog_thread_error;
347 struct tog_log_thread_args {
348 pthread_cond_t need_commits;
349 pthread_cond_t commit_loaded;
350 int commits_needed;
351 int load_all;
352 struct got_commit_graph *graph;
353 struct commit_queue *commits;
354 const char *in_repo_path;
355 struct got_object_id *start_id;
356 struct got_repository *repo;
357 int *pack_fds;
358 int log_complete;
359 sig_atomic_t *quit;
360 struct commit_queue_entry **first_displayed_entry;
361 struct commit_queue_entry **selected_entry;
362 int *searching;
363 int *search_next_done;
364 regex_t *regex;
365 };
367 struct tog_log_view_state {
368 struct commit_queue commits;
369 struct commit_queue_entry *first_displayed_entry;
370 struct commit_queue_entry *last_displayed_entry;
371 struct commit_queue_entry *selected_entry;
372 int selected;
373 char *in_repo_path;
374 char *head_ref_name;
375 int log_branches;
376 struct got_repository *repo;
377 struct got_object_id *start_id;
378 sig_atomic_t quit;
379 pthread_t thread;
380 struct tog_log_thread_args thread_args;
381 struct commit_queue_entry *matched_entry;
382 struct commit_queue_entry *search_entry;
383 struct tog_colors colors;
384 int use_committer;
385 };
387 #define TOG_COLOR_DIFF_MINUS 1
388 #define TOG_COLOR_DIFF_PLUS 2
389 #define TOG_COLOR_DIFF_CHUNK_HEADER 3
390 #define TOG_COLOR_DIFF_META 4
391 #define TOG_COLOR_TREE_SUBMODULE 5
392 #define TOG_COLOR_TREE_SYMLINK 6
393 #define TOG_COLOR_TREE_DIRECTORY 7
394 #define TOG_COLOR_TREE_EXECUTABLE 8
395 #define TOG_COLOR_COMMIT 9
396 #define TOG_COLOR_AUTHOR 10
397 #define TOG_COLOR_DATE 11
398 #define TOG_COLOR_REFS_HEADS 12
399 #define TOG_COLOR_REFS_TAGS 13
400 #define TOG_COLOR_REFS_REMOTES 14
401 #define TOG_COLOR_REFS_BACKUP 15
403 struct tog_blame_cb_args {
404 struct tog_blame_line *lines; /* one per line */
405 int nlines;
407 struct tog_view *view;
408 struct got_object_id *commit_id;
409 int *quit;
410 };
412 struct tog_blame_thread_args {
413 const char *path;
414 struct got_repository *repo;
415 struct tog_blame_cb_args *cb_args;
416 int *complete;
417 got_cancel_cb cancel_cb;
418 void *cancel_arg;
419 };
421 struct tog_blame {
422 FILE *f;
423 off_t filesize;
424 struct tog_blame_line *lines;
425 int nlines;
426 off_t *line_offsets;
427 pthread_t thread;
428 struct tog_blame_thread_args thread_args;
429 struct tog_blame_cb_args cb_args;
430 const char *path;
431 int *pack_fds;
432 };
434 struct tog_blame_view_state {
435 int first_displayed_line;
436 int last_displayed_line;
437 int selected_line;
438 int last_diffed_line;
439 int blame_complete;
440 int eof;
441 int done;
442 struct got_object_id_queue blamed_commits;
443 struct got_object_qid *blamed_commit;
444 char *path;
445 struct got_repository *repo;
446 struct got_object_id *commit_id;
447 struct tog_blame blame;
448 int matched_line;
449 struct tog_colors colors;
450 };
452 struct tog_parent_tree {
453 TAILQ_ENTRY(tog_parent_tree) entry;
454 struct got_tree_object *tree;
455 struct got_tree_entry *first_displayed_entry;
456 struct got_tree_entry *selected_entry;
457 int selected;
458 };
460 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
462 struct tog_tree_view_state {
463 char *tree_label;
464 struct got_object_id *commit_id;/* commit which this tree belongs to */
465 struct got_tree_object *root; /* the commit's root tree entry */
466 struct got_tree_object *tree; /* currently displayed (sub-)tree */
467 struct got_tree_entry *first_displayed_entry;
468 struct got_tree_entry *last_displayed_entry;
469 struct got_tree_entry *selected_entry;
470 int ndisplayed, selected, show_ids;
471 struct tog_parent_trees parents; /* parent trees of current sub-tree */
472 char *head_ref_name;
473 struct got_repository *repo;
474 struct got_tree_entry *matched_entry;
475 struct tog_colors colors;
476 };
478 struct tog_reflist_entry {
479 TAILQ_ENTRY(tog_reflist_entry) entry;
480 struct got_reference *ref;
481 int idx;
482 };
484 TAILQ_HEAD(tog_reflist_head, tog_reflist_entry);
486 struct tog_ref_view_state {
487 struct tog_reflist_head refs;
488 struct tog_reflist_entry *first_displayed_entry;
489 struct tog_reflist_entry *last_displayed_entry;
490 struct tog_reflist_entry *selected_entry;
491 int nrefs, ndisplayed, selected, show_date, show_ids, sort_by_date;
492 struct got_repository *repo;
493 struct tog_reflist_entry *matched_entry;
494 struct tog_colors colors;
495 };
497 /*
498 * We implement two types of views: parent views and child views.
500 * The 'Tab' key switches focus between a parent view and its child view.
501 * Child views are shown side-by-side to their parent view, provided
502 * there is enough screen estate.
504 * When a new view is opened from within a parent view, this new view
505 * becomes a child view of the parent view, replacing any existing child.
507 * When a new view is opened from within a child view, this new view
508 * becomes a parent view which will obscure the views below until the
509 * user quits the new parent view by typing 'q'.
511 * This list of views contains parent views only.
512 * Child views are only pointed to by their parent view.
513 */
514 TAILQ_HEAD(tog_view_list_head, tog_view);
516 struct tog_view {
517 TAILQ_ENTRY(tog_view) entry;
518 WINDOW *window;
519 PANEL *panel;
520 int nlines, ncols, begin_y, begin_x; /* based on split height/width */
521 int resized_y, resized_x; /* begin_y/x based on user resizing */
522 int maxx, x; /* max column and current start column */
523 int lines, cols; /* copies of LINES and COLS */
524 int nscrolled, offset; /* lines scrolled and hsplit line offset */
525 int ch, count; /* current keymap and count prefix */
526 int resized; /* set when in a resize event */
527 int focussed; /* Only set on one parent or child view at a time. */
528 int dying;
529 struct tog_view *parent;
530 struct tog_view *child;
532 /*
533 * This flag is initially set on parent views when a new child view
534 * is created. It gets toggled when the 'Tab' key switches focus
535 * between parent and child.
536 * The flag indicates whether focus should be passed on to our child
537 * view if this parent view gets picked for focus after another parent
538 * view was closed. This prevents child views from losing focus in such
539 * situations.
540 */
541 int focus_child;
543 enum tog_view_mode mode;
544 /* type-specific state */
545 enum tog_view_type type;
546 union {
547 struct tog_diff_view_state diff;
548 struct tog_log_view_state log;
549 struct tog_blame_view_state blame;
550 struct tog_tree_view_state tree;
551 struct tog_ref_view_state ref;
552 } state;
554 const struct got_error *(*show)(struct tog_view *);
555 const struct got_error *(*input)(struct tog_view **,
556 struct tog_view *, int);
557 const struct got_error *(*reset)(struct tog_view *);
558 const struct got_error *(*resize)(struct tog_view *, int);
559 const struct got_error *(*close)(struct tog_view *);
561 const struct got_error *(*search_start)(struct tog_view *);
562 const struct got_error *(*search_next)(struct tog_view *);
563 int search_started;
564 int searching;
565 #define TOG_SEARCH_FORWARD 1
566 #define TOG_SEARCH_BACKWARD 2
567 int search_next_done;
568 #define TOG_SEARCH_HAVE_MORE 1
569 #define TOG_SEARCH_NO_MORE 2
570 #define TOG_SEARCH_HAVE_NONE 3
571 regex_t regex;
572 regmatch_t regmatch;
573 };
575 static const struct got_error *open_diff_view(struct tog_view *,
576 struct got_object_id *, struct got_object_id *,
577 const char *, const char *, int, int, int, struct tog_view *,
578 struct got_repository *);
579 static const struct got_error *show_diff_view(struct tog_view *);
580 static const struct got_error *input_diff_view(struct tog_view **,
581 struct tog_view *, int);
582 static const struct got_error *reset_diff_view(struct tog_view *);
583 static const struct got_error* close_diff_view(struct tog_view *);
584 static const struct got_error *search_start_diff_view(struct tog_view *);
585 static const struct got_error *search_next_diff_view(struct tog_view *);
587 static const struct got_error *open_log_view(struct tog_view *,
588 struct got_object_id *, struct got_repository *,
589 const char *, const char *, int);
590 static const struct got_error * show_log_view(struct tog_view *);
591 static const struct got_error *input_log_view(struct tog_view **,
592 struct tog_view *, int);
593 static const struct got_error *resize_log_view(struct tog_view *, int);
594 static const struct got_error *close_log_view(struct tog_view *);
595 static const struct got_error *search_start_log_view(struct tog_view *);
596 static const struct got_error *search_next_log_view(struct tog_view *);
598 static const struct got_error *open_blame_view(struct tog_view *, char *,
599 struct got_object_id *, struct got_repository *);
600 static const struct got_error *show_blame_view(struct tog_view *);
601 static const struct got_error *input_blame_view(struct tog_view **,
602 struct tog_view *, int);
603 static const struct got_error *reset_blame_view(struct tog_view *);
604 static const struct got_error *close_blame_view(struct tog_view *);
605 static const struct got_error *search_start_blame_view(struct tog_view *);
606 static const struct got_error *search_next_blame_view(struct tog_view *);
608 static const struct got_error *open_tree_view(struct tog_view *,
609 struct got_object_id *, const char *, struct got_repository *);
610 static const struct got_error *show_tree_view(struct tog_view *);
611 static const struct got_error *input_tree_view(struct tog_view **,
612 struct tog_view *, int);
613 static const struct got_error *close_tree_view(struct tog_view *);
614 static const struct got_error *search_start_tree_view(struct tog_view *);
615 static const struct got_error *search_next_tree_view(struct tog_view *);
617 static const struct got_error *open_ref_view(struct tog_view *,
618 struct got_repository *);
619 static const struct got_error *show_ref_view(struct tog_view *);
620 static const struct got_error *input_ref_view(struct tog_view **,
621 struct tog_view *, int);
622 static const struct got_error *close_ref_view(struct tog_view *);
623 static const struct got_error *search_start_ref_view(struct tog_view *);
624 static const struct got_error *search_next_ref_view(struct tog_view *);
626 static volatile sig_atomic_t tog_sigwinch_received;
627 static volatile sig_atomic_t tog_sigpipe_received;
628 static volatile sig_atomic_t tog_sigcont_received;
629 static volatile sig_atomic_t tog_sigint_received;
630 static volatile sig_atomic_t tog_sigterm_received;
632 static void
633 tog_sigwinch(int signo)
635 tog_sigwinch_received = 1;
638 static void
639 tog_sigpipe(int signo)
641 tog_sigpipe_received = 1;
644 static void
645 tog_sigcont(int signo)
647 tog_sigcont_received = 1;
650 static void
651 tog_sigint(int signo)
653 tog_sigint_received = 1;
656 static void
657 tog_sigterm(int signo)
659 tog_sigterm_received = 1;
662 static int
663 tog_fatal_signal_received(void)
665 return (tog_sigpipe_received ||
666 tog_sigint_received || tog_sigint_received);
669 static const struct got_error *
670 view_close(struct tog_view *view)
672 const struct got_error *err = NULL, *child_err = NULL;
674 if (view->child) {
675 child_err = view_close(view->child);
676 view->child = NULL;
678 if (view->close)
679 err = view->close(view);
680 if (view->panel)
681 del_panel(view->panel);
682 if (view->window)
683 delwin(view->window);
684 free(view);
685 return err ? err : child_err;
688 static struct tog_view *
689 view_open(int nlines, int ncols, int begin_y, int begin_x,
690 enum tog_view_type type)
692 struct tog_view *view = calloc(1, sizeof(*view));
694 if (view == NULL)
695 return NULL;
697 view->type = type;
698 view->lines = LINES;
699 view->cols = COLS;
700 view->nlines = nlines ? nlines : LINES - begin_y;
701 view->ncols = ncols ? ncols : COLS - begin_x;
702 view->begin_y = begin_y;
703 view->begin_x = begin_x;
704 view->window = newwin(nlines, ncols, begin_y, begin_x);
705 if (view->window == NULL) {
706 view_close(view);
707 return NULL;
709 view->panel = new_panel(view->window);
710 if (view->panel == NULL ||
711 set_panel_userptr(view->panel, view) != OK) {
712 view_close(view);
713 return NULL;
716 keypad(view->window, TRUE);
717 return view;
720 static int
721 view_split_begin_x(int begin_x)
723 if (begin_x > 0 || COLS < 120)
724 return 0;
725 return (COLS - MAX(COLS / 2, 80));
728 /* XXX Stub till we decide what to do. */
729 static int
730 view_split_begin_y(int lines)
732 return lines * HSPLIT_SCALE;
735 static const struct got_error *view_resize(struct tog_view *);
737 static const struct got_error *
738 view_splitscreen(struct tog_view *view)
740 const struct got_error *err = NULL;
742 if (!view->resized && view->mode == TOG_VIEW_SPLIT_HRZN) {
743 if (view->resized_y && view->resized_y < view->lines)
744 view->begin_y = view->resized_y;
745 else
746 view->begin_y = view_split_begin_y(view->nlines);
747 view->begin_x = 0;
748 } else if (!view->resized) {
749 if (view->resized_x && view->resized_x < view->cols - 1 &&
750 view->cols > 119)
751 view->begin_x = view->resized_x;
752 else
753 view->begin_x = view_split_begin_x(0);
754 view->begin_y = 0;
756 view->nlines = LINES - view->begin_y;
757 view->ncols = COLS - view->begin_x;
758 view->lines = LINES;
759 view->cols = COLS;
760 err = view_resize(view);
761 if (err)
762 return err;
764 if (view->parent && view->mode == TOG_VIEW_SPLIT_HRZN)
765 view->parent->nlines = view->begin_y;
767 if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
768 return got_error_from_errno("mvwin");
770 return NULL;
773 static const struct got_error *
774 view_fullscreen(struct tog_view *view)
776 const struct got_error *err = NULL;
778 view->begin_x = 0;
779 view->begin_y = view->resized ? view->begin_y : 0;
780 view->nlines = view->resized ? view->nlines : LINES;
781 view->ncols = COLS;
782 view->lines = LINES;
783 view->cols = COLS;
784 err = view_resize(view);
785 if (err)
786 return err;
788 if (mvwin(view->window, view->begin_y, view->begin_x) == ERR)
789 return got_error_from_errno("mvwin");
791 return NULL;
794 static int
795 view_is_parent_view(struct tog_view *view)
797 return view->parent == NULL;
800 static int
801 view_is_splitscreen(struct tog_view *view)
803 return view->begin_x > 0 || view->begin_y > 0;
806 static int
807 view_is_fullscreen(struct tog_view *view)
809 return view->nlines == LINES && view->ncols == COLS;
812 static int
813 view_is_hsplit_top(struct tog_view *view)
815 return view->mode == TOG_VIEW_SPLIT_HRZN && view->child &&
816 view_is_splitscreen(view->child);
819 static void
820 view_border(struct tog_view *view)
822 PANEL *panel;
823 const struct tog_view *view_above;
825 if (view->parent)
826 return view_border(view->parent);
828 panel = panel_above(view->panel);
829 if (panel == NULL)
830 return;
832 view_above = panel_userptr(panel);
833 if (view->mode == TOG_VIEW_SPLIT_HRZN)
834 mvwhline(view->window, view_above->begin_y - 1,
835 view->begin_x, got_locale_is_utf8() ?
836 ACS_HLINE : '-', view->ncols);
837 else
838 mvwvline(view->window, view->begin_y, view_above->begin_x - 1,
839 got_locale_is_utf8() ? ACS_VLINE : '|', view->nlines);
842 static const struct got_error *view_init_hsplit(struct tog_view *, int);
843 static const struct got_error *request_log_commits(struct tog_view *);
844 static const struct got_error *offset_selection_down(struct tog_view *);
845 static void offset_selection_up(struct tog_view *);
846 static void view_get_split(struct tog_view *, int *, int *);
848 static const struct got_error *
849 view_resize(struct tog_view *view)
851 const struct got_error *err = NULL;
852 int dif, nlines, ncols;
854 dif = LINES - view->lines; /* line difference */
856 if (view->lines > LINES)
857 nlines = view->nlines - (view->lines - LINES);
858 else
859 nlines = view->nlines + (LINES - view->lines);
860 if (view->cols > COLS)
861 ncols = view->ncols - (view->cols - COLS);
862 else
863 ncols = view->ncols + (COLS - view->cols);
865 if (view->child) {
866 int hs = view->child->begin_y;
868 if (!view_is_fullscreen(view))
869 view->child->begin_x = view_split_begin_x(view->begin_x);
870 if (view->mode == TOG_VIEW_SPLIT_HRZN ||
871 view->child->begin_x == 0) {
872 ncols = COLS;
874 view_fullscreen(view->child);
875 if (view->child->focussed)
876 show_panel(view->child->panel);
877 else
878 show_panel(view->panel);
879 } else {
880 ncols = view->child->begin_x;
882 view_splitscreen(view->child);
883 show_panel(view->child->panel);
885 /*
886 * XXX This is ugly and needs to be moved into the above
887 * logic but "works" for now and my attempts at moving it
888 * break either 'tab' or 'F' key maps in horizontal splits.
889 */
890 if (hs) {
891 err = view_splitscreen(view->child);
892 if (err)
893 return err;
894 if (dif < 0) { /* top split decreased */
895 err = offset_selection_down(view);
896 if (err)
897 return err;
899 view_border(view);
900 update_panels();
901 doupdate();
902 show_panel(view->child->panel);
903 nlines = view->nlines;
905 } else if (view->parent == NULL)
906 ncols = COLS;
908 if (view->resize && dif > 0) {
909 err = view->resize(view, dif);
910 if (err)
911 return err;
914 if (wresize(view->window, nlines, ncols) == ERR)
915 return got_error_from_errno("wresize");
916 if (replace_panel(view->panel, view->window) == ERR)
917 return got_error_from_errno("replace_panel");
918 wclear(view->window);
920 view->nlines = nlines;
921 view->ncols = ncols;
922 view->lines = LINES;
923 view->cols = COLS;
925 return NULL;
928 static const struct got_error *
929 resize_log_view(struct tog_view *view, int increase)
931 struct tog_log_view_state *s = &view->state.log;
932 const struct got_error *err = NULL;
933 int n = 0;
935 if (s->selected_entry)
936 n = s->selected_entry->idx + view->lines - s->selected;
938 /*
939 * Request commits to account for the increased
940 * height so we have enough to populate the view.
941 */
942 if (s->commits.ncommits < n) {
943 view->nscrolled = n - s->commits.ncommits + increase + 1;
944 err = request_log_commits(view);
947 return err;
950 static void
951 view_adjust_offset(struct tog_view *view, int n)
953 if (n == 0)
954 return;
956 if (view->parent && view->parent->offset) {
957 if (view->parent->offset + n >= 0)
958 view->parent->offset += n;
959 else
960 view->parent->offset = 0;
961 } else if (view->offset) {
962 if (view->offset - n >= 0)
963 view->offset -= n;
964 else
965 view->offset = 0;
969 static const struct got_error *
970 view_resize_split(struct tog_view *view, int resize)
972 const struct got_error *err = NULL;
973 struct tog_view *v = NULL;
975 if (view->parent)
976 v = view->parent;
977 else
978 v = view;
980 if (!v->child || !view_is_splitscreen(v->child))
981 return NULL;
983 v->resized = v->child->resized = resize; /* lock for resize event */
985 if (view->mode == TOG_VIEW_SPLIT_HRZN) {
986 if (v->child->resized_y)
987 v->child->begin_y = v->child->resized_y;
988 if (view->parent)
989 v->child->begin_y -= resize;
990 else
991 v->child->begin_y += resize;
992 if (v->child->begin_y < 3) {
993 view->count = 0;
994 v->child->begin_y = 3;
995 } else if (v->child->begin_y > LINES - 1) {
996 view->count = 0;
997 v->child->begin_y = LINES - 1;
999 v->ncols = COLS;
1000 v->child->ncols = COLS;
1001 view_adjust_offset(view, resize);
1002 err = view_init_hsplit(v, v->child->begin_y);
1003 if (err)
1004 return err;
1005 v->child->resized_y = v->child->begin_y;
1006 } else {
1007 if (v->child->resized_x)
1008 v->child->begin_x = v->child->resized_x;
1009 if (view->parent)
1010 v->child->begin_x -= resize;
1011 else
1012 v->child->begin_x += resize;
1013 if (v->child->begin_x < 11) {
1014 view->count = 0;
1015 v->child->begin_x = 11;
1016 } else if (v->child->begin_x > COLS - 1) {
1017 view->count = 0;
1018 v->child->begin_x = COLS - 1;
1020 v->child->resized_x = v->child->begin_x;
1023 v->child->mode = v->mode;
1024 v->child->nlines = v->lines - v->child->begin_y;
1025 v->child->ncols = v->cols - v->child->begin_x;
1026 v->focus_child = 1;
1028 err = view_fullscreen(v);
1029 if (err)
1030 return err;
1031 err = view_splitscreen(v->child);
1032 if (err)
1033 return err;
1035 if (v->mode == TOG_VIEW_SPLIT_HRZN) {
1036 err = offset_selection_down(v->child);
1037 if (err)
1038 return err;
1041 if (v->resize)
1042 err = v->resize(v, 0);
1043 else if (v->child->resize)
1044 err = v->child->resize(v->child, 0);
1046 v->resized = v->child->resized = 0;
1048 return err;
1051 static void
1052 view_transfer_size(struct tog_view *dst, struct tog_view *src)
1054 struct tog_view *v = src->child ? src->child : src;
1056 dst->resized_x = v->resized_x;
1057 dst->resized_y = v->resized_y;
1060 static const struct got_error *
1061 view_close_child(struct tog_view *view)
1063 const struct got_error *err = NULL;
1065 if (view->child == NULL)
1066 return NULL;
1068 err = view_close(view->child);
1069 view->child = NULL;
1070 return err;
1073 static const struct got_error *
1074 view_set_child(struct tog_view *view, struct tog_view *child)
1076 const struct got_error *err = NULL;
1078 view->child = child;
1079 child->parent = view;
1081 err = view_resize(view);
1082 if (err)
1083 return err;
1085 if (view->child->resized_x || view->child->resized_y)
1086 err = view_resize_split(view, 0);
1088 return err;
1091 static void
1092 tog_resizeterm(void)
1094 int cols, lines;
1095 struct winsize size;
1097 if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &size) < 0) {
1098 cols = 80; /* Default */
1099 lines = 24;
1100 } else {
1101 cols = size.ws_col;
1102 lines = size.ws_row;
1104 resize_term(lines, cols);
1107 static const struct got_error *
1108 view_search_start(struct tog_view *view)
1110 const struct got_error *err = NULL;
1111 struct tog_view *v = view;
1112 char pattern[1024];
1113 int ret;
1115 if (view->search_started) {
1116 regfree(&view->regex);
1117 view->searching = 0;
1118 memset(&view->regmatch, 0, sizeof(view->regmatch));
1120 view->search_started = 0;
1122 if (view->nlines < 1)
1123 return NULL;
1125 if (view_is_hsplit_top(view))
1126 v = view->child;
1128 mvwaddstr(v->window, v->nlines - 1, 0, "/");
1129 wclrtoeol(v->window);
1131 nodelay(view->window, FALSE); /* block for search term input */
1132 nocbreak();
1133 echo();
1134 ret = wgetnstr(v->window, pattern, sizeof(pattern));
1135 wrefresh(v->window);
1136 cbreak();
1137 noecho();
1138 nodelay(view->window, TRUE);
1139 if (ret == ERR)
1140 return NULL;
1142 if (regcomp(&view->regex, pattern, REG_EXTENDED | REG_NEWLINE) == 0) {
1143 err = view->search_start(view);
1144 if (err) {
1145 regfree(&view->regex);
1146 return err;
1148 view->search_started = 1;
1149 view->searching = TOG_SEARCH_FORWARD;
1150 view->search_next_done = 0;
1151 view->search_next(view);
1154 return NULL;
1157 /* Switch split mode. If view is a parent or child, draw the new splitscreen. */
1158 static const struct got_error *
1159 switch_split(struct tog_view *view)
1161 const struct got_error *err = NULL;
1162 struct tog_view *v = NULL;
1164 if (view->parent)
1165 v = view->parent;
1166 else
1167 v = view;
1169 if (v->mode == TOG_VIEW_SPLIT_HRZN)
1170 v->mode = TOG_VIEW_SPLIT_VERT;
1171 else
1172 v->mode = TOG_VIEW_SPLIT_HRZN;
1174 if (!v->child)
1175 return NULL;
1176 else if (v->mode == TOG_VIEW_SPLIT_VERT && v->cols < 120)
1177 v->mode = TOG_VIEW_SPLIT_NONE;
1179 view_get_split(v, &v->child->begin_y, &v->child->begin_x);
1180 if (v->mode == TOG_VIEW_SPLIT_HRZN && v->child->resized_y)
1181 v->child->begin_y = v->child->resized_y;
1182 else if (v->mode == TOG_VIEW_SPLIT_VERT && v->child->resized_x)
1183 v->child->begin_x = v->child->resized_x;
1186 if (v->mode == TOG_VIEW_SPLIT_HRZN) {
1187 v->ncols = COLS;
1188 v->child->ncols = COLS;
1189 v->child->nscrolled = LINES - v->child->nlines;
1191 err = view_init_hsplit(v, v->child->begin_y);
1192 if (err)
1193 return err;
1195 v->child->mode = v->mode;
1196 v->child->nlines = v->lines - v->child->begin_y;
1197 v->focus_child = 1;
1199 err = view_fullscreen(v);
1200 if (err)
1201 return err;
1202 err = view_splitscreen(v->child);
1203 if (err)
1204 return err;
1206 if (v->mode == TOG_VIEW_SPLIT_NONE)
1207 v->mode = TOG_VIEW_SPLIT_VERT;
1208 if (v->mode == TOG_VIEW_SPLIT_HRZN) {
1209 err = offset_selection_down(v);
1210 err = offset_selection_down(v->child);
1211 } else {
1212 offset_selection_up(v);
1213 offset_selection_up(v->child);
1215 if (v->resize)
1216 err = v->resize(v, 0);
1217 else if (v->child->resize)
1218 err = v->child->resize(v->child, 0);
1220 return err;
1224 * Compute view->count from numeric input. Assign total to view->count and
1225 * return first non-numeric key entered.
1227 static int
1228 get_compound_key(struct tog_view *view, int c)
1230 struct tog_view *v = view;
1231 int x, n = 0;
1233 if (view_is_hsplit_top(view))
1234 v = view->child;
1235 else if (view->mode == TOG_VIEW_SPLIT_VERT && view->parent)
1236 v = view->parent;
1238 view->count = 0;
1239 cbreak(); /* block for input */
1240 wmove(v->window, v->nlines - 1, 0);
1241 wclrtoeol(v->window);
1242 waddch(v->window, ':');
1244 do {
1245 x = getcurx(v->window);
1246 if (x != ERR && x < view->ncols) {
1247 waddch(v->window, c);
1248 wrefresh(v->window);
1252 * Don't overflow. Max valid request should be the greatest
1253 * between the longest and total lines; cap at 10 million.
1255 if (n >= 9999999)
1256 n = 9999999;
1257 else
1258 n = n * 10 + (c - '0');
1259 } while (((c = wgetch(view->window))) >= '0' && c <= '9' && c != ERR);
1261 /* Massage excessive or inapplicable values at the input handler. */
1262 view->count = n;
1264 return c;
1267 static const struct got_error *
1268 view_input(struct tog_view **new, int *done, struct tog_view *view,
1269 struct tog_view_list_head *views)
1271 const struct got_error *err = NULL;
1272 struct tog_view *v;
1273 int ch, errcode;
1275 *new = NULL;
1277 /* Clear "no matches" indicator. */
1278 if (view->search_next_done == TOG_SEARCH_NO_MORE ||
1279 view->search_next_done == TOG_SEARCH_HAVE_NONE) {
1280 view->search_next_done = TOG_SEARCH_HAVE_MORE;
1281 view->count = 0;
1284 if (view->searching && !view->search_next_done) {
1285 errcode = pthread_mutex_unlock(&tog_mutex);
1286 if (errcode)
1287 return got_error_set_errno(errcode,
1288 "pthread_mutex_unlock");
1289 sched_yield();
1290 errcode = pthread_mutex_lock(&tog_mutex);
1291 if (errcode)
1292 return got_error_set_errno(errcode,
1293 "pthread_mutex_lock");
1294 view->search_next(view);
1295 return NULL;
1298 nodelay(view->window, FALSE);
1299 /* Allow threads to make progress while we are waiting for input. */
1300 errcode = pthread_mutex_unlock(&tog_mutex);
1301 if (errcode)
1302 return got_error_set_errno(errcode, "pthread_mutex_unlock");
1303 /* If we have an unfinished count, let C-g or backspace abort. */
1304 if (view->count && --view->count) {
1305 cbreak();
1306 nodelay(view->window, TRUE);
1307 ch = wgetch(view->window);
1308 if (ch == CTRL('g') || ch == KEY_BACKSPACE)
1309 view->count = 0;
1310 else
1311 ch = view->ch;
1312 } else {
1313 ch = wgetch(view->window);
1314 if (ch >= '1' && ch <= '9')
1315 view->ch = ch = get_compound_key(view, ch);
1317 errcode = pthread_mutex_lock(&tog_mutex);
1318 if (errcode)
1319 return got_error_set_errno(errcode, "pthread_mutex_lock");
1320 nodelay(view->window, TRUE);
1322 if (tog_sigwinch_received || tog_sigcont_received) {
1323 tog_resizeterm();
1324 tog_sigwinch_received = 0;
1325 tog_sigcont_received = 0;
1326 TAILQ_FOREACH(v, views, entry) {
1327 err = view_resize(v);
1328 if (err)
1329 return err;
1330 err = v->input(new, v, KEY_RESIZE);
1331 if (err)
1332 return err;
1333 if (v->child) {
1334 err = view_resize(v->child);
1335 if (err)
1336 return err;
1337 err = v->child->input(new, v->child,
1338 KEY_RESIZE);
1339 if (err)
1340 return err;
1341 if (v->child->resized_x || v->child->resized_y) {
1342 err = view_resize_split(v, 0);
1343 if (err)
1344 return err;
1350 switch (ch) {
1351 case '\t':
1352 view->count = 0;
1353 if (view->child) {
1354 view->focussed = 0;
1355 view->child->focussed = 1;
1356 view->focus_child = 1;
1357 } else if (view->parent) {
1358 view->focussed = 0;
1359 view->parent->focussed = 1;
1360 view->parent->focus_child = 0;
1361 if (!view_is_splitscreen(view)) {
1362 if (view->parent->resize) {
1363 err = view->parent->resize(view->parent,
1364 0);
1365 if (err)
1366 return err;
1368 offset_selection_up(view->parent);
1369 err = view_fullscreen(view->parent);
1370 if (err)
1371 return err;
1374 break;
1375 case 'q':
1376 if (view->parent && view->mode == TOG_VIEW_SPLIT_HRZN) {
1377 if (view->parent->resize) {
1378 /* might need more commits to fill fullscreen */
1379 err = view->parent->resize(view->parent, 0);
1380 if (err)
1381 break;
1383 offset_selection_up(view->parent);
1385 err = view->input(new, view, ch);
1386 view->dying = 1;
1387 break;
1388 case 'Q':
1389 *done = 1;
1390 break;
1391 case 'F':
1392 view->count = 0;
1393 if (view_is_parent_view(view)) {
1394 if (view->child == NULL)
1395 break;
1396 if (view_is_splitscreen(view->child)) {
1397 view->focussed = 0;
1398 view->child->focussed = 1;
1399 err = view_fullscreen(view->child);
1400 } else {
1401 err = view_splitscreen(view->child);
1402 if (!err)
1403 err = view_resize_split(view, 0);
1405 if (err)
1406 break;
1407 err = view->child->input(new, view->child,
1408 KEY_RESIZE);
1409 } else {
1410 if (view_is_splitscreen(view)) {
1411 view->parent->focussed = 0;
1412 view->focussed = 1;
1413 err = view_fullscreen(view);
1414 } else {
1415 err = view_splitscreen(view);
1416 if (!err && view->mode != TOG_VIEW_SPLIT_HRZN)
1417 err = view_resize(view->parent);
1418 if (!err)
1419 err = view_resize_split(view, 0);
1421 if (err)
1422 break;
1423 err = view->input(new, view, KEY_RESIZE);
1425 if (err)
1426 break;
1427 if (view->resize) {
1428 err = view->resize(view, 0);
1429 if (err)
1430 break;
1432 if (view->parent)
1433 err = offset_selection_down(view->parent);
1434 if (!err)
1435 err = offset_selection_down(view);
1436 break;
1437 case 'S':
1438 view->count = 0;
1439 err = switch_split(view);
1440 break;
1441 case '-':
1442 err = view_resize_split(view, -1);
1443 break;
1444 case '+':
1445 err = view_resize_split(view, 1);
1446 break;
1447 case KEY_RESIZE:
1448 break;
1449 case '/':
1450 view->count = 0;
1451 if (view->search_start)
1452 view_search_start(view);
1453 else
1454 err = view->input(new, view, ch);
1455 break;
1456 case 'N':
1457 case 'n':
1458 if (view->search_started && view->search_next) {
1459 view->searching = (ch == 'n' ?
1460 TOG_SEARCH_FORWARD : TOG_SEARCH_BACKWARD);
1461 view->search_next_done = 0;
1462 view->search_next(view);
1463 } else
1464 err = view->input(new, view, ch);
1465 break;
1466 case 'A':
1467 if (tog_diff_algo == GOT_DIFF_ALGORITHM_MYERS)
1468 tog_diff_algo = GOT_DIFF_ALGORITHM_PATIENCE;
1469 else
1470 tog_diff_algo = GOT_DIFF_ALGORITHM_MYERS;
1471 TAILQ_FOREACH(v, views, entry) {
1472 if (v->reset) {
1473 err = v->reset(v);
1474 if (err)
1475 return err;
1477 if (v->child && v->child->reset) {
1478 err = v->child->reset(v->child);
1479 if (err)
1480 return err;
1483 break;
1484 default:
1485 err = view->input(new, view, ch);
1486 break;
1489 return err;
1492 static int
1493 view_needs_focus_indication(struct tog_view *view)
1495 if (view_is_parent_view(view)) {
1496 if (view->child == NULL || view->child->focussed)
1497 return 0;
1498 if (!view_is_splitscreen(view->child))
1499 return 0;
1500 } else if (!view_is_splitscreen(view))
1501 return 0;
1503 return view->focussed;
1506 static const struct got_error *
1507 view_loop(struct tog_view *view)
1509 const struct got_error *err = NULL;
1510 struct tog_view_list_head views;
1511 struct tog_view *new_view;
1512 char *mode;
1513 int fast_refresh = 10;
1514 int done = 0, errcode;
1516 mode = getenv("TOG_VIEW_SPLIT_MODE");
1517 if (!mode || !(*mode == 'h' || *mode == 'H'))
1518 view->mode = TOG_VIEW_SPLIT_VERT;
1519 else
1520 view->mode = TOG_VIEW_SPLIT_HRZN;
1522 errcode = pthread_mutex_lock(&tog_mutex);
1523 if (errcode)
1524 return got_error_set_errno(errcode, "pthread_mutex_lock");
1526 TAILQ_INIT(&views);
1527 TAILQ_INSERT_HEAD(&views, view, entry);
1529 view->focussed = 1;
1530 err = view->show(view);
1531 if (err)
1532 return err;
1533 update_panels();
1534 doupdate();
1535 while (!TAILQ_EMPTY(&views) && !done && !tog_thread_error &&
1536 !tog_fatal_signal_received()) {
1537 /* Refresh fast during initialization, then become slower. */
1538 if (fast_refresh && fast_refresh-- == 0)
1539 halfdelay(10); /* switch to once per second */
1541 err = view_input(&new_view, &done, view, &views);
1542 if (err)
1543 break;
1544 if (view->dying) {
1545 struct tog_view *v, *prev = NULL;
1547 if (view_is_parent_view(view))
1548 prev = TAILQ_PREV(view, tog_view_list_head,
1549 entry);
1550 else if (view->parent)
1551 prev = view->parent;
1553 if (view->parent) {
1554 view->parent->child = NULL;
1555 view->parent->focus_child = 0;
1556 /* Restore fullscreen line height. */
1557 view->parent->nlines = view->parent->lines;
1558 err = view_resize(view->parent);
1559 if (err)
1560 break;
1561 /* Make resized splits persist. */
1562 view_transfer_size(view->parent, view);
1563 } else
1564 TAILQ_REMOVE(&views, view, entry);
1566 err = view_close(view);
1567 if (err)
1568 goto done;
1570 view = NULL;
1571 TAILQ_FOREACH(v, &views, entry) {
1572 if (v->focussed)
1573 break;
1575 if (view == NULL && new_view == NULL) {
1576 /* No view has focus. Try to pick one. */
1577 if (prev)
1578 view = prev;
1579 else if (!TAILQ_EMPTY(&views)) {
1580 view = TAILQ_LAST(&views,
1581 tog_view_list_head);
1583 if (view) {
1584 if (view->focus_child) {
1585 view->child->focussed = 1;
1586 view = view->child;
1587 } else
1588 view->focussed = 1;
1592 if (new_view) {
1593 struct tog_view *v, *t;
1594 /* Only allow one parent view per type. */
1595 TAILQ_FOREACH_SAFE(v, &views, entry, t) {
1596 if (v->type != new_view->type)
1597 continue;
1598 TAILQ_REMOVE(&views, v, entry);
1599 err = view_close(v);
1600 if (err)
1601 goto done;
1602 break;
1604 TAILQ_INSERT_TAIL(&views, new_view, entry);
1605 view = new_view;
1607 if (view) {
1608 if (view_is_parent_view(view)) {
1609 if (view->child && view->child->focussed)
1610 view = view->child;
1611 } else {
1612 if (view->parent && view->parent->focussed)
1613 view = view->parent;
1615 show_panel(view->panel);
1616 if (view->child && view_is_splitscreen(view->child))
1617 show_panel(view->child->panel);
1618 if (view->parent && view_is_splitscreen(view)) {
1619 err = view->parent->show(view->parent);
1620 if (err)
1621 goto done;
1623 err = view->show(view);
1624 if (err)
1625 goto done;
1626 if (view->child) {
1627 err = view->child->show(view->child);
1628 if (err)
1629 goto done;
1631 update_panels();
1632 doupdate();
1635 done:
1636 while (!TAILQ_EMPTY(&views)) {
1637 const struct got_error *close_err;
1638 view = TAILQ_FIRST(&views);
1639 TAILQ_REMOVE(&views, view, entry);
1640 close_err = view_close(view);
1641 if (close_err && err == NULL)
1642 err = close_err;
1645 errcode = pthread_mutex_unlock(&tog_mutex);
1646 if (errcode && err == NULL)
1647 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
1649 return err;
1652 __dead static void
1653 usage_log(void)
1655 endwin();
1656 fprintf(stderr,
1657 "usage: %s log [-b] [-c commit] [-r repository-path] [path]\n",
1658 getprogname());
1659 exit(1);
1662 /* Create newly allocated wide-character string equivalent to a byte string. */
1663 static const struct got_error *
1664 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
1666 char *vis = NULL;
1667 const struct got_error *err = NULL;
1669 *ws = NULL;
1670 *wlen = mbstowcs(NULL, s, 0);
1671 if (*wlen == (size_t)-1) {
1672 int vislen;
1673 if (errno != EILSEQ)
1674 return got_error_from_errno("mbstowcs");
1676 /* byte string invalid in current encoding; try to "fix" it */
1677 err = got_mbsavis(&vis, &vislen, s);
1678 if (err)
1679 return err;
1680 *wlen = mbstowcs(NULL, vis, 0);
1681 if (*wlen == (size_t)-1) {
1682 err = got_error_from_errno("mbstowcs"); /* give up */
1683 goto done;
1687 *ws = calloc(*wlen + 1, sizeof(**ws));
1688 if (*ws == NULL) {
1689 err = got_error_from_errno("calloc");
1690 goto done;
1693 if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
1694 err = got_error_from_errno("mbstowcs");
1695 done:
1696 free(vis);
1697 if (err) {
1698 free(*ws);
1699 *ws = NULL;
1700 *wlen = 0;
1702 return err;
1705 static const struct got_error *
1706 expand_tab(char **ptr, const char *src)
1708 char *dst;
1709 size_t len, n, idx = 0, sz = 0;
1711 *ptr = NULL;
1712 n = len = strlen(src);
1713 dst = malloc(n + 1);
1714 if (dst == NULL)
1715 return got_error_from_errno("malloc");
1717 while (idx < len && src[idx]) {
1718 const char c = src[idx];
1720 if (c == '\t') {
1721 size_t nb = TABSIZE - sz % TABSIZE;
1722 char *p;
1724 p = realloc(dst, n + nb);
1725 if (p == NULL) {
1726 free(dst);
1727 return got_error_from_errno("realloc");
1730 dst = p;
1731 n += nb;
1732 memset(dst + sz, ' ', nb);
1733 sz += nb;
1734 } else
1735 dst[sz++] = src[idx];
1736 ++idx;
1739 dst[sz] = '\0';
1740 *ptr = dst;
1741 return NULL;
1745 * Advance at most n columns from wline starting at offset off.
1746 * Return the index to the first character after the span operation.
1747 * Return the combined column width of all spanned wide character in
1748 * *rcol.
1750 static int
1751 span_wline(int *rcol, int off, wchar_t *wline, int n, int col_tab_align)
1753 int width, i, cols = 0;
1755 if (n == 0) {
1756 *rcol = cols;
1757 return off;
1760 for (i = off; wline[i] != L'\0'; ++i) {
1761 if (wline[i] == L'\t')
1762 width = TABSIZE - ((cols + col_tab_align) % TABSIZE);
1763 else
1764 width = wcwidth(wline[i]);
1766 if (width == -1) {
1767 width = 1;
1768 wline[i] = L'.';
1771 if (cols + width > n)
1772 break;
1773 cols += width;
1776 *rcol = cols;
1777 return i;
1781 * Format a line for display, ensuring that it won't overflow a width limit.
1782 * With scrolling, the width returned refers to the scrolled version of the
1783 * line, which starts at (*wlinep)[*scrollxp]. The caller must free *wlinep.
1785 static const struct got_error *
1786 format_line(wchar_t **wlinep, int *widthp, int *scrollxp,
1787 const char *line, int nscroll, int wlimit, int col_tab_align, int expand)
1789 const struct got_error *err = NULL;
1790 int cols;
1791 wchar_t *wline = NULL;
1792 char *exstr = NULL;
1793 size_t wlen;
1794 int i, scrollx;
1796 *wlinep = NULL;
1797 *widthp = 0;
1799 if (expand) {
1800 err = expand_tab(&exstr, line);
1801 if (err)
1802 return err;
1805 err = mbs2ws(&wline, &wlen, expand ? exstr : line);
1806 free(exstr);
1807 if (err)
1808 return err;
1810 scrollx = span_wline(&cols, 0, wline, nscroll, col_tab_align);
1812 if (wlen > 0 && wline[wlen - 1] == L'\n') {
1813 wline[wlen - 1] = L'\0';
1814 wlen--;
1816 if (wlen > 0 && wline[wlen - 1] == L'\r') {
1817 wline[wlen - 1] = L'\0';
1818 wlen--;
1821 i = span_wline(&cols, scrollx, wline, wlimit, col_tab_align);
1822 wline[i] = L'\0';
1824 if (widthp)
1825 *widthp = cols;
1826 if (scrollxp)
1827 *scrollxp = scrollx;
1828 if (err)
1829 free(wline);
1830 else
1831 *wlinep = wline;
1832 return err;
1835 static const struct got_error*
1836 build_refs_str(char **refs_str, struct got_reflist_head *refs,
1837 struct got_object_id *id, struct got_repository *repo)
1839 static const struct got_error *err = NULL;
1840 struct got_reflist_entry *re;
1841 char *s;
1842 const char *name;
1844 *refs_str = NULL;
1846 TAILQ_FOREACH(re, refs, entry) {
1847 struct got_tag_object *tag = NULL;
1848 struct got_object_id *ref_id;
1849 int cmp;
1851 name = got_ref_get_name(re->ref);
1852 if (strcmp(name, GOT_REF_HEAD) == 0)
1853 continue;
1854 if (strncmp(name, "refs/", 5) == 0)
1855 name += 5;
1856 if (strncmp(name, "got/", 4) == 0 &&
1857 strncmp(name, "got/backup/", 11) != 0)
1858 continue;
1859 if (strncmp(name, "heads/", 6) == 0)
1860 name += 6;
1861 if (strncmp(name, "remotes/", 8) == 0) {
1862 name += 8;
1863 s = strstr(name, "/" GOT_REF_HEAD);
1864 if (s != NULL && s[strlen(s)] == '\0')
1865 continue;
1867 err = got_ref_resolve(&ref_id, repo, re->ref);
1868 if (err)
1869 break;
1870 if (strncmp(name, "tags/", 5) == 0) {
1871 err = got_object_open_as_tag(&tag, repo, ref_id);
1872 if (err) {
1873 if (err->code != GOT_ERR_OBJ_TYPE) {
1874 free(ref_id);
1875 break;
1877 /* Ref points at something other than a tag. */
1878 err = NULL;
1879 tag = NULL;
1882 cmp = got_object_id_cmp(tag ?
1883 got_object_tag_get_object_id(tag) : ref_id, id);
1884 free(ref_id);
1885 if (tag)
1886 got_object_tag_close(tag);
1887 if (cmp != 0)
1888 continue;
1889 s = *refs_str;
1890 if (asprintf(refs_str, "%s%s%s", s ? s : "",
1891 s ? ", " : "", name) == -1) {
1892 err = got_error_from_errno("asprintf");
1893 free(s);
1894 *refs_str = NULL;
1895 break;
1897 free(s);
1900 return err;
1903 static const struct got_error *
1904 format_author(wchar_t **wauthor, int *author_width, char *author, int limit,
1905 int col_tab_align)
1907 char *smallerthan;
1909 smallerthan = strchr(author, '<');
1910 if (smallerthan && smallerthan[1] != '\0')
1911 author = smallerthan + 1;
1912 author[strcspn(author, "@>")] = '\0';
1913 return format_line(wauthor, author_width, NULL, author, 0, limit,
1914 col_tab_align, 0);
1917 static const struct got_error *
1918 draw_commit(struct tog_view *view, struct got_commit_object *commit,
1919 struct got_object_id *id, const size_t date_display_cols,
1920 int author_display_cols)
1922 struct tog_log_view_state *s = &view->state.log;
1923 const struct got_error *err = NULL;
1924 char datebuf[12]; /* YYYY-MM-DD + SPACE + NUL */
1925 char *logmsg0 = NULL, *logmsg = NULL;
1926 char *author = NULL;
1927 wchar_t *wlogmsg = NULL, *wauthor = NULL;
1928 int author_width, logmsg_width;
1929 char *newline, *line = NULL;
1930 int col, limit, scrollx;
1931 const int avail = view->ncols;
1932 struct tm tm;
1933 time_t committer_time;
1934 struct tog_color *tc;
1936 committer_time = got_object_commit_get_committer_time(commit);
1937 if (gmtime_r(&committer_time, &tm) == NULL)
1938 return got_error_from_errno("gmtime_r");
1939 if (strftime(datebuf, sizeof(datebuf), "%G-%m-%d ", &tm) == 0)
1940 return got_error(GOT_ERR_NO_SPACE);
1942 if (avail <= date_display_cols)
1943 limit = MIN(sizeof(datebuf) - 1, avail);
1944 else
1945 limit = MIN(date_display_cols, sizeof(datebuf) - 1);
1946 tc = get_color(&s->colors, TOG_COLOR_DATE);
1947 if (tc)
1948 wattr_on(view->window,
1949 COLOR_PAIR(tc->colorpair), NULL);
1950 waddnstr(view->window, datebuf, limit);
1951 if (tc)
1952 wattr_off(view->window,
1953 COLOR_PAIR(tc->colorpair), NULL);
1954 col = limit;
1955 if (col > avail)
1956 goto done;
1958 if (avail >= 120) {
1959 char *id_str;
1960 err = got_object_id_str(&id_str, id);
1961 if (err)
1962 goto done;
1963 tc = get_color(&s->colors, TOG_COLOR_COMMIT);
1964 if (tc)
1965 wattr_on(view->window,
1966 COLOR_PAIR(tc->colorpair), NULL);
1967 wprintw(view->window, "%.8s ", id_str);
1968 if (tc)
1969 wattr_off(view->window,
1970 COLOR_PAIR(tc->colorpair), NULL);
1971 free(id_str);
1972 col += 9;
1973 if (col > avail)
1974 goto done;
1977 if (s->use_committer)
1978 author = strdup(got_object_commit_get_committer(commit));
1979 else
1980 author = strdup(got_object_commit_get_author(commit));
1981 if (author == NULL) {
1982 err = got_error_from_errno("strdup");
1983 goto done;
1985 err = format_author(&wauthor, &author_width, author, avail - col, col);
1986 if (err)
1987 goto done;
1988 tc = get_color(&s->colors, TOG_COLOR_AUTHOR);
1989 if (tc)
1990 wattr_on(view->window,
1991 COLOR_PAIR(tc->colorpair), NULL);
1992 waddwstr(view->window, wauthor);
1993 if (tc)
1994 wattr_off(view->window,
1995 COLOR_PAIR(tc->colorpair), NULL);
1996 col += author_width;
1997 while (col < avail && author_width < author_display_cols + 2) {
1998 waddch(view->window, ' ');
1999 col++;
2000 author_width++;
2002 if (col > avail)
2003 goto done;
2005 err = got_object_commit_get_logmsg(&logmsg0, commit);
2006 if (err)
2007 goto done;
2008 logmsg = logmsg0;
2009 while (*logmsg == '\n')
2010 logmsg++;
2011 newline = strchr(logmsg, '\n');
2012 if (newline)
2013 *newline = '\0';
2014 limit = avail - col;
2015 if (view->child && !view_is_hsplit_top(view) && limit > 0)
2016 limit--; /* for the border */
2017 err = format_line(&wlogmsg, &logmsg_width, &scrollx, logmsg, view->x,
2018 limit, col, 1);
2019 if (err)
2020 goto done;
2021 waddwstr(view->window, &wlogmsg[scrollx]);
2022 col += MAX(logmsg_width, 0);
2023 while (col < avail) {
2024 waddch(view->window, ' ');
2025 col++;
2027 done:
2028 free(logmsg0);
2029 free(wlogmsg);
2030 free(author);
2031 free(wauthor);
2032 free(line);
2033 return err;
2036 static struct commit_queue_entry *
2037 alloc_commit_queue_entry(struct got_commit_object *commit,
2038 struct got_object_id *id)
2040 struct commit_queue_entry *entry;
2042 entry = calloc(1, sizeof(*entry));
2043 if (entry == NULL)
2044 return NULL;
2046 entry->id = id;
2047 entry->commit = commit;
2048 return entry;
2051 static void
2052 pop_commit(struct commit_queue *commits)
2054 struct commit_queue_entry *entry;
2056 entry = TAILQ_FIRST(&commits->head);
2057 TAILQ_REMOVE(&commits->head, entry, entry);
2058 got_object_commit_close(entry->commit);
2059 commits->ncommits--;
2060 /* Don't free entry->id! It is owned by the commit graph. */
2061 free(entry);
2064 static void
2065 free_commits(struct commit_queue *commits)
2067 while (!TAILQ_EMPTY(&commits->head))
2068 pop_commit(commits);
2071 static const struct got_error *
2072 match_commit(int *have_match, struct got_object_id *id,
2073 struct got_commit_object *commit, regex_t *regex)
2075 const struct got_error *err = NULL;
2076 regmatch_t regmatch;
2077 char *id_str = NULL, *logmsg = NULL;
2079 *have_match = 0;
2081 err = got_object_id_str(&id_str, id);
2082 if (err)
2083 return err;
2085 err = got_object_commit_get_logmsg(&logmsg, commit);
2086 if (err)
2087 goto done;
2089 if (regexec(regex, got_object_commit_get_author(commit), 1,
2090 &regmatch, 0) == 0 ||
2091 regexec(regex, got_object_commit_get_committer(commit), 1,
2092 &regmatch, 0) == 0 ||
2093 regexec(regex, id_str, 1, &regmatch, 0) == 0 ||
2094 regexec(regex, logmsg, 1, &regmatch, 0) == 0)
2095 *have_match = 1;
2096 done:
2097 free(id_str);
2098 free(logmsg);
2099 return err;
2102 static const struct got_error *
2103 queue_commits(struct tog_log_thread_args *a)
2105 const struct got_error *err = NULL;
2108 * We keep all commits open throughout the lifetime of the log
2109 * view in order to avoid having to re-fetch commits from disk
2110 * while updating the display.
2112 do {
2113 struct got_object_id *id;
2114 struct got_commit_object *commit;
2115 struct commit_queue_entry *entry;
2116 int errcode;
2118 err = got_commit_graph_iter_next(&id, a->graph, a->repo,
2119 NULL, NULL);
2120 if (err || id == NULL)
2121 break;
2123 err = got_object_open_as_commit(&commit, a->repo, id);
2124 if (err)
2125 break;
2126 entry = alloc_commit_queue_entry(commit, id);
2127 if (entry == NULL) {
2128 err = got_error_from_errno("alloc_commit_queue_entry");
2129 break;
2132 errcode = pthread_mutex_lock(&tog_mutex);
2133 if (errcode) {
2134 err = got_error_set_errno(errcode,
2135 "pthread_mutex_lock");
2136 break;
2139 entry->idx = a->commits->ncommits;
2140 TAILQ_INSERT_TAIL(&a->commits->head, entry, entry);
2141 a->commits->ncommits++;
2143 if (*a->searching == TOG_SEARCH_FORWARD &&
2144 !*a->search_next_done) {
2145 int have_match;
2146 err = match_commit(&have_match, id, commit, a->regex);
2147 if (err)
2148 break;
2149 if (have_match)
2150 *a->search_next_done = TOG_SEARCH_HAVE_MORE;
2153 errcode = pthread_mutex_unlock(&tog_mutex);
2154 if (errcode && err == NULL)
2155 err = got_error_set_errno(errcode,
2156 "pthread_mutex_unlock");
2157 if (err)
2158 break;
2159 } while (*a->searching == TOG_SEARCH_FORWARD && !*a->search_next_done);
2161 return err;
2164 static void
2165 select_commit(struct tog_log_view_state *s)
2167 struct commit_queue_entry *entry;
2168 int ncommits = 0;
2170 entry = s->first_displayed_entry;
2171 while (entry) {
2172 if (ncommits == s->selected) {
2173 s->selected_entry = entry;
2174 break;
2176 entry = TAILQ_NEXT(entry, entry);
2177 ncommits++;
2181 static const struct got_error *
2182 draw_commits(struct tog_view *view)
2184 const struct got_error *err = NULL;
2185 struct tog_log_view_state *s = &view->state.log;
2186 struct commit_queue_entry *entry = s->selected_entry;
2187 const int limit = view->nlines;
2188 int width;
2189 int ncommits, author_cols = 4;
2190 char *id_str = NULL, *header = NULL, *ncommits_str = NULL;
2191 char *refs_str = NULL;
2192 wchar_t *wline;
2193 struct tog_color *tc;
2194 static const size_t date_display_cols = 12;
2196 if (s->selected_entry &&
2197 !(view->searching && view->search_next_done == 0)) {
2198 struct got_reflist_head *refs;
2199 err = got_object_id_str(&id_str, s->selected_entry->id);
2200 if (err)
2201 return err;
2202 refs = got_reflist_object_id_map_lookup(tog_refs_idmap,
2203 s->selected_entry->id);
2204 if (refs) {
2205 err = build_refs_str(&refs_str, refs,
2206 s->selected_entry->id, s->repo);
2207 if (err)
2208 goto done;
2212 if (s->thread_args.commits_needed == 0)
2213 halfdelay(10); /* disable fast refresh */
2215 if (s->thread_args.commits_needed > 0 || s->thread_args.load_all) {
2216 if (asprintf(&ncommits_str, " [%d/%d] %s",
2217 entry ? entry->idx + 1 : 0, s->commits.ncommits,
2218 (view->searching && !view->search_next_done) ?
2219 "searching..." : "loading...") == -1) {
2220 err = got_error_from_errno("asprintf");
2221 goto done;
2223 } else {
2224 const char *search_str = NULL;
2226 if (view->searching) {
2227 if (view->search_next_done == TOG_SEARCH_NO_MORE)
2228 search_str = "no more matches";
2229 else if (view->search_next_done == TOG_SEARCH_HAVE_NONE)
2230 search_str = "no matches found";
2231 else if (!view->search_next_done)
2232 search_str = "searching...";
2235 if (asprintf(&ncommits_str, " [%d/%d] %s",
2236 entry ? entry->idx + 1 : 0, s->commits.ncommits,
2237 search_str ? search_str :
2238 (refs_str ? refs_str : "")) == -1) {
2239 err = got_error_from_errno("asprintf");
2240 goto done;
2244 if (s->in_repo_path && strcmp(s->in_repo_path, "/") != 0) {
2245 if (asprintf(&header, "commit %s %s%s", id_str ? id_str :
2246 "........................................",
2247 s->in_repo_path, ncommits_str) == -1) {
2248 err = got_error_from_errno("asprintf");
2249 header = NULL;
2250 goto done;
2252 } else if (asprintf(&header, "commit %s%s",
2253 id_str ? id_str : "........................................",
2254 ncommits_str) == -1) {
2255 err = got_error_from_errno("asprintf");
2256 header = NULL;
2257 goto done;
2259 err = format_line(&wline, &width, NULL, header, 0, view->ncols, 0, 0);
2260 if (err)
2261 goto done;
2263 werase(view->window);
2265 if (view_needs_focus_indication(view))
2266 wstandout(view->window);
2267 tc = get_color(&s->colors, TOG_COLOR_COMMIT);
2268 if (tc)
2269 wattr_on(view->window,
2270 COLOR_PAIR(tc->colorpair), NULL);
2271 waddwstr(view->window, wline);
2272 if (tc)
2273 wattr_off(view->window,
2274 COLOR_PAIR(tc->colorpair), NULL);
2275 while (width < view->ncols) {
2276 waddch(view->window, ' ');
2277 width++;
2279 if (view_needs_focus_indication(view))
2280 wstandend(view->window);
2281 free(wline);
2282 if (limit <= 1)
2283 goto done;
2285 /* Grow author column size if necessary, and set view->maxx. */
2286 entry = s->first_displayed_entry;
2287 ncommits = 0;
2288 view->maxx = 0;
2289 while (entry) {
2290 struct got_commit_object *c = entry->commit;
2291 char *author, *eol, *msg, *msg0;
2292 wchar_t *wauthor, *wmsg;
2293 int width;
2294 if (ncommits >= limit - 1)
2295 break;
2296 if (s->use_committer)
2297 author = strdup(got_object_commit_get_committer(c));
2298 else
2299 author = strdup(got_object_commit_get_author(c));
2300 if (author == NULL) {
2301 err = got_error_from_errno("strdup");
2302 goto done;
2304 err = format_author(&wauthor, &width, author, COLS,
2305 date_display_cols);
2306 if (author_cols < width)
2307 author_cols = width;
2308 free(wauthor);
2309 free(author);
2310 if (err)
2311 goto done;
2312 err = got_object_commit_get_logmsg(&msg0, c);
2313 if (err)
2314 goto done;
2315 msg = msg0;
2316 while (*msg == '\n')
2317 ++msg;
2318 if ((eol = strchr(msg, '\n')))
2319 *eol = '\0';
2320 err = format_line(&wmsg, &width, NULL, msg, 0, INT_MAX,
2321 date_display_cols + author_cols, 0);
2322 if (err)
2323 goto done;
2324 view->maxx = MAX(view->maxx, width);
2325 free(msg0);
2326 free(wmsg);
2327 ncommits++;
2328 entry = TAILQ_NEXT(entry, entry);
2331 entry = s->first_displayed_entry;
2332 s->last_displayed_entry = s->first_displayed_entry;
2333 ncommits = 0;
2334 while (entry) {
2335 if (ncommits >= limit - 1)
2336 break;
2337 if (ncommits == s->selected)
2338 wstandout(view->window);
2339 err = draw_commit(view, entry->commit, entry->id,
2340 date_display_cols, author_cols);
2341 if (ncommits == s->selected)
2342 wstandend(view->window);
2343 if (err)
2344 goto done;
2345 ncommits++;
2346 s->last_displayed_entry = entry;
2347 entry = TAILQ_NEXT(entry, entry);
2350 view_border(view);
2351 done:
2352 free(id_str);
2353 free(refs_str);
2354 free(ncommits_str);
2355 free(header);
2356 return err;
2359 static void
2360 log_scroll_up(struct tog_log_view_state *s, int maxscroll)
2362 struct commit_queue_entry *entry;
2363 int nscrolled = 0;
2365 entry = TAILQ_FIRST(&s->commits.head);
2366 if (s->first_displayed_entry == entry)
2367 return;
2369 entry = s->first_displayed_entry;
2370 while (entry && nscrolled < maxscroll) {
2371 entry = TAILQ_PREV(entry, commit_queue_head, entry);
2372 if (entry) {
2373 s->first_displayed_entry = entry;
2374 nscrolled++;
2379 static const struct got_error *
2380 trigger_log_thread(struct tog_view *view, int wait)
2382 struct tog_log_thread_args *ta = &view->state.log.thread_args;
2383 int errcode;
2385 halfdelay(1); /* fast refresh while loading commits */
2387 while (!ta->log_complete && !tog_thread_error &&
2388 (ta->commits_needed > 0 || ta->load_all)) {
2389 /* Wake the log thread. */
2390 errcode = pthread_cond_signal(&ta->need_commits);
2391 if (errcode)
2392 return got_error_set_errno(errcode,
2393 "pthread_cond_signal");
2396 * The mutex will be released while the view loop waits
2397 * in wgetch(), at which time the log thread will run.
2399 if (!wait)
2400 break;
2402 /* Display progress update in log view. */
2403 show_log_view(view);
2404 update_panels();
2405 doupdate();
2407 /* Wait right here while next commit is being loaded. */
2408 errcode = pthread_cond_wait(&ta->commit_loaded, &tog_mutex);
2409 if (errcode)
2410 return got_error_set_errno(errcode,
2411 "pthread_cond_wait");
2413 /* Display progress update in log view. */
2414 show_log_view(view);
2415 update_panels();
2416 doupdate();
2419 return NULL;
2422 static const struct got_error *
2423 request_log_commits(struct tog_view *view)
2425 struct tog_log_view_state *state = &view->state.log;
2426 const struct got_error *err = NULL;
2428 if (state->thread_args.log_complete)
2429 return NULL;
2431 state->thread_args.commits_needed += view->nscrolled;
2432 err = trigger_log_thread(view, 1);
2433 view->nscrolled = 0;
2435 return err;
2438 static const struct got_error *
2439 log_scroll_down(struct tog_view *view, int maxscroll)
2441 struct tog_log_view_state *s = &view->state.log;
2442 const struct got_error *err = NULL;
2443 struct commit_queue_entry *pentry;
2444 int nscrolled = 0, ncommits_needed;
2446 if (s->last_displayed_entry == NULL)
2447 return NULL;
2449 ncommits_needed = s->last_displayed_entry->idx + 1 + maxscroll;
2450 if (s->commits.ncommits < ncommits_needed &&
2451 !s->thread_args.log_complete) {
2453 * Ask the log thread for required amount of commits.
2455 s->thread_args.commits_needed += maxscroll;
2456 err = trigger_log_thread(view, 1);
2457 if (err)
2458 return err;
2461 do {
2462 pentry = TAILQ_NEXT(s->last_displayed_entry, entry);
2463 if (pentry == NULL && view->mode != TOG_VIEW_SPLIT_HRZN)
2464 break;
2466 s->last_displayed_entry = pentry ?
2467 pentry : s->last_displayed_entry;;
2469 pentry = TAILQ_NEXT(s->first_displayed_entry, entry);
2470 if (pentry == NULL)
2471 break;
2472 s->first_displayed_entry = pentry;
2473 } while (++nscrolled < maxscroll);
2475 if (view->mode == TOG_VIEW_SPLIT_HRZN && !s->thread_args.log_complete)
2476 view->nscrolled += nscrolled;
2477 else
2478 view->nscrolled = 0;
2480 return err;
2483 static const struct got_error *
2484 open_diff_view_for_commit(struct tog_view **new_view, int begin_y, int begin_x,
2485 struct got_commit_object *commit, struct got_object_id *commit_id,
2486 struct tog_view *log_view, struct got_repository *repo)
2488 const struct got_error *err;
2489 struct got_object_qid *parent_id;
2490 struct tog_view *diff_view;
2492 diff_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_DIFF);
2493 if (diff_view == NULL)
2494 return got_error_from_errno("view_open");
2496 parent_id = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
2497 err = open_diff_view(diff_view, parent_id ? &parent_id->id : NULL,
2498 commit_id, NULL, NULL, 3, 0, 0, log_view, repo);
2499 if (err == NULL)
2500 *new_view = diff_view;
2501 return err;
2504 static const struct got_error *
2505 tree_view_visit_subtree(struct tog_tree_view_state *s,
2506 struct got_tree_object *subtree)
2508 struct tog_parent_tree *parent;
2510 parent = calloc(1, sizeof(*parent));
2511 if (parent == NULL)
2512 return got_error_from_errno("calloc");
2514 parent->tree = s->tree;
2515 parent->first_displayed_entry = s->first_displayed_entry;
2516 parent->selected_entry = s->selected_entry;
2517 parent->selected = s->selected;
2518 TAILQ_INSERT_HEAD(&s->parents, parent, entry);
2519 s->tree = subtree;
2520 s->selected = 0;
2521 s->first_displayed_entry = NULL;
2522 return NULL;
2525 static const struct got_error *
2526 tree_view_walk_path(struct tog_tree_view_state *s,
2527 struct got_commit_object *commit, const char *path)
2529 const struct got_error *err = NULL;
2530 struct got_tree_object *tree = NULL;
2531 const char *p;
2532 char *slash, *subpath = NULL;
2534 /* Walk the path and open corresponding tree objects. */
2535 p = path;
2536 while (*p) {
2537 struct got_tree_entry *te;
2538 struct got_object_id *tree_id;
2539 char *te_name;
2541 while (p[0] == '/')
2542 p++;
2544 /* Ensure the correct subtree entry is selected. */
2545 slash = strchr(p, '/');
2546 if (slash == NULL)
2547 te_name = strdup(p);
2548 else
2549 te_name = strndup(p, slash - p);
2550 if (te_name == NULL) {
2551 err = got_error_from_errno("strndup");
2552 break;
2554 te = got_object_tree_find_entry(s->tree, te_name);
2555 if (te == NULL) {
2556 err = got_error_path(te_name, GOT_ERR_NO_TREE_ENTRY);
2557 free(te_name);
2558 break;
2560 free(te_name);
2561 s->first_displayed_entry = s->selected_entry = te;
2563 if (!S_ISDIR(got_tree_entry_get_mode(s->selected_entry)))
2564 break; /* jump to this file's entry */
2566 slash = strchr(p, '/');
2567 if (slash)
2568 subpath = strndup(path, slash - path);
2569 else
2570 subpath = strdup(path);
2571 if (subpath == NULL) {
2572 err = got_error_from_errno("strdup");
2573 break;
2576 err = got_object_id_by_path(&tree_id, s->repo, commit,
2577 subpath);
2578 if (err)
2579 break;
2581 err = got_object_open_as_tree(&tree, s->repo, tree_id);
2582 free(tree_id);
2583 if (err)
2584 break;
2586 err = tree_view_visit_subtree(s, tree);
2587 if (err) {
2588 got_object_tree_close(tree);
2589 break;
2591 if (slash == NULL)
2592 break;
2593 free(subpath);
2594 subpath = NULL;
2595 p = slash;
2598 free(subpath);
2599 return err;
2602 static const struct got_error *
2603 browse_commit_tree(struct tog_view **new_view, int begin_y, int begin_x,
2604 struct commit_queue_entry *entry, const char *path,
2605 const char *head_ref_name, struct got_repository *repo)
2607 const struct got_error *err = NULL;
2608 struct tog_tree_view_state *s;
2609 struct tog_view *tree_view;
2611 tree_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_TREE);
2612 if (tree_view == NULL)
2613 return got_error_from_errno("view_open");
2615 err = open_tree_view(tree_view, entry->id, head_ref_name, repo);
2616 if (err)
2617 return err;
2618 s = &tree_view->state.tree;
2620 *new_view = tree_view;
2622 if (got_path_is_root_dir(path))
2623 return NULL;
2625 return tree_view_walk_path(s, entry->commit, path);
2628 static const struct got_error *
2629 block_signals_used_by_main_thread(void)
2631 sigset_t sigset;
2632 int errcode;
2634 if (sigemptyset(&sigset) == -1)
2635 return got_error_from_errno("sigemptyset");
2637 /* tog handles SIGWINCH, SIGCONT, SIGINT, SIGTERM */
2638 if (sigaddset(&sigset, SIGWINCH) == -1)
2639 return got_error_from_errno("sigaddset");
2640 if (sigaddset(&sigset, SIGCONT) == -1)
2641 return got_error_from_errno("sigaddset");
2642 if (sigaddset(&sigset, SIGINT) == -1)
2643 return got_error_from_errno("sigaddset");
2644 if (sigaddset(&sigset, SIGTERM) == -1)
2645 return got_error_from_errno("sigaddset");
2647 /* ncurses handles SIGTSTP */
2648 if (sigaddset(&sigset, SIGTSTP) == -1)
2649 return got_error_from_errno("sigaddset");
2651 errcode = pthread_sigmask(SIG_BLOCK, &sigset, NULL);
2652 if (errcode)
2653 return got_error_set_errno(errcode, "pthread_sigmask");
2655 return NULL;
2658 static void *
2659 log_thread(void *arg)
2661 const struct got_error *err = NULL;
2662 int errcode = 0;
2663 struct tog_log_thread_args *a = arg;
2664 int done = 0;
2667 * Sync startup with main thread such that we begin our
2668 * work once view_input() has released the mutex.
2670 errcode = pthread_mutex_lock(&tog_mutex);
2671 if (errcode) {
2672 err = got_error_set_errno(errcode, "pthread_mutex_lock");
2673 return (void *)err;
2676 err = block_signals_used_by_main_thread();
2677 if (err) {
2678 pthread_mutex_unlock(&tog_mutex);
2679 goto done;
2682 while (!done && !err && !tog_fatal_signal_received()) {
2683 errcode = pthread_mutex_unlock(&tog_mutex);
2684 if (errcode) {
2685 err = got_error_set_errno(errcode,
2686 "pthread_mutex_unlock");
2687 goto done;
2689 err = queue_commits(a);
2690 if (err) {
2691 if (err->code != GOT_ERR_ITER_COMPLETED)
2692 goto done;
2693 err = NULL;
2694 done = 1;
2695 } else if (a->commits_needed > 0 && !a->load_all)
2696 a->commits_needed--;
2698 errcode = pthread_mutex_lock(&tog_mutex);
2699 if (errcode) {
2700 err = got_error_set_errno(errcode,
2701 "pthread_mutex_lock");
2702 goto done;
2703 } else if (*a->quit)
2704 done = 1;
2705 else if (*a->first_displayed_entry == NULL) {
2706 *a->first_displayed_entry =
2707 TAILQ_FIRST(&a->commits->head);
2708 *a->selected_entry = *a->first_displayed_entry;
2711 errcode = pthread_cond_signal(&a->commit_loaded);
2712 if (errcode) {
2713 err = got_error_set_errno(errcode,
2714 "pthread_cond_signal");
2715 pthread_mutex_unlock(&tog_mutex);
2716 goto done;
2719 if (done)
2720 a->commits_needed = 0;
2721 else {
2722 if (a->commits_needed == 0 && !a->load_all) {
2723 errcode = pthread_cond_wait(&a->need_commits,
2724 &tog_mutex);
2725 if (errcode) {
2726 err = got_error_set_errno(errcode,
2727 "pthread_cond_wait");
2728 pthread_mutex_unlock(&tog_mutex);
2729 goto done;
2731 if (*a->quit)
2732 done = 1;
2736 a->log_complete = 1;
2737 errcode = pthread_mutex_unlock(&tog_mutex);
2738 if (errcode)
2739 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
2740 done:
2741 if (err) {
2742 tog_thread_error = 1;
2743 pthread_cond_signal(&a->commit_loaded);
2745 return (void *)err;
2748 static const struct got_error *
2749 stop_log_thread(struct tog_log_view_state *s)
2751 const struct got_error *err = NULL, *thread_err = NULL;
2752 int errcode;
2754 if (s->thread) {
2755 s->quit = 1;
2756 errcode = pthread_cond_signal(&s->thread_args.need_commits);
2757 if (errcode)
2758 return got_error_set_errno(errcode,
2759 "pthread_cond_signal");
2760 errcode = pthread_mutex_unlock(&tog_mutex);
2761 if (errcode)
2762 return got_error_set_errno(errcode,
2763 "pthread_mutex_unlock");
2764 errcode = pthread_join(s->thread, (void **)&thread_err);
2765 if (errcode)
2766 return got_error_set_errno(errcode, "pthread_join");
2767 errcode = pthread_mutex_lock(&tog_mutex);
2768 if (errcode)
2769 return got_error_set_errno(errcode,
2770 "pthread_mutex_lock");
2771 s->thread = 0; //NULL;
2774 if (s->thread_args.repo) {
2775 err = got_repo_close(s->thread_args.repo);
2776 s->thread_args.repo = NULL;
2779 if (s->thread_args.pack_fds) {
2780 const struct got_error *pack_err =
2781 got_repo_pack_fds_close(s->thread_args.pack_fds);
2782 if (err == NULL)
2783 err = pack_err;
2784 s->thread_args.pack_fds = NULL;
2787 if (s->thread_args.graph) {
2788 got_commit_graph_close(s->thread_args.graph);
2789 s->thread_args.graph = NULL;
2792 return err ? err : thread_err;
2795 static const struct got_error *
2796 close_log_view(struct tog_view *view)
2798 const struct got_error *err = NULL;
2799 struct tog_log_view_state *s = &view->state.log;
2800 int errcode;
2802 err = stop_log_thread(s);
2804 errcode = pthread_cond_destroy(&s->thread_args.need_commits);
2805 if (errcode && err == NULL)
2806 err = got_error_set_errno(errcode, "pthread_cond_destroy");
2808 errcode = pthread_cond_destroy(&s->thread_args.commit_loaded);
2809 if (errcode && err == NULL)
2810 err = got_error_set_errno(errcode, "pthread_cond_destroy");
2812 free_commits(&s->commits);
2813 free(s->in_repo_path);
2814 s->in_repo_path = NULL;
2815 free(s->start_id);
2816 s->start_id = NULL;
2817 free(s->head_ref_name);
2818 s->head_ref_name = NULL;
2819 return err;
2822 static const struct got_error *
2823 search_start_log_view(struct tog_view *view)
2825 struct tog_log_view_state *s = &view->state.log;
2827 s->matched_entry = NULL;
2828 s->search_entry = NULL;
2829 return NULL;
2832 static const struct got_error *
2833 search_next_log_view(struct tog_view *view)
2835 const struct got_error *err = NULL;
2836 struct tog_log_view_state *s = &view->state.log;
2837 struct commit_queue_entry *entry;
2839 /* Display progress update in log view. */
2840 show_log_view(view);
2841 update_panels();
2842 doupdate();
2844 if (s->search_entry) {
2845 int errcode, ch;
2846 errcode = pthread_mutex_unlock(&tog_mutex);
2847 if (errcode)
2848 return got_error_set_errno(errcode,
2849 "pthread_mutex_unlock");
2850 ch = wgetch(view->window);
2851 errcode = pthread_mutex_lock(&tog_mutex);
2852 if (errcode)
2853 return got_error_set_errno(errcode,
2854 "pthread_mutex_lock");
2855 if (ch == CTRL('g') || ch == KEY_BACKSPACE) {
2856 view->search_next_done = TOG_SEARCH_HAVE_MORE;
2857 return NULL;
2859 if (view->searching == TOG_SEARCH_FORWARD)
2860 entry = TAILQ_NEXT(s->search_entry, entry);
2861 else
2862 entry = TAILQ_PREV(s->search_entry,
2863 commit_queue_head, entry);
2864 } else if (s->matched_entry) {
2865 int matched_idx = s->matched_entry->idx;
2866 int selected_idx = s->selected_entry->idx;
2869 * If the user has moved the cursor after we hit a match,
2870 * the position from where we should continue searching
2871 * might have changed.
2873 if (view->searching == TOG_SEARCH_FORWARD) {
2874 if (matched_idx > selected_idx)
2875 entry = TAILQ_NEXT(s->selected_entry, entry);
2876 else
2877 entry = TAILQ_NEXT(s->matched_entry, entry);
2878 } else {
2879 if (matched_idx < selected_idx)
2880 entry = TAILQ_PREV(s->selected_entry,
2881 commit_queue_head, entry);
2882 else
2883 entry = TAILQ_PREV(s->matched_entry,
2884 commit_queue_head, entry);
2886 } else {
2887 entry = s->selected_entry;
2890 while (1) {
2891 int have_match = 0;
2893 if (entry == NULL) {
2894 if (s->thread_args.log_complete ||
2895 view->searching == TOG_SEARCH_BACKWARD) {
2896 view->search_next_done =
2897 (s->matched_entry == NULL ?
2898 TOG_SEARCH_HAVE_NONE : TOG_SEARCH_NO_MORE);
2899 s->search_entry = NULL;
2900 return NULL;
2903 * Poke the log thread for more commits and return,
2904 * allowing the main loop to make progress. Search
2905 * will resume at s->search_entry once we come back.
2907 s->thread_args.commits_needed++;
2908 return trigger_log_thread(view, 0);
2911 err = match_commit(&have_match, entry->id, entry->commit,
2912 &view->regex);
2913 if (err)
2914 break;
2915 if (have_match) {
2916 view->search_next_done = TOG_SEARCH_HAVE_MORE;
2917 s->matched_entry = entry;
2918 break;
2921 s->search_entry = entry;
2922 if (view->searching == TOG_SEARCH_FORWARD)
2923 entry = TAILQ_NEXT(entry, entry);
2924 else
2925 entry = TAILQ_PREV(entry, commit_queue_head, entry);
2928 if (s->matched_entry) {
2929 int cur = s->selected_entry->idx;
2930 while (cur < s->matched_entry->idx) {
2931 err = input_log_view(NULL, view, KEY_DOWN);
2932 if (err)
2933 return err;
2934 cur++;
2936 while (cur > s->matched_entry->idx) {
2937 err = input_log_view(NULL, view, KEY_UP);
2938 if (err)
2939 return err;
2940 cur--;
2944 s->search_entry = NULL;
2946 return NULL;
2949 static const struct got_error *
2950 open_log_view(struct tog_view *view, struct got_object_id *start_id,
2951 struct got_repository *repo, const char *head_ref_name,
2952 const char *in_repo_path, int log_branches)
2954 const struct got_error *err = NULL;
2955 struct tog_log_view_state *s = &view->state.log;
2956 struct got_repository *thread_repo = NULL;
2957 struct got_commit_graph *thread_graph = NULL;
2958 int errcode;
2960 if (in_repo_path != s->in_repo_path) {
2961 free(s->in_repo_path);
2962 s->in_repo_path = strdup(in_repo_path);
2963 if (s->in_repo_path == NULL)
2964 return got_error_from_errno("strdup");
2967 /* The commit queue only contains commits being displayed. */
2968 TAILQ_INIT(&s->commits.head);
2969 s->commits.ncommits = 0;
2971 s->repo = repo;
2972 if (head_ref_name) {
2973 s->head_ref_name = strdup(head_ref_name);
2974 if (s->head_ref_name == NULL) {
2975 err = got_error_from_errno("strdup");
2976 goto done;
2979 s->start_id = got_object_id_dup(start_id);
2980 if (s->start_id == NULL) {
2981 err = got_error_from_errno("got_object_id_dup");
2982 goto done;
2984 s->log_branches = log_branches;
2986 STAILQ_INIT(&s->colors);
2987 if (has_colors() && getenv("TOG_COLORS") != NULL) {
2988 err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
2989 get_color_value("TOG_COLOR_COMMIT"));
2990 if (err)
2991 goto done;
2992 err = add_color(&s->colors, "^$", TOG_COLOR_AUTHOR,
2993 get_color_value("TOG_COLOR_AUTHOR"));
2994 if (err) {
2995 free_colors(&s->colors);
2996 goto done;
2998 err = add_color(&s->colors, "^$", TOG_COLOR_DATE,
2999 get_color_value("TOG_COLOR_DATE"));
3000 if (err) {
3001 free_colors(&s->colors);
3002 goto done;
3006 view->show = show_log_view;
3007 view->input = input_log_view;
3008 view->resize = resize_log_view;
3009 view->close = close_log_view;
3010 view->search_start = search_start_log_view;
3011 view->search_next = search_next_log_view;
3013 if (s->thread_args.pack_fds == NULL) {
3014 err = got_repo_pack_fds_open(&s->thread_args.pack_fds);
3015 if (err)
3016 goto done;
3018 err = got_repo_open(&thread_repo, got_repo_get_path(repo), NULL,
3019 s->thread_args.pack_fds);
3020 if (err)
3021 goto done;
3022 err = got_commit_graph_open(&thread_graph, s->in_repo_path,
3023 !s->log_branches);
3024 if (err)
3025 goto done;
3026 err = got_commit_graph_iter_start(thread_graph, s->start_id,
3027 s->repo, NULL, NULL);
3028 if (err)
3029 goto done;
3031 errcode = pthread_cond_init(&s->thread_args.need_commits, NULL);
3032 if (errcode) {
3033 err = got_error_set_errno(errcode, "pthread_cond_init");
3034 goto done;
3036 errcode = pthread_cond_init(&s->thread_args.commit_loaded, NULL);
3037 if (errcode) {
3038 err = got_error_set_errno(errcode, "pthread_cond_init");
3039 goto done;
3042 s->thread_args.commits_needed = view->nlines;
3043 s->thread_args.graph = thread_graph;
3044 s->thread_args.commits = &s->commits;
3045 s->thread_args.in_repo_path = s->in_repo_path;
3046 s->thread_args.start_id = s->start_id;
3047 s->thread_args.repo = thread_repo;
3048 s->thread_args.log_complete = 0;
3049 s->thread_args.quit = &s->quit;
3050 s->thread_args.first_displayed_entry = &s->first_displayed_entry;
3051 s->thread_args.selected_entry = &s->selected_entry;
3052 s->thread_args.searching = &view->searching;
3053 s->thread_args.search_next_done = &view->search_next_done;
3054 s->thread_args.regex = &view->regex;
3055 done:
3056 if (err)
3057 close_log_view(view);
3058 return err;
3061 static const struct got_error *
3062 show_log_view(struct tog_view *view)
3064 const struct got_error *err;
3065 struct tog_log_view_state *s = &view->state.log;
3067 if (s->thread == 0) { //NULL) {
3068 int errcode = pthread_create(&s->thread, NULL, log_thread,
3069 &s->thread_args);
3070 if (errcode)
3071 return got_error_set_errno(errcode, "pthread_create");
3072 if (s->thread_args.commits_needed > 0) {
3073 err = trigger_log_thread(view, 1);
3074 if (err)
3075 return err;
3079 return draw_commits(view);
3082 static void
3083 log_move_cursor_up(struct tog_view *view, int page, int home)
3085 struct tog_log_view_state *s = &view->state.log;
3087 if (s->selected_entry->idx == 0)
3088 view->count = 0;
3089 if (s->first_displayed_entry == NULL)
3090 return;
3092 if ((page && TAILQ_FIRST(&s->commits.head) == s->first_displayed_entry)
3093 || home)
3094 s->selected = home ? 0 : MAX(0, s->selected - page - 1);
3096 if (!page && !home && s->selected > 0)
3097 --s->selected;
3098 else
3099 log_scroll_up(s, home ? s->commits.ncommits : MAX(page, 1));
3101 select_commit(s);
3102 return;
3105 static const struct got_error *
3106 log_move_cursor_down(struct tog_view *view, int page)
3108 struct tog_log_view_state *s = &view->state.log;
3109 struct commit_queue_entry *first;
3110 const struct got_error *err = NULL;
3112 first = s->first_displayed_entry;
3113 if (first == NULL) {
3114 view->count = 0;
3115 return NULL;
3118 if (s->thread_args.log_complete &&
3119 s->selected_entry->idx >= s->commits.ncommits - 1)
3120 return NULL;
3122 if (!page) {
3123 int eos = view->nlines - 2;
3125 if (view_is_hsplit_top(view))
3126 --eos; /* border consumes the last line */
3127 if (s->selected < MIN(eos, s->commits.ncommits - 1))
3128 ++s->selected;
3129 else
3130 err = log_scroll_down(view, 1);
3131 } else if (s->thread_args.load_all) {
3132 if (s->last_displayed_entry->idx == s->commits.ncommits - 1)
3133 s->selected += MIN(s->last_displayed_entry->idx -
3134 s->selected_entry->idx, page + 1);
3135 else
3136 err = log_scroll_down(view, MIN(page,
3137 s->commits.ncommits - s->selected_entry->idx - 1));
3138 s->selected = MIN(view->nlines - 2, s->commits.ncommits - 1);
3139 } else {
3140 err = log_scroll_down(view, page);
3141 if (err)
3142 return err;
3143 if (first == s->first_displayed_entry && s->selected <
3144 MIN(view->nlines - 2, s->commits.ncommits - 1)) {
3145 s->selected = MIN(s->commits.ncommits - 1, page);
3148 if (err)
3149 return err;
3152 * We might necessarily overshoot in horizontal
3153 * splits; if so, select the last displayed commit.
3155 s->selected = MIN(s->selected,
3156 s->last_displayed_entry->idx - s->first_displayed_entry->idx);
3158 select_commit(s);
3160 if (s->thread_args.log_complete &&
3161 s->selected_entry->idx == s->commits.ncommits - 1)
3162 view->count = 0;
3164 return NULL;
3167 static void
3168 view_get_split(struct tog_view *view, int *y, int *x)
3170 *x = 0;
3171 *y = 0;
3173 if (view->mode == TOG_VIEW_SPLIT_HRZN) {
3174 if (view->child && view->child->resized_y)
3175 *y = view->child->resized_y;
3176 else if (view->resized_y)
3177 *y = view->resized_y;
3178 else
3179 *y = view_split_begin_y(view->lines);
3180 } else if (view->mode == TOG_VIEW_SPLIT_VERT) {
3181 if (view->child && view->child->resized_x)
3182 *x = view->child->resized_x;
3183 else if (view->resized_x)
3184 *x = view->resized_x;
3185 else
3186 *x = view_split_begin_x(view->begin_x);
3190 /* Split view horizontally at y and offset view->state->selected line. */
3191 static const struct got_error *
3192 view_init_hsplit(struct tog_view *view, int y)
3194 const struct got_error *err = NULL;
3196 view->nlines = y;
3197 view->ncols = COLS;
3198 err = view_resize(view);
3199 if (err)
3200 return err;
3202 err = offset_selection_down(view);
3204 return err;
3207 static const struct got_error *
3208 input_log_view(struct tog_view **new_view, struct tog_view *view, int ch)
3210 const struct got_error *err = NULL;
3211 struct tog_log_view_state *s = &view->state.log;
3212 struct tog_view *diff_view = NULL, *tree_view = NULL;
3213 struct tog_view *ref_view = NULL;
3214 struct commit_queue_entry *entry;
3215 int begin_x = 0, begin_y = 0, eos, n, nscroll;
3217 if (s->thread_args.load_all) {
3218 if (ch == CTRL('g') || ch == KEY_BACKSPACE)
3219 s->thread_args.load_all = 0;
3220 else if (s->thread_args.log_complete) {
3221 err = log_move_cursor_down(view, s->commits.ncommits);
3222 s->thread_args.load_all = 0;
3224 return err;
3227 eos = nscroll = view->nlines - 1;
3228 if (view_is_hsplit_top(view))
3229 --eos; /* border */
3231 switch (ch) {
3232 case 'q':
3233 s->quit = 1;
3234 break;
3235 case '0':
3236 view->x = 0;
3237 break;
3238 case '$':
3239 view->x = MAX(view->maxx - view->ncols / 2, 0);
3240 view->count = 0;
3241 break;
3242 case KEY_RIGHT:
3243 case 'l':
3244 if (view->x + view->ncols / 2 < view->maxx)
3245 view->x += 2; /* move two columns right */
3246 else
3247 view->count = 0;
3248 break;
3249 case KEY_LEFT:
3250 case 'h':
3251 view->x -= MIN(view->x, 2); /* move two columns back */
3252 if (view->x <= 0)
3253 view->count = 0;
3254 break;
3255 case 'k':
3256 case KEY_UP:
3257 case '<':
3258 case ',':
3259 case CTRL('p'):
3260 log_move_cursor_up(view, 0, 0);
3261 break;
3262 case 'g':
3263 case KEY_HOME:
3264 log_move_cursor_up(view, 0, 1);
3265 view->count = 0;
3266 break;
3267 case CTRL('u'):
3268 case 'u':
3269 nscroll /= 2;
3270 /* FALL THROUGH */
3271 case KEY_PPAGE:
3272 case CTRL('b'):
3273 case 'b':
3274 log_move_cursor_up(view, nscroll, 0);
3275 break;
3276 case 'j':
3277 case KEY_DOWN:
3278 case '>':
3279 case '.':
3280 case CTRL('n'):
3281 err = log_move_cursor_down(view, 0);
3282 break;
3283 case '@':
3284 s->use_committer = !s->use_committer;
3285 break;
3286 case 'G':
3287 case KEY_END: {
3288 /* We don't know yet how many commits, so we're forced to
3289 * traverse them all. */
3290 view->count = 0;
3291 if (!s->thread_args.log_complete) {
3292 s->thread_args.load_all = 1;
3293 return trigger_log_thread(view, 0);
3296 s->selected = 0;
3297 entry = TAILQ_LAST(&s->commits.head, commit_queue_head);
3298 for (n = 0; n < eos; n++) {
3299 if (entry == NULL)
3300 break;
3301 s->first_displayed_entry = entry;
3302 entry = TAILQ_PREV(entry, commit_queue_head, entry);
3304 if (n > 0)
3305 s->selected = n - 1;
3306 select_commit(s);
3307 break;
3309 case CTRL('d'):
3310 case 'd':
3311 nscroll /= 2;
3312 /* FALL THROUGH */
3313 case KEY_NPAGE:
3314 case CTRL('f'):
3315 case 'f':
3316 case ' ':
3317 err = log_move_cursor_down(view, nscroll);
3318 break;
3319 case KEY_RESIZE:
3320 if (s->selected > view->nlines - 2)
3321 s->selected = view->nlines - 2;
3322 if (s->selected > s->commits.ncommits - 1)
3323 s->selected = s->commits.ncommits - 1;
3324 select_commit(s);
3325 if (s->commits.ncommits < view->nlines - 1 &&
3326 !s->thread_args.log_complete) {
3327 s->thread_args.commits_needed += (view->nlines - 1) -
3328 s->commits.ncommits;
3329 err = trigger_log_thread(view, 1);
3331 break;
3332 case KEY_ENTER:
3333 case '\r':
3334 view->count = 0;
3335 if (s->selected_entry == NULL)
3336 break;
3338 /* get dimensions--don't split till initialisation succeeds */
3339 if (view_is_parent_view(view))
3340 view_get_split(view, &begin_y, &begin_x);
3342 err = open_diff_view_for_commit(&diff_view, begin_y, begin_x,
3343 s->selected_entry->commit, s->selected_entry->id,
3344 view, s->repo);
3345 if (err)
3346 break;
3348 if (view_is_parent_view(view) &&
3349 view->mode == TOG_VIEW_SPLIT_HRZN) { /* safe to split */
3350 err = view_init_hsplit(view, begin_y);
3351 if (err)
3352 break;
3355 view->focussed = 0;
3356 diff_view->focussed = 1;
3357 diff_view->mode = view->mode;
3358 diff_view->nlines = view->lines - begin_y;
3360 if (view_is_parent_view(view)) {
3361 view_transfer_size(diff_view, view);
3362 err = view_close_child(view);
3363 if (err)
3364 return err;
3365 err = view_set_child(view, diff_view);
3366 if (err)
3367 return err;
3368 view->focus_child = 1;
3369 } else
3370 *new_view = diff_view;
3371 break;
3372 case 't':
3373 view->count = 0;
3374 if (s->selected_entry == NULL)
3375 break;
3376 if (view_is_parent_view(view))
3377 view_get_split(view, &begin_y, &begin_x);
3378 err = browse_commit_tree(&tree_view, begin_y, begin_x,
3379 s->selected_entry, s->in_repo_path, s->head_ref_name,
3380 s->repo);
3381 if (err)
3382 break;
3383 if (view_is_parent_view(view) &&
3384 view->mode == TOG_VIEW_SPLIT_HRZN) {
3385 err = view_init_hsplit(view, begin_y);
3386 if (err)
3387 break;
3389 view->focussed = 0;
3390 tree_view->focussed = 1;
3391 tree_view->mode = view->mode;
3392 tree_view->nlines = view->lines - begin_y;
3393 if (view_is_parent_view(view)) {
3394 view_transfer_size(tree_view, view);
3395 err = view_close_child(view);
3396 if (err)
3397 return err;
3398 err = view_set_child(view, tree_view);
3399 if (err)
3400 return err;
3401 view->focus_child = 1;
3402 } else
3403 *new_view = tree_view;
3404 break;
3405 case KEY_BACKSPACE:
3406 case CTRL('l'):
3407 case 'B':
3408 view->count = 0;
3409 if (ch == KEY_BACKSPACE &&
3410 got_path_is_root_dir(s->in_repo_path))
3411 break;
3412 err = stop_log_thread(s);
3413 if (err)
3414 return err;
3415 if (ch == KEY_BACKSPACE) {
3416 char *parent_path;
3417 err = got_path_dirname(&parent_path, s->in_repo_path);
3418 if (err)
3419 return err;
3420 free(s->in_repo_path);
3421 s->in_repo_path = parent_path;
3422 s->thread_args.in_repo_path = s->in_repo_path;
3423 } else if (ch == CTRL('l')) {
3424 struct got_object_id *start_id;
3425 err = got_repo_match_object_id(&start_id, NULL,
3426 s->head_ref_name ? s->head_ref_name : GOT_REF_HEAD,
3427 GOT_OBJ_TYPE_COMMIT, &tog_refs, s->repo);
3428 if (err)
3429 return err;
3430 free(s->start_id);
3431 s->start_id = start_id;
3432 s->thread_args.start_id = s->start_id;
3433 } else /* 'B' */
3434 s->log_branches = !s->log_branches;
3436 if (s->thread_args.pack_fds == NULL) {
3437 err = got_repo_pack_fds_open(&s->thread_args.pack_fds);
3438 if (err)
3439 return err;
3441 err = got_repo_open(&s->thread_args.repo,
3442 got_repo_get_path(s->repo), NULL,
3443 s->thread_args.pack_fds);
3444 if (err)
3445 return err;
3446 tog_free_refs();
3447 err = tog_load_refs(s->repo, 0);
3448 if (err)
3449 return err;
3450 err = got_commit_graph_open(&s->thread_args.graph,
3451 s->in_repo_path, !s->log_branches);
3452 if (err)
3453 return err;
3454 err = got_commit_graph_iter_start(s->thread_args.graph,
3455 s->start_id, s->repo, NULL, NULL);
3456 if (err)
3457 return err;
3458 free_commits(&s->commits);
3459 s->first_displayed_entry = NULL;
3460 s->last_displayed_entry = NULL;
3461 s->selected_entry = NULL;
3462 s->selected = 0;
3463 s->thread_args.log_complete = 0;
3464 s->quit = 0;
3465 s->thread_args.commits_needed = view->lines;
3466 s->matched_entry = NULL;
3467 s->search_entry = NULL;
3468 break;
3469 case 'r':
3470 view->count = 0;
3471 if (view_is_parent_view(view))
3472 view_get_split(view, &begin_y, &begin_x);
3473 ref_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_REF);
3474 if (ref_view == NULL)
3475 return got_error_from_errno("view_open");
3476 err = open_ref_view(ref_view, s->repo);
3477 if (err) {
3478 view_close(ref_view);
3479 return err;
3481 if (view_is_parent_view(view) &&
3482 view->mode == TOG_VIEW_SPLIT_HRZN) {
3483 err = view_init_hsplit(view, begin_y);
3484 if (err)
3485 break;
3487 view->focussed = 0;
3488 ref_view->focussed = 1;
3489 ref_view->mode = view->mode;
3490 ref_view->nlines = view->lines - begin_y;
3491 if (view_is_parent_view(view)) {
3492 view_transfer_size(ref_view, view);
3493 err = view_close_child(view);
3494 if (err)
3495 return err;
3496 err = view_set_child(view, ref_view);
3497 if (err)
3498 return err;
3499 view->focus_child = 1;
3500 } else
3501 *new_view = ref_view;
3502 break;
3503 default:
3504 view->count = 0;
3505 break;
3508 return err;
3511 static const struct got_error *
3512 apply_unveil(const char *repo_path, const char *worktree_path)
3514 const struct got_error *error;
3516 #ifdef PROFILE
3517 if (unveil("gmon.out", "rwc") != 0)
3518 return got_error_from_errno2("unveil", "gmon.out");
3519 #endif
3520 if (repo_path && unveil(repo_path, "r") != 0)
3521 return got_error_from_errno2("unveil", repo_path);
3523 if (worktree_path && unveil(worktree_path, "rwc") != 0)
3524 return got_error_from_errno2("unveil", worktree_path);
3526 if (unveil(GOT_TMPDIR_STR, "rwc") != 0)
3527 return got_error_from_errno2("unveil", GOT_TMPDIR_STR);
3529 error = got_privsep_unveil_exec_helpers();
3530 if (error != NULL)
3531 return error;
3533 if (unveil(NULL, NULL) != 0)
3534 return got_error_from_errno("unveil");
3536 return NULL;
3539 static void
3540 init_curses(void)
3543 * Override default signal handlers before starting ncurses.
3544 * This should prevent ncurses from installing its own
3545 * broken cleanup() signal handler.
3547 signal(SIGWINCH, tog_sigwinch);
3548 signal(SIGPIPE, tog_sigpipe);
3549 signal(SIGCONT, tog_sigcont);
3550 signal(SIGINT, tog_sigint);
3551 signal(SIGTERM, tog_sigterm);
3553 initscr();
3554 cbreak();
3555 halfdelay(1); /* Do fast refresh while initial view is loading. */
3556 noecho();
3557 nonl();
3558 intrflush(stdscr, FALSE);
3559 keypad(stdscr, TRUE);
3560 curs_set(0);
3561 if (getenv("TOG_COLORS") != NULL) {
3562 start_color();
3563 use_default_colors();
3567 static const struct got_error *
3568 get_in_repo_path_from_argv0(char **in_repo_path, int argc, char *argv[],
3569 struct got_repository *repo, struct got_worktree *worktree)
3571 const struct got_error *err = NULL;
3573 if (argc == 0) {
3574 *in_repo_path = strdup("/");
3575 if (*in_repo_path == NULL)
3576 return got_error_from_errno("strdup");
3577 return NULL;
3580 if (worktree) {
3581 const char *prefix = got_worktree_get_path_prefix(worktree);
3582 char *p;
3584 err = got_worktree_resolve_path(&p, worktree, argv[0]);
3585 if (err)
3586 return err;
3587 if (asprintf(in_repo_path, "%s%s%s", prefix,
3588 (p[0] != '\0' && !got_path_is_root_dir(prefix)) ? "/" : "",
3589 p) == -1) {
3590 err = got_error_from_errno("asprintf");
3591 *in_repo_path = NULL;
3593 free(p);
3594 } else
3595 err = got_repo_map_path(in_repo_path, repo, argv[0]);
3597 return err;
3600 static const struct got_error *
3601 cmd_log(int argc, char *argv[])
3603 const struct got_error *error;
3604 struct got_repository *repo = NULL;
3605 struct got_worktree *worktree = NULL;
3606 struct got_object_id *start_id = NULL;
3607 char *in_repo_path = NULL, *repo_path = NULL, *cwd = NULL;
3608 char *start_commit = NULL, *label = NULL;
3609 struct got_reference *ref = NULL;
3610 const char *head_ref_name = NULL;
3611 int ch, log_branches = 0;
3612 struct tog_view *view;
3613 int *pack_fds = NULL;
3615 while ((ch = getopt(argc, argv, "bc:r:")) != -1) {
3616 switch (ch) {
3617 case 'b':
3618 log_branches = 1;
3619 break;
3620 case 'c':
3621 start_commit = optarg;
3622 break;
3623 case 'r':
3624 repo_path = realpath(optarg, NULL);
3625 if (repo_path == NULL)
3626 return got_error_from_errno2("realpath",
3627 optarg);
3628 break;
3629 default:
3630 usage_log();
3631 /* NOTREACHED */
3635 argc -= optind;
3636 argv += optind;
3638 if (argc > 1)
3639 usage_log();
3641 error = got_repo_pack_fds_open(&pack_fds);
3642 if (error != NULL)
3643 goto done;
3645 if (repo_path == NULL) {
3646 cwd = getcwd(NULL, 0);
3647 if (cwd == NULL)
3648 return got_error_from_errno("getcwd");
3649 error = got_worktree_open(&worktree, cwd);
3650 if (error && error->code != GOT_ERR_NOT_WORKTREE)
3651 goto done;
3652 if (worktree)
3653 repo_path =
3654 strdup(got_worktree_get_repo_path(worktree));
3655 else
3656 repo_path = strdup(cwd);
3657 if (repo_path == NULL) {
3658 error = got_error_from_errno("strdup");
3659 goto done;
3663 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
3664 if (error != NULL)
3665 goto done;
3667 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
3668 repo, worktree);
3669 if (error)
3670 goto done;
3672 init_curses();
3674 error = apply_unveil(got_repo_get_path(repo),
3675 worktree ? got_worktree_get_root_path(worktree) : NULL);
3676 if (error)
3677 goto done;
3679 /* already loaded by tog_log_with_path()? */
3680 if (TAILQ_EMPTY(&tog_refs)) {
3681 error = tog_load_refs(repo, 0);
3682 if (error)
3683 goto done;
3686 if (start_commit == NULL) {
3687 error = got_repo_match_object_id(&start_id, &label,
3688 worktree ? got_worktree_get_head_ref_name(worktree) :
3689 GOT_REF_HEAD, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
3690 if (error)
3691 goto done;
3692 head_ref_name = label;
3693 } else {
3694 error = got_ref_open(&ref, repo, start_commit, 0);
3695 if (error == NULL)
3696 head_ref_name = got_ref_get_name(ref);
3697 else if (error->code != GOT_ERR_NOT_REF)
3698 goto done;
3699 error = got_repo_match_object_id(&start_id, NULL,
3700 start_commit, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
3701 if (error)
3702 goto done;
3705 view = view_open(0, 0, 0, 0, TOG_VIEW_LOG);
3706 if (view == NULL) {
3707 error = got_error_from_errno("view_open");
3708 goto done;
3710 error = open_log_view(view, start_id, repo, head_ref_name,
3711 in_repo_path, log_branches);
3712 if (error)
3713 goto done;
3714 if (worktree) {
3715 /* Release work tree lock. */
3716 got_worktree_close(worktree);
3717 worktree = NULL;
3719 error = view_loop(view);
3720 done:
3721 free(in_repo_path);
3722 free(repo_path);
3723 free(cwd);
3724 free(start_id);
3725 free(label);
3726 if (ref)
3727 got_ref_close(ref);
3728 if (repo) {
3729 const struct got_error *close_err = got_repo_close(repo);
3730 if (error == NULL)
3731 error = close_err;
3733 if (worktree)
3734 got_worktree_close(worktree);
3735 if (pack_fds) {
3736 const struct got_error *pack_err =
3737 got_repo_pack_fds_close(pack_fds);
3738 if (error == NULL)
3739 error = pack_err;
3741 tog_free_refs();
3742 return error;
3745 __dead static void
3746 usage_diff(void)
3748 endwin();
3749 fprintf(stderr, "usage: %s diff [-a] [-C number] [-r repository-path] "
3750 "[-w] object1 object2\n", getprogname());
3751 exit(1);
3754 static int
3755 match_line(const char *line, regex_t *regex, size_t nmatch,
3756 regmatch_t *regmatch)
3758 return regexec(regex, line, nmatch, regmatch, 0) == 0;
3761 static struct tog_color *
3762 match_color(struct tog_colors *colors, const char *line)
3764 struct tog_color *tc = NULL;
3766 STAILQ_FOREACH(tc, colors, entry) {
3767 if (match_line(line, &tc->regex, 0, NULL))
3768 return tc;
3771 return NULL;
3774 static const struct got_error *
3775 add_matched_line(int *wtotal, const char *line, int wlimit, int col_tab_align,
3776 WINDOW *window, int skipcol, regmatch_t *regmatch)
3778 const struct got_error *err = NULL;
3779 char *exstr = NULL;
3780 wchar_t *wline = NULL;
3781 int rme, rms, n, width, scrollx;
3782 int width0 = 0, width1 = 0, width2 = 0;
3783 char *seg0 = NULL, *seg1 = NULL, *seg2 = NULL;
3785 *wtotal = 0;
3787 rms = regmatch->rm_so;
3788 rme = regmatch->rm_eo;
3790 err = expand_tab(&exstr, line);
3791 if (err)
3792 return err;
3794 /* Split the line into 3 segments, according to match offsets. */
3795 seg0 = strndup(exstr, rms);
3796 if (seg0 == NULL) {
3797 err = got_error_from_errno("strndup");
3798 goto done;
3800 seg1 = strndup(exstr + rms, rme - rms);
3801 if (seg1 == NULL) {
3802 err = got_error_from_errno("strndup");
3803 goto done;
3805 seg2 = strdup(exstr + rme);
3806 if (seg2 == NULL) {
3807 err = got_error_from_errno("strndup");
3808 goto done;
3811 /* draw up to matched token if we haven't scrolled past it */
3812 err = format_line(&wline, &width0, NULL, seg0, 0, wlimit,
3813 col_tab_align, 1);
3814 if (err)
3815 goto done;
3816 n = MAX(width0 - skipcol, 0);
3817 if (n) {
3818 free(wline);
3819 err = format_line(&wline, &width, &scrollx, seg0, skipcol,
3820 wlimit, col_tab_align, 1);
3821 if (err)
3822 goto done;
3823 waddwstr(window, &wline[scrollx]);
3824 wlimit -= width;
3825 *wtotal += width;
3828 if (wlimit > 0) {
3829 int i = 0, w = 0;
3830 size_t wlen;
3832 free(wline);
3833 err = format_line(&wline, &width1, NULL, seg1, 0, wlimit,
3834 col_tab_align, 1);
3835 if (err)
3836 goto done;
3837 wlen = wcslen(wline);
3838 while (i < wlen) {
3839 width = wcwidth(wline[i]);
3840 if (width == -1) {
3841 /* should not happen, tabs are expanded */
3842 err = got_error(GOT_ERR_RANGE);
3843 goto done;
3845 if (width0 + w + width > skipcol)
3846 break;
3847 w += width;
3848 i++;
3850 /* draw (visible part of) matched token (if scrolled into it) */
3851 if (width1 - w > 0) {
3852 wattron(window, A_STANDOUT);
3853 waddwstr(window, &wline[i]);
3854 wattroff(window, A_STANDOUT);
3855 wlimit -= (width1 - w);
3856 *wtotal += (width1 - w);
3860 if (wlimit > 0) { /* draw rest of line */
3861 free(wline);
3862 if (skipcol > width0 + width1) {
3863 err = format_line(&wline, &width2, &scrollx, seg2,
3864 skipcol - (width0 + width1), wlimit,
3865 col_tab_align, 1);
3866 if (err)
3867 goto done;
3868 waddwstr(window, &wline[scrollx]);
3869 } else {
3870 err = format_line(&wline, &width2, NULL, seg2, 0,
3871 wlimit, col_tab_align, 1);
3872 if (err)
3873 goto done;
3874 waddwstr(window, wline);
3876 *wtotal += width2;
3878 done:
3879 free(wline);
3880 free(exstr);
3881 free(seg0);
3882 free(seg1);
3883 free(seg2);
3884 return err;
3887 static const struct got_error *
3888 draw_file(struct tog_view *view, const char *header)
3890 struct tog_diff_view_state *s = &view->state.diff;
3891 regmatch_t *regmatch = &view->regmatch;
3892 const struct got_error *err;
3893 int nprinted = 0;
3894 char *line;
3895 size_t linesize = 0;
3896 ssize_t linelen;
3897 struct tog_color *tc;
3898 wchar_t *wline;
3899 int width;
3900 int max_lines = view->nlines;
3901 int nlines = s->nlines;
3902 off_t line_offset;
3904 line_offset = s->line_offsets[s->first_displayed_line - 1];
3905 if (fseeko(s->f, line_offset, SEEK_SET) == -1)
3906 return got_error_from_errno("fseek");
3908 werase(view->window);
3910 if (header) {
3911 if (asprintf(&line, "[%d/%d] %s",
3912 s->first_displayed_line - 1 + s->selected_line, nlines,
3913 header) == -1)
3914 return got_error_from_errno("asprintf");
3915 err = format_line(&wline, &width, NULL, line, 0, view->ncols,
3916 0, 0);
3917 free(line);
3918 if (err)
3919 return err;
3921 if (view_needs_focus_indication(view))
3922 wstandout(view->window);
3923 waddwstr(view->window, wline);
3924 free(wline);
3925 wline = NULL;
3926 if (view_needs_focus_indication(view))
3927 wstandend(view->window);
3928 if (width <= view->ncols - 1)
3929 waddch(view->window, '\n');
3931 if (max_lines <= 1)
3932 return NULL;
3933 max_lines--;
3936 s->eof = 0;
3937 view->maxx = 0;
3938 line = NULL;
3939 while (max_lines > 0 && nprinted < max_lines) {
3940 linelen = getline(&line, &linesize, s->f);
3941 if (linelen == -1) {
3942 if (feof(s->f)) {
3943 s->eof = 1;
3944 break;
3946 free(line);
3947 return got_ferror(s->f, GOT_ERR_IO);
3950 /* Set view->maxx based on full line length. */
3951 err = format_line(&wline, &width, NULL, line, 0, INT_MAX, 0,
3952 view->x ? 1 : 0);
3953 if (err) {
3954 free(line);
3955 return err;
3957 view->maxx = MAX(view->maxx, width);
3958 free(wline);
3959 wline = NULL;
3961 tc = match_color(&s->colors, line);
3962 if (tc)
3963 wattr_on(view->window,
3964 COLOR_PAIR(tc->colorpair), NULL);
3965 if (s->first_displayed_line + nprinted == s->matched_line &&
3966 regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
3967 err = add_matched_line(&width, line, view->ncols, 0,
3968 view->window, view->x, regmatch);
3969 if (err) {
3970 free(line);
3971 return err;
3973 } else {
3974 int skip;
3975 err = format_line(&wline, &width, &skip, line,
3976 view->x, view->ncols, 0, view->x ? 1 : 0);
3977 if (err) {
3978 free(line);
3979 return err;
3981 waddwstr(view->window, &wline[skip]);
3982 free(wline);
3983 wline = NULL;
3985 if (tc)
3986 wattr_off(view->window,
3987 COLOR_PAIR(tc->colorpair), NULL);
3988 if (width <= view->ncols - 1)
3989 waddch(view->window, '\n');
3990 nprinted++;
3992 free(line);
3993 if (nprinted >= 1)
3994 s->last_displayed_line = s->first_displayed_line +
3995 (nprinted - 1);
3996 else
3997 s->last_displayed_line = s->first_displayed_line;
3999 view_border(view);
4001 if (s->eof) {
4002 while (nprinted < view->nlines) {
4003 waddch(view->window, '\n');
4004 nprinted++;
4007 err = format_line(&wline, &width, NULL, TOG_EOF_STRING, 0,
4008 view->ncols, 0, 0);
4009 if (err) {
4010 return err;
4013 wstandout(view->window);
4014 waddwstr(view->window, wline);
4015 free(wline);
4016 wline = NULL;
4017 wstandend(view->window);
4020 return NULL;
4023 static char *
4024 get_datestr(time_t *time, char *datebuf)
4026 struct tm mytm, *tm;
4027 char *p, *s;
4029 tm = gmtime_r(time, &mytm);
4030 if (tm == NULL)
4031 return NULL;
4032 s = asctime_r(tm, datebuf);
4033 if (s == NULL)
4034 return NULL;
4035 p = strchr(s, '\n');
4036 if (p)
4037 *p = '\0';
4038 return s;
4041 static const struct got_error *
4042 get_changed_paths(struct got_pathlist_head *paths,
4043 struct got_commit_object *commit, struct got_repository *repo)
4045 const struct got_error *err = NULL;
4046 struct got_object_id *tree_id1 = NULL, *tree_id2 = NULL;
4047 struct got_tree_object *tree1 = NULL, *tree2 = NULL;
4048 struct got_object_qid *qid;
4050 qid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
4051 if (qid != NULL) {
4052 struct got_commit_object *pcommit;
4053 err = got_object_open_as_commit(&pcommit, repo,
4054 &qid->id);
4055 if (err)
4056 return err;
4058 tree_id1 = got_object_id_dup(
4059 got_object_commit_get_tree_id(pcommit));
4060 if (tree_id1 == NULL) {
4061 got_object_commit_close(pcommit);
4062 return got_error_from_errno("got_object_id_dup");
4064 got_object_commit_close(pcommit);
4068 if (tree_id1) {
4069 err = got_object_open_as_tree(&tree1, repo, tree_id1);
4070 if (err)
4071 goto done;
4074 tree_id2 = got_object_commit_get_tree_id(commit);
4075 err = got_object_open_as_tree(&tree2, repo, tree_id2);
4076 if (err)
4077 goto done;
4079 err = got_diff_tree(tree1, tree2, NULL, NULL, -1, -1, "", "", repo,
4080 got_diff_tree_collect_changed_paths, paths, 0);
4081 done:
4082 if (tree1)
4083 got_object_tree_close(tree1);
4084 if (tree2)
4085 got_object_tree_close(tree2);
4086 free(tree_id1);
4087 return err;
4090 static const struct got_error *
4091 add_line_offset(off_t **line_offsets, size_t *nlines, off_t off)
4093 off_t *p;
4095 p = reallocarray(*line_offsets, *nlines + 1, sizeof(off_t));
4096 if (p == NULL)
4097 return got_error_from_errno("reallocarray");
4098 *line_offsets = p;
4099 (*line_offsets)[*nlines] = off;
4100 (*nlines)++;
4101 return NULL;
4104 static const struct got_error *
4105 write_commit_info(off_t **line_offsets, size_t *nlines,
4106 struct got_object_id *commit_id, struct got_reflist_head *refs,
4107 struct got_repository *repo, FILE *outfile)
4109 const struct got_error *err = NULL;
4110 char datebuf[26], *datestr;
4111 struct got_commit_object *commit;
4112 char *id_str = NULL, *logmsg = NULL, *s = NULL, *line;
4113 time_t committer_time;
4114 const char *author, *committer;
4115 char *refs_str = NULL;
4116 struct got_pathlist_head changed_paths;
4117 struct got_pathlist_entry *pe;
4118 off_t outoff = 0;
4119 int n;
4121 TAILQ_INIT(&changed_paths);
4123 if (refs) {
4124 err = build_refs_str(&refs_str, refs, commit_id, repo);
4125 if (err)
4126 return err;
4129 err = got_object_open_as_commit(&commit, repo, commit_id);
4130 if (err)
4131 return err;
4133 err = got_object_id_str(&id_str, commit_id);
4134 if (err) {
4135 err = got_error_from_errno("got_object_id_str");
4136 goto done;
4139 err = add_line_offset(line_offsets, nlines, 0);
4140 if (err)
4141 goto done;
4143 n = fprintf(outfile, "commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
4144 refs_str ? refs_str : "", refs_str ? ")" : "");
4145 if (n < 0) {
4146 err = got_error_from_errno("fprintf");
4147 goto done;
4149 outoff += n;
4150 err = add_line_offset(line_offsets, nlines, outoff);
4151 if (err)
4152 goto done;
4154 n = fprintf(outfile, "from: %s\n",
4155 got_object_commit_get_author(commit));
4156 if (n < 0) {
4157 err = got_error_from_errno("fprintf");
4158 goto done;
4160 outoff += n;
4161 err = add_line_offset(line_offsets, nlines, outoff);
4162 if (err)
4163 goto done;
4165 committer_time = got_object_commit_get_committer_time(commit);
4166 datestr = get_datestr(&committer_time, datebuf);
4167 if (datestr) {
4168 n = fprintf(outfile, "date: %s UTC\n", datestr);
4169 if (n < 0) {
4170 err = got_error_from_errno("fprintf");
4171 goto done;
4173 outoff += n;
4174 err = add_line_offset(line_offsets, nlines, outoff);
4175 if (err)
4176 goto done;
4178 author = got_object_commit_get_author(commit);
4179 committer = got_object_commit_get_committer(commit);
4180 if (strcmp(author, committer) != 0) {
4181 n = fprintf(outfile, "via: %s\n", committer);
4182 if (n < 0) {
4183 err = got_error_from_errno("fprintf");
4184 goto done;
4186 outoff += n;
4187 err = add_line_offset(line_offsets, nlines, outoff);
4188 if (err)
4189 goto done;
4191 if (got_object_commit_get_nparents(commit) > 1) {
4192 const struct got_object_id_queue *parent_ids;
4193 struct got_object_qid *qid;
4194 int pn = 1;
4195 parent_ids = got_object_commit_get_parent_ids(commit);
4196 STAILQ_FOREACH(qid, parent_ids, entry) {
4197 err = got_object_id_str(&id_str, &qid->id);
4198 if (err)
4199 goto done;
4200 n = fprintf(outfile, "parent %d: %s\n", pn++, id_str);
4201 if (n < 0) {
4202 err = got_error_from_errno("fprintf");
4203 goto done;
4205 outoff += n;
4206 err = add_line_offset(line_offsets, nlines, outoff);
4207 if (err)
4208 goto done;
4209 free(id_str);
4210 id_str = NULL;
4214 err = got_object_commit_get_logmsg(&logmsg, commit);
4215 if (err)
4216 goto done;
4217 s = logmsg;
4218 while ((line = strsep(&s, "\n")) != NULL) {
4219 n = fprintf(outfile, "%s\n", line);
4220 if (n < 0) {
4221 err = got_error_from_errno("fprintf");
4222 goto done;
4224 outoff += n;
4225 err = add_line_offset(line_offsets, nlines, outoff);
4226 if (err)
4227 goto done;
4230 err = get_changed_paths(&changed_paths, commit, repo);
4231 if (err)
4232 goto done;
4233 TAILQ_FOREACH(pe, &changed_paths, entry) {
4234 struct got_diff_changed_path *cp = pe->data;
4235 n = fprintf(outfile, "%c %s\n", cp->status, pe->path);
4236 if (n < 0) {
4237 err = got_error_from_errno("fprintf");
4238 goto done;
4240 outoff += n;
4241 err = add_line_offset(line_offsets, nlines, outoff);
4242 if (err)
4243 goto done;
4244 free((char *)pe->path);
4245 free(pe->data);
4248 fputc('\n', outfile);
4249 outoff++;
4250 err = add_line_offset(line_offsets, nlines, outoff);
4251 done:
4252 got_pathlist_free(&changed_paths);
4253 free(id_str);
4254 free(logmsg);
4255 free(refs_str);
4256 got_object_commit_close(commit);
4257 if (err) {
4258 free(*line_offsets);
4259 *line_offsets = NULL;
4260 *nlines = 0;
4262 return err;
4265 static const struct got_error *
4266 create_diff(struct tog_diff_view_state *s)
4268 const struct got_error *err = NULL;
4269 FILE *f = NULL;
4270 int obj_type;
4272 free(s->line_offsets);
4273 s->line_offsets = malloc(sizeof(off_t));
4274 if (s->line_offsets == NULL)
4275 return got_error_from_errno("malloc");
4276 s->nlines = 0;
4278 f = got_opentemp();
4279 if (f == NULL) {
4280 err = got_error_from_errno("got_opentemp");
4281 goto done;
4283 if (s->f && fclose(s->f) == EOF) {
4284 err = got_error_from_errno("fclose");
4285 goto done;
4287 s->f = f;
4289 if (s->id1)
4290 err = got_object_get_type(&obj_type, s->repo, s->id1);
4291 else
4292 err = got_object_get_type(&obj_type, s->repo, s->id2);
4293 if (err)
4294 goto done;
4296 switch (obj_type) {
4297 case GOT_OBJ_TYPE_BLOB:
4298 err = got_diff_objects_as_blobs(&s->line_offsets, &s->nlines,
4299 s->f1, s->f2, s->fd1, s->fd2, s->id1, s->id2,
4300 s->label1, s->label2, tog_diff_algo, s->diff_context,
4301 s->ignore_whitespace, s->force_text_diff, s->repo, s->f);
4302 break;
4303 case GOT_OBJ_TYPE_TREE:
4304 err = got_diff_objects_as_trees(&s->line_offsets, &s->nlines,
4305 s->f1, s->f2, s->fd1, s->fd2, s->id1, s->id2, NULL, "", "",
4306 tog_diff_algo, s->diff_context, s->ignore_whitespace,
4307 s->force_text_diff, s->repo, s->f);
4308 break;
4309 case GOT_OBJ_TYPE_COMMIT: {
4310 const struct got_object_id_queue *parent_ids;
4311 struct got_object_qid *pid;
4312 struct got_commit_object *commit2;
4313 struct got_reflist_head *refs;
4315 err = got_object_open_as_commit(&commit2, s->repo, s->id2);
4316 if (err)
4317 goto done;
4318 refs = got_reflist_object_id_map_lookup(tog_refs_idmap, s->id2);
4319 /* Show commit info if we're diffing to a parent/root commit. */
4320 if (s->id1 == NULL) {
4321 err = write_commit_info(&s->line_offsets, &s->nlines,
4322 s->id2, refs, s->repo, s->f);
4323 if (err)
4324 goto done;
4325 } else {
4326 parent_ids = got_object_commit_get_parent_ids(commit2);
4327 STAILQ_FOREACH(pid, parent_ids, entry) {
4328 if (got_object_id_cmp(s->id1, &pid->id) == 0) {
4329 err = write_commit_info(
4330 &s->line_offsets, &s->nlines,
4331 s->id2, refs, s->repo, s->f);
4332 if (err)
4333 goto done;
4334 break;
4338 got_object_commit_close(commit2);
4340 err = got_diff_objects_as_commits(&s->line_offsets, &s->nlines,
4341 s->f1, s->f2, s->fd1, s->fd2, s->id1, s->id2, NULL,
4342 tog_diff_algo, s->diff_context, s->ignore_whitespace,
4343 s->force_text_diff, s->repo, s->f);
4344 break;
4346 default:
4347 err = got_error(GOT_ERR_OBJ_TYPE);
4348 break;
4350 if (err)
4351 goto done;
4352 done:
4353 if (s->f && fflush(s->f) != 0 && err == NULL)
4354 err = got_error_from_errno("fflush");
4355 return err;
4358 static void
4359 diff_view_indicate_progress(struct tog_view *view)
4361 mvwaddstr(view->window, 0, 0, "diffing...");
4362 update_panels();
4363 doupdate();
4366 static const struct got_error *
4367 search_start_diff_view(struct tog_view *view)
4369 struct tog_diff_view_state *s = &view->state.diff;
4371 s->matched_line = 0;
4372 return NULL;
4375 static const struct got_error *
4376 search_next_diff_view(struct tog_view *view)
4378 struct tog_diff_view_state *s = &view->state.diff;
4379 const struct got_error *err = NULL;
4380 int lineno;
4381 char *line = NULL;
4382 size_t linesize = 0;
4383 ssize_t linelen;
4385 if (!view->searching) {
4386 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4387 return NULL;
4390 if (s->matched_line) {
4391 if (view->searching == TOG_SEARCH_FORWARD)
4392 lineno = s->matched_line + 1;
4393 else
4394 lineno = s->matched_line - 1;
4395 } else
4396 lineno = s->first_displayed_line;
4398 while (1) {
4399 off_t offset;
4401 if (lineno <= 0 || lineno > s->nlines) {
4402 if (s->matched_line == 0) {
4403 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4404 break;
4407 if (view->searching == TOG_SEARCH_FORWARD)
4408 lineno = 1;
4409 else
4410 lineno = s->nlines;
4413 offset = s->line_offsets[lineno - 1];
4414 if (fseeko(s->f, offset, SEEK_SET) != 0) {
4415 free(line);
4416 return got_error_from_errno("fseeko");
4418 linelen = getline(&line, &linesize, s->f);
4419 if (linelen != -1) {
4420 char *exstr;
4421 err = expand_tab(&exstr, line);
4422 if (err)
4423 break;
4424 if (match_line(exstr, &view->regex, 1,
4425 &view->regmatch)) {
4426 view->search_next_done = TOG_SEARCH_HAVE_MORE;
4427 s->matched_line = lineno;
4428 free(exstr);
4429 break;
4431 free(exstr);
4433 if (view->searching == TOG_SEARCH_FORWARD)
4434 lineno++;
4435 else
4436 lineno--;
4438 free(line);
4440 if (s->matched_line) {
4441 s->first_displayed_line = s->matched_line;
4442 s->selected_line = 1;
4445 return err;
4448 static const struct got_error *
4449 close_diff_view(struct tog_view *view)
4451 const struct got_error *err = NULL;
4452 struct tog_diff_view_state *s = &view->state.diff;
4454 free(s->id1);
4455 s->id1 = NULL;
4456 free(s->id2);
4457 s->id2 = NULL;
4458 if (s->f && fclose(s->f) == EOF)
4459 err = got_error_from_errno("fclose");
4460 s->f = NULL;
4461 if (s->f1 && fclose(s->f1) == EOF && err == NULL)
4462 err = got_error_from_errno("fclose");
4463 s->f1 = NULL;
4464 if (s->f2 && fclose(s->f2) == EOF && err == NULL)
4465 err = got_error_from_errno("fclose");
4466 s->f2 = NULL;
4467 if (s->fd1 != -1 && close(s->fd1) == -1 && err == NULL)
4468 err = got_error_from_errno("close");
4469 s->fd1 = -1;
4470 if (s->fd2 != -1 && close(s->fd2) == -1 && err == NULL)
4471 err = got_error_from_errno("close");
4472 s->fd2 = -1;
4473 free_colors(&s->colors);
4474 free(s->line_offsets);
4475 s->line_offsets = NULL;
4476 s->nlines = 0;
4477 return err;
4480 static const struct got_error *
4481 open_diff_view(struct tog_view *view, struct got_object_id *id1,
4482 struct got_object_id *id2, const char *label1, const char *label2,
4483 int diff_context, int ignore_whitespace, int force_text_diff,
4484 struct tog_view *parent_view, struct got_repository *repo)
4486 const struct got_error *err;
4487 struct tog_diff_view_state *s = &view->state.diff;
4489 memset(s, 0, sizeof(*s));
4490 s->fd1 = -1;
4491 s->fd2 = -1;
4493 if (id1 != NULL && id2 != NULL) {
4494 int type1, type2;
4495 err = got_object_get_type(&type1, repo, id1);
4496 if (err)
4497 return err;
4498 err = got_object_get_type(&type2, repo, id2);
4499 if (err)
4500 return err;
4502 if (type1 != type2)
4503 return got_error(GOT_ERR_OBJ_TYPE);
4505 s->first_displayed_line = 1;
4506 s->last_displayed_line = view->nlines;
4507 s->selected_line = 1;
4508 s->repo = repo;
4509 s->id1 = id1;
4510 s->id2 = id2;
4511 s->label1 = label1;
4512 s->label2 = label2;
4514 if (id1) {
4515 s->id1 = got_object_id_dup(id1);
4516 if (s->id1 == NULL)
4517 return got_error_from_errno("got_object_id_dup");
4518 } else
4519 s->id1 = NULL;
4521 s->id2 = got_object_id_dup(id2);
4522 if (s->id2 == NULL) {
4523 err = got_error_from_errno("got_object_id_dup");
4524 goto done;
4527 s->f1 = got_opentemp();
4528 if (s->f1 == NULL) {
4529 err = got_error_from_errno("got_opentemp");
4530 goto done;
4533 s->f2 = got_opentemp();
4534 if (s->f2 == NULL) {
4535 err = got_error_from_errno("got_opentemp");
4536 goto done;
4539 s->fd1 = got_opentempfd();
4540 if (s->fd1 == -1) {
4541 err = got_error_from_errno("got_opentempfd");
4542 goto done;
4545 s->fd2 = got_opentempfd();
4546 if (s->fd2 == -1) {
4547 err = got_error_from_errno("got_opentempfd");
4548 goto done;
4551 s->first_displayed_line = 1;
4552 s->last_displayed_line = view->nlines;
4553 s->diff_context = diff_context;
4554 s->ignore_whitespace = ignore_whitespace;
4555 s->force_text_diff = force_text_diff;
4556 s->parent_view = parent_view;
4557 s->repo = repo;
4559 STAILQ_INIT(&s->colors);
4560 if (has_colors() && getenv("TOG_COLORS") != NULL) {
4561 err = add_color(&s->colors,
4562 "^-", TOG_COLOR_DIFF_MINUS,
4563 get_color_value("TOG_COLOR_DIFF_MINUS"));
4564 if (err)
4565 goto done;
4566 err = add_color(&s->colors, "^\\+",
4567 TOG_COLOR_DIFF_PLUS,
4568 get_color_value("TOG_COLOR_DIFF_PLUS"));
4569 if (err)
4570 goto done;
4571 err = add_color(&s->colors,
4572 "^@@", TOG_COLOR_DIFF_CHUNK_HEADER,
4573 get_color_value("TOG_COLOR_DIFF_CHUNK_HEADER"));
4574 if (err)
4575 goto done;
4577 err = add_color(&s->colors,
4578 "^(commit [0-9a-f]|parent [0-9]|"
4579 "(blob|file|tree|commit) [-+] |"
4580 "[MDmA] [^ ])", TOG_COLOR_DIFF_META,
4581 get_color_value("TOG_COLOR_DIFF_META"));
4582 if (err)
4583 goto done;
4585 err = add_color(&s->colors,
4586 "^(from|via): ", TOG_COLOR_AUTHOR,
4587 get_color_value("TOG_COLOR_AUTHOR"));
4588 if (err)
4589 goto done;
4591 err = add_color(&s->colors,
4592 "^date: ", TOG_COLOR_DATE,
4593 get_color_value("TOG_COLOR_DATE"));
4594 if (err)
4595 goto done;
4598 if (parent_view && parent_view->type == TOG_VIEW_LOG &&
4599 view_is_splitscreen(view))
4600 show_log_view(parent_view); /* draw border */
4601 diff_view_indicate_progress(view);
4603 err = create_diff(s);
4605 view->show = show_diff_view;
4606 view->input = input_diff_view;
4607 view->reset = reset_diff_view;
4608 view->close = close_diff_view;
4609 view->search_start = search_start_diff_view;
4610 view->search_next = search_next_diff_view;
4611 done:
4612 if (err)
4613 close_diff_view(view);
4614 return err;
4617 static const struct got_error *
4618 show_diff_view(struct tog_view *view)
4620 const struct got_error *err;
4621 struct tog_diff_view_state *s = &view->state.diff;
4622 char *id_str1 = NULL, *id_str2, *header;
4623 const char *label1, *label2;
4625 if (s->id1) {
4626 err = got_object_id_str(&id_str1, s->id1);
4627 if (err)
4628 return err;
4629 label1 = s->label1 ? : id_str1;
4630 } else
4631 label1 = "/dev/null";
4633 err = got_object_id_str(&id_str2, s->id2);
4634 if (err)
4635 return err;
4636 label2 = s->label2 ? : id_str2;
4638 if (asprintf(&header, "diff %s %s", label1, label2) == -1) {
4639 err = got_error_from_errno("asprintf");
4640 free(id_str1);
4641 free(id_str2);
4642 return err;
4644 free(id_str1);
4645 free(id_str2);
4647 err = draw_file(view, header);
4648 free(header);
4649 return err;
4652 static const struct got_error *
4653 set_selected_commit(struct tog_diff_view_state *s,
4654 struct commit_queue_entry *entry)
4656 const struct got_error *err;
4657 const struct got_object_id_queue *parent_ids;
4658 struct got_commit_object *selected_commit;
4659 struct got_object_qid *pid;
4661 free(s->id2);
4662 s->id2 = got_object_id_dup(entry->id);
4663 if (s->id2 == NULL)
4664 return got_error_from_errno("got_object_id_dup");
4666 err = got_object_open_as_commit(&selected_commit, s->repo, entry->id);
4667 if (err)
4668 return err;
4669 parent_ids = got_object_commit_get_parent_ids(selected_commit);
4670 free(s->id1);
4671 pid = STAILQ_FIRST(parent_ids);
4672 s->id1 = pid ? got_object_id_dup(&pid->id) : NULL;
4673 got_object_commit_close(selected_commit);
4674 return NULL;
4677 static const struct got_error *
4678 reset_diff_view(struct tog_view *view)
4680 struct tog_diff_view_state *s = &view->state.diff;
4682 view->count = 0;
4683 wclear(view->window);
4684 s->first_displayed_line = 1;
4685 s->last_displayed_line = view->nlines;
4686 s->matched_line = 0;
4687 diff_view_indicate_progress(view);
4688 return create_diff(s);
4691 static struct got_object_id *get_selected_commit_id(struct tog_blame_line *,
4692 int, int, int);
4693 static struct got_object_id *get_annotation_for_line(struct tog_blame_line *,
4694 int, int);
4696 static const struct got_error *
4697 input_diff_view(struct tog_view **new_view, struct tog_view *view, int ch)
4699 const struct got_error *err = NULL;
4700 struct tog_diff_view_state *s = &view->state.diff;
4701 struct tog_log_view_state *ls;
4702 struct commit_queue_entry *old_selected_entry;
4703 char *line = NULL;
4704 size_t linesize = 0;
4705 ssize_t linelen;
4706 int i, nscroll = view->nlines - 1, up = 0;
4708 switch (ch) {
4709 case '0':
4710 view->x = 0;
4711 break;
4712 case '$':
4713 view->x = MAX(view->maxx - view->ncols / 3, 0);
4714 view->count = 0;
4715 break;
4716 case KEY_RIGHT:
4717 case 'l':
4718 if (view->x + view->ncols / 3 < view->maxx)
4719 view->x += 2; /* move two columns right */
4720 else
4721 view->count = 0;
4722 break;
4723 case KEY_LEFT:
4724 case 'h':
4725 view->x -= MIN(view->x, 2); /* move two columns back */
4726 if (view->x <= 0)
4727 view->count = 0;
4728 break;
4729 case 'a':
4730 case 'w':
4731 if (ch == 'a')
4732 s->force_text_diff = !s->force_text_diff;
4733 if (ch == 'w')
4734 s->ignore_whitespace = !s->ignore_whitespace;
4735 err = reset_diff_view(view);
4736 break;
4737 case 'g':
4738 case KEY_HOME:
4739 s->first_displayed_line = 1;
4740 view->count = 0;
4741 break;
4742 case 'G':
4743 case KEY_END:
4744 view->count = 0;
4745 if (s->eof)
4746 break;
4748 s->first_displayed_line = (s->nlines - view->nlines) + 2;
4749 s->eof = 1;
4750 break;
4751 case 'k':
4752 case KEY_UP:
4753 case CTRL('p'):
4754 if (s->first_displayed_line > 1)
4755 s->first_displayed_line--;
4756 else
4757 view->count = 0;
4758 break;
4759 case CTRL('u'):
4760 case 'u':
4761 nscroll /= 2;
4762 /* FALL THROUGH */
4763 case KEY_PPAGE:
4764 case CTRL('b'):
4765 case 'b':
4766 if (s->first_displayed_line == 1) {
4767 view->count = 0;
4768 break;
4770 i = 0;
4771 while (i++ < nscroll && s->first_displayed_line > 1)
4772 s->first_displayed_line--;
4773 break;
4774 case 'j':
4775 case KEY_DOWN:
4776 case CTRL('n'):
4777 if (!s->eof)
4778 s->first_displayed_line++;
4779 else
4780 view->count = 0;
4781 break;
4782 case CTRL('d'):
4783 case 'd':
4784 nscroll /= 2;
4785 /* FALL THROUGH */
4786 case KEY_NPAGE:
4787 case CTRL('f'):
4788 case 'f':
4789 case ' ':
4790 if (s->eof) {
4791 view->count = 0;
4792 break;
4794 i = 0;
4795 while (!s->eof && i++ < nscroll) {
4796 linelen = getline(&line, &linesize, s->f);
4797 s->first_displayed_line++;
4798 if (linelen == -1) {
4799 if (feof(s->f)) {
4800 s->eof = 1;
4801 } else
4802 err = got_ferror(s->f, GOT_ERR_IO);
4803 break;
4806 free(line);
4807 break;
4808 case '[':
4809 if (s->diff_context > 0) {
4810 s->diff_context--;
4811 s->matched_line = 0;
4812 diff_view_indicate_progress(view);
4813 err = create_diff(s);
4814 if (s->first_displayed_line + view->nlines - 1 >
4815 s->nlines) {
4816 s->first_displayed_line = 1;
4817 s->last_displayed_line = view->nlines;
4819 } else
4820 view->count = 0;
4821 break;
4822 case ']':
4823 if (s->diff_context < GOT_DIFF_MAX_CONTEXT) {
4824 s->diff_context++;
4825 s->matched_line = 0;
4826 diff_view_indicate_progress(view);
4827 err = create_diff(s);
4828 } else
4829 view->count = 0;
4830 break;
4831 case '<':
4832 case ',':
4833 case 'K':
4834 up = 1;
4835 /* FALL THROUGH */
4836 case '>':
4837 case '.':
4838 case 'J':
4839 if (s->parent_view == NULL) {
4840 view->count = 0;
4841 break;
4843 s->parent_view->count = view->count;
4845 if (s->parent_view->type == TOG_VIEW_LOG) {
4846 ls = &s->parent_view->state.log;
4847 old_selected_entry = ls->selected_entry;
4849 err = input_log_view(NULL, s->parent_view,
4850 up ? KEY_UP : KEY_DOWN);
4851 if (err)
4852 break;
4853 view->count = s->parent_view->count;
4855 if (old_selected_entry == ls->selected_entry)
4856 break;
4858 err = set_selected_commit(s, ls->selected_entry);
4859 if (err)
4860 break;
4861 } else if (s->parent_view->type == TOG_VIEW_BLAME) {
4862 struct tog_blame_view_state *bs;
4863 struct got_object_id *id, *prev_id;
4865 bs = &s->parent_view->state.blame;
4866 prev_id = get_annotation_for_line(bs->blame.lines,
4867 bs->blame.nlines, bs->last_diffed_line);
4869 err = input_blame_view(&view, s->parent_view,
4870 up ? KEY_UP : KEY_DOWN);
4871 if (err)
4872 break;
4873 view->count = s->parent_view->count;
4875 if (prev_id == NULL)
4876 break;
4877 id = get_selected_commit_id(bs->blame.lines,
4878 bs->blame.nlines, bs->first_displayed_line,
4879 bs->selected_line);
4880 if (id == NULL)
4881 break;
4883 if (!got_object_id_cmp(prev_id, id))
4884 break;
4886 err = input_blame_view(&view, s->parent_view, KEY_ENTER);
4887 if (err)
4888 break;
4890 s->first_displayed_line = 1;
4891 s->last_displayed_line = view->nlines;
4892 s->matched_line = 0;
4893 view->x = 0;
4895 diff_view_indicate_progress(view);
4896 err = create_diff(s);
4897 break;
4898 default:
4899 view->count = 0;
4900 break;
4903 return err;
4906 static const struct got_error *
4907 cmd_diff(int argc, char *argv[])
4909 const struct got_error *error = NULL;
4910 struct got_repository *repo = NULL;
4911 struct got_worktree *worktree = NULL;
4912 struct got_object_id *id1 = NULL, *id2 = NULL;
4913 char *repo_path = NULL, *cwd = NULL;
4914 char *id_str1 = NULL, *id_str2 = NULL;
4915 char *label1 = NULL, *label2 = NULL;
4916 int diff_context = 3, ignore_whitespace = 0;
4917 int ch, force_text_diff = 0;
4918 const char *errstr;
4919 struct tog_view *view;
4920 int *pack_fds = NULL;
4922 while ((ch = getopt(argc, argv, "aC:r:w")) != -1) {
4923 switch (ch) {
4924 case 'a':
4925 force_text_diff = 1;
4926 break;
4927 case 'C':
4928 diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
4929 &errstr);
4930 if (errstr != NULL)
4931 errx(1, "number of context lines is %s: %s",
4932 errstr, errstr);
4933 break;
4934 case 'r':
4935 repo_path = realpath(optarg, NULL);
4936 if (repo_path == NULL)
4937 return got_error_from_errno2("realpath",
4938 optarg);
4939 got_path_strip_trailing_slashes(repo_path);
4940 break;
4941 case 'w':
4942 ignore_whitespace = 1;
4943 break;
4944 default:
4945 usage_diff();
4946 /* NOTREACHED */
4950 argc -= optind;
4951 argv += optind;
4953 if (argc == 0) {
4954 usage_diff(); /* TODO show local worktree changes */
4955 } else if (argc == 2) {
4956 id_str1 = argv[0];
4957 id_str2 = argv[1];
4958 } else
4959 usage_diff();
4961 error = got_repo_pack_fds_open(&pack_fds);
4962 if (error)
4963 goto done;
4965 if (repo_path == NULL) {
4966 cwd = getcwd(NULL, 0);
4967 if (cwd == NULL)
4968 return got_error_from_errno("getcwd");
4969 error = got_worktree_open(&worktree, cwd);
4970 if (error && error->code != GOT_ERR_NOT_WORKTREE)
4971 goto done;
4972 if (worktree)
4973 repo_path =
4974 strdup(got_worktree_get_repo_path(worktree));
4975 else
4976 repo_path = strdup(cwd);
4977 if (repo_path == NULL) {
4978 error = got_error_from_errno("strdup");
4979 goto done;
4983 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
4984 if (error)
4985 goto done;
4987 init_curses();
4989 error = apply_unveil(got_repo_get_path(repo), NULL);
4990 if (error)
4991 goto done;
4993 error = tog_load_refs(repo, 0);
4994 if (error)
4995 goto done;
4997 error = got_repo_match_object_id(&id1, &label1, id_str1,
4998 GOT_OBJ_TYPE_ANY, &tog_refs, repo);
4999 if (error)
5000 goto done;
5002 error = got_repo_match_object_id(&id2, &label2, id_str2,
5003 GOT_OBJ_TYPE_ANY, &tog_refs, repo);
5004 if (error)
5005 goto done;
5007 view = view_open(0, 0, 0, 0, TOG_VIEW_DIFF);
5008 if (view == NULL) {
5009 error = got_error_from_errno("view_open");
5010 goto done;
5012 error = open_diff_view(view, id1, id2, label1, label2, diff_context,
5013 ignore_whitespace, force_text_diff, NULL, repo);
5014 if (error)
5015 goto done;
5016 error = view_loop(view);
5017 done:
5018 free(label1);
5019 free(label2);
5020 free(repo_path);
5021 free(cwd);
5022 if (repo) {
5023 const struct got_error *close_err = got_repo_close(repo);
5024 if (error == NULL)
5025 error = close_err;
5027 if (worktree)
5028 got_worktree_close(worktree);
5029 if (pack_fds) {
5030 const struct got_error *pack_err =
5031 got_repo_pack_fds_close(pack_fds);
5032 if (error == NULL)
5033 error = pack_err;
5035 tog_free_refs();
5036 return error;
5039 __dead static void
5040 usage_blame(void)
5042 endwin();
5043 fprintf(stderr,
5044 "usage: %s blame [-c commit] [-r repository-path] path\n",
5045 getprogname());
5046 exit(1);
5049 struct tog_blame_line {
5050 int annotated;
5051 struct got_object_id *id;
5054 static const struct got_error *
5055 draw_blame(struct tog_view *view)
5057 struct tog_blame_view_state *s = &view->state.blame;
5058 struct tog_blame *blame = &s->blame;
5059 regmatch_t *regmatch = &view->regmatch;
5060 const struct got_error *err;
5061 int lineno = 0, nprinted = 0;
5062 char *line = NULL;
5063 size_t linesize = 0;
5064 ssize_t linelen;
5065 wchar_t *wline;
5066 int width;
5067 struct tog_blame_line *blame_line;
5068 struct got_object_id *prev_id = NULL;
5069 char *id_str;
5070 struct tog_color *tc;
5072 err = got_object_id_str(&id_str, &s->blamed_commit->id);
5073 if (err)
5074 return err;
5076 rewind(blame->f);
5077 werase(view->window);
5079 if (asprintf(&line, "commit %s", id_str) == -1) {
5080 err = got_error_from_errno("asprintf");
5081 free(id_str);
5082 return err;
5085 err = format_line(&wline, &width, NULL, line, 0, view->ncols, 0, 0);
5086 free(line);
5087 line = NULL;
5088 if (err)
5089 return err;
5090 if (view_needs_focus_indication(view))
5091 wstandout(view->window);
5092 tc = get_color(&s->colors, TOG_COLOR_COMMIT);
5093 if (tc)
5094 wattr_on(view->window,
5095 COLOR_PAIR(tc->colorpair), NULL);
5096 waddwstr(view->window, wline);
5097 if (tc)
5098 wattr_off(view->window,
5099 COLOR_PAIR(tc->colorpair), NULL);
5100 if (view_needs_focus_indication(view))
5101 wstandend(view->window);
5102 free(wline);
5103 wline = NULL;
5104 if (width < view->ncols - 1)
5105 waddch(view->window, '\n');
5107 if (asprintf(&line, "[%d/%d] %s%s",
5108 s->first_displayed_line - 1 + s->selected_line, blame->nlines,
5109 s->blame_complete ? "" : "annotating... ", s->path) == -1) {
5110 free(id_str);
5111 return got_error_from_errno("asprintf");
5113 free(id_str);
5114 err = format_line(&wline, &width, NULL, line, 0, view->ncols, 0, 0);
5115 free(line);
5116 line = NULL;
5117 if (err)
5118 return err;
5119 waddwstr(view->window, wline);
5120 free(wline);
5121 wline = NULL;
5122 if (width < view->ncols - 1)
5123 waddch(view->window, '\n');
5125 s->eof = 0;
5126 view->maxx = 0;
5127 while (nprinted < view->nlines - 2) {
5128 linelen = getline(&line, &linesize, blame->f);
5129 if (linelen == -1) {
5130 if (feof(blame->f)) {
5131 s->eof = 1;
5132 break;
5134 free(line);
5135 return got_ferror(blame->f, GOT_ERR_IO);
5137 if (++lineno < s->first_displayed_line)
5138 continue;
5140 /* Set view->maxx based on full line length. */
5141 err = format_line(&wline, &width, NULL, line, 0, INT_MAX, 9, 1);
5142 if (err) {
5143 free(line);
5144 return err;
5146 free(wline);
5147 wline = NULL;
5148 view->maxx = MAX(view->maxx, width);
5150 if (nprinted == s->selected_line - 1)
5151 wstandout(view->window);
5153 if (blame->nlines > 0) {
5154 blame_line = &blame->lines[lineno - 1];
5155 if (blame_line->annotated && prev_id &&
5156 got_object_id_cmp(prev_id, blame_line->id) == 0 &&
5157 !(nprinted == s->selected_line - 1)) {
5158 waddstr(view->window, " ");
5159 } else if (blame_line->annotated) {
5160 char *id_str;
5161 err = got_object_id_str(&id_str,
5162 blame_line->id);
5163 if (err) {
5164 free(line);
5165 return err;
5167 tc = get_color(&s->colors, TOG_COLOR_COMMIT);
5168 if (tc)
5169 wattr_on(view->window,
5170 COLOR_PAIR(tc->colorpair), NULL);
5171 wprintw(view->window, "%.8s", id_str);
5172 if (tc)
5173 wattr_off(view->window,
5174 COLOR_PAIR(tc->colorpair), NULL);
5175 free(id_str);
5176 prev_id = blame_line->id;
5177 } else {
5178 waddstr(view->window, "........");
5179 prev_id = NULL;
5181 } else {
5182 waddstr(view->window, "........");
5183 prev_id = NULL;
5186 if (nprinted == s->selected_line - 1)
5187 wstandend(view->window);
5188 waddstr(view->window, " ");
5190 if (view->ncols <= 9) {
5191 width = 9;
5192 } else if (s->first_displayed_line + nprinted ==
5193 s->matched_line &&
5194 regmatch->rm_so >= 0 && regmatch->rm_so < regmatch->rm_eo) {
5195 err = add_matched_line(&width, line, view->ncols - 9, 9,
5196 view->window, view->x, regmatch);
5197 if (err) {
5198 free(line);
5199 return err;
5201 width += 9;
5202 } else {
5203 int skip;
5204 err = format_line(&wline, &width, &skip, line,
5205 view->x, view->ncols - 9, 9, 1);
5206 if (err) {
5207 free(line);
5208 return err;
5210 waddwstr(view->window, &wline[skip]);
5211 width += 9;
5212 free(wline);
5213 wline = NULL;
5216 if (width <= view->ncols - 1)
5217 waddch(view->window, '\n');
5218 if (++nprinted == 1)
5219 s->first_displayed_line = lineno;
5221 free(line);
5222 s->last_displayed_line = lineno;
5224 view_border(view);
5226 return NULL;
5229 static const struct got_error *
5230 blame_cb(void *arg, int nlines, int lineno,
5231 struct got_commit_object *commit, struct got_object_id *id)
5233 const struct got_error *err = NULL;
5234 struct tog_blame_cb_args *a = arg;
5235 struct tog_blame_line *line;
5236 int errcode;
5238 if (nlines != a->nlines ||
5239 (lineno != -1 && lineno < 1) || lineno > a->nlines)
5240 return got_error(GOT_ERR_RANGE);
5242 errcode = pthread_mutex_lock(&tog_mutex);
5243 if (errcode)
5244 return got_error_set_errno(errcode, "pthread_mutex_lock");
5246 if (*a->quit) { /* user has quit the blame view */
5247 err = got_error(GOT_ERR_ITER_COMPLETED);
5248 goto done;
5251 if (lineno == -1)
5252 goto done; /* no change in this commit */
5254 line = &a->lines[lineno - 1];
5255 if (line->annotated)
5256 goto done;
5258 line->id = got_object_id_dup(id);
5259 if (line->id == NULL) {
5260 err = got_error_from_errno("got_object_id_dup");
5261 goto done;
5263 line->annotated = 1;
5264 done:
5265 errcode = pthread_mutex_unlock(&tog_mutex);
5266 if (errcode)
5267 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
5268 return err;
5271 static void *
5272 blame_thread(void *arg)
5274 const struct got_error *err, *close_err;
5275 struct tog_blame_thread_args *ta = arg;
5276 struct tog_blame_cb_args *a = ta->cb_args;
5277 int errcode, fd1 = -1, fd2 = -1;
5278 FILE *f1 = NULL, *f2 = NULL;
5280 fd1 = got_opentempfd();
5281 if (fd1 == -1)
5282 return (void *)got_error_from_errno("got_opentempfd");
5284 fd2 = got_opentempfd();
5285 if (fd2 == -1) {
5286 err = got_error_from_errno("got_opentempfd");
5287 goto done;
5290 f1 = got_opentemp();
5291 if (f1 == NULL) {
5292 err = (void *)got_error_from_errno("got_opentemp");
5293 goto done;
5295 f2 = got_opentemp();
5296 if (f2 == NULL) {
5297 err = (void *)got_error_from_errno("got_opentemp");
5298 goto done;
5301 err = block_signals_used_by_main_thread();
5302 if (err)
5303 goto done;
5305 err = got_blame(ta->path, a->commit_id, ta->repo,
5306 tog_diff_algo, blame_cb, ta->cb_args,
5307 ta->cancel_cb, ta->cancel_arg, fd1, fd2, f1, f2);
5308 if (err && err->code == GOT_ERR_CANCELLED)
5309 err = NULL;
5311 errcode = pthread_mutex_lock(&tog_mutex);
5312 if (errcode) {
5313 err = got_error_set_errno(errcode, "pthread_mutex_lock");
5314 goto done;
5317 close_err = got_repo_close(ta->repo);
5318 if (err == NULL)
5319 err = close_err;
5320 ta->repo = NULL;
5321 *ta->complete = 1;
5323 errcode = pthread_mutex_unlock(&tog_mutex);
5324 if (errcode && err == NULL)
5325 err = got_error_set_errno(errcode, "pthread_mutex_unlock");
5327 done:
5328 if (fd1 != -1 && close(fd1) == -1 && err == NULL)
5329 err = got_error_from_errno("close");
5330 if (fd2 != -1 && close(fd2) == -1 && err == NULL)
5331 err = got_error_from_errno("close");
5332 if (f1 && fclose(f1) == EOF && err == NULL)
5333 err = got_error_from_errno("fclose");
5334 if (f2 && fclose(f2) == EOF && err == NULL)
5335 err = got_error_from_errno("fclose");
5337 return (void *)err;
5340 static struct got_object_id *
5341 get_selected_commit_id(struct tog_blame_line *lines, int nlines,
5342 int first_displayed_line, int selected_line)
5344 struct tog_blame_line *line;
5346 if (nlines <= 0)
5347 return NULL;
5349 line = &lines[first_displayed_line - 1 + selected_line - 1];
5350 if (!line->annotated)
5351 return NULL;
5353 return line->id;
5356 static struct got_object_id *
5357 get_annotation_for_line(struct tog_blame_line *lines, int nlines,
5358 int lineno)
5360 struct tog_blame_line *line;
5362 if (nlines <= 0 || lineno >= nlines)
5363 return NULL;
5365 line = &lines[lineno - 1];
5366 if (!line->annotated)
5367 return NULL;
5369 return line->id;
5372 static const struct got_error *
5373 stop_blame(struct tog_blame *blame)
5375 const struct got_error *err = NULL;
5376 int i;
5378 if (blame->thread) {
5379 int errcode;
5380 errcode = pthread_mutex_unlock(&tog_mutex);
5381 if (errcode)
5382 return got_error_set_errno(errcode,
5383 "pthread_mutex_unlock");
5384 errcode = pthread_join(blame->thread, (void **)&err);
5385 if (errcode)
5386 return got_error_set_errno(errcode, "pthread_join");
5387 errcode = pthread_mutex_lock(&tog_mutex);
5388 if (errcode)
5389 return got_error_set_errno(errcode,
5390 "pthread_mutex_lock");
5391 if (err && err->code == GOT_ERR_ITER_COMPLETED)
5392 err = NULL;
5393 blame->thread = 0; //NULL;
5395 if (blame->thread_args.repo) {
5396 const struct got_error *close_err;
5397 close_err = got_repo_close(blame->thread_args.repo);
5398 if (err == NULL)
5399 err = close_err;
5400 blame->thread_args.repo = NULL;
5402 if (blame->f) {
5403 if (fclose(blame->f) == EOF && err == NULL)
5404 err = got_error_from_errno("fclose");
5405 blame->f = NULL;
5407 if (blame->lines) {
5408 for (i = 0; i < blame->nlines; i++)
5409 free(blame->lines[i].id);
5410 free(blame->lines);
5411 blame->lines = NULL;
5413 free(blame->cb_args.commit_id);
5414 blame->cb_args.commit_id = NULL;
5415 if (blame->pack_fds) {
5416 const struct got_error *pack_err =
5417 got_repo_pack_fds_close(blame->pack_fds);
5418 if (err == NULL)
5419 err = pack_err;
5420 blame->pack_fds = NULL;
5422 return err;
5425 static const struct got_error *
5426 cancel_blame_view(void *arg)
5428 const struct got_error *err = NULL;
5429 int *done = arg;
5430 int errcode;
5432 errcode = pthread_mutex_lock(&tog_mutex);
5433 if (errcode)
5434 return got_error_set_errno(errcode,
5435 "pthread_mutex_unlock");
5437 if (*done)
5438 err = got_error(GOT_ERR_CANCELLED);
5440 errcode = pthread_mutex_unlock(&tog_mutex);
5441 if (errcode)
5442 return got_error_set_errno(errcode,
5443 "pthread_mutex_lock");
5445 return err;
5448 static const struct got_error *
5449 run_blame(struct tog_view *view)
5451 struct tog_blame_view_state *s = &view->state.blame;
5452 struct tog_blame *blame = &s->blame;
5453 const struct got_error *err = NULL;
5454 struct got_commit_object *commit = NULL;
5455 struct got_blob_object *blob = NULL;
5456 struct got_repository *thread_repo = NULL;
5457 struct got_object_id *obj_id = NULL;
5458 int obj_type, fd = -1;
5459 int *pack_fds = NULL;
5461 err = got_object_open_as_commit(&commit, s->repo,
5462 &s->blamed_commit->id);
5463 if (err)
5464 return err;
5466 fd = got_opentempfd();
5467 if (fd == -1) {
5468 err = got_error_from_errno("got_opentempfd");
5469 goto done;
5472 err = got_object_id_by_path(&obj_id, s->repo, commit, s->path);
5473 if (err)
5474 goto done;
5476 err = got_object_get_type(&obj_type, s->repo, obj_id);
5477 if (err)
5478 goto done;
5480 if (obj_type != GOT_OBJ_TYPE_BLOB) {
5481 err = got_error(GOT_ERR_OBJ_TYPE);
5482 goto done;
5485 err = got_object_open_as_blob(&blob, s->repo, obj_id, 8192, fd);
5486 if (err)
5487 goto done;
5488 blame->f = got_opentemp();
5489 if (blame->f == NULL) {
5490 err = got_error_from_errno("got_opentemp");
5491 goto done;
5493 err = got_object_blob_dump_to_file(&blame->filesize, &blame->nlines,
5494 &blame->line_offsets, blame->f, blob);
5495 if (err)
5496 goto done;
5497 if (blame->nlines == 0) {
5498 s->blame_complete = 1;
5499 goto done;
5502 /* Don't include \n at EOF in the blame line count. */
5503 if (blame->line_offsets[blame->nlines - 1] == blame->filesize)
5504 blame->nlines--;
5506 blame->lines = calloc(blame->nlines, sizeof(*blame->lines));
5507 if (blame->lines == NULL) {
5508 err = got_error_from_errno("calloc");
5509 goto done;
5512 err = got_repo_pack_fds_open(&pack_fds);
5513 if (err)
5514 goto done;
5515 err = got_repo_open(&thread_repo, got_repo_get_path(s->repo), NULL,
5516 pack_fds);
5517 if (err)
5518 goto done;
5520 blame->pack_fds = pack_fds;
5521 blame->cb_args.view = view;
5522 blame->cb_args.lines = blame->lines;
5523 blame->cb_args.nlines = blame->nlines;
5524 blame->cb_args.commit_id = got_object_id_dup(&s->blamed_commit->id);
5525 if (blame->cb_args.commit_id == NULL) {
5526 err = got_error_from_errno("got_object_id_dup");
5527 goto done;
5529 blame->cb_args.quit = &s->done;
5531 blame->thread_args.path = s->path;
5532 blame->thread_args.repo = thread_repo;
5533 blame->thread_args.cb_args = &blame->cb_args;
5534 blame->thread_args.complete = &s->blame_complete;
5535 blame->thread_args.cancel_cb = cancel_blame_view;
5536 blame->thread_args.cancel_arg = &s->done;
5537 s->blame_complete = 0;
5539 if (s->first_displayed_line + view->nlines - 1 > blame->nlines) {
5540 s->first_displayed_line = 1;
5541 s->last_displayed_line = view->nlines;
5542 s->selected_line = 1;
5544 s->matched_line = 0;
5546 done:
5547 if (commit)
5548 got_object_commit_close(commit);
5549 if (fd != -1 && close(fd) == -1 && err == NULL)
5550 err = got_error_from_errno("close");
5551 if (blob)
5552 got_object_blob_close(blob);
5553 free(obj_id);
5554 if (err)
5555 stop_blame(blame);
5556 return err;
5559 static const struct got_error *
5560 open_blame_view(struct tog_view *view, char *path,
5561 struct got_object_id *commit_id, struct got_repository *repo)
5563 const struct got_error *err = NULL;
5564 struct tog_blame_view_state *s = &view->state.blame;
5566 STAILQ_INIT(&s->blamed_commits);
5568 s->path = strdup(path);
5569 if (s->path == NULL)
5570 return got_error_from_errno("strdup");
5572 err = got_object_qid_alloc(&s->blamed_commit, commit_id);
5573 if (err) {
5574 free(s->path);
5575 return err;
5578 STAILQ_INSERT_HEAD(&s->blamed_commits, s->blamed_commit, entry);
5579 s->first_displayed_line = 1;
5580 s->last_displayed_line = view->nlines;
5581 s->selected_line = 1;
5582 s->blame_complete = 0;
5583 s->repo = repo;
5584 s->commit_id = commit_id;
5585 memset(&s->blame, 0, sizeof(s->blame));
5587 STAILQ_INIT(&s->colors);
5588 if (has_colors() && getenv("TOG_COLORS") != NULL) {
5589 err = add_color(&s->colors, "^", TOG_COLOR_COMMIT,
5590 get_color_value("TOG_COLOR_COMMIT"));
5591 if (err)
5592 return err;
5595 view->show = show_blame_view;
5596 view->input = input_blame_view;
5597 view->reset = reset_blame_view;
5598 view->close = close_blame_view;
5599 view->search_start = search_start_blame_view;
5600 view->search_next = search_next_blame_view;
5602 return run_blame(view);
5605 static const struct got_error *
5606 close_blame_view(struct tog_view *view)
5608 const struct got_error *err = NULL;
5609 struct tog_blame_view_state *s = &view->state.blame;
5611 if (s->blame.thread)
5612 err = stop_blame(&s->blame);
5614 while (!STAILQ_EMPTY(&s->blamed_commits)) {
5615 struct got_object_qid *blamed_commit;
5616 blamed_commit = STAILQ_FIRST(&s->blamed_commits);
5617 STAILQ_REMOVE_HEAD(&s->blamed_commits, entry);
5618 got_object_qid_free(blamed_commit);
5621 free(s->path);
5622 free_colors(&s->colors);
5623 return err;
5626 static const struct got_error *
5627 search_start_blame_view(struct tog_view *view)
5629 struct tog_blame_view_state *s = &view->state.blame;
5631 s->matched_line = 0;
5632 return NULL;
5635 static const struct got_error *
5636 search_next_blame_view(struct tog_view *view)
5638 struct tog_blame_view_state *s = &view->state.blame;
5639 const struct got_error *err = NULL;
5640 int lineno;
5641 char *line = NULL;
5642 size_t linesize = 0;
5643 ssize_t linelen;
5645 if (!view->searching) {
5646 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5647 return NULL;
5650 if (s->matched_line) {
5651 if (view->searching == TOG_SEARCH_FORWARD)
5652 lineno = s->matched_line + 1;
5653 else
5654 lineno = s->matched_line - 1;
5655 } else
5656 lineno = s->first_displayed_line - 1 + s->selected_line;
5658 while (1) {
5659 off_t offset;
5661 if (lineno <= 0 || lineno > s->blame.nlines) {
5662 if (s->matched_line == 0) {
5663 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5664 break;
5667 if (view->searching == TOG_SEARCH_FORWARD)
5668 lineno = 1;
5669 else
5670 lineno = s->blame.nlines;
5673 offset = s->blame.line_offsets[lineno - 1];
5674 if (fseeko(s->blame.f, offset, SEEK_SET) != 0) {
5675 free(line);
5676 return got_error_from_errno("fseeko");
5678 linelen = getline(&line, &linesize, s->blame.f);
5679 if (linelen != -1) {
5680 char *exstr;
5681 err = expand_tab(&exstr, line);
5682 if (err)
5683 break;
5684 if (match_line(exstr, &view->regex, 1,
5685 &view->regmatch)) {
5686 view->search_next_done = TOG_SEARCH_HAVE_MORE;
5687 s->matched_line = lineno;
5688 free(exstr);
5689 break;
5691 free(exstr);
5693 if (view->searching == TOG_SEARCH_FORWARD)
5694 lineno++;
5695 else
5696 lineno--;
5698 free(line);
5700 if (s->matched_line) {
5701 s->first_displayed_line = s->matched_line;
5702 s->selected_line = 1;
5705 return err;
5708 static const struct got_error *
5709 show_blame_view(struct tog_view *view)
5711 const struct got_error *err = NULL;
5712 struct tog_blame_view_state *s = &view->state.blame;
5713 int errcode;
5715 if (s->blame.thread == 0 && !s->blame_complete) {
5716 errcode = pthread_create(&s->blame.thread, NULL, blame_thread,
5717 &s->blame.thread_args);
5718 if (errcode)
5719 return got_error_set_errno(errcode, "pthread_create");
5721 halfdelay(1); /* fast refresh while annotating */
5724 if (s->blame_complete)
5725 halfdelay(10); /* disable fast refresh */
5727 err = draw_blame(view);
5729 view_border(view);
5730 return err;
5733 static const struct got_error *
5734 log_annotated_line(struct tog_view **new_view, int begin_y, int begin_x,
5735 struct got_repository *repo, struct got_object_id *id)
5737 struct tog_view *log_view;
5738 const struct got_error *err = NULL;
5740 *new_view = NULL;
5742 log_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_LOG);
5743 if (log_view == NULL)
5744 return got_error_from_errno("view_open");
5746 err = open_log_view(log_view, id, repo, GOT_REF_HEAD, "", 0);
5747 if (err)
5748 view_close(log_view);
5749 else
5750 *new_view = log_view;
5752 return err;
5755 static const struct got_error *
5756 input_blame_view(struct tog_view **new_view, struct tog_view *view, int ch)
5758 const struct got_error *err = NULL, *thread_err = NULL;
5759 struct tog_view *diff_view, *log_view;
5760 struct tog_blame_view_state *s = &view->state.blame;
5761 int eos, nscroll, begin_y = 0, begin_x = 0;
5763 eos = nscroll = view->nlines - 2;
5764 if (view_is_hsplit_top(view))
5765 --eos; /* border */
5767 switch (ch) {
5768 case '0':
5769 view->x = 0;
5770 break;
5771 case '$':
5772 view->x = MAX(view->maxx - view->ncols / 3, 0);
5773 view->count = 0;
5774 break;
5775 case KEY_RIGHT:
5776 case 'l':
5777 if (view->x + view->ncols / 3 < view->maxx)
5778 view->x += 2; /* move two columns right */
5779 else
5780 view->count = 0;
5781 break;
5782 case KEY_LEFT:
5783 case 'h':
5784 view->x -= MIN(view->x, 2); /* move two columns back */
5785 if (view->x <= 0)
5786 view->count = 0;
5787 break;
5788 case 'q':
5789 s->done = 1;
5790 break;
5791 case 'g':
5792 case KEY_HOME:
5793 s->selected_line = 1;
5794 s->first_displayed_line = 1;
5795 view->count = 0;
5796 break;
5797 case 'G':
5798 case KEY_END:
5799 if (s->blame.nlines < eos) {
5800 s->selected_line = s->blame.nlines;
5801 s->first_displayed_line = 1;
5802 } else {
5803 s->selected_line = eos;
5804 s->first_displayed_line = s->blame.nlines - (eos - 1);
5806 view->count = 0;
5807 break;
5808 case 'k':
5809 case KEY_UP:
5810 case CTRL('p'):
5811 if (s->selected_line > 1)
5812 s->selected_line--;
5813 else if (s->selected_line == 1 &&
5814 s->first_displayed_line > 1)
5815 s->first_displayed_line--;
5816 else
5817 view->count = 0;
5818 break;
5819 case CTRL('u'):
5820 case 'u':
5821 nscroll /= 2;
5822 /* FALL THROUGH */
5823 case KEY_PPAGE:
5824 case CTRL('b'):
5825 case 'b':
5826 if (s->first_displayed_line == 1) {
5827 if (view->count > 1)
5828 nscroll += nscroll;
5829 s->selected_line = MAX(1, s->selected_line - nscroll);
5830 view->count = 0;
5831 break;
5833 if (s->first_displayed_line > nscroll)
5834 s->first_displayed_line -= nscroll;
5835 else
5836 s->first_displayed_line = 1;
5837 break;
5838 case 'j':
5839 case KEY_DOWN:
5840 case CTRL('n'):
5841 if (s->selected_line < eos && s->first_displayed_line +
5842 s->selected_line <= s->blame.nlines)
5843 s->selected_line++;
5844 else if (s->first_displayed_line < s->blame.nlines - (eos - 1))
5845 s->first_displayed_line++;
5846 else
5847 view->count = 0;
5848 break;
5849 case 'c':
5850 case 'p': {
5851 struct got_object_id *id = NULL;
5853 view->count = 0;
5854 id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
5855 s->first_displayed_line, s->selected_line);
5856 if (id == NULL)
5857 break;
5858 if (ch == 'p') {
5859 struct got_commit_object *commit, *pcommit;
5860 struct got_object_qid *pid;
5861 struct got_object_id *blob_id = NULL;
5862 int obj_type;
5863 err = got_object_open_as_commit(&commit,
5864 s->repo, id);
5865 if (err)
5866 break;
5867 pid = STAILQ_FIRST(
5868 got_object_commit_get_parent_ids(commit));
5869 if (pid == NULL) {
5870 got_object_commit_close(commit);
5871 break;
5873 /* Check if path history ends here. */
5874 err = got_object_open_as_commit(&pcommit,
5875 s->repo, &pid->id);
5876 if (err)
5877 break;
5878 err = got_object_id_by_path(&blob_id, s->repo,
5879 pcommit, s->path);
5880 got_object_commit_close(pcommit);
5881 if (err) {
5882 if (err->code == GOT_ERR_NO_TREE_ENTRY)
5883 err = NULL;
5884 got_object_commit_close(commit);
5885 break;
5887 err = got_object_get_type(&obj_type, s->repo,
5888 blob_id);
5889 free(blob_id);
5890 /* Can't blame non-blob type objects. */
5891 if (obj_type != GOT_OBJ_TYPE_BLOB) {
5892 got_object_commit_close(commit);
5893 break;
5895 err = got_object_qid_alloc(&s->blamed_commit,
5896 &pid->id);
5897 got_object_commit_close(commit);
5898 } else {
5899 if (got_object_id_cmp(id,
5900 &s->blamed_commit->id) == 0)
5901 break;
5902 err = got_object_qid_alloc(&s->blamed_commit,
5903 id);
5905 if (err)
5906 break;
5907 s->done = 1;
5908 thread_err = stop_blame(&s->blame);
5909 s->done = 0;
5910 if (thread_err)
5911 break;
5912 STAILQ_INSERT_HEAD(&s->blamed_commits,
5913 s->blamed_commit, entry);
5914 err = run_blame(view);
5915 if (err)
5916 break;
5917 break;
5919 case 'C': {
5920 struct got_object_qid *first;
5922 view->count = 0;
5923 first = STAILQ_FIRST(&s->blamed_commits);
5924 if (!got_object_id_cmp(&first->id, s->commit_id))
5925 break;
5926 s->done = 1;
5927 thread_err = stop_blame(&s->blame);
5928 s->done = 0;
5929 if (thread_err)
5930 break;
5931 STAILQ_REMOVE_HEAD(&s->blamed_commits, entry);
5932 got_object_qid_free(s->blamed_commit);
5933 s->blamed_commit =
5934 STAILQ_FIRST(&s->blamed_commits);
5935 err = run_blame(view);
5936 if (err)
5937 break;
5938 break;
5940 case 'L': {
5941 struct got_object_id *id = NULL;
5943 view->count = 0;
5944 id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
5945 s->first_displayed_line, s->selected_line);
5946 if (id == NULL)
5947 break;
5949 if (view_is_parent_view(view))
5950 view_get_split(view, &begin_y, &begin_x);
5951 err = log_annotated_line(&log_view, begin_y, begin_x,
5952 s->repo, id);
5953 if (err)
5954 break;
5955 if (view_is_parent_view(view) &&
5956 view->mode == TOG_VIEW_SPLIT_HRZN) {
5957 err = view_init_hsplit(view, begin_y);
5958 if (err)
5959 break;
5962 view->focussed = 0;
5963 log_view->focussed = 1;
5964 log_view->mode = view->mode;
5965 log_view->nlines = view->lines - begin_y;
5966 if (view_is_parent_view(view)) {
5967 view_transfer_size(log_view, view);
5968 err = view_close_child(view);
5969 if (err)
5970 return err;
5971 err = view_set_child(view, log_view);
5972 if (err)
5973 return err;
5974 view->focus_child = 1;
5975 } else
5976 *new_view = log_view;
5977 break;
5979 case KEY_ENTER:
5980 case '\r': {
5981 struct got_object_id *id = NULL;
5982 struct got_object_qid *pid;
5983 struct got_commit_object *commit = NULL;
5985 view->count = 0;
5986 id = get_selected_commit_id(s->blame.lines, s->blame.nlines,
5987 s->first_displayed_line, s->selected_line);
5988 if (id == NULL)
5989 break;
5990 err = got_object_open_as_commit(&commit, s->repo, id);
5991 if (err)
5992 break;
5993 pid = STAILQ_FIRST(got_object_commit_get_parent_ids(commit));
5994 if (*new_view) {
5995 /* traversed from diff view, release diff resources */
5996 err = close_diff_view(*new_view);
5997 if (err)
5998 break;
5999 diff_view = *new_view;
6000 } else {
6001 if (view_is_parent_view(view))
6002 view_get_split(view, &begin_y, &begin_x);
6004 diff_view = view_open(0, 0, begin_y, begin_x,
6005 TOG_VIEW_DIFF);
6006 if (diff_view == NULL) {
6007 got_object_commit_close(commit);
6008 err = got_error_from_errno("view_open");
6009 break;
6012 err = open_diff_view(diff_view, pid ? &pid->id : NULL,
6013 id, NULL, NULL, 3, 0, 0, view, s->repo);
6014 got_object_commit_close(commit);
6015 if (err) {
6016 view_close(diff_view);
6017 break;
6019 s->last_diffed_line = s->first_displayed_line - 1 +
6020 s->selected_line;
6021 if (*new_view)
6022 break; /* still open from active diff view */
6023 if (view_is_parent_view(view) &&
6024 view->mode == TOG_VIEW_SPLIT_HRZN) {
6025 err = view_init_hsplit(view, begin_y);
6026 if (err)
6027 break;
6030 view->focussed = 0;
6031 diff_view->focussed = 1;
6032 diff_view->mode = view->mode;
6033 diff_view->nlines = view->lines - begin_y;
6034 if (view_is_parent_view(view)) {
6035 view_transfer_size(diff_view, view);
6036 err = view_close_child(view);
6037 if (err)
6038 break;
6039 err = view_set_child(view, diff_view);
6040 if (err)
6041 break;
6042 view->focus_child = 1;
6043 } else
6044 *new_view = diff_view;
6045 if (err)
6046 break;
6047 break;
6049 case CTRL('d'):
6050 case 'd':
6051 nscroll /= 2;
6052 /* FALL THROUGH */
6053 case KEY_NPAGE:
6054 case CTRL('f'):
6055 case 'f':
6056 case ' ':
6057 if (s->last_displayed_line >= s->blame.nlines &&
6058 s->selected_line >= MIN(s->blame.nlines,
6059 view->nlines - 2)) {
6060 view->count = 0;
6061 break;
6063 if (s->last_displayed_line >= s->blame.nlines &&
6064 s->selected_line < view->nlines - 2) {
6065 s->selected_line +=
6066 MIN(nscroll, s->last_displayed_line -
6067 s->first_displayed_line - s->selected_line + 1);
6069 if (s->last_displayed_line + nscroll <= s->blame.nlines)
6070 s->first_displayed_line += nscroll;
6071 else
6072 s->first_displayed_line =
6073 s->blame.nlines - (view->nlines - 3);
6074 break;
6075 case KEY_RESIZE:
6076 if (s->selected_line > view->nlines - 2) {
6077 s->selected_line = MIN(s->blame.nlines,
6078 view->nlines - 2);
6080 break;
6081 default:
6082 view->count = 0;
6083 break;
6085 return thread_err ? thread_err : err;
6088 static const struct got_error *
6089 reset_blame_view(struct tog_view *view)
6091 const struct got_error *err;
6092 struct tog_blame_view_state *s = &view->state.blame;
6094 view->count = 0;
6095 s->done = 1;
6096 err = stop_blame(&s->blame);
6097 s->done = 0;
6098 if (err)
6099 return err;
6100 return run_blame(view);
6103 static const struct got_error *
6104 cmd_blame(int argc, char *argv[])
6106 const struct got_error *error;
6107 struct got_repository *repo = NULL;
6108 struct got_worktree *worktree = NULL;
6109 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
6110 char *link_target = NULL;
6111 struct got_object_id *commit_id = NULL;
6112 struct got_commit_object *commit = NULL;
6113 char *commit_id_str = NULL;
6114 int ch;
6115 struct tog_view *view;
6116 int *pack_fds = NULL;
6118 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
6119 switch (ch) {
6120 case 'c':
6121 commit_id_str = optarg;
6122 break;
6123 case 'r':
6124 repo_path = realpath(optarg, NULL);
6125 if (repo_path == NULL)
6126 return got_error_from_errno2("realpath",
6127 optarg);
6128 break;
6129 default:
6130 usage_blame();
6131 /* NOTREACHED */
6135 argc -= optind;
6136 argv += optind;
6138 if (argc != 1)
6139 usage_blame();
6141 error = got_repo_pack_fds_open(&pack_fds);
6142 if (error != NULL)
6143 goto done;
6145 if (repo_path == NULL) {
6146 cwd = getcwd(NULL, 0);
6147 if (cwd == NULL)
6148 return got_error_from_errno("getcwd");
6149 error = got_worktree_open(&worktree, cwd);
6150 if (error && error->code != GOT_ERR_NOT_WORKTREE)
6151 goto done;
6152 if (worktree)
6153 repo_path =
6154 strdup(got_worktree_get_repo_path(worktree));
6155 else
6156 repo_path = strdup(cwd);
6157 if (repo_path == NULL) {
6158 error = got_error_from_errno("strdup");
6159 goto done;
6163 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
6164 if (error != NULL)
6165 goto done;
6167 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv, repo,
6168 worktree);
6169 if (error)
6170 goto done;
6172 init_curses();
6174 error = apply_unveil(got_repo_get_path(repo), NULL);
6175 if (error)
6176 goto done;
6178 error = tog_load_refs(repo, 0);
6179 if (error)
6180 goto done;
6182 if (commit_id_str == NULL) {
6183 struct got_reference *head_ref;
6184 error = got_ref_open(&head_ref, repo, worktree ?
6185 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD, 0);
6186 if (error != NULL)
6187 goto done;
6188 error = got_ref_resolve(&commit_id, repo, head_ref);
6189 got_ref_close(head_ref);
6190 } else {
6191 error = got_repo_match_object_id(&commit_id, NULL,
6192 commit_id_str, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
6194 if (error != NULL)
6195 goto done;
6197 view = view_open(0, 0, 0, 0, TOG_VIEW_BLAME);
6198 if (view == NULL) {
6199 error = got_error_from_errno("view_open");
6200 goto done;
6203 error = got_object_open_as_commit(&commit, repo, commit_id);
6204 if (error)
6205 goto done;
6207 error = got_object_resolve_symlinks(&link_target, in_repo_path,
6208 commit, repo);
6209 if (error)
6210 goto done;
6212 error = open_blame_view(view, link_target ? link_target : in_repo_path,
6213 commit_id, repo);
6214 if (error)
6215 goto done;
6216 if (worktree) {
6217 /* Release work tree lock. */
6218 got_worktree_close(worktree);
6219 worktree = NULL;
6221 error = view_loop(view);
6222 done:
6223 free(repo_path);
6224 free(in_repo_path);
6225 free(link_target);
6226 free(cwd);
6227 free(commit_id);
6228 if (commit)
6229 got_object_commit_close(commit);
6230 if (worktree)
6231 got_worktree_close(worktree);
6232 if (repo) {
6233 const struct got_error *close_err = got_repo_close(repo);
6234 if (error == NULL)
6235 error = close_err;
6237 if (pack_fds) {
6238 const struct got_error *pack_err =
6239 got_repo_pack_fds_close(pack_fds);
6240 if (error == NULL)
6241 error = pack_err;
6243 tog_free_refs();
6244 return error;
6247 static const struct got_error *
6248 draw_tree_entries(struct tog_view *view, const char *parent_path)
6250 struct tog_tree_view_state *s = &view->state.tree;
6251 const struct got_error *err = NULL;
6252 struct got_tree_entry *te;
6253 wchar_t *wline;
6254 struct tog_color *tc;
6255 int width, n, i, nentries;
6256 int limit = view->nlines;
6258 s->ndisplayed = 0;
6259 if (view_is_hsplit_top(view))
6260 --limit; /* border */
6262 werase(view->window);
6264 if (limit == 0)
6265 return NULL;
6267 err = format_line(&wline, &width, NULL, s->tree_label, 0, view->ncols,
6268 0, 0);
6269 if (err)
6270 return err;
6271 if (view_needs_focus_indication(view))
6272 wstandout(view->window);
6273 tc = get_color(&s->colors, TOG_COLOR_COMMIT);
6274 if (tc)
6275 wattr_on(view->window,
6276 COLOR_PAIR(tc->colorpair), NULL);
6277 waddwstr(view->window, wline);
6278 if (tc)
6279 wattr_off(view->window,
6280 COLOR_PAIR(tc->colorpair), NULL);
6281 if (view_needs_focus_indication(view))
6282 wstandend(view->window);
6283 free(wline);
6284 wline = NULL;
6285 if (width < view->ncols - 1)
6286 waddch(view->window, '\n');
6287 if (--limit <= 0)
6288 return NULL;
6289 err = format_line(&wline, &width, NULL, parent_path, 0, view->ncols,
6290 0, 0);
6291 if (err)
6292 return err;
6293 waddwstr(view->window, wline);
6294 free(wline);
6295 wline = NULL;
6296 if (width < view->ncols - 1)
6297 waddch(view->window, '\n');
6298 if (--limit <= 0)
6299 return NULL;
6300 waddch(view->window, '\n');
6301 if (--limit <= 0)
6302 return NULL;
6304 if (s->first_displayed_entry == NULL) {
6305 te = got_object_tree_get_first_entry(s->tree);
6306 if (s->selected == 0) {
6307 if (view->focussed)
6308 wstandout(view->window);
6309 s->selected_entry = NULL;
6311 waddstr(view->window, " ..\n"); /* parent directory */
6312 if (s->selected == 0 && view->focussed)
6313 wstandend(view->window);
6314 s->ndisplayed++;
6315 if (--limit <= 0)
6316 return NULL;
6317 n = 1;
6318 } else {
6319 n = 0;
6320 te = s->first_displayed_entry;
6323 nentries = got_object_tree_get_nentries(s->tree);
6324 for (i = got_tree_entry_get_index(te); i < nentries; i++) {
6325 char *line = NULL, *id_str = NULL, *link_target = NULL;
6326 const char *modestr = "";
6327 mode_t mode;
6329 te = got_object_tree_get_entry(s->tree, i);
6330 mode = got_tree_entry_get_mode(te);
6332 if (s->show_ids) {
6333 err = got_object_id_str(&id_str,
6334 got_tree_entry_get_id(te));
6335 if (err)
6336 return got_error_from_errno(
6337 "got_object_id_str");
6339 if (got_object_tree_entry_is_submodule(te))
6340 modestr = "$";
6341 else if (S_ISLNK(mode)) {
6342 int i;
6344 err = got_tree_entry_get_symlink_target(&link_target,
6345 te, s->repo);
6346 if (err) {
6347 free(id_str);
6348 return err;
6350 for (i = 0; i < strlen(link_target); i++) {
6351 if (!isprint((unsigned char)link_target[i]))
6352 link_target[i] = '?';
6354 modestr = "@";
6356 else if (S_ISDIR(mode))
6357 modestr = "/";
6358 else if (mode & S_IXUSR)
6359 modestr = "*";
6360 if (asprintf(&line, "%s %s%s%s%s", id_str ? id_str : "",
6361 got_tree_entry_get_name(te), modestr,
6362 link_target ? " -> ": "",
6363 link_target ? link_target : "") == -1) {
6364 free(id_str);
6365 free(link_target);
6366 return got_error_from_errno("asprintf");
6368 free(id_str);
6369 free(link_target);
6370 err = format_line(&wline, &width, NULL, line, 0, view->ncols,
6371 0, 0);
6372 if (err) {
6373 free(line);
6374 break;
6376 if (n == s->selected) {
6377 if (view->focussed)
6378 wstandout(view->window);
6379 s->selected_entry = te;
6381 tc = match_color(&s->colors, line);
6382 if (tc)
6383 wattr_on(view->window,
6384 COLOR_PAIR(tc->colorpair), NULL);
6385 waddwstr(view->window, wline);
6386 if (tc)
6387 wattr_off(view->window,
6388 COLOR_PAIR(tc->colorpair), NULL);
6389 if (width < view->ncols - 1)
6390 waddch(view->window, '\n');
6391 if (n == s->selected && view->focussed)
6392 wstandend(view->window);
6393 free(line);
6394 free(wline);
6395 wline = NULL;
6396 n++;
6397 s->ndisplayed++;
6398 s->last_displayed_entry = te;
6399 if (--limit <= 0)
6400 break;
6403 return err;
6406 static void
6407 tree_scroll_up(struct tog_tree_view_state *s, int maxscroll)
6409 struct got_tree_entry *te;
6410 int isroot = s->tree == s->root;
6411 int i = 0;
6413 if (s->first_displayed_entry == NULL)
6414 return;
6416 te = got_tree_entry_get_prev(s->tree, s->first_displayed_entry);
6417 while (i++ < maxscroll) {
6418 if (te == NULL) {
6419 if (!isroot)
6420 s->first_displayed_entry = NULL;
6421 break;
6423 s->first_displayed_entry = te;
6424 te = got_tree_entry_get_prev(s->tree, te);
6428 static const struct got_error *
6429 tree_scroll_down(struct tog_view *view, int maxscroll)
6431 struct tog_tree_view_state *s = &view->state.tree;
6432 struct got_tree_entry *next, *last;
6433 int n = 0;
6435 if (s->first_displayed_entry)
6436 next = got_tree_entry_get_next(s->tree,
6437 s->first_displayed_entry);
6438 else
6439 next = got_object_tree_get_first_entry(s->tree);
6441 last = s->last_displayed_entry;
6442 while (next && n++ < maxscroll) {
6443 if (last)
6444 last = got_tree_entry_get_next(s->tree, last);
6445 if (last || (view->mode == TOG_VIEW_SPLIT_HRZN && next)) {
6446 s->first_displayed_entry = next;
6447 next = got_tree_entry_get_next(s->tree, next);
6451 return NULL;
6454 static const struct got_error *
6455 tree_entry_path(char **path, struct tog_parent_trees *parents,
6456 struct got_tree_entry *te)
6458 const struct got_error *err = NULL;
6459 struct tog_parent_tree *pt;
6460 size_t len = 2; /* for leading slash and NUL */
6462 TAILQ_FOREACH(pt, parents, entry)
6463 len += strlen(got_tree_entry_get_name(pt->selected_entry))
6464 + 1 /* slash */;
6465 if (te)
6466 len += strlen(got_tree_entry_get_name(te));
6468 *path = calloc(1, len);
6469 if (path == NULL)
6470 return got_error_from_errno("calloc");
6472 (*path)[0] = '/';
6473 pt = TAILQ_LAST(parents, tog_parent_trees);
6474 while (pt) {
6475 const char *name = got_tree_entry_get_name(pt->selected_entry);
6476 if (strlcat(*path, name, len) >= len) {
6477 err = got_error(GOT_ERR_NO_SPACE);
6478 goto done;
6480 if (strlcat(*path, "/", len) >= len) {
6481 err = got_error(GOT_ERR_NO_SPACE);
6482 goto done;
6484 pt = TAILQ_PREV(pt, tog_parent_trees, entry);
6486 if (te) {
6487 if (strlcat(*path, got_tree_entry_get_name(te), len) >= len) {
6488 err = got_error(GOT_ERR_NO_SPACE);
6489 goto done;
6492 done:
6493 if (err) {
6494 free(*path);
6495 *path = NULL;
6497 return err;
6500 static const struct got_error *
6501 blame_tree_entry(struct tog_view **new_view, int begin_y, int begin_x,
6502 struct got_tree_entry *te, struct tog_parent_trees *parents,
6503 struct got_object_id *commit_id, struct got_repository *repo)
6505 const struct got_error *err = NULL;
6506 char *path;
6507 struct tog_view *blame_view;
6509 *new_view = NULL;
6511 err = tree_entry_path(&path, parents, te);
6512 if (err)
6513 return err;
6515 blame_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_BLAME);
6516 if (blame_view == NULL) {
6517 err = got_error_from_errno("view_open");
6518 goto done;
6521 err = open_blame_view(blame_view, path, commit_id, repo);
6522 if (err) {
6523 if (err->code == GOT_ERR_CANCELLED)
6524 err = NULL;
6525 view_close(blame_view);
6526 } else
6527 *new_view = blame_view;
6528 done:
6529 free(path);
6530 return err;
6533 static const struct got_error *
6534 log_selected_tree_entry(struct tog_view **new_view, int begin_y, int begin_x,
6535 struct tog_tree_view_state *s)
6537 struct tog_view *log_view;
6538 const struct got_error *err = NULL;
6539 char *path;
6541 *new_view = NULL;
6543 log_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_LOG);
6544 if (log_view == NULL)
6545 return got_error_from_errno("view_open");
6547 err = tree_entry_path(&path, &s->parents, s->selected_entry);
6548 if (err)
6549 return err;
6551 err = open_log_view(log_view, s->commit_id, s->repo, s->head_ref_name,
6552 path, 0);
6553 if (err)
6554 view_close(log_view);
6555 else
6556 *new_view = log_view;
6557 free(path);
6558 return err;
6561 static const struct got_error *
6562 open_tree_view(struct tog_view *view, struct got_object_id *commit_id,
6563 const char *head_ref_name, struct got_repository *repo)
6565 const struct got_error *err = NULL;
6566 char *commit_id_str = NULL;
6567 struct tog_tree_view_state *s = &view->state.tree;
6568 struct got_commit_object *commit = NULL;
6570 TAILQ_INIT(&s->parents);
6571 STAILQ_INIT(&s->colors);
6573 s->commit_id = got_object_id_dup(commit_id);
6574 if (s->commit_id == NULL)
6575 return got_error_from_errno("got_object_id_dup");
6577 err = got_object_open_as_commit(&commit, repo, commit_id);
6578 if (err)
6579 goto done;
6582 * The root is opened here and will be closed when the view is closed.
6583 * Any visited subtrees and their path-wise parents are opened and
6584 * closed on demand.
6586 err = got_object_open_as_tree(&s->root, repo,
6587 got_object_commit_get_tree_id(commit));
6588 if (err)
6589 goto done;
6590 s->tree = s->root;
6592 err = got_object_id_str(&commit_id_str, commit_id);
6593 if (err != NULL)
6594 goto done;
6596 if (asprintf(&s->tree_label, "commit %s", commit_id_str) == -1) {
6597 err = got_error_from_errno("asprintf");
6598 goto done;
6601 s->first_displayed_entry = got_object_tree_get_entry(s->tree, 0);
6602 s->selected_entry = got_object_tree_get_entry(s->tree, 0);
6603 if (head_ref_name) {
6604 s->head_ref_name = strdup(head_ref_name);
6605 if (s->head_ref_name == NULL) {
6606 err = got_error_from_errno("strdup");
6607 goto done;
6610 s->repo = repo;
6612 if (has_colors() && getenv("TOG_COLORS") != NULL) {
6613 err = add_color(&s->colors, "\\$$",
6614 TOG_COLOR_TREE_SUBMODULE,
6615 get_color_value("TOG_COLOR_TREE_SUBMODULE"));
6616 if (err)
6617 goto done;
6618 err = add_color(&s->colors, "@$", TOG_COLOR_TREE_SYMLINK,
6619 get_color_value("TOG_COLOR_TREE_SYMLINK"));
6620 if (err)
6621 goto done;
6622 err = add_color(&s->colors, "/$",
6623 TOG_COLOR_TREE_DIRECTORY,
6624 get_color_value("TOG_COLOR_TREE_DIRECTORY"));
6625 if (err)
6626 goto done;
6628 err = add_color(&s->colors, "\\*$",
6629 TOG_COLOR_TREE_EXECUTABLE,
6630 get_color_value("TOG_COLOR_TREE_EXECUTABLE"));
6631 if (err)
6632 goto done;
6634 err = add_color(&s->colors, "^$", TOG_COLOR_COMMIT,
6635 get_color_value("TOG_COLOR_COMMIT"));
6636 if (err)
6637 goto done;
6640 view->show = show_tree_view;
6641 view->input = input_tree_view;
6642 view->close = close_tree_view;
6643 view->search_start = search_start_tree_view;
6644 view->search_next = search_next_tree_view;
6645 done:
6646 free(commit_id_str);
6647 if (commit)
6648 got_object_commit_close(commit);
6649 if (err)
6650 close_tree_view(view);
6651 return err;
6654 static const struct got_error *
6655 close_tree_view(struct tog_view *view)
6657 struct tog_tree_view_state *s = &view->state.tree;
6659 free_colors(&s->colors);
6660 free(s->tree_label);
6661 s->tree_label = NULL;
6662 free(s->commit_id);
6663 s->commit_id = NULL;
6664 free(s->head_ref_name);
6665 s->head_ref_name = NULL;
6666 while (!TAILQ_EMPTY(&s->parents)) {
6667 struct tog_parent_tree *parent;
6668 parent = TAILQ_FIRST(&s->parents);
6669 TAILQ_REMOVE(&s->parents, parent, entry);
6670 if (parent->tree != s->root)
6671 got_object_tree_close(parent->tree);
6672 free(parent);
6675 if (s->tree != NULL && s->tree != s->root)
6676 got_object_tree_close(s->tree);
6677 if (s->root)
6678 got_object_tree_close(s->root);
6679 return NULL;
6682 static const struct got_error *
6683 search_start_tree_view(struct tog_view *view)
6685 struct tog_tree_view_state *s = &view->state.tree;
6687 s->matched_entry = NULL;
6688 return NULL;
6691 static int
6692 match_tree_entry(struct got_tree_entry *te, regex_t *regex)
6694 regmatch_t regmatch;
6696 return regexec(regex, got_tree_entry_get_name(te), 1, &regmatch,
6697 0) == 0;
6700 static const struct got_error *
6701 search_next_tree_view(struct tog_view *view)
6703 struct tog_tree_view_state *s = &view->state.tree;
6704 struct got_tree_entry *te = NULL;
6706 if (!view->searching) {
6707 view->search_next_done = TOG_SEARCH_HAVE_MORE;
6708 return NULL;
6711 if (s->matched_entry) {
6712 if (view->searching == TOG_SEARCH_FORWARD) {
6713 if (s->selected_entry)
6714 te = got_tree_entry_get_next(s->tree,
6715 s->selected_entry);
6716 else
6717 te = got_object_tree_get_first_entry(s->tree);
6718 } else {
6719 if (s->selected_entry == NULL)
6720 te = got_object_tree_get_last_entry(s->tree);
6721 else
6722 te = got_tree_entry_get_prev(s->tree,
6723 s->selected_entry);
6725 } else {
6726 if (s->selected_entry)
6727 te = s->selected_entry;
6728 else if (view->searching == TOG_SEARCH_FORWARD)
6729 te = got_object_tree_get_first_entry(s->tree);
6730 else
6731 te = got_object_tree_get_last_entry(s->tree);
6734 while (1) {
6735 if (te == NULL) {
6736 if (s->matched_entry == NULL) {
6737 view->search_next_done = TOG_SEARCH_HAVE_MORE;
6738 return NULL;
6740 if (view->searching == TOG_SEARCH_FORWARD)
6741 te = got_object_tree_get_first_entry(s->tree);
6742 else
6743 te = got_object_tree_get_last_entry(s->tree);
6746 if (match_tree_entry(te, &view->regex)) {
6747 view->search_next_done = TOG_SEARCH_HAVE_MORE;
6748 s->matched_entry = te;
6749 break;
6752 if (view->searching == TOG_SEARCH_FORWARD)
6753 te = got_tree_entry_get_next(s->tree, te);
6754 else
6755 te = got_tree_entry_get_prev(s->tree, te);
6758 if (s->matched_entry) {
6759 s->first_displayed_entry = s->matched_entry;
6760 s->selected = 0;
6763 return NULL;
6766 static const struct got_error *
6767 show_tree_view(struct tog_view *view)
6769 const struct got_error *err = NULL;
6770 struct tog_tree_view_state *s = &view->state.tree;
6771 char *parent_path;
6773 err = tree_entry_path(&parent_path, &s->parents, NULL);
6774 if (err)
6775 return err;
6777 err = draw_tree_entries(view, parent_path);
6778 free(parent_path);
6780 view_border(view);
6781 return err;
6784 static const struct got_error *
6785 input_tree_view(struct tog_view **new_view, struct tog_view *view, int ch)
6787 const struct got_error *err = NULL;
6788 struct tog_tree_view_state *s = &view->state.tree;
6789 struct tog_view *log_view, *ref_view;
6790 struct got_tree_entry *te;
6791 int begin_y = 0, begin_x = 0, n, nscroll = view->nlines - 3;
6793 switch (ch) {
6794 case 'i':
6795 s->show_ids = !s->show_ids;
6796 view->count = 0;
6797 break;
6798 case 'l':
6799 view->count = 0;
6800 if (!s->selected_entry)
6801 break;
6802 if (view_is_parent_view(view))
6803 view_get_split(view, &begin_y, &begin_x);
6804 err = log_selected_tree_entry(&log_view, begin_y, begin_x, s);
6805 if (view_is_parent_view(view) &&
6806 view->mode == TOG_VIEW_SPLIT_HRZN) {
6807 err = view_init_hsplit(view, begin_y);
6808 if (err)
6809 break;
6811 view->focussed = 0;
6812 log_view->focussed = 1;
6813 log_view->mode = view->mode;
6814 log_view->nlines = view->lines - begin_y;
6815 if (view_is_parent_view(view)) {
6816 view_transfer_size(log_view, view);
6817 err = view_close_child(view);
6818 if (err)
6819 return err;
6820 err = view_set_child(view, log_view);
6821 if (err)
6822 return err;
6823 view->focus_child = 1;
6824 } else
6825 *new_view = log_view;
6826 break;
6827 case 'r':
6828 view->count = 0;
6829 if (view_is_parent_view(view))
6830 view_get_split(view, &begin_y, &begin_x);
6831 ref_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_REF);
6832 if (ref_view == NULL)
6833 return got_error_from_errno("view_open");
6834 err = open_ref_view(ref_view, s->repo);
6835 if (err) {
6836 view_close(ref_view);
6837 return err;
6839 if (view_is_parent_view(view) &&
6840 view->mode == TOG_VIEW_SPLIT_HRZN) {
6841 err = view_init_hsplit(view, begin_y);
6842 if (err)
6843 break;
6845 view->focussed = 0;
6846 ref_view->focussed = 1;
6847 ref_view->mode = view->mode;
6848 ref_view->nlines = view->lines - begin_y;
6849 if (view_is_parent_view(view)) {
6850 view_transfer_size(ref_view, view);
6851 err = view_close_child(view);
6852 if (err)
6853 return err;
6854 err = view_set_child(view, ref_view);
6855 if (err)
6856 return err;
6857 view->focus_child = 1;
6858 } else
6859 *new_view = ref_view;
6860 break;
6861 case 'g':
6862 case KEY_HOME:
6863 s->selected = 0;
6864 view->count = 0;
6865 if (s->tree == s->root)
6866 s->first_displayed_entry =
6867 got_object_tree_get_first_entry(s->tree);
6868 else
6869 s->first_displayed_entry = NULL;
6870 break;
6871 case 'G':
6872 case KEY_END: {
6873 int eos = view->nlines - 3;
6875 if (view->mode == TOG_VIEW_SPLIT_HRZN)
6876 --eos; /* border */
6877 s->selected = 0;
6878 view->count = 0;
6879 te = got_object_tree_get_last_entry(s->tree);
6880 for (n = 0; n < eos; n++) {
6881 if (te == NULL) {
6882 if (s->tree != s->root) {
6883 s->first_displayed_entry = NULL;
6884 n++;
6886 break;
6888 s->first_displayed_entry = te;
6889 te = got_tree_entry_get_prev(s->tree, te);
6891 if (n > 0)
6892 s->selected = n - 1;
6893 break;
6895 case 'k':
6896 case KEY_UP:
6897 case CTRL('p'):
6898 if (s->selected > 0) {
6899 s->selected--;
6900 break;
6902 tree_scroll_up(s, 1);
6903 if (s->selected_entry == NULL ||
6904 (s->tree == s->root && s->selected_entry ==
6905 got_object_tree_get_first_entry(s->tree)))
6906 view->count = 0;
6907 break;
6908 case CTRL('u'):
6909 case 'u':
6910 nscroll /= 2;
6911 /* FALL THROUGH */
6912 case KEY_PPAGE:
6913 case CTRL('b'):
6914 case 'b':
6915 if (s->tree == s->root) {
6916 if (got_object_tree_get_first_entry(s->tree) ==
6917 s->first_displayed_entry)
6918 s->selected -= MIN(s->selected, nscroll);
6919 } else {
6920 if (s->first_displayed_entry == NULL)
6921 s->selected -= MIN(s->selected, nscroll);
6923 tree_scroll_up(s, MAX(0, nscroll));
6924 if (s->selected_entry == NULL ||
6925 (s->tree == s->root && s->selected_entry ==
6926 got_object_tree_get_first_entry(s->tree)))
6927 view->count = 0;
6928 break;
6929 case 'j':
6930 case KEY_DOWN:
6931 case CTRL('n'):
6932 if (s->selected < s->ndisplayed - 1) {
6933 s->selected++;
6934 break;
6936 if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
6937 == NULL) {
6938 /* can't scroll any further */
6939 view->count = 0;
6940 break;
6942 tree_scroll_down(view, 1);
6943 break;
6944 case CTRL('d'):
6945 case 'd':
6946 nscroll /= 2;
6947 /* FALL THROUGH */
6948 case KEY_NPAGE:
6949 case CTRL('f'):
6950 case 'f':
6951 case ' ':
6952 if (got_tree_entry_get_next(s->tree, s->last_displayed_entry)
6953 == NULL) {
6954 /* can't scroll any further; move cursor down */
6955 if (s->selected < s->ndisplayed - 1)
6956 s->selected += MIN(nscroll,
6957 s->ndisplayed - s->selected - 1);
6958 else
6959 view->count = 0;
6960 break;
6962 tree_scroll_down(view, nscroll);
6963 break;
6964 case KEY_ENTER:
6965 case '\r':
6966 case KEY_BACKSPACE:
6967 if (s->selected_entry == NULL || ch == KEY_BACKSPACE) {
6968 struct tog_parent_tree *parent;
6969 /* user selected '..' */
6970 if (s->tree == s->root) {
6971 view->count = 0;
6972 break;
6974 parent = TAILQ_FIRST(&s->parents);
6975 TAILQ_REMOVE(&s->parents, parent,
6976 entry);
6977 got_object_tree_close(s->tree);
6978 s->tree = parent->tree;
6979 s->first_displayed_entry =
6980 parent->first_displayed_entry;
6981 s->selected_entry =
6982 parent->selected_entry;
6983 s->selected = parent->selected;
6984 if (s->selected > view->nlines - 3) {
6985 err = offset_selection_down(view);
6986 if (err)
6987 break;
6989 free(parent);
6990 } else if (S_ISDIR(got_tree_entry_get_mode(
6991 s->selected_entry))) {
6992 struct got_tree_object *subtree;
6993 view->count = 0;
6994 err = got_object_open_as_tree(&subtree, s->repo,
6995 got_tree_entry_get_id(s->selected_entry));
6996 if (err)
6997 break;
6998 err = tree_view_visit_subtree(s, subtree);
6999 if (err) {
7000 got_object_tree_close(subtree);
7001 break;
7003 } else if (S_ISREG(got_tree_entry_get_mode(
7004 s->selected_entry))) {
7005 struct tog_view *blame_view;
7006 int begin_x = 0, begin_y = 0;
7008 if (view_is_parent_view(view))
7009 view_get_split(view, &begin_y, &begin_x);
7011 err = blame_tree_entry(&blame_view, begin_y, begin_x,
7012 s->selected_entry, &s->parents,
7013 s->commit_id, s->repo);
7014 if (err)
7015 break;
7017 if (view_is_parent_view(view) &&
7018 view->mode == TOG_VIEW_SPLIT_HRZN) {
7019 err = view_init_hsplit(view, begin_y);
7020 if (err)
7021 break;
7024 view->count = 0;
7025 view->focussed = 0;
7026 blame_view->focussed = 1;
7027 blame_view->mode = view->mode;
7028 blame_view->nlines = view->lines - begin_y;
7029 if (view_is_parent_view(view)) {
7030 view_transfer_size(blame_view, view);
7031 err = view_close_child(view);
7032 if (err)
7033 return err;
7034 err = view_set_child(view, blame_view);
7035 if (err)
7036 return err;
7037 view->focus_child = 1;
7038 } else
7039 *new_view = blame_view;
7041 break;
7042 case KEY_RESIZE:
7043 if (view->nlines >= 4 && s->selected >= view->nlines - 3)
7044 s->selected = view->nlines - 4;
7045 view->count = 0;
7046 break;
7047 default:
7048 view->count = 0;
7049 break;
7052 return err;
7055 __dead static void
7056 usage_tree(void)
7058 endwin();
7059 fprintf(stderr,
7060 "usage: %s tree [-c commit] [-r repository-path] [path]\n",
7061 getprogname());
7062 exit(1);
7065 static const struct got_error *
7066 cmd_tree(int argc, char *argv[])
7068 const struct got_error *error;
7069 struct got_repository *repo = NULL;
7070 struct got_worktree *worktree = NULL;
7071 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
7072 struct got_object_id *commit_id = NULL;
7073 struct got_commit_object *commit = NULL;
7074 const char *commit_id_arg = NULL;
7075 char *label = NULL;
7076 struct got_reference *ref = NULL;
7077 const char *head_ref_name = NULL;
7078 int ch;
7079 struct tog_view *view;
7080 int *pack_fds = NULL;
7082 while ((ch = getopt(argc, argv, "c:r:")) != -1) {
7083 switch (ch) {
7084 case 'c':
7085 commit_id_arg = optarg;
7086 break;
7087 case 'r':
7088 repo_path = realpath(optarg, NULL);
7089 if (repo_path == NULL)
7090 return got_error_from_errno2("realpath",
7091 optarg);
7092 break;
7093 default:
7094 usage_tree();
7095 /* NOTREACHED */
7099 argc -= optind;
7100 argv += optind;
7102 if (argc > 1)
7103 usage_tree();
7105 error = got_repo_pack_fds_open(&pack_fds);
7106 if (error != NULL)
7107 goto done;
7109 if (repo_path == NULL) {
7110 cwd = getcwd(NULL, 0);
7111 if (cwd == NULL)
7112 return got_error_from_errno("getcwd");
7113 error = got_worktree_open(&worktree, cwd);
7114 if (error && error->code != GOT_ERR_NOT_WORKTREE)
7115 goto done;
7116 if (worktree)
7117 repo_path =
7118 strdup(got_worktree_get_repo_path(worktree));
7119 else
7120 repo_path = strdup(cwd);
7121 if (repo_path == NULL) {
7122 error = got_error_from_errno("strdup");
7123 goto done;
7127 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
7128 if (error != NULL)
7129 goto done;
7131 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
7132 repo, worktree);
7133 if (error)
7134 goto done;
7136 init_curses();
7138 error = apply_unveil(got_repo_get_path(repo), NULL);
7139 if (error)
7140 goto done;
7142 error = tog_load_refs(repo, 0);
7143 if (error)
7144 goto done;
7146 if (commit_id_arg == NULL) {
7147 error = got_repo_match_object_id(&commit_id, &label,
7148 worktree ? got_worktree_get_head_ref_name(worktree) :
7149 GOT_REF_HEAD, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
7150 if (error)
7151 goto done;
7152 head_ref_name = label;
7153 } else {
7154 error = got_ref_open(&ref, repo, commit_id_arg, 0);
7155 if (error == NULL)
7156 head_ref_name = got_ref_get_name(ref);
7157 else if (error->code != GOT_ERR_NOT_REF)
7158 goto done;
7159 error = got_repo_match_object_id(&commit_id, NULL,
7160 commit_id_arg, GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
7161 if (error)
7162 goto done;
7165 error = got_object_open_as_commit(&commit, repo, commit_id);
7166 if (error)
7167 goto done;
7169 view = view_open(0, 0, 0, 0, TOG_VIEW_TREE);
7170 if (view == NULL) {
7171 error = got_error_from_errno("view_open");
7172 goto done;
7174 error = open_tree_view(view, commit_id, head_ref_name, repo);
7175 if (error)
7176 goto done;
7177 if (!got_path_is_root_dir(in_repo_path)) {
7178 error = tree_view_walk_path(&view->state.tree, commit,
7179 in_repo_path);
7180 if (error)
7181 goto done;
7184 if (worktree) {
7185 /* Release work tree lock. */
7186 got_worktree_close(worktree);
7187 worktree = NULL;
7189 error = view_loop(view);
7190 done:
7191 free(repo_path);
7192 free(cwd);
7193 free(commit_id);
7194 free(label);
7195 if (ref)
7196 got_ref_close(ref);
7197 if (repo) {
7198 const struct got_error *close_err = got_repo_close(repo);
7199 if (error == NULL)
7200 error = close_err;
7202 if (pack_fds) {
7203 const struct got_error *pack_err =
7204 got_repo_pack_fds_close(pack_fds);
7205 if (error == NULL)
7206 error = pack_err;
7208 tog_free_refs();
7209 return error;
7212 static const struct got_error *
7213 ref_view_load_refs(struct tog_ref_view_state *s)
7215 struct got_reflist_entry *sre;
7216 struct tog_reflist_entry *re;
7218 s->nrefs = 0;
7219 TAILQ_FOREACH(sre, &tog_refs, entry) {
7220 if (strncmp(got_ref_get_name(sre->ref),
7221 "refs/got/", 9) == 0 &&
7222 strncmp(got_ref_get_name(sre->ref),
7223 "refs/got/backup/", 16) != 0)
7224 continue;
7226 re = malloc(sizeof(*re));
7227 if (re == NULL)
7228 return got_error_from_errno("malloc");
7230 re->ref = got_ref_dup(sre->ref);
7231 if (re->ref == NULL)
7232 return got_error_from_errno("got_ref_dup");
7233 re->idx = s->nrefs++;
7234 TAILQ_INSERT_TAIL(&s->refs, re, entry);
7237 s->first_displayed_entry = TAILQ_FIRST(&s->refs);
7238 return NULL;
7241 static void
7242 ref_view_free_refs(struct tog_ref_view_state *s)
7244 struct tog_reflist_entry *re;
7246 while (!TAILQ_EMPTY(&s->refs)) {
7247 re = TAILQ_FIRST(&s->refs);
7248 TAILQ_REMOVE(&s->refs, re, entry);
7249 got_ref_close(re->ref);
7250 free(re);
7254 static const struct got_error *
7255 open_ref_view(struct tog_view *view, struct got_repository *repo)
7257 const struct got_error *err = NULL;
7258 struct tog_ref_view_state *s = &view->state.ref;
7260 s->selected_entry = 0;
7261 s->repo = repo;
7263 TAILQ_INIT(&s->refs);
7264 STAILQ_INIT(&s->colors);
7266 err = ref_view_load_refs(s);
7267 if (err)
7268 return err;
7270 if (has_colors() && getenv("TOG_COLORS") != NULL) {
7271 err = add_color(&s->colors, "^refs/heads/",
7272 TOG_COLOR_REFS_HEADS,
7273 get_color_value("TOG_COLOR_REFS_HEADS"));
7274 if (err)
7275 goto done;
7277 err = add_color(&s->colors, "^refs/tags/",
7278 TOG_COLOR_REFS_TAGS,
7279 get_color_value("TOG_COLOR_REFS_TAGS"));
7280 if (err)
7281 goto done;
7283 err = add_color(&s->colors, "^refs/remotes/",
7284 TOG_COLOR_REFS_REMOTES,
7285 get_color_value("TOG_COLOR_REFS_REMOTES"));
7286 if (err)
7287 goto done;
7289 err = add_color(&s->colors, "^refs/got/backup/",
7290 TOG_COLOR_REFS_BACKUP,
7291 get_color_value("TOG_COLOR_REFS_BACKUP"));
7292 if (err)
7293 goto done;
7296 view->show = show_ref_view;
7297 view->input = input_ref_view;
7298 view->close = close_ref_view;
7299 view->search_start = search_start_ref_view;
7300 view->search_next = search_next_ref_view;
7301 done:
7302 if (err)
7303 free_colors(&s->colors);
7304 return err;
7307 static const struct got_error *
7308 close_ref_view(struct tog_view *view)
7310 struct tog_ref_view_state *s = &view->state.ref;
7312 ref_view_free_refs(s);
7313 free_colors(&s->colors);
7315 return NULL;
7318 static const struct got_error *
7319 resolve_reflist_entry(struct got_object_id **commit_id,
7320 struct tog_reflist_entry *re, struct got_repository *repo)
7322 const struct got_error *err = NULL;
7323 struct got_object_id *obj_id;
7324 struct got_tag_object *tag = NULL;
7325 int obj_type;
7327 *commit_id = NULL;
7329 err = got_ref_resolve(&obj_id, repo, re->ref);
7330 if (err)
7331 return err;
7333 err = got_object_get_type(&obj_type, repo, obj_id);
7334 if (err)
7335 goto done;
7337 switch (obj_type) {
7338 case GOT_OBJ_TYPE_COMMIT:
7339 *commit_id = obj_id;
7340 break;
7341 case GOT_OBJ_TYPE_TAG:
7342 err = got_object_open_as_tag(&tag, repo, obj_id);
7343 if (err)
7344 goto done;
7345 free(obj_id);
7346 err = got_object_get_type(&obj_type, repo,
7347 got_object_tag_get_object_id(tag));
7348 if (err)
7349 goto done;
7350 if (obj_type != GOT_OBJ_TYPE_COMMIT) {
7351 err = got_error(GOT_ERR_OBJ_TYPE);
7352 goto done;
7354 *commit_id = got_object_id_dup(
7355 got_object_tag_get_object_id(tag));
7356 if (*commit_id == NULL) {
7357 err = got_error_from_errno("got_object_id_dup");
7358 goto done;
7360 break;
7361 default:
7362 err = got_error(GOT_ERR_OBJ_TYPE);
7363 break;
7366 done:
7367 if (tag)
7368 got_object_tag_close(tag);
7369 if (err) {
7370 free(*commit_id);
7371 *commit_id = NULL;
7373 return err;
7376 static const struct got_error *
7377 log_ref_entry(struct tog_view **new_view, int begin_y, int begin_x,
7378 struct tog_reflist_entry *re, struct got_repository *repo)
7380 struct tog_view *log_view;
7381 const struct got_error *err = NULL;
7382 struct got_object_id *commit_id = NULL;
7384 *new_view = NULL;
7386 err = resolve_reflist_entry(&commit_id, re, repo);
7387 if (err) {
7388 if (err->code != GOT_ERR_OBJ_TYPE)
7389 return err;
7390 else
7391 return NULL;
7394 log_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_LOG);
7395 if (log_view == NULL) {
7396 err = got_error_from_errno("view_open");
7397 goto done;
7400 err = open_log_view(log_view, commit_id, repo,
7401 got_ref_get_name(re->ref), "", 0);
7402 done:
7403 if (err)
7404 view_close(log_view);
7405 else
7406 *new_view = log_view;
7407 free(commit_id);
7408 return err;
7411 static void
7412 ref_scroll_up(struct tog_ref_view_state *s, int maxscroll)
7414 struct tog_reflist_entry *re;
7415 int i = 0;
7417 if (s->first_displayed_entry == TAILQ_FIRST(&s->refs))
7418 return;
7420 re = TAILQ_PREV(s->first_displayed_entry, tog_reflist_head, entry);
7421 while (i++ < maxscroll) {
7422 if (re == NULL)
7423 break;
7424 s->first_displayed_entry = re;
7425 re = TAILQ_PREV(re, tog_reflist_head, entry);
7429 static const struct got_error *
7430 ref_scroll_down(struct tog_view *view, int maxscroll)
7432 struct tog_ref_view_state *s = &view->state.ref;
7433 struct tog_reflist_entry *next, *last;
7434 int n = 0;
7436 if (s->first_displayed_entry)
7437 next = TAILQ_NEXT(s->first_displayed_entry, entry);
7438 else
7439 next = TAILQ_FIRST(&s->refs);
7441 last = s->last_displayed_entry;
7442 while (next && n++ < maxscroll) {
7443 if (last)
7444 last = TAILQ_NEXT(last, entry);
7445 if (last || (view->mode == TOG_VIEW_SPLIT_HRZN)) {
7446 s->first_displayed_entry = next;
7447 next = TAILQ_NEXT(next, entry);
7451 return NULL;
7454 static const struct got_error *
7455 search_start_ref_view(struct tog_view *view)
7457 struct tog_ref_view_state *s = &view->state.ref;
7459 s->matched_entry = NULL;
7460 return NULL;
7463 static int
7464 match_reflist_entry(struct tog_reflist_entry *re, regex_t *regex)
7466 regmatch_t regmatch;
7468 return regexec(regex, got_ref_get_name(re->ref), 1, &regmatch,
7469 0) == 0;
7472 static const struct got_error *
7473 search_next_ref_view(struct tog_view *view)
7475 struct tog_ref_view_state *s = &view->state.ref;
7476 struct tog_reflist_entry *re = NULL;
7478 if (!view->searching) {
7479 view->search_next_done = TOG_SEARCH_HAVE_MORE;
7480 return NULL;
7483 if (s->matched_entry) {
7484 if (view->searching == TOG_SEARCH_FORWARD) {
7485 if (s->selected_entry)
7486 re = TAILQ_NEXT(s->selected_entry, entry);
7487 else
7488 re = TAILQ_PREV(s->selected_entry,
7489 tog_reflist_head, entry);
7490 } else {
7491 if (s->selected_entry == NULL)
7492 re = TAILQ_LAST(&s->refs, tog_reflist_head);
7493 else
7494 re = TAILQ_PREV(s->selected_entry,
7495 tog_reflist_head, entry);
7497 } else {
7498 if (s->selected_entry)
7499 re = s->selected_entry;
7500 else if (view->searching == TOG_SEARCH_FORWARD)
7501 re = TAILQ_FIRST(&s->refs);
7502 else
7503 re = TAILQ_LAST(&s->refs, tog_reflist_head);
7506 while (1) {
7507 if (re == NULL) {
7508 if (s->matched_entry == NULL) {
7509 view->search_next_done = TOG_SEARCH_HAVE_MORE;
7510 return NULL;
7512 if (view->searching == TOG_SEARCH_FORWARD)
7513 re = TAILQ_FIRST(&s->refs);
7514 else
7515 re = TAILQ_LAST(&s->refs, tog_reflist_head);
7518 if (match_reflist_entry(re, &view->regex)) {
7519 view->search_next_done = TOG_SEARCH_HAVE_MORE;
7520 s->matched_entry = re;
7521 break;
7524 if (view->searching == TOG_SEARCH_FORWARD)
7525 re = TAILQ_NEXT(re, entry);
7526 else
7527 re = TAILQ_PREV(re, tog_reflist_head, entry);
7530 if (s->matched_entry) {
7531 s->first_displayed_entry = s->matched_entry;
7532 s->selected = 0;
7535 return NULL;
7538 static const struct got_error *
7539 show_ref_view(struct tog_view *view)
7541 const struct got_error *err = NULL;
7542 struct tog_ref_view_state *s = &view->state.ref;
7543 struct tog_reflist_entry *re;
7544 char *line = NULL;
7545 wchar_t *wline;
7546 struct tog_color *tc;
7547 int width, n;
7548 int limit = view->nlines;
7550 werase(view->window);
7552 s->ndisplayed = 0;
7553 if (view_is_hsplit_top(view))
7554 --limit; /* border */
7556 if (limit == 0)
7557 return NULL;
7559 re = s->first_displayed_entry;
7561 if (asprintf(&line, "references [%d/%d]", re->idx + s->selected + 1,
7562 s->nrefs) == -1)
7563 return got_error_from_errno("asprintf");
7565 err = format_line(&wline, &width, NULL, line, 0, view->ncols, 0, 0);
7566 if (err) {
7567 free(line);
7568 return err;
7570 if (view_needs_focus_indication(view))
7571 wstandout(view->window);
7572 waddwstr(view->window, wline);
7573 if (view_needs_focus_indication(view))
7574 wstandend(view->window);
7575 free(wline);
7576 wline = NULL;
7577 free(line);
7578 line = NULL;
7579 if (width < view->ncols - 1)
7580 waddch(view->window, '\n');
7581 if (--limit <= 0)
7582 return NULL;
7584 n = 0;
7585 while (re && limit > 0) {
7586 char *line = NULL;
7587 char ymd[13]; /* YYYY-MM-DD + " " + NUL */
7589 if (s->show_date) {
7590 struct got_commit_object *ci;
7591 struct got_tag_object *tag;
7592 struct got_object_id *id;
7593 struct tm tm;
7594 time_t t;
7596 err = got_ref_resolve(&id, s->repo, re->ref);
7597 if (err)
7598 return err;
7599 err = got_object_open_as_tag(&tag, s->repo, id);
7600 if (err) {
7601 if (err->code != GOT_ERR_OBJ_TYPE) {
7602 free(id);
7603 return err;
7605 err = got_object_open_as_commit(&ci, s->repo,
7606 id);
7607 if (err) {
7608 free(id);
7609 return err;
7611 t = got_object_commit_get_committer_time(ci);
7612 got_object_commit_close(ci);
7613 } else {
7614 t = got_object_tag_get_tagger_time(tag);
7615 got_object_tag_close(tag);
7617 free(id);
7618 if (gmtime_r(&t, &tm) == NULL)
7619 return got_error_from_errno("gmtime_r");
7620 if (strftime(ymd, sizeof(ymd), "%G-%m-%d ", &tm) == 0)
7621 return got_error(GOT_ERR_NO_SPACE);
7623 if (got_ref_is_symbolic(re->ref)) {
7624 if (asprintf(&line, "%s%s -> %s", s->show_date ?
7625 ymd : "", got_ref_get_name(re->ref),
7626 got_ref_get_symref_target(re->ref)) == -1)
7627 return got_error_from_errno("asprintf");
7628 } else if (s->show_ids) {
7629 struct got_object_id *id;
7630 char *id_str;
7631 err = got_ref_resolve(&id, s->repo, re->ref);
7632 if (err)
7633 return err;
7634 err = got_object_id_str(&id_str, id);
7635 if (err) {
7636 free(id);
7637 return err;
7639 if (asprintf(&line, "%s%s: %s", s->show_date ? ymd : "",
7640 got_ref_get_name(re->ref), id_str) == -1) {
7641 err = got_error_from_errno("asprintf");
7642 free(id);
7643 free(id_str);
7644 return err;
7646 free(id);
7647 free(id_str);
7648 } else if (asprintf(&line, "%s%s", s->show_date ? ymd : "",
7649 got_ref_get_name(re->ref)) == -1)
7650 return got_error_from_errno("asprintf");
7652 err = format_line(&wline, &width, NULL, line, 0, view->ncols,
7653 0, 0);
7654 if (err) {
7655 free(line);
7656 return err;
7658 if (n == s->selected) {
7659 if (view->focussed)
7660 wstandout(view->window);
7661 s->selected_entry = re;
7663 tc = match_color(&s->colors, got_ref_get_name(re->ref));
7664 if (tc)
7665 wattr_on(view->window,
7666 COLOR_PAIR(tc->colorpair), NULL);
7667 waddwstr(view->window, wline);
7668 if (tc)
7669 wattr_off(view->window,
7670 COLOR_PAIR(tc->colorpair), NULL);
7671 if (width < view->ncols - 1)
7672 waddch(view->window, '\n');
7673 if (n == s->selected && view->focussed)
7674 wstandend(view->window);
7675 free(line);
7676 free(wline);
7677 wline = NULL;
7678 n++;
7679 s->ndisplayed++;
7680 s->last_displayed_entry = re;
7682 limit--;
7683 re = TAILQ_NEXT(re, entry);
7686 view_border(view);
7687 return err;
7690 static const struct got_error *
7691 browse_ref_tree(struct tog_view **new_view, int begin_y, int begin_x,
7692 struct tog_reflist_entry *re, struct got_repository *repo)
7694 const struct got_error *err = NULL;
7695 struct got_object_id *commit_id = NULL;
7696 struct tog_view *tree_view;
7698 *new_view = NULL;
7700 err = resolve_reflist_entry(&commit_id, re, repo);
7701 if (err) {
7702 if (err->code != GOT_ERR_OBJ_TYPE)
7703 return err;
7704 else
7705 return NULL;
7709 tree_view = view_open(0, 0, begin_y, begin_x, TOG_VIEW_TREE);
7710 if (tree_view == NULL) {
7711 err = got_error_from_errno("view_open");
7712 goto done;
7715 err = open_tree_view(tree_view, commit_id,
7716 got_ref_get_name(re->ref), repo);
7717 if (err)
7718 goto done;
7720 *new_view = tree_view;
7721 done:
7722 free(commit_id);
7723 return err;
7725 static const struct got_error *
7726 input_ref_view(struct tog_view **new_view, struct tog_view *view, int ch)
7728 const struct got_error *err = NULL;
7729 struct tog_ref_view_state *s = &view->state.ref;
7730 struct tog_view *log_view, *tree_view;
7731 struct tog_reflist_entry *re;
7732 int begin_y = 0, begin_x = 0, n, nscroll = view->nlines - 1;
7734 switch (ch) {
7735 case 'i':
7736 s->show_ids = !s->show_ids;
7737 view->count = 0;
7738 break;
7739 case 'm':
7740 s->show_date = !s->show_date;
7741 view->count = 0;
7742 break;
7743 case 'o':
7744 s->sort_by_date = !s->sort_by_date;
7745 view->count = 0;
7746 err = got_reflist_sort(&tog_refs, s->sort_by_date ?
7747 got_ref_cmp_by_commit_timestamp_descending :
7748 tog_ref_cmp_by_name, s->repo);
7749 if (err)
7750 break;
7751 got_reflist_object_id_map_free(tog_refs_idmap);
7752 err = got_reflist_object_id_map_create(&tog_refs_idmap,
7753 &tog_refs, s->repo);
7754 if (err)
7755 break;
7756 ref_view_free_refs(s);
7757 err = ref_view_load_refs(s);
7758 break;
7759 case KEY_ENTER:
7760 case '\r':
7761 view->count = 0;
7762 if (!s->selected_entry)
7763 break;
7764 if (view_is_parent_view(view))
7765 view_get_split(view, &begin_y, &begin_x);
7767 err = log_ref_entry(&log_view, begin_y, begin_x,
7768 s->selected_entry, s->repo);
7769 if (err)
7770 break;
7772 if (view_is_parent_view(view) &&
7773 view->mode == TOG_VIEW_SPLIT_HRZN) {
7774 err = view_init_hsplit(view, begin_y);
7775 if (err)
7776 break;
7779 view->focussed = 0;
7780 log_view->focussed = 1;
7781 log_view->mode = view->mode;
7782 log_view->nlines = view->lines - begin_y;
7783 if (view_is_parent_view(view)) {
7784 view_transfer_size(log_view, view);
7785 err = view_close_child(view);
7786 if (err)
7787 return err;
7788 err = view_set_child(view, log_view);
7789 if (err)
7790 return err;
7791 view->focus_child = 1;
7792 } else
7793 *new_view = log_view;
7794 break;
7795 case 't':
7796 view->count = 0;
7797 if (!s->selected_entry)
7798 break;
7799 if (view_is_parent_view(view))
7800 view_get_split(view, &begin_y, &begin_x);
7801 err = browse_ref_tree(&tree_view, begin_y, begin_x,
7802 s->selected_entry, s->repo);
7803 if (err || tree_view == NULL)
7804 break;
7805 if (view_is_parent_view(view) &&
7806 view->mode == TOG_VIEW_SPLIT_HRZN) {
7807 err = view_init_hsplit(view, begin_y);
7808 if (err)
7809 break;
7811 view->focussed = 0;
7812 tree_view->focussed = 1;
7813 tree_view->mode = view->mode;
7814 tree_view->nlines = view->lines - begin_y;
7815 if (view_is_parent_view(view)) {
7816 view_transfer_size(tree_view, view);
7817 err = view_close_child(view);
7818 if (err)
7819 return err;
7820 err = view_set_child(view, tree_view);
7821 if (err)
7822 return err;
7823 view->focus_child = 1;
7824 } else
7825 *new_view = tree_view;
7826 break;
7827 case 'g':
7828 case KEY_HOME:
7829 s->selected = 0;
7830 view->count = 0;
7831 s->first_displayed_entry = TAILQ_FIRST(&s->refs);
7832 break;
7833 case 'G':
7834 case KEY_END: {
7835 int eos = view->nlines - 1;
7837 if (view->mode == TOG_VIEW_SPLIT_HRZN)
7838 --eos; /* border */
7839 s->selected = 0;
7840 view->count = 0;
7841 re = TAILQ_LAST(&s->refs, tog_reflist_head);
7842 for (n = 0; n < eos; n++) {
7843 if (re == NULL)
7844 break;
7845 s->first_displayed_entry = re;
7846 re = TAILQ_PREV(re, tog_reflist_head, entry);
7848 if (n > 0)
7849 s->selected = n - 1;
7850 break;
7852 case 'k':
7853 case KEY_UP:
7854 case CTRL('p'):
7855 if (s->selected > 0) {
7856 s->selected--;
7857 break;
7859 ref_scroll_up(s, 1);
7860 if (s->selected_entry == TAILQ_FIRST(&s->refs))
7861 view->count = 0;
7862 break;
7863 case CTRL('u'):
7864 case 'u':
7865 nscroll /= 2;
7866 /* FALL THROUGH */
7867 case KEY_PPAGE:
7868 case CTRL('b'):
7869 case 'b':
7870 if (s->first_displayed_entry == TAILQ_FIRST(&s->refs))
7871 s->selected -= MIN(nscroll, s->selected);
7872 ref_scroll_up(s, MAX(0, nscroll));
7873 if (s->selected_entry == TAILQ_FIRST(&s->refs))
7874 view->count = 0;
7875 break;
7876 case 'j':
7877 case KEY_DOWN:
7878 case CTRL('n'):
7879 if (s->selected < s->ndisplayed - 1) {
7880 s->selected++;
7881 break;
7883 if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL) {
7884 /* can't scroll any further */
7885 view->count = 0;
7886 break;
7888 ref_scroll_down(view, 1);
7889 break;
7890 case CTRL('d'):
7891 case 'd':
7892 nscroll /= 2;
7893 /* FALL THROUGH */
7894 case KEY_NPAGE:
7895 case CTRL('f'):
7896 case 'f':
7897 case ' ':
7898 if (TAILQ_NEXT(s->last_displayed_entry, entry) == NULL) {
7899 /* can't scroll any further; move cursor down */
7900 if (s->selected < s->ndisplayed - 1)
7901 s->selected += MIN(nscroll,
7902 s->ndisplayed - s->selected - 1);
7903 if (view->count > 1 && s->selected < s->ndisplayed - 1)
7904 s->selected += s->ndisplayed - s->selected - 1;
7905 view->count = 0;
7906 break;
7908 ref_scroll_down(view, nscroll);
7909 break;
7910 case CTRL('l'):
7911 view->count = 0;
7912 tog_free_refs();
7913 err = tog_load_refs(s->repo, s->sort_by_date);
7914 if (err)
7915 break;
7916 ref_view_free_refs(s);
7917 err = ref_view_load_refs(s);
7918 break;
7919 case KEY_RESIZE:
7920 if (view->nlines >= 2 && s->selected >= view->nlines - 1)
7921 s->selected = view->nlines - 2;
7922 break;
7923 default:
7924 view->count = 0;
7925 break;
7928 return err;
7931 __dead static void
7932 usage_ref(void)
7934 endwin();
7935 fprintf(stderr, "usage: %s ref [-r repository-path]\n",
7936 getprogname());
7937 exit(1);
7940 static const struct got_error *
7941 cmd_ref(int argc, char *argv[])
7943 const struct got_error *error;
7944 struct got_repository *repo = NULL;
7945 struct got_worktree *worktree = NULL;
7946 char *cwd = NULL, *repo_path = NULL;
7947 int ch;
7948 struct tog_view *view;
7949 int *pack_fds = NULL;
7951 while ((ch = getopt(argc, argv, "r:")) != -1) {
7952 switch (ch) {
7953 case 'r':
7954 repo_path = realpath(optarg, NULL);
7955 if (repo_path == NULL)
7956 return got_error_from_errno2("realpath",
7957 optarg);
7958 break;
7959 default:
7960 usage_ref();
7961 /* NOTREACHED */
7965 argc -= optind;
7966 argv += optind;
7968 if (argc > 1)
7969 usage_ref();
7971 error = got_repo_pack_fds_open(&pack_fds);
7972 if (error != NULL)
7973 goto done;
7975 if (repo_path == NULL) {
7976 cwd = getcwd(NULL, 0);
7977 if (cwd == NULL)
7978 return got_error_from_errno("getcwd");
7979 error = got_worktree_open(&worktree, cwd);
7980 if (error && error->code != GOT_ERR_NOT_WORKTREE)
7981 goto done;
7982 if (worktree)
7983 repo_path =
7984 strdup(got_worktree_get_repo_path(worktree));
7985 else
7986 repo_path = strdup(cwd);
7987 if (repo_path == NULL) {
7988 error = got_error_from_errno("strdup");
7989 goto done;
7993 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
7994 if (error != NULL)
7995 goto done;
7997 init_curses();
7999 error = apply_unveil(got_repo_get_path(repo), NULL);
8000 if (error)
8001 goto done;
8003 error = tog_load_refs(repo, 0);
8004 if (error)
8005 goto done;
8007 view = view_open(0, 0, 0, 0, TOG_VIEW_REF);
8008 if (view == NULL) {
8009 error = got_error_from_errno("view_open");
8010 goto done;
8013 error = open_ref_view(view, repo);
8014 if (error)
8015 goto done;
8017 if (worktree) {
8018 /* Release work tree lock. */
8019 got_worktree_close(worktree);
8020 worktree = NULL;
8022 error = view_loop(view);
8023 done:
8024 free(repo_path);
8025 free(cwd);
8026 if (repo) {
8027 const struct got_error *close_err = got_repo_close(repo);
8028 if (close_err)
8029 error = close_err;
8031 if (pack_fds) {
8032 const struct got_error *pack_err =
8033 got_repo_pack_fds_close(pack_fds);
8034 if (error == NULL)
8035 error = pack_err;
8037 tog_free_refs();
8038 return error;
8042 * If view was scrolled down to move the selected line into view when opening a
8043 * horizontal split, scroll back up when closing the split/toggling fullscreen.
8045 static void
8046 offset_selection_up(struct tog_view *view)
8048 switch (view->type) {
8049 case TOG_VIEW_BLAME: {
8050 struct tog_blame_view_state *s = &view->state.blame;
8051 if (s->first_displayed_line == 1) {
8052 s->selected_line = MAX(s->selected_line - view->offset,
8053 1);
8054 break;
8056 if (s->first_displayed_line > view->offset)
8057 s->first_displayed_line -= view->offset;
8058 else
8059 s->first_displayed_line = 1;
8060 s->selected_line += view->offset;
8061 break;
8063 case TOG_VIEW_LOG:
8064 log_scroll_up(&view->state.log, view->offset);
8065 view->state.log.selected += view->offset;
8066 break;
8067 case TOG_VIEW_REF:
8068 ref_scroll_up(&view->state.ref, view->offset);
8069 view->state.ref.selected += view->offset;
8070 break;
8071 case TOG_VIEW_TREE:
8072 tree_scroll_up(&view->state.tree, view->offset);
8073 view->state.tree.selected += view->offset;
8074 break;
8075 default:
8076 break;
8079 view->offset = 0;
8083 * If the selected line is in the section of screen covered by the bottom split,
8084 * scroll down offset lines to move it into view and index its new position.
8086 static const struct got_error *
8087 offset_selection_down(struct tog_view *view)
8089 const struct got_error *err = NULL;
8090 const struct got_error *(*scrolld)(struct tog_view *, int);
8091 int *selected = NULL;
8092 int header, offset;
8094 switch (view->type) {
8095 case TOG_VIEW_BLAME: {
8096 struct tog_blame_view_state *s = &view->state.blame;
8097 header = 3;
8098 scrolld = NULL;
8099 if (s->selected_line > view->nlines - header) {
8100 offset = abs(view->nlines - s->selected_line - header);
8101 s->first_displayed_line += offset;
8102 s->selected_line -= offset;
8103 view->offset = offset;
8105 break;
8107 case TOG_VIEW_LOG: {
8108 struct tog_log_view_state *s = &view->state.log;
8109 scrolld = &log_scroll_down;
8110 header = view_is_parent_view(view) ? 3 : 2;
8111 selected = &s->selected;
8112 break;
8114 case TOG_VIEW_REF: {
8115 struct tog_ref_view_state *s = &view->state.ref;
8116 scrolld = &ref_scroll_down;
8117 header = 3;
8118 selected = &s->selected;
8119 break;
8121 case TOG_VIEW_TREE: {
8122 struct tog_tree_view_state *s = &view->state.tree;
8123 scrolld = &tree_scroll_down;
8124 header = 5;
8125 selected = &s->selected;
8126 break;
8128 default:
8129 selected = NULL;
8130 scrolld = NULL;
8131 header = 0;
8132 break;
8135 if (selected && *selected > view->nlines - header) {
8136 offset = abs(view->nlines - *selected - header);
8137 view->offset = offset;
8138 if (scrolld && offset) {
8139 err = scrolld(view, offset);
8140 *selected -= offset;
8144 return err;
8147 static void
8148 list_commands(FILE *fp)
8150 size_t i;
8152 fprintf(fp, "commands:");
8153 for (i = 0; i < nitems(tog_commands); i++) {
8154 const struct tog_cmd *cmd = &tog_commands[i];
8155 fprintf(fp, " %s", cmd->name);
8157 fputc('\n', fp);
8160 __dead static void
8161 usage(int hflag, int status)
8163 FILE *fp = (status == 0) ? stdout : stderr;
8165 fprintf(fp, "usage: %s [-h] [-V | --version] [command] [arg ...]\n",
8166 getprogname());
8167 if (hflag) {
8168 fprintf(fp, "lazy usage: %s path\n", getprogname());
8169 list_commands(fp);
8171 exit(status);
8174 static char **
8175 make_argv(int argc, ...)
8177 va_list ap;
8178 char **argv;
8179 int i;
8181 va_start(ap, argc);
8183 argv = calloc(argc, sizeof(char *));
8184 if (argv == NULL)
8185 err(1, "calloc");
8186 for (i = 0; i < argc; i++) {
8187 argv[i] = strdup(va_arg(ap, char *));
8188 if (argv[i] == NULL)
8189 err(1, "strdup");
8192 va_end(ap);
8193 return argv;
8197 * Try to convert 'tog path' into a 'tog log path' command.
8198 * The user could simply have mistyped the command rather than knowingly
8199 * provided a path. So check whether argv[0] can in fact be resolved
8200 * to a path in the HEAD commit and print a special error if not.
8201 * This hack is for mpi@ <3
8203 static const struct got_error *
8204 tog_log_with_path(int argc, char *argv[])
8206 const struct got_error *error = NULL, *close_err;
8207 const struct tog_cmd *cmd = NULL;
8208 struct got_repository *repo = NULL;
8209 struct got_worktree *worktree = NULL;
8210 struct got_object_id *commit_id = NULL, *id = NULL;
8211 struct got_commit_object *commit = NULL;
8212 char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
8213 char *commit_id_str = NULL, **cmd_argv = NULL;
8214 int *pack_fds = NULL;
8216 cwd = getcwd(NULL, 0);
8217 if (cwd == NULL)
8218 return got_error_from_errno("getcwd");
8220 error = got_repo_pack_fds_open(&pack_fds);
8221 if (error != NULL)
8222 goto done;
8224 error = got_worktree_open(&worktree, cwd);
8225 if (error && error->code != GOT_ERR_NOT_WORKTREE)
8226 goto done;
8228 if (worktree)
8229 repo_path = strdup(got_worktree_get_repo_path(worktree));
8230 else
8231 repo_path = strdup(cwd);
8232 if (repo_path == NULL) {
8233 error = got_error_from_errno("strdup");
8234 goto done;
8237 error = got_repo_open(&repo, repo_path, NULL, pack_fds);
8238 if (error != NULL)
8239 goto done;
8241 error = get_in_repo_path_from_argv0(&in_repo_path, argc, argv,
8242 repo, worktree);
8243 if (error)
8244 goto done;
8246 error = tog_load_refs(repo, 0);
8247 if (error)
8248 goto done;
8249 error = got_repo_match_object_id(&commit_id, NULL, worktree ?
8250 got_worktree_get_head_ref_name(worktree) : GOT_REF_HEAD,
8251 GOT_OBJ_TYPE_COMMIT, &tog_refs, repo);
8252 if (error)
8253 goto done;
8255 if (worktree) {
8256 got_worktree_close(worktree);
8257 worktree = NULL;
8260 error = got_object_open_as_commit(&commit, repo, commit_id);
8261 if (error)
8262 goto done;
8264 error = got_object_id_by_path(&id, repo, commit, in_repo_path);
8265 if (error) {
8266 if (error->code != GOT_ERR_NO_TREE_ENTRY)
8267 goto done;
8268 fprintf(stderr, "%s: '%s' is no known command or path\n",
8269 getprogname(), argv[0]);
8270 usage(1, 1);
8271 /* not reached */
8274 close_err = got_repo_close(repo);
8275 if (error == NULL)
8276 error = close_err;
8277 repo = NULL;
8279 error = got_object_id_str(&commit_id_str, commit_id);
8280 if (error)
8281 goto done;
8283 cmd = &tog_commands[0]; /* log */
8284 argc = 4;
8285 cmd_argv = make_argv(argc, cmd->name, "-c", commit_id_str, argv[0]);
8286 error = cmd->cmd_main(argc, cmd_argv);
8287 done:
8288 if (repo) {
8289 close_err = got_repo_close(repo);
8290 if (error == NULL)
8291 error = close_err;
8293 if (commit)
8294 got_object_commit_close(commit);
8295 if (worktree)
8296 got_worktree_close(worktree);
8297 if (pack_fds) {
8298 const struct got_error *pack_err =
8299 got_repo_pack_fds_close(pack_fds);
8300 if (error == NULL)
8301 error = pack_err;
8303 free(id);
8304 free(commit_id_str);
8305 free(commit_id);
8306 free(cwd);
8307 free(repo_path);
8308 free(in_repo_path);
8309 if (cmd_argv) {
8310 int i;
8311 for (i = 0; i < argc; i++)
8312 free(cmd_argv[i]);
8313 free(cmd_argv);
8315 tog_free_refs();
8316 return error;
8319 int
8320 main(int argc, char *argv[])
8322 const struct got_error *error = NULL;
8323 const struct tog_cmd *cmd = NULL;
8324 int ch, hflag = 0, Vflag = 0;
8325 char **cmd_argv = NULL;
8326 static const struct option longopts[] = {
8327 { "version", no_argument, NULL, 'V' },
8328 { NULL, 0, NULL, 0}
8330 char *diff_algo_str = NULL;
8332 setlocale(LC_CTYPE, "");
8334 while ((ch = getopt_long(argc, argv, "+hV", longopts, NULL)) != -1) {
8335 switch (ch) {
8336 case 'h':
8337 hflag = 1;
8338 break;
8339 case 'V':
8340 Vflag = 1;
8341 break;
8342 default:
8343 usage(hflag, 1);
8344 /* NOTREACHED */
8348 argc -= optind;
8349 argv += optind;
8350 optind = 1;
8351 optreset = 1;
8353 if (Vflag) {
8354 got_version_print_str();
8355 return 0;
8358 #ifndef PROFILE
8359 if (pledge("stdio rpath wpath cpath flock proc tty exec sendfd unveil",
8360 NULL) == -1)
8361 err(1, "pledge");
8362 #endif
8364 if (argc == 0) {
8365 if (hflag)
8366 usage(hflag, 0);
8367 /* Build an argument vector which runs a default command. */
8368 cmd = &tog_commands[0];
8369 argc = 1;
8370 cmd_argv = make_argv(argc, cmd->name);
8371 } else {
8372 size_t i;
8374 /* Did the user specify a command? */
8375 for (i = 0; i < nitems(tog_commands); i++) {
8376 if (strncmp(tog_commands[i].name, argv[0],
8377 strlen(argv[0])) == 0) {
8378 cmd = &tog_commands[i];
8379 break;
8384 diff_algo_str = getenv("TOG_DIFF_ALGORITHM");
8385 if (diff_algo_str) {
8386 if (strcasecmp(diff_algo_str, "patience") == 0)
8387 tog_diff_algo = GOT_DIFF_ALGORITHM_PATIENCE;
8388 if (strcasecmp(diff_algo_str, "myers") == 0)
8389 tog_diff_algo = GOT_DIFF_ALGORITHM_MYERS;
8392 if (cmd == NULL) {
8393 if (argc != 1)
8394 usage(0, 1);
8395 /* No command specified; try log with a path */
8396 error = tog_log_with_path(argc, argv);
8397 } else {
8398 if (hflag)
8399 cmd->cmd_usage();
8400 else
8401 error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
8404 endwin();
8405 putchar('\n');
8406 if (cmd_argv) {
8407 int i;
8408 for (i = 0; i < argc; i++)
8409 free(cmd_argv[i]);
8410 free(cmd_argv);
8413 if (error && error->code != GOT_ERR_CANCELLED)
8414 fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
8415 return 0;