Blame


1 5c860e29 2018-03-12 stsp /*
2 f42b1b34 2018-03-12 stsp * Copyright (c) 2017 Martin Pieuchot <mpi@openbsd.org>
3 5d56da81 2019-01-13 stsp * Copyright (c) 2018, 2019 Stefan Sperling <stsp@openbsd.org>
4 5c860e29 2018-03-12 stsp *
5 5c860e29 2018-03-12 stsp * Permission to use, copy, modify, and distribute this software for any
6 5c860e29 2018-03-12 stsp * purpose with or without fee is hereby granted, provided that the above
7 5c860e29 2018-03-12 stsp * copyright notice and this permission notice appear in all copies.
8 5c860e29 2018-03-12 stsp *
9 5c860e29 2018-03-12 stsp * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10 5c860e29 2018-03-12 stsp * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11 5c860e29 2018-03-12 stsp * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12 5c860e29 2018-03-12 stsp * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13 5c860e29 2018-03-12 stsp * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14 5c860e29 2018-03-12 stsp * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15 5c860e29 2018-03-12 stsp * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16 5c860e29 2018-03-12 stsp */
17 5c860e29 2018-03-12 stsp
18 f42b1b34 2018-03-12 stsp #include <sys/queue.h>
19 64a96a6d 2018-04-01 stsp #include <sys/limits.h>
20 c0768b0f 2018-06-10 stsp #include <sys/types.h>
21 5de5890b 2018-10-18 stsp #include <sys/stat.h>
22 f42b1b34 2018-03-12 stsp
23 5c860e29 2018-03-12 stsp #include <err.h>
24 5c860e29 2018-03-12 stsp #include <errno.h>
25 5c860e29 2018-03-12 stsp #include <locale.h>
26 99437157 2018-11-11 stsp #include <signal.h>
27 5c860e29 2018-03-12 stsp #include <stdio.h>
28 5c860e29 2018-03-12 stsp #include <stdlib.h>
29 5c860e29 2018-03-12 stsp #include <string.h>
30 5c860e29 2018-03-12 stsp #include <unistd.h>
31 c09a553d 2018-03-12 stsp #include <libgen.h>
32 c0768b0f 2018-06-10 stsp #include <time.h>
33 5c860e29 2018-03-12 stsp
34 f42b1b34 2018-03-12 stsp #include "got_error.h"
35 f42b1b34 2018-03-12 stsp #include "got_object.h"
36 5261c201 2018-04-01 stsp #include "got_reference.h"
37 f42b1b34 2018-03-12 stsp #include "got_repository.h"
38 c09a553d 2018-03-12 stsp #include "got_worktree.h"
39 79109fed 2018-03-27 stsp #include "got_diff.h"
40 372ccdbb 2018-06-10 stsp #include "got_commit_graph.h"
41 404c43c4 2018-06-21 stsp #include "got_blame.h"
42 63219cd2 2019-01-04 stsp #include "got_privsep.h"
43 5c860e29 2018-03-12 stsp
44 5c860e29 2018-03-12 stsp #ifndef nitems
45 5c860e29 2018-03-12 stsp #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
46 5c860e29 2018-03-12 stsp #endif
47 99437157 2018-11-11 stsp
48 99437157 2018-11-11 stsp static volatile sig_atomic_t sigint_received;
49 99437157 2018-11-11 stsp static volatile sig_atomic_t sigpipe_received;
50 99437157 2018-11-11 stsp
51 99437157 2018-11-11 stsp static void
52 99437157 2018-11-11 stsp catch_sigint(int signo)
53 99437157 2018-11-11 stsp {
54 99437157 2018-11-11 stsp sigint_received = 1;
55 99437157 2018-11-11 stsp }
56 99437157 2018-11-11 stsp
57 99437157 2018-11-11 stsp static void
58 99437157 2018-11-11 stsp catch_sigpipe(int signo)
59 99437157 2018-11-11 stsp {
60 99437157 2018-11-11 stsp sigpipe_received = 1;
61 99437157 2018-11-11 stsp }
62 5c860e29 2018-03-12 stsp
63 99437157 2018-11-11 stsp
64 5c860e29 2018-03-12 stsp struct cmd {
65 5c860e29 2018-03-12 stsp const char *cmd_name;
66 d7d4f210 2018-03-12 stsp const struct got_error *(*cmd_main)(int, char *[]);
67 1b6b95a8 2018-03-12 stsp void (*cmd_usage)(void);
68 46a0db7d 2018-03-12 stsp const char *cmd_descr;
69 5c860e29 2018-03-12 stsp };
70 5c860e29 2018-03-12 stsp
71 4ed7e80c 2018-05-20 stsp __dead static void usage(void);
72 4ed7e80c 2018-05-20 stsp __dead static void usage_checkout(void);
73 507dc3bb 2018-12-29 stsp __dead static void usage_update(void);
74 4ed7e80c 2018-05-20 stsp __dead static void usage_log(void);
75 4ed7e80c 2018-05-20 stsp __dead static void usage_diff(void);
76 404c43c4 2018-06-21 stsp __dead static void usage_blame(void);
77 5de5890b 2018-10-18 stsp __dead static void usage_tree(void);
78 6bad629b 2019-02-04 stsp __dead static void usage_status(void);
79 5c860e29 2018-03-12 stsp
80 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_checkout(int, char *[]);
81 507dc3bb 2018-12-29 stsp static const struct got_error* cmd_update(int, char *[]);
82 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_log(int, char *[]);
83 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_diff(int, char *[]);
84 404c43c4 2018-06-21 stsp static const struct got_error* cmd_blame(int, char *[]);
85 5de5890b 2018-10-18 stsp static const struct got_error* cmd_tree(int, char *[]);
86 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_status(int, char *[]);
87 5c860e29 2018-03-12 stsp
88 4ed7e80c 2018-05-20 stsp static struct cmd got_commands[] = {
89 c09a553d 2018-03-12 stsp { "checkout", cmd_checkout, usage_checkout,
90 0bb8a95e 2018-03-12 stsp "check out a new work tree from a repository" },
91 507dc3bb 2018-12-29 stsp { "update", cmd_update, usage_update,
92 507dc3bb 2018-12-29 stsp "update a work tree to a different commit" },
93 1b6b95a8 2018-03-12 stsp { "log", cmd_log, usage_log,
94 1b6b95a8 2018-03-12 stsp "show repository history" },
95 b00d56cd 2018-04-01 stsp { "diff", cmd_diff, usage_diff,
96 b00d56cd 2018-04-01 stsp "compare files and directories" },
97 404c43c4 2018-06-21 stsp { "blame", cmd_blame, usage_blame,
98 404c43c4 2018-06-21 stsp " show when lines in a file were changed" },
99 5de5890b 2018-10-18 stsp { "tree", cmd_tree, usage_tree,
100 5de5890b 2018-10-18 stsp " list files and directories in repository" },
101 1b6b95a8 2018-03-12 stsp { "status", cmd_status, usage_status,
102 1b6b95a8 2018-03-12 stsp "show modification status of files" },
103 5c860e29 2018-03-12 stsp };
104 5c860e29 2018-03-12 stsp
105 5c860e29 2018-03-12 stsp int
106 5c860e29 2018-03-12 stsp main(int argc, char *argv[])
107 5c860e29 2018-03-12 stsp {
108 5c860e29 2018-03-12 stsp struct cmd *cmd;
109 5c860e29 2018-03-12 stsp unsigned int i;
110 5c860e29 2018-03-12 stsp int ch;
111 1b6b95a8 2018-03-12 stsp int hflag = 0;
112 5c860e29 2018-03-12 stsp
113 289e3cbf 2019-02-04 stsp setlocale(LC_CTYPE, "");
114 5c860e29 2018-03-12 stsp
115 1b6b95a8 2018-03-12 stsp while ((ch = getopt(argc, argv, "h")) != -1) {
116 5c860e29 2018-03-12 stsp switch (ch) {
117 1b6b95a8 2018-03-12 stsp case 'h':
118 1b6b95a8 2018-03-12 stsp hflag = 1;
119 1b6b95a8 2018-03-12 stsp break;
120 5c860e29 2018-03-12 stsp default:
121 5c860e29 2018-03-12 stsp usage();
122 5c860e29 2018-03-12 stsp /* NOTREACHED */
123 5c860e29 2018-03-12 stsp }
124 5c860e29 2018-03-12 stsp }
125 5c860e29 2018-03-12 stsp
126 5c860e29 2018-03-12 stsp argc -= optind;
127 5c860e29 2018-03-12 stsp argv += optind;
128 1e70621d 2018-03-27 stsp optind = 0;
129 5c860e29 2018-03-12 stsp
130 5c860e29 2018-03-12 stsp if (argc <= 0)
131 5c860e29 2018-03-12 stsp usage();
132 5c860e29 2018-03-12 stsp
133 99437157 2018-11-11 stsp signal(SIGINT, catch_sigint);
134 99437157 2018-11-11 stsp signal(SIGPIPE, catch_sigpipe);
135 99437157 2018-11-11 stsp
136 5c860e29 2018-03-12 stsp for (i = 0; i < nitems(got_commands); i++) {
137 d7d4f210 2018-03-12 stsp const struct got_error *error;
138 d7d4f210 2018-03-12 stsp
139 5c860e29 2018-03-12 stsp cmd = &got_commands[i];
140 5c860e29 2018-03-12 stsp
141 5c860e29 2018-03-12 stsp if (strncmp(cmd->cmd_name, argv[0], strlen(argv[0])))
142 5c860e29 2018-03-12 stsp continue;
143 5c860e29 2018-03-12 stsp
144 1b6b95a8 2018-03-12 stsp if (hflag)
145 1b6b95a8 2018-03-12 stsp got_commands[i].cmd_usage();
146 1b6b95a8 2018-03-12 stsp
147 d7d4f210 2018-03-12 stsp error = got_commands[i].cmd_main(argc, argv);
148 80d5f134 2018-11-11 stsp if (error && !(sigint_received || sigpipe_received)) {
149 d7d4f210 2018-03-12 stsp fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
150 d7d4f210 2018-03-12 stsp return 1;
151 d7d4f210 2018-03-12 stsp }
152 d7d4f210 2018-03-12 stsp
153 d7d4f210 2018-03-12 stsp return 0;
154 5c860e29 2018-03-12 stsp }
155 5c860e29 2018-03-12 stsp
156 20ecf764 2018-03-12 stsp fprintf(stderr, "%s: unknown command '%s'\n", getprogname(), argv[0]);
157 5c860e29 2018-03-12 stsp return 1;
158 5c860e29 2018-03-12 stsp }
159 5c860e29 2018-03-12 stsp
160 4ed7e80c 2018-05-20 stsp __dead static void
161 5c860e29 2018-03-12 stsp usage(void)
162 5c860e29 2018-03-12 stsp {
163 46a0db7d 2018-03-12 stsp int i;
164 46a0db7d 2018-03-12 stsp
165 987e94ba 2018-03-12 stsp fprintf(stderr, "usage: %s [-h] command [arg ...]\n\n"
166 987e94ba 2018-03-12 stsp "Available commands:\n", getprogname());
167 46a0db7d 2018-03-12 stsp for (i = 0; i < nitems(got_commands); i++) {
168 46a0db7d 2018-03-12 stsp struct cmd *cmd = &got_commands[i];
169 46a0db7d 2018-03-12 stsp fprintf(stderr, " %s: %s\n", cmd->cmd_name, cmd->cmd_descr);
170 46a0db7d 2018-03-12 stsp }
171 5c860e29 2018-03-12 stsp exit(1);
172 5c860e29 2018-03-12 stsp }
173 5c860e29 2018-03-12 stsp
174 0266afb7 2019-01-04 stsp static const struct got_error *
175 0266afb7 2019-01-04 stsp apply_unveil(const char *repo_path, const char *worktree_path)
176 0266afb7 2019-01-04 stsp {
177 0266afb7 2019-01-04 stsp const struct got_error *error;
178 0266afb7 2019-01-04 stsp
179 0266afb7 2019-01-04 stsp if (repo_path && unveil(repo_path, "r") != 0)
180 0266afb7 2019-01-04 stsp return got_error_from_errno();
181 0266afb7 2019-01-04 stsp
182 0266afb7 2019-01-04 stsp if (worktree_path && unveil(worktree_path, "rwc") != 0)
183 0266afb7 2019-01-04 stsp return got_error_from_errno();
184 0266afb7 2019-01-04 stsp
185 f12d0dbe 2019-01-04 stsp if (unveil("/tmp", "rwc") != 0)
186 0266afb7 2019-01-04 stsp return got_error_from_errno();
187 0266afb7 2019-01-04 stsp
188 0266afb7 2019-01-04 stsp error = got_privsep_unveil_exec_helpers();
189 0266afb7 2019-01-04 stsp if (error != NULL)
190 0266afb7 2019-01-04 stsp return error;
191 0266afb7 2019-01-04 stsp
192 0266afb7 2019-01-04 stsp if (unveil(NULL, NULL) != 0)
193 0266afb7 2019-01-04 stsp return got_error_from_errno();
194 0266afb7 2019-01-04 stsp
195 0266afb7 2019-01-04 stsp return NULL;
196 0266afb7 2019-01-04 stsp }
197 0266afb7 2019-01-04 stsp
198 4ed7e80c 2018-05-20 stsp __dead static void
199 c09a553d 2018-03-12 stsp usage_checkout(void)
200 c09a553d 2018-03-12 stsp {
201 0bb8a95e 2018-03-12 stsp fprintf(stderr, "usage: %s checkout [-p prefix] repository-path "
202 0bb8a95e 2018-03-12 stsp "[worktree-path]\n", getprogname());
203 c09a553d 2018-03-12 stsp exit(1);
204 92a684f4 2018-03-12 stsp }
205 92a684f4 2018-03-12 stsp
206 92a684f4 2018-03-12 stsp static void
207 a0eb853d 2018-12-29 stsp checkout_progress(void *arg, unsigned char status, const char *path)
208 92a684f4 2018-03-12 stsp {
209 92a684f4 2018-03-12 stsp char *worktree_path = arg;
210 92a684f4 2018-03-12 stsp
211 92a684f4 2018-03-12 stsp while (path[0] == '/')
212 92a684f4 2018-03-12 stsp path++;
213 92a684f4 2018-03-12 stsp
214 d7b62c98 2018-12-27 stsp printf("%c %s/%s\n", status, worktree_path, path);
215 99437157 2018-11-11 stsp }
216 99437157 2018-11-11 stsp
217 99437157 2018-11-11 stsp static const struct got_error *
218 6bad629b 2019-02-04 stsp check_cancelled(void *arg)
219 99437157 2018-11-11 stsp {
220 99437157 2018-11-11 stsp if (sigint_received || sigpipe_received)
221 99437157 2018-11-11 stsp return got_error(GOT_ERR_CANCELLED);
222 99437157 2018-11-11 stsp return NULL;
223 8069f636 2019-01-12 stsp }
224 8069f636 2019-01-12 stsp
225 8069f636 2019-01-12 stsp static const struct got_error *
226 8069f636 2019-01-12 stsp check_ancestry(struct got_worktree *worktree, struct got_object_id *commit_id,
227 8069f636 2019-01-12 stsp struct got_repository *repo)
228 8069f636 2019-01-12 stsp {
229 8069f636 2019-01-12 stsp const struct got_error *err;
230 8069f636 2019-01-12 stsp struct got_reference *head_ref = NULL;
231 8069f636 2019-01-12 stsp struct got_object_id *head_commit_id = NULL;
232 8069f636 2019-01-12 stsp struct got_commit_graph *graph = NULL;
233 8069f636 2019-01-12 stsp
234 8069f636 2019-01-12 stsp head_ref = got_worktree_get_head_ref(worktree);
235 8069f636 2019-01-12 stsp if (head_ref == NULL)
236 8069f636 2019-01-12 stsp return got_error_from_errno();
237 8069f636 2019-01-12 stsp
238 8069f636 2019-01-12 stsp /* TODO: Check the reflog. The head ref may have been rebased. */
239 8069f636 2019-01-12 stsp err = got_ref_resolve(&head_commit_id, repo, head_ref);
240 8069f636 2019-01-12 stsp if (err)
241 8069f636 2019-01-12 stsp goto done;
242 8069f636 2019-01-12 stsp
243 8069f636 2019-01-12 stsp err = got_commit_graph_open(&graph, head_commit_id, "/", 1, repo);
244 8069f636 2019-01-12 stsp if (err)
245 8069f636 2019-01-12 stsp goto done;
246 8069f636 2019-01-12 stsp
247 8069f636 2019-01-12 stsp err = got_commit_graph_iter_start(graph, head_commit_id, repo);
248 8069f636 2019-01-12 stsp if (err)
249 8069f636 2019-01-12 stsp goto done;
250 8069f636 2019-01-12 stsp while (1) {
251 8069f636 2019-01-12 stsp struct got_object_id *id;
252 8069f636 2019-01-12 stsp
253 8069f636 2019-01-12 stsp if (sigint_received || sigpipe_received)
254 8069f636 2019-01-12 stsp break;
255 8069f636 2019-01-12 stsp
256 8069f636 2019-01-12 stsp err = got_commit_graph_iter_next(&id, graph);
257 8069f636 2019-01-12 stsp if (err) {
258 8069f636 2019-01-12 stsp if (err->code == GOT_ERR_ITER_COMPLETED) {
259 8069f636 2019-01-12 stsp err = got_error(GOT_ERR_ANCESTRY);
260 8069f636 2019-01-12 stsp break;
261 8069f636 2019-01-12 stsp }
262 8069f636 2019-01-12 stsp if (err->code != GOT_ERR_ITER_NEED_MORE)
263 8069f636 2019-01-12 stsp break;
264 8069f636 2019-01-12 stsp err = got_commit_graph_fetch_commits(graph, 1, repo);
265 8069f636 2019-01-12 stsp if (err)
266 8069f636 2019-01-12 stsp break;
267 8069f636 2019-01-12 stsp else
268 8069f636 2019-01-12 stsp continue;
269 8069f636 2019-01-12 stsp }
270 8069f636 2019-01-12 stsp if (id == NULL)
271 8069f636 2019-01-12 stsp break;
272 8069f636 2019-01-12 stsp if (got_object_id_cmp(id, commit_id) == 0)
273 8069f636 2019-01-12 stsp break;
274 8069f636 2019-01-12 stsp }
275 8069f636 2019-01-12 stsp done:
276 8069f636 2019-01-12 stsp if (head_ref)
277 8069f636 2019-01-12 stsp got_ref_close(head_ref);
278 8069f636 2019-01-12 stsp if (graph)
279 8069f636 2019-01-12 stsp got_commit_graph_close(graph);
280 8069f636 2019-01-12 stsp return err;
281 c09a553d 2018-03-12 stsp }
282 c09a553d 2018-03-12 stsp
283 8069f636 2019-01-12 stsp
284 4ed7e80c 2018-05-20 stsp static const struct got_error *
285 c09a553d 2018-03-12 stsp cmd_checkout(int argc, char *argv[])
286 c09a553d 2018-03-12 stsp {
287 c09a553d 2018-03-12 stsp const struct got_error *error = NULL;
288 c09a553d 2018-03-12 stsp struct got_repository *repo = NULL;
289 c09a553d 2018-03-12 stsp struct got_reference *head_ref = NULL;
290 c09a553d 2018-03-12 stsp struct got_worktree *worktree = NULL;
291 c09a553d 2018-03-12 stsp char *repo_path = NULL;
292 c09a553d 2018-03-12 stsp char *worktree_path = NULL;
293 0bb8a95e 2018-03-12 stsp const char *path_prefix = "";
294 8069f636 2019-01-12 stsp char *commit_id_str = NULL;
295 e5dc7198 2018-12-29 stsp int ch, same_path_prefix;
296 c09a553d 2018-03-12 stsp
297 8069f636 2019-01-12 stsp while ((ch = getopt(argc, argv, "c:p:")) != -1) {
298 0bb8a95e 2018-03-12 stsp switch (ch) {
299 8069f636 2019-01-12 stsp case 'c':
300 8069f636 2019-01-12 stsp commit_id_str = strdup(optarg);
301 8069f636 2019-01-12 stsp if (commit_id_str == NULL)
302 8069f636 2019-01-12 stsp return got_error_from_errno();
303 8069f636 2019-01-12 stsp break;
304 0bb8a95e 2018-03-12 stsp case 'p':
305 0bb8a95e 2018-03-12 stsp path_prefix = optarg;
306 0bb8a95e 2018-03-12 stsp break;
307 0bb8a95e 2018-03-12 stsp default:
308 0bb8a95e 2018-03-12 stsp usage();
309 0bb8a95e 2018-03-12 stsp /* NOTREACHED */
310 0bb8a95e 2018-03-12 stsp }
311 0bb8a95e 2018-03-12 stsp }
312 0bb8a95e 2018-03-12 stsp
313 0bb8a95e 2018-03-12 stsp argc -= optind;
314 0bb8a95e 2018-03-12 stsp argv += optind;
315 0bb8a95e 2018-03-12 stsp
316 6715a751 2018-03-16 stsp #ifndef PROFILE
317 68ed9ba5 2019-02-10 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
318 68ed9ba5 2019-02-10 stsp "unveil", NULL) == -1)
319 c09a553d 2018-03-12 stsp err(1, "pledge");
320 6715a751 2018-03-16 stsp #endif
321 0bb8a95e 2018-03-12 stsp if (argc == 1) {
322 c09a553d 2018-03-12 stsp char *cwd, *base, *dotgit;
323 76089277 2018-04-01 stsp repo_path = realpath(argv[0], NULL);
324 76089277 2018-04-01 stsp if (repo_path == NULL)
325 76089277 2018-04-01 stsp return got_error_from_errno();
326 c09a553d 2018-03-12 stsp cwd = getcwd(NULL, 0);
327 76089277 2018-04-01 stsp if (cwd == NULL) {
328 76089277 2018-04-01 stsp error = got_error_from_errno();
329 76089277 2018-04-01 stsp goto done;
330 76089277 2018-04-01 stsp }
331 5d7c1dab 2018-04-01 stsp if (path_prefix[0])
332 5d7c1dab 2018-04-01 stsp base = basename(path_prefix);
333 5d7c1dab 2018-04-01 stsp else
334 5d7c1dab 2018-04-01 stsp base = basename(repo_path);
335 76089277 2018-04-01 stsp if (base == NULL) {
336 76089277 2018-04-01 stsp error = got_error_from_errno();
337 76089277 2018-04-01 stsp goto done;
338 76089277 2018-04-01 stsp }
339 c09a553d 2018-03-12 stsp dotgit = strstr(base, ".git");
340 c09a553d 2018-03-12 stsp if (dotgit)
341 c09a553d 2018-03-12 stsp *dotgit = '\0';
342 c09a553d 2018-03-12 stsp if (asprintf(&worktree_path, "%s/%s", cwd, base) == -1) {
343 0a585a0d 2018-03-17 stsp error = got_error_from_errno();
344 c09a553d 2018-03-12 stsp free(cwd);
345 76089277 2018-04-01 stsp goto done;
346 c09a553d 2018-03-12 stsp }
347 c09a553d 2018-03-12 stsp free(cwd);
348 0bb8a95e 2018-03-12 stsp } else if (argc == 2) {
349 76089277 2018-04-01 stsp repo_path = realpath(argv[0], NULL);
350 76089277 2018-04-01 stsp if (repo_path == NULL) {
351 76089277 2018-04-01 stsp error = got_error_from_errno();
352 76089277 2018-04-01 stsp goto done;
353 76089277 2018-04-01 stsp }
354 f7b38925 2018-04-01 stsp worktree_path = realpath(argv[1], NULL);
355 76089277 2018-04-01 stsp if (worktree_path == NULL) {
356 76089277 2018-04-01 stsp error = got_error_from_errno();
357 76089277 2018-04-01 stsp goto done;
358 76089277 2018-04-01 stsp }
359 c09a553d 2018-03-12 stsp } else
360 c09a553d 2018-03-12 stsp usage_checkout();
361 63219cd2 2019-01-04 stsp
362 0266afb7 2019-01-04 stsp error = apply_unveil(repo_path, worktree_path);
363 0266afb7 2019-01-04 stsp if (error)
364 63219cd2 2019-01-04 stsp goto done;
365 c09a553d 2018-03-12 stsp
366 c09a553d 2018-03-12 stsp error = got_repo_open(&repo, repo_path);
367 c09a553d 2018-03-12 stsp if (error != NULL)
368 c09a553d 2018-03-12 stsp goto done;
369 8069f636 2019-01-12 stsp
370 c09a553d 2018-03-12 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
371 c09a553d 2018-03-12 stsp if (error != NULL)
372 c09a553d 2018-03-12 stsp goto done;
373 c09a553d 2018-03-12 stsp
374 0bb8a95e 2018-03-12 stsp error = got_worktree_init(worktree_path, head_ref, path_prefix, repo);
375 d70b8e30 2018-12-27 stsp if (error != NULL && !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
376 c09a553d 2018-03-12 stsp goto done;
377 c09a553d 2018-03-12 stsp
378 c09a553d 2018-03-12 stsp error = got_worktree_open(&worktree, worktree_path);
379 c09a553d 2018-03-12 stsp if (error != NULL)
380 c09a553d 2018-03-12 stsp goto done;
381 c09a553d 2018-03-12 stsp
382 e5dc7198 2018-12-29 stsp error = got_worktree_match_path_prefix(&same_path_prefix, worktree,
383 e5dc7198 2018-12-29 stsp path_prefix);
384 e5dc7198 2018-12-29 stsp if (error != NULL)
385 e5dc7198 2018-12-29 stsp goto done;
386 e5dc7198 2018-12-29 stsp if (!same_path_prefix) {
387 49520a32 2018-12-29 stsp error = got_error(GOT_ERR_PATH_PREFIX);
388 49520a32 2018-12-29 stsp goto done;
389 49520a32 2018-12-29 stsp }
390 49520a32 2018-12-29 stsp
391 8069f636 2019-01-12 stsp if (commit_id_str) {
392 8069f636 2019-01-12 stsp struct got_object_id *commit_id;
393 8069f636 2019-01-12 stsp error = got_object_resolve_id_str(&commit_id, repo,
394 8069f636 2019-01-12 stsp commit_id_str);
395 8069f636 2019-01-12 stsp if (error != NULL)
396 8069f636 2019-01-12 stsp goto done;
397 8069f636 2019-01-12 stsp error = check_ancestry(worktree, commit_id, repo);
398 8069f636 2019-01-12 stsp if (error != NULL) {
399 8069f636 2019-01-12 stsp free(commit_id);
400 8069f636 2019-01-12 stsp goto done;
401 8069f636 2019-01-12 stsp }
402 8069f636 2019-01-12 stsp error = got_worktree_set_base_commit_id(worktree, repo,
403 8069f636 2019-01-12 stsp commit_id);
404 8069f636 2019-01-12 stsp free(commit_id);
405 8069f636 2019-01-12 stsp if (error)
406 8069f636 2019-01-12 stsp goto done;
407 8069f636 2019-01-12 stsp }
408 8069f636 2019-01-12 stsp
409 93a30277 2018-12-24 stsp error = got_worktree_checkout_files(worktree, repo,
410 6bad629b 2019-02-04 stsp checkout_progress, worktree_path, check_cancelled, NULL);
411 c09a553d 2018-03-12 stsp if (error != NULL)
412 c09a553d 2018-03-12 stsp goto done;
413 c09a553d 2018-03-12 stsp
414 b65ae19a 2018-04-24 stsp printf("Now shut up and hack\n");
415 c09a553d 2018-03-12 stsp
416 c09a553d 2018-03-12 stsp done:
417 8069f636 2019-01-12 stsp free(commit_id_str);
418 76089277 2018-04-01 stsp free(repo_path);
419 507dc3bb 2018-12-29 stsp free(worktree_path);
420 507dc3bb 2018-12-29 stsp return error;
421 507dc3bb 2018-12-29 stsp }
422 507dc3bb 2018-12-29 stsp
423 507dc3bb 2018-12-29 stsp __dead static void
424 507dc3bb 2018-12-29 stsp usage_update(void)
425 507dc3bb 2018-12-29 stsp {
426 507dc3bb 2018-12-29 stsp fprintf(stderr, "usage: %s update [-c commit] [worktree-path]\n",
427 507dc3bb 2018-12-29 stsp getprogname());
428 507dc3bb 2018-12-29 stsp exit(1);
429 507dc3bb 2018-12-29 stsp }
430 507dc3bb 2018-12-29 stsp
431 507dc3bb 2018-12-29 stsp static void
432 507dc3bb 2018-12-29 stsp update_progress(void *arg, unsigned char status, const char *path)
433 507dc3bb 2018-12-29 stsp {
434 784955db 2019-01-12 stsp int *did_something = arg;
435 784955db 2019-01-12 stsp
436 507dc3bb 2018-12-29 stsp if (status == GOT_STATUS_EXISTS)
437 507dc3bb 2018-12-29 stsp return;
438 507dc3bb 2018-12-29 stsp
439 1545c615 2019-02-10 stsp *did_something = 1;
440 507dc3bb 2018-12-29 stsp while (path[0] == '/')
441 507dc3bb 2018-12-29 stsp path++;
442 507dc3bb 2018-12-29 stsp printf("%c %s\n", status, path);
443 be7061eb 2018-12-30 stsp }
444 be7061eb 2018-12-30 stsp
445 be7061eb 2018-12-30 stsp static const struct got_error *
446 507dc3bb 2018-12-29 stsp cmd_update(int argc, char *argv[])
447 507dc3bb 2018-12-29 stsp {
448 507dc3bb 2018-12-29 stsp const struct got_error *error = NULL;
449 507dc3bb 2018-12-29 stsp struct got_repository *repo = NULL;
450 507dc3bb 2018-12-29 stsp struct got_worktree *worktree = NULL;
451 507dc3bb 2018-12-29 stsp char *worktree_path = NULL;
452 507dc3bb 2018-12-29 stsp struct got_object_id *commit_id = NULL;
453 9c4b8182 2019-01-02 stsp char *commit_id_str = NULL;
454 784955db 2019-01-12 stsp int ch, did_something = 0;
455 507dc3bb 2018-12-29 stsp
456 507dc3bb 2018-12-29 stsp while ((ch = getopt(argc, argv, "c:")) != -1) {
457 507dc3bb 2018-12-29 stsp switch (ch) {
458 507dc3bb 2018-12-29 stsp case 'c':
459 9c4b8182 2019-01-02 stsp commit_id_str = strdup(optarg);
460 9c4b8182 2019-01-02 stsp if (commit_id_str == NULL)
461 9c4b8182 2019-01-02 stsp return got_error_from_errno();
462 507dc3bb 2018-12-29 stsp break;
463 507dc3bb 2018-12-29 stsp default:
464 507dc3bb 2018-12-29 stsp usage();
465 507dc3bb 2018-12-29 stsp /* NOTREACHED */
466 507dc3bb 2018-12-29 stsp }
467 507dc3bb 2018-12-29 stsp }
468 507dc3bb 2018-12-29 stsp
469 507dc3bb 2018-12-29 stsp argc -= optind;
470 507dc3bb 2018-12-29 stsp argv += optind;
471 507dc3bb 2018-12-29 stsp
472 507dc3bb 2018-12-29 stsp #ifndef PROFILE
473 68ed9ba5 2019-02-10 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
474 68ed9ba5 2019-02-10 stsp "unveil", NULL) == -1)
475 507dc3bb 2018-12-29 stsp err(1, "pledge");
476 507dc3bb 2018-12-29 stsp #endif
477 507dc3bb 2018-12-29 stsp if (argc == 0) {
478 507dc3bb 2018-12-29 stsp worktree_path = getcwd(NULL, 0);
479 507dc3bb 2018-12-29 stsp if (worktree_path == NULL) {
480 507dc3bb 2018-12-29 stsp error = got_error_from_errno();
481 507dc3bb 2018-12-29 stsp goto done;
482 507dc3bb 2018-12-29 stsp }
483 507dc3bb 2018-12-29 stsp } else if (argc == 1) {
484 507dc3bb 2018-12-29 stsp worktree_path = realpath(argv[0], NULL);
485 507dc3bb 2018-12-29 stsp if (worktree_path == NULL) {
486 507dc3bb 2018-12-29 stsp error = got_error_from_errno();
487 507dc3bb 2018-12-29 stsp goto done;
488 507dc3bb 2018-12-29 stsp }
489 507dc3bb 2018-12-29 stsp } else
490 507dc3bb 2018-12-29 stsp usage_update();
491 507dc3bb 2018-12-29 stsp
492 507dc3bb 2018-12-29 stsp error = got_worktree_open(&worktree, worktree_path);
493 507dc3bb 2018-12-29 stsp if (error != NULL)
494 507dc3bb 2018-12-29 stsp goto done;
495 507dc3bb 2018-12-29 stsp
496 507dc3bb 2018-12-29 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
497 507dc3bb 2018-12-29 stsp if (error != NULL)
498 507dc3bb 2018-12-29 stsp goto done;
499 507dc3bb 2018-12-29 stsp
500 8c02d095 2019-02-05 stsp error = apply_unveil(got_repo_get_path(repo),
501 8c02d095 2019-02-05 stsp got_worktree_get_root_path(worktree));
502 0266afb7 2019-01-04 stsp if (error)
503 0266afb7 2019-01-04 stsp goto done;
504 0266afb7 2019-01-04 stsp
505 507dc3bb 2018-12-29 stsp if (commit_id_str == NULL) {
506 507dc3bb 2018-12-29 stsp struct got_reference *head_ref;
507 507dc3bb 2018-12-29 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
508 507dc3bb 2018-12-29 stsp if (error != NULL)
509 507dc3bb 2018-12-29 stsp goto done;
510 507dc3bb 2018-12-29 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
511 9c4b8182 2019-01-02 stsp if (error != NULL)
512 9c4b8182 2019-01-02 stsp goto done;
513 9c4b8182 2019-01-02 stsp error = got_object_id_str(&commit_id_str, commit_id);
514 507dc3bb 2018-12-29 stsp if (error != NULL)
515 507dc3bb 2018-12-29 stsp goto done;
516 507dc3bb 2018-12-29 stsp } else {
517 507dc3bb 2018-12-29 stsp error = got_object_resolve_id_str(&commit_id, repo,
518 507dc3bb 2018-12-29 stsp commit_id_str);
519 507dc3bb 2018-12-29 stsp if (error != NULL)
520 507dc3bb 2018-12-29 stsp goto done;
521 507dc3bb 2018-12-29 stsp }
522 35c965b2 2018-12-29 stsp
523 be7061eb 2018-12-30 stsp error = check_ancestry(worktree, commit_id, repo);
524 be7061eb 2018-12-30 stsp if (error != NULL)
525 be7061eb 2018-12-30 stsp goto done;
526 507dc3bb 2018-12-29 stsp
527 507dc3bb 2018-12-29 stsp if (got_object_id_cmp(got_worktree_get_base_commit_id(worktree),
528 507dc3bb 2018-12-29 stsp commit_id) != 0) {
529 507dc3bb 2018-12-29 stsp error = got_worktree_set_base_commit_id(worktree, repo,
530 507dc3bb 2018-12-29 stsp commit_id);
531 507dc3bb 2018-12-29 stsp if (error)
532 507dc3bb 2018-12-29 stsp goto done;
533 507dc3bb 2018-12-29 stsp }
534 507dc3bb 2018-12-29 stsp
535 507dc3bb 2018-12-29 stsp error = got_worktree_checkout_files(worktree, repo,
536 6bad629b 2019-02-04 stsp update_progress, &did_something, check_cancelled, NULL);
537 507dc3bb 2018-12-29 stsp if (error != NULL)
538 507dc3bb 2018-12-29 stsp goto done;
539 9c4b8182 2019-01-02 stsp
540 784955db 2019-01-12 stsp if (did_something)
541 784955db 2019-01-12 stsp printf("Updated to commit %s\n", commit_id_str);
542 784955db 2019-01-12 stsp else
543 784955db 2019-01-12 stsp printf("Already up-to-date\n");
544 507dc3bb 2018-12-29 stsp done:
545 c09a553d 2018-03-12 stsp free(worktree_path);
546 507dc3bb 2018-12-29 stsp free(commit_id);
547 9c4b8182 2019-01-02 stsp free(commit_id_str);
548 c09a553d 2018-03-12 stsp return error;
549 c09a553d 2018-03-12 stsp }
550 c09a553d 2018-03-12 stsp
551 f42b1b34 2018-03-12 stsp static const struct got_error *
552 79109fed 2018-03-27 stsp print_patch(struct got_commit_object *commit, struct got_object_id *id,
553 c0cc5c62 2018-10-18 stsp int diff_context, struct got_repository *repo)
554 5c860e29 2018-03-12 stsp {
555 f42b1b34 2018-03-12 stsp const struct got_error *err = NULL;
556 79109fed 2018-03-27 stsp struct got_tree_object *tree1 = NULL, *tree2;
557 79f35eb3 2018-06-11 stsp struct got_object_qid *qid;
558 0f2b3dca 2018-12-22 stsp char *id_str1 = NULL, *id_str2;
559 79109fed 2018-03-27 stsp
560 45d799e2 2018-12-23 stsp err = got_object_open_as_tree(&tree2, repo,
561 45d799e2 2018-12-23 stsp got_object_commit_get_tree_id(commit));
562 79109fed 2018-03-27 stsp if (err)
563 79109fed 2018-03-27 stsp return err;
564 79109fed 2018-03-27 stsp
565 45d799e2 2018-12-23 stsp qid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
566 79f35eb3 2018-06-11 stsp if (qid != NULL) {
567 79109fed 2018-03-27 stsp struct got_commit_object *pcommit;
568 79109fed 2018-03-27 stsp
569 117e771c 2018-07-23 stsp err = got_object_open_as_commit(&pcommit, repo, qid->id);
570 79109fed 2018-03-27 stsp if (err)
571 79109fed 2018-03-27 stsp return err;
572 79109fed 2018-03-27 stsp
573 45d799e2 2018-12-23 stsp err = got_object_open_as_tree(&tree1, repo,
574 45d799e2 2018-12-23 stsp got_object_commit_get_tree_id(pcommit));
575 79109fed 2018-03-27 stsp got_object_commit_close(pcommit);
576 0f2b3dca 2018-12-22 stsp if (err)
577 0f2b3dca 2018-12-22 stsp return err;
578 0f2b3dca 2018-12-22 stsp
579 0f2b3dca 2018-12-22 stsp err = got_object_id_str(&id_str1, qid->id);
580 79109fed 2018-03-27 stsp if (err)
581 79109fed 2018-03-27 stsp return err;
582 79109fed 2018-03-27 stsp }
583 79109fed 2018-03-27 stsp
584 0f2b3dca 2018-12-22 stsp err = got_object_id_str(&id_str2, id);
585 0f2b3dca 2018-12-22 stsp if (err)
586 0f2b3dca 2018-12-22 stsp goto done;
587 0f2b3dca 2018-12-22 stsp
588 56765ebb 2018-12-23 stsp printf("diff %s %s\n", id_str1 ? id_str1 : "/dev/null", id_str2);
589 54156555 2018-12-24 stsp err = got_diff_tree(tree1, tree2, "", "", diff_context, repo, stdout);
590 0f2b3dca 2018-12-22 stsp done:
591 79109fed 2018-03-27 stsp if (tree1)
592 79109fed 2018-03-27 stsp got_object_tree_close(tree1);
593 79109fed 2018-03-27 stsp got_object_tree_close(tree2);
594 0f2b3dca 2018-12-22 stsp free(id_str1);
595 0f2b3dca 2018-12-22 stsp free(id_str2);
596 79109fed 2018-03-27 stsp return err;
597 79109fed 2018-03-27 stsp }
598 79109fed 2018-03-27 stsp
599 4bb494d5 2018-06-16 stsp static char *
600 6c281f94 2018-06-11 stsp get_datestr(time_t *time, char *datebuf)
601 6c281f94 2018-06-11 stsp {
602 6c281f94 2018-06-11 stsp char *p, *s = ctime_r(time, datebuf);
603 6c281f94 2018-06-11 stsp p = strchr(s, '\n');
604 6c281f94 2018-06-11 stsp if (p)
605 6c281f94 2018-06-11 stsp *p = '\0';
606 6c281f94 2018-06-11 stsp return s;
607 6c281f94 2018-06-11 stsp }
608 6c281f94 2018-06-11 stsp
609 79109fed 2018-03-27 stsp static const struct got_error *
610 79109fed 2018-03-27 stsp print_commit(struct got_commit_object *commit, struct got_object_id *id,
611 199a4027 2019-02-02 stsp struct got_repository *repo, int show_patch, int diff_context,
612 199a4027 2019-02-02 stsp struct got_reflist_head *refs)
613 79109fed 2018-03-27 stsp {
614 79109fed 2018-03-27 stsp const struct got_error *err = NULL;
615 621015ac 2018-07-23 stsp char *id_str, *datestr, *logmsg0, *logmsg, *line;
616 6c281f94 2018-06-11 stsp char datebuf[26];
617 45d799e2 2018-12-23 stsp time_t committer_time;
618 45d799e2 2018-12-23 stsp const char *author, *committer;
619 199a4027 2019-02-02 stsp char *refs_str = NULL;
620 199a4027 2019-02-02 stsp struct got_reflist_entry *re;
621 5c860e29 2018-03-12 stsp
622 199a4027 2019-02-02 stsp SIMPLEQ_FOREACH(re, refs, entry) {
623 199a4027 2019-02-02 stsp char *s;
624 199a4027 2019-02-02 stsp const char *name;
625 199a4027 2019-02-02 stsp if (got_object_id_cmp(re->id, id) != 0)
626 199a4027 2019-02-02 stsp continue;
627 199a4027 2019-02-02 stsp name = got_ref_get_name(re->ref);
628 d9498b20 2019-02-05 stsp if (strcmp(name, GOT_REF_HEAD) == 0)
629 d9498b20 2019-02-05 stsp continue;
630 199a4027 2019-02-02 stsp if (strncmp(name, "refs/", 5) == 0)
631 199a4027 2019-02-02 stsp name += 5;
632 e34f9ed6 2019-02-02 stsp if (strncmp(name, "heads/", 6) == 0)
633 e34f9ed6 2019-02-02 stsp name += 6;
634 141c2bff 2019-02-04 stsp if (strncmp(name, "remotes/", 8) == 0)
635 141c2bff 2019-02-04 stsp name += 8;
636 199a4027 2019-02-02 stsp s = refs_str;
637 199a4027 2019-02-02 stsp if (asprintf(&refs_str, "%s%s%s", s ? s : "", s ? ", " : "",
638 199a4027 2019-02-02 stsp name) == -1) {
639 199a4027 2019-02-02 stsp err = got_error_from_errno();
640 199a4027 2019-02-02 stsp free(s);
641 199a4027 2019-02-02 stsp break;
642 199a4027 2019-02-02 stsp }
643 199a4027 2019-02-02 stsp free(s);
644 199a4027 2019-02-02 stsp }
645 832c249c 2018-06-10 stsp err = got_object_id_str(&id_str, id);
646 8bf5b3c9 2018-03-17 stsp if (err)
647 8bf5b3c9 2018-03-17 stsp return err;
648 788c352e 2018-06-16 stsp
649 33d869be 2018-12-22 stsp printf("-----------------------------------------------\n");
650 199a4027 2019-02-02 stsp printf("commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
651 199a4027 2019-02-02 stsp refs_str ? refs_str : "", refs_str ? ")" : "");
652 832c249c 2018-06-10 stsp free(id_str);
653 45d799e2 2018-12-23 stsp printf("from: %s\n", got_object_commit_get_author(commit));
654 45d799e2 2018-12-23 stsp committer_time = got_object_commit_get_committer_time(commit);
655 45d799e2 2018-12-23 stsp datestr = get_datestr(&committer_time, datebuf);
656 dab5fe87 2018-09-14 stsp printf("date: %s UTC\n", datestr);
657 45d799e2 2018-12-23 stsp author = got_object_commit_get_author(commit);
658 45d799e2 2018-12-23 stsp committer = got_object_commit_get_committer(commit);
659 45d799e2 2018-12-23 stsp if (strcmp(author, committer) != 0)
660 45d799e2 2018-12-23 stsp printf("via: %s\n", committer);
661 45d799e2 2018-12-23 stsp if (got_object_commit_get_nparents(commit) > 1) {
662 45d799e2 2018-12-23 stsp const struct got_object_id_queue *parent_ids;
663 79f35eb3 2018-06-11 stsp struct got_object_qid *qid;
664 3fe1abad 2018-06-10 stsp int n = 1;
665 45d799e2 2018-12-23 stsp parent_ids = got_object_commit_get_parent_ids(commit);
666 45d799e2 2018-12-23 stsp SIMPLEQ_FOREACH(qid, parent_ids, entry) {
667 79f35eb3 2018-06-11 stsp err = got_object_id_str(&id_str, qid->id);
668 a0603db2 2018-06-10 stsp if (err)
669 a0603db2 2018-06-10 stsp return err;
670 3fe1abad 2018-06-10 stsp printf("parent %d: %s\n", n++, id_str);
671 a0603db2 2018-06-10 stsp free(id_str);
672 a0603db2 2018-06-10 stsp }
673 a0603db2 2018-06-10 stsp }
674 832c249c 2018-06-10 stsp
675 45d799e2 2018-12-23 stsp logmsg0 = strdup(got_object_commit_get_logmsg(commit));
676 621015ac 2018-07-23 stsp if (logmsg0 == NULL)
677 832c249c 2018-06-10 stsp return got_error_from_errno();
678 8bf5b3c9 2018-03-17 stsp
679 621015ac 2018-07-23 stsp logmsg = logmsg0;
680 832c249c 2018-06-10 stsp do {
681 832c249c 2018-06-10 stsp line = strsep(&logmsg, "\n");
682 832c249c 2018-06-10 stsp if (line)
683 832c249c 2018-06-10 stsp printf(" %s\n", line);
684 832c249c 2018-06-10 stsp } while (line);
685 621015ac 2018-07-23 stsp free(logmsg0);
686 832c249c 2018-06-10 stsp
687 971751ac 2018-03-27 stsp if (show_patch) {
688 c0cc5c62 2018-10-18 stsp err = print_patch(commit, id, diff_context, repo);
689 971751ac 2018-03-27 stsp if (err == 0)
690 971751ac 2018-03-27 stsp printf("\n");
691 971751ac 2018-03-27 stsp }
692 07862c20 2018-09-15 stsp
693 d82de447 2018-09-15 stsp fflush(stdout);
694 07862c20 2018-09-15 stsp return err;
695 f42b1b34 2018-03-12 stsp }
696 5c860e29 2018-03-12 stsp
697 f42b1b34 2018-03-12 stsp static const struct got_error *
698 15a94983 2018-12-23 stsp print_commits(struct got_object_id *root_id, struct got_repository *repo,
699 15a94983 2018-12-23 stsp char *path, int show_patch, int diff_context, int limit,
700 199a4027 2019-02-02 stsp int first_parent_traversal, struct got_reflist_head *refs)
701 f42b1b34 2018-03-12 stsp {
702 f42b1b34 2018-03-12 stsp const struct got_error *err;
703 372ccdbb 2018-06-10 stsp struct got_commit_graph *graph;
704 372ccdbb 2018-06-10 stsp
705 31cedeaf 2018-09-15 stsp err = got_commit_graph_open(&graph, root_id, path,
706 31cedeaf 2018-09-15 stsp first_parent_traversal, repo);
707 f42b1b34 2018-03-12 stsp if (err)
708 f42b1b34 2018-03-12 stsp return err;
709 31cedeaf 2018-09-15 stsp err = got_commit_graph_iter_start(graph, root_id, repo);
710 372ccdbb 2018-06-10 stsp if (err)
711 fcc85cad 2018-07-22 stsp goto done;
712 31cedeaf 2018-09-15 stsp while (1) {
713 372ccdbb 2018-06-10 stsp struct got_commit_object *commit;
714 372ccdbb 2018-06-10 stsp struct got_object_id *id;
715 8bf5b3c9 2018-03-17 stsp
716 84453469 2018-11-11 stsp if (sigint_received || sigpipe_received)
717 84453469 2018-11-11 stsp break;
718 84453469 2018-11-11 stsp
719 b43fbaa0 2018-06-11 stsp err = got_commit_graph_iter_next(&id, graph);
720 372ccdbb 2018-06-10 stsp if (err) {
721 9ba79e04 2018-06-11 stsp if (err->code == GOT_ERR_ITER_COMPLETED) {
722 9ba79e04 2018-06-11 stsp err = NULL;
723 9ba79e04 2018-06-11 stsp break;
724 9ba79e04 2018-06-11 stsp }
725 372ccdbb 2018-06-10 stsp if (err->code != GOT_ERR_ITER_NEED_MORE)
726 372ccdbb 2018-06-10 stsp break;
727 31cedeaf 2018-09-15 stsp err = got_commit_graph_fetch_commits(graph, 1, repo);
728 372ccdbb 2018-06-10 stsp if (err)
729 372ccdbb 2018-06-10 stsp break;
730 372ccdbb 2018-06-10 stsp else
731 372ccdbb 2018-06-10 stsp continue;
732 7e665116 2018-04-02 stsp }
733 b43fbaa0 2018-06-11 stsp if (id == NULL)
734 7e665116 2018-04-02 stsp break;
735 b43fbaa0 2018-06-11 stsp
736 f8e900f3 2018-06-11 stsp err = got_object_open_as_commit(&commit, repo, id);
737 b43fbaa0 2018-06-11 stsp if (err)
738 fcc85cad 2018-07-22 stsp break;
739 199a4027 2019-02-02 stsp err = print_commit(commit, id, repo, show_patch, diff_context,
740 199a4027 2019-02-02 stsp refs);
741 b43fbaa0 2018-06-11 stsp got_object_commit_close(commit);
742 372ccdbb 2018-06-10 stsp if (err || (limit && --limit == 0))
743 7e665116 2018-04-02 stsp break;
744 31cedeaf 2018-09-15 stsp }
745 fcc85cad 2018-07-22 stsp done:
746 372ccdbb 2018-06-10 stsp got_commit_graph_close(graph);
747 f42b1b34 2018-03-12 stsp return err;
748 f42b1b34 2018-03-12 stsp }
749 5c860e29 2018-03-12 stsp
750 4ed7e80c 2018-05-20 stsp __dead static void
751 6f3d1eb0 2018-03-12 stsp usage_log(void)
752 6f3d1eb0 2018-03-12 stsp {
753 c0cc5c62 2018-10-18 stsp fprintf(stderr, "usage: %s log [-c commit] [-C number] [-f] [ -l N ] [-p] "
754 04ca23f4 2018-07-16 stsp "[-r repository-path] [path]\n", getprogname());
755 6f3d1eb0 2018-03-12 stsp exit(1);
756 6f3d1eb0 2018-03-12 stsp }
757 6f3d1eb0 2018-03-12 stsp
758 4ed7e80c 2018-05-20 stsp static const struct got_error *
759 f42b1b34 2018-03-12 stsp cmd_log(int argc, char *argv[])
760 f42b1b34 2018-03-12 stsp {
761 f42b1b34 2018-03-12 stsp const struct got_error *error;
762 04ca23f4 2018-07-16 stsp struct got_repository *repo = NULL;
763 cffc0aa4 2019-02-05 stsp struct got_worktree *worktree = NULL;
764 15a94983 2018-12-23 stsp struct got_commit_object *commit = NULL;
765 3235492e 2018-04-01 stsp struct got_object_id *id = NULL;
766 04ca23f4 2018-07-16 stsp char *repo_path = NULL, *path = NULL, *cwd = NULL, *in_repo_path = NULL;
767 d142fc45 2018-04-01 stsp char *start_commit = NULL;
768 c0cc5c62 2018-10-18 stsp int diff_context = 3, ch;
769 1fd6d7ea 2018-06-13 stsp int show_patch = 0, limit = 0, first_parent_traversal = 0;
770 64a96a6d 2018-04-01 stsp const char *errstr;
771 199a4027 2019-02-02 stsp struct got_reflist_head refs;
772 5c860e29 2018-03-12 stsp
773 6715a751 2018-03-16 stsp #ifndef PROFILE
774 6098196c 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
775 6098196c 2019-01-04 stsp NULL)
776 ad242220 2018-09-08 stsp == -1)
777 f42b1b34 2018-03-12 stsp err(1, "pledge");
778 6715a751 2018-03-16 stsp #endif
779 79109fed 2018-03-27 stsp
780 c0cc5c62 2018-10-18 stsp while ((ch = getopt(argc, argv, "pc:C:l:fr:")) != -1) {
781 79109fed 2018-03-27 stsp switch (ch) {
782 79109fed 2018-03-27 stsp case 'p':
783 79109fed 2018-03-27 stsp show_patch = 1;
784 d142fc45 2018-04-01 stsp break;
785 d142fc45 2018-04-01 stsp case 'c':
786 d142fc45 2018-04-01 stsp start_commit = optarg;
787 64a96a6d 2018-04-01 stsp break;
788 c0cc5c62 2018-10-18 stsp case 'C':
789 4a8520aa 2018-10-18 stsp diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
790 4a8520aa 2018-10-18 stsp &errstr);
791 c0cc5c62 2018-10-18 stsp if (errstr != NULL)
792 c0cc5c62 2018-10-18 stsp err(1, "-C option %s", errstr);
793 c0cc5c62 2018-10-18 stsp break;
794 64a96a6d 2018-04-01 stsp case 'l':
795 64a96a6d 2018-04-01 stsp limit = strtonum(optarg, 1, INT_MAX, &errstr);
796 64a96a6d 2018-04-01 stsp if (errstr != NULL)
797 64a96a6d 2018-04-01 stsp err(1, "-l option %s", errstr);
798 79109fed 2018-03-27 stsp break;
799 0ed6ed4c 2018-06-13 stsp case 'f':
800 0ed6ed4c 2018-06-13 stsp first_parent_traversal = 1;
801 a0603db2 2018-06-10 stsp break;
802 04ca23f4 2018-07-16 stsp case 'r':
803 04ca23f4 2018-07-16 stsp repo_path = realpath(optarg, NULL);
804 04ca23f4 2018-07-16 stsp if (repo_path == NULL)
805 04ca23f4 2018-07-16 stsp err(1, "-r option");
806 04ca23f4 2018-07-16 stsp break;
807 79109fed 2018-03-27 stsp default:
808 79109fed 2018-03-27 stsp usage();
809 79109fed 2018-03-27 stsp /* NOTREACHED */
810 79109fed 2018-03-27 stsp }
811 79109fed 2018-03-27 stsp }
812 79109fed 2018-03-27 stsp
813 79109fed 2018-03-27 stsp argc -= optind;
814 79109fed 2018-03-27 stsp argv += optind;
815 79109fed 2018-03-27 stsp
816 04ca23f4 2018-07-16 stsp if (argc == 0)
817 04ca23f4 2018-07-16 stsp path = strdup("");
818 04ca23f4 2018-07-16 stsp else if (argc == 1)
819 04ca23f4 2018-07-16 stsp path = strdup(argv[0]);
820 04ca23f4 2018-07-16 stsp else
821 6f3d1eb0 2018-03-12 stsp usage_log();
822 04ca23f4 2018-07-16 stsp if (path == NULL)
823 04ca23f4 2018-07-16 stsp return got_error_from_errno();
824 f42b1b34 2018-03-12 stsp
825 04ca23f4 2018-07-16 stsp cwd = getcwd(NULL, 0);
826 04ca23f4 2018-07-16 stsp if (cwd == NULL) {
827 04ca23f4 2018-07-16 stsp error = got_error_from_errno();
828 04ca23f4 2018-07-16 stsp goto done;
829 04ca23f4 2018-07-16 stsp }
830 cffc0aa4 2019-02-05 stsp
831 cffc0aa4 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
832 cffc0aa4 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
833 cffc0aa4 2019-02-05 stsp goto done;
834 cffc0aa4 2019-02-05 stsp error = NULL;
835 cffc0aa4 2019-02-05 stsp
836 cffc0aa4 2019-02-05 stsp repo_path = worktree ?
837 cffc0aa4 2019-02-05 stsp strdup(got_worktree_get_repo_path(worktree)) : strdup(cwd);
838 04ca23f4 2018-07-16 stsp if (repo_path == NULL) {
839 cffc0aa4 2019-02-05 stsp error = got_error_from_errno();
840 cffc0aa4 2019-02-05 stsp goto done;
841 04ca23f4 2018-07-16 stsp }
842 04ca23f4 2018-07-16 stsp
843 6098196c 2019-01-04 stsp error = apply_unveil(repo_path, NULL);
844 6098196c 2019-01-04 stsp if (error)
845 6098196c 2019-01-04 stsp goto done;
846 6098196c 2019-01-04 stsp
847 f42b1b34 2018-03-12 stsp error = got_repo_open(&repo, repo_path);
848 d7d4f210 2018-03-12 stsp if (error != NULL)
849 04ca23f4 2018-07-16 stsp goto done;
850 f42b1b34 2018-03-12 stsp
851 d142fc45 2018-04-01 stsp if (start_commit == NULL) {
852 3235492e 2018-04-01 stsp struct got_reference *head_ref;
853 3235492e 2018-04-01 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
854 3235492e 2018-04-01 stsp if (error != NULL)
855 3235492e 2018-04-01 stsp return error;
856 3235492e 2018-04-01 stsp error = got_ref_resolve(&id, repo, head_ref);
857 3235492e 2018-04-01 stsp got_ref_close(head_ref);
858 3235492e 2018-04-01 stsp if (error != NULL)
859 3235492e 2018-04-01 stsp return error;
860 15a94983 2018-12-23 stsp error = got_object_open_as_commit(&commit, repo, id);
861 3235492e 2018-04-01 stsp } else {
862 d1f2edc9 2018-06-13 stsp struct got_reference *ref;
863 d1f2edc9 2018-06-13 stsp error = got_ref_open(&ref, repo, start_commit);
864 3235492e 2018-04-01 stsp if (error == NULL) {
865 1caad61b 2019-02-01 stsp int obj_type;
866 d1f2edc9 2018-06-13 stsp error = got_ref_resolve(&id, repo, ref);
867 d1f2edc9 2018-06-13 stsp got_ref_close(ref);
868 d1f2edc9 2018-06-13 stsp if (error != NULL)
869 1caad61b 2019-02-01 stsp goto done;
870 1caad61b 2019-02-01 stsp error = got_object_get_type(&obj_type, repo, id);
871 1caad61b 2019-02-01 stsp if (error != NULL)
872 1caad61b 2019-02-01 stsp goto done;
873 1caad61b 2019-02-01 stsp if (obj_type == GOT_OBJ_TYPE_TAG) {
874 1caad61b 2019-02-01 stsp struct got_tag_object *tag;
875 1caad61b 2019-02-01 stsp error = got_object_open_as_tag(&tag, repo, id);
876 1caad61b 2019-02-01 stsp if (error != NULL)
877 1caad61b 2019-02-01 stsp goto done;
878 1caad61b 2019-02-01 stsp if (got_object_tag_get_object_type(tag) !=
879 1caad61b 2019-02-01 stsp GOT_OBJ_TYPE_COMMIT) {
880 1caad61b 2019-02-01 stsp got_object_tag_close(tag);
881 1caad61b 2019-02-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
882 1caad61b 2019-02-01 stsp goto done;
883 1caad61b 2019-02-01 stsp }
884 1caad61b 2019-02-01 stsp free(id);
885 1caad61b 2019-02-01 stsp id = got_object_id_dup(
886 1caad61b 2019-02-01 stsp got_object_tag_get_object_id(tag));
887 1caad61b 2019-02-01 stsp if (id == NULL)
888 1caad61b 2019-02-01 stsp error = got_error_from_errno();
889 1caad61b 2019-02-01 stsp got_object_tag_close(tag);
890 1caad61b 2019-02-01 stsp if (error)
891 1caad61b 2019-02-01 stsp goto done;
892 1caad61b 2019-02-01 stsp } else if (obj_type != GOT_OBJ_TYPE_COMMIT) {
893 1caad61b 2019-02-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
894 1caad61b 2019-02-01 stsp goto done;
895 1caad61b 2019-02-01 stsp }
896 15a94983 2018-12-23 stsp error = got_object_open_as_commit(&commit, repo, id);
897 d1f2edc9 2018-06-13 stsp if (error != NULL)
898 1caad61b 2019-02-01 stsp goto done;
899 d1f2edc9 2018-06-13 stsp }
900 15a94983 2018-12-23 stsp if (commit == NULL) {
901 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&id, repo,
902 d1f2edc9 2018-06-13 stsp start_commit);
903 d1f2edc9 2018-06-13 stsp if (error != NULL)
904 d1f2edc9 2018-06-13 stsp return error;
905 3235492e 2018-04-01 stsp }
906 3235492e 2018-04-01 stsp }
907 d7d4f210 2018-03-12 stsp if (error != NULL)
908 04ca23f4 2018-07-16 stsp goto done;
909 04ca23f4 2018-07-16 stsp
910 23721109 2018-10-22 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
911 04ca23f4 2018-07-16 stsp if (error != NULL)
912 04ca23f4 2018-07-16 stsp goto done;
913 04ca23f4 2018-07-16 stsp if (in_repo_path) {
914 04ca23f4 2018-07-16 stsp free(path);
915 04ca23f4 2018-07-16 stsp path = in_repo_path;
916 04ca23f4 2018-07-16 stsp }
917 199a4027 2019-02-02 stsp
918 199a4027 2019-02-02 stsp SIMPLEQ_INIT(&refs);
919 199a4027 2019-02-02 stsp error = got_ref_list(&refs, repo);
920 199a4027 2019-02-02 stsp if (error)
921 199a4027 2019-02-02 stsp goto done;
922 04ca23f4 2018-07-16 stsp
923 15a94983 2018-12-23 stsp error = print_commits(id, repo, path, show_patch,
924 199a4027 2019-02-02 stsp diff_context, limit, first_parent_traversal, &refs);
925 04ca23f4 2018-07-16 stsp done:
926 04ca23f4 2018-07-16 stsp free(path);
927 04ca23f4 2018-07-16 stsp free(repo_path);
928 04ca23f4 2018-07-16 stsp free(cwd);
929 f42b1b34 2018-03-12 stsp free(id);
930 cffc0aa4 2019-02-05 stsp if (worktree)
931 cffc0aa4 2019-02-05 stsp got_worktree_close(worktree);
932 ad242220 2018-09-08 stsp if (repo) {
933 ad242220 2018-09-08 stsp const struct got_error *repo_error;
934 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
935 ad242220 2018-09-08 stsp if (error == NULL)
936 ad242220 2018-09-08 stsp error = repo_error;
937 ad242220 2018-09-08 stsp }
938 8bf5b3c9 2018-03-17 stsp return error;
939 5c860e29 2018-03-12 stsp }
940 5c860e29 2018-03-12 stsp
941 4ed7e80c 2018-05-20 stsp __dead static void
942 3f8b7d6a 2018-04-01 stsp usage_diff(void)
943 3f8b7d6a 2018-04-01 stsp {
944 b72f483a 2019-02-05 stsp fprintf(stderr, "usage: %s diff [-C number] [-r repository-path] "
945 927df6b7 2019-02-10 stsp "[object1 object2 | path]\n", getprogname());
946 3f8b7d6a 2018-04-01 stsp exit(1);
947 b00d56cd 2018-04-01 stsp }
948 b00d56cd 2018-04-01 stsp
949 b72f483a 2019-02-05 stsp struct print_diff_arg {
950 b72f483a 2019-02-05 stsp struct got_repository *repo;
951 b72f483a 2019-02-05 stsp struct got_worktree *worktree;
952 b72f483a 2019-02-05 stsp int diff_context;
953 3fc0c068 2019-02-10 stsp const char *id_str;
954 3fc0c068 2019-02-10 stsp int header_shown;
955 b72f483a 2019-02-05 stsp };
956 b72f483a 2019-02-05 stsp
957 4ed7e80c 2018-05-20 stsp static const struct got_error *
958 b72f483a 2019-02-05 stsp print_diff(void *arg, unsigned char status, const char *path,
959 b72f483a 2019-02-05 stsp struct got_object_id *id)
960 b72f483a 2019-02-05 stsp {
961 b72f483a 2019-02-05 stsp struct print_diff_arg *a = arg;
962 b72f483a 2019-02-05 stsp const struct got_error *err = NULL;
963 b72f483a 2019-02-05 stsp struct got_blob_object *blob1 = NULL;
964 b72f483a 2019-02-05 stsp FILE *f2 = NULL;
965 b72f483a 2019-02-05 stsp char *abspath = NULL;
966 b72f483a 2019-02-05 stsp struct stat sb;
967 b72f483a 2019-02-05 stsp
968 276262e8 2019-02-08 stsp if (status != GOT_STATUS_MODIFY)
969 b72f483a 2019-02-05 stsp return NULL;
970 3fc0c068 2019-02-10 stsp
971 3fc0c068 2019-02-10 stsp if (!a->header_shown) {
972 3fc0c068 2019-02-10 stsp printf("diff %s %s\n", a->id_str,
973 3fc0c068 2019-02-10 stsp got_worktree_get_root_path(a->worktree));
974 3fc0c068 2019-02-10 stsp a->header_shown = 1;
975 3fc0c068 2019-02-10 stsp }
976 b72f483a 2019-02-05 stsp
977 b72f483a 2019-02-05 stsp err = got_object_open_as_blob(&blob1, a->repo, id, 8192);
978 b72f483a 2019-02-05 stsp if (err)
979 b72f483a 2019-02-05 stsp goto done;
980 b72f483a 2019-02-05 stsp
981 b72f483a 2019-02-05 stsp if (asprintf(&abspath, "%s/%s",
982 b72f483a 2019-02-05 stsp got_worktree_get_root_path(a->worktree), path) == -1) {
983 b72f483a 2019-02-05 stsp err = got_error_from_errno();
984 b72f483a 2019-02-05 stsp goto done;
985 b72f483a 2019-02-05 stsp }
986 b72f483a 2019-02-05 stsp
987 b72f483a 2019-02-05 stsp f2 = fopen(abspath, "r");
988 b72f483a 2019-02-05 stsp if (f2 == NULL) {
989 b72f483a 2019-02-05 stsp err = got_error_from_errno();
990 b72f483a 2019-02-05 stsp goto done;
991 b72f483a 2019-02-05 stsp }
992 b72f483a 2019-02-05 stsp if (lstat(abspath, &sb) == -1) {
993 b72f483a 2019-02-05 stsp err = got_error_from_errno();
994 b72f483a 2019-02-05 stsp goto done;
995 b72f483a 2019-02-05 stsp }
996 b72f483a 2019-02-05 stsp
997 b72f483a 2019-02-05 stsp err = got_diff_blob_file(blob1, f2, sb.st_size, path, a->diff_context,
998 b72f483a 2019-02-05 stsp stdout);
999 b72f483a 2019-02-05 stsp done:
1000 b72f483a 2019-02-05 stsp if (blob1)
1001 b72f483a 2019-02-05 stsp got_object_blob_close(blob1);
1002 b72f483a 2019-02-05 stsp if (f2)
1003 b72f483a 2019-02-05 stsp fclose(f2);
1004 b72f483a 2019-02-05 stsp free(abspath);
1005 927df6b7 2019-02-10 stsp return err;
1006 927df6b7 2019-02-10 stsp }
1007 927df6b7 2019-02-10 stsp
1008 927df6b7 2019-02-10 stsp static const struct got_error *
1009 927df6b7 2019-02-10 stsp get_status_path(char **status_path, struct got_worktree *worktree,
1010 927df6b7 2019-02-10 stsp const char *arg)
1011 927df6b7 2019-02-10 stsp {
1012 927df6b7 2019-02-10 stsp const struct got_error *err = NULL;
1013 927df6b7 2019-02-10 stsp char *resolved, *path = NULL;
1014 927df6b7 2019-02-10 stsp size_t len;
1015 927df6b7 2019-02-10 stsp
1016 927df6b7 2019-02-10 stsp *status_path = NULL;
1017 927df6b7 2019-02-10 stsp
1018 927df6b7 2019-02-10 stsp resolved = realpath(arg, NULL);
1019 927df6b7 2019-02-10 stsp if (resolved == NULL)
1020 927df6b7 2019-02-10 stsp return got_error_from_errno();
1021 927df6b7 2019-02-10 stsp
1022 927df6b7 2019-02-10 stsp if (strncmp(got_worktree_get_root_path(worktree), resolved,
1023 927df6b7 2019-02-10 stsp strlen(got_worktree_get_root_path(worktree)))) {
1024 927df6b7 2019-02-10 stsp err = got_error(GOT_ERR_BAD_PATH);
1025 927df6b7 2019-02-10 stsp goto done;
1026 927df6b7 2019-02-10 stsp }
1027 927df6b7 2019-02-10 stsp
1028 927df6b7 2019-02-10 stsp path = strdup(resolved + strlen(got_worktree_get_root_path(worktree)));
1029 927df6b7 2019-02-10 stsp if (path == NULL) {
1030 927df6b7 2019-02-10 stsp err = got_error_from_errno();
1031 927df6b7 2019-02-10 stsp goto done;
1032 927df6b7 2019-02-10 stsp }
1033 927df6b7 2019-02-10 stsp
1034 927df6b7 2019-02-10 stsp /* XXX status walk can't deal with trailing slash! */
1035 927df6b7 2019-02-10 stsp len = strlen(path);
1036 927df6b7 2019-02-10 stsp while (path[len - 1] == '/') {
1037 927df6b7 2019-02-10 stsp path[len - 1] = '\0';
1038 927df6b7 2019-02-10 stsp len--;
1039 927df6b7 2019-02-10 stsp }
1040 927df6b7 2019-02-10 stsp done:
1041 927df6b7 2019-02-10 stsp free(resolved);
1042 927df6b7 2019-02-10 stsp if (err == NULL)
1043 927df6b7 2019-02-10 stsp *status_path = path;
1044 927df6b7 2019-02-10 stsp else
1045 927df6b7 2019-02-10 stsp free(path);
1046 b72f483a 2019-02-05 stsp return err;
1047 b72f483a 2019-02-05 stsp }
1048 b72f483a 2019-02-05 stsp
1049 b72f483a 2019-02-05 stsp static const struct got_error *
1050 b00d56cd 2018-04-01 stsp cmd_diff(int argc, char *argv[])
1051 b00d56cd 2018-04-01 stsp {
1052 b00d56cd 2018-04-01 stsp const struct got_error *error;
1053 b00d56cd 2018-04-01 stsp struct got_repository *repo = NULL;
1054 b72f483a 2019-02-05 stsp struct got_worktree *worktree = NULL;
1055 b72f483a 2019-02-05 stsp char *cwd = NULL, *repo_path = NULL;
1056 15a94983 2018-12-23 stsp struct got_object_id *id1 = NULL, *id2 = NULL;
1057 15a94983 2018-12-23 stsp char *id_str1 = NULL, *id_str2 = NULL;
1058 15a94983 2018-12-23 stsp int type1, type2;
1059 c0cc5c62 2018-10-18 stsp int diff_context = 3, ch;
1060 c0cc5c62 2018-10-18 stsp const char *errstr;
1061 927df6b7 2019-02-10 stsp char *path = NULL;
1062 b00d56cd 2018-04-01 stsp
1063 b00d56cd 2018-04-01 stsp #ifndef PROFILE
1064 25eccc22 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1065 25eccc22 2019-01-04 stsp NULL) == -1)
1066 b00d56cd 2018-04-01 stsp err(1, "pledge");
1067 b00d56cd 2018-04-01 stsp #endif
1068 b00d56cd 2018-04-01 stsp
1069 b72f483a 2019-02-05 stsp while ((ch = getopt(argc, argv, "C:r:")) != -1) {
1070 b00d56cd 2018-04-01 stsp switch (ch) {
1071 c0cc5c62 2018-10-18 stsp case 'C':
1072 c0cc5c62 2018-10-18 stsp diff_context = strtonum(optarg, 1, INT_MAX, &errstr);
1073 c0cc5c62 2018-10-18 stsp if (errstr != NULL)
1074 c0cc5c62 2018-10-18 stsp err(1, "-C option %s", errstr);
1075 c0cc5c62 2018-10-18 stsp break;
1076 b72f483a 2019-02-05 stsp case 'r':
1077 b72f483a 2019-02-05 stsp repo_path = realpath(optarg, NULL);
1078 b72f483a 2019-02-05 stsp if (repo_path == NULL)
1079 b72f483a 2019-02-05 stsp err(1, "-r option");
1080 b72f483a 2019-02-05 stsp break;
1081 b00d56cd 2018-04-01 stsp default:
1082 b00d56cd 2018-04-01 stsp usage();
1083 b00d56cd 2018-04-01 stsp /* NOTREACHED */
1084 b00d56cd 2018-04-01 stsp }
1085 b00d56cd 2018-04-01 stsp }
1086 b00d56cd 2018-04-01 stsp
1087 b00d56cd 2018-04-01 stsp argc -= optind;
1088 b00d56cd 2018-04-01 stsp argv += optind;
1089 b00d56cd 2018-04-01 stsp
1090 b72f483a 2019-02-05 stsp cwd = getcwd(NULL, 0);
1091 b72f483a 2019-02-05 stsp if (cwd == NULL) {
1092 b72f483a 2019-02-05 stsp error = got_error_from_errno();
1093 b72f483a 2019-02-05 stsp goto done;
1094 b72f483a 2019-02-05 stsp }
1095 927df6b7 2019-02-10 stsp error = got_worktree_open(&worktree, cwd);
1096 927df6b7 2019-02-10 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1097 927df6b7 2019-02-10 stsp goto done;
1098 927df6b7 2019-02-10 stsp if (argc <= 1) {
1099 927df6b7 2019-02-10 stsp if (worktree == NULL) {
1100 927df6b7 2019-02-10 stsp error = got_error(GOT_ERR_NOT_WORKTREE);
1101 927df6b7 2019-02-10 stsp goto done;
1102 927df6b7 2019-02-10 stsp }
1103 b72f483a 2019-02-05 stsp if (repo_path)
1104 b72f483a 2019-02-05 stsp errx(1,
1105 b72f483a 2019-02-05 stsp "-r option can't be used when diffing a work tree");
1106 b72f483a 2019-02-05 stsp repo_path = strdup(got_worktree_get_repo_path(worktree));
1107 927df6b7 2019-02-10 stsp if (repo_path == NULL) {
1108 927df6b7 2019-02-10 stsp error = got_error_from_errno();
1109 927df6b7 2019-02-10 stsp goto done;
1110 927df6b7 2019-02-10 stsp }
1111 927df6b7 2019-02-10 stsp if (argc == 1) {
1112 927df6b7 2019-02-10 stsp error = get_status_path(&path, worktree, argv[0]);
1113 927df6b7 2019-02-10 stsp if (error)
1114 927df6b7 2019-02-10 stsp goto done;
1115 927df6b7 2019-02-10 stsp } else {
1116 927df6b7 2019-02-10 stsp path = strdup("");
1117 927df6b7 2019-02-10 stsp if (path == NULL) {
1118 927df6b7 2019-02-10 stsp error = got_error_from_errno();
1119 927df6b7 2019-02-10 stsp goto done;
1120 927df6b7 2019-02-10 stsp }
1121 927df6b7 2019-02-10 stsp }
1122 b72f483a 2019-02-05 stsp } else if (argc == 2) {
1123 15a94983 2018-12-23 stsp id_str1 = argv[0];
1124 15a94983 2018-12-23 stsp id_str2 = argv[1];
1125 b00d56cd 2018-04-01 stsp } else
1126 b00d56cd 2018-04-01 stsp usage_diff();
1127 25eccc22 2019-01-04 stsp
1128 b72f483a 2019-02-05 stsp if (repo_path == NULL) {
1129 b72f483a 2019-02-05 stsp repo_path = getcwd(NULL, 0);
1130 b72f483a 2019-02-05 stsp if (repo_path == NULL)
1131 b72f483a 2019-02-05 stsp return got_error_from_errno();
1132 b72f483a 2019-02-05 stsp }
1133 b72f483a 2019-02-05 stsp
1134 b72f483a 2019-02-05 stsp error = apply_unveil(repo_path,
1135 b72f483a 2019-02-05 stsp worktree ? got_worktree_get_root_path(worktree) : NULL);
1136 25eccc22 2019-01-04 stsp if (error)
1137 25eccc22 2019-01-04 stsp goto done;
1138 b00d56cd 2018-04-01 stsp
1139 b00d56cd 2018-04-01 stsp error = got_repo_open(&repo, repo_path);
1140 76089277 2018-04-01 stsp free(repo_path);
1141 b00d56cd 2018-04-01 stsp if (error != NULL)
1142 b00d56cd 2018-04-01 stsp goto done;
1143 b00d56cd 2018-04-01 stsp
1144 b72f483a 2019-02-05 stsp if (worktree) {
1145 b72f483a 2019-02-05 stsp struct print_diff_arg arg;
1146 b72f483a 2019-02-05 stsp char *id_str;
1147 b72f483a 2019-02-05 stsp error = got_object_id_str(&id_str,
1148 b72f483a 2019-02-05 stsp got_worktree_get_base_commit_id(worktree));
1149 b72f483a 2019-02-05 stsp if (error)
1150 b72f483a 2019-02-05 stsp goto done;
1151 b72f483a 2019-02-05 stsp arg.repo = repo;
1152 b72f483a 2019-02-05 stsp arg.worktree = worktree;
1153 b72f483a 2019-02-05 stsp arg.diff_context = diff_context;
1154 3fc0c068 2019-02-10 stsp arg.id_str = id_str;
1155 3fc0c068 2019-02-10 stsp arg.header_shown = 0;
1156 b72f483a 2019-02-05 stsp
1157 927df6b7 2019-02-10 stsp error = got_worktree_status(worktree, path, repo, print_diff,
1158 b72f483a 2019-02-05 stsp &arg, check_cancelled, NULL);
1159 b72f483a 2019-02-05 stsp free(id_str);
1160 b72f483a 2019-02-05 stsp goto done;
1161 b72f483a 2019-02-05 stsp }
1162 b72f483a 2019-02-05 stsp
1163 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&id1, repo, id_str1);
1164 6402fb3c 2018-09-15 stsp if (error)
1165 b00d56cd 2018-04-01 stsp goto done;
1166 b00d56cd 2018-04-01 stsp
1167 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&id2, repo, id_str2);
1168 6402fb3c 2018-09-15 stsp if (error)
1169 b00d56cd 2018-04-01 stsp goto done;
1170 b00d56cd 2018-04-01 stsp
1171 15a94983 2018-12-23 stsp error = got_object_get_type(&type1, repo, id1);
1172 15a94983 2018-12-23 stsp if (error)
1173 15a94983 2018-12-23 stsp goto done;
1174 15a94983 2018-12-23 stsp
1175 15a94983 2018-12-23 stsp error = got_object_get_type(&type2, repo, id2);
1176 15a94983 2018-12-23 stsp if (error)
1177 15a94983 2018-12-23 stsp goto done;
1178 15a94983 2018-12-23 stsp
1179 15a94983 2018-12-23 stsp if (type1 != type2) {
1180 b00d56cd 2018-04-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1181 b00d56cd 2018-04-01 stsp goto done;
1182 b00d56cd 2018-04-01 stsp }
1183 b00d56cd 2018-04-01 stsp
1184 15a94983 2018-12-23 stsp switch (type1) {
1185 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_BLOB:
1186 15a94983 2018-12-23 stsp error = got_diff_objects_as_blobs(id1, id2, NULL, NULL,
1187 54156555 2018-12-24 stsp diff_context, repo, stdout);
1188 b00d56cd 2018-04-01 stsp break;
1189 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_TREE:
1190 15a94983 2018-12-23 stsp error = got_diff_objects_as_trees(id1, id2, "", "",
1191 54156555 2018-12-24 stsp diff_context, repo, stdout);
1192 b00d56cd 2018-04-01 stsp break;
1193 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_COMMIT:
1194 15a94983 2018-12-23 stsp printf("diff %s %s\n", id_str1 ? id_str1 : "/dev/null",
1195 15a94983 2018-12-23 stsp id_str2);
1196 15a94983 2018-12-23 stsp error = got_diff_objects_as_commits(id1, id2, diff_context,
1197 c0cc5c62 2018-10-18 stsp repo, stdout);
1198 b00d56cd 2018-04-01 stsp break;
1199 b00d56cd 2018-04-01 stsp default:
1200 b00d56cd 2018-04-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1201 b00d56cd 2018-04-01 stsp }
1202 b00d56cd 2018-04-01 stsp
1203 b00d56cd 2018-04-01 stsp done:
1204 15a94983 2018-12-23 stsp free(id1);
1205 15a94983 2018-12-23 stsp free(id2);
1206 927df6b7 2019-02-10 stsp free(path);
1207 b72f483a 2019-02-05 stsp if (worktree)
1208 b72f483a 2019-02-05 stsp got_worktree_close(worktree);
1209 ad242220 2018-09-08 stsp if (repo) {
1210 ad242220 2018-09-08 stsp const struct got_error *repo_error;
1211 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
1212 ad242220 2018-09-08 stsp if (error == NULL)
1213 ad242220 2018-09-08 stsp error = repo_error;
1214 ad242220 2018-09-08 stsp }
1215 b00d56cd 2018-04-01 stsp return error;
1216 404c43c4 2018-06-21 stsp }
1217 404c43c4 2018-06-21 stsp
1218 404c43c4 2018-06-21 stsp __dead static void
1219 404c43c4 2018-06-21 stsp usage_blame(void)
1220 404c43c4 2018-06-21 stsp {
1221 9270e621 2019-02-05 stsp fprintf(stderr,
1222 9270e621 2019-02-05 stsp "usage: %s blame [-c commit] [-r repository-path] path\n",
1223 404c43c4 2018-06-21 stsp getprogname());
1224 404c43c4 2018-06-21 stsp exit(1);
1225 b00d56cd 2018-04-01 stsp }
1226 b00d56cd 2018-04-01 stsp
1227 404c43c4 2018-06-21 stsp static const struct got_error *
1228 404c43c4 2018-06-21 stsp cmd_blame(int argc, char *argv[])
1229 404c43c4 2018-06-21 stsp {
1230 404c43c4 2018-06-21 stsp const struct got_error *error;
1231 404c43c4 2018-06-21 stsp struct got_repository *repo = NULL;
1232 0c06baac 2019-02-05 stsp struct got_worktree *worktree = NULL;
1233 66bea077 2018-08-02 stsp char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
1234 404c43c4 2018-06-21 stsp struct got_object_id *commit_id = NULL;
1235 404c43c4 2018-06-21 stsp char *commit_id_str = NULL;
1236 404c43c4 2018-06-21 stsp int ch;
1237 404c43c4 2018-06-21 stsp
1238 404c43c4 2018-06-21 stsp #ifndef PROFILE
1239 36e2fb66 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1240 36e2fb66 2019-01-04 stsp NULL) == -1)
1241 404c43c4 2018-06-21 stsp err(1, "pledge");
1242 404c43c4 2018-06-21 stsp #endif
1243 404c43c4 2018-06-21 stsp
1244 66bea077 2018-08-02 stsp while ((ch = getopt(argc, argv, "c:r:")) != -1) {
1245 404c43c4 2018-06-21 stsp switch (ch) {
1246 404c43c4 2018-06-21 stsp case 'c':
1247 404c43c4 2018-06-21 stsp commit_id_str = optarg;
1248 404c43c4 2018-06-21 stsp break;
1249 66bea077 2018-08-02 stsp case 'r':
1250 66bea077 2018-08-02 stsp repo_path = realpath(optarg, NULL);
1251 66bea077 2018-08-02 stsp if (repo_path == NULL)
1252 66bea077 2018-08-02 stsp err(1, "-r option");
1253 66bea077 2018-08-02 stsp break;
1254 404c43c4 2018-06-21 stsp default:
1255 404c43c4 2018-06-21 stsp usage();
1256 404c43c4 2018-06-21 stsp /* NOTREACHED */
1257 404c43c4 2018-06-21 stsp }
1258 404c43c4 2018-06-21 stsp }
1259 404c43c4 2018-06-21 stsp
1260 404c43c4 2018-06-21 stsp argc -= optind;
1261 404c43c4 2018-06-21 stsp argv += optind;
1262 404c43c4 2018-06-21 stsp
1263 a39318fd 2018-08-02 stsp if (argc == 1)
1264 404c43c4 2018-06-21 stsp path = argv[0];
1265 a39318fd 2018-08-02 stsp else
1266 404c43c4 2018-06-21 stsp usage_blame();
1267 404c43c4 2018-06-21 stsp
1268 66bea077 2018-08-02 stsp cwd = getcwd(NULL, 0);
1269 66bea077 2018-08-02 stsp if (cwd == NULL) {
1270 66bea077 2018-08-02 stsp error = got_error_from_errno();
1271 66bea077 2018-08-02 stsp goto done;
1272 66bea077 2018-08-02 stsp }
1273 66bea077 2018-08-02 stsp if (repo_path == NULL) {
1274 0c06baac 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
1275 0c06baac 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1276 66bea077 2018-08-02 stsp goto done;
1277 0c06baac 2019-02-05 stsp else
1278 0c06baac 2019-02-05 stsp error = NULL;
1279 0c06baac 2019-02-05 stsp if (worktree) {
1280 0c06baac 2019-02-05 stsp repo_path =
1281 0c06baac 2019-02-05 stsp strdup(got_worktree_get_repo_path(worktree));
1282 0c06baac 2019-02-05 stsp if (repo_path == NULL)
1283 0c06baac 2019-02-05 stsp error = got_error_from_errno();
1284 0c06baac 2019-02-05 stsp if (error)
1285 0c06baac 2019-02-05 stsp goto done;
1286 0c06baac 2019-02-05 stsp } else {
1287 0c06baac 2019-02-05 stsp repo_path = strdup(cwd);
1288 0c06baac 2019-02-05 stsp if (repo_path == NULL) {
1289 0c06baac 2019-02-05 stsp error = got_error_from_errno();
1290 0c06baac 2019-02-05 stsp goto done;
1291 0c06baac 2019-02-05 stsp }
1292 66bea077 2018-08-02 stsp }
1293 66bea077 2018-08-02 stsp }
1294 36e2fb66 2019-01-04 stsp
1295 36e2fb66 2019-01-04 stsp error = apply_unveil(repo_path, NULL);
1296 36e2fb66 2019-01-04 stsp if (error)
1297 36e2fb66 2019-01-04 stsp goto done;
1298 66bea077 2018-08-02 stsp
1299 404c43c4 2018-06-21 stsp error = got_repo_open(&repo, repo_path);
1300 404c43c4 2018-06-21 stsp if (error != NULL)
1301 404c43c4 2018-06-21 stsp goto done;
1302 404c43c4 2018-06-21 stsp
1303 0c06baac 2019-02-05 stsp if (worktree) {
1304 6efaaa2d 2019-02-05 stsp const char *prefix = got_worktree_get_path_prefix(worktree);
1305 0c06baac 2019-02-05 stsp char *p, *worktree_subdir = cwd +
1306 0c06baac 2019-02-05 stsp strlen(got_worktree_get_root_path(worktree));
1307 6efaaa2d 2019-02-05 stsp if (asprintf(&p, "%s%s%s%s%s",
1308 6efaaa2d 2019-02-05 stsp prefix, (strcmp(prefix, "/") != 0) ? "/" : "",
1309 6efaaa2d 2019-02-05 stsp worktree_subdir, worktree_subdir[0] ? "/" : "",
1310 6efaaa2d 2019-02-05 stsp path) == -1) {
1311 0c06baac 2019-02-05 stsp error = got_error_from_errno();
1312 0c06baac 2019-02-05 stsp goto done;
1313 0c06baac 2019-02-05 stsp }
1314 6efaaa2d 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, p, 0);
1315 0c06baac 2019-02-05 stsp free(p);
1316 0c06baac 2019-02-05 stsp } else {
1317 0c06baac 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
1318 0c06baac 2019-02-05 stsp }
1319 0c06baac 2019-02-05 stsp if (error)
1320 66bea077 2018-08-02 stsp goto done;
1321 66bea077 2018-08-02 stsp
1322 404c43c4 2018-06-21 stsp if (commit_id_str == NULL) {
1323 404c43c4 2018-06-21 stsp struct got_reference *head_ref;
1324 404c43c4 2018-06-21 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1325 404c43c4 2018-06-21 stsp if (error != NULL)
1326 66bea077 2018-08-02 stsp goto done;
1327 404c43c4 2018-06-21 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
1328 404c43c4 2018-06-21 stsp got_ref_close(head_ref);
1329 404c43c4 2018-06-21 stsp if (error != NULL)
1330 66bea077 2018-08-02 stsp goto done;
1331 404c43c4 2018-06-21 stsp } else {
1332 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&commit_id, repo,
1333 15a94983 2018-12-23 stsp commit_id_str);
1334 404c43c4 2018-06-21 stsp if (error != NULL)
1335 66bea077 2018-08-02 stsp goto done;
1336 404c43c4 2018-06-21 stsp }
1337 404c43c4 2018-06-21 stsp
1338 66bea077 2018-08-02 stsp error = got_blame(in_repo_path, commit_id, repo, stdout);
1339 404c43c4 2018-06-21 stsp done:
1340 66bea077 2018-08-02 stsp free(in_repo_path);
1341 66bea077 2018-08-02 stsp free(repo_path);
1342 66bea077 2018-08-02 stsp free(cwd);
1343 404c43c4 2018-06-21 stsp free(commit_id);
1344 0c06baac 2019-02-05 stsp if (worktree)
1345 0c06baac 2019-02-05 stsp got_worktree_close(worktree);
1346 ad242220 2018-09-08 stsp if (repo) {
1347 ad242220 2018-09-08 stsp const struct got_error *repo_error;
1348 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
1349 ad242220 2018-09-08 stsp if (error == NULL)
1350 ad242220 2018-09-08 stsp error = repo_error;
1351 ad242220 2018-09-08 stsp }
1352 404c43c4 2018-06-21 stsp return error;
1353 5de5890b 2018-10-18 stsp }
1354 5de5890b 2018-10-18 stsp
1355 5de5890b 2018-10-18 stsp __dead static void
1356 5de5890b 2018-10-18 stsp usage_tree(void)
1357 5de5890b 2018-10-18 stsp {
1358 c1669e2e 2019-01-09 stsp fprintf(stderr,
1359 c1669e2e 2019-01-09 stsp "usage: %s tree [-c commit] [-r repository-path] [-iR] path\n",
1360 5de5890b 2018-10-18 stsp getprogname());
1361 5de5890b 2018-10-18 stsp exit(1);
1362 5de5890b 2018-10-18 stsp }
1363 5de5890b 2018-10-18 stsp
1364 c1669e2e 2019-01-09 stsp static void
1365 c1669e2e 2019-01-09 stsp print_entry(struct got_tree_entry *te, const char *id, const char *path,
1366 c1669e2e 2019-01-09 stsp const char *root_path)
1367 c1669e2e 2019-01-09 stsp {
1368 c1669e2e 2019-01-09 stsp int is_root_path = (strcmp(path, root_path) == 0);
1369 5de5890b 2018-10-18 stsp
1370 c1669e2e 2019-01-09 stsp path += strlen(root_path);
1371 c1669e2e 2019-01-09 stsp while (path[0] == '/')
1372 c1669e2e 2019-01-09 stsp path++;
1373 c1669e2e 2019-01-09 stsp
1374 c1669e2e 2019-01-09 stsp printf("%s%s%s%s%s\n", id ? id : "", path,
1375 d6e648b4 2019-02-10 stsp is_root_path ? "" : "/", te->name,
1376 d6e648b4 2019-02-10 stsp S_ISDIR(te->mode) ? "/" : ((te->mode & S_IXUSR) ? "*" : ""));
1377 c1669e2e 2019-01-09 stsp }
1378 c1669e2e 2019-01-09 stsp
1379 5de5890b 2018-10-18 stsp static const struct got_error *
1380 5de5890b 2018-10-18 stsp print_tree(const char *path, struct got_object_id *commit_id,
1381 c1669e2e 2019-01-09 stsp int show_ids, int recurse, const char *root_path,
1382 c1669e2e 2019-01-09 stsp struct got_repository *repo)
1383 5de5890b 2018-10-18 stsp {
1384 5de5890b 2018-10-18 stsp const struct got_error *err = NULL;
1385 5de5890b 2018-10-18 stsp struct got_object_id *tree_id = NULL;
1386 5de5890b 2018-10-18 stsp struct got_tree_object *tree = NULL;
1387 5de5890b 2018-10-18 stsp const struct got_tree_entries *entries;
1388 5de5890b 2018-10-18 stsp struct got_tree_entry *te;
1389 5de5890b 2018-10-18 stsp
1390 5de5890b 2018-10-18 stsp err = got_object_id_by_path(&tree_id, repo, commit_id, path);
1391 5de5890b 2018-10-18 stsp if (err)
1392 5de5890b 2018-10-18 stsp goto done;
1393 5de5890b 2018-10-18 stsp
1394 5de5890b 2018-10-18 stsp err = got_object_open_as_tree(&tree, repo, tree_id);
1395 5de5890b 2018-10-18 stsp if (err)
1396 5de5890b 2018-10-18 stsp goto done;
1397 5de5890b 2018-10-18 stsp entries = got_object_tree_get_entries(tree);
1398 5de5890b 2018-10-18 stsp te = SIMPLEQ_FIRST(&entries->head);
1399 5de5890b 2018-10-18 stsp while (te) {
1400 5de5890b 2018-10-18 stsp char *id = NULL;
1401 84453469 2018-11-11 stsp
1402 84453469 2018-11-11 stsp if (sigint_received || sigpipe_received)
1403 84453469 2018-11-11 stsp break;
1404 84453469 2018-11-11 stsp
1405 5de5890b 2018-10-18 stsp if (show_ids) {
1406 5de5890b 2018-10-18 stsp char *id_str;
1407 5de5890b 2018-10-18 stsp err = got_object_id_str(&id_str, te->id);
1408 5de5890b 2018-10-18 stsp if (err)
1409 5de5890b 2018-10-18 stsp goto done;
1410 5de5890b 2018-10-18 stsp if (asprintf(&id, "%s ", id_str) == -1) {
1411 5de5890b 2018-10-18 stsp err = got_error_from_errno();
1412 5de5890b 2018-10-18 stsp free(id_str);
1413 5de5890b 2018-10-18 stsp goto done;
1414 5de5890b 2018-10-18 stsp }
1415 5de5890b 2018-10-18 stsp free(id_str);
1416 5de5890b 2018-10-18 stsp }
1417 c1669e2e 2019-01-09 stsp print_entry(te, id, path, root_path);
1418 5de5890b 2018-10-18 stsp free(id);
1419 c1669e2e 2019-01-09 stsp
1420 c1669e2e 2019-01-09 stsp if (recurse && S_ISDIR(te->mode)) {
1421 c1669e2e 2019-01-09 stsp char *child_path;
1422 c1669e2e 2019-01-09 stsp if (asprintf(&child_path, "%s%s%s", path,
1423 c1669e2e 2019-01-09 stsp path[0] == '/' && path[1] == '\0' ? "" : "/",
1424 c1669e2e 2019-01-09 stsp te->name) == -1) {
1425 c1669e2e 2019-01-09 stsp err = got_error_from_errno();
1426 c1669e2e 2019-01-09 stsp goto done;
1427 c1669e2e 2019-01-09 stsp }
1428 c1669e2e 2019-01-09 stsp err = print_tree(child_path, commit_id, show_ids, 1,
1429 c1669e2e 2019-01-09 stsp root_path, repo);
1430 c1669e2e 2019-01-09 stsp free(child_path);
1431 c1669e2e 2019-01-09 stsp if (err)
1432 c1669e2e 2019-01-09 stsp goto done;
1433 c1669e2e 2019-01-09 stsp }
1434 c1669e2e 2019-01-09 stsp
1435 c1669e2e 2019-01-09 stsp te = SIMPLEQ_NEXT(te, entry);
1436 5de5890b 2018-10-18 stsp }
1437 5de5890b 2018-10-18 stsp done:
1438 5de5890b 2018-10-18 stsp if (tree)
1439 5de5890b 2018-10-18 stsp got_object_tree_close(tree);
1440 5de5890b 2018-10-18 stsp free(tree_id);
1441 5de5890b 2018-10-18 stsp return err;
1442 404c43c4 2018-06-21 stsp }
1443 404c43c4 2018-06-21 stsp
1444 5de5890b 2018-10-18 stsp static const struct got_error *
1445 5de5890b 2018-10-18 stsp cmd_tree(int argc, char *argv[])
1446 5de5890b 2018-10-18 stsp {
1447 5de5890b 2018-10-18 stsp const struct got_error *error;
1448 5de5890b 2018-10-18 stsp struct got_repository *repo = NULL;
1449 7a2c19d6 2019-02-05 stsp struct got_worktree *worktree = NULL;
1450 7a2c19d6 2019-02-05 stsp const char *path;
1451 7a2c19d6 2019-02-05 stsp char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
1452 5de5890b 2018-10-18 stsp struct got_object_id *commit_id = NULL;
1453 5de5890b 2018-10-18 stsp char *commit_id_str = NULL;
1454 c1669e2e 2019-01-09 stsp int show_ids = 0, recurse = 0;
1455 5de5890b 2018-10-18 stsp int ch;
1456 5de5890b 2018-10-18 stsp
1457 5de5890b 2018-10-18 stsp #ifndef PROFILE
1458 0f8d269b 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1459 0f8d269b 2019-01-04 stsp NULL) == -1)
1460 5de5890b 2018-10-18 stsp err(1, "pledge");
1461 5de5890b 2018-10-18 stsp #endif
1462 5de5890b 2018-10-18 stsp
1463 c1669e2e 2019-01-09 stsp while ((ch = getopt(argc, argv, "c:r:iR")) != -1) {
1464 5de5890b 2018-10-18 stsp switch (ch) {
1465 5de5890b 2018-10-18 stsp case 'c':
1466 5de5890b 2018-10-18 stsp commit_id_str = optarg;
1467 5de5890b 2018-10-18 stsp break;
1468 5de5890b 2018-10-18 stsp case 'r':
1469 5de5890b 2018-10-18 stsp repo_path = realpath(optarg, NULL);
1470 5de5890b 2018-10-18 stsp if (repo_path == NULL)
1471 5de5890b 2018-10-18 stsp err(1, "-r option");
1472 5de5890b 2018-10-18 stsp break;
1473 5de5890b 2018-10-18 stsp case 'i':
1474 5de5890b 2018-10-18 stsp show_ids = 1;
1475 5de5890b 2018-10-18 stsp break;
1476 c1669e2e 2019-01-09 stsp case 'R':
1477 c1669e2e 2019-01-09 stsp recurse = 1;
1478 c1669e2e 2019-01-09 stsp break;
1479 5de5890b 2018-10-18 stsp default:
1480 5de5890b 2018-10-18 stsp usage();
1481 5de5890b 2018-10-18 stsp /* NOTREACHED */
1482 5de5890b 2018-10-18 stsp }
1483 5de5890b 2018-10-18 stsp }
1484 5de5890b 2018-10-18 stsp
1485 5de5890b 2018-10-18 stsp argc -= optind;
1486 5de5890b 2018-10-18 stsp argv += optind;
1487 5de5890b 2018-10-18 stsp
1488 5de5890b 2018-10-18 stsp if (argc == 1)
1489 5de5890b 2018-10-18 stsp path = argv[0];
1490 5de5890b 2018-10-18 stsp else if (argc > 1)
1491 5de5890b 2018-10-18 stsp usage_tree();
1492 5de5890b 2018-10-18 stsp else
1493 7a2c19d6 2019-02-05 stsp path = NULL;
1494 7a2c19d6 2019-02-05 stsp
1495 9bf7a39b 2019-02-05 stsp cwd = getcwd(NULL, 0);
1496 9bf7a39b 2019-02-05 stsp if (cwd == NULL) {
1497 9bf7a39b 2019-02-05 stsp error = got_error_from_errno();
1498 9bf7a39b 2019-02-05 stsp goto done;
1499 9bf7a39b 2019-02-05 stsp }
1500 5de5890b 2018-10-18 stsp if (repo_path == NULL) {
1501 7a2c19d6 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
1502 8994de28 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1503 7a2c19d6 2019-02-05 stsp goto done;
1504 7a2c19d6 2019-02-05 stsp else
1505 7a2c19d6 2019-02-05 stsp error = NULL;
1506 7a2c19d6 2019-02-05 stsp if (worktree) {
1507 7a2c19d6 2019-02-05 stsp repo_path =
1508 7a2c19d6 2019-02-05 stsp strdup(got_worktree_get_repo_path(worktree));
1509 7a2c19d6 2019-02-05 stsp if (repo_path == NULL)
1510 7a2c19d6 2019-02-05 stsp error = got_error_from_errno();
1511 7a2c19d6 2019-02-05 stsp if (error)
1512 7a2c19d6 2019-02-05 stsp goto done;
1513 7a2c19d6 2019-02-05 stsp } else {
1514 7a2c19d6 2019-02-05 stsp repo_path = strdup(cwd);
1515 7a2c19d6 2019-02-05 stsp if (repo_path == NULL) {
1516 7a2c19d6 2019-02-05 stsp error = got_error_from_errno();
1517 7a2c19d6 2019-02-05 stsp goto done;
1518 7a2c19d6 2019-02-05 stsp }
1519 5de5890b 2018-10-18 stsp }
1520 5de5890b 2018-10-18 stsp }
1521 5de5890b 2018-10-18 stsp
1522 0f8d269b 2019-01-04 stsp error = apply_unveil(repo_path, NULL);
1523 0f8d269b 2019-01-04 stsp if (error)
1524 0f8d269b 2019-01-04 stsp goto done;
1525 0f8d269b 2019-01-04 stsp
1526 5de5890b 2018-10-18 stsp error = got_repo_open(&repo, repo_path);
1527 5de5890b 2018-10-18 stsp if (error != NULL)
1528 5de5890b 2018-10-18 stsp goto done;
1529 5de5890b 2018-10-18 stsp
1530 9bf7a39b 2019-02-05 stsp if (path == NULL) {
1531 9bf7a39b 2019-02-05 stsp if (worktree) {
1532 9bf7a39b 2019-02-05 stsp char *p, *worktree_subdir = cwd +
1533 9bf7a39b 2019-02-05 stsp strlen(got_worktree_get_root_path(worktree));
1534 9bf7a39b 2019-02-05 stsp if (asprintf(&p, "%s/%s",
1535 9bf7a39b 2019-02-05 stsp got_worktree_get_path_prefix(worktree),
1536 9bf7a39b 2019-02-05 stsp worktree_subdir) == -1) {
1537 9bf7a39b 2019-02-05 stsp error = got_error_from_errno();
1538 9bf7a39b 2019-02-05 stsp goto done;
1539 9bf7a39b 2019-02-05 stsp }
1540 9bf7a39b 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, p, 1);
1541 9bf7a39b 2019-02-05 stsp free(p);
1542 9bf7a39b 2019-02-05 stsp if (error)
1543 9bf7a39b 2019-02-05 stsp goto done;
1544 9bf7a39b 2019-02-05 stsp } else
1545 9bf7a39b 2019-02-05 stsp path = "/";
1546 9bf7a39b 2019-02-05 stsp }
1547 9bf7a39b 2019-02-05 stsp if (in_repo_path == NULL) {
1548 9bf7a39b 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
1549 9bf7a39b 2019-02-05 stsp if (error != NULL)
1550 9bf7a39b 2019-02-05 stsp goto done;
1551 9bf7a39b 2019-02-05 stsp }
1552 5de5890b 2018-10-18 stsp
1553 5de5890b 2018-10-18 stsp if (commit_id_str == NULL) {
1554 5de5890b 2018-10-18 stsp struct got_reference *head_ref;
1555 5de5890b 2018-10-18 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1556 5de5890b 2018-10-18 stsp if (error != NULL)
1557 5de5890b 2018-10-18 stsp goto done;
1558 5de5890b 2018-10-18 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
1559 5de5890b 2018-10-18 stsp got_ref_close(head_ref);
1560 5de5890b 2018-10-18 stsp if (error != NULL)
1561 5de5890b 2018-10-18 stsp goto done;
1562 5de5890b 2018-10-18 stsp } else {
1563 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&commit_id, repo,
1564 15a94983 2018-12-23 stsp commit_id_str);
1565 5de5890b 2018-10-18 stsp if (error != NULL)
1566 5de5890b 2018-10-18 stsp goto done;
1567 5de5890b 2018-10-18 stsp }
1568 5de5890b 2018-10-18 stsp
1569 c1669e2e 2019-01-09 stsp error = print_tree(in_repo_path, commit_id, show_ids, recurse,
1570 c1669e2e 2019-01-09 stsp in_repo_path, repo);
1571 5de5890b 2018-10-18 stsp done:
1572 5de5890b 2018-10-18 stsp free(in_repo_path);
1573 5de5890b 2018-10-18 stsp free(repo_path);
1574 5de5890b 2018-10-18 stsp free(cwd);
1575 5de5890b 2018-10-18 stsp free(commit_id);
1576 7a2c19d6 2019-02-05 stsp if (worktree)
1577 7a2c19d6 2019-02-05 stsp got_worktree_close(worktree);
1578 5de5890b 2018-10-18 stsp if (repo) {
1579 5de5890b 2018-10-18 stsp const struct got_error *repo_error;
1580 5de5890b 2018-10-18 stsp repo_error = got_repo_close(repo);
1581 5de5890b 2018-10-18 stsp if (error == NULL)
1582 5de5890b 2018-10-18 stsp error = repo_error;
1583 5de5890b 2018-10-18 stsp }
1584 5de5890b 2018-10-18 stsp return error;
1585 5de5890b 2018-10-18 stsp }
1586 5de5890b 2018-10-18 stsp
1587 6bad629b 2019-02-04 stsp __dead static void
1588 6bad629b 2019-02-04 stsp usage_status(void)
1589 6bad629b 2019-02-04 stsp {
1590 927df6b7 2019-02-10 stsp fprintf(stderr, "usage: %s status [path]\n", getprogname());
1591 6bad629b 2019-02-04 stsp exit(1);
1592 6bad629b 2019-02-04 stsp }
1593 5c860e29 2018-03-12 stsp
1594 b72f483a 2019-02-05 stsp static const struct got_error *
1595 b72f483a 2019-02-05 stsp print_status(void *arg, unsigned char status, const char *path,
1596 b72f483a 2019-02-05 stsp struct got_object_id *id)
1597 6bad629b 2019-02-04 stsp {
1598 6bad629b 2019-02-04 stsp printf("%c %s\n", status, path);
1599 b72f483a 2019-02-05 stsp return NULL;
1600 6bad629b 2019-02-04 stsp }
1601 5c860e29 2018-03-12 stsp
1602 6bad629b 2019-02-04 stsp static const struct got_error *
1603 6bad629b 2019-02-04 stsp cmd_status(int argc, char *argv[])
1604 6bad629b 2019-02-04 stsp {
1605 6bad629b 2019-02-04 stsp const struct got_error *error = NULL;
1606 6bad629b 2019-02-04 stsp struct got_repository *repo = NULL;
1607 6bad629b 2019-02-04 stsp struct got_worktree *worktree = NULL;
1608 927df6b7 2019-02-10 stsp char *cwd = NULL, *path = NULL;
1609 6bad629b 2019-02-04 stsp int ch;
1610 5c860e29 2018-03-12 stsp
1611 6bad629b 2019-02-04 stsp while ((ch = getopt(argc, argv, "")) != -1) {
1612 6bad629b 2019-02-04 stsp switch (ch) {
1613 5c860e29 2018-03-12 stsp default:
1614 6bad629b 2019-02-04 stsp usage();
1615 6bad629b 2019-02-04 stsp /* NOTREACHED */
1616 5c860e29 2018-03-12 stsp }
1617 5c860e29 2018-03-12 stsp }
1618 5c860e29 2018-03-12 stsp
1619 6bad629b 2019-02-04 stsp argc -= optind;
1620 6bad629b 2019-02-04 stsp argv += optind;
1621 5c860e29 2018-03-12 stsp
1622 6bad629b 2019-02-04 stsp #ifndef PROFILE
1623 6bad629b 2019-02-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1624 6bad629b 2019-02-04 stsp NULL) == -1)
1625 6bad629b 2019-02-04 stsp err(1, "pledge");
1626 f42b1b34 2018-03-12 stsp #endif
1627 927df6b7 2019-02-10 stsp cwd = getcwd(NULL, 0);
1628 927df6b7 2019-02-10 stsp if (cwd == NULL) {
1629 927df6b7 2019-02-10 stsp error = got_error_from_errno();
1630 927df6b7 2019-02-10 stsp goto done;
1631 927df6b7 2019-02-10 stsp }
1632 927df6b7 2019-02-10 stsp
1633 927df6b7 2019-02-10 stsp error = got_worktree_open(&worktree, cwd);
1634 927df6b7 2019-02-10 stsp if (error != NULL)
1635 927df6b7 2019-02-10 stsp goto done;
1636 927df6b7 2019-02-10 stsp
1637 6bad629b 2019-02-04 stsp if (argc == 0) {
1638 927df6b7 2019-02-10 stsp path = strdup("");
1639 927df6b7 2019-02-10 stsp if (path == NULL) {
1640 6bad629b 2019-02-04 stsp error = got_error_from_errno();
1641 6bad629b 2019-02-04 stsp goto done;
1642 6bad629b 2019-02-04 stsp }
1643 6bad629b 2019-02-04 stsp } else if (argc == 1) {
1644 927df6b7 2019-02-10 stsp error = get_status_path(&path, worktree, argv[0]);
1645 927df6b7 2019-02-10 stsp if (error)
1646 6bad629b 2019-02-04 stsp goto done;
1647 6bad629b 2019-02-04 stsp } else
1648 6bad629b 2019-02-04 stsp usage_status();
1649 6bad629b 2019-02-04 stsp
1650 6bad629b 2019-02-04 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
1651 6bad629b 2019-02-04 stsp if (error != NULL)
1652 6bad629b 2019-02-04 stsp goto done;
1653 6bad629b 2019-02-04 stsp
1654 c7f4312f 2019-02-05 stsp error = apply_unveil(got_repo_get_path(repo),
1655 c7f4312f 2019-02-05 stsp got_worktree_get_root_path(worktree));
1656 6bad629b 2019-02-04 stsp if (error)
1657 6bad629b 2019-02-04 stsp goto done;
1658 6bad629b 2019-02-04 stsp
1659 927df6b7 2019-02-10 stsp error = got_worktree_status(worktree, path, repo, print_status, NULL,
1660 6bad629b 2019-02-04 stsp check_cancelled, NULL);
1661 6bad629b 2019-02-04 stsp done:
1662 927df6b7 2019-02-10 stsp free(cwd);
1663 927df6b7 2019-02-10 stsp free(path);
1664 6bad629b 2019-02-04 stsp return error;
1665 6bad629b 2019-02-04 stsp }