Blame


1 9f7d7167 2018-04-29 stsp /*
2 9f7d7167 2018-04-29 stsp * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
3 9f7d7167 2018-04-29 stsp *
4 9f7d7167 2018-04-29 stsp * Permission to use, copy, modify, and distribute this software for any
5 9f7d7167 2018-04-29 stsp * purpose with or without fee is hereby granted, provided that the above
6 9f7d7167 2018-04-29 stsp * copyright notice and this permission notice appear in all copies.
7 9f7d7167 2018-04-29 stsp *
8 9f7d7167 2018-04-29 stsp * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9 9f7d7167 2018-04-29 stsp * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10 9f7d7167 2018-04-29 stsp * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11 9f7d7167 2018-04-29 stsp * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12 9f7d7167 2018-04-29 stsp * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13 9f7d7167 2018-04-29 stsp * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14 9f7d7167 2018-04-29 stsp * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15 9f7d7167 2018-04-29 stsp */
16 9f7d7167 2018-04-29 stsp
17 80ddbec8 2018-04-29 stsp #include <sys/queue.h>
18 80ddbec8 2018-04-29 stsp
19 31120ada 2018-04-30 stsp #include <errno.h>
20 9f7d7167 2018-04-29 stsp #include <curses.h>
21 9f7d7167 2018-04-29 stsp #include <panel.h>
22 9f7d7167 2018-04-29 stsp #include <locale.h>
23 9f7d7167 2018-04-29 stsp #include <stdlib.h>
24 26ed57b2 2018-05-19 stsp #include <stdio.h>
25 9f7d7167 2018-04-29 stsp #include <getopt.h>
26 9f7d7167 2018-04-29 stsp #include <string.h>
27 9f7d7167 2018-04-29 stsp #include <err.h>
28 80ddbec8 2018-04-29 stsp #include <unistd.h>
29 26ed57b2 2018-05-19 stsp #include <util.h>
30 26ed57b2 2018-05-19 stsp #include <limits.h>
31 9f7d7167 2018-04-29 stsp
32 9f7d7167 2018-04-29 stsp #include "got_error.h"
33 80ddbec8 2018-04-29 stsp #include "got_object.h"
34 80ddbec8 2018-04-29 stsp #include "got_reference.h"
35 80ddbec8 2018-04-29 stsp #include "got_repository.h"
36 80ddbec8 2018-04-29 stsp #include "got_diff.h"
37 9f7d7167 2018-04-29 stsp
38 881b2d3e 2018-04-30 stsp #ifndef MIN
39 881b2d3e 2018-04-30 stsp #define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
40 881b2d3e 2018-04-30 stsp #endif
41 881b2d3e 2018-04-30 stsp
42 9f7d7167 2018-04-29 stsp #ifndef nitems
43 9f7d7167 2018-04-29 stsp #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
44 9f7d7167 2018-04-29 stsp #endif
45 9f7d7167 2018-04-29 stsp
46 9f7d7167 2018-04-29 stsp enum tog_view_id {
47 9f7d7167 2018-04-29 stsp TOG_VIEW_LOG,
48 9f7d7167 2018-04-29 stsp TOG_VIEW_DIFF,
49 9f7d7167 2018-04-29 stsp TOG_VIEW_BLAME,
50 9f7d7167 2018-04-29 stsp };
51 9f7d7167 2018-04-29 stsp
52 9f7d7167 2018-04-29 stsp struct tog_cmd {
53 c2301be8 2018-04-30 stsp const char *name;
54 9f7d7167 2018-04-29 stsp const struct got_error *(*cmd_main)(int, char *[]);
55 9f7d7167 2018-04-29 stsp void (*cmd_usage)(void);
56 9f7d7167 2018-04-29 stsp enum tog_view_id view;
57 c2301be8 2018-04-30 stsp const char *descr;
58 9f7d7167 2018-04-29 stsp };
59 9f7d7167 2018-04-29 stsp
60 9f7d7167 2018-04-29 stsp __dead void usage(void);
61 9f7d7167 2018-04-29 stsp __dead void usage_log(void);
62 9f7d7167 2018-04-29 stsp __dead void usage_diff(void);
63 9f7d7167 2018-04-29 stsp __dead void usage_blame(void);
64 9f7d7167 2018-04-29 stsp
65 9f7d7167 2018-04-29 stsp const struct got_error* cmd_log(int, char *[]);
66 9f7d7167 2018-04-29 stsp const struct got_error* cmd_diff(int, char *[]);
67 9f7d7167 2018-04-29 stsp const struct got_error* cmd_blame(int, char *[]);
68 9f7d7167 2018-04-29 stsp
69 9f7d7167 2018-04-29 stsp struct tog_cmd tog_commands[] = {
70 9f7d7167 2018-04-29 stsp { "log", cmd_log, usage_log, TOG_VIEW_LOG,
71 9f7d7167 2018-04-29 stsp "show repository history" },
72 9f7d7167 2018-04-29 stsp { "diff", cmd_diff, usage_diff, TOG_VIEW_DIFF,
73 9f7d7167 2018-04-29 stsp "compare files and directories" },
74 9f7d7167 2018-04-29 stsp { "blame", cmd_blame, usage_blame, TOG_VIEW_BLAME,
75 9f7d7167 2018-04-29 stsp "show line-by-line file history" },
76 9f7d7167 2018-04-29 stsp };
77 9f7d7167 2018-04-29 stsp
78 9f7d7167 2018-04-29 stsp /* globals */
79 9f7d7167 2018-04-29 stsp WINDOW *tog_main_win;
80 9f7d7167 2018-04-29 stsp PANEL *tog_main_panel;
81 80ddbec8 2018-04-29 stsp static struct tog_log_view {
82 80ddbec8 2018-04-29 stsp WINDOW *window;
83 80ddbec8 2018-04-29 stsp PANEL *panel;
84 80ddbec8 2018-04-29 stsp } tog_log_view;
85 26ed57b2 2018-05-19 stsp static struct tog_diff_view {
86 26ed57b2 2018-05-19 stsp WINDOW *window;
87 26ed57b2 2018-05-19 stsp PANEL *panel;
88 26ed57b2 2018-05-19 stsp } tog_diff_view;
89 9f7d7167 2018-04-29 stsp
90 26ed57b2 2018-05-19 stsp int
91 26ed57b2 2018-05-19 stsp tog_opentempfd(void)
92 26ed57b2 2018-05-19 stsp {
93 26ed57b2 2018-05-19 stsp char name[PATH_MAX];
94 26ed57b2 2018-05-19 stsp int fd;
95 26ed57b2 2018-05-19 stsp
96 26ed57b2 2018-05-19 stsp if (strlcpy(name, "/tmp/tog.XXXXXXXX", sizeof(name)) >= sizeof(name))
97 26ed57b2 2018-05-19 stsp return -1;
98 26ed57b2 2018-05-19 stsp
99 26ed57b2 2018-05-19 stsp fd = mkstemp(name);
100 26ed57b2 2018-05-19 stsp unlink(name);
101 26ed57b2 2018-05-19 stsp return fd;
102 26ed57b2 2018-05-19 stsp }
103 26ed57b2 2018-05-19 stsp
104 26ed57b2 2018-05-19 stsp FILE *
105 26ed57b2 2018-05-19 stsp tog_opentemp(void)
106 26ed57b2 2018-05-19 stsp {
107 26ed57b2 2018-05-19 stsp int fd;
108 26ed57b2 2018-05-19 stsp FILE *f;
109 26ed57b2 2018-05-19 stsp
110 26ed57b2 2018-05-19 stsp fd = tog_opentempfd();
111 26ed57b2 2018-05-19 stsp if (fd < 0)
112 26ed57b2 2018-05-19 stsp return NULL;
113 26ed57b2 2018-05-19 stsp
114 26ed57b2 2018-05-19 stsp f = fdopen(fd, "w+");
115 26ed57b2 2018-05-19 stsp if (f == NULL) {
116 26ed57b2 2018-05-19 stsp close(fd);
117 26ed57b2 2018-05-19 stsp return NULL;
118 26ed57b2 2018-05-19 stsp }
119 26ed57b2 2018-05-19 stsp
120 26ed57b2 2018-05-19 stsp return f;
121 26ed57b2 2018-05-19 stsp }
122 26ed57b2 2018-05-19 stsp
123 9f7d7167 2018-04-29 stsp __dead void
124 9f7d7167 2018-04-29 stsp usage_log(void)
125 9f7d7167 2018-04-29 stsp {
126 80ddbec8 2018-04-29 stsp endwin();
127 80ddbec8 2018-04-29 stsp fprintf(stderr, "usage: %s log [-c commit] [repository-path]\n",
128 9f7d7167 2018-04-29 stsp getprogname());
129 9f7d7167 2018-04-29 stsp exit(1);
130 80ddbec8 2018-04-29 stsp }
131 80ddbec8 2018-04-29 stsp
132 80ddbec8 2018-04-29 stsp static const struct got_error *
133 0553a4e3 2018-04-30 stsp draw_commit(struct got_commit_object *commit, struct got_object_id *id)
134 80ddbec8 2018-04-29 stsp {
135 80ddbec8 2018-04-29 stsp const struct got_error *err = NULL;
136 80ddbec8 2018-04-29 stsp char *logmsg0 = NULL, *logmsg = NULL;
137 6d9fbc00 2018-04-29 stsp char *author0 = NULL, *author = NULL;
138 6d9fbc00 2018-04-29 stsp char *newline, *smallerthan;
139 80ddbec8 2018-04-29 stsp char *line = NULL;
140 6d9fbc00 2018-04-29 stsp char *id_str = NULL;
141 881b2d3e 2018-04-30 stsp const size_t id_display_len = 8;
142 881b2d3e 2018-04-30 stsp const size_t author_display_len = 16;
143 881b2d3e 2018-04-30 stsp size_t id_len, author_len, logmsg_len, avail;
144 881b2d3e 2018-04-30 stsp int i, col;
145 80ddbec8 2018-04-29 stsp
146 80ddbec8 2018-04-29 stsp err = got_object_id_str(&id_str, id);
147 80ddbec8 2018-04-29 stsp if (err)
148 80ddbec8 2018-04-29 stsp return err;
149 881b2d3e 2018-04-30 stsp id_len = strlen(id_str);
150 881b2d3e 2018-04-30 stsp
151 80ddbec8 2018-04-29 stsp logmsg0 = strdup(commit->logmsg);
152 6d9fbc00 2018-04-29 stsp if (logmsg0 == NULL) {
153 6d9fbc00 2018-04-29 stsp err = got_error_from_errno();
154 6d9fbc00 2018-04-29 stsp goto done;
155 6d9fbc00 2018-04-29 stsp }
156 80ddbec8 2018-04-29 stsp logmsg = logmsg0;
157 80ddbec8 2018-04-29 stsp while (*logmsg == '\n')
158 80ddbec8 2018-04-29 stsp logmsg++;
159 80ddbec8 2018-04-29 stsp newline = strchr(logmsg, '\n');
160 6d9fbc00 2018-04-29 stsp if (newline)
161 80ddbec8 2018-04-29 stsp *newline = '\0';
162 881b2d3e 2018-04-30 stsp logmsg_len = strlen(logmsg);
163 80ddbec8 2018-04-29 stsp
164 6d9fbc00 2018-04-29 stsp author0 = strdup(commit->author);
165 6d9fbc00 2018-04-29 stsp if (author0 == NULL) {
166 80ddbec8 2018-04-29 stsp err = got_error_from_errno();
167 80ddbec8 2018-04-29 stsp goto done;
168 80ddbec8 2018-04-29 stsp }
169 6d9fbc00 2018-04-29 stsp author = author0;
170 6d9fbc00 2018-04-29 stsp smallerthan = strchr(author, '<');
171 6d9fbc00 2018-04-29 stsp if (smallerthan)
172 6d9fbc00 2018-04-29 stsp *smallerthan = '\0';
173 6d9fbc00 2018-04-29 stsp else {
174 6d9fbc00 2018-04-29 stsp char *at = strchr(author, '@');
175 6d9fbc00 2018-04-29 stsp if (at)
176 6d9fbc00 2018-04-29 stsp *at = '\0';
177 6d9fbc00 2018-04-29 stsp }
178 881b2d3e 2018-04-30 stsp author_len = strlen(author);
179 80ddbec8 2018-04-29 stsp
180 881b2d3e 2018-04-30 stsp avail = COLS - 1;
181 881b2d3e 2018-04-30 stsp line = calloc(avail + 1, sizeof(*line));
182 881b2d3e 2018-04-30 stsp if (line == NULL) {
183 6d9fbc00 2018-04-29 stsp err = got_error_from_errno();
184 6d9fbc00 2018-04-29 stsp goto done;
185 6d9fbc00 2018-04-29 stsp }
186 6d9fbc00 2018-04-29 stsp
187 881b2d3e 2018-04-30 stsp col = 0;
188 881b2d3e 2018-04-30 stsp for (i = 0; i < MIN(id_display_len, id_len); i++) {
189 881b2d3e 2018-04-30 stsp if (col >= avail)
190 881b2d3e 2018-04-30 stsp goto draw;
191 881b2d3e 2018-04-30 stsp line[col++] = id_str[i];
192 80ddbec8 2018-04-29 stsp }
193 881b2d3e 2018-04-30 stsp while (i < id_display_len) {
194 881b2d3e 2018-04-30 stsp if (col >= avail)
195 881b2d3e 2018-04-30 stsp goto draw;
196 881b2d3e 2018-04-30 stsp line[col++] = ' ';
197 881b2d3e 2018-04-30 stsp i++;
198 881b2d3e 2018-04-30 stsp }
199 881b2d3e 2018-04-30 stsp if (col >= avail)
200 881b2d3e 2018-04-30 stsp goto draw;
201 881b2d3e 2018-04-30 stsp line[col++] = ' ';
202 881b2d3e 2018-04-30 stsp for (i = 0; i < MIN(author_display_len, author_len); i++) {
203 881b2d3e 2018-04-30 stsp if (col >= avail)
204 881b2d3e 2018-04-30 stsp goto draw;
205 881b2d3e 2018-04-30 stsp line[col++] = author[i];
206 881b2d3e 2018-04-30 stsp }
207 881b2d3e 2018-04-30 stsp while (i < author_display_len) {
208 881b2d3e 2018-04-30 stsp if (col >= avail)
209 881b2d3e 2018-04-30 stsp goto draw;
210 881b2d3e 2018-04-30 stsp line[col++] = ' ';
211 881b2d3e 2018-04-30 stsp i++;
212 881b2d3e 2018-04-30 stsp }
213 881b2d3e 2018-04-30 stsp if (col >= avail)
214 881b2d3e 2018-04-30 stsp goto draw;
215 881b2d3e 2018-04-30 stsp line[col++] = ' ';
216 881b2d3e 2018-04-30 stsp
217 881b2d3e 2018-04-30 stsp while (col < avail && *logmsg)
218 881b2d3e 2018-04-30 stsp line[col++] = *logmsg++;
219 881b2d3e 2018-04-30 stsp while (col < avail)
220 881b2d3e 2018-04-30 stsp line[col++] = ' ';
221 881b2d3e 2018-04-30 stsp draw:
222 881b2d3e 2018-04-30 stsp waddstr(tog_log_view.window, line);
223 80ddbec8 2018-04-29 stsp waddch(tog_log_view.window, '\n');
224 80ddbec8 2018-04-29 stsp done:
225 80ddbec8 2018-04-29 stsp free(logmsg0);
226 6d9fbc00 2018-04-29 stsp free(author0);
227 80ddbec8 2018-04-29 stsp free(line);
228 6d9fbc00 2018-04-29 stsp free(id_str);
229 80ddbec8 2018-04-29 stsp return err;
230 80ddbec8 2018-04-29 stsp }
231 26ed57b2 2018-05-19 stsp
232 80ddbec8 2018-04-29 stsp struct commit_queue_entry {
233 80ddbec8 2018-04-29 stsp TAILQ_ENTRY(commit_queue_entry) entry;
234 80ddbec8 2018-04-29 stsp struct got_object_id *id;
235 80ddbec8 2018-04-29 stsp struct got_commit_object *commit;
236 80ddbec8 2018-04-29 stsp };
237 0553a4e3 2018-04-30 stsp TAILQ_HEAD(commit_queue, commit_queue_entry);
238 80ddbec8 2018-04-29 stsp
239 899d86c2 2018-05-10 stsp static struct commit_queue_entry *
240 899d86c2 2018-05-10 stsp alloc_commit_queue_entry(struct got_commit_object *commit,
241 899d86c2 2018-05-10 stsp struct got_object_id *id)
242 80ddbec8 2018-04-29 stsp {
243 80ddbec8 2018-04-29 stsp struct commit_queue_entry *entry;
244 80ddbec8 2018-04-29 stsp
245 80ddbec8 2018-04-29 stsp entry = calloc(1, sizeof(*entry));
246 80ddbec8 2018-04-29 stsp if (entry == NULL)
247 899d86c2 2018-05-10 stsp return NULL;
248 99db9666 2018-05-07 stsp
249 899d86c2 2018-05-10 stsp entry->id = id;
250 99db9666 2018-05-07 stsp entry->commit = commit;
251 899d86c2 2018-05-10 stsp return entry;
252 99db9666 2018-05-07 stsp }
253 80ddbec8 2018-04-29 stsp
254 99db9666 2018-05-07 stsp static void
255 99db9666 2018-05-07 stsp pop_commit(struct commit_queue *commits)
256 99db9666 2018-05-07 stsp {
257 99db9666 2018-05-07 stsp struct commit_queue_entry *entry;
258 99db9666 2018-05-07 stsp
259 99db9666 2018-05-07 stsp entry = TAILQ_FIRST(commits);
260 99db9666 2018-05-07 stsp TAILQ_REMOVE(commits, entry, entry);
261 99db9666 2018-05-07 stsp got_object_commit_close(entry->commit);
262 99db9666 2018-05-07 stsp free(entry->id);
263 99db9666 2018-05-07 stsp free(entry);
264 99db9666 2018-05-07 stsp }
265 99db9666 2018-05-07 stsp
266 99db9666 2018-05-07 stsp static void
267 99db9666 2018-05-07 stsp free_commits(struct commit_queue *commits)
268 99db9666 2018-05-07 stsp {
269 99db9666 2018-05-07 stsp while (!TAILQ_EMPTY(commits))
270 99db9666 2018-05-07 stsp pop_commit(commits);
271 c4972b91 2018-05-07 stsp }
272 c4972b91 2018-05-07 stsp
273 c4972b91 2018-05-07 stsp
274 c4972b91 2018-05-07 stsp static const struct got_error *
275 899d86c2 2018-05-10 stsp fetch_parent_commit(struct commit_queue_entry **pentry,
276 899d86c2 2018-05-10 stsp struct commit_queue_entry *entry, struct got_repository *repo)
277 c4972b91 2018-05-07 stsp {
278 899d86c2 2018-05-10 stsp const struct got_error *err = NULL;
279 c4972b91 2018-05-07 stsp struct got_object *obj = NULL;
280 899d86c2 2018-05-10 stsp struct got_commit_object *commit;
281 899d86c2 2018-05-10 stsp struct got_object_id *id;
282 c4972b91 2018-05-07 stsp struct got_parent_id *pid;
283 c4972b91 2018-05-07 stsp
284 899d86c2 2018-05-10 stsp *pentry = NULL;
285 c4972b91 2018-05-07 stsp
286 c4972b91 2018-05-07 stsp /* Follow the first parent (TODO: handle merge commits). */
287 c4972b91 2018-05-07 stsp pid = SIMPLEQ_FIRST(&entry->commit->parent_ids);
288 c4972b91 2018-05-07 stsp if (pid == NULL)
289 c4972b91 2018-05-07 stsp return NULL;
290 c4972b91 2018-05-07 stsp err = got_object_open(&obj, repo, pid->id);
291 c4972b91 2018-05-07 stsp if (err)
292 c4972b91 2018-05-07 stsp return err;
293 c4972b91 2018-05-07 stsp if (got_object_get_type(obj) != GOT_OBJ_TYPE_COMMIT) {
294 c4972b91 2018-05-07 stsp err = got_error(GOT_ERR_OBJ_TYPE);
295 c4972b91 2018-05-07 stsp got_object_close(obj);
296 c4972b91 2018-05-07 stsp return err;
297 c4972b91 2018-05-07 stsp }
298 c4972b91 2018-05-07 stsp
299 899d86c2 2018-05-10 stsp err = got_object_commit_open(&commit, repo, obj);
300 c4972b91 2018-05-07 stsp got_object_close(obj);
301 c4972b91 2018-05-07 stsp if (err)
302 c4972b91 2018-05-07 stsp return err;
303 c4972b91 2018-05-07 stsp
304 899d86c2 2018-05-10 stsp id = got_object_id_dup(pid->id);
305 899d86c2 2018-05-10 stsp if (id == NULL) {
306 c4972b91 2018-05-07 stsp err = got_error_from_errno();
307 899d86c2 2018-05-10 stsp got_object_commit_close(commit);
308 899d86c2 2018-05-10 stsp return err;;
309 c4972b91 2018-05-07 stsp }
310 899d86c2 2018-05-10 stsp
311 899d86c2 2018-05-10 stsp *pentry = alloc_commit_queue_entry(commit, id);
312 899d86c2 2018-05-10 stsp if (*pentry == NULL) {
313 c4972b91 2018-05-07 stsp err = got_error_from_errno();
314 899d86c2 2018-05-10 stsp got_object_commit_close(commit);
315 c4972b91 2018-05-07 stsp }
316 c4972b91 2018-05-07 stsp
317 899d86c2 2018-05-10 stsp return err;;
318 899d86c2 2018-05-10 stsp }
319 899d86c2 2018-05-10 stsp
320 899d86c2 2018-05-10 stsp static const struct got_error *
321 899d86c2 2018-05-10 stsp get_head_commit_id(struct got_object_id **head_id, struct got_repository *repo)
322 899d86c2 2018-05-10 stsp {
323 899d86c2 2018-05-10 stsp const struct got_error *err = NULL;
324 899d86c2 2018-05-10 stsp struct got_reference *head_ref;
325 899d86c2 2018-05-10 stsp
326 899d86c2 2018-05-10 stsp *head_id = NULL;
327 899d86c2 2018-05-10 stsp
328 899d86c2 2018-05-10 stsp err = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
329 899d86c2 2018-05-10 stsp if (err)
330 899d86c2 2018-05-10 stsp return err;
331 899d86c2 2018-05-10 stsp
332 899d86c2 2018-05-10 stsp err = got_ref_resolve(head_id, repo, head_ref);
333 899d86c2 2018-05-10 stsp got_ref_close(head_ref);
334 899d86c2 2018-05-10 stsp if (err) {
335 899d86c2 2018-05-10 stsp *head_id = NULL;
336 899d86c2 2018-05-10 stsp return err;
337 899d86c2 2018-05-10 stsp }
338 899d86c2 2018-05-10 stsp
339 c4972b91 2018-05-07 stsp return NULL;
340 99db9666 2018-05-07 stsp }
341 99db9666 2018-05-07 stsp
342 99db9666 2018-05-07 stsp static const struct got_error *
343 899d86c2 2018-05-10 stsp prepend_commits(int *ncommits, struct commit_queue *commits,
344 899d86c2 2018-05-10 stsp struct got_object_id *first_id, struct got_object_id *last_id,
345 899d86c2 2018-05-10 stsp int limit, struct got_repository *repo)
346 899d86c2 2018-05-10 stsp {
347 899d86c2 2018-05-10 stsp const struct got_error *err = NULL;
348 899d86c2 2018-05-10 stsp struct got_object *first_obj = NULL, *last_obj = NULL;
349 899d86c2 2018-05-10 stsp struct got_commit_object *commit = NULL;
350 899d86c2 2018-05-10 stsp struct got_object_id *id = NULL;
351 899d86c2 2018-05-10 stsp struct commit_queue_entry *entry, *old_head_entry;
352 899d86c2 2018-05-10 stsp
353 899d86c2 2018-05-10 stsp *ncommits = 0;
354 899d86c2 2018-05-10 stsp
355 899d86c2 2018-05-10 stsp err = got_object_open(&first_obj, repo, first_id);
356 899d86c2 2018-05-10 stsp if (err)
357 899d86c2 2018-05-10 stsp goto done;
358 899d86c2 2018-05-10 stsp if (got_object_get_type(first_obj) != GOT_OBJ_TYPE_COMMIT) {
359 899d86c2 2018-05-10 stsp err = got_error(GOT_ERR_OBJ_TYPE);
360 899d86c2 2018-05-10 stsp goto done;
361 899d86c2 2018-05-10 stsp }
362 899d86c2 2018-05-10 stsp err = got_object_open(&last_obj, repo, last_id);
363 899d86c2 2018-05-10 stsp if (err)
364 899d86c2 2018-05-10 stsp goto done;
365 899d86c2 2018-05-10 stsp if (got_object_get_type(last_obj) != GOT_OBJ_TYPE_COMMIT) {
366 899d86c2 2018-05-10 stsp err = got_error(GOT_ERR_OBJ_TYPE);
367 899d86c2 2018-05-10 stsp goto done;
368 899d86c2 2018-05-10 stsp }
369 899d86c2 2018-05-10 stsp
370 899d86c2 2018-05-10 stsp err = got_object_commit_open(&commit, repo, first_obj);
371 899d86c2 2018-05-10 stsp if (err)
372 899d86c2 2018-05-10 stsp goto done;
373 899d86c2 2018-05-10 stsp
374 899d86c2 2018-05-10 stsp id = got_object_id_dup(first_id);
375 899d86c2 2018-05-10 stsp if (id == NULL) {
376 899d86c2 2018-05-10 stsp err = got_error_from_errno();
377 899d86c2 2018-05-10 stsp goto done;
378 899d86c2 2018-05-10 stsp }
379 899d86c2 2018-05-10 stsp
380 899d86c2 2018-05-10 stsp entry = alloc_commit_queue_entry(commit, id);
381 899d86c2 2018-05-10 stsp if (entry == NULL)
382 899d86c2 2018-05-10 stsp return got_error_from_errno();
383 899d86c2 2018-05-10 stsp
384 899d86c2 2018-05-10 stsp old_head_entry = TAILQ_FIRST(commits);
385 899d86c2 2018-05-10 stsp if (old_head_entry)
386 899d86c2 2018-05-10 stsp TAILQ_INSERT_BEFORE(old_head_entry, entry, entry);
387 899d86c2 2018-05-10 stsp else
388 899d86c2 2018-05-10 stsp TAILQ_INSERT_HEAD(commits, entry, entry);
389 899d86c2 2018-05-10 stsp
390 899d86c2 2018-05-10 stsp *ncommits = 1;
391 899d86c2 2018-05-10 stsp
392 899d86c2 2018-05-10 stsp /*
393 899d86c2 2018-05-10 stsp * Fetch parent commits.
394 899d86c2 2018-05-10 stsp * XXX If first and last commit aren't ancestrally related this loop
395 899d86c2 2018-05-10 stsp * we will keep iterating until a root commit is encountered.
396 899d86c2 2018-05-10 stsp */
397 899d86c2 2018-05-10 stsp while (1) {
398 899d86c2 2018-05-10 stsp struct commit_queue_entry *pentry;
399 899d86c2 2018-05-10 stsp
400 899d86c2 2018-05-10 stsp err = fetch_parent_commit(&pentry, entry, repo);
401 899d86c2 2018-05-10 stsp if (err)
402 899d86c2 2018-05-10 stsp goto done;
403 899d86c2 2018-05-10 stsp if (pentry == NULL)
404 899d86c2 2018-05-10 stsp break;
405 899d86c2 2018-05-10 stsp
406 899d86c2 2018-05-10 stsp /*
407 899d86c2 2018-05-10 stsp * Fill up to old HEAD commit if commit queue was not empty.
408 899d86c2 2018-05-10 stsp * We must not leave a gap in history.
409 899d86c2 2018-05-10 stsp */
410 899d86c2 2018-05-10 stsp if (old_head_entry &&
411 899d86c2 2018-05-10 stsp got_object_id_cmp(pentry->id, old_head_entry->id) == 0)
412 899d86c2 2018-05-10 stsp break;
413 899d86c2 2018-05-10 stsp
414 899d86c2 2018-05-10 stsp TAILQ_INSERT_AFTER(commits, entry, pentry, entry);
415 899d86c2 2018-05-10 stsp (*ncommits)++;
416 899d86c2 2018-05-10 stsp if (*ncommits >= limit)
417 899d86c2 2018-05-10 stsp break;
418 899d86c2 2018-05-10 stsp
419 899d86c2 2018-05-10 stsp /* Fill up to last requested commit if queue was empty. */
420 899d86c2 2018-05-10 stsp if (old_head_entry == NULL &&
421 899d86c2 2018-05-10 stsp got_object_id_cmp(pentry->id, last_id) == 0)
422 899d86c2 2018-05-10 stsp break;
423 899d86c2 2018-05-10 stsp
424 899d86c2 2018-05-10 stsp entry = pentry;
425 899d86c2 2018-05-10 stsp }
426 899d86c2 2018-05-10 stsp
427 899d86c2 2018-05-10 stsp done:
428 899d86c2 2018-05-10 stsp if (first_obj)
429 899d86c2 2018-05-10 stsp got_object_close(first_obj);
430 899d86c2 2018-05-10 stsp if (last_obj)
431 899d86c2 2018-05-10 stsp got_object_close(last_obj);
432 899d86c2 2018-05-10 stsp return err;
433 899d86c2 2018-05-10 stsp }
434 899d86c2 2018-05-10 stsp
435 899d86c2 2018-05-10 stsp static const struct got_error *
436 899d86c2 2018-05-10 stsp fetch_commits(struct commit_queue_entry **start_entry,
437 899d86c2 2018-05-10 stsp struct got_object_id *start_id, struct commit_queue *commits,
438 899d86c2 2018-05-10 stsp int limit, struct got_repository *repo)
439 99db9666 2018-05-07 stsp {
440 99db9666 2018-05-07 stsp const struct got_error *err;
441 99db9666 2018-05-07 stsp struct commit_queue_entry *entry;
442 99db9666 2018-05-07 stsp int ncommits = 0;
443 899d86c2 2018-05-10 stsp struct got_object_id *head_id = NULL;
444 99db9666 2018-05-07 stsp
445 899d86c2 2018-05-10 stsp *start_entry = NULL;
446 899d86c2 2018-05-10 stsp
447 899d86c2 2018-05-10 stsp err = get_head_commit_id(&head_id, repo);
448 99db9666 2018-05-07 stsp if (err)
449 99db9666 2018-05-07 stsp return err;
450 99db9666 2018-05-07 stsp
451 899d86c2 2018-05-10 stsp /* Prepend HEAD commit and all ancestors up to start commit. */
452 899d86c2 2018-05-10 stsp err = prepend_commits(&ncommits, commits, head_id, start_id, limit,
453 899d86c2 2018-05-10 stsp repo);
454 99db9666 2018-05-07 stsp if (err)
455 99db9666 2018-05-07 stsp return err;
456 99db9666 2018-05-07 stsp
457 899d86c2 2018-05-10 stsp if (got_object_id_cmp(head_id, start_id) == 0)
458 899d86c2 2018-05-10 stsp *start_entry = TAILQ_FIRST(commits);
459 899d86c2 2018-05-10 stsp else
460 899d86c2 2018-05-10 stsp *start_entry = TAILQ_LAST(commits, commit_queue);
461 899d86c2 2018-05-10 stsp
462 899d86c2 2018-05-10 stsp if (ncommits >= limit)
463 899d86c2 2018-05-10 stsp return NULL;
464 899d86c2 2018-05-10 stsp
465 899d86c2 2018-05-10 stsp /* Append more commits from start commit up to the requested limit. */
466 899d86c2 2018-05-10 stsp entry = TAILQ_LAST(commits, commit_queue);
467 899d86c2 2018-05-10 stsp while (entry && ncommits < limit) {
468 899d86c2 2018-05-10 stsp struct commit_queue_entry *pentry;
469 899d86c2 2018-05-10 stsp
470 899d86c2 2018-05-10 stsp err = fetch_parent_commit(&pentry, entry, repo);
471 80ddbec8 2018-04-29 stsp if (err)
472 80ddbec8 2018-04-29 stsp break;
473 899d86c2 2018-05-10 stsp if (pentry)
474 899d86c2 2018-05-10 stsp TAILQ_INSERT_TAIL(commits, pentry, entry);
475 899d86c2 2018-05-10 stsp entry = pentry;
476 0553a4e3 2018-04-30 stsp ncommits++;
477 0553a4e3 2018-04-30 stsp }
478 80ddbec8 2018-04-29 stsp
479 899d86c2 2018-05-10 stsp if (err)
480 899d86c2 2018-05-10 stsp *start_entry = NULL;
481 0553a4e3 2018-04-30 stsp return err;
482 0553a4e3 2018-04-30 stsp }
483 0553a4e3 2018-04-30 stsp
484 0553a4e3 2018-04-30 stsp static const struct got_error *
485 4a7f7875 2018-05-10 stsp draw_commits(struct commit_queue_entry **last, struct commit_queue_entry *first,
486 4a7f7875 2018-05-10 stsp int selected, int limit)
487 0553a4e3 2018-04-30 stsp {
488 0553a4e3 2018-04-30 stsp const struct got_error *err = NULL;
489 0553a4e3 2018-04-30 stsp struct commit_queue_entry *entry;
490 0553a4e3 2018-04-30 stsp int ncommits = 0;
491 0553a4e3 2018-04-30 stsp
492 0553a4e3 2018-04-30 stsp wclear(tog_log_view.window);
493 0553a4e3 2018-04-30 stsp
494 899d86c2 2018-05-10 stsp entry = first;
495 899d86c2 2018-05-10 stsp *last = first;
496 899d86c2 2018-05-10 stsp while (entry) {
497 899d86c2 2018-05-10 stsp if (ncommits == limit)
498 899d86c2 2018-05-10 stsp break;
499 0553a4e3 2018-04-30 stsp if (ncommits == selected)
500 0553a4e3 2018-04-30 stsp wstandout(tog_log_view.window);
501 0553a4e3 2018-04-30 stsp err = draw_commit(entry->commit, entry->id);
502 0553a4e3 2018-04-30 stsp if (ncommits == selected)
503 0553a4e3 2018-04-30 stsp wstandend(tog_log_view.window);
504 0553a4e3 2018-04-30 stsp if (err)
505 0553a4e3 2018-04-30 stsp break;
506 0553a4e3 2018-04-30 stsp ncommits++;
507 899d86c2 2018-05-10 stsp *last = entry;
508 899d86c2 2018-05-10 stsp entry = TAILQ_NEXT(entry, entry);
509 80ddbec8 2018-04-29 stsp }
510 80ddbec8 2018-04-29 stsp
511 80ddbec8 2018-04-29 stsp update_panels();
512 80ddbec8 2018-04-29 stsp doupdate();
513 0553a4e3 2018-04-30 stsp
514 80ddbec8 2018-04-29 stsp return err;
515 9f7d7167 2018-04-29 stsp }
516 07b55e75 2018-05-10 stsp
517 07b55e75 2018-05-10 stsp static void
518 07b55e75 2018-05-10 stsp scroll_up(struct commit_queue_entry **first_displayed_entry, int n,
519 07b55e75 2018-05-10 stsp struct commit_queue *commits)
520 07b55e75 2018-05-10 stsp {
521 07b55e75 2018-05-10 stsp struct commit_queue_entry *entry;
522 07b55e75 2018-05-10 stsp int nscrolled = 0;
523 07b55e75 2018-05-10 stsp
524 07b55e75 2018-05-10 stsp entry = TAILQ_FIRST(commits);
525 07b55e75 2018-05-10 stsp if (*first_displayed_entry == entry)
526 07b55e75 2018-05-10 stsp return;
527 9f7d7167 2018-04-29 stsp
528 07b55e75 2018-05-10 stsp entry = *first_displayed_entry;
529 07b55e75 2018-05-10 stsp while (entry && nscrolled < n) {
530 07b55e75 2018-05-10 stsp entry = TAILQ_PREV(entry, commit_queue, entry);
531 07b55e75 2018-05-10 stsp if (entry) {
532 07b55e75 2018-05-10 stsp *first_displayed_entry = entry;
533 07b55e75 2018-05-10 stsp nscrolled++;
534 07b55e75 2018-05-10 stsp }
535 07b55e75 2018-05-10 stsp }
536 aa075928 2018-05-10 stsp }
537 aa075928 2018-05-10 stsp
538 aa075928 2018-05-10 stsp static const struct got_error *
539 aa075928 2018-05-10 stsp scroll_down(struct commit_queue_entry **first_displayed_entry, int n,
540 aa075928 2018-05-10 stsp struct commit_queue_entry *last_displayed_entry,
541 aa075928 2018-05-10 stsp struct commit_queue *commits, struct got_repository *repo)
542 aa075928 2018-05-10 stsp {
543 aa075928 2018-05-10 stsp const struct got_error *err = NULL;
544 aa075928 2018-05-10 stsp struct commit_queue_entry *entry;
545 aa075928 2018-05-10 stsp int nscrolled = 0;
546 aa075928 2018-05-10 stsp
547 aa075928 2018-05-10 stsp if (last_displayed_entry->commit->nparents == 0)
548 aa075928 2018-05-10 stsp return NULL;
549 aa075928 2018-05-10 stsp
550 aa075928 2018-05-10 stsp entry = *first_displayed_entry;
551 aa075928 2018-05-10 stsp do {
552 aa075928 2018-05-10 stsp struct commit_queue_entry *pentry;
553 aa075928 2018-05-10 stsp
554 aa075928 2018-05-10 stsp pentry = TAILQ_NEXT(entry, entry);
555 aa075928 2018-05-10 stsp if (pentry == NULL) {
556 aa075928 2018-05-10 stsp err = fetch_parent_commit(&pentry, entry, repo);
557 aa075928 2018-05-10 stsp if (err)
558 aa075928 2018-05-10 stsp break;
559 aa075928 2018-05-10 stsp if (pentry == NULL) {
560 aa075928 2018-05-10 stsp *first_displayed_entry = entry;
561 aa075928 2018-05-10 stsp return NULL;
562 aa075928 2018-05-10 stsp }
563 aa075928 2018-05-10 stsp TAILQ_INSERT_TAIL(commits, pentry, entry);
564 aa075928 2018-05-10 stsp last_displayed_entry = pentry;
565 aa075928 2018-05-10 stsp }
566 aa075928 2018-05-10 stsp
567 aa075928 2018-05-10 stsp *first_displayed_entry = pentry;
568 aa075928 2018-05-10 stsp entry = pentry;
569 aa075928 2018-05-10 stsp
570 aa075928 2018-05-10 stsp if (TAILQ_LAST(commits, commit_queue) == last_displayed_entry) {
571 aa075928 2018-05-10 stsp err = fetch_parent_commit(&pentry, last_displayed_entry,
572 aa075928 2018-05-10 stsp repo);
573 aa075928 2018-05-10 stsp if (err)
574 aa075928 2018-05-10 stsp break;
575 aa075928 2018-05-10 stsp if (pentry) {
576 aa075928 2018-05-10 stsp TAILQ_INSERT_TAIL(commits, pentry, entry);
577 aa075928 2018-05-10 stsp last_displayed_entry = pentry;
578 aa075928 2018-05-10 stsp }
579 aa075928 2018-05-10 stsp }
580 aa075928 2018-05-10 stsp } while (++nscrolled < n);
581 aa075928 2018-05-10 stsp
582 aa075928 2018-05-10 stsp return NULL;
583 07b55e75 2018-05-10 stsp }
584 4a7f7875 2018-05-10 stsp
585 4a7f7875 2018-05-10 stsp static int
586 4a7f7875 2018-05-10 stsp num_parents(struct commit_queue_entry *entry)
587 4a7f7875 2018-05-10 stsp {
588 4a7f7875 2018-05-10 stsp int nparents = 0;
589 07b55e75 2018-05-10 stsp
590 4a7f7875 2018-05-10 stsp while (entry) {
591 4a7f7875 2018-05-10 stsp entry = TAILQ_NEXT(entry, entry);
592 4a7f7875 2018-05-10 stsp nparents++;
593 4a7f7875 2018-05-10 stsp }
594 4a7f7875 2018-05-10 stsp
595 4a7f7875 2018-05-10 stsp return nparents;
596 4a7f7875 2018-05-10 stsp }
597 4a7f7875 2018-05-10 stsp
598 80ddbec8 2018-04-29 stsp static const struct got_error *
599 80ddbec8 2018-04-29 stsp show_log_view(struct got_object_id *start_id, struct got_repository *repo)
600 80ddbec8 2018-04-29 stsp {
601 80ddbec8 2018-04-29 stsp const struct got_error *err = NULL;
602 c4972b91 2018-05-07 stsp struct got_object_id *id;
603 4a7f7875 2018-05-10 stsp int ch, done = 0, selected = 0, nparents;
604 0553a4e3 2018-04-30 stsp struct commit_queue commits;
605 899d86c2 2018-05-10 stsp struct commit_queue_entry *first_displayed_entry = NULL;
606 899d86c2 2018-05-10 stsp struct commit_queue_entry *last_displayed_entry = NULL;
607 80ddbec8 2018-04-29 stsp
608 c4972b91 2018-05-07 stsp id = got_object_id_dup(start_id);
609 c4972b91 2018-05-07 stsp if (id == NULL)
610 c4972b91 2018-05-07 stsp return got_error_from_errno();
611 c4972b91 2018-05-07 stsp
612 80ddbec8 2018-04-29 stsp if (tog_log_view.window == NULL) {
613 80ddbec8 2018-04-29 stsp tog_log_view.window = newwin(0, 0, 0, 0);
614 80ddbec8 2018-04-29 stsp if (tog_log_view.window == NULL)
615 80ddbec8 2018-04-29 stsp return got_error_from_errno();
616 00a838fa 2018-04-29 stsp keypad(tog_log_view.window, TRUE);
617 80ddbec8 2018-04-29 stsp }
618 80ddbec8 2018-04-29 stsp if (tog_log_view.panel == NULL) {
619 80ddbec8 2018-04-29 stsp tog_log_view.panel = new_panel(tog_log_view.window);
620 80ddbec8 2018-04-29 stsp if (tog_log_view.panel == NULL)
621 80ddbec8 2018-04-29 stsp return got_error_from_errno();
622 80ddbec8 2018-04-29 stsp }
623 80ddbec8 2018-04-29 stsp
624 899d86c2 2018-05-10 stsp TAILQ_INIT(&commits);
625 899d86c2 2018-05-10 stsp err = fetch_commits(&first_displayed_entry, id, &commits, LINES, repo);
626 80ddbec8 2018-04-29 stsp if (err)
627 80ddbec8 2018-04-29 stsp goto done;
628 899d86c2 2018-05-10 stsp while (!done) {
629 899d86c2 2018-05-10 stsp err = draw_commits(&last_displayed_entry, first_displayed_entry,
630 899d86c2 2018-05-10 stsp selected, LINES);
631 80ddbec8 2018-04-29 stsp if (err)
632 d0f709cb 2018-04-30 stsp goto done;
633 80ddbec8 2018-04-29 stsp
634 80ddbec8 2018-04-29 stsp nodelay(stdscr, FALSE);
635 80ddbec8 2018-04-29 stsp ch = wgetch(tog_log_view.window);
636 80ddbec8 2018-04-29 stsp switch (ch) {
637 31120ada 2018-04-30 stsp case ERR:
638 31120ada 2018-04-30 stsp if (errno) {
639 31120ada 2018-04-30 stsp err = got_error_from_errno();
640 31120ada 2018-04-30 stsp goto done;
641 31120ada 2018-04-30 stsp }
642 31120ada 2018-04-30 stsp break;
643 80ddbec8 2018-04-29 stsp case 'q':
644 80ddbec8 2018-04-29 stsp done = 1;
645 80ddbec8 2018-04-29 stsp break;
646 80ddbec8 2018-04-29 stsp case 'k':
647 80ddbec8 2018-04-29 stsp case KEY_UP:
648 80ddbec8 2018-04-29 stsp if (selected > 0)
649 80ddbec8 2018-04-29 stsp selected--;
650 8ec9698d 2018-05-10 stsp if (selected > 0)
651 d91e25cb 2018-05-10 stsp break;
652 07b55e75 2018-05-10 stsp scroll_up(&first_displayed_entry, 1, &commits);
653 80ddbec8 2018-04-29 stsp break;
654 48531068 2018-05-10 stsp case KEY_PPAGE:
655 dfc1d240 2018-05-10 stsp if (TAILQ_FIRST(&commits) ==
656 dfc1d240 2018-05-10 stsp first_displayed_entry) {
657 dfc1d240 2018-05-10 stsp selected = 0;
658 dfc1d240 2018-05-10 stsp break;
659 dfc1d240 2018-05-10 stsp }
660 48531068 2018-05-10 stsp scroll_up(&first_displayed_entry, LINES,
661 48531068 2018-05-10 stsp &commits);
662 48531068 2018-05-10 stsp break;
663 80ddbec8 2018-04-29 stsp case 'j':
664 80ddbec8 2018-04-29 stsp case KEY_DOWN:
665 80ee4603 2018-05-10 stsp nparents = num_parents(first_displayed_entry);
666 06abe2cd 2018-05-10 stsp if (selected < LINES - 1 &&
667 06abe2cd 2018-05-10 stsp selected < nparents - 1)
668 80ddbec8 2018-04-29 stsp selected++;
669 06abe2cd 2018-05-10 stsp if (selected < LINES - 1 &&
670 06abe2cd 2018-05-10 stsp selected < nparents - 1)
671 c4972b91 2018-05-07 stsp break;
672 aa075928 2018-05-10 stsp err = scroll_down(&first_displayed_entry, 1,
673 aa075928 2018-05-10 stsp last_displayed_entry, &commits, repo);
674 4a7f7875 2018-05-10 stsp if (err)
675 4a7f7875 2018-05-10 stsp goto done;
676 4a7f7875 2018-05-10 stsp break;
677 4a7f7875 2018-05-10 stsp case KEY_NPAGE:
678 4a7f7875 2018-05-10 stsp nparents = num_parents(first_displayed_entry);
679 dfc1d240 2018-05-10 stsp if (nparents < LINES - 1 &&
680 dfc1d240 2018-05-10 stsp selected < nparents - 1) {
681 dfc1d240 2018-05-10 stsp selected = nparents - 1;
682 dfc1d240 2018-05-10 stsp break;
683 dfc1d240 2018-05-10 stsp }
684 e50ee4f1 2018-05-10 stsp err = scroll_down(&first_displayed_entry, LINES,
685 e50ee4f1 2018-05-10 stsp last_displayed_entry, &commits, repo);
686 899d86c2 2018-05-10 stsp if (err)
687 aa075928 2018-05-10 stsp goto done;
688 4a7f7875 2018-05-10 stsp nparents = num_parents(first_displayed_entry);
689 dfc1d240 2018-05-10 stsp if (selected > nparents)
690 dfc1d240 2018-05-10 stsp selected = nparents - 1;
691 80ddbec8 2018-04-29 stsp break;
692 d6df9be4 2018-04-30 stsp case KEY_RESIZE:
693 31120ada 2018-04-30 stsp if (selected > LINES)
694 31120ada 2018-04-30 stsp selected = LINES - 1;
695 31120ada 2018-04-30 stsp break;
696 80ddbec8 2018-04-29 stsp default:
697 80ddbec8 2018-04-29 stsp break;
698 80ddbec8 2018-04-29 stsp }
699 80ddbec8 2018-04-29 stsp nodelay(stdscr, TRUE);
700 899d86c2 2018-05-10 stsp }
701 80ddbec8 2018-04-29 stsp done:
702 0553a4e3 2018-04-30 stsp free_commits(&commits);
703 80ddbec8 2018-04-29 stsp return err;
704 80ddbec8 2018-04-29 stsp }
705 80ddbec8 2018-04-29 stsp
706 9f7d7167 2018-04-29 stsp const struct got_error *
707 9f7d7167 2018-04-29 stsp cmd_log(int argc, char *argv[])
708 9f7d7167 2018-04-29 stsp {
709 80ddbec8 2018-04-29 stsp const struct got_error *error;
710 80ddbec8 2018-04-29 stsp struct got_repository *repo;
711 899d86c2 2018-05-10 stsp struct got_object_id *start_id = NULL;
712 80ddbec8 2018-04-29 stsp char *repo_path = NULL;
713 80ddbec8 2018-04-29 stsp char *start_commit = NULL;
714 80ddbec8 2018-04-29 stsp int ch;
715 80ddbec8 2018-04-29 stsp
716 80ddbec8 2018-04-29 stsp #ifndef PROFILE
717 80ddbec8 2018-04-29 stsp if (pledge("stdio rpath wpath cpath flock proc tty", NULL) == -1)
718 80ddbec8 2018-04-29 stsp err(1, "pledge");
719 80ddbec8 2018-04-29 stsp #endif
720 80ddbec8 2018-04-29 stsp
721 80ddbec8 2018-04-29 stsp while ((ch = getopt(argc, argv, "c:")) != -1) {
722 80ddbec8 2018-04-29 stsp switch (ch) {
723 80ddbec8 2018-04-29 stsp case 'c':
724 80ddbec8 2018-04-29 stsp start_commit = optarg;
725 80ddbec8 2018-04-29 stsp break;
726 80ddbec8 2018-04-29 stsp default:
727 80ddbec8 2018-04-29 stsp usage();
728 80ddbec8 2018-04-29 stsp /* NOTREACHED */
729 80ddbec8 2018-04-29 stsp }
730 80ddbec8 2018-04-29 stsp }
731 80ddbec8 2018-04-29 stsp
732 80ddbec8 2018-04-29 stsp argc -= optind;
733 80ddbec8 2018-04-29 stsp argv += optind;
734 80ddbec8 2018-04-29 stsp
735 80ddbec8 2018-04-29 stsp if (argc == 0) {
736 80ddbec8 2018-04-29 stsp repo_path = getcwd(NULL, 0);
737 80ddbec8 2018-04-29 stsp if (repo_path == NULL)
738 80ddbec8 2018-04-29 stsp return got_error_from_errno();
739 80ddbec8 2018-04-29 stsp } else if (argc == 1) {
740 80ddbec8 2018-04-29 stsp repo_path = realpath(argv[0], NULL);
741 80ddbec8 2018-04-29 stsp if (repo_path == NULL)
742 80ddbec8 2018-04-29 stsp return got_error_from_errno();
743 80ddbec8 2018-04-29 stsp } else
744 80ddbec8 2018-04-29 stsp usage_log();
745 80ddbec8 2018-04-29 stsp
746 80ddbec8 2018-04-29 stsp error = got_repo_open(&repo, repo_path);
747 80ddbec8 2018-04-29 stsp free(repo_path);
748 80ddbec8 2018-04-29 stsp if (error != NULL)
749 80ddbec8 2018-04-29 stsp return error;
750 80ddbec8 2018-04-29 stsp
751 80ddbec8 2018-04-29 stsp if (start_commit == NULL) {
752 899d86c2 2018-05-10 stsp error = get_head_commit_id(&start_id, repo);
753 80ddbec8 2018-04-29 stsp if (error != NULL)
754 80ddbec8 2018-04-29 stsp return error;
755 80ddbec8 2018-04-29 stsp } else {
756 80ddbec8 2018-04-29 stsp struct got_object *obj;
757 80ddbec8 2018-04-29 stsp error = got_object_open_by_id_str(&obj, repo, start_commit);
758 80ddbec8 2018-04-29 stsp if (error == NULL) {
759 899d86c2 2018-05-10 stsp start_id = got_object_get_id(obj);
760 899d86c2 2018-05-10 stsp if (start_id == NULL)
761 80ddbec8 2018-04-29 stsp error = got_error_from_errno();
762 80ddbec8 2018-04-29 stsp }
763 80ddbec8 2018-04-29 stsp }
764 80ddbec8 2018-04-29 stsp if (error != NULL)
765 80ddbec8 2018-04-29 stsp return error;
766 899d86c2 2018-05-10 stsp error = show_log_view(start_id, repo);
767 899d86c2 2018-05-10 stsp free(start_id);
768 80ddbec8 2018-04-29 stsp got_repo_close(repo);
769 80ddbec8 2018-04-29 stsp return error;
770 9f7d7167 2018-04-29 stsp }
771 9f7d7167 2018-04-29 stsp
772 9f7d7167 2018-04-29 stsp __dead void
773 9f7d7167 2018-04-29 stsp usage_diff(void)
774 9f7d7167 2018-04-29 stsp {
775 80ddbec8 2018-04-29 stsp endwin();
776 9f7d7167 2018-04-29 stsp fprintf(stderr, "usage: %s diff [repository-path] object1 object2\n",
777 9f7d7167 2018-04-29 stsp getprogname());
778 9f7d7167 2018-04-29 stsp exit(1);
779 26ed57b2 2018-05-19 stsp }
780 26ed57b2 2018-05-19 stsp
781 26ed57b2 2018-05-19 stsp static const struct got_error *
782 26ed57b2 2018-05-19 stsp diff_blobs(FILE *outfile, struct got_object *obj1, struct got_object *obj2,
783 26ed57b2 2018-05-19 stsp struct got_repository *repo)
784 26ed57b2 2018-05-19 stsp {
785 26ed57b2 2018-05-19 stsp const struct got_error *err;
786 26ed57b2 2018-05-19 stsp struct got_blob_object *blob1 = NULL, *blob2 = NULL;
787 26ed57b2 2018-05-19 stsp
788 26ed57b2 2018-05-19 stsp err = got_object_blob_open(&blob1, repo, obj1, 8192);
789 26ed57b2 2018-05-19 stsp if (err)
790 26ed57b2 2018-05-19 stsp goto done;
791 26ed57b2 2018-05-19 stsp err = got_object_blob_open(&blob2, repo, obj2, 8192);
792 26ed57b2 2018-05-19 stsp if (err)
793 26ed57b2 2018-05-19 stsp goto done;
794 26ed57b2 2018-05-19 stsp
795 26ed57b2 2018-05-19 stsp err = got_diff_blob(blob1, blob2, NULL, NULL, outfile);
796 26ed57b2 2018-05-19 stsp done:
797 26ed57b2 2018-05-19 stsp if (blob1)
798 26ed57b2 2018-05-19 stsp got_object_blob_close(blob1);
799 26ed57b2 2018-05-19 stsp if (blob2)
800 26ed57b2 2018-05-19 stsp got_object_blob_close(blob2);
801 26ed57b2 2018-05-19 stsp return err;
802 26ed57b2 2018-05-19 stsp }
803 26ed57b2 2018-05-19 stsp
804 26ed57b2 2018-05-19 stsp static const struct got_error *
805 26ed57b2 2018-05-19 stsp diff_trees(FILE *outfile, struct got_object *obj1, struct got_object *obj2,
806 26ed57b2 2018-05-19 stsp struct got_repository *repo)
807 26ed57b2 2018-05-19 stsp {
808 26ed57b2 2018-05-19 stsp const struct got_error *err;
809 26ed57b2 2018-05-19 stsp struct got_tree_object *tree1 = NULL, *tree2 = NULL;
810 26ed57b2 2018-05-19 stsp
811 26ed57b2 2018-05-19 stsp err = got_object_tree_open(&tree1, repo, obj1);
812 26ed57b2 2018-05-19 stsp if (err)
813 26ed57b2 2018-05-19 stsp goto done;
814 26ed57b2 2018-05-19 stsp err = got_object_tree_open(&tree2, repo, obj2);
815 26ed57b2 2018-05-19 stsp if (err)
816 26ed57b2 2018-05-19 stsp goto done;
817 26ed57b2 2018-05-19 stsp
818 26ed57b2 2018-05-19 stsp err = got_diff_tree(tree1, tree2, repo, outfile);
819 26ed57b2 2018-05-19 stsp done:
820 26ed57b2 2018-05-19 stsp if (tree1)
821 26ed57b2 2018-05-19 stsp got_object_tree_close(tree1);
822 26ed57b2 2018-05-19 stsp if (tree2)
823 26ed57b2 2018-05-19 stsp got_object_tree_close(tree2);
824 26ed57b2 2018-05-19 stsp return err;
825 26ed57b2 2018-05-19 stsp }
826 26ed57b2 2018-05-19 stsp
827 26ed57b2 2018-05-19 stsp static const struct got_error *
828 26ed57b2 2018-05-19 stsp diff_commits(FILE *outfile, struct got_object *obj1, struct got_object *obj2,
829 26ed57b2 2018-05-19 stsp struct got_repository *repo)
830 26ed57b2 2018-05-19 stsp {
831 26ed57b2 2018-05-19 stsp const struct got_error *err;
832 26ed57b2 2018-05-19 stsp struct got_commit_object *commit1 = NULL, *commit2 = NULL;
833 26ed57b2 2018-05-19 stsp struct got_object *tree_obj1 = NULL, *tree_obj2 = NULL;
834 26ed57b2 2018-05-19 stsp
835 26ed57b2 2018-05-19 stsp err = got_object_commit_open(&commit1, repo, obj1);
836 26ed57b2 2018-05-19 stsp if (err)
837 26ed57b2 2018-05-19 stsp goto done;
838 26ed57b2 2018-05-19 stsp err = got_object_commit_open(&commit2, repo, obj2);
839 26ed57b2 2018-05-19 stsp if (err)
840 26ed57b2 2018-05-19 stsp goto done;
841 26ed57b2 2018-05-19 stsp
842 26ed57b2 2018-05-19 stsp err = got_object_open(&tree_obj1, repo, commit1->tree_id);
843 26ed57b2 2018-05-19 stsp if (err)
844 26ed57b2 2018-05-19 stsp goto done;
845 26ed57b2 2018-05-19 stsp err = got_object_open(&tree_obj2, repo, commit2->tree_id);
846 26ed57b2 2018-05-19 stsp if (err)
847 26ed57b2 2018-05-19 stsp goto done;
848 26ed57b2 2018-05-19 stsp
849 26ed57b2 2018-05-19 stsp err = diff_trees(outfile, tree_obj1, tree_obj2, repo);
850 26ed57b2 2018-05-19 stsp done:
851 26ed57b2 2018-05-19 stsp if (tree_obj1)
852 26ed57b2 2018-05-19 stsp got_object_close(tree_obj1);
853 26ed57b2 2018-05-19 stsp if (tree_obj2)
854 26ed57b2 2018-05-19 stsp got_object_close(tree_obj2);
855 26ed57b2 2018-05-19 stsp if (commit1)
856 26ed57b2 2018-05-19 stsp got_object_commit_close(commit1);
857 26ed57b2 2018-05-19 stsp if (commit2)
858 26ed57b2 2018-05-19 stsp got_object_commit_close(commit2);
859 26ed57b2 2018-05-19 stsp return err;
860 26ed57b2 2018-05-19 stsp
861 26ed57b2 2018-05-19 stsp }
862 26ed57b2 2018-05-19 stsp
863 26ed57b2 2018-05-19 stsp const struct got_error *
864 26ed57b2 2018-05-19 stsp draw_diff(FILE *f, int *first_displayed_line, int *last_displayed_line,
865 26ed57b2 2018-05-19 stsp int *eof, int max_lines)
866 26ed57b2 2018-05-19 stsp {
867 26ed57b2 2018-05-19 stsp int nlines = 0, nprinted = 0;
868 26ed57b2 2018-05-19 stsp
869 26ed57b2 2018-05-19 stsp rewind(f);
870 26ed57b2 2018-05-19 stsp wclear(tog_diff_view.window);
871 26ed57b2 2018-05-19 stsp
872 26ed57b2 2018-05-19 stsp *eof = 0;
873 26ed57b2 2018-05-19 stsp while (nprinted < max_lines) {
874 26ed57b2 2018-05-19 stsp char *line;
875 26ed57b2 2018-05-19 stsp size_t lineno;
876 26ed57b2 2018-05-19 stsp size_t linelen;
877 26ed57b2 2018-05-19 stsp const char delim[3] = { '\0', '\0', '\0'};
878 26ed57b2 2018-05-19 stsp
879 26ed57b2 2018-05-19 stsp line = fparseln(f, &linelen, &lineno, delim, 0);
880 26ed57b2 2018-05-19 stsp if (line == NULL) {
881 26ed57b2 2018-05-19 stsp *eof = 1;
882 26ed57b2 2018-05-19 stsp break;
883 26ed57b2 2018-05-19 stsp }
884 26ed57b2 2018-05-19 stsp if (++nlines < *first_displayed_line) {
885 26ed57b2 2018-05-19 stsp free(line);
886 26ed57b2 2018-05-19 stsp continue;
887 26ed57b2 2018-05-19 stsp }
888 26ed57b2 2018-05-19 stsp
889 26ed57b2 2018-05-19 stsp if (linelen > COLS - 1)
890 26ed57b2 2018-05-19 stsp line[COLS - 1] = '\0';
891 26ed57b2 2018-05-19 stsp waddstr(tog_diff_view.window, line);
892 26ed57b2 2018-05-19 stsp waddch(tog_diff_view.window, '\n');
893 26ed57b2 2018-05-19 stsp if (++nprinted == 1)
894 26ed57b2 2018-05-19 stsp *first_displayed_line = nlines;
895 26ed57b2 2018-05-19 stsp free(line);
896 26ed57b2 2018-05-19 stsp }
897 26ed57b2 2018-05-19 stsp *last_displayed_line = nlines;
898 26ed57b2 2018-05-19 stsp
899 26ed57b2 2018-05-19 stsp update_panels();
900 26ed57b2 2018-05-19 stsp doupdate();
901 26ed57b2 2018-05-19 stsp
902 26ed57b2 2018-05-19 stsp return NULL;
903 9f7d7167 2018-04-29 stsp }
904 9f7d7167 2018-04-29 stsp
905 9f7d7167 2018-04-29 stsp const struct got_error *
906 26ed57b2 2018-05-19 stsp show_diff_view(struct got_object *obj1, struct got_object *obj2,
907 26ed57b2 2018-05-19 stsp struct got_repository *repo)
908 26ed57b2 2018-05-19 stsp {
909 26ed57b2 2018-05-19 stsp const struct got_error *err;
910 26ed57b2 2018-05-19 stsp FILE *f;
911 26ed57b2 2018-05-19 stsp int ch, done = 0, first_displayed_line = 1, last_displayed_line = LINES;
912 26ed57b2 2018-05-19 stsp int eof;
913 26ed57b2 2018-05-19 stsp
914 26ed57b2 2018-05-19 stsp if (got_object_get_type(obj1) != got_object_get_type(obj2))
915 26ed57b2 2018-05-19 stsp return got_error(GOT_ERR_OBJ_TYPE);
916 26ed57b2 2018-05-19 stsp
917 26ed57b2 2018-05-19 stsp f = tog_opentemp();
918 26ed57b2 2018-05-19 stsp if (f == NULL)
919 26ed57b2 2018-05-19 stsp return got_error_from_errno();
920 26ed57b2 2018-05-19 stsp
921 26ed57b2 2018-05-19 stsp switch (got_object_get_type(obj1)) {
922 26ed57b2 2018-05-19 stsp case GOT_OBJ_TYPE_BLOB:
923 26ed57b2 2018-05-19 stsp err = diff_blobs(f, obj1, obj2, repo);
924 26ed57b2 2018-05-19 stsp break;
925 26ed57b2 2018-05-19 stsp case GOT_OBJ_TYPE_TREE:
926 26ed57b2 2018-05-19 stsp err = diff_trees(f, obj1, obj2, repo);
927 26ed57b2 2018-05-19 stsp break;
928 26ed57b2 2018-05-19 stsp case GOT_OBJ_TYPE_COMMIT:
929 26ed57b2 2018-05-19 stsp err = diff_commits(f, obj1, obj2, repo);
930 26ed57b2 2018-05-19 stsp break;
931 26ed57b2 2018-05-19 stsp default:
932 26ed57b2 2018-05-19 stsp return got_error(GOT_ERR_OBJ_TYPE);
933 26ed57b2 2018-05-19 stsp }
934 26ed57b2 2018-05-19 stsp
935 26ed57b2 2018-05-19 stsp fflush(f);
936 26ed57b2 2018-05-19 stsp
937 26ed57b2 2018-05-19 stsp if (tog_diff_view.window == NULL) {
938 26ed57b2 2018-05-19 stsp tog_diff_view.window = newwin(0, 0, 0, 0);
939 26ed57b2 2018-05-19 stsp if (tog_diff_view.window == NULL)
940 26ed57b2 2018-05-19 stsp return got_error_from_errno();
941 26ed57b2 2018-05-19 stsp keypad(tog_diff_view.window, TRUE);
942 26ed57b2 2018-05-19 stsp }
943 26ed57b2 2018-05-19 stsp if (tog_diff_view.panel == NULL) {
944 26ed57b2 2018-05-19 stsp tog_diff_view.panel = new_panel(tog_diff_view.window);
945 26ed57b2 2018-05-19 stsp if (tog_diff_view.panel == NULL)
946 26ed57b2 2018-05-19 stsp return got_error_from_errno();
947 26ed57b2 2018-05-19 stsp } else
948 26ed57b2 2018-05-19 stsp show_panel(tog_diff_view.panel);
949 26ed57b2 2018-05-19 stsp
950 26ed57b2 2018-05-19 stsp while (!done) {
951 26ed57b2 2018-05-19 stsp err = draw_diff(f, &first_displayed_line, &last_displayed_line,
952 26ed57b2 2018-05-19 stsp &eof, LINES);
953 26ed57b2 2018-05-19 stsp if (err)
954 26ed57b2 2018-05-19 stsp break;
955 26ed57b2 2018-05-19 stsp nodelay(stdscr, FALSE);
956 26ed57b2 2018-05-19 stsp ch = wgetch(tog_diff_view.window);
957 26ed57b2 2018-05-19 stsp switch (ch) {
958 26ed57b2 2018-05-19 stsp case 'q':
959 26ed57b2 2018-05-19 stsp done = 1;
960 26ed57b2 2018-05-19 stsp break;
961 26ed57b2 2018-05-19 stsp case 'k':
962 26ed57b2 2018-05-19 stsp case KEY_UP:
963 26ed57b2 2018-05-19 stsp if (first_displayed_line > 1)
964 26ed57b2 2018-05-19 stsp first_displayed_line--;
965 26ed57b2 2018-05-19 stsp break;
966 26ed57b2 2018-05-19 stsp case 'j':
967 26ed57b2 2018-05-19 stsp case KEY_DOWN:
968 26ed57b2 2018-05-19 stsp if (!eof)
969 26ed57b2 2018-05-19 stsp first_displayed_line++;
970 26ed57b2 2018-05-19 stsp break;
971 26ed57b2 2018-05-19 stsp default:
972 26ed57b2 2018-05-19 stsp break;
973 26ed57b2 2018-05-19 stsp }
974 26ed57b2 2018-05-19 stsp nodelay(stdscr, TRUE);
975 26ed57b2 2018-05-19 stsp }
976 26ed57b2 2018-05-19 stsp fclose(f);
977 26ed57b2 2018-05-19 stsp return err;
978 26ed57b2 2018-05-19 stsp }
979 26ed57b2 2018-05-19 stsp
980 26ed57b2 2018-05-19 stsp const struct got_error *
981 9f7d7167 2018-04-29 stsp cmd_diff(int argc, char *argv[])
982 9f7d7167 2018-04-29 stsp {
983 26ed57b2 2018-05-19 stsp const struct got_error *error = NULL;
984 26ed57b2 2018-05-19 stsp struct got_repository *repo = NULL;
985 26ed57b2 2018-05-19 stsp struct got_object *obj1 = NULL, *obj2 = NULL;
986 26ed57b2 2018-05-19 stsp char *repo_path = NULL;
987 26ed57b2 2018-05-19 stsp char *obj_id_str1 = NULL, *obj_id_str2 = NULL;
988 26ed57b2 2018-05-19 stsp int ch;
989 26ed57b2 2018-05-19 stsp
990 26ed57b2 2018-05-19 stsp #ifndef PROFILE
991 26ed57b2 2018-05-19 stsp if (pledge("stdio rpath wpath cpath flock proc tty", NULL) == -1)
992 26ed57b2 2018-05-19 stsp err(1, "pledge");
993 26ed57b2 2018-05-19 stsp #endif
994 26ed57b2 2018-05-19 stsp
995 26ed57b2 2018-05-19 stsp while ((ch = getopt(argc, argv, "")) != -1) {
996 26ed57b2 2018-05-19 stsp switch (ch) {
997 26ed57b2 2018-05-19 stsp default:
998 26ed57b2 2018-05-19 stsp usage();
999 26ed57b2 2018-05-19 stsp /* NOTREACHED */
1000 26ed57b2 2018-05-19 stsp }
1001 26ed57b2 2018-05-19 stsp }
1002 26ed57b2 2018-05-19 stsp
1003 26ed57b2 2018-05-19 stsp argc -= optind;
1004 26ed57b2 2018-05-19 stsp argv += optind;
1005 26ed57b2 2018-05-19 stsp
1006 26ed57b2 2018-05-19 stsp if (argc == 0) {
1007 26ed57b2 2018-05-19 stsp usage_diff(); /* TODO show local worktree changes */
1008 26ed57b2 2018-05-19 stsp } else if (argc == 2) {
1009 26ed57b2 2018-05-19 stsp repo_path = getcwd(NULL, 0);
1010 26ed57b2 2018-05-19 stsp if (repo_path == NULL)
1011 26ed57b2 2018-05-19 stsp return got_error_from_errno();
1012 26ed57b2 2018-05-19 stsp obj_id_str1 = argv[0];
1013 26ed57b2 2018-05-19 stsp obj_id_str2 = argv[1];
1014 26ed57b2 2018-05-19 stsp } else if (argc == 3) {
1015 26ed57b2 2018-05-19 stsp repo_path = realpath(argv[0], NULL);
1016 26ed57b2 2018-05-19 stsp if (repo_path == NULL)
1017 26ed57b2 2018-05-19 stsp return got_error_from_errno();
1018 26ed57b2 2018-05-19 stsp obj_id_str1 = argv[1];
1019 26ed57b2 2018-05-19 stsp obj_id_str2 = argv[2];
1020 26ed57b2 2018-05-19 stsp } else
1021 26ed57b2 2018-05-19 stsp usage_diff();
1022 26ed57b2 2018-05-19 stsp
1023 26ed57b2 2018-05-19 stsp error = got_repo_open(&repo, repo_path);
1024 26ed57b2 2018-05-19 stsp free(repo_path);
1025 26ed57b2 2018-05-19 stsp if (error)
1026 26ed57b2 2018-05-19 stsp goto done;
1027 26ed57b2 2018-05-19 stsp
1028 26ed57b2 2018-05-19 stsp error = got_object_open_by_id_str(&obj1, repo, obj_id_str1);
1029 26ed57b2 2018-05-19 stsp if (error)
1030 26ed57b2 2018-05-19 stsp goto done;
1031 26ed57b2 2018-05-19 stsp
1032 26ed57b2 2018-05-19 stsp error = got_object_open_by_id_str(&obj2, repo, obj_id_str2);
1033 26ed57b2 2018-05-19 stsp if (error)
1034 26ed57b2 2018-05-19 stsp goto done;
1035 26ed57b2 2018-05-19 stsp
1036 26ed57b2 2018-05-19 stsp error = show_diff_view(obj1, obj2, repo);
1037 26ed57b2 2018-05-19 stsp done:
1038 26ed57b2 2018-05-19 stsp got_repo_close(repo);
1039 26ed57b2 2018-05-19 stsp if (obj1)
1040 26ed57b2 2018-05-19 stsp got_object_close(obj1);
1041 26ed57b2 2018-05-19 stsp if (obj2)
1042 26ed57b2 2018-05-19 stsp got_object_close(obj2);
1043 26ed57b2 2018-05-19 stsp return error;
1044 9f7d7167 2018-04-29 stsp }
1045 9f7d7167 2018-04-29 stsp
1046 9f7d7167 2018-04-29 stsp __dead void
1047 9f7d7167 2018-04-29 stsp usage_blame(void)
1048 9f7d7167 2018-04-29 stsp {
1049 80ddbec8 2018-04-29 stsp endwin();
1050 9f7d7167 2018-04-29 stsp fprintf(stderr, "usage: %s blame [repository-path] blob-object\n",
1051 9f7d7167 2018-04-29 stsp getprogname());
1052 9f7d7167 2018-04-29 stsp exit(1);
1053 9f7d7167 2018-04-29 stsp }
1054 9f7d7167 2018-04-29 stsp
1055 9f7d7167 2018-04-29 stsp const struct got_error *
1056 9f7d7167 2018-04-29 stsp cmd_blame(int argc, char *argv[])
1057 9f7d7167 2018-04-29 stsp {
1058 9f7d7167 2018-04-29 stsp return got_error(GOT_ERR_NOT_IMPL);
1059 9f7d7167 2018-04-29 stsp }
1060 9f7d7167 2018-04-29 stsp
1061 9f7d7167 2018-04-29 stsp static const struct got_error *
1062 9f7d7167 2018-04-29 stsp init_curses(void)
1063 9f7d7167 2018-04-29 stsp {
1064 9f7d7167 2018-04-29 stsp initscr();
1065 9f7d7167 2018-04-29 stsp cbreak();
1066 9f7d7167 2018-04-29 stsp noecho();
1067 9f7d7167 2018-04-29 stsp nonl();
1068 9f7d7167 2018-04-29 stsp intrflush(stdscr, FALSE);
1069 9f7d7167 2018-04-29 stsp keypad(stdscr, TRUE);
1070 1f475ad8 2018-05-10 stsp curs_set(0);
1071 9f7d7167 2018-04-29 stsp
1072 9f7d7167 2018-04-29 stsp tog_main_win = newwin(0, 0, 0, 0);
1073 9f7d7167 2018-04-29 stsp if (tog_main_win == NULL)
1074 9f7d7167 2018-04-29 stsp return got_error_from_errno();
1075 9f7d7167 2018-04-29 stsp tog_main_panel = new_panel(tog_main_win);
1076 9f7d7167 2018-04-29 stsp if (tog_main_panel == NULL)
1077 9f7d7167 2018-04-29 stsp return got_error_from_errno();
1078 9f7d7167 2018-04-29 stsp
1079 9f7d7167 2018-04-29 stsp return NULL;
1080 9f7d7167 2018-04-29 stsp }
1081 9f7d7167 2018-04-29 stsp
1082 9f7d7167 2018-04-29 stsp __dead void
1083 9f7d7167 2018-04-29 stsp usage(void)
1084 9f7d7167 2018-04-29 stsp {
1085 9f7d7167 2018-04-29 stsp int i;
1086 9f7d7167 2018-04-29 stsp
1087 c2301be8 2018-04-30 stsp fprintf(stderr, "usage: %s [-h] [command] [arg ...]\n\n"
1088 9f7d7167 2018-04-29 stsp "Available commands:\n", getprogname());
1089 9f7d7167 2018-04-29 stsp for (i = 0; i < nitems(tog_commands); i++) {
1090 9f7d7167 2018-04-29 stsp struct tog_cmd *cmd = &tog_commands[i];
1091 c2301be8 2018-04-30 stsp fprintf(stderr, " %s: %s\n", cmd->name, cmd->descr);
1092 9f7d7167 2018-04-29 stsp }
1093 9f7d7167 2018-04-29 stsp exit(1);
1094 9f7d7167 2018-04-29 stsp }
1095 9f7d7167 2018-04-29 stsp
1096 c2301be8 2018-04-30 stsp static char **
1097 c2301be8 2018-04-30 stsp make_argv(const char *arg0, const char *arg1)
1098 c2301be8 2018-04-30 stsp {
1099 c2301be8 2018-04-30 stsp char **argv;
1100 c2301be8 2018-04-30 stsp int argc = (arg1 == NULL ? 1 : 2);
1101 c2301be8 2018-04-30 stsp
1102 c2301be8 2018-04-30 stsp argv = calloc(argc, sizeof(char *));
1103 c2301be8 2018-04-30 stsp if (argv == NULL)
1104 c2301be8 2018-04-30 stsp err(1, "calloc");
1105 c2301be8 2018-04-30 stsp argv[0] = strdup(arg0);
1106 c2301be8 2018-04-30 stsp if (argv[0] == NULL)
1107 c2301be8 2018-04-30 stsp err(1, "calloc");
1108 c2301be8 2018-04-30 stsp if (arg1) {
1109 c2301be8 2018-04-30 stsp argv[1] = strdup(arg1);
1110 c2301be8 2018-04-30 stsp if (argv[1] == NULL)
1111 c2301be8 2018-04-30 stsp err(1, "calloc");
1112 c2301be8 2018-04-30 stsp }
1113 c2301be8 2018-04-30 stsp
1114 c2301be8 2018-04-30 stsp return argv;
1115 c2301be8 2018-04-30 stsp }
1116 c2301be8 2018-04-30 stsp
1117 9f7d7167 2018-04-29 stsp int
1118 9f7d7167 2018-04-29 stsp main(int argc, char *argv[])
1119 9f7d7167 2018-04-29 stsp {
1120 9f7d7167 2018-04-29 stsp const struct got_error *error = NULL;
1121 9f7d7167 2018-04-29 stsp struct tog_cmd *cmd = NULL;
1122 9f7d7167 2018-04-29 stsp int ch, hflag = 0;
1123 c2301be8 2018-04-30 stsp char **cmd_argv = NULL;
1124 9f7d7167 2018-04-29 stsp
1125 9f7d7167 2018-04-29 stsp setlocale(LC_ALL, "");
1126 9f7d7167 2018-04-29 stsp
1127 9f7d7167 2018-04-29 stsp while ((ch = getopt(argc, argv, "h")) != -1) {
1128 9f7d7167 2018-04-29 stsp switch (ch) {
1129 9f7d7167 2018-04-29 stsp case 'h':
1130 9f7d7167 2018-04-29 stsp hflag = 1;
1131 9f7d7167 2018-04-29 stsp break;
1132 9f7d7167 2018-04-29 stsp default:
1133 9f7d7167 2018-04-29 stsp usage();
1134 9f7d7167 2018-04-29 stsp /* NOTREACHED */
1135 9f7d7167 2018-04-29 stsp }
1136 9f7d7167 2018-04-29 stsp }
1137 9f7d7167 2018-04-29 stsp
1138 9f7d7167 2018-04-29 stsp argc -= optind;
1139 9f7d7167 2018-04-29 stsp argv += optind;
1140 9f7d7167 2018-04-29 stsp optind = 0;
1141 c2301be8 2018-04-30 stsp optreset = 1;
1142 9f7d7167 2018-04-29 stsp
1143 c2301be8 2018-04-30 stsp if (argc == 0) {
1144 c2301be8 2018-04-30 stsp /* Build an argument vector which runs a default command. */
1145 9f7d7167 2018-04-29 stsp cmd = &tog_commands[0];
1146 c2301be8 2018-04-30 stsp cmd_argv = make_argv(cmd->name, NULL);
1147 c2301be8 2018-04-30 stsp argc = 1;
1148 c2301be8 2018-04-30 stsp } else {
1149 9f7d7167 2018-04-29 stsp int i;
1150 9f7d7167 2018-04-29 stsp
1151 c2301be8 2018-04-30 stsp /* Did the user specific a command? */
1152 9f7d7167 2018-04-29 stsp for (i = 0; i < nitems(tog_commands); i++) {
1153 c2301be8 2018-04-30 stsp if (strncmp(tog_commands[i].name, argv[0],
1154 9f7d7167 2018-04-29 stsp strlen(argv[0])) == 0) {
1155 9f7d7167 2018-04-29 stsp cmd = &tog_commands[i];
1156 9f7d7167 2018-04-29 stsp if (hflag)
1157 9f7d7167 2018-04-29 stsp tog_commands[i].cmd_usage();
1158 9f7d7167 2018-04-29 stsp break;
1159 9f7d7167 2018-04-29 stsp }
1160 9f7d7167 2018-04-29 stsp }
1161 9f7d7167 2018-04-29 stsp if (cmd == NULL) {
1162 c2301be8 2018-04-30 stsp /* Did the user specify a repository? */
1163 c2301be8 2018-04-30 stsp char *repo_path = realpath(argv[0], NULL);
1164 c2301be8 2018-04-30 stsp if (repo_path) {
1165 c2301be8 2018-04-30 stsp struct got_repository *repo;
1166 c2301be8 2018-04-30 stsp error = got_repo_open(&repo, repo_path);
1167 c2301be8 2018-04-30 stsp if (error == NULL)
1168 c2301be8 2018-04-30 stsp got_repo_close(repo);
1169 c2301be8 2018-04-30 stsp } else
1170 ad7de8d9 2018-04-30 stsp error = got_error_from_errno();
1171 c2301be8 2018-04-30 stsp if (error) {
1172 ad7de8d9 2018-04-30 stsp fprintf(stderr, "%s: '%s' is neither a known "
1173 ad7de8d9 2018-04-30 stsp "command nor a path to a repository\n",
1174 c2301be8 2018-04-30 stsp getprogname(), argv[0]);
1175 ad7de8d9 2018-04-30 stsp free(repo_path);
1176 c2301be8 2018-04-30 stsp return 1;
1177 c2301be8 2018-04-30 stsp }
1178 c2301be8 2018-04-30 stsp cmd = &tog_commands[0];
1179 c2301be8 2018-04-30 stsp cmd_argv = make_argv(cmd->name, repo_path);
1180 c2301be8 2018-04-30 stsp argc = 2;
1181 c2301be8 2018-04-30 stsp free(repo_path);
1182 9f7d7167 2018-04-29 stsp }
1183 9f7d7167 2018-04-29 stsp }
1184 9f7d7167 2018-04-29 stsp
1185 9f7d7167 2018-04-29 stsp error = init_curses();
1186 9f7d7167 2018-04-29 stsp if (error) {
1187 80ddbec8 2018-04-29 stsp fprintf(stderr, "cannot initialize ncurses: %s\n", error->msg);
1188 9f7d7167 2018-04-29 stsp return 1;
1189 9f7d7167 2018-04-29 stsp }
1190 9f7d7167 2018-04-29 stsp
1191 c2301be8 2018-04-30 stsp error = cmd->cmd_main(argc, cmd_argv ? cmd_argv : argv);
1192 9f7d7167 2018-04-29 stsp if (error)
1193 9f7d7167 2018-04-29 stsp goto done;
1194 9f7d7167 2018-04-29 stsp done:
1195 9f7d7167 2018-04-29 stsp endwin();
1196 c2301be8 2018-04-30 stsp free(cmd_argv);
1197 9f7d7167 2018-04-29 stsp if (error)
1198 9f7d7167 2018-04-29 stsp fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
1199 9f7d7167 2018-04-29 stsp return 0;
1200 9f7d7167 2018-04-29 stsp }