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>
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.
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.
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>
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>
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"
44 5c860e29 2018-03-12 stsp #ifndef nitems
45 5c860e29 2018-03-12 stsp #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
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;
52 99437157 2018-11-11 stsp catch_sigint(int signo)
54 99437157 2018-11-11 stsp sigint_received = 1;
58 99437157 2018-11-11 stsp catch_sigpipe(int signo)
60 99437157 2018-11-11 stsp sigpipe_received = 1;
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;
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 d0eebce4 2019-03-11 stsp __dead static void usage_ref(void);
80 d00136be 2019-03-26 stsp __dead static void usage_add(void);
81 2ec1f75b 2019-03-26 stsp __dead static void usage_rm(void);
82 a129376b 2019-03-28 stsp __dead static void usage_revert(void);
84 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_checkout(int, char *[]);
85 507dc3bb 2018-12-29 stsp static const struct got_error* cmd_update(int, char *[]);
86 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_log(int, char *[]);
87 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_diff(int, char *[]);
88 404c43c4 2018-06-21 stsp static const struct got_error* cmd_blame(int, char *[]);
89 5de5890b 2018-10-18 stsp static const struct got_error* cmd_tree(int, char *[]);
90 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_status(int, char *[]);
91 d0eebce4 2019-03-11 stsp static const struct got_error* cmd_ref(int, char *[]);
92 d00136be 2019-03-26 stsp static const struct got_error* cmd_add(int, char *[]);
93 2ec1f75b 2019-03-26 stsp static const struct got_error* cmd_rm(int, char *[]);
94 a129376b 2019-03-28 stsp static const struct got_error* cmd_revert(int, char *[]);
96 4ed7e80c 2018-05-20 stsp static struct cmd got_commands[] = {
97 c09a553d 2018-03-12 stsp { "checkout", cmd_checkout, usage_checkout,
98 0bb8a95e 2018-03-12 stsp "check out a new work tree from a repository" },
99 507dc3bb 2018-12-29 stsp { "update", cmd_update, usage_update,
100 507dc3bb 2018-12-29 stsp "update a work tree to a different commit" },
101 1b6b95a8 2018-03-12 stsp { "log", cmd_log, usage_log,
102 1b6b95a8 2018-03-12 stsp "show repository history" },
103 b00d56cd 2018-04-01 stsp { "diff", cmd_diff, usage_diff,
104 b00d56cd 2018-04-01 stsp "compare files and directories" },
105 404c43c4 2018-06-21 stsp { "blame", cmd_blame, usage_blame,
106 a67e2392 2019-03-26 stsp "show when lines in a file were changed" },
107 5de5890b 2018-10-18 stsp { "tree", cmd_tree, usage_tree,
108 a67e2392 2019-03-26 stsp "list files and directories in repository" },
109 1b6b95a8 2018-03-12 stsp { "status", cmd_status, usage_status,
110 1b6b95a8 2018-03-12 stsp "show modification status of files" },
111 d0eebce4 2019-03-11 stsp { "ref", cmd_ref, usage_ref,
112 d0eebce4 2019-03-11 stsp "manage references in repository" },
113 d00136be 2019-03-26 stsp { "add", cmd_add, usage_add,
114 d00136be 2019-03-26 stsp "add a new file to version control" },
115 2ec1f75b 2019-03-26 stsp { "rm", cmd_rm, usage_rm,
116 2ec1f75b 2019-03-26 stsp "remove a versioned file" },
117 a129376b 2019-03-28 stsp { "revert", cmd_revert, usage_revert,
118 a129376b 2019-03-28 stsp "revert uncommitted changes" },
122 5c860e29 2018-03-12 stsp main(int argc, char *argv[])
124 5c860e29 2018-03-12 stsp struct cmd *cmd;
125 5c860e29 2018-03-12 stsp unsigned int i;
127 1b6b95a8 2018-03-12 stsp int hflag = 0;
129 289e3cbf 2019-02-04 stsp setlocale(LC_CTYPE, "");
131 1b6b95a8 2018-03-12 stsp while ((ch = getopt(argc, argv, "h")) != -1) {
132 5c860e29 2018-03-12 stsp switch (ch) {
138 5c860e29 2018-03-12 stsp /* NOTREACHED */
142 5c860e29 2018-03-12 stsp argc -= optind;
143 5c860e29 2018-03-12 stsp argv += optind;
144 1e70621d 2018-03-27 stsp optind = 0;
146 5c860e29 2018-03-12 stsp if (argc <= 0)
149 99437157 2018-11-11 stsp signal(SIGINT, catch_sigint);
150 99437157 2018-11-11 stsp signal(SIGPIPE, catch_sigpipe);
152 5c860e29 2018-03-12 stsp for (i = 0; i < nitems(got_commands); i++) {
153 d7d4f210 2018-03-12 stsp const struct got_error *error;
155 5c860e29 2018-03-12 stsp cmd = &got_commands[i];
157 5c860e29 2018-03-12 stsp if (strncmp(cmd->cmd_name, argv[0], strlen(argv[0])))
161 1b6b95a8 2018-03-12 stsp got_commands[i].cmd_usage();
163 d7d4f210 2018-03-12 stsp error = got_commands[i].cmd_main(argc, argv);
164 80d5f134 2018-11-11 stsp if (error && !(sigint_received || sigpipe_received)) {
165 d7d4f210 2018-03-12 stsp fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
172 20ecf764 2018-03-12 stsp fprintf(stderr, "%s: unknown command '%s'\n", getprogname(), argv[0]);
176 4ed7e80c 2018-05-20 stsp __dead static void
177 5c860e29 2018-03-12 stsp usage(void)
181 987e94ba 2018-03-12 stsp fprintf(stderr, "usage: %s [-h] command [arg ...]\n\n"
182 987e94ba 2018-03-12 stsp "Available commands:\n", getprogname());
183 46a0db7d 2018-03-12 stsp for (i = 0; i < nitems(got_commands); i++) {
184 46a0db7d 2018-03-12 stsp struct cmd *cmd = &got_commands[i];
185 46a0db7d 2018-03-12 stsp fprintf(stderr, " %s: %s\n", cmd->cmd_name, cmd->cmd_descr);
190 0266afb7 2019-01-04 stsp static const struct got_error *
191 d0eebce4 2019-03-11 stsp apply_unveil(const char *repo_path, int repo_read_only,
192 d0eebce4 2019-03-11 stsp const char *worktree_path)
194 0266afb7 2019-01-04 stsp const struct got_error *error;
196 d0eebce4 2019-03-11 stsp if (repo_path && unveil(repo_path, repo_read_only ? "r" : "rwc") != 0)
197 0266afb7 2019-01-04 stsp return got_error_from_errno();
199 0266afb7 2019-01-04 stsp if (worktree_path && unveil(worktree_path, "rwc") != 0)
200 0266afb7 2019-01-04 stsp return got_error_from_errno();
202 f12d0dbe 2019-01-04 stsp if (unveil("/tmp", "rwc") != 0)
203 0266afb7 2019-01-04 stsp return got_error_from_errno();
205 0266afb7 2019-01-04 stsp error = got_privsep_unveil_exec_helpers();
206 0266afb7 2019-01-04 stsp if (error != NULL)
207 0266afb7 2019-01-04 stsp return error;
209 0266afb7 2019-01-04 stsp if (unveil(NULL, NULL) != 0)
210 0266afb7 2019-01-04 stsp return got_error_from_errno();
212 0266afb7 2019-01-04 stsp return NULL;
215 4ed7e80c 2018-05-20 stsp __dead static void
216 c09a553d 2018-03-12 stsp usage_checkout(void)
218 0bb8a95e 2018-03-12 stsp fprintf(stderr, "usage: %s checkout [-p prefix] repository-path "
219 0bb8a95e 2018-03-12 stsp "[worktree-path]\n", getprogname());
223 92a684f4 2018-03-12 stsp static void
224 a0eb853d 2018-12-29 stsp checkout_progress(void *arg, unsigned char status, const char *path)
226 92a684f4 2018-03-12 stsp char *worktree_path = arg;
228 92a684f4 2018-03-12 stsp while (path[0] == '/')
231 d7b62c98 2018-12-27 stsp printf("%c %s/%s\n", status, worktree_path, path);
234 99437157 2018-11-11 stsp static const struct got_error *
235 6bad629b 2019-02-04 stsp check_cancelled(void *arg)
237 99437157 2018-11-11 stsp if (sigint_received || sigpipe_received)
238 99437157 2018-11-11 stsp return got_error(GOT_ERR_CANCELLED);
239 99437157 2018-11-11 stsp return NULL;
242 8069f636 2019-01-12 stsp static const struct got_error *
243 8069f636 2019-01-12 stsp check_ancestry(struct got_worktree *worktree, struct got_object_id *commit_id,
244 8069f636 2019-01-12 stsp struct got_repository *repo)
246 8069f636 2019-01-12 stsp const struct got_error *err;
247 8069f636 2019-01-12 stsp struct got_reference *head_ref = NULL;
248 8069f636 2019-01-12 stsp struct got_object_id *head_commit_id = NULL;
249 8069f636 2019-01-12 stsp struct got_commit_graph *graph = NULL;
251 8069f636 2019-01-12 stsp head_ref = got_worktree_get_head_ref(worktree);
252 8069f636 2019-01-12 stsp if (head_ref == NULL)
253 8069f636 2019-01-12 stsp return got_error_from_errno();
255 8069f636 2019-01-12 stsp /* TODO: Check the reflog. The head ref may have been rebased. */
256 8069f636 2019-01-12 stsp err = got_ref_resolve(&head_commit_id, repo, head_ref);
260 8069f636 2019-01-12 stsp err = got_commit_graph_open(&graph, head_commit_id, "/", 1, repo);
264 8069f636 2019-01-12 stsp err = got_commit_graph_iter_start(graph, head_commit_id, repo);
267 8069f636 2019-01-12 stsp while (1) {
268 8069f636 2019-01-12 stsp struct got_object_id *id;
270 8069f636 2019-01-12 stsp if (sigint_received || sigpipe_received)
273 8069f636 2019-01-12 stsp err = got_commit_graph_iter_next(&id, graph);
275 8069f636 2019-01-12 stsp if (err->code == GOT_ERR_ITER_COMPLETED) {
276 8069f636 2019-01-12 stsp err = got_error(GOT_ERR_ANCESTRY);
279 8069f636 2019-01-12 stsp if (err->code != GOT_ERR_ITER_NEED_MORE)
281 8069f636 2019-01-12 stsp err = got_commit_graph_fetch_commits(graph, 1, repo);
287 8069f636 2019-01-12 stsp if (id == NULL)
289 8069f636 2019-01-12 stsp if (got_object_id_cmp(id, commit_id) == 0)
293 8069f636 2019-01-12 stsp if (head_ref)
294 8069f636 2019-01-12 stsp got_ref_close(head_ref);
296 8069f636 2019-01-12 stsp got_commit_graph_close(graph);
297 8069f636 2019-01-12 stsp return err;
301 4ed7e80c 2018-05-20 stsp static const struct got_error *
302 c09a553d 2018-03-12 stsp cmd_checkout(int argc, char *argv[])
304 c09a553d 2018-03-12 stsp const struct got_error *error = NULL;
305 c09a553d 2018-03-12 stsp struct got_repository *repo = NULL;
306 c09a553d 2018-03-12 stsp struct got_reference *head_ref = NULL;
307 c09a553d 2018-03-12 stsp struct got_worktree *worktree = NULL;
308 c09a553d 2018-03-12 stsp char *repo_path = NULL;
309 c09a553d 2018-03-12 stsp char *worktree_path = NULL;
310 0bb8a95e 2018-03-12 stsp const char *path_prefix = "";
311 8069f636 2019-01-12 stsp char *commit_id_str = NULL;
312 e5dc7198 2018-12-29 stsp int ch, same_path_prefix;
314 8069f636 2019-01-12 stsp while ((ch = getopt(argc, argv, "c:p:")) != -1) {
315 0bb8a95e 2018-03-12 stsp switch (ch) {
317 8069f636 2019-01-12 stsp commit_id_str = strdup(optarg);
318 8069f636 2019-01-12 stsp if (commit_id_str == NULL)
319 8069f636 2019-01-12 stsp return got_error_from_errno();
322 0bb8a95e 2018-03-12 stsp path_prefix = optarg;
325 2deda0b9 2019-03-07 stsp usage_checkout();
326 0bb8a95e 2018-03-12 stsp /* NOTREACHED */
330 0bb8a95e 2018-03-12 stsp argc -= optind;
331 0bb8a95e 2018-03-12 stsp argv += optind;
333 6715a751 2018-03-16 stsp #ifndef PROFILE
334 68ed9ba5 2019-02-10 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
335 68ed9ba5 2019-02-10 stsp "unveil", NULL) == -1)
336 c09a553d 2018-03-12 stsp err(1, "pledge");
338 0bb8a95e 2018-03-12 stsp if (argc == 1) {
339 c09a553d 2018-03-12 stsp char *cwd, *base, *dotgit;
340 76089277 2018-04-01 stsp repo_path = realpath(argv[0], NULL);
341 76089277 2018-04-01 stsp if (repo_path == NULL)
342 76089277 2018-04-01 stsp return got_error_from_errno();
343 c09a553d 2018-03-12 stsp cwd = getcwd(NULL, 0);
344 76089277 2018-04-01 stsp if (cwd == NULL) {
345 76089277 2018-04-01 stsp error = got_error_from_errno();
348 5d7c1dab 2018-04-01 stsp if (path_prefix[0])
349 5d7c1dab 2018-04-01 stsp base = basename(path_prefix);
351 5d7c1dab 2018-04-01 stsp base = basename(repo_path);
352 76089277 2018-04-01 stsp if (base == NULL) {
353 76089277 2018-04-01 stsp error = got_error_from_errno();
356 c09a553d 2018-03-12 stsp dotgit = strstr(base, ".git");
357 c09a553d 2018-03-12 stsp if (dotgit)
358 c09a553d 2018-03-12 stsp *dotgit = '\0';
359 c09a553d 2018-03-12 stsp if (asprintf(&worktree_path, "%s/%s", cwd, base) == -1) {
360 0a585a0d 2018-03-17 stsp error = got_error_from_errno();
365 0bb8a95e 2018-03-12 stsp } else if (argc == 2) {
366 76089277 2018-04-01 stsp repo_path = realpath(argv[0], NULL);
367 76089277 2018-04-01 stsp if (repo_path == NULL) {
368 76089277 2018-04-01 stsp error = got_error_from_errno();
371 f7b38925 2018-04-01 stsp worktree_path = realpath(argv[1], NULL);
372 76089277 2018-04-01 stsp if (worktree_path == NULL) {
373 76089277 2018-04-01 stsp error = got_error_from_errno();
377 c09a553d 2018-03-12 stsp usage_checkout();
379 0cd1c46a 2019-03-11 stsp error = apply_unveil(repo_path, 0, worktree_path);
383 c09a553d 2018-03-12 stsp error = got_repo_open(&repo, repo_path);
384 c09a553d 2018-03-12 stsp if (error != NULL)
387 c09a553d 2018-03-12 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
388 c09a553d 2018-03-12 stsp if (error != NULL)
391 0bb8a95e 2018-03-12 stsp error = got_worktree_init(worktree_path, head_ref, path_prefix, repo);
392 d70b8e30 2018-12-27 stsp if (error != NULL && !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
395 c09a553d 2018-03-12 stsp error = got_worktree_open(&worktree, worktree_path);
396 c09a553d 2018-03-12 stsp if (error != NULL)
399 e5dc7198 2018-12-29 stsp error = got_worktree_match_path_prefix(&same_path_prefix, worktree,
400 e5dc7198 2018-12-29 stsp path_prefix);
401 e5dc7198 2018-12-29 stsp if (error != NULL)
403 e5dc7198 2018-12-29 stsp if (!same_path_prefix) {
404 49520a32 2018-12-29 stsp error = got_error(GOT_ERR_PATH_PREFIX);
408 8069f636 2019-01-12 stsp if (commit_id_str) {
409 8069f636 2019-01-12 stsp struct got_object_id *commit_id;
410 8069f636 2019-01-12 stsp error = got_object_resolve_id_str(&commit_id, repo,
411 8069f636 2019-01-12 stsp commit_id_str);
412 8069f636 2019-01-12 stsp if (error != NULL)
414 8069f636 2019-01-12 stsp error = check_ancestry(worktree, commit_id, repo);
415 8069f636 2019-01-12 stsp if (error != NULL) {
416 8069f636 2019-01-12 stsp free(commit_id);
419 8069f636 2019-01-12 stsp error = got_worktree_set_base_commit_id(worktree, repo,
420 8069f636 2019-01-12 stsp commit_id);
421 8069f636 2019-01-12 stsp free(commit_id);
426 93a30277 2018-12-24 stsp error = got_worktree_checkout_files(worktree, repo,
427 6bad629b 2019-02-04 stsp checkout_progress, worktree_path, check_cancelled, NULL);
428 c09a553d 2018-03-12 stsp if (error != NULL)
431 b65ae19a 2018-04-24 stsp printf("Now shut up and hack\n");
434 8069f636 2019-01-12 stsp free(commit_id_str);
435 76089277 2018-04-01 stsp free(repo_path);
436 507dc3bb 2018-12-29 stsp free(worktree_path);
437 507dc3bb 2018-12-29 stsp return error;
440 507dc3bb 2018-12-29 stsp __dead static void
441 507dc3bb 2018-12-29 stsp usage_update(void)
443 507dc3bb 2018-12-29 stsp fprintf(stderr, "usage: %s update [-c commit] [worktree-path]\n",
444 507dc3bb 2018-12-29 stsp getprogname());
448 507dc3bb 2018-12-29 stsp static void
449 507dc3bb 2018-12-29 stsp update_progress(void *arg, unsigned char status, const char *path)
451 784955db 2019-01-12 stsp int *did_something = arg;
453 507dc3bb 2018-12-29 stsp if (status == GOT_STATUS_EXISTS)
456 1545c615 2019-02-10 stsp *did_something = 1;
457 507dc3bb 2018-12-29 stsp while (path[0] == '/')
459 507dc3bb 2018-12-29 stsp printf("%c %s\n", status, path);
462 be7061eb 2018-12-30 stsp static const struct got_error *
463 507dc3bb 2018-12-29 stsp cmd_update(int argc, char *argv[])
465 507dc3bb 2018-12-29 stsp const struct got_error *error = NULL;
466 507dc3bb 2018-12-29 stsp struct got_repository *repo = NULL;
467 507dc3bb 2018-12-29 stsp struct got_worktree *worktree = NULL;
468 507dc3bb 2018-12-29 stsp char *worktree_path = NULL;
469 507dc3bb 2018-12-29 stsp struct got_object_id *commit_id = NULL;
470 9c4b8182 2019-01-02 stsp char *commit_id_str = NULL;
471 784955db 2019-01-12 stsp int ch, did_something = 0;
473 507dc3bb 2018-12-29 stsp while ((ch = getopt(argc, argv, "c:")) != -1) {
474 507dc3bb 2018-12-29 stsp switch (ch) {
476 9c4b8182 2019-01-02 stsp commit_id_str = strdup(optarg);
477 9c4b8182 2019-01-02 stsp if (commit_id_str == NULL)
478 9c4b8182 2019-01-02 stsp return got_error_from_errno();
481 2deda0b9 2019-03-07 stsp usage_update();
482 507dc3bb 2018-12-29 stsp /* NOTREACHED */
486 507dc3bb 2018-12-29 stsp argc -= optind;
487 507dc3bb 2018-12-29 stsp argv += optind;
489 507dc3bb 2018-12-29 stsp #ifndef PROFILE
490 68ed9ba5 2019-02-10 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
491 68ed9ba5 2019-02-10 stsp "unveil", NULL) == -1)
492 507dc3bb 2018-12-29 stsp err(1, "pledge");
494 507dc3bb 2018-12-29 stsp if (argc == 0) {
495 507dc3bb 2018-12-29 stsp worktree_path = getcwd(NULL, 0);
496 507dc3bb 2018-12-29 stsp if (worktree_path == NULL) {
497 507dc3bb 2018-12-29 stsp error = got_error_from_errno();
500 507dc3bb 2018-12-29 stsp } else if (argc == 1) {
501 507dc3bb 2018-12-29 stsp worktree_path = realpath(argv[0], NULL);
502 507dc3bb 2018-12-29 stsp if (worktree_path == NULL) {
503 507dc3bb 2018-12-29 stsp error = got_error_from_errno();
507 507dc3bb 2018-12-29 stsp usage_update();
509 507dc3bb 2018-12-29 stsp error = got_worktree_open(&worktree, worktree_path);
510 507dc3bb 2018-12-29 stsp if (error != NULL)
513 507dc3bb 2018-12-29 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
514 507dc3bb 2018-12-29 stsp if (error != NULL)
517 97430839 2019-03-11 stsp error = apply_unveil(got_repo_get_path(repo), 0,
518 8c02d095 2019-02-05 stsp got_worktree_get_root_path(worktree));
522 507dc3bb 2018-12-29 stsp if (commit_id_str == NULL) {
523 507dc3bb 2018-12-29 stsp struct got_reference *head_ref;
524 507dc3bb 2018-12-29 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
525 507dc3bb 2018-12-29 stsp if (error != NULL)
527 507dc3bb 2018-12-29 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
528 9c4b8182 2019-01-02 stsp if (error != NULL)
530 9c4b8182 2019-01-02 stsp error = got_object_id_str(&commit_id_str, commit_id);
531 507dc3bb 2018-12-29 stsp if (error != NULL)
534 507dc3bb 2018-12-29 stsp error = got_object_resolve_id_str(&commit_id, repo,
535 507dc3bb 2018-12-29 stsp commit_id_str);
536 507dc3bb 2018-12-29 stsp if (error != NULL)
540 be7061eb 2018-12-30 stsp error = check_ancestry(worktree, commit_id, repo);
541 be7061eb 2018-12-30 stsp if (error != NULL)
544 507dc3bb 2018-12-29 stsp if (got_object_id_cmp(got_worktree_get_base_commit_id(worktree),
545 507dc3bb 2018-12-29 stsp commit_id) != 0) {
546 507dc3bb 2018-12-29 stsp error = got_worktree_set_base_commit_id(worktree, repo,
547 507dc3bb 2018-12-29 stsp commit_id);
552 507dc3bb 2018-12-29 stsp error = got_worktree_checkout_files(worktree, repo,
553 6bad629b 2019-02-04 stsp update_progress, &did_something, check_cancelled, NULL);
554 507dc3bb 2018-12-29 stsp if (error != NULL)
557 784955db 2019-01-12 stsp if (did_something)
558 784955db 2019-01-12 stsp printf("Updated to commit %s\n", commit_id_str);
560 784955db 2019-01-12 stsp printf("Already up-to-date\n");
562 c09a553d 2018-03-12 stsp free(worktree_path);
563 507dc3bb 2018-12-29 stsp free(commit_id);
564 9c4b8182 2019-01-02 stsp free(commit_id_str);
565 c09a553d 2018-03-12 stsp return error;
568 f42b1b34 2018-03-12 stsp static const struct got_error *
569 79109fed 2018-03-27 stsp print_patch(struct got_commit_object *commit, struct got_object_id *id,
570 c0cc5c62 2018-10-18 stsp int diff_context, struct got_repository *repo)
572 f42b1b34 2018-03-12 stsp const struct got_error *err = NULL;
573 79109fed 2018-03-27 stsp struct got_tree_object *tree1 = NULL, *tree2;
574 79f35eb3 2018-06-11 stsp struct got_object_qid *qid;
575 0f2b3dca 2018-12-22 stsp char *id_str1 = NULL, *id_str2;
577 45d799e2 2018-12-23 stsp err = got_object_open_as_tree(&tree2, repo,
578 45d799e2 2018-12-23 stsp got_object_commit_get_tree_id(commit));
580 79109fed 2018-03-27 stsp return err;
582 45d799e2 2018-12-23 stsp qid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
583 79f35eb3 2018-06-11 stsp if (qid != NULL) {
584 79109fed 2018-03-27 stsp struct got_commit_object *pcommit;
586 117e771c 2018-07-23 stsp err = got_object_open_as_commit(&pcommit, repo, qid->id);
588 79109fed 2018-03-27 stsp return err;
590 45d799e2 2018-12-23 stsp err = got_object_open_as_tree(&tree1, repo,
591 45d799e2 2018-12-23 stsp got_object_commit_get_tree_id(pcommit));
592 79109fed 2018-03-27 stsp got_object_commit_close(pcommit);
594 0f2b3dca 2018-12-22 stsp return err;
596 0f2b3dca 2018-12-22 stsp err = got_object_id_str(&id_str1, qid->id);
598 79109fed 2018-03-27 stsp return err;
601 0f2b3dca 2018-12-22 stsp err = got_object_id_str(&id_str2, id);
605 56765ebb 2018-12-23 stsp printf("diff %s %s\n", id_str1 ? id_str1 : "/dev/null", id_str2);
606 54156555 2018-12-24 stsp err = got_diff_tree(tree1, tree2, "", "", diff_context, repo, stdout);
609 79109fed 2018-03-27 stsp got_object_tree_close(tree1);
610 79109fed 2018-03-27 stsp got_object_tree_close(tree2);
611 0f2b3dca 2018-12-22 stsp free(id_str1);
612 0f2b3dca 2018-12-22 stsp free(id_str2);
613 79109fed 2018-03-27 stsp return err;
616 4bb494d5 2018-06-16 stsp static char *
617 6c281f94 2018-06-11 stsp get_datestr(time_t *time, char *datebuf)
619 6c281f94 2018-06-11 stsp char *p, *s = ctime_r(time, datebuf);
620 6c281f94 2018-06-11 stsp p = strchr(s, '\n');
626 79109fed 2018-03-27 stsp static const struct got_error *
627 79109fed 2018-03-27 stsp print_commit(struct got_commit_object *commit, struct got_object_id *id,
628 199a4027 2019-02-02 stsp struct got_repository *repo, int show_patch, int diff_context,
629 199a4027 2019-02-02 stsp struct got_reflist_head *refs)
631 79109fed 2018-03-27 stsp const struct got_error *err = NULL;
632 621015ac 2018-07-23 stsp char *id_str, *datestr, *logmsg0, *logmsg, *line;
633 6c281f94 2018-06-11 stsp char datebuf[26];
634 45d799e2 2018-12-23 stsp time_t committer_time;
635 45d799e2 2018-12-23 stsp const char *author, *committer;
636 199a4027 2019-02-02 stsp char *refs_str = NULL;
637 199a4027 2019-02-02 stsp struct got_reflist_entry *re;
639 199a4027 2019-02-02 stsp SIMPLEQ_FOREACH(re, refs, entry) {
641 199a4027 2019-02-02 stsp const char *name;
642 199a4027 2019-02-02 stsp if (got_object_id_cmp(re->id, id) != 0)
644 199a4027 2019-02-02 stsp name = got_ref_get_name(re->ref);
645 d9498b20 2019-02-05 stsp if (strcmp(name, GOT_REF_HEAD) == 0)
647 199a4027 2019-02-02 stsp if (strncmp(name, "refs/", 5) == 0)
649 7143d404 2019-03-12 stsp if (strncmp(name, "got/", 4) == 0)
651 e34f9ed6 2019-02-02 stsp if (strncmp(name, "heads/", 6) == 0)
653 141c2bff 2019-02-04 stsp if (strncmp(name, "remotes/", 8) == 0)
655 199a4027 2019-02-02 stsp s = refs_str;
656 199a4027 2019-02-02 stsp if (asprintf(&refs_str, "%s%s%s", s ? s : "", s ? ", " : "",
657 199a4027 2019-02-02 stsp name) == -1) {
658 199a4027 2019-02-02 stsp err = got_error_from_errno();
664 832c249c 2018-06-10 stsp err = got_object_id_str(&id_str, id);
666 8bf5b3c9 2018-03-17 stsp return err;
668 33d869be 2018-12-22 stsp printf("-----------------------------------------------\n");
669 199a4027 2019-02-02 stsp printf("commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
670 199a4027 2019-02-02 stsp refs_str ? refs_str : "", refs_str ? ")" : "");
671 832c249c 2018-06-10 stsp free(id_str);
672 d3d493d7 2019-02-21 stsp id_str = NULL;
673 d3d493d7 2019-02-21 stsp free(refs_str);
674 d3d493d7 2019-02-21 stsp refs_str = NULL;
675 45d799e2 2018-12-23 stsp printf("from: %s\n", got_object_commit_get_author(commit));
676 45d799e2 2018-12-23 stsp committer_time = got_object_commit_get_committer_time(commit);
677 45d799e2 2018-12-23 stsp datestr = get_datestr(&committer_time, datebuf);
678 dab5fe87 2018-09-14 stsp printf("date: %s UTC\n", datestr);
679 45d799e2 2018-12-23 stsp author = got_object_commit_get_author(commit);
680 45d799e2 2018-12-23 stsp committer = got_object_commit_get_committer(commit);
681 45d799e2 2018-12-23 stsp if (strcmp(author, committer) != 0)
682 45d799e2 2018-12-23 stsp printf("via: %s\n", committer);
683 45d799e2 2018-12-23 stsp if (got_object_commit_get_nparents(commit) > 1) {
684 45d799e2 2018-12-23 stsp const struct got_object_id_queue *parent_ids;
685 79f35eb3 2018-06-11 stsp struct got_object_qid *qid;
687 45d799e2 2018-12-23 stsp parent_ids = got_object_commit_get_parent_ids(commit);
688 45d799e2 2018-12-23 stsp SIMPLEQ_FOREACH(qid, parent_ids, entry) {
689 79f35eb3 2018-06-11 stsp err = got_object_id_str(&id_str, qid->id);
691 a0603db2 2018-06-10 stsp return err;
692 3fe1abad 2018-06-10 stsp printf("parent %d: %s\n", n++, id_str);
693 a0603db2 2018-06-10 stsp free(id_str);
697 45d799e2 2018-12-23 stsp logmsg0 = strdup(got_object_commit_get_logmsg(commit));
698 621015ac 2018-07-23 stsp if (logmsg0 == NULL)
699 832c249c 2018-06-10 stsp return got_error_from_errno();
701 621015ac 2018-07-23 stsp logmsg = logmsg0;
703 832c249c 2018-06-10 stsp line = strsep(&logmsg, "\n");
705 832c249c 2018-06-10 stsp printf(" %s\n", line);
706 832c249c 2018-06-10 stsp } while (line);
707 621015ac 2018-07-23 stsp free(logmsg0);
709 971751ac 2018-03-27 stsp if (show_patch) {
710 c0cc5c62 2018-10-18 stsp err = print_patch(commit, id, diff_context, repo);
711 971751ac 2018-03-27 stsp if (err == 0)
712 971751ac 2018-03-27 stsp printf("\n");
715 cbe7f848 2019-02-11 stsp if (fflush(stdout) != 0 && err == NULL)
716 cbe7f848 2019-02-11 stsp err = got_error_from_errno();
717 07862c20 2018-09-15 stsp return err;
720 f42b1b34 2018-03-12 stsp static const struct got_error *
721 15a94983 2018-12-23 stsp print_commits(struct got_object_id *root_id, struct got_repository *repo,
722 15a94983 2018-12-23 stsp char *path, int show_patch, int diff_context, int limit,
723 199a4027 2019-02-02 stsp int first_parent_traversal, struct got_reflist_head *refs)
725 f42b1b34 2018-03-12 stsp const struct got_error *err;
726 372ccdbb 2018-06-10 stsp struct got_commit_graph *graph;
728 31cedeaf 2018-09-15 stsp err = got_commit_graph_open(&graph, root_id, path,
729 31cedeaf 2018-09-15 stsp first_parent_traversal, repo);
731 f42b1b34 2018-03-12 stsp return err;
732 31cedeaf 2018-09-15 stsp err = got_commit_graph_iter_start(graph, root_id, repo);
735 31cedeaf 2018-09-15 stsp while (1) {
736 372ccdbb 2018-06-10 stsp struct got_commit_object *commit;
737 372ccdbb 2018-06-10 stsp struct got_object_id *id;
739 84453469 2018-11-11 stsp if (sigint_received || sigpipe_received)
742 b43fbaa0 2018-06-11 stsp err = got_commit_graph_iter_next(&id, graph);
744 9ba79e04 2018-06-11 stsp if (err->code == GOT_ERR_ITER_COMPLETED) {
745 9ba79e04 2018-06-11 stsp err = NULL;
748 372ccdbb 2018-06-10 stsp if (err->code != GOT_ERR_ITER_NEED_MORE)
750 31cedeaf 2018-09-15 stsp err = got_commit_graph_fetch_commits(graph, 1, repo);
756 b43fbaa0 2018-06-11 stsp if (id == NULL)
759 f8e900f3 2018-06-11 stsp err = got_object_open_as_commit(&commit, repo, id);
762 199a4027 2019-02-02 stsp err = print_commit(commit, id, repo, show_patch, diff_context,
764 b43fbaa0 2018-06-11 stsp got_object_commit_close(commit);
765 372ccdbb 2018-06-10 stsp if (err || (limit && --limit == 0))
769 372ccdbb 2018-06-10 stsp got_commit_graph_close(graph);
770 f42b1b34 2018-03-12 stsp return err;
773 4ed7e80c 2018-05-20 stsp __dead static void
774 6f3d1eb0 2018-03-12 stsp usage_log(void)
776 c0cc5c62 2018-10-18 stsp fprintf(stderr, "usage: %s log [-c commit] [-C number] [-f] [ -l N ] [-p] "
777 04ca23f4 2018-07-16 stsp "[-r repository-path] [path]\n", getprogname());
781 4ed7e80c 2018-05-20 stsp static const struct got_error *
782 f42b1b34 2018-03-12 stsp cmd_log(int argc, char *argv[])
784 f42b1b34 2018-03-12 stsp const struct got_error *error;
785 04ca23f4 2018-07-16 stsp struct got_repository *repo = NULL;
786 cffc0aa4 2019-02-05 stsp struct got_worktree *worktree = NULL;
787 15a94983 2018-12-23 stsp struct got_commit_object *commit = NULL;
788 3235492e 2018-04-01 stsp struct got_object_id *id = NULL;
789 04ca23f4 2018-07-16 stsp char *repo_path = NULL, *path = NULL, *cwd = NULL, *in_repo_path = NULL;
790 d142fc45 2018-04-01 stsp char *start_commit = NULL;
791 c0cc5c62 2018-10-18 stsp int diff_context = 3, ch;
792 1fd6d7ea 2018-06-13 stsp int show_patch = 0, limit = 0, first_parent_traversal = 0;
793 64a96a6d 2018-04-01 stsp const char *errstr;
794 199a4027 2019-02-02 stsp struct got_reflist_head refs;
796 1b3893a2 2019-03-18 stsp SIMPLEQ_INIT(&refs);
798 6715a751 2018-03-16 stsp #ifndef PROFILE
799 6098196c 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
802 f42b1b34 2018-03-12 stsp err(1, "pledge");
805 c0cc5c62 2018-10-18 stsp while ((ch = getopt(argc, argv, "pc:C:l:fr:")) != -1) {
806 79109fed 2018-03-27 stsp switch (ch) {
808 79109fed 2018-03-27 stsp show_patch = 1;
811 d142fc45 2018-04-01 stsp start_commit = optarg;
814 4a8520aa 2018-10-18 stsp diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
816 c0cc5c62 2018-10-18 stsp if (errstr != NULL)
817 c0cc5c62 2018-10-18 stsp err(1, "-C option %s", errstr);
820 64a96a6d 2018-04-01 stsp limit = strtonum(optarg, 1, INT_MAX, &errstr);
821 64a96a6d 2018-04-01 stsp if (errstr != NULL)
822 64a96a6d 2018-04-01 stsp err(1, "-l option %s", errstr);
825 0ed6ed4c 2018-06-13 stsp first_parent_traversal = 1;
828 04ca23f4 2018-07-16 stsp repo_path = realpath(optarg, NULL);
829 04ca23f4 2018-07-16 stsp if (repo_path == NULL)
830 04ca23f4 2018-07-16 stsp err(1, "-r option");
833 2deda0b9 2019-03-07 stsp usage_log();
834 79109fed 2018-03-27 stsp /* NOTREACHED */
838 79109fed 2018-03-27 stsp argc -= optind;
839 79109fed 2018-03-27 stsp argv += optind;
841 04ca23f4 2018-07-16 stsp cwd = getcwd(NULL, 0);
842 04ca23f4 2018-07-16 stsp if (cwd == NULL) {
843 04ca23f4 2018-07-16 stsp error = got_error_from_errno();
847 cffc0aa4 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
848 cffc0aa4 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
850 cffc0aa4 2019-02-05 stsp error = NULL;
852 e7301579 2019-03-18 stsp if (argc == 0) {
853 cbd1af7a 2019-03-18 stsp path = strdup("");
854 e7301579 2019-03-18 stsp if (path == NULL) {
855 e7301579 2019-03-18 stsp error = got_error_from_errno();
858 e7301579 2019-03-18 stsp } else if (argc == 1) {
859 e7301579 2019-03-18 stsp if (worktree) {
860 e7301579 2019-03-18 stsp error = got_worktree_resolve_path(&path, worktree,
865 e7301579 2019-03-18 stsp path = strdup(argv[0]);
866 e7301579 2019-03-18 stsp if (path == NULL) {
867 e7301579 2019-03-18 stsp error = got_error_from_errno();
872 cbd1af7a 2019-03-18 stsp usage_log();
874 cffc0aa4 2019-02-05 stsp repo_path = worktree ?
875 cffc0aa4 2019-02-05 stsp strdup(got_worktree_get_repo_path(worktree)) : strdup(cwd);
876 04ca23f4 2018-07-16 stsp if (repo_path == NULL) {
877 cffc0aa4 2019-02-05 stsp error = got_error_from_errno();
881 d0eebce4 2019-03-11 stsp error = apply_unveil(repo_path, 1,
882 efcae1c6 2019-03-07 stsp worktree ? got_worktree_get_root_path(worktree) : NULL);
886 f42b1b34 2018-03-12 stsp error = got_repo_open(&repo, repo_path);
887 d7d4f210 2018-03-12 stsp if (error != NULL)
890 d142fc45 2018-04-01 stsp if (start_commit == NULL) {
891 3235492e 2018-04-01 stsp struct got_reference *head_ref;
892 3235492e 2018-04-01 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
893 3235492e 2018-04-01 stsp if (error != NULL)
894 3235492e 2018-04-01 stsp return error;
895 3235492e 2018-04-01 stsp error = got_ref_resolve(&id, repo, head_ref);
896 3235492e 2018-04-01 stsp got_ref_close(head_ref);
897 3235492e 2018-04-01 stsp if (error != NULL)
898 3235492e 2018-04-01 stsp return error;
899 15a94983 2018-12-23 stsp error = got_object_open_as_commit(&commit, repo, id);
901 d1f2edc9 2018-06-13 stsp struct got_reference *ref;
902 d1f2edc9 2018-06-13 stsp error = got_ref_open(&ref, repo, start_commit);
903 3235492e 2018-04-01 stsp if (error == NULL) {
904 1caad61b 2019-02-01 stsp int obj_type;
905 d1f2edc9 2018-06-13 stsp error = got_ref_resolve(&id, repo, ref);
906 d1f2edc9 2018-06-13 stsp got_ref_close(ref);
907 d1f2edc9 2018-06-13 stsp if (error != NULL)
909 1caad61b 2019-02-01 stsp error = got_object_get_type(&obj_type, repo, id);
910 1caad61b 2019-02-01 stsp if (error != NULL)
912 1caad61b 2019-02-01 stsp if (obj_type == GOT_OBJ_TYPE_TAG) {
913 1caad61b 2019-02-01 stsp struct got_tag_object *tag;
914 1caad61b 2019-02-01 stsp error = got_object_open_as_tag(&tag, repo, id);
915 1caad61b 2019-02-01 stsp if (error != NULL)
917 1caad61b 2019-02-01 stsp if (got_object_tag_get_object_type(tag) !=
918 1caad61b 2019-02-01 stsp GOT_OBJ_TYPE_COMMIT) {
919 1caad61b 2019-02-01 stsp got_object_tag_close(tag);
920 1caad61b 2019-02-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
924 1caad61b 2019-02-01 stsp id = got_object_id_dup(
925 1caad61b 2019-02-01 stsp got_object_tag_get_object_id(tag));
926 1caad61b 2019-02-01 stsp if (id == NULL)
927 1caad61b 2019-02-01 stsp error = got_error_from_errno();
928 1caad61b 2019-02-01 stsp got_object_tag_close(tag);
931 1caad61b 2019-02-01 stsp } else if (obj_type != GOT_OBJ_TYPE_COMMIT) {
932 1caad61b 2019-02-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
935 15a94983 2018-12-23 stsp error = got_object_open_as_commit(&commit, repo, id);
936 d1f2edc9 2018-06-13 stsp if (error != NULL)
939 15a94983 2018-12-23 stsp if (commit == NULL) {
940 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&id, repo,
941 d1f2edc9 2018-06-13 stsp start_commit);
942 d1f2edc9 2018-06-13 stsp if (error != NULL)
943 d1f2edc9 2018-06-13 stsp return error;
946 d7d4f210 2018-03-12 stsp if (error != NULL)
949 23721109 2018-10-22 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
950 04ca23f4 2018-07-16 stsp if (error != NULL)
952 04ca23f4 2018-07-16 stsp if (in_repo_path) {
953 04ca23f4 2018-07-16 stsp free(path);
954 04ca23f4 2018-07-16 stsp path = in_repo_path;
957 199a4027 2019-02-02 stsp error = got_ref_list(&refs, repo);
961 15a94983 2018-12-23 stsp error = print_commits(id, repo, path, show_patch,
962 199a4027 2019-02-02 stsp diff_context, limit, first_parent_traversal, &refs);
964 04ca23f4 2018-07-16 stsp free(path);
965 04ca23f4 2018-07-16 stsp free(repo_path);
968 cffc0aa4 2019-02-05 stsp if (worktree)
969 cffc0aa4 2019-02-05 stsp got_worktree_close(worktree);
970 ad242220 2018-09-08 stsp if (repo) {
971 ad242220 2018-09-08 stsp const struct got_error *repo_error;
972 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
973 ad242220 2018-09-08 stsp if (error == NULL)
974 ad242220 2018-09-08 stsp error = repo_error;
976 e2e879a0 2019-03-11 stsp got_ref_list_free(&refs);
977 8bf5b3c9 2018-03-17 stsp return error;
980 4ed7e80c 2018-05-20 stsp __dead static void
981 3f8b7d6a 2018-04-01 stsp usage_diff(void)
983 b72f483a 2019-02-05 stsp fprintf(stderr, "usage: %s diff [-C number] [-r repository-path] "
984 927df6b7 2019-02-10 stsp "[object1 object2 | path]\n", getprogname());
988 b72f483a 2019-02-05 stsp struct print_diff_arg {
989 b72f483a 2019-02-05 stsp struct got_repository *repo;
990 b72f483a 2019-02-05 stsp struct got_worktree *worktree;
991 b72f483a 2019-02-05 stsp int diff_context;
992 3fc0c068 2019-02-10 stsp const char *id_str;
993 3fc0c068 2019-02-10 stsp int header_shown;
996 4ed7e80c 2018-05-20 stsp static const struct got_error *
997 b72f483a 2019-02-05 stsp print_diff(void *arg, unsigned char status, const char *path,
998 b72f483a 2019-02-05 stsp struct got_object_id *id)
1000 b72f483a 2019-02-05 stsp struct print_diff_arg *a = arg;
1001 b72f483a 2019-02-05 stsp const struct got_error *err = NULL;
1002 b72f483a 2019-02-05 stsp struct got_blob_object *blob1 = NULL;
1003 b72f483a 2019-02-05 stsp FILE *f2 = NULL;
1004 b72f483a 2019-02-05 stsp char *abspath = NULL;
1005 b72f483a 2019-02-05 stsp struct stat sb;
1007 2ec1f75b 2019-03-26 stsp if (status != GOT_STATUS_MODIFY && status != GOT_STATUS_ADD &&
1008 7154f6ce 2019-03-27 stsp status != GOT_STATUS_DELETE && status != GOT_STATUS_CONFLICT)
1009 b72f483a 2019-02-05 stsp return NULL;
1011 3fc0c068 2019-02-10 stsp if (!a->header_shown) {
1012 3fc0c068 2019-02-10 stsp printf("diff %s %s\n", a->id_str,
1013 3fc0c068 2019-02-10 stsp got_worktree_get_root_path(a->worktree));
1014 3fc0c068 2019-02-10 stsp a->header_shown = 1;
1017 7154f6ce 2019-03-27 stsp if (status != GOT_STATUS_ADD) {
1018 d00136be 2019-03-26 stsp err = got_object_open_as_blob(&blob1, a->repo, id, 8192);
1020 d00136be 2019-03-26 stsp goto done;
1024 7154f6ce 2019-03-27 stsp if (status != GOT_STATUS_DELETE) {
1025 2ec1f75b 2019-03-26 stsp if (asprintf(&abspath, "%s/%s",
1026 2ec1f75b 2019-03-26 stsp got_worktree_get_root_path(a->worktree), path) == -1) {
1027 2ec1f75b 2019-03-26 stsp err = got_error_from_errno();
1028 2ec1f75b 2019-03-26 stsp goto done;
1031 2ec1f75b 2019-03-26 stsp f2 = fopen(abspath, "r");
1032 2ec1f75b 2019-03-26 stsp if (f2 == NULL) {
1033 2ec1f75b 2019-03-26 stsp err = got_error_from_errno();
1034 2ec1f75b 2019-03-26 stsp goto done;
1036 2ec1f75b 2019-03-26 stsp if (lstat(abspath, &sb) == -1) {
1037 2ec1f75b 2019-03-26 stsp err = got_error_from_errno();
1038 2ec1f75b 2019-03-26 stsp goto done;
1041 2ec1f75b 2019-03-26 stsp sb.st_size = 0;
1043 b72f483a 2019-02-05 stsp err = got_diff_blob_file(blob1, f2, sb.st_size, path, a->diff_context,
1046 b72f483a 2019-02-05 stsp if (blob1)
1047 b72f483a 2019-02-05 stsp got_object_blob_close(blob1);
1048 fb43ecf1 2019-02-11 stsp if (f2 && fclose(f2) != 0 && err == NULL)
1049 fb43ecf1 2019-02-11 stsp err = got_error_from_errno();
1050 b72f483a 2019-02-05 stsp free(abspath);
1051 927df6b7 2019-02-10 stsp return err;
1054 927df6b7 2019-02-10 stsp static const struct got_error *
1055 b00d56cd 2018-04-01 stsp cmd_diff(int argc, char *argv[])
1057 b00d56cd 2018-04-01 stsp const struct got_error *error;
1058 b00d56cd 2018-04-01 stsp struct got_repository *repo = NULL;
1059 b72f483a 2019-02-05 stsp struct got_worktree *worktree = NULL;
1060 b72f483a 2019-02-05 stsp char *cwd = NULL, *repo_path = NULL;
1061 15a94983 2018-12-23 stsp struct got_object_id *id1 = NULL, *id2 = NULL;
1062 15a94983 2018-12-23 stsp char *id_str1 = NULL, *id_str2 = NULL;
1063 15a94983 2018-12-23 stsp int type1, type2;
1064 c0cc5c62 2018-10-18 stsp int diff_context = 3, ch;
1065 c0cc5c62 2018-10-18 stsp const char *errstr;
1066 927df6b7 2019-02-10 stsp char *path = NULL;
1068 b00d56cd 2018-04-01 stsp #ifndef PROFILE
1069 25eccc22 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1070 25eccc22 2019-01-04 stsp NULL) == -1)
1071 b00d56cd 2018-04-01 stsp err(1, "pledge");
1074 b72f483a 2019-02-05 stsp while ((ch = getopt(argc, argv, "C:r:")) != -1) {
1075 b00d56cd 2018-04-01 stsp switch (ch) {
1077 c0cc5c62 2018-10-18 stsp diff_context = strtonum(optarg, 1, INT_MAX, &errstr);
1078 c0cc5c62 2018-10-18 stsp if (errstr != NULL)
1079 c0cc5c62 2018-10-18 stsp err(1, "-C option %s", errstr);
1082 b72f483a 2019-02-05 stsp repo_path = realpath(optarg, NULL);
1083 b72f483a 2019-02-05 stsp if (repo_path == NULL)
1084 b72f483a 2019-02-05 stsp err(1, "-r option");
1087 2deda0b9 2019-03-07 stsp usage_diff();
1088 b00d56cd 2018-04-01 stsp /* NOTREACHED */
1092 b00d56cd 2018-04-01 stsp argc -= optind;
1093 b00d56cd 2018-04-01 stsp argv += optind;
1095 b72f483a 2019-02-05 stsp cwd = getcwd(NULL, 0);
1096 b72f483a 2019-02-05 stsp if (cwd == NULL) {
1097 b72f483a 2019-02-05 stsp error = got_error_from_errno();
1098 b72f483a 2019-02-05 stsp goto done;
1100 927df6b7 2019-02-10 stsp error = got_worktree_open(&worktree, cwd);
1101 927df6b7 2019-02-10 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1102 927df6b7 2019-02-10 stsp goto done;
1103 927df6b7 2019-02-10 stsp if (argc <= 1) {
1104 927df6b7 2019-02-10 stsp if (worktree == NULL) {
1105 927df6b7 2019-02-10 stsp error = got_error(GOT_ERR_NOT_WORKTREE);
1106 927df6b7 2019-02-10 stsp goto done;
1108 b72f483a 2019-02-05 stsp if (repo_path)
1110 b72f483a 2019-02-05 stsp "-r option can't be used when diffing a work tree");
1111 b72f483a 2019-02-05 stsp repo_path = strdup(got_worktree_get_repo_path(worktree));
1112 927df6b7 2019-02-10 stsp if (repo_path == NULL) {
1113 927df6b7 2019-02-10 stsp error = got_error_from_errno();
1114 927df6b7 2019-02-10 stsp goto done;
1116 927df6b7 2019-02-10 stsp if (argc == 1) {
1117 6c7ab921 2019-03-18 stsp error = got_worktree_resolve_path(&path, worktree,
1119 927df6b7 2019-02-10 stsp if (error)
1120 927df6b7 2019-02-10 stsp goto done;
1122 927df6b7 2019-02-10 stsp path = strdup("");
1123 927df6b7 2019-02-10 stsp if (path == NULL) {
1124 927df6b7 2019-02-10 stsp error = got_error_from_errno();
1125 927df6b7 2019-02-10 stsp goto done;
1128 b72f483a 2019-02-05 stsp } else if (argc == 2) {
1129 15a94983 2018-12-23 stsp id_str1 = argv[0];
1130 15a94983 2018-12-23 stsp id_str2 = argv[1];
1132 b00d56cd 2018-04-01 stsp usage_diff();
1134 b72f483a 2019-02-05 stsp if (repo_path == NULL) {
1135 b72f483a 2019-02-05 stsp repo_path = getcwd(NULL, 0);
1136 b72f483a 2019-02-05 stsp if (repo_path == NULL)
1137 b72f483a 2019-02-05 stsp return got_error_from_errno();
1140 d0eebce4 2019-03-11 stsp error = apply_unveil(repo_path, 1,
1141 b72f483a 2019-02-05 stsp worktree ? got_worktree_get_root_path(worktree) : NULL);
1142 25eccc22 2019-01-04 stsp if (error)
1143 25eccc22 2019-01-04 stsp goto done;
1145 b00d56cd 2018-04-01 stsp error = got_repo_open(&repo, repo_path);
1146 76089277 2018-04-01 stsp free(repo_path);
1147 b00d56cd 2018-04-01 stsp if (error != NULL)
1148 b00d56cd 2018-04-01 stsp goto done;
1150 b72f483a 2019-02-05 stsp if (worktree) {
1151 b72f483a 2019-02-05 stsp struct print_diff_arg arg;
1152 b72f483a 2019-02-05 stsp char *id_str;
1153 b72f483a 2019-02-05 stsp error = got_object_id_str(&id_str,
1154 b72f483a 2019-02-05 stsp got_worktree_get_base_commit_id(worktree));
1155 b72f483a 2019-02-05 stsp if (error)
1156 b72f483a 2019-02-05 stsp goto done;
1157 b72f483a 2019-02-05 stsp arg.repo = repo;
1158 b72f483a 2019-02-05 stsp arg.worktree = worktree;
1159 b72f483a 2019-02-05 stsp arg.diff_context = diff_context;
1160 3fc0c068 2019-02-10 stsp arg.id_str = id_str;
1161 3fc0c068 2019-02-10 stsp arg.header_shown = 0;
1163 927df6b7 2019-02-10 stsp error = got_worktree_status(worktree, path, repo, print_diff,
1164 b72f483a 2019-02-05 stsp &arg, check_cancelled, NULL);
1165 b72f483a 2019-02-05 stsp free(id_str);
1166 b72f483a 2019-02-05 stsp goto done;
1169 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&id1, repo, id_str1);
1170 6402fb3c 2018-09-15 stsp if (error)
1171 b00d56cd 2018-04-01 stsp goto done;
1173 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&id2, repo, id_str2);
1174 6402fb3c 2018-09-15 stsp if (error)
1175 b00d56cd 2018-04-01 stsp goto done;
1177 15a94983 2018-12-23 stsp error = got_object_get_type(&type1, repo, id1);
1178 15a94983 2018-12-23 stsp if (error)
1179 15a94983 2018-12-23 stsp goto done;
1181 15a94983 2018-12-23 stsp error = got_object_get_type(&type2, repo, id2);
1182 15a94983 2018-12-23 stsp if (error)
1183 15a94983 2018-12-23 stsp goto done;
1185 15a94983 2018-12-23 stsp if (type1 != type2) {
1186 b00d56cd 2018-04-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1187 b00d56cd 2018-04-01 stsp goto done;
1190 15a94983 2018-12-23 stsp switch (type1) {
1191 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_BLOB:
1192 15a94983 2018-12-23 stsp error = got_diff_objects_as_blobs(id1, id2, NULL, NULL,
1193 54156555 2018-12-24 stsp diff_context, repo, stdout);
1195 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_TREE:
1196 15a94983 2018-12-23 stsp error = got_diff_objects_as_trees(id1, id2, "", "",
1197 54156555 2018-12-24 stsp diff_context, repo, stdout);
1199 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_COMMIT:
1200 15a94983 2018-12-23 stsp printf("diff %s %s\n", id_str1 ? id_str1 : "/dev/null",
1202 15a94983 2018-12-23 stsp error = got_diff_objects_as_commits(id1, id2, diff_context,
1203 c0cc5c62 2018-10-18 stsp repo, stdout);
1206 b00d56cd 2018-04-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1210 15a94983 2018-12-23 stsp free(id1);
1211 15a94983 2018-12-23 stsp free(id2);
1212 927df6b7 2019-02-10 stsp free(path);
1213 b72f483a 2019-02-05 stsp if (worktree)
1214 b72f483a 2019-02-05 stsp got_worktree_close(worktree);
1215 ad242220 2018-09-08 stsp if (repo) {
1216 ad242220 2018-09-08 stsp const struct got_error *repo_error;
1217 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
1218 ad242220 2018-09-08 stsp if (error == NULL)
1219 ad242220 2018-09-08 stsp error = repo_error;
1221 b00d56cd 2018-04-01 stsp return error;
1224 404c43c4 2018-06-21 stsp __dead static void
1225 404c43c4 2018-06-21 stsp usage_blame(void)
1227 9270e621 2019-02-05 stsp fprintf(stderr,
1228 9270e621 2019-02-05 stsp "usage: %s blame [-c commit] [-r repository-path] path\n",
1229 404c43c4 2018-06-21 stsp getprogname());
1233 404c43c4 2018-06-21 stsp static const struct got_error *
1234 404c43c4 2018-06-21 stsp cmd_blame(int argc, char *argv[])
1236 404c43c4 2018-06-21 stsp const struct got_error *error;
1237 404c43c4 2018-06-21 stsp struct got_repository *repo = NULL;
1238 0c06baac 2019-02-05 stsp struct got_worktree *worktree = NULL;
1239 66bea077 2018-08-02 stsp char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
1240 404c43c4 2018-06-21 stsp struct got_object_id *commit_id = NULL;
1241 404c43c4 2018-06-21 stsp char *commit_id_str = NULL;
1244 404c43c4 2018-06-21 stsp #ifndef PROFILE
1245 36e2fb66 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1246 36e2fb66 2019-01-04 stsp NULL) == -1)
1247 404c43c4 2018-06-21 stsp err(1, "pledge");
1250 66bea077 2018-08-02 stsp while ((ch = getopt(argc, argv, "c:r:")) != -1) {
1251 404c43c4 2018-06-21 stsp switch (ch) {
1253 404c43c4 2018-06-21 stsp commit_id_str = optarg;
1256 66bea077 2018-08-02 stsp repo_path = realpath(optarg, NULL);
1257 66bea077 2018-08-02 stsp if (repo_path == NULL)
1258 66bea077 2018-08-02 stsp err(1, "-r option");
1261 2deda0b9 2019-03-07 stsp usage_blame();
1262 404c43c4 2018-06-21 stsp /* NOTREACHED */
1266 404c43c4 2018-06-21 stsp argc -= optind;
1267 404c43c4 2018-06-21 stsp argv += optind;
1269 a39318fd 2018-08-02 stsp if (argc == 1)
1270 404c43c4 2018-06-21 stsp path = argv[0];
1272 404c43c4 2018-06-21 stsp usage_blame();
1274 66bea077 2018-08-02 stsp cwd = getcwd(NULL, 0);
1275 66bea077 2018-08-02 stsp if (cwd == NULL) {
1276 66bea077 2018-08-02 stsp error = got_error_from_errno();
1277 66bea077 2018-08-02 stsp goto done;
1279 66bea077 2018-08-02 stsp if (repo_path == NULL) {
1280 0c06baac 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
1281 0c06baac 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1282 66bea077 2018-08-02 stsp goto done;
1284 0c06baac 2019-02-05 stsp error = NULL;
1285 0c06baac 2019-02-05 stsp if (worktree) {
1286 0c06baac 2019-02-05 stsp repo_path =
1287 0c06baac 2019-02-05 stsp strdup(got_worktree_get_repo_path(worktree));
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 if (error)
1291 0c06baac 2019-02-05 stsp goto done;
1293 0c06baac 2019-02-05 stsp repo_path = strdup(cwd);
1294 0c06baac 2019-02-05 stsp if (repo_path == NULL) {
1295 0c06baac 2019-02-05 stsp error = got_error_from_errno();
1296 0c06baac 2019-02-05 stsp goto done;
1301 d0eebce4 2019-03-11 stsp error = apply_unveil(repo_path, 1, NULL);
1302 36e2fb66 2019-01-04 stsp if (error)
1303 36e2fb66 2019-01-04 stsp goto done;
1305 404c43c4 2018-06-21 stsp error = got_repo_open(&repo, repo_path);
1306 404c43c4 2018-06-21 stsp if (error != NULL)
1307 404c43c4 2018-06-21 stsp goto done;
1309 0c06baac 2019-02-05 stsp if (worktree) {
1310 6efaaa2d 2019-02-05 stsp const char *prefix = got_worktree_get_path_prefix(worktree);
1311 0c06baac 2019-02-05 stsp char *p, *worktree_subdir = cwd +
1312 0c06baac 2019-02-05 stsp strlen(got_worktree_get_root_path(worktree));
1313 6efaaa2d 2019-02-05 stsp if (asprintf(&p, "%s%s%s%s%s",
1314 6efaaa2d 2019-02-05 stsp prefix, (strcmp(prefix, "/") != 0) ? "/" : "",
1315 6efaaa2d 2019-02-05 stsp worktree_subdir, worktree_subdir[0] ? "/" : "",
1316 6efaaa2d 2019-02-05 stsp path) == -1) {
1317 0c06baac 2019-02-05 stsp error = got_error_from_errno();
1318 0c06baac 2019-02-05 stsp goto done;
1320 6efaaa2d 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, p, 0);
1323 0c06baac 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
1325 0c06baac 2019-02-05 stsp if (error)
1326 66bea077 2018-08-02 stsp goto done;
1328 404c43c4 2018-06-21 stsp if (commit_id_str == NULL) {
1329 404c43c4 2018-06-21 stsp struct got_reference *head_ref;
1330 404c43c4 2018-06-21 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1331 404c43c4 2018-06-21 stsp if (error != NULL)
1332 66bea077 2018-08-02 stsp goto done;
1333 404c43c4 2018-06-21 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
1334 404c43c4 2018-06-21 stsp got_ref_close(head_ref);
1335 404c43c4 2018-06-21 stsp if (error != NULL)
1336 66bea077 2018-08-02 stsp goto done;
1338 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&commit_id, repo,
1339 15a94983 2018-12-23 stsp commit_id_str);
1340 404c43c4 2018-06-21 stsp if (error != NULL)
1341 66bea077 2018-08-02 stsp goto done;
1344 66bea077 2018-08-02 stsp error = got_blame(in_repo_path, commit_id, repo, stdout);
1346 66bea077 2018-08-02 stsp free(in_repo_path);
1347 66bea077 2018-08-02 stsp free(repo_path);
1348 66bea077 2018-08-02 stsp free(cwd);
1349 404c43c4 2018-06-21 stsp free(commit_id);
1350 0c06baac 2019-02-05 stsp if (worktree)
1351 0c06baac 2019-02-05 stsp got_worktree_close(worktree);
1352 ad242220 2018-09-08 stsp if (repo) {
1353 ad242220 2018-09-08 stsp const struct got_error *repo_error;
1354 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
1355 ad242220 2018-09-08 stsp if (error == NULL)
1356 ad242220 2018-09-08 stsp error = repo_error;
1358 404c43c4 2018-06-21 stsp return error;
1361 5de5890b 2018-10-18 stsp __dead static void
1362 5de5890b 2018-10-18 stsp usage_tree(void)
1364 c1669e2e 2019-01-09 stsp fprintf(stderr,
1365 c1669e2e 2019-01-09 stsp "usage: %s tree [-c commit] [-r repository-path] [-iR] path\n",
1366 5de5890b 2018-10-18 stsp getprogname());
1370 c1669e2e 2019-01-09 stsp static void
1371 c1669e2e 2019-01-09 stsp print_entry(struct got_tree_entry *te, const char *id, const char *path,
1372 c1669e2e 2019-01-09 stsp const char *root_path)
1374 c1669e2e 2019-01-09 stsp int is_root_path = (strcmp(path, root_path) == 0);
1376 c1669e2e 2019-01-09 stsp path += strlen(root_path);
1377 c1669e2e 2019-01-09 stsp while (path[0] == '/')
1380 c1669e2e 2019-01-09 stsp printf("%s%s%s%s%s\n", id ? id : "", path,
1381 d6e648b4 2019-02-10 stsp is_root_path ? "" : "/", te->name,
1382 d6e648b4 2019-02-10 stsp S_ISDIR(te->mode) ? "/" : ((te->mode & S_IXUSR) ? "*" : ""));
1385 5de5890b 2018-10-18 stsp static const struct got_error *
1386 5de5890b 2018-10-18 stsp print_tree(const char *path, struct got_object_id *commit_id,
1387 c1669e2e 2019-01-09 stsp int show_ids, int recurse, const char *root_path,
1388 c1669e2e 2019-01-09 stsp struct got_repository *repo)
1390 5de5890b 2018-10-18 stsp const struct got_error *err = NULL;
1391 5de5890b 2018-10-18 stsp struct got_object_id *tree_id = NULL;
1392 5de5890b 2018-10-18 stsp struct got_tree_object *tree = NULL;
1393 5de5890b 2018-10-18 stsp const struct got_tree_entries *entries;
1394 5de5890b 2018-10-18 stsp struct got_tree_entry *te;
1396 5de5890b 2018-10-18 stsp err = got_object_id_by_path(&tree_id, repo, commit_id, path);
1398 5de5890b 2018-10-18 stsp goto done;
1400 5de5890b 2018-10-18 stsp err = got_object_open_as_tree(&tree, repo, tree_id);
1402 5de5890b 2018-10-18 stsp goto done;
1403 5de5890b 2018-10-18 stsp entries = got_object_tree_get_entries(tree);
1404 5de5890b 2018-10-18 stsp te = SIMPLEQ_FIRST(&entries->head);
1405 5de5890b 2018-10-18 stsp while (te) {
1406 5de5890b 2018-10-18 stsp char *id = NULL;
1408 84453469 2018-11-11 stsp if (sigint_received || sigpipe_received)
1411 5de5890b 2018-10-18 stsp if (show_ids) {
1412 5de5890b 2018-10-18 stsp char *id_str;
1413 5de5890b 2018-10-18 stsp err = got_object_id_str(&id_str, te->id);
1415 5de5890b 2018-10-18 stsp goto done;
1416 5de5890b 2018-10-18 stsp if (asprintf(&id, "%s ", id_str) == -1) {
1417 5de5890b 2018-10-18 stsp err = got_error_from_errno();
1418 5de5890b 2018-10-18 stsp free(id_str);
1419 5de5890b 2018-10-18 stsp goto done;
1421 5de5890b 2018-10-18 stsp free(id_str);
1423 c1669e2e 2019-01-09 stsp print_entry(te, id, path, root_path);
1426 c1669e2e 2019-01-09 stsp if (recurse && S_ISDIR(te->mode)) {
1427 c1669e2e 2019-01-09 stsp char *child_path;
1428 c1669e2e 2019-01-09 stsp if (asprintf(&child_path, "%s%s%s", path,
1429 c1669e2e 2019-01-09 stsp path[0] == '/' && path[1] == '\0' ? "" : "/",
1430 c1669e2e 2019-01-09 stsp te->name) == -1) {
1431 c1669e2e 2019-01-09 stsp err = got_error_from_errno();
1432 c1669e2e 2019-01-09 stsp goto done;
1434 c1669e2e 2019-01-09 stsp err = print_tree(child_path, commit_id, show_ids, 1,
1435 c1669e2e 2019-01-09 stsp root_path, repo);
1436 c1669e2e 2019-01-09 stsp free(child_path);
1438 c1669e2e 2019-01-09 stsp goto done;
1441 c1669e2e 2019-01-09 stsp te = SIMPLEQ_NEXT(te, entry);
1445 5de5890b 2018-10-18 stsp got_object_tree_close(tree);
1446 5de5890b 2018-10-18 stsp free(tree_id);
1447 5de5890b 2018-10-18 stsp return err;
1450 5de5890b 2018-10-18 stsp static const struct got_error *
1451 5de5890b 2018-10-18 stsp cmd_tree(int argc, char *argv[])
1453 5de5890b 2018-10-18 stsp const struct got_error *error;
1454 5de5890b 2018-10-18 stsp struct got_repository *repo = NULL;
1455 7a2c19d6 2019-02-05 stsp struct got_worktree *worktree = NULL;
1456 7a2c19d6 2019-02-05 stsp const char *path;
1457 7a2c19d6 2019-02-05 stsp char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
1458 5de5890b 2018-10-18 stsp struct got_object_id *commit_id = NULL;
1459 5de5890b 2018-10-18 stsp char *commit_id_str = NULL;
1460 c1669e2e 2019-01-09 stsp int show_ids = 0, recurse = 0;
1463 5de5890b 2018-10-18 stsp #ifndef PROFILE
1464 0f8d269b 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1465 0f8d269b 2019-01-04 stsp NULL) == -1)
1466 5de5890b 2018-10-18 stsp err(1, "pledge");
1469 c1669e2e 2019-01-09 stsp while ((ch = getopt(argc, argv, "c:r:iR")) != -1) {
1470 5de5890b 2018-10-18 stsp switch (ch) {
1472 5de5890b 2018-10-18 stsp commit_id_str = optarg;
1475 5de5890b 2018-10-18 stsp repo_path = realpath(optarg, NULL);
1476 5de5890b 2018-10-18 stsp if (repo_path == NULL)
1477 5de5890b 2018-10-18 stsp err(1, "-r option");
1480 5de5890b 2018-10-18 stsp show_ids = 1;
1483 c1669e2e 2019-01-09 stsp recurse = 1;
1486 2deda0b9 2019-03-07 stsp usage_tree();
1487 5de5890b 2018-10-18 stsp /* NOTREACHED */
1491 5de5890b 2018-10-18 stsp argc -= optind;
1492 5de5890b 2018-10-18 stsp argv += optind;
1494 5de5890b 2018-10-18 stsp if (argc == 1)
1495 5de5890b 2018-10-18 stsp path = argv[0];
1496 5de5890b 2018-10-18 stsp else if (argc > 1)
1497 5de5890b 2018-10-18 stsp usage_tree();
1499 7a2c19d6 2019-02-05 stsp path = NULL;
1501 9bf7a39b 2019-02-05 stsp cwd = getcwd(NULL, 0);
1502 9bf7a39b 2019-02-05 stsp if (cwd == NULL) {
1503 9bf7a39b 2019-02-05 stsp error = got_error_from_errno();
1504 9bf7a39b 2019-02-05 stsp goto done;
1506 5de5890b 2018-10-18 stsp if (repo_path == NULL) {
1507 7a2c19d6 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
1508 8994de28 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1509 7a2c19d6 2019-02-05 stsp goto done;
1511 7a2c19d6 2019-02-05 stsp error = NULL;
1512 7a2c19d6 2019-02-05 stsp if (worktree) {
1513 7a2c19d6 2019-02-05 stsp repo_path =
1514 7a2c19d6 2019-02-05 stsp strdup(got_worktree_get_repo_path(worktree));
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 if (error)
1518 7a2c19d6 2019-02-05 stsp goto done;
1520 7a2c19d6 2019-02-05 stsp repo_path = strdup(cwd);
1521 7a2c19d6 2019-02-05 stsp if (repo_path == NULL) {
1522 7a2c19d6 2019-02-05 stsp error = got_error_from_errno();
1523 7a2c19d6 2019-02-05 stsp goto done;
1528 d0eebce4 2019-03-11 stsp error = apply_unveil(repo_path, 1, NULL);
1529 0f8d269b 2019-01-04 stsp if (error)
1530 0f8d269b 2019-01-04 stsp goto done;
1532 5de5890b 2018-10-18 stsp error = got_repo_open(&repo, repo_path);
1533 5de5890b 2018-10-18 stsp if (error != NULL)
1534 5de5890b 2018-10-18 stsp goto done;
1536 9bf7a39b 2019-02-05 stsp if (path == NULL) {
1537 9bf7a39b 2019-02-05 stsp if (worktree) {
1538 9bf7a39b 2019-02-05 stsp char *p, *worktree_subdir = cwd +
1539 9bf7a39b 2019-02-05 stsp strlen(got_worktree_get_root_path(worktree));
1540 9bf7a39b 2019-02-05 stsp if (asprintf(&p, "%s/%s",
1541 9bf7a39b 2019-02-05 stsp got_worktree_get_path_prefix(worktree),
1542 9bf7a39b 2019-02-05 stsp worktree_subdir) == -1) {
1543 9bf7a39b 2019-02-05 stsp error = got_error_from_errno();
1544 9bf7a39b 2019-02-05 stsp goto done;
1546 9bf7a39b 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, p, 1);
1548 9bf7a39b 2019-02-05 stsp if (error)
1549 9bf7a39b 2019-02-05 stsp goto done;
1551 9bf7a39b 2019-02-05 stsp path = "/";
1553 9bf7a39b 2019-02-05 stsp if (in_repo_path == NULL) {
1554 9bf7a39b 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
1555 9bf7a39b 2019-02-05 stsp if (error != NULL)
1556 9bf7a39b 2019-02-05 stsp goto done;
1559 5de5890b 2018-10-18 stsp if (commit_id_str == NULL) {
1560 5de5890b 2018-10-18 stsp struct got_reference *head_ref;
1561 5de5890b 2018-10-18 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD);
1562 5de5890b 2018-10-18 stsp if (error != NULL)
1563 5de5890b 2018-10-18 stsp goto done;
1564 5de5890b 2018-10-18 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
1565 5de5890b 2018-10-18 stsp got_ref_close(head_ref);
1566 5de5890b 2018-10-18 stsp if (error != NULL)
1567 5de5890b 2018-10-18 stsp goto done;
1569 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&commit_id, repo,
1570 15a94983 2018-12-23 stsp commit_id_str);
1571 5de5890b 2018-10-18 stsp if (error != NULL)
1572 5de5890b 2018-10-18 stsp goto done;
1575 c1669e2e 2019-01-09 stsp error = print_tree(in_repo_path, commit_id, show_ids, recurse,
1576 c1669e2e 2019-01-09 stsp in_repo_path, repo);
1578 5de5890b 2018-10-18 stsp free(in_repo_path);
1579 5de5890b 2018-10-18 stsp free(repo_path);
1580 5de5890b 2018-10-18 stsp free(cwd);
1581 5de5890b 2018-10-18 stsp free(commit_id);
1582 7a2c19d6 2019-02-05 stsp if (worktree)
1583 7a2c19d6 2019-02-05 stsp got_worktree_close(worktree);
1584 5de5890b 2018-10-18 stsp if (repo) {
1585 5de5890b 2018-10-18 stsp const struct got_error *repo_error;
1586 5de5890b 2018-10-18 stsp repo_error = got_repo_close(repo);
1587 5de5890b 2018-10-18 stsp if (error == NULL)
1588 5de5890b 2018-10-18 stsp error = repo_error;
1590 5de5890b 2018-10-18 stsp return error;
1593 6bad629b 2019-02-04 stsp __dead static void
1594 6bad629b 2019-02-04 stsp usage_status(void)
1596 927df6b7 2019-02-10 stsp fprintf(stderr, "usage: %s status [path]\n", getprogname());
1600 b72f483a 2019-02-05 stsp static const struct got_error *
1601 b72f483a 2019-02-05 stsp print_status(void *arg, unsigned char status, const char *path,
1602 b72f483a 2019-02-05 stsp struct got_object_id *id)
1604 6bad629b 2019-02-04 stsp printf("%c %s\n", status, path);
1605 b72f483a 2019-02-05 stsp return NULL;
1608 6bad629b 2019-02-04 stsp static const struct got_error *
1609 6bad629b 2019-02-04 stsp cmd_status(int argc, char *argv[])
1611 6bad629b 2019-02-04 stsp const struct got_error *error = NULL;
1612 6bad629b 2019-02-04 stsp struct got_repository *repo = NULL;
1613 6bad629b 2019-02-04 stsp struct got_worktree *worktree = NULL;
1614 927df6b7 2019-02-10 stsp char *cwd = NULL, *path = NULL;
1617 6bad629b 2019-02-04 stsp while ((ch = getopt(argc, argv, "")) != -1) {
1618 6bad629b 2019-02-04 stsp switch (ch) {
1620 2deda0b9 2019-03-07 stsp usage_status();
1621 6bad629b 2019-02-04 stsp /* NOTREACHED */
1625 6bad629b 2019-02-04 stsp argc -= optind;
1626 6bad629b 2019-02-04 stsp argv += optind;
1628 6bad629b 2019-02-04 stsp #ifndef PROFILE
1629 6bad629b 2019-02-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1630 6bad629b 2019-02-04 stsp NULL) == -1)
1631 6bad629b 2019-02-04 stsp err(1, "pledge");
1633 927df6b7 2019-02-10 stsp cwd = getcwd(NULL, 0);
1634 927df6b7 2019-02-10 stsp if (cwd == NULL) {
1635 927df6b7 2019-02-10 stsp error = got_error_from_errno();
1636 927df6b7 2019-02-10 stsp goto done;
1639 927df6b7 2019-02-10 stsp error = got_worktree_open(&worktree, cwd);
1640 927df6b7 2019-02-10 stsp if (error != NULL)
1641 927df6b7 2019-02-10 stsp goto done;
1643 6bad629b 2019-02-04 stsp if (argc == 0) {
1644 927df6b7 2019-02-10 stsp path = strdup("");
1645 927df6b7 2019-02-10 stsp if (path == NULL) {
1646 6bad629b 2019-02-04 stsp error = got_error_from_errno();
1647 6bad629b 2019-02-04 stsp goto done;
1649 6bad629b 2019-02-04 stsp } else if (argc == 1) {
1650 6c7ab921 2019-03-18 stsp error = got_worktree_resolve_path(&path, worktree, argv[0]);
1651 927df6b7 2019-02-10 stsp if (error)
1652 6bad629b 2019-02-04 stsp goto done;
1654 6bad629b 2019-02-04 stsp usage_status();
1656 6bad629b 2019-02-04 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
1657 6bad629b 2019-02-04 stsp if (error != NULL)
1658 6bad629b 2019-02-04 stsp goto done;
1660 d0eebce4 2019-03-11 stsp error = apply_unveil(got_repo_get_path(repo), 1,
1661 c7f4312f 2019-02-05 stsp got_worktree_get_root_path(worktree));
1662 6bad629b 2019-02-04 stsp if (error)
1663 6bad629b 2019-02-04 stsp goto done;
1665 927df6b7 2019-02-10 stsp error = got_worktree_status(worktree, path, repo, print_status, NULL,
1666 6bad629b 2019-02-04 stsp check_cancelled, NULL);
1668 927df6b7 2019-02-10 stsp free(cwd);
1669 927df6b7 2019-02-10 stsp free(path);
1670 d0eebce4 2019-03-11 stsp return error;
1673 d0eebce4 2019-03-11 stsp __dead static void
1674 d0eebce4 2019-03-11 stsp usage_ref(void)
1676 d0eebce4 2019-03-11 stsp fprintf(stderr,
1677 d0eebce4 2019-03-11 stsp "usage: %s ref [-r repository] -l | -d name | name object\n",
1678 d0eebce4 2019-03-11 stsp getprogname());
1682 d0eebce4 2019-03-11 stsp static const struct got_error *
1683 d0eebce4 2019-03-11 stsp list_refs(struct got_repository *repo)
1685 d0eebce4 2019-03-11 stsp static const struct got_error *err = NULL;
1686 d0eebce4 2019-03-11 stsp struct got_reflist_head refs;
1687 d0eebce4 2019-03-11 stsp struct got_reflist_entry *re;
1689 d0eebce4 2019-03-11 stsp SIMPLEQ_INIT(&refs);
1690 d0eebce4 2019-03-11 stsp err = got_ref_list(&refs, repo);
1692 d0eebce4 2019-03-11 stsp return err;
1694 d0eebce4 2019-03-11 stsp SIMPLEQ_FOREACH(re, &refs, entry) {
1695 d0eebce4 2019-03-11 stsp char *refstr;
1696 d0eebce4 2019-03-11 stsp refstr = got_ref_to_str(re->ref);
1697 d0eebce4 2019-03-11 stsp if (refstr == NULL)
1698 d0eebce4 2019-03-11 stsp return got_error_from_errno();
1699 d0eebce4 2019-03-11 stsp printf("%s: %s\n", got_ref_get_name(re->ref), refstr);
1700 d0eebce4 2019-03-11 stsp free(refstr);
1703 e2e879a0 2019-03-11 stsp got_ref_list_free(&refs);
1704 d0eebce4 2019-03-11 stsp return NULL;
1707 d0eebce4 2019-03-11 stsp static const struct got_error *
1708 d0eebce4 2019-03-11 stsp delete_ref(struct got_repository *repo, const char *refname)
1710 d0eebce4 2019-03-11 stsp const struct got_error *err = NULL;
1711 d0eebce4 2019-03-11 stsp struct got_reference *ref;
1713 d0eebce4 2019-03-11 stsp err = got_ref_open(&ref, repo, refname);
1715 d0eebce4 2019-03-11 stsp return err;
1717 d0eebce4 2019-03-11 stsp err = got_ref_delete(ref, repo);
1718 d0eebce4 2019-03-11 stsp got_ref_close(ref);
1719 d0eebce4 2019-03-11 stsp return err;
1722 d0eebce4 2019-03-11 stsp static const struct got_error *
1723 d0eebce4 2019-03-11 stsp add_ref(struct got_repository *repo, const char *refname, const char *id_str)
1725 d0eebce4 2019-03-11 stsp const struct got_error *err = NULL;
1726 d0eebce4 2019-03-11 stsp struct got_object_id *id;
1727 d0eebce4 2019-03-11 stsp struct got_reference *ref = NULL;
1729 d0eebce4 2019-03-11 stsp err = got_object_resolve_id_str(&id, repo, id_str);
1731 d0eebce4 2019-03-11 stsp return err;
1733 d0eebce4 2019-03-11 stsp err = got_ref_alloc(&ref, refname, id);
1735 d0eebce4 2019-03-11 stsp goto done;
1737 d0eebce4 2019-03-11 stsp err = got_ref_write(ref, repo);
1740 d0eebce4 2019-03-11 stsp got_ref_close(ref);
1742 d0eebce4 2019-03-11 stsp return err;
1745 d0eebce4 2019-03-11 stsp static const struct got_error *
1746 d0eebce4 2019-03-11 stsp cmd_ref(int argc, char *argv[])
1748 d0eebce4 2019-03-11 stsp const struct got_error *error = NULL;
1749 d0eebce4 2019-03-11 stsp struct got_repository *repo = NULL;
1750 d0eebce4 2019-03-11 stsp struct got_worktree *worktree = NULL;
1751 d0eebce4 2019-03-11 stsp char *cwd = NULL, *repo_path = NULL;
1752 d0eebce4 2019-03-11 stsp int ch, do_list = 0;
1753 d0eebce4 2019-03-11 stsp const char *delref = NULL;
1755 d0eebce4 2019-03-11 stsp /* TODO: Add -s option for adding symbolic references. */
1756 d0eebce4 2019-03-11 stsp while ((ch = getopt(argc, argv, "d:r:l")) != -1) {
1757 d0eebce4 2019-03-11 stsp switch (ch) {
1759 d0eebce4 2019-03-11 stsp delref = optarg;
1762 d0eebce4 2019-03-11 stsp repo_path = realpath(optarg, NULL);
1763 d0eebce4 2019-03-11 stsp if (repo_path == NULL)
1764 d0eebce4 2019-03-11 stsp err(1, "-r option");
1767 d0eebce4 2019-03-11 stsp do_list = 1;
1770 d0eebce4 2019-03-11 stsp usage_ref();
1771 d0eebce4 2019-03-11 stsp /* NOTREACHED */
1775 d0eebce4 2019-03-11 stsp if (do_list && delref)
1776 d0eebce4 2019-03-11 stsp errx(1, "-l and -d options are mutually exclusive\n");
1778 d0eebce4 2019-03-11 stsp argc -= optind;
1779 d0eebce4 2019-03-11 stsp argv += optind;
1781 d0eebce4 2019-03-11 stsp if (do_list || delref) {
1782 d0eebce4 2019-03-11 stsp if (argc > 0)
1783 d0eebce4 2019-03-11 stsp usage_ref();
1784 d0eebce4 2019-03-11 stsp } else if (argc != 2)
1785 d0eebce4 2019-03-11 stsp usage_ref();
1787 d0eebce4 2019-03-11 stsp #ifndef PROFILE
1788 e0b57350 2019-03-12 stsp if (do_list) {
1789 e0b57350 2019-03-12 stsp if (pledge("stdio rpath wpath flock proc exec sendfd unveil",
1790 e0b57350 2019-03-12 stsp NULL) == -1)
1791 e0b57350 2019-03-12 stsp err(1, "pledge");
1793 e0b57350 2019-03-12 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec "
1794 e0b57350 2019-03-12 stsp "sendfd unveil", NULL) == -1)
1795 e0b57350 2019-03-12 stsp err(1, "pledge");
1798 d0eebce4 2019-03-11 stsp cwd = getcwd(NULL, 0);
1799 d0eebce4 2019-03-11 stsp if (cwd == NULL) {
1800 d0eebce4 2019-03-11 stsp error = got_error_from_errno();
1801 d0eebce4 2019-03-11 stsp goto done;
1804 d0eebce4 2019-03-11 stsp if (repo_path == NULL) {
1805 d0eebce4 2019-03-11 stsp error = got_worktree_open(&worktree, cwd);
1806 d0eebce4 2019-03-11 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1807 d0eebce4 2019-03-11 stsp goto done;
1809 d0eebce4 2019-03-11 stsp error = NULL;
1810 d0eebce4 2019-03-11 stsp if (worktree) {
1811 d0eebce4 2019-03-11 stsp repo_path =
1812 d0eebce4 2019-03-11 stsp strdup(got_worktree_get_repo_path(worktree));
1813 d0eebce4 2019-03-11 stsp if (repo_path == NULL)
1814 d0eebce4 2019-03-11 stsp error = got_error_from_errno();
1815 d0eebce4 2019-03-11 stsp if (error)
1816 d0eebce4 2019-03-11 stsp goto done;
1818 d0eebce4 2019-03-11 stsp repo_path = strdup(cwd);
1819 d0eebce4 2019-03-11 stsp if (repo_path == NULL) {
1820 d0eebce4 2019-03-11 stsp error = got_error_from_errno();
1821 d0eebce4 2019-03-11 stsp goto done;
1826 d0eebce4 2019-03-11 stsp error = apply_unveil(repo_path, do_list,
1827 d0eebce4 2019-03-11 stsp worktree ? got_worktree_get_root_path(worktree) : NULL);
1828 d0eebce4 2019-03-11 stsp if (error)
1829 d0eebce4 2019-03-11 stsp goto done;
1831 d0eebce4 2019-03-11 stsp error = got_repo_open(&repo, repo_path);
1832 d0eebce4 2019-03-11 stsp if (error != NULL)
1833 d0eebce4 2019-03-11 stsp goto done;
1835 d0eebce4 2019-03-11 stsp if (do_list)
1836 d0eebce4 2019-03-11 stsp error = list_refs(repo);
1837 d0eebce4 2019-03-11 stsp else if (delref)
1838 d0eebce4 2019-03-11 stsp error = delete_ref(repo, delref);
1840 d0eebce4 2019-03-11 stsp error = add_ref(repo, argv[0], argv[1]);
1843 d0eebce4 2019-03-11 stsp got_repo_close(repo);
1844 d0eebce4 2019-03-11 stsp if (worktree)
1845 d0eebce4 2019-03-11 stsp got_worktree_close(worktree);
1846 d0eebce4 2019-03-11 stsp free(cwd);
1847 d0eebce4 2019-03-11 stsp free(repo_path);
1848 d00136be 2019-03-26 stsp return error;
1851 d00136be 2019-03-26 stsp __dead static void
1852 d00136be 2019-03-26 stsp usage_add(void)
1854 d00136be 2019-03-26 stsp fprintf(stderr, "usage: %s add file-path\n", getprogname());
1858 d00136be 2019-03-26 stsp static const struct got_error *
1859 d00136be 2019-03-26 stsp cmd_add(int argc, char *argv[])
1861 d00136be 2019-03-26 stsp const struct got_error *error = NULL;
1862 031a5338 2019-03-26 stsp struct got_repository *repo = NULL;
1863 d00136be 2019-03-26 stsp struct got_worktree *worktree = NULL;
1864 d00136be 2019-03-26 stsp char *cwd = NULL, *path = NULL, *relpath = NULL;
1867 d00136be 2019-03-26 stsp while ((ch = getopt(argc, argv, "")) != -1) {
1868 d00136be 2019-03-26 stsp switch (ch) {
1870 d00136be 2019-03-26 stsp usage_add();
1871 d00136be 2019-03-26 stsp /* NOTREACHED */
1875 d00136be 2019-03-26 stsp argc -= optind;
1876 d00136be 2019-03-26 stsp argv += optind;
1878 d00136be 2019-03-26 stsp if (argc != 1)
1879 d00136be 2019-03-26 stsp usage_add();
1881 d00136be 2019-03-26 stsp path = realpath(argv[0], NULL);
1882 d00136be 2019-03-26 stsp if (path == NULL) {
1883 d00136be 2019-03-26 stsp error = got_error_from_errno();
1884 d00136be 2019-03-26 stsp goto done;
1887 d00136be 2019-03-26 stsp cwd = getcwd(NULL, 0);
1888 d00136be 2019-03-26 stsp if (cwd == NULL) {
1889 d00136be 2019-03-26 stsp error = got_error_from_errno();
1890 d00136be 2019-03-26 stsp goto done;
1892 d00136be 2019-03-26 stsp error = got_worktree_open(&worktree, cwd);
1893 d00136be 2019-03-26 stsp if (error)
1894 d00136be 2019-03-26 stsp goto done;
1896 031a5338 2019-03-26 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
1897 031a5338 2019-03-26 stsp if (error != NULL)
1898 031a5338 2019-03-26 stsp goto done;
1900 031a5338 2019-03-26 stsp error = apply_unveil(got_repo_get_path(repo), 1,
1901 031a5338 2019-03-26 stsp got_worktree_get_root_path(worktree));
1902 d00136be 2019-03-26 stsp if (error)
1903 d00136be 2019-03-26 stsp goto done;
1905 031a5338 2019-03-26 stsp error = got_worktree_schedule_add(worktree, path, print_status, NULL,
1907 d00136be 2019-03-26 stsp if (error)
1908 d00136be 2019-03-26 stsp goto done;
1911 031a5338 2019-03-26 stsp got_repo_close(repo);
1912 d00136be 2019-03-26 stsp if (worktree)
1913 d00136be 2019-03-26 stsp got_worktree_close(worktree);
1914 d00136be 2019-03-26 stsp free(path);
1915 d00136be 2019-03-26 stsp free(relpath);
1916 2ec1f75b 2019-03-26 stsp free(cwd);
1917 2ec1f75b 2019-03-26 stsp return error;
1920 2ec1f75b 2019-03-26 stsp __dead static void
1921 2ec1f75b 2019-03-26 stsp usage_rm(void)
1923 2ec1f75b 2019-03-26 stsp fprintf(stderr, "usage: %s rm [-f] file-path\n", getprogname());
1927 2ec1f75b 2019-03-26 stsp static const struct got_error *
1928 2ec1f75b 2019-03-26 stsp cmd_rm(int argc, char *argv[])
1930 2ec1f75b 2019-03-26 stsp const struct got_error *error = NULL;
1931 2ec1f75b 2019-03-26 stsp struct got_worktree *worktree = NULL;
1932 2ec1f75b 2019-03-26 stsp struct got_repository *repo = NULL;
1933 2ec1f75b 2019-03-26 stsp char *cwd = NULL, *path = NULL;
1934 2ec1f75b 2019-03-26 stsp int ch, delete_local_mods = 0;
1936 2ec1f75b 2019-03-26 stsp while ((ch = getopt(argc, argv, "f")) != -1) {
1937 2ec1f75b 2019-03-26 stsp switch (ch) {
1939 2ec1f75b 2019-03-26 stsp delete_local_mods = 1;
1942 2ec1f75b 2019-03-26 stsp usage_add();
1943 2ec1f75b 2019-03-26 stsp /* NOTREACHED */
1947 2ec1f75b 2019-03-26 stsp argc -= optind;
1948 2ec1f75b 2019-03-26 stsp argv += optind;
1950 2ec1f75b 2019-03-26 stsp if (argc != 1)
1951 2ec1f75b 2019-03-26 stsp usage_rm();
1953 2ec1f75b 2019-03-26 stsp path = realpath(argv[0], NULL);
1954 2ec1f75b 2019-03-26 stsp if (path == NULL) {
1955 2ec1f75b 2019-03-26 stsp error = got_error_from_errno();
1956 2ec1f75b 2019-03-26 stsp goto done;
1959 2ec1f75b 2019-03-26 stsp cwd = getcwd(NULL, 0);
1960 2ec1f75b 2019-03-26 stsp if (cwd == NULL) {
1961 2ec1f75b 2019-03-26 stsp error = got_error_from_errno();
1962 2ec1f75b 2019-03-26 stsp goto done;
1964 2ec1f75b 2019-03-26 stsp error = got_worktree_open(&worktree, cwd);
1965 2ec1f75b 2019-03-26 stsp if (error)
1966 2ec1f75b 2019-03-26 stsp goto done;
1968 2ec1f75b 2019-03-26 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
1969 2ec1f75b 2019-03-26 stsp if (error != NULL)
1970 2ec1f75b 2019-03-26 stsp goto done;
1972 c2253644 2019-03-26 stsp error = apply_unveil(got_repo_get_path(repo), 1,
1973 c2253644 2019-03-26 stsp got_worktree_get_root_path(worktree));
1974 2ec1f75b 2019-03-26 stsp if (error)
1975 2ec1f75b 2019-03-26 stsp goto done;
1977 2ec1f75b 2019-03-26 stsp error = got_worktree_schedule_delete(worktree, path, delete_local_mods,
1978 2ec1f75b 2019-03-26 stsp print_status, NULL, repo);
1979 a129376b 2019-03-28 stsp if (error)
1980 a129376b 2019-03-28 stsp goto done;
1983 a129376b 2019-03-28 stsp got_repo_close(repo);
1984 a129376b 2019-03-28 stsp if (worktree)
1985 a129376b 2019-03-28 stsp got_worktree_close(worktree);
1986 a129376b 2019-03-28 stsp free(path);
1987 a129376b 2019-03-28 stsp free(cwd);
1988 a129376b 2019-03-28 stsp return error;
1991 a129376b 2019-03-28 stsp __dead static void
1992 a129376b 2019-03-28 stsp usage_revert(void)
1994 a129376b 2019-03-28 stsp fprintf(stderr, "usage: %s revert file-path\n", getprogname());
1998 a129376b 2019-03-28 stsp static void
1999 a129376b 2019-03-28 stsp revert_progress(void *arg, unsigned char status, const char *path)
2001 a129376b 2019-03-28 stsp while (path[0] == '/')
2003 a129376b 2019-03-28 stsp printf("%c %s\n", status, path);
2006 a129376b 2019-03-28 stsp static const struct got_error *
2007 a129376b 2019-03-28 stsp cmd_revert(int argc, char *argv[])
2009 a129376b 2019-03-28 stsp const struct got_error *error = NULL;
2010 a129376b 2019-03-28 stsp struct got_worktree *worktree = NULL;
2011 a129376b 2019-03-28 stsp struct got_repository *repo = NULL;
2012 a129376b 2019-03-28 stsp char *cwd = NULL, *path = NULL;
2015 a129376b 2019-03-28 stsp while ((ch = getopt(argc, argv, "")) != -1) {
2016 a129376b 2019-03-28 stsp switch (ch) {
2018 a129376b 2019-03-28 stsp usage_revert();
2019 a129376b 2019-03-28 stsp /* NOTREACHED */
2023 a129376b 2019-03-28 stsp argc -= optind;
2024 a129376b 2019-03-28 stsp argv += optind;
2026 a129376b 2019-03-28 stsp if (argc != 1)
2027 a129376b 2019-03-28 stsp usage_revert();
2029 a129376b 2019-03-28 stsp path = realpath(argv[0], NULL);
2030 a129376b 2019-03-28 stsp if (path == NULL) {
2031 a129376b 2019-03-28 stsp error = got_error_from_errno();
2032 a129376b 2019-03-28 stsp goto done;
2035 a129376b 2019-03-28 stsp cwd = getcwd(NULL, 0);
2036 a129376b 2019-03-28 stsp if (cwd == NULL) {
2037 a129376b 2019-03-28 stsp error = got_error_from_errno();
2038 a129376b 2019-03-28 stsp goto done;
2040 a129376b 2019-03-28 stsp error = got_worktree_open(&worktree, cwd);
2041 2ec1f75b 2019-03-26 stsp if (error)
2042 2ec1f75b 2019-03-26 stsp goto done;
2044 a129376b 2019-03-28 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
2045 a129376b 2019-03-28 stsp if (error != NULL)
2046 a129376b 2019-03-28 stsp goto done;
2048 a129376b 2019-03-28 stsp error = apply_unveil(got_repo_get_path(repo), 1,
2049 a129376b 2019-03-28 stsp got_worktree_get_root_path(worktree));
2050 a129376b 2019-03-28 stsp if (error)
2051 a129376b 2019-03-28 stsp goto done;
2053 a129376b 2019-03-28 stsp error = got_worktree_revert(worktree, path,
2054 a129376b 2019-03-28 stsp revert_progress, NULL, repo);
2055 a129376b 2019-03-28 stsp if (error)
2056 a129376b 2019-03-28 stsp goto done;
2059 2ec1f75b 2019-03-26 stsp got_repo_close(repo);
2060 2ec1f75b 2019-03-26 stsp if (worktree)
2061 2ec1f75b 2019-03-26 stsp got_worktree_close(worktree);
2062 2ec1f75b 2019-03-26 stsp free(path);
2063 d00136be 2019-03-26 stsp free(cwd);
2064 6bad629b 2019-02-04 stsp return error;
