2 * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
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.
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.
17 #include <sys/queue.h>
21 #define _XOPEN_SOURCE_EXTENDED
23 #undef _XOPEN_SOURCE_EXTENDED
37 #include "got_error.h"
38 #include "got_object.h"
39 #include "got_reference.h"
40 #include "got_repository.h"
42 #include "got_opentemp.h"
43 #include "got_commit_graph.h"
45 #include "got_blame.h"
48 #define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
52 #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
57 const struct got_error *(*cmd_main)(int, char *[]);
58 void (*cmd_usage)(void);
62 __dead static void usage(void);
63 __dead static void usage_log(void);
64 __dead static void usage_diff(void);
65 __dead static void usage_blame(void);
66 __dead static void usage_tree(void);
68 static const struct got_error* cmd_log(int, char *[]);
69 static const struct got_error* cmd_diff(int, char *[]);
70 static const struct got_error* cmd_blame(int, char *[]);
71 static const struct got_error* cmd_tree(int, char *[]);
73 static struct tog_cmd tog_commands[] = {
74 { "log", cmd_log, usage_log,
75 "show repository history" },
76 { "diff", cmd_diff, usage_diff,
77 "compare files and directories" },
78 { "blame", cmd_blame, usage_blame,
79 "show line-by-line file history" },
80 { "tree", cmd_tree, usage_tree,
81 "browse trees in repository" },
84 static struct tog_view {
87 } tog_log_view, tog_diff_view, tog_blame_view, tog_tree_view;
89 static const struct got_error *
90 show_diff_view(struct got_object *, struct got_object *,
91 struct got_repository *);
92 static const struct got_error *
93 show_log_view(struct got_object_id *, struct got_repository *);
94 static const struct got_error *
95 show_blame_view(const char *, struct got_object_id *, struct got_repository *);
96 static const struct got_error *
97 show_tree_view(struct got_tree_object *, struct got_object_id *,
98 struct got_repository *);
104 fprintf(stderr, "usage: %s log [-c commit] [repository-path]\n",
109 /* Create newly allocated wide-character string equivalent to a byte string. */
110 static const struct got_error *
111 mbs2ws(wchar_t **ws, size_t *wlen, const char *s)
114 const struct got_error *err = NULL;
117 *wlen = mbstowcs(NULL, s, 0);
118 if (*wlen == (size_t)-1) {
121 return got_error_from_errno();
123 /* byte string invalid in current encoding; try to "fix" it */
124 err = got_mbsavis(&vis, &vislen, s);
127 *wlen = mbstowcs(NULL, vis, 0);
128 if (*wlen == (size_t)-1) {
129 err = got_error_from_errno(); /* give up */
134 *ws = calloc(*wlen + 1, sizeof(*ws));
136 err = got_error_from_errno();
140 if (mbstowcs(*ws, vis ? vis : s, *wlen) != *wlen)
141 err = got_error_from_errno();
152 /* Format a line for display, ensuring that it won't overflow a width limit. */
153 static const struct got_error *
154 format_line(wchar_t **wlinep, int *widthp, const char *line, int wlimit)
156 const struct got_error *err = NULL;
158 wchar_t *wline = NULL;
164 err = mbs2ws(&wline, &wlen, line);
169 while (i < wlen && cols <= wlimit) {
170 int width = wcwidth(wline[i]);
179 if (wline[i] == L'\t')
183 err = got_error_from_errno();
201 static const struct got_error *
202 draw_commit(struct got_commit_object *commit, struct got_object_id *id)
204 const struct got_error *err = NULL;
205 char *logmsg0 = NULL, *logmsg = NULL;
206 char *author0 = NULL, *author = NULL;
207 wchar_t *wlogmsg = NULL, *wauthor = NULL;
208 int author_width, logmsg_width;
209 char *newline, *smallerthan;
214 static const size_t id_display_cols = 8;
215 static const size_t author_display_cols = 16;
216 const int avail = COLS;
218 err = got_object_id_str(&id_str, id);
221 id_len = strlen(id_str);
222 if (avail < id_display_cols) {
223 limit = MIN(id_len, avail);
224 waddnstr(tog_log_view.window, id_str, limit);
226 limit = MIN(id_display_cols, id_len);
227 waddnstr(tog_log_view.window, id_str, limit);
230 while (col <= avail && col < id_display_cols + 2) {
231 waddch(tog_log_view.window, ' ');
237 author0 = strdup(commit->author);
238 if (author0 == NULL) {
239 err = got_error_from_errno();
243 smallerthan = strchr(author, '<');
247 char *at = strchr(author, '@');
251 limit = MIN(avail, author_display_cols);
252 err = format_line(&wauthor, &author_width, author, limit);
255 waddwstr(tog_log_view.window, wauthor);
257 while (col <= avail && author_width < author_display_cols + 1) {
258 waddch(tog_log_view.window, ' ');
265 logmsg0 = strdup(commit->logmsg);
266 if (logmsg0 == NULL) {
267 err = got_error_from_errno();
271 while (*logmsg == '\n')
273 newline = strchr(logmsg, '\n');
277 err = format_line(&wlogmsg, &logmsg_width, logmsg, limit);
280 waddwstr(tog_log_view.window, wlogmsg);
282 while (col <= avail) {
283 waddch(tog_log_view.window, ' ');
296 struct commit_queue_entry {
297 TAILQ_ENTRY(commit_queue_entry) entry;
298 struct got_object_id *id;
299 struct got_commit_object *commit;
301 TAILQ_HEAD(commit_queue, commit_queue_entry);
303 static struct commit_queue_entry *
304 alloc_commit_queue_entry(struct got_commit_object *commit,
305 struct got_object_id *id)
307 struct commit_queue_entry *entry;
309 entry = calloc(1, sizeof(*entry));
314 entry->commit = commit;
319 pop_commit(struct commit_queue *commits)
321 struct commit_queue_entry *entry;
323 entry = TAILQ_FIRST(commits);
324 TAILQ_REMOVE(commits, entry, entry);
325 got_object_commit_close(entry->commit);
326 /* Don't free entry->id! It is owned by the commit graph. */
331 free_commits(struct commit_queue *commits)
333 while (!TAILQ_EMPTY(commits))
337 static const struct got_error *
338 queue_commits(struct got_commit_graph *graph, struct commit_queue *commits,
339 struct got_object_id *start_id, struct got_repository *repo)
341 const struct got_error *err = NULL;
342 struct got_object_id *id;
343 struct commit_queue_entry *entry;
345 err = got_commit_graph_iter_start(graph, start_id);
349 entry = TAILQ_LAST(commits, commit_queue);
350 if (entry && got_object_id_cmp(entry->id, start_id) == 0) {
353 /* Start ID's commit is already on the queue; skip over it. */
354 err = got_commit_graph_iter_next(&id, graph);
355 if (err && err->code != GOT_ERR_ITER_NEED_MORE)
358 err = got_commit_graph_fetch_commits(&nfetched, graph, 1, repo);
364 struct got_commit_object *commit;
366 err = got_commit_graph_iter_next(&id, graph);
368 if (err->code == GOT_ERR_ITER_NEED_MORE)
373 err = got_object_open_as_commit(&commit, repo, id);
377 entry = alloc_commit_queue_entry(commit, id);
379 err = got_error_from_errno();
383 TAILQ_INSERT_TAIL(commits, entry, entry);
389 static const struct got_error *
390 fetch_next_commit(struct commit_queue_entry **pentry,
391 struct commit_queue_entry *entry, struct commit_queue *commits,
392 struct got_commit_graph *graph, struct got_repository *repo)
394 const struct got_error *err = NULL;
395 struct got_object_qid *qid;
399 /* Populate commit graph with entry's parent commits. */
400 SIMPLEQ_FOREACH(qid, &entry->commit->parent_ids, entry) {
402 err = got_commit_graph_fetch_commits_up_to(&nfetched,
403 graph, qid->id, repo);
408 /* Append outstanding commits to queue in graph sort order. */
409 err = queue_commits(graph, commits, entry->id, repo);
411 if (err->code == GOT_ERR_ITER_COMPLETED)
416 /* Next entry to display should now be available. */
417 *pentry = TAILQ_NEXT(entry, entry);
419 return got_error(GOT_ERR_NO_OBJ);
424 static const struct got_error *
425 get_head_commit_id(struct got_object_id **head_id, struct got_repository *repo)
427 const struct got_error *err = NULL;
428 struct got_reference *head_ref;
432 err = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
436 err = got_ref_resolve(head_id, repo, head_ref);
437 got_ref_close(head_ref);
446 static const struct got_error *
447 draw_commits(struct commit_queue_entry **last, struct commit_queue_entry **selected,
448 struct commit_queue_entry *first, int selected_idx, int limit)
450 const struct got_error *err = NULL;
451 struct commit_queue_entry *entry;
454 werase(tog_log_view.window);
459 if (ncommits == limit)
461 if (ncommits == selected_idx) {
462 wstandout(tog_log_view.window);
465 err = draw_commit(entry->commit, entry->id);
466 if (ncommits == selected_idx)
467 wstandend(tog_log_view.window);
472 entry = TAILQ_NEXT(entry, entry);
482 scroll_up(struct commit_queue_entry **first_displayed_entry, int maxscroll,
483 struct commit_queue *commits)
485 struct commit_queue_entry *entry;
488 entry = TAILQ_FIRST(commits);
489 if (*first_displayed_entry == entry)
492 entry = *first_displayed_entry;
493 while (entry && nscrolled < maxscroll) {
494 entry = TAILQ_PREV(entry, commit_queue, entry);
496 *first_displayed_entry = entry;
502 static const struct got_error *
503 scroll_down(struct commit_queue_entry **first_displayed_entry, int maxscroll,
504 struct commit_queue_entry *last_displayed_entry,
505 struct commit_queue *commits, struct got_commit_graph *graph,
506 struct got_repository *repo)
508 const struct got_error *err = NULL;
509 struct commit_queue_entry *pentry;
513 pentry = TAILQ_NEXT(last_displayed_entry, entry);
514 if (pentry == NULL) {
515 err = fetch_next_commit(&pentry, last_displayed_entry,
516 commits, graph, repo);
517 if (err || pentry == NULL)
520 last_displayed_entry = pentry;
522 pentry = TAILQ_NEXT(*first_displayed_entry, entry);
525 *first_displayed_entry = pentry;
526 } while (++nscrolled < maxscroll);
532 num_parents(struct commit_queue_entry *entry)
537 entry = TAILQ_NEXT(entry, entry);
544 static const struct got_error *
545 show_commit(struct commit_queue_entry *entry, struct got_repository *repo)
547 const struct got_error *err;
548 struct got_object *obj1 = NULL, *obj2 = NULL;
549 struct got_object_qid *parent_id;
551 err = got_object_open(&obj2, repo, entry->id);
555 parent_id = SIMPLEQ_FIRST(&entry->commit->parent_ids);
557 err = got_object_open(&obj1, repo, parent_id->id);
562 err = show_diff_view(obj1, obj2, repo);
565 got_object_close(obj1);
567 got_object_close(obj2);
571 static const struct got_error *
572 browse_commit(struct commit_queue_entry *entry, struct got_repository *repo)
574 const struct got_error *err = NULL;
575 struct got_tree_object *tree;
577 err = got_object_open_as_tree(&tree, repo, entry->commit->tree_id);
581 err = show_tree_view(tree, entry->id, repo);
582 got_object_tree_close(tree);
586 static const struct got_error *
587 show_log_view(struct got_object_id *start_id, struct got_repository *repo)
589 const struct got_error *err = NULL;
590 struct got_object_id *head_id = NULL;
591 int ch, done = 0, selected = 0, nparents, nfetched;
592 struct got_commit_graph *graph;
593 struct commit_queue commits;
594 struct commit_queue_entry *entry = NULL;
595 struct commit_queue_entry *first_displayed_entry = NULL;
596 struct commit_queue_entry *last_displayed_entry = NULL;
597 struct commit_queue_entry *selected_entry = NULL;
599 if (tog_log_view.window == NULL) {
600 tog_log_view.window = newwin(0, 0, 0, 0);
601 if (tog_log_view.window == NULL)
602 return got_error_from_errno();
603 keypad(tog_log_view.window, TRUE);
605 if (tog_log_view.panel == NULL) {
606 tog_log_view.panel = new_panel(tog_log_view.window);
607 if (tog_log_view.panel == NULL)
608 return got_error_from_errno();
610 show_panel(tog_log_view.panel);
612 err = get_head_commit_id(&head_id, repo);
616 TAILQ_INIT(&commits);
618 err = got_commit_graph_open(&graph, head_id, 0, repo);
622 /* Populate commit graph with a sufficient number of commits. */
623 err = got_commit_graph_fetch_commits_up_to(&nfetched, graph, start_id,
627 err = got_commit_graph_fetch_commits(&nfetched, graph, LINES, repo);
632 * Open the initial batch of commits, sorted in commit graph order.
633 * We keep all commits open throughout the lifetime of the log view
634 * in order to avoid having to re-fetch commits from disk while
635 * updating the display.
637 err = queue_commits(graph, &commits, head_id, repo);
638 if (err && err->code != GOT_ERR_ITER_COMPLETED)
641 /* Find entry corresponding to the first commit to display. */
642 TAILQ_FOREACH(entry, &commits, entry) {
643 if (got_object_id_cmp(entry->id, start_id) == 0) {
644 first_displayed_entry = entry;
648 if (first_displayed_entry == NULL) {
649 err = got_error(GOT_ERR_NO_OBJ);
654 err = draw_commits(&last_displayed_entry, &selected_entry,
655 first_displayed_entry, selected, LINES);
659 nodelay(stdscr, FALSE);
660 ch = wgetch(tog_log_view.window);
661 nodelay(stdscr, TRUE);
665 err = got_error_from_errno();
678 scroll_up(&first_displayed_entry, 1, &commits);
681 if (TAILQ_FIRST(&commits) ==
682 first_displayed_entry) {
686 scroll_up(&first_displayed_entry, LINES,
691 nparents = num_parents(first_displayed_entry);
692 if (selected < LINES - 1 &&
693 selected < nparents - 1) {
697 err = scroll_down(&first_displayed_entry, 1,
698 last_displayed_entry, &commits, graph,
704 err = scroll_down(&first_displayed_entry, LINES,
705 last_displayed_entry, &commits, graph,
709 if (last_displayed_entry->commit->nparents > 0)
711 /* can't scroll any further; move cursor down */
712 nparents = num_parents(first_displayed_entry);
713 if (selected < LINES - 1 ||
714 selected < nparents - 1)
715 selected = MIN(LINES - 1, nparents - 1);
718 if (selected > LINES)
719 selected = LINES - 1;
723 err = show_commit(selected_entry, repo);
726 show_panel(tog_log_view.panel);
729 err = browse_commit(selected_entry, repo);
732 show_panel(tog_log_view.panel);
741 got_commit_graph_close(graph);
742 free_commits(&commits);
746 static const struct got_error *
747 cmd_log(int argc, char *argv[])
749 const struct got_error *error;
750 struct got_repository *repo;
751 struct got_object_id *start_id = NULL;
752 char *repo_path = NULL;
753 char *start_commit = NULL;
757 if (pledge("stdio rpath wpath cpath flock proc tty", NULL) == -1)
761 while ((ch = getopt(argc, argv, "c:")) != -1) {
764 start_commit = optarg;
776 repo_path = getcwd(NULL, 0);
777 if (repo_path == NULL)
778 return got_error_from_errno();
779 } else if (argc == 1) {
780 repo_path = realpath(argv[0], NULL);
781 if (repo_path == NULL)
782 return got_error_from_errno();
786 error = got_repo_open(&repo, repo_path);
791 if (start_commit == NULL) {
792 error = get_head_commit_id(&start_id, repo);
796 struct got_object *obj;
797 error = got_object_open_by_id_str(&obj, repo, start_commit);
799 start_id = got_object_get_id(obj);
800 if (start_id == NULL)
801 error = got_error_from_errno();
806 error = show_log_view(start_id, repo);
808 got_repo_close(repo);
816 fprintf(stderr, "usage: %s diff [repository-path] object1 object2\n",
822 parse_next_line(FILE *f, size_t *len)
827 const char delim[3] = { '\0', '\0', '\0'};
829 line = fparseln(f, &linelen, &lineno, delim, 0);
835 static const struct got_error *
836 draw_file(WINDOW *window, FILE *f, int *first_displayed_line,
837 int *last_displayed_line, int *eof, int max_lines)
839 const struct got_error *err;
840 int nlines = 0, nprinted = 0;
850 while (nprinted < max_lines) {
851 line = parse_next_line(f, &len);
856 if (++nlines < *first_displayed_line) {
861 err = format_line(&wline, &width, line, COLS);
866 waddwstr(window, wline);
868 waddch(window, '\n');
870 *first_displayed_line = nlines;
873 *last_displayed_line = nlines;
881 static const struct got_error *
882 show_diff_view(struct got_object *obj1, struct got_object *obj2,
883 struct got_repository *repo)
885 const struct got_error *err;
887 int ch, done = 0, first_displayed_line = 1, last_displayed_line = LINES;
890 if (obj1 != NULL && obj2 != NULL &&
891 got_object_get_type(obj1) != got_object_get_type(obj2))
892 return got_error(GOT_ERR_OBJ_TYPE);
896 return got_error_from_errno();
898 switch (got_object_get_type(obj1 ? obj1 : obj2)) {
899 case GOT_OBJ_TYPE_BLOB:
900 err = got_diff_objects_as_blobs(obj1, obj2, repo, f);
902 case GOT_OBJ_TYPE_TREE:
903 err = got_diff_objects_as_trees(obj1, obj2, repo, f);
905 case GOT_OBJ_TYPE_COMMIT:
906 err = got_diff_objects_as_commits(obj1, obj2, repo, f);
909 return got_error(GOT_ERR_OBJ_TYPE);
914 if (tog_diff_view.window == NULL) {
915 tog_diff_view.window = newwin(0, 0, 0, 0);
916 if (tog_diff_view.window == NULL)
917 return got_error_from_errno();
918 keypad(tog_diff_view.window, TRUE);
920 if (tog_diff_view.panel == NULL) {
921 tog_diff_view.panel = new_panel(tog_diff_view.window);
922 if (tog_diff_view.panel == NULL)
923 return got_error_from_errno();
925 show_panel(tog_diff_view.panel);
928 err = draw_file(tog_diff_view.window, f, &first_displayed_line,
929 &last_displayed_line, &eof, LINES);
932 nodelay(stdscr, FALSE);
933 ch = wgetch(tog_diff_view.window);
934 nodelay(stdscr, TRUE);
942 if (first_displayed_line > 1)
943 first_displayed_line--;
947 while (i++ < LINES - 1 &&
948 first_displayed_line > 1)
949 first_displayed_line--;
956 first_displayed_line++;
961 while (!eof && i++ < LINES - 1) {
962 char *line = parse_next_line(f, NULL);
963 first_displayed_line++;
976 static const struct got_error *
977 cmd_diff(int argc, char *argv[])
979 const struct got_error *error = NULL;
980 struct got_repository *repo = NULL;
981 struct got_object *obj1 = NULL, *obj2 = NULL;
982 char *repo_path = NULL;
983 char *obj_id_str1 = NULL, *obj_id_str2 = NULL;
987 if (pledge("stdio rpath wpath cpath flock proc tty", NULL) == -1)
991 while ((ch = getopt(argc, argv, "")) != -1) {
1003 usage_diff(); /* TODO show local worktree changes */
1004 } else if (argc == 2) {
1005 repo_path = getcwd(NULL, 0);
1006 if (repo_path == NULL)
1007 return got_error_from_errno();
1008 obj_id_str1 = argv[0];
1009 obj_id_str2 = argv[1];
1010 } else if (argc == 3) {
1011 repo_path = realpath(argv[0], NULL);
1012 if (repo_path == NULL)
1013 return got_error_from_errno();
1014 obj_id_str1 = argv[1];
1015 obj_id_str2 = argv[2];
1019 error = got_repo_open(&repo, repo_path);
1024 error = got_object_open_by_id_str(&obj1, repo, obj_id_str1);
1028 error = got_object_open_by_id_str(&obj2, repo, obj_id_str2);
1032 error = show_diff_view(obj1, obj2, repo);
1034 got_repo_close(repo);
1036 got_object_close(obj1);
1038 got_object_close(obj2);
1046 fprintf(stderr, "usage: %s blame [-c commit] [repository-path] path\n",
1051 static const struct got_error *
1052 show_blame_view(const char *path, struct got_object_id *commit_id,
1053 struct got_repository *repo)
1055 const struct got_error *err = NULL;
1057 int ch, done = 0, first_displayed_line = 1, last_displayed_line = LINES;
1062 return got_error_from_errno();
1064 err = got_blame(path, commit_id, repo, f);
1070 if (tog_blame_view.window == NULL) {
1071 tog_blame_view.window = newwin(0, 0, 0, 0);
1072 if (tog_blame_view.window == NULL)
1073 return got_error_from_errno();
1074 keypad(tog_blame_view.window, TRUE);
1076 if (tog_blame_view.panel == NULL) {
1077 tog_blame_view.panel = new_panel(tog_blame_view.window);
1078 if (tog_blame_view.panel == NULL)
1079 return got_error_from_errno();
1081 show_panel(tog_blame_view.panel);
1084 err = draw_file(tog_blame_view.window, f, &first_displayed_line,
1085 &last_displayed_line, &eof, LINES);
1088 nodelay(stdscr, FALSE);
1089 ch = wgetch(tog_blame_view.window);
1090 nodelay(stdscr, TRUE);
1098 if (first_displayed_line > 1)
1099 first_displayed_line--;
1103 while (i++ < LINES - 1 &&
1104 first_displayed_line > 1)
1105 first_displayed_line--;
1112 first_displayed_line++;
1117 while (!eof && i++ < LINES - 1) {
1118 char *line = parse_next_line(f, NULL);
1119 first_displayed_line++;
1133 static const struct got_error *
1134 cmd_blame(int argc, char *argv[])
1136 const struct got_error *error;
1137 struct got_repository *repo = NULL;
1138 char *repo_path = NULL;
1140 struct got_object_id *commit_id = NULL;
1141 char *commit_id_str = NULL;
1145 if (pledge("stdio rpath wpath cpath flock proc tty", NULL) == -1)
1149 while ((ch = getopt(argc, argv, "c:")) != -1) {
1152 commit_id_str = optarg;
1165 } else if (argc == 1) {
1166 repo_path = getcwd(NULL, 0);
1167 if (repo_path == NULL)
1168 return got_error_from_errno();
1170 } else if (argc == 2) {
1171 repo_path = realpath(argv[0], NULL);
1172 if (repo_path == NULL)
1173 return got_error_from_errno();
1178 error = got_repo_open(&repo, repo_path);
1183 if (commit_id_str == NULL) {
1184 struct got_reference *head_ref;
1185 error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1188 error = got_ref_resolve(&commit_id, repo, head_ref);
1189 got_ref_close(head_ref);
1191 struct got_object *obj;
1192 error = got_object_open_by_id_str(&obj, repo, commit_id_str);
1195 commit_id = got_object_get_id(obj);
1196 if (commit_id == NULL)
1197 error = got_error_from_errno();
1198 got_object_close(obj);
1203 error = show_blame_view(path, commit_id, repo);
1207 got_repo_close(repo);
1211 static const struct got_error *
1212 draw_tree_entries(struct got_tree_entry **first_displayed_entry,
1213 struct got_tree_entry **last_displayed_entry,
1214 struct got_tree_entry **selected_entry, int *ndisplayed,
1215 WINDOW *window, const char *label, const char *parent_path,
1216 const struct got_tree_entries *entries, int selected, int limit, int isroot)
1218 const struct got_error *err = NULL;
1219 struct got_tree_entry *te;
1230 err = format_line(&wline, &width, label, COLS);
1233 waddwstr(window, wline);
1235 waddch(window, '\n');
1238 err = format_line(&wline, &width, parent_path, COLS);
1241 waddwstr(window, wline);
1243 waddch(window, '\n');
1246 waddch(window, '\n');
1250 te = SIMPLEQ_FIRST(&entries->head);
1251 if (*first_displayed_entry == NULL) {
1252 if (selected == 0) {
1254 *selected_entry = NULL;
1256 waddstr(window, " ..\n"); /* parent directory */
1265 while (te != *first_displayed_entry)
1266 te = SIMPLEQ_NEXT(te, entry);
1271 if (asprintf(&line, " %s%s",
1272 te->name, S_ISDIR(te->mode) ? "/" : "") == -1)
1273 return got_error_from_errno();
1274 err = format_line(&wline, &width, line, COLS);
1279 if (n == selected) {
1281 *selected_entry = te;
1283 waddwstr(window, wline);
1285 waddch(window, '\n');
1291 *last_displayed_entry = te;
1294 te = SIMPLEQ_NEXT(te, entry);
1301 tree_scroll_up(struct got_tree_entry **first_displayed_entry, int maxscroll,
1302 const struct got_tree_entries *entries, int isroot)
1304 struct got_tree_entry *te, *prev;
1307 if (*first_displayed_entry == NULL)
1310 te = SIMPLEQ_FIRST(&entries->head);
1311 if (*first_displayed_entry == te) {
1313 *first_displayed_entry = NULL;
1317 /* XXX this is stupid... switch to TAILQ? */
1318 for (i = 0; i < maxscroll; i++) {
1319 while (te != *first_displayed_entry) {
1321 te = SIMPLEQ_NEXT(te, entry);
1323 *first_displayed_entry = prev;
1324 te = SIMPLEQ_FIRST(&entries->head);
1326 if (!isroot && te == SIMPLEQ_FIRST(&entries->head) && i < maxscroll)
1327 *first_displayed_entry = NULL;
1331 tree_scroll_down(struct got_tree_entry **first_displayed_entry, int maxscroll,
1332 struct got_tree_entry *last_displayed_entry,
1333 const struct got_tree_entries *entries)
1335 struct got_tree_entry *next;
1338 if (SIMPLEQ_NEXT(last_displayed_entry, entry) == NULL)
1341 if (*first_displayed_entry)
1342 next = SIMPLEQ_NEXT(*first_displayed_entry, entry);
1344 next = SIMPLEQ_FIRST(&entries->head);
1346 *first_displayed_entry = next;
1347 if (++n >= maxscroll)
1349 next = SIMPLEQ_NEXT(next, entry);
1353 struct tog_parent_tree {
1354 TAILQ_ENTRY(tog_parent_tree) entry;
1355 struct got_tree_object *tree;
1356 struct got_tree_entry *first_displayed_entry;
1357 struct got_tree_entry *selected_entry;
1361 TAILQ_HEAD(tog_parent_trees, tog_parent_tree);
1363 static const struct got_error *
1364 tree_entry_path(char **path, struct tog_parent_trees *parents,
1365 struct got_tree_entry *te)
1367 const struct got_error *err = NULL;
1368 struct tog_parent_tree *pt;
1369 size_t len = 2; /* for leading slash and NUL */
1371 TAILQ_FOREACH(pt, parents, entry)
1372 len += strlen(pt->selected_entry->name) + 1 /* slash */;
1374 len += strlen(te->name);
1376 *path = calloc(1, len);
1378 return got_error_from_errno();
1381 pt = TAILQ_LAST(parents, tog_parent_trees);
1383 if (strlcat(*path, pt->selected_entry->name, len) >= len) {
1384 err = got_error(GOT_ERR_NO_SPACE);
1387 if (strlcat(*path, "/", len) >= len) {
1388 err = got_error(GOT_ERR_NO_SPACE);
1391 pt = TAILQ_PREV(pt, tog_parent_trees, entry);
1394 if (strlcat(*path, te->name, len) >= len) {
1395 err = got_error(GOT_ERR_NO_SPACE);
1407 static const struct got_error *
1408 blame_tree_entry(struct got_tree_entry *te, struct tog_parent_trees *parents,
1409 struct got_object_id *commit_id, struct got_repository *repo)
1411 const struct got_error *err = NULL;
1414 err = tree_entry_path(&path, parents, te);
1418 err = show_blame_view(path, commit_id, repo);
1423 static const struct got_error *
1424 show_tree_view(struct got_tree_object *root, struct got_object_id *commit_id,
1425 struct got_repository *repo)
1427 const struct got_error *err = NULL;
1428 int ch, done = 0, selected = 0;
1429 struct got_tree_object *tree = root;
1430 const struct got_tree_entries *entries;
1431 struct got_tree_entry *first_displayed_entry = NULL;
1432 struct got_tree_entry *last_displayed_entry = NULL;
1433 struct got_tree_entry *selected_entry = NULL;
1434 char *commit_id_str = NULL, *tree_label = NULL;
1435 int nentries, ndisplayed;
1436 struct tog_parent_trees parents;
1438 TAILQ_INIT(&parents);
1440 err = got_object_id_str(&commit_id_str, commit_id);
1444 if (asprintf(&tree_label, "tree of commit %s", commit_id_str) == -1) {
1445 err = got_error_from_errno();
1449 if (tog_tree_view.window == NULL) {
1450 tog_tree_view.window = newwin(0, 0, 0, 0);
1451 if (tog_tree_view.window == NULL)
1452 return got_error_from_errno();
1453 keypad(tog_tree_view.window, TRUE);
1455 if (tog_tree_view.panel == NULL) {
1456 tog_tree_view.panel = new_panel(tog_tree_view.window);
1457 if (tog_tree_view.panel == NULL)
1458 return got_error_from_errno();
1460 show_panel(tog_tree_view.panel);
1462 entries = got_object_tree_get_entries(root);
1463 first_displayed_entry = SIMPLEQ_FIRST(&entries->head);
1466 entries = got_object_tree_get_entries(tree);
1467 nentries = entries->nentries;
1469 nentries++; /* '..' directory */
1471 err = tree_entry_path(&parent_path, &parents, NULL);
1475 err = draw_tree_entries(&first_displayed_entry,
1476 &last_displayed_entry, &selected_entry, &ndisplayed,
1477 tog_tree_view.window, tree_label, parent_path, entries,
1478 selected, LINES, tree == root);
1483 nodelay(stdscr, FALSE);
1484 ch = wgetch(tog_tree_view.window);
1485 nodelay(stdscr, TRUE);
1496 tree_scroll_up(&first_displayed_entry, 1,
1497 entries, tree == root);
1500 if (SIMPLEQ_FIRST(&entries->head) ==
1501 first_displayed_entry) {
1503 first_displayed_entry = NULL;
1507 tree_scroll_up(&first_displayed_entry, LINES,
1508 entries, tree == root);
1512 if (selected < ndisplayed - 1) {
1516 tree_scroll_down(&first_displayed_entry, 1,
1517 last_displayed_entry, entries);
1520 tree_scroll_down(&first_displayed_entry, LINES,
1521 last_displayed_entry, entries);
1522 if (SIMPLEQ_NEXT(last_displayed_entry, entry))
1524 /* can't scroll any further; move cursor down */
1525 if (selected < ndisplayed - 1)
1526 selected = ndisplayed - 1;
1530 if (selected_entry == NULL) {
1531 struct tog_parent_tree *parent;
1533 /* user selected '..' */
1536 parent = TAILQ_FIRST(&parents);
1537 TAILQ_REMOVE(&parents, parent, entry);
1538 got_object_tree_close(tree);
1539 tree = parent->tree;
1540 first_displayed_entry =
1541 parent->first_displayed_entry;
1542 selected_entry = parent->selected_entry;
1543 selected = parent->selected;
1545 } else if (S_ISDIR(selected_entry->mode)) {
1546 struct tog_parent_tree *parent;
1547 struct got_tree_object *child;
1548 err = got_object_open_as_tree(
1549 &child, repo, selected_entry->id);
1552 parent = calloc(1, sizeof(*parent));
1553 if (parent == NULL) {
1554 err = got_error_from_errno();
1557 parent->tree = tree;
1558 parent->first_displayed_entry =
1559 first_displayed_entry;
1560 parent->selected_entry = selected_entry;
1561 parent->selected = selected;
1562 TAILQ_INSERT_HEAD(&parents, parent,
1566 first_displayed_entry = NULL;
1567 } else if (S_ISREG(selected_entry->mode)) {
1568 err = blame_tree_entry(selected_entry,
1569 &parents, commit_id, repo);
1575 if (selected > LINES)
1576 selected = ndisplayed - 1;
1584 free(commit_id_str);
1585 while (!TAILQ_EMPTY(&parents)) {
1586 struct tog_parent_tree *parent;
1587 parent = TAILQ_FIRST(&parents);
1588 TAILQ_REMOVE(&parents, parent, entry);
1593 got_object_tree_close(tree);
1601 fprintf(stderr, "usage: %s tree [-c commit] [repository-path]\n",
1606 static const struct got_error *
1607 cmd_tree(int argc, char *argv[])
1609 const struct got_error *error;
1610 struct got_repository *repo = NULL;
1611 char *repo_path = NULL;
1612 struct got_object_id *commit_id = NULL;
1613 char *commit_id_arg = NULL;
1614 struct got_commit_object *commit = NULL;
1615 struct got_tree_object *tree = NULL;
1619 if (pledge("stdio rpath wpath cpath flock proc tty", NULL) == -1)
1623 while ((ch = getopt(argc, argv, "c:")) != -1) {
1626 commit_id_arg = optarg;
1638 repo_path = getcwd(NULL, 0);
1639 if (repo_path == NULL)
1640 return got_error_from_errno();
1641 } else if (argc == 1) {
1642 repo_path = realpath(argv[0], NULL);
1643 if (repo_path == NULL)
1644 return got_error_from_errno();
1648 error = got_repo_open(&repo, repo_path);
1653 if (commit_id_arg == NULL) {
1654 error = get_head_commit_id(&commit_id, repo);
1658 struct got_object *obj;
1659 error = got_object_open_by_id_str(&obj, repo, commit_id_arg);
1660 if (error == NULL) {
1661 commit_id = got_object_get_id(obj);
1662 if (commit_id == NULL)
1663 error = got_error_from_errno();
1669 error = got_object_open_as_commit(&commit, repo, commit_id);
1673 error = got_object_open_as_tree(&tree, repo, commit->tree_id);
1677 error = show_tree_view(tree, commit_id, repo);
1681 got_object_commit_close(commit);
1683 got_object_tree_close(tree);
1685 got_repo_close(repo);
1695 intrflush(stdscr, FALSE);
1696 keypad(stdscr, TRUE);
1705 fprintf(stderr, "usage: %s [-h] [command] [arg ...]\n\n"
1706 "Available commands:\n", getprogname());
1707 for (i = 0; i < nitems(tog_commands); i++) {
1708 struct tog_cmd *cmd = &tog_commands[i];
1709 fprintf(stderr, " %s: %s\n", cmd->name, cmd->descr);
1715 make_argv(const char *arg0, const char *arg1)
1718 int argc = (arg1 == NULL ? 1 : 2);
1720 argv = calloc(argc, sizeof(char *));
1723 argv[0] = strdup(arg0);
1724 if (argv[0] == NULL)
1727 argv[1] = strdup(arg1);
1728 if (argv[1] == NULL)
1736 main(int argc, char *argv[])
1738 const struct got_error *error = NULL;
1739 struct tog_cmd *cmd = NULL;
1741 char **cmd_argv = NULL;
1743 setlocale(LC_ALL, "");
1745 while ((ch = getopt(argc, argv, "h")) != -1) {
1764 /* Build an argument vector which runs a default command. */
1765 cmd = &tog_commands[0];
1766 cmd_argv = make_argv(cmd->name, NULL);
1771 /* Did the user specific a command? */
1772 for (i = 0; i < nitems(tog_commands); i++) {
1773 if (strncmp(tog_commands[i].name, argv[0],
1774 strlen(argv[0])) == 0) {
1775 cmd = &tog_commands[i];
1777 tog_commands[i].cmd_usage();
1782 /* Did the user specify a repository? */
1783 char *repo_path = realpath(argv[0], NULL);
1785 struct got_repository *repo;
1786 error = got_repo_open(&repo, repo_path);
1788 got_repo_close(repo);
1790 error = got_error_from_errno();
1793 fprintf(stderr, "%s: '%s' is not a "
1794 "known command\n", getprogname(),
1798 fprintf(stderr, "%s: '%s' is neither a known "
1799 "command nor a path to a repository\n",
1800 getprogname(), argv[0]);
1804 cmd = &tog_commands[0];
1805 cmd_argv = make_argv(cmd->name, repo_path);
1813 error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
1820 fprintf(stderr, "%s: %s\n", getprogname(), error->msg);