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 3c45a30a 2019-05-12 jcs #include <sys/param.h>
23 33ad4cbe 2019-05-12 jcs #include <sys/wait.h>
24 f42b1b34 2018-03-12 stsp
25 5c860e29 2018-03-12 stsp #include <err.h>
26 5c860e29 2018-03-12 stsp #include <errno.h>
27 5c860e29 2018-03-12 stsp #include <locale.h>
28 99437157 2018-11-11 stsp #include <signal.h>
29 5c860e29 2018-03-12 stsp #include <stdio.h>
30 5c860e29 2018-03-12 stsp #include <stdlib.h>
31 5c860e29 2018-03-12 stsp #include <string.h>
32 5c860e29 2018-03-12 stsp #include <unistd.h>
33 c09a553d 2018-03-12 stsp #include <libgen.h>
34 c0768b0f 2018-06-10 stsp #include <time.h>
35 33ad4cbe 2019-05-12 jcs #include <paths.h>
36 5c860e29 2018-03-12 stsp
37 f42b1b34 2018-03-12 stsp #include "got_error.h"
38 f42b1b34 2018-03-12 stsp #include "got_object.h"
39 5261c201 2018-04-01 stsp #include "got_reference.h"
40 f42b1b34 2018-03-12 stsp #include "got_repository.h"
41 1dd54920 2019-05-11 stsp #include "got_path.h"
42 c09a553d 2018-03-12 stsp #include "got_worktree.h"
43 79109fed 2018-03-27 stsp #include "got_diff.h"
44 372ccdbb 2018-06-10 stsp #include "got_commit_graph.h"
45 404c43c4 2018-06-21 stsp #include "got_blame.h"
46 63219cd2 2019-01-04 stsp #include "got_privsep.h"
47 793c30b5 2019-05-13 stsp #include "got_opentemp.h"
48 5c860e29 2018-03-12 stsp
49 5c860e29 2018-03-12 stsp #ifndef nitems
50 5c860e29 2018-03-12 stsp #define nitems(_a) (sizeof((_a)) / sizeof((_a)[0]))
51 5c860e29 2018-03-12 stsp #endif
52 99437157 2018-11-11 stsp
53 99437157 2018-11-11 stsp static volatile sig_atomic_t sigint_received;
54 99437157 2018-11-11 stsp static volatile sig_atomic_t sigpipe_received;
55 99437157 2018-11-11 stsp
56 99437157 2018-11-11 stsp static void
57 99437157 2018-11-11 stsp catch_sigint(int signo)
58 99437157 2018-11-11 stsp {
59 99437157 2018-11-11 stsp sigint_received = 1;
60 99437157 2018-11-11 stsp }
61 99437157 2018-11-11 stsp
62 99437157 2018-11-11 stsp static void
63 99437157 2018-11-11 stsp catch_sigpipe(int signo)
64 99437157 2018-11-11 stsp {
65 99437157 2018-11-11 stsp sigpipe_received = 1;
66 99437157 2018-11-11 stsp }
67 5c860e29 2018-03-12 stsp
68 99437157 2018-11-11 stsp
69 5c860e29 2018-03-12 stsp struct cmd {
70 5c860e29 2018-03-12 stsp const char *cmd_name;
71 d7d4f210 2018-03-12 stsp const struct got_error *(*cmd_main)(int, char *[]);
72 1b6b95a8 2018-03-12 stsp void (*cmd_usage)(void);
73 46a0db7d 2018-03-12 stsp const char *cmd_descr;
74 5c860e29 2018-03-12 stsp };
75 5c860e29 2018-03-12 stsp
76 4ed7e80c 2018-05-20 stsp __dead static void usage(void);
77 2c7829a4 2019-06-17 stsp __dead static void usage_init(void);
78 4ed7e80c 2018-05-20 stsp __dead static void usage_checkout(void);
79 507dc3bb 2018-12-29 stsp __dead static void usage_update(void);
80 4ed7e80c 2018-05-20 stsp __dead static void usage_log(void);
81 4ed7e80c 2018-05-20 stsp __dead static void usage_diff(void);
82 404c43c4 2018-06-21 stsp __dead static void usage_blame(void);
83 5de5890b 2018-10-18 stsp __dead static void usage_tree(void);
84 6bad629b 2019-02-04 stsp __dead static void usage_status(void);
85 d0eebce4 2019-03-11 stsp __dead static void usage_ref(void);
86 d00136be 2019-03-26 stsp __dead static void usage_add(void);
87 2ec1f75b 2019-03-26 stsp __dead static void usage_rm(void);
88 a129376b 2019-03-28 stsp __dead static void usage_revert(void);
89 c4296144 2019-05-09 stsp __dead static void usage_commit(void);
90 234035bc 2019-06-01 stsp __dead static void usage_cherrypick(void);
91 5ef14e63 2019-06-02 stsp __dead static void usage_backout(void);
92 5c860e29 2018-03-12 stsp
93 2c7829a4 2019-06-17 stsp static const struct got_error* cmd_init(int, char *[]);
94 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_checkout(int, char *[]);
95 507dc3bb 2018-12-29 stsp static const struct got_error* cmd_update(int, char *[]);
96 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_log(int, char *[]);
97 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_diff(int, char *[]);
98 404c43c4 2018-06-21 stsp static const struct got_error* cmd_blame(int, char *[]);
99 5de5890b 2018-10-18 stsp static const struct got_error* cmd_tree(int, char *[]);
100 4ed7e80c 2018-05-20 stsp static const struct got_error* cmd_status(int, char *[]);
101 d0eebce4 2019-03-11 stsp static const struct got_error* cmd_ref(int, char *[]);
102 d00136be 2019-03-26 stsp static const struct got_error* cmd_add(int, char *[]);
103 2ec1f75b 2019-03-26 stsp static const struct got_error* cmd_rm(int, char *[]);
104 a129376b 2019-03-28 stsp static const struct got_error* cmd_revert(int, char *[]);
105 c4296144 2019-05-09 stsp static const struct got_error* cmd_commit(int, char *[]);
106 234035bc 2019-06-01 stsp static const struct got_error* cmd_cherrypick(int, char *[]);
107 5ef14e63 2019-06-02 stsp static const struct got_error* cmd_backout(int, char *[]);
108 5c860e29 2018-03-12 stsp
109 4ed7e80c 2018-05-20 stsp static struct cmd got_commands[] = {
110 2c7829a4 2019-06-17 stsp { "init", cmd_init, usage_init,
111 2c7829a4 2019-06-17 stsp "create a new empty repository" },
112 c09a553d 2018-03-12 stsp { "checkout", cmd_checkout, usage_checkout,
113 0bb8a95e 2018-03-12 stsp "check out a new work tree from a repository" },
114 507dc3bb 2018-12-29 stsp { "update", cmd_update, usage_update,
115 507dc3bb 2018-12-29 stsp "update a work tree to a different commit" },
116 1b6b95a8 2018-03-12 stsp { "log", cmd_log, usage_log,
117 1b6b95a8 2018-03-12 stsp "show repository history" },
118 b00d56cd 2018-04-01 stsp { "diff", cmd_diff, usage_diff,
119 b00d56cd 2018-04-01 stsp "compare files and directories" },
120 404c43c4 2018-06-21 stsp { "blame", cmd_blame, usage_blame,
121 a67e2392 2019-03-26 stsp "show when lines in a file were changed" },
122 5de5890b 2018-10-18 stsp { "tree", cmd_tree, usage_tree,
123 a67e2392 2019-03-26 stsp "list files and directories in repository" },
124 1b6b95a8 2018-03-12 stsp { "status", cmd_status, usage_status,
125 1b6b95a8 2018-03-12 stsp "show modification status of files" },
126 d0eebce4 2019-03-11 stsp { "ref", cmd_ref, usage_ref,
127 d0eebce4 2019-03-11 stsp "manage references in repository" },
128 d00136be 2019-03-26 stsp { "add", cmd_add, usage_add,
129 bd14628f 2019-05-12 stsp "add new files to version control" },
130 2ec1f75b 2019-03-26 stsp { "rm", cmd_rm, usage_rm,
131 2ec1f75b 2019-03-26 stsp "remove a versioned file" },
132 a129376b 2019-03-28 stsp { "revert", cmd_revert, usage_revert,
133 a129376b 2019-03-28 stsp "revert uncommitted changes" },
134 c4296144 2019-05-09 stsp { "commit", cmd_commit, usage_commit,
135 5501382f 2019-05-10 stsp "write changes from work tree to repository" },
136 234035bc 2019-06-01 stsp { "cherrypick", cmd_cherrypick, usage_cherrypick,
137 234035bc 2019-06-01 stsp "merge a single commit from another branch into a work tree" },
138 5ef14e63 2019-06-02 stsp { "backout", cmd_backout, usage_backout,
139 5ef14e63 2019-06-02 stsp "reverse-merge changes from a commit into a work tree" },
140 5c860e29 2018-03-12 stsp };
141 5c860e29 2018-03-12 stsp
142 5c860e29 2018-03-12 stsp int
143 5c860e29 2018-03-12 stsp main(int argc, char *argv[])
144 5c860e29 2018-03-12 stsp {
145 5c860e29 2018-03-12 stsp struct cmd *cmd;
146 5c860e29 2018-03-12 stsp unsigned int i;
147 5c860e29 2018-03-12 stsp int ch;
148 1b6b95a8 2018-03-12 stsp int hflag = 0;
149 5c860e29 2018-03-12 stsp
150 289e3cbf 2019-02-04 stsp setlocale(LC_CTYPE, "");
151 5c860e29 2018-03-12 stsp
152 1b6b95a8 2018-03-12 stsp while ((ch = getopt(argc, argv, "h")) != -1) {
153 5c860e29 2018-03-12 stsp switch (ch) {
154 1b6b95a8 2018-03-12 stsp case 'h':
155 1b6b95a8 2018-03-12 stsp hflag = 1;
156 1b6b95a8 2018-03-12 stsp break;
157 5c860e29 2018-03-12 stsp default:
158 5c860e29 2018-03-12 stsp usage();
159 5c860e29 2018-03-12 stsp /* NOTREACHED */
160 5c860e29 2018-03-12 stsp }
161 5c860e29 2018-03-12 stsp }
162 5c860e29 2018-03-12 stsp
163 5c860e29 2018-03-12 stsp argc -= optind;
164 5c860e29 2018-03-12 stsp argv += optind;
165 1e70621d 2018-03-27 stsp optind = 0;
166 5c860e29 2018-03-12 stsp
167 5c860e29 2018-03-12 stsp if (argc <= 0)
168 5c860e29 2018-03-12 stsp usage();
169 5c860e29 2018-03-12 stsp
170 99437157 2018-11-11 stsp signal(SIGINT, catch_sigint);
171 99437157 2018-11-11 stsp signal(SIGPIPE, catch_sigpipe);
172 99437157 2018-11-11 stsp
173 5c860e29 2018-03-12 stsp for (i = 0; i < nitems(got_commands); i++) {
174 d7d4f210 2018-03-12 stsp const struct got_error *error;
175 d7d4f210 2018-03-12 stsp
176 5c860e29 2018-03-12 stsp cmd = &got_commands[i];
177 5c860e29 2018-03-12 stsp
178 5c860e29 2018-03-12 stsp if (strncmp(cmd->cmd_name, argv[0], strlen(argv[0])))
179 5c860e29 2018-03-12 stsp continue;
180 5c860e29 2018-03-12 stsp
181 1b6b95a8 2018-03-12 stsp if (hflag)
182 1b6b95a8 2018-03-12 stsp got_commands[i].cmd_usage();
183 1b6b95a8 2018-03-12 stsp
184 d7d4f210 2018-03-12 stsp error = got_commands[i].cmd_main(argc, argv);
185 80d5f134 2018-11-11 stsp if (error && !(sigint_received || sigpipe_received)) {
186 d7d4f210 2018-03-12 stsp fprintf(stderr, "%s: %s\n", getprogname(), error->msg);
187 d7d4f210 2018-03-12 stsp return 1;
188 d7d4f210 2018-03-12 stsp }
189 d7d4f210 2018-03-12 stsp
190 d7d4f210 2018-03-12 stsp return 0;
191 5c860e29 2018-03-12 stsp }
192 5c860e29 2018-03-12 stsp
193 20ecf764 2018-03-12 stsp fprintf(stderr, "%s: unknown command '%s'\n", getprogname(), argv[0]);
194 5c860e29 2018-03-12 stsp return 1;
195 5c860e29 2018-03-12 stsp }
196 5c860e29 2018-03-12 stsp
197 4ed7e80c 2018-05-20 stsp __dead static void
198 5c860e29 2018-03-12 stsp usage(void)
199 5c860e29 2018-03-12 stsp {
200 46a0db7d 2018-03-12 stsp int i;
201 46a0db7d 2018-03-12 stsp
202 987e94ba 2018-03-12 stsp fprintf(stderr, "usage: %s [-h] command [arg ...]\n\n"
203 987e94ba 2018-03-12 stsp "Available commands:\n", getprogname());
204 46a0db7d 2018-03-12 stsp for (i = 0; i < nitems(got_commands); i++) {
205 46a0db7d 2018-03-12 stsp struct cmd *cmd = &got_commands[i];
206 46a0db7d 2018-03-12 stsp fprintf(stderr, " %s: %s\n", cmd->cmd_name, cmd->cmd_descr);
207 46a0db7d 2018-03-12 stsp }
208 5c860e29 2018-03-12 stsp exit(1);
209 5c860e29 2018-03-12 stsp }
210 5c860e29 2018-03-12 stsp
211 0266afb7 2019-01-04 stsp static const struct got_error *
212 0ee7065d 2019-05-13 stsp get_editor(char **abspath)
213 e2ba3d07 2019-05-13 stsp {
214 0ee7065d 2019-05-13 stsp const struct got_error *err = NULL;
215 e2ba3d07 2019-05-13 stsp const char *editor;
216 e2ba3d07 2019-05-13 stsp
217 e2ba3d07 2019-05-13 stsp editor = getenv("VISUAL");
218 e2ba3d07 2019-05-13 stsp if (editor == NULL)
219 e2ba3d07 2019-05-13 stsp editor = getenv("EDITOR");
220 e2ba3d07 2019-05-13 stsp
221 0ee7065d 2019-05-13 stsp if (editor) {
222 0ee7065d 2019-05-13 stsp err = got_path_find_prog(abspath, editor);
223 0ee7065d 2019-05-13 stsp if (err)
224 0ee7065d 2019-05-13 stsp return err;
225 0ee7065d 2019-05-13 stsp }
226 e2ba3d07 2019-05-13 stsp
227 0ee7065d 2019-05-13 stsp if (*abspath == NULL) {
228 0ee7065d 2019-05-13 stsp *abspath = strdup("/bin/ed");
229 0ee7065d 2019-05-13 stsp if (*abspath == NULL)
230 0ee7065d 2019-05-13 stsp return got_error_from_errno("strdup");
231 0ee7065d 2019-05-13 stsp }
232 0ee7065d 2019-05-13 stsp
233 e2ba3d07 2019-05-13 stsp return NULL;
234 e2ba3d07 2019-05-13 stsp }
235 e2ba3d07 2019-05-13 stsp
236 e2ba3d07 2019-05-13 stsp static const struct got_error *
237 d0eebce4 2019-03-11 stsp apply_unveil(const char *repo_path, int repo_read_only,
238 314a6357 2019-05-13 stsp const char *worktree_path, int create_worktree)
239 0266afb7 2019-01-04 stsp {
240 163ce85a 2019-05-13 stsp const struct got_error *err;
241 0266afb7 2019-01-04 stsp
242 a0937847 2019-05-11 stsp if (create_worktree) {
243 a0937847 2019-05-11 stsp /* Pre-create work tree path to avoid unveiling its parents. */
244 163ce85a 2019-05-13 stsp err = got_path_mkdir(worktree_path);
245 3c45a30a 2019-05-12 jcs
246 3c45a30a 2019-05-12 jcs if (errno == EEXIST) {
247 280f921b 2019-05-12 stsp if (got_path_dir_is_empty(worktree_path)) {
248 3c45a30a 2019-05-12 jcs errno = 0;
249 163ce85a 2019-05-13 stsp err = NULL;
250 3c45a30a 2019-05-12 jcs } else {
251 df056ada 2019-05-15 stsp err = got_error_path(worktree_path,
252 df056ada 2019-05-15 stsp GOT_ERR_DIR_NOT_EMPTY);
253 3c45a30a 2019-05-12 jcs }
254 3c45a30a 2019-05-12 jcs }
255 3c45a30a 2019-05-12 jcs
256 163ce85a 2019-05-13 stsp if (err && (err->code != GOT_ERR_ERRNO || errno != EISDIR))
257 163ce85a 2019-05-13 stsp return err;
258 a0937847 2019-05-11 stsp }
259 a0937847 2019-05-11 stsp
260 d0eebce4 2019-03-11 stsp if (repo_path && unveil(repo_path, repo_read_only ? "r" : "rwc") != 0)
261 638f9024 2019-05-13 stsp return got_error_from_errno2("unveil", repo_path);
262 0266afb7 2019-01-04 stsp
263 0266afb7 2019-01-04 stsp if (worktree_path && unveil(worktree_path, "rwc") != 0)
264 638f9024 2019-05-13 stsp return got_error_from_errno2("unveil", worktree_path);
265 0266afb7 2019-01-04 stsp
266 f12d0dbe 2019-01-04 stsp if (unveil("/tmp", "rwc") != 0)
267 638f9024 2019-05-13 stsp return got_error_from_errno2("unveil", "/tmp");
268 0266afb7 2019-01-04 stsp
269 163ce85a 2019-05-13 stsp err = got_privsep_unveil_exec_helpers();
270 163ce85a 2019-05-13 stsp if (err != NULL)
271 163ce85a 2019-05-13 stsp return err;
272 0266afb7 2019-01-04 stsp
273 0266afb7 2019-01-04 stsp if (unveil(NULL, NULL) != 0)
274 638f9024 2019-05-13 stsp return got_error_from_errno("unveil");
275 0266afb7 2019-01-04 stsp
276 0266afb7 2019-01-04 stsp return NULL;
277 2c7829a4 2019-06-17 stsp }
278 2c7829a4 2019-06-17 stsp
279 2c7829a4 2019-06-17 stsp __dead static void
280 2c7829a4 2019-06-17 stsp usage_init(void)
281 2c7829a4 2019-06-17 stsp {
282 2c7829a4 2019-06-17 stsp fprintf(stderr, "usage: %s init path\n", getprogname());
283 2c7829a4 2019-06-17 stsp exit(1);
284 2c7829a4 2019-06-17 stsp }
285 2c7829a4 2019-06-17 stsp
286 2c7829a4 2019-06-17 stsp static const struct got_error *
287 2c7829a4 2019-06-17 stsp cmd_init(int argc, char *argv[])
288 2c7829a4 2019-06-17 stsp {
289 2c7829a4 2019-06-17 stsp const struct got_error *error = NULL;
290 2c7829a4 2019-06-17 stsp char *repo_path = NULL;
291 2c7829a4 2019-06-17 stsp int ch;
292 2c7829a4 2019-06-17 stsp
293 2c7829a4 2019-06-17 stsp while ((ch = getopt(argc, argv, "")) != -1) {
294 2c7829a4 2019-06-17 stsp switch (ch) {
295 2c7829a4 2019-06-17 stsp default:
296 2c7829a4 2019-06-17 stsp usage_init();
297 2c7829a4 2019-06-17 stsp /* NOTREACHED */
298 2c7829a4 2019-06-17 stsp }
299 2c7829a4 2019-06-17 stsp }
300 2c7829a4 2019-06-17 stsp
301 2c7829a4 2019-06-17 stsp argc -= optind;
302 2c7829a4 2019-06-17 stsp argv += optind;
303 2c7829a4 2019-06-17 stsp
304 2c7829a4 2019-06-17 stsp #ifndef PROFILE
305 2c7829a4 2019-06-17 stsp if (pledge("stdio rpath wpath cpath unveil", NULL) == -1)
306 2c7829a4 2019-06-17 stsp err(1, "pledge");
307 2c7829a4 2019-06-17 stsp #endif
308 2c7829a4 2019-06-17 stsp if (argc != 1)
309 bc20e173 2019-06-17 stsp usage_init();
310 2c7829a4 2019-06-17 stsp
311 2c7829a4 2019-06-17 stsp repo_path = strdup(argv[0]);
312 2c7829a4 2019-06-17 stsp if (repo_path == NULL)
313 2c7829a4 2019-06-17 stsp return got_error_from_errno("strdup");
314 2c7829a4 2019-06-17 stsp
315 2c7829a4 2019-06-17 stsp got_path_strip_trailing_slashes(repo_path);
316 2c7829a4 2019-06-17 stsp
317 2c7829a4 2019-06-17 stsp error = got_path_mkdir(repo_path);
318 2c7829a4 2019-06-17 stsp if (error &&
319 2c7829a4 2019-06-17 stsp !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
320 2c7829a4 2019-06-17 stsp goto done;
321 2c7829a4 2019-06-17 stsp
322 2c7829a4 2019-06-17 stsp error = apply_unveil(repo_path, 0, NULL, 0);
323 2c7829a4 2019-06-17 stsp if (error)
324 2c7829a4 2019-06-17 stsp goto done;
325 2c7829a4 2019-06-17 stsp
326 2c7829a4 2019-06-17 stsp error = got_repo_init(repo_path);
327 2c7829a4 2019-06-17 stsp if (error != NULL)
328 2c7829a4 2019-06-17 stsp goto done;
329 2c7829a4 2019-06-17 stsp
330 2c7829a4 2019-06-17 stsp done:
331 2c7829a4 2019-06-17 stsp free(repo_path);
332 2c7829a4 2019-06-17 stsp return error;
333 0266afb7 2019-01-04 stsp }
334 0266afb7 2019-01-04 stsp
335 4ed7e80c 2018-05-20 stsp __dead static void
336 c09a553d 2018-03-12 stsp usage_checkout(void)
337 c09a553d 2018-03-12 stsp {
338 08573d5b 2019-05-14 stsp fprintf(stderr, "usage: %s checkout [-b branch] [-c commit] "
339 08573d5b 2019-05-14 stsp "[-p prefix] repository-path [worktree-path]\n", getprogname());
340 c09a553d 2018-03-12 stsp exit(1);
341 92a684f4 2018-03-12 stsp }
342 92a684f4 2018-03-12 stsp
343 92a684f4 2018-03-12 stsp static void
344 a0eb853d 2018-12-29 stsp checkout_progress(void *arg, unsigned char status, const char *path)
345 92a684f4 2018-03-12 stsp {
346 92a684f4 2018-03-12 stsp char *worktree_path = arg;
347 a484d721 2019-06-10 stsp
348 a484d721 2019-06-10 stsp /* Base commit bump happens silently. */
349 a484d721 2019-06-10 stsp if (status == GOT_STATUS_BUMP_BASE)
350 a484d721 2019-06-10 stsp return;
351 92a684f4 2018-03-12 stsp
352 92a684f4 2018-03-12 stsp while (path[0] == '/')
353 92a684f4 2018-03-12 stsp path++;
354 92a684f4 2018-03-12 stsp
355 d7b62c98 2018-12-27 stsp printf("%c %s/%s\n", status, worktree_path, path);
356 99437157 2018-11-11 stsp }
357 99437157 2018-11-11 stsp
358 99437157 2018-11-11 stsp static const struct got_error *
359 6bad629b 2019-02-04 stsp check_cancelled(void *arg)
360 99437157 2018-11-11 stsp {
361 99437157 2018-11-11 stsp if (sigint_received || sigpipe_received)
362 99437157 2018-11-11 stsp return got_error(GOT_ERR_CANCELLED);
363 99437157 2018-11-11 stsp return NULL;
364 8069f636 2019-01-12 stsp }
365 8069f636 2019-01-12 stsp
366 8069f636 2019-01-12 stsp static const struct got_error *
367 024e9686 2019-05-14 stsp check_linear_ancestry(struct got_object_id *commit_id,
368 024e9686 2019-05-14 stsp struct got_object_id *base_commit_id, struct got_repository *repo)
369 8069f636 2019-01-12 stsp {
370 d5bea539 2019-05-13 stsp const struct got_error *err = NULL;
371 024e9686 2019-05-14 stsp struct got_object_id *yca_id;
372 8069f636 2019-01-12 stsp
373 d5bea539 2019-05-13 stsp err = got_commit_graph_find_youngest_common_ancestor(&yca_id,
374 d5bea539 2019-05-13 stsp commit_id, base_commit_id, repo);
375 36a38700 2019-05-10 stsp if (err)
376 36a38700 2019-05-10 stsp return err;
377 8069f636 2019-01-12 stsp
378 d5bea539 2019-05-13 stsp if (yca_id == NULL)
379 d5bea539 2019-05-13 stsp return got_error(GOT_ERR_ANCESTRY);
380 8069f636 2019-01-12 stsp
381 d5bea539 2019-05-13 stsp /*
382 d5bea539 2019-05-13 stsp * Require a straight line of history between the target commit
383 d5bea539 2019-05-13 stsp * and the work tree's base commit.
384 d5bea539 2019-05-13 stsp *
385 d5751d49 2019-05-14 stsp * Non-linear situations such as this require a rebase:
386 d5bea539 2019-05-13 stsp *
387 d5bea539 2019-05-13 stsp * (commit) D F (base_commit)
388 d5bea539 2019-05-13 stsp * \ /
389 d5bea539 2019-05-13 stsp * C E
390 d5bea539 2019-05-13 stsp * \ /
391 d5bea539 2019-05-13 stsp * B (yca)
392 d5bea539 2019-05-13 stsp * |
393 d5bea539 2019-05-13 stsp * A
394 d5bea539 2019-05-13 stsp *
395 d5bea539 2019-05-13 stsp * 'got update' only handles linear cases:
396 d5bea539 2019-05-13 stsp * Update forwards in time: A (base/yca) - B - C - D (commit)
397 efa2b6f7 2019-05-14 stsp * Update backwards in time: D (base) - C - B - A (commit/yca)
398 d5bea539 2019-05-13 stsp */
399 d5bea539 2019-05-13 stsp if (got_object_id_cmp(commit_id, yca_id) != 0 &&
400 d5bea539 2019-05-13 stsp got_object_id_cmp(base_commit_id, yca_id) != 0)
401 d5bea539 2019-05-13 stsp return got_error(GOT_ERR_ANCESTRY);
402 8069f636 2019-01-12 stsp
403 d5bea539 2019-05-13 stsp free(yca_id);
404 d5bea539 2019-05-13 stsp return NULL;
405 c09a553d 2018-03-12 stsp }
406 a367ff0f 2019-05-14 stsp
407 a367ff0f 2019-05-14 stsp static const struct got_error *
408 a367ff0f 2019-05-14 stsp check_same_branch(struct got_object_id *commit_id,
409 a367ff0f 2019-05-14 stsp struct got_reference *head_ref, struct got_repository *repo)
410 a367ff0f 2019-05-14 stsp {
411 a367ff0f 2019-05-14 stsp const struct got_error *err = NULL;
412 a367ff0f 2019-05-14 stsp struct got_commit_graph *graph = NULL;
413 a367ff0f 2019-05-14 stsp struct got_object_id *head_commit_id = NULL;
414 a367ff0f 2019-05-14 stsp int is_same_branch = 0;
415 a367ff0f 2019-05-14 stsp
416 a367ff0f 2019-05-14 stsp err = got_ref_resolve(&head_commit_id, repo, head_ref);
417 a367ff0f 2019-05-14 stsp if (err)
418 a367ff0f 2019-05-14 stsp goto done;
419 a367ff0f 2019-05-14 stsp
420 a367ff0f 2019-05-14 stsp err = got_commit_graph_open(&graph, head_commit_id, "/", 1, repo);
421 a367ff0f 2019-05-14 stsp if (err)
422 a367ff0f 2019-05-14 stsp goto done;
423 a367ff0f 2019-05-14 stsp
424 a367ff0f 2019-05-14 stsp err = got_commit_graph_iter_start(graph, head_commit_id, repo);
425 a367ff0f 2019-05-14 stsp if (err)
426 a367ff0f 2019-05-14 stsp goto done;
427 a367ff0f 2019-05-14 stsp
428 a367ff0f 2019-05-14 stsp for (;;) {
429 a367ff0f 2019-05-14 stsp struct got_object_id *id;
430 a367ff0f 2019-05-14 stsp err = got_commit_graph_iter_next(&id, graph);
431 a367ff0f 2019-05-14 stsp if (err) {
432 a367ff0f 2019-05-14 stsp if (err->code == GOT_ERR_ITER_COMPLETED) {
433 a367ff0f 2019-05-14 stsp err = NULL;
434 a367ff0f 2019-05-14 stsp break;
435 a367ff0f 2019-05-14 stsp }
436 a367ff0f 2019-05-14 stsp else if (err->code != GOT_ERR_ITER_NEED_MORE)
437 a367ff0f 2019-05-14 stsp break;
438 a367ff0f 2019-05-14 stsp err = got_commit_graph_fetch_commits(graph, 1,
439 a367ff0f 2019-05-14 stsp repo);
440 a367ff0f 2019-05-14 stsp if (err)
441 a367ff0f 2019-05-14 stsp break;
442 a367ff0f 2019-05-14 stsp }
443 c09a553d 2018-03-12 stsp
444 a367ff0f 2019-05-14 stsp if (id) {
445 a367ff0f 2019-05-14 stsp if (got_object_id_cmp(id, commit_id) == 0) {
446 a367ff0f 2019-05-14 stsp is_same_branch = 1;
447 a367ff0f 2019-05-14 stsp break;
448 a367ff0f 2019-05-14 stsp }
449 a367ff0f 2019-05-14 stsp }
450 a367ff0f 2019-05-14 stsp }
451 a367ff0f 2019-05-14 stsp done:
452 a367ff0f 2019-05-14 stsp if (graph)
453 a367ff0f 2019-05-14 stsp got_commit_graph_close(graph);
454 a367ff0f 2019-05-14 stsp free(head_commit_id);
455 a367ff0f 2019-05-14 stsp if (!err && !is_same_branch)
456 a367ff0f 2019-05-14 stsp err = got_error(GOT_ERR_ANCESTRY);
457 a367ff0f 2019-05-14 stsp return err;
458 a367ff0f 2019-05-14 stsp }
459 8069f636 2019-01-12 stsp
460 4ed7e80c 2018-05-20 stsp static const struct got_error *
461 c09a553d 2018-03-12 stsp cmd_checkout(int argc, char *argv[])
462 c09a553d 2018-03-12 stsp {
463 c09a553d 2018-03-12 stsp const struct got_error *error = NULL;
464 c09a553d 2018-03-12 stsp struct got_repository *repo = NULL;
465 c09a553d 2018-03-12 stsp struct got_reference *head_ref = NULL;
466 c09a553d 2018-03-12 stsp struct got_worktree *worktree = NULL;
467 c09a553d 2018-03-12 stsp char *repo_path = NULL;
468 c09a553d 2018-03-12 stsp char *worktree_path = NULL;
469 0bb8a95e 2018-03-12 stsp const char *path_prefix = "";
470 08573d5b 2019-05-14 stsp const char *branch_name = GOT_REF_HEAD;
471 8069f636 2019-01-12 stsp char *commit_id_str = NULL;
472 72151b04 2019-05-11 stsp int ch, same_path_prefix;
473 c09a553d 2018-03-12 stsp
474 08573d5b 2019-05-14 stsp while ((ch = getopt(argc, argv, "b:c:p:")) != -1) {
475 0bb8a95e 2018-03-12 stsp switch (ch) {
476 08573d5b 2019-05-14 stsp case 'b':
477 08573d5b 2019-05-14 stsp branch_name = optarg;
478 08573d5b 2019-05-14 stsp break;
479 8069f636 2019-01-12 stsp case 'c':
480 8069f636 2019-01-12 stsp commit_id_str = strdup(optarg);
481 8069f636 2019-01-12 stsp if (commit_id_str == NULL)
482 638f9024 2019-05-13 stsp return got_error_from_errno("strdup");
483 8069f636 2019-01-12 stsp break;
484 0bb8a95e 2018-03-12 stsp case 'p':
485 0bb8a95e 2018-03-12 stsp path_prefix = optarg;
486 0bb8a95e 2018-03-12 stsp break;
487 0bb8a95e 2018-03-12 stsp default:
488 2deda0b9 2019-03-07 stsp usage_checkout();
489 0bb8a95e 2018-03-12 stsp /* NOTREACHED */
490 0bb8a95e 2018-03-12 stsp }
491 0bb8a95e 2018-03-12 stsp }
492 0bb8a95e 2018-03-12 stsp
493 0bb8a95e 2018-03-12 stsp argc -= optind;
494 0bb8a95e 2018-03-12 stsp argv += optind;
495 0bb8a95e 2018-03-12 stsp
496 6715a751 2018-03-16 stsp #ifndef PROFILE
497 68ed9ba5 2019-02-10 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
498 68ed9ba5 2019-02-10 stsp "unveil", NULL) == -1)
499 c09a553d 2018-03-12 stsp err(1, "pledge");
500 6715a751 2018-03-16 stsp #endif
501 0bb8a95e 2018-03-12 stsp if (argc == 1) {
502 c09a553d 2018-03-12 stsp char *cwd, *base, *dotgit;
503 76089277 2018-04-01 stsp repo_path = realpath(argv[0], NULL);
504 76089277 2018-04-01 stsp if (repo_path == NULL)
505 638f9024 2019-05-13 stsp return got_error_from_errno2("realpath", argv[0]);
506 c09a553d 2018-03-12 stsp cwd = getcwd(NULL, 0);
507 76089277 2018-04-01 stsp if (cwd == NULL) {
508 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
509 76089277 2018-04-01 stsp goto done;
510 76089277 2018-04-01 stsp }
511 230a42bd 2019-05-11 jcs if (path_prefix[0]) {
512 230a42bd 2019-05-11 jcs base = basename(path_prefix);
513 230a42bd 2019-05-11 jcs if (base == NULL) {
514 638f9024 2019-05-13 stsp error = got_error_from_errno2("basename",
515 230a42bd 2019-05-11 jcs path_prefix);
516 230a42bd 2019-05-11 jcs goto done;
517 230a42bd 2019-05-11 jcs }
518 230a42bd 2019-05-11 jcs } else {
519 5d7c1dab 2018-04-01 stsp base = basename(repo_path);
520 230a42bd 2019-05-11 jcs if (base == NULL) {
521 638f9024 2019-05-13 stsp error = got_error_from_errno2("basename",
522 230a42bd 2019-05-11 jcs repo_path);
523 230a42bd 2019-05-11 jcs goto done;
524 230a42bd 2019-05-11 jcs }
525 76089277 2018-04-01 stsp }
526 c09a553d 2018-03-12 stsp dotgit = strstr(base, ".git");
527 c09a553d 2018-03-12 stsp if (dotgit)
528 c09a553d 2018-03-12 stsp *dotgit = '\0';
529 c09a553d 2018-03-12 stsp if (asprintf(&worktree_path, "%s/%s", cwd, base) == -1) {
530 638f9024 2019-05-13 stsp error = got_error_from_errno("asprintf");
531 c09a553d 2018-03-12 stsp free(cwd);
532 76089277 2018-04-01 stsp goto done;
533 c09a553d 2018-03-12 stsp }
534 c09a553d 2018-03-12 stsp free(cwd);
535 0bb8a95e 2018-03-12 stsp } else if (argc == 2) {
536 76089277 2018-04-01 stsp repo_path = realpath(argv[0], NULL);
537 76089277 2018-04-01 stsp if (repo_path == NULL) {
538 638f9024 2019-05-13 stsp error = got_error_from_errno2("realpath", argv[0]);
539 76089277 2018-04-01 stsp goto done;
540 76089277 2018-04-01 stsp }
541 f7b38925 2018-04-01 stsp worktree_path = realpath(argv[1], NULL);
542 76089277 2018-04-01 stsp if (worktree_path == NULL) {
543 638f9024 2019-05-13 stsp error = got_error_from_errno2("realpath", argv[1]);
544 76089277 2018-04-01 stsp goto done;
545 76089277 2018-04-01 stsp }
546 c09a553d 2018-03-12 stsp } else
547 c09a553d 2018-03-12 stsp usage_checkout();
548 c09a553d 2018-03-12 stsp
549 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
550 72151b04 2019-05-11 stsp got_path_strip_trailing_slashes(worktree_path);
551 13bfb272 2019-05-10 jcs
552 c09a553d 2018-03-12 stsp error = got_repo_open(&repo, repo_path);
553 c09a553d 2018-03-12 stsp if (error != NULL)
554 c02c541e 2019-03-29 stsp goto done;
555 c02c541e 2019-03-29 stsp
556 314a6357 2019-05-13 stsp error = apply_unveil(got_repo_get_path(repo), 0, worktree_path, 1);
557 c02c541e 2019-03-29 stsp if (error)
558 c09a553d 2018-03-12 stsp goto done;
559 8069f636 2019-01-12 stsp
560 08573d5b 2019-05-14 stsp error = got_ref_open(&head_ref, repo, branch_name, 0);
561 c09a553d 2018-03-12 stsp if (error != NULL)
562 c09a553d 2018-03-12 stsp goto done;
563 c09a553d 2018-03-12 stsp
564 0bb8a95e 2018-03-12 stsp error = got_worktree_init(worktree_path, head_ref, path_prefix, repo);
565 d70b8e30 2018-12-27 stsp if (error != NULL && !(error->code == GOT_ERR_ERRNO && errno == EEXIST))
566 c09a553d 2018-03-12 stsp goto done;
567 c09a553d 2018-03-12 stsp
568 c09a553d 2018-03-12 stsp error = got_worktree_open(&worktree, worktree_path);
569 c09a553d 2018-03-12 stsp if (error != NULL)
570 c09a553d 2018-03-12 stsp goto done;
571 c09a553d 2018-03-12 stsp
572 e5dc7198 2018-12-29 stsp error = got_worktree_match_path_prefix(&same_path_prefix, worktree,
573 e5dc7198 2018-12-29 stsp path_prefix);
574 e5dc7198 2018-12-29 stsp if (error != NULL)
575 e5dc7198 2018-12-29 stsp goto done;
576 e5dc7198 2018-12-29 stsp if (!same_path_prefix) {
577 49520a32 2018-12-29 stsp error = got_error(GOT_ERR_PATH_PREFIX);
578 49520a32 2018-12-29 stsp goto done;
579 49520a32 2018-12-29 stsp }
580 49520a32 2018-12-29 stsp
581 8069f636 2019-01-12 stsp if (commit_id_str) {
582 8069f636 2019-01-12 stsp struct got_object_id *commit_id;
583 8069f636 2019-01-12 stsp error = got_object_resolve_id_str(&commit_id, repo,
584 8069f636 2019-01-12 stsp commit_id_str);
585 8069f636 2019-01-12 stsp if (error != NULL)
586 8069f636 2019-01-12 stsp goto done;
587 024e9686 2019-05-14 stsp error = check_linear_ancestry(commit_id,
588 024e9686 2019-05-14 stsp got_worktree_get_base_commit_id(worktree), repo);
589 8069f636 2019-01-12 stsp if (error != NULL) {
590 8069f636 2019-01-12 stsp free(commit_id);
591 8069f636 2019-01-12 stsp goto done;
592 8069f636 2019-01-12 stsp }
593 45d344f6 2019-05-14 stsp error = check_same_branch(commit_id, head_ref, repo);
594 45d344f6 2019-05-14 stsp if (error)
595 45d344f6 2019-05-14 stsp goto done;
596 8069f636 2019-01-12 stsp error = got_worktree_set_base_commit_id(worktree, repo,
597 8069f636 2019-01-12 stsp commit_id);
598 8069f636 2019-01-12 stsp free(commit_id);
599 8069f636 2019-01-12 stsp if (error)
600 8069f636 2019-01-12 stsp goto done;
601 8069f636 2019-01-12 stsp }
602 8069f636 2019-01-12 stsp
603 c4cdcb68 2019-04-03 stsp error = got_worktree_checkout_files(worktree, "", repo,
604 6bad629b 2019-02-04 stsp checkout_progress, worktree_path, check_cancelled, NULL);
605 c09a553d 2018-03-12 stsp if (error != NULL)
606 c09a553d 2018-03-12 stsp goto done;
607 c09a553d 2018-03-12 stsp
608 b65ae19a 2018-04-24 stsp printf("Now shut up and hack\n");
609 c09a553d 2018-03-12 stsp
610 c09a553d 2018-03-12 stsp done:
611 8069f636 2019-01-12 stsp free(commit_id_str);
612 76089277 2018-04-01 stsp free(repo_path);
613 507dc3bb 2018-12-29 stsp free(worktree_path);
614 507dc3bb 2018-12-29 stsp return error;
615 507dc3bb 2018-12-29 stsp }
616 507dc3bb 2018-12-29 stsp
617 507dc3bb 2018-12-29 stsp __dead static void
618 507dc3bb 2018-12-29 stsp usage_update(void)
619 507dc3bb 2018-12-29 stsp {
620 024e9686 2019-05-14 stsp fprintf(stderr, "usage: %s update [-b branch] [-c commit] [path]\n",
621 507dc3bb 2018-12-29 stsp getprogname());
622 507dc3bb 2018-12-29 stsp exit(1);
623 507dc3bb 2018-12-29 stsp }
624 507dc3bb 2018-12-29 stsp
625 507dc3bb 2018-12-29 stsp static void
626 507dc3bb 2018-12-29 stsp update_progress(void *arg, unsigned char status, const char *path)
627 507dc3bb 2018-12-29 stsp {
628 784955db 2019-01-12 stsp int *did_something = arg;
629 784955db 2019-01-12 stsp
630 507dc3bb 2018-12-29 stsp if (status == GOT_STATUS_EXISTS)
631 507dc3bb 2018-12-29 stsp return;
632 507dc3bb 2018-12-29 stsp
633 1545c615 2019-02-10 stsp *did_something = 1;
634 a484d721 2019-06-10 stsp
635 a484d721 2019-06-10 stsp /* Base commit bump happens silently. */
636 a484d721 2019-06-10 stsp if (status == GOT_STATUS_BUMP_BASE)
637 a484d721 2019-06-10 stsp return;
638 a484d721 2019-06-10 stsp
639 507dc3bb 2018-12-29 stsp while (path[0] == '/')
640 507dc3bb 2018-12-29 stsp path++;
641 507dc3bb 2018-12-29 stsp printf("%c %s\n", status, path);
642 be7061eb 2018-12-30 stsp }
643 be7061eb 2018-12-30 stsp
644 be7061eb 2018-12-30 stsp static const struct got_error *
645 a1fb16d8 2019-05-24 stsp switch_head_ref(struct got_reference *head_ref,
646 a1fb16d8 2019-05-24 stsp struct got_object_id *commit_id, struct got_worktree *worktree,
647 a1fb16d8 2019-05-24 stsp struct got_repository *repo)
648 a1fb16d8 2019-05-24 stsp {
649 a1fb16d8 2019-05-24 stsp const struct got_error *err = NULL;
650 a1fb16d8 2019-05-24 stsp char *base_id_str;
651 a1fb16d8 2019-05-24 stsp int ref_has_moved = 0;
652 a1fb16d8 2019-05-24 stsp
653 a1fb16d8 2019-05-24 stsp /* Trivial case: switching between two different references. */
654 a1fb16d8 2019-05-24 stsp if (strcmp(got_ref_get_name(head_ref),
655 a1fb16d8 2019-05-24 stsp got_worktree_get_head_ref_name(worktree)) != 0) {
656 a1fb16d8 2019-05-24 stsp printf("Switching work tree from %s to %s\n",
657 a1fb16d8 2019-05-24 stsp got_worktree_get_head_ref_name(worktree),
658 a1fb16d8 2019-05-24 stsp got_ref_get_name(head_ref));
659 a1fb16d8 2019-05-24 stsp return got_worktree_set_head_ref(worktree, head_ref);
660 a1fb16d8 2019-05-24 stsp }
661 a1fb16d8 2019-05-24 stsp
662 a1fb16d8 2019-05-24 stsp err = check_linear_ancestry(commit_id,
663 a1fb16d8 2019-05-24 stsp got_worktree_get_base_commit_id(worktree), repo);
664 a1fb16d8 2019-05-24 stsp if (err) {
665 a1fb16d8 2019-05-24 stsp if (err->code != GOT_ERR_ANCESTRY)
666 a1fb16d8 2019-05-24 stsp return err;
667 a1fb16d8 2019-05-24 stsp ref_has_moved = 1;
668 a1fb16d8 2019-05-24 stsp }
669 a1fb16d8 2019-05-24 stsp if (!ref_has_moved)
670 a1fb16d8 2019-05-24 stsp return NULL;
671 a1fb16d8 2019-05-24 stsp
672 a1fb16d8 2019-05-24 stsp /* Switching to a rebased branch with the same reference name. */
673 a1fb16d8 2019-05-24 stsp err = got_object_id_str(&base_id_str,
674 a1fb16d8 2019-05-24 stsp got_worktree_get_base_commit_id(worktree));
675 a1fb16d8 2019-05-24 stsp if (err)
676 a1fb16d8 2019-05-24 stsp return err;
677 a1fb16d8 2019-05-24 stsp printf("Reference %s now points at a different branch\n",
678 a1fb16d8 2019-05-24 stsp got_worktree_get_head_ref_name(worktree));
679 a1fb16d8 2019-05-24 stsp printf("Switching work tree from %s to %s\n", base_id_str,
680 a1fb16d8 2019-05-24 stsp got_worktree_get_head_ref_name(worktree));
681 a1fb16d8 2019-05-24 stsp return NULL;
682 a1fb16d8 2019-05-24 stsp }
683 a1fb16d8 2019-05-24 stsp
684 a1fb16d8 2019-05-24 stsp static const struct got_error *
685 507dc3bb 2018-12-29 stsp cmd_update(int argc, char *argv[])
686 507dc3bb 2018-12-29 stsp {
687 507dc3bb 2018-12-29 stsp const struct got_error *error = NULL;
688 507dc3bb 2018-12-29 stsp struct got_repository *repo = NULL;
689 507dc3bb 2018-12-29 stsp struct got_worktree *worktree = NULL;
690 c4cdcb68 2019-04-03 stsp char *worktree_path = NULL, *path = NULL;
691 507dc3bb 2018-12-29 stsp struct got_object_id *commit_id = NULL;
692 9c4b8182 2019-01-02 stsp char *commit_id_str = NULL;
693 024e9686 2019-05-14 stsp const char *branch_name = NULL;
694 024e9686 2019-05-14 stsp struct got_reference *head_ref = NULL;
695 784955db 2019-01-12 stsp int ch, did_something = 0;
696 507dc3bb 2018-12-29 stsp
697 024e9686 2019-05-14 stsp while ((ch = getopt(argc, argv, "b:c:")) != -1) {
698 507dc3bb 2018-12-29 stsp switch (ch) {
699 024e9686 2019-05-14 stsp case 'b':
700 024e9686 2019-05-14 stsp branch_name = optarg;
701 024e9686 2019-05-14 stsp break;
702 507dc3bb 2018-12-29 stsp case 'c':
703 9c4b8182 2019-01-02 stsp commit_id_str = strdup(optarg);
704 9c4b8182 2019-01-02 stsp if (commit_id_str == NULL)
705 638f9024 2019-05-13 stsp return got_error_from_errno("strdup");
706 507dc3bb 2018-12-29 stsp break;
707 507dc3bb 2018-12-29 stsp default:
708 2deda0b9 2019-03-07 stsp usage_update();
709 507dc3bb 2018-12-29 stsp /* NOTREACHED */
710 507dc3bb 2018-12-29 stsp }
711 507dc3bb 2018-12-29 stsp }
712 507dc3bb 2018-12-29 stsp
713 507dc3bb 2018-12-29 stsp argc -= optind;
714 507dc3bb 2018-12-29 stsp argv += optind;
715 507dc3bb 2018-12-29 stsp
716 507dc3bb 2018-12-29 stsp #ifndef PROFILE
717 68ed9ba5 2019-02-10 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec sendfd "
718 68ed9ba5 2019-02-10 stsp "unveil", NULL) == -1)
719 507dc3bb 2018-12-29 stsp err(1, "pledge");
720 507dc3bb 2018-12-29 stsp #endif
721 c4cdcb68 2019-04-03 stsp worktree_path = getcwd(NULL, 0);
722 c4cdcb68 2019-04-03 stsp if (worktree_path == NULL) {
723 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
724 c4cdcb68 2019-04-03 stsp goto done;
725 c4cdcb68 2019-04-03 stsp }
726 c4cdcb68 2019-04-03 stsp error = got_worktree_open(&worktree, worktree_path);
727 c4cdcb68 2019-04-03 stsp if (error)
728 c4cdcb68 2019-04-03 stsp goto done;
729 c4cdcb68 2019-04-03 stsp
730 507dc3bb 2018-12-29 stsp if (argc == 0) {
731 c4cdcb68 2019-04-03 stsp path = strdup("");
732 c4cdcb68 2019-04-03 stsp if (path == NULL) {
733 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
734 507dc3bb 2018-12-29 stsp goto done;
735 507dc3bb 2018-12-29 stsp }
736 507dc3bb 2018-12-29 stsp } else if (argc == 1) {
737 c4cdcb68 2019-04-03 stsp error = got_worktree_resolve_path(&path, worktree, argv[0]);
738 c4cdcb68 2019-04-03 stsp if (error)
739 507dc3bb 2018-12-29 stsp goto done;
740 507dc3bb 2018-12-29 stsp } else
741 507dc3bb 2018-12-29 stsp usage_update();
742 507dc3bb 2018-12-29 stsp
743 507dc3bb 2018-12-29 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
744 507dc3bb 2018-12-29 stsp if (error != NULL)
745 507dc3bb 2018-12-29 stsp goto done;
746 507dc3bb 2018-12-29 stsp
747 97430839 2019-03-11 stsp error = apply_unveil(got_repo_get_path(repo), 0,
748 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
749 0266afb7 2019-01-04 stsp if (error)
750 0266afb7 2019-01-04 stsp goto done;
751 0266afb7 2019-01-04 stsp
752 a1fb16d8 2019-05-24 stsp error = got_ref_open(&head_ref, repo, branch_name ? branch_name :
753 a1fb16d8 2019-05-24 stsp got_worktree_get_head_ref_name(worktree), 0);
754 024e9686 2019-05-14 stsp if (error != NULL)
755 024e9686 2019-05-14 stsp goto done;
756 507dc3bb 2018-12-29 stsp if (commit_id_str == NULL) {
757 507dc3bb 2018-12-29 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
758 9c4b8182 2019-01-02 stsp if (error != NULL)
759 9c4b8182 2019-01-02 stsp goto done;
760 9c4b8182 2019-01-02 stsp error = got_object_id_str(&commit_id_str, commit_id);
761 507dc3bb 2018-12-29 stsp if (error != NULL)
762 507dc3bb 2018-12-29 stsp goto done;
763 507dc3bb 2018-12-29 stsp } else {
764 507dc3bb 2018-12-29 stsp error = got_object_resolve_id_str(&commit_id, repo,
765 507dc3bb 2018-12-29 stsp commit_id_str);
766 507dc3bb 2018-12-29 stsp if (error != NULL)
767 507dc3bb 2018-12-29 stsp goto done;
768 507dc3bb 2018-12-29 stsp }
769 35c965b2 2018-12-29 stsp
770 a1fb16d8 2019-05-24 stsp if (branch_name) {
771 024e9686 2019-05-14 stsp struct got_object_id *head_commit_id;
772 024e9686 2019-05-14 stsp if (strlen(path) != 0) {
773 a1fb16d8 2019-05-24 stsp fprintf(stderr, "%s: switching between branches "
774 a1fb16d8 2019-05-24 stsp "requires that the entire work tree "
775 024e9686 2019-05-14 stsp "gets updated, not just '%s'\n",
776 024e9686 2019-05-14 stsp getprogname(), path);
777 024e9686 2019-05-14 stsp error = got_error(GOT_ERR_BAD_PATH);
778 024e9686 2019-05-14 stsp goto done;
779 024e9686 2019-05-14 stsp }
780 024e9686 2019-05-14 stsp error = got_ref_resolve(&head_commit_id, repo, head_ref);
781 024e9686 2019-05-14 stsp if (error)
782 024e9686 2019-05-14 stsp goto done;
783 024e9686 2019-05-14 stsp error = check_linear_ancestry(commit_id, head_commit_id, repo);
784 a367ff0f 2019-05-14 stsp free(head_commit_id);
785 024e9686 2019-05-14 stsp if (error != NULL)
786 024e9686 2019-05-14 stsp goto done;
787 a367ff0f 2019-05-14 stsp error = check_same_branch(commit_id, head_ref, repo);
788 a367ff0f 2019-05-14 stsp if (error)
789 a367ff0f 2019-05-14 stsp goto done;
790 a1fb16d8 2019-05-24 stsp error = switch_head_ref(head_ref, commit_id, worktree, repo);
791 024e9686 2019-05-14 stsp if (error)
792 024e9686 2019-05-14 stsp goto done;
793 024e9686 2019-05-14 stsp } else {
794 024e9686 2019-05-14 stsp error = check_linear_ancestry(commit_id,
795 024e9686 2019-05-14 stsp got_worktree_get_base_commit_id(worktree), repo);
796 a367ff0f 2019-05-14 stsp if (error != NULL) {
797 a367ff0f 2019-05-14 stsp if (error->code == GOT_ERR_ANCESTRY)
798 a367ff0f 2019-05-14 stsp error = got_error(GOT_ERR_BRANCH_MOVED);
799 024e9686 2019-05-14 stsp goto done;
800 a367ff0f 2019-05-14 stsp }
801 a367ff0f 2019-05-14 stsp error = check_same_branch(commit_id, head_ref, repo);
802 a367ff0f 2019-05-14 stsp if (error)
803 a367ff0f 2019-05-14 stsp goto done;
804 024e9686 2019-05-14 stsp }
805 507dc3bb 2018-12-29 stsp
806 507dc3bb 2018-12-29 stsp if (got_object_id_cmp(got_worktree_get_base_commit_id(worktree),
807 507dc3bb 2018-12-29 stsp commit_id) != 0) {
808 507dc3bb 2018-12-29 stsp error = got_worktree_set_base_commit_id(worktree, repo,
809 507dc3bb 2018-12-29 stsp commit_id);
810 507dc3bb 2018-12-29 stsp if (error)
811 507dc3bb 2018-12-29 stsp goto done;
812 507dc3bb 2018-12-29 stsp }
813 507dc3bb 2018-12-29 stsp
814 c4cdcb68 2019-04-03 stsp error = got_worktree_checkout_files(worktree, path, repo,
815 6bad629b 2019-02-04 stsp update_progress, &did_something, check_cancelled, NULL);
816 507dc3bb 2018-12-29 stsp if (error != NULL)
817 507dc3bb 2018-12-29 stsp goto done;
818 9c4b8182 2019-01-02 stsp
819 784955db 2019-01-12 stsp if (did_something)
820 784955db 2019-01-12 stsp printf("Updated to commit %s\n", commit_id_str);
821 784955db 2019-01-12 stsp else
822 784955db 2019-01-12 stsp printf("Already up-to-date\n");
823 507dc3bb 2018-12-29 stsp done:
824 c09a553d 2018-03-12 stsp free(worktree_path);
825 c4cdcb68 2019-04-03 stsp free(path);
826 507dc3bb 2018-12-29 stsp free(commit_id);
827 9c4b8182 2019-01-02 stsp free(commit_id_str);
828 c09a553d 2018-03-12 stsp return error;
829 c09a553d 2018-03-12 stsp }
830 c09a553d 2018-03-12 stsp
831 f42b1b34 2018-03-12 stsp static const struct got_error *
832 79109fed 2018-03-27 stsp print_patch(struct got_commit_object *commit, struct got_object_id *id,
833 c0cc5c62 2018-10-18 stsp int diff_context, struct got_repository *repo)
834 5c860e29 2018-03-12 stsp {
835 f42b1b34 2018-03-12 stsp const struct got_error *err = NULL;
836 79109fed 2018-03-27 stsp struct got_tree_object *tree1 = NULL, *tree2;
837 79f35eb3 2018-06-11 stsp struct got_object_qid *qid;
838 0f2b3dca 2018-12-22 stsp char *id_str1 = NULL, *id_str2;
839 aaa13589 2019-06-01 stsp struct got_diff_blob_output_unidiff_arg arg;
840 79109fed 2018-03-27 stsp
841 45d799e2 2018-12-23 stsp err = got_object_open_as_tree(&tree2, repo,
842 45d799e2 2018-12-23 stsp got_object_commit_get_tree_id(commit));
843 79109fed 2018-03-27 stsp if (err)
844 79109fed 2018-03-27 stsp return err;
845 79109fed 2018-03-27 stsp
846 45d799e2 2018-12-23 stsp qid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
847 79f35eb3 2018-06-11 stsp if (qid != NULL) {
848 79109fed 2018-03-27 stsp struct got_commit_object *pcommit;
849 79109fed 2018-03-27 stsp
850 117e771c 2018-07-23 stsp err = got_object_open_as_commit(&pcommit, repo, qid->id);
851 79109fed 2018-03-27 stsp if (err)
852 79109fed 2018-03-27 stsp return err;
853 79109fed 2018-03-27 stsp
854 45d799e2 2018-12-23 stsp err = got_object_open_as_tree(&tree1, repo,
855 45d799e2 2018-12-23 stsp got_object_commit_get_tree_id(pcommit));
856 79109fed 2018-03-27 stsp got_object_commit_close(pcommit);
857 0f2b3dca 2018-12-22 stsp if (err)
858 0f2b3dca 2018-12-22 stsp return err;
859 0f2b3dca 2018-12-22 stsp
860 0f2b3dca 2018-12-22 stsp err = got_object_id_str(&id_str1, qid->id);
861 79109fed 2018-03-27 stsp if (err)
862 79109fed 2018-03-27 stsp return err;
863 79109fed 2018-03-27 stsp }
864 79109fed 2018-03-27 stsp
865 0f2b3dca 2018-12-22 stsp err = got_object_id_str(&id_str2, id);
866 0f2b3dca 2018-12-22 stsp if (err)
867 0f2b3dca 2018-12-22 stsp goto done;
868 0f2b3dca 2018-12-22 stsp
869 56765ebb 2018-12-23 stsp printf("diff %s %s\n", id_str1 ? id_str1 : "/dev/null", id_str2);
870 aaa13589 2019-06-01 stsp arg.diff_context = diff_context;
871 aaa13589 2019-06-01 stsp arg.outfile = stdout;
872 aaa13589 2019-06-01 stsp err = got_diff_tree(tree1, tree2, "", "", repo,
873 aaa13589 2019-06-01 stsp got_diff_blob_output_unidiff, &arg);
874 0f2b3dca 2018-12-22 stsp done:
875 79109fed 2018-03-27 stsp if (tree1)
876 79109fed 2018-03-27 stsp got_object_tree_close(tree1);
877 79109fed 2018-03-27 stsp got_object_tree_close(tree2);
878 0f2b3dca 2018-12-22 stsp free(id_str1);
879 0f2b3dca 2018-12-22 stsp free(id_str2);
880 79109fed 2018-03-27 stsp return err;
881 79109fed 2018-03-27 stsp }
882 79109fed 2018-03-27 stsp
883 4bb494d5 2018-06-16 stsp static char *
884 6c281f94 2018-06-11 stsp get_datestr(time_t *time, char *datebuf)
885 6c281f94 2018-06-11 stsp {
886 6c281f94 2018-06-11 stsp char *p, *s = ctime_r(time, datebuf);
887 6c281f94 2018-06-11 stsp p = strchr(s, '\n');
888 6c281f94 2018-06-11 stsp if (p)
889 6c281f94 2018-06-11 stsp *p = '\0';
890 6c281f94 2018-06-11 stsp return s;
891 6c281f94 2018-06-11 stsp }
892 6c281f94 2018-06-11 stsp
893 79109fed 2018-03-27 stsp static const struct got_error *
894 79109fed 2018-03-27 stsp print_commit(struct got_commit_object *commit, struct got_object_id *id,
895 199a4027 2019-02-02 stsp struct got_repository *repo, int show_patch, int diff_context,
896 199a4027 2019-02-02 stsp struct got_reflist_head *refs)
897 79109fed 2018-03-27 stsp {
898 79109fed 2018-03-27 stsp const struct got_error *err = NULL;
899 621015ac 2018-07-23 stsp char *id_str, *datestr, *logmsg0, *logmsg, *line;
900 6c281f94 2018-06-11 stsp char datebuf[26];
901 45d799e2 2018-12-23 stsp time_t committer_time;
902 45d799e2 2018-12-23 stsp const char *author, *committer;
903 199a4027 2019-02-02 stsp char *refs_str = NULL;
904 199a4027 2019-02-02 stsp struct got_reflist_entry *re;
905 5c860e29 2018-03-12 stsp
906 199a4027 2019-02-02 stsp SIMPLEQ_FOREACH(re, refs, entry) {
907 199a4027 2019-02-02 stsp char *s;
908 199a4027 2019-02-02 stsp const char *name;
909 199a4027 2019-02-02 stsp if (got_object_id_cmp(re->id, id) != 0)
910 199a4027 2019-02-02 stsp continue;
911 199a4027 2019-02-02 stsp name = got_ref_get_name(re->ref);
912 d9498b20 2019-02-05 stsp if (strcmp(name, GOT_REF_HEAD) == 0)
913 d9498b20 2019-02-05 stsp continue;
914 199a4027 2019-02-02 stsp if (strncmp(name, "refs/", 5) == 0)
915 199a4027 2019-02-02 stsp name += 5;
916 7143d404 2019-03-12 stsp if (strncmp(name, "got/", 4) == 0)
917 7143d404 2019-03-12 stsp continue;
918 e34f9ed6 2019-02-02 stsp if (strncmp(name, "heads/", 6) == 0)
919 e34f9ed6 2019-02-02 stsp name += 6;
920 141c2bff 2019-02-04 stsp if (strncmp(name, "remotes/", 8) == 0)
921 141c2bff 2019-02-04 stsp name += 8;
922 199a4027 2019-02-02 stsp s = refs_str;
923 199a4027 2019-02-02 stsp if (asprintf(&refs_str, "%s%s%s", s ? s : "", s ? ", " : "",
924 199a4027 2019-02-02 stsp name) == -1) {
925 638f9024 2019-05-13 stsp err = got_error_from_errno("asprintf");
926 199a4027 2019-02-02 stsp free(s);
927 199a4027 2019-02-02 stsp break;
928 199a4027 2019-02-02 stsp }
929 199a4027 2019-02-02 stsp free(s);
930 199a4027 2019-02-02 stsp }
931 832c249c 2018-06-10 stsp err = got_object_id_str(&id_str, id);
932 8bf5b3c9 2018-03-17 stsp if (err)
933 8bf5b3c9 2018-03-17 stsp return err;
934 788c352e 2018-06-16 stsp
935 33d869be 2018-12-22 stsp printf("-----------------------------------------------\n");
936 199a4027 2019-02-02 stsp printf("commit %s%s%s%s\n", id_str, refs_str ? " (" : "",
937 199a4027 2019-02-02 stsp refs_str ? refs_str : "", refs_str ? ")" : "");
938 832c249c 2018-06-10 stsp free(id_str);
939 d3d493d7 2019-02-21 stsp id_str = NULL;
940 d3d493d7 2019-02-21 stsp free(refs_str);
941 d3d493d7 2019-02-21 stsp refs_str = NULL;
942 45d799e2 2018-12-23 stsp printf("from: %s\n", got_object_commit_get_author(commit));
943 45d799e2 2018-12-23 stsp committer_time = got_object_commit_get_committer_time(commit);
944 45d799e2 2018-12-23 stsp datestr = get_datestr(&committer_time, datebuf);
945 dab5fe87 2018-09-14 stsp printf("date: %s UTC\n", datestr);
946 45d799e2 2018-12-23 stsp author = got_object_commit_get_author(commit);
947 45d799e2 2018-12-23 stsp committer = got_object_commit_get_committer(commit);
948 45d799e2 2018-12-23 stsp if (strcmp(author, committer) != 0)
949 45d799e2 2018-12-23 stsp printf("via: %s\n", committer);
950 45d799e2 2018-12-23 stsp if (got_object_commit_get_nparents(commit) > 1) {
951 45d799e2 2018-12-23 stsp const struct got_object_id_queue *parent_ids;
952 79f35eb3 2018-06-11 stsp struct got_object_qid *qid;
953 3fe1abad 2018-06-10 stsp int n = 1;
954 45d799e2 2018-12-23 stsp parent_ids = got_object_commit_get_parent_ids(commit);
955 45d799e2 2018-12-23 stsp SIMPLEQ_FOREACH(qid, parent_ids, entry) {
956 79f35eb3 2018-06-11 stsp err = got_object_id_str(&id_str, qid->id);
957 a0603db2 2018-06-10 stsp if (err)
958 a0603db2 2018-06-10 stsp return err;
959 3fe1abad 2018-06-10 stsp printf("parent %d: %s\n", n++, id_str);
960 a0603db2 2018-06-10 stsp free(id_str);
961 a0603db2 2018-06-10 stsp }
962 a0603db2 2018-06-10 stsp }
963 832c249c 2018-06-10 stsp
964 45d799e2 2018-12-23 stsp logmsg0 = strdup(got_object_commit_get_logmsg(commit));
965 621015ac 2018-07-23 stsp if (logmsg0 == NULL)
966 638f9024 2019-05-13 stsp return got_error_from_errno("strdup");
967 8bf5b3c9 2018-03-17 stsp
968 621015ac 2018-07-23 stsp logmsg = logmsg0;
969 832c249c 2018-06-10 stsp do {
970 832c249c 2018-06-10 stsp line = strsep(&logmsg, "\n");
971 832c249c 2018-06-10 stsp if (line)
972 832c249c 2018-06-10 stsp printf(" %s\n", line);
973 832c249c 2018-06-10 stsp } while (line);
974 621015ac 2018-07-23 stsp free(logmsg0);
975 832c249c 2018-06-10 stsp
976 971751ac 2018-03-27 stsp if (show_patch) {
977 c0cc5c62 2018-10-18 stsp err = print_patch(commit, id, diff_context, repo);
978 971751ac 2018-03-27 stsp if (err == 0)
979 971751ac 2018-03-27 stsp printf("\n");
980 971751ac 2018-03-27 stsp }
981 07862c20 2018-09-15 stsp
982 cbe7f848 2019-02-11 stsp if (fflush(stdout) != 0 && err == NULL)
983 638f9024 2019-05-13 stsp err = got_error_from_errno("fflush");
984 07862c20 2018-09-15 stsp return err;
985 f42b1b34 2018-03-12 stsp }
986 5c860e29 2018-03-12 stsp
987 f42b1b34 2018-03-12 stsp static const struct got_error *
988 15a94983 2018-12-23 stsp print_commits(struct got_object_id *root_id, struct got_repository *repo,
989 15a94983 2018-12-23 stsp char *path, int show_patch, int diff_context, int limit,
990 199a4027 2019-02-02 stsp int first_parent_traversal, struct got_reflist_head *refs)
991 f42b1b34 2018-03-12 stsp {
992 f42b1b34 2018-03-12 stsp const struct got_error *err;
993 372ccdbb 2018-06-10 stsp struct got_commit_graph *graph;
994 372ccdbb 2018-06-10 stsp
995 31cedeaf 2018-09-15 stsp err = got_commit_graph_open(&graph, root_id, path,
996 31cedeaf 2018-09-15 stsp first_parent_traversal, repo);
997 f42b1b34 2018-03-12 stsp if (err)
998 f42b1b34 2018-03-12 stsp return err;
999 31cedeaf 2018-09-15 stsp err = got_commit_graph_iter_start(graph, root_id, repo);
1000 372ccdbb 2018-06-10 stsp if (err)
1001 fcc85cad 2018-07-22 stsp goto done;
1002 656b1f76 2019-05-11 jcs for (;;) {
1003 372ccdbb 2018-06-10 stsp struct got_commit_object *commit;
1004 372ccdbb 2018-06-10 stsp struct got_object_id *id;
1005 8bf5b3c9 2018-03-17 stsp
1006 84453469 2018-11-11 stsp if (sigint_received || sigpipe_received)
1007 84453469 2018-11-11 stsp break;
1008 84453469 2018-11-11 stsp
1009 b43fbaa0 2018-06-11 stsp err = got_commit_graph_iter_next(&id, graph);
1010 372ccdbb 2018-06-10 stsp if (err) {
1011 9ba79e04 2018-06-11 stsp if (err->code == GOT_ERR_ITER_COMPLETED) {
1012 9ba79e04 2018-06-11 stsp err = NULL;
1013 9ba79e04 2018-06-11 stsp break;
1014 9ba79e04 2018-06-11 stsp }
1015 372ccdbb 2018-06-10 stsp if (err->code != GOT_ERR_ITER_NEED_MORE)
1016 372ccdbb 2018-06-10 stsp break;
1017 31cedeaf 2018-09-15 stsp err = got_commit_graph_fetch_commits(graph, 1, repo);
1018 372ccdbb 2018-06-10 stsp if (err)
1019 372ccdbb 2018-06-10 stsp break;
1020 372ccdbb 2018-06-10 stsp else
1021 372ccdbb 2018-06-10 stsp continue;
1022 7e665116 2018-04-02 stsp }
1023 b43fbaa0 2018-06-11 stsp if (id == NULL)
1024 7e665116 2018-04-02 stsp break;
1025 b43fbaa0 2018-06-11 stsp
1026 f8e900f3 2018-06-11 stsp err = got_object_open_as_commit(&commit, repo, id);
1027 b43fbaa0 2018-06-11 stsp if (err)
1028 fcc85cad 2018-07-22 stsp break;
1029 199a4027 2019-02-02 stsp err = print_commit(commit, id, repo, show_patch, diff_context,
1030 199a4027 2019-02-02 stsp refs);
1031 b43fbaa0 2018-06-11 stsp got_object_commit_close(commit);
1032 372ccdbb 2018-06-10 stsp if (err || (limit && --limit == 0))
1033 7e665116 2018-04-02 stsp break;
1034 31cedeaf 2018-09-15 stsp }
1035 fcc85cad 2018-07-22 stsp done:
1036 372ccdbb 2018-06-10 stsp got_commit_graph_close(graph);
1037 f42b1b34 2018-03-12 stsp return err;
1038 f42b1b34 2018-03-12 stsp }
1039 5c860e29 2018-03-12 stsp
1040 4ed7e80c 2018-05-20 stsp __dead static void
1041 6f3d1eb0 2018-03-12 stsp usage_log(void)
1042 6f3d1eb0 2018-03-12 stsp {
1043 499d7ecc 2019-05-22 stsp fprintf(stderr, "usage: %s log [-b] [-c commit] [-C number] [ -l N ] [-p] "
1044 04ca23f4 2018-07-16 stsp "[-r repository-path] [path]\n", getprogname());
1045 6f3d1eb0 2018-03-12 stsp exit(1);
1046 6f3d1eb0 2018-03-12 stsp }
1047 6f3d1eb0 2018-03-12 stsp
1048 4ed7e80c 2018-05-20 stsp static const struct got_error *
1049 f42b1b34 2018-03-12 stsp cmd_log(int argc, char *argv[])
1050 f42b1b34 2018-03-12 stsp {
1051 f42b1b34 2018-03-12 stsp const struct got_error *error;
1052 04ca23f4 2018-07-16 stsp struct got_repository *repo = NULL;
1053 cffc0aa4 2019-02-05 stsp struct got_worktree *worktree = NULL;
1054 15a94983 2018-12-23 stsp struct got_commit_object *commit = NULL;
1055 3235492e 2018-04-01 stsp struct got_object_id *id = NULL;
1056 04ca23f4 2018-07-16 stsp char *repo_path = NULL, *path = NULL, *cwd = NULL, *in_repo_path = NULL;
1057 d142fc45 2018-04-01 stsp char *start_commit = NULL;
1058 c0cc5c62 2018-10-18 stsp int diff_context = 3, ch;
1059 1fd6d7ea 2018-06-13 stsp int show_patch = 0, limit = 0, first_parent_traversal = 0;
1060 64a96a6d 2018-04-01 stsp const char *errstr;
1061 199a4027 2019-02-02 stsp struct got_reflist_head refs;
1062 1b3893a2 2019-03-18 stsp
1063 1b3893a2 2019-03-18 stsp SIMPLEQ_INIT(&refs);
1064 5c860e29 2018-03-12 stsp
1065 6715a751 2018-03-16 stsp #ifndef PROFILE
1066 6098196c 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1067 6098196c 2019-01-04 stsp NULL)
1068 ad242220 2018-09-08 stsp == -1)
1069 f42b1b34 2018-03-12 stsp err(1, "pledge");
1070 6715a751 2018-03-16 stsp #endif
1071 79109fed 2018-03-27 stsp
1072 499d7ecc 2019-05-22 stsp while ((ch = getopt(argc, argv, "bpc:C:l:r:")) != -1) {
1073 79109fed 2018-03-27 stsp switch (ch) {
1074 499d7ecc 2019-05-22 stsp case 'b':
1075 499d7ecc 2019-05-22 stsp first_parent_traversal = 1;
1076 499d7ecc 2019-05-22 stsp break;
1077 79109fed 2018-03-27 stsp case 'p':
1078 79109fed 2018-03-27 stsp show_patch = 1;
1079 d142fc45 2018-04-01 stsp break;
1080 d142fc45 2018-04-01 stsp case 'c':
1081 d142fc45 2018-04-01 stsp start_commit = optarg;
1082 64a96a6d 2018-04-01 stsp break;
1083 c0cc5c62 2018-10-18 stsp case 'C':
1084 4a8520aa 2018-10-18 stsp diff_context = strtonum(optarg, 0, GOT_DIFF_MAX_CONTEXT,
1085 4a8520aa 2018-10-18 stsp &errstr);
1086 c0cc5c62 2018-10-18 stsp if (errstr != NULL)
1087 c0cc5c62 2018-10-18 stsp err(1, "-C option %s", errstr);
1088 c0cc5c62 2018-10-18 stsp break;
1089 64a96a6d 2018-04-01 stsp case 'l':
1090 64a96a6d 2018-04-01 stsp limit = strtonum(optarg, 1, INT_MAX, &errstr);
1091 64a96a6d 2018-04-01 stsp if (errstr != NULL)
1092 64a96a6d 2018-04-01 stsp err(1, "-l option %s", errstr);
1093 a0603db2 2018-06-10 stsp break;
1094 04ca23f4 2018-07-16 stsp case 'r':
1095 04ca23f4 2018-07-16 stsp repo_path = realpath(optarg, NULL);
1096 04ca23f4 2018-07-16 stsp if (repo_path == NULL)
1097 04ca23f4 2018-07-16 stsp err(1, "-r option");
1098 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
1099 04ca23f4 2018-07-16 stsp break;
1100 79109fed 2018-03-27 stsp default:
1101 2deda0b9 2019-03-07 stsp usage_log();
1102 79109fed 2018-03-27 stsp /* NOTREACHED */
1103 79109fed 2018-03-27 stsp }
1104 79109fed 2018-03-27 stsp }
1105 79109fed 2018-03-27 stsp
1106 79109fed 2018-03-27 stsp argc -= optind;
1107 79109fed 2018-03-27 stsp argv += optind;
1108 f42b1b34 2018-03-12 stsp
1109 04ca23f4 2018-07-16 stsp cwd = getcwd(NULL, 0);
1110 04ca23f4 2018-07-16 stsp if (cwd == NULL) {
1111 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
1112 04ca23f4 2018-07-16 stsp goto done;
1113 04ca23f4 2018-07-16 stsp }
1114 cffc0aa4 2019-02-05 stsp
1115 cffc0aa4 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
1116 cffc0aa4 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1117 cffc0aa4 2019-02-05 stsp goto done;
1118 cffc0aa4 2019-02-05 stsp error = NULL;
1119 cffc0aa4 2019-02-05 stsp
1120 e7301579 2019-03-18 stsp if (argc == 0) {
1121 cbd1af7a 2019-03-18 stsp path = strdup("");
1122 e7301579 2019-03-18 stsp if (path == NULL) {
1123 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1124 cbd1af7a 2019-03-18 stsp goto done;
1125 e7301579 2019-03-18 stsp }
1126 e7301579 2019-03-18 stsp } else if (argc == 1) {
1127 e7301579 2019-03-18 stsp if (worktree) {
1128 e7301579 2019-03-18 stsp error = got_worktree_resolve_path(&path, worktree,
1129 e7301579 2019-03-18 stsp argv[0]);
1130 e7301579 2019-03-18 stsp if (error)
1131 e7301579 2019-03-18 stsp goto done;
1132 e7301579 2019-03-18 stsp } else {
1133 e7301579 2019-03-18 stsp path = strdup(argv[0]);
1134 e7301579 2019-03-18 stsp if (path == NULL) {
1135 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1136 e7301579 2019-03-18 stsp goto done;
1137 e7301579 2019-03-18 stsp }
1138 e7301579 2019-03-18 stsp }
1139 cbd1af7a 2019-03-18 stsp } else
1140 cbd1af7a 2019-03-18 stsp usage_log();
1141 cbd1af7a 2019-03-18 stsp
1142 5486daa2 2019-05-11 stsp if (repo_path == NULL) {
1143 5486daa2 2019-05-11 stsp repo_path = worktree ?
1144 5486daa2 2019-05-11 stsp strdup(got_worktree_get_repo_path(worktree)) : strdup(cwd);
1145 5486daa2 2019-05-11 stsp }
1146 04ca23f4 2018-07-16 stsp if (repo_path == NULL) {
1147 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1148 cffc0aa4 2019-02-05 stsp goto done;
1149 04ca23f4 2018-07-16 stsp }
1150 6098196c 2019-01-04 stsp
1151 f42b1b34 2018-03-12 stsp error = got_repo_open(&repo, repo_path);
1152 d7d4f210 2018-03-12 stsp if (error != NULL)
1153 04ca23f4 2018-07-16 stsp goto done;
1154 f42b1b34 2018-03-12 stsp
1155 c02c541e 2019-03-29 stsp error = apply_unveil(got_repo_get_path(repo), 1,
1156 314a6357 2019-05-13 stsp worktree ? got_worktree_get_root_path(worktree) : NULL, 0);
1157 c02c541e 2019-03-29 stsp if (error)
1158 c02c541e 2019-03-29 stsp goto done;
1159 c02c541e 2019-03-29 stsp
1160 d142fc45 2018-04-01 stsp if (start_commit == NULL) {
1161 3235492e 2018-04-01 stsp struct got_reference *head_ref;
1162 1cc14b9f 2019-05-14 stsp error = got_ref_open(&head_ref, repo,
1163 1cc14b9f 2019-05-14 stsp worktree ? got_worktree_get_head_ref_name(worktree)
1164 1cc14b9f 2019-05-14 stsp : GOT_REF_HEAD, 0);
1165 3235492e 2018-04-01 stsp if (error != NULL)
1166 3235492e 2018-04-01 stsp return error;
1167 3235492e 2018-04-01 stsp error = got_ref_resolve(&id, repo, head_ref);
1168 3235492e 2018-04-01 stsp got_ref_close(head_ref);
1169 3235492e 2018-04-01 stsp if (error != NULL)
1170 3235492e 2018-04-01 stsp return error;
1171 15a94983 2018-12-23 stsp error = got_object_open_as_commit(&commit, repo, id);
1172 3235492e 2018-04-01 stsp } else {
1173 d1f2edc9 2018-06-13 stsp struct got_reference *ref;
1174 2f17228e 2019-05-12 stsp error = got_ref_open(&ref, repo, start_commit, 0);
1175 3235492e 2018-04-01 stsp if (error == NULL) {
1176 1caad61b 2019-02-01 stsp int obj_type;
1177 d1f2edc9 2018-06-13 stsp error = got_ref_resolve(&id, repo, ref);
1178 d1f2edc9 2018-06-13 stsp got_ref_close(ref);
1179 d1f2edc9 2018-06-13 stsp if (error != NULL)
1180 1caad61b 2019-02-01 stsp goto done;
1181 1caad61b 2019-02-01 stsp error = got_object_get_type(&obj_type, repo, id);
1182 1caad61b 2019-02-01 stsp if (error != NULL)
1183 1caad61b 2019-02-01 stsp goto done;
1184 1caad61b 2019-02-01 stsp if (obj_type == GOT_OBJ_TYPE_TAG) {
1185 1caad61b 2019-02-01 stsp struct got_tag_object *tag;
1186 1caad61b 2019-02-01 stsp error = got_object_open_as_tag(&tag, repo, id);
1187 1caad61b 2019-02-01 stsp if (error != NULL)
1188 1caad61b 2019-02-01 stsp goto done;
1189 1caad61b 2019-02-01 stsp if (got_object_tag_get_object_type(tag) !=
1190 1caad61b 2019-02-01 stsp GOT_OBJ_TYPE_COMMIT) {
1191 1caad61b 2019-02-01 stsp got_object_tag_close(tag);
1192 1caad61b 2019-02-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1193 1caad61b 2019-02-01 stsp goto done;
1194 1caad61b 2019-02-01 stsp }
1195 1caad61b 2019-02-01 stsp free(id);
1196 1caad61b 2019-02-01 stsp id = got_object_id_dup(
1197 1caad61b 2019-02-01 stsp got_object_tag_get_object_id(tag));
1198 1caad61b 2019-02-01 stsp if (id == NULL)
1199 638f9024 2019-05-13 stsp error = got_error_from_errno(
1200 230a42bd 2019-05-11 jcs "got_object_id_dup");
1201 1caad61b 2019-02-01 stsp got_object_tag_close(tag);
1202 1caad61b 2019-02-01 stsp if (error)
1203 1caad61b 2019-02-01 stsp goto done;
1204 1caad61b 2019-02-01 stsp } else if (obj_type != GOT_OBJ_TYPE_COMMIT) {
1205 1caad61b 2019-02-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1206 1caad61b 2019-02-01 stsp goto done;
1207 1caad61b 2019-02-01 stsp }
1208 15a94983 2018-12-23 stsp error = got_object_open_as_commit(&commit, repo, id);
1209 d1f2edc9 2018-06-13 stsp if (error != NULL)
1210 1caad61b 2019-02-01 stsp goto done;
1211 d1f2edc9 2018-06-13 stsp }
1212 15a94983 2018-12-23 stsp if (commit == NULL) {
1213 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&id, repo,
1214 d1f2edc9 2018-06-13 stsp start_commit);
1215 d1f2edc9 2018-06-13 stsp if (error != NULL)
1216 d1f2edc9 2018-06-13 stsp return error;
1217 3235492e 2018-04-01 stsp }
1218 3235492e 2018-04-01 stsp }
1219 d7d4f210 2018-03-12 stsp if (error != NULL)
1220 04ca23f4 2018-07-16 stsp goto done;
1221 04ca23f4 2018-07-16 stsp
1222 23721109 2018-10-22 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
1223 04ca23f4 2018-07-16 stsp if (error != NULL)
1224 04ca23f4 2018-07-16 stsp goto done;
1225 04ca23f4 2018-07-16 stsp if (in_repo_path) {
1226 04ca23f4 2018-07-16 stsp free(path);
1227 04ca23f4 2018-07-16 stsp path = in_repo_path;
1228 04ca23f4 2018-07-16 stsp }
1229 199a4027 2019-02-02 stsp
1230 199a4027 2019-02-02 stsp error = got_ref_list(&refs, repo);
1231 199a4027 2019-02-02 stsp if (error)
1232 199a4027 2019-02-02 stsp goto done;
1233 04ca23f4 2018-07-16 stsp
1234 15a94983 2018-12-23 stsp error = print_commits(id, repo, path, show_patch,
1235 199a4027 2019-02-02 stsp diff_context, limit, first_parent_traversal, &refs);
1236 04ca23f4 2018-07-16 stsp done:
1237 04ca23f4 2018-07-16 stsp free(path);
1238 04ca23f4 2018-07-16 stsp free(repo_path);
1239 04ca23f4 2018-07-16 stsp free(cwd);
1240 f42b1b34 2018-03-12 stsp free(id);
1241 cffc0aa4 2019-02-05 stsp if (worktree)
1242 cffc0aa4 2019-02-05 stsp got_worktree_close(worktree);
1243 ad242220 2018-09-08 stsp if (repo) {
1244 ad242220 2018-09-08 stsp const struct got_error *repo_error;
1245 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
1246 ad242220 2018-09-08 stsp if (error == NULL)
1247 ad242220 2018-09-08 stsp error = repo_error;
1248 ad242220 2018-09-08 stsp }
1249 e2e879a0 2019-03-11 stsp got_ref_list_free(&refs);
1250 8bf5b3c9 2018-03-17 stsp return error;
1251 5c860e29 2018-03-12 stsp }
1252 5c860e29 2018-03-12 stsp
1253 4ed7e80c 2018-05-20 stsp __dead static void
1254 3f8b7d6a 2018-04-01 stsp usage_diff(void)
1255 3f8b7d6a 2018-04-01 stsp {
1256 b72f483a 2019-02-05 stsp fprintf(stderr, "usage: %s diff [-C number] [-r repository-path] "
1257 927df6b7 2019-02-10 stsp "[object1 object2 | path]\n", getprogname());
1258 3f8b7d6a 2018-04-01 stsp exit(1);
1259 b00d56cd 2018-04-01 stsp }
1260 b00d56cd 2018-04-01 stsp
1261 b72f483a 2019-02-05 stsp struct print_diff_arg {
1262 b72f483a 2019-02-05 stsp struct got_repository *repo;
1263 b72f483a 2019-02-05 stsp struct got_worktree *worktree;
1264 b72f483a 2019-02-05 stsp int diff_context;
1265 3fc0c068 2019-02-10 stsp const char *id_str;
1266 3fc0c068 2019-02-10 stsp int header_shown;
1267 b72f483a 2019-02-05 stsp };
1268 b72f483a 2019-02-05 stsp
1269 4ed7e80c 2018-05-20 stsp static const struct got_error *
1270 b72f483a 2019-02-05 stsp print_diff(void *arg, unsigned char status, const char *path,
1271 016a88dd 2019-05-13 stsp struct got_object_id *blob_id, struct got_object_id *commit_id)
1272 b72f483a 2019-02-05 stsp {
1273 b72f483a 2019-02-05 stsp struct print_diff_arg *a = arg;
1274 b72f483a 2019-02-05 stsp const struct got_error *err = NULL;
1275 b72f483a 2019-02-05 stsp struct got_blob_object *blob1 = NULL;
1276 b72f483a 2019-02-05 stsp FILE *f2 = NULL;
1277 b72f483a 2019-02-05 stsp char *abspath = NULL;
1278 b72f483a 2019-02-05 stsp struct stat sb;
1279 b72f483a 2019-02-05 stsp
1280 2ec1f75b 2019-03-26 stsp if (status != GOT_STATUS_MODIFY && status != GOT_STATUS_ADD &&
1281 7154f6ce 2019-03-27 stsp status != GOT_STATUS_DELETE && status != GOT_STATUS_CONFLICT)
1282 b72f483a 2019-02-05 stsp return NULL;
1283 3fc0c068 2019-02-10 stsp
1284 3fc0c068 2019-02-10 stsp if (!a->header_shown) {
1285 3fc0c068 2019-02-10 stsp printf("diff %s %s\n", a->id_str,
1286 3fc0c068 2019-02-10 stsp got_worktree_get_root_path(a->worktree));
1287 3fc0c068 2019-02-10 stsp a->header_shown = 1;
1288 3fc0c068 2019-02-10 stsp }
1289 b72f483a 2019-02-05 stsp
1290 7154f6ce 2019-03-27 stsp if (status != GOT_STATUS_ADD) {
1291 016a88dd 2019-05-13 stsp err = got_object_open_as_blob(&blob1, a->repo, blob_id, 8192);
1292 d00136be 2019-03-26 stsp if (err)
1293 d00136be 2019-03-26 stsp goto done;
1294 b72f483a 2019-02-05 stsp
1295 d00136be 2019-03-26 stsp }
1296 d00136be 2019-03-26 stsp
1297 7154f6ce 2019-03-27 stsp if (status != GOT_STATUS_DELETE) {
1298 2ec1f75b 2019-03-26 stsp if (asprintf(&abspath, "%s/%s",
1299 2ec1f75b 2019-03-26 stsp got_worktree_get_root_path(a->worktree), path) == -1) {
1300 638f9024 2019-05-13 stsp err = got_error_from_errno("asprintf");
1301 2ec1f75b 2019-03-26 stsp goto done;
1302 2ec1f75b 2019-03-26 stsp }
1303 b72f483a 2019-02-05 stsp
1304 2ec1f75b 2019-03-26 stsp f2 = fopen(abspath, "r");
1305 2ec1f75b 2019-03-26 stsp if (f2 == NULL) {
1306 638f9024 2019-05-13 stsp err = got_error_from_errno2("fopen", abspath);
1307 2ec1f75b 2019-03-26 stsp goto done;
1308 2ec1f75b 2019-03-26 stsp }
1309 2ec1f75b 2019-03-26 stsp if (lstat(abspath, &sb) == -1) {
1310 638f9024 2019-05-13 stsp err = got_error_from_errno2("lstat", abspath);
1311 2ec1f75b 2019-03-26 stsp goto done;
1312 2ec1f75b 2019-03-26 stsp }
1313 2ec1f75b 2019-03-26 stsp } else
1314 2ec1f75b 2019-03-26 stsp sb.st_size = 0;
1315 b72f483a 2019-02-05 stsp
1316 b72f483a 2019-02-05 stsp err = got_diff_blob_file(blob1, f2, sb.st_size, path, a->diff_context,
1317 b72f483a 2019-02-05 stsp stdout);
1318 b72f483a 2019-02-05 stsp done:
1319 b72f483a 2019-02-05 stsp if (blob1)
1320 b72f483a 2019-02-05 stsp got_object_blob_close(blob1);
1321 fb43ecf1 2019-02-11 stsp if (f2 && fclose(f2) != 0 && err == NULL)
1322 638f9024 2019-05-13 stsp err = got_error_from_errno("fclose");
1323 b72f483a 2019-02-05 stsp free(abspath);
1324 927df6b7 2019-02-10 stsp return err;
1325 927df6b7 2019-02-10 stsp }
1326 927df6b7 2019-02-10 stsp
1327 927df6b7 2019-02-10 stsp static const struct got_error *
1328 b00d56cd 2018-04-01 stsp cmd_diff(int argc, char *argv[])
1329 b00d56cd 2018-04-01 stsp {
1330 b00d56cd 2018-04-01 stsp const struct got_error *error;
1331 b00d56cd 2018-04-01 stsp struct got_repository *repo = NULL;
1332 b72f483a 2019-02-05 stsp struct got_worktree *worktree = NULL;
1333 b72f483a 2019-02-05 stsp char *cwd = NULL, *repo_path = NULL;
1334 15a94983 2018-12-23 stsp struct got_object_id *id1 = NULL, *id2 = NULL;
1335 cd628e99 2019-05-28 stsp const char *id_str1 = NULL, *id_str2 = NULL;
1336 5e70831e 2019-05-28 stsp char *label1 = NULL, *label2 = NULL;
1337 15a94983 2018-12-23 stsp int type1, type2;
1338 c0cc5c62 2018-10-18 stsp int diff_context = 3, ch;
1339 c0cc5c62 2018-10-18 stsp const char *errstr;
1340 927df6b7 2019-02-10 stsp char *path = NULL;
1341 b00d56cd 2018-04-01 stsp
1342 b00d56cd 2018-04-01 stsp #ifndef PROFILE
1343 25eccc22 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1344 25eccc22 2019-01-04 stsp NULL) == -1)
1345 b00d56cd 2018-04-01 stsp err(1, "pledge");
1346 b00d56cd 2018-04-01 stsp #endif
1347 b00d56cd 2018-04-01 stsp
1348 b72f483a 2019-02-05 stsp while ((ch = getopt(argc, argv, "C:r:")) != -1) {
1349 b00d56cd 2018-04-01 stsp switch (ch) {
1350 c0cc5c62 2018-10-18 stsp case 'C':
1351 c0cc5c62 2018-10-18 stsp diff_context = strtonum(optarg, 1, INT_MAX, &errstr);
1352 c0cc5c62 2018-10-18 stsp if (errstr != NULL)
1353 c0cc5c62 2018-10-18 stsp err(1, "-C option %s", errstr);
1354 c0cc5c62 2018-10-18 stsp break;
1355 b72f483a 2019-02-05 stsp case 'r':
1356 b72f483a 2019-02-05 stsp repo_path = realpath(optarg, NULL);
1357 b72f483a 2019-02-05 stsp if (repo_path == NULL)
1358 b72f483a 2019-02-05 stsp err(1, "-r option");
1359 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
1360 b72f483a 2019-02-05 stsp break;
1361 b00d56cd 2018-04-01 stsp default:
1362 2deda0b9 2019-03-07 stsp usage_diff();
1363 b00d56cd 2018-04-01 stsp /* NOTREACHED */
1364 b00d56cd 2018-04-01 stsp }
1365 b00d56cd 2018-04-01 stsp }
1366 b00d56cd 2018-04-01 stsp
1367 b00d56cd 2018-04-01 stsp argc -= optind;
1368 b00d56cd 2018-04-01 stsp argv += optind;
1369 b00d56cd 2018-04-01 stsp
1370 b72f483a 2019-02-05 stsp cwd = getcwd(NULL, 0);
1371 b72f483a 2019-02-05 stsp if (cwd == NULL) {
1372 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
1373 b72f483a 2019-02-05 stsp goto done;
1374 b72f483a 2019-02-05 stsp }
1375 927df6b7 2019-02-10 stsp error = got_worktree_open(&worktree, cwd);
1376 927df6b7 2019-02-10 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1377 927df6b7 2019-02-10 stsp goto done;
1378 927df6b7 2019-02-10 stsp if (argc <= 1) {
1379 927df6b7 2019-02-10 stsp if (worktree == NULL) {
1380 927df6b7 2019-02-10 stsp error = got_error(GOT_ERR_NOT_WORKTREE);
1381 927df6b7 2019-02-10 stsp goto done;
1382 927df6b7 2019-02-10 stsp }
1383 b72f483a 2019-02-05 stsp if (repo_path)
1384 b72f483a 2019-02-05 stsp errx(1,
1385 b72f483a 2019-02-05 stsp "-r option can't be used when diffing a work tree");
1386 b72f483a 2019-02-05 stsp repo_path = strdup(got_worktree_get_repo_path(worktree));
1387 927df6b7 2019-02-10 stsp if (repo_path == NULL) {
1388 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1389 927df6b7 2019-02-10 stsp goto done;
1390 927df6b7 2019-02-10 stsp }
1391 927df6b7 2019-02-10 stsp if (argc == 1) {
1392 6c7ab921 2019-03-18 stsp error = got_worktree_resolve_path(&path, worktree,
1393 cbd1af7a 2019-03-18 stsp argv[0]);
1394 927df6b7 2019-02-10 stsp if (error)
1395 927df6b7 2019-02-10 stsp goto done;
1396 927df6b7 2019-02-10 stsp } else {
1397 927df6b7 2019-02-10 stsp path = strdup("");
1398 927df6b7 2019-02-10 stsp if (path == NULL) {
1399 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1400 927df6b7 2019-02-10 stsp goto done;
1401 927df6b7 2019-02-10 stsp }
1402 927df6b7 2019-02-10 stsp }
1403 b72f483a 2019-02-05 stsp } else if (argc == 2) {
1404 15a94983 2018-12-23 stsp id_str1 = argv[0];
1405 15a94983 2018-12-23 stsp id_str2 = argv[1];
1406 30db809c 2019-06-05 stsp if (worktree && repo_path == NULL) {
1407 30db809c 2019-06-05 stsp repo_path =
1408 30db809c 2019-06-05 stsp strdup(got_worktree_get_repo_path(worktree));
1409 30db809c 2019-06-05 stsp if (repo_path == NULL) {
1410 30db809c 2019-06-05 stsp error = got_error_from_errno("strdup");
1411 30db809c 2019-06-05 stsp goto done;
1412 30db809c 2019-06-05 stsp }
1413 30db809c 2019-06-05 stsp }
1414 b00d56cd 2018-04-01 stsp } else
1415 b00d56cd 2018-04-01 stsp usage_diff();
1416 25eccc22 2019-01-04 stsp
1417 b72f483a 2019-02-05 stsp if (repo_path == NULL) {
1418 b72f483a 2019-02-05 stsp repo_path = getcwd(NULL, 0);
1419 b72f483a 2019-02-05 stsp if (repo_path == NULL)
1420 638f9024 2019-05-13 stsp return got_error_from_errno("getcwd");
1421 b72f483a 2019-02-05 stsp }
1422 b00d56cd 2018-04-01 stsp
1423 b00d56cd 2018-04-01 stsp error = got_repo_open(&repo, repo_path);
1424 76089277 2018-04-01 stsp free(repo_path);
1425 b00d56cd 2018-04-01 stsp if (error != NULL)
1426 b00d56cd 2018-04-01 stsp goto done;
1427 b00d56cd 2018-04-01 stsp
1428 c02c541e 2019-03-29 stsp error = apply_unveil(got_repo_get_path(repo), 1,
1429 314a6357 2019-05-13 stsp worktree ? got_worktree_get_root_path(worktree) : NULL, 0);
1430 c02c541e 2019-03-29 stsp if (error)
1431 c02c541e 2019-03-29 stsp goto done;
1432 c02c541e 2019-03-29 stsp
1433 30db809c 2019-06-05 stsp if (argc <= 1) {
1434 b72f483a 2019-02-05 stsp struct print_diff_arg arg;
1435 b72f483a 2019-02-05 stsp char *id_str;
1436 b72f483a 2019-02-05 stsp error = got_object_id_str(&id_str,
1437 b72f483a 2019-02-05 stsp got_worktree_get_base_commit_id(worktree));
1438 b72f483a 2019-02-05 stsp if (error)
1439 b72f483a 2019-02-05 stsp goto done;
1440 b72f483a 2019-02-05 stsp arg.repo = repo;
1441 b72f483a 2019-02-05 stsp arg.worktree = worktree;
1442 b72f483a 2019-02-05 stsp arg.diff_context = diff_context;
1443 3fc0c068 2019-02-10 stsp arg.id_str = id_str;
1444 3fc0c068 2019-02-10 stsp arg.header_shown = 0;
1445 b72f483a 2019-02-05 stsp
1446 927df6b7 2019-02-10 stsp error = got_worktree_status(worktree, path, repo, print_diff,
1447 b72f483a 2019-02-05 stsp &arg, check_cancelled, NULL);
1448 b72f483a 2019-02-05 stsp free(id_str);
1449 b72f483a 2019-02-05 stsp goto done;
1450 b72f483a 2019-02-05 stsp }
1451 b72f483a 2019-02-05 stsp
1452 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&id1, repo, id_str1);
1453 e02e74af 2019-05-28 stsp if (error) {
1454 e02e74af 2019-05-28 stsp struct got_reference *ref;
1455 e02e74af 2019-05-28 stsp if (error->code != GOT_ERR_BAD_OBJ_ID_STR)
1456 e02e74af 2019-05-28 stsp goto done;
1457 e02e74af 2019-05-28 stsp error = got_ref_open(&ref, repo, id_str1, 0);
1458 e02e74af 2019-05-28 stsp if (error != NULL)
1459 e02e74af 2019-05-28 stsp goto done;
1460 5e70831e 2019-05-28 stsp label1 = strdup(got_ref_get_name(ref));
1461 5e70831e 2019-05-28 stsp if (label1 == NULL) {
1462 5e70831e 2019-05-28 stsp error = got_error_from_errno("strdup");
1463 5e70831e 2019-05-28 stsp goto done;
1464 5e70831e 2019-05-28 stsp }
1465 e02e74af 2019-05-28 stsp error = got_ref_resolve(&id1, repo, ref);
1466 e02e74af 2019-05-28 stsp got_ref_close(ref);
1467 e02e74af 2019-05-28 stsp if (error != NULL)
1468 5e70831e 2019-05-28 stsp goto done;
1469 5e70831e 2019-05-28 stsp } else {
1470 5e70831e 2019-05-28 stsp label1 = strdup(id_str1);
1471 5e70831e 2019-05-28 stsp if (label1 == NULL) {
1472 5e70831e 2019-05-28 stsp error = got_error_from_errno("strdup");
1473 e02e74af 2019-05-28 stsp goto done;
1474 5e70831e 2019-05-28 stsp }
1475 e02e74af 2019-05-28 stsp }
1476 b00d56cd 2018-04-01 stsp
1477 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&id2, repo, id_str2);
1478 e02e74af 2019-05-28 stsp if (error) {
1479 e02e74af 2019-05-28 stsp struct got_reference *ref;
1480 e02e74af 2019-05-28 stsp if (error->code != GOT_ERR_BAD_OBJ_ID_STR)
1481 e02e74af 2019-05-28 stsp goto done;
1482 e02e74af 2019-05-28 stsp error = got_ref_open(&ref, repo, id_str2, 0);
1483 e02e74af 2019-05-28 stsp if (error != NULL)
1484 5e70831e 2019-05-28 stsp goto done;
1485 5e70831e 2019-05-28 stsp label2 = strdup(got_ref_get_name(ref));
1486 5e70831e 2019-05-28 stsp if (label2 == NULL) {
1487 5e70831e 2019-05-28 stsp error = got_error_from_errno("strdup");
1488 e02e74af 2019-05-28 stsp goto done;
1489 5e70831e 2019-05-28 stsp }
1490 e02e74af 2019-05-28 stsp error = got_ref_resolve(&id2, repo, ref);
1491 e02e74af 2019-05-28 stsp got_ref_close(ref);
1492 e02e74af 2019-05-28 stsp if (error != NULL)
1493 e02e74af 2019-05-28 stsp goto done;
1494 5e70831e 2019-05-28 stsp } else {
1495 5e70831e 2019-05-28 stsp label2 = strdup(id_str2);
1496 5e70831e 2019-05-28 stsp if (label2 == NULL) {
1497 5e70831e 2019-05-28 stsp error = got_error_from_errno("strdup");
1498 5e70831e 2019-05-28 stsp goto done;
1499 5e70831e 2019-05-28 stsp }
1500 e02e74af 2019-05-28 stsp }
1501 b00d56cd 2018-04-01 stsp
1502 15a94983 2018-12-23 stsp error = got_object_get_type(&type1, repo, id1);
1503 15a94983 2018-12-23 stsp if (error)
1504 15a94983 2018-12-23 stsp goto done;
1505 15a94983 2018-12-23 stsp
1506 15a94983 2018-12-23 stsp error = got_object_get_type(&type2, repo, id2);
1507 15a94983 2018-12-23 stsp if (error)
1508 15a94983 2018-12-23 stsp goto done;
1509 15a94983 2018-12-23 stsp
1510 15a94983 2018-12-23 stsp if (type1 != type2) {
1511 b00d56cd 2018-04-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1512 b00d56cd 2018-04-01 stsp goto done;
1513 b00d56cd 2018-04-01 stsp }
1514 b00d56cd 2018-04-01 stsp
1515 15a94983 2018-12-23 stsp switch (type1) {
1516 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_BLOB:
1517 15a94983 2018-12-23 stsp error = got_diff_objects_as_blobs(id1, id2, NULL, NULL,
1518 54156555 2018-12-24 stsp diff_context, repo, stdout);
1519 b00d56cd 2018-04-01 stsp break;
1520 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_TREE:
1521 15a94983 2018-12-23 stsp error = got_diff_objects_as_trees(id1, id2, "", "",
1522 54156555 2018-12-24 stsp diff_context, repo, stdout);
1523 b00d56cd 2018-04-01 stsp break;
1524 b00d56cd 2018-04-01 stsp case GOT_OBJ_TYPE_COMMIT:
1525 5e70831e 2019-05-28 stsp printf("diff %s %s\n", label1, label2);
1526 15a94983 2018-12-23 stsp error = got_diff_objects_as_commits(id1, id2, diff_context,
1527 c0cc5c62 2018-10-18 stsp repo, stdout);
1528 b00d56cd 2018-04-01 stsp break;
1529 b00d56cd 2018-04-01 stsp default:
1530 b00d56cd 2018-04-01 stsp error = got_error(GOT_ERR_OBJ_TYPE);
1531 b00d56cd 2018-04-01 stsp }
1532 b00d56cd 2018-04-01 stsp
1533 b00d56cd 2018-04-01 stsp done:
1534 5e70831e 2019-05-28 stsp free(label1);
1535 5e70831e 2019-05-28 stsp free(label2);
1536 15a94983 2018-12-23 stsp free(id1);
1537 15a94983 2018-12-23 stsp free(id2);
1538 927df6b7 2019-02-10 stsp free(path);
1539 b72f483a 2019-02-05 stsp if (worktree)
1540 b72f483a 2019-02-05 stsp got_worktree_close(worktree);
1541 ad242220 2018-09-08 stsp if (repo) {
1542 ad242220 2018-09-08 stsp const struct got_error *repo_error;
1543 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
1544 ad242220 2018-09-08 stsp if (error == NULL)
1545 ad242220 2018-09-08 stsp error = repo_error;
1546 ad242220 2018-09-08 stsp }
1547 b00d56cd 2018-04-01 stsp return error;
1548 404c43c4 2018-06-21 stsp }
1549 404c43c4 2018-06-21 stsp
1550 404c43c4 2018-06-21 stsp __dead static void
1551 404c43c4 2018-06-21 stsp usage_blame(void)
1552 404c43c4 2018-06-21 stsp {
1553 9270e621 2019-02-05 stsp fprintf(stderr,
1554 9270e621 2019-02-05 stsp "usage: %s blame [-c commit] [-r repository-path] path\n",
1555 404c43c4 2018-06-21 stsp getprogname());
1556 404c43c4 2018-06-21 stsp exit(1);
1557 b00d56cd 2018-04-01 stsp }
1558 b00d56cd 2018-04-01 stsp
1559 404c43c4 2018-06-21 stsp static const struct got_error *
1560 404c43c4 2018-06-21 stsp cmd_blame(int argc, char *argv[])
1561 404c43c4 2018-06-21 stsp {
1562 404c43c4 2018-06-21 stsp const struct got_error *error;
1563 404c43c4 2018-06-21 stsp struct got_repository *repo = NULL;
1564 0c06baac 2019-02-05 stsp struct got_worktree *worktree = NULL;
1565 66bea077 2018-08-02 stsp char *path, *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
1566 404c43c4 2018-06-21 stsp struct got_object_id *commit_id = NULL;
1567 404c43c4 2018-06-21 stsp char *commit_id_str = NULL;
1568 404c43c4 2018-06-21 stsp int ch;
1569 404c43c4 2018-06-21 stsp
1570 404c43c4 2018-06-21 stsp #ifndef PROFILE
1571 36e2fb66 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1572 36e2fb66 2019-01-04 stsp NULL) == -1)
1573 404c43c4 2018-06-21 stsp err(1, "pledge");
1574 404c43c4 2018-06-21 stsp #endif
1575 404c43c4 2018-06-21 stsp
1576 66bea077 2018-08-02 stsp while ((ch = getopt(argc, argv, "c:r:")) != -1) {
1577 404c43c4 2018-06-21 stsp switch (ch) {
1578 404c43c4 2018-06-21 stsp case 'c':
1579 404c43c4 2018-06-21 stsp commit_id_str = optarg;
1580 404c43c4 2018-06-21 stsp break;
1581 66bea077 2018-08-02 stsp case 'r':
1582 66bea077 2018-08-02 stsp repo_path = realpath(optarg, NULL);
1583 66bea077 2018-08-02 stsp if (repo_path == NULL)
1584 66bea077 2018-08-02 stsp err(1, "-r option");
1585 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
1586 66bea077 2018-08-02 stsp break;
1587 404c43c4 2018-06-21 stsp default:
1588 2deda0b9 2019-03-07 stsp usage_blame();
1589 404c43c4 2018-06-21 stsp /* NOTREACHED */
1590 404c43c4 2018-06-21 stsp }
1591 404c43c4 2018-06-21 stsp }
1592 404c43c4 2018-06-21 stsp
1593 404c43c4 2018-06-21 stsp argc -= optind;
1594 404c43c4 2018-06-21 stsp argv += optind;
1595 404c43c4 2018-06-21 stsp
1596 a39318fd 2018-08-02 stsp if (argc == 1)
1597 404c43c4 2018-06-21 stsp path = argv[0];
1598 a39318fd 2018-08-02 stsp else
1599 404c43c4 2018-06-21 stsp usage_blame();
1600 404c43c4 2018-06-21 stsp
1601 66bea077 2018-08-02 stsp cwd = getcwd(NULL, 0);
1602 66bea077 2018-08-02 stsp if (cwd == NULL) {
1603 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
1604 66bea077 2018-08-02 stsp goto done;
1605 66bea077 2018-08-02 stsp }
1606 66bea077 2018-08-02 stsp if (repo_path == NULL) {
1607 0c06baac 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
1608 0c06baac 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1609 66bea077 2018-08-02 stsp goto done;
1610 0c06baac 2019-02-05 stsp else
1611 0c06baac 2019-02-05 stsp error = NULL;
1612 0c06baac 2019-02-05 stsp if (worktree) {
1613 0c06baac 2019-02-05 stsp repo_path =
1614 0c06baac 2019-02-05 stsp strdup(got_worktree_get_repo_path(worktree));
1615 0c06baac 2019-02-05 stsp if (repo_path == NULL)
1616 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1617 0c06baac 2019-02-05 stsp if (error)
1618 0c06baac 2019-02-05 stsp goto done;
1619 0c06baac 2019-02-05 stsp } else {
1620 0c06baac 2019-02-05 stsp repo_path = strdup(cwd);
1621 0c06baac 2019-02-05 stsp if (repo_path == NULL) {
1622 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1623 0c06baac 2019-02-05 stsp goto done;
1624 0c06baac 2019-02-05 stsp }
1625 66bea077 2018-08-02 stsp }
1626 66bea077 2018-08-02 stsp }
1627 36e2fb66 2019-01-04 stsp
1628 c02c541e 2019-03-29 stsp error = got_repo_open(&repo, repo_path);
1629 c02c541e 2019-03-29 stsp if (error != NULL)
1630 36e2fb66 2019-01-04 stsp goto done;
1631 66bea077 2018-08-02 stsp
1632 314a6357 2019-05-13 stsp error = apply_unveil(got_repo_get_path(repo), 1, NULL, 0);
1633 c02c541e 2019-03-29 stsp if (error)
1634 404c43c4 2018-06-21 stsp goto done;
1635 404c43c4 2018-06-21 stsp
1636 0c06baac 2019-02-05 stsp if (worktree) {
1637 6efaaa2d 2019-02-05 stsp const char *prefix = got_worktree_get_path_prefix(worktree);
1638 0c06baac 2019-02-05 stsp char *p, *worktree_subdir = cwd +
1639 0c06baac 2019-02-05 stsp strlen(got_worktree_get_root_path(worktree));
1640 6efaaa2d 2019-02-05 stsp if (asprintf(&p, "%s%s%s%s%s",
1641 6efaaa2d 2019-02-05 stsp prefix, (strcmp(prefix, "/") != 0) ? "/" : "",
1642 6efaaa2d 2019-02-05 stsp worktree_subdir, worktree_subdir[0] ? "/" : "",
1643 6efaaa2d 2019-02-05 stsp path) == -1) {
1644 638f9024 2019-05-13 stsp error = got_error_from_errno("asprintf");
1645 0c06baac 2019-02-05 stsp goto done;
1646 0c06baac 2019-02-05 stsp }
1647 6efaaa2d 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, p, 0);
1648 0c06baac 2019-02-05 stsp free(p);
1649 0c06baac 2019-02-05 stsp } else {
1650 0c06baac 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
1651 0c06baac 2019-02-05 stsp }
1652 0c06baac 2019-02-05 stsp if (error)
1653 66bea077 2018-08-02 stsp goto done;
1654 66bea077 2018-08-02 stsp
1655 404c43c4 2018-06-21 stsp if (commit_id_str == NULL) {
1656 404c43c4 2018-06-21 stsp struct got_reference *head_ref;
1657 2f17228e 2019-05-12 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD, 0);
1658 404c43c4 2018-06-21 stsp if (error != NULL)
1659 66bea077 2018-08-02 stsp goto done;
1660 404c43c4 2018-06-21 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
1661 404c43c4 2018-06-21 stsp got_ref_close(head_ref);
1662 404c43c4 2018-06-21 stsp if (error != NULL)
1663 66bea077 2018-08-02 stsp goto done;
1664 404c43c4 2018-06-21 stsp } else {
1665 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&commit_id, repo,
1666 15a94983 2018-12-23 stsp commit_id_str);
1667 404c43c4 2018-06-21 stsp if (error != NULL)
1668 66bea077 2018-08-02 stsp goto done;
1669 404c43c4 2018-06-21 stsp }
1670 404c43c4 2018-06-21 stsp
1671 66bea077 2018-08-02 stsp error = got_blame(in_repo_path, commit_id, repo, stdout);
1672 404c43c4 2018-06-21 stsp done:
1673 66bea077 2018-08-02 stsp free(in_repo_path);
1674 66bea077 2018-08-02 stsp free(repo_path);
1675 66bea077 2018-08-02 stsp free(cwd);
1676 404c43c4 2018-06-21 stsp free(commit_id);
1677 0c06baac 2019-02-05 stsp if (worktree)
1678 0c06baac 2019-02-05 stsp got_worktree_close(worktree);
1679 ad242220 2018-09-08 stsp if (repo) {
1680 ad242220 2018-09-08 stsp const struct got_error *repo_error;
1681 ad242220 2018-09-08 stsp repo_error = got_repo_close(repo);
1682 ad242220 2018-09-08 stsp if (error == NULL)
1683 ad242220 2018-09-08 stsp error = repo_error;
1684 ad242220 2018-09-08 stsp }
1685 404c43c4 2018-06-21 stsp return error;
1686 5de5890b 2018-10-18 stsp }
1687 5de5890b 2018-10-18 stsp
1688 5de5890b 2018-10-18 stsp __dead static void
1689 5de5890b 2018-10-18 stsp usage_tree(void)
1690 5de5890b 2018-10-18 stsp {
1691 c1669e2e 2019-01-09 stsp fprintf(stderr,
1692 c1669e2e 2019-01-09 stsp "usage: %s tree [-c commit] [-r repository-path] [-iR] path\n",
1693 5de5890b 2018-10-18 stsp getprogname());
1694 5de5890b 2018-10-18 stsp exit(1);
1695 5de5890b 2018-10-18 stsp }
1696 5de5890b 2018-10-18 stsp
1697 c1669e2e 2019-01-09 stsp static void
1698 c1669e2e 2019-01-09 stsp print_entry(struct got_tree_entry *te, const char *id, const char *path,
1699 c1669e2e 2019-01-09 stsp const char *root_path)
1700 c1669e2e 2019-01-09 stsp {
1701 c1669e2e 2019-01-09 stsp int is_root_path = (strcmp(path, root_path) == 0);
1702 5de5890b 2018-10-18 stsp
1703 c1669e2e 2019-01-09 stsp path += strlen(root_path);
1704 c1669e2e 2019-01-09 stsp while (path[0] == '/')
1705 c1669e2e 2019-01-09 stsp path++;
1706 c1669e2e 2019-01-09 stsp
1707 c1669e2e 2019-01-09 stsp printf("%s%s%s%s%s\n", id ? id : "", path,
1708 d6e648b4 2019-02-10 stsp is_root_path ? "" : "/", te->name,
1709 d6e648b4 2019-02-10 stsp S_ISDIR(te->mode) ? "/" : ((te->mode & S_IXUSR) ? "*" : ""));
1710 c1669e2e 2019-01-09 stsp }
1711 c1669e2e 2019-01-09 stsp
1712 5de5890b 2018-10-18 stsp static const struct got_error *
1713 5de5890b 2018-10-18 stsp print_tree(const char *path, struct got_object_id *commit_id,
1714 c1669e2e 2019-01-09 stsp int show_ids, int recurse, const char *root_path,
1715 c1669e2e 2019-01-09 stsp struct got_repository *repo)
1716 5de5890b 2018-10-18 stsp {
1717 5de5890b 2018-10-18 stsp const struct got_error *err = NULL;
1718 5de5890b 2018-10-18 stsp struct got_object_id *tree_id = NULL;
1719 5de5890b 2018-10-18 stsp struct got_tree_object *tree = NULL;
1720 5de5890b 2018-10-18 stsp const struct got_tree_entries *entries;
1721 5de5890b 2018-10-18 stsp struct got_tree_entry *te;
1722 5de5890b 2018-10-18 stsp
1723 5de5890b 2018-10-18 stsp err = got_object_id_by_path(&tree_id, repo, commit_id, path);
1724 5de5890b 2018-10-18 stsp if (err)
1725 5de5890b 2018-10-18 stsp goto done;
1726 5de5890b 2018-10-18 stsp
1727 5de5890b 2018-10-18 stsp err = got_object_open_as_tree(&tree, repo, tree_id);
1728 5de5890b 2018-10-18 stsp if (err)
1729 5de5890b 2018-10-18 stsp goto done;
1730 5de5890b 2018-10-18 stsp entries = got_object_tree_get_entries(tree);
1731 5de5890b 2018-10-18 stsp te = SIMPLEQ_FIRST(&entries->head);
1732 5de5890b 2018-10-18 stsp while (te) {
1733 5de5890b 2018-10-18 stsp char *id = NULL;
1734 84453469 2018-11-11 stsp
1735 84453469 2018-11-11 stsp if (sigint_received || sigpipe_received)
1736 84453469 2018-11-11 stsp break;
1737 84453469 2018-11-11 stsp
1738 5de5890b 2018-10-18 stsp if (show_ids) {
1739 5de5890b 2018-10-18 stsp char *id_str;
1740 5de5890b 2018-10-18 stsp err = got_object_id_str(&id_str, te->id);
1741 5de5890b 2018-10-18 stsp if (err)
1742 5de5890b 2018-10-18 stsp goto done;
1743 5de5890b 2018-10-18 stsp if (asprintf(&id, "%s ", id_str) == -1) {
1744 638f9024 2019-05-13 stsp err = got_error_from_errno("asprintf");
1745 5de5890b 2018-10-18 stsp free(id_str);
1746 5de5890b 2018-10-18 stsp goto done;
1747 5de5890b 2018-10-18 stsp }
1748 5de5890b 2018-10-18 stsp free(id_str);
1749 5de5890b 2018-10-18 stsp }
1750 c1669e2e 2019-01-09 stsp print_entry(te, id, path, root_path);
1751 5de5890b 2018-10-18 stsp free(id);
1752 c1669e2e 2019-01-09 stsp
1753 c1669e2e 2019-01-09 stsp if (recurse && S_ISDIR(te->mode)) {
1754 c1669e2e 2019-01-09 stsp char *child_path;
1755 c1669e2e 2019-01-09 stsp if (asprintf(&child_path, "%s%s%s", path,
1756 c1669e2e 2019-01-09 stsp path[0] == '/' && path[1] == '\0' ? "" : "/",
1757 c1669e2e 2019-01-09 stsp te->name) == -1) {
1758 638f9024 2019-05-13 stsp err = got_error_from_errno("asprintf");
1759 c1669e2e 2019-01-09 stsp goto done;
1760 c1669e2e 2019-01-09 stsp }
1761 c1669e2e 2019-01-09 stsp err = print_tree(child_path, commit_id, show_ids, 1,
1762 c1669e2e 2019-01-09 stsp root_path, repo);
1763 c1669e2e 2019-01-09 stsp free(child_path);
1764 c1669e2e 2019-01-09 stsp if (err)
1765 c1669e2e 2019-01-09 stsp goto done;
1766 c1669e2e 2019-01-09 stsp }
1767 c1669e2e 2019-01-09 stsp
1768 c1669e2e 2019-01-09 stsp te = SIMPLEQ_NEXT(te, entry);
1769 5de5890b 2018-10-18 stsp }
1770 5de5890b 2018-10-18 stsp done:
1771 5de5890b 2018-10-18 stsp if (tree)
1772 5de5890b 2018-10-18 stsp got_object_tree_close(tree);
1773 5de5890b 2018-10-18 stsp free(tree_id);
1774 5de5890b 2018-10-18 stsp return err;
1775 404c43c4 2018-06-21 stsp }
1776 404c43c4 2018-06-21 stsp
1777 5de5890b 2018-10-18 stsp static const struct got_error *
1778 5de5890b 2018-10-18 stsp cmd_tree(int argc, char *argv[])
1779 5de5890b 2018-10-18 stsp {
1780 5de5890b 2018-10-18 stsp const struct got_error *error;
1781 5de5890b 2018-10-18 stsp struct got_repository *repo = NULL;
1782 7a2c19d6 2019-02-05 stsp struct got_worktree *worktree = NULL;
1783 7a2c19d6 2019-02-05 stsp const char *path;
1784 7a2c19d6 2019-02-05 stsp char *cwd = NULL, *repo_path = NULL, *in_repo_path = NULL;
1785 5de5890b 2018-10-18 stsp struct got_object_id *commit_id = NULL;
1786 5de5890b 2018-10-18 stsp char *commit_id_str = NULL;
1787 c1669e2e 2019-01-09 stsp int show_ids = 0, recurse = 0;
1788 5de5890b 2018-10-18 stsp int ch;
1789 5de5890b 2018-10-18 stsp
1790 5de5890b 2018-10-18 stsp #ifndef PROFILE
1791 0f8d269b 2019-01-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1792 0f8d269b 2019-01-04 stsp NULL) == -1)
1793 5de5890b 2018-10-18 stsp err(1, "pledge");
1794 5de5890b 2018-10-18 stsp #endif
1795 5de5890b 2018-10-18 stsp
1796 c1669e2e 2019-01-09 stsp while ((ch = getopt(argc, argv, "c:r:iR")) != -1) {
1797 5de5890b 2018-10-18 stsp switch (ch) {
1798 5de5890b 2018-10-18 stsp case 'c':
1799 5de5890b 2018-10-18 stsp commit_id_str = optarg;
1800 5de5890b 2018-10-18 stsp break;
1801 5de5890b 2018-10-18 stsp case 'r':
1802 5de5890b 2018-10-18 stsp repo_path = realpath(optarg, NULL);
1803 5de5890b 2018-10-18 stsp if (repo_path == NULL)
1804 5de5890b 2018-10-18 stsp err(1, "-r option");
1805 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
1806 5de5890b 2018-10-18 stsp break;
1807 5de5890b 2018-10-18 stsp case 'i':
1808 5de5890b 2018-10-18 stsp show_ids = 1;
1809 5de5890b 2018-10-18 stsp break;
1810 c1669e2e 2019-01-09 stsp case 'R':
1811 c1669e2e 2019-01-09 stsp recurse = 1;
1812 c1669e2e 2019-01-09 stsp break;
1813 5de5890b 2018-10-18 stsp default:
1814 2deda0b9 2019-03-07 stsp usage_tree();
1815 5de5890b 2018-10-18 stsp /* NOTREACHED */
1816 5de5890b 2018-10-18 stsp }
1817 5de5890b 2018-10-18 stsp }
1818 5de5890b 2018-10-18 stsp
1819 5de5890b 2018-10-18 stsp argc -= optind;
1820 5de5890b 2018-10-18 stsp argv += optind;
1821 5de5890b 2018-10-18 stsp
1822 5de5890b 2018-10-18 stsp if (argc == 1)
1823 5de5890b 2018-10-18 stsp path = argv[0];
1824 5de5890b 2018-10-18 stsp else if (argc > 1)
1825 5de5890b 2018-10-18 stsp usage_tree();
1826 5de5890b 2018-10-18 stsp else
1827 7a2c19d6 2019-02-05 stsp path = NULL;
1828 7a2c19d6 2019-02-05 stsp
1829 9bf7a39b 2019-02-05 stsp cwd = getcwd(NULL, 0);
1830 9bf7a39b 2019-02-05 stsp if (cwd == NULL) {
1831 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
1832 9bf7a39b 2019-02-05 stsp goto done;
1833 9bf7a39b 2019-02-05 stsp }
1834 5de5890b 2018-10-18 stsp if (repo_path == NULL) {
1835 7a2c19d6 2019-02-05 stsp error = got_worktree_open(&worktree, cwd);
1836 8994de28 2019-02-05 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
1837 7a2c19d6 2019-02-05 stsp goto done;
1838 7a2c19d6 2019-02-05 stsp else
1839 7a2c19d6 2019-02-05 stsp error = NULL;
1840 7a2c19d6 2019-02-05 stsp if (worktree) {
1841 7a2c19d6 2019-02-05 stsp repo_path =
1842 7a2c19d6 2019-02-05 stsp strdup(got_worktree_get_repo_path(worktree));
1843 7a2c19d6 2019-02-05 stsp if (repo_path == NULL)
1844 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1845 7a2c19d6 2019-02-05 stsp if (error)
1846 7a2c19d6 2019-02-05 stsp goto done;
1847 7a2c19d6 2019-02-05 stsp } else {
1848 7a2c19d6 2019-02-05 stsp repo_path = strdup(cwd);
1849 7a2c19d6 2019-02-05 stsp if (repo_path == NULL) {
1850 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1851 7a2c19d6 2019-02-05 stsp goto done;
1852 7a2c19d6 2019-02-05 stsp }
1853 5de5890b 2018-10-18 stsp }
1854 5de5890b 2018-10-18 stsp }
1855 5de5890b 2018-10-18 stsp
1856 5de5890b 2018-10-18 stsp error = got_repo_open(&repo, repo_path);
1857 5de5890b 2018-10-18 stsp if (error != NULL)
1858 c02c541e 2019-03-29 stsp goto done;
1859 c02c541e 2019-03-29 stsp
1860 314a6357 2019-05-13 stsp error = apply_unveil(got_repo_get_path(repo), 1, NULL, 0);
1861 c02c541e 2019-03-29 stsp if (error)
1862 5de5890b 2018-10-18 stsp goto done;
1863 5de5890b 2018-10-18 stsp
1864 9bf7a39b 2019-02-05 stsp if (path == NULL) {
1865 9bf7a39b 2019-02-05 stsp if (worktree) {
1866 9bf7a39b 2019-02-05 stsp char *p, *worktree_subdir = cwd +
1867 9bf7a39b 2019-02-05 stsp strlen(got_worktree_get_root_path(worktree));
1868 9bf7a39b 2019-02-05 stsp if (asprintf(&p, "%s/%s",
1869 9bf7a39b 2019-02-05 stsp got_worktree_get_path_prefix(worktree),
1870 9bf7a39b 2019-02-05 stsp worktree_subdir) == -1) {
1871 638f9024 2019-05-13 stsp error = got_error_from_errno("asprintf");
1872 9bf7a39b 2019-02-05 stsp goto done;
1873 9bf7a39b 2019-02-05 stsp }
1874 9bf7a39b 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, p, 1);
1875 9bf7a39b 2019-02-05 stsp free(p);
1876 9bf7a39b 2019-02-05 stsp if (error)
1877 9bf7a39b 2019-02-05 stsp goto done;
1878 9bf7a39b 2019-02-05 stsp } else
1879 9bf7a39b 2019-02-05 stsp path = "/";
1880 9bf7a39b 2019-02-05 stsp }
1881 9bf7a39b 2019-02-05 stsp if (in_repo_path == NULL) {
1882 9bf7a39b 2019-02-05 stsp error = got_repo_map_path(&in_repo_path, repo, path, 1);
1883 9bf7a39b 2019-02-05 stsp if (error != NULL)
1884 9bf7a39b 2019-02-05 stsp goto done;
1885 9bf7a39b 2019-02-05 stsp }
1886 5de5890b 2018-10-18 stsp
1887 5de5890b 2018-10-18 stsp if (commit_id_str == NULL) {
1888 5de5890b 2018-10-18 stsp struct got_reference *head_ref;
1889 2f17228e 2019-05-12 stsp error = got_ref_open(&head_ref, repo, GOT_REF_HEAD, 0);
1890 5de5890b 2018-10-18 stsp if (error != NULL)
1891 5de5890b 2018-10-18 stsp goto done;
1892 5de5890b 2018-10-18 stsp error = got_ref_resolve(&commit_id, repo, head_ref);
1893 5de5890b 2018-10-18 stsp got_ref_close(head_ref);
1894 5de5890b 2018-10-18 stsp if (error != NULL)
1895 5de5890b 2018-10-18 stsp goto done;
1896 5de5890b 2018-10-18 stsp } else {
1897 15a94983 2018-12-23 stsp error = got_object_resolve_id_str(&commit_id, repo,
1898 15a94983 2018-12-23 stsp commit_id_str);
1899 5de5890b 2018-10-18 stsp if (error != NULL)
1900 5de5890b 2018-10-18 stsp goto done;
1901 5de5890b 2018-10-18 stsp }
1902 5de5890b 2018-10-18 stsp
1903 c1669e2e 2019-01-09 stsp error = print_tree(in_repo_path, commit_id, show_ids, recurse,
1904 c1669e2e 2019-01-09 stsp in_repo_path, repo);
1905 5de5890b 2018-10-18 stsp done:
1906 5de5890b 2018-10-18 stsp free(in_repo_path);
1907 5de5890b 2018-10-18 stsp free(repo_path);
1908 5de5890b 2018-10-18 stsp free(cwd);
1909 5de5890b 2018-10-18 stsp free(commit_id);
1910 7a2c19d6 2019-02-05 stsp if (worktree)
1911 7a2c19d6 2019-02-05 stsp got_worktree_close(worktree);
1912 5de5890b 2018-10-18 stsp if (repo) {
1913 5de5890b 2018-10-18 stsp const struct got_error *repo_error;
1914 5de5890b 2018-10-18 stsp repo_error = got_repo_close(repo);
1915 5de5890b 2018-10-18 stsp if (error == NULL)
1916 5de5890b 2018-10-18 stsp error = repo_error;
1917 5de5890b 2018-10-18 stsp }
1918 5de5890b 2018-10-18 stsp return error;
1919 5de5890b 2018-10-18 stsp }
1920 5de5890b 2018-10-18 stsp
1921 6bad629b 2019-02-04 stsp __dead static void
1922 6bad629b 2019-02-04 stsp usage_status(void)
1923 6bad629b 2019-02-04 stsp {
1924 927df6b7 2019-02-10 stsp fprintf(stderr, "usage: %s status [path]\n", getprogname());
1925 6bad629b 2019-02-04 stsp exit(1);
1926 6bad629b 2019-02-04 stsp }
1927 5c860e29 2018-03-12 stsp
1928 b72f483a 2019-02-05 stsp static const struct got_error *
1929 b72f483a 2019-02-05 stsp print_status(void *arg, unsigned char status, const char *path,
1930 016a88dd 2019-05-13 stsp struct got_object_id *blob_id, struct got_object_id *commit_id)
1931 6bad629b 2019-02-04 stsp {
1932 6bad629b 2019-02-04 stsp printf("%c %s\n", status, path);
1933 b72f483a 2019-02-05 stsp return NULL;
1934 6bad629b 2019-02-04 stsp }
1935 5c860e29 2018-03-12 stsp
1936 6bad629b 2019-02-04 stsp static const struct got_error *
1937 6bad629b 2019-02-04 stsp cmd_status(int argc, char *argv[])
1938 6bad629b 2019-02-04 stsp {
1939 6bad629b 2019-02-04 stsp const struct got_error *error = NULL;
1940 6bad629b 2019-02-04 stsp struct got_repository *repo = NULL;
1941 6bad629b 2019-02-04 stsp struct got_worktree *worktree = NULL;
1942 927df6b7 2019-02-10 stsp char *cwd = NULL, *path = NULL;
1943 6bad629b 2019-02-04 stsp int ch;
1944 5c860e29 2018-03-12 stsp
1945 6bad629b 2019-02-04 stsp while ((ch = getopt(argc, argv, "")) != -1) {
1946 6bad629b 2019-02-04 stsp switch (ch) {
1947 5c860e29 2018-03-12 stsp default:
1948 2deda0b9 2019-03-07 stsp usage_status();
1949 6bad629b 2019-02-04 stsp /* NOTREACHED */
1950 5c860e29 2018-03-12 stsp }
1951 5c860e29 2018-03-12 stsp }
1952 5c860e29 2018-03-12 stsp
1953 6bad629b 2019-02-04 stsp argc -= optind;
1954 6bad629b 2019-02-04 stsp argv += optind;
1955 5c860e29 2018-03-12 stsp
1956 6bad629b 2019-02-04 stsp #ifndef PROFILE
1957 6bad629b 2019-02-04 stsp if (pledge("stdio rpath wpath cpath flock proc exec sendfd unveil",
1958 6bad629b 2019-02-04 stsp NULL) == -1)
1959 6bad629b 2019-02-04 stsp err(1, "pledge");
1960 f42b1b34 2018-03-12 stsp #endif
1961 927df6b7 2019-02-10 stsp cwd = getcwd(NULL, 0);
1962 927df6b7 2019-02-10 stsp if (cwd == NULL) {
1963 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
1964 927df6b7 2019-02-10 stsp goto done;
1965 927df6b7 2019-02-10 stsp }
1966 927df6b7 2019-02-10 stsp
1967 927df6b7 2019-02-10 stsp error = got_worktree_open(&worktree, cwd);
1968 927df6b7 2019-02-10 stsp if (error != NULL)
1969 927df6b7 2019-02-10 stsp goto done;
1970 927df6b7 2019-02-10 stsp
1971 6bad629b 2019-02-04 stsp if (argc == 0) {
1972 927df6b7 2019-02-10 stsp path = strdup("");
1973 927df6b7 2019-02-10 stsp if (path == NULL) {
1974 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
1975 6bad629b 2019-02-04 stsp goto done;
1976 6bad629b 2019-02-04 stsp }
1977 6bad629b 2019-02-04 stsp } else if (argc == 1) {
1978 6c7ab921 2019-03-18 stsp error = got_worktree_resolve_path(&path, worktree, argv[0]);
1979 927df6b7 2019-02-10 stsp if (error)
1980 6bad629b 2019-02-04 stsp goto done;
1981 6bad629b 2019-02-04 stsp } else
1982 6bad629b 2019-02-04 stsp usage_status();
1983 6bad629b 2019-02-04 stsp
1984 6bad629b 2019-02-04 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
1985 6bad629b 2019-02-04 stsp if (error != NULL)
1986 6bad629b 2019-02-04 stsp goto done;
1987 6bad629b 2019-02-04 stsp
1988 d0eebce4 2019-03-11 stsp error = apply_unveil(got_repo_get_path(repo), 1,
1989 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
1990 6bad629b 2019-02-04 stsp if (error)
1991 6bad629b 2019-02-04 stsp goto done;
1992 6bad629b 2019-02-04 stsp
1993 927df6b7 2019-02-10 stsp error = got_worktree_status(worktree, path, repo, print_status, NULL,
1994 6bad629b 2019-02-04 stsp check_cancelled, NULL);
1995 6bad629b 2019-02-04 stsp done:
1996 927df6b7 2019-02-10 stsp free(cwd);
1997 927df6b7 2019-02-10 stsp free(path);
1998 d0eebce4 2019-03-11 stsp return error;
1999 d0eebce4 2019-03-11 stsp }
2000 d0eebce4 2019-03-11 stsp
2001 d0eebce4 2019-03-11 stsp __dead static void
2002 d0eebce4 2019-03-11 stsp usage_ref(void)
2003 d0eebce4 2019-03-11 stsp {
2004 d0eebce4 2019-03-11 stsp fprintf(stderr,
2005 d83d9d5c 2019-05-13 stsp "usage: %s ref [-r repository] -l | -d name | name target\n",
2006 d0eebce4 2019-03-11 stsp getprogname());
2007 d0eebce4 2019-03-11 stsp exit(1);
2008 d0eebce4 2019-03-11 stsp }
2009 d0eebce4 2019-03-11 stsp
2010 d0eebce4 2019-03-11 stsp static const struct got_error *
2011 d0eebce4 2019-03-11 stsp list_refs(struct got_repository *repo)
2012 d0eebce4 2019-03-11 stsp {
2013 d0eebce4 2019-03-11 stsp static const struct got_error *err = NULL;
2014 d0eebce4 2019-03-11 stsp struct got_reflist_head refs;
2015 d0eebce4 2019-03-11 stsp struct got_reflist_entry *re;
2016 d0eebce4 2019-03-11 stsp
2017 d0eebce4 2019-03-11 stsp SIMPLEQ_INIT(&refs);
2018 d0eebce4 2019-03-11 stsp err = got_ref_list(&refs, repo);
2019 d0eebce4 2019-03-11 stsp if (err)
2020 d0eebce4 2019-03-11 stsp return err;
2021 d0eebce4 2019-03-11 stsp
2022 d0eebce4 2019-03-11 stsp SIMPLEQ_FOREACH(re, &refs, entry) {
2023 d0eebce4 2019-03-11 stsp char *refstr;
2024 d0eebce4 2019-03-11 stsp refstr = got_ref_to_str(re->ref);
2025 d0eebce4 2019-03-11 stsp if (refstr == NULL)
2026 638f9024 2019-05-13 stsp return got_error_from_errno("got_ref_to_str");
2027 d0eebce4 2019-03-11 stsp printf("%s: %s\n", got_ref_get_name(re->ref), refstr);
2028 d0eebce4 2019-03-11 stsp free(refstr);
2029 d0eebce4 2019-03-11 stsp }
2030 d0eebce4 2019-03-11 stsp
2031 e2e879a0 2019-03-11 stsp got_ref_list_free(&refs);
2032 d0eebce4 2019-03-11 stsp return NULL;
2033 d0eebce4 2019-03-11 stsp }
2034 d0eebce4 2019-03-11 stsp
2035 d0eebce4 2019-03-11 stsp static const struct got_error *
2036 d0eebce4 2019-03-11 stsp delete_ref(struct got_repository *repo, const char *refname)
2037 d0eebce4 2019-03-11 stsp {
2038 d0eebce4 2019-03-11 stsp const struct got_error *err = NULL;
2039 d0eebce4 2019-03-11 stsp struct got_reference *ref;
2040 d0eebce4 2019-03-11 stsp
2041 2f17228e 2019-05-12 stsp err = got_ref_open(&ref, repo, refname, 0);
2042 d0eebce4 2019-03-11 stsp if (err)
2043 d0eebce4 2019-03-11 stsp return err;
2044 d0eebce4 2019-03-11 stsp
2045 d0eebce4 2019-03-11 stsp err = got_ref_delete(ref, repo);
2046 d0eebce4 2019-03-11 stsp got_ref_close(ref);
2047 d0eebce4 2019-03-11 stsp return err;
2048 d0eebce4 2019-03-11 stsp }
2049 d0eebce4 2019-03-11 stsp
2050 d0eebce4 2019-03-11 stsp static const struct got_error *
2051 d83d9d5c 2019-05-13 stsp add_ref(struct got_repository *repo, const char *refname, const char *target)
2052 d0eebce4 2019-03-11 stsp {
2053 d0eebce4 2019-03-11 stsp const struct got_error *err = NULL;
2054 d0eebce4 2019-03-11 stsp struct got_object_id *id;
2055 d0eebce4 2019-03-11 stsp struct got_reference *ref = NULL;
2056 d0eebce4 2019-03-11 stsp
2057 d83d9d5c 2019-05-13 stsp err = got_object_resolve_id_str(&id, repo, target);
2058 d83d9d5c 2019-05-13 stsp if (err) {
2059 d83d9d5c 2019-05-13 stsp struct got_reference *target_ref;
2060 d0eebce4 2019-03-11 stsp
2061 d83d9d5c 2019-05-13 stsp if (err->code != GOT_ERR_BAD_OBJ_ID_STR)
2062 d83d9d5c 2019-05-13 stsp return err;
2063 d83d9d5c 2019-05-13 stsp err = got_ref_open(&target_ref, repo, target, 0);
2064 d83d9d5c 2019-05-13 stsp if (err)
2065 d83d9d5c 2019-05-13 stsp return err;
2066 d83d9d5c 2019-05-13 stsp err = got_ref_resolve(&id, repo, target_ref);
2067 d83d9d5c 2019-05-13 stsp got_ref_close(target_ref);
2068 d83d9d5c 2019-05-13 stsp if (err)
2069 d83d9d5c 2019-05-13 stsp return err;
2070 d83d9d5c 2019-05-13 stsp }
2071 d83d9d5c 2019-05-13 stsp
2072 d0eebce4 2019-03-11 stsp err = got_ref_alloc(&ref, refname, id);
2073 d0eebce4 2019-03-11 stsp if (err)
2074 d0eebce4 2019-03-11 stsp goto done;
2075 d0eebce4 2019-03-11 stsp
2076 d0eebce4 2019-03-11 stsp err = got_ref_write(ref, repo);
2077 d0eebce4 2019-03-11 stsp done:
2078 d0eebce4 2019-03-11 stsp if (ref)
2079 d0eebce4 2019-03-11 stsp got_ref_close(ref);
2080 d0eebce4 2019-03-11 stsp free(id);
2081 d0eebce4 2019-03-11 stsp return err;
2082 d0eebce4 2019-03-11 stsp }
2083 d0eebce4 2019-03-11 stsp
2084 d0eebce4 2019-03-11 stsp static const struct got_error *
2085 d0eebce4 2019-03-11 stsp cmd_ref(int argc, char *argv[])
2086 d0eebce4 2019-03-11 stsp {
2087 d0eebce4 2019-03-11 stsp const struct got_error *error = NULL;
2088 d0eebce4 2019-03-11 stsp struct got_repository *repo = NULL;
2089 d0eebce4 2019-03-11 stsp struct got_worktree *worktree = NULL;
2090 d0eebce4 2019-03-11 stsp char *cwd = NULL, *repo_path = NULL;
2091 d0eebce4 2019-03-11 stsp int ch, do_list = 0;
2092 d0eebce4 2019-03-11 stsp const char *delref = NULL;
2093 d0eebce4 2019-03-11 stsp
2094 d0eebce4 2019-03-11 stsp /* TODO: Add -s option for adding symbolic references. */
2095 d0eebce4 2019-03-11 stsp while ((ch = getopt(argc, argv, "d:r:l")) != -1) {
2096 d0eebce4 2019-03-11 stsp switch (ch) {
2097 d0eebce4 2019-03-11 stsp case 'd':
2098 d0eebce4 2019-03-11 stsp delref = optarg;
2099 d0eebce4 2019-03-11 stsp break;
2100 d0eebce4 2019-03-11 stsp case 'r':
2101 d0eebce4 2019-03-11 stsp repo_path = realpath(optarg, NULL);
2102 d0eebce4 2019-03-11 stsp if (repo_path == NULL)
2103 d0eebce4 2019-03-11 stsp err(1, "-r option");
2104 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(repo_path);
2105 d0eebce4 2019-03-11 stsp break;
2106 d0eebce4 2019-03-11 stsp case 'l':
2107 d0eebce4 2019-03-11 stsp do_list = 1;
2108 d0eebce4 2019-03-11 stsp break;
2109 d0eebce4 2019-03-11 stsp default:
2110 d0eebce4 2019-03-11 stsp usage_ref();
2111 d0eebce4 2019-03-11 stsp /* NOTREACHED */
2112 d0eebce4 2019-03-11 stsp }
2113 d0eebce4 2019-03-11 stsp }
2114 d0eebce4 2019-03-11 stsp
2115 d0eebce4 2019-03-11 stsp if (do_list && delref)
2116 d0eebce4 2019-03-11 stsp errx(1, "-l and -d options are mutually exclusive\n");
2117 d0eebce4 2019-03-11 stsp
2118 d0eebce4 2019-03-11 stsp argc -= optind;
2119 d0eebce4 2019-03-11 stsp argv += optind;
2120 d0eebce4 2019-03-11 stsp
2121 d0eebce4 2019-03-11 stsp if (do_list || delref) {
2122 d0eebce4 2019-03-11 stsp if (argc > 0)
2123 d0eebce4 2019-03-11 stsp usage_ref();
2124 d0eebce4 2019-03-11 stsp } else if (argc != 2)
2125 d0eebce4 2019-03-11 stsp usage_ref();
2126 d0eebce4 2019-03-11 stsp
2127 d0eebce4 2019-03-11 stsp #ifndef PROFILE
2128 e0b57350 2019-03-12 stsp if (do_list) {
2129 e0b57350 2019-03-12 stsp if (pledge("stdio rpath wpath flock proc exec sendfd unveil",
2130 e0b57350 2019-03-12 stsp NULL) == -1)
2131 e0b57350 2019-03-12 stsp err(1, "pledge");
2132 e0b57350 2019-03-12 stsp } else {
2133 e0b57350 2019-03-12 stsp if (pledge("stdio rpath wpath cpath fattr flock proc exec "
2134 e0b57350 2019-03-12 stsp "sendfd unveil", NULL) == -1)
2135 e0b57350 2019-03-12 stsp err(1, "pledge");
2136 e0b57350 2019-03-12 stsp }
2137 d0eebce4 2019-03-11 stsp #endif
2138 d0eebce4 2019-03-11 stsp cwd = getcwd(NULL, 0);
2139 d0eebce4 2019-03-11 stsp if (cwd == NULL) {
2140 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
2141 d0eebce4 2019-03-11 stsp goto done;
2142 d0eebce4 2019-03-11 stsp }
2143 d0eebce4 2019-03-11 stsp
2144 d0eebce4 2019-03-11 stsp if (repo_path == NULL) {
2145 d0eebce4 2019-03-11 stsp error = got_worktree_open(&worktree, cwd);
2146 d0eebce4 2019-03-11 stsp if (error && error->code != GOT_ERR_NOT_WORKTREE)
2147 d0eebce4 2019-03-11 stsp goto done;
2148 d0eebce4 2019-03-11 stsp else
2149 d0eebce4 2019-03-11 stsp error = NULL;
2150 d0eebce4 2019-03-11 stsp if (worktree) {
2151 d0eebce4 2019-03-11 stsp repo_path =
2152 d0eebce4 2019-03-11 stsp strdup(got_worktree_get_repo_path(worktree));
2153 d0eebce4 2019-03-11 stsp if (repo_path == NULL)
2154 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
2155 d0eebce4 2019-03-11 stsp if (error)
2156 d0eebce4 2019-03-11 stsp goto done;
2157 d0eebce4 2019-03-11 stsp } else {
2158 d0eebce4 2019-03-11 stsp repo_path = strdup(cwd);
2159 d0eebce4 2019-03-11 stsp if (repo_path == NULL) {
2160 638f9024 2019-05-13 stsp error = got_error_from_errno("strdup");
2161 d0eebce4 2019-03-11 stsp goto done;
2162 d0eebce4 2019-03-11 stsp }
2163 d0eebce4 2019-03-11 stsp }
2164 d0eebce4 2019-03-11 stsp }
2165 d0eebce4 2019-03-11 stsp
2166 d0eebce4 2019-03-11 stsp error = got_repo_open(&repo, repo_path);
2167 d0eebce4 2019-03-11 stsp if (error != NULL)
2168 d0eebce4 2019-03-11 stsp goto done;
2169 d0eebce4 2019-03-11 stsp
2170 c02c541e 2019-03-29 stsp error = apply_unveil(got_repo_get_path(repo), do_list,
2171 314a6357 2019-05-13 stsp worktree ? got_worktree_get_root_path(worktree) : NULL, 0);
2172 c02c541e 2019-03-29 stsp if (error)
2173 c02c541e 2019-03-29 stsp goto done;
2174 c02c541e 2019-03-29 stsp
2175 d0eebce4 2019-03-11 stsp if (do_list)
2176 d0eebce4 2019-03-11 stsp error = list_refs(repo);
2177 d0eebce4 2019-03-11 stsp else if (delref)
2178 d0eebce4 2019-03-11 stsp error = delete_ref(repo, delref);
2179 d0eebce4 2019-03-11 stsp else
2180 d0eebce4 2019-03-11 stsp error = add_ref(repo, argv[0], argv[1]);
2181 d0eebce4 2019-03-11 stsp done:
2182 d0eebce4 2019-03-11 stsp if (repo)
2183 d0eebce4 2019-03-11 stsp got_repo_close(repo);
2184 d0eebce4 2019-03-11 stsp if (worktree)
2185 d0eebce4 2019-03-11 stsp got_worktree_close(worktree);
2186 d0eebce4 2019-03-11 stsp free(cwd);
2187 d0eebce4 2019-03-11 stsp free(repo_path);
2188 d00136be 2019-03-26 stsp return error;
2189 d00136be 2019-03-26 stsp }
2190 d00136be 2019-03-26 stsp
2191 d00136be 2019-03-26 stsp __dead static void
2192 d00136be 2019-03-26 stsp usage_add(void)
2193 d00136be 2019-03-26 stsp {
2194 fbb7e5c7 2019-05-11 stsp fprintf(stderr, "usage: %s add file-path ...\n", getprogname());
2195 d00136be 2019-03-26 stsp exit(1);
2196 d00136be 2019-03-26 stsp }
2197 d00136be 2019-03-26 stsp
2198 d00136be 2019-03-26 stsp static const struct got_error *
2199 d00136be 2019-03-26 stsp cmd_add(int argc, char *argv[])
2200 d00136be 2019-03-26 stsp {
2201 d00136be 2019-03-26 stsp const struct got_error *error = NULL;
2202 031a5338 2019-03-26 stsp struct got_repository *repo = NULL;
2203 d00136be 2019-03-26 stsp struct got_worktree *worktree = NULL;
2204 1dd54920 2019-05-11 stsp char *cwd = NULL;
2205 1dd54920 2019-05-11 stsp struct got_pathlist_head paths;
2206 1dd54920 2019-05-11 stsp struct got_pathlist_entry *pe;
2207 723c305c 2019-05-11 jcs int ch, x;
2208 1dd54920 2019-05-11 stsp
2209 1dd54920 2019-05-11 stsp TAILQ_INIT(&paths);
2210 d00136be 2019-03-26 stsp
2211 d00136be 2019-03-26 stsp while ((ch = getopt(argc, argv, "")) != -1) {
2212 d00136be 2019-03-26 stsp switch (ch) {
2213 d00136be 2019-03-26 stsp default:
2214 d00136be 2019-03-26 stsp usage_add();
2215 d00136be 2019-03-26 stsp /* NOTREACHED */
2216 d00136be 2019-03-26 stsp }
2217 d00136be 2019-03-26 stsp }
2218 d00136be 2019-03-26 stsp
2219 d00136be 2019-03-26 stsp argc -= optind;
2220 d00136be 2019-03-26 stsp argv += optind;
2221 d00136be 2019-03-26 stsp
2222 723c305c 2019-05-11 jcs if (argc < 1)
2223 d00136be 2019-03-26 stsp usage_add();
2224 d00136be 2019-03-26 stsp
2225 723c305c 2019-05-11 jcs /* make sure each file exists before doing anything halfway */
2226 723c305c 2019-05-11 jcs for (x = 0; x < argc; x++) {
2227 1dd54920 2019-05-11 stsp char *path = realpath(argv[x], NULL);
2228 723c305c 2019-05-11 jcs if (path == NULL) {
2229 638f9024 2019-05-13 stsp error = got_error_from_errno2("realpath", argv[x]);
2230 723c305c 2019-05-11 jcs goto done;
2231 723c305c 2019-05-11 jcs }
2232 1dd54920 2019-05-11 stsp free(path);
2233 d00136be 2019-03-26 stsp }
2234 d00136be 2019-03-26 stsp
2235 d00136be 2019-03-26 stsp cwd = getcwd(NULL, 0);
2236 d00136be 2019-03-26 stsp if (cwd == NULL) {
2237 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
2238 d00136be 2019-03-26 stsp goto done;
2239 d00136be 2019-03-26 stsp }
2240 723c305c 2019-05-11 jcs
2241 d00136be 2019-03-26 stsp error = got_worktree_open(&worktree, cwd);
2242 d00136be 2019-03-26 stsp if (error)
2243 d00136be 2019-03-26 stsp goto done;
2244 d00136be 2019-03-26 stsp
2245 031a5338 2019-03-26 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
2246 031a5338 2019-03-26 stsp if (error != NULL)
2247 031a5338 2019-03-26 stsp goto done;
2248 031a5338 2019-03-26 stsp
2249 031a5338 2019-03-26 stsp error = apply_unveil(got_repo_get_path(repo), 1,
2250 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
2251 d00136be 2019-03-26 stsp if (error)
2252 d00136be 2019-03-26 stsp goto done;
2253 d00136be 2019-03-26 stsp
2254 723c305c 2019-05-11 jcs for (x = 0; x < argc; x++) {
2255 1dd54920 2019-05-11 stsp char *path = realpath(argv[x], NULL);
2256 723c305c 2019-05-11 jcs if (path == NULL) {
2257 638f9024 2019-05-13 stsp error = got_error_from_errno2("realpath", argv[x]);
2258 723c305c 2019-05-11 jcs goto done;
2259 723c305c 2019-05-11 jcs }
2260 723c305c 2019-05-11 jcs
2261 723c305c 2019-05-11 jcs got_path_strip_trailing_slashes(path);
2262 1dd54920 2019-05-11 stsp error = got_pathlist_insert(&pe, &paths, path, NULL);
2263 1dd54920 2019-05-11 stsp if (error) {
2264 1dd54920 2019-05-11 stsp free(path);
2265 723c305c 2019-05-11 jcs goto done;
2266 1dd54920 2019-05-11 stsp }
2267 723c305c 2019-05-11 jcs }
2268 1dd54920 2019-05-11 stsp error = got_worktree_schedule_add(worktree, &paths, print_status,
2269 1dd54920 2019-05-11 stsp NULL, repo);
2270 d00136be 2019-03-26 stsp done:
2271 031a5338 2019-03-26 stsp if (repo)
2272 031a5338 2019-03-26 stsp got_repo_close(repo);
2273 d00136be 2019-03-26 stsp if (worktree)
2274 d00136be 2019-03-26 stsp got_worktree_close(worktree);
2275 1dd54920 2019-05-11 stsp TAILQ_FOREACH(pe, &paths, entry)
2276 1dd54920 2019-05-11 stsp free((char *)pe->path);
2277 1dd54920 2019-05-11 stsp got_pathlist_free(&paths);
2278 2ec1f75b 2019-03-26 stsp free(cwd);
2279 2ec1f75b 2019-03-26 stsp return error;
2280 2ec1f75b 2019-03-26 stsp }
2281 2ec1f75b 2019-03-26 stsp
2282 2ec1f75b 2019-03-26 stsp __dead static void
2283 2ec1f75b 2019-03-26 stsp usage_rm(void)
2284 2ec1f75b 2019-03-26 stsp {
2285 cc11e7e3 2019-06-04 stsp fprintf(stderr, "usage: %s rm [-f] file-path ...\n", getprogname());
2286 2ec1f75b 2019-03-26 stsp exit(1);
2287 2ec1f75b 2019-03-26 stsp }
2288 2ec1f75b 2019-03-26 stsp
2289 2ec1f75b 2019-03-26 stsp static const struct got_error *
2290 2ec1f75b 2019-03-26 stsp cmd_rm(int argc, char *argv[])
2291 2ec1f75b 2019-03-26 stsp {
2292 2ec1f75b 2019-03-26 stsp const struct got_error *error = NULL;
2293 2ec1f75b 2019-03-26 stsp struct got_worktree *worktree = NULL;
2294 2ec1f75b 2019-03-26 stsp struct got_repository *repo = NULL;
2295 17ed4618 2019-06-02 stsp char *cwd = NULL;
2296 17ed4618 2019-06-02 stsp struct got_pathlist_head paths;
2297 17ed4618 2019-06-02 stsp struct got_pathlist_entry *pe;
2298 17ed4618 2019-06-02 stsp int ch, i, delete_local_mods = 0;
2299 2ec1f75b 2019-03-26 stsp
2300 17ed4618 2019-06-02 stsp TAILQ_INIT(&paths);
2301 17ed4618 2019-06-02 stsp
2302 2ec1f75b 2019-03-26 stsp while ((ch = getopt(argc, argv, "f")) != -1) {
2303 2ec1f75b 2019-03-26 stsp switch (ch) {
2304 2ec1f75b 2019-03-26 stsp case 'f':
2305 2ec1f75b 2019-03-26 stsp delete_local_mods = 1;
2306 2ec1f75b 2019-03-26 stsp break;
2307 2ec1f75b 2019-03-26 stsp default:
2308 2ec1f75b 2019-03-26 stsp usage_add();
2309 2ec1f75b 2019-03-26 stsp /* NOTREACHED */
2310 2ec1f75b 2019-03-26 stsp }
2311 2ec1f75b 2019-03-26 stsp }
2312 2ec1f75b 2019-03-26 stsp
2313 2ec1f75b 2019-03-26 stsp argc -= optind;
2314 2ec1f75b 2019-03-26 stsp argv += optind;
2315 2ec1f75b 2019-03-26 stsp
2316 17ed4618 2019-06-02 stsp if (argc < 1)
2317 2ec1f75b 2019-03-26 stsp usage_rm();
2318 2ec1f75b 2019-03-26 stsp
2319 17ed4618 2019-06-02 stsp /* make sure each file exists before doing anything halfway */
2320 17ed4618 2019-06-02 stsp for (i = 0; i < argc; i++) {
2321 17ed4618 2019-06-02 stsp char *path = realpath(argv[i], NULL);
2322 17ed4618 2019-06-02 stsp if (path == NULL) {
2323 17ed4618 2019-06-02 stsp error = got_error_from_errno2("realpath", argv[i]);
2324 17ed4618 2019-06-02 stsp goto done;
2325 17ed4618 2019-06-02 stsp }
2326 17ed4618 2019-06-02 stsp free(path);
2327 2ec1f75b 2019-03-26 stsp }
2328 2ec1f75b 2019-03-26 stsp
2329 2ec1f75b 2019-03-26 stsp cwd = getcwd(NULL, 0);
2330 2ec1f75b 2019-03-26 stsp if (cwd == NULL) {
2331 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
2332 2ec1f75b 2019-03-26 stsp goto done;
2333 2ec1f75b 2019-03-26 stsp }
2334 2ec1f75b 2019-03-26 stsp error = got_worktree_open(&worktree, cwd);
2335 2ec1f75b 2019-03-26 stsp if (error)
2336 2ec1f75b 2019-03-26 stsp goto done;
2337 2ec1f75b 2019-03-26 stsp
2338 2ec1f75b 2019-03-26 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
2339 2af4a041 2019-05-11 jcs if (error)
2340 2ec1f75b 2019-03-26 stsp goto done;
2341 2ec1f75b 2019-03-26 stsp
2342 c2253644 2019-03-26 stsp error = apply_unveil(got_repo_get_path(repo), 1,
2343 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
2344 2ec1f75b 2019-03-26 stsp if (error)
2345 2ec1f75b 2019-03-26 stsp goto done;
2346 2ec1f75b 2019-03-26 stsp
2347 17ed4618 2019-06-02 stsp for (i = 0; i < argc; i++) {
2348 17ed4618 2019-06-02 stsp char *path = realpath(argv[i], NULL);
2349 17ed4618 2019-06-02 stsp if (path == NULL) {
2350 17ed4618 2019-06-02 stsp error = got_error_from_errno2("realpath", argv[i]);
2351 17ed4618 2019-06-02 stsp goto done;
2352 17ed4618 2019-06-02 stsp }
2353 17ed4618 2019-06-02 stsp
2354 17ed4618 2019-06-02 stsp got_path_strip_trailing_slashes(path);
2355 17ed4618 2019-06-02 stsp error = got_pathlist_insert(&pe, &paths, path, NULL);
2356 17ed4618 2019-06-02 stsp if (error) {
2357 17ed4618 2019-06-02 stsp free(path);
2358 17ed4618 2019-06-02 stsp goto done;
2359 17ed4618 2019-06-02 stsp }
2360 17ed4618 2019-06-02 stsp }
2361 17ed4618 2019-06-02 stsp error = got_worktree_schedule_delete(worktree, &paths,
2362 17ed4618 2019-06-02 stsp delete_local_mods, print_status, NULL, repo);
2363 a129376b 2019-03-28 stsp if (error)
2364 a129376b 2019-03-28 stsp goto done;
2365 a129376b 2019-03-28 stsp done:
2366 a129376b 2019-03-28 stsp if (repo)
2367 a129376b 2019-03-28 stsp got_repo_close(repo);
2368 a129376b 2019-03-28 stsp if (worktree)
2369 a129376b 2019-03-28 stsp got_worktree_close(worktree);
2370 17ed4618 2019-06-02 stsp TAILQ_FOREACH(pe, &paths, entry)
2371 17ed4618 2019-06-02 stsp free((char *)pe->path);
2372 17ed4618 2019-06-02 stsp got_pathlist_free(&paths);
2373 a129376b 2019-03-28 stsp free(cwd);
2374 a129376b 2019-03-28 stsp return error;
2375 a129376b 2019-03-28 stsp }
2376 a129376b 2019-03-28 stsp
2377 a129376b 2019-03-28 stsp __dead static void
2378 a129376b 2019-03-28 stsp usage_revert(void)
2379 a129376b 2019-03-28 stsp {
2380 e20a8b6f 2019-06-04 stsp fprintf(stderr, "usage: %s revert file-path ...\n", getprogname());
2381 a129376b 2019-03-28 stsp exit(1);
2382 a129376b 2019-03-28 stsp }
2383 a129376b 2019-03-28 stsp
2384 a129376b 2019-03-28 stsp static void
2385 a129376b 2019-03-28 stsp revert_progress(void *arg, unsigned char status, const char *path)
2386 a129376b 2019-03-28 stsp {
2387 a129376b 2019-03-28 stsp while (path[0] == '/')
2388 a129376b 2019-03-28 stsp path++;
2389 a129376b 2019-03-28 stsp printf("%c %s\n", status, path);
2390 a129376b 2019-03-28 stsp }
2391 a129376b 2019-03-28 stsp
2392 a129376b 2019-03-28 stsp static const struct got_error *
2393 a129376b 2019-03-28 stsp cmd_revert(int argc, char *argv[])
2394 a129376b 2019-03-28 stsp {
2395 a129376b 2019-03-28 stsp const struct got_error *error = NULL;
2396 a129376b 2019-03-28 stsp struct got_worktree *worktree = NULL;
2397 a129376b 2019-03-28 stsp struct got_repository *repo = NULL;
2398 a129376b 2019-03-28 stsp char *cwd = NULL, *path = NULL;
2399 e20a8b6f 2019-06-04 stsp struct got_pathlist_head paths;
2400 e20a8b6f 2019-06-04 stsp struct got_pathlist_entry *pe;
2401 e20a8b6f 2019-06-04 stsp int ch, i;
2402 a129376b 2019-03-28 stsp
2403 e20a8b6f 2019-06-04 stsp TAILQ_INIT(&paths);
2404 e20a8b6f 2019-06-04 stsp
2405 a129376b 2019-03-28 stsp while ((ch = getopt(argc, argv, "")) != -1) {
2406 a129376b 2019-03-28 stsp switch (ch) {
2407 a129376b 2019-03-28 stsp default:
2408 a129376b 2019-03-28 stsp usage_revert();
2409 a129376b 2019-03-28 stsp /* NOTREACHED */
2410 a129376b 2019-03-28 stsp }
2411 a129376b 2019-03-28 stsp }
2412 a129376b 2019-03-28 stsp
2413 a129376b 2019-03-28 stsp argc -= optind;
2414 a129376b 2019-03-28 stsp argv += optind;
2415 a129376b 2019-03-28 stsp
2416 e20a8b6f 2019-06-04 stsp if (argc < 1)
2417 a129376b 2019-03-28 stsp usage_revert();
2418 a129376b 2019-03-28 stsp
2419 e20a8b6f 2019-06-04 stsp for (i = 0; i < argc; i++) {
2420 e20a8b6f 2019-06-04 stsp char *path = realpath(argv[i], NULL);
2421 e20a8b6f 2019-06-04 stsp if (path == NULL) {
2422 e20a8b6f 2019-06-04 stsp error = got_error_from_errno2("realpath", argv[i]);
2423 e20a8b6f 2019-06-04 stsp goto done;
2424 e20a8b6f 2019-06-04 stsp }
2425 e20a8b6f 2019-06-04 stsp
2426 e20a8b6f 2019-06-04 stsp got_path_strip_trailing_slashes(path);
2427 e20a8b6f 2019-06-04 stsp error = got_pathlist_insert(&pe, &paths, path, NULL);
2428 e20a8b6f 2019-06-04 stsp if (error) {
2429 e20a8b6f 2019-06-04 stsp free(path);
2430 e20a8b6f 2019-06-04 stsp goto done;
2431 e20a8b6f 2019-06-04 stsp }
2432 a129376b 2019-03-28 stsp }
2433 a129376b 2019-03-28 stsp
2434 a129376b 2019-03-28 stsp cwd = getcwd(NULL, 0);
2435 a129376b 2019-03-28 stsp if (cwd == NULL) {
2436 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
2437 a129376b 2019-03-28 stsp goto done;
2438 a129376b 2019-03-28 stsp }
2439 a129376b 2019-03-28 stsp error = got_worktree_open(&worktree, cwd);
2440 2ec1f75b 2019-03-26 stsp if (error)
2441 2ec1f75b 2019-03-26 stsp goto done;
2442 a129376b 2019-03-28 stsp
2443 a129376b 2019-03-28 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
2444 a129376b 2019-03-28 stsp if (error != NULL)
2445 a129376b 2019-03-28 stsp goto done;
2446 a129376b 2019-03-28 stsp
2447 a129376b 2019-03-28 stsp error = apply_unveil(got_repo_get_path(repo), 1,
2448 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
2449 a129376b 2019-03-28 stsp if (error)
2450 a129376b 2019-03-28 stsp goto done;
2451 a129376b 2019-03-28 stsp
2452 e20a8b6f 2019-06-04 stsp error = got_worktree_revert(worktree, &paths,
2453 a129376b 2019-03-28 stsp revert_progress, NULL, repo);
2454 a129376b 2019-03-28 stsp if (error)
2455 a129376b 2019-03-28 stsp goto done;
2456 2ec1f75b 2019-03-26 stsp done:
2457 2ec1f75b 2019-03-26 stsp if (repo)
2458 2ec1f75b 2019-03-26 stsp got_repo_close(repo);
2459 2ec1f75b 2019-03-26 stsp if (worktree)
2460 2ec1f75b 2019-03-26 stsp got_worktree_close(worktree);
2461 2ec1f75b 2019-03-26 stsp free(path);
2462 d00136be 2019-03-26 stsp free(cwd);
2463 6bad629b 2019-02-04 stsp return error;
2464 c4296144 2019-05-09 stsp }
2465 c4296144 2019-05-09 stsp
2466 c4296144 2019-05-09 stsp __dead static void
2467 c4296144 2019-05-09 stsp usage_commit(void)
2468 c4296144 2019-05-09 stsp {
2469 c6fc0acd 2019-05-09 stsp fprintf(stderr, "usage: %s commit [-m msg] file-path\n", getprogname());
2470 c4296144 2019-05-09 stsp exit(1);
2471 33ad4cbe 2019-05-12 jcs }
2472 33ad4cbe 2019-05-12 jcs
2473 33ad4cbe 2019-05-12 jcs int
2474 e2ba3d07 2019-05-13 stsp spawn_editor(const char *editor, const char *file)
2475 33ad4cbe 2019-05-12 jcs {
2476 33ad4cbe 2019-05-12 jcs pid_t pid;
2477 33ad4cbe 2019-05-12 jcs sig_t sighup, sigint, sigquit;
2478 33ad4cbe 2019-05-12 jcs int st = -1;
2479 33ad4cbe 2019-05-12 jcs
2480 33ad4cbe 2019-05-12 jcs sighup = signal(SIGHUP, SIG_IGN);
2481 33ad4cbe 2019-05-12 jcs sigint = signal(SIGINT, SIG_IGN);
2482 33ad4cbe 2019-05-12 jcs sigquit = signal(SIGQUIT, SIG_IGN);
2483 33ad4cbe 2019-05-12 jcs
2484 33ad4cbe 2019-05-12 jcs switch (pid = fork()) {
2485 33ad4cbe 2019-05-12 jcs case -1:
2486 33ad4cbe 2019-05-12 jcs goto doneediting;
2487 33ad4cbe 2019-05-12 jcs case 0:
2488 e2ba3d07 2019-05-13 stsp execl(editor, editor, file, (char *)NULL);
2489 33ad4cbe 2019-05-12 jcs _exit(127);
2490 33ad4cbe 2019-05-12 jcs }
2491 33ad4cbe 2019-05-12 jcs
2492 33ad4cbe 2019-05-12 jcs while (waitpid(pid, &st, 0) == -1)
2493 33ad4cbe 2019-05-12 jcs if (errno != EINTR)
2494 33ad4cbe 2019-05-12 jcs break;
2495 33ad4cbe 2019-05-12 jcs
2496 33ad4cbe 2019-05-12 jcs doneediting:
2497 33ad4cbe 2019-05-12 jcs (void)signal(SIGHUP, sighup);
2498 33ad4cbe 2019-05-12 jcs (void)signal(SIGINT, sigint);
2499 33ad4cbe 2019-05-12 jcs (void)signal(SIGQUIT, sigquit);
2500 33ad4cbe 2019-05-12 jcs
2501 33ad4cbe 2019-05-12 jcs if (!WIFEXITED(st)) {
2502 33ad4cbe 2019-05-12 jcs errno = EINTR;
2503 33ad4cbe 2019-05-12 jcs return -1;
2504 33ad4cbe 2019-05-12 jcs }
2505 33ad4cbe 2019-05-12 jcs
2506 33ad4cbe 2019-05-12 jcs return WEXITSTATUS(st);
2507 33ad4cbe 2019-05-12 jcs }
2508 33ad4cbe 2019-05-12 jcs
2509 e2ba3d07 2019-05-13 stsp struct collect_commit_logmsg_arg {
2510 e2ba3d07 2019-05-13 stsp const char *cmdline_log;
2511 e2ba3d07 2019-05-13 stsp const char *editor;
2512 e0870e44 2019-05-13 stsp const char *worktree_path;
2513 76d98825 2019-06-03 stsp const char *branch_name;
2514 314a6357 2019-05-13 stsp const char *repo_path;
2515 e0870e44 2019-05-13 stsp char *logmsg_path;
2516 e2ba3d07 2019-05-13 stsp
2517 e2ba3d07 2019-05-13 stsp };
2518 e0870e44 2019-05-13 stsp
2519 33ad4cbe 2019-05-12 jcs static const struct got_error *
2520 33ad4cbe 2019-05-12 jcs collect_commit_logmsg(struct got_pathlist_head *commitable_paths, char **logmsg,
2521 33ad4cbe 2019-05-12 jcs void *arg)
2522 33ad4cbe 2019-05-12 jcs {
2523 76d98825 2019-06-03 stsp char *initial_content = NULL;
2524 33ad4cbe 2019-05-12 jcs struct got_pathlist_entry *pe;
2525 33ad4cbe 2019-05-12 jcs const struct got_error *err = NULL;
2526 e0870e44 2019-05-13 stsp char *template = NULL;
2527 e2ba3d07 2019-05-13 stsp struct collect_commit_logmsg_arg *a = arg;
2528 33ad4cbe 2019-05-12 jcs char buf[1024];
2529 33ad4cbe 2019-05-12 jcs struct stat st, st2;
2530 33ad4cbe 2019-05-12 jcs FILE *fp;
2531 33ad4cbe 2019-05-12 jcs size_t len;
2532 e0870e44 2019-05-13 stsp int fd, content_changed = 0;
2533 33ad4cbe 2019-05-12 jcs
2534 33ad4cbe 2019-05-12 jcs /* if a message was specified on the command line, just use it */
2535 e2ba3d07 2019-05-13 stsp if (a->cmdline_log != NULL && strlen(a->cmdline_log) != 0) {
2536 e2ba3d07 2019-05-13 stsp len = strlen(a->cmdline_log) + 1;
2537 33ad4cbe 2019-05-12 jcs *logmsg = malloc(len + 1);
2538 9f42ff69 2019-05-13 stsp if (*logmsg == NULL)
2539 638f9024 2019-05-13 stsp return got_error_from_errno("malloc");
2540 e2ba3d07 2019-05-13 stsp strlcpy(*logmsg, a->cmdline_log, len);
2541 33ad4cbe 2019-05-12 jcs return NULL;
2542 33ad4cbe 2019-05-12 jcs }
2543 33ad4cbe 2019-05-12 jcs
2544 e0870e44 2019-05-13 stsp if (asprintf(&template, "%s/logmsg", a->worktree_path) == -1)
2545 76d98825 2019-06-03 stsp return got_error_from_errno("asprintf");
2546 76d98825 2019-06-03 stsp
2547 76d98825 2019-06-03 stsp if (asprintf(&initial_content,
2548 76d98825 2019-06-03 stsp "\n# changes to be committed on branch %s:\n",
2549 76d98825 2019-06-03 stsp a->branch_name) == -1)
2550 638f9024 2019-05-13 stsp return got_error_from_errno("asprintf");
2551 e0870e44 2019-05-13 stsp
2552 e0870e44 2019-05-13 stsp err = got_opentemp_named_fd(&a->logmsg_path, &fd, template);
2553 793c30b5 2019-05-13 stsp if (err)
2554 e0870e44 2019-05-13 stsp goto done;
2555 33ad4cbe 2019-05-12 jcs
2556 e0870e44 2019-05-13 stsp dprintf(fd, initial_content);
2557 33ad4cbe 2019-05-12 jcs
2558 33ad4cbe 2019-05-12 jcs TAILQ_FOREACH(pe, commitable_paths, entry) {
2559 33ad4cbe 2019-05-12 jcs struct got_commitable *ct = pe->data;
2560 8656d6c4 2019-05-20 stsp dprintf(fd, "# %c %s\n",
2561 8656d6c4 2019-05-20 stsp got_commitable_get_status(ct),
2562 8656d6c4 2019-05-20 stsp got_commitable_get_path(ct));
2563 33ad4cbe 2019-05-12 jcs }
2564 33ad4cbe 2019-05-12 jcs close(fd);
2565 33ad4cbe 2019-05-12 jcs
2566 e0870e44 2019-05-13 stsp if (stat(a->logmsg_path, &st) == -1) {
2567 638f9024 2019-05-13 stsp err = got_error_from_errno2("stat", a->logmsg_path);
2568 33ad4cbe 2019-05-12 jcs goto done;
2569 33ad4cbe 2019-05-12 jcs }
2570 33ad4cbe 2019-05-12 jcs
2571 e0870e44 2019-05-13 stsp if (spawn_editor(a->editor, a->logmsg_path) == -1) {
2572 638f9024 2019-05-13 stsp err = got_error_from_errno("failed spawning editor");
2573 33ad4cbe 2019-05-12 jcs goto done;
2574 33ad4cbe 2019-05-12 jcs }
2575 33ad4cbe 2019-05-12 jcs
2576 e0870e44 2019-05-13 stsp if (stat(a->logmsg_path, &st2) == -1) {
2577 638f9024 2019-05-13 stsp err = got_error_from_errno("stat");
2578 33ad4cbe 2019-05-12 jcs goto done;
2579 33ad4cbe 2019-05-12 jcs }
2580 33ad4cbe 2019-05-12 jcs
2581 33ad4cbe 2019-05-12 jcs if (st.st_mtime == st2.st_mtime && st.st_size == st2.st_size) {
2582 e0870e44 2019-05-13 stsp unlink(a->logmsg_path);
2583 e0870e44 2019-05-13 stsp free(a->logmsg_path);
2584 e0870e44 2019-05-13 stsp a->logmsg_path = NULL;
2585 33ad4cbe 2019-05-12 jcs err = got_error_msg(GOT_ERR_COMMIT_MSG_EMPTY,
2586 33ad4cbe 2019-05-12 jcs "no changes made to commit message, aborting");
2587 33ad4cbe 2019-05-12 jcs goto done;
2588 33ad4cbe 2019-05-12 jcs }
2589 33ad4cbe 2019-05-12 jcs
2590 33ad4cbe 2019-05-12 jcs *logmsg = malloc(st2.st_size + 1);
2591 fcde04d9 2019-05-13 stsp if (*logmsg == NULL) {
2592 638f9024 2019-05-13 stsp err = got_error_from_errno("malloc");
2593 fcde04d9 2019-05-13 stsp goto done;
2594 fcde04d9 2019-05-13 stsp }
2595 cc79381d 2019-05-22 stsp (*logmsg)[0] = '\0';
2596 33ad4cbe 2019-05-12 jcs len = 0;
2597 33ad4cbe 2019-05-12 jcs
2598 e0870e44 2019-05-13 stsp fp = fopen(a->logmsg_path, "r");
2599 d4592c7c 2019-05-22 stsp if (fp == NULL) {
2600 d4592c7c 2019-05-22 stsp err = got_error_from_errno("fopen");
2601 d4592c7c 2019-05-22 stsp goto done;
2602 d4592c7c 2019-05-22 stsp }
2603 33ad4cbe 2019-05-12 jcs while (fgets(buf, sizeof(buf), fp) != NULL) {
2604 e0870e44 2019-05-13 stsp if (!content_changed && strcmp(buf, initial_content) != 0)
2605 e0870e44 2019-05-13 stsp content_changed = 1;
2606 33ad4cbe 2019-05-12 jcs if (buf[0] == '#' || (len == 0 && buf[0] == '\n'))
2607 78527a0a 2019-05-22 stsp continue; /* remove comments and leading empty lines */
2608 33ad4cbe 2019-05-12 jcs len = strlcat(*logmsg, buf, st2.st_size);
2609 33ad4cbe 2019-05-12 jcs }
2610 33ad4cbe 2019-05-12 jcs fclose(fp);
2611 33ad4cbe 2019-05-12 jcs
2612 33ad4cbe 2019-05-12 jcs while (len > 0 && (*logmsg)[len - 1] == '\n') {
2613 33ad4cbe 2019-05-12 jcs (*logmsg)[len - 1] = '\0';
2614 33ad4cbe 2019-05-12 jcs len--;
2615 33ad4cbe 2019-05-12 jcs }
2616 33ad4cbe 2019-05-12 jcs
2617 e0870e44 2019-05-13 stsp if (len == 0 || !content_changed) {
2618 e0870e44 2019-05-13 stsp unlink(a->logmsg_path);
2619 e0870e44 2019-05-13 stsp free(a->logmsg_path);
2620 e0870e44 2019-05-13 stsp a->logmsg_path = NULL;
2621 33ad4cbe 2019-05-12 jcs err = got_error_msg(GOT_ERR_COMMIT_MSG_EMPTY,
2622 33ad4cbe 2019-05-12 jcs "commit message cannot be empty, aborting");
2623 33ad4cbe 2019-05-12 jcs goto done;
2624 33ad4cbe 2019-05-12 jcs }
2625 33ad4cbe 2019-05-12 jcs done:
2626 76d98825 2019-06-03 stsp free(initial_content);
2627 e0870e44 2019-05-13 stsp free(template);
2628 314a6357 2019-05-13 stsp
2629 314a6357 2019-05-13 stsp /* Editor is done; we can now apply unveil(2) */
2630 314a6357 2019-05-13 stsp if (err == NULL)
2631 314a6357 2019-05-13 stsp err = apply_unveil(a->repo_path, 0, a->worktree_path, 0);
2632 33ad4cbe 2019-05-12 jcs return err;
2633 6bad629b 2019-02-04 stsp }
2634 c4296144 2019-05-09 stsp
2635 c4296144 2019-05-09 stsp static const struct got_error *
2636 c4296144 2019-05-09 stsp cmd_commit(int argc, char *argv[])
2637 c4296144 2019-05-09 stsp {
2638 c4296144 2019-05-09 stsp const struct got_error *error = NULL;
2639 c4296144 2019-05-09 stsp struct got_worktree *worktree = NULL;
2640 c4296144 2019-05-09 stsp struct got_repository *repo = NULL;
2641 c4296144 2019-05-09 stsp char *cwd = NULL, *path = NULL, *id_str = NULL;
2642 c4296144 2019-05-09 stsp struct got_object_id *id = NULL;
2643 33ad4cbe 2019-05-12 jcs const char *logmsg = NULL;
2644 35bd8fed 2019-05-09 stsp const char *got_author = getenv("GOT_AUTHOR");
2645 e2ba3d07 2019-05-13 stsp struct collect_commit_logmsg_arg cl_arg;
2646 e2ba3d07 2019-05-13 stsp char *editor = NULL;
2647 c4296144 2019-05-09 stsp int ch;
2648 c4296144 2019-05-09 stsp
2649 c4296144 2019-05-09 stsp while ((ch = getopt(argc, argv, "m:")) != -1) {
2650 c4296144 2019-05-09 stsp switch (ch) {
2651 c4296144 2019-05-09 stsp case 'm':
2652 c4296144 2019-05-09 stsp logmsg = optarg;
2653 c4296144 2019-05-09 stsp break;
2654 c4296144 2019-05-09 stsp default:
2655 c4296144 2019-05-09 stsp usage_commit();
2656 c4296144 2019-05-09 stsp /* NOTREACHED */
2657 c4296144 2019-05-09 stsp }
2658 c4296144 2019-05-09 stsp }
2659 c4296144 2019-05-09 stsp
2660 c4296144 2019-05-09 stsp argc -= optind;
2661 c4296144 2019-05-09 stsp argv += optind;
2662 c4296144 2019-05-09 stsp
2663 c4296144 2019-05-09 stsp if (argc == 1) {
2664 c4296144 2019-05-09 stsp path = realpath(argv[0], NULL);
2665 c4296144 2019-05-09 stsp if (path == NULL) {
2666 638f9024 2019-05-13 stsp error = got_error_from_errno2("realpath", argv[0]);
2667 c4296144 2019-05-09 stsp goto done;
2668 c4296144 2019-05-09 stsp }
2669 7fbaa4f3 2019-05-11 stsp got_path_strip_trailing_slashes(path);
2670 c4296144 2019-05-09 stsp } else if (argc != 0)
2671 c4296144 2019-05-09 stsp usage_commit();
2672 c4296144 2019-05-09 stsp
2673 35bd8fed 2019-05-09 stsp if (got_author == NULL) {
2674 35bd8fed 2019-05-09 stsp /* TODO: Look current user up in password database */
2675 35bd8fed 2019-05-09 stsp error = got_error(GOT_ERR_COMMIT_NO_AUTHOR);
2676 35bd8fed 2019-05-09 stsp goto done;
2677 35bd8fed 2019-05-09 stsp }
2678 c4296144 2019-05-09 stsp
2679 c4296144 2019-05-09 stsp cwd = getcwd(NULL, 0);
2680 c4296144 2019-05-09 stsp if (cwd == NULL) {
2681 638f9024 2019-05-13 stsp error = got_error_from_errno("getcwd");
2682 c4296144 2019-05-09 stsp goto done;
2683 c4296144 2019-05-09 stsp }
2684 c4296144 2019-05-09 stsp error = got_worktree_open(&worktree, cwd);
2685 c4296144 2019-05-09 stsp if (error)
2686 c4296144 2019-05-09 stsp goto done;
2687 c4296144 2019-05-09 stsp
2688 c4296144 2019-05-09 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
2689 c4296144 2019-05-09 stsp if (error != NULL)
2690 c4296144 2019-05-09 stsp goto done;
2691 c4296144 2019-05-09 stsp
2692 314a6357 2019-05-13 stsp /*
2693 314a6357 2019-05-13 stsp * unveil(2) traverses exec(2); if an editor is used we have
2694 314a6357 2019-05-13 stsp * to apply unveil after the log message has been written.
2695 314a6357 2019-05-13 stsp */
2696 314a6357 2019-05-13 stsp if (logmsg == NULL || strlen(logmsg) == 0)
2697 314a6357 2019-05-13 stsp error = get_editor(&editor);
2698 314a6357 2019-05-13 stsp else
2699 314a6357 2019-05-13 stsp error = apply_unveil(got_repo_get_path(repo), 0,
2700 314a6357 2019-05-13 stsp got_worktree_get_root_path(worktree), 0);
2701 c4296144 2019-05-09 stsp if (error)
2702 c4296144 2019-05-09 stsp goto done;
2703 c4296144 2019-05-09 stsp
2704 e2ba3d07 2019-05-13 stsp cl_arg.editor = editor;
2705 e2ba3d07 2019-05-13 stsp cl_arg.cmdline_log = logmsg;
2706 e0870e44 2019-05-13 stsp cl_arg.worktree_path = got_worktree_get_root_path(worktree);
2707 76d98825 2019-06-03 stsp cl_arg.branch_name = got_worktree_get_head_ref_name(worktree);
2708 76d98825 2019-06-03 stsp if (strncmp(cl_arg.branch_name, "refs/", 5) == 0)
2709 76d98825 2019-06-03 stsp cl_arg.branch_name += 5;
2710 76d98825 2019-06-03 stsp if (strncmp(cl_arg.branch_name, "heads/", 6) == 0)
2711 76d98825 2019-06-03 stsp cl_arg.branch_name += 6;
2712 314a6357 2019-05-13 stsp cl_arg.repo_path = got_repo_get_path(repo);
2713 e0870e44 2019-05-13 stsp cl_arg.logmsg_path = NULL;
2714 35bd8fed 2019-05-09 stsp error = got_worktree_commit(&id, worktree, path, got_author, NULL,
2715 e2ba3d07 2019-05-13 stsp collect_commit_logmsg, &cl_arg, print_status, NULL, repo);
2716 e0870e44 2019-05-13 stsp if (error) {
2717 e0870e44 2019-05-13 stsp if (cl_arg.logmsg_path)
2718 e0870e44 2019-05-13 stsp fprintf(stderr, "%s: log message preserved in %s\n",
2719 e0870e44 2019-05-13 stsp getprogname(), cl_arg.logmsg_path);
2720 c4296144 2019-05-09 stsp goto done;
2721 e0870e44 2019-05-13 stsp }
2722 c4296144 2019-05-09 stsp
2723 e0870e44 2019-05-13 stsp if (cl_arg.logmsg_path)
2724 e0870e44 2019-05-13 stsp unlink(cl_arg.logmsg_path);
2725 e0870e44 2019-05-13 stsp
2726 c4296144 2019-05-09 stsp error = got_object_id_str(&id_str, id);
2727 c4296144 2019-05-09 stsp if (error)
2728 c4296144 2019-05-09 stsp goto done;
2729 a7648d7a 2019-06-02 stsp printf("Created commit %s\n", id_str);
2730 c4296144 2019-05-09 stsp done:
2731 c4296144 2019-05-09 stsp if (repo)
2732 c4296144 2019-05-09 stsp got_repo_close(repo);
2733 c4296144 2019-05-09 stsp if (worktree)
2734 c4296144 2019-05-09 stsp got_worktree_close(worktree);
2735 c4296144 2019-05-09 stsp free(path);
2736 c4296144 2019-05-09 stsp free(cwd);
2737 c4296144 2019-05-09 stsp free(id_str);
2738 e2ba3d07 2019-05-13 stsp free(editor);
2739 234035bc 2019-06-01 stsp return error;
2740 234035bc 2019-06-01 stsp }
2741 234035bc 2019-06-01 stsp
2742 234035bc 2019-06-01 stsp __dead static void
2743 234035bc 2019-06-01 stsp usage_cherrypick(void)
2744 234035bc 2019-06-01 stsp {
2745 234035bc 2019-06-01 stsp fprintf(stderr, "usage: %s cherrypick commit-id\n", getprogname());
2746 234035bc 2019-06-01 stsp exit(1);
2747 234035bc 2019-06-01 stsp }
2748 234035bc 2019-06-01 stsp
2749 234035bc 2019-06-01 stsp static const struct got_error *
2750 234035bc 2019-06-01 stsp cmd_cherrypick(int argc, char *argv[])
2751 234035bc 2019-06-01 stsp {
2752 234035bc 2019-06-01 stsp const struct got_error *error = NULL;
2753 234035bc 2019-06-01 stsp struct got_worktree *worktree = NULL;
2754 234035bc 2019-06-01 stsp struct got_repository *repo = NULL;
2755 234035bc 2019-06-01 stsp char *cwd = NULL, *commit_id_str = NULL;
2756 234035bc 2019-06-01 stsp struct got_object_id *commit_id = NULL;
2757 234035bc 2019-06-01 stsp struct got_commit_object *commit = NULL;
2758 234035bc 2019-06-01 stsp struct got_object_qid *pid;
2759 234035bc 2019-06-01 stsp struct got_reference *head_ref = NULL;
2760 234035bc 2019-06-01 stsp int ch, did_something = 0;
2761 234035bc 2019-06-01 stsp
2762 234035bc 2019-06-01 stsp while ((ch = getopt(argc, argv, "")) != -1) {
2763 234035bc 2019-06-01 stsp switch (ch) {
2764 234035bc 2019-06-01 stsp default:
2765 234035bc 2019-06-01 stsp usage_cherrypick();
2766 234035bc 2019-06-01 stsp /* NOTREACHED */
2767 234035bc 2019-06-01 stsp }
2768 234035bc 2019-06-01 stsp }
2769 234035bc 2019-06-01 stsp
2770 234035bc 2019-06-01 stsp argc -= optind;
2771 234035bc 2019-06-01 stsp argv += optind;
2772 234035bc 2019-06-01 stsp
2773 234035bc 2019-06-01 stsp if (argc != 1)
2774 234035bc 2019-06-01 stsp usage_cherrypick();
2775 234035bc 2019-06-01 stsp
2776 234035bc 2019-06-01 stsp cwd = getcwd(NULL, 0);
2777 234035bc 2019-06-01 stsp if (cwd == NULL) {
2778 234035bc 2019-06-01 stsp error = got_error_from_errno("getcwd");
2779 234035bc 2019-06-01 stsp goto done;
2780 234035bc 2019-06-01 stsp }
2781 234035bc 2019-06-01 stsp error = got_worktree_open(&worktree, cwd);
2782 234035bc 2019-06-01 stsp if (error)
2783 234035bc 2019-06-01 stsp goto done;
2784 234035bc 2019-06-01 stsp
2785 234035bc 2019-06-01 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
2786 234035bc 2019-06-01 stsp if (error != NULL)
2787 234035bc 2019-06-01 stsp goto done;
2788 234035bc 2019-06-01 stsp
2789 234035bc 2019-06-01 stsp error = apply_unveil(got_repo_get_path(repo), 0,
2790 234035bc 2019-06-01 stsp got_worktree_get_root_path(worktree), 0);
2791 234035bc 2019-06-01 stsp if (error)
2792 234035bc 2019-06-01 stsp goto done;
2793 234035bc 2019-06-01 stsp
2794 234035bc 2019-06-01 stsp error = got_object_resolve_id_str(&commit_id, repo, argv[0]);
2795 234035bc 2019-06-01 stsp if (error != NULL) {
2796 234035bc 2019-06-01 stsp struct got_reference *ref;
2797 234035bc 2019-06-01 stsp if (error->code != GOT_ERR_BAD_OBJ_ID_STR)
2798 234035bc 2019-06-01 stsp goto done;
2799 234035bc 2019-06-01 stsp error = got_ref_open(&ref, repo, argv[0], 0);
2800 234035bc 2019-06-01 stsp if (error != NULL)
2801 234035bc 2019-06-01 stsp goto done;
2802 234035bc 2019-06-01 stsp error = got_ref_resolve(&commit_id, repo, ref);
2803 234035bc 2019-06-01 stsp got_ref_close(ref);
2804 234035bc 2019-06-01 stsp if (error != NULL)
2805 234035bc 2019-06-01 stsp goto done;
2806 234035bc 2019-06-01 stsp }
2807 234035bc 2019-06-01 stsp error = got_object_id_str(&commit_id_str, commit_id);
2808 234035bc 2019-06-01 stsp if (error)
2809 234035bc 2019-06-01 stsp goto done;
2810 234035bc 2019-06-01 stsp
2811 234035bc 2019-06-01 stsp error = got_ref_open(&head_ref, repo,
2812 234035bc 2019-06-01 stsp got_worktree_get_head_ref_name(worktree), 0);
2813 234035bc 2019-06-01 stsp if (error != NULL)
2814 234035bc 2019-06-01 stsp goto done;
2815 234035bc 2019-06-01 stsp
2816 234035bc 2019-06-01 stsp error = check_same_branch(commit_id, head_ref, repo);
2817 234035bc 2019-06-01 stsp if (error) {
2818 234035bc 2019-06-01 stsp if (error->code != GOT_ERR_ANCESTRY)
2819 234035bc 2019-06-01 stsp goto done;
2820 234035bc 2019-06-01 stsp error = NULL;
2821 234035bc 2019-06-01 stsp } else {
2822 234035bc 2019-06-01 stsp error = got_error(GOT_ERR_SAME_BRANCH);
2823 234035bc 2019-06-01 stsp goto done;
2824 234035bc 2019-06-01 stsp }
2825 234035bc 2019-06-01 stsp
2826 234035bc 2019-06-01 stsp error = got_object_open_as_commit(&commit, repo, commit_id);
2827 234035bc 2019-06-01 stsp if (error)
2828 234035bc 2019-06-01 stsp goto done;
2829 234035bc 2019-06-01 stsp pid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
2830 03415a1a 2019-06-02 stsp error = got_worktree_merge_files(worktree, pid ? pid->id : NULL,
2831 03415a1a 2019-06-02 stsp commit_id, repo, update_progress, &did_something, check_cancelled,
2832 03415a1a 2019-06-02 stsp NULL);
2833 234035bc 2019-06-01 stsp if (error != NULL)
2834 234035bc 2019-06-01 stsp goto done;
2835 234035bc 2019-06-01 stsp
2836 234035bc 2019-06-01 stsp if (did_something)
2837 a7648d7a 2019-06-02 stsp printf("Merged commit %s\n", commit_id_str);
2838 234035bc 2019-06-01 stsp done:
2839 234035bc 2019-06-01 stsp if (commit)
2840 234035bc 2019-06-01 stsp got_object_commit_close(commit);
2841 234035bc 2019-06-01 stsp free(commit_id_str);
2842 234035bc 2019-06-01 stsp if (head_ref)
2843 234035bc 2019-06-01 stsp got_ref_close(head_ref);
2844 234035bc 2019-06-01 stsp if (worktree)
2845 234035bc 2019-06-01 stsp got_worktree_close(worktree);
2846 234035bc 2019-06-01 stsp if (repo)
2847 234035bc 2019-06-01 stsp got_repo_close(repo);
2848 c4296144 2019-05-09 stsp return error;
2849 c4296144 2019-05-09 stsp }
2850 5ef14e63 2019-06-02 stsp
2851 5ef14e63 2019-06-02 stsp __dead static void
2852 5ef14e63 2019-06-02 stsp usage_backout(void)
2853 5ef14e63 2019-06-02 stsp {
2854 5ef14e63 2019-06-02 stsp fprintf(stderr, "usage: %s backout commit-id\n", getprogname());
2855 5ef14e63 2019-06-02 stsp exit(1);
2856 5ef14e63 2019-06-02 stsp }
2857 5ef14e63 2019-06-02 stsp
2858 5ef14e63 2019-06-02 stsp static const struct got_error *
2859 5ef14e63 2019-06-02 stsp cmd_backout(int argc, char *argv[])
2860 5ef14e63 2019-06-02 stsp {
2861 5ef14e63 2019-06-02 stsp const struct got_error *error = NULL;
2862 5ef14e63 2019-06-02 stsp struct got_worktree *worktree = NULL;
2863 5ef14e63 2019-06-02 stsp struct got_repository *repo = NULL;
2864 5ef14e63 2019-06-02 stsp char *cwd = NULL, *commit_id_str = NULL;
2865 5ef14e63 2019-06-02 stsp struct got_object_id *commit_id = NULL;
2866 5ef14e63 2019-06-02 stsp struct got_commit_object *commit = NULL;
2867 5ef14e63 2019-06-02 stsp struct got_object_qid *pid;
2868 5ef14e63 2019-06-02 stsp struct got_reference *head_ref = NULL;
2869 5ef14e63 2019-06-02 stsp int ch, did_something = 0;
2870 5ef14e63 2019-06-02 stsp
2871 5ef14e63 2019-06-02 stsp while ((ch = getopt(argc, argv, "")) != -1) {
2872 5ef14e63 2019-06-02 stsp switch (ch) {
2873 5ef14e63 2019-06-02 stsp default:
2874 5ef14e63 2019-06-02 stsp usage_backout();
2875 5ef14e63 2019-06-02 stsp /* NOTREACHED */
2876 5ef14e63 2019-06-02 stsp }
2877 5ef14e63 2019-06-02 stsp }
2878 5ef14e63 2019-06-02 stsp
2879 5ef14e63 2019-06-02 stsp argc -= optind;
2880 5ef14e63 2019-06-02 stsp argv += optind;
2881 5ef14e63 2019-06-02 stsp
2882 5ef14e63 2019-06-02 stsp if (argc != 1)
2883 5ef14e63 2019-06-02 stsp usage_backout();
2884 5ef14e63 2019-06-02 stsp
2885 5ef14e63 2019-06-02 stsp cwd = getcwd(NULL, 0);
2886 5ef14e63 2019-06-02 stsp if (cwd == NULL) {
2887 5ef14e63 2019-06-02 stsp error = got_error_from_errno("getcwd");
2888 5ef14e63 2019-06-02 stsp goto done;
2889 5ef14e63 2019-06-02 stsp }
2890 5ef14e63 2019-06-02 stsp error = got_worktree_open(&worktree, cwd);
2891 5ef14e63 2019-06-02 stsp if (error)
2892 5ef14e63 2019-06-02 stsp goto done;
2893 5ef14e63 2019-06-02 stsp
2894 5ef14e63 2019-06-02 stsp error = got_repo_open(&repo, got_worktree_get_repo_path(worktree));
2895 5ef14e63 2019-06-02 stsp if (error != NULL)
2896 5ef14e63 2019-06-02 stsp goto done;
2897 5ef14e63 2019-06-02 stsp
2898 5ef14e63 2019-06-02 stsp error = apply_unveil(got_repo_get_path(repo), 0,
2899 5ef14e63 2019-06-02 stsp got_worktree_get_root_path(worktree), 0);
2900 5ef14e63 2019-06-02 stsp if (error)
2901 5ef14e63 2019-06-02 stsp goto done;
2902 5ef14e63 2019-06-02 stsp
2903 5ef14e63 2019-06-02 stsp error = got_object_resolve_id_str(&commit_id, repo, argv[0]);
2904 5ef14e63 2019-06-02 stsp if (error != NULL) {
2905 5ef14e63 2019-06-02 stsp struct got_reference *ref;
2906 5ef14e63 2019-06-02 stsp if (error->code != GOT_ERR_BAD_OBJ_ID_STR)
2907 5ef14e63 2019-06-02 stsp goto done;
2908 5ef14e63 2019-06-02 stsp error = got_ref_open(&ref, repo, argv[0], 0);
2909 5ef14e63 2019-06-02 stsp if (error != NULL)
2910 5ef14e63 2019-06-02 stsp goto done;
2911 5ef14e63 2019-06-02 stsp error = got_ref_resolve(&commit_id, repo, ref);
2912 5ef14e63 2019-06-02 stsp got_ref_close(ref);
2913 5ef14e63 2019-06-02 stsp if (error != NULL)
2914 5ef14e63 2019-06-02 stsp goto done;
2915 5ef14e63 2019-06-02 stsp }
2916 5ef14e63 2019-06-02 stsp error = got_object_id_str(&commit_id_str, commit_id);
2917 5ef14e63 2019-06-02 stsp if (error)
2918 5ef14e63 2019-06-02 stsp goto done;
2919 5ef14e63 2019-06-02 stsp
2920 5ef14e63 2019-06-02 stsp error = got_ref_open(&head_ref, repo,
2921 5ef14e63 2019-06-02 stsp got_worktree_get_head_ref_name(worktree), 0);
2922 5ef14e63 2019-06-02 stsp if (error != NULL)
2923 5ef14e63 2019-06-02 stsp goto done;
2924 5ef14e63 2019-06-02 stsp
2925 5ef14e63 2019-06-02 stsp error = check_same_branch(commit_id, head_ref, repo);
2926 5ef14e63 2019-06-02 stsp if (error)
2927 5ef14e63 2019-06-02 stsp goto done;
2928 5ef14e63 2019-06-02 stsp
2929 5ef14e63 2019-06-02 stsp error = got_object_open_as_commit(&commit, repo, commit_id);
2930 5ef14e63 2019-06-02 stsp if (error)
2931 5ef14e63 2019-06-02 stsp goto done;
2932 5ef14e63 2019-06-02 stsp pid = SIMPLEQ_FIRST(got_object_commit_get_parent_ids(commit));
2933 5ef14e63 2019-06-02 stsp if (pid == NULL) {
2934 5ef14e63 2019-06-02 stsp error = got_error(GOT_ERR_ROOT_COMMIT);
2935 5ef14e63 2019-06-02 stsp goto done;
2936 5ef14e63 2019-06-02 stsp }
2937 5ef14e63 2019-06-02 stsp
2938 5ef14e63 2019-06-02 stsp error = got_worktree_merge_files(worktree, commit_id, pid->id, repo,
2939 5ef14e63 2019-06-02 stsp update_progress, &did_something, check_cancelled, NULL);
2940 5ef14e63 2019-06-02 stsp if (error != NULL)
2941 5ef14e63 2019-06-02 stsp goto done;
2942 5ef14e63 2019-06-02 stsp
2943 5ef14e63 2019-06-02 stsp if (did_something)
2944 a7648d7a 2019-06-02 stsp printf("Backed out commit %s\n", commit_id_str);
2945 5ef14e63 2019-06-02 stsp done:
2946 5ef14e63 2019-06-02 stsp if (commit)
2947 5ef14e63 2019-06-02 stsp got_object_commit_close(commit);
2948 5ef14e63 2019-06-02 stsp free(commit_id_str);
2949 5ef14e63 2019-06-02 stsp if (head_ref)
2950 5ef14e63 2019-06-02 stsp got_ref_close(head_ref);
2951 5ef14e63 2019-06-02 stsp if (worktree)
2952 5ef14e63 2019-06-02 stsp got_worktree_close(worktree);
2953 5ef14e63 2019-06-02 stsp if (repo)
2954 5ef14e63 2019-06-02 stsp got_repo_close(repo);
2955 5ef14e63 2019-06-02 stsp return error;
2956 5ef14e63 2019-06-02 stsp }